Mercurial > prosody-modules
diff mod_ircd/mod_ircd.lua @ 111:3de60860adca
mod_ircd: Initial commit of a wonderfully hacky but working IRC->XMPP interface for Prosody
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Wed, 23 Dec 2009 14:22:02 +0000 |
parents | |
children | d4ff1cd414e5 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mod_ircd/mod_ircd.lua Wed Dec 23 14:22:02 2009 +0000 @@ -0,0 +1,178 @@ +local irc_listener = { default_port = 6667, default_mode = "*l" }; + +local sessions = {}; +local commands = {}; + +local nicks = {}; + +local st = require "util.stanza"; + +local conference_server = module:get_option("conference_server") or "conference.jabber.org"; + +local function irc_close_session(session) + session.conn:close(); +end + +function irc_listener.onincoming(conn, data) + local session = sessions[conn]; + if not session then + session = { conn = conn, host = module.host, reset_stream = function () end, + close = irc_close_session, log = logger.init("irc"..(conn.id or "1")), + roster = {} }; + sessions[conn] = session; + function session.data(data) + module:log("debug", "Received: %s", data); + local command, args = data:match("^%s*([^ ]+) *(.*)%s*$"); + if not command then + module:log("warn", "Invalid command: %s", data); + return; + end + command = command:upper(); + module:log("debug", "Received command: %s", command); + if commands[command] then + local ret = commands[command](session, args); + if ret then + session.send(ret.."\r\n"); + end + end + end + function session.send(data) + module:log("debug", "sending: %s", data); + return conn:write(data.."\r\n"); + end + end + if data then + session.data(data); + end +end + +function irc_listener.ondisconnect(conn, error) + module:log("debug", "Client disconnected"); + sessions[conn] = nil; +end + +function commands.NICK(session, nick) + nick = nick:match("^[%w_]+"); + if nicks[nick] then + session.send(":"..session.host.." 433 * The nickname "..nick.." is already in use"); + return; + end + nicks[nick] = session; + session.nick = nick; + session.full_jid = nick.."@"..module.host.."/ircd"; + session.type = "c2s"; + module:log("debug", "Client bound to %s", session.full_jid); + session.send(":"..session.host.." 001 "..session.nick.." :Welcome to XMPP via the "..session.host.." gateway "..session.nick); +end + +local joined_mucs = {}; +function commands.JOIN(session, channel) + if not joined_mucs[channel] then + joined_mucs[channel] = { occupants = {}, sessions = {} }; + end + joined_mucs[channel].sessions[session] = true; + local join_stanza = st.presence({ from = session.full_jid, to = channel:gsub("^#", "").."@"..conference_server.."/"..session.nick }); + core_process_stanza(session, join_stanza); + session.send(":"..session.nick.." JOIN :"..channel); + session.send(":"..session.host.." 332 "..session.nick.." "..channel.." :Connection in progress..."); + session.send(":"..session.host.." 353 "..session.nick.." = "..channel.." :"..session.nick); + session.send(":"..session.host.." 366 "..session.nick.." "..channel.." :End of /NAMES list."); +end + +function commands.PART(session, channel) + local channel, part_message = channel:match("^(%S+) :(.+)$"); + core_process_stanza(session, st.presence{ type = "unavailable", from = session.full_jid, + to = channel:gsub("^#", "").."@"..conference_server.."/"..session.nick }:tag("status"):text(part_message)); +end + +function commands.PRIVMSG(session, message) + local who, message = message:match("^(%S+) :(.+)$"); + if joined_mucs[who] then + core_process_stanza(session, st.message{to=who:gsub("^#", "").."@"..conference_server, type="groupchat"}:tag("body"):text(message)); + end +end + +function commands.WHO(session, channel) + if joined_mucs[channel] then + for nick in pairs(joined_mucs[channel].occupants) do + --n=MattJ 91.85.191.50 irc.freenode.net MattJ H :0 Matthew Wild + session.send(":"..session.host.." 352 "..session.nick.." "..channel.." "..nick.." "..nick.." "..session.host.." "..nick.." H :0 "..nick); + end + session.send(":"..session.host.." 315 "..session.nick.." "..channel.. " :End of /WHO list"); + end +end + +function commands.MODE(session, channel) + session.send(":"..session.host.." 324 "..session.nick.." "..channel.." +J"); +end + +--- Component (handle stanzas from the server for IRC clients) +function irc_component(origin, stanza) + local from, from_bare = stanza.attr.from, jid.bare(stanza.attr.from); + local from_node = "#"..jid.split(stanza.attr.from); + + if joined_mucs[from_node] and from_bare == from then + -- From room itself + local joined_muc = joined_mucs[from_node]; + if stanza.name == "message" then + local subject = stanza:get_child("subject"); + if subject then + local subject_text = subject:get_text(); + for session in pairs(joined_muc.sessions) do + session.send(":"..session.host.." 332 "..session.nick.." "..from_node.." :"..subject_text); + end + end + end + elseif joined_mucs[from_node] then + -- From room occupant + local joined_muc = joined_mucs[from_node]; + local nick = select(3, jid.split(from)):gsub(" ", "_"); + if stanza.name == "presence" then + local what; + if not stanza.attr.type then + if joined_muc.occupants[nick] then + return; + end + joined_muc.occupants[nick] = true; + what = "JOIN"; + else + joined_muc.occupants[nick] = nil; + what = "PART"; + end + for session in pairs(joined_muc.sessions) do + if nick ~= session.nick then + session.send(":"..nick.."!"..nick.." "..what.." :"..from_node); + end + end + elseif stanza.name == "message" then + local body = stanza:get_child("body"); + local hasdelay = stanza:get_child("delay", "urn:xmpp:delay"); + if body then + for session in pairs(joined_muc.sessions) do + if nick ~= session.nick or hasdelay then + session.send(":"..nick.." PRIVMSG "..from_node.." :"..body:get_text()); + end + end + end + end + end +end + +require "core.componentmanager".register_component(module.host, irc_component); + +prosody.events.add_handler("server-stopping", function (shutdown) + module:log("debug", "Closing IRC connections prior to shutdown"); + for channel, joined_muc in pairs(joined_mucs) do + for session in pairs(joined_muc.sessions) do + core_process_stanza(session, + st.presence{ type = "unavailable", from = session.full_jid, + to = channel:gsub("^#", "").."@"..conference_server.."/"..session.nick } + :tag("status") + :text("Connection closed: Server is shutting down"..(shutdown.reason and (": "..shutdown.reason) or ""))); + session:close(); + end + end +end); + +require "net.connlisteners".register("irc", irc_listener); +require "net.connlisteners".start("irc", { port = module:get_option("port") });