Reader::getIterator() - 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 15:15 UTC

Reader::getIterator() A

↳ Parent: Reader

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	0
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 BadMethodCallException;
use CallbackFilterIterator;
use Countable;
use Iterator;
use IteratorAggregate;
use JsonSerializable;
use SplFileObject;
use TypeError;
use const STREAM_FILTER_READ;
use function array_combine;
use function array_filter;
use function array_pad;
use function array_slice;
use function array_unique;
use function gettype;
use function is_array;
use function iterator_count;
use function iterator_to_array;
use function mb_strlen;
use function mb_substr;
use function sprintf;
use function strlen;
use function substr;

/**
 * A class to select records from a CSV document.
 *
 * @package League.csv
 * @since  3.0.0
 *
 * @method array fetchOne(int $nth_record = 0) Returns a single record from the CSV
 * @method Generator fetchColumn(string|int $column_index) Returns the next value from a single CSV record field
 * @method Generator fetchPairs(string|int $offset_index = 0, string|int $value_index = 1) Fetches the next key-value pairs from the CSV document
 */
class Reader extends AbstractCsv implements Countable, IteratorAggregate, JsonSerializable
{
    /**
     * header offset.
     *
     * @var int|null
     */
    protected $header_offset;

    /**
     * header record.
     *
     * @var string[]
     */
    protected $header = [];

    /**
     * records count.
     *
     * @var int
     */
    protected $nb_records = -1;

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

    /**
     * {@inheritdoc}
     */
    public static function createFromPath(string $path, string $open_mode = 'r', $context = null)
    {
        return new static(Stream::createFromPath($path, $open_mode, $context));
    }

    /**
     * Returns the header offset.
     *
     * If no CSV header offset is set this method MUST return null
     *
     * @return int|null
     */
    public function getHeaderOffset()
    {
        return $this->header_offset;
    }

    /**
     * Returns the CSV record used as header.
     *
     * The returned header is represented as an array of string values
     *
     * @return string[]
     */
    public function getHeader(): array
    {
        if (null === $this->header_offset) {
            return $this->header;
        }

        if ([] !== $this->header) {
            return $this->header;
        }

        $this->header = $this->setHeader($this->header_offset);

        return $this->header;
    }

    /**
     * Determine the CSV record header.
     *
     * @throws Exception If the header offset is set and no record is found or is the empty array
     *
     * @return string[]
     */
    protected function setHeader(int $offset): array
    {
        $header = $this->seekRow($offset);
        if (false === $header || [] === $header) {
            throw new Exception(sprintf('The header record does not exist or is empty at offset: `%s`', $offset));
        }

        if (0 === $offset) {
            return $this->removeBOM($header, mb_strlen($this->getInputBOM()), $this->enclosure);
        }

        return $header;
    }

    /**
     * Returns the row at a given offset.
     */
    protected function seekRow(int $offset)
    {
        $this->document->setFlags(SplFileObject::READ_CSV | SplFileObject::READ_AHEAD | SplFileObject::SKIP_EMPTY);
        $this->document->setCsvControl($this->delimiter, $this->enclosure, $this->escape);
        $this->document->rewind();

        //Workaround for SplFileObject::seek bug in PHP7.2+ see https://bugs.php.net/bug.php?id=75917
        if (PHP_VERSION_ID >= 70200 && !$this->document instanceof Stream) {
            while ($offset !== $this->document->key() && $this->document->valid()) {
                $this->document->next();
            }

            return $this->document->current();
        }

        $this->document->seek($offset);

        return $this->document->current();
    }

    /**
     * Strip the BOM sequence from a record.
     *
     * @param string[] $record
     *
     * @return string[]
     */
    protected function removeBOM(array $record, int $bom_length, string $enclosure): array
    {
        if (0 == $bom_length) {
            return $record;
        }

        $record[0] = mb_substr($record[0], $bom_length);
        if ($enclosure.$enclosure != substr($record[0].$record[0], strlen($record[0]) - 1, 2)) {
            return $record;
        }

        $record[0] = substr($record[0], 1, -1);

        return $record;
    }

    /**
     * {@inheritdoc}
     */
    public function __call($method, array $arguments)
    {
        static $whitelisted = ['fetchColumn' => 1, 'fetchOne' => 1, 'fetchPairs' => 1];
        if (isset($whitelisted[$method])) {
            return (new ResultSet($this->getRecords(), $this->getHeader()))->$method(...$arguments);
        }

        throw new BadMethodCallException(sprintf('%s::%s() method does not exist', __CLASS__, $method));
    }

    /**
     * {@inheritdoc}
     */
    public function count(): int
    {
        if (-1 === $this->nb_records) {
            $this->nb_records = iterator_count($this->getRecords());
        }

        return $this->nb_records;
    }

