summaryrefslogtreecommitdiff
path: root/data/rbot/plugins/imdb.rb
blob: 432d0efc47ea4d87f537f6d431a9f3abb7073e0a (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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# IMDB plugin for RubyBot
# (c) 2005 Arnaud Cornet <arnaud.cornet@gmail.com>
# Licensed under MIT License.

require 'net/http'
require 'cgi'
require 'uri/common'

class Imdb
  def initialize(bot)
    @bot = bot
  end

  def search(rawstr)
    str = URI.escape(rawstr)
    @http = @bot.httputil.get_proxy(URI.parse("http://us.imdb.com/find?q=#{str}"))
    @http.start
    begin
    resp, data = @http.get("/find?q=#{str}", "User-Agent" => "Mozilla/5.0")
    rescue Net::ProtoRetriableError => detail
      head = detail.data
      if head.code == "301" or head.code == "302"
            return head['location'].gsub(/http:\/\/us.imdb.com/, "").gsub(/\?.*/, "")
        end
    end
    if resp.code == "200"
      m = /<a href="(\/title\/tt[0-9]+\/?)[^"]*"(:?[^>]*)>([^<]*)<\/a>/.match(resp.body)
      if m
        url = m[1]
        title = m[2]
        return url
      end
    elsif resp.code == "302"
      return resp['location'].gsub(/http:\/\/us.imdb.com/, "").gsub(/\?.*/, "")
    end
    return nil
  end

  def info(rawstr)
    sr = search(rawstr)
    if !sr
      debug "IMDB: search returned NIL"
      return nil
    end
    resp, data = @http.get(sr, "User-Agent" =>
      "Mozilla/5.0 (compatible; Konqueror/3.1; Linux)")
    if resp.code == "200"
      m = /<title>([^<]*)<\/title>/.match(resp.body)
      return nil if !m
      title = CGI.unescapeHTML(m[1])

      m = /<b>([0-9.]+)\/10<\/b> \(([0-9,]+) votes?\)/.match(resp.body)
      return nil if !m
      score = m[1]
      votes = m[2]

      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]
    end
    return nil
  end
end

class ImdbPlugin < Plugin
  def help(plugin, topic="")
    "imdb <string> => search http://www.imdb.org for <string>"
  end

  def privmsg(m)
    unless(m.params && m.params.length > 0)
      m.reply "incorrect usage: " + help(m.plugin)
      return
    end

    i = Imdb.new(@bot)
    info = i.info(m.params)
    if !info
      m.reply "Nothing found for #{m.params}"
      return nil
    end
    m.reply "#{info[1]} : #{info[0]}"
    m.reply "Ratings: #{info[2]}/10 (#{info[3]} voters). Genre: #{info[4].join('/')}"
  end
end

plugin = ImdbPlugin.new
plugin.register("imdb")