add extended state machines compiler

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    boba (0.0.2)
+    boba (0.0.5)
       sorbet-static-and-runtime (~> 0.5)
       tapioca (~> 0.16.2)
 

History.md

@@ -1,5 +1,9 @@
 # Boba History
 
+## 0.0.5
+
+- Add extended state machines compiler to fix typing on state machines class methods
+
 ## 0.0.4
 
 - Add railtie to make `PrivateRelation` type available at runtime

Makefile

@@ -14,8 +14,7 @@ clean:
 docs:
 	bundle exec rake generate_dsl_documentation
 
-
 # release
-#   make clean && make build
+#   make clean && make docs && make build
 # 	gem push boba-[version].gem
 #		gh release create [version]

lib/boba/version.rb

@@ -2,5 +2,5 @@
 # frozen_string_literal: true
 
 module Boba
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

lib/tapioca/dsl/compilers/state_machines_extended.rb

@@ -0,0 +1,37 @@
+# typed: ignore
+# frozen_string_literal: true
+
+return unless defined?(Tapioca)
+
+require "tapioca/dsl/compilers/state_machines"
+
+module Tapioca
+  module Dsl
+    module Compilers
+      # `Tapioca::Dsl::Compilers::StateMachinesExtended` extends the default state machines compiler provided by Tapioca
+      # to allow for calling `with_state` and `without_state` on all Active Record relations. This is a temporary fix
+      # until a more durable solution can be found for this type of issue.
+      # See https://github.com/Shopify/tapioca/pull/1994#issuecomment-2302624697.
+      class StateMachinesExtended < ::Tapioca::Dsl::Compilers::StateMachines
+        ACTIVE_RECORD_RELATION_MODULE_NAMES = [
+          "GeneratedRelationMethods",
+          "GeneratedAssociationRelationMethods",
+        ].freeze
+
+        def decorate
+          return if constant.state_machines.empty?
+
+          super()
+
+          root.create_path(T.unsafe(constant)) do |klass|
+            class_module_name = "StateMachineClassHelperModule"
+
+            ACTIVE_RECORD_RELATION_MODULE_NAMES.each do |module_name|
+              klass.create_module(module_name).create_include(class_module_name)
+            end
+          end
+        end
+      end
+    end
+  end
+end

manual/compiler_statemachinesextended.md

@@ -0,0 +1,6 @@
+## StateMachinesExtended
+
+`Tapioca::Dsl::Compilers::StateMachinesExtended` extends the default state machines compiler provided by Tapioca
+to allow for calling `with_state` and `without_state` on all Active Record relations. This is a temporary fix
+until a more durable solution can be found for this type of issue.
+See https://github.com/Shopify/tapioca/pull/1994#issuecomment-2302624697.

manual/compilers.md

@@ -8,4 +8,5 @@ This list is an evergeen list of currently available compilers.
 * [ActiveRecordAssociationsPersisted](compiler_activerecordassociationspersisted.md)
 * [ActiveRecordColumnsPersisted](compiler_activerecordcolumnspersisted.md)
 * [MoneyRails](compiler_moneyrails.md)
+* [StateMachinesExtended](compiler_statemachinesextended.md)
 <!-- END_COMPILER_LIST -->