Merge pull request #210 from gammarers/feature/add-machine-logging-option

feat: add machine logging option

Author: yicr

API.md

@@ -1,10 +1,16 @@
 import { Stack } from 'aws-cdk-lib';
 import * as iam from 'aws-cdk-lib/aws-iam';
+import * as logs from 'aws-cdk-lib/aws-logs';
 import * as scheduler from 'aws-cdk-lib/aws-scheduler';
 import * as sns from 'aws-cdk-lib/aws-sns';
+import { LogLevel as EC2InstanceRunningScheduleStackMachineLogLevel } from 'aws-cdk-lib/aws-stepfunctions';
 import { Construct } from 'constructs';
 import { RunningControlStateMachine } from './resources/running-control-state-machine';
 
+export {
+  EC2InstanceRunningScheduleStackMachineLogLevel,
+};
+
 export interface TargetResource {
   readonly tagKey: string;
   readonly tagValues: string[];
@@ -22,11 +28,16 @@ export interface Notifications {
   // readonly slack?: Slack;
 }
 
+export interface LogOption {
+  readonly machineLogLevel?: EC2InstanceRunningScheduleStackMachineLogLevel;
+}
+
 export interface EC2InstanceRunningScheduleStackProps {
   readonly targetResource: TargetResource;
   readonly stopSchedule?: Schedule;
   readonly startSchedule?: Schedule;
   readonly notifications?: Notifications;
+  readonly logOption?: LogOption;
 }
 
 export class EC2InstanceRunningScheduleStack extends Stack {
@@ -51,6 +62,22 @@ export class EC2InstanceRunningScheduleStack extends Stack {
     const machine = new RunningControlStateMachine(this, 'StateMachine', {
       stateMachineName: undefined,
       notificationTopic: topic,
+      logs: (() => {
+        if (props.logOption?.machineLogLevel) {
+          return {
+            destination: new logs.LogGroup(this, 'StateMachineLogGroup', {
+              logGroupName: (() => {
+                // if (names.stateMachineName) {
+                //   return `/aws/states/${names.stateMachineName}`;
+                // }
+                return undefined;
+              })(),
+            }),
+            level: props.logOption.machineLogLevel,
+          };
+        }
+        return undefined;
+      })(),
     });
 
     // 👇 EventBridge Scheduler IAM Role

src/index.ts

@@ -1,6 +1,6 @@
 import { App } from 'aws-cdk-lib';
 import { Template } from 'aws-cdk-lib/assertions';
-import { EC2InstanceRunningScheduleStack } from '../src';
+import { EC2InstanceRunningScheduleStack, EC2InstanceRunningScheduleStackMachineLogLevel } from '../src';
 
 describe('Ec2InstanceRunningScheduleStack specific Testing', () => {
 
@@ -29,6 +29,9 @@ describe('Ec2InstanceRunningScheduleStack specific Testing', () => {
         '[email protected]',
       ],
     },
+    logOption: {
+      machineLogLevel: EC2InstanceRunningScheduleStackMachineLogLevel.ALL,
+    },
   });
 
   const template = Template.fromStack(stack);