91 lines
1.8 KiB
Go
91 lines
1.8 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) Msg(m *irc.Message) {
|
||
|
|
if m.Params[0] == w.Nick && m.Prefix.Host == "tripsit/operator/hibs" {
|
||
|
|
if "join" == m.Params[1] {
|
||
|
|
w.write("JOIN", "##wat")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if strings.Contains(m.Prefix.Host, "tripsit") && m.Params[0] == "##wat" {
|
||
|
|
args := strings.FieldsFunc(m.Params[1], func(c rune) bool {return c == ' '})
|
||
|
|
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,
|
||
|
|
})
|
||
|
|
}
|