import asyncio
import sys
import types
import unittest
from unittest.mock import Mock


class PluginMigrationTests(unittest.TestCase):
    def _load_plugin(self):
        decky = types.SimpleNamespace(
            DECKY_HOME="/decky",
            DECKY_USER_HOME="/home/deck",
            migrate_logs=Mock(),
            migrate_settings=Mock(),
            migrate_runtime=Mock(),
            logger=Mock(),
        )
        previous_decky = sys.modules.get("decky")
        previous_tomllib = sys.modules.get("tomllib")
        sys.modules["decky"] = decky
        sys.modules["tomllib"] = types.SimpleNamespace(loads=Mock())
        sys.path.insert(0, "py_modules")
        from lsfg_vk.plugin import Plugin
        return Plugin, decky, previous_decky, previous_tomllib

    def _restore(self, previous_decky, previous_tomllib):
        sys.path.remove("py_modules")
        if previous_decky is None:
            sys.modules.pop("decky", None)
        else:
            sys.modules["decky"] = previous_decky
        if previous_tomllib is None:
            sys.modules.pop("tomllib", None)
        else:
            sys.modules["tomllib"] = previous_tomllib

    def test_migration_only_runs_decky_path_migrations(self):
        Plugin, decky, previous_decky, previous_tomllib = self._load_plugin()
        try:
            plugin = Plugin.__new__(Plugin)
            plugin.installation_service = Mock()
            plugin.flatpak_service = Mock()

            asyncio.run(plugin._migration())

            decky.migrate_logs.assert_called_once()
            decky.migrate_settings.assert_called_once()
            decky.migrate_runtime.assert_called_once()
            plugin.installation_service.install.assert_not_called()
            plugin.flatpak_service.prepare_app.assert_not_called()
        finally:
            self._restore(previous_decky, previous_tomllib)

    def test_uninstall_cleans_owned_flatpak_state_and_profiles(self):
        Plugin, _decky, previous_decky, previous_tomllib = self._load_plugin()
        try:
            plugin = Plugin.__new__(Plugin)
            plugin.installation_service = Mock()
            plugin.flatpak_service = Mock()
            plugin.configuration_service = Mock()
            plugin.flatpak_service.remove_plugin_owned_environment.return_value = {"success": True}

            asyncio.run(plugin._uninstall())

            plugin.flatpak_service.remove_plugin_owned_environment.assert_called_once_with()
            plugin.configuration_service.reset_all_flatpak_configs.assert_called_once_with()
            plugin.installation_service.cleanup_on_uninstall.assert_called_once_with()
        finally:
            self._restore(previous_decky, previous_tomllib)

    def test_uninstall_preserves_profiles_when_flatpak_cleanup_fails(self):
        Plugin, _decky, previous_decky, previous_tomllib = self._load_plugin()
        try:
            plugin = Plugin.__new__(Plugin)
            plugin.installation_service = Mock()
            plugin.flatpak_service = Mock()
            plugin.configuration_service = Mock()
            plugin.flatpak_service.remove_plugin_owned_environment.return_value = {"success": False, "error": "changed"}

            asyncio.run(plugin._uninstall())

            plugin.configuration_service.reset_all_flatpak_configs.assert_not_called()
            plugin.installation_service.cleanup_on_uninstall.assert_called_once_with()
        finally:
            self._restore(previous_decky, previous_tomllib)


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