blob: 1b14e4b02634b908c803d2d739a9285829276f7a [file] [log] [blame]
// Copyright 2018 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package vgo
import (
"bytes"
"encoding/json"
"fmt"
"go/build"
"io/ioutil"
"os"
"path/filepath"
"sort"
"strings"
"cmd/go/internal/base"
"cmd/go/internal/cfg"
"cmd/go/internal/imports"
"cmd/go/internal/modconv"
"cmd/go/internal/modfetch"
"cmd/go/internal/modfile"
"cmd/go/internal/module"
"cmd/go/internal/mvs"
"cmd/go/internal/search"
"cmd/go/internal/semver"
)
type importLevel int
const (
levelNone importLevel = 0
levelBuild importLevel = 1
levelTest importLevel = 2
levelTestRecursive importLevel = 3
)
var (
buildList []module.Version
tags map[string]bool
importmap map[string]string
pkgdir map[string]string
pkgmod map[string]module.Version
isGetU bool
)
func AddImports(gofiles []string) {
if Init(); !Enabled() {
return
}
InitMod()
imports, testImports, err := imports.ScanFiles(gofiles, tags)
if err != nil {
base.Fatalf("vgo: %v", err)
}
iterate(func(ld *loader) {
ld.importList(imports, levelBuild)
ld.importList(testImports, levelBuild)
})
writeGoMod()
}
func ImportPaths(args []string) []string {
if Init(); !Enabled() {
return search.ImportPaths(args)
}
InitMod()
paths := importPaths(args)
writeGoMod()
return paths
}
func importPaths(args []string) []string {
level := levelBuild
switch cfg.CmdName {
case "test", "vet":
level = levelTest
}
cleaned := search.CleanImportPaths(args)
iterate(func(ld *loader) {
args = expandImportPaths(cleaned)
for i, pkg := range args {
if pkg == "." || pkg == ".." || strings.HasPrefix(pkg, "./") || strings.HasPrefix(pkg, "../") {
dir := filepath.Join(cwd, pkg)
if dir == ModRoot {
pkg = Target.Path
} else if strings.HasPrefix(dir, ModRoot+string(filepath.Separator)) {
pkg = Target.Path + filepath.ToSlash(dir[len(ModRoot):])
} else {
base.Errorf("vgo: package %s outside module root", pkg)
continue
}
args[i] = pkg
}
ld.importPkg(pkg, level)
}
})
return args
}
func Lookup(parentPath, path string) (dir, realPath string, err error) {
realPath = importmap[path]
if realPath == "" {
if isStandardImportPath(path) {
dir := filepath.Join(cfg.GOROOT, "src", path)
if _, err := os.Stat(dir); err == nil {
return dir, path, nil
}
}
return "", "", fmt.Errorf("no such package in module")
}
return pkgdir[realPath], realPath, nil
}
func iterate(doImports func(*loader)) {
var err error
mvsOp := mvs.BuildList
if isGetU {
mvsOp = mvs.UpgradeAll
}
buildList, err = mvsOp(Target, newReqs())
if err != nil {
base.Fatalf("vgo: %v", err)
}
var ld *loader
for {
ld = newLoader()
doImports(ld)
if len(ld.missing) == 0 {
break
}
for _, m := range ld.missing {
findMissing(m)
}
base.ExitIfErrors()
buildList, err = mvsOp(Target, newReqs())
if err != nil {
base.Fatalf("vgo: %v", err)
}
}
base.ExitIfErrors()
importmap = ld.importmap
pkgdir = ld.pkgdir
pkgmod = ld.pkgmod
}
type loader struct {
imported map[string]importLevel
importmap map[string]string
pkgdir map[string]string
pkgmod map[string]module.Version
tags map[string]bool
missing []missing
imports []string
stack []string
}
type missing struct {
path string
stack string
}
func newLoader() *loader {
ld := &loader{
imported: make(map[string]importLevel),
importmap: make(map[string]string),
pkgdir: make(map[string]string),
pkgmod: make(map[string]module.Version),
tags: imports.Tags(),
}
ld.imported["C"] = 100
return ld
}
func (ld *loader) stackText() string {
var buf bytes.Buffer
for _, p := range ld.stack[:len(ld.stack)-1] {
fmt.Fprintf(&buf, "import %q ->\n\t", p)
}
fmt.Fprintf(&buf, "import %q", ld.stack[len(ld.stack)-1])
return buf.String()
}
func (ld *loader) importList(pkgs []string, level importLevel) {
for _, pkg := range pkgs {
ld.importPkg(pkg, level)
}
}
func (ld *loader) importPkg(path string, level importLevel) {
if ld.imported[path] >= level {
return
}
ld.stack = append(ld.stack, path)
defer func() {
ld.stack = ld.stack[:len(ld.stack)-1]
}()
// Any rewritings go here.
realPath := path
ld.imported[path] = level
ld.importmap[path] = realPath
if realPath != path && ld.imported[realPath] >= level {
// Already handled.
return
}
dir := ld.importDir(realPath)
if dir == "" {
return
}
ld.pkgdir[realPath] = dir
imports, testImports, err := imports.ScanDir(dir, ld.tags)
if err != nil {
base.Errorf("vgo: %s [%s]: %v", ld.stackText(), dir, err)
return
}
nextLevel := level
if level == levelTest {
nextLevel = levelBuild
}
for _, pkg := range imports {
ld.importPkg(pkg, nextLevel)
}
if level >= levelTest {
for _, pkg := range testImports {
ld.importPkg(pkg, nextLevel)
}
}
}
func (ld *loader) importDir(path string) string {
if importPathInModule(path, Target.Path) {
dir := ModRoot
if len(path) > len(Target.Path) {
dir = filepath.Join(dir, path[len(Target.Path)+1:])
}
ld.pkgmod[path] = Target
return dir
}
i := strings.Index(path, "/")
if i < 0 || !strings.Contains(path[:i], ".") {
if strings.HasPrefix(path, "golang_org/") {
return filepath.Join(cfg.GOROOT, "src/vendor", path)
}
dir := filepath.Join(cfg.GOROOT, "src", path)
if _, err := os.Stat(dir); err == nil {
return dir
}
}
var mod1 module.Version
var dir1 string
for _, mod := range buildList {
if !importPathInModule(path, mod.Path) {
continue
}
dir, err := fetch(mod)
if err != nil {
base.Errorf("vgo: %s: %v", ld.stackText(), err)
return ""
}
if len(path) > len(mod.Path) {
dir = filepath.Join(dir, path[len(mod.Path)+1:])
}
if dir1 != "" {
base.Errorf("vgo: %s: found in both %v %v and %v %v", ld.stackText(),
mod1.Path, mod1.Version, mod.Path, mod.Version)
return ""
}
dir1 = dir
mod1 = mod
}
if dir1 != "" {
ld.pkgmod[path] = mod1
return dir1
}
ld.missing = append(ld.missing, missing{path, ld.stackText()})
return ""
}
func replaced(mod module.Version) *modfile.Replace {
var found *modfile.Replace
for _, r := range modFile.Replace {
if r.Old == mod {
found = r // keep going
}
}
return found
}
func importPathInModule(path, mpath string) bool {
return mpath == path ||
len(path) > len(mpath) && path[len(mpath)] == '/' && path[:len(mpath)] == mpath
}
var found = make(map[string]bool)
func findMissing(m missing) {
for _, mod := range buildList {
if importPathInModule(m.path, mod.Path) {
// Leave for ordinary build to complain about the missing import.
return
}
}
if build.IsLocalImport(m.path) {
base.Errorf("vgo: relative import is not supported: %s", m.path)
return
}
fmt.Fprintf(os.Stderr, "vgo: resolving import %q\n", m.path)
repo, info, err := modfetch.Import(m.path, allowed)
if err != nil {
base.Errorf("vgo: %s: %v", m.stack, err)
return
}
root := repo.ModulePath()
fmt.Fprintf(os.Stderr, "vgo: finding %s (latest)\n", root)
if found[root] {
base.Fatalf("internal error: findmissing loop on %s", root)
}
found[root] = true
fmt.Fprintf(os.Stderr, "vgo: adding %s %s\n", root, info.Version)
buildList = append(buildList, module.Version{root, info.Version})
modFile.AddRequire(root, info.Version)
}
type mvsReqs struct {
extra []module.Version
}
func newReqs(extra ...module.Version) *mvsReqs {
r := &mvsReqs{
extra: extra,
}
return r
}
func (r *mvsReqs) Required(mod module.Version) ([]module.Version, error) {
list, err := r.required(mod)
if err != nil {
return nil, err
}
if *getU {
for i := range list {
list[i].Version = "none"
}
return list, nil
}
for i, mv := range list {
for excluded[mv] {
mv1, err := r.Next(mv)
if err != nil {
return nil, err
}
if mv1.Version == "" {
return nil, fmt.Errorf("%s(%s) depends on excluded %s(%s) with no newer version available", mod.Path, mod.Version, mv.Path, mv.Version)
}
mv = mv1
}
list[i] = mv
}
return list, nil
}
var vgoVersion = []byte(modconv.Prefix)
func (r *mvsReqs) required(mod module.Version) ([]module.Version, error) {
if mod == Target {
var list []module.Version
if buildList != nil {
list = append(list, buildList[1:]...)
return list, nil
}
for _, r := range modFile.Require {
list = append(list, r.Mod)
}
list = append(list, r.extra...)
return list, nil
}
origPath := mod.Path
if repl := replaced(mod); repl != nil {
if repl.New.Version == "" {
// TODO: need to slip the new version into the tags list etc.
dir := repl.New.Path
if !filepath.IsAbs(dir) {
dir = filepath.Join(ModRoot, dir)
}
gomod := filepath.Join(dir, "go.mod")
data, err := ioutil.ReadFile(gomod)
if err != nil {
return nil, err
}
f, err := modfile.Parse(gomod, data, nil)
if err != nil {
return nil, err
}
var list []module.Version
for _, r := range f.Require {
list = append(list, r.Mod)
}
return list, nil
}
mod = repl.New
}
if mod.Version == "none" {
return nil, nil
}
if !semver.IsValid(mod.Version) {
// Disallow the broader queries supported by fetch.Lookup.
panic(fmt.Errorf("invalid semantic version %q for %s", mod.Version, mod.Path))
return nil, fmt.Errorf("invalid semantic version %q", mod.Version)
}
gomod := filepath.Join(srcV, "cache", mod.Path, "@v", mod.Version+".mod")
infofile := filepath.Join(srcV, "cache", mod.Path, "@v", mod.Version+".info")
var f *modfile.File
if data, err := ioutil.ReadFile(gomod); err == nil {
// If go.mod has a //vgo comment at the start,
// it was auto-converted from a legacy lock file.
// The auto-conversion details may have bugs and
// may be fixed in newer versions of vgo.
// We ignore cached go.mod files if they do not match
// our own vgoVersion.
if !bytes.HasPrefix(data, vgoVersion[:len("//vgo")]) || bytes.HasPrefix(data, vgoVersion) {
f, err := modfile.Parse(gomod, data, nil)
if err != nil {
return nil, err
}
var list []module.Version
for _, r := range f.Require {
list = append(list, r.Mod)
}
return list, nil
}
f, err = modfile.Parse("go.mod", data, nil)
if err != nil {
return nil, fmt.Errorf("parsing downloaded go.mod: %v", err)
}
} else {
if !quietLookup {
fmt.Fprintf(os.Stderr, "vgo: finding %s %s\n", mod.Path, mod.Version)
}
repo, err := modfetch.Lookup(mod.Path)
if err != nil {
base.Errorf("vgo: %s: %v\n", mod.Path, err)
return nil, err
}
info, err := repo.Stat(mod.Version)
if err != nil {
base.Errorf("vgo: %s %s: %v\n", mod.Path, mod.Version, err)
return nil, err
}
data, err := repo.GoMod(info.Version)
if err != nil {
base.Errorf("vgo: %s %s: %v\n", mod.Path, mod.Version, err)
return nil, err
}
f, err = modfile.Parse("go.mod", data, nil)
if err != nil {
return nil, fmt.Errorf("parsing downloaded go.mod: %v", err)
}
dir := filepath.Dir(gomod)
if err := os.MkdirAll(dir, 0777); err != nil {
return nil, fmt.Errorf("caching go.mod: %v", err)
}
js, err := json.Marshal(info)
if err != nil {
return nil, fmt.Errorf("internal error: json failure: %v", err)
}
if err := ioutil.WriteFile(infofile, js, 0666); err != nil {
return nil, fmt.Errorf("caching info: %v", err)
}
if err := ioutil.WriteFile(gomod, data, 0666); err != nil {
return nil, fmt.Errorf("caching go.mod: %v", err)
}
}
if mpath := f.Module.Mod.Path; mpath != origPath && mpath != mod.Path {
return nil, fmt.Errorf("downloaded %q and got module %q", mod.Path, mpath)
}
var list []module.Version
for _, req := range f.Require {
list = append(list, req.Mod)
}
if false {
fmt.Fprintf(os.Stderr, "REQLIST %v:\n", mod)
for _, req := range list {
fmt.Fprintf(os.Stderr, "\t%v\n", req)
}
}
return list, nil
}
var quietLookup bool
func (*mvsReqs) Max(v1, v2 string) string {
if semver.Compare(v1, v2) == -1 {
return v2
}
return v1
}
func (*mvsReqs) Latest(path string) (module.Version, error) {
// Note that query "latest" is not the same as
// using repo.Latest.
// The query only falls back to untagged versions
// if nothing is tagged. The Latest method
// only ever returns untagged versions,
// which is not what we want.
fmt.Fprintf(os.Stderr, "vgo: finding %s latest\n", path)
info, err := modfetch.Query(path, "latest", allowed)
if err != nil {
return module.Version{}, err
}
return module.Version{Path: path, Version: info.Version}, nil
}
var versionCache = make(map[string][]string)
func versions(path string) ([]string, error) {
list, ok := versionCache[path]
if !ok {
var err error
repo, err := modfetch.Lookup(path)
if err != nil {
return nil, err
}
list, err = repo.Versions("")
if err != nil {
return nil, err
}
versionCache[path] = list
}
return list, nil
}
func (*mvsReqs) Previous(m module.Version) (module.Version, error) {
list, err := versions(m.Path)
if err != nil {
return module.Version{}, err
}
i := sort.Search(len(list), func(i int) bool { return semver.Compare(list[i], m.Version) >= 0 })
if i > 0 {
return module.Version{Path: m.Path, Version: list[i-1]}, nil
}
return module.Version{Path: m.Path, Version: "none"}, nil
}
func (*mvsReqs) Next(m module.Version) (module.Version, error) {
list, err := versions(m.Path)
if err != nil {
return module.Version{}, err
}
i := sort.Search(len(list), func(i int) bool { return semver.Compare(list[i], m.Version) > 0 })
if i < len(list) {
return module.Version{Path: m.Path, Version: list[i]}, nil
}
return module.Version{Path: m.Path, Version: "none"}, nil
}