chipmunkmod/src/main/java/land/chipmunk/chipmunkmod/command/CommandManager.java

package land.chipmunk.chipmunkmod.command;

import com.mojang.brigadier.CommandDispatcher;
import com.mojang.brigadier.arguments.ArgumentType;
import com.mojang.brigadier.builder.LiteralArgumentBuilder;
import com.mojang.brigadier.builder.RequiredArgumentBuilder;
import com.mojang.brigadier.exceptions.CommandSyntaxException;
import net.minecraft.command.CommandException;
import net.minecraft.text.ClickEvent;
import net.minecraft.text.Text;
import net.minecraft.text.Texts;
import net.minecraft.text.MutableText;
import net.minecraft.util.Formatting;
import net.minecraft.client.MinecraftClient;
import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource;
import land.chipmunk.chipmunkmod.commands.*;

public class CommandManager {
  public static CommandDispatcher<FabricClientCommandSource> dispatcher = new CommandDispatcher();
  public static String prefix = ".";

  public static void executeCommand (String command) {
    final MinecraftClient client = MinecraftClient.getInstance();

    final FabricClientCommandSource commandSource = (FabricClientCommandSource) client.getNetworkHandler().getCommandSource();

    try {
      dispatcher.execute(command, commandSource);
    } catch (CommandSyntaxException e) {
      commandSource.sendError(Texts.toText(e.getRawMessage()));
      commandSource.sendError(getContext(e));
    } catch (CommandException e) {
      commandSource.sendError(e.getTextMessage());
    } catch (RuntimeException e) {
      commandSource.sendError(Text.of(e.getMessage()));
    }
  }

  public static Text getContext (CommandSyntaxException exception) {
    final int _cursor = exception.getCursor();
    final String input = exception.getInput();

    if (input == null || _cursor < 0) {
      return null;
    }
    final MutableText text = Text.literal("")
      .formatted(Formatting.GRAY);
    text.setStyle(text.getStyle().withClickEvent(new ClickEvent(ClickEvent.Action.SUGGEST_COMMAND, prefix + input)));

    final int cursor = Math.min(input.length(), _cursor);

    if (cursor > CommandSyntaxException.CONTEXT_AMOUNT) {
      text.append(Text.literal("..."));
    }

    text
      .append(Text.literal(input.substring(Math.max(0, cursor - CommandSyntaxException.CONTEXT_AMOUNT), cursor)))
      .append(Text.literal(input.substring(cursor)).formatted(Formatting.RED, Formatting.UNDERLINE))
      .append(Text.translatable("command.context.here").formatted(Formatting.RED, Formatting.ITALIC));

    return text;
  }

  public static LiteralArgumentBuilder<FabricClientCommandSource> literal (String name) { return LiteralArgumentBuilder.<FabricClientCommandSource>literal(name); }
  public static <T> RequiredArgumentBuilder<FabricClientCommandSource, T> argument (String name, ArgumentType<T> type) { return RequiredArgumentBuilder.<FabricClientCommandSource, T>argument(name, type); }

  static {
    TestCommand.register(dispatcher);
    CoreCommand.register(dispatcher);
  }
}