cloudformation-ruby-dsl-addedvars 1.2.4

data/lib/cloudformation-ruby-dsl-addedvars/dsl.rb

@@ -0,0 +1,270 @@
+require 'json'
+
+############################ Utility functions
+
+# Formats a template as JSON
+def generate_template(template)
+  format_json template, !template.nopretty
+end
+
+def generate_json(data, pretty = true)
+  # Raw formatting
+  return JSON.generate(data) unless pretty
+
+  # Pretty formatting
+  JSON.pretty_generate(data)
+end
+
+
+############################# Generic DSL
+
+class JsonObjectDSL
+  def initialize(&block)
+    @dict = {}
+    instance_eval &block
+  end
+
+  def value(values)
+    @dict.update(values)
+  end
+
+  def default(key, value)
+    @dict[key] ||= value
+  end
+
+  def to_json(*args)
+    @dict.to_json(*args)
+  end
+
+  def print()
+    puts JSON.pretty_generate(self)
+  end
+end
+
+############################# CloudFormation DSL
+
+# Main entry point
+def raw_template(parameters = {}, stack_name = nil, aws_region = default_region, aws_profile = nil, nopretty = false, &block)
+  TemplateDSL.new(parameters, stack_name, aws_region, aws_profile, nopretty, &block)
+end
+
+def default_region
+  ENV['EC2_REGION'] || ENV['AWS_DEFAULT_REGION'] || 'us-east-1'
+end
+
+# Core interpreter for the DSL
+class TemplateDSL < JsonObjectDSL
+  attr_reader :parameters, :aws_region, :nopretty, :stack_name, :aws_profile
+
+  def initialize(parameters = {}, stack_name = nil, aws_region = default_region, aws_profile = nil, nopretty = false)
+    @parameters = parameters
+    @stack_name = stack_name
+    @aws_region = aws_region
+    @aws_profile = aws_profile
+    @nopretty = nopretty
+    super()
+  end
+
+  def exec!()
+    cfn(self)
+  end
+
+  def parameter(name, options)
+    default(:Parameters, {})[name] = options
+    @parameters[name] ||= options[:Default]
+  end
+
+  # Find parameters where the specified attribute is true then remove the attribute from the cfn template.
+  def excise_parameter_attribute!(attribute)
+    marked_parameters = []
+    @dict.fetch(:Parameters, {}).each do |param, options|
+      if options.delete(attribute.to_sym) or options.delete(attribute.to_s)
+        marked_parameters << param
+      end
+    end
+    marked_parameters
+  end
+
+  def mapping(name, options)
+    # if options is a string and a valid file then the script will process the external file.
+    default(:Mappings, {})[name] = \
+      if options.is_a?(Hash); options
+      elsif options.is_a?(String); load_from_file(options)['Mappings'][name]
+      else; raise("Options for mapping #{name} is neither a string or a hash.  Error!")
+    end
+  end
+
+  def load_from_file(filename)
+    file = File.open(filename)
+
+    begin
+      # Figure out what the file extension is and process accordingly.
+      contents = case File.extname(filename)
+        when ".rb"; eval(file.read, nil, filename)
+        when ".json"; JSON.load(file)
+        when ".yaml"; YAML::load(file)
+        else; raise("Do not recognize extension of #{filename}.")
+      end
+    ensure
+      file.close
+    end
+    contents
+  end
+
+  # Find tags where the specified attribute is true then remove this attribute.
+  def get_tag_attribute(tags, attribute)
+    marked_tags = []
+    tags.each do |tag, options|
+      if options.delete(attribute.to_sym) or options.delete(attribute.to_s)
+        marked_tags << tag
+      end
+    end
+    marked_tags
+  end
+
+  def excise_tags!
+    tags = @dict.fetch(:Tags, {})
+    @dict.delete(:Tags)
+    tags
+  end
+
+  def tag(tag, *args)
+    if (tag.is_a?(String) || tag.is_a?(Symbol)) && !args.empty?
+      default(:Tags, {})[tag.to_s] = args[0]
+    # For backward-compatibility, transform `tag_name=>value` format to `tag_name, :Value=>value, :Immutable=>true`
+    # Tags declared this way remain immutable and won't be updated.
+    elsif tag.is_a?(Hash) && tag.size == 1 && args.empty?
+      $stderr.puts "WARNING: #{tag} tag declaration format is deprecated and will be removed in a future version. Please use resource-like style instead."
+      tag.each do |name, value|
+        default(:Tags, {})[name.to_s] = {:Value => value, :Immutable => true}
+      end
+    else
+      $stderr.puts "Error: #{tag} tag validation error. Please verify tag's declaration format."
+      exit(false)
+    end
+  end
+
+  def condition(name, options) default(:Conditions, {})[name] = options end
+
+  def resource(name, options) default(:Resources, {})[name] = options end
+
+  def output(name, options) default(:Outputs, {})[name] = options end
+
+  def find_in_map(map, key, name)
+    # Eagerly evaluate mappings when all keys are known at template expansion time
+    if map.is_a?(String) && key.is_a?(String) && name.is_a?(String)
+      # We don't know whether the map was built with string keys or symbol keys.  Try both.
+      def get(map, key) map[key] || map.fetch(key.to_sym) end
+      get(get(@dict.fetch(:Mappings).fetch(map), key), name)
+    else
+      { :'Fn::FindInMap' => [ map, key, name ] }
+    end
+  end
+end
+
+def base64(value) { :'Fn::Base64' => value } end
+
+def find_in_map(map, key, name) { :'Fn::FindInMap' => [ map, key, name ] } end
+
+def get_att(resource, attribute) { :'Fn::GetAtt' => [ resource, attribute ] } end
+
+def get_azs(region = '') { :'Fn::GetAZs' => region } end
+
+def join(delim, *list)
+  case list.length
+    when 0 then ''
+    else join_list(delim,list)
+  end
+end
+
+# Variant of join that matches the native CFN syntax.
+def join_list(delim, list) { :'Fn::Join' => [ delim, list ] } end
+
+def equal(one, two) { :'Fn::Equals' => [one, two] } end
+
+def fn_not(condition) { :'Fn::Not' => [condition] } end
+
+def fn_or(*condition_list)
+  case condition_list.length
+    when 0..1 then raise "fn_or needs at least 2 items."
+    when 2..10 then  { :'Fn::Or' => condition_list }
+    else raise "fn_or needs a list of 2-10 items that evaluate to true/false."
+  end
+end
+
+def fn_and(*condition_list)
+  case condition_list.length
+    when 0..1 then raise "fn_and needs at least 2 items."
+    when 2..10 then  { :'Fn::And' => condition_list }
+    else raise "fn_and needs a list of 2-10 items that evaluate to true/false."
+  end
+end
+
+def fn_if(cond, if_true, if_false) { :'Fn::If' => [cond, if_true, if_false] } end
+
+def not_equal(one, two) fn_not(equal(one,two)) end
+
+def select(index, list) { :'Fn::Select' => [ index, list ] } end
+
+def ref(name) { :Ref => name } end
+
+def aws_account_id() ref("AWS::AccountId") end
+
+def aws_notification_arns() ref("AWS::NotificationARNs") end
+
+def aws_no_value() ref("AWS::NoValue") end
+
+def aws_stack_id() ref("AWS::StackId") end
+
+def aws_stack_name() ref("AWS::StackName") end
+
+# deprecated, for backward compatibility
+def no_value()
+    warn_deprecated('no_value()', 'aws_no_value()')
+    aws_no_value()
+end
+
+# Read the specified file and return its value as a string literal
+def file(filename) File.read(File.absolute_path(filename, File.dirname($PROGRAM_NAME))) end
+
+# Interpolates a string like "NAME={{ref('Service')}}" and returns a CloudFormation "Fn::Join"
+# operation to collect the results.  Anything between {{ and }} is interpreted as a Ruby expression
+# and eval'd.  This is especially useful with Ruby "here" documents.
+# Local variables may also be exposed to the string via the `locals` hash.
+def interpolate(string, locals={})
+  list = []
+  while string.length > 0
+    head, match, string = string.partition(/\{\{.*?\}\}/)
+    list << head if head.length > 0
+    list << eval(match[2..-3], nil, 'interpolated string') if match.length > 0
+  end
+
+  # Split out strings in an array by newline, for visibility
+  list = list.flat_map {|value| value.is_a?(String) ? value.lines.to_a : value }
+  join('', *list)
+end
+
+def join_interpolate(delim, string)
+  $stderr.puts "join_interpolate(delim,string) has been deprecated; use interpolate(string) instead"
+  interpolate(string)
+end
+
+# This class is used by erb templates so they can access the parameters passed
+class Namespace
+  attr_accessor :params
+  def initialize(hash)
+    @params = hash
+  end
+  def get_binding
+    binding
+  end
+end
+
+# Combines the provided ERB template with optional parameters
+def erb_template(filename, params = {})
+  ERB.new(file(filename), nil, '-').result(Namespace.new(params).get_binding)
+end
+
+def warn_deprecated(old, new)
+    $stderr.puts "Warning: '#{old}' has been deprecated.  Please update your template to use '#{new}' instead."
+end

