From dda8d67140cd4086c1fbf3e1846d59bc0a62fae7 Mon Sep 17 00:00:00 2001
From: Cem Nura <cem.nura@gmail.com>
Date: Sun, 3 Oct 2021 03:13:36 +0300
Subject: [PATCH] Added oidc cache extension template

Signed-off-by: Cem Nura <cem.nura@gmail.com>
---
 bom/application/pom.xml                       |  5 ++
 extensions/oidc-cache/deployment/pom.xml      | 43 +++++++++++++++++
 .../cache/deployment/OidcCacheProcessor.java  | 27 +++++++++++
 extensions/oidc-cache/pom.xml                 | 19 ++++++++
 extensions/oidc-cache/runtime/pom.xml         | 46 +++++++++++++++++++
 .../oidc/runtime/CacheTokenStateManager.java  | 33 +++++++++++++
 .../resources/META-INF/quarkus-extension.yaml |  9 ++++
 extensions/pom.xml                            |  1 +
 8 files changed, 183 insertions(+)
 create mode 100644 extensions/oidc-cache/deployment/pom.xml
 create mode 100644 extensions/oidc-cache/deployment/src/main/java/io/quarkus/oidc/cache/deployment/OidcCacheProcessor.java
 create mode 100644 extensions/oidc-cache/pom.xml
 create mode 100644 extensions/oidc-cache/runtime/pom.xml
 create mode 100644 extensions/oidc-cache/runtime/src/main/java/io/quarkus/oidc/runtime/CacheTokenStateManager.java
 create mode 100644 extensions/oidc-cache/runtime/src/main/resources/META-INF/quarkus-extension.yaml

diff --git a/bom/application/pom.xml b/bom/application/pom.xml
index a3a2b7a8ce871..6257aa17a8294 100644
--- a/bom/application/pom.xml
+++ b/bom/application/pom.xml
@@ -723,6 +723,11 @@
                 <artifactId>quarkus-oidc</artifactId>
                 <version>${project.version}</version>
             </dependency>
+            <dependency>
+                <groupId>io.quarkus</groupId>
+                <artifactId>quarkus-oidc-cache</artifactId>
+                <version>${project.version}</version>
+            </dependency>
             <dependency>
                 <groupId>io.quarkus</groupId>
                 <artifactId>quarkus-test-oidc-server</artifactId>
