# frozen_string_literal: true # typed: true module T::Types # Validates that an object belongs to the specified class. class Simple < Base attr_reader :raw_type def initialize(raw_type) @raw_type = raw_type end # overrides Base def name # Memoize to mitigate pathological performance with anonymous modules (https://bugs.ruby-lang.org/issues/11119) # # `name` isn't normally a hot path for types, but it is used in initializing a T::Types::Union, # and so in `T.nilable`, and so in runtime constructions like `x = T.let(nil, T.nilable(Integer))`. @name ||= @raw_type.name.freeze end # overrides Base def valid?(obj) obj.is_a?(@raw_type) end # overrides Base private def subtype_of_single?(other) case other when Simple @raw_type <= other.raw_type else false end end def to_nilable @nilable ||= T::Types::Union.new([self, T::Utils::Nilable::NIL_TYPE]) end module Private module Pool @cache = ObjectSpace::WeakMap.new def self.type_for_module(mod) cached = @cache[mod] return cached if cached type = if mod == ::Array T::Array[T.untyped] elsif mod == ::Hash T::Hash[T.untyped, T.untyped] elsif mod == ::Enumerable T::Enumerable[T.untyped] elsif mod == ::Enumerator T::Enumerator[T.untyped] elsif mod == ::Range T::Range[T.untyped] elsif !Object.autoload?(:Set) && Object.const_defined?(:Set) && mod == ::Set T::Set[T.untyped] else Simple.new(mod) end # Unfortunately, we still need to check if the module is frozen, # since WeakMap adds a finalizer to the key that is added # to the map, so that it can clear the map entry when the key is # garbage collected. # For a frozen object, though, adding a finalizer is not a valid # operation, so this still raises if `mod` is frozen. @cache[mod] = type unless mod.frozen? type end end end end end