bamboo_domain/session/

persistence.rs

use std::io;
use std::sync::Arc;

use crate::session::task::TaskList;
use crate::session::types::Session;
use crate::session::PermissionAuditSeed;

/// Result of the retrieval-window-specific execute-boundary checkpoint.
///
/// `Rebased` means persistence observed a newer durable transcript, made no
/// write, and replaced the caller's staged value with a clean reconciled base
/// that must be planned and prepared again before dispatch.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RetrievalWindowCheckpointOutcome {
    Committed,
    Rebased,
}

/// Immutable execution provenance for an already-persisted untagged Child wait.
/// This capability is captured once from the execution's initial snapshot; it
/// is not Session metadata and must never be reconstructed at a later save.
#[derive(Debug, Clone)]
pub struct InheritedChildWait {
    session_id: String,
    created_at: chrono::DateTime<chrono::Utc>,
    wait: crate::WaitingForChildrenState,
}

impl InheritedChildWait {
    pub fn capture(session: &Session) -> Option<Self> {
        let wait = session
            .agent_runtime_state
            .as_ref()?
            .waiting_for_children
            .as_ref()?;
        if wait.registered_by_tool_call_id.is_some() {
            return None;
        }
        Some(Self {
            session_id: session.id.clone(),
            created_at: session.created_at,
            wait: wait.clone(),
        })
    }

    pub fn session_id(&self) -> &str {
        &self.session_id
    }

    pub fn wait(&self) -> &crate::WaitingForChildrenState {
        &self.wait
    }

    pub fn validate_session(&self, session: &Session) -> io::Result<()> {
        if session.id != self.session_id || session.created_at != self.created_at {
            return Err(io::Error::other(crate::SessionAuthorityConflict(
                "inherited child wait session changed identity or birth".into(),
            )));
        }
        Ok(())
    }
}

/// Provider-boundary admission through the bound Root writer. None from the
/// port below means this execution does not use ordinary Root authority.
#[derive(Debug, Default)]
pub struct RootInboxAdmission {
    pub merged: usize,
    pub committed_messages: Vec<crate::Message>,
    pub admission_error: Option<String>,
}

/// Merge messages from a live runner snapshot into an already-durable
/// transcript without ever removing or rewriting a durable message.
///
/// Runtime sessions are append-oriented, and every newly-created message has a
/// stable id.  A runner may nevertheless be holding a snapshot that predates a
/// concurrent append (for example, an injected child-completion message).  A
/// terminal/error checkpoint must not full-save that stale snapshot: doing so
/// would shrink the transcript.  Keep the durable ordering and append only the
/// live messages whose ids are not durable yet.
pub fn append_missing_runtime_messages(session: &mut Session, durable: &Session) -> usize {
    let mut seen = durable
        .messages
        .iter()
        .map(|message| message.id.clone())
        .collect::<std::collections::HashSet<_>>();
    let missing = session
        .messages
        .iter()
        .filter(|message| seen.insert(message.id.clone()))
        .cloned()
        .collect::<Vec<_>>();
    let appended = missing.len();
    session.messages = durable.messages.iter().cloned().chain(missing).collect();
    // Provider-native groups are message-anchored and append-only as well. A
    // concurrent durable prefix must not be erased by a stale runner save, and
    // a runner's newly completed group must remain paired with its new message.
    session.merge_provider_transcript_from_durable(durable);
    appended
}

/// Merge the durable SessionInbox admitted-id cursor into a writer snapshot.
///
/// Runtime writers can hold a session clone from before another run admitted
/// an inbox message. No later full save may erase that durable dedupe state.
pub fn merge_session_inbox_admission(session: &mut Session, durable: &Session) {
    let Some(durable_state) = durable.session_inbox_admission().cloned() else {
        return;
    };
    session
        .session_inbox_admission_mut()
        .merge_from(&durable_state);
}

