@basementstudio/sanity-ai-image-plugin 0.0.1 → 0.1.1

package/dist/studio/components/asset-source.js

@@ -0,0 +1,169 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { useSecrets } from "@sanity/studio-secrets";
+import { Box, Button, Card, Flex, Select, Stack, Text, TextArea, } from "@sanity/ui";
+import { useEffect, useMemo, useRef, useState } from "react";
+import { useClient, } from "sanity";
+import { getDefaultAllowedAiImageModel, getSupportedAiImageModelOptions, } from "../../utils/models";
+import { MAX_REFERENCE_IMAGES, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE, SOURCE_NAME, SOURCE_TITLE, } from "../../utils/shared";
+import { getMissingSharedSecretMessage, getSharedSecretValue, SHARED_SECRET_HEADER_NAME, SHARED_SECRET_NAMESPACE, } from "../shared-secret";
+import { composePrompt } from "../../utils/prompts";
+import { buildSettingsReferenceImages, fetchSettingsDocument, findTargetSettings, getErrorMessage, resolveSettingsModel, } from "../settings-data";
+import { formatRateLimitRemainingTime, useFrontendRateLimit, } from "./frontend-rate-limit";
+import { convertImageFileToPng, createGeneratedFile, formatFileSize, } from "../files";
+function getGenerationPhaseLabel(phase) {
+    switch (phase) {
+        case "preparing":
+            return "Preparing...";
+        case "generating":
+            return "Generating...";
+        case "uploading":
+            return "Uploading to Sanity...";
+        case "updating":
+            return "Updating field...";
+        default:
+            return "Generate with AI Image Plugin";
+    }
+}
+export function createAssetSource(options) {
+    const assetSourceTarget = options.assetSourceTarget;
+    function AssetSourceComponent({ onClose, onSelect, }) {
+        const client = useClient({ apiVersion: options.apiVersion });
+        const { loading: isLoadingSharedSecret, secrets } = useSecrets(SHARED_SECRET_NAMESPACE);
+        const [prompt, setPrompt] = useState("");
+        const [selectedModel, setSelectedModel] = useState(getDefaultAllowedAiImageModel(options.allowedModels));
+        const [referenceImages, setReferenceImages] = useState([]);
+        const [error, setError] = useState(null);
+        const [generationPhase, setGenerationPhase] = useState("idle");
+        const isGenerating = generationPhase !== "idle";
+        const sharedSecret = getSharedSecretValue(secrets);
+        const hasCustomSelectedModelRef = useRef(false);
+        const frontendRateLimit = useFrontendRateLimit(options.frontendRateLimit);
+        const allowedModelOptions = useMemo(() => getSupportedAiImageModelOptions(options.allowedModels), [options.allowedModels]);
+        useEffect(() => {
+            let isMounted = true;
+            setSelectedModel(getDefaultAllowedAiImageModel(options.allowedModels));
+            hasCustomSelectedModelRef.current = false;
+            async function loadDefaultModel() {
+                try {
+                    const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
+                    if (!isMounted || hasCustomSelectedModelRef.current) {
+                        return;
+                    }
+                    setSelectedModel(resolveSettingsModel({
+                        allowedModels: options.allowedModels,
+                        settings,
+                    }));
+                }
+                catch {
+                    // Keep the first allowed model selected if settings loading fails.
+                }
+            }
+            void loadDefaultModel();
+            return () => {
+                isMounted = false;
+            };
+        }, [client, options.allowedModels]);
+        function handleReferenceImageChange(event) {
+            const nextFiles = Array.from(event.currentTarget.files ?? []);
+            if (nextFiles.length === 0) {
+                return;
+            }
+            setReferenceImages((currentFiles) => [...currentFiles, ...nextFiles].slice(0, MAX_REFERENCE_IMAGES));
+            event.currentTarget.value = "";
+        }
+        function handleRemoveReferenceImage(indexToRemove) {
+            setReferenceImages((currentFiles) => currentFiles.filter((_, index) => index !== indexToRemove));
+        }
+        async function handleGenerate() {
+            setError(null);
+            setGenerationPhase("preparing");
+            try {
+                if (isLoadingSharedSecret) {
+                    throw new Error("Loading AI Image Plugin shared secret...");
+                }
+                if (!sharedSecret) {
+                    throw new Error(getMissingSharedSecretMessage());
+                }
+                const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
+                const targetSettings = assetSourceTarget
+                    ? findTargetSettings(settings, assetSourceTarget.id)
+                    : null;
+                const composedPrompt = composePrompt([
+                    settings?.globalPrompt
+                        ? `Global direction:\n${settings.globalPrompt}`
+                        : null,
+                    targetSettings?.prompt
+                        ? `Asset source direction:\n${targetSettings.prompt}`
+                        : null,
+                    prompt.trim() ? `Editor prompt:\n${prompt.trim()}` : null,
+                ]);
+                if (!composedPrompt) {
+                    throw new Error("Add a prompt or configure a default prompt first.");
+                }
+                const formData = new FormData();
+                const normalizedReferenceImages = await Promise.all(referenceImages.map(convertImageFileToPng));
+                const settingsReferenceImages = await buildSettingsReferenceImages(settings, assetSourceTarget?.id);
+                const allReferenceImages = [
+                    ...settingsReferenceImages,
+                    ...normalizedReferenceImages,
+                ].slice(0, MAX_REFERENCE_IMAGES);
+                formData.set("prompt", composedPrompt);
+                formData.set("model", selectedModel);
+                for (const referenceImage of allReferenceImages) {
+                    formData.append("references", referenceImage);
+                }
+                const rateLimitSnapshot = frontendRateLimit.consumeAttempt();
+                if (!rateLimitSnapshot.allowed) {
+                    return;
+                }
+                setGenerationPhase("generating");
+                const response = await fetch(options.apiEndpoint, {
+                    method: "POST",
+                    body: formData,
+                    headers: {
+                        [SHARED_SECRET_HEADER_NAME]: sharedSecret,
+                    },
+                });
+                const payload = (await response.json());
+                if (!response.ok) {
+                    throw new Error("error" in payload && payload.error
+                        ? payload.error
+                        : "AI image generation failed.");
+                }
+                if (!("data" in payload) ||
+                    typeof payload.data !== "string" ||
+                    typeof payload.mimeType !== "string") {
+                    throw new Error("AI Image Plugin returned an invalid image payload.");
+                }
+                const generatedFile = createGeneratedFile(payload, prompt, "ai-image-plugin");
+                setGenerationPhase("uploading");
+                const uploadedAsset = await client.assets.upload("image", generatedFile, {
+                    filename: generatedFile.name,
+                });
+                setGenerationPhase("updating");
+                onSelect([{ kind: "assetDocumentId", value: uploadedAsset._id }]);
+                onClose();
+            }
+            catch (nextError) {
+                setError(getErrorMessage(nextError));
+            }
+            finally {
+                setGenerationPhase("idle");
+            }
+        }
+        return (_jsx(Card, { padding: 4, children: _jsxs(Stack, { space: 4, children: [_jsxs(Stack, { space: 2, children: [_jsx(Text, { size: 2, weight: "semibold", children: "Generate an image with AI Image Plugin and drop it straight into this field." }), _jsx(Text, { muted: true, size: 1, children: "Project defaults from the AI Image Plugin settings tool are included automatically when configured." })] }), _jsxs(Box, { children: [_jsx(Text, { size: 1, weight: "medium", children: "Model" }), _jsx(Box, { marginTop: 2, children: _jsx(Select, { onChange: (event) => {
+                                        hasCustomSelectedModelRef.current = true;
+                                        setSelectedModel(event.currentTarget.value);
+                                    }, value: selectedModel, children: allowedModelOptions.map((modelOption) => (_jsx("option", { value: modelOption.value, children: modelOption.title }, modelOption.value))) }) })] }), _jsxs(Box, { children: [_jsx(Text, { size: 1, weight: "medium", children: assetSourceTarget?.promptLabel || "Prompt" }), _jsx(Box, { marginTop: 2, children: _jsx(TextArea, { onChange: (event) => setPrompt(event.currentTarget.value), placeholder: assetSourceTarget?.promptPlaceholder ||
+                                        "Turn these references into a clean homepage hero image with warm daylight and subtle depth.", rows: 6, value: prompt }) })] }), _jsxs(Box, { children: [_jsx(Text, { size: 1, weight: "medium", children: "Reference images" }), _jsx(Box, { marginTop: 2, children: _jsx("input", { accept: "image/*", multiple: true, onChange: handleReferenceImageChange, type: "file" }) }), referenceImages.length > 0 ? (_jsx(Box, { marginTop: 3, children: _jsx(Stack, { space: 2, children: referenceImages.map((referenceImage, index) => (_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Flex, { align: "center", gap: 3, justify: "space-between", children: [_jsxs(Box, { flex: 1, children: [_jsx(Text, { size: 1, weight: "medium", children: referenceImage.name }), _jsx(Text, { muted: true, size: 1, children: formatFileSize(referenceImage.size) })] }), _jsx(Button, { mode: "bleed", onClick: () => handleRemoveReferenceImage(index), text: "Remove" })] }) }, `${referenceImage.name}-${index}`))) }) })) : null] }), error ? (_jsx(Card, { padding: 3, radius: 2, tone: "critical", children: _jsx(Text, { size: 1, children: error }) })) : null, frontendRateLimit.blocked && !isGenerating ? (_jsx(Card, { padding: 3, radius: 2, tone: "caution", children: _jsxs(Text, { size: 1, children: ["Please wait", " ", formatRateLimitRemainingTime(frontendRateLimit.remainingMs), " ", "before generating again."] }) })) : null, isGenerating ? (_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Text, { size: 1, children: [getGenerationPhaseLabel(generationPhase), " This can take a minute or two for slower models."] }) })) : null, _jsxs(Flex, { gap: 3, justify: "flex-end", children: [_jsx(Button, { mode: "ghost", onClick: onClose, text: "Cancel" }), _jsx(Button, { disabled: isGenerating ||
+                                    isLoadingSharedSecret ||
+                                    frontendRateLimit.blocked, onClick: () => void handleGenerate(), text: getGenerationPhaseLabel(generationPhase), tone: "primary" })] })] }) }));
+    }
+    return {
+        name: SOURCE_NAME,
+        title: SOURCE_TITLE,
+        component: AssetSourceComponent,
+    };
+}
+//# sourceMappingURL=asset-source.js.map

