# -*- coding: utf-8 -*-
"""截当前/指定页的图：python .claude/shot.py <pageUuid> <out.png>"""
import base64, json, sys, os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from eda import run
page, out = sys.argv[1], sys.argv[2]
js = """
const sleep = ms => new Promise(r => setTimeout(r, ms));
await eda.dmt_EditorControl.openDocument('%s');
await sleep(1500);
await eda.dmt_EditorControl.zoomToAllPrimitives();
await sleep(1500);
const img = await eda.dmt_EditorControl.getCurrentRenderedAreaImage();
const bytes = new Uint8Array(await new Response(img).arrayBuffer());
let s=''; for (let i=0;i<bytes.length;i++) s += String.fromCharCode(bytes[i]);
return {page:(await eda.dmt_Schematic.getCurrentSchematicPageInfo()).name, b64: btoa(s)};
""" % page
r = run(js, timeout=300)
assert r.get('success'), json.dumps(r, ensure_ascii=False)[:400]
open(out, 'wb').write(base64.b64decode(r['result']['b64']))
print(r['result']['page'], '->', out)