use ApiPlatform\OpenApi\OpenApi;
use Symfony\Component\DependencyInjection\Attribute\Autowire;
use Symfony\Component\Serializer\Normalizer\NormalizerInterface;

class OpenApiNormalizer implements NormalizerInterface
{
    public function __construct(
        #[Autowire(service: 'api_platform.openapi.normalizer')]
        private readonly NormalizerInterface $normalizer,
    ) {
    }

    public function normalize(mixed $object, ?string $format = null, array $context = []): array|string|int|float|bool|\ArrayObject|null
    {
        /**
         * @var array{paths: array<string, array<string, array{tags?: string[], parameters: array<string, array{in: string, allowEmptyValue: bool, allowReserved: bool}>}>>, components: array{schemas: array<string, array<string, string>>}} $result
         */
        $result = $this->normalizer->normalize($object, $format, $context);

        if ($object instanceof OpenApi && '3' == $context['spec_version']) {
            // Remove allowEmptyValue & allowReserved on parameters see https://spec.openapis.org/oas/v3.1.0#fixed-fields-9
            foreach ($result['paths'] as &$path) {
                foreach ($path as &$method) {
                    foreach ($method['parameters'] as &$parameters) {
                        // allowEmptyValue & allowReserved only applicable for query parameters
                        if ('query' !== $parameters['in']) {
                            unset($parameters['allowEmptyValue']);
                            unset($parameters['allowReserved']);
                        }
                    }
                }
            }

            // Remove unused components, could be optimized with list of depencies per schema and progressive elemination of schema without dependencies
            do {
                $refs = [];
                $removed = 0;
                foreach (new \RecursiveIteratorIterator(new \RecursiveArrayIterator($result), \RecursiveIteratorIterator::LEAVES_ONLY) as $key => $value) {
                    if ('$ref' === $key) {
                        $refs[] = $value;
                    }
                }
                foreach ($result['components']['schemas'] as $key => $schema) {
                    if (!\in_array('#/components/schemas/'.$key, $refs)) {
                        unset($result['components']['schemas'][$key]);
                        ++$removed;
                    }
                }
            } while ($removed > 0);

            // Add global tags
            $tags = [];
            foreach ($result['paths'] as &$path) {
                foreach ($path as &$method) {
                    $tags = array_merge($tags, $method['tags'] ?? []);
                }
            }
            $result['tags'] = array_map(fn (string $tag) => [
                'name' => $tag,
            ], array_values(array_unique($tags)));
        }

        return $result;
    }

    public function supportsNormalization(mixed $data, ?string $format = null): bool
    {
        return $data instanceof OpenApi && 'json' === $format;
    }

    /**
     * @return array<class-string, bool>
     */
    public function getSupportedTypes(?string $format): array
    {
        return [
            OpenApi::class => true,
        ];
    }
}