rbs_activesupport 1.0.0

data/lib/rbs_activesupport/include.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require "active_support/concern"
+
+module RbsActivesupport
+  class Include
+    attr_reader :context, :module_path, :options
+
+    def initialize(context, module_path, options)
+      @context = context
+      @module_path = module_path
+      @options = options
+    end
+
+    def argument
+      if module_path.first.nil?
+        RBS::Namespace.new(path: module_path[1...], absolute: true) # steep:ignore ArgumentTypeMismatch
+      else
+        RBS::Namespace.new(path: module_path, absolute: false) # steep:ignore ArgumentTypeMismatch
+      end
+    end
+
+    def module_name
+      namespace = @context
+
+      loop do
+        modname = namespace + argument
+        return modname if Object.const_defined?(modname.to_s.delete_suffix("::"))
+
+        break if namespace.empty?
+
+        namespace = namespace.parent
+      end
+    end
+
+    def concern?
+      return false unless module_name
+
+      modname = module_name.to_s.delete_suffix("::")
+      return false unless Object.const_defined?(modname)
+
+      mod = Object.const_get(modname)
+      mod&.singleton_class&.include?(ActiveSupport::Concern)
+    end
+
+    def classmethods?
+      return false unless module_name
+
+      modname = module_name.append(:ClassMethods).to_s.delete_suffix("::")
+      Object.const_defined?(modname)
+    end
+
+    def public?
+      !private?
+    end
+
+    def private?
+      options.fetch(:private, false)
+    end
+  end
+end

data/lib/rbs_activesupport/method_searcher.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+require "rbs"
+
+module RbsActivesupport
+  class MethodSearcher
+    attr_reader :rbs_builder
+
+    def initialize(rbs_builder)
+      @rbs_builder = rbs_builder
+    end
+
+    def method_types_for(delegate)
+      delegate_to = lookup_method_types(delegate.namespace.to_type_name, delegate.to)
+      return ["() -> untyped"] if delegate_to.any? { |t| t.type.return_type.is_a?(RBS::Types::Bases::Any) }
+
+      return_types = delegate_to
+                     .filter_map { |t| t.type.return_type.name } # steep:ignore NoMethod
+                     .uniq
+                     .flat_map { |t| lookup_method_types(t, delegate.method) }
+                     .map(&:to_s)
+      return_types << "() -> untyped" if return_types.empty?
+      return_types
+    end
+
+    private
+
+    def lookup_method_types(type_name, method)
+      instance = rbs_builder.build_instance(type_name)
+      method_def = instance.methods[method]
+      return [] unless method_def
+
+      method_def.defs.map(&:type)
+    rescue StandardError
+      []
+    end
+  end
+end

data/lib/rbs_activesupport/parser.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require "pathname"
+require "rbs"
+require "rbs/cli"
+require "rbs/prototype/rb"
+
+module RbsActivesupport
+  class Parser < ::RBS::Prototype::RB
+    class MethodCall
+      attr_reader :name, :args
+
+      def initialize(name, args, private)
+        @name = name
+        @args = args
+        @private = private
+      end
+
+      def private?
+        @private
+      end
+    end
+
+    METHODS = %i[
+      class_attribute delegate cattr_accessor mattr_accessor cattr_reader mattr_reader cattr_writer mattr_writer include
+    ].freeze # steep:ignore IncompatibleAssignment
+
+    alias process_orig process
+
+    attr_reader :method_calls
+
+    def initialize
+      super
+      @method_calls = Hash.new { |hash, key| hash[key] = [] }
+    end
+
+    def process(node, decls:, comments:, context:)
+      case node.type
+      when :FCALL, :VCALL
+        args = node.children[1]&.children || []
+        case node.children[0]
+        when *METHODS
+          @method_calls[context.namespace] << MethodCall.new(node.children[0], args, private?(decls))
+        else
+          process_orig(node, decls: decls, comments: comments, context: context)
+        end
+      else
+        process_orig(node, decls: decls, comments: comments, context: context)
+      end
+    end
+
+    def private?(decls)
+      current_accessibility(decls) == private
+    end
+  end
+end

