main: use GOPATH from the environment

Be more compatible with the Go toolchain by setting GOPATH in the same
way. This makes it possible to flash and run examples from the standard
GOPATH instead of only from the source tree.

1 files changed, 21 insertions, 0 deletions

diff --git a/target.go b/target.go
index 43a9700d5..8f7a44cc2 100644
--- a/target.go
+++ b/target.go
@@ -58,3 +58,24 @@ func sourceDir() string {
 	_, path, _, _ := runtime.Caller(0)
 	return filepath.Dir(path)
 }
+
+func getGopath() string {
+	gopath := os.Getenv("GOPATH")
+	if gopath != "" {
+		return gopath
+	}
+
+	// fallback
+	var home string
+	if runtime.GOOS == "windows" {
+		home = os.Getenv("USERPROFILE")
+	} else {
+		home = os.Getenv("HOME")
+	}
+	if home == "" {
+		// This is very unlikely, so panic here.
+		// Not the nicest solution, however.
+		panic("no $HOME or %USERPROFILE% found")
+	}
+	return filepath.Join(home, "go")
+}