pixelfederation/swoole-bundle

<?php

declare(strict_types=1);

namespace K911\Swoole\Bridge\Symfony\Cache;

use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\CompilerPass\StatefulServices\CompileProcessor;
use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\CompilerPass\StatefulServices\Proxifier;
use K911\Swoole\Bridge\Symfony\Bundle\DependencyInjection\ContainerConstants;
use K911\Swoole\Bridge\Symfony\Container\SimpleResetter;
use Symfony\Component\Cache\Adapter\AbstractAdapter;
use Symfony\Component\DependencyInjection\Argument\AbstractArgument;
use Symfony\Component\DependencyInjection\ContainerBuilder;
use Symfony\Component\DependencyInjection\Definition;

final class CacheAdapterProcessor implements CompileProcessor
{
    public function process(ContainerBuilder $container, Proxifier $proxifier): void
    {
        $resetterDefId = 'swoole_bundle.coroutines_support.cache_adapter_resetter';
        $resetterDef = new Definition(SimpleResetter::class);
        $resetterDef->setArguments(['reset']);
        $taggedCount = 0;

        foreach ($container->getDefinitions() as $definition) {
            try {
                if (!$definition->isAbstract() && is_subclass_of($definition->getClass(), AbstractAdapter::class)) {
                    foreach ($definition->getArguments() as $argument) {
                        if ($argument instanceof AbstractArgument) {
                            continue 2;
                        }
                    }

                    $definition->addTag(ContainerConstants::TAG_STATEFUL_SERVICE, ['resetter' => $resetterDefId]);
                    ++$taggedCount;
                }
            } catch (\Throwable $e) {
                // ignore
            }
        }

        if (0 === $taggedCount) {
            return;
        }

        $container->setDefinition($resetterDefId, $resetterDef);
    }
}