Merge pull request #918 from zapier/PDE-5540

feat(core): Add support for providing the cache expiration for hydrators

packages/core/src/tools/create-dehydrator.js

@@ -9,19 +9,25 @@ const wrapHydrate = require('./wrap-hydrate');
 const createDehydrator = (input, type = 'method') => {
   const app = _.get(input, '_zapier.app');
 
-  return (func, inputData) => {
+  return (func, inputData, cacheExpiration) => {
     inputData = inputData || {};
     if (inputData.inputData) {
       throw new DehydrateError(
         'Oops! You passed a full `bundle` - really you should pass what you want under `inputData`!'
       );
     }
-    return wrapHydrate({
+    const payload = {
       type,
       method: resolveMethodPath(app, func),
       // inputData vs. bundle is a legacy oddity
       bundle: _.omit(inputData, 'environment'), // don't leak the environment
-    });
+    };
+
+    if (cacheExpiration) {
+      payload.cacheExpiration = cacheExpiration;
+    }
+
+    return wrapHydrate(payload);
   };
 };
 

packages/core/test/hydration.js

@@ -52,6 +52,13 @@ describe('hydration', () => {
       );
     });
 
+    it('should allow passing of cache expiration argument along in the dehydrated data', () => {
+      const result = dehydrate(funcToFind, {}, 60);
+      result.should.eql(
+        'hydrate|||{"type":"method","method":"some.path.to","bundle":{},"cacheExpiration":60}|||hydrate'
+      );
+    });
+
     it('should not accept payload size bigger than 12000 bytes.', () => {
       const inputData = { key: 'a'.repeat(12001) };
       (() => {
@@ -91,6 +98,14 @@ describe('hydration', () => {
       );
     });
 
+    it('should allow passing of cache expiration argument along in the dehydrated data', () => {
+      const inputData = { key: 'value' };
+      const result = dehydrateFile(funcToFind, inputData, 60);
+      result.should.eql(
+        'hydrate|||{"type":"file","method":"some.path.to","bundle":{"key":"value"},"cacheExpiration":60}|||hydrate'
+      );
+    });
+
     it('should not accept payload size bigger than 12000 bytes.', () => {
       const inputData = { key: 'a'.repeat(12001) };
       (() => {

packages/core/types/zapier.custom.d.ts

@@ -133,7 +133,8 @@ export interface RawHttpResponse<T = any> extends BaseHttpResponse {
 
 type DehydrateFunc = <T>(
   func: (z: ZObject, bundle: Bundle<T>) => any,
-  inputData: T
+  inputData?: T,
+  cacheExpiration?: number,
 ) => string;
 
 export interface ZObject {