Val & Carly - OO Ride Share - Edges

lib/driver.rb

@@ -0,0 +1,85 @@
+require 'csv'
+require 'time'
+require_relative "user"
+
+module RideShare
+
+  class Driver < User
+    attr_reader :vin, :driven_trips
+    attr_accessor :status
+
+    def initialize(input)
+      super(input)
+
+      @vin = input[:vin].to_s
+
+      if @vin.nil? || @vin == 0
+        raise ArgumentError, 'ID cannot be blank or less than zero.'
+      end
+
+
+      if input[:vin].length != 17 || input[:vin].length == 0
+        raise ArgumentError, 'vin numbers must be 17 characters in length'
+      end
+
+      @status = input[:status]
+
+      unless @status == :AVAILABLE
+        @status == :UNAVAILABLE
+
+      end
+
+
+      @driven_trips = input[:trips].nil? ? [] : input[:trips]
+    end
+
+    def add_driven_trip(driven_trip)
+      @driven_trips << driven_trip
+
+      unless driven_trip.is_a? Trip
+        raise ArgumentError
+      end
+
+    end
+
+
+    def average_rating
+      total_rating = 0
+
+      @driven_trips.each do |trip|
+        total_rating += trip.rating
+      end
+
+      if @driven_trips.length == 0
+        average = 0
+      else
+        average = total_rating.to_f/@driven_trips.length
+      end
+      return average
+    end
+
+
+    def total_revenue
+      total_revenue = 0
+
+      @driven_trips.each do |trip|
+        total_revenue += trip.cost
+      end
+
+      if @driven_trips.length == 0
+        total = 0
+      else
+        total = (total_revenue.to_f - 1.65)*0.8
+      end
+      return total
+    end
+
+
+    def net_expenditures
+      return super - total_revenue.round(2)
+
+    end
+
+
+  end
+end

lib/trip.rb

@@ -1,8 +1,9 @@
 require 'csv'
+require 'time'
 
 module RideShare
   class Trip
-    attr_reader :id, :passenger, :start_time, :end_time, :cost, :rating
+    attr_reader :id, :passenger, :start_time, :end_time, :cost, :rating, :driver
 
     def initialize(input)
       @id = input[:id]
@@ -11,10 +12,23 @@ def initialize(input)
       @end_time = input[:end_time]
       @cost = input[:cost]
       @rating = input[:rating]
+      @driver = input[:driver]
 
-      if @rating > 5 || @rating < 1
+      valid_ratings = [*1..5, nil]
+      unless valid_ratings.include? (rating)
         raise ArgumentError.new("Invalid rating #{@rating}")
       end
+
+      if end_time != nil
+        if @end_time < @start_time
+          raise ArgumentError.new
+        end
+      end
+    end
+
+    def calculate_duration
+      duration = @end_time - @start_time
+      return duration
     end
 
     def inspect

lib/trip_dispatcher.rb

@@ -1,16 +1,20 @@
 require 'csv'
 require 'time'
+require 'pry'
 
 require_relative 'user'
 require_relative 'trip'
+require_relative 'driver'
 
 module RideShare
   class TripDispatcher
     attr_reader :drivers, :passengers, :trips
 
     def initialize(user_file = 'support/users.csv',
-                   trip_file = 'support/trips.csv')
+      trip_file = 'support/trips.csv',
+      driver_file = 'support/drivers.csv')
       @passengers = load_users(user_file)
+      @drivers = load_drivers(driver_file)
       @trips = load_trips(trip_file)
     end
 
@@ -33,44 +37,106 @@ def load_users(filename)
     def load_trips(filename)
       trips = []
       trip_data = CSV.open(filename, 'r', headers: true,
-                                          header_converters: :symbol)
+        header_converters: :symbol)
+
+        trip_data.each do |raw_trip|
+          passenger = find_passenger(raw_trip[:passenger_id].to_i)
+          driver = find_driver(raw_trip[:driver_id].to_i)
+
+          parsed_trip = {
+            id: raw_trip[:id].to_i,
+            passenger: passenger,
+            #2018-06-07 04:18:47 -0700
+            start_time: Time.parse(raw_trip[:start_time]),
+            #,2018-06-07 04:19:25 -0700
+            end_time: Time.parse(raw_trip[:end_time]),
+            cost: raw_trip[:cost].to_f,
+            rating: raw_trip[:rating].to_i,
+            driver: driver
+          }
+
+          create_trip(parsed_trip, driver, passenger, trips)
+        end
+
+        return trips
+      end
 
