From 84e44fe4f833b8b669f4adb4ab7699df0d14ca72 Mon Sep 17 00:00:00 2001
From: Aleksei <a.gryczewski999@gmail.com>
Date: Fri, 28 Feb 2020 17:44:03 -0500
Subject: [PATCH] New StatsCommand

---
 .../fsbot/commands/general/StatsCommand.java  | 46 +++++++++++++++++++
 .../dev/salmonllama/fsbot/guthix/Guthix.java  |  4 +-
 2 files changed, 49 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/dev/salmonllama/fsbot/commands/general/StatsCommand.java

diff --git a/src/main/java/dev/salmonllama/fsbot/commands/general/StatsCommand.java b/src/main/java/dev/salmonllama/fsbot/commands/general/StatsCommand.java
new file mode 100644
index 0000000..c2338f2
--- /dev/null
+++ b/src/main/java/dev/salmonllama/fsbot/commands/general/StatsCommand.java
@@ -0,0 +1,46 @@
+package dev.salmonllama.fsbot.commands.general;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collection;
+
+import org.javacord.api.entity.message.embed.EmbedBuilder;
+
+import dev.salmonllama.fsbot.database.controllers.OutfitController;
+import dev.salmonllama.fsbot.guthix.Command;
+import dev.salmonllama.fsbot.guthix.CommandContext;
+import dev.salmonllama.fsbot.guthix.CommandPermission;
+import dev.salmonllama.fsbot.guthix.PermissionType;
+
+public class StatsCommand extends Command {
+    @Override public String name() { return "Stats"; }
+    @Override public String description() { return "Shows various stats from Fashionscape Bot"; }
+    @Override public String usage() { return "stats"; }
+    @Override public String category() { return "General"; }
+    @Override public CommandPermission permission() { return new CommandPermission(PermissionType.NONE); }
+    @Override public Collection<String> aliases() { return new ArrayList<>(Arrays.asList("stats")); }
+
+    @Override
+    public void onCommand(CommandContext ctx) {
+        // Stats to display:
+        // Number of users
+        // Number of servers
+        // Number of galleries
+        // Number of non-deleted images
+        // Number of total images
+        // CPU and RAM?
+        int userCount = ctx.getApi().getCachedUsers().size(); // Will these be accurate with sharding?
+        int serverCount = ctx.getApi().getServers().size();
+
+        EmbedBuilder embed = new EmbedBuilder(); // TODO: Standard embeds yeah?
+        embed.setTitle("Stats");
+        embed.addField("Users:", String.valueOf(userCount));
+        embed.addField("Servers:", String.valueOf(serverCount));
+
+        OutfitController.countActiveOutfits().thenAccept(count -> embed.addField("Active outfits:", String.valueOf(count))).join();
+        OutfitController.countTotalOutfits().thenAccept(count -> embed.addField("Total outifts:", String.valueOf(count))).join();
+        OutfitController.countFeaturedOutfits().thenAccept(count -> embed.addField("Featured outfits:", String.valueOf(count))).join();
+
+        ctx.reply(embed);
+    }
+}
\ No newline at end of file
diff --git a/src/main/java/dev/salmonllama/fsbot/guthix/Guthix.java b/src/main/java/dev/salmonllama/fsbot/guthix/Guthix.java
index 186e549..f6c6a9f 100644
--- a/src/main/java/dev/salmonllama/fsbot/guthix/Guthix.java
+++ b/src/main/java/dev/salmonllama/fsbot/guthix/Guthix.java
@@ -22,9 +22,10 @@ import java.util.Collection;
 import java.util.HashMap;
 
 /*
-* Guthix is Fashionscape Bot's command repository and dispatcher
+ * Guthix is Fashionscape Bot's command repository and dispatcher
  */
 public class Guthix implements MessageCreateListener {
+    @SuppressWarnings("unused")
     private DiscordApi api;
     private DatabaseUtilities db;
 
@@ -68,6 +69,7 @@ public class Guthix implements MessageCreateListener {
         addCommand(new ColorsCommand());
         addCommand(new OutfitCommand());
         addCommand(new HelpCommand(this));
+        addCommand(new StatsCommand());
     }
 
     public void addCommand(Command cmd) {