forked from ChomeNS/chomens-bot-java
Revive EvalCommand. On this side it is pretty much done. Backend is still very broken as of committing this
This commit is contained in:
parent
d0c46ee143
commit
a3a126f7a8
12 changed files with 298 additions and 1 deletions
|
@ -66,6 +66,7 @@ public class Bot {
|
|||
public BossbarManagerPlugin bossbar;
|
||||
public MusicPlayerPlugin music;
|
||||
public TPSPlugin tps;
|
||||
public EvalPlugin eval;
|
||||
public TrustedPlugin trusted;
|
||||
public BruhifyPlugin bruhify;
|
||||
public CloopPlugin cloop;
|
||||
|
@ -119,6 +120,7 @@ public class Bot {
|
|||
this.bossbar = new BossbarManagerPlugin(this);
|
||||
this.music = new MusicPlayerPlugin(this);
|
||||
this.tps = new TPSPlugin(this);
|
||||
this.eval = new EvalPlugin(this);
|
||||
this.trusted = new TrustedPlugin(this);
|
||||
this.bruhify = new BruhifyPlugin(this);
|
||||
this.cloop = new CloopPlugin(this);
|
||||
|
|
|
@ -26,6 +26,8 @@ public class Configuration {
|
|||
public IRC irc = new IRC();
|
||||
public Music music = new Music();
|
||||
|
||||
public Eval eval = new Eval();
|
||||
|
||||
public ColorPalette colorPalette = new ColorPalette();
|
||||
|
||||
public String ownerName = "chayapak"; // mabe mabe
|
||||
|
@ -125,6 +127,10 @@ public class Configuration {
|
|||
public String error = "#FF0000";
|
||||
}
|
||||
|
||||
public static class Eval {
|
||||
public String address = "ws://localhost:3069";
|
||||
}
|
||||
|
||||
public static class SelfCare {
|
||||
public int delay = 225;
|
||||
|
||||
|
|
|
@ -0,0 +1,58 @@
|
|||
package me.chayapak1.chomens_bot.commands;
|
||||
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.command.Command;
|
||||
import me.chayapak1.chomens_bot.command.CommandContext;
|
||||
import me.chayapak1.chomens_bot.command.CommandException;
|
||||
import me.chayapak1.chomens_bot.command.TrustLevel;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalOutput;
|
||||
import me.chayapak1.chomens_bot.util.ColorUtilities;
|
||||
import net.kyori.adventure.text.Component;
|
||||
import net.kyori.adventure.text.format.NamedTextColor;
|
||||
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
|
||||
public class EvalCommand extends Command {
|
||||
public EvalCommand () {
|
||||
super(
|
||||
"eval",
|
||||
"Evaluate JavaScript codes on a separate Node.JS server running isolated-vm",
|
||||
new String[] { "run <code>", "reset" },
|
||||
new String[] {},
|
||||
TrustLevel.PUBLIC,
|
||||
false
|
||||
);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Component execute(CommandContext context) throws CommandException {
|
||||
final Bot bot = context.bot;
|
||||
|
||||
if (!bot.eval.connected) throw new CommandException(Component.text("Eval server is not online"));
|
||||
|
||||
final String action = context.getString(false, true);
|
||||
|
||||
switch (action) {
|
||||
case "run" -> {
|
||||
final String command = context.getString(true, true);
|
||||
|
||||
final CompletableFuture<EvalOutput> future = bot.eval.run(command);
|
||||
|
||||
future.thenApply((output) -> {
|
||||
if (output.isError()) context.sendOutput(Component.text(output.output()).color(NamedTextColor.RED));
|
||||
else context.sendOutput(Component.text(output.output()));
|
||||
|
||||
return output;
|
||||
});
|
||||
}
|
||||
case "reset" -> {
|
||||
bot.eval.reset();
|
||||
|
||||
return Component.text("Reset the eval worker").color(ColorUtilities.getColorByString(bot.config.colorPalette.defaultColor));
|
||||
}
|
||||
default -> throw new CommandException(Component.text("Invalid action"));
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,29 @@
|
|||
package me.chayapak1.chomens_bot.data.eval;
|
||||
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
|
||||
public class EvalFunction {
|
||||
public final String name;
|
||||
|
||||
protected final Bot bot;
|
||||
|
||||
public EvalFunction (
|
||||
String name,
|
||||
Bot bot
|
||||
) {
|
||||
this.name = name;
|
||||
this.bot = bot;
|
||||
}
|
||||
|
||||
public Output execute (Object ...args) { return null; }
|
||||
|
||||
public static class Output {
|
||||
public final String message;
|
||||
public final boolean parseJSON;
|
||||
|
||||
public Output (String message, boolean parseJSON) {
|
||||
this.message = message;
|
||||
this.parseJSON = parseJSON;
|
||||
}
|
||||
}
|
||||
}
|
|
@ -1,4 +1,4 @@
|
|||
package me.chayapak1.chomens_bot.data;
|
||||
package me.chayapak1.chomens_bot.data.eval;
|
||||
|
||||
public record EvalOutput (
|
||||
boolean isError,
|
|
@ -0,0 +1,19 @@
|
|||
package me.chayapak1.chomens_bot.evalFunctions;
|
||||
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalFunction;
|
||||
|
||||
public class ChatFunction extends EvalFunction {
|
||||
public ChatFunction (Bot bot) {
|
||||
super("chat", bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Output execute(Object... args) {
|
||||
final String message = (String) args[0];
|
||||
|
||||
bot.chat.send(message);
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,19 @@
|
|||
package me.chayapak1.chomens_bot.evalFunctions;
|
||||
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalFunction;
|
||||
|
||||
public class CoreFunction extends EvalFunction {
|
||||
public CoreFunction (Bot bot) {
|
||||
super("core", bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Output execute(Object... args) {
|
||||
final String command = (String) args[0];
|
||||
|
||||
bot.core.run(command);
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,19 @@
|
|||
package me.chayapak1.chomens_bot.evalFunctions;
|
||||
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalFunction;
|
||||
|
||||
public class CorePlaceBlockFunction extends EvalFunction {
|
||||
public CorePlaceBlockFunction (Bot bot) {
|
||||
super("corePlaceBlock", bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Output execute(Object... args) {
|
||||
final String command = (String) args[0];
|
||||
|
||||
bot.core.runPlaceBlock(command);
|
||||
|
||||
return null;
|
||||
}
|
||||
}
|
|
@ -0,0 +1,35 @@
|
|||
package me.chayapak1.chomens_bot.evalFunctions;
|
||||
|
||||
import com.google.gson.JsonArray;
|
||||
import com.google.gson.JsonObject;
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.data.PlayerEntry;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalFunction;
|
||||
import net.kyori.adventure.text.serializer.gson.GsonComponentSerializer;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class GetPlayerListFunction extends EvalFunction {
|
||||
public GetPlayerListFunction(Bot bot) {
|
||||
super("getPlayerList", bot);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Output execute(Object... args) {
|
||||
final List<PlayerEntry> list = bot.players.list;
|
||||
|
||||
final JsonArray array = new JsonArray();
|
||||
|
||||
for (PlayerEntry entry : list) {
|
||||
final JsonObject object = new JsonObject();
|
||||
|
||||
object.addProperty("uuid", entry.profile.getIdAsString());
|
||||
object.addProperty("username", entry.profile.getName());
|
||||
if (entry.displayName != null) object.addProperty("displayName", GsonComponentSerializer.gson().serialize(entry.displayName));
|
||||
|
||||
array.add(object);
|
||||
}
|
||||
|
||||
return new Output(array.toString(), true);
|
||||
}
|
||||
}
|
|
@ -48,6 +48,7 @@ public class CommandHandlerPlugin {
|
|||
registerCommand(new ClearChatQueueCommand());
|
||||
registerCommand(new FilterCommand());
|
||||
registerCommand(new MailCommand());
|
||||
registerCommand(new EvalCommand());
|
||||
registerCommand(new InfoCommand());
|
||||
registerCommand(new ConsoleCommand());
|
||||
registerCommand(new PCrashCommand());
|
||||
|
|
105
src/main/java/me/chayapak1/chomens_bot/plugins/EvalPlugin.java
Normal file
105
src/main/java/me/chayapak1/chomens_bot/plugins/EvalPlugin.java
Normal file
|
@ -0,0 +1,105 @@
|
|||
package me.chayapak1.chomens_bot.plugins;
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import com.google.gson.JsonArray;
|
||||
import io.socket.client.IO;
|
||||
import io.socket.client.Socket;
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalFunction;
|
||||
import me.chayapak1.chomens_bot.data.eval.EvalOutput;
|
||||
import me.chayapak1.chomens_bot.evalFunctions.ChatFunction;
|
||||
import me.chayapak1.chomens_bot.evalFunctions.CoreFunction;
|
||||
import me.chayapak1.chomens_bot.evalFunctions.CorePlaceBlockFunction;
|
||||
import me.chayapak1.chomens_bot.evalFunctions.GetPlayerListFunction;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
|
||||
public class EvalPlugin {
|
||||
public static final String BRIDGE_PREFIX = "function:";
|
||||
|
||||
public boolean connected = false;
|
||||
|
||||
private Socket socket = null;
|
||||
|
||||
private int transactionId = 0;
|
||||
|
||||
private final Map<Integer, CompletableFuture<EvalOutput>> futures = new HashMap<>();
|
||||
|
||||
public final List<EvalFunction> functions = new ArrayList<>();
|
||||
|
||||
private final Gson gson = new Gson();
|
||||
|
||||
public EvalPlugin (Bot bot) {
|
||||
functions.add(new CoreFunction(bot));
|
||||
functions.add(new CorePlaceBlockFunction(bot));
|
||||
functions.add(new ChatFunction(bot));
|
||||
functions.add(new GetPlayerListFunction(bot));
|
||||
|
||||
try {
|
||||
socket = IO.socket(bot.config.eval.address);
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
socket.on(Socket.EVENT_CONNECT, (args) -> {
|
||||
connected = true;
|
||||
|
||||
final JsonArray array = new JsonArray();
|
||||
|
||||
for (EvalFunction function : functions) array.add(function.name);
|
||||
|
||||
socket.emit(
|
||||
"setFunctions",
|
||||
gson.toJson(array)
|
||||
);
|
||||
});
|
||||
socket.on(Socket.EVENT_DISCONNECT, (args) -> connected = false);
|
||||
socket.on(Socket.EVENT_CONNECT_ERROR, (args) -> connected = false);
|
||||
|
||||
for (EvalFunction function : functions) {
|
||||
socket.on(BRIDGE_PREFIX + function.name, args -> {
|
||||
final EvalFunction.Output output = function.execute(args);
|
||||
|
||||
if (output == null) return;
|
||||
|
||||
socket.emit("functionOutput:" + function.name, output.message, output.parseJSON);
|
||||
});
|
||||
}
|
||||
|
||||
socket.on("codeOutput", (args) -> {
|
||||
final int id = (int) args[0];
|
||||
final boolean isError = (boolean) args[1];
|
||||
final String output = (String) args[2];
|
||||
|
||||
final CompletableFuture<EvalOutput> future = futures.get(id);
|
||||
|
||||
future.complete(new EvalOutput(isError, output));
|
||||
});
|
||||
|
||||
socket.connect();
|
||||
}
|
||||
|
||||
public CompletableFuture<EvalOutput> run (String code) {
|
||||
final CompletableFuture<EvalOutput> future = new CompletableFuture<>();
|
||||
|
||||
if (!connected) return null;
|
||||
|
||||
socket.emit("runCode", transactionId, code);
|
||||
|
||||
futures.put(transactionId, future);
|
||||
|
||||
transactionId++;
|
||||
|
||||
return future;
|
||||
}
|
||||
|
||||
public void reset () {
|
||||
if (!connected) return;
|
||||
|
||||
socket.emit("reset");
|
||||
}
|
||||
}
|
|
@ -94,6 +94,10 @@ core:
|
|||
# PLEASE give valid JSON component here else the core don't refill at all
|
||||
customName: '[{"text":"ChomeNS ","color":"yellow"},{"text":"Core","color":"green"},{"text":"™","color":"gold"}]'
|
||||
|
||||
# https://code.chipmunk.land/ChomeNS/eval-server
|
||||
eval:
|
||||
address: 'ws://localhost:3069'
|
||||
|
||||
selfCare:
|
||||
delay: 225 # NOTE: self care does NOT use chat queue
|
||||
|
||||
|
|
Loading…
Reference in a new issue