feat: appsync datasource to eventbus connector (#3311)

Co-authored-by: Slava Senchenko <[email protected]>

Author: paulhcsun

integration/combination/test_connectors.py

@@ -62,6 +62,7 @@ def tearDown(self):
             ("combination/connector_appsync_api_to_lambda",),
             ("combination/connector_appsync_to_lambda",),
             ("combination/connector_appsync_to_table",),
+            ("combination/connector_appsync_to_eventbus",),
             ("combination/connector_function_to_function",),
             ("combination/connector_restapi_to_function",),
             ("combination/connector_httpapi_to_function",),

integration/resources/expected/combination/connector_appsync_to_eventbus.json

@@ -0,0 +1,38 @@
+[
+  {
+    "LogicalResourceId": "ApiKey",
+    "ResourceType": "AWS::AppSync::ApiKey"
+  },
+  {
+    "LogicalResourceId": "ApiSchema",
+    "ResourceType": "AWS::AppSync::GraphQLSchema"
+  },
+  {
+    "LogicalResourceId": "AppSyncApi",
+    "ResourceType": "AWS::AppSync::GraphQLApi"
+  },
+  {
+    "LogicalResourceId": "AppSyncEventBusDataSource",
+    "ResourceType": "AWS::AppSync::DataSource"
+  },
+  {
+    "LogicalResourceId": "AppSyncSayHelloResolver",
+    "ResourceType": "AWS::AppSync::Resolver"
+  },
+  {
+    "LogicalResourceId": "ConnectorPolicy",
+    "ResourceType": "AWS::IAM::ManagedPolicy"
+  },
+  {
+    "LogicalResourceId": "EventBridgeRole",
+    "ResourceType": "AWS::IAM::Role"
+  },
+  {
+    "LogicalResourceId": "EventBus",
+    "ResourceType": "AWS::Events::EventBus"
+  },
+  {
+    "LogicalResourceId": "TriggerFunction",
+    "ResourceType": "AWS::Lambda::Function"
+  }
+]

integration/resources/templates/combination/connector_appsync_to_eventbus.yaml

@@ -0,0 +1,192 @@
+Resources:
+  EventBus:
+    Type: AWS::Events::EventBus
+    Properties:
+      Name: !Sub "${AWS::StackName}-EventBus"
+
+  EventBridgeRole:
+    Type: AWS::IAM::Role
+    Properties:
+      AssumeRolePolicyDocument:
+        Version: '2012-10-17'
+        Statement:
+        - Effect: Allow
+          Action:
+          - sts:AssumeRole
+          Principal:
+            Service:
+            - appsync.amazonaws.com
+            - lambda.amazonaws.com
+
+  AppSyncApi:
+    Type: AWS::AppSync::GraphQLApi
+    Properties:
+      Name: AppSyncApi
+      AuthenticationType: API_KEY
+
+  ApiSchema:
+    Type: AWS::AppSync::GraphQLSchema
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+      Definition: |
+        type EntryDetails {
+          ErrorCode: String
+          ErrorMessage: String
+          EventId: String!
+        }
+
+        type PutEventsResult {
+          Entries: [EntryDetails!]!
+          FailedEntry: Int
+        }
+
+        type Query {
+          sayHello: PutEventsResult!
+        }
+
+        schema {
+          query: Query
+        }
+
+  AppSyncEventBusDataSource:
+    Type: AWS::AppSync::DataSource
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+      Name: AppSyncEventBusDataSource
+      Type: AMAZON_EVENTBRIDGE
+      ServiceRoleArn: !GetAtt EventBridgeRole.Arn
+      EventBridgeConfig:
+        EventBusArn: !GetAtt 'EventBus.Arn'
+
+  AppSyncSayHelloResolver:
+    DependsOn: ApiSchema
+    Type: AWS::AppSync::Resolver
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+      TypeName: Query
+      FieldName: sayHello
+      DataSourceName: !GetAtt AppSyncEventBusDataSource.Name
+      Runtime:
+        Name: APPSYNC_JS
+        RuntimeVersion: 1.0.0
+      Code: |
+        import { util } from '@aws-appsync/utils';
+        export function request(ctx) {
+          return {
+            "operation" : "PutEvents",
+            "events" : [{
+              "source": "com.mycompany.myapp",
+              "detail": {
+                "key1" : "value1",
+                "key2" : "value2"
+              },
+              "resources": ["Resource1", "Resource2"],
+              "detailType": "myDetailType"
+            }]
+          }
+        }
+
+        export function response(ctx) {
+          if(ctx.error)
+            util.error(ctx.error.message, ctx.error.type, ctx.result)
+          else
+            return ctx.result
+        }
+
+  Connector:
+    Type: AWS::Serverless::Connector
+    Properties:
+      Source:
+        Id: AppSyncEventBusDataSource
+      Destination:
+        Id: EventBus
+      Permissions:
+      - Write
+
+  ApiKey:
+    Type: AWS::AppSync::ApiKey
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+
+  TriggerFunction:
+    Type: AWS::Serverless::Function
+    Properties:
+      Role: !GetAtt EventBridgeRole.Arn
+      Environment:
+        Variables:
+          API_KEY: !GetAtt ApiKey.ApiKey
+          GRAPHQL_URL: !GetAtt AppSyncApi.GraphQLUrl
+          EventBusName: !Ref EventBus
+      Runtime: nodejs16.x
+      Handler: index.handler
+      InlineCode: |
+        const https = require("https");
+
+        exports.handler = async () => {
+          const queries = {
+            sayHello: /* GraphQL */ `
+              query {
+                sayHello {
+                  Entries {
+                    ErrorCode
+                    EventId
+                    ErrorMessage
+                  }
+                  FailedEntry
+                }
+              }
+            `,
+          };
+
+          const fetch = async (url, options) =>
+            new Promise((resolve, reject) => {
+              const req = https.request(url, options, (res) => {
+                const body = [];
+                res.on("data", (chunk) => body.push(chunk));
+                res.on("end", () => {
+                  const resString = Buffer.concat(body).toString();
+                  resolve(resString);
+                });
+              });
+              req.on("error", (err) => {
+                reject(err);
+              });
+              req.on("timeout", () => {
+                req.destroy();
+                reject(new Error("Request time out"));
+              });
+              req.write(options.body);
+              req.end();
+            });
+
+          const makeRequest = async (queryName) => {
+            const options = {
+              method: "POST",
+              headers: {
+                "x-api-key": process.env.API_KEY,
+              },
+              body: JSON.stringify({ query: queries[queryName] }),
+              timeout: 600000, // ms
+            };
+
+            const response = await fetch(process.env.GRAPHQL_URL, options);
+            let body = JSON.parse(response);
+            const data = body.data?.[queryName];
+            
+            if (body.errors !== undefined) {
+              throw JSON.stringify(body.errors);
+            }
+            
+            if (data.FailedEntry != null || data.ErrorCode != null ) {
+              throw new Error(
+                `${queryName} error: failed to send event to eventbus ${process.env.EventBusName}`);
+            }
+            
+            return body.data;
+          };
+
+          await makeRequest("sayHello");
+        };
+
+Metadata:
+  SamTransformTest: true

samtranslator/model/connector_profiles/profiles.json

@@ -790,6 +790,23 @@
             }
           }
         }
