From 17326764a42574a57b178d2983408186b40688b1 Mon Sep 17 00:00:00 2001
From: "Antonio J. Delgado" <ad@susurrando.com>
Date: Fri, 21 Feb 2025 08:17:58 +0200
Subject: [PATCH] fix home

---
 classify_videos.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/classify_videos.sh b/classify_videos.sh
index 0c88013..8cb9407 100755
--- a/classify_videos.sh
+++ b/classify_videos.sh
@@ -1,4 +1,11 @@
 #!/bin/bash
+if [ -z "${HOME}" ]; then
+    if [ "$(whoami)" == "root" ]; then
+        HOME="/root"
+    else
+        HOME=$(grep "$(whoami)" /etc/passwd | awk 'BEGIN {FS=":"} {print($6)}')
+    fi
+fi
 download_dir="$(grep download_dir "${CONFIG_FILE}" | awk 'BEGIN {FS="="} {print($2)}' | sed 's/"//g')"
 if [ "${download_dir}" == "" ]; then
     download_dir="${HOME}/downloaded_youtube_videos"