cheesy-arena-lite/web/setup_lower_thirds.go

// Copyright 2014 Team 254. All Rights Reserved.
// Author: pat@patfairbank.com (Patrick Fairbank)
//
// Web routes for managing lower thirds.

package web

import (
	"fmt"
	"github.com/Team254/cheesy-arena/model"
	"github.com/Team254/cheesy-arena/websocket"
	"github.com/mitchellh/mapstructure"
	"io"
	"log"
	"net/http"
)

// Shows the lower third configuration page.
func (web *Web) lowerThirdsGetHandler(w http.ResponseWriter, r *http.Request) {
	if !web.userIsAdmin(w, r) {
		return
	}

	template, err := web.parseFiles("templates/setup_lower_thirds.html", "templates/base.html")
	if err != nil {
		handleWebErr(w, err)
		return
	}
	lowerThirds, err := web.arena.Database.GetAllLowerThirds()
	if err != nil {
		handleWebErr(w, err)
		return
	}
	data := struct {
		*model.EventSettings
		LowerThirds []model.LowerThird
	}{web.arena.EventSettings, lowerThirds}
	err = template.ExecuteTemplate(w, "base", data)
	if err != nil {
		handleWebErr(w, err)
		return
	}
}

// The websocket endpoint for the lower thirds client to send control commands.
func (web *Web) lowerThirdsWebsocketHandler(w http.ResponseWriter, r *http.Request) {
	if !web.userIsAdmin(w, r) {
		return
	}

	ws, err := websocket.NewWebsocket(w, r)
	if err != nil {
		handleWebErr(w, err)
		return
	}
	defer ws.Close()

	// Loop, waiting for commands and responding to them, until the client closes the connection.
	for {
		messageType, data, err := ws.Read()
		if err != nil {
			if err == io.EOF {
				// Client has closed the connection; nothing to do here.
				return
			}
			log.Println(err)
			return
		}

		switch messageType {
		case "saveLowerThird":
			var lowerThird model.LowerThird
			err = mapstructure.Decode(data, &lowerThird)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
			web.saveLowerThird(&lowerThird)
		case "deleteLowerThird":
			var lowerThird model.LowerThird
			err = mapstructure.Decode(data, &lowerThird)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
			err = web.arena.Database.DeleteLowerThird(&lowerThird)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
		case "showLowerThird":
			var lowerThird model.LowerThird
			err = mapstructure.Decode(data, &lowerThird)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
			web.saveLowerThird(&lowerThird)
			web.arena.LowerThirdNotifier.NotifyWithMessage(lowerThird)
			web.arena.AudienceDisplayMode = "lowerThird"
			web.arena.AudienceDisplayModeNotifier.Notify()
			continue
		case "hideLowerThird":
			var lowerThird model.LowerThird
			err = mapstructure.Decode(data, &lowerThird)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
			web.saveLowerThird(&lowerThird)
			web.arena.AudienceDisplayMode = "blank"
			web.arena.AudienceDisplayModeNotifier.Notify()
			continue
		case "reorderLowerThird":
			args := struct {
				Id     int
				MoveUp bool
			}{}
			err = mapstructure.Decode(data, &args)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
			err = web.reorderLowerThird(args.Id, args.MoveUp)
			if err != nil {
				ws.WriteError(err.Error())
				continue
			}
		default:
			ws.WriteError(fmt.Sprintf("Invalid message type '%s'.", messageType))
			continue
		}

		// Force a reload of the client to render the updated lower thirds list.
		err = ws.WriteNotifier(web.arena.ReloadDisplaysNotifier)
		if err != nil {
			log.Println(err)
			return
		}
	}
}

func (web *Web) saveLowerThird(lowerThird *model.LowerThird) error {
	oldLowerThird, err := web.arena.Database.GetLowerThirdById(lowerThird.Id)
	if err != nil {
		return err
	}

	// Create or update lower third.
	if oldLowerThird == nil {
		err = web.arena.Database.CreateLowerThird(lowerThird)
	} else {
		err = web.arena.Database.SaveLowerThird(lowerThird)
	}
	if err != nil {
		return err
	}
	return nil
}

func (web *Web) reorderLowerThird(id int, moveUp bool) error {
	lowerThird, err := web.arena.Database.GetLowerThirdById(id)
	if err != nil {
		return err
	}

	// Get the lower third to swap positions with.
	lowerThirds, err := web.arena.Database.GetAllLowerThirds()
	if err != nil {
		return err
	}
	var lowerThirdIndex int
	for i, third := range lowerThirds {
		if third.Id == lowerThird.Id {
			lowerThirdIndex = i
			break
		}
	}
	if moveUp {
		lowerThirdIndex--
	} else {
		lowerThirdIndex++
	}
	if lowerThirdIndex < 0 || lowerThirdIndex == len(lowerThirds) {
		// The one to move is already at the limit; return an error to prevent a page reload.
		return fmt.Errorf("Already at the limit.")
	}
	adjacentLowerThird, err := web.arena.Database.GetLowerThirdById(lowerThirds[lowerThirdIndex].Id)
	if err != nil {
		return err
	}

	// Swap their display orders and save.
	lowerThird.DisplayOrder, adjacentLowerThird.DisplayOrder =
		adjacentLowerThird.DisplayOrder, lowerThird.DisplayOrder
	err = web.arena.Database.SaveLowerThird(lowerThird)
	if err != nil {
		return err
	}
	err = web.arena.Database.SaveLowerThird(adjacentLowerThird)
	if err != nil {
		return err
	}

	return nil
}