whats-for-dinner 1.0.0

data/lib/command_line_interface.rb

@@ -0,0 +1,114 @@
+class CommandLineInterface
+
+  CATEGORIES = Category.get_category_list
+
+  def run
+    location = get_location
+    users = get_users
+    whats_for_dinner = WhatsForDinner.new(location, users)
+    loop do
+      restaurant = whats_for_dinner.ask
+      make_recommendation(restaurant)
+      break if yes?("Sound good?")
+    end
+    puts "Great!  Thanks for using What's For Dinner?, powered by Yelp."
+  end
+
+  def get_location
+    puts "Hi!  Where do you want to have dinner tonight?"
+    location = gets.strip
+  end
+
+  def get_users
+    users = []
+    loop do
+      puts "What's your name?"
+      name = gets.strip
+      puts "What are you in the mood for today?"
+      likes = get_preferences(gets)
+      puts "What don't you want to eat?"
+      dislikes = get_preferences(gets)
+      user = User.new(name, likes, dislikes)
+      users << user
+      break unless yes?("Is anyone else joining you?")
+    end
+    users
+  end
+
+  def make_recommendation(restaurant)
+    puts "How about?"
+    puts ""
+    puts restaurant.name
+    puts "#{restaurant.rating} star rating, #{restaurant.review_count} reviews"
+    puts restaurant.categories.map{|category| category.title}.join(", ")
+    puts ""
+    puts restaurant.display_address
+    puts restaurant.display_phone
+    puts ""
+    puts "\"#{restaurant.snippet_text}\""
+    puts ""
+  end
+
+  def get_preferences(input)
+    validated_categories = []
+    if input.strip == ""
+      validated_categories
+    else
+      user_categories = input
+        .split(/[,][ ]*/)
+        .select {|string| !string.include? ","}
+        .map {|string| string.strip}
+      user_categories.each do |user_category|
+        if CATEGORIES.map{|category| category.title.downcase}.include? user_category.downcase
+          validated_categories << user_category.downcase
+        else
+          suggested_categories = suggested_categories(user_category).map{|category| category.title}
+          formatted_categories = format_categories(suggested_categories)
+          puts "I'm sorry, I didn't understand \"#{user_category}\".  Did you mean #{formatted_categories}?"
+          recover_category = gets
+            .split(/[,][ ]*/)
+            .select {|string| !string.include? ","}
+            .first
+          get_preferences(recover_category).each {|category| validated_categories << category}
+        end
+      end
+      validated_categories.uniq
+    end
+  end
+
+  def suggested_categories(input, categories = CATEGORIES, characters = 0)
+    suggestions = categories
+      .select {|category| category.title.downcase[0, characters] == input.downcase[0, characters]}
+      .compact
+    if suggestions.empty? || suggested_categories(input, suggestions, characters + 1).empty?
+      suggestions
+    else
+      suggested_categories(input, suggestions, characters + 1)
+    end
+  end
+
+  def format_categories(array, string = "")
+    head = "\"" + array[0].split.map{|word| word.capitalize}.join(" ") + "\""
+    tail = array[1..-1]
+    if tail.empty?
+      string == ""? head : string + ", or " + head
+    else
+      string = string == ""? head : string + ", " + head
+      format_categories(tail, string)
+    end
+  end
+
+  def yes?(prompt)
+    puts prompt
+    case gets.strip.downcase
+    when "yes"
+      true
+    when "no"
+      false
+    else
+      puts "I'm sorry, I didn't understand you.  Please say \"Yes\" or \"No\"."
+      yes?(prompt)
+    end
+  end
+
+end

data/lib/restaurant.rb

@@ -0,0 +1,33 @@
+class Restaurant
+  attr_accessor :id, :is_claimed, :is_closed, :name, :image_url, :url, :mobile_url, :phone, :display_phone, :review_count, :categories, :distance, :rating, :rating_img_url, :rating_img_url_small, :rating_img_url_large, :snippet_text, :snippet_image_url, :address, :display_address, :city, :state_code, :postal_code, :country_code, :cross_streets, :neighborhoods, :latitude, :longitude, :menu_provider, :menu_date_updated, :reservation_url, :eat24_url
+
+  @@all = []
+
+  def initialize(restaurant)
+    location = restaurant.location
+    coordinate = location.coordinate
+    [restaurant, location, coordinate].each do |instance|
+      instance.instance_variables.each do |variable|
+        instance_has = instance.instance_variable_defined?(variable)
+        self_has = self.instance_variables.find(variable)
+        if instance_has && self_has
+          if variable.to_s == "@categories"
+            categories = instance.categories
+              .map {|category| Category.find_by_title(category[0])}
+              .compact
+            @categories = categories
+            categories.each {|category| category.restaurants << self}
+          else
+            self.instance_variable_set(variable, instance.instance_variable_get(variable))
+          end
+        end
+      end
+    end
+    @@all << self
+  end
+
+  def self.all
+    @@all
+  end
+
+end

