sqlite refactor

2026-05-08 06:28:16 +02:00
parent b94de9fa7e
commit 1411f11e7d
4 changed files with 77 additions and 17 deletions
@@ -1,6 +1,57 @@
 #!/bin/bash
-apk update && apk upgrade
-cp /music/archive.dat /download
-./yt-dlp -U
-./yt-dlp -P "/download" --download-archive "/download/archive.dat" --sleep-interval 5 -x -o ${FILE_NAME_FORMAT} --add-metadata --embed-thumbnail --embed-chapters ${PLAYLIST_URL} 
-cp /download/* /music
+#./yt-dlp -U
+
+TABLE="downloads"
+CACHE="cache"
+S_NEW=0
+S_DOWNLOADED=1
+S_MISSING_FROM_PLAYLIST=2
+S_DELETED=3
+S_FAILED=4
+
+#CREATE database
+if [ ! -f "$DB" ]; then
+    echo "Database not found. Creating database"
+    sqlite3 "$DB" "CREATE TABLE $TABLE (id INTEGER PRIMARY KEY,yt_id varchar(20) UNIQUE, file_name TEXT, status INTEGER);"
+    sqlite3 "$DB" "CREATE TABLE $CACHE (id INTEGER PRIMARY KEY,yt_id varchar(20) UNIQUE);"
+fi
+
+#Clear Cache
+sqlite3 "$DB" "DELETE FROM $CACHE ;"
+
+#Fill Cache
+./yt-dlp --get-id --flat-playlist "${PLAYLIST_URL}" | while read -r ID; do
+    #echo "Inserting video ID: $ID"
+    sqlite3 "$DB" "INSERT INTO $CACHE (yt_id) VALUES ('$ID');" 2>/dev/null
+done
+
+#Download music existing in cache but not in downloads table
+sqlite3 -list -noheader "$DB" "SELECT yt_id FROM $CACHE" | while IFS='|' read -r ID; do
+    count=$(sqlite3 "$DB" "SELECT count(*) FROM $TABLE WHERE yt_id = '$ID' AND status = $S_DOWNLOADED")
+
+    if [ $count -eq 0 ];then
+        ./yt-dlp -P "/download" --sleep-interval 5 -x -o ${FILE_NAME_FORMAT} --add-metadata --embed-thumbnail --embed-chapters $ID
+        name=$(./yt-dlp --get-filename -o ${FILE_NAME_FORMAT} $ID)
+
+        if [ $? -eq 0 ]; then
+            sqlite3 "$DB" "INSERT INTO $TABLE (yt_id,status,file_name) VALUES ('$ID', $S_DOWNLOADED,${name@Q});"
+            mv /download/* /music/
+        else
+            sqlite3 "$DB" "INSERT INTO $TABLE (yt_id,status,file_name) VALUES ('$ID', $S_FAILED,${name@Q});"
+        fi
+    fi
+done
+
+#Delete files existing in downloads but not in cache
+sqlite3 -list -noheader "$DB" "SELECT $TABLE.file_name FROM $TABLE LEFT OUTER JOIN $CACHE ON $TABLE.yt_id = $CACHE.yt_id WHERE $CACHE.ID IS NULL AND $TABLE.status != '3' " | while IFS='|' read -r file_name; do
+
+    echo "DELETING FILE WITH NAME: ${file_name@Q}"
+    sqlite3 "$DB" "UPDATE $TABLE SET status = '$S_DELETED' where file_name=${file_name@Q};"
+    echo "/music/${file_name@Q}*"
+    rm "/music/${file_name@Q}"*
+done
+#cp /music/archive.dat /download
+#./yt-dlp -U
+#./yt-dlp -P "/download" --download-archive "/download/archive.dat" --sleep-interval 5 -x -o ${FILE_NAME_FORMAT} --add-metadata --embed-thumbnail --embed-chapters ${PLAYLIST_URL} 
+#cp /download/* /music
+