Add function to get a list of users in a channel

trunk
mio 2022-04-28 03:20:09 +00:00
parent f5302105f0
commit 20652e5e31
1 changed files with 59 additions and 9 deletions

View File

@ -51,9 +51,10 @@ itte.config = {
dc_svr = { "conn", "Connection to {{server}} closed." },
exit = { "conn", "Exiting ..." },
listen = { "listen" },
names = { "names", "channel", "users" },
nickserv = { "auth_nickserv" },
privmsg = { "sender", "recipient", "reply_to", "code", "code_params",
"body" },
privmsg = { "privmsg", "sender", "recipient", "reply_to", "code",
"code_params", "body" },
redact = { "********" },
send = { "send" },
svrs_not_found = { "config", "Error: servers not found." },
@ -217,6 +218,11 @@ function itte.get_commands(svr)
check = "MODE",
resp = "JOIN ",
},
names = {
-- Check and response are reversed
check = "353",
resp = "NAMES",
},
nick = {
check = "NOTICE",
resp = "NICK " .. svr.nick,
@ -551,6 +557,7 @@ itte.docs.traverse_channels = [[ (context_table, mode_str, channels_table)
]]
function itte.traverse_channels(cxt, mode, chans)
local channels = cxt.channels
cxt.chan_meta = {}
if chans ~= nil then channels = chans end
if mode == "join" then
@ -560,6 +567,7 @@ function itte.traverse_channels(cxt, mode, chans)
for c = 1, #channels do
if not util.has_key(cxt.channels, channels[c]) then
table.insert(cxt.channels, channels[c])
cxt.chan_meta[channels[c]:sub(2)] = {}
end
end
@ -570,6 +578,7 @@ function itte.traverse_channels(cxt, mode, chans)
for c = 1, #channels do
if util.has_key(cxt.channels, channels[c]) then
table.remove(cxt.channels, util.find_key(cxt.channels, channels[c]))
cxt.chan_meta[channels[c]:sub(2)] = nil
end
end
end
@ -620,18 +629,52 @@ function itte.parse_privmsg(cxt, str)
body = itte.config.debugs.redact[1]
code_params = itte.config.debugs.redact[1]
end
util.debug("privmsg", "{ " ..
itte.config.debugs.privmsg[1] .. ": " .. msg.sender ..
", " .. itte.config.debugs.privmsg[2] .. ": " .. msg.recipient ..
", " .. itte.config.debugs.privmsg[3] .. ": " .. msg.reply_to ..
", " .. itte.config.debugs.privmsg[4] .. ": " .. msg.code ..
", " .. itte.config.debugs.privmsg[5] .. ": " .. code_params ..
", " .. itte.config.debugs.privmsg[6] .. ": " .. body ..
util.debug(itte.config.debugs.privmsg[1], "{ " ..
itte.config.debugs.privmsg[2] .. ": " .. msg.sender ..
", " .. itte.config.debugs.privmsg[3] .. ": " .. msg.recipient ..
", " .. itte.config.debugs.privmsg[4] .. ": " .. msg.reply_to ..
", " .. itte.config.debugs.privmsg[5] .. ": " .. msg.code ..
", " .. itte.config.debugs.privmsg[6] .. ": " .. code_params ..
", " .. itte.config.debugs.privmsg[7] .. ": " .. body ..
" }", itte.config.debug)
return msg
end
itte.docs.parse_names = [[ (context_table, data_str)
Given the context table and a NAMES data string, parse the string and
return the channel name and a table of users.
]]
function itte.parse_names(cxt, str)
-- Separator marks the start of the message body
body_sep, _ = string.find(str, ":", 2)
local channel = string.sub(str, string.find(str, "#"),
string.find(str, ":", 2) - 2)
local names = util.split_str(string.sub(str, body_sep + 1))
util.debug(itte.config.debugs.names[1],
itte.config.debugs.names[2] .. ": " .. channel ..
", " .. itte.config.debugs.names[3] .. ": { " .. table.concat(names, ", ")
.. " }", itte.config.debug)
return channel, names
end
itte.docs.get_users = [[ (context_table, channel_str)
Given the context table and a channel, return a table of users in the
channel.
]]
function itte.get_users(cxt, str)
local chan = str
if string.find(chan, "#") == 1 then chan = str:sub(2) end
if not util.has_key(cxt.chan_meta, chan) then
do return end
end
-- Query server for the latest list of names
itte.send_command(cxt.con, cxt.cmds.names.resp .. " #" .. chan)
return cxt.chan_meta[chan].users
end
-- ---------------------------------------------------------------------------
-- Service code handlers
-- ---------------------------------------------------------------------------
@ -840,6 +883,13 @@ function itte.listen(name, str)
itte.send_command(cxt.con, string.gsub(str, cxt.cmds.ping.check,
cxt.cmds.ping.resp))
-- Update channel names list
elseif string.find(str, ":" .. cxt.host .. " " .. cxt.cmds.names.check) == 1
then
util.debug(itte.config.debugs.listen[1], str, itte.config.debug)
local channel, names = itte.parse_names(cxt, str)
cxt.chan_meta[channel:sub(2)] = { users = names }
-- Respond to service code
elseif util.is_substr(str, cxt.cmds.privmsg.check) then
local msg = itte.parse_privmsg(cxt, str)