summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--data/rbot/plugins/imdb.rb77
1 files changed, 71 insertions, 6 deletions
diff --git a/data/rbot/plugins/imdb.rb b/data/rbot/plugins/imdb.rb
index cfadd3c8..2388ed61 100644
--- a/data/rbot/plugins/imdb.rb
+++ b/data/rbot/plugins/imdb.rb
@@ -32,10 +32,9 @@ class Imdb
end
if resp.code == "200"
- m = /<a href="(\/title\/tt[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/.match(resp.body)
+ m = /<a href="(\/(?:title|name)\/(?:tt|nm)[0-9]+\/?)[^"]*"(?:[^>]*)>(?:[^<]*)<\/a>/.match(resp.body)
if m
url = m[1]
- title = m[2]
return url
end
elsif resp.code == "302"
@@ -50,6 +49,22 @@ class Imdb
debug "IMDB: search returned NIL"
return nil
end
+ type = sr.match(/^\/([^\/]+)\//)[1].downcase.intern rescue nil
+ case type
+ when :title
+ return info_title(sr)
+ when :name
+ return info_name(sr)
+ else
+ return "#{sr}"
+ end
+ end
+
+ def grab_info(info, body)
+ /<div class="info">\s+<h5>#{info}:<\/h5>\s+(.*?)<\/div>/mi.match(body)[1] rescue nil
+ end
+
+ def info_title(sr)
resp = nil
begin
resp = @bot.httputil.get_response('http://us.imdb.com' + sr,
@@ -70,12 +85,63 @@ class Imdb
score = m[1]
votes = m[2]
+ plot = nil
+ data = grab_info(/Plot (?:Outline|Summary)/, resp.body)
+ if data
+ plot = "Plot: #{data.ircify_html.gsub(/\s+more$/,'')}"
+ end
+
genre = Array.new
resp.body.scan(/<a href="\/Sections\/Genres\/[^\/]+\/">([^<]+)<\/a>/) do |gnr|
genre << gnr
end
- return ["http://us.imdb.com" + sr, title, score, votes,
- genre]
+ info = "#{title} : http://us.imdb.com#{sr}\n"
+ info << "Ratings: #{score}/10 (#{votes} voters). Genre: #{genre.join('/')}\n"
+ info << plot if plot
+ return info
+ end
+ return nil
+ end
+
+ def info_name(sr)
+ resp = nil
+ begin
+ resp = @bot.httputil.get_response('http://us.imdb.com' + sr,
+ :max_redir => -1)
+ rescue Exception => e
+ error e.message
+ warning e.backtrace.join("\n")
+ return nil
+ end
+
+ if resp.code == "200"
+ m = /<title>([^<]*)<\/title>/.match(resp.body)
+ return nil if !m
+ name = CGI.unescapeHTML(m[1])
+
+ birth = nil
+ data = grab_info("Date of Birth", resp.body)
+ if data
+ birth = "Birth: #{data.ircify_html.gsub(/\s+more$/,'')}"
+ end
+
+ death = nil
+ data = grab_info("Date of Death", resp.body)
+ if data
+ death = "Death: #{data.ircify_html.gsub(/\s+more$/,'')}"
+ end
+
+ awards = nil
+ data = grab_info("Awards", resp.body)
+ if data
+ awards = "Awards: #{data.ircify_html.gsub(/\s+more$/,'')}"
+ end
+
+ info = "#{name} : http://us.imdb.com#{sr}\n"
+ info << [birth, death].compact.join('. ') << "\n"
+ info << awards if awards
+ return info
+
end
return nil
end
@@ -94,8 +160,7 @@ class ImdbPlugin < Plugin
m.reply "Nothing found for #{what}"
return nil
end
- m.reply "#{info[1]} : #{info[0]}"
- m.reply "Ratings: #{info[2]}/10 (#{info[3]} voters). Genre: #{info[4].join('/')}"
+ m.reply info
end
end