#!/usr/bin/env python3
import json, os, socketserver, threading, http.server, time, urllib.parse, math
from pathlib import Path
from playwright.sync_api import sync_playwright

ROOT = Path('/mnt/data/nex_v303_sequential_finger_contact/NEXT11_V303_SEQUENTIAL_FINGER_CONTACT_ROOT')
OUT = ROOT / 'diagnostics/V303_BROWSER_RUNTIME_AUDIT.json'
SHOT_DIR = ROOT / 'diagnostics/v303_browser_screens'
SHOT_DIR.mkdir(parents=True, exist_ok=True)

class QuietHandler(http.server.SimpleHTTPRequestHandler):
    def log_message(self,*args): pass

def start_server():
    os.chdir(ROOT)
    srv=socketserver.TCPServer(('127.0.0.1',0), QuietHandler)
    threading.Thread(target=srv.serve_forever,daemon=True).start()
    return srv, srv.server_address[1]

EVAL_JS = """() => {
  const keys=Object.keys(window).filter(k=>k.startsWith('__NEXPROD01')).sort();
  const grip=window.__NEXPROD01_GRIP_SOLVER_V1 || null;
  const finger=window.__NEXPROD01_V303_SEQUENTIAL_FINGER_CONTACT || window.__NEXPROD01_V301_FINGER_CHILD_LAYER_GATE || null;
  const v151=window.__NEXPROD01_V151_ACTUAL_MESH_TRUTH || null;
  const probe=window.__NEXPROD01_V93_ACTIVE_HAND_CONTROL_PROBE || null;
  const debug=window.__NEXPROD01_DEBUG_SCENE || null;
  const model=debug?.model || null;
  let boneCount=null, skinnedMeshCount=0, canvasCount=document.querySelectorAll('canvas').length;
  let fingerBones={};
  try { if(model){ boneCount=0; model.traverse(o=>{ if(o?.isBone){ boneCount++; if(/^[LR]_(Index|Mid|Ring|Pinky|Thumb)[123]$/.test(o.name)){ const p=new THREE.Vector3(); o.getWorldPosition(p); fingerBones[o.name]={x:+p.x.toFixed(5),y:+p.y.toFixed(5),z:+p.z.toFixed(5)}; }} if(o?.isSkinnedMesh) skinnedMeshCount++; }); } } catch(e) {}
  return {href:location.href, ready:document.readyState, hasDebug:!!debug, hasModel:!!model, boneCount, skinnedMeshCount, canvasCount, windowMarkers:keys, markerCount:keys.length, gripSolverV1:grip, v303FingerContact:finger, v151MeshTruth:v151, activeHandProbe:probe, fingerBones, bodyText:(document.body?.innerText||'').slice(0,1000)};
}"""

def summarize(final, side):
    final=final or {}
    grip=final.get('gripSolverV1') if isinstance(final,dict) else None
    finger=final.get('v303FingerContact') if isinstance(final,dict) else None
    stages=(grip or {}).get('stages') or {} if isinstance(grip,dict) else {}
    side_gate=stages.get('v301SideSurfaceContactGate') or {}
    rows=(finger.get('rows') or []) if isinstance(finger,dict) else []
    active_rows=[r for r in rows if isinstance(r,dict) and r.get('active') is True]
    changed=(finger.get('parentDrift') or {}).get('changed') if isinstance(finger,dict) else None
    finger_results=finger.get('fingerResults') if isinstance(finger,dict) else []
    return {
      'hasDebug': bool(final.get('hasDebug')),
      'hasModel': bool(final.get('hasModel')),
      'boneCount': final.get('boneCount'),
      'skinnedMeshCount': final.get('skinnedMeshCount'),
      'hasGripSolverV1': bool(grip),
      'runtimeGripVersion': grip.get('version') if isinstance(grip,dict) else None,
      'gripSide': grip.get('side') if isinstance(grip,dict) else None,
      'gripPass': grip.get('pass') if isinstance(grip,dict) else None,
      'gripStatus': grip.get('status') if isinstance(grip,dict) else None,
      'sideSurfaceGatePass': side_gate.get('pass'),
      'fingerSchema': finger.get('schema') if isinstance(finger,dict) else None,
      'fingerActive': finger.get('active') if isinstance(finger,dict) else None,
      'fingerBlocked': finger.get('blocked') if isinstance(finger,dict) else None,
      'fingerReady': finger.get('ready') if isinstance(finger,dict) else None,
      'fingerReleaseEnabled': finger.get('releaseEnabled') if isinstance(finger,dict) else None,
      'fingerPass': finger.get('pass') if isinstance(finger,dict) else None,
      'activeFingerJointRows': len(active_rows),
      'parentDriftPass': (finger.get('parentDrift') or {}).get('pass') if isinstance(finger,dict) else None,
      'parentDriftChanged': changed,
      'thumbActiveRows': len([r for r in rows if isinstance(r,dict) and r.get('finger')=='thumb' and r.get('active') is True]),
      'perFinger': [{k:fr.get(k) for k in ['finger','contactOk','worstPenetrationM','terms']} for fr in finger_results] if isinstance(finger_results,list) else None,
    }

