mirror of
https://github.com/fluffle/goirc
synced 2025-11-03 19:48:04 +00:00
402 lines
13 KiB
Go
402 lines
13 KiB
Go
package irc
|
|
|
|
// this file contains the basic set of event handlers
|
|
// to manage tracking an irc connection etc.
|
|
|
|
import (
|
|
"strings";
|
|
"strconv";
|
|
)
|
|
|
|
// Add an event handler for a specific IRC command
|
|
func (conn *Conn) AddHandler(name string, f func (*Conn, *Line)) {
|
|
n := strings.ToUpper(name);
|
|
if e, ok := conn.events[n]; ok {
|
|
if len(e) == cap(e) {
|
|
// crap, we're full. expand e by another 10 handler slots
|
|
ne := make([]func (*Conn, *Line), len(e), len(e)+10);
|
|
for i := 0; i<len(e); i++ {
|
|
ne[i] = e[i];
|
|
}
|
|
e = ne;
|
|
}
|
|
e = e[0:len(e)+1];
|
|
e[len(e)-1] = f;
|
|
} else {
|
|
e := make([]func (*Conn, *Line), 1, 10);
|
|
e[0] = f;
|
|
conn.events[n] = e;
|
|
}
|
|
}
|
|
|
|
// loops through all event handlers for line.Cmd, running each in a goroutine
|
|
func (conn *Conn) dispatchEvent(line *Line) {
|
|
// seems that we end up dispatching an event with a nil line when receiving
|
|
// EOF from the server. Until i've tracked down why....
|
|
if line == nil {
|
|
conn.error("irc.dispatchEvent(): buh? line == nil :-(");
|
|
return
|
|
}
|
|
|
|
// So, I think CTCP and (in particular) CTCP ACTION are better handled as
|
|
// separate events as opposed to forcing people to have gargantuan PRIVMSG
|
|
// handlers to cope with the possibilities.
|
|
if line.Cmd == "PRIVMSG" && len(line.Text) > 2 &&
|
|
line.Text[0] == '\001' && line.Text[len(line.Text)-1] == '\001' {
|
|
// WOO, it's a CTCP message
|
|
t := strings.Split(line.Text[1:len(line.Text)-1], " ", 2);
|
|
if c := strings.ToUpper(t[0]); c == "ACTION" {
|
|
// make a CTCP ACTION it's own event a-la PRIVMSG
|
|
line.Cmd = c;
|
|
} else {
|
|
// otherwise, dispatch a generic CTCP event that
|
|
// contains the type of CTCP in line.Args[0]
|
|
line.Cmd = "CTCP";
|
|
a := make([]string, len(line.Args)+1);
|
|
a[0] = c;
|
|
for i:=0; i<len(line.Args); i++ {
|
|
a[i+1] = line.Args[i];
|
|
}
|
|
line.Args = a;
|
|
}
|
|
if len(t) > 1 {
|
|
// for some CTCP messages this could make more sense
|
|
// in line.Args[], but meh. MEH, I say.
|
|
line.Text = t[1];
|
|
}
|
|
}
|
|
if funcs, ok := conn.events[line.Cmd]; ok {
|
|
for _, f := range funcs {
|
|
go f(conn, line)
|
|
}
|
|
}
|
|
}
|
|
|
|
// sets up the internal event handlers to do useful things with lines
|
|
// XXX: is there a better way of doing this?
|
|
// Turns out there may be but it's not actually implemented in the language yet
|
|
// according to the people on freenode/#go-nuts ... :-(
|
|
// see: http://golang.org/doc/go_spec.html#Method_expressions for details
|
|
// I think this means we should be able to do something along the lines of:
|
|
// conn.AddHandler("event", (*Conn).h_handler);
|
|
// where h_handler is declared in the irc package as:
|
|
// func (conn *Conn) h_handler(line *Line) {}
|
|
// in the future, but for now the compiler throws a hissy fit.
|
|
func (conn *Conn) setupEvents() {
|
|
conn.events = make(map[string] []func(*Conn, *Line));
|
|
|
|
// Basic ping/pong handler
|
|
conn.AddHandler("PING", func(conn *Conn, line *Line) {
|
|
conn.Raw("PONG :" + line.Text);
|
|
});
|
|
|
|
// Handler to trigger a "CONNECTED" event on receipt of numeric 001
|
|
conn.AddHandler("001", func(conn *Conn, line *Line) {
|
|
// we're connected!
|
|
conn.connected = true;
|
|
conn.dispatchEvent(&Line{Cmd: "CONNECTED"});
|
|
// and we're being given our hostname (from the server's perspective)
|
|
if ridx := strings.LastIndex(line.Text, " "); ridx != -1 {
|
|
h := line.Text[ridx+1:len(line.Text)];
|
|
if idx := strings.Index(h, "@"); idx != -1 {
|
|
conn.Me.Host = h[idx+1:len(h)];
|
|
}
|
|
}
|
|
});
|
|
|
|
// XXX: do we need 005 protocol support message parsing here?
|
|
// probably in the future, but I can't quite be arsed yet.
|
|
/*
|
|
:irc.pl0rt.org 005 GoTest CMDS=KNOCK,MAP,DCCALLOW,USERIP UHNAMES NAMESX SAFELIST HCN MAXCHANNELS=20 CHANLIMIT=#:20 MAXLIST=b:60,e:60,I:60 NICKLEN=30 CHANNELLEN=32 TOPICLEN=307 KICKLEN=307 AWAYLEN=307 :are supported by this server
|
|
:irc.pl0rt.org 005 GoTest MAXTARGETS=20 WALLCHOPS WATCH=128 WATCHOPTS=A SILENCE=15 MODES=12 CHANTYPES=# PREFIX=(qaohv)~&@%+ CHANMODES=beI,kfL,lj,psmntirRcOAQKVCuzNSMT NETWORK=bb101.net CASEMAPPING=ascii EXTBAN=~,cqnr ELIST=MNUCT :are supported by this server
|
|
:irc.pl0rt.org 005 GoTest STATUSMSG=~&@%+ EXCEPTS INVEX :are supported by this server
|
|
*/
|
|
|
|
// Handler to deal with "433 :Nickname already in use"
|
|
conn.AddHandler("433", func(conn *Conn, line *Line) {
|
|
// Args[1] is the new nick we were attempting to acquire
|
|
conn.Nick(line.Args[1] + "_");
|
|
});
|
|
|
|
// Handler NICK messages to inform us about nick changes
|
|
conn.AddHandler("NICK", func(conn *Conn, line *Line) {
|
|
// all nicks should be handled the same way, our own included
|
|
if n := conn.GetNick(line.Nick); n != nil {
|
|
n.ReNick(line.Text);
|
|
} else {
|
|
conn.error("irc.NICK(): buh? unknown nick %s.", line.Nick);
|
|
}
|
|
});
|
|
|
|
// Handle VERSION requests and CTCP PING
|
|
conn.AddHandler("CTCP", func(conn *Conn, line *Line) {
|
|
if line.Args[0] == "VERSION" {
|
|
conn.CtcpReply(line.Nick, "VERSION", "powered by goirc...");
|
|
} else if line.Args[0] == "PING" {
|
|
conn.CtcpReply(line.Nick, "PING", line.Text);
|
|
}
|
|
});
|
|
|
|
// Handle JOINs to channels to maintain state
|
|
conn.AddHandler("JOIN", func(conn *Conn, line *Line) {
|
|
ch := conn.GetChannel(line.Text);
|
|
n := conn.GetNick(line.Nick);
|
|
if ch == nil {
|
|
// first we've seen of this channel, so should be us joining it
|
|
// NOTE this will also take care of n == nil && ch == nil
|
|
if n != conn.Me {
|
|
conn.error("irc.JOIN(): buh? JOIN to unknown channel %s recieved from (non-me) nick %s", line.Text, line.Nick);
|
|
return;
|
|
}
|
|
ch = conn.NewChannel(line.Text);
|
|
// since we don't know much about this channel, ask server for info
|
|
// we get the channel users automatically in 353 and the channel
|
|
// topic in 332 on join, so we just need to get the modes
|
|
conn.Mode(ch.Name);
|
|
}
|
|
if n == nil {
|
|
// this is the first we've seen of this nick
|
|
n = conn.NewNick(line.Nick, line.Ident, "", line.Host);
|
|
// since we don't know much about this nick, ask server for info
|
|
conn.Whois(n.Nick);
|
|
}
|
|
// this takes care of both nick and channel linking \o/
|
|
ch.AddNick(n);
|
|
});
|
|
|
|
// Handle PARTs from channels to maintain state
|
|
conn.AddHandler("PART", func(conn *Conn, line *Line) {
|
|
ch := conn.GetChannel(line.Args[0]);
|
|
n := conn.GetNick(line.Nick);
|
|
if ch != nil && n != nil {
|
|
ch.DelNick(n);
|
|
} else {
|
|
conn.error("irc.PART(): buh? PART of channel %s by nick %s", line.Args[0], line.Nick);
|
|
}
|
|
});
|
|
|
|
// Handle KICKs from channels to maintain state
|
|
conn.AddHandler("KICK", func(conn *Conn, line *Line) {
|
|
// XXX: this won't handle autorejoining channels on KICK
|
|
// it's trivial to do this in a seperate handler...
|
|
ch := conn.GetChannel(line.Args[0]);
|
|
n := conn.GetNick(line.Args[1]);
|
|
if ch != nil && n != nil {
|
|
ch.DelNick(n);
|
|
} else {
|
|
conn.error("irc.KICK(): buh? KICK from channel %s of nick %s", line.Args[0], line.Args[1]);
|
|
}
|
|
});
|
|
|
|
// Handle other people's QUITs
|
|
conn.AddHandler("QUIT", func(conn *Conn, line *Line) {
|
|
if n := conn.GetNick(line.Nick); n != nil {
|
|
n.Delete();
|
|
} else {
|
|
conn.error("irc.QUIT(): buh? QUIT from unknown nick %s", line.Nick);
|
|
}
|
|
});
|
|
|
|
// Handle MODE changes for channels we know about (and our nick personally)
|
|
// this is moderately ugly. suggestions for improvement welcome
|
|
conn.AddHandler("MODE", func(conn *Conn, line *Line) {
|
|
// channel modes first
|
|
if ch := conn.GetChannel(line.Args[0]); ch != nil {
|
|
modeargs := line.Args[2:len(line.Args)];
|
|
var modeop bool; // true => add mode, false => remove mode
|
|
var modestr string;
|
|
for i := 0; i < len(line.Args[1]); i++ {
|
|
switch m := line.Args[1][i]; m {
|
|
case '+':
|
|
modeop = true;
|
|
modestr = string(m);
|
|
case '-':
|
|
modeop = false;
|
|
modestr = string(m);
|
|
case 'i': ch.Modes.InviteOnly = modeop;
|
|
case 'm': ch.Modes.Moderated = modeop;
|
|
case 'n': ch.Modes.NoExternalMsg = modeop;
|
|
case 'p': ch.Modes.Private = modeop;
|
|
case 's': ch.Modes.Secret = modeop;
|
|
case 't': ch.Modes.ProtectedTopic = modeop;
|
|
case 'z': ch.Modes.SSLOnly = modeop;
|
|
case 'O': ch.Modes.OperOnly = modeop;
|
|
case 'k':
|
|
if len(modeargs) != 0 {
|
|
ch.Modes.Key, modeargs = modeargs[0], modeargs[1:len(modeargs)]
|
|
} else {
|
|
conn.error("irc.MODE(): buh? not enough arguments to process MODE %s %s%s", ch.Name, modestr, m)
|
|
}
|
|
case 'l':
|
|
if len(modeargs) != 0 {
|
|
ch.Modes.Limit, _ = strconv.Atoi(modeargs[0]);
|
|
modeargs = modeargs[1:len(modeargs)];
|
|
} else {
|
|
conn.error("irc.MODE(): buh? not enough arguments to process MODE %s %s%s", ch.Name, modestr, m)
|
|
}
|
|
case 'q', 'a', 'o', 'h', 'v':
|
|
if len(modeargs) != 0 {
|
|
n := conn.GetNick(modeargs[0]);
|
|
if p, ok := ch.Nicks[n]; ok && n != nil {
|
|
switch m {
|
|
case 'q': p.Owner = modeop;
|
|
case 'a': p.Admin = modeop;
|
|
case 'o': p.Op = modeop;
|
|
case 'h': p.HalfOp = modeop;
|
|
case 'v': p.Voice = modeop;
|
|
}
|
|
modeargs = modeargs[1:len(modeargs)];
|
|
} else {
|
|
conn.error("irc.MODE(): MODE %s %s%s %s: buh? state tracking failure.", ch.Name, modestr, m, modeargs[0]);
|
|
}
|
|
} else {
|
|
conn.error("irc.MODE(): buh? not enough arguments to process MODE %s %s%s", ch.Name, modestr, m);
|
|
}
|
|
}
|
|
}
|
|
} else if n := conn.GetNick(line.Args[0]); n != nil {
|
|
// nick mode change, should be us
|
|
if n != conn.Me {
|
|
conn.error("irc.MODE(): buh? recieved MODE %s for (non-me) nick %s", line.Args[1], n.Nick);
|
|
return;
|
|
}
|
|
var modeop bool; // true => add mode, false => remove mode
|
|
for i := 0; i < len(line.Text); i++ {
|
|
switch m := line.Text[i]; m {
|
|
case '+':
|
|
modeop = true;
|
|
case '-':
|
|
modeop = false;
|
|
case 'i': n.Modes.Invisible = modeop;
|
|
case 'o': n.Modes.Oper = modeop;
|
|
case 'w': n.Modes.WallOps = modeop;
|
|
case 'x': n.Modes.HiddenHost = modeop;
|
|
case 'z': n.Modes.SSL = modeop;
|
|
}
|
|
}
|
|
} else {
|
|
conn.error("irc.MODE(): buh? not sure what to do with MODE %s %s", line.Args[0], line.Args[1]);
|
|
}
|
|
});
|
|
|
|
// Handle TOPIC changes for channels
|
|
conn.AddHandler("TOPIC", func(conn *Conn, line *Line) {
|
|
if ch := conn.GetChannel(line.Args[0]); ch != nil {
|
|
ch.Topic = line.Text;
|
|
} else {
|
|
conn.error("irc.TOPIC(): buh? topic change on unknown channel %s", line.Args[0]);
|
|
}
|
|
});
|
|
|
|
// Handle 311 whois reply
|
|
conn.AddHandler("311", func(conn *Conn, line *Line) {
|
|
if n := conn.GetNick(line.Args[1]); n != nil {
|
|
n.Ident = line.Args[2];
|
|
n.Host = line.Args[3];
|
|
n.Name = line.Text;
|
|
} else {
|
|
conn.error("irc.311(): buh? received WHOIS info for unknown nick %s", line.Args[1]);
|
|
}
|
|
});
|
|
|
|
// Handle 324 mode reply
|
|
conn.AddHandler("324", func(conn *Conn, line *Line) {
|
|
// XXX: copypasta from MODE, needs tidying.
|
|
if ch := conn.GetChannel(line.Args[1]); ch != nil {
|
|
modeargs := line.Args[3:len(line.Args)];
|
|
var modeop bool; // true => add mode, false => remove mode
|
|
var modestr string;
|
|
for i := 0; i < len(line.Args[2]); i++ {
|
|
switch m := line.Args[2][i]; m {
|
|
case '+':
|
|
modeop = true;
|
|
modestr = string(m);
|
|
case '-':
|
|
modeop = false;
|
|
modestr = string(m);
|
|
case 'i': ch.Modes.InviteOnly = modeop;
|
|
case 'm': ch.Modes.Moderated = modeop;
|
|
case 'n': ch.Modes.NoExternalMsg = modeop;
|
|
case 'p': ch.Modes.Private = modeop;
|
|
case 's': ch.Modes.Secret = modeop;
|
|
case 't': ch.Modes.ProtectedTopic = modeop;
|
|
case 'z': ch.Modes.SSLOnly = modeop;
|
|
case 'O': ch.Modes.OperOnly = modeop;
|
|
case 'k':
|
|
if len(modeargs) != 0 {
|
|
ch.Modes.Key, modeargs = modeargs[0], modeargs[1:len(modeargs)]
|
|
} else {
|
|
conn.error("irc.324(): buh? not enough arguments to process MODE %s %s%s", ch.Name, modestr, m)
|
|
}
|
|
case 'l':
|
|
if len(modeargs) != 0 {
|
|
ch.Modes.Limit, _ = strconv.Atoi(modeargs[0]);
|
|
modeargs = modeargs[1:len(modeargs)];
|
|
} else {
|
|
conn.error("irc.324(): buh? not enough arguments to process MODE %s %s%s", ch.Name, modestr, m)
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
conn.error("irc.324(): buh? received MODE settings for unknown channel %s", line.Args[1]);
|
|
}
|
|
});
|
|
|
|
// Handle 332 topic reply on join to channel
|
|
conn.AddHandler("332", func(conn *Conn, line *Line) {
|
|
if ch := conn.GetChannel(line.Args[1]); ch != nil {
|
|
ch.Topic = line.Text;
|
|
} else {
|
|
conn.error("irc.332(): buh? received TOPIC value for unknown channel %s", line.Args[1]);
|
|
}
|
|
});
|
|
|
|
// Handle 353 names reply
|
|
conn.AddHandler("353", func(conn *Conn, line *Line) {
|
|
if ch := conn.GetChannel(line.Args[2]); ch != nil {
|
|
nicks := strings.Split(line.Text, " ", 0);
|
|
for _, nick := range nicks {
|
|
// UnrealIRCd's coders are lazy and leave a trailing space
|
|
if nick == "" {
|
|
continue
|
|
}
|
|
switch c := nick[0]; c {
|
|
case '~', '&', '@', '%', '+':
|
|
nick = nick[1:len(nick)];
|
|
fallthrough;
|
|
default:
|
|
n := conn.GetNick(nick);
|
|
if n == nil {
|
|
// we don't know this nick yet!
|
|
n = conn.NewNick(nick, "", "", "");
|
|
conn.Whois(nick);
|
|
}
|
|
if n != conn.Me {
|
|
// we will be in the names list, but should also be in
|
|
// the channel's nick list from the JOIN handler above
|
|
ch.AddNick(n);
|
|
}
|
|
p := ch.Nicks[n];
|
|
switch c {
|
|
case '~': p.Owner = true;
|
|
case '&': p.Admin = true;
|
|
case '@': p.Op = true;
|
|
case '%': p.HalfOp = true;
|
|
case '+': p.Voice = true;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
conn.error("irc.353(): buh? received NAMES list for unknown channel %s", line.Args[2]);
|
|
}
|
|
});
|
|
|
|
// Handle 671 whois reply (nick connected via SSL)
|
|
conn.AddHandler("671", func(conn *Conn, line *Line) {
|
|
if n := conn.GetNick(line.Args[1]); n != nil {
|
|
n.Modes.SSL = true;
|
|
} else {
|
|
conn.error("irc.671(): buh? received WHOIS SSL info for unknown nick %s", line.Args[1]);
|
|
}
|
|
});
|
|
}
|