Refactor web handlers into separate package.

web/setup_lower_thirds.go

@@ -0,0 +1,205 @@
+// 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/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
+	}
+
+	websocket, err := NewWebsocket(w, r)
+	if err != nil {
+		handleWebErr(w, err)
+		return
+	}
+	defer websocket.Close()
+
+	// Loop, waiting for commands and responding to them, until the client closes the connection.
+	for {
+		messageType, data, err := websocket.Read()
+		if err != nil {
+			if err == io.EOF {
+				// Client has closed the connection; nothing to do here.
+				return
+			}
+			log.Printf("Websocket error: %s", err)
+			return
+		}
+
+		switch messageType {
+		case "saveLowerThird":
+			var lowerThird model.LowerThird
+			err = mapstructure.Decode(data, &lowerThird)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+			web.saveLowerThird(&lowerThird)
+		case "deleteLowerThird":
+			var lowerThird model.LowerThird
+			err = mapstructure.Decode(data, &lowerThird)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+			err = web.arena.Database.DeleteLowerThird(&lowerThird)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+		case "showLowerThird":
+			var lowerThird model.LowerThird
+			err = mapstructure.Decode(data, &lowerThird)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+			web.saveLowerThird(&lowerThird)
+			web.arena.LowerThirdNotifier.Notify(lowerThird)
+			web.arena.AudienceDisplayScreen = "lowerThird"
+			web.arena.AudienceDisplayNotifier.Notify(nil)
+			continue
+		case "hideLowerThird":
+			var lowerThird model.LowerThird
+			err = mapstructure.Decode(data, &lowerThird)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+			web.saveLowerThird(&lowerThird)
+			web.arena.AudienceDisplayScreen = "blank"
+			web.arena.AudienceDisplayNotifier.Notify(nil)
+			continue
+		case "reorderLowerThird":
+			args := struct {
+				Id     int
+				MoveUp bool
+			}{}
+			err = mapstructure.Decode(data, &args)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+			err = web.reorderLowerThird(args.Id, args.MoveUp)
+			if err != nil {
+				websocket.WriteError(err.Error())
+				continue
+			}
+		default:
+			websocket.WriteError(fmt.Sprintf("Invalid message type '%s'.", messageType))
+			continue
+		}
+
+		// Force a reload of the client to render the updated lower thirds list.
+		err = websocket.Write("reload", nil)
+		if err != nil {
+			log.Printf("Websocket error: %s", 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
+}