#!/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_v314_mcp_only_authority/NEXT11_V314_ACTUAL_MCP_ONLY_PALM_AUTHORITY_ROOT')
OUT=ROOT/'diagnostics/v314'; OUT.mkdir(parents=True, exist_ok=True)
class Quiet(http.server.SimpleHTTPRequestHandler):
    def log_message(self,*args): pass
EVAL_JS=r'''() => {
 const keys=Object.keys(window).filter(k=>k.startsWith('__NEXPROD01')).sort();
 const grip=window.__NEXPROD01_GRIP_SOLVER_V1||null;
 const finalTruth=window.__NEXPROD01_V314_FINAL_PREFINGER_TRUTH||window.__NEXPROD01_V311_FINAL_PREFINGER_TRUTH||null;
 const v313Global=window.__NEXPROD01_V314_ACTUAL_MCP_ONLY_PALM_AUTHORITY||null;
 const gate=window.__NEXPROD01_V314_ACTUAL_MCP_PALM_PREFINGER_GATE||window.__NEXPROD01_V311_ACTUAL_MCP_PALM_PREFINGER_GATE||window.__NEXPROD01_V308_ACTUAL_MCP_PALM_PREFINGER_GATE||null;
 const finger=window.__NEXPROD01_V301_FINGER_CHILD_LAYER_GATE||window.__NEXPROD01_V304_ZPLUS_SEQUENTIAL_FINGER_CONTACT||null;
 const debug=window.__NEXPROD01_DEBUG_SCENE||null;
 const model=debug?.model||null;
 let boneCount=null, skinnedMeshCount=0;
 let bones={};
 try{ if(model){ boneCount=0; model.traverse(o=>{ if(o?.isBone){boneCount++; if(/^L_(Forearm|Hand|Index1|Pinky1|Mid1|Index2|Pinky2|Mid2|Thumb1|Thumb2)/.test(o.name)){ const p=new THREE.Vector3(); o.getWorldPosition(p); bones[o.name]={x:+p.x.toFixed(5),y:+p.y.toFixed(5),z:+p.z.toFixed(5), q:{x:+o.quaternion.x.toFixed(6),y:+o.quaternion.y.toFixed(6),z:+o.quaternion.z.toFixed(6),w:+o.quaternion.w.toFixed(6)}}; }} if(o?.isSkinnedMesh) skinnedMeshCount++; }); }}catch(e){bones.error=String(e)}
 const commit = (debug&&debug.lastResult&&debug.lastResult.commit)||null;
 const v314Step = v313Global || commit?.steps?.find?.(s=>s?.schema==='NEXPROD01_V314_WRIST_NODE_180_RESTART') || null;
 return {ready:document.readyState, href:location.href, hasDebug:!!debug, hasModel:!!model,boneCount,skinnedMeshCount,canvasCount:document.querySelectorAll('canvas').length,windowMarkers:keys, gripSolverV1:grip, finalTruth, actualGate:gate, fingerLayer:finger, commitMode:commit?.mode||null, v314Step, bones, bodyText:(document.body?.innerText||'').slice(0,800)};
}'''
def summarise(val):
    val=val or {}; grip=val.get('gripSolverV1') or {}; stages=grip.get('stages') or {}; final=val.get('finalTruth') or {}; gate=val.get('actualGate') or {}; mcp=(gate.get('v307ActualMcpPalmGate') or {}) if isinstance(gate,dict) else {}; metrics=mcp.get('metrics') or {}; step=val.get('v314Step') or {}; finger=val.get('fingerLayer') or {}
    return {'hasModel':val.get('hasModel'),'boneCount':val.get('boneCount'),'skinnedMeshCount':val.get('skinnedMeshCount'),'runtimeVersion':grip.get('version'),'side':grip.get('side'),'gripPass':grip.get('pass'),'gripStatus':grip.get('status'),'v314StepApplied':step.get('applied'),'v314StepPass':step.get('pass'),'v313ChangedFingerLocals':len(step.get('changedFingerLocals') or []),'v313RotationDeg':step.get('rotationDeg'),'actualGatePass':gate.get('pass'),'actualGateStatus':gate.get('status'),'wrongWay':gate.get('wrongWay'),'orientationFail':gate.get('orientationFail'),'indexPen':((metrics.get('index') or {}).get('penetrationM')),'pinkyPen':((metrics.get('pinky') or {}).get('penetrationM')),'indexDist':((metrics.get('index') or {}).get('distanceToSurfaceM')),'pinkyDist':((metrics.get('pinky') or {}).get('distanceToSurfaceM')),'fingerBlocked':finger.get('blocked'),'fingerActive':finger.get('active'),'fingerReady':finger.get('ready')}
