local irc = require("irc.main") irc.handlers = {} local handlers = irc.handlers handlers["PING"] = function(conn, msg) conn:send(irc.Message({command="PONG", args=msg.args})) end handlers["001"] = function(conn, msg) conn.authed = true conn.nick = msg.args[1] end handlers["PRIVMSG"] = function(conn, msg) conn:invoke("OnChat", msg.user, msg.args[1], msg.args[2]) end handlers["NOTICE"] = function(conn, msg) conn:invoke("OnNotice", msg.user, msg.args[1], msg.args[2]) end handlers["JOIN"] = function(conn, msg) local channel = msg.args[1] if conn.track_users then if msg.user.nick == conn.nick then conn.channels[channel] = {users = {}} else conn.channels[channel].users[msg.user.nick] = msg.user end end conn:invoke("OnJoin", msg.user, msg.args[1]) end handlers["PART"] = function(conn, msg) local channel = msg.args[1] if conn.track_users then if msg.user.nick == conn.nick then conn.channels[channel] = nil else conn.channels[channel].users[msg.user.nick] = nil end end conn:invoke("OnPart", msg.user, msg.args[1], msg.args[2]) end handlers["QUIT"] = function(conn, msg) if conn.track_users then for chanName, chan in pairs(conn.channels) do chan.users[msg.user.nick] = nil end end conn:invoke("OnQuit", msg.user, msg.args[1], msg.args[2]) end handlers["NICK"] = function(conn, msg) local newNick = msg.args[1] if conn.track_users then for chanName, chan in pairs(conn.channels) do local users = chan.users local oldinfo = users[msg.user.nick] if oldinfo then users[newNick] = oldinfo users[msg.user.nick] = nil conn:invoke("NickChange", msg.user, newNick, chanName) end end else conn:invoke("NickChange", msg.user, newNick) end if msg.user.nick == conn.nick then conn.nick = newnick end end local function needNewNick(conn, msg) local newnick = conn.nickGenerator(msg.args[2]) if msg.nick then conn:queue(msg.nick(newnick)) end -- Ugly fix end -- ERR_ERRONEUSNICKNAME (Misspelt but remains for historical reasons) handlers["432"] = needNewNick -- ERR_NICKNAMEINUSE handlers["433"] = needNewNick -- RPL_ISUPPORT handlers["005"] = function(conn, msg) local arglen = #msg.args -- Skip first and last parameters (nick and info) for i = 2, arglen - 1 do local item = msg.args[i] local pos = item:find("=") if pos then conn.supports[item:sub(1, pos - 1)] = item:sub(pos + 1) else conn.supports[item] = true end end end -- RPL_MOTDSTART handlers["375"] = function(conn, msg) conn.motd = "" end -- RPL_MOTD handlers["372"] = function(conn, msg) -- MOTD lines have a "- " prefix, strip it. conn.motd = conn.motd .. msg.args[2]:sub(3) .. '\n' end -- NAMES list handlers["353"] = function(conn, msg) local chanType = msg.args[2] local channel = msg.args[3] local names = msg.args[4] if conn.track_users then conn.channels[channel] = conn.channels[channel] or {users = {}, type = chanType} local users = conn.channels[channel].users for nick in names:gmatch("(%S+)") do local access, name = irc.parseNick(conn, nick) users[name] = {access = access} end end end -- End of NAMES list handlers["366"] = function(conn, msg) if conn.track_users then conn:invoke("NameList", msg.args[2], msg.args[3]) end end -- No topic handlers["331"] = function(conn, msg) conn:invoke("OnTopic", msg.args[2], nil) end handlers["TOPIC"] = function(conn, msg) conn:invoke("OnTopic", msg.args[1], msg.args[2]) end handlers["332"] = function(conn, msg) conn:invoke("OnTopic", msg.args[2], msg.args[3]) end -- Topic creation info handlers["333"] = function(conn, msg) conn:invoke("OnTopicInfo", msg.args[2], msg.args[3], tonumber(msg.args[4])) end handlers["KICK"] = function(conn, msg) conn:invoke("OnKick", msg.args[1], msg.args[2], msg.user, msg.args[3]) end -- RPL_UMODEIS -- To answer a query about a client's own mode, RPL_UMODEIS is sent back handlers["221"] = function(conn, msg) conn:invoke("OnUserMode", msg.args[2]) end -- RPL_CHANNELMODEIS -- The result from common irc servers differs from that defined by the rfc handlers["324"] = function(conn, msg) conn:invoke("OnChannelMode", msg.args[2], msg.args[3]) end handlers["MODE"] = function(conn, msg) local target = msg.args[1] local modes = msg.args[2] local optList = {} for i = 3, #msg.args do table.insert(optList, msg.args[i]) end if conn.track_users and target ~= conn.nick then local add = true local argNum = 1 irc.updatePrefixModes(conn) for c in modes:gmatch(".") do if c == "+" then add = true elseif c == "-" then add = false elseif conn.modeprefix[c] then local nick = optList[argNum] argNum = argNum + 1 local user = conn.channels[target].users[nick] user.access = user.access or {} local access = user.access access[c] = add if c == "o" then access.op = add elseif c == "v" then access.voice = add end end end end conn:invoke("OnModeChange", msg.user, target, modes, unpack(optList)) end handlers["ERROR"] = function(conn, msg) conn:invoke("OnDisconnect", msg.args[1], true) conn:shutdown() error(msg.args[1], 3) end