From eeb2ef960b2bc26cdce9779b22fbf6b076f1c617 Mon Sep 17 00:00:00 2001
From: HeroBrineGoat <76707404+MasterOfTheFish@users.noreply.github.com>
Date: Thu, 6 Jan 2022 21:22:47 -0500
Subject: [PATCH] Added bstats

---
 build.gradle                                               | 2 ++
 .../io/github/fisher2911/hmccosmetics/HMCCosmetics.java    | 4 ++++
 .../java/io/github/fisher2911/hmccosmetics/user/User.java  | 7 +++++++
 .../java/io/github/fisher2911/hmccosmetics/util/Keys.java  | 1 +
 4 files changed, 14 insertions(+)

diff --git a/build.gradle b/build.gradle
index 700ef245..a8eef230 100644
--- a/build.gradle
+++ b/build.gradle
@@ -28,6 +28,7 @@ dependencies {
     implementation 'dev.triumphteam:triumph-gui:3.0.3'
     implementation 'me.mattstudios.utils:matt-framework:1.4.6'
     implementation 'org.spongepowered:configurate-yaml:4.1.2'
+    implementation 'org.bstats:bstats-bukkit:2.2.1'
 }
 
 test {
@@ -40,6 +41,7 @@ shadowJar {
     relocate 'net.kyori.adventure.text.minimessage', 'io.github.fisher2911.hmccosmetics.adventure.minimessage'
     relocate 'net.kyori.adventure.platform', 'io.github.fisher2911.hmccosmetics.adventure.platform'
     relocate 'org.spongepowered.configurate', 'io.github.fisher2911.hmccosmetics.configurate'
+    relocate 'org.bstats', 'io.github.fisher2911.hmccosmetics.bstats'
 }
 
 shadowJar {
diff --git a/src/main/java/io/github/fisher2911/hmccosmetics/HMCCosmetics.java b/src/main/java/io/github/fisher2911/hmccosmetics/HMCCosmetics.java
index a0fda8ae..33f8a28c 100644
--- a/src/main/java/io/github/fisher2911/hmccosmetics/HMCCosmetics.java
+++ b/src/main/java/io/github/fisher2911/hmccosmetics/HMCCosmetics.java
@@ -14,6 +14,7 @@ import io.github.fisher2911.hmccosmetics.message.MessageHandler;
 import io.github.fisher2911.hmccosmetics.message.Messages;
 import io.github.fisher2911.hmccosmetics.user.UserManager;
 import me.mattstudios.mf.base.CommandManager;
+import org.bstats.bukkit.Metrics;
 import org.bukkit.plugin.java.JavaPlugin;
 
 import java.util.Arrays;
@@ -30,6 +31,9 @@ public class HMCCosmetics extends JavaPlugin {
 
     @Override
     public void onEnable() {
+        final int pluginId = 13873;
+        final Metrics metrics = new Metrics(this, pluginId);
+
         protocolManager = ProtocolLibrary.getProtocolManager();
         this.messageHandler = new MessageHandler(this);
         this.userManager = new UserManager(this);
diff --git a/src/main/java/io/github/fisher2911/hmccosmetics/user/User.java b/src/main/java/io/github/fisher2911/hmccosmetics/user/User.java
index b20605e0..0553910f 100644
--- a/src/main/java/io/github/fisher2911/hmccosmetics/user/User.java
+++ b/src/main/java/io/github/fisher2911/hmccosmetics/user/User.java
@@ -4,12 +4,14 @@ import io.github.fisher2911.hmccosmetics.gui.ArmorItem;
 import io.github.fisher2911.hmccosmetics.inventory.PlayerArmor;
 import io.github.fisher2911.hmccosmetics.message.MessageHandler;
 import io.github.fisher2911.hmccosmetics.message.Messages;
+import io.github.fisher2911.hmccosmetics.util.Keys;
 import org.bukkit.Bukkit;
 import org.bukkit.Material;
 import org.bukkit.entity.ArmorStand;
 import org.bukkit.entity.Player;
 import org.bukkit.inventory.EntityEquipment;
 import org.bukkit.inventory.ItemStack;
+import org.bukkit.persistence.PersistentDataType;
 import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
@@ -161,6 +163,11 @@ public class User {
                     armorStand -> {
                         armorStand.setVisible(false);
                         armorStand.setMarker(true);
+                        armorStand.getPersistentDataContainer().set(
+                                Keys.ARMOR_STAND_KEY,
+                                PersistentDataType.BYTE,
+                                (byte) 1
+                        );
                         player.addPassenger(armorStand);
                     });
         }
diff --git a/src/main/java/io/github/fisher2911/hmccosmetics/util/Keys.java b/src/main/java/io/github/fisher2911/hmccosmetics/util/Keys.java
index 61418b51..89de2b91 100644
--- a/src/main/java/io/github/fisher2911/hmccosmetics/util/Keys.java
+++ b/src/main/java/io/github/fisher2911/hmccosmetics/util/Keys.java
@@ -15,6 +15,7 @@ public class Keys {
     }
 
     public static final NamespacedKey ITEM_KEY = new NamespacedKey(plugin, "cosmetic");
+    public static final NamespacedKey ARMOR_STAND_KEY = new NamespacedKey(plugin, "armor-stand");
 
     public static void setKey(final ItemStack itemStack) {
         final ItemMeta itemMeta = itemStack.getItemMeta();