def main():
    os.chdir(ROOT)
    srv=socketserver.TCPServer(('127.0.0.1',0),Quiet); threading.Thread(target=srv.serve_forever,daemon=True).start(); port=srv.server_address[1]
    cases=[('right','v94_right_full','0.75'),('left','v94_left_full','0.25')]
    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=*'])
        ctx=browser.new_context(viewport={'width':1400,'height':1000},device_scale_factor=1)
        for name,caseId,phase in cases:
          page=ctx.new_page()
          params={'exercise':'hurdle-stretch','caseId':caseId,'phase':phase,'play':'0','debug':'0','visualMode':'nex','nativeRuntimeParity':'1','nativeSkin':'1','skinPreset':'freeridehuman','readyTimeoutMs':'12000','hostLifecycle':'v314-mcp-only-authority','build':'NEXPROD01_RUNTIME_V314_WRIST_NODE_180_RESTART_2026_08_19','safeBoot':'0','lowPower':'1','pixelRatioCap':'1.15','antialias':'1','maxFPS':'24','staticAfterReady':'0','idleSleepAfterReady':'0','v':'V314_'+name}
          url=f'http://127.0.0.1:{port}/assets/nexprod/nexprod01-player.html?'+urllib.parse.urlencode(params)
          navErr=None; final=None; polls=[]
          try: page.goto(url,wait_until='domcontentloaded',timeout=15000)
          except Exception as e: navErr=repr(e)
          for i in range(24):
            page.wait_for_timeout(650)
            try: val=page.evaluate(EVAL_JS)
            except Exception as e: polls.append({'i':i,'evalErr':repr(e)}); continue
            final=val; polls.append({'i':i,'hasModel':val.get('hasModel'),'hasGrip':bool(val.get('gripSolverV1'))})
            if val.get('hasModel') and val.get('gripSolverV1'):
              page.wait_for_timeout(500); final=page.evaluate(EVAL_JS); break
          shot_canvas=str(OUT/f'V314_{name}_canvas.png')
          shot_page=str(OUT/f'V314_{name}_page.png')
          try: page.locator('canvas').first.screenshot(path=shot_canvas,timeout=8000)
          except Exception:
            try: page.screenshot(path=shot_canvas,full_page=False)
            except Exception: shot_canvas=None
          try: page.screenshot(path=shot_page,full_page=False)
          except Exception: shot_page=None
          results.append({'name':name,'caseId':caseId,'phase':phase,'url':url,'navigationError':navErr,'polls':polls,'final':final,'summary':summarise(final),'screenshotCanvas':shot_canvas,'screenshotPage':shot_page})
          page.close()
        browser.close()
    finally:
      srv.shutdown(); srv.server_close()
    audit={'schema':'NEXT11_V314_BROWSER_MCP_ONLY_AUTHORITY_AUDIT','generatedAt':time.strftime('%Y-%m-%dT%H:%M:%SZ',time.gmtime()),'root':str(ROOT),'results':results,'summaries':[r['summary'] for r in results]}
    audit['passBrowserRuntime']=all(r['summary']['hasModel'] and r['summary']['runtimeVersion']=='V314_ACTUAL_MCP_ONLY_PALM_AUTHORITY' for r in results)
    audit['leftWristNodeStepObserved']=any(r['summary']['side']=='L' and r['summary']['v314StepApplied'] for r in results)
    audit['noFingerRelease']=all(r['summary']['fingerBlocked'] is True and r['summary']['fingerActive'] is not True for r in results)
    audit['noFingerLocalRotationInV312Step']=all((r['summary'].get('v313ChangedFingerLocals') in (None,0)) for r in results)
    (OUT/'V314_BROWSER_MCP_ONLY_AUTHORITY_AUDIT.json').write_text(json.dumps(audit,indent=2),encoding='utf-8')
    print(json.dumps({'audit':str(OUT/'V314_BROWSER_MCP_ONLY_AUTHORITY_AUDIT.json'),'passBrowserRuntime':audit['passBrowserRuntime'],'leftWristNodeStepObserved':audit['leftWristNodeStepObserved'],'noFingerRelease':audit['noFingerRelease'],'noFingerLocalRotationInV312Step':audit['noFingerLocalRotationInV312Step'],'summaries':audit['summaries']},indent=2))
if __name__=='__main__': main()