/
main.go
105 lines (91 loc) · 2.57 KB
/
main.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
94
95
96
97
98
99
100
101
102
103
104
105
package main
import (
"bytes"
"fmt"
"os"
"os/exec"
"strings"
"github.com/urfave/cli/v2"
"github.com/xyproto/textoutput"
)
const versionString = "aget 1.3.3"
func run(o *textoutput.TextOutput, commandString string) error {
var stdoutBuf, stderrBuf bytes.Buffer
o.Println("<green>" + commandString + "</green>")
words := strings.Split(commandString, " ")
cmd := exec.Command(words[0], words[1:]...)
cmd.Stdout = &stdoutBuf
cmd.Stderr = &stderrBuf
if err := cmd.Start(); err != nil {
o.Printf("<yellow>%s</yellow>\n", err)
o.Printf("<yellow>%s</yellow>\n", stdoutBuf.String())
o.Printf("<red>%s</red>\n", stderrBuf.String())
return err
}
if err := cmd.Wait(); err != nil {
o.Printf("<yellow>%s</yellow>\n", err)
o.Printf("<yellow>%s</yellow>\n", stdoutBuf.String())
o.Printf("<red>%s</red>\n", stderrBuf.String())
return err
}
return nil
}
func main() {
o := textoutput.New()
if appErr := (&cli.App{
Name: "aget",
Usage: "clone AUR packages with git",
Flags: []cli.Flag{
&cli.BoolFlag{Name: "silent", Aliases: []string{"s"}},
&cli.BoolFlag{Name: "https", Aliases: []string{"http", "web"}},
&cli.BoolFlag{Name: "version", Aliases: []string{"V"}},
},
Action: func(c *cli.Context) error {
if c.Bool("version") {
o.Println(versionString)
os.Exit(0)
}
// Check if text output should be disabled
if c.Bool("silent") {
o.Disable()
}
// Check if any arguments are given
if c.NArg() == 0 {
o.ErrExit("Please supply a package name as an argument")
}
// Interpret the arguments as package names
packageNames := c.Args().Slice()
// Treat all arguments as AUR packages that should be cloned
var err error
for _, packageName := range packageNames {
if _, err := os.Stat(packageName); err == nil {
o.Print("<darkred>Directory already exists:</darkred> ")
o.Println("<yellow>" + packageName + "</yellow>")
continue
}
protocol := "ssh://"
if c.Bool("https") {
protocol = "https://"
}
url := fmt.Sprintf("%saur@aur.archlinux.org/%s.git", protocol, packageName)
// git clone
if err := run(o, "git clone "+url); err != nil {
continue
}
// cd packageName
o.Println("<green>cd " + packageName + "</green>")
if err := os.Chdir(packageName); err != nil {
o.Printf("<red>%s</red>\n", err)
continue
}
// switch to the master branch, in case the default branch name is ie. "main"
if err := run(o, "git switch -C master"); err != nil {
continue
}
}
return err
},
}).Run(os.Args); appErr != nil {
o.ErrExit(appErr.Error())
}
}