139 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
		
			4.2 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
| #!/usr/bin/env bash
 | |
| 
 | |
| # Copyright (c) 2021-2023 tteck
 | |
| # Author: tteck (tteckster)
 | |
| # License: MIT
 | |
| # https://github.com/tteck/Proxmox/raw/main/LICENSE
 | |
| 
 | |
| if [ "$VERBOSE" == "yes" ]; then set -x; fi
 | |
| YW=$(echo "\033[33m")
 | |
| BL=$(echo "\033[36m")
 | |
| RD=$(echo "\033[01;31m")
 | |
| GN=$(echo "\033[1;92m")
 | |
| CL=$(echo "\033[m")
 | |
| CM="${GN}✓${CL}"
 | |
| CROSS="${RD}✗${CL}"
 | |
| BFR="\\r\\033[K"
 | |
| HOLD="-"
 | |
| set -Eeuo pipefail
 | |
| trap 'error_handler $LINENO "$BASH_COMMAND"' ERR
 | |
| function error_handler() {
 | |
|   local exit_code="$?"
 | |
|   local line_number="$1"
 | |
|   local command="$2"
 | |
|   local error_message="${RD}[ERROR]${CL} in line ${RD}$line_number${CL}: exit code ${RD}$exit_code${CL}: while executing command ${YW}$command${CL}"
 | |
|   echo -e "\n$error_message\n"
 | |
| }
 | |
| function msg_info() {
 | |
|   local msg="$1"
 | |
|   echo -ne " ${HOLD} ${YW}${msg}..."
 | |
| }
 | |
| 
 | |
| function msg_ok() {
 | |
|   local msg="$1"
 | |
|   echo -e "${BFR} ${CM} ${GN}${msg}${CL}"
 | |
| }
 | |
| function msg_error() {
 | |
|   local msg="$1"
 | |
|   echo -e "${BFR} ${CROSS} ${RD}${msg}${CL}"
 | |
| }
 | |
| 
 | |
| msg_info "Validating Storage"
 | |
| VALIDCT=$(pvesm status -content rootdir | awk 'NR>1')
 | |
| if [ -z "$VALIDCT" ]; then
 | |
|   msg_error "Unable to detect a valid Container Storage location."
 | |
|   exit 1
 | |
| fi
 | |
| VALIDTMP=$(pvesm status -content vztmpl | awk 'NR>1')
 | |
| if [ -z "$VALIDTMP" ]; then
 | |
|   msg_error "Unable to detect a valid Template Storage location."
 | |
|   exit 1
 | |
| fi
 | |
| 
 | |
| 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 || exit "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 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 ${APPLICATION}?\nTo make a selection, use the Spacebar.\n" \
 | |
|         16 $(($MSG_MAX_LENGTH + 23)) 6 \
 | |
|         "${MENU[@]}" 3>&1 1>&2 2>&3) || exit "Menu aborted."
 | |
|     done
 | |
|     printf $STORAGE
 | |
|   fi
 | |
| }
 | |
| 
 | |
| [[ "${CTID:-}" ]] || exit "You need to set 'CTID' variable."
 | |
| [[ "${PCT_OSTYPE:-}" ]] || exit "You need to set 'PCT_OSTYPE' variable."
 | |
| 
 | |
| [ "$CTID" -ge "100" ] || exit "ID cannot be less than 100."
 | |
| 
 | |
| if pct status $CTID &>/dev/null; then
 | |
|   echo -e "ID '$CTID' is already in use."
 | |
|   unset CTID
 | |
|   exit "Cannot use ID that is already in use."
 | |
| fi
 | |
| 
 | |
| TEMPLATE_STORAGE=$(select_storage template) || exit
 | |
| msg_ok "Using ${BL}$TEMPLATE_STORAGE${CL} ${GN}for Template Storage."
 | |
| 
 | |
| CONTAINER_STORAGE=$(select_storage container) || exit
 | |
| msg_ok "Using ${BL}$CONTAINER_STORAGE${CL} ${GN}for Container Storage."
 | |
| 
 | |
| msg_info "Updating LXC Template List"
 | |
| pveam update >/dev/null
 | |
| msg_ok "Updated LXC Template List"
 | |
| 
 | |
| 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 ] || exit "Unable to find a template when searching for '$TEMPLATE_SEARCH'."
 | |
| TEMPLATE="${TEMPLATES[-1]}"
 | |
| 
 | |
| if ! pveam list $TEMPLATE_STORAGE | grep -q $TEMPLATE; then
 | |
|   msg_info "Downloading LXC Template"
 | |
|   pveam download $TEMPLATE_STORAGE $TEMPLATE >/dev/null ||
 | |
|     exit "A problem occured while downloading the LXC template."
 | |
|   msg_ok "Downloaded 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})
 | |
| 
 | |
| msg_info "Creating LXC Container"
 | |
| pct create $CTID ${TEMPLATE_STORAGE}:vztmpl/${TEMPLATE} ${PCT_OPTIONS[@]} >/dev/null ||
 | |
|   exit "A problem occured while trying to create container."
 | |
| msg_ok "LXC Container ${BL}$CTID${CL} ${GN}was successfully created."
 | 
