90 lines
3.2 KiB
Java
90 lines
3.2 KiB
Java
package com.djrapitops.plan.command.commands.manage;
|
|
|
|
import com.djrapitops.plan.PlanPlugin;
|
|
import com.djrapitops.plan.system.database.DBSystem;
|
|
import com.djrapitops.plan.system.database.databases.Database;
|
|
import com.djrapitops.plan.system.locale.Locale;
|
|
import com.djrapitops.plan.system.locale.lang.CmdHelpLang;
|
|
import com.djrapitops.plan.system.locale.lang.CommandLang;
|
|
import com.djrapitops.plan.system.locale.lang.ManageLang;
|
|
import com.djrapitops.plan.system.settings.Permissions;
|
|
import com.djrapitops.plan.system.settings.Settings;
|
|
import com.djrapitops.plan.system.settings.config.PlanConfig;
|
|
import com.djrapitops.plugin.command.CommandNode;
|
|
import com.djrapitops.plugin.command.CommandType;
|
|
import com.djrapitops.plugin.command.Sender;
|
|
import com.djrapitops.plugin.logging.L;
|
|
import com.djrapitops.plugin.logging.error.ErrorHandler;
|
|
import com.djrapitops.plugin.utilities.Verify;
|
|
|
|
import javax.inject.Inject;
|
|
import java.io.IOException;
|
|
import java.util.Arrays;
|
|
|
|
/**
|
|
* This manage SubCommand is used to swap to a different database and reload the
|
|
* plugin if the connection to the new database can be established.
|
|
*
|
|
* @author Rsl1122
|
|
* @since 2.3.0
|
|
*/
|
|
public class ManageHotSwapCommand extends CommandNode {
|
|
|
|
private final PlanPlugin plugin;
|
|
private final Locale locale;
|
|
private final DBSystem dbSystem;
|
|
private final PlanConfig config;
|
|
private final ErrorHandler errorHandler;
|
|
|
|
@Inject
|
|
public ManageHotSwapCommand(PlanPlugin plugin, Locale locale, DBSystem dbSystem, PlanConfig config, ErrorHandler errorHandler) {
|
|
super("hotswap", Permissions.MANAGE.getPermission(), CommandType.PLAYER_OR_ARGS);
|
|
|
|
this.plugin = plugin;
|
|
this.locale = locale;
|
|
this.dbSystem = dbSystem;
|
|
this.config = config;
|
|
this.errorHandler = errorHandler;
|
|
|
|
setArguments("<DB>");
|
|
setShortHelp(locale.getString(CmdHelpLang.MANAGE_HOTSWAP));
|
|
}
|
|
|
|
@Override
|
|
public void onCommand(Sender sender, String commandLabel, String[] args) {
|
|
Verify.isTrue(args.length >= 1,
|
|
() -> new IllegalArgumentException(locale.getString(CommandLang.FAIL_REQ_ONE_ARG, Arrays.toString(this.getArguments()))));
|
|
|
|
String dbName = args[0].toLowerCase();
|
|
|
|
boolean isCorrectDB = Verify.equalsOne(dbName, "sqlite", "mysql");
|
|
Verify.isTrue(isCorrectDB,
|
|
() -> new IllegalArgumentException(locale.getString(ManageLang.FAIL_INCORRECT_DB, dbName)));
|
|
|
|
Verify.isFalse(dbName.equals(dbSystem.getDatabase().getConfigName()),
|
|
() -> new IllegalArgumentException(locale.getString(ManageLang.FAIL_SAME_DB)));
|
|
|
|
try {
|
|
Database database = dbSystem.getActiveDatabaseByName(dbName);
|
|
database.init();
|
|
|
|
if (!database.isOpen()) {
|
|
return;
|
|
}
|
|
} catch (Exception e) {
|
|
errorHandler.log(L.ERROR, this.getClass(), e);
|
|
sender.sendMessage(locale.getString(ManageLang.PROGRESS_FAIL, e.getMessage()));
|
|
return;
|
|
}
|
|
|
|
try {
|
|
config.set(Settings.DB_TYPE, dbName);
|
|
config.save();
|
|
} catch (IOException e) {
|
|
errorHandler.log(L.ERROR, this.getClass(), e);
|
|
return;
|
|
}
|
|
plugin.reloadPlugin(true);
|
|
}
|
|
}
|