mqtt-install.sh 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. #!/usr/bin/env bash
  2. YW=`echo "\033[33m"`
  3. RD=`echo "\033[01;31m"`
  4. BL=`echo "\033[36m"`
  5. GN=`echo "\033[1;92m"`
  6. CL=`echo "\033[m"`
  7. RETRY_NUM=10
  8. RETRY_EVERY=3
  9. NUM=$RETRY_NUM
  10. CM="${GN}✓${CL}"
  11. CROSS="${RD}✗${CL}"
  12. BFR="\\r\\033[K"
  13. HOLD="-"
  14. set -o errexit
  15. set -o errtrace
  16. set -o nounset
  17. set -o pipefail
  18. shopt -s expand_aliases
  19. alias die='EXIT=$? LINE=$LINENO error_exit'
  20. trap die ERR
  21. function error_exit() {
  22. trap - ERR
  23. local reason="Unknown failure occurred."
  24. local msg="${1:-$reason}"
  25. local flag="${RD}‼ ERROR ${CL}$EXIT@$LINE"
  26. echo -e "$flag $msg" 1>&2
  27. exit $EXIT
  28. }
  29. function msg_info() {
  30. local msg="$1"
  31. echo -ne " ${HOLD} ${YW}${msg}..."
  32. }
  33. function msg_ok() {
  34. local msg="$1"
  35. echo -e "${BFR} ${CM} ${GN}${msg}${CL}"
  36. }
  37. function msg_error() {
  38. local msg="$1"
  39. echo -e "${BFR} ${CROSS} ${RD}${msg}${CL}"
  40. }
  41. msg_info "Setting up Container OS "
  42. sed -i "/$LANG/ s/\(^# \)//" /etc/locale.gen
  43. locale-gen >/dev/null
  44. while [ "$(hostname -I)" = "" ]; do
  45. 1>&2 echo -en "${CROSS}${RD} No Network! "
  46. sleep $RETRY_EVERY
  47. ((NUM--))
  48. if [ $NUM -eq 0 ]
  49. then
  50. 1>&2 echo -e "${CROSS}${RD} No Network After $RETRY_NUM Tries${CL}"
  51. exit 1
  52. fi
  53. done
  54. msg_ok "Set up Container OS"
  55. msg_ok "Network Connected: ${BL}$(hostname -I)"
  56. if nc -zw1 8.8.8.8 443; then msg_ok "Internet Connected"; else msg_error "Internet NOT Connected"; exit 1; fi;
  57. RESOLVEDIP=$(nslookup "github.com" | awk -F':' '/^Address: / { matched = 1 } matched { print $2}' | xargs)
  58. if [[ -z "$RESOLVEDIP" ]]; then msg_error "DNS Lookup Failure"; else msg_ok "DNS Resolved github.com to $RESOLVEDIP"; fi;
  59. msg_info "Updating Container OS"
  60. apt-get update &>/dev/null
  61. apt-get -y upgrade &>/dev/null
  62. msg_ok "Updated Container OS"
  63. msg_info "Installing Dependencies"
  64. apt-get install -y curl &>/dev/null
  65. apt-get install -y sudo &>/dev/null
  66. apt-get install -y gnupg &>/dev/null
  67. msg_ok "Installed Dependencies"
  68. msg_info "Installing Mosquitto MQTT Broker"
  69. wget http://repo.mosquitto.org/debian/mosquitto-repo.gpg.key &>/dev/null
  70. apt-key add mosquitto-repo.gpg.key &>/dev/null
  71. cd /etc/apt/sources.list.d/
  72. wget http://repo.mosquitto.org/debian/mosquitto-bullseye.list &>/dev/null
  73. apt-get update >/dev/null
  74. apt-get -y install mosquitto &>/dev/null
  75. apt-get -y install mosquitto-clients &>/dev/null
  76. msg_ok "Installed Mosquitto MQTT Broker"
  77. PASS=$(grep -w "root" /etc/shadow | cut -b6);
  78. if [[ $PASS != $ ]]; then
  79. msg_info "Customizing Container"
  80. rm /etc/motd
  81. rm /etc/update-motd.d/10-uname
  82. touch ~/.hushlogin
  83. GETTY_OVERRIDE="/etc/systemd/system/container-getty@1.service.d/override.conf"
  84. mkdir -p $(dirname $GETTY_OVERRIDE)
  85. cat << EOF > $GETTY_OVERRIDE
  86. [Service]
  87. ExecStart=
  88. ExecStart=-/sbin/agetty --autologin root --noclear --keep-baud tty%I 115200,38400,9600 \$TERM
  89. EOF
  90. systemctl daemon-reload
  91. systemctl restart $(basename $(dirname $GETTY_OVERRIDE) | sed 's/\.d//')
  92. msg_ok "Customized Container"
  93. fi
  94. msg_info "Cleaning up"
  95. apt-get autoremove >/dev/null
  96. apt-get autoclean >/dev/null
  97. rm -rf /var/{cache,log}/* /var/lib/apt/lists/*
  98. msg_ok "Cleaned"