diff --git a/unpriv/create_lxc.sh b/unpriv/create_lxc.sh new file mode 100644 index 00000000..509a73d4 --- /dev/null +++ b/unpriv/create_lxc.sh @@ -0,0 +1,121 @@ +#!/usr/bin/env bash + +set -o errexit +set -o errtrace +set -o nounset +set -o pipefail +shopt -s expand_aliases +alias die='EXIT=$? LINE=$LINENO error_exit' +BL=`echo "\033[36m"` +RD=`echo "\033[01;31m"` +CM='\xE2\x9C\x94\033' +GN=`echo "\033[1;92m"` +CL=`echo "\033[m"` +trap die ERR + +function error_exit() { + trap - ERR + local DEFAULT='Unknown failure occured.' + local REASON="\e[97m${1:-$DEFAULT}\e[39m" + local FLAG="\e[91m[ERROR] \e[93m$EXIT@$LINE" + msg "$FLAG $REASON" 1>&2 + exit $EXIT +} +function warn() { + local REASON="\e[97m$1\e[39m" + local FLAG="\e[93m[WARNING]\e[39m" + msg "$FLAG $REASON" +} +function info() { + local REASON="$1" + local FLAG="\e[36m[INFO]\e[39m" + msg "$FLAG $REASON" +} +function msg() { + local TEXT="$1" + echo -e "$TEXT" +} +function select_storage() { + local CLASS=$1 + local CONTENT + local CONTENT_LABEL + case $CLASS in + container) CONTENT='rootdir'; CONTENT_LABEL='Container';; + template) CONTENT='vztmpl'; CONTENT_LABEL='Container template';; + *) false || die "Invalid storage class.";; + esac + + local -a MENU + while read -r line; do + local TAG=$(echo $line | awk '{print $1}') + local TYPE=$(echo $line | awk '{printf "%-10s", $2}') + local FREE=$(echo $line | numfmt --field 4-6 --from-unit=K --to=iec --format %.2f | awk '{printf( "%9sB", $6)}') + local ITEM=" Type: $TYPE Free: $FREE " + local OFFSET=2 + if [[ $((${#ITEM} + $OFFSET)) -gt ${MSG_MAX_LENGTH:-} ]]; then + local MSG_MAX_LENGTH=$((${#ITEM} + $OFFSET)) + fi + MENU+=( "$TAG" "$ITEM" "OFF" ) + done < <(pvesm status -content $CONTENT | awk 'NR>1') + + if [ $((${#MENU[@]}/3)) -eq 0 ]; then + warn "'$CONTENT_LABEL' needs to be selected for at least one storage location." + die "Unable to detect valid storage location." + elif [ $((${#MENU[@]}/3)) -eq 1 ]; then + printf ${MENU[0]} + else + local STORAGE + while [ -z "${STORAGE:+x}" ]; do + STORAGE=$(whiptail --title "Storage Pools" --radiolist \ + "Which storage pool you would like to use for the ${CONTENT_LABEL,,}?\n\n" \ + 16 $(($MSG_MAX_LENGTH + 23)) 6 \ + "${MENU[@]}" 3>&1 1>&2 2>&3) || die "Menu aborted." + done + printf $STORAGE + fi +} + +[[ "${CTID:-}" ]] || die "You need to set 'CTID' variable." +[[ "${PCT_OSTYPE:-}" ]] || die "You need to set 'PCT_OSTYPE' variable." + +[ "$CTID" -ge "100" ] || die "ID cannot be less than 100." + +if pct status $CTID &>/dev/null; then + warn "ID '$CTID' is already in use." + unset CTID + die "Cannot use ID that is already in use." +fi + +TEMPLATE_STORAGE=$(select_storage template) || exit +info "Using ${BL}$TEMPLATE_STORAGE${CL} for Template Storage." + +CONTAINER_STORAGE=$(select_storage container) || exit +info "Using ${BL}$CONTAINER_STORAGE${CL} for Container Storage." + +echo -en "${GN} Updating LXC Template List... " +pveam update >/dev/null +echo -e "${CM}${CL} \r" + +TEMPLATE_SEARCH=${PCT_OSTYPE}-${PCT_OSVERSION:-} +mapfile -t TEMPLATES < <(pveam available -section system | sed -n "s/.*\($TEMPLATE_SEARCH.*\)/\1/p" | sort -t - -k 2 -V) +[ ${#TEMPLATES[@]} -gt 0 ] || die "Unable to find a template when searching for '$TEMPLATE_SEARCH'." +TEMPLATE="${TEMPLATES[-1]}" + +if ! pveam list $TEMPLATE_STORAGE | grep -q $TEMPLATE; then + echo -en "${GN} Downloading LXC Template... " + echo -e "${CM}${CL} \r" + pveam download $TEMPLATE_STORAGE $TEMPLATE >/dev/null || + die "A problem occured while downloading the LXC template." +fi + +DEFAULT_PCT_OPTIONS=( + -arch $(dpkg --print-architecture)) + +PCT_OPTIONS=( ${PCT_OPTIONS[@]:-${DEFAULT_PCT_OPTIONS[@]}} ) +[[ " ${PCT_OPTIONS[@]} " =~ " -rootfs " ]] || PCT_OPTIONS+=( -rootfs $CONTAINER_STORAGE:${PCT_DISK_SIZE:-8} ) + +echo -en "${GN} Creating LXC Container... " +pct create $CTID ${TEMPLATE_STORAGE}:vztmpl/${TEMPLATE} ${PCT_OPTIONS[@]} >/dev/null || + die "A problem occured while trying to create container." +echo -e "${CM}${CL} \r" +info "LXC Container ${BL}$CTID${CL} was successfully created."