build talent name translation map

2 files changed, 35 insertions, 2 deletions

diff --git a/src/maps/maps.js b/src/maps/maps.js
index 0b121cd..01c89ad 100644
--- a/src/maps/maps.js
+++ b/src/maps/maps.js
@@ -1,18 +1,34 @@
 const rarities = require('./rarities.json');
+const names = require('./names.json');
 
 function getTalentName(talent) {
-  if (talent == 'NoTalent' || talent == '' || talent == undefined) {
+  if (talent == 'NoTalent') {
     return 'No Talent';
   }
 
-  return talent.substring(talent.lastIndexOf('_') + 1, talent.length - 1) // snake case to words
+  switch (talent.slice(talent.length - 7 - 1, talent.length - 1)) {
+    case 'TalentA':
+    case 'TalentB':
+    case 'TalentC':
+      if (Object.keys(names).includes(talent)) {
+        return names[talent];
+      } else {
+        return 'unknown';
+      }
+    default:
+      return talent.substring(talent.lastIndexOf('_') + 1, talent.length - 1) // snake case to words
                .replace(/([A-Z])/g, (s) => ' ' + s.toLowerCase())
                .split(' ') // capitalize first letter
                .map((s) => s.charAt(0).toUpperCase() + s.slice(1))
                .join(' ');
+  }
 }
 
 function getTalentRarity(talent) {
+  if (talent == 'No Talent') {
+    return 'None';
+  }
+
   switch (talent.slice(talent.length - 7 - 1, talent.length - 1)) {
     case 'TalentA': return 'Rare';
     case 'TalentB': return 'Epic';
diff --git a/src/maps/names.json b/src/maps/names.json
new file mode 100644
index 0000000..0c43e3e
--- /dev/null
+++ b/src/maps/names.json
@@ -0,0 +1,17 @@
+{
+  "*Talent_Churnwalker_TalentA*": "TODO",
+  "*Talent_Churnwalker_TalentB*": "TODO",
+  "*Talent_Churnwalker_TalentC*": "TODO",
+  "*Talent_Lorelai_TalentA*": "TODO",
+  "*Talent_Lorelai_TalentB*": "TODO",
+  "*Talent_Lorelai_TalentC*": "TODO",
+  "*Talent_Malene_TalentA*": "TODO",
+  "*Talent_Malene_TalentB*": "TODO",
+  "*Talent_Malene_TalentC*": "TODO",
+  "*Talent_Tony_TalentA*": "TODO",
+  "*Talent_Tony_TalentB*": "TODO",
+  "*Talent_Tony_TalentC*": "TODO",
+  "*Talent_Varya_TalentA*": "TODO",
+  "*Talent_Varya_TalentB*": "TODO",
+  "*Talent_Varya_TalentC*": "TODO"
+}