irc/src/callback.lua

-- IRC Mod for Minetest
-- By Diego Martínez <kaeza@users.sf.net>
--
-- This mod allows to tie a Minetest server to an IRC channel.
--
-- This program is free software. It comes without any warranty, to
-- the extent permitted by applicable law. You can redistribute it
-- and/or modify it under the terms of the Do What The Fuck You Want
-- To Public License, Version 2, as published by Sam Hocevar. See
-- http://sam.zoy.org/wtfpl/COPYING for more details.
--

local irc = require("irc");

mt_irc.callbacks = { };

mt_irc._callback = function ( name, breakonreturn, ... )
    local list = mt_irc.callbacks[name];
    if (not list) then return; end
    for n = 1, #list do
        local r = list[n](...);
        if (breakonreturn and (r ~= nil)) then return r; end
    end
end

mt_irc.register_callback = function ( name, func )
    local list = mt_irc.callbacks[name];
    if (not list) then
        list = { };
        mt_irc.callbacks[name] = list;
    end
    list[#list + 1] = func;
end

minetest.register_on_joinplayer(function ( player )
    local name = player:get_player_name();
    mt_irc.connected_players[name] = mt_irc.auto_join;
    if (not mt_irc.connect_ok) then return; end
    mt_irc.say("*** "..name.." joined the game");
end);

minetest.register_on_leaveplayer(function ( player )
    local name = player:get_player_name();
    mt_irc.connected_players[name] = nil;
    if (not mt_irc.connect_ok) then return; end
    mt_irc.say("*** "..name.." left the game");
end);

irc.register_callback("connect", function ( )
    mt_irc.got_motd = true;
    irc.join(mt_irc.channel);
end);

irc.register_callback("channel_msg", function ( channel, from, message )
    if (not mt_irc.connect_ok) then return; end
    local t = {
        name=(from or "<BUG:no one is saying this>");
        message=(message or "<BUG:there is no message>");
        server=mt_irc.server;
        port=mt_irc.port;
        channel=mt_irc.channel;
    };
    local text = mt_irc.message_format_in:gsub("%$%(([^)]+)%)", t)
    if (mt_irc._callback("channel_msg", from, message, text)) then return; end
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, text); end
    end
end);

local function bot_command ( from, message )

    local pos = message:find(" ", 1, true);
    local cmd, args;
    if (pos) then
        cmd = message:sub(1, pos - 1);
        args = message:sub(pos + 1);
    else
        cmd = message;
        args = "";
    end

    if (not mt_irc.bot_commands[cmd]) then
        mt_irc.say(from, "Unknown command `"..cmd.."'. Try `!help'.");
        return;
    end

    mt_irc.bot_commands[cmd].func(from, args);

end

irc.register_callback("private_msg", function ( from, message )
    if (not mt_irc.connect_ok) then return; end
    local player_to;
    local msg;
    if (message:sub(1, 1) == "@") then
        local pos = message:find(" ", 1, true);
        if (not pos) then return; end
        player_to = message:sub(2, pos - 1);
        msg = message:sub(pos + 1);
    elseif (message:sub(1, 1) == "!") then
        bot_command(from, message:sub(2));
        return;
    else
        irc.say(from, 'Message not sent! Please use "!help" to see possible commands.');
        irc.say(from, '    Or use the "@playername Message" syntax to send a private message.');
        return;
    end
    if (not mt_irc.connected_players[player_to]) then
        irc.say(from, "User `"..player_to.."' is not connected to IRC.");
        return;
    end
    local t = {
        name=(from or "<BUG:no one is saying this>");
        message=(msg or "<BUG:there is no message>");
        server=mt_irc.server;
        port=mt_irc.port;
        channel=mt_irc.channel;
    };
    local text = mt_irc.message_format_in:expandvars(t);
    if (mt_irc._callback("private_msg", from, player_to, message, text)) then return; end
    minetest.chat_send_player(player_to, "PRIVATE: "..text);
    mt_irc.say(from, "Message sent!")
end);

irc.register_callback("kick", function(chaninfo, nick, kicker)
    if nick == mt_irc.server_nick then
        minetest.chat_send_all("IRC: Bot was kicked by "..kicker..".");
        mt_irc.got_motd = false;
        mt_irc.connect_ok = false;
        irc.quit("Kicked");
    end
end);

irc.register_callback("nick_change", function ( from, old_nick )
    if (not mt_irc.connect_ok) then return; end
    mt_irc._callback("nick_change", false, old_nick, from);
    local text = "["..old_nick.." changed his nick to "..from.."]";
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, text); end
    end
end);

irc.register_callback("join", function ( servinfo, from )
    local text = "*** "..from.." joined "..mt_irc.channel;
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, text); end
    end
end);

irc.register_callback("part", function ( servinfo, from, part_msg )
    mt_irc._callback("part", false, from, part_msg);
    local text
    if part_msg then
        text = "*** "..from.." left "..mt_irc.channel.." ("..part_msg..")";
    else
        text = "*** "..from.." left "..mt_irc.channel;
    end
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, text); end
    end
end);

irc.register_callback("channel_act", function ( servinfo, from, message)
    if (not mt_irc.connect_ok) then return; end
    local text = "*** "..from.." "..message;
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, text); end
    end
end);

minetest.register_on_chat_message(function ( name, message )
    if (not mt_irc.connect_ok) then return; end
    if (message:sub(1, 1) == "/") then return; end
    if (not mt_irc.connected_players[name]) then return; end
    if (not minetest.check_player_privs(name, {shout=true})) then
        return;
    end
    if (not mt_irc.buffered_messages) then
        mt_irc.buffered_messages = { };
    end
    mt_irc.buffered_messages[#mt_irc.buffered_messages + 1] = {
        name = name;
        message = message;
    };
end);

minetest.register_on_shutdown(function ( )
    irc.quit("Game shutting down.");
    for n = 1, 5 do
        irc.poll();
    end
end);

irc.handlers.on_error = function (...) --( from, respond_to )
    for k, v in pairs(mt_irc.connected_players) do
        if (v) then minetest.chat_send_player(k, "IRC: Bot had a network error. Reconnecting in 5 seconds..."); end
    end
    for _, v in ipairs({...}) do
        minetest.chat_send_all(dump(v));
    end
    irc.quit("Network error");
    for n = 1, 5 do
        irc.poll();
    end
    mt_irc.got_motd = false;
    mt_irc.connect_ok = false;
    minetest.after(5, mt_irc.connect);
end

irc.handlers.on_err_nicknameinuse = function ( from, respond_to )
    irc.quit("Nick in use");
    for n = 1, 5 do
        irc.poll();
    end
    mt_irc.got_motd = false;
    mt_irc.connect_ok = false;
    local n = (tonumber(mt_irc.server_nick:sub(-1)) or 0) + 1;
    if (n == 10) then n = 1; end
    mt_irc.server_nick = mt_irc.server_nick:sub(1, -2)..n;
    mt_irc.connect();
end

-- TESTING
--[[
mt_irc.register_callback("part", function ( nick, part_msg )
  mt_irc.say("TEST: "..nick.." has left the building!");
end)

mt_irc.register_callback("nick_change", function ( old_nick, new_nick )
  mt_irc.say("TEST: "..old_nick.." -> "..new_nick);
end)
]]