data/lib/user.rb

@@ -0,0 +1,17 @@
+class User
+  attr_accessor :name, :likes, :dislikes
+
+  @@all = []
+
+  def initialize(name, likes, dislikes)
+    @name = name
+    @likes = likes.map {|title| Category.find_by_title(title)}
+    @dislikes = dislikes.map {|title| Category.find_by_title(title)}
+    @@all << self
+  end
+
+  def self.all
+    @@all
+  end
+
+end

data/lib/whats_for_dinner.rb

@@ -0,0 +1,51 @@
+class WhatsForDinner
+  attr_accessor :location, :users, :yelp_client, :categories, :preferred_categories, :restaurants
+
+  def initialize(location, users)
+    @location = location
+    @users = users
+    @yelp_client = YelpClient.new
+    @categories = Category.all
+    @preferred_categories = []
+    @restaurants = []
+  end
+
+  def ask
+    if self.restaurants.empty?
+      category_filters = get_category_filters
+      self.categories = self.categories - self.preferred_categories
+      params = {category_filter: category_filters}
+      results = yelp_client.search(location, params)
+      self.restaurants = results.map{|result| Restaurant.new(result)}
+      if self.restaurants.empty?
+        ask
+      else
+        recommendation = self.restaurants.first
+        self.restaurants = self.restaurants[1..-1]
+        recommendation
+      end
+    else
+      recommendation = self.restaurants.first
+      self.restaurants = self.restaurants[1..-1]
+      recommendation
+    end
+  end
+
+  def get_category_filters
+    categories_with_votes = self.categories.map{|category| [category, 0]}.to_h
+    users.each do |user|
+      user.likes.each do |category|
+        categories_with_votes[category] += 1 if categories_with_votes.has_key?(category)
+      end
+      user.dislikes.each do |category|
+        categories_with_votes[category] -= 100 if categories_with_votes.has_key?(category)
+      end
+    end
+    max_votes = categories_with_votes.values.max
+    self.preferred_categories = categories_with_votes
+      .select {|category, votes| votes == max_votes}
+      .keys
+    self.preferred_categories.map {|category| category.filter}.join(",")
+  end
+
+end

data/lib/yelp_client.rb

@@ -0,0 +1,18 @@
+class YelpClient
+  attr_accessor :client
+
+  def initialize
+    @client = Yelp::Client.new
+    client.configure do |config|
+      config.consumer_key = "iet5BTns6urx8_MPMSN9lg"
+      config.consumer_secret = "LUwtv6MjjikYbt7BrHUxJV6oJ0E"
+      config.token = "K3EIm60p2YqhX5H5DOyZoqfFJua3TGCS"
+      config.token_secret = "oVvc-K6pQR31nGvujojMjty07hY"
+    end
+  end
+
+  def search(location, params = {}, locale = {})
+    client.search(location, params, locale).businesses
+  end
+
+end

metadata

@@ -0,0 +1,53 @@
+--- !ruby/object:Gem::Specification
+name: whats-for-dinner
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Adam Fraser
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-01-09 00:00:00.000000000 Z
+dependencies: []
+description: A simple gem to help groups find restaurants that optimize their collective
+  preferences.
+email: adam.fraser@gmail.com
+executables:
+- whats-for-dinner
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/whats-for-dinner
+- lib/category.rb
+- lib/category_list.rb
+- lib/command_line_interface.rb
+- lib/restaurant.rb
+- lib/user.rb
+- lib/whats_for_dinner.rb
+- lib/yelp_client.rb
+homepage: https://github.com/adamgfraser/whats-for-dinner
+licenses:
+- None
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5.1
+signing_key: 
+specification_version: 4
+summary: What's For Dinner?
+test_files: []