feat: Add support for table alias when injecting ONLY clause (#45)

* feat: Add support for table alias when injecting ONLY clause

* chore: Bumped version to 0.15.1

* fix: Linting

lib/hoardable/arel_visitors.rb

@@ -40,10 +40,16 @@ def visit_Arel_Nodes_InnerJoin(o, collector)
     end
 
     private def hoardable_maybe_add_only(o, collector)
-      return unless o.left.instance_variable_get("@klass").in?(Hoardable::REGISTRY)
-      return if Hoardable.instance_variable_get("@at")
+      left = o.left
 
-      collector << "ONLY "
+      if left.is_a?(Arel::Nodes::TableAlias)
+        hoardable_maybe_add_only(left, collector)
+      else
+        return unless left.instance_variable_get("@klass").in?(Hoardable::REGISTRY)
+        return if Hoardable.instance_variable_get("@at")
+
+        collector << "ONLY "
+      end
     end
   end
 end

lib/hoardable/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Hoardable
-  VERSION = "0.15.0"
+  VERSION = "0.15.1"
 end

test/support/models.rb

@@ -84,6 +84,8 @@ class Like < ActiveRecord::Base
 class UserWithTrashedPosts < ActiveRecord::Base
   self.table_name = "users"
   has_many :posts, -> { include_versions }, foreign_key: "user_id"
+
+  has_one :bio, class_name: "Profile", foreign_key: "user_id"
 end
 
 class Current < ActiveSupport::CurrentAttributes

test/test_model.rb

@@ -663,4 +663,11 @@ def create_comments_and_destroy_post
     assert_empty user.posts
     assert_empty User.joins(:posts)
   end
+
+  test "applys ONLY clause on joined relationship with aliased name" do
+    assert_equal(
+      "SELECT \"users\".* FROM \"users\" INNER JOIN ONLY \"profiles\" \"bio\" ON \"bio\".\"user_id\" = \"users\".\"id\" WHERE \"bio\".\"id\" = 999",
+      UserWithTrashedPosts.joins(:bio).where(bio: { id: 999 }).to_sql
+    )
+  end
 end