module Aws
  class RiKanjoo
    class Mode
      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'
          @price_url      = "https://a0.awsstatic.com/pricing/1/rds/#{@rdbms}"
          @multiaz        = multiaz
        end

        def om_price_from_contents contents
          region = self.convert_region(@region)

          # parse
          json = self.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
        end

        def ri_price_from_contents contents
          region = self.convert_region(@region)
          mtype  = (@multiaz) ? 'multiAZdeployRes' : 'stdDeployRes'

          # parse
          json = self.parse_contents(contents)

          ri_info = {}

          json["config"]["regions"].each do |r|
            next unless r["region"] == region

            r["instanceTypes"].each do |type|
              # beauty
              type["type"].gsub!(/Single-AZ Deployment \(Reserved\)/, 'stdDeployRes')

              next unless type["type"] == mtype

              type["tiers"].each do |i|
                next unless i["size"] == @instance_type

                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

          return ri_info
        end

        def price_url
          @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 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 "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
end