services/management/enclave/src/error.rs

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

use teaclave_rpc::{Code, Status};
use thiserror::Error;

#[derive(Error, Debug)]
pub(crate) enum ManagementServiceError {
    #[error("service internal error")]
    Service(#[from] anyhow::Error),
    #[error("permission denied")]
    PermissionDenied,
    #[error("missing user id")]
    MissingUserId,
    #[error("missing user role")]
    MissingUserRole,
    #[error("invalid data id")]
    InvalidDataId,
    #[error("invalid output file")]
    InvalidOutputFile,
    #[error("invalid function id")]
    InvalidFunctionId,
    #[error("invalid task id")]
    InvalidTaskId,
    #[error("invalid task")]
    InvalidTask,
    #[error("failed to assign data to task")]
    TaskAssignDataError,
    #[error("failed to approve task")]
    TaskApproveError,
    #[error("failed to invoke task")]
    TaskInvokeError,
    #[error("failed to cancel task, reason: {0}")]
    TaskCancelError(String),
    #[error("function quota has been used up")]
    FunctionQuotaError,
    #[error("audit log error, reason: {0}")]
    AuditError(String),
}

impl From<ManagementServiceError> for Status {
    fn from(error: ManagementServiceError) -> Self {
        log::debug!("ManagementServiceError: {:?}", error);
        let msg = error.to_string();
        let code = match error {
            ManagementServiceError::PermissionDenied => Code::PermissionDenied,
            ManagementServiceError::Service(_) => Code::Internal,
            ManagementServiceError::InvalidDataId
            | ManagementServiceError::InvalidOutputFile
            | ManagementServiceError::InvalidFunctionId
            | ManagementServiceError::InvalidTaskId
            | ManagementServiceError::InvalidTask => Code::InvalidArgument,
            _ => Code::Unknown,
        };
        Status::new(code, msg)
    }
}