|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
namespace Nopolabs\Yabot; |
|
4
|
|
|
|
|
5
|
|
|
use DateTime; |
|
6
|
|
|
use Exception; |
|
7
|
|
|
use Nopolabs\Yabot\Helpers\ConfigTrait; |
|
8
|
|
|
use Nopolabs\Yabot\Helpers\LogTrait; |
|
9
|
|
|
use Nopolabs\Yabot\Helpers\LoopTrait; |
|
10
|
|
|
use Nopolabs\Yabot\Helpers\SlackTrait; |
|
11
|
|
|
use Nopolabs\Yabot\Message\MessageFactory; |
|
12
|
|
|
use Nopolabs\Yabot\Plugin\PluginInterface; |
|
13
|
|
|
use Nopolabs\Yabot\Plugin\PluginManager; |
|
14
|
|
|
use Nopolabs\Yabot\Slack\Client; |
|
15
|
|
|
use Psr\Log\LoggerInterface; |
|
16
|
|
|
use React\EventLoop\LoopInterface; |
|
17
|
|
|
use React\EventLoop\Timer\TimerInterface; |
|
18
|
|
|
use React\Promise\Timer; |
|
19
|
|
|
use Slack\Payload; |
|
20
|
|
|
use Slack\User; |
|
21
|
|
|
use Throwable; |
|
22
|
|
|
|
|
23
|
|
|
class Yabot |
|
24
|
|
|
{ |
|
25
|
|
|
use LogTrait; |
|
26
|
|
|
use LoopTrait; |
|
27
|
|
|
use SlackTrait; |
|
28
|
|
|
use ConfigTrait; |
|
29
|
|
|
|
|
30
|
|
|
/** @var MessageFactory */ |
|
31
|
|
|
private $messageFactory; |
|
32
|
|
|
|
|
33
|
|
|
/** @var PluginManager */ |
|
34
|
|
|
private $pluginManager; |
|
35
|
|
|
|
|
36
|
|
|
/** @var string */ |
|
37
|
|
|
private $messageLog; |
|
38
|
|
|
|
|
39
|
|
|
/** @var TimerInterface */ |
|
40
|
|
|
private $monitor; |
|
41
|
|
|
|
|
42
|
|
|
/** @var bool */ |
|
43
|
|
|
private $pong; |
|
44
|
|
|
|
|
45
|
|
|
public function __construct( |
|
46
|
|
|
LoggerInterface $logger, |
|
47
|
|
|
LoopInterface $eventLoop, |
|
48
|
|
|
Client $slackClient, |
|
49
|
|
|
MessageFactory $messageFactory, |
|
50
|
|
|
PluginManager $pluginManager, |
|
51
|
|
|
array $config = [] |
|
52
|
|
|
) { |
|
53
|
|
|
$this->setLog($logger); |
|
54
|
|
|
$this->setLoop($eventLoop); |
|
55
|
|
|
$this->setSlack($slackClient); |
|
56
|
|
|
$this->setConfig($config); |
|
57
|
|
|
$this->messageFactory = $messageFactory; |
|
58
|
|
|
$this->pluginManager = $pluginManager; |
|
59
|
|
|
$this->messageLog = null; |
|
60
|
|
|
} |
|
61
|
|
|
|
|
62
|
|
|
public function getMessageLog() |
|
63
|
|
|
{ |
|
64
|
|
|
return $this->messageLog; |
|
65
|
|
|
} |
|
66
|
|
|
|
|
67
|
|
|
public function setMessageLog(string $messageLog = null) |
|
68
|
|
|
{ |
|
69
|
|
|
$this->messageLog = $messageLog ?? null; |
|
70
|
|
|
} |
|
71
|
|
|
|
|
72
|
|
|
public function init(array $plugins) |
|
73
|
|
|
{ |
|
74
|
|
|
foreach ($plugins as $pluginId => $plugin) { |
|
75
|
|
|
/** @var PluginInterface $plugin */ |
|
76
|
|
|
|
|
77
|
|
|
$this->info("loading $pluginId"); |
|
78
|
|
|
|
|
79
|
|
|
try { |
|
80
|
|
|
$this->pluginManager->loadPlugin($pluginId, $plugin); |
|
81
|
|
|
} catch (Exception $e) { |
|
82
|
|
|
$this->warning("Unhandled Exception while loading $pluginId: ".$e->getMessage()); |
|
83
|
|
|
$this->warning($e->getTraceAsString()); |
|
84
|
|
|
} |
|
85
|
|
|
} |
|
86
|
|
|
} |
|
87
|
|
|
|
|
88
|
|
|
public function run() |
|
89
|
|
|
{ |
|
90
|
|
|
$this->getSlack()->init(); |
|
91
|
|
|
|
|
92
|
|
|
$this->getLog()->info('Connecting...'); |
|
93
|
|
|
|
|
94
|
|
|
$this->connect(); |
|
95
|
|
|
|
|
96
|
|
|
$this->addMemoryReporting(); |
|
97
|
|
|
|
|
98
|
|
|
$this->getLoop()->run(); |
|
99
|
|
|
} |
|
100
|
|
|
|
|
101
|
|
|
public function shutDown() |
|
102
|
|
|
{ |
|
103
|
|
|
$this->getLog()->error('Shutting down...'); |
|
104
|
|
|
|
|
105
|
|
|
$this->getSlack()->disconnect(); |
|
106
|
|
|
$this->getLoop()->stop(); |
|
107
|
|
|
} |
|
108
|
|
|
|
|
109
|
|
|
public function reconnect() |
|
110
|
|
|
{ |
|
111
|
|
|
$this->getLog()->error('Reconnecting...'); |
|
112
|
|
|
|
|
113
|
|
|
if ($this->monitor) { |
|
114
|
|
|
$this->loop->cancelTimer($this->monitor); |
|
115
|
|
|
} |
|
116
|
|
|
|
|
117
|
|
|
$this->getSlack()->reconnect()->then( |
|
118
|
|
|
function () { |
|
119
|
|
|
$this->getLog()->info('Reconnected'); |
|
120
|
|
|
$this->monitor = $this->startConnectionMonitor(); |
|
121
|
|
|
}, |
|
122
|
|
|
function () { |
|
123
|
|
|
$this->getLog()->error('Reconnect failed, shutting down.'); |
|
124
|
|
|
$this->shutDown(); |
|
125
|
|
|
} |
|
126
|
|
|
); |
|
127
|
|
|
} |
|
128
|
|
|
|
|
129
|
|
|
public function connected() |
|
130
|
|
|
{ |
|
131
|
|
|
$slack = $this->getSlack(); |
|
132
|
|
|
|
|
133
|
|
|
$slack->update(function(User $authedUser) { |
|
134
|
|
|
$this->pluginManager->setAuthedUser($authedUser); |
|
135
|
|
|
}); |
|
136
|
|
|
|
|
137
|
|
|
$slack->onEvent('message', [$this, 'onMessage']); |
|
138
|
|
|
$slack->onEvent('team_join', [$this, 'onTeamJoin']); |
|
139
|
|
|
|
|
140
|
|
|
$this->monitor = $this->startConnectionMonitor(); |
|
141
|
|
|
} |
|
142
|
|
|
|
|
143
|
|
|
public function onMessage(Payload $payload) |
|
144
|
|
|
{ |
|
145
|
|
|
$data = $payload->getData(); |
|
146
|
|
|
|
|
147
|
|
|
$this->debug('Received message', $data); |
|
148
|
|
|
|
|
149
|
|
|
try { |
|
150
|
|
|
$this->logMessage($data); |
|
151
|
|
|
$message = $this->messageFactory->create($data); |
|
152
|
|
|
} catch (Throwable $throwable) { |
|
153
|
|
|
$errmsg = $throwable->getMessage()."\n" |
|
154
|
|
|
.$throwable->getTraceAsString()."\n" |
|
155
|
|
|
."Payload data: ".json_encode($data); |
|
156
|
|
|
$this->warning($errmsg); |
|
157
|
|
|
return; |
|
158
|
|
|
} |
|
159
|
|
|
|
|
160
|
|
|
if ($message->isSelf()) { |
|
161
|
|
|
return; |
|
162
|
|
|
} |
|
163
|
|
|
|
|
164
|
|
|
$this->pluginManager->dispatchMessage($message); |
|
165
|
|
|
} |
|
166
|
|
|
|
|
167
|
|
|
public function onTeamJoin(Payload $payload) |
|
|
|
|
|
|
168
|
|
|
{ |
|
169
|
|
|
$this->getSlack()->updateUsers(); |
|
170
|
|
|
} |
|
171
|
|
|
|
|
172
|
|
|
public function getHelp() : string |
|
173
|
|
|
{ |
|
174
|
|
|
return implode("\n", $this->pluginManager->getHelp()); |
|
175
|
|
|
} |
|
176
|
|
|
|
|
177
|
|
|
public function getStatus() : string |
|
178
|
|
|
{ |
|
179
|
|
|
$statuses = $this->pluginManager->getStatuses(); |
|
180
|
|
|
|
|
181
|
|
|
array_unshift($statuses, $this->getFormattedMemoryUsage()); |
|
182
|
|
|
|
|
183
|
|
|
return implode("\n", $statuses); |
|
184
|
|
|
} |
|
185
|
|
|
|
|
186
|
|
|
protected function addMemoryReporting() |
|
187
|
|
|
{ |
|
188
|
|
|
$now = new DateTime(); |
|
189
|
|
|
$then = new DateTime('+1 hour'); |
|
190
|
|
|
$then->setTime($then->format('H'), 0, 0); |
|
191
|
|
|
$delay = $then->getTimestamp() - $now->getTimestamp(); |
|
192
|
|
|
|
|
193
|
|
|
$this->addTimer($delay, function() { |
|
194
|
|
|
$this->info($this->getFormattedMemoryUsage()); |
|
195
|
|
|
$this->addPeriodicTimer(3600, function() { |
|
196
|
|
|
$this->info($this->getFormattedMemoryUsage()); |
|
197
|
|
|
}); |
|
198
|
|
|
}); |
|
199
|
|
|
} |
|
200
|
|
|
|
|
201
|
|
|
protected function getFormattedMemoryUsage() : string |
|
202
|
|
|
{ |
|
203
|
|
|
$memory = memory_get_usage() / 1024; |
|
204
|
|
|
$formatted = number_format($memory, 3).'K'; |
|
205
|
|
|
return "Current memory usage: {$formatted}"; |
|
206
|
|
|
} |
|
207
|
|
|
|
|
208
|
|
|
protected function logMessage($data) |
|
209
|
|
|
{ |
|
210
|
|
|
if ($this->messageLog !== null) { |
|
211
|
|
|
file_put_contents($this->messageLog, json_encode($data) . "\n", FILE_APPEND); |
|
212
|
|
|
} |
|
213
|
|
|
} |
|
214
|
|
|
|
|
215
|
|
|
/** |
|
216
|
|
|
* @return TimerInterface|null |
|
217
|
|
|
*/ |
|
218
|
|
|
protected function startConnectionMonitor() |
|
219
|
|
|
{ |
|
220
|
|
|
if ($interval = $this->get('connection_monitor.interval')) { |
|
221
|
|
|
|
|
222
|
|
|
$this->getLog()->info("Monitoring websocket connection every $interval seconds."); |
|
223
|
|
|
$this->notify("Monitoring websocket connection every $interval seconds."); |
|
224
|
|
|
|
|
225
|
|
|
$this->ping(); |
|
226
|
|
|
|
|
227
|
|
|
return $this->loop->addPeriodicTimer($interval, function () { |
|
228
|
|
|
$this->checkPong(); |
|
229
|
|
|
$this->ping(); |
|
230
|
|
|
}); |
|
231
|
|
|
} |
|
232
|
|
|
} |
|
233
|
|
|
|
|
234
|
|
|
protected function checkPong() |
|
235
|
|
|
{ |
|
236
|
|
|
if (!$this->pong) { |
|
237
|
|
|
$this->getLog()->error('No pong.'); |
|
238
|
|
|
|
|
239
|
|
|
$failureStrategy = $this->get('connection_monitor.failure_strategy', 'reconnect'); |
|
240
|
|
|
|
|
241
|
|
|
if ($failureStrategy === 'reconnect') { |
|
242
|
|
|
$this->reconnect(); |
|
243
|
|
|
return; |
|
244
|
|
|
} |
|
245
|
|
|
|
|
246
|
|
|
if ($failureStrategy !== 'shutdown') { |
|
247
|
|
|
$this->getLog()->error("Unknown connection_monitor.failure_strategy '$failureStrategy'"); |
|
248
|
|
|
} |
|
249
|
|
|
|
|
250
|
|
|
$this->shutDown(); |
|
251
|
|
|
} |
|
252
|
|
|
} |
|
253
|
|
|
|
|
254
|
|
|
protected function ping() |
|
255
|
|
|
{ |
|
256
|
|
|
$this->pong = false; |
|
257
|
|
|
|
|
258
|
|
|
$this->getSlack()->ping() |
|
259
|
|
|
->then( |
|
260
|
|
|
function (Payload $payload) { |
|
261
|
|
|
$this->getLog()->info($payload->toJson()); |
|
262
|
|
|
$this->pong = true; |
|
263
|
|
|
} |
|
264
|
|
|
); |
|
265
|
|
|
} |
|
266
|
|
|
|
|
267
|
|
|
protected function notify(string $message) |
|
268
|
|
|
{ |
|
269
|
|
|
if ($user = $this->get('notify.user')) { |
|
270
|
|
|
$this->getSlack()->directMessage($message, $user); |
|
271
|
|
|
} |
|
272
|
|
|
} |
|
273
|
|
|
|
|
274
|
|
|
protected function connect() |
|
275
|
|
|
{ |
|
276
|
|
|
Timer\timeout($this->getSlack()->connect(), 30, $this->getLoop()) |
|
|
|
|
|
|
277
|
|
|
->then(function () { |
|
278
|
|
|
$this->getLog()->info('Connected.'); |
|
279
|
|
|
$this->connected(); |
|
280
|
|
|
}) |
|
281
|
|
|
->otherwise(function (Timer\TimeoutException $error) { |
|
282
|
|
|
$this->getLog()->error($error->getMessage()); |
|
283
|
|
|
$this->getLog()->error('Connection failed, shutting down.'); |
|
284
|
|
|
$this->shutDown(); |
|
285
|
|
|
}) |
|
286
|
|
|
->otherwise(function ($error) { |
|
|
|
|
|
|
287
|
|
|
$this->getLog()->error('Connection failed, shutting down.'); |
|
288
|
|
|
$this->shutDown(); |
|
289
|
|
|
}); |
|
290
|
|
|
} |
|
291
|
|
|
} |
|
292
|
|
|
|
This check looks from parameters that have been defined for a function or method, but which are not used in the method body.