Handle both JWK and JWKS

Handle classpath: prefix for resource paths
eclipse · May 3, 2018 · 587b657 · 587b657
1 parent 5ef1c7b
commit 587b657
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tck/src/test/java/org/eclipse/microprofile/jwt/tck/config/SimpleTokenUtils.java b/tck/src/test/java/org/eclipse/microprofile/jwt/tck/config/SimpleTokenUtils.java
@@ -58,14 +58,20 @@ public static PublicKey decodePublicKey(String pemEncoded) throws Exception {
     }
 
     /**
-     * Decode a JWKS encoded public key string to an RSA PublicKey
+     * Decode a JWK(S) encoded public key string to an RSA PublicKey
      * @param jwksValue - JWKS string value
      * @return PublicKey from RSAPublicKeySpec
      */
     public static PublicKey decodeJWKSPublicKey(String jwksValue) throws Exception {
         JsonObject jwks = Json.createReader(new StringReader(jwksValue)).readObject();
         JsonArray keys = jwks.getJsonArray("keys");
-        JsonObject jwk = keys.getJsonObject(0);
+        JsonObject jwk;
+        if(keys != null) {
+            jwk = keys.getJsonObject(0);
+        }
+        else {
+            jwk = jwks;
+        }
         String e = jwk.getString("e");
         String n = jwk.getString("n");
 
@@ -97,6 +103,10 @@ private static String removeBeginEnd(String pem) {
      * @throws IOException - on failure
      */
     public static String readResource(String resName) throws IOException {
+        // Strip any classpath: prefix
+        if(resName.startsWith("classpath:")) {
+            resName = resName.substring(10);
+        }
         InputStream is = SimpleTokenUtils.class.getResourceAsStream(resName);
         StringWriter sw = new StringWriter();
         try(BufferedReader br = new BufferedReader(new InputStreamReader(is))) {