cache: scaffold kv integration via context

packages/core/package.json

@@ -15,6 +15,7 @@
     "typescript": "^5.2.2"
   },
   "dependencies": {
+    "crypto-js": "^4.2.0",
     "graphql-tag": "^2.12.6"
   }
 }

packages/core/src/index.ts

@@ -1 +1,2 @@
-export {GraphQLHandler} from "./handlers/graphql";
+export {GraphQLHandler} from "./handlers/graphql";
+export {generateGraphQLCacheKey} from "./invalidation/invalidation";

services/itty-hydra/hydra.config.ts

@@ -0,0 +1,17 @@
+export const HydraConfig = {
+    rateLimiting: {
+      default: {
+        budget: 100,
+      },
+    },
+    publicQueries: [
+      {
+        name: "health",
+      },
+      {
+        name: "hydraDevQuery",
+      }
+    ],
+    jwksUri: "https://id.authdog.com/oidc/.well-known/jwks.json",
+  };
+

services/itty-hydra/main.ts

@@ -4,17 +4,90 @@ import { withDurables } from "itty-durable";
 import { NotFound } from "./handlers/notFound";
 import { Health } from "./handlers/health";
 import {GraphQLHandler} from "@authdog/hydra-core";
+import { HydraConfig } from "./hydra.config";
+import { extractedAllQueryIdentifiersInRawQuery, generateGraphQLCacheKey } from "@authdog/hydra-core/src/invalidation/invalidation";
 
 const { preflight, corsify } = createCors();
 
 const router = Router();
 router
-  .all("*", withDurables())
+  // .all("*", withDurables())
   .options("*", preflight)
   .get("/", Health)
   .get("/health", Health)
+  // serves playground
   .get("/graphql", GraphQLHandler)
-  .post("/graphql", GraphQLHandler)
+  .post("/graphql", async (req, env, ctx) => {
+    const {kv} = ctx;
+
+    let extractedQueries = [];
+    let cacheKey = null;
+
+    // const clonedRequest = req.clone();
+
+    // // const requestHeaders = clonedRequest?.headers;
+    const requestBody = await req.clone()?.json();
+
+    let isIntrospection = true;
+    if (requestBody?.operationName !== "IntrospectionQuery") {
+      isIntrospection = false;
+
+
+    // const isMutation = requestBody?.query?.startsWith("mutation");
+
+    extractedQueries = extractedAllQueryIdentifiersInRawQuery(
+      requestBody?.query
+    );
+
+    const variables = JSON.stringify(requestBody?.variables);
+
+
+    const requiresAuthorization = extractedQueries.some((query) => {
+      return !HydraConfig.publicQueries.some((publicQuery) => {
+        return publicQuery.name === query;
+      });
+    });
+
+    if (!requiresAuthorization) {
+      cacheKey = await generateGraphQLCacheKey({
+        query: requestBody?.query,
+        variables,
+      });
+    } else if (requiresAuthorization) {
+      return new Response(
+        JSON.stringify({
+          errors: [
+            {
+              message: "Unauthorized",
+            },
+          ],
+        }),
+        {
+          status: 401,
+        }
+      );
+    }
+
+    if (cacheKey) {
+      const cachedResponse = await kv.get(cacheKey);
+      if (cachedResponse) {
+        return new Response(cachedResponse, {
+          status: 200,
+        });
+      }
+    }
+  }
+
+    const response = await GraphQLHandler(req, env, ctx);
+
+      if (cacheKey) {
+        const responseBody = await response?.clone()?.json();
+        const responseBodyString = JSON.stringify(responseBody);
+        await kv.put(cacheKey, responseBodyString);
+      }
+
+      return response;
+  })
   .get("*", NotFound);
 
 const handleRequest = (req, env, ctx) => {
@@ -23,7 +96,7 @@ const handleRequest = (req, env, ctx) => {
   const enrichedContext = {
     ...ctx,
     kv: HYDRA_ACME,
-    rateLimiter: null,
+    // rateLimiter: null,
   };
 
   return router