Create start.sh

start.sh

@@ -0,0 +1,101 @@
+#!/bin/bash
+
+# 设置默认值
+MUSIC_DIR=${MUSIC_DIR:-/music}
+DATASET_MUSIC_NAME=${DATASET_MUSIC_NAME:-"your-username/music-dataset"}
+MUSIC_TOKEN=${MUSIC_TOKEN:-""}
+BACKUP_DATASET_ID=${BACKUP_DATASET_ID:-"your-username/navidrome-backup"}
+BACKUP_INTERVAL=${BACKUP_INTERVAL:-3600}  # 默认3600秒备份一次
+
+echo "[INFO] Starting Navidrome setup"
+
+# 确保目录存在并有正确权限
+mkdir -p ${MUSIC_DIR}
+mkdir -p /data/cache
+mkdir -p /.cache
+chmod -R 755 ${MUSIC_DIR}
+chmod -R 755 /data
+chmod -R 755 /.cache
+
+# 激活Python虚拟环境
+source /venv/bin/activate
+
+# 恢复备份
+if [ -n "$BACKUP_DATASET_ID" ] && [ -n "$MUSIC_TOKEN" ]; then
+    echo "[INFO] Attempting to restore from backup..."
+    python /backup.py download "$MUSIC_TOKEN" "$BACKUP_DATASET_ID" "/data"
+fi
+
+# 下载数据集中的音乐文件到音乐目录
+if [ -n "$DATASET_MUSIC_NAME" ] && [ -n "$MUSIC_TOKEN" ]; then
+    echo "[INFO] Downloading music files from Hugging Face Dataset..."
+    python -c "
+from huggingface_hub import snapshot_download
+import os
+import shutil
+# 设置环境变量
+os.environ['HF_TOKEN'] = '${MUSIC_TOKEN}'
+music_dir = '${MUSIC_DIR}'
+dataset_name = '${DATASET_MUSIC_NAME}'
+try:
+    # 确保缓存目录存在并有正确权限
+    os.makedirs('/.cache', exist_ok=True)
+    
+    # 下载数据集
+    print(f'Downloading dataset {dataset_name} to {music_dir}')
+    snapshot_download(
+        repo_id=dataset_name,
+        repo_type='dataset',
+        local_dir=music_dir,
+        token=os.environ['HF_TOKEN']
+    )
+    print('Download completed successfully')
+except Exception as e:
+    print(f'Error downloading dataset: {str(e)}')
+    # 创建测试文件确保目录不为空
+    with open(os.path.join(music_dir, 'test.txt'), 'w') as f:
+        f.write('This is a test file')
+"
+fi
+
+# 检查音乐目录
+echo "[INFO] Checking music directory..."
+ls -la ${MUSIC_DIR}
+
+# 备份函数
+backup_data() {
+    while true; do
+        echo "[INFO] Starting backup process $(date)"
+        
+        if [ -d "/data" ]; then
+            echo "[INFO] Backing up Navidrome data to HuggingFace..."
+            python /backup.py upload "$MUSIC_TOKEN" "$BACKUP_DATASET_ID" "/data"
+        else
+            echo "[WARNING] Data directory does not exist, skipping backup..."
+        fi
+        
+        echo "[INFO] Next backup in ${BACKUP_INTERVAL} seconds..."
+        sleep $BACKUP_INTERVAL
+    done
+}
+
+# 如果设置了备份数据集ID，则启动备份进程
+if [ -n "$BACKUP_DATASET_ID" ] && [ -n "$MUSIC_TOKEN" ]; then
+    echo "[INFO] Starting backup process with dataset: ${BACKUP_DATASET_ID}"
+    backup_data &
+else
+    echo "[WARNING] Backup disabled. Set BACKUP_DATASET_ID and MUSIC_TOKEN to enable."
+fi
+
+# 找到navidrome可执行文件的路径
+NAVIDROME_PATH=$(which navidrome || find /app -name navidrome -type f 2>/dev/null | head -1)
+
+if [ -z "$NAVIDROME_PATH" ]; then
+    echo "[ERROR] Could not find navidrome executable"
+    exit 1
+else
+    echo "[INFO] Found navidrome at: $NAVIDROME_PATH"
+    # 启动Navidrome
+    echo "[INFO] Starting Navidrome..."
+    exec $NAVIDROME_PATH
+fi