]> git.netwichtig.de Git - user/henk/code/ruby/rbot.git/commitdiff
con:
authorGiuseppe Bilotta <giuseppe.bilotta@gmail.com>
Mon, 24 Jul 2006 20:40:56 +0000 (20:40 +0000)
committerGiuseppe Bilotta <giuseppe.bilotta@gmail.com>
Mon, 24 Jul 2006 20:40:56 +0000 (20:40 +0000)
data/rbot/plugins/rss.rb

index af7901275b2e5a211552a63bd88c0efbc2ef881c..0e4fd6b934ae3d3f8f7cd58eb97a0afda163892c 100644 (file)
@@ -92,6 +92,10 @@ class RSSFeedsPlugin < Plugin
     :default => 90, :validate => Proc.new{|v| v > 0 && v < 400},\r
     :desc => "How many characters to use of a RSS item text")\r
 \r
+  BotConfig.register BotConfigIntegerValue.new('rss.thread_sleep',\r
+    :default => 120, :validate => Proc.new{|v| v > 30},\r
+    :desc => "How many characters to use of a RSS item text")\r
+\r
   @@watchThreads = Hash.new\r
   @@mutex = Mutex.new\r
 \r
@@ -156,12 +160,16 @@ class RSSFeedsPlugin < Plugin
     end\r
   end\r
 \r
-  def report_problem(report, m=nil)\r
-    if m\r
+  def report_problem(report, e=nil, m=nil)\r
+    if m && m.respond_to?(reply)\r
       m.reply report\r
     else\r
       warning report\r
     end\r
+    if e\r
+      debug e.inspect\r
+      debug e.backtrace.join("\n") if e.class >= Exception\r
+    end\r
   end\r
 \r
   def show_rss(m, params)\r
@@ -183,7 +191,7 @@ class RSSFeedsPlugin < Plugin
     m.reply("Channel : #{title}")\r
     # TODO: optional by-date sorting if dates present\r
     items[0...limit].reverse.each do |item|\r
-      printRssItem(m.replyto,item)\r
+      printFormattedRss(feed, item, {:places=>[m.replyto],:handle=>nil,:date=>true})\r
     end\r
   end\r
 \r
@@ -225,6 +233,10 @@ class RSSFeedsPlugin < Plugin
   def add_rss(m, params, force=false)\r
     handle = params[:handle]\r
     url = params[:url]\r
+    unless url.match(/https?/)\r
+      m.reply "I only deal with feeds from HTTP sources, so I can't use #{url} (maybe you forgot the handle?)"\r
+      return\r
+    end\r
     type = params[:type]\r
     if @feeds.fetch(handle, nil) && !force\r
       m.reply "There is already a feed named #{handle} (URL: #{@feeds[handle].url})"\r
@@ -336,16 +348,16 @@ class RSSFeedsPlugin < Plugin
   private\r
   def watchRss(feed, m=nil)\r
     if @@watchThreads.has_key?(feed.handle)\r
-      report_problem("watcher thread for #{feed.handle} is already running", m)\r
+      report_problem("watcher thread for #{feed.handle} is already running", nil, m)\r
       return\r
     end\r
     @@watchThreads[feed.handle] = Thread.new do\r
-      debug 'watchRss thread started.'\r
+      debug "watcher for #{feed} started"\r
       oldItems = []\r
       firstRun = true\r
       loop do\r
         begin\r
-          debug 'Fetching rss feed...'\r
+          debug "fetching #{feed}"\r
           title = newItems = nil\r
           @@mutex.synchronize {\r
             title, newItems = fetchRss(feed)\r
@@ -354,7 +366,7 @@ class RSSFeedsPlugin < Plugin
             m.reply "no items in feed"\r
             break\r
           end\r
-          debug "Checking if new items are available"\r
+          debug "Checking if new items are available for #{feed}"\r
           if firstRun\r
             debug "First run, we'll see next time"\r
             firstRun = false\r
@@ -364,15 +376,14 @@ class RSSFeedsPlugin < Plugin
               otxt.include?(item.to_s)\r
             }\r
             if dispItems.length > 0\r
