class RBS::Types::Union

def ==(other)
  other.is_a?(Union) && other.types == types
end

def each_type(&block)
  if block
    types.each(&block)
  else
    enum_for :each_type
  end
end

def free_variables(set = Set.new)
  set.tap do
    types.each do |type|
      type.free_variables set
    end
  end
end

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

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

def hash
  self.class.hash ^ types.hash
end

def initialize(types:, location:)
  @types = types
  @location = location
end

def map_type(&block)
  if block
    Union.new(types: types.map(&block), location: location)
  else
    enum_for :map_type
  end
end

def map_type_name(&block)
  Union.new(
    types: types.map {|type| type.map_type_name(&block) },
    location: location
  )
end

def sub(s)
  return self if s.empty?
  self.class.new(types: types.map {|ty| ty.sub(s) },
                 location: location)
end

def to_json(state = _ = nil)
  { class: :union, types: types, location: location }.to_json(state)
end

def to_s(level = 0)
  strs = types.map do |ty|
    case ty
    when Intersection
      ty.to_s([1, level].max)
    else
      ty.to_s
    end
  end
  if level > 0
    "(#{strs.join(" | ")})"
  else
    strs.join(" | ")
  end
end

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