From fff574c0dd3477556fff3e5543b41f0978012755 Mon Sep 17 00:00:00 2001
From: HexedHero <6012891+HexedHero@users.noreply.github.com>
Date: Sat, 26 Sep 2020 23:51:03 +0100
Subject: [PATCH] Load chunks async when regenning blocks closes #75

---
 pom.xml                                       | 37 ++++++++++++++++++-
 .../mmocore/manager/CustomBlockManager.java   | 11 ++++--
 2 files changed, 43 insertions(+), 5 deletions(-)

diff --git a/pom.xml b/pom.xml
index 813046d6..969ad011 100644
--- a/pom.xml
+++ b/pom.xml
@@ -51,6 +51,27 @@
                     <argLine>-ea -Dfile.encoding=UTF-8</argLine>
                 </configuration>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-shade-plugin</artifactId>
+                <version>3.2.4</version>
+                <configuration>
+                    <relocations>
+                        <relocation>
+                            <pattern>io.papermc.lib</pattern>
+                            <shadedPattern>net.indyuce.mmocore.shaded.paperlib</shadedPattern>
+                        </relocation>
+                    </relocations>
+                </configuration>
+                <executions>
+                    <execution>
+                        <phase>package</phase>
+                        <goals>
+                            <goal>shade</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
         </plugins>
     </build>
     <repositories>
@@ -66,6 +87,10 @@
             <id>sk89q-repo</id>
             <url>https://maven.enginehub.org/repo/</url>
         </repository>
+        <repository>
+            <id>papermc</id>
+            <url>https://papermc.io/repo/repository/maven-public/</url>
+        </repository>
     </repositories>
     <dependencies>
 
@@ -93,19 +118,27 @@
             <scope>system</scope>
             <systemPath>${basedir}/lib/spigot.jar</systemPath>
         </dependency>
-        
+
         <!-- APIs -->
         <dependency>
             <groupId>org.jetbrains</groupId>
             <artifactId>annotations</artifactId>
             <version>19.0.0</version>
+            <scope>provided</scope>
         </dependency>
         <dependency>
             <groupId>mysql</groupId>
             <artifactId>mysql-connector-java</artifactId>
             <version>5.1.41</version>
+            <scope>provided</scope>
         </dependency>
-        
+        <dependency>
+            <groupId>io.papermc</groupId>
+            <artifactId>paperlib</artifactId>
+            <version>1.0.5</version>
+            <scope>compile</scope>
+        </dependency>
+
         <!-- Other Plugins -->
         <dependency>
             <groupId>com.bekvon.bukkit</groupId>
diff --git a/src/main/java/net/Indyuce/mmocore/manager/CustomBlockManager.java b/src/main/java/net/Indyuce/mmocore/manager/CustomBlockManager.java
index 8c394fec..4a6d4fae 100644
--- a/src/main/java/net/Indyuce/mmocore/manager/CustomBlockManager.java
+++ b/src/main/java/net/Indyuce/mmocore/manager/CustomBlockManager.java
@@ -16,6 +16,7 @@ import org.bukkit.block.Block;
 import org.bukkit.configuration.ConfigurationSection;
 import org.bukkit.entity.Entity;
 
+import io.papermc.lib.PaperLib;
 import net.Indyuce.mmocore.MMOCore;
 import net.Indyuce.mmocore.api.block.BlockInfo;
 import net.Indyuce.mmocore.api.block.BlockInfo.RegeneratingBlock;
@@ -91,9 +92,13 @@ public class CustomBlockManager extends MMOManager {
 	}
 
 	private void regen(RegeneratingBlock info, boolean shutdown) {
-		info.getRegeneratingBlock().getBlock().place(info.getLocation(), info);
-		info.getLocation().getBlock().getState().update();
-		if(!shutdown) active.remove(info);
+        Location infoLocation = info.getLocation();
+		// Get the chunk and load it async if needed.
+        PaperLib.getChunkAtAsync(infoLocation).whenComplete((chunk, ex) -> {
+                info.getRegeneratingBlock().getBlock().place(infoLocation, info);
+                info.getLocation().getBlock().getState().update();
+                if(!shutdown) active.remove(info);
+        });
 	}
 
 	/*