src/api2/config/job.rs: add job api
This commit is contained in:
		@ -3,9 +3,11 @@ use proxmox::list_subdirs_api_method;
 | 
			
		||||
 | 
			
		||||
pub mod datastore;
 | 
			
		||||
pub mod remote;
 | 
			
		||||
pub mod job;
 | 
			
		||||
 | 
			
		||||
const SUBDIRS: SubdirMap = &[
 | 
			
		||||
    ("datastore", &datastore::ROUTER),
 | 
			
		||||
    ("job", &job::ROUTER),
 | 
			
		||||
    ("remote", &remote::ROUTER),
 | 
			
		||||
];
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										277
									
								
								src/api2/config/job.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										277
									
								
								src/api2/config/job.rs
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,277 @@
 | 
			
		||||
use anyhow::{bail, Error};
 | 
			
		||||
use serde_json::Value;
 | 
			
		||||
use ::serde::{Deserialize, Serialize};
 | 
			
		||||
 | 
			
		||||
use proxmox::api::{api, Router, RpcEnvironment};
 | 
			
		||||
 | 
			
		||||
use crate::api2::types::*;
 | 
			
		||||
use crate::config::jobs::{self, PullJobConfig};
 | 
			
		||||
 | 
			
		||||
// fixme: add access permissions
 | 
			
		||||
 | 
			
		||||
#[api(
 | 
			
		||||
    input: {
 | 
			
		||||
        properties: {},
 | 
			
		||||
    },
 | 
			
		||||
    returns: {
 | 
			
		||||
        description: "List configured jobs.",
 | 
			
		||||
        type: Array,
 | 
			
		||||
        items: { type: jobs::PullJobConfig },
 | 
			
		||||
    },
 | 
			
		||||
)]
 | 
			
		||||
/// List all pull jobs
 | 
			
		||||
