rustfs/ecstore/src/store_err.rs

use crate::{
    disk::error::{is_err_file_not_found, DiskError},
    error::Error,
    utils::path::decode_dir_object,
};

#[derive(Debug, thiserror::Error, PartialEq, Eq)]
pub enum StorageError {
    #[error("not implemented")]
    NotImplemented,

    #[error("Invalid arguments provided for {0}/{1}-{2}")]
    InvalidArgument(String, String, String),

    #[error("method not allowed")]
    MethodNotAllowed,

    #[error("Bucket not found: {0}")]
    BucketNotFound(String),

    #[error("Bucket not empty: {0}")]
    BucketNotEmpty(String),

    #[error("Bucket name invalid: {0}")]
    BucketNameInvalid(String),

    #[error("Object name invalid: {0}/{1}")]
    ObjectNameInvalid(String, String),

    #[error("Bucket exists: {0}")]
    BucketExists(String),
    #[error("Storage reached its minimum free drive threshold.")]
    StorageFull,
    #[error("Please reduce your request rate")]
    SlowDown,

    #[error("Prefix access is denied:{0}/{1}")]
    PrefixAccessDenied(String, String),

    #[error("Invalid UploadID KeyCombination: {0}/{1}")]
    InvalidUploadIDKeyCombination(String, String),

    #[error("Malformed UploadID: {0}")]
    MalformedUploadID(String),

    #[error("Object name too long: {0}/{1}")]
    ObjectNameTooLong(String, String),

    #[error("Object name contains forward slash as prefix: {0}/{1}")]
    ObjectNamePrefixAsSlash(String, String),

    #[error("Object not found: {0}/{1}")]
    ObjectNotFound(String, String),

    #[error("volume not found: {0}")]
    VolumeNotFound(String),

    #[error("Version not found: {0}/{1}-{2}")]
    VersionNotFound(String, String, String),

    #[error("Invalid upload id: {0}/{1}-{2}")]
    InvalidUploadID(String, String, String),

    #[error("Specified part could not be found. PartNumber {0}, Expected {1}, got {2}")]
    InvalidPart(usize, String, String),

    #[error("Invalid version id: {0}/{1}-{2}")]
    InvalidVersionID(String, String, String),
    #[error("invalid data movement operation, source and destination pool are the same for : {0}/{1}-{2}")]
    DataMovementOverwriteErr(String, String, String),

    #[error("Object exists on :{0} as directory {1}")]
    ObjectExistsAsDirectory(String, String),

    #[error("Storage resources are insufficient for the read operation")]
    InsufficientReadQuorum,

    #[error("Storage resources are insufficient for the write operation")]
    InsufficientWriteQuorum,

    #[error("Decommission not started")]
    DecommissionNotStarted,
}

impl StorageError {
    pub fn to_u32(&self) -> u32 {
        match self {
            StorageError::NotImplemented => 0x01,
            StorageError::InvalidArgument(_, _, _) => 0x02,
            StorageError::MethodNotAllowed => 0x03,
            StorageError::BucketNotFound(_) => 0x04,
            StorageError::BucketNotEmpty(_) => 0x05,
            StorageError::BucketNameInvalid(_) => 0x06,
            StorageError::ObjectNameInvalid(_, _) => 0x07,
            StorageError::BucketExists(_) => 0x08,
            StorageError::StorageFull => 0x09,
            StorageError::SlowDown => 0x0A,
            StorageError::PrefixAccessDenied(_, _) => 0x0B,
            StorageError::InvalidUploadIDKeyCombination(_, _) => 0x0C,
            StorageError::MalformedUploadID(_) => 0x0D,
            StorageError::ObjectNameTooLong(_, _) => 0x0E,
            StorageError::ObjectNamePrefixAsSlash(_, _) => 0x0F,
            StorageError::ObjectNotFound(_, _) => 0x10,
            StorageError::VersionNotFound(_, _, _) => 0x11,
            StorageError::InvalidUploadID(_, _, _) => 0x12,
            StorageError::InvalidVersionID(_, _, _) => 0x13,
            StorageError::DataMovementOverwriteErr(_, _, _) => 0x14,
            StorageError::ObjectExistsAsDirectory(_, _) => 0x15,
            StorageError::InsufficientReadQuorum => 0x16,
            StorageError::InsufficientWriteQuorum => 0x17,
            StorageError::DecommissionNotStarted => 0x18,
            StorageError::InvalidPart(_, _, _) => 0x19,
            StorageError::VolumeNotFound(_) => 0x20,
        }
    }

    pub fn from_u32(error: u32) -> Option<Self> {
        match error {
            0x01 => Some(StorageError::NotImplemented),
            0x02 => Some(StorageError::InvalidArgument(Default::default(), Default::default(), Default::default())),
            0x03 => Some(StorageError::MethodNotAllowed),
            0x04 => Some(StorageError::BucketNotFound(Default::default())),
            0x05 => Some(StorageError::BucketNotEmpty(Default::default())),
            0x06 => Some(StorageError::BucketNameInvalid(Default::default())),
            0x07 => Some(StorageError::ObjectNameInvalid(Default::default(), Default::default())),
            0x08 => Some(StorageError::BucketExists(Default::default())),
            0x09 => Some(StorageError::StorageFull),
            0x0A => Some(StorageError::SlowDown),
            0x0B => Some(StorageError::PrefixAccessDenied(Default::default(), Default::default())),
            0x0C => Some(StorageError::InvalidUploadIDKeyCombination(Default::default(), Default::default())),
            0x0D => Some(StorageError::MalformedUploadID(Default::default())),
            0x0E => Some(StorageError::ObjectNameTooLong(Default::default(), Default::default())),
            0x0F => Some(StorageError::ObjectNamePrefixAsSlash(Default::default(), Default::default())),
            0x10 => Some(StorageError::ObjectNotFound(Default::default(), Default::default())),
            0x11 => Some(StorageError::VersionNotFound(Default::default(), Default::default(), Default::default())),
            0x12 => Some(StorageError::InvalidUploadID(Default::default(), Default::default(), Default::default())),
            0x13 => Some(StorageError::InvalidVersionID(Default::default(), Default::default(), Default::default())),
            0x14 => Some(StorageError::DataMovementOverwriteErr(
                Default::default(),
                Default::default(),
                Default::default(),
            )),
            0x15 => Some(StorageError::ObjectExistsAsDirectory(Default::default(), Default::default())),
            0x16 => Some(StorageError::InsufficientReadQuorum),
            0x17 => Some(StorageError::InsufficientWriteQuorum),
            0x18 => Some(StorageError::DecommissionNotStarted),
            0x19 => Some(StorageError::InvalidPart(Default::default(), Default::default(), Default::default())),
            0x20 => Some(StorageError::VolumeNotFound(Default::default())),
            _ => None,
        }
    }
}

