steamcannon-deltacloud-core 0.0.7.1-java

data/lib/deltacloud/base_driver/features.rb

@@ -0,0 +1,173 @@
+require 'deltacloud/validation'
+
+# Add advertising of optional features to the base driver
+module Deltacloud
+
+  class FeatureError < StandardError; end
+  class DuplicateFeatureDeclError < FeatureError; end
+  class UndeclaredFeatureError < FeatureError; end
+
+  class BaseDriver
+
+    # An operation on a collection like cretae or show. Features
+    # can add parameters to operations
+    class Operation
+      attr_reader :name
+
+      include Deltacloud::Validation
+
+      def initialize(name, &block)
+        @name = name
+        @params = {}
+        instance_eval &block
+      end
+    end
+
+    # The declaration of a feature, defines what operations
+    # are modified by it
+    class FeatureDecl
+      attr_reader :name, :operations
+
+      def initialize(name, &block)
+        @name = name
+        @operations = []
+        instance_eval &block
+      end
+
+      def description(text=nil)
+        @description = text if text
+        @description
+      end
+
+      # Add a new operation or modify an existing one through BLOCK
+      def operation(name, &block)
+        unless op = @operations.find { |op| op.name == name }
+          op = Operation.new(name, &block)
+          @operations << op
+        else
+          op.instance_eval(&block) if block_given?
+        end
+        op
+      end
+    end
+
+    # A specific feature enabled by a driver (see +feature+)
+    class Feature
+      attr_reader :decl
+
+      def initialize(decl, &block)
+        @decl = decl
+        instance_eval &block if block_given?
+      end
+
+      def name
+        decl.name
+      end
+
+      def operations
+        decl.operations
+      end
+
+      def description
+        decl.description
+      end
+    end
+
+    def self.feature_decls
+      @@feature_decls ||= {}
+    end
+
+    def self.feature_decl_for(collection, name)
+      decls = feature_decls[collection]
+      if decls
+        decls.find { |dcl| dcl.name == name }
+      else
+        nil
+      end
+    end
+
+    # Declare a new feature
+    def self.declare_feature(collection, name, &block)
+      feature_decls[collection] ||= []
+      raise DuplicateFeatureDeclError if feature_decl_for(collection, name)
+      feature_decls[collection] << FeatureDecl.new(name, &block)
+    end
+
+    def self.features
+      @@features ||= {}
+    end
+
+    # Declare in a driver that it supports a specific feature
+    #
+    # The same feature can be declared multiple times in a driver, so that
+    # it can be changed successively by passing in different blocks.
+    def self.feature(collection, name, &block)
+      features[collection] ||= []
+      if f = features[collection].find { |f| f.name == name }
+        f.instance_eval(&block) if block_given?
+        return f
+      end
+      unless decl = feature_decl_for(collection, name)
+        raise UndeclaredFeatureError, "No feature #{name} for #{collection}"
+      end
+      features[collection] << Feature.new(decl, &block)
+    end
+
+    def features(collection)
+      self.class.features[collection] || []
+    end
+
+    #
+    # Declaration of optional features
+    #
+    declare_feature :images,  :owner_id do
+      description "Filter images using owner id"
+      operation :index do
+        param :owner_id,  :string,  :optional,  nil,  "Owner ID"
+      end
+    end
+
+    declare_feature :instances, :user_name do
+      description "Accept a user-defined name on instance creation"
+      operation :create do
+        param :name, :string, :optional, nil,
+        "The user-defined name"
+      end
+    end
+
+    declare_feature :instances, :user_data do
+      description "Make user-defined data available on a special webserver"
+      operation :create do
+        param :user_data, :string, :optional, nil,
+        "Base64 encoded user data will be published to internal webserver"
+      end
+    end
+
+    declare_feature :instances, :authentication_key do
+      operation :create do
+        param :keyname, :string,  :optional, nil
+        "EC2 key authentification method"
+      end
+      operation :show do
+      end
+    end
+
+    declare_feature :instances, :authentication_password do
+      operation :create do
+        param :password, :string, :optional
+      end
+    end
+
+    declare_feature :instances, :hardware_profiles do
+      description "Size instances according to changes to a hardware profile"
+      # The parameters are filled in from the hardware profiles
+    end
+
+    declare_feature :buckets, :bucket_location do
+      description "Take extra location parameter for Bucket creation (e.g. S3, 'eu' or 'us-west-1')"
+      operation :create do
+        param :location, :string, :optional
+      end
+    end
+  end
+end

data/lib/deltacloud/base_driver/mock_driver.rb

