Db instance

lib/harp-runtime/cloud/cloud_mutator.rb

@@ -17,6 +17,7 @@ def initialize(options)
         @secret = options[:secret]
         @cloud_type = options[:cloud_type]
         @compute = nil
+        @rds = nil
         @mock = (options.include? :mock) ? true : false
       end
 
@@ -31,6 +32,15 @@ def establish_connect(resource_type)
           @compute = Fog::Compute.new(:provider => 'AWS',
             :aws_access_key_id => @access, :aws_secret_access_key => @secret)
           return @compute
+        elsif Harp::Resources::RESOURCES_RDS.include? resource_type.class
+          if ! @rds.nil?
+            return @rds
+          end
+          if @mock
+            Fog.mock!
+          end
+          @rds = Fog::AWS::RDS.new({:aws_access_key_id => @access, :aws_secret_access_key => @secret})
+          return @rds
         end
       end
 

lib/harp-runtime/models/rds.rb

@@ -0,0 +1,8 @@
+require 'data_mapper'
+require 'harp-runtime/models/base'
+
+class DBSecurityGroup < HarpResource
+end
+
+class DBInstance < HarpResource
+end

lib/harp-runtime/resources/available_resources.rb

@@ -13,6 +13,9 @@ module Resources
 
     # Set to contain all compute resources
     RESOURCES_COMPUTE = Set.new
+
+    # Set to contain all rds resources
+    RESOURCES_RDS = Set.new
 
     class AvailableResource < Fog::Model
 
@@ -123,6 +126,7 @@ def output_token(args={})
 end
 
 require "harp-runtime/resources/compute/types"
+require "harp-runtime/resources/rds/types"
 
 class String
   def underscore

lib/harp-runtime/resources/rds/db_instance.rb

@@ -0,0 +1,74 @@
+require 'set'
+require 'fog/core/model'
+require 'harp-runtime/models/rds'
+
+module Harp
+  module Resources
+
+    class DBInstance < AvailableResource
+
+      include Harp::Resources
+
+      attribute :id,                           :aliases => 'DBInstanceIdentifier'
+      attribute :engine,                       :aliases => 'Engine'
+      attribute :engine_version,               :aliases => 'EngineVersion'
+      attribute :state,                        :aliases => 'DBInstanceStatus'
+      attribute :allocated_storage,            :aliases => 'AllocatedStorage'
+      attribute :availability_zone ,           :aliases => 'AvailabilityZone'
+      attribute :flavor_id,                    :aliases => 'DBInstanceClass'
+      attribute :endpoint,                     :aliases => 'Endpoint'
+      attribute :read_replica_source,          :aliases => 'ReadReplicaSourceDBInstanceIdentifier'
+      attribute :read_replica_identifiers,     :aliases => 'ReadReplicaDBInstanceIdentifiers'
+      attribute :master_username,              :aliases => 'MasterUsername'
+      attribute :multi_az,                     :aliases => 'MultiAZ'
+      attribute :created_at,                   :aliases => 'InstanceCreateTime'
+      attribute :last_restorable_time,         :aliases => 'LatestRestorableTime'
+      attribute :auto_minor_version_upgrade,   :aliases => 'AutoMinorVersionUpgrade'
+      attribute :pending_modified_values,      :aliases => 'PendingModifiedValues'
+      attribute :preferred_backup_window,      :aliases => 'PreferredBackupWindow'
+      attribute :preferred_maintenance_window, :aliases => 'PreferredMaintenanceWindow'
+      attribute :db_name,                      :aliases => 'DBName'
+      attribute :db_security_groups,           :aliases => 'DBSecurityGroups'
+      attribute :db_parameter_groups,          :aliases => 'DBParameterGroups'
+      attribute :backup_retention_period,      :aliases => 'BackupRetentionPeriod'
+      attribute :license_model,                :aliases => 'LicenseModel'
+      attribute :db_subnet_group_name,         :aliases => 'DBSubnetGroupName'
+      attribute :publicly_accessible,          :aliases => 'PubliclyAccessible'
+      attribute :vpc_security_groups,          :aliases => 'VpcSecurityGroups'
+      attribute :password
+
+      attr_accessor :parameter_group_name, :security_group_names, :port
+
+      register_resource :db_instance, RESOURCES_RDS
+
+      # Only keeping a few properties, simplest define keeps.
+      @keeps = /^id$|^name$/
+
+      @output = false
+
+      def self.persistent_type()
+        ::DBInstance
+      end
+
+      def create(service)
+        create_attribs = self.attribs[:attributes]
+        create_attribs[:tags] = {"Name" => @name}
+        create_attribs[:id] = @name
+        instance = service.servers.create(create_attribs)
+        @id = instance.id
+        return instance
+      end
+
+      def destroy(service)
+        destroy_attribs = self.attribs
+        if @id
+          instance = service.servers.destroy(destroy_attribs)
+        else
+          puts "No ID set, cannot delete."
+        end
+        return instance
+      end
+
+    end
+  end
+end