pub fn to_object_err(err: Error, params: Vec<&str>) -> Error {
    if let Some(e) = err.downcast_ref::<DiskError>() {
        match e {
            DiskError::DiskFull => {
                return Error::new(StorageError::StorageFull);
            }

            DiskError::FileNotFound => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                let object = params.get(1).cloned().map(decode_dir_object).unwrap_or_default();

                return Error::new(StorageError::ObjectNotFound(bucket, object));
            }
            DiskError::FileVersionNotFound => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                let object = params.get(1).cloned().map(decode_dir_object).unwrap_or_default();
                let version = params.get(2).cloned().unwrap_or_default().to_owned();

                return Error::new(StorageError::VersionNotFound(bucket, object, version));
            }
            DiskError::TooManyOpenFiles => {
                return Error::new(StorageError::SlowDown);
            }
            DiskError::FileNameTooLong => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                let object = params.get(1).cloned().map(decode_dir_object).unwrap_or_default();

                return Error::new(StorageError::ObjectNameInvalid(bucket, object));
            }
            DiskError::VolumeExists => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                return Error::new(StorageError::BucketExists(bucket));
            }
            DiskError::IsNotRegular => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                let object = params.get(1).cloned().map(decode_dir_object).unwrap_or_default();

                return Error::new(StorageError::ObjectExistsAsDirectory(bucket, object));
            }

            DiskError::VolumeNotFound => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                return Error::new(StorageError::BucketNotFound(bucket));
            }
            DiskError::VolumeNotEmpty => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                return Error::new(StorageError::BucketNotEmpty(bucket));
            }

            DiskError::FileAccessDenied => {
                let bucket = params.first().cloned().unwrap_or_default().to_owned();
                let object = params.get(1).cloned().map(decode_dir_object).unwrap_or_default();

                return Error::new(StorageError::PrefixAccessDenied(bucket, object));
            }
            // DiskError::MaxVersionsExceeded => todo!(),
            // DiskError::Unexpected => todo!(),
            // DiskError::CorruptedFormat => todo!(),
            // DiskError::CorruptedBackend => todo!(),
            // DiskError::UnformattedDisk => todo!(),
            // DiskError::InconsistentDisk => todo!(),
            // DiskError::UnsupportedDisk => todo!(),
            // DiskError::DiskNotDir => todo!(),
            // DiskError::DiskNotFound => todo!(),
            // DiskError::DiskOngoingReq => todo!(),
            // DiskError::DriveIsRoot => todo!(),
            // DiskError::FaultyRemoteDisk => todo!(),
            // DiskError::FaultyDisk => todo!(),
            // DiskError::DiskAccessDenied => todo!(),
            // DiskError::FileCorrupt => todo!(),
            // DiskError::BitrotHashAlgoInvalid => todo!(),
            // DiskError::CrossDeviceLink => todo!(),
            // DiskError::LessData => todo!(),
            // DiskError::MoreData => todo!(),
            // DiskError::OutdatedXLMeta => todo!(),
            // DiskError::PartMissingOrCorrupt => todo!(),
            // DiskError::PathNotFound => todo!(),
            // DiskError::VolumeAccessDenied => todo!(),
            _ => (),
        }
    }

    err
}

pub fn is_err_read_quorum(err: &Error) -> bool {
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::InsufficientReadQuorum)
    } else {
        false
    }
}

pub fn is_err_invalid_upload_id(err: &Error) -> bool {
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::InvalidUploadID(_, _, _))
    } else {
        false
    }
}

pub fn is_err_version_not_found(err: &Error) -> bool {
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::VersionNotFound(_, _, _))
    } else {
        false
    }
}

pub fn is_err_bucket_exists(err: &Error) -> bool {
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::BucketExists(_))
    } else {
        false
    }
}

pub fn is_err_bucket_not_found(err: &Error) -> bool {
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::VolumeNotFound(_)) || matches!(e, StorageError::BucketNotFound(_))
    } else {
        false
    }
}

pub fn is_err_object_not_found(err: &Error) -> bool {
    if is_err_file_not_found(err) {
        return true;
    }
    if let Some(e) = err.downcast_ref::<StorageError>() {
        matches!(e, StorageError::ObjectNotFound(_, _))
    } else {
        false
    }
}

#[test]
fn test_storage_error() {
    let e1 = Error::new(StorageError::BucketExists("ss".into()));
    let e2 = Error::new(StorageError::ObjectNotFound("ss".into(), "sdf".to_owned()));
    assert!(is_err_bucket_exists(&e1));
    assert!(!is_err_object_not_found(&e1));
    assert!(is_err_object_not_found(&e2));
}