draw_cloud 0.0.1

data/lib/draw_cloud/base64_func.rb

@@ -0,0 +1,30 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this work except in compliance with the License. You may
+# obtain a copy of the License in the LICENSE file, or 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 DrawCloud
+  class Base64Func
+    attr_accessor :arg
+    def initialize(arg=nil)
+      @arg = arg
+    end
+
+    def ref
+      {"Fn::Base64" => DrawCloud.ref(arg) }
+    end
+  end
+end

data/lib/draw_cloud/configuration.rb

@@ -0,0 +1,61 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this work except in compliance with the License. You may
+# obtain a copy of the License in the LICENSE file, or 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 DrawCloud
+  class Configuration < Base
+    attr_accessor :description
+    def self.draw(&block)
+      c = Configuration.new
+      c.instance_exec(c, &block)
+      puts JSON.pretty_generate(c.to_cf)
+    end
+
+    def initialize(options={}, &block)
+      super(options, &block)
+    end
+
+    def to_cf
+      h = {"AWSTemplateFormatVersion" => "2010-09-09"}
+      h["Description"] = description if description
+
+      c = Configuration.new
+      self.load_into_config(c)
+
+      { "Mappings" => c.mappings,
+        "Parameters" => c.parameters,
+        "Resources" => c.resources,
+        "Outputs" => c.outputs }.each do |(key, values)|
+        h[key] = Hash[*values.collect {|k,v| [k, v.to_h]}.flatten] unless values.empty?
+      end
+      h
+    end
+
+    def cf_add_mapping(name, map)
+      mappings[name] = map
+    end
+    def cf_add_parameter(name, param)
+      parameters[name] = param
+    end
+    def cf_add_resource(name, res)
+      resources[name] = res
+    end
+    def cf_add_output(name, out)
+      outputs[name] = out
+    end
+  end
+end

data/lib/draw_cloud/configuration.rb~

@@ -0,0 +1,64 @@
+module DrawCloud
+  class Configuration
+    attr_accessor :mappings, :parameters, :resources, :outputs, :vpcs
+    def self.draw(&block)
+      c = Configuration.new
+      c.instance_exec(c, &block)
+      puts JSON.pretty_generate(c.to_cf)
+    end
+
+    def initialize
+      @mappings = {}
+      @parameters = {}
+      @resources = {}
+      @outputs = {}
+
+      @vpcs = []
+    end
+
+    def description(s)
+      @description = s
+    end
+
+    def mapping(name, map_by_function, values={})
+      m = Map.new(name, map_by_function, values)
+      @mappings[m.resource_name] = m
+      m
+    end
+
+    def vpc(name, cidr, &block)
+      v = Vpc.new(name, cidr, &block)
+      @vpcs << v
+      v
+    end
+
+    def to_cf
+      h = {"AWSTemplateFormatVersion" => "2010-09-09"}
+      h["Description"] = @description if @description
+
+      c = Configuration.new
+      [@mappings, @parameters, @resources, @outputs].each {|i| i.each {|k,v| v.load_into_config(c) } }
+      @vpcs.each {|v| v.load_into_config(c) }
+      { "Mappings" => c.mappings,
+        "Parameters" => c.parameters,
+        "Resources" => c.resources,
+        "Outputs" => c.outputs }.each do |(key, values)|
+        h[key] = Hash[*values.collect {|k,v| [k, v.to_h]}.flatten] unless values.empty?
+      end
+      h
+    end
+
+    def cf_add_mapping(name, map)
+      mappings[name] = map
+    end
+    def cf_add_parameter(name, param)
+      parameters[name] = param
+    end
+    def cf_add_resource(name, res)
+      resources[name] = res
+    end
+    def cf_add_output(name, out)
+      outputs[name] = out
+    end
+  end
+end

data/lib/draw_cloud/ec2_instance.rb

