more rdoc-ification
[rbot] / data / rbot / plugins / rss.rb
1 #-- vim:sw=2:et\r
2 #++\r
3 #\r
4 # :title: RSS feed plugin for rbot\r
5 #\r
6 # Author:: Stanislav Karchebny <berkus@madfire.net>\r
7 # Author:: Ian Monroe <ian@monroe.nu>\r
8 # Author:: Mark Kretschmann <markey@web.de>\r
9 # Author:: Giuseppe Bilotta <giuseppe.bilotta@gmail.com>\r
10 #\r
11 # Copyright:: (C) 2004 Stanislav Karchebny\r
12 # Copyright:: (C) 2005 Ian Monroe, Mark Kretschmann\r
13 # Copyright:: (C) 2006-2007 Giuseppe Bilotta\r
14 #\r
15 # License:: MIT license\r
16 \r
17 require 'rss'\r
18 \r
19 module ::RSS\r
20 \r
21   # Make an  'unique' ID for a given item, based on appropriate bot options\r
22   # Currently only suppored is bot.config['rss.show_updated']: when true, the\r
23   # description is included in the uid hashing, otherwise it's not\r
24   #\r
25   def RSS.item_uid_for_bot(item, opts={})\r
26     options = { :show_updated => true}.merge(opts)\r
27     desc = options[:show_updated] ? item.description : nil\r
28     [item.title, item.link, desc].hash\r
29   end\r
30 \r
31   # Add support for Slashdot namespace in RDF. The code is just an adaptation\r
32   # of the DublinCore code.\r
33   unless defined?(SLASH_PREFIX)\r
34     SLASH_PREFIX = 'slash'\r
35     SLASH_URI = "http://purl.org/rss/1.0/modules/slash/"\r
36 \r
37     RDF.install_ns(SLASH_PREFIX, SLASH_URI)\r
38 \r
39     module BaseSlashModel\r
40       def append_features(klass)\r
41         super\r
42 \r
43         return if klass.instance_of?(Module)\r
44         SlashModel::ELEMENT_NAME_INFOS.each do |name, plural_name|\r
45           plural = plural_name || "#{name}s"\r
46           full_name = "#{SLASH_PREFIX}_#{name}"\r
47           full_plural_name = "#{SLASH_PREFIX}_#{plural}"\r
48           klass_name = "Slash#{Utils.to_class_name(name)}"\r
49 \r
50           # This will fail with older version of the Ruby RSS module\r
51           begin\r
52             klass.install_have_children_element(name, SLASH_URI, "*",\r
53                                                 full_name, full_plural_name)\r
54             klass.install_must_call_validator(SLASH_PREFIX, SLASH_URI)\r
55           rescue ArgumentError\r
56             klass.module_eval("install_have_children_element(#{full_name.dump}, #{full_plural_name.dump})")\r
57           end\r
58 \r
59           klass.module_eval(<<-EOC, *get_file_and_line_from_caller(0))\r
60           remove_method :#{full_name}     if method_defined? :#{full_name}\r
61           remove_method :#{full_name}=    if method_defined? :#{full_name}=\r
62           remove_method :set_#{full_name} if method_defined? :set_#{full_name}\r
63 \r
64           def #{full_name}\r
65             @#{full_name}.first and @#{full_name}.first.value\r
66           end\r
67 \r
68           def #{full_name}=(new_value)\r
69             @#{full_name}[0] = Utils.new_with_value_if_need(#{klass_name}, new_value)\r
70           end\r
71           alias set_#{full_name} #{full_name}=\r
72         EOC\r
73         end\r
74       end\r
75     end\r
76 \r
77     module SlashModel\r
78       extend BaseModel\r
79       extend BaseSlashModel\r
80 \r
81       TEXT_ELEMENTS = {\r
82       "department" => nil,\r
83       "section" => nil,\r
84       "comments" =>  nil,\r
85       "hit_parade" => nil\r
86       }\r
87 \r
88       ELEMENT_NAME_INFOS = SlashModel::TEXT_ELEMENTS.to_a\r
89 \r
90       ELEMENTS = TEXT_ELEMENTS.keys\r
91 \r
92       ELEMENTS.each do |name, plural_name|\r
93         module_eval(<<-EOC, *get_file_and_line_from_caller(0))\r
94         class Slash#{Utils.to_class_name(name)} < Element\r
95           include RSS10\r
96 \r
97           content_setup\r
98 \r
99           class << self\r
100             def required_prefix\r
101               SLASH_PREFIX\r
102             end\r
103 \r
104             def required_uri\r
105               SLASH_URI\r
106             end\r
107           end\r
108 \r
109           @tag_name = #{name.dump}\r
110 \r
111           alias_method(:value, :content)\r
112           alias_method(:value=, :content=)\r
113 \r
114           def initialize(*args)\r
115             begin\r
116               if Utils.element_initialize_arguments?(args)\r
117                 super\r
118               else\r
119                 super()\r
120                 self.content = args[0]\r
121               end\r
122             # Older Ruby RSS module\r
123             rescue NoMethodError\r
124               super()\r
125               self.content = args[0]\r
126             end\r
127           end\r
128 \r
129           def full_name\r
130             tag_name_with_prefix(SLASH_PREFIX)\r
131           end\r
132 \r
133           def maker_target(target)\r
134             target.new_#{name}\r
135           end\r
136 \r
137           def setup_maker_attributes(#{name})\r
138             #{name}.content = content\r
139           end\r
140         end\r
141       EOC\r
142       end\r
143     end\r
144 \r
145     class RDF\r
146       class Item; include SlashModel; end\r
147     end\r
148 \r
149     SlashModel::ELEMENTS.each do |name|\r
150       class_name = Utils.to_class_name(name)\r
151       BaseListener.install_class_name(SLASH_URI, name, "Slash#{class_name}")\r
152     end\r
153 \r
154     SlashModel::ELEMENTS.collect! {|name| "#{SLASH_PREFIX}_#{name}"}\r
155   end\r
156 end\r
157 \r
158 \r
159 class ::RssBlob\r
160   attr_accessor :url\r
161   attr_accessor :handle\r
162   attr_accessor :type\r
163   attr :watchers\r
164   attr_accessor :refresh_rate\r
165   attr_accessor :xml\r
166   attr_accessor :title\r
167   attr_accessor :items\r
168   attr_accessor :mutex\r
169 \r
170   def initialize(url,handle=nil,type=nil,watchers=[], xml=nil)\r
171     @url = url\r
172     if handle\r
173       @handle = handle\r
174     else\r
175       @handle = url\r
176     end\r
177     @type = type\r
178     @watchers=[]\r
179     @refresh_rate = nil\r
180     @xml = xml\r
181     @title = nil\r
182     @items = nil\r
183     @mutex = Mutex.new\r
184     sanitize_watchers(watchers)\r
185   end\r
186 \r
187   def dup\r
188     @mutex.synchronize do\r
189       self.class.new(@url,\r
190                      @handle,\r
191                      @type ? @type.dup : nil,\r
192                      @watchers.dup,\r
193                      @xml ? @xml.dup : nil)\r
194     end\r
195   end\r
196 \r
197   # Downcase all watchers, possibly turning them into Strings if they weren't\r
198   def sanitize_watchers(list=@watchers)\r
199     ls = list.dup\r
200     @watchers.clear\r
201     ls.each { |w|\r
202       add_watch(w)\r
203     }\r
204   end\r
205 \r
206   def watched?\r
207     !@watchers.empty?\r
208   end\r
209 \r
210   def watched_by?(who)\r
211     @watchers.include?(who.downcase)\r
212   end\r
213 \r
214   def add_watch(who)\r
215     if watched_by?(who)\r
216       return nil\r
217     end\r
218     @mutex.synchronize do\r
219       @watchers << who.downcase\r
220     end\r
221     return who\r
222   end\r
223 \r
224   def rm_watch(who)\r
225     @mutex.synchronize do\r
226       @watchers.delete(who.downcase)\r
227     end\r
228   end\r
229 \r
230   def to_a\r
231     [@handle,@url,@type,@refresh_rate,@watchers]\r
232   end\r
233 \r
234   def to_s(watchers=false)\r
235     if watchers\r
236       a = self.to_a.flatten\r
237     else\r
238       a = self.to_a[0,3]\r
239     end\r
240     a.compact.join(" | ")\r
241   end\r
242 end\r
243 \r
244 class RSSFeedsPlugin < Plugin\r
245   Config.register Config::IntegerValue.new('rss.head_max',\r
246     :default => 100, :validate => Proc.new{|v| v > 0 && v < 200},\r
247     :desc => "How many characters to use of a RSS item header")\r
248 \r
249   Config.register Config::IntegerValue.new('rss.text_max',\r
250     :default => 200, :validate => Proc.new{|v| v > 0 && v < 400},\r
251     :desc => "How many characters to use of a RSS item text")\r
252 \r
253   Config.register Config::IntegerValue.new('rss.thread_sleep',\r
254     :default => 300, :validate => Proc.new{|v| v > 30},\r
255     :desc => "How many seconds to sleep before checking RSS feeds again")\r
256 \r
257   Config.register Config::BooleanValue.new('rss.show_updated',\r
258     :default => true,\r
259     :desc => "Whether feed items for which the description was changed should be shown as new")\r
260 \r
261   # We used to save the Mutex with the RssBlob, which was idiotic. And\r
262   # since Mutexes dumped in one version might not be resotrable in another,\r
263   # we need a few tricks to be able to restore data from other versions of Ruby\r
264   #\r
265   # When migrating 1.8.6 => 1.8.5, all we need to do is define an empty\r
266   # #marshal_load() method for Mutex. For 1.8.5 => 1.8.6 we need something\r
267   # dirtier, as seen later on in the initialization code.\r
268   unless Mutex.new.respond_to?(:marshal_load)\r
269     class ::Mutex\r
270       def marshal_load(str)\r
271         return\r
272       end\r
273     end\r
274   end\r
275 \r
276   attr_reader :feeds\r
277 \r
278   def initialize\r
279     super\r
280     if @registry.has_key?(:feeds)\r
281       # When migrating from Ruby 1.8.5 to 1.8.6, dumped Mutexes may render the\r
282       # data unrestorable. If this happens, we patch the data, thus allowing\r
283       # the restore to work.\r
284       #\r
285       # This is actually pretty safe for a number of reasons:\r
286       # * the code is only called if standard marshalling fails\r
287       # * the string we look for is quite unlikely to appear randomly\r
288       # * if the string appears somewhere and the patched string isn't recoverable\r
289       #   either, we'll get another (unrecoverable) error, which makes the rss\r
290       #   plugin unsable, just like it was if no recovery was attempted\r
291       # * if the string appears somewhere and the patched string is recoverable,\r
292       #   we may get a b0rked feed, which is eventually overwritten by a clean\r
293       #   one, so the worst thing that can happen is that a feed update spams\r
294       #   the watchers once\r
295       @registry.recovery = Proc.new { |val|\r
296         patched = val.sub(":\v@mutexo:\nMutex", ":\v@mutexo:\vObject")\r
297         ret = Marshal.restore(patched)\r
298         ret.each_value { |blob|\r
299           blob.mutex = nil\r
300           blob\r
301         }\r
302       }\r
303 \r
304       @feeds = @registry[:feeds]\r
305       raise unless @feeds\r
306 \r
307       @registry.recovery = nil\r
308 \r
309       @feeds.keys.grep(/[A-Z]/) { |k|\r
310         @feeds[k.downcase] = @feeds[k]\r
311         @feeds.delete(k)\r
312       }\r
313       @feeds.each { |k, f|\r
314         f.mutex = Mutex.new\r
315         f.sanitize_watchers\r
316         parseRss(f) if f.xml\r
317       }\r
318     else\r
319       @feeds = Hash.new\r
320     end\r
321     @watch = Hash.new\r
322     rewatch_rss\r
323   end\r
324 \r
325   def name\r
326     "rss"\r
327   end\r
328 \r
329   def watchlist\r
330     @feeds.select { |h, f| f.watched? }\r
331   end\r
332 \r
333   def cleanup\r
334     stop_watches\r
335     super\r
336   end\r
337 \r
338   def save\r
339     unparsed = Hash.new()\r
340     @feeds.each { |k, f|\r
341       unparsed[k] = f.dup\r
342       # we don't want to save the mutex\r
343       unparsed[k].mutex = nil\r
344     }\r
345     @registry[:feeds] = unparsed\r
346   end\r
347 \r
348   def stop_watch(handle)\r
349     if @watch.has_key?(handle)\r
350       begin\r
351         debug "Stopping watch #{handle}"\r
352         @bot.timer.remove(@watch[handle])\r
353         @watch.delete(handle)\r
354       rescue Exception => e\r
355         report_problem("Failed to stop watch for #{handle}", e, nil)\r
356       end\r
357     end\r
358   end\r
359 \r
360   def stop_watches\r
361     @watch.each_key { |k|\r
362       stop_watch(k)\r
363     }\r
364   end\r
365 \r
366   def help(plugin,topic="")\r
367     case topic\r
368     when "show"\r
369       "rss show #{Bold}handle#{Bold} [#{Bold}limit#{Bold}] : show #{Bold}limit#{Bold} (default: 5, max: 15) entries from rss #{Bold}handle#{Bold}; #{Bold}limit#{Bold} can also be in the form a..b, to display a specific range of items"\r
370     when "list"\r
371       "rss list [#{Bold}handle#{Bold}] : list all rss feeds (matching #{Bold}handle#{Bold})"\r
372     when "watched"\r
373       "rss watched [#{Bold}handle#{Bold}] [in #{Bold}chan#{Bold}]: list all watched rss feeds (matching #{Bold}handle#{Bold}) (in channel #{Bold}chan#{Bold})"\r
374     when "who", "watches", "who watches"\r
375       "rss who watches [#{Bold}handle#{Bold}]]: list all watchers for rss feeds (matching #{Bold}handle#{Bold})"\r
376     when "add"\r
377       "rss add #{Bold}handle#{Bold} #{Bold}url#{Bold} [#{Bold}type#{Bold}] : add a new rss called #{Bold}handle#{Bold} from url #{Bold}url#{Bold} (of type #{Bold}type#{Bold})"\r
378     when "change"\r
379       "rss change #{Bold}what#{Bold} of #{Bold}handle#{Bold} to #{Bold}new#{Bold} : change the #{Underline}handle#{Underline}, #{Underline}url#{Underline}, #{Underline}type#{Underline} or #{Underline}refresh#{Underline} rate of rss called #{Bold}handle#{Bold} to value #{Bold}new#{Bold}"\r
380     when /^(del(ete)?|rm)$/\r
381       "rss del(ete)|rm #{Bold}handle#{Bold} : delete rss feed #{Bold}handle#{Bold}"\r
382     when "replace"\r
383       "rss replace #{Bold}handle#{Bold} #{Bold}url#{Bold} [#{Bold}type#{Bold}] : try to replace the url of rss called #{Bold}handle#{Bold} with #{Bold}url#{Bold} (of type #{Bold}type#{Bold}); only works if nobody else is watching it"\r
384     when "forcereplace"\r
385       "rss forcereplace #{Bold}handle#{Bold} #{Bold}url#{Bold} [#{Bold}type#{Bold}] : replace the url of rss called #{Bold}handle#{Bold} with #{Bold}url#{Bold} (of type #{Bold}type#{Bold})"\r
386     when "watch"\r
387       "rss watch #{Bold}handle#{Bold} [#{Bold}url#{Bold} [#{Bold}type#{Bold}]]  [in #{Bold}chan#{Bold}]: watch rss #{Bold}handle#{Bold} for changes (in channel #{Bold}chan#{Bold}); when the other parameters are present, the feed will be created if it doesn't exist yet"\r
388     when /(un|rm)watch/\r
389       "rss unwatch|rmwatch #{Bold}handle#{Bold} [in #{Bold}chan#{Bold}]: stop watching rss #{Bold}handle#{Bold} (in channel #{Bold}chan#{Bold}) for changes"\r
390     when "rewatch"\r
391       "rss rewatch : restart threads that watch for changes in watched rss"\r
392     else\r
393       "manage RSS feeds: rss show|list|watched|add|change|del(ete)|rm|(force)replace|watch|unwatch|rmwatch|rewatch"\r
394     end\r
395   end\r
396 \r
397   def report_problem(report, e=nil, m=nil)\r
398     if m && m.respond_to?(:reply)\r
399       m.reply report\r
400     else\r
401       warning report\r
402     end\r
403     if e\r
404       debug e.inspect\r
405       debug e.backtrace.join("\n") if e.respond_to?(:backtrace)\r
406     end\r
407   end\r
408 \r
409   def show_rss(m, params)\r
410     handle = params[:handle]\r
411     lims = params[:limit].to_s.match(/(\d+)(?:..(\d+))?/)\r
412     debug lims.to_a.inspect\r
413     if lims[2]\r
414       ll = [[lims[1].to_i-1,lims[2].to_i-1].min,  0].max\r
415       ul = [[lims[1].to_i-1,lims[2].to_i-1].max, 14].min\r
416       rev = lims[1].to_i > lims[2].to_i\r
417     else\r
418       ll = 0\r
419       ul = [[lims[1].to_i-1, 0].max, 14].min\r
420       rev = false\r
421     end\r
422 \r
423     feed = @feeds.fetch(handle.downcase, nil)\r
424     unless feed\r
425       m.reply "I don't know any feeds named #{handle}"\r
426       return\r
427     end\r
428 \r
429     m.reply "lemme fetch it..."\r
430     title = items = nil\r
431     we_were_watching = false\r
432 \r
433     if @watch.key?(feed.handle)\r
434       # If a feed is being watched, we run the watcher thread\r
435       # so that all watchers can be informed of changes to\r
436       # the feed. Before we do that, though, we remove the\r
437       # show requester from the watchlist, if present, lest\r
438       # he gets the update twice.\r
439       if feed.watched_by?(m.replyto)\r
440         we_were_watching = true\r
441         feed.rm_watch(m.replyto)\r
442       end\r
443       @bot.timer.reschedule(@watch[feed.handle], 0)\r
444       if we_were_watching\r
445         feed.add_watch(m.replyto)\r
446       end\r
447     else\r
448       fetched = fetchRss(feed, m, false)\r
449     end\r
450     return unless fetched or feed.xml\r
451     if not fetched and feed.items\r
452       m.reply "using old data"\r
453     else\r
454       parsed = parseRss(feed, m)\r
455       m.reply "using old data" unless parsed\r
456     end\r
457     return unless feed.items\r
458     title = feed.title\r
459     items = feed.items\r
460 \r
461     # We sort the feeds in freshness order (newer ones first)\r
462     items = freshness_sort(items)\r
463     disp = items[ll..ul]\r
464     disp.reverse! if rev\r
465 \r
466     m.reply "Channel : #{title}"\r
467     disp.each do |item|\r
468       printFormattedRss(feed, item, {:places=>[m.replyto],:handle=>nil,:date=>true})\r
469     end\r
470   end\r
471 \r
472   def itemDate(item,ex=nil)\r
473     return item.pubDate if item.respond_to?(:pubDate) and item.pubDate\r
474     return item.date if item.respond_to?(:date) and item.date\r
475     return ex\r
476   end\r
477 \r
478   def freshness_sort(items)\r
479     notime = Time.at(0)\r
480     items.sort { |a, b|\r
481       itemDate(b, notime) <=> itemDate(a, notime)\r
482     }\r
483   end\r
484 \r
485   def list_rss(m, params)\r
486     wanted = params[:handle]\r
487     reply = String.new\r
488     @feeds.each { |handle, feed|\r
489       next if wanted and !handle.match(/#{wanted}/i)\r
490       reply << "#{feed.handle}: #{feed.url} (in format: #{feed.type ? feed.type : 'default'})"\r
491       (reply << " refreshing every #{Utils.secs_to_string(feed.refresh_rate)}") if feed.refresh_rate\r
492       (reply << " (watched)") if feed.watched_by?(m.replyto)\r
493       reply << "\n"\r
494     }\r
495     if reply.empty?\r
496       reply = "no feeds found"\r
497       reply << " matching #{wanted}" if wanted\r
498     end\r
499     m.reply reply, :max_lines => reply.length\r
500   end\r
501 \r
502   def watched_rss(m, params)\r
503     wanted = params[:handle]\r
504     chan = params[:chan] || m.replyto\r
505     reply = String.new\r
506     watchlist.each { |handle, feed|\r
507       next if wanted and !handle.match(/#{wanted}/i)\r
508       next unless feed.watched_by?(chan)\r
509       reply << "#{feed.handle}: #{feed.url} (in format: #{feed.type ? feed.type : 'default'})"\r
510       (reply << " refreshing every #{Utils.secs_to_string(feed.refresh_rate)}") if feed.refresh_rate\r
511       reply << "\n"\r
512     }\r
513     if reply.empty?\r
514       reply = "no watched feeds"\r
515       reply << " matching #{wanted}" if wanted\r
516     end\r
517     m.reply reply\r
518   end\r
519 \r
520   def who_watches(m, params)\r
521     wanted = params[:handle]\r
522     reply = String.new\r
523     watchlist.each { |handle, feed|\r
524       next if wanted and !handle.match(/#{wanted}/i)\r
525       reply << "#{feed.handle}: #{feed.url} (in format: #{feed.type ? feed.type : 'default'})"\r
526       (reply << " refreshing every #{Utils.secs_to_string(feed.refresh_rate)}") if feed.refresh_rate\r
527       reply << ": watched by #{feed.watchers.join(', ')}"\r
528       reply << "\n"\r
529     }\r
530     if reply.empty?\r
531       reply = "no watched feeds"\r
532       reply << " matching #{wanted}" if wanted\r
533     end\r
534     m.reply reply\r
535   end\r
536 \r
537   def add_rss(m, params, force=false)\r
538     handle = params[:handle]\r
539     url = params[:url]\r
540     unless url.match(/https?/)\r
541       m.reply "I only deal with feeds from HTTP sources, so I can't use #{url} (maybe you forgot the handle?)"\r
542       return\r
543     end\r
544     type = params[:type]\r
545     if @feeds.fetch(handle.downcase, nil) && !force\r
546       m.reply "There is already a feed named #{handle} (URL: #{@feeds[handle.downcase].url})"\r
547       return\r
548     end\r
549     unless url\r
550       m.reply "You must specify both a handle and an url to add an RSS feed"\r
551       return\r
552     end\r
553     @feeds[handle.downcase] = RssBlob.new(url,handle,type)\r
554     reply = "Added RSS #{url} named #{handle}"\r
555     if type\r
556       reply << " (format: #{type})"\r
557     end\r
558     m.reply reply\r
559     return handle\r
560   end\r
561 \r
562   def change_rss(m, params)\r
563     handle = params[:handle].downcase\r
564     feed = @feeds.fetch(handle, nil)\r
565     unless feed\r
566       m.reply "No such feed with handle #{handle}"\r
567       return\r
568     end\r
569     case params[:what].intern\r
570     when :handle\r
571       new = params[:new].downcase\r
572       if @feeds.key?(new) and @feeds[new]\r
573         m.reply "There already is a feed with handle #{new}"\r
574         return\r
575       else\r
576         feed.mutex.synchronize do\r
577           @feeds[new] = feed\r
578           @feeds.delete(handle)\r
579           feed.handle = new\r
580         end\r
581         handle = new\r
582       end\r
583     when :url\r
584       new = params[:new]\r
585       feed.mutex.synchronize do\r
586         feed.url = new\r
587       end\r
588     when :format, :type\r
589       new = params[:new]\r
590       new = nil if new == 'default'\r
591       feed.mutex.synchronize do\r
592         feed.type = new\r
593       end\r
594     when :refresh\r
595       new = params[:new].to_i\r
596       new = nil if new == 0\r
597       feed.mutex.synchronize do\r
598         feed.refresh_rate = new\r
599       end\r
600     else\r
601       m.reply "Don't know how to change #{params[:what]} for feeds"\r
602       return\r
603     end\r
604     m.reply "Feed changed:"\r
605     list_rss(m, {:handle => handle})\r
606   end\r
607 \r
608   def del_rss(m, params, pass=false)\r
609     feed = unwatch_rss(m, params, true)\r
610     return unless feed\r
611     if feed.watched?\r
612       m.reply "someone else is watching #{feed.handle}, I won't remove it from my list"\r
613       return\r
614     end\r
615     @feeds.delete(feed.handle.downcase)\r
616     m.okay unless pass\r
617     return\r
618   end\r
619 \r
620   def replace_rss(m, params)\r
621     handle = params[:handle]\r
622     if @feeds.key?(handle.downcase)\r
623       del_rss(m, {:handle => handle}, true)\r
624     end\r
625     if @feeds.key?(handle.downcase)\r
626       m.reply "can't replace #{feed.handle}"\r
627     else\r
628       add_rss(m, params, true)\r
629     end\r
630   end\r
631 \r
632   def forcereplace_rss(m, params)\r
633     add_rss(m, params, true)\r
634   end\r
635 \r
636   def watch_rss(m, params)\r
637     handle = params[:handle]\r
638     chan = params[:chan] || m.replyto\r
639     url = params[:url]\r
640     type = params[:type]\r
641     if url\r
642       add_rss(m, params)\r
643     end\r
644     feed = @feeds.fetch(handle.downcase, nil)\r
645     if feed\r
646       if feed.add_watch(chan)\r
647         watchRss(feed, m)\r
648         m.okay\r
649       else\r
650         m.reply "Already watching #{feed.handle} in #{chan}"\r
651       end\r
652     else\r
653       m.reply "Couldn't watch feed #{handle} (no such feed found)"\r
654     end\r
655   end\r
656 \r
657   def unwatch_rss(m, params, pass=false)\r
658     handle = params[:handle].downcase\r
659     chan = params[:chan] || m.replyto\r
660     unless @feeds.has_key?(handle)\r
661       m.reply("dunno that feed")\r
662       return\r
663     end\r
664     feed = @feeds[handle]\r
665     if feed.rm_watch(chan)\r
666       m.reply "#{chan} has been removed from the watchlist for #{feed.handle}"\r
667     else\r
668       m.reply("#{chan} wasn't watching #{feed.handle}") unless pass\r
669     end\r
670     if !feed.watched?\r
671       stop_watch(handle)\r
672     end\r
673     return feed\r
674   end\r
675 \r
676   def rewatch_rss(m=nil, params=nil)\r
677     stop_watches\r
678 \r
679     # Read watches from list.\r
680     watchlist.each{ |handle, feed|\r
681       watchRss(feed, m)\r
682     }\r
683     m.okay if m\r
684   end\r
685 \r
686   private\r
687   def watchRss(feed, m=nil)\r
688     if @watch.has_key?(feed.handle)\r
689       report_problem("watcher thread for #{feed.handle} is already running", nil, m)\r
690       return\r
691     end\r
692     status = Hash.new\r
693     status[:failures] = 0\r
694     status[:first_run] = true\r
695     @watch[feed.handle] = @bot.timer.add(0) {\r
696       debug "watcher for #{feed} started"\r
697       failures = status[:failures]\r
698       first_run = status.delete(:first_run)\r
699       begin\r
700         debug "fetching #{feed}"\r
701         oldxml = feed.xml ? feed.xml.dup : nil\r
702         unless fetchRss(feed)\r
703           failures += 1\r
704         else\r
705           if first_run\r
706             debug "first run for #{feed}, getting items"\r
707             parseRss(feed)\r
708           elsif oldxml and oldxml == feed.xml\r
709             debug "xml for #{feed} didn't change"\r
710             failures -= 1 if failures > 0\r
711           else\r
712             if not feed.items\r
713               debug "no previous items in feed #{feed}"\r
714               parseRss(feed)\r
715               failures -= 1 if failures > 0\r
716             else\r
717               # This one is used for debugging\r
718               otxt = []\r
719 \r
720               # These are used for checking new items vs old ones\r
721               uid_opts = { :show_updated => @bot.config['rss.show_updated'] }\r
722               oids = Set.new feed.items.map { |item|\r
723                 uid = RSS.item_uid_for_bot(item, uid_opts)\r
724                 otxt << item.to_s\r
725                 debug [uid, item].inspect\r
726                 debug [uid, otxt.last].inspect\r
727                 uid\r
728               }\r
729 \r
730               unless parseRss(feed)\r
731                 debug "no items in feed #{feed}"\r
732                 failures += 1\r
733               else\r
734                 debug "Checking if new items are available for #{feed}"\r
735                 failures -= 1 if failures > 0\r
736                 # debug "Old:"\r
737                 # debug oldxml\r
738                 # debug "New:"\r
739                 # debug feed.xml\r
740 \r
741                 dispItems = feed.items.reject { |item|\r
742                   uid = RSS.item_uid_for_bot(item, uid_opts)\r
743                   txt = item.to_s\r
744                   if oids.include?(uid)\r
745                     debug "rejecting old #{uid} #{item.inspect}"\r
746                     debug [uid, txt].inspect\r
747                     true\r
748                   else\r
749                     debug "accepting new #{uid} #{item.inspect}"\r
750                     debug [uid, txt].inspect\r
751                     warning "same text! #{txt}" if otxt.include?(txt)\r
752                     false\r
753                   end\r
754                 }\r
755 \r
756                 if dispItems.length > 0\r
757                   debug "Found #{dispItems.length} new items in #{feed}"\r
758                   # When displaying watched feeds, publish them from older to newer\r
759                   dispItems.reverse.each { |item|\r
760                     printFormattedRss(feed, item)\r
761                   }\r
762                 else\r
763                   debug "No new items found in #{feed}"\r
764                 end\r
765               end\r
766             end\r
767           end\r
768         end\r
769       rescue Exception => e\r
770         error "Error watching #{feed}: #{e.inspect}"\r
771         debug e.backtrace.join("\n")\r
772         failures += 1\r
773       end\r
774 \r
775       status[:failures] = failures\r
776 \r
777       timer = nil\r
778       seconds = @bot.config['rss.thread_sleep']\r
779       feed.mutex.synchronize do\r
780         timer = @watch[feed.handle]\r
781         seconds = feed.refresh_rate if feed.refresh_rate\r
782       end\r
783       seconds *= failures + 1\r
784       seconds += seconds * (rand(100)-50)/100\r
785       debug "watcher for #{feed} going to sleep #{seconds} seconds.."\r
786       @bot.timer.reschedule(timer, seconds) rescue warning "watcher for #{feed} failed to reschedule: #{$!.inspect}"\r
787     }\r
788     debug "watcher for #{feed} added"\r
789   end\r
790 \r
791   def printFormattedRss(feed, item, opts=nil)\r
792     places = feed.watchers\r
793     handle = "::#{feed.handle}:: "\r
794     date = String.new\r
795     if opts\r
796       places = opts[:places] if opts.key?(:places)\r
797       handle = opts[:handle].to_s if opts.key?(:handle)\r
798       if opts.key?(:date) && opts[:date]\r
799         if item.respond_to?(:pubDate) \r
800           if item.pubDate.class <= Time\r
801             date = item.pubDate.strftime("%Y/%m/%d %H:%M")\r
802           else\r
803             date = item.pubDate.to_s\r
804           end\r
805         elsif  item.respond_to?(:date)\r
806           if item.date.class <= Time\r
807             date = item.date.strftime("%Y/%m/%d %H:%M")\r
808           else\r
809             date = item.date.to_s\r
810           end\r
811         else\r
812           date = "(no date)"\r
813         end\r
814         date += " :: "\r
815       end\r
816     end\r
817 \r
818     title = "#{Bold}#{item.title.ircify_html :limit => @bot.config['rss.head_max']}#{Bold}" if item.title\r
819 \r
820     desc = item.description.ircify_html(:limit => @bot.config['rss.text_max'], :a_href => :link_out) if item.description\r
821 \r
822     link = item.link.chomp if item.link\r
823 \r
824     debug item.inspect\r
825     category = item.dc_subject rescue item.category.content rescue nil\r
826     category = nil if category and category.empty?\r
827     author = item.dc_creator rescue item.author rescue nil\r
828     author = nil if author and author.empty?\r
829 \r
830     line1 = nil\r
831     line2 = nil\r
832 \r
833     at = ((item.title && item.link) ? ' @ ' : '')\r
834     case feed.type\r
835     when 'blog'\r
836       author << " " if author\r
837       abt = category ? "about #{category} " : ""\r
838       line1 = "#{handle}#{date}#{author}blogged #{abt}at #{link}"\r
839       line2 = "#{handle}#{title} - #{desc}"\r
840     when 'forum'\r
841       line1 = "#{handle}#{date}#{title}#{at}#{link}"\r
842     when 'wiki'\r
843       line1 = "#{handle}#{date}#{title}#{at}#{link} has been edited by #{author}. #{desc}"\r
844     when 'gmane'\r
845       line1 = "#{handle}#{date}Message #{title} sent by #{author}. #{desc}"\r
846     when 'trac'\r
847       line1 = "#{handle}#{date}#{title} @ #{link}"\r
848       unless item.title =~ /^Changeset \[(\d+)\]/\r
849         line2 = "#{handle}#{date}#{desc}"\r
850       end\r
851     when '/.'\r
852       dept = "(from the #{item.slash_department} dept) " rescue nil\r
853       sec = " in section #{item.slash_section}" rescue nil\r
854 \r
855       line1 = "#{handle}#{date}#{dept}#{title}#{at}#{link} (posted by #{author}#{sec})"\r
856     else\r
857       line1 = "#{handle}#{date}#{title}#{at}#{link}"\r
858     end\r
859     places.each { |loc|\r
860       @bot.say loc, line1, :overlong => :truncate\r
861       next unless line2\r
862       @bot.say loc, line2, :overlong => :truncate\r
863     }\r
864   end\r
865 \r
866   def fetchRss(feed, m=nil, cache=true)\r
867     begin\r
868       # Use 60 sec timeout, cause the default is too low\r
869       xml = @bot.httputil.get(feed.url,\r
870                               :read_timeout => 60,\r
871                               :open_timeout => 60,\r
872                               :cache => cache)\r
873     rescue URI::InvalidURIError, URI::BadURIError => e\r
874       report_problem("invalid rss feed #{feed.url}", e, m)\r
875       return nil\r
876     rescue => e\r
877       report_problem("error getting #{feed.url}", e, m)\r
878       return nil\r
879     end\r
880     debug "fetched #{feed}"\r
881     unless xml\r
882       report_problem("reading feed #{feed} failed", nil, m)\r
883       return nil\r
884     end\r
885     # Ok, 0.9 feeds are not supported, maybe because\r
886     # Netscape happily removed the DTD. So what we do is just to\r
887     # reassign the 0.9 RDFs to 1.0, and hope it goes right.\r
888     xml.gsub!("xmlns=\"http://my.netscape.com/rdf/simple/0.9/\"",\r
889               "xmlns=\"http://purl.org/rss/1.0/\"")\r
890     feed.mutex.synchronize do\r
891       feed.xml = xml\r
892     end\r
893     return true\r
894   end\r
895 \r
896   def parseRss(feed, m=nil)\r
897     return nil unless feed.xml\r
898     feed.mutex.synchronize do\r
899       xml = feed.xml\r
900       begin\r
901         ## do validate parse\r
902         rss = RSS::Parser.parse(xml)\r
903         debug "parsed and validated #{feed}"\r
904       rescue RSS::InvalidRSSError\r
905         ## do non validate parse for invalid RSS 1.0\r
906         begin\r
907           rss = RSS::Parser.parse(xml, false)\r
908           debug "parsed but not validated #{feed}"\r
909         rescue RSS::Error => e\r
910           report_problem("parsing rss stream failed, whoops =(", e, m)\r
911           return nil\r
912         end\r
913       rescue RSS::Error => e\r
914         report_problem("parsing rss stream failed, oioi", e, m)\r
915         return nil\r
916       rescue => e\r
917         report_problem("processing error occured, sorry =(", e, m)\r
918         return nil\r
919       end\r
920       items = []\r
921       if rss.nil?\r
922         report_problem("#{feed} does not include RSS 1.0 or 0.9x/2.0", nil, m)\r
923       else\r
924         begin\r
925           rss.output_encoding = 'UTF-8'\r
926         rescue RSS::UnknownConvertMethod => e\r
927           report_problem("bah! something went wrong =(", e, m)\r
928           return nil\r
929         end\r
930         rss.channel.title ||= "Unknown"\r
931         title = rss.channel.title\r
932         rss.items.each do |item|\r
933           item.title ||= "Unknown"\r
934           items << item\r
935         end\r
936       end\r
937 \r
938       if items.empty?\r
939         report_problem("no items found in the feed, maybe try weed?", e, m)\r
940         return nil\r
941       end\r
942       feed.title = title\r
943       feed.items = items\r
944       return true\r
945     end\r
946   end\r
947 end\r
948 \r
949 plugin = RSSFeedsPlugin.new\r
950 \r
951 plugin.map 'rss show :handle :limit',\r
952   :action => 'show_rss',\r
953   :requirements => {:limit => /^\d+(?:\.\.\d+)?$/},\r
954   :defaults => {:limit => 5}\r
955 plugin.map 'rss list :handle',\r
956   :action => 'list_rss',\r
957   :defaults => {:handle => nil}\r
958 plugin.map 'rss watched :handle [in :chan]',\r
959   :action => 'watched_rss',\r
960   :defaults => {:handle => nil}\r
961 plugin.map 'rss who watches :handle',\r
962   :action => 'who_watches',\r
963   :defaults => {:handle => nil}\r
964 plugin.map 'rss add :handle :url :type',\r
965   :action => 'add_rss',\r
966   :defaults => {:type => nil}\r
967 plugin.map 'rss change :what of :handle to :new',\r
968   :action => 'change_rss',\r
969   :requirements => { :what => /handle|url|format|type|refresh/ }\r
970 plugin.map 'rss change :what for :handle to :new',\r
971   :action => 'change_rss',\r
972   :requirements => { :what => /handle|url|format|type|refesh/ }\r
973 plugin.map 'rss del :handle',\r
974   :action => 'del_rss'\r
975 plugin.map 'rss delete :handle',\r
976   :action => 'del_rss'\r
977 plugin.map 'rss rm :handle',\r
978   :action => 'del_rss'\r
979 plugin.map 'rss replace :handle :url :type',\r
980   :action => 'replace_rss',\r
981   :defaults => {:type => nil}\r
982 plugin.map 'rss forcereplace :handle :url :type',\r
983   :action => 'forcereplace_rss',\r
984   :defaults => {:type => nil}\r
985 plugin.map 'rss watch :handle [in :chan]',\r
986   :action => 'watch_rss',\r
987   :defaults => {:url => nil, :type => nil}\r
988 plugin.map 'rss watch :handle :url :type [in :chan]',\r
989   :action => 'watch_rss',\r
990   :defaults => {:url => nil, :type => nil}\r
991 plugin.map 'rss unwatch :handle [in :chan]',\r
992   :action => 'unwatch_rss'\r
993 plugin.map 'rss rmwatch :handle [in :chan]',\r
994   :action => 'unwatch_rss'\r
995 plugin.map 'rss rewatch',\r
996   :action => 'rewatch_rss'\r
997 \r