Committing before copilot touches this

breadpad-shared/src/config.rs

@@ -0,0 +1,180 @@
+use anyhow::Result;
+use serde::{Deserialize, Serialize};
+use std::fs;
+use std::path::PathBuf;
+
+fn default_type_str() -> String { "note".into() }
+fn default_workspace_tag() -> bool { true }
+fn default_snooze_options() -> Vec<String> {
+    vec!["15m".into(), "1h".into(), "tomorrow_morning".into()]
+}
+fn default_archive_after_days() -> i64 { 30 }
+fn default_model_path() -> String { "~/.local/share/breadpad/model/classifier.onnx".into() }
+fn default_tokenizer_path() -> String { "~/.local/share/breadpad/model/tokenizer.json".into() }
+fn default_execution_provider() -> String { "auto".into() }
+fn default_morning_time() -> String { "08:00".into() }
+fn default_missed_grace_minutes() -> i64 { 60 }
+fn default_ollama_endpoint() -> String { "http://localhost:11434".into() }
+fn default_ollama_model() -> String { "llama3.2:3b".into() }
+fn default_ollama_confidence_threshold() -> f32 { 0.6 }
+fn default_ollama_enabled() -> bool { true }
+fn default_calendar_enabled() -> bool { false }
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct Settings {
+    #[serde(default = "default_type_str")]
+    pub default_type: String,
+    #[serde(default = "default_workspace_tag")]
+    pub workspace_tag: bool,
+    #[serde(default = "default_snooze_options")]
+    pub snooze_options: Vec<String>,
+    #[serde(default = "default_archive_after_days")]
+    pub archive_after_days: i64,
+}
+
+impl Default for Settings {
+    fn default() -> Self {
+        Settings {
+            default_type: default_type_str(),
+            workspace_tag: default_workspace_tag(),
+            snooze_options: default_snooze_options(),
+            archive_after_days: default_archive_after_days(),
+        }
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct OllamaConfig {
+    #[serde(default = "default_ollama_endpoint")]
+    pub endpoint: String,
+    #[serde(default = "default_ollama_model")]
+    pub model: String,
+    #[serde(default = "default_ollama_confidence_threshold")]
+    pub confidence_threshold: f32,
+    #[serde(default = "default_ollama_enabled")]
+    pub enabled: bool,
+}
+
+impl Default for OllamaConfig {
+    fn default() -> Self {
+        OllamaConfig {
+            endpoint: default_ollama_endpoint(),
+            model: default_ollama_model(),
+            confidence_threshold: default_ollama_confidence_threshold(),
+            enabled: default_ollama_enabled(),
+        }
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ModelConfig {
+    #[serde(default = "default_model_path")]
+    pub path: String,
+    #[serde(default = "default_tokenizer_path")]
+    pub tokenizer: String,
+    #[serde(default = "default_execution_provider")]
+    pub execution_provider: String,
+    #[serde(default)]
+    pub ollama: OllamaConfig,
+}
+
+impl Default for ModelConfig {
+    fn default() -> Self {
+        ModelConfig {
+            path: default_model_path(),
+            tokenizer: default_tokenizer_path(),
+            execution_provider: default_execution_provider(),
+            ollama: OllamaConfig::default(),
+        }
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct RemindersConfig {
+    #[serde(default = "default_morning_time")]
+    pub default_morning: String,
+    #[serde(default = "default_missed_grace_minutes")]
+    pub missed_grace_minutes: i64,
+}
+
+impl Default for RemindersConfig {
+    fn default() -> Self {
+        RemindersConfig {
+            default_morning: default_morning_time(),
+            missed_grace_minutes: default_missed_grace_minutes(),
+        }
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct CalendarConfig {
+    #[serde(default = "default_calendar_enabled")]
+    pub enabled: bool,
+    #[serde(default)]
+    pub url: String,
+    #[serde(default)]
+    pub username: String,
+    #[serde(default)]
+    pub password: String,
+}
+
+impl Default for CalendarConfig {
+    fn default() -> Self {
+        CalendarConfig {
+            enabled: false,
+            url: String::new(),
+            username: String::new(),
+            password: String::new(),
+        }
+    }
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, Default)]
+pub struct Config {
+    #[serde(default)]
+    pub settings: Settings,
+    #[serde(default)]
+    pub model: ModelConfig,
+    #[serde(default)]
+    pub reminders: RemindersConfig,
+    #[serde(default)]
+    pub calendar: CalendarConfig,
+}
+
+impl Config {
+    pub fn load() -> Result<Self> {
+        let path = config_path();
+        if !path.exists() {
+            let cfg = Config::default();
+            cfg.save()?;
+            return Ok(cfg);
+        }
+        let text = fs::read_to_string(&path)?;
+        let cfg: Config = toml::from_str(&text)?;
+        Ok(cfg)
+    }
+
+    pub fn save(&self) -> Result<()> {
+        let path = config_path();
+        if let Some(parent) = path.parent() {
+            fs::create_dir_all(parent)?;
+        }
+        let text = toml::to_string_pretty(self)?;
+        fs::write(&path, text)?;
+        Ok(())
+    }
+}
+
+pub fn config_path() -> PathBuf {
+    dirs::config_dir()
+        .unwrap_or_else(|| PathBuf::from("~/.config"))
+        .join("breadpad")
+        .join("breadpad.toml")
+}
+
+pub fn style_css_path() -> PathBuf {
+    dirs::config_dir()
+        .unwrap_or_else(|| PathBuf::from("~/.config"))
+        .join("breadpad")
+        .join("style.css")
+}