"""Local trusted-zone URL tests. No database."""
from __future__ import annotations

import os
import sys
import tempfile
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))
os.environ.setdefault("KANSHO_PROVIDER_KEY", "")
os.environ.setdefault("KANSHO_DETECT_PROVIDER_KEY", "")
Path(tempfile.gettempdir()).mkdir(parents=True, exist_ok=True)

from providers import is_local_url


def expect(ok: bool, message: str) -> None:
    if not ok:
        raise SystemExit(f"FAIL: {message}")
    print(f"OK  {message}")


def main() -> None:
    expect(is_local_url("http://127.0.0.1:11434/v1/chat/completions"), "loopback is local")
    expect(is_local_url("http://localhost:11434/v1/chat/completions"), "localhost is local")
    expect(is_local_url("http://192.168.2.144:11434/v1/chat/completions"), "RFC1918 Ollama host is local")
    expect(is_local_url("http://10.0.0.8:11434/v1/chat/completions"), "10/8 is local")
    expect(is_local_url("http://ollama:11434/v1/chat/completions"), "compose hostname ollama is local")
    expect(not is_local_url("https://openrouter.ai/api/v1/chat/completions"), "OpenRouter is not local")
    expect(not is_local_url("https://8.8.8.8/v1/chat/completions"), "public IP is not local")
    print("local url tests passed.")


if __name__ == "__main__":
    main()