List available items and pack sizes. Move shop methods out into their own class

Author: SelenaSmall

app.rb

@@ -7,7 +7,7 @@
 
 # Read user input
 loop do
-  $stdout.print "Main menu: LIST available products, SHOP, VIEW basket, EXIT without placing an order? \n"
+  $stdout.print "Main menu: LIST available products, SHOP (add items to your order), VIEW basket, EXIT \n"
 
   input = gets.chomp
 

lib/handle_input.rb

@@ -1,17 +1,21 @@
 require_relative 'item'
 require_relative 'invoice'
+require_relative 'list'
+require_relative 'shop'
 require_relative 'order_line'
 
 # HandleInput class
 class HandleInput
-  attr_reader :order, :invoice
+  attr_reader :order, :invoice, :list, :shop
 
   # Actions
   ACTIONS = %w[LIST SHOP VIEW].freeze
 
   def initialize(order)
     @order = order
     @invoice = Invoice.new
+    @list = List.new
+    @shop = Shop.new(order)
   end
 
   # Interpret method
@@ -20,57 +24,10 @@ def initialize(order)
   def interpret(command)
     return unless ACTIONS.detect { |a| a == command }
 
-    if command.match?('LIST')
-      # list all availble items and package sizes
-    end
+    return list.selection if command.match?('LIST')
 
-    return shop_menu if command.match?('SHOP')
+    return shop.menu if command.match?('SHOP')
 
     return invoice.print_order(order) if command.match?('VIEW')
   end
-
-  private
-
-  def shop_menu
-    $stdout.print "Type BACK at any time to return to the main menu. \nAdd qty and items to backet, example input: 3 watermelons \n"
-
-    loop do
-      input = $stdin.gets.chomp
-
-      next if input.empty?
-
-      break if 'BACK'.match?(input)
-
-      shop(input)
-      next
-    end
-  end
-
-  # Order input pattern
-  PATTERN = /^\s*\d+\s*(watermelons||pineapples||rockmelons)$/
-
-  # Order items
-  def shop(input)
-    unless input.match?(PATTERN)
-      puts "That's not a valid input"
-      return
-    end
-
-    # Break input down into qty, item
-    line = input.split(/\W+/)
-
-    exec(OrderLine.new(line[0].to_i, Item.new(line[1])))
-  end
-
-  # Add item order_line to order
-  def exec(order_line)
-    order.add_item(
-      order_line,
-      order_line.present_line(
-        order_line.optimal(order_line.order_item.name)
-      )
-    )
-
-    order_line
-  end
 end

lib/invoice.rb

@@ -1,11 +1,5 @@
 # Invoice class
 class Invoice
-  attr_reader :printorder
-
-  def initialize
-    @printorder = []
-  end
-
   def print_order(order)
     puts "Customer Invoice \n"
     puts "Items rounded up to nearest available qty\n"

lib/list.rb

@@ -0,0 +1,24 @@
+# List class
+class List
+  def selection
+    puts "Products Available \n"
+
+    puts 'Watermelons'
+    Item.new('watermelons').packs('watermelons').each do |i|
+      puts "#{i.qty} pack @ #{i.price}"
+    end
+    puts "---------------\n"
+
+    puts 'Pineapples'
+    Item.new('pineapples').packs('pineapples').each do |i|
+      puts "#{i.qty} pack @ #{i.price}"
+    end
+    puts "---------------\n"
+
+    puts 'Rockmelons'
+    Item.new('rockmelons').packs('rockmelons').each do |i|
+      puts "#{i.qty} pack @ #{i.price}"
+    end
+    puts "---------------\n"
+  end
+end

lib/shop.rb

