Panda, Tiger & Eagle

lib/api.rb

@@ -9,11 +9,15 @@ class Api
   def self.search_by_title(title)
     url = "http://api.rottentomatoes.com/api/public/v1.0/movies.json?apikey=#{APIKEY}&q=#{URI.encode(title)}&page_limit=1"
     struct = OpenStruct.new(get_url_as_json(url).fetch("movies").first)
-    Movie.new(id: struct.id.to_i,
-              title: struct.title,
-              year: struct.year,
-              score: struct.ratings["critics_score"]
-             )
+    if struct.id.nil?
+      :no_movie_found
+    else
+      Movie.new(id: struct.id.to_i,
+                title: struct.title,
+                year: struct.year,
+                score: struct.ratings["critics_score"]
+               )
+    end
   end
 
 

lib/text_color.rb

@@ -0,0 +1,28 @@
+# Found at: http://www.linuxgazette.net/issue65/padala.html
+
+# ATTRIBUTES
+RESET      = 0
+BRIGHT     = 1
+DIM        = 2
+UNDERLINE  = 3
+BLINK      = 4
+REVERSE    = 7
+HIDDEN     = 8
+
+# COLORS
+BLACK      = 0
+RED        = 1
+GREEN      = 2
+YELLOW     = 3
+BLUE       = 4
+MAGENTA    = 5
+CYAN       = 6
+WHITE      = 7
+
+def text_color(attr, color)
+  command = String.new
+
+  # Command is the control command to the terminal
+  command << "\e[%d;%dm" % [attr, color + 30]
+  printf(command)
+end

movie_json.rb

@@ -1,20 +1,83 @@
 require_relative "lib/movie"
 require_relative "lib/api"
+require_relative "lib/text_color"
 
-def find_movie
-  puts "OH HAI. Search?"
-  movie_title = gets
-  movie = Api.search_by_title(movie_title)
-  puts "Found: #{movie.title}. Score: #{movie.score}"
+def movies_average(movies_arr)
+  scores = []
+  movies_arr.each { |movie| scores << movie.score }
+  scores.inject{ |sum, el| sum + el }.to_f / scores.size
 end
 
-find_movie
+def movies_slope(movies_arr)
+  movies_by_year = movies_arr.sort_by! { |movie| movie.year }
+  y2_arr = []
+  y1_arr = []
+  movies_by_year.each do |movie|
+    if movie.year == movies_by_year.first.year
+      y2_arr << movie
+    end
+  end
+  movies_by_year.each do |movie|
+    if movie.year == movies_by_year.last.year
+      y1_arr << movie
+    end
+  end
+  y1 = movies_average(y1_arr)
+  y2 = movies_average(y2_arr)
+  x1 = movies_by_year.last.year
+  x2 = movies_by_year.first.year
+  (y1 - y2).to_f / (x1 - x2).to_f
+end
+
+def show_average(movies_arr)
+  text_color(BRIGHT, BLUE)
+  printf "Average: #{movies_average(movies_arr)}\n"
+  text_color(RESET, WHITE)
+end
+
+def show_contentment(movies_arr)
+  if movies_slope(movies_arr) == 0
+    puts "Staying Content"
+  elsif movies_slope(movies_arr) > 0
+    text_color(BRIGHT, YELLOW)
+    puts "Getting Happier"
+  else
+    text_color(BRIGHT, MAGENTA)
+    puts "Getting Madder"
+  end
+  text_color(RESET, WHITE)
+end
+
+def find_movie(movies_arr)
+  print "Search: "
+  searched_movie = gets.chomp
+  movie = Api.search_by_title(searched_movie)
+  if movie == :no_movie_found
+    text_color(BRIGHT, RED)
+    printf "\tCould not find #{searched_movie}\n"
+  else
+    text_color(BRIGHT, GREEN)
+    printf "\tFound: #{movie.title}\n\tScore: #{movie.score}\n"
+    movies_arr << movie
+  end
+  text_color(RESET, WHITE)
+end
+
+movies_arr = []
+
+puts "HI! Welcome to rssll's #{$0}."
+printf "Search for movies that you really like!\n\n"
+find_movie(movies_arr)
 
-while true do
-  puts "Search Again (Y/N)" 
-  answer = gets.upcase[0]
-  if answer == "Y"
-    find_movie
+loop do
+  print "Search Again? (Y/N) "
+  answer = gets.chomp
+  if answer =~ /Y/i
+    find_movie(movies_arr)
+    if movies_arr.size > 1
+      show_average(movies_arr)
+      show_contentment(movies_arr)
+    end
   else
     break
   end