tapioca 0.9.4 → 0.10.0

data/lib/tapioca/dsl/compiler.rb

@@ -22,6 +22,40 @@ module Tapioca
       sig { returns(RBI::Tree) }
       attr_reader :root
 
+      class << self
+        extend T::Sig
+
+        sig { params(constant: Module).returns(T::Boolean) }
+        def handles?(constant)
+          processable_constants.include?(constant)
+        end
+
+        sig { abstract.returns(T::Enumerable[Module]) }
+        def gather_constants; end
+
+        sig { returns(T::Set[Module]) }
+        def processable_constants
+          @processable_constants ||= T.let(
+            T::Set[Module].new(gather_constants).compare_by_identity,
+            T.nilable(T::Set[Module])
+          )
+        end
+
+        private
+
+        sig { returns(T::Enumerable[Class]) }
+        def all_classes
+          @all_classes = T.let(@all_classes, T.nilable(T::Enumerable[Class]))
+          @all_classes ||= T.cast(ObjectSpace.each_object(Class), T::Enumerable[Class]).each
+        end
+
+        sig { returns(T::Enumerable[Module]) }
+        def all_modules
+          @all_modules = T.let(@all_modules, T.nilable(T::Enumerable[Module]))
+          @all_modules ||= T.cast(ObjectSpace.each_object(Module), T::Enumerable[Module]).each
+        end
+      end
+
       sig { params(pipeline: Tapioca::Dsl::Pipeline, root: RBI::Tree, constant: ConstantType).void }
       def initialize(pipeline, root, constant)
         @pipeline = pipeline
@@ -30,11 +64,6 @@ module Tapioca
         @errors = T.let([], T::Array[String])
       end
 
-      sig { params(constant: Module).returns(T::Boolean) }
-      def self.handles?(constant)
-        processable_constants.include?(constant)
-      end
-
       sig { params(compiler_name: String).returns(T::Boolean) }
       def compiler_enabled?(compiler_name)
         @pipeline.compiler_enabled?(compiler_name)
@@ -43,17 +72,6 @@ module Tapioca
       sig { abstract.void }
       def decorate; end
 
-      sig { abstract.returns(T::Enumerable[Module]) }
-      def self.gather_constants; end
-
-      sig { returns(T::Set[Module]) }
-      def self.processable_constants
-        @processable_constants ||= T.let(
-          T::Set[Module].new(gather_constants).compare_by_identity,
-          T.nilable(T::Set[Module])
-        )
-      end
-
       # NOTE: This should eventually accept an `Error` object or `Exception` rather than simply a `String`.
       sig { params(error: String).void }
       def add_error(error)
@@ -62,18 +80,6 @@ module Tapioca
 
       private
 
-      sig { returns(T::Enumerable[Class]) }
-      private_class_method def self.all_classes
-        @all_classes = T.let(@all_classes, T.nilable(T::Enumerable[Class]))
-        @all_classes ||= T.cast(ObjectSpace.each_object(Class), T::Enumerable[Class]).each
-      end
-
-      sig { returns(T::Enumerable[Module]) }
-      private_class_method def self.all_modules
-        @all_modules = T.let(@all_modules, T.nilable(T::Enumerable[Module]))
-        @all_modules ||= T.cast(ObjectSpace.each_object(Module), T::Enumerable[Module]).each
-      end
-
       # Get the types of each parameter from a method signature
       sig do
         params(
@@ -158,15 +164,6 @@ module Tapioca
         return_type = "T.untyped" if return_type == "<NOT-TYPED>"
         return_type
       end
-
-      sig { params(type: String).returns(String) }
-      def as_nilable_type(type)
-        if type.start_with?("T.nilable(", "::T.nilable(") || type == "T.untyped" || type == "::T.untyped"
-          type
-        else
-          "T.nilable(#{type})"
-        end
-      end
     end
   end
 end

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

@@ -115,9 +115,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          T.cast(ObjectSpace.each_object(::AASM::ClassMethods), T::Enumerable[Module])
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            T.cast(ObjectSpace.each_object(::AASM::ClassMethods), T::Enumerable[Module])
+          end
         end
       end
     end

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

@@ -121,9 +121,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActionController::Base).reject(&:abstract?).select(&:name)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActionController::Base).reject(&:abstract?).select(&:name)
+          end
         end
 
         private

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

