From 14106565864b929c91643d989e3b26eb16d268dc Mon Sep 17 00:00:00 2001
From: tteckster <tteckster@gmail.com>
Date: Mon, 20 Mar 2023 21:31:42 -0400
Subject: [PATCH] Update function.sh

---
 misc/function.sh | 58 +++++++++++++++++++++++++++++++++++-------------
 1 file changed, 43 insertions(+), 15 deletions(-)

diff --git a/misc/function.sh b/misc/function.sh
index 55f72f50..0ccc6af6 100644
--- a/misc/function.sh
+++ b/misc/function.sh
@@ -1,3 +1,16 @@
+color() {
+  YW=$(echo "\033[33m")
+  RD=$(echo "\033[01;31m")
+  BL=$(echo "\033[36m")
+  GN=$(echo "\033[1;92m")
+  CL=$(echo "\033[m")
+  RETRY_NUM=10
+  RETRY_EVERY=3
+  CM="${GN}✓${CL}"
+  CROSS="${RD}✗${CL}"
+  BFR="\\r\\033[K"
+  HOLD="-"
+}
 
 error_handler() {
   local exit_code="$?"
@@ -58,22 +71,25 @@ install_script() {
 }
 
 setting_up_container() {
-    sed -i "/$LANG/ s/\(^# \)//" /etc/locale.gen
-    locale-gen >/dev/null
-    echo $tz > /etc/timezone
-    ln -sf /usr/share/zoneinfo/$tz /etc/localtime
-    for ((i=RETRY_NUM; i>0; i--)); do
-      if [ "$(hostname -I)" != "" ]; then
-        break
-      fi
-      echo 1>&2 -en "${CROSS}${RD} No Network! "
-      sleep $RETRY_EVERY
-    done
-    if [ "$(hostname -I)" = "" ]; then
-      echo 1>&2 -e "\n${CROSS}${RD} No Network After $RETRY_NUM Tries${CL}"
-      echo -e " 🖧  Check Network Settings"
-      exit 1
+  msg_info "Setting up Container OS"
+  sed -i "/$LANG/ s/\(^# \)//" /etc/locale.gen
+  locale-gen >/dev/null
+  echo $tz > /etc/timezone
+  ln -sf /usr/share/zoneinfo/$tz /etc/localtime
+  for ((i=RETRY_NUM; i>0; i--)); do
+    if [ "$(hostname -I)" != "" ]; then
+      break
     fi
+  echo 1>&2 -en "${CROSS}${RD} No Network! "
+  sleep $RETRY_EVERY
+  done
+  if [ "$(hostname -I)" = "" ]; then
+    echo 1>&2 -e "\n${CROSS}${RD} No Network After $RETRY_NUM Tries${CL}"
+    echo -e " 🖧  Check Network Settings"
+    exit 1
+  fi
+  msg_ok "Set up Container OS"
+  msg_ok "Network Connected: ${BL}$(hostname -I)"
 }
 
 network_ckeck() {
@@ -94,6 +110,18 @@ network_ckeck() {
   set -e
   trap 'error_handler $LINENO "$BASH_COMMAND"' ERR
 }
+update_os() {
+  msg_info "Updating Container OS"
+  $STD apt-get update
+  $STD apt-get -y upgrade
+  msg_ok "Updated Container OS"
+}
+motd() {
+echo "export TERM='xterm-256color'" >>/root/.bashrc
+echo -e "$APPLICATION LXC provided by https://tteck.github.io/Proxmox/\n" > /etc/motd
+chmod -x /etc/update-motd.d/*
+if [[ "${SSH_ROOT}" == "yes" ]]; then sed -i "s/#PermitRootLogin prohibit-password/PermitRootLogin yes/g" /etc/ssh/sshd_config; systemctl restart sshd; fi
+}
 
 customize() {
   msg_info "Customizing Container"