from __future__ import annotations

import io
import plistlib
import struct
import sys
import tarfile
import tempfile
import unittest
import zipfile
from pathlib import Path


RELEASE_SCRIPTS = Path(__file__).resolve().parents[1] / "release"
sys.path.insert(0, str(RELEASE_SCRIPTS))

import verify_native_package  # noqa: E402


def fake_pe(machine: int) -> bytes:
    data = bytearray(512)
    data[:2] = b"MZ"
    struct.pack_into("<I", data, 0x3C, 0x80)
    data[0x80:0x84] = b"PE\0\0"
    struct.pack_into("<H", data, 0x84, machine)
    return bytes(data)


def fake_macho(cpu_type: int) -> bytes:
    return b"\xcf\xfa\xed\xfe" + cpu_type.to_bytes(4, "little") + bytes(504)


class VerifyNativePackageTests(unittest.TestCase):
    def test_archive_paths_reject_parent_traversal_and_absolute_paths(self) -> None:
        for path in ("../secret", "dir/../../secret", "/absolute/file"):
            with self.subTest(path=path), self.assertRaises(RuntimeError):
                verify_native_package.require_safe_archive_path(path)
        verify_native_package.require_safe_archive_path("NyaTerm/dir/file")

    def test_windows_portable_has_required_entries(self) -> None:
        with tempfile.TemporaryDirectory() as directory:
            path = Path(directory) / "portable.zip"
            root = "NyaTerm-portable"
            with zipfile.ZipFile(path, "w") as archive:
                archive.writestr(f"{root}/NyaTerm.exe", fake_pe(0x8664))
                archive.writestr(f"{root}/nyaterm-portable", b"")
                archive.writestr(f"{root}/LICENSE", b"license")
                archive.writestr(f"{root}/VERSION", b"2.0.0\n")
                archive.writestr(f"{root}/data/.keep", b"")
            verify_native_package.verify_windows_portable(
                path, "x86_64-pc-windows-msvc", "2.0.0"
            )

    def test_windows_portable_rejects_wrong_architecture(self) -> None:
        with tempfile.TemporaryDirectory() as directory:
            path = Path(directory) / "portable.zip"
            root = "NyaTerm-portable"
            with zipfile.ZipFile(path, "w") as archive:
                archive.writestr(f"{root}/NyaTerm.exe", fake_pe(0xAA64))
                archive.writestr(f"{root}/nyaterm-portable", b"")
                archive.writestr(f"{root}/LICENSE", b"license")
                archive.writestr(f"{root}/VERSION", b"2.0.0\n")
                archive.writestr(f"{root}/data/.keep", b"")
            with self.assertRaisesRegex(RuntimeError, "PE machine"):
                verify_native_package.verify_windows_portable(
                    path, "x86_64-pc-windows-msvc", "2.0.0"
                )

    def test_macos_archive_validates_bundle_metadata_and_architecture(self) -> None:
        with tempfile.TemporaryDirectory() as directory:
            path = Path(directory) / "NyaTerm.app.tar.gz"
            entries = {
                "NyaTerm.app/Contents/MacOS/NyaTerm": fake_macho(0x0100000C),
                "NyaTerm.app/Contents/Info.plist": plistlib.dumps(
                    {
                        "CFBundleIdentifier": "com.kang.nyaterm",
                        "CFBundleShortVersionString": "2.0.0",
                    }
                ),
                "NyaTerm.app/Contents/Resources/VERSION": b"2.0.0\n",
                "NyaTerm.app/Contents/Resources/LICENSE": b"license",
                "NyaTerm.app/Contents/Resources/icon.icns": b"icon",
            }
            with tarfile.open(path, "w:gz") as archive:
                for name, data in entries.items():
                    item = tarfile.TarInfo(name)
                    item.size = len(data)
                    archive.addfile(item, io.BytesIO(data))
            verify_native_package.verify_macos_archive(
                path, "aarch64-apple-darwin", "2.0.0"
            )

    def test_release_verification_fails_before_platform_tools_when_asset_missing(self) -> None:
        with tempfile.TemporaryDirectory() as directory:
            with self.assertRaisesRegex(RuntimeError, "missing release artifacts"):
                verify_native_package.verify_release(
                    Path(directory), "x86_64-unknown-linux-gnu", "2.0.0"
                )

    def test_binary_header_helpers_reject_invalid_formats(self) -> None:
        with self.assertRaises(RuntimeError):
            verify_native_package.pe_machine(b"not-pe")
        with self.assertRaises(RuntimeError):
            verify_native_package.elf_machine(b"not-elf")
        with self.assertRaises(RuntimeError):
            verify_native_package.macho_cpu_type(b"not-macho")


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