Merge pull request #506 from CodeNow/lambda-functions

adding script that is sending autoscaling events to slack

Author: anandkumarpatel

lambda/asg-events/index.js

@@ -0,0 +1,53 @@
+'use strict'
+const https = require('https')
+
+exports.handler = function (event, context) {
+  let message
+  try {
+    message = JSON.parse(event.Records[0].Sns.Message)
+  } catch (e) {
+    message = { Description: event.Records[0].Sns.Message }
+  }
+
+  const postData = {
+    channel: '#aws-notifications',
+    username: 'AWS',
+    text: '*' + event.Records[0].Sns.Subject + '*',
+    icon_emoji: ':cloud:'
+  }
+
+  let slackMessage = [
+    message.Description,
+    '',
+    message.Cause
+  ].join('\n')
+
+  if (!message.Description && !message.Cause) {
+    slackMessage = 'test notification :partyparrot:'
+  }
+
+  postData.attachments = [{
+    color: 'good',
+    text: slackMessage
+  }]
+
+  const options = {
+    method: 'POST',
+    hostname: 'hooks.slack.com',
+    port: 443,
+    path: '/services/T029DEC10/B141U5BHT/UaAwljeWydJaiW6RmD09C4Wx'
+  }
+
+  const req = https.request(options, (res) => {
+    res.setEncoding('utf8')
+    res.on('data', () => {})
+    res.on('end', () => { context.done(null) })
+  })
+
+  req.on('error', (e) => {
+    console.error('problem with request: ' + e.message)
+  })
+
+  req.write(JSON.stringify(postData))
+  req.end()
+}

lambda/asg-events/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "asg-event-handler",
+  "version": "1.0.0",
+  "description": "AWS Lambda Handler for ASG Events.",
+  "main": "index.js",
+  "directories": {
+    "test": "test"
+  },
+  "scripts": {
+    "test": "standard && $npm_package_options_testCommand",
+    "testonly": "$npm_package_options_testCommand"
+  },
+  "options": {
+    "testCommand": "ava --serial --fail-fast --timeout 5000 --verbose"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/CodeNow/devops-scripts.git"
+  },
+  "author": "Bryan Kendall <[email protected]>",
+  "license": "UNLICENSED",
+  "bugs": {
+    "url": "https://github.com/CodeNow/devops-scripts/issues"
+  },
+  "homepage": "https://github.com/CodeNow/devops-scripts#readme",
+  "devDependencies": {
+    "ava": "^0.15.2",
+    "sinon": "^1.17.4",
+    "standard": "^7.1.2"
+  }
+}

lambda/asg-events/test/fixtures/new-ec2-instances.json

