From dc77419b090ba87bb7ee239f1067d483cfda38c2 Mon Sep 17 00:00:00 2001
From: CanbiZ <47820557+MickLesk@users.noreply.github.com>
Date: Fri, 16 May 2025 08:54:37 +0200
Subject: [PATCH] finalize

---
 ct/babybuddy.sh              | 71 ++++++++++++++++++++++++++++--------
 install/babybuddy-install.sh |  1 +
 2 files changed, 57 insertions(+), 15 deletions(-)

diff --git a/ct/babybuddy.sh b/ct/babybuddy.sh
index 0646742..a3fadfd 100644
--- a/ct/babybuddy.sh
+++ b/ct/babybuddy.sh
@@ -5,7 +5,7 @@ source <(curl -s https://git.community-scripts.org/community-scripts/ProxmoxVED/
 # License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
 # Source:
 
-APP="BabyBuddy"
+APP="Baby Buddy"
 var_tags="${var_tags:-baby}"
 var_disk="${var_disk:-5}"
 var_cpu="${var_cpu:-2}"
@@ -20,19 +20,60 @@ color
 catch_errors
 
 function update_script() {
-    header_info
-    check_container_storage
-    check_container_resources
-    if [[ ! -d /opt/maxun ]]; then
-        msg_error "No ${APP} Installation Found!"
-        exit
-    fi
-    RELEASE=$(curl -s https://api.github.com/repos/xxxxx/xxxxx/releases/latest | grep "tag_name" | awk '{print substr($2, 2, length($2)-3) }')
-    if [[ ! -f /opt/${APP}_version.txt ]] || [[ "${RELEASE}" != "$(cat /opt/${APP}_version.txt)" ]]; then
-        msg_info "Stopping Services"
-        systemctl stop APP
-        msg_ok "Services Stopped"
-    fi
+  header_info
+  check_container_storage
+  check_container_resources
+  if [[ ! -d /opt/babybuddy ]]; then
+    msg_error "No ${APP} Installation Found!"
+    exit
+  fi
+
+  RELEASE=$(curl -fsSL https://api.github.com/repos/babybuddy/babybuddy/releases/latest | grep "tag_name" | awk '{print substr($2, 3, length($2)-4) }')
+  if [[ ! -f /opt/${APP}_version.txt ]] || [[ "${RELEASE}" != "$(cat /opt/babybuddy_version.txt)" ]]; then
+    setup_uv
+
+    msg_info "Stopping Services"
+    systemctl stop nginx
+    systemctl stop uwsgi
+    msg_ok "Services Stopped"
+
+    msg_info "Cleaning old files"
+    cp babybuddy/settings/production.py /tmp/production.py.bak
+    find . -mindepth 1 -maxdepth 1 ! -name '.venv' -exec rm -rf {} +
+    msg_ok "Cleaned old files"
+
+    msg_info "Updating ${APP} to v${RELEASE}"
+    temp_file=$(mktemp)
+    curl -fsSL "https://github.com/babybuddy/babybuddy/archive/refs/tags/v${RELEASE}.tar.gz" -o "$temp_file"
+    cd /opt/babybuddy
+    tar zxf "$temp_file" --strip-components=1 -C /opt/babybuddy
+    mv /tmp/production.py.bak babybuddy/settings/production.py
+    cd /opt/babybuddy
+    source .venv/bin/activate
+    $STD uv pip install -r requirements.txt
+    $STD python manage.py migrate
+    echo "${RELEASE}" >/opt/${APP}_version.txt
+    msg_ok "Updated ${APP} to v${RELEASE}"
+
+    msg_info "Fixing permissions"
+    chown -R www-data:www-data /opt/data
+    chmod 640 /opt/data/db.sqlite3
+    chmod 750 /opt/data
+    msg_ok "Permissions fixed"
+
+    msg_info "Starting Services"
+    systemctl start uwsgi
+    systemctl start nginx
+    msg_ok "Services Started"
+
+    msg_info "Cleaning up"
+    rm -f "$temp_file"
+    msg_ok "Cleaned"
+    msg_ok "Updated Successfully"
+  else
+    msg_ok "No update required. ${APP} is already at v${RELEASE}"
+  fi
+  exit
 }
 
 start
@@ -42,4 +83,4 @@ description
 msg_ok "Completed Successfully!\n"
 echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"
 echo -e "${INFO}${YW} Access it using the following URL:${CL}"
-echo -e "${TAB}${GATEWAY}${BGN}http://${IP}:8080${CL}"
+echo -e "${TAB}${GATEWAY}${BGN}http://${IP}${CL}"
diff --git a/install/babybuddy-install.sh b/install/babybuddy-install.sh
index 97c36d6..e0b5e09 100644
--- a/install/babybuddy-install.sh
+++ b/install/babybuddy-install.sh
@@ -67,6 +67,7 @@ EOF
   echo "Django Admin User: $DJANGO_ADMIN_USER"
   echo "Django Admin Password: $DJANGO_ADMIN_PASS"
 } >>~/babybuddy.creds
+echo "${RELEASE}" >/opt/babybuddy_version.txt
 msg_ok "Setup Django Admin"
 
 msg_info "Configuring uWSGI"