/// Restore durable provider messages identified by their typed
/// `metadata.session_message` marker into a stale writer without preserving
/// unrelated durable suffixes. The bounded cursor is only a fast recent index;
/// the transcript marker is the unbounded source of truth after cursor
/// eviction.
///
/// Insertion follows durable transcript neighbors so an admitted user/runtime
/// message remains ahead of any later assistant output held by the stale
/// runner. This is narrower than [`append_missing_runtime_messages`], retaining
/// the historical shrink semantics for unrelated concurrent messages while
/// making a cursor/tombstone incapable of outliving its transcript entry.
pub fn restore_missing_admitted_inbox_messages(session: &mut Session, durable: &Session) -> usize {
    let admission = durable.session_inbox_admission();
    let mut restored = 0;
    for (durable_index, message) in durable.messages.iter().enumerate() {
        let typed_marker = message
            .metadata
            .as_ref()
            .and_then(|metadata| metadata.get("session_message"))
            .is_some_and(|marker| {
                marker.get("id").and_then(serde_json::Value::as_str) == Some(message.id.as_str())
                    && marker
                        .get("target_session_id")
                        .and_then(serde_json::Value::as_str)
                        == Some(durable.id.as_str())
                    && crate::SessionMessageId::parse(message.id.clone()).is_ok()
            });
        let recent_cursor = admission.is_some_and(|state| state.contains_str(&message.id));
        if !(typed_marker || recent_cursor)
            || session
                .messages
                .iter()
                .any(|current| current.id == message.id)
        {
            continue;
        }

        let insertion = durable.messages[..durable_index]
            .iter()
            .rev()
            .find_map(|predecessor| {
                session
                    .messages
                    .iter()
                    .position(|current| current.id == predecessor.id)
                    .map(|index| index + 1)
            })
            .or_else(|| {
                durable.messages[durable_index + 1..]
                    .iter()
                    .find_map(|successor| {
                        session
                            .messages
                            .iter()
                            .position(|current| current.id == successor.id)
                    })
            })
            .unwrap_or(session.messages.len());
        session.messages.insert(insertion, message.clone());
        restored += 1;
    }
    restored
}

/// Port for runtime (non-authoritative) session persistence.
///
/// Implementors must:
/// - Serialize concurrent saves per session ID.
/// - Merge on-disk authoritative metadata (`title`, `title_generated`, `pinned`, `title_version`,
///   `metadata_version`) before writing, so UI edits are never clobbered.
#[async_trait::async_trait]
pub trait RuntimeSessionPersistence: Send + Sync {
    /// An execution-private binding, never reconstructed from mutable state.
    fn inherited_child_wait(&self) -> Option<InheritedChildWait> {
        None
    }

