Writer::addRecord() - Code Metrics - Inspection of "bug fix #307" - thephpleague/csv - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Pull Request — master (#309)

by ignace nyamagana

created 2018-09-10 16:10 UTC

Writer::addRecord() A

↳ Parent: Writer

Complexity

Conditions	1
Paths	1

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	2
CRAP Score	1

Importance

Changes

Metric	Value
cc	1
nc	1
nop	1
dl	0
loc	4
ccs	2
cts	2
cp	1
crap	1
rs	10
c	0
b	0
f	0

<?php

/**
 * League.Csv (https://csv.thephpleague.com).
 *
 * @author  Ignace Nyamagana Butera <[email protected]>
 * @license https://github.com/thephpleague/csv/blob/master/LICENSE (MIT License)
 * @version 9.1.5
 * @link    https://github.com/thephpleague/csv
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

declare(strict_types=1);

namespace League\Csv;

use Traversable;
use TypeError;
use const SEEK_CUR;
use const STREAM_FILTER_WRITE;
use function array_reduce;
use function gettype;
use function implode;
use function is_iterable;
use function sprintf;
use function str_replace;
use function strlen;

/**
 * A class to insert records into a CSV Document.
 *
 * @package League.csv
 * @since   4.0.0
 * @author  Ignace Nyamagana Butera <[email protected]>
 */
class Writer extends AbstractCsv
{
    const MODE_PHP = 'MODE_PHP';

    const MODE_RFC4180 = 'MODE_RFC4180';

    /**
     * callable collection to format the record before insertion.
     *
     * @var callable[]
     */
    protected $formatters = [];

    /**
     * callable collection to validate the record before insertion.
     *
     * @var callable[]
     */
    protected $validators = [];

    /**
     * newline character.
     *
     * @var string
     */
    protected $newline = "\n";

    /**
     * Insert records count for flushing.
     *
     * @var int
     */
    protected $flush_counter = 0;

    /**
     * Buffer flush threshold.
     *
     * @var int|null
     */
    protected $flush_threshold;

    /**
     * {@inheritdoc}
     */
    protected $stream_filter_mode = STREAM_FILTER_WRITE;

    /**
     * Regular expression used to detect if RFC4180 formatting is necessary.
     *
     * @var string
     */
    protected $rfc4180_regexp;

    /**
     * double enclosure for RFC4180 compliance.
     *
     * @var string
     */
    protected $rfc4180_enclosure;

    /**
     * Returns the current newline sequence characters.
     */
    public function getNewline(): string
    {
        return $this->newline;
    }

    /**
     * Get the flush threshold.
     *
     * @return int|null
     */
    public function getFlushThreshold()
    {
        return $this->flush_threshold;
    }

    /**
     * Adds multiple records to the CSV document.
     *
     * @see Writer::insertOne
     *
     * @param Traversable|array $records
     */
    public function insertAll($records, string $mode = self::MODE_PHP): int
    {
        if (!is_iterable($records)) {
            throw new TypeError(sprintf('%s() expects argument passed to be iterable, %s given', __METHOD__, gettype($records)));
        }

        $bytes = 0;
        foreach ($records as $record) {
            $bytes += $this->insertOne($record, $mode);
        }

        $this->flush_counter = 0;
        $this->document->fflush();

        return $bytes;
    }

    /**
     * Adds a single record to a CSV document.
     *
     * A record is an array that can contains scalar types values, NULL values
     * or objects implementing the __toString method.
     *
     * @throws CannotInsertRecord If the record can not be inserted
     */
    public function insertOne(array $record, string $mode = self::MODE_PHP): int
    {
        static $methodList = [self::MODE_PHP => 'addRecord', self::MODE_RFC4180 => 'addRFC4180CompliantRecord'];
        $method = $methodList[$mode] ?? null;
        if (null === $method) {
            throw new Exception(sprintf('Unknown or unsupported writing mode %s', $mode));
        }

        $record = array_reduce($this->formatters, [$this, 'formatRecord'], $record);
        $this->validateRecord($record);
        $bytes = $this->$method($record);
        if (false !== $bytes && 0 !== $bytes) {
            return $bytes + $this->consolidate();
        }

        throw CannotInsertRecord::triggerOnInsertion($record);
    }

    /**
     * Adds a single record to a CSV Document using PHP algorithm.
     *
     * @see https://php.net/manual/en/function.fputcsv.php
     *
     * @return int|bool
     */
    protected function addRecord(array $record)
    {
        return $this->document->fputcsv($record, $this->delimiter, $this->enclosure, $this->escape);
    }

    /**
     * Adds a single record to a CSV Document using RFC4180 algorithm.
     *
     * @see https://php.net/manual/en/function.fputcsv.php
     * @see https://php.net/manual/en/function.fwrite.php
     * @see https://tools.ietf.org/html/rfc4180
     * @see http://edoceo.com/utilitas/csv-file-format
     *
     * Fields must be delimited with enclosures if they contains :
     *
     *     - Leading or trailing whitespaces
     *     - Embedded delimiters
     *     - Embedded line-breaks
     *     - Embedded enclosures.
     *
     * Embedded enclosures must be doubled.
     *
     * String conversion is done without any check like fputcsv:
     *
     *     - Emits E_NOTICE on Array conversion (returns the 'Array' string)
     *     - Throws catchable fatal error on objects that can not be converted
     *     - Returns resource id without notice or error (returns 'Resource id #2')
     *     - Converts boolean true to '1', boolean false to the empty string
     *     - Converts null value to the empty string
     *
     * the LF character is added at the end of each record to mimic fputcsv behavior
     *
     * @return int|bool
     */
    protected function addRFC4180CompliantRecord(array $record)
    {
        foreach ($record as &$field) {
            $field = (string) $field;
            if (preg_match($this->rfc4180_regexp, $field)) {
                $field = $this->enclosure.str_replace($this->enclosure, $this->rfc4180_enclosure, $field).$this->enclosure;
            }
        }
        unset($field);

        return $this->document->fwrite(implode($this->delimiter, $record)."\n");
    }

    /**
     * {@inheritdoc}
     */
    protected function resetProperties()
    {
        parent::resetProperties();
        $embedded_characters = "\n|\r".preg_quote($this->delimiter, '/').'|'.preg_quote($this->enclosure, '/');
        $this->rfc4180_regexp = '/^
            (\ +)                          # leading whitespaces
            |(['.$embedded_characters.'])  # embedded characters
            |(\ +)                         # trailing whitespaces
        $/x';
        $this->rfc4180_enclosure = $this->enclosure.$this->enclosure;
    }

    /**
     * Format a record.
     *
     * The returned array must contain
     *   - scalar types values,
     *   - NULL values,
     *   - or objects implementing the __toString() method.
     */
    protected function formatRecord(array $record, callable $formatter): array
    {
        return $formatter($record);
    }

    /**
     * Validate a record.
     *
     * @throws CannotInsertRecord If the validation failed
     */
    protected function validateRecord(array $record)
    {
        foreach ($this->validators as $name => $validator) {
            if (true !== $validator($record)) {
                throw CannotInsertRecord::triggerOnValidation($name, $record);
            }
        }
    }

    /**
     * Apply post insertion actions.
     */
    protected function consolidate(): int
    {
        $bytes = 0;
        if ("\n" !== $this->newline) {
            $this->document->fseek(-1, SEEK_CUR);
            $bytes = $this->document->fwrite($this->newline, strlen($this->newline)) - 1;
        }

        if (null === $this->flush_threshold) {
            return $bytes;
        }

        ++$this->flush_counter;
        if (0 === $this->flush_counter % $this->flush_threshold) {
            $this->flush_counter = 0;
            $this->document->fflush();
        }

        return $bytes;
    }

    /**
     * Adds a record formatter.
     */
    public function addFormatter(callable $formatter): self
    {
        $this->formatters[] = $formatter;

        return $this;
    }

    /**
     * Adds a record validator.
     */
    public function addValidator(callable $validator, string $validator_name): self
    {
        $this->validators[$validator_name] = $validator;

        return $this;
    }

    /**
     * Sets the newline sequence.
     */
    public function setNewline(string $newline): self
    {
        $this->newline = $newline;

        return $this;
    }

    /**
     * Set the flush threshold.
     *
     * @param int|null $threshold
     *
     * @throws Exception if the threshold is a integer lesser than 1
     */
    public function setFlushThreshold($threshold): self
    {
        if ($threshold === $this->flush_threshold) {
            return $this;
        }

        if (!is_nullable_int($threshold)) {
            throw new TypeError(sprintf(__METHOD__.'() expects 1 Argument to be null or an integer %s given', gettype($threshold)));
        }

        if (null !== $threshold && 1 > $threshold) {
            throw new Exception(__METHOD__.'() expects 1 Argument to be null or a valid integer greater or equal to 1');
        }

        $this->flush_threshold = $threshold;
        $this->flush_counter = 0;
        $this->document->fflush();

        return $this;
    }
}


1		<?php
2
3		/**
4		* League.Csv (https://csv.thephpleague.com).
5		*
6		* @author Ignace Nyamagana Butera <[email protected]>
7		* @license https://github.com/thephpleague/csv/blob/master/LICENSE (MIT License)
8		* @version 9.1.5
9		* @link https://github.com/thephpleague/csv
10		*
11		* For the full copyright and license information, please view the LICENSE
12		* file that was distributed with this source code.
13		*/
14
15		declare(strict_types=1);
16
17		namespace League\Csv;
18
19		use Traversable;
20		use TypeError;
21		use const SEEK_CUR;
22		use const STREAM_FILTER_WRITE;
23		use function array_reduce;
24		use function gettype;
25		use function implode;
26		use function is_iterable;
27		use function sprintf;
28		use function str_replace;
29		use function strlen;
30
31		/**
32		* A class to insert records into a CSV Document.
33		*
34		* @package League.csv
35		* @since 4.0.0
36		* @author Ignace Nyamagana Butera <[email protected]>
37		*/
38		class Writer extends AbstractCsv
39		{
40		const MODE_PHP = 'MODE_PHP';
41
42		const MODE_RFC4180 = 'MODE_RFC4180';
43
44		/**
45		* callable collection to format the record before insertion.
46		*
47		* @var callable[]
48		*/
49		protected $formatters = [];
50
51		/**
52		* callable collection to validate the record before insertion.
53		*
54		* @var callable[]
55		*/
56		protected $validators = [];
57
58		/**
59		* newline character.
60		*
61		* @var string
62		*/
63		protected $newline = "\n";
64
65		/**
66		* Insert records count for flushing.
67		*
68		* @var int
69		*/
70		protected $flush_counter = 0;
71
72		/**
73		* Buffer flush threshold.
74		*
75		* @var int\|null
76		*/
77		protected $flush_threshold;
78
79		/**
80		* {@inheritdoc}
81		*/
82		protected $stream_filter_mode = STREAM_FILTER_WRITE;
83
84		/**
85		* Regular expression used to detect if RFC4180 formatting is necessary.
86		*
87		* @var string
88		*/
89		protected $rfc4180_regexp;
90
91		/**
92		* double enclosure for RFC4180 compliance.
93		*
94		* @var string
95		*/
96		protected $rfc4180_enclosure;
97
98		/**
99		* Returns the current newline sequence characters.
100		*/
101	3	public function getNewline(): string
102		{
103	3	return $this->newline;
104		}
105
106		/**
107		* Get the flush threshold.
108		*
109		* @return int\|null
110		*/
111	3	public function getFlushThreshold()
112		{
113	3	return $this->flush_threshold;
114		}
115
116		/**
117		* Adds multiple records to the CSV document.
118		*
119		* @see Writer::insertOne
120		*
121		* @param Traversable\|array $records
122		*/
123	9	public function insertAll($records, string $mode = self::MODE_PHP): int
124		{
125	9	if (!is_iterable($records)) {
126	3	throw new TypeError(sprintf('%s() expects argument passed to be iterable, %s given', __METHOD__, gettype($records)));
127		}
128
129	6	$bytes = 0;
130	6	foreach ($records as $record) {
131	6	$bytes += $this->insertOne($record, $mode);
132		}
133
134	6	$this->flush_counter = 0;
135	6	$this->document->fflush();
136
137	6	return $bytes;
138		}
139
140		/**
141		* Adds a single record to a CSV document.
142		*
143		* A record is an array that can contains scalar types values, NULL values
144		* or objects implementing the __toString method.
145		*
146		* @throws CannotInsertRecord If the record can not be inserted
147		*/
148	57	public function insertOne(array $record, string $mode = self::MODE_PHP): int
149		{
150	57	static $methodList = [self::MODE_PHP => 'addRecord', self::MODE_RFC4180 => 'addRFC4180CompliantRecord'];
151	57	$method = $methodList[$mode] ?? null;
152	57	if (null === $method) {
153	3	throw new Exception(sprintf('Unknown or unsupported writing mode %s', $mode));
154		}
155
156	54	$record = array_reduce($this->formatters, [$this, 'formatRecord'], $record);
157	54	$this->validateRecord($record);
158	51	$bytes = $this->$method($record);
159	51	if (false !== $bytes && 0 !== $bytes) {
160	45	return $bytes + $this->consolidate();
161		}
162
163	6	throw CannotInsertRecord::triggerOnInsertion($record);
164		}
165
166		/**
167		* Adds a single record to a CSV Document using PHP algorithm.
168		*
169		* @see https://php.net/manual/en/function.fputcsv.php
170		*
171		* @return int\|bool
172		*/
173	12	protected function addRecord(array $record)
174		{
175	12	return $this->document->fputcsv($record, $this->delimiter, $this->enclosure, $this->escape);
176		}
177
178		/**
179		* Adds a single record to a CSV Document using RFC4180 algorithm.
180		*
181		* @see https://php.net/manual/en/function.fputcsv.php
182		* @see https://php.net/manual/en/function.fwrite.php
183		* @see https://tools.ietf.org/html/rfc4180
184		* @see http://edoceo.com/utilitas/csv-file-format
185		*
186		* Fields must be delimited with enclosures if they contains :
187		*
188		* - Leading or trailing whitespaces
189		* - Embedded delimiters
190		* - Embedded line-breaks
191		* - Embedded enclosures.
192		*
193		* Embedded enclosures must be doubled.
194		*
195		* String conversion is done without any check like fputcsv:
196		*
197		* - Emits E_NOTICE on Array conversion (returns the 'Array' string)
198		* - Throws catchable fatal error on objects that can not be converted
199		* - Returns resource id without notice or error (returns 'Resource id #2')
200		* - Converts boolean true to '1', boolean false to the empty string
201		* - Converts null value to the empty string
202		*
203		* the LF character is added at the end of each record to mimic fputcsv behavior
204		*
205		* @return int\|bool
206		*/
207	27	protected function addRFC4180CompliantRecord(array $record)
208		{
209	27	foreach ($record as &$field) {
210	27	$field = (string) $field;
211	27	if (preg_match($this->rfc4180_regexp, $field)) {
212	27	$field = $this->enclosure.str_replace($this->enclosure, $this->rfc4180_enclosure, $field).$this->enclosure;
213		}
214		}
215	27	unset($field);
216
217	27	return $this->document->fwrite(implode($this->delimiter, $record)."\n");
218		}
219
220		/**
221		* {@inheritdoc}
222		*/
223	15	protected function resetProperties()
224		{
225	15	parent::resetProperties();
226	15	$embedded_characters = "\n\|\r".preg_quote($this->delimiter, '/').'\|'.preg_quote($this->enclosure, '/');
227	15	$this->rfc4180_regexp = '/^
228		(\ +) # leading whitespaces
229	15	\|(['.$embedded_characters.']) # embedded characters
230		\|(\ +) # trailing whitespaces
231		$/x';
232	15	$this->rfc4180_enclosure = $this->enclosure.$this->enclosure;
233	15	}
234
235		/**
236		* Format a record.
237		*
238		* The returned array must contain
239		* - scalar types values,
240		* - NULL values,
241		* - or objects implementing the __toString() method.
242		*/
243	3	protected function formatRecord(array $record, callable $formatter): array
244		{
245	3	return $formatter($record);
246		}
247
248		/**
249		* Validate a record.
250		*
251		* @throws CannotInsertRecord If the validation failed
252		*/
253	12	protected function validateRecord(array $record)
254		{
255	12	foreach ($this->validators as $name => $validator) {
256	3	if (true !== $validator($record)) {
257	3	throw CannotInsertRecord::triggerOnValidation($name, $record);
258		}
259		}
260	9	}
261
262		/**
263		* Apply post insertion actions.
264		*/
265	12	protected function consolidate(): int
266		{
267	12	$bytes = 0;
268	12	if ("\n" !== $this->newline) {
269	3	$this->document->fseek(-1, SEEK_CUR);
270	3	$bytes = $this->document->fwrite($this->newline, strlen($this->newline)) - 1;
271		}
272
273	12	if (null === $this->flush_threshold) {
274	9	return $bytes;
275		}
276
277	3	++$this->flush_counter;
278	3	if (0 === $this->flush_counter % $this->flush_threshold) {
279	3	$this->flush_counter = 0;
280	3	$this->document->fflush();
281		}
282
283	3	return $bytes;
284		}
285
286		/**
287		* Adds a record formatter.
288		*/
289	3	public function addFormatter(callable $formatter): self
290		{
291	3	$this->formatters[] = $formatter;
292
293	3	return $this;
294		}
295
296		/**
297		* Adds a record validator.
298		*/
299	3	public function addValidator(callable $validator, string $validator_name): self
300		{
301	3	$this->validators[$validator_name] = $validator;
302
303	3	return $this;
304		}
305
306		/**
307		* Sets the newline sequence.
308		*/
309	3	public function setNewline(string $newline): self
310		{
311	3	$this->newline = $newline;
312
313	3	return $this;
314		}
315
316		/**
317		* Set the flush threshold.
318		*
319		* @param int\|null $threshold
320		*
321		* @throws Exception if the threshold is a integer lesser than 1
322		*/
323	12	public function setFlushThreshold($threshold): self
324		{
325	12	if ($threshold === $this->flush_threshold) {
326	3	return $this;
327		}
328
329	12	if (!is_nullable_int($threshold)) {
330	3	throw new TypeError(sprintf(__METHOD__.'() expects 1 Argument to be null or an integer %s given', gettype($threshold)));
331		}
332
333	9	if (null !== $threshold && 1 > $threshold) {
334	3	throw new Exception(__METHOD__.'() expects 1 Argument to be null or a valid integer greater or equal to 1');
335		}
336
337	9	$this->flush_threshold = $threshold;
338	9	$this->flush_counter = 0;
339	9	$this->document->fflush();
340
341	9	return $this;
342		}
343		}
344

thephpleague / csv

Pull Request — master (#309)

Writer::addRecord() A

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like