Added checklists to database, rendering and toggling items

node.go

@@ -8,6 +8,22 @@ import (
 	"time"
 )
 
+type ChecklistItem struct {
+	ID      int
+	GroupID int `db:"checklist_group_id"`
+	Order   int
+	Label   string
+	Checked bool
+}
+
+type ChecklistGroup struct {
+	ID     int
+	NodeID int `db:"node_id"`
+	Order  int
+	Label  string
+	Items  []ChecklistItem
+}
+
 type Node struct {
 	ID          int
 	UserID      int `db:"user_id"`
@@ -22,6 +38,8 @@ type Node struct {
 	Complete    bool
 	Level       int
 
+	ChecklistGroups []ChecklistGroup
+
 	ContentEncrypted string `db:"content_encrypted" json:"-"`
 	Markdown         bool
 }
@@ -211,6 +229,8 @@ func RetrieveNode(userID, nodeID int) (node Node, err error) { // {{{
 			} else {
 				node.Content = row.Content
 			}
+
+			node.retrieveChecklist()
 		}
 
 		if row.Level == 1 {
@@ -421,5 +441,96 @@ func SearchNodes(userID int, search string) (nodes []Node, err error) { // {{{
 
 	return
 } // }}}
+func ChecklistItemState(userID, checklistItemID int, state bool) (err error) {// {{{
+	_, err = service.Db.Conn.Exec(
+		`
+		UPDATE checklist_item i
+		SET checked = $3
+		FROM checklist_group g, node n
+		WHERE
+			i.checklist_group_id = g.id AND
+			g.node_id = n.id AND
+			n.user_id = $1 AND
+			i.id = $2;	
+		`,
+		userID,
+		checklistItemID,
+		state,
+	)
+	return
+}// }}}
+
+func (node *Node) retrieveChecklist() (err error) { // {{{
+	var rows *sqlx.Rows
+	rows, err = service.Db.Conn.Queryx(`
+		SELECT
+			g.id AS group_id,
+			g.order AS group_order,
+			g.label AS group_label,
+
+			i.id AS item_id,
+			i.order AS item_order,
+			i.label AS item_label,
+			i.checked
+
+		FROM public.checklist_group g
+		LEFT JOIN public.checklist_item i ON i.checklist_group_id = g.id
+		WHERE
+			g.node_id = $1
+		ORDER BY
+			g.order DESC,
+			i.order DESC
+	`, node.ID)
+	if err != nil {
+		return
+	}
+	defer rows.Close()
+
+	groups := make(map[int]*ChecklistGroup)
+	var found bool
+	var group *ChecklistGroup
+	var item ChecklistItem
+	for rows.Next() {
+		row := struct {
+			GroupID    int    `db:"group_id"`
+			GroupOrder int    `db:"group_order"`
+			GroupLabel string `db:"group_label"`
+
+			ItemID    int    `db:"item_id"`
+			ItemOrder int    `db:"item_order"`
+			ItemLabel string `db:"item_label"`
+			Checked   bool
+		}{}
+		err = rows.StructScan(&row)
+		if err != nil {
+			return
+		}
+
+		if group, found = groups[row.GroupID]; !found {
+			group = new(ChecklistGroup)
+			group.ID = row.GroupID
+			group.NodeID = node.ID
+			group.Order = row.GroupOrder
+			group.Label = row.GroupLabel
+			group.Items = []ChecklistItem{}
+			groups[group.ID] = group
+		}
+
+		item = ChecklistItem{}
+		item.ID = row.ItemID
+		item.GroupID = row.GroupID
+		item.Order = row.ItemOrder
+		item.Label = row.ItemLabel
+		item.Checked = row.Checked
+		group.Items = append(group.Items, item)
+	}
+
+	node.ChecklistGroups = []ChecklistGroup{}
+	for _, group := range groups {
+		node.ChecklistGroups = append(node.ChecklistGroups, *group)
+	}
+
+	return
+} // }}}
 
 // vim: foldmethod=marker