package/dist/studio/components/asset-source.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"asset-source.js","sourceRoot":"","sources":["../../../src/studio/components/asset-source.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAEb,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EACL,GAAG,EACH,MAAM,EACN,IAAI,EACJ,IAAI,EACJ,MAAM,EACN,KAAK,EACL,IAAI,EACJ,QAAQ,GACT,MAAM,YAAY,CAAC;AACpB,OAAO,EAAoB,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC/E,OAAO,EAGL,SAAS,GACV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,6BAA6B,EAC7B,+BAA+B,GAEhC,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,WAAW,EACX,YAAY,GAEb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAEL,6BAA6B,EAC7B,oBAAoB,EACpB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EACL,4BAA4B,EAC5B,qBAAqB,EACrB,kBAAkB,EAClB,eAAe,EACf,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,4BAA4B,EAC5B,oBAAoB,GACrB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,cAAc,GACf,MAAM,UAAU,CAAC;AAelB,SAAS,uBAAuB,CAAC,KAAsB;IACrD,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,WAAW;YACd,OAAO,cAAc,CAAC;QACxB,KAAK,YAAY;YACf,OAAO,eAAe,CAAC;QACzB,KAAK,WAAW;YACd,OAAO,wBAAwB,CAAC;QAClC,KAAK,UAAU;YACb,OAAO,mBAAmB,CAAC;QAC7B;YACE,OAAO,+BAA+B,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,OAAwB;IACxD,MAAM,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IAEpD,SAAS,oBAAoB,CAAC,EAC5B,OAAO,EACP,QAAQ,GACkB;QAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;QAC7D,MAAM,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,GAC/C,UAAU,CAAuB,uBAAuB,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;QACzC,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CACrD,CAAC;QACF,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAS,EAAE,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GACzC,QAAQ,CAAkB,MAAM,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,eAAe,KAAK,MAAM,CAAC;QAChD,MAAM,YAAY,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,yBAAyB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC1E,MAAM,mBAAmB,GAAG,OAAO,CACjC,GAAG,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,aAAa,CAAC,EAC5D,CAAC,OAAO,CAAC,aAAa,CAAC,CACxB,CAAC;QAEF,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,SAAS,GAAG,IAAI,CAAC;YAErB,gBAAgB,CAAC,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC;YACvE,yBAAyB,CAAC,OAAO,GAAG,KAAK,CAAC;YAE1C,KAAK,UAAU,gBAAgB;gBAC7B,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAC;oBAEF,IAAI,CAAC,SAAS,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC;wBACpD,OAAO;oBACT,CAAC;oBAED,gBAAgB,CACd,oBAAoB,CAAC;wBACnB,aAAa,EAAE,OAAO,CAAC,aAAa;wBACpC,QAAQ;qBACT,CAAC,CACH,CAAC;gBACJ,CAAC;gBAAC,MAAM,CAAC;oBACP,mEAAmE;gBACrE,CAAC;YACH,CAAC;YAED,KAAK,gBAAgB,EAAE,CAAC;YAExB,OAAO,GAAG,EAAE;gBACV,SAAS,GAAG,KAAK,CAAC;YACpB,CAAC,CAAC;QACJ,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC;QAEpC,SAAS,0BAA0B,CAAC,KAAoC;YACtE,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAE9D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3B,OAAO;YACT,CAAC;YAED,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE,CAClC,CAAC,GAAG,YAAY,EAAE,GAAG,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAC/D,CAAC;YACF,KAAK,CAAC,aAAa,CAAC,KAAK,GAAG,EAAE,CAAC;QACjC,CAAC;QAED,SAAS,0BAA0B,CAAC,aAAqB;YACvD,kBAAkB,CAAC,CAAC,YAAY,EAAE,EAAE,CAClC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,aAAa,CAAC,CAC3D,CAAC;QACJ,CAAC;QAED,KAAK,UAAU,cAAc;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAC;YACf,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAEhC,IAAI,CAAC;gBACH,IAAI,qBAAqB,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;gBAC9D,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,EAAE,CAAC,CAAC;gBACnD,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAC;gBACF,MAAM,cAAc,GAAG,iBAAiB;oBACtC,CAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE,iBAAiB,CAAC,EAAE,CAAC;oBACpD,CAAC,CAAC,IAAI,CAAC;gBACT,MAAM,cAAc,GAAG,aAAa,CAAC;oBACnC,QAAQ,EAAE,YAAY;wBACpB,CAAC,CAAC,sBAAsB,QAAQ,CAAC,YAAY,EAAE;wBAC/C,CAAC,CAAC,IAAI;oBACR,cAAc,EAAE,MAAM;wBACpB,CAAC,CAAC,4BAA4B,cAAc,CAAC,MAAM,EAAE;wBACrD,CAAC,CAAC,IAAI;oBACR,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,mBAAmB,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI;iBAC1D,CAAC,CAAC;gBAEH,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;gBACvE,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAChC,MAAM,yBAAyB,GAAG,MAAM,OAAO,CAAC,GAAG,CACjD,eAAe,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAC3C,CAAC;gBACF,MAAM,uBAAuB,GAAG,MAAM,4BAA4B,CAChE,QAAQ,EACR,iBAAiB,EAAE,EAAE,CACtB,CAAC;gBACF,MAAM,kBAAkB,GAAG;oBACzB,GAAG,uBAAuB;oBAC1B,GAAG,yBAAyB;iBAC7B,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAC;gBAEjC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;gBACvC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBAErC,KAAK,MAAM,cAAc,IAAI,kBAAkB,EAAE,CAAC;oBAChD,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;gBAChD,CAAC;gBAED,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,cAAc,EAAE,CAAC;gBAE7D,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC;oBAC/B,OAAO;gBACT,CAAC;gBAED,kBAAkB,CAAC,YAAY,CAAC,CAAC;gBACjC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE;oBAChD,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE;wBACP,CAAC,yBAAyB,CAAC,EAAE,YAAY;qBAC1C;iBACF,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAEhB,CAAC;gBAEvB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,IAAI,KAAK,CACb,OAAO,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK;wBACjC,CAAC,CAAC,OAAO,CAAC,KAAK;wBACf,CAAC,CAAC,6BAA6B,CAClC,CAAC;gBACJ,CAAC;gBAED,IACE,CAAC,CAAC,MAAM,IAAI,OAAO,CAAC;oBACpB,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;oBAChC,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,EACpC,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;gBACxE,CAAC;gBAED,MAAM,aAAa,GAAG,mBAAmB,CACvC,OAAO,EACP,MAAM,EACN,iBAAiB,CAClB,CAAC;gBACF,kBAAkB,CAAC,WAAW,CAAC,CAAC;gBAChC,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAC9C,OAAO,EACP,aAAa,EACb;oBACE,QAAQ,EAAE,aAAa,CAAC,IAAI;iBAC7B,CACF,CAAC;gBAEF,kBAAkB,CAAC,UAAU,CAAC,CAAC;gBAC/B,QAAQ,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAClE,OAAO,EAAE,CAAC;YACZ,CAAC;YAAC,OAAO,SAAS,EAAE,CAAC;gBACnB,QAAQ,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAC;YACvC,CAAC;oBAAS,CAAC;gBACT,kBAAkB,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QAED,OAAO,CACL,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,YACd,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,UAAU,6FAGzB,EACP,KAAC,IAAI,IAAC,KAAK,QAAC,IAAI,EAAE,CAAC,oHAGZ,IACD,EAER,MAAC,GAAG,eACF,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,sBAEvB,EACP,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,KAAC,MAAM,IACL,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE;wCAClB,yBAAyB,CAAC,OAAO,GAAG,IAAI,CAAC;wCACzC,gBAAgB,CACd,KAAK,CAAC,aAAa,CAAC,KAAgC,CACrD,CAAC;oCACJ,CAAC,EACD,KAAK,EAAE,aAAa,YAEnB,mBAAmB,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CACxC,iBAAgC,KAAK,EAAE,WAAW,CAAC,KAAK,YACrD,WAAW,CAAC,KAAK,IADP,WAAW,CAAC,KAAK,CAErB,CACV,CAAC,GACK,GACL,IACF,EAEN,MAAC,GAAG,eACF,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,YAC3B,iBAAiB,EAAE,WAAW,IAAI,QAAQ,GACtC,EACP,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,KAAC,QAAQ,IACP,QAAQ,EAAE,CAAC,KAAuC,EAAE,EAAE,CACpD,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,EAEtC,WAAW,EACT,iBAAiB,EAAE,iBAAiB;wCACpC,6FAA6F,EAE/F,IAAI,EAAE,CAAC,EACP,KAAK,EAAE,MAAM,GACb,GACE,IACF,EAEN,MAAC,GAAG,eACF,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,iCAEvB,EACP,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,gBACE,MAAM,EAAC,SAAS,EAChB,QAAQ,QACR,QAAQ,EAAE,0BAA0B,EACpC,IAAI,EAAC,MAAM,GACX,GACE,EACL,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAC5B,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,KAAC,KAAK,IAAC,KAAK,EAAE,CAAC,YACZ,eAAe,CAAC,GAAG,CAAC,CAAC,cAAc,EAAE,KAAK,EAAE,EAAE,CAAC,CAC9C,KAAC,IAAI,IAEH,OAAO,EAAE,CAAC,EACV,MAAM,EAAE,CAAC,EACT,IAAI,EAAC,aAAa,YAElB,MAAC,IAAI,IAAC,KAAK,EAAC,QAAQ,EAAC,GAAG,EAAE,CAAC,EAAE,OAAO,EAAC,eAAe,aAClD,MAAC,GAAG,IAAC,IAAI,EAAE,CAAC,aACV,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,YAC3B,cAAc,CAAC,IAAI,GACf,EACP,KAAC,IAAI,IAAC,KAAK,QAAC,IAAI,EAAE,CAAC,YAChB,cAAc,CAAC,cAAc,CAAC,IAAI,CAAC,GAC/B,IACH,EACN,KAAC,MAAM,IACL,IAAI,EAAC,OAAO,EACZ,OAAO,EAAE,GAAG,EAAE,CAAC,0BAA0B,CAAC,KAAK,CAAC,EAChD,IAAI,EAAC,QAAQ,GACb,IACG,IAnBF,GAAG,cAAc,CAAC,IAAI,IAAI,KAAK,EAAE,CAoBjC,CACR,CAAC,GACI,GACJ,CACP,CAAC,CAAC,CAAC,IAAI,IACJ,EAEL,KAAK,CAAC,CAAC,CAAC,CACP,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,UAAU,YAC1C,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,YAAG,KAAK,GAAQ,GACxB,CACR,CAAC,CAAC,CAAC,IAAI,EAEP,iBAAiB,CAAC,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAC5C,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,SAAS,YACzC,MAAC,IAAI,IAAC,IAAI,EAAE,CAAC,4BACC,GAAG,EACd,4BAA4B,CAAC,iBAAiB,CAAC,WAAW,CAAC,EAAE,GAAG,gCAE5D,GACF,CACR,CAAC,CAAC,CAAC,IAAI,EAEP,YAAY,CAAC,CAAC,CAAC,CACd,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,aAAa,YAC7C,MAAC,IAAI,IAAC,IAAI,EAAE,CAAC,aACV,uBAAuB,CAAC,eAAe,CAAC,yDAEpC,GACF,CACR,CAAC,CAAC,CAAC,IAAI,EAER,MAAC,IAAI,IAAC,GAAG,EAAE,CAAC,EAAE,OAAO,EAAC,UAAU,aAC9B,KAAC,MAAM,IAAC,IAAI,EAAC,OAAO,EAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAC,QAAQ,GAAG,EACvD,KAAC,MAAM,IACL,QAAQ,EACN,YAAY;oCACZ,qBAAqB;oCACrB,iBAAiB,CAAC,OAAO,EAE3B,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,cAAc,EAAE,EACpC,IAAI,EAAE,uBAAuB,CAAC,eAAe,CAAC,EAC9C,IAAI,EAAC,SAAS,GACd,IACG,IACD,GACH,CACR,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,oBAAoB;KAChC,CAAC;AACJ,CAAC"}

