tapioca 0.3.1 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 81caba98bdbde19ffbb78a1860f994a8b5c39a82dcd48832ce0daabcd9f6d77b
-  data.tar.gz: 803fe5c9b0ac3838f984455adf53fd2a470e13a877ea218d80bb74b2c309afbe
+  metadata.gz: f9b10c318332131210706b023ccbe3a63a5889153e3b365a9c454ee7311d2498
+  data.tar.gz: 6c5a5ef82c6a952b535f6f04b3c3127968bca4d8d101e321e13e8e5968ea3222
 SHA512:
-  metadata.gz: 75c2b49c410a6b9f558470a5ab6b59f464b41e4b03a88344fd8967d06e3b36b8b29669bd92c8103d106e234a91fb4ead51d617a365059342e03542cfe113c497
-  data.tar.gz: 4261f1657b92a65ba119f95e1e9be3211af056fe2ba5a5b225b0d920a2adda1fd7619d9e21bb8e034c3946b4cb931d8ef45beebd3bb429c2c5d8446bce09efa8
+  metadata.gz: f1df6d85dd2924d25e3d6460749cc5a5ce75022bf999e663c813d6dd483e130b6a1c9873872f2412046d6d15f87effc125bc2f3941323f0ac3eca27a0bfa89a7
+  data.tar.gz: f6b81c3d32e1cec689ed022c17f57ada8655c0594c944c396d26a481e32dce029b77ed8f877063bfee138405019f6bb5408f2de3148d437a2b8db01a699063c5

data/Gemfile

@@ -7,5 +7,29 @@ gemspec
 gem 'rubocop-shopify', require: false
 
 group(:deployment, :development) do
-  gem("rake", "~> 12.3")
+  gem("rake")
+end
+
+gem("bundler", "~> 1.17")
+gem("pry-byebug")
+gem("minitest")
+gem("minitest-hooks")
+gem("minitest-fork_executor")
+gem("minitest-reporters")
+gem("sorbet")
+
+group(:development, :test) do
+  gem("smart_properties", ">= 1.15.0", require: false)
+  gem("frozen_record", ">= 0.17", require: false)
+  gem("sprockets", "~> 3.7", require: false)
+  gem("rails", "~> 5.2", require: false)
+  gem("state_machines", "~> 0.5.0", require: false)
+  gem("activerecord-typedstore", "~> 1.3", require: false)
+  gem("sqlite3")
+  gem("identity_cache", "~> 1.0", require: false)
+  gem('cityhash', git: 'https://github.com/csfrancis/cityhash.git',
+                  ref: '3cfc7d01f333c01811d5e834f1495eaa29f87c36', require: false)
+  gem("activemodel-serializers-xml", "~> 1.0", require: false)
+  gem("activeresource", "~> 5.1", require: false)
+  gem("google-protobuf", "~>3.12.0", require: false)
 end

data/README.md

