Mercurial > prosody-modules
view mod_muc_eventsource/mod_muc_eventsource.lua @ 3568:6b3181fe5617
mod_auth_token: Timezone fix for TOTP checking
luatz.time() returns milliseconds since epoch which is in UTC time, so we don't
need to convert to UTC with gmtime.
By calling gmtime, TOTP validation was failing when this module wasn't running
on machine set to UTC time.
author | JC Brand <jc@opkode.com> |
---|---|
date | Thu, 02 May 2019 11:07:27 +0200 |
parents | 39485b9bbdd6 |
children |
line wrap: on
line source
module:depends("http"); local nodeprep = require "util.encodings".stringprep.nodeprep; local jid_split = require "util.jid".split; local json = require "util.json"; local streams = {}; function client_closed(response) local node = response._eventsource_node; module:log("debug", "Destroying client for %q", node); streams[node][response] = nil; if next(streams[node]) == nil then streams[node] = nil; end end function serve_stream(event, node) local response = event.response; node = nodeprep(node); if node == nil then return 400; end module:log("debug", "Client subscribed to: %s", node); response.on_destroy = client_closed; response._eventsource_node = node; response.conn:write(table.concat({ "HTTP/1.1 200 OK"; "Content-Type: text/event-stream"; "Access-Control-Allow-Origin: *"; "Access-Control-Allow-Methods: GET"; "Access-Control-Max-Age: 7200"; ""; ""; }, "\r\n")); local clientlist = streams[node]; if not clientlist then clientlist = {}; streams[node] = clientlist; end clientlist[response] = response.conn; return true; end function handle_message(event) local room, stanza = event.room, event.stanza; local node = (jid_split(event.room.jid)); local clientlist = streams[node]; if not clientlist then module:log("debug", "No clients for %q", node); return; end -- Extract body from message local body = event.stanza:get_child_text("body"); if not body then return; end local nick = select(3, jid_split(stanza.attr.from)); -- Encode body and broadcast to eventsource subscribers local json_data = json.encode({ nick = nick; body = body; }); local data = "data: "..json_data:gsub("\n", "\ndata: \n").."\n\n"; for response, conn in pairs(clientlist) do conn:write(data); end end module:provides("http", { name = "eventsource"; route = { ["GET /*"] = serve_stream; }; }); module:hook("muc-broadcast-message", handle_message);