Add file to configure aws authentication settings (#482)

Author: abetomo

lib/aws.js

@@ -0,0 +1,38 @@
+'use strict'
+
+const aws = require('aws-sdk')
+const proxy = require('proxy-agent')
+
+module.exports = {
+  sdk: aws,
+  updateConfig (config, region) {
+    const awsSecurity = { region: region }
+
+    if (config.profile) {
+      aws.config.credentials = new aws.SharedIniFileCredentials({
+        profile: config.profile
+      })
+    } else {
+      awsSecurity.accessKeyId = config.accessKey
+      awsSecurity.secretAccessKey = config.secretKey
+    }
+
+    if (config.sessionToken) {
+      awsSecurity.sessionToken = config.sessionToken
+    }
+
+    if (config.deployTimeout) {
+      aws.config.httpOptions.timeout = parseInt(config.deployTimeout)
+    }
+
+    if (config.proxy) {
+      aws.config.httpOptions.agent = proxy(config.proxy)
+    }
+
+    if (config.endpoint) {
+      aws.config.endpoint = config.endpoint
+    }
+
+    aws.config.update(awsSecurity)
+  }
+}

lib/main.js

@@ -2,15 +2,14 @@
 
 const path = require('path')
 const os = require('os')
-const aws = require('aws-sdk')
+const aws = require(path.join(__dirname, 'aws'))
 const { exec, execSync, execFile } = require('child_process')
 const fs = require('fs-extra')
 const klaw = require('klaw')
 const packageJson = require(path.join(__dirname, '..', 'package.json'))
 const minimatch = require('minimatch')
 const archiver = require('archiver')
 const dotenv = require('dotenv')
-const proxy = require('proxy-agent')
 const ScheduleEvents = require(path.join(__dirname, 'schedule_events'))
 const S3Events = require(path.join(__dirname, 's3_events'))
 const S3Deploy = require(path.join(__dirname, 's3_deploy'))
@@ -822,51 +821,20 @@ they may not work as expected in the Lambda environment.
     })
   }
 
-  _awsConfigUpdate (program, region) {
-    const awsSecurity = { region: region }
-
-    if (program.profile) {
-      aws.config.credentials = new aws.SharedIniFileCredentials({
-        profile: program.profile
-      })
-    } else {
-      awsSecurity.accessKeyId = program.accessKey
-      awsSecurity.secretAccessKey = program.secretKey
-    }
-
-    if (program.sessionToken) {
-      awsSecurity.sessionToken = program.sessionToken
-    }
-
-    if (program.deployTimeout) {
-      aws.config.httpOptions.timeout = parseInt(program.deployTimeout)
-    }
-
-    if (program.proxy) {
-      aws.config.httpOptions.agent = proxy(program.proxy)
-    }
-
-    if (program.endpoint) {
-      aws.config.endpoint = program.endpoint
-    }
-
-    aws.config.update(awsSecurity)
-  }
-
   _isFunctionDoesNotExist (err) {
     return err.code === 'ResourceNotFoundException' &&
       !!err.message.match(/^Function not found:/)
   }
 
   _deployToRegion (program, params, region, buffer) {
-    this._awsConfigUpdate(program, region)
+    aws.updateConfig(program, region)
 
     console.log('=> Reading event source file to memory')
     const eventSourceList = this._eventSourceList(program)
 
     return Promise.resolve().then(() => {
       if (this._isUseS3(program)) {
-        const s3Deploy = new S3Deploy(aws, region)
+        const s3Deploy = new S3Deploy(aws.sdk, region)
         return s3Deploy.putPackage(params, region, buffer)
       }
       return null
@@ -880,13 +848,13 @@ they may not work as expected in the Lambda environment.
       }
       console.log(params)
 
-      const lambda = new aws.Lambda({
+      const lambda = new aws.sdk.Lambda({
         region: region,
         apiVersion: '2015-03-31'
       })
-      const scheduleEvents = new ScheduleEvents(aws, region)
-      const s3Events = new S3Events(aws, region)
-      const cloudWatchLogs = new CloudWatchLogs(aws, region)
+      const scheduleEvents = new ScheduleEvents(aws.sdk, region)
+      const s3Events = new S3Events(aws.sdk, region)
+      const cloudWatchLogs = new CloudWatchLogs(aws.sdk, region)
 
       // Checking function
       return lambda.getFunction({