클라우드 브라우저로 웹 페이지 찍기

#!/usr/bin/env node
// 클라우드 브라우저로 웹 페이지 찍기 — endue OS 샘플.
//   chrome 가상 기기를 띄우고 → 주소로 이동하고 → 탭을 읽고 → 화면 한 장을 PNG 로 저장한다.
//
//   node chrome-screenshot.mjs https://example.com shot.png
//   ENDUE_OS_TOKEN=… ENDUE_OS_URL=… node …   운영 — endue 계정의 토큰
import { writeFile } from 'node:fs/promises';

const [url = 'https://example.com', out = 'shot.png'] = process.argv.slice(2);
const BASE = (process.env.ENDUE_OS_URL ?? 'http://127.0.0.1:5173').replace(/\/+$/, '');
const API = `${BASE}/api/v1`;
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));

const token =
  process.env.ENDUE_OS_TOKEN ??
  (await (await fetch(`${BASE}/idp/token?sub=${process.env.ENDUE_OS_SUB ?? `sample-${Date.now().toString(36)}`}&ttl=3600`, { method: 'POST' })).json()).access_token;

async function api(method, path, body, { raw = false } = {}) {
  const res = await fetch(`${API}${path}`, {
    method,
    headers: { Authorization: `Bearer ${token}`, ...(body ? { 'Content-Type': 'application/json' } : {}) },
    body: body ? JSON.stringify(body) : undefined,
  });
  if (raw) return res;
  const json = res.status === 204 ? null : await res.json();
  if (res.status === 402) throw new Error(`${method} ${path} → 402 ${json?.error?.code}: 크레딧이 부족해요 (잔액 ${json?.error?.details?.balance ?? '?'}). 충전한 뒤 다시 돌려 주세요.`);
  if (!res.ok) throw new Error(`${method} ${path} → ${res.status} ${json?.error?.code}`);
  return { status: res.status, data: json?.data };
}

const station = (await api('POST', '/stations', { name: 'web-shots' })).data;
let device;
try {
  device = (await api('POST', `/stations/${station.station_id}/devices`, { type: 'virtual', kind: 'chrome', name: 'browser' })).data.device;
  while (!(device.compute_status === 'running' && device.online)) {
    if (device.compute_status === 'failed') throw new Error(device.status_reason);
    await sleep(2000);
    device = (await api('GET', `/devices/${device.device_id}`)).data;
  }
  const rid = device.device_id;
  console.log(`chrome running — ${rid}`);

  const nav = (await api('POST', `/devices/${rid}/commands`, { command: 'navigate', input: { url } })).data;
  if (nav.status !== 'succeeded') throw new Error(`navigate: ${nav.status} ${nav.error}`);
  const tabs = (await api('POST', `/devices/${rid}/commands`, { command: 'tab_list', input: {} })).data.output?.tabs ?? [];
  const active = tabs.find((t) => t.active);
  console.log(`활성 탭: ${active?.title ?? '?'} — ${active?.url}`);

  await sleep(1000); // 페이지가 그려질 틈
  // 화면 API — 명령 기록에 남지 않고, max_age_ms 안의 화면은 다시 찍지 않는다.
  const res = await api('GET', `/devices/${rid}/screen?max_age_ms=500`, undefined, { raw: true });
  if (res.status !== 200) throw new Error(`screen → ${res.status} ${await res.text()}`);
  const bytes = Buffer.from(await res.arrayBuffer());
  await writeFile(out, bytes);
  console.log(`${out} 저장 — ${bytes.length} bytes, ${res.headers.get('content-type')}, 찍은 시각 ${res.headers.get('x-frame-at')}`);
} finally {
  if (device) {
    await api('POST', `/devices/${device.device_id}/stop`).catch(() => {});
    await api('DELETE', `/devices/${device.device_id}`).catch(() => {});
  }
  for (let i = 0; i < 15 && !(await api('DELETE', `/stations/${station.station_id}`).then(() => true, () => false)); i++) await sleep(2000);
}