xyz-rust 0.4.6

// 错误分类学是三个前端共享的单一错误模型：一个带 Kind 的错误同时驱动
// CLI 退出码、HTTP 状态码与 MCP JSON-RPC 错误码，传输实现永远不需要去
// 解释命令专属的错误字符串。
//
// 零第三方依赖：CodedError 手写 std::error::Error 实现（不引 thiserror）。

use std::fmt;

/// Kind 给错误分类。前端通过 http_status、exit_code、jsonrpc_code 把
/// Kind 翻译成各自渠道的错误表示。
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize)]
#[serde(rename_all = "snake_case")]
pub enum Kind {
    /// 调用方提供了畸形或非法的入参（缺必填、校验失败、枚举越界）。
    InvalidInput,
    /// 调用方必须先行认证。
    Unauthorized,
    /// 调用方已认证但缺少权限。
    Forbidden,
    /// 操作目标不存在。
    NotFound,
    /// 操作与现有状态冲突。
    Conflict,
    /// 操作被调用方取消。
    Canceled,
    /// 依赖的服务暂时不可用。
    Unavailable,
    /// 未分类失败的兜底分类。
    Internal,
}

impl Kind {
    pub fn as_str(&self) -> &'static str {
        match self {
            Kind::InvalidInput => "invalid_input",
            Kind::Unauthorized => "unauthorized",
            Kind::Forbidden => "forbidden",
            Kind::NotFound => "not_found",
            Kind::Conflict => "conflict",
            Kind::Canceled => "canceled",
            Kind::Unavailable => "unavailable",
            Kind::Internal => "internal",
        }
    }
}

impl fmt::Display for Kind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(self.as_str())
    }
}

/// Error 是库内统一的错误类型（CodedError 的 Rust 对应物）。
/// kind 缺省为 Internal；用户在自己的 handler 里也可以直接返回任何
/// `std::error::Error`（由 classify 兜底分类为 Internal）。
///
/// spec §8.5：可选富化层由 builder 链挂载——
/// `Error::new(NotFound, "user 7").with_code("USER_NOT_FOUND")
///  .with_detail("user_id", json!(7)).with_status(410)`。
#[derive(Debug)]
pub struct Error {
    kind: Kind,
    message: String,
    cause: Option<Box<dyn std::error::Error + Send + Sync + 'static>>,
    /// §8.5 富化层（code/detail/status）；Box 保持 Result 的小体积
    /// （clippy::result_large_err）。
    rich: Option<Box<Rich>>,
}

/// §8.5 的三层可选富化上下文。
#[derive(Debug, Clone, Default)]
pub struct Rich {
    /// 自由域标识（绝不参与传输映射）。
    pub code: Option<String>,
    /// 结构化上下文，原样进 §8.6 错误体。
    pub detail: Option<serde_json::Map<String, serde_json::Value>>,
    /// HTTP 状态覆盖（仅 HTTP 通道）。
    pub status: Option<u16>,
}

impl Error {
    fn base(
        kind: Kind,
        message: String,
        cause: Option<Box<dyn std::error::Error + Send + Sync + 'static>>,
    ) -> Self {
        Error {
            kind,
            message,
            cause,
            rich: None,
        }
    }

    /// 新建一条没有 cause 的分类错误。
    pub fn new(kind: Kind, msg: impl Into<String>) -> Self {
        Error::base(kind, msg.into(), None)
    }

    /// 用格式串新建一条分类错误。
    pub fn errorf(kind: Kind, format: std::fmt::Arguments<'_>) -> Self {
        Error::base(kind, fmt::format(format), None)
    }

    /// 把任意错误包成分类错误：消息取 cause 的消息。
    pub fn wrap(kind: Kind, cause: impl std::error::Error + Send + Sync + 'static) -> Self {
        Error::base(kind, String::new(), Some(Box::new(cause)))
    }

    /// 既有消息又有 cause 的分类错误。
    pub fn wrap_msg(
        kind: Kind,
        cause: impl std::error::Error + Send + Sync + 'static,
        msg: impl Into<String>,
    ) -> Self {
        Error::base(kind, msg.into(), Some(Box::new(cause)))
    }

