MidnightLib/common/src/main/java/eu/midnightdust/lib/config/AutoCommand.java

package eu.midnightdust.lib.config;

import com.mojang.brigadier.arguments.DoubleArgumentType;
import com.mojang.brigadier.arguments.IntegerArgumentType;
import com.mojang.brigadier.arguments.StringArgumentType;
import com.mojang.brigadier.builder.LiteralArgumentBuilder;
import eu.midnightdust.lib.util.PlatformFunctions;
import net.minecraft.server.command.CommandManager;
import net.minecraft.server.command.ServerCommandSource;
import net.minecraft.text.Text;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class AutoCommand {
    public static List<LiteralArgumentBuilder<ServerCommandSource>> commands = new ArrayList<>();
    private LiteralArgumentBuilder<ServerCommandSource> command;
    final Field entry;
    final String modid;

    public AutoCommand(Field entry, String modid) {
        this.entry = entry;
        this.modid = modid;
    }

    public void register() {
        command = CommandManager.literal(modid);
        command();
        LiteralArgumentBuilder<ServerCommandSource> finalized = CommandManager.literal("midnightconfig").requires(source -> source.hasPermissionLevel(2)).then(command);

        PlatformFunctions.registerCommand(finalized); commands.add(finalized);
    }

    private void command() {
        if (entry.getType() == int.class)
            command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                    CommandManager.argument("value", IntegerArgumentType.integer((int) entry.getAnnotation(MidnightConfig.Entry.class).min(),(int) entry.getAnnotation(MidnightConfig.Entry.class).max()))
                            .executes(ctx -> this.setValue(ctx.getSource(), IntegerArgumentType.getInteger(ctx, "value")))
            ));
        else if (entry.getType() == double.class)
            command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                    CommandManager.argument("value", DoubleArgumentType.doubleArg(entry.getAnnotation(MidnightConfig.Entry.class).min(),entry.getAnnotation(MidnightConfig.Entry.class).max()))
                            .executes(ctx -> this.setValue(ctx.getSource(), DoubleArgumentType.getDouble(ctx, "value")))
            ));
        else if (entry.getType() == boolean.class) {
            command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                    CommandManager.literal("true")
                            .executes(ctx -> this.setValue(ctx.getSource(), true))
            ));
            command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                    CommandManager.literal("false")
                            .executes(ctx -> this.setValue(ctx.getSource(), false))
            ));
        }
        else if (entry.getType().isEnum()) {
            for (int i = 0; i < entry.getType().getEnumConstants().length; ++i) {
                Object enumValue = Arrays.stream(entry.getType().getEnumConstants()).toList().get(i);
                command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                        CommandManager.literal(enumValue.toString())
                                .executes(ctx -> this.setValue(ctx.getSource(), enumValue))
                ));
            }
        }
        else
            command = command.then(CommandManager.literal(this.entry.getName()).executes(ctx -> getValue(ctx.getSource())).then(
                    CommandManager.argument("value", StringArgumentType.string())
                            .executes(ctx -> this.setValue(ctx.getSource(), StringArgumentType.getString(ctx, "value")))
            ));
    }

    private int setValue(ServerCommandSource source, Object value) {
        try {
            entry.set(null,value);
            MidnightConfig.write(modid);
        }
        catch (Exception e) {
            source.sendError(Text.literal("Could not set "+entry.getName()+" to value "+value+": " + e));
            return 0;
        }

        source.sendFeedback(() -> Text.literal("Successfully set " + entry.getName()+" to "+value), true);
        return 1;
    }
    private int getValue(ServerCommandSource source) {
        source.sendFeedback(() -> {
            try {
                return Text.literal("The value of "+entry.getName()+" is "+entry.get(null));
            } catch (IllegalAccessException e) {
                throw new RuntimeException(e);
            }
            }, false);
        return 0;
    }
}