package/dist/studio/components/frontend-rate-limit.d.ts

@@ -0,0 +1,29 @@
+import type { ResolvedFrontendRateLimitOptions } from "../../utils/shared";
+export declare const FRONTEND_RATE_LIMIT_STORAGE_KEY = "ai-image-plugin.frontend-rate-limit.v1";
+type FrontendRateLimitState = {
+    attempts: number[];
+};
+type FrontendRateLimitEnvironment = {
+    fallbackState?: FrontendRateLimitState;
+    storage?: Pick<Storage, "getItem" | "setItem"> | null;
+};
+export type FrontendRateLimitSnapshot = {
+    attempts: number[];
+    blocked: boolean;
+    remainingMs: number;
+};
+export type FrontendRateLimitAttemptResult = FrontendRateLimitSnapshot & {
+    allowed: boolean;
+};
+export declare function createFrontendRateLimiter(options: ResolvedFrontendRateLimitOptions | false, environment?: FrontendRateLimitEnvironment): {
+    consumeAttempt(now?: number): FrontendRateLimitAttemptResult;
+    getSnapshot(now?: number): FrontendRateLimitSnapshot;
+};
+export declare function formatRateLimitRemainingTime(remainingMs: number): string;
+export declare function useFrontendRateLimit(options: ResolvedFrontendRateLimitOptions | false): {
+    blocked: boolean;
+    consumeAttempt: () => FrontendRateLimitAttemptResult;
+    remainingMs: number;
+};
+export {};
+//# sourceMappingURL=frontend-rate-limit.d.ts.map

