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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
package main
import (
"fmt"
"io/ioutil"
"os"
"strings"
"time"
"github.com/teddywing/timetasker/timetask"
"gopkg.in/alecthomas/kingpin.v2"
)
var VERSION string = "0.1.0"
var config Config
func main() {
var err error
// Parse command line arguments
project_alias := kingpin.Flag(
"project",
"Project alias defined in config.toml.",
).
Short('p').
String()
time_spent := kingpin.Flag("time", "Time spent working on project.").
Short('t').
Default("7").
Float()
date_str := kingpin.Flag("date", "Date when work was done (e.g. 2017-01-31)").
Short('d').
String()
description := kingpin.Flag("description", "Description of work.").
Short('m').
String()
write_config_description := fmt.Sprintf(
"Initialise a new config file template at %s",
configFile(),
)
write_config := kingpin.Flag("write-config", write_config_description).
Bool()
list_modules := kingpin.Flag("list-modules", "List sprints with IDs").Bool()
kingpin.Version(VERSION)
kingpin.Parse()
// Error if no --project unless --write-config was passed
if *project_alias == "" && !*write_config {
kingpin.Fatalf("required flag --project not provided, try --help")
}
if *write_config {
err = maybeWriteConfig()
kingpin.FatalIfError(err, "could not write config file")
os.Exit(0)
}
err = loadConfig()
kingpin.FatalIfError(err, "could not load config file, try --write-config")
// Submit time entry
project, ok := config.Projects[*project_alias]
if !ok {
kingpin.Errorf("project '%s' not found", *project_alias)
os.Exit(1)
}
var date time.Time
// If the date argument isn't sent, default to today
if *date_str == "" {
date = time.Now()
} else {
date, err = time.Parse("2006-01-02", *date_str)
kingpin.FatalIfError(
err,
"date '%s' could not be parsed. Example: -d 2017-01-31\n",
*date_str,
)
}
time_entry := timetask.NewTimeEntry(
config.Profile,
project,
date,
*time_spent,
*description,
)
password, err := passwordCmd(config.Auth.PasswordCmd)
kingpin.FatalIfError(err, "password command failed")
resp, client, err := timetask.Login(
config.Auth.Username,
password,
)
kingpin.FatalIfError(err, "login request failed")
defer resp.Body.Close()
body, err := ioutil.ReadAll(resp.Body)
if strings.Contains(
string(body),
"The username and password don't appear to be valid.",
) {
kingpin.Errorf("TimeTask authentication failed")
os.Exit(1)
}
// List modules
if *list_modules {
modules, err := timetask.RequestModules(*client, time_entry)
kingpin.FatalIfError(err, "could not retrieve sprints")
fmt.Println(modules)
os.Exit(0)
}
resp, err = timetask.SubmitTimeEntry(*client, time_entry)
kingpin.FatalIfError(err, "time entry submission request failed")
defer resp.Body.Close()
body, err = ioutil.ReadAll(resp.Body)
if strings.Contains(
string(body),
"No time entries were created.",
) {
kingpin.Errorf("time entry creation failed")
os.Exit(1)
}
}
|
