클라우드 리눅스에서 파이썬 돌리기

#!/usr/bin/env node
// 클라우드 리눅스에서 파이썬 돌리기 — endue OS 샘플.
//   Station 을 만들고 → linux 가상 기기(virtual Device)를 띄우고 → running 을 기다려 → 스크립트를 써서 실행하고
//   → 오래 걸리는 명령은 202 + 롱폴로 받고 → 정리한다.
//
//   node linux-python-runner.mjs              로컬 스택 (가짜 IdP 로 토큰을 받는다)
//   ENDUE_OS_TOKEN=… ENDUE_OS_URL=… node …    운영 — endue 계정의 토큰
//   KEEP=1 node …                             끝나도 기기·Station 을 남긴다
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));

async function getToken() {
  if (process.env.ENDUE_OS_TOKEN) return process.env.ENDUE_OS_TOKEN;
  // 로컬 스택 전용 — 가짜 IdP 가 아무 이름으로 토큰을 준다.
  const res = await fetch(`${BASE}/idp/token?sub=${process.env.ENDUE_OS_SUB ?? `sample-${Date.now().toString(36)}`}&ttl=3600`, { method: 'POST' });
  return (await res.json()).access_token;
}

const token = await getToken();

/** 공개 API 한 번. 성공이면 { status, data, headers }, 아니면 오류 코드를 담아 던진다. */
async function api(method, path, body) {
  const res = await fetch(`${API}${path}`, {
    method,
    headers: { Authorization: `Bearer ${token}`, ...(body ? { 'Content-Type': 'application/json' } : {}) },
    body: body ? JSON.stringify(body) : undefined,
  });
  const json = res.status === 204 ? null : await res.json();
  if (res.status === 402) {
    // 비용이 생기는 호출(가상 기기 만들기·시작, 명령)은 계정 잔액이 0 이하면 막힌다.
    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}: ${json?.error?.message}`);
  return { status: res.status, data: json?.data, headers: res.headers };
}

/** 명령 하나. 25초 안에 끝나면 201 로 결과가 오고, 아니면 202 — 그때는 롱폴로 끝을 기다린다. */
async function run(deviceId, command, input, opts = {}) {
  const first = await api('POST', `/devices/${deviceId}/commands`, { command, input, ...opts });
  let cmd = first.data;
  if (first.status === 202) {
    console.log(`  … 202 Accepted (${cmd.command_id}) — 롱폴로 기다린다`);
    while (!['succeeded', 'failed', 'timed_out', 'cancelled', 'expired', 'unknown'].includes(cmd.status)) {
      cmd = (await api('GET', `/commands/${cmd.command_id}?wait_ms=25000`)).data;
    }
    if (cmd.status === 'succeeded') cmd.output = (await api('GET', `/commands/${cmd.command_id}/output`)).data.output;
  }
  if (cmd.status !== 'succeeded') throw new Error(`${command}: ${cmd.status} ${cmd.error_code ?? ''} ${cmd.error ?? ''}`);
  return cmd.output;
}

const station = (await api('POST', '/stations', { name: 'python-lab' })).data;
console.log(`Station ${station.station_id}`);
let device;
try {
  // 가상 기기 — VM·Storage·에이전트가 함께 생긴다. 사용자 JWT 로만 (API 키면 403).
  device = (await api('POST', `/stations/${station.station_id}/devices`, { type: 'virtual', kind: 'linux', name: 'py-box', size: 'small' })).data.device;
  console.log(`Device ${device.device_id} — ${device.compute_status}`);

  // 에이전트가 붙으면 running. 단건 조회는 전이 중인 가상 기기를 공급자에게 한 번 확인해 준다.
  const until = Date.now() + 180_000;
  while (!(device.compute_status === 'running' && device.online)) {
    if (device.compute_status === 'failed') throw new Error(`Device failed: ${device.status_reason}`);
    if (Date.now() > until) throw new Error('3분 안에 running 이 되지 않았다');
    await sleep(2000);
    device = (await api('GET', `/devices/${device.device_id}`)).data;
  }
  const rid = device.device_id;
  console.log(`running — device ${rid}`);

  const script = ['import platform, statistics', 'xs = [3, 1, 4, 1, 5, 9, 2, 6]', 'print(platform.machine(), statistics.mean(xs), sorted(xs))'].join('\n');
  await run(rid, 'write_file', { path: 'stats.py', content: script });
  const out = await run(rid, 'run_program', { program: 'python3', args: ['stats.py'] });
  console.log(`python3 stats.py → ${String(out.stdout).trim()}`);

  // wait_ms: 0 이면 바로 202 — 오래 걸리는 작업을 보내 두고 나중에 받는 모양.
  const listing = await run(rid, 'list_dir', { path: '.' }, { wait_ms: 0 });
  console.log(`작업 폴더: ${(listing.entries ?? []).map((e) => e.name).join(', ')}`);
} finally {
  if (process.env.KEEP) {
    console.log('KEEP=1 — 남겨 둔다');
  } else {
    if (device) {
      await api('POST', `/devices/${device.device_id}/stop`).catch(() => {});
      await api('DELETE', `/devices/${device.device_id}`).catch((e) => console.warn(String(e)));
    }
    for (let i = 0; i < 15; i++) {
      const ok = await api('DELETE', `/stations/${station.station_id}`).then(() => true, () => false);
      if (ok) break;
      await sleep(2000);
    }
    console.log('정리했다');
  }
}