Tài liệu API Pilio

const API_BASE = "https://pilio.ai";
const apiKey = process.env.PILIO_API_KEY!;

async function pilioFetch(path: string, init: RequestInit = {}) {
  const response = await fetch(API_BASE + path, {
    ...init,
    headers: {
      Authorization: `Bearer ${apiKey}`,
      "Content-Type": "application/json",
      ...init.headers,
    },
  });

  const payload = await response.json();
  if (!response.ok || payload.code !== 200) {
    throw new Error(payload.message || `Pilio API failed: ${payload.code}`);
  }
  return payload.data;
}

async function removeWatermark(file: File) {
  const created = await pilioFetch("/v1/files/batch-create", {
    method: "POST",
    body: JSON.stringify({
      files: [{
        nanoid: crypto.randomUUID().replaceAll("-", "").slice(0, 21),
        name: file.name,
        type: file.name.split(".").pop()?.toLowerCase(),
        size: file.size,
      }],
      upload_mode: "simple",
      storage_intent: "temporary",
    }),
  });

  const source = created.items[0];

  if (source.upload_url) {
    // 预签名地址不需要 Pilio API Key
    await fetch(source.upload_url, {
      method: "PUT",
      headers: { "Content-Type": file.type || "application/octet-stream" },
      body: file,
    });
  } else {
    throw new Error("This sample only covers single upload_url uploads. Use upload_urls + complete for multipart files.");
  }

  const task = await pilioFetch("/v1/images/remove-watermark", {
    method: "POST",
    body: JSON.stringify({ image_file_id: source.id, mode: "auto" }),
  });

  while (true) {
    await new Promise((resolve) => setTimeout(resolve, 2000));
    const status = await pilioFetch(task.status_url);
    if (status.status === "Failed") throw new Error(status.error_message || "Task failed");
    if (status.status === "Succeeded") break;
  }

  const result = await pilioFetch(task.result_url);
  return result.files?.[0]?.download_url || result.zip_file?.download_url;
}