-      trip_data.each do |raw_trip|
-        passenger = find_passenger(raw_trip[:passenger_id].to_i)
 
-        parsed_trip = {
-          id: raw_trip[:id].to_i,
-          passenger: passenger,
-          start_time: raw_trip[:start_time],
-          end_time: raw_trip[:end_time],
-          cost: raw_trip[:cost].to_f,
-          rating: raw_trip[:rating].to_i
-        }
+      def load_drivers(filename)
+        drivers = []
+        driver_data = CSV.open(filename, 'r', headers: true, header_converters: :symbol)
+
+        driver_data.each do |raw_driver|
+          user_with_same_id = find_passenger(raw_driver[0].to_i)
+
+          parsed_driver = {
+            id: raw_driver[0].to_i,
+            vin: raw_driver[1],
+            name: user_with_same_id.name,
+            phone_number: user_with_same_id.phone_number,
+            status: raw_driver[2].to_sym
+          }
+
+          driver = Driver.new(parsed_driver)
+          drivers << driver
+        end
+
+        return drivers
+      end
+
+
+      def find_passenger(id)
+        check_id(id)
+        return @passengers.find { |passenger| passenger.id == id }
+      end
+
+
+      def find_driver(id)
+        check_id(id)
+        return @drivers.find { |driver| driver.id == id }
+      end
 
-        trip = Trip.new(parsed_trip)
+      def create_trip(trip_data, driver, passenger, trip_list)
+        trip = Trip.new(trip_data)
         passenger.add_trip(trip)
-        trips << trip
+        driver.add_driven_trip(trip)
+        trip_list << trip
+        return trip
       end
 
-      return trips
-    end
 
-    def find_passenger(id)
-      check_id(id)
-      return @passengers.find { |passenger| passenger.id == id }
-    end
+      def request_trip(user_id)
+        available_drivers = @drivers.select { |driver| driver.status == :AVAILABLE}
 
-    def inspect
-      return "#<#{self.class.name}:0x#{self.object_id.to_s(16)} \
-              #{trips.count} trips, \
-              #{drivers.count} drivers, \
-              #{passengers.count} passengers>"
-    end
+        passenger = find_passenger(user_id)
+        driver = available_drivers.first
+
+        trip_data = {
+          start_time: Time.now,
+          end_time: nil,
+          passenger: passenger,
+          driver: driver,
+          cost: nil,
+          rating: nil
+        }
 
-    private
+        trip_data[:driver].status = :UNAVAILABLE
+        new_trip = create_trip(trip_data, driver, passenger, @trips)
 
-    def check_id(id)
-      raise ArgumentError, "ID cannot be blank or less than zero. (got #{id})" if id.nil? || id <= 0
+        return new_trip
+      end
+
+      def inspect
+        return "#<#{self.class.name}:0x#{self.object_id.to_s(16)} \
+        #{trips.count} trips, \
+        #{drivers.count} drivers, \
+        #{passengers.count} passengers>"
+      end
+
+      private
+
+      def check_id(id)
+        raise ArgumentError, "ID cannot be blank or less than zero. (got #{id})" if id.nil? || id <= 0
+      end
     end
   end
-end

lib/user.rb

@@ -16,5 +16,24 @@ def initialize(input)
     def add_trip(trip)
       @trips << trip
     end
+
+    def net_expenditures
+      # all_trips = @user.trips
+      array = []
+      @trips.each do |trip|
+        array << trip.cost
+      end
+      return array.sum
+
+    end
+
+    def total_time_spent
+      array = []
+      @trips.each do |trip|
+        array << trip.calculate_duration
+      end
+      return array.sum
+    end
+
   end
 end