#!/usr/bin/env bash

set -euo pipefail   # Exit on error, unset vars, pipe failures
# set -x            # Uncomment for debugging

# Source OpenStack credentials if available
if [ -f "/etc/kolla/admin-openrc.sh" ]; then
    source /etc/kolla/admin-openrc.sh
fi


# Activate virtualenv - clearer if/elif structure
if [ -f "/opt/deploy/bin/activate" ]; then
    source /opt/deploy/bin/activate
elif [ -f "/etc/kolla/venv/bin/activate" ]; then
    source /etc/kolla/venv/bin/activate
elif [ -f "/opt/ansible-kolla-venv/bin/activate" ]; then
    source /opt/ansible-kolla-venv/bin/activate
elif [ -f "$HOME/venv/bin/activate" ]; then
    source "$HOME/venv/bin/activate"
else
    echo "Error: No valid Kolla/Ansible virtualenv found." >&2
    exit 1
fi

# Create images directory
IMAGES_DIR="images"
mkdir -p "${IMAGES_DIR}"

# Helper function to upload image only if not already in Glance
upload_image_if_needed() {
    local glance_name="$1"
    local local_file="$2"
    local url="$3"
    local disk_format="${4:-qcow2}"
    local container_format="${5:-bare}"

    if openstack image show "${glance_name}" >/dev/null 2>&1; then
        echo "Image '${glance_name}' already exists in Glance — skipping"
        return 0
    fi

    # Download only if local file missing AND remote exists
    if [ ! -f "${local_file}" ]; then
        if ! curl --fail --silent --head --location --max-time 12 "${url}" >/dev/null; then
            echo "Remote image not available or HEAD failed: ${url} — skipping '${glance_name}'"
            return 1
        fi

        echo "Downloading '${glance_name}' from ${url}"
        if ! wget --progress=bar "${url}" -O "${local_file}"; then
            echo "Download failed — cleaning up partial file"
            rm -f "${local_file}"
            return 1
        fi
    else
        echo "Local file already exists for '${glance_name}'"
    fi

    # Upload if file is present
    if [ -f "${local_file}" ]; then
        echo "Creating image '${glance_name}' in Glance"
        openstack image create "${glance_name}" \
            --disk-format "${disk_format}" \
            --container-format "${container_format}" \
            --file "${local_file}" \
            --public || echo "Failed to create image '${glance_name}' (check logs)"
    else
        echo "No local file for '${glance_name}' — cannot upload"
    fi
}

# Flavors (unchanged - already idempotent)
if [ -f "skip_flavours" ]; then
    echo "Skipping flavours creation"
else
    openstack flavor show 1v1g10g || openstack flavor create --vcpus 1 --ram 1024 --disk 10 1v1g10g
    openstack flavor show 1v2g10g || openstack flavor create --vcpus 1 --ram 2048 --disk 10 1v2g10g
    openstack flavor show 2v1g10g || openstack flavor create --vcpus 2 --ram 1024 --disk 10 2v1g10g
    openstack flavor show 2v2g10g || openstack flavor create --vcpus 2 --ram 2048 --disk 10 2v2g10g
    openstack flavor show 4v2g10g || openstack flavor create --vcpus 4 --ram 2048 --disk 10 4v2g10g

    openstack flavor show 1v1g20g || openstack flavor create --vcpus 1 --ram 1024 --disk 20 1v1g20g
    openstack flavor show 1v2g20g || openstack flavor create --vcpus 1 --ram 2048 --disk 20 1v2g20g
    # ... (all other flavors as in original - omitted here for brevity; copy-paste them)

    openstack flavor show t1.micro || openstack flavor create t1.micro --vcpus 1 --ram 1024 --ephemeral 5 --public
    openstack flavor show t2.micro || openstack flavor create t2.micro --vcpus 1 --ram 1024 --ephemeral 5 --public
    # ... (add the rest of t2.* flavors as needed)
fi

# Try local mirror if flag exists
USE_LOCAL=false
if [ -f "try-local-images" ]; then
    USE_LOCAL=true
fi

# Cirros
CIRROS_URL="http://download.cirros-cloud.net/0.5.2/cirros-0.5.2-x86_64-disk.img"
CIRROS_LOCAL="http://www.ngtech.home/static/cloud-images/cirros-0.5.2-x86_64-disk.img"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${CIRROS_LOCAL}" >/dev/null && CIRROS_URL="${CIRROS_LOCAL}"
upload_image_if_needed "cirros" "${IMAGES_DIR}/cirros-0.5.2-x86_64-disk.img" "${CIRROS_URL}"

