orthoses-rails 0.1.0 → 0.4.0

data/examples/rails/patch/pg/simple_decoder.rbs

@@ -0,0 +1,4 @@
+module PG
+  class SimpleDecoder
+  end
+end

data/examples/rails/patch/pg.rbs

@@ -0,0 +1,2 @@
+module PG
+end

data/examples/rails/patch/qc/queue.rbs

@@ -0,0 +1,2 @@
+class QC::Queue
+end

data/examples/rails/patch/qc.rbs

@@ -0,0 +1,2 @@
+class QC
+end

data/examples/rails/patch/que/job.rbs

@@ -0,0 +1,2 @@
+class Que::Job
+end

data/examples/rails/patch/que.rbs

@@ -0,0 +1,2 @@
+module Que
+end

data/examples/rails/patch/rack/cache/entity_store.rbs

@@ -0,0 +1,2 @@
+class Rack::Cache::EntityStore
+end

data/examples/rails/patch/rack/cache/meta_store.rbs

@@ -0,0 +1,2 @@
+class Rack::Cache::MetaStore
+end

data/examples/rails/patch/rack/cache.rbs

@@ -0,0 +1,2 @@
+module Rack::Cache
+end

data/examples/rails/patch/rack/session/dalli.rbs

@@ -0,0 +1,2 @@
+class Rack::Session::Dalli
+end

data/examples/rails/patch/rack/session.rbs

@@ -0,0 +1,2 @@
+module Rack::Session
+end

data/examples/rails/patch/rboc.rbs

@@ -0,0 +1,2 @@
+module RDoc
+end

data/examples/rails/patch/rdoc/generator/sdoc.rbs

@@ -0,0 +1,3 @@
+# defined in sdoc gem
+class RDoc::Generator::SDoc
+end

data/examples/rails/patch/rdoc/generator.rbs

@@ -0,0 +1,2 @@
+module RDoc::Generator
+end

data/examples/rails/patch/rdoc/task.rbs

@@ -0,0 +1,2 @@
+class RDoc::Task
+end

data/examples/rails/patch/ripper.rbs

@@ -0,0 +1,2 @@
+class Ripper
+end

data/examples/rails/patch/sidekiq/worker.rbs

@@ -0,0 +1,2 @@
+module Sidekiq::Worker
+end

data/examples/rails/patch/sidekiq.rbs

@@ -0,0 +1,2 @@
+module Sidekiq
+end

data/examples/rails/patch/sneakers/worker.rbs

@@ -0,0 +1,2 @@
+module Sneakers::Worker
+end

data/examples/rails/patch/sneakers.rbs

@@ -0,0 +1,2 @@
+module Sneakers
+end

data/examples/rails/patch/sucker_punch/job.rbs

@@ -0,0 +1,2 @@
+module SuckerPunch::Job
+end

data/examples/rails/patch/sucker_punch.rbs

@@ -0,0 +1,2 @@
+module SuckerPunch
+end

data/examples/rails/patch/thor/actions/create_file.rbs

@@ -0,0 +1,2 @@
+class Thor::Actions::CreateFile
+end

data/examples/rails/patch/thor/actions.rbs

@@ -0,0 +1,2 @@
+module Thor::Actions
+end

data/examples/rails/patch/thor/error.rbs

@@ -0,0 +1,2 @@
+class Thor::Error < ::StandardError
+end

data/examples/rails/patch/thor/group.rbs

@@ -0,0 +1,2 @@
+class Thor::Group
+end

data/examples/rails/patch/thor.rbs

@@ -0,0 +1,2 @@
+class Thor
+end

data/lib/orthoses/active_model/has_secure_password.rb

@@ -13,35 +13,46 @@ module Orthoses
       end
 
       def call
-        target_method = ::ActiveRecord::Base.method(:has_secure_password)
+        target_method = begin
+          ::ActiveModel::SecurePassword::ClassMethods.instance_method(:has_secure_password)
+        rescue NameError => err
+          Orthoses.logger.warn("Run `require 'active_support/concern'; require 'active_model/secure_password'` and retry because #{err}")
+          require 'active_support/concern'
+          require 'active_model/secure_password'
+          retry
+        end
         call_tracer = Orthoses::CallTracer.new
 