data/lib/rbs_activesupport/rake_task.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require "pathname"
+require "rake/tasklib"
+
+module RbsActivesupport
+  class RakeTask < Rake::TaskLib
+    attr_accessor :name, :signature_root_dir
+
+    def initialize(name = :'rbs:activesupport', &block)
+      super()
+
+      @name = name
+      @signature_root_dir = Pathname(Rails.root / "sig/activesupport")
+
+      block&.call(self)
+
+      define_clean_task
+      define_generate_task
+      define_setup_task
+    end
+
+    def define_setup_task
+      desc "Run all tasks of rbs_activesupport"
+
+      deps = [:"#{name}:clean", :"#{name}:generate"]
+      task("#{name}:setup" => deps)
+    end
+
+    def define_generate_task
+      desc "Generate RBS files for activesupport gem"
+      task("#{name}:generate": :environment) do
+        require "rbs_activesupport" # load RbsActivesupport lazily
+
+        Rails.application.eager_load!
+
+        signature_root_dir.mkpath
+
+        (Rails.root / "app").glob("**/*.rb").each do |file|
+          rbs = Generator.new(file, rbs_builder).generate
+          next unless rbs
+
+          rbs_path = signature_root_dir / file.sub_ext(".rbs").relative_path_from(Rails.root)
+          rbs_path.dirname.mkpath
+          rbs_path.write(rbs)
+        end
+      end
+    end
+
+    def define_clean_task
+      desc "Clean RBS files for config gem"
+      task "#{name}:clean" do
+        signature_root_dir.rmtree if signature_root_dir.exist?
+      end
+    end
+
+    private
+
+    def rbs_builder
+      @rbs_builder ||= begin
+        loader = RBS::CLI::LibraryOptions.new.loader
+        loader.add(path: Pathname("sig"))
+        env = RBS::Environment.from_loader(loader).resolve_type_names
+        RBS::DefinitionBuilder.new(env: env)
+      end
+    end
+  end
+end

data/lib/rbs_activesupport/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module RbsActivesupport
+  VERSION = "1.0.0"
+end

data/lib/rbs_activesupport.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require_relative "rbs_activesupport/ast"
+require_relative "rbs_activesupport/attribute_accessor"
+require_relative "rbs_activesupport/class_attribute"
+require_relative "rbs_activesupport/declaration_builder"
+require_relative "rbs_activesupport/delegate"
+require_relative "rbs_activesupport/generator"
+require_relative "rbs_activesupport/include"
+require_relative "rbs_activesupport/method_searcher"
+require_relative "rbs_activesupport/parser"
+require_relative "rbs_activesupport/version"
+
+module RbsActivesupport
+  class Error < StandardError; end
+  # Your code goes here...
+end

data/rbs_collection.lock.yaml

@@ -0,0 +1,272 @@
+---
+path: ".gem_rbs_collection"
+gems:
+- name: actionpack
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: actionview
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activesupport
+  version: '7.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: ast
+  version: '2.4'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: base64
+  version: '0'
+  source:
+    type: stdlib
+- name: bigdecimal
+  version: '0'
+  source:
+    type: stdlib
+- name: cgi
+  version: '0'
+  source:
+    type: stdlib
+- name: concurrent-ruby
+  version: '1.1'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: connection_pool
+  version: '2.4'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: date
+  version: '0'
+  source:
+    type: stdlib
+- name: dbm
+  version: '0'
+  source:
+    type: stdlib
+- name: erb
+  version: '0'
+  source:
+    type: stdlib
+- name: fileutils
+  version: '0'
+  source:
+    type: stdlib
+- name: i18n
+  version: '1.10'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: io-console
+  version: '0'
+  source:
+    type: stdlib
+- name: json
+  version: '0'
+  source:
+    type: stdlib
+- name: logger
+  version: '0'
+  source:
+    type: stdlib
+- name: minitest
+  version: '0'
+  source:
+    type: stdlib
+- name: monitor
+  version: '0'
+  source:
+    type: stdlib
+- name: mutex_m
+  version: '0'
+  source:
+    type: stdlib
+- name: nokogiri
+  version: '1.11'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: optparse
+  version: '0'
+  source:
+    type: stdlib
+- name: parallel
+  version: '1.20'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: parser
+  version: '3.2'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: pathname
+  version: '0'
+  source:
+    type: stdlib
+- name: pstore
+  version: '0'
+  source:
+    type: stdlib
+- name: psych
+  version: '0'
+  source:
+    type: stdlib
+- name: rack
+  version: '2.2'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rails-dom-testing
+  version: '2.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: railties
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rainbow
+  version: '3.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rake
+  version: '13.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rbs
+  version: 3.5.2
+  source:
+    type: rubygems
+- name: rdoc
+  version: '0'
+  source:
+    type: stdlib
+- name: regexp_parser
+  version: '2.8'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rubocop
+  version: '1.57'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rubocop-ast
+  version: '1.30'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: securerandom
+  version: '0'
+  source:
+    type: stdlib
+- name: singleton
+  version: '0'
+  source:
+    type: stdlib
+- name: strscan
+  version: '0'
+  source:
+    type: stdlib
+- name: tempfile
+  version: '0'
+  source:
+    type: stdlib
+- name: thor
+  version: '1.2'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: time
+  version: '0'
+  source:
+    type: stdlib
+- name: timeout
+  version: '0'
+  source:
+    type: stdlib
+- name: tsort
+  version: '0'
+  source:
+    type: stdlib
+- name: tzinfo
+  version: '2.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: a6ed3d2c66aa743a4f6f71b70e0c7e912b94d054
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: uri
+  version: '0'
+  source:
+    type: stdlib
+gemfile_lock_path: Gemfile.lock