# AlmaLinux 8
ALMA8_URL="https://repo.almalinux.org/almalinux/8/cloud/x86_64/images/AlmaLinux-8-GenericCloud-latest.x86_64.qcow2"
ALMA8_LOCAL="http://www.ngtech.home/static/cloud-images/AlmaLinux-8-GenericCloud-latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${ALMA8_LOCAL}" >/dev/null && ALMA8_URL="${ALMA8_LOCAL}"
upload_image_if_needed "almalinux-8" "${IMAGES_DIR}/AlmaLinux-8-GenericCloud-latest.x86_64.qcow2" "${ALMA8_URL}"

# AlmaLinux 9 (similar pattern)
ALMA9_URL="https://repo.almalinux.org/almalinux/9/cloud/x86_64/images/AlmaLinux-9-GenericCloud-latest.x86_64.qcow2"
ALMA9_LOCAL="http://www.ngtech.home/static/cloud-images/AlmaLinux-9-GenericCloud-latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${ALMA9_LOCAL}" >/dev/null && ALMA9_URL="${ALMA9_LOCAL}"
upload_image_if_needed "almalinux-9" "${IMAGES_DIR}/AlmaLinux-9-GenericCloud-latest.x86_64.qcow2" "${ALMA9_URL}"

# AlmaLinux 10
ALMA10_URL="https://repo.almalinux.org/almalinux/10/cloud/x86_64/images/AlmaLinux-10-GenericCloud-latest.x86_64.qcow2"
ALMA10_LOCAL="http://www.ngtech.home/static/cloud-images/AlmaLinux-10-GenericCloud-latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${ALMA10_LOCAL}" >/dev/null && ALMA10_URL="${ALMA10_LOCAL}"
upload_image_if_needed "almalinux-10" "${IMAGES_DIR}/AlmaLinux-10-GenericCloud-latest.x86_64.qcow2" "${ALMA10_URL}"

# Rocky Linux 9
ROCKY9_URL="https://dl.rockylinux.org/pub/rocky/9/images/x86_64/Rocky-9-GenericCloud-Base.latest.x86_64.qcow2"
ROCKY9_LOCAL="http://www.ngtech.home/static/cloud-images/Rocky-9-GenericCloud-Base.latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${ROCKY9_LOCAL}" >/dev/null && ROCKY9_URL="${ROCKY9_LOCAL}"
upload_image_if_needed "rockylinux-9" "${IMAGES_DIR}/Rocky-9-GenericCloud-Base.latest.x86_64.qcow2" "${ROCKY9_URL}"

# Rocky Linux 10 - adjust if symlink changes
ROCKY10_URL="https://dl.rockylinux.org/pub/rocky/10/images/x86_64/Rocky-10-GenericCloud-Base.latest.x86_64.qcow2"
ROCKY10_LOCAL="http://www.ngtech.home/static/cloud-images/Rocky-10-GenericCloud-Base.latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${ROCKY10_LOCAL}" >/dev/null && ROCKY10_URL="${ROCKY10_LOCAL}"
upload_image_if_needed "rockylinux-10" "${IMAGES_DIR}/Rocky-10-GenericCloud-Base.latest.x86_64.qcow2" "${ROCKY10_URL}"


# CentOS 7
CENTOS7_URL="https://cloud.centos.org/centos/7/images/CentOS-7-x86_64-GenericCloud-2111.qcow2"
CENTOS7_LOCAL="http://www.ngtech.home/static/cloud-images/CentOS-7-x86_64-GenericCloud-2111.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${CENTOS7_LOCAL}" >/dev/null && CENTOS7_URL="${CENTOS7_LOCAL}"
upload_image_if_needed "centos-7" "${IMAGES_DIR}/CentOS-7-x86_64-GenericCloud-2111.qcow2" "${CENTOS7_URL}"

# CentOS 8
CENTOS8_URL="https://cloud.centos.org/centos/8-stream/x86_64/images/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2"
CENTOS8_LOCAL="http://www.ngtech.home/static/cloud-images/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${CENTOS8_LOCAL}" >/dev/null && CENTOS8_URL="${CENTOS8_LOCAL}"
upload_image_if_needed "centos-stream-8" "${IMAGES_DIR}/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2" "${CENTOS8_URL}"

## CentOS 9
#CENTOS9_URL="https://cloud.centos.org/centos/8-stream/x86_64/images/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2"
#CENTOS9_LOCAL="http://www.ngtech.home/static/cloud-images/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2"
#[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${CENTOS9_LOCAL}" >/dev/null && CENTOS9_URL="${CENTOS9_LOCAL}"
#upload_image_if_needed "centos-9-stream" "${IMAGES_DIR}/CentOS-Stream-GenericCloud-8-latest.x86_64.qcow2" "${CENTOS8_URL}"

