2020-06-27 15:36:48 +00:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2019-2020 GeyserMC. http://geysermc.org
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*
|
|
|
|
* @author GeyserMC
|
|
|
|
* @link https://github.com/GeyserMC/Geyser
|
|
|
|
*/
|
|
|
|
|
|
|
|
package org.geysermc.connector.command.defaults;
|
|
|
|
|
|
|
|
import com.fasterxml.jackson.databind.JsonNode;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
|
|
import com.fasterxml.jackson.databind.ser.impl.SimpleBeanPropertyFilter;
|
|
|
|
import com.fasterxml.jackson.databind.ser.impl.SimpleFilterProvider;
|
2020-06-27 22:27:00 +00:00
|
|
|
import org.geysermc.connector.common.ChatColor;
|
2020-06-27 15:36:48 +00:00
|
|
|
import org.geysermc.connector.GeyserConnector;
|
|
|
|
import org.geysermc.connector.command.CommandSender;
|
|
|
|
import org.geysermc.connector.command.GeyserCommand;
|
2020-08-15 20:06:50 +00:00
|
|
|
import org.geysermc.connector.common.serializer.AsteriskSerializer;
|
2020-06-27 15:36:48 +00:00
|
|
|
import org.geysermc.connector.dump.DumpInfo;
|
2020-07-05 23:35:51 +00:00
|
|
|
import org.geysermc.connector.utils.LanguageUtils;
|
2020-06-27 15:36:48 +00:00
|
|
|
import org.geysermc.connector.utils.WebUtils;
|
|
|
|
|
2020-08-15 20:06:50 +00:00
|
|
|
import java.io.FileOutputStream;
|
2020-06-27 15:36:48 +00:00
|
|
|
import java.io.IOException;
|
|
|
|
|
|
|
|
public class DumpCommand extends GeyserCommand {
|
|
|
|
|
|
|
|
private final GeyserConnector connector;
|
|
|
|
private static final ObjectMapper MAPPER = new ObjectMapper();
|
|
|
|
private static final String DUMP_URL = "https://dump.geysermc.org/";
|
|
|
|
|
|
|
|
public DumpCommand(GeyserConnector connector, String name, String description, String permission) {
|
|
|
|
super(name, description, permission);
|
|
|
|
|
|
|
|
this.connector = connector;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void execute(CommandSender sender, String[] args) {
|
2020-08-15 20:06:50 +00:00
|
|
|
boolean showSensitive = false;
|
|
|
|
boolean offlineDump = false;
|
|
|
|
if (args.length >= 1) {
|
|
|
|
for (String arg : args) {
|
|
|
|
switch (arg) {
|
|
|
|
case "full":
|
|
|
|
showSensitive = true;
|
|
|
|
break;
|
|
|
|
case "offline":
|
|
|
|
offlineDump = true;
|
|
|
|
break;
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
AsteriskSerializer.showSensitive = showSensitive;
|
|
|
|
|
2020-07-05 23:35:51 +00:00
|
|
|
sender.sendMessage(LanguageUtils.getLocaleStringLog("geyser.commands.dump.collecting"));
|
2020-06-27 15:36:48 +00:00
|
|
|
String dumpData = "";
|
|
|
|
try {
|
2020-08-15 20:06:50 +00:00
|
|
|
if (offlineDump) {
|
|
|
|
dumpData = MAPPER.writerWithDefaultPrettyPrinter().writeValueAsString(new DumpInfo());
|
|
|
|
} else {
|
|
|
|
dumpData = MAPPER.writeValueAsString(new DumpInfo());
|
|
|
|
}
|
2020-06-27 15:36:48 +00:00
|
|
|
} catch (IOException e) {
|
2020-07-05 23:35:51 +00:00
|
|
|
sender.sendMessage(ChatColor.RED + LanguageUtils.getLocaleStringLog("geyser.commands.dump.collect_error"));
|
|
|
|
connector.getLogger().error(LanguageUtils.getLocaleStringLog("geyser.commands.dump.collect_error_short"), e);
|
2020-06-27 15:36:48 +00:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2020-08-15 20:06:50 +00:00
|
|
|
String uploadedDumpUrl = "";
|
2020-06-27 15:36:48 +00:00
|
|
|
|
2020-08-15 20:06:50 +00:00
|
|
|
if (offlineDump) {
|
|
|
|
sender.sendMessage(LanguageUtils.getLocaleStringLog("geyser.commands.dump.writing"));
|
|
|
|
|
|
|
|
try {
|
|
|
|
FileOutputStream outputStream = new FileOutputStream(GeyserConnector.getInstance().getBootstrap().getConfigFolder().resolve("dump.json").toFile());
|
|
|
|
outputStream.write(dumpData.getBytes());
|
|
|
|
outputStream.close();
|
|
|
|
} catch (IOException e) {
|
|
|
|
sender.sendMessage(ChatColor.RED + LanguageUtils.getLocaleStringLog("geyser.commands.dump.write_error"));
|
|
|
|
connector.getLogger().error(LanguageUtils.getLocaleStringLog("geyser.commands.dump.write_error_short"), e);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
uploadedDumpUrl = "dump.json";
|
|
|
|
} else {
|
|
|
|
sender.sendMessage(LanguageUtils.getLocaleStringLog("geyser.commands.dump.uploading"));
|
|
|
|
|
|
|
|
String response;
|
|
|
|
JsonNode responseNode;
|
|
|
|
try {
|
|
|
|
response = WebUtils.post(DUMP_URL + "documents", dumpData);
|
|
|
|
responseNode = MAPPER.readTree(response);
|
|
|
|
} catch (IOException e) {
|
|
|
|
sender.sendMessage(ChatColor.RED + LanguageUtils.getLocaleStringLog("geyser.commands.dump.upload_error"));
|
|
|
|
connector.getLogger().error(LanguageUtils.getLocaleStringLog("geyser.commands.dump.upload_error_short"), e);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!responseNode.has("key")) {
|
|
|
|
sender.sendMessage(ChatColor.RED + LanguageUtils.getLocaleStringLog("geyser.commands.dump.upload_error_short") + ": " + (responseNode.has("message") ? responseNode.get("message").asText() : response));
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
uploadedDumpUrl = DUMP_URL + responseNode.get("key").asText();
|
2020-06-27 15:36:48 +00:00
|
|
|
}
|
|
|
|
|
2020-07-05 23:35:51 +00:00
|
|
|
sender.sendMessage(LanguageUtils.getLocaleStringLog("geyser.commands.dump.message") + " " + ChatColor.DARK_AQUA + uploadedDumpUrl);
|
2020-06-27 15:36:48 +00:00
|
|
|
if (!sender.isConsole()) {
|
2020-07-05 23:35:51 +00:00
|
|
|
connector.getLogger().info(LanguageUtils.getLocaleStringLog("geyser.commands.dump.created", sender.getName(), uploadedDumpUrl));
|
2020-06-27 15:36:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|