summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--data/rbot/plugins/google.rb61
1 files changed, 25 insertions, 36 deletions
diff --git a/data/rbot/plugins/google.rb b/data/rbot/plugins/google.rb
index 604f44cb..8c573c39 100644
--- a/data/rbot/plugins/google.rb
+++ b/data/rbot/plugins/google.rb
@@ -1,53 +1,42 @@
-require 'net/http'
-require 'uri/common'
+require 'uri'
Net::HTTP.version_1_2
+GOOGLE_WAP_LINK = /<a accesskey="(\d)" href=".*?u=(.*?)">(.*?)<\/a>/im
+
class GooglePlugin < Plugin
def help(plugin, topic="")
- "search <string> => search google for <string>"
+ "google <string> => search google for <string>"
end
- def privmsg(m)
- unless(m.params && m.params.length > 0)
- m.reply "incorrect usage: " + help(m.plugin)
- return
- end
- searchfor = URI.escape m.params
- query = "/search?q=#{searchfor}&btnI=I%27m%20feeling%20lucky"
- result = "not found!"
+ def google(m, params)
+ what = params[:words].to_s
+ searchfor = URI.escape what
- proxy_host = nil
- proxy_port = nil
+ url = "http://www.google.com/wml/search?q=#{searchfor}"
- if(ENV['http_proxy'])
- if(ENV['http_proxy'] =~ /^http:\/\/(.+):(\d+)$/)
- proxy_host = $1
- proxy_port = $2
- end
- end
-
- http = @bot.httputil.get_proxy(URI.parse("http://www.google.com"))
begin
- http.start {|http|
- resp = http.get(query)
- if resp.code == "302"
- result = resp['location']
- end
- }
+ wml = @bot.httputil.get(url)
rescue => e
- p e
- if e.response && e.response['location']
- result = e.response['location']
- else
- result = "error!"
- end
+ m.reply "error googling for #{what}"
+ return
end
- m.reply "#{m.params}: #{result}"
+ results = wml.scan(GOOGLE_WAP_LINK)
+ if results.length == 0
+ m.reply "no results found for #{what}"
+ return
+ end
+ results = results[0...3].map { |res|
+ "#{res[0]}. #{Bold}#{Utils.decode_html_entities res[2].strip}#{Bold}: #{URI.unescape res[1].strip}"
+ }.join(" | ")
+
+ m.reply "Results for #{what}: #{results}"
end
end
+
plugin = GooglePlugin.new
-plugin.register("search")
-plugin.register("google")
+
+plugin.map "search *words", :action => 'google'
+plugin.map "google *words", :action => 'google'