imdb_og 0.5.0 → 0.5.1

data/VERSION

@@ -1 +1 @@
-0.5.0
+0.5.1

data/{imdb.gemspec → imdb_og.gemspec}

@@ -4,12 +4,12 @@
 # -*- encoding: utf-8 -*-
 
 Gem::Specification.new do |s|
-  s.name = %q{imdb}
-  s.version = "0.5.0"
+  s.name = %q{imdb_og}
+  s.version = "0.5.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jon Maddox"]
-  s.date = %q{2009-10-10}
+  s.date = %q{2009-11-03}
   s.description = %q{Simple library to look up movies on IMDB}
   s.email = %q{jon@mustacheinc.com}
   s.extra_rdoc_files = [
@@ -22,12 +22,14 @@ Gem::Specification.new do |s|
      "README.textile",
      "Rakefile",
      "VERSION",
+     "imdb_og.gemspec",
      "lib/imdb.rb",
      "lib/imdb/imdb.rb",
      "lib/imdb/imdb_company.rb",
      "lib/imdb/imdb_genre.rb",
      "lib/imdb/imdb_movie.rb",
      "lib/imdb/imdb_name.rb",
+     "lib/imdb/patches.rb",
      "test/imdb_test.rb",
      "test/test_helper.rb"
   ]

data/lib/imdb/imdb.rb

@@ -6,6 +6,18 @@ class Imdb
   IMDB_GENRE_BASE_URL = "http://www.imdb.com/Sections/Genres/"
   IMDB_SEARCH_BASE_URL = "http://imdb.com/find?s=all&q="
 
+
+  def self.search_movies_by_title(title)
+    document = Hpricot(open("#{IMDB_SEARCH_BASE_URL}#{CGI::escape(title)};s=tt").read)
+
+    results = document.search('a[@href^="/title/tt"]').reject do |element|
+      element.innerHTML.strip_tags.empty?
+    end.map do |element|
+      {:imdb_id => element['href'][/tt\d+/], :title => element.innerHTML.strip_tags.unescape_html}
+    end
+    results.uniq
+  end
+
   def self.find_movie_by_id(id)
     coder = HTMLEntities.new
 

data/lib/imdb/patches.rb

@@ -0,0 +1,13 @@
+require 'iconv'
+
+module StringPatches
+  def unescape_html
+    Iconv.conv("UTF-8", 'ISO-8859-1', CGI::unescapeHTML(self))
+  end
+
+  def strip_tags
+    gsub(/<\/?[^>]*>/, "")
+  end
+end
+
+String.send :include, StringPatches

data/lib/imdb.rb

@@ -2,9 +2,11 @@ require 'rubygems'
 require 'hpricot'
 require 'open-uri'
 require 'date'
+require 'cgi'
 require 'htmlentities'
 require 'imdb/imdb'
 require 'imdb/imdb_company'
 require 'imdb/imdb_movie'
 require 'imdb/imdb_name'
 require 'imdb/imdb_genre'
+require 'imdb/patches'

data/test/imdb_test.rb

@@ -10,6 +10,25 @@ class ImdbTest < Test::Unit::TestCase
       assert_equal "http://imdb.com/find?s=all&q=", Imdb::IMDB_SEARCH_BASE_URL
     end
   end
+  context "when searching" do
+    setup do
+      @results = Imdb.search_movies_by_title('transformers')
+    end
+    
+    should "return an array of results" do
+      assert_equal Array, @results.class
+    end
+
+    should "return an array of hashes" do
+      assert_equal Hash, @results.first.class
+    end
+
+    should "return an array of hashes with the right keys" do
+      assert @results.first.has_key?(:title)
+      assert @results.first.has_key?(:imdb_id)
+    end
+
+  end
 
   context "ImdbMovie" do
     context "when first created" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: imdb_og
 version: !ruby/object:Gem::Version 
-  version: 0.5.0
+  version: 0.5.1
 platform: ruby
 authors: 
 - Jon Maddox
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-10 00:00:00 -04:00
+date: 2009-11-03 00:00:00 -05:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -57,13 +57,14 @@ files:
 - README.textile
 - Rakefile
 - VERSION
-- imdb.gemspec
+- imdb_og.gemspec
 - lib/imdb.rb
 - lib/imdb/imdb.rb
 - lib/imdb/imdb_company.rb
 - lib/imdb/imdb_genre.rb
 - lib/imdb/imdb_movie.rb
 - lib/imdb/imdb_name.rb
+- lib/imdb/patches.rb
 - test/imdb_test.rb
 - test/test_helper.rb
 has_rdoc: true