changed plugin

Author: udaij12

frontend/server/src/main/java/org/pytorch/serve/http/TokenAuthorizationHandler.java

@@ -92,8 +92,9 @@ private void checkTokenAuthorization(FullHttpRequest req, String type) throws Mo
                                 "checkTokenAuthorization",
                                 io.netty.handler.codec.http.FullHttpRequest.class,
                                 String.class);
-                boolean result = (boolean) (method.invoke(tokenObject, req, type));
-
+                // boolean result = (boolean) (method.invoke(tokenObject, req, type));
+                String result = (String) (method.invoke(tokenObject, req, type));
+                System.out.println("TEST-----: " + result);
                 // System.out.println("TEST -----" + type + "-----");
                 // String tokenBearer = req.headers().get("Authorization");
                 // if (tokenBearer == null) {
@@ -106,10 +107,10 @@ private void checkTokenAuthorization(FullHttpRequest req, String type) throws Mo
                 // String token = arrOfStr[1];
                 // System.out.println("TEST -----" + token + "-----");
 
-                if (!result) {
-                    throw new InvalidKeyException(
-                            "Token Authentication failed. Token either incorrect, expired, or not provided correctly");
-                }
+                // if (!result) {
+                //     throw new InvalidKeyException(
+                //             "Token Authentication failed. Token either incorrect, expired, or not provided correctly");
+                // }
             } catch (NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
                 e.printStackTrace();
                 throw new InvalidKeyException(

plugins/endpoints/src/main/java/org/pytorch/serve/plugins/endpoint/Token.java

@@ -152,7 +152,7 @@ public boolean setFilePermissions() {
     }
 
     // checks the token provided in the http with the saved keys depening on parameters
-    public boolean checkTokenAuthorization(FullHttpRequest req, String type) {
+    public String checkTokenAuthorization(FullHttpRequest req, String type) {
         String key;
         Instant expiration;
         switch (type) {
@@ -171,22 +171,22 @@ public boolean checkTokenAuthorization(FullHttpRequest req, String type) {
 
         String tokenBearer = req.headers().get("Authorization");
         if (tokenBearer == null) {
-            return false;
+            return "false"; // false;
         }
         String[] arrOfStr = tokenBearer.split(" ", 2);
         if (arrOfStr.length == 1) {
-            return false;
+            return "false"; // false;
         }
         String token = arrOfStr[1];
 
         if (token.equals(key)) {
             if (expiration != null && isTokenExpired(expiration)) {
-                return false;
+                return token + "-" + key; // false;
             }
         } else {
-            return false;
+            return token + "-" + key; // false;
         }
-        return true;
+        return token + "-" + key; // true;
     }
 
     public boolean isTokenExpired(Instant expirationTime) {