cheese-chef-provisioning-azurerm 1.0.4 → 1.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e46d02c74aa1bca38bc1b1b2d8217c40b9f8935f
-  data.tar.gz: 464f2ec0d5a2e92a2b9f51266116990a9a290731
+  metadata.gz: 2a42dcbfec41ef166c44fda1953c68b13cd2ba05
+  data.tar.gz: 7394b5f6a7df6532a772040b026852ae7b1a30fb
 SHA512:
-  metadata.gz: 3ccdcfe0d51fd4eea33a43b5d7623ac5add998fb9791d38edd58052795235627eefd14422f17cb6f6f29852b94ecc786290e1bd170417ffcdac7579ea46a9251
-  data.tar.gz: 024a999c276bcd3db08a097f82126ef8c15d3c3dd7edc9e54046dc30f6b02be56d759f7a277b1ebf5668097780d7f3cb1a17e9abe22d58540870db3572e3bdb3
+  metadata.gz: 51c90e3f4f5c0de61fc4969035908d23cfb3fb9a29f8dc27b3b01535b2dc710a77a32c9bc185005afa39af01df652e07a7ad0539d165bf57ebd191d89b3171cd
+  data.tar.gz: 9463071f66eba7fd6eca9cf1237bc0223cf3d0af94013be9886d9d91e1c30a5b759ff09f9855e17e43f33640868736f05cd62d219ba19e40ddcf6811fec1dae4

data/lib/chef/provider/azure_data_disk.rb

@@ -0,0 +1,188 @@
+require 'chef/provisioning/azurerm/azure_provider'
+
+class Chef
+  class Provider
+    class AzureDataDisk < Chef::Provisioning::AzureRM::AzureProvider
+      provides :azure_data_disk
+
+      def whyrun_supported?
+        true
+      end
+
+      action :create do
+        vm = get_vm_silent
+        if vm && !data_disk_attached_to?(vm)
+          attach_data_disk(vm)
+        else
+          create_disk unless data_disk_exist?
+        end
+      end
+
+      action :destroy do
+        vm = get_vm_silent
+        if data_disk_exist?
+          action_handler.report_progress "Data disk #{new_resource.name} was found."
+          if data_disk_owner_id
+            if vm && vm.id == data_disk_owner_id
+              action_handler.report_progress "Data disk #{new_resource.name} is attached to #{new_resource.vm} VM."
+              detach_data_disk(vm)
+            else
+              raise "Data disk #{new_resource.name} is attached to a VM. Please provide correct VM information to destroy the disk."
+            end
+          end
+          result = destroy_data_disk
+          Chef::Log.debug("result: #{result.inspect}")
+        end
+      end
+
+      action :detach do
+        vm = get_vm_silent
+        if vm
+          if data_disk_exist? && data_disk_attached_to?(vm)
+            detach_data_disk(vm)
+          elsif !data_disk_exist?
+            raise "Data disk #{new_resource.name} not found."
+          end
+        else
+          raise "VM not available."
+        end
+      end
+
+      action :attach do
+        vm = get_vm_silent
+        if vm
+          if data_disk_exist? && !data_disk_attached_to?(vm)
+            attach_data_disk(vm)
+          elsif !data_disk_exist?
+            raise "Data disk #{new_resource.name} not found."
+          end
+        else
+          raise "VM not available."
+        end
+      end
+
+      def create_disk
+        action_handler.report_progress "Creating new data disk #{new_resource.name}..."
+        disk_params = Azure::ARM::Compute::Models::Disk.new.tap do |disk|
+          disk.account_type = new_resource.storage_account_type
+          disk.disk_size_gb = new_resource.size
+          disk.tags = new_resource.tags if new_resource.tags
+          disk.location = new_resource.location
+          disk.creation_data = Azure::ARM::Compute::Models::CreationData.new.tap do |creationData|
+            creationData.create_option = 'empty'
+          end
+        end
+        compute_management_client.disks.create_or_update(new_resource.resource_group, new_resource.name, disk_params)
+      end
+
+      def get_vm_silent
+        try_azure_operation('getting vm', true) do
+          compute_management_client.virtual_machines.get(new_resource.resource_group, new_resource.vm) if new_resource.vm
+        end
+      end
+
+      def data_disk_exist?
+        try_azure_operation('getting data disk', true) do
+          compute_management_client.disks.get(new_resource.resource_group, new_resource.name)
+        end
+      end
+
+      def data_disk_id
+        try_azure_operation('getting data disk') do
+          compute_management_client.disks.get(new_resource.resource_group, new_resource.name).id
+        end
+      end
+
+      def data_disk_size
+        try_azure_operation('getting data disk') do
+          compute_management_client.disks.get(new_resource.resource_group, new_resource.name).disk_size_gb
+        end
+      end
+
+      def data_disk_owner_id
+        try_azure_operation('getting data disk') do
+          compute_management_client.disks.get(new_resource.resource_group, new_resource.name).owner_id
+        end
+      end
+
+      def data_disk_owner_id_silent
+        try_azure_operation('getting data disk', true) do
+          compute_management_client.disks.get(new_resource.resource_group, new_resource.name).owner_id
+        end
+      end
+
+      def data_disk_attached_to?(vm)
+        if data_disk_owner_id_silent
+          if data_disk_owner_id == vm.id
+            return true
+          else
+            raise "Disk #{new_resource.name} is attached to some other VM."
+          end
+        end
+        false
+      end
+
+      def data_disk_setup_for(vm)
+        data_disk = Azure::ARM::Compute::Models::DataDisk.new.tap do |dd_disk|
+          dd_disk.name = new_resource.name
+          dd_disk.disk_size_gb = new_resource.size if !data_disk_exist? || new_resource.size > data_disk_size
+          Chef::Log.warn("Disk size reduction is not supported") if data_disk_exist? && new_resource.size < data_disk_size
+          dd_disk.lun = disk_lun(vm)
+          dd_disk.caching = new_resource.caching
+          dd_disk.create_option = data_disk_exist? ? 'attach' : 'empty'
+          managedDiskInfo = Azure::ARM::Compute::Models::ManagedDiskParameters.new.tap do |managedDisk|
+            if data_disk_exist?
+              action_handler.report_progress "Data disk with name #{new_resource.name} was found. Using this..."
+              managedDisk.id = data_disk_id
+            else
+              managedDisk.storage_account_type = new_resource.storage_account_type
+            end
+          end
+          dd_disk.managed_disk = managedDiskInfo
+        end
+      end
+
+      def detach_data_disk(vm)
+        if vm
+          vm.storage_profile.data_disks.each do |ddisk|
+            if ddisk.name == new_resource.name
+              action_handler.report_progress "Removing data disk #{new_resource.name} from VM #{new_resource.vm}."
+              (vm.storage_profile.data_disks).delete(ddisk)
+              result = compute_management_client.virtual_machines.create_or_update(new_resource.resource_group, new_resource.vm, vm)
+              Chef::Log.debug("result: #{result.inspect}")
+            end
+          end
+        else
+          raise "VM not available."
+        end
+      end
+
+      def attach_data_disk(vm)
+        if vm
+          data_disk = data_disk_setup_for(vm)
+          vm.storage_profile.data_disks.push(data_disk)
+          action_handler.report_progress "Updating VM #{new_resource.vm} with data disk #{new_resource.name}..."
+          result = compute_management_client.virtual_machines.create_or_update(new_resource.resource_group, new_resource.vm, vm)
+          Chef::Log.debug("result: #{result.inspect}")
+        else
+          raise "VM not available."
+        end
+      end
+
+      def destroy_data_disk
+        action_handler.report_progress "Deleting data disk #{new_resource.name}."
+        compute_management_client.disks.delete(new_resource.resource_group, new_resource.name)
+      end
+
+      def disk_lun(vm)
+        if new_resource.lun
+          new_resource.lun
+        elsif (vm.storage_profile.data_disks.last).nil?
+          2
+        else
+          vm.storage_profile.data_disks.last.lun + 1
+        end
+      end
+    end
+  end
+end

