#!/bin/bash
# generates user data, meta data, and seed.iso for cloud init. for Ubuntu
# requires the name of the vm as an argument

image_url="https://cloud-images.ubuntu.com/noble/current/noble-server-cloudimg-amd64.img"
sshkeysdir="/root/k"
scriptdir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
dir_path="/var/lib/libvirt/images/.image_store"
src_file="${dir_path}/ubuntu24.04.qcow2"
config_dir="/var/lib/libvirt/images/.temp"
user_data="${config_dir}/user-data"
meta_data="${config_dir}/meta-data"
seed_iso="${config_dir}/seed.iso"
temp_img="${dir_path}/ubuntu_temp.img"

if [ $# -ne 1 ]; then
    echo "Usage: $0 <vm-name>"
    exit 1
fi

sudo mkdir -p "$dir_path"
sudo mkdir -p "$config_dir"

if [ ! -f "$src_file" ]; then
    echo "source image does not exist! downloading..."
    sudo wget -O "$temp_img" "$image_url"
    echo "Converting to qcow2 format..."
    sudo qemu-img convert -f raw -O qcow2 "$temp_img" "$src_file"
    sudo rm "$temp_img"
fi

cat > "$user_data" << EOF
#cloud-config
users:
  - name: ubuntu
    ssh-authorized-keys:
EOF

for key in $sshkeysdir/*.pub; do
    echo "      - $(cat "$key")" >> $user_data
done

cat >> "$user_data" << 'EOF'
    sudo: ['ALL=(ALL) NOPASSWD:ALL']
    groups: sudo
    shell: /bin/bash
runcmd:
  - sudo growpart /dev/sda 1
  - sudo resize2fs /dev/sda1
  - sudo apt-get update
  - sudo apt-get install -y vim git
  - cd /home/ubuntu
  - touch /home/ubuntu/runcmd_done
EOF

cat > "$meta_data" << EOF
instance-id: vm_id
local-hostname: $1
EOF

genisoimage -output "$seed_iso" -volid cidata -joliet -rock "$user_data" "$meta_data" &> /dev/null || { echo "Failed to create seed.iso."; exit 1; }
echo "Configuration files generated successfully"