Class: RBS::DefinitionBuilder::MethodBuilder

# File 'lib/rbs/definition_builder/method_builder.rb', line 81

attr_reader :instance_methods

# File 'lib/rbs/definition_builder/method_builder.rb', line 83

attr_reader :interface_methods

# File 'lib/rbs/definition_builder/method_builder.rb', line 82

attr_reader :singleton_methods

# File 'lib/rbs/definition_builder/method_builder.rb', line 185

def build_attribute(methods, type, member:, accessibility:)
  if member.is_a?(AST::Members::AttrReader) || member.is_a?(AST::Members::AttrAccessor)
    defn = methods.methods[member.name] ||= Methods::Definition.empty(type: type, name: member.name)

    defn.accessibilities << accessibility
    defn.originals << member
  end

  if member.is_a?(AST::Members::AttrWriter) || member.is_a?(AST::Members::AttrAccessor)
    defn = methods.methods[:"#{member.name}="] ||= Methods::Definition.empty(type: type, name: :"#{member.name}=")

    defn.accessibilities << accessibility
    defn.originals << member
  end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 93

def build_instance(type_name)
  instance_methods[type_name] ||=
    begin
      entry = env.class_decls[type_name]
      args = Types::Variable.build(entry.type_params.each.map(&:name))
      type = Types::ClassInstance.new(name: type_name, args: args, location: nil)
      Methods.new(type: type).tap do |methods|
        entry.decls.each do |d|
          subst = Substitution.build(d.decl.type_params.each.map(&:name), args)
          each_member_with_accessibility(d.decl.members) do |member, accessibility|
            case member
            when AST::Members::MethodDefinition
              if member.instance?
                build_method(methods,
                             type,
                             member: member.update(types: member.types.map {|type| type.sub(subst) }),
                             accessibility: accessibility)
              end
            when AST::Members::AttrReader, AST::Members::AttrWriter, AST::Members::AttrAccessor
              if member.kind == :instance
                build_attribute(methods,
                                type,
                                member: member.update(type: member.type.sub(subst)),
                                accessibility: accessibility)
              end
            when AST::Members::Alias
              if member.kind == :instance
                build_alias(methods, type, member: member, accessibility: accessibility)
              end
            end
          end
        end
      end.validate!
    end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 158

def build_interface(type_name)
  interface_methods[type_name] ||=
    begin
      entry = env.interface_decls[type_name]
      args = Types::Variable.build(entry.decl.type_params.each.map(&:name))
      type = Types::Interface.new(name: type_name, args: args, location: nil)

      Methods.new(type: type).tap do |methods|
        entry.decl.members.each do |member|
          case member
          when AST::Members::MethodDefinition
            build_method(methods, type, member: member, accessibility: :public)
          when AST::Members::Alias
            build_alias(methods, type, member: member, accessibility: :public)
          end
        end
      end.validate!
    end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 201

def build_method(methods, type, member:, accessibility:)
  defn = methods.methods[member.name] ||= Methods::Definition.empty(type: type, name: member.name)

  if member.overload?
    defn.overloads << member
  else
    defn.accessibilities << accessibility
    defn.originals << member
  end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 129

def build_singleton(type_name)
  singleton_methods[type_name] ||=
    begin
      entry = env.class_decls[type_name]
      type = Types::ClassSingleton.new(name: type_name, location: nil)

      Methods.new(type: type).tap do |methods|
        entry.decls.each do |d|
          each_member_with_accessibility(d.decl.members) do |member, accessibility|
            case member
            when AST::Members::MethodDefinition
              if member.singleton?
                build_method(methods, type, member: member, accessibility: accessibility)
              end
            when AST::Members::AttrReader, AST::Members::AttrWriter, AST::Members::AttrAccessor
              if member.kind == :singleton
                build_attribute(methods, type, member: member, accessibility: accessibility)
              end
            when AST::Members::Alias
              if member.kind == :singleton
                build_alias(methods, type, member: member, accessibility: accessibility)
              end
            end
          end
        end
      end.validate!
    end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 212

def each_member_with_accessibility(members, accessibility: :public)
  members.each do |member|
    case member
    when AST::Members::Public
      accessibility = :public
    when AST::Members::Private
      accessibility = :private
    else
      yield member, accessibility
    end
  end
end

# File 'lib/rbs/definition_builder/method_builder.rb', line 225

def update(env:, except:)
  MethodBuilder.new(env: env).tap do |copy|
    copy.instance_methods.merge!(instance_methods)
    copy.singleton_methods.merge!(singleton_methods)
    copy.interface_methods.merge!(interface_methods)

    except.each do |type_name|
      copy.instance_methods.delete(type_name)
      copy.singleton_methods.delete(type_name)
      copy.interface_methods.delete(type_name)
    end
  end
end