class RBS::MethodType

# File rbs-1.4.0/lib/rbs/method_type.rb, line 8
def initialize(type_params:, type:, block:, location:)
  @type_params = type_params
  @type = type
  @block = block
  @location = location
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 15
def ==(other)
  other.is_a?(MethodType) &&
    other.type_params == type_params &&
    other.type == type &&
    other.block == block
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 65
def each_type(&block)
  if block
    type.each_type(&block)
    self.block&.yield_self do |b|
      b.type.each_type(&block)
    end
  else
    enum_for :each_type
  end
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 48
def free_variables(set = Set.new)
  type.free_variables(set)
  block&.type&.free_variables(set)
  set.subtract(type_params)
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 54
def map_type(&block)
  self.class.new(
    type_params: type_params,
    type: type.map_type(&block),
    block: self.block&.yield_self do |b|
      Types::Block.new(type: b.type.map_type(&block), required: b.required)
    end,
    location: location
  )
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 31
def sub(s)
  s.without(*type_params).yield_self do |sub|
    map_type do |ty|
      ty.sub(sub)
    end
  end
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 22
def to_json(state = _ = nil)
  {
    type_params: type_params,
    type: type,
    block: block,
    location: location
  }.to_json(state)
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 76
def to_s
  s = case
      when (b = block) && b.required
        "(#{type.param_to_s}) { (#{b.type.param_to_s}) -> #{b.type.return_to_s} } -> #{type.return_to_s}"
      when b = block
        "(#{type.param_to_s}) ?{ (#{b.type.param_to_s}) -> #{b.type.return_to_s} } -> #{type.return_to_s}"
      else
        "(#{type.param_to_s}) -> #{type.return_to_s}"
      end

  if type_params.empty?
    s
  else
    "[#{type_params.join(", ")}] #{s}"
  end
end

# File rbs-1.4.0/lib/rbs/method_type.rb, line 39
def update(type_params: self.type_params, type: self.type, block: self.block, location: self.location)
  self.class.new(
    type_params: type_params,
    type: type,
    block: block,
    location: location
  )
end