    /// 把任意用户错误升级为库错误：只包一层（返回 Box 进 source 链），
    /// classify 会沿链找用户内层已有的分类。用户错误本身不丢。
    pub fn upgrade(cause: impl std::error::Error + Send + Sync + 'static) -> Self {
        Error::base(Kind::Internal, String::new(), Some(Box::new(cause)))
    }

    fn rich_mut(&mut self) -> &mut Rich {
        self.rich.get_or_insert_with(|| Box::new(Rich::default()))
    }

    /// §8.5 code：自由域标识，链式挂载（不改变 Kind 映射）。
    pub fn with_code(mut self, code: impl Into<String>) -> Self {
        self.rich_mut().code = Some(code.into());
        self
    }

    /// §8.5 detail：单条结构化上下文，链式挂载。
    pub fn with_detail(mut self, key: impl Into<String>, value: serde_json::Value) -> Self {
        self.rich_mut()
            .detail
            .get_or_insert_with(serde_json::Map::new)
            .insert(key.into(), value);
        self
    }

    /// §8.5 detail：批量合并结构化上下文，链式挂载。
    pub fn with_details(mut self, details: serde_json::Map<String, serde_json::Value>) -> Self {
        self.rich_mut()
            .detail
            .get_or_insert_with(serde_json::Map::new)
            .extend(details);
        self
    }

    /// §8.5 status：覆盖 Kind 推导的 HTTP 状态（仅 HTTP 通道生效）。
    pub fn with_status(mut self, status: u16) -> Self {
        self.rich_mut().status = Some(status);
        self
    }

    /// 从错误链继承富化层（code/detail/status 各取首个非空）——管线包装
    /// 壳用：handler 已是富化 errors::Error 时，包装不丢 §8.5 语义。
    pub fn inherit_rich(self, src: &(dyn std::error::Error + 'static)) -> Self {
        self.inherit_values(rich_fields(src))
    }

    /// inherit_rich 的取值版本（管线里 cause 已 move 时用）。
    pub(crate) fn inherit_values(self, (code, detail, status): RichValues) -> Self {
        if code.is_none() && detail.is_none() && status.is_none() {
            return self;
        }
        let mut me = self;
        let rich = me.rich_mut();
        if rich.code.is_none() {
            rich.code = code;
        }
        if rich.detail.is_none() {
            rich.detail = detail;
        }
        if rich.status.is_none() {
            rich.status = status;
        }
        me
    }

    pub fn kind(&self) -> Kind {
        self.kind
    }

    /// §8.5 HTTP 状态覆盖（None = 按 Kind 推导）。
    pub fn status_override(&self) -> Option<u16> {
        self.rich.as_ref().and_then(|r| r.status)
    }

    /// 解开一层到最内层已知 cause（Go 的 Cause）。
    pub fn cause(&self) -> Option<&(dyn std::error::Error + Send + Sync)> {
        self.cause.as_deref()
    }
}

impl fmt::Display for Error {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match (&self.message.is_empty(), &self.cause) {
            (false, Some(cause)) => write!(f, "{}: {}", self.message, cause),
            (false, None) => f.write_str(&self.message),
            (true, Some(cause)) => write!(f, "{}", cause),
            (true, None) => f.write_str(self.kind.as_str()),
        }
    }
}

