diff --git a/config.example.yaml b/config.example.yaml index 5d3fbac..02dc9dd 100644 --- a/config.example.yaml +++ b/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 diff --git a/main.go b/main.go index c242571..fb71444 100644 --- a/main.go +++ b/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, diff --git a/wat/bot.go b/wat/bot.go index a555de4..41c030f 100644 --- a/wat/bot.go +++ b/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,11 @@ func (w *WatBot) Msg(m *irc.Message) { args = args[1:] } + // integration with games in other bots + if w.integration.HandleIntegration(m, args) { + return + } + // check if command char (or something weird) is present if args[0] != "wat" && args[0][0] != '#' { return diff --git a/wat/integration.go b/wat/integration.go new file mode 100644 index 0000000..b4ff8e9 --- /dev/null +++ b/wat/integration.go @@ -0,0 +1,94 @@ +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) bool { + isBot, games := w.Bot(m) + if isBot { + // 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 + } + } + // not an authorized bot or no integration matched the given message + return false +} + +func (w *WatIntegration) Jeopardy(m *irc.Message, msgargs []string) { + // hey, I avoided regex! + // 1. Starts parsing an array of message arguments containing "Top finishers: (nick1: 1000) (nick2: 2000)", where + // the "($nick: $value)" pairs can contain arbitrary nicknames + integer values and can repeat one to any amount of times + // 2. Join the array on spaces to a string, but skip the first two elements to remove "Top" and "finishers:" + // 3. Replace ") (" in the string with ";" - the semicolon is chosen as a temporary delimiter because it does not conflict with any other characters in the message + // 4. Replace ": " in the string with ":" + // 5. Replace "(" in the string with "" (relevant for the first nick/value pair) + // 6. Replace ")" in the string with "" (relevant for the last nick/value pair) + // 7. Now, we have a string like "nick1:1000;nick2:2000" - split it back into an array on ";" + // 8. The result is an array like "[nick1:1000, nick2:2000]" + 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) + // iterate over the "$nick:$value" string elements + for _, pair := range finisherPrizes { + // turn the string element into an array, where the first entry is the nickname, and the second the value + 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) + } + } +}