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
|
diff --git a/main.go b/main.go
index 1e74a63d6204131dd23e34603cf200402a4727c5..e903f4bb19e3f70ad14d473264e7fa29ded812a7 100644
--- a/main.go
+++ b/main.go
@@ -11,35 +11,54 @@
"github.com/peterbourgon/ff/v3"
)
+var Version = "develop"
+
func main() {
fs := flag.NewFlagSet("gomodinit", flag.ExitOnError)
- base := fs.String("base", "", "Base URL prefix for module")
+ versionFlag := fs.Bool("version", false, "Print version and exit")
if err := ff.Parse(fs, os.Args[1:], ff.WithEnvVarPrefix("GMI")); err != nil {
fmt.Println(err)
return
}
- if *base == "" {
- fmt.Println("Base URL is required")
+ if *versionFlag {
+ fmt.Printf("gomodinit %s\n", Version)
return
}
- dir, err := os.Getwd()
- if err != nil {
- fmt.Println(err)
- return
- }
-
- name := filepath.Base(dir)
+ var name string
if fs.NArg() > 0 {
name = fs.Arg(0)
+ } else {
+ n, err := module()
+ if err != nil {
+ fmt.Println(err)
+ return
+ }
+ name = n
}
- *base = strings.TrimSuffix(*base, "/")
-
- cmd := exec.Command("go", "mod", "init", fmt.Sprintf("%s/%s", *base, name))
+ cmd := exec.Command("go", "mod", "init", name)
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
_ = cmd.Run()
}
+
+func module() (string, error) {
+ dir, err := os.Getwd()
+ if err != nil {
+ return "", err
+ }
+
+ list := strings.Split(dir, string(filepath.Separator))
+ name := list[len(list)-1]
+ for idx := len(list) - 2; idx >= 0; idx-- {
+ part := list[idx]
+ name = fmt.Sprintf("%s/%s", part, name)
+ if strings.Contains(part, ".") {
+ break
+ }
+ }
+ return name, nil
+}
|