serverless/lib/utils/aws/Misc.js

'use strict';

/**
 * Serverless Services: AWS: Misc
 * - Prefix custom methods with "s"
 */

let path      = require('path'),
    os        = require('os'),
    AWS       = require('aws-sdk'),
    SError    = require('../../Error'),
    SUtils    = require('../../utils'),
    SCli      = require('../../utils/cli'),
    async     = require('async'),
    dotenv    = require('dotenv'),
    fs        = require('fs');

module.exports.validRegions = [
  'us-east-1',
  'us-west-2',      // Oregon
  'eu-west-1',      // Ireland
  'ap-northeast-1'  // Tokyo
];

/**
 * Get ENV Files
 * - Get env files for a region or all regions, for a given stage
 * - Region param can be "all"
 */

module.exports.getEnvFiles = function(serverless, region, stage) {

  let _this      = this,
      regionGets = [];

  if (region != 'all' || stage == 'local') {  //single region
    if (stage == 'local') {
      region = 'local';
    }

    regionGets.push(_this.getEnvFileAsMap(serverless, region, stage)
        .then(envVars => {
          return {region: region, vars: envVars.map, raw: envVars.raw};
        }));

  } else {
    // All regions
    if (!serverless.state.meta.get().stages[stage]) {
      return Promise.reject(new SError(`Invalid stage ${stage}`, SError.errorCodes.UNKNOWN));
    }
    Object.keys(serverless.state.meta.get().stages[stage].regions).forEach(region => {
      regionGets.push(
          _this.getEnvFileAsMap(serverless, region, stage)
              .then(envVars => {
                return {region: region, vars: envVars.map, raw: envVars.raw};
              }));
    });
  }

  return Promise.all(regionGets);
};

/**
 * Get Env File As Map
 */

module.exports.getEnvFileAsMap = function(Serverless, region, stage) {

  let deferred;

  if (stage == 'local') {
    deferred = Promise.resolve(fs.readFileSync(Serverless.getProject().getFilePath( '.env' )));
  } else {
    let projectName  = Serverless.state.meta.get().variables.project,
        bucketName   = Serverless.state.meta.get().variables.projectBucket,
        bucketRegion = SUtils.getProjectBucketRegion(Serverless.state.meta.get().variables);

    let awsConfig = {
      region:          bucketRegion,
      accessKeyId:     Serverless.config.awsAdminKeyId,
      secretAccessKey: Serverless.config.awsAdminSecretKey
    };

    let S3 = require('./S3')(awsConfig);

    SCli.log(`Getting ENV file from S3 bucket: ${bucketName}`);

    deferred = S3.sGetEnvFile(bucketName, projectName, stage, region)
        .then(function(s3ObjData) {
          return (!s3ObjData.Body) ? '' : s3ObjData.Body;
        });
  }

  return deferred
      .then(function(envFileBuffer) {
        return {raw: envFileBuffer, map: dotenv.parse(envFileBuffer)};
      })
      .catch(function(err) {
        console.error(`Warning: trouble getting env for stage: ${stage} region: ${region}`, err);
        return {};
      });
};