googlefight plugin

1 files changed, 64 insertions, 0 deletions

diff --git a/data/rbot/plugins/googlefight.rb b/data/rbot/plugins/googlefight.rb
new file mode 100755
index 00000000..3f7cf5a1
--- /dev/null
+++ b/data/rbot/plugins/googlefight.rb
@@ -0,0 +1,64 @@
+#-- vim:sw=2:et
+#++
+#
+# :title: Googlefight plugin for rbot
+#
+# Author:: Raine Virta <rane@kapsi.fi
+# Copyright:: (C) 2009 Raine Virta
+# License:: GPL v2
+
+class GoogleFightPlugin < Plugin
+  def help(plugin, topic)
+    "googlefight <keyword 1> <keyword 2> [... <keyword n+1>] => battles given keywords based on amount of google search results and announces the winner!"
+  end
+  
+  def fight(m, params)
+    keywords = parse_keywords(params)
+    return if keywords.nil?
+    
+    keywords.map! do |k|
+      [k, google_count(k)]
+    end
+    
+    m.reply output(keywords)
+  end
+  
+  def output(result)
+    result = result.sort_by { |e| e[1] }.reverse
+    str = result.map do |kw|
+      "%{keyword} (%{count})" % {
+        :keyword => Bold+kw[0]+Bold,
+        :count   => kw[1].to_s.gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
+      }
+    end.join(" vs. ")
+    
+    unless result[0][1].zero?
+      str << _(" -- %{keyword} wins!") % {
+        :keyword => Bold+result[0][0]+Bold
+      }
+    else
+      str << _(" -- no winner here!")
+    end 
+  end
+  
+  def parse_keywords(params)
+    str = params[:keywords].join(" ")
+    
+    # foo "foo bar" bar
+    # no separators so assume they're all separate keywords
+    if str.match(/(?:"[\w\s]+"|\w+)(?: (?:"[\w\s]+"|\w+))+/)
+      str.scan(/"[^"]+"|\S+/).flatten
+    end
+  end
+  
+  def google_count(query)
+    url  = 'http://www.google.com/search?hl=en&safe=off&btnG=Search&q=' << CGI.escape(query)
+    html = Net::HTTP.get(URI.parse((url)))
+    res  = html.scan(%r{Results <b>1</b> - <b>\d+</b> of(?: about)? <b>([\d,]+)</b>})
+    res.to_s.tr(",", "").to_i
+  end
+end
+
+plugin = GoogleFightPlugin.new
+plugin.map "googlefight *keywords", :action => "fight",
+  :requirements => { :keywords => /^[\w\s"]+? (?:(?:(?:\||vs\.) )?[\w\s"]+?)+/ }
\ No newline at end of file