add CFN custom resource files

Author: kylesomers

CloudFormation/CustomResources/APIGatewayCreateScript/WK305_Gateway.zip

@@ -0,0 +1,58 @@
+var AWS = require('aws-sdk');
+
+var apigateway;
+var apigatewayuuid;
+
+module.exports = {
+		deleteGateway:function(event, context, callback)
+	{
+		console.log('deleting gateway, parameters passed in: ');
+		console.log(event);
+		
+		region = event.ResourceProperties.region;
+		apigatewayuuid = event.StackId;
+		
+		AWS.config.update({region: region});
+		apigateway = new AWS.APIGateway();
+		
+		deleteGatewayImplementation(callback);
+	}
+}
+
+function deleteGatewayImplementation(callback)
+{
+	var params = {};
+	apigateway.getRestApis(params, function(err, data) {
+	  if (!err)
+	  {
+		  console.log('finding api gateway in list');
+		  for (var i = 0; i < data.items.length; i++) 
+		  {
+			  if(data.items[i].description == apigatewayuuid)
+			  {
+				  	console.log('found gateway, deleting...');
+				  	var params = {
+					  restApiId: data.items[i].id
+					};
+					apigateway.deleteRestApi(params, function(err, data) {
+					  if (!err)
+					  {
+						  console.log('successfully deleted api gateway');
+						  callback(null);
+						  
+					  }
+					  else
+					  {
+						  callback(err);
+					  }
+					}); 
+			  }
+		  }
+	  }
+	  else
+	  {
+		  console.log(err, err.stack); 
+		  callback(err);
+	  }
+	});
+}                

CloudFormation/CustomResources/APIGatewayCreateScript/src/ApiGatewayCreate.js

@@ -0,0 +1,85 @@
+/**
+ * New node file
+ */
+
+var AWS = require('aws-sdk');
+var apigatewaycreate = require('./ApiGatewayCreate');
+var apigatewaydelete = require('./ApiGatewayDelete');
+var cloudformationsender = require('./cloudformationsender');
+var zombieconfigupdater = require('./ZombieWebsiteConfigUpdate');
+
+var theRegion;
+var theS3Bucket;
+var theS3Key
+var theEvent;
+var theContext;
+
+exports.handleGatewayEvent = function(event, context)
+{
+
+	theEvent = event;
+	theContext = context;
+	
+	if(event.RequestType == 'Create')
+	{
+		theRegion = event.ResourceProperties.region;
+		theS3Bucket = event.ResourceProperties.s3bucket;
+		theS3Key = event.ResourceProperties.s3key;
+		
+		AWS.config.update({region: theRegion});
+		
+		apigatewaycreate.createGateway(event, context, createApiGatewayCallback);
+	}
+	else if(event.RequestType == 'Delete')
+	{
+		apigatewaydelete.deleteGateway(event, context, finishedUpdatingCallback);
+	}
+	else
+	{
+		console.log(event);
+		console.log('non-create requst type, sending suceed');
+		cloudformationsender.sendResponse(event, context, "SUCCESS", {});
+	}
+}
+
+function createApiGatewayCallback(err, restAPIId)
+{
+	if(err)
+	{
+		console.log(err, err.stack);
+		cloudformationsender.sendResponse(theEvent, 
+	            theContext, "FAILED", {});
+	}
+	else
+	{
+		var url = "https://" + restAPIId + ".execute-api." + theRegion + 
+		".amazonaws.com/ZombieWorkshopStage/zombie/message";
+		
+		console.log("calling s3 helper to create constants with url: " + url);
+		
+		zombieconfigupdater.updateConfig(theRegion,
+										 theS3Key,
+										 theS3Bucket,
+										 url,
+										 finishedUpdatingCallback);
+	}
+}
+
+/**
+ * callback that looks if it's an error and updates cloudformation appropriately...
+ * @param err
+ */
+function finishedUpdatingCallback(err)
+{
+
+	if(err)
+	{
+		cloudformationsender.sendResponse(theEvent, 
+	            theContext, "FAILED", {});
+	}
+	else
+	{	
+		cloudformationsender.sendResponse(theEvent, 
+	            theContext, "SUCCESS", {});
+	}
+}

CloudFormation/CustomResources/APIGatewayCreateScript/src/ApiGatewayDelete.js

@@ -0,0 +1,26 @@
+/**
+ * This module writes the file out in the format needed for the website.
+ * the S3 bucket/key and the url is passed in, and it writes the javascript file out.
+ */
+var AWS = require('aws-sdk');
+module.exports = {
+		updateConfig:function(region, s3key, s3bucketName, url, callback)
+	{
+		console.log('updating Zombie S3 web page located at: ' + s3key + ' and ' + s3bucketName);
+		
+		AWS.config.update({region: region});
+		
+		var s3bucket = new AWS.S3({params: {Bucket: s3bucketName}});
+		s3bucket.createBucket(function() {
+		  var params = {Key: s3key, Body: 'var MESSAGES_ENDPOINT = "' + url + '";'};
+		  s3bucket.upload(params, function(err, data) {
+		    if (err) {
+		      callback(err);
+		    } else {
+		      callback(null);
+		    }
+		  });
+		});
+	}
+}
+

CloudFormation/CustomResources/APIGatewayCreateScript/src/LambdaHandler.js

@@ -0,0 +1,54 @@
+/**
+ * New node file
+ */
+
+module.exports = {
+		sendResponse: function(event, context, responseStatus, responseData) {
+ 
+	var responseBody = JSON.stringify({
+	    Status: responseStatus,
+	    Reason: "See the details in CloudWatch Log Stream: " + context.logStreamName,
+	        PhysicalResourceId: context.logStreamName,
+	        StackId: event.StackId,
+	        RequestId: event.RequestId,
+	        LogicalResourceId: event.LogicalResourceId,
+	        Data: responseData
+	    });
+	 
+	    console.log("RESPONSE BODY:\n", responseBody);
+	 
+	    var https = require("https");
+	var url = require("url");
+	 
+	    var parsedUrl = url.parse(event.ResponseURL);
+	    var options = {
+	        hostname: parsedUrl.hostname,
+	        port: 443,
+	        path: parsedUrl.path,
+	        method: "PUT",
+	    headers: {
+	        "content-type": "",
+	        "content-length": responseBody.length
+	        }
+	    };
+	 
+	    console.log("SENDING RESPONSE...\n");
+	 
+	    var request = https.request(options, function(response) {
+	        console.log("STATUS: " + response.statusCode);
+	    console.log("HEADERS: " + JSON.stringify(response.headers));
+	    // Tell AWS Lambda that the function execution is done  
+	        context.done();
+	    });
+	 
+	    request.on("error", function(error) {
+	    console.log("sendResponse Error:" + error);
+	    // Tell AWS Lambda that the function execution is done  
+	        context.done();
+	    });
+	  
+	    // write data to request body
+	    request.write(responseBody);
+	    request.end();
+	}
+}