Notes/schedule.go

package main

import (
	// Standard
	"encoding/json"
	"fmt"
	"regexp"
	"strings"
	"time"
)

func init() {
	fmt.Printf("")
}

type Schedule struct {
	ID           int
	UserID       int `json:"user_id"`
	Node         Node
	ScheduleUUID string `db:"schedule_uuid"`
	Time         time.Time
	Description  string
	Acknowledged bool
}

func ScanForSchedules(content string) (schedules []Schedule) {
	schedules = []Schedule{}

	rxp := regexp.MustCompile(`\{\s*([0-9]{4}-[0-9]{2}-[0-9]{2}\s+[0-9]{2}:[0-9]{2}(?::[0-9]{2})?)\s*\,\s*([^\]]+?)\s*\}`)
	foundSchedules := rxp.FindAllStringSubmatch(content, -1)

	for _, data := range foundSchedules {
		// Missing seconds
		if strings.Count(data[1], ":") == 1 {
			data[1] = data[1] + ":00"
		}

		timestamp, err := time.Parse("2006-01-02 15:04:05", data[1])
		if err != nil {
			continue
		}

		schedule := Schedule{
			Time:        timestamp,
			Description: data[2],
		}
		schedules = append(schedules, schedule)
	}

	return
}

func (a Schedule) IsEqual(b Schedule) bool {
	return a.UserID == b.UserID &&
		a.Node.ID == b.Node.ID &&
		a.Time.Equal(b.Time) &&
		a.Description == b.Description
}

func (s *Schedule) Insert(queryable Queryable) error {
	res := queryable.QueryRow(`
		INSERT INTO schedule(user_id, node_id, time, description)
		VALUES($1, $2, $3, $4)
		RETURNING id
		`,
		s.UserID,
		s.Node.ID,
		s.Time,
		s.Description,
	)

	return res.Scan(&s.ID)
}

func (s *Schedule) Delete(queryable Queryable) error {
	_, err := queryable.Exec(`
		DELETE FROM schedule
		WHERE
			user_id = $1 AND
			id = $2
		`,
		s.UserID,
		s.ID,
	)
	return err
}

func RetrieveSchedules(userID int, nodeID int) (schedules []Schedule, err error) {
	schedules = []Schedule{}

	res := service.Db.Conn.QueryRow(`
		WITH schedule_events AS (
			SELECT
				id,
				user_id,
				json_build_object('id', node_id) AS node,
				schedule_uuid,
				time::timestamptz,
				description,
				acknowledged
			FROM schedule
			WHERE
				user_id=$1 AND
				CASE
				WHEN $2 > 0 THEN node_id = $2
				ELSE true
				END
		)
		SELECT
			COALESCE(jsonb_agg(s.*), '[]'::jsonb)
		FROM schedule_events s
		`,
		userID,
		nodeID,
	)

	var data []byte
	err = res.Scan(&data)
	if err != nil {
		return
	}

	err = json.Unmarshal(data, &schedules)
	return
}

func ExpiredSchedules() []Schedule {
	schedules := []Schedule{}
	/*
	res, err := service.Db.Conn.Query(`
		SELECT
			*
		FROM schedule
		WHERE
			time < NOW() AND
			NOT acknowledged
		ORDER BY
			time ASC
	`)
	*/
	return schedules
}