79 lines
2.3 KiB
PHP
79 lines
2.3 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This file is part of the PropelBundle package.
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*
|
|
* @license MIT License
|
|
*/
|
|
|
|
namespace Propel\PropelBundle\DataFixtures\Loader;
|
|
|
|
use Symfony\Component\DependencyInjection\ContainerInterface;
|
|
use Symfony\Component\Yaml\ParseException;
|
|
use Symfony\Component\Yaml\Yaml;
|
|
|
|
/**
|
|
* YAML fixtures loader.
|
|
*
|
|
* @author William Durand <william.durand1@gmail.com>
|
|
*/
|
|
class YamlDataLoader extends AbstractDataLoader
|
|
{
|
|
/**
|
|
* @var \Symfony\Component\DependencyInjection\ContainerInterface
|
|
*/
|
|
private $container;
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function __construct($rootDir, ContainerInterface $container)
|
|
{
|
|
parent::__construct($rootDir, $container->get('propel'), $container->getParameter('propel.configuration'));
|
|
|
|
$this->container = $container;
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
protected function transformDataToArray($file)
|
|
{
|
|
if (strpos($file, "\n") === false && is_file($file)) {
|
|
if (false === is_readable($file)) {
|
|
throw new ParseException(sprintf('Unable to parse "%s" as the file is not readable.', $file));
|
|
}
|
|
|
|
if (null !== $this->container && $this->container->has('faker.generator')) {
|
|
$generator = $this->container->get('faker.generator');
|
|
$faker = function($type) use ($generator) {
|
|
$args = func_get_args();
|
|
array_shift($args);
|
|
|
|
echo Yaml::dump(call_user_func_array(array($generator, $type), $args)) . "\n";
|
|
};
|
|
} else {
|
|
$faker = function($text) {
|
|
echo $text . "\n";
|
|
};
|
|
}
|
|
|
|
ob_start();
|
|
$retval = include($file);
|
|
$content = ob_get_clean();
|
|
|
|
// if an array is returned by the config file assume it's in plain php form else in YAML
|
|
$file = is_array($retval) ? $retval : $content;
|
|
|
|
// if an array is returned by the config file assume it's in plain php form else in YAML
|
|
if (is_array($file)) {
|
|
return $file;
|
|
}
|
|
}
|
|
|
|
return Yaml::parse($file);
|
|
}
|
|
}
|