@orion-js/graphql 2.6.5 → 2.6.7

package/lib/getOperationName.js

@@ -0,0 +1,28 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = getOperationName;
+
+async function getOperationName(params) {
+  const {
+    request,
+    getBodyJSON
+  } = params;
+
+  try {
+    if (request.method !== 'GET') {
+      const data = await getBodyJSON();
+      return data.operationName;
+    }
+
+    if (request.method === 'GET') {
+      // handle persisted queries
+      const queryParams = new URLSearchParams(request.url.split('?')[1]);
+      return queryParams.get('operationName');
+    }
+  } catch (error) {
+    return null;
+  }
+}

package/lib/startGraphQL.js

@@ -7,10 +7,10 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.default = _default;
 
-var _app = require("@orion-js/app");
-
 var _apolloServerMicro = require("apollo-server-micro");
 
+var _app = require("@orion-js/app");
+
 var _startGraphiQL = _interopRequireDefault(require("./startGraphiQL"));
 
 var _getApolloOptions = _interopRequireDefault(require("./getApolloOptions"));
@@ -21,6 +21,12 @@ var _micro = _interopRequireDefault(require("micro"));
 
 var _apolloServerCore = require("apollo-server-core");
 
+var _apolloServerCaching = require("apollo-server-caching");
+
+var _getOperationName = _interopRequireDefault(require("./getOperationName"));
+
+var _storePersistedQuery = _interopRequireDefault(require("./storePersistedQuery"));
+
 global.globalMicro = _micro.default;
 
 async function _default(options) {
@@ -35,6 +41,7 @@ async function _default(options) {
   await apolloServer.start();
   const handler = apolloServer.createHandler(); // highlight-line
 
+  const cacheControlCache = apolloOptions.persistedQueries?.cache ? apolloOptions.persistedQueries.cache : new _apolloServerCaching.InMemoryLRUCache();
   (0, _app.route)('/graphql', async function (params) {
     const {
       request,
@@ -42,6 +49,8 @@ async function _default(options) {
       viewer,
       getBodyJSON
     } = params;
+    const operationName = await (0, _getOperationName.default)(params);
+    const cacheKey = operationName ? `orion-cache-control-${operationName}` : null;
 
     if (options.executeGraphQLCache) {
       try {
@@ -61,6 +70,13 @@ async function _default(options) {
         const result = await options.executeGraphQLCache(params, fallback);
 
         if (result) {
+          await (0, _storePersistedQuery.default)(apolloOptions, params);
+
+          if (result.cacheControl && cacheKey) {
+            cacheControlCache.set(cacheKey, result.cacheControl);
+            response.setHeader('Cache-Control', result.cacheControl);
+          }
+
           return result;
         }
       } catch (error) {
@@ -68,6 +84,22 @@ async function _default(options) {
       }
     }
 
+    if (cacheKey) {
+      let shouldCache = true;
+
+      if (apolloOptions?.shouldAddCacheControl) {
+        shouldCache = apolloOptions.shouldAddCacheControl(operationName);
+      }
+
+      if (shouldCache) {
+        const cacheControl = await cacheControlCache.get(cacheKey);
+
+        if (cacheControl) {
+          response.setHeader('Cache-Control', cacheControl);
+        }
+      }
+    }
+
     request._orionjsViewer = viewer;
     handler(request, response);
   });

package/lib/storePersistedQuery.js

@@ -0,0 +1,41 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = storePersistedQuery;
+exports.APQ_CACHE_PREFIX = void 0;
+const APQ_CACHE_PREFIX = 'apq:';
+exports.APQ_CACHE_PREFIX = APQ_CACHE_PREFIX;
+
+function createSHA(algorithm) {
+  return require('crypto').createHash(algorithm);
+}
+
+function computeQueryHash(query) {
+  return createSHA('sha256').update(query).digest('hex');
+}
+
+async function storePersistedQuery(apolloConfig, params) {
+  const queryCache = apolloConfig.persistedQueries.cache;
+
+  if (!queryCache) {
+    return;
+  }
+
+  if (params.request.method !== 'POST') {
+    return;
+  }
+
+  const data = await params.getBodyJSON();
+
+  if (!data?.query) {
+    return;
+  }
+
+  const hash = APQ_CACHE_PREFIX + computeQueryHash(data.query);
+  const ttl = apolloConfig.persistedQueries?.ttl !== undefined ? apolloConfig.persistedQueries.ttl : 60 * 15;
+  await queryCache.set(hash, data.query, {
+    ttl
+  });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@orion-js/graphql",
-  "version": "2.6.5",
+  "version": "2.6.7",
   "main": "lib/index.js",
   "author": "nicolaslopezj",
   "license": "MIT",
@@ -11,7 +11,7 @@
     "test": "jest"
   },
   "dependencies": {
-    "@orion-js/schema": "^2.6.5",
+    "@orion-js/schema": "^2.6.7",
     "apollo-server-core": "3.4.0",
     "apollo-server-micro": "3.4.0",
     "deep-sort-object": "^1.0.2",
@@ -26,7 +26,7 @@
     "@babel/plugin-transform-runtime": "^7.11.0",
     "@babel/preset-env": "^7.11.0",
     "@babel/runtime": "^7.11.2",
-    "@orion-js/app": "^2.6.5",
+    "@orion-js/app": "^2.6.7",
     "babel-jest": "^26.2.2",
     "graphql": "^15.6.0",
     "jest": "^26.2.2"
@@ -38,5 +38,5 @@
   "publishConfig": {
     "access": "public"
   },
-  "gitHead": "64c548a6f2a849ec016b1731c1659ef8bd7f6bee"
+  "gitHead": "c96b65cd08a81c7ac14e3d02df38efd0482a791e"
 }