From 8770debfef28b9a83d0a9c2224043ce2c5dcdbf5 Mon Sep 17 00:00:00 2001
From: NorbiPeti <szatmari.norbert.peter@gmail.com>
Date: Wed, 3 Mar 2021 16:48:43 +0100
Subject: [PATCH] Add bStats metrics

---
 pom.xml                                       | 28 +++++++++++++++++++
 .../customdimensions/CustomDimensions.java    | 25 +++++++++++++----
 2 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/pom.xml b/pom.xml
index 5d24cc7..bbec711 100644
--- a/pom.xml
+++ b/pom.xml
@@ -23,6 +23,28 @@
                     </compilerArgs>
                 </configuration>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-shade-plugin</artifactId>
+                <version>3.2.4</version>
+                <configuration>
+                    <relocations>
+                        <relocation>
+                            <pattern>org.bstats</pattern>
+                            <shadedPattern>buttondevteam.customdimensions.bstats</shadedPattern>
+                        </relocation>
+                    </relocations>
+                    <createDependencyReducedPom>false</createDependencyReducedPom>
+                </configuration>
+                <executions>
+                    <execution>
+                        <phase>package</phase>
+                        <goals>
+                            <goal>shade</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
         </plugins>
     </build>
 
@@ -50,6 +72,12 @@
             <version>0.3.0</version>
             <scope>provided</scope>
         </dependency>
+        <dependency>
+            <groupId>org.bstats</groupId>
+            <artifactId>bstats-bukkit</artifactId>
+            <version>2.2.1</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 
     <profiles>
diff --git a/src/main/java/buttondevteam/customdimensions/CustomDimensions.java b/src/main/java/buttondevteam/customdimensions/CustomDimensions.java
index 02b46d8..05efb88 100644
--- a/src/main/java/buttondevteam/customdimensions/CustomDimensions.java
+++ b/src/main/java/buttondevteam/customdimensions/CustomDimensions.java
@@ -3,8 +3,11 @@ package buttondevteam.customdimensions;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Lists;
+import com.google.common.util.concurrent.Callables;
 import com.mojang.serialization.Lifecycle;
 import net.minecraft.server.v1_16_R3.*;
+import org.bstats.bukkit.Metrics;
+import org.bstats.charts.SimplePie;
 import org.bukkit.Bukkit;
 import org.bukkit.WorldType;
 import org.bukkit.craftbukkit.v1_16_R3.CraftServer;
@@ -18,8 +21,11 @@ import java.io.IOException;
 import java.util.*;
 
 public class CustomDimensions extends JavaPlugin implements Listener {
+	private Metrics metrics;
+
 	@Override
 	public void onEnable() {
+		metrics = new Metrics(this, 10545);
 		getLogger().info("Loading custom dimensions...");
 		try {
 			load();
@@ -46,26 +52,33 @@ public class CustomDimensions extends JavaPlugin implements Listener {
 			saveConfig();
 		}
 		var ignored = getConfig().getStringList("ignored");
+		int allCount = -3, loadedCount = 0, ignoredCount = 0; //-3: overworld, nether, end
 		for (var dimEntry : dimensionRegistry.d()) {
+			allCount++;
 			if (ignored.contains(dimEntry.getKey().a().getKey())) {
 				getLogger().info(dimEntry.getKey() + " is on the ignore list");
+				ignoredCount++;
 				continue;
 			}
 			try {
-				loadDimension(dimEntry.getKey(), dimEntry.getValue(), convertable, console, mainWorld);
+				if (loadDimension(dimEntry.getKey(), dimEntry.getValue(), convertable, console, mainWorld))
+					loadedCount++;
 			} catch (Exception e) {
 				getLogger().warning("Failed to load dimension " + dimEntry.getKey());
 				e.printStackTrace();
 			}
 		}
+		metrics.addCustomChart(new SimplePie("all_custom_dimensions", Callables.returning(allCount + "")));
+		metrics.addCustomChart(new SimplePie("loaded_custom_dimensions", Callables.returning(loadedCount + "")));
+		metrics.addCustomChart(new SimplePie("ignored_custom_dimensions", Callables.returning(ignoredCount + "")));
 	}
 
-	private void loadDimension(ResourceKey<WorldDimension> dimKey, WorldDimension dimension,
-	                           Convertable convertable, DedicatedServer console, org.bukkit.World mainWorld) throws IOException {
+	private boolean loadDimension(ResourceKey<WorldDimension> dimKey, WorldDimension dimension,
+	                              Convertable convertable, DedicatedServer console, org.bukkit.World mainWorld) throws IOException {
 		if (dimKey == WorldDimension.OVERWORLD //The default dimensions are already loaded
 				|| dimKey == WorldDimension.THE_NETHER
 				|| dimKey == WorldDimension.THE_END)
-			return;
+			return false;
 		ResourceKey<World> worldKey = ResourceKey.a(IRegistry.L, dimKey.a());
 		DimensionManager dimensionmanager = dimension.b();
 		ChunkGenerator chunkgenerator = dimension.c();
@@ -74,7 +87,7 @@ public class CustomDimensions extends JavaPlugin implements Listener {
 			name = dimKey.a().getKey();
 		if (Bukkit.getWorld(name) != null) {
 			getLogger().info(name + " already loaded");
-			return;
+			return false;
 		}
 		getLogger().info("Loading " + name);
 		var session = convertable.new ConversionSession(name, dimKey) { //The original session isn't prepared for custom dimensions
@@ -135,6 +148,7 @@ public class CustomDimensions extends JavaPlugin implements Listener {
 
 		if (Bukkit.getWorld(name.toLowerCase(Locale.ENGLISH)) == null) {
 			getLogger().warning("Failed to load custom dimension " + name);
+			return false;
 		} else {
 			console.initWorld(worldserver, worlddata, worlddata, worlddata.getGeneratorSettings());
 			worldserver.setSpawnFlags(true, true);
@@ -142,6 +156,7 @@ public class CustomDimensions extends JavaPlugin implements Listener {
 			Bukkit.getPluginManager().callEvent(new WorldInitEvent(worldserver.getWorld()));
 			console.loadSpawn(worldserver.getChunkProvider().playerChunkMap.worldLoadListener, worldserver);
 			Bukkit.getPluginManager().callEvent(new WorldLoadEvent(worldserver.getWorld()));
+			return true;
 		}
 	}
 }