-        result = call_tracer.trace(target_method) do
+        store = call_tracer.trace(target_method) do
           @loader.call
         end
 
-        call_tracer.result.each do |method, argument|
-          base = method.receiver.to_s
-          if argument[:attribute].nil? # < 6.0
-            result[base].delete_if { |line| line == "include ActiveModel::SecurePassword::InstanceMethodsOnActivation" }
+        call_tracer.captures.each do |capture|
+          next unless capture.method.receiver.kind_of?(Class)
+          base_name = Utils.module_name(capture.method.receiver)
+          next unless base_name
+
+          attribute = capture.argument[:attribute] || :password
+          full_name = if ::ActiveModel::VERSION::MAJOR < 6
+            "ActiveModel::SecurePassword::InstanceMethodsOnActivation"
           else
-            result[base].delete_if { |line| line.start_with?("include #<InstanceMethodsOnActivation:") }
+            "#{base_name}::ActiveModel_SecurePassword_InstanceMethodsOnActivation_#{attribute}"
           end
-          attribute = argument[:attribute] || :password
-          mod_name = "ActiveModel_SecurePassword_InstanceMethodsOnActivation_#{attribute}"
+
           lines = []
           lines << "attr_reader #{attribute}: String?"
           lines << "def #{attribute}=: (String) -> String"
           lines << "def #{attribute}_confirmation=: (String) -> String"
-          lines << "def authenticate_#{attribute}: (String) -> (#{base} | false)"
+          lines << "def authenticate_#{attribute}: (String) -> (#{base_name} | false)"
           if attribute == :password
             lines << "alias authenticate authenticate_password"
           end
-          result["module #{base}::#{mod_name}"].concat(lines)
-          result[base] << "include #{mod_name}"
+          store[full_name].header = "module #{full_name}"
+          store[full_name].concat(lines)
+          store[base_name] << "include #{full_name}"
         end
 
-        result
+        store
       end
     end
   end

data/lib/orthoses/active_model/known_sig/7.0/serialization.rbs

@@ -0,0 +1,4 @@
+module ActiveModel::Serialization
+  # https://github.com/ruby/rbs/issues/1008
+  def send: (Object::name name, *untyped args) ?{ (*untyped) -> untyped } -> untyped
+end

data/lib/orthoses/active_model/known_sig/7.0/validations.rbs

@@ -0,0 +1,4 @@
+module ActiveModel::Validations
+  # https://github.com/ruby/rbs/issues/1008
+  def send: (Object::name name, *untyped args) ?{ (*untyped) -> untyped } -> untyped
+end

data/lib/orthoses/active_model/known_sig.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Orthoses
+  module ActiveModel
+    class KnownSig
+      include Orthoses::Rails::KnownSigHelper
+
+      def initialize(loader)
+        @loader = loader
+      end
+
+      def call
+        @loader.call.tap do |store|
+          paths = best_version_paths(::ActiveModel::VERSION::STRING, __dir__)
+          env = Content::Environment.load_from_paths(paths)
+          env.write_to(store: store)
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/active_model.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require_relative 'active_model/has_secure_password'
+require_relative 'active_model/known_sig'

data/lib/orthoses/active_record/belongs_to.rb

@@ -11,6 +11,7 @@ module Orthoses
         @loader.call.tap do |store|
           ::ActiveRecord::Base.descendants.each do |base|
             next if base.abstract_class?
+            base_name = Orthoses::Utils.module_name(base) || next
 
             lines = base.reflect_on_all_associations(:belongs_to).flat_map do |ref|
               # FIXME: Can I get list of class for polymorphic?
@@ -30,11 +31,16 @@ module Orthoses
               end
             end
 
