aws-rikanjo 0.0.8 → 0.0.9

data/lib/aws/rikanjo/cli.rb

@@ -0,0 +1,113 @@
+require 'optparse'
+require 'aws/rikanjo/base'
+
+module Aws
+  module RiKanjoo
+    class CLI
+
+      def initialize
+        @options = {}
+        @optparse = nil
+        @cost = nil
+      end
+
+      def start
+        parse
+        rikanjo
+        output
+      end
+
+      def parse
+        optparse = OptionParser.new do |opts|
+          opts.banner = 'Usage: rikanjo ec2/rds [options]'
+
+          # subcommand
+          mode = ARGV.shift
+          if mode == 'ec2'
+          elsif mode == 'rds'
+            opts.on('--multiaz', 'enable multi-az') do |value|
+              @options[:multiaz] = value
+            end
+          else
+            $stderr.puts "no such subcommand: #{mode}"
+            puts opts
+            exit 1
+          end
+          @options[:mode] = mode
+
+          region_values = %w(us-east-1 us-west-1 us-west-2 eu-west-1 ap-southeast-1 ap-northeast-1 ap-southeast-2 sa-east-1 )
+          opts.on('-r', '--region=VALUE', region_values, "specify aws-region (#{region_values.join('/')})") do |value|
+            @options[:region] = value
+          end
+
+          opts.on('-t', '--instance_type=VALUE', 'specify ec2-instance-type') do |value|
+            @options[:instance_type] = value
+          end
+
+          ri_util_values = %w(light medium heavy)
+          opts.on('-u', '--ri_util=VALUE', ri_util_values, "specify ri-util (#{ri_util_values.join('/')})") do |value|
+            @options[:ri_util] = value
+          end
+
+          opts.on('-j', '--output_json', "specify output_json flg") do |value|
+            @options[:output_json] = value
+          end
+
+          opts.on('-h', '--help') do
+            puts opts
+            exit
+          end
+        end
+
+        # validation
+        begin
+          optparse.parse!
+          require_args = [:region, :instance_type, :ri_util]
+          error_args = require_args.select { |param| @options[param].nil? }
+          unless error_args.empty?
+            puts "require arguments: #{error_args.join(', ')}"
+            puts
+            puts optparse
+            exit
+          end
+        rescue
+          puts $!.to_s
+          puts
+          puts optparse
+          exit
+        end
+      end
+
+      def rikanjo
+        # rikanjo
+        a = Aws::RiKanjoo::Base.new(
+            mode = @options[:mode],
+            region = @options[:region],
+            instance_type = @options[:instance_type],
+            ri_util = @options[:ri_util],
+            multiaz = @options[:multiaz],
+        )
+        @cost = a.total_cost_year
+      end
+
+      def output
+        if @options[:output_json]
+          output_json()
+        else
+          output_text()
+        end
+      end
+
+      def output_json
+        puts @cost.to_json
+      end
+
+      def output_text
+        @cost.keys.each do |key|
+          puts "#{key}: #{@cost[key]}"
+        end
+      end
+
+    end
+  end
+end

data/lib/aws/rikanjo/ec2.rb

