Implement profile management features: add profile selection, saving, and loading functionality; enhance workspace settings with profile support; update UI components for profile handling.

Author: gramistella

src/core/workspace.rs

@@ -5,20 +5,55 @@ use std::{
 
 use serde::{Deserialize, Serialize};
 
-/// On-disk workspace settings for a project folder.
-/// Stored at `<project>/.stitchworkspace/workspace.json`.
+/* ============================ Workspace settings ============================ */
+
 #[derive(Debug, Clone, Serialize, Deserialize, Default)]
 pub struct WorkspaceSettings {
-    pub version: u32, // for future migrations; currently 1
+    pub version: u32,
     pub ext_filter: String,
     pub exclude_dirs: String,
     pub exclude_files: String,
     pub remove_prefix: String,
     pub remove_regex: String,
     pub hierarchy_only: bool,
     pub dirs_only: bool,
+
+    /// Optional name of the currently-selected profile (if any).
+    #[serde(default)]
+    pub current_profile: Option<String>,
+}
+
+/* ================================ Profiles ================================= */
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum ProfileScope {
+    Shared,
+    Local,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, Default)]
+pub struct ProfileSelection {
+    /// Project-relative path using forward slashes.
+    pub path: String,
+    pub state: bool,
+}
+
+#[derive(Debug, Clone, Serialize, Deserialize, Default)]
+pub struct Profile {
+    pub name: String,
+    pub settings: WorkspaceSettings,
+    /// Explicit on/off checks captured relative to project root.
+    pub explicit: Vec<ProfileSelection>,
 }
 
+#[derive(Debug, Clone)]
+pub struct ProfileMeta {
+    pub name: String,
+    pub scope: ProfileScope,
+}
+
+/* ========================= Paths & basic workspace ========================= */
+
 pub fn workspace_dir(project_root: &Path) -> PathBuf {
     project_root.join(".stitchworkspace")
 }
@@ -27,7 +62,6 @@ pub fn workspace_file(project_root: &Path) -> PathBuf {
     workspace_dir(project_root).join("workspace.json")
 }
 
