defect-config 0.1.0-alpha.2

use super::*;

use std::fs;

use tempfile::TempDir;

/// Write a profile at `<root>/.config/defect/agents/<name>/` (user) or
/// `<root>/proj/.defect/agents/<name>/` (project).
fn write_profile(agents_dir: &Path, name: &str, config_toml: &str, system_md: Option<&str>) {
    let dir = agents_dir.join(name);
    fs::create_dir_all(&dir).expect("mkdir profile");
    fs::write(dir.join("config.toml"), config_toml).expect("write config.toml");
    if let Some(md) = system_md {
        fs::write(dir.join("system.md"), md).expect("write system.md");
    }
}

/// Write a single-file profile to `<agents_dir>/<name>.md`.
fn write_single_file(agents_dir: &Path, name: &str, contents: &str) {
    fs::create_dir_all(agents_dir).expect("mkdir agents");
    fs::write(agents_dir.join(format!("{name}.md")), contents).expect("write .md");
}

/// Create a repo root containing `.git` (so that `find_repo_root` hits it), returning
/// `(tmp, repo_root)`.
fn repo(tmp: &TempDir) -> PathBuf {
    let root = tmp.path().join("proj");
    fs::create_dir_all(root.join(".git")).expect("mkdir .git");
    root
}

fn opts_with(tmp: &TempDir, repo_root: &Path) -> LoadConfigOptions {
    LoadConfigOptions {
        cwd: repo_root.to_path_buf(),
        xdg_config_home: Some(tmp.path().join("xdg")),
        ..LoadConfigOptions::default()
    }
}

#[test]
fn discovers_project_and_user_profiles() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);

    let user_agents = tmp.path().join("xdg/defect/agents");
    write_profile(
        &user_agents,
        "userbot",
        "description = \"a user-layer agent\"\n",
        Some("you are userbot"),
    );

    let project_agents = repo_root.join(".defect/agents");
    write_profile(
        &project_agents,
        "reviewer",
        "description = \"review diffs\"\n[tools]\nallow = [\"read_file\"]\n",
        Some("you are reviewer"),
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles.len(), 2);
    assert_eq!(profiles["userbot"].description, "a user-layer agent");
    assert_eq!(profiles["reviewer"].tool_allow, vec!["read_file"]);
    assert_eq!(profiles["reviewer"].system_prompt_text, "you are reviewer");
}

#[test]
fn project_overrides_user_on_name_collision() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);

    write_profile(
        &tmp.path().join("xdg/defect/agents"),
        "bot",
        "description = \"user version\"\n",
        Some("user prompt"),
    );
    write_profile(
        &repo_root.join(".defect/agents"),
        "bot",
        "description = \"project version\"\n",
        Some("project prompt"),
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles.len(), 1);
    assert_eq!(profiles["bot"].description, "project version");
    assert_eq!(profiles["bot"].system_prompt_text, "project prompt");
}

#[test]
fn missing_description_is_hard_error() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "bad",
        "model = \"x\"\n",
        Some("prompt"),
    );

    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn allow_omitted_defaults_to_read_only_set() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "reader",
        "description = \"reads\"\n",
        Some("prompt"),
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles["reader"].tool_allow, vec!["read_file", "search"]);
}

#[test]
fn hooks_omitted_yields_empty() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "plain",
        "description = \"no hooks\"\n",
        Some("prompt"),
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert!(profiles["plain"].hooks.is_empty());
}

#[test]
fn profile_hooks_parsed_with_name_and_source() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "guard",
        "description = \"hooked\"\n\
         [[hooks.before_tool_apply]]\n\
         name = \"redact\"\n\
         match = { tool = \"bash\" }\n\
         handler = { type = \"builtin\", name = \"redact-secrets\" }\n",
        Some("prompt"),
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    let entries = profiles["guard"].hooks.get("before_tool_apply");
    assert_eq!(entries.len(), 1);
    assert_eq!(entries[0].name.as_deref(), Some("redact"));
    assert_eq!(entries[0].matcher.tool.as_deref(), Some("bash"));
    // Project-level discovery ⇒ source = Project.
    assert_eq!(entries[0].source, crate::types::ConfigSource::Project);
}

#[test]
fn profile_hooks_unknown_event_is_hard_error() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "typo",
        "description = \"bad event\"\n\
         [[hooks.before_tool_aply]]\n\
         handler = { type = \"builtin\", name = \"redact-secrets\" }\n",
        Some("prompt"),
    );

    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn single_file_profile_hooks_parsed() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "inline",
        "+++\n\
         description = \"inline hooked\"\n\
         [[hooks.after_session_enter]]\n\
         handler = { type = \"builtin\", name = \"skill-manifest\" }\n\
         +++\nyou are inline\n",
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    let entries = profiles["inline"].hooks.get("after_session_enter");
    assert_eq!(entries.len(), 1);
    // Single-file variant without name ⇒ None (assembly-time fallback to anonymous).
    assert_eq!(entries[0].name, None);
}

#[test]
fn unknown_key_is_hard_error() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_profile(
        &repo_root.join(".defect/agents"),
        "typo",
        "description = \"x\"\nmdoel = \"oops\"\n",
        Some("prompt"),
    );

    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn prompt_file_escaping_profile_dir_is_rejected() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    // Place a secret in the repo root that the profile attempts to read via
    // `../../secret.md`.
    fs::write(repo_root.join("secret.md"), "TOPSECRET").expect("write secret");
    write_profile(
        &repo_root.join(".defect/agents"),
        "escaper",
        "description = \"x\"\n[prompt]\nfile = \"../../secret.md\"\n",
        Some("decoy"),
    );

    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must reject escape");
    match err {
        ConfigError::Invalid { message, .. } => {
            assert!(message.contains("prompt.file"), "got: {message}");
        }
        other => panic!("expected Invalid, got {other:?}"),
    }
}

