feat: add Step Functions enrichiment for pipes

Author: mazyu36

packages/@aws-cdk/aws-pipes-enrichments-alpha/README.md

@@ -46,3 +46,24 @@ const pipe = new pipes.Pipe(this, 'Pipe', {
   target: new SomeTarget(targetQueue),
 });
 ```
+
+### Step Functions statemachine
+
+A Step Functions statemachine can be used to enrich events of a pipe.
+Express workflows are only supported and can only be invoked synchronously.
+For more information, see [Amazon EventBridge Pipes event enrichment](https://docs.aws.amazon.com/eventbridge/latest/userguide/pipes-enrichment.html).
+
+```ts
+declare const sourceQueue: sqs.Queue;
+declare const targetQueue: sqs.Queue;
+
+declare const enrichmentStateMachine: stepfunctions.StateMachine;
+
+const enrichment = new enrichments.StepFunctionsEnrichment(enrichmentStateMachine);
+
+const pipe = new pipes.Pipe(this, 'Pipe', {
+  source: new SomeSource(sourceQueue),
+  enrichment,
+  target: new SomeTarget(targetQueue),
+});
+```

packages/@aws-cdk/aws-pipes-enrichments-alpha/lib/index.ts

@@ -1 +1,2 @@
 export * from './lambda';
+export * from './stepfunctions';

packages/@aws-cdk/aws-pipes-enrichments-alpha/lib/stepfunctions.ts

@@ -0,0 +1,49 @@
+import { EnrichmentParametersConfig, IEnrichment, IPipe, InputTransformation } from '@aws-cdk/aws-pipes-alpha';
+import { IRole } from 'aws-cdk-lib/aws-iam';
+import { IStateMachine, StateMachine, StateMachineType } from 'aws-cdk-lib/aws-stepfunctions';
+
+/**
+ * Properties for a StepFunctionsEnrichment
+ */
+export interface StepFunctionsEnrichmentProps {
+  /**
+   * The input transformation for the enrichment
+   * @see https://docs.aws.amazon.com/eventbridge/latest/userguide/eb-pipes-input-transformation.html
+   * @default - None
+   */
+  readonly inputTransformation?: InputTransformation;
+}
+
+/**
+ * A StepFunctions enrichment for a pipe
+ */
+export class StepFunctionsEnrichment implements IEnrichment {
+  public readonly enrichmentArn: string;
+
+  private readonly inputTransformation?: InputTransformation;
+  private readonly stateMachine: IStateMachine;
+
+  constructor(stateMachine: IStateMachine, props?: StepFunctionsEnrichmentProps) {
+    this.stateMachine=stateMachine;
+    if (this.stateMachine instanceof StateMachine
+      && (this.stateMachine.stateMachineType === StateMachineType.STANDARD)
+    ) {
+      throw new Error('Enrichiment does not support STANDARD state machine workflows. Use EXPRESS instead.');
+    }
+    this.enrichmentArn = stateMachine.stateMachineArn;
+    this.inputTransformation = props?.inputTransformation;
+  }
+
+  bind(pipe: IPipe): EnrichmentParametersConfig {
+    return {
+      enrichmentParameters: {
+        inputTemplate: this.inputTransformation?.bind(pipe).inputTemplate,
+      },
+    };
+  }
+
+  grantInvoke(pipeRole: IRole): void {
+    this.stateMachine.grantStartSyncExecution(pipeRole);
+  }
+}
+

packages/@aws-cdk/aws-pipes-enrichments-alpha/rosetta/default.ts-fixture

@@ -2,6 +2,7 @@
 import * as cdk from 'aws-cdk-lib';
 import * as sqs from 'aws-cdk-lib/aws-sqs';
 import * as lambda from 'aws-cdk-lib/aws-lambda';
+import * as stepfunctions from 'aws-cdk-lib/aws-stepfunctions';
 import { Construct } from 'constructs';
 import * as pipes from '@aws-cdk/aws-pipes-alpha';
 import * as enrichments from '@aws-cdk/aws-pipes-enrichments-alpha';

packages/@aws-cdk/aws-pipes-enrichments-alpha/test/__snapshots__/stepfunctions.test.ts.snap

@@ -0,0 +1,76 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`stepfunctions should grant pipe role invoke access 1`] = `
+{
+  "EnrichmentStateMachineRoleDE810FCA": {
+    "Properties": {
+      "AssumeRolePolicyDocument": {
+        "Statement": [
+          {
+            "Action": "sts:AssumeRole",
+            "Effect": "Allow",
+            "Principal": {
+              "Service": {
+                "Fn::FindInMap": [
+                  "ServiceprincipalMap",
+                  {
+                    "Ref": "AWS::Region",
+                  },
+                  "states",
+                ],
+              },
+            },
+          },
+        ],
+        "Version": "2012-10-17",
+      },
+    },
+    "Type": "AWS::IAM::Role",
+  },
+  "MyPipeRoleCBC8E9AB": {
+    "Properties": {
+      "AssumeRolePolicyDocument": {
+        "Statement": [
+          {
+            "Action": "sts:AssumeRole",
+            "Effect": "Allow",
+            "Principal": {
+              "Service": "pipes.amazonaws.com",
+            },
+          },
+        ],
+        "Version": "2012-10-17",
+      },
+    },
+    "Type": "AWS::IAM::Role",
+  },
+}
+`;
+
+exports[`stepfunctions should grant pipe role invoke access 2`] = `
+{
+  "MyPipeRoleDefaultPolicy31387C20": {
+    "Properties": {
+      "PolicyDocument": {
+        "Statement": [
+          {
+            "Action": "states:StartSyncExecution",
+            "Effect": "Allow",
+            "Resource": {
+              "Ref": "EnrichmentStateMachine8BED6C4E",
+            },
+          },
+        ],
+        "Version": "2012-10-17",
+      },
+      "PolicyName": "MyPipeRoleDefaultPolicy31387C20",
+      "Roles": [
+        {
+          "Ref": "MyPipeRoleCBC8E9AB",
+        },
+      ],
+    },
+    "Type": "AWS::IAM::Policy",
+  },
+}
+`;