search plugin: fix gcalc regexp

[user/henk/code/ruby/rbot.git] / data / rbot / plugins / grouphug.rb

diff --git a/data/rbot/plugins/grouphug.rb b/data/rbot/plugins/grouphug.rb
index 73c21eee54aba6fd88de5e32aad3cb08e06e72e5..f08f57533829dfd3943615a2e881691dbf11870f 100644 (file)
--- a/data/rbot/plugins/grouphug.rb
+++ b/data/rbot/plugins/grouphug.rb
@@ -1,33 +1,61 @@
-# Plugin for the Ruby IRC bot (http://linuxbrit.co.uk/rbot/)
-# (c) 2005 Mark Kretschmann <markey@web.de>
-# Licensed under GPL V2.
-
-require "net/http"
-
+#-- vim:sw=2:et
+#++
+#
+# :title: Grouphug Plugin for rbot
+#
+# Author:: Mark Kretschmann <markey@web.de>
+# Author:: Casey Link <unnamedrambler@gmail.com>
+# Copyright:: (C) 2005 Mark Kretschmann
+# Copyright:: (C) 2008 Casey Link
+# License:: GPL v2
 
 class GrouphugPlugin < Plugin
-    def help( plugin, topic="" )
-        "Grouphug plugin. Confess! Usage: 'confess' for random confession, 'confess <number>' for specific one."
+  REG  = Regexp.new('<div class="content">\s*<p>(.*?)</p>\s+</div>', Regexp::MULTILINE)
+  def initialize
+    super
+    @confessions = Array.new
+  end
+
+  def help( plugin, topic="" )
+    return "Grouphug plugin. Confess! Usage: 'confess' for random confession, 'confess <number>' for specific one."
+  end
+
+  def confess(m, params)
+    opts = { :cache => false }
+    path = "random"
+    begin
+      # Fetch a specific question - separate from cache
+      if params[:num]
+        path = "confessions/#{params[:num]}"
+        opts.delete(:cache)
+        data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
+
+        res = data.scan(REG)
+        confession = res[0][0].ircify_html
+        confession = "no confession ##{params[:num]} found" if confession.empty? and params[:num]
+        m.reply confession
+      else # Cache random confessions
+        if @confessions.empty?
+          data = @bot.httputil.get("http://grouphug.us/#{path}", opts)
+          res = data.scan(REG)
+          res.each do |quote|
+            @confessions << quote[0].ircify_html
+          end
+        end
+        confession = @confessions.pop
+        m.reply confession
+      end
+    rescue
+      m.reply "failed to connect to grouphug.us"
     end
-
-    def privmsg( m )
-        h = Net::HTTP.new( "grouphug.us", 80 )
-        path = "/random"
-        path = "/confessions/#{m.params()}" if m.params()
-        data = bot.httputil.get(URI.parse("http://grouphug.us/#{path}"))
-
-        reg = Regexp.new( '(<td class="conf-text")(.*?)(<p>)(.*?)(</p>)', Regexp::MULTILINE )
-        confession = reg.match( data )[4]
-        confession.gsub!( /<.*?>/, "" ) # Remove html tags
-        confession.gsub!( "\t", "" ) # Remove tab characters
-
-        @bot.say(m.replyto, confession)
-   end
+  end
 end
 
 
 plugin = GrouphugPlugin.new
 
-plugin.register("grouphug")
-plugin.register("confess")
+plugin.map "grouphug [:num]",
+  :thread => true, :action => :confess, :requirements => { :num => /\d+/ }
+plugin.map "confess [:num]",
+  :thread => true, :action => :confess, :requirements => { :num => /\d+/ }