require "json"
require "rest-client"
require "date"
require "yaml"

class Movies
  attr_reader :title, :year, :rated, :plot, :genres, :director, :writers, :actors, :poster, :runtime, :rating, :votes, :id
  
  def initialize(url, params = {})
    if params.keys.include?(:callback)
      raise ArgumentError.new("Passing the callback option makes not sense.")
    end
    @url = "#{url}&#{(@params = params).map{|key, value| "#{key}=#{value}"}.join("&")}"
  end

  def self.find_by_id(id, params = {})
    unless id.to_s.match(/tt\d{4,}/)
      raise ArgumentError.new("The id is not valid.")
    end
    Movies.new("http://www.imdbapi.com/?i=#{id}", params).prepare
  end
  
  def self.find_by_title(title, params = {})
    if title.nil? or title.empty?
      raise ArgumentError.new("Title can not be blank.")
    end
    Movies.new("http://www.imdbapi.com/?t=#{URI.encode(title)}", params).prepare
  end
  
  def self.find_by_release_name(title, params = {})
    if title.nil? or title.empty?
      raise ArgumentError.new("Title can not be blank.")
    end
    
    params.merge!(y: $1) if title =~ /((19|20)\d{2})/
    Movies.new("http://www.imdbapi.com/?t=#{URI.encode(Movies.cleaner(title))}", params).prepare
  end
  
  def prepare
    tap do
      return self unless found?
      
      content.keys.each do |name| 
        instance_variable_set "@" + name.to_s.downcase, (content[name] == "N/A" ? nil : content[name])
      end
      
      @year    = @year.to_i
      @genres  = @genre.split(",")
      @writers = @writer.split(",")
      @actors  = @actors.split(", ")
      @rating  = @rating.to_f
      @votes   = @votes.to_i
      
      if @runtime =~ /(\d+).+?(\d+)/
        @runtime = $1.to_i * 60 + $1.to_i
      elsif @runtime =~ /(\d+) hrs/
        @runtime = $1.to_i * 60
      else
        @runtime = 0
      end 
    end
  end
  
  def found?
    content["Response"] == "True"
  end

  def released
    Date.parse(@released)
  rescue
    return nil
  end
  
  def tomato
    unless @params[:tomatoes]
      raise ArgumentError.new("You have to set 'tomatoes' to true to get this data.")
    end
    
    @_tomato ||= Struct.new(
      :meter, 
      :image,
      :rating,
      :reviews,
      :fresh,
      :rotten
    ).new(
      @tomatometer.to_i, 
      @tomatoimage,
      @tomatorating.to_f,
      @tomatoreviews.to_i,
      @tomatofresh.to_i,
      @tomatorotten.to_i
    )
  end
  
  def href
    "http://www.imdb.com/title/#{@id}/"
  end
  
  def self.excluded
    @_excluded ||= YAML.load_file("#{File.dirname(__FILE__)}/movies/exclude.yml")["excluded"]
  end
  
  def self.cleaner(string)
    excluded.each do |clean|
      string = string.gsub(/#{clean}/i, ' ')
    end

    [/((19|20)\d{2})/, /\./, /\s*-\s*/, /\s{2,}/].each do |regex|
      string = string.gsub(regex, ' ')
    end

    string.strip
  end
  
  private
    def content
      @_content ||= JSON.parse(download)
    end

    def download
      @_download ||= RestClient.get(@url, timeout: 3)
    end
end