use crate::files::{MacFiles, mutation::supervision, policy, protocol, store::Store};
use cueward_core::files::mutation::MutationStatus;
use cueward_core::files::trash::execution::TrashPlatform;
use cueward_core::files::trash::restore;
use cueward_core::files::trash::restore::RestoreRequest;
pub use cueward_core::files::trash::restore::RestoreRequest as Request;
pub use cueward_core::files::trash::restore::{RestoreReceipt, RestoreStage};
use cueward_core::files::*;
use serde::{Deserialize, Serialize};
use std::path::Path;
const STORE: Store = Store("files");
#[derive(Debug, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct RestoreWorkerRequest {
pub operation_id: String,
}
pub fn run(
executable: &Path,
request: &RestoreRequest,
timeout: u64,
) -> Result<RestoreReceipt, FileError> {
protocol::payload(request, timeout)?;
restore::validate_request(request)?;
let prepared = STORE.create(|id, path| RestoreReceipt::new(id, path, request.clone()))?;
supervision::announce(&prepared.operation_id, "trash restore-receipt")?;
let result = supervision::run(
executable,
"files-trash-restore-worker",
&RestoreWorkerRequest {
operation_id: prepared.operation_id.clone(),
},
timeout,
)
.and_then(|receipt| {
protocol::verify_receipt(
&prepared,
receipt,
&read_receipt(&prepared.operation_id)?,
"restore worker response differs from prepared/stored evidence",
)
});
match result {
Ok(receipt) => Ok(receipt),
Err(mut error) => {
error.message.push_str("; inspect the receipt, restore staging and original destination; never automatically retry or delete");
let mut receipt = read_receipt(&prepared.operation_id).unwrap_or(prepared);
restore::record_error(&mut receipt, error);
receipt.status = MutationStatus::Uncertain;
save(&receipt)?;
Ok(receipt)
}
}
}
pub fn read_receipt(id: &str) -> Result<RestoreReceipt, FileError> {
let receipt: RestoreReceipt = STORE.load(id)?;
if receipt.operation != "trash_restore"
|| receipt.operation_id != id
|| STORE.directory(id)?.join("receipt.json").to_str() != Some(&receipt.receipt_path)
{
return Err(FileError::new(
FileErrorCode::InvalidOptions,
"restore receipt schema/identity/path mismatch",
));
}
Ok(receipt)
}
pub fn read_supervised_request() -> Result<RestoreWorkerRequest, FileError> {
supervision::read_request()
}
pub fn execute_worker(input: &RestoreWorkerRequest) -> Result<RestoreReceipt, FileError> {
execute_prepared(&MacFiles, input)
}
fn execute_prepared(
platform: &impl TrashPlatform,
input: &RestoreWorkerRequest,
) -> Result<RestoreReceipt, FileError> {
STORE.claim(&input.operation_id)?;
let mut receipt = read_receipt(&input.operation_id)?;
receipt.validate_fresh()?;
let result = policy::NoMaterialization::enter().and_then(|_policy| {
let original = super::execution::read_receipt(&receipt.request.trash_operation_id)?;
let expected = restore::fingerprint(&original)?;
restore::execute(platform, &mut receipt, &original, &mut |r| {
let current = super::execution::read_receipt(&r.request.trash_operation_id)?;
if restore::fingerprint(¤t)? != expected {
return Err(FileError::new(
FileErrorCode::Changed,
"original trash receipt changed during restore",
));
}
save(r)
})
});
if let Err(error) = result {
restore::record_error(&mut receipt, error);
}
save(&receipt)?;
Ok(receipt)
}
fn save(receipt: &RestoreReceipt) -> Result<(), FileError> {
STORE.save(&receipt.operation_id, &receipt.receipt_path, receipt)
}
#[cfg(test)]
#[path = "restore_tests.rs"]
mod tests;
#[cfg(test)]
#[path = "restore_safety_tests.rs"]
mod safety_tests;
#[cfg(test)]
#[path = "restore_lifetime_tests.rs"]
mod lifetime_tests;