From f8af29a62f9b56bc126acbe2f90b097c34a60ed7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Franti=C5=A1ek=20Ma=C5=A1a?= <frantisekmasa1@gmail.com>
Date: Thu, 7 Jul 2022 18:02:24 +0200
Subject: [PATCH] Migrate legacy armour to trappings

---
 .../migrations/v0.0.6__drop_legacy_armour.ts  | 80 +++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 firebase/migrations/v0.0.6__drop_legacy_armour.ts

diff --git a/firebase/migrations/v0.0.6__drop_legacy_armour.ts b/firebase/migrations/v0.0.6__drop_legacy_armour.ts
new file mode 100644
index 000000000..e7abf3550
--- /dev/null
+++ b/firebase/migrations/v0.0.6__drop_legacy_armour.ts
@@ -0,0 +1,80 @@
+import { uuid } from 'uuidv4';
+
+const ARMOUR_PARTS = {
+      "leftLeg": "LEFT_LEG",
+      "head": "HEAD",
+      "rightArm": "RIGHT_ARM",
+      "body": "BODY",
+      "rightLeg": "RIGHT_LEG",
+      "leftArm": "LEFT_ARM",
+      "shield": "SHIELD",
+};
+
+export async function migrate({firestore}: { firestore: firebase.firestore.Firestore }): Promise<void> {
+    const characters = await firestore.collection("parties").doc("bbc08a6e-38dd-48de-8608-0e5461c5d190")
+    .collection("characters").get();
+    //const characters = await firestore.collectionGroup("characters").get();
+
+    for (const document of characters.docs) {
+        const character = document.data();
+
+        const partyId = document.ref.parent.parent.id;
+
+        const armourDoc = document.ref.collection("features").doc("armor");
+
+        const armour = (await armourDoc.get()).data();
+
+        if (armour == undefined) {
+            continue;
+        }
+
+        const newArmourData = {};
+
+        for (const [key, value] of Object.entries(armour)) {
+            newArmourData[key] = 0;
+
+            if (value == 0) {
+                continue;
+            }
+
+            const part = ARMOUR_PARTS[key];
+
+            if (!part) {
+                throw `Unknown armour part ${part}`;
+            }
+
+            await createArmourTrapping(document.ref, part, value);
+        }
+
+        await armourDoc.set(newArmourData);
+    }
+}
+
+async function createArmourTrapping(
+    character: firebase.firestore.DocumentReference,
+    armourPart: string,
+    points: number
+): Promise<void> {
+    const id = uuid();
+
+    await character.collection("inventory")
+        .doc(id)
+        .set({
+            id: id,
+            name: "Migrated " + armourPart.replace("_", " ").toLowerCase() + " armour",
+            encumbrance: 0,
+            containerId: null,
+            description: "This armour was automatically created from old Armour mechanism unrelated to trappings.\n" +
+                         "From now on, equipped armour is automatically calculated from equipped Armour trappings",
+            quantity: 1,
+            trappingType: {
+                kind: "ARMOUR",
+                flaws: {},
+                locations: [armourPart],
+                points: points,
+                qualities: {},
+                type: "OTHER",
+                worn: true,
+            },
+        });
+}
\ No newline at end of file