Backup, #289

src/main/java/nightkosh/gravestone/config/Config.java

@@ -62,6 +62,7 @@ public final void getConfigs() {
     public static boolean generateEmptyPlayerGraves;
     public static boolean dropGraveBlockAtDestruction;
     public static List<Integer> playerGravesDimensionalBlackList;
+    public static boolean createBackups;
 
     public static List<GraveStoneHelper.RestrictedArea> restrictGraveGenerationInArea;
 
@@ -106,6 +107,8 @@ private static void gravesConfig() {
 
         playerGravesDimensionalBlackList = ConfigsHelper.getDimensionList(config, CATEGORY_GRAVES, "PlayerGravesDimensionalBlackList", "",
                 "List of dimension ids in which player's graves will not be generated at death");
+
+        createBackups = config.get(CATEGORY_GRAVES, "CreateBackups", true).getBoolean();
     }
 
 

src/main/java/nightkosh/gravestone/helper/BackupsHandler.java

@@ -0,0 +1,66 @@
+package nightkosh.gravestone.helper;
+
+import net.minecraft.item.ItemStack;
+import net.minecraft.util.math.BlockPos;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * GraveStone mod
+ *
+ * @author NightKosh
+ * @license Lesser GNU Public License v3 (http://www.gnu.org/licenses/lgpl.html)
+ */
+public class BackupsHandler {
+
+    public static final Map<String, Backup> BACKUPS = new HashMap<>();
+
+    public static class Backup {
+        private int dimensionId;
+
+        private BlockPos pos;
+
+        private List<ItemStack> items = new ArrayList<>();
+
+        public Backup() {
+
+        }
+
+        public Backup(int dimensionId, BlockPos pos, List<ItemStack> items) {
+            this.dimensionId = dimensionId;
+            this.pos = pos;
+            setItems(items);
+        }
+
+        public int getDimensionId() {
+            return dimensionId;
+        }
+
+        public void setDimensionId(int dimensionId) {
+            this.dimensionId = dimensionId;
+        }
+
+        public BlockPos getPos() {
+            return pos;
+        }
+
+        public void setPos(BlockPos pos) {
+            this.pos = pos;
+        }
+
+        public List<ItemStack> getItems() {
+            return items;
+        }
+
+        public void setItems(List<ItemStack> items) {
+            for (ItemStack stack : items) {
+                if (stack != null && !stack.isEmpty()) {
+                    this.items.add(stack.copy());
+                }
+            }
+        }
+    }
+}

src/main/java/nightkosh/gravestone/helper/GraveGenerationHelper.java

@@ -375,6 +375,15 @@ private static void createOnDeath(Entity entity, World world, BlockPos pos, Deat
         EnumFacing direction = EnumFacing.getHorizontal(MathHelper.floor((double) (entity.rotationYaw * 4 / 360F) + 0.5) & 3);
 
         BlockPos newPos = findPlaceForGrave(world, entity, pos, damageSource);
+
+        if (Config.createBackups && entity instanceof EntityPlayer) {
+            try {
+                BackupsHandler.BACKUPS.put(entity.getName(), new BackupsHandler.Backup(world.provider.getDimension(), newPos, items));
+            } catch (Exception e) {
+                GSLogger.logError("Can't create backup!");
+            }
+        }
+
         if (newPos != null) {
             world.setBlockState(newPos, GSBlock.GRAVE_STONE.getDefaultState().withProperty(BlockGraveStone.FACING, direction), 2);
             TileEntityGraveStone tileEntity = (TileEntityGraveStone) world.getTileEntity(newPos);