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
|
package main
import (
"flag"
"fmt"
"os"
)
func main() {
if len(os.Args) < 2 {
fmt.Fprintln(os.Stderr, "usage: nilink <serve|add|remove|list>")
os.Exit(1)
}
switch os.Args[1] {
case "serve":
cmdServe(os.Args[2:])
case "add":
cmdAdd(os.Args[2:])
case "remove":
cmdRemove(os.Args[2:])
case "list":
cmdList(os.Args[2:])
default:
fmt.Fprintf(os.Stderr, "unknown command: %s\n", os.Args[1])
os.Exit(1)
}
}
func cmdAdd(args []string) {
flags := flag.NewFlagSet("add", flag.ExitOnError)
dbPath := flags.String("db", "data/nilink.db", "database path")
flags.Parse(args)
if flags.NArg() != 1 {
fmt.Fprintln(os.Stderr, "usage: nilink add [-db path] <url>")
os.Exit(1)
}
url := flags.Arg(0)
db, err := openDB(*dbPath)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
defer db.Close()
id, err := insertLink(db, url)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
short, err := encodeID(id)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
fmt.Printf("%s %s\n", short, url)
}
func cmdRemove(args []string) {
flags := flag.NewFlagSet("remove", flag.ExitOnError)
dbPath := flags.String("db", "data/nilink.db", "database path")
flags.Parse(args)
if flags.NArg() != 1 {
fmt.Fprintln(os.Stderr, "usage: nilink remove [-db path] <short-id>")
os.Exit(1)
}
shortID := flags.Arg(0)
id, err := decodeID(shortID)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
db, err := openDB(*dbPath)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
defer db.Close()
if err := deleteLink(db, id); err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
}
func cmdList(args []string) {
flags := flag.NewFlagSet("list", flag.ExitOnError)
dbPath := flags.String("db", "data/nilink.db", "database path")
flags.Parse(args)
db, err := openDB(*dbPath)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
defer db.Close()
links, err := listLinks(db)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
for _, l := range links {
short, err := encodeID(l.ID)
if err != nil {
fmt.Fprintf(os.Stderr, "error: %v\n", err)
os.Exit(1)
}
fmt.Printf("%s %s %s\n", short, l.URL, l.CreatedAt)
}
}
|
