31 lines
904 B
Rust
31 lines
904 B
Rust
|
use std::collections::HashMap;
|
||
|
|
use std::sync::Mutex;
|
||
|
|
use std::sync::OnceLock;
|
||
|
|
|
||
|
|
use super::job::BashJob;
|
||
|
|
|
||
|
|
pub(crate) fn bash_jobs_map() -> &'static Mutex<HashMap<String, BashJob>> {
|
||
|
|
static JOBS: OnceLock<Mutex<HashMap<String, BashJob>>> = OnceLock::new();
|
||
|
|
JOBS.get_or_init(|| Mutex::new(HashMap::new()))
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn bash_output(id: &str) -> Option<Vec<String>> {
|
||
|
|
let mut map = bash_jobs_map().lock().ok()?;
|
||
|
|
let job = map.get_mut(id)?;
|
||
|
|
let mut lines = Vec::new();
|
||
|
|
while let Some(line) = job.try_read_line() {
|
||
|
|
lines.push(line);
|
||
|
|
}
|
||
|
|
if lines.is_empty() { None } else { Some(lines) }
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn bash_kill(id: &str) -> anyhow::Result<()> {
|
||
|
|
let mut map = bash_jobs_map().lock().map_err(|e| anyhow::anyhow!("lock error: {}", e))?;
|
||
|
|
let job = map.remove(id);
|
||
|
|
if job.is_some() {
|
||
|
|
Ok(())
|
||
|
|
} else {
|
||
|
|
anyhow::bail!("bash job '{}' not found", id)
|
||
|
|
}
|
||
|
|
}
|