Allow Custom Commands
This commit is contained in:
parent
72e2c4947d
commit
5ca2a63b5e
@ -3,6 +3,7 @@
|
|||||||
#include <symbols/minecraft.h>
|
#include <symbols/minecraft.h>
|
||||||
#include <mods/chat/chat.h>
|
#include <mods/chat/chat.h>
|
||||||
#include <mods/misc/misc.h>
|
#include <mods/misc/misc.h>
|
||||||
|
#include <mods/server/server.h>
|
||||||
|
|
||||||
// The Actual Mod
|
// The Actual Mod
|
||||||
HOOK(chat_handle_packet_send, void, (const Minecraft *minecraft, ChatPacket *packet)) {
|
HOOK(chat_handle_packet_send, void, (const Minecraft *minecraft, ChatPacket *packet)) {
|
||||||
@ -23,3 +24,20 @@ HOOK(chat_handle_packet_send, void, (const Minecraft *minecraft, ChatPacket *pac
|
|||||||
real_chat_handle_packet_send(minecraft, packet);
|
real_chat_handle_packet_send(minecraft, packet);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Add Server Command
|
||||||
|
HOOK(server_get_commands, std::vector<ServerCommand> *, (Minecraft *minecraft, ServerSideNetworkHandler *server_side_network_handler)) {
|
||||||
|
// Call Original Method
|
||||||
|
ensure_server_get_commands();
|
||||||
|
std::vector<ServerCommand> *commands = real_server_get_commands(minecraft, server_side_network_handler);
|
||||||
|
// Add Command
|
||||||
|
commands->push_back({
|
||||||
|
.name = "greet",
|
||||||
|
.comment = "Example Custom Command",
|
||||||
|
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
INFO("Hello World!");
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Return
|
||||||
|
return commands;
|
||||||
|
}
|
@ -1,7 +1,19 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#include <symbols/minecraft.h>
|
||||||
|
|
||||||
#include "server_properties.h"
|
#include "server_properties.h"
|
||||||
|
|
||||||
|
struct ServerCommand {
|
||||||
|
const std::string name;
|
||||||
|
const std::string comment;
|
||||||
|
const std::function<void(const std::string &)> callback;
|
||||||
|
[[nodiscard]] bool has_args() const;
|
||||||
|
[[nodiscard]] std::string get_lhs_help() const;
|
||||||
|
[[nodiscard]] std::string get_full_help(int max_lhs_length) const;
|
||||||
|
};
|
||||||
|
|
||||||
extern "C" {
|
extern "C" {
|
||||||
|
std::vector<ServerCommand> *server_get_commands(Minecraft *minecraft, ServerSideNetworkHandler *server_side_network_handler);
|
||||||
ServerProperties &get_server_properties();
|
ServerProperties &get_server_properties();
|
||||||
}
|
}
|
@ -271,14 +271,10 @@ static void handle_server_stop(Minecraft *minecraft) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Handle Commands
|
// Handle Commands
|
||||||
struct Command {
|
bool ServerCommand::has_args() const {
|
||||||
const std::string name;
|
|
||||||
const std::string comment;
|
|
||||||
const std::function<void(const std::string &)> callback;
|
|
||||||
[[nodiscard]] bool has_args() const {
|
|
||||||
return name[name.length() - 1] == ' ';
|
return name[name.length() - 1] == ' ';
|
||||||
}
|
}
|
||||||
[[nodiscard]] std::string get_lhs_help() const {
|
std::string ServerCommand::get_lhs_help() const {
|
||||||
std::string out;
|
std::string out;
|
||||||
out.append(4, ' ');
|
out.append(4, ' ');
|
||||||
out += name;
|
out += name;
|
||||||
@ -286,15 +282,104 @@ struct Command {
|
|||||||
out += "<Arguments>";
|
out += "<Arguments>";
|
||||||
}
|
}
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
[[nodiscard]] std::string get_full_help(const int max_lhs_length) const {
|
std::string ServerCommand::get_full_help(const int max_lhs_length) const {
|
||||||
std::string out = get_lhs_help();
|
std::string out = get_lhs_help();
|
||||||
out.append(max_lhs_length - out.length(), ' ');
|
out.append(max_lhs_length - out.length(), ' ');
|
||||||
out += " - ";
|
out += " - ";
|
||||||
out += comment;
|
out += comment;
|
||||||
return out;
|
return out;
|
||||||
|
}
|
||||||
|
std::vector<ServerCommand> *server_get_commands(Minecraft *minecraft, ServerSideNetworkHandler *server_side_network_handler) {
|
||||||
|
std::vector<ServerCommand> *commands = new std::vector<ServerCommand>;
|
||||||
|
// Ban Player
|
||||||
|
if (!is_whitelist()) {
|
||||||
|
commands->push_back({
|
||||||
|
.name = "ban ",
|
||||||
|
.comment = "IP-Ban All Players With Specified Username",
|
||||||
|
.callback = [minecraft](const std::string &cmd) {
|
||||||
|
find_players(minecraft, cmd, ban_callback, false);
|
||||||
}
|
}
|
||||||
};
|
});
|
||||||
|
}
|
||||||
|
// Reload White/Blacklist
|
||||||
|
commands->push_back({
|
||||||
|
.name = "reload",
|
||||||
|
.comment = std::string("Reload The ") + (is_whitelist() ? "Whitelist" : "Blacklist"),
|
||||||
|
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
INFO("Reloading %s", is_whitelist() ? "Whitelist" : "Blacklist");
|
||||||
|
is_ip_in_blacklist(nullptr);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Kill Player
|
||||||
|
commands->push_back({
|
||||||
|
.name = "kill ",
|
||||||
|
.comment = "Kill All Players With Specified Username",
|
||||||
|
.callback = [minecraft](const std::string &cmd) {
|
||||||
|
find_players(minecraft, cmd, kill_callback, false);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Post Message
|
||||||
|
commands->push_back({
|
||||||
|
.name = "say ",
|
||||||
|
.comment = "Print Specified Message To Chat",
|
||||||
|
.callback = [server_side_network_handler](const std::string &cmd) {
|
||||||
|
// Format Message
|
||||||
|
const std::string message = "[Server] " + cmd;
|
||||||
|
char *safe_message = to_cp437(message.c_str());
|
||||||
|
std::string cpp_string = safe_message;
|
||||||
|
// Post Message To Chat
|
||||||
|
server_side_network_handler->displayGameMessage(cpp_string);
|
||||||
|
// Free
|
||||||
|
free(safe_message);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// List Players
|
||||||
|
commands->push_back({
|
||||||
|
.name = "list",
|
||||||
|
.comment = "List All Players",
|
||||||
|
.callback = [minecraft](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
INFO("All Players:");
|
||||||
|
find_players(minecraft, "", list_callback, true);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Ticks-Per-Second
|
||||||
|
commands->push_back({
|
||||||
|
.name = "tps",
|
||||||
|
.comment = "Print TPS",
|
||||||
|
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
INFO("TPS: %f", tps);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Stop
|
||||||
|
commands->push_back({
|
||||||
|
.name = "stop",
|
||||||
|
.comment = "Stop Server",
|
||||||
|
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
compat_request_exit();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Help Page
|
||||||
|
commands->push_back({
|
||||||
|
.name = "help",
|
||||||
|
.comment = "Print This Message",
|
||||||
|
.callback = [commands](__attribute__((unused)) const std::string &cmd) {
|
||||||
|
INFO("All Commands:");
|
||||||
|
int max_lhs_length = 0;
|
||||||
|
for (ServerCommand command : *commands) {
|
||||||
|
const int lhs_length = command.get_lhs_help().length();
|
||||||
|
if (lhs_length > max_lhs_length) {
|
||||||
|
max_lhs_length = lhs_length;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (ServerCommand command : *commands) {
|
||||||
|
INFO("%s", command.get_full_help(max_lhs_length).c_str());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
// Return
|
||||||
|
return commands;
|
||||||
|
}
|
||||||
static void handle_commands(Minecraft *minecraft) {
|
static void handle_commands(Minecraft *minecraft) {
|
||||||
// Check If Level Is Generated
|
// Check If Level Is Generated
|
||||||
if (minecraft->isLevelGenerated() && stdin_line_ready) {
|
if (minecraft->isLevelGenerated() && stdin_line_ready) {
|
||||||
@ -306,87 +391,10 @@ static void handle_commands(Minecraft *minecraft) {
|
|||||||
ServerSideNetworkHandler *server_side_network_handler = get_server_side_network_handler(minecraft);
|
ServerSideNetworkHandler *server_side_network_handler = get_server_side_network_handler(minecraft);
|
||||||
if (server_side_network_handler != nullptr) {
|
if (server_side_network_handler != nullptr) {
|
||||||
// Generate Command List
|
// Generate Command List
|
||||||
std::vector<Command> commands;
|
std::vector<ServerCommand> *commands = server_get_commands(minecraft, server_side_network_handler);
|
||||||
if (!is_whitelist()) {
|
|
||||||
commands.push_back({
|
|
||||||
.name = "ban ",
|
|
||||||
.comment = "IP-Ban All Players With Specified Username",
|
|
||||||
.callback = [&minecraft](const std::string &cmd) {
|
|
||||||
find_players(minecraft, cmd, ban_callback, false);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
commands.push_back({
|
|
||||||
.name = "reload",
|
|
||||||
.comment = std::string("Reload The ") + (is_whitelist() ? "Whitelist" : "Blacklist"),
|
|
||||||
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
|
||||||
INFO("Reloading %s", is_whitelist() ? "Whitelist" : "Blacklist");
|
|
||||||
is_ip_in_blacklist(nullptr);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "kill ",
|
|
||||||
.comment = "Kill All Players With Specified Username",
|
|
||||||
.callback = [&minecraft](const std::string &cmd) {
|
|
||||||
find_players(minecraft, cmd, kill_callback, false);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "say ",
|
|
||||||
.comment = "Print Specified Message To Chat",
|
|
||||||
.callback = [&server_side_network_handler](const std::string &cmd) {
|
|
||||||
// Format Message
|
|
||||||
const std::string message = "[Server] " + cmd;
|
|
||||||
char *safe_message = to_cp437(message.c_str());
|
|
||||||
std::string cpp_string = safe_message;
|
|
||||||
// Post Message To Chat
|
|
||||||
server_side_network_handler->displayGameMessage(cpp_string);
|
|
||||||
// Free
|
|
||||||
free(safe_message);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "list",
|
|
||||||
.comment = "List All Players",
|
|
||||||
.callback = [&minecraft](__attribute__((unused)) const std::string &cmd) {
|
|
||||||
INFO("All Players:");
|
|
||||||
find_players(minecraft, "", list_callback, true);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "tps",
|
|
||||||
.comment = "Print TPS",
|
|
||||||
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
|
||||||
INFO("TPS: %f", tps);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "stop",
|
|
||||||
.comment = "Stop Server",
|
|
||||||
.callback = [](__attribute__((unused)) const std::string &cmd) {
|
|
||||||
compat_request_exit();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
commands.push_back({
|
|
||||||
.name = "help",
|
|
||||||
.comment = "Print This Message",
|
|
||||||
.callback = [&commands](__attribute__((unused)) const std::string &cmd) {
|
|
||||||
INFO("All Commands:");
|
|
||||||
int max_lhs_length = 0;
|
|
||||||
for (Command command : commands) {
|
|
||||||
const int lhs_length = command.get_lhs_help().length();
|
|
||||||
if (lhs_length > max_lhs_length) {
|
|
||||||
max_lhs_length = lhs_length;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for (Command command : commands) {
|
|
||||||
INFO("%s", command.get_full_help(max_lhs_length).c_str());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
});
|
|
||||||
// Run
|
// Run
|
||||||
bool success = false;
|
bool success = false;
|
||||||
for (Command command : commands) {
|
for (ServerCommand command : *commands) {
|
||||||
const bool valid = command.has_args() ? data.rfind(command.name, 0) == 0 : data == command.name;
|
const bool valid = command.has_args() ? data.rfind(command.name, 0) == 0 : data == command.name;
|
||||||
if (valid) {
|
if (valid) {
|
||||||
command.callback(data.substr(command.name.length()));
|
command.callback(data.substr(command.name.length()));
|
||||||
@ -397,6 +405,8 @@ static void handle_commands(Minecraft *minecraft) {
|
|||||||
if (!success) {
|
if (!success) {
|
||||||
INFO("Invalid Command: %s", data.c_str());
|
INFO("Invalid Command: %s", data.c_str());
|
||||||
}
|
}
|
||||||
|
// Free
|
||||||
|
delete commands;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user