blob: da156fda4e95047b36070fc28b7be1e84247876f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
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{About ([\d,]+) results})
res[0][0].to_s.tr(",", "").to_i
end
end
plugin = GoogleFightPlugin.new
plugin.map "googlefight *keywords", :action => "fight",
:requirements => { :keywords => /^[\w\s"]+? (?:(?:(?:\||vs\.) )?[\w\s"]+?)+/ }
|