llm_manager/config/
profiles.rs

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

llm_manager/config/
profiles.rs
