knife-azure 1.0.0

data/Gemfile

@@ -0,0 +1,22 @@
+source "http://rubygems.org"
+# Add dependencies required to use your gem here.
+# Example:
+#   gem "activesupport", ">= 2.3.5"
+
+gemspec
+  #gem "chef", "~> 0.10.8"
+
+# Add dependencies to develop your gem here.
+# Include everything needed to run rake, tests, features, etc.
+group :development do
+  gem "rspec", "~> 2.8.0"
+  gem "rake"
+  gem "rdoc", "~> 3.12"
+  gem "bundler"
+  gem "guard-rspec"
+  gem "libnotify"
+  gem "rubygems-bundler", "~> 0.2.8"
+  gem "interactive_editor"
+  gem "equivalent-xml", "~> 0.2.9"
+end
+

data/Guardfile

@@ -0,0 +1,8 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+guard :rspec, :version => 2 do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/(.+)\.rb$})    # { |m| "spec/lib/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+end

data/LICENSE

@@ -0,0 +1,201 @@
+                              Apache License
+                        Version 2.0, January 2004
+                     http://www.apache.org/licenses/
+
+TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+1. Definitions.
+
+   "License" shall mean the terms and conditions for use, reproduction,
+   and distribution as defined by Sections 1 through 9 of this document.
+
+   "Licensor" shall mean the copyright owner or entity authorized by
+   the copyright owner that is granting the License.
+
+   "Legal Entity" shall mean the union of the acting entity and all
+   other entities that control, are controlled by, or are under common
+   control with that entity. For the purposes of this definition,
+   "control" means (i) the power, direct or indirect, to cause the
+   direction or management of such entity, whether by contract or
+   otherwise, or (ii) ownership of fifty percent (50%) or more of the
+   outstanding shares, or (iii) beneficial ownership of such entity.
+
+   "You" (or "Your") shall mean an individual or Legal Entity
+   exercising permissions granted by this License.
+
+   "Source" form shall mean the preferred form for making modifications,
+   including but not limited to software source code, documentation
+   source, and configuration files.
+
+   "Object" form shall mean any form resulting from mechanical
+   transformation or translation of a Source form, including but
+   not limited to compiled object code, generated documentation,
+   and conversions to other media types.
+
+   "Work" shall mean the work of authorship, whether in Source or
+   Object form, made available under the License, as indicated by a
+   copyright notice that is included in or attached to the work
+   (an example is provided in the Appendix below).
+
+   "Derivative Works" shall mean any work, whether in Source or Object
+   form, that is based on (or derived from) the Work and for which the
+   editorial revisions, annotations, elaborations, or other modifications
+   represent, as a whole, an original work of authorship. For the purposes
+   of this License, Derivative Works shall not include works that remain
+   separable from, or merely link (or bind by name) to the interfaces of,
+   the Work and Derivative Works thereof.
+
+   "Contribution" shall mean any work of authorship, including
+   the original version of the Work and any modifications or additions
+   to that Work or Derivative Works thereof, that is intentionally
+   submitted to Licensor for inclusion in the Work by the copyright owner
+   or by an individual or Legal Entity authorized to submit on behalf of
+   the copyright owner. For the purposes of this definition, "submitted"
+   means any form of electronic, verbal, or written communication sent
+   to the Licensor or its representatives, including but not limited to
+   communication on electronic mailing lists, source code control systems,
+   and issue tracking systems that are managed by, or on behalf of, the
+   Licensor for the purpose of discussing and improving the Work, but
+   excluding communication that is conspicuously marked or otherwise
+   designated in writing by the copyright owner as "Not a Contribution."
+
+   "Contributor" shall mean Licensor and any individual or Legal Entity
+   on behalf of whom a Contribution has been received by Licensor and
+   subsequently incorporated within the Work.
+
+2. Grant of Copyright License. Subject to the terms and conditions of
+   this License, each Contributor hereby grants to You a perpetual,
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+   copyright license to reproduce, prepare Derivative Works of,
+   publicly display, publicly perform, sublicense, and distribute the
+   Work and such Derivative Works in Source or Object form.
+
+3. Grant of Patent License. Subject to the terms and conditions of
+   this License, each Contributor hereby grants to You a perpetual,
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+   (except as stated in this section) patent license to make, have made,
+   use, offer to sell, sell, import, and otherwise transfer the Work,
+   where such license applies only to those patent claims licensable
+   by such Contributor that are necessarily infringed by their
+   Contribution(s) alone or by combination of their Contribution(s)
+   with the Work to which such Contribution(s) was submitted. If You
+   institute patent litigation against any entity (including a
+   cross-claim or counterclaim in a lawsuit) alleging that the Work
+   or a Contribution incorporated within the Work constitutes direct
+   or contributory patent infringement, then any patent licenses
+   granted to You under this License for that Work shall terminate
+   as of the date such litigation is filed.
+
+4. Redistribution. You may reproduce and distribute copies of the
+   Work or Derivative Works thereof in any medium, with or without
+   modifications, and in Source or Object form, provided that You
+   meet the following conditions:
+
+   (a) You must give any other recipients of the Work or
+       Derivative Works a copy of this License; and
+
+   (b) You must cause any modified files to carry prominent notices
+       stating that You changed the files; and
+
+   (c) You must retain, in the Source form of any Derivative Works
+       that You distribute, all copyright, patent, trademark, and
+       attribution notices from the Source form of the Work,
+       excluding those notices that do not pertain to any part of
+       the Derivative Works; and
+
+   (d) If the Work includes a "NOTICE" text file as part of its
+       distribution, then any Derivative Works that You distribute must
+       include a readable copy of the attribution notices contained
+       within such NOTICE file, excluding those notices that do not
+       pertain to any part of the Derivative Works, in at least one
+       of the following places: within a NOTICE text file distributed
+       as part of the Derivative Works; within the Source form or
+       documentation, if provided along with the Derivative Works; or,
+       within a display generated by the Derivative Works, if and
+       wherever such third-party notices normally appear. The contents
+       of the NOTICE file are for informational purposes only and
+       do not modify the License. You may add Your own attribution
+       notices within Derivative Works that You distribute, alongside
+       or as an addendum to the NOTICE text from the Work, provided
+       that such additional attribution notices cannot be construed
+       as modifying the License.
+
+   You may add Your own copyright statement to Your modifications and
+   may provide additional or different license terms and conditions
+   for use, reproduction, or distribution of Your modifications, or
+   for any such Derivative Works as a whole, provided Your use,
+   reproduction, and distribution of the Work otherwise complies with
+   the conditions stated in this License.
+
+5. Submission of Contributions. Unless You explicitly state otherwise,
+   any Contribution intentionally submitted for inclusion in the Work
+   by You to the Licensor shall be under the terms and conditions of
+   this License, without any additional terms or conditions.
+   Notwithstanding the above, nothing herein shall supersede or modify
+   the terms of any separate license agreement you may have executed
+   with Licensor regarding such Contributions.
+
+6. Trademarks. This License does not grant permission to use the trade
+   names, trademarks, service marks, or product names of the Licensor,
+   except as required for reasonable and customary use in describing the
+   origin of the Work and reproducing the content of the NOTICE file.
+
+7. Disclaimer of Warranty. Unless required by applicable law or
+   agreed to in writing, Licensor provides the Work (and each
+   Contributor provides its Contributions) on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+   implied, including, without limitation, any warranties or conditions
+   of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+   PARTICULAR PURPOSE. You are solely responsible for determining the
+   appropriateness of using or redistributing the Work and assume any
+   risks associated with Your exercise of permissions under this License.
+
+8. Limitation of Liability. In no event and under no legal theory,
+   whether in tort (including negligence), contract, or otherwise,
+   unless required by applicable law (such as deliberate and grossly
+   negligent acts) or agreed to in writing, shall any Contributor be
+   liable to You for damages, including any direct, indirect, special,
+   incidental, or consequential damages of any character arising as a
+   result of this License or out of the use or inability to use the
+   Work (including but not limited to damages for loss of goodwill,
+   work stoppage, computer failure or malfunction, or any and all
+   other commercial damages or losses), even if such Contributor
+   has been advised of the possibility of such damages.
+
+9. Accepting Warranty or Additional Liability. While redistributing
+   the Work or Derivative Works thereof, You may choose to offer,
+   and charge a fee for, acceptance of support, warranty, indemnity,
+   or other liability obligations and/or rights consistent with this
+   License. However, in accepting such obligations, You may act only
+   on Your own behalf and on Your sole responsibility, not on behalf
+   of any other Contributor, and only if You agree to indemnify,
+   defend, and hold each Contributor harmless for any liability
+   incurred by, or claims asserted against, such Contributor by reason
+   of your accepting any such warranty or additional liability.
+
+END OF TERMS AND CONDITIONS
+
+APPENDIX: How to apply the Apache License to your work.
+
+   To apply the Apache License to your work, attach the following
+   boilerplate notice, with the fields enclosed by brackets "[]"
+   replaced with your own identifying information. (Don't include
+   the brackets!)  The text should be enclosed in the appropriate
+   comment syntax for the file format. We also recommend that a
+   file or class name and description of purpose be included on the
+   same "printed page" as the copyright notice for easier
+   identification within third-party archives.
+
+Copyright [yyyy] [name of copyright owner]
+
+Licensed 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.