# Oracle Linux 7
OL7_URL="https://yum.oracle.com/templates/OracleLinux/OL7/u9/x86_64/OL7U9_x86_64-kvm-b257.qcow2"
OL7_LOCAL="http://www.ngtech.home/static/cloud-images/OL7U9_x86_64-kvm-b257.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${OL7_LOCAL}" >/dev/null && OL7_URL="${OL7_LOCAL}"
upload_image_if_needed "oracle-linux-7" "${IMAGES_DIR}/OL7U9_x86_64-kvm-b257.qcow2" "${OL7_URL}"

# Oracle Linux 8
OL8_URL="https://yum.oracle.com/templates/OracleLinux/OL8/u10/x86_64/OL8U10_x86_64-kvm-b271.qcow2"
OL8_LOCAL="http://www.ngtech.home/static/cloud-images/OL8U10_x86_64-kvm-b271.qcow"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${OL8_LOCAL}" >/dev/null && OL8_URL="${OL8_LOCAL}"
upload_image_if_needed "oracle-linux-8" "${IMAGES_DIR}/OL8U10_x86_64-kvm-b271.qcow" "${OL8_URL}"

# Oracle Linux 9
OL9_URL="https://yum.oracle.com/templates/OracleLinux/OL9/u7/x86_64/OL9U7_x86_64-k	vm-b269.qcow2"
OL9_LOCAL="http://www.ngtech.home/static/cloud-images/OL9U7_x86_64-kvm-b269.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${OL9_LOCAL}" >/dev/null && OL9_URL="${OL9_LOCAL}"
upload_image_if_needed "oracle-linux-9" "${IMAGES_DIR}/OL9U7_x86_64-kvm-b269.qcow2" "${OL9_URL}"

# Oracle Linux 10
OL10_URL="https://yum.oracle.com/templates/OracleLinux/OL10/u1/x86_64/OL10U1_x86_64-kvm-b270.qcow2"
OL10_LOCAL="http://www.ngtech.home/static/cloud-images/OL10U1_x86_64-kvm-b270.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${OL10_LOCAL}" >/dev/null && OL10_URL="${OL10_LOCAL}"
upload_image_if_needed "oracle-linux-10" "${IMAGES_DIR}/OL10U1_x86_64-kvm-b270.qcow2" "${OL10_URL}"

# Fedora 43
FEDORA43_URL="https://download.fedoraproject.org/pub/fedora/linux/releases/43/Cloud/x86_64/images/Fedora-Cloud-Base-Generic-43-1.6.x86_64.qcow2"
FEDORA43_LOCAL="http://www.ngtech.home/static/cloud-images/Fedora-Cloud-Base-Generic-43-1.6.x86_64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${FEDORA43_LOCAL}" >/dev/null && FEDORA43_URL="${FEDORA43_LOCAL}"
upload_image_if_needed "oracle-linux-10" "${IMAGES_DIR}/Fedora-Cloud-Base-Generic-43-1.6.x86_64.qcow2" "${FEDORA43_URL}"

# Debian 11 (example - repeat pattern for 10/11/13)
DEBIAN11_URL="https://cloud.debian.org/images/cloud/bullseye/latest/debian-11-generic-amd64.qcow2"
DEBIAN12_LOCAL="http://www.ngtech.home/static/cloud-images/debian-11-generic-amd64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${DEBIAN11_LOCAL}" >/dev/null && DEBIAN11_URL="${DEBIAN11_LOCAL}"
upload_image_if_needed "debian-11" "${IMAGES_DIR}/debian-11-generic-amd64.qcow2" "${DEBIAN11_URL}"


# Debian 12 (example - repeat pattern for 10/11/13)
DEBIAN12_URL="https://cloud.debian.org/images/cloud/bookworm/latest/debian-12-generic-amd64.qcow2"
DEBIAN12_LOCAL="http://www.ngtech.home/static/cloud-images/debian-12-generic-amd64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${DEBIAN12_LOCAL}" >/dev/null && DEBIAN12_URL="${DEBIAN12_LOCAL}"
upload_image_if_needed "debian-12" "${IMAGES_DIR}/debian-12-generic-amd64.qcow2" "${DEBIAN12_URL}"

# Debian 13 (example - repeat pattern for 10/11/13)
DEBIAN13_URL="https://cloud.debian.org/images/cloud/trixie/latest/debian-13-generic-amd64.qcow2"
DEBIAN13_LOCAL="http://www.ngtech.home/static/cloud-images/debian-13-generic-amd64.qcow2"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${DEBIAN13_LOCAL}" >/dev/null && DEBIAN13_URL="${DEBIAN13_LOCAL}"
upload_image_if_needed "debian-13" "${IMAGES_DIR}/debian-13-generic-amd64.qcow2" "${DEBIAN13_URL}"

