more classification

[alttp.git] / app / TwitchBot / TokenizedMessage.php

diff --git a/app/TwitchBot/TokenizedMessage.php b/app/TwitchBot/TokenizedMessage.php
index 99947374d134a8301773cf866153a22d179ab6d1..0e241d5cd804c490e72f761d01cba7ae70357e5d 100644 (file)
--- a/app/TwitchBot/TokenizedMessage.php
+++ b/app/TwitchBot/TokenizedMessage.php
@@ -22,7 +22,7 @@ class TokenizedMessage {
                                $positions = explode(',', $set[1]);
                                foreach ($positions as $position) {
                                        $coords = explode('-', $position);
-                                       $this->emotes[] = strtolower(substr($this->text, $coords[0], $coords[1] - $coords[0] + 1));
+                                       $this->emotes[] = preg_replace('/\d+$/', '', strtolower(substr($this->text, $coords[0], $coords[1] - $coords[0] + 1)));
                                        for ($i = intval($coords[0]); $i <= intval($coords[1]); ++$i) {
                                                $this->emoteless[$i] = ' ';
                                        }
@@ -51,6 +51,10 @@ class TokenizedMessage {
                return Str::contains($this->text, $text);
        }
 
+       public function containsEmoteless($text) {
+               return Str::contains($this->emoteless, $text);
+       }
+
        public function containsRaw($text) {
                return Str::contains($this->raw, $text);
        }
@@ -59,6 +63,10 @@ class TokenizedMessage {
                return Str::endsWith($this->text, $text);
        }
 
+       public function endsWithEmoteless($text) {
+               return Str::endsWith($this->emoteless, $text);
+       }
+
        public function endsWithRaw($text) {
                return Str::endsWith($this->raw, $text);
        }
@@ -68,6 +76,14 @@ class TokenizedMessage {
        }
 
        public function hasEmote($text) {
+               if (is_array($text)) {
+                       foreach ($text as $token) {
+                               if (in_array($token, $this->emotes)) {
+                                       return true;
+                               }
+                       }
+                       return false;
+               }
                return in_array($text, $this->emotes);
        }
 
@@ -108,6 +124,14 @@ class TokenizedMessage {
        }
 
        public function hasToken($text) {
+               if (is_array($text)) {
+                       foreach ($text as $token) {
+                               if (in_array($token, $this->tokens)) {
+                                       return true;
+                               }
+                       }
+                       return false;
+               }
                return in_array($text, $this->tokens);
        }
 
@@ -159,6 +183,10 @@ class TokenizedMessage {
                return Str::startsWith($this->text, $text);
        }
 
+       public function startsWithEmoteless($text) {
+               return Str::startsWith($this->emoteless, $text);
+       }
+
        public function startsWithRaw($text) {
                return Str::startsWith($this->raw, $text);
        }
@@ -177,7 +205,7 @@ class TokenizedMessage {
                if ($this->containsRaw('horstie')) {
                        return true;
                }
-               if ($this->containsRaw(['vielendankfürdenraid', 'willkommenaufstarbase47'])) {
+               if ($this->containsRaw(['vielendankfürdenraid', 'thanksfortheraid', 'willkommenaufstarbase47'])) {
                        return true;
                }
                return false;
@@ -188,22 +216,32 @@ class TokenizedMessage {
                if (is_null($this->classification)) {
                        if (empty($this->raw)) {
                                $this->classification = 'unclassified';
+                       } else if ($this->startsWith('!')) {
+                               $this->classification = 'cmd';
                        } else if (is_numeric($this->raw)) {
                                $this->classification = 'number';
-                       } else if ($this->hasTokenThatStartsOrEndsWith(['gg'])) {
+                       } else if ($this->hasTokenThatStartsOrEndsWith(['gg']) || $this->hasEmoteThatEndsWith(['gg'])) {
                                $this->classification = 'gg';
-                       } else if ($this->containsRaw(['glgl', 'glhf', 'hfgl'])) {
+                       } else if ($this->containsRaw(['glgl', 'glhf', 'goodluck', 'hfgl'])) {
                                $this->classification = 'gl';
-                       } else if ($this->containsRaw(['haha', 'hehe', 'hihi', 'kekw', 'lol', 'lul', 'xd']) || $this->hasTokenThatStartsWith(':d')) {
-                               $this->classification = 'lol';
-                       } else if ($this->startsWithRaw(['ahoi', 'hallo', 'hello', 'huhu']) || $this->hasEmoteThatEndsWith(['hello', 'hi', 'wave']) || $this->hasToken('hi')) {
+                       } else if ($this->startsWithRaw(['ahoi', 'hallo', 'hello', 'hey', 'huhu', 'moin']) || $this->hasEmoteThatEndsWith(['hello', 'heyguys', 'hi', 'wave']) || $this->hasToken(['hi', 'hey']) || $this->containsRaw(['gutenmorgen', 'gutenabend'])) {
                                $this->classification = 'hi';
-                       } else if ($this->containsRaw(['pog', 'wow'])) {
+                       } else if ($this->hasTokenThatStartsOrEndsWith(['pog', 'wow'])) {
                                $this->classification = 'pog';
                        } else if ($this->containsRaw(['hype'])) {
                                $this->classification = 'hype';
+                       } else if ($this->hasToken(['danke', 'thanks', 'thx', 'ty'])) {
+                               $this->classification = 'thx';
+                       } else if ($this->hasToken(['<3']) || $this->hasEmoteThatEndsWith(['herz', 'hug', 'love'])) {
+                               $this->classification = 'love';
+                       } else if ($this->hasTokenThatStartsWith(['wat', 'wtf']) || $this->hasEmoteThatStartsWith(['wat', 'wtf'])) {
+                               $this->classification = 'wtf';
+                       } else if ($this->endsWithEmoteless('?')) {
+                               $this->classification = 'question';
                        } else if ($this->startsOrEndsWithRaw(['o7']) || $this->hasEmoteThatContains('salut')) {
                                $this->classification = 'o7';
+                       } else if ($this->containsRaw(['haha', 'hehe', 'hihi', 'kekw', 'lol', 'lul']) || $this->hasTokenThatStartsWith([':d', 'xd'])) {
+                               $this->classification = 'lol';
                        } else {
                                $this->classification = 'unclassified';
                        }