data/lib/chef/provisioning/azurerm.rb

@@ -10,7 +10,7 @@ require 'azure_mgmt_network'
 Chef::Log.info("chef-provisioning-azurerm #{Chef::Provisioning::AzureRM::VERSION}")
 Chef::Log.info("chef-provisioning #{Chef::Provisioning::VERSION}")
 
-resources = %w(resource_group resource_template storage_account virtual_network network_interface public_ip_address)
+resources = %w(resource_group resource_template storage_account virtual_network network_interface public_ip_address data_disk)
 resources.each do |r|
   require "chef/resource/azure_#{r}"
   require "chef/provider/azure_#{r}"

data/lib/chef/provisioning/azurerm/version.rb

@@ -1,7 +1,7 @@
 class Chef
   module Provisioning
     module AzureRM
-      VERSION = '1.0.4'.freeze
+      VERSION = '1.0.5'.freeze
     end
   end
 end

data/lib/chef/resource/azure_data_disk.rb

@@ -0,0 +1,20 @@
+require 'chef/provisioning/azurerm/azure_resource'
+
+class Chef
+  class Resource
+    class AzureDataDisk < Chef::Provisioning::AzureRM::AzureResource
+      resource_name :azure_data_disk
+      actions :create, :destroy, :detach, :attach, :nothing
+      default_action :create
+      attribute :name, kind_of: String, name_attribute: true, regex: /^[\w\-\(\)\.]{0,80}$+(?<!\.)$/i
+      attribute :size, kind_of: Integer, default: 10
+      attribute :storage_account_type, String, default: 'Standard_LRS'
+      attribute :caching, String, default: 'none'
+      attribute :location, String
+      attribute :lun, Integer
+      attribute :vm, String
+      attribute :tags, kind_of: Hash
+      attribute :resource_group, String, required: true
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cheese-chef-provisioning-azurerm
 version: !ruby/object:Gem::Version
-  version: 1.0.4
+  version: 1.0.5
 platform: ruby
 authors:
 - Stuart Preston
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-04-10 00:00:00.000000000 Z
+date: 2017-04-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chef
@@ -156,6 +156,7 @@ files:
 - CHANGELOG.md
 - LICENSE.txt
 - README.md
+- lib/chef/provider/azure_data_disk.rb
 - lib/chef/provider/azure_network_interface.rb
 - lib/chef/provider/azure_public_ip_address.rb
 - lib/chef/provider/azure_resource_group.rb
@@ -170,6 +171,7 @@ files:
 - lib/chef/provisioning/azurerm/driver.rb
 - lib/chef/provisioning/azurerm/version.rb
 - lib/chef/provisioning/driver_init/azurerm.rb
+- lib/chef/resource/azure_data_disk.rb
 - lib/chef/resource/azure_network_interface.rb
 - lib/chef/resource/azure_public_ip_address.rb
 - lib/chef/resource/azure_resource_group.rb