Support for images

quiz/sheets.go

@@ -1,8 +1,13 @@
 package quiz
 
 import (
+	"bytes"
+	"crypto/sha256"
 	"encoding/json"
 	"fmt"
+	"image"
+	_ "image/jpeg"
+	"image/png"
 	"io"
 	"math"
 	"net/http"
@@ -114,7 +119,7 @@ func getQuestionFromRow(row *sheets.RowData) (qq *Question, err error) {
 			continue
 		}
 		cellContent := getContentFromCell(cell)
-		if cellContent == (DisplayableContent{}) {
+		if cellContent.Text == "" {
 			continue
 		}
 
@@ -138,7 +143,7 @@ func getQuestionFromRow(row *sheets.RowData) (qq *Question, err error) {
 	}
 
 	// validation
-	if qq.Question == (DisplayableContent{}) {
+	if qq.Question.Text == "" {
 		if len(qq.Correct) == 0 && len(qq.Wrong) == 0 {
 			return nil, nil
 		}
@@ -224,16 +229,26 @@ func parseCellFormula(formula, parameter string) (content DisplayableContent) {
 			log.Printf("Error: get image from url '%s': %v", url, err)
 			return
 		}
-		data, err := io.ReadAll(resp.Body)
+		if resp.StatusCode < 200 || resp.StatusCode >= 300 {
+			data, _ := io.ReadAll(resp.Body)
+			log.Printf("Error: could not get image from url: got '%s': %s", resp.Status, string(data))
+			return
+		}
+
+		img, imgFormat, err := image.Decode(resp.Body)
 		if err != nil {
-			log.Printf("Error: reading image response: %v", err)
+			log.Printf("Error: decoding image from '%s': %v", url, err)
 			return
 		}
-		if resp.StatusCode < 200 || resp.StatusCode >= 300 {
-			log.Printf("Error: could not get image from url: got '%s': %s", resp.Status, string(data))
+		hash := sha256.New()
+		imgData := &bytes.Buffer{}
+		err = png.Encode(io.MultiWriter(hash, imgData), img)
+		if err != nil {
+			log.Printf("Error: encoding image (%s to png): %v", imgFormat, err)
 			return
 		}
-		content.Text = string(data)
+		content.Text = fmt.Sprintf("%x", hash.Sum(nil))
+		content.Media = imgData.Bytes()
 	}
 	return
 }

quiz/types.go

@@ -73,8 +73,9 @@ type Question struct {
 }
 
 type DisplayableContent struct {
-	Type ContentType
-	Text string
+	Type  ContentType `json:"type"`
+	Text  string      `json:"text"`
+	Media []byte      `json:"-"`
 }
 
 type ContentType uint8
@@ -85,8 +86,8 @@ const (
 )
 
 type Round struct {
-	Question string                  `json:"question"`
-	Answers  []string                `json:"answers"`
+	Question DisplayableContent      `json:"question"`
+	Answers  []DisplayableContent    `json:"answers"`
 	Correct  int                     `json:"correct,omitempty"`
 	Current  int                     `json:"current_round"`
 	Max      int                     `json:"max_round"`
@@ -277,16 +278,16 @@ func (c Category) GetRounds(n int) []*Round {
 }
 
 func (q Question) ToRound() Round {
-	var answers []string
+	var answers []DisplayableContent
 
 	// select one correct answer
 	if len(q.Correct) > 1 {
 		rand.Shuffle(len(q.Correct), func(i, j int) {
 			q.Correct[i], q.Correct[j] = q.Correct[j], q.Correct[i]
 		})
-		answers = append(answers, q.Correct[rand.Intn(len(q.Correct)-1)].Text)
+		answers = append(answers, q.Correct[rand.Intn(len(q.Correct)-1)])
 	} else {
-		answers = append(answers, q.Correct[0].Text)
+		answers = append(answers, q.Correct[0])
 	}
 
 	// select up to 3 wrong answers
@@ -296,9 +297,7 @@ func (q Question) ToRound() Round {
 		})
 	}
 	num_wrong := int(math.Min(float64(cap(q.Wrong)), 3))
-	for _, a := range q.Wrong[:num_wrong] {
-		answers = append(answers, a.Text)
-	}
+	answers = append(answers, q.Wrong[:num_wrong]...)
 
 	var correct int
 	rand.Shuffle(len(answers), func(i, j int) {
@@ -309,7 +308,7 @@ func (q Question) ToRound() Round {
 	})
 
 	return Round{
-		Question: q.Question.Text,
+		Question: q.Question,
 		Answers:  answers,
 		Correct:  correct + 1,
 	}

webserver/handle.go

@@ -11,6 +11,7 @@ import (
 	"strings"
 
 	"github.com/google/uuid"
+	"github.com/gorilla/mux"
 	"github.com/kesuaheli/twitchgo"
 	"github.com/spf13/viper"
 )
@@ -300,6 +301,38 @@ func getRound(w http.ResponseWriter, r *http.Request) {
 
 	w.Write(b)
 }
+func getRoundMedia(w http.ResponseWriter, r *http.Request) {
+	c, ok := isAuthorized(r)
+	if !ok {
+		w.WriteHeader(http.StatusUnauthorized)
+		return
+	}
+
+	if c.Game == nil {
+		w.WriteHeader(http.StatusBadRequest)
+		return
+	}
+
+	if c.Game.Current == 0 {
+		http.Error(w, "no active round", http.StatusNotFound)
+		return
+	}
+
+	media := mux.Vars(r)["media"]
+	round := *c.Game.Rounds[c.Game.Current-1]
+	for _, answer := range round.Answers {
+		if answer.Type == quiz.CONTENTTEXT || answer.Text != media {
+			continue
+		}
+		w.Write(answer.Media)
+		return
+	}
+	if round.Question.Type != quiz.CONTENTTEXT && round.Question.Text == media {
+		w.Write(round.Question.Media)
+		return
+	}
+	http.Error(w, "media not found", http.StatusNotFound)
+}
 
 func nextRound(w http.ResponseWriter, r *http.Request) {
 	c, ok := isAuthorized(r)

webserver/webserver.go

@@ -72,6 +72,7 @@ func initHandler() http.Handler {
 	r.HandleFunc("/game", handleGame)
 	r.HandleFunc("/vote/streamer", handleStreamerVote).Methods(http.MethodPost)
 	r.HandleFunc("/round", getRound).Methods(http.MethodGet)
+	r.HandleFunc("/round/media/{media}", getRoundMedia).Methods(http.MethodGet)
 	r.HandleFunc("/round/next", nextRound).Methods(http.MethodPost)
 
 	return r