cfn-model 0.4.12 → 0.4.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a1706430755f12f30280c2a2baebb5566d50898ede81ec20e503ef91642e08b6
-  data.tar.gz: 30075abf14b456b2ebed049d9af2b68cfee7be938f8bcf9b8936170e651f8eeb
+  metadata.gz: 82992d31bca207f0fdccac49ce35cd26042703ea19c9cccd6c84166c534fcb30
+  data.tar.gz: c73dfff8dc47c2112b3d7c31ed9ff9c88d00cb7e9519495383b54d924d4110e9
 SHA512:
-  metadata.gz: a7223fce1ad56a488c37cfc27c0b35a2b5f0df76c4d3dc7b55ca2d46139b8f2f479e5ae981549ac91844d710cabf2519a73fbc1507426e0ee254976cf0a328d6
-  data.tar.gz: a1e9d39a1f17aa3f0ed0238cb8abc098a2880f3873ff3270ff9f2e6dc6c8dbdc017c96e3c0e6ca30531de5be8adf7a3494cd97b843476a8300ebda657fea977a
+  metadata.gz: 0de79a70b411c0655e60e0fbc61454b4fd8b662a132a66ec56512c5ce9dce7a338c3994dd8df1fa988bc281defed281e0300197b23383034bb147b05e8dec399
+  data.tar.gz: 02a845a5e38a29d9d097ec1a31fdd6be34ce28760150c381e7070776fb864864a70659b58ed409eacb1b02185ec17d8debdfe00d18348c0f9da2b20e9c1f8d47

data/lib/cfn-model/model/cfn_model.rb

@@ -3,7 +3,7 @@
 require_relative 'references'
 
 class CfnModel
-  attr_reader :resources, :parameters, :line_numbers
+  attr_reader :resources, :parameters, :line_numbers, :conditions
 
   ##
   # if you really want it, here it is - the raw Hash from YAML.load.  you'll have to mess with structural nits of
@@ -14,6 +14,7 @@ class CfnModel
   def initialize
     @parameters = {}
     @resources = {}
+    @conditions = {}
     @raw_model = nil
     @line_numbers = {}
   end
@@ -24,6 +25,9 @@ class CfnModel
   # the Hash is a clone
   def copy
     new_cfn_model = CfnModel.new
+    @conditions.each do |k,v|
+      new_cfn_model.conditions[k] = v
+    end
     @parameters.each do |k,v|
       new_cfn_model.parameters[k] = v
     end

data/lib/cfn-model/parser/cfn_parser.rb

@@ -38,8 +38,8 @@ class CfnParser
   ##
   # Given raw json/yml CloudFormation template, returns a CfnModel object
   # or raise ParserErrors if something is amiss with the format
-  def parse(cloudformation_yml, parameter_values_json=nil, with_line_numbers=false)
-    cfn_model = parse_without_parameters(cloudformation_yml, with_line_numbers)
+  def parse(cloudformation_yml, parameter_values_json=nil, with_line_numbers=false, condition_values_json=nil)
+    cfn_model = parse_without_parameters(cloudformation_yml, with_line_numbers, condition_values_json)
 
     apply_parameter_values(cfn_model, parameter_values_json)
 
@@ -54,7 +54,7 @@ class CfnParser
     ToRubyWithLineNumbers.create.accept(handler.root).first
   end
 
-  def parse_without_parameters(cloudformation_yml, with_line_numbers=false)
+  def parse_without_parameters(cloudformation_yml, with_line_numbers=false, condition_values_json=nil)
     pre_validate_model cloudformation_yml
 
     cfn_hash =
@@ -73,6 +73,8 @@ class CfnParser
     cfn_model = CfnModel.new
     cfn_model.raw_model = cfn_hash
 
+    process_conditions cfn_hash, cfn_model, condition_values_json
+
     # pass 1: wire properties into ModelElement objects
     if with_line_numbers
       transform_hash_into_model_elements_with_numbers cfn_hash, cfn_model
@@ -89,6 +91,24 @@ class CfnParser
 
   private
 
