class ClassDiscriminatorFromClassMetadata

class ClassDiscriminatorFromClassMetadata implements ClassDiscriminatorResolverInterface {
    private array $mappingForMappedObjectCache = [];
    public function __construct(ClassMetadataFactoryInterface $classMetadataFactory) {
    }
    public function getMappingForClass(string $class) : ?ClassDiscriminatorMapping {
        if ($this->classMetadataFactory
            ->hasMetadataFor($class)) {
            return $this->classMetadataFactory
                ->getMetadataFor($class)
                ->getClassDiscriminatorMapping();
        }
        return null;
    }
    public function getMappingForMappedObject(object|string $object) : ?ClassDiscriminatorMapping {
        if ($this->classMetadataFactory
            ->hasMetadataFor($object)) {
            $metadata = $this->classMetadataFactory
                ->getMetadataFor($object);
            if (null !== $metadata->getClassDiscriminatorMapping()) {
                return $metadata->getClassDiscriminatorMapping();
            }
        }
        $cacheKey = \is_object($object) ? $object::class : $object;
        if (!\array_key_exists($cacheKey, $this->mappingForMappedObjectCache)) {
            $this->mappingForMappedObjectCache[$cacheKey] = $this->resolveMappingForMappedObject($object);
        }
        return $this->mappingForMappedObjectCache[$cacheKey];
    }
    public function getTypeForMappedObject(object|string $object) : ?string {
        if (null === ($mapping = $this->getMappingForMappedObject($object))) {
            return null;
        }
        return $mapping->getMappedObjectType($object);
    }
    private function resolveMappingForMappedObject(object|string $object) : ?ClassDiscriminatorMapping {
        $reflectionClass = new \ReflectionClass($object);
        if ($parentClass = $reflectionClass->getParentClass()) {
            if (null !== ($parentMapping = $this->getMappingForMappedObject($parentClass->getName()))) {
                return $parentMapping;
            }
        }
        foreach ($reflectionClass->getInterfaceNames() as $interfaceName) {
            if (null !== ($interfaceMapping = $this->getMappingForMappedObject($interfaceName))) {
                return $interfaceMapping;
            }
        }
        return null;
    }

}