Mercurial > prosody-modules
annotate mod_archive_muc/mod_archive_muc.lua @ 554:a2b0174b5c48
mod_muc_limits: New module to impose overall rate-limits on a MUC (not on individual users)
| author | Matthew Wild <mwild1@gmail.com> |
|---|---|
| date | Sun, 15 Jan 2012 01:08:15 +0000 |
| parents | bbe4df968099 |
| children |
| rev | line source |
|---|---|
| 237 | 1 -- Prosody IM |
| 2 -- Copyright (C) 2010 Dai Zhiwei | |
| 3 -- | |
| 4 -- This project is MIT/X11 licensed. Please see the | |
| 5 -- COPYING file in the source package for more information. | |
| 6 -- | |
| 7 | |
| 8 local st = require "util.stanza"; | |
| 9 local dm = require "util.datamanager"; | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
10 local jid_compare, jid_split, jid_bare = require "util.jid".compare, require "util.jid".bare, require "util.jid".split; |
|
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
11 local datetime = require "util.datetime".datetime; |
|
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
12 local user_exists = require "core.usermanager".user_exists; |
|
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
13 local is_contact_subscribed = require "core.rostermanager".is_contact_subscribed; |
| 237 | 14 |
| 15 local PREFS_DIR = "archive_muc_prefs"; | |
| 16 local ARCHIVE_DIR = "archive_muc"; | |
| 17 | |
|
243
6202ce4d12d6
mod_archive: added some config options.
shinysky<shinysky1986(AT)gmail.com>
parents:
240
diff
changeset
|
18 local AUTO_MUC_ARCHIVING_ENABLED = module:get_option_boolean("auto_muc_archiving_enabled", true); |
| 237 | 19 |
|
476
2c85635318a5
mod_archive_muc: Fixed a nil global access.
Waqas Hussain <waqas20@gmail.com>
parents:
243
diff
changeset
|
20 local NULL = {}; |
|
2c85635318a5
mod_archive_muc: Fixed a nil global access.
Waqas Hussain <waqas20@gmail.com>
parents:
243
diff
changeset
|
21 |
| 237 | 22 module:add_feature("urn:xmpp:archive#preferences"); |
| 23 module:add_feature("urn:xmpp:archive#management"); | |
| 24 | |
| 25 ------------------------------------------------------------ | |
| 26 -- Utils | |
| 27 ------------------------------------------------------------ | |
|
240
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
28 local function trim(s) |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
29 return (string.gsub(s, "^%s*(.-)%s*$", "%1")) |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
30 end |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
31 |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
32 local function clean_up(t) |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
33 for i = #t, 1, -1 do |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
34 if type(t[i]) == 'table' then |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
35 clean_up(t[i]); |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
36 elseif type(t[i]) == 'string' and trim(t[i]) == '' then |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
37 table.remove(t, i); |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
38 end |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
39 end |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
40 end |
|
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
41 |
| 237 | 42 local function load_prefs(node, host) |
| 43 return st.deserialize(dm.load(node, host, PREFS_DIR)); | |
| 44 end | |
| 45 | |
| 46 local function store_prefs(data, node, host) | |
|
240
ef0b580f434d
mod_archive_muc: clean up '\n ' in preference stanza
shinysky<shinysky1986(AT)gmail.com>
parents:
238
diff
changeset
|
47 clean_up(data); |
| 237 | 48 dm.store(node, host, PREFS_DIR, st.preserialize(data)); |
| 49 end | |
| 50 | |
| 51 local function match_jid(rule, id) | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
52 return not rule or jid_compare(id, rule); |
| 237 | 53 end |
| 54 | |
| 55 local function is_earlier(start, coll_start) | |
| 56 return not start or start <= coll_start; | |
| 57 end | |
| 58 | |
| 59 local function is_later(endtime, coll_start) | |
| 60 return not endtime or endtime >= coll_start; | |
| 61 end | |
| 62 | |
| 63 ------------------------------------------------------------ | |
| 64 -- Preferences | |
| 65 ------------------------------------------------------------ | |
| 66 local function preferences_handler(event) | |
| 67 local origin, stanza = event.origin, event.stanza; | |
| 68 module:log("debug", "-- Enter muc preferences_handler()"); | |
| 69 module:log("debug", "-- muc pref:\n%s", tostring(stanza)); | |
| 70 if stanza.attr.type == "get" then | |
| 71 local data = load_prefs(origin.username, origin.host); | |
| 72 if data then | |
| 73 origin.send(st.reply(stanza):add_child(data)); | |
| 74 else | |
| 75 origin.send(st.reply(stanza)); | |
| 76 end | |
| 77 elseif stanza.attr.type == "set" then | |
| 78 local node, host = origin.username, origin.host; | |
| 79 if stanza.tags[1] and stanza.tags[1].name == 'prefs' then | |
| 80 store_prefs(stanza.tags[1], node, host); | |
| 81 origin.send(st.reply(stanza)); | |
| 82 local user = bare_sessions[node.."@"..host]; | |
| 83 local push = st.iq({type="set"}); | |
| 84 push:add_child(stanza.tags[1]); | |
| 85 for _, res in pairs(user and user.sessions or NULL) do -- broadcast to all resources | |
| 86 if res.presence then -- to resource | |
| 87 push.attr.to = res.full_jid; | |
| 88 res.send(push); | |
| 89 end | |
| 90 end | |
| 91 end | |
| 92 end | |
| 93 return true; | |
| 94 end | |
| 95 | |
| 96 ------------------------------------------------------------ | |
| 97 -- Archive Management | |
| 98 ------------------------------------------------------------ | |
| 99 local function management_handler(event) | |
| 100 module:log("debug", "-- Enter muc management_handler()"); | |
| 101 local origin, stanza = event.origin, event.stanza; | |
| 102 local node, host = origin.username, origin.host; | |
| 103 local data = dm.list_load(node, host, ARCHIVE_DIR); | |
| 104 local elem = stanza.tags[1]; | |
| 105 local resset = {} | |
| 106 if data then | |
| 107 for i = #data, 1, -1 do | |
| 108 local forwarded = st.deserialize(data[i]); | |
| 109 local res = (match_jid(elem.attr["with"], forwarded.tags[2].attr.from) | |
| 110 or match_jid(elem.attr["with"], forwarded.tags[2].attr.to)) | |
| 111 and is_earlier(elem.attr["start"], forwarded.tags[1].attr["stamp"]) | |
| 112 and is_later(elem.attr["end"], forwarded.tags[1].attr["stamp"]); | |
| 113 if res then | |
| 114 table.insert(resset, forwarded); | |
| 115 end | |
| 116 end | |
| 117 for i = #resset, 1, -1 do | |
| 118 local res = st.message({to = stanza.attr.from, id=st.new_id()}); | |
| 119 res:add_child(resset[i]); | |
| 120 origin.send(res); | |
| 121 end | |
| 122 end | |
| 123 origin.send(st.reply(stanza)); | |
| 124 return true; | |
| 125 end | |
| 126 | |
| 127 ------------------------------------------------------------ | |
| 128 -- Message Handler | |
| 129 ------------------------------------------------------------ | |
| 130 local function is_in(list, jid) | |
| 131 for _,v in ipairs(list) do | |
| 132 if match_jid(v:get_text(), jid) then -- JID Matching | |
| 133 return true; | |
| 134 end | |
| 135 end | |
| 136 return false; | |
| 137 end | |
| 138 | |
| 139 local function apply_pref(node, host, jid) | |
| 140 local pref = load_prefs(node, host); | |
| 141 if not pref then | |
|
243
6202ce4d12d6
mod_archive: added some config options.
shinysky<shinysky1986(AT)gmail.com>
parents:
240
diff
changeset
|
142 return AUTO_MUC_ARCHIVING_ENABLED; |
| 237 | 143 end |
| 144 local always = pref:child_with_name('always'); | |
| 145 if always and is_in(always, jid) then | |
| 146 return true; | |
| 147 end | |
| 148 local never = pref:child_with_name('never'); | |
| 149 if never and is_in(never, jid) then | |
| 150 return false; | |
| 151 end | |
| 152 local default = pref.attr['default']; | |
| 153 if default == 'roster' then | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
154 return is_contact_subscribed(node, host, jid_bare(jid)); |
| 237 | 155 elseif default == 'always' then |
| 156 return true; | |
| 157 elseif default == 'never' then | |
| 158 return false; | |
| 159 end | |
|
243
6202ce4d12d6
mod_archive: added some config options.
shinysky<shinysky1986(AT)gmail.com>
parents:
240
diff
changeset
|
160 return AUTO_MUC_ARCHIVING_ENABLED; |
| 237 | 161 end |
| 162 | |
| 163 local function store_msg(msg, node, host) | |
| 164 local forwarded = st.stanza('forwarded', {xmlns='urn:xmpp:forward:tmp'}); | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
165 forwarded:tag('delay', {xmlns='urn:xmpp:delay',stamp=datetime()}):up(); |
| 237 | 166 forwarded:add_child(msg); |
| 167 dm.list_append(node, host, ARCHIVE_DIR, st.preserialize(forwarded)); | |
| 168 end | |
| 169 | |
| 170 local function msg_handler(data) | |
| 171 module:log("debug", "-- Enter muc msg_handler()"); | |
| 172 local origin, stanza = data.origin, data.stanza; | |
| 173 local body = stanza:child_with_name("body"); | |
| 174 if body then | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
175 local from_node, from_host = jid_split(stanza.attr.from); |
|
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
176 local to_node, to_host = jid_split(stanza.attr.to); |
|
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
177 if user_exists(from_node, from_host) and apply_pref(from_node, from_host, stanza.attr.to) then |
| 237 | 178 store_msg(stanza, from_node, from_host); |
| 179 end | |
|
477
bbe4df968099
mod_archive_muc: A little refactoring to improve code search.
Waqas Hussain <waqas20@gmail.com>
parents:
476
diff
changeset
|
180 if user_exists(to_node, to_host) and apply_pref(to_node, to_host, stanza.attr.from) then |
| 237 | 181 store_msg(stanza, to_node, to_host); |
| 182 end | |
| 183 end | |
| 184 | |
| 185 return nil; | |
| 186 end | |
| 187 | |
| 188 -- Preferences | |
| 189 module:hook("iq/self/urn:xmpp:archive#preferences:prefs", preferences_handler); | |
| 190 -- Archive management | |
| 191 module:hook("iq/self/urn:xmpp:archive#management:query", management_handler); | |
| 192 | |
| 193 module:hook("message/bare", msg_handler, 20); | |
| 194 |