#!/bin/bash
# generates configuration for FreeBSD VM deployment
# requires the name of the vm as an argument

image_url="https://download.freebsd.org/releases/VM-IMAGES/14.2-RELEASE/amd64/Latest/FreeBSD-14.2-RELEASE-amd64-BASIC-CLOUDINIT.zfs.qcow2.xz"
sshkeysdir="/root/k"
scriptdir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
dir_path="/var/lib/libvirt/images/.image_store"
src_file="${dir_path}/freebsd14.0.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"

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

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

# Download and extract FreeBSD image if it doesn't exist
if [ ! -f "$src_file" ]; then
    echo "source image does not exist! downloading..."
    sudo wget -O "${src_file}.xz" "$image_url"
    sudo xz -d "${src_file}.xz"
fi

# Note: FreeBSD uses a different user setup mechanism
cat > "$user_data" << EOF
#cloud-config
users:
  - name: user
    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: wheel
    shell: /bin/sh
runcmd:
  - pw usermod user -s /bin/sh
  - pkg update -f
  - pkg install -y vim git sudo
  - touch /home/user/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"