    /**
     * {@inheritdoc}
     */
    public function getIterator(): Iterator
    {
        return $this->getRecords();
    }

    /**
     * {@inheritdoc}
     */
    public function jsonSerialize(): array
    {
        return iterator_to_array($this->getRecords(), false);
    }

    /**
     * Returns the CSV records as an iterator object.
     *
     * Each CSV record is represented as a simple array containig strings or null values.
     *
     * If the CSV document has a header record then each record is combined
     * to the header record and the header record is removed from the iterator.
     *
     * If the CSV document is inconsistent. Missing record fields are
     * filled with null values while extra record fields are strip from
     * the returned object.
     *
     * @param string[] $header an optional header to use instead of the CSV document header
     */
    public function getRecords(array $header = []): Iterator
    {
        $header = $this->computeHeader($header);
        $normalized = function ($record): bool {
            return is_array($record) && $record != [null];
        };
        $bom = $this->getInputBOM();
        $this->document->setFlags(SplFileObject::READ_CSV | SplFileObject::READ_AHEAD | SplFileObject::SKIP_EMPTY);
        $this->document->setCsvControl($this->delimiter, $this->enclosure, $this->escape);

        $records = $this->stripBOM(new CallbackFilterIterator($this->document, $normalized), $bom);
        if (null !== $this->header_offset) {
            $records = new CallbackFilterIterator($records, function (array $record, int $offset): bool {
                return $offset !== $this->header_offset;
            });
        }

        return $this->combineHeader($records, $header);
    }

    /**
     * Returns the header to be used for iteration.
     *
     * @param string[] $header
     *
     * @throws Exception If the header contains non unique column name
     *
     * @return string[]
     */
    protected function computeHeader(array $header)
    {
        if ([] === $header) {
            $header = $this->getHeader();
        }

        if ($header === array_unique(array_filter($header, 'is_string'))) {
            return $header;
        }

        throw new Exception('The header record must be empty or a flat array with unique string values');
    }

    /**
     * Combine the CSV header to each record if present.
     *
     * @param string[] $header
     */
    protected function combineHeader(Iterator $iterator, array $header): Iterator
    {
        if ([] === $header) {
            return $iterator;
        }

        $field_count = count($header);
        $mapper = function (array $record) use ($header, $field_count): array {
            if (count($record) != $field_count) {
                $record = array_slice(array_pad($record, $field_count, null), 0, $field_count);
            }

            return array_combine($header, $record);
        };

        return new MapIterator($iterator, $mapper);
    }

    /**
     * Strip the BOM sequence from the returned records if necessary.
     */
    protected function stripBOM(Iterator $iterator, string $bom): Iterator
    {
        if ('' === $bom) {
            return $iterator;
        }

        $bom_length = mb_strlen($bom);
        $mapper = function (array $record, int $index) use ($bom_length): array {
            if (0 != $index) {
                return $record;
            }

            return $this->removeBOM($record, $bom_length, $this->enclosure);
        };

        return new MapIterator($iterator, $mapper);
    }

    /**
     * Selects the record to be used as the CSV header.
     *
     * Because the header is represented as an array, to be valid
     * a header MUST contain only unique string value.
     *
     * @param int|null $offset the header record offset
     *
     * @throws Exception if the offset is a negative integer
     *
     * @return static
     */
    public function setHeaderOffset($offset): self
    {
        if ($offset === $this->header_offset) {
            return $this;
        }

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

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

        $this->header_offset = $offset;
        $this->resetProperties();

        return $this;
    }

