feat: support v3 new log api

lopburny · lopburny · commit 87bf472b46ae · 2021-12-07T17:13:12.000Z
diff --git a/lib/deploy/stepFunctions/compileAlarms.js b/lib/deploy/stepFunctions/compileAlarms.js
@@ -5,6 +5,8 @@ const BbPromise = require('bluebird');
 const Joi = require('@hapi/joi');
 const schema = require('./compileAlarms.schema');
 
+let log;
+
 const cloudWatchMetricNames = {
   executionsTimedOut: 'ExecutionsTimedOut',
   executionsFailed: 'ExecutionsFailed',
@@ -39,7 +41,7 @@ function getCloudWatchAlarms(
   );
 
   if (!_.isEmpty(invalid)) {
-    serverless.cli.consoleLog(
+    log(
       `state machine [${stateMachineName}] : alarms.metrics has invalid metrics `,
       `[${invalid.join(',')}]. `
       + 'No CloudWatch Alarms would be created for these. '
@@ -96,7 +98,7 @@ function validateConfig(serverless, stateMachineName, alarmsObj) {
   const { error } = Joi.validate(alarmsObj, schema, { allowUnknown: false });
 
   if (error) {
-    serverless.cli.consoleLog(
+    log(
       `State machine [${stateMachineName}] : alarms config is malformed. `
       + 'Please see https://github.com/horike37/serverless-step-functions for examples. '
       + `${error}`,
@@ -109,6 +111,7 @@ function validateConfig(serverless, stateMachineName, alarmsObj) {
 
 module.exports = {
   compileAlarms() {
+    log = this.log;
     const cloudWatchAlarms = _.flatMap(this.getAllStateMachines(), (name) => {
       const stateMachineObj = this.getStateMachine(name);
       const stateMachineLogicalId = this.getStateMachineLogicalId(name, stateMachineObj);
diff --git a/lib/deploy/stepFunctions/compileIamRole.js b/lib/deploy/stepFunctions/compileIamRole.js
@@ -6,6 +6,8 @@ const path = require('path');
 const { isIntrinsic, translateLocalFunctionNames, trimAliasFromLambdaArn } = require('../../utils/aws');
 const { getArnPartition } = require('../../utils/arn');
 
+let log;
+
 function getTaskStates(states) {
   return _.flatMap(states, (state) => {
     switch (state.Type) {
@@ -49,7 +51,7 @@ function sqsQueueUrlToArn(serverless, queueUrl) {
     // we need to use "*" as ARN
     return '*';
   }
-  serverless.cli.consoleLog(`Unable to parse SQS queue url [${queueUrl}]`);
+  log(`Unable to parse SQS queue url [${queueUrl}]`);
   return [];
 }
 
@@ -62,7 +64,7 @@ function getSqsPermissions(serverless, state) {
       : sqsQueueUrlToArn(serverless, state.Parameters.QueueUrl);
     return [{ action: 'sqs:SendMessage', resource: queueArn }];
   }
-  serverless.cli.consoleLog('SQS task missing Parameters.QueueUrl or Parameters.QueueUrl.$');
+  log('SQS task missing Parameters.QueueUrl or Parameters.QueueUrl.$');
   return [];
 }
 
@@ -73,7 +75,7 @@ function getSnsPermissions(serverless, state) {
     const topicArn = state.Parameters['TopicArn.$'] ? '*' : state.Parameters.TopicArn;
     return [{ action: 'sns:Publish', resource: topicArn }];
   }
-  serverless.cli.consoleLog('SNS task missing Parameters.TopicArn or Parameters.TopicArn.$');
+  log('SNS task missing Parameters.TopicArn or Parameters.TopicArn.$');
   return [];
 }
 
@@ -478,7 +480,7 @@ function getIamPermissions(taskStates) {
             ],
           }];
         }
-        this.serverless.cli.consoleLog('Cannot generate IAM policy statement for Task state', state);
+        this.log('Cannot generate IAM policy statement for Task state', state);
         return [];
     }
   });
@@ -505,6 +507,7 @@ function getIamStatements(iamPermissions) {
 
 module.exports = {
   compileIamRole() {
+    log = this.log;
     this.getAllStateMachines().forEach((stateMachineName) => {
       const stateMachineObj = this.getStateMachine(stateMachineName);
       if (stateMachineObj.role) {
diff --git a/lib/deploy/stepFunctions/compileNotifications.js b/lib/deploy/stepFunctions/compileNotifications.js
@@ -6,6 +6,8 @@ const crypto = require('crypto');
 const BbPromise = require('bluebird');
 const schema = require('./compileNotifications.schema');
 
+let log;
+
 const executionStatuses = [
   'ABORTED', 'FAILED', 'RUNNING', 'SUCCEEDED', 'TIMED_OUT',
 ];
@@ -318,7 +320,7 @@ function validateConfig(serverless, stateMachineName, stateMachineObj, notificat
   }
 
   if (stateMachineObj.type === 'EXPRESS') {
-    serverless.cli.consoleLog(
+    log(
       `State machine [${stateMachineName}] : notifications are not supported on Express Workflows. `
       + 'Please see https://docs.aws.amazon.com/step-functions/latest/dg/concepts-standard-vs-express.html for difference between Step Functions Standard and Express Workflow.',
     );
@@ -330,7 +332,7 @@ function validateConfig(serverless, stateMachineName, stateMachineObj, notificat
   );
 
   if (error) {
-    serverless.cli.consoleLog(
+    log(
       `State machine [${stateMachineName}] : notifications config is malformed. `
       + 'Please see https://github.com/horike37/serverless-step-functions for examples. '
       + `${error}`,
@@ -343,6 +345,7 @@ function validateConfig(serverless, stateMachineName, stateMachineObj, notificat
 
 module.exports = {
   compileNotifications() {
+    log = this.log;
     const newResourcePairs = _.flatMap(this.getAllStateMachines(), (name) => {
       const stateMachineObj = this.getStateMachine(name);
       const stateMachineLogicalId = this.getStateMachineLogicalId(name, stateMachineObj);
diff --git a/lib/deploy/stepFunctions/compileStateMachines.js b/lib/deploy/stepFunctions/compileStateMachines.js
@@ -119,7 +119,7 @@ module.exports = {
           if (this.serverless.service.stepFunctions.validate) {
             const { isValid, errorsText } = aslValidator(stateMachineObj.definition);
             if (isValid) {
-              this.serverless.cli.consoleLog(`✓ State machine "${stateMachineName}" definition is valid`);
+              this.log(`✓ State machine "${stateMachineName}" definition is valid`);
             } else {
               const errorMessage = [
                 `✕ State machine "${stateMachineName}" definition is invalid:`,
diff --git a/lib/deploy/stepFunctions/compileStateMachines.test.js b/lib/deploy/stepFunctions/compileStateMachines.test.js
@@ -3,6 +3,7 @@
 const _ = require('lodash');
 const expect = require('chai').expect;
 const Serverless = require('serverless/lib/Serverless');
+const CLI = require('serverless/lib/classes/CLI');
 const AwsProvider = require('serverless/lib/plugins/aws/provider');
 const ServerlessStepFunctions = require('./../../index');
 
@@ -12,9 +13,7 @@ describe('#compileStateMachines', () => {
 
   beforeEach(() => {
     serverless = new Serverless();
-    serverless.cli = {
-      consoleLog: () => {},
-    };
+    serverless.cli = new CLI(serverless);
     serverless.configSchemaHandler = {
       // eslint-disable-next-line no-unused-vars
       defineTopLevelProperty: (propertyName, propertySchema) => {},
diff --git a/lib/index.js b/lib/index.js
@@ -29,14 +29,17 @@ const invoke = require('./invoke/invoke');
 const yamlParser = require('./yamlParser');
 const naming = require('./naming');
 
+const Logger = require('./utils/logger');
+
 class ServerlessStepFunctions {
-  constructor(serverless, options) {
+  constructor(serverless, options, v3Api) {
     this.serverless = serverless;
     this.options = options || {};
     this.provider = this.serverless.getProvider('aws');
     this.service = this.serverless.service.service;
     this.region = this.provider.getRegion();
     this.stage = this.provider.getStage();
+    this.log = new Logger(serverless, v3Api).log;
     Object.assign(
       this,
       compileStateMachines,
@@ -171,17 +174,17 @@ class ServerlessStepFunctions {
       .then(this.startExecution)
       .then(this.describeExecution)
       .then((result) => {
-        this.serverless.cli.consoleLog('');
+        this.log('');
         if (result.status === 'FAILED') {
           return this.getExecutionHistory()
             .then((error) => {
-              this.serverless.cli.consoleLog(_.merge(result, error.events[error.events.length - 1]
+              this.log(_.merge(result, error.events[error.events.length - 1]
                 .executionFailedEventDetails));
               process.exitCode = 1;
             });
         }
 
-        this.serverless.cli.consoleLog(result);
+        this.log(result);
         return BbPromise.resolve();
       });
   }
@@ -225,7 +228,7 @@ class ServerlessStepFunctions {
     message += stateMachineMessages;
     message += '\n';
 
-    this.serverless.cli.consoleLog(message);
+    this.log(message);
 
     return message;
   }
diff --git a/lib/index.test.js b/lib/index.test.js
@@ -287,22 +287,23 @@ describe('#index', () => {
   });
 
   describe('#deploy()', () => {
-    let consoleLogStub;
+    let logStub;
 
     beforeEach(() => {
-      consoleLogStub = sinon.stub(serverless.cli, 'consoleLog').returns();
+      logStub = sinon.stub();
       serverlessStepFunctions.endpointInfo = 'https://example.com';
+      serverlessStepFunctions.log = logStub;
     });
 
     afterEach(() => {
-      serverlessStepFunctions.serverless.cli.consoleLog.restore();
+      serverlessStepFunctions.log = null;
     });
 
     it('should not display endpoints if http event not given', () => {
       serverlessStepFunctions.serverless.service.stepFunctions = {};
       const expectedMessage = '';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(false);
+      expect(logStub.calledOnce).to.equal(false);
       expect(message).to.equal(expectedMessage);
     });
 
@@ -328,7 +329,7 @@ describe('#index', () => {
       expectedMessage += '\n  POST - https://example.com/foo/bar';
       expectedMessage += '\n';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(logStub.calledOnce).to.equal(true);
       expect(message).to.equal(expectedMessage);
     });
 
@@ -351,7 +352,7 @@ describe('#index', () => {
       expectedMessage += '\n  POST - https://example.com/foo/bar';
       expectedMessage += '\n';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(logStub.calledOnce).to.equal(true);
       expect(message).to.equal(expectedMessage);
     });
 
@@ -374,7 +375,7 @@ describe('#index', () => {
       expectedMessage += '\n  POST - https://example.com';
       expectedMessage += '\n';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(logStub.calledOnce).to.equal(true);
       expect(message).to.equal(expectedMessage);
     });
 
@@ -389,7 +390,7 @@ describe('#index', () => {
       };
       const expectedMessage = '';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(false);
+      expect(logStub.calledOnce).to.equal(false);
       expect(message).to.equal(expectedMessage);
     });
 
@@ -408,7 +409,7 @@ describe('#index', () => {
       };
       const expectedMessage = '';
       const message = serverlessStepFunctions.display();
-      expect(consoleLogStub.calledOnce).to.equal(false);
+      expect(logStub.calledOnce).to.equal(false);
       expect(message).to.equal(expectedMessage);
     });
   });
diff --git a/lib/utils/logger.js b/lib/utils/logger.js
@@ -0,0 +1,13 @@
+'use strict';
+
+class Logger {
+  constructor(serverless, v3Api) {
+    if (v3Api) {
+      this.log = v3Api.log;
+    } else {
+      this.log = serverless.cli && serverless.cli.consoleLog;
+    }
+  }
+}
+
+module.exports = Logger;
diff --git a/lib/utils/logger.test.js b/lib/utils/logger.test.js
@@ -0,0 +1,39 @@
+'use strict';
+
+const expect = require('chai').expect;
+const sinon = require('sinon');
+const Serverless = require('serverless/lib/Serverless');
+const Logger = require('../utils/logger');
+
+describe('logger', () => {
+  let serverless;
+  let v3LogSpy;
+
+  beforeEach(() => {
+    serverless = new Serverless();
+    serverless.cli = { consoleLog: sinon.spy() };
+
+    v3LogSpy = sinon.spy();
+  });
+
+  afterEach(() => {
+    serverless.cli = null;
+    v3LogSpy = null;
+  });
+
+  it('should be assigned legacy logger (serverless.cli.consoleLog) if v3 log api is not supplied', () => {
+    const log = new Logger(serverless).log;
+    log('Testing purpose');
+
+    expect(serverless.cli.consoleLog.callCount).to.equal(1);
+    expect(v3LogSpy.callCount).to.equal(0);
+  });
+
+  it('should be assigned new log API if new log api is supplied', () => {
+    const log = new Logger(serverless, { log: v3LogSpy }).log;
+    log('Testing purpose');
+
+    expect(serverless.cli.consoleLog.callCount).to.equal(0);
+    expect(v3LogSpy.callCount).to.equal(1);
+  });
+});
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,8 @@ const path = require('path');`
`6`	`6`	`const { isIntrinsic, translateLocalFunctionNames, trimAliasFromLambdaArn } = require('../../utils/aws');`
`7`	`7`	`const { getArnPartition } = require('../../utils/arn');`
`8`	`8`
	`9`	`+let log;`
	`10`	`+`
`9`	`11`	`function getTaskStates(states) {`
`10`	`12`	`return _.flatMap(states, (state) => {`
`11`	`13`	`switch (state.Type) {`
`@@ -49,7 +51,7 @@ function sqsQueueUrlToArn(serverless, queueUrl) {`
`49`	`51`	`// we need to use "*" as ARN`
`50`	`52`	`return '*';`
`51`	`53`	`}`
`52`		- serverless.cli.consoleLog(`Unable to parse SQS queue url [${queueUrl}]`);
	`54`	+ log(`Unable to parse SQS queue url [${queueUrl}]`);
`53`	`55`	`return [];`
`54`	`56`	`}`
`55`	`57`
`@@ -62,7 +64,7 @@ function getSqsPermissions(serverless, state) {`
`62`	`64`	`: sqsQueueUrlToArn(serverless, state.Parameters.QueueUrl);`
`63`	`65`	`return [{ action: 'sqs:SendMessage', resource: queueArn }];`
`64`	`66`	`}`
`65`		`- serverless.cli.consoleLog('SQS task missing Parameters.QueueUrl or Parameters.QueueUrl.$');`
	`67`	`+ log('SQS task missing Parameters.QueueUrl or Parameters.QueueUrl.$');`
`66`	`68`	`return [];`
`67`	`69`	`}`
`68`	`70`
`@@ -73,7 +75,7 @@ function getSnsPermissions(serverless, state) {`
`73`	`75`	`const topicArn = state.Parameters['TopicArn.$'] ? '*' : state.Parameters.TopicArn;`
`74`	`76`	`return [{ action: 'sns:Publish', resource: topicArn }];`
`75`	`77`	`}`
`76`		`- serverless.cli.consoleLog('SNS task missing Parameters.TopicArn or Parameters.TopicArn.$');`
	`78`	`+ log('SNS task missing Parameters.TopicArn or Parameters.TopicArn.$');`
`77`	`79`	`return [];`
`78`	`80`	`}`
`79`	`81`
`@@ -478,7 +480,7 @@ function getIamPermissions(taskStates) {`
`478`	`480`	`],`
`479`	`481`	`}];`
`480`	`482`	`}`
`481`		`- this.serverless.cli.consoleLog('Cannot generate IAM policy statement for Task state', state);`
	`483`	`+ this.log('Cannot generate IAM policy statement for Task state', state);`
`482`	`484`	`return [];`
`483`	`485`	`}`
`484`	`486`	`});`
`@@ -505,6 +507,7 @@ function getIamStatements(iamPermissions) {`
`505`	`507`
`506`	`508`	`module.exports = {`
`507`	`509`	`compileIamRole() {`
	`510`	`+ log = this.log;`
`508`	`511`	`this.getAllStateMachines().forEach((stateMachineName) => {`
`509`	`512`	`const stateMachineObj = this.getStateMachine(stateMachineName);`
`510`	`513`	`if (stateMachineObj.role) {`