    /// Return an execution-private view retaining the captured wait provenance
    /// through every intermediate save. Unsupported persisters fail closed.
    fn bind_inherited_child_wait(
        &self,
        _inherited: InheritedChildWait,
    ) -> io::Result<Arc<dyn RuntimeSessionPersistence>> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            crate::SessionAuthorityConflict(
                "execution-scoped inherited child wait persistence is unsupported".into(),
            ),
        ))
    }

    /// Immutable capability already captured by this concrete execution.
    /// An unbound Host/default persister never supplies a current owner.
    fn root_actor_writer(&self) -> Option<crate::RootActorRuntimeWrite> {
        None
    }

    /// Execution-private input restriction. A selector carries no authority;
    /// only a persister already bound to a real Root writer may install it.
    /// Unsupported hosts fail closed. This introduces no durable state.
    fn bind_root_inbox_receipt(
        &self,
        _receipt: crate::SessionInboxReceipt,
    ) -> io::Result<Arc<dyn RuntimeSessionPersistence>> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "Root receipt binding unsupported",
        ))
    }

    fn root_inbox_receipt(&self) -> Option<&crate::SessionInboxReceipt> {
        None
    }

    /// Whether this Host requires an explicit execution capability for the
    /// proposed Root. A route without a bound event/persistence handoff must
    /// reject before claiming or making a provider call.
    fn root_actor_execution_required(&self, _session: &Session) -> bool {
        false
    }

    /// Host execution binding, never a request DTO. Legacy embedders have no
    /// ActorDirectory; a host that enables one must reject unsupported storage
    /// before claiming rather than returning an unfenced writer.
    async fn bind_root_actor_execution(
        &self,
        _session: &Session,
        _run_id: &str,
    ) -> io::Result<Option<RootActorExecutionBinding>> {
        Ok(None)
    }

    /// Persist the session, merging any newer authoritative metadata from disk.
    async fn save_runtime_session(&self, session: &mut Session) -> io::Result<()>;

    /// Whether this private writer supports the physical profile check.
    fn supports_bound_startup_session(&self) -> bool {
        false
    }

    /// Persist a Root-bound transition only if its original execution profile
    /// still matches under the backend's physical writer and Root authority.
    /// Unsupported persisters must never emulate this with a separate read/save.
    async fn save_bound_startup_session(
        &self,
        _session: &mut Session,
        _expected: &Session,
    ) -> io::Result<()> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            crate::SessionAuthorityConflict(
                "conditional bound startup persistence is unsupported".into(),
            ),
        ))
    }

    /// Persist a runner's final snapshot. Implementations with a per-session
    /// lock should reconcile a tool-registered child wait against the latest
    /// durable wait inside that lock: a child may finish after the runner's
    /// last read but before this save. The default retains compatibility for
    /// in-memory and test persisters without that concurrent completion path.
    async fn save_finalized_runtime_session(&self, session: &mut Session) -> io::Result<()> {
        self.save_runtime_session(session).await
    }

    /// Finalize a wait inherited by this execution. Backends must reconcile
    /// against durable state inside their cross-process write transaction.
    /// Unsupported adapters fail closed rather than reopen a read/save race.
    async fn save_finalized_runtime_with_inherited_child_wait(
        &self,
        session: &mut Session,
        inherited: &crate::session::runtime_state::WaitingForChildrenState,
    ) -> io::Result<()> {
        let _ = (session, inherited);
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            crate::SessionAuthorityConflict(
                "atomic inherited child wait finalization is unsupported".into(),
            ),
        ))
    }

    /// Authoritatively seed one validated actor activation.
    ///
    /// Unlike an ordinary runtime save, the incoming RunSpec posture and its
    /// complete audit record must replace any posture left by a previous warm
    /// activation. Implementations must still preserve durable SessionInbox
    /// admission/transcript proof and serialize the operation per session.
    ///
    /// There is no safe generic implementation through
    /// [`Self::save_runtime_session`]: that primitive is explicitly allowed to
    /// adopt a newer disk posture, which would make warm workers sticky across
    /// runs. Custom persisters therefore fail closed until they implement this
    /// authority boundary deliberately.
    async fn seed_runtime_activation(&self, _session: &mut Session) -> io::Result<()> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support authoritative activation seeding",
        ))
    }

    /// Atomically persist a worker-declared executor mapping for the current
    /// host-authoritative permission posture.
    ///
    /// The caller supplies the audit revision it observed before dispatch.
    /// Implementations must load and compare that revision while holding the
    /// per-session lock, reject a concurrent posture update, and allocate a new
    /// host revision/timestamp themselves. Remote audit clocks are never an
    /// authority at this boundary.
    async fn record_permission_posture_activation(
        &self,
        _session_id: &str,
        _expected_audit_revision: Option<u64>,
        _seed: &PermissionAuditSeed,
    ) -> io::Result<Option<Session>> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support atomic permission posture activation",
        ))
    }

    /// Persist only the runtime control-plane for a session.
    ///
    /// Task lists and other runtime metadata belong to the control-plane and do
    /// not require rewriting the potentially large message transcript. Built-in
    /// persistence implementations with a runtime sidecar should override this
    /// operation with their sidecar-only path. Custom/legacy implementations
    /// remain source-compatible and safely fall back to the full runtime save.
    ///
    /// Callers must not rely on this operation to persist message or
    /// `model_context_state` changes. The durable ledger is checkpoint-owned;
    /// sidecar implementations must preserve its latest committed value while
    /// applying the caller's narrow control-plane mutation.
    async fn save_runtime_control_plane(&self, session: &mut Session) -> io::Result<()> {
        self.save_runtime_session(session).await
    }

    /// Load the representation paired with
    /// [`save_runtime_control_plane`](Self::save_runtime_control_plane).
    ///
    /// Sidecar-capable implementations should return their message-free
    /// control-plane snapshot. The default deliberately returns the full
    /// runtime session: when the paired save also falls back to a full save,
    /// retaining the transcript makes that fallback safe rather than replacing
    /// durable messages with an empty sidecar-shaped snapshot.
    async fn load_runtime_control_plane(&self, session_id: &str) -> io::Result<Option<Session>> {
        self.load_runtime_session(session_id).await
    }

    /// Atomically update only the shared Task list and its version.
    ///
    /// The default is safe for custom/legacy persistence: it loads the full
    /// runtime session, changes only Task-owned fields, then uses the paired
    /// control-plane save (which itself defaults to a full save). Returning
    /// `false` means the implementation could not load the target; callers that
    /// also hold a [`Storage`](crate::storage::Storage) may retain legacy
    /// behavior with an explicit full-load/full-save fallback.
    ///
    /// Implementations with per-session transactions should override this so
    /// the load, narrow mutation and save share one critical section.
    async fn update_task_list_control_plane(
        &self,
        session_id: &str,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        let Some(mut session) = self.load_runtime_session(session_id).await? else {
            return Ok(false);
        };
        session.set_task_list(task_list.clone());
        session.set_task_list_version_meta(version.to_string());
        self.save_runtime_control_plane(&mut session).await?;
        Ok(true)
    }

    /// Atomically update Task-owned control-plane fields only when the durable
    /// Task generation and exact list still match the expected snapshot.
    ///
    /// `false` covers an unsupported atomic compare-and-patch, a missing target,
    /// or a version conflict. Callers must treat it as a stale write and must
    /// not publish their staged Task state. The default fails closed because a
    /// load followed by a separately locked save is not an atomic CAS.
    async fn update_task_list_control_plane_if_version(
        &self,
        session_id: &str,
        expected_version: &str,
        expected_task_list: &TaskList,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        let _ = (
            session_id,
            expected_version,
            expected_task_list,
            task_list,
            version,
        );
        Ok(false)
    }

    /// Recoverably compare-and-patch the executing session and its shared root.
    /// Implementations must validate both generations before either target is
    /// written and may return `Ok(true)` only after both Task generations are
    /// durable with no undo record that could later revert them. An error after
    /// one physical write must restore both originals before returning or retain
    /// durable recovery state and fail subsequent paired access closed until
    /// recovery completes. Root-session callers pass the same id twice and
    /// receive the single-target CAS semantics above.
    async fn update_task_list_control_planes_if_version(
        &self,
        session_id: &str,
        shared_session_id: &str,
        expected_version: &str,
        expected_task_list: &TaskList,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        if session_id == shared_session_id {
            return self
                .update_task_list_control_plane_if_version(
                    session_id,
                    expected_version,
                    expected_task_list,
                    task_list,
                    version,
                )
                .await;
        }
        let _ = (
            session_id,
            shared_session_id,
            expected_version,
            expected_task_list,
            task_list,
            version,
        );
        Ok(false)
    }

    /// Append-safe checkpoint used at the shared engine execute boundary.
    ///
    /// Unlike [`save_runtime_session`](Self::save_runtime_session), this must
    /// preserve messages that were appended durably by a concurrent writer
    /// after the runner loaded its snapshot.  Implementations that can provide
    /// a per-session transaction should override this method and perform the
    /// load/merge/save under one lock.  The default still reconciles against a
    /// latest snapshot for lightweight/custom SDK persisters; the built-in
    /// storage implementation supplies the atomic variant.
    async fn checkpoint_runtime_session(&self, session: &mut Session) -> io::Result<()> {
        if let Some(durable) = self.load_runtime_session(&session.id).await? {
            session
                .check_conversation_generation(&durable)
                .map_err(|error| {
                    io::Error::new(
                        io::ErrorKind::WouldBlock,
                        crate::SessionAuthorityConflict(error.to_string()),
                    )
                })?;
            append_missing_runtime_messages(session, &durable);
            merge_session_inbox_admission(session, &durable);
        }
        self.save_runtime_session(session).await
    }

    /// Bound execution only; unsupported/custom adapters cannot authorize a
    /// continuation by falling back to generic checkpoint or snapshot reads.
    async fn checkpoint_live_bootstrap(
        &self,
        _session: &Session,
        _inbox: Arc<dyn crate::SessionInboxPort>,
        _gate: crate::AdmissionGate,
    ) -> io::Result<crate::LiveBootstrapProof> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "live bootstrap checkpoint unsupported",
        ))
    }

    async fn confirm_live_bootstrap(
        &self,
        _proof: &crate::LiveBootstrapProof,
        _inbox: Arc<dyn crate::SessionInboxPort>,
        _publish: crate::RootActorRuntimeEventPublisher,
    ) -> io::Result<()> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "live bootstrap confirmation unsupported",
        ))
    }

    /// Only the existing owner may claim structured Loop inputs. Unsupported
    /// adapters do not fall back to ordinary provider-message admission.
    async fn claim_scoped_loop_controls(
        &self,
        _session: &Session,
        _inbox: Arc<dyn crate::SessionInboxPort>,
        _active_run_id: &str,
        _limit: usize,
    ) -> io::Result<Vec<crate::ScopedLoopControlClaim>> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "scoped Loop claims unsupported",
        ))
    }

    /// State, immutable control outcome and typed consumption proof share the
    /// actual target checkpoint; ACK is last. Err never means rollback.
    async fn checkpoint_scoped_loop_control(
        &self,
        _session: &mut Session,
        _bindings: &[crate::ScopedLoopControlBinding],
        _activation: Arc<dyn crate::SessionActivationPort>,
        _input: Option<(
            Arc<dyn crate::SessionInboxPort>,
            crate::ScopedLoopControlClaim,
        )>,
    ) -> io::Result<()> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "scoped Loop checkpoint unsupported",
        ))
    }

    /// Canonical read after a failed/uncertain checkpoint. A cached Session or
    /// the proposed caller snapshot is not confirmation of what was written.
    async fn load_loop_control_checkpoint(&self, _session_id: &str) -> io::Result<Option<Session>> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "strict Loop checkpoint reload unsupported",
        ))
    }

    async fn admit_root_inbox(
        &self,
        _session: &mut Session,
        _inbox: Arc<dyn crate::SessionInboxPort>,
        _active_run_id: Option<&str>,
    ) -> io::Result<Option<RootInboxAdmission>> {
        Ok(None)
    }

    /// Atomically commit a staged archive transcript rewrite.
    ///
    /// `expected_base` is the exact pre-archive Session used for planning.
    /// Implementations must compare it with the latest durable transcript while
    /// holding their per-session serialization lock. If a concurrent append or
    /// rewrite is present, they must perform no save, rebase `staged` onto that
    /// durable snapshot, and return [`RetrievalWindowCheckpointOutcome::Rebased`].
    /// Otherwise they must preserve the staged message archive flags and commit
    /// them with the compression event and model-context reset. The built-in
    /// implementation also accepts a MessageSummary archive through this same
    /// boundary, preserving raw history while replacing its derived summary.
    ///
    /// There is no safe fallback through the ordinary append-only checkpoint,
    /// because that path deliberately restores durable message clones and would
    /// erase the new archive flags. Custom persisters therefore fail closed
    /// until they implement this boundary explicitly.
    async fn checkpoint_retrieval_window(
        &self,
        _expected_base: &Session,
        _staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support retrieval-window checkpoints",
        ))
    }

    /// Atomically commit a provider-visible System-prompt rewrite.
    ///
    /// This is deliberately separate from both the append-safe runtime
    /// checkpoint (which must restore durable message content) and the
    /// retrieval-window archive checkpoint (which requires a new archive
    /// event). Implementations compare `expected_base` under their per-session
    /// lock, return `Rebased` without writing on conflict, and accept only the
    /// bounded prompt rewrite plus its provider/model-context reset.
    async fn checkpoint_prompt_rewrite(
        &self,
        _expected_base: &Session,
        _staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support prompt-rewrite checkpoints",
        ))
    }

    /// Atomically commit one permanently rejected `archive_context` result.
    ///
    /// This boundary permits only the correlated Tool result rewrite, the
    /// bounded consumed/rejection metadata, and the provider/model-context
    /// reset required to make that rewrite visible. Implementations compare
    /// `expected_base` under their per-session lock and return `Rebased`
    /// without writing when the durable transcript changed concurrently.
    async fn checkpoint_manual_archive_rejection(
        &self,
        _expected_base: &Session,
        _staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support manual archive-rejection checkpoints",
        ))
    }

    /// Atomically consume one successful no-op `archive_context` request.
    ///
    /// This boundary permits only the correlated consumed-occurrence marker.
    /// Implementations compare `expected_base` under their per-session lock
    /// and return `Rebased` without writing when any durable transcript,
    /// metadata, runtime metadata, or execution-profile field changed. The
    /// caller must then restage the marker from the returned durable snapshot.
    async fn checkpoint_manual_archive_consumption(
        &self,
        _expected_base: &Session,
        _staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        Err(io::Error::new(
            io::ErrorKind::Unsupported,
            "runtime persistence does not support manual archive-consumption checkpoints",
        ))
    }

    /// Load the latest runtime-visible session snapshot when the persistence
    /// implementation can coordinate reads. Tools may update a repository-owned
    /// clone while an agent loop holds its own live Session; the loop uses this
    /// hook to merge narrowly-scoped tool side effects before its next save.
    async fn load_runtime_session(&self, _session_id: &str) -> io::Result<Option<Session>> {
        Ok(None)
    }

    /// Read a canonical snapshot for execution admission without publishing or
    /// trusting a cached clone. Preserve the same bound-reader checks as the
    /// runtime loader. A persister with a cache must override this operation to
    /// read its durable source (or return Unsupported); the default serves
    /// uncached/legacy loaders. This read never grants execution/write authority.
    async fn load_runtime_admission_snapshot(
        &self,
        session_id: &str,
    ) -> io::Result<Option<Session>> {
        self.load_runtime_session(session_id).await
    }

    /// Clear the bounded compatibility queue iff it still equals the entries
    /// that were durably copied into SessionInbox. Implementations with a
    /// per-session transaction should override this method.
    async fn clear_legacy_pending_messages(
        &self,
        session_id: &str,
        expected: &[serde_json::Value],
    ) -> io::Result<bool> {
        let Some(mut latest) = self.load_runtime_session(session_id).await? else {
            return Ok(false);
        };
        if latest.pending_injected_messages().as_deref() != Some(expected) {
            return Ok(false);
        }
        latest.clear_pending_injected_messages();
        self.save_runtime_session(&mut latest).await?;
        Ok(true)
    }

    /// Append one JSON-line analysis record to the session's append-only
    /// token-usage log (see [`Storage::append_token_usage_record`]). Defaults to
    /// a no-op so non-file-backed persisters are unaffected.
    ///
    /// [`Storage::append_token_usage_record`]: crate::storage::Storage::append_token_usage_record
    async fn append_token_usage_record(&self, session_id: &str, json_line: &str) -> io::Result<()> {
        let _ = (session_id, json_line);
        Ok(())
    }
}

