sneak/go-ircevent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #62 from alaska/master

Browse Source 
Add kick/multi kick. Properly randomize. Catch invalid CTCP messages and log
This commit is contained in:
Thomas Jager 2015-07-31 07:05:02 +02:00
commit 35f8ae857e
2 changed files with 44 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
irc.go
View File

@ -20,6 +20,7 @@ package irc
import ( import (
"bufio" "bufio"
"bytes"
"crypto/tls" "crypto/tls"
"errors" "errors"
"fmt" "fmt"
@ -245,6 +246,29 @@ func (irc *Connection) Privmsgf(target, format string, a ...interface{}) {
irc.Privmsg(target, fmt.Sprintf(format, a...)) irc.Privmsg(target, fmt.Sprintf(format, a...))
} }
// Kick <user> from <channel> with <msg>. For no message, pass empty string ("")
func (irc *Connection) Kick(user, channel, msg string) {
var cmd bytes.Buffer
cmd.WriteString(fmt.Sprintf("KICK %s %s", channel, user))
if msg != "" {
cmd.WriteString(fmt.Sprintf(" :%s", msg))
}
cmd.WriteString("\r\n")
irc.pwrite <- cmd.String()
}
// Kick all <users> from <channel> with <msg>. For no message, pass
// empty string ("")
func (irc *Connection) MultiKick(users []string, channel string, msg string) {
var cmd bytes.Buffer
cmd.WriteString(fmt.Sprintf("KICK %s %s", channel, strings.Join(users, ",")))
if msg != "" {
cmd.WriteString(fmt.Sprintf(" :%s", msg))
}
cmd.WriteString("\r\n")
irc.pwrite <- cmd.String()
}
// Send raw string. // Send raw string.
func (irc *Connection) SendRaw(message string) { func (irc *Connection) SendRaw(message string) {
irc.pwrite <- message + "\r\n" irc.pwrite <- message + "\r\n"

33
irc_callback.go
View File

@ -80,8 +80,11 @@ func (irc *Connection) RunCallbacks(event *Event) {
if event.Code == "PRIVMSG" && len(msg) > 2 && msg[0] == '\x01' { if event.Code == "PRIVMSG" && len(msg) > 2 && msg[0] == '\x01' {
event.Code = "CTCP" //Unknown CTCP event.Code = "CTCP" //Unknown CTCP
if i := strings.LastIndex(msg, "\x01"); i > -1 { if i := strings.LastIndex(msg, "\x01"); i > 0 {
msg = msg[1:i] msg = msg[1:i]
} else {
irc.Log.Printf("Invalid CTCP Message: %s\n", strconv.Quote(msg))
return
} }
if msg == "VERSION" { if msg == "VERSION" {
@ -161,14 +164,14 @@ func (irc *Connection) setupCallbacks() {
irc.AddCallback("CTCP_PING", func(e *Event) { irc.SendRawf("NOTICE %s :\x01%s\x01", e.Nick, e.Message()) }) irc.AddCallback("CTCP_PING", func(e *Event) { irc.SendRawf("NOTICE %s :\x01%s\x01", e.Nick, e.Message()) })
// 437: ERR_UNAVAILRESOURCE "<nick/channel> :Nick/channel is temporarily unavailable" // 437: ERR_UNAVAILRESOURCE "<nick/channel> :Nick/channel is temporarily unavailable"
// Add a _ to current nick. If irc.nickcurrent is empty this cannot // Add a _ to current nick. If irc.nickcurrent is empty this cannot
// work. It has to be set somewhere first in case the nick is already // work. It has to be set somewhere first in case the nick is already
// taken or unavailable from the beginning. // taken or unavailable from the beginning.
irc.AddCallback("437", func(e *Event) { irc.AddCallback("437", func(e *Event) {
// If irc.nickcurrent hasn't been set yet, set to irc.nick // If irc.nickcurrent hasn't been set yet, set to irc.nick
if irc.nickcurrent == "" { if irc.nickcurrent == "" {
irc.nickcurrent = irc.nick irc.nickcurrent = irc.nick
} }
if len(irc.nickcurrent) > 8 { if len(irc.nickcurrent) > 8 {
irc.nickcurrent = "_" + irc.nickcurrent irc.nickcurrent = "_" + irc.nickcurrent
@ -181,10 +184,10 @@ func (irc *Connection) setupCallbacks() {
// 433: ERR_NICKNAMEINUSE "<nick> :Nickname is already in use" // 433: ERR_NICKNAMEINUSE "<nick> :Nickname is already in use"
// Add a _ to current nick. // Add a _ to current nick.
irc.AddCallback("433", func(e *Event) { irc.AddCallback("433", func(e *Event) {
// If irc.nickcurrent hasn't been set yet, set to irc.nick // If irc.nickcurrent hasn't been set yet, set to irc.nick
if irc.nickcurrent == "" { if irc.nickcurrent == "" {
irc.nickcurrent = irc.nick irc.nickcurrent = irc.nick
} }
if len(irc.nickcurrent) > 8 { if len(irc.nickcurrent) > 8 {
irc.nickcurrent = "_" + irc.nickcurrent irc.nickcurrent = "_" + irc.nickcurrent
@ -216,3 +219,7 @@ func (irc *Connection) setupCallbacks() {
irc.nickcurrent = e.Arguments[0] irc.nickcurrent = e.Arguments[0]
}) })
} }
func init() {
rand.Seed(time.Now().UnixNano())
}