data/Rakefile

@@ -0,0 +1,51 @@
+# encoding: utf-8
+
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'rake'
+
+spec = eval(File.read("knife-azure.gemspec"))
+
+require 'rubygems/package_task'
+
+Gem::PackageTask.new(spec) do |pkg|
+  pkg.gem_spec = spec
+end
+
+require 'rspec/core'
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec) do |spec|
+  spec.pattern = FileList['spec/unit/**/*_spec.rb']
+end
+
+RSpec::Core::RakeTask.new(:functional) do |spec|
+  spec.pattern = FileList['spec/functional/**/*_test.rb']
+end
+
+RSpec::Core::RakeTask.new(:integration) do |spec|
+  spec.pattern = FileList['spec/integration/**/*_test.rb']
+end
+
+RSpec::Core::RakeTask.new(:rcov) do |spec|
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :default => :spec
+
+require 'rdoc/task'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "knife-azure #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/knife-azure.gemspec

@@ -0,0 +1,106 @@
+$:.unshift(File.dirname(__FILE__) + '/lib')
+require 'knife-azure/version'
+
+Gem::Specification.new do |s|
+  s.name = "knife-azure"
+  s.version = Knife::Azure::VERSION
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Barry Davis"]
+  s.date = "2012-06-06"
+  s.summary = "A plugin to Opscode knife for creating instances on the Microsoft Azure platform"
+  s.description = s.summary
+  s.email = "oss@opscode.com"
+  s.licenses = ["Apache 2.0"]
+  s.extra_rdoc_files = [
+    "LICENSE"
+  ]
+  s.files = [
+    "Gemfile",
+    "Guardfile",
+    "LICENSE",
+    "Rakefile",
+    "knife-azure.gemspec",
+    "lib/azure/connection.rb",
+    "lib/azure/deploy.rb",
+    "lib/azure/disk.rb",
+    "lib/azure/host.rb",
+    "lib/azure/image.rb",
+    "lib/azure/rest.rb",
+    "lib/azure/role.rb",
+    "lib/azure/utility.rb",
+    "lib/chef/knife/azure_base.rb",
+    "lib/chef/knife/azure_image_list.rb",
+    "lib/chef/knife/azure_server_create.rb",
+    "lib/chef/knife/azure_server_delete.rb",
+    "lib/chef/knife/azure_server_describe.rb",
+    "lib/chef/knife/azure_server_list.rb",
+    "lib/knife-azure/version.rb",
+    "readme.rdoc",
+    "spec/functional/deploys_test.rb",
+    "spec/functional/host_test.rb",
+    "spec/functional/images_list_test.rb",
+    "spec/functional/role_test.rb",
+    "spec/integration/role_lifecycle_test.rb",
+    "spec/spec_helper.rb",
+    "spec/unit/assets/create_deployment.xml",
+    "spec/unit/assets/create_deployment_in_progress.xml",
+    "spec/unit/assets/create_host.xml",
+    "spec/unit/assets/create_role.xml",
+    "spec/unit/assets/list_deployments_for_service000.xml",
+    "spec/unit/assets/list_deployments_for_service001.xml",
+    "spec/unit/assets/list_deployments_for_service002.xml",
+    "spec/unit/assets/list_deployments_for_service003.xml",
+    "spec/unit/assets/list_disks.xml",
+    "spec/unit/assets/list_hosts.xml",
+    "spec/unit/assets/list_images.xml",
+    "spec/unit/assets/post_success.xml",
+    "spec/unit/deploys_list_spec.rb",
+    "spec/unit/disks_spec.rb",
+    "spec/unit/hosts_spec.rb",
+    "spec/unit/images_spec.rb",
+    "spec/unit/query_azure_mock.rb",
+    "spec/unit/roles_create_spec.rb",
+    "spec/unit/roles_list_spec.rb"
+  ]
+  s.homepage = "http://github.com/opscode/knife-azure"
+  s.require_paths = ["lib"]
+  s.rubygems_version = "1.8.23"
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<knife-azure>, [">= 0"])
+      s.add_development_dependency(%q<rspec>, ["~> 2.8.0"])
+      s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_development_dependency(%q<bundler>, [">= 0"])
+      s.add_development_dependency(%q<guard-rspec>, [">= 0"])
+      s.add_development_dependency(%q<libnotify>, [">= 0"])
+      s.add_development_dependency(%q<rubygems-bundler>, ["~> 0.2.8"])
+      s.add_development_dependency(%q<interactive_editor>, [">= 0"])
+      s.add_development_dependency(%q<equivalent-xml>, ["~> 0.2.9"])
+    else
+      s.add_dependency(%q<knife-azure>, [">= 0"])
+      s.add_dependency(%q<rspec>, ["~> 2.8.0"])
+      s.add_dependency(%q<rdoc>, ["~> 3.12"])
+      s.add_dependency(%q<bundler>, [">= 0"])
+      s.add_dependency(%q<guard-rspec>, [">= 0"])
+      s.add_dependency(%q<libnotify>, [">= 0"])
+      s.add_dependency(%q<rubygems-bundler>, ["~> 0.2.8"])
+      s.add_dependency(%q<interactive_editor>, [">= 0"])
+      s.add_dependency(%q<equivalent-xml>, ["~> 0.2.9"])
+    end
+  else
+    s.add_dependency(%q<knife-azure>, [">= 0"])
+    s.add_dependency(%q<rspec>, ["~> 2.8.0"])
+    s.add_dependency(%q<rdoc>, ["~> 3.12"])
+    s.add_dependency(%q<bundler>, [">= 0"])
+    s.add_dependency(%q<guard-rspec>, [">= 0"])
+    s.add_dependency(%q<libnotify>, [">= 0"])
+    s.add_dependency(%q<rubygems-bundler>, ["~> 0.2.8"])
+    s.add_dependency(%q<interactive_editor>, [">= 0"])
+    s.add_dependency(%q<equivalent-xml>, ["~> 0.2.9"])
+  end
+end
+