-            generated_association_methods = "#{base}::GeneratedAssociationMethods"
-            store["module #{generated_association_methods}"].concat(lines)
+            generated_association_methods = "#{base_name}::GeneratedAssociationMethods"
 
-            code = "include #{generated_association_methods}"
-            store[base.to_s] << code
+            store[generated_association_methods].tap do |content|
+              content.header = "module #{generated_association_methods}"
+              content.concat(lines)
+            end
+
+            store[base_name].tap do |content|
+              store[base_name] << "include #{generated_association_methods}"
+            end
           end
         end
       end

data/lib/orthoses/active_record/generated_attribute_methods.rb

@@ -12,9 +12,7 @@ module Orthoses
           ::ActiveRecord::Base.descendants.each do |klass|
             next if klass.abstract_class?
 
-            name = klass.to_s
-            generated_attribute_methods = "#{name}::AttributeMethods::GeneratedAttributeMethods"
-            store[name] << "include #{generated_attribute_methods}"
+            name = Utils.module_name(klass) || next
             lines = klass.columns.flat_map do |col|
               req = sql_type_to_rbs(col.type)
               opt = "#{req}?"
@@ -41,7 +39,11 @@ module Orthoses
                 "def clear_#{col.name}_change: () -> void",
               ]
             end
-            store["module #{generated_attribute_methods}"].concat(lines)
+            generated_attribute_methods = "#{name}::AttributeMethods::GeneratedAttributeMethods"
+            store[name] << "include #{generated_attribute_methods}"
+
+            store[generated_attribute_methods].header = "module #{generated_attribute_methods}"
+            store[generated_attribute_methods].concat(lines)
           end
         end
       end

data/lib/orthoses/active_record/has_many.rb

@@ -11,9 +11,9 @@ module Orthoses
         @loader.call.tap do |store|
           ::ActiveRecord::Base.descendants.each do |base|
             next if base.abstract_class?
+            base_name = Utils.module_name(base) || next
 
-            generated_association_methods = "#{base}::GeneratedAssociationMethods"
-            collection_proxy = "#{base}::ActiveRecord_Associations_CollectionProxy"
+            collection_proxy = "#{base_name}::ActiveRecord_Associations_CollectionProxy"
 
             lines = base.reflect_on_all_associations(:has_many).flat_map do |ref|
               singular_name = ref.name.to_s.singularize
@@ -26,10 +26,10 @@ module Orthoses
               ]
             end
 
-            store["module #{generated_association_methods}"].concat(lines)
-
-            code = "include #{generated_association_methods}"
-            store[base.to_s] << code
+            generated_association_methods = "#{base_name}::GeneratedAssociationMethods"
+            store[generated_association_methods].header = "module #{generated_association_methods}"
+            store[generated_association_methods].concat(lines)
+            store[base_name] << "include #{generated_association_methods}"
           end
         end
       end

data/lib/orthoses/active_record/has_one.rb

@@ -11,6 +11,7 @@ module Orthoses
         @loader.call.tap do |store|
           ::ActiveRecord::Base.descendants.each do |base|
             next if base.abstract_class?
+            base_name = Utils.module_name(base) || next
 
             lines = base.reflect_on_all_associations(:has_one).flat_map do |ref|
               type = ref.klass.to_s
@@ -26,11 +27,11 @@ module Orthoses
               ]
             end
 
-            generated_association_methods = "#{base}::GeneratedAssociationMethods"
-            store["module #{generated_association_methods}"].concat(lines)
+            generated_association_methods = "#{base_name}::GeneratedAssociationMethods"
+            store[generated_association_methods].header = "module #{generated_association_methods}"
+            store[generated_association_methods].concat(lines)
 
-            code = "include #{generated_association_methods}"
-            store[base.to_s] << code
+            store[base_name] << "include #{generated_association_methods}"
           end
         end
       end

data/lib/orthoses/active_record/known_sig/6.0/result.rbs

