from apiflask import APIFlask, HTTPError
from apiflask.exceptions import _ValidationError
from iti.applications.common.utils import fail
from iti.applications.common.exceptions.biz_exp import BizException
from iti.applications.common.exceptions.permission import PermissionDeniedException


def init_error_handler(app: APIFlask):
    """
    全局异常处理
    """

    @app.errorhandler(Exception)
    def handle_exception(error):
        """
        未处理的异常处理
        """
        try:
            # 安全地获取错误信息
            error_msg = str(error)
            error_type = type(error).__name__
            app.logger.error(
                f"服务器错误 [{error_type}]: {error_msg}",
                exc_info=True  # 使用 exc_info 代替 stack_info，更安全
            )
        except Exception as log_error:
            # 如果日志记录失败，使用最基本的方式记录
            try:
                app.logger.error(f"服务器错误（日志记录失败）: {type(error).__name__}")
            except:
                pass  # 完全失败则放弃日志记录
        
        # 安全地返回错误信息
        try:
            error_data = str(error)
        except:
            error_data = type(error).__name__
        
        return fail(message="服务器错误", code=500, data=error_data)

    @app.errorhandler(400)
    def handle_400(error):
        """
        参数错误
        """
        try:
            app.logger.error(f"参数错误: {error}", exc_info=True)
        except:
            app.logger.error("参数错误（日志记录失败）")
        
        return fail(
            message=error.data.message
            if error.data and "message" in error.data
            else "参数错误",
            code=400,
            data=str(error),
        )

    @app.errorhandler(BizException)
    def handle_biz_exception(error: BizException):
        """
        业务异常处理
        """
        try:
            app.logger.error(f"业务异常: {error}")
        except:
            app.logger.error("业务异常（日志记录失败）")
        return fail(error.message, code=error.code, data=error.data)

    @app.errorhandler(PermissionDeniedException)
    def handle_permission_denied_exception(error: PermissionDeniedException):
        """
        权限不足异常处理
        """
        try:
            app.logger.error(f"权限不足: {error}")
        except:
            app.logger.error("权限不足（日志记录失败）")
        return fail(message=error.message, code=error.code)

    @app.error_processor
    def handler_http_error(error: HTTPError):
        """
        http异常处理
        """
        try:
            if isinstance(error, _ValidationError):
                app.logger.error(f"参数验证错误: {error.detail}")
                error.message = "参数验证错误"
            else:
                app.logger.error(
                    f"HTTP异常: {error.message} {error.status_code} {error.detail} {error.headers} {error.extra_data}"
                )
        except:
            app.logger.error("HTTP异常（日志记录失败）")
        
        return (
            fail(message=error.message, code=error.status_code, data=error.detail),
            200,
            error.headers,
        )