接口参考

import { readFile } from 'node:fs/promises';
import { TUTUJIN } from './config.js';

/**
 * @param {string} prompt  改图指令
 * @param {Array<{buffer: Buffer, mime: string, filename: string}>} images
 *        参考图。第 0 张是底图（要被改的那张），顺序有语义，不要打乱。
 * @param {string} size    必须是像素串；"跟随原图" 的算法见本页尺寸一节
 */
export async function imageToImage({ prompt, images, size, quality = 'high', mask }) {
  if (!images?.length) throw new Error('图生图至少需要一张参考图');

  const form = new FormData();
  form.append('model', TUTUJIN.model);
  form.append('prompt', prompt);
  form.append('size', size);          // 恒传，省略会塌成 1024x1024
  form.append('quality', quality);
  form.append('response_format', 'url');

  // 多图字段名先按 image[] 发；若实测中转层不透传数组，改成单个 'image' 只发 images[0]
  for (const img of images) {
    form.append('image[]', new Blob([img.buffer], { type: img.mime }), img.filename);
  }
  // 单图写法（fallback）：
  // form.append('image', new Blob([images[0].buffer], { type: images[0].mime }), images[0].filename);

  if (mask) {
    form.append('mask', new Blob([mask.buffer], { type: mask.mime }), mask.filename);
  }

  const resp = await fetch(`${TUTUJIN.baseUrl}/images/edits`, {
    method: 'POST',
    headers: { Authorization: `Bearer ${TUTUJIN.apiKey}` },  // ← 只放这一个，别手写 Content-Type
    body: form,
    signal: AbortSignal.timeout(TUTUJIN.timeoutMs),
  });

  if (!resp.ok) {
    const text = await resp.text();
    throw new Error(`图像接口返回 ${resp.status}: ${text.slice(0, 500)}`);
  }
  return await extractImage(await resp.json());
}

// 从本地文件构造参考图：后缀 → MIME，并在本地先把不合规的挡掉
export async function loadImage(path) {
  const ext = path.toLowerCase().split('.').pop();
  const mime = { png: 'image/png', jpg: 'image/jpeg', jpeg: 'image/jpeg', webp: 'image/webp' }[ext];
  if (!mime) throw new Error(`不支持的图片格式: ${ext}`);
  const buffer = await readFile(path);
  if (buffer.length > 25 * 1024 * 1024) throw new Error('参考图超过 25MB');
  return { buffer, mime, filename: path.split('/').pop() };
}

const MAX_EDGE = 3840, MIN_PX = 655_360, MAX_PX = 8_294_400, MAX_RATIO = 3;
const round16 = (n) => Math.max(16, Math.round(n / 16) * 16);

/**
 * @param {number} w @param {number} h
 * @param {number} [maxEdge] 想限制在 ~1.5MP 常规档就传 1536
 * @returns {string} "宽x高"
 */
export function normalizeSize(w, h, maxEdge = MAX_EDGE) {
  // ① 比例钳到 3:1 以内
  if (w / h > MAX_RATIO) h = w / MAX_RATIO;
  else if (h / w > MAX_RATIO) w = h / MAX_RATIO;
  // ② 最长边
  const longest = Math.max(w, h);
  if (longest > maxEdge) { const s = maxEdge / longest; w *= s; h *= s; }
  // ③ 总像素上限
  let px = w * h;
  if (px > MAX_PX) { const s = Math.sqrt(MAX_PX / px); w *= s; h *= s; }
  // ④ 总像素下限（放大后若又超边长，边长优先）
  px = w * h;
  if (px < MIN_PX) {
    const s = Math.sqrt(MIN_PX / px); w *= s; h *= s;
    const l2 = Math.max(w, h);
    if (l2 > maxEdge) { const s2 = maxEdge / l2; w *= s2; h *= s2; }
  }
  // ⑤ 16 的倍数（最后做）
  return `${round16(w)}x${round16(h)}`;
}

import sharp from 'sharp';   // npm i sharp

export async function sizeFollowingSource(buffer) {
  try {
    const meta = await sharp(buffer, { failOn: 'none' }).metadata();
    let w = meta.width, h = meta.height;
    if (!w || !h) return '1024x1024';
    // ★ EXIF 方向 5-8 表示图被旋转了 90°，metadata 给的是"存储"宽高、和肉眼看到的相反
    //   （手机竖拍 JPEG 常见 orientation=6）。不换轴会把竖拍照片当成横图发出去。
    if (meta.orientation >= 5 && meta.orientation <= 8) [w, h] = [h, w];
    // 常规档：长边压到 1536（约 1.5MP）
    return normalizeSize(w, h, 1536);
  } catch {
    return '1024x1024';   // 读不出尺寸（损坏图 / HEIC 等）→ 安全回落，绝不因此中断
  }
}

export async function extractImage(json) {
  const d = json?.data?.[0];
  if (!d) throw new Error('上游响应里没有 data');

  if (typeof d.b64_json === 'string' && d.b64_json) {
    return Buffer.from(d.b64_json, 'base64');
  }
  if (typeof d.url === 'string' && /^https?:\/\//.test(d.url)) {
    // ★ 这个 URL 是临时的（存活以小时计，域名也不固定）。
    //   必须在拿到的当下就下载下来，存进你自己的存储。
    const r = await fetch(d.url, { signal: AbortSignal.timeout(60_000) });
    if (!r.ok) throw new Error(`下载生成图失败: ${r.status}`);
    return Buffer.from(await r.arrayBuffer());
  }
  throw new Error('data[0] 里既没有 url 也没有 b64_json');
}