data/lib/azure/connection.rb

@@ -0,0 +1,75 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed 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 File.expand_path('../utility', __FILE__)
+require File.expand_path('../rest', __FILE__)
+require File.expand_path('../host', __FILE__)
+require File.expand_path('../deploy', __FILE__)
+require File.expand_path('../role', __FILE__)
+require File.expand_path('../disk', __FILE__)
+require File.expand_path('../image', __FILE__)
+
+class Azure
+  class Connection
+    include AzureAPI
+    attr_accessor :hosts, :rest, :images, :deploys, :roles, :disks
+    def initialize(params={})
+      @rest = Rest.new(params)
+      @hosts = Hosts.new(self)
+      @images = Images.new(self)
+      @deploys = Deploys.new(self)
+      @roles = Roles.new(self)
+      @disks = Disks.new(self)
+    end
+    def query_azure(service_name, verb = 'get', body = '')
+      Chef::Log.info 'calling ' + verb + ' ' + service_name
+      Chef::Log.debug body unless body == ''
+      response = @rest.query_azure(service_name, verb, body)
+      if response.code.to_i == 200
+        ret_val = Nokogiri::XML response.body
+      elsif response.code.to_i >= 201 && response.code.to_i <= 299
+        ret_val = wait_for_completion()
+      else
+        if response.body
+          ret_val = Nokogiri::XML response.body
+          Chef::Log.warn ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+        else
+          Chef::Log.warn 'http error: ' + response.code
+        end
+      end
+      ret_val
+    end
+    def wait_for_completion()
+      status = 'InProgress'
+      Chef::Log.info 'Waiting while status returns InProgress'
+      while status == 'InProgress'
+        response = @rest.query_for_completion()       
+        ret_val = Nokogiri::XML response.body
+        status = ret_val.at_css('Status').content
+        if status == 'InProgress'
+          print '.'
+          sleep(0.5)
+        elsif status == 'Succeeded'
+          Chef::Log.debug 'not InProgress : ' + ret_val.to_xml
+        else
+          Chef::Log.warn status + ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+        end
+      end
+      ret_val
+    end
+  end
+end

