Mercurial > prosody-hg
comparison plugins/muc/muc.lib.lua @ 6138:fa746d834424
plugins/muc/muc.lib: Add muc-get-history event; it uses an iterator in the event object so that messages don't need to be all in memory at once
| author | daurnimator <quae@daurnimator.com> |
|---|---|
| date | Mon, 24 Mar 2014 13:34:06 -0400 |
| parents | 7db24f237a83 |
| children | 544f75256883 |
comparison
equal
deleted
inserted
replaced
| 6137:7db24f237a83 | 6138:fa746d834424 |
|---|---|
| 191 end | 191 end |
| 192 end | 192 end |
| 193 | 193 |
| 194 return maxchars, maxstanzas, since | 194 return maxchars, maxstanzas, since |
| 195 end | 195 end |
| 196 -- Get history for 'to' | 196 |
| 197 function room_mt:get_history(to, maxchars, maxstanzas, since) | 197 module:hook("muc-get-history", function(event) |
| 198 local history = self._data['history']; -- send discussion history | 198 local room = event.room |
| 199 if not history then return function() end end | 199 local history = room._data['history']; -- send discussion history |
| 200 if not history then return nil end | |
| 200 local history_len = #history | 201 local history_len = #history |
| 201 | 202 |
| 202 maxstanzas = maxstanzas or history_len | 203 local to = event.to |
| 204 local maxchars = event.maxchars | |
| 205 local maxstanzas = event.maxstanzas or history_len | |
| 206 local since = event.since | |
| 203 local n = 0; | 207 local n = 0; |
| 204 local charcount = 0; | 208 local charcount = 0; |
| 205 for i=history_len,1,-1 do | 209 for i=history_len,1,-1 do |
| 206 local entry = history[i]; | 210 local entry = history[i]; |
| 207 if maxchars then | 211 if maxchars then |
| 216 if n + 1 > maxstanzas then break; end | 220 if n + 1 > maxstanzas then break; end |
| 217 n = n + 1; | 221 n = n + 1; |
| 218 end | 222 end |
| 219 | 223 |
| 220 local i = history_len-n+1 | 224 local i = history_len-n+1 |
| 221 return function() | 225 function event:next_stanza() |
| 222 if i > history_len then return nil end | 226 if i > history_len then return nil end |
| 223 local entry = history[i] | 227 local entry = history[i] |
| 224 local msg = entry.stanza | 228 local msg = entry.stanza |
| 225 msg.attr.to = to; | 229 msg.attr.to = to; |
| 226 i = i + 1 | 230 i = i + 1 |
| 227 return msg | 231 return msg |
| 228 end | 232 end |
| 229 end | 233 return true; |
| 230 function room_mt:send_history(to, stanza) | 234 end) |
| 235 | |
| 236 function room_mt:send_history(stanza) | |
| 231 local maxchars, maxstanzas, since = parse_history(stanza) | 237 local maxchars, maxstanzas, since = parse_history(stanza) |
| 232 for msg in self:get_history(to, maxchars, maxstanzas, since) do | 238 local event = { |
| 239 room = self; | |
| 240 to = stanza.attr.from; -- `to` is required to calculate the character count for `maxchars` | |
| 241 maxchars = maxchars, maxstanzas = maxstanzas, since = since; | |
| 242 next_stanza = function() end; -- events should define this iterator | |
| 243 } | |
| 244 module:fire_event("muc-get-history", event) | |
| 245 for msg in event.next_stanza , event do | |
| 233 self:_route_stanza(msg); | 246 self:_route_stanza(msg); |
| 234 end | 247 end |
| 235 end | 248 end |
| 236 function room_mt:send_subject(to) | 249 function room_mt:send_subject(to) |
| 237 if self._data['subject'] then | 250 if self._data['subject'] then |