Customizable uncaught exception handling

Author: NichtStudioCode

invui/src/main/java/xyz/xenondevs/invui/InvUI.java

@@ -10,7 +10,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
-import java.util.logging.Logger;
+import java.util.function.BiConsumer;
 
 /**
  * Main class of InvUI, managing the plugin instance.
@@ -21,6 +21,7 @@ public final class InvUI implements Listener {
 
     private final List<Runnable> disableHandlers = new ArrayList<>();
     private @Nullable Plugin plugin;
+    private BiConsumer<? super String, ? super Throwable> uncaughtExceptionHandler = (msg, e) -> getPlugin().getComponentLogger().error(msg, e);
 
     private InvUI() {}
 
@@ -86,13 +87,23 @@ public void setPlugin(@Nullable Plugin plugin) {
     }
 
     /**
-     * Gets the logger of the configured plugin instance.
+     * Sets a handler for uncaught exceptions in InvUI.
      *
-     * @return The logger of the plugin instance.
-     * @throws IllegalStateException If the plugin instance is not set and cannot be inferred.
+     * @param uncaughtExceptionHandler The new uncaught exception handler.
+     */
+    public void setUncaughtExceptionHandler(BiConsumer<? super String, ? super Throwable> uncaughtExceptionHandler) {
+        this.uncaughtExceptionHandler = uncaughtExceptionHandler;
+    }
+    
+    /**
+     * Handles an uncaught exception using the configured
+     * {@link #setUncaughtExceptionHandler(BiConsumer) uncaught exception handler}.
+     *
+     * @param msg An additional message that provides more context.
+     * @param t   The uncaught exception that was thrown.
      */
-    public Logger getLogger() {
-        return getPlugin().getLogger();
+    public void handleUncaughtException(String msg, Throwable t) {
+        uncaughtExceptionHandler.accept(msg, t);
     }
 
     /**

invui/src/main/java/xyz/xenondevs/invui/gui/SlotElement.java

@@ -14,7 +14,6 @@
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
-import java.util.logging.Level;
 
 /**
  * Represents an element in a slot in a {@link Gui}.
@@ -66,7 +65,7 @@ record Item(xyz.xenondevs.invui.item.Item item) implements SlotElement {
             try {
                 return item.getItemProvider(player).get(Languages.getInstance().getLocale(player));
             } catch (Throwable t) {
-                InvUI.getInstance().getLogger().log(Level.SEVERE, "Failed to get item stack for item slot element", t);
+                InvUI.getInstance().handleUncaughtException("Failed to get item stack for item slot element", t);
             }
 
             return null;

invui/src/main/java/xyz/xenondevs/invui/internal/menu/CustomContainerMenu.java

@@ -56,7 +56,6 @@
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
-import java.util.logging.Level;
 
 /**
  * A packet-based container menu.
@@ -606,7 +605,7 @@ protected void runInInteractionContext(Runnable run) {
             interactionContext.set(true);
             run.run();
         } catch (Throwable t) {
-            InvUI.getInstance().getLogger().log(Level.SEVERE, "An exception occurred while handling a window interaction", t);
+            InvUI.getInstance().handleUncaughtException("An exception occurred while handling a window interaction", t);
         } finally {
             interactionContext.set(false);
         }

invui/src/main/java/xyz/xenondevs/invui/internal/util/CollectionUtils.java

@@ -43,7 +43,7 @@ public static <T> void forEachCatching(Iterable<? extends T> iterable, Consumer<
                 action.accept(obj);
                 i++;
             } catch (Throwable t) {
-                InvUI.getInstance().getLogger().log(Level.SEVERE, message + " (" + i + ")", t);
+                InvUI.getInstance().handleUncaughtException(message + " (" + i + ")", t);
             }
         }
     }

invui/src/main/java/xyz/xenondevs/invui/internal/util/FuncUtils.java

@@ -23,7 +23,7 @@ public static <T> T getSafely(@NonNull Supplier<? extends T> supplier, T fallbac
         try {
             return supplier.get();
         } catch (Throwable t) {
-            InvUI.getInstance().getLogger().log(Level.SEVERE, "Failed to get value from supplier", t);
+            InvUI.getInstance().handleUncaughtException("Failed to get value from supplier", t);
             return fallback;
         }
     }

invui/src/main/java/xyz/xenondevs/invui/inventory/Inventory.java

@@ -451,7 +451,7 @@ public boolean callClickEvent(int slot, Click click) {
             try {
                 handler.accept(clickEvent);
             } catch (Throwable t) {
-                InvUI.getInstance().getLogger().log(Level.SEVERE, "An exception occurred while handling an inventory event", t);
+                InvUI.getInstance().handleUncaughtException("An exception occurred while handling an inventory event", t);
             }
         }
 
@@ -478,7 +478,7 @@ public ItemPreUpdateEvent callPreUpdateEvent(@Nullable UpdateReason updateReason
             try {
                 handler.accept(event);
             } catch (Throwable t) {
-                InvUI.getInstance().getLogger().log(Level.SEVERE, "An exception occurred while handling an inventory event", t);
+                InvUI.getInstance().handleUncaughtException("An exception occurred while handling an inventory event", t);
             }
         }
         return event;
@@ -503,7 +503,7 @@ public void callPostUpdateEvent(@Nullable UpdateReason updateReason, int slot, @
             try {
                 handler.accept(event);
             } catch (Throwable t) {
-                InvUI.getInstance().getLogger().log(Level.SEVERE, "An exception occurred while handling an inventory event", t);
+                InvUI.getInstance().handleUncaughtException("An exception occurred while handling an inventory event", t);
             }
         }
     }

invui/src/main/java/xyz/xenondevs/invui/inventory/VirtualInventoryManager.java

@@ -11,7 +11,6 @@
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.*;
-import java.util.logging.Level;
 
 /**
  * Automatically reads and writes {@link VirtualInventory VirtualInventories} to files when the server starts and stops.
@@ -156,8 +155,7 @@ private void deserializeAll() {
                 VirtualInventory inventory = VirtualInventory.deserialize(in);
                 inventories.put(inventory.getUuid(), inventory);
             } catch (IOException e) {
-                InvUI.getInstance().getLogger().log(
-                    Level.SEVERE,
+                InvUI.getInstance().handleUncaughtException(
                     "Failed to deserialize a VirtualInventory from file " + file.getPath(),
                     e
                 );
@@ -176,8 +174,7 @@ private void serializeAll() {
             try (FileOutputStream out = new FileOutputStream(file)) {
                 inventory.serialize(out);
             } catch (IOException e) {
-                InvUI.getInstance().getLogger().log(
-                    Level.SEVERE,
+                InvUI.getInstance().handleUncaughtException(
                     "Failed to serialize a VirtualInventory to file " + file.getPath(),
                     e
                 );