fix: Initialize waystones instantly in WaystonesAPI place methods

Author: BlayTheNinth

common/src/main/java/net/blay09/mods/waystones/InternalMethodsImpl.java

@@ -8,6 +8,8 @@
 import net.blay09.mods.waystones.api.trait.IAttunementItem;
 import net.blay09.mods.waystones.block.ModBlocks;
 import net.blay09.mods.waystones.block.WaystoneBlock;
+import net.blay09.mods.waystones.block.WaystoneBlockBase;
+import net.blay09.mods.waystones.block.entity.WaystoneBlockEntityBase;
 import net.blay09.mods.waystones.config.WaystonesConfig;
 import net.blay09.mods.waystones.core.*;
 import net.blay09.mods.waystones.requirement.RequirementModifierParser;
@@ -131,8 +133,19 @@ public ItemStack createBoundScroll(Waystone waystone) {
     @Override
     public Optional<Waystone> placeWaystone(ServerLevel level, BlockPos pos, WaystoneStyle style) {
         Block block = BuiltInRegistries.BLOCK.getValue(style.getBlockRegistryName());
-        level.setBlock(pos, block.defaultBlockState().setValue(WaystoneBlock.HALF, DoubleBlockHalf.LOWER), 3);
-        level.setBlock(pos.above(), block.defaultBlockState().setValue(WaystoneBlock.HALF, DoubleBlockHalf.UPPER), 3);
+        level.setBlock(pos, block.defaultBlockState()
+                .setValue(WaystoneBlock.HALF, DoubleBlockHalf.LOWER)
+                .setValue(WaystoneBlockBase.ORIGIN, WaystoneOrigin.PLAYER), 3);
+        level.setBlock(pos.above(), block.defaultBlockState()
+                .setValue(WaystoneBlock.HALF, DoubleBlockHalf.UPPER)
+                .setValue(WaystoneBlockBase.ORIGIN, WaystoneOrigin.PLAYER), 3);
+        if (level.getBlockEntity(pos) instanceof WaystoneBlockEntityBase waystoneBlockEntity) {
+            waystoneBlockEntity.initializeWaystone(level, null, WaystoneOrigin.PLAYER);
+            if (level.getBlockEntity(pos.above()) instanceof WaystoneBlockEntityBase waystoneBlockEntityAbove) {
+                waystoneBlockEntityAbove.initializeFromBase(waystoneBlockEntity);
+            }
+            return Optional.of(waystoneBlockEntity.getWaystone());
+        }
         return getWaystoneAt(level, pos);
     }
 
@@ -143,14 +156,30 @@ public Optional<Waystone> placeSharestone(ServerLevel level, BlockPos pos, DyeCo
             return Optional.empty();
         }
 
-        level.setBlock(pos, sharestone.defaultBlockState().setValue(WaystoneBlock.HALF, DoubleBlockHalf.LOWER), 3);
-        level.setBlock(pos.above(), sharestone.defaultBlockState().setValue(WaystoneBlock.HALF, DoubleBlockHalf.UPPER), 3);
+        level.setBlock(pos, sharestone.defaultBlockState()
+                .setValue(WaystoneBlock.HALF, DoubleBlockHalf.LOWER)
+                .setValue(WaystoneBlockBase.ORIGIN, WaystoneOrigin.PLAYER), 3);
+        level.setBlock(pos.above(), sharestone.defaultBlockState()
+                .setValue(WaystoneBlock.HALF, DoubleBlockHalf.UPPER)
+                .setValue(WaystoneBlockBase.ORIGIN, WaystoneOrigin.PLAYER), 3);
+        if (level.getBlockEntity(pos) instanceof WaystoneBlockEntityBase waystoneBlockEntity) {
+            waystoneBlockEntity.initializeWaystone(level, null, WaystoneOrigin.PLAYER);
+            if (level.getBlockEntity(pos.above()) instanceof WaystoneBlockEntityBase waystoneBlockEntityAbove) {
+                waystoneBlockEntityAbove.initializeFromBase(waystoneBlockEntity);
+            }
+            return Optional.of(waystoneBlockEntity.getWaystone());
+        }
         return getWaystoneAt(level, pos);
     }
 
     @Override
     public Optional<Waystone> placeWarpPlate(ServerLevel level, BlockPos pos) {
-        level.setBlock(pos, ModBlocks.warpPlate.defaultBlockState(), 3);
+        level.setBlock(pos, ModBlocks.warpPlate.defaultBlockState()
+                .setValue(WaystoneBlockBase.ORIGIN, WaystoneOrigin.PLAYER), 3);
+        if (level.getBlockEntity(pos) instanceof WaystoneBlockEntityBase waystoneBlockEntity) {
+            waystoneBlockEntity.initializeWaystone(level, null, WaystoneOrigin.PLAYER);
+            return Optional.of(waystoneBlockEntity.getWaystone());
+        }
         return getWaystoneAt(level, pos);
     }