app/TwitchBot/IRCMessage.php

   1 <?php
   2
   3 namespace App\TwitchBot;
   4
   5 use App\Models\ChatLog;
   6
   7 class IRCMessage {
   8
   9         public $command = null;
  10         public $params = [];
  11         public $nick = null;
  12         public $user = null;
  13         public $host = null;
  14         public $servername = null;
  15         public $tags = [];
  16
  17         public static function fromString($message) {
  18                 $msg = new static();
  19
  20                 $raw = rtrim($message);
  21                 $processed = $raw;
  22
  23                 if ($processed[0] == '@') {
  24                         $tags = explode(' ', $processed, 2);
  25                         $processed = $tags[1];
  26                         $tags = explode(';', ltrim(trim($tags[0]), '@'));
  27                         foreach ($tags as $tag) {
  28                                 $parts = explode('=', $tag, 2);
  29                                 $msg->tags[$parts[0]] = $parts[1] == '' ? null : $parts[1];
  30                         }
  31                 }
  32
  33                 if ($processed[0] == ':') {
  34                         $has_user = false;
  35                         $has_host = false;
  36                         $prefix = explode(' ', $processed, 2);
  37                         $processed = $prefix[1];
  38                         $prefix = $prefix[0][0] === ':' ? substr($prefix[0], 1) : $prefix[0];
  39                         if (strpos($prefix, '!') !== false) {
  40                                 $has_user = true;
  41                         }
  42                         if (strpos($prefix, '@') !== false) {
  43                                 $has_host = true;
  44                         }
  45                         if ($has_user && $has_host) {
  46                                 $prefix_part = explode('!', $prefix, 2);
  47                                 $msg->nick = $prefix_part[0];
  48                                 $prefix_part = explode('@', $prefix_part[1], 2);
  49                                 $msg->user = $prefix_part[0];
  50                                 $msg->host = $prefix_part[1];
  51                         } else if ($has_user) {
  52                                 $prefix_part = explode('!', $prefix, 2);
  53                                 $msg->nick = $prefix_part[0];
  54                                 $msg->user = $prefix_part[1];
  55                         } else if ($has_host) {
  56                                 $prefix_part = explode('@', $prefix, 2);
  57                                 $msg->nick = $prefix_part[0];
  58                                 $msg->host = $prefix_part[1];
  59                         } else {
  60                                 $msg->servername = $prefix;
  61                         }
  62                 }
  63
  64                 $command = explode(' ', $processed, 2);
  65                 $msg->command = $command[0];
  66                 $processed = $command[1] ?? '';
  67
  68                 while (strlen($processed)) {
  69                         if ($processed[0] == ':') {
  70                                 $msg->params[] = substr($processed, 1);
  71                                 break;
  72                         }
  73                         $e = explode(' ', $processed, 2);
  74                         $msg->params[] = $e[0];
  75                         if (isset($e[1])) {
  76                                 $processed = $e[1];
  77                         } else {
  78                                 break;
  79                         }
  80                 }
  81
  82                 return $msg;
  83         }
  84
  85         public function encode() {
  86                 $str = '';
  87                 if (!empty($this->tags)) {
  88                         $str .= '@';
  89                         $first = true;
  90                         foreach ($this->tags as $name => $value) {
  91                                 if ($first) {
  92                                         $first = false;
  93                                 } else {
  94                                         $str .= ';';
  95                                 }
  96                                 $str .= $name.'=';
  97                                 if (!empty($value)) {
  98                                         $str .= $value;
  99                                 }
 100                         }
 101                         $str .= ' ';
 102                 }
 103
 104                 if (!empty($this->servername)) {
 105                         $str .= ':'.$this->servername.' ';
 106                 } else if (!empty($this->nick)) {
 107                         $str .= ':'.$this->nick;
 108                         if (!empty($this->user)) {
 109                                 $str .= '!'.$this->user;
 110                         }
 111                         if (!empty($this->host)) {
 112                                 $str .= '@'.$this->host;
 113                         }
 114                         $str .= ' ';
 115                 }
 116
 117                 $str .= $this->command;
 118
 119                 if (!empty($this->params)) {
 120                         $n = count($this->params) - 1;
 121                         for ($i = 0; $i < $n; ++$i) {
 122                                 $str .= ' '.$this->params[$i];
 123                         }
 124                         $str .= ' :'.$this->params[$n];
 125                 }
 126
 127                 return $str;
 128         }
 129
 130         public function log() {
 131                 ChatLog::create([
 132                         'command' => $this->command,
 133                         'nick' => $this->nick,
 134                         'params' => $this->params,
 135                         'tags' => $this->tags,
 136                 ]);
 137         }
 138
 139         public function tokenize() {
 140                 return TokenizedMessage::fromIRC($this);
 141         }
 142
 143         public static function capReq($cap) {
 144                 $msg = new IRCMessage();
 145                 $msg->command = 'CAP REQ';
 146                 $msg->params[] = $cap;
 147                 return $msg;
 148         }
 149
 150         public static function join($channels) {
 151                 $msg = new IRCMessage();
 152                 $msg->command = 'JOIN';
 153                 $msg->params[] = implode(',', $channels);
 154                 return $msg;
 155         }
 156
 157         public static function part($channels) {
 158                 $msg = new IRCMessage();
 159                 $msg->command = 'PART';
 160                 $msg->params[] = implode(',', $channels);
 161                 return $msg;
 162         }
 163
 164         public static function ping($token = 'localhorsttv') {
 165                 $msg = new IRCMessage();
 166                 $msg->command = 'PING';
 167                 $msg->params[] = $token;
 168                 return $msg;
 169         }
 170
 171         public static function privmsg($target, $message) {
 172                 $msg = new IRCMessage();
 173                 $msg->command = 'PRIVMSG';
 174                 $msg->params[] = $target;
 175                 $msg->params[] = $message;
 176                 return $msg;
 177         }
 178
 179         public function getPrivMsgTarget() {
 180                 if (!empty($this->params)) {
 181                         return $this->params[0];
 182                 }
 183                 return '';
 184         }
 185
 186         public function getText() {
 187                 if (!empty($this->params)) {
 188                         return $this->params[count($this->params) - 1];
 189                 }
 190                 return '';
 191         }
 192
 193         public function isNotice() {
 194                 return $this->command == 'NOTICE';
 195         }
 196
 197         public function isPing() {
 198                 return $this->command == 'PING';
 199         }
 200
 201         public function isPong() {
 202                 return $this->command == 'PONG';
 203         }
 204
 205         public function isPrivMsg() {
 206                 return $this->command == 'PRIVMSG';
 207         }
 208
 209         public function isWhisper() {
 210                 return $this->command == 'WHISPER';
 211         }
 212
 213         public function isOwner() {
 214                 return substr($this->getPrivMsgTarget(), 1) == $this->nick;
 215         }
 216
 217         public function isMod() {
 218                 return $this->isOwner() || (isset($this->tags['mod']) && $this->tags['mod'] == '1');
 219         }
 220
 221         public function makePong() {
 222                 $msg = new IRCMessage();
 223                 $msg->command = 'PONG';
 224                 $msg->params = array_values($this->params);
 225                 return $msg;
 226         }
 227
 228 }
 229
 230 ?>