<?php

declare(strict_types=1);
/**
 * This file is part of Hyperf.
 *
 * @link     https://www.hyperf.io
 * @document https://hyperf.wiki
 * @contact  group@hyperf.io
 * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
 */

namespace App\Exception\Handler;

use App\Exception\ApiException;
use Hyperf\Contract\StdoutLoggerInterface;
use Hyperf\ExceptionHandler\ExceptionHandler;
use Hyperf\HttpMessage\Stream\SwooleStream;
use Hyperf\RateLimit\Exception\RateLimitException;
use Hyperf\Validation\ValidationException;
use Psr\Http\Message\MessageInterface;
use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\StreamInterface;
use Throwable;
use function Hyperf\Support\env;

/**
 * Author: ykxiao
 * Date: 2025/6/3
 * Time: 下午8:57
 * Description: 默认异常处理
 *
 * (c) ykxiao <yk_9001@hotmail.com>
 *
 * This source file is subject to the MIT license that is bundled
 * with this source code in the file LICENSE.
 */
class AppExceptionHandler extends ExceptionHandler
{
    public function __construct(protected StdoutLoggerInterface $logger)
    {
    }

    public function handle(Throwable $throwable, ResponseInterface $response): MessageInterface|ResponseInterface
    {
        $this->logException($throwable);

        if ($throwable instanceof RateLimitException) {
            return $this->handleRateLimitException($response);
        }

        return $this->handleOtherExceptions($throwable, $response);
    }

    public function isValid(Throwable $throwable): bool
    {
        return ! $throwable instanceof ApiException;
    }

    protected function createResponseBody($message, $code): StreamInterface
    {
        $data = [
            'code' => $code,
            'message' => $message,
        ];
        return new SwooleStream(json_encode($data, JSON_UNESCAPED_UNICODE));
    }

    protected function logException(Throwable $throwable): void
    {
        $this->logger->error(sprintf('%s[%s] in %s', $throwable->getMessage(), $throwable->getLine(), $throwable->getFile()));
        $this->logger->error($throwable->getTraceAsString());
    }

    protected function handleRateLimitException(ResponseInterface $response): ResponseInterface
    {
        $message = '触发请求频率限流规则';
        return $response->withStatus(429)
            ->withHeader('Content-Type', 'application/json')
            ->withBody($this->createResponseBody($message, 429));
    }

    protected function handleOtherExceptions(Throwable $throwable, ResponseInterface $response): ResponseInterface
    {
        $message = $throwable->getMessage();

        if (env('APP_ENV') === 'production' && ! $throwable instanceof ValidationException) {
            return $response->withStatus(200)
                ->withHeader('Content-Type', 'application/json')
                ->withBody($this->createResponseBody($message, 0));
        }

        return $response->withHeader('Server', 'mes-auto')
            ->withStatus(200)
            ->withBody($this->createResponseBody($message, 0));
    }
}