vendor/ashiina/socket.io-emitter/src/Emitter.php line 48

Open in your IDE?
  1. <?php
  3. namespace SocketIO;
  5. define('EVENT', 2);
  6. define('BINARY_EVENT', 5);
  8. if (!function_exists('msgpack_pack')) {
  9.   require(__DIR__ . '/msgpack_pack.php');
  10. }
  12. class Emitter {
  13.   private $uid = 'emitter';
  15.   public function __construct($redis = FALSE, $opts = array()) {
  16.     if (is_array($redis)) {
  17.       $opts = $redis;
  18.       $redis = FALSE;
  19.     }
  21.     // Apply default arguments
  22.     $opts = array_merge(array('host' => 'localhost', 'port' => 6379), $opts);
  24.     if (!$redis) {
  25.       // Default to phpredis
  26.       if (extension_loaded('redis')) {
  27.         if (!isset($opts['socket']) && !isset($opts['host'])) throw new \Exception('Host should be provided when not providing a redis instance');
  28.         if (!isset($opts['socket']) && !isset($opts['port'])) throw new \Exception('Port should be provided when not providing a redis instance');
  30.         $redis = new \Redis();
  31.         if (isset($opts['socket'])) {
  32.           $redis->connect($opts['socket']);
  33.         } else {
  34.           $redis->connect($opts['host'], $opts['port']);
  35.         }
  36.       } else {
  37.         $redis = new \TinyRedisClient($opts['host'].':'.$opts['port']);
  38.       }
  39.     }
  41.     if (!is_callable(array($redis, 'publish'))) {
  42.       throw new \Exception('The Redis client provided is invalid. The client needs to implement the publish method. Try using the default client.');
  43.     }
  45.     $this->redis = $redis;
  46.     $this->prefix = isset($opts['key']) ? $opts['key'] : 'socket.io';
  48.     $this->_rooms = array();
  49.     $this->_flags = array();
  50.   }
  52.   /*
  53.    * Flags
  54.    */
  56.   public function __get($flag) {
  57.     $this->_flags[$flag] = TRUE;
  58.     return $this;
  59.   }
  61.   private function readFlag($flag) {
  62.     return isset($this->_flags[$flag]) ? $this->_flags[$flag] : false;
  63.   }
  65.   /*
  66.    * Broadcasting
  67.    */
  69.   public function in($room) {
  70.     if (!in_array($room, $this->_rooms)) {
  71.       $this->_rooms[] = $room;
  72.     }
  74.     return $this;
  75.   }
  77.   // Alias for in
  78.   public function to($room) {
  79.     return $this->in($room);
  80.   }
  82.   /*
  83.    * Namespace
  84.    */
  86.   public function of($nsp) {
  87.     $this->_flags['nsp'] = $nsp;
  88.     return $this;
  89.   }
  91.   /*
  92.    * Emitting
  93.    */
  95.   public function emit() {
  96.     $args = func_get_args();
  97.     $packet = array();
  99.     $packet['type'] = EVENT;
  100.     // handle binary wrapper args
  101.     for ($i = 0; $i < count($args); $i++) {
  102.       $arg = $args[$i];
  103.       if ($arg instanceof Binary) {
  104.         $args[$i] = strval($arg);
  105.         $this->binary;
  106.       }
  107.     }
  109.     if ($this->readFlag('binary')) $packet['type'] = BINARY_EVENT;
  111.     $packet['data'] = $args;
  113.     // set namespace
  114.     if (isset($this->_flags['nsp'])) {
  115.       $packet['nsp'] = $this->_flags['nsp'];
  116.       unset($this->_flags['nsp']);
  117.     } else {
  118.       $packet['nsp'] = '/';
  119.     }
  121.     $opts = array(
  122.       'rooms' => $this->_rooms,
  123.       'flags' => $this->_flags
  124.     );
  125.     $chn = $this->prefix . '#' . $packet['nsp'] . '#';
  126.     $packed = msgpack_pack(array($this->uid,$packet,$opts));
  128.     // hack buffer extensions for msgpack with binary
  129.     if ($packet['type'] == BINARY_EVENT) {
  130.       $packed = str_replace(pack('c', 0xda), pack('c', 0xd8), $packed);
  131.       $packed = str_replace(pack('c', 0xdb), pack('c', 0xd9), $packed);
  132.     }
  134.     // publish
  135.     if (is_array($this->_rooms) && count($this->_rooms) > 0) {
  136.         foreach ($this->_rooms as $room) {
  137.             $chnRoom = $chn . $room . '#';
  138.             $this->redis->publish($chnRoom, $packed);
  139.         }
  140.     } else {
  141.         $this->redis->publish($chn, $packed);
  142.     }
  144.     // reset state
  145.     $this->_rooms = array();
  146.     $this->_flags = array();
  148.     return $this;
  149.   }
  150. }