pubsub: add counter pub

bot/handlers.go

@@ -3,10 +3,14 @@
 package bot
 
 import (
+	"bytes"
 	"database/sql"
+	"encoding/json"
 	"errors"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
+	"net/http"
 	"reflect"
 	"regexp"
 	"strconv"
@@ -263,3 +267,25 @@ func (b *bot) selfSaid(conn Connector, channel, message string, action bool) {
 		}
 	}
 }
+
+// PubToASub sends updates to subscribed URLs
+func (b *bot) PubToASub(subject string, payload interface{}) {
+	key := fmt.Sprintf("pubsub.%s.url", subject)
+	subs := b.config.GetArray(key, []string{})
+	if len(subs) == 0 {
+		return
+	}
+	encodedBody, _ := json.Marshal(struct {
+		Payload interface{} `json:"payload"`
+	}{payload})
+	body := bytes.NewBuffer(encodedBody)
+	for _, url := range subs {
+		r, err := http.Post(url, "text/json", body)
+		if err != nil {
+			log.Error().Err(err).Msg("")
+		} else {
+			body, _ := ioutil.ReadAll(r.Body)
+			log.Debug().Msgf("Response body: %s", string(body))
+		}
+	}
+}

bot/interfaces.go

@@ -172,6 +172,9 @@ type Bot interface {
 
 	// Check valid password
 	CheckPassword(secret, password string) bool
+
+	// PubToASub publishes a message to any subscribers
+	PubToASub(subject string, payload interface{})
 }
 
 // Connector represents a server connection to a chat service

bot/mock.go

@@ -118,11 +118,12 @@ func NewMockBot() *MockBot {
 	return &b
 }
 
-func (mb *MockBot) GetPluginNames() []string                   { return nil }
+func (mb *MockBot) GetPluginNames() []string                      { return nil }
-func (mb *MockBot) RefreshPluginBlacklist() error              { return nil }
+func (mb *MockBot) RefreshPluginBlacklist() error                 { return nil }
-func (mb *MockBot) RefreshPluginWhitelist() error              { return nil }
+func (mb *MockBot) RefreshPluginWhitelist() error                 { return nil }
-func (mb *MockBot) GetWhitelist() []string                     { return []string{} }
+func (mb *MockBot) GetWhitelist() []string                        { return []string{} }
-func (mb *MockBot) OnBlacklist(ch, p string) bool              { return false }
+func (mb *MockBot) OnBlacklist(ch, p string) bool                 { return false }
-func (mb *MockBot) URLFormat(title, url string) string         { return title + url }
+func (mb *MockBot) URLFormat(title, url string) string            { return title + url }
-func (mb *MockBot) CheckPassword(secret, password string) bool { return true }
+func (mb *MockBot) CheckPassword(secret, password string) bool    { return true }
-func (mb *MockBot) ListenAndServe()                            {}
+func (mb *MockBot) ListenAndServe()                               {}
+func (mb *MockBot) PubToASub(subject string, payload interface{}) {}

plugins/counter/api.go

@@ -160,7 +160,7 @@ func (p *CounterPlugin) handleCounterAPI(w http.ResponseWriter, r *http.Request)
 }
 
 type Update struct {
-	Who    string
+	Who    string `json:"who"`
-	What   string
+	What   string `json:"what"`
-	Amount int
+	Amount int    `json:"amount"`
 }

plugins/counter/counter.go

@@ -344,6 +344,11 @@ func New(b bot.Bot) *CounterPlugin {
 	b.Register(cp, bot.Help, cp.help)
 	cp.registerWeb()
 
+	RegisterUpdate(func(r bot.Request, u Update) {
+		log.Debug().Msgf("Publishing update %v", u)
+		b.PubToASub("counter", u)
+	})
+
 	return cp
 }