Water: Leah & Sophia

lib/driver.rb

@@ -0,0 +1,56 @@
+require_relative 'csv_record'
+
+module RideShare
+  class Driver < CsvRecord
+    attr_reader :id, :name, :vin, :trips
+    attr_accessor :status
+
+    def initialize(id:, name:, vin:, status: :AVAILABLE, trips: [])
+      super(id)
+
+      @name = name
+      @vin = vin
+      raise ArgumentError.new("Invalid VIN") if (vin.length) != 17
+
+      @status = status
+      raise ArgumentError.new("Invalid status") if ![:AVAILABLE, :UNAVAILABLE].include?(@status)
+
+      @trips = trips
+
+    end
+
+    def accept_new_trip(trip)
+      add_trip(trip)
+      @status = :UNAVAILABLE
+    end
+
+    def add_trip(trip)
+      @trips << trip
+    end
+
+    def average_rating
+      ratings_array = @trips.reject { |trip| trip.rating == nil }.map(&:rating)
+      return 0 if (@trips) == [] || @trips == nil
+      return (ratings_array.sum/ratings_array.length).to_f
+    end
+
+    def total_revenue
+      return 0 if (@trips) == [] || @trips == nil
+      trip_cost = @trips.map(&:cost)
+      return 0 if trip_cost.sum < 1.65
+      driver_revenue = (trip_cost.sum - 1.65) * 0.80
+      return driver_revenue.to_f
+    end
+
+    private
+
+    def self.from_csv(record)
+      return new(
+          id: record[:id],
+          name: record[:name],
+          vin: record[:vin],
+          status: record[:status].to_sym
+      )
+    end
+  end
+end

lib/main.rb

@@ -0,0 +1,26 @@
+require_relative 'passenger'
+require_relative 'trip'
+require_relative 'csv_record'
+require_relative 'driver'
+
+# passenger1 = Passenger.new(2, "Merten Noles", 8347387434, [1,20,54,2018-12-27 02:39:05 -0800,2018-12-27 03:38:08 -0800,10,4])
+#
+# passenger1_expenditures = passenger1.net_expenditures
+#
+# pp passenger1_expenditures
+#
+# arr = [5,6,1]
+#
+# arr_sum = arr.reduce(:+)
+#
+# pp arr_sum
+#
+# @trips =
+#
+# def net_expenditures
+#   array_trip_costs = @trips.map(&:cost)
+#   total_expenditure = array_trip_costs.reduce(:+)
+#   return total_expenditure
+# end
+
+driver_instance = Driver.new(1,"Paul Klee","WBS76FYD47DJF7206",AVAILABLE)

lib/passenger.rb

@@ -1,4 +1,5 @@
 require_relative 'csv_record'
+require_relative 'trip'
 
 module RideShare
   class Passenger < CsvRecord
@@ -16,6 +17,15 @@ def add_trip(trip)
       @trips << trip
     end
 
+    def net_expenditures
+      total_expenditure = @trips.sum(&:cost)
+      return total_expenditure
+    end
+
+    def total_time_spent
+      return @trips.sum(&:duration_trip)
+    end
+
     private
 
     def self.from_csv(record)

lib/trip.rb

@@ -1,19 +1,22 @@
 require 'csv'
+require 'time'
 
 require_relative 'csv_record'
 
 module RideShare
   class Trip < CsvRecord
-    attr_reader :id, :passenger, :passenger_id, :start_time, :end_time, :cost, :rating
+    attr_reader :id, :passenger, :passenger_id, :start_time, :end_time, :cost, :rating, :driver_id, :driver
 
     def initialize(
           id:,
           passenger: nil,
           passenger_id: nil,
           start_time:,
-          end_time:,
+          end_time: nil,
           cost: nil,
-          rating:
+          rating: nil,
+          driver_id: nil,
+          driver: nil
         )
       super(id)
 
@@ -32,10 +35,18 @@ def initialize(
       @end_time = end_time
       @cost = cost
       @rating = rating
+      @driver_id = driver_id
+      @driver = driver
 
-      if @rating > 5 || @rating < 1
+
+      if @rating != nil && (@rating > 5 || @rating < 1)
         raise ArgumentError.new("Invalid rating #{@rating}")
       end
+
+      if @end_time != nil && (@end_time < @start_time)
+        raise ArgumentError.new("Invalid Time")
+      end
+
     end
 
     def inspect
@@ -50,21 +61,31 @@ def inspect
         "rating=#{rating}>"
     end
 
-    def connect(passenger)
+    def connect(passenger, driver)
       @passenger = passenger
       passenger.add_trip(self)
+      @driver = driver
+      driver.add_trip(self)
     end
 
+    def duration_trip
+      duration = @end_time - @start_time
+      return duration
+    end
+
+
     private
 
     def self.from_csv(record)
       return self.new(
                id: record[:id],
                passenger_id: record[:passenger_id],
-               start_time: record[:start_time],
-               end_time: record[:end_time],
+               start_time: Time.parse(record[:start_time]),
+               end_time: Time.parse(record[:end_time]),
                cost: record[:cost],
-               rating: record[:rating]
+               rating: record[:rating],
+               driver_id: record[:driver_id],
+               driver: record[:driver]
              )
     end
   end

lib/trip_dispatcher.rb

@@ -3,6 +3,7 @@
 
 require_relative 'passenger'
 require_relative 'trip'
+require_relative 'driver'
 
 module RideShare
   class TripDispatcher
@@ -11,6 +12,7 @@ class TripDispatcher
     def initialize(directory: './support')
       @passengers = Passenger.load_all(directory: directory)
       @trips = Trip.load_all(directory: directory)
+      @drivers = Driver.load_all(directory: directory)
       connect_trips
     end
 
@@ -19,6 +21,45 @@ def find_passenger(id)
       return @passengers.find { |passenger| passenger.id == id }
     end
 
+    def find_driver(id)
+      Driver.validate_id(id)
+      return @drivers.find { |driver| driver.id == id }
+    end
+
+    def driver_assigned
+      driver = @drivers.find { |driver| driver.status == :AVAILABLE }
+      return driver
+    end
+
+    def request_trip(passenger_id)
+      driver = driver_assigned
+      return nil if driver == nil
+
+      start_time = Time.now
+      end_time = nil
+      passenger = find_passenger(passenger_id)
+      trip_data = {
+          id: @trips.length + 1,
+          passenger: passenger,
+          start_time: start_time,
+          end_time: end_time,
+          cost: nil,
+          rating: nil,
+          driver_id: driver.id,
+          driver: driver
+      }
+
+      new_trip_instance = RideShare::Trip.new(trip_data)
+
+
+      driver.accept_new_trip(new_trip_instance)
+      passenger.add_trip(new_trip_instance)
+      @trips << new_trip_instance
+
+      return new_trip_instance
+
+    end
+
     def inspect
       # Make puts output more useful
       return "#<#{self.class.name}:0x#{object_id.to_s(16)} \
@@ -30,9 +71,11 @@ def inspect
     private
 
     def connect_trips
+
       @trips.each do |trip|
         passenger = find_passenger(trip.passenger_id)
-        trip.connect(passenger)
+        driver = find_driver(trip.driver_id)
+        trip.connect(passenger, driver)
       end
 
       return trips