cases=[
  {'name':'left','side':'L','caseId':'v94_left_full','exercise':'hurdle-stretch','phase':0.25},
  {'name':'right','side':'R','caseId':'v94_right_full','exercise':'hurdle-stretch','phase':0.75},
]

srv, port = start_server()
results=[]
try:
  with sync_playwright() as p:
    env=os.environ.copy(); env['LIBGL_ALWAYS_SOFTWARE']='1'; env['VK_ICD_FILENAMES']='/usr/lib/chromium/vk_swiftshader_icd.json'; env['DISPLAY']=''
    browser=p.chromium.launch(executable_path='/usr/bin/chromium',headless=True,env=env,args=['--no-sandbox','--disable-dev-shm-usage','--ignore-gpu-blocklist','--enable-webgl','--enable-webgl2','--enable-unsafe-swiftshader','--disable-gpu-sandbox','--ozone-platform=headless','--use-gl=angle','--use-angle=swiftshader-webgl','--remote-allow-origins=*'])
    context=browser.new_context(viewport={'width':1280,'height':900}, device_scale_factor=1)
    for case in cases:
      page=context.new_page()
      params={'exercise':case['exercise'],'caseId':case['caseId'],'phase':str(case['phase']),'play':'0','debug':'0','visualMode':'nex','nativeRuntimeParity':'1','nativeSkin':'1','skinPreset':'freeridehuman','readyTimeoutMs':'12000','hostLifecycle':'v303-sequential-finger-contact-browser-audit','build':'NEXPROD01_RUNTIME_V303_SEQUENTIAL_FINGER_CONTACT_2026_08_19','safeBoot':'0','lowPower':'1','pixelRatioCap':'1.15','antialias':'1','maxFPS':'24','staticAfterReady':'0','idleSleepAfterReady':'0','v':'V303_'+case['name']}
      url=f'http://127.0.0.1:{port}/assets/nexprod/nexprod01-player.html?'+urllib.parse.urlencode(params)
      nav_error=None
      try: page.goto(url, wait_until='domcontentloaded', timeout=15000)
      except Exception as e: nav_error=repr(e)
      polls=[]; final=None
      for i in range(12):
        page.wait_for_timeout(1000)
        try: val=page.evaluate(EVAL_JS)
        except Exception as e:
          polls.append({'t':i+1,'evalError':repr(e)}); continue
        final=val
        finger=val.get('v303FingerContact') or {}
        polls.append({'t':i+1,'hasDebug':val.get('hasDebug'),'hasModel':val.get('hasModel'),'boneCount':val.get('boneCount'),'markerCount':val.get('markerCount'),'gripSchema':(val.get('gripSolverV1') or {}).get('schema'),'fingerSchema':finger.get('schema'),'fingerActive':finger.get('active'),'fingerPass':finger.get('pass')})
        if val.get('hasModel') and val.get('gripSolverV1') and finger.get('schema'):
          page.wait_for_timeout(1000)
          final=page.evaluate(EVAL_JS)
          break
      shot=str(SHOT_DIR / f"V303_{case['name']}_sequential_finger_contact_player.png")
      try: page.screenshot(path=shot, full_page=False, timeout=5000)
      except Exception: shot=None
      results.append({'case':case,'url':url,'navigationError':nav_error,'polls':polls,'final':final,'screenshot':shot,'summary':summarize(final,case['side'])})
      page.close()
    browser.close()
finally:
  srv.shutdown(); srv.server_close()

audit={'schema':'NEXT11_V303_BROWSER_RUNTIME_SEQUENTIAL_FINGER_CONTACT_AUDIT','generatedAt':time.strftime('%Y-%m-%dT%H:%M:%SZ',time.gmtime()),'root':str(ROOT),'cases':results}
audit['passBrowserRuntime']=all(r['summary']['hasModel'] and r['summary']['hasGripSolverV1'] for r in results)
audit['passSideSurfaceGate']=all(r['summary']['sideSurfaceGatePass'] is True for r in results)
audit['passFingerLayerActive']=all(r['summary']['fingerSchema']=='NEXPROD01_V303_SEQUENTIAL_FINGER_CONTACT_CHILD_LAYER' and r['summary']['fingerActive'] is True and r['summary']['fingerBlocked'] is False and r['summary']['fingerReleaseEnabled'] is True for r in results)
audit['passChildOnly']=all(r['summary']['parentDriftPass'] is True and r['summary']['thumbActiveRows']==0 for r in results)
audit['passHasActiveJointRows']=all(r['summary']['activeFingerJointRows']>0 for r in results)
audit['pass']=audit['passBrowserRuntime'] and audit['passSideSurfaceGate'] and audit['passFingerLayerActive'] and audit['passChildOnly'] and audit['passHasActiveJointRows']
OUT.write_text(json.dumps(audit,indent=2),encoding='utf-8')
print(json.dumps({'out':str(OUT),'pass':audit['pass'],'flags':{k:v for k,v in audit.items() if k.startswith('pass')},'summaries':[r['summary'] for r in results]},indent=2))
