64 lines
1.8 KiB
Rust
64 lines
1.8 KiB
Rust
use std::path::{Path, PathBuf};
|
|
|
|
use color_eyre::Result;
|
|
|
|
/// ~/.bore directory.
|
|
pub fn bore_dir() -> Result<PathBuf> {
|
|
let home = dirs::home_dir()
|
|
.ok_or_else(|| color_eyre::eyre::eyre!("could not determine home directory"))?;
|
|
Ok(home.join(".bore"))
|
|
}
|
|
|
|
fn secrets_path() -> Result<PathBuf> {
|
|
Ok(bore_dir()?.join("secrets"))
|
|
}
|
|
|
|
pub fn load_secret(server_key: &str) -> Result<Option<String>> {
|
|
let path = secrets_path()?;
|
|
load_entry(&path, server_key)
|
|
}
|
|
|
|
pub fn save_secret(server_key: &str, secret: &str) -> Result<()> {
|
|
let path = secrets_path()?;
|
|
save_entry(&path, server_key, secret)
|
|
}
|
|
|
|
pub fn load_entry(path: &Path, key: &str) -> Result<Option<String>> {
|
|
let contents = match std::fs::read_to_string(path) {
|
|
Ok(c) => c,
|
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(None),
|
|
Err(e) => return Err(e.into()),
|
|
};
|
|
for line in contents.lines() {
|
|
let line = line.trim();
|
|
if line.is_empty() || line.starts_with('#') {
|
|
continue;
|
|
}
|
|
if let Some((k, v)) = line.split_once(' ')
|
|
&& k == key
|
|
{
|
|
return Ok(Some(v.to_string()));
|
|
}
|
|
}
|
|
Ok(None)
|
|
}
|
|
|
|
pub fn save_entry(path: &Path, key: &str, value: &str) -> Result<()> {
|
|
if let Some(parent) = path.parent() {
|
|
std::fs::create_dir_all(parent)?;
|
|
}
|
|
|
|
let mut lines: Vec<String> = match std::fs::read_to_string(path) {
|
|
Ok(c) => c
|
|
.lines()
|
|
.filter(|l| l.split_once(' ').is_none_or(|(k, _)| k != key))
|
|
.map(String::from)
|
|
.collect(),
|
|
Err(e) if e.kind() == std::io::ErrorKind::NotFound => Vec::new(),
|
|
Err(e) => return Err(e.into()),
|
|
};
|
|
lines.push(format!("{key} {value}"));
|
|
std::fs::write(path, lines.join("\n") + "\n")?;
|
|
Ok(())
|
|
}
|