TITLE_OR_NAME_MATCH = /<a href="(\/(?:title|name)\/(?:tt|nm)[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
TITLE_MATCH = /<a href="(\/title\/tt[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
NAME_MATCH = /<a href="(\/name\/nm[0-9]+\/?)[^"]*"(?:[^>]*)>([^<]*)<\/a>/
+ FINAL_ARTICLE_MATCH = /, ([A-Z]\S{0,2})$/
def initialize(bot)
@bot = bot
end
def search(rawstr)
- str = URI.escape(rawstr) << ";site=aka"
+ str = URI.escape(rawstr)
+ str << ";site=aka" if @bot.config['imdb.aka']
return do_search(str)
end
if resp.code == "200"
m = []
- m << TITLE_OR_NAME_MATCH.match(resp.body)
- if resp.body.match(/\(Exact Matches\)<\/b>/)
+ m << TITLE_OR_NAME_MATCH.match(resp.body) if @bot.config['imdb.popular']
+ if resp.body.match(/\(Exact Matches\)<\/b>/) and @bot.config['imdb.exact']
m << TITLE_OR_NAME_MATCH.match($')
end
m.compact!
/<div class="info">\s+<h5>#{info}:<\/h5>\s+(.*?)<\/div>/mi.match(body)[1] rescue nil
end
+ def fix_article(org_tit)
+ title = org_tit.dup
+ if @bot.config['imdb.fix_article'] and title.gsub!(FINAL_ARTICLE_MATCH, '')
+ art = $1.dup
+ debug art.inspect
+ if art[-1,1].match(/[a-z]/)
+ art << " "
+ end
+ return art + title
+ end
+ return title
+ end
+
def info_title(sr)
resp = nil
begin
m = /<title>([^<]*)<\/title>/.match(resp.body)
return nil if !m
title_date = m[1]
- title, date, extra = title_date.scan(/^(.*)\((\d\d\d\d(?:[IV]+)?)\)\s*(.+)?$/).first
- title.strip!
+ pre_title, date, extra = title_date.scan(/^(.*)\((\d\d\d\d(?:\/[IV]+)?)\)\s*(.+)?$/).first
+ pre_title.strip!
+ title = fix_article(pre_title)
dir = nil
data = grab_info(/Directors?/, resp.body)
country = nil
data = grab_info(/Country/, resp.body)
if data
- country = data.ircify_html
+ country = data.ircify_html.gsub(' / ','/')
end
info << [title, "(#{country}, #{date})", extra, dir ? "[#{dir}]" : nil, ": http://us.imdb.com#{sr}"].compact.join(" ")
what = str.match(/<a name="[^"]+">([^<]+)<\/a>/)[1] rescue nil
next unless what
movies[what] = str.scan(TITLE_MATCH)[0..2].map { |url, tit|
- tit
+ fix_article(tit)
}
}
end
end
class ImdbPlugin < Plugin
+ BotConfig.register BotConfigBooleanValue.new('imdb.aka',
+ :default => true,
+ :desc => "Look for IMDB matches also in translated titles and other 'also known as' information")
+ BotConfig.register BotConfigBooleanValue.new('imdb.popular',
+ :default => true,
+ :desc => "Display info on popular IMDB entries matching the request closely")
+ BotConfig.register BotConfigBooleanValue.new('imdb.exact',
+ :default => true,
+ :desc => "Display info on IMDB entries matching the request exactly")
+ BotConfig.register BotConfigBooleanValue.new('imdb.fix_article',
+ :default => false,
+ :desc => "Try to detect an article placed at the end and move it in front of the title")
+
def help(plugin, topic="")
"imdb <string> => search http://www.imdb.org for <string>"
end