@@ -0,0 +1,48 @@
+# Shop class
+class Shop
+  attr_reader :order
+  # Order input pattern
+  PATTERN = /^\s*\d+\s*(watermelons||pineapples||rockmelons)$/
+
+  def initialize(order)
+    @order = order
+  end
+
+  def menu
+    $stdout.print "Type BACK at any time to return to the main menu. \nAdd qty and items to backet, example input: 3 watermelons \n"
+
+    loop do
+      input = $stdin.gets.chomp
+
+      next if input.empty?
+
+      break if 'BACK'.match?(input)
+
+      choose_items(input)
+      next
+    end
+  end
+
+  def choose_items(input)
+    unless input.match?(PATTERN)
+      puts "That's not a valid input"
+      return
+    end
+
+    # Break input down into qty, item
+    line = input.split(/\W+/)
+
+    exec(OrderLine.new(line[0].to_i, Item.new(line[1])))
+  end
+
+  def exec(order_line)
+    order.add_item(
+      order_line,
+      order_line.present_line(
+        order_line.optimal(order_line.order_item.name)
+      )
+    )
+
+    order_line
+  end
+end

spec/handle_input_spec.rb

@@ -41,43 +41,4 @@
       expect(instance.interpret('SHOP')).to be_nil
     end
   end
-
-  describe '#shop_menu' do
-    before do
-      $stdin = StringIO.new('3 watermelons')
-      $stdin = StringIO.new('BACK')
-    end
-
-    it 'should return nil' do
-      instance = HandleInput.new(Order.new)
-
-      expect(instance.send(:shop_menu)).to be_nil
-    end
-  end
-
-  describe '#shop' do
-    before do
-      $stdin = StringIO.new('3 watermelons')
-    end
-
-    it 'should return an instance of OrderLine' do
-      instance = HandleInput.new(Order.new)
-
-      expect(instance.send(:shop, '10 watermelons')).to be_a OrderLine
-    end
-
-    it 'should return nil if input does not match PATTERN' do
-      instance = HandleInput.new(Order.new)
-
-      expect(instance.send(:shop, 'fruit')).to be_nil
-    end
-  end
-
-  describe '#exec' do
-    it 'should return an instance of OrderLine' do
-      instance = HandleInput.new(Order.new)
-
-      expect(instance.send(:exec, OrderLine.new(10, Item.new('watermelons')))).to be_a OrderLine
-    end
-  end
 end

spec/invoice_spec.rb

@@ -3,14 +3,6 @@
 require './lib/invoice'
 
 describe Invoice do
-  describe '#initialize' do
-    it 'should have a printorder attribute which is an Array' do
-      instance = Invoice.new
-
-      expect(instance.printorder).to be_a Array
-    end
-  end
-
   describe '#print_order' do
     it 'should return nil' do
       instance = Invoice.new

spec/list_spec.rb

@@ -0,0 +1,13 @@
+require 'rspec'
+require 'spec_helper'
+require './lib/list'
+
+describe List do
+  describe '#selection' do
+    it 'should return nil' do
+      instance = List.new
+
+      expect(instance.selection).to be_nil
+    end
+  end
+end

spec/shop_spec.rb

@@ -0,0 +1,52 @@
+require 'rspec'
+require 'spec_helper'
+require './lib/shop'
+
+describe Shop do
+  describe '#initialize' do
+    it 'should have an order attribute which is an instance of Order' do
+      instance = Shop.new(Order.new)
+
+      expect(instance.order).to be_a Order
+    end
+  end
+
+  describe '#menu' do
+    before do
+      $stdin = StringIO.new('3 watermelons')
+      $stdin = StringIO.new('BACK')
+    end
+
+    it 'should return nil' do
+      instance = Shop.new(Order.new)
+
+      expect(instance.send(:menu)).to be_nil
+    end
+  end
+
+  describe '#choose_items' do
+    before do
+      $stdin = StringIO.new('3 watermelons')
+    end
+
+    it 'should return an instance of OrderLine' do
+      instance = Shop.new(Order.new)
+
+      expect(instance.send(:choose_items, '10 watermelons')).to be_a OrderLine
+    end
+
+    it 'should return nil if input does not match PATTERN' do
+      instance = Shop.new(Order.new)
+
+      expect(instance.send(:choose_items, 'fruit')).to be_nil
+    end
+  end
+
+  describe '#exec' do
+    it 'should return an instance of OrderLine' do
+      instance = Shop.new(Order.new)
+
+      expect(instance.send(:exec, OrderLine.new(10, Item.new('watermelons')))).to be_a OrderLine
+    end
+  end
+end