sockudo_core/

error.rs

use thiserror::Error;

#[derive(Error, Debug)]
pub enum Error {
    // 4000-4099: Don't reconnect errors
    #[error("Application only accepts SSL connections, reconnect using wss://")]
    SSLRequired,

    #[error("Application does not exist")]
    ApplicationNotFound,

    #[error("Application disabled")]
    ApplicationDisabled,

    #[error("Application is over adapter quota")]
    OverConnectionQuota,

    #[error("Path not found")]
    PathNotFound,

    #[error("Invalid version string format")]
    InvalidVersionFormat,

    #[error("Unsupported protocol version: {0}")]
    UnsupportedProtocolVersion(String),

    #[error("No protocol version supplied")]
    NoProtocolVersion,

    #[error("Connection is unauthorized")]
    Unauthorized,

    #[error("Origin not allowed")]
    OriginNotAllowed,

    // 4100-4199: Reconnect with backoff errors
    #[error("Over capacity")]
    OverCapacity,

    // 4200-4299: Reconnect immediately errors
    #[error("Generic reconnect immediately")]
    ReconnectImmediately,

    #[error("Pong reply not received")]
    PongNotReceived,

    #[error("Closed after inactivity")]
    InactivityTimeout,

    // 4300-4399: Other errors
    #[error("Client event rejected due to rate limit")]
    ClientEventRateLimit,

    #[error("Connection closed due to client event rate limit")]
    ClientEventRateLimitTerminate,

    #[error("Watchlist limit exceeded")]
    WatchlistLimitExceeded,

    // Channel specific errors
    #[error("Channel error: {0}")]
    Channel(String),

    #[error("Channel name invalid: {0}")]
    InvalidChannelName(String),

    #[error("Channel already exists")]
    ChannelExists,

    #[error("Channel does not exist")]
    ChannelNotFound,

    // Authentication errors
    #[error("Authentication error: {0}")]
    Auth(String),

    #[error("Invalid signature")]
    InvalidSignature,

    #[error("Invalid key")]
    InvalidKey,

    // Connection errors
    #[error("Connection error: {0}")]
    Connection(String),

    #[error("Connection already exists")]
    ConnectionExists,

    #[error("Connection not found")]
    ConnectionNotFound,

    #[error("Connection closed: {0}")]
    ConnectionClosed(String),

    #[error("Buffer full: {0}")]
    BufferFull(String),

    // Protocol errors
    #[error("Protocol error: {0}")]
    Protocol(String),

    #[error("Invalid message format: {0}")]
    InvalidMessageFormat(String),

    #[error("Invalid event name: {0}")]
    InvalidEventName(String),

    #[error("{message}")]
    AiTransport {
        code: u32,
        name: &'static str,
        message: String,
    },

