<?php


namespace HttpServer;


use Exception;
use Snowflake\Abstracts\Input;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;
use Swoole\WebSocket\Server;

/**
 * Class Action
 * @package HttpServer
 */
trait Action
{

	/**
	 * @param \HttpServer\Server $socket
	 * @return mixed
	 * @throws Exception
	 */
	public function restart($socket)
	{
		$this->_shutdown($socket);

		return $this->start();
	}


	/**
	 * @param \HttpServer\Server $socket
	 * @throws Exception
	 */
	public function stop($socket)
	{
		$this->_shutdown($socket);
	}

	/**
	 * @param $server
	 * @return void
	 * @throws Exception
	 */
	private function _shutdown($server)
	{
		$socket = storage('socket.sock');
		if (!file_exists($socket)) {
			$this->close($server);
		} else {
			$pathId = file_get_contents($socket);
			@unlink($socket);

			if (empty($pathId)) {
				$this->close($server);
			} else {
				exec("ps -ef $pathId | grep $pathId", $output);
				if (!empty($output)) {
					exec("kill -TERM $pathId");
				}
				$this->close($server);
			}
		}
		Snowflake::clearWorkerId();
	}


	/**
	 * @param \HttpServer\Server $server
	 * @return void
	 * @throws Exception
	 */
	private function close(\HttpServer\Server $server)
	{
		echo 'waite.';
		while ($server->isRunner()) {
			echo '.';
			$pods = glob(storage(null, 'worker') . '/*');
			if (count($pods) < 1) {
				break;
			}
			foreach ($pods as $pid) {
				if (!file_exists($pid)) {
					continue;
				}
				$content = file_get_contents($pid);
				exec("ps -ax | awk '{ print $1 }' | grep -e '^{$content}$'", $output);
				if (count($output) > 0) {
					$this->closeByPid($content);
				} else {
					file_exists($pid) && @unlink($pid);
				}
			}
			usleep(100);
		}
		echo PHP_EOL;
	}


	/**
	 * @param $port
	 * @return bool|array
	 */
	private function isUse($port)
	{
		if (empty($port)) {
			return false;
		}
		if (Snowflake::isLinux()) {
			exec('netstat -tunlp | grep ' . $port, $output);
		} else {
			exec('lsof -i :' . $port . ' | grep -i "LISTEN"', $output);
		}
		if (empty($output)) {
			return false;
		}
		return $output;
	}

	/**
	 * @param $pid
	 */
	private function closeByPid($pid)
	{
		shell_exec('kill -TERM ' . $pid);
	}


}