-              debug "Found #{dispItems.length} new items"\r
+              debug "Found #{dispItems.length} new items in #{feed}"\r
               dispItems.each { |item|\r
-                debug "showing #{item.title}"\r
                 @@mutex.synchronize {\r
                   printFormattedRss(feed, item)\r
                 }\r
               }\r
             else\r
-              debug "No new items found"\r
+              debug "No new items found in #{feed}"\r
             end\r
           end\r
           oldItems = newItems.dup\r
@@ -381,43 +392,44 @@ class RSSFeedsPlugin < Plugin
           debug e.backtrace.join("\n")\r
         end\r
 \r
-        seconds = 150 + rand(100)\r
-        debug "Thread going to sleep #{seconds} seconds.."\r
+        seconds = @bot.config['rss.thread_sleep']\r
+        seconds += seconds * rand(50)/100\r
+        debug "watcher for #{feed} going to sleep #{seconds} seconds.."\r
         sleep seconds\r
       end\r
     end\r
   end\r
 \r
-  def printRssItem(loc,item)\r
-    if item.kind_of?(RSS::RDF::Item)\r
-      @bot.say loc, item.title.chomp.riphtml.shorten(@bot.config['rss.head_max']) + " @ " + item.link\r
-    else\r
-      desc = String.new\r
-      desc << item.pubDate.to_s.chomp + ": " if item.pubDate\r
-      desc << item.title.chomp.riphtml.shorten(@bot.config['rss.head_max']) + " :: " if item.title\r
-      desc << " @ " + item.link.chomp if item.link\r
-      @bot.say loc, desc\r
+  def printFormattedRss(feed, item, opts=nil)\r
+    places = feed.watchers\r
+    handle = "::#{feed.handle}:: "\r
+    date = String.new\r
+    if opts\r
+      places = opts[:places] if opts.key?(:places)\r
+      handle = opts[:handle].to_s if opts.key?(:handle)\r
+      date = item.pubDate.strftime("%Y/%m/%d %H.%M.%S") + " :: " if (opts.key?(:date) && opts[:date] && item.pubDate)\r
     end\r