@@ -54,9 +54,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActionMailer::Base).reject(&:abstract?)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActionMailer::Base).reject(&:abstract?)
+          end
         end
       end
     end

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

@@ -67,9 +67,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveJob::Base)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveJob::Base)
+          end
         end
       end
     end

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

@@ -53,9 +53,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          all_classes.grep(::ActiveModel::Attributes::ClassMethods)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            all_classes.grep(::ActiveModel::Attributes::ClassMethods)
+          end
         end
 
         private
@@ -71,11 +75,11 @@ module Tapioca
             constant.attribute_method_matchers
           end
           patterns.flat_map do |pattern|
-            constant.attribute_types.map do |name, value|
+            constant.attribute_types.filter_map do |name, value|
               next unless handle_method_pattern?(pattern)
 
               [pattern.method_name(name), type_for(value)]
-            end.compact
+            end
           end
         end
 

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

@@ -85,13 +85,17 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          # This selects all classes that are `ActiveModel::SecurePassword::ClassMethods === klass`.
-          # In other words, we select all classes that have `ActiveModel::SecurePassword::ClassMethods`
-          # as an ancestor of its singleton class, i.e. all classes that have extended the
-          # `ActiveModel::SecurePassword::ClassMethods` module.
-          all_classes.grep(::ActiveModel::SecurePassword::ClassMethods)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            # This selects all classes that are `ActiveModel::SecurePassword::ClassMethods === klass`.
+            # In other words, we select all classes that have `ActiveModel::SecurePassword::ClassMethods`
+            # as an ancestor of its singleton class, i.e. all classes that have extended the
+            # `ActiveModel::SecurePassword::ClassMethods` module.
+            all_classes.grep(::ActiveModel::SecurePassword::ClassMethods)
+          end
         end
       end
     end

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

@@ -135,9 +135,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+          end
         end
 
         private

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

@@ -135,9 +135,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+          end
         end
 
         private

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

@@ -80,9 +80,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base)
+          end
         end
 
         private

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

@@ -57,11 +57,15 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          return [] unless Rails.application
+        class << self
+          extend T::Sig
 
-          [ActiveSupport::TestCase]
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            return [] unless Rails.application
+
+            [ActiveSupport::TestCase]
+          end
         end
 
         private

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

@@ -166,9 +166,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          ActiveRecord::Base.descendants.reject(&:abstract_class?)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            ActiveRecord::Base.descendants.reject(&:abstract_class?)
+          end
         end
 
         ASSOCIATION_METHODS = T.let(

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

@@ -80,9 +80,11 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+        class << self
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base).reject(&:abstract_class?)
+          end
         end
 
         private

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

@@ -113,10 +113,14 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base).select do |klass|
-            klass.include?(ActiveRecord::TypedStore::Behavior)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base).select do |klass|
+              klass.include?(ActiveRecord::TypedStore::Behavior)
+            end
           end
         end
 

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

@@ -74,9 +74,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveResource::Base)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveResource::Base)
+          end
         end
 
         private

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

@@ -74,11 +74,15 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveRecord::Base)
-            .reject(&:abstract_class?)
-            .grep(::ActiveStorage::Reflection::ActiveRecordExtensions::ClassMethods)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveRecord::Base)
+              .reject(&:abstract_class?)
+              .grep(::ActiveStorage::Reflection::ActiveRecordExtensions::ClassMethods)
+          end
         end
 
         private

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

@@ -52,10 +52,9 @@ module Tapioca
 
           mixed_in_class_methods = dependencies
             .uniq # Deduplicate
-            .map do |concern| # Map to class methods module name, if exists
+            .filter_map do |concern| # Map to class methods module name, if exists
               "#{qualified_name_of(concern)}::ClassMethods" if concern.const_defined?(:ClassMethods)
             end
-            .compact # Remove non-existent records
 
           return if mixed_in_class_methods.empty?
 
