diff --git a/main.go b/main.go index 011e882..5df810c 100644 --- a/main.go +++ b/main.go @@ -16,15 +16,10 @@ func main() { log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile) binDirs := []string{} srcDirs := []string{} - for _, env := range os.Environ() { - if strings.HasPrefix(env, "GOPATH=") { - i := strings.Index(env, "=") - paths := filepath.SplitList(env[i+1:]) - for _, path := range paths { - binDirs = append(binDirs, filepath.Join(path, "bin")) - srcDirs = append(srcDirs, filepath.Join(path, "src")) - } - } + + for _, path := range filepath.SplitList(os.Getenv("GOPATH")) { + binDirs = append(binDirs, filepath.Join(path, "bin")) + srcDirs = append(srcDirs, filepath.Join(path, "src")) } files := []string{}