module support

utils/utils.go

@@ -0,0 +1,53 @@
+package utils
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+var SearchPaths []string
+
+func init() {
+	cwd, err := os.Getwd()
+	if err != nil {
+		log.Fatalf("error getting cwd: %s", err)
+	}
+
+	if e := os.Getenv("MONKEYPATH"); e != "" {
+		tokens := strings.Split(e, ":")
+		for _, token := range tokens {
+			AddPath(token) // ignore errors
+		}
+	} else {
+		SearchPaths = append(SearchPaths, cwd)
+	}
+}
+
+func AddPath(path string) error {
+	path = os.ExpandEnv(filepath.Clean(path))
+	absPath, err := filepath.Abs(path)
+	if err != nil {
+		return err
+	}
+	SearchPaths = append(SearchPaths, absPath)
+	return nil
+}
+
+func Exists(path string) bool {
+	_, err := os.Stat(path)
+	return err == nil
+}
+
+func FindModule(name string) string {
+	basename := fmt.Sprintf("%s.monkey", name)
+	for _, p := range SearchPaths {
+		filename := filepath.Join(p, basename)
+		if Exists(filename) {
+			return filename
+		}
+	}
+	return ""
+}