pixelkiln 0.74.0 → 0.76.0

package/dist/cli.js

@@ -6139,6 +6139,25 @@ function cleanupResults(images, sent, operation) {
   }
   return images.map((image) => Buffer.from(image.base64, "base64"));
 }
+function parseProFlashQuote(raw) {
+  const body = raw && typeof raw === "object" ? raw : {};
+  const nested = [body, body.usage, body.cost, body.estimate].filter((v) => Boolean(v) && typeof v === "object");
+  const pick = (...names) => {
+    for (const obj of nested) {
+      for (const name of names) {
+        const value = obj[name];
+        if (typeof value === "number" && Number.isFinite(value) && value >= 0) return value;
+      }
+    }
+    return null;
+  };
+  const image = pick("image", "image_generations", "first_image", "first_image_generations");
+  const rotations = pick("rotations", "rotation", "rotations_generations", "rotation_generations");
+  const total = pick("total", "total_generations", "generations");
+  if (total !== null) return { image, rotations, total };
+  if (image !== null || rotations !== null) return { image, rotations, total: (image ?? 0) + (rotations ?? 0) };
+  throw new Error(`GET /pro-flash/cost returned no price: ${JSON.stringify(raw).slice(0, 200)}`);
+}
 var PixelLabError = class extends ProviderError {
   constructor(message6, status, body) {
     super("pixellab", message6, { status });
@@ -6197,6 +6216,19 @@ var PixelLabClient = class {
       plan: raw.subscription?.plan ?? "unknown"
     };
   }
+  /**
+   * `GET /pro-flash/cost`: PixelLab's own quote for one Pro Flash job, split
+   * into the first image and the v3 rotations. Free, and provisional by
+   * PixelLab's own description; the billed amount on the finished job is
+   * the real one. The response schema is undocumented (`{}` in the spec),
+   * so this reads it leniently: a total if one is given, else the sum of
+   * the parts, and refuses an answer with no number in it.
+   */
+  async proFlashCost(args) {
+    const query = new URLSearchParams({ operation: args.operation, width: String(args.width), height: String(args.height) });
+    if (args.nDirections !== void 0) query.set("n_directions", String(args.nDirections));
+    return parseProFlashQuote(await this.request(`/pro-flash/cost?${query}`));
+  }
   /**
    * Square objects that persist indefinitely.
    *
@@ -7333,6 +7365,65 @@ function proFlashImageCost(width, height) {
   const side = Math.max(width, height);
   return side <= 96 ? 5 : side <= 208 ? 6 : 9;
 }
+function proFlashQuoteQuery(spec) {
+  if (spec.provider !== "pixellab" || spec.mirror) return null;
+  if (spec.revision) {
+    if (spec.revision.engine !== "pro-flash") return null;
+    return {
+      operation: spec.revision.mode === "inpaint" ? "inpaint" : "edit",
+      width: spec.revision.sourceWidth ?? spec.width,
+      height: spec.revision.sourceHeight ?? spec.height,
+      part: "total"
+    };
+  }
+  if (spec.generator === "imageProFlash") return { operation: "create", width: spec.width, height: spec.height, part: "total" };
+  const character = spec.character;
+  if (spec.generator === "character" && character?.kind === "base" && character.mode === "pro-flash") {
+    const south = character.reference?.south;
+    return south ? { operation: "character", width: south.width, height: south.height, nDirections: 8, part: "rotations" } : { operation: "character", width: spec.width, height: spec.height, nDirections: 8, part: "total" };
+  }
+  const object = spec.objectPro;
+  if (spec.generator === "objectPro" && object?.kind === "base") {
+    const south = object.reference;
+    return south ? { operation: "object", width: south.width, height: south.height, nDirections: object.directions, part: "rotations" } : { operation: "object", width: spec.width, height: spec.height, nDirections: object.directions, part: "total" };
+  }
+  return null;
+}
+function createProFlashQuoter(client2, opts = {}) {
+  const ttl = opts.ttlMs ?? 10 * 6e4;
+  const timeoutMs = opts.timeoutMs ?? 5e3;
+  const now = opts.now ?? Date.now;
+  const cache = /* @__PURE__ */ new Map();
+  return async (spec) => {
+    const query = proFlashQuoteQuery(spec);
+    if (!query) return null;
+    const key = `${query.operation}:${query.width}x${query.height}:${query.nDirections ?? ""}`;
+    let hit = cache.get(key);
+    if (!hit || now() - hit.at > ttl) {
+      let timer;
+      const answer = Promise.race([
+        client2.proFlashCost({
+          operation: query.operation,
+          width: query.width,
+          height: query.height,
+          ...query.nDirections !== void 0 ? { nDirections: query.nDirections } : {}
+        }),
+        new Promise((_, reject) => {
+          timer = setTimeout(() => reject(new Error(`PixelLab's Pro Flash quote did not answer within ${timeoutMs / 1e3}s`)), timeoutMs);
+        })
+      ]).finally(() => clearTimeout(timer));
+      hit = { at: now(), answer };
+      cache.set(key, hit);
+      answer.catch(() => {
+        if (cache.get(key) === hit) cache.delete(key);
+      });
+    }
+    const quote = await hit.answer;
+    if (query.part === "total") return quote.total;
+    if (quote.rotations === null) throw new Error("PixelLab's Pro Flash quote gave no rotations figure for a base drawn from a sprite");
+    return quote.rotations;
+  };
+}
 function characterAnimationName(spec) {
   return `pixelkiln:${spec.styleId}/${spec.assetId}`;
 }