@@ -0,0 +1,141 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this work except in compliance with the License. You may
+# obtain a copy of the License in the LICENSE file, or 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 DrawCloud
+  class EC2Instance < Base
+    attr_accessor(:name,
+                  :availability_zone,
+                  :disable_api_termination,
+                  :image_id,
+                  :instance_type,
+                  :key_name,
+                  :monitoring,
+                  :placement_group_name,
+                  :private_ip_address,
+                  :source_dest_check,
+                  :subnet_id,
+                  :tags,
+                  :user_data,
+                  :metadata,
+                  :template,
+                  :eip_name)
+    alias :instance_class :instance_type
+    alias :instance_class= :instance_type=
+    alias :instance_monitoring :monitoring
+    alias :instance_monitoring= :monitoring=
+    alias :ami :image_id
+    alias :ami= :image_id=
+    alias :subnet :subnet_id
+    alias :subnet= :subnet_id=
+    def initialize(name, options={}, &block)
+      @name = name
+      @tags = {}
+      @template = options.fetch(:template, nil)
+      super(options, &block)
+    end
+
+    def ec2_instance
+      self
+    end
+
+    def elastic_ip=(eip)
+      case eip
+      when DrawCloud::ElasticIp
+        eip.instance_id = self
+        self.eip_name = nil
+      else
+        self.eip_name = eip
+      end
+    end
+
+    def elastic_ip_association
+      DrawCloud::ElasticIp::ElasticIpAssociation.new(eip_name, self, vpc)
+    end
+
+    def load_into_config(config)
+      config.cf_add_resource resource_name, self
+      config.cf_add_resource(elastic_ip_association.resource_name, elastic_ip_association) if eip_name
+      super(config)
+    end
+
+    def resource_name
+      resource_style(name) + "EC2"
+    end
+
+    def fetchmergeprop(name)
+      s = {}
+      s.deep_merge!(template.fetchmergeprop(name)) if template
+      s.deep_merge!(self.send(name))
+      s
+    end
+
+    def fetchprop(name)
+      s = self.send(name)
+      return s unless s.nil?
+      return template.fetchprop(name) unless template.nil?
+      nil
+    end
+
+    def fetchunionprop(name)
+      if template
+        p = template.fetchunionprop(name).clone
+        p.concat self.send(name)
+        p.uniq
+      else
+        self.send(name)
+      end
+    end
+
+    def default_tags
+      {"Name" => resource_style(name)}
+    end
+
+    def to_h
+      h = {
+        "Type" => "AWS::EC2::Instance",
+        "Properties" => {
+          "ImageId" => DrawCloud.ref(fetchprop :image_id),
+          "InstanceType" => DrawCloud.ref(fetchprop :instance_type),
+        }
+      }
+      p = h["Properties"]
+      %w(availability_zone disable_api_termination key_name
+         monitoring placement_group_name private_ip_address
+         source_dest_check subnet_id user_data).each do |prop_str|
+        prop = prop_str.intern
+        p[resource_style(prop)] = DrawCloud.ref(fetchprop(prop)) unless fetchprop(prop).nil?
+      end
+      p["Tags"] = hash_to_tag_array(default_tags.merge(fetchmergeprop(:tags)))
+      h["DependsOn"] = DrawCloud.resource_name(fetchprop(:depends_on)) unless fetchprop(:depends_on).nil?
+      h["Metadata"] = DrawCloud.ref(fetchmergeprop(:metadata)) unless fetchmergeprop(:metadata).empty?
+
+      enis = fetchunionprop(:network_interfaces)
+      p["NetworkInterfaces"] = enis.enum_for(:each_with_index).collect do |e, i|
+        { "NetworkInterfaceId" => DrawCloud.ref(e),
+          "DeviceIndex" => (i+1).to_s }
+      end unless enis.empty?
+
+      security_groups = fetchunionprop(:security_groups)
+      vpc_security_groups = security_groups.find_all(&:vpc)
+      regular_security_groups = security_groups.reject(&:vpc)
+      p["SecurityGroups"] = regular_security_groups.collect {|s| DrawCloud.ref(s) } unless regular_security_groups.empty?
+      p["SecurityGroupIds"] = vpc_security_groups.collect {|s| DrawCloud.ref(s) } unless vpc_security_groups.empty?
+      h
+    end
+  end
+end

data/lib/draw_cloud/ec2_instance.rb~

@@ -0,0 +1,78 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+# License::   All rights reserved
+#
+
+module DrawCloud
+  class EC2Instance
+    include Base
+
+    attr_accessor(:availability_zone,
+                  :disable_api_termination,
+                  :image_id,
+                  :instance_type,
+                  :key_name,
+                  :monitoring,
+                  :placement_group_name,
+                  :private_ip_address,
+                  :security_groups,
+                  :security_group_ids,
+                  :source_dest_check,
+                  :subnet_id,
+                  :tags,
+                  :template)
+    def initialize(name, options={}, &block)
+      @name = name
+      @security_groups = []
+      @security_group_ids = []
+      @tags = []
+      @template = options.fetch(:template, nil)
+      self.instance_exec(self, &block) if block
+    end
+
+    def load_into_config(config)
+      config.cf_add_resource resource_name, self
+    end
+
+    def resource_name
+      resource_style(name) + "EC2"
+    end
+
+    def fetchprop(name)
+      s = self.send(name)
+      return s unless s.nil?
+      return template.fetchprop(name) unless template.nil?
+      nil
+    end
+
+    def fetchunionprop(name)
+      if template
+        p = template.send(name).clone
+        p.concat self.send(name)
+      else
+        self.send(name)
+      end
+    end
+
+    def to_h
+      h = {
+        "Type" => "AWS::EC2::Instance",
+        "Properties" => {
+          "ImageId" => DrawCloud.ref(fetchprop :image_id),
+          "InstanceType" => DrawCloud.ref(fetchprop :instance_type),
+        }
+      }
+      p = h["Properties"]
+      %s(availability_zone disable_api_termination key_name
+         monitoring placement_group_name private_ip_address
+         source_dest_check subnet_id).each do |prop|
+        p[resource_style(prop)] = DrawCloud.ref(fetchprop(prop)) unless fetchprop(prop).nil?
+      end
+      p["Tags"] = fetchunionprop(:tags) unless fetchunionprop(:tags).empty?
+      p["SecurityGroups"] = fetchunionprop(:security_groups).collect {|s| DrawCloud.ref(s) } unless fetchunionprop(:security_groups).empty?
+      p["SecurityGroupIds"] = fetchunionprop(:security_group_ids).collect {|s| DrawCloud.ref(s) } unless fetchunionprop(:security_group_ids).empty?
+      h
+    end
+  end
+end

