goctl added

tools/goctl/util/path.go

@@ -0,0 +1,53 @@
+package util
+
+import (
+	"fmt"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"zero/tools/goctl/vars"
+)
+
+func MkdirIfNotExist(dir string) error {
+	if len(dir) == 0 {
+		return nil
+	}
+
+	if _, err := os.Stat(dir); os.IsNotExist(err) {
+		return os.MkdirAll(dir, os.ModePerm)
+	}
+
+	return nil
+}
+
+func PathFromGoSrc() (string, error) {
+	dir, err := os.Getwd()
+	if err != nil {
+		return "", err
+	}
+
+	gopath := os.Getenv("GOPATH")
+	parent := path.Join(gopath, "src", vars.ProjectName)
+	pos := strings.Index(dir, parent)
+	if pos < 0 {
+		return "", fmt.Errorf("%s is not in GOPATH", dir)
+	}
+
+	// skip slash
+	return dir[len(parent)+1:], nil
+}
+
+func GetParentPackage(dir string) (string, error) {
+	absDir, err := filepath.Abs(dir)
+	if err != nil {
+		return "", err
+	}
+	pos := strings.Index(absDir, vars.ProjectName)
+	if pos < 0 {
+		return "", fmt.Errorf("error dir:[%s],please make sure that your project is in the %s directory", vars.ProjectName, dir)
+	}
+
+	return absDir[pos:], nil
+}