Skip to main content

orch/config/
mod.rs

1//! Config reader — loads YAML config files and resolves dot-separated keys.
2//!
3//! Reads `~/.orch/config.yml` (global) and `.orch.yml` (project).
4//! Project config overrides global config for the same key.
5//!
6//! Features:
7//! - In-memory cache: parsed YAML is cached for the process lifetime
8//! - File watching: uses notify crate to watch for config file changes
9//! - Hot reload: a changed file is re-parsed into the cache; one that does not
10//!   parse leaves the last good values in place
11//! - Change notifications: subscribers receive events when config files change
12
13pub mod projects_yaml;
14
15use anyhow::Context;
16use notify::{Config, RecommendedWatcher, RecursiveMode, Watcher};
17use std::collections::{HashMap, HashSet};
18use std::path::{Path, PathBuf};
19use std::sync::{Arc, Mutex, RwLock};
20use std::time::Duration;
21use tokio::sync::broadcast;
22
23/// Cached YAML values — parsed once per file, reused for all key lookups.
24/// Protected by RwLock for concurrent read access.
25static CACHE: std::sync::LazyLock<RwLock<HashMap<PathBuf, serde_norway::Value>>> =
26    std::sync::LazyLock::new(|| RwLock::new(HashMap::new()));
27
28/// Files currently being watched for changes.
29static WATCHER: std::sync::LazyLock<RwLock<HashMap<PathBuf, ()>>> =
30    std::sync::LazyLock::new(|| RwLock::new(HashMap::new()));
31
32/// Global file watcher instance (started on first use).
33static FILE_WATCHER: std::sync::LazyLock<Arc<Mutex<Option<RecommendedWatcher>>>> =
34    std::sync::LazyLock::new(|| Arc::new(Mutex::new(None)));
35
36/// Config files whose content on disk does not read or parse right now.
37/// Tracked so the parse error is logged once per breakage, not once per
38/// watcher event.
39static BAD_FILES: std::sync::LazyLock<RwLock<HashSet<PathBuf>>> =
40    std::sync::LazyLock::new(|| RwLock::new(HashSet::new()));
41
42/// Broadcast sender for config change notifications.
43/// Sends the path of the changed config file.
44static CHANGE_TX: std::sync::LazyLock<broadcast::Sender<PathBuf>> =
45    std::sync::LazyLock::new(|| {
46        let (tx, _) = broadcast::channel(16);
47        tx
48    });
49
50/// Subscribe to config change notifications.
51///
52/// Returns a receiver that fires whenever a watched config file changes.
53/// The receiver yields the path of the changed file.
54pub fn subscribe() -> broadcast::Receiver<PathBuf> {
55    CHANGE_TX.subscribe()
56}
57
58/// Pre-populate the config cache using async I/O.
59///
60/// Call once at engine startup so subsequent synchronous [`get`] / [`get_list`]
61/// calls always hit the in-memory cache and never block a Tokio thread with
62/// `std::fs::read_to_string`.
63pub async fn warm_cache() -> anyhow::Result<()> {
64    let global_path = global_config_path()?;
65    if global_path.exists() {
66        match tokio::fs::read_to_string(&global_path).await {
67            Ok(content) => {
68                if let Ok(parsed) = serde_norway::from_str::<serde_norway::Value>(&content) {
69                    if let Ok(mut cache) = CACHE.write() {
70                        cache.insert(global_path.clone(), parsed);
71                    }
72                    watch_file(&global_path);
73                }
74            }
75            Err(e) => tracing::warn!("failed to warm global config cache: {e}"),
76        }
77    }
78
79    let project_path = std::env::current_dir()
80        .unwrap_or_else(|_| PathBuf::from("."))
81        .join(".orch.yml");
82    if project_path.exists() {
83        match tokio::fs::read_to_string(&project_path).await {
84            Ok(content) => {
85                if let Ok(parsed) = serde_norway::from_str::<serde_norway::Value>(&content) {
86                    if let Ok(mut cache) = CACHE.write() {
87                        cache.insert(project_path.clone(), parsed);
88                    }
89                    watch_file(&project_path);
90                }
91            }
92            Err(e) => tracing::warn!("failed to warm project config cache: {e}"),
93        }
94    }
95
96    Ok(())
97}
98
99/// Clear the entire config cache. Exposed for use by integration tests that
100/// need to run in an isolated config environment (e.g. a temp dir with no
101/// project `.orch.yml`).
102#[cfg(test)]
103pub fn clear_test_cache() {
104    if let Ok(mut cache) = CACHE.write() {
105        cache.clear();
106    }
107}
108
109/// Re-read a changed config file and notify subscribers.
110///
111/// A file that cannot be read or parsed leaves the last good value in the
112/// cache and sends no notification, so a broken or half-written edit cannot
113/// reset the running config to defaults.
114///
115/// Returns true when this call logged a newly invalid file, so repeated
116/// watcher events for one breakage produce one error line
117fn reload_file(path: &PathBuf) -> bool {
118    let parsed = std::fs::read_to_string(path)
119        .map_err(|e| e.to_string())
120        .and_then(|content| {
121            serde_norway::from_str::<serde_norway::Value>(&content).map_err(|e| e.to_string())
122        });
123
124    match parsed {
125        Ok(value) => {
126            if let Ok(mut cache) = CACHE.write() {
127                cache.insert(path.clone(), value);
128            }
129            if let Ok(mut bad) = BAD_FILES.write() {
130                if bad.remove(path) {
131                    tracing::info!(path = %path.display(), "config file parses again, reloaded");
132                }
133            }
134            tracing::debug!("config cache refreshed for: {}", path.display());
135            // Notify subscribers (ignore error if no active receivers)
136            let _ = CHANGE_TX.send(path.clone());
137            false
138        }
139        Err(err) => {
140            let first_failure = BAD_FILES
141                .write()
142                .map(|mut bad| bad.insert(path.clone()))
143                .unwrap_or(false);
144            if first_failure {
145                tracing::error!(
146                    path = %path.display(),
147                    %err,
148                    "config file is invalid, keeping the last good values"
149                );
150            }
151            first_failure
152        }
153    }
154}
155
156/// Start the file watcher if not already running.
157fn ensure_watcher() {
158    let mut watcher_guard = FILE_WATCHER.lock().unwrap_or_else(|e| e.into_inner());
159    if watcher_guard.is_some() {
160        return;
161    }
162
163    // Create the watcher
164    let watcher = RecommendedWatcher::new(
165        move |res: Result<notify::Event, notify::Error>| {
166            if let Ok(event) = res {
167                if !is_content_change(&event.kind) {
168                    return;
169                }
170                for path in event.paths {
171                    // Check if it's a config file we care about
172                    let filename = path.file_name().and_then(|n| n.to_str()).unwrap_or("");
173
174                    if filename == "config.yml" || filename == ".orch.yml" {
175                        let _ = reload_file(&path);
176                    }
177                }
178            }
179        },
180        Config::default().with_poll_interval(Duration::from_secs(2)),
181    );
182
183    if let Ok(w) = watcher {
184        *watcher_guard = Some(w);
185    }
186}
187
188// inotify reports every read as an access event, and reload_file reads the file, so reacting to
189// access events reloads forever on Linux
190fn is_content_change(kind: &notify::EventKind) -> bool {
191    kind.is_create() || kind.is_modify() || kind.is_remove()
192}
193
194/**
195 * Register a config file for change watching. `get` / `get_list` watch the files they read, but a
196 * project's `.orch.yml` is read directly by `get_projects_with_paths`, so the caller that knows the
197 * project list registers it here. A missing path is skipped: notify cannot watch one, and a project
198 * may be configured through the legacy `.orchestrator.yml` name or the global `gh.repo` fallback,
199 * which has no project directory at all
200 **/
201pub fn watch(path: &Path) {
202    if !path.exists() {
203        tracing::debug!(path = %path.display(), "no config file to watch here");
204        return;
205    }
206    watch_file(&path.to_path_buf());
207}
208
209/// Watch a config file for changes.
210fn watch_file(path: &PathBuf) {
211    // First ensure watcher is running
212    ensure_watcher();
213
214    // Track that we want to watch this file
215    if let Ok(mut watched) = WATCHER.write() {
216        if !watched.contains_key(path) {
217            watched.insert(path.clone(), ());
218
219            // Add to notify watcher
220            if let Ok(mut guard) = FILE_WATCHER.lock() {
221                if let Some(ref mut watcher) = *guard {
222                    match watcher.watch(path, RecursiveMode::NonRecursive) {
223                        Ok(()) => tracing::debug!("now watching config file: {}", path.display()),
224                        Err(e) => tracing::warn!(
225                            "failed to register config file watcher for {}: {}",
226                            path.display(),
227                            e
228                        ),
229                    }
230                }
231            }
232        }
233    }
234}
235
236fn read_file_to_string(path: &Path) -> anyhow::Result<String> {
237    match tokio::runtime::Handle::try_current() {
238        Ok(handle) if handle.runtime_flavor() == tokio::runtime::RuntimeFlavor::MultiThread => {
239            Ok(tokio::task::block_in_place(|| {
240                std::fs::read_to_string(path)
241            })?)
242        }
243        _ => Ok(std::fs::read_to_string(path)?),
244    }
245}
246
247fn load_cached_root(path: &PathBuf) -> anyhow::Result<serde_norway::Value> {
248    if let Ok(cache) = CACHE.read() {
249        if let Some(cached) = cache.get(path) {
250            return Ok(cached.clone());
251        }
252    }
253
254    let content =
255        read_file_to_string(path).with_context(|| format!("reading {}", path.display()))?;
256    let parsed: serde_norway::Value =
257        serde_norway::from_str(&content).with_context(|| format!("parsing {}", path.display()))?;
258
259    if let Ok(mut cache) = CACHE.write() {
260        cache.insert(path.clone(), parsed.clone());
261    }
262
263    Ok(parsed)
264}
265
266/// Resolve the global config path: `~/.orch/config.yml`
267fn global_config_path() -> anyhow::Result<PathBuf> {
268    crate::home::config_path()
269}
270
271/// Get a config value as a list of strings by dot-separated key.
272///
273/// Supports both YAML sequences and comma-separated strings:
274/// ```yaml
275/// workflow:
276///   allowed_tools:
277///     - Edit
278///     - Write
279///     - git
280/// ```
281/// or:
282/// ```yaml
283/// workflow:
284///   allowed_tools: "Edit,Write,git"
285/// ```
286///
287/// Returns empty vec if key not found.
288pub fn get_list(key: &str) -> anyhow::Result<Vec<String>> {
289    // Try project config first
290    // Use an absolute path so the cache key is stable across tests/processes
291    // that change the current working directory.
292    let project_path = std::env::current_dir()
293        .unwrap_or_else(|_| PathBuf::from("."))
294        .join(".orch.yml");
295    if project_path.exists() {
296        if let Ok(val) = resolve_list(&project_path, key) {
297            if !val.is_empty() {
298                return Ok(val);
299            }
300        }
301    }
302
303    // Fall back to global config
304    let global_path = global_config_path()?;
305    if global_path.exists() {
306        return resolve_list(&global_path, key);
307    }
308
309    Ok(vec![])
310}
311
312/// Resolve a dot-separated key as a list from a YAML file.
313fn resolve_list(path: &PathBuf, key: &str) -> anyhow::Result<Vec<String>> {
314    watch_file(path);
315
316    let root = load_cached_root(path)?;
317
318    extract_list(&root, key)
319}
320
321/// Extract a list of strings from a YAML tree by dot-separated key.
322fn extract_list(root: &serde_norway::Value, key: &str) -> anyhow::Result<Vec<String>> {
323    let mut current = root;
324    for part in key.split('.') {
325        match current.get(part) {
326            Some(v) => current = v,
327            None => return Ok(vec![]),
328        }
329    }
330
331    match current {
332        serde_norway::Value::Sequence(seq) => Ok(seq
333            .iter()
334            .filter_map(|item| item.as_str().map(String::from))
335            .collect()),
336        serde_norway::Value::String(s) => {
337            // Support comma-separated string format
338            Ok(s.split(',')
339                .map(|s| s.trim().to_string())
340                .filter(|s| !s.is_empty())
341                .collect())
342        }
343        _ => Ok(vec![]),
344    }
345}
346
347/// Get a config value by dot-separated key (e.g. "agents.claude.model").
348///
349/// Lookup order:
350/// 1. `.orch.yml` in the current directory (project config)
351/// 2. `~/.orch/config.yml` (global config)
352///
353/// Returns the first match as a string.
354/// Files are parsed once and cached for the process lifetime.
355pub fn get(key: &str) -> anyhow::Result<String> {
356    // Try project config first
357    // Use an absolute path so the cache key is stable across tests/processes
358    // that change the current working directory.
359    let project_path = std::env::current_dir()
360        .unwrap_or_else(|_| PathBuf::from("."))
361        .join(".orch.yml");
362    if project_path.exists() {
363        if let Ok(val) = resolve_key(&project_path, key) {
364            return Ok(val);
365        }
366    }
367
368    // Fall back to global config
369    let global_path = global_config_path()?;
370    if global_path.exists() {
371        return resolve_key(&global_path, key);
372    }
373
374    anyhow::bail!("config key not found: {key}")
375}
376
377/// Get all project paths from the global config `projects:` list.
378///
379/// Returns a list of absolute paths to project directories.
380/// Each directory should contain a `.orch.yml` with `gh.repo`.
381pub fn get_project_paths() -> anyhow::Result<Vec<String>> {
382    let global_path = global_config_path()?;
383
384    if global_path.exists() {
385        if let Ok(paths) = resolve_projects_list(&global_path, "projects") {
386            if !paths.is_empty() {
387                return Ok(paths);
388            }
389        }
390    }
391
392    Ok(vec![])
393}
394
395/// Like `get_projects()` but also returns the project directory path.
396///
397/// Returns `(repo_slug, project_dir)` pairs. Used by the engine to resolve
398/// per-project config (e.g., jobs from each project's `.orch.yml`).
399pub fn get_projects_with_paths() -> anyhow::Result<Vec<(String, std::path::PathBuf)>> {
400    let paths = get_project_paths()?;
401
402    if !paths.is_empty() {
403        let mut results = Vec::new();
404        for path_str in &paths {
405            let path = std::path::PathBuf::from(path_str);
406            let orch_yml = path.join(".orch.yml");
407
408            let config_file = if orch_yml.exists() {
409                orch_yml
410            } else {
411                let legacy = path.join(".orchestrator.yml");
412                if legacy.exists() {
413                    legacy
414                } else {
415                    continue;
416                }
417            };
418
419            if let Ok(content) = std::fs::read_to_string(&config_file) {
420                if let Ok(doc) = serde_norway::from_str::<serde_norway::Value>(&content) {
421                    if let Some(repo) = doc
422                        .get("gh")
423                        .and_then(|gh| gh.get("repo"))
424                        .and_then(|r| r.as_str())
425                    {
426                        results.push((repo.to_string(), path));
427                    }
428                }
429            }
430        }
431
432        if !results.is_empty() {
433            return Ok(results);
434        }
435    }
436
437    // Fall back to global gh.repo (no project dir available)
438    if let Ok(repo) = get("gh.repo") {
439        if !repo.is_empty() {
440            return Ok(vec![(repo, std::path::PathBuf::from("."))]);
441        }
442    }
443
444    Ok(vec![])
445}
446
447/// Resolve the repo slug for a specific project path.
448///
449/// Reads `gh.repo` from the project's `.orch.yml`.
450pub fn get_repo_for_project(project_path: &std::path::Path) -> anyhow::Result<String> {
451    let orch_yml = project_path.join(".orch.yml");
452    let config_file = if orch_yml.exists() {
453        orch_yml
454    } else {
455        let legacy = project_path.join(".orchestrator.yml");
456        if legacy.exists() {
457            legacy
458        } else {
459            anyhow::bail!("no .orch.yml found in {}", project_path.display());
460        }
461    };
462
463    let content = std::fs::read_to_string(&config_file)?;
464    let doc: serde_norway::Value = serde_norway::from_str(&content)?;
465
466    doc.get("gh")
467        .and_then(|gh| gh.get("repo"))
468        .and_then(|r| r.as_str())
469        .map(String::from)
470        .ok_or_else(|| anyhow::anyhow!("gh.repo not set in {}", config_file.display()))
471}
472
473/// Resolve project path from CWD by finding the nearest `.orch.yml`.
474///
475/// Walks up from the current directory looking for `.orch.yml` or legacy `.orchestrator.yml`.
476pub fn find_project_root() -> anyhow::Result<std::path::PathBuf> {
477    let mut dir = std::env::current_dir()?;
478
479    loop {
480        if dir.join(".orch.yml").exists() || dir.join(".orchestrator.yml").exists() {
481            return Ok(dir);
482        }
483
484        if !dir.pop() {
485            break;
486        }
487    }
488
489    anyhow::bail!("no .orch.yml found in current directory or parents — run `orch init`")
490}
491
492/// Resolve the repo slug for the current project (from CWD).
493///
494/// Walks up from CWD to find `.orch.yml`, reads `gh.repo` from it.
495/// Falls back to global `gh.repo` for backward compatibility.
496pub fn get_current_repo() -> anyhow::Result<String> {
497    // Try to find project root from CWD
498    if let Ok(project_root) = find_project_root() {
499        if let Ok(repo) = get_repo_for_project(&project_root) {
500            return Ok(repo);
501        }
502    }
503
504    // Fall back to per-project config in CWD (.orch.yml)
505    if let Ok(repo) = get("gh.repo") {
506        if !repo.is_empty() {
507            return Ok(repo);
508        }
509    }
510
511    // Legacy fallback
512    if let Ok(repo) = get("repo") {
513        if !repo.is_empty() {
514            return Ok(repo);
515        }
516    }
517
518    anyhow::bail!("could not determine repo — ensure .orch.yml has gh.repo set")
519}
520
521/// Skill repository configuration.
522#[derive(Debug, Clone)]
523pub struct SkillConfig {
524    /// GitHub repository in "owner/repo" format.
525    pub repo: String,
526}
527
528/// Get skills configuration from the global config.
529///
530/// Reads the `skills:` list from `~/.orch/config.yml`:
531/// ```yaml
532/// skills:
533///   - repo: "owner/skill-repo"
534///   - repo: "owner/another-repo"
535/// ```
536///
537/// Returns an empty vector if no skills are configured.
538pub fn get_skills() -> anyhow::Result<Vec<SkillConfig>> {
539    let global_path = global_config_path()?;
540
541    if !global_path.exists() {
542        return Ok(vec![]);
543    }
544
545    let root = load_cached_root(&global_path)?;
546
547    extract_skills_list(&root)
548}
549
550/// Extract skills list from a YAML tree.
551fn extract_skills_list(root: &serde_norway::Value) -> anyhow::Result<Vec<SkillConfig>> {
552    let Some(serde_norway::Value::Sequence(seq)) = root.get("skills") else {
553        return Ok(vec![]);
554    };
555
556    let mut skills = Vec::new();
557    for item in seq {
558        if let serde_norway::Value::Mapping(map) = item {
559            let repo = map
560                .get("repo")
561                .and_then(|v| v.as_str())
562                .map(String::from)
563                .unwrap_or_default();
564
565            if !repo.is_empty() {
566                skills.push(SkillConfig { repo });
567            }
568        }
569    }
570
571    Ok(skills)
572}
573
574/// Resolve a projects list from a config file.
575fn resolve_projects_list(path: &PathBuf, key: &str) -> anyhow::Result<Vec<String>> {
576    // First, ensure we're watching this file for changes
577    watch_file(path);
578
579    let root = load_cached_root(path)?;
580
581    // Try to extract projects list - if key doesn't exist, return empty vector
582    match extract_projects_list(&root, key) {
583        Ok(projects) => Ok(projects),
584        Err(_) => Ok(vec![]),
585    }
586}
587
588/// Extract a projects list from a YAML tree.
589fn extract_projects_list(root: &serde_norway::Value, key: &str) -> anyhow::Result<Vec<String>> {
590    let mut current = root;
591    for part in key.split('.') {
592        // If key doesn't exist, return empty vector
593        let next = match current.get(part) {
594            Some(v) => v,
595            None => return Ok(vec![]),
596        };
597        current = next;
598    }
599
600    match current {
601        serde_norway::Value::Sequence(seq) => {
602            let mut projects = Vec::new();
603            for item in seq {
604                if let serde_norway::Value::String(s) = item {
605                    projects.push(s.clone());
606                } else if let serde_norway::Value::Mapping(map) = item {
607                    // Support map format: { repo: "owner/repo", project_dir: "/path" }
608                    if let Some(serde_norway::Value::String(repo)) = map.get("repo") {
609                        projects.push(repo.clone());
610                    }
611                }
612            }
613            Ok(projects)
614        }
615        _ => Ok(vec![]),
616    }
617}
618
619/// Resolve a dot-separated key from a YAML file.
620///
621/// Caches the parsed YAML so repeated lookups don't re-read disk.
622/// Sets up file watching for hot reload when first loading.
623fn resolve_key(path: &PathBuf, key: &str) -> anyhow::Result<String> {
624    // First, ensure we're watching this file for changes
625    watch_file(path);
626
627    let root = load_cached_root(path)?;
628
629    extract_value(&root, key)
630}
631
632/// Extract a value from a YAML tree by dot-separated key.
633fn extract_value(root: &serde_norway::Value, key: &str) -> anyhow::Result<String> {
634    let mut current = root;
635    for part in key.split('.') {
636        current = current
637            .get(part)
638            .with_context(|| format!("key not found: {key}"))?;
639    }
640
641    match current {
642        serde_norway::Value::String(s) => Ok(s.clone()),
643        serde_norway::Value::Number(n) => Ok(n.to_string()),
644        serde_norway::Value::Bool(b) => Ok(b.to_string()),
645        serde_norway::Value::Null => Ok(String::new()),
646        _ => Ok(serde_norway::to_string(current)?),
647    }
648}
649
650#[cfg(test)]
651mod tests {
652    use super::*;
653    use std::io::Write;
654
655    fn write_yaml(dir: &std::path::Path, name: &str, content: &str) -> PathBuf {
656        let path = dir.join(name);
657        let mut f = std::fs::File::create(&path).unwrap();
658        f.write_all(content.as_bytes()).unwrap();
659        path
660    }
661
662    #[test]
663    fn resolve_simple_key() {
664        let dir = tempfile::tempdir().unwrap();
665        let path = write_yaml(dir.path(), "config.yml", "repo: owner/repo\n");
666        let val = resolve_key(&path, "repo").unwrap();
667        assert_eq!(val, "owner/repo");
668    }
669
670    #[test]
671    fn resolve_nested_key() {
672        let dir = tempfile::tempdir().unwrap();
673        let path = write_yaml(
674            dir.path(),
675            "config.yml",
676            "agents:\n  claude:\n    model: opus\n",
677        );
678        let val = resolve_key(&path, "agents.claude.model").unwrap();
679        assert_eq!(val, "opus");
680    }
681
682    #[test]
683    fn resolve_boolean_value() {
684        let dir = tempfile::tempdir().unwrap();
685        let path = write_yaml(dir.path(), "config.yml", "enabled: true\n");
686        let val = resolve_key(&path, "enabled").unwrap();
687        assert_eq!(val, "true");
688    }
689
690    #[test]
691    fn resolve_number_value() {
692        let dir = tempfile::tempdir().unwrap();
693        let path = write_yaml(dir.path(), "config.yml", "timeout: 300\n");
694        let val = resolve_key(&path, "timeout").unwrap();
695        assert_eq!(val, "300");
696    }
697
698    #[test]
699    fn resolve_missing_key_errors() {
700        let dir = tempfile::tempdir().unwrap();
701        let path = write_yaml(dir.path(), "config.yml", "repo: owner/repo\n");
702        let result = resolve_key(&path, "missing.key");
703        assert!(result.is_err());
704    }
705
706    #[test]
707    fn resolve_missing_file_errors() {
708        let path = PathBuf::from("/nonexistent/config.yml");
709        let result = resolve_key(&path, "repo");
710        assert!(result.is_err());
711    }
712
713    #[test]
714    fn resolve_projects_list_string_array() {
715        let dir = tempfile::tempdir().unwrap();
716        let path = write_yaml(
717            dir.path(),
718            "config.yml",
719            "projects:\n  - owner/repo1\n  - owner/repo2\n",
720        );
721        let projects = resolve_projects_list(&path, "projects").unwrap();
722        assert_eq!(projects, vec!["owner/repo1", "owner/repo2"]);
723    }
724
725    #[test]
726    fn resolve_projects_list_map_array() {
727        let dir = tempfile::tempdir().unwrap();
728        let path = write_yaml(
729            dir.path(),
730            "config.yml",
731            "projects:\n  - repo: owner/repo1\n    project_dir: /path/to/repo1\n  - repo: owner/repo2\n    project_dir: /path/to/repo2\n",
732        );
733        let projects = resolve_projects_list(&path, "projects").unwrap();
734        assert_eq!(projects, vec!["owner/repo1", "owner/repo2"]);
735    }
736
737    #[test]
738    fn resolve_projects_list_fallback_to_repo() {
739        let dir = tempfile::tempdir().unwrap();
740        let path = write_yaml(dir.path(), "config.yml", "repo: owner/single-repo\n");
741        let projects = resolve_projects_list(&path, "projects").unwrap();
742        assert!(projects.is_empty());
743    }
744
745    #[test]
746    fn extract_projects_list_from_sequence() {
747        let yaml =
748            serde_norway::from_str::<serde_norway::Value>("projects:\n  - a\n  - b\n  - c\n")
749                .unwrap();
750        let projects = extract_projects_list(&yaml, "projects").unwrap();
751        assert_eq!(projects, vec!["a", "b", "c"]);
752    }
753
754    #[test]
755    fn extract_projects_list_from_mapping_sequence() {
756        let yaml =
757            serde_norway::from_str::<serde_norway::Value>("projects:\n  - repo: a\n  - repo: b\n")
758                .unwrap();
759        let projects = extract_projects_list(&yaml, "projects").unwrap();
760        assert_eq!(projects, vec!["a", "b"]);
761    }
762
763    #[test]
764    fn extract_projects_list_empty_for_missing() {
765        let yaml = serde_norway::from_str::<serde_norway::Value>("foo: bar").unwrap();
766        let projects = extract_projects_list(&yaml, "projects").unwrap();
767        assert!(projects.is_empty());
768    }
769
770    #[test]
771    fn subscribe_receives_change_notifications() {
772        let mut rx = subscribe();
773        let dir = tempfile::tempdir().unwrap();
774        let path = write_yaml(dir.path(), "config.yml", "repo: owner/repo\n");
775
776        // Simulate the watcher firing on a file change
777        assert!(!reload_file(&path));
778
779        // Drain until we find our path (other tests may send on the shared channel)
780        loop {
781            match rx.try_recv() {
782                Ok(received_path) if received_path == path => break,
783                Ok(_) => continue, // skip notifications from other tests
784                Err(e) => panic!("expected config change notification, got error: {e:?}"),
785            }
786        }
787    }
788
789    #[test]
790    fn subscribe_multiple_receivers() {
791        let mut rx1 = subscribe();
792        let mut rx2 = subscribe();
793        let dir = tempfile::tempdir().unwrap();
794        let path = write_yaml(dir.path(), "config.yml", "repo: owner/repo\n");
795
796        assert!(!reload_file(&path));
797
798        // Drain until we find our path on both receivers
799        let find_path = |rx: &mut tokio::sync::broadcast::Receiver<PathBuf>, expected: &PathBuf| loop {
800            match rx.try_recv() {
801                Ok(p) if &p == expected => return,
802                Ok(_) => continue,
803                Err(e) => panic!("expected config change notification, got error: {e:?}"),
804            }
805        };
806        find_path(&mut rx1, &path);
807        find_path(&mut rx2, &path);
808    }
809
810    #[test]
811    fn resolve_list_from_sequence() {
812        let dir = tempfile::tempdir().unwrap();
813        let path = write_yaml(
814            dir.path(),
815            "config.yml",
816            "workflow:\n  allowed_tools:\n    - Edit\n    - Write\n    - git\n    - npm\n",
817        );
818        let list = resolve_list(&path, "workflow.allowed_tools").unwrap();
819        assert_eq!(list, vec!["Edit", "Write", "git", "npm"]);
820    }
821
822    #[test]
823    fn resolve_list_from_comma_string() {
824        let dir = tempfile::tempdir().unwrap();
825        let path = write_yaml(
826            dir.path(),
827            "config.yml",
828            "workflow:\n  allowed_tools: \"Edit,Write,git,npm\"\n",
829        );
830        let list = resolve_list(&path, "workflow.allowed_tools").unwrap();
831        assert_eq!(list, vec!["Edit", "Write", "git", "npm"]);
832    }
833
834    #[test]
835    fn resolve_list_missing_key_returns_empty() {
836        let dir = tempfile::tempdir().unwrap();
837        let path = write_yaml(dir.path(), "config.yml", "foo: bar\n");
838        let list = resolve_list(&path, "workflow.allowed_tools").unwrap();
839        assert!(list.is_empty());
840    }
841
842    #[test]
843    fn extract_skills_list_from_sequence() {
844        let yaml = serde_norway::from_str::<serde_norway::Value>(
845            "skills:\n  - repo: owner/skills1\n  - repo: owner/skills2\n",
846        )
847        .unwrap();
848        let skills = extract_skills_list(&yaml).unwrap();
849        assert_eq!(skills.len(), 2);
850        assert_eq!(skills[0].repo, "owner/skills1");
851        assert_eq!(skills[1].repo, "owner/skills2");
852    }
853
854    #[test]
855    fn extract_skills_list_single_repo() {
856        let yaml =
857            serde_norway::from_str::<serde_norway::Value>("skills:\n  - repo: owner/skills\n")
858                .unwrap();
859        let skills = extract_skills_list(&yaml).unwrap();
860        assert_eq!(skills.len(), 1);
861        assert_eq!(skills[0].repo, "owner/skills");
862    }
863
864    /**
865     * Read a key straight out of the cache. The reload tests must not call `resolve_key` or
866     * `resolve_projects_list`: those register the file with the real notify watcher, whose thread
867     * then races the test for the first `reload_file` call on a broken edit
868     **/
869    fn cached_value(path: &PathBuf, key: &str) -> String {
870        extract_value(&load_cached_root(path).unwrap(), key).unwrap()
871    }
872
873    fn cached_projects(path: &PathBuf) -> Vec<String> {
874        extract_projects_list(&load_cached_root(path).unwrap(), "projects").unwrap()
875    }
876
877    #[test]
878    fn reading_a_config_file_does_not_trigger_a_reload() {
879        use notify::event::{AccessKind, AccessMode, CreateKind, DataChange, ModifyKind};
880        assert!(!is_content_change(&notify::EventKind::Access(
881            AccessKind::Open(AccessMode::Read)
882        )));
883        assert!(!is_content_change(&notify::EventKind::Access(
884            AccessKind::Close(AccessMode::Read)
885        )));
886        assert!(is_content_change(&notify::EventKind::Modify(
887            ModifyKind::Data(DataChange::Content)
888        )));
889        assert!(is_content_change(&notify::EventKind::Create(
890            CreateKind::File
891        )));
892    }
893
894    #[test]
895    fn invalid_edit_keeps_last_good_values_and_sends_no_notification() {
896        let dir = tempfile::tempdir().unwrap();
897        let path = write_yaml(dir.path(), "config.yml", "engine:\n  max_parallel: 8\n");
898        assert!(!reload_file(&path));
899        assert_eq!(cached_value(&path, "engine.max_parallel"), "8");
900
901        let mut rx = subscribe();
902        std::fs::write(&path, "engine:\n  max_parallel: [unclosed\n").unwrap();
903        assert!(reload_file(&path));
904
905        assert_eq!(cached_value(&path, "engine.max_parallel"), "8");
906        assert!(rx.try_recv().is_err(), "a broken edit must not notify");
907
908        // A later good edit is picked up again
909        std::fs::write(&path, "engine:\n  max_parallel: 12\n").unwrap();
910        assert!(!reload_file(&path));
911        assert_eq!(cached_value(&path, "engine.max_parallel"), "12");
912    }
913
914    #[test]
915    fn adding_a_project_is_visible_after_reload() {
916        let dir = tempfile::tempdir().unwrap();
917        let path = write_yaml(dir.path(), "config.yml", "projects:\n  - /srv/one\n");
918        assert_eq!(cached_projects(&path), vec!["/srv/one"]);
919
920        std::fs::write(&path, "projects:\n  - /srv/one\n  - /srv/two\n").unwrap();
921        assert!(!reload_file(&path));
922
923        assert_eq!(cached_projects(&path), vec!["/srv/one", "/srv/two"]);
924    }
925
926    #[test]
927    fn repeated_watcher_events_for_one_breakage_log_once() {
928        let dir = tempfile::tempdir().unwrap();
929        let path = write_yaml(dir.path(), "config.yml", "engine:\n  max_parallel: 4\n");
930        reload_file(&path);
931
932        std::fs::write(&path, "engine: [broken\n").unwrap();
933        assert!(reload_file(&path), "first bad read logs");
934        assert!(!reload_file(&path), "same breakage does not log again");
935    }
936
937    #[test]
938    fn extract_skills_list_empty_for_missing() {
939        let yaml = serde_norway::from_str::<serde_norway::Value>("foo: bar").unwrap();
940        let skills = extract_skills_list(&yaml).unwrap();
941        assert!(skills.is_empty());
942    }
943}