import json
import os
from typing import Any

SNAPSHOTS_DIR = "snapshots"


def get_snapshot_path(name: str) -> str:
    """Get the file path for a snapshot."""
    os.makedirs(SNAPSHOTS_DIR, exist_ok=True)
    return os.path.join(SNAPSHOTS_DIR, f"{name}.json")


def list_snapshots() -> list[dict[str, Any]]:
    """List all available snapshots."""
    os.makedirs(SNAPSHOTS_DIR, exist_ok=True)
    snapshots = []

    for filename in os.listdir(SNAPSHOTS_DIR):
        if filename.endswith(".json"):
            path = os.path.join(SNAPSHOTS_DIR, filename)
            with open(path, "r") as f:
                data = json.load(f)
                snapshots.append(
                    {
                        "name": filename[:-5],
                        "metadata": data.get("metadata", {}),
                        "requests_count": len(data.get("requests", [])),
                    }
                )

    return snapshots


def load_snapshot(name: str) -> dict[str, Any]:
    """Load a snapshot from file."""
    path = get_snapshot_path(name)

    if not os.path.exists(path):
        raise FileNotFoundError(f"Snapshot '{name}' not found")

    with open(path, "r") as f:
        return json.load(f)


def save_snapshot(
    name: str,
    session_data: dict[str, Any],
    description: str | None = None,
    tags: list[str] | None = None,
    source_url: str | None = None,
) -> str:
    """Save a snapshot to file."""
    path = get_snapshot_path(name)

    import time

    metadata = {
        "version": "1.0",
        "timestamp": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
        "description": description or "",
        "source_url": source_url or "",
        "tags": tags or [],
        "latency_mode": "original",
        "custom_latency_ms": None,
    }

    snapshot = {
        "metadata": metadata,
        "requests": [session_data],
    }

    with open(path, "w") as f:
        json.dump(snapshot, f, indent=2)

    return path


def delete_snapshot(name: str) -> bool:
    """Delete a snapshot."""
    path = get_snapshot_path(name)

    if not os.path.exists(path):
        return False

    os.remove(path)
    return True