llm_manager/config/
presets.rs

1use std::collections::HashMap;
2use std::path::PathBuf;
3
4use serde::{Deserialize, Serialize};
5
6use crate::config::config_base_dir;
7use crate::config::SystemPromptPreset;
8use crate::config::builtin_system_prompt_presets;
9use crate::config::store::{load_all_from_dir, move_to_unused, save_yaml};
10
11/// Directory for per-preset YAML configs.
12pub fn presets_config_dir() -> PathBuf {
13    config_base_dir()
14        .join("llm-manager")
15        .join("presets")
16}
17
18/// Directory for unused (deleted) preset configs.
19pub fn unused_presets_dir() -> PathBuf {
20    config_base_dir()
21        .join("llm-manager")
22        .join("unused_presets")
23}
24
25/// System prompt preset store — manages per-preset YAML configs.
26#[derive(Debug, Clone, Serialize, Deserialize)]
27pub struct PresetStore {
28    presets_dir: PathBuf,
29    unused_dir: PathBuf,
30    cache: HashMap<String, SystemPromptPreset>,
31}
32
33impl PresetStore {
34    pub fn new() -> Self {
35        let presets_dir = presets_config_dir();
36        let unused_dir = unused_presets_dir();
37        let cache = load_all_from_dir(&presets_dir);
38        Self {
39            presets_dir,
40            unused_dir,
41            cache,
42        }
43    }
44
45    /// Get all user-defined presets (excluding built-ins).
46    pub fn user_presets(&self) -> Vec<SystemPromptPreset> {
47        let builtin = builtin_system_prompt_presets();
48        self.cache
49            .values()
50            .filter(|p| !builtin.iter().any(|b| b.name == p.name))
51            .cloned()
52            .collect()
53    }
54
55    /// Save (or update) a preset.
56    pub fn save(&mut self, preset: &SystemPromptPreset) {
57        save_yaml(&preset.name, preset, &self.presets_dir, &self.unused_dir);
58        self.cache.insert(preset.name.clone(), preset.clone());
59    }
60
61    /// Insert a built-in preset into the in-memory cache only (no disk I/O).
62    pub fn insert_builtin(&mut self, preset: SystemPromptPreset) {
63        self.cache.insert(preset.name.clone(), preset);
64    }
65
66    /// Delete a preset by moving it to the unused directory.
67    pub fn delete(&mut self, name: &str) -> bool {
68        let builtin = builtin_system_prompt_presets();
69        if builtin.iter().any(|b| b.name == name) {
70            return false;
71        }
72        move_to_unused(name, &self.presets_dir, &self.unused_dir);
73        self.cache.remove(name);
74        true
75    }
76
77    /// Get a preset by name.
78    pub fn get(&self, name: &str) -> Option<&SystemPromptPreset> {
79        self.cache.get(name)
80    }
81
82    /// Get all presets (built-in + user).
83    pub fn all(&self) -> Vec<SystemPromptPreset> {
84        let builtin = builtin_system_prompt_presets();
85        let mut all: Vec<SystemPromptPreset> = builtin.clone();
86        for p in self.cache.values() {
87            if !builtin.iter().any(|b| b.name == p.name) {
88                all.push(p.clone());
89            }
90        }
91        all
92    }
93}
94
95impl Default for PresetStore {
96    fn default() -> Self {
97        Self::new()
98    }
99}
llm_manager/config/presets.rs

llm_manager/config/
presets.rs
