Contents

require "imdb_party"
require 'imdb_party/levenshtein'

class MovieSearcher
  attr_accessor :options
  
  def initialize(args)
    args.keys.each { |name| instance_variable_set "@" + name.to_s, args[name] unless name == :options }
    
    @options = {
      :long => 15,
      :split => /\s+|\./,
      :imdb => ImdbParty::Imdb.new,
      :limit => 0.4
    }
    
    @options.merge!(args[:options]) unless args[:options].nil?
  end
  
  def self.find_by_release_name(search_value, options = {})
    this = MovieSearcher.new(options.merge(:search_value => search_value.to_s))
    return if this.to_long?
    
    movie = this.find_the_movie!
    return if movie.nil?
    
    # If the user wants for information about the movie, the {options[:details]} option should be true
    this.options[:details] ? self.find_movie_by_id(movie.imdb_id) : movie
  end
  
  def to_long?
    @split = @search_value.split(@options[:split])
    @split.length > @options[:long]
  end
  
  def find_the_movie!
    current =  @split.length
    
    until current <= 0 do
      title = @split.take(current).join(' ')
      movies = @options[:imdb].find_by_title(title)
      break if movies.any?
      current -= 1 
    end
    
    return if movies.nil? or not movies.any?
    
    # Cleaning up the title, no years allowed
    title = title.gsub(/[^a-z0-9]/i, '').gsub(/(19|20)\d{2}/, '')
    
    movie = movies.map do |movie| 
      [movie, Levenshtein.distance(movie[:title].gsub(/[^a-z0-9]/i, ''), title, @options[:limit])]
    end.reject do |value|
      value.last.nil?
    end.sort_by do |_,value|
      value
    end.first
    
    return if movie.nil?
    
    return ImdbParty::Movie.new(movie.first)
  end
  
  def self.method_missing(method, *args, &block)  
    result = ImdbParty::Imdb.new.send(method, *args)
    return if result.nil?
    result.class == Array ? result.map{|r| ImdbParty::Movie.new(r)} : result
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
movie_searcher-0.0.6	lib/movie_searcher.rb