import importlib.util
import json
import os
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch


ROOT = Path(__file__).resolve().parents[1]
MODULE_PATH = ROOT / "platform/llama-dashboard/app.py"


class _Response:
    status = 202

    def __enter__(self):
        return self

    def __exit__(self, *_args):
        return False

    def read(self):
        return b'{"status":"accepted"}'


class DashboardModeTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.tempdir = tempfile.TemporaryDirectory()
        with patch.dict(os.environ, {
            "DASHBOARD_HISTORY_DB": str(Path(cls.tempdir.name) / "history.sqlite3"),
            "DASHBOARD_BACKUP_DIR": str(Path(cls.tempdir.name) / "backups"),
            "ROUTER_URL": "http://router.test:8081",
            "ROUTER_API_KEY": "test-key",
        }):
            spec = importlib.util.spec_from_file_location("dashboard_app_test", MODULE_PATH)
            cls.dashboard = importlib.util.module_from_spec(spec)
            assert spec.loader is not None
            spec.loader.exec_module(cls.dashboard)
        cls.backup_dir = Path(cls.tempdir.name) / "backups"
        cls.backup_dir.mkdir()

    @classmethod
    def tearDownClass(cls):
        cls.tempdir.cleanup()

    def test_separation_mode_is_forwarded_to_router(self):
        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
            status, body = self.dashboard.change_mode("separation")

        self.assertEqual(status, 202)
        self.assertEqual(body, {"status": "accepted"})
        request = urlopen.call_args.args[0]
        self.assertEqual(json.loads(request.data), {"mode": "separation"})
        self.assertEqual(request.get_header("Authorization"), "Bearer test-key")

    def test_voice_mode_is_forwarded_to_router(self):
        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
            status, body = self.dashboard.change_mode("voice")

        self.assertEqual(status, 202)
        self.assertEqual(body, {"status": "accepted"})
        request = urlopen.call_args.args[0]
        self.assertEqual(json.loads(request.data), {"mode": "voice"})

    def test_voice_change_mode_is_forwarded_to_router(self):
        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
            status, body = self.dashboard.change_mode("voicechange")

        self.assertEqual(status, 202)
        self.assertEqual(body, {"status": "accepted"})
        request = urlopen.call_args.args[0]
        self.assertEqual(json.loads(request.data), {"mode": "voicechange"})

    def test_applio_mode_is_forwarded_to_router(self):
        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
            status, body = self.dashboard.change_mode("applio")
        self.assertEqual(status, 202)
        self.assertEqual(body, {"status": "accepted"})
        self.assertEqual(json.loads(urlopen.call_args.args[0].data), {"mode": "applio"})

    def test_yue2_mode_is_forwarded_to_router(self):
        with patch.object(self.dashboard.urllib.request, "urlopen", return_value=_Response()) as urlopen:
            status, body = self.dashboard.change_mode("yue2")
        self.assertEqual(status, 202)
        self.assertEqual(body, {"status": "accepted"})
        self.assertEqual(json.loads(urlopen.call_args.args[0].data), {"mode": "yue2"})

    def test_unknown_mode_is_rejected_without_router_request(self):
        with patch.object(self.dashboard.urllib.request, "urlopen") as urlopen:
            status, body = self.dashboard.change_mode("unknown")

        self.assertEqual(status, 400)
        self.assertEqual(body, {"error": "invalid mode"})
        urlopen.assert_not_called()

    def test_dashboard_uses_one_status_poll_for_all_mode_labels(self):
        html = self.dashboard.HTML

        self.assertEqual(html.count("fetch('/api/status'"), 1)
        self.assertNotIn("refreshVoiceChange", html)
        self.assertIn("voicechange:'X-VC Voice Changer'", html)
        self.assertIn("applio:'Applio / RVC'", html)

    def test_dashboard_offers_both_applio_frontends(self):
        html = self.dashboard.HTML

        self.assertIn("Original Applio UI", html)
        self.assertIn("Mikes Applio UI", html)
        self.assertIn("http://192.168.1.212:8011/", html)
        self.assertIn("http://192.168.1.212:8012/", html)

    def test_dashboard_keeps_ace_step_and_offers_yue2_separately(self):
        html = self.dashboard.HTML

        self.assertIn("ACE-Step Studio", html)
        self.assertIn("YuE2 Studio", html)
        self.assertIn("setMode('music')", html)
        self.assertIn("setMode('yue2')", html)
        self.assertIn("http://192.168.1.212:8014/", html)

    def test_dashboard_lists_only_portable_encrypted_backups(self):
        valid = self.backup_dir / "athena-portable-2026-09-10T10-00-00Z.tar.zst.age"
        valid.write_bytes(b"encrypted")
        valid.with_name(valid.name + ".sha256").write_text(
            "a" * 64 + "  " + valid.name + "\n", encoding="utf-8"
        )
        (self.backup_dir / "unrelated.txt").write_text("ignore", encoding="utf-8")

        backups = self.dashboard.backup_inventory()

        self.assertEqual([item["name"] for item in backups], [valid.name])
        self.assertEqual(backups[0]["sha256"], "a" * 64)
        self.assertTrue(backups[0]["download_url"].startswith("/api/backups/download/"))


if __name__ == "__main__":
    unittest.main()