mqtt.sh 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159
  1. #!/usr/bin/env bash
  2. YW=`echo "\033[33m"`
  3. BL=`echo "\033[36m"`
  4. RD=`echo "\033[01;31m"`
  5. CM='\xE2\x9C\x94\033'
  6. GN=`echo "\033[1;92m"`
  7. CL=`echo "\033[m"`
  8. while true; do
  9. read -p "This will create a New MQTT LXC. Proceed(y/n)?" yn
  10. case $yn in
  11. [Yy]* ) break;;
  12. [Nn]* ) exit;;
  13. * ) echo "Please answer yes or no.";;
  14. esac
  15. done
  16. clear
  17. function header_info {
  18. echo -e "${BL}
  19. __ __ ____ _______ _______
  20. | \/ |/ __ \__ __|__ __|
  21. | \ / | | | | | | | |
  22. | |\/| | | | | | | | |
  23. | | | | |__| | | | | |
  24. |_| |_|\___\_\ |_| |_|
  25. ${CL}"
  26. }
  27. header_info
  28. show_menu(){
  29. printf " ${YW} 1)${YW} Privileged ${CL}\n"
  30. printf " ${YW} 2)${GN} Unprivileged ${CL}\n"
  31. printf "Please choose a Install Method and hit enter or ${RD}x${CL} to exit."
  32. read opt
  33. }
  34. option_picked(){
  35. message=${@:-"${CL}Error: No message passed"}
  36. printf " ${YW}${message}${CL}\n"
  37. }
  38. show_menu
  39. while [ $opt != '' ]
  40. do
  41. if [ $opt = '' ]; then
  42. exit;
  43. else
  44. case $opt in
  45. 1) clear;
  46. header_info;
  47. option_picked "Using Privileged Install";
  48. IM=0
  49. break;
  50. ;;
  51. 2) clear;
  52. header_info;
  53. option_picked "Using Unprivileged Install";
  54. IM=1
  55. break;
  56. ;;
  57. x)exit;
  58. ;;
  59. \n)exit;
  60. ;;
  61. *)clear;
  62. option_picked "Please choose a Install Method from the menu";
  63. show_menu;
  64. ;;
  65. esac
  66. fi
  67. done
  68. set -o errexit
  69. set -o errtrace
  70. set -o nounset
  71. set -o pipefail
  72. shopt -s expand_aliases
  73. alias die='EXIT=$? LINE=$LINENO error_exit'
  74. trap die ERR
  75. trap cleanup EXIT
  76. function error_exit() {
  77. trap - ERR
  78. local DEFAULT='Unknown failure occured.'
  79. local REASON="\e[97m${1:-$DEFAULT}\e[39m"
  80. local FLAG="\e[91m[ERROR] \e[93m$EXIT@$LINE"
  81. msg "$FLAG $REASON"
  82. [ ! -z ${CTID-} ] && cleanup_ctid
  83. exit $EXIT
  84. }
  85. function warn() {
  86. local REASON="\e[97m$1\e[39m"
  87. local FLAG="\e[93m[WARNING]\e[39m"
  88. msg "$FLAG $REASON"
  89. }
  90. function info() {
  91. local REASON="$1"
  92. local FLAG="\e[36m[INFO]\e[39m"
  93. msg "$FLAG $REASON"
  94. }
  95. function msg() {
  96. local TEXT="$1"
  97. echo -e "$TEXT"
  98. }
  99. function cleanup_ctid() {
  100. if $(pct status $CTID &>/dev/null); then
  101. if [ "$(pct status $CTID | awk '{print $2}')" == "running" ]; then
  102. pct stop $CTID
  103. fi
  104. pct destroy $CTID
  105. elif [ "$(pvesm list $STORAGE --vmid $CTID)" != "" ]; then
  106. pvesm free $ROOTFS
  107. fi
  108. }
  109. function cleanup() {
  110. popd >/dev/null
  111. rm -rf $TEMP_DIR
  112. }
  113. if [ "$IM" == "1" ]; then
  114. FEATURES="nesting=1,keyctl=1,mknod=1"
  115. else
  116. FEATURES="nesting=1"
  117. fi
  118. TEMP_DIR=$(mktemp -d)
  119. pushd $TEMP_DIR >/dev/null
  120. export CTID=$(pvesh get /cluster/nextid)
  121. export PCT_OSTYPE=debian
  122. export PCT_OSVERSION=11
  123. export PCT_DISK_SIZE=2
  124. export PCT_OPTIONS="
  125. -features $FEATURES
  126. -hostname mqtt
  127. -net0 name=eth0,bridge=vmbr0,ip=dhcp
  128. -onboot 1
  129. -cores 1
  130. -memory 512
  131. -unprivileged ${IM}
  132. "
  133. bash -c "$(wget -qLO - https://raw.githubusercontent.com/tteck/Proxmox/main/ct/create_lxc.sh)" || exit
  134. STORAGE_TYPE=$(pvesm status -storage $(pct config $CTID | grep rootfs | awk -F ":" '{print $2}') | awk 'NR>1 {print $2}')
  135. if [ "$STORAGE_TYPE" == "zfspool" ]; then
  136. warn "Some addons may not work due to ZFS not supporting 'fallocate'."
  137. fi
  138. echo -en "${GN} Starting LXC Container... "
  139. pct start $CTID
  140. echo -e "${CM}${CL} \r"
  141. alias lxc-cmd="lxc-attach -n $CTID --"
  142. lxc-cmd bash -c "$(wget -qLO - https://raw.githubusercontent.com/tteck/Proxmox/main/setup/mqtt-install.sh)" || exit
  143. IP=$(pct exec $CTID ip a s dev eth0 | sed -n '/inet / s/\// /p' | awk '{print $2}')
  144. echo -e "${GN}Successfully created MQTT LXC to${CL} ${BL}$CTID${CL}. \n"