From 08e2cd0153b6b0d1a0f9d3f4ab967fb5efba2c02 Mon Sep 17 00:00:00 2001
From: Jiri Lojda <lojdaj@seznam.cz>
Date: Wed, 6 Dec 2023 12:23:18 +0100
Subject: [PATCH] Add taxonomy ids into context

---
 src/commands/import.ts                        |  4 ++-
 .../entities/taxonomies.ts                    | 27 +++++++++++++----
 .../importExportEntities/entityDefinition.ts  |  4 +++
 src/utils/array.ts                            | 29 +++++++++++++++++++
 4 files changed, 57 insertions(+), 7 deletions(-)
 create mode 100644 src/utils/array.ts

diff --git a/src/commands/import.ts b/src/commands/import.ts
index 364efe33..ee0925e0 100644
--- a/src/commands/import.ts
+++ b/src/commands/import.ts
@@ -57,7 +57,9 @@ const importEntities = async (params: ImportEntitiesParams) => {
 
   let context: ImportContext = {
     collectionIdsByOldIds: new Map(),
-    languageIdsByOldIds: new Map()
+    languageIdsByOldIds: 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
index acb86ee4..5b705f70 100644
--- a/src/commands/importExportEntities/entities/taxonomies.ts
+++ b/src/commands/importExportEntities/entities/taxonomies.ts
@@ -1,5 +1,6 @@
 import { TaxonomyContracts } from "@kontent-ai/management-sdk";
 
+import { zip } from "../../../utils/array.js";
 import { serially } from "../../../utils/requests.js";
 import { EntityDefinition } from "../entityDefinition.js";
 
@@ -7,12 +8,21 @@ export const taxonomiesEntity: EntityDefinition<ReadonlyArray<TaxonomyContracts.
   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
index d8946b75..5653f369 100644
--- a/src/commands/importExportEntities/entityDefinition.ts
+++ b/src/commands/importExportEntities/entityDefinition.ts
@@ -13,4 +13,8 @@ export type EntityDefinition<T> = Readonly<{
 export type ImportContext = Readonly<{
   collectionIdsByOldIds: ReadonlyMap<string, string>;
   languageIdsByOldIds: ReadonlyMap<string, string>;
+  taxonomyGroupIdsByOldIds: IdsMap;
+  taxonomyTermIdsByOldIds: IdsMap;
 }>;
+
+type IdsMap = ReadonlyMap<string, string>;
diff --git a/src/utils/array.ts b/src/utils/array.ts
new file mode 100644
index 00000000..779d4b10
--- /dev/null
+++ b/src/utils/array.ts
@@ -0,0 +1,29 @@
+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;
+
+/**
+ * Handles zip of two types where at least one is an array of unknown length.
+*/
+type ZipArrays<T1 extends ReadonlyArray<any>, T2 extends ReadonlyArray<any>> = ReadonlyArray<readonly [T1[number], T2[number]]>;
+
+/**
+ * Handles zip of two tuples (arrays of known length and exact types for different positions).
+ * This type expects two tuples and doesn't work well with arrays of unknown length.
+*/
+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;