Compare commits
26 Commits
Author | SHA1 | Date | |
---|---|---|---|
67ef6fec1b | |||
930839ab58 | |||
91b0e21b7a | |||
dfe7deff72 | |||
e300f71370 | |||
5ce4d4bfe4 | |||
c4ff70da70 | |||
7ec49a9769 | |||
b6187e0077 | |||
accf16c92a | |||
394d6ca2fa | |||
d99a0d84bf | |||
a5cc5e0317 | |||
89ed59a9c7 | |||
d06e724f06 | |||
97e9d7d0c2 | |||
a475bc2f42 | |||
0a90b6e483 | |||
cc322e87e6 | |||
bde9a8defb | |||
f4a9607770 | |||
dbaf6f046a | |||
331d38694e | |||
bd7d3abf24 | |||
be1b15b6f9 | |||
84696672cd |
3
.gitignore
vendored
3
.gitignore
vendored
@ -8,6 +8,9 @@
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Normal binary, `go build`
|
||||
watbot
|
||||
|
||||
# Test binary, build with `go test -c`
|
||||
*.test
|
||||
|
||||
|
27
config.example.yaml
Normal file
27
config.example.yaml
Normal file
@ -0,0 +1,27 @@
|
||||
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'
|
||||
|
6
go.mod
6
go.mod
@ -3,8 +3,12 @@ module git.circuitco.de/self/watbot
|
||||
go 1.15
|
||||
|
||||
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/namsral/flag v1.7.4-pre
|
||||
github.com/stretchr/testify v1.9.0 // indirect
|
||||
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1
|
||||
gorm.io/driver/sqlite v1.1.4
|
||||
gorm.io/gorm v1.20.11
|
||||
)
|
||||
|
28
go.sum
28
go.sum
@ -1,4 +1,10 @@
|
||||
github.com/go-irc/irc v1.3.0 h1:IMD+d/+EzY51ecMLOz73r/NXTZrEp8khrePxRCvX71M=
|
||||
github.com/creasty/defaults v1.8.0 h1:z27FJxCAa0JKt3utc0sCImAEb+spPucmKoOdLHvHYKk=
|
||||
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/go.mod h1:jJILTRy8s/qOvusiKifAEfhQMVwft1ZwQaVJnnzmyX4=
|
||||
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
|
||||
@ -7,8 +13,24 @@ 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/mattn/go-sqlite3 v1.14.5 h1:1IdxlwTNazvbKJQSxoJ5/9ECbEeaTTyeU7sEAZ5KKTQ=
|
||||
github.com/mattn/go-sqlite3 v1.14.5/go.mod h1:WVKg1VTActs4Qso6iwGbiFih2UIHo0ENGwNd0Lj+XmI=
|
||||
github.com/namsral/flag v1.7.4-pre h1:b2ScHhoCUkbsq0d2C15Mv+VU8bl8hAXV8arnWiOHNZs=
|
||||
github.com/namsral/flag v1.7.4-pre/go.mod h1:OXldTctbM6SWH1K899kPZcf65KxJiD7MsceFUpB5yDo=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
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/go.mod h1:mJCeTFr7+crvS+TRnWc5Z3UvwxUN1BGBLMrf5LA9DYw=
|
||||
gorm.io/gorm v1.20.7/go.mod h1:0HFTzE/SqkGTzK6TlDPPQbAYCluiVvhzoA1+aVyzenw=
|
||||
|
125
main.go
125
main.go
@ -1,40 +1,121 @@
|
||||
package main
|
||||
|
||||
import "fmt"
|
||||
import "crypto/tls"
|
||||
import (
|
||||
"crypto/tls"
|
||||
"errors"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
import "github.com/go-irc/irc"
|
||||
import "github.com/namsral/flag"
|
||||
"flag"
|
||||
"github.com/go-irc/irc"
|
||||
|
||||
import "git.circuitco.de/self/watbot/wat"
|
||||
"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() {
|
||||
pass := flag.String("pass", "", "password")
|
||||
var configPathArg string
|
||||
flag.StringVar(&configPathArg, "config", "config.yaml", "Path to configuration file")
|
||||
flag.Parse()
|
||||
fmt.Printf("PASS len %d\n", len(*pass))
|
||||
config := irc.ClientConfig{
|
||||
Nick: "watt",
|
||||
Pass: *pass,
|
||||
User: "wat",
|
||||
Name: "wat",
|
||||
log.Println("Starting with configuration:", configPathArg)
|
||||
|
||||
config, err := readConfig(configPathArg)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
ircConfig := irc.ClientConfig{
|
||||
Nick: config.Nick,
|
||||
Pass: config.Pass,
|
||||
User: config.User,
|
||||
Name: config.Name,
|
||||
}
|
||||
watConfig := wat.WatConfig{
|
||||
PermittedChannels: []string{
|
||||
"#lucy",
|
||||
"#sweden",
|
||||
},
|
||||
IgnoredHosts: []string{
|
||||
"tripsit/user/creatonez",
|
||||
},
|
||||
DatabasePath: config.Database,
|
||||
AutoJoinChannels: config.Channels.Join,
|
||||
PermittedChannels: config.Channels.Permitted,
|
||||
IgnoredHosts: config.Ignores.Hosts,
|
||||
AdminHosts: config.Admins.Hosts,
|
||||
BotHosts: config.Bots.Hosts,
|
||||
BotGames: config.Bots.Games,
|
||||
}
|
||||
tcpConf := &tls.Config{
|
||||
InsecureSkipVerify: true,
|
||||
InsecureSkipVerify: !config.Server.TlsVerify,
|
||||
}
|
||||
conn, err := tls.Dial("tcp", "127.0.0.1:6697", tcpConf)
|
||||
conn, err := tls.Dial("tcp", fmt.Sprintf("%s:%d", config.Server.Host, config.Server.Port), tcpConf)
|
||||
if err != nil {
|
||||
fmt.Println("err " + err.Error())
|
||||
return
|
||||
}
|
||||
wwat := wat.NewWatBot(&config, &watConfig, conn)
|
||||
wwat := wat.NewWatBot(&ircConfig, &watConfig, conn)
|
||||
wwat.Run()
|
||||
}
|
||||
|
35
wat/bot.go
35
wat/bot.go
@ -6,6 +6,7 @@ import (
|
||||
"strings"
|
||||
|
||||
"github.com/go-irc/irc"
|
||||
"github.com/ergochat/irc-go/ircfmt"
|
||||
)
|
||||
|
||||
type WatBot struct {
|
||||
@ -13,19 +14,26 @@ type WatBot struct {
|
||||
conn *tls.Conn
|
||||
c *WatConfig
|
||||
game *WatGame
|
||||
integration *WatIntegration
|
||||
Db *WatDb
|
||||
Nick string
|
||||
}
|
||||
|
||||
type WatConfig struct {
|
||||
PermittedChannels []string
|
||||
DatabasePath string
|
||||
BotHosts []string
|
||||
BotGames BotGameConfig
|
||||
AdminHosts []string
|
||||
IgnoredHosts []string
|
||||
AutoJoinChannels []string
|
||||
PermittedChannels []string
|
||||
}
|
||||
|
||||
func NewWatBot(config *irc.ClientConfig, watConfig *WatConfig, serverConn *tls.Conn) *WatBot {
|
||||
wat := WatBot{conn: serverConn, Nick: config.Nick, c: watConfig}
|
||||
wat.Db = NewWatDb()
|
||||
wat.Db = NewWatDb(watConfig.DatabasePath)
|
||||
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
|
||||
@ -35,17 +43,29 @@ func CleanNick(nick string) string {
|
||||
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) {
|
||||
switch cmd := m.Command; cmd {
|
||||
case "PING":
|
||||
w.write("PONG", m.Params[0])
|
||||
case "PRIVMSG":
|
||||
w.Msg(m)
|
||||
case "001":
|
||||
for _, channel := range w.c.AutoJoinChannels {
|
||||
w.write("JOIN", PrefixChannel(channel))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (w *WatBot) Admin(m *irc.Message) bool {
|
||||
return m.Prefix.Host == "mph.monster"
|
||||
return w.Allowed(m.Prefix.Host, w.c.AdminHosts)
|
||||
}
|
||||
|
||||
func (w *WatBot) Allowed(c string, r []string) bool {
|
||||
@ -67,7 +87,7 @@ func (w *WatBot) CanRespond(m *irc.Message) bool {
|
||||
// if !strings.Contains(m.Prefix.Host, "") {
|
||||
// return false
|
||||
// }
|
||||
if !w.Allowed(m.Params[0], w.c.PermittedChannels) {
|
||||
if !w.Allowed(PrefixChannel(m.Params[0]), w.c.PermittedChannels) {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
@ -86,7 +106,7 @@ func (w *WatBot) Msg(m *irc.Message) {
|
||||
}
|
||||
|
||||
// fieldsfunc allows you to obtain rune separated fields/args
|
||||
args := strings.FieldsFunc(m.Params[1], func(c rune) bool { return c == ' ' })
|
||||
args := strings.FieldsFunc(ircfmt.Strip(m.Params[1]), func(c rune) bool { return c == ' ' })
|
||||
|
||||
if len(args) == 0 {
|
||||
return
|
||||
@ -109,6 +129,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
|
||||
|
10
wat/db.go
10
wat/db.go
@ -52,10 +52,10 @@ type WatDb struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func NewWatDb() *WatDb {
|
||||
func NewWatDb(dbpath string) *WatDb {
|
||||
w := WatDb{}
|
||||
var err error
|
||||
w.db, err = gorm.Open(sqlite.Open("wat.db"), &gorm.Config{})
|
||||
w.db, err = gorm.Open(sqlite.Open(dbpath), &gorm.Config{})
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
@ -65,8 +65,12 @@ func NewWatDb() *WatDb {
|
||||
|
||||
func (w *WatDb) User(nick, host string, create bool) Player {
|
||||
var player Player
|
||||
query := "nick = ?"
|
||||
if host != "" {
|
||||
query = query + " or host = ?"
|
||||
}
|
||||
// Try and get a user
|
||||
if err := w.db.First(&player, "nick = ? or host = ?", nick, host).Error; err != nil && create {
|
||||
if err := w.db.First(&player, query, nick, host).Error; err != nil && create {
|
||||
fmt.Printf("Creating user: %s\n", err.Error())
|
||||
// No user, make another
|
||||
player.Nick = nick
|
||||
|
111
wat/game.go
111
wat/game.go
@ -28,7 +28,7 @@ var unconscious = "wat, your hands fumble and fail you. try resting, weakling."
|
||||
|
||||
func NewWatGame(bot *WatBot, db *WatDb) *WatGame {
|
||||
g := WatGame{bot, db, Player{}, nil, nil, nil, nil, map[string]int{}}
|
||||
g.me = g.db.User(bot.Nick, "amia8t89xfp8y.liberta.casa", true)
|
||||
g.me = g.db.User(bot.Nick, "", true)
|
||||
g.commands = map[string](func(*Player, []string) string){
|
||||
//"wat": g.megaWat,
|
||||
"steroid": g.Steroid,
|
||||
@ -121,9 +121,17 @@ func (g *WatGame) help() string {
|
||||
return ret
|
||||
}
|
||||
|
||||
func (g *WatGame) RandInt(max int64) uint64 {
|
||||
i, _ := rand.Int(rand.Reader, big.NewInt(max))
|
||||
return i.Uint64()
|
||||
func (g *WatGame) RandInt(maxx int64) (uint64, error) {
|
||||
bi := big.NewInt(maxx)
|
||||
// prevent panic of rand.Int on big numbers
|
||||
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 {
|
||||
@ -166,8 +174,11 @@ func (g *WatGame) Dice(player *Player, fields []string) string {
|
||||
roll = i
|
||||
}
|
||||
}
|
||||
answer := g.RandInt(int64(roll)) + 1
|
||||
return fmt.Sprintf("1d%d - %d", roll, answer)
|
||||
answer, err := g.RandInt(int64(roll))
|
||||
if e := handleError(err); e != "" {
|
||||
return e
|
||||
}
|
||||
return fmt.Sprintf("1d%d - %d", roll + 1, answer)
|
||||
}
|
||||
|
||||
type PositiveError struct{}
|
||||
@ -206,12 +217,20 @@ func (g *WatGame) Roll(player *Player, fields []string) string {
|
||||
}
|
||||
lotteryNum := int64(-1)
|
||||
if dieSize > 100 {
|
||||
lotteryNum = int64(g.RandInt(dieSize)) + 1
|
||||
lotteryNumRand, randErr := g.RandInt(dieSize)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
lotteryNum = int64(lotteryNumRand) + 1
|
||||
}
|
||||
if amount > player.Coins {
|
||||
return "wat? brokeass"
|
||||
}
|
||||
n := int64(g.RandInt(dieSize)) + 1
|
||||
nRand, randErr := g.RandInt(dieSize)
|
||||
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)
|
||||
if n == lotteryNum {
|
||||
player.Coins += player.Coins
|
||||
@ -255,11 +274,18 @@ func (g *WatGame) Punch(player *Player, fields []string) string {
|
||||
if !target.Conscious() {
|
||||
return "wat? you're punching someone who is already unconscious. u crazy?"
|
||||
}
|
||||
chance := g.RandInt(6) + 1
|
||||
dmg := g.RandInt(6) + 1
|
||||
chance, randErr := g.RandInt(6)
|
||||
if e := handleError(randErr); e != "" {
|
||||
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)
|
||||
dmg += uint64(player.Level(player.Anarchy))
|
||||
if chance > 3 {
|
||||
if chance + 1 > 3 {
|
||||
ret += fmt.Sprintf("hits %s for %d points of damage! ", target.Nick, dmg)
|
||||
target.Health -= int64(dmg)
|
||||
g.db.Update(target)
|
||||
@ -300,7 +326,11 @@ func (g *WatGame) Frame(player *Player, fields []string) string {
|
||||
if target.Coins < amount {
|
||||
return fmt.Sprintf("wat? %s is too poor for this.", target.Nick)
|
||||
}
|
||||
n := g.RandInt(6) + 1
|
||||
n, randErr := g.RandInt(6)
|
||||
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)
|
||||
if n < 3 {
|
||||
ret += fmt.Sprintf("You frame %s for a minor crime. They pay me %d.", target.Nick, amount)
|
||||
@ -335,7 +365,11 @@ func (g *WatGame) Steal(player *Player, fields []string) string {
|
||||
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)
|
||||
}
|
||||
n := g.RandInt(6) + 1
|
||||
n, randErr := g.RandInt(6)
|
||||
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)
|
||||
if n < 3 {
|
||||
ret += "You did it! Sneaky bastard!"
|
||||
@ -380,7 +414,11 @@ func (g *WatGame) Leech(player *Player, fields []string) string {
|
||||
if err != "" {
|
||||
return err
|
||||
}
|
||||
r := g.RandInt(10) + 1
|
||||
r, randErr := g.RandInt(10)
|
||||
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)
|
||||
hpDown := amount / divisor
|
||||
player.Coins -= amount
|
||||
@ -408,7 +446,11 @@ func (g *WatGame) Rest(player *Player, fields []string) string {
|
||||
} else if delta < minRest {
|
||||
ret = fmt.Sprintf("wat were you thinking, sleeping at a time like this (%d until next rest)", minRest-delta)
|
||||
} else {
|
||||
value := g.RandInt(10) + 1
|
||||
value, randErr := g.RandInt(10)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
value = value + 1
|
||||
if player.Health < -5 {
|
||||
player.Health = 1
|
||||
ret = fmt.Sprintf("wow ur beat up. i pity u, ur health is now 1.")
|
||||
@ -435,8 +477,15 @@ func (g *WatGame) Bench(player *Player, fields []string) string {
|
||||
if !g.CanAct(player, Action_Lift, minTime) {
|
||||
return "you're tired. no more lifting for now."
|
||||
}
|
||||
weight := g.RandInt(370) + 50
|
||||
reps := g.RandInt(10)
|
||||
weight, randErr := g.RandInt(370)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
weight = weight + 50
|
||||
reps, randErr := g.RandInt(10)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
value := int64(0)
|
||||
reply := fmt.Sprintf("%s benches %dwatts for %d reps, ", player.Nick, weight, reps)
|
||||
if weight < 150 {
|
||||
@ -452,7 +501,10 @@ func (g *WatGame) Bench(player *Player, fields []string) string {
|
||||
}
|
||||
if g.roid[player.Nick] != 0 {
|
||||
delete(g.roid, player.Nick)
|
||||
success := g.RandInt(2)
|
||||
success, randErr := g.RandInt(2)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
if success != 0 {
|
||||
player.Health = 0
|
||||
player.Anarchy -= 10
|
||||
@ -475,7 +527,10 @@ func (g *WatGame) Riot(player *Player, fields []string) string {
|
||||
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)"
|
||||
}
|
||||
r := g.RandInt(100)
|
||||
r, randErr := g.RandInt(100)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e
|
||||
}
|
||||
reply := ""
|
||||
if r > 40 {
|
||||
player.Anarchy += 3
|
||||
@ -621,9 +676,21 @@ func PrintTwo(nick string, value uint64) string {
|
||||
}
|
||||
|
||||
func (g *WatGame) megaWat(player *Player, _ []string) string {
|
||||
mega := g.RandInt(1000000) + 1
|
||||
kilo := g.RandInt(1000) + 1
|
||||
ten := g.RandInt(100) + 1
|
||||
mega, randErr := g.RandInt(1000000)
|
||||
if e := handleError(randErr); e != "" {
|
||||
return e + " mega fail"
|
||||
}
|
||||
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 := ""
|
||||
if mega == 23 {
|
||||
player.Coins += 1000000
|
||||
|
126
wat/integration.go
Normal file
126
wat/integration.go
Normal file
@ -0,0 +1,126 @@
|
||||
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)
|
||||
}
|
||||
}
|
22
wat/utils.go
Normal file
22
wat/utils.go
Normal file
@ -0,0 +1,22 @@
|
||||
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 ""
|
||||
}
|
Loading…
Reference in New Issue
Block a user