Time - Lola - Yoyo

Gemfile

@@ -1,6 +1,6 @@
 source 'http://rubygems.org'
 
-ruby '2.5.5'
+ruby '2.6.3'
 
 gem 'rake'
 

Gemfile.lock

@@ -0,0 +1,34 @@
+GEM
+  remote: http://rubygems.org/
+  specs:
+    ansi (1.5.0)
+    awesome_print (1.8.0)
+    builder (3.2.4)
+    csv (3.1.2)
+    minitest (5.14.0)
+    minitest-reporters (1.4.2)
+      ansi
+      builder
+      minitest (>= 5.0)
+      ruby-progressbar
+    minitest-skip (0.0.1)
+      minitest (~> 5.0)
+    rake (13.0.1)
+    ruby-progressbar (1.10.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  awesome_print
+  csv
+  minitest
+  minitest-reporters
+  minitest-skip
+  rake
+
+RUBY VERSION
+   ruby 2.6.3p62
+
+BUNDLED WITH
+   2.1.4

lib/csv_record.rb

@@ -1,4 +1,4 @@
-require 'csv'
+require "csv"
 
 module RideShare
   class CsvRecord
@@ -8,30 +8,30 @@ def initialize(id)
       self.class.validate_id(id)
       @id = id
     end
-    
+
     # Takes either full_path or directory and optional file_name
     # Default file name matches class name
     def self.load_all(full_path: nil, directory: nil, file_name: nil)
       full_path ||= build_path(directory, file_name)
-
       return CSV.read(
-        full_path,
-        headers: true,
-        header_converters: :symbol,
-        converters: :numeric
-      ).map { |record| from_csv(record) }
+               full_path,
+               headers: true,
+               header_converters: :symbol,
+               converters: :numeric,
+             ).map { |record| from_csv(record) }
     end
 
     def self.validate_id(id)
       if id.nil? || id <= 0
-        raise ArgumentError, 'ID cannot be blank or less than one.'
+        raise ArgumentError, "ID cannot be blank or less than one."
       end
     end
 
     private
-
+
+    # Template method
     def self.from_csv(record)
-      raise NotImplementedError, 'Implement me in a child class!'
+      raise NotImplementedError, "Implement me in a child class!"
     end
 
     def self.build_path(directory, file_name)
@@ -40,10 +40,9 @@ def self.build_path(directory, file_name)
       end
 
       unless file_name
-        class_name = self.to_s.split('::').last
+        class_name = self.to_s.split("::").last
         file_name = "#{class_name.downcase}s.csv"
       end
-
       return "#{directory}/#{file_name}"
     end
   end

lib/driver.rb

@@ -0,0 +1,62 @@
+require_relative "csv_record"
+
+module RideShare
+  class Driver < CsvRecord
+    attr_reader :name, :vin, :trips
+    attr_accessor :status
+
+    def initialize(id:, name:, vin: nil, status: :AVAILABLE, trips: nil)
+      super(id)
+      @name = name
+      if vin.length != 17
+        raise ArgumentError, "vin must be 17 characters long"
+      end
+      @vin = vin
+      @status = status
+      @trips = trips || []
+    end
+
+    def add_trip(trip)
+      @trips << trip
+    end
+
+    def average_rating
+      rating_total = 0
+      if @trips.empty?
+        return 0
+      else
+        @trips.each do |trip|
+          rating_total += trip.rating
+        end
+        return rating_total / @trips.length.to_f.round(2)
+      end
+    end
+
+    def total_revenue
+      total_revenue = 0
+      if @trips.empty?
+        return 0
+      else
+        @trips.each do |trip|
+          if trip.cost >= 1.65
+            total_revenue += trip.cost * 0.8 - 1.65
+          else
+            total_revenue
+          end
+        end
+      end
+      return total_revenue
+    end
+
+    private
+
+    def self.from_csv(record)
+      return self.new(
+               id: record[:id],
+               name: record[:name],
+               vin: record[:vin],
+               status: record[:status].to_sym,
+             )
+    end
+  end
+end

