Fix auto update

build/COPY_ROOT_1/opt/ai-dock/bin/preflight.d/10-default.sh

@@ -3,7 +3,7 @@
 
 function preflight_main() {
     preflight_update_fooocus
-    printf "%s" "${FOOOCUS_FLAGS}" > /etc/fooocus_flags.conf
+    printf "%s" "${FOOOCUS_ARGS}" > /etc/fooocus_args.conf
 }
 
 function preflight_update_fooocus() {

build/COPY_ROOT_1/opt/ai-dock/bin/set-fooocus-flags.sh → build/COPY_ROOT_1/opt/ai-dock/bin/set-fooocus-args.sh

@@ -1,4 +1,4 @@
 #!/bin/bash
 
-echo "$@" > /etc/fooocus_flags.conf
+echo "$@" > /etc/fooocus_args.conf
 supervisorctl restart fooocus

build/COPY_ROOT_1/opt/ai-dock/bin/supervisor-fooocus.sh

@@ -41,14 +41,14 @@ function start() {
 
     printf "Starting $SERVICE_NAME...\n"
 
-    PLATFORM_FLAGS=
+    PLATFORM_ARGS=
     if [[ $XPU_TARGET = "CPU" ]]; then
-        PLATFORM_FLAGS="--always-cpu --disable-xformers"
+        PLATFORM_ARGS="--always-cpu --disable-xformers"
     elif [[ $XPU_TARGET = "AMD_GPU" ]]; then
-        PLATFORM_FLAGS="--disable-xformers"
+        PLATFORM_ARGS="--disable-xformers"
     fi
 
-    BASE_FLAGS=""
+    BASE_ARGS=""
 
     # Delay launch until provisioning completes
     if [[ -f /run/workspace_sync || -f /run/container_config ]]; then
@@ -72,13 +72,13 @@ function start() {
     fuser -k -SIGKILL ${LISTEN_PORT}/tcp > /dev/null 2>&1 &
     wait -n
 
-    FLAGS_COMBINED="${PLATFORM_FLAGS} ${BASE_FLAGS} $(cat /etc/fooocus_flags.conf)"
+    ARGS_COMBINED="${PLATFORM_ARGS} ${BASE_ARGS} $(cat /etc/fooocus_args.conf)"
     printf "Starting %s...\n" "${SERVICE_NAME}"
 
     cd /opt/Fooocus
     source "$FOOOCUS_VENV/bin/activate"
     LD_PRELOAD=libtcmalloc.so python launch.py \
-        ${FLAGS_COMBINED} --port ${LISTEN_PORT}
+        ${ARGS_COMBINED} --port ${LISTEN_PORT}
 }
 
 start 2>&1

build/COPY_ROOT_1/opt/ai-dock/bin/update-fooocus.sh

@@ -1,11 +1,13 @@
 #!/bin/bash
 umask 002
-branch=main
 
 source /opt/ai-dock/bin/venv-set.sh fooocus
 
 if [[ -n "${FOOOCUS_BRANCH}" ]]; then
     branch="${FOOOCUS_BRANCH}"
+else
+    branch="$(curl -s https://api.github.com/repos/lllyasviel/Fooocus/tags | \
+            jq -r '.[0].name')"
 fi
 
 # -b flag has priority
@@ -16,6 +18,8 @@ do
     esac
 done
 
+[[ -n $branch ]] || echo "Failed to get update target"; exit 1
+
 printf "Updating Fooocus (${branch})...\n"
 
 cd /opt/Fooocus