<?php

namespace App\TwitchBot;

use App\Models\Channel;
use App\Models\TwitchToken;
use Monolog\Handler\StreamHandler;
use Monolog\Logger;
use Ratchet\Client\Connector;
use Ratchet\Client\WebSocket;
use Ratchet\RFC6455\Messaging\Message;
use React\EventLoop\Loop;

class TwitchBot {

	public function __construct() {
		$this->logger = new Logger('TwitchBot');
		$this->logger->pushHandler(new StreamHandler('php://stdout', Logger::INFO));

		$this->token = TwitchToken::firstWhere('nick', 'localhorsttv');
		if (!$this->token) {
			throw new \Exception('unable to find access token');
		}

		$this->connector = new Connector();
		$this->connect();
	}

	public function getLogger() {
		return $this->logger;
	}

	public function getLoop() {
		return Loop::get();
	}

	public function run() {
		$this->shutting_down = false;
		$this->getLoop()->run();
	}

	public function stop() {
		$this->logger->info('shutting down');
		$this->shutting_down = true;
		$this->disconnect();
		$this->getLoop()->stop();
	}


	public function connect() {
		($this->connector)('wss://irc-ws.chat.twitch.tv:443')->done(
			[$this, 'handleWsConnect'],
			[$this, 'handleWsConnectError'],
		);
	}

	public function disconnect() {
		$this->ws->close();
	}

	public function handleWsConnect(WebSocket $ws) {
		$this->logger->info('websocket connection established');
		$this->ws = $ws;
		$ws->on('message', [$this, 'handleWsMessage']);
		$ws->on('close', [$this, 'handleWsClose']);
		$ws->on('error', [$this, 'handleWsError']);
		$ws->send('CAP REQ :twitch.tv/tags twitch.tv/commands');
		$this->login();
	}

	public function handleWsConnectError(WebSocket $ws) {
		$this->logger->error('failed to establish websocket connection');
	}

	public function handleWsMessage(Message $message, WebSocket $ws) {
		$irc_messages = explode("\r\n", rtrim($message->getPayload(), "\r\n"));
		foreach ($irc_messages as $irc_message) {
			$this->logger->debug('received IRC message '.$irc_message);
			$this->handleIRCMessage(IRCMessage::fromString($irc_message));
		}
	}

	public function handleWsClose(int $op, string $reason) {
		$this->logger->info('websocket connection closed: '.$reason.' ['.$op.']');
		if (!$this->shutting_down) {
			$this->logger->info('reconnecting in 10 seconds');
			Loop::addTimer(10, [$this, 'connect']);
		}
	}

	public function handleWsError(\Exception $e, WebSocket $ws) {
		$this->logger->error('websocket error '.$e->getMessage());
	}


	public function handleIRCMessage(IRCMessage $msg) {
		if ($msg->isPrivMsg()) {
			$this->handlePrivMsg($msg);
			return;
		}
		if ($msg->isPing()) {
			$this->sendIRCMessage($msg->makePong());
			return;
		}
		if ($msg->isNotice() && $msg->getText() == 'Login authentication failed') {
			$this->logger->notice('login failed, refreshing access token');
			$this->token->refresh();
			$this->login();
			return;
		}
		if ($msg->command == '001') {
			// successful login
			$this->joinChannels();
			return;
		}
	}

	public function handlePrivMsg(IRCMessage $msg) {
		$target = $msg->getPrivMsgTarget();
		if ($target[0] != '#') return;
		$text = $msg->getText();
		if ($text[0] != '!') return;
		$channel = Channel::firstWhere('twitch_chat', '=', $target);
		if (!$channel) return;
		$this->handleChatCommand($channel, $msg);
	}

	public function handleChatCommand(Channel $channel, IRCMessage $msg) {
		$cmd = explode(' ', ltrim($msg->getText(), '!'), 2);
		if (!isset($channel->chat_commands[$cmd[0]])) return;
		$config = $channel->chat_commands[$cmd[0]];
		$this->logger->info('got command '.$cmd[0].' on channel '.$channel->title);
		try {
			$command = ChatCommand::create($this, $channel, $config);
			$command->execute($cmd[1] ?? '');
		} catch (\Exception $e) {
			$this->logger->warning('error executing command '.$cmd[0].' on channel '.$channel->title.': '.$e->getMessage());
		}
	}

	public function login() {
		$this->ws->send('PASS oauth:'.$this->token->access);
		$this->ws->send('NICK localhorsttv');
	}

	public function joinChannels() {
		$this->logger->info('joining channels');
		$channels = Channel::where('twitch_chat', '!=', '')->get();
		$names = [];
		foreach ($channels as $channel) {
			$names[] = $channel->twitch_chat;
		}
		$chunks = array_chunk($names, 10);
		foreach ($chunks as $chunk) {
			$this->sendIRCMessage(IRCMessage::join($chunk));
		}
	}

	public function sendIRCMessage(IRCMessage $msg) {
		$irc_message = $msg->encode();
		$this->logger->debug('sending IRC message '.$irc_message);
		$this->ws->send($irc_message);
	}


	private $logger;

	private $token;

	private $connector;
	private $ws;
	private $shutting_down = false;

}

?>