@@ -0,0 +1,117 @@
+require 'yajl'
+
+module Aws
+  module RiKanjoo
+    class Ec2
+      attr_reader :region, :instance_type, :ri_util
+
+      def initialize(region, instance_type, ri_util)
+        @region             = region
+        @instance_type      = instance_type
+        @ri_util            = ri_util
+        @os                 = 'linux'
+        @current_price_url  = 'http://a0.awsstatic.com/pricing/1/ec2'
+        @previous_price_url = 'http://a0.awsstatic.com/pricing/1/ec2/previous-generation'
+      end
+
+      def om_price_from_contents(contents)
+        region = convert_region(@region)
+
+        # parse
+        json = parse_contents(contents)
+
+        # select 'region' and 'instance_type'
+        json['config']['regions'].each do |r|
+          # r = {"region"=>"us-east", "instanceTypes"=>[{"type"=>"generalCurrentGen", ...
+          next unless r['region'] == region
+          r['instanceTypes'].each do |type|
+            # type = {"type"=>"generalCurrentGen", "sizes"=>[{"size"=>"m3.medium", "vCPU"=>"1" ...
+            type['sizes'].each do |i|
+              next unless i['size'] == @instance_type
+
+              return { hr_price: i['valueColumns'][0]['prices']['USD'] }
+            end
+          end
+        end
+
+        abort("[ec2][#{region}][#{@instance_type}] Not found om-price?")
+      end
+
+      def ri_price_from_contents(contents)
+        region = clean_region(@region)
+
+        # parse
+        json = parse_contents(contents)
+
+        ri_info = nil
+
+        json['config']['regions'].each do |r|
+          next unless r['region'] == region
+
+          r['instanceTypes'].each do |type|
+            type['sizes'].each do |i|
+              next unless i['size'] == @instance_type
+
+              ri_info = ri_info ||= {}
+              i['valueColumns'].each do |y|
+                case y['name']
+                when 'yrTerm1'
+                  ri_info[@ri_util] = { upfront: y['prices']['USD'] }
+                when 'yrTerm1Hourly'
+                  ri_info[@ri_util].store(:hr_price, y['prices']['USD'])
+                end
+              end
+            end
+          end
+        end
+
+        abort("[ec2][#{region}][#{@instance_type}] Not found ri-price?") unless ri_info
+
+        return ri_info
+      end
+
+      def price_url
+        return (self.previous_generation_type) ? @previous_price_url : @current_price_url
+      end
+
+      def om_price_file
+        return "#{@os}-od.min.js"
+      end
+
+      def ri_price_file
+        return (self.previous_generation_type) ? "#{@ri_util}_#{@os}.min.js" : "#{@os}-ri-#{@ri_util}.min.js"
+      end
+
+      def previous_generation_type
+        case @instance_type
+        when /^(c1|m2|cc2\.8xlarge|cr1\.8xlarge|hi1\.4xlarge|cg1\.4xlarge)/ then true
+        when /^m1/ then (@instance_type == 'm1.small') ? false : true
+        else false
+        end
+      end
+
+      def parse_contents(data)
+        data = data.gsub("/*\n * This file is intended for use only on aws.amazon.com. We do not guarantee its availability or accuracy.\n *\n * Copyright 2014 Amazon.com, Inc. or its affiliates. All rights reserved.\n */\ncallback({",'{').gsub("\);", '').gsub(/([a-zA-Z]+):/, '"\1":')
+        return Yajl::Parser.parse(data)
+      end
+
+      def convert_region(region)
+        # not same om region
+        case region
+        when 'us-east-1'      then 'us-east'
+        when 'us-west-1'      then 'us-west'
+        when 'us-west-2'      then 'us-west-2'
+        when 'eu-west-1'      then 'eu-ireland'
+        when 'ap-southeast-1' then 'apac-sin'
+        when 'ap-northeast-1' then 'apac-tokyo'
+        when 'ap-southeast-2' then 'apac-syd'
+        when 'sa-east-1'      then 'sa-east-1'
+        end
+      end
+
+      def clean_region(region)
+        return region.gsub(/^us-east-1$/, 'us-east')
+      end
+    end
+  end
+end

data/lib/aws/rikanjo/rds.rb

@@ -0,0 +1,120 @@
+module Aws
+  module RiKanjoo
+    class Rds
+      attr_reader :region, :instance_type, :ri_util
+
+      def initialize(region, instance_type, ri_util, multiaz)
+        @region         = region
+        @instance_type  = "db.#{instance_type}"
+        @ri_util        = ri_util
+        @rdbms          = 'mysql'
+        @current_price_url  = "http://a0.awsstatic.com/pricing/1/rds/#{@rdbms}"
+        @previous_price_url = "http://a0.awsstatic.com/pricing/1/rds/#{@rdbms}/previous-generation"
+        @multiaz        = multiaz
+      end
+
+      def om_price_from_contents(contents)
+        region = convert_region(@region)
+
+        # parse
+        json = parse_contents(contents)
+
+        # select 'region' and 'instance_type'
+        json['config']['regions'].each do |r|
+          # r = {"region"=>"us-east", "instanceTypes"=>[{"type"=>"generalCurrentGen", ...
+          next unless r['region'] == region
+          r['types'].each do |type|
+            # type = {"type"=>"generalCurrentGen", "sizes"=>[{"size"=>"m3.medium", "vCPU"=>"1" ...
+            type['tiers'].each do |i|
+              next unless i['name'] == @instance_type
+
+              return { hr_price: i['prices']['USD'] }
+            end
+          end
+        end
+
+        abort("[rds][#{region}][#{@instance_type}] Not found om-price?")
+      end
+
+      def ri_price_from_contents(contents)
+        region = convert_region(@region)
+        mtype  = (@multiaz) ? 'Multi-AZ' : 'Single-AZ'
+
+        # parse
+        json = parse_contents(contents)
+
+        ri_info = nil
+
+        json['config']['regions'].each do |r|
+          next unless r['region'] == region
+
+          r['instanceTypes'].each do |type|
+            next unless type['type'] =~ /#{mtype}/
+
+            type['tiers'].each do |i|
+              next unless i['size'] == @instance_type
+
+              ri_info = ri_info ||= {}
+              i['valueColumns'].each do |y|
+                # beauty
+                y['name'].gsub!(/^year/, 'yr')
+                y['prices']['USD'].gsub!(/,/, '')
+
+                case y['name']
+                when 'yrTerm1'
+                  ri_info[@ri_util] = { upfront: y['prices']['USD'] }
+                when 'yrTerm1Hourly'
+                  ri_info[@ri_util].store(:hr_price, y['prices']['USD'])
+                end
+              end
+
+            end
+          end
+        end
+
+        abort("[rds][#{region}][#{@instance_type}] Not found ri-price?") unless ri_info
+
+        return ri_info
+      end
+
+      def price_url
+        return (self.previous_generation_type) ? @previous_price_url : @current_price_url
+      end
+
+      def om_price_file
+        return (@multiaz) ? "pricing-multiAZ-deployments.min.js" : "pricing-standard-deployments.min.js"
+      end
+
+      def ri_price_file
+        return "pricing-#{@ri_util}-utilization-reserved-instances.min.js"
+      end
+
+      def previous_generation_type
+        case @instance_type
+        when /^db.(c1|m2|cc2\.8xlarge|cr1\.8xlarge|hi1\.4xlarge|cg1\.4xlarge)/ then true
+        when /^db.m1/ then (@instance_type == 'db.m1.small') ? false : true
+        else false
+        end
+      end
+
+      def parse_contents(data)
+        data = data.gsub("/*\n * This file is intended for use only on aws.amazon.com. We do not guarantee its availability or accuracy.\n *\n * Copyright 2014 Amazon.com, Inc. or its affiliates. All rights reserved.\n */\ncallback({",'{').gsub("\);", '').gsub(/([a-zA-Z]+):/, '"\1":')
+        return Yajl::Parser.parse(data)
+      end
+
+      def convert_region(region)
+        # not same om region
+        case region
+        when 'us-east-1'      then 'us-east'
+        when 'us-west-1'      then 'us-west'
+        when 'us-west-2'      then 'us-west-2'
+        when 'eu-west-1'      then 'eu-ireland'
+        when 'ap-southeast-1' then 'apac-sin'
+        when 'ap-northeast-1' then 'apac-tokyo'
+        when 'ap-southeast-2' then 'apac-syd'
+        when 'sa-east-1'      then 'sa-east-1'
+        end
+      end
+    end
+  end
+end

