use std::env;
use std::fs;
use std::path::{Path, PathBuf};
use serde::{Deserialize, Serialize};
use thiserror::Error;
const DEFAULT_DB_PATH: &str = "~/.mempal/palace.db";
const DEFAULT_EMBED_BACKEND: &str = "model2vec";
#[derive(Debug, Clone, PartialEq, Eq, Deserialize, Serialize)]
#[serde(default)]
pub struct Config {
pub db_path: String,
pub embed: EmbedConfig,
pub context: ContextConfig,
}
impl Default for Config {
fn default() -> Self {
Self {
db_path: DEFAULT_DB_PATH.to_string(),
embed: EmbedConfig::default(),
context: ContextConfig::default(),
}
}
}
impl Config {
pub fn load() -> Result<Self, ConfigError> {
Self::load_from(&default_config_path())
}
pub fn load_from(path: &Path) -> Result<Self, ConfigError> {
match fs::read_to_string(path) {
Ok(contents) => Ok(toml::from_str(&contents)?),
Err(error) if error.kind() == std::io::ErrorKind::NotFound => Ok(Self::default()),
Err(source) => Err(ConfigError::Read {
path: path.to_path_buf(),
source,
}),
}
}
pub fn save_to(&self, path: &Path) -> Result<(), ConfigError> {
if let Some(parent) = path.parent() {
fs::create_dir_all(parent).map_err(|source| ConfigError::Write {
path: parent.to_path_buf(),
source,
})?;
}
let contents = toml::to_string_pretty(self)?;
fs::write(path, contents).map_err(|source| ConfigError::Write {
path: path.to_path_buf(),
source,
})
}
pub fn save_default(&self) -> Result<(), ConfigError> {
self.save_to(&default_config_path())
}
}
#[derive(Debug, Clone, PartialEq, Eq, Deserialize, Serialize)]
#[serde(default)]
pub struct EmbedConfig {
pub backend: String,
pub model: Option<String>,
pub api_endpoint: Option<String>,
pub api_model: Option<String>,
}
impl Default for EmbedConfig {
fn default() -> Self {
Self {
backend: DEFAULT_EMBED_BACKEND.to_string(),
model: None,
api_endpoint: None,
api_model: None,
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Deserialize, Serialize, Default)]
#[serde(default)]
pub struct ContextConfig {
pub include_cards_default: bool,
}
#[derive(Debug, Error)]
pub enum ConfigError {
#[error("failed to read config from {path}")]
Read {
path: PathBuf,
#[source]
source: std::io::Error,
},
#[error("failed to write config to {path}")]
Write {
path: PathBuf,
#[source]
source: std::io::Error,
},
#[error("failed to parse config TOML")]
Parse(#[from] toml::de::Error),
#[error("failed to serialize config TOML")]
Serialize(#[from] toml::ser::Error),
}
fn default_config_path() -> PathBuf {
env::var_os("HOME")
.map(PathBuf::from)
.map(|home| home.join(".mempal").join("config.toml"))
.unwrap_or_else(|| PathBuf::from("~/.mempal/config.toml"))
}