1pub mod converge;
89pub mod fingerprint;
90pub mod gates;
91pub mod hold;
92pub mod host;
93pub mod journal;
94pub mod preflight;
95pub mod record;
96pub mod rollback;
97pub mod run_store;
98
99use std::collections::{BTreeMap, BTreeSet};
100use std::path::{Path, PathBuf};
101
102use serde::Serialize;
103use serde_json::{Value, json};
104
105use crate::actions::backup::backup_download;
106use crate::actions::deploy::{
107 Declared, DeployPlanResult, PlanOptions, VerifyOptions, deploy_plan, deploy_verify,
108};
109use crate::actions::workspace::rfc3339_now_utc;
110use crate::client::backup::BackupType;
111use crate::client::deploy_api::scan_and_wait;
112use crate::client::scan::{LockVerdict, ScanKind, ScanOutcome};
113use crate::client::{DeployApi, GatewayApi};
114use crate::deploy::layout::RelPath;
115use crate::deploy::ledger::{Ledger, LedgerRoot, journal_location, ledger_location};
116use crate::deploy::pair::{BackupNode, BackupPolicy, BackupReport, wall_clock_ms};
117use crate::deploy::plan::PlanStatus;
118use crate::deploy::target::NodeTarget;
119use crate::deploy::transport::TreeReader;
120use crate::deploy::transport::batch::BUDGET_MAX_S;
121use crate::deploy::tree::sha256_hex;
122use crate::error::{CoreError, PreflightFailure, RollbackFailure};
123use crate::poll::PollConfig;
124
125use self::fingerprint::{LEDGER_KEY, fingerprint};
126use self::gates::GateFlags;
127use self::hold::{Clock, HeldLock, HoldModel, LOCK_MARGIN_S};
128use self::host::{ApplyHost, BatchSpec, DeleteOp, WriteOp};
129use self::journal::{Journal, JournalPhase, journal_on_box};
130use self::preflight::{DesignersPolicy, PreflightInput, run_preflights};
131use self::record::{RecordState, RunRecord, TouchedEntry};
132use self::run_store::{EventLog, RunDir, RunId};
133
134pub use self::rollback::{
135 InterruptedRun, RollbackChecked, RollbackContext, RollbackKind, RollbackOptions,
136 RollbackOutcome, check_rollback, restore_set, run_rollback,
137};
138
139const LOCKED_SAMPLE_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(5);
143
144pub const ROLLED_BACK_SLUG: &str = "deploy_rolled_back";
152
153pub const BACKUP_UNSYNCED_SLUG: &str = "deploy_backup_unsynced";
162
163const GWBK_CHECK: &str = "gwbk";
165
166const RESTART_UNAVAILABLE: &str = "restart_pending_unavailable";
169
170const EVENTS_UNWRITABLE: &str = "events_log_unwritable";
173
174#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize)]
176#[serde(rename_all = "snake_case")]
177pub enum OutcomeKind {
178 Applied,
180 Noop,
182 RolledBack,
184 AppliedBackupUnsynced,
188}
189
190impl OutcomeKind {
191 pub fn as_str(self) -> &'static str {
193 match self {
194 OutcomeKind::Applied => "applied",
195 OutcomeKind::Noop => "noop",
196 OutcomeKind::RolledBack => "rolled_back",
197 OutcomeKind::AppliedBackupUnsynced => "applied_backup_unsynced",
198 }
199 }
200}
201
202#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize)]
204#[serde(rename_all = "snake_case")]
205pub enum RollbackCause {
206 LockExpired,
208 WriteFailed,
210 ScanIncomplete,
212 VerifyFailed,
214}
215
216impl RollbackCause {
217 pub fn as_str(self) -> &'static str {
219 match self {
220 RollbackCause::LockExpired => "lock_expired",
221 RollbackCause::WriteFailed => "write_failed",
222 RollbackCause::ScanIncomplete => "scan_incomplete",
223 RollbackCause::VerifyFailed => "verify_failed",
224 }
225 }
226}
227
228#[derive(Debug, Clone, PartialEq, Eq)]
230pub struct WriteSet {
231 pub writes: Vec<WriteOp>,
233 pub deletes: Vec<DeleteOp>,
235}
236
237pub fn write_set(plan: &DeployPlanResult, declared: &Declared) -> Result<WriteSet, CoreError> {
249 let defect = |why: String| CoreError::Internal(format!("write set: {why}"));
250 let mut writes = Vec::new();
251 let mut live: BTreeMap<&str, (bool, Option<&str>)> = BTreeMap::new();
253 for entry in &plan.entries {
254 let in_mode = entry.live_in.iter().any(|c| c == "mode");
255 for file in &entry.files {
256 live.insert(file.path.as_str(), (in_mode, file.live_sha256.as_deref()));
257 }
258 if !matches!(entry.status, PlanStatus::Added | PlanStatus::Changed) {
259 continue;
260 }
261 for file in &entry.files {
262 let Some(desired) = &file.desired_sha256 else {
263 continue;
264 };
265 let path = RelPath::new(&file.path)
266 .map_err(|why| defect(format!("the plan names an invalid path ({why})")))?;
267 let bytes =
268 declared.outcome.tree.files.get(&path).ok_or_else(|| {
269 defect(format!("{path} is in the plan but not in the render"))
270 })?;
271 if sha256_hex(bytes) != *desired {
272 return Err(defect(format!(
273 "{path} was rendered differently from the plan"
274 )));
275 }
276 writes.push(WriteOp {
277 path,
278 sha256: desired.clone(),
279 pre: if in_mode {
280 file.live_sha256.clone()
281 } else {
282 None
283 },
284 });
285 }
286 }
287 let mut deletes = Vec::new();
288 for path in &plan.delete_set.files {
289 let Some((true, Some(pre))) = live.get(path.as_str()) else {
290 return Err(defect(format!(
291 "{path} is in the delete set without a pre-image in the plan"
292 )));
293 };
294 deletes.push(DeleteOp {
295 path: path.clone(),
296 pre: (*pre).to_string(),
297 });
298 }
299 writes.sort_by(|a, b| a.path.as_str().as_bytes().cmp(b.path.as_str().as_bytes()));
300 deletes.sort_by(|a, b| a.path.as_str().as_bytes().cmp(b.path.as_str().as_bytes()));
301 Ok(WriteSet { writes, deletes })
302}
303
304fn new_ledger(
307 plan: &DeployPlanResult,
308 old: Option<&Ledger>,
309 set: &WriteSet,
310 run_id: &RunId,
311) -> Ledger {
312 let mut files: BTreeSet<String> = old
313 .map(|ledger| ledger.files.iter().cloned().collect())
314 .unwrap_or_default();
315 for op in &set.deletes {
316 files.remove(op.path.as_str());
317 }
318 for op in &set.writes {
319 files.insert(op.path.as_str().to_string());
320 }
321 Ledger {
322 schema_version: 1,
323 gateway: plan.gateway.clone(),
324 mode: plan.mode.clone(),
325 render_sha256: plan.render_sha256.clone(),
326 run_id: run_id.as_str().to_string(),
327 files: files.into_iter().collect(),
328 }
329}
330
331fn commit_budget_s(hold: &HoldModel) -> u32 {
338 let seconds = hold.hold_s.saturating_sub(LOCK_MARGIN_S as u64);
339 u32::try_from(seconds.clamp(1, u64::from(BUDGET_MAX_S))).unwrap_or(BUDGET_MAX_S)
340}
341
342fn batch_spec(
345 plan: &DeployPlanResult,
346 run_id: &RunId,
347 set: &WriteSet,
348 ledger_new: Vec<u8>,
349 ledger_pre: Option<String>,
350 commit_budget_s: u32,
351) -> BatchSpec {
352 let mut touched: Vec<TouchedEntry> = set
353 .writes
354 .iter()
355 .map(|w| TouchedEntry {
356 path: w.path.as_str().to_string(),
357 pre_sha256: w.pre.clone(),
358 post_sha256: Some(w.sha256.clone()),
359 })
360 .chain(set.deletes.iter().map(|d| TouchedEntry {
361 path: d.path.as_str().to_string(),
362 pre_sha256: Some(d.pre.clone()),
363 post_sha256: None,
364 }))
365 .collect();
366 touched.push(TouchedEntry {
367 path: LEDGER_KEY.to_string(),
368 pre_sha256: ledger_pre.clone(),
369 post_sha256: Some(sha256_hex(&ledger_new)),
370 });
371 touched.sort_by(|a, b| a.path.as_bytes().cmp(b.path.as_bytes()));
372 let pre_fingerprint = fingerprint(
373 &touched
374 .iter()
375 .map(|e| (e.path.clone(), e.pre_sha256.clone()))
376 .collect(),
377 );
378 let run = run_id.as_str().to_string();
379 BatchSpec {
380 run_id: run_id.clone(),
381 gateway: plan.gateway.clone(),
382 mode: plan.mode.clone(),
383 writes: set.writes.clone(),
384 deletes: set.deletes.clone(),
385 ledger_new: Some(ledger_new),
386 ledger_pre,
387 commit_budget_s: Some(commit_budget_s),
388 journal: Journal {
389 schema_version: 1,
390 run_id: run.clone(),
391 gateway: plan.gateway.clone(),
392 mode: plan.mode.clone(),
393 phase: JournalPhase::Snapshotting,
394 pre_fingerprint: pre_fingerprint.clone(),
395 touched: touched.iter().map(|e| e.path.clone()).collect(),
396 },
397 record: RunRecord {
398 schema_version: 1,
399 run_id: run,
400 gateway: plan.gateway.clone(),
401 mode: plan.mode.clone(),
402 render_sha256: plan.render_sha256.clone(),
403 plan_sha256: plan.plan_sha256.clone(),
404 state: RecordState::Applying,
405 rolled_back_by: None,
406 touched,
407 pre_fingerprint,
408 post_fingerprint: None,
409 },
410 }
411}
412
413#[derive(Debug, Clone, PartialEq, Eq)]
415pub enum ApplyEvent {
416 RunStarted {
418 run_id: String,
420 writes: usize,
422 deletes: usize,
424 },
425 Staged {
427 files: usize,
429 },
430 LockAcquired {
432 hold_s: u64,
434 },
435 Committed {
437 moved: usize,
439 deleted: usize,
441 },
442 LockVerdict {
444 held: bool,
446 },
447 ScanDone,
449 VerifyDone {
451 ok: bool,
453 },
454 Finalized,
456 CommitInterrupted {
460 code: String,
462 },
463 VerifyChecks {
465 checks: Vec<(String, String)>,
467 },
468 RestoreStarted {
471 cause: Option<RollbackCause>,
473 },
474 LiveChangesDiscarded {
477 count: usize,
479 paths: Vec<String>,
481 },
482 Restored,
484 Cleared,
487 InterruptedRun {
491 phase: &'static str,
493 action: &'static str,
495 hashes: BTreeMap<String, Option<String>>,
497 },
498 PreflightChecks {
500 checks: Vec<(String, String)>,
502 },
503 DesignersPoll {
505 open: usize,
507 },
508 DesignersProceed {
510 sessions: Vec<preflight::DesignerSeen>,
512 },
513 GwbkDownloaded {
515 bytes: u64,
517 },
518 GwbkSkipped,
520 BackupSampled {
524 last_sync: Option<i64>,
526 },
527 BackupWaitStarted {
529 budget_s: u64,
531 },
532 BackupPoll {
534 peer_connected: bool,
536 sync_pending: bool,
538 advanced: bool,
540 },
541 BackupResync {
543 ok: bool,
545 code: Option<String>,
547 },
548 BackupVerdict {
550 converged: bool,
552 reason: Option<&'static str>,
554 },
555 Failed {
557 code: String,
559 },
560}
561
562impl ApplyEvent {
563 pub fn name(&self) -> &'static str {
565 match self {
566 ApplyEvent::RunStarted { .. } => "run_started",
567 ApplyEvent::Staged { .. } => "staged",
568 ApplyEvent::LockAcquired { .. } => "lock_acquired",
569 ApplyEvent::Committed { .. } => "committed",
570 ApplyEvent::LockVerdict { .. } => "lock_verdict",
571 ApplyEvent::ScanDone => "scan_done",
572 ApplyEvent::VerifyDone { .. } => "verify_done",
573 ApplyEvent::VerifyChecks { .. } => "verify_checks",
574 ApplyEvent::Finalized => "finalized",
575 ApplyEvent::CommitInterrupted { .. } => "commit_interrupted",
576 ApplyEvent::RestoreStarted { .. } => "restore_started",
577 ApplyEvent::Restored => "restored",
578 ApplyEvent::Cleared => "cleared",
579 ApplyEvent::InterruptedRun { .. } => "interrupted_run",
580 ApplyEvent::LiveChangesDiscarded { .. } => "live_changes_discarded",
581 ApplyEvent::PreflightChecks { .. } => "preflight_checks",
582 ApplyEvent::DesignersPoll { .. } => "designers_poll",
583 ApplyEvent::DesignersProceed { .. } => "designers_proceed",
584 ApplyEvent::GwbkDownloaded { .. } => "gwbk_downloaded",
585 ApplyEvent::GwbkSkipped => "gwbk_skipped",
586 ApplyEvent::BackupSampled { .. } => "backup_sampled",
587 ApplyEvent::BackupWaitStarted { .. } => "backup_wait_started",
588 ApplyEvent::BackupPoll { .. } => "backup_poll",
589 ApplyEvent::BackupResync { .. } => "backup_resync",
590 ApplyEvent::BackupVerdict { .. } => "backup_verdict",
591 ApplyEvent::Failed { .. } => "failed",
592 }
593 }
594
595 pub fn fields(&self) -> Value {
597 match self {
598 ApplyEvent::RunStarted {
599 run_id,
600 writes,
601 deletes,
602 } => json!({"run_id": run_id, "writes": writes, "deletes": deletes}),
603 ApplyEvent::Staged { files } => json!({"files": files}),
604 ApplyEvent::LockAcquired { hold_s } => json!({"hold_s": hold_s}),
605 ApplyEvent::Committed { moved, deleted } => {
606 json!({"moved": moved, "deleted": deleted})
607 }
608 ApplyEvent::LockVerdict { held } => json!({"held": held}),
609 ApplyEvent::VerifyDone { ok } => json!({"ok": ok}),
610 ApplyEvent::VerifyChecks { checks } => json!({
611 "checks": checks
612 .iter()
613 .map(|(id, status)| json!({"id": id, "status": status}))
614 .collect::<Vec<_>>(),
615 }),
616 ApplyEvent::PreflightChecks { checks } => json!({
617 "checks": checks
618 .iter()
619 .map(|(id, status)| json!({"id": id, "status": status}))
620 .collect::<Vec<_>>(),
621 }),
622 ApplyEvent::DesignersPoll { open } => json!({"open": open}),
623 ApplyEvent::DesignersProceed { sessions } => json!({
624 "sessions": sessions
625 .iter()
626 .map(|(user, project, host)| json!({"user": user, "project": project, "host": host}))
627 .collect::<Vec<_>>(),
628 }),
629 ApplyEvent::RestoreStarted { cause } => {
630 json!({"cause": cause.map(RollbackCause::as_str)})
631 }
632 ApplyEvent::LiveChangesDiscarded { count, paths } => {
633 json!({"count": count, "paths": paths})
634 }
635 ApplyEvent::InterruptedRun {
636 phase,
637 action,
638 hashes,
639 } => json!({
640 "phase": phase,
641 "action": action,
642 "touched": hashes.len(),
643 "hashes": hashes,
644 }),
645 ApplyEvent::GwbkDownloaded { bytes } => json!({"bytes": bytes}),
646 ApplyEvent::BackupSampled { last_sync } => json!({"last_sync": last_sync}),
647 ApplyEvent::BackupWaitStarted { budget_s } => json!({"budget_s": budget_s}),
648 ApplyEvent::BackupPoll {
649 peer_connected,
650 sync_pending,
651 advanced,
652 } => json!({
653 "peer_connected": peer_connected,
654 "sync_pending": sync_pending,
655 "advanced": advanced,
656 }),
657 ApplyEvent::BackupResync { ok, code } => json!({"ok": ok, "code": code}),
658 ApplyEvent::BackupVerdict { converged, reason } => {
659 json!({"converged": converged, "reason": reason})
660 }
661 ApplyEvent::CommitInterrupted { code } | ApplyEvent::Failed { code } => {
662 json!({"code": code})
663 }
664 ApplyEvent::ScanDone
665 | ApplyEvent::Finalized
666 | ApplyEvent::Restored
667 | ApplyEvent::Cleared
668 | ApplyEvent::GwbkSkipped => Value::Null,
669 }
670 }
671}
672
673pub trait ApplyObserver: Send + Sync {
675 fn event(&self, e: &ApplyEvent);
677}
678
679#[derive(Debug, Clone, Copy, Default)]
681pub struct NoObserver;
682
683impl ApplyObserver for NoObserver {
684 fn event(&self, _e: &ApplyEvent) {}
685}
686
687pub struct ApplyContext<'a> {
689 pub declared: &'a Declared,
691 pub repo: &'a std::path::Path,
693 pub reader: &'a dyn TreeReader,
695 pub host: &'a dyn ApplyHost,
697 pub deploy: &'a dyn DeployApi,
699 pub gateway: &'a dyn GatewayApi,
701 pub backup: Option<&'a BackupNode<'a>>,
704 pub clock: &'a dyn Clock,
706 pub observer: &'a dyn ApplyObserver,
708}
709
710impl<'a> ApplyContext<'a> {
711 fn restore_context(&'a self, opts: &'a ApplyOptions) -> RestoreContext<'a> {
713 RestoreContext {
714 env: &self.declared.env,
715 master: &self.declared.target.nodes.master,
716 host: self.host,
717 deploy: self.deploy,
718 clock: self.clock,
719 hold: &opts.hold,
720 poll: &opts.poll,
721 }
722 }
723}
724
725#[derive(Debug, Clone)]
727pub struct ApplyOptions {
728 pub base: Option<PathBuf>,
730 pub gates: GateFlags,
732 pub project: String,
734 pub hold: HoldModel,
736 pub poll: PollConfig,
738 pub verify: VerifyOptions,
740 pub designers: DesignersPolicy,
742 pub skip_gwbk: bool,
744 pub backup: BackupPolicy,
747}
748
749#[derive(Debug, Clone, PartialEq, Serialize)]
751pub struct ApplyOutcome {
752 #[serde(rename = "outcome")]
754 pub kind: OutcomeKind,
755 pub run_id: Option<String>,
757 pub run_dir: Option<String>,
759 pub cause: Option<RollbackCause>,
761 pub render_sha256: String,
763 pub plan_sha256: String,
765 pub written: Vec<String>,
767 pub deleted: Vec<String>,
769 pub post_fingerprint: Option<String>,
771 pub warnings: Vec<String>,
773 pub gwbk: Option<String>,
776 pub skip_gwbk: bool,
778 pub restart_pending: Vec<String>,
780 pub restart_pending_new: Vec<String>,
782 pub snapshot_sha256: BTreeMap<String, String>,
786 pub backup: Option<BackupReport>,
789}
790
791pub async fn run_apply(
799 ctx: &ApplyContext<'_>,
800 plan: &DeployPlanResult,
801 opts: &ApplyOptions,
802) -> Result<ApplyOutcome, CoreError> {
803 if !plan.summary.has_changes {
804 return Ok(ApplyOutcome {
805 kind: OutcomeKind::Noop,
806 run_id: None,
807 run_dir: None,
808 cause: None,
809 render_sha256: plan.render_sha256.clone(),
810 plan_sha256: plan.plan_sha256.clone(),
811 written: Vec::new(),
812 deleted: Vec::new(),
813 post_fingerprint: None,
814 warnings: Vec::new(),
815 gwbk: None,
816 skip_gwbk: false,
817 restart_pending: Vec::new(),
818 restart_pending_new: Vec::new(),
819 snapshot_sha256: BTreeMap::new(),
820 backup: None,
821 });
822 }
823 let (run_id, run_dir) = RunDir::create_fresh(ctx.repo, &plan.gateway)?;
824 let mut steps = Steps::new(None, ctx.observer);
825 let result = drive(ctx, plan, opts, &run_id, &run_dir, &mut steps).await;
826
827 if let Err(e) = &result {
829 let _ = steps.emit(ApplyEvent::Failed {
830 code: e.code().to_string(),
831 });
832 }
833 let doc = match &result {
834 Ok(outcome) => {
835 let mut doc = serde_json::to_value(outcome).unwrap_or(Value::Null);
836 doc["ok"] = Value::Bool(!matches!(
837 outcome.kind,
838 OutcomeKind::RolledBack | OutcomeKind::AppliedBackupUnsynced
839 ));
840 match outcome.kind {
841 OutcomeKind::RolledBack => {
842 doc["slug"] = Value::String(ROLLED_BACK_SLUG.to_string());
843 }
844 OutcomeKind::AppliedBackupUnsynced => {
845 doc["slug"] = Value::String(BACKUP_UNSYNCED_SLUG.to_string());
846 }
847 OutcomeKind::Applied | OutcomeKind::Noop => {}
848 }
849 doc
850 }
851 Err(e) => json!({
852 "ok": false,
853 "run_id": run_id.as_str(),
854 "error": {"code": e.code(), "message": e.to_string()},
855 }),
856 };
857 let written = run_dir.write_json("result.json", &doc);
858 match result {
859 Ok(mut outcome) => {
860 if written.is_err() {
861 outcome.warnings.push("result_json_unwritable".to_string());
862 }
863 Ok(outcome)
864 }
865 Err(e) => Err(e),
866 }
867}
868
869pub struct Steps<'a> {
872 log: Option<EventLog>,
873 observer: &'a dyn ApplyObserver,
874}
875
876impl<'a> Steps<'a> {
877 pub fn new(log: Option<EventLog>, observer: &'a dyn ApplyObserver) -> Self {
879 Steps { log, observer }
880 }
881
882 pub fn emit(&mut self, event: ApplyEvent) -> Result<(), CoreError> {
884 self.observer.event(&event);
885 match &mut self.log {
886 Some(log) => log.append(event.name(), event.fields()),
887 None => Ok(()),
888 }
889 }
890}
891
892fn with_env(err: CoreError, env: &str) -> CoreError {
894 match err {
895 CoreError::DeployJournalPresent {
896 run_id, location, ..
897 } => CoreError::DeployJournalPresent {
898 run_id,
899 env: env.to_string(),
900 location,
901 },
902 other => other,
903 }
904}
905
906enum BoxState {
909 Mine(Journal),
911 Other,
913 Absent,
915 Unknown,
917}
918
919async fn box_state(ctx: &ApplyContext<'_>, plan: &DeployPlanResult, run_id: &RunId) -> BoxState {
920 let master = &ctx.declared.target.nodes.master;
921 match journal_on_box(ctx.reader, master, &plan.gateway, &plan.mode).await {
922 Ok(Some(journal)) if journal.run_id == run_id.as_str() => BoxState::Mine(journal),
923 Ok(Some(_)) => BoxState::Other,
924 Ok(None) => BoxState::Absent,
925 Err(_) => BoxState::Unknown,
926 }
927}
928
929fn interrupted(
933 ctx: &ApplyContext<'_>,
934 plan: &DeployPlanResult,
935 steps: &mut Steps<'_>,
936 journal: &Journal,
937 err: &CoreError,
938) -> CoreError {
939 let _ = steps.emit(ApplyEvent::CommitInterrupted {
940 code: err.code().to_string(),
941 });
942 journal::journal_present(
943 journal,
944 &ctx.declared.env,
945 &ctx.declared.target.nodes.master,
946 &plan.gateway,
947 &plan.mode,
948 )
949}
950
951async fn commit_failed(
967 ctx: &ApplyContext<'_>,
968 run_id: &RunId,
969 plan: &DeployPlanResult,
970 steps: &mut Steps<'_>,
971 err: CoreError,
972) -> CoreError {
973 if matches!(err, CoreError::DeployJournalPresent { .. }) {
974 return err;
975 }
976 match box_state(ctx, plan, run_id).await {
977 BoxState::Mine(journal) => interrupted(ctx, plan, steps, &journal, &err),
978 BoxState::Unknown => {
979 let _ = steps.emit(ApplyEvent::CommitInterrupted {
983 code: format!("{}; journal state unknown", err.code()),
984 });
985 recovery_pointer(ctx, plan, run_id, err)
986 }
987 BoxState::Other | BoxState::Absent => err,
988 }
989}
990
991async fn unfinished_after_green(
998 ctx: &ApplyContext<'_>,
999 run_id: &RunId,
1000 plan: &DeployPlanResult,
1001 steps: &mut Steps<'_>,
1002 err: CoreError,
1003) -> CoreError {
1004 match box_state(ctx, plan, run_id).await {
1005 BoxState::Mine(journal) => interrupted(ctx, plan, steps, &journal, &err),
1006 _ => recovery_pointer(ctx, plan, run_id, err),
1007 }
1008}
1009
1010fn recovery_pointer(
1015 ctx: &ApplyContext<'_>,
1016 plan: &DeployPlanResult,
1017 run_id: &RunId,
1018 err: CoreError,
1019) -> CoreError {
1020 let env = &ctx.declared.env;
1021 match err {
1022 CoreError::TransportFailed {
1023 operation,
1024 node,
1025 detail,
1026 } => CoreError::TransportFailed {
1027 operation,
1028 node,
1029 detail: format!(
1030 "{detail}; this run's journal may be on the host: recover it with: \
1031 ign deploy rollback --run {run} --env {env}",
1032 run = run_id.as_str()
1033 ),
1034 },
1035 _ => CoreError::DeployJournalPresent {
1036 run_id: run_id.as_str().to_string(),
1037 env: env.clone(),
1038 location: journal_location(
1039 LedgerRoot::for_node(&ctx.declared.target.nodes.master),
1040 &plan.gateway,
1041 &plan.mode,
1042 ),
1043 },
1044 }
1045}
1046
1047async fn drive(
1051 ctx: &ApplyContext<'_>,
1052 plan: &DeployPlanResult,
1053 opts: &ApplyOptions,
1054 run_id: &RunId,
1055 run_dir: &RunDir,
1056 steps: &mut Steps<'_>,
1057) -> Result<ApplyOutcome, CoreError> {
1058 let declared = ctx.declared;
1059 let env = declared.env.as_str();
1060 let (gateway, mode) = (plan.gateway.as_str(), plan.mode.as_str());
1061
1062 let started_at = rfc3339_now_utc();
1063 steps.log = Some(EventLog::open(run_dir)?);
1064
1065 let set = write_set(plan, declared)?;
1066 steps.emit(ApplyEvent::RunStarted {
1067 run_id: run_id.as_str().to_string(),
1068 writes: set.writes.len(),
1069 deletes: set.deletes.len(),
1070 })?;
1071
1072 let touched_paths: Vec<String> = set
1076 .writes
1077 .iter()
1078 .map(|w| w.path.as_str().to_string())
1079 .chain(set.deletes.iter().map(|d| d.path.as_str().to_string()))
1080 .collect();
1081 let preflights = run_preflights(
1082 &PreflightInput {
1083 target: &declared.target,
1084 declared: Some(declared),
1085 master: ctx.deploy,
1086 backup: ctx.backup,
1087 gateway: ctx.gateway,
1088 reader: ctx.reader,
1089 host: ctx.host,
1090 touched_files: set.writes.len() + set.deletes.len(),
1091 touched_paths: &touched_paths,
1092 hold: &opts.hold,
1093 designers: &opts.designers,
1094 clock: ctx.clock,
1095 },
1096 steps,
1097 )
1098 .await;
1099 let seen = preflights.as_ref().ok();
1100 run_dir.write_json(
1101 "run.json",
1102 &json!({
1103 "schema_version": 1,
1104 "run_id": run_id.as_str(),
1105 "env": env,
1106 "gateway": gateway,
1107 "mode": mode,
1108 "node": plan.node,
1109 "transport": plan.transport,
1110 "started_at": started_at,
1111 "run_dir_git": run_dir.git_status().as_record(),
1112 "render_sha256": plan.render_sha256,
1113 "plan_sha256": plan.plan_sha256,
1114 "hold_s": opts.hold.hold_s,
1115 "allow_destructive": opts.gates.allow_destructive,
1116 "allow_drift": opts.gates.allow_drift,
1117 "skip_gwbk": opts.skip_gwbk,
1118 "designers": opts.designers.mode.as_str(),
1119 "designers_timeout_s": opts.designers.timeout.as_secs(),
1120 "designers_seen": seen.map(|r| r.designers_seen
1121 .iter()
1122 .map(|(user, project, host)| json!({"user": user, "project": project, "host": host}))
1123 .collect::<Vec<_>>()),
1124 "preflights": seen.map(|r| r.findings
1125 .iter()
1126 .map(|f| json!({"check": f.check.as_str(), "status": f.status.as_str()}))
1127 .collect::<Vec<_>>()),
1128 }),
1129 )?;
1130 run_dir.write_json("plan.json", plan)?;
1131 let preflights = preflights?;
1132 if !preflights.passed() {
1133 steps.emit(ApplyEvent::PreflightChecks {
1134 checks: preflights
1135 .findings
1136 .iter()
1137 .map(|f| (f.check.as_str().to_string(), f.status.as_str().to_string()))
1138 .collect(),
1139 })?;
1140 return Err(CoreError::DeployPreflightFailed {
1141 failures: preflights.failures().into(),
1142 });
1143 }
1144
1145 let mut backup_before = match ctx.backup {
1151 Some(backup) => {
1152 let sampled = backup.status().await?.last_sync_epoch_ms();
1153 steps.emit(ApplyEvent::BackupSampled { last_sync: sampled })?;
1154 sampled
1155 }
1156 None => None,
1157 };
1158 let mut backup_log_since_ms = wall_clock_ms();
1162
1163 let gwbk = take_gwbk(ctx, opts, run_id, run_dir, steps).await?;
1166
1167 let restart_before = ctx.deploy.restart_pending().await.ok();
1171
1172 let old_bytes = ctx.reader.read_ledger(gateway, mode).await?;
1174 let old_ledger = match &old_bytes {
1175 Some(bytes) => Some(Ledger::parse(
1176 &ledger_location(
1177 LedgerRoot::for_node(&declared.target.nodes.master),
1178 gateway,
1179 mode,
1180 ),
1181 bytes,
1182 gateway,
1183 mode,
1184 )?),
1185 None => None,
1186 };
1187 let ledger = new_ledger(plan, old_ledger.as_ref(), &set, run_id);
1188 let ledger_bytes = ledger.to_bytes();
1189 Ledger::parse(
1190 &ledger_location(
1191 LedgerRoot::for_node(&declared.target.nodes.master),
1192 gateway,
1193 mode,
1194 ),
1195 &ledger_bytes,
1196 gateway,
1197 mode,
1198 )?;
1199 let spec = batch_spec(
1200 plan,
1201 run_id,
1202 &set,
1203 ledger_bytes,
1204 old_bytes.as_deref().map(sha256_hex),
1205 commit_budget_s(&opts.hold),
1206 );
1207 if let Some(problem) = spec.problem() {
1208 return Err(CoreError::Internal(format!(
1209 "the batch the engine built is invalid: {problem}"
1210 )));
1211 }
1212
1213 let scope = RunScope {
1214 plan,
1215 run_id,
1216 run_dir,
1217 gwbk: gwbk.as_deref(),
1218 restart_before: restart_before.as_deref(),
1219 record: &spec.record,
1220 };
1221
1222 let prepared = ctx
1224 .host
1225 .prepare(&spec, &declared.outcome.tree)
1226 .await
1227 .map_err(|e| with_env(e, env))?;
1228 steps.emit(ApplyEvent::Staged {
1229 files: set.writes.len(),
1230 })?;
1231 let lock =
1232 match HeldLock::acquire(ctx.deploy, ScanKind::Config, opts.hold.hold_s, ctx.clock).await {
1233 Ok(lock) => lock,
1234 Err(refused) => {
1235 let _ = ctx.host.discard_interrupted(run_id).await;
1239 return Err(refused);
1240 }
1241 };
1242 steps.emit(ApplyEvent::LockAcquired {
1243 hold_s: opts.hold.hold_s,
1244 })?;
1245 if let Some(backup) = ctx.backup {
1257 let since_ms = wall_clock_ms();
1258 if let Ok(Ok(status)) = tokio::time::timeout(LOCKED_SAMPLE_TIMEOUT, backup.status()).await {
1259 let sampled = status.last_sync_epoch_ms();
1260 steps.emit(ApplyEvent::BackupSampled { last_sync: sampled })?;
1261 backup_before = sampled;
1262 backup_log_since_ms = since_ms;
1263 }
1264 }
1265 let report = match ctx.host.commit(&prepared).await {
1266 Ok(report) => report,
1267 Err(e) => return Err(commit_failed(ctx, run_id, plan, steps, with_env(e, env)).await),
1268 };
1269 if let Err(e) = steps.emit(ApplyEvent::Committed {
1270 moved: report.moved.len(),
1271 deleted: report.deleted.len(),
1272 }) {
1273 return Err(commit_failed(ctx, run_id, plan, steps, e).await);
1276 }
1277 if report.stale {
1278 return Err(plan_stale(ctx, plan, opts, run_id).await);
1279 }
1280 if report.failure.is_some() {
1281 let cause = if report.deadline {
1286 RollbackCause::LockExpired
1287 } else {
1288 RollbackCause::WriteFailed
1289 };
1290 return roll_back(ctx, opts, &scope, Some(&lock), cause, steps, Vec::new()).await;
1291 }
1292
1293 let mut warnings = match confirm(ctx, opts, &lock, steps).await {
1296 Ok(Confirmed::Green { warnings }) => warnings,
1297 Ok(Confirmed::Abort { cause, warnings }) => {
1298 return roll_back(ctx, opts, &scope, Some(&lock), cause, steps, warnings).await;
1299 }
1300 Err(Unconfirmed::Restore { cause, step, error }) => {
1301 let note = format!("post_commit_error ({step}: {})", error.code());
1304 return roll_back(ctx, opts, &scope, Some(&lock), cause, steps, vec![note]).await;
1305 }
1306 Err(Unconfirmed::Green(error)) => {
1307 return Err(unfinished_after_green(ctx, run_id, plan, steps, error).await);
1308 }
1309 };
1310
1311 let post = match ctx.host.fingerprint(&spec.touched_paths()).await {
1313 Ok(post) => post,
1314 Err(e) => return Err(unfinished_after_green(ctx, run_id, plan, steps, e).await),
1315 };
1316 let post_fingerprint = fingerprint(&post);
1317 let mut record = spec.record.clone();
1318 record.state = RecordState::Applied;
1319 record.post_fingerprint = Some(post_fingerprint.clone());
1320 if let Err(e) = ctx.host.finalize(&record).await {
1321 if !matches!(box_state(ctx, plan, run_id).await, BoxState::Absent) {
1324 return Err(unfinished_after_green(ctx, run_id, plan, steps, e).await);
1325 }
1326 warnings.push(format!("finalize_reported_error ({})", e.code()));
1327 }
1328 if steps.emit(ApplyEvent::Finalized).is_err() {
1329 warnings.push(EVENTS_UNWRITABLE.to_string());
1330 }
1331
1332 let (restart_pending, restart_pending_new) =
1333 restart_report(ctx.deploy, restart_before.as_deref(), &mut warnings).await;
1334 let mut snapshot_sha256 = BTreeMap::new();
1335 match ctx.host.fetch_snapshot(run_id, run_dir.path()).await {
1336 Ok(hashes) => {
1337 warnings.extend(snapshot_warnings(&spec.record, &hashes));
1338 snapshot_sha256 = hashes;
1339 }
1340 Err(e) => warnings.push(format!("snapshot_download_failed ({})", e.code())),
1341 }
1342
1343 let backup_report = match ctx.backup {
1346 Some(backup) => Some(
1347 converge::converge_backup(
1348 ctx,
1349 backup,
1350 opts,
1351 &set,
1352 converge::Baseline {
1353 last_sync: backup_before,
1354 log_since_ms: backup_log_since_ms,
1355 },
1356 steps,
1357 &mut warnings,
1358 )
1359 .await,
1360 ),
1361 None => None,
1362 };
1363 let kind = match &backup_report {
1364 Some(report) if !report.converged => OutcomeKind::AppliedBackupUnsynced,
1365 _ => OutcomeKind::Applied,
1366 };
1367 Ok(ApplyOutcome {
1368 kind,
1369 run_id: Some(run_id.as_str().to_string()),
1370 run_dir: Some(run_dir.path().to_string_lossy().into_owned()),
1371 cause: None,
1372 render_sha256: plan.render_sha256.clone(),
1373 plan_sha256: plan.plan_sha256.clone(),
1374 written: set
1375 .writes
1376 .iter()
1377 .map(|w| w.path.as_str().to_string())
1378 .collect(),
1379 deleted: set
1380 .deletes
1381 .iter()
1382 .map(|d| d.path.as_str().to_string())
1383 .collect(),
1384 post_fingerprint: Some(post_fingerprint),
1385 warnings,
1386 gwbk,
1387 skip_gwbk: opts.skip_gwbk,
1388 restart_pending,
1389 restart_pending_new,
1390 snapshot_sha256,
1391 backup: backup_report,
1392 })
1393}
1394
1395enum Confirmed {
1398 Green { warnings: Vec<String> },
1400 Abort {
1402 cause: RollbackCause,
1403 warnings: Vec<String>,
1404 },
1405}
1406
1407enum Unconfirmed {
1409 Restore {
1413 cause: RollbackCause,
1414 step: &'static str,
1415 error: CoreError,
1416 },
1417 Green(CoreError),
1419}
1420
1421async fn confirm(
1428 ctx: &ApplyContext<'_>,
1429 opts: &ApplyOptions,
1430 lock: &HeldLock,
1431 steps: &mut Steps<'_>,
1432) -> Result<Confirmed, Unconfirmed> {
1433 fn undone(cause: RollbackCause, step: &'static str) -> impl FnOnce(CoreError) -> Unconfirmed {
1434 move |error| Unconfirmed::Restore { cause, step, error }
1435 }
1436 let abort = |cause, warnings| Ok(Confirmed::Abort { cause, warnings });
1437
1438 let verdict = lock
1440 .check(ctx.deploy, LOCK_MARGIN_S, ctx.clock)
1441 .await
1442 .map_err(undone(RollbackCause::LockExpired, "lock_check"))?;
1443 let held = matches!(verdict, LockVerdict::Held { .. });
1444 steps
1445 .emit(ApplyEvent::LockVerdict { held })
1446 .map_err(undone(RollbackCause::LockExpired, "events"))?;
1447 if !held {
1448 return abort(RollbackCause::LockExpired, Vec::new());
1450 }
1451 match scan_and_wait(ctx.deploy, ScanKind::Config, opts.poll.clone())
1453 .await
1454 .map_err(undone(RollbackCause::ScanIncomplete, "scan"))?
1455 {
1456 ScanOutcome::Complete(_) => steps
1457 .emit(ApplyEvent::ScanDone)
1458 .map_err(undone(RollbackCause::ScanIncomplete, "events"))?,
1459 ScanOutcome::NotComplete { .. } => {
1460 return abort(RollbackCause::ScanIncomplete, Vec::new());
1461 }
1462 }
1463
1464 let verify = deploy_verify(
1465 ctx.declared,
1466 &opts.project,
1467 ctx.deploy,
1468 ctx.gateway,
1469 ctx.reader,
1470 &opts.verify,
1471 )
1472 .await
1473 .map_err(undone(RollbackCause::VerifyFailed, "verify"))?;
1474 if !verify.ok {
1475 steps
1476 .emit(ApplyEvent::VerifyDone { ok: false })
1477 .map_err(undone(RollbackCause::VerifyFailed, "events"))?;
1478 steps
1480 .emit(ApplyEvent::VerifyChecks {
1481 checks: verify
1482 .checks
1483 .iter()
1484 .map(|c| (c.id.word().to_string(), c.status.word().to_string()))
1485 .collect(),
1486 })
1487 .map_err(undone(RollbackCause::VerifyFailed, "events"))?;
1488 return abort(RollbackCause::VerifyFailed, verify.warnings);
1489 }
1490 steps
1491 .emit(ApplyEvent::VerifyDone { ok: true })
1492 .map_err(Unconfirmed::Green)?;
1493 Ok(Confirmed::Green {
1494 warnings: verify.warnings,
1495 })
1496}
1497
1498async fn restart_report(
1505 deploy: &dyn DeployApi,
1506 before: Option<&[String]>,
1507 warnings: &mut Vec<String>,
1508) -> (Vec<String>, Vec<String>) {
1509 let after = deploy.restart_pending().await.ok();
1510 if before.is_none() || after.is_none() {
1511 warnings.push(RESTART_UNAVAILABLE.to_string());
1512 }
1513 let pending = after.unwrap_or_default();
1514 let known = before.unwrap_or_default();
1515 let added = pending
1516 .iter()
1517 .filter(|entry| !known.contains(entry))
1518 .cloned()
1519 .collect();
1520 (pending, added)
1521}
1522
1523fn snapshot_warnings(record: &RunRecord, copied: &BTreeMap<String, String>) -> Vec<String> {
1527 record
1528 .pre_map()
1529 .into_iter()
1530 .filter_map(|(path, pre)| {
1531 let pre = pre?;
1532 let key = if path == LEDGER_KEY {
1533 "ledger.json".to_string()
1534 } else {
1535 format!("files/{path}")
1536 };
1537 (copied.get(&key) != Some(&pre)).then(|| format!("snapshot_copy_differs ({path})"))
1538 })
1539 .collect()
1540}
1541
1542async fn take_gwbk(
1559 ctx: &ApplyContext<'_>,
1560 opts: &ApplyOptions,
1561 run_id: &RunId,
1562 run_dir: &RunDir,
1563 steps: &mut Steps<'_>,
1564) -> Result<Option<String>, CoreError> {
1565 if opts.skip_gwbk {
1566 steps.emit(ApplyEvent::GwbkSkipped)?;
1567 return Ok(None);
1568 }
1569 let refuse = |detail: String| CoreError::DeployPreflightFailed {
1570 failures: vec![PreflightFailure {
1571 check: GWBK_CHECK.to_string(),
1572 detail,
1573 }]
1574 .into(),
1575 };
1576 let name = format!("{}.gwbk", run_id.as_str());
1577 drop(run_dir.create_private_file(&name)?);
1578 let path = run_dir.path().join(&name);
1579 let verdict = match backup_download(ctx.gateway, Some(&path), "", BackupType::All).await {
1580 Ok(_) => validate_gwbk(&path),
1581 Err(e) => Err(format!("the .gwbk was not downloaded ({})", e.code())),
1582 };
1583 match verdict {
1584 Ok(bytes) => {
1585 steps.emit(ApplyEvent::GwbkDownloaded { bytes })?;
1586 Ok(Some(name))
1587 }
1588 Err(detail) => {
1589 let _ = std::fs::remove_file(&path);
1590 Err(refuse(detail))
1591 }
1592 }
1593}
1594
1595fn validate_gwbk(path: &Path) -> Result<u64, String> {
1598 let bytes = std::fs::metadata(path)
1599 .map_err(|_| "the downloaded .gwbk cannot be read back".to_string())?
1600 .len();
1601 if bytes == 0 {
1602 return Err("the downloaded .gwbk is empty".to_string());
1603 }
1604 let file = std::fs::File::open(path)
1605 .map_err(|_| "the downloaded .gwbk cannot be read back".to_string())?;
1606 let archive = zip::ZipArchive::new(file).map_err(|_| {
1607 "the downloaded .gwbk is not a readable ZIP (no central directory), so the download is incomplete"
1608 .to_string()
1609 })?;
1610 if archive.is_empty() {
1611 return Err("the downloaded .gwbk is a ZIP with no entries".to_string());
1612 }
1613 Ok(bytes)
1614}
1615
1616struct RunScope<'a> {
1618 plan: &'a DeployPlanResult,
1619 run_id: &'a RunId,
1620 run_dir: &'a RunDir,
1621 gwbk: Option<&'a str>,
1623 restart_before: Option<&'a [String]>,
1625 record: &'a RunRecord,
1628}
1629
1630async fn roll_back(
1633 ctx: &ApplyContext<'_>,
1634 opts: &ApplyOptions,
1635 scope: &RunScope<'_>,
1636 held: Option<&HeldLock>,
1637 cause: RollbackCause,
1638 steps: &mut Steps<'_>,
1639 mut warnings: Vec<String>,
1640) -> Result<ApplyOutcome, CoreError> {
1641 let restored = restore_snapshot(
1642 &ctx.restore_context(opts),
1643 &RestoreInput {
1644 run_id: scope.run_id,
1645 run_dir: scope.run_dir,
1646 record: scope.record,
1647 held,
1648 cause: Some(cause),
1649 },
1650 steps,
1651 )
1652 .await?;
1653 warnings.extend(restored.warnings);
1654 let (restart_pending, restart_pending_new) =
1655 restart_report(ctx.deploy, scope.restart_before, &mut warnings).await;
1656 Ok(ApplyOutcome {
1657 kind: OutcomeKind::RolledBack,
1658 run_id: Some(scope.run_id.as_str().to_string()),
1659 run_dir: Some(scope.run_dir.path().to_string_lossy().into_owned()),
1660 cause: Some(cause),
1661 render_sha256: scope.plan.render_sha256.clone(),
1662 plan_sha256: scope.plan.plan_sha256.clone(),
1663 written: Vec::new(),
1664 deleted: Vec::new(),
1665 post_fingerprint: Some(scope.record.pre_fingerprint.clone()),
1666 warnings,
1667 gwbk: scope.gwbk.map(str::to_string),
1668 skip_gwbk: opts.skip_gwbk,
1669 restart_pending,
1670 restart_pending_new,
1671 snapshot_sha256: restored.snapshot_sha256,
1672 backup: None,
1673 })
1674}
1675
1676pub struct RestoreInput<'a> {
1678 pub run_id: &'a RunId,
1682 pub run_dir: &'a RunDir,
1685 pub record: &'a RunRecord,
1687 pub held: Option<&'a HeldLock>,
1689 pub cause: Option<RollbackCause>,
1692}
1693
1694pub struct RestoreContext<'a> {
1697 pub env: &'a str,
1699 pub master: &'a NodeTarget,
1701 pub host: &'a dyn ApplyHost,
1703 pub deploy: &'a dyn DeployApi,
1705 pub clock: &'a dyn Clock,
1707 pub hold: &'a HoldModel,
1709 pub poll: &'a PollConfig,
1711}
1712
1713#[derive(Debug, Clone, PartialEq, Eq)]
1715pub struct RestoreReport {
1716 pub restored: usize,
1718 pub removed: usize,
1720 pub fresh_lock: bool,
1723 pub warnings: Vec<String>,
1725 pub snapshot_sha256: BTreeMap<String, String>,
1728}
1729
1730pub async fn restore_snapshot(
1754 ctx: &RestoreContext<'_>,
1755 input: &RestoreInput<'_>,
1756 steps: &mut Steps<'_>,
1757) -> Result<RestoreReport, CoreError> {
1758 let record = input.record;
1759 match restore_steps(ctx, input, steps).await {
1760 Ok(report) => Ok(report),
1761 Err(cause) => Err(CoreError::RollbackFailed(Box::new(RollbackFailure {
1762 run_id: record.run_id.clone(),
1763 env: ctx.env.to_string(),
1764 run_dir: Some(input.run_dir.path().to_string_lossy().into_owned()),
1765 journal: journal_location(
1766 LedgerRoot::for_node(ctx.master),
1767 &record.gateway,
1768 &record.mode,
1769 ),
1770 cause,
1771 }))),
1772 }
1773}
1774
1775async fn restore_steps(
1777 ctx: &RestoreContext<'_>,
1778 input: &RestoreInput<'_>,
1779 steps: &mut Steps<'_>,
1780) -> Result<RestoreReport, String> {
1781 let gw = |step: &'static str| move |e: CoreError| format!("{step}: {}", e.code());
1784 let host = |step: &'static str| move |e: CoreError| format!("{step}: {}", with_env(e, ctx.env));
1785 let event = |e: CoreError| format!("events: {}", e.code());
1786
1787 let record = input.record;
1788 let deploy = ctx.deploy;
1789 steps
1790 .emit(ApplyEvent::RestoreStarted { cause: input.cause })
1791 .map_err(event)?;
1792
1793 let need_s = ctx.hold.estimate_commit_s(record.touched.len()).ceil() as i64 + LOCK_MARGIN_S;
1795 let reuse = match input.held {
1796 Some(lock) => match lock
1797 .check_gateway(deploy, need_s)
1798 .await
1799 .map_err(gw("checking the scan-lock"))?
1800 {
1801 LockVerdict::Held { .. } => true,
1802 LockVerdict::Expiring { .. } => {
1803 lock.wait_for_lapse(deploy, ctx.poll.interval)
1804 .await
1805 .map_err(gw("waiting for the scan-lock to lapse"))?;
1806 false
1807 }
1808 LockVerdict::NotHeld => false,
1809 },
1810 None => false,
1811 };
1812 if !reuse {
1813 HeldLock::acquire(deploy, ScanKind::Config, ctx.hold.hold_s, ctx.clock)
1814 .await
1815 .map_err(gw("taking the scan-lock"))?;
1816 steps
1817 .emit(ApplyEvent::LockAcquired {
1818 hold_s: ctx.hold.hold_s,
1819 })
1820 .map_err(event)?;
1821 }
1822
1823 let report = ctx.host.restore(record).await.map_err(host("restore"))?;
1825 if let Some(failure) = &report.failure {
1826 return Err(format!("restore: {failure}"));
1827 }
1828 if report.stale {
1829 return Err("restore: the host reported a stale pre-image".to_string());
1830 }
1831 match scan_and_wait(deploy, ScanKind::Config, ctx.poll.clone())
1832 .await
1833 .map_err(gw("the scan after the restore"))?
1834 {
1835 ScanOutcome::Complete(_) => steps.emit(ApplyEvent::ScanDone).map_err(event)?,
1836 ScanOutcome::NotComplete { .. } => {
1837 return Err("the scan after the restore did not complete".to_string());
1838 }
1839 }
1840
1841 let touched: Vec<String> = record.touched.iter().map(|e| e.path.clone()).collect();
1843 let live = ctx
1844 .host
1845 .fingerprint(&touched)
1846 .await
1847 .map_err(host("reading the restored files"))?;
1848 let pre = record.pre_map();
1849 let differing: Vec<String> = pre
1850 .iter()
1851 .filter(|(path, hash)| live.get(*path) != Some(*hash))
1852 .map(|(path, _)| path.clone())
1853 .collect();
1854 if !differing.is_empty() {
1855 return Err(format!(
1856 "{} restored path(s) differ from the snapshot: {}",
1857 differing.len(),
1858 listed(&differing)
1859 ));
1860 }
1861 let mode = deploy
1862 .deployment_mode()
1863 .await
1864 .map_err(gw("reading the deployment mode"))?;
1865 if mode.as_deref() != Some(record.mode.as_str()) {
1866 return Err("the gateway's deployment mode is not the target's".to_string());
1867 }
1868
1869 let mut rolled_back = record.clone();
1871 rolled_back.state = RecordState::RolledBack;
1872 rolled_back.rolled_back_by = Some(input.run_id.as_str().to_string());
1873 rolled_back.post_fingerprint = Some(record.pre_fingerprint.clone());
1874 ctx.host
1875 .finalize(&rolled_back)
1876 .await
1877 .map_err(host("finalize"))?;
1878 steps.emit(ApplyEvent::Restored).map_err(event)?;
1879
1880 let mut warnings = Vec::new();
1881 let mut snapshot_sha256 = BTreeMap::new();
1882 match RunId::parse(&record.run_id) {
1883 Ok(restored_run) => {
1884 match ctx
1885 .host
1886 .fetch_snapshot(&restored_run, input.run_dir.path())
1887 .await
1888 {
1889 Ok(hashes) => {
1890 warnings.extend(snapshot_warnings(record, &hashes));
1891 snapshot_sha256 = hashes;
1892 }
1893 Err(e) => warnings.push(format!("snapshot_download_failed ({})", e.code())),
1894 }
1895 }
1896 Err(e) => warnings.push(format!("snapshot_download_failed ({})", e.code())),
1897 }
1898 Ok(RestoreReport {
1899 restored: report.moved.len(),
1900 removed: report.deleted.len(),
1901 fresh_lock: !reuse,
1902 warnings,
1903 snapshot_sha256,
1904 })
1905}
1906
1907fn listed(paths: &[String]) -> String {
1909 let shown = paths.len().min(crate::error::APPLY_LIST_SHOWN);
1910 let mut out = paths[..shown].join(", ");
1911 if paths.len() > shown {
1912 out.push_str(&format!(" and {} more", paths.len() - shown));
1913 }
1914 out
1915}
1916
1917async fn plan_stale(
1922 ctx: &ApplyContext<'_>,
1923 plan: &DeployPlanResult,
1924 opts: &ApplyOptions,
1925 run_id: &RunId,
1926) -> CoreError {
1927 let _ = ctx.host.discard_interrupted(run_id).await;
1930 let _ = ctx.deploy.scan_request(ScanKind::Config).await;
1931 let actual = match deploy_plan(
1932 ctx.declared,
1933 ctx.reader,
1934 &PlanOptions {
1935 base: opts.base.clone(),
1936 diff: false,
1937 },
1938 )
1939 .await
1940 {
1941 Ok(now) => now.plan_sha256,
1942 Err(_) => "unavailable".to_string(),
1943 };
1944 CoreError::PlanStale {
1945 expected: plan.plan_sha256.clone(),
1946 actual,
1947 detail: "a live file changed between the plan and the lock; nothing was written"
1948 .to_string(),
1949 }
1950}
1951
1952#[cfg(test)]
1953mod tests {
1954 use super::*;
1955
1956 fn record(touched: &[(&str, Option<&str>)]) -> RunRecord {
1957 let touched: Vec<TouchedEntry> = touched
1958 .iter()
1959 .map(|(path, pre)| TouchedEntry {
1960 path: (*path).to_string(),
1961 pre_sha256: pre.map(str::to_string),
1962 post_sha256: Some("f".repeat(64)),
1963 })
1964 .collect();
1965 RunRecord {
1966 schema_version: 1,
1967 run_id: "20261016T000000Z-aaaaaa".to_string(),
1968 gateway: "gw".to_string(),
1969 mode: "dev".to_string(),
1970 render_sha256: "a".repeat(64),
1971 plan_sha256: "b".repeat(64),
1972 state: RecordState::Applying,
1973 rolled_back_by: None,
1974 touched,
1975 pre_fingerprint: "c".repeat(64),
1976 post_fingerprint: None,
1977 }
1978 }
1979
1980 #[test]
1981 fn the_commit_budget_is_the_hold_less_its_margin_and_never_zero() {
1982 let hold = |hold_s| HoldModel {
1983 hold_s,
1984 ..HoldModel::MEASURED
1985 };
1986 assert_eq!(commit_budget_s(&hold(176)), 174);
1989 assert_eq!(commit_budget_s(&HoldModel::MEASURED), 174);
1990 assert_eq!(commit_budget_s(&hold(2)), 1);
1993 assert_eq!(commit_budget_s(&hold(0)), 1);
1994 assert_eq!(commit_budget_s(&hold(1_000_000)), BUDGET_MAX_S);
1995 }
1996
1997 #[test]
1998 fn snapshot_warnings_name_a_differing_or_missing_copy_and_skip_created_paths() {
1999 let (h1, h2) = ("1".repeat(64), "2".repeat(64));
2000 let rec = record(&[
2001 ("a/config.json", Some(&h1)),
2002 ("b/config.json", Some(&h1)),
2003 ("c/config.json", None),
2004 (LEDGER_KEY, Some(&h2)),
2005 ]);
2006 let mut copied = BTreeMap::new();
2007 copied.insert("files/a/config.json".to_string(), h1.clone());
2008 copied.insert("files/b/config.json".to_string(), h2.clone());
2009 copied.insert("ledger.json".to_string(), h2);
2010 assert_eq!(
2013 snapshot_warnings(&rec, &copied),
2014 ["snapshot_copy_differs (b/config.json)"]
2015 );
2016 copied.remove("files/a/config.json");
2018 assert_eq!(
2019 snapshot_warnings(&rec, &copied),
2020 [
2021 "snapshot_copy_differs (a/config.json)",
2022 "snapshot_copy_differs (b/config.json)"
2023 ]
2024 );
2025 }
2026}