boba 0.0.11 → 0.0.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0a299592dce0ee68ddd4ad96f0c6d35a665739102bfa02c8e5d3f3cf26ed0531
-  data.tar.gz: 71d4ebe6c98f11586fef81e8d7b8892adce83bb2fbeafa516369947cbcd0dd9e
+  metadata.gz: 0c88ef5656da55d5a864ad1bf3b5231907260c62f2239964f47921368dbffea8
+  data.tar.gz: c32f3fe6f3063407146db5559cee2d6714474e9f69eb769ce39278c961908537
 SHA512:
-  metadata.gz: b34ab72d0801c768da549880ad247e978538e29197c51ff09b8d0e3c6d18b23e4433a808948613b98f3621ed0f486cd5fc7f44dad2f0e4587aec1070bf14cb0a
-  data.tar.gz: 1fca80c8c8a75f2763e4682b4b3e5543af3700abe8b2bd3c0fcb097c749d9292b67375c4121fc0ec43c49502678e0c8ee5fd3da59f02ce930a39ec5a1971f302
+  metadata.gz: ba9b4ccc7ffb27011848c62428a8cbed4cf9556bc3b63c04a4db867ed08ca12c2f16f48bc9f4c1f0ade04bd769d5033a977bc1e34f85115995c31dcdd65d21e0
+  data.tar.gz: dfe9d28722a6191722271c44b64ba2aec62eca438534b9617b2806d460c35c171635a662d6a6033d96220de6f3ff40cf0d4dc5e4122643ecad4fe3b6b6ac9b52

data/README.md

@@ -1,6 +1,6 @@
 # Boba
 
-> :warning: This gem is in pre-release and is not ready for use.
+> :warning: This software is currently under active development. It should not be considered stable until 1.0.0.
 
 Boba is a collection of compilers for Sorbet & Tapioca.
 
@@ -36,7 +36,8 @@ If you'd like to use relation types in your sigs that are less broad than `Activ
   gem 'boba'
 ```
 
-The railtie will automatically define the `PrivateRelation` constant on each model that inherits from `ActiveRecord::Base`. It can then be used in typing, like thus:
+The railtie will automatically define the `PrivateRelation`, `PrivateAssociationRelation`, and `PrivateCollectionProxy` constants on each model that inherits from `ActiveRecord::Base`. These are defined as their corresponding private `ActiveRecord` classes, so runtime type checking works as expected. They can then be used in typing, like so:
+
 ```ruby
 class Post < ::ActiveRecord::Base
   scope :recent -> { where('created_at > ?', Date.current) }
@@ -49,7 +50,7 @@ sig { params(author: Author).returns(Post::PrivateRelation) }
 def posts_from_author(author); end
 ```
 
-and the following should not raise an error:
+and the following should not raise a Sorbet error:
 
 ```ruby
 sig { params(author: Author).returns(Post::PrivateRelation) }