pub fn list_pull_jobs(
 | 
			
		||||
    _param: Value,
 | 
			
		||||
    mut rpcenv: &mut dyn RpcEnvironment,
 | 
			
		||||
) -> Result<Vec<PullJobConfig>, Error> {
 | 
			
		||||
 | 
			
		||||
    let (config, digest) = jobs::config()?;
 | 
			
		||||
 | 
			
		||||
    let list = config.convert_to_typed_array("pull")?;
 | 
			
		||||
 | 
			
		||||
    rpcenv["digest"] = proxmox::tools::digest_to_hex(&digest).into();
 | 
			
		||||
 | 
			
		||||
    Ok(list)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[api(
 | 
			
		||||
    protected: true,
 | 
			
		||||
    input: {
 | 
			
		||||
        properties: {
 | 
			
		||||
            id: {
 | 
			
		||||
                schema: JOB_ID_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            store: {
 | 
			
		||||
                schema: DATASTORE_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            remote: {
 | 
			
		||||
                schema: REMOTE_ID_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            "remote-store": {
 | 
			
		||||
                schema: DATASTORE_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            "remove-vanished": {
 | 
			
		||||
                schema: REMOVE_VANISHED_BACKUPS_SCHEMA,
 | 
			
		||||
                optional: true,
 | 
			
		||||
            },
 | 
			
		||||
            comment: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: SINGLE_LINE_COMMENT_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            schedule: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: GC_SCHEDULE_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
        },
 | 
			
		||||
    },
 | 
			
		||||
)]
 | 
			
		||||
/// Create a new pull job.
 | 
			
		||||
pub fn create_pull_job(param: Value) -> Result<(), Error> {
 | 
			
		||||
 | 
			
		||||
    let _lock = crate::tools::open_file_locked(jobs::JOB_CFG_LOCKFILE, std::time::Duration::new(10, 0))?;
 | 
			
		||||
 | 
			
		||||
    let pull_job: jobs::PullJobConfig = serde_json::from_value(param.clone())?;
 | 
			
		||||
 | 
			
		||||
    let (mut config, _digest) = jobs::config()?;
 | 
			
		||||
 | 
			
		||||
    if let Some(_) = config.sections.get(&pull_job.id) {
 | 
			
		||||
        bail!("job '{}' already exists.", pull_job.id);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    config.set_data(&pull_job.id, "pull", &pull_job)?;
 | 
			
		||||
 | 
			
		||||
    jobs::save_config(&config)?;
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[api(
 | 
			
		||||
   input: {
 | 
			
		||||
        properties: {
 | 
			
		||||
            id: {
 | 
			
		||||
                schema: JOB_ID_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
        },
 | 
			
		||||
    },
 | 
			
		||||
    returns: {
 | 
			
		||||
        description: "The pull job configuration.",
 | 
			
		||||
        type: jobs::PullJobConfig,
 | 
			
		||||
    },
 | 
			
		||||
)]
 | 
			
		||||
/// Read a pull job configuration.
 | 
			
		||||
pub fn read_pull_job(
 | 
			
		||||
    id: String,
 | 
			
		||||
    mut rpcenv: &mut dyn RpcEnvironment,
 | 
			
		||||
) -> Result<PullJobConfig, Error> {
 | 
			
		||||
    let (config, digest) = jobs::config()?;
 | 
			
		||||
 | 
			
		||||
    let pull_job = config.lookup("pull", &id)?;
 | 
			
		||||
    rpcenv["digest"] = proxmox::tools::digest_to_hex(&digest).into();
 | 
			
		||||
 | 
			
		||||
    Ok(pull_job)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[api()]
 | 
			
		||||
#[derive(Serialize, Deserialize)]
 | 
			
		||||
#[serde(rename_all="kebab-case")]
 | 
			
		||||
#[allow(non_camel_case_types)]
 | 
			
		||||
/// Deletable property name
 | 
			
		||||
pub enum DeletableProperty {
 | 
			
		||||
    /// Delete the comment property.
 | 
			
		||||
    comment,
 | 
			
		||||
    /// Delete the job schedule.
 | 
			
		||||
    schedule,
 | 
			
		||||
    /// Delete the remove-vanished flag.
 | 
			
		||||
    remove_vanished,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[api(
 | 
			
		||||
    protected: true,
 | 
			
		||||
    input: {
 | 
			
		||||
        properties: {
 | 
			
		||||
            id: {
 | 
			
		||||
                schema: JOB_ID_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            store: {
 | 
			
		||||
                schema: DATASTORE_SCHEMA,
 | 
			
		||||
                optional: true,
 | 
			
		||||
            },
 | 
			
		||||
            remote: {
 | 
			
		||||
                schema: REMOTE_ID_SCHEMA,
 | 
			
		||||
                optional: true,
 | 
			
		||||
            },
 | 
			
		||||
            "remote-store": {
 | 
			
		||||
                schema: DATASTORE_SCHEMA,
 | 
			
		||||
                optional: true,
 | 
			
		||||
            },
 | 
			
		||||
            "remove-vanished": {
 | 
			
		||||
                schema: REMOVE_VANISHED_BACKUPS_SCHEMA,
 | 
			
		||||
                optional: true,
 | 
			
		||||
            },
 | 
			
		||||
            comment: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: SINGLE_LINE_COMMENT_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            schedule: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: GC_SCHEDULE_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            delete: {
 | 
			
		||||
                description: "List of properties to delete.",
 | 
			
		||||
                type: Array,
 | 
			
		||||
                optional: true,
 | 
			
		||||
                items: {
 | 
			
		||||
                    type: DeletableProperty,
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
            digest: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: PROXMOX_CONFIG_DIGEST_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
        },
 | 
			
		||||
    },
 | 
			
		||||
)]
 | 
			
		||||
/// Update pull job config.
 | 
			
		||||
pub fn update_pull_job(
 | 
			
		||||
    id: String,
 | 
			
		||||
    store: Option<String>,
 | 
			
		||||
    remote: Option<String>,
 | 
			
		||||
    remote_store: Option<String>,
 | 
			
		||||
    remove_vanished: Option<bool>,
 | 
			
		||||
    comment: Option<String>,
 | 
			
		||||
    schedule: Option<String>,
 | 
			
		||||
    delete: Option<Vec<DeletableProperty>>,
 | 
			
		||||
    digest: Option<String>,
 | 
			
		||||
) -> Result<(), Error> {
 | 
			
		||||
 | 
			
		||||
    let _lock = crate::tools::open_file_locked(jobs::JOB_CFG_LOCKFILE, std::time::Duration::new(10, 0))?;
 | 
			
		||||
 | 
			
		||||
    // pass/compare digest
 | 
			
		||||
    let (mut config, expected_digest) = jobs::config()?;
 | 
			
		||||
 | 
			
		||||
    if let Some(ref digest) = digest {
 | 
			
		||||
        let digest = proxmox::tools::hex_to_digest(digest)?;
 | 
			
		||||
        crate::tools::detect_modified_configuration_file(&digest, &expected_digest)?;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let mut data: jobs::PullJobConfig = config.lookup("pull", &id)?;
 | 
			
		||||
 | 
			
		||||
     if let Some(delete) = delete {
 | 
			
		||||
        for delete_prop in delete {
 | 
			
		||||
            match delete_prop {
 | 
			
		||||
                DeletableProperty::comment => { data.comment = None; },
 | 
			
		||||
                DeletableProperty::schedule => { data.schedule = None; },
 | 
			
		||||
                DeletableProperty::remove_vanished => { data.remove_vanished = None; },
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if let Some(comment) = comment {
 | 
			
		||||
        let comment = comment.trim().to_string();
 | 
			
		||||
        if comment.is_empty() {
 | 
			
		||||
            data.comment = None;
 | 
			
		||||
        } else {
 | 
			
		||||
            data.comment = Some(comment);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if let Some(store) = store { data.store = store; }
 | 
			
		||||
    if let Some(remote) = remote { data.remote = remote; }
 | 
			
		||||
    if let Some(remote_store) = remote_store { data.remote_store = remote_store; }
 | 
			
		||||
        
 | 
			
		||||
    
 | 
			
		||||
    if schedule.is_some() { data.schedule = schedule; }
 | 
			
		||||
    if remove_vanished.is_some() { data.remove_vanished = remove_vanished; }
 | 
			
		||||
 | 
			
		||||
    config.set_data(&id, "pull", &data)?;
 | 
			
		||||
 | 
			
		||||
    jobs::save_config(&config)?;
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[api(
 | 
			
		||||
    protected: true,
 | 
			
		||||
    input: {
 | 
			
		||||
        properties: {
 | 
			
		||||
            id: {
 | 
			
		||||
                schema: JOB_ID_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
            digest: {
 | 
			
		||||
                optional: true,
 | 
			
		||||
                schema: PROXMOX_CONFIG_DIGEST_SCHEMA,
 | 
			
		||||
            },
 | 
			
		||||
        },
 | 
			
		||||
    },
 | 
			
		||||
)]
 | 
			
		||||
/// Remove a job configuration
 | 
			
		||||
pub fn delete_job(id: String, digest: Option<String>) -> Result<(), Error> {
 | 
			
		||||
 | 
			
		||||
    let _lock = crate::tools::open_file_locked(jobs::JOB_CFG_LOCKFILE, std::time::Duration::new(10, 0))?;
 | 
			
		||||
 | 
			
		||||
    let (mut config, expected_digest) = jobs::config()?;
 | 
			
		||||
 | 
			
		||||
    if let Some(ref digest) = digest {
 | 
			
		||||
        let digest = proxmox::tools::hex_to_digest(digest)?;
 | 
			
		||||
        crate::tools::detect_modified_configuration_file(&digest, &expected_digest)?;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    match config.sections.get(&id) {
 | 
			
		||||
        Some(_) => { config.sections.remove(&id); },
 | 
			
		||||
        None => bail!("job '{}' does not exist.", id),
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    jobs::save_config(&config)?;
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
const ITEM_ROUTER: Router = Router::new()
 | 
			
		||||
    .get(&API_METHOD_READ_PULL_JOB)
 | 
			
		||||
    .put(&API_METHOD_UPDATE_PULL_JOB)
 | 
			
		||||
    .delete(&API_METHOD_DELETE_JOB);
 | 
			
		||||
 | 
			
		||||
pub const ROUTER: Router = Router::new()
 | 
			
		||||
    .get(&API_METHOD_LIST_PULL_JOBS)
 | 
			
		||||
    .post(&API_METHOD_CREATE_PULL_JOB)
 | 
			
		||||
    .match_all("name", &ITEM_ROUTER);
 | 
			
		||||
		Reference in New Issue
	
	Block a user