Contents

# frozen_string_literal: true

require "rbs"

module RbsActivesupport
  class MethodSearcher
    attr_reader :rbs_builder #: RBS::DefinitionBuilder

    # @rbs rbs_builder: RBS::DefinitionBuilder
    def initialize(rbs_builder) #: void
      @rbs_builder = rbs_builder
    end

    # @rbs delegate: Delegate
    def method_types_for(delegate) #: Array[String]
      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 = return_type_names_for(delegate_to).uniq
                                                       .flat_map { |t| lookup_method_types(t, delegate.method) }
                                                       .map(&:to_s)
      return_types << "() -> untyped" if return_types.empty?
      return_types
    end

    private

    # @rbs delegate_to: Array[RBS::MethodType]
    def return_type_names_for(delegate_to) #: Array[RBS::TypeName]
      delegate_to.filter_map do |t|
        type_name_for(t.type.return_type)
      end
    end

    # @rbs type: RBS::Types::t
    def type_name_for(type) #: RBS::TypeName?
      case type
      when RBS::Types::Optional
        type_name_for(type.type)
      when RBS::Types::ClassSingleton, RBS::Types::ClassInstance, RBS::Types::Interface, RBS::Types::Alias
        type.name
      end
    end

    # @rbs type_name: RBS::TypeName
    # @rbs method: Symbol
    def lookup_method_types(type_name, method) #: Array[RBS::MethodType]
      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

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
rbs_activesupport-1.4.3	lib/rbs_activesupport/method_searcher.rb
rbs_activesupport-1.4.2	lib/rbs_activesupport/method_searcher.rb
rbs_activesupport-1.4.1	lib/rbs_activesupport/method_searcher.rb
rbs_activesupport-1.4.0	lib/rbs_activesupport/method_searcher.rb
rbs_activesupport-1.3.0	lib/rbs_activesupport/method_searcher.rb
rbs_activesupport-1.2.1	lib/rbs_activesupport/method_searcher.rb