kiri-http-server/Abstracts/WebSocketServer.php

<?php

namespace Kiri\Server\Abstracts;

use Exception;
use Kiri\Abstracts\Component;
use Kiri\Router\DataGrip;
use Kiri\Router\Handler;
use Kiri\Router\RouterCollector;
use Kiri\Server\Contract\OnCloseInterface;
use Kiri\Server\Contract\OnDisconnectInterface;
use Kiri\Server\Contract\OnHandshakeInterface;
use Kiri\Server\Contract\OnMessageInterface;
use ReflectionException;
use Swoole\Http\Request;
use Swoole\Http\Response;
use Swoole\Server;
use Swoole\WebSocket\Frame;


class WebSocketServer extends Component implements OnHandshakeInterface, OnMessageInterface, OnDisconnectInterface, OnCloseInterface
{


	public string $host = '127.0.0.1';


	public int $port = 6086;


	public array $events = [];

	public array $settings = [];

	public int $socket_type = SWOOLE_SOCK_TCP6;


	public Handler $handler;


	/**
	 * @var RouterCollector
	 */
	public RouterCollector $collector;


	/**
	 * @return void
	 * @throws ReflectionException
	 */
	public function init(): void
	{
		/** @var Server $server */
		$application = \Kiri::service();
		if (!$application->has('server')) {
			$server = new \Swoole\WebSocket\Server($this->host, $this->port, SWOOLE_PROCESS);
			$application->set('server', $server);
		} else {
			$server = $application->get('server');
		}

		$socket = $server->addlistener($this->host, $this->port, $this->socket_type);
		$socket->set($this->settings);
		$socket->on('handshake', [$this, 'onHandshake']);
		$socket->on('message', [$this, 'onMessage']);
		$socket->on('disconnect', [$this, 'onDisconnect']);

		$this->collector = \Kiri::getDi()->get(DataGrip::class)->get('wss');

		$this->handler = $this->collector->query('/', 'GET');
	}


	/**
	 * @param Request $request
	 * @param Response $response
	 * @return void
	 * @throws Exception
	 */
	public function onHandshake(Request $request, Response $response): void
	{
		// TODO: Implement onHandshake() method.
		$secWebSocketKey = $request->header['sec-websocket-key'];
		$patten = '#^[+/0-9A-Za-z]{21}[AQgw]==$#';
		if (0 === preg_match($patten, $secWebSocketKey) || 16 !== strlen(base64_decode($secWebSocketKey))) {
			throw new Exception('protocol error.', 500);
		}
		$key = base64_encode(sha1($request->header['sec-websocket-key'] . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11', TRUE));
		$headers = [
			'Upgrade'               => 'websocket',
			'Connection'            => 'Upgrade',
			'Sec-websocket-Accept'  => $key,
			'Sec-websocket-Version' => '13',
		];
		if (isset($request->header['sec-websocket-protocol'])) {
			$explode = explode(',', $request->header['sec-websocket-protocol']);
			$headers['Sec-websocket-Protocol'] = $explode[0];
		}
		foreach ($headers as $key => $val) {
			$response->setHeader($key, $val);
		}
		if ($this->handler->implement(OnHandshakeInterface::class)) {
			$handler = $this->handler->getClass();

			call_user_func([$handler, 'onHandshake'], $request, $response);
		} else {
			$response->setStatusCode(101);
			$response->end();
		}
	}


	/**
	 * @param Server $server
	 * @param Frame $frame
	 * @return void
	 */
	public function onMessage(Server $server, Frame $frame): void
	{
		// TODO: Implement onMessage() method.
		if (!$this->handler->implement(OnMessageInterface::class)) {
			return;
		}

		$handler = $this->handler->getClass();

		call_user_func([$handler, 'onMessage'], $server, $frame);
	}


	/**
	 * @param \Swoole\WebSocket\Server $server
	 * @param int $fd
	 * @return void
	 */
	public function onDisconnect(\Swoole\WebSocket\Server $server, int $fd): void
	{
		// TODO: Implement onDisconnect() method.
		if (!$this->handler->implement(OnDisconnectInterface::class)) {
			return;
		}

		$handler = $this->handler->getClass();

		call_user_func([$handler, 'onDisconnect'], $server, $fd);
	}


	/**
	 * @param Server $server
	 * @param int $fd
	 * @return void
	 */
	public function onClose(Server $server, int $fd): void
	{
		// TODO: Implement onDisconnect() method.
		if (!$this->handler->implement(OnCloseInterface::class)) {
			return;
		}

		$handler = $this->handler->getClass();

		call_user_func([$handler, 'onClose'], $server, $fd);
	}

}