]> git.localhorst.tv Git - alttp.git/blob - app/TwitchBot/TwitchBot.php
more spam
[alttp.git] / app / TwitchBot / TwitchBot.php
1 <?php
2
3 namespace App\TwitchBot;
4
5 use App\Models\Channel;
6 use App\Models\TwitchBotCommand;
7 use App\Models\TwitchToken;
8 use Monolog\Handler\StreamHandler;
9 use Monolog\Logger;
10 use Ratchet\Client\Connector;
11 use Ratchet\Client\WebSocket;
12 use Ratchet\RFC6455\Messaging\Message;
13 use React\EventLoop\Loop;
14
15 class TwitchBot {
16
17         public function __construct($nick) {
18                 $this->nick = $nick;
19                 $this->logger = new Logger('TwitchBot');
20                 $this->logger->pushHandler(new StreamHandler('php://stdout', Logger::INFO));
21
22                 $this->token = TwitchToken::firstWhere('nick', $nick);
23                 if (!$this->token) {
24                         throw new \Exception('unable to find access token');
25                 }
26                 if ($this->token->hasExpired()) {
27                         $this->token->refresh();
28                 }
29
30                 $this->connector = new Connector();
31                 $this->connect();
32                 $this->startPinger();
33         }
34
35         public function getLogger() {
36                 return $this->logger;
37         }
38
39         public function getLoop() {
40                 return Loop::get();
41         }
42
43         public function isReady() {
44                 return $this->ready;
45         }
46
47         public function run() {
48                 $this->shutting_down = false;
49                 $this->getLoop()->run();
50         }
51
52         public function stop() {
53                 $this->logger->info('shutting down');
54                 $this->shutting_down = true;
55                 $this->disconnect();
56                 $this->getLoop()->stop();
57         }
58
59
60         public function connect() {
61                 ($this->connector)('wss://irc-ws.chat.twitch.tv:443')->done(
62                         [$this, 'handleWsConnect'],
63                         [$this, 'handleWsConnectError'],
64                 );
65         }
66
67         public function disconnect() {
68                 $this->ws->close();
69         }
70
71         public function handleWsConnect(WebSocket $ws) {
72                 $this->logger->info('websocket connection established');
73                 $this->ws = $ws;
74                 $ws->on('message', [$this, 'handleWsMessage']);
75                 $ws->on('close', [$this, 'handleWsClose']);
76                 $ws->on('error', [$this, 'handleWsError']);
77                 $ws->send('CAP REQ :twitch.tv/tags twitch.tv/commands');
78                 $this->login();
79         }
80
81         public function handleWsConnectError(WebSocket $ws) {
82                 $this->logger->error('failed to establish websocket connection');
83         }
84
85         public function handleWsMessage(Message $message, WebSocket $ws) {
86                 $irc_messages = explode("\r\n", rtrim($message->getPayload(), "\r\n"));
87                 foreach ($irc_messages as $irc_message) {
88                         $this->logger->info('received IRC message '.$irc_message);
89                         $this->handleIRCMessage(IRCMessage::fromString($irc_message));
90                 }
91         }
92
93         public function handleWsClose(int $op, string $reason) {
94                 $this->ready = false;
95                 $this->logger->info('websocket connection closed: '.$reason.' ['.$op.']');
96                 if (!$this->shutting_down) {
97                         $this->logger->info('reconnecting in 5 seconds');
98                         Loop::addTimer(5, [$this, 'connect']);
99                 }
100         }
101
102         public function handleWsError(\Exception $e, WebSocket $ws) {
103                 $this->logger->error('websocket error '.$e->getMessage());
104         }
105
106
107         public function handleIRCMessage(IRCMessage $msg) {
108                 $this->last_contact = time();
109                 if ($msg->isPing()) {
110                         $this->sendIRCMessage($msg->makePong());
111                         return;
112                 }
113                 if ($msg->isPong()) {
114                         return;
115                 }
116                 $this->logMessage($msg);
117                 if ($msg->isPrivMsg()) {
118                         $this->handlePrivMsg($msg);
119                         return;
120                 }
121                 if ($msg->isNotice() && $msg->getText() == 'Login authentication failed') {
122                         $this->logger->notice('login failed, refreshing access token');
123                         $this->token->refresh();
124                         $this->login();
125                         return;
126                 }
127                 if ($msg->command == '001') {
128                         // successful login
129                         $this->joinChannels();
130                         $this->ready = true;
131                         return;
132                 }
133         }
134
135         public function getMessageChannel(IRCMessage $msg) {
136                 $target = $msg->getPrivMsgTarget();
137                 if (substr($target, 0, 1) !== '#') {
138                         $target = '#'.$target;
139                 }
140                 return Channel::firstWhere('twitch_chat', '=', $target);
141         }
142
143         public function logMessage(IRCMessage $msg) {
144         }
145
146         public function handlePrivMsg(IRCMessage $msg) {
147         }
148
149         public function login() {
150                 $this->ws->send('PASS oauth:'.$this->token->access);
151                 $this->ws->send('NICK '.$this->nick);
152         }
153
154         public function joinChannels() {
155         }
156
157         private function startPinger() {
158                 $this->getLoop()->addPeriodicTimer(15, function () {
159                         if (!$this->ready) return;
160                         if (time() - $this->last_contact < 60) return;
161                         try {
162                                 $this->sendIRCMessage(IRCMessage::ping($this->nick));
163                         } catch (\Exception $e) {
164                         }
165                 });
166         }
167
168         public function sendIRCMessage(IRCMessage $msg) {
169                 $irc_message = $msg->encode();
170                 $this->logger->info('sending IRC message '.$irc_message);
171                 $this->ws->send($irc_message);
172                 $this->last_contact = time();
173         }
174
175
176         protected function listenCommands() {
177                 $this->getLoop()->addPeriodicTimer(1, function () {
178                         if (!$this->isReady()) return;
179                         $command = TwitchBotCommand::where('bot_nick', '=', $this->nick)->where('status', '=', 'pending')->oldest()->first();
180                         if ($command) {
181                                 try {
182                                         $command->execute($this);
183                                 } catch (\Exception $e) {
184                                 }
185                         }
186                 });
187         }
188
189
190         private $logger;
191
192         private $nick;
193         private $token;
194
195         private $connector;
196         private $ws;
197         private $ready = false;
198         private $shutting_down = false;
199
200         private $last_contact;
201
202 }
203
204 ?>