@civitai/blocks-react 0.71.0 → 0.72.2

package/dist/internal/mockHost.d.ts

@@ -37,7 +37,8 @@
  * called). Only the postMessage bridge round-trips are exercised. Never import
  * this from production code.
  */
-import { type BlockContext, type BlockResourceInfo, type BlockResourcePickerType, type BlockUploadedImageInfo, type BlockGenerationSourceImageInfo, type BuzzAccountType, type BlockBuzzTransaction, type BlockBuzzAccount, type BlockDailyCompensationResource, type BlockViewer, type BlockWildcardPack, type BlockWildcardPackErrorCode, type AppWorkflow, type BlockPublishedModel, type BlockGatedImage, type BlockCollectionFollowErrorCode, type BlockCreatePostHostError, type BlockCreatePostResult, type BlockPrepareTrainingDatasetHostError, type BlockRunTrainingHostError, type BlockTrainingRejectionReason, type ColorDomain, type SharedStorageValue, type Theme, type ViewerInfo, type WorkflowBody } from '@civitai/app-sdk/blocks';
+import { type BlockContext, type BlockResourceInfo, type BlockResourcePickerType, type BlockUploadedImageInfo, type BlockGenerationSourceImageInfo, type BuzzAccountType, type BlockBuzzTransaction, type BlockBuzzAccount, type BlockDailyCompensationResource, type BlockViewer, type BlockWildcardPack, type BlockWildcardPackErrorCode, type AppWorkflow, type BlockModel3dOutput, type BlockPublishedModel, type BlockRunConfirmedStepHostError, type BlockGatedImage, type BlockCollectionFollowErrorCode, type BlockCreatePostHostError, type BlockCreatePostResult, type BlockPrepareTrainingDatasetHostError, type BlockRunTrainingHostError, type BlockTrainingRejectionReason, type ColorDomain, type SharedStorageValue, type Theme, type ViewerInfo, type WorkflowBody } from '@civitai/app-sdk/blocks';
+import { type MockWorkflowQuoteFailure } from './mockWorkflowGraph.js';
 /**
  * How submits behave. `'none'` = everything succeeds; `'all'` / `'insufficient'`
  * = every submit fails as if the viewer were OUT OF BUZZ, until an
@@ -84,6 +85,7 @@ export type CostSpec = number | ((req: WorkflowBody, step?: MockWorkflowStepInfo
  * {@link MockGenerationScenario.costPerGen} once per step and sums the results.
  * `undefined` for every other kind, which is priced with one call.
  */
