Refactor integrations

Move to a separate file for better code structure and to avoid huge
branching inside Msg().

Signed-off-by: Georg Pfuetzenreuter <mail@georg-pfuetzenreuter.net>

config.example.yaml

@@ -6,6 +6,12 @@ watbot:
   name: watest
   nick: watest  # nick is name by default
   user: watest  # user is nick by default
+  bots:  # optional, no default
+    games:  # mapping of bot names to games
+      katyusha:
+        - jeopardy  # currently jeopardy is the only integrated game
+    hosts:  # hostmasks considered as valid bots
+      - bot.example.com
   admins:  # optional, no default
     hosts:
       - admin.example.com

main.go

@@ -24,6 +24,10 @@ type watConfig struct {
 	Pass		string		`yaml:"pass"`
 	User		string		`yaml:"user"`
 	Name		string		`yaml:"name"`
+	Bots            struct {
+		Hosts []string		`yaml:"hosts"`
+		Games wat.BotGameConfig	`yaml:"games"`
+	} `yaml:"bots"`
 	Admins		struct {
 		Hosts []string		`yaml:"hosts"`
 	}				`yaml:"admins"`
@@ -99,6 +103,8 @@ func main() {
 		PermittedChannels: config.Channels.Permitted,
 		IgnoredHosts:      config.Ignores.Hosts,
 		AdminHosts:        config.Admins.Hosts,
+		BotHosts:          config.Bots.Hosts,
+		BotGames:          config.Bots.Games,
 	}
 	tcpConf := &tls.Config{
 		InsecureSkipVerify: !config.Server.TlsVerify,

wat/bot.go

@@ -13,11 +13,14 @@ type WatBot struct {
 	conn   *tls.Conn
 	c      *WatConfig
 	game   *WatGame
+	integration *WatIntegration
 	Db     *WatDb
 	Nick   string
 }
 
 type WatConfig struct {
+	BotHosts          []string
+	BotGames          BotGameConfig
 	AdminHosts        []string
 	IgnoredHosts      []string
 	AutoJoinChannels  []string
@@ -28,6 +31,7 @@ func NewWatBot(config *irc.ClientConfig, watConfig *WatConfig, serverConn *tls.C
 	wat := WatBot{conn: serverConn, Nick: config.Nick, c: watConfig}
 	wat.Db = NewWatDb()
 	wat.game = NewWatGame(&wat, wat.Db)
+	wat.integration = NewWatIntegration(&wat, wat.Db, &WatIntegrationConfig{BotHosts: watConfig.BotHosts, BotGames: watConfig.BotGames})
 	config.Handler = irc.HandlerFunc(wat.HandleIrcMsg)
 	wat.client = irc.NewClient(wat.conn, *config)
 	return &wat
@@ -123,6 +127,14 @@ func (w *WatBot) Msg(m *irc.Message) {
 		args = args[1:]
 	}
 
+	// integration with games in other bots
+	isBot, games := w.integration.Bot(m)
+	if isBot {
+		if w.integration.HandleIntegration(m, args, games) {
+			return
+		}
+	}
+
 	// check if command char (or something weird) is present
 	if args[0] != "wat" && args[0][0] != '#' {
 		return

wat/integration.go

@@ -0,0 +1,81 @@
+package wat
+
+import (
+	"fmt"
+	"strconv"
+	"strings"
+
+	"github.com/go-irc/irc"
+)
+
+type BotGameConfig map[string][]string
+
+type WatIntegrationConfig struct {
+	BotHosts []string
+	BotGames BotGameConfig
+}
+
+type WatIntegration struct {
+	bot *WatBot
+	db  *WatDb
+	c   *WatIntegrationConfig
+}
+
+func NewWatIntegration(bot *WatBot, db *WatDb, c *WatIntegrationConfig) *WatIntegration {
+	return &WatIntegration{bot, db, c}
+}
+
+func (w *WatIntegration) Bot(m *irc.Message) (bool, []string) {
+	isBot := w.bot.Allowed(m.Prefix.Host, w.c.BotHosts)
+	var games []string
+	if isBot {
+		for b, g := range w.c.BotGames {
+			if b == m.Prefix.Name {
+				games = g
+				break
+			}
+		}
+	}
+	return isBot, games
+}
+
+func (w *WatIntegration) HandleIntegration(m *irc.Message, msgargs []string, games []string) bool {
+
+	// handles a message "Top finishers: (nick1: 1300) (nick2: 1200)" from an authorized Jeopardy game bot
+	if msgargs[0] == "Top" && msgargs[1] == "finishers:" && w.bot.Allowed("jeopardy", games) {
+		w.Jeopardy(m, msgargs)
+		return true
+	}
+
+	return false
+}
+
+func (w *WatIntegration) Jeopardy(m *irc.Message, msgargs []string) {
+	// hey, I avoided regex!
+	finisherPrizes := strings.Split(strings.Replace(strings.Replace(strings.Replace(strings.Replace(strings.Join(msgargs[2:], " "), ") (", ";", -1), ": ", ":", -1), "(", "", 1), ")", "", 1), ";")
+	fmt.Printf("Processing Jeopardy: %s\n", finisherPrizes)
+	for _, pair := range finisherPrizes {
+		nameCoinPair := strings.Split(pair, ":")
+		coins, err := strconv.ParseUint(nameCoinPair[1], 10, 64)
+		if err != nil {
+			fmt.Printf("Invalid coins, cannot process pair for cashout: %s.\n", nameCoinPair)
+			continue
+		}
+		name := nameCoinPair[0]
+		// Jeopardy prizes are quite a lot of $$$, make it a bit more sane
+		coins = coins / 40
+		// name   = we assume the Jeopardy player name to match a Watbot player name
+		// host   = we could use some WHO logic to find the host, but assuming nickname lookup to be sufficient here
+		// create = based on the above, maybe rather not create Watbot players based on only a nick?
+		//          but it expects someone to have played with Watbot before to be eligible for Jeopardy cashout ..
+		player := w.db.User(name, "", false)
+		if player.Nick == "" {
+			fmt.Printf("Player %s does not exist in Watbot, skipping cashout.\n", name)
+			continue
+		} else {
+			w.bot.reply(m, fmt.Sprintf("smartass %s, gave u %d :)", player.Nick, coins))
+			player.Coins += coins
+			w.db.Update(player)
+		}
+	}
+}