package wat import ( "crypto/tls" "fmt" "strings" "github.com/go-irc/irc" ) type WatBot struct { client *irc.Client conn *tls.Conn c *WatConfig game *WatGame Db *WatDb Nick string } type WatConfig struct { PermittedChannels []string IgnoredHosts []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.game = NewWatGame(&wat, wat.Db) config.Handler = irc.HandlerFunc(wat.HandleIrcMsg) wat.client = irc.NewClient(wat.conn, *config) return &wat } func CleanNick(nick string) string { return string(nick[0]) + "\u200c" + nick[1:] } 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) } } func (w *WatBot) Admin(m *irc.Message) bool { return m.Prefix.Host == "mph.monster" } func (w *WatBot) Allowed(c string, r []string) bool { for _, allowed := range r { if c == allowed { return true } } return false } func (w *WatBot) CanRespond(m *irc.Message) bool { if w.Admin(m) { return true } if w.Allowed(m.Prefix.Host, w.c.IgnoredHosts) { return false } // if !strings.Contains(m.Prefix.Host, "") { // return false // } if !w.Allowed(m.Params[0], w.c.PermittedChannels) { return false } return true } func (w *WatBot) Msg(m *irc.Message) { // bail out if you're not yves, if you're not tripsit or if you're not in an allowed channel // but if you're an admin you can do whatever if !w.CanRespond(m) { return } // make sure there's actually some text to process if len(m.Params[1]) == 0 { return } // fieldsfunc allows you to obtain rune separated fields/args args := strings.FieldsFunc(m.Params[1], func(c rune) bool { return c == ' ' }) if len(args) == 0 { return } if w.Admin(m) { // allow impersonation of the robot from anywhere if (args[0] == "imp" || args[0] == "imps") && len(args) > 2 { if args[0] == "imps" { w.write(args[1], args[2], strings.Join(args[3:], " ")) } else { w.write(args[1], args[2:]...) } return } } // strip offline message prefix from znc for handling offline buffer if args[0][0] == '[' && len(args) > 1 { args = args[1:] } // check if command char (or something weird) is present if args[0] != "wat" && args[0][0] != '#' { return } // clean input if args[0][0] == '#' { args[0] = args[0][1:] } user := strings.ToLower(m.Prefix.Name) player := w.Db.User(user, m.Prefix.Host, true) w.game.Msg(m, &player, args) } func (w *WatBot) Run() { defer w.conn.Close() err := w.client.Run() if err != nil { fmt.Println("Error returned while running client: " + err.Error()) } } func (w *WatBot) say(dest, msg string) { if len(msg) == 0 { return } //fmt.Printf("MSG %s: %s\n", dest, msg) w.write("PRIVMSG", dest, msg) } func (w *WatBot) reply(s *irc.Message, r string) { sender := s.Params[0] if sender == w.Nick { sender = s.Prefix.Name } w.say(sender, r) } func (w *WatBot) write(cmd string, params ...string) { w.client.WriteMessage(&irc.Message{ Command: cmd, Params: params, }) }