better quota system for contextual messages

[alttp.git] / app / Models / ChatLog.php

diff --git a/app/Models/ChatLog.php b/app/Models/ChatLog.php
index 3b4df15f2ac9596846729b91fc564c5780ca68b8..d5fc76d9325c565b803ec16eaa92c465366822ac 100644 (file)
--- a/app/Models/ChatLog.php
+++ b/app/Models/ChatLog.php
@@ -2,6 +2,7 @@
 
 namespace App\Models;
 
+use App\TwitchBot\TokenizedMessage;
 use Illuminate\Database\Eloquent\Factories\HasFactory;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\Arr;
@@ -20,6 +21,10 @@ class ChatLog extends Model {
                return $this->belongsTo(User::class);
        }
 
+       public function tokenize() {
+               return TokenizedMessage::fromLog($this);
+       }
+
        public function getTextWithoutEmotes() {
                $text = $this->text_content;
                if (isset($this->tags['emotes']) && !empty($this->tags['emotes'])) {
@@ -61,10 +66,11 @@ class ChatLog extends Model {
                        }
                        $this->text_content = $this->params[1];
                        $this->detectLanguage();
-                       if ($this->scanForSpam()) {
+                       $tokenized = $this->tokenize();
+                       if ($tokenized->isSpammy()) {
                                $this->banned = true;
                        }
-                       $this->classification = static::classify($this->text_content);
+                       $this->classification = $tokenized->classify();
                        return;
                }
 
@@ -86,39 +92,6 @@ class ChatLog extends Model {
                ]);
        }
 
-       public static function classify($text) {
-               if (empty($text)) {
-                       return 'unclassified';
-               }
-               if (is_numeric(trim($text))) {
-                       return 'number';
-               }
-               $rawText = strtolower(preg_replace('/[^\w]/', '', $text));
-               $tokenizedText = preg_split('/\s+/', strtolower(trim($text)));
-               if (Str::startsWith($rawText, 'gg') || Str::endsWith($rawText, 'gg')) {
-                       return 'gg';
-               }
-               if (Str::contains($rawText, ['glgl', 'glhf', 'hfgl'])) {
-                       return 'gl';
-               }
-               if (Str::contains($rawText, ['haha', 'hehe', 'hihi', 'kekw', 'lol', 'lul', 'xd'])) {
-                       return 'lol';
-               }
-               if (Str::startsWith($rawText, ['ahoi', 'hallo', 'hello', 'hi ', 'huhu']) || Str::endsWith($rawText, ['hi', 'wave'])) {
-                       return 'hi';
-               }
-               if (Str::contains($rawText, ['pog', 'wow'])) {
-                       return 'pog';
-               }
-               if (Str::contains($rawText, ['hype'])) {
-                       return 'hype';
-               }
-               if (Str::startsWith($rawText, 'o7') || Str::endsWith($rawText, 'o7') || Str::contains($rawText, 'salut')) {
-                       return 'o7';
-               }
-               return 'unclassified';
-       }
-
        protected function evaluateUser() {
        }
 
@@ -152,54 +125,6 @@ class ChatLog extends Model {
                }
        }
 
-       public static function spammyText($raw_text) {
-               $text = strtolower($raw_text);
-               if (substr($text, 0, 1) == '!') {
-                       return true;
-               }
-               if (strpos($text, '$') !== false) {
-                       return true;
-               }
-               if (strpos($text, '€') !== false) {
-                       return true;
-               }
-               if (strpos($text, '@') !== false) {
-                       return true;
-               }
-               if (strpos($text, '://') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'followers') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'horstie') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'promotion') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'viewers') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'view ers') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'vielen dank für den raid') !== false) {
-                       return true;
-               }
-               if (strpos($text, 'willkommen auf starbase 47') !== false) {
-                       return true;
-               }
-               return false;
-       }
-
-       protected function scanForSpam() {
-               if (is_numeric($this->text_content)) {
-                       return true;
-               }
-               return static::spammyText($this->text_content);
-       }
-
        protected $casts = [
                'banned' => 'boolean',
                'params' => 'array',