lita-spendo 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0dee1763aed5676a874615dbdaa9ae9609d0b728
-  data.tar.gz: 4b237c902a4fdca531c2248fc3cf1b92b371e291
+  metadata.gz: 529456b3bdf3fc77c3beaa9b53fa8637414df145
+  data.tar.gz: 43805a4f1e72389fb1cfd734200f98cb25611c77
 SHA512:
-  metadata.gz: 9eb5d1886ec72288ff5587c6448e68309516e95e2234b39c549cd8f5619196144a452b7b43e63b47425c026fb5203646ce34c86ede91eec5fd6cf1c5dffc68f6
-  data.tar.gz: 22855627345554166f97e8cc76e0e10d344af3509abb31b909c96c01199a6a4ddfcc4a50675d99345952f153963dbbd03e20e4551fe63771ebd979166429f169
+  metadata.gz: 3541b6ba32df59527cdaa955216f298b50be57dadef8661169fb56e06be5de7aad2da7a2113a236b6588d63e2842c734d6d47f86168919f11ca80e74a5b51cfc
+  data.tar.gz: a93c84f6d65caf9bab6815b953454f4900074d6792ae0685ee6946919f47b411b37c90eebd5bd5e7c49140e18e7e8b618da6a5842dfcc70aaf479003ef0e99d9

data/lib/lita/handlers/spendo.rb

@@ -1,4 +1,5 @@
 require 'aws-sdk'
+require 'json'
 
 module Lita
   module Handlers
@@ -26,15 +27,51 @@ module Lita
 
     class Spendo < Handler
 
+      LAST_RECORD_KEY = 'last_record'
+
       config :aws_account_id, type: String
       config :dynamodb_table, type: String, default: 'BillingHistory'
       config :base_image_url, type: String
+      config :room,           type: String
+      config :time_between_polls, type: Integer, default: 60*60
 
       route(/^spendo$/, :show, command: true, help: {
         "spendo" => "show current billing level"
       })
 
+      on(:connected) do |payload|
+        if config.room
+          # join the alert room
+          robot.join config.room
+        end
+
+        # set up a timer to poll DynamoDB
+        every(config.time_between_polls) do |timer|
+          check_for_alerts
+        end
+      end
+
       def show(response)
+        message, url = create_message
+
+        response.reply message
+        response.reply url
+      end
+
+      attr_writer :billing_history
+
+      def billing_history
+        if @billing_history.nil?
+          params = {
+            aws_account_id: config.aws_account_id,
+            dynamodb_table: config.dynamodb_table
+          }
+          @billing_history = BillingHistory.new(params)
+        end
+        @billing_history
+      end
+
+      def create_message
         data = billing_history.latest
 
         account          = data['Account']
@@ -58,21 +95,42 @@ module Lita
 
         url = config.base_image_url + "/#{alert_level}.jpg"
 
-        response.reply message
-        response.reply url
+        return [message, url]
       end
 
-      attr_writer :billing_history
+      # write current data to redis
+      # BigDecimals get saved as strings which fail to
+      # deserialize as integers (but do deserialize as floats)
+      def save_billing_data(data)
+        redis.set LAST_RECORD_KEY, data.to_json
+      end
 
-      def billing_history
-        if @billing_history.nil?
-          params = {
-            aws_account_id: config.aws_account_id,
-            dynamodb_table: config.dynamodb_table
-          }
-          @billing_history = BillingHistory.new(params)
+      # read previous data from redis
+      def load_billing_data
+        data = redis.get LAST_RECORD_KEY
+        return nil if data.nil?
+
+        JSON.parse(data)
+      end
+
+      def alert_level_changed?(previous, current)
+        previous['AlertLevel'].to_f != current['AlertLevel'].to_f
+      end
+
+      def check_for_alerts
+        log.debug "checking for alerts"
+        current_data = billing_history.latest
+        last_data    = load_billing_data
+
+        if last_data && alert_level_changed?(last_data, current_data)
+          message, url = create_message
+          target = Source.new(room: config.room)
+          robot.send_messages(target, message, url)
+        else
+          log.debug "alert level unchanged"
         end
-        @billing_history
+
+        save_billing_data current_data
       end
     end
 

data/lita-spendo.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-spendo"
-  spec.version       = "0.1.2"
+  spec.version       = "0.2.0"
   spec.authors       = ["Chris Chalfant"]
   spec.email         = ["cchalfant@leafsoftwaresolutions.com"]
   spec.description   = "Lita handler for displaying AWS billing"

data/spec/lita/handlers/spendo_spec.rb

@@ -31,6 +31,19 @@ describe Lita::Handlers::Spendo, lita_handler: true do
     it 'shows the url' do
       expect(replies).to include("http://foo.com/3.jpg")
     end
+  end
+
+  describe '#alert_level_changed?' do
+    it 'is true if the levels are different' do
+      prev = {'AlertLevel' => "1"}
+      curr = {'AlertLevel' => "2"}
+      expect(subject.alert_level_changed?(prev, curr)).to be_truthy
+    end
 
+    it 'is false if the levels are same' do
+      prev = {'AlertLevel' => "1"}
+      curr = {'AlertLevel' => "1"}
+      expect(subject.alert_level_changed?(prev, curr)).to be_falsey
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lita-spendo
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Chris Chalfant
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-30 00:00:00.000000000 Z
+date: 2015-05-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: lita