Work on relationship to allow for to/from addresses on rides

app/models/address.rb

@@ -1,7 +1,9 @@
 # frozen_string_literal: true
 
 class Address < ApplicationRecord
-  belongs_to :owner, polymorphic: true, optional: false
+  has_many :driver_addresses, dependent: :destroy
+  has_one :current_driver_address, -> { where(current: true) }, class_name: "DriverAddress", dependent: :destroy
+  has_one :current_driver, through: :current_driver_address, source: :address, dependent: :destroy
 
-  validates :line_1, :city, :state, :zip_code, :latitude, :longitude, presence: true
+  validates :line_1, :city, :state, :zip_code, :place_id, :latitude, :longitude, presence: true
 end

app/models/driver.rb

@@ -1,7 +1,9 @@
 # frozen_string_literal: true
 
 class Driver < ApplicationRecord
-  has_one :address, as: :owner, dependent: :destroy
+  has_many :driver_addresses, dependent: :destroy
+  has_one :current_driver_address, -> { where(current: true) }, class_name: "DriverAddress", dependent: :destroy
+  has_one :current_address, through: :current_driver_address, source: :driver, dependent: :destroy
 
   validates :first_name, :last_name, presence: true
 end

app/models/driver_address.rb

@@ -0,0 +1,9 @@
+class DriverAddress < ApplicationRecord
+  belongs_to :driver
+  belongs_to :address
+
+  validates :current,
+    if: -> { current },
+    uniqueness: { scope: :driver_id, message: "can only have one current Driver" },
+    inclusion: { in: [true, false] }
+end

app/models/ride.rb

@@ -2,6 +2,8 @@
 
 class Ride < ApplicationRecord
   belongs_to :driver, optional: true
+  belongs_to :from_address, class_name: "Address"
+  belongs_to :to_address, class_name: "Address"
 
   validates :duration, :distance, :commute_duration, :amount_cents, presence: true
   monetize :amount_cents,

db/migrate/20240702182238_create_addresses.rb

@@ -11,7 +11,6 @@ def change
       t.float :latitude, null: false
       t.float :longitude, null: false
       t.string :place_id, null: true, index: true
-      t.references :owner, polymorphic: true, null: false, index: true
 
       t.timestamps
     end

db/migrate/20240702193305_create_rides.rb

@@ -6,6 +6,8 @@ def change
       t.float :commute_duration, index: true, null: false
       t.monetize :amount
       t.references :driver, null: true, index: true
+      t.references :from_address, foreign_key: {to_table: :addresses}, index: true, null: false
+      t.references :to_address, foreign_key: {to_table: :addresses}, index: true, null: false
       t.timestamps
     end
   end

db/migrate/20240702210002_create_driver_addresses.rb

@@ -0,0 +1,11 @@
+class CreateDriverAddresses < ActiveRecord::Migration[7.1]
+  def change
+    create_table :driver_addresses do |t|
+      t.boolean :current, null: false, default: false
+      t.references :driver, null: false, foreign_key: true
+      t.references :address, null: false, foreign_key: true
+
+      t.timestamps
+    end
+  end
+end

db/migrate/20240702210854_add_uniq_index_to_driver_addresses.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddUniqIndexToDriverAddresses < ActiveRecord::Migration[7.1]
+  def change
+    add_index :driver_addresses, %i[current driver_id], unique: true, where: "(current IS TRUE)"
+  end
+end

spec/factories/addresses.rb

@@ -10,6 +10,5 @@
     place_id { Faker::Internet.unique.device_token }
     latitude { Faker::Address.latitude }
     longitude { Faker::Address.longitude }
-    association :owner, factory: [:driver]
   end
 end

spec/factories/drivers.rb

@@ -4,6 +4,5 @@
   factory :driver do
     first_name { Faker::Name.first_name }
     last_name { Faker::Name.last_name }
-    association :address, factory: :address
   end
 end

spec/factories/rides.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :ride do
+    duration { 2.3 }
+    commute_duration { 1.0 }
+    distance { 30.1 }
+    amount_cents { 1200 }
+
+    driver { nil }
+
+    association :from_address, factory: :address
+    association :to_address, factory: :address
+  end
+end

spec/models/address_spec.rb

@@ -3,11 +3,17 @@
 require "rails_helper"
 
 RSpec.describe Address, type: :model do
-  it { is_expected.to belong_to(:owner) }
+  it { is_expected.to have_one(:current_driver) }
+
   it { is_expected.to validate_presence_of(:line_1) }
   it { is_expected.to validate_presence_of(:city) }
   it { is_expected.to validate_presence_of(:state) }
   it { is_expected.to validate_presence_of(:zip_code) }
   it { is_expected.to validate_presence_of(:latitude) }
   it { is_expected.to validate_presence_of(:longitude) }
+
+  describe "instance_methods" do
+    it "can query for associated rides" do
+    end
+  end
 end