fix(lib:LootTable): constructor can't handle number[]

PFiS1737 · Jul 14, 2024 · 99724ab · 99724ab
1 parent a517ce8
commit 99724ab
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/lib/wrapper/LootTable.class.ts b/src/lib/wrapper/LootTable.class.ts
@@ -6,9 +6,17 @@ export interface ILootTableItemConfig<T> {
 }
 
 export class LootTable<T = number> {
-  constructor(items: Array<ILootTableItemConfig<T>>) {
+  constructor(items: Array<ILootTableItemConfig<T>> | number[]) {
     if (items) {
-      for (const item of items) this.addItem(item)
+      for (const item of items) {
+        if (typeof item === "number")
+          this.addItem({
+            weight: 1,
+            // @ts-ignore
+            value: item,
+          })
+        else this.addItem(item)
+      }
     }
   }