Graze\DataFile\Format\Formatter\CsvFormatter

Complexity

Total Complexity

16

Size/Duplication

Total Lines	141
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Test Coverage

Coverage

100%

Importance

Changes	3
Bugs	0	Features	1

8 Methods

Rating	Name	Duplication	Size	Complexity
A	escape()	0	4	1
A	encode()	0	4	1
A	getInitialBlock()	0	4	1
A	getRowSeparator()	0	4	1
A	getClosingBlock()	0	4	1
A	__construct()	0	12	2
B	buildReplacements()	0	26	6
A	format()	0	14	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\Formatter;

use Graze\DataFile\Format\CsvFormatInterface;
use Graze\DataFile\Format\Processor\BoolProcessor;
use Graze\DataFile\Format\Processor\DateTimeProcessor;
use Graze\DataFile\Format\Processor\ObjectToStringProcessor;
use Graze\DataFile\Format\Processor\RowProcessor;

class CsvFormatter implements FormatterInterface
{
    use RowProcessor;
    use InvokeFormatter;

    /**
     * @var CsvFormatInterface
     */
    private $csvFormat;

    /**
     * @var string[]
     */
    private $escapeChars;

    /**
     * @var string[]
     */
    private $replaceChars;

    /**
     * @var string
     */
    private $initial;

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

        $this->buildReplacements();

        $this->initial = (!is_null($this->csvFormat->getBom())) ? $this->csvFormat->getBom() : '';

        $this->addProcessor(new DateTimeProcessor());
        $this->addProcessor(new BoolProcessor());
        $this->addProcessor(new ObjectToStringProcessor());
    }

    /**
     * Build replacements to perform for each entry
     */
    private function buildReplacements()
    {
        if ($this->csvFormat->getEscapeCharacter()) {
            $this->escapeChars = [
                $this->csvFormat->getEscapeCharacter(), // escape escape first so that it doesn't re-escape later on
                $this->csvFormat->getDelimiter(),
                "\n",
                "\r",
                "\t",
            ];
            if ($this->csvFormat->hasQuotes() && !$this->csvFormat->isDoubleQuote()) {
                $this->escapeChars[] = $this->csvFormat->getQuoteCharacter();
            }

            $this->escapeChars = array_unique($this->escapeChars);

            $this->replaceChars = array_map(function ($char) {
                return $this->csvFormat->getEscapeCharacter() . $char;
            }, $this->escapeChars);
        }

        if ($this->csvFormat->hasQuotes() && $this->csvFormat->isDoubleQuote()) {
            $this->escapeChars[] = $this->csvFormat->getQuoteCharacter();
            $this->replaceChars[] = str_repeat($this->csvFormat->getQuoteCharacter(), 2);
        }
    }

    /**
     * @param array $data
     *
     * @return string
     */
    public function format(array $data)
    {
        $data = $this->process($data);

        foreach ($data as &$element) {
            if (is_null($element)) {
                $element = $this->csvFormat->getNullOutput();
            } else {
                $element = $this->csvFormat->getQuoteCharacter() . $this->escape($element) . $this->csvFormat->getQuoteCharacter();
            }
        }

        return $this->encode(implode($this->csvFormat->getDelimiter(), $data));
    }

    /**
     * @param string $string
     *
     * @return string
     */
    protected function escape($string)
    {
        return str_replace($this->escapeChars, $this->replaceChars, $string);
    }

    /**
     * @param string $string
     *
     * @return string
     */
    private function encode($string)
    {
        return mb_convert_encoding($string, $this->csvFormat->getEncoding());
    }

    /**
     * Return an initial block if required
     *
     * @return string
     */
    public function getInitialBlock()
    {
        return $this->initial;
    }

    /**
     * Get a separator between each row
     *
     * @return string
     */
    public function getRowSeparator()
    {
        return $this->encode($this->csvFormat->getLineTerminator());
    }

    /**
     * Return a closing block if required
     *
     * @return string
     */
    public function getClosingBlock()
    {
        return '';
    }
}