symfony/src/Symfony/Components/DependencyInjection/Loader/YamlFileLoader.php

<?php

namespace Symfony\Components\DependencyInjection\Loader;

use Symfony\Components\DependencyInjection\Container;
use Symfony\Components\DependencyInjection\Definition;
use Symfony\Components\DependencyInjection\Reference;
use Symfony\Components\DependencyInjection\BuilderConfiguration;
use Symfony\Components\YAML\YAML;

/*
 * This file is part of the symfony framework.
 *
 * (c) Fabien Potencier <fabien.potencier@symfony-project.com>
 *
 * This source file is subject to the MIT license that is bundled
 * with this source code in the file LICENSE.
 */

/**
 * YamlFileLoader loads YAML files service definitions.
 *
 * The YAML format does not support anonymous services (cf. the XML loader).
 *
 * @package    symfony
 * @subpackage dependency_injection
 * @author     Fabien Potencier <fabien.potencier@symfony-project.com>
 */
class YamlFileLoader extends FileLoader
{
  /**
   * Loads an array of XML files.
   *
   * @param  array $files An array of XML files
   *
   * @return array An array of definitions and parameters
   */
  public function load($files)
  {
    if (!is_array($files))
    {
      $files = array($files);
    }

    return $this->parse($this->getFilesAsArray($files));
  }

  protected function parse($data)
  {
    $configuration = new BuilderConfiguration();

    foreach ($data as $file => $content)
    {
      if (!$content)
      {
        continue;
      }

      // imports
      $this->parseImports($configuration, $content, $file);

      // parameters
      if (isset($content['parameters']))
      {
        foreach ($content['parameters'] as $key => $value)
        {
          $configuration->setParameter(strtolower($key), $this->resolveServices($value));
        }
      }

      // services
      $this->parseDefinitions($configuration, $content, $file);

      // extensions
      $this->loadFromExtensions($configuration, $content);
    }

    return $configuration;
  }

  protected function parseImports(BuilderConfiguration $configuration, $content, $file)
  {
    if (!isset($content['imports']))
    {
      return;
    }

    foreach ($content['imports'] as $import)
    {
      $configuration->merge($this->parseImport($import, $file));
    }
  }

  protected function parseImport($import, $file)
  {
    if (isset($import['class']) && $import['class'] != get_class($this))
    {
      $class = $import['class'];
      $loader = new $class($this->paths);
    }
    else
    {
      $loader = $this;
    }

    $importedFile = $this->getAbsolutePath($import['resource'], dirname($file));

    return $loader->load($importedFile);
  }

  protected function parseDefinitions(BuilderConfiguration $configuration, $content, $file)
  {
    if (!isset($content['services']))
    {
      return;
    }

    foreach ($content['services'] as $id => $service)
    {
      $this->parseDefinition($configuration, $id, $service, $file);
    }
  }

  protected function parseDefinition(BuilderConfiguration $configuration, $id, $service, $file)
  {
    if (is_string($service) && 0 === strpos($service, '@'))
    {
      $configuration->setAlias($id, substr($service, 1));

      return;
    }

    $definition = new Definition($service['class']);

    if (isset($service['shared']))
    {
      $definition->setShared($service['shared']);
    }

    if (isset($service['constructor']))
    {
      $definition->setConstructor($service['constructor']);
    }

    if (isset($service['file']))
    {
      $definition->setFile($service['file']);
    }

    if (isset($service['arguments']))
    {
      $definition->setArguments($this->resolveServices($service['arguments']));
    }

    if (isset($service['configurator']))
    {
      if (is_string($service['configurator']))
      {
        $definition->setConfigurator($service['configurator']);
      }
      else
      {
        $definition->setConfigurator(array($this->resolveServices($service['configurator'][0]), $service['configurator'][1]));
      }
    }

    if (isset($service['calls']))
    {
      foreach ($service['calls'] as $call)
      {
        $definition->addMethodCall($call[0], $this->resolveServices($call[1]));
      }
    }

    $configuration->setDefinition($id, $definition);
  }

  protected function getFilesAsArray(array $files)
  {
    $yamls = array();
    foreach ($files as $file)
    {
      $path = $this->getAbsolutePath($file);

      if (!file_exists($path))
      {
        throw new \InvalidArgumentException(sprintf('The service file "%s" does not exist (in: %s).', $file, implode(', ', $this->paths)));
      }

      $yamls[$path] = $this->validate(YAML::load($path), $path);
    }

    return $yamls;
  }

  protected function validate($content, $file)
  {
    if (null === $content)
    {
      return $content;
    }

    if (!is_array($content))
    {
      throw new \InvalidArgumentException(sprintf('The service file "%s" is not valid.', $file));
    }

    foreach (array_keys($content) as $key)
    {
      if (in_array($key, array('imports', 'parameters', 'services')))
      {
        continue;
      }

      // can it be handled by an extension?
      if (false !== strpos($key, '.'))
      {
        list($namespace, $tag) = explode('.', $key);
        if (!static::getExtension($namespace))
        {
          throw new \InvalidArgumentException(sprintf('There is no extension able to load the configuration for "%s" (in %s).', $key, $file));
        }

        continue;
      }

      throw new \InvalidArgumentException(sprintf('The "%s" tag is not valid (in %s).', $key, $file));
    }

    return $content;
  }

  protected function resolveServices($value)
  {
    if (is_array($value))
    {
      $value = array_map(array($this, 'resolveServices'), $value);
    }
    else if (is_string($value) && 0 === strpos($value, '@@'))
    {
      $value = new Reference(substr($value, 2), Container::IGNORE_ON_INVALID_REFERENCE);
    }
    else if (is_string($value) && 0 === strpos($value, '@'))
    {
      $value = new Reference(substr($value, 1));
    }

    return $value;
  }

  protected function loadFromExtensions(BuilderConfiguration $configuration, $content)
  {
    foreach ($content as $key => $config)
    {
      if (in_array($key, array('imports', 'parameters', 'services')))
      {
        continue;
      }

      list($namespace, $tag) = explode('.', $key);

      if (!is_array($config))
      {
        $config = array();
      }

      $configuration->merge(static::getExtension($namespace)->load($tag, $config));
    }
  }
}