diff --git a/pbs-config/src/datastore.rs b/pbs-config/src/datastore.rs index dc5bb3da..396dcb37 100644 --- a/pbs-config/src/datastore.rs +++ b/pbs-config/src/datastore.rs @@ -62,6 +62,20 @@ pub fn complete_datastore_name(_arg: &str, _param: &HashMap) -> } } +pub fn complete_removable_datastore_name( + _arg: &str, + _param: &HashMap, +) -> Vec { + match config() { + Ok((data, _digest)) => data + .sections + .into_iter() + .filter_map(|(name, (_, c))| c.get("backing-device").map(|_| name)) + .collect(), + Err(_) => Vec::new(), + } +} + pub fn complete_acl_path(_arg: &str, _param: &HashMap) -> Vec { let mut list = vec![ String::from("/"), diff --git a/src/bin/proxmox_backup_manager/datastore.rs b/src/bin/proxmox_backup_manager/datastore.rs index 3a349451..32a55fb9 100644 --- a/src/bin/proxmox_backup_manager/datastore.rs +++ b/src/bin/proxmox_backup_manager/datastore.rs @@ -42,6 +42,34 @@ fn list_datastores(param: Value, rpcenv: &mut dyn RpcEnvironment) -> Result Result<(), Error> { + param["node"] = "localhost".into(); + + let info = &api2::admin::datastore::API_METHOD_MOUNT; + let result = match info.handler { + ApiHandler::Sync(handler) => (handler)(param, info, rpcenv)?, + _ => unreachable!(), + }; + + crate::wait_for_local_worker(result.as_str().unwrap()).await?; + Ok(()) +} + #[api( input: { properties: { @@ -101,6 +129,34 @@ async fn create_datastore(mut param: Value) -> Result { Ok(Value::Null) } +#[api( + protected: true, + input: { + properties: { + store: { + schema: DATASTORE_SCHEMA, + }, + digest: { + optional: true, + schema: PROXMOX_CONFIG_DIGEST_SCHEMA, + }, + }, + }, +)] +/// Unmount a removable datastore. +async fn unmount_datastore(mut param: Value, rpcenv: &mut dyn RpcEnvironment) -> Result<(), Error> { + param["node"] = "localhost".into(); + + let info = &api2::admin::datastore::API_METHOD_UNMOUNT; + let result = match info.handler { + ApiHandler::Async(handler) => (handler)(param, info, rpcenv).await?, + _ => unreachable!(), + }; + + crate::wait_for_local_worker(result.as_str().unwrap()).await?; + Ok(()) +} + #[api( protected: true, input: { @@ -191,6 +247,15 @@ async fn update_datastore(name: String, mut param: Value) -> Result<(), Error> { pub fn datastore_commands() -> CommandLineInterface { let cmd_def = CliCommandMap::new() .insert("list", CliCommand::new(&API_METHOD_LIST_DATASTORES)) + .insert( + "mount", + CliCommand::new(&API_METHOD_MOUNT_DATASTORE) + .arg_param(&["store"]) + .completion_cb( + "store", + pbs_config::datastore::complete_removable_datastore_name, + ), + ) .insert( "show", CliCommand::new(&API_METHOD_SHOW_DATASTORE) @@ -201,6 +266,15 @@ pub fn datastore_commands() -> CommandLineInterface { "create", CliCommand::new(&API_METHOD_CREATE_DATASTORE).arg_param(&["name", "path"]), ) + .insert( + "unmount", + CliCommand::new(&API_METHOD_UNMOUNT_DATASTORE) + .arg_param(&["store"]) + .completion_cb( + "store", + pbs_config::datastore::complete_removable_datastore_name, + ), + ) .insert( "update", CliCommand::new(&API_METHOD_UPDATE_DATASTORE)