#[test]
fn empty_when_no_agents_dirs() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert!(profiles.is_empty());
}

#[test]
fn subdir_without_config_toml_is_skipped() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    let agents = repo_root.join(".defect/agents");
    fs::create_dir_all(agents.join("not-a-profile")).expect("mkdir");
    write_profile(&agents, "real", "description = \"r\"\n", Some("p"));

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles.len(), 1);
    assert!(profiles.contains_key("real"));
}

// --- single-file variant (+++ TOML frontmatter) ---

#[test]
fn discovers_single_file_profile() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "reviewer",
        "+++\ndescription = \"review diffs\"\nmodel = \"opus\"\n[tools]\nallow = [\"read_file\"]\n+++\nYou are a reviewer.\n",
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles.len(), 1);
    let p = &profiles["reviewer"];
    assert_eq!(p.description, "review diffs");
    assert_eq!(p.model.as_deref(), Some("opus"));
    assert_eq!(p.tool_allow, vec!["read_file"]);
    assert_eq!(p.system_prompt_text, "You are a reviewer.");
}

#[test]
fn single_file_allow_omitted_defaults_read_only() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "reader",
        "+++\ndescription = \"reads\"\n+++\nbody\n",
    );
    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles["reader"].tool_allow, vec!["read_file", "search"]);
}

#[test]
fn single_file_missing_frontmatter_errors() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "bad",
        "no frontmatter here\njust text\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn single_file_missing_description_errors() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "bad",
        "+++\nmodel = \"x\"\n+++\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn single_file_prompt_table_is_rejected() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "bad",
        "+++\ndescription = \"d\"\n[prompt]\nfile = \"x.md\"\n+++\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must reject [prompt]");
    match err {
        ConfigError::Invalid { message, .. } => {
            assert!(message.contains("[prompt]"), "got: {message}");
        }
        other => panic!("expected Invalid, got {other:?}"),
    }
}

#[test]
fn single_file_unknown_key_errors() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "typo",
        "+++\ndescription = \"d\"\nmdoel = \"oops\"\n+++\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[test]
fn folder_and_single_file_same_name_same_layer_conflicts() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    let agents = repo_root.join(".defect/agents");
    write_profile(
        &agents,
        "dup",
        "description = \"folder\"\n",
        Some("folder prompt"),
    );
    write_single_file(
        &agents,
        "dup",
        "+++\ndescription = \"file\"\n+++\nfile prompt\n",
    );

    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must conflict");
    match err {
        ConfigError::Invalid { message, .. } => {
            assert!(message.contains("duplicate"), "got: {message}");
        }
        other => panic!("expected Invalid, got {other:?}"),
    }
}

#[test]
fn single_file_project_overrides_user() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &tmp.path().join("xdg/defect/agents"),
        "bot",
        "+++\ndescription = \"user\"\n+++\nuser prompt\n",
    );
    write_single_file(
        &repo_root.join(".defect/agents"),
        "bot",
        "+++\ndescription = \"project\"\n+++\nproject prompt\n",
    );
    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    assert_eq!(profiles.len(), 1);
    assert_eq!(profiles["bot"].description, "project");
    assert_eq!(profiles["bot"].system_prompt_text, "project prompt");
}

// Single-file YAML frontmatter (delimited by `---`, requires the `yaml` feature)

#[cfg(feature = "yaml")]
#[test]
fn discovers_yaml_frontmatter_profile() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "reviewer",
        "---\ndescription: review diffs\nmodel: opus\ntools:\n  allow: [read_file, search]\n---\nYou are a reviewer.\n",
    );

    let profiles = discover_profiles(&opts_with(&tmp, &repo_root)).expect("discover");
    let p = &profiles["reviewer"];
    assert_eq!(p.description, "review diffs");
    assert_eq!(p.model.as_deref(), Some("opus"));
    assert_eq!(p.tool_allow, vec!["read_file", "search"]);
    assert_eq!(p.system_prompt_text, "You are a reviewer.");
}

#[cfg(feature = "yaml")]
#[test]
fn yaml_unknown_key_errors() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "typo",
        "---\ndescription: d\nmdoel: oops\n---\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail");
    assert!(matches!(err, ConfigError::Invalid { .. }));
}

#[cfg(feature = "yaml")]
#[test]
fn yaml_prompt_table_is_rejected() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "bad",
        "---\ndescription: d\nprompt:\n  file: x.md\n---\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must reject prompt");
    match err {
        ConfigError::Invalid { message, .. } => {
            assert!(message.contains("[prompt]"), "got: {message}");
        }
        other => panic!("expected Invalid, got {other:?}"),
    }
}

/// When the `yaml` feature is disabled, `---` frontmatter must hard-fail with an
/// actionable error (no silent degradation).
#[cfg(not(feature = "yaml"))]
#[test]
fn yaml_frontmatter_without_feature_errors() {
    let tmp = TempDir::new().expect("tmp");
    let repo_root = repo(&tmp);
    write_single_file(
        &repo_root.join(".defect/agents"),
        "y",
        "---\ndescription: d\n---\nbody\n",
    );
    let err = discover_profiles(&opts_with(&tmp, &repo_root)).expect_err("must fail without yaml");
    match err {
        ConfigError::Invalid { message, .. } => {
            assert!(message.contains("yaml"), "got: {message}");
        }
        other => panic!("expected Invalid, got {other:?}"),
    }
}