-  end\r
-\r
-  def printFormattedRss(feed, item)\r
-    debug "Printing formatted item #{item.inspect} for feed #{feed.to_s}"\r
-    feed.watchers.each { |loc|\r
+    title = "#{Bold}#{item.title.chomp.riphtml}#{Bold}" if item.title\r
+    desc = item.description.gsub(/\s+/,' ').strip.riphtml.shorten(@bot.config['rss.text_max']) if item.description\r
+    link = item.link.chomp if item.link\r
+    places.each { |loc|\r
       case feed.type\r
       when 'blog'\r
-        @bot.say loc, "::#{feed.handle}:: #{item.category.content} just blogged at #{item.link}::"\r
-        @bot.say loc, "::#{feed.handle}:: #{item.title.chomp.riphtml} - #{item.description.chomp.riphtml.shorten(@bot.config['rss.text_max'])}::"\r
+        @bot.say loc, "#{handle}#{date}#{item.category.content} blogged at #{link}"\r
+        @bot.say loc, "#{handle}#{title} - #{desc}"\r
       when 'forum'\r
-        @bot.say loc, "::#{feed.handle}:: #{item.pubDate.to_s.chomp+": " if item.pubDate}#{item.title.chomp.riphtml+" :: " if item.title}#{" @ "+item.link.chomp if item.link}"\r
+        @bot.say loc, "#{handle}#{date}#{title}#{' @ ' if item.title && item.link}#{link}"\r
       when 'wiki'\r
-        @bot.say loc, "::#{feed.handle}:: #{item.title} has been edited by #{item.dc_creator}. #{item.description.split("\n")[0].chomp.riphtml.shorten(@bot.config['rss.text_max'])} #{item.link} ::"\r
+        @bot.say loc, "#{handle}#{date}#{item.title} has been edited by #{item.dc_creator}. #{desc} #{link}"\r
       when 'gmame'\r
-        @bot.say loc, "::#{feed.handle}:: Message #{item.title} sent by #{item.dc_creator}. #{item.description.split("\n")[0].chomp.riphtml.shorten(@bot.config['rss.text_max'])} ::"\r
+        @bot.say loc, "#{handle}#{date}Message #{title} sent by #{item.dc_creator}. #{desc}"\r
       when 'trac'\r
-        @bot.say loc, "::#{feed.handle}:: #{item.title} :: #{item.link}"\r
-        @bot.say loc, "::#{feed.handle}:: #{item.description.gsub(/\s+/,' ').strip.riphtml.shorten(@bot.config['rss.text_max'])}"\r
+        @bot.say loc, "#{handle}#{date}#{title} @ #{link}"\r
+        unless item.title =~ /^Changeset \[(\d+)\]/\r
+          @bot.say loc, "#{handle}#{desc}"\r
+        end\r
       else\r
-        printRssItem(loc,item)\r
+        @bot.say loc, "#{handle}#{date}#{title}#{' @ ' if item.title && item.link}#{link}"\r
       end\r
     }\r
   end\r
@@ -427,44 +439,42 @@ class RSSFeedsPlugin < Plugin
       # Use 60 sec timeout, cause the default is too low\r
       xml = @bot.httputil.get_cached(feed.url,60,60)\r
     rescue URI::InvalidURIError, URI::BadURIError => e\r
-      report_problem("invalid rss feed #{feed.url}", m)\r
+      report_problem("invalid rss feed #{feed.url}", e, m)\r
       return\r
     end\r
-    debug 'fetched'\r
+    debug "fetched #{feed}"\r
     unless xml\r
-      report_problem("reading feed #{url} failed", m)\r
+      report_problem("reading feed #{feed} failed", nil, m)\r
       return\r
     end\r
 \r
     begin\r
       ## do validate parse\r
       rss = RSS::Parser.parse(xml)\r
-      debug 'parsed'\r
+      debug "parsed #{feed}"\r
     rescue RSS::InvalidRSSError\r
       ## do non validate parse for invalid RSS 1.0\r
       begin\r
         rss = RSS::Parser.parse(xml, false)\r
-      rescue RSS::Error\r
-        report_problem("parsing rss stream failed, whoops =(", m)\r
+      rescue RSS::Error => e\r
+        report_problem("parsing rss stream failed, whoops =(", e, m)\r
         return\r
       end\r
-    rescue RSS::Error\r
-      report_problem("parsing rss stream failed, oioi", m)\r
+    rescue RSS::Error => e\r
+      report_problem("parsing rss stream failed, oioi", e, m)\r
       return\r
     rescue => e\r
-      report_problem("processing error occured, sorry =(", m)\r
-      debug e.inspect\r
-      debug e.backtrace.join("\n")\r
+      report_problem("processing error occured, sorry =(", e, m)\r
       return\r
     end\r
     items = []\r
     if rss.nil?\r
-      report_problem("#{feed.url} does not include RSS 1.0 or 0.9x/2.0",m)\r
+      report_problem("#{feed} does not include RSS 1.0 or 0.9x/2.0", nil, m)\r
     else\r
       begin\r
-        rss.output_encoding = "euc-jp"\r
-      rescue RSS::UnknownConvertMethod\r
-        report_problem("bah! something went wrong =(",m)\r
+        rss.output_encoding = 'UTF-8'\r
+      rescue RSS::UnknownConvertMethod => e\r
+        report_problem("bah! something went wrong =(", e, m)\r
         return\r
       end\r
       rss.channel.title ||= "Unknown"\r
@@ -476,7 +486,7 @@ class RSSFeedsPlugin < Plugin
     end\r
 \r
     if items.empty?\r
-      report_problem("no items found in the feed, maybe try weed?",m)\r
+      report_problem("no items found in the feed, maybe try weed?", e, m)\r
       return\r
     end\r
     return [title, items]\r