diff --git a/extensions/oidc-cache/deployment/pom.xml b/extensions/oidc-cache/deployment/pom.xml
new file mode 100644
index 0000000000000..26041ec6dd4b3
--- /dev/null
+++ b/extensions/oidc-cache/deployment/pom.xml
@@ -0,0 +1,43 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd" xmlns="http://maven.apache.org/POM/4.0.0"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+  <modelVersion>4.0.0</modelVersion>
+  <parent>
+    <groupId>io.quarkus</groupId>
+    <artifactId>quarkus-oidc-cache-parent</artifactId>
+    <version>999-SNAPSHOT</version>
+  </parent>
+  <artifactId>quarkus-oidc-cache-deployment</artifactId>
+  <name>Quarkus - OpenID Cache - Deployment</name>
+  <dependencies>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-arc-deployment</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-oidc-cache</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-junit5-internal</artifactId>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+  <build>
+    <plugins>
+      <plugin>
+        <artifactId>maven-compiler-plugin</artifactId>
+        <configuration>
+          <annotationProcessorPaths>
+            <path>
+              <groupId>io.quarkus</groupId>
+              <artifactId>quarkus-extension-processor</artifactId>
+              <version>${project.version}</version>
+            </path>
+          </annotationProcessorPaths>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
+</project>
diff --git a/extensions/oidc-cache/deployment/src/main/java/io/quarkus/oidc/cache/deployment/OidcCacheProcessor.java b/extensions/oidc-cache/deployment/src/main/java/io/quarkus/oidc/cache/deployment/OidcCacheProcessor.java
new file mode 100644
index 0000000000000..c5547738e74c8
--- /dev/null
+++ b/extensions/oidc-cache/deployment/src/main/java/io/quarkus/oidc/cache/deployment/OidcCacheProcessor.java
@@ -0,0 +1,27 @@
+package io.quarkus.oidc.cache.deployment;
+
+import io.quarkus.arc.deployment.AdditionalBeanBuildItem;
+import io.quarkus.deployment.Capabilities;
+import io.quarkus.deployment.annotations.BuildStep;
+import io.quarkus.deployment.builditem.FeatureBuildItem;
+import io.quarkus.oidc.runtime.CacheTokenStateManager;
+
+class OidcCacheProcessor {
+
+    private static final String FEATURE = "oidc-cache";
+
+    @BuildStep
+    FeatureBuildItem feature() {
+        return new FeatureBuildItem(FEATURE);
+    }
+
+    AdditionalBeanBuildItem cacheTokenStateManager(Capabilities capabilities) {
+
+        AdditionalBeanBuildItem.Builder builder = AdditionalBeanBuildItem.builder().setUnremovable();
+
+        builder.addBeanClass(CacheTokenStateManager.class);
+
+        return builder.build();
+    }
+
+}
diff --git a/extensions/oidc-cache/pom.xml b/extensions/oidc-cache/pom.xml
new file mode 100644
index 0000000000000..338a1fbd3bb09
--- /dev/null
+++ b/extensions/oidc-cache/pom.xml
@@ -0,0 +1,19 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <parent>
+    <groupId>io.quarkus</groupId>
+    <artifactId>quarkus-extensions-parent</artifactId>
+    <version>999-SNAPSHOT</version>
+  </parent>
+  <modelVersion>4.0.0</modelVersion>
+
+  <artifactId>quarkus-oidc-cache-parent</artifactId>
+  <packaging>pom</packaging>
+  <name>Quarkus - OpenID Cache - Parent</name>
+  <modules>
+    <module>deployment</module>
+    <module>runtime</module>
+  </modules>
+</project>
\ No newline at end of file
diff --git a/extensions/oidc-cache/runtime/pom.xml b/extensions/oidc-cache/runtime/pom.xml
new file mode 100644
index 0000000000000..317b89daec4d8
--- /dev/null
+++ b/extensions/oidc-cache/runtime/pom.xml
@@ -0,0 +1,46 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd" xmlns="http://maven.apache.org/POM/4.0.0"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
+  <modelVersion>4.0.0</modelVersion>
+  <parent>
+    <groupId>io.quarkus</groupId>
+    <artifactId>quarkus-oidc-cache-parent</artifactId>
+    <version>999-SNAPSHOT</version>
+  </parent>
+  <artifactId>quarkus-oidc-cache</artifactId>
+  <name>Quarkus - OpenID Cache - Runtime</name>
+  <dependencies>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-arc</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-oidc</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>io.quarkus</groupId>
+      <artifactId>quarkus-cache</artifactId>
+    </dependency>
+  </dependencies>
+  <build>
+    <plugins>
+      <plugin>
+        <groupId>io.quarkus</groupId>
+        <artifactId>quarkus-bootstrap-maven-plugin</artifactId>
+      </plugin>
+      <plugin>
+        <artifactId>maven-compiler-plugin</artifactId>
+        <configuration>
+          <annotationProcessorPaths>
+            <path>
+              <groupId>io.quarkus</groupId>
+              <artifactId>quarkus-extension-processor</artifactId>
+              <version>${project.version}</version>
+            </path>
+          </annotationProcessorPaths>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
+</project>
diff --git a/extensions/oidc-cache/runtime/src/main/java/io/quarkus/oidc/runtime/CacheTokenStateManager.java b/extensions/oidc-cache/runtime/src/main/java/io/quarkus/oidc/runtime/CacheTokenStateManager.java
new file mode 100644
index 0000000000000..58af0ef6ab97f
--- /dev/null
+++ b/extensions/oidc-cache/runtime/src/main/java/io/quarkus/oidc/runtime/CacheTokenStateManager.java
@@ -0,0 +1,33 @@
+package io.quarkus.oidc.runtime;
+
+import io.quarkus.cache.CacheInvalidate;
+import io.quarkus.cache.CacheKey;
+import io.quarkus.cache.CacheResult;
+import io.quarkus.oidc.AuthorizationCodeTokens;
+import io.quarkus.oidc.OidcRequestContext;
+import io.quarkus.oidc.OidcTenantConfig;
+import io.quarkus.oidc.TokenStateManager;
+import io.smallrye.mutiny.Uni;
+import io.vertx.ext.web.RoutingContext;
+
+import javax.enterprise.context.ApplicationScoped;
+
+@ApplicationScoped
+public class CacheTokenStateManager implements TokenStateManager {
+    @Override
+    public Uni<String> createTokenState(RoutingContext routingContext, OidcTenantConfig oidcConfig, AuthorizationCodeTokens tokens, OidcRequestContext<String> requestContext) {
+        return TokenStateManager.super.createTokenState(routingContext, oidcConfig, tokens, requestContext);
+    }
+
+    @Override
+    @CacheResult(cacheName = "tokenState")
+    public Uni<AuthorizationCodeTokens> getTokens(RoutingContext routingContext, OidcTenantConfig oidcConfig, @CacheKey String tokenState, OidcRequestContext<AuthorizationCodeTokens> requestContext) {
+        return TokenStateManager.super.getTokens(routingContext, oidcConfig, tokenState, requestContext);
+    }
+
+    @Override
+    @CacheInvalidate(cacheName = "tokenState")
+    public Uni<Void> deleteTokens(RoutingContext routingContext, OidcTenantConfig oidcConfig, @CacheKey String tokenState, OidcRequestContext<Void> requestContext) {
+        return TokenStateManager.super.deleteTokens(routingContext, oidcConfig, tokenState, requestContext);
+    }
+}
diff --git a/extensions/oidc-cache/runtime/src/main/resources/META-INF/quarkus-extension.yaml b/extensions/oidc-cache/runtime/src/main/resources/META-INF/quarkus-extension.yaml
new file mode 100644
index 0000000000000..6b42bc364d270
--- /dev/null
+++ b/extensions/oidc-cache/runtime/src/main/resources/META-INF/quarkus-extension.yaml
@@ -0,0 +1,9 @@
+name: Oidc Cache
+#description: OpenID Cache ...
+metadata:
+#  keywords:
+#    - oidc-cache
+#  guide: ...
+#  categories:
+#    - "miscellaneous"
+#  status: "preview"
\ No newline at end of file
diff --git a/extensions/pom.xml b/extensions/pom.xml
index a979a9d04fa3e..731e96f21532f 100644
--- a/extensions/pom.xml
+++ b/extensions/pom.xml
@@ -129,6 +129,7 @@
         <module>smallrye-jwt</module>
         <module>oidc-common</module>
         <module>oidc</module>
+        <module>oidc-cache</module>
         <module>oidc-token-propagation</module>
         <module>oidc-client</module>
         <module>oidc-client-filter</module>