/// Ownership passed from the canonical repository into one concrete runtime.
/// The runtime owns renewal and finish; dropping an execution must release its
/// own fence without ever finishing a replacement owner.
pub struct RootActorExecutionBinding {
    pub persistence: Arc<dyn RuntimeSessionPersistence>,
    pub directory: Arc<dyn crate::ActorDirectoryPort>,
    pub owner: crate::RootActorRuntimeWrite,
    pub lease_duration: std::time::Duration,
    abandon: Option<Box<dyn FnOnce() + Send + Sync>>,
}

impl RootActorExecutionBinding {
    pub fn new(
        persistence: Arc<dyn RuntimeSessionPersistence>,
        directory: Arc<dyn crate::ActorDirectoryPort>,
        owner: crate::RootActorRuntimeWrite,
        lease_duration: std::time::Duration,
        abandon: impl FnOnce() + Send + Sync + 'static,
    ) -> Self {
        Self {
            persistence,
            directory,
            owner,
            lease_duration,
            abandon: Some(Box::new(abandon)),
        }
    }

    /// The runtime calls this only after handing finish to an owned detached
    /// job, so cancellation of the waiter cannot strand this execution.
    pub fn disarm_abandonment(&mut self) {
        self.abandon.take();
    }
}

impl Drop for RootActorExecutionBinding {
    fn drop(&mut self) {
        if let Some(abandon) = self.abandon.take() {
            abandon();
        }
    }
}