+export type { MockWorkflowQuoteFailure };
 export interface MockWorkflowStepInfo {
     /** Zero-based position in the body's `steps`. */
     index: number;
@@ -307,17 +309,27 @@ export interface MockGenerationScenario {
      */
     failWorkflowStep?: string;
     /**
-     * Simulate the host getting NO price quote for a `kind: 'workflow'` body. The
-     * host then refuses, before any reservation, a workflow with a training step
-     * (`errorCode: 'training-quote-unavailable'`) or a `repeat` step at any depth
-     * (`'workflow_repeat_unbounded'`); any other workflow proceeds reserved at the
-     * CEILING — {@link MockHostOptions.buzzBudget} × the number of steps — which
-     * the estimate reports as `cost.total`, with `authorFee: 0` (no quote, no
-     * fee base). Ignored by every other kind. Live-tunable via
+     * Simulate the host getting NO price quote for a `kind: 'workflow'` body, and
+     * say why: `'rejected'` (the orchestrator refused the body), `'unavailable'`
+     * (no usable answer — the one cause a retry can clear) or `'unpriced'` (an
+     * answer without a price). `true` means `'unavailable'`.
+     *
+     *  - A workflow with a training step (`errorCode: 'training-quote-<cause>'`)
+     *    or a `repeat` step at any depth (`'workflow_repeat_unbounded'`) is
+     *    refused on BOTH `estimate()` and `submit()`, before any reservation.
+     *  - Any other workflow: `estimate()` FAILS with `'workflow_quote_<cause>'`
+     *    and no `cost`, while `submit()` still PROCEEDS, reserved at the CEILING —
+     *    {@link MockHostOptions.buzzBudget} × the number of steps — with no author
+     *    fee (no quote, no fee base). A failed estimate does not predict a refused
+     *    submit.
+     *
+     * `error` is the host's sentence for the cause. The mock has no orchestrator
+     * reason to pass on, so `'rejected'` always reads the host's reason-less form.
+     * Ignored by every other kind. Live-tunable via
      * `setScenario({ generation: { workflowUnpriced } })`. Default: unset (every
      * workflow is quoted at its per-step `costPerGen`).
      */
-    workflowUnpriced?: boolean;
+    workflowUnpriced?: boolean | MockWorkflowQuoteFailure;
     /**
      * A single result image url (or `(body) => url`). For a `kind: 'workflow'`
      * run this (and {@link images}) is the LAST step that did not set
@@ -371,6 +383,22 @@ export interface MockGenerationScenario {
      * `appWorkflows`.
      */
     trainingPublishedModel?: BlockPublishedModel;
+    /**
+     * The 3D models a succeeded PASS-THROUGH step reports on
+     * {@link BlockWorkflowSnapshot.model3dOutputs} — what a `polyGen` job returns.
+     * As on the real host, every labelled `url` is ALSO in the snapshot's
+     * `imageUrls`: when set, the mock answers with exactly these urls there, in
+     * this order, unless {@link images} / {@link image} say otherwise (then the
+     * labels are sent as given and it is on you to keep the two consistent).
+     *
+     * Applies only to a pass-through `{ kind: 'step', $type }` body — the only arm
+     * the host labels. Default: unset, and the field is OMITTED, as the host omits
+     * it for a workflow that produced no 3D model.
+     *
+     * Snapshot-only, like {@link trainedEpochs}: for the app-queue read put
+     * `model3dOutputs` on the rows you pass as {@link MockHostOptions.appWorkflows}.
+     */
+    model3dOutputs?: BlockModel3dOutput[];
 }
 /**
  * The training bounds the mock enforces — COPIES of `@civitai/app-sdk/blocks`'s
@@ -405,6 +433,41 @@ export declare const MOCK_TRAINING_DATASET_GONE_ERROR = "training dataset not fo
  */
 export declare const MOCK_TRAINING_NONE_ELIGIBLE_ERROR = "none of the requested images can be used for training";
 export declare const MOCK_TRAINING_NONE_IMPORTED_ERROR = "none of the requested images could be prepared for training";
+/**
+ * The per-job ceiling of a confirmed step — a COPY of `@civitai/app-sdk/blocks`'s
+ * `BLOCK_CONFIRMED_STEP_MAX_BUZZ_PER_JOB`, not value-imported for the reason
+ * given on {@link MOCK_TRAINING_BOUNDS}. Pinned to the SDK's value by
+ * `test/confirmedStepFlow.test.tsx`, which imports both.
+ */
+export declare const MOCK_CONFIRMED_STEP_MAX_BUZZ_PER_JOB = 5000;
+/**
+ * The real server's refusal when the host reads back a step quote it does not
+ * hold — unknown, expired, or already used by an earlier run. This is the
+ * PREVIEW's message, which is the one `RUN_CONFIRMED_STEP` reaches first.
+ */
+export declare const MOCK_STEP_QUOTE_GONE_ERROR = "step quote not found or expired";
+/** The real server's refusal for a confirmed run whose body is not the quoted one. */
+export declare const MOCK_STEP_BODY_MISMATCH_ERROR = "the step body differs from the one that was quoted \u2014 estimate again";
+/**
+ * The real server's refusal for a `quoteId` body sent as a plain
+ * `SUBMIT_WORKFLOW`: nothing but `RUN_CONFIRMED_STEP`'s dialog records the
+ * viewer's confirmation.
+ */
+export declare const MOCK_STEP_NOT_CONFIRMED_ERROR = "this step has not been confirmed by the viewer in this session";
+/**
+ * The real server's refusal for a TRAINING `$type` (`training`,
+ * `imageResourceTraining`) on the confirmed-step path — at the estimate, where a
+ * quote would otherwise be granted, and at the confirmed submit. Training runs
+ * have their own kind.
+ */
+export declare const MOCK_STEP_TRAINING_TYPE_ERROR = "a training step type cannot run as a confirmed step \u2014 use kind:'training'";
+/**
+ * The fixed PREFIX of the real server's refusal for a confirmed-step body whose
+ * `input` is not plain JSON. The full message continues with the offending path
+ * and what was found there (`body.input.when is a Date`), which varies — match on
+ * this prefix.
+ */
+export declare const MOCK_STEP_NON_JSON_INPUT_ERROR_PREFIX = "a confirmed step needs plain JSON input (objects, arrays, strings, finite numbers, booleans, null) \u2014 ";
 /**
  * BUZZ scenario controls — simulate a viewer who runs out of Buzz. The mock host
  * treats `balance` as a spendable wallet: each succeeding generation DEBITS its
@@ -968,6 +1031,51 @@ export interface MockHostOptions {
      * `setScenario({ runTrainingCapRefusal: undefined })` clears it.
      */
     runTrainingCapRefusal?: string;
+    /**
+     * Force `RUN_CONFIRMED_STEP` to reply with an `error` instead of starting the
+     * job — a {@link BlockRunConfirmedStepHostError} or any other string for a
+     * server message. Absent → the viewer "confirms" and the job starts.
+     *
+     *  - `'declined'` — the viewer dismissed the confirm. No job.
+     *  - `'daily-limit-exceeded'` — the job would exceed the viewer's own daily
+     *    Buzz limit for the app. No job; the real host explains it in its chrome.
+     *  - `'submission-unconfirmed'` — the job MAY exist.
+     *
+     * A CONFIRMED STEP IS QUOTED BY THE ESTIMATE, with no knob of its own: a
+     * pass-through `{ kind: 'step', $type }` estimate carries
+     * {@link BlockWorkflowSnapshot.stepQuote} whenever the mock's price for it
+     * ({@link MockGenerationScenario.costPerGen}, or `cost`) is above the token's
+     * per-call budget ({@link MockHostOptions.buzzBudget}, default 200) and at most
+     * 5,000, for a signed-in viewer. So `generation: { costPerGen: 800 }` is all it
+     * takes to reach this path. ONLY a single `ESTIMATE_WORKFLOW` quotes one: a
+     * batch-estimate cell never carries `stepQuote`, as on the server. And where a
+     * quote would be granted, two bodies make the estimate FAIL instead, with the
+     * server's message: a training `$type` (`training`, `imageResourceTraining`,
+     * in any casing), and an `input` that is not plain JSON (a `Date`, `Map`,
+     * `Set`, `BigInt`, `NaN`, an empty slot in a sparse array, …).
+     *
+     * Whether the QUOTE survives follows the real flow: the host's own gate,
+     * dialog and daily-limit codes settle before the server claims the quote, so
+     * it stays runnable; `submission-unconfirmed` and any server message settle
+     * after the claim, so it is used up — running it again gets the quote-gone
+     * error. Live-tunable via {@link MockHost.setScenario}; `undefined` clears it.
+     * URL: `?confirmedStep=declined|daily-limit|unconfirmed`.
+     *
+     * 🔴 Same real gap as {@link runTrainingError}: the confirm is HOST chrome, so
+     * the mock settles immediately where the real host waits on a click. The mock
+     * also prices every estimate as FIXED — the real host gives a variable or
+     * metered price no `stepQuote` at all.
+     */
+    runConfirmedStepError?: BlockRunConfirmedStepHostError | string;
+    /**
+     * Make `RUN_CONFIRMED_STEP` resolve the server's spend-cap refusal instead of
+     * a job: a snapshot `{ workflowId: 'failed', status: 'failed', cost: { total:
+     * <quote> }, error: <this string> }` (refunded, no job).
+     * `useRunConfirmedStep()` rejects it with `.refused`. 🔴 The quote is CONSUMED,
+     * as on the server. Absent → the job starts. Live-tunable via
+     * {@link MockHost.setScenario}; `undefined` clears it.
+     */
+    runConfirmedStepCapRefusal?: string;
     /**
      * Force `SAVE_IMAGE` to reply with this `error` instead of saving — any of
      * the host's free-text refusals (`'image url is not allowed'`,
@@ -1156,6 +1264,9 @@ export interface MockHostOptions {
      * shape carries the field, so a context you pass WITHOUT a `theme` key still
      * gets the harness theme (and still follows a later `setTheme`). Only a slot
      * this SDK has no shape for is left alone — there is no `theme` field to set.
+     *
+     * The slot also decides `SET_USER_CHECKPOINT`, as in production: refused on
+     * `app.page` (the default), persisted on a model slot.
      */
     context?: BlockContext;
     /**
@@ -1230,7 +1341,7 @@ export interface MockHostOptions {
  * {@link MockHost.setTheme}, which pushes a real `THEME_CHANGE` — the mock of
  * the viewer toggling dark mode with the block already mounted.)
  */
-export type MockHostScenarioPatch = Pick<MockHostOptions, 'consentGrantable' | 'failMode' | 'cost' | 'pollsUntilDone' | 'cannedPicks' | 'cannedMultiPicks' | 'cannedImageUpload' | 'cannedGenerationSourceUpload' | 'cannedImageScan' | 'generation' | 'buzz' | 'storage' | 'shared' | 'buzzBalanceError' | 'viewerResult' | 'viewerError' | 'buzzTransactions' | 'buzzAccounts' | 'dailyCompensation' | 'buzzReadError' | 'wildcardPack' | 'wildcardPackError' | 'collectionFollowError' | 'createPostResult' | 'createPostError' | 'trainingDatasetRejected' | 'trainingDatasetError' | 'trainingQuoteTotal' | 'runTrainingError' | 'runTrainingCapRefusal' | 'appWorkflows' | 'appWorkflowsError' | 'publishImageIds' | 'publishError' | 'gatedImages' | 'gatedImagesError' | 'disallowedAccountTypes' | 'saveImageError' | 'uploadImageBytesResult' | 'uploadImageBytesError'>;
+export type MockHostScenarioPatch = Pick<MockHostOptions, 'consentGrantable' | 'failMode' | 'cost' | 'pollsUntilDone' | 'cannedPicks' | 'cannedMultiPicks' | 'cannedImageUpload' | 'cannedGenerationSourceUpload' | 'cannedImageScan' | 'generation' | 'buzz' | 'storage' | 'shared' | 'buzzBalanceError' | 'viewerResult' | 'viewerError' | 'buzzTransactions' | 'buzzAccounts' | 'dailyCompensation' | 'buzzReadError' | 'wildcardPack' | 'wildcardPackError' | 'collectionFollowError' | 'createPostResult' | 'createPostError' | 'trainingDatasetRejected' | 'trainingDatasetError' | 'trainingQuoteTotal' | 'runTrainingError' | 'runTrainingCapRefusal' | 'runConfirmedStepError' | 'runConfirmedStepCapRefusal' | 'appWorkflows' | 'appWorkflowsError' | 'publishImageIds' | 'publishError' | 'gatedImages' | 'gatedImagesError' | 'disallowedAccountTypes' | 'saveImageError' | 'uploadImageBytesResult' | 'uploadImageBytesError'>;
 /** Runtime Buzz-balance handle exposed on {@link MockHost.buzz}. */
 export interface MockBuzzHandle {
     /** Current simulated balance, or `undefined` when balance isn't simulated. */
@@ -1279,6 +1390,8 @@ export declare function disallowedAccountError(accountType: BuzzAccountType): st
  * map onto the new scenario groups so a dev can flip out-of-Buzz /
  * failures / latency without editing code. `?capRefusal=1` (or `=<text>`) sets
  * {@link MockGenerationScenario.submitCapRefusal}.
+ * `?confirmedStep=declined|daily-limit|unconfirmed` sets
+ * {@link MockHostOptions.runConfirmedStepError}.
  *
  * Returns a partial overlay applied ON TOP of explicit {@link MockHostOptions}
  * (URL wins — it's the interactive dev knob). No-op outside a browser.