LuaIRC/messages.lua

144 lines
2.8 KiB
Lua

local assert = assert
local setmetatable = setmetatable
local unpack = unpack
local pairs = pairs
module "irc"
msgs = {}
local msg_meta = {}
msg_meta.__index = msg_meta
function Message(cmd, args)
return setmetatable({
command = cmd,
args = args or {},
}, msg_meta)
end
function msg_meta:toRFC1459()
s = ""
if self.tags then
s = s.."@"
for key, value in pairs(self.tags) do
s = s..key
if value ~= true then
assert(not value:find("[%z\07\r\n; ]"),
"NUL, BELL, CR, LF, semicolon, and"
.." space are not allowed in RFC1459"
.." formated tag values.")
s = s.."="..value
end
s = s..";"
end
-- Strip trailing semicolon
s = s:sub(1, -2)
s = s.." "
end
s = s..self.command
argnum = #self.args
for i = 1, argnum do
local arg = self.args[i]
local startsWithColon = (arg:sub(1, 1) == ":")
local hasSpace = arg:find(" ")
if i == argnum and (hasSpace or startsWithColon) then
s = s.." :"
else
assert(not hasSpace and not startsWithColon,
"Message arguments can not be "
.."serialized to RFC1459 format")
s = s.." "
end
s = s..arg
end
return s
end
function msgs.privmsg(to, text)
return Message("PRIVMSG", {to, text})
end
function msgs.notice(to, text)
return Message("NOTICE", {to, text})
end
function msgs.action(to, text)
return Message("PRIVMSG", {to, ("\x01ACTION %s\x01"):format(text)})
end
function msgs.ctcp(command, to, args)
s = "\x01"..command
if args then
s = ' '..args
end
s = s..'\x01'
return Message("PRIVMSG", {to, s})
end
function msgs.kick(channel, target, reason)
return Message("KICK", {channel, target, reason})
end
function msgs.join(channel, key)
return Message("JOIN", {channel, key})
end
function msgs.part(channel, reason)
return Message("PART", {channel, reason})
end
function msgs.quit(reason)
return Message("QUIT", {reason})
end
function msgs.kill(target, reason)
return Message("KILL", {target, reason})
end
function msgs.kline(time, mask, reason, operreason)
local args = nil
if time then
args = {time, mask, reason..'|'..operreason}
else
args = {mask, reason..'|'..operreason}
end
return Message("KLINE", args)
end
function msgs.whois(nick, server)
local args = nil
if server then
args = {server, nick}
else
args = {nick}
end
return Message("WHOIS", args)
end
function msgs.topic(channel, text)
return Message("TOPIC", {channel, text})
end
function msgs.invite(channel, target)
return Message("INVITE", {channel, target})
end
function msgs.nick(nick)
return Message("NICK", {nick})
end
function msgs.mode(target, modes)
-- We have to split the modes parameter because the mode string and
-- each parameter are seperate arguments (The first command is incorrect)
-- MODE :+ov Nick1 Nick2
-- MODE +ov Nick1 Nick2
mt = split(modes)
return Message("MODE", {target, unpack(mt)})
end