irc/src/hooks.lua

-- This file is licensed under the terms of the BSD 2-clause license.
-- See LICENSE.txt for details.


mt_irc.hooks = {}
mt_irc.registered_hooks = {}


function mt_irc:doHook(conn)
	for name, hook in pairs(self.registered_hooks) do
		for _, func in pairs(hook) do
			conn:hook(name, func)
		end
	end
end


function mt_irc:register_hook(name, func)
	self.registered_hooks[name] = self.registered_hooks[name] or {}
	table.insert(self.registered_hooks[name], func)
end


function mt_irc.hooks.raw(line)
	if mt_irc.config.debug then
		print("RECV: "..line)
	end
end


function mt_irc.hooks.send(line)
	if mt_irc.config.debug then
		print("SEND: "..line)
	end
end


function mt_irc.hooks.chat(user, channel, message)
	-- Strip bold, underline, and colors
	message = message:gsub('\2', '')
	message = message:gsub('\31', '')
	message = message:gsub('\3[0-9][0-9,]*', '')

	if string.sub(message, 1, 1) == string.char(1) then
		mt_irc.conn:invoke("OnCTCP", user, channel, message)
		return
	end

	if channel == mt_irc.conn.nick then
		mt_irc.conn:invoke("PrivateMessage", user, message)
	else
		mt_irc.conn:invoke("OnChannelChat", user, channel, message)
	end
end


function mt_irc.hooks.ctcp(user, channel, message)
	if message:sub(2, 7):upper() == "ACTION" and
			channel ~= mt_irc.conn.nick then
		local action = message:sub(9, -2)
		mt_irc:sendLocal(("* %s@IRC %s"):format(user.nick, action))
	elseif message:sub(2, 8):upper() == "VERSION" then
		mt_irc:queueMsg(mt_irc.msgs.notice(user.nick,
				("\1VERSION Minetest IRC mod %s\1")
				:format(mt_irc.version)))
	elseif message:sub(2, 5):upper() == "PING" then
		local ts = message:sub(7, -2)
		mt_irc:queueMsg(mt_irc.msgs.notice(user.nick,
				("\1PING %s\1"):format(ts)))
	elseif message:sub(2, 5):upper() == "TIME" then
		mt_irc:queueMsg(mt_irc.msgs.notice(user.nick,
				("\1TIME %s\1"):format(os.date())))
	end
end


function mt_irc.hooks.channelChat(user, channel, message)
	-- Support multiple servers in a channel better by converting:
	-- "<server@IRC> <player> message" into "<player@server> message"
	-- "<server@IRC> *** player joined/left the game" into "*** player@server joined/left the game"
	-- and "<server@IRC> * player orders a pizza" into "* player@server orders a pizza"
	local foundchat, _, chatnick, chatmessage
		= message:find("^<([^>]+)> (.*)$")
	local foundjoin, _, joinnick
		= message:find("^%*%*%* ([^%s]+) joined the game$")
	local foundleave, _, leavenick
		= message:find("^%*%*%* ([^%s]+) left the game$")
	local foundaction, _, actionnick, actionmessage
		= message:find("^%* ([^%s]+) (.*)$")

	if foundchat then
		mt_irc:sendLocal(("<%s@%s> %s")
				:format(chatnick, user.nick, chatmessage))
	elseif foundjoin then
		mt_irc:sendLocal(("*** %s@%s joined the game")
				:format(joinnick, user.nick))
	elseif foundleave then
		mt_irc:sendLocal(("*** %s@%s left the game")
				:format(leavenick, user.nick))
	elseif foundaction then
		mt_irc:sendLocal(("* %s@%s %s")
				:format(actionnick, user.nick, actionmessage))
	else
		mt_irc:sendLocal(("<%s@IRC> %s"):format(user.nick, message))
	end
end


function mt_irc.hooks.pm(user, message)
	local player_to
	local msg
	if message:sub(1, 1) == "@" then
		local found, _, player_to, message = message:find("^.([^%s]+)%s(.+)$")
		if not mt_irc.joined_players[player_to] then
			mt_irc:say(user.nick, "User '"..player_to.."' has parted.")
			return
		elseif not minetest.get_player_by_name(player_to) then
			mt_irc:say(user.nick, "User '"..player_to.."' is not in the game.")
			return
		end
		minetest.chat_send_player(player_to,
				"PM from "..user.nick.."@IRC: "..message, false)
		mt_irc:say(user.nick, "Message sent!")
	elseif message:sub(1, 1) == "!" then
		mt_irc:bot_command(user, message:sub(2))
		return
	else
		mt_irc:say(user.nick, "Invalid command. Use '"
				..mt_irc.config.command_prefix
				.."list' to see possible commands.")
		return
	end
