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

ROOT = Path('/mnt/data/nex_v301_side_surface_gate/NEXT11_V301_SIDE_SURFACE_CONTACT_GATE_ROOT')
OUT = ROOT / 'diagnostics/V301_BROWSER_RUNTIME_AUDIT.json'
SHOT_DIR = ROOT / 'diagnostics/v301_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_V301_FINGER_CHILD_LAYER_GATE || window.__NEXPROD01_V297_PREFINGER_CHILD_LAYER_GATE || null;
  const v110=window.__NEXPROD01_V110_DISTAL_GRIP_COUPLED || 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;
  try { if(model){ boneCount=0; model.traverse(o=>{ if(o?.isBone) boneCount++; 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, v301FingerGate:finger, v110DistalGrip:v110, v151MeshTruth:v151, activeHandProbe:probe, bodyText:(document.body?.innerText||'').slice(0,1000)};
}"""

def nested(d,*keys):
    cur=d
    for k in keys:
        if not isinstance(cur,dict): return None
        cur=cur.get(k)
    return cur

def summarize(final):
    final=final or {}
    grip=final.get('gripSolverV1') if isinstance(final,dict) else None
    finger=final.get('v301FingerGate') if isinstance(final,dict) else None
    probe=final.get('activeHandProbe') if isinstance(final,dict) else None
    v151=final.get('v151MeshTruth') if isinstance(final,dict) else None
    stages=(grip or {}).get('stages') or {} if isinstance(grip,dict) else {}
    entry=stages.get('v297EntryFrameGate') or {}
    side_gate=stages.get('v301SideSurfaceContactGate') or {}
    roll=stages.get('v300ForearmAxisSecondPalmRoll') or {}
    meet=stages.get('v300ObjectMeetGate') or {}
    checks=side_gate.get('checks') or {}
    return {
      'hasDebug': bool(final.get('hasDebug')),
      'hasModel': bool(final.get('hasModel')),
      'boneCount': final.get('boneCount'),
      'skinnedMeshCount': final.get('skinnedMeshCount'),
      'canvasCount': final.get('canvasCount'),
      '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,
      'palmContactReady': grip.get('palmContactReady') if isinstance(grip,dict) else None,
      'preFingerReady': grip.get('preFingerReady') if isinstance(grip,dict) else None,
      'entryGate': entry.get('gate'),
      'entryGatePass': entry.get('pass'),
      'entryRequiredOutwardShiftM': entry.get('requiredOutwardShiftM'),
      'sideSurfaceGateSchema': side_gate.get('schema'),
      'sideSurfaceGatePass': side_gate.get('pass'),
      'handPalmTowardObject': checks.get('handPalmTowardObject'),
      'outsideContact': checks.get('outsideContact'),
      'noGrossPenetration': checks.get('noGrossPenetration'),
      'firstPenetrationM': checks.get('firstPenetrationM'),
      'secondPenetrationM': checks.get('secondPenetrationM'),
      'firstOutsideScore': checks.get('firstOutsideScore'),
      'secondOutsideScore': checks.get('secondOutsideScore'),
      'v300RollPass': roll.get('pass'),
      'v300RollDeg': roll.get('rollDeg'),
      'v300ObjectMeetM': meet.get('meetM'),
      'v300ObjectMeetPass': meet.get('pass'),
      'fingerGateSchema': finger.get('schema') 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,
      'fingerReason': finger.get('reason') if isinstance(finger,dict) else None,
      'probeSide': probe.get('side') if isinstance(probe,dict) else None,
      'meshTruthSide': v151.get('side') if isinstance(v151,dict) else None,
    }

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

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':'v301-side-surface-contact-gate-browser-audit','build':'NEXPROD01_RUNTIME_V301_SIDE_SURFACE_CONTACT_GATE_2026_08_19','safeBoot':'0','lowPower':'1','pixelRatioCap':'1.15','antialias':'1','maxFPS':'24','staticAfterReady':'0','idleSleepAfterReady':'0','v':'V301_'+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(8):
        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
        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'),'fingerGateSchema':(val.get('v301FingerGate') or {}).get('schema')})
        if val.get('hasModel') and val.get('gripSolverV1'):
          page.wait_for_timeout(1000)
          final=page.evaluate(EVAL_JS)
          break
      shot=str(SHOT_DIR / f"V301_{case['name']}_side_surface_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)})
      page.close()
    browser.close()
finally:
  srv.shutdown(); srv.server_close()

audit={'schema':'NEXT11_V301_BROWSER_RUNTIME_SIDE_SURFACE_CONTACT_GATE_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'] and r['summary']['sideSurfaceGateSchema']=='NEXPROD01_V301_SIDE_SURFACE_CONTACT_GATE' for r in results)
audit['passSideSurfaceGate']=all(r['summary']['sideSurfaceGatePass'] is True and r['summary']['outsideContact'] is True and r['summary']['noGrossPenetration'] is True and r['summary']['handPalmTowardObject'] is True for r in results)
audit['passNoFingerReleaseByDefault']=all(r['summary']['fingerGateSchema']=='NEXPROD01_V301_FINGER_CHILD_LAYER_GATE' and r['summary']['fingerBlocked'] is True and r['summary']['fingerReady'] is True and r['summary']['fingerReleaseEnabled'] is False for r in results)
audit['passSideDiagnostics']=all((r['summary']['probeSide'] in (None,r['summary']['gripSide'])) and (r['summary']['meshTruthSide'] in (None,r['summary']['gripSide'])) for r in results)
OUT.write_text(json.dumps(audit,indent=2),encoding='utf-8')
print(json.dumps({'out':str(OUT),'passBrowserRuntime':audit['passBrowserRuntime'],'passSideSurfaceGate':audit['passSideSurfaceGate'],'passNoFingerReleaseByDefault':audit['passNoFingerReleaseByDefault'],'passSideDiagnostics':audit['passSideDiagnostics'],'summaries':[r['summary'] for r in results]},indent=2))