data/lib/azure/deploy.rb

@@ -0,0 +1,114 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed 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.
+#
+
+class Azure
+  class Deploys
+    def initialize(connection)
+      @connection=connection
+    end
+    def all
+      deploys = Array.new
+      hosts = @connection.hosts.all
+      hosts.each do |host|
+        deploy = Deploy.new(@connection)
+        deploy.retrieve(host.name)
+        unless deploy.name == nil
+          deploys << deploy
+        end
+      end
+      deploys
+    end
+    def find(hostedservicename)
+      deployName = nil
+      self.all.each do |deploy|
+        next unless deploy.hostedservicename == hostedservicename
+        deployName = deploy.name
+      end
+      deployName
+    end
+    def create(params)
+      unless @connection.hosts.exists(params[:hosted_service_name])
+        @connection.hosts.create(params)
+      end 
+      params['deploy_name'] = find(params[:hosted_service_name])
+      if params['deploy_name'] != nil
+        role = Role.new(@connection)
+        roleXML = role.setup(params)
+        ret_val = role.create(params, roleXML)
+      else
+        params['deploy_name'] = params[:hosted_service_name]
+        deploy = Deploy.new(@connection)
+        deployXML = deploy.setup(params)
+        ret_val = deploy.create(params, deployXML)
+      end
+      if ret_val.css('Error Code').length > 0
+          Chef::Log.fatal 'Unable to create role:' + ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+          exit 1
+      end
+      @connection.roles.find(params[:role_name])
+    end
+    def delete(rolename)
+    end
+  end
+
+  class Deploy
+    include AzureUtility
+    attr_accessor :connection, :name, :status, :url, :roles, :hostedservicename
+    def initialize(connection)
+      @connection = connection
+    end
+    def retrieve(hostedservicename)
+      @hostedservicename = hostedservicename
+      deployXML = @connection.query_azure("hostedservices/#{hostedservicename}/deploymentslots/Production")
+      if deployXML.at_css('Deployment Name') != nil
+        @name = xml_content(deployXML, 'Deployment Name')
+        @status = xml_content(deployXML,'Deployment Status')
+        @url = xml_content(deployXML, 'Deployment Url')
+        @roles = Array.new
+        rolesXML = deployXML.css('Deployment RoleInstanceList RoleInstance')
+        rolesXML.each do |roleXML|
+          role = Role.new(@connection)
+          role.parse(roleXML, hostedservicename, @name)
+          @roles << role
+        end
+      end
+    end
+    def setup(params)
+      role = Role.new(@connection)
+      roleXML = role.setup(params)
+      #roleXML = Nokogiri::XML role.setup(params)
+      builder = Nokogiri::XML::Builder.new do |xml|
+        xml.Deployment(
+          'xmlns'=>'http://schemas.microsoft.com/windowsazure',
+          'xmlns:i'=>'http://www.w3.org/2001/XMLSchema-instance' 
+        ) {
+          xml.Name params['deploy_name']
+          xml.DeploymentSlot 'Production'
+          xml.Label Base64.encode64(params['deploy_name']).strip
+          xml.RoleList { xml.Role('i:type'=>'PersistentVMRole') }
+        }
+      end
+      builder.doc.at_css('Role') << roleXML.at_css('PersistentVMRole').children.to_s
+      builder.doc
+    end
+    def create(params, deployXML)
+      servicecall = "hostedservices/#{params[:hosted_service_name]}/deployments"
+      @connection.query_azure(servicecall, "post", deployXML.to_xml) 
+    end
+  end
+end

data/lib/azure/disk.rb

@@ -0,0 +1,62 @@
+#
+# Author:: Barry Davis (barryd@jetstreamsoftware.com)
+# Copyright:: Copyright (c) 2010-2011 Opscode, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed 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.
+#
+
+class Azure
+  class Disks
+    def initialize(connection)
+      @connection=connection
+    end
+    def all
+      disks = Array.new
+      response = @connection.query_azure('disks')
+      founddisks = response.css('Disk')
+      founddisks.each do |disk|
+        item = Disk.new(disk)
+        disks << item
+      end
+      disks
+    end
+    def find(name)
+      founddisk = nil
+      self.all.each do |disk|
+        next unless disk.name == name
+        founddisk = disk
+      end
+      founddisk
+    end
+    def exists(name)
+      find(name) != nil
+    end
+    def clear_unattached
+      self.all.each do |disk|
+        next unless disk.attached == false
+        @connection.query_azure('disks/' + disk.name, 'delete')
+      end
+    end
+  end
+end
+
+class Azure
+  class Disk
+    attr_accessor :name, :attached
+    def initialize(disk)
+      @name = disk.at_css('Name').content
+      @attached = disk.at_css('AttachedTo') != nil
+    end
+  end
+end