|
2455
|
1 -- Copyright (C) 2016 Kim Alvefur |
|
|
2 -- |
|
|
3 |
|
|
4 module:depends"csi" |
|
|
5 module:depends"track_muc_joins" |
|
|
6 local jid = require "util.jid"; |
|
|
7 local new_queue = require "util.queue".new; |
|
|
8 |
|
|
9 local function new_pump(output, ...) |
|
|
10 -- luacheck: ignore 212/self |
|
|
11 local q = new_queue(...); |
|
|
12 local flush = true; |
|
|
13 function q:pause() |
|
|
14 flush = false; |
|
|
15 end |
|
|
16 function q:resume() |
|
|
17 flush = true; |
|
|
18 return q:flush(); |
|
|
19 end |
|
|
20 local push = q.push; |
|
|
21 function q:push(item) |
|
|
22 local ok = push(self, item); |
|
|
23 if not ok then |
|
|
24 q:flush(); |
|
|
25 output(item, self); |
|
|
26 elseif flush then |
|
|
27 return q:flush(); |
|
|
28 end |
|
|
29 return true; |
|
|
30 end |
|
|
31 function q:flush() |
|
|
32 local item = self:pop(); |
|
|
33 while item do |
|
|
34 output(item, self); |
|
|
35 item = self:pop(); |
|
|
36 end |
|
|
37 return true; |
|
|
38 end |
|
|
39 return q; |
|
|
40 end |
|
|
41 |
|
|
42 -- TODO delay stamps |
|
|
43 -- local dt = require "util.datetime"; |
|
|
44 |
|
|
45 local function is_important(stanza, session) |
|
|
46 local st_name = stanza.name; |
|
|
47 if not st_name then return false; end |
|
|
48 local st_type = stanza.attr.type; |
|
|
49 if st_name == "presence" then |
|
|
50 -- TODO check for MUC status codes? |
|
|
51 return false; |
|
|
52 elseif st_name == "message" then |
|
|
53 if st_type == "headline" then |
|
|
54 return false; |
|
|
55 end |
|
|
56 local body = stanza:get_child_text("body"); |
|
|
57 if not body then return false; end |
|
|
58 if st_type == "groupchat" then |
|
|
59 if body:find(session.username, 1, true) then return true; end |
|
|
60 local rooms = session.rooms_joined; |
|
|
61 if not rooms then return false; end |
|
|
62 local room_nick = rooms[jid.bare(stanza.attr.from)]; |
|
|
63 if room_nick and body:find(room_nick, 1, true) then return true; end |
|
|
64 return false; |
|
|
65 end |
|
|
66 return body; |
|
|
67 end |
|
|
68 return true; |
|
|
69 end |
|
|
70 |
|
|
71 module:hook("csi-client-inactive", function (event) |
|
|
72 local session = event.origin; |
|
|
73 if session.pump then |
|
|
74 session.pump:pause(); |
|
|
75 else |
|
|
76 session._orig_send = session.send; |
|
|
77 local pump = new_pump(session.send, 100); |
|
|
78 pump:pause(); |
|
|
79 session.pump = pump; |
|
|
80 function session.send(stanza) |
|
|
81 pump:push(stanza); |
|
|
82 if is_important(stanza, session) then |
|
|
83 pump:flush(); |
|
|
84 end |
|
|
85 return true; |
|
|
86 end |
|
|
87 end |
|
|
88 end); |
|
|
89 |
|
|
90 module:hook("csi-client-active", function (event) |
|
|
91 local session = event.origin; |
|
|
92 if session.pump then |
|
|
93 session.pump:resume(); |
|
|
94 end |
|
|
95 end); |
|
|
96 |
