class Steep::AST::Types::Union

def self.build(types:, location: nil)
  types.flat_map do |type|
    if type.is_a?(Union)
      type.types
    else
      [type]
    end
  end.map do |type|
    case type
    when AST::Types::Any
      return AST::Types::Any.new(location: location)
    when AST::Types::Top
      return AST::Types::Top.new(location: location)
    when AST::Types::Bot
      nil
    else
      type
    end
  end.compact.uniq.yield_self do |tys|
    if tys.length == 1
      tys.first
    else
      new(types: tys.sort_by(&:hash), location: location)
    end
  end
end

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

def free_variables
  types.each.with_object(Set.new) do |type, set|
    set.merge(type.free_variables)
  end
end

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

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

def level
  [0] + level_of_children(types)
end

def subst(s)
  self.class.build(location: location,
                   types: types.map {|ty| ty.subst(s) })
end

def to_s
  "(#{types.map(&:to_s).sort.join(" | ")})"
end

def with_location(new_location)
  self.class.new(types: types, location: new_location)
end