100 lines
2.0 KiB
Go
100 lines
2.0 KiB
Go
package wat
|
|
|
|
import (
|
|
"github.com/go-irc/irc"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
type WatBot struct {
|
|
client *irc.Client
|
|
conn *tls.Conn
|
|
game *WatGame
|
|
Db *WatDb
|
|
Nick string
|
|
}
|
|
|
|
func NewWatBot(config *irc.ClientConfig, serverConn *tls.Conn) *WatBot {
|
|
wat := WatBot{conn:serverConn, Nick:config.Nick}
|
|
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) {
|
|
fmt.Println(m)
|
|
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 == "tripsit/operator/hibs"
|
|
}
|
|
|
|
func (w *WatBot) Msg(m *irc.Message) {
|
|
if !strings.Contains(m.Prefix.Host, "tripsit") || (m.Params[0] != "##wat" && m.Params[0] != "##test" && !w.Admin(m)) {
|
|
return
|
|
}
|
|
|
|
args := strings.FieldsFunc(m.Params[1], func(c rune) bool {return c == ' '})
|
|
|
|
if w.Admin(m) {
|
|
// Do a special admin command and return, or continue
|
|
if args[0] == "imp" {
|
|
w.write(args[1], args[2:]...)
|
|
}
|
|
}
|
|
|
|
if len(args) < 1 && args[0] != "wat" && args[0][0] != '#' {
|
|
return
|
|
}
|
|
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(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,
|
|
})
|
|
}
|