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