import os
import tempfile
import unittest
from unittest import mock

import dxvk_fps
import setup_tool_dynamic
import setup_tool_responsive


class FakeVar:
    def __init__(self, value):
        self.value = value

    def get(self):
        return self.value

    def set(self, value):
        self.value = value


class DxvkFpsLimitTests(unittest.TestCase):
    def write_config(self, root, text):
        path = os.path.join(root, "dxvk.conf")
        with open(path, "w", encoding="utf-8", newline="\n") as handle:
            handle.write(text)
        return path

    def read_config(self, path):
        with open(path, "r", encoding="utf-8") as handle:
            return handle.read()

    def test_enable_replaces_bundled_commented_value_and_preserves_other_options(self):
        with tempfile.TemporaryDirectory() as root:
            path = self.write_config(
                root,
                "# DXVK configuration\n"
                "# d3d9.maxFrameRate = 1000\n"
                "dxvk.allowFse = False\n",
            )

            dxvk_fps.apply_dxvk_fps_limit(path, True, 165)
            updated = self.read_config(path)

            self.assertIn("d3d9.maxFrameRate = 165\n", updated)
            self.assertNotIn("# d3d9.maxFrameRate = 1000", updated)
            self.assertIn("dxvk.allowFse = False\n", updated)
            self.assertEqual(updated.count("d3d9.maxFrameRate"), 1)
            self.assertEqual(dxvk_fps.read_dxvk_fps_limit(path), (True, 165))

    def test_disable_comments_the_setting(self):
        with tempfile.TemporaryDirectory() as root:
            path = self.write_config(
                root,
                "d3d9.maxFrameRate = 144\n"
                "dxvk.numCompilerThreads = 4\n",
            )

            dxvk_fps.apply_dxvk_fps_limit(path, False, 144)
            updated = self.read_config(path)

            self.assertIn("# d3d9.maxFrameRate = 144\n", updated)
            self.assertNotIn("\nd3d9.maxFrameRate = 144\n", "\n" + updated)
            self.assertEqual(dxvk_fps.read_dxvk_fps_limit(path), (False, 144))

    def test_duplicate_limit_lines_are_collapsed_to_one(self):
        with tempfile.TemporaryDirectory() as root:
            path = self.write_config(
                root,
                "d3d9.maxFrameRate = 60\n"
                "# d3d9.maxFrameRate = 120\n"
                "dxvk.allowFse = False\n",
            )

            dxvk_fps.apply_dxvk_fps_limit(path, True, 240)
            updated = self.read_config(path)

            self.assertEqual(updated.count("d3d9.maxFrameRate"), 1)
            self.assertIn("d3d9.maxFrameRate = 240\n", updated)
            self.assertIn("dxvk.allowFse = False\n", updated)

    def test_missing_limit_line_is_appended(self):
        with tempfile.TemporaryDirectory() as root:
            path = self.write_config(root, "dxvk.allowFse = False\n")

            dxvk_fps.apply_dxvk_fps_limit(path, True, 75)
            updated = self.read_config(path)

            self.assertIn("dxvk.allowFse = False\n", updated)
            self.assertTrue(updated.endswith("d3d9.maxFrameRate = 75\n"))

    def test_invalid_values_are_rejected_without_modifying_file(self):
        with tempfile.TemporaryDirectory() as root:
            original = "# d3d9.maxFrameRate = 1000\n"
            path = self.write_config(root, original)

            with self.assertRaises(ValueError):
                dxvk_fps.apply_dxvk_fps_limit(path, True, 0)

            self.assertEqual(self.read_config(path), original)

    def test_refresh_detection_always_returns_a_positive_integer(self):
        detected = dxvk_fps.detect_max_refresh_rate(default=77)
        self.assertIsInstance(detected, int)
        self.assertGreater(detected, 0)

    @mock.patch.object(setup_tool_dynamic.ModernWowSetupTool, "configure_dxvk")
    @mock.patch("setup_tool_responsive.dxvk_fps.apply_dxvk_fps_limit")
    def test_directx9_never_applies_fps_config(self, apply_limit, parent_configure):
        tool = setup_tool_responsive.ResponsiveModernWowSetupTool.__new__(
            setup_tool_responsive.ResponsiveModernWowSetupTool
        )
        tool.rendering_mode = FakeVar("directx9")
        tool.limit_dxvk_fps = FakeVar(True)
        tool.dxvk_fps_limit = FakeVar(165)

        tool.configure_dxvk("C:/WoW")

        parent_configure.assert_called_once_with("C:/WoW")
        apply_limit.assert_not_called()

    @mock.patch.object(setup_tool_dynamic.ModernWowSetupTool, "configure_dxvk")
    @mock.patch("setup_tool_responsive.dxvk_fps.apply_dxvk_fps_limit")
    def test_dxvk_applies_selected_fps_value(self, apply_limit, parent_configure):
        tool = setup_tool_responsive.ResponsiveModernWowSetupTool.__new__(
            setup_tool_responsive.ResponsiveModernWowSetupTool
        )
        tool.rendering_mode = FakeVar("dxvk")
        tool.limit_dxvk_fps = FakeVar(True)
        tool.dxvk_fps_limit = FakeVar(144)

        tool.configure_dxvk("C:/WoW")

        parent_configure.assert_called_once_with("C:/WoW")
        apply_limit.assert_called_once_with(
            os.path.join("C:/WoW", "dxvk.conf"),
            True,
            144,
        )


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