    /**
     * {@inheritdoc}
     */
    protected function resetProperties()
    {
        parent::resetProperties();
        $this->nb_records = -1;
        $this->header = [];
    }
}


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 BadMethodCallException;
20		use CallbackFilterIterator;
21		use Countable;
22		use Iterator;
23		use IteratorAggregate;
24		use JsonSerializable;
25		use SplFileObject;
26		use TypeError;
27		use const STREAM_FILTER_READ;
28		use function array_combine;
29		use function array_filter;
30		use function array_pad;
31		use function array_slice;
32		use function array_unique;
33		use function gettype;
34		use function is_array;
35		use function iterator_count;
36		use function iterator_to_array;
37		use function mb_strlen;
38		use function mb_substr;
39		use function sprintf;
40		use function strlen;
41		use function substr;
42
43		/**
44		* A class to select records from a CSV document.
45		*
46		* @package League.csv
47		* @since 3.0.0
48		*
49		* @method array fetchOne(int $nth_record = 0) Returns a single record from the CSV
50		* @method Generator fetchColumn(string\|int $column_index) Returns the next value from a single CSV record field
51		* @method Generator fetchPairs(string\|int $offset_index = 0, string\|int $value_index = 1) Fetches the next key-value pairs from the CSV document
52		*/
53		class Reader extends AbstractCsv implements Countable, IteratorAggregate, JsonSerializable
54		{
55		/**
56		* header offset.
57		*
58		* @var int\|null
59		*/
60		protected $header_offset;
61
62		/**
63		* header record.
64		*
65		* @var string[]
66		*/
67		protected $header = [];
68
69		/**
70		* records count.
71		*
72		* @var int
73		*/
74		protected $nb_records = -1;
75
76		/**
77		* {@inheritdoc}
78		*/
79		protected $stream_filter_mode = STREAM_FILTER_READ;
80
81		/**
82		* {@inheritdoc}
83		*/
84	3	public static function createFromPath(string $path, string $open_mode = 'r', $context = null)
85		{
86	3	return new static(Stream::createFromPath($path, $open_mode, $context));
87		}
88
89		/**
90		* Returns the header offset.
91		*
92		* If no CSV header offset is set this method MUST return null
93		*
94		* @return int\|null
95		*/
96	15	public function getHeaderOffset()
97		{
98	15	return $this->header_offset;
99		}
100
101		/**
102		* Returns the CSV record used as header.
103		*
104		* The returned header is represented as an array of string values
105		*
106		* @return string[]
107		*/
108	15	public function getHeader(): array
109		{
110	15	if (null === $this->header_offset) {
111	12	return $this->header;
112		}
113
114	6	if ([] !== $this->header) {
115	3	return $this->header;
116		}
117
118	6	$this->header = $this->setHeader($this->header_offset);
119
120	6	return $this->header;
121		}
122
123		/**
124		* Determine the CSV record header.
125		*
126		* @throws Exception If the header offset is set and no record is found or is the empty array
127		*
128		* @return string[]
129		*/
130	12	protected function setHeader(int $offset): array
131		{
132	12	$header = $this->seekRow($offset);
133	12	if (false === $header \|\| [] === $header) {
134	6	throw new Exception(sprintf('The header record does not exist or is empty at offset: `%s`', $offset));
135		}
136
137	6	if (0 === $offset) {
138	3	return $this->removeBOM($header, mb_strlen($this->getInputBOM()), $this->enclosure);
139		}
140
141	3	return $header;
142		}
143
144		/**
145		* Returns the row at a given offset.
146		*/
147	6	protected function seekRow(int $offset)
148		{
149	6	$this->document->setFlags(SplFileObject::READ_CSV \| SplFileObject::READ_AHEAD \| SplFileObject::SKIP_EMPTY);
150	6	$this->document->setCsvControl($this->delimiter, $this->enclosure, $this->escape);
151	6	$this->document->rewind();
152
153		//Workaround for SplFileObject::seek bug in PHP7.2+ see https://bugs.php.net/bug.php?id=75917
154	6	if (PHP_VERSION_ID >= 70200 && !$this->document instanceof Stream) {
155		while ($offset !== $this->document->key() && $this->document->valid()) {
156		$this->document->next();
157		}
158
159		return $this->document->current();
160		}
161
162	6	$this->document->seek($offset);
163
164	6	return $this->document->current();
165		}
166
167		/**
168		* Strip the BOM sequence from a record.
169		*
170		* @param string[] $record
171		*
172		* @return string[]
173		*/
174	12	protected function removeBOM(array $record, int $bom_length, string $enclosure): array
175		{
176	12	if (0 == $bom_length) {
177	3	return $record;
178		}
179
180	9	$record[0] = mb_substr($record[0], $bom_length);
181	9	if ($enclosure.$enclosure != substr($record[0].$record[0], strlen($record[0]) - 1, 2)) {
182	6	return $record;
183		}
184
185	3	$record[0] = substr($record[0], 1, -1);
186
187	3	return $record;
188		}
189
190		/**
191		* {@inheritdoc}
192		*/
193	9	public function __call($method, array $arguments)
194		{
195	9	static $whitelisted = ['fetchColumn' => 1, 'fetchOne' => 1, 'fetchPairs' => 1];
196	9	if (isset($whitelisted[$method])) {
197	3	return (new ResultSet($this->getRecords(), $this->getHeader()))->$method(...$arguments);
198		}
199
200	6	throw new BadMethodCallException(sprintf('%s::%s() method does not exist', __CLASS__, $method));
201		}
202
203		/**
204		* {@inheritdoc}
205		*/
206	3	public function count(): int
207		{
208	3	if (-1 === $this->nb_records) {
209	3	$this->nb_records = iterator_count($this->getRecords());
210		}
211
212	3	return $this->nb_records;
213		}
214
215		/**
216		* {@inheritdoc}
217		*/
218	3	public function getIterator(): Iterator
219		{
220	3	return $this->getRecords();
221		}
222
223		/**
224		* {@inheritdoc}
225		*/
226	3	public function jsonSerialize(): array
227		{
228	3	return iterator_to_array($this->getRecords(), false);
229		}
230
231		/**
232		* Returns the CSV records as an iterator object.
233		*
234		* Each CSV record is represented as a simple array containig strings or null values.
235		*
236		* If the CSV document has a header record then each record is combined
237		* to the header record and the header record is removed from the iterator.
238		*
239		* If the CSV document is inconsistent. Missing record fields are
240		* filled with null values while extra record fields are strip from
241		* the returned object.
242		*
243		* @param string[] $header an optional header to use instead of the CSV document header
244		*/
245	18	public function getRecords(array $header = []): Iterator
246		{
247	18	$header = $this->computeHeader($header);
248	15	$normalized = function ($record): bool {
249	15	return is_array($record) && $record != [null];
250	15	};
251	15	$bom = $this->getInputBOM();
252	15	$this->document->setFlags(SplFileObject::READ_CSV \| SplFileObject::READ_AHEAD \| SplFileObject::SKIP_EMPTY);
253	15	$this->document->setCsvControl($this->delimiter, $this->enclosure, $this->escape);
254
255	15	$records = $this->stripBOM(new CallbackFilterIterator($this->document, $normalized), $bom);
256	15	if (null !== $this->header_offset) {
257	6	$records = new CallbackFilterIterator($records, function (array $record, int $offset): bool {
258	6	return $offset !== $this->header_offset;
259	6	});
260		}
261
262	15	return $this->combineHeader($records, $header);
263		}
264
265		/**
266		* Returns the header to be used for iteration.
267		*
268		* @param string[] $header
269		*
270		* @throws Exception If the header contains non unique column name
271		*
272		* @return string[]
273		*/
274	24	protected function computeHeader(array $header)
275		{
276	24	if ([] === $header) {
277	21	$header = $this->getHeader();
278		}
279
280	24	if ($header === array_unique(array_filter($header, 'is_string'))) {
281	21	return $header;
282		}
283
284	3	throw new Exception('The header record must be empty or a flat array with unique string values');
285		}
286
287		/**
288		* Combine the CSV header to each record if present.
289		*
290		* @param string[] $header
291		*/
292	30	protected function combineHeader(Iterator $iterator, array $header): Iterator
293		{
294	30	if ([] === $header) {
295	21	return $iterator;
296		}
297
298	12	$field_count = count($header);
299	12	$mapper = function (array $record) use ($header, $field_count): array {
300	12	if (count($record) != $field_count) {
301	6	$record = array_slice(array_pad($record, $field_count, null), 0, $field_count);
302		}
303
304	12	return array_combine($header, $record);
305	12	};
306
307	12	return new MapIterator($iterator, $mapper);
308		}
309
310		/**
311		* Strip the BOM sequence from the returned records if necessary.
312		*/
313	24	protected function stripBOM(Iterator $iterator, string $bom): Iterator
314		{
315	24	if ('' === $bom) {
316	15	return $iterator;
317		}
318
319	9	$bom_length = mb_strlen($bom);
320	9	$mapper = function (array $record, int $index) use ($bom_length): array {
321	9	if (0 != $index) {
322	3	return $record;
323		}
324
325	9	return $this->removeBOM($record, $bom_length, $this->enclosure);
326	9	};
327
328	9	return new MapIterator($iterator, $mapper);
329		}
330
331		/**
332		* Selects the record to be used as the CSV header.
333		*
334		* Because the header is represented as an array, to be valid
335		* a header MUST contain only unique string value.
336		*
337		* @param int\|null $offset the header record offset
338		*
339		* @throws Exception if the offset is a negative integer
340		*
341		* @return static
342		*/
343	21	public function setHeaderOffset($offset): self
344		{
345	21	if ($offset === $this->header_offset) {
346	12	return $this;
347		}
348
349	9	if (!is_nullable_int($offset)) {
350	3	throw new TypeError(sprintf(__METHOD__.'() expects 1 Argument to be null or an integer %s given', gettype($offset)));
351		}
352
353	6	if (null !== $offset && 0 > $offset) {
354	3	throw new Exception(__METHOD__.'() expects 1 Argument to be greater or equal to 0');
355		}
356
357	3	$this->header_offset = $offset;
358	3	$this->resetProperties();
359
360	3	return $this;
361		}
362
363		/**
364		* {@inheritdoc}
365		*/
366	24	protected function resetProperties()
367		{
368	24	parent::resetProperties();
369	24	$this->nb_records = -1;
370	24	$this->header = [];
371	24	}
372		}
373

thephpleague / csv

Pull Request — master (#309)

Reader::getIterator() A

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like