fn fingerprint(data: &str) -> String {
const FNV_OFFSET: u64 = 0xcbf2_9ce4_8422_2325;
const FNV_PRIME: u64 = 0x0100_0000_01b3;
let mut hash = FNV_OFFSET;
for byte in data.as_bytes() {
hash ^= u64::from(*byte);
hash = hash.wrapping_mul(FNV_PRIME);
}
format!("{hash:016x}")
}
fn render_managed(contents: &str) -> String {
format!(
"{MANAGED_HEADER_PREFIX}{}\n{contents}",
fingerprint(contents)
)
}
fn parse_managed(text: &str) -> Option<(&str, &str)> {
let rest = text.strip_prefix(MANAGED_HEADER_PREFIX)?;
rest.split_once('\n')
}
pub fn ensure_default_agents() {
ensure_default_agents_in(&agents_dir());
}
pub(crate) fn ensure_default_agents_in(dir: &Path) {
if let Err(err) = crate::utils::fs_perms::create_private_dir_all(dir) {
log::warn!(
"ensure_default_agents: failed to create {}: {err}",
dir.display()
);
return;
}
for (name, contents) in DEFAULT_AGENT_CONFIGS {
let path = dir.join(format!("{name}.toml"));
match std::fs::read_to_string(&path) {
Err(err) if err.kind() == ErrorKind::NotFound => {
if let Err(err) = std::fs::write(&path, render_managed(contents)) {
log::warn!(
"ensure_default_agents: failed to write {}: {err}",
path.display()
);
}
}
Err(err) => {
log::warn!(
"ensure_default_agents: failed to read {}: {err}",
path.display()
);
}
Ok(existing) => {
let Some((recorded_fingerprint, body)) = parse_managed(&existing) else {
continue;
};
if body == *contents {
continue; }
if fingerprint(body) != recorded_fingerprint {
continue;
}
if let Err(err) = std::fs::write(&path, render_managed(contents)) {
log::warn!(
"ensure_default_agents: failed to rewrite stale default {}: {err}",
path.display()
);
} else {
log::info!(
"ensure_default_agents: upgraded stale default {} to the current built-in",
path.display()
);
}
}
}
}
}
#[cfg(test)]
#[path = "agents_tests.rs"]
mod agents_tests;