Compare commits

..

No commits in common. "master" and "systemd" have entirely different histories.

12 changed files with 58 additions and 437 deletions

3
.gitignore vendored
View File

@ -8,9 +8,6 @@
*.so *.so
*.dylib *.dylib
# Normal binary, `go build`
watbot
# Test binary, build with `go test -c` # Test binary, build with `go test -c`
*.test *.test

View File

@ -1,27 +0,0 @@
watbot:
database: wat.db # wat.db (in the working directory) is the default
server:
host: irc.casa # mandatory, no default
port: 6697
tls_verify: true
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
ignores: # optional, no default
hosts:
- annoying.example.com
channels: # optional, no default
join:
- crantest # channels without a prefix character will be prefixed with "#"
permitted:
- '#lucy'

2
count Executable file
View File

@ -0,0 +1,2 @@
wc -l osrc/watbot_*.py
wc -l *.go wat/*.go

6
go.mod
View File

@ -3,12 +3,8 @@ module git.circuitco.de/self/watbot
go 1.15 go 1.15
require ( require (
github.com/creasty/defaults v1.8.0
github.com/ergochat/irc-go v0.4.0
github.com/go-irc/irc v2.1.0+incompatible github.com/go-irc/irc v2.1.0+incompatible
github.com/stretchr/testify v1.9.0 // indirect github.com/namsral/flag v1.7.4-pre
gopkg.in/yaml.v2 v2.4.0 // indirect
gopkg.in/yaml.v3 v3.0.1
gorm.io/driver/sqlite v1.1.4 gorm.io/driver/sqlite v1.1.4
gorm.io/gorm v1.20.11 gorm.io/gorm v1.20.11
) )

28
go.sum
View File

@ -1,10 +1,4 @@
github.com/creasty/defaults v1.8.0 h1:z27FJxCAa0JKt3utc0sCImAEb+spPucmKoOdLHvHYKk= github.com/go-irc/irc v1.3.0 h1:IMD+d/+EzY51ecMLOz73r/NXTZrEp8khrePxRCvX71M=
github.com/creasty/defaults v1.8.0/go.mod h1:iGzKe6pbEHnpMPtfDXZEr0NVxWnPTjb1bbDy08fPzYM=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/ergochat/irc-go v0.4.0 h1:0YibCKfAAtwxQdNjLQd9xpIEPisLcJ45f8FNsMHAuZc=
github.com/ergochat/irc-go v0.4.0/go.mod h1:2vi7KNpIPWnReB5hmLpl92eMywQvuIeIIGdt/FQCph0=
github.com/go-irc/irc v2.1.0+incompatible h1:pg7pMVq5OYQbqTxceByD/EN8VIsba7DtKn49rsCnG8Y= github.com/go-irc/irc v2.1.0+incompatible h1:pg7pMVq5OYQbqTxceByD/EN8VIsba7DtKn49rsCnG8Y=
github.com/go-irc/irc v2.1.0+incompatible/go.mod h1:jJILTRy8s/qOvusiKifAEfhQMVwft1ZwQaVJnnzmyX4= github.com/go-irc/irc v2.1.0+incompatible/go.mod h1:jJILTRy8s/qOvusiKifAEfhQMVwft1ZwQaVJnnzmyX4=
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
@ -13,24 +7,8 @@ github.com/jinzhu/now v1.1.1 h1:g39TucaRWyV3dwDO++eEc6qf8TVIQ/Da48WmqjZ3i7E=
github.com/jinzhu/now v1.1.1/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= github.com/jinzhu/now v1.1.1/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
github.com/mattn/go-sqlite3 v1.14.5 h1:1IdxlwTNazvbKJQSxoJ5/9ECbEeaTTyeU7sEAZ5KKTQ= github.com/mattn/go-sqlite3 v1.14.5 h1:1IdxlwTNazvbKJQSxoJ5/9ECbEeaTTyeU7sEAZ5KKTQ=
github.com/mattn/go-sqlite3 v1.14.5/go.mod h1:WVKg1VTActs4Qso6iwGbiFih2UIHo0ENGwNd0Lj+XmI= github.com/mattn/go-sqlite3 v1.14.5/go.mod h1:WVKg1VTActs4Qso6iwGbiFih2UIHo0ENGwNd0Lj+XmI=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/namsral/flag v1.7.4-pre h1:b2ScHhoCUkbsq0d2C15Mv+VU8bl8hAXV8arnWiOHNZs=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/namsral/flag v1.7.4-pre/go.mod h1:OXldTctbM6SWH1K899kPZcf65KxJiD7MsceFUpB5yDo=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
gorm.io/driver/sqlite v1.1.4 h1:PDzwYE+sI6De2+mxAneV9Xs11+ZyKV6oxD3wDGkaNvM= gorm.io/driver/sqlite v1.1.4 h1:PDzwYE+sI6De2+mxAneV9Xs11+ZyKV6oxD3wDGkaNvM=
gorm.io/driver/sqlite v1.1.4/go.mod h1:mJCeTFr7+crvS+TRnWc5Z3UvwxUN1BGBLMrf5LA9DYw= gorm.io/driver/sqlite v1.1.4/go.mod h1:mJCeTFr7+crvS+TRnWc5Z3UvwxUN1BGBLMrf5LA9DYw=
gorm.io/gorm v1.20.7/go.mod h1:0HFTzE/SqkGTzK6TlDPPQbAYCluiVvhzoA1+aVyzenw= gorm.io/gorm v1.20.7/go.mod h1:0HFTzE/SqkGTzK6TlDPPQbAYCluiVvhzoA1+aVyzenw=

125
main.go
View File

@ -1,121 +1,40 @@
package main package main
import ( import "fmt"
"crypto/tls" import "crypto/tls"
"errors"
"fmt"
"log"
"os"
"flag" import "github.com/go-irc/irc"
"github.com/go-irc/irc" import "github.com/namsral/flag"
"git.circuitco.de/self/watbot/wat" import "git.circuitco.de/self/watbot/wat"
"github.com/creasty/defaults"
"gopkg.in/yaml.v3"
)
type Config struct {
Watbot watConfig `yaml:"watbot"`
}
type watConfig struct {
Database string `default:"wat.db" yaml:"database"`
Nick string `yaml:"nick"`
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"`
Channels struct {
Join []string `yaml:"join"`
Permitted []string `yaml:"permitted"`
} `yaml:"channels"`
Ignores struct {
Hosts []string `yaml:"hosts"`
} `yaml:"ignores"`
Server struct {
Host string `yaml:"host"`
Port int `default:"6697" yaml:"port"`
TlsVerify bool `default:"true" yaml:"tls_verify"`
} `yaml:"server"`
}
func readConfig(configPath string) (*watConfig, error) {
allConfig := Config{}
buffer, err := os.ReadFile(configPath)
if err != nil {
return nil, fmt.Errorf("Could not read configuration file: %s", err)
}
err = yaml.Unmarshal(buffer, &allConfig)
if err != nil {
return nil, fmt.Errorf("Could not parse configuration file: %s", err)
}
config := &allConfig.Watbot
defaults.Set(config)
if config.Server.Host == "" {
return nil, errors.New("Shall I play wattery to know where to connect to?")
}
if config.Name != "" && config.Nick == "" {
config.Nick = config.Name
}
if config.Nick != "" && config.User == "" {
config.User = config.Nick
}
if config.Name == "" || config.Nick == "" || config.User == "" {
return nil, errors.New("Don't know who I am.")
}
return config, nil
}
func main() { func main() {
var configPathArg string pass := flag.String("pass", "", "password")
flag.StringVar(&configPathArg, "config", "config.yaml", "Path to configuration file")
flag.Parse() flag.Parse()
log.Println("Starting with configuration:", configPathArg) fmt.Printf("PASS len %d\n", len(*pass))
config := irc.ClientConfig{
config, err := readConfig(configPathArg) Nick: "watt",
if err != nil { Pass: *pass,
log.Fatalln(err) User: "wat",
os.Exit(1) Name: "wat",
}
ircConfig := irc.ClientConfig{
Nick: config.Nick,
Pass: config.Pass,
User: config.User,
Name: config.Name,
} }
watConfig := wat.WatConfig{ watConfig := wat.WatConfig{
DatabasePath: config.Database, PermittedChannels: []string{
AutoJoinChannels: config.Channels.Join, "#lucy",
PermittedChannels: config.Channels.Permitted, "#sweden",
IgnoredHosts: config.Ignores.Hosts, },
AdminHosts: config.Admins.Hosts, IgnoredHosts: []string{
BotHosts: config.Bots.Hosts, "tripsit/user/creatonez",
BotGames: config.Bots.Games, },
} }
tcpConf := &tls.Config{ tcpConf := &tls.Config{
InsecureSkipVerify: !config.Server.TlsVerify, InsecureSkipVerify: true,
} }
conn, err := tls.Dial("tcp", fmt.Sprintf("%s:%d", config.Server.Host, config.Server.Port), tcpConf) conn, err := tls.Dial("tcp", "127.0.0.1:6697", tcpConf)
if err != nil { if err != nil {
fmt.Println("err " + err.Error()) fmt.Println("err " + err.Error())
return return
} }
wwat := wat.NewWatBot(&ircConfig, &watConfig, conn) wwat := wat.NewWatBot(&config, &watConfig, conn)
wwat.Run() wwat.Run()
} }

View File

@ -6,7 +6,6 @@ import (
"strings" "strings"
"github.com/go-irc/irc" "github.com/go-irc/irc"
"github.com/ergochat/irc-go/ircfmt"
) )
type WatBot struct { type WatBot struct {
@ -14,26 +13,19 @@ type WatBot struct {
conn *tls.Conn conn *tls.Conn
c *WatConfig c *WatConfig
game *WatGame game *WatGame
integration *WatIntegration
Db *WatDb Db *WatDb
Nick string Nick string
} }
type WatConfig struct { type WatConfig struct {
DatabasePath string
BotHosts []string
BotGames BotGameConfig
AdminHosts []string
IgnoredHosts []string
AutoJoinChannels []string
PermittedChannels []string PermittedChannels []string
IgnoredHosts []string
} }
func NewWatBot(config *irc.ClientConfig, watConfig *WatConfig, serverConn *tls.Conn) *WatBot { func NewWatBot(config *irc.ClientConfig, watConfig *WatConfig, serverConn *tls.Conn) *WatBot {
wat := WatBot{conn: serverConn, Nick: config.Nick, c: watConfig} wat := WatBot{conn: serverConn, Nick: config.Nick, c: watConfig}
wat.Db = NewWatDb(watConfig.DatabasePath) wat.Db = NewWatDb()
wat.game = NewWatGame(&wat, wat.Db) 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) config.Handler = irc.HandlerFunc(wat.HandleIrcMsg)
wat.client = irc.NewClient(wat.conn, *config) wat.client = irc.NewClient(wat.conn, *config)
return &wat return &wat
@ -43,29 +35,17 @@ func CleanNick(nick string) string {
return string(nick[0]) + "\u200c" + nick[1:] return string(nick[0]) + "\u200c" + nick[1:]
} }
func PrefixChannel(channel string) string {
// there could theoretically be other channel prefixes ..
if channel[0] != '#' && channel[0] != '!' {
channel = "#" + channel
}
return channel
}
func (w *WatBot) HandleIrcMsg(c *irc.Client, m *irc.Message) { func (w *WatBot) HandleIrcMsg(c *irc.Client, m *irc.Message) {
switch cmd := m.Command; cmd { switch cmd := m.Command; cmd {
case "PING": case "PING":
w.write("PONG", m.Params[0]) w.write("PONG", m.Params[0])
case "PRIVMSG": case "PRIVMSG":
w.Msg(m) w.Msg(m)
case "001":
for _, channel := range w.c.AutoJoinChannels {
w.write("JOIN", PrefixChannel(channel))
}
} }
} }
func (w *WatBot) Admin(m *irc.Message) bool { func (w *WatBot) Admin(m *irc.Message) bool {
return w.Allowed(m.Prefix.Host, w.c.AdminHosts) return m.Prefix.Host == "mph.monster"
} }
func (w *WatBot) Allowed(c string, r []string) bool { func (w *WatBot) Allowed(c string, r []string) bool {
@ -87,7 +67,7 @@ func (w *WatBot) CanRespond(m *irc.Message) bool {
// if !strings.Contains(m.Prefix.Host, "") { // if !strings.Contains(m.Prefix.Host, "") {
// return false // return false
// } // }
if !w.Allowed(PrefixChannel(m.Params[0]), w.c.PermittedChannels) { if !w.Allowed(m.Params[0], w.c.PermittedChannels) {
return false return false
} }
return true return true
@ -106,7 +86,7 @@ func (w *WatBot) Msg(m *irc.Message) {
} }
// fieldsfunc allows you to obtain rune separated fields/args // fieldsfunc allows you to obtain rune separated fields/args
args := strings.FieldsFunc(ircfmt.Strip(m.Params[1]), func(c rune) bool { return c == ' ' }) args := strings.FieldsFunc(m.Params[1], func(c rune) bool { return c == ' ' })
if len(args) == 0 { if len(args) == 0 {
return return
@ -129,11 +109,6 @@ func (w *WatBot) Msg(m *irc.Message) {
args = args[1:] 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 // check if command char (or something weird) is present
if args[0] != "wat" && args[0][0] != '#' { if args[0] != "wat" && args[0][0] != '#' {
return return

View File

@ -52,10 +52,10 @@ type WatDb struct {
db *gorm.DB db *gorm.DB
} }
func NewWatDb(dbpath string) *WatDb { func NewWatDb() *WatDb {
w := WatDb{} w := WatDb{}
var err error var err error
w.db, err = gorm.Open(sqlite.Open(dbpath), &gorm.Config{}) w.db, err = gorm.Open(sqlite.Open("wat.db"), &gorm.Config{})
if err != nil { if err != nil {
panic(err) panic(err)
} }
@ -65,12 +65,8 @@ func NewWatDb(dbpath string) *WatDb {
func (w *WatDb) User(nick, host string, create bool) Player { func (w *WatDb) User(nick, host string, create bool) Player {
var player Player var player Player
query := "nick = ?"
if host != "" {
query = query + " or host = ?"
}
// Try and get a user // Try and get a user
if err := w.db.First(&player, query, nick, host).Error; err != nil && create { if err := w.db.First(&player, "nick = ? or host = ?", nick, host).Error; err != nil && create {
fmt.Printf("Creating user: %s\n", err.Error()) fmt.Printf("Creating user: %s\n", err.Error())
// No user, make another // No user, make another
player.Nick = nick player.Nick = nick

View File

@ -28,7 +28,7 @@ var unconscious = "wat, your hands fumble and fail you. try resting, weakling."
func NewWatGame(bot *WatBot, db *WatDb) *WatGame { func NewWatGame(bot *WatBot, db *WatDb) *WatGame {
g := WatGame{bot, db, Player{}, nil, nil, nil, nil, map[string]int{}} g := WatGame{bot, db, Player{}, nil, nil, nil, nil, map[string]int{}}
g.me = g.db.User(bot.Nick, "", true) g.me = g.db.User(bot.Nick, "amia8t89xfp8y.liberta.casa", true)
g.commands = map[string](func(*Player, []string) string){ g.commands = map[string](func(*Player, []string) string){
//"wat": g.megaWat, //"wat": g.megaWat,
"steroid": g.Steroid, "steroid": g.Steroid,
@ -121,17 +121,9 @@ func (g *WatGame) help() string {
return ret return ret
} }
func (g *WatGame) RandInt(maxx int64) (uint64, error) { func (g *WatGame) RandInt(max int64) uint64 {
bi := big.NewInt(maxx) i, _ := rand.Int(rand.Reader, big.NewInt(max))
// prevent panic of rand.Int on big numbers return i.Uint64()
if bi.BitLen() < 2 {
return 0, fmt.Errorf("overflow, got %d", bi)
}
i, err := rand.Int(rand.Reader, bi)
if err != nil {
return 0, err
}
return i.Uint64(), nil
} }
func (g *WatGame) Heal(player *Player, fields []string) string { func (g *WatGame) Heal(player *Player, fields []string) string {
@ -174,11 +166,8 @@ func (g *WatGame) Dice(player *Player, fields []string) string {
roll = i roll = i
} }
} }
answer, err := g.RandInt(int64(roll)) answer := g.RandInt(int64(roll)) + 1
if e := handleError(err); e != "" { return fmt.Sprintf("1d%d - %d", roll, answer)
return e
}
return fmt.Sprintf("1d%d - %d", roll + 1, answer)
} }
type PositiveError struct{} type PositiveError struct{}
@ -217,20 +206,12 @@ func (g *WatGame) Roll(player *Player, fields []string) string {
} }
lotteryNum := int64(-1) lotteryNum := int64(-1)
if dieSize > 100 { if dieSize > 100 {
lotteryNumRand, randErr := g.RandInt(dieSize) lotteryNum = int64(g.RandInt(dieSize)) + 1
if e := handleError(randErr); e != "" {
return e
}
lotteryNum = int64(lotteryNumRand) + 1
} }
if amount > player.Coins { if amount > player.Coins {
return "wat? brokeass" return "wat? brokeass"
} }
nRand, randErr := g.RandInt(dieSize) n := int64(g.RandInt(dieSize)) + 1
if e := handleError(randErr); e != "" {
return e
}
n := int64(nRand) + 1
ret := fmt.Sprintf("%s rolls the %d sided die... %d! ", player.Nick, dieSize, n) ret := fmt.Sprintf("%s rolls the %d sided die... %d! ", player.Nick, dieSize, n)
if n == lotteryNum { if n == lotteryNum {
player.Coins += player.Coins player.Coins += player.Coins
@ -274,18 +255,11 @@ func (g *WatGame) Punch(player *Player, fields []string) string {
if !target.Conscious() { if !target.Conscious() {
return "wat? you're punching someone who is already unconscious. u crazy?" return "wat? you're punching someone who is already unconscious. u crazy?"
} }
chance, randErr := g.RandInt(6) chance := g.RandInt(6) + 1
if e := handleError(randErr); e != "" { dmg := g.RandInt(6) + 1
return e
}
dmg, randErr := g.RandInt(6)
if e := handleError(randErr); e != "" {
return e
}
dmg = dmg + 1
ret := fmt.Sprintf("%s rolls a d6... %s ", player.Nick, player.Nick) ret := fmt.Sprintf("%s rolls a d6... %s ", player.Nick, player.Nick)
dmg += uint64(player.Level(player.Anarchy)) dmg += uint64(player.Level(player.Anarchy))
if chance + 1 > 3 { if chance > 3 {
ret += fmt.Sprintf("hits %s for %d points of damage! ", target.Nick, dmg) ret += fmt.Sprintf("hits %s for %d points of damage! ", target.Nick, dmg)
target.Health -= int64(dmg) target.Health -= int64(dmg)
g.db.Update(target) g.db.Update(target)
@ -326,11 +300,7 @@ func (g *WatGame) Frame(player *Player, fields []string) string {
if target.Coins < amount { if target.Coins < amount {
return fmt.Sprintf("wat? %s is too poor for this.", target.Nick) return fmt.Sprintf("wat? %s is too poor for this.", target.Nick)
} }
n, randErr := g.RandInt(6) n := g.RandInt(6) + 1
if e := handleError(randErr); e != "" {
return e
}
n = n + 1
ret := fmt.Sprintf("%s rolls a d6 to frame %s with %d %s: It's a %d! (<3 wins). ", player.Nick, target.Nick, amount, currency, n) ret := fmt.Sprintf("%s rolls a d6 to frame %s with %d %s: It's a %d! (<3 wins). ", player.Nick, target.Nick, amount, currency, n)
if n < 3 { if n < 3 {
ret += fmt.Sprintf("You frame %s for a minor crime. They pay me %d.", target.Nick, amount) ret += fmt.Sprintf("You frame %s for a minor crime. They pay me %d.", target.Nick, amount)
@ -365,11 +335,7 @@ func (g *WatGame) Steal(player *Player, fields []string) string {
if target.Coins < amount { if target.Coins < amount {
return fmt.Sprintf("wat? %s is poor and doesn't have that much to steal. (%d %s)", target.Nick, target.Coins, currency) return fmt.Sprintf("wat? %s is poor and doesn't have that much to steal. (%d %s)", target.Nick, target.Coins, currency)
} }
n, randErr := g.RandInt(6) n := g.RandInt(6) + 1
if e := handleError(randErr); e != "" {
return e
}
n = n + 1
ret := fmt.Sprintf("%s is trying to steal %d %s from %s... ", player.Nick, amount, currency, target.Nick) ret := fmt.Sprintf("%s is trying to steal %d %s from %s... ", player.Nick, amount, currency, target.Nick)
if n < 3 { if n < 3 {
ret += "You did it! Sneaky bastard!" ret += "You did it! Sneaky bastard!"
@ -414,11 +380,7 @@ func (g *WatGame) Leech(player *Player, fields []string) string {
if err != "" { if err != "" {
return err return err
} }
r, randErr := g.RandInt(10) r := g.RandInt(10) + 1
if e := handleError(randErr); e != "" {
return e
}
r = r + 1
reply := fmt.Sprintf("You muster your wealth and feed it to %s. ", g.bot.Nick) reply := fmt.Sprintf("You muster your wealth and feed it to %s. ", g.bot.Nick)
hpDown := amount / divisor hpDown := amount / divisor
player.Coins -= amount player.Coins -= amount
@ -446,11 +408,7 @@ func (g *WatGame) Rest(player *Player, fields []string) string {
} else if delta < minRest { } else if delta < minRest {
ret = fmt.Sprintf("wat were you thinking, sleeping at a time like this (%d until next rest)", minRest-delta) ret = fmt.Sprintf("wat were you thinking, sleeping at a time like this (%d until next rest)", minRest-delta)
} else { } else {
value, randErr := g.RandInt(10) value := g.RandInt(10) + 1
if e := handleError(randErr); e != "" {
return e
}
value = value + 1
if player.Health < -5 { if player.Health < -5 {
player.Health = 1 player.Health = 1
ret = fmt.Sprintf("wow ur beat up. i pity u, ur health is now 1.") ret = fmt.Sprintf("wow ur beat up. i pity u, ur health is now 1.")
@ -477,15 +435,8 @@ func (g *WatGame) Bench(player *Player, fields []string) string {
if !g.CanAct(player, Action_Lift, minTime) { if !g.CanAct(player, Action_Lift, minTime) {
return "you're tired. no more lifting for now." return "you're tired. no more lifting for now."
} }
weight, randErr := g.RandInt(370) weight := g.RandInt(370) + 50
if e := handleError(randErr); e != "" { reps := g.RandInt(10)
return e
}
weight = weight + 50
reps, randErr := g.RandInt(10)
if e := handleError(randErr); e != "" {
return e
}
value := int64(0) value := int64(0)
reply := fmt.Sprintf("%s benches %dwatts for %d reps, ", player.Nick, weight, reps) reply := fmt.Sprintf("%s benches %dwatts for %d reps, ", player.Nick, weight, reps)
if weight < 150 { if weight < 150 {
@ -501,10 +452,7 @@ func (g *WatGame) Bench(player *Player, fields []string) string {
} }
if g.roid[player.Nick] != 0 { if g.roid[player.Nick] != 0 {
delete(g.roid, player.Nick) delete(g.roid, player.Nick)
success, randErr := g.RandInt(2) success := g.RandInt(2)
if e := handleError(randErr); e != "" {
return e
}
if success != 0 { if success != 0 {
player.Health = 0 player.Health = 0
player.Anarchy -= 10 player.Anarchy -= 10
@ -527,10 +475,7 @@ func (g *WatGame) Riot(player *Player, fields []string) string {
if !g.CanAct(player, Action_Riot, int64((48 * time.Hour).Seconds())) { if !g.CanAct(player, Action_Riot, int64((48 * time.Hour).Seconds())) {
return "Planning a riot takes time and the right circumstances. Be prepared. (nothing happens)" return "Planning a riot takes time and the right circumstances. Be prepared. (nothing happens)"
} }
r, randErr := g.RandInt(100) r := g.RandInt(100)
if e := handleError(randErr); e != "" {
return e
}
reply := "" reply := ""
if r > 40 { if r > 40 {
player.Anarchy += 3 player.Anarchy += 3
@ -676,21 +621,9 @@ func PrintTwo(nick string, value uint64) string {
} }
func (g *WatGame) megaWat(player *Player, _ []string) string { func (g *WatGame) megaWat(player *Player, _ []string) string {
mega, randErr := g.RandInt(1000000) mega := g.RandInt(1000000) + 1
if e := handleError(randErr); e != "" { kilo := g.RandInt(1000) + 1
return e + " mega fail" ten := g.RandInt(100) + 1
}
mega = mega + 1
kilo, randErr := g.RandInt(1000)
if e := handleError(randErr); e != "" {
return e + " kilo fail"
}
kilo = kilo + 1
ten, randErr := g.RandInt(100)
if e := handleError(randErr); e != "" {
return e + " ten fail"
}
ten = ten + 1
reply := "" reply := ""
if mega == 23 { if mega == 23 {
player.Coins += 1000000 player.Coins += 1000000

View File

@ -1,126 +0,0 @@
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)
var msg string
var many bool
fiprcount := len(finisherPrizes)
cashoutcount := 0
// only a single winner
if fiprcount == 1 {
msg = "smartass %s :) gave u %d"
many = false
// multiple winners
} else if fiprcount > 1 {
msg = "gang of smartasses :) gave %s %d"
many = true
// no winners (should never get here)
} else {
fmt.Printf("Empty finishers, aborting Jeopardy processing")
return
}
// 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
if coins == 0 {
continue
}
cashoutcount += 1
// 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(strings.ToLower(name), "", false)
if player.Nick == "" {
fmt.Printf("Player %s does not exist in Watbot, skipping cashout.\n", name)
continue
}
// fill previous format placeholders
msg = fmt.Sprintf(msg, player.Nick, coins)
if many {
// append additional ones for filling in the next loop iteration
msg = msg + ", %s %d"
}
player.Coins += coins
w.db.Update(player)
}
if many {
// remove format placeholders from last loop iteration
msg = strings.Replace(msg, ", %s %d", ".", 1)
}
if cashoutcount > 0 {
w.bot.reply(m, msg)
}
}

View File

@ -1,22 +0,0 @@
package wat
import (
"fmt"
"runtime"
)
func handleError(err error) string {
if err != nil {
pc, _, _, ok := runtime.Caller(1)
details := runtime.FuncForPC(pc)
var cFun string
if ok && details != nil {
cFun = details.Name()
} else {
cFun = "???"
}
fmt.Printf("caught error in %s: %v\n", cFun, err)
return "u wat"
}
return ""
}

BIN
watbot Executable file

Binary file not shown.