use std::collections::HashMap; use anyhow::Error; use lazy_static::lazy_static; use proxmox_schema::*; use proxmox_section_config::{SectionConfig, SectionConfigData, SectionConfigPlugin}; use pbs_api_types::{JOB_ID_SCHEMA, VerificationJobConfig}; use crate::{open_backup_lockfile, replace_backup_config, BackupLockGuard}; lazy_static! { pub static ref CONFIG: SectionConfig = init(); } fn init() -> SectionConfig { let obj_schema = match VerificationJobConfig::API_SCHEMA { Schema::Object(ref obj_schema) => obj_schema, _ => unreachable!(), }; let plugin = SectionConfigPlugin::new("verification".to_string(), Some(String::from("id")), obj_schema); let mut config = SectionConfig::new(&JOB_ID_SCHEMA); config.register_plugin(plugin); config } pub const VERIFICATION_CFG_FILENAME: &str = "/etc/proxmox-backup/verification.cfg"; pub const VERIFICATION_CFG_LOCKFILE: &str = "/etc/proxmox-backup/.verification.lck"; /// Get exclusive lock pub fn lock_config() -> Result { open_backup_lockfile(VERIFICATION_CFG_LOCKFILE, None, true) } pub fn config() -> Result<(SectionConfigData, [u8;32]), Error> { let content = proxmox_sys::fs::file_read_optional_string(VERIFICATION_CFG_FILENAME)?; let content = content.unwrap_or_else(String::new); let digest = openssl::sha::sha256(content.as_bytes()); let data = CONFIG.parse(VERIFICATION_CFG_FILENAME, &content)?; Ok((data, digest)) } pub fn save_config(config: &SectionConfigData) -> Result<(), Error> { let raw = CONFIG.write(VERIFICATION_CFG_FILENAME, config)?; replace_backup_config(VERIFICATION_CFG_FILENAME, raw.as_bytes()) } // shell completion helper pub fn complete_verification_job_id(_arg: &str, _param: &HashMap) -> Vec { match config() { Ok((data, _digest)) => data.sections.iter().map(|(id, _)| id.to_string()).collect(), Err(_) => return vec![], } }