From 74efe02c75ae624d9815543feee233f21700b97f Mon Sep 17 00:00:00 2001
From: Chris Sexton <chris.sexton@gmail.com>
Date: Sat, 9 Feb 2019 08:32:27 -0500
Subject: [PATCH] twitch: make stream announcements more complicated

And fun. Now they're templates.
---
 plugins/twitch/twitch.go | 75 +++++++++++++++++++++++++++++++++++-----
 1 file changed, 67 insertions(+), 8 deletions(-)

diff --git a/plugins/twitch/twitch.go b/plugins/twitch/twitch.go
index a088d40..3e3081a 100644
--- a/plugins/twitch/twitch.go
+++ b/plugins/twitch/twitch.go
@@ -1,14 +1,15 @@
 package twitch
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
-	"html/template"
 	"io/ioutil"
 	"log"
 	"net/http"
 	"net/url"
 	"strings"
+	"text/template"
 	"time"
 
 	"github.com/velour/catbase/bot"
@@ -16,6 +17,12 @@ import (
 	"github.com/velour/catbase/config"
 )
 
+const (
+	isStreamingTplFallback      = "{{.Name}} is streaming {{.Game}} at {{.URL}}"
+	notStreamingTplFallback     = "{{.Name}} is not streaming"
+	stoppedStreamingTplFallback = "{{.Name}} just stopped streaming"
+)
+
 type TwitchPlugin struct {
 	Bot        bot.Bot
 	config     *config.Config
@@ -112,7 +119,8 @@ func (p *TwitchPlugin) serveStreaming(w http.ResponseWriter, r *http.Request) {
 }
 
 func (p *TwitchPlugin) message(kind bot.Kind, message msg.Message, args ...interface{}) bool {
-	if strings.ToLower(message.Body) == "twitch status" {
+	body := strings.ToLower(message.Body)
+	if body == "twitch status" {
 		channel := message.Channel
 		if users := p.config.GetArray("Twitch."+channel+".Users", []string{}); len(users) > 0 {
 			for _, twitcherName := range users {
@@ -122,13 +130,21 @@ func (p *TwitchPlugin) message(kind bot.Kind, message msg.Message, args ...inter
 			}
 		}
 		return true
+	} else if body == "reset twitch" {
+		p.config.Set("twitch.istpl", isStreamingTplFallback)
+		p.config.Set("twitch.nottpl", notStreamingTplFallback)
+		p.config.Set("twitch.stoppedtpl", stoppedStreamingTplFallback)
 	}
 
 	return false
 }
 
 func (p *TwitchPlugin) help(kind bot.Kind, message msg.Message, args ...interface{}) bool {
-	msg := "There's no help for you here."
+	msg := "You can set the templates for streams with\n"
+	msg += fmt.Sprintf("twitch.istpl (default: %s)\n", isStreamingTplFallback)
+	msg += fmt.Sprintf("twitch.nottpl (default: %s)\n", notStreamingTplFallback)
+	msg += fmt.Sprintf("twitch.stoppedtpl (default: %s)\n", stoppedStreamingTplFallback)
+	msg += "And you can ask who is streaming with `!twitch status`"
 	p.Bot.Send(bot.Message, message.Channel, msg)
 	return true
 }
@@ -216,21 +232,64 @@ func (p *TwitchPlugin) checkTwitch(channel string, twitcher *Twitcher, alwaysPri
 		gameID = games[0].GameID
 		title = games[0].Title
 	}
-	streamWord := p.config.Get("Twitch.StreamWord", "streaming")
+
+	notStreamingTpl := p.config.Get("Twitch.NotTpl", notStreamingTplFallback)
+	isStreamingTpl := p.config.Get("Twitch.IsTpl", isStreamingTplFallback)
+	stoppedStreamingTpl := p.config.Get("Twitch.StoppedTpl", stoppedStreamingTplFallback)
+	buf := bytes.Buffer{}
+
+	info := struct {
+		Name string
+		Game string
+		URL  string
+	}{
+		twitcher.name,
+		title,
+		twitcher.URL(),
+	}
+
 	if alwaysPrintStatus {
 		if gameID == "" {
-			p.Bot.Send(bot.Message, channel, fmt.Sprintf("%s is not %s.", twitcher.name, streamWord))
+			t, err := template.New("notStreaming").Parse(notStreamingTpl)
+			if err != nil {
+				log.Println(err)
+				p.Bot.Send(bot.Message, channel, err)
+				t = template.Must(template.New("notStreaming").Parse(notStreamingTplFallback))
+			}
+			t.Execute(&buf, info)
+			p.Bot.Send(bot.Message, channel, buf.String())
 		} else {
-			p.Bot.Send(bot.Message, channel, fmt.Sprintf("%s is %s %s at %s", twitcher.name, streamWord, title, twitcher.URL()))
+			t, err := template.New("isStreaming").Parse(isStreamingTpl)
+			if err != nil {
+				log.Println(err)
+				p.Bot.Send(bot.Message, channel, err)
+				t = template.Must(template.New("isStreaming").Parse(isStreamingTplFallback))
+			}
+			t.Execute(&buf, info)
+			p.Bot.Send(bot.Message, channel, buf.String())
 		}
 	} else if gameID == "" {
 		if twitcher.gameID != "" {
-			p.Bot.Send(bot.Message, channel, fmt.Sprintf("%s just stopped %s.", twitcher.name, streamWord))
+			t, err := template.New("stoppedStreaming").Parse(stoppedStreamingTpl)
+			if err != nil {
+				log.Println(err)
+				p.Bot.Send(bot.Message, channel, err)
+				t = template.Must(template.New("stoppedStreaming").Parse(stoppedStreamingTplFallback))
+			}
+			t.Execute(&buf, info)
+			p.Bot.Send(bot.Message, channel, buf.String())
 		}
 		twitcher.gameID = ""
 	} else {
 		if twitcher.gameID != gameID {
-			p.Bot.Send(bot.Message, channel, fmt.Sprintf("%s is %s %s at %s", twitcher.name, streamWord, title, twitcher.URL()))
+			t, err := template.New("isStreaming").Parse(isStreamingTpl)
+			if err != nil {
+				log.Println(err)
+				p.Bot.Send(bot.Message, channel, err)
+				t = template.Must(template.New("isStreaming").Parse(isStreamingTplFallback))
+			}
+			t.Execute(&buf, info)
+			p.Bot.Send(bot.Message, channel, buf.String())
 		}
 		twitcher.gameID = gameID
 	}