add first patch

kubejs/startup_scripts/globalMachineOwnerUtils.js

@@ -0,0 +1,81 @@
+/**
+ * wrapper so that arbitrary stage data can be read using the has() method same as player.stages
+ * @param {string[]} stages
+ * @return {object}
+ */
+function SocietyStages(stages) {
+  return {
+    value: stages,
+    has: function (x) {
+      return this.value.toString().includes(String(x));
+    },
+  };
+}
+
+global.NO_STAGES = SocietyStages([]);
+
+const TAG_STRING = Java.loadClass("net.minecraft.nbt.Tag").TAG_STRING;
+
+/**
+ * reads stages set in block entity NBT data by global.cacheOwner()
+ * @param {Internal.BlockEntityJS} blockEntity
+ * @return {SocietyStages}
+ */
+global.getBlockEntityStages = (blockEntity) => {
+  return SocietyStages(
+    blockEntity.data.getList("stages", TAG_STRING).toArray().map((x) => x.toString())
+  );
+}
+
+/**
+ * cache the stages of a block entity's owner in the block entity's NBT data
+ * cache the existence of the block entity's owner's attributes, regardless of attribute value
+ * if the owner is online, that player is returned, else null is returned
+ * if nothing has ever been cached before, and the owner is not online, stages=[] attributes=[]
+ * @param {Internal.BlockEntityJS} entity
+ * @param {string[]|null|undefined} stagesToFind
+ * @param {string[]|null|undefined} attributesToFind
+ * @return {Internal.Player|null} owner
+ */
+global.cacheOwner = (entity, stagesToFind, attributesToFind) => {
+  const { level, block } = entity;
+  let binPlayer = null;
+  level.getServer().players.forEach((p) => {
+    if (p.getUuid().toString() === block.getEntityData().data.owner) {
+      binPlayer = p; 
+    }
+  });
+  let nbt = block.getEntityData();
+  if (binPlayer) {
+    if (attributesToFind) {
+      let newAttributes = [];
+      attributesToFind.forEach((attr) => {
+        newAttributes.push({
+          Base: Number(
+            binPlayer.nbt.Attributes.filter((obj) => {
+              return obj.Name === attr;
+            })[0]?.Base
+          ),
+          Name: attr,
+        });
+      });
+      nbt.data.attributes = newAttributes;
+    }
+    if (stagesToFind) {
+      let stages = [];
+      stagesToFind.forEach((stage) => {
+        if (binPlayer.stages.has(stage)) stages.push(stage);
+      });
+      nbt.data.stages = stages;
+    }
+  } else {
+    if ((attributesToFind !== null) && !nbt.data.contains("attributes")) {
+      nbt.data.attributes = [];
+    }
+    if ((stagesToFind !== null) && !nbt.data.contains("stages")) {
+      nbt.data.stages = [];
+    }
+  }
+  global.setBlockEntityData(block, nbt);
+  return binPlayer;
+};