57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package cover
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"errors"
|
||
"io"
|
||
"os/exec"
|
||
|
||
"github.com/qiniu/goc/v2/pkg/config"
|
||
"github.com/qiniu/goc/v2/pkg/log"
|
||
)
|
||
|
||
var (
|
||
// ErrCoverPkgFailed represents the error that fails to inject the package
|
||
ErrCoverPkgFailed = errors.New("fail to inject code to project")
|
||
// ErrCoverListFailed represents the error that fails to list package dependencies
|
||
ErrCoverListFailed = errors.New("fail to list package dependencies")
|
||
)
|
||
|
||
// ListPackages list all packages under specific via go list command.
|
||
func ListPackages(dir string) map[string]*config.Package {
|
||
cmd := exec.Command("go", "list", "-json", "./...")
|
||
cmd.Dir = dir
|
||
|
||
var errBuf bytes.Buffer
|
||
cmd.Stderr = &errBuf
|
||
out, err := cmd.Output()
|
||
if err != nil {
|
||
log.Fatalf("execute go list -json failed, err: %v, stdout: %v, stderr: %v", err, string(out), errBuf.String())
|
||
}
|
||
// 有些时候 go 命令会打印一些信息到 stderr，但其实命令整体是成功运行了
|
||
if errBuf.String() != "" {
|
||
log.Errorf("%v", errBuf.String())
|
||
}
|
||
|
||
dec := json.NewDecoder(bytes.NewBuffer(out))
|
||
pkgs := make(map[string]*config.Package, 0)
|
||
|
||
for {
|
||
var pkg config.Package
|
||
if err := dec.Decode(&pkg); err != nil {
|
||
if err == io.EOF {
|
||
break
|
||
}
|
||
log.Fatalf("reading go list output error: %v", err)
|
||
}
|
||
if pkg.Error != nil {
|
||
log.Fatalf("list package %s failed with output: %v", pkg.ImportPath, pkg.Error)
|
||
}
|
||
|
||
pkgs[pkg.ImportPath] = &pkg
|
||
}
|
||
|
||
return pkgs
|
||
}
|