"""AppRun puts a `solidcraft-cli` wrapper on PATH (~/.local/bin) and the wrapper runs the bundled CLI."""
import os
import shutil
import subprocess
import tempfile
import unittest
from pathlib import Path

ROOT = Path(__file__).resolve().parents[2]


class AppRunCli(unittest.TestCase):
    def setUp(self):
        self.tmp = Path(tempfile.mkdtemp())
        self.addCleanup(shutil.rmtree, self.tmp, True)
        self.appdir = self.tmp / "app" / "AppDir"
        (self.appdir / "usr/bin").mkdir(parents=True)
        (self.appdir / "usr/share/applications").mkdir(parents=True)
        shutil.copy(ROOT / "packaging/linux/AppRun", self.appdir / "AppRun")
        (self.appdir / "usr/share/applications/io.tangencylabs.solidcraft.desktop").write_text("[Desktop Entry]\nExec=solidcraft\n")
        for name in ("solidcraft", "solidcraft-cli"):
            p = self.appdir / "usr/bin" / name
            p.write_text(f"#!/bin/sh\necho {name} \"$@\"\n")
            p.chmod(0o755)
        # The AppImage file: running it runs AppRun, as the runtime does.
        self.image = self.tmp / "My App" / "SolidCraft.AppImage"
        self.image.parent.mkdir()
        self.image.write_text(f'#!/bin/sh\nexec "{self.appdir}/AppRun" "$@"\n')
        self.image.chmod(0o755)
        self.home = self.tmp / "home"
        self.home.mkdir()
        self.env = {"PATH": os.environ["PATH"], "HOME": str(self.home), "APPIMAGE": str(self.image)}
        self.wrapper = self.home / ".local/bin/solidcraft-cli"

    def run_apprun(self, *args, **extra):
        env = {**self.env, **extra}
        return subprocess.run([str(self.appdir / "AppRun"), *args], env=env, capture_output=True, text=True, check=True).stdout.strip()

    def test_wrapper_runs_the_bundled_cli(self):
        self.assertEqual(self.run_apprun("a"), "solidcraft a")
        self.assertTrue(self.wrapper.is_file())
        out = subprocess.run([str(self.wrapper), "mcp", "setup"], env=self.env, capture_output=True, text=True, check=True).stdout.strip()
        self.assertEqual(out, "solidcraft-cli mcp setup")

    def test_wrapper_follows_a_moved_appimage(self):
        self.run_apprun()
        moved = self.tmp / "moved.AppImage"
        shutil.move(self.image, moved)
        self.env["APPIMAGE"] = str(moved)
        self.run_apprun()
        self.assertIn(str(moved), self.wrapper.read_text())

    def test_a_file_that_is_not_ours_is_left_alone(self):
        self.wrapper.parent.mkdir(parents=True)
        self.wrapper.write_text("#!/bin/sh\necho mine\n")
        self.run_apprun()
        self.assertEqual(self.wrapper.read_text(), "#!/bin/sh\necho mine\n")
        other = self.home / ".local/bin/solidcraft-cli"
        other.unlink()
        other.symlink_to("/bin/true")
        self.run_apprun()
        self.assertTrue(other.is_symlink())

    def test_entries_under_earlier_app_ids_are_replaced(self):
        apps = self.home / ".local/share/applications"
        apps.mkdir(parents=True)
        icons = self.home / ".local/share/icons/hicolor/256x256/apps"
        icons.mkdir(parents=True)
        for old in ("ai.storyteller.solidcraft", "ai.storyteller.solvecraft"):
            (apps / f"{old}.desktop").write_text(f'[Desktop Entry]\nExec="/old/SolidCraft.AppImage" %F\nIcon={old}\n')
            (icons / f"{old}.png").write_text("png")
        # An entry this script did not write (a package's, Exec on PATH) is left alone.
        (apps / "ai.storyteller.other.desktop").write_text("[Desktop Entry]\nExec=other\nIcon=ai.storyteller.other\n")
        self.run_apprun()
        self.assertTrue((apps / "io.tangencylabs.solidcraft.desktop").is_file())
        for old in ("ai.storyteller.solidcraft", "ai.storyteller.solvecraft"):
            self.assertFalse((apps / f"{old}.desktop").exists())
            self.assertFalse((icons / f"{old}.png").exists())
        self.assertTrue((apps / "ai.storyteller.other.desktop").is_file())

    def test_opt_out_installs_nothing(self):
        self.run_apprun(SOLIDCRAFT_NO_DESKTOP_INTEGRATION="1")
        self.assertFalse(self.wrapper.exists())

    def test_desktop_entry_icon_is_an_absolute_png_path(self):
        # Ship a hicolor mark so AppRun can copy it and point Icon= at the install (#10673).
        src = self.appdir / "usr/share/icons/hicolor/256x256/apps"
        src.mkdir(parents=True)
        (src / "io.tangencylabs.solidcraft.png").write_bytes(b"\x89PNG\r\n\x1a\n")
        (self.appdir / "usr/share/applications/io.tangencylabs.solidcraft.desktop").write_text(
            "[Desktop Entry]\nExec=solidcraft\nIcon=io.tangencylabs.solidcraft\n"
        )
        self.run_apprun()
        desk = (self.home / ".local/share/applications/io.tangencylabs.solidcraft.desktop").read_text()
        installed = self.home / ".local/share/icons/hicolor/256x256/apps/io.tangencylabs.solidcraft.png"
        self.assertTrue(installed.is_file())
        self.assertIn(f"Icon={installed}", desk)
        self.assertNotIn("Icon=io.tangencylabs.solidcraft\n", desk)


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