diff options
Diffstat (limited to 'model/timer.go')
-rw-r--r-- | model/timer.go | 110 |
1 files changed, 0 insertions, 110 deletions
diff --git a/model/timer.go b/model/timer.go deleted file mode 100644 index 3f13d0d..0000000 --- a/model/timer.go +++ /dev/null @@ -1,110 +0,0 @@ -package model - -import ( - "database/sql" - "log" - "time" - - "stevenlr.com/timer/utils" -) - -func GenerateTimerToken() (string, error) { - return utils.GenerateRandomString(66) -} - -type Timer struct { - Id UUID - Name string - StartTime Time - EndTime Time - Owner UUID - Token string -} - -func (self Timer) IsFinished() bool { - return MakeTimeNow().Compare(self.EndTime) >= 0 -} - -func InsertTimer(tx *sql.Tx, name string, seconds int, ownerId UUID) error { - now := MakeTimeNow() - end := Time(time.Time(now).Add(time.Duration(seconds) * time.Second)) - id := MakeUUID() - token, _ := GenerateTimerToken() - _, err := tx.Exec(` - INSERT INTO Timer VALUES ($1, $2, $3, $4, $5, $6)`, id, name, now, end, ownerId, token) - return err -} - -func GetTimersForUser(db *sql.DB, owner UUID) []Timer { - rows, err := db.Query("SELECT Id, Name FROM Timer WHERE Owner=$1", owner) - if err != nil { - log.Fatalln(err) - } - - timers := []Timer{} - for rows.Next() { - var t Timer - if err := rows.Scan(&t.Id, &t.Name); err == nil { - timers = append(timers, t) - } - } - - return timers -} - -func GetTimerForUser(db *sql.DB, id UUID, userId UUID) *Timer { - row := db.QueryRow("SELECT Id, Name, StartTime, EndTime, Owner, Token FROM Timer WHERE Id=$1 AND Owner=$2", id, userId) - - var t Timer - if err := row.Scan(&t.Id, &t.Name, &t.StartTime, &t.EndTime, &t.Owner, &t.Token); err == nil { - return &t - } - - return nil -} - -func GetTimerWithToken(db *sql.DB, id UUID, token string) *Timer { - row := db.QueryRow("SELECT Id, Name, StartTime, EndTime, Owner, Token FROM Timer WHERE Id=$1 AND Token=$2", id, token) - - var t Timer - if err := row.Scan(&t.Id, &t.Name, &t.StartTime, &t.EndTime, &t.Owner, &t.Token); err == nil { - return &t - } - - return nil -} - -func DeleteTimer(db *sql.DB, id UUID, userId UUID) bool { - res, err := db.Exec("DELETE FROM Timer WHERE Id=$1 AND Owner=$2", id, userId) - if err != nil { - return false - } - - affected, err := res.RowsAffected() - return err == nil && affected == 1 -} - -func UpdateTimerEndTime(db *sql.DB, id UUID, endTime Time, userId UUID) bool { - res, err := db.Exec("UPDATE Timer SET EndTime=$1 WHERE Id=$2 AND Owner=$3", endTime, id, userId) - if err != nil { - return false - } - - affected, err := res.RowsAffected() - return err == nil && affected == 1 -} - -func RegenerateTimerToken(db *sql.DB, id UUID, userId UUID) bool { - newToken, err := GenerateTimerToken() - if err != nil { - return false - } - - res, err := db.Exec("UPDATE Timer SET Token=$1 WHERE Id=$2 AND Owner=$3", newToken, id, userId) - if err != nil { - return false - } - - affected, err := res.RowsAffected() - return err == nil && affected == 1 -} |