@braze/sanity-plugin-ai-translation 0.2.79 → 0.2.81

package/dist/index.js

@@ -1,6 +1,6 @@
 import { getPublishedId, pathToString, useClient, useValidationStatus, defineType, defineField, definePlugin } from "sanity";
 import { jsx, jsxs, Fragment } from "react/jsx-runtime";
-import { WarningOutlineIcon, ErrorOutlineIcon, CheckmarkIcon, CloseIcon, PanelRightIcon, SyncIcon, SearchIcon, CircleIcon, ArrowRightIcon, ResetIcon, TranslateIcon } from "@sanity/icons";
+import { WarningOutlineIcon, ErrorOutlineIcon, CheckmarkIcon, CloseIcon, SyncIcon, PanelRightIcon, SearchIcon, CircleIcon, ArrowRightIcon, ResetIcon, TranslateIcon } from "@sanity/icons";
 import { createContext, useReducer, useRef, useEffect, useContext, useMemo, useState, useCallback } from "react";
 import defaults from "lodash.defaults";
 import { Card, Stack, Flex, Text, Box, Heading, Button, Spinner, Badge, Inline, Radio, Checkbox, Grid, TextInput, Select, Label, Code, Tooltip } from "@sanity/ui";
@@ -385,32 +385,63 @@ const isLoggingEnabled = () => process.env.NEXT_PUBLIC_SANITY_STUDIO_AI_TRANSLAT
   return devLog.debug("Source references with unpublished changes", {
     count: results.length
   }), results;