@@ -14208,31 +14299,43 @@ async function applyManifestEdit(manifestPath, edit) {
     (absolute, original) => ({ manifestPath: absolute, sha256: sha256(original), changed: false })
   );
 }
-async function priceManifestEdit(manifestPath, edit) {
+async function priceManifestEdit(manifestPath, edit, opts = {}) {
   return withCandidate(
     manifestPath,
     edit,
     async ({ absolute, specs }) => {
       const before = new Map((await resolveSpecs(await loadManifest(absolute))).map((spec) => [`${spec.styleId}/${spec.assetId}`, spec.specHash]));
+      const priced = specs.filter((spec) => before.get(`${spec.styleId}/${spec.assetId}`) !== spec.specHash);
+      let quoteError;
+      const live = await Promise.all(priced.map(async (spec) => {
+        if (!opts.quote) return null;
+        try {
+          return await opts.quote(spec);
+        } catch (error) {
+          quoteError ??= error instanceof Error ? error.message : String(error);
+          return null;
+        }
+      }));
       const items = [];
       const totals = {};
-      for (const spec of specs) {
+      priced.forEach((spec, index) => {
         const key = `${spec.styleId}/${spec.assetId}`;
-        const was = before.get(key);
-        if (was === spec.specHash) continue;
+        const quoted = live[index];
+        const cost = quoted ?? spec.cost;
         items.push({
           key,
           styleId: spec.styleId,
           assetId: spec.assetId,
-          change: was === void 0 ? "new" : "changed",
+          change: before.has(key) ? "changed" : "new",
           provider: spec.provider,
-          cost: spec.cost,
+          cost,
           costUnit: spec.costUnit,
-          candidates: spec.candidates
+          candidates: spec.candidates,
+          ...quoted !== null && quoted !== void 0 ? { quote: "live", estimate: spec.cost } : {}
         });
-        totals[spec.costUnit] = (totals[spec.costUnit] ?? 0) + spec.cost;
-      }
-      return { items, totals };
+        totals[spec.costUnit] = (totals[spec.costUnit] ?? 0) + cost;
+      });
+      return quoteError ? { items, totals, quoteError } : { items, totals };
     },
     () => ({ items: [], totals: {} })
   );
@@ -14895,7 +14998,15 @@ function createGalleryPriceHandler(opts) {
     } catch (error) {
       throw new ManifestEditError(error instanceof Error ? error.message : String(error));
     }
-    return priceManifestEdit(manifestPath, parsed.data);
+    let quote = null;
+    let quoteError;
+    try {
+      quote = opts.quoteFor ? await opts.quoteFor(parsed.data.project) : null;
+    } catch (error) {
+      quoteError = error instanceof Error ? error.message : String(error);
+    }
+    const price = await priceManifestEdit(manifestPath, parsed.data, quote ? { quote } : {});
+    return quoteError && !price.quoteError ? { ...price, quoteError } : price;
   };
 }
 
@@ -17879,7 +17990,10 @@ async function serveUntilStopped(initial, reload, args, access2) {
   };
   const loadProject = async (project) => {
     const ctx = await access2.loadProject(project);
-    const dir = path34.dirname(ctx.loaded.path);
+    loadProjectEnv(path34.dirname(ctx.loaded.path));
+    return ctx;
+  };
+  const loadProjectEnv = (dir) => {
     const declared = readEnvFiles(dir);
     const credentialNames = new Set(availableProviders().flatMap((id) => providerCredentialEnvs(providerFactory(id))));
     for (const [name, value] of Object.entries(declared)) {
@@ -17890,7 +18004,18 @@ async function serveUntilStopped(initial, reload, args, access2) {
       }
     }
     loadEnvFiles(dir);
-    return ctx;
+  };
+  const quoters = /* @__PURE__ */ new Map();
+  const quoteFor = async (project) => {
+    loadProjectEnv(path34.dirname(path34.resolve(await access2.manifestFor(project))));
+    const key = process.env.PIXELLAB_API_KEY;
+    if (!key) return null;
+    let quoter = quoters.get(key);
+    if (!quoter) {
+      quoter = createProFlashQuoter(new PixelLabClient(key));
+      quoters.set(key, quoter);
+    }
+    return quoter;
   };
   const server = await serveGallery({
     load: () => {
@@ -17913,7 +18038,7 @@ async function serveUntilStopped(initial, reload, args, access2) {
     ),
     ...args.edit ? { edit: createGalleryEditHandler({ manifestFor: access2.manifestFor, loadProject: access2.loadProject, reload, onProgress: log }) } : {},
     ...budget ? { generate: createGenerateHandlers({ loadProject, providerFor, budget, reload, onProgress: log }) } : {},
-    ...args.edit ? { price: createGalleryPriceHandler({ manifestFor: access2.manifestFor }) } : {},
+    ...args.edit ? { price: createGalleryPriceHandler({ manifestFor: access2.manifestFor, quoteFor }) } : {},
     // Estimating acts on the author's behalf and feeds a manifest edit, so it follows the write gate.
     ...args.edit ? { skeleton: createGallerySkeletonHandlers({ loadProject, limit: args.estimateLimit, onProgress: log }) } : {},
     // The editor exists to write hand edits back, so it follows the write gate.