imdb 0.1.0 → 0.2.0

data/History.txt

@@ -1,3 +1,8 @@
+== 0.2.0 2009-06-04
+
+* Added console tool 'imdb' for searching and getting movie info.
+* Fixed issue #2
+
 == 0.1.0 2009-06-03
 
 * Added Imdb::Search that allows search IMDB for a specific movie.

data/Manifest.txt

@@ -2,13 +2,16 @@ History.txt
 Manifest.txt
 README.rdoc
 Rakefile
+bin/imdb
 lib/imdb.rb
+lib/imdb/cli.rb
 lib/imdb/movie.rb
 lib/imdb/search.rb
 lib/imdb/string_extensions.rb
 script/console
 script/destroy
 script/generate
+spec/imdb_cli_spec.rb
 spec/imdb_movie_spec.rb
 spec/imdb_search_spec.rb
 spec/spec.opts

data/bin/imdb

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+#
+#  Created on 2009-6-4.
+#  Copyright (c) 2009. All rights reserved.
+
+require File.expand_path(File.dirname(__FILE__) + "/../lib/imdb")
+
+require "imdb/cli"
+
+Imdb::CLI.execute(STDOUT, ARGV)

data/lib/imdb.rb

@@ -10,5 +10,5 @@ require 'imdb/search'
 require 'imdb/string_extensions'
 
 module Imdb
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

data/lib/imdb/cli.rb

@@ -0,0 +1,100 @@
+require 'optparse'
+
+module Imdb
+  class CLI
+    
+    # Run the imdb command
+    #
+    # Searching
+    #
+    #   imdb Star Trek
+    #
+    # Get a movie, supply a 7 digit IMDB id
+    #
+    #   imdb 0095016
+    #
+    def self.execute(stdout, arguments=[])
+      
+      @stdout = stdout
+      
+      options = {
+      }
+      mandatory_options = %w(  )
+      
+      parser = OptionParser.new do |opts|
+        opts.banner = <<-BANNER.gsub(/^          /,'')
+          IMDB #{Imdb::VERSION}
+
+          Usage: #{File.basename($0)} Search Query
+                 #{File.basename($0)} 0095016
+
+        BANNER
+        opts.separator ""
+        opts.on("-v", "--version",
+                "Show the current version.") { stdout.puts "IMDB #{Imdb::VERSION}"; exit }
+        opts.on("-h", "--help",
+                "Show this help message.") { stdout.puts opts; exit }
+        opts.parse!(arguments)
+
+        if mandatory_options && mandatory_options.find { |option| options[option.to_sym].nil? }
+          stdout.puts opts; exit
+        end
+      end
+
+      query = arguments.join(" ").strip
+      exit if query.blank?
+      
+      movie, search = nil, nil
+      
+      # If ID, fetch movie
+      if query.match(/\d\d\d\d\d\d\d/)
+        fetch_movie(query)
+      else
+        search_movie(query)
+      end
+    end
+    
+    def self.fetch_movie(imdb_id)
+      @stdout.puts ">> Fetching movie #{imdb_id}"
+      
+      movie = Imdb::Movie.new(imdb_id)
+      
+      display_movie_details(movie)
+    end
+    
+    def self.search_movie(query)
+      @stdout.puts ">> Searching for \"#{query}\""
+      
+      search = Imdb::Search.new(query)
+      
+      if search.movies.size == 1
+        display_movie_details(search.movies.first)
+      else
+        display_search_results(search.movies)
+      end
+    end
+    
+    def self.display_movie_details(movie)
+      @stdout.puts
+      @stdout.puts "#{movie.title} (#{movie.year})"
+      @stdout.puts "=" * 72
+      @stdout.puts "Rating: #{movie.rating}"
+      @stdout.puts "Duration: #{movie.length} minutes"
+      @stdout.puts "Directed by: #{movie.director}"
+      @stdout.puts "Cast by: #{movie.cast_members[0..4].join(", ")}"
+      @stdout.puts "Genres: #{movie.genres.join(", ")}"
+      @stdout.puts "#{movie.plot}"
+      @stdout.puts "=" * 72
+      @stdout.puts
+    end
+    
+    def self.display_search_results(movies = [])
+      movies = movies[0..9] # limit to ten top hits
+      
+      movies.each do |movie|
+        @stdout.puts " > #{movie.id} | #{movie.title}"
+      end
+    end
+    
+  end
+end

data/lib/imdb/movie.rb

@@ -17,7 +17,7 @@ module Imdb
     def initialize(imdb_id, title = nil)
       @id = imdb_id
       @url = "http://www.imdb.com/title/tt#{imdb_id}/"
-      @title = title
+      @title = title.gsub(/"/, "")
     end
     
     # Returns an array with cast members

data/spec/imdb_cli_spec.rb

@@ -0,0 +1,34 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+require 'imdb/cli'
+
+describe Imdb::CLI, "execute" do
+  
+  describe "yield search results" do
+    before(:each) do
+      @stdout_io = StringIO.new
+      Imdb::CLI.execute(@stdout_io, ["Star Trek"])
+      @stdout_io.rewind
+      @stdout = @stdout_io.read
+    end
+  
+    it "report data" do
+      @stdout.should =~ /0060028/
+      @stdout.should =~ /Star Trek/
+      @stdout.should =~ /1966/
+    end
+  end
+
+  describe "yield one movie" do
+    before(:each) do
+      @stdout_io = StringIO.new
+      Imdb::CLI.execute(@stdout_io, ["0117731"])
+      @stdout_io.rewind
+      @stdout = @stdout_io.read
+    end
+  
+    it "report data" do
+      @stdout.should =~ /Star Trek\: First Contact \(1996\)/
+      @stdout.should =~ /Jonathan Frakes/
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: imdb
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Ariejan de Vroom
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-06-03 00:00:00 +02:00
+date: 2009-06-04 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -55,8 +55,8 @@ dependencies:
 description: This packages allows you to easy access publicly available data from IMDB.
 email: 
 - ariejan@ariejan.net
-executables: []
-
+executables: 
+- imdb
 extensions: []
 
 extra_rdoc_files: 
@@ -68,13 +68,16 @@ files:
 - Manifest.txt
 - README.rdoc
 - Rakefile
+- bin/imdb
 - lib/imdb.rb
+- lib/imdb/cli.rb
 - lib/imdb/movie.rb
 - lib/imdb/search.rb
 - lib/imdb/string_extensions.rb
 - script/console
 - script/destroy
 - script/generate
+- spec/imdb_cli_spec.rb
 - spec/imdb_movie_spec.rb
 - spec/imdb_search_spec.rb
 - spec/spec.opts