#!/bin/sh

# This will generate a openstack-style config drive image suitable for
# use with cloud-init.  You may optionally pass in an ssh public key
# (using the -k/--ssh-key option) and a user-data blog (using the
# -u/--user-data option).

usage () {
        echo "usage: ${0##*/}: [--ssh-key <pubkey>] [--gpg-key <file>] [--vendor-data <file>] [--user-data <file>] [--hostname <hostname>] [--model <model>] [--mk-pipelines <mk-pipelines>] [--pipeline-library <pipeline-library>] <imagename>"
}

ARGS=$(getopt \
        -o k:g:u:v:h:m:mp:p \
        --long help,hostname:,ssh-key:,gpg-key:,user-data:,vendor-data:,model:,mk-pipelines:,pipeline-library: -n ${0##*/} \
        -- "$@")

if [ $? -ne 0 ]; then
        usage >&2
        exit 2
fi

eval set -- "$ARGS"

while :; do
        case "$1" in
                --help)
                        usage
                        exit 0
                        ;;
                -k|--ssh-key)
                        ssh_key="$2"
                        shift 2
                        ;;
                -g|--gpg-key)
                        gpg_key="$2"
                        shift 2
                        ;;
                -u|--user-data)
                        user_data="$2"
                        shift 2
                        ;;
                -v|--vendor-data)
                        vendor_data="$2"
                        shift 2
                        ;;
                -h|--hostname)
                        hostname="$2"
                        shift 2
                        ;;
                -m|--model)
                        model="$2"
                        shift 2
                        ;;
                -mp|--mk-pipelines)
                        mk_pipelines="$2"
                        shift 2
                        ;;
                -p|--pipeline-library)
                        pipeline_library="$2"
                        shift 2
                        ;;
                --)     shift
                        break
                        ;;
        esac
done

config_image=$1
shift


uuid=$(uuidgen)
if ! [ "$hostname" ]; then
        hostname="$uuid"
fi

trap 'rm -rf $config_dir' EXIT
config_dir=$(mktemp -t -d configXXXXXX)

if [ "$ssh_key" ] && [ -f "$ssh_key" ]; then
        echo "adding pubkey from $ssh_key"
        cp $ssh_key $config_dir/root_auth_keys
fi

if [ "$gpg_key" ] && [ -f "$gpg_key" ]; then
        echo "adding gpg key from $gpg_key"
        mkdir $config_dir/gpg
        cp $gpg_key $config_dir/gpg/salt_master_pillar.asc
fi

if [ "$user_data" ] && [ -f "$user_data" ]; then
        echo "adding user data from $user_data"
        cp $user_data $config_dir/user-data
else
        touch $config_dir/user-data
fi

if [ "$vendor_data" ] && [ -f "$vendor_data" ]; then
        echo "adding vendor data from $vendor_data"
        cp $vendor_data $config_dir/vendor-data
fi

if [ "$model" ] ; then
        echo "adding reclass model directory $model"
        mkdir $config_dir/model
        cp -r $model $config_dir/model/
fi

if [ "$mk_pipelines" ] ; then
        echo "adding mk_pipelines directory $mk_pipelines"
        mkdir $config_dir/mk_pipelines
        cp -r $mk_pipelines $config_dir/mk-pipelines/
fi

if [ "$pipeline_library" ] ; then
        echo "adding pipeline_library directory $pipeline_library"
        mkdir $config_dir/pipeline_library
        cp -r $pipeline_library $config_dir/pipeline-library/
fi

cat > $config_dir/meta-data <<-EOF
instance-id: $uuid
hostname: $hostname
local-hostname: $hostname
EOF

#PS1="debug> " bash --norc

echo "generating configuration image at $config_image"
if ! mkisofs -o $config_image -V cidata -r -J --quiet $config_dir; then
        echo "ERROR: failed to create $config_image" >&2
        exit 1
fi

chmod a+r $config_image