File size: 2,378 Bytes
ea39c0e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | use codex_http_client::HttpClientFactory;
use crate::EnvironmentManager;
use crate::ExecServerError;
use crate::ExecServerRuntimePaths;
use crate::environment_provider::EnvironmentDefault;
use crate::environment_provider::EnvironmentProviderSnapshot;
use crate::remote::NoiseRendezvousEnvironmentConfig;
#[derive(Debug)]
pub(crate) enum PreparedEnvironmentSource {
Noise(NoiseRendezvousEnvironmentConfig),
Snapshot(EnvironmentProviderSnapshot),
}
/// Holds discovered execution environments before their HTTP policy is resolved.
///
/// Preparing environments does not start remote connections. Callers can inspect
/// the default environment to choose config-loading behavior and then build the
/// manager with the effective outbound HTTP policy.
#[derive(Debug)]
pub struct PreparedEnvironmentManager {
pub(crate) source: PreparedEnvironmentSource,
}
impl PreparedEnvironmentManager {
/// Returns whether the discovered default environment is remote.
pub fn default_environment_is_remote(&self) -> bool {
match &self.source {
PreparedEnvironmentSource::Noise(_) => true,
PreparedEnvironmentSource::Snapshot(snapshot) => match &snapshot.default {
EnvironmentDefault::Disabled => false,
EnvironmentDefault::EnvironmentId(default_id) => snapshot
.environments
.iter()
.any(|(environment_id, _)| environment_id == default_id),
},
}
}
/// Builds the manager and starts remote connections using the supplied policy.
pub fn build(
self,
local_runtime_paths: Option<ExecServerRuntimePaths>,
http_client_factory: HttpClientFactory,
) -> Result<EnvironmentManager, ExecServerError> {
match self.source {
PreparedEnvironmentSource::Noise(config) => {
EnvironmentManager::from_noise_environment_config(
config,
local_runtime_paths,
http_client_factory,
)
}
PreparedEnvironmentSource::Snapshot(snapshot) => EnvironmentManager::from_snapshot(
snapshot,
local_runtime_paths,
http_client_factory,
),
}
}
}
#[cfg(test)]
#[path = "environment_bootstrap_tests.rs"]
mod tests;
|