feat(EventBridge Scheduler): ✨ add logic for eventbridge scheduler

Author: notanmay

lib/deploy/events/schedule/compileScheduledEvents.js

@@ -10,6 +10,8 @@ module.exports = {
     _.forEach(this.getAllStateMachines(), (stateMachineName) => {
       const stateMachineObj = this.getStateMachine(stateMachineName);
       let scheduleNumberInFunction = 0;
+      const METHOD_SCHEDULER = 'scheduler';
+      const METHOD_EVENT_BUS = 'eventBus';
 
       if (stateMachineObj.events) {
         _.forEach(stateMachineObj.events, (event) => {
@@ -24,6 +26,8 @@ module.exports = {
             let InputPathsMap;
             let Name;
             let Description;
+            let method;
+            let timezone;
 
             // TODO validate rate syntax
             if (typeof event.schedule === 'object') {
@@ -49,6 +53,8 @@ module.exports = {
               InputTemplate = InputTransformer && event.schedule.inputTransformer.inputTemplate;
               Name = event.schedule.name;
               Description = event.schedule.description;
+              method = event.schedule.method || METHOD_EVENT_BUS;
+              timezone = event.schedule.timezone;
 
               if ([Input, InputPath, InputTransformer].filter(Boolean).length > 1) {
                 const errorMessage = [
@@ -76,6 +82,34 @@ module.exports = {
                 // escape quotes to favor JSON.parse
                 InputTemplate = InputTemplate.replace(/\"/g, '\\"'); // eslint-disable-line
               }
+              if (InputTransformer) {
+                if (method === METHOD_SCHEDULER) {
+                  const errorMessage = [
+                    'Cannot setup "schedule" event: "inputTransformer" is not supported with "scheduler" mode',
+                    'SCHEDULE_PARAMETER_NOT_SUPPORTED',
+                  ].join('');
+                  throw new this.serverless.classes
+                    .Error(errorMessage);
+                } else {
+                  InputTransformer = this.formatInputTransformer(InputTransformer);
+                }
+              }
+              if (InputPath && method === METHOD_SCHEDULER) {
+                const errorMessage = [
+                  'Cannot setup "schedule" event: "inputPath" is not supported with "scheduler" mode',
+                  'SCHEDULE_PARAMETER_NOT_SUPPORTED',
+                ].join('');
+                throw new this.serverless.classes
+                  .Error(errorMessage);
+              }
+              if (timezone && method !== METHOD_SCHEDULER) {
+                const errorMessage = [
+                  'Cannot setup "schedule" event: "timezone" is only supported with "scheduler" mode',
+                  'SCHEDULE_PARAMETER_NOT_SUPPORTED',
+                ].join('');
+                throw new this.serverless.classes
+                  .Error(errorMessage);
+              }
             } else if (typeof event.schedule === 'string') {
               ScheduleExpression = event.schedule;
               State = 'ENABLED';
@@ -110,8 +144,66 @@ module.exports = {
                 }
               `;
 
-            const scheduleTemplate = `
-              {
+            let scheduleTemplate;
+            let iamRoleTemplate;
+
+            if (method === METHOD_SCHEDULER) {
+              scheduleTemplate = `{
+              "Type": "AWS::Scheduler::Schedule",
+              "Properties": {
+                "ScheduleExpression": "${ScheduleExpression}",
+                "State": "${State}",
+                ${timezone ? `"ScheduleExpressionTimezone": "${timezone}",` : ''}
+                ${Name ? `"Name": "${Name}",` : ''}
+                ${Description ? `"Description": "${Description}",` : ''}
+                "Target": {
+                  "Arn": { "Ref": "${stateMachineLogicalId}" },
+                  "RoleArn": ${roleArn}
+                },
+                "FlexibleTimeWindow": {
+                  "Mode": "OFF"
+                }
+              }
+            }`;
+
+              iamRoleTemplate = `{
+            "Type": "AWS::IAM::Role",
+            "Properties": {
+              "AssumeRolePolicyDocument": {
+                "Version": "2012-10-17",
+                "Statement": [
+                  {
+                    "Effect": "Allow",
+                    "Principal": {
+                        "Service": "scheduler.amazonaws.com"
+                    },
+                    "Action": "sts:AssumeRole"
+                  }
+                ]
+              },
+              "Policies": [
+                {
+                  "PolicyName": "${policyName}",
+                  "PolicyDocument": {
+                    "Version": "2012-10-17",
+                    "Statement": [
+                      {
+                        "Effect": "Allow",
+                        "Action": [
+                          "states:StartExecution"
+                        ],
+                        "Resource": {
+                          "Ref": "${stateMachineLogicalId}"
+                        }
+                      }
+                    ]
+                  }
+                }
+              ]
+            }
+          }`;
+            } else {
+              scheduleTemplate = `{
                 "Type": "AWS::Events::Rule",
                 "Properties": {
                   "ScheduleExpression": "${ScheduleExpression}",
@@ -130,11 +222,8 @@ module.exports = {
                     "RoleArn": ${roleArn}
                   }]
                 }
-              }
-            `;
-
-            let iamRoleTemplate = `
-            {
+              }`;
+              iamRoleTemplate = `{
               "Type": "AWS::IAM::Role",
               "Properties": {
                 "AssumeRolePolicyDocument": {
@@ -169,8 +258,8 @@ module.exports = {
                   }
                 ]
               }
+            }`;
             }
-            `;
             if (permissionsBoundary) {
               const jsonIamRole = JSON.parse(iamRoleTemplate);
               jsonIamRole.Properties.PermissionsBoundary = permissionsBoundary;