From 898057cc20edf918211074fcd81416494ed37db3 Mon Sep 17 00:00:00 2001
From: Kane York <rikingcoding@gmail.com>
Date: Mon, 26 Oct 2015 10:23:53 -0700
Subject: [PATCH] rework /pubmsg to use MessageTargetType

---
 socketserver/internal/server/backend.go    | 33 ++++++++++++---
 socketserver/internal/server/backlog.go    | 49 +++++++++++++---------
 socketserver/internal/server/handlecore.go |  3 +-
 3 files changed, 58 insertions(+), 27 deletions(-)

diff --git a/socketserver/internal/server/backend.go b/socketserver/internal/server/backend.go
index 70c826b2..00a55293 100644
--- a/socketserver/internal/server/backend.go
+++ b/socketserver/internal/server/backend.go
@@ -76,17 +76,38 @@ func HBackendPublishRequest(w http.ResponseWriter, r *http.Request) {
 	json := formData.Get("args")
 	channel := formData.Get("channel")
 	scope := formData.Get("scope")
+
+	target := MessageTargetTypeByName(scope)
+
+	if cmd == "" {
+		w.WriteHeader(422)
+		fmt.Fprintf(w, "Error: cmd cannot be blank")
+		return
+	}
+	if channel == "" && (target == MsgTargetTypeChat || target == MsgTargetTypeMultichat || target == MsgTargetTypeWatching) {
+		w.WriteHeader(422)
+		fmt.Fprintf(w, "Error: channel must be specified")
+		return
+	}
+
 	cm := ClientMessage{MessageID: -1, Command: Command(cmd), origArguments: json}
 	var count int
-	if scope == "chat" {
+
+	switch target {
+	case MsgTargetTypeSingle:
+		// TODO
+	case MsgTargetTypeChat:
 		count = PublishToChat(channel, cm)
-	} else if scope == "channel" {
+	case MsgTargetTypeMultichat:
+		// TODO
+	case MsgTargetTypeWatching:
 		count = PublishToWatchers(channel, cm)
-	} else if scope == "global" {
+	case MsgTargetTypeGlobal:
 		count = PublishToAll(cm)
-	} else {
-		w.WriteHeader(400)
-		fmt.Fprint(w, "Need to specify either chat or channel")
+	case MsgTargetTypeInvalid:
+	default:
+		w.WriteHeader(422)
+		fmt.Fprint(w, "Invalid 'scope'. must be single, chat, multichat, channel, or global")
 		return
 	}
 	fmt.Fprint(w, count)
diff --git a/socketserver/internal/server/backlog.go b/socketserver/internal/server/backlog.go
index 82fd500c..24e0b6be 100644
--- a/socketserver/internal/server/backlog.go
+++ b/socketserver/internal/server/backlog.go
@@ -6,33 +6,37 @@ import (
 	"net/http"
 )
 
+type PushCommandCacheInfo struct {
+	Caching BacklogCacheType
+	Target  MessageTargetType
+}
+
 // this value is just docs right now
-var ServerInitiatedCommands = []string{
+var ServerInitiatedCommands = map[string]PushCommandCacheInfo{
 	/// Global updates & notices
-	"update_news", // timecache:global
-	"message",     // timecache:global
-	"reload_ff",   // timecache:global
+	"update_news": {CacheTypeTimestamps, MsgTargetTypeGlobal}, // timecache:global
+	"message":     {CacheTypeTimestamps, MsgTargetTypeGlobal}, // timecache:global
+	"reload_ff":   {CacheTypeTimestamps, MsgTargetTypeGlobal}, // timecache:global
 
 	/// Emote updates
-	"reload_badges", // timecache:global
-	"set_badge",     // timecache:multichat
-	"reload_set",    // timecache:multichat
-	"load_set",      // TODO what are the semantics of this?
+	"reload_badges": {CacheTypeTimestamps, MsgTargetTypeGlobal},    // timecache:global
+	"set_badge":     {CacheTypeTimestamps, MsgTargetTypeMultichat}, // timecache:multichat
+	"reload_set":    {CacheTypeTimestamps, MsgTargetTypeMultichat}, // timecache:multichat
+	"load_set":      {},                                            // TODO what are the semantics of this?
 
 	/// User auth
-	"do_authorize", // nocache:single
+	"do_authorize": {CacheTypeNever, MsgTargetTypeSingle}, // nocache:single
 
 	/// Channel data
-	// extra emote sets included in the chat
-	"follow_sets", // mustcache:chat
-	// extra follow buttons below the stream
-	"follow_buttons", // mustcache:watching
-	// SRL race data
-	"srl_race", // cachelast:watching
+	// follow_sets: extra emote sets included in the chat
+	// follow_buttons: extra follow buttons below the stream
+	"follow_sets":    {CacheTypePersistent, MsgTargetTypeChat},     // mustcache:chat
+	"follow_buttons": {CacheTypePersistent, MsgTargetTypeWatching}, // mustcache:watching
+	"srl_race":       {CacheTypeLastOnly, MsgTargetTypeWatching},   // cachelast:watching
 
 	/// Chatter/viewer counts
-	"chatters", // cachelast:watching
-	"viewers",  // cachelast:watching
+	"chatters": {CacheTypeLastOnly, MsgTargetTypeWatching}, // cachelast:watching
+	"viewers":  {CacheTypeLastOnly, MsgTargetTypeWatching}, // cachelast:watching
 }
 var _ = ServerInitiatedCommands
 
@@ -70,15 +74,15 @@ const (
 	MsgTargetTypeGlobal
 )
 
+// note: see types.go for methods on these
+
 // Returned by BacklogCacheType.UnmarshalJSON()
 var ErrorUnrecognizedCacheType = errors.New("Invalid value for cachetype")
 
 // Returned by MessageTargetType.UnmarshalJSON()
 var ErrorUnrecognizedTargetType = errors.New("Invalid value for message target")
 
-// note: see types.go for methods on these
-
-func HBackendSaveBacklog(w http.ResponseWriter, r *http.Request) {
+func HBackendUpdateAndPublish(w http.ResponseWriter, r *http.Request) {
 	formData, err := UnsealRequest(r.Form)
 	if err != nil {
 		w.WriteHeader(403)
@@ -86,4 +90,9 @@ func HBackendSaveBacklog(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	cmd := formData.Get("command")
+	cacheinfo, ok := ServerInitiatedCommands[cmd]
+	if !ok {
+		w.WriteHeader(422)
+	}
 }
diff --git a/socketserver/internal/server/handlecore.go b/socketserver/internal/server/handlecore.go
index 910bb4d6..f4b40653 100644
--- a/socketserver/internal/server/handlecore.go
+++ b/socketserver/internal/server/handlecore.go
@@ -127,7 +127,8 @@ func SetupServerAndHandle(config *Config, tlsConfig *tls.Config, serveMux *http.
 		serveMux = http.DefaultServeMux
 	}
 	serveMux.HandleFunc("/", sockServer.ServeHTTP)
-	serveMux.HandleFunc("/pub", HBackendPublishRequest)
+	serveMux.HandleFunc("/pubmsg", HBackendPublishRequest)
+	serveMux.HandleFunc("/updatepub", HBackendUpdateAndPublish)
 }
 
 // Handle a new websocket connection from a FFZ client.