1 files changed, 150 insertions, 0 deletions
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/auth/AuthManager.java b/src/main/java/kr/syeyoung/dungeonsguide/auth/AuthManager.java
new file mode 100644
index 00000000..5588a4a2
--- /dev/null
+++ b/src/main/java/kr/syeyoung/dungeonsguide/auth/AuthManager.java
@@ -0,0 +1,150 @@
+package kr.syeyoung.dungeonsguide.auth;
+
+import com.google.common.base.Throwables;
+import com.google.common.util.concurrent.ThreadFactoryBuilder;
+import com.google.gson.JsonObject;
+import com.mojang.authlib.exceptions.AuthenticationException;
+import kr.syeyoung.dungeonsguide.auth.authprovider.AuthProvider;
+import kr.syeyoung.dungeonsguide.auth.authprovider.DgAuth.DgAuth;
+import kr.syeyoung.dungeonsguide.auth.authprovider.DgAuth.DgAuthUtil;
+import kr.syeyoung.dungeonsguide.mod.chat.ChatTransmitter;
+import kr.syeyoung.dungeonsguide.mod.events.impl.AuthChangedEvent;
+import kr.syeyoung.dungeonsguide.mod.stomp.StompManager;
+import lombok.Setter;
+import net.minecraft.client.Minecraft;
+import net.minecraftforge.common.MinecraftForge;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+import net.minecraftforge.fml.common.gameevent.TickEvent;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+import java.io.IOException;
+import java.security.KeyPair;
+import java.security.NoSuchAlgorithmException;
+import java.util.Objects;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ThreadFactory;
+import java.util.concurrent.TimeUnit;
+
+
+public class AuthManager {
+    Logger logger = LogManager.getLogger("AuthManger");
+
+    private static AuthManager instance;
+
+    public static AuthManager getInstance() {
+        if(instance == null) instance = new AuthManager();
+        return instance;
+    }
+
+
+    @Setter
+    private String baseserverurl = "https://dungeons.guide";
+
+    private AuthProvider currentProvider;
+
+    public String getToken() {
+        if (currentProvider != null && currentProvider.getToken() != null) {
+            return currentProvider.getToken();
+        }
+        return null;
+    }
+
+    public KeyPair getKeyPair(){
+        if (currentProvider != null && currentProvider.getToken() != null) {
+            return currentProvider.getRsaKey();
+        }
+        return null;
+    }
+
+
+    boolean initlock = false;
+
+    public void init() {
+        if (initlock) {
+            logger.info("Cannot init AuthManger twice");
+            return;
+        }
+
+        reauth();
+
+        initlock = true;
+
+
+        MinecraftForge.EVENT_BUS.register(this);
+
+        ThreadFactory namedThreadFactory = new ThreadFactoryBuilder().setNameFormat("DgAuth Pool").build();
+        final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1, namedThreadFactory);
+        scheduler.scheduleAtFixedRate(() -> {
+            if (getToken() != null) {
+                JsonObject obj = DgAuthUtil.getJwtPayload(getToken());
+                if (!obj.get("uuid").getAsString().replace("-", "").equals(Minecraft.getMinecraft().getSession().getPlayerID())) {
+                    shouldReAuth = true;
+                }
+            }
+        }, 10,2000, TimeUnit.MILLISECONDS);
+    }
+
+    boolean shouldReAuth = true;
+    int tickCounter;
+
+    @SubscribeEvent
+    public void onTickClientTick(TickEvent.ClientTickEvent event) {
+        if (event.phase != TickEvent.Phase.START) return;
+
+        if (tickCounter % 200 == 0) {
+            tickCounter = 0;
+            reauth();
+        }
+        tickCounter++;
+
+    }
+
+    public boolean isPlebUser(){
+        return Objects.equals(getInstance().getPlanType(), "OPENSOURCE");
+    }
+
+    public String getPlanType(){
+        if(getToken() == null) return null;
+
+
+       JsonObject jwt = DgAuthUtil.getJwtPayload(getToken());
+
+       if(!jwt.has("plan")) return null;
+
+       return jwt.get("plan").getAsString();
+
+    }
+
+    void reauth() {
+        if (!shouldReAuth) return;
+
+        shouldReAuth = false;
+
+        currentProvider = null;
+        try {
+            currentProvider = new DgAuth(baseserverurl).createAuthProvider();
+            if (currentProvider.getToken() == null) {
+                shouldReAuth = true;
+                currentProvider = null;
+                ChatTransmitter.addToQueue("§eDungeons Guide §7:: §r§cDG auth failed, trying again in ten seconds", true);
+                logger.info("DG auth failed, trying again in a second");
+            } else {
+                // RE-AUTHed SUCCESSFULLY HOORAY
+                // for some reason the forge events don't work in pre init, so I call the callback directly
+                StompManager.getInstance().init();
+                MinecraftForge.EVENT_BUS.post(new AuthChangedEvent());
+            }
+        } catch (NoSuchAlgorithmException | AuthenticationException | IOException e) {
+
+            shouldReAuth = true;
+            currentProvider = null;
+            ChatTransmitter.addToQueue("§eDungeons Guide §7:: §r§cDG auth failed, trying again in ten seconds", true);
+            logger.error("Re-auth failed with message {}, trying again in a ten seconds", String.valueOf(Throwables.getRootCause(e)));
+        }
+
+    }
+
+
+}