Update AuthUtil.java

Author: zhou9584

center/src/main/java/com/microsoft/hydralab/center/util/AuthUtil.java

@@ -66,6 +66,7 @@ public class AuthUtil {
 
     Map<String, Boolean> urlMapping = null;
     private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(AuthUtil.class);
+    private static Boolean isMiseLogInitialized = false;
 
     public boolean isValidToken(String token) {
         LOGGER.info("Starting token validation...");
@@ -118,14 +119,39 @@ private boolean validateTokenWithPublicKey(String token) {
         }
     }
 
-    public boolean validateTokenWithMISE(String token) {
+    private boolean validateTokenWithMISE(String token) {
         LOGGER.info("Starting MISE token validation...");
 
         try {
             // Mise mise = Mise.createClient();
             Class<?> miseClass = Class.forName("com.microsoft.identity.service.essentials.Mise");
             Object mise = miseClass.getMethod("createClient").invoke(null);
 
+            if (!isMiseLogInitialized) {
+                LOGGER.info("Initializing MISE...");
+                // mise.assignLogMessageCallback(new Mise.ILogCallback() {...}, null);
+                Class<?> iLogCallbackClass = Class.forName("com.microsoft.identity.service.essentials.Mise$ILogCallback");
+
+                Object logCallback = java.lang.reflect.Proxy.newProxyInstance(
+                        iLogCallbackClass.getClassLoader(),
+                        new Class<?>[]{iLogCallbackClass},
+                        (proxy, method, args) -> {
+                            String methodName = method.getName();
+                            if ("callback".equals(methodName)) {
+                                Object level = args[0];
+                                String message = (String) args[1];
+                                // Print all log levels for simplicity
+                                LOGGER.info(message);
+                            }
+                            return null;
+                        }
+                );
+
+                miseClass.getMethod("assignLogMessageCallback", iLogCallbackClass, Object.class)
+                        .invoke(mise, logCallback, null);
+                isMiseLogInitialized = true;
+            }
+
             // Configure MISE
             JSONObject config = new JSONObject();
             JSONObject azureAd = new JSONObject();