-/// Ensure `.stitchworkspace/` exists; return its path.
 pub fn ensure_workspace_dir(project_root: &Path) -> io::Result<PathBuf> {
     let dir = workspace_dir(project_root);
     if !dir.exists() {
@@ -36,14 +70,55 @@ pub fn ensure_workspace_dir(project_root: &Path) -> io::Result<PathBuf> {
     Ok(dir)
 }
 
-/// Try to load settings; returns `None` if file is missing or invalid.
+/* ============================ Profiles locations ============================ */
+
+fn profiles_shared_dir(project_root: &Path) -> PathBuf {
+    workspace_dir(project_root).join("profiles")
+}
+
+fn profiles_local_dir(project_root: &Path) -> PathBuf {
+    workspace_dir(project_root).join("local").join("profiles")
+}
+
+pub fn ensure_profiles_dirs(project_root: &Path) -> io::Result<()> {
+    fs::create_dir_all(profiles_shared_dir(project_root))?;
+    fs::create_dir_all(profiles_local_dir(project_root))?;
+    Ok(())
+}
+
+fn sanitize_profile_name(name: &str) -> String {
+    // keep it simple & predictable for file names
+    let mut s = name.trim().to_string();
+    if s.is_empty() {
+        s.push_str("unnamed");
+    }
+    s.chars()
+        .map(|c| {
+            if c.is_ascii_alphanumeric() || c == '-' || c == '_' || c == ' ' {
+                c
+            } else {
+                '_'
+            }
+        })
+        .collect()
+}
+
+fn profile_path(project_root: &Path, scope: ProfileScope, name: &str) -> PathBuf {
+    let base = match scope {
+        ProfileScope::Shared => profiles_shared_dir(project_root),
+        ProfileScope::Local => profiles_local_dir(project_root),
+    };
+    base.join(format!("{}.json", sanitize_profile_name(name)))
+}
+
+/* =============================== Workspace IO ============================== */
+
 pub fn load_workspace(project_root: &Path) -> Option<WorkspaceSettings> {
     let path = workspace_file(project_root);
     let data = fs::read(&path).ok()?;
     serde_json::from_slice::<WorkspaceSettings>(&data).ok()
 }
 
-/// Save settings atomically to `workspace.json`.
 pub fn save_workspace(project_root: &Path, settings: &WorkspaceSettings) -> io::Result<()> {
     ensure_workspace_dir(project_root)?;
 
@@ -53,7 +128,82 @@ pub fn save_workspace(project_root: &Path, settings: &WorkspaceSettings) -> io::
     let data = serde_json::to_vec_pretty(settings).map_err(|e| io::Error::other(e.to_string()))?;
 
     fs::write(&tmp, data)?;
-    // Atomic on most platforms when same directory
     fs::rename(&tmp, &path)?;
     Ok(())
 }
+
+/* =============================== Profiles IO =============================== */
+
+pub fn save_profile(
+    project_root: &Path,
+    profile: &Profile,
+    scope: ProfileScope,
+) -> io::Result<()> {
+    ensure_profiles_dirs(project_root)?;
+    let path = profile_path(project_root, scope, &profile.name);
+    let tmp = path.with_extension("tmp");
+    let data = serde_json::to_vec_pretty(profile).map_err(|e| io::Error::other(e.to_string()))?;
+    fs::write(&tmp, data)?;
+    fs::rename(&tmp, &path)?;
+    Ok(())
+}
+
+/// Returns (Profile, Scope) preferring Local if both exist.
+pub fn load_profile(project_root: &Path, name: &str) -> Option<(Profile, ProfileScope)> {
+    let local = profile_path(project_root, ProfileScope::Local, name);
+    if let Ok(bytes) = fs::read(&local) {
+        if let Ok(p) = serde_json::from_slice::<Profile>(&bytes) {
+            return Some((p, ProfileScope::Local));
+        }
+    }
+    let shared = profile_path(project_root, ProfileScope::Shared, name);
+    if let Ok(bytes) = fs::read(&shared) {
+        if let Ok(p) = serde_json::from_slice::<Profile>(&bytes) {
+            return Some((p, ProfileScope::Shared));
+        }
+    }
+    None
+}
+
+/// Lists all profiles found. If a name exists in both scopes, only the Local one is returned.
+pub fn list_profiles(project_root: &Path) -> Vec<ProfileMeta> {
+    fn scan(dir: &Path, scope: ProfileScope, out: &mut Vec<(String, ProfileScope)>) {
+        if let Ok(rd) = fs::read_dir(dir) {
+            for ent in rd.flatten() {
+                if let Some(ext) = ent.path().extension() {
+                    if ext == "json" {
+                        if let Some(os) = ent.path().file_stem() {
+                            let name = os.to_string_lossy().to_string();
+                            out.push((name, scope));
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    let mut raw: Vec<(String, ProfileScope)> = Vec::new();
+    scan(&profiles_shared_dir(project_root), ProfileScope::Shared, &mut raw);
+    scan(&profiles_local_dir(project_root), ProfileScope::Local, &mut raw);
+
+    // keep a single entry per name, preferring Local
+    use std::collections::BTreeMap;
+    let mut by_name: BTreeMap<String, ProfileScope> = BTreeMap::new();
+    for (n, s) in raw {
+        match by_name.get(&n) {
+            None => {
+                by_name.insert(n, s);
+            }
+            Some(prev) => {
+                if *prev == ProfileScope::Shared && s == ProfileScope::Local {
+                    by_name.insert(n, s);
+                }
+            }
+        }
+    }
+
+    by_name
+        .into_iter()
+        .map(|(name, scope)| ProfileMeta { name, scope })
+        .collect()
+}

src/main.rs

@@ -14,14 +14,13 @@ use slint::ComponentHandle;
 use ui::{
     AppState, AppWindow, Row, SelectFromTextDialog, apply_selection_from_text, on_check_updates,
     on_copy_output, on_filter_changed, on_generate_output, on_select_folder, on_toggle_check,
-    on_toggle_expand,
+    on_toggle_expand, on_save_profile_as, on_select_profile, on_save_profile_current
 };
 
 #[cfg(feature = "ui")]
 fn spawn_window(registry: Rc<RefCell<Vec<AppWindow>>>) -> anyhow::Result<()> {
     let app = AppWindow::new()?;
 
-    // Initialize UI properties
     app.set_app_version(env!("CARGO_PKG_VERSION").into());
     app.set_ext_filter("".into());
     app.set_exclude_dirs(".git, node_modules, target, _target, .elan, .lake, .idea, .vscode, _app, .svelte-kit, .sqlx, venv, .venv, __pycache__, LICENSES, fixtures".into());
@@ -37,18 +36,19 @@ fn spawn_window(registry: Rc<RefCell<Vec<AppWindow>>>) -> anyhow::Result<()> {
     app.set_copy_toast_text("".into());
     app.set_output_stats("0 chars • 0 tokens".into());
 
-    // Per-window state
+    // Profiles UI defaults
+    app.set_profiles(slint::ModelRc::new(slint::VecModel::from(Vec::<slint::SharedString>::new())));
+    app.set_selected_profile_index(-1);
+
     let state = Rc::new(RefCell::new(AppState {
         poll_interval_ms: 45_000,
         ..Default::default()
     }));
 
-    // Periodic poll timer: capture a Weak to avoid moving `state` while borrowed
     {
         let app_weak = app.as_weak();
         let interval_ms = { state.borrow().poll_interval_ms };
         let state_weak = Rc::downgrade(&state);
-        // borrow only long enough to call `start`, then drop before closure capture
         {
             let st = state.borrow();
             st.poll_timer.start(
@@ -64,7 +64,6 @@ fn spawn_window(registry: Rc<RefCell<Vec<AppWindow>>>) -> anyhow::Result<()> {
         }
     }
 
-    // UI callbacks (per window)
     {
         let app_weak = app.as_weak();
         let state = Rc::clone(&state);
@@ -120,7 +119,6 @@ fn spawn_window(registry: Rc<RefCell<Vec<AppWindow>>>) -> anyhow::Result<()> {
         });
     }
     {
-        // "Select from Text…" dialog
         let app_weak = app.as_weak();
         let state = Rc::clone(&state);
 
@@ -158,15 +156,42 @@ fn spawn_window(registry: Rc<RefCell<Vec<AppWindow>>>) -> anyhow::Result<()> {
         });
     }
 
-    // Multi-window: hook "New Window" button
+    // Profiles: selection + save + save as
+    {
+        let app_weak = app.as_weak();
+        let state = Rc::clone(&state);
+        app.on_select_profile(move |idx| {
+            if let Some(app) = app_weak.upgrade() {
+                on_select_profile(&app, &state, idx);
+            }
+        });
+    }
+    {
+        let app_weak = app.as_weak();
+        let state = Rc::clone(&state);
+        app.on_save_profile(move || {
+            if let Some(app) = app_weak.upgrade() {
+                on_save_profile_current(&app, &state);
+            }
+        });
+    }
+    {
+        let app_weak = app.as_weak();
+        let state = Rc::clone(&state);
+        app.on_save_profile_as(move || {
+            if let Some(app) = app_weak.upgrade() {
+                on_save_profile_as(&app, &state);
+            }
+        });
+    }
+
     {
         let registry_clone = Rc::clone(&registry);
         app.on_new_window(move || {
             let _ = spawn_window(Rc::clone(&registry_clone));
         });
     }
 
-    // Show this window and keep the handle alive
     app.show()?;
     registry.borrow_mut().push(app);