#Python 与 Node.js 示例

import { createWriteStream } from 'node:fs';
import { writeFile } from 'node:fs/promises';
import { Readable } from 'node:stream';
import { pipeline } from 'node:stream/promises';

const base = (process.env.THQ_VIDEO_BASE_URL || 'https://new.thqllm.com').replace(/\/+$/, '');
const key = process.env.THQ_VIDEO_API_KEY;
if (!key) throw new Error('请设置 THQ_VIDEO_API_KEY');
const headers = { Authorization: `Bearer ${key}` };
const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
const transient = new Set([429, 500, 502, 503, 504, 522]);
let id = process.env.THQ_VIDEO_TASK_ID;

if (!id) {
  const {
    THQ_VIDEO_MODEL: model,
    THQ_VIDEO_SECONDS: seconds,
    THQ_VIDEO_SIZE: size,
    THQ_VIDEO_RESOLUTION: resolution,
  } = process.env;
  if (!model || !seconds || !size || !resolution) {
    throw new Error('请设置模型、时长、画面比例和清晰度环境变量');
  }
  // 创建请求只发一次，不为超时或 5xx 配置自动重试。
  const response = await fetch(`${base}/v1/videos`, {
    method: 'POST',
    headers: { ...headers, 'Content-Type': 'application/json' },
    body: JSON.stringify({ model, prompt: '雨后的街道，镜头缓慢前进', seconds, size, resolution }),
    signal: AbortSignal.timeout(180_000),
  });
  if (!response.ok) throw new Error(`创建请求 HTTP ${response.status}: ${await response.text()}`);
  id = (await response.json()).id;
  if (typeof id !== 'string' || !id) throw new Error('响应缺少有效 id，请核对控制台任务记录');
}

console.log('已保存任务 ID:', id);
await writeFile('thq-video-task-id.txt', id, 'utf8');
const deadline = Date.now() + 60 * 60 * 1000;
let resultUrl;

while (Date.now() < deadline) {
  await sleep(5000);
  let response;
  try {
    response = await fetch(`${base}/v1/videos/${encodeURIComponent(id)}`, {
      headers,
      signal: AbortSignal.timeout(60_000),
    });
  } catch {
    continue; // 网络异常只重试查询。
  }
  if (transient.has(response.status)) {
    const value = response.headers.get('Retry-After');
    await response.body?.cancel();
    const delay =
      value && /^\d+$/.test(value)
        ? Number(value) * 1000
        : value
          ? Date.parse(value) - Date.now()
          : 10_000;
    await sleep(
      Math.min(
        Math.max(5000, Number.isFinite(delay) ? delay : 10_000),
        Math.max(0, deadline - Date.now()),
      ),
    );
    continue;
  }
  if (!response.ok) throw new Error(`查询 HTTP ${response.status}: ${await response.text()}`);
  const state = await response.json();
  const status = String(state.status || '').toLowerCase();
  console.log('任务状态:', status, '进度:', state.progress);
  if (['failed', 'failure', 'cancelled', 'canceled'].includes(status)) {
    throw new Error(JSON.stringify(state.error || state.fail_reason || state));
  }
  if (
    ['completed', 'success', 'succeeded'].includes(status) &&
    typeof state.url === 'string' &&
    state.url
  ) {
    resultUrl = state.url;
    break;
  }
}
if (!resultUrl) throw new Error(`本地等待已到期，请继续查询原任务: ${id}`);
const parsedUrl = new URL(resultUrl);
if (parsedUrl.protocol !== 'https:') throw new Error('结果 URL 必须使用 HTTPS');

// 直接使用返回的 URL；不附带 API Key，不调用 /content。
const media = await fetch(resultUrl, { signal: AbortSignal.timeout(180_000) });
if (!media.ok) throw new Error(`下载 HTTP ${media.status}`);
const contentType = (media.headers.get('content-type') || '').split(';')[0].trim().toLowerCase();
if (
  (!contentType.startsWith('video/') && contentType !== 'application/octet-stream') ||
  !media.body
) {
  throw new Error(`结果未返回视频文件，Content-Type: ${contentType}`);
}
await pipeline(Readable.fromWeb(media.body), createWriteStream('output.mp4'));
console.log('视频已保存为 output.mp4');