package/dist/studio/components/frontend-rate-limit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-rate-limit.d.ts","sourceRoot":"","sources":["../../../src/studio/components/frontend-rate-limit.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,oBAAoB,CAAC;AAE3E,eAAO,MAAM,+BAA+B,2CACF,CAAC;AAI3C,KAAK,sBAAsB,GAAG;IAC5B,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAAC;AAEF,KAAK,4BAA4B,GAAG;IAClC,aAAa,CAAC,EAAE,sBAAsB,CAAC;IACvC,OAAO,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,SAAS,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;CACvD,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,8BAA8B,GAAG,yBAAyB,GAAG;IACvE,OAAO,EAAE,OAAO,CAAC;CAClB,CAAC;AAiKF,wBAAgB,yBAAyB,CACvC,OAAO,EAAE,gCAAgC,GAAG,KAAK,EACjD,WAAW,CAAC,EAAE,4BAA4B;kCAGN,8BAA8B;+BAsBjC,yBAAyB;EAI3D;AAED,wBAAgB,4BAA4B,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAMxE;AAED,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,gCAAgC,GAAG,KAAK;;0BAuDtB,8BAA8B;;EAa1D"}

package/dist/studio/components/frontend-rate-limit.js

@@ -0,0 +1,197 @@
+"use client";
+import { useEffect, useState } from "react";
+export const FRONTEND_RATE_LIMIT_STORAGE_KEY = "ai-image-plugin.frontend-rate-limit.v1";
+const FRONTEND_RATE_LIMIT_SYNC_EVENT = "ai-image-plugin:frontend-rate-limit";
+const memoryFallbackState = {
+    attempts: [],
+};
+function createEmptySnapshot() {
+    return {
+        attempts: [],
+        blocked: false,
+        remainingMs: 0,
+    };
+}
+function getWindowMs(options) {
+    return options.windowSecs * 1000;
+}
+function hasSameAttempts(left, right) {
+    if (left.length !== right.length) {
+        return false;
+    }
+    return left.every((value, index) => value === right[index]);
+}
+function pruneAttempts(attempts, windowMs, now) {
+    return attempts.filter((attempt) => now - attempt < windowMs);
+}
+function parseAttempts(value) {
+    if (!value) {
+        return [];
+    }
+    try {
+        const parsedValue = JSON.parse(value);
+        if (!Array.isArray(parsedValue.attempts)) {
+            return [];
+        }
+        return parsedValue.attempts
+            .filter((attempt) => typeof attempt === "number" && Number.isFinite(attempt))
+            .sort((left, right) => left - right);
+    }
+    catch {
+        return [];
+    }
+}
+function resolveStorage(storage) {
+    if (storage !== undefined) {
+        return storage;
+    }
+    if (typeof window === "undefined") {
+        return null;
+    }
+    try {
+        return window.localStorage;
+    }
+    catch {
+        return null;
+    }
+}
+function readAttempts(storage, fallbackState) {
+    if (!storage) {
+        return [...fallbackState.attempts];
+    }
+    try {
+        return parseAttempts(storage.getItem(FRONTEND_RATE_LIMIT_STORAGE_KEY));
+    }
+    catch {
+        return [...fallbackState.attempts];
+    }
+}
+function persistAttempts(attempts, storage, fallbackState) {
+    fallbackState.attempts = [...attempts];
+    if (!storage) {
+        return;
+    }
+    try {
+        storage.setItem(FRONTEND_RATE_LIMIT_STORAGE_KEY, JSON.stringify({ attempts }));
+    }
+    catch {
+        // Keep the in-memory fallback current when storage writes fail.
+    }
+}
+function notifyAttemptListeners() {
+    if (typeof window === "undefined") {
+        return;
+    }
+    window.dispatchEvent(new Event(FRONTEND_RATE_LIMIT_SYNC_EVENT));
+}
+function buildSnapshot(attempts, options, now) {
+    const windowMs = getWindowMs(options);
+    const blocked = attempts.length >= options.maxRequests;
+    const relevantAttempt = blocked
+        ? attempts[attempts.length - options.maxRequests]
+        : null;
+    return {
+        attempts,
+        blocked,
+        remainingMs: relevantAttempt === null
+            ? 0
+            : Math.max(0, relevantAttempt + windowMs - now),
+    };
+}
+function readSnapshot(options, environment, now = Date.now()) {
+    if (!options) {
+        return createEmptySnapshot();
+    }
+    const fallbackState = environment?.fallbackState || memoryFallbackState;
+    const storage = resolveStorage(environment?.storage);
+    const storedAttempts = readAttempts(storage, fallbackState);
+    const attempts = pruneAttempts(storedAttempts, getWindowMs(options), now);
+    if (!hasSameAttempts(storedAttempts, attempts)) {
+        persistAttempts(attempts, storage, fallbackState);
+    }
+    return buildSnapshot(attempts, options, now);
+}
+export function createFrontendRateLimiter(options, environment) {
+    return {
+        consumeAttempt(now = Date.now()) {
+            const currentSnapshot = readSnapshot(options, environment, now);
+            if (!options || currentSnapshot.blocked) {
+                return {
+                    ...currentSnapshot,
+                    allowed: !currentSnapshot.blocked,
+                };
+            }
+            const fallbackState = environment?.fallbackState || memoryFallbackState;
+            const storage = resolveStorage(environment?.storage);
+            const attempts = [...currentSnapshot.attempts, now];
+            persistAttempts(attempts, storage, fallbackState);
+            notifyAttemptListeners();
+            return {
+                ...buildSnapshot(attempts, options, now),
+                allowed: true,
+            };
+        },
+        getSnapshot(now = Date.now()) {
+            return readSnapshot(options, environment, now);
+        },
+    };
+}
+export function formatRateLimitRemainingTime(remainingMs) {
+    const totalSeconds = Math.max(0, Math.ceil(remainingMs / 1000));
+    const minutes = String(Math.floor(totalSeconds / 60)).padStart(2, "0");
+    const seconds = String(totalSeconds % 60).padStart(2, "0");
+    return `${minutes}:${seconds}`;
+}
+export function useFrontendRateLimit(options) {
+    const maxRequests = options ? options.maxRequests : 0;
+    const windowSecs = options ? options.windowSecs : 0;
+    const [snapshot, setSnapshot] = useState(() => createFrontendRateLimiter(options).getSnapshot());
+    useEffect(() => {
+        setSnapshot(createFrontendRateLimiter(options).getSnapshot());
+    }, [maxRequests, windowSecs]);
+    useEffect(() => {
+        if (typeof window === "undefined") {
+            return;
+        }
+        function syncSnapshot() {
+            setSnapshot(createFrontendRateLimiter(options).getSnapshot());
+        }
+        function handleStorage(event) {
+            if (!event.key || event.key === FRONTEND_RATE_LIMIT_STORAGE_KEY) {
+                syncSnapshot();
+            }
+        }
+        function handleSync() {
+            syncSnapshot();
+        }
+        syncSnapshot();
+        window.addEventListener("storage", handleStorage);
+        window.addEventListener(FRONTEND_RATE_LIMIT_SYNC_EVENT, handleSync);
+        return () => {
+            window.removeEventListener("storage", handleStorage);
+            window.removeEventListener(FRONTEND_RATE_LIMIT_SYNC_EVENT, handleSync);
+        };
+    }, [maxRequests, windowSecs]);
+    useEffect(() => {
+        if (!snapshot.blocked || typeof window === "undefined") {
+            return;
+        }
+        const intervalId = window.setInterval(() => {
+            setSnapshot(createFrontendRateLimiter(options).getSnapshot());
+        }, 1000);
+        return () => {
+            window.clearInterval(intervalId);
+        };
+    }, [snapshot.blocked, maxRequests, windowSecs]);
+    function consumeAttempt() {
+        const nextSnapshot = createFrontendRateLimiter(options).consumeAttempt();
+        setSnapshot(nextSnapshot);
+        return nextSnapshot;
+    }
+    return {
+        blocked: snapshot.blocked,
+        consumeAttempt,
+        remainingMs: snapshot.remainingMs,
+    };
+}
+//# sourceMappingURL=frontend-rate-limit.js.map