@@ -0,0 +1,58 @@
+require 'deltacloud/method_serializer'
+
+# Create 'mock' version of original driver client/gem:
+
+module Mock
+
+  class S3 < RightAws::S3
+    include MethodSerializer::Cache
+
+    def self.cached_methods
+      [
+        :buckets
+      ]
+    end
+
+    MethodSerializer::Cache::wrap_methods(self, :cache_dir => File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'tests', 'ec2', 'support'))
+  end
+
+  class EC2 < AWS::EC2::Base
+
+    include MethodSerializer::Cache
+
+    def self.cached_methods
+      [
+        :describe_images,
+        :describe_availability_zones,
+        :describe_keypairs,
+        :create_keypair,
+        :run_instances,
+        :describe_instances,
+        :reboot_instances,
+        :terminate_instances,
+        :delete_keypair
+      ]
+    end
+
+    MethodSerializer::Cache::wrap_methods(self, :cache_dir => File.join(File.dirname(__FILE__), '..', '..', '..', '..', 'tests', 'ec2', 'support'))
+  end
+end
+
+
+# Replace original client with mock client
+Deltacloud::Drivers::EC2::EC2Driver.class_eval do
+  alias_method :original_new_client, :new_client
+  alias_method :original_s3_client, :s3_client
+
+  def new_client(credentials, opts={})
+    Mock::EC2.new(
+      :access_key_id => credentials.user,
+      :secret_access_key => credentials.password
+    )
+  end
+
+  def s3_client(credentials)
+    Mock::S3.new(credentials.user, credentials.password)
+  end
+
+end

data/lib/deltacloud/base_driver.rb

@@ -0,0 +1,20 @@
+#
+# Copyright (C) 2009  Red Hat, Inc.
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.  The
+# ASF licenses this file to you 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 'deltacloud/base_driver/base_driver'
+require 'deltacloud/base_driver/features'

data/lib/deltacloud/drivers/azure/azure_driver.rb

@@ -0,0 +1,127 @@
+#
+# Copyright (C) 2010  Red Hat, Inc.
+#
+# Licensed to the Apache Software Foundation (ASF) under one or more
+# contributor license agreements.  See the NOTICE file distributed with
+# this work for additional information regarding copyright ownership.  The
+# ASF licenses this file to you 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.
+
+#Windows Azure (WAZ) gem at http://github.com/johnnyhalife/waz-storage
+require 'waz-blobs'
+require 'deltacloud/base_driver'
+module Deltacloud
+  module Drivers
+    module Azure
+
+class AzureDriver < Deltacloud::BaseDriver
+
+  def supported_collections; [:buckets]
+  end
+
+#--
+# Buckets
+#--
+  def buckets(credentials, opts)
+    buckets = []
+    azure_connect(credentials)
+    safely do
+      WAZ::Blobs::Container.list.each do |waz_container|
+        buckets << convert_container(waz_container)
+      end
+    end
+    buckets = filter_on(buckets, :id, opts)
+  end
+
+#--
+# Create bucket
+#--
+  def create_bucket(credentials, name, opts)
+    bucket = nil
+    azure_connect(credentials)
+    safely do
+      waz_container = WAZ::Blobs::Container.create(name)
+      bucket = convert_container(waz_container)
+    end
+    bucket
+  end
+
+#--
+# Delete bucket
+#--
+  def delete_bucket(credentials, name, opts)
+    azure_connect(credentials)
+    safely do
+      WAZ::Blobs::Container.find(name).destroy!
+    end
+  end
+
+#--
+# Blobs
+#--
+  def blobs(credentials, opts)
+    blob_list = []
+    azure_connect(credentials)
+    safely do
+      the_bucket = WAZ::Blobs::Container.find(opts['bucket'])
+      the_bucket.blobs.each do |waz_blob|
+        blob_list << convert_blob(waz_blob)
+      end
+    end
+    blob_list = filter_on(blob_list, :id, opts)
+    blob_list
+  end
+
+  def blob_data(credentials, bucket_id, blob_id, opts)
+    azure_connect(credentials)
+    # WAZ get blob data methods cant accept blocks for 'streaming'... FIXME
+      yield WAZ::Blobs::Container.find(bucket_id)[blob_id].value
+  end
+
+  private
+
+  def azure_connect(credentials)
+    options = {:account_name => credentials.user, :access_key => credentials.password}
+    safely do
+      WAZ::Storage::Base.establish_connection!(options)
+    end
+  end
+
+  def convert_container(waz_container)
+    blob_list = []
+    waz_container.blobs.each do |blob|
+      blob_list << blob.name
+    end
+    Bucket.new({ :id => waz_container.name,
+                    :name => waz_container.name,
+                    :size => blob_list.size,
+                    :blob_list => blob_list
+                  })
+  end
+
+  def convert_blob(waz_blob)
+    url = waz_blob.url.split('/')
+    bucket = url[url.length-2] #FIXME
+    Blob.new({   :id => waz_blob.name,
+                 :bucket => bucket,
+                 :content_length => waz_blob.metadata[:content_length],
+                 :content_type => waz_blob.metadata[:content_type],
+                 :last_modified => waz_blob.metadata[:last_modified]
+              })
+  end
+
+
+end
+
+    end #module Azure
+  end #module Drivers
+end #module Deltacloud