Movie JSON submission

Gemfile

@@ -0,0 +1,4 @@
+source 'http://rubygems.org'
+
+
+gem 'rspec'

Gemfile.lock

@@ -0,0 +1,18 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    diff-lcs (1.2.5)
+    rspec (2.14.1)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    rspec-core (2.14.7)
+    rspec-expectations (2.14.5)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.14.5)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  rspec

README.md

@@ -42,7 +42,7 @@ Slope of line is `(y1 - y2) / (x1 - x2)` ... In this case, if my average rating
 
 ```
 (45 - 50).to_f / (2012 - 1990).to_f
-=> -0.22727272727272727  
+=> -0.22727272727272727
 # getting madder
 ```
 

lib/api.rb

@@ -2,6 +2,7 @@
 require "json"
 require "ostruct"
 require_relative "./movie"
+require_relative "./movie_library"
 class Api
 
   APIKEY="4t6456xa33z8qhcqyuqgnkjh"
@@ -16,7 +17,6 @@ def self.search_by_title(title)
              )
   end
 
-
   def self.get_url_as_json(url)
     JSON.parse(open(url).read)
   end

lib/movie_library.rb

@@ -0,0 +1,44 @@
+class MovieLibrary
+
+  attr_reader :movies, :library, :slope
+  def initialize
+    @movies = []
+    @library = []
+    @slope = 0
+  end
+
+  def calculate_slope(catalog)
+    first = catalog.first
+    last = catalog.last
+    return 0 if first == last
+    (first.score - last.score).to_f / (last.year - first.year).to_f
+  end
+
+  def sort_by_year(movies)
+    movies.sort { |movie,another_movie| movie.year <=> another_movie.year }
+  end
+
+  def average_rating_by_year(year)
+    average_rating all_movies.select {|movie| movie.year == year}
+  end
+
+  def average_rating(movies)
+    return 0 if count == 0
+    movies.inject(0.0) { |sum, movie| sum + movie.score } / movies.size
+  end
+
+  def catalog(movie)
+    @movies << movie
+    @library = sort_by_year(@movies)
+    @slope = calculate_slope(@library)
+    return @library
+  end
+
+  def all_movies
+    @movies
+  end
+
+  def count
+    movies.count
+  end
+end

movie_json.rb

@@ -1,21 +1,42 @@
 require_relative "lib/movie"
+require_relative "lib/movie_library"
 require_relative "lib/api"
 
 def find_movie
-  puts "OH HAI. Search?"
-  movie_title = gets
-  movie = Api.search_by_title(movie_title)
-  puts "Found: #{movie.title}. Score: #{movie.score}"
+  begin
+    puts "OH HAI. Add a movie you really like?"
+    movie_title = gets
+    movie = Api.search_by_title(movie_title)
+  rescue NoMethodError
+    puts "Not found."
+  end
 end
 
-find_movie
+movie_library = MovieLibrary.new
+movie = find_movie
+unless movie.nil?
+movie_library.catalog(movie)
+puts "Found: #{movie.title}. Score: #{movie.score}"
+end
 
 while true do
-  puts "Search Again (Y/N)" 
+  puts "Search Again (Y/N)"
   answer = gets.upcase[0]
   if answer == "Y"
-    find_movie
+    movie = find_movie
+    unless movie.nil?
+    movie_library.catalog(movie)
+    puts "Found: #{movie.title}. Score: #{movie.score}"
+    end
   else
+    puts "\n--------------------------"
+    puts "Summary for your Movie Library"
+    puts "-----------------------------"
+    movie_library.library.each do |movie|
+      puts "ID: #{movie.id}\tMovie: #{movie.title}\tYear: #{movie.year} Score: #{movie.score}\tAverage: #{movie_library.average_rating_by_year(movie.year)}"
+    end
+    puts "Your library has a Happiness Index of: #{movie_library.slope}"
+    puts "Goodbye!!!"
     break
   end
 end

spec/api_spec.rb

@@ -3,25 +3,30 @@
 
 describe Api do
 
-  let(:movie) { Api.search_by_title("Forrest Gump") }
+    let(:movie) { Api.search_by_title("Forrest Gump") }
 
-  before do
-    Api.stub(:get_url_as_json) { JSON.parse(File.read("spec/fixtures/forrest.json")) }
-  end
+    before do
+      Api.stub(:get_url_as_json) { JSON.parse(File.read("spec/fixtures/forrest.json")) }
+    end
 