#[async_trait::async_trait]
impl<T: RuntimeSessionPersistence + ?Sized> RuntimeSessionPersistence for Arc<T> {
    fn inherited_child_wait(&self) -> Option<InheritedChildWait> {
        (**self).inherited_child_wait()
    }

    fn bind_inherited_child_wait(
        &self,
        inherited: InheritedChildWait,
    ) -> io::Result<Arc<dyn RuntimeSessionPersistence>> {
        (**self).bind_inherited_child_wait(inherited)
    }

    fn root_actor_writer(&self) -> Option<crate::RootActorRuntimeWrite> {
        (**self).root_actor_writer()
    }

    fn bind_root_inbox_receipt(
        &self,
        receipt: crate::SessionInboxReceipt,
    ) -> io::Result<Arc<dyn RuntimeSessionPersistence>> {
        (**self).bind_root_inbox_receipt(receipt)
    }

    fn root_inbox_receipt(&self) -> Option<&crate::SessionInboxReceipt> {
        (**self).root_inbox_receipt()
    }

    fn root_actor_execution_required(&self, session: &Session) -> bool {
        (**self).root_actor_execution_required(session)
    }

    async fn bind_root_actor_execution(
        &self,
        session: &Session,
        run_id: &str,
    ) -> io::Result<Option<RootActorExecutionBinding>> {
        (**self).bind_root_actor_execution(session, run_id).await
    }
    async fn save_runtime_session(&self, session: &mut Session) -> io::Result<()> {
        (**self).save_runtime_session(session).await
    }

