@basementstudio/sanity-ai-image-plugin 0.0.0 → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +281 -257
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +3 -0
- package/dist/index.js.map +1 -0
- package/dist/presets/article-featured-image.d.ts +3 -0
- package/dist/presets/article-featured-image.d.ts.map +1 -0
- package/dist/presets/article-featured-image.js +17 -0
- package/dist/presets/article-featured-image.js.map +1 -0
- package/dist/server/constants.d.ts +6 -0
- package/dist/server/constants.d.ts.map +1 -0
- package/{src/server/constants.ts → dist/server/constants.js} +17 -20
- package/dist/server/constants.js.map +1 -0
- package/dist/server/handle-request.d.ts +5 -0
- package/dist/server/handle-request.d.ts.map +1 -0
- package/dist/server/handle-request.js +122 -0
- package/dist/server/handle-request.js.map +1 -0
- package/dist/server/types.d.ts +28 -0
- package/dist/server/types.d.ts.map +1 -0
- package/dist/server/types.js +2 -0
- package/dist/server/types.js.map +1 -0
- package/dist/server/utils.d.ts +50 -0
- package/dist/server/utils.d.ts.map +1 -0
- package/dist/server/utils.js +274 -0
- package/dist/server/utils.js.map +1 -0
- package/dist/server.d.ts +3 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +3 -0
- package/dist/server.js.map +1 -0
- package/dist/studio/components/asset-source.d.ts +4 -0
- package/dist/studio/components/asset-source.d.ts.map +1 -0
- package/dist/studio/components/asset-source.js +163 -0
- package/dist/studio/components/asset-source.js.map +1 -0
- package/dist/studio/components/generate-button-input.d.ts +8 -0
- package/dist/studio/components/generate-button-input.d.ts.map +1 -0
- package/dist/studio/components/generate-button-input.js +186 -0
- package/dist/studio/components/generate-button-input.js.map +1 -0
- package/dist/studio/components/input-router.d.ts +7 -0
- package/dist/studio/components/input-router.d.ts.map +1 -0
- package/dist/studio/components/input-router.js +21 -0
- package/dist/studio/components/input-router.js.map +1 -0
- package/dist/studio/files.d.ts +9 -0
- package/dist/studio/files.d.ts.map +1 -0
- package/dist/studio/files.js +86 -0
- package/dist/studio/files.js.map +1 -0
- package/dist/studio/plugin.d.ts +3 -0
- package/dist/studio/plugin.d.ts.map +1 -0
- package/dist/studio/plugin.js +47 -0
- package/dist/studio/plugin.js.map +1 -0
- package/dist/studio/settings/schema.d.ts +8 -0
- package/dist/studio/settings/schema.d.ts.map +1 -0
- package/dist/studio/settings/schema.js +110 -0
- package/dist/studio/settings/schema.js.map +1 -0
- package/dist/studio/settings/tool.d.ts +3 -0
- package/dist/studio/settings/tool.d.ts.map +1 -0
- package/dist/studio/settings/tool.js +292 -0
- package/dist/studio/settings/tool.js.map +1 -0
- package/dist/studio/settings-data.d.ts +24 -0
- package/dist/studio/settings-data.d.ts.map +1 -0
- package/dist/studio/settings-data.js +99 -0
- package/dist/studio/settings-data.js.map +1 -0
- package/dist/studio/shared-secret.d.ts +9 -0
- package/dist/studio/shared-secret.d.ts.map +1 -0
- package/dist/studio/shared-secret.js +37 -0
- package/dist/studio/shared-secret.js.map +1 -0
- package/dist/utils/config.d.ts +3 -0
- package/dist/utils/config.d.ts.map +1 -0
- package/dist/utils/config.js +41 -0
- package/dist/utils/config.js.map +1 -0
- package/dist/utils/context-fields.d.ts +16 -0
- package/dist/utils/context-fields.d.ts.map +1 -0
- package/dist/utils/context-fields.js +104 -0
- package/dist/utils/context-fields.js.map +1 -0
- package/dist/utils/document-paths.d.ts +10 -0
- package/dist/utils/document-paths.d.ts.map +1 -0
- package/dist/utils/document-paths.js +33 -0
- package/dist/utils/document-paths.js.map +1 -0
- package/dist/utils/models.d.ts +22 -0
- package/dist/utils/models.d.ts.map +1 -0
- package/dist/utils/models.js +76 -0
- package/dist/utils/models.js.map +1 -0
- package/dist/utils/prompts.d.ts +2 -0
- package/dist/utils/prompts.d.ts.map +1 -0
- package/dist/utils/prompts.js +7 -0
- package/dist/utils/prompts.js.map +1 -0
- package/dist/utils/shared.d.ts +82 -0
- package/dist/utils/shared.d.ts.map +1 -0
- package/dist/utils/shared.js +13 -0
- package/dist/utils/shared.js.map +1 -0
- package/package.json +53 -47
- package/src/index.ts +0 -23
- package/src/presets/article-featured-image.ts +0 -23
- package/src/server/handle-request.ts +0 -207
- package/src/server/types.ts +0 -30
- package/src/server/utils.ts +0 -395
- package/src/server.ts +0 -14
- package/src/studio/components/asset-source.tsx +0 -297
- package/src/studio/components/generate-button-input.tsx +0 -380
- package/src/studio/components/input-router.tsx +0 -41
- package/src/studio/files.ts +0 -114
- package/src/studio/plugin.tsx +0 -54
- package/src/studio/settings/schema.ts +0 -122
- package/src/studio/settings/tool.tsx +0 -587
- package/src/studio/settings-data.ts +0 -172
- package/src/utils/config.ts +0 -55
- package/src/utils/context-fields.ts +0 -172
- package/src/utils/document-paths.ts +0 -51
- package/src/utils/models.ts +0 -126
- package/src/utils/prompts.ts +0 -6
- package/src/utils/shared.ts +0 -88
|
@@ -0,0 +1,163 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
|
|
3
|
+
import { useSecrets } from "@sanity/studio-secrets";
|
|
4
|
+
import { Box, Button, Card, Flex, Select, Stack, Text, TextArea } from "@sanity/ui";
|
|
5
|
+
import { useEffect, useMemo, useRef, useState } from "react";
|
|
6
|
+
import { useClient, } from "sanity";
|
|
7
|
+
import { getDefaultAllowedAiImageModel, getSupportedAiImageModelOptions, } from "../../utils/models";
|
|
8
|
+
import { MAX_REFERENCE_IMAGES, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE, SOURCE_NAME, SOURCE_TITLE, } from "../../utils/shared";
|
|
9
|
+
import { getMissingSharedSecretMessage, getSharedSecretValue, SHARED_SECRET_HEADER_NAME, SHARED_SECRET_NAMESPACE, } from "../shared-secret";
|
|
10
|
+
import { composePrompt } from "../../utils/prompts";
|
|
11
|
+
import { buildSettingsReferenceImages, fetchSettingsDocument, findTargetSettings, getErrorMessage, resolveSettingsModel, } from "../settings-data";
|
|
12
|
+
import { convertImageFileToPng, createGeneratedFile, formatFileSize, } from "../files";
|
|
13
|
+
function getGenerationPhaseLabel(phase) {
|
|
14
|
+
switch (phase) {
|
|
15
|
+
case "preparing":
|
|
16
|
+
return "Preparing...";
|
|
17
|
+
case "generating":
|
|
18
|
+
return "Generating...";
|
|
19
|
+
case "uploading":
|
|
20
|
+
return "Uploading to Sanity...";
|
|
21
|
+
case "updating":
|
|
22
|
+
return "Updating field...";
|
|
23
|
+
default:
|
|
24
|
+
return "Generate with AI Image Plugin";
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
export function createAssetSource(options) {
|
|
28
|
+
const assetSourceTarget = options.assetSourceTarget;
|
|
29
|
+
function AssetSourceComponent({ onClose, onSelect, }) {
|
|
30
|
+
const client = useClient({ apiVersion: options.apiVersion });
|
|
31
|
+
const { loading: isLoadingSharedSecret, secrets } = useSecrets(SHARED_SECRET_NAMESPACE);
|
|
32
|
+
const [prompt, setPrompt] = useState("");
|
|
33
|
+
const [selectedModel, setSelectedModel] = useState(getDefaultAllowedAiImageModel(options.allowedModels));
|
|
34
|
+
const [referenceImages, setReferenceImages] = useState([]);
|
|
35
|
+
const [error, setError] = useState(null);
|
|
36
|
+
const [generationPhase, setGenerationPhase] = useState("idle");
|
|
37
|
+
const isGenerating = generationPhase !== "idle";
|
|
38
|
+
const sharedSecret = getSharedSecretValue(secrets);
|
|
39
|
+
const hasCustomSelectedModelRef = useRef(false);
|
|
40
|
+
const allowedModelOptions = useMemo(() => getSupportedAiImageModelOptions(options.allowedModels), [options.allowedModels]);
|
|
41
|
+
useEffect(() => {
|
|
42
|
+
let isMounted = true;
|
|
43
|
+
setSelectedModel(getDefaultAllowedAiImageModel(options.allowedModels));
|
|
44
|
+
hasCustomSelectedModelRef.current = false;
|
|
45
|
+
async function loadDefaultModel() {
|
|
46
|
+
try {
|
|
47
|
+
const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
|
|
48
|
+
if (!isMounted || hasCustomSelectedModelRef.current) {
|
|
49
|
+
return;
|
|
50
|
+
}
|
|
51
|
+
setSelectedModel(resolveSettingsModel({
|
|
52
|
+
allowedModels: options.allowedModels,
|
|
53
|
+
settings,
|
|
54
|
+
}));
|
|
55
|
+
}
|
|
56
|
+
catch {
|
|
57
|
+
// Keep the first allowed model selected if settings loading fails.
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
void loadDefaultModel();
|
|
61
|
+
return () => {
|
|
62
|
+
isMounted = false;
|
|
63
|
+
};
|
|
64
|
+
}, [client, options.allowedModels]);
|
|
65
|
+
function handleReferenceImageChange(event) {
|
|
66
|
+
const nextFiles = Array.from(event.currentTarget.files ?? []);
|
|
67
|
+
if (nextFiles.length === 0) {
|
|
68
|
+
return;
|
|
69
|
+
}
|
|
70
|
+
setReferenceImages((currentFiles) => [...currentFiles, ...nextFiles].slice(0, MAX_REFERENCE_IMAGES));
|
|
71
|
+
event.currentTarget.value = "";
|
|
72
|
+
}
|
|
73
|
+
function handleRemoveReferenceImage(indexToRemove) {
|
|
74
|
+
setReferenceImages((currentFiles) => currentFiles.filter((_, index) => index !== indexToRemove));
|
|
75
|
+
}
|
|
76
|
+
async function handleGenerate() {
|
|
77
|
+
setError(null);
|
|
78
|
+
setGenerationPhase("preparing");
|
|
79
|
+
try {
|
|
80
|
+
if (isLoadingSharedSecret) {
|
|
81
|
+
throw new Error("Loading AI Image Plugin shared secret...");
|
|
82
|
+
}
|
|
83
|
+
if (!sharedSecret) {
|
|
84
|
+
throw new Error(getMissingSharedSecretMessage());
|
|
85
|
+
}
|
|
86
|
+
const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
|
|
87
|
+
const targetSettings = assetSourceTarget
|
|
88
|
+
? findTargetSettings(settings, assetSourceTarget.id)
|
|
89
|
+
: null;
|
|
90
|
+
const composedPrompt = composePrompt([
|
|
91
|
+
settings?.globalPrompt
|
|
92
|
+
? `Global direction:\n${settings.globalPrompt}`
|
|
93
|
+
: null,
|
|
94
|
+
targetSettings?.prompt
|
|
95
|
+
? `Asset source direction:\n${targetSettings.prompt}`
|
|
96
|
+
: null,
|
|
97
|
+
prompt.trim()
|
|
98
|
+
? `Editor prompt:\n${prompt.trim()}`
|
|
99
|
+
: null,
|
|
100
|
+
]);
|
|
101
|
+
if (!composedPrompt) {
|
|
102
|
+
throw new Error("Add a prompt or configure a default prompt first.");
|
|
103
|
+
}
|
|
104
|
+
const formData = new FormData();
|
|
105
|
+
const normalizedReferenceImages = await Promise.all(referenceImages.map(convertImageFileToPng));
|
|
106
|
+
const settingsReferenceImages = await buildSettingsReferenceImages(settings, assetSourceTarget?.id);
|
|
107
|
+
const allReferenceImages = [
|
|
108
|
+
...settingsReferenceImages,
|
|
109
|
+
...normalizedReferenceImages,
|
|
110
|
+
].slice(0, MAX_REFERENCE_IMAGES);
|
|
111
|
+
formData.set("prompt", composedPrompt);
|
|
112
|
+
formData.set("model", selectedModel);
|
|
113
|
+
for (const referenceImage of allReferenceImages) {
|
|
114
|
+
formData.append("references", referenceImage);
|
|
115
|
+
}
|
|
116
|
+
setGenerationPhase("generating");
|
|
117
|
+
const response = await fetch(options.apiEndpoint, {
|
|
118
|
+
method: "POST",
|
|
119
|
+
body: formData,
|
|
120
|
+
headers: {
|
|
121
|
+
[SHARED_SECRET_HEADER_NAME]: sharedSecret,
|
|
122
|
+
},
|
|
123
|
+
});
|
|
124
|
+
const payload = (await response.json());
|
|
125
|
+
if (!response.ok) {
|
|
126
|
+
throw new Error("error" in payload && payload.error
|
|
127
|
+
? payload.error
|
|
128
|
+
: "AI image generation failed.");
|
|
129
|
+
}
|
|
130
|
+
if (!("data" in payload) ||
|
|
131
|
+
typeof payload.data !== "string" ||
|
|
132
|
+
typeof payload.mimeType !== "string") {
|
|
133
|
+
throw new Error("AI Image Plugin returned an invalid image payload.");
|
|
134
|
+
}
|
|
135
|
+
const generatedFile = createGeneratedFile(payload, prompt, "ai-image-plugin");
|
|
136
|
+
setGenerationPhase("uploading");
|
|
137
|
+
const uploadedAsset = await client.assets.upload("image", generatedFile, {
|
|
138
|
+
filename: generatedFile.name,
|
|
139
|
+
});
|
|
140
|
+
setGenerationPhase("updating");
|
|
141
|
+
onSelect([{ kind: "assetDocumentId", value: uploadedAsset._id }]);
|
|
142
|
+
onClose();
|
|
143
|
+
}
|
|
144
|
+
catch (nextError) {
|
|
145
|
+
setError(getErrorMessage(nextError));
|
|
146
|
+
}
|
|
147
|
+
finally {
|
|
148
|
+
setGenerationPhase("idle");
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
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) => {
|
|
152
|
+
hasCustomSelectedModelRef.current = true;
|
|
153
|
+
setSelectedModel(event.currentTarget.value);
|
|
154
|
+
}, 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 ||
|
|
155
|
+
"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, 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, onClick: () => void handleGenerate(), text: getGenerationPhaseLabel(generationPhase), tone: "primary" })] })] }) }));
|
|
156
|
+
}
|
|
157
|
+
return {
|
|
158
|
+
name: SOURCE_NAME,
|
|
159
|
+
title: SOURCE_TITLE,
|
|
160
|
+
component: AssetSourceComponent,
|
|
161
|
+
};
|
|
162
|
+
}
|
|
163
|
+
//# sourceMappingURL=asset-source.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"asset-source.js","sourceRoot":"","sources":["../../../src/studio/components/asset-source.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAA;;AAEZ,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAA;AACnD,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AACnF,OAAO,EAAoB,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAA;AAC9E,OAAO,EAGL,SAAS,GACV,MAAM,QAAQ,CAAA;AACf,OAAO,EACL,6BAA6B,EAC7B,+BAA+B,GAEhC,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,WAAW,EACX,YAAY,GAEb,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAEL,6BAA6B,EAC7B,oBAAoB,EACpB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,kBAAkB,CAAA;AACzB,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AACnD,OAAO,EACL,4BAA4B,EAC5B,qBAAqB,EACrB,kBAAkB,EAClB,eAAe,EACf,oBAAoB,GACrB,MAAM,kBAAkB,CAAA;AACzB,OAAO,EACL,qBAAqB,EACrB,mBAAmB,EACnB,cAAc,GACf,MAAM,UAAU,CAAA;AAUjB,SAAS,uBAAuB,CAAC,KAAsB;IACrD,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,WAAW;YACd,OAAO,cAAc,CAAA;QACvB,KAAK,YAAY;YACf,OAAO,eAAe,CAAA;QACxB,KAAK,WAAW;YACd,OAAO,wBAAwB,CAAA;QACjC,KAAK,UAAU;YACb,OAAO,mBAAmB,CAAA;QAC5B;YACE,OAAO,+BAA+B,CAAA;IAC1C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,OAAwB;IACxD,MAAM,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAA;IAEnD,SAAS,oBAAoB,CAAC,EAC5B,OAAO,EACP,QAAQ,GACkB;QAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAA;QAC5D,MAAM,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,GAC/C,UAAU,CAAuB,uBAAuB,CAAC,CAAA;QAC3D,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAA;QACxC,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CACrD,CAAA;QACD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAS,EAAE,CAAC,CAAA;QAClE,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAA;QACvD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAkB,MAAM,CAAC,CAAA;QAC/E,MAAM,YAAY,GAAG,eAAe,KAAK,MAAM,CAAA;QAC/C,MAAM,YAAY,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAA;QAClD,MAAM,yBAAyB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;QAC/C,MAAM,mBAAmB,GAAG,OAAO,CACjC,GAAG,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,aAAa,CAAC,EAC5D,CAAC,OAAO,CAAC,aAAa,CAAC,CACxB,CAAA;QAED,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,SAAS,GAAG,IAAI,CAAA;YAEpB,gBAAgB,CAAC,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAA;YACtE,yBAAyB,CAAC,OAAO,GAAG,KAAK,CAAA;YAEzC,KAAK,UAAU,gBAAgB;gBAC7B,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAA;oBAED,IAAI,CAAC,SAAS,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC;wBACpD,OAAM;oBACR,CAAC;oBAED,gBAAgB,CACd,oBAAoB,CAAC;wBACnB,aAAa,EAAE,OAAO,CAAC,aAAa;wBACpC,QAAQ;qBACT,CAAC,CACH,CAAA;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,mEAAmE;gBACrE,CAAC;YACH,CAAC;YAED,KAAK,gBAAgB,EAAE,CAAA;YAEvB,OAAO,GAAG,EAAE;gBACV,SAAS,GAAG,KAAK,CAAA;YACnB,CAAC,CAAA;QACH,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAA;QAEnC,SAAS,0BAA0B,CAAC,KAAoC;YACtE,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE,CAAC,CAAA;YAE7D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3B,OAAM;YACR,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,CAAA;YACD,KAAK,CAAC,aAAa,CAAC,KAAK,GAAG,EAAE,CAAA;QAChC,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,CAAA;QACH,CAAC;QAED,KAAK,UAAU,cAAc;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAA;YACd,kBAAkB,CAAC,WAAW,CAAC,CAAA;YAE/B,IAAI,CAAC;gBACH,IAAI,qBAAqB,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;gBAC7D,CAAC;gBAED,IAAI,CAAC,YAAY,EAAE,CAAC;oBAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,EAAE,CAAC,CAAA;gBAClD,CAAC;gBAED,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAA;gBACD,MAAM,cAAc,GAAG,iBAAiB;oBACtC,CAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE,iBAAiB,CAAC,EAAE,CAAC;oBACpD,CAAC,CAAC,IAAI,CAAA;gBACR,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;wBACX,CAAC,CAAC,mBAAmB,MAAM,CAAC,IAAI,EAAE,EAAE;wBACpC,CAAC,CAAC,IAAI;iBACT,CAAC,CAAA;gBAEF,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;gBACtE,CAAC;gBAED,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAA;gBAC/B,MAAM,yBAAyB,GAAG,MAAM,OAAO,CAAC,GAAG,CACjD,eAAe,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAC3C,CAAA;gBACD,MAAM,uBAAuB,GAAG,MAAM,4BAA4B,CAChE,QAAQ,EACR,iBAAiB,EAAE,EAAE,CACtB,CAAA;gBACD,MAAM,kBAAkB,GAAG;oBACzB,GAAG,uBAAuB;oBAC1B,GAAG,yBAAyB;iBAC7B,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,CAAC,CAAA;gBAEhC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAA;gBACtC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAA;gBAEpC,KAAK,MAAM,cAAc,IAAI,kBAAkB,EAAE,CAAC;oBAChD,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAA;gBAC/C,CAAC;gBAED,kBAAkB,CAAC,YAAY,CAAC,CAAA;gBAChC,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,CAAA;gBACF,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAEhB,CAAA;gBAEtB,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,CAAA;gBACH,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,CAAA;gBACvE,CAAC;gBAED,MAAM,aAAa,GAAG,mBAAmB,CACvC,OAAO,EACP,MAAM,EACN,iBAAiB,CAClB,CAAA;gBACD,kBAAkB,CAAC,WAAW,CAAC,CAAA;gBAC/B,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,aAAa,EAAE;oBACvE,QAAQ,EAAE,aAAa,CAAC,IAAI;iBAC7B,CAAC,CAAA;gBAEF,kBAAkB,CAAC,UAAU,CAAC,CAAA;gBAC9B,QAAQ,CAAC,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,aAAa,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA;gBACjE,OAAO,EAAE,CAAA;YACX,CAAC;YAAC,OAAO,SAAS,EAAE,CAAC;gBACnB,QAAQ,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC,CAAA;YACtC,CAAC;oBAAS,CAAC;gBACT,kBAAkB,CAAC,MAAM,CAAC,CAAA;YAC5B,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,CAAA;wCACxC,gBAAgB,CACd,KAAK,CAAC,aAAa,CAAC,KAAgC,CACrD,CAAA;oCACH,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,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,EAAE,YAAY,IAAI,qBAAqB,EAC/C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,cAAc,EAAE,EACpC,IAAI,EAAE,uBAAuB,CAAC,eAAe,CAAC,EAC9C,IAAI,EAAC,SAAS,GACd,IACG,IACD,GACH,CACR,CAAA;IACH,CAAC;IAED,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,oBAAoB;KAChC,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { type ImageValue, type ObjectInputProps } from "sanity";
|
|
2
|
+
import { type GenerateButtonTarget, type ResolvedOptions } from "../../utils/shared";
|
|
3
|
+
export declare function GenerateButtonInput({ options, props, target, }: {
|
|
4
|
+
options: ResolvedOptions;
|
|
5
|
+
props: ObjectInputProps<ImageValue>;
|
|
6
|
+
target: GenerateButtonTarget;
|
|
7
|
+
}): import("react/jsx-runtime").JSX.Element;
|
|
8
|
+
//# sourceMappingURL=generate-button-input.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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,CAAA;AAkBf,OAAO,EAGL,KAAK,oBAAoB,EACzB,KAAK,eAAe,EACrB,MAAM,oBAAoB,CAAA;AAiC3B,wBAAgB,mBAAmB,CAAC,EAClC,OAAO,EACP,KAAK,EACL,MAAM,GACP,EAAE;IACD,OAAO,EAAE,eAAe,CAAA;IACxB,KAAK,EAAE,gBAAgB,CAAC,UAAU,CAAC,CAAA;IACnC,MAAM,EAAE,oBAAoB,CAAA;CAC7B,2CAkYA"}
|
|
@@ -0,0 +1,186 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
|
|
3
|
+
import { Box, Button, Card, Dialog, Flex, Select, Stack, Text, TextArea, } from "@sanity/ui";
|
|
4
|
+
import { useSecrets } from "@sanity/studio-secrets";
|
|
5
|
+
import { useEffect, useMemo, useRef, useState } from "react";
|
|
6
|
+
import { PatchEvent, set, setIfMissing, useClient, useFormValue, useSchema, } from "sanity";
|
|
7
|
+
import { getMissingSharedSecretMessage, getSharedSecretValue, SHARED_SECRET_HEADER_NAME, SHARED_SECRET_NAMESPACE, } from "../shared-secret";
|
|
8
|
+
import { buildContextFieldPrompt, getDefaultSelectedContextFieldPaths, getSelectableContextFields, } from "../../utils/context-fields";
|
|
9
|
+
import { getDefaultAllowedAiImageModel, getSupportedAiImageModelOptions, } from "../../utils/models";
|
|
10
|
+
import { SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE, } from "../../utils/shared";
|
|
11
|
+
import { composePrompt } from "../../utils/prompts";
|
|
12
|
+
import { buildSettingsReferenceImages, fetchSettingsDocument, findTargetSettings, getErrorMessage, resolveSettingsModel, } from "../settings-data";
|
|
13
|
+
import { createGeneratedFile } from "../files";
|
|
14
|
+
function getGenerationPhaseLabel(phase) {
|
|
15
|
+
switch (phase) {
|
|
16
|
+
case "preparing":
|
|
17
|
+
return "Preparing...";
|
|
18
|
+
case "generating":
|
|
19
|
+
return "Generating...";
|
|
20
|
+
case "uploading":
|
|
21
|
+
return "Uploading to Sanity...";
|
|
22
|
+
case "updating":
|
|
23
|
+
return "Updating field...";
|
|
24
|
+
default:
|
|
25
|
+
return "Generate image";
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
export function GenerateButtonInput({ options, props, target, }) {
|
|
29
|
+
const client = useClient({ apiVersion: options.apiVersion });
|
|
30
|
+
const { loading: isLoadingSharedSecret, secrets } = useSecrets(SHARED_SECRET_NAMESPACE);
|
|
31
|
+
const schema = useSchema();
|
|
32
|
+
const documentValue = useFormValue([]);
|
|
33
|
+
const documentType = useFormValue(["_type"]);
|
|
34
|
+
const [isDialogOpen, setIsDialogOpen] = useState(false);
|
|
35
|
+
const [userPrompt, setUserPrompt] = useState("");
|
|
36
|
+
const [selectedModel, setSelectedModel] = useState(getDefaultAllowedAiImageModel(options.allowedModels));
|
|
37
|
+
const [selectedContextFieldPaths, setSelectedContextFieldPaths] = useState([]);
|
|
38
|
+
const [error, setError] = useState(null);
|
|
39
|
+
const [generationPhase, setGenerationPhase] = useState("idle");
|
|
40
|
+
const isGenerating = generationPhase !== "idle";
|
|
41
|
+
const currentAssetRef = props.value?.asset?._ref || "empty";
|
|
42
|
+
const hasCustomSelectedModelRef = useRef(false);
|
|
43
|
+
const selectableContextFields = useMemo(() => getSelectableContextFields(documentType ? schema.get(documentType) : undefined), [documentType, schema]);
|
|
44
|
+
const defaultSelectedContextFieldPaths = useMemo(() => getDefaultSelectedContextFieldPaths({
|
|
45
|
+
selectableContextFields,
|
|
46
|
+
suggestedContextFieldPaths: target.suggestedContextFieldPaths,
|
|
47
|
+
}), [selectableContextFields, target.suggestedContextFieldPaths]);
|
|
48
|
+
const runtimePrompt = buildContextFieldPrompt({
|
|
49
|
+
contextFieldPaths: selectedContextFieldPaths,
|
|
50
|
+
documentValue,
|
|
51
|
+
});
|
|
52
|
+
const sharedSecret = getSharedSecretValue(secrets);
|
|
53
|
+
const allowedModelOptions = useMemo(() => getSupportedAiImageModelOptions(options.allowedModels), [options.allowedModels]);
|
|
54
|
+
useEffect(() => {
|
|
55
|
+
if (!isDialogOpen) {
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
let isMounted = true;
|
|
59
|
+
setSelectedModel(getDefaultAllowedAiImageModel(options.allowedModels));
|
|
60
|
+
hasCustomSelectedModelRef.current = false;
|
|
61
|
+
async function loadDefaultModel() {
|
|
62
|
+
try {
|
|
63
|
+
const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
|
|
64
|
+
if (!isMounted || hasCustomSelectedModelRef.current) {
|
|
65
|
+
return;
|
|
66
|
+
}
|
|
67
|
+
setSelectedModel(resolveSettingsModel({
|
|
68
|
+
allowedModels: options.allowedModels,
|
|
69
|
+
settings,
|
|
70
|
+
}));
|
|
71
|
+
}
|
|
72
|
+
catch {
|
|
73
|
+
// Keep the first allowed model selected if settings loading fails.
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
void loadDefaultModel();
|
|
77
|
+
return () => {
|
|
78
|
+
isMounted = false;
|
|
79
|
+
};
|
|
80
|
+
}, [client, isDialogOpen, options.allowedModels]);
|
|
81
|
+
function toggleContextFieldPath(fieldPath) {
|
|
82
|
+
setSelectedContextFieldPaths((currentFieldPaths) => currentFieldPaths.includes(fieldPath)
|
|
83
|
+
? currentFieldPaths.filter((currentFieldPath) => currentFieldPath !== fieldPath)
|
|
84
|
+
: [...currentFieldPaths, fieldPath]);
|
|
85
|
+
}
|
|
86
|
+
async function handleGenerate() {
|
|
87
|
+
setError(null);
|
|
88
|
+
setGenerationPhase("preparing");
|
|
89
|
+
try {
|
|
90
|
+
if (isLoadingSharedSecret) {
|
|
91
|
+
throw new Error("Loading AI Image Plugin shared secret...");
|
|
92
|
+
}
|
|
93
|
+
if (!sharedSecret) {
|
|
94
|
+
throw new Error(getMissingSharedSecretMessage());
|
|
95
|
+
}
|
|
96
|
+
const settings = await fetchSettingsDocument(client, SETTINGS_DOCUMENT_ID, SETTINGS_SCHEMA_TYPE);
|
|
97
|
+
const targetSettings = findTargetSettings(settings, target.id);
|
|
98
|
+
const prompt = composePrompt([
|
|
99
|
+
settings?.globalPrompt
|
|
100
|
+
? `Global direction:\n${settings.globalPrompt}`
|
|
101
|
+
: null,
|
|
102
|
+
targetSettings?.prompt
|
|
103
|
+
? `Target direction:\n${targetSettings.prompt}`
|
|
104
|
+
: null,
|
|
105
|
+
runtimePrompt ? `Document context:\n${runtimePrompt}` : null,
|
|
106
|
+
userPrompt.trim()
|
|
107
|
+
? `Additional editor instructions:\n${userPrompt.trim()}`
|
|
108
|
+
: null,
|
|
109
|
+
]);
|
|
110
|
+
if (!prompt) {
|
|
111
|
+
throw new Error("Add a prompt or configure target defaults first.");
|
|
112
|
+
}
|
|
113
|
+
const referenceImages = await buildSettingsReferenceImages(settings, target.id);
|
|
114
|
+
const formData = new FormData();
|
|
115
|
+
formData.set("prompt", prompt);
|
|
116
|
+
formData.set("model", selectedModel);
|
|
117
|
+
for (const referenceImage of referenceImages) {
|
|
118
|
+
formData.append("references", referenceImage);
|
|
119
|
+
}
|
|
120
|
+
setGenerationPhase("generating");
|
|
121
|
+
const response = await fetch(options.apiEndpoint, {
|
|
122
|
+
method: "POST",
|
|
123
|
+
body: formData,
|
|
124
|
+
headers: {
|
|
125
|
+
[SHARED_SECRET_HEADER_NAME]: sharedSecret,
|
|
126
|
+
},
|
|
127
|
+
});
|
|
128
|
+
const payload = (await response.json());
|
|
129
|
+
if (!response.ok) {
|
|
130
|
+
throw new Error("error" in payload && payload.error
|
|
131
|
+
? payload.error
|
|
132
|
+
: "AI image generation failed.");
|
|
133
|
+
}
|
|
134
|
+
if (!("data" in payload) ||
|
|
135
|
+
typeof payload.data !== "string" ||
|
|
136
|
+
typeof payload.mimeType !== "string") {
|
|
137
|
+
throw new Error("AI Image Plugin returned an invalid image payload.");
|
|
138
|
+
}
|
|
139
|
+
const basename = String(documentValue?.title ||
|
|
140
|
+
documentValue?._type ||
|
|
141
|
+
target.title ||
|
|
142
|
+
"generated").trim() || "generated";
|
|
143
|
+
const generatedFile = createGeneratedFile(payload, basename, "ai-image-plugin-field");
|
|
144
|
+
setGenerationPhase("uploading");
|
|
145
|
+
const uploadedAsset = await client.assets.upload("image", generatedFile, {
|
|
146
|
+
filename: generatedFile.name,
|
|
147
|
+
});
|
|
148
|
+
setGenerationPhase("updating");
|
|
149
|
+
props.onChange(PatchEvent.from([
|
|
150
|
+
setIfMissing({ _type: "image" }),
|
|
151
|
+
set({ _type: "reference", _ref: uploadedAsset._id }, ["asset"]),
|
|
152
|
+
]));
|
|
153
|
+
setIsDialogOpen(false);
|
|
154
|
+
setUserPrompt("");
|
|
155
|
+
}
|
|
156
|
+
catch (nextError) {
|
|
157
|
+
setError(getErrorMessage(nextError, "Unable to generate an image for this field."));
|
|
158
|
+
}
|
|
159
|
+
finally {
|
|
160
|
+
setGenerationPhase("idle");
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
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 ||
|
|
164
|
+
"Generate an image with AI Image Plugin for this field." }), _jsx(Flex, { gap: 3, children: _jsx(Button, { onClick: () => {
|
|
165
|
+
setError(null);
|
|
166
|
+
setSelectedContextFieldPaths(defaultSelectedContextFieldPaths);
|
|
167
|
+
setIsDialogOpen(true);
|
|
168
|
+
}, 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: () => {
|
|
169
|
+
if (isGenerating) {
|
|
170
|
+
return;
|
|
171
|
+
}
|
|
172
|
+
setError(null);
|
|
173
|
+
setIsDialogOpen(false);
|
|
174
|
+
}, 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) => {
|
|
175
|
+
hasCustomSelectedModelRef.current = true;
|
|
176
|
+
setSelectedModel(event.currentTarget.value);
|
|
177
|
+
}, 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 ||
|
|
178
|
+
"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) => {
|
|
179
|
+
const isSelected = selectedContextFieldPaths.includes(field.path);
|
|
180
|
+
return (_jsx(Button, { mode: isSelected ? "default" : "ghost", onClick: () => toggleContextFieldPath(field.path), text: field.label, tone: isSelected ? "primary" : "default" }, field.path));
|
|
181
|
+
}) })] }) }) })) : (_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, 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: () => {
|
|
182
|
+
setError(null);
|
|
183
|
+
setIsDialogOpen(false);
|
|
184
|
+
}, text: "Cancel" }), _jsx(Button, { disabled: isGenerating || isLoadingSharedSecret, onClick: () => void handleGenerate(), text: getGenerationPhaseLabel(generationPhase), tone: "primary" })] })] }) }) })) : null] }));
|
|
185
|
+
}
|
|
186
|
+
//# sourceMappingURL=generate-button-input.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generate-button-input.js","sourceRoot":"","sources":["../../../src/studio/components/generate-button-input.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAA;;AAEZ,OAAO,EACL,GAAG,EACH,MAAM,EACN,IAAI,EACJ,MAAM,EACN,IAAI,EACJ,MAAM,EACN,KAAK,EACL,IAAI,EACJ,QAAQ,GACT,MAAM,YAAY,CAAA;AACnB,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAA;AACnD,OAAO,EAAoB,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAA;AAC9E,OAAO,EACL,UAAU,EAGV,GAAG,EACH,YAAY,EACZ,SAAS,EACT,YAAY,EACZ,SAAS,GACV,MAAM,QAAQ,CAAA;AACf,OAAO,EAEL,6BAA6B,EAC7B,oBAAoB,EACpB,yBAAyB,EACzB,uBAAuB,GACxB,MAAM,kBAAkB,CAAA;AACzB,OAAO,EACL,uBAAuB,EACvB,mCAAmC,EACnC,0BAA0B,GAC3B,MAAM,4BAA4B,CAAA;AACnC,OAAO,EACL,6BAA6B,EAC7B,+BAA+B,GAEhC,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EACL,oBAAoB,EACpB,oBAAoB,GAGrB,MAAM,oBAAoB,CAAA;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAA;AACnD,OAAO,EACL,4BAA4B,EAC5B,qBAAqB,EACrB,kBAAkB,EAClB,eAAe,EACf,oBAAoB,GACrB,MAAM,kBAAkB,CAAA;AACzB,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAA;AAS9C,SAAS,uBAAuB,CAAC,KAAsB;IACrD,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,WAAW;YACd,OAAO,cAAc,CAAA;QACvB,KAAK,YAAY;YACf,OAAO,eAAe,CAAA;QACxB,KAAK,WAAW;YACd,OAAO,wBAAwB,CAAA;QACjC,KAAK,UAAU;YACb,OAAO,mBAAmB,CAAA;QAC5B;YACE,OAAO,gBAAgB,CAAA;IAC3B,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,EAClC,OAAO,EACP,KAAK,EACL,MAAM,GAKP;IACC,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAA;IAC5D,MAAM,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,GAC/C,UAAU,CAAuB,uBAAuB,CAAC,CAAA;IAC3D,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,MAAM,aAAa,GAAG,YAAY,CAAC,EAAE,CAAwC,CAAA;IAC7E,MAAM,YAAY,GAAG,YAAY,CAAC,CAAC,OAAO,CAAC,CAAuB,CAAA;IAClE,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAA;IACvD,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAA;IAChD,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CACrD,CAAA;IACD,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,QAAQ,CAExE,EAAE,CAAC,CAAA;IACL,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAA;IACvD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAkB,MAAM,CAAC,CAAA;IAC/E,MAAM,YAAY,GAAG,eAAe,KAAK,MAAM,CAAA;IAC/C,MAAM,eAAe,GAAG,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,IAAI,OAAO,CAAA;IAC3D,MAAM,yBAAyB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAA;IAE/C,MAAM,uBAAuB,GAAG,OAAO,CACrC,GAAG,EAAE,CACH,0BAA0B,CACxB,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS,CACpD,EACH,CAAC,YAAY,EAAE,MAAM,CAAC,CACvB,CAAA;IACD,MAAM,gCAAgC,GAAG,OAAO,CAC9C,GAAG,EAAE,CACH,mCAAmC,CAAC;QAClC,uBAAuB;QACvB,0BAA0B,EAAE,MAAM,CAAC,0BAA0B;KAC9D,CAAC,EACJ,CAAC,uBAAuB,EAAE,MAAM,CAAC,0BAA0B,CAAC,CAC7D,CAAA;IACD,MAAM,aAAa,GAAG,uBAAuB,CAAC;QAC5C,iBAAiB,EAAE,yBAAyB;QAC5C,aAAa;KACd,CAAC,CAAA;IACF,MAAM,YAAY,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAA;IAClD,MAAM,mBAAmB,GAAG,OAAO,CACjC,GAAG,EAAE,CAAC,+BAA+B,CAAC,OAAO,CAAC,aAAa,CAAC,EAC5D,CAAC,OAAO,CAAC,aAAa,CAAC,CACxB,CAAA;IAED,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAM;QACR,CAAC;QAED,IAAI,SAAS,GAAG,IAAI,CAAA;QAEpB,gBAAgB,CAAC,6BAA6B,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAA;QACtE,yBAAyB,CAAC,OAAO,GAAG,KAAK,CAAA;QAEzC,KAAK,UAAU,gBAAgB;YAC7B,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAA;gBAED,IAAI,CAAC,SAAS,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC;oBACpD,OAAM;gBACR,CAAC;gBAED,gBAAgB,CACd,oBAAoB,CAAC;oBACnB,aAAa,EAAE,OAAO,CAAC,aAAa;oBACpC,QAAQ;iBACT,CAAC,CACH,CAAA;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,mEAAmE;YACrE,CAAC;QACH,CAAC;QAED,KAAK,gBAAgB,EAAE,CAAA;QAEvB,OAAO,GAAG,EAAE;YACV,SAAS,GAAG,KAAK,CAAA;QACnB,CAAC,CAAA;IACH,CAAC,EAAE,CAAC,MAAM,EAAE,YAAY,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC,CAAA;IAEjD,SAAS,sBAAsB,CAAC,SAAiB;QAC/C,4BAA4B,CAAC,CAAC,iBAAiB,EAAE,EAAE,CACjD,iBAAiB,CAAC,QAAQ,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,iBAAiB,CAAC,MAAM,CACtB,CAAC,gBAAgB,EAAE,EAAE,CAAC,gBAAgB,KAAK,SAAS,CACrD;YACH,CAAC,CAAC,CAAC,GAAG,iBAAiB,EAAE,SAAS,CAAC,CACtC,CAAA;IACH,CAAC;IAED,KAAK,UAAU,cAAc;QAC3B,QAAQ,CAAC,IAAI,CAAC,CAAA;QACd,kBAAkB,CAAC,WAAW,CAAC,CAAA;QAE/B,IAAI,CAAC;YACH,IAAI,qBAAqB,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAA;YAC7D,CAAC;YAED,IAAI,CAAC,YAAY,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,6BAA6B,EAAE,CAAC,CAAA;YAClD,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAC1C,MAAM,EACN,oBAAoB,EACpB,oBAAoB,CACrB,CAAA;YACD,MAAM,cAAc,GAAG,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,CAAA;YAC9D,MAAM,MAAM,GAAG,aAAa,CAAC;gBAC3B,QAAQ,EAAE,YAAY;oBACpB,CAAC,CAAC,sBAAsB,QAAQ,CAAC,YAAY,EAAE;oBAC/C,CAAC,CAAC,IAAI;gBACR,cAAc,EAAE,MAAM;oBACpB,CAAC,CAAC,sBAAsB,cAAc,CAAC,MAAM,EAAE;oBAC/C,CAAC,CAAC,IAAI;gBACR,aAAa,CAAC,CAAC,CAAC,sBAAsB,aAAa,EAAE,CAAC,CAAC,CAAC,IAAI;gBAC5D,UAAU,CAAC,IAAI,EAAE;oBACf,CAAC,CAAC,oCAAoC,UAAU,CAAC,IAAI,EAAE,EAAE;oBACzD,CAAC,CAAC,IAAI;aACT,CAAC,CAAA;YAEF,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAA;YACrE,CAAC;YAED,MAAM,eAAe,GAAG,MAAM,4BAA4B,CACxD,QAAQ,EACR,MAAM,CAAC,EAAE,CACV,CAAA;YACD,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAA;YAE/B,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;YAC9B,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAA;YAEpC,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE,CAAC;gBAC7C,QAAQ,CAAC,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAAA;YAC/C,CAAC;YAED,kBAAkB,CAAC,YAAY,CAAC,CAAA;YAChC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE;gBAChD,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE;oBACP,CAAC,yBAAyB,CAAC,EAAE,YAAY;iBAC1C;aACF,CAAC,CAAA;YACF,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAEhB,CAAA;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CACb,OAAO,IAAI,OAAO,IAAI,OAAO,CAAC,KAAK;oBACjC,CAAC,CAAC,OAAO,CAAC,KAAK;oBACf,CAAC,CAAC,6BAA6B,CAClC,CAAA;YACH,CAAC;YAED,IACE,CAAC,CAAC,MAAM,IAAI,OAAO,CAAC;gBACpB,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ;gBAChC,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,EACpC,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAA;YACvE,CAAC;YAED,MAAM,QAAQ,GACZ,MAAM,CACJ,aAAa,EAAE,KAAK;gBAClB,aAAa,EAAE,KAAK;gBACpB,MAAM,CAAC,KAAK;gBACZ,WAAW,CACd,CAAC,IAAI,EAAE,IAAI,WAAW,CAAA;YACzB,MAAM,aAAa,GAAG,mBAAmB,CACvC,OAAO,EACP,QAAQ,EACR,uBAAuB,CACxB,CAAA;YACD,kBAAkB,CAAC,WAAW,CAAC,CAAA;YAC/B,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,aAAa,EAAE;gBACvE,QAAQ,EAAE,aAAa,CAAC,IAAI;aAC7B,CAAC,CAAA;YAEF,kBAAkB,CAAC,UAAU,CAAC,CAAA;YAC9B,KAAK,CAAC,QAAQ,CACZ,UAAU,CAAC,IAAI,CAAC;gBACd,YAAY,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;gBAChC,GAAG,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC;aAChE,CAAC,CACH,CAAA;YACD,eAAe,CAAC,KAAK,CAAC,CAAA;YACtB,aAAa,CAAC,EAAE,CAAC,CAAA;QACnB,CAAC;QAAC,OAAO,SAAS,EAAE,CAAC;YACnB,QAAQ,CACN,eAAe,CAAC,SAAS,EAAE,6CAA6C,CAAC,CAC1E,CAAA;QACH,CAAC;gBAAS,CAAC;YACT,kBAAkB,CAAC,MAAM,CAAC,CAAA;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,CACL,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,aAAa,YAC7C,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,YACV,MAAM,CAAC,WAAW;gCACjB,wDAAwD,GACrD,EACP,KAAC,IAAI,IAAC,GAAG,EAAE,CAAC,YACV,KAAC,MAAM,IACL,OAAO,EAAE,GAAG,EAAE;oCACZ,QAAQ,CAAC,IAAI,CAAC,CAAA;oCACd,4BAA4B,CAAC,gCAAgC,CAAC,CAAA;oCAC9D,eAAe,CAAC,IAAI,CAAC,CAAA;gCACvB,CAAC,EACD,IAAI,EAAC,UAAU,EACf,IAAI,EAAC,SAAS,GACd,GACG,IACD,GACH,EAEP,KAAC,GAAG,cAAwB,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,IAA5C,eAAe,CAAoC,EAE5D,YAAY,CAAC,CAAC,CAAC,CACd,KAAC,MAAM,IACL,MAAM,EAAE,MAAM,CAAC,WAAW,IAAI,gBAAgB,EAC9C,EAAE,EAAE,GAAG,MAAM,CAAC,EAAE,kBAAkB,EAClC,OAAO,EAAE,GAAG,EAAE;oBACZ,IAAI,YAAY,EAAE,CAAC;wBACjB,OAAM;oBACR,CAAC;oBAED,QAAQ,CAAC,IAAI,CAAC,CAAA;oBACd,eAAe,CAAC,KAAK,CAAC,CAAA;gBACxB,CAAC,EACD,KAAK,EAAE,CAAC,YAER,KAAC,GAAG,IAAC,OAAO,EAAE,CAAC,YACb,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,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;gDAClB,yBAAyB,CAAC,OAAO,GAAG,IAAI,CAAA;gDACxC,gBAAgB,CACd,KAAK,CAAC,aAAa,CAAC,KAAgC,CACrD,CAAA;4CACH,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,MAAM,CAAC,WAAW,IAAI,eAAe,GACjC,EACP,KAAC,GAAG,IAAC,SAAS,EAAE,CAAC,YACf,KAAC,QAAQ,IACP,QAAQ,EAAE,CAAC,KAAuC,EAAE,EAAE,CACpD,aAAa,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,EAE1C,WAAW,EACT,MAAM,CAAC,iBAAiB;gDACxB,8CAA8C,EAEhD,IAAI,EAAE,CAAC,EACP,KAAK,EAAE,UAAU,GACjB,GACE,IACF,EAEL,uBAAuB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CACpC,KAAC,GAAG,cACF,KAAC,KAAK,IAAC,KAAK,EAAE,CAAC,YACb,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,wCAEvB,EACP,KAAC,IAAI,IAAC,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,YACtC,uBAAuB,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oDACrC,MAAM,UAAU,GAAG,yBAAyB,CAAC,QAAQ,CACnD,KAAK,CAAC,IAAI,CACX,CAAA;oDAED,OAAO,CACL,KAAC,MAAM,IAEL,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,EACtC,OAAO,EAAE,GAAG,EAAE,CAAC,sBAAsB,CAAC,KAAK,CAAC,IAAI,CAAC,EACjD,IAAI,EAAE,KAAK,CAAC,KAAK,EACjB,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,IAJnC,KAAK,CAAC,IAAI,CAKf,CACH,CAAA;gDACH,CAAC,CAAC,GACG,IACD,GACF,GACJ,CACP,CAAC,CAAC,CAAC,CACF,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,aAAa,YAC7C,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,iCAEvB,EACP,KAAC,IAAI,IAAC,KAAK,QAAC,IAAI,EAAE,CAAC,wFAGZ,IACD,GACH,CACR,EAEA,uBAAuB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CACpC,KAAC,IAAI,IAAC,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAC,aAAa,YAC7C,MAAC,KAAK,IAAC,KAAK,EAAE,CAAC,aACb,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAC,QAAQ,yCAEvB,EACN,aAAa,CAAC,CAAC,CAAC,CACf,KAAC,IAAI,IAAC,IAAI,EAAE,CAAC,YAAG,aAAa,GAAQ,CACtC,CAAC,CAAC,CAAC,CACF,KAAC,IAAI,IAAC,KAAK,QAAC,IAAI,EAAE,CAAC,yEAEZ,CACR,IACK,GACH,CACR,CAAC,CAAC,CAAC,IAAI,EAEP,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,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,IACL,QAAQ,EAAE,YAAY,EACtB,IAAI,EAAC,OAAO,EACZ,OAAO,EAAE,GAAG,EAAE;4CACZ,QAAQ,CAAC,IAAI,CAAC,CAAA;4CACd,eAAe,CAAC,KAAK,CAAC,CAAA;wCACxB,CAAC,EACD,IAAI,EAAC,QAAQ,GACb,EACF,KAAC,MAAM,IACL,QAAQ,EAAE,YAAY,IAAI,qBAAqB,EAC/C,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,cAAc,EAAE,EACpC,IAAI,EAAE,uBAAuB,CAAC,eAAe,CAAC,EAC9C,IAAI,EAAC,SAAS,GACd,IACG,IACD,GACJ,GACC,CACV,CAAC,CAAC,CAAC,IAAI,IACF,CACT,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { type InputProps } from "sanity";
|
|
2
|
+
import type { ResolvedOptions } from "../../utils/shared";
|
|
3
|
+
export declare function InputRouter({ options, props, }: {
|
|
4
|
+
options: ResolvedOptions;
|
|
5
|
+
props: InputProps;
|
|
6
|
+
}): import("react/jsx-runtime").JSX.Element;
|
|
7
|
+
//# sourceMappingURL=input-router.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"input-router.d.ts","sourceRoot":"","sources":["../../../src/studio/components/input-router.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAgB,KAAK,UAAU,EAAyB,MAAM,QAAQ,CAAA;AAI7E,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAA;AAEzD,wBAAgB,WAAW,CAAC,EAC1B,OAAO,EACP,KAAK,GACN,EAAE;IACD,OAAO,EAAE,eAAe,CAAA;IACxB,KAAK,EAAE,UAAU,CAAA;CAClB,2CA0BA"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import { jsx as _jsx } from "react/jsx-runtime";
|
|
3
|
+
import { useFormValue } from "sanity";
|
|
4
|
+
import { GenerateButtonInput } from "./generate-button-input";
|
|
5
|
+
import { doesTargetMatchField } from "../../utils/document-paths";
|
|
6
|
+
export function InputRouter({ options, props, }) {
|
|
7
|
+
const documentType = useFormValue(["_type"]);
|
|
8
|
+
if (props.schemaType?.name !== "image") {
|
|
9
|
+
return props.renderDefault(props);
|
|
10
|
+
}
|
|
11
|
+
const target = options.targets.find((candidate) => doesTargetMatchField({
|
|
12
|
+
documentType,
|
|
13
|
+
path: props.path,
|
|
14
|
+
target: candidate,
|
|
15
|
+
}));
|
|
16
|
+
if (!target) {
|
|
17
|
+
return props.renderDefault(props);
|
|
18
|
+
}
|
|
19
|
+
return (_jsx(GenerateButtonInput, { options: options, props: props, target: target }));
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=input-router.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"input-router.js","sourceRoot":"","sources":["../../../src/studio/components/input-router.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAA;;AAEZ,OAAO,EAAE,YAAY,EAA0C,MAAM,QAAQ,CAAA;AAE7E,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAA;AAC7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAA;AAGjE,MAAM,UAAU,WAAW,CAAC,EAC1B,OAAO,EACP,KAAK,GAIN;IACC,MAAM,YAAY,GAAG,YAAY,CAAC,CAAC,OAAO,CAAC,CAAuB,CAAA;IAElE,IAAI,KAAK,CAAC,UAAU,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;QACvC,OAAO,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAA;IACnC,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAChD,oBAAoB,CAAC;QACnB,YAAY;QACZ,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,SAAS;KAClB,CAAC,CACH,CAAA;IAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAA;IACnC,CAAC;IAED,OAAO,CACL,KAAC,mBAAmB,IAClB,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,KAAqC,EAC5C,MAAM,EAAE,MAAM,GACd,CACH,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export declare function convertImageFileToPng(file: File): Promise<File>;
|
|
2
|
+
export declare function base64ToBlob(base64: string, mimeType: string): Blob;
|
|
3
|
+
export declare function fileExtensionFromMimeType(mimeType: string): string;
|
|
4
|
+
export declare function formatFileSize(bytes: number): string;
|
|
5
|
+
export declare function createGeneratedFile(result: {
|
|
6
|
+
data: string;
|
|
7
|
+
mimeType: string;
|
|
8
|
+
}, basename: string, prefix?: string): File;
|
|
9
|
+
//# sourceMappingURL=files.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"files.d.ts","sourceRoot":"","sources":["../../src/studio/files.ts"],"names":[],"mappings":"AAOA,wBAAsB,qBAAqB,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAiDrE;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CASnE;AAED,wBAAgB,yBAAyB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CASlE;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAYpD;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,EAC1C,QAAQ,EAAE,MAAM,EAChB,MAAM,SAAoB,GACzB,IAAI,CAeN"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
function getPngFilename(filename) {
|
|
3
|
+
const basename = filename.replace(/\.[^/.]+$/, "");
|
|
4
|
+
return `${basename || "reference"}.png`;
|
|
5
|
+
}
|
|
6
|
+
export async function convertImageFileToPng(file) {
|
|
7
|
+
if (file.type === "image/png") {
|
|
8
|
+
return file;
|
|
9
|
+
}
|
|
10
|
+
const imageUrl = URL.createObjectURL(file);
|
|
11
|
+
try {
|
|
12
|
+
const image = await new Promise((resolve, reject) => {
|
|
13
|
+
const nextImage = new Image();
|
|
14
|
+
nextImage.onload = () => resolve(nextImage);
|
|
15
|
+
nextImage.onerror = () => reject(new Error(`Unable to decode ${file.name} as an image.`));
|
|
16
|
+
nextImage.src = imageUrl;
|
|
17
|
+
});
|
|
18
|
+
const canvas = document.createElement("canvas");
|
|
19
|
+
const width = image.naturalWidth || image.width;
|
|
20
|
+
const height = image.naturalHeight || image.height;
|
|
21
|
+
canvas.width = width;
|
|
22
|
+
canvas.height = height;
|
|
23
|
+
const context = canvas.getContext("2d");
|
|
24
|
+
if (!context) {
|
|
25
|
+
throw new Error("Unable to convert the reference image to PNG.");
|
|
26
|
+
}
|
|
27
|
+
context.drawImage(image, 0, 0, width, height);
|
|
28
|
+
const blob = await new Promise((resolve, reject) => {
|
|
29
|
+
canvas.toBlob((nextBlob) => {
|
|
30
|
+
if (nextBlob) {
|
|
31
|
+
resolve(nextBlob);
|
|
32
|
+
return;
|
|
33
|
+
}
|
|
34
|
+
reject(new Error("Unable to convert the reference image to PNG."));
|
|
35
|
+
}, "image/png", 1);
|
|
36
|
+
});
|
|
37
|
+
return new File([blob], getPngFilename(file.name), { type: "image/png" });
|
|
38
|
+
}
|
|
39
|
+
finally {
|
|
40
|
+
URL.revokeObjectURL(imageUrl);
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
export function base64ToBlob(base64, mimeType) {
|
|
44
|
+
const binaryString = atob(base64);
|
|
45
|
+
const bytes = new Uint8Array(binaryString.length);
|
|
46
|
+
for (const [index, character] of Array.from(binaryString).entries()) {
|
|
47
|
+
bytes[index] = character.charCodeAt(0);
|
|
48
|
+
}
|
|
49
|
+
return new Blob([bytes], { type: mimeType });
|
|
50
|
+
}
|
|
51
|
+
export function fileExtensionFromMimeType(mimeType) {
|
|
52
|
+
switch (mimeType) {
|
|
53
|
+
case "image/jpeg":
|
|
54
|
+
return "jpg";
|
|
55
|
+
case "image/webp":
|
|
56
|
+
return "webp";
|
|
57
|
+
default:
|
|
58
|
+
return "png";
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
export function formatFileSize(bytes) {
|
|
62
|
+
if (bytes < 1024) {
|
|
63
|
+
return `${bytes} B`;
|
|
64
|
+
}
|
|
65
|
+
const kilobytes = bytes / 1024;
|
|
66
|
+
if (kilobytes < 1024) {
|
|
67
|
+
return `${kilobytes.toFixed(1)} KB`;
|
|
68
|
+
}
|
|
69
|
+
return `${(kilobytes / 1024).toFixed(1)} MB`;
|
|
70
|
+
}
|
|
71
|
+
export function createGeneratedFile(result, basename, prefix = "ai-image-plugin") {
|
|
72
|
+
const extension = fileExtensionFromMimeType(result.mimeType);
|
|
73
|
+
const timestamp = new Date().toISOString().replaceAll(/[:.]/g, "-");
|
|
74
|
+
const safeBasename = basename
|
|
75
|
+
.trim()
|
|
76
|
+
.toLowerCase()
|
|
77
|
+
.replaceAll(/[^a-z0-9]+/g, "-")
|
|
78
|
+
.replaceAll(/^-+|-+$/g, "")
|
|
79
|
+
.slice(0, 40);
|
|
80
|
+
const suffix = safeBasename || "generated";
|
|
81
|
+
const blob = base64ToBlob(result.data, result.mimeType);
|
|
82
|
+
return new File([blob], `${prefix}-${suffix}-${timestamp}.${extension}`, {
|
|
83
|
+
type: result.mimeType,
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=files.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"files.js","sourceRoot":"","sources":["../../src/studio/files.ts"],"names":[],"mappings":"AAAA,YAAY,CAAA;AAEZ,SAAS,cAAc,CAAC,QAAgB;IACtC,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAA;IAClD,OAAO,GAAG,QAAQ,IAAI,WAAW,MAAM,CAAA;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,IAAU;IACpD,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAA;IACb,CAAC;IAED,MAAM,QAAQ,GAAG,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAA;IAE1C,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACpE,MAAM,SAAS,GAAG,IAAI,KAAK,EAAE,CAAA;YAC7B,SAAS,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;YAC3C,SAAS,CAAC,OAAO,GAAG,GAAG,EAAE,CACvB,MAAM,CAAC,IAAI,KAAK,CAAC,oBAAoB,IAAI,CAAC,IAAI,eAAe,CAAC,CAAC,CAAA;YACjE,SAAS,CAAC,GAAG,GAAG,QAAQ,CAAA;QAC1B,CAAC,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAA;QAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,KAAK,CAAA;QAC/C,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,IAAI,KAAK,CAAC,MAAM,CAAA;QAElD,MAAM,CAAC,KAAK,GAAG,KAAK,CAAA;QACpB,MAAM,CAAC,MAAM,GAAG,MAAM,CAAA;QAEtB,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QAEvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAA;QAClE,CAAC;QAED,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAA;QAE7C,MAAM,IAAI,GAAG,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACvD,MAAM,CAAC,MAAM,CACX,CAAC,QAAQ,EAAE,EAAE;gBACX,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,CAAC,QAAQ,CAAC,CAAA;oBACjB,OAAM;gBACR,CAAC;gBAED,MAAM,CAAC,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC,CAAA;YACpE,CAAC,EACD,WAAW,EACX,CAAC,CACF,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,OAAO,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAA;IAC3E,CAAC;YAAS,CAAC;QACT,GAAG,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAA;IAC/B,CAAC;AACH,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,QAAgB;IAC3D,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,CAAA;IACjC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;IAEjD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;QACpE,KAAK,CAAC,KAAK,CAAC,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAA;IACxC,CAAC;IAED,OAAO,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAA;AAC9C,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAgB;IACxD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,YAAY;YACf,OAAO,KAAK,CAAA;QACd,KAAK,YAAY;YACf,OAAO,MAAM,CAAA;QACf;YACE,OAAO,KAAK,CAAA;IAChB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,KAAa;IAC1C,IAAI,KAAK,GAAG,IAAI,EAAE,CAAC;QACjB,OAAO,GAAG,KAAK,IAAI,CAAA;IACrB,CAAC;IAED,MAAM,SAAS,GAAG,KAAK,GAAG,IAAI,CAAA;IAE9B,IAAI,SAAS,GAAG,IAAI,EAAE,CAAC;QACrB,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAA;IACrC,CAAC;IAED,OAAO,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAA;AAC9C,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAA0C,EAC1C,QAAgB,EAChB,MAAM,GAAG,iBAAiB;IAE1B,MAAM,SAAS,GAAG,yBAAyB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IAC5D,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAA;IACnE,MAAM,YAAY,GAAG,QAAQ;SAC1B,IAAI,EAAE;SACN,WAAW,EAAE;SACb,UAAU,CAAC,aAAa,EAAE,GAAG,CAAC;SAC9B,UAAU,CAAC,UAAU,EAAE,EAAE,CAAC;SAC1B,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;IACf,MAAM,MAAM,GAAG,YAAY,IAAI,WAAW,CAAA;IAC1C,MAAM,IAAI,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;IAEvD,OAAO,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,MAAM,IAAI,MAAM,IAAI,SAAS,IAAI,SAAS,EAAE,EAAE;QACvE,IAAI,EAAE,MAAM,CAAC,QAAQ;KACtB,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin.d.ts","sourceRoot":"","sources":["../../src/studio/plugin.tsx"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAA;AAEpD,eAAO,MAAM,aAAa,wCA6CxB,CAAA"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
import { jsx as _jsx } from "react/jsx-runtime";
|
|
2
|
+
import { definePlugin } from "sanity";
|
|
3
|
+
import { createAssetSource } from "./components/asset-source";
|
|
4
|
+
import { InputRouter } from "./components/input-router";
|
|
5
|
+
import { createSettingsSchema } from "./settings/schema";
|
|
6
|
+
import { createSettingsTool } from "./settings/tool";
|
|
7
|
+
import { normalizeOptions } from "../utils/config";
|
|
8
|
+
export const aiImagePlugin = definePlugin((rawOptions) => {
|
|
9
|
+
const options = normalizeOptions(rawOptions);
|
|
10
|
+
const assetSource = options.assetSourceTarget
|
|
11
|
+
? createAssetSource(options)
|
|
12
|
+
: null;
|
|
13
|
+
const settingsTool = createSettingsTool(options);
|
|
14
|
+
const settingsSchema = createSettingsSchema(options);
|
|
15
|
+
return {
|
|
16
|
+
name: "ai-image-plugin",
|
|
17
|
+
...(options.registerSettingsSchemaType
|
|
18
|
+
? {
|
|
19
|
+
schema: {
|
|
20
|
+
types: [settingsSchema],
|
|
21
|
+
},
|
|
22
|
+
}
|
|
23
|
+
: {}),
|
|
24
|
+
tools: [
|
|
25
|
+
{
|
|
26
|
+
name: options.settingsToolName,
|
|
27
|
+
title: options.settingsToolTitle,
|
|
28
|
+
component: settingsTool,
|
|
29
|
+
},
|
|
30
|
+
],
|
|
31
|
+
form: {
|
|
32
|
+
...(assetSource
|
|
33
|
+
? {
|
|
34
|
+
image: {
|
|
35
|
+
assetSources: (previousAssetSources) => previousAssetSources.some((previousAssetSource) => previousAssetSource.name === assetSource.name)
|
|
36
|
+
? previousAssetSources
|
|
37
|
+
: [...previousAssetSources, assetSource],
|
|
38
|
+
},
|
|
39
|
+
}
|
|
40
|
+
: {}),
|
|
41
|
+
components: {
|
|
42
|
+
input: (props) => (_jsx(InputRouter, { options: options, props: props })),
|
|
43
|
+
},
|
|
44
|
+
},
|
|
45
|
+
};
|
|
46
|
+
});
|
|
47
|
+
//# sourceMappingURL=plugin.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../../src/studio/plugin.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,YAAY,EAAqC,MAAM,QAAQ,CAAA;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAA;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAA;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAA;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAA;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAA;AAGlD,MAAM,CAAC,MAAM,aAAa,GAAG,YAAY,CAAgB,CAAC,UAAU,EAAE,EAAE;IACtE,MAAM,OAAO,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAA;IAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,iBAAiB;QAC3C,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC5B,CAAC,CAAC,IAAI,CAAA;IACR,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAA;IAChD,MAAM,cAAc,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAA;IAEpD,OAAO;QACL,IAAI,EAAE,iBAAiB;QACvB,GAAG,CAAC,OAAO,CAAC,0BAA0B;YACpC,CAAC,CAAC;gBACE,MAAM,EAAE;oBACN,KAAK,EAAE,CAAC,cAAc,CAAC;iBACxB;aACF;YACH,CAAC,CAAC,EAAE,CAAC;QACP,KAAK,EAAE;YACL;gBACE,IAAI,EAAE,OAAO,CAAC,gBAAgB;gBAC9B,KAAK,EAAE,OAAO,CAAC,iBAAiB;gBAChC,SAAS,EAAE,YAAY;aACxB;SACF;QACD,IAAI,EAAE;YACJ,GAAG,CAAC,WAAW;gBACb,CAAC,CAAC;oBACE,KAAK,EAAE;wBACL,YAAY,EAAE,CAAC,oBAAmC,EAAE,EAAE,CACpD,oBAAoB,CAAC,IAAI,CACvB,CAAC,mBAAmB,EAAE,EAAE,CACtB,mBAAmB,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,CAChD;4BACC,CAAC,CAAC,oBAAoB;4BACtB,CAAC,CAAC,CAAC,GAAG,oBAAoB,EAAE,WAAW,CAAC;qBAC7C;iBACF;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,UAAU,EAAE;gBACV,KAAK,EAAE,CAAC,KAAiB,EAAE,EAAE,CAAC,CAC5B,KAAC,WAAW,IAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,GAAI,CAChD;aACF;SACF;KACF,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { type ResolvedOptions } from "../../utils/shared";
|
|
2
|
+
export declare function createSettingsSchema(options: ResolvedOptions): {
|
|
3
|
+
type: "document";
|
|
4
|
+
name: "aiImagePluginSettings";
|
|
5
|
+
} & Omit<import("sanity").DocumentDefinition, "preview"> & {
|
|
6
|
+
preview?: import("sanity").PreviewConfig<Record<string, string>, Record<never, any>> | undefined;
|
|
7
|
+
};
|
|
8
|
+
//# sourceMappingURL=schema.d.ts.map
|