@@ -0,0 +1,22 @@
+{
+  "Records": [
+    {
+      "EventSource": "aws:sns",
+      "EventVersion": "1.0",
+      "EventSubscriptionArn": "arn:aws:sns:us-west-2:437258487404:asg-events-production-delta:2467eda4-9382-441d-b00f-4baec59c583a",
+      "Sns": {
+        "Type": "Notification",
+        "MessageId": "302ece4d-4377-52fc-87da-81b353f6530d",
+        "TopicArn": "arn:aws:sns:us-west-2:437258487404:asg-events-production-delta",
+        "Subject": "Auto Scaling: launch for group \"asg-production-delta-2513635\"",
+        "Message": "{\"Progress\":50,\"AccountId\":\"437258487404\",\"Description\":\"Launching a new EC2 instance: i-8844d555\",\"RequestId\":\"9d2dc323-d5e1-42a3-9945-515767249fa7\",\"EndTime\":\"2016-07-11T21:23:13.840Z\",\"AutoScalingGroupARN\":\"arn:aws:autoscaling:us-west-2:437258487404:autoScalingGroup:a1b93c0d-4566-420a-a753-4cb9df982cff:autoScalingGroupName/asg-production-delta-2513635\",\"ActivityId\":\"9d2dc323-d5e1-42a3-9945-515767249fa7\",\"StartTime\":\"2016-07-11T21:22:07.785Z\",\"Service\":\"AWS Auto Scaling\",\"Time\":\"2016-07-11T21:23:13.840Z\",\"EC2InstanceId\":\"i-8844d555\",\"StatusCode\":\"InProgress\",\"StatusMessage\":\"\",\"Details\":{\"Subnet ID\":\"subnet-3343206a\",\"Availability Zone\":\"us-west-2c\",\"InvokingAlarms\":[{\"Trigger\":{\"MetricName\":\"Swarm Reserved Memory Maximum Available\",\"ComparisonOperator\":\"LessThanThreshold\",\"Statistic\":\"AVERAGE\",\"Dimensions\":[{\"name\":\"AutoScalingGroupName\",\"value\":\"asg-production-delta-2513635\"}],\"Period\":300,\"EvaluationPeriods\":1,\"Unit\":null,\"Namespace\":\"Runnable/Swarm\",\"Threshold\":2},\"AlarmName\":\"asg-production-delta-2513635-max-available-memory\",\"AlarmDescription\":null,\"AWSAccountId\":\"437258487404\",\"OldStateValue\":\"OK\",\"NewStateReason\":\"Threshold Crossed: 1 datapoint (1.7499999999999996) was less than the threshold (2.0).\",\"Region\":\"US West - Oregon\",\"NewStateValue\":\"ALARM\",\"StateChangeTime\":1468272105152}]},\"AutoScalingGroupName\":\"asg-production-delta-2513635\",\"Cause\":\"At 2016-07-11T21:21:45Z a monitor alarm asg-production-delta-2513635-max-available-memory in state ALARM triggered policy scale-out changing the desired capacity from 11 to 12.  At 2016-07-11T21:22:06Z an instance was started in response to a difference between desired and actual capacity, increasing the capacity from 11 to 12.\",\"Event\":\"autoscaling:EC2_INSTANCE_LAUNCH\"}",
+        "Timestamp": "2016-07-11T21:23:13.907Z",
+        "SignatureVersion": "1",
+        "Signature": "RBO7jTwwXwNHyW2x4O39FN2Drz7FwvbmfdbUsm+OFFjGXITAkwGms6OFMVYsaGClTXbeycfFcCCCjsSZPeY6o2jZSrfeteYPHUcLs/xSyz1FCaUJN99qxwGYqRVdP/O3wXi65NwRqOTHSK9Ptm9SjRGzCnJfwwLQIjQimmaVaUsVMIGbkVgLPlCJ59qALXXeEJ5zNVVrym3E4b/LsO8D+WEmLySsJQ/92UTBX7ZwQUb/aoEVnqFqgCrCaQQpxztyURjbjl1C4BkXAuC1bHqfJ3czXv/WbuwjLbhvUNK//eutXBaqN+k7aO/VU3wwWQ8Tv5TPQ8qvDaiAbnWUrgUVaA==",
+        "SigningCertUrl": "https://sns.us-west-2.amazonaws.com/SimpleNotificationService-bb750dd426d95ee9390147a5624348ee.pem",
+        "UnsubscribeUrl": "https://sns.us-west-2.amazonaws.com/?Action=Unsubscribe&SubscriptionArn=arn:aws:sns:us-west-2:437258487404:asg-events-production-delta:2467eda4-9382-441d-b00f-4baec59c583a",
+        "MessageAttributes": {}
+      }
+    }
+  ]
+}

lambda/asg-events/test/fixtures/test-notification.json

@@ -0,0 +1,22 @@
+{
+  "Records": [
+    {
+      "EventSource": "aws:sns",
+      "EventVersion": "1.0",
+      "EventSubscriptionArn": "arn:aws:sns:us-west-2:437258487404:asg-events-production-delta:2467eda4-9382-441d-b00f-4baec59c583a",
+      "Sns": {
+        "Type": "Notification",
+        "MessageId": "ab6d7f62-fd75-5025-bfe8-309cfbd0f8eb",
+        "TopicArn": "arn:aws:sns:us-west-2:437258487404:asg-events-production-delta",
+        "Subject": "Auto Scaling: test notification for group \"asg-production-delta-2685575\"",
+        "Message": "{\"AccountId\":\"437258487404\",\"RequestId\":\"cc86deb5-47a7-11e6-b529-354c7d92db8f\",\"AutoScalingGroupARN\":\"arn:aws:autoscaling:us-west-2:437258487404:autoScalingGroup:886e6b80-b398-4857-9fe8-16d45b429e2b:autoScalingGroupName/asg-production-delta-2685575\",\"AutoScalingGroupName\":\"asg-production-delta-2685575\",\"Service\":\"AWS Auto Scaling\",\"Event\":\"autoscaling:TEST_NOTIFICATION\",\"Time\":\"2016-07-11T20:41:09.815Z\"}",
+        "Timestamp": "2016-07-11T20:41:09.950Z",
+        "SignatureVersion": "1",
+        "Signature": "V/grO2ol3DgM17q87x+ltyp6C2U5EfEpTeeKI4iwCXsUTfVvARkuRRnAuHTcsAZ4cl2mXQEZXVfJIdXfM6jAYx7RVpQQ+Yn5FpuFR8aBcZmDZWCJxNMVGX/HubQXHI9M33/e2sLPZAD8Bok38o8PyuTS77Wfy//jJMtaJ7KsIToHNxrNERFv1zPiURQFQ3xUA9bo87aaU/8MXha91UWglsyJWxZRKdIc/v2dcRQnPHfOFxAbi5wThOHFoNHmrJupaimKwfjf/QVv9kEfVaz8/vQb6Uz/RxLZPxlgfeOTm8HckyizzZsdmYWANk2rD1xlVHZZAU6TNuXoL9by7yiYWQ==",
+        "SigningCertUrl": "https://sns.us-west-2.amazonaws.com/SimpleNotificationService-bb750dd426d95ee9390147a5624348ee.pem",
+        "UnsubscribeUrl": "https://sns.us-west-2.amazonaws.com/?Action=Unsubscribe&SubscriptionArn=arn:aws:sns:us-west-2:437258487404:asg-events-production-delta:2467eda4-9382-441d-b00f-4baec59c583a",
+        "MessageAttributes": {}
+      }
+    }
+  ]
+}

