PollRepository.php
2.88 KB
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
127
<?php
namespace FootyRoom\Repositories\Poll;
use MongoDB\BSON\ObjectId;
use FootyRoom\Support\MongoClient;
use MongoDB\BSON\UTCDateTime;
use FootyRoom\Support\AutoMapper;
use FootyRoom\Core\Poll\Poll;
class PollRepository
{
/**
* @var \FootyRoom\Support\MongoClient
*/
protected $mongo;
/**
* Constructor.
*
* @param \FootyRoom\Support\MongoClient $mongo
*/
public function __construct(MongoClient $mongo)
{
$this->mongo = $mongo->footyroom;
}
/**
* Finds poll by specified id.
*
* @param string $pollId
*
* @return \FootyRoom\Core\Poll\Poll|null
*/
public function findById($pollId)
{
$pollDto = $this->mongo->polls->findOne([
'_id' => new ObjectId($pollId),
]);
if (!$pollDto) {
return null;
}
$poll = AutoMapper::map($pollDto, Poll::class);
AutoMapper::setValue($poll, 'id', (string) $pollDto->_id);
return $poll;
}
/**
* Maps Poll object to array so that we can persist it.
*
* @param \FootyRoom\Core\Poll\Poll $poll
*
* @return array
*/
protected function mapToArray(Poll $poll)
{
$pollDTO = [
'userId' => $poll->getUserId(),
'isPublished' => $poll->getIsPublished(),
];
foreach ($poll->getChoices() as $key => $choice) {
$pollDTO['choices'][$key]['id'] = $choice->getId();
$pollDTO['choices'][$key]['text'] = $choice->getText();
$pollDTO['choices'][$key]['image'] = $choice->getImage();
}
return $pollDTO;
}
/**
* Create poll.
*
* @param \FootyRoom\Core\Poll\Poll $poll
*/
public function create(Poll $poll)
{
$pollDTO = $this->mapToArray($poll);
// Automatically expire unpublished polls after 1 day.
$pollDTO['expireAt'] = new UTCDateTime(strtotime('+30 day') * 1000);
$result = $this->mongo->polls->insertOne($pollDTO);
AutoMapper::setValue($poll, 'id', (string) $result->getInsertedId());
}
/**
* Update poll.
*
* @param \FootyRoom\Core\Poll\Poll $poll
*/
public function update(Poll $poll)
{
$pollDTO = $this->mapToArray($poll);
$this->mongo->polls->updateOne(
['_id' => new ObjectId($poll->getId())],
['$set' => [
'choices' => $pollDTO['choices'],
'isPublished' => $pollDTO['isPublished'],
]]
);
}
/**
* Marks poll as published.
*
* @param string $id
*/
public function markPublished($id)
{
$this->mongo->polls->updateOne(
['_id' => new ObjectId($id)],
['$set' => [
'isPublished' => true,
],
'$unset' => [
'expireAt' => 1,
], ]
);
}
}