@@ -0,0 +1,5 @@
+class ActiveRecord::Result
+  include Enumerable[Hash[String, untyped]]
+  def each: () -> Enumerator[Hash[String, untyped], Array[Hash[String, untyped]]]
+          | () { (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+end

data/lib/orthoses/active_record/known_sig/6.1/result.rbs

@@ -0,0 +1,5 @@
+class ActiveRecord::Result
+  include Enumerable[Hash[String, untyped]]
+  def each: () -> Enumerator[Hash[String, untyped], Array[Hash[String, untyped]]]
+          | () { (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+end

data/lib/orthoses/active_record/known_sig/7.0/encryption/context.rbs

@@ -0,0 +1,9 @@
+# attr_accessor in #each
+class ActiveRecord::Encryption::Context
+  attr_accessor key_provider: untyped
+  attr_accessor key_generator: untyped
+  attr_accessor cipher: untyped
+  attr_accessor message_serializer: untyped
+  attr_accessor encryptor: untyped
+  attr_accessor frozen_encryption: untyped
+end

data/lib/orthoses/active_record/known_sig/7.0/result.rbs

@@ -0,0 +1,5 @@
+class ActiveRecord::Result
+  include Enumerable[Hash[String, untyped]]
+  def each: () -> Enumerator[Hash[String, untyped], Array[Hash[String, untyped]]]
+          | () { (Hash[String, untyped]) -> void } -> Array[Hash[String, untyped]]
+end

data/lib/orthoses/active_record/known_sig.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Orthoses
+  module ActiveRecord
+    class KnownSig
+      include Orthoses::Rails::KnownSigHelper
+
+      def initialize(loader)
+        @loader = loader
+      end
+
+      def call
+        @loader.call.tap do |store|
+          paths = best_version_paths(::ActiveRecord::VERSION::STRING, __dir__)
+          env = Content::Environment.load_from_paths(paths)
+          env.write_to(store: store)
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/active_record/query_methods.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module Orthoses
+  module ActiveRecord
+    class QueryMethods
+      def initialize(loader)
+        @loader = loader
+      end
+
+      def call
+        @loader.call.tap do |store|
+          content = store["ActiveRecord::QueryMethods"]
+
+          ::ActiveRecord::Relation::VALUE_METHODS.each do |name|
+            method_name, type =
+              case name
+              when *::ActiveRecord::Relation::MULTI_VALUE_METHODS
+                ["#{name}_values", "::Array[untyped]"]
+              when *::ActiveRecord::Relation::SINGLE_VALUE_METHODS
+                ["#{name}_value", name == :create_with ? "::Hash[untyped, untyped]?" : "untyped"]
+              when *::ActiveRecord::Relation::CLAUSE_METHODS
+                ["#{name}_clause", name == :from ? "::ActiveRecord::Relation::FromClause" : "::ActiveRecord::Relation::WhereClause"]
+              end
+
+            content << "def #{method_name}: () -> #{type}"
+            content << "def #{method_name}=: (#{type} value) -> #{type}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/active_record.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require_relative 'active_record/belongs_to'
+require_relative 'active_record/generated_attribute_methods'
+require_relative 'active_record/has_many'
+require_relative 'active_record/has_one'
+require_relative 'active_record/known_sig'
+require_relative 'active_record/query_methods'

data/lib/orthoses/active_support/class_attribute.rb

@@ -15,24 +15,35 @@ module Orthoses
       end
 
       def call
-        require 'active_support/core_ext/class/attribute'
-        target_method = ::Class.instance_method(:class_attribute)
+        target_method = begin
+          ::Class.instance_method(:class_attribute)
+        rescue NameError => err
+          Orthoses.logger.warn("Run `require 'active_support/core_ext/class/attribute'` and retry because #{err}")
+          require 'active_support/core_ext/class/attribute'
+          retry
+        end
         call_tracer = Orthoses::CallTracer.new
 
-        result = call_tracer.trace(target_method) do
+        store = call_tracer.trace(target_method) do
           @loader.call
         end
 
-        call_tracer.result.each do |method, argument|
+        call_tracer.captures.each do |capture|
+          receiver_name = Orthoses::Utils.module_name(capture.method.receiver)
+          next unless receiver_name
+
           methods = []
-          if ::ActiveSupport.version < Gem::Version.new("6.0")
-            options = argument[:attrs].extract_options!
-            argument[:instance_reader]    = options.fetch(:instance_accessor, true) && options.fetch(:instance_reader, true)
-            argument[:instance_writer]    = options.fetch(:instance_accessor, true) && options.fetch(:instance_writer, true)
-            argument[:instance_predicate] = options.fetch(:instance_predicate, true)
-            argument[:default_value]      = options.fetch(:default, nil)
+          if ::ActiveSupport::VERSION::MAJOR < 6
+            options = capture.argument[:attrs].extract_options!
+            capture.argument[:instance_reader]    = options.fetch(:instance_accessor, true) && options.fetch(:instance_reader, true)
+            capture.argument[:instance_writer]    = options.fetch(:instance_accessor, true) && options.fetch(:instance_writer, true)
+            capture.argument[:instance_predicate] = options.fetch(:instance_predicate, true)
+            capture.argument[:default_value]      = options.fetch(:default, nil)
           end
-          argument[:attrs].each do |name|
+
+          content = store[receiver_name]
+
+          capture.argument[:attrs].each do |name|
             next unless @if.nil? || @if.call(method, name)
 
             # skip internal attribute
@@ -41,18 +52,20 @@ module Orthoses
             next if name == :attributes_to_define_after_schema_loads
 
             methods << "def self.#{name}: () -> untyped"
-            methods << "def self.#{name}?: () -> bool" if argument[:instance_predicate]
+            methods << "def self.#{name}?: () -> bool" if capture.argument[:instance_predicate]
             methods << "def self.#{name}=: (untyped value) -> untyped"
-            methods << "def #{name}: () -> untyped" if argument[:instance_reader]
-            methods << "def #{name}?: () -> bool" if argument[:instance_predicate] && argument[:instance_reader]
-            methods << "attr_writer #{name}: untyped" if argument[:instance_writer]
+            methods << "def #{name}: () -> untyped" if capture.argument[:instance_reader]
+            methods << "def #{name}?: () -> bool" if capture.argument[:instance_predicate] && capture.argument[:instance_reader]
+            methods << "def #{name}=: (untyped value) -> untyped" if capture.argument[:instance_writer]
+            # In RBS, `foo=` and attr_writer :foo cannot live together.
+            content.body.delete_if { |line| line.start_with?("attr_writer #{name}:") }
           end
           next if methods.empty?
 
-          result[method.receiver.to_s].concat(methods)
+          content.concat(methods)
         end
 
-        result
+        store
       end
     end
   end

data/lib/orthoses/active_support/configurable.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Orthoses
+  module ActiveSupport
+    # <= 6.1
+    #   def config_accessor(*names, instance_reader: true, instance_writer: true, instance_accessor: true)
+    # >= 7
+    #   def config_accessor(*names, instance_reader: true, instance_writer: true, instance_accessor: true, default: nil)
+    class Configurable
+      def initialize(loader)
+        @loader = loader
+      end
+
+      def call
+        config_accessor = CallTracer.new
+        store = config_accessor.trace(::ActiveSupport::Configurable::ClassMethods.instance_method(:config_accessor)) do
+          @loader.call
+        end
+        config_accessor.captures.each do |capture|
+          mod_name = Utils.module_name(capture.method.receiver) or next
+          content = store[mod_name]
+          capture.argument[:names].each do |name|
+            content << "def self.#{name}: () -> untyped"
+            content << "def self.#{name}=: (untyped value) -> untyped"
+            if capture.argument[:instance_accessor]
+              content << "def #{name}: () -> untyped" if capture.argument[:instance_reader]
+              content << "def #{name}=: (untyped value) -> untyped" if capture.argument[:instance_writer]
+            end
+          end
+        end
+
+        store
+      end
+    end
+  end
+end