<?php
/**
 * Copyright 2023 Adobe
 * All Rights Reserved.
 */
declare(strict_types=1);

namespace Magento\CustomerGraphQl\Model\Resolver\Cache\Customer;

use Magento\Customer\Model\Data\Customer;
use Magento\Framework\EntityManager\HydratorPool;
use Magento\Framework\EntityManager\TypeResolver;
use Magento\GraphQlResolverCache\Model\Resolver\Result\DehydratorInterface;

/**
 * Customer resolver data dehydrator to create snapshot data necessary to restore model.
 */
class ModelDehydrator implements DehydratorInterface
{
    /**
     * @var TypeResolver
     */
    private TypeResolver $typeResolver;

    /**
     * @var HydratorPool
     */
    private HydratorPool $hydratorPool;

    /**
     * @param HydratorPool $hydratorPool
     * @param TypeResolver $typeResolver
     */
    public function __construct(
        HydratorPool $hydratorPool,
        TypeResolver $typeResolver
    ) {
        $this->typeResolver = $typeResolver;
        $this->hydratorPool = $hydratorPool;
    }

    /**
     * @inheritdoc
     */
    public function dehydrate(array &$resolvedValue): void
    {
        if (isset($resolvedValue['model']) && $resolvedValue['model'] instanceof Customer) {
            /** @var Customer $model */
            $model = $resolvedValue['model'];
            $entityType = $this->typeResolver->resolve($model);
            $resolvedValue['model_data'] = $this->hydratorPool->getHydrator($entityType)
                ->extract($model);
            $resolvedValue['model_entity_type'] = $entityType;
            $resolvedValue['model_id'] = $model->getId();
        }
    }
}