-}, getExistingTranslationId = (translationMetadata, targetLanguage) => translationMetadata?.translations?.length && translationMetadata.translations.find(
+}, TRANSLATION_METADATA_TYPE = "translation.metadata", getExistingTranslationId = (translationMetadata, targetLanguage) => translationMetadata?.translations?.length && translationMetadata.translations.find(
   (translation) => translation._key === targetLanguage
 )?.value._ref || null, checkTargetDrafts = async ({
   client,
   translationMetadata,
   selectedTargetLanguages,
-  sourceDocumentId
+  sourceDocumentId,
+  references
 }) => {
-  if (!translationMetadata?.translations?.length)
-    return devLog.debug("No translation metadata \u2014 skipping draft check", {
-      sourceDocumentId
-    }), [];
   const targetEntries = [];
-  for (const lang of selectedTargetLanguages) {
-    const targetId = getExistingTranslationId(translationMetadata, lang);
-    targetId && targetEntries.push({ targetId, targetLanguage: lang });
+  if (translationMetadata?.translations?.length)
+    for (const lang of selectedTargetLanguages) {
+      const targetId = getExistingTranslationId(translationMetadata, lang);
+      targetId && targetEntries.push({ targetId, targetLanguage: lang, sourceDocumentId, isReference: !1 });
+    }
+  else
+    devLog.debug("No translation metadata \u2014 skipping main document draft check", {
+      sourceDocumentId
+    });
+  if (references && references.length > 0) {
+    const refIds = references.map((r) => r._id), refMetadataDocs = await client.fetch(
+      "*[_type == $metadataType && array::intersects(translations[].value._ref, $refIds)] { translations }",
+      { metadataType: TRANSLATION_METADATA_TYPE, refIds }
+    );
+    devLog.debug("Fetched reference metadata for draft check", {
+      refCount: refIds.length,
+      metadataDocsFound: refMetadataDocs.length
+    });
+    const refIdSet = new Set(refIds);
+    for (const metadataDoc of refMetadataDocs) {
+      if (!metadataDoc.translations?.length) continue;
+      const matchingRefId = metadataDoc.translations.find(
+        (t) => t.value?._ref && refIdSet.has(t.value._ref)
+      )?.value._ref;
+      if (matchingRefId)
+        for (const lang of selectedTargetLanguages) {
+          const targetId = getExistingTranslationId(metadataDoc, lang);
+          targetId && targetEntries.push({
+            targetId,
+            targetLanguage: lang,
+            sourceDocumentId: matchingRefId,
+            isReference: !0
+          });
+        }
+    }
   }
-  if (targetEntries.length === 0)
+  const seen = /* @__PURE__ */ new Set(), deduped = targetEntries.filter((e) => seen.has(e.targetId) ? !1 : (seen.add(e.targetId), !0));
+  if (deduped.length === 0)
     return devLog.debug("No existing translations found for selected languages", {
       sourceDocumentId,
       selectedTargetLanguages
     }), [];
-  const draftIds = targetEntries.map((e) => `drafts.${e.targetId}`);
+  const draftIds = deduped.map((e) => `drafts.${e.targetId}`);
   devLog.debug("Checking for stale drafts on target documents", {
     sourceDocumentId,
-    draftIdsToCheck: draftIds.length
+    draftIdsToCheck: draftIds.length,
+    referenceTargets: deduped.filter((e) => e.isReference).length
   });
   const drafts = await client.fetch(
     "*[_id in $draftIds]{ _id, _type, title, name }",
@@ -419,23 +450,25 @@ const isLoggingEnabled = () => process.env.NEXT_PUBLIC_SANITY_STUDIO_AI_TRANSLAT
   if (drafts.length === 0)
     return devLog.debug("No stale drafts found", { sourceDocumentId }), [];
   const draftMap = new Map(drafts.map((d) => [d._id, d])), results = [];
-  for (const entry of targetEntries) {
+  for (const entry of deduped) {
     const draftId = `drafts.${entry.targetId}`, draft = draftMap.get(draftId);
     draft && results.push({
       targetId: entry.targetId,
       draftId,
       _type: draft._type,
       title: draft.title || draft.name,
-      sourceDocumentId,
-      targetLanguage: entry.targetLanguage
+      sourceDocumentId: entry.sourceDocumentId,
+      targetLanguage: entry.targetLanguage,
+      isReference: entry.isReference || void 0
     });
   }
   return devLog.info("Stale drafts detected on target documents", {
     sourceDocumentId,
     staleDraftCount: results.length,
+    referenceDraftCount: results.filter((r) => r.isReference).length,
     affectedLanguages: results.map((r) => r.targetLanguage)
   }), results;
-}, TRANSLATION_METADATA_TYPE = "translation.metadata", getTranslationMetadata = async ({
+}, getTranslationMetadata = async ({
   client,
   sourceDocument,
   sourceLanguage
@@ -3379,15 +3412,16 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     }
   );
 }, VALIDATION_TIMEOUT_MS = 2e3, DocumentValidator = ({ documentId, documentType, onResult }) => {
-  const publishedId = getPublishedId(documentId), { isValidating, validation } = useValidationStatus(publishedId, documentType), hasReceivedRealResult = useRef(!1), [timedOut, setTimedOut] = useState(!1);
+  const publishedId = getPublishedId(documentId), { isValidating, validation } = useValidationStatus(publishedId, documentType), hasReceivedRealResult = useRef(!1), hasSettled = useRef(!1), [timedOut, setTimedOut] = useState(!1);
   return useEffect(() => {
     const timer = setTimeout(() => setTimedOut(!0), VALIDATION_TIMEOUT_MS);
     return () => clearTimeout(timer);
   }, []), useEffect(() => {
+    if (hasSettled.current) return;
     (isValidating || validation.length > 0) && (hasReceivedRealResult.current = !0);
-    const stillWaitingForFirstResult = !hasReceivedRealResult.current && !timedOut, errors = validation.filter((v) => v.level === "error");
-    onResult(documentId, {
-      isValidating: isValidating || stillWaitingForFirstResult,
+    const stillWaitingForFirstResult = !hasReceivedRealResult.current && !timedOut, errors = validation.filter((v) => v.level === "error"), computedIsValidating = isValidating || stillWaitingForFirstResult;
+    !computedIsValidating && hasReceivedRealResult.current && (hasSettled.current = !0), onResult(documentId, {
+      isValidating: computedIsValidating,
       errors
     });
   }, [isValidating, validation, documentId, onResult, timedOut]), null;
@@ -3404,7 +3438,10 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     translationMode,
     releaseConfig,
     documentSource
-  } = useTranslationContext(), client = useClient({ apiVersion: "vX" }), [validationResults, setValidationResults] = useState(/* @__PURE__ */ new Map()), [sourceRefsWithDrafts, setSourceRefsWithDrafts] = useState([]), [isSourceDraftCheckLoading, setIsSourceDraftCheckLoading] = useState(!0), [staleDrafts, setStaleDrafts] = useState([]), [isDraftCheckLoading, setIsDraftCheckLoading] = useState(!0), [draftAction, setDraftAction] = useState("none"), [isDraftActionInProgress, setIsDraftActionInProgress] = useState(!1), [conflictContext, setConflictContext] = useState(null), [isCheckingConflicts, setIsCheckingConflicts] = useState(!0), filteredReferences = references.filter((ref) => selectedReferenceIds.has(ref._id)), documentsToValidate = useMemo(() => {
+  } = useTranslationContext(), client = useClient({ apiVersion: "vX" }), [scanKey, setScanKey] = useState(0), [validationResults, setValidationResults] = useState(/* @__PURE__ */ new Map()), [sourceRefsWithDrafts, setSourceRefsWithDrafts] = useState([]), [isSourceDraftCheckLoading, setIsSourceDraftCheckLoading] = useState(!0), [staleDrafts, setStaleDrafts] = useState([]), [isDraftCheckLoading, setIsDraftCheckLoading] = useState(!0), [draftAction, setDraftAction] = useState("none"), [isDraftActionInProgress, setIsDraftActionInProgress] = useState(!1), [conflictContext, setConflictContext] = useState(null), [isCheckingConflicts, setIsCheckingConflicts] = useState(!0), filteredReferences = useMemo(
+    () => references.filter((ref) => selectedReferenceIds.has(ref._id)),
+    [references, selectedReferenceIds]
+  ), documentsToValidate = useMemo(() => {
     const docs = [
       {
         _id: document._id,
@@ -3418,6 +3455,9 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     return docs;
   }, [document, references, selectedReferenceIds, shouldTranslateReferences]), handleValidationResult = useCallback((id, result) => {
     setValidationResults((prev) => {
+      const existing = prev.get(id);
+      if (existing && existing.isValidating === result.isValidating && existing.errors.length === result.errors.length)
+        return prev;
       const next = new Map(prev);
       return next.set(id, result), next;
     });
@@ -3460,7 +3500,8 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
         client,
         translationMetadata,
         selectedTargetLanguages,
-        sourceDocumentId: document._id
+        sourceDocumentId: document._id,
+        references: shouldTranslateReferences ? filteredReferences : void 0
       });
       setStaleDrafts(drafts), drafts.length === 0 && setDraftAction("none");
     } catch (error) {
@@ -3471,7 +3512,7 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     } finally {
       setIsDraftCheckLoading(!1);
     }
-  }, [client, translationMetadata, selectedTargetLanguages, document._id]), runConflictCheck = useCallback(async () => {
+  }, [client, translationMetadata, selectedTargetLanguages, document._id, shouldTranslateReferences, filteredReferences]), runConflictCheck = useCallback(async () => {
     if (releaseConfig.mode === "new") {
       setIsCheckingConflicts(!1), setConflictContext(null);
       return;
@@ -3488,10 +3529,10 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
       });
       foundConflicts.length > 0 ? setConflictContext({ releaseId: checkedReleaseId, conflicts: foundConflicts }) : setConflictContext(null), setIsCheckingConflicts(!1);
     }
-  }, [releaseConfig, client, document, selectedTargetLanguages, config.languageFieldName, filteredReferences]), hasRunChecks = useRef(!1);
+  }, [releaseConfig, client, document, selectedTargetLanguages, config.languageFieldName, filteredReferences]);
   useEffect(() => {
-    hasRunChecks.current || (hasRunChecks.current = !0, runSourceDraftCheck(), runDraftCheck(), runConflictCheck());
-  }, [runSourceDraftCheck, runDraftCheck, runConflictCheck]);
+    runSourceDraftCheck(), runDraftCheck(), runConflictCheck();
+  }, [scanKey]);
   const handleDiscardDrafts = useCallback(async () => {
     setIsDraftActionInProgress(!0);
     try {
@@ -3584,7 +3625,9 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     conflictContext || startTranslation(!1);
   }, [conflictContext, startTranslation]), handleBack = useCallback(() => {
     dispatch({ type: "SET_OVERALL_STATUS", payload: "idle" });
-  }, [dispatch]), sourceRefDraftIds = useMemo(
+  }, [dispatch]), handleRescan = useCallback(() => {
+    setValidationResults(/* @__PURE__ */ new Map()), setSourceRefsWithDrafts([]), setIsSourceDraftCheckLoading(!0), setStaleDrafts([]), setIsDraftCheckLoading(!0), setDraftAction("none"), setConflictContext(null), setIsCheckingConflicts(!0), setScanKey((k) => k + 1);
+  }, []), sourceRefDraftIds = useMemo(
     () => new Set(sourceRefsWithDrafts.map((d) => d.publishedId)),
     [sourceRefsWithDrafts]
   ), isLoading = someStillValidating || isSourceDraftCheckLoading || isDraftCheckLoading || isCheckingConflicts, hasStaleDrafts = staleDrafts.length > 0, draftsDismissed = draftAction === "continue", canStartTranslation = !isLoading && !hasValidationErrors && !conflictContext && (!hasStaleDrafts || draftsDismissed), handleCancelConflict = useCallback(() => {
@@ -3611,7 +3654,7 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
         documentType: doc._type,
         onResult: handleValidationResult
       },
-      doc._id
+      `${doc._id}-${scanKey}`
     )),
     /* @__PURE__ */ jsx(Text, { weight: "semibold", size: 2, children: "Pre-flight Checks" }),
     /* @__PURE__ */ jsxs(Text, { size: 1, muted: !0, children: [
@@ -3687,7 +3730,8 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
         /* @__PURE__ */ jsx(Card, { padding: 3, border: !0, radius: 2, children: /* @__PURE__ */ jsx(Box, { style: { maxHeight: "300px", overflowY: "auto" }, children: /* @__PURE__ */ jsx(Stack, { space: 3, children: staleDrafts.map((draft) => /* @__PURE__ */ jsxs(Flex, { align: "center", justify: "space-between", gap: 2, children: [
           /* @__PURE__ */ jsxs(Flex, { align: "center", gap: 2, flex: 1, style: { minWidth: 0 }, children: [
             /* @__PURE__ */ jsx(Text, { size: 1, children: draft.title || draft.targetId }),
-            /* @__PURE__ */ jsx(Badge, { tone: "primary", fontSize: 0, children: draft.targetLanguage })
+            /* @__PURE__ */ jsx(Badge, { tone: "primary", fontSize: 0, children: draft.targetLanguage }),
+            draft.isReference && /* @__PURE__ */ jsx(Badge, { fontSize: 0, children: "reference" })
           ] }),
           /* @__PURE__ */ jsx(DocumentOpenerButton, { documentId: draft.targetId, documentType: draft._type })
         ] }, draft.draftId)) }) }) }),
@@ -3731,16 +3775,28 @@ ${protectedPhrases.join(", ")}`, buildStyleGuideSection = (globalPrompt, localeP
     ] }) }),
     /* @__PURE__ */ jsxs(Flex, { justify: "space-between", gap: 2, children: [
       /* @__PURE__ */ jsx(Button, { mode: "ghost", text: "Back", onClick: handleBack }),
-      /* @__PURE__ */ jsx(
-        Button,
-        {
-          mode: "ghost",
-          tone: "positive",
-          text: isLoading ? "Checking\u2026" : "Start Translation",
-          onClick: handleStartTranslation,
-          disabled: !canStartTranslation
-        }
-      )
+      /* @__PURE__ */ jsxs(Flex, { gap: 2, children: [
+        /* @__PURE__ */ jsx(
+          Button,
+          {
+            mode: "ghost",
+            text: "Rescan",
+            icon: SyncIcon,
+            onClick: handleRescan,
+            disabled: isLoading
+          }
+        ),
+        /* @__PURE__ */ jsx(
+          Button,
+          {
+            mode: "ghost",
+            tone: "positive",
+            text: isLoading ? "Checking\u2026" : "Start Translation",
+            onClick: handleStartTranslation,
+            disabled: !canStartTranslation
+          }
+        )
+      ] })
     ] })
   ] });
 }, ActionButtonBar = ({ setIsDialogOpen }) => {