Add support for importing language variants

kontent-ai · Jan 4, 2024 · 6071468 · 6071468
1 parent ba50e12
commit 6071468
Show file tree

Hide file tree

Showing 11 changed files with 523 additions and 68 deletions.
diff --git a/src/commands/export.ts b/src/commands/export.ts
@@ -12,7 +12,7 @@ import { contentItemsExportEntity } from "./importExportEntities/entities/conten
 import { contentTypesEntity } from "./importExportEntities/entities/contentTypes.js";
 import { contentTypesSnippetsEntity } from "./importExportEntities/entities/contentTypesSnippets.js";
 import { languagesEntity } from "./importExportEntities/entities/languages.js";
-import { languageVariantsExportEntity } from "./importExportEntities/entities/languageVariants.js";
+import { languageVariantsEntity } from "./importExportEntities/entities/languageVariants.js";
 import { previewUrlsExportEntity } from "./importExportEntities/entities/previewUrls.js";
 import { rolesExportEntity } from "./importExportEntities/entities/roles.js";
 import { spacesExportEntity } from "./importExportEntities/entities/spaces.js";
@@ -60,7 +60,7 @@ const entityDefinitions: ReadonlyArray<EntityExportDefinition<any>> = [
   contentTypesSnippetsEntity,
   contentTypesEntity,
   contentItemsExportEntity,
-  languageVariantsExportEntity,
+  languageVariantsEntity,
   assetFoldersEntity,
   assetsEntity,
 ];

diff --git a/src/commands/import.ts b/src/commands/import.ts
@@ -17,6 +17,7 @@ import {
   updateItemAndTypeReferencesInSnippetsImportEntity,
 } from "./importExportEntities/entities/contentTypesSnippets.js";
 import { languagesEntity } from "./importExportEntities/entities/languages.js";
+import { languageVariantsEntity } from "./importExportEntities/entities/languageVariants.js";
 import { taxonomiesEntity } from "./importExportEntities/entities/taxonomies.js";
 import { workflowsEntity } from "./importExportEntities/entities/workflows.js";
 import { EntityImportDefinition, ImportContext } from "./importExportEntities/entityDefinition.js";
@@ -59,6 +60,7 @@ const entityDefinitions: ReadonlyArray<EntityImportDefinition<any>> = [
   updateItemAndTypeReferencesInSnippetsImportEntity,
   updateItemAndTypeReferencesInTypesImportEntity,
   workflowsEntity,
+  languageVariantsEntity,
 ];
 
 type ImportEntitiesParams = Readonly<{
@@ -105,9 +107,9 @@ const createInitialContext = (): ImportContext => ({
   taxonomyTermIdsByOldIds: new Map(),
   assetFolderIdsByOldIds: new Map(),
   assetIdsByOldIds: new Map(),
-  contentTypeSnippetIdsWithElementsByOldIds: new Map(),
-  contentItemIdsByOldIds: new Map(),
-  contentTypeIdsWithElementsByOldIds: new Map(),
+  contentTypeSnippetContextWithElementsByOldIds: new Map(),
+  contentItemContextByOldIds: new Map(),
+  contentTypeContextWithElementsByOldIds: new Map(),
   workflowIdsByOldIds: new Map(),
-  worfklowStepsIdsByOldIds: new Map(),
+  worfklowStepsIdsWithTransitionsByOldIds: new Map(),
 });
diff --git a/src/commands/importExportEntities/entities/contentItems.ts b/src/commands/importExportEntities/entities/contentItems.ts
@@ -15,7 +15,10 @@ export const contentItemsExportEntity: EntityDefinition<ReadonlyArray<ContentIte
 
       return {
         ...context,
-        contentItemIdsByOldIds: new Map(zip(fileItems, projectItems).map(([fItem, pItem]) => [fItem.id, pItem.id])),
+        contentItemContextByOldIds: new Map(
+          zip(fileItems, projectItems)
+            .map(([fItem, pItem]) => [fItem.id, { selfId: pItem.id, oldTypeId: fItem.type.id ?? "" }]),
+        ),
       };
     },
   };
