support context

main.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	_ "embed"
 	"flag"
 	"fmt"
@@ -61,9 +62,9 @@ func main() {
 
 	nodePath := ""
 	if *onlyPrint {
-		nodePath = node.GetNodePath(ver, nil)
+		nodePath = node.GetNodePath(context.Background(), ver, nil)
 	} else {
-		nodePath = node.GetNodePath(ver, log.New(os.Stdout, "", 0))
+		nodePath = node.GetNodePath(context.Background(), ver, log.New(os.Stdout, "", 0))
 	}
 
 	binPath := node.BinPath(nodePath)

pkg/node/node.go

@@ -1,6 +1,7 @@
 package node
 
 import (
+	"context"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -15,7 +16,7 @@ import (
 var cacheDir = filepath.Join(fetchup.CacheDir(), "use-node")
 var versionsFile = filepath.Join(cacheDir, "versions.json")
 
-func GetNodePath(required string, logger fetchup.Logger) string {
+func GetNodePath(ctx context.Context, required string, logger fetchup.Logger) string {
 	utils.E(os.MkdirAll(cacheDir, 0755))
 
 	n := getNodeInfo(required)
@@ -29,6 +30,7 @@ func GetNodePath(required string, logger fetchup.Logger) string {
 	_ = os.RemoveAll(nodePath)
 
 	fu := fetchup.New(nodePath, n.URLs()...)
+	fu.Ctx = ctx
 
 	if logger == nil {
 		fu.Logger = fetchup.LoggerQuiet

pkg/node/node_test.go

@@ -39,8 +39,8 @@ func TestGetVersion(t *testing.T) {
 func TestGetNode(t *testing.T) {
 	g := got.T(t)
 
-	GetNodePath("", nil)
-	p := GetNodePath("", nil)
+	GetNodePath(g.Context(), "", nil)
+	p := GetNodePath(g.Context(), "", nil)
 
 	g.Has(p, "use-node")
 }