@@ -58,6 +59,15 @@ def recent_posts_from_author(author)
 end
 ```
 
+Boba also defines a type alias `RelationType` on each such class, which is defined as the union of the three relation types. This is useful because the relation types are often used interchangeably and so you may expect to return or pass any of the three classes as an argument. To use this, you will also need to use the `ActiveRecordRelationTypes` compiler to generate the type alias in the signatures as well (or define them manually in shims).
+
+```ruby
+sig { params(author: Author).returns(Post::RelationType) }
+def recent_posts_from_author(author)
+  posts_from_author(author).recent
+end
+```
+
 ## Contributing
 
 Bugs and feature requests are welcome and should be [filed as issues on github](https://github.com/angellist/boba/issues).

data/lib/boba/relations_railtie.rb

@@ -6,13 +6,30 @@ require "rails/railtie"
 class Boba::RelationsRailtie < Rails::Railtie
   railtie_name(:boba)
 
-  initializer("boba.add_private_relation_constant") do
+  initializer("boba.add_private_relation_constants") do
     ActiveSupport.on_load(:active_record) do
       module AciveRecordInheritDefineRelationTypes
         def inherited(child)
           super(child)
 
-          child.const_set("PrivateRelation", Object)
+          # Tapioca defines these three classes for each active record model as proxies for the actual AR internal
+          # classes. In order to be able to use these as types in signatures, we need to expose them as actual constants
+          # at runtime. Tapioca intentionally obfuscates these classes because they're private, so exposing them is
+          # _slightly_ dangerous in that someone could do something naughty. But we're not super worried about it.
+          child.const_set("PrivateRelation", child.const_get(:ActiveRecord_Relation))
+          child.const_set("PrivateAssociationRelation", child.const_get(:ActiveRecord_AssociationRelation))
+          child.const_set("PrivateCollectionProxy", child.const_get(:ActiveRecord_Associations_CollectionProxy))
+
+          # Expose a common type so that signatures can be typed to the broader `RelationType` since the three are often
+          # used interchangeably.
+          relation_type = T.type_alias do
+            T.any(
+              child.const_get(:PrivateRelation),
+              child.const_get(:PrivateAssociationRelation),
+              child.const_get(:PrivateCollectionProxy),
+            )
+          end
+          child.const_set(:RelationType, relation_type)
         end
       end
 

data/lib/boba/version.rb

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

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

@@ -0,0 +1,67 @@
+# typed: strict
+# frozen_string_literal: true
+
+return unless defined?(ActiveRecord::Base)
+
+require "tapioca/dsl/helpers/active_record_constants_helper"
+require "tapioca/dsl/compilers/active_record_relations"
+
+return unless defined?(Tapioca::Dsl::Compilers::ActiveRecordRelations)
+
+module Tapioca
+  module Dsl
+    module Compilers
+      # `Tapioca::Dsl::Compilers::ActiveRecordRelationTypes` extends `Tapioca::Dsl::Compilers::ActiveRecordRelationTypes`
+      # to generate a `RelationType` type alias for each class. This type alias is defined a runtime through the Boba
+      # railtie, and is useful for typing signatures to accept or return relations. For instance, with the following
+      # `ActiveRecord::Base` subclass:
+      # ~~~rb
+      # class Post < ApplicationRecord
+      # end
+      # ~~~
+      #
+      # This compiler will produce the RBI file `post.rbi` with the following content:
+      # ~~~rbi
+      # # post.rbi
+      # # typed: true
+      #
+      # class Post
+      #   RelationType = T.any(PrivateRelation, PrivateAssociationRelation, PrivateCollectionProxy)
+      # end
+      # ~~~
+      # So that the following method will accept any of the private relation types as an argument:
+      # ~~~rb
+      # sig { params(posts: Post::RelationType).void }
+      # def process_posts(posts)
+      #   # ...
+      # end
+      # ~~~
+      class ActiveRecordRelationTypes < Compiler
+        extend T::Sig
+
+        ConstantType = type_member { { fixed: T.class_of(::ActiveRecord::Base) } }
+
+        sig { override.void }
+        def decorate
+          root.create_path(constant) do |rbi_class|
+            relation_type_alias = "T.any(" \
+              "#{Tapioca::Dsl::Helpers::ActiveRecordConstantsHelper::RelationClassName}, " \
+              "#{Tapioca::Dsl::Helpers::ActiveRecordConstantsHelper::AssociationRelationClassName}, " \
+              "#{Tapioca::Dsl::Helpers::ActiveRecordConstantsHelper::AssociationsCollectionProxyClassName}" \
+              ")"
+            rbi_class.create_type_variable("RelationType", type: "T.type_alias { #{relation_type_alias} }")
+          end
+        end
+
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            Tapioca::Dsl::Compilers::ActiveRecordRelations.gather_constants
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -1,7 +1,7 @@
-# frozen_string_literal: true
 # typed: true
+# frozen_string_literal: true
 
-return if !defined?(AttrJson::Record)
+return unless defined?(AttrJson::Record)
 
 module Tapioca
   module Dsl
@@ -39,7 +39,7 @@ module Tapioca
         # Class methods module is already defined in the gem rbi, so just reference it here.
         ClassMethodsModuleName = "AttrJson::Record::ClassMethods"
         InstanceMethodModuleName = "AttrJsonGeneratedMethods"
-        ConstantType = type_member {{ fixed: T.any(T.class_of(::AttrJson::Record), T.class_of(::AttrJson::Model)) }}
+        ConstantType = type_member { { fixed: T.any(T.class_of(::AttrJson::Record), T.class_of(::AttrJson::Model)) } }
 
         class << self
           extend T::Sig
@@ -65,7 +65,7 @@ module Tapioca
         private
 
         def decorate_attributes(rbi_scope)
-          T.unsafe(constant).attr_json_registry
+          constant.attr_json_registry
             .definitions
             .sort_by(&:name) # this is annoying, but we need to sort to force consistent ordering or the rbi checks fail
             .each do |definition|
@@ -122,7 +122,7 @@ module Tapioca
 
           sorbet_type = "::#{sorbet_type}"
           sorbet_type = "T::Array[#{sorbet_type}]" if array
-          sorbet_type = "T.nilable(#{sorbet_type})" if nilable # todo: improve this
+          sorbet_type = "T.nilable(#{sorbet_type})" if nilable # TODO: improve this
 
           sorbet_type
         end

data/lib/tapioca/dsl/compilers/{state_machines.rb → state_machines_extended.rb}

@@ -111,7 +111,7 @@ module Tapioca
       #   end
       # end
       # ~~~
-      class StateMachines < Compiler
+      class StateMachinesExtended < Compiler
         extend T::Sig
 
         ACTIVE_RECORD_RELATION_MODULE_NAMES = [

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: boba
 version: !ruby/object:Gem::Version
-  version: 0.0.11
+  version: 0.0.13
 platform: ruby
 authors:
 - Angellist
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-28 00:00:00.000000000 Z
+date: 2024-12-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sorbet-static-and-runtime
@@ -28,16 +28,16 @@ dependencies:
   name: tapioca
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - "<="
       - !ruby/object:Gem::Version
-        version: 0.16.4
+        version: 0.16.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - "<="
       - !ruby/object:Gem::Version
-        version: 0.16.4
+        version: 0.16.5
 description:
 email:
 - alex.stathis@angellist.com
@@ -54,18 +54,19 @@ files:
 - lib/boba/version.rb
 - lib/tapioca/dsl/compilers/active_record_associations_persisted.rb
 - lib/tapioca/dsl/compilers/active_record_columns_persisted.rb
+- lib/tapioca/dsl/compilers/active_record_relation_types.rb
 - lib/tapioca/dsl/compilers/attr_json.rb
 - lib/tapioca/dsl/compilers/money_rails.rb
 - lib/tapioca/dsl/compilers/paperclip.rb
-- lib/tapioca/dsl/compilers/state_machines.rb
+- lib/tapioca/dsl/compilers/state_machines_extended.rb
 homepage: https://github.com/angellist/boba
 licenses:
 - MIT
 metadata:
   bug_tracker_uri: https://github.com/angellist/boba/issues
-  changelog_uri: https://github.com/angellist/boba/blob/0.0.11/History.md
+  changelog_uri: https://github.com/angellist/boba/blob/0.0.13/History.md
   homepage_uri: https://github.com/angellist/boba
-  source_code_uri: https://github.com/angellist/boba/tree/0.0.11
+  source_code_uri: https://github.com/angellist/boba/tree/0.0.13
   rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []