lingshu-tools 0.10.0

Tool registry, ToolHandler trait, and 50+ tool implementations
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

//! Serializes env + `EDGECRAB_HOME` mutations in web search / extract tests.

use std::sync::{Mutex, MutexGuard};
use tempfile::TempDir;

static WEB_CONFIG_TEST_LOCK: Mutex<()> = Mutex::new(());

pub fn web_config_test_lock() -> MutexGuard<'static, ()> {
    WEB_CONFIG_TEST_LOCK
        .lock()
        .unwrap_or_else(|e| e.into_inner())
}

/// Temporarily point `EDGECRAB_HOME` at an isolated `config.yaml`.
pub struct EdgecrabHomeGuard {
    prev_home: Option<String>,
    _dir: TempDir,
}

impl EdgecrabHomeGuard {
    pub fn isolated(yaml: Option<&str>) -> Self {
        let dir = TempDir::new().expect("tempdir");
        let content = yaml.unwrap_or(
            r#"
web_search:
  primary: ""
"#,
        );
        std::fs::write(dir.path().join("config.yaml"), content).expect("write config");
        let prev_home = std::env::var("EDGECRAB_HOME").ok();
        unsafe { std::env::set_var("EDGECRAB_HOME", dir.path()) };
        Self {
            prev_home,
            _dir: dir,
        }
    }
}

impl Drop for EdgecrabHomeGuard {
    fn drop(&mut self) {
        unsafe { std::env::remove_var("EDGECRAB_HOME") };
        if let Some(v) = &self.prev_home {
            unsafe { std::env::set_var("EDGECRAB_HOME", v) };
        }
    }
}