<?php


namespace AutoAdmin\Bundle\BrickBuilder\Doctrine\EventListener;

use AutoAdmin\Bundle\BrickBuilder\Brick\BrickConfigManager;
use AutoAdmin\Bundle\BrickBuilder\Brick\Config\BrickConfiguration;
use AutoAdmin\Bundle\BrickBuilder\Brick\ConfigSetting\EntityCollectionSetting;
use AutoAdmin\Bundle\BrickBuilder\Brick\ConfigSetting\EntitySetting;
use AutoAdmin\Bundle\BrickBuilder\Brick\ConfigSetting\ScalarCollectionSetting;
use AutoAdmin\Bundle\BrickBuilder\Brick\ConfigSetting\ScalarSetting;
use AutoAdmin\Bundle\BrickBuilder\Brick\ConfigSetting\SubEntityCollectionSetting;
use AutoAdmin\Bundle\BrickBuilder\Exception\BadSettingFormatException;
use AutoAdmin\Bundle\BrickBuilder\Helper\DatabaseBrickFormat;
use AutoAdmin\Bundle\BrickBuilder\Model\AbstractBrickSubContent;
use AutoAdmin\Bundle\BrickBuilder\Model\BrickInterface;
use AutoAdmin\Bundle\Content\Model\SequencedInterface;
use Doctrine\Common\EventSubscriber;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Events;
use Doctrine\Persistence\Event\LifecycleEventArgs;
use Exception;
use ReflectionClass;
use RuntimeException;

class BrickListener implements EventSubscriber
{
    private array $trackedBricks = [];
    private array $toUpdate = [];
    private array $toRemove = [];

    public function __construct(
        protected BrickConfigManager $configManager,
        protected EntityManagerInterface $em,
        protected DatabaseBrickFormat $brickFormat,
    )
    {
    }


    public function getSubscribedEvents(): array
    {
        return [
            // load
            Events::postLoad,
            // update
            Events::preUpdate,
            Events::postUpdate,
            // persist
            Events::prePersist,
            Events::postPersist,
        ];
    }

    /**
     * @throws Exception
     */
    public function postLoad(LifecycleEventArgs $args): void
    {
        $object = $args->getObject();
        if ($object instanceof BrickInterface) {
            $this->retrieve($args);
        }
    }

    /**
     * @throws Exception
     */
    public function preUpdate(LifecycleEventArgs $args): void
    {
        $this->store($args);
    }

    public function postUpdate(LifecycleEventArgs $args): void
    {
        $this->removeObjects($args);
        $this->updateObjects($args);
    }

    /**
     * @throws Exception
     */
    public function prePersist(LifecycleEventArgs $args): void
    {
        $this->store($args);
    }

    public function postPersist(LifecycleEventArgs $args): void
    {
        $this->removeObjects($args);
        $this->updateObjects($args);
    }

    protected function updateObjects(LifecycleEventArgs $args): void
    {
        $object = $args->getObject();
        if ($object instanceof BrickInterface && $this->toUpdate) {
            foreach ($this->toUpdate as $key => $item) {
                $this->em->persist($item);
                unset($this->toUpdate[$key]);
            }
            $this->em->flush();
        }
    }

    protected function removeObjects(LifecycleEventArgs $args): void
    {
        $object = $args->getObject();
        if ($object instanceof BrickInterface && $this->toRemove) {
            foreach ($this->toRemove as $key => $item) {
                $this->em->remove($item);
                unset($this->toRemove[$key]);
            }
            $this->em->flush();
        }
    }

    /**
     * @throws Exception
     */
    protected function store(LifecycleEventArgs $args): void
    {
        $entity = $args->getObject();

        if ($entity instanceof BrickInterface) {
            $splHash = spl_object_hash($entity);
            // only handle brick storage once per flush, avoid recursion
            if (in_array($splHash, $this->trackedBricks)) {
                return;
            }

//            $changeSet = $this->em->getUnitOfWork()->getEntityChangeSet($entity);
//            if (empty($changeSet)) {
//                return;
//            }

            $this->trackedBricks[] = $splHash;

            $this->checkRemovedItems($entity);
            $this->toDatabaseFormat($entity);
        }
    }

