feat: MSK, MSK CONNECT and SES

packages/api/src/adapter-aws/AwsServerlessAdapter.ts

@@ -5,6 +5,7 @@ import {
   AddPermissionCommand,
   DeleteFunctionCommand,
   DeleteEventSourceMappingCommand,
+  UpdateEventSourceMappingCommand,
   GetFunctionCommand,
   InvokeCommand,
   type InvokeCommandOutput,
@@ -34,6 +35,8 @@ import type {
   EventBridgeLambdaTriggerOverview,
   SqsLambdaEventSourceMappingInput,
   SqsLambdaEventSourceMappingOverview,
+  LambdaEventSourceMappingInput,
+  LambdaEventSourceMappingOverview,
   ServerlessInvokeResult,
   ServiceSchema,
   UpdateResourceInput,
@@ -385,6 +388,57 @@ export class AwsServerlessAdapter implements CloudServiceAdapter {
     await this.lambda.send(new DeleteEventSourceMappingCommand({UUID: uuid}))
   }
 
+  async createLambdaEventSourceMapping(input: LambdaEventSourceMappingInput): Promise<LambdaEventSourceMappingOverview> {
+    const functionName = stringValue(input.functionName)
+    const eventSourceArn = stringValue(input.eventSourceArn)
+    if (!functionName) throw new ValidationError('functionName is required')
+    if (!eventSourceArn) throw new ValidationError('eventSourceArn is required')
+    const isKinesis = eventSourceArn.includes(':kinesis:')
+    if (isKinesis && !input.startingPosition) throw new ValidationError('startingPosition is required for Kinesis event sources')
+    const mapping = await this.lambda.send(new CreateEventSourceMappingCommand({
+      FunctionName: functionName,
+      EventSourceArn: eventSourceArn,
+      ...(input.batchSize === undefined ? {} : {BatchSize: eventSourceBatchSize(input.batchSize, isKinesis ? 10_000 : 10)}),
+      ...(input.startingPosition ? {StartingPosition: input.startingPosition} : {}),
+    }))
+    return lambdaEventSourceMappingOverview(mapping)
+  }
+
+  async listLambdaEventSourceMappings(filter: {functionName?: string; eventSourceArn?: string} = {}): Promise<LambdaEventSourceMappingOverview[]> {
+    const mappings: LambdaEventSourceMappingOverview[] = []
+    let marker: string | undefined
+    do {
+      const result = await this.lambda.send(new ListEventSourceMappingsCommand({
+        ...(filter.functionName ? {FunctionName: filter.functionName} : {}),
+        ...(filter.eventSourceArn ? {EventSourceArn: filter.eventSourceArn} : {}),
+        Marker: marker,
+      }))
+      for (const mapping of result.EventSourceMappings ?? []) {
+        if (mapping.UUID && mapping.EventSourceArn) mappings.push(lambdaEventSourceMappingOverview(mapping))
+      }
+      marker = result.NextMarker
+    } while (marker)
+    return mappings.sort((left, right) => left.uuid.localeCompare(right.uuid))
+  }
+
+  async updateLambdaEventSourceMapping(mappingId: string, input: {batchSize?: number; enabled?: boolean}): Promise<LambdaEventSourceMappingOverview> {
+    const uuid = stringValue(mappingId)
+    if (!uuid) throw new ValidationError('mappingId is required')
+    if (input.batchSize === undefined && input.enabled === undefined) throw new ValidationError('batchSize or enabled is required')
+    const mapping = await this.lambda.send(new UpdateEventSourceMappingCommand({
+      UUID: uuid,
+      ...(input.batchSize === undefined ? {} : {BatchSize: eventSourceBatchSize(input.batchSize, 10_000)}),
+      ...(input.enabled === undefined ? {} : {Enabled: input.enabled}),
+    }))
+    return lambdaEventSourceMappingOverview(mapping)
+  }
+
+  async deleteLambdaEventSourceMapping(mappingId: string): Promise<void> {
+    const uuid = stringValue(mappingId)
+    if (!uuid) throw new ValidationError('mappingId is required')
+    await this.lambda.send(new DeleteEventSourceMappingCommand({UUID: uuid}))
+  }
+
   private async listEventBridgeTargets(ruleName: string) {
     const targets = []
     let nextToken: string | undefined
@@ -457,6 +511,22 @@ function sqsBatchSize(value: number): number {
   return value
 }
 
+function eventSourceBatchSize(value: number, maximum: number): number {
+  if (!Number.isInteger(value) || value < 1 || value > maximum) throw new ValidationError(`batchSize must be an integer between 1 and ${maximum}`)
+  return value
+}
+
+function lambdaEventSourceMappingOverview(
+  mapping: {UUID?: string; FunctionArn?: string; EventSourceArn?: string; BatchSize?: number; StartingPosition?: string; State?: string; StateTransitionReason?: string},
+): LambdaEventSourceMappingOverview {
+  if (!mapping.UUID || !mapping.EventSourceArn) throw new ValidationError('Lambda event source mapping did not return a UUID and event source ARN')
+  return {
+    id: mapping.UUID, uuid: mapping.UUID, functionName: mapping.FunctionArn ? lambdaFunctionName(mapping.FunctionArn) : null,
+    functionArn: mapping.FunctionArn ?? null, eventSourceArn: mapping.EventSourceArn, batchSize: mapping.BatchSize ?? null,
+    startingPosition: mapping.StartingPosition ?? null, state: mapping.State ?? null, stateTransitionReason: mapping.StateTransitionReason ?? null,
+  }
+}
+
 function sqsEventSourceMappingOverview(
   mapping: {UUID?: string; FunctionArn?: string; EventSourceArn?: string; BatchSize?: number; State?: string; StateTransitionReason?: string},
   functionName: string,