data-mapper/src/Collection/CollectionAbstract.php at 59e405d91933f035e9c981837f2eef0ff243515c · g4code/data-mapper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
<?php

namespace G4\DataMapper\Collection;

abstract class CollectionAbstract implements \Iterator, \Countable
{
    protected $_keyMap = array();

    protected $_objects = array();

    protected $_pointer = 0;

    protected $_rawData = array();

    protected $_total = 0;

    public function count(): int
    {
        return $this->_total;
    }

    public function current(): mixed
    {
        return $this->_getObject();
    }

    public function getRawData()
    {
        return $this->_rawData;
    }

    public function key(): mixed
    {
        return $this->_pointer;
    }

    public function next(): void
    {
        $row = $this->_getObject();

        if (!empty($row)) {
            $this->_incrementPointer();
        }
    }

    public function rewind(): void
    {
        $this->_pointer = 0;
    }

    public function valid(): bool
    {
        return !is_null($this->current());
    }

    abstract protected function _factory();

    protected function _addObject($object)
    {
        $this->_objects[$this->_pointer] = $object;
        return $this;
    }

    protected function _getCurrentObject()
    {
        return $this->_objects[$this->_pointer];
    }

    protected function _getCurrentRawData()
    {
        return $this->_rawData[$this->_keyMap[$this->_pointer]];
    }

    /**
     * TODO:Drasko - Do this better!
     */
    protected function _getObject()
    {
        if ($this->_pointer >= $this->_total || $this->_pointer < 0) {
            return null;
        }

        if ($this->_hasCurrentObject()) {
            return $this->_getCurrentObject();
        }

        if ($this->_hasCurrentRawData()) {
            $object = $this->_factory();

            if ($object === null) {
                return null;
            }

            $this->_addObject($object);

            return $this->_getCurrentObject();
        }

        return null;
    }

    protected function _hasCurrentObject()
    {
        return isset($this->_objects[$this->_pointer]);
    }

    protected function _hasCurrentRawData()
    {
        return isset($this->_rawData[$this->_keyMap[$this->_pointer]]);
    }

    public function _incrementPointer()
    {
        $this->_pointer++;
        return $this;
    }

    protected function _setRawData(array $rawData = null)
    {
        if (!is_null($rawData)) {
            $this->_rawData = $rawData;
            $this->_total   = count($rawData);
            $this->_keyMap  = array_keys($rawData);
        }
    }
}