Merge pull request #16 from gabeguz/master

Don't exit if $GOPATH/src is a symlink
kujtimiihoxha · Apr 1, 2019 · 543fc76 · 543fc76
2 parents 37c59ca + 95f98a6
commit 543fc76
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -3,6 +3,7 @@ package main
 import (
 	"os"
 	"path"
+	"path/filepath"
 	"runtime"
 	"strings"
 
@@ -22,6 +23,16 @@ func main() {
 		logrus.Error(err)
 		return
 	}
+	gosrc, err = filepath.EvalSymlinks(gosrc)
+	if err != nil {
+		logrus.Error(err)
+		return
+	}
+	pwd, err = filepath.EvalSymlinks(pwd)
+	if err != nil {
+		logrus.Error(err)
+		return
+	}
 	if !strings.HasPrefix(pwd, gosrc) {
 		logrus.Error("The project must be in the $GOPATH/src folder for the generator to work.")
 		return