import asyncio
from playwright.async_api import async_playwright

JS_OVERLAY = r"""
async () => {
    const T=window.THREE, grp=window.__dzw._grp;
    const r=await fetch('jeonban.obj'); const txt=await r.text();
    const lines=txt.split('
'), vBuf=[], posArr=[];
    for(const line of lines){
        const p=line.trim().split(/\s+/);
        if(p[0]==='v'){ vBuf.push(parseFloat(p[1]),parseFloat(p[2]),parseFloat(p[3])); }
        else if(p[0]==='f'){
            const verts=[];
            for(let i=1;i<p.length;i++){ const sp=p[i].split('/'); verts.push((parseInt(sp[0])-1)*3); }
            for(let i=1;i<verts.length-1;i++)
                for(const vi of [verts[0],verts[i],verts[i+1]])
                    posArr.push(vBuf[vi],vBuf[vi+1],vBuf[vi+2]);
        }
    }
    const g=new T.BufferGeometry();
    g.setAttribute('position',new T.BufferAttribute(new Float32Array(posArr),3));
    g.computeVertexNormals();
    const m=new T.Mesh(g,new T.MeshPhongMaterial({color:0xff0000,transparent:true,opacity:0.7,wireframe:true}));
    m.raycast=()=>{};
    grp.add(m);
    return 'done';
}
"""

async def shot():
    async with async_playwright() as p:
        br = await p.chromium.launch(channel='msedge', headless=True)
        pg = await br.new_page(viewport={'width':1536,'height':864})
        await pg.goto('http://localhost:8090/DOZIKWORKS_OS_v5_locked.dc.html')
        print('로딩 45초...')
        await pg.wait_for_timeout(45000)
        await pg.evaluate(JS_OVERLAY)
        await pg.wait_for_timeout(2000)
        # 위에서 본 뷰
        await pg.evaluate("(()=>{ const a=window.__dzw; a.orbit.phi=0.02; a.orbit.r=8; a.updateCam(); })()")
        await pg.wait_for_timeout(800)
        await pg.screenshot(path='overlay_top.png')
        print('top 저장')
        # 정면
        await pg.evaluate("(()=>{ const a=window.__dzw; a.orbit.phi=1.3; a.orbit.theta=-1.57; a.orbit.r=6; a.updateCam(); })()")
        await pg.wait_for_timeout(800)
        await pg.screenshot(path='overlay_front.png')
        print('front 저장')
        await br.close()

asyncio.run(shot())