Skip to content

Commit

Permalink
feat: Add Cloudinit support using iso image (#294)
Browse files Browse the repository at this point in the history
fixes #296
  • Loading branch information
Manisha15 authored Oct 10, 2023
1 parent abd790a commit 55e22a1
Show file tree
Hide file tree
Showing 2 changed files with 74 additions and 0 deletions.
73 changes: 73 additions & 0 deletions app/helpers/proxmox_vm_cloudinit_helper.rb
Original file line number Diff line number Diff line change
Expand Up @@ -41,4 +41,77 @@ def parse_server_cloudinit(args)
end
cloudinit_h
end

def create_cloudinit_iso(vm_name, configs)
ssh = Fog::SSH.new(URI.parse(fog_credentials[:proxmox_url]).host, fog_credentials[:proxmox_username].split('@')[0], { password: fog_credentials[:proxmox_password] })
files = []
wd = create_temp_directory(ssh)
configs.each do |config|
begin
ssh.run([%({echo "#{config[1]}" >> "#{wd}/#{config[0]}"})])
rescue ::Foreman::Exception => e
logger.error("Error writing to the file #{config[0]}: #{e}")
raise e
end
files.append(File.join(wd, config[0]))
end
raise Foreman::Exception, N_('ISO build failed') unless ssh.run(generate_iso_command(vm_name, files))
delete_config_files(wd)
iso
end

def generate_iso_command(vm_name, files)
iso = File.join(default_iso_path, "#{vm_name}_cloudinit.iso")
arguments = ["genisoimage", "-output #{iso}", '-volid', 'cidata', '-joliet', '-rock']
logger.warn("iso image generation args: #{arguments}")
arguments.concat(files).join(' ')
end

def create_temp_directory(ssh)
res = ssh.run([%({mktemp -d})])
raise ::Foreman::Exception, "Could not create working directory to store cloudinit config data: #{res.first.stdout}." unless res.first.status.zero?
res.first.stdout.chomp
end

def delete_config_files(working_dir)
ssh.run("rm -rf #{working_dir}")
rescue Foreman::Exception => e
logger.warn("Could not delete directory for config files: #{e}. Please delete it manually at #{working_dir}")
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

def default_iso_path
"/var/lib/vz/template/iso"
end
end
1 change: 1 addition & 0 deletions app/models/foreman_fog_proxmox/proxmox_vm_commands.rb
Original file line number Diff line number Diff line change
Expand Up @@ -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] }
Expand Down

0 comments on commit 55e22a1

Please sign in to comment.