lita-aws-cloudwatch 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: bd3b6f952230d48f6f56dc7745bcb2a9f33c2687
+  data.tar.gz: 101e2b55f44ecbc6564251c02c7fb4c6950f0c1e
+SHA512:
+  metadata.gz: 81e6f1b04d7f80b68fb9158e909b6c6e6f4699e28006a60f9a2437417e5692338dc39392b25628f0de96ecd8bbefc12edfb1ac61748eefba3d6a436e3ce0cbfa
+  data.tar.gz: dc6728dbf2bb417220652346d33070b840b1183fbe3113705bd69b0e0a1e817ac6c516f019bb3951a008d03da2fb8474a5af23372b1cbcb2100dbb46f3fba241

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--colour
+--profile

data/.travis.yml

@@ -0,0 +1,5 @@
+language: ruby
+rvm:
+  - 2.0.0
+services:
+  - redis-server

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec

data/README.md

@@ -0,0 +1,73 @@
+# lita-aws-cloudwatch
+
+[![Build Status](https://travis-ci.org/5fpro/lita-aws-cloudwatch.svg?branch=master)](https://travis-ci.org/5fpro/lita-aws-cloudwatch)
+
+Receive AWS CloudWatch alarm from AWS SNS (Simple Notification Service), and messaging to room.
+
+# Features
+
+- Supporting multiple AWS accounts.
+- Notify to different room for each AWS acount.
+- Can reveice SNS confirmation data.
+- Show `AWS AccountID` while receiving SNS confirmation data.
+
+## Installation
+
+- Add lita-aws-cloudwatch to your Lita instance's Gemfile:
+``` ruby
+gem "lita-aws-cloudwatch"
+```
+
+- See <a href="#configuration">Configuration</a>.
+
+- Restart lita.
+
+(goto AWS web console...)
+
+- Create `Topic` in AWS SNS.
+
+- Create SNS `Subscription` from `Topic`
+  - choose Protocol to `HTTP`.
+  - set Endpoint to `http://123.123.123.123:8888/aws-cloudwatch/receive`
+  - You will receive confirmation link from lita notify (click the link to finish confirmation).
+
+- Set CloudWatch notification to topic.
+
+- Done :)
+
+## Configuration
+
+- You must enable lita `http routing` and `redis` in `lita_config.rb`
+```
+  config.redis['host'] = "127.0.0.1"
+  config.redis['port'] = 6379
+
+  config.http.port = 8888
+```
+
+- Default room name while account is not set yet.
+```
+  config.handlers.aws_cloudwatch.default_room = "general"
+```
+
+
+## Usage
+
+- list all aws accounts. Including account id, name, room.
+```
+aws account list
+```
+
+- `aws account set [account id] [account name]` : set account name.
+```
+aws account set 123123 5Fpro co. ltd.
+```
+
+- `aws account set [account id] [room name]` : set notify room for account. If you use Slack, it need to invite lita robot to room.
+```
+aws account set 123123 #server-state.
+```
+
+## License
+
+[MIT](http://opensource.org/licenses/MIT)

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/lib/lita/handlers/aws_cloudwatch.rb

@@ -0,0 +1,120 @@
+module Lita
+  module Handlers
+    class AwsCloudwatch < Handler
+      config :default_room, type: String, default: "general"
+
+      route(/^aws account set ([0-9]+) (.+)$/, :set_account, command: true, help: { "aws account set [account id] [account name]" => "Set AWS account name" })
+      def set_account(response)
+        account_id = response.matches[0][0]
+        account_name = response.matches[0][1]
+        redis.hset(redis_key, account_id, account_name)
+        response.reply("AWS account: #{account_id} has been set name: #{account_name}")
+      end
+
+      route(/^aws account room ([0-9]+) ([^\s]+)$/, :set_account_room, command: true, help: { "aws account room [account id] [room_name]" => "Set room for AWS account" })
+      def set_account_room(response)
+        account_id = response.matches[0][0]
+        room_name = response.matches[0][1].gsub("#", "")
+        redis.hset(redis_key_for_room, account_id, room_name)
+        response.reply("AWS account: #{account_id} has set room to: ##{room_name} , please invite robot to the room.")
+      end
+
+      route(/^aws account list/, :list_accounts, command: true, help: { "aws account list" => "List all AWS accounts" })
+      def list_accounts(response)
+        message = accounts.keys.map{ |key| "#{key}: #{accounts[key]} ( ##{fetch_room(key)} )" }.join("\n")
+        response.reply(message)
+      end
+
+      http.post "/aws-cloudwatch/receive", :receive
+      def receive(request, response)
+        @request = request
+        case params_data["Type"]
+        when "SubscriptionConfirmation"
+          alert_confirmation!
+        when "Notification"
+          alert_notification!
+        else
+          alert_unknowtype!
+        end
+        response.write("ok")
+      end
+
+      private
+
+      def alert_confirmation!
+        message = params_data["Message"].gsub("SubscribeURL", params_data["SubscribeURL"])
+        account_id = params_data["TopicArn"].split(":")[-2]
+        room = fetch_room(account_id)
+        send_message_to_room(message, room)
+        unless accounts[account_id]
+          send_message_to_room("Your AWS Account ID is `#{account_id}`, type `aws account set #{account_id} [account name]` for sett name, and type `aws account room #{account_id} [room name]` to set reporting room", room)
+        end
+      end
+
+      def alert_unknowtype!
+        room = config.default_room
+        message = "unknow type: #{params_data["Type"].inspect} , send to robot.\n DEBUG: #{params_data.inspect}"
+        send_message_to_room(message, room)
+      end
+
+      def alert_notification!
+        data = JSON.parse(params_data["Message"])
+        account_id = data["AWSAccountId"]
+        name = data["AlarmName"]
+        state = data["NewStateValue"]
+        reason = data["NewStateReason"]
+        messages = []
+        messages << "Account: #{accounts[account_id] || account_id}"
+        messages << "State: #{state}"
+        messages << "Name: #{name}"
+        messages << "Reason: #{reason}"
+        room = fetch_room(account_id)
+        send_message_to_room(messages.join("\n"), room)
+      end
+
+      def redis_key_for_room
+        "aws-cloudwatch-accounts-room"
+      end
+
+      def redis_key
+        "aws-cloudwatch"
+      end
+
+      def accounts
+        @accounts ||= redis.hgetall(redis_key)
+        @accounts
+      end
+
+      def fetch_room(account_id)
+        @rooms ||= redis.hgetall(redis_key_for_room)
+        @rooms[account_id] || config.default_room
+      end
+
+      def params_data
+        @data ||= JSON.parse(@request.body.string)
+        @data
+      end
+
+      def send_message_to_room(message, room_name = nil)
+        room_name ||= config.default_room
+        target = Source.new(room: find_room_id_by_name(room_name))
+        robot.send_messages(target, message)
+      end
+
+      def find_room_id_by_name(room_name)
+        case robot.config.robot.adapter.to_s.to_sym
+        when :slack
+          if room = ::Lita::Room.find_by_name(room_name)
+            return room.id
+          else
+            ::Lita::Room.find_by_name("general").id
+          end
+        else
+          room_name
+        end
+      end
+
+      Lita.register_handler(self)
+    end
+  end
+end

data/lib/lita-aws-cloudwatch.rb

@@ -0,0 +1,12 @@
+require "lita"
+
+Lita.load_locales Dir[File.expand_path(
+  File.join("..", "..", "locales", "*.yml"), __FILE__
+)]
+
+require "lita/handlers/aws_cloudwatch"
+
+Lita::Handlers::AwsCloudwatch.template_root File.expand_path(
+  File.join("..", "..", "templates"),
+ __FILE__
+)

data/lita-aws-cloudwatch.gemspec

@@ -0,0 +1,23 @@
+Gem::Specification.new do |spec|
+  spec.name          = "lita-aws-cloudwatch"
+  spec.version       = "0.1.1"
+  spec.authors       = ["marsz"]
+  spec.email         = ["marsz330@gmail.com"]
+  spec.description   = "Receive AWS CloudWatch alarm from AWS SNS (Simple Notification Service), and messaging to room."
+  spec.summary       = "AWS CloudWatch integration with AWS SNS"
+  spec.homepage      = "https://github.com/5fpro/lita-aws-cloudwatch"
+  spec.license       = "MIT"
+  spec.metadata      = { "lita_plugin_type" => "handler" }
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "lita", ">= 4"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rack-test"
+  spec.add_development_dependency "rspec", ">= 3.0.0"
+end

data/locales/en.yml

@@ -0,0 +1,4 @@
+en:
+  lita:
+    handlers:
+      aws_cloudwatch:

data/spec/lita/handlers/aws_cloudwatch_spec.rb

@@ -0,0 +1,86 @@
+require "spec_helper"
+
+describe Lita::Handlers::AwsCloudwatch, lita_handler: true do
+  it "#set_account" do
+    send_command "aws account set 1234 5Fpro HaHa"
+    expect(replies.size).to be > 0
+    send_command "aws account list"
+    expect(replies.last).to match("1234: 5Fpro HaHa")
+  end
+
+  it "#set_account_room" do
+    send_command "aws account set 1234 5Fpro HaHa"
+    send_command "aws account room 1234 #haha"
+    expect(replies.size).to be > 0
+    send_command "aws account list"
+    expect(replies.last).to match("1234: 5Fpro HaHa ( #haha )")
+  end
+
+  context "#receive" do
+    let(:account_id){ "863886017929" }
+    let(:body_confirmation){ '{"Type": "SubscriptionConfirmation", "MessageId": "cb304f8d-740d-4810-b9dd-7d47a1184eb8", "Token": "2336412f37fb687f5d51e6e241d7700bdc2fd74df0fbef7c6648870dee1c4c87ab59fe66548b509ff18165e81a2a4c6b4b5803444bf84a10365c1104202a9d010019385e91a7c871bf84959b919966358f3e2ced05bd9bc1d08a86ac8bae46a1420da82c2b308a395058cb4cad8ba00cdf5efada5812e150171241d95acd5153", "TopicArn": "arn:aws:sns:ap-northeast-1:863886017929:developers", "Message": "You have chosen to subscribe to the topic arn:aws:sns:ap-northeast-1:863886017929:developers.\nTo confirm the subscription, visit the SubscribeURL included in this message.", "SubscribeURL": "https://sns.ap-northeast-1.amazonaws.com/?Action=ConfirmSubscription&TopicArn=arn:aws:sns:ap-northeast-1:863886017929:developers&Token=2336412f37fb687f5d51e6e241d7700bdc2fd74df0fbef7c6648870dee1c4c87ab59fe66548b509ff18165e81a2a4c6b4b5803444bf84a10365c1104202a9d010019385e91a7c871bf84959b919966358f3e2ced05bd9bc1d08a86ac8bae46a1420da82c2b308a395058cb4cad8ba00cdf5efada5812e150171241d95acd5153", "Timestamp": "2015-12-06T07:13:41.370Z", "SignatureVersion": "1", "Signature": "bkvpfe9br6g41XQRJOgDg99C2evpXTGsXJ8lbLfZbmt4rfloORWO5Kw0Dh7QKIijd9y/QyvU6Yo80k8DXVWfJfj/vLysjplHOsrxSmvR1F7VMZil1cyoag3c/LNHJIq0WRSJxH8O6HkT1c5SNCSWfAiqz+4tw/brtCl8gK9vmGr0eGPOcFhT46xrz0Ucc+clFlwuhAjrRoK9vRRsIyTl4NEPgtC7HyRlB+Qm+ooXi7Bk0AUnGMdTKPYB/c1bCq+A0VKCzxLKjSzIme0i59n3lxFh4zSSygbX68gay7A2FRxYfl2j6/agH5gC0jsnQh1Xgry7Y/+z0fdUwzAZoWw5Ng==", "SigningCertURL": "https://sns.ap-northeast-1.amazonaws.com/SimpleNotificationService-bb750dd426d95ee9390147a5624348ee.pem"}' }
+    let(:body_notification){ '{"Type": "Notification", "MessageId": "f1cd789a-df69-5dc4-bd54-36419e0f72c4", "TopicArn": "arn:aws:sns:ap-northeast-1:863886017929:developers", "Subject": "ALARM: \"api-1 - Network In\" in APAC - Tokyo", "Message": "{\"AlarmName\":\"api-1 - Network In\",\"AlarmDescription\":null,\"AWSAccountId\":\"863886017929\",\"NewStateValue\":\"ALARM\",\"NewStateReason\":\"Threshold Crossed: 1 datapoint (2024818.75) was greater than or equal to the threshold (2000000.0).\",\"StateChangeTime\":\"2015-12-04T07:22:14.394+0000\",\"Region\":\"APAC - Tokyo\",\"OldStateValue\":\"OK\",\"Trigger\":{\"MetricName\":\"NetworkIn\",\"Namespace\":\"AWS/EC2\",\"Statistic\":\"AVERAGE\",\"Unit\":null,\"Dimensions\":[{\"name\":\"InstanceId\",\"value\":\"i-bc8dcaa5\"}],\"Period\":300,\"EvaluationPeriods\":1,\"ComparisonOperator\":\"GreaterThanOrEqualToThreshold\",\"Threshold\":2000000.0}}", "Timestamp": "2015-12-04T07:22:14.461Z", "SignatureVersion": "1", "Signature": "cW8eyf5k/+QaVZna/nhMh8HSnN7Y0BFR1fOvyYTG9seBsW4f6tGMj/wVK9aQVqVvEI3Vt54+OM0RSiiJHGOaanej2IxH+DiVo7k7VnN9uK25ejYbcp0pXusz7z7+VfzDr26QRCGF1qmq+TTx61q4PmfTIdigPZ9H9wUzQdujLxJy7vjRjJTLnhJ6fQtGCOyTfl0FTpYbVPeRT8/vu6G6XYw53hAa+sRbGiONmirwF0OzLQaW7M0FRO/Y/W8Yyr8GTNcT9egSIRED79fmkzWz2oxmEjVCJBUtvO6BBmjckCSsFB7oSDjuOKhGv5TNP29qSE36pwox0AoOnn6SoTnAlw==", "SigningCertURL": "https://sns.ap-northeast-1.amazonaws.com/SimpleNotificationService-bb750dd426d95ee9390147a5624348ee.pem", "UnsubscribeURL": "https://sns.ap-northeast-1.amazonaws.com/?Action=Unsubscribe&SubscriptionArn=arn:aws:sns:ap-northeast-1:863886017929:developers:dc2735b1-b160-4ec3-901e-393fd971a28a"}' }
+    context "no account name, no room" do
+      it "confirmation" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_confirmation
+        end
+        expect( response.body ).to match("ok")
+      end
+      it "notification" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_notification
+        end
+        expect( response.body ).to match("ok")
+      end
+    end
+    context "only account id" do
+      before{ send_command "aws account set #{account_id} 5Fpro" }
+      it "confirmation" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_confirmation
+        end
+        expect( response.body ).to match("ok")
+      end
+      it "notification" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_notification
+        end
+        expect( response.body ).to match("ok")
+      end
+    end
+
+    context "only room" do
+      before{ send_command "aws account room #{account_id} haha" }
+      it "confirmation" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_confirmation
+        end
+        expect( response.body ).to match("ok")
+      end
+      it "notification" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_notification
+        end
+        expect( response.body ).to match("ok")
+      end
+    end
+
+    context "both room and account_id" do
+      before{ send_command "aws account room #{account_id} haha" }
+      before{ send_command "aws account set #{account_id} 5Fpro" }
+      it "confirmation" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_confirmation
+        end
+        expect( response.body ).to match("ok")
+      end
+      it "notification" do
+        response = http.post("/aws-cloudwatch/receive") do |req|
+          req.body = body_notification
+        end
+        expect( response.body ).to match("ok")
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'json'
+require "lita-aws-cloudwatch"
+require "lita/rspec"
+
+# A compatibility mode is provided for older plugins upgrading from Lita 3. Since this plugin
+# was generated with Lita 4, the compatibility mode should be left disabled.
+Lita.version_3_compatibility_mode = false

metadata

@@ -0,0 +1,131 @@
+--- !ruby/object:Gem::Specification
+name: lita-aws-cloudwatch
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- marsz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-12-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: lita
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+description: Receive AWS CloudWatch alarm from AWS SNS (Simple Notification Service),
+  and messaging to room.
+email:
+- marsz330@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- README.md
+- Rakefile
+- lib/lita-aws-cloudwatch.rb
+- lib/lita/handlers/aws_cloudwatch.rb
+- lita-aws-cloudwatch.gemspec
+- locales/en.yml
+- spec/lita/handlers/aws_cloudwatch_spec.rb
+- spec/spec_helper.rb
+- templates/.gitkeep
+homepage: https://github.com/5fpro/lita-aws-cloudwatch
+licenses:
+- MIT
+metadata:
+  lita_plugin_type: handler
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.6
+signing_key: 
+specification_version: 4
+summary: AWS CloudWatch integration with AWS SNS
+test_files:
+- spec/lita/handlers/aws_cloudwatch_spec.rb
+- spec/spec_helper.rb