@sanity/assist 1.2.15-lang.2 → 1.2.15-lang.3

package/dist/index.js

@@ -984,6 +984,7 @@ function useTranslate(apiClient) {
     let {
       documentId,
       languagePath,
+      translatePath,
       fieldLanguageMap
     } = _ref4;
     setLoading(true);
@@ -995,6 +996,7 @@ function useTranslate(apiClient) {
         types,
         languagePath,
         fieldLanguageMap,
+        translatePath: translatePath.length === 0 ? documentRootKey : sanity.pathToString(translatePath),
         userId: user == null ? void 0 : user.id
       }
     }).catch(e => {
@@ -3329,7 +3331,7 @@ const defaultLanguageOutputs = function (member, enclosingType, translateFromLan
   }
   return void 0;
 };
-function getTranslationMap(documentSchema, documentMembers, translateFromLanguageId, outputLanguageIds, langFn) {
+function getFieldLanguageMap(documentSchema, documentMembers, translateFromLanguageId, outputLanguageIds, langFn) {
   var _a, _b, _c;
   const translationMaps = [];
   for (const member of documentMembers) {
@@ -4561,13 +4563,12 @@ const getLanguageParams = (select, document) => {
 const FieldTranslationContext = react.createContext({
   openFieldTranslation: () => {},
   translationLoading: false
-  //loadLanguages: () => {},
 });
 function useFieldTranslation() {
   return react.useContext(FieldTranslationContext);
 }
 function FieldTranslationProvider(props) {
-  var _a, _b;
+  var _a, _b, _c;
   const {
     config: assistConfig
   } = useAiAssistanceConfig();
@@ -4577,40 +4578,42 @@ function FieldTranslationProvider(props) {
     translate: runTranslate
   } = useTranslate(apiClient);
   const [dialogOpen, setDialogOpen] = react.useState(false);
-  const [document, setDocument] = react.useState();
-  const [documentSchema, setDocumentSchema] = react.useState();
+  const [fieldTranslationParams, setFieldTranslationParams] = react.useState();
   const [languages, setLanguages] = react.useState();
   const [fromLanguage, setFromLanguage] = react.useState(void 0);
   const [toLanguages, setToLanguages] = react.useState(void 0);
-  const [translationMap, setTranslationMap] = react.useState();
+  const [fieldLanguageMaps, setFieldLanguageMaps] = react.useState();
   const close = react.useCallback(() => {
     setDialogOpen(false);
     setLanguages(void 0);
-    setDocument(void 0);
-    setDocument(void 0);
+    setFieldTranslationParams(void 0);
   }, []);
   const languageClient = sanity.useClient({
     apiVersion: (_b = config == null ? void 0 : config.apiVersion) != null ? _b : "2022-11-27"
   });
-  const documentId = document == null ? void 0 : document._id;
+  const documentId = (_c = fieldTranslationParams == null ? void 0 : fieldTranslationParams.document) == null ? void 0 : _c._id;
   const id = react.useId();
-  const selectFromLanguage = react.useCallback((from, languages2, document2, documentSchema2) => {
+  const selectFromLanguage = react.useCallback((from, languages2, params) => {
     var _a2, _b2;
+    const {
+      document,
+      documentSchema
+    } = params != null ? params : {};
     setFromLanguage(from);
-    if (!document2 || !documentSchema2 || !languages2) {
-      setTranslationMap(void 0);
+    if (!document || !documentSchema || !params || !languages2) {
+      setFieldLanguageMaps(void 0);
       return;
     }
     const to = languages2.filter(l => l.id !== (from == null ? void 0 : from.id));
     setToLanguages(to);
     const fromId = from == null ? void 0 : from.id;
     const toIds = (_a2 = to == null ? void 0 : to.map(l => l.id)) != null ? _a2 : [];
-    const docMembers = getDocumentMembersFlat(document2, documentSchema2);
+    const docMembers = getDocumentMembersFlat(document, documentSchema);
     if (fromId && (toIds == null ? void 0 : toIds.length)) {
-      const transMap = getTranslationMap(documentSchema2, docMembers, fromId, toIds, (_b2 = config == null ? void 0 : config.translationOutputs) != null ? _b2 : defaultLanguageOutputs);
-      setTranslationMap(transMap);
+      const transMap = getFieldLanguageMap(documentSchema, docMembers, fromId, toIds, (_b2 = config == null ? void 0 : config.translationOutputs) != null ? _b2 : defaultLanguageOutputs);
+      setFieldLanguageMaps(transMap);
     } else {
-      setTranslationMap(void 0);
+      setFieldLanguageMaps(void 0);
     }
   }, [config]);
   const toggleToLanguage = react.useCallback((toggledLang, toLanguages2, languages2) => {
@@ -4621,20 +4624,15 @@ function FieldTranslationProvider(props) {
     const newToLangs = languages2.filter(anyLang => !!(toLanguages2 == null ? void 0 : toLanguages2.find(selectedLang => toggledLang.id !== selectedLang.id && selectedLang.id === anyLang.id)) || toggledLang.id === anyLang.id && !wasSelected);
     setToLanguages(newToLangs);
   }, []);
-  const openFieldTranslation = react.useCallback(async _ref11 => {
-    let {
-      document: document2,
-      documentSchema: documentSchema2
-    } = _ref11;
+  const openFieldTranslation = react.useCallback(async params => {
     setDialogOpen(true);
-    const languageParams = getLanguageParams(config == null ? void 0 : config.selectLanguageParams, document2);
+    const languageParams = getLanguageParams(config == null ? void 0 : config.selectLanguageParams, params.document);
     const languages2 = await (typeof (config == null ? void 0 : config.languages) === "function" ? config == null ? void 0 : config.languages(languageClient, languageParams) : Promise.resolve(config == null ? void 0 : config.languages));
     setLanguages(languages2);
-    setDocument(document2);
-    setDocumentSchema(documentSchema2);
+    setFieldTranslationParams(params);
     const fromLanguage2 = languages2 == null ? void 0 : languages2[0];
     if (fromLanguage2) {
-      selectFromLanguage(fromLanguage2, languages2, document2, documentSchema2);
+      selectFromLanguage(fromLanguage2, languages2, params);
     } else {
       console.error("No languages available for selected language params", languageParams);
     }
@@ -4645,12 +4643,14 @@ function FieldTranslationProvider(props) {
       translationLoading: false
     };
   }, [openFieldTranslation]);
-  const runDisabled = !fromLanguage || !(toLanguages == null ? void 0 : toLanguages.length) || !(translationMap == null ? void 0 : translationMap.length) || !documentId;
+  const runDisabled = !fromLanguage || !(toLanguages == null ? void 0 : toLanguages.length) || !(fieldLanguageMaps == null ? void 0 : fieldLanguageMaps.length) || !documentId;
   const onRunTranslation = react.useCallback(() => {
-    if (translationMap && documentId) {
+    const translatePath = fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath;
+    if (fieldLanguageMaps && documentId && translatePath) {
       runTranslate({
         documentId,
-        fieldLanguageMap: translationMap.map(map => ({
+        translatePath: fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath,
+        fieldLanguageMap: fieldLanguageMaps.map(map => ({
           ...map,
           // eslint-disable-next-line max-nested-callbacks
           outputs: map.outputs.filter(out => !!(toLanguages == null ? void 0 : toLanguages.find(l => l.id === out.id)))
@@ -4658,7 +4658,7 @@ function FieldTranslationProvider(props) {
       });
     }
     close();
-  }, [translationMap, documentId, runTranslate, close, toLanguages]);
+  }, [fieldLanguageMaps, documentId, runTranslate, close, toLanguages, fieldTranslationParams == null ? void 0 : fieldTranslationParams.translatePath]);
   const runButton = /* @__PURE__ */jsxRuntime.jsx(ui.Button, {
     text: "Translate",
     tone: "primary",
@@ -4717,7 +4717,7 @@ function FieldTranslationProvider(props) {
                 name: "fromLang",
                 value: l.id,
                 checked: l.id === (fromLanguage == null ? void 0 : fromLanguage.id),
-                onClick: () => selectFromLanguage(l, languages, document, documentSchema)
+                onClick: () => selectFromLanguage(l, languages, fieldTranslationParams)
               }), /* @__PURE__ */jsxRuntime.jsx(ui.Text, {
                 children: (_a2 = l.title) != null ? _a2 : l.id
               })]
@@ -4741,7 +4741,7 @@ function FieldTranslationProvider(props) {
                 value: l.id,
                 checked: !!(toLanguages == null ? void 0 : toLanguages.find(tl => tl.id === l.id)),
                 onClick: () => toggleToLanguage(l, toLanguages, languages),
-                disabled: !(translationMap == null ? void 0 : translationMap.find(tm => tm.outputs.find(o => o.id === l.id)))
+                disabled: !(fieldLanguageMaps == null ? void 0 : fieldLanguageMaps.find(tm => tm.outputs.find(o => o.id === l.id)))
               }), /* @__PURE__ */jsxRuntime.jsx(ui.Text, {
                 children: (_a2 = l.title) != null ? _a2 : l.id
               })]
@@ -4806,7 +4806,10 @@ function ErrorWrapper(props) {
   const catchError = react.useCallback(params => {
     setError(params.error);
   }, [setError]);
-  const unsetValue = react.useCallback(() => onChange(sanity.PatchEvent.from(sanity.unset())), [onChange]);
+  const unsetValue = react.useCallback(() => {
+    onChange(sanity.PatchEvent.from(sanity.unset()));
+    setError(void 0);
+  }, [onChange]);
   const dismiss = react.useCallback(() => setError(void 0), []);
   const catcher = /* @__PURE__ */jsxRuntime.jsx(ui.ErrorBoundary, {
     onCatch: catchError,
@@ -4938,11 +4941,11 @@ function InstructionInput(props) {
     })]
   });
 }
-function ObjectMember(_ref12) {
+function ObjectMember(_ref11) {
   let {
     fieldName,
     ...props
-  } = _ref12;
+  } = _ref11;
   const member = findFieldMember(props.members, fieldName);
   return member ? /* @__PURE__ */jsxRuntime.jsx(sanity.ObjectInputMember, {
     ...props,
@@ -5081,8 +5084,8 @@ function IconInput(props) {
     onChange
   } = props;
   const id = react.useId();
-  const items = react.useMemo(() => Object.entries(icons.icons).map(_ref13 => {
-    let [key, icon] = _ref13;
+  const items = react.useMemo(() => Object.entries(icons.icons).map(_ref12 => {
+    let [key, icon] = _ref12;
     return /* @__PURE__ */jsxRuntime.jsx(IconItem, {
       iconKey: key,
       icon,
@@ -5110,12 +5113,12 @@ function IconInput(props) {
     }
   });
 }
-function IconItem(_ref14) {
+function IconItem(_ref13) {
   let {
     icon,
     iconKey: key,
     onChange
-  } = _ref14;
+  } = _ref13;
   const onClick = react.useCallback(() => onChange(sanity.set(key)), [onChange, key]);
   return /* @__PURE__ */jsxRuntime.jsx(ui.MenuItem, {
     icon,
@@ -5126,8 +5129,8 @@ function IconItem(_ref14) {
 }
 function getIcon(iconName) {
   var _a, _b;
-  return (_b = (_a = Object.entries(icons.icons).find(_ref15 => {
-    let [key] = _ref15;
+  return (_b = (_a = Object.entries(icons.icons).find(_ref14 => {
+    let [key] = _ref14;
     return key === iconName;
   })) == null ? void 0 : _a[1]) != null ? _b : icons.icons.sparkles;
 }
@@ -5286,11 +5289,11 @@ const contextDocumentSchema = sanity.defineType({
       title: "title",
       context: "context"
     },
-    prepare(_ref16) {
+    prepare(_ref15) {
       let {
         title,
         context
-      } = _ref16;
+      } = _ref15;
       var _a;
       const text = context == null ? void 0 : context.flatMap(block => block == null ? void 0 : block.children).flatMap(child => {
         var _a2;
@@ -5390,11 +5393,11 @@ function InstructionOutputField(props) {
     children: props.children
   });
 }
-function EnabledOutputField(_ref17) {
+function EnabledOutputField(_ref16) {
   let {
     fieldSchema,
     ...props
-  } = _ref17;
+  } = _ref16;
   var _a;
   const [open, setOpen] = react.useState(!!((_a = props.value) == null ? void 0 : _a.length));
   const onExpand = react.useCallback(() => setOpen(true), []);
@@ -5442,11 +5445,11 @@ function useEmptySelectAllValue(value, allowedValues, onChange) {
     }
   }, [allowedValues, value, onChange]);
 }
-function ObjectOutputInput(_ref18) {
+function ObjectOutputInput(_ref17) {
   let {
     fieldSchema,
     ...props
-  } = _ref18;
+  } = _ref17;
   const {
     value,
     onChange
@@ -5493,11 +5496,11 @@ function ObjectOutputInput(_ref18) {
     })
   });
 }
-function ArrayOutputInput(_ref19) {
+function ArrayOutputInput(_ref18) {
   let {
     fieldSchema,
     ...props
-  } = _ref19;
+  } = _ref18;
   const {
     value,
     onChange
@@ -5542,13 +5545,13 @@ function ArrayOutputInput(_ref19) {
     })
   });
 }
-function Selectable(_ref20) {
+function Selectable(_ref19) {
   let {
     title,
     arrayValue,
     value,
     onChange
-  } = _ref20;
+  } = _ref19;
   const checked = !(arrayValue == null ? void 0 : arrayValue.length) || !!(arrayValue == null ? void 0 : arrayValue.find(v => v._key === value));
   const handleChange = react.useCallback(() => onChange(checked, value), [onChange, checked, value]);
   return /* @__PURE__ */jsxRuntime.jsxs(ui.Flex, {
@@ -5613,10 +5616,10 @@ const fieldReference = sanity.defineType({
     select: {
       path: "path"
     },
-    prepare(_ref21) {
+    prepare(_ref20) {
       let {
         path
-      } = _ref21;
+      } = _ref20;
       return {
         title: path,
         path,
@@ -5761,12 +5764,12 @@ const instruction = sanity.defineType({
       title: "title",
       userId: "userId"
     },
-    prepare: _ref22 => {
+    prepare: _ref21 => {
       let {
         icon,
         title,
         userId
-      } = _ref22;
+      } = _ref21;
       return {
         title,
         icon: icon ? icons.icons[icon] : icons.SparklesIcon,
@@ -6218,21 +6221,25 @@ function node$1(node2) {
 const translateActions = {
   name: "sanity-assist-translate",
   useAction(props) {
-    var _a, _b, _c, _d, _e, _f, _g;
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const {
       config
     } = useAiAssistanceConfig();
     const apiClient = useApiClient(config == null ? void 0 : config.__customApiClient);
-    const isDocumentLevel = props.path.length === 0;
     const {
-      schemaType,
+      schemaType: fieldSchemaType,
+      path,
       documentId,
+      documentSchemaType,
       documentIsAssistable
     } = props;
-    const fieldTransEnabled = (_c = (_b = (_a = config.translate) == null ? void 0 : _a.field) == null ? void 0 : _b.documentTypes) == null ? void 0 : _c.includes(schemaType.name);
-    const docTransTypes = (_e = (_d = config.translate) == null ? void 0 : _d.document) == null ? void 0 : _e.documentTypes;
-    const documentTranslationEnabled = isDocumentLevel && (!docTransTypes && isAssistSupported(schemaType) || (docTransTypes == null ? void 0 : docTransTypes.includes(schemaType.name)));
-    if (documentTranslationEnabled || fieldTransEnabled) {
+    const isDocumentLevel = path.length === 0;
+    const docTransTypes = (_b = (_a = config.translate) == null ? void 0 : _a.document) == null ? void 0 : _b.documentTypes;
+    const options = fieldSchemaType == null ? void 0 : fieldSchemaType.options;
+    const addFieldAction = isDocumentLevel || ((_c = options == null ? void 0 : options.aiWritingAssistance) == null ? void 0 : _c.translateAction);
+    const fieldTransEnabled = addFieldAction && documentSchemaType && ((_f = (_e = (_d = config.translate) == null ? void 0 : _d.field) == null ? void 0 : _e.documentTypes) == null ? void 0 : _f.includes(documentSchemaType.name));
+    const documentTranslationEnabled = addFieldAction && documentSchemaType && (!docTransTypes && isAssistSupported(fieldSchemaType) || (docTransTypes == null ? void 0 : docTransTypes.includes(documentSchemaType.name)));
+    if (documentSchemaType && (documentTranslationEnabled || fieldTransEnabled)) {
       const {
         value: documentValue,
         onChange: documentOnChange
@@ -6245,32 +6252,45 @@ const translateActions = {
         task: translationApi.translate
       });
       docRef.current = documentValue;
-      const languagePath = (_g = (_f = config.translate) == null ? void 0 : _f.document) == null ? void 0 : _g.languageField;
-      const translateDocumentAction = react.useMemo(() => languagePath && documentTranslationEnabled ? node$1({
-        type: "action",
-        icon: translationApi.loading ? () => /* @__PURE__ */jsxRuntime.jsx(ui.Box, {
-          style: {
-            height: 17
-          },
-          children: /* @__PURE__ */jsxRuntime.jsx(ui.Spinner, {
+      const languagePath = (_h = (_g = config.translate) == null ? void 0 : _g.document) == null ? void 0 : _h.languageField;
+      const translateDocumentAction = react.useMemo(() => {
+        var _a2;
+        if (!languagePath || !documentTranslationEnabled) {
+          return void 0;
+        }
+        const {
+          value: languageId
+        } = (_a2 = mutator.extractWithPath(languagePath, documentValue)[0]) != null ? _a2 : {};
+        const title = path.length ? "Translate" : "Translate document";
+        return node$1({
+          type: "action",
+          icon: translationApi.loading ? () => /* @__PURE__ */jsxRuntime.jsx(ui.Box, {
             style: {
-              transform: "translateY(6px)"
+              height: 17
+            },
+            children: /* @__PURE__ */jsxRuntime.jsx(ui.Spinner, {
+              style: {
+                transform: "translateY(6px)"
+              }
+            })
+          }) : icons.TranslateIcon,
+          title,
+          onAction: () => {
+            if (translationApi.loading || !languagePath || !documentId) {
+              return;
             }
-          })
-        }) : icons.TranslateIcon,
-        title: "Translate document",
-        onAction: () => {
-          if (translationApi.loading || !languagePath || !documentId) {
-            return;
-          }
-          translate({
-            languagePath,
-            documentId: documentId != null ? documentId : ""
-          });
-        },
-        renderAsButton: true,
-        disabled: translationApi.loading
-      }) : void 0, [languagePath, translate, documentId, translationApi.loading, documentTranslationEnabled]);
+            translate({
+              languagePath,
+              translatePath: path,
+              documentId: documentId != null ? documentId : ""
+            });
+          },
+          renderAsButton: true,
+          disabled: translationApi.loading || !languageId ? {
+            reason: "Language is not set for this document"
+          } : void 0
+        });
+      }, [languagePath, translate, documentId, translationApi.loading, documentTranslationEnabled, path]);
       const fieldTranslate = useFieldTranslation();
       const openFieldTranslation = useDraftDelayedTask({
         documentOnChange,
@@ -6289,24 +6309,25 @@ const translateActions = {
             }
           })
         }) : icons.TranslateIcon,
-        title: "Translate fields",
+        title: "Translate fields...",
         onAction: () => {
           if (fieldTranslate.translationLoading || !documentId) {
             return;
           }
           openFieldTranslation({
             document: docRef.current,
-            documentSchema: schemaType
+            documentSchema: documentSchemaType,
+            translatePath: path
           });
         },
         renderAsButton: true,
         disabled: fieldTranslate.translationLoading
-      }) : void 0, [openFieldTranslation, schemaType, documentId, fieldTranslate.translationLoading, fieldTransEnabled]);
+      }) : void 0, [openFieldTranslation, documentSchemaType, documentId, fieldTranslate.translationLoading, fieldTransEnabled, path]);
       return react.useMemo(() => {
         return node$1({
           type: "group",
           icon: () => null,
-          title: "Translate",
+          title: "Translation",
           children: [translateDocumentAction, translateFieldsAction].filter(c => !!c),
           expanded: true
         });
@@ -6374,7 +6395,8 @@ const assistFieldActions = {
     const translateAction = translateActions.useAction(sanity.typed({
       ...props,
       documentId: assistableDocumentId,
-      documentIsAssistable
+      documentIsAssistable,
+      documentSchemaType
     }));
     const manageInstructions = react.useCallback(() => isSelected ? closeInspector(aiInspectorId) : openInspector(aiInspectorId, {
       [fieldPathParam]: pathKey,
@@ -6429,7 +6451,7 @@ const assistFieldActions = {
       type: "group",
       icon: icons.SparklesIcon,
       title: pluginTitleShort,
-      children: [runInstructionsGroup, translateAction, assistSupported && manageInstructionsItem].filter(c => !!c),
+      children: [runInstructionsGroup, translateAction, assistSupported && manageInstructionsItem].filter(c => !!c).filter(c => c.type === "group" ? c.children.length : true),
       expanded: false,
       renderAsButton: true,
       hidden: !assistSupported && !imageCaptionAction && !translateAction
@@ -6558,18 +6580,34 @@ function AssistDocumentInputWrapper(props) {
     documentId
   });
 }
-function AssistDocumentInput(_ref23) {
+function AssistDocumentInput(_ref22) {
   let {
     documentId,
     ...props
-  } = _ref23;
+  } = _ref22;
   useInstructionToaster(documentId, props.schemaType);
+  const schemaType = react.useMemo(() => {
+    if (props.schemaType.name !== assistDocumentTypeName) {
+      return props.schemaType;
+    }
+    return {
+      ...props.schemaType,
+      type: {
+        ...props.schemaType.type,
+        // compatability with i18nArrays plugin that requires this to be document
+        name: "document"
+      }
+    };
+  }, [props.schemaType]);
   return /* @__PURE__ */jsxRuntime.jsx(FirstAssistedPathProvider, {
     members: props.members,
     children: /* @__PURE__ */jsxRuntime.jsx(AssistDocumentContextProvider, {
-      schemaType: props.schemaType,
+      schemaType,
       documentId,
-      children: props.renderDefault(props)
+      children: props.renderDefault({
+        ...props,
+        schemaType
+      })
     })
   });
 }
@@ -6645,29 +6683,36 @@ const assist = sanity.definePlugin(config => {
     },
     document: {
       inspectors: (prev, context) => {
-        const docSchema = context.schema.get(context.documentType);
+        const documentType = context.documentType;
+        const docSchema = context.schema.get(documentType);
         if (docSchema && isSchemaAssistEnabled(docSchema)) {
           return [...prev, assistInspector];
         }
         return prev;
       },
-      unstable_fieldActions: (prev, _ref24) => {
+      unstable_fieldActions: (prev, _ref23) => {
         let {
           documentType,
           schema
-        } = _ref24;
+        } = _ref23;
+        if (documentType === assistDocumentTypeName) {
+          return [];
+        }
         const docSchema = schema.get(documentType);
         if (docSchema && isSchemaAssistEnabled(docSchema)) {
           return [...prev, assistFieldActions];
         }
         return prev;
       },
-      unstable_languageFilter: (prev, _ref25) => {
+      unstable_languageFilter: (prev, _ref24) => {
         let {
           documentId,
           schema,
           schemaType
-        } = _ref25;
+        } = _ref24;
+        if (schemaType === assistDocumentTypeName) {
+          return [];
+        }
         const docSchema = schema.get(schemaType);
         if (docSchema && sanity.isObjectSchemaType(docSchema) && isSchemaAssistEnabled(docSchema)) {
           return [...prev, createAssistDocumentPresence(documentId, docSchema)];