data/lib/cloudformation-ruby-dsl-addedvars/spotprice.rb

@@ -0,0 +1,50 @@
+# Copyright 2013-2014 Bazaarvoice, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# This list of prices was sourced from the on-demand prices current as of 12/3/2012.
+# We expect the actual price we pay per instance to be roughly 1/10 the prices below.
+SPOT_PRICES_BY_INSTANCE_TYPE = {
+    "m1.small"    => 0.065,
+    "m1.medium"   => 0.130,
+    "m1.large"    => 0.260,
+    "m1.xlarge"   => 0.520,
+    "m3.xlarge"   => 0.580,
+    "m3.2xlarge"  => 1.160,
+    "t1.micro"    => 0.020,
+    "m2.xlarge"   => 0.450,
+    "m2.2xlarge"  => 0.900,
+    "m2.4xlarge"  => 1.800,
+    "c1.medium"   => 0.165,
+    "c1.xlarge"   => 0.660,
+    "cc1.4xlarge" => 1.300,
+    "cc2.8xlarge" => 2.400,
+    "cg1.4xlarge" => 2.100,
+    "hi1.4xlarge" => 3.100,
+    "hs1.8xlarge" => 4.600,
+    "cr1.8xlarge" => 4.000,
+}
+
+def spot_price(spot_price_string, instance_type)
+  case spot_price_string
+    when 'false', '' then aws_no_value()
+    when 'true' then spot_price_for_instance_type(instance_type)
+    else spot_price_string
+  end
+end
+
+def spot_price_for_instance_type(instance_type)
+  # Add 10% to ensure that we have a small buffer against current spot prices increasing
+  # to the on-demand prices, which theoretically could happen often.
+  SPOT_PRICES_BY_INSTANCE_TYPE[instance_type] * 1.10
+end