data/lib/aws/rikanjo/version.rb

@@ -1,5 +1,5 @@
 module Aws
   module Rikanjo
-    VERSION = "0.0.8"
+    VERSION = "0.0.9"
   end
 end

data/spec/lib/rikanjo_ec2_spec.rb

@@ -1,14 +1,14 @@
 require "spec_helper"
-require "aws/rikanjo/mode/ec2"
+require "aws/rikanjo/ec2"
 
 include RikanjoSpecHelper
 
-describe 'AWS::Rikanjo::Mode::Ec2' do
+describe 'AWS::Rikanjo::Ec2' do
 
   describe 'previous type' do
     it "is previous instance type" do
       %w{ m1.medium c1.medium m2.xlarge cc2.8xlarge cr1.8xlarge hi1.4xlarge cg1.4xlarge }.each do |t|
-          a = Aws::RiKanjoo::Mode::Ec2.new(
+          a = Aws::RiKanjoo::Ec2.new(
               region        = 'ap-northeast-1',
               instance_type = t,
               ri_util       = 'light',
@@ -19,7 +19,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
 
     it "is current instance type" do
       %w{ t1.micro m1.small m3.medium c3.medium r3.xlarge i2.xlarge }.each do |t|
-          a = Aws::RiKanjoo::Mode::Ec2.new(
+          a = Aws::RiKanjoo::Ec2.new(
               region        = 'ap-northeast-1',
               instance_type = t,
               ri_util       = 'light',
@@ -31,7 +31,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
 
   describe 'url' do
     it "build previous price url" do
-      a = Aws::RiKanjoo::Mode::Ec2.new(
+      a = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm1.medium',
           ri_util       = 'light',
@@ -40,7 +40,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
     end
 
     it "build current price url" do
-      a = Aws::RiKanjoo::Mode::Ec2.new(
+      a = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm3.medium',
           ri_util       = 'light',
@@ -49,7 +49,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
     end
 
     it "build previous ri price file" do
-      a = Aws::RiKanjoo::Mode::Ec2.new(
+      a = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm1.medium',
           ri_util       = 'light',
@@ -58,7 +58,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
     end
 
     it "build current ri price file" do
-      a = Aws::RiKanjoo::Mode::Ec2.new(
+      a = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm3.medium',
           ri_util       = 'heavy',
@@ -72,13 +72,13 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
 
     before :all do
       # rikanjo (current)
-      a1 = Aws::RiKanjoo::Mode::Ec2.new(
+      a1 = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm3.medium',
           ri_util       = 'medium',
       )
       # rikanjo (previous)
-      a2 = Aws::RiKanjoo::Mode::Ec2.new(
+      a2 = Aws::RiKanjoo::Ec2.new(
           region        = 'ap-northeast-1',
           instance_type = 'm1.medium',
           ri_util       = 'medium',
@@ -91,7 +91,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
 
     it "is able to get the price" do
       regions.each do |region|
-        a = Aws::RiKanjoo::Mode::Ec2.new(
+        a = Aws::RiKanjoo::Ec2.new(
             region        = region,
             instance_type = 'm3.large',
             ri_util       = 'medium',
@@ -108,7 +108,7 @@ describe 'AWS::Rikanjo::Mode::Ec2' do
 
     it "an exception is raised when there are no instance-type" do
       regions.each do |region|
-        a = Aws::RiKanjoo::Mode::Ec2.new(
+        a = Aws::RiKanjoo::Ec2.new(
             region        = region,
             instance_type = 'm3.large.not.exists',
             ri_util       = 'medium',