gt-sanity 1.0.10 → 1.1.0

package/dist/index.mjs

@@ -652,13 +652,17 @@ async function initProject(uploadResult, options, secrets) {
     const { setupJobId } = setupResult, start = Date.now(), pollInterval = 2e3;
     let setupCompleted = !1, setupFailedMessage = null;
     for (; ; ) {
-      const status = await gt.checkSetupStatus(setupJobId);
-      if (status.status === "completed") {
+      const status = await gt.checkJobStatus([setupJobId]);
+      if (!status[0]) {
+        setupFailedMessage = "Unknown error";
+        break;
+      }
+      if (status[0].status === "completed") {
         setupCompleted = !0;
         break;
       }
-      if (status.status === "failed") {
-        setupFailedMessage = status.error?.message || "Unknown error";
+      if (status[0].status === "failed") {
+        setupFailedMessage = status[0].error?.message || "Unknown error";
         break;
       }
       if (Date.now() - start > setupTimeoutMs) {
@@ -680,22 +684,20 @@ async function createJobs(uploadResult, localeIds, secrets) {
 async function downloadTranslations(files, secrets, maxRetries = 3, retryDelay = 1e3) {
   overrideConfig(secrets);
   let retries = 0;
-  const fileIds = files.map((file) => file.translationId), map = new Map(files.map((file) => [file.translationId, file])), result = [];
   for (; retries <= maxRetries; )
     try {
-      const downloadedFiles = (await gt.downloadFileBatch(fileIds)).files || [];
-      for (const file of downloadedFiles) {
-        const documentData = map.get(file.id);
-        documentData && result.push({
-          docData: documentData,
-          data: file.data
-        });
-      }
-      return result;
+      return (await gt.downloadFileBatch(
+        files.map((file) => ({
+          fileId: file.fileId,
+          branchId: file.branchId,
+          versionId: file.versionId,
+          locale: file.locale
+        }))
+      )).files || [];
     } catch {
       retries++, await new Promise((resolve) => setTimeout(resolve, retryDelay));
     }
-  return result;
+  return [];
 }
 async function checkTranslationStatus(fileQueryData, downloadStatus, secrets) {
   overrideConfig(secrets);
@@ -703,8 +705,10 @@ async function checkTranslationStatus(fileQueryData, downloadStatus, secrets) {
     const currentQueryData = fileQueryData.filter(
       (item) => !downloadStatus.downloaded.has(`${item.fileId}:${item.locale}`) && !downloadStatus.failed.has(`${item.fileId}:${item.locale}`) && !downloadStatus.skipped.has(`${item.fileId}:${item.locale}`)
     );
-    return currentQueryData.length === 0 ? !0 : ((await gt.checkFileTranslations(currentQueryData)).translations || []).filter(
-      (translation) => translation.isReady && translation.fileId
+    return currentQueryData.length === 0 ? !0 : ((await gt.queryFileData({
+      translatedFiles: currentQueryData
+    })).translatedFiles || []).filter(
+      (translation) => translation.completedAt
     );
   } catch (error) {
     return console.error("Error checking translation status", error), [];
@@ -1022,20 +1026,15 @@ async function processImportBatch(items, options = {}) {
     successfulImports
   };
 }
-async function getReadyFilesForImport(documents, translationStatuses, options = {}) {
+async function getReadyFilesForImport(translationStatuses, options = {}) {
   const { filterReadyFiles = () => !0 } = options, readyFiles = [];
   for (const [key, status] of translationStatuses.entries())
-    if (status.isReady && status.translationId && filterReadyFiles(key, status)) {
-      const [documentId, locale] = key.split(":"), document2 = documents.find(
-        (doc) => (doc._id?.replace("drafts.", "") || doc._id) === documentId
-      );
-      document2 && readyFiles.push({
-        documentId,
-        versionId: document2._rev,
-        translationId: status.translationId,
-        locale
-      });
-    }
+    status.isReady && filterReadyFiles(key, status) && readyFiles.push({
+      fileId: status.fileData.fileId,
+      versionId: status.fileData.versionId,
+      branchId: status.fileData.branchId,
+      locale: status.fileData.locale
+    });
   return readyFiles;
 }
 async function importTranslations(readyFiles, secrets, translationContext, options = {}) {
@@ -1043,13 +1042,13 @@ async function importTranslations(readyFiles, secrets, translationContext, optio
     return { successCount: 0, failureCount: 0, successfulImports: [] };
   const importItems = (await downloadTranslations(readyFiles, secrets)).map((file) => ({
     docInfo: {
-      documentId: file.docData.documentId,
-      versionId: file.docData.versionId
+      documentId: file.fileId,
+      versionId: file.versionId
     },
-    locale: file.docData.locale,
+    locale: file.locale,
     data: file.data,
     translationContext,
-    key: `${file.docData.documentId}:${file.docData.locale}`
+    key: `${file.branchId}:${file.fileId}:${file.versionId}:${file.locale}`
   })), result = await processImportBatch(importItems, {
     onProgress: options.onProgress,
     onItemSuccess: (item, key) => {
@@ -1104,7 +1103,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
   children,
   singleDocument
 }) => {
-  const [isBusy, setIsBusy] = useState(!1), [documents, setDocuments] = useState([]), [locales, setLocales] = useState([]), [autoRefresh, setAutoRefresh] = useState(!1), [loadingDocuments, setLoadingDocuments] = useState(!1), [importProgress, setImportProgress] = useState({
+  const [isBusy, setIsBusy] = useState(!1), [documents, setDocuments] = useState([]), [locales, setLocales] = useState([]), [autoRefresh, setAutoRefresh] = useState(!1), [autoImport, setAutoImport] = useState(!1), [autoPatchReferences, setAutoPatchReferences] = useState(!1), [autoPublish, setAutoPublish] = useState(!1), [loadingDocuments, setLoadingDocuments] = useState(!1), [importProgress, setImportProgress] = useState({
     current: 0,
     total: 0,
     isImporting: !1
@@ -1118,7 +1117,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     skipped: /* @__PURE__ */ new Set()
   }), [translationStatuses, setTranslationStatuses] = useState(/* @__PURE__ */ new Map()), [isRefreshing, setIsRefreshing] = useState(!1), client = useClient(), schema = useSchema(), translationContext = { client, schema }, toast = useToast(), { loading: loadingSecrets, secrets } = useSecrets(
     pluginConfig.getSecretsNamespace()
-  ), fetchDocuments = useCallback(async () => {
+  ), [branchId, setBranchId] = useState(void 0), fetchDocuments = useCallback(async () => {
     setLoadingDocuments(!0);
     try {
       if (singleDocument) {
@@ -1216,13 +1215,10 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
       }
     }
   }, [secrets, documents, locales, schema]), handleImportAll = useCallback(async () => {
-    if (!(!secrets || documents.length === 0)) {
+    if (!(!secrets || documents.length === 0 || !branchId)) {
       setIsBusy(!0);
       try {
-        const readyFiles = await getReadyFilesForImport(
-          documents,
-          translationStatuses
-        );
+        const readyFiles = await getReadyFilesForImport(translationStatuses);
         if (readyFiles.length === 0) {
           toast.push({
             title: "No ready translations to import",
@@ -1279,13 +1275,15 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     documents,
     translationStatuses,
     downloadStatus,
-    translationContext
-  ]), getMissingTranslations = useCallback(
-    async (documentIds, localeIds) => {
+    translationContext,
+    branchId
+  ]), getExistingTranslations = useCallback(
+    async (documentIds, localeIds, branchId2) => {
       const sourceLocale = pluginConfig.getSourceLocale(), existingMetadata = await client.fetch(`*[
       _type == 'translation.metadata' &&
       translations[_key == $sourceLocale][0].value._ref in $documentIds
     ] {
+      _rev,
       'sourceDocId': translations[_key == $sourceLocale][0].value._ref,
       'existingTranslations': translations[_key in $localeIds]._key
     }`, {
@@ -1293,40 +1291,28 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
         documentIds,
         localeIds
       }), existing = /* @__PURE__ */ new Set();
-      existingMetadata.forEach((metadata) => {
+      return existingMetadata.forEach((metadata) => {
         metadata.existingTranslations?.forEach((localeId) => {
-          localeId !== sourceLocale && existing.add(`${metadata.sourceDocId}:${localeId}`);
-        });
-      });
-      const missing = /* @__PURE__ */ new Set();
-      return documentIds.forEach((docId) => {
-        localeIds.forEach((localeId) => {
-          if (localeId !== sourceLocale) {
-            const key = `${docId}:${localeId}`;
-            existing.has(key) || missing.add(key);
-          }
+          localeId !== sourceLocale && existing.add(
+            `${branchId2}:${metadata.sourceDocId}:${metadata._rev}:${localeId}`
+          );
         });
-      }), missing;
+      }), existing;
     },
     [client]
   ), handleImportMissing = useCallback(async () => {
-    if (!(!secrets || documents.length === 0)) {
+    if (!(!secrets || documents.length === 0 || !branchId)) {
       setIsBusy(!0);
       try {
         const availableLocaleIds = locales.filter((locale) => locale.enabled !== !1).map((locale) => locale.localeId), documentIds = documents.map(
           (doc) => doc._id?.replace("drafts.", "") || doc._id
-        ), missingTranslations = await getMissingTranslations(
+        ), existingTranslations2 = await getExistingTranslations(
           documentIds,
-          availableLocaleIds
-        );
-        console.log("missingTranslations", missingTranslations);
-        const readyFiles = await getReadyFilesForImport(
-          documents,
-          translationStatuses,
-          {
-            filterReadyFiles: (key) => missingTranslations.has(key)
-          }
-        );
+          availableLocaleIds,
+          branchId
+        ), readyFiles = await getReadyFilesForImport(translationStatuses, {
+          filterReadyFiles: (key) => !existingTranslations2.has(key)
+        });
         if (readyFiles.length === 0) {
           toast.push({
             title: "No missing translations to import",
@@ -1385,9 +1371,20 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     translationStatuses,
     downloadStatus,
     translationContext,
-    getMissingTranslations
-  ]), handleRefreshAll = useCallback(async () => {
-    if (!(!secrets || documents.length === 0)) {
+    getExistingTranslations,
+    branchId
+  ]), handleGetBranchId = useCallback(
+    async (secrets2) => {
+      overrideConfig(secrets2);
+      const defaultBranch = await gt.createBranch({
+        branchName: "main",
+        defaultBranch: !0
+      });
+      setBranchId(defaultBranch.branch.id);
+    },
+    [secrets]
+  ), handleRefreshAll = useCallback(async () => {
+    if (!(!secrets || documents.length === 0 || !branchId)) {
       setIsRefreshing(!0);
       try {
         const availableLocaleIds = locales.filter((locale) => locale.enabled !== !1).map((locale) => locale.localeId), fileQueryData = [];
@@ -1397,6 +1394,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
             fileQueryData.push({
               versionId: doc._rev,
               fileId: documentId,
+              branchId,
               locale: localeId
             });
           }
@@ -1409,16 +1407,21 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
           const newStatuses = /* @__PURE__ */ new Map();
           for (const doc of documents)
             for (const localeId of availableLocaleIds) {
-              const key = `${doc._id?.replace("drafts.", "") || doc._id}:${localeId}`;
+              const documentId = doc._id?.replace("drafts.", "") || doc._id, versionId = doc._rev, key = `${branchId}:${documentId}:${versionId}:${localeId}`;
               newStatuses.set(key, { progress: 0, isReady: !1 });
             }
           if (Array.isArray(readyTranslations))
             for (const translation of readyTranslations) {
-              const key = `${translation.fileId}:${translation.locale}`;
+              const key = `${branchId}:${translation.fileId}:${translation.versionId}:${translation.locale}`;
               newStatuses.set(key, {
                 progress: 100,
                 isReady: !0,
-                translationId: translation.id
+                fileData: {
+                  versionId: translation.versionId,
+                  fileId: translation.fileId,
+                  branchId: translation.branchId,
+                  locale: translation.locale
+                }
               });
             }
           return newStatuses;
@@ -1437,11 +1440,11 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
         setIsRefreshing(!1);
       }
     }
-  }, [secrets, documents, locales]), handleImportDocument = useCallback(
-    async (documentId, localeId) => {
+  }, [secrets, documents, locales, branchId]), handleImportDocument = useCallback(
+    async (documentId, versionId, localeId) => {
       if (!secrets) return;
-      const key = `${documentId}:${localeId}`, status = translationStatuses.get(key);
-      if (!status?.isReady || !status.translationId) {
+      const key = `${branchId}:${documentId}:${versionId}:${localeId}`, status = translationStatuses.get(key);
+      if (!status?.isReady || !status.fileData) {
         toast.push({
           title: `Translation not ready for ${documentId} (${localeId})`,
           status: "warning",
@@ -1464,10 +1467,10 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
         const downloadedFiles = await downloadTranslations(
           [
             {
-              documentId,
-              versionId: document2._rev,
-              translationId: status.translationId,
-              locale: localeId
+              fileId: status.fileData.fileId,
+              branchId: status.fileData.branchId,
+              versionId: status.fileData.versionId,
+              locale: status.fileData.locale
             }
           ],
           secrets
@@ -1513,7 +1516,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
         });
       }
     },
-    [secrets, documents, translationContext, translationStatuses]
+    [secrets, documents, translationContext, translationStatuses, branchId]
   ), handlePatchDocumentReferences = useCallback(async () => {
     if (!secrets || documents.length === 0) return 0;
     setIsBusy(!0);
@@ -1588,7 +1591,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     } finally {
       setIsBusy(!1), setImportProgress({ current: 0, total: 0, isImporting: !1 });
     }
-  }, [secrets, documents, locales, client]), handlePublishAllTranslations = useCallback(async () => {
+  }, [secrets, documents, locales, client, branchId]), handlePublishAllTranslations = useCallback(async () => {
     if (!secrets || documents.length === 0) return 0;
     setIsBusy(!0);
     try {
@@ -1640,7 +1643,7 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     } finally {
       setIsBusy(!1);
     }
-  }, [secrets, documents, client]);
+  }, [secrets, documents, client, branchId]);
   useEffect(() => {
     fetchDocuments();
   }, [fetchDocuments]), useEffect(() => {
@@ -1657,13 +1660,16 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     return () => clearInterval(interval);
   }, [autoRefresh, documents.length, secrets, handleRefreshAll]), useEffect(() => {
     setImportedTranslations(new Set(downloadStatus.downloaded));
-  }, [downloadStatus.downloaded]);
+  }, [downloadStatus.downloaded]), secrets && handleGetBranchId(secrets);
   const contextValue = {
     // State
     isBusy,
     documents,
     locales,
     autoRefresh,
+    autoImport,
+    autoPatchReferences,
+    autoPublish,
     loadingDocuments,
     importProgress,
     importedTranslations,
@@ -1673,9 +1679,13 @@ const getLocales = async (secrets) => pluginConfig.getLocales().map((locale) =>
     isRefreshing,
     loadingSecrets,
     secrets,
+    branchId,
     // Actions
     setLocales,
     setAutoRefresh,
+    setAutoImport,
+    setAutoPatchReferences,
+    setAutoPublish,
     handleTranslateAll,
     handleImportAll,
     handleImportMissing,
@@ -2939,6 +2949,7 @@ const WrapText = dt(Box)`
     documents,
     locales,
     translationStatuses,
+    branchId,
     isBusy,
     handleTranslateAll,
     handleImportDocument,
@@ -2947,8 +2958,16 @@ const WrapText = dt(Box)`
     importedTranslations,
     setLocales,
     handlePatchDocumentReferences,
-    handlePublishAllTranslations
-  } = useTranslations(), [autoImport, setAutoImport] = useState(!1), [isImporting, setIsImporting] = useState(!1), [autoRefresh, setAutoRefresh] = useState(!0), [autoPatchReferences, setAutoPatchReferences] = useState(!0), [autoPublish, setAutoPublish] = useState(!0), [isPublishing, setIsPublishing] = useState(!1), toast = useToast(), document2 = documents[0], currentDocumentLanguage = useMemo(() => {
+    handlePublishAllTranslations,
+    autoRefresh,
+    setAutoRefresh,
+    autoImport,
+    setAutoImport,
+    autoPatchReferences,
+    setAutoPatchReferences,
+    autoPublish,
+    setAutoPublish
+  } = useTranslations(), [isImporting, setIsImporting] = useState(!1), [isPublishing, setIsPublishing] = useState(!1), toast = useToast(), document2 = documents[0], currentDocumentLanguage = useMemo(() => {
     if (!document2) return null;
     const languageField = pluginConfig.getLanguageField();
     return document2[languageField] || pluginConfig.getSourceLocale();
@@ -2962,7 +2981,7 @@ const WrapText = dt(Box)`
       const { autoOnly = !1 } = options;
       if (isImporting || !documentId || autoOnly && !autoImport) return;
       const readyTranslations = availableLocales.filter((locale) => {
-        const key = `${documentId}:${locale.localeId}`;
+        const key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`;
         return translationStatuses.get(key)?.isReady && !importedTranslations.has(key);
       });
       if (readyTranslations.length !== 0) {
@@ -2970,7 +2989,7 @@ const WrapText = dt(Box)`
         try {
           await Promise.all(
             readyTranslations.map(
-              (locale) => handleImportDocument(documentId, locale.localeId)
+              (locale) => handleImportDocument(documentId, document2._rev, locale.localeId)
             )
           ), autoPatchReferences && await handlePatchDocumentReferences(), autoPublish && await handlePublishAllTranslations();
         } finally {
@@ -2996,20 +3015,8 @@ const WrapText = dt(Box)`
   useEffect(() => {
     handleImportTranslations({ autoOnly: !0 });
   }, [handleImportTranslations]), useEffect(() => {
-    if (!autoRefresh || !documentId || availableLocales.length === 0) return;
-    const interval = setInterval(async () => {
-      await handleRefreshAll(), await handleImportTranslations({ autoOnly: !0 });
-    }, 1e4);
-    return () => clearInterval(interval);
-  }, [
-    autoRefresh,
-    documentId,
-    availableLocales.length,
-    handleRefreshAll,
-    handleImportTranslations
-  ]), useEffect(() => {
-    (async () => (await handleRefreshAll(), await handleImportTranslations({ autoOnly: !0 })))();
-  }, []);
+    setAutoRefresh(!0), setAutoPatchReferences(!0), setAutoPublish(!0);
+  }, [setAutoRefresh, setAutoPatchReferences, setAutoPublish]);
   const toggleLocale = useCallback(
     (localeId, shouldEnable) => {
       const updatedLocales = locales.map(
@@ -3089,13 +3096,13 @@ const WrapText = dt(Box)`
               padding: 2,
               text: "Refresh Status",
               onClick: handleRefreshAll,
-              disabled: isRefreshing
+              disabled: isRefreshing || isBusy
             }
           )
         ] })
       ] }),
       /* @__PURE__ */ jsx(Box, { children: availableLocales.map((locale) => {
-        const key = `${documentId}:${locale.localeId}`, status = translationStatuses.get(key), progress = status?.progress || 0, isImported = importedTranslations.has(key);
+        const key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`, status = translationStatuses.get(key), progress = status?.progress || 0, isImported = importedTranslations.has(key);
         return /* @__PURE__ */ jsx(
           LanguageStatus,
           {
@@ -3103,7 +3110,11 @@ const WrapText = dt(Box)`
             progress,
             isImported,
             importFile: async () => {
-              !isImported && status?.isReady && await handleImportDocument(documentId, locale.localeId);
+              !isImported && status?.isReady && await handleImportDocument(
+                documentId,
+                document2._rev,
+                locale.localeId
+              );
             }
           },
           key
@@ -3121,7 +3132,7 @@ const WrapText = dt(Box)`
                 text: isImporting ? "Importing..." : "Import All",
                 icon: DownloadIcon,
                 disabled: isImporting || availableLocales.every((locale) => {
-                  const key = `${documentId}:${locale.localeId}`;
+                  const key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`;
                   return !translationStatuses.get(key)?.isReady || importedTranslations.has(key);
                 }),
                 style: { minWidth: "180px" }
@@ -3143,12 +3154,12 @@ const WrapText = dt(Box)`
             "Imported",
             " ",
             availableLocales.filter((locale) => {
-              const key = `${documentId}:${locale.localeId}`;
+              const key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`;
               return importedTranslations.has(key);
             }).length,
             "/",
             availableLocales.filter((locale) => {
-              const key = `${documentId}:${locale.localeId}`;
+              const key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`;
               return translationStatuses.get(key)?.isReady;
             }).length
           ] })
@@ -3260,7 +3271,8 @@ const WrapText = dt(Box)`
     translationStatuses,
     downloadStatus,
     importedTranslations,
-    handleImportDocument
+    handleImportDocument,
+    branchId
   } = useTranslations();
   return loadingDocuments ? /* @__PURE__ */ jsx(Flex, { align: "center", justify: "center", padding: 4, children: /* @__PURE__ */ jsx(Spinner, {}) }) : /* @__PURE__ */ jsx(Box, { style: { maxHeight: "60vh", overflowY: "auto" }, children: /* @__PURE__ */ jsx(Stack, { space: 2, children: documents.map((document2) => /* @__PURE__ */ jsx(Card, { shadow: 1, padding: 3, children: /* @__PURE__ */ jsxs(Stack, { space: 3, children: [
     /* @__PURE__ */ jsx(Flex, { justify: "space-between", align: "flex-start", children: /* @__PURE__ */ jsxs(Box, { flex: 1, children: [
@@ -3268,7 +3280,7 @@ const WrapText = dt(Box)`
       /* @__PURE__ */ jsx(Text, { size: 0, muted: !0, style: { marginTop: "2px" }, children: document2._type })
     ] }) }),
     /* @__PURE__ */ jsx(Stack, { space: 2, children: locales.length > 0 ? locales.filter((locale) => locale.enabled !== !1).map((locale) => {
-      const documentId = document2._id?.replace("drafts.", "") || document2._id, key = `${documentId}:${locale.localeId}`, status = translationStatuses.get(key), isDownloaded = downloadStatus.downloaded.has(key), isImported = importedTranslations.has(key);
+      const documentId = document2._id?.replace("drafts.", "") || document2._id, key = `${branchId}:${documentId}:${document2._rev}:${locale.localeId}`, status = translationStatuses.get(key), isDownloaded = downloadStatus.downloaded.has(key), isImported = importedTranslations.has(key);
       return /* @__PURE__ */ jsx(
         LanguageStatus,
         {
@@ -3278,11 +3290,12 @@ const WrapText = dt(Box)`
           importFile: async () => {
             await handleImportDocument(
               documentId,
+              document2._rev,
               locale.localeId
             );
           }
         },
-        `${document2._id}-${locale.localeId}`
+        `${document2._id}-${document2._rev}-${locale.localeId}`
       );
     }) : /* @__PURE__ */ jsx(Text, { size: 1, muted: !0, children: "No locales configured" }) })
   ] }) }, document2._id)) }) });