@@ -66,24 +65,28 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          # Find all Modules that are:
-          all_modules.select do |mod|
-            # named (i.e. not anonymous)
-            name_of(mod) &&
-              # not singleton classes
-              !mod.singleton_class? &&
-              # extend ActiveSupport::Concern, and
-              mod.singleton_class < ActiveSupport::Concern &&
-              # have dependencies (i.e. include another concern)
-              !dependencies_of(mod).empty?
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            # Find all Modules that are:
+            all_modules.select do |mod|
+              # named (i.e. not anonymous)
+              name_of(mod) &&
+                # not singleton classes
+                !mod.singleton_class? &&
+                # extend ActiveSupport::Concern, and
+                mod.singleton_class < ActiveSupport::Concern &&
+                # have dependencies (i.e. include another concern)
+                !dependencies_of(mod).empty?
+            end
           end
-        end
 
-        sig { params(concern: Module).returns(T::Array[Module]) }
-        def self.dependencies_of(concern)
-          concern.instance_variable_get(:@_dependencies)
+          sig { params(concern: Module).returns(T::Array[Module]) }
+          def dependencies_of(concern)
+            concern.instance_variable_get(:@_dependencies)
+          end
         end
 
         private

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

@@ -88,9 +88,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::ActiveSupport::CurrentAttributes)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::ActiveSupport::CurrentAttributes)
+          end
         end
 
         private

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

@@ -96,16 +96,20 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          name = ::Config.const_name
-          return [] unless Object.const_defined?(name)
+        class << self
+          extend T::Sig
 
-          config_object = Object.const_get(name)
-          options_class_name = "#{name}#{CONFIG_OPTIONS_SUFFIX}"
-          Object.const_set(options_class_name, config_object.singleton_class)
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            name = ::Config.const_name
+            return [] unless Object.const_defined?(name)
 
-          Array(config_object.singleton_class)
+            config_object = Object.const_get(name)
+            options_class_name = "#{name}#{CONFIG_OPTIONS_SUFFIX}"
+            Object.const_set(options_class_name, config_object.singleton_class)
+
+            Array(config_object.singleton_class)
+          end
         end
       end
     end

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

@@ -88,9 +88,13 @@ module Tapioca
           end
         end
 
-        sig { override.returns(T::Enumerable[Module]) }
-        def self.gather_constants
-          descendants_of(::FrozenRecord::Base).reject(&:abstract_class?)
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            descendants_of(::FrozenRecord::Base).reject(&:abstract_class?)
+          end
         end
 
         private

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

@@ -0,0 +1,71 @@
+# typed: strict
+# frozen_string_literal: true
+
+begin
+  require "graphql"
+rescue LoadError
+  return
+end
+
+require "tapioca/dsl/helpers/graphql_type_helper"
+
+module Tapioca
+  module Dsl
+    module Compilers
+      # `Tapioca::Dsl::Compilers::GraphqlInputObject` generates RBI files for subclasses of
+      # [`GraphQL::Schema::InputObject`](https://graphql-ruby.org/api-doc/2.0.11/GraphQL/Schema/InputObject).
+      #
+      # For example, with the following `GraphQL::Schema::InputObject` subclass:
+      #
+      # ~~~rb
+      # class CreateCommentInput < GraphQL::Schema::InputObject
+      #   argument :body, String, required: true
+      #   argument :post_id, ID, required: true
+      # end
+      # ~~~
+      #
+      # this compiler will produce the RBI file `notify_user_job.rbi` with the following content:
+      #
+      # ~~~rbi
+      # # create_comment.rbi
+      # # typed: true
+      # class CreateCommentInput
+      #   sig { returns(String) }
+      #   def body; end
+      #
+      #   sig { returns(String) }
+      #   def post_id; end
+      # end
+      # ~~~
+      class GraphqlInputObject < Compiler
+        extend T::Sig
+
+        ConstantType = type_member { { fixed: T.class_of(GraphQL::Schema::InputObject) } }
+
+        sig { override.void }
+        def decorate
+          arguments = constant.all_argument_definitions
+          return if arguments.empty?
+
+          graphql_type_helper = Helpers::GraphqlTypeHelper.new
+
+          root.create_path(constant) do |input_object|
+            arguments.each do |argument|
+              name = argument.keyword.to_s
+              input_object.create_method(name, return_type: graphql_type_helper.type_for(argument.type))
+            end
+          end
+        end
+
+        class << self
+          extend T::Sig
+
+          sig { override.returns(T::Enumerable[Module]) }
+          def gather_constants
+            all_classes.select { |c| c < GraphQL::Schema::InputObject }
+          end
+        end
+      end
+    end
+  end
+end