link weapon to ammo

Vendare · Aug 4, 2024 · 9cf69db · 9cf69db
1 parent 4a0cb38
commit 9cf69db
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 4 deletions.
diff --git a/script/data/item/ammunitionData.js b/script/data/item/ammunitionData.js
@@ -0,0 +1,52 @@
+import EquipmentItemData from "./equipmentItemData.js";
+
+const fields = foundry.data.fields;
+
+export default class AmmunitionData extends EquipmentItemData {
+
+    static defineSchema() {
+
+        const equipmentItemData = super.defineSchema();
+        return {
+            // Using destructuring to effectively append our additional data here
+            ...equipmentItemData,
+            quantity: new fields.NumberField({ initial: 0 }),
+            effect: new fields.SchemaField({
+                damage: new fields.SchemaField({
+                    modifier: new fields.NumberField({ initial: 0 }),
+                    type: new fields.StringField({ initial: "impact" })
+                }),
+                special: new fields.StringField({ initial: "" }),
+                penetration: new fields.StringField({ initial: "0" }),
+                attack: new fields.SchemaField({
+                    modifier: new fields.NumberField({ initial: 0 })
+                })
+            }),
+            weapon: new fields.StringField({ initial: "" }),
+            weaponId: new fields.StringField({ initial: "" })
+        };
+
+    }
+
+    prepareDerivedData() {
+        super.prepareDerivedData();
+
+        this.prepareWeaponFetch();
+
+    }
+
+    prepareWeaponFetch() {
+        // We only store a reference to the weapon, here we get the whole item and store it in memory only
+        // Weapons can only be connected to ammo for actor owned ammo
+        if (this.parent.actor && this.weaponId !== "") {
+            this.weaponItem = this.parent.actor.items.get(this.weaponId);
+            this.weapon = this.weaponItem.name;
+        }
+
+        if (this.parent.actor && this.weaponId === "") {
+            this.weaponItem = null;
+            this.weapon = "";
+        }
+    }
+
+}
diff --git a/script/sheet/weapon.js b/script/sheet/weapon.js
@@ -44,10 +44,12 @@ export class WeaponSheet extends DarkHeresyItemSheet {
 
         // It has to be ammunition from the same actor
         if (item?.type === "ammunition" && item?.actor.uuid === this.item.actor.uuid) {
-            if (this.item.ammo !== "") {
-                // remove old ammo
+            if (this.item.system.ammo !== "") {
+                let oldAmmo = this.item.actor.items.get(this.item.system.ammo);
+                oldAmmo.update({ "system.weaponId": "" });
             }
 
+            item.update({ "system.weaponId": this.item.id });
             this.item.update({ "system.ammo": item.id });
         }
     }

diff --git a/template/sheet/actor/tab/gear.hbs b/template/sheet/actor/tab/gear.hbs
@@ -27,8 +27,8 @@
             </div>
             <div class="items">
                 {{#each items.weapons as |item|}}
-                    <div class="gear item">
-                        <div class="flex row" data-item-id="{{item.id}}">
+                    <div class="gear item" data-item-id="{{item.id}}">
+                        <div class="flex row">
                             <div class="flex name item-edit">
                                 <div class="image-container">
                                     <div class="image" style="background-image: url('{{item.img}}')"></div>