#!/usr/bin/env bats

setup() {
    REPO_ROOT="$(cd "$BATS_TEST_DIRNAME/../../../../../" && pwd)"
    load "$REPO_ROOT/tests/test_helper/bats-support/load"
    load "$REPO_ROOT/tests/test_helper/bats-assert/load"

    SCRIPT="$(cd "$BATS_TEST_DIRNAME/../scripts" && pwd)/new-agent.sh"
    ROOT="$(mktemp -d)"
}

teardown() {
    rm -rf "$ROOT"
}

# ---------------------------------------------------------------------------
# Help
# ---------------------------------------------------------------------------

@test "--help exits 0" {
    run bash "$SCRIPT" --help
    assert_success
    assert_output --partial "Usage:"
}

# ---------------------------------------------------------------------------
# Plugin scope (plugin.json present at root)
# ---------------------------------------------------------------------------

@test "plugin scope: creates both agent files in agents/" {
    touch "$ROOT/plugin.json"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -f "$ROOT/agents/my-agent.md" ]
    assert [ -f "$ROOT/agents/my-agent.agent.md" ]
}

@test "plugin scope: creates agents/ directory if missing" {
    touch "$ROOT/plugin.json"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -d "$ROOT/agents" ]
}

@test "plugin scope: creates sources.md in agents/" {
    touch "$ROOT/plugin.json"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -f "$ROOT/agents/sources.md" ]
}

@test "plugin scope: no-op if claude code file already exists" {
    touch "$ROOT/plugin.json"
    mkdir -p "$ROOT/agents"
    echo "existing" > "$ROOT/agents/my-agent.md"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    run grep "existing" "$ROOT/agents/my-agent.md"
    assert_success
}

@test "plugin scope: no-op if copilot file already exists" {
    touch "$ROOT/plugin.json"
    mkdir -p "$ROOT/agents"
    echo "existing" > "$ROOT/agents/my-agent.agent.md"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    run grep "existing" "$ROOT/agents/my-agent.agent.md"
    assert_success
}

@test "plugin scope: still creates missing file when one already exists" {
    touch "$ROOT/plugin.json"
    mkdir -p "$ROOT/agents"
    echo "existing" > "$ROOT/agents/my-agent.md"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -f "$ROOT/agents/my-agent.agent.md" ]
}

# ---------------------------------------------------------------------------
# Non-plugin scope (no plugin.json)
# ---------------------------------------------------------------------------

@test "non-plugin scope: creates claude code file in .claude/agents/" {
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -f "$ROOT/.claude/agents/my-agent.md" ]
}

@test "non-plugin scope: creates copilot file in .github/agents/" {
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -f "$ROOT/.github/agents/my-agent.agent.md" ]
}

@test "non-plugin scope: creates .claude/agents/ directory if missing" {
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -d "$ROOT/.claude/agents" ]
}

@test "non-plugin scope: creates .github/agents/ directory if missing" {
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ -d "$ROOT/.github/agents" ]
}

@test "non-plugin scope: no sources.md created" {
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    assert [ ! -f "$ROOT/.claude/agents/sources.md" ]
    assert [ ! -f "$ROOT/.github/agents/sources.md" ]
}

@test "non-plugin scope: no-op if claude code file already exists" {
    mkdir -p "$ROOT/.claude/agents"
    echo "existing" > "$ROOT/.claude/agents/my-agent.md"
    run bash "$SCRIPT" my-agent "$ROOT"
    assert_success
    run grep "existing" "$ROOT/.claude/agents/my-agent.md"
    assert_success
}

# ---------------------------------------------------------------------------
# Name validation
# ---------------------------------------------------------------------------

@test "fails when no arguments given" {
    run bash "$SCRIPT"
    assert_failure
}

@test "fails when agent name contains uppercase" {
    run bash "$SCRIPT" MyAgent "$ROOT"
    assert_failure
}

@test "fails when agent name has consecutive hyphens" {
    run bash "$SCRIPT" my--agent "$ROOT"
    assert_failure
}

@test "fails when agent name has a leading hyphen" {
    run bash "$SCRIPT" -my-agent "$ROOT"
    assert_failure
}

@test "fails when agent name has a trailing hyphen" {
    run bash "$SCRIPT" my-agent- "$ROOT"
    assert_failure
}

@test "agent name with numbers is valid" {
    run bash "$SCRIPT" agent-v2 "$ROOT"
    assert_success
}

# ---------------------------------------------------------------------------
# Root validation
# ---------------------------------------------------------------------------

@test "fails when root directory does not exist" {
    run bash "$SCRIPT" my-agent "/nonexistent/path"
    assert_failure
}

# ---------------------------------------------------------------------------
# Template content
# ---------------------------------------------------------------------------

@test "plugin scope: claude code template contains AGENT_NAME substituted" {
    touch "$ROOT/plugin.json"
    bash "$SCRIPT" my-agent "$ROOT"
    run grep "my-agent" "$ROOT/agents/my-agent.md"
    assert_success
}

@test "plugin scope: copilot template contains AGENT_NAME substituted" {
    touch "$ROOT/plugin.json"
    bash "$SCRIPT" my-agent "$ROOT"
    run grep "my-agent" "$ROOT/agents/my-agent.agent.md"
    assert_success
}