CloudHealth/amazon-pricing

#--
# Amazon Web Services Pricing Ruby library
#
# Ruby Gem Name::  amazon-pricing
# Author::    Joe Kinsella (mailto:joe.kinsella@gmail.com)
# Copyright:: Copyright (c) 2011-2013 CloudHealth
# License::   Distributes under the same terms as Ruby
# Home::      http://github.com/CloudHealth/amazon-pricing
#++
module AwsPricing
  # Region represents a geographic area in which Amazon Web Services has
  # one or more data centers. Different regions will offer difference services
  # and pricing.
  #
  # e.g. us-east, us-west
  #
  class Region
    attr_accessor :name, :ebs_price, :ec2_instance_types, :rds_instance_types, :elasticache_node_types, :ec2_dh_types

    def initialize(name)
      @name = name
      @ec2_instance_types     = {}
      @rds_instance_types     = {}
      @elasticache_node_types = {}
      @ec2_dh_types           = {}
    end

    # Maintained for backward compatibility reasons (retrieves EC2 instance type)
    def instance_types
      ec2_instance_types
    end

    def ec2_instance_types
      @ec2_instance_types.values
    end

    def rds_instance_types
      @rds_instance_types.values
    end

    def elasticache_node_types
      @elasticache_node_types.values
    end

    def ec2_dh_types
      @ec2_dh_types.values
    end

    # Returns whether an instance_type is available. 
    # operating_system = :linux, :mswin, :rhel, :sles, :mswinSQL, :mswinSQLWeb
    # type_of_instance = :ondemand, :light, :medium, :heavy
    def instance_type_available?(api_name, type_of_instance = :ondemand, operating_system = :linux)
      instance = @ec2_instance_types[api_name]
      return false if instance.nil?
      instance.available?(type_of_instance, operating_system)
    end

    # type_of_instance = :ondemand, :light, :medium, :heavy
    def add_or_update_ec2_instance_type(api_name, name)
      current = get_ec2_instance_type(api_name)
      if current.nil?
        current = Ec2InstanceType.new(self, api_name, name)
        @ec2_instance_types[api_name] = current
      end
      current
    end

    def add_or_update_rds_instance_type(api_name, name)
      current = get_rds_instance_type(api_name)
      if current.nil?
        current = RdsInstanceType.new(self, api_name, name)
        @rds_instance_types[api_name] = current
      end
      current
    end

    def add_or_update_elasticache_node_type(api_name, name)
      current = get_elasticache_node_type(api_name)
      if current.nil?
        current = ElastiCacheNodeType.new(self, api_name, name)
        @elasticache_node_types[api_name] = current
      end
      current
    end

    def add_or_update_ec2_dh_type(family)
      current = get_ec2_dh_type(family)
      if current.nil?
        current = Ec2DedicatedHostType.new(self, family)
        @ec2_dh_types[family] = current
      end
      current
    end

    # Maintained for backward compatibility reasons (retrieves EC2 instance type)
    def get_instance_type(api_name)
      get_ec2_instance_type(api_name)
    end

    def get_ec2_instance_type(api_name)
      @ec2_instance_types[api_name]
    end

    def get_rds_instance_type(api_name)
      @rds_instance_types[api_name]
    end

    def get_elasticache_node_type(api_name)
      @elasticache_node_types[api_name]
    end

    def get_ec2_dh_type(family)
      @ec2_dh_types[family]
    end

  end

end