Set NODE_PATH environment variable

flimzy · flimzy · commit 9e8fb4ad55f7 · 2015-10-19T17:19:45.000+02:00
diff --git a/tool.go b/tool.go
@@ -658,13 +658,27 @@ func runNode(script string, args []string, dir string) error {
 	node.Stdin = os.Stdin
 	node.Stdout = os.Stdout
 	node.Stderr = os.Stderr
+	node.Env = setNodeEnv(dir)
 	err := node.Run()
 	if _, ok := err.(*exec.ExitError); err != nil && !ok {
 		err = fmt.Errorf("could not run Node.js: %s", err.Error())
 	}
 	return err
 }
 
+func setNodeEnv(dir string) []string {
+    nodePath := dir + "/node_modules"
+
+    env := os.Environ()
+    for i,e := range env {
+        if strings.HasPrefix(e, "NODE_PATH=") {
+            env[i] = e + ":" + nodePath
+            return env
+        }
+    }
+    return append(env, "NODE_PATH=" + nodePath)
+}
+
 type testFuncs struct {
 	Tests      []testFunc
 	Benchmarks []testFunc
