Mercurial > prosody-modules
view mod_muc_eventsource/mod_muc_eventsource.lua @ 4895:2542fd80cd15
mod_turn_external: Fix type of config option (thanks mirux)
There was a separate boolean option to enable TLS before, but it was
merged with the port number option and it seems the typed API interface
got confused.
Backport of trunk rev aa7a8aa64d3f
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Tue, 15 Feb 2022 21:51:52 +0100 |
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);