use redb::Database;
use std::collections::HashMap;
use std::path::PathBuf;
use std::sync::Mutex;
use std::sync::atomic::{AtomicBool, Ordering};
use tokio::runtime::Runtime;
use tokio::sync::watch;
use crate::errors;
pub struct BuildContext {
runtime: Runtime,
interrupted: AtomicBool,
interrupt_tx: watch::Sender<bool>,
interrupt_rx: watch::Receiver<bool>,
pub(crate) checksum_cache: Mutex<Option<HashMap<PathBuf, String>>>,
pub(crate) mtime_db: Mutex<Option<Database>>,
pub(crate) mtime_enabled: AtomicBool,
pub(crate) max_arg_len: std::sync::atomic::AtomicUsize,
pub(crate) webcache_ttl_secs: std::sync::atomic::AtomicU64,
pub(crate) command_timeout_secs: std::sync::atomic::AtomicU64,
}
impl BuildContext {
pub(crate) fn new() -> Self {
let runtime = Runtime::new().expect(errors::TOKIO_RUNTIME);
let (interrupt_tx, interrupt_rx) = watch::channel(false);
Self {
runtime,
interrupted: AtomicBool::new(false),
interrupt_tx,
interrupt_rx,
checksum_cache: Mutex::new(None),
mtime_db: Mutex::new(None),
mtime_enabled: AtomicBool::new(true),
max_arg_len: std::sync::atomic::AtomicUsize::new(1_000_000),
webcache_ttl_secs: std::sync::atomic::AtomicU64::new(7 * 24 * 60 * 60),
command_timeout_secs: std::sync::atomic::AtomicU64::new(0),
}
}
pub(crate) fn set_command_timeout_secs(&self, n: u64) {
self.command_timeout_secs.store(n, Ordering::Relaxed);
}
pub fn command_timeout(&self) -> Option<std::time::Duration> {
match self.command_timeout_secs.load(Ordering::Relaxed) {
0 => None,
secs => Some(std::time::Duration::from_secs(secs)),
}
}
pub(crate) fn set_mtime_check(&self, enabled: bool) {
self.mtime_enabled.store(enabled, Ordering::Relaxed);
}
pub(crate) fn set_max_arg_len(&self, n: usize) {
self.max_arg_len.store(n, Ordering::Relaxed);
}
pub(crate) fn set_webcache_ttl_secs(&self, n: u64) {
self.webcache_ttl_secs.store(n, Ordering::Relaxed);
}
pub(crate) fn webcache_ttl_secs(&self) -> u64 {
self.webcache_ttl_secs.load(Ordering::Relaxed)
}
pub fn max_arg_len(&self) -> usize {
self.max_arg_len.load(Ordering::Relaxed)
}
pub(crate) const fn runtime(&self) -> &Runtime {
&self.runtime
}
pub(crate) fn is_interrupted(&self) -> bool {
self.interrupted.load(Ordering::SeqCst)
}
pub(crate) fn interrupt(&self) {
self.interrupted.store(true, Ordering::SeqCst);
let _ = self.interrupt_tx.send(true);
}
pub(crate) fn interrupt_receiver(&self) -> watch::Receiver<bool> {
self.interrupt_rx.clone()
}
}