lib/harp-runtime/resources/rds/db_security_group.rb

@@ -0,0 +1,50 @@
+require 'set'
+require 'fog/core/model'
+require 'harp-runtime/models/rds'
+
+module Harp
+  module Resources
+
+    class DBSecurityGroup < AvailableResource
+
+      include Harp::Resources
+
+      attribute  :id,                   :aliases => 'DBSecurityGroupName'
+      attribute  :description,          :aliases => 'DBSecurityGroupDescription'
+      attribute  :ec2_security_groups,  :aliases => 'EC2SecurityGroups'
+      attribute  :ip_ranges,            :aliases => 'IPRanges'
+      attribute  :owner_id,             :aliases => 'OwnerId'
+
+      register_resource :db_security_group, RESOURCES_RDS
+
+      # Only keeping a few properties, simplest define keeps.
+      @keeps = /^id$|description/
+
+      @output = false
+
+      def self.persistent_type()
+        ::DBSecurityGroup
+      end
+
+      def create(service)
+        create_attribs = self.attribs[:attributes]
+        create_attribs[:tags] = {"Name" => @name}
+        create_attribs[:id] = @name
+        security_group = service.security_groups.create(create_attribs)
+        @id = security_group.id
+        return security_group
+      end
+
+      def destroy(service)
+        destroy_attribs = self.attribs
+        if @id
+          security_group = service.security_groups.destroy(destroy_attribs)
+        else
+          puts "No ID set, cannot delete."
+        end
+        return security_group
+      end
+
+    end
+  end
+end

lib/harp-runtime/resources/rds/types.rb

@@ -0,0 +1,11 @@
+require 'set'
+
+require 'harp-runtime/resources/rds/db_security_group'
+require 'harp-runtime/resources/rds/db_instance'
+
+module Harp
+  module Resources
+    extend self
+  end
+end
+

spec/compute_cloud_spec.rb

@@ -45,5 +45,4 @@
 
     expect(result.description).to eq("A web security group")
   end
-
 end

spec/rds_cloud_spec.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/spec_helper'
+require "rubygems"
+require "harp_runtime"
+require "harp-runtime/cloud/cloud_mutator"
+
+db_security_group_resource = {
+  "type" => "Std::DBSecurityGroup",
+  "description" => "A web db security group"
+}
+
+db_instance_resource = {
+  "type" => "Std::DBInstance",
+  "engine" => "MySQL",
+  "allocated_storage" => "5",
+  "master_username" => "masterdbusername",
+  "password" => "masterpassword"
+}
+
+describe Harp::Cloud::CloudMutator, "#create" do
+  it "creates a db security group" do
+    context = {}
+    context[:cloud_type] = :aws # for the moment, assume AWS cloud
+    context[:mock] = true
+    context[:debug] = true
+    context[:access] = "test"
+    context[:secret] = "test"
+    mutator = Harp::Cloud::CloudMutator.new(context)
+
+    result = mutator.create("test_db_sg1", db_security_group_resource)
+    expect(result.class).to eq(DBSecurityGroup)
+    expect(result.name).to eq("test_db_sg1")
+
+    expect(result.description).to eq("A web db security group")
+  end
+  it "creates a db instance" do
+    context = {}
+    context[:cloud_type] = :aws # for the moment, assume AWS cloud
+    context[:mock] = true
+    context[:debug] = true
+    context[:access] = "test"
+    context[:secret] = "test"
+    mutator = Harp::Cloud::CloudMutator.new(context)
+
+    result = mutator.create("test_db_db1", db_instance_resource)
+    expect(result.class).to eq(DBInstance)
+    expect(result.name).to eq("test_db_db1")
+  end
+end

spec/spec_helper.rb

@@ -21,6 +21,7 @@
 
 require 'harp-runtime/models/base'
 require 'harp-runtime/models/compute'
+require 'harp-runtime/models/rds'
 
 DataMapper.finalize
 DataMapper.auto_migrate!