Handle missing cases for MFA errors

Author: carlodrift

src/main/java/net/clementraynaud/skoice/listeners/session/ReadyListener.java

@@ -26,10 +26,15 @@
 import net.dv8tion.jda.api.Permission;
 import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.events.session.ReadyEvent;
+import net.dv8tion.jda.api.exceptions.ErrorResponseException;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
+import net.dv8tion.jda.api.requests.ErrorResponse;
+import net.dv8tion.jda.api.requests.RestAction;
 import net.kyori.adventure.text.event.HoverEvent;
 import org.bukkit.entity.Player;
 
+import java.util.function.Consumer;
+
 public class ReadyListener extends ListenerAdapter {
 
     private final Skoice plugin;
@@ -56,6 +61,19 @@ public void onReady(ReadyEvent event) {
 
             this.setup(tokenManager);
         });
+
+        Consumer<? super Throwable> defaultFailure = RestAction.getDefaultFailure();
+        RestAction.setDefaultFailure(throwable -> {
+            if (throwable instanceof ErrorResponseException) {
+                ErrorResponseException error = (ErrorResponseException) throwable;
+                if (error.getErrorCode() == ErrorResponse.MFA_NOT_ENABLED.getCode()) {
+                    this.plugin.getListenerManager().update();
+                    return;
+                }
+            }
+
+            defaultFailure.accept(throwable);
+        });
     }
 
     private void handlePublicBot(Player tokenManager) {