    fn supports_bound_startup_session(&self) -> bool {
        (**self).supports_bound_startup_session()
    }

    async fn save_bound_startup_session(
        &self,
        session: &mut Session,
        expected: &Session,
    ) -> io::Result<()> {
        (**self).save_bound_startup_session(session, expected).await
    }

    async fn save_finalized_runtime_session(&self, session: &mut Session) -> io::Result<()> {
        (**self).save_finalized_runtime_session(session).await
    }

    async fn save_finalized_runtime_with_inherited_child_wait(
        &self,
        session: &mut Session,
        inherited: &crate::session::runtime_state::WaitingForChildrenState,
    ) -> io::Result<()> {
        (**self)
            .save_finalized_runtime_with_inherited_child_wait(session, inherited)
            .await
    }

    async fn seed_runtime_activation(&self, session: &mut Session) -> io::Result<()> {
        (**self).seed_runtime_activation(session).await
    }

    async fn record_permission_posture_activation(
        &self,
        session_id: &str,
        expected_audit_revision: Option<u64>,
        seed: &PermissionAuditSeed,
    ) -> io::Result<Option<Session>> {
        (**self)
            .record_permission_posture_activation(session_id, expected_audit_revision, seed)
            .await
    }

    async fn save_runtime_control_plane(&self, session: &mut Session) -> io::Result<()> {
        (**self).save_runtime_control_plane(session).await
    }

