-
Notifications
You must be signed in to change notification settings - Fork 1
/
list-cmd.go
95 lines (72 loc) · 1.92 KB
/
list-cmd.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
package main
import (
"io"
"fmt"
"github.com/ivartj/minn/args"
)
func init() {
cmdRegister("list", listCmd)
}
func listCmdUsage(w io.Writer) {
fmt.Fprintf(w, "Usage: %s list\n", mainProgramName)
fmt.Fprintln(w, `
Description:
Lists cards by card IDs, ordered so that the earliest scheduled come first.
Options can be used to control the selection of cards listed.
Options:
-h, --help Prints help message.
--select-scheduled Limit selection to scheduled cards.
--select-new Limit selection to new (unreviewed) cards.
`)
}
func listCmdArgs(cmd *cmdContext) (whereClause string) {
tok := args.NewTokenizer(cmd.Args)
whereExpressions := []string{}
for tok.Next() {
if !tok.IsOption() {
cmd.Fatalf("Unexpected non-option, '%s'.\n", tok.Arg())
}
switch tok.Arg() {
case "-h", "--help":
listCmdUsage(cmd.Stdout)
cmd.Exit(0)
case "--select-scheduled":
whereExpressions = append(whereExpressions, fmt.Sprintf("schedule_time < '%s'", cmd.Now().Format(utilTimeFormat)))
case "--select-new":
whereExpressions = append(whereExpressions, "state is 0")
default:
cmd.Fatalf("Unrecognized option, '%s'.", tok.Arg())
}
}
if tok.Err() != nil {
cmd.Fatalf("Error on processing command-line options: %s.\n", tok.Err().Error())
}
if len(whereExpressions) == 0 {
whereClause = ""
} else {
whereClause = "where "
for i, expr := range whereExpressions {
if i != 0 {
whereClause += " and "
}
whereClause += expr
}
}
return whereClause
}
func listCmd(cmd *cmdContext) {
whereClause := listCmdArgs(cmd)
rows, err := cmd.Query("select card_id from cards " + whereClause + " order by schedule_time asc;")
if err != nil {
cmd.Fatalf("Database query failed: %s.\n", err.Error())
}
var cardID int
for rows.Next() {
err = rows.Scan(&cardID)
if err != nil {
cmd.Fatalf("Failed to scan card ID: %s.\n", err.Error())
}
cmd.Println(cardID)
}
cmd.Commit()
}