feat: support for yarn (#581)

fix #579

Author: abetomo

bin/node-lambda

@@ -11,6 +11,7 @@ const fs = require('fs')
 const packageJson = fs.existsSync(path.join(process.cwd(), 'package.json')) ? require(path.join(process.cwd(), 'package.json')) : {}
 const packageJsonName = packageJson.name || 'UnnamedFunction'
 
+const PACKAGE_MANAGER = process.env.PACKAGE_MANAGER || 'npm'
 const AWS_ENVIRONMENT = process.env.AWS_ENVIRONMENT || ''
 const AWS_ENDPOINT = process.env.AWS_ENDPOINT || ''
 const CONFIG_FILE = process.env.CONFIG_FILE || ''
@@ -66,6 +67,7 @@ program.showSuggestionAfterError()
 program
   .command('deploy')
   .description('Deploy your application to Amazon Lambda')
+  .option('--packageManager [PACKAGE_MANAGER]', 'Package manager', PACKAGE_MANAGER)
   .option('-e, --environment [AWS_ENVIRONMENT]', 'Choose environment {dev, staging, production}',
     AWS_ENVIRONMENT)
   .option('-E, --endpoint [AWS_ENDPOINT]', 'Choose endpoint (e.g. localstack, "http://127.0.0.1:4574")',
@@ -123,6 +125,7 @@ program
   .command('package')
   .alias('zip')
   .description('Create zipped package for Amazon Lambda deployment')
+  .option('--packageManager [PACKAGE_MANAGER]', 'Package manager', PACKAGE_MANAGER)
   .option('-A, --packageDirectory [PACKAGE_DIRECTORY]', 'Local Package Directory', PACKAGE_DIRECTORY)
   .option('-I, --dockerImage [DOCKER_IMAGE]', 'Docker image for npm ci', DOCKER_IMAGE)
   .option('-n, --functionName [AWS_FUNCTION_NAME]', 'Lambda FunctionName', AWS_FUNCTION_NAME)

lib/main.js

@@ -407,11 +407,38 @@ Emulate only the body of the API Gateway event.
     }
   }
 
+  _getYarnInstallCommand (program, codeDirectory) {
+    const installOptions = [
+      '-s',
+      'install',
+      '--production'
+    ]
+
+    if (program.optionalDependencies === false) {
+      installOptions.push('--ignore-optional')
+    }
+
+    if (!program.dockerImage) {
+      installOptions.push('--cwd', codeDirectory)
+    }
+
+    return {
+      packageManager: 'yarn',
+      installOptions
+    }
+  }
+
   _packageInstall (program, codeDirectory) {
     // Run on windows:
     // https://nodejs.org/api/child_process.html#child_process_spawning_bat_and_cmd_files_on_windows
 
-    const { packageManager, installOptions } = this._getNpmInstallCommand(program, codeDirectory)
+    const { packageManager, installOptions } = (() => {
+      // default npm
+      if (program.packageManager === 'yarn') {
+        return this._getYarnInstallCommand(program, codeDirectory)
+      }
+      return this._getNpmInstallCommand(program, codeDirectory)
+    })()
 
     const paramsOnContainer = (() => {
       // with docker
@@ -459,7 +486,7 @@ Emulate only the body of the API Gateway event.
         env: process.env
       }, (err) => {
         if (err) return reject(err)
-        resolve()
+        resolve(packageManager)
       })
     })
   }
@@ -701,7 +728,8 @@ they may not work as expected in the Lambda environment.
     await this._fileCopy(program, lambdaSrcDirectory, codeDirectory, true)
     if (!program.keepNodeModules) {
       console.log('=> Running package install')
-      await this._packageInstall(program, codeDirectory)
+      const usedPackageManager = await this._packageInstall(program, codeDirectory)
+      console.log(`(Package manager used was '${usedPackageManager}'.)`)
     }
     await this._postInstallScript(program, codeDirectory)
     console.log('=> Zipping deployment package')