aws-lambda/lib/main.js

"use strict";

var 
aws = require( "aws-sdk" ),
exec = require( "child_process" ).exec,
fs = require( "fs" ),
os = require( "os" ),
packageJson = require( "./../package.json" ),
path = require( "path" );

var Lambda = function( ) { 
  this.version = packageJson.version;

  return this;
};

Lambda.prototype._createEnvFile = function( ) {
  var 
  envfile = process.cwd() + "/.env",
  envexamplefile = __dirname + "/.env.example";

  if ( !fs.existsSync( envfile ) ) {
    console.log( ".env file successfully created" )
    fs.writeFileSync( envfile, fs.readFileSync( envexamplefile) );
  }

};

Lambda.prototype._createEventJsonFile = function( ) {
  var 
  eventfile = process.cwd() + "/event.json",
  eventexamplefile = __dirname + "/event.json.example";

  if ( !fs.existsSync( eventfile ) ) {
    console.log( "event.json file successfully created" )
    fs.writeFileSync( eventfile, fs.readFileSync( eventexamplefile ) );
  }

};

Lambda.prototype.setup = function( ) {
  console.log( "Running setup." );
  this._createEnvFile( );
  this._createEventJsonFile( );
  console.log( "Setup done. Edit the .env and event.json files as needed." );
};

Lambda.prototype.run = function( program ) {
  this._createEventJsonFile( );

  var 
  dir = program.directory,
  splitHandler = program.handler.split( "." ),
  filename = splitHandler[0] + ".js",
  handlername = splitHandler[1];

  var 
  handler = require( process.cwd() + "/" + filename )[handlername],
  event = require( process.cwd() + "/event.json" );

  this._runHandler( handler, event );
}

Lambda.prototype._runHandler = function( handler, event ) {
  var context = {
    done: function( ) {
      process.exit(0);
    }
  }

  handler( event, context );
};

Lambda.prototype._zipfileTmpPath = function( program ) {
  var
  ms_since_epoch = +new Date,
  filename = program.functionName + "-" + ms_since_epoch + ".zip",
  zipfile = path.join( os.tmpDir(), filename );

  return zipfile;
};

Lambda.prototype._params = function( program, buffer ) { 
  var params = {               
    FunctionName: program.functionName + "-" + program.environment,
    FunctionZip: buffer,       
    Handler: program.handler,  
    Mode: program.mode,        
    Role: program.role,
    Runtime: program.runtime,
    Description: program.description,
    MemorySize: program.memorySize, 
    Timeout: program.timeout
  };

  return params;
};

Lambda.prototype.deploy = function( program ) {
  this._createEnvFile( );

  aws.config.update({
    accessKeyId: program.accessKey,
    secretAccessKey: program.secretKey,
    region: program.region
  });

  var 
  _this = this,
  lambda = new aws.Lambda( { apiVersion: "2014-11-11" } ),
  zipfile = _this._zipfileTmpPath( program );

  // zip up but ignore .git, .dotfiles, .log files, and the event.json file.
  console.log( "Generating zip file" );
  exec( "zip -roq " + zipfile + " * -x '.git' -x '*/\.*' -x '*/\*.log' -x 'event.json'", function( err, stdout, stderr ) {
    if ( err ) {
      throw err;
    }

    console.log( "Reading zip file to memory" );
    var 
    buffer = fs.readFileSync( zipfile ),
    params = _this._params( program, buffer );

    console.log( "Uploading zip file to AWS Lambda with parameters:" );
    console.log( params );

    lambda.uploadFunction( params, function( err, data ) {
      if ( err ) {
        console.log( err );
      } else {
        console.log( "Zip file done uploading. Results follow:" );
        console.log( data );
      }
    } );

  } );
}

module.exports = new Lambda( );