data/rbs_collection.yaml

@@ -0,0 +1,19 @@
+# Download sources
+sources:
+  - type: git
+    name: ruby/gem_rbs_collection
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    revision: main
+    repo_dir: gems
+
+# You can specify local directories as sources also.
+# - type: local
+#   path: path/to/your/local/repository
+
+# A directory to install the downloaded RBSs
+path: .gem_rbs_collection
+
+# gems:
+#   # If you want to avoid installing rbs files for gems, you can specify them here.
+#   - name: GEM_NAME
+#     ignore: true

data/sig/rbs_activesupport/ast.rbs

@@ -0,0 +1,7 @@
+module RbsActivesupport
+  module AST
+    def eval_args: (Array[untyped] args) -> Array[Array[Symbol?]]
+    def eval_args_with_options: (Array[untyped] args) -> [Array[Symbol], Hash[Symbol, untyped]]
+    def eval_node: (untyped node) -> untyped
+  end
+end

data/sig/rbs_activesupport/attribute_accessor.rbs

@@ -0,0 +1,15 @@
+module RbsActivesupport
+  class AttributeAccessor
+    attr_reader name: Symbol
+    attr_reader options: Hash[untyped, untyped]
+
+    def initialize: (Symbol name, Hash[untyped, untyped] options) -> void
+    def singleton_reader?: () -> bool
+    def singleton_writer?: () -> bool
+    def instance_accessor?: () -> bool
+    def instance_reader?: () -> bool
+    def instance_writer?: () -> bool
+    def public?: () -> bool
+    def private?: () -> bool
+  end
+end

data/sig/rbs_activesupport/class_attribute.rbs

@@ -0,0 +1,14 @@
+module RbsActivesupport
+  class ClassAttribute
+    attr_reader name: Symbol
+    attr_reader options: Hash[untyped, untyped]
+
+    def initialize: (Symbol name, Hash[untyped, untyped] options) -> void
+    def instance_accessor?: () -> bool
+    def instance_reader?: () -> bool
+    def instance_writer?: () -> bool
+    def instance_predicate?: () -> bool
+    def public?: () -> bool
+    def private?: () -> bool
+  end
+end

data/sig/rbs_activesupport/declaration_builder.rbs

