Add cloud-init support

Fixes #68

Author: tristanrobert

lib/fog/proxmox/compute/models/disk.rb

@@ -18,6 +18,7 @@
 # frozen_string_literal: true
 
 require 'fog/proxmox/helpers/disk_helper'
+require 'fog/proxmox/helpers/controller_helper'
 
 module Fog
   module Proxmox
@@ -56,11 +57,19 @@ def rootfs?
         end
 
         def mount_point?
-          id.match(/(mp)(\d+)/)
+          Fog::Proxmox::DiskHelper.mount_point?(id)
         end
 
         def controller?
-          id.match(/(scsi|ide|sata|virtio)(\d+)/)
+          Fog::Proxmox::DiskHelper.server_disk?(id)
+        end
+
+        def cloud_init?
+          id != 'ide2' && media == 'cdrom'
+        end
+
+        def hard_disk?
+          controller? && !cdrom? && !cloud_init?
         end
 
         def template?

lib/fog/proxmox/compute/models/disks.rb

@@ -45,6 +45,10 @@ def cdrom
         def rootfs
           find(&:rootfs?)
         end
+
+        def cloudinit
+          find(&:cloud_init?)
+        end
       end
     end
   end

lib/fog/proxmox/compute/models/server_config.rb

@@ -69,6 +69,14 @@ class ServerConfig < Fog::Model
         attribute :unprivileged
         attribute :interfaces
         attribute :disks
+        attribute :ciuser
+        attribute :cipassword
+        attribute :cicustom
+        attribute :citype
+        attribute :ipconfigs
+        attribute :sshkeys
+        attribute :searchdomain
+        attribute :nameserver
 
         def initialize(new_attributes = {})
           prepare_service_value(new_attributes)

lib/fog/proxmox/helpers/disk_helper.rb

@@ -23,15 +23,15 @@ module Fog
   module Proxmox
     # module Disk mixins
     module DiskHelper
-      
       DISKS_REGEXP = /^(scsi|sata|mp|rootfs|virtio|ide)(\d+){0,1}$/
       SERVER_DISK_REGEXP = /^(scsi|sata|virtio|ide)(\d+)$/
       MOUNT_POINT_REGEXP = /^(mp)(\d+)$/
       ROOTFS_REGEXP = /^(rootfs)$/
       CDROM_REGEXP = /^(.*)[,]{0,1}(media=cdrom)[,]{0,1}(.*)$/
       TEMPLATE_REGEXP = /^(.*)(base-)(.*)$/
+      CLOUD_INIT_REGEXP = /^(.*)(cloudinit)(.*)$/
 
-      # Convert disk attributes hash into API Proxmox parameters string 
+      # Convert disk attributes hash into API Proxmox parameters string
       def self.flatten(disk)
         id = disk[:id]
         value = ''
@@ -161,6 +161,19 @@ def self.container_disk?(id)
       def self.template?(volid)
         TEMPLATE_REGEXP.match(volid) ? true : false
       end
+
+      def self.cloud_init?(volid)
+        CLOUD_INIT_REGEXP.match(volid) ? true : false
+      end
+
+      def self.of_type?(disk_h, vm_type)
+        id = disk_h['id'] if disk_h.key?('id')
+        id = disk_h[:id] if disk_h.key?(:id)
+        result = false
+        result = server_disk?(id) if vm_type == 'qemu'
+        result = container_disk?(id) if vm_type == 'lxc'
+        result
+      end
     end
   end
 end