feat: enable custom functions

Author: dreamorosi

packages/idempotency/src/IdempotencyHandler.ts

@@ -17,6 +17,7 @@ import { IdempotencyRecord } from './persistence/IdempotencyRecord.js';
 import { IdempotencyConfig } from './IdempotencyConfig.js';
 import { MAX_RETRIES, IdempotencyRecordStatus } from './constants.js';
 import { search } from '@aws-lambda-powertools/jmespath';
+import { PowertoolsFunctions } from '@aws-lambda-powertools/jmespath/functions';
 
 /**
  * @internal
@@ -276,7 +277,8 @@ export class IdempotencyHandler<Func extends AnyFunction> {
     ) {
       const selection = search(
         this.#idempotencyConfig.eventKeyJmesPath,
-        this.#functionPayloadToBeHashed
+        this.#functionPayloadToBeHashed,
+        { customFunctions: new PowertoolsFunctions() }
       );
 
       return selection === undefined || selection === null;

packages/idempotency/src/persistence/BasePersistenceLayer.ts

@@ -1,5 +1,6 @@
 import { createHash, Hash } from 'node:crypto';
 import { search } from '@aws-lambda-powertools/jmespath';
+import { PowertoolsFunctions } from '@aws-lambda-powertools/jmespath/functions';
 import type {
   BasePersistenceLayerOptions,
   BasePersistenceLayerInterface,
@@ -279,7 +280,9 @@ abstract class BasePersistenceLayer implements BasePersistenceLayerInterface {
    */
   private getHashedIdempotencyKey(data: JSONValue): string {
     if (this.eventKeyJmesPath) {
-      data = search(this.eventKeyJmesPath, data) as JSONValue;
+      data = search(this.eventKeyJmesPath, data, {
+        customFunctions: new PowertoolsFunctions(),
+      }) as JSONValue;
     }
 
     if (BasePersistenceLayer.isMissingIdempotencyKey(data)) {
@@ -305,7 +308,9 @@ abstract class BasePersistenceLayer implements BasePersistenceLayerInterface {
    */
   private getHashedPayload(data: JSONValue): string {
     if (this.isPayloadValidationEnabled() && this.validationKeyJmesPath) {
-      data = search(this.validationKeyJmesPath, data) as JSONValue;
+      data = search(this.validationKeyJmesPath, data, {
+        customFunctions: new PowertoolsFunctions(),
+      }) as JSONValue;
 
       return this.generateHash(JSON.stringify(data));
     } else {

packages/idempotency/tests/e2e/makeIdempotent.test.FunctionCode.ts

@@ -90,16 +90,17 @@ export const handlerCustomized = async (
  * Test idempotent Lambda handler with JMESPath expression to extract event key.
  */
 export const handlerLambda = makeIdempotent(
-  async (event: { foo: string }, context: Context) => {
+  async (event: { body: string }, context: Context) => {
     logger.addContext(context);
-    logger.info(`foo`, { details: event.foo });
+    const body = JSON.parse(event.body);
+    logger.info('foo', { details: body.foo });
 
-    return event.foo;
+    return body.foo;
   },
   {
     persistenceStore: dynamoDBPersistenceLayer,
     config: new IdempotencyConfig({
-      eventKeyJmesPath: 'foo',
+      eventKeyJmesPath: 'powertools_json(body).foo',
       useLocalCache: true,
     }),
   }

packages/idempotency/tests/e2e/makeIdempotent.test.ts

@@ -268,10 +268,12 @@ describe(`Idempotency E2E tests, wrapper function usage`, () => {
     async () => {
       // Prepare
       const payload = {
-        foo: 'bar',
+        body: JSON.stringify({
+          foo: 'bar',
+        }),
       };
       const payloadHash = createHash('md5')
-        .update(JSON.stringify(payload.foo))
+        .update(JSON.stringify('bar'))
         .digest('base64');
 
       // Act