systemprompt_cli/
env_overrides.rs

1//! Process-environment snapshot for the CLI.
2//!
3//! [`EnvOverrides`] captures every environment variable the CLI consults, read
4//! once at process start ([`EnvOverrides::from_process_env`]) and threaded
5//! through [`crate::context::CommandContext`]. Command code never calls
6//! `std::env::var` directly — tests construct the snapshot with
7//! [`EnvOverrides::from_vars`] instead of mutating process state.
8//!
9//! Copyright (c) systemprompt.io — Business Source License 1.1.
10//! See <https://systemprompt.io> for licensing details.
11
12use std::collections::HashMap;
13
14use systemprompt_identifiers::{ContextId, SessionId, UserId};
15
16#[derive(Debug, Clone, Default)]
17pub struct EnvOverrides {
18    pub output_format: Option<String>,
19    pub log_level: Option<String>,
20    pub no_color: bool,
21    pub non_interactive: bool,
22    pub profile: Option<String>,
23    pub rust_log: Option<String>,
24    pub is_fly: bool,
25    pub is_remote_cli: bool,
26    pub editor: Option<String>,
27    pub database_url: Option<String>,
28    pub services_path: Option<String>,
29    pub session: SessionEnv,
30}
31
32#[derive(Debug, Clone, Default)]
33pub struct SessionEnv {
34    pub user_id: Option<UserId>,
35    pub session_id: Option<SessionId>,
36    pub context_id: Option<ContextId>,
37    pub auth_token: Option<String>,
38}
39
40impl EnvOverrides {
41    #[must_use]
42    pub fn from_process_env() -> Self {
43        Self::from_lookup(|key| std::env::var(key).ok())
44    }
45
46    #[must_use]
47    pub fn from_vars<I, K, V>(vars: I) -> Self
48    where
49        I: IntoIterator<Item = (K, V)>,
50        K: Into<String>,
51        V: Into<String>,
52    {
53        let map: HashMap<String, String> = vars
54            .into_iter()
55            .map(|(k, v)| (k.into(), v.into()))
56            .collect();
57        Self::from_lookup(|key| map.get(key).cloned())
58    }
59
60    fn from_lookup(lookup: impl Fn(&str) -> Option<String>) -> Self {
61        Self {
62            output_format: lookup("SYSTEMPROMPT_OUTPUT_FORMAT"),
63            log_level: lookup("SYSTEMPROMPT_LOG_LEVEL"),
64            no_color: lookup("SYSTEMPROMPT_NO_COLOR").is_some() || lookup("NO_COLOR").is_some(),
65            non_interactive: lookup("SYSTEMPROMPT_NON_INTERACTIVE").is_some(),
66            profile: lookup("SYSTEMPROMPT_PROFILE"),
67            rust_log: lookup("RUST_LOG"),
68            is_fly: lookup("FLY_APP_NAME").is_some(),
69            is_remote_cli: lookup("SYSTEMPROMPT_CLI_REMOTE").is_some(),
70            editor: lookup("VISUAL").or_else(|| lookup("EDITOR")),
71            database_url: lookup("DATABASE_URL"),
72            services_path: lookup("SYSTEMPROMPT_SERVICES_PATH"),
73            session: SessionEnv {
74                user_id: lookup("SYSTEMPROMPT_USER_ID").map(UserId::new),
75                session_id: lookup("SYSTEMPROMPT_SESSION_ID").map(SessionId::new),
76                context_id: lookup("SYSTEMPROMPT_CONTEXT_ID").and_then(|value| {
77                    ContextId::try_new(value)
78                        .inspect_err(|error| {
79                            tracing::warn!(
80                                error = %error,
81                                "ignoring malformed SYSTEMPROMPT_CONTEXT_ID; expected a UUID"
82                            );
83                        })
84                        .ok()
85                }),
86                auth_token: lookup("SYSTEMPROMPT_AUTH_TOKEN"),
87            },
88        }
89    }
90}
systemprompt_cli/env_overrides.rs

systemprompt_cli/
env_overrides.rs
