kiri-core/kiri-engine/Error/ErrorHandler.php

<?php
/**
 * Created by PhpStorm.
 * User: whwyy
 * Date: 2018/4/26 0026
 * Time: 10:00
 */
declare(strict_types=1);

namespace Kiri\Error;

use Closure;
use Exception;
use Kiri;
use Kiri\Abstracts\Component;
use Kiri\Core\Json;
use Kiri\Annotation\Inject;
use Kiri\Events\EventDispatch;
use Kiri\Message\Events\OnAfterRequest;
use Kiri\Message\Handler\Formatter\IFormatter;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use Kiri\Server\Events\OnShutdown;
use ReflectionException;

/**
 * Class ErrorHandler
 *
 * @package Kiri\Base
 * @property-read $asError
 */
class ErrorHandler extends Component implements ErrorInterface
{

	/** @var ?IFormatter $message */
	private ?IFormatter $message = NULL;


	/**
	 * @var string
	 */
	public string $category = 'app';


	/**
	 * @var EventDispatch
	 */
	#[Inject(EventDispatch::class)]
	public EventDispatch $dispatch;


	/**
	 * @param array|Closure|null $callback
	 * @return void
	 */
	public function registerExceptionHandler(null|array|Closure $callback): void
	{
		if (empty($callback)) {
			$callback = [$this, 'exceptionHandler'];
		} else if (is_array($callback) && is_string($callback[0])) {
			$callback[0] = Kiri::getDi()->get($callback[0]);
		}
		set_exception_handler($callback);
	}


	/**
	 * @param array|Closure|null $callback
	 * @return void
	 */
	public function registerErrorHandler(null|array|Closure $callback): void
	{
		if (empty($callback)) {
			$callback = [$this, 'errorHandler'];
		} else if (is_array($callback) && is_string($callback[0])) {
			$callback[0] = Kiri::getDi()->get($callback[0]);
		}
		set_error_handler($callback);
	}


	/**
	 * @param array|Closure|null $callback
	 * @return void
	 */
	public function registerShutdownHandler(null|array|Closure $callback): void
	{
		if (empty($callback)) {
			$callback = [$this, 'shutdown'];
		} else if (is_array($callback) && is_string($callback[0])) {
			$callback[0] = Kiri::getDi()->get($callback[0]);
		}
		register_shutdown_function($callback);
	}


	/**
	 * @return void
	 * @throws ContainerExceptionInterface
	 * @throws NotFoundExceptionInterface
	 * @throws ReflectionException
	 * @throws Exception
	 */
	public function shutdown(): void
	{
		$lastError = error_get_last();
		if (empty($lastError) || $lastError['type'] !== E_ERROR) {
			return;
		}

		$this->category = 'shutdown';

		$messages = explode(PHP_EOL, $lastError['message']);

		$message = array_shift($messages);

		$this->dispatch->dispatch(new OnShutdown());

		$this->sendError($message, $lastError['file'], $lastError['line']);
	}


	/**
	 * @param \Throwable $exception
	 *
	 * @throws ContainerExceptionInterface
	 * @throws NotFoundExceptionInterface
	 * @throws ReflectionException
	 * @throws Exception
	 */
	public function exceptionHandler(\Throwable $exception)
	{
		$this->category = 'exception';

		$this->dispatch->dispatch(new OnAfterRequest());

		$this->sendError($exception->getMessage(), $exception->getFile(), $exception->getLine());
	}


	/**
	 * @throws \ErrorException
	 * @throws ContainerExceptionInterface
	 * @throws NotFoundExceptionInterface
	 * @throws ReflectionException
	 */
	public function errorHandler()
	{
		$error = func_get_args();

		$path = ['file' => $error[2], 'line' => $error[3]];

		if ($error[0] === 0) {
			$error[0] = 500;
		}

		$data = Json::to(500, $error[1], $path);

		$this->logger->error('On error handler', [$data]);

		$this->dispatch->dispatch(new OnAfterRequest());

		throw new \ErrorException($error[1], $error[0], 1, $error[2], $error[3]);
	}

	/**
	 * @param $message
	 * @param $file
	 * @param $line
	 * @param int $code
	 * @return false|string
	 * @throws Exception
	 */
	public function sendError($message, $file, $line, int $code = 500): bool|string
	{
		$path = ['file' => $file, 'line' => $line];

		$data = Json::to($code, $this->category . ': ' . $message, $path);

		file_put_contents('php://output', $data . PHP_EOL, FILE_APPEND);

		return $data;
	}

	/**
	 * @return mixed
	 */
	public function getErrorMessage(): mixed
	{
		$message = $this->message;
		$this->message = NULL;
		return $message->getData();
	}

	/**
	 * @return bool
	 */
	public function getAsError(): bool
	{
		return $this->message !== NULL;
	}

	/**
	 * @param $message
	 * @param string $category
	 *
	 * @throws Exception
	 */
	public function writer($message, string $category = 'app')
	{
		Kiri::getLogger()->debug($category, [$message]);
	}
}