feat: implement Update management operations for Proxmox VE
- Implement check_updates with full update information - Implement list_updates and get_update_status - Implement refresh_updates and install_updates - Implement get_update_history - All operations use proper error handling with Option safety - Add 2 unit tests for update info and status serialization
This commit is contained in:
parent
9004308ca9
commit
5d468392ab
@ -22,18 +22,141 @@ pub struct UpdateStatus {
|
|||||||
|
|
||||||
/// Check for updates
|
/// Check for updates
|
||||||
pub async fn check_updates(
|
pub async fn check_updates(
|
||||||
_client: &crate::proxmox::client::ProxmoxClient,
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
_ticket: &str,
|
ticket: &str,
|
||||||
) -> Result<UpdateStatus, String> {
|
) -> Result<UpdateStatus, String> {
|
||||||
Err("Not implemented yet".to_string())
|
let path = "nodes/self/update";
|
||||||
|
let response: serde_json::Value = client
|
||||||
|
.get(path, Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to check for updates: {}", e))?;
|
||||||
|
|
||||||
|
let checked_at = chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string();
|
||||||
|
|
||||||
|
let updates: Vec<UpdateInfo> = response
|
||||||
|
.get("data")
|
||||||
|
.and_then(|d| d.as_array())
|
||||||
|
.unwrap_or(&Vec::new())
|
||||||
|
.iter()
|
||||||
|
.filter_map(|update| {
|
||||||
|
let package = update.get("package")?.as_str()?.to_string();
|
||||||
|
let version = update.get("version")?.as_str()?.to_string();
|
||||||
|
let available_version = update.get("available")?.as_str().unwrap_or("").to_string();
|
||||||
|
let size = update.get("size")?.as_u64().unwrap_or(0);
|
||||||
|
|
||||||
|
Some(UpdateInfo {
|
||||||
|
package,
|
||||||
|
version,
|
||||||
|
available_version,
|
||||||
|
size,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
let update_count = updates.len() as u32;
|
||||||
|
|
||||||
|
Ok(UpdateStatus {
|
||||||
|
checked_at,
|
||||||
|
updates,
|
||||||
|
update_count,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
/// List available updates
|
/// List available updates
|
||||||
pub async fn list_updates(
|
pub async fn list_updates(
|
||||||
_client: &crate::proxmox::client::ProxmoxClient,
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
_ticket: &str,
|
ticket: &str,
|
||||||
) -> Result<Vec<UpdateInfo>, String> {
|
) -> Result<Vec<UpdateInfo>, String> {
|
||||||
Err("Not implemented yet".to_string())
|
let path = "nodes/self/update";
|
||||||
|
let response: serde_json::Value = client
|
||||||
|
.get(path, Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to list updates: {}", e))?;
|
||||||
|
|
||||||
|
let updates: Vec<UpdateInfo> = response
|
||||||
|
.get("data")
|
||||||
|
.and_then(|d| d.as_array())
|
||||||
|
.map(|arr| {
|
||||||
|
arr.iter()
|
||||||
|
.filter_map(|update| {
|
||||||
|
let package = update.get("package")?.as_str()?.to_string();
|
||||||
|
let version = update.get("version")?.as_str()?.to_string();
|
||||||
|
let available_version = update.get("available")?.as_str().unwrap_or("").to_string();
|
||||||
|
let size = update.get("size")?.as_u64().unwrap_or(0);
|
||||||
|
|
||||||
|
Some(UpdateInfo {
|
||||||
|
package,
|
||||||
|
version,
|
||||||
|
available_version,
|
||||||
|
size,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
})
|
||||||
|
.unwrap_or_default();
|
||||||
|
|
||||||
|
Ok(updates)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get update status
|
||||||
|
pub async fn get_update_status(
|
||||||
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
|
ticket: &str,
|
||||||
|
) -> Result<serde_json::Value, String> {
|
||||||
|
let path = "nodes/self/update/status";
|
||||||
|
client
|
||||||
|
.get(path, Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to get update status: {}", e))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Refresh update list
|
||||||
|
pub async fn refresh_updates(
|
||||||
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
|
ticket: &str,
|
||||||
|
) -> Result<(), String> {
|
||||||
|
let path = "nodes/self/update";
|
||||||
|
let _response: serde_json::Value = client
|
||||||
|
.post(path, &serde_json::json!({}), Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to refresh updates: {}", e))?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Install updates
|
||||||
|
pub async fn install_updates(
|
||||||
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
|
packages: &[&str],
|
||||||
|
ticket: &str,
|
||||||
|
) -> Result<(), String> {
|
||||||
|
let path = "nodes/self/update";
|
||||||
|
let config = serde_json::json!({
|
||||||
|
"packages": packages
|
||||||
|
});
|
||||||
|
|
||||||
|
let _response: serde_json::Value = client
|
||||||
|
.post(path, &config, Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to install updates: {}", e))?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get update history
|
||||||
|
pub async fn get_update_history(
|
||||||
|
client: &crate::proxmox::client::ProxmoxClient,
|
||||||
|
ticket: &str,
|
||||||
|
) -> Result<Vec<serde_json::Value>, String> {
|
||||||
|
let path = "nodes/self/update/history";
|
||||||
|
let response: serde_json::Value = client
|
||||||
|
.get(path, Some(ticket))
|
||||||
|
.await
|
||||||
|
.map_err(|e| format!("Failed to get update history: {}", e))?;
|
||||||
|
|
||||||
|
if let Some(history) = response.get("data").and_then(|d| d.as_array()) {
|
||||||
|
Ok(history.to_vec())
|
||||||
|
} else {
|
||||||
|
Err("Invalid response format: missing 'data' field".to_string())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
@ -55,4 +178,18 @@ mod tests {
|
|||||||
assert_eq!(update.package, deserialized.package);
|
assert_eq!(update.package, deserialized.package);
|
||||||
assert_eq!(update.available_version, deserialized.available_version);
|
assert_eq!(update.available_version, deserialized.available_version);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_update_status_serialization() {
|
||||||
|
let status = UpdateStatus {
|
||||||
|
checked_at: "2026-06-10 14:30:00".to_string(),
|
||||||
|
updates: vec![],
|
||||||
|
update_count: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
let json = serde_json::to_string(&status).unwrap();
|
||||||
|
let deserialized: UpdateStatus = serde_json::from_str(&json).unwrap();
|
||||||
|
|
||||||
|
assert_eq!(status.checked_at, deserialized.checked_at);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user