Fix database creation on first startup

Author: Innectic

API/src/main/java/me/innectic/permissify/api/PermissifyAPI.java

@@ -58,10 +58,13 @@ public void initialize(HandlerType type, Optional<ConnectionInformation> connect
             e.printStackTrace();
         }
         if (!databaseHandler.isPresent()) throw new Exception("No data handler present.");
-        boolean connected = databaseHandler.get().connect();
-        if (connected) System.out.println("Connected to the database.");
-        else System.out.println("Unable to connect to the database.");
-        databaseHandler.ifPresent(handler -> handler.clear(new ArrayList<>()));
+        databaseHandler.ifPresent(handler -> {
+            handler.initialize();
+            handler.clear(new ArrayList<>());
+            boolean connected = handler.connect();
+            if (connected) System.out.println("Connected to the database.");
+            else System.out.println("Unable to connect to the database.");
+        });
     }
 
     public static Optional<PermissifyAPI> get() {

API/src/main/java/me/innectic/permissify/api/database/handlers/MySQLHandler.java

@@ -66,6 +66,59 @@ private Optional<Connection> getConnection() {
     public void initialize() {
         // Make sure that the cache is empty
         this.cachedPermissions = new HashMap<>();
+        // Make sure the database needed actually exists.
+        if (!connectionInformation.isPresent()) return;
+
+        try {
+            String connectionURL = "jdbc:mysql://" + connectionInformation.get().getUrl() + ":" + connectionInformation.get().getPort();
+            Optional<Connection> connection = Optional.ofNullable(DriverManager.getConnection(connectionURL, connectionInformation.get().getUsername(), connectionInformation.get().getPassword()));
+            if (!connection.isPresent()) return;
+            String database = connectionInformation.get().getDatabase();
+            // TODO: This should all be prepared statements, but that breaks for some reason
+            PreparedStatement databaseStatement = connection.get().prepareStatement("CREATE DATABASE IF NOT EXISTS " + database);
+            databaseStatement.execute();
+            databaseStatement.close();
+
+            PreparedStatement formattingStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database + ".formatting (`format` VARCHAR(400) NOT NULL, formatter VARCHAR(200) NOT NULL)");
+            formattingStatement.execute();
+            formattingStatement.close();
+
+            PreparedStatement groupMembersStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database + ".groupMembers (uuid VARCHAR(767) NOT NULL, `group` VARCHAR(700) NOT NULL, `primary` TINYINT NOT NULL)");
+            groupMembersStatement.execute();
+            groupMembersStatement.close();
+
+            PreparedStatement groupPermissionsStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database +".groupPermissions (groupName VARCHAR(767) NOT NULL, permission VARCHAR(767) NOT NULL)");
+            groupPermissionsStatement.execute();
+            groupPermissionsStatement.close();
+
+            PreparedStatement groupsStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database + ".groups (name VARCHAR(100) NOT NULL UNIQUE, prefix VARCHAR(100) NOT NULL, suffix VARCHAR(100) NOT NULL, chatcolor VARCHAR(4) NOT NULL, defaultGroup TINYINT NOT NULL)");
+            groupsStatement.execute();
+            groupsStatement.close();
+
+            PreparedStatement playerPermissionsStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database + ".playerPermissions (uuid VARCHAR(767) NOT NULL, permission VARCHAR(767) NOT NULL, granted TINYINT NOT NULL)");
+            playerPermissionsStatement.execute();
+            playerPermissionsStatement.close();
+
+            PreparedStatement superAdminStatement = connection.get().prepareStatement("CREATE TABLE IF NOT EXISTS " + database + ".superAdmin (uuid VARCHAR(767) NOT NULL)");
+            superAdminStatement.execute();
+            superAdminStatement.close();
+
+            PreparedStatement defaultChatFormat = connection.get().prepareStatement("INSERT INTO " + database + ".formatting (`format`, formatter) VALUES (?,?)");
+            defaultChatFormat.setString(1, "{group} {username}: {message}");
+            defaultChatFormat.setString(2, "chat");
+            defaultChatFormat.execute();
+            defaultChatFormat.close();
+
+            PreparedStatement defaultWhisperFormat = connection.get().prepareStatement("INSERT INTO " + database + ".formatting (`format`, formatter) VALUES (?,?)");
+            defaultWhisperFormat.setString(1, "{senderGroup} {username} > {receiverGroup} {to}: {message}");
+            defaultWhisperFormat.setString(2, "whisper");
+            defaultWhisperFormat.execute();
+            defaultWhisperFormat.close();
+
+            connection.get().close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
     }
 
     @Override

Spigot/src/main/java/me/innectic/permissify/spigot/commands/permissify/CacheCommand.java

@@ -26,6 +26,7 @@
 
 import me.innectic.permissify.api.PermissifyConstants;
 import me.innectic.permissify.api.database.DatabaseHandler;
+import me.innectic.permissify.api.permission.PermissionGroup;
 import me.innectic.permissify.api.util.ArgumentUtil;
 import me.innectic.permissify.spigot.PermissifyMain;
 import me.innectic.permissify.spigot.commands.CommandResponse;
@@ -51,8 +52,8 @@ public CommandResponse handleCache(CommandSender sender, String[] args) {
             return new CommandResponse(PermissifyConstants.UNABLE_OTHER.replace("<REASON>", "No database handler"), false);
         // Show information about the current cache
         DatabaseHandler handler = PermissifyMain.getInstance().getPermissifyAPI().getDatabaseHandler().get();
-        String cacheInformation = String.format("Cached groups: %d, cached permissions: %d", handler.getCachedGroups().size(),
-                handler.getCachedPermissions().size());
+        String cacheInformation = String.format("Cached groups: %d, cached permissions: %d, default group: %s", handler.getCachedGroups().size(),
+                handler.getCachedPermissions().size(), handler.getDefaultGroup().map(PermissionGroup::getName).orElse("NONE"));
         return new CommandResponse(cacheInformation, false);
     }
 

Spigot/src/main/java/me/innectic/permissify/spigot/commands/permissify/FormatCommand.java

@@ -69,12 +69,14 @@ private CommandResponse handleWhisperFormat(CommandSender sender, String[] args)
 
     private CommandResponse handleDisableFormat(CommandSender sender, String[] args) {
         PermissifyMain.getInstance().getConfig().set("handleChat", false);
+        PermissifyMain.getInstance().saveConfig();
         PermissifyMain.getInstance().setHandleChat(false);
         return new CommandResponse(PermissifyConstants.TOGGLED_CHAT_HANDLE.replace("<STATE>", "Disabled"), true);
     }
 
     private CommandResponse handleEnableFormat(CommandSender sender, String[] args) {
         PermissifyMain.getInstance().getConfig().set("handleChat", true);
+        PermissifyMain.getInstance().saveConfig();
         PermissifyMain.getInstance().setHandleChat(true);
         return new CommandResponse(PermissifyConstants.TOGGLED_CHAT_HANDLE.replace("<STATE>", "Enabled"), true);
     }