lewis.moe
Monorepo for Tangled
appview/models: move db.PunchCard into models
Signed-off-by: oppiliappan <me@oppi.li>
+7
-16
appview/db/punchcard.go
+7
-16
appview/db/punchcard.go
···
5
5
"fmt"
6
6
"strings"
7
7
"time"
8
+
9
+
"tangled.org/core/appview/models"
8
10
)
9
11
10
-
type Punch struct {
11
-
Did string
12
-
Date time.Time
13
-
Count int
14
-
}
15
-
16
12
// this adds to the existing count
17
-
func AddPunch(e Execer, punch Punch) error {
13
+
func AddPunch(e Execer, punch models.Punch) error {
18
14
_, err := e.Exec(`
19
15
insert into punchcard (did, date, count)
20
16
values (?, ?, ?)
···
24
20
return err
25
21
}
26
22
27
-
type Punchcard struct {
28
-
Total int
29
-
Punches []Punch
30
-
}
31
-
32
-
func MakePunchcard(e Execer, filters ...filter) (*Punchcard, error) {
33
-
punchcard := &Punchcard{}
23
+
func MakePunchcard(e Execer, filters ...filter) (*models.Punchcard, error) {
24
+
punchcard := &models.Punchcard{}
34
25
now := time.Now()
35
26
startOfYear := time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.UTC)
36
27
endOfYear := time.Date(now.Year(), 12, 31, 0, 0, 0, 0, time.UTC)
37
28
for d := startOfYear; d.Before(endOfYear) || d.Equal(endOfYear); d = d.AddDate(0, 0, 1) {
38
-
punchcard.Punches = append(punchcard.Punches, Punch{
29
+
punchcard.Punches = append(punchcard.Punches, models.Punch{
39
30
Date: d,
40
31
Count: 0,
41
32
})
···
68
59
defer rows.Close()
69
60
70
61
for rows.Next() {
71
-
var punch Punch
62
+
var punch models.Punch
72
63
var date string
73
64
var count sql.NullInt64
74
65
if err := rows.Scan(&date, &count); err != nil {
+14
appview/models/punchcard.go
+14
appview/models/punchcard.go
+1
-1
appview/pages/pages.go
+1
-1
appview/pages/pages.go