goirc/irc/handlers.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]);
		}
	});
}