impl std::error::Error for Error {
    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
        self.cause
            .as_ref()
            .map(|c| c.as_ref() as &(dyn std::error::Error + 'static))
    }
}

impl From<std::io::Error> for Error {
    fn from(e: std::io::Error) -> Self {
        // 保留底层 io::Error 进 cause 链：显示文本与历史形态完全一致
        //（"io: …"），同时让 EPIPE 之类的类型信息可被沿链检测。
        Error::wrap_msg(Kind::Internal, e, "io")
    }
}

/// 错误链上是否为管道早关（EPIPE）：CLI 出口据此静默终止（对齐 Go
/// 进程被 SIGPIPE 终止的 128+13=141 行为），而不是打印错误体。
pub fn is_broken_pipe(err: &(dyn std::error::Error + 'static)) -> bool {
    let mut cur: Option<&(dyn std::error::Error + 'static)> = Some(err);
    while let Some(e) = cur {
        if let Some(io) = e.downcast_ref::<std::io::Error>()
            && io.kind() == std::io::ErrorKind::BrokenPipe
        {
            return true;
        }
        cur = e.source();
    }
    false
}

/// 沿错误链找到第一条带分类的错误并返回其 Kind：
/// 未分类的非空错误兜底 Internal。入参需 'static 具体错误（或经
/// source() 拿到的 &dyn Error + 'static）。
pub fn classify(err: &(dyn std::error::Error + 'static)) -> Option<Kind> {
    let mut cur: Option<&(dyn std::error::Error + 'static)> = Some(err);
    while let Some(e) = cur {
        if let Some(ce) = e.downcast_ref::<Error>() {
            return Some(ce.kind);
        }
        // 用户错误链里可能嵌套我们自己的分类错误（例如 io::Error 包了 Error）。
        cur = e.source();
    }
    Some(Kind::Internal)
}

/// 把 Kind 翻译成自然的 HTTP 状态码。
pub fn http_status(kind: Kind) -> u16 {
    match kind {
        Kind::InvalidInput => 400,
        Kind::Unauthorized => 401,
        Kind::Forbidden => 403,
        Kind::NotFound => 404,
        Kind::Conflict => 409,
        Kind::Unavailable => 503,
        Kind::Canceled => 499, // 非标准但广为人知；前端可自行覆盖
        Kind::Internal => 500,
    }
}

/// 把 Kind 翻译成 CLI 进程退出码。
pub fn exit_code(kind: Kind) -> i32 {
    if kind == Kind::InvalidInput {
        2 // 与常规 flag 解析失败同码
    } else {
        1
    }
}

/// 把 Kind 翻译成 JSON-RPC 2.0 错误码。-32000..-32099 是 JSON-RPC 与 MCP
/// 保留给应用自定义服务端错误的区间。
pub fn jsonrpc_code(kind: Kind) -> i64 {
    match kind {
        Kind::InvalidInput => -32602, // Invalid params
        Kind::NotFound => -32001,
        Kind::Conflict => -32009,
        Kind::Unauthorized => -32010,
        Kind::Forbidden => -32011,
        Kind::Canceled => -32012,
        _ => -32603, // Internal error
    }
}

/// 命令 handler 的 canonical 返回类型。
pub type Result<T> = std::result::Result<T, Error>;

/// 沿错误链找第一个分类载体（Go errors.From 的对应物）。
pub fn from_chain<'a>(err: &'a (dyn std::error::Error + 'static)) -> Option<&'a Error> {
    let mut cur: Option<&(dyn std::error::Error + 'static)> = Some(err);
    while let Some(e) = cur {
        if let Some(coded) = e.downcast_ref::<Error>() {
            return Some(coded);
        }
        cur = e.source();
    }
    None
}

/// 沿链合并富化字段：各字段取首个非空（包装壳不遮蔽内层 handler 错误）。
pub(crate) type RichValues = (
    Option<String>,
    Option<serde_json::Map<String, serde_json::Value>>,
    Option<u16>,
);

pub(crate) fn rich_fields(err: &(dyn std::error::Error + 'static)) -> RichValues {
    let mut cur: Option<&(dyn std::error::Error + 'static)> = Some(err);
    let (mut code, mut detail, mut status) = (None, None, None);
    while let Some(e) = cur {
        if let Some(ce) = e.downcast_ref::<Error>()
            && let Some(r) = &ce.rich
        {
            if code.is_none() {
                code = r.code.clone();
            }
            if detail.is_none() {
                detail = r.detail.clone();
            }
            if status.is_none() {
                status = r.status;
            }
        }
        cur = e.source();
    }
    (code, detail, status)
}

/// 最内层 cause 的消息（spec §8.4 的 most-specific 语义；无 cause 即自身）。
fn deepest_message(err: &(dyn std::error::Error + 'static)) -> String {
    let mut cur: &(dyn std::error::Error + 'static) = err;
    while let Some(src) = cur.source() {
        cur = src;
    }
    cur.to_string()
}

/// §8.6 共享错误体：HTTP 响应体、CLI 机器格式 stderr 与 MCP `_meta.xyz.error`
/// 共用的单一形状。平铺 `error` 键与旧契约字节兼容，`code`/`detail` 纯增量。
#[derive(Debug, Clone, serde::Serialize)]
pub struct Body {
    pub error: String,
    pub kind: Kind,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub code: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub detail: Option<serde_json::Map<String, serde_json::Value>>,
}

impl Default for Body {
    fn default() -> Self {
        Body {
            error: String::new(),
            kind: Kind::Internal,
            code: None,
            detail: None,
        }
    }
}

/// 把任意错误投射为共享错误体（spec §8.6）。消息取最内层 cause，
/// 富化层（code/detail）从链上第一个分类载体上取。
pub fn error_body(err: &(dyn std::error::Error + 'static)) -> Body {
    let mut b = Body {
        error: deepest_message(err),
        kind: classify(err).unwrap_or(Kind::Internal),
        code: None,
        detail: None,
    };
    let (code, detail, _) = rich_fields(err);
    b.code = code;
    b.detail = detail;
    b
}

/// HTTP 状态：显式 §8.5 status 覆盖优先，否则按 Kind 推导。
pub fn error_status(err: &(dyn std::error::Error + 'static)) -> u16 {
    let (_, _, status) = rich_fields(err);
    if let Some(s) = status {
        return s;
    }
    http_status(classify(err).unwrap_or(Kind::Internal))
}

/// 便捷构造函数（对齐 Go 的 errs.New/errs.Errorf/errs.Wrap）。
pub fn new(kind: Kind, msg: impl Into<String>) -> Error {
    Error::new(kind, msg)
}

/// 返回带格式化消息的分类错误：
/// `errs::errorf!(errs::Kind::NotFound, "user {} not found", name)`
#[macro_export]
macro_rules! errorf {
    ($kind:expr, $($arg:tt)*) => {
        $crate::errors::Error::errorf($kind, format_args!($($arg)*))
    };
}

pub use errorf;
#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn mappings() {
        assert_eq!(http_status(Kind::InvalidInput), 400);
        assert_eq!(http_status(Kind::Unauthorized), 401);
        assert_eq!(http_status(Kind::Forbidden), 403);
        assert_eq!(http_status(Kind::NotFound), 404);
        assert_eq!(http_status(Kind::Conflict), 409);
        assert_eq!(http_status(Kind::Unavailable), 503);
        assert_eq!(http_status(Kind::Canceled), 499);
        assert_eq!(http_status(Kind::Internal), 500);

        assert_eq!(exit_code(Kind::InvalidInput), 2);
        assert_eq!(exit_code(Kind::NotFound), 1);

        assert_eq!(jsonrpc_code(Kind::InvalidInput), -32602);
        assert_eq!(jsonrpc_code(Kind::NotFound), -32001);
        assert_eq!(jsonrpc_code(Kind::Conflict), -32009);
        assert_eq!(jsonrpc_code(Kind::Unauthorized), -32010);
        assert_eq!(jsonrpc_code(Kind::Forbidden), -32011);
        assert_eq!(jsonrpc_code(Kind::Canceled), -32012);
        assert_eq!(jsonrpc_code(Kind::Internal), -32603);
    }

    #[test]
    fn classify_chain() {
        let e = Error::wrap_msg(Kind::NotFound, Error::new(Kind::Conflict, "inner"), "outer");
        assert_eq!(classify(&e), Some(Kind::NotFound));
        // 内层 cause 类型也保留在链上（Cause 语义；Error::source() 在
        // std::error::Error 上，给出 &dyn Error + 'static）。
        let cause = std::error::Error::source(&e).unwrap();
        assert_eq!(classify(cause), Some(Kind::Conflict));
    }

    #[test]
    fn error_display_forms() {
        assert_eq!(Error::new(Kind::NotFound, "x").to_string(), "x");
        let wrapped = Error::wrap(Kind::InvalidInput, std::io::Error::other("ioerr"));
        assert_eq!(wrapped.to_string(), "ioerr");
        let both = Error::wrap_msg(Kind::Internal, std::io::Error::other("ioerr"), "boom");
        assert_eq!(both.to_string(), "boom: ioerr");
    }
}