"""Pack tiny shader fixtures through build_ipa.sh's archive step, without Xcode or game files."""
import os
from pathlib import Path
import subprocess
import sys
import tempfile
import unittest
import zipfile

REPO = Path(__file__).resolve().parents[3]
sys.path.insert(0, str(REPO / 'dm/games'))
import game
sys.path.insert(0, str(REPO / 'dm/tools'))
import shader_lib


class ProfilePackTests(unittest.TestCase):
    def test_an_image_without_pulling_libraries_stops_an_app_that_needs_them(self):
        with tempfile.TemporaryDirectory() as temporary:
            root = Path(temporary); lib = root / 'b-fixture'; lib.mkdir(); (lib / 'k_b.metallib').write_text('lib'); (lib / 'k_bp.metallib').write_text('pull')
            shader_lib.pack(lib, leave_out=True)
            def pack(*options): return subprocess.run([sys.executable, str(REPO / 'dm/ios/tools/pack_ipa.py'), str(root / 'x.ipa'), *options, f'Payload/DmIOS.app/msl/b-fixture={lib}'], capture_output=True, text=True)
            r = pack(); self.assertNotEqual(r.returncode, 0); self.assertIn('no pulling vertex libraries', r.stdout + r.stderr)
            r = pack('--skip-suffix', '_bp.metallib'); self.assertEqual(r.returncode, 0, r.stdout + r.stderr)
            with zipfile.ZipFile(root / 'x.ipa') as archive: self.assertIn('Payload/DmIOS.app/msl/b-fixture.img', archive.namelist())
            self.assertEqual([n for n in os.listdir(root) if '.msl-' in n], [])

    def test_pulling_libraries_follow_compiled_profile_and_explicit_override(self):
        # (game, IOS_PULL, with a pipeline archive, _bp libraries packed): an archive's bases are made of the pulling libraries, so an IPA with one carries them whatever IOS_PULL says
        cases = [('01007EF00011E000', None, False, True), ('010015100B514000', None, False, False),
                 ('01007EF00011E000', '0', False, False), ('010015100B514000', '1', False, True),
                 ('010015100B514000', None, True, True), ('01007EF00011E000', '0', True, True)]
        for gid, override, with_pso, included, packed in [c + (p,) for c in cases for p in (False, True)]:
            with self.subTest(game=gid, IOS_PULL=override, archive=with_pso, one_file=packed), tempfile.TemporaryDirectory() as temporary:
                root = Path(temporary)
                dm = root / 'dm'
                header = dm / 'host/gen' / gid / 'game_profile.h'
                header.parent.mkdir(parents=True)
                header.write_text(game.header(game.Game(REPO / 'dm/games' / gid)))
                app, shaders, romfs = [root / name for name in ('DmIOS.app', 'shaders', 'romfs')]
                for folder in (app, shaders / 'b-fixture', romfs): folder.mkdir(parents=True)
                for name in ('fixture_b.metallib', 'fixture_bp.metallib', 'fixture_bv.metallib', 'fixture_b.json'):
                    (shaders / 'b-fixture' / name).write_text(name)
                if packed: self.assertTrue(shader_lib.pack(shaders / 'b-fixture'))
                (app / 'executable').write_text('app')
                (romfs / 'marker').write_text('romfs')
                (root / 'tools').symlink_to(REPO / 'dm/ios/tools', target_is_directory=True)
                out = root / 'fixture.ipa'
                env = {k: v for k, v in os.environ.items() if k not in ('IOS_PULL', 'DM_GAMES_DIR', 'DM_GAMES_FIRST')}
                env.update(GAME=gid, DM=str(dm), APP=str(app), MSL_IOS=str(shaders), MSLDIR='b-fixture',
                           AOC='', ROMFS=str(romfs), OUT_IPA=str(out), PROFILE_TOOL=str(REPO / 'dm/games/game.py'))
                if override is not None: env['IOS_PULL'] = override
                if with_pso:
                    pso = root / 'pso'; pso.mkdir(); (pso / 'prebuilt.tsv').write_text('fixture')
                    env['PSO'] = str(pso)
                # Exercise the actual archive step; only the preceding Xcode build is bypassed.
                script = (REPO / 'dm/ios/tools/build_ipa.sh').read_text()
                packing = script[script.index('\nP=Payload/DmIOS.app\n'):]
                result = subprocess.run(['/bin/bash', '-c', 'set -u\nt() { :; }; die() { echo "$*" >&2; exit 1; }\n'
                                         'gp() { python3 "$PROFILE_TOOL" "$GAME" "$1"; }\n' + packing],
                                        cwd=root, env=env, capture_output=True, text=True)
                self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
                with zipfile.ZipFile(out) as archive:
                    names = set(archive.namelist())
                    prefix = 'Payload/DmIOS.app/msl/b-fixture/'
                    if packed:
                        self.assertFalse([n for n in names if n.startswith(prefix)])
                        (root / 'taken.img').write_bytes(archive.read('Payload/DmIOS.app/msl/b-fixture.img'))
                        names = {prefix + p.name for p in shader_lib.romfs.Path(root / 'taken.img').iterdir()} | names
                        self.assertEqual([n for n in os.listdir(root) if '.msl-' in n], [], 'the copy without the pulling libraries is gone')
                    self.assertEqual(prefix + 'fixture_bp.metallib' in names, included)
                    for name in ('fixture_b.metallib', 'fixture_bv.metallib', 'fixture_b.json'):
                        self.assertIn(prefix + name, names)
                    # the fork's slim package: the RomFS stays out of the IPA (DM_FULL_ROMFS=1 packs it)
                    self.assertFalse(any(n.startswith('Payload/DmIOS.app/romfs') for n in names))
                    self.assertEqual('Payload/DmIOS.app/pso/prebuilt.tsv' in names, with_pso)


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