From 1f9aa37b0457c0beeffcc7d1f35ac55f103d3f46 Mon Sep 17 00:00:00 2001
From: Manisha Singhal <singhal@atix.de>
Date: Thu, 28 Sep 2023 08:59:40 +0200
Subject: [PATCH] Add Cloudinit support using iso image

---
 app/helpers/proxmox_vm_cloudinit_helper.rb    | 51 +++++++++++++++++++
 .../proxmox_vm_commands.rb                    |  1 +
 2 files changed, 52 insertions(+)

diff --git a/app/helpers/proxmox_vm_cloudinit_helper.rb b/app/helpers/proxmox_vm_cloudinit_helper.rb
index 9d12c4ac3..6d2ce080f 100644
--- a/app/helpers/proxmox_vm_cloudinit_helper.rb
+++ b/app/helpers/proxmox_vm_cloudinit_helper.rb
@@ -41,4 +41,55 @@ def parse_server_cloudinit(args)
     end
     cloudinit_h
   end
+
+  def create_cloudinit_iso(vm_name, configs)
+    iso = File.join('/var/lib/vz/template/iso', "#{vm_name}_cloudinit.iso")
+    arguments = ['genisoimage', "-output #{iso}", '-volid', 'cidata', '-joliet', '-rock']
+    ssh = Fog::SSH.new(URI.parse(fog_credentials[:proxmox_url]).host, fog_credentials[:proxmox_username].split('@')[0], { password: fog_credentials[:proxmox_password] })
+    wd = ssh.run([%({mktemp -d})]).first.stdout.chomp
+    configs.each do |config|
+      begin
+        ssh.run([%({echo "#{config[1]}" >> "#{wd}/#{config[0]}"})])
+      rescue ::Foreman::Exception => e
+        logger.warn("Error writing to the file #{config[0]}: #{e}")
+      end
+      arguments.append(File.join(wd, config[0]))
+    end
+    logger.warn("iso image generation args: #{arguments}")
+    raise Foreman::Exception.new(N_("ISO build failed")) unless ssh.run([%{#{arguments.join(' ')}}])
+    ssh.run([%(rm -rf #{wd})])
+    iso
+  end
+
+  def parse_cloudinit_config(args)
+    filenames = ["meta-data"]
+    config_data = ["instance-id: #{args[:name]}"]
+    user_data = args.delete(:user_data)
+
+    return args if user_data == ''
+
+    if user_data.include?('#network-config') && user_data.include?('#cloud-config')
+      config_data.concat(user_data.split('#network-config'))
+      filenames.append(['user-data', 'network-config'])
+    elsif user_data.include?('#network-config') && !user_data.include?('#cloud-config')
+      config_data.append(user_data.split('#network-config')[1])
+      filenames.append("network-config")
+    elsif !user_data.include?('#network-config') && user_data.include?('#cloud-config')
+      config_data.append(user_data)
+      filenames.append("user-data")
+    end
+
+    return args if config_data.length == 1
+
+    configs = filenames.zip(config_data).to_h
+
+    iso = create_cloudinit_iso(args[:name], configs)
+    args.merge(attach_cloudinit_iso(args[:node_id], iso))
+  end
+
+  def attach_cloudinit_iso(node, iso)
+    storage = storages(node, 'iso')[0]
+    volume = storage.volumes.detect { |v| v.volid.include? File.basename(iso) }
+    { ide2: "#{volume.volid},media=cdrom" }
+  end
 end
diff --git a/app/models/foreman_fog_proxmox/proxmox_vm_commands.rb b/app/models/foreman_fog_proxmox/proxmox_vm_commands.rb
index cfe5ddfa7..8287cb81f 100644
--- a/app/models/foreman_fog_proxmox/proxmox_vm_commands.rb
+++ b/app/models/foreman_fog_proxmox/proxmox_vm_commands.rb
@@ -56,6 +56,7 @@ def create_vm(args = {})
     end
 
     def compute_clone_attributes(args, container, type)
+      args = parse_cloudinit_config(args) if args[:user_data]
       parsed_args = parse_typed_vm(args, type)
       if container
         options = { :hostname => args[:name] }