From 3cf81afc4aa594d0870b41e8f2c77056d1f94d4d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Mon, 27 Nov 2023 12:26:51 -0600
Subject: [PATCH] feat: tree type

---
 scripts/buildTankopedia.ts | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/scripts/buildTankopedia.ts b/scripts/buildTankopedia.ts
index 78da26c1b..daa7d3cec 100644
--- a/scripts/buildTankopedia.ts
+++ b/scripts/buildTankopedia.ts
@@ -3,6 +3,7 @@ import { load } from 'js-yaml';
 import { argv } from 'process';
 import { ElementCompact, xml2js } from 'xml-js';
 import { parse } from 'yaml';
+import { TreeTypeString } from '../src/components/Tanks';
 import { NATION_IDS } from '../src/constants/nations';
 
 const VEHICLE_DEFS = 'assets/Data/XML/item_defs/vehicles';
@@ -42,8 +43,7 @@ interface BlitzkriegTankopedia {
   nation: string;
   name: string;
   name_short?: string;
-  is_premium: boolean;
-  is_collector: boolean;
+  tree_type: TreeTypeString;
   tier: number;
   type: string;
 }
@@ -74,8 +74,8 @@ Promise.all(
             // TODO: use userString property instead
             const name = strings[`#${nation}_vehicles:${vehicle}`];
             const name_short = strings[`#${nation}_vehicles:${vehicle}_short`];
-            const is_premium = 'gold' in vehicleData.price;
-            const is_collector = vehicleData.sellPrice
+            const isPremium = 'gold' in vehicleData.price;
+            const isCollector = vehicleData.sellPrice
               ? 'gold' in vehicleData.sellPrice
               : false;
             const tier = parseInt(vehicleData.level._text);
@@ -87,8 +87,11 @@ Promise.all(
               name,
               name_short,
               nation,
-              is_premium,
-              is_collector,
+              tree_type: isCollector
+                ? 'collector'
+                : isPremium
+                ? 'premium'
+                : 'tech-tree',
               tier,
               type,
             };