317 lines
15 KiB
Java
Executable File
317 lines
15 KiB
Java
Executable File
package uk.co.angrybee.joe;
|
|
|
|
import org.bukkit.Server;
|
|
import org.bukkit.configuration.file.FileConfiguration;
|
|
import org.bukkit.plugin.java.JavaPlugin;
|
|
import com.earth2me.essentials.Essentials;
|
|
import org.kitteh.vanish.VanishPlugin;
|
|
import org.bukkit.plugin.PluginManager;
|
|
import uk.co.angrybee.joe.commands.minecraft.CommandAbout;
|
|
import uk.co.angrybee.joe.commands.minecraft.CommandReload;
|
|
import uk.co.angrybee.joe.commands.minecraft.CommandStatus;
|
|
import uk.co.angrybee.joe.configs.*;
|
|
import uk.co.angrybee.joe.events.JoinLeaveEvents;
|
|
import uk.co.angrybee.joe.events.OnBanEvent;
|
|
import uk.co.angrybee.joe.events.OnWhitelistEvents;
|
|
import uk.co.angrybee.joe.stores.InGameRemovedList;
|
|
import uk.co.angrybee.joe.stores.RemovedList;
|
|
import uk.co.angrybee.joe.stores.UserList;
|
|
import uk.co.angrybee.joe.stores.WhitelistedPlayers;
|
|
import uk.co.angrybee.joe.events.EssentialsVanishEvents;
|
|
import uk.co.angrybee.joe.events.SuperVanishEvents;
|
|
import uk.co.angrybee.joe.events.VanishNoPacketEvents;
|
|
|
|
import java.util.List;
|
|
import java.util.logging.Logger;
|
|
import java.util.stream.Stream;
|
|
|
|
public class DiscordWhitelister extends JavaPlugin {
|
|
public static String botToken;
|
|
|
|
private static boolean configCreated = false;
|
|
|
|
public static boolean initialized = false;
|
|
public static boolean useCustomMessages = false;
|
|
public static boolean useIdForRoles = false;
|
|
public static boolean useCustomPrefixes = false;
|
|
public static boolean showPlayerSkin = true;
|
|
public static boolean showVanishedPlayersInCount = false;
|
|
public static boolean useInGameAddRemoves = true;
|
|
public static boolean useOnBanEvents = true;
|
|
public static boolean useUltraPerms = false;
|
|
public static boolean useLuckPerms = false;
|
|
public static boolean useOnWhitelistCommands = false;
|
|
public static boolean removeUnnecessaryMessages = false;
|
|
|
|
public static boolean botEnabled;
|
|
|
|
public static String[] bannedRoles;
|
|
|
|
private static JavaPlugin thisPlugin;
|
|
private static Server thisServer;
|
|
private static Logger pluginLogger;
|
|
|
|
// Plugins
|
|
public static Essentials essentialsPlugin;
|
|
public static VanishPlugin vanishNoPacketPlugin;
|
|
public static boolean hasSuperVanishOrPremiumVanish;
|
|
|
|
// For not counting vanished players when other players join/leave
|
|
private static int vanishedPlayersCount;
|
|
|
|
public static int removeMessageWaitTime = 5;
|
|
|
|
public static MainConfig mainConfig;
|
|
|
|
public static CustomPrefixConfig customPrefixConfig;
|
|
public static CustomMessagesConfig customMessagesConfig;
|
|
public static PermissionsConfig permissionsConfig;
|
|
public static OnWhitelistCommandsConfig onWhitelistCommandsConfig;
|
|
|
|
|
|
@Override
|
|
public void onEnable() {
|
|
thisPlugin = this;
|
|
thisServer = thisPlugin.getServer();
|
|
pluginLogger = thisPlugin.getLogger();
|
|
|
|
// Get/check for plugin
|
|
PluginManager pluginManager = getServer().getPluginManager();
|
|
essentialsPlugin = (Essentials) pluginManager.getPlugin("Essentials");
|
|
vanishNoPacketPlugin = (VanishPlugin) pluginManager.getPlugin("VanishNoPacket");
|
|
hasSuperVanishOrPremiumVanish = pluginManager.getPlugin("SuperVanish") != null || pluginManager.getPlugin("PremiumVanish") != null;
|
|
|
|
int initSuccess = InitBot(true);
|
|
|
|
if (initSuccess == 0) {
|
|
pluginLogger.info("Successfully initialized Discord client");
|
|
initialized = true;
|
|
} else if (initSuccess == 1) {
|
|
pluginLogger.severe("Discord Client failed to initialize, please check if your config file is valid");
|
|
initialized = false;
|
|
return;
|
|
}
|
|
|
|
// Check for leavers if enabled
|
|
if (botEnabled) {
|
|
DiscordClient.ServerLeaveStartupCheck();
|
|
DiscordClient.RequiredRoleStartupCheck();
|
|
}
|
|
|
|
this.getCommand("discordwhitelister").setExecutor(new CommandStatus());
|
|
this.getCommand("discordwhitelisterabout").setExecutor(new CommandAbout());
|
|
this.getCommand("discordwhitelisterreload").setExecutor(new CommandReload());
|
|
}
|
|
|
|
@Override
|
|
public void onDisable() {
|
|
if (initialized ) {
|
|
DiscordClient.javaDiscordAPI.shutdownNow();
|
|
}
|
|
}
|
|
|
|
public static JavaPlugin getPlugin() {
|
|
return thisPlugin;
|
|
}
|
|
|
|
public static FileConfiguration getCustomMessagesConfig() {
|
|
return customMessagesConfig.getFileConfiguration();
|
|
}
|
|
|
|
public static Logger getPluginLogger() {
|
|
return pluginLogger;
|
|
}
|
|
|
|
public static boolean getUseCustomPrefixes() {
|
|
return useCustomPrefixes;
|
|
}
|
|
|
|
public static void addVanishedPlayer() {
|
|
vanishedPlayersCount++;
|
|
}
|
|
|
|
public static void removeVanishedPlayer() {
|
|
vanishedPlayersCount--;
|
|
}
|
|
|
|
public static int getOnlineUsers() {
|
|
return thisPlugin.getServer().getOnlinePlayers().size() - vanishedPlayersCount;
|
|
}
|
|
|
|
public static int getMaximumAllowedPlayers() {
|
|
return thisPlugin.getServer().getMaxPlayers();
|
|
}
|
|
|
|
public static int InitBot(boolean firstInit) {
|
|
if (firstInit)
|
|
vanishedPlayersCount = 0;
|
|
|
|
ConfigSetup();
|
|
|
|
botToken = mainConfig.getFileConfiguration().getString("discord-bot-token");
|
|
botEnabled = mainConfig.getFileConfiguration().getBoolean("bot-enabled");
|
|
showPlayerSkin = mainConfig.getFileConfiguration().getBoolean("show-player-skin-on-whitelist");
|
|
configCreated = mainConfig.fileCreated;
|
|
showVanishedPlayersInCount = mainConfig.getFileConfiguration().getBoolean("show-vanished-players-in-player-count");
|
|
useInGameAddRemoves = mainConfig.getFileConfiguration().getBoolean("add-in-game-adds-and-removes-to-list");
|
|
useOnBanEvents = mainConfig.getFileConfiguration().getBoolean("use-on-ban-events");
|
|
removeUnnecessaryMessages = mainConfig.getFileConfiguration().getBoolean("remove-unnecessary-messages-from-whitelist-channel");
|
|
removeMessageWaitTime = mainConfig.getFileConfiguration().getInt("seconds-to-remove-message-from-whitelist-channel");
|
|
useOnWhitelistCommands = mainConfig.getFileConfiguration().getBoolean("use-on-whitelist-commands");
|
|
|
|
if (!botEnabled) {
|
|
pluginLogger.info("Bot is disabled as per the config, doing nothing");
|
|
} else if (configCreated || botToken.equals(MainConfig.default_token)) {
|
|
pluginLogger.warning("Config newly created. Please paste your bot token into the config file, doing nothing until next server start");
|
|
} else {
|
|
// Check for LuckPerms first
|
|
if (mainConfig.getFileConfiguration().getBoolean("assign-perms-with-luck-perms")) {
|
|
if (getPlugin().getServer().getPluginManager().getPlugin("LuckPerms") != null) {
|
|
useLuckPerms = true;
|
|
getPluginLogger().info("LuckPerms found!");
|
|
} else {
|
|
getPluginLogger().warning("LuckPerms was not found but is enabled in the config. Doing nothing...");
|
|
useLuckPerms = false;
|
|
}
|
|
}
|
|
if (mainConfig.getFileConfiguration().getBoolean("assign-perms-with-ultra-perms")) {
|
|
if (getPlugin().getServer().getPluginManager().getPlugin("UltraPermissions") != null) {
|
|
useUltraPerms = true;
|
|
getPluginLogger().info("Ultra Permissions found!");
|
|
} else {
|
|
getPluginLogger().warning("Ultra Permissions was not found but is enabled in the config. Doing nothing...");
|
|
useUltraPerms = false;
|
|
}
|
|
}
|
|
|
|
// TODO: remove in favour of split versions
|
|
DiscordClient.customWhitelistAddPrefix = customPrefixConfig.getFileConfiguration().getString("whitelist-add-prefix").toLowerCase();
|
|
DiscordClient.customWhitelistRemovePrefix = customPrefixConfig.getFileConfiguration().getString("whitelist-remove-prefix").toLowerCase();
|
|
DiscordClient.customClearNamePrefix = customPrefixConfig.getFileConfiguration().getString("clear-name-prefix").toLowerCase();
|
|
DiscordClient.customLimitedWhitelistClearPrefix = customPrefixConfig.getFileConfiguration().getString("limited-whitelist-clear-prefix").toLowerCase();
|
|
DiscordClient.customClearBanPrefix = customPrefixConfig.getFileConfiguration().getString("clear-ban-prefix").toLowerCase();
|
|
|
|
// Split versions
|
|
DiscordClient.customWhitelistAddPrefixSplit = customPrefixConfig.getFileConfiguration().getString("whitelist-add-prefix").toLowerCase().trim().split(" ");
|
|
DiscordClient.customWhitelistRemovePrefixSplit = customPrefixConfig.getFileConfiguration().getString("whitelist-remove-prefix").toLowerCase().trim().split(" ");
|
|
DiscordClient.customClearNamePrefixSplit = customPrefixConfig.getFileConfiguration().getString("clear-name-prefix").toLowerCase().trim().split(" ");
|
|
DiscordClient.customLimitedWhitelistClearPrefixSplit = customPrefixConfig.getFileConfiguration().getString("limited-whitelist-clear-prefix").toLowerCase().trim().split(" ");
|
|
DiscordClient.customClearBanPrefixSplit = customPrefixConfig.getFileConfiguration().getString("clear-ban-prefix").toLowerCase().trim().split(" ");
|
|
DiscordClient.customWhoIsPrefix = customPrefixConfig.getFileConfiguration().getString("whitelist-whois-prefix").toLowerCase().trim().split(" ");
|
|
|
|
|
|
pluginLogger.info("Initializing Discord client...");
|
|
|
|
// TODO: below role section could be moved to DiscordClient class
|
|
useIdForRoles = mainConfig.getFileConfiguration().getBoolean("use-id-for-roles");
|
|
|
|
// set add & remove roles
|
|
DiscordClient.allowedToAddRemoveRoles = new String[mainConfig.getFileConfiguration().getList("add-remove-roles").size()];
|
|
for (int roles = 0; roles < DiscordClient.allowedToAddRemoveRoles.length; ++roles) {
|
|
DiscordClient.allowedToAddRemoveRoles[roles] = mainConfig.getFileConfiguration().getList("add-remove-roles").get(roles).toString();
|
|
}
|
|
|
|
// set add roles
|
|
DiscordClient.allowedToAddRoles = new String[mainConfig.getFileConfiguration().getList("add-roles").size()];
|
|
for (int roles = 0; roles < DiscordClient.allowedToAddRoles.length; ++roles) {
|
|
DiscordClient.allowedToAddRoles[roles] = mainConfig.getFileConfiguration().getList("add-roles").get(roles).toString();
|
|
}
|
|
|
|
// set limited add roles
|
|
DiscordClient.allowedToAddLimitedRoles = new String[mainConfig.getFileConfiguration().getList("limited-add-roles").size()];
|
|
for (int roles = 0; roles < DiscordClient.allowedToAddLimitedRoles.length; ++roles) {
|
|
DiscordClient.allowedToAddLimitedRoles[roles] = mainConfig.getFileConfiguration().getList("limited-add-roles").get(roles).toString();
|
|
}
|
|
|
|
// Get banned roles
|
|
if (useOnBanEvents) {
|
|
List<String> tempBannedRoles = mainConfig.getFileConfiguration().getStringList("banned-roles");
|
|
bannedRoles = new String[tempBannedRoles.size()];
|
|
for (int i = 0; i < tempBannedRoles.size(); i++) {
|
|
bannedRoles[i] = tempBannedRoles.get(i);
|
|
}
|
|
}
|
|
|
|
// Allowed to clear name roles
|
|
DiscordClient.allowedToClearNamesRoles = new String[mainConfig.getFileConfiguration().getStringList("clear-command-roles").size()];
|
|
for (int roles = 0; roles < DiscordClient.allowedToClearNamesRoles.length; roles++) {
|
|
DiscordClient.allowedToClearNamesRoles[roles] = mainConfig.getFileConfiguration().getStringList("clear-command-roles").get(roles);
|
|
}
|
|
|
|
// All roles combined for role check
|
|
DiscordClient.combinedRoles = Stream.of(DiscordClient.allowedToAddRemoveRoles, DiscordClient.allowedToAddRoles,
|
|
DiscordClient.allowedToAddLimitedRoles, DiscordClient.allowedToClearNamesRoles)
|
|
.flatMap(Stream::of).toArray(String[]::new);
|
|
|
|
// Custom messages check
|
|
useCustomMessages = mainConfig.getFileConfiguration().getBoolean("use-custom-messages");
|
|
useCustomPrefixes = mainConfig.getFileConfiguration().getBoolean("use-custom-prefixes");
|
|
|
|
int initSuccess = DiscordClient.InitializeClient(botToken);
|
|
|
|
if (initSuccess == 1)
|
|
return 1;
|
|
|
|
|
|
// No need for an if here statement anymore as this code will not run if the client has not been initialized
|
|
// Only attempt to set player count if the bot successfully initialized
|
|
if (mainConfig.getFileConfiguration().getBoolean("show-player-count")) {
|
|
if (firstInit) {
|
|
// Register events if enabled
|
|
thisServer.getPluginManager().registerEvents(new JoinLeaveEvents(), thisPlugin);
|
|
//pluginLogger.info("Registered join/leave events!");
|
|
if (hasSuperVanishOrPremiumVanish) {
|
|
thisServer.getPluginManager().registerEvents(new SuperVanishEvents(), thisPlugin);
|
|
//pluginLogger.info("Registered SuperVanish events!");
|
|
}
|
|
if (vanishNoPacketPlugin != null) {
|
|
thisServer.getPluginManager().registerEvents(new VanishNoPacketEvents(), thisPlugin);
|
|
//pluginLogger.info("Registered VanishNoPacket events!");
|
|
}
|
|
if (essentialsPlugin != null) {
|
|
thisServer.getPluginManager().registerEvents(new EssentialsVanishEvents(), thisPlugin);
|
|
//pluginLogger.info("Registered Essentials vanish events!");
|
|
}
|
|
}
|
|
|
|
// Set initial player count
|
|
DiscordClient.SetPlayerCountStatus(getOnlineUsers());
|
|
}
|
|
|
|
// Register whitelist events if enabled
|
|
if (useInGameAddRemoves)
|
|
thisServer.getPluginManager().registerEvents(new OnWhitelistEvents(), thisPlugin);
|
|
|
|
// Register ban events if enabled
|
|
if (useOnBanEvents)
|
|
thisServer.getPluginManager().registerEvents(new OnBanEvent(), thisPlugin);
|
|
|
|
return 0;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
public static void ConfigSetup() {
|
|
mainConfig = new MainConfig();
|
|
customPrefixConfig = new CustomPrefixConfig();
|
|
customMessagesConfig = new CustomMessagesConfig();
|
|
permissionsConfig = new PermissionsConfig();
|
|
onWhitelistCommandsConfig = new OnWhitelistCommandsConfig();
|
|
|
|
// Run this first, as it creates the root folder if it does not exist
|
|
mainConfig.ConfigSetup();
|
|
customPrefixConfig.ConfigSetup();
|
|
customMessagesConfig.ConfigSetup();
|
|
permissionsConfig.ConfigSetup();
|
|
onWhitelistCommandsConfig.ConfigSetup();
|
|
|
|
// Init Stores
|
|
UserList.StoreSetup();
|
|
InGameRemovedList.StoreSetup();
|
|
RemovedList.StoreSetup();
|
|
|
|
WhitelistedPlayers.Setup();
|
|
}
|
|
}
|