package/dist/studio/components/frontend-rate-limit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontend-rate-limit.js","sourceRoot":"","sources":["../../../src/studio/components/frontend-rate-limit.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAG5C,MAAM,CAAC,MAAM,+BAA+B,GAC1C,wCAAwC,CAAC;AAE3C,MAAM,8BAA8B,GAAG,qCAAqC,CAAC;AAqB7E,MAAM,mBAAmB,GAA2B;IAClD,QAAQ,EAAE,EAAE;CACb,CAAC;AAEF,SAAS,mBAAmB;IAC1B,OAAO;QACL,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,OAAyC;IAC5D,OAAO,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;AACnC,CAAC;AAED,SAAS,eAAe,CAAC,IAAc,EAAE,KAAe;IACtD,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,aAAa,CACpB,QAAkB,EAClB,QAAgB,EAChB,GAAW;IAEX,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,GAAG,QAAQ,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,aAAa,CAAC,KAAoB;IACzC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAA2B,CAAC;QAEhE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,WAAW,CAAC,QAAQ;aACxB,MAAM,CACL,CAAC,OAAO,EAAqB,EAAE,CAC7B,OAAO,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAC1D;aACA,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC;IACzC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CACrB,OAAgD;IAEhD,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,CAAC;QACH,OAAO,MAAM,CAAC,YAAY,CAAC;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CACnB,OAAoD,EACpD,aAAqC;IAErC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;IAED,IAAI,CAAC;QACH,OAAO,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC,CAAC;IACzE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CACtB,QAAkB,EAClB,OAAoD,EACpD,aAAqC;IAErC,aAAa,CAAC,QAAQ,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IAEvC,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,OAAO,CAAC,OAAO,CACb,+BAA+B,EAC/B,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC,CAC7B,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,gEAAgE;IAClE,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB;IAC7B,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO;IACT,CAAC;IAED,MAAM,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,aAAa,CACpB,QAAkB,EAClB,OAAyC,EACzC,GAAW;IAEX,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,IAAI,OAAO,CAAC,WAAW,CAAC;IACvD,MAAM,eAAe,GAAG,OAAO;QAC7B,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC;QACjD,CAAC,CAAC,IAAI,CAAC;IAET,OAAO;QACL,QAAQ;QACR,OAAO;QACP,WAAW,EACT,eAAe,KAAK,IAAI;YACtB,CAAC,CAAC,CAAC;YACH,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,GAAG,QAAQ,GAAG,GAAG,CAAC;KACpD,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,OAAiD,EACjD,WAA0C,EAC1C,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAEhB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,mBAAmB,EAAE,CAAC;IAC/B,CAAC;IAED,MAAM,aAAa,GAAG,WAAW,EAAE,aAAa,IAAI,mBAAmB,CAAC;IACxE,MAAM,OAAO,GAAG,cAAc,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IACrD,MAAM,cAAc,GAAG,YAAY,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAC5D,MAAM,QAAQ,GAAG,aAAa,CAAC,cAAc,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,CAAC;IAE1E,IAAI,CAAC,eAAe,CAAC,cAAc,EAAE,QAAQ,CAAC,EAAE,CAAC;QAC/C,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,OAAO,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,OAAiD,EACjD,WAA0C;IAE1C,OAAO;QACL,cAAc,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;YAC7B,MAAM,eAAe,GAAG,YAAY,CAAC,OAAO,EAAE,WAAW,EAAE,GAAG,CAAC,CAAC;YAEhE,IAAI,CAAC,OAAO,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;gBACxC,OAAO;oBACL,GAAG,eAAe;oBAClB,OAAO,EAAE,CAAC,eAAe,CAAC,OAAO;iBAClC,CAAC;YACJ,CAAC;YAED,MAAM,aAAa,GAAG,WAAW,EAAE,aAAa,IAAI,mBAAmB,CAAC;YACxE,MAAM,OAAO,GAAG,cAAc,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAEpD,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;YAClD,sBAAsB,EAAE,CAAC;YAEzB,OAAO;gBACL,GAAG,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC;gBACxC,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QACD,WAAW,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;YAC1B,OAAO,YAAY,CAAC,OAAO,EAAE,WAAW,EAAE,GAAG,CAAC,CAAC;QACjD,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,WAAmB;IAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACvE,MAAM,OAAO,GAAG,MAAM,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAE3D,OAAO,GAAG,OAAO,IAAI,OAAO,EAAE,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,OAAiD;IAEjD,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IACtD,MAAM,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;IACpD,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAA4B,GAAG,EAAE,CACvE,yBAAyB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CACjD,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,WAAW,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;IAChE,CAAC,EAAE,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;IAE9B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QAED,SAAS,YAAY;YACnB,WAAW,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,SAAS,aAAa,CAAC,KAAmB;YACxC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,KAAK,+BAA+B,EAAE,CAAC;gBAChE,YAAY,EAAE,CAAC;YACjB,CAAC;QACH,CAAC;QAED,SAAS,UAAU;YACjB,YAAY,EAAE,CAAC;QACjB,CAAC;QAED,YAAY,EAAE,CAAC;QACf,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;QAClD,MAAM,CAAC,gBAAgB,CAAC,8BAA8B,EAAE,UAAU,CAAC,CAAC;QAEpE,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,mBAAmB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;YACrD,MAAM,CAAC,mBAAmB,CAAC,8BAA8B,EAAE,UAAU,CAAC,CAAC;QACzE,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;IAE9B,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;YACvD,OAAO;QACT,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,EAAE;YACzC,WAAW,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAChE,CAAC,EAAE,IAAI,CAAC,CAAC;QAET,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC;IAEhD,SAAS,cAAc;QACrB,MAAM,YAAY,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,CAAC;QAEzE,WAAW,CAAC,YAAY,CAAC,CAAC;QAE1B,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO;QACL,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,cAAc;QACd,WAAW,EAAE,QAAQ,CAAC,WAAW;KAClC,CAAC;AACJ,CAAC"}

