cheesy-arena-lite/model/lower_third.go

// Copyright 2014 Team 254. All Rights Reserved.
// Author: pat@patfairbank.com (Patrick Fairbank)
//
// Model and datastore CRUD methods for the text on a lower third slide.

package model

import (
	"sort"
)

type LowerThird struct {
	Id           int64 `db:"id"`
	TopText      string
	BottomText   string
	DisplayOrder int
	AwardId      int64
}

func (database *Database) CreateLowerThird(lowerThird *LowerThird) error {
	return database.tables[LowerThird{}].create(lowerThird)
}

func (database *Database) GetLowerThirdById(id int64) (*LowerThird, error) {
	var lowerThird *LowerThird
	err := database.tables[LowerThird{}].getById(id, &lowerThird)
	return lowerThird, err
}

func (database *Database) UpdateLowerThird(lowerThird *LowerThird) error {
	return database.tables[LowerThird{}].update(lowerThird)
}

func (database *Database) DeleteLowerThird(id int64) error {
	return database.tables[LowerThird{}].delete(id)
}

func (database *Database) TruncateLowerThirds() error {
	return database.tables[LowerThird{}].truncate()
}

func (database *Database) GetAllLowerThirds() ([]LowerThird, error) {
	var lowerThirds []LowerThird
	if err := database.tables[LowerThird{}].getAll(&lowerThirds); err != nil {
		return nil, err
	}
	sort.Slice(lowerThirds, func(i, j int) bool {
		return lowerThirds[i].DisplayOrder < lowerThirds[j].DisplayOrder
	})
	return lowerThirds, nil
}

func (database *Database) GetLowerThirdsByAwardId(awardId int64) ([]LowerThird, error) {
	lowerThirds, err := database.GetAllLowerThirds()
	if err != nil {
		return nil, err
	}

	var matchingLowerThirds []LowerThird
	for _, lowerThird := range lowerThirds {
		if lowerThird.AwardId == awardId {
			matchingLowerThirds = append(matchingLowerThirds, lowerThird)
		}
	}
	return matchingLowerThirds, nil
}

func (database *Database) GetNextLowerThirdDisplayOrder() int {
	lowerThirds, err := database.GetAllLowerThirds()
	if err != nil {
		return 0
	}
	if len(lowerThirds) == 0 {
		return 1
	}
	return lowerThirds[len(lowerThirds)-1].DisplayOrder + 1
}