Completed
Push — master ( 6067eb...3f355c )
by Samuel
15:57 queued 01:01
created

DateBehaviour::__construct()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 5
Code Lines 3

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 5
rs 9.4285
cc 1
eloc 3
nc 1
nop 2
1
<?php
2
3
declare(strict_types=1);
4
5
namespace SimpleMapper\Behaviour;
6
7
use Nette\Utils\DateTime;
8
use SimpleMapper\ActiveRow;
9
10
class DateBehaviour extends AbstractBehaviour
11
{
12
    /** @var string */
13
    private $createdAtField;
14
15
    /** @var string */
16
    private $updatedAtField;
17
18
    /**
19
     * @param string $createdAtField
20
     * @param string $updatedAtField
21
     */
22
    public function __construct(string $createdAtField = 'created_at', string $updatedAtField = 'updated_at')
23
    {
24
        $this->createdAtField = $createdAtField;
25
        $this->updatedAtField = $updatedAtField;
26
    }
27
28
    /**
29
     * @param array $data
30
     * @return array
31
     */
32
    public function beforeInsert(array $data): array
33
    {
34
        $now = new DateTime('now');
35
        if ($this->createdAtField) {
36
            $data[$this->createdAtField] = $now;
37
        }
38
        if ($this->updatedAtField) {
39
            $data[$this->updatedAtField] = $now;
40
        }
41
        return $data;
42
    }
43
44
    /**
45
     * @param ActiveRow $record
46
     * @param array $data
47
     * @return array
48
     */
49
    public function beforeUpdate(ActiveRow $record, array $data): array
50
    {
51
        if ($this->updatedAtField) {
52
            $data[$this->updatedAtField] = new DateTime('now');
53
        }
54
        return $data;
55
    }
56
}
57