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
|
package main
import (
"errors"
"flag"
"fmt"
"io/fs"
"os"
"os/exec"
"path/filepath"
"strings"
)
const codeDir = "/nezuko/code/"
func checkRepoName(name string) (string, error) {
if strings.Contains(name, "/") {
return "", fmt.Errorf("invalid name %q: contains slashes")
}
if !strings.HasSuffix(name, ".git") {
name += ".git"
}
return name, nil
}
func create() error {
flag.Usage = func() {
fmt.Fprintf(os.Stderr, `Usage: create [REPO]
Create a new git repo named REPO.
`)
}
flag.Parse()
args := flag.Args()
if len(args) == 0 {
return fmt.Errorf("missing positional argument")
}
if len(args) > 1 {
return fmt.Errorf("too many positional arguments")
}
name, err := checkRepoName(args[0])
if err != nil {
return err
}
dir := filepath.Join(codeDir, name)
fmt.Printf("mkdir %s\n", dir)
if err := os.Mkdir(dir, 0755); err != nil {
return err
}
fmt.Printf("cd %s && git init --bare\n", dir)
git := exec.Command("git", "init", "--bare")
git.Dir = dir
git.Stdout = os.Stdout
git.Stderr = os.Stderr
if err := git.Run(); err != nil {
return err
}
hook := filepath.Join(dir, "hooks/post-update")
hookSample := hook + ".sample"
fmt.Printf("mv %s %s\n", hookSample, hook)
if err := os.Rename(hookSample, hook); err != nil {
return err
}
return nil
}
func fileExists(path string) bool {
_, err := os.Stat(path)
return !errors.Is(err, fs.ErrNotExist)
}
func delete() error {
flag.Usage = func() {
fmt.Fprintf(os.Stderr, `Usage: delete [REPO]
Delete the git repo named REPO.
`)
}
flag.Parse()
args := flag.Args()
if len(args) == 0 {
return fmt.Errorf("missing positional argument")
}
if len(args) > 1 {
return fmt.Errorf("too many positional arguments")
}
name, err := checkRepoName(args[0])
if err != nil {
return err
}
dir := filepath.Join(codeDir, name)
fmt.Printf("rm -rf %s\n", dir)
if !fileExists(dir) {
return fmt.Errorf("repo %q does not exist", name)
}
if err := os.RemoveAll(dir); err != nil {
return err
}
return nil
}
func main() {
var err error
if filepath.Base(os.Args[0]) == "delete" {
err = delete()
} else {
err = create()
}
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}
|
