import os
import tempfile
import types
import unittest

from mudfish_decky.system import api_base, ensure_daemon, find_mudfish


def make_install(root, version):
    os.makedirs(os.path.join(root, version, "bin"))
    os.makedirs(os.path.join(root, version, "var"))
    open(os.path.join(root, version, "bin", "mudrun-headless"), "w").close()


class SystemTest(unittest.TestCase):
    def test_find_mudfish_picks_highest_version(self):
        with tempfile.TemporaryDirectory() as root:
            for v in ["5.11.0", "6.6.0", "6.10.1"]:
                make_install(root, v)
            os.makedirs(os.path.join(root, "6.99.0"))  # no binary: ignored
            self.assertEqual(find_mudfish(root),
                             (os.path.join(root, "6.10.1", "bin", "mudrun-headless"), os.path.join(root, "6.10.1", "var")))

    def test_find_mudfish_none_when_missing(self):
        with tempfile.TemporaryDirectory() as root:
            self.assertIsNone(find_mudfish(os.path.join(root, "nope")))

    def test_api_base_reads_ipport(self):
        with tempfile.TemporaryDirectory() as var:
            with open(os.path.join(var, ".ipport"), "w") as f:
                f.write("127.0.0.1:9123")
            self.assertEqual(api_base(var), "http://127.0.0.1:9123")

    def test_api_base_default(self):
        with tempfile.TemporaryDirectory() as var:
            self.assertEqual(api_base(var), "http://127.0.0.1:8282")

    def test_ensure_daemon_noop_when_up(self):
        calls = []
        ensure_daemon("/bin/x", lambda: True, run=lambda cmd, **kw: calls.append(cmd))
        self.assertEqual(calls, [])

    def test_ensure_daemon_starts_transient_unit(self):
        calls, ups = [], iter([False, False, True])

        def run(cmd, **kw):
            calls.append(cmd)
            return types.SimpleNamespace(returncode=0)

        ensure_daemon("/opt/mudfish/6.6.0/bin/mudrun-headless", lambda: next(ups), run=run, sleep=lambda s: None)
        self.assertEqual(calls, [["systemd-run", "--unit=mudfish-decky", "--collect",
                                  "/opt/mudfish/6.6.0/bin/mudrun-headless", "-B"]])


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