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..b361137 100644 --- a/wat/bot.go +++ b/wat/bot.go @@ -3,6 +3,7 @@ package wat import ( "crypto/tls" "fmt" + "strconv" "strings" "github.com/go-irc/irc" @@ -17,7 +18,16 @@ type WatBot struct { Nick string } +type BotGame struct { + Games []string + Name string +} + +type BotGameConfig map[string][]string + type WatConfig struct { + BotHosts []string + BotGames BotGameConfig AdminHosts []string IgnoredHosts []string AutoJoinChannels []string @@ -62,6 +72,20 @@ func (w *WatBot) Admin(m *irc.Message) bool { return w.Allowed(m.Prefix.Host, w.c.AdminHosts) } +func (w *WatBot) Bot(m *irc.Message) (bool, []string) { + isBot := w.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 *WatBot) Allowed(c string, r []string) bool { for _, allowed := range r { if c == allowed { @@ -123,6 +147,44 @@ func (w *WatBot) Msg(m *irc.Message) { args = args[1:] } + // integration with games in other bots + isBot, games := w.Bot(m) + if isBot { + // Jeopardy + // parses a message "Top finishers: (nick1: 1300) (nick2: 1200)" from an authorized Jeopardy game bot + if args[0] == "Top" && args[1] == "finishers:" && w.Allowed("jeopardy", games) { + // hey, I avoided regex! + finisherPrizes := strings.Split(strings.Replace(strings.Replace(strings.Replace(strings.Replace(strings.Join(args[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.reply(m, fmt.Sprintf("smartass %s, gave u %d :)", player.Nick, coins)) + player.Coins += coins + w.Db.Update(player) + } + } + + return + } + } + // check if command char (or something weird) is present if args[0] != "wat" && args[0][0] != '#' { return