#!/usr/bin/env python3
"""v0.322.0 burndown-2 browser check: the Data tab's export key field and the 2nd-copy part. Minimal CDP client: log in to the controller through its PUBLIC address in headless Chrome, open each
app-page tab, assert computed styles, collect console errors, save screenshots. The password is read
from a 0600 file and only ever sent inside a CDP message; it is never printed."""
import base64, json, os, socket, struct, subprocess, sys, time, urllib.request

CHROME = os.path.expanduser('~/.cache/ms-playwright/chromium_headless_shell-1243/chrome-headless-shell-linux64/chrome-headless-shell')
BASE = sys.argv[1]            # https://felhom.enkisfelhom.hu
PWFILE = sys.argv[2]          # the controller password (0600 file)
FBPWFILE = sys.argv[3]        # FileBrowser's admin password (0600 file) — for the cold „Fajlok" click
OUT = sys.argv[4]
PORT = 9333
os.makedirs(OUT, exist_ok=True)
prof = os.path.join(OUT, '.profile')
proc = subprocess.Popen([CHROME, f'--remote-debugging-port={PORT}', '--no-sandbox', f'--user-data-dir={prof}',
                         '--window-size=1280,900', 'about:blank'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
try:
    for _ in range(50):
        try:
            tabs = json.load(urllib.request.urlopen(f'http://127.0.0.1:{PORT}/json'))
            break
        except Exception:
            time.sleep(0.2)
    ws_url = [t for t in tabs if t['type'] == 'page'][0]['webSocketDebuggerUrl']
    host, path = ws_url[len('ws://'):].split('/', 1)
    h, p = host.split(':')
    sock = socket.create_connection((h, int(p)))
    key = base64.b64encode(os.urandom(16)).decode()
    sock.send(f'GET /{path} HTTP/1.1\r\nHost: {host}\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: {key}\r\nSec-WebSocket-Version: 13\r\n\r\n'.encode())
    buf = b''
    while b'\r\n\r\n' not in buf:
        buf += sock.recv(4096)
    st = {'pending': buf.split(b'\r\n\r\n', 1)[1]}

    def send(obj):
        data = json.dumps(obj).encode()
        hdr = bytearray([0x81])
        n = len(data)
        if n < 126: hdr.append(0x80 | n)
        elif n < 65536: hdr.append(0x80 | 126); hdr += struct.pack('>H', n)
        else: hdr.append(0x80 | 127); hdr += struct.pack('>Q', n)
        mask = os.urandom(4); hdr += mask
        sock.sendall(bytes(hdr) + bytes(b ^ mask[i % 4] for i, b in enumerate(data)))

    def recv_exact(n):
        while len(st['pending']) < n:
            st['pending'] += sock.recv(1 << 20)
        out, st['pending'] = st['pending'][:n], st['pending'][n:]
        return out

    def recv():
        msg = b''
        while True:
            b0, b1 = recv_exact(2)
            n = b1 & 0x7f
            if n == 126: n = struct.unpack('>H', recv_exact(2))[0]
            elif n == 127: n = struct.unpack('>Q', recv_exact(8))[0]
            msg += recv_exact(n)
            if b0 & 0x80: return json.loads(msg)

    console = []
    mid = [0]
    def call(method, **params):
        mid[0] += 1; i = mid[0]
        send({'id': i, 'method': method, 'params': params})
        while True:
            m = recv()
            if m.get('method') == 'Runtime.exceptionThrown':
                console.append(('exception', m['params']['exceptionDetails'].get('text', ''), m['params']['exceptionDetails'].get('exception', {}).get('description', '')[:200]))
            elif m.get('method') == 'Runtime.consoleAPICalled' and m['params']['type'] in ('error', 'warning'):
                console.append((m['params']['type'], ' '.join(str(a.get('value', a.get('description', ''))) for a in m['params']['args'])[:200]))
            elif m.get('method') == 'Log.entryAdded' and m['params']['entry']['level'] in ('error', 'warning'):
                e = m['params']['entry']; console.append(('log-' + e['level'], e.get('text', '')[:200], e.get('url', '')))
            if m.get('id') == i:
                if 'error' in m: raise RuntimeError(f'{method}: {m["error"]}')
                return m.get('result', {})

    def ev(expr):
        r = call('Runtime.evaluate', expression=expr, returnByValue=True, awaitPromise=True)
        return r.get('result', {}).get('value')

    def goto(url):
        call('Page.navigate', url=url)
        for _ in range(100):
            time.sleep(0.2)
            if ev('document.readyState') == 'complete':
                break
        time.sleep(1.5)  # images' onerror, layout

    for d in ('Page', 'Runtime', 'Log', 'Network'):
        call(d + '.enable')
    pw = open(PWFILE).read().strip()
    goto(BASE + '/login')
    ev("(function(){var f=document.querySelector('input[type=password]'); f.value=%s; f.form.submit(); return 1})()" % json.dumps(pw))
    del pw
    time.sleep(3)
    print('after login url:', ev('location.pathname'))
    console.clear()  # the login page is not under test

    def shot(name):
        s = call('Page.captureScreenshot', format='png', captureBeyondViewport=True)
        open(os.path.join(OUT, name + '.png'), 'wb').write(base64.b64decode(s['data']))
    goto(BASE + '/apps/calibre-web/data')
    time.sleep(2)  # give any autofill its moment
    r = ev("""(function(){ var k=document.getElementById('fabEncryptionKey');
        return {path: location.pathname, keyField: k ? {name: k.name, autocomplete: k.getAttribute('autocomplete'), type: k.type, valueLength: k.value.length} : null,
          oldIdPresent: !!document.getElementById('exportPassword'),
          statusRows: document.querySelectorAll('[data-backup-row="second"],[data-backup-row="target"]').length,
          targetLabels: document.querySelectorAll('label[for="target"]').length,
          form: !!document.querySelector('[data-second-backup-form]'),
          warnCallout: !!document.querySelector('#app-backup-section .app-callout-warn'),
          scrollWidth: document.documentElement.scrollWidth}; })()""")
    r['console'] = console[:]
    shot('calibre-web-data')
    print(json.dumps(r, indent=1, ensure_ascii=False))
finally:
    proc.terminate()
    subprocess.run(['rm', '-rf', prof])