Simplify default exception handler registration

Incendo · Jan 27, 2024 · 26acbf9 · 26acbf9
1 parent 88f7b8f
commit 26acbf9
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 8 deletions.
diff --git a/cloud-fabric/src/main/java/org/incendo/cloud/fabric/FabricServerCommandManager.java b/cloud-fabric/src/main/java/org/incendo/cloud/fabric/FabricServerCommandManager.java
@@ -107,11 +107,7 @@ public FabricServerCommandManager(
         }
 
         ModdedParserMappings.registerServer(this);
-        ModdedExceptionHandler.registerDefaults(
-            this,
-            CommandSourceStack::sendFailure,
-            CommandSourceStack::getTextName
-        );
+        ModdedExceptionHandler.registerDefaults(this);
     }
 
     /**

diff --git a/...mon/src/main/java/org/incendo/cloud/minecraft/modded/internal/ModdedExceptionHandler.java b/...mon/src/main/java/org/incendo/cloud/minecraft/modded/internal/ModdedExceptionHandler.java
@@ -30,6 +30,7 @@
 import java.util.function.BiConsumer;
 import java.util.function.Function;
 import net.minecraft.ChatFormatting;
+import net.minecraft.commands.CommandSourceStack;
 import net.minecraft.commands.SharedSuggestionProvider;
 import net.minecraft.network.chat.ClickEvent;
 import net.minecraft.network.chat.Component;
@@ -40,6 +41,7 @@
 import org.checkerframework.checker.nullness.qual.NonNull;
 import org.checkerframework.framework.qual.DefaultQualifier;
 import org.incendo.cloud.CommandManager;
+import org.incendo.cloud.brigadier.BrigadierManagerHolder;
 import org.incendo.cloud.exception.ArgumentParseException;
 import org.incendo.cloud.exception.CommandExecutionException;
 import org.incendo.cloud.exception.InvalidCommandSenderException;
@@ -81,17 +83,32 @@ default void handle(final ExceptionContext<C, T> context) throws Throwable {
      */
     void handle(S source, C sender, T throwable) throws Throwable;
 
+    /**
+     * Registers the default handlers.
+     *
+     * @param commandManager the command manager
+     * @param <M>            command manager type
+     * @param <C>            command sender type
+     */
+    static <C, M extends CommandManager<C> & BrigadierManagerHolder<C, CommandSourceStack>> void registerDefaults(
+            final M commandManager
+    ) {
+        registerDefaults(commandManager, CommandSourceStack::sendFailure, CommandSourceStack::getTextName);
+    }
+
     /**
      * Registers the default handlers.
      *
      * @param commandManager the command manager
      * @param sendError      error message sender
      * @param getName        name getter
+     * @param <M>            command manager type
      * @param <C>            command sender type
      * @param <S>            command source type
      */
-    static <C, S extends SharedSuggestionProvider> void registerDefaults(
-        final CommandManager<C> commandManager,
+    static <C, S extends SharedSuggestionProvider,
+            M extends CommandManager<C> & BrigadierManagerHolder<C, S>> void registerDefaults(
+            final M commandManager,
         final BiConsumer<S, Component> sendError,
         final Function<S, String> getName
     ) {

diff --git a/cloud-neoforge/src/main/java/org/incendo/cloud/neoforge/NeoForgeCommandManager.java b/cloud-neoforge/src/main/java/org/incendo/cloud/neoforge/NeoForgeCommandManager.java
@@ -70,7 +70,7 @@ protected NeoForgeCommandManager(
             this.senderMapper.map(dummyCommandSourceProvider.get()),
             this
         ), senderMapper);
-        ModdedExceptionHandler.registerDefaults(this, CommandSourceStack::sendFailure, CommandSourceStack::getTextName);
+        ModdedExceptionHandler.registerDefaults(this);
         registrationHandler.initialize(this);
         this.captionRegistry().registerProvider(new ModdedDefaultCaptionsProvider<>());
         this.registerCommandPreProcessor(new ModdedPreprocessor<>(senderMapper));