serverless/lib/actions/CodeUploadLambdaNodeJs.js

'use strict';

/**
 * Action: Code Upload: Lambda: Nodejs
 * - Uploads a single Lambda's code to their Serverless region bucket
 * - Don't attach "evt" to context, it will be overwritten in concurrent operations
 */

const SPlugin    = require('../ServerlessPlugin'),
    SError       = require('../ServerlessError'),
    SUtils       = require('../utils/index'),
    BbPromise    = require('bluebird'),
    path         = require('path'),
    fs           = require('fs'),
    os           = require('os');

// Promisify fs module
BbPromise.promisifyAll(fs);

class CodeUploadLambdaNodejs extends SPlugin {

  /**
   * Constructor
   */

  constructor(S, config) {
    super(S, config);
  }

  /**
   * Get Name
   */

  static getName() {
    return 'serverless.core.' + CodeUploadLambdaNodejs.name;
  }

  /**
   * Register Plugin Actions
   */

  registerActions() {

    this.S.addAction(this.codeUploadLambdaNodejs.bind(this), {
      handler:       'codeUploadLambdaNodejs',
      description:   'Uploads a Lambda\'s code to S3'
    });

    return BbPromise.resolve();
  }

  /**
   * Function Deploy
   */

  codeUploadLambdaNodejs(evt) {

    let _this = this;

    // Load AWS Service Instances
    let awsConfig = {
      region:          evt.region.region,
      accessKeyId:     _this.S._awsAdminKeyId,
      secretAccessKey: _this.S._awsAdminSecretKey,
    };
    _this.S3              = require('../utils/aws/S3')(awsConfig);
    _this.AwsMisc         = require('../utils/aws/Misc');

    // Flow
    return _this._validateAndPrepare(evt)
        .bind(_this)
        .then(_this._upload)
        .then(function() {
          return evt;
        })
        .catch(function(e) {
          console.log(e.stack)
        });
  }

  /**
   * Validate And Prepare
   */

  _validateAndPrepare(evt) {
    return BbPromise.resolve(evt);
  }

  /**
   * Upload
   * - Upload zip file to S3
   */

  _upload(evt) {

    let _this = this;

    SUtils.sDebug(`Uploading ${evt.function.name} to ${evt.region.regionBucket}`);

    return _this.S3.sPutLambdaZip(
        evt.region.regionBucket,
        _this.S._projectJson.name,
        evt.stage,
        evt.function.name,
        fs.createReadStream(evt.function.pathCompressed))
        .then(function(s3Key) {

          // Add S3Key and Bucket to Lambda.Function.Properties.Code
          for (let r in evt.function.cloudFormation.lambda) {

            if (evt.function.cloudFormation.lambda[r].Type === 'AWS::Lambda::Function') {
              evt.function.cloudFormation.lambda[r].Properties.Code = {
                S3Bucket: evt.region.regionBucket,
                S3Key: s3Key
              }
            }
          }

          return BbPromise.resolve(evt);
        });
  }
}

module.exports = CodeUploadLambdaNodejs;