-
Notifications
You must be signed in to change notification settings - Fork 546
/
tidy.go
93 lines (80 loc) · 2.12 KB
/
tidy.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
* Copyright (c) 2021 The GoPlus Authors (goplus.org). All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package mod
import (
"fmt"
"log"
"os"
"os/exec"
"syscall"
"github.com/goplus/gop"
"github.com/goplus/gop/cmd/internal/base"
"github.com/goplus/gop/x/gopenv"
"github.com/goplus/mod"
"github.com/goplus/mod/gopmod"
"github.com/goplus/mod/modfetch"
)
// gop mod tidy
var cmdTidy = &base.Command{
UsageLine: "gop mod tidy [-e -v]",
Short: "add missing and remove unused modules",
}
func init() {
cmdTidy.Run = runTidy
}
func runTidy(cmd *base.Command, args []string) {
mod, err := gopmod.Load(".", mod.GopModOnly)
if err != nil {
if err == syscall.ENOENT {
fmt.Fprintln(os.Stderr, "gop.mod not found")
os.Exit(1)
}
log.Panicln(err)
}
depMods, err := gop.GenDepMods(mod, mod.Root(), true)
check(err)
tidy(mod, depMods)
}
func tidy(mod *gopmod.Module, depMods map[string]struct{}) {
old := mod.DepMods()
for modPath := range old {
if _, ok := depMods[modPath]; !ok { // removed
mod.DropRequire(modPath)
}
}
for modPath := range depMods {
if _, ok := old[modPath]; !ok { // added
if newMod, err := modfetch.Get(modPath); err != nil {
log.Fatalln(err)
} else {
mod.AddRequire(newMod.Path, newMod.Version)
}
}
}
err := mod.Save()
check(err)
_, _, err = gop.GenGo(mod.Root()+"/...", &gop.Config{DontUpdateGoMod: true})
check(err)
err = mod.UpdateGoMod(gopenv.Get(), true)
check(err)
cmd := exec.Command("go", "mod", "tidy")
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
cmd.Stdin = os.Stdin
cmd.Dir = mod.Root()
err = cmd.Run()
check(err)
}