summaryrefslogtreecommitdiff
path: root/page.go
blob: b58436d11b7a7e828721fa840e593e010255e7cf (plain)
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
package main

import "time"

type Service struct {
	Db Database
}

type DateInfo struct {
	Date      Date
	Tomorrow  Date
	Yesterday Date
	IsToday   bool
}

type EntryPage struct {
	Entry    *Entry
	Detached bool
	Tracking *Entry
	DateInfo DateInfo
}

type RootPage struct {
	DateInfo DateInfo
	Entry    *EntryPage
	Entries  []*Entry
}

type TasksPage struct {
	Tasks []*Task
}

func NewService(db Database) *Service {
	return &Service{
		Db: db,
	}
}

func NewDateInfo(t time.Time) DateInfo {
	return DateInfo{
		Date:      DateFromStd(t),
		Tomorrow:  DateFromStd(t.Add(24 * time.Hour)),
		Yesterday: DateFromStd(t.Add(-24 * time.Hour)),
		IsToday:   DateFromStd(time.Now()) == DateFromStd(t),
	}
}

func (srv *Service) GetTasksPage() (*TasksPage, error) {
	tasks, err := srv.Db.GetTasks()
	if err != nil {
		return nil, err
	}
	return &TasksPage{
		Tasks: tasks,
	}, nil
}
func (srv *Service) GetEntryPage(detached *Entry, dt DateInfo) (*EntryPage, error) {
	tracking, err := srv.Db.GetTracking()
	if err != nil {
		return nil, err
	}
	page := &EntryPage{
		Entry:    tracking,
		Detached: false,
		Tracking: tracking,
		DateInfo: dt,
	}

	if detached != nil {
		page.Entry = detached
		page.Detached = true
	}

	return page, nil
}

func (srv *Service) GetRootPage(date Date) (*RootPage, error) {
	var entry *Entry = nil

	if string(date) != time.Now().Format("2006-01-02") {
		// If not today detach
		entry = NewEntry(date)
	}
	
	entryPage, err := srv.GetEntryPage(entry, NewDateInfo(date.ToStd()))
	if err != nil {
		return nil, err
	}
	entries, err := srv.Db.QueryEntry(nil, &date)
	if err != nil {
		return nil, err
	}

	return &RootPage{
		DateInfo: entryPage.DateInfo,
		Entry:    entryPage,
		Entries:  entries,
	}, nil
}