+      },
+      "AWS::Events::EventBus": {
+        "Type": "AWS_IAM_ROLE_MANAGED_POLICY",
+        "Properties": {
+          "SourcePolicy": true,
+          "AccessCategories": {
+            "Write": {
+              "Statement": [
+                {
+                  "Effect": "Allow",
+                  "Action": ["events:PutEvents"],
+                  "Resource": ["%{Destination.Arn}"]
+                }
+              ]
+            }
+          }
+        }
       }
     },
     "AWS::AppSync::GraphQLApi": {

tests/translator/input/connector_appsync_to_eventbus.yaml

@@ -0,0 +1,66 @@
+Resources:
+  EventBus:
+    Type: AWS::Events::EventBus
+    Properties:
+      Name: !Sub '${AWS::StackName}-EventBus'
+
+  EventBridgeRole:
+    Type: AWS::IAM::Role
+    Properties:
+      RoleName: appsync-eventbridge-role
+      AssumeRolePolicyDocument:
+        Version: '2012-10-17'
+        Statement:
+        - Effect: Allow
+          Action:
+          - sts:AssumeRole
+          Principal:
+            Service:
+            - appsync.amazonaws.com
+
+  AppSyncEventBusDataSource:
+    Type: AWS::AppSync::DataSource
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+      Name: MyDataSource
+      Type: AMAZON_EVENTBRIDGE
+      ServiceRoleArn: !GetAtt EventBridgeRole.Arn
+      EventBridgeConfig:
+        EventBusArn: !GetAtt 'EventBus.Arn'
+
+  AppSyncApi:
+    Type: AWS::AppSync::GraphQLApi
+    Properties:
+      AuthenticationType: AWS_IAM
+      Name: AppSyncApi
+
+  ApiSchema:
+    Type: AWS::AppSync::GraphQLSchema
+    Properties:
+      ApiId: !GetAtt AppSyncApi.ApiId
+      Definition: |
+        type Note {
+          NoteId: ID!
+          title: String
+          content: String
+        }
+        type Query {
+          getNote(NoteId: ID!): Note
+        }
+        type Mutation {
+          saveNote(NoteId: ID!, title: String!, content: String!): Note!
+        }
+        type Schema {
+          query: Query
+          mutation: Mutation
+        }
+
+  Connector:
+    Type: AWS::Serverless::Connector
+    Properties:
+      Source:
+        Id: AppSyncEventBusDataSource
+      Destination:
+        Id: EventBus
+      Permissions:
+      - Write