@@ -2,12 +2,14 @@
 
 [![Build Status](https://travis-ci.org/Shopify/tapioca.svg?branch=master)](https://travis-ci.org/Shopify/tapioca)
 
-Tapioca is a library used to generate RBI (Ruby interface) files for use with [Sorbet](https://sorbet.org). RBI files provide the structure (classes, modules, methods, parameters) of the gem/library to Sorbet to assist with typechecking. 
+Tapioca is a library used to generate RBI (Ruby interface) files for use with [Sorbet](https://sorbet.org). RBI files provide the structure (classes, modules, methods, parameters) of the gem/library to Sorbet to assist with typechecking.
 
 As yet, no gem exports type information in a consumable format and it would be a huge effort to manually maintain such an interface file for all the gems that your codebase depends on. Thus, there is a need for an automated way to generate the appropriate RBI file for a given gem. The `tapioca` gem, developed at Shopify, is able to do exactly that to almost 99% accuracy. It can generate the definitions for all statically defined types and most of the runtime defined types exported from Ruby gems (non-Ruby gems are not handled yet).
 
 When you run `tapioca sync` in a project, `tapioca` loads all the gems that are in your dependency list from the Gemfile into memory. It then performs runtime introspection on the loaded types to understand their structure and generates an appropriate RBI file for each gem with a versioned filename.
 
+## Manual gem requires
+
 For gems that have a normal default `require` and load all of their constants through such a require, everything works seamlessly. However, for gems that are marked as `require: false` in the Gemfile, or for gems that export optionally loaded types via different requires, where a single require does not load the whole gem code into memory, `tapioca` will not be able to load some of the types into memory and, thus, won't be able to generate complete RBIs for them. For this reason, we need to keep a small external file named `sorbet/tapioca/require.rb` that is executed after all the gems in the Gemfile have been required and before generation of gem RBIs have started. This file is responsible for adding the requires for additional files from gems, which are not covered by the default require.
 
 For example, suppose you are using the class `BetterHtml::Parser` exported from the `better_html` gem. Just doing a `require "better_html"` (which is the default require) does not load that type:
@@ -27,10 +29,17 @@ from (pry):3:in `__pry__`
 => BetterHtml::Parser
 ```
 
-In order to make sure that `tapioca` can reflect on that type, we need to add the line `require "better_html/parser"` to the `sorbet/tapioca/require.rb` file. This will make sure `BetterHtml::Parser` is loaded into memory and a type annotation is generated for it in the `better_html.rbi` file. If this extra `require` line is not added to `sorbet/tapioca/require.rb` file, then the definition for that type will be missing from the RBI file. 
+In order to make sure that `tapioca` can reflect on that type, we need to add the line `require "better_html/parser"` to the `sorbet/tapioca/require.rb` file. This will make sure `BetterHtml::Parser` is loaded into memory and a type annotation is generated for it in the `better_html.rbi` file. If this extra `require` line is not added to `sorbet/tapioca/require.rb` file, then the definition for that type will be missing from the RBI file.
 
 If you ever run into a case, where you add a gem or update the version of a gem and run `tapioca sync` but don't have some types you expect in the generated gem RBI files, you will need to make sure you have added the necessary requires to the `sorbet/tapioca/require.rb` file.
 
+You can use the command `tapioca require` to auto-populate the `sorbet/tapioca/require.rb` file with all the requires found
+in your application. Once the file generated, you should review it, remove all unnecessary requires and commit it.
+
+## How does tapioca compare to "srb rbi gems" ?
+
+[Please see the detailed answer on our wiki](https://github.com/Shopify/tapioca/wiki/How-does-tapioca-compare-to-%22srb-rbi-gems%22-%3F)
+
 ## Installation
 
 Add this line to your application's `Gemfile`:
@@ -49,7 +58,9 @@ Commands:
   tapioca generate [gem...]  # generate RBIs from gems
   tapioca help [COMMAND]     # Describe available commands or one specific command
   tapioca init               # initializes folder structure
+  tapioca require            # generate the list of files to be required by tapioca
   tapioca sync               # sync RBIs to Gemfile
+  tapioca todo               # generate the list of unresolved constants
 
 Options:
   --pre, -b, [--prerequire=file]              # A file to be required before Bundler.require is called
@@ -80,6 +91,12 @@ Command: `tapioca sync`
 
 This will sync the RBIs with the gems in the Gemfile and will add, update, and remove RBIs as necessary.
 
+### Generate the list of all unresolved constants
+
+Command: `tapioca todo`
+
+This will generate the file `sorbet/rbi/todo.rbi` defining all unresolved constants as empty modules.
+
 ### Flags
 
 - `--prerequire [file]`: A file to be required before `Bundler.require` is called.
@@ -88,6 +105,10 @@ This will sync the RBIs with the gems in the Gemfile and will add, update, and r
 - `--generate-command [command]`: The command to run to regenerate RBI files (used in header comment of the RBI files), defaults to the current command.
 - `--typed-overrides [gem:level]`: Overrides typed sigils for generated gem RBIs for gem `gem` to level `level` (`level` can be one of `ignore`, `false`, `true`, `strict`, or `strong`, see [the Sorbet docs](https://sorbet.org/docs/static#file-level-granularity-strictness-levels) for more details).
 
+### Strong typing option for ActiveRecord column methods
+
+`tapioca` gives you the option to generate stricter type signatures for your ActiveRecord column types. By default, methods generated for columns that are defined in the schema have signatures of T.untyped. However, if the object extends a module with name StrongTypeGeneration, tapioca will generate stricter signatures that follow closely with the types defined in the schema. Expectation is the StrongTypeGeneration module you define in your application won't allow objects to be initialized with "bad state". It will check all the attributes that are not nillable to ensure they are not nil.
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/Shopify/tapioca. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](https://github.com/Shopify/tapioca/blob/master/CODE_OF_CONDUCT.md) code of conduct.

data/Rakefile

@@ -3,10 +3,21 @@
 require "bundler/gem_tasks"
 require "rake/testtask"
 
-begin
-  require 'rspec/core/rake_task'
-  RSpec::Core::RakeTask.new(:spec)
-rescue LoadError # rubocop:disable Lint/SuppressedException
+Rake.application.options.trace = false
+
+Rake::TestTask.new do |t|
+  t.libs << "lib"
+  t.libs << "spec"
+  t.warning = false
+  t.test_files = FileList['spec/**/*_spec.rb']
+end
+
+task(:spec) do
+  begin
+    Rake::Task[:test].execute
+  rescue RuntimeError
+    exit(1)
+  end
 end
 
 task(default: :spec)

data/lib/tapioca.rb

@@ -4,10 +4,12 @@
 require "sorbet-runtime"
 
 module Tapioca
-  def self.silence_warnings
+  def self.silence_warnings(&blk)
     original_verbosity = $VERBOSE
     $VERBOSE = nil
-    yield
+    Gem::DefaultUserInteraction.use_ui(Gem::SilentUI.new) do
+      blk.call
+    end
   ensure
     $VERBOSE = original_verbosity
   end
@@ -35,7 +37,11 @@ require "tapioca/config_builder"
 require "tapioca/generator"
 require "tapioca/cli"
 require "tapioca/gemfile"
+require "tapioca/compilers/sorbet"
+require "tapioca/compilers/requires_compiler"
 require "tapioca/compilers/symbol_table_compiler"
 require "tapioca/compilers/symbol_table/symbol_generator"
 require "tapioca/compilers/symbol_table/symbol_loader"
+require "tapioca/compilers/todos_compiler"
+require "tapioca/compilers/dsl_compiler"
 require "tapioca/version"

data/lib/tapioca/cli.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
-# typed: false
+# typed: true
 
 require 'thor'
 
@@ -44,14 +44,40 @@ module Tapioca
       end
       create_file(Config::DEFAULT_POSTREQUIRE, skip: true) do
         <<~CONTENT
-          # frozen_string_literal: true
           # typed: false
+          # frozen_string_literal: true
 
           # Add your extra requires here
         CONTENT
       end
     end
 
+    desc "require", "generate the list of files to be required by tapioca"
+    def require
+      Tapioca.silence_warnings do
+        generator.build_requires
+      end
+    end
+
+    desc "todo", "generate the list of unresolved constants"
+    def todo
+      Tapioca.silence_warnings do
+        generator.build_todos
+      end
+    end
+
+    desc "dsl [constant...]", "generate RBIs for dynamic methods"
+    option :generators,
+      type: :array,
+      aliases: ["--gen", "-g"],
+      banner: "generator [generator ...]",
+      desc: "Only run supplied DSL generators"
+    def dsl(*constants)
+      Tapioca.silence_warnings do
+        generator.build_dsl(constants)
+      end
+    end
+
     desc "generate [gem...]", "generate RBIs from gems"
     def generate(*gems)
       Tapioca.silence_warnings do
@@ -72,7 +98,10 @@ module Tapioca
       end
 
       def generator
-        @generator ||= Generator.new(ConfigBuilder.from_options(options))
+        current_command = T.must(current_command_chain.first)
+        @generator ||= Generator.new(
+          ConfigBuilder.from_options(current_command, options)
+        )
       end
     end
   end

data/lib/tapioca/compilers/dsl/action_controller_helpers.rb

@@ -0,0 +1,129 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "parlour"
+
+begin
+  require "action_controller"
+rescue LoadError
+  return
+end
+
+module Tapioca
+  module Compilers
+    module Dsl
+      # `Tapioca::Compilers::Dsl::ActionControllerHelpers` decorates RBI files for all
+      # subclasses of `::ActionController::Base`
+      # to add helper methods (see https://api.rubyonrails.org/classes/ActionController/Helpers.html).
+      #
+      # For example, with the following `MyHelper` module:
+      #
+      # ~~~rb
+      # module MyHelper
+      #   def greet(user)
+      #     # ...
+      #   end
+      #
+      #  def localized_time
+      #     # ...
+      #   end
+      # end
+      # ~~~
+      #
+      # and the following controller:
+      #
+      # ~~~rb
+      # class UserController < ActionController::Base
+      #   helper MyHelper
+      #   helper { def age(user) "99" end }
+      #   helper_method :current_user_name
+      #
+      #   def current_user_name
+      #     # ...
+      #   end
+      # end
+      #
+      # this generator will produce an RBI file `user_controller.rbi` with the following content:
+      #
+      # ~~~rbi
+      # # user_controller.rbi
+      # # typed: strong
+      # class UserController
+      #   sig { returns(UserController::HelperProxy) }
+      #   def helpers; end
+      # end
+      #
+      # module UserController::HelperMethods
+      #    include MyHelper
+      #
+      #    sig { params(user: T.untyped).returns(T.untyped) }
+      #    def age(user); end
+      #
+      #    sig { returns(T.untyped) }
+      #    def current_user_name; end
+      #  end
+      #
+      # class UserController::HelperProxy < ::ActionView::Base
+      #   include UserController::HelperMethods
+      # end
+      # ~~~
+      class ActionControllerHelpers < Base
+        extend T::Sig
+
+        sig do
+          override
+            .params(root: Parlour::RbiGenerator::Namespace, constant: T.class_of(::ActionController::Base))
+            .void
+        end
+        def decorate(root, constant)
+          helper_proxy_name = "#{constant}::HelperProxy"
+          helper_methods_name = "#{constant}::HelperMethods"
+          proxied_helper_methods = constant._helper_methods.map(&:to_s).map(&:to_sym)
+
+          # Create helper method module
+          root.create_module(helper_methods_name) do |helper_methods|
+            helpers_module = constant._helpers
+
+            gather_includes(helpers_module).each do |ancestor|
+              helper_methods.create_include(ancestor)
+            end
+
+            helpers_module.instance_methods(false).each do |method_name|
+              method = if proxied_helper_methods.include?(method_name)
+                constant.instance_method(method_name)
+              else
+                helpers_module.instance_method(method_name)
+              end
+              create_method_from_def(helper_methods, method)
+            end
+          end
+
+          # Create helper proxy class
+          root.create_class(helper_proxy_name, superclass: "::ActionView::Base") do |proxy|
+            proxy.create_include(helper_methods_name)
+          end
+
+          # Define the helpers method
+          root.path(constant) do |controller|
+            create_method(controller, 'helpers', return_type: helper_proxy_name)
+          end
+        end
+
+        sig { override.returns(T::Enumerable[Module]) }
+        def gather_constants
+          ::ActionController::Base.descendants.reject(&:abstract?)
+        end
+
+        private
+
+        sig { params(mod: Module).returns(T::Array[String]) }
+        def gather_includes(mod)
+          mod.ancestors
+            .reject { |ancestor| ancestor.is_a?(Class) || ancestor == mod || ancestor.name.nil? }
+            .map { |ancestor| T.must(ancestor.name) }
+            .reverse
+        end
+      end
+    end
+  end
+end

data/lib/tapioca/compilers/dsl/action_mailer.rb

@@ -0,0 +1,65 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "parlour"
+
+begin
+  require "action_mailer"
+rescue LoadError
+  return
+end
+
+module Tapioca
+  module Compilers
+    module Dsl
+      # `Tapioca::Compilers::Dsl::ActionMailer` generates RBI files for subclasses of `ActionMailer::Base`
+      # (see https://api.rubyonrails.org/classes/ActionMailer/Base.html).
+      #
+      # For example, with the following `ActionMailer` subclass:
+      #
+      # ~~~rb
+      # class NotifierMailer < ActionMailer::Base
+      #   def notify_customer(customer_id)
+      #     # ...
+      #   end
+      # end
+      # ~~~
+      #
+      # this generator will produce the RBI file `notifier_mailer.rbi` with the following content:
+      #
+      # ~~~rbi
+      # # notifier_mailer.rbi
+      # # typed: true
+      # class NotifierMailer
+      #   sig { params(customer_id: T.untyped).returns(::ActionMailer::MessageDelivery) }
+      #   def self.notify_customer(customer_id); end
+      # end
+      # ~~~
+      class ActionMailer < Base
+        extend T::Sig
+
+        sig { override.params(root: Parlour::RbiGenerator::Namespace, constant: T.class_of(::ActionMailer::Base)).void }
+        def decorate(root, constant)
+          root.path(constant) do |k|
+            constant.action_methods.to_a.each do |mailer_method|
+              method_def = constant.instance_method(mailer_method)
+              parameters = compile_method_parameters_to_parlour(method_def)
+              create_method(
+                k,
+                mailer_method,
+                parameters: parameters,
+                return_type: '::ActionMailer::MessageDelivery',
+                class_method: true
+              )
+            end
+          end
+        end
+
+        sig { override.returns(T::Enumerable[Module]) }
+        def gather_constants
+          ::ActionMailer::Base.descendants.reject(&:abstract?)
+        end
+      end
+    end
+  end
+end

data/lib/tapioca/compilers/dsl/active_record_associations.rb

@@ -0,0 +1,267 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "parlour"
+
+begin
+  require "active_record"
+rescue LoadError
+  return
+end
+
+module Tapioca
+  module Compilers
+    module Dsl
+      # `Tapioca::Compilers::Dsl::ActiveRecordAssociations` refines RBI files for subclasses of `ActiveRecord::Base`
+      # (see https://api.rubyonrails.org/classes/ActiveRecord/Base.html). This generator is only
+      # responsible for defining the methods that would be created for the association that
+      # are defined in the Active Record model.
+      #
+      # For example, with the following model class:
+      #
+      # ~~~rb
+      # class Post < ActiveRecord::Base
+      #   belongs_to :category
+      #   has_many :comments
+      #   has_one :author, class_name: "User"
+      # end
+      # ~~~
+      #
+      # this generator will produce the following methods in the RBI file
+      # `post.rbi`:
+      #
+      # ~~~rbi
+      # # post.rbi
+      # # typed: true
+      #
+      # class Post
+      #   include Post::GeneratedAssociationMethods
+      # end
+      #
+      # module Post::GeneratedAssociationMethods
+      #   sig { returns(T.nilable(::User)) }
+      #   def author; end
+      #
+      #   sig { params(value: T.nilable(::User)).void }
+      #   def author=(value); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::User)) }
+      #   def build_author(*args, &blk); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::Category)) }
+      #   def build_category(*args, &blk); end
+      #
+      #   sig { returns(T.nilable(::Category)) }
+      #   def category; end
+      #
+      #   sig { params(value: T.nilable(::Category)).void }
+      #   def category=(value); end
+      #
+      #   sig { returns(T::Array[T.untyped]) }
+      #   def comment_ids; end
+      #
+      #   sig { params(ids: T::Array[T.untyped]).returns(T::Array[T.untyped]) }
+      #   def comment_ids=(ids); end
+      #
+      #   sig { returns(::ActiveRecord::Associations::CollectionProxy[Comment]) }
+      #   def comments; end
+      #
+      #   sig { params(value: T::Enumerable[::Comment]).void }
+      #   def comments=(value); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::User)) }
+      #   def create_author(*args, &blk); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::User)) }
+      #   def create_author!(*args, &blk); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::Category)) }
+      #   def create_category(*args, &blk); end
+      #
+      #   sig { params(args: T.untyped, blk: T.untyped).returns(T.nilable(::Category)) }
+      #   def create_category!(*args, &blk); end
+      #
+      #   sig { returns(T.nilable(::User)) }
+      #   def reload_author; end
+      #
+      #   sig { returns(T.nilable(::Category)) }
+      #   def reload_category; end
+      # end
+      # ~~~
+      class ActiveRecordAssociations < Base
+        extend T::Sig
+
+        ReflectionType = T.type_alias do
+          T.any(::ActiveRecord::Reflection::ThroughReflection, ::ActiveRecord::Reflection::AssociationReflection)
+        end
+
+        sig { override.params(root: Parlour::RbiGenerator::Namespace, constant: T.class_of(ActiveRecord::Base)).void }
+        def decorate(root, constant)
+          return if constant.reflections.empty?
+
+          module_name = "#{constant}::GeneratedAssociationMethods"
+          root.create_module(module_name) do |mod|
+            constant.reflections.each do |association_name, reflection|
+              if reflection.collection?
+                populate_collection_assoc_getter_setter(mod, constant, association_name, reflection)
+              else
+                populate_single_assoc_getter_setter(mod, constant, association_name, reflection)
+              end
+            end
+          end
+
+          root.path(constant) do |klass|
+            klass.create_include(module_name)
+          end
+        end
+
+        sig { override.returns(T::Enumerable[Module]) }
+        def gather_constants
+          ActiveRecord::Base.descendants.reject(&:abstract_class?)
+        end
+
+        private
+
+        sig do
+          params(
+            klass: Parlour::RbiGenerator::Namespace,
+            constant: T.class_of(ActiveRecord::Base),
+            association_name: T.any(String, Symbol),
+            reflection: ReflectionType
+          ).void
+        end
+        def populate_single_assoc_getter_setter(klass, constant, association_name, reflection)
+          association_class = type_for(constant, reflection)
+          association_type = "T.nilable(#{association_class})"
+
+          create_method(
+            klass,
+            association_name.to_s,
+            return_type: association_type,
+          )
+          create_method(
+            klass,
+            "#{association_name}=",
+            parameters: [
+              Parlour::RbiGenerator::Parameter.new("value", type: association_type),
+            ],
+            return_type: nil
+          )
+          create_method(
+            klass,
+            "reload_#{association_name}",
+            return_type: association_type,
+          )
+          if reflection.constructable?
+            create_method(
+              klass,
+              "build_#{association_name}",
+              parameters: [
+                Parlour::RbiGenerator::Parameter.new("*args", type: "T.untyped"),
+                Parlour::RbiGenerator::Parameter.new("&blk", type: "T.untyped"),
+              ],
+              return_type: association_type
+            )
+            create_method(
+              klass,
+              "create_#{association_name}",
+              parameters: [
+                Parlour::RbiGenerator::Parameter.new("*args", type: "T.untyped"),
+                Parlour::RbiGenerator::Parameter.new("&blk", type: "T.untyped"),
+              ],
+              return_type: association_type
+            )
+            create_method(
+              klass,
+              "create_#{association_name}!",
+              parameters: [
+                Parlour::RbiGenerator::Parameter.new("*args", type: "T.untyped"),
+                Parlour::RbiGenerator::Parameter.new("&blk", type: "T.untyped"),
+              ],
+              return_type: association_type
+            )
+          end
+        end
+
+        sig do
+          params(
+            klass: Parlour::RbiGenerator::Namespace,
+            constant: T.class_of(ActiveRecord::Base),
+            association_name: T.any(String, Symbol),
+            reflection: ReflectionType
+          ).void
+        end
+        def populate_collection_assoc_getter_setter(klass, constant, association_name, reflection)
+          association_class = type_for(constant, reflection)
+          relation_class = relation_type_for(constant, reflection)
+
+          create_method(
+            klass,
+            association_name.to_s,
+            return_type: relation_class,
+          )
+          create_method(
+            klass,
+            "#{association_name}=",
+            parameters: [
+              Parlour::RbiGenerator::Parameter.new("value", type: "T::Enumerable[#{association_class}]"),
+            ],
+            return_type: nil,
+          )
+          create_method(
+            klass,
+            "#{association_name.to_s.singularize}_ids",
+            return_type: "T::Array[T.untyped]"
+          )
+          create_method(
+            klass,
+            "#{association_name.to_s.singularize}_ids=",
+            parameters: [
+              Parlour::RbiGenerator::Parameter.new("ids", type: "T::Array[T.untyped]"),
+            ],
+            return_type: "T::Array[T.untyped]"
+          )
+        end
+
+        sig do
+          params(
+            constant: T.class_of(ActiveRecord::Base),
+            reflection: ReflectionType
+          ).returns(String)
+        end
+        def type_for(constant, reflection)
+          return "T.untyped" if !constant.table_exists? || polymorphic_association?(reflection)
+
+          "::#{reflection.klass.name}"
+        end
+
+        sig do
+          params(
+            constant: T.class_of(ActiveRecord::Base),
+            reflection: ReflectionType
+          ).returns(String)
+        end
+        def relation_type_for(constant, reflection)
+          "ActiveRecord::Associations::CollectionProxy" if !constant.table_exists? ||
+                                                            polymorphic_association?(reflection)
+
+          # Change to: "::#{reflection.klass.name}::ActiveRecord_Associations_CollectionProxy"
+          "::ActiveRecord::Associations::CollectionProxy[#{reflection.klass.name}]"
+        end
+
+        sig do
+          params(
+            reflection: ReflectionType
+          ).returns(T::Boolean)
+        end
+        def polymorphic_association?(reflection)
+          if reflection.through_reflection?
+            polymorphic_association?(reflection.source_reflection)
+          else
+            !!reflection.polymorphic?
+          end
+        end
+      end
+    end
+  end
+end