Michelle and Katrina - Edges - OO-Ride-Share

lib/driver.rb

@@ -0,0 +1,77 @@
+require_relative "user"
+
+
+module RideShare
+  class Driver < User
+    attr_reader :id, :name, :vehicle_id, :driven_trips, :phone_number
+    attr_accessor :status
+
+    def initialize(input)
+      super(input)
+
+      @vehicle_id = input[:vin]
+      @driven_trips = input[:trips].nil? ? [] : input[:trips]
+      @status = input[:status] ? input[:status] : :AVAILABLE
+
+
+      unless @vehicle_id.length == 17
+        raise ArgumentError, 'Vehicle ID must contain 17 characters'
+      end
+
+      unless @status == :AVAILABLE || @status == :UNAVAILABLE
+        raise ArgumentError, 'Not a valid status.'
+      end
+
+      if input[:id].nil? || input[:id] <= 0
+        raise ArgumentError, 'ID cannot be blank or less than zero.'
+      end
+    end
+
+    def average_rating()
+
+      sum = 0
+
+      if @driven_trips.empty?
+        return 0
+      else
+        trips_amount = @driven_trips.length
+
+        valid_trips = @driven_trips.reject { |trip| trip.rating.nil? }
+
+        return valid_trips.sum {|trip| trip.rating} / trips_amount.to_f
+      end
+    end
+
+    def add_driven_trip(trip)
+
+      unless trip.class == RideShare::Trip
+        raise ArgumentError, 'This is not a valid trip'
+      end
+
+      @driven_trips << trip
+    end
+
+
+    def total_revenue()
+      if @driven_trips.empty?
+        return 0
+      else
+        valid_trips = @driven_trips.reject { |trip| trip.cost.nil? }
+
+        revenue_made = valid_trips.sum {|trip| (trip.cost - 1.65) } * 0.80
+
+        return ("%.2f" % revenue_made).to_f
+      end
+    end
+
+
+    def net_expenditures()
+
+      total_driver_revenue = total_revenue()
+
+      if !(super.nil? && total_driver_revenue.nil?)
+	       return ("%.2f" % (super - total_driver_revenue)).to_f
+      end
+    end
+  end
+end

lib/trip.rb

@@ -1,19 +1,31 @@
 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]
       @passenger = input[:passenger]
       @start_time = input[:start_time]
       @end_time = input[:end_time]
       @cost = input[:cost]
       @rating = input[:rating]
+      @driver = input[:driver]
+
+      if @rating != nil
+        if @rating > 5 || @rating < 1
+          raise ArgumentError.new("Invalid rating #{@rating}")
+        end
+      end
+
 
-      if @rating > 5 || @rating < 1
-        raise ArgumentError.new("Invalid rating #{@rating}")
+      if @end_time != nil
+        if @end_time - @start_time < 0
+          raise ArgumentError.new("Trip end time is before start time.")
+        end
       end
     end
 
@@ -22,5 +34,11 @@ def inspect
       "ID=#{id.inspect} " +
       "PassengerID=#{passenger&.id.inspect}>"
     end
+
+    def calculate_trip_duration
+      if @end_time != nil
+        return @end_time - @start_time
+      end
+    end
   end
 end

lib/trip_dispatcher.rb

@@ -3,15 +3,19 @@
 
 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
 
     def load_users(filename)
@@ -37,29 +41,87 @@ def load_trips(filename)
 
       trip_data.each do |raw_trip|
         passenger = find_passenger(raw_trip[:passenger_id].to_i)
+        driver = find_driver(raw_trip[:driver_id].to_i)
+
+        start_time = Time.parse(raw_trip[:start_time])
+        end_time = Time.parse(raw_trip[:end_time])
 
         parsed_trip = {
           id: raw_trip[:id].to_i,
           passenger: passenger,
-          start_time: raw_trip[:start_time],
-          end_time: raw_trip[:end_time],
+          start_time: start_time,
+          end_time: end_time,
           cost: raw_trip[:cost].to_f,
-          rating: raw_trip[:rating].to_i
+          rating: raw_trip[:rating].to_i,
+          driver: driver
         }
 
         trip = Trip.new(parsed_trip)
+
         passenger.add_trip(trip)
+        driver.add_driven_trip(trip)
         trips << trip
       end
 
       return trips
     end
 
+    def load_drivers(filename)
+      drivers = []
+
+      driver_data = CSV.open(filename, "r", headers:true, header_converters: :symbol)
+
+      driver_data.each do |raw_driver|
+        user = find_passenger(raw_driver[:id].to_i)
+
+        parsed_driver = {
+          id: raw_driver[:id].to_i,
+          name: user.name,
+          phone: user.phone_number,
+          vin: raw_driver[:vin],
+          status: raw_driver[:status].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
+
+    def request_trip(user_id)
+      current_passenger = find_passenger(user_id)
+      available_driver = @drivers.find { |driver| driver.status == :AVAILABLE }
+
+      available_driver.status = :UNAVAILABLE
+
+      parsed_trip = {
+        id: (@trips.max_by {|trip| trip.id}).id + 1,
+        passenger: current_passenger,
+        start_time: Time.now,
+        end_time: nil,
+        cost: nil,
+        rating: nil,
+        driver: available_driver
+      }
+
+      trip_in_progress = Trip.new(parsed_trip)
+      current_passenger.add_trip(trip_in_progress)
+      available_driver.add_driven_trip(trip_in_progress)
+      @trips << trip_in_progress
+
+      return trip_in_progress
+    end
+
     def inspect
       return "#<#{self.class.name}:0x#{self.object_id.to_s(16)} \
               #{trips.count} trips, \
@@ -70,7 +132,9 @@ def inspect
     private
 
     def check_id(id)
+      id = id.to_i
       raise ArgumentError, "ID cannot be blank or less than zero. (got #{id})" if id.nil? || id <= 0
+      return id
     end
   end
 end

lib/user.rb

@@ -1,3 +1,7 @@
+require_relative "trip"
+require "time"
+
+
 module RideShare
   class User
     attr_reader :id, :name, :phone_number, :trips
@@ -16,5 +20,19 @@ def initialize(input)
     def add_trip(trip)
       @trips << trip
     end
+
+    def net_expenditures()
+      valid_trips = @trips.reject { |trip| trip.cost.nil? }
+      return valid_trips.sum { |trip| trip.cost} if valid_trips.length > 0
+    end
+
+    def total_time_spent()
+
+      valid_trips = @trips.reject { |trip| trip.calculate_trip_duration.nil? }
+
+      return valid_trips.sum {|trip| trip.calculate_trip_duration} if valid_trips.length > 0
+
+    end
+
   end
 end