khepin / KhepinYamlFixturesBundle

Fixture/MongoYamlFixture.php

<?php

namespace Khepin\YamlFixturesBundle\Fixture;

use Doctrine\Common\Util\Inflector;

class MongoYamlFixture extends AbstractFixture
{
    /**
     * Creates and returns one object based on the given data and metadata
     *
     * @param $class object's class name
     * @param $data array of the object's fixture data
     * @param $metadata the class metadata for doctrine
     * @param $embedded true for embedded documents
     * @return Object
     */
    public function createObject($class, $data, $metadata, $options = array())
    {
        // options to state if a document is to be embedded or persisted on its own
        $embedded = isset($options['embedded']);
        $mapping = array_keys($metadata->fieldMappings);
        // Instantiate new object
        $object = $this->makeInstance($class, $data);
        unset($data['__construct']);

        foreach ($data as $field => $value) {
            // Add the fields defined in the fixtures file
            $method = Inflector::camelize('set_' . $field);
            // This is a standard field
            if (in_array($field, $mapping)) {
                // Dates need to be converted to DateTime objects
                $type = $metadata->fieldMappings[$field]['type'];
                if ($type == 'many') {
                    $method = Inflector::camelize('add_'.$field);
                    // EmbedMany
                    if (
                        isset($metadata->fieldMappings[$field]['embedded']) &&
                        $metadata->fieldMappings[$field]['embedded']) {
                        foreach ($value as $embedded_value) {
                            $embed_class = $metadata->fieldMappings[$field]['targetDocument'];
                            $embed_data = $embedded_value;
                            $embed_meta = $this->getMetaDataForClass($embed_class);
                            $value = $this->createObject($embed_class, $embed_data, $embed_meta, array(
                                'embedded' => true));
                            $object->$method($value);
                        }
                    //ReferenceMany
                    } else {
                        foreach ($value as $reference_object) {
                            $object->$method($this->loader->getReference($reference_object));
                        }
                    }
                } else {
                    if ($type == 'datetime' || $type == 'date' || $type == 'time') {

This code seems to be duplicated across your project.

                        $value = new \DateTime($value);
                    }
                    if ($type == 'one') {
                        // EmbedOne
                        if (
                            isset($metadata->fieldMappings[$field]['embedded']) &&
                            $metadata->fieldMappings[$field]['embedded']) {
                            $embed_class = $metadata->fieldMappings[$field]['targetDocument'];
                            $embed_data = $value;
                            $embed_meta = $this->getMetaDataForClass($embed_class);
                            $value = $this->createObject($embed_class, $embed_data, $embed_meta, array(
                                'embedded' => true));
                        // ReferenceOne
                        } else {
                            $value = $this->loader->getReference($value);
                        }
                    }
                    $object->$method($value);
                }
            } else {
                // The key is not a field's name but the name of a method to be called
                $object->$method($value);
            }
        }

        // Save a reference to the current object
        if (!$embedded) {
            $this->runServiceCalls($object);
        }

        return $object;
    }
}