data/rbot/plugins/search.rb

   1 require 'uri'
   2
   3 Net::HTTP.version_1_2
   4
   5 GOOGLE_WAP_LINK = /<a accesskey="(\d)" href=".*?u=(.*?)">(.*?)<\/a>/im
   6
   7 class SearchPlugin < Plugin
   8   def help(plugin, topic="")
   9     case topic
  10     when "search"
  11     "search <string> => search google for <string>"
  12     when "google"
  13     "google <string> => search google for <string>"
  14     when "wp"
  15       "wp [<code>] <string> => search for <string> on Wikipedia. You can select a national <code> to only search the national Wikipedia"
  16     else
  17     "search <string> (or: google <string>) => search google for <string> | wp <string> => search for <string> on Wikipedia"
  18     end
  19   end
  20
  21   def google(m, params)
  22     what = params[:words].to_s
  23     searchfor = URI.escape what
  24     if params[:site]
  25       site = "site:#{params[:site]}+"
  26     else
  27       site = ""
  28     end
  29
  30     url = "http://www.google.com/wml/search?q=#{site}#{searchfor}"
  31
  32
  33     begin
  34       wml = @bot.httputil.get(url)
  35     rescue => e
  36       m.reply "error googling for #{what}"
  37       return
  38     end
  39     results = wml.scan(GOOGLE_WAP_LINK)
  40     if results.length == 0
  41       m.reply "no results found for #{what}"
  42       return
  43     end
  44     results = results[0...3].map { |res|
  45       "#{res[0]}. #{Bold}#{Utils.decode_html_entities res[2].strip}#{Bold}: #{URI.unescape res[1].strip}"
  46     }.join(" | ")
  47
  48     m.reply "Results for #{what}: #{results}"
  49   end
  50
  51   def wikipedia(m, params)
  52     lang = params[:lang]
  53     site = "#{lang.nil? ? '' : lang + '.'}wikipedia.org"
  54     params[:site] = site
  55     return google(m, params)
  56   end
  57 end
  58
  59 plugin = SearchPlugin.new
  60
  61 plugin.map "search *words", :action => 'google'
  62 plugin.map "google *words", :action => 'google'
  63 plugin.map "wp :lang *words", :action => 'wikipedia', :requirements => { :lang => /^\w\w\w?$/ }
  64 plugin.map "wp *words", :action => 'wikipedia'
  65