@@ -0,0 +1,25 @@
+module RbsActivesupport
+  class DeclarationBuilder
+    type t = AttributeAccessor | ClassAttribute | Delegate | Include
+
+    include AST
+
+    attr_reader method_searcher: MethodSearcher
+
+    def initialize: (MethodSearcher method_searcher) -> void
+    def build: (RBS::Namespace namespace, Array[Parser::MethodCall] method_calls) -> [Array[String], Array[String]]
+
+    private
+
+    def build_method_calls: (RBS::Namespace namespace, Array[Parser::MethodCall] method_calls) -> Array[t]
+    def build_attribute_accessor: (Parser::MethodCall method_call) -> Array[AttributeAccessor]
+    def build_class_attribute: (Parser::MethodCall method_calls) -> Array[ClassAttribute]
+    def build_delegate: (RBS::Namespace namespace, Parser::MethodCall method_calls) -> Array[Delegate]
+    def build_include: (RBS::Namespace namespace, Parser::MethodCall method_calls) -> Array[Include]
+    def render: (t decl) -> String?
+    def render_attribute_accessor: (AttributeAccessor decl) -> String
+    def render_class_attribute: (ClassAttribute decl) -> String
+    def render_delegate: (Delegate decl) -> String
+    def render_include: (Include decl) -> String?
+  end
+end

data/sig/rbs_activesupport/delegate.rbs

@@ -0,0 +1,13 @@
+module RbsActivesupport
+  class Delegate
+    attr_reader namespace: RBS::Namespace
+    attr_reader method: Symbol
+    attr_reader options: Hash[Symbol, untyped]
+
+    def initialize: (RBS::Namespace namespace, Symbol method, Hash[Symbol, untyped] options) -> void
+    def to: () -> Symbol
+    def method_name: () -> Symbol
+    def public?: () -> bool
+    def private?: () -> bool
+  end
+end

data/sig/rbs_activesupport/generator.rbs

@@ -0,0 +1,20 @@
+module RbsActivesupport
+  class Generator
+    def self.generate: (Pathname pathname, RBS::DefinitionBuilder rbs_builder) -> String?
+
+    include AST
+
+    attr_reader pathname: Pathname
+    attr_reader declaration_builder: DeclarationBuilder
+
+    def initialize: (Pathname pathname, RBS::DefinitionBuilder rbs_builder) -> void
+    def generate: () -> String?
+
+    private
+
+    def format: (String rbs) -> String
+    def parse_source_code: () -> Hash[RBS::Namespace, Array[Parser::MethodCall]]
+    def header: (RBS::Namespace namespace) -> String
+    def footer: (RBS::Namespace namespace) -> String
+  end
+end

data/sig/rbs_activesupport/include.rbs

@@ -0,0 +1,16 @@
+module RbsActivesupport
+  class Include
+    attr_reader context: RBS::Namespace
+    attr_reader module_path: Array[Symbol?]
+    attr_reader options: Hash[Symbol, untyped]
+
+    def initialize: (RBS::Namespace context, Array[Symbol?] module_path, Hash[Symbol, untyped] options) -> void
+    def argument: () -> RBS::Namespace
+    %a{pure}
+    def module_name: () -> RBS::Namespace?
+    def concern?: () -> bool
+    def classmethods?: () -> bool
+    def public?: () -> bool
+    def private?: () -> bool
+  end
+end

data/sig/rbs_activesupport/method_searcher.rbs

@@ -0,0 +1,12 @@
+module RbsActivesupport
+  class MethodSearcher
+    attr_reader rbs_builder: RBS::DefinitionBuilder
+
+    def initialize: (RBS::DefinitionBuilder rbs_builder) -> void
+    def method_types_for: (Delegate delegate) -> Array[String]
+
+    private
+
+    def lookup_method_types: (RBS::TypeName namespace, Symbol method) -> Array[RBS::MethodType]
+  end
+end

data/sig/rbs_activesupport/parser.rbs

@@ -0,0 +1,22 @@
+module RbsActivesupport
+  class Parser < ::RBS::Prototype::RB
+    class MethodCall
+      attr_reader name: t
+      attr_reader args: Array[RubyVM::AbstractSyntaxTree::Node]
+
+      @private: bool
+
+      def initialize: (t name, Array[RubyVM::AbstractSyntaxTree::Node] args, bool private) -> void
+      def private?: () -> bool
+    end
+
+    type t = :class_attribute | :delegate | :cattr_accessor | :mattr_accessor | :cattr_reader | :mattr_reader | :cattr_writer | :mattr_writer | :include
+    METHODS: Array[t]
+
+    alias process_orig process
+
+    attr_reader method_calls: Hash[RBS::Namespace, Array[MethodCall]]
+
+    def private?: (Array[RBS::AST::Declarations::t | RBS::AST::Members::t] decls) -> bool
+  end
+end