"""Step definitions for CLI plan and context workflow tests."""

import os
import shutil
import subprocess
import sys
import tempfile
from pathlib import Path

from behave import given, then
from behave.runner import Context


def add_cleanup(context: Context, handler):
    """Add a cleanup handler to the context."""
    if not hasattr(context, "_cleanup_handlers"):
        context._cleanup_handlers = []
    context._cleanup_handlers.append(handler)


@given("I have a minimal test environment")
def step_minimal_test_environment(context: Context) -> None:
    """Ensure we have a minimal test environment."""
    context.test_dir = None
    context.original_dir = os.getcwd()
    # Set up mock AI provider for testing
    os.environ["CLEVERAGENTS_TESTING_USE_MOCK_AI"] = "true"


@given("I am in a temporary directory")
def step_in_temp_directory(context: Context) -> None:
    """Create and change to a temporary directory."""
    context.test_dir = tempfile.mkdtemp(prefix="cleveragents_test_")
    context.original_dir = os.getcwd()
    previous_search_root = os.environ.get("CLEVERAGENTS_PROJECT_SEARCH_ROOT")
    os.environ["CLEVERAGENTS_PROJECT_SEARCH_ROOT"] = context.test_dir
    os.chdir(context.test_dir)

    # Register cleanup
    def cleanup():
        os.chdir(context.original_dir)
        if previous_search_root is None:
            os.environ.pop("CLEVERAGENTS_PROJECT_SEARCH_ROOT", None)
        else:
            os.environ["CLEVERAGENTS_PROJECT_SEARCH_ROOT"] = previous_search_root
        if context.test_dir and os.path.exists(context.test_dir):
            shutil.rmtree(context.test_dir)

    add_cleanup(context, cleanup)


@given('I have initialized a project "{name}"')
def step_initialize_project_simple(context: Context, name: str) -> None:
    """Initialize a project in the current directory."""

    result = subprocess.run(
        [sys.executable, "-m", "cleveragents", "init", name],
        capture_output=True,
        text=True,
        timeout=60,
        cwd=context.test_dir if hasattr(context, "test_dir") else None,
    )
    assert result.returncode == 0, f"Failed to initialize project: {result.stderr}"


@given('I have created a file "{filename}" with content "{content}"')
def step_create_file_with_content(
    context: Context, filename: str, content: str
) -> None:
    """Create a file with specific content."""
    Path(filename).write_text(content)


@given('I have added path "{path}" to context')
def step_add_path_to_context(context: Context, path: str) -> None:
    """Add a file or directory path to context."""
    # First ensure we have a plan
    if not hasattr(context, "has_plan"):
        # Create a default plan first
        result = subprocess.run(
            [sys.executable, "-m", "cleveragents", "tell", "Default test plan"],
            capture_output=True,
            text=True,
            timeout=60,
            cwd=context.test_dir if hasattr(context, "test_dir") else None,
        )
        assert result.returncode == 0, f"Failed to create plan: {result.stderr}"
        context.has_plan = True

    result = subprocess.run(
        [sys.executable, "-m", "cleveragents", "context", "add", path],
        capture_output=True,
        text=True,
        timeout=30,
        cwd=context.test_dir if hasattr(context, "test_dir") else None,
    )
    assert result.returncode == 0, f"Failed to add to context: {result.stderr}"


@then('the directory "{dirname}" should exist')
def step_directory_should_exist(context: Context, dirname: str) -> None:
    """Check if a directory exists."""
    assert Path(dirname).is_dir(), f"Directory {dirname} does not exist"


@then('the test file "{filename}" should exist')
def step_test_file_should_exist(context: Context, filename: str) -> None:
    """Check if a test file exists."""
    assert Path(filename).exists(), f"File {filename} does not exist"


@then('the file "{filename}" should contain "{content}"')
def step_file_should_contain(context: Context, filename: str, content: str) -> None:
    """Check if a file contains specific content."""
    file_content = Path(filename).read_text()
    assert content in file_content, f"File {filename} does not contain '{content}'"


def after_scenario(context: Context, scenario) -> None:
    """Clean up after each scenario."""
    # Clean up environment variables
    if "CLEVERAGENTS_TESTING_USE_MOCK_AI" in os.environ:
        del os.environ["CLEVERAGENTS_TESTING_USE_MOCK_AI"]

    # Execute any cleanup handlers
    if hasattr(context, "_cleanup_handlers"):
        for handler in context._cleanup_handlers:
            handler()