end


function mt_irc.hooks.kick(channel, target, prefix, reason)
	if target == mt_irc.conn.nick then
		minetest.chat_send_all("IRC: kicked from "..channel.." by "..prefix.nick..".")
		mt_irc:disconnect("Kicked")
	else
		mt_irc:sendLocal(("-!- %s was kicked from %s by %s [%s]")
				:format(target, channel, prefix.nick, reason))
	end
end


function mt_irc.hooks.notice(user, target, message)
	if not user.nick then return end --Server NOTICEs
	if target == mt_irc.conn.nick then return end
	mt_irc:sendLocal("--"..user.nick.."@IRC-- "..message)
end


function mt_irc.hooks.mode(user, target, modes, ...)
	local by = ""
	if user.nick then
		by = " by "..user.nick
	end
	local options = ""
	for _, option in pairs({...}) do
		options = options.." "..option
	end
	minetest.chat_send_all(("-!- mode/%s [%s%s]%s")
			:format(target, modes, options, by))
end


function mt_irc.hooks.nick(user, newNick)
	mt_irc:sendLocal(("-!- %s is now known as %s")
			:format(user.nick, newNick))
end


function mt_irc.hooks.join(user, channel)
	mt_irc:sendLocal(("-!- %s joined %s")
			:format(user.nick, channel))
end


function mt_irc.hooks.part(user, channel, reason)
	reason = reason or ""
	mt_irc:sendLocal(("-!- %s has left %s [%s]")
			:format(user.nick, channel, reason))
end


function mt_irc.hooks.quit(user, reason)
	mt_irc:sendLocal(("-!- %s has quit [%s]")
			:format(user.nick, reason))
end


function mt_irc.hooks.disconnect(message, isError)
	mt_irc.connected = false
	if isError then
		minetest.log("error",  "IRC: Error: Disconnected, reconnecting in one minute.")
		minetest.chat_send_all("IRC: Error: Disconnected, reconnecting in one minute.")
		minetest.after(60, mt_irc.connect, mt_irc)
	else
		minetest.log("action", "IRC: Disconnected.")
		minetest.chat_send_all("IRC: Disconnected.")
	end
end


function mt_irc.hooks.preregister(conn)
	if not (mt_irc.config.SASLUser and mt_irc.config.SASLPass) then return end
	local authString = mt_irc.b64e(
		("%s\x00%s\x00%s"):format(
		mt_irc.config.SASLUser,
		mt_irc.config.SASLUser,
		mt_irc.config.SASLPass)
	)
	conn:send("CAP REQ sasl")
	conn:send("AUTHENTICATE PLAIN")
	conn:send("AUTHENTICATE "..authString)
	--LuaIRC will send CAP END
end


mt_irc:register_hook("PreRegister",     mt_irc.hooks.preregister)
mt_irc:register_hook("OnRaw",           mt_irc.hooks.raw)
mt_irc:register_hook("OnSend",          mt_irc.hooks.send)
mt_irc:register_hook("OnChat",          mt_irc.hooks.chat)
mt_irc:register_hook("OnPart",          mt_irc.hooks.part)
mt_irc:register_hook("OnKick",          mt_irc.hooks.kick)
mt_irc:register_hook("OnJoin",          mt_irc.hooks.join)
mt_irc:register_hook("OnQuit",          mt_irc.hooks.quit)
mt_irc:register_hook("NickChange",      mt_irc.hooks.nick)
mt_irc:register_hook("OnCTCP",          mt_irc.hooks.ctcp)
mt_irc:register_hook("PrivateMessage",  mt_irc.hooks.pm)
mt_irc:register_hook("OnNotice",        mt_irc.hooks.notice)
mt_irc:register_hook("OnChannelChat",   mt_irc.hooks.channelChat)
mt_irc:register_hook("OnModeChange",    mt_irc.hooks.mode)
mt_irc:register_hook("OnDisconnect",    mt_irc.hooks.disconnect)