@@ -28,7 +31,7 @@ const createImportItemFetcher =
       .addContentItem()
       .withData({
         ...fileItem,
-        type: { id: context.contentTypeIdsWithElementsByOldIds.get(fileItem.type.id ?? "")?.selfId },
+        type: { id: context.contentTypeContextWithElementsByOldIds.get(fileItem.type.id ?? "")?.selfId },
         collection: { id: context.collectionIdsByOldIds.get(fileItem.collection.id ?? "") },
         external_id: fileItem.external_id ?? fileItem.codename,
       })

diff --git a/src/commands/importExportEntities/entities/contentTypes.ts b/src/commands/importExportEntities/entities/contentTypes.ts
@@ -1,4 +1,4 @@
-import { ContentTypeContracts, ManagementClient } from "@kontent-ai/management-sdk";
+import { ContentTypeContracts, ContentTypeElements, ManagementClient } from "@kontent-ai/management-sdk";
 
 import { zip } from "../../../utils/array.js";
 import { serially } from "../../../utils/requests.js";
@@ -18,13 +18,58 @@ export const contentTypesEntity: EntityDefinition<ReadonlyArray<ContentTypeContr
 
     return {
       ...context,
-      contentTypeIdsWithElementsByOldIds: new Map(
+      contentTypeContextWithElementsByOldIds: new Map(
         zip(fileTypes, projectTypes)
           .map(([fileType, projectType]) => {
             const elementIdEntries = zip(fileType.elements, projectType.elements)
               .map(([fileEl, projectEl]) => [fileEl.id ?? "", projectEl.id ?? ""] as const);
 
-            return [fileType.id, { selfId: projectType.id, elementIdsByOldIds: new Map(elementIdEntries) }];
+            return [fileType.id, {
+              selfId: projectType.id,
+              elementIdsByOldIds: new Map(elementIdEntries),
+              elementTypeByOldIds: new Map(
+                fileType.elements.flatMap(el => {
+                  if (el.type === "snippet") {
+                    const typedEl = el as unknown as ContentTypeElements.ISnippetElement;
+                    return [
+                      ...context.contentTypeSnippetContextWithElementsByOldIds.get(typedEl.snippet.id ?? "")
+                        ?.elementTypeByOldIds ?? [],
+                    ];
+                  }
+
+                  return [[el.id ?? "", el.type]];
+                }),
+              ),
+              multiChoiceOptionIdsByOldIdsByOldElementId: new Map(
+                fileType.elements
+                  .flatMap(el => {
+                    switch (el.type) {
+                      case "snippet": {
+                        const typedEl = el as unknown as ContentTypeElements.ISnippetElement;
+                        return [
+                          ...context.contentTypeSnippetContextWithElementsByOldIds.get(typedEl.snippet.id ?? "")
+                            ?.multiChoiceOptionIdsByOldIdsByOldElementId ?? [],
+                        ];
+                      }
+                      case "multiple_choice": {
+                        const typedEl = el as unknown as ContentTypeElements.IMultipleChoiceElement;
+                        const typedProjectEl = projectType.elements.find(e =>
+                          e.id === el.id
+                        ) as ContentTypeElements.IMultipleChoiceElement;
+
+                        return [[
+                          el.id ?? "",
+                          new Map(
+                            zip(typedEl.options, typedProjectEl.options).map(([fO, pO]) => [fO.id ?? "", pO.id ?? ""]),
+                          ),
+                        ]];
+                      }
+                      default:
+                        return [];
+                    }
+                  }),
+              ),
+            }];
           }),
       ),
     };
@@ -82,7 +127,7 @@ const createUpdateTypeItemReferencesFetcher =
       .flatMap(
         createPatchItemAndTypeReferencesInTypeElement(
           params.context,
-          c => c.contentTypeIdsWithElementsByOldIds.get(type.id)?.elementIdsByOldIds,
+          c => c.contentTypeContextWithElementsByOldIds.get(type.id)?.elementIdsByOldIds,
         ),
       );
 
@@ -92,7 +137,7 @@ const createUpdateTypeItemReferencesFetcher =
 
     return params.client
       .modifyContentType()
-      .byTypeId(params.context.contentTypeIdsWithElementsByOldIds.get(type.id)?.selfId ?? "")
+      .byTypeId(params.context.contentTypeContextWithElementsByOldIds.get(type.id)?.selfId ?? "")
       .withData(patchOps)
       .toPromise();
   };