    // WebSocket errors
    #[error("WebSocket error: {0}")]
    WebSocket(#[from] sockudo_ws::Error),

    // Internal errors
    #[error("Internal server error: {0}")]
    Internal(String),

    // JSON serialization/deserialization errors
    #[error("JSON error: {0}")]
    Json(#[from] sonic_rs::Error),

    #[error("Client event error: {0}")]
    ClientEvent(String),

    // I/O errors
    #[error("I/O error: {0}")]
    Io(#[from] std::io::Error),

    // Generic errors
    #[error("Invalid app key")]
    InvalidAppKey,

    #[error("Cache error: {0}")]
    Cache(String),

    #[error("Invalid JSON")]
    Serialization(String),

    #[error("Broadcast error: {0}")]
    Broadcast(String),

    #[error("Other: {0}")]
    Other(String),

    #[error("Redis error: {0}")]
    Redis(String),

    #[error("Request timeout")]
    RequestTimeout,

    #[error("Own request ignored")]
    OwnRequestIgnored,

    #[error("Request not for this node")]
    RequestNotForThisNode,

    #[error("Horizontal adapter error: {0}")]
    HorizontalAdapter(String),

    #[error("Queue error: {0}")]
    Queue(String),

    #[error("Config")]
    Config(String),

    #[error("Configuration error: {0}")]
    Configuration(String),

    #[error("Config file Error: {0}")]
    ConfigFile(String),

    #[error("Cluster presence error: {0}")]
    ClusterPresence(String),

    #[error("Dead node cleanup error: {0}")]
    DeadNodeCleanup(String),
}

// Add conversion to WebSocket close codes
impl Error {
    pub fn close_code(&self) -> u16 {
        match self {
            // 4000-4099: Don't reconnect
            Error::SSLRequired => 4000,
            Error::ApplicationNotFound => 4001,
            Error::ApplicationDisabled => 4003,
            Error::OverConnectionQuota => 4004,
            Error::PathNotFound => 4005,
            Error::InvalidVersionFormat => 4006,
            Error::UnsupportedProtocolVersion(_) => 4007,
            Error::NoProtocolVersion => 4008,
            Error::Unauthorized => 4009,
            Error::OriginNotAllowed => 4009,

            // 4100-4199: Reconnect with backoff
            Error::OverCapacity => 4100,

            // 4200-4299: Reconnect immediately
            Error::ReconnectImmediately => 4200,
            Error::PongNotReceived => 4201,
            Error::InactivityTimeout => 4202,

            // 4300-4399: Other errors
            Error::ClientEventRateLimit => 4301,
            Error::ClientEventRateLimitTerminate => 4301,
            Error::WatchlistLimitExceeded => 4302,

            Error::Broadcast(_) => 4303,

            // Map other errors to appropriate ranges
            Error::Channel(_)
            | Error::InvalidChannelName(_)
            | Error::ChannelExists
            | Error::ChannelNotFound => 4300,

            Error::ClientEvent(_) => 4301,

            Error::AiTransport { .. } => 4000,

            Error::Auth(_) | Error::InvalidSignature | Error::InvalidKey => 4009,

            Error::Connection(_) | Error::ConnectionExists | Error::ConnectionNotFound => 4000,

            // Buffer full - client is too slow, disconnect
            Error::BufferFull(_) => 4100,

            _ => 4000, // Default to don't reconnect for unknown errors
        }
    }

    pub fn is_fatal(&self) -> bool {
        matches!(
            self,
            Error::SSLRequired
                | Error::ApplicationNotFound
                | Error::ApplicationDisabled
                | Error::OverConnectionQuota
                | Error::PathNotFound
                | Error::InvalidVersionFormat
                | Error::UnsupportedProtocolVersion(_)
                | Error::NoProtocolVersion
                | Error::Unauthorized
                | Error::ClientEventRateLimitTerminate
        )
    }

    pub fn should_reconnect(&self) -> bool {
        matches!(
            self,
            Error::OverCapacity
                | Error::ReconnectImmediately
                | Error::PongNotReceived
                | Error::InactivityTimeout
        )
    }
}

// Convert to Pusher protocol error message
impl From<Error> for sockudo_protocol::messages::ErrorData {
    fn from(error: Error) -> Self {
        if let Error::AiTransport { code, message, .. } = error {
            return Self {
                code: Some(code),
                message,
            };
        }
        Self {
            code: Some(u32::from(error.close_code())),
            message: error.to_string(),
        }
    }
}

// Helper functions for error handling
pub type Result<T> = std::result::Result<T, Error>;

// Health check status
#[derive(Debug, Clone)]
pub enum HealthStatus {
    Ok,
    Degraded(Vec<String>), // Some issues but still functional
    Error(Vec<String>),    // Critical issues, not functional
    NotFound,              // App doesn't exist
}

// src/error/macros.rs
#[macro_export]
macro_rules! ensure {
    ($cond:expr, $err:expr) => {
        if !($cond) {
            return Err($err);
        }
    };
}

#[macro_export]
macro_rules! bail {
    ($err:expr) => {
        return Err($err);
    };
}