Completed
Pull Request — master (#210)
by ignace nyamagana
02:41
created

Statement::computeHeader()   B

Complexity

Conditions 4
Paths 4

Size

Total Lines 24
Code Lines 15

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 15
CRAP Score 4

Importance

Changes 0
Metric Value
dl 0
loc 24
ccs 15
cts 15
cp 1
rs 8.6845
c 0
b 0
f 0
cc 4
eloc 15
nc 4
nop 1
crap 4
1
<?php
2
/**
3
* This file is part of the League.csv library
4
*
5
* @license http://opensource.org/licenses/MIT
6
* @link https://github.com/thephpleague/csv/
7
* @version 9.0.0
8
* @package League.csv
9
*
10
* For the full copyright and license information, please view the LICENSE
11
* file that was distributed with this source code.
12
*/
13
declare(strict_types=1);
14
15
namespace League\Csv;
16
17
use ArrayIterator;
18
use CallbackFilterIterator;
19
use Iterator;
20
use League\Csv\Config\ValidatorTrait;
21
use LimitIterator;
22
use SplFileObject;
23
24
/**
25
 *  A trait to manage filtering a CSV
26
 *
27
 * @package League.csv
28
 * @since  9.0.0
29
 *
30
 */
31
class Statement
32
{
33
    use ValidatorTrait;
34
35
    /**
36
     * Callables to filter the iterator
37
     *
38
     * @var callable[]
39
     */
40
    protected $where = [];
41
42
    /**
43
     * Callables to sort the iterator
44
     *
45
     * @var callable[]
46
     */
47
    protected $order_by = [];
48
49
    /**
50
     * iterator Offset
51
     *
52
     * @var int
53
     */
54
    protected $offset = 0;
55
56
    /**
57
     * iterator maximum length
58
     *
59
     * @var int
60
     */
61
    protected $limit = -1;
62
63
    /**
64
     * CSV headers
65
     *
66
     * @var string[]
67
     */
68
    protected $header = [];
69
70
    /**
71
     * Set LimitIterator Offset
72
     *
73
     * @param $offset
74
     *
75
     * @return self
76
     */
77 12
    public function offset(int $offset = 0): self
78
    {
79 12
        $offset = $this->filterInteger($offset, 0, 'the offset must be a positive integer or 0');
80 12
        if ($offset === $this->offset) {
81 2
            return $this;
82
        }
83
84 10
        $clone = clone $this;
85 10
        $clone->offset = $offset;
86
87 10
        return $clone;
88
    }
89
90
    /**
91
     * Set LimitIterator Count
92
     *
93
     * @param int $limit
94
     *
95
     * @return self
96
     */
97 14
    public function limit(int $limit = -1): self
98
    {
99 14
        $limit = $this->filterInteger($limit, -1, 'the limit must an integer greater or equals to -1');
100 12
        if ($limit === $this->limit) {
101 2
            return $this;
102
        }
103
104 10
        $clone = clone $this;
105 10
        $clone->limit = $limit;
106
107 10
        return $clone;
108
    }
109
110
    /**
111
     * Set an Iterator sorting callable function
112
     *
113
     * @param callable $callable
114
     *
115
     * @return self
116
     */
117 2
    public function orderBy(callable $callable): self
118
    {
119 2
        $clone = clone $this;
120 2
        $clone->order_by[] = $callable;
121
122 2
        return $clone;
123
    }
124
125
    /**
126
     * Set the Iterator filter method
127
     *
128
     * @param callable $callable
129
     *
130
     * @return self
131
     */
132 2
    public function where(callable $callable): self
133
    {
134 2
        $clone = clone $this;
135 2
        $clone->where[] = $callable;
136
137 2
        return $clone;
138
    }
139
140
    /**
141
     * Set the headers to be used by the RecordSet object
142
     *
143
     * @param string[] $header
144
     *
145
     * @return self
146
     */
147 18
    public function header(array $header): self
148
    {
149 18
        $header = $this->filterHeader($header);
150 16
        if ($header === $this->header) {
151 2
            return $this;
152
        }
153
154 14
        $clone = clone $this;
155 14
        $clone->header = $header;
156
157 14
        return $clone;
158
    }
159
160
    /**
161
     * Returns the inner CSV Document Iterator object
162
     *
163
     * @return RecordSet
164
     */
165 120
    public function process(Reader $reader)
166
    {
167 120
        $csv = $reader->getDocument();
168 120
        $csv->setFlags(SplFileObject::READ_CSV | SplFileObject::READ_AHEAD | SplFileObject::SKIP_EMPTY);
169 120
        $csv->setCsvControl($reader->getDelimiter(), $reader->getEnclosure(), $reader->getEscape());
170 120
        $header = !empty($this->header) ? $this->header : $this->computeHeader($reader);
171
        $normalized = function ($row) {
172 102
            return is_array($row) && $row != [null];
173 118
        };
174
175 118
        $iterator = $this->stripBOM(
176 118
            new CallbackFilterIterator($csv, $normalized),
177 118
            $reader->getInputBOM(),
178 118
            $reader->getEnclosure()
179
        );
180 118
        $iterator = $this->combineHeader($iterator, $header);
181 118
        $iterator = $this->filterRecords($iterator);
182 118
        $iterator = $this->orderRecords($iterator);
183
184 118
        return new RecordSet(new LimitIterator($iterator, $this->offset, $this->limit), $header);
185
    }
186
187
    /**
188
     * Set the computed RecordSet headers
189
     *
190
     * @param Reader $reader The CSV document Reader object
191
     *
192
     * @throws Exception If the header is not found
193
     *
194
     * @return string[]
195
     */
196 106
    protected function computeHeader(Reader $reader)
197
    {
198 106
        $offset = $reader->getHeaderOffset();
199 106
        if (null === $offset) {
200 92
            return [];
201
        }
202
203 14
        $csv = $reader->getDocument();
204 14
        $csv->seek($offset);
205 14
        $header = $csv->current();
206 14
        if (empty($header)) {
207 2
            throw new Exception('The header record specified by `Reader::setHeaderOffset` does not exist or is empty');
208
        }
209
210 12
        if (0 === $offset) {
211 8
            $header = $this->removeBOM($header, mb_strlen($reader->getInputBOM()), $reader->getEnclosure());
212
        }
213 12
        $header = $this->filterHeader($header);
214
        array_unshift($this->where, function ($row, $index) use ($offset) {
215 8
            return $index !== $offset;
216 12
        });
217
218 12
        return $header;
219
    }
220
221
    /**
222
     * Remove the BOM sequence from the CSV
223
     *
224
     * @param Iterator $iterator
225
     *
226
     * @return Iterator
227
     */
228 118
    protected function stripBOM(Iterator $iterator, string $bom, string $enclosure): Iterator
229
    {
230 118
        if ('' == $bom) {
231 100
            return $iterator;
232
        }
233
234 18
        $bom_length = mb_strlen($bom);
235
        $strip_bom = function ($row, $index) use ($bom_length, $enclosure) {
236 18
            if (0 != $index || !is_array($row)) {
237 14
                return $row;
238
            }
239
240 18
            return $this->removeBOM($row, $bom_length, $enclosure);
241 18
        };
242
243 18
        return new MapIterator($iterator, $strip_bom);
244
    }
245
246
    /**
247
     * Add the CSV header if present
248
     *
249
     * @param Iterator $iterator
250
     * @param string[] $header
251
     *
252
     * @return Iterator
253
     */
254 118
    protected function combineHeader(Iterator $iterator, array $header): Iterator
255
    {
256 118
        if (empty($header)) {
257 92
            return $iterator;
258
        }
259
260 26
        $header_count = count($header);
261
        $combine = function (array $row) use ($header_count, $header) {
262 18
            if ($header_count != count($row)) {
263 4
                $row = array_slice(array_pad($row, $header_count, null), 0, $header_count);
264
            }
265
266 18
            return array_combine($header, $row);
267 26
        };
268
269 26
        return new MapIterator($iterator, $combine);
270
    }
271
272
    /**
273
    * Filter the Iterator
274
    *
275
    * @param Iterator $iterator
276
    *
277
    * @return Iterator
278
    */
279 118
    protected function filterRecords(Iterator $iterator): Iterator
280
    {
281
        $reducer = function ($iterator, $callable) {
282 14
            return new CallbackFilterIterator($iterator, $callable);
283 118
        };
284
285 118
        return array_reduce($this->where, $reducer, $iterator);
286
    }
287
288
    /**
289
    * Sort the Iterator
290
    *
291
    * @param Iterator $iterator
292
    *
293
    * @return Iterator
294
    */
295 118
    protected function orderRecords(Iterator $iterator): Iterator
296
    {
297 118
        if (empty($this->order_by)) {
298 116
            return $iterator;
299
        }
300
301 2
        $obj = new ArrayIterator(iterator_to_array($iterator));
302 2
        $obj->uasort(function ($row_a, $row_b) {
303 2
            $res = 0;
304 2
            foreach ($this->order_by as $compare) {
305 2
                if (0 !== ($res = $compare($row_a, $row_b))) {
306 2
                    break;
307
                }
308
            }
309
310 2
            return $res;
311 2
        });
312
313 2
        return $obj;
314
    }
315
}
316