-  it "should search for movies" do
-    movie.title.should eq("Forrest Gump")
-  end
+    it "should search for movies" do
+      movie.title.should eq("Forrest Gump")
+    end
 
-  it "should return the score" do
-    movie.score.should eq(71)
-  end
+    it "should return the score" do
+      movie.score.should eq(71)
+    end
 
-  it "should return the id" do
-    movie.id.should eq(10036)
-  end
+    it "should return the id" do
+      movie.id.should eq(10036)
+    end
+
+    it "should return the year" do
+      movie.year.should eq(1994)
+    end
+
+    it "should not raise an error when searching for a non-existent movie" do
+      expect { Api.search_by_title("NOTHINGFOUNDHERE") }.to_not raise_error
+    end
 
-  it "should return the year" do
-    movie.year.should eq(1994)
-  end
 end

spec/movie_library_spec.rb

@@ -0,0 +1,81 @@
+require 'rspec'
+require_relative "../lib/movie"
+require_relative "../lib/movie_library"
+describe MovieLibrary do
+
+  let(:movie_library) {MovieLibrary.new}
+  let (:movie){Movie.new(id: "the-id", title: "the-title", year: 1998, score: 50)}
+  let (:another_movie){Movie.new(id: "the-id", title: "the-title", year: 2012, score: 50)}
+  let (:yet_another_movie){Movie.new(id: "the-id", title: "the-title", year: 1990, score: 45)}
+  describe "#new" do
+
+    it 'should be a movie library object ' do
+      movie_library.is_a?(MovieLibrary).should be_true
+    end
+
+    it 'should have no movies when initialized' do
+      movie_library.count.should eq(0)
+    end
+  end
+
+  describe "#catalog" do
+
+    it 'should add a new movie to the library whenever a search happens' do
+      movie_library.catalog(movie)
+      expect(movie_library.all_movies).to include (movie)
+    end
+    it 'should increase the movie count' do
+      movie_library.catalog(movie)
+      movie_library.count.should eq(1)
+    end
+  end
+
+  describe "#average_rating" do
+
+    it "should have no average rating when no movies are cataloged" do
+      movie_library.average_rating(Array.new).should eq(0)
+    end
+
+    it 'should calculate the average rating' do
+      movie_library.catalog(movie)
+      movie_library.average_rating(movie_library.all_movies).should eq(50)
+    end
+  end
+
+  describe "#average_rating_by_year" do
+
+    it 'should have no average yearly rating when no movies are cataloged' do
+      movie_library.average_rating_by_year(movie.year).should eq(0)
+    end
+
+    it 'should calculate the average rating by year when the library has' do
+      movie_library.catalog(movie)
+      movie_library.catalog(another_movie)
+      movie_library.catalog(yet_another_movie)
+      movie_library.average_rating_by_year(1998).should eq(50)
+    end
+  end
+
+  describe "#sort_by_year" do
+
+    it 'should sort the movies by year' do
+      movie_library.catalog(another_movie)
+      movie_library.catalog(yet_another_movie)
+      movie_library.library.should == [yet_another_movie, another_movie]
+    end
+  end
+
+  describe "#calculate_slope" do
+
+    it "should have no slope when the library only contains one movie" do
+      movie_library.catalog(movie)
+      movie_library.slope.should eq(0)
+    end
+    it 'should calculate the slope of the average ratings from the first year to the last year' do
+      movie_library.catalog(another_movie)
+      movie_library.catalog(yet_another_movie)
+      movie_library.slope.should eq(-0.22727272727272727)
+    end
+
+  end
+end

spec/movie_spec.rb

@@ -1,12 +1,21 @@
+require 'rspec'
 require_relative "../lib/movie"
 describe Movie do
 
-  it "should store the title, year, and score" do
-    movie = Movie.new(id: "the-id", title: "the-title", year: 1998, score: 50)
-    movie.id.should eq("the-id")
-    movie.title.should eq("the-title")
-    movie.year.should eq(1998)
-    movie.score.should eq(50)
-  end
-
+	describe "#new" do
+
+    let (:movie){Movie.new(id: "the-id", title: "the-title", year: 1998, score: 50)}
+
+	  it "should store the title, year, and score" do
+	    movie.id.should eq("the-id")
+	    movie.title.should eq("the-title")
+	    movie.year.should eq(1998)
+	    movie.score.should eq(50)
+	  end
+
+    it 'should a movie object' do
+      movie.is_a?(Movie).should be_true
+    end
+	end
+
 end