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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
package main
import (
"errors"
"os"
"path/filepath"
"strings"
"go.wit.com/dev/alexflint/arg"
"go.wit.com/lib/protobuf/forgepb"
"go.wit.com/lib/protobuf/gitpb"
"go.wit.com/log"
)
// sent via -ldflags
var VERSION string
var BUILDTIME string
var pp *arg.Parser
var forge *forgepb.Forge
// var check *gitpb.Repo
var configSave bool
func main() {
var check *gitpb.Repo
log.Info("go-mod-clean version", VERSION, "built on", BUILDTIME)
pp = arg.MustParse(&argv)
// load the ~/.config/forge/ config
// this lets you configure repos you have read/write access too
forge = forgepb.Init()
if argv.All {
// run this on every single repo
// do this before publishing new golang versions
all := forge.Repos.SortByFullPath()
for all.Scan() {
check = all.Next()
if err := doMain(check); err != nil {
badExit(check, err)
}
}
} else {
// figure out what directory we are running in
check = findPwdRepo()
if check == nil {
log.Info("this directory isn't in a golang project (not in ~/go/src nor a go.work file)")
badExit(nil, nil)
}
if err := doMain(check); err != nil {
badExit(check, err)
}
if argv.Force {
if err := doForce(check); err != nil {
badExit(check, err)
}
} else {
if err := doSmart(check); err != nil {
badExit(check, err)
}
}
}
if configSave {
forge.ConfigSave()
}
log.Info("forge.FinalGoDepsCheck() worked :", check.GetGoPath())
okExit(check, "go.sum seems clean")
}
func findPwdRepo() *gitpb.Repo {
var check *gitpb.Repo
// attempt to use the working directory
// this is probably what happens most of the time
pwd, _ := os.Getwd()
if strings.HasPrefix(pwd, forge.GetGoSrc()) {
gopath := strings.TrimPrefix(pwd, forge.GetGoSrc())
gopath = strings.Trim(gopath, "/")
log.Info("findRepo() trying gopath", gopath)
check = forge.Repos.FindByFullPath(pwd)
if check != nil {
log.Info("findRepo() worked", check.GetGoPath())
return check
}
}
return nil
}
func saveAsMetadata(repo *gitpb.Repo) error {
cname := repo.GetCurrentBranchName()
if repo.GoInfo.GoPrimitive {
if err := repo.AutogenSave([]string{"go.mod"}, cname, true); err != nil {
return err
}
} else {
if err := repo.AutogenSave([]string{"go.mod", "go.sum"}, cname, true); err != nil {
return err
}
}
return nil
}
func restoreFromGoPkg(repo *gitpb.Repo) error {
homedir, err := os.UserHomeDir()
if err != nil {
badExit(nil, err)
}
rver := repo.GetMasterVersion()
if rver == "" {
return errors.New("could not get master version")
}
modfile := filepath.Join(homedir, "go/pkg/mod", repo.GetGoPath()+"@"+rver, "go.mod")
log.Info("mod path should be", modfile)
data, err := os.ReadFile(modfile)
if err != nil {
return err
}
modf, err := os.OpenFile("go.mod", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
return err
}
defer modf.Close()
modf.Write(data)
modfile = filepath.Join(homedir, "go/pkg/mod", repo.GetGoPath()+"@"+rver, "go.sum")
log.Info("mod path should be", modfile)
data, err = os.ReadFile(modfile)
if err == nil {
sumf, _ := os.OpenFile("go.sum", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
defer sumf.Close()
sumf.Write(data)
}
// try go.sum, but no error checking since it might not be there
return nil
}
func doMain(repo *gitpb.Repo) error {
if argv.Strict {
return doStrict(repo)
}
if argv.Force {
err := doForce(repo)
return err
}
// if --force or --strict is not set, fall back to a "smart" guess
// at what the user probably would want
return doSmart(repo)
}
|
