From 06e3b0def23dc61d075f45c1191290cfc914455b Mon Sep 17 00:00:00 2001
From: DDSRem <73049927+DDSRem@users.noreply.github.com>
Date: Sat, 23 Nov 2024 12:25:46 +0800
Subject: [PATCH] fix(update): useless tmp directory when not updated

---
 update | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/update b/update
index 03c8cba0..a4664aa0 100644
--- a/update
+++ b/update
@@ -20,16 +20,6 @@ function WARN() {
     echo -e "${WARN} ${1}"
 }
 
-TMP_PATH=$(mktemp -d)
-if [ ! -d "${TMP_PATH}" ]; then
-    # 如果自动生成 tmp 文件夹失败则手动指定，避免出现数据丢失等情况
-    TMP_PATH=/tmp/mp_update_path
-    if [ -d /tmp/mp_update_path ]; then
-        rm -rf /tmp/mp_update_path
-    fi
-    mkdir -p /tmp/mp_update_path
-fi
-
 # 下载及解压
 function download_and_unzip() {
     local retries=0
@@ -275,6 +265,15 @@ function get_priority() {
 }
 
 if [[ "${MOVIEPILOT_AUTO_UPDATE}" = "true" ]] || [[ "${MOVIEPILOT_AUTO_UPDATE}" = "release" ]] || [[ "${MOVIEPILOT_AUTO_UPDATE}" = "dev" ]]; then
+    TMP_PATH=$(mktemp -d)
+    if [ ! -d "${TMP_PATH}" ]; then
+        # 如果自动生成 tmp 文件夹失败则手动指定，避免出现数据丢失等情况
+        TMP_PATH=/tmp/mp_update_path
+        if [ -d /tmp/mp_update_path ]; then
+            rm -rf /tmp/mp_update_path
+        fi
+        mkdir -p /tmp/mp_update_path
+    fi
     # 优先级：镜像站 > 全局 > 不代理
     # pip
     retries=0
@@ -324,6 +323,9 @@ if [[ "${MOVIEPILOT_AUTO_UPDATE}" = "true" ]] || [[ "${MOVIEPILOT_AUTO_UPDATE}"
             WARN "当前版本号获取失败，继续启动..."
         fi
     fi
+    if [ -d "${TMP_PATH}" ]; then
+        rm -rf "${TMP_PATH}"
+    fi
 elif [[ "${MOVIEPILOT_AUTO_UPDATE}" = "false" ]]; then
     INFO "程序自动升级已关闭，如需自动升级请在创建容器时设置环境变量：MOVIEPILOT_AUTO_UPDATE=release"
 else