3 namespace App\TwitchBot;
5 use App\Models\Channel;
6 use App\Models\ChatBotLog;
7 use App\Models\ChatLog;
8 use Illuminate\Support\Arr;
9 use Illuminate\Support\Str;
11 class TwitchChatBot extends TwitchBot {
13 public function __construct() {
14 parent::__construct('horstiebot');
15 $this->updateChannels();
17 $this->listenCommands();
20 public function joinChannels() {
21 $this->getLogger()->info('joining channels');
23 foreach ($this->channels as $channel) {
24 $names[] = $channel->twitch_chat;
26 $chunks = array_chunk($names, 10);
27 foreach ($chunks as $chunk) {
28 $this->sendIRCMessage(IRCMessage::join($chunk));
32 public function logMessage(IRCMessage $msg) {
33 $channel = $this->getMessageChannel($msg);
34 if ($channel && !$channel->join) {
39 public function handlePrivMsg(IRCMessage $msg) {
40 if ($msg->nick == 'horstiebot') return;
41 $channel = $this->getMessageChannel($msg);
42 if (!$channel) return;
43 $this->tagChannelRead($channel, $msg);
47 private function startTimer() {
48 $this->getLoop()->addPeriodicTimer(1, function () {
49 if (!$this->isReady()) return;
50 foreach ($this->channels as $channel) {
51 $this->decideSend($channel);
54 $this->getLoop()->addPeriodicTimer(60, function () {
55 $this->updateChannels();
59 private function updateChannels() {
60 $this->channels = Channel::where('twitch_chat', '!=', '')->where('chat', '=', true)->get();
63 private function decideSend(Channel $channel) {
64 $notes = $this->getNotes($channel);
65 if ($notes['read_since_last_write'] < $notes['wait_msgs']) {
68 if (time() - $notes['last_write'] < $notes['wait_time']) {
71 if ($notes['read_since_last_write'] == $notes['wait_msgs'] && time() - $notes['last_read'] < 3) {
72 // don't immediately respond if we crossed the msg threshold last
75 $text = $this->contextualMsg($channel);
76 if (!$text) $text = $this->randomChat($channel);
78 $actual_text = is_object($text) ? $text->text_content : $text;
79 $this->tagChannelWrite($channel);
80 $this->sendIRCMessage(IRCMessage::privmsg($channel->twitch_chat, $actual_text));
81 $log = new ChatBotLog();
82 $log->channel()->associate($channel);
83 if (is_object($text)) {
84 $log->origin()->associate($text);
86 $log->text = $actual_text;
90 private function getNotes(Channel $channel) {
91 if (!isset($this->notes[$channel->id])) {
92 $this->notes[$channel->id] = [
95 'last_write' => time(),
97 'read_since_last_write' => 0,
98 'wait_msgs' => $this->randomWaitMsgs($channel),
99 'wait_time' => $this->randomWaitTime($channel),
102 return $this->notes[$channel->id];
105 private function getNote(Channel $channel, $name, $default = null) {
106 $notes = $this->getNotes($channel);
107 if (array_key_exists($name, $notes)) {
108 return $notes[$name];
113 private function setNote(Channel $channel, $name, $value) {
114 $this->getNotes($channel);
115 $this->notes[$channel->id][$name] = $value;
118 private function collectClassifications(Channel $channel) {
119 $classifications = [];
120 $notes = $this->getNotes($channel);
121 foreach ($notes['latest_msgs'] as $msg) {
122 $classification = $msg->classify();
123 if ($classification == 'unclassified') continue;
124 if (isset($classifications[$classification])) {
125 ++$classifications[$classification];
127 $classifications[$classification] = 1;
130 arsort($classifications);
131 return $classifications;
134 private function contextualMsg(Channel $channel) {
135 $last = $this->getNote($channel, 'last_special');
136 $classifications = $this->collectClassifications($channel);
157 foreach ($classifications as $classification => $count) {
158 if (!isset($count_quotas[$classification]) || $count < $count_quotas[$classification]) continue;
159 if (!isset($time_quotas[$classification]) || $this->getTimeSinceSpecial($channel, $classification) < $time_quotas[$classification]) continue;
160 $this->tagChannelSpecialSent($channel, $classification);
161 if ($classification == 'number') {
162 return $this->randomContextualNumber($channel);
164 if ($classification == 'lol') {
165 return $this->randomLaughter($channel);
167 return $channel->randomOfClass($classification);
172 private function randomChat(Channel $channel) {
173 return $channel->queryChatlog()
174 ->whereIn('classification', ['hi', 'hype', 'lol', 'pog', 'unclassified'])
178 private function randomContextualNumber(Channel $channel) {
179 $notes = $this->getNotes($channel);
182 foreach ($notes['latest_msgs'] as $msg) {
183 if ($msg->classify() == 'number') {
184 $number = $msg->getNumericValue();
185 $min = min($min, $number);
186 $max = max($max, $number);
189 return random_int($min, $max);
192 private function randomLaughter(Channel $channel) {
222 $channel->randomOfClass('lol'),
226 private function randomMsg(Channel $channel) {
227 return $channel->queryChatlog()->first();
230 private function randomWaitMsgs(Channel $channel) {
231 $min = $channel->getChatSetting('wait_msgs_min', 1);
232 $max = $channel->getChatSetting('wait_msgs_max', 10);
233 return random_int($min, $max);
236 private function randomWaitTime(Channel $channel) {
237 $min = $channel->getChatSetting('wait_time_min', 1);
238 $max = $channel->getChatSetting('wait_time_max', 900);
239 return random_int($min, $max);
242 private function tagChannelRead(Channel $channel, IRCMessage $msg) {
243 $this->getNotes($channel);
244 $this->notes[$channel->id]['last_read'] = time();
245 ++$this->notes[$channel->id]['read_since_last_write'];
247 $tokenized = $msg->tokenize();
248 if (!ChatLog::isKnownBot($msg->nick) && !$tokenized->isSpammy()) {
249 $this->notes[$channel->id]['latest_msgs'][] = $tokenized;
250 if (count($this->notes[$channel->id]['latest_msgs']) > 10) {
251 array_shift($this->notes[$channel->id]['latest_msgs']);
254 if ($this->isDirectedAtMe($msg->getText()) && $this->shouldRespond($channel)) {
255 $this->notes[$channel->id]['wait_msgs'] = 0;
256 $this->notes[$channel->id]['wait_time'] = 0;
260 private function tagChannelWrite(Channel $channel) {
261 $this->getNotes($channel);
262 $this->notes[$channel->id]['last_write'] = time();
263 $this->notes[$channel->id]['read_since_last_write'] = 0;
264 $this->notes[$channel->id]['wait_msgs'] = $this->randomWaitMsgs($channel);
265 $this->notes[$channel->id]['wait_time'] = $this->randomWaitTime($channel);
268 private function tagChannelSpecialSent(Channel $channel, $classification) {
269 $this->getNotes($channel);
270 $this->notes[$channel->id]['last_special'][$classification] = time();
273 private function getTimeSinceSpecial(Channel $channel, $classification) {
274 $notes = $this->getNotes($channel);
275 if (isset($notes['last_special'][$classification])) {
276 return time() - $notes['last_special'][$classification];
281 private function isDirectedAtMe($raw_text) {
282 $text = strtolower($raw_text);
283 if (strpos($text, 'horsti') !== false) {
289 private function shouldRespond(Channel $channel) {
290 $setting = $channel->getChatSetting('respond', 'yes');
291 if ($setting == 'yes') {
294 if ($setting == '50') {
295 return random_int(0, 1);