lib/passenger.rb

@@ -1,12 +1,11 @@
-require_relative 'csv_record'
+require_relative "csv_record"
 
 module RideShare
   class Passenger < CsvRecord
     attr_reader :name, :phone_number, :trips
 
-    def initialize(id:, name:, phone_number:, trips: nil)
+    def initialize(id:, name:, phone_number:, trips: [])
       super(id)
-
       @name = name
       @phone_number = phone_number
       @trips = trips || []
@@ -16,14 +15,32 @@ def add_trip(trip)
       @trips << trip
     end
 
+    # Wave 1: method to return the total amount of money that passenger has spent on their trips + tests
+    def net_expenditures
+      total_cost = 0
+      @trips.each do |trip|
+        total_cost += trip.cost
+      end
+      return total_cost
+    end
+
+    # Wave 1: method to return the total amount of time that passenger has spent on their trips + tests
+    def total_time_spent
+      total_time = 0
+      @trips.each do |trip|
+        total_time += trip.calculate_duration
+      end
+      return total_time
+    end
+
     private
 
     def self.from_csv(record)
       return new(
-        id: record[:id],
-        name: record[:name],
-        phone_number: record[:phone_num]
-      )
+               id: record[:id],
+               name: record[:name],
+               phone_number: record[:phone_num],
+             )
     end
   end
 end

lib/trip.rb

@@ -1,46 +1,66 @@
-require 'csv'
+require "csv"
+require "time"
 
-require_relative 'csv_record'
+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, :driver_id
 
     def initialize(
-          id:,
-          passenger: nil,
-          passenger_id: nil,
-          start_time:,
-          end_time:,
-          cost: nil,
-          rating:
-        )
+      id:,
+      passenger: nil,
+      passenger_id: nil,
+      start_time:,
+      end_time:,
+      cost: nil,
+      rating:,
+      driver: nil,
+      driver_id: nil
+    )
       super(id)
 
       if passenger
         @passenger = passenger
         @passenger_id = passenger.id
-
       elsif passenger_id
         @passenger_id = passenger_id
-
       else
-        raise ArgumentError, 'Passenger or passenger_id is required'
+        raise ArgumentError, "Passenger or passenger_id is required"
+      end
+
+      # Raises an ArgumentError if the end time is before the start time
+      if end_time != nil
+        if end_time > start_time
+          @start_time = start_time
+          @end_time = end_time
+        else
+          raise ArgumentError, "Those are invalid times"
+        end
       end
 
-      @start_time = start_time
-      @end_time = end_time
       @cost = cost
       @rating = rating
 
-      if @rating > 5 || @rating < 1
-        raise ArgumentError.new("Invalid rating #{@rating}")
+      if rating != nil
+        if @rating > 5 || @rating < 1
+          raise ArgumentError.new("Invalid rating #{@rating}")
+        end
+      end
+
+      # Wave 2: When a Trip is constructed, either driver_id or driver must be provided.
+      if driver
+        @driver = driver
+        @driver_id = driver.id
+      elsif driver_id
+        @driver_id = driver_id
+      else
+        raise ArgumentError, "Driver or driver_id is required"
       end
     end
 
     def inspect
       # Prevent infinite loop when puts-ing a Trip
-      # trip contains a passenger contains a trip contains a passenger...
       "#<#{self.class.name}:0x#{self.object_id.to_s(16)} " +
         "ID=#{id.inspect} " +
         "PassengerID=#{passenger&.id.inspect}>"
@@ -51,16 +71,28 @@ def connect(passenger)
       passenger.add_trip(self)
     end
 
+    def connect_driver(driver)
+      @driver = driver
+      driver.add_trip(self)
+    end
+
+    # Wave 1: Instance method to calculate the duration of the trip in seconds
+    def calculate_duration
+      duration = @end_time - @start_time
+      return duration
+    end
+
     private
 
     def self.from_csv(record)
       return self.new(
                id: record[:id],
+               driver_id: record[:driver_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],
              )
     end
   end