prosody-modules: mod_s2soutinjection/mod_s2soutinjection.lua annotate

annotate mod_s2soutinjection/mod_s2soutinjection.lua @ 4579:b305814bd930

mod_muc_dicebot: A thing to roll dice Do you see what happens, Jitsi? Do you see what happens when you make it hard for me to use a proper bot? This is what happens, Jitsi. This is what happens when you meet a stranger in the alps! Ahem. In all seriousness, this is more of a quick hack than anything else. It will look for `.r` in MUC messages and if it finds it, it'll interpret it as an instruction to roll a few dice. Injects the results in the body of the message. Eats the message alive if it is malformed.

author	Jonas Schäfer <jonas@wielicki.name>
date	Sat, 29 May 2021 15:17:05 +0200
parents	864fefec1c07
children	f4a9e804c457

rev	line source
1089 4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	1 local st = require"util.stanza";
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	2 local new_ip = require"util.ip".new_ip;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	3 local new_outgoing = require"core.s2smanager".new_outgoing;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	4 local bounce_sendq = module:depends"s2s".route_to_new_session.bounce_sendq;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	5 local s2sout = module:depends"s2s".route_to_new_session.s2sout;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	6
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	7 local injected = module:get_option("s2s_connect_overrides");
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	8
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	9 local function isip(addr)
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	10 return not not (addr and addr:match("^%d+%.%d+%.%d+%.%d+$") or addr:match("^[%x:]:[%x:]-:[%x:]$"));
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	11 end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	12
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	13 module:hook("route/remote", function(event)
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	14 local from_host, to_host, stanza = event.from_host, event.to_host, event.stanza;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	15 local inject = injected and injected[to_host];
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	16 if not inject then return end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	17 log("debug", "opening a new outgoing connection for this stanza");
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	18 local host_session = new_outgoing(from_host, to_host);
4557 864fefec1c07 mod_s2soutinjection: Set version 1.0 for outgoing streams (thanks moparisthebest) Matthew Wild <mwild1@gmail.com> parents: 1089 diff changeset	19 host_session.version = 1;
1089 4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	20
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	21 -- Store in buffer
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	22 host_session.bounce_sendq = bounce_sendq;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	23 host_session.sendq = { {tostring(stanza), stanza.attr.type ~= "error" and stanza.attr.type ~= "result" and st.reply(stanza)} };
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	24 log("debug", "stanza [%s] queued until connection complete", tostring(stanza.name));
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	25
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	26 local ip_hosts, srv_hosts = {}, {};
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	27 host_session.srv_hosts = srv_hosts;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	28 host_session.srv_choice = 0;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	29
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	30 if type(inject) == "string" then inject = { inject } end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	31
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	32 for _, item in ipairs(inject) do
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	33 local host, port = item[1] or item, tonumber(item[2]) or 5269;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	34 if isip(host) then
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	35 ip_hosts[#ip_hosts+1] = { ip = new_ip(host), port = port }
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	36 else
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	37 srv_hosts[#srv_hosts+1] = { target = host, port = port }
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	38 end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	39 end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	40 if #ip_hosts > 0 then
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	41 host_session.ip_hosts = ip_hosts;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	42 host_session.ip_choice = 0; -- Incremented by try_next_ip
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	43 s2sout.try_next_ip(host_session);
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	44 return true;
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	45 end
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	46
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	47 return s2sout.try_connect(host_session, host_session.srv_hosts[1].target, host_session.srv_hosts[1].port);
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	48 end, -2);
4057f176be7b mod_s2soutinjection: Initial commit, variant of mod_srvinjection using 0.9+ APIs Kim Alvefur <zash@zash.se> parents: diff changeset	49

Mercurial > prosody-modules

annotate mod_s2soutinjection/mod_s2soutinjection.lua @ 4579:b305814bd930