create_custom_iso.sh

#!/bin/bash
set -eo pipefail

K3OS_RELEASE_VERSION=$1
K3OS_FILENAME=$2
K3OS_RELEASE_URL="https://github.com/rancher/k3os/releases/download/${K3OS_RELEASE_VERSION}/${K3OS_FILENAME}"
K3OS_REMASTERED_FILENAME="k3os-custom.iso"

COLOR_CYAN="\033[1;36m"
COLOR_WHITE="\033[1;37m"
COLOR_RED="\033[0;31m"
NO_COLOR="\033[0m"

main () {
    echo
    echo -e "${COLOR_CYAN}###################################################################################################################${NO_COLOR}"
    echo -e "${COLOR_WHITE}Downloading k3os release version ${K3OS_RELEASE_VERSION} ${NO_COLOR}"
    echo

    wget -q --show-progress $K3OS_RELEASE_URL -O /tmp/$K3OS_FILENAME
    
    if [[ ! -f /tmp/$K3OS_FILENAME ]]; then
        echo -e "${COLOR_RED}File ${K3OS_RELEASE_URL} could not be downloaded. ${NO_COLOR}"
        exit 1
    fi

    create_iso
}

create_iso () {
    echo
    echo -e "${COLOR_CYAN}###################################################################################################################${NO_COLOR}"
    echo -e "${COLOR_WHITE}Creating remastered k3os ${K3OS_RELEASE_VERSION} ISO file ${NO_COLOR}"
    echo

    mkdir -p /source
    mkdir -p /iso/boot/grub
    mount -o loop /tmp/$K3OS_FILENAME /source

    cp -rf /source/k3os /iso/
    cp -f config.yml /iso/k3os/system/config.yaml
    cp -f grub.cfg /iso/boot/grub/grub.cfg
    grub-mkrescue -o ${K3OS_REMASTERED_FILENAME} /iso/ -- -volid K3OS

    if [[ ! -f ./${K3OS_REMASTERED_FILENAME} ]]; then
        echo -e "${COLOR_RED}Remastered ISO file could not be created. ${NO_COLOR}"
        exit 1
    fi

    echo
    echo -e "${COLOR_CYAN}###################################################################################################################${NO_COLOR}"
    echo -e "${COLOR_WHITE}Remastered k3os ISO file ${K3OS_REMASTERED_FILENAME} successfully created. ${NO_COLOR}"
    echo
}

main "$@"; exit