from __future__ import annotations

import importlib
import os
import sys
import tempfile
import unittest
from pathlib import Path

from fastapi.testclient import TestClient


BACKEND_DIR = Path(__file__).resolve().parents[1]
if str(BACKEND_DIR) not in sys.path:
    sys.path.insert(0, str(BACKEND_DIR))


class CapabilitiesApiTests(unittest.TestCase):
    def _load_client(
        self,
        runtime_mode: str,
        enabled_features: str | None = None,
    ) -> TestClient:
        self.temp_dir = tempfile.TemporaryDirectory()
        os.environ["OMNIVOICE_RUNTIME_MODE"] = runtime_mode
        os.environ["OMNIVOICE_DATA_ROOT"] = self.temp_dir.name
        if enabled_features is None:
            os.environ.pop("OMNIVOICE_ENABLED_FEATURES", None)
        else:
            os.environ["OMNIVOICE_ENABLED_FEATURES"] = enabled_features

        sys.modules.pop("main", None)
        main = importlib.import_module("main")
        main = importlib.reload(main)
        self.addCleanup(self.temp_dir.cleanup)
        return TestClient(main.app)

    def tearDown(self) -> None:
        os.environ.pop("OMNIVOICE_RUNTIME_MODE", None)
        os.environ.pop("OMNIVOICE_DATA_ROOT", None)
        os.environ.pop("OMNIVOICE_ENABLED_FEATURES", None)

    def test_self_hosted_mode_enables_all_local_features(self) -> None:
        client = self._load_client(runtime_mode="self_hosted")

        response = client.get("/api/capabilities")

        self.assertEqual(response.status_code, 200)
        payload = response.json()
        self.assertEqual(payload["runtime_mode"], "self_hosted")
        self.assertEqual(payload["license"]["state"], "dev_override")
        self.assertFalse(payload["ui"]["show_runtime_help"])
        self.assertTrue(payload["features"]["playground"])
        self.assertTrue(payload["features"]["audio_studio"])
        self.assertTrue(payload["features"]["video_studio"])
        self.assertTrue(payload["features"]["post_production"])
        self.assertTrue(payload["features"]["audio_export"])
        self.assertTrue(payload["features"]["video_export"])

    def test_production_mode_defaults_to_playground_only(self) -> None:
        client = self._load_client(runtime_mode="production")

        response = client.get("/api/capabilities")

        self.assertEqual(response.status_code, 200)
        payload = response.json()
        self.assertEqual(payload["runtime_mode"], "production")
        self.assertTrue(payload["features"]["playground"])
        self.assertFalse(payload["features"]["audio_studio"])
        self.assertFalse(payload["features"]["video_studio"])
        self.assertFalse(payload["features"]["post_production"])
        self.assertTrue(payload["ui"]["show_runtime_help"])