data/lib/cloudformation-ruby-dsl-addedvars/table.rb

@@ -0,0 +1,123 @@
+# Copyright 2013-2014 Bazaarvoice, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'detabulator'
+
+class Table
+  def self.load(filename)
+    self.new File.read filename
+  end
+
+  def initialize(table_as_text)
+    raw_header, *raw_data = Detabulator.new.detabulate table_as_text
+    @header = raw_header.map(&:to_sym)
+    @records = raw_data.map { |row| Hash[@header.zip(row)] }
+  end
+
+  # Selects all rows in the table which match the name/value pairs of the predicate object and returns
+  # the single distinct value from those rows for the specified key.
+  def get(key, predicate)
+    distinct_values(filter(predicate), key, false)
+  end
+
+  # Select the headers as list. Argument(s) will be excluded from output.
+  def get_header(*exclude)
+    @header.reject{ |key| key if exclude.include?(key) }
+  end
+
+  # Selects all rows in the table which match the name/value pairs of the predicate object and returns
+  # all distinct values from those rows for the specified key.
+  def get_list(key, predicate)
+    distinct_values(filter(predicate), key, true)
+  end
+
+  # Selects all rows in the table which match the name/value pairs of the predicate object and returns a
+  # hash of hashes, where the key for the top-level hash is the key paramter and the second-level hash keys are
+  # those in the keys paramter. This is useful when you want multiple column values for a given row.
+  def get_multihash(key, predicate, *keys)
+    build_nested_hash(filter(predicate), key, keys)
+  end
+
+  # Selects all rows in the table which match the name/value pairs of the predicate object and returns a
+  # set of nested maps, where the key for the map at level n is the key at index n in the specified keys,
+  # except for the last key in the specified keys which is used to determine the value of the leaf-level map.
+  # In the simple case where keys is a list of 2 elements, this returns a map from key[0] to key[1].
+  def get_map(predicate, *keys)
+    build_nested_map(filter(predicate), keys, false)
+  end
+
+  # Selects all rows in the table which match the name/value pairs of the predicate object and returns a
+  # set of nested maps, where the key for the map at level n is the key at index n in the specified keys,
+  # except for the last key in the specified keys which is used to determine the list of values in the
+  # leaf-level map.  In the simple case where keys is a list of 2 elements, this returns a map from key[0]
+  # to a list of values for key[1].
+  def get_multimap(predicate, *keys)
+    build_nested_map(filter(predicate), keys, true)
+  end
+
+  private
+
+  # Return the subset of records that match the predicate for all keys in the predicate.
+  # The predicate is expected to be a map of key/value or key/[value,...] pairs.
+  def filter(predicate)
+    def matches(predicate_value, record_value)
+      if predicate_value.is_a?(Array); predicate_value.include?(record_value)
+      else; predicate_value == record_value
+      end
+    end
+
+    @records.select { |record| predicate.all? { |key, value| matches(value, record[key]) } }
+  end
+
+  def build_nested_hash(records, key, keys)
+    hash = {}
+    records.each do |record|
+      hash[record[key]] = {}
+      keys.each do |hash_key|
+        hash[record[key]][hash_key] = record[hash_key]
+      end
+    end
+    return hash
+  end
+
+  def build_nested_map(records, path, multi)
+    key, *rest = path
+    if rest.empty?
+      # Build the leaf level of the data structure
+      distinct_values(records, key, multi)
+    else
+      # Return a hash keyed by the distinct values of the first key and values are the result of a
+      # recursive invocation of arrange() with the rest of the keys
+      result = {}
+      records.group_by do |record|
+        record[key]
+      end.map do |value, group|
+        result[value] = build_nested_map(group, rest, multi)
+      end
+      result
+    end
+  end
+
+  def distinct_values(records, key, multi)
+    values = records.map { |record| record[key] }.uniq
+    if multi
+      # In a multimap the leaf level is a list of string values
+      values
+    else
+      # In a non-multimap the leaf level is a single string value
+      raise "Multiple distinct values for the same key '#{key}': #{records.inspect}" if values.length > 1
+      values[0]
+    end
+  end
+end

data/lib/cloudformation-ruby-dsl-addedvars/version.rb

@@ -0,0 +1,21 @@
+# Copyright 2013-2014 Bazaarvoice, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+module Cfn
+  module Ruby
+    module Dsl
+      VERSION = "1.2.4"
+    end
+  end
+end