lambda/asg-events/test/index.js

@@ -0,0 +1,110 @@
+const https = require('https')
+const EventEmitter = require('events')
+
+class MockRequest extends EventEmitter {
+  constructor () {
+    super()
+    this._written_data = []
+  }
+
+  write (data) {
+    this._written_data.push(data)
+  }
+
+  end () {
+    this._ended = true
+  }
+}
+
+class MockResponse extends EventEmitter {
+  setEncoding (encoding) {
+    this.encoding = encoding
+  }
+}
+
+const test = require('ava')
+const sinon = require('sinon')
+const NEW_EC2_INSTANCE = require('./fixtures/new-ec2-instances.json')
+const TEST_NOTIFICATION = require('./fixtures/test-notification.json')
+
+const handler = require('../').handler
+
+test.before((t) => {
+  sinon.stub(https, 'request')
+})
+
+test.beforeEach((t) => {
+  https.request.reset()
+})
+
+test.cb('it should call done in the context', (t) => {
+  const request = new MockRequest()
+  const response = new MockResponse()
+  https.request.returns(request)
+  https.request.yieldsAsync(response)
+  sinon.stub(request, 'end', () => {
+    setTimeout(() => { response.emit('end') })
+  })
+  handler(NEW_EC2_INSTANCE, {
+    done: () => {
+      t.pass()
+      t.end()
+    }
+  })
+})
+
+test.cb('it should post to slack', (t) => {
+  const request = new MockRequest()
+  const response = new MockResponse()
+  https.request.returns(request)
+  https.request.yieldsAsync(response)
+  sinon.stub(request, 'end', () => {
+    setTimeout(() => { response.emit('end') })
+  })
+  handler(NEW_EC2_INSTANCE, { done: () => {
+    sinon.assert.calledOnce(https.request)
+    sinon.assert.calledWithExactly(
+      https.request,
+      {
+        method: 'POST',
+        hostname: 'hooks.slack.com',
+        port: 443,
+        path: sinon.match(/\/services\/[^\/]+\/[^\/]+\/[^\/]+/)
+      },
+      sinon.match.func
+    )
+    t.end()
+  } })
+})
+
+test.cb('it should send the description and cause', (t) => {
+  const request = new MockRequest()
+  const response = new MockResponse()
+  https.request.returns(request)
+  https.request.yieldsAsync(response)
+  sinon.stub(request, 'end', () => {
+    setTimeout(() => { response.emit('end') })
+  })
+  handler(NEW_EC2_INSTANCE, { done: () => {
+    t.is(request._written_data.length, 1, 'one write')
+    const s = JSON.parse(request._written_data.pop())
+    t.regex(s.attachments[0].text, /Launching a new EC2 instance/)
+    t.end()
+  } })
+})
+
+test.cb('it should send a test message', (t) => {
+  const request = new MockRequest()
+  const response = new MockResponse()
+  https.request.returns(request)
+  https.request.yieldsAsync(response)
+  sinon.stub(request, 'end', () => {
+    setTimeout(() => { response.emit('end') })
+  })
+  handler(TEST_NOTIFICATION, { done: () => {
+    t.is(request._written_data.length, 1, 'one write')
+    const s = JSON.parse(request._written_data.pop())
+    t.regex(s.attachments[0].text, /test notification :partyparrot:/)
+    t.end()
+  } })
+})