From ead86e6242be129163feb2723e0aa5cea9545397 Mon Sep 17 00:00:00 2001
From: ksurl <ksurl@users.noreply.github.com>
Date: Tue, 25 Jul 2023 16:18:45 +0000
Subject: [PATCH] validate config file extension for lock file env

---
 root/etc/s6-overlay/s6-rc.d/init-flexget-config/run | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/root/etc/s6-overlay/s6-rc.d/init-flexget-config/run b/root/etc/s6-overlay/s6-rc.d/init-flexget-config/run
index e6ad064..687fd35 100755
--- a/root/etc/s6-overlay/s6-rc.d/init-flexget-config/run
+++ b/root/etc/s6-overlay/s6-rc.d/init-flexget-config/run
@@ -6,7 +6,18 @@ mkdir -p \
     /data
 
 FG_CONFIG_FILE="${FG_CONFIG_FILE:-/config/.flexget/config.yml}"
-FG_LOCK_FILE="${FG_CONFIG_FILE/config.yml/.config-lock}"
+case "${FG_CONFIG_FILE}" in
+  *yml)
+    FG_LOCK_FILE="${FG_CONFIG_FILE/config.yml/.config-lock}"
+    ;;
+  *yaml)
+    FG_LOCK_FILE="${FG_CONFIG_FILE/config.yaml/.config-lock}"
+    ;;
+  default)
+    echo "invalid config file extension"
+    exit 1
+    ;;
+esac
 # clean up config-lock from unclean shutdown
 if [[ -f "${FG_LOCK_FILE}" ]]; then
     rm -rf "${FG_LOCK_FILE}"