135 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
			
		
		
	
	
			135 lines
		
	
	
		
			4.1 KiB
		
	
	
	
		
			Bash
		
	
	
	
	
	
| #!/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'
 | |
| 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}"
 | |
| BFR="\\r\\033[K"
 | |
| HOLD="-"
 | |
| trap die ERR
 | |
| 
 | |
| 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 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
 | |
| 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 ] || die "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"
 | |
|   msg_ok "Downloading LXC Template"
 | |
|   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} )
 | |
| 
 | |
| msg_info "Creating LXC Container"
 | |
| pct create $CTID ${TEMPLATE_STORAGE}:vztmpl/${TEMPLATE} ${PCT_OPTIONS[@]} >/dev/null ||
 | |
|   die "A problem occured while trying to create container."
 | |
| msg_ok "LXC Container ${BL}$CTID${CL} ${GN}was successfully created."
 | |
| 
 | 
