ruboty-kanjo 0.1.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 94f52aa724a6748935c8a8864eee93b941c1c03d
-  data.tar.gz: b12af63a2675dec4f218b269ba97ef0d48039cec
+  metadata.gz: 5599f410097e6f482e82e73e5f37d29631e0810f
+  data.tar.gz: 8802c5c4153d3144f92559ba1fbf5f4549a19ff8
 SHA512:
-  metadata.gz: 23ed9e548c954598e1a399e8153a547d913a66d64f1a6bc4f741a2423b179be867709f3bc715b2e4f08951718c47766c25176854c87caee2214d7409d3158bd7
-  data.tar.gz: 68ae94550d2a88965525303939970f8e52f88c5950881e56cd2f4f9a2f5fdde3e1b58e65f9dd6e3e5bf7584627b91d5a513b1d6d608d5396e73dddcfe03be039
+  metadata.gz: 88551c8c9e2f167e58eab58f9fdc9e248d11190673790202c374dc367bf1a3bc6413df24d964349cf7622d168b54d7bfc1e705f9a64350af5c4619e315e9947d
+  data.tar.gz: cb2136b734b54bd18a08bfea0a6a15b60539a5fb8e1bd938fa863adf9f40e3a74729022038f680f0680e4bca8f431437d803ff1a1b12d6477a2cc69c7b52c6d0

data/README.md

@@ -17,7 +17,8 @@ And then execute:
 ## Usage
 
 ```
-ruboty kanjo aws - fetch pricing by AWS
+ruboty /kanjo aws billing\z/ - fetch aws billing
+ruboty /kanjo aws spot (?<availability_zone>[a-z0-9-]+) (?<instances>[a-z0-9.]+(?:\s*,\s*[a-z0-9.]+)*)\z/ - fetch aws spot instacne prices
 ```
 
 ## Development

data/lib/ruboty/handlers/kanjo.rb

@@ -3,10 +3,17 @@ require 'ruboty/kanjo/actions/aws'
 
 module Ruboty::Handlers
   class Kanjo < Ruboty::Handlers::Base
-    on /kanjo aws\z/, name: :aws, description: 'fetch aws billing'
+    on /kanjo aws billing\z/,
+      name: :aws_billing, description: 'fetch aws billing'
+    on /kanjo aws spot (?<availability_zone>[a-z0-9-]+) (?<instances>[a-z0-9.]+(?:\s*,\s*[a-z0-9.]+)*)\z/,
+      name: :aws_spot_instance, description: 'fetch aws spot instacne prices'
 
-    def aws(message)
-      Ruboty::Kanjo::Actions::AWS.new(message).call
+    def aws_billing(message)
+      Ruboty::Kanjo::Actions::AWS.new(message).billing
+    end
+
+    def aws_spot_instance(message)
+      Ruboty::Kanjo::Actions::AWS.new(message).spot_instance
     end
   end
 end

data/lib/ruboty/kanjo/actions/aws.rb

@@ -2,29 +2,84 @@ require 'aws-sdk'
 
 module Ruboty::Kanjo::Actions
   class AWS < Ruboty::Actions::Base
-    ONE_HOUR = 60 * 60
-
-    def call
+    ONE_HOUR = 60 * 60 * 6
+
+    INSTANCE_TYPES = %w(
+      t1.micro t2.nano t2.micro t2.small t2.medium t2.large
+      m1.small m1.medium m1.large m1.xlarge
+      m3.medium m3.large m3.xlarge m3.2xlarge
+      m4.large m4.xlarge m4.2xlarge m4.4xlarge m4.10xlarge m4.16xlarge
+      m2.xlarge m2.2xlarge m2.4xlarge
+      cr1.8xlarge
+      r3.large r3.xlarge r3.2xlarge r3.4xlarge r3.8xlarge
+      x1.16xlarge x1.32xlarge
+      i2.xlarge i2.2xlarge i2.4xlarge i2.8xlarge
+      hi1.4xlarge
+      hs1.8xlarge
+      c1.medium c1.xlarge
+      c3.large c3.xlarge c3.2xlarge c3.4xlarge c3.8xlarge
+      c4.large c4.xlarge c4.2xlarge c4.4xlarge c4.8xlarge
+      cc1.4xlarge cc2.8xlarge
+      g2.2xlarge g2.8xlarge
+      cg1.4xlarge
+      p2.xlarge p2.8xlarge p2.16xlarge
+      d2.xlarge d2.2xlarge d2.4xlarge d2.8xlarge
+    )
+
+    def billing
       services = list_services
 
-      col = (services + ['Total']).map(&:size).max
-
-      total_price = get_price
-      msg = ["#{'Total'.ljust(col)} : #{total_price} USD"]
+      prices = {}
+      prices['Total'] = get_price
 
       services.each do |service|
-        msg.push("#{service.ljust(col)} : #{get_price(service)} USD")
+        prices[service] = get_price(service)
+      end
+
+      message.reply(table(prices), code: true)
+    end
+
+    def spot_instance
+      az = message[:availability_zone]
+      types = message[:instances].split(/\s*,\s*/)
+
+      types = types.keep_if { |type| INSTANCE_TYPES.include?(type) }
+
+      if types.empty?
+        message.reply('Not available instance types')
+        return
+      end
+
+      unless availability_zones.include?(az)
+        message.reply('Availability zone is invalid')
+        return
+      end
+
+      res = ec2.describe_spot_price_history({
+        dry_run: false,
+        start_time: Time.now,
+        end_time: Time.now,
+        instance_types: types,
+        availability_zone: az,
+        max_results: 5,
+      })
+
+      prices = {}
+      res.spot_price_history.sort_by(&:instance_type).each do |history|
+        prices[history.instance_type + ' '+ history.product_description] = "#{history.spot_price} USD"
       end
 
-      message.reply(msg.join("\n"))
+      message.reply(table(prices), code: true)
     end
 
     def list_services
-      resp = cloudwatch.list_metrics(namespace: 'AWS/Billing',
-                                     metric_name: 'EstimatedCharges',
-                                     dimensions: [
-                                       { name: 'Currency', value: 'USD' }
-                                     ])
+      resp = cloudwatch.list_metrics(
+        namespace: 'AWS/Billing',
+        metric_name: 'EstimatedCharges',
+        dimensions: [
+          { name: 'Currency', value: 'USD' }
+        ]
+      )
 
       services = []
       resp.metrics.each do |metric|
@@ -56,7 +111,7 @@ module Ruboty::Kanjo::Actions
         unit: 'None'
       )
 
-      resp.datapoints.last.maximum
+      resp.datapoints.last&.maximum
     end
 
     private
@@ -64,5 +119,24 @@ module Ruboty::Kanjo::Actions
     def cloudwatch
       @cloudwatch ||= Aws::CloudWatch::Client.new(region: 'us-east-1')
     end
+
+    def ec2
+      @ec2 ||= Aws::EC2::Client.new()
+    end
+
+    def availability_zones
+      ec2.describe_availability_zones.availability_zones.map(&:zone_name)
+    end
+
+    def table(data)
+      col = data.keys.map(&:size).max
+
+      lines = []
+      data.each_pair do |key, val|
+        lines.push("#{key.ljust(col)} : #{val}")
+      end
+
+      lines.join("\n")
+    end
   end
 end

data/lib/ruboty/kanjo/version.rb

@@ -1,5 +1,5 @@
 module Ruboty
   module Kanjo
-    VERSION = '0.1.0'.freeze
+    VERSION = '1.0.0'.freeze
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruboty-kanjo
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Sho Kusano