Split the client up into resource-specific files

2021-09-23 04:30:50 +00:00
parent 31985ee5c6
commit e85fbd5df9
10 changed files with 459 additions and 432 deletions
--- a/client/search.go
+++ b/client/search.go
@@ -0,0 +1,119 @@
+package client
+
+import "fmt"
+import "net/url"
+import "strings"
+
+import "cloud.google.com/go/civil"
+
+type SearchQuery struct {
+	SectionsAny []*Section
+	Completed   *bool
+	Due         *bool
+	DueOn       *civil.Date
+	DueBefore   *civil.Date
+	DueAfter    *civil.Date
+	TagsAny     []*Tag
+	TagsNot     []*Tag
+}
+
+var _TRUE = true
+var TRUE = &_TRUE
+var _FALSE = false
+var FALSE = &_FALSE
+
+func (wc *WorkspaceClient) Search(q *SearchQuery) ([]*Task, error) {
+	path := fmt.Sprintf("workspaces/%s/tasks/search", wc.workspace.GID)
+
+	values := &url.Values{
+		"sort_by":        []string{"created_at"},
+		"sort_ascending": []string{"true"},
+	}
+
+	values.Add("opt_fields", "created_at,due_on,html_notes,name")
+
+	if len(q.SectionsAny) > 0 {
+		gids := []string{}
+		for _, sec := range q.SectionsAny {
+			gids = append(gids, sec.GID)
+		}
+		values.Add("sections.any", strings.Join(gids, ","))
+	}
+
+	if q.Completed != nil {
+		values.Add("completed", fmt.Sprintf("%t", *q.Completed))
+	}
+
+	if q.Due != nil {
+		if *q.Due {
+			values.Add("due_on.after", "1970-01-01")
+		} else {
+			values.Add("due_on", "null")
+		}
+	}
+
+	if q.DueOn != nil {
+		values.Add("due_on", q.DueOn.String())
+	}
+
+	if q.DueBefore != nil {
+		values.Add("due_on.before", q.DueBefore.String())
+	}
+
+	if q.DueAfter != nil {
+		values.Add("due_on.after", q.DueAfter.String())
+	}
+
+	if len(q.TagsAny) > 0 {
+		gids := []string{}
+		for _, sec := range q.TagsAny {
+			gids = append(gids, sec.GID)
+		}
+		values.Add("tags.any", strings.Join(gids, ","))
+	}
+
+	if len(q.TagsNot) > 0 {
+		gids := []string{}
+		for _, sec := range q.TagsNot {
+			gids = append(gids, sec.GID)
+		}
+		values.Add("tags.not", strings.Join(gids, ","))
+	}
+
+	tasksByGID := map[string]*Task{}
+
+	for {
+		resp := &tasksResponse{}
+		err := wc.client.get(path, values, resp)
+		if err != nil {
+			return nil, err
+		}
+
+		maxCreatedAt := ""
+
+		for _, task := range resp.Data {
+			err := task.parse()
+			if err != nil {
+				return nil, err
+			}
+			tasksByGID[task.GID] = task
+
+			if task.CreatedAt > maxCreatedAt {
+				maxCreatedAt = task.CreatedAt
+			}
+		}
+
+		if len(resp.Data) < perPage {
+			break
+		}
+
+		values.Set("created_at.after", maxCreatedAt)
+	}
+
+	tasks := []*Task{}
+	for _, task := range tasksByGID {
+		tasks = append(tasks, task)
+	}
+
+	return tasks, nil
+}