Class: RBS::Types::Function

# File 'lib/rbs/types.rb', line 1167

def has_classish_type?
  each_type.any? {|type| type.has_classish_type? }
end

# File 'lib/rbs/types.rb', line 1155

def has_keyword?
  if !required_keywords.empty? || !optional_keywords.empty? || rest_keywords
    true
  else
    false
  end
end

# File 'lib/rbs/types.rb', line 1163

def has_self_type?
  each_type.any? {|type| type.has_self_type? }
end

# File 'lib/rbs/types.rb', line 902

attr_reader :optional_keywords

# File 'lib/rbs/types.rb', line 898

attr_reader :optional_positionals

# File 'lib/rbs/types.rb', line 901

attr_reader :required_keywords

# File 'lib/rbs/types.rb', line 897

attr_reader :required_positionals

# File 'lib/rbs/types.rb', line 903

attr_reader :rest_keywords

# File 'lib/rbs/types.rb', line 899

attr_reader :rest_positionals

# File 'lib/rbs/types.rb', line 904

attr_reader :return_type

# File 'lib/rbs/types.rb', line 900

attr_reader :trailing_positionals

# File 'lib/rbs/types.rb', line 1171

def with_nonreturn_void?
  if each_param.any? {|param| param.type.with_nonreturn_void? }
    true
  else
    if return_type.is_a?(Bases::Void)
      false
    else
      return_type.with_nonreturn_void?
    end
  end
end

# File 'lib/rbs/types.rb', line 988

def amap(array, &block)
  if array.empty?
    _ = array
  else
    array.map(&block)
  end
end

# File 'lib/rbs/types.rb', line 1125

def drop_head
  case
  when !required_positionals.empty?
    [
      required_positionals[0],
      update(required_positionals: required_positionals.drop(1))
    ]
  when !optional_positionals.empty?
    [
      optional_positionals[0],
      update(optional_positionals: optional_positionals.drop(1))
    ]
  else
    raise "Cannot #drop_head"
  end
end

# File 'lib/rbs/types.rb', line 1142

def drop_tail
  case
  when !trailing_positionals.empty?
    last = trailing_positionals.last or raise
    [
      last,
      update(trailing_positionals: trailing_positionals.take(trailing_positionals.size - 1))
    ]
  else
    raise "Cannot #drop_tail"
  end
end

# File 'lib/rbs/types.rb', line 1025

def each_param(&block)
  if block
    required_positionals.each(&block)
    optional_positionals.each(&block)
    rest_positionals&.yield_self(&block)
    trailing_positionals.each(&block)
    required_keywords.each_value(&block)
    optional_keywords.each_value(&block)
    rest_keywords&.yield_self(&block)
  else
    enum_for :each_param
  end
end

# File 'lib/rbs/types.rb', line 1010

def each_type
  if block_given?
    required_positionals.each {|param| yield param.type }
    optional_positionals.each {|param| yield param.type }
    rest_positionals&.yield_self {|param| yield param.type }
    trailing_positionals.each {|param| yield param.type }
    required_keywords.each_value {|param| yield param.type }
    optional_keywords.each_value {|param| yield param.type }
    rest_keywords&.yield_self {|param| yield param.type }
    yield(return_type)
  else
    enum_for :each_type
  end
end

# File 'lib/rbs/types.rb', line 929

alias eql? ==

# File 'lib/rbs/types.rb', line 943

def free_variables(set = Set.new)
  set.tap do
    required_positionals.each do |param|
      param.type.free_variables(set)
    end
    optional_positionals.each do |param|
      param.type.free_variables(set)
    end
    rest_positionals&.yield_self do |param|
      param.type.free_variables(set)
    end
    trailing_positionals.each do |param|
      param.type.free_variables(set)
    end
    required_keywords.each_value do |param|
      param.type.free_variables(set)
    end
    optional_keywords.each_value do |param|
      param.type.free_variables(set)
    end
    rest_keywords&.yield_self do |param|
      param.type.free_variables(set)
    end

    return_type.free_variables(set)
  end
end

# File 'lib/rbs/types.rb', line 931

def hash
  self.class.hash ^
    required_positionals.hash ^
    optional_positionals.hash ^
    rest_positionals.hash ^
    trailing_positionals.hash ^
    required_keywords.hash ^
    optional_keywords.hash ^
    rest_keywords.hash ^
    return_type.hash
end

# File 'lib/rbs/types.rb', line 996

def hmapv(hash, &block)
  if hash.empty?
    _ = hash
  else
    hash.transform_values(&block)
  end
end

# File 'lib/rbs/types.rb', line 971

def map_type(&block)
  if block
    Function.new(
      required_positionals: amap(required_positionals) {|param| param.map_type(&block) },
      optional_positionals: amap(optional_positionals) {|param| param.map_type(&block) },
      rest_positionals: rest_positionals&.yield_self {|param| param.map_type(&block) },
      trailing_positionals: amap(trailing_positionals) {|param| param.map_type(&block) },
      required_keywords: hmapv(required_keywords) {|param| param.map_type(&block) },
      optional_keywords: hmapv(optional_keywords) {|param| param.map_type(&block) },
      rest_keywords: rest_keywords&.yield_self {|param| param.map_type(&block) },
      return_type: yield(return_type)
    )
  else
    enum_for :map_type
  end
end

# File 'lib/rbs/types.rb', line 1004

def map_type_name(&block)
  map_type do |type|
    type.map_type_name(&block)
  end
end

# File 'lib/rbs/types.rb', line 1106

def param_to_s
  # @type var params: Array[String]
  params = []

  params.push(*required_positionals.map(&:to_s))
  params.push(*optional_positionals.map {|p| "?#{p}"})
  params.push("*#{rest_positionals}") if rest_positionals
  params.push(*trailing_positionals.map(&:to_s))
  params.push(*required_keywords.map {|name, param| "#{name}: #{param}" })
  params.push(*optional_keywords.map {|name, param| "?#{name}: #{param}" })
  params.push("**#{rest_keywords}") if rest_keywords

  params.join(", ")
end

# File 'lib/rbs/types.rb', line 1121

def return_to_s
  return_type.to_s(1)
end

# File 'lib/rbs/types.rb', line 1052

def sub(s)
  map_type {|ty| ty.sub(s) }
end

# File 'lib/rbs/types.rb', line 1039

def to_json(state = _ = nil)
  {
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: return_type
  }.to_json(state)
end

# File 'lib/rbs/types.rb', line 1082

def update(required_positionals: self.required_positionals, optional_positionals: self.optional_positionals, rest_positionals: self.rest_positionals, trailing_positionals: self.trailing_positionals,
           required_keywords: self.required_keywords, optional_keywords: self.optional_keywords, rest_keywords: self.rest_keywords, return_type: self.return_type)
  Function.new(
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: return_type
  )
end

# File 'lib/rbs/types.rb', line 1069

def with_return_type(type)
  Function.new(
    required_positionals: required_positionals,
    optional_positionals: optional_positionals,
    rest_positionals: rest_positionals,
    trailing_positionals: trailing_positionals,
    required_keywords: required_keywords,
    optional_keywords: optional_keywords,
    rest_keywords: rest_keywords,
    return_type: type
  )
end