go-zero/tools/goctl/rpc/ctx/project.go

package ctx

import (
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"path/filepath"
	"regexp"
	"runtime"
	"strings"

	"github.com/tal-tech/go-zero/tools/goctl/rpc/execx"
	"github.com/tal-tech/go-zero/tools/goctl/util"
	"github.com/tal-tech/go-zero/tools/goctl/util/console"
)

var errProtobufNotFound = errors.New("github.com/golang/protobuf is not found,please ensure you has already [go get github.com/golang/protobuf]")

const (
	constGo          = "go"
	constProtoC      = "protoc"
	constGoModOn     = "go env GO111MODULE"
	constGoMod       = "go env GOMOD"
	constGoModCache  = "go env GOMODCACHE"
	constGoPath      = "go env GOPATH"
	constProtoCGenGo = "protoc-gen-go"
)

type (
	Project struct {
		Path     string
		Name     string
		GoPath   string
		Protobuf Protobuf
		GoMod    GoMod
	}

	GoMod struct {
		ModOn      bool
		GoModCache string
		GoMod      string
		Module     string
	}
	Protobuf struct {
		Path string
	}
)

func prepare(log console.Console) (*Project, error) {
	log.Info("checking go env...")
	_, err := exec.LookPath(constGo)
	if err != nil {
		return nil, err
	}

	_, err = exec.LookPath(constProtoC)
	if err != nil {
		return nil, err
	}

	var (
		goModOn                   bool
		goMod, goModCache, module string
		goPath                    string
		name, path                string
		protobufModule            string
	)
	ret, err := execx.Run(constGoModOn)
	if err != nil {
		return nil, err
	}

	goModOn = strings.TrimSpace(ret) == "on"
	ret, err = execx.Run(constGoMod)
	if err != nil {
		return nil, err
	}

	goMod = strings.TrimSpace(ret)
	ret, err = execx.Run(constGoModCache)
	if err != nil {
		return nil, err
	}

	goModCache = strings.TrimSpace(ret)
	ret, err = execx.Run(constGoPath)
	if err != nil {
		return nil, err
	}

	goPath = strings.TrimSpace(ret)
	src := filepath.Join(goPath, "src")
	if len(goMod) > 0 {
		if goModCache == "" {
			goModCache = filepath.Join(goPath, "pkg", "mod")
		}
		path = filepath.Dir(goMod)
		name = filepath.Base(path)
		data, err := ioutil.ReadFile(goMod)
		if err != nil {
			return nil, err
		}

		module, err = matchModule(data)
		if err != nil {
			return nil, err
		}
	} else {
		if goModCache == "" {
			goModCache = src
		}
		pwd, err := os.Getwd()
		if err != nil {
			return nil, err
		}

		if !strings.HasPrefix(pwd, src) {
			return nil, fmt.Errorf("%s: project is not in go mod and go path", pwd)
		}
		r := strings.TrimPrefix(pwd, src+string(filepath.Separator))
		name = filepath.Dir(r)
		if name == "." {
			name = r
		}
		path = filepath.Join(src, name)
		module = name
	}

	protobuf := filepath.Join(goModCache, protobufModule)
	if !util.FileExists(protobuf) {
		return nil, fmt.Errorf("expected protobuf module in path: %s,please ensure you has already [go get github.com/golang/protobuf]", protobuf)
	}

	var protoCGenGoFilename string
	os := runtime.GOOS
	switch os {
	case "darwin":
		protoCGenGoFilename = filepath.Join(goPath, "bin", "protoc-gen-go")
	case "windows":
		protoCGenGoFilename = filepath.Join(goPath, "bin", "protoc-gen-go.exe")
	default:
		return nil, fmt.Errorf("unexpeted os: %s", os)
	}

	if !util.FileExists(protoCGenGoFilename) {
		sh := "go install " + filepath.Join(protobuf, constProtoCGenGo)
		log.Warning(sh)
		stdout, err := execx.Run(sh)
		if err != nil {
			return nil, err
		}

		log.Info(stdout)
	}
	if !util.FileExists(protoCGenGoFilename) {
		return nil, fmt.Errorf("protoc-gen-go is not found")
	}
	return &Project{
		Name:   name,
		Path:   path,
		GoPath: goPath,
		Protobuf: Protobuf{
			Path: protobuf,
		},
		GoMod: GoMod{
			ModOn:      goModOn,
			GoModCache: goModCache,
			GoMod:      goMod,
			Module:     module,
		},
	}, nil
}

func matchModule(data []byte) (string, error) {
	text := string(data)
	re := regexp.MustCompile(`(?m)^\s*module\s+[a-z0-9/\-.]+$`)
	matches := re.FindAllString(text, -1)
	if len(matches) == 1 {
		target := matches[0]
		index := strings.Index(target, "module")
		return strings.TrimSpace(target[index+6:]), nil
	}

	return "", nil
}