Smart\EtlBundle\Extractor\YamlEntityExtractor

Complexity

Total Complexity

17

Size/Duplication

Total Lines	126
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

0%

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	getProcessEntities()	0	4	1
A	setProcessEntities()	0	4	1
A	extract()	0	14	3
A	check()	0	8	2
A	processFile()	0	13	3
A	processObject()	0	21	4
A	getEntity()	0	11	3

<?php

namespace Smart\EtlBundle\Extractor;

use Symfony\Component\PropertyAccess\PropertyAccessor;
use Symfony\Component\Yaml\Yaml;

/**
 * Nicolas Bastien <nicolas.bastien@smartbooster.io>
 */
class YamlEntityExtractor extends AbstractFolderExtrator implements ExtractorInterface
{
    /**
     * List of entities to extract
     * [
     *      'entity_code' => 'Model Classname'
     * ]
     * @var array
     */
    protected $processEntities;

    /**
     * @var array
     */
    protected $entities = [];

    /**
     * @var PropertyAccessor
     */
    protected $accessor;

    /**
     * @return array
     */
    public function getProcessEntities()
    {
        return $this->processEntities;
    }

    /**
     * @param array $processEntities
     */
    public function setProcessEntities($processEntities)
    {
        $this->processEntities = $processEntities;
    }

    /**
     * @inheritDoc
     */
    public function extract()
    {
        try {
            $this->check();


            $files = $this->getFiles();

The call to getFiles() misses a required argument $extension.

            foreach ($files as $filename) {
                $this->processFile($filename);
            }
        } catch (\Exception $e) {
            throw $e;
        }
    }
    
    protected function check()
    {
        parent::check();
        
        if (count($this->processEntities) === 0) {
            throw new \BadMethodCallException('Nothing to process');
        }
    }

    /**
     * @param string $filename
     */
    protected function processFile($filename)
    {
        $filepath   = sprintf('%s/%s.yml', $this->folderToExtract, $filename);

        $data = Yaml::parse(file_get_contents($filepath));
        foreach ($data as $values) {
            $object = $this->processObject($filename, $values);
            if ($object !== null) {
                //todo add dynamic identifier instead of code
                $this->entities[$object->getCode()] = $object;
            }
        }
    }

    /**
     * @param string $entityType
     * @param array $data
     * @return mixed
     * @throws \Exception
     */
    protected function processObject($entityType, array $data)
    {
        if (!isset($this->processEntities[$entityType])) {
            throw new \Exception('Entity type ' . $entityType . ' is not handled');
        }

        $objectClass = $this->processEntities[$entityType];
        $object = new $objectClass();

        foreach ($data as $key => $value) {
            $valueToSet = $value;
            if (strpos($value, '@') === 0) {
                //handle relations
                $valueToSet = $this->getEntity($value);
            }

            $this->accessor->setValue($object, $key, $valueToSet);
        }

        return $object;
    }

    /**
     * @param string $identifier
     * @return mixed
     * @throws \Exception
     */
    protected function getEntity($identifier)
    {
        if (strpos($identifier, '@') === 0) {
            $identifier = substr($identifier, 1);
        }
        if (!isset($this->entities[$identifier])) {
            throw new \Exception('Entity identifed by ' . $identifier . ' not found');
        }

        return $this->entities[$identifier];
    }
}