package/dist/studio/components/generate-button-input.d.ts

@@ -0,0 +1,8 @@
+import { type ImageValue, type ObjectInputProps } from "sanity";
+import { type GenerateButtonTarget, type ResolvedOptions } from "../../utils/shared";
+export declare function GenerateButtonInput({ options, props, target, }: {
+    options: ResolvedOptions;
+    props: ObjectInputProps<ImageValue>;
+    target: GenerateButtonTarget;
+}): import("react/jsx-runtime").JSX.Element;
+//# sourceMappingURL=generate-button-input.d.ts.map

package/dist/studio/components/generate-button-input.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-button-input.d.ts","sourceRoot":"","sources":["../../../src/studio/components/generate-button-input.tsx"],"names":[],"mappings":"AAeA,OAAO,EAEL,KAAK,UAAU,EACf,KAAK,gBAAgB,EAMtB,MAAM,QAAQ,CAAC;AAkBhB,OAAO,EAGL,KAAK,oBAAoB,EACzB,KAAK,eAAe,EACrB,MAAM,oBAAoB,CAAC;AA0C5B,wBAAgB,mBAAmB,CAAC,EAClC,OAAO,EACP,KAAK,EACL,MAAM,GACP,EAAE;IACD,OAAO,EAAE,eAAe,CAAC;IACzB,KAAK,EAAE,gBAAgB,CAAC,UAAU,CAAC,CAAC;IACpC,MAAM,EAAE,oBAAoB,CAAC;CAC9B,2CA6ZA"}

