Fix relation ids generation for composite primary keys

lib/tapioca/dsl/compilers/active_record_relations.rb

@@ -794,8 +794,7 @@ def create_common_methods
                   constant,
                   column_type_option: Tapioca::Dsl::Helpers::ActiveRecordColumnTypeHelper::ColumnTypeOption::Persisted,
                 )
-                primary_key = constant.primary_key
-                getter_type, _setter_type = column_type_helper.type_for(primary_key)
+                getter_type, _setter_type = column_type_helper.type_for("id")
                 type = getter_type
                 create_common_method("ids", return_type: "T::Array[#{type}]")
               else

spec/tapioca/dsl/compilers/active_record_relations_spec.rb

@@ -778,6 +778,54 @@ def not(opts, *rest); end
               assert_equal(expected, rbi_for(:Post))
             end
 
+            it "generates typed ids for a nonstandard scalar primary key" do
+              add_ruby_file("schema.rb", <<~RUBY)
+                ActiveRecord::Migration.suppress_messages do
+                  ActiveRecord::Schema.define do
+                    create_table :posts, id: :string, primary_key: :code
+                  end
+                end
+              RUBY
+
+              add_ruby_file("post.rb", <<~RUBY)
+                class Post < ActiveRecord::Base
+                end
+              RUBY
+
+              expected = indented(<<~RBI, 4)
+                sig { returns(T::Array[::String]) }
+                def ids; end
+              RBI
+
+              assert_includes(rbi_for(:Post), expected)
+            end
+
+            it "generates typed ids for composite primary keys" do
+              add_ruby_file("schema.rb", <<~RUBY)
+                ActiveRecord::Migration.suppress_messages do
+                  ActiveRecord::Schema.define do
+                    create_table :posts, primary_key: [:listing_id, :checkin_on, :checkout_on] do |t|
+                      t.string :listing_id, null: false
+                      t.date :checkin_on, null: false
+                      t.date :checkout_on, null: false
+                    end
+                  end
+                end
+              RUBY
+
+              add_ruby_file("post.rb", <<~RUBY)
+                class Post < ActiveRecord::Base
+                end
+              RUBY
+
+              expected = indented(<<~RBI, 4)
+                sig { returns(T::Array[[::String, ::Date, ::Date]]) }
+                def ids; end
+              RBI
+
+              assert_includes(rbi_for(:Post), expected)
+            end
+
             it "generates handles composite primary keys" do
               add_ruby_file("post.rb", <<~RUBY)
                 class Post < ActiveRecord::Base