+  def process_conditions(cfn_hash, cfn_model, condition_values_json)
+    if cfn_hash.key?('Conditions')
+      if condition_values_json.nil?
+        condition_values = {}
+      else
+        condition_values = JSON.load condition_values_json
+      end
+
+      cfn_hash['Conditions'].each do |condition_key, _|
+        if condition_values.key?(condition_key) && [true, false].include?(condition_values[condition_key])
+          cfn_model.conditions[condition_key] = condition_values[condition_key]
+        else
+          cfn_model.conditions[condition_key] = true
+        end
+      end
+    end
+  end
+
   def apply_parameter_values(cfn_model, parameter_values_json)
     ParameterSubstitution.new.apply_parameter_values(
       cfn_model,
@@ -119,7 +139,7 @@ class CfnParser
       resource_object.resource_type = resource['Type']
       resource_object.metadata = resource['Metadata']
 
-      assign_fields_based_upon_properties resource_object, resource
+      assign_fields_based_upon_properties resource_object, resource, cfn_model
 
       cfn_model.resources[resource_name] = resource_object
     end
@@ -135,7 +155,7 @@ class CfnParser
       resource_object.resource_type = resource['Type']['value']
       resource_object.metadata = resource['Metadata']
 
-      assign_fields_based_upon_properties resource_object, resource
+      assign_fields_based_upon_properties resource_object, resource, cfn_model
 
       cfn_model.resources[resource_name] = resource_object
       cfn_model.line_numbers[resource_name] = resource['Type']['line']
@@ -175,11 +195,31 @@ class CfnParser
     end
   end
 
-  def assign_fields_based_upon_properties(resource_object, resource)
+  def deal_with_conditional_property_definitions(resource, cfn_model)
+    all_extra_concrete_properties = []
+    resource['Properties'].each do |property_name, property_value|
+      next if %w(Fn::Transform).include? property_name
+      if property_name == 'Fn::If'
+        concrete_properties = ExpressionEvaluator.new.evaluate(
+          {'Fn::If'=>property_value},
+          cfn_model.conditions
+        )
+        all_extra_concrete_properties << concrete_properties
+      end
+    end
+    all_extra_concrete_properties.each do |extra_concrete_properties|
+      resource['Properties'].merge!(extra_concrete_properties)
+    end
+    resource['Properties'].delete('Fn::If')
+  end
+
+  def assign_fields_based_upon_properties(resource_object, resource, cfn_model)
     unless resource['Properties'].nil?
+      deal_with_conditional_property_definitions(resource, cfn_model)
+
       resource['Properties'].each do |property_name, property_value|
-        next if %w(Fn::Transform Fn::If).include? property_name
-        resource_object.send("#{map_property_name_to_attribute(property_name)}=", property_value)
+        next if %w(Fn::Transform).include? property_name
+        resource_object.send("#{map_property_name_to_attribute(property_name)}=", map_property_value(property_value, cfn_model))
       end
     end
   end
@@ -194,6 +234,10 @@ class CfnParser
     resource_class
   end
 
+  def map_property_value(property_value, cfn_model)
+    ExpressionEvaluator.new.evaluate(property_value, cfn_model.conditions)
+  end
+
   def map_property_name_to_attribute(str)
     (str.slice(0).downcase + str[1..(str.length)]).gsub /-/, '_'
   end

data/lib/cfn-model/parser/expression_evaluator.rb

@@ -0,0 +1,39 @@
+class ExpressionEvaluator
+  FN_IF = 'Fn::If'
+
+  ##
+  # {'Fn::If'=>[Condition,X,Y]} returns X if conditions doesn't include Condition, otherwise it return X or Y depending
+  #
+  # Other than Fn::If, it just returns the value itself
+  def evaluate(expression, conditions)
+    if if_condition?(expression)
+      outcome(expression, conditions)
+    else
+      expression
+    end
+  end
+
+  private
+
+  def outcome(expression, conditions)
+    if if_condition?(expression)
+      if_expression = expression[FN_IF]
+      condition_name = if_expression[0]
+      if conditions[condition_name]
+        outcome(if_expression[1], conditions)
+      else
+        outcome(if_expression[2], conditions)
+      end
+    elsif expression.is_a?(Hash) # plain dict
+      expression.each do |k,v|
+        expression[k] = outcome(v, conditions)
+      end
+    else
+      expression
+    end
+  end
+
+  def if_condition?(property_value)
+    property_value.is_a?(Hash) && property_value.key?(FN_IF) && property_value.size == 1
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cfn-model
 version: !ruby/object:Gem::Version
-  version: 0.4.12
+  version: 0.4.13
 platform: ruby
 authors:
 - Eric Kascic
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-01-13 00:00:00.000000000 Z
+date: 2020-01-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubocop
@@ -89,6 +89,7 @@ files:
 - lib/cfn-model/parser/cfn_parser.rb
 - lib/cfn-model/parser/ec2_instance_parser.rb
 - lib/cfn-model/parser/ec2_network_interface_parser.rb
+- lib/cfn-model/parser/expression_evaluator.rb
 - lib/cfn-model/parser/iam_group_parser.rb
 - lib/cfn-model/parser/iam_role_parser.rb
 - lib/cfn-model/parser/iam_user_parser.rb