| // 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 mvs implements Minimal Version Selection. |
| // See https://research.swtch.com/vgo3. |
| package mvs |
| |
| import ( |
| "fmt" |
| "sort" |
| |
| "cmd/go/internal/module" |
| ) |
| |
| type Reqs interface { |
| Required(m module.Version) ([]module.Version, error) |
| Max(v1, v2 string) string |
| Latest(path string) (module.Version, error) |
| Previous(m module.Version) (module.Version, error) |
| } |
| |
| type MissingModuleError struct { |
| Module module.Version |
| } |
| |
| func (e *MissingModuleError) Error() string { |
| return fmt.Sprintf("missing module: %v", e.Module) |
| } |
| |
| // BuildList returns the build list for the target module. |
| func BuildList(target module.Version, reqs Reqs) ([]module.Version, error) { |
| return buildList(target, reqs, nil, nil) |
| } |
| |
| func buildList(target module.Version, reqs Reqs, uses map[module.Version][]module.Version, vers map[string]string) ([]module.Version, error) { |
| var ( |
| min = map[string]string{target.Path: target.Version} |
| todo = []module.Version{target} |
| seen = map[module.Version]bool{target: true} |
| ) |
| for len(todo) > 0 { |
| m := todo[len(todo)-1] |
| todo = todo[:len(todo)-1] |
| required, _ := reqs.Required(m) |
| for _, r := range required { |
| if uses != nil { |
| uses[r] = append(uses[r], m) |
| } |
| if !seen[r] { |
| if v, ok := min[r.Path]; !ok { |
| min[r.Path] = r.Version |
| } else if max := reqs.Max(v, r.Version); max != v { |
| min[r.Path] = max |
| } |
| todo = append(todo, r) |
| seen[r] = true |
| } |
| } |
| } |
| |
| if min[target.Path] != target.Version { |
| panic("unbuildable") // TODO |
| } |
| |
| if vers == nil { |
| vers = make(map[string]string) |
| } |
| list := []module.Version{target} |
| for i := 0; i < len(list); i++ { |
| m := list[i] |
| required, err := reqs.Required(m) |
| if err != nil { |
| // TODO: Check error is decent. |
| return nil, err |
| } |
| for _, r := range required { |
| v := min[r.Path] |
| if reqs.Max(v, r.Version) != v { |
| panic("mistake") // TODO |
| } |
| if _, ok := vers[r.Path]; !ok { |
| vers[r.Path] = v |
| list = append(list, module.Version{r.Path, v}) |
| } |
| } |
| } |
| tail := list[1:] |
| sort.Slice(tail, func(i, j int) bool { |
| return tail[i].Path < tail[j].Path |
| }) |
| return list, nil |
| } |
| |
| // Req returns the minimal requirement list for the target module |
| // that result in the given build list. |
| func Req(target module.Version, list []module.Version, reqs Reqs) ([]module.Version, error) { |
| // Compute postorder, cache requirements. |
| var postorder []module.Version |
| reqCache := map[module.Version][]module.Version{} |
| reqCache[target] = nil |
| var walk func(module.Version) error |
| walk = func(m module.Version) error { |
| _, ok := reqCache[m] |
| if ok { |
| return nil |
| } |
| required, err := reqs.Required(m) |
| if err != nil { |
| return err |
| } |
| reqCache[m] = required |
| for _, m1 := range required { |
| if err := walk(m1); err != nil { |
| return err |
| } |
| } |
| postorder = append(postorder, m) |
| return nil |
| } |
| for _, m := range list { |
| if err := walk(m); err != nil { |
| return nil, err |
| } |
| } |
| |
| // Walk modules in reverse post-order, only adding those not implied already. |
| have := map[string]string{} |
| walk = func(m module.Version) error { |
| if v, ok := have[m.Path]; ok && reqs.Max(m.Version, v) == v { |
| return nil |
| } |
| have[m.Path] = m.Version |
| for _, m1 := range reqCache[m] { |
| walk(m1) |
| } |
| return nil |
| } |
| max := map[string]string{} |
| for _, m := range list { |
| if max[m.Path] == "" { |
| max[m.Path] = m.Version |
| } else { |
| max[m.Path] = reqs.Max(m.Version, max[m.Path]) |
| } |
| } |
| var min []module.Version |
| for i := len(postorder) - 1; i >= 0; i-- { |
| m := postorder[i] |
| if max[m.Path] != m.Version { |
| // Older version. |
| continue |
| } |
| if have[m.Path] != m.Version { |
| min = append(min, m) |
| walk(m) |
| } |
| } |
| sort.Slice(min, func(i, j int) bool { |
| return min[i].Path < min[j].Path |
| }) |
| return min, nil |
| } |
| |
| // UpgradeAll returns a build list for the target module |
| // in which every module is upgraded to its latest version. |
| func UpgradeAll(target module.Version, reqs Reqs) ([]module.Version, error) { |
| have := map[string]bool{target.Path: true} |
| list := []module.Version{target} |
| for i := 0; i < len(list); i++ { |
| m := list[i] |
| required, err := reqs.Required(m) |
| if err != nil { |
| panic(err) // TODO |
| } |
| for _, r := range required { |
| latest, err := reqs.Latest(r.Path) |
| if err != nil { |
| panic(err) // TODO |
| } |
| if reqs.Max(latest.Version, r.Version) != latest.Version { |
| panic("mistake") // TODO |
| } |
| if !have[r.Path] { |
| have[r.Path] = true |
| list = append(list, module.Version{r.Path, latest.Version}) |
| } |
| } |
| } |
| tail := list[1:] |
| sort.Slice(tail, func(i, j int) bool { |
| return tail[i].Path < tail[j].Path |
| }) |
| return list, nil |
| } |
| |
| // Upgrade returns a build list for the target module |
| // in which the given additional modules are upgraded. |
| func Upgrade(target module.Version, reqs Reqs, upgrade ...module.Version) ([]module.Version, error) { |
| list, err := reqs.Required(target) |
| if err != nil { |
| panic(err) // TODO |
| } |
| // TODO: Maybe if an error is given, |
| // rerun with BuildList(upgrade[0], reqs) etc |
| // to find which ones are the buggy ones. |
| list = append([]module.Version(nil), list...) |
| list = append(list, upgrade...) |
| return BuildList(target, &override{target, list, reqs}) |
| } |
| |
| // Downgrade returns a build list for the target module |
| // in which the given additional modules are downgraded. |
| func Downgrade(target module.Version, reqs Reqs, downgrade ...module.Version) ([]module.Version, error) { |
| list, err := reqs.Required(target) |
| if err != nil { |
| panic(err) // TODO |
| } |
| max := make(map[string]string) |
| for _, r := range list { |
| max[r.Path] = r.Version |
| } |
| for _, d := range downgrade { |
| if v, ok := max[d.Path]; !ok || reqs.Max(v, d.Version) != d.Version { |
| max[d.Path] = d.Version |
| } |
| } |
| |
| var ( |
| added = make(map[module.Version]bool) |
| rdeps = make(map[module.Version][]module.Version) |
| excluded = make(map[module.Version]bool) |
| ) |
| var exclude func(module.Version) |
| exclude = func(m module.Version) { |
| if excluded[m] { |
| return |
| } |
| excluded[m] = true |
| for _, p := range rdeps[m] { |
| exclude(p) |
| } |
| } |
| var add func(module.Version) |
| add = func(m module.Version) { |
| if added[m] { |
| return |
| } |
| added[m] = true |
| if v, ok := max[m.Path]; ok && reqs.Max(m.Version, v) != v { |
| exclude(m) |
| return |
| } |
| list, err := reqs.Required(m) |
| if err != nil { |
| panic(err) // TODO |
| } |
| for _, r := range list { |
| add(r) |
| if excluded[r] { |
| exclude(m) |
| return |
| } |
| rdeps[r] = append(rdeps[r], m) |
| } |
| } |
| |
| var out []module.Version |
| out = append(out, target) |
| List: |
| for _, r := range list { |
| add(r) |
| for excluded[r] { |
| p, err := reqs.Previous(r) |
| if err != nil { |
| return nil, err // TODO |
| } |
| if p.Version == "none" { |
| continue List |
| } |
| add(p) |
| r = p |
| } |
| out = append(out, r) |
| } |
| |
| return out, nil |
| } |
| |
| type override struct { |
| target module.Version |
| list []module.Version |
| Reqs |
| } |
| |
| func (r *override) Required(m module.Version) ([]module.Version, error) { |
| if m == r.target { |
| return r.list, nil |
| } |
| return r.Reqs.Required(m) |
| } |