    async fn load_runtime_control_plane(&self, session_id: &str) -> io::Result<Option<Session>> {
        (**self).load_runtime_control_plane(session_id).await
    }

    async fn update_task_list_control_plane(
        &self,
        session_id: &str,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        (**self)
            .update_task_list_control_plane(session_id, task_list, version)
            .await
    }

    async fn update_task_list_control_plane_if_version(
        &self,
        session_id: &str,
        expected_version: &str,
        expected_task_list: &TaskList,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        (**self)
            .update_task_list_control_plane_if_version(
                session_id,
                expected_version,
                expected_task_list,
                task_list,
                version,
            )
            .await
    }

    async fn update_task_list_control_planes_if_version(
        &self,
        session_id: &str,
        shared_session_id: &str,
        expected_version: &str,
        expected_task_list: &TaskList,
        task_list: &TaskList,
        version: &str,
    ) -> io::Result<bool> {
        (**self)
            .update_task_list_control_planes_if_version(
                session_id,
                shared_session_id,
                expected_version,
                expected_task_list,
                task_list,
                version,
            )
            .await
    }

    async fn checkpoint_runtime_session(&self, session: &mut Session) -> io::Result<()> {
        (**self).checkpoint_runtime_session(session).await
    }

    async fn checkpoint_live_bootstrap(
        &self,
        session: &Session,
        inbox: Arc<dyn crate::SessionInboxPort>,
        gate: crate::AdmissionGate,
    ) -> io::Result<crate::LiveBootstrapProof> {
        (**self)
            .checkpoint_live_bootstrap(session, inbox, gate)
            .await
    }

