]> git.localhorst.tv Git - alttp.git/blobdiff - app/TwitchBot/TwitchChatBot.php
improved responses
[alttp.git] / app / TwitchBot / TwitchChatBot.php
index 470cb24f5f34fc717402051143213a6bc9842b42..442173d8c60c82228d66ffdd52e2b1ff5744ef07 100644 (file)
@@ -3,6 +3,7 @@
 namespace App\TwitchBot;
 
 use App\Models\Channel;
+use App\Models\ChatBotLog;
 use App\Models\ChatLog;
 use Illuminate\Support\Arr;
 use Illuminate\Support\Str;
@@ -72,26 +73,28 @@ class TwitchChatBot extends TwitchBot {
                        return;
                }
                $text = $this->contextualMsg($channel);
-               if (!$text) $text = $this->randomMsg($channel);
+               if (!$text) $text = $this->randomChat($channel);
                if (!$text) return;
+               $actual_text = is_object($text) ? $text->text_content : $text;
                $this->tagChannelWrite($channel);
-               $this->sendIRCMessage(IRCMessage::privmsg($channel->twitch_chat, $text));
-       }
-
-       private function getChatSetting(Channel $channel, $name, $default = null) {
-               if (array_key_exists($name, $channel->chat_settings)) {
-                       return $channel->chat_settings[$name];
+               $this->sendIRCMessage(IRCMessage::privmsg($channel->twitch_chat, $actual_text));
+               $log = new ChatBotLog();
+               $log->channel()->associate($channel);
+               if (is_object($text)) {
+                       $log->origin()->associate($text);
                }
-               return $default;
+               $log->text = $actual_text;
+               $log->save();
        }
 
        private function getNotes(Channel $channel) {
                if (!isset($this->notes[$channel->id])) {
                        $this->notes[$channel->id] = [
                                'last_read' => 0,
-                               'last_special' => '',
+                               'last_special' => [],
                                'last_write' => time(),
                                'latest_msgs' => [],
+                               'queued_special' => false,
                                'read_since_last_write' => 0,
                                'wait_msgs' => $this->randomWaitMsgs($channel),
                                'wait_time' => $this->randomWaitTime($channel),
@@ -113,92 +116,86 @@ class TwitchChatBot extends TwitchBot {
                $this->notes[$channel->id][$name] = $value;
        }
 
-       private function checkForGG(Channel $channel) {
-               $notes = $this->getNotes($channel);
-               $ggs = 0;
-               foreach ($notes['latest_msgs'] as $text) {
-                       $rawText = strtolower(preg_replace('/[^\w]/', '', $text));
-                       if (Str::startsWith($rawText, 'gg') || Str::endsWith($rawText, 'gg')) {
-                               ++$ggs;
-                       }
-               }
-               return $ggs > 2;
-       }
-
-       private function checkForGLHF(Channel $channel) {
-               $notes = $this->getNotes($channel);
-               $gls = 0;
-               foreach ($notes['latest_msgs'] as $text) {
-                       $rawText = strtolower(preg_replace('/[^\w]/', '', $text));
-                       if (Str::contains($rawText, ['glgl', 'glhf', 'hfgl'])) {
-                               ++$gls;
-                       }
-               }
-               return $gls > 2;
-       }
-
-       private function checkForLaughter(Channel $channel) {
-               $notes = $this->getNotes($channel);
-               $lulz = 0;
-               foreach ($notes['latest_msgs'] as $text) {
-                       $rawText = strtolower(preg_replace('/[^\w]/', '', $text));
-                       if (Str::contains($rawText, ['haha', 'kekw', 'lol', 'lul', 'xd'])) {
-                               ++$lulz;
-                       }
-               }
-               return $lulz > 2;
-       }
-
-       private function checkForNumbers(Channel $channel) {
+       private function collectClassifications(Channel $channel) {
+               $classifications = [];
                $notes = $this->getNotes($channel);
-               $numbers = 0;
-               foreach ($notes['latest_msgs'] as $text) {
-                       if (is_numeric(trim($text))) {
-                               ++$numbers;
+               foreach ($notes['latest_msgs'] as $msg) {
+                       $classification = $msg->classify();
+                       if ($classification == 'unclassified') continue;
+                       if (isset($classifications[$classification])) {
+                               ++$classifications[$classification];
+                       } else {
+                               $classifications[$classification] = 1;
                        }
                }
-               return $numbers > 2;
+               arsort($classifications);
+               return $classifications;
        }
 
        private function contextualMsg(Channel $channel) {
-               $last = $this->getNote($channel, 'last_special');
-               if ($last != 'gg' && $this->checkForGG($channel)) {
-                       $this->setNote($channel, 'last_special', 'gg');
-                       return $this->randomGG($channel);
-               }
-               if ($last != 'number' && $this->checkForNumbers($channel)) {
-                       $this->setNote($channel, 'last_special', 'number');
-                       return $this->randomContextualNumber($channel);
+               if ($this->hasQueuedSpecial($channel)) {
+                       $classification = $this->getQueuedSpecial($channel);
+                       if (is_string($classification)) {
+                               $this->tagChannelSpecialSent($channel, $classification);
+                       }
+                       $this->clearQueuedSpecial($channel);
+                       return $this->getRandomOfClass($channel, $classification);
                }
-               if ($last != 'lol' && $this->checkForLaughter($channel)) {
-                       $this->setNote($channel, 'last_special', 'lol');
-                       return $this->randomLaughter($channel);
+               $latest_msg = $this->getLatestMessage($channel);
+               if ($latest_msg->classify() == 'question') {
+                       $response = $latest_msg->getResponseCategory();
+                       return $this->getRandomOfClass($channel, $response);
                }
-               if ($last != 'glhf' && $this->checkForGLHF($channel)) {
-                       $this->setNote($channel, 'last_special', 'glhf');
-                       return $this->randomGLHF($channel);
+               $last = $this->getLastSpecialSent($channel);
+               $classifications = $this->collectClassifications($channel);
+               $count_quotas = [
+                       'gg' => 2,
+                       'gl' => 2,
+                       'hi' => 2,
+                       'hype' => 2,
+                       'lol' => 2,
+                       'love' => 2,
+                       'number' => 2,
+                       'pog' => 2,
+                       'o7' => 2,
+                       'wtf' => 2,
+               ];
+               $time_quotas = [
+                       'gg' => 600,
+                       'gl' => 900,
+                       'hi' => 60,
+                       'hype' => 60,
+                       'lol' => 60,
+                       'love' => 60,
+                       'number' => 300,
+                       'pog' => 60,
+                       'o7' => 300,
+                       'wtf' => 60,
+               ];
+               foreach ($classifications as $classification => $count) {
+                       if ($classification == $last) continue;
+                       if (!isset($count_quotas[$classification]) || $count < $count_quotas[$classification]) continue;
+                       if (!isset($time_quotas[$classification]) || $this->getTimeSinceSpecial($channel, $classification) < $time_quotas[$classification]) continue;
+                       $this->tagChannelSpecialSent($channel, $classification);
+                       $reaction = $this->getChimeInReaction($channel, $classification);
+                       return $this->getRandomOfClass($channel, $reaction);
                }
                return false;
        }
 
-       private function queryChatlog(Channel $channel) {
-               return ChatLog::where('type', '=', 'chat')
-                       ->where('banned', '=', false)
-                       ->where('created_at', '<', now()->sub(1, 'day'))
-                       ->where(function ($query) use ($channel) {
-                               $query->whereNull('detected_language');
-                               $query->orWhereIn('detected_language', $channel->languages);
-                       })
-                       ->inRandomOrder();
+       private function randomChat(Channel $channel) {
+               return $channel->queryChatlog()
+                       ->whereNotIn('classification', ['gg', 'gl', 'number', 'o7'])
+                       ->first();
        }
 
        private function randomContextualNumber(Channel $channel) {
                $notes = $this->getNotes($channel);
                $min = 100000;
                $max = 0;
-               foreach ($notes['latest_msgs'] as $text) {
-                       if (is_numeric(trim($text))) {
-                               $number = intval(trim($text));
+               foreach ($notes['latest_msgs'] as $msg) {
+                       if ($msg->classify() == 'number') {
+                               $number = $msg->getNumericValue();
                                $min = min($min, $number);
                                $max = max($max, $number);
                        }
@@ -206,23 +203,10 @@ class TwitchChatBot extends TwitchBot {
                return random_int($min, $max);
        }
 
-       private function randomGG(Channel $channel) {
-               $line = $this->queryChatlog($channel)
-                       ->where('text_content', 'LIKE', '%gg')
-                       ->whereRaw('LENGTH(`text_content`) < 30')
-                       ->first();
-               return $line->text_content;
-       }
-
-       private function randomGLHF(Channel $channel) {
-               $line = $this->queryChatlog($channel)
-                       ->where('text_content', 'LIKE', '%glhf%')
-                       ->whereRaw('LENGTH(`text_content`) < 30')
-                       ->first();
-               return $line->text_content;
-       }
-
        private function randomLaughter(Channel $channel) {
+               if (!random_int(0, 2)) {
+                       return $channel->randomOfClass('lol');
+               }
                return Arr::random([
                        ':tf:',
                        '4Head',
@@ -233,7 +217,6 @@ class TwitchChatBot extends TwitchBot {
                        'GunRun',
                        'heh',
                        'Hhhehehe',
-                       'HypeLUL',
                        'Jebaited',
                        'Jebasted',
                        'KEKW',
@@ -256,29 +239,59 @@ class TwitchChatBot extends TwitchBot {
        }
 
        private function randomMsg(Channel $channel) {
-               $line = $this->queryChatlog($channel)->first();
-               return $line->text_content;
+               return $channel->queryChatlog()->first();
        }
 
        private function randomWaitMsgs(Channel $channel) {
-               $min = $this->getChatSetting($channel, 'wait_msgs_min', 1);
-               $max = $this->getChatSetting($channel, 'wait_msgs_max', 10);
+               $min = $channel->getChatSetting('wait_msgs_min', 1);
+               $max = $channel->getChatSetting('wait_msgs_max', 10);
                return random_int($min, $max);
        }
 
        private function randomWaitTime(Channel $channel) {
-               $min = $this->getChatSetting($channel, 'wait_time_min', 1);
-               $max = $this->getChatSetting($channel, 'wait_time_max', 900);
+               $min = $channel->getChatSetting('wait_time_min', 1);
+               $max = $channel->getChatSetting('wait_time_max', 900);
                return random_int($min, $max);
        }
 
+       private function queueSpecial(Channel $channel, $classification) {
+               $this->getNotes($channel);
+               $this->notes[$channel->id]['queued_special'] = $classification;
+       }
+
+       private function hasQueuedSpecial(Channel $channel) {
+               return !!$this->getQueuedSpecial($channel);
+       }
+
+       private function getQueuedSpecial(Channel $channel) {
+               $notes = $this->getNotes($channel);
+               return $notes['queued_special'];
+       }
+
+       private function clearQueuedSpecial(Channel $channel) {
+               $this->getNotes($channel);
+               $this->notes[$channel->id]['queued_special'] = false;
+       }
+
        private function tagChannelRead(Channel $channel, IRCMessage $msg) {
                $this->getNotes($channel);
                $this->notes[$channel->id]['last_read'] = time();
                ++$this->notes[$channel->id]['read_since_last_write'];
-               $this->notes[$channel->id]['latest_msgs'][] = $msg->getText();
-               if (count($this->notes[$channel->id]['latest_msgs']) > 10) {
-                       array_shift($this->notes[$channel->id]['latest_msgs']);
+
+               $tokenized = $msg->tokenize();
+               if (!ChatLog::isKnownBot($msg->nick) && !$tokenized->isSpammy()) {
+                       $this->notes[$channel->id]['latest_msgs'][] = $tokenized;
+                       if (count($this->notes[$channel->id]['latest_msgs']) > 10) {
+                               array_shift($this->notes[$channel->id]['latest_msgs']);
+                       }
+               }
+               if ($this->isDirectedAtMe($msg->getText()) && $this->shouldRespond($channel)) {
+                       $this->notes[$channel->id]['wait_msgs'] = 0;
+                       $this->notes[$channel->id]['wait_time'] = 0;
+                       $response = $tokenized->getResponseCategory();
+                       if ($response) {
+                               $this->queueSpecial($channel, $response);
+                       }
                }
        }
 
@@ -290,6 +303,85 @@ class TwitchChatBot extends TwitchBot {
                $this->notes[$channel->id]['wait_time'] = $this->randomWaitTime($channel);
        }
 
+       private function tagChannelSpecialSent(Channel $channel, $classification) {
+               $this->getNotes($channel);
+               $this->notes[$channel->id]['last_special'][$classification] = time();
+       }
+
+       private function getLatestMessage(Channel $channel) {
+               $this->getNotes($channel);
+               if (!empty($notes['latest_msgs'])) {
+                       return $notes['latest_msgs'][count($notes['latest_msgs']) - 1];
+               }
+               return TokenizedMessage::fromString('');
+       }
+
+       private function getLastSpecialSent(Channel $channel) {
+               $notes = $this->getNotes($channel);
+               $max_time = 0;
+               $max_classification = '';
+               foreach ($notes['last_special'] as $classification => $time) {
+                       if ($time > $max_time) {
+                               $max_time = $time;
+                               $max_classification = $classification;
+                       }
+               }
+               return $max_classification;
+       }
+
+       private function getTimeSinceSpecial(Channel $channel, $classification) {
+               $notes = $this->getNotes($channel);
+               if (isset($notes['last_special'][$classification])) {
+                       return time() - $notes['last_special'][$classification];
+               }
+               return 999999;
+       }
+
+       private function isDirectedAtMe($raw_text) {
+               $text = strtolower($raw_text);
+               if (strpos($text, 'horsti') !== false) {
+                       return true;
+               }
+               return false;
+       }
+
+       private function shouldRespond(Channel $channel) {
+               $setting = $channel->getChatSetting('respond', 'yes');
+               if ($setting == 'yes') {
+                       return true;
+               }
+               if ($setting == '50') {
+                       return random_int(0, 1);
+               }
+               return false;
+       }
+
+       private function getRandomOfClass(Channel $channel, $classification) {
+               if ($classification == 'number') {
+                       return $this->randomContextualNumber($channel);
+               }
+               if ($classification == 'lol') {
+                       return $this->randomLaughter($channel);
+               }
+               return $channel->randomOfClass($classification);
+       }
+
+       private function getChimeInReaction(Channel $channel, $classification) {
+               switch ($classification) {
+                       case 'hi':
+                               return ['hi', 'love'];
+                       case 'hype':
+                               return ['hype', 'love', 'pog'];
+                       case 'lol':
+                               return ['kappa', 'lol'];
+                       case 'pog':
+                               return ['hype', 'pog'];
+                       case 'wtf':
+                               return ['lol', 'wtf'];
+               }
+               return $classification;
+       }
+
        private $channels;
        private $notes = [];