Skip to content

Commit

Permalink
refactor: split up DfxConfigError into method-specific error types
Browse files Browse the repository at this point in the history
  • Loading branch information
ericswanson-dfinity committed Sep 25, 2023
1 parent b24cc40 commit 99d37d1
Show file tree
Hide file tree
Showing 3 changed files with 67 additions and 36 deletions.
54 changes: 29 additions & 25 deletions src/dfx-core/src/config/model/dfinity.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,17 @@
use crate::config::directories::get_config_dfx_dir_path;
use crate::config::model::bitcoin_adapter::BitcoinAdapterLogLevel;
use crate::config::model::canister_http_adapter::HttpAdapterLogLevel;
use crate::error::dfx_config::DfxConfigError;
use crate::error::dfx_config::DfxConfigError::{
CanisterCircularDependency, CanisterNotFound, CanistersFieldDoesNotExist,
GetCanistersWithDependenciesFailed, GetComputeAllocationFailed, GetFreezingThresholdFailed,
GetMemoryAllocationFailed, GetRemoteCanisterIdFailed, PullCanistersSameId,
use crate::error::dfx_config::AddDependenciesError::CanisterCircularDependency;
use crate::error::dfx_config::GetCanisterNamesWithDependenciesError::AddDependenciesFailed;
use crate::error::dfx_config::GetComputeAllocationError::GetComputeAllocationFailed;
use crate::error::dfx_config::GetFreezingThresholdError::GetFreezingThresholdFailed;
use crate::error::dfx_config::GetMemoryAllocationError::GetMemoryAllocationFailed;
use crate::error::dfx_config::GetPullCanistersError::PullCanistersSameId;
use crate::error::dfx_config::GetRemoteCanisterIdError::GetRemoteCanisterIdFailed;
use crate::error::dfx_config::{
AddDependenciesError, GetCanisterConfigError, GetCanisterNamesWithDependenciesError,
GetComputeAllocationError, GetFreezingThresholdError, GetMemoryAllocationError,
GetPullCanistersError, GetRemoteCanisterIdError,
};
use crate::error::load_dfx_config::LoadDfxConfigError;
use crate::error::load_dfx_config::LoadDfxConfigError::{
Expand Down Expand Up @@ -749,36 +755,34 @@ impl ConfigInterface {
pub fn get_canister_names_with_dependencies(
&self,
some_canister: Option<&str>,
) -> Result<Vec<String>, DfxConfigError> {
) -> Result<Vec<String>, GetCanisterNamesWithDependenciesError> {
self.canisters
.as_ref()
.ok_or(CanistersFieldDoesNotExist())
.ok_or(GetCanisterNamesWithDependenciesError::CanistersFieldDoesNotExist())
.and_then(|canister_map| match some_canister {
Some(specific_canister) => {
let mut names = HashSet::new();
let mut path = vec![];
add_dependencies(canister_map, &mut names, &mut path, specific_canister)
.map(|_| names.into_iter().collect())
.map_err(|err| AddDependenciesFailed(specific_canister.to_string(), err))
}
None => Ok(canister_map.keys().cloned().collect()),
})
.map_err(|cause| {
GetCanistersWithDependenciesFailed(some_canister.map(String::from), Box::new(cause))
})
}

pub fn get_remote_canister_id(
&self,
canister: &str,
network: &str,
) -> Result<Option<Principal>, DfxConfigError> {
) -> Result<Option<Principal>, GetRemoteCanisterIdError> {
let maybe_principal = self
.get_canister_config(canister)
.map_err(|e| {
GetRemoteCanisterIdFailed(
Box::new(canister.to_string()),
Box::new(network.to_string()),
Box::new(e),
e,
)
})?
.remote
Expand All @@ -793,17 +797,17 @@ impl ConfigInterface {
&self,
canister: &str,
network: &str,
) -> Result<bool, DfxConfigError> {
) -> Result<bool, GetRemoteCanisterIdError> {
Ok(self.get_remote_canister_id(canister, network)?.is_some())
}

pub fn get_compute_allocation(
&self,
canister_name: &str,
) -> Result<Option<u64>, DfxConfigError> {
) -> Result<Option<u64>, GetComputeAllocationError> {
Ok(self
.get_canister_config(canister_name)
.map_err(|e| GetComputeAllocationFailed(canister_name.to_string(), Box::new(e)))?
.map_err(|e| GetComputeAllocationFailed(canister_name.to_string(), e))?
.initialization_values
.compute_allocation
.map(|x| x.0))
Expand All @@ -812,37 +816,37 @@ impl ConfigInterface {
pub fn get_memory_allocation(
&self,
canister_name: &str,
) -> Result<Option<Byte>, DfxConfigError> {
) -> Result<Option<Byte>, GetMemoryAllocationError> {
Ok(self
.get_canister_config(canister_name)
.map_err(|e| GetMemoryAllocationFailed(canister_name.to_string(), Box::new(e)))?
.map_err(|e| GetMemoryAllocationFailed(canister_name.to_string(), e))?
.initialization_values
.memory_allocation)
}

pub fn get_freezing_threshold(
&self,
canister_name: &str,
) -> Result<Option<Duration>, DfxConfigError> {
) -> Result<Option<Duration>, GetFreezingThresholdError> {
Ok(self
.get_canister_config(canister_name)
.map_err(|e| GetFreezingThresholdFailed(canister_name.to_string(), Box::new(e)))?
.map_err(|e| GetFreezingThresholdFailed(canister_name.to_string(), e))?
.initialization_values
.freezing_threshold)
}

fn get_canister_config(
&self,
canister_name: &str,
) -> Result<&ConfigCanistersCanister, DfxConfigError> {
) -> Result<&ConfigCanistersCanister, GetCanisterConfigError> {
self.canisters
.as_ref()
.ok_or(CanistersFieldDoesNotExist())?
.ok_or(GetCanisterConfigError::CanistersFieldDoesNotExist())?
.get(canister_name)
.ok_or_else(|| CanisterNotFound(canister_name.to_string()))
.ok_or_else(|| GetCanisterConfigError::CanisterNotFound(canister_name.to_string()))
}

pub fn get_pull_canisters(&self) -> Result<BTreeMap<String, Principal>, DfxConfigError> {
pub fn get_pull_canisters(&self) -> Result<BTreeMap<String, Principal>, GetPullCanistersError> {
let mut res = BTreeMap::new();
let mut id_to_name: BTreeMap<Principal, &String> = BTreeMap::new();
if let Some(map) = &self.canisters {
Expand All @@ -865,7 +869,7 @@ fn add_dependencies(
names: &mut HashSet<String>,
path: &mut Vec<String>,
canister_name: &str,
) -> Result<(), DfxConfigError> {
) -> Result<(), AddDependenciesError> {
let inserted = names.insert(String::from(canister_name));

if !inserted {
Expand All @@ -879,7 +883,7 @@ fn add_dependencies(

let canister_config = all_canisters
.get(canister_name)
.ok_or_else(|| CanisterNotFound(canister_name.to_string()))?;
.ok_or_else(|| AddDependenciesError::CanisterNotFound(canister_name.to_string()))?;

path.push(String::from(canister_name));

Expand Down
4 changes: 2 additions & 2 deletions src/dfx-core/src/error/canister_id_store.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
use crate::error::{
dfx_config::DfxConfigError, fs::FsError, structured_file::StructuredFileError,
dfx_config::GetPullCanistersError, fs::FsError, structured_file::StructuredFileError,
unified_io::UnifiedIoError,
};
use thiserror::Error;
Expand Down Expand Up @@ -37,7 +37,7 @@ pub enum CanisterIdStoreError {
},

#[error(transparent)]
DfxConfigError(#[from] DfxConfigError),
GetPullCanistersFailed(#[from] GetPullCanistersError),
}

impl From<FsError> for CanisterIdStoreError {
Expand Down
45 changes: 36 additions & 9 deletions src/dfx-core/src/error/dfx_config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,31 +2,58 @@ use candid::Principal;
use thiserror::Error;

#[derive(Error, Debug)]
pub enum DfxConfigError {
pub enum AddDependenciesError {
#[error("Circular canister dependencies: {}", _0.join(" -> "))]
CanisterCircularDependency(Vec<String>),

#[error("Canister '{0}' not found in dfx.json.")]
CanisterNotFound(String),
}

#[derive(Error, Debug)]
pub enum GetCanisterConfigError {
#[error("No canisters in the configuration file.")]
CanistersFieldDoesNotExist(),

#[error("Canister '{0}' not found in dfx.json.")]
CanisterNotFound(String),
}

#[derive(Error, Debug)]
pub enum GetCanisterNamesWithDependenciesError {
#[error("No canisters in the configuration file.")]
CanistersFieldDoesNotExist(),

#[error("Failed to get canisters with their dependencies (for {}): {1}", _0.as_deref().unwrap_or("all canisters"))]
GetCanistersWithDependenciesFailed(Option<String>, Box<DfxConfigError>),
#[error("Failed to add dependencies for canister '{0}': {1}")]
AddDependenciesFailed(String, AddDependenciesError),
}

#[derive(Error, Debug)]
pub enum GetComputeAllocationError {
#[error("Failed to get compute allocation for canister '{0}': {1}")]
GetComputeAllocationFailed(String, Box<DfxConfigError>),
GetComputeAllocationFailed(String, GetCanisterConfigError),
}

#[derive(Error, Debug)]
pub enum GetFreezingThresholdError {
#[error("Failed to get freezing threshold for canister '{0}': {1}")]
GetFreezingThresholdFailed(String, Box<DfxConfigError>),
GetFreezingThresholdFailed(String, GetCanisterConfigError),
}

#[derive(Error, Debug)]
pub enum GetMemoryAllocationError {
#[error("Failed to get memory allocation for canister '{0}': {1}")]
GetMemoryAllocationFailed(String, Box<DfxConfigError>),

#[error("Failed to figure out if canister '{0}' has a remote id on network '{1}': {2}")]
GetRemoteCanisterIdFailed(Box<String>, Box<String>, Box<DfxConfigError>),
GetMemoryAllocationFailed(String, GetCanisterConfigError),
}

#[derive(Error, Debug)]
pub enum GetPullCanistersError {
#[error("Pull dependencies '{0}' and '{1}' have the same canister ID: {2}")]
PullCanistersSameId(String, String, Principal),
}

#[derive(Error, Debug)]
pub enum GetRemoteCanisterIdError {
#[error("Failed to figure out if canister '{0}' has a remote id on network '{1}': {2}")]
GetRemoteCanisterIdFailed(Box<String>, Box<String>, GetCanisterConfigError),
}

0 comments on commit 99d37d1

Please sign in to comment.