Smart\EtlBundle\Extractor\AbstractFolderExtrator

Complexity

Total Complexity

6

Size/Duplication

Total Lines	56
Duplicated Lines	0 %

Test Coverage

Coverage

81.25%

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	getFolderToExtract()	0	3	1
A	setFolderToExtract()	0	3	1
A	check()	0	4	2
A	getFiles()	0	11	2

<?php

namespace Smart\EtlBundle\Extractor;

use Symfony\Component\Finder\Finder;
use Symfony\Component\Finder\SplFileInfo;

/**
 * Nicolas Bastien <nicolas.bastien@smartbooster.io>
 */
abstract class AbstractFolderExtrator extends AbstractExtractor
{
    /**
     * @var string
     */
    protected $folderToExtract;

    /**
     * @var string
     */
    abstract protected function getFileExtension();

    /**
     * @param string $filepath
     * @return mixed
     */
    abstract protected function extractFileContent($filepath);

    /**
     * @return string
     */
    public function getFolderToExtract()
    {
        return $this->folderToExtract;
    }

    /**
     * @param string $folderToExtract
     */
    public function setFolderToExtract($folderToExtract)
    {
        $this->folderToExtract = $folderToExtract;
    }

    protected function check()
    {
        if (!is_dir($this->folderToExtract)) {
            throw new \BadMethodCallException('Invalid folder to extract : ' . $this->folderToExtract);
        }
    }

    /**
     * @param string $extension
     * @return array
     */
    protected function getFiles($extension)
    {
        $finder = new Finder();
        $finder->in($this->folderToExtract);
        $files = [];
        /* @var SplFileInfo $file */
        foreach ($finder->files()->name('*.' . $extension) as $file) {
            $files[] = substr($file->getFilename(), 0, -(strlen($extension) + 1));
        }

        return array_combine($files, $files);
    }
}