go-ircevent/irc_callback.go

168 lines
4.1 KiB
Go
Raw Normal View History

2012-11-05 22:46:47 +00:00
package irc
import (
"strconv"
"strings"
"time"
"crypto/sha1"
"fmt"
"reflect"
"math/rand"
2012-11-05 22:46:47 +00:00
)
func (irc *Connection) AddCallback(eventcode string, callback func(*Event)) string {
2012-11-05 22:46:47 +00:00
eventcode = strings.ToUpper(eventcode)
2012-03-22 03:50:21 +00:00
if _, ok := irc.events[eventcode]; !ok {
irc.events[eventcode] = make(map[string]func(*Event))
}
h := sha1.New()
rawId := []byte(fmt.Sprintf("%v%d", reflect.ValueOf(callback).Pointer(), rand.Int63()))
h.Write(rawId)
id := fmt.Sprintf("%x", h.Sum(nil))
irc.events[eventcode][id] = callback
return id
}
func (irc *Connection) RemoveCallback(eventcode string, i string) bool {
eventcode = strings.ToUpper(eventcode)
if event, ok := irc.events[eventcode]; ok{
if _, ok := event[i]; ok {
delete(irc.events[eventcode], i)
return true
}
irc.Log.Printf("Event found, but no callback found at id %s\n", i)
return false
2012-11-05 22:46:47 +00:00
}
irc.Log.Println("Event not found")
return false
2012-11-05 22:46:47 +00:00
}
func (irc *Connection) ReplaceCallback(eventcode string, i string, callback func(*Event)) {
2012-11-05 22:46:47 +00:00
eventcode = strings.ToUpper(eventcode)
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
if event, ok := irc.events[eventcode]; ok {
if _, ok := event[i]; ok {
2012-11-05 22:46:47 +00:00
event[i] = callback
return
}
irc.Log.Printf("Event found, but no callback found at id %s\n", i)
2012-11-05 22:46:47 +00:00
}
irc.Log.Printf("Event not found. Use AddCallBack\n")
2012-11-05 22:46:47 +00:00
}
func (irc *Connection) RunCallbacks(event *Event) {
if event.Code == "PRIVMSG" && len(event.Message) > 0 && event.Message[0] == '\x01' {
event.Code = "CTCP" //Unknown CTCP
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
if i := strings.LastIndex(event.Message, "\x01"); i > -1 {
event.Message = event.Message[1:i]
}
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
if event.Message == "VERSION" {
event.Code = "CTCP_VERSION"
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
} else if event.Message == "TIME" {
event.Code = "CTCP_TIME"
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
} else if event.Message[0:4] == "PING" {
event.Code = "CTCP_PING"
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
} else if event.Message == "USERINFO" {
event.Code = "CTCP_USERINFO"
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
} else if event.Message == "CLIENTINFO" {
event.Code = "CTCP_CLIENTINFO"
2014-02-02 05:38:45 +00:00
} else if event.Message[0:6] == "ACTION" {
event.Code = "CTCP_ACTION"
event.Message = event.Message[7:]
2012-11-05 22:46:47 +00:00
}
}
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
if callbacks, ok := irc.events[event.Code]; ok {
if irc.VerboseCallbackHandler {
irc.Log.Printf("%v (%v) >> %#v\n", event.Code, len(callbacks), event)
2012-11-05 22:46:47 +00:00
}
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
for _, callback := range callbacks {
go callback(event)
}
} else if irc.VerboseCallbackHandler {
irc.Log.Printf("%v (0) >> %#v\n", event.Code, event)
2012-11-05 22:46:47 +00:00
}
if callbacks, ok := irc.events["*"]; ok {
if irc.VerboseCallbackHandler {
irc.Log.Printf("Wildcard %v (%v) >> %#v\n", event.Code, len(callbacks), event)
}
for _, callback := range callbacks {
go callback(event)
}
}
2012-11-05 22:46:47 +00:00
}
func (irc *Connection) setupCallbacks() {
irc.events = make(map[string]map[string]func(*Event))
2012-11-05 22:46:47 +00:00
//Handle ping events
irc.AddCallback("PING", func(e *Event) { irc.SendRaw("PONG :" + e.Message) })
//Version handler
irc.AddCallback("CTCP_VERSION", func(e *Event) {
irc.SendRawf("NOTICE %s :\x01VERSION %s\x01", e.Nick, irc.Version)
2012-11-05 22:46:47 +00:00
})
irc.AddCallback("CTCP_USERINFO", func(e *Event) {
irc.SendRawf("NOTICE %s :\x01USERINFO %s\x01", e.Nick, irc.user)
})
irc.AddCallback("CTCP_CLIENTINFO", func(e *Event) {
irc.SendRawf("NOTICE %s :\x01CLIENTINFO PING VERSION TIME USERINFO CLIENTINFO\x01", e.Nick)
})
irc.AddCallback("CTCP_TIME", func(e *Event) {
ltime := time.Now()
irc.SendRawf("NOTICE %s :\x01TIME %s\x01", e.Nick, ltime.String())
})
irc.AddCallback("CTCP_PING", func(e *Event) { irc.SendRawf("NOTICE %s :\x01%s\x01", e.Nick, e.Message) })
irc.AddCallback("437", func(e *Event) {
irc.nickcurrent = irc.nickcurrent + "_"
irc.SendRawf("NICK %s", irc.nickcurrent)
})
irc.AddCallback("433", func(e *Event) {
if len(irc.nickcurrent) > 8 {
irc.nickcurrent = "_" + irc.nickcurrent
2012-03-22 03:50:21 +00:00
2012-11-05 22:46:47 +00:00
} else {
irc.nickcurrent = irc.nickcurrent + "_"
}
irc.SendRawf("NICK %s", irc.nickcurrent)
})
irc.AddCallback("PONG", func(e *Event) {
ns, _ := strconv.ParseInt(e.Message, 10, 64)
delta := time.Duration(time.Now().UnixNano() - ns)
if irc.Debug {
irc.Log.Printf("Lag: %vs\n", delta)
}
2012-11-05 22:46:47 +00:00
})
irc.AddCallback("NICK", func(e *Event) {
if e.Nick == irc.nick {
2013-03-13 11:52:31 +00:00
irc.nickcurrent = e.Message
2012-11-05 22:46:47 +00:00
}
})
irc.AddCallback("001", func(e *Event) {
irc.nickcurrent = e.Arguments[0]
})
}