diff --git a/src/commands/importExportEntities/entities/contentTypesSnippets.ts b/src/commands/importExportEntities/entities/contentTypesSnippets.ts
@@ -1,4 +1,4 @@
-import { ContentTypeSnippetContracts, ManagementClient } from "@kontent-ai/management-sdk";
+import { ContentTypeElements, ContentTypeSnippetContracts, ManagementClient } from "@kontent-ai/management-sdk";
 
 import { zip } from "../../../utils/array.js";
 import { serially } from "../../../utils/requests.js";
@@ -21,15 +21,42 @@ export const contentTypesSnippetsEntity: EntityDefinition<
 
     return {
       ...context,
-      contentTypeSnippetIdsWithElementsByOldIds: new Map(
+      contentTypeSnippetContextWithElementsByOldIds: new Map(
         zip(fileSnippets, projectSnippets)
           .map(([fileSnippet, projectSnippet]) => {
             const elementIdEntries = zip(fileSnippet.elements, projectSnippet.elements)
               .map(([fileEl, projectEl]) => [fileEl.id ?? "", projectEl.id ?? ""] as const);
 
-            return [fileSnippet.id, { selfId: projectSnippet.id, elementIdsByOldIds: new Map(elementIdEntries) }];
+            return [fileSnippet.id, {
+              selfId: projectSnippet.id,
+              elementIdsByOldIds: new Map(elementIdEntries),
+              elementTypeByOldIds: new Map(fileSnippet.elements.map(el => [el.id ?? "", el.type])),
+              multiChoiceOptionIdsByOldIdsByOldElementId: new Map(
+                fileSnippet.elements
+                  .flatMap(el => {
+                    switch (el.type) {
+                      case "multiple_choice": {
+                        const typedEl = el as ContentTypeElements.IMultipleChoiceElement;
+                        const projectTypedEl = projectSnippet.elements
+                          .find(e => e.id === el.id) as ContentTypeElements.IMultipleChoiceElement;
+                        return [[
+                          el.id ?? "",
+                          new Map(
+                            zip(typedEl.options, projectTypedEl.options).map(([fO, pO]) => [fO.id ?? "", pO.id ?? ""]),
+                          ),
+                        ]];
+                      }
+                      default:
+                        return [];
+                    }
+                  }),
+              ),
+            }];
           }),
       ),
+      elementTypesByOldElementIdsByOldSnippetIds: new Map(
+        fileSnippets.map(snippet => [snippet.id, new Map(snippet.elements.map(el => [el.id ?? "", el.type]))]),
+      ),
     };
   },
 };
@@ -81,7 +108,7 @@ const createUpdateSnippetItemAndTypeReferencesFetcher =
       .flatMap(
         createPatchItemAndTypeReferencesInTypeElement(
           params.context,
-          c => c.contentTypeSnippetIdsWithElementsByOldIds.get(snippet.id)?.elementIdsByOldIds,
+          c => c.contentTypeSnippetContextWithElementsByOldIds.get(snippet.id)?.elementIdsByOldIds,
         ),
       );
 
@@ -91,7 +118,7 @@ const createUpdateSnippetItemAndTypeReferencesFetcher =
 
     return params.client
       .modifyContentTypeSnippet()
-      .byTypeId(params.context.contentTypeSnippetIdsWithElementsByOldIds.get(snippet.id)?.selfId ?? "")
+      .byTypeId(params.context.contentTypeSnippetContextWithElementsByOldIds.get(snippet.id)?.selfId ?? "")
       .withData(patchOps)
       .toPromise();
   };