    async fn confirm_live_bootstrap(
        &self,
        proof: &crate::LiveBootstrapProof,
        inbox: Arc<dyn crate::SessionInboxPort>,
        publish: crate::RootActorRuntimeEventPublisher,
    ) -> io::Result<()> {
        (**self).confirm_live_bootstrap(proof, inbox, publish).await
    }

    async fn claim_scoped_loop_controls(
        &self,
        session: &Session,
        inbox: Arc<dyn crate::SessionInboxPort>,
        active_run_id: &str,
        limit: usize,
    ) -> io::Result<Vec<crate::ScopedLoopControlClaim>> {
        (**self)
            .claim_scoped_loop_controls(session, inbox, active_run_id, limit)
            .await
    }

    async fn checkpoint_scoped_loop_control(
        &self,
        session: &mut Session,
        bindings: &[crate::ScopedLoopControlBinding],
        activation: Arc<dyn crate::SessionActivationPort>,
        input: Option<(
            Arc<dyn crate::SessionInboxPort>,
            crate::ScopedLoopControlClaim,
        )>,
    ) -> io::Result<()> {
        (**self)
            .checkpoint_scoped_loop_control(session, bindings, activation, input)
            .await
    }

    async fn load_loop_control_checkpoint(&self, session_id: &str) -> io::Result<Option<Session>> {
        (**self).load_loop_control_checkpoint(session_id).await
    }

    async fn admit_root_inbox(
        &self,
        session: &mut Session,
        inbox: Arc<dyn crate::SessionInboxPort>,
        active_run_id: Option<&str>,
    ) -> io::Result<Option<RootInboxAdmission>> {
        (**self)
            .admit_root_inbox(session, inbox, active_run_id)
            .await
    }

    async fn checkpoint_retrieval_window(
        &self,
        expected_base: &Session,
        staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        (**self)
            .checkpoint_retrieval_window(expected_base, staged)
            .await
    }

    async fn checkpoint_prompt_rewrite(
        &self,
        expected_base: &Session,
        staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        (**self)
            .checkpoint_prompt_rewrite(expected_base, staged)
            .await
    }

    async fn checkpoint_manual_archive_rejection(
        &self,
        expected_base: &Session,
        staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        (**self)
            .checkpoint_manual_archive_rejection(expected_base, staged)
            .await
    }

    async fn checkpoint_manual_archive_consumption(
        &self,
        expected_base: &Session,
        staged: &mut Session,
    ) -> io::Result<RetrievalWindowCheckpointOutcome> {
        (**self)
            .checkpoint_manual_archive_consumption(expected_base, staged)
            .await
    }

    async fn load_runtime_session(&self, session_id: &str) -> io::Result<Option<Session>> {
        (**self).load_runtime_session(session_id).await
    }

    async fn load_runtime_admission_snapshot(
        &self,
        session_id: &str,
    ) -> io::Result<Option<Session>> {
        (**self).load_runtime_admission_snapshot(session_id).await
    }

    async fn clear_legacy_pending_messages(
        &self,
        session_id: &str,
        expected: &[serde_json::Value],
    ) -> io::Result<bool> {
        (**self)
            .clear_legacy_pending_messages(session_id, expected)
            .await
    }

    async fn append_token_usage_record(&self, session_id: &str, json_line: &str) -> io::Result<()> {
        (**self)
            .append_token_usage_record(session_id, json_line)
            .await
    }
}