fix: Extends functionality works from Statblocks (close #100)

javalent · Apr 22, 2023 · 055f07e · 055f07e
1 parent e9108ac
commit 055f07e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/utils/creature.ts b/src/utils/creature.ts
@@ -47,7 +47,7 @@ export class Creature {
         if (this.creature.cr) {
             return XP_PER_CR[this.creature.cr] ?? 0;
         }
-        const base = plugin.bestiary.find((c) => c.name == this.name);
+        const base = plugin.getBaseCreatureFromBestiary(this.name);
         if (base && base.cr) {
             return XP_PER_CR[base.cr] ?? 0;
         }

diff --git a/src/utils/suggester.ts b/src/utils/suggester.ts
@@ -336,14 +336,13 @@ export class SRDMonsterSuggestionModal extends SuggestionModal<
     }
     onChooseItem(item: HomebrewCreature | SRDMonster) {
         this.inputEl.value = item.name;
-        this.creature = item;
+        this.creature = this.plugin.getBaseCreatureFromBestiary(item.name);
         this.onClose();
         this.close();
     }
     selectSuggestion({ item }: FuzzyMatch<HomebrewCreature | SRDMonster>) {
         this.inputEl.value = item.name;
-        this.creature = item;
-
+        this.creature = this.plugin.getBaseCreatureFromBestiary(item.name);
         this.onClose();
         this.close();
     }