456c883987
1. 去掉全局的配置 config 2. 合并 pkg/build, pkg/flag, pkg/cover, pkg/config 几个包（这几个包有强相关性，适合放一处。并且分开也容易造成循环依赖）
139 lines
3.1 KiB
Go
139 lines
3.1 KiB
Go
package build
|
||
|
||
import (
|
||
"os"
|
||
"os/exec"
|
||
"strings"
|
||
|
||
"github.com/qiniu/goc/v2/pkg/log"
|
||
"github.com/spf13/pflag"
|
||
)
|
||
|
||
// Build struct a build
|
||
type Build struct {
|
||
Args []string // all goc + go command line args + flags
|
||
FlagSets *pflag.FlagSet
|
||
BuildType int
|
||
|
||
Debug bool
|
||
Host string
|
||
Mode string // cover mode
|
||
|
||
GOPATH string
|
||
GOBIN string
|
||
CurWd string
|
||
TmpWd string
|
||
CurModProjectDir string
|
||
TmpModProjectDir string
|
||
|
||
Goflags []string // go command line flags
|
||
GoArgs []string // go command line args
|
||
Packages []string // go command line [packages]
|
||
|
||
ImportPath string // the whole import path of the project
|
||
Pkgs map[string]*Package
|
||
GlobalCoverVarImportPath string
|
||
GlobalCoverVarImportPathDir string
|
||
}
|
||
|
||
// NewBuild creates a Build struct
|
||
//
|
||
func NewBuild(opts ...GocOption) *Build {
|
||
b := &Build{}
|
||
|
||
for _, opt := range opts {
|
||
opt(b)
|
||
}
|
||
|
||
// 1. 解析 goc 命令行和 go 命令行
|
||
b.buildCmdArgsParse()
|
||
// 2. 解析 go 包位置
|
||
b.getPackagesDir()
|
||
// 3. 读取工程元信息：go.mod, pkgs list ...
|
||
b.readProjectMetaInfo()
|
||
// 4. 展示元信息
|
||
b.displayProjectMetaInfo()
|
||
|
||
return b
|
||
}
|
||
|
||
// Build starts go build
|
||
//
|
||
// 1. copy project to temp,
|
||
// 2. inject cover variables and functions into the project,
|
||
// 3. build the project in temp.
|
||
func (b *Build) Build() {
|
||
// 1. 拷贝至临时目录
|
||
b.copyProjectToTmp()
|
||
defer b.clean()
|
||
|
||
log.Donef("project copied to temporary directory")
|
||
|
||
// 2. update go.mod file if needed
|
||
b.updateGoModFile()
|
||
// 3. inject cover vars
|
||
b.Inject()
|
||
// 4. build in the temp project
|
||
b.doBuildInTemp()
|
||
}
|
||
|
||
func (b *Build) doBuildInTemp() {
|
||
log.StartWait("building the injected project")
|
||
|
||
goflags := b.Goflags
|
||
// 检查用户是否自定义了 -o
|
||
oSet := false
|
||
for _, flag := range goflags {
|
||
if flag == "-o" {
|
||
oSet = true
|
||
}
|
||
}
|
||
|
||
// 如果没被设置就加一个至原命令执行的目录
|
||
if !oSet {
|
||
goflags = append(goflags, "-o", b.CurWd)
|
||
}
|
||
|
||
pacakges := b.Packages
|
||
|
||
goflags = append(goflags, pacakges...)
|
||
|
||
args := []string{"build"}
|
||
args = append(args, goflags...)
|
||
// go 命令行由 go build [-o output] [build flags] [packages] 组成
|
||
cmd := exec.Command("go", args...)
|
||
cmd.Dir = b.TmpWd
|
||
cmd.Stdout = os.Stdout
|
||
cmd.Stderr = os.Stderr
|
||
|
||
log.Infof("go build cmd is: %v, in path [%v]", nicePrintArgs(cmd.Args), cmd.Dir)
|
||
if err := cmd.Start(); err != nil {
|
||
log.Fatalf("fail to execute go build: %v", err)
|
||
}
|
||
if err := cmd.Wait(); err != nil {
|
||
log.Fatalf("fail to execute go build: %v", err)
|
||
}
|
||
|
||
// done
|
||
log.StopWait()
|
||
log.Donef("go build done")
|
||
}
|
||
|
||
// nicePrintArgs 优化 args 打印内容
|
||
//
|
||
// 假如：go build -ldflags "-X my/package/config.Version=1.0.0" -o /home/lyy/gitdown/gin-test/cmd .
|
||
//
|
||
// 实际输出会变为：go build -ldflags -X my/package/config.Version=1.0.0 -o /home/lyy/gitdown/gin-test/cmd .
|
||
func nicePrintArgs(args []string) []string {
|
||
output := make([]string, 0)
|
||
for _, arg := range args {
|
||
if strings.Contains(arg, " ") {
|
||
output = append(output, "\""+arg+"\"")
|
||
} else {
|
||
output = append(output, arg)
|
||
}
|
||
}
|
||
|
||
return output
|
||
}
|