Graze\DataFile\Format\Parser\CsvParser

Complexity

Total Complexity

14

Size/Duplication

Total Lines	105
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	4

Test Coverage

Coverage

100%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	parse()	0	8	1
A	parseIterator()	0	12	3
A	parseHeaderRow()	0	8	2
A	handleHeaderRow()	0	9	4
A	mapHeaders()	0	16	3

<?php
/**
 * This file is part of graze/data-file
 *
 * Copyright (c) 2016 Nature Delivered Ltd. <https://www.graze.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 *
 * @license https://github.com/graze/data-file/blob/master/LICENSE.md
 * @link    https://github.com/graze/data-file
 */

namespace Graze\DataFile\Format\Parser;

use CallbackFilterIterator;
use Graze\CsvToken\Parser;
use Graze\CsvToken\Tokeniser\StreamTokeniser;
use Graze\DataFile\Format\CsvFormatInterface;
use Graze\DataFile\Helper\MapIterator;
use Iterator;
use LimitIterator;
use RuntimeException;

class CsvParser implements ParserInterface
{
    /** @var CsvFormatInterface */
    private $csvFormat;
    /** @var array */
    private $headerRow = [];

    /**
     * @param CsvFormatInterface $csvFormat
     */
    public function __construct(CsvFormatInterface $csvFormat)
    {
        $this->csvFormat = $csvFormat;
    }

    /**
     * @param resource $stream
     *
     * @return Iterator
     */
    public function parse($stream)
    {
        $tokeniser = new StreamTokeniser($this->csvFormat, $stream);
        $parser = new Parser();
        return $this->parseIterator(
            $parser->parse($tokeniser->getTokens())
        );
    }

    /**
     * Parse a supplied iterator
     *
     * @param Iterator $iterator
     *
     * @return Iterator
     */
    private function parseIterator(Iterator $iterator)
    {
        $iterator = $this->parseHeaderRow($iterator);
        if ($this->csvFormat->getDataStart() > 1 || $this->csvFormat->getLimit() !== -1) {
            $iterator = new LimitIterator(
                $iterator,
                max(0, $this->csvFormat->getDataStart() - 1),
                $this->csvFormat->getLimit()
            );
        }
        return $iterator;
    }

    /**
     * @param Iterator $iterator
     *
     * @return Iterator
     */
    private function parseHeaderRow(Iterator $iterator)
    {
        if ($this->csvFormat->hasHeaderRow()) {
            $iterator = new CallbackFilterIterator($iterator, [$this, 'handleHeaderRow']);
            $iterator = new MapIterator($iterator, [$this, 'mapHeaders']);
        }
        return $iterator;
    }

    /**
     * Parse the rows looking for the header row and storing it locally
     *
     * @param array $current
     * @param int   $key
     *
     * @return bool
     */
    public function handleHeaderRow(array $current, $key)
    {
        if ($key == $this->csvFormat->getHeaderRow() - 1) {
            if (count($current) > 1 || strlen($current[0]) > 0) {
                $this->headerRow = $current;
            }
        }
        return true;
    }

    /**
     * Map any headers found onto each element in the data
     *
     * @param array $current
     *
     * @return array
     */
    public function mapHeaders(array $current)
    {
        if (count($this->headerRow) > 0) {
            if (count($this->headerRow) !== count($current)) {
                throw new RuntimeException(
                    "The number of entries in: " . implode(',', $current) .
                    " does not match the header: " . implode(',', $this->headerRow)
                );
            }
            return array_combine(
                $this->headerRow,
                $current
            );
        }
        return $current;
    }
}