    /**
     * @throws Exception
     */
    protected function retrieve(LifecycleEventArgs $args): void
    {
        $entity = $args->getObject();
        if ($entity instanceof BrickInterface) {
            $this->fromDatabaseFormat($entity);
        }
    }

    /**
     * @throws Exception
     */
    protected function checkRemovedItems(BrickInterface $brick): void
    {
        $changeSet = $this->em->getUnitOfWork()->getEntityChangeSet($brick);
        if (empty($changeSet['settings'])) {
            return;
        }
        $settingsChangeSet = $changeSet['settings'];
        [$before, $after] = $settingsChangeSet;
        /** @var BrickConfiguration $configuration */
        $configuration = $this->configManager->getHandlerConfiguration($brick->getType());
        $settingsConfig = $configuration->getSettings();
        $settings = $brick->getSettings();
        foreach ($settings as $key => $val) {
            if (!array_key_exists($key, $configuration->getSettings())) {
                continue;
            }
            $settingConfig = $settingsConfig[$key];
            switch ($currentType = $settingConfig->getType()) {
                case EntitySetting::class:
                    $old = $before[$key] ?? null;
                    $new = $this->brickFormat->objectToString($after[$key]);

                    // if we may have a divergence, try to get source object
                    if (!empty($old) && $new !== $old) {
                        if (is_string($old)) {
                            $old = $this->brickFormat->stringToObject($old);
                            if (null !== $old) {
                                $this->removeIfNeeded($old, $new);
                            }
                        }
                    }
                    break;
                case EntityCollectionSetting::class:
                case SubEntityCollectionSetting::class:
                    $val = array_filter($val);
                    $oldValues = [];
                    if (!empty($before[$key])) {
                        foreach ($before[$key] as $item) {
                            if (!is_string($item)) {
                                throw new BadSettingFormatException($brick, $key, 'in old content building');
                            }
                            $oldValues[] = $this->brickFormat->stringToObject($item);
                        }
                    }

                    $newValues = [];
                    foreach ($after[$key] as $item) {
                        $newValues[] = $item;
                    }

                    foreach ($oldValues as $item) {
                        $found = false;
                        foreach ($newValues as $newValue) {
                            if ($newValue === $item) {
                                $found = true;
                            }
                        }

                        if ($found) {
                            continue;
                        }

                        // only cascade delete objects for SubEntityCollectionSetting
                        if ($currentType === SubEntityCollectionSetting::class && $item) {
                            foreach ($val as $k => $v) {
                                if ($item === $v) {
                                    unset($val[$k]);
                                }
                            }
                            $this->toRemove[] = $item;
                        }
                    }
                    $settings[$key] = $val;
                    break;
                case ScalarSetting::class:
                case ScalarCollectionSetting::class:
                default:
                    break;
            }
        }
        $brick->setSettings($settings);
    }

    /**
     * Retrieve from database
     * @throws Exception
     */
    protected function fromDatabaseFormat(BrickInterface $brick): void
    {
        /** @var BrickConfiguration $configuration */
        $configuration = $this->configManager->getHandlerConfiguration($brick->getType());
        $settings = $brick->getSettings();

        foreach ($settings as $key => $val) {
            if (!array_key_exists($key, $settingsConfigs = $configuration->getSettings())) {
                continue;
            }
            $settingConfig = $settingsConfigs[$key];
            //$settings[$key] = $settingConfig->getType();
            switch ($settingConfig->getType()) {
                case EntitySetting::class:
                    if ($val !== null && !is_string($val)) {
                        throw new BadSettingFormatException($brick, $key);
                    }
                    $settings[$key] = $this->brickFormat->stringToObject($val);

                    break;
                case EntityCollectionSetting::class:
                case SubEntityCollectionSetting::class:
                    foreach ($val as $index => $obj) {
                        if (!is_string($obj)) {
                            throw new BadSettingFormatException($brick, $key);
                        }
                        $settings[$key][$index] = $this->brickFormat->stringToObject($obj);
                    }
                    break;
                case ScalarSetting::class:
                case ScalarCollectionSetting::class:
                default:
                    break;
            }
        }
        $brick->setSettings($settings);
    }

