selfware 0.6.1

Your personal AI workshop — software you own, software that lasts
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

//! Shared test helpers for env-variable isolation.
//!
//! Both `config::tests` and `cli::tests` mutate `SELFWARE_*` env vars.  When
//! the lib test suite runs with multiple threads (the default), those
//! mutations race.  The `test_env_guard()` helper acquires a global mutex and
//! returns an RAII guard that **clears** every `SELFWARE_*` variable on drop,
//! so even panicking tests don't leave pollution behind.

use std::sync::Mutex;

pub static ENV_MUTEX: Mutex<()> = Mutex::new(());

pub struct EnvGuard(#[allow(dead_code)] std::sync::MutexGuard<'static, ()>);

impl Drop for EnvGuard {
    fn drop(&mut self) {
        for var in &[
            "SELFWARE_CONFIG",
            "SELFWARE_ENDPOINT",
            "SELFWARE_MODEL",
            "SELFWARE_API_KEY",
            "OPENROUTER_API_KEY",
            "SELFWARE_MAX_TOKENS",
            "SELFWARE_TEMPERATURE",
            "SELFWARE_TIMEOUT",
            "SELFWARE_THEME",
            "SELFWARE_LOG_LEVEL",
            "SELFWARE_MODE",
            "SELFWARE_STRICT_PERMISSIONS",
        ] {
            std::env::remove_var(var);
        }
    }
}

/// Lock the global env mutex, clear all `SELFWARE_*` vars, and return a
/// guard that re-clears them on drop.
pub fn clear_env() -> EnvGuard {
    let guard = ENV_MUTEX.lock().unwrap_or_else(|e| e.into_inner());
    for var in &[
        "SELFWARE_CONFIG",
        "SELFWARE_ENDPOINT",
        "SELFWARE_MODEL",
        "SELFWARE_API_KEY",
        "OPENROUTER_API_KEY",
        "SELFWARE_MAX_TOKENS",
        "SELFWARE_TEMPERATURE",
        "SELFWARE_TIMEOUT",
        "SELFWARE_THEME",
        "SELFWARE_LOG_LEVEL",
        "SELFWARE_MODE",
        "SELFWARE_STRICT_PERMISSIONS",
    ] {
        std::env::remove_var(var);
    }
    EnvGuard(guard)
}