data/lib/draw_cloud/ec2_instance_template.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this work except in compliance with the License. You may
+# obtain a copy of the License in the LICENSE file, or 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 DrawCloud
+  class EC2InstanceTemplate < EC2Instance
+    def initialize(name, options={}, &block)
+      super(name, options, &block)
+    end
+
+    def load_into_config(config)
+      # template - do NOT add to config
+    end
+  end
+end

data/lib/draw_cloud/ec2_instance_template.rb~

@@ -0,0 +1,36 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+# License::   All rights reserved
+#
+
+module DrawCloud
+  class EC2Instance
+    include Base
+
+    attr_accessor(:availability_zone,
+                  :disable_api_termination,
+                  :image_id,
+                  :instance_type,
+                  :key_name,
+                  :monitoring,
+                  :placement_group_name,
+                  :private_ip_address,
+                  :security_groups,
+                  :security_group_ids,
+                  :source_dest_check,
+                  :subnet_id,
+                  :tags)
+    def initialize(name, options={}, &block)
+      @name = name
+      @security_groups = []
+      @security_group_ids = []
+      @tags
+      self.instance_exec(self, &block) if block
+    end
+
+    def load_into_config(config)
+      # template!
+    end
+  end
+end

data/lib/draw_cloud/elastic_ip.rb

@@ -0,0 +1,97 @@
+# encoding: utf-8
+#
+# Copyright:: Copyright (c) 2012, SweetSpot Diabetes Care, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you
+# may not use this work except in compliance with the License. You may
+# obtain a copy of the License in the LICENSE file, or 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 DrawCloud
+  class ElasticIp < Base
+    class ElasticIpAssociation
+      attr_accessor(:elastic_ip, :instance, :vpc)
+      def initialize(elastic_ip, instance, vpc)
+        @elastic_ip = elastic_ip
+        @instance = instance
+        @vpc = vpc
+      end
+
+      def resource_name
+        elastic_ip.resource_name + "Association"
+      end
+
+      def to_h
+        h = {
+          "Type" => "AWS::EC2::EIPAssociation",
+          "Properties" => { },
+        }
+        if instance.ec2_instance
+          h["Properties"]["InstanceId"] = DrawCloud.ref(instance)
+        elsif instance.network_interface
+          h["Properties"]["NetworkInterfaceId"] = DrawCloud.ref(instance)
+        else
+          raise ArgumentError, "Unknown instance or network interface type #{instance.inspect}"
+        end
+
+        if vpc
+          case elastic_ip
+          when DrawCloud::ElasticIp
+            h["Properties"]["AllocationId"] = DrawCloud.ref(elastic_ip[:allocation_id])
+          else
+            h["Properties"]["AllocationId"] = DrawCloud.ref(elastic_ip)
+          end
+        else
+          h["Properties"]["EIP"] = DrawCloud.ref(elastic_ip)
+        end
+        h
+      end
+    end
+
+    attr_accessor(:name, :instance_id, :domain)
+    def initialize(name, options={}, &block)
+      @name = name
+      @domain = options.fetch(:domain, nil)
+      @instance_id = options.fetch(:instance_id, nil)
+      super(options, &block)
+    end
+
+    def elastic_ip
+      self
+    end
+
+    def association
+      ElasticIpAssociation.new(self, instance_id, vpc)
+    end
+
+    def load_into_config(config)
+      config.cf_add_resource resource_name, self
+      config.cf_add_resource(association.resource_name, association) if instance_id
+      super(config)
+    end
+
+    def resource_name
+      resource_style(name) + "EIP"
+    end
+
+    def to_h
+      h = {
+        "Type" => "AWS::EC2::EIP",
+        "Properties" => {}
+      }
+      h["Properties"]["Domain"] = domain unless domain.nil?
+      h["Properties"]["Domain"] = "vpc" if (domain.nil? && vpc)
+      add_standard_properties(h)
+    end
+  end
+end
+