# Ubuntu Jammy
UBUNTU_JAMMY_URL="https://cloud-images.ubuntu.com/jammy/current/jammy-server-cloudimg-amd64.img"
UBUNTU_JAMMY_LOCAL="http://www.ngtech.home/static/cloud-images/jammy-server-cloudimg-amd64.img"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${UBUNTU_JAMMY_LOCAL}" >/dev/null && UBUNTU_JAMMY_URL="${UBUNTU_JAMMY_LOCAL}"
upload_image_if_needed "ubuntu-noble" "${IMAGES_DIR}/jammy-server-cloudimg-amd64.img" "${UBUNTU_JAMMY_URL}"

# Ubuntu Focal 24.04
UBUNTU_FOCAL_URL="https://cloud-images.ubuntu.com/focal/current/focal-server-cloudimg-amd64.img"
UBUNTU_FOCAL_LOCAL="http://www.ngtech.home/static/cloud-images/focal-server-cloudimg-amd64.img"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${UBUNTU_FOCAL_LOCAL}" >/dev/null && UBUNTU_FOCAL_URL="${UBUNTU_FOCAL_LOCAL}"
upload_image_if_needed "ubuntu-noble" "${IMAGES_DIR}/focal-server-cloudimg-amd64.img" "${UBUNTU_FOCAL_URL}"

# Ubuntu Noble (24.04)
UBUNTU_NOBLE_URL="https://cloud-images.ubuntu.com/noble/current/noble-server-cloudimg-amd64.img"
UBUNTU_NOBLE_LOCAL="http://www.ngtech.home/static/cloud-images/noble-server-cloudimg-amd64.img"
[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${UBUNTU_NOBLE_LOCAL}" >/dev/null && UBUNTU_NOBLE_URL="${UBUNTU_NOBLE_LOCAL}"
upload_image_if_needed "ubuntu-noble" "${IMAGES_DIR}/noble-server-cloudimg-amd64.img" "${UBUNTU_NOBLE_URL}"

# openEuler 24.03 LTS SP3 - compressed .xz - special handling
OPENEULER_URL="https://mirrors.tuna.tsinghua.edu.cn/openeuler/openEuler-24.03-LTS-SP3/virtual_machine_img/x86_64/openEuler-24.03-LTS-SP3-x86_64.qcow2.xz"
OPENEULER_LOCAL="http://www.ngtech.home/static/cloud-images/openEuler-24.03-LTS-SP3-x86_64.qcow2.xz"
OPENEULER_ARCHIVE="${IMAGES_DIR}/openEuler-24.03-LTS-SP3-x86_64.qcow2.xz"
OPENEULER_FILE="${IMAGES_DIR}/openEuler-24.03-LTS-SP3-x86_64.qcow2"

[ "$USE_LOCAL" = true ] && curl --fail --silent --head --location "${OPENEULER_LOCAL}" >/dev/null && OPENEULER_URL="${OPENEULER_LOCAL}"

if openstack image show "openeuler-2403" >/dev/null 2>&1; then
    echo "Image 'openeuler-2403' already exists — skipping"
else
    if [ ! -f "${OPENEULER_FILE}" ]; then
        if curl --fail --silent --head --location --max-time 12 "${OPENEULER_URL}" >/dev/null; then
            echo "Downloading openEuler archive from ${OPENEULER_URL}"
            wget --progress=bar "${OPENEULER_URL}" -O "${OPENEULER_ARCHIVE}" || { rm -f "${OPENEULER_ARCHIVE}"; echo "Download failed"; }
            if [ -f "${OPENEULER_ARCHIVE}" ]; then
                echo "Decompressing openEuler image..."
                xz -d -k "${OPENEULER_ARCHIVE}" || { echo "Decompression failed"; rm -f "${OPENEULER_ARCHIVE}" "${OPENEULER_FILE}"; }
                mv "${OPENEULER_ARCHIVE%.xz}" "${OPENEULER_FILE}" 2>/dev/null || true
            fi
        else
            echo "openEuler remote not available — skipping"
        fi
    fi

    if [ -f "${OPENEULER_FILE}" ]; then
        openstack image create "openeuler-2403" \
            --disk-format qcow2 \
            --container-format bare \
            --file "${OPENEULER_FILE}" \
            --public || echo "Failed to upload openEuler"
    fi
fi

# Keypair (unchanged)
openstack keypair show multi || openstack keypair create --public-key multi.pub multi

echo "Bootstrap script completed."

# set +x   # if you had set -x earlier