    /**
     * Storing to database
     * @throws Exception
     */
    protected function toDatabaseFormat(BrickInterface $brick): void
    {
        /** @var BrickConfiguration $configuration */
        $configuration = $this->configManager->getHandlerConfiguration($brick->getType());

        /** @var BrickInterface $child */
        $index = 0;
        foreach ($brick->getChildren() as $child) {
            $child->setPosition($index);
            $index++;
        }

        $settings = $brick->getSettings();

        foreach ($settings as $key => $val) {
            if (!array_key_exists($key, $settingsConfigs = $configuration->getSettings())) {
                continue;
            }

            // already transformed this brick, don't redo it.
            if (!empty($this->trackedBricks[spl_object_hash($brick)][$key])) {
                continue;
            }

            $settingConfig = $settingsConfigs[$key];

            switch ($settingConfig->getType()) {
                case EntitySetting::class:
                    if(is_object($val)) {
                        $settings[$key] = $this->toDatabaseValue($val);
                    }
                    break;
                case EntityCollectionSetting::class:
                case SubEntityCollectionSetting::class:

                    /*
                     * If already string-format, we don't need to transform it
                     */
                    foreach ($val as $index => $obj) {
                        if (!is_string($obj) && !is_object($obj)) {
                            throw new RuntimeException(
                                'Setting encountered during save was neither object nor string.'
                            );
                        }

                        $converted = is_object($obj) ? $this->toDatabaseValue($obj) : $obj;
                        $settings[$key][$index] = $converted;
                    }
                    $settings[$key] = array_filter(array_values($settings[$key]));// reindex

                    // try to order items following their sequence
                    uasort(
                        $settings[$key],
                        function ($a, $b) {
                            if (
                                !$a instanceof SequencedInterface
                                || !$b instanceof SequencedInterface
                            ) {
                                return 0;
                            }

                            return $a->getSequence() < $b->getSequence();
                        }
                    );
                    break;
                case ScalarCollectionSetting::class:
                    $settings[$key] = array_filter(array_values($val));// reindex
                    break;
                case ScalarSetting::class:
                default:
                    break;
            }

            // mark brick setting index as transformed to avoid reprocessing if multiple flush are occurring
            $this->trackedBricks[spl_object_hash($brick)][$key] = [$val, $settings[$key]];
        }

        $brick->setSettings($settings);
    }

    /**
     * @throws Exception
     */
    protected function toDatabaseValue($object): ?string
    {
        if (is_string($object)) {
            $object = $this->brickFormat->stringToObject($object);
        }

        if (is_null($object)) {
            return null;
        }

        $em = $this->em;
        if (empty($object->getId())) {
//            throw new \RuntimeException('Brick setting value of type "'.get_class($object).'" is not persisted.');

            // save un-persisted entity as we need its id to transform
            $em->persist($object);
            $em->flush();
            $em->refresh($object);
        } else {
            $this->toUpdate[] = $object;
        }

        return $this->brickFormat->objectToString($object);
    }

    /**
     * @throws Exception
     */
    private function removeIfNeeded(object $old, ?string $new): void
    {
        $refl = new ReflectionClass($old);
        if (!$refl->isSubclassOf(AbstractBrickSubContent::class)) {
            return;
        }

        // 2 cases:
        if ($new === null) {
            // - $old is an object with a given id and $new is null, always delete
            $this->toRemove[] = $old;
        } else {
            $new = $this->brickFormat->stringToObject($new);
            // - $old is an object with a given id and $new is an object with a different id, delete old only if id is different
            if ($old->getId() != $new->getId()) {
                $this->toRemove[] = $old;
            }
        }
    }
}
