"""Headset panel packaging, ownership checks and window targeting."""
import sandbox  # noqa: F401
import io
import json
import os
from pathlib import Path
import secrets
import subprocess
import sys
import tarfile
import tempfile
import unittest
from unittest.mock import patch

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / 'ui'))
import frame_headset
import frame_headset_runtime


class HeadsetBundle(unittest.TestCase):
    def test_bundle_is_deterministic_and_contains_the_local_ui(self):
        first = frame_headset.bundle()
        self.assertEqual(first, frame_headset.bundle())
        with tarfile.open(fileobj=io.BytesIO(first), mode='r:gz') as archive:
            names = set(archive.getnames())
        self.assertIn('ui/headset.html', names)
        self.assertIn('ui/frame_headset_runtime.py', names)
        self.assertIn('ui/server.py', names)
        self.assertFalse(any('\\' in name for name in names))
        self.assertFalse(any('__pycache__' in name for name in names))

    def test_incompatible_device_is_rejected_before_upload(self):
        state = {'arch': 'x86_64', 'variant': 'vr'}
        calls = []
        def ssh(*args, **kwargs):
            calls.append(args)
            return json.dumps(state)
        with patch.object(frame_headset, 'bundle') as bundle:
            with self.assertRaisesRegex(ValueError, 'Steam Frame'):
                frame_headset.change(ssh, 'install')
            bundle.assert_not_called()
        self.assertEqual(len(calls), 1)

    @unittest.skipIf(os.name == 'nt', 'remote script runs on Linux')
    def test_unowned_install_is_never_removed(self):
        with tempfile.TemporaryDirectory() as home:
            token = secrets.token_hex(12)
            result = subprocess.run([sys.executable, '-', 'remove', token, 'none'],
                                    input=frame_headset.APPLY, text=True, capture_output=True,
                                    env={**os.environ, 'HOME': home}, timeout=5)
            self.assertNotEqual(result.returncode, 0)
            self.assertIn('No Frame Control managed', result.stderr)


class HeadsetWindow(unittest.TestCase):
    def test_uses_only_its_marker_title(self):
        tree = ' 0x101 "Other panel"\n 0xabc "Frame Control VR [fc-headset]"\n'
        with patch.object(frame_headset_runtime, 'command', return_value=tree):
            self.assertEqual(frame_headset_runtime.panel_window(), '0xabc')
        with patch.object(frame_headset_runtime, 'command', return_value=' 0x101 "Other panel"'):
            self.assertIsNone(frame_headset_runtime.panel_window())


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