Add taxonomy ids into context

kontent-ai · Dec 7, 2023 · 306692b · 306692b
1 parent d4d84bc
commit 306692b
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 7 deletions.
diff --git a/src/commands/import.ts b/src/commands/import.ts
@@ -54,6 +54,8 @@ const importEntities = async (params: ImportEntitiesParams) => {
 
   let context: ImportContext = {
     collectionIdsByOldIds: new Map(),
+    taxonomyGroupIdsByOldIds: new Map(),
+    taxonomyTermIdsByOldIds: new Map(),
   };
 
   await serially(entityDefinitions.map(def => async () => {

diff --git a/src/commands/importExportEntities/entities/taxonomies.ts b/src/commands/importExportEntities/entities/taxonomies.ts
@@ -1,18 +1,28 @@
 import { TaxonomyContracts } from "@kontent-ai/management-sdk";
 
+import { zip } from "../../../utils/array.js";
 import { serially } from "../../../utils/requests.js";
 import { EntityDefinition } from "../entityDefinition.js";
 
 export const taxonomiesEntity: EntityDefinition<ReadonlyArray<TaxonomyContracts.ITaxonomyContract>> = {
   name: "taxonomies",
   fetchEntities: client => client.listTaxonomies().toAllPromise().then(res => res.data.items.map(t => t._raw)),
   serializeEntities: taxonomies => JSON.stringify(taxonomies),
-  importEntities: async (client, fileTaxonomies) => {
-    await serially(fileTaxonomies.map(taxonomy => () =>
-      client
-        .addTaxonomy()
-        .withData(addExternalIds(taxonomy))
-        .toPromise()));
+  importEntities: async (client, fileTaxonomies, context) => {
+    const projectTaxonomies = await serially<ReadonlyArray<() => Promise<TaxonomyContracts.ITaxonomyContract>>>(
+      fileTaxonomies.map(taxonomy => () =>
+        client
+          .addTaxonomy()
+          .withData(addExternalIds(taxonomy))
+          .toPromise()
+          .then(res => res.data._raw))
+    );
+
+    return {
+      ...context,
+      taxonomyGroupIdsByOldIds: new Map(zip(fileTaxonomies.map(t => t.id), projectTaxonomies.map(t => t.id))),
+      taxonomyTermIdsByOldIds: new Map(zip(fileTaxonomies.flatMap(t => t.terms), projectTaxonomies.flatMap(t => t.terms)).flatMap(extractTermIdsEntries)),
+    };
   },
   deserializeEntities: JSON.parse,
 };
@@ -22,3 +32,8 @@ const addExternalIds = (taxonomy: TaxonomyContracts.ITaxonomyContract): Taxonomy
   external_id: taxonomy.external_id ?? taxonomy.codename,
   terms: taxonomy.terms.map(addExternalIds),
 });
+
+const extractTermIdsEntries = ([fileTaxonomy, projectTaxonomy]: readonly [TaxonomyContracts.ITaxonomyContract, TaxonomyContracts.ITaxonomyContract]): ReadonlyArray<readonly [string, string]> => [
+  [fileTaxonomy.id, projectTaxonomy.id] as const,
+  ...zip(fileTaxonomy.terms, projectTaxonomy.terms).flatMap(extractTermIdsEntries),
+];
diff --git a/src/commands/importExportEntities/entityDefinition.ts b/src/commands/importExportEntities/entityDefinition.ts
@@ -11,5 +11,9 @@ export type EntityDefinition<T> = Readonly<{
 }>;
 
 export type ImportContext = Readonly<{
-  collectionIdsByOldIds: ReadonlyMap<string, string>;
+  collectionIdsByOldIds: IdsMap;
+  taxonomyGroupIdsByOldIds: IdsMap;
+  taxonomyTermIdsByOldIds: IdsMap;
 }>;
+
+type IdsMap = ReadonlyMap<string, string>;
diff --git a/src/utils/array.ts b/src/utils/array.ts
@@ -0,0 +1,22 @@
+export const zip = <T1 extends ReadonlyArray<any>, T2 extends ReadonlyArray<any>>(arr1: T1, arr2: T2): Zip<T1, T2> =>
+  arr1
+    .slice(0, Math.min(arr1.length, arr2.length))
+    .map((el1, i) => [el1, arr2[i]] as const) as unknown as Zip<T1, T2>;
+
+type Zip<T1 extends ReadonlyArray<any>, T2 extends ReadonlyArray<any>> =
+  true extends IsEmptyTuple<T1> | IsEmptyTuple<T2>
+    ? readonly []
+    : [IsNonEmptyTuple<T1>, IsNonEmptyTuple<T2>] extends [true, true]
+      ? ZipTuples<T1, T2>
+      : ZipArrays<T1, T2>;
+
+type IsEmptyTuple<T extends ReadonlyArray<any>> = T extends readonly [] ? true : false;
+
+type IsNonEmptyTuple<T extends ReadonlyArray<any>> = T extends readonly [any, ...ReadonlyArray<any>] ? true : false;
+
+type ZipArrays<T1 extends ReadonlyArray<any>, T2 extends ReadonlyArray<any>> = ReadonlyArray<readonly [T1[number], T2[number]]>;
+
+type ZipTuples<T1 extends ReadonlyArray<any>, T2 extends ReadonlyArray<any>, Accum extends ReadonlyArray<readonly [any, any]> = readonly []> =
+  [T1, T2] extends [readonly [infer First1, ...infer Rest1 extends ReadonlyArray<any>], readonly [infer First2, ...infer Rest2 extends ReadonlyArray<any>]]
+    ? ZipTuples<Rest1, Rest2, readonly [...Accum, readonly [First1, First2]]>
+    : Accum;