package/dist/studio/components/generate-button-input.js

@@ -0,0 +1,194 @@
+"use client";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
+import { Box, Button, Card, Dialog, Flex, Select, Stack, Text, TextArea, } from "@sanity/ui";
+import { useSecrets } from "@sanity/studio-secrets";
+import { useEffect, useMemo, useRef, useState } from "react";
+import { PatchEvent, set, setIfMissing, useClient, useFormValue, useSchema, } from "sanity";
+import { getMissingSharedSecretMessage, getSharedSecretValue, SHARED_SECRET_HEADER_NAME, SHARED_SECRET_NAMESPACE, } from "../shared-secret";
+import { buildContextFieldPrompt, getDefaultSelectedContextFieldPaths, getSelectableContextFields, } from "../../utils/context-fields";
+import { getDefaultAllowedAiImageModel, getSupportedAiImageModelOptions, } from "../../utils/models";
+import { SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE, } from "../../utils/shared";
+import { composePrompt } from "../../utils/prompts";
+import { buildSettingsReferenceImages, fetchSettingsDocument, findTargetSettings, getErrorMessage, resolveSettingsModel, } from "../settings-data";
+import { formatRateLimitRemainingTime, useFrontendRateLimit, } from "./frontend-rate-limit";
+import { createGeneratedFile } from "../files";
+function getGenerationPhaseLabel(phase) {
+    switch (phase) {
+        case "preparing":
+            return "Preparing...";
+        case "generating":
+            return "Generating...";
+        case "uploading":
+            return "Uploading to Sanity...";
+        case "updating":
+            return "Updating field...";
+        default:
+            return "Generate image";
+    }
+}
+export function GenerateButtonInput({ options, props, target, }) {
+    const client = useClient({ apiVersion: options.apiVersion });
+    const { loading: isLoadingSharedSecret, secrets } = useSecrets(SHARED_SECRET_NAMESPACE);
+    const schema = useSchema();
+    const documentValue = useFormValue([]);
+    const documentType = useFormValue(["_type"]);
+    const [isDialogOpen, setIsDialogOpen] = useState(false);
+    const [userPrompt, setUserPrompt] = useState("");
+    const [selectedModel, setSelectedModel] = useState(getDefaultAllowedAiImageModel(options.allowedModels));
+    const [selectedContextFieldPaths, setSelectedContextFieldPaths] = useState([]);
+    const [error, setError] = useState(null);
+    const [generationPhase, setGenerationPhase] = useState("idle");
+    const isGenerating = generationPhase !== "idle";
+    const currentAssetRef = props.value?.asset?._ref || "empty";
+    const hasCustomSelectedModelRef = useRef(false);
+    const frontendRateLimit = useFrontendRateLimit(options.frontendRateLimit);
+    const selectableContextFields = useMemo(() => getSelectableContextFields(documentType ? schema.get(documentType) : undefined), [documentType, schema]);
+    const defaultSelectedContextFieldPaths = useMemo(() => getDefaultSelectedContextFieldPaths({
+        selectableContextFields,
+        suggestedContextFieldPaths: target.suggestedContextFieldPaths,
+    }), [selectableContextFields, target.suggestedContextFieldPaths]);
+    const runtimePrompt = buildContextFieldPrompt({
+        contextFieldPaths: selectedContextFieldPaths,
+        documentValue,
+    });
+    const sharedSecret = getSharedSecretValue(secrets);
+    const allowedModelOptions = useMemo(() => getSupportedAiImageModelOptions(options.allowedModels), [options.allowedModels]);
+    useEffect(() => {
+        if (!isDialogOpen) {
+            return;
+        }
+        let isMounted = true;
+        setSelectedModel(getDefaultAllowedAiImageModel(options.allowedModels));
+        hasCustomSelectedModelRef.current = false;
+        async function loadDefaultModel() {
+            try {
+                const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
+                if (!isMounted || hasCustomSelectedModelRef.current) {
+                    return;
+                }
+                setSelectedModel(resolveSettingsModel({
+                    allowedModels: options.allowedModels,
+                    settings,
+                }));
+            }
+            catch {
+                // Keep the first allowed model selected if settings loading fails.
+            }
+        }
+        void loadDefaultModel();
+        return () => {
+            isMounted = false;
+        };
+    }, [client, isDialogOpen, options.allowedModels]);
+    function toggleContextFieldPath(fieldPath) {
+        setSelectedContextFieldPaths((currentFieldPaths) => currentFieldPaths.includes(fieldPath)
+            ? currentFieldPaths.filter((currentFieldPath) => currentFieldPath !== fieldPath)
+            : [...currentFieldPaths, fieldPath]);
+    }
+    async function handleGenerate() {
+        setError(null);
+        setGenerationPhase("preparing");
+        try {
+            if (isLoadingSharedSecret) {
+                throw new Error("Loading AI Image Plugin shared secret...");
+            }
+            if (!sharedSecret) {
+                throw new Error(getMissingSharedSecretMessage());
+            }
+            const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
+            const targetSettings = findTargetSettings(settings, target.id);
+            const prompt = composePrompt([
+                settings?.globalPrompt
+                    ? `Global direction:\n${settings.globalPrompt}`
+                    : null,
+                targetSettings?.prompt
+                    ? `Target direction:\n${targetSettings.prompt}`
+                    : null,
+                runtimePrompt ? `Document context:\n${runtimePrompt}` : null,
+                userPrompt.trim()
+                    ? `Additional editor instructions:\n${userPrompt.trim()}`
+                    : null,
+            ]);
+            if (!prompt) {
+                throw new Error("Add a prompt or configure target defaults first.");
+            }
+            const referenceImages = await buildSettingsReferenceImages(settings, target.id);
+            const formData = new FormData();
+            formData.set("prompt", prompt);
+            formData.set("model", selectedModel);
+            for (const referenceImage of referenceImages) {
+                formData.append("references", referenceImage);
+            }
+            const rateLimitSnapshot = frontendRateLimit.consumeAttempt();
+            if (!rateLimitSnapshot.allowed) {
+                return;
+            }
+            setGenerationPhase("generating");
+            const response = await fetch(options.apiEndpoint, {
+                method: "POST",
+                body: formData,
+                headers: {
+                    [SHARED_SECRET_HEADER_NAME]: sharedSecret,
+                },
+            });
+            const payload = (await response.json());
+            if (!response.ok) {
+                throw new Error("error" in payload && payload.error
+                    ? payload.error
+                    : "AI image generation failed.");
+            }
+            if (!("data" in payload) ||
+                typeof payload.data !== "string" ||
+                typeof payload.mimeType !== "string") {
+                throw new Error("AI Image Plugin returned an invalid image payload.");
+            }
+            const basename = String(documentValue?.title ||
+                documentValue?._type ||
+                target.title ||
+                "generated").trim() || "generated";
+            const generatedFile = createGeneratedFile(payload, basename, "ai-image-plugin-field");
+            setGenerationPhase("uploading");
+            const uploadedAsset = await client.assets.upload("image", generatedFile, {
+                filename: generatedFile.name,
+            });
+            setGenerationPhase("updating");
+            props.onChange(PatchEvent.from([
+                setIfMissing({ _type: "image" }),
+                set({ _type: "reference", _ref: uploadedAsset._id }, ["asset"]),
+            ]));
+            setIsDialogOpen(false);
+            setUserPrompt("");
+        }
+        catch (nextError) {
+            setError(getErrorMessage(nextError, "Unable to generate an image for this field."));
+        }
+        finally {
+            setGenerationPhase("idle");
+        }
+    }
+    return (_jsxs(Stack, { space: 4, children: [_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Stack, { space: 3, children: [_jsx(Text, { size: 1, children: target.description ||
+                                "Generate an image with AI Image Plugin for this field." }), _jsx(Flex, { gap: 3, children: _jsx(Button, { onClick: () => {
+                                    setError(null);
+                                    setSelectedContextFieldPaths(defaultSelectedContextFieldPaths);
+                                    setIsDialogOpen(true);
+                                }, text: "Generate", tone: "primary" }) })] }) }), _jsx(Box, { children: props.renderDefault(props) }, currentAssetRef), isDialogOpen ? (_jsx(Dialog, { header: target.dialogTitle || "Generate Image", id: `${target.id}-generate-dialog`, onClose: () => {
+                    if (isGenerating) {
+                        return;
+                    }
+                    setError(null);
+                    setIsDialogOpen(false);
+                }, width: 1, children: _jsx(Box, { padding: 4, children: _jsxs(Stack, { space: 4, children: [_jsxs(Box, { children: [_jsx(Text, { size: 1, weight: "medium", children: "Model" }), _jsx(Box, { marginTop: 2, children: _jsx(Select, { onChange: (event) => {
+                                                hasCustomSelectedModelRef.current = true;
+                                                setSelectedModel(event.currentTarget.value);
+                                            }, value: selectedModel, children: allowedModelOptions.map((modelOption) => (_jsx("option", { value: modelOption.value, children: modelOption.title }, modelOption.value))) }) })] }), _jsxs(Box, { children: [_jsx(Text, { size: 1, weight: "medium", children: target.promptLabel || "Custom prompt" }), _jsx(Box, { marginTop: 2, children: _jsx(TextArea, { onChange: (event) => setUserPrompt(event.currentTarget.value), placeholder: target.promptPlaceholder ||
+                                                "Optional custom instructions for this image.", rows: 6, value: userPrompt }) })] }), selectableContextFields.length > 0 ? (_jsx(Box, { children: _jsx(Stack, { space: 3, children: _jsxs(Stack, { space: 2, children: [_jsx(Text, { size: 1, weight: "medium", children: "Include document fields" }), _jsx(Flex, { gap: 2, style: { flexWrap: "wrap" }, children: selectableContextFields.map((field) => {
+                                                    const isSelected = selectedContextFieldPaths.includes(field.path);
+                                                    return (_jsx(Button, { mode: isSelected ? "default" : "ghost", onClick: () => toggleContextFieldPath(field.path), text: field.label, tone: isSelected ? "primary" : "default" }, field.path));
+                                                }) })] }) }) })) : (_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Stack, { space: 2, children: [_jsx(Text, { size: 1, weight: "medium", children: "Document context" }), _jsx(Text, { muted: true, size: 1, children: "No supported top-level document fields are available for this type yet." })] }) })), selectableContextFields.length > 0 ? (_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Stack, { space: 2, children: [_jsx(Text, { size: 1, weight: "medium", children: "Document context preview" }), runtimePrompt ? (_jsx(Text, { size: 1, children: runtimePrompt })) : (_jsx(Text, { muted: true, size: 1, children: "No document fields are selected with a usable value yet." }))] }) })) : null, error ? (_jsx(Card, { padding: 3, radius: 2, tone: "critical", children: _jsx(Text, { size: 1, children: error }) })) : null, frontendRateLimit.blocked && !isGenerating ? (_jsx(Card, { padding: 3, radius: 2, tone: "caution", children: _jsxs(Text, { size: 1, children: ["Please wait", " ", formatRateLimitRemainingTime(frontendRateLimit.remainingMs), " ", "before generating again."] }) })) : null, isGenerating ? (_jsx(Card, { padding: 3, radius: 2, tone: "transparent", children: _jsxs(Text, { size: 1, children: [getGenerationPhaseLabel(generationPhase), " This can take a minute or two for slower models."] }) })) : null, _jsxs(Flex, { gap: 3, justify: "flex-end", children: [_jsx(Button, { disabled: isGenerating, mode: "ghost", onClick: () => {
+                                            setError(null);
+                                            setIsDialogOpen(false);
+                                        }, text: "Cancel" }), _jsx(Button, { disabled: isGenerating ||
+                                            isLoadingSharedSecret ||
+                                            frontendRateLimit.blocked, onClick: () => void handleGenerate(), text: getGenerationPhaseLabel(generationPhase), tone: "primary" })] })] }) }) })) : null] }));
+}
+//# sourceMappingURL=generate-button-input.js.map