JMSSerializerBundle/DependencyInjection/Compiler/SetVisitorsPass.php

<?php

/*
 * Copyright 2011 Johannes M. Schmitt <schmittjoh@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

namespace JMS\SerializerBundle\DependencyInjection\Compiler;

use JMS\SerializerBundle\Exception\RuntimeException;
use Symfony\Component\DependencyInjection\ContainerBuilder;
use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;

class SetVisitorsPass implements CompilerPassInterface
{
    public function process(ContainerBuilder $container)
    {
        if (!$container->hasDefinition('jms_serializer.serializer')) {
            return;
        }

        $serializationVisitors = array();
        foreach ($container->findTaggedServiceIds('jms_serializer.serialization_visitor') as $id => $attributes) {
            if (!isset($attributes[0]['format'])) {
                throw new RuntimeException('You must set a "format" for the tag named "jms_serializer.serialization_visitor".');
            }

            $container
                ->getDefinition($id)
                ->setPublic(true)
            ;

            $serializationVisitors[$attributes[0]['format']] = $id;
        }

        $deserializationVisitors = array();
        foreach ($container->findTaggedServiceIds('jms_serializer.deserialization_visitor') as $id => $attributes) {
            if (!isset($attributes[0]['format'])) {
                throw new RuntimeException('You must set a "format" for the tag named "jms_serializer.deserialization_visitor".');
            }

            $container
                ->getDefinition($id)
                ->setPublic(true)
            ;

            $deserializationVisitors[$attributes[0]['format']] = $id;
        }

        foreach ($container->findTaggedServiceIds('jms_serializer.serializer') as $id => $attr) {
            $container
                ->getDefinition($id)
                ->replaceArgument(1, $serializationVisitors)
                ->replaceArgument(2, $deserializationVisitors)
            ;
        }
    }
}