ec2.rb

task :classify_nodes do
  @dns_names.each do |description|
    description =~ /^(.+?)\..+$/
    hostname = $1
    node = @ic.get_node(hostname)
    if exists?(:puppet_env)
      EC2_DATA[:iclassify_attribs][:puppet_env] = puppet_env
    end
    node.tags = EC2_DATA[:iclassify_tags]
     EC2_DATA[:iclassify_attribs].each do |name, values|
       exists = node.attribs.detect { |a| a[:name] == name }
       if exists
         exists[:values] = values.kind_of?(Array) ? values : [ values ]
       else
         node.attribs << { :name => name, :values => values.kind_of?(Array) ? values : [ values ] }
       end
     end
     logger.info("Classifying #{node.attrib?('fqdn')} as #{EC2_DATA[:iclassify_tags].join(', ')}")
     puts node.inspect
     @ic.update_node(node)
  end
end