everruns 0.18.2

// LocalRuntimeBuilder — optional convenience sugar over InProcessRuntimeBuilder.
//
// This wires a LocalProfile + LocalBackends (task registry + schedule store)
// and a real-disk workspace file store into an `InProcessRuntimeBuilder`. It is
// strictly optional: everything it does can be done by composing the pieces in
// `LocalBackends` directly. The platform store is intentionally NOT wired here
// because the local platform store needs a `LocalSessionRunner` that usually
// wraps the built runtime (a chicken/egg). Attach it after build via
// `LocalBackends::with_platform_runner` and rebuild, or use the standalone
// `LocalPlatformStore`.

use everruns_host::HostComposition;
use std::sync::Arc;

use everruns_host::{
    InProcessRuntime, InProcessRuntimeBuilder, RealDiskSessionFileSystemFactory,
    SessionFileSystemFactory, SessionFileSystemFactoryContext,
};
use everruns_provider::error::Result;
use everruns_provider::model_spec::ModelSpec;

use super::backends::LocalBackends;
use super::profile::LocalProfile;

/// Capability registry for a local profile: the hosted product catalog whose
/// services `LocalBackends` supplies from SQLite (EVE-885), plus whatever
/// portable policy capabilities and environment integrations this host selected
/// (EVE-884 and EVE-883).
///
/// Callers that assemble their own [`HostComposition`] for a local profile
/// should start here, so build-time capability validation sees the same set the
/// local runtime executes.
pub fn local_capability_registry() -> everruns_core::CapabilityRegistry {
    let registry = everruns_platform::capabilities::hosted_capability_registry();
    everruns_host::compose_runtime_capability_registry(registry)
}

/// Convenience wrapper around [`InProcessRuntimeBuilder`] that wires a local
/// profile + SQLite-backed task/schedule stores and a workspace file store.
pub struct LocalRuntimeBuilder {
    profile: LocalProfile,
    inner: InProcessRuntimeBuilder,
    file_system_factory: Option<Arc<dyn SessionFileSystemFactory>>,
    /// Caller-supplied platform definition override. When `Some`, `build()`
    /// installs it as-is instead of the default local one; the caller owns the
    /// session filesystem factory in that case.
    host_composition: Option<HostComposition>,
}

impl LocalRuntimeBuilder {
    /// Start from a profile. Defaults the platform definition to built-ins via
    /// `InProcessRuntimeBuilder::new()`.
    pub fn new(profile: LocalProfile) -> Self {
        Self {
            profile,
            inner: InProcessRuntimeBuilder::new(),
            file_system_factory: None,
            host_composition: None,
        }
    }

    /// Replace the platform definition (capabilities, drivers, ...). When set,
    /// `build()` respects this definition instead of constructing the default
    /// local one, so capability/driver overrides take effect. The caller is
    /// then responsible for the session filesystem factory on their definition.
    pub fn host_composition(mut self, host_composition: HostComposition) -> Self {
        self.host_composition = Some(host_composition);
        self
    }

    /// Register `provider` (replacing any same-name registration) and default
    /// the runtime model to `model_id` on it when nothing else set one.
    ///
    /// Deterministic local runs pass the `llmsim` provider built by
    /// `everruns_llmsim::llm_sim_provider(...)` here; testing/demo helpers are
    /// not part of the production dependency tree.
    pub fn provider_with_default_model(
        mut self,
        provider: everruns_provider::runtime_provider::Provider,
        model_id: impl Into<String>,
    ) -> Self {
        self.inner = self.inner.provider_with_default_model(provider, model_id);
        self
    }

    /// Set the runtime default model.
    pub fn default_model(mut self, model: ModelSpec) -> Self {
        self.inner = self.inner.default_model(model);
        self
    }

    /// Seed a harness definition (under its embedder-chosen id).
    pub fn harness(mut self, harness: everruns_host::SeededHarness) -> Self {
        self.inner = self.inner.harness(harness);
        self
    }

    /// Seed an agent.
    pub fn agent(mut self, agent: everruns_core::AgentDefinition) -> Self {
        self.inner = self.inner.agent(agent);
        self
    }

    /// Seed a session.
    pub fn session(mut self, session: everruns_core::ExecutionSession) -> Self {
        self.inner = self.inner.session(session);
        self
    }

    /// Override the session filesystem factory. By default a real-disk factory
    /// rooted at `profile.workspace_root` is used.
    pub fn session_file_system_factory(
        mut self,
        factory: Arc<dyn SessionFileSystemFactory>,
    ) -> Self {
        self.file_system_factory = Some(factory);
        self
    }

    /// Access the underlying `InProcessRuntimeBuilder` for advanced wiring.
    pub fn inner_mut(&mut self) -> &mut InProcessRuntimeBuilder {
        &mut self.inner
    }

    /// Build the runtime and return it along with the constructed
    /// [`LocalBackends`] so the embedder can attach a platform runner and reuse
    /// the local stores. The task registry and schedule store factory are
    /// installed on the runtime; the platform store is left to the embedder.
    pub async fn build(self) -> Result<(InProcessRuntime, LocalBackends)> {
        self.profile
            .ensure_dirs()
            .map_err(|e| everruns_provider::error::AgentLoopError::config(e.to_string()))?;

        let local = LocalBackends::new(
            self.profile.clone(),
            everruns_host::HostBackends::in_memory(),
        )?;

        // Respect a caller-supplied platform definition; otherwise build the
        // default local one rooted at the profile workspace. Only install the
        // default when the caller has not overridden it, so capability/driver
        // overrides via `host_composition(...)` are not silently discarded.
        let host_composition = match self.host_composition {
            Some(pd) => pd,
            None => {
                let factory = self.file_system_factory.unwrap_or_else(|| {
                    Arc::new(RealDiskSessionFileSystemFactory::new(
                        self.profile.workspace_root.clone(),
                    ))
                });
                HostComposition::builder()
                    .capability_registry(local_capability_registry())
                    .driver_registry(everruns_provider::DriverRegistry::new())
                    .egress_service(everruns_host::runtime_egress_service())
                    .session_file_system_factory(factory)
                    .build()
            }
        };

        let runtime = self
            .inner
            .host_composition(host_composition)
            .session_file_system_factory_context(SessionFileSystemFactoryContext::new())
            .backends(local.runtime_backends.clone())
            .build()
            .await?;

        Ok((runtime, local))
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn default_local_registry_keeps_core_and_full_portable_catalogs() {
        let registry = local_capability_registry();

        for capability_id in [
            "session_schedule",
            "current_time",
            "compaction",
            "tool_call_repair",
            "usage_limit_auto_continue",
        ] {
            assert!(
                registry.has(capability_id),
                "local registry is missing `{capability_id}`"
            );
        }
    }
}