EntityData.php

<?php

declare (strict_types=1);
namespace mglaman\PHPStanDrupal\Drupal;

use Drupal\Core\Config\Entity\ConfigEntityInterface;
use Drupal\Core\Config\Entity\ConfigEntityStorageInterface;
use Drupal\Core\Entity\ContentEntityInterface;
use Drupal\Core\Entity\ContentEntityStorageInterface;
use Drupal\Core\Entity\EntityStorageInterface;
use mglaman\PHPStanDrupal\Type\EntityStorage\ConfigEntityStorageType;
use mglaman\PHPStanDrupal\Type\EntityStorage\ContentEntityStorageType;
use mglaman\PHPStanDrupal\Type\EntityStorage\EntityStorageType;
use PHPStan\Type\ObjectType;
final class EntityData {
    
    /**
     * @var string
     */
    private $entityTypeId;
    
    /**
     * @var string|null
     */
    private $className;
    
    /**
     * @var string|null
     */
    private $storageClassName;
    public function __construct(string $entityTypeId, array $definition) {
        $this->entityTypeId = $entityTypeId;
        $this->className = $definition['class'] ?? null;
        $this->storageClassName = $definition['storage'] ?? null;
    }
    public function getClassType() : ?ObjectType {
        return $this->className === null ? null : new ObjectType($this->className);
    }
    public function getStorageType() : ?EntityStorageType {
        if ($this->storageClassName === null) {
            $classType = $this->getClassType();
            if ($classType === null) {
                return null;
            }
            if ((new ObjectType(ConfigEntityInterface::class))->isSuperTypeOf($classType)
                ->yes()) {
                $this->storageClassName = 'Drupal\\Core\\Config\\Entity\\ConfigEntityStorage';
            }
            elseif ((new ObjectType(ContentEntityInterface::class))->isSuperTypeOf($classType)
                ->yes()) {
                $this->storageClassName = 'Drupal\\Core\\Entity\\Sql\\SqlContentEntityStorage';
            }
            else {
                return null;
            }
        }
        $storageType = new ObjectType($this->storageClassName);
        if ((new ObjectType(EntityStorageInterface::class))->isSuperTypeOf($storageType)
            ->no()) {
            return null;
        }
        if ((new ObjectType(ConfigEntityStorageInterface::class))->isSuperTypeOf($storageType)
            ->yes()) {
            return new ConfigEntityStorageType($this->entityTypeId, $this->storageClassName);
        }
        if ((new ObjectType(ContentEntityStorageInterface::class))->isSuperTypeOf($storageType)
            ->yes()) {
            return new ContentEntityStorageType($this->entityTypeId, $this->storageClassName);
        }
        return new EntityStorageType($this->entityTypeId, $this->storageClassName);
    }

}