使用 Rclone + Emby + OpenList 实现 OneDrive 302 直链播放

#!/bin/bash

# --- 1. 参数配置 ---
torrent_name=$1
content_dir=$2
file_hash=$7
torrent_category=$8

qb_username="admin"               # qbit用户名
qb_password="123"          # qbit密码
qb_web_url="http://localhost:8080"  # qbit webui地址 (docker映射后用localhost)
log_dir="/config/log"        # **修改:** 统一的日志目录

rclone_dest="onedrive"           # **修改:** 你的 rclone 远程名称
rclone_path="qbit"             # **修改:** 你想上传到的网盘目录
rclone_parallel="8"               # **修改:** 32太高了，8是更合理的值

auto_del_flag="rclone"
qb_version="5.1.2"

# --- 2. 你的选择 (做种 还是 自动清理?) ---
# true  = 自动清理 (不能做种)
# false = 可以做种 (不能自动清理，未来需手动删除)
leeching_mode="true"
# 是否做种智能判断逻辑：
# 如果分类名称包含 "keep" 或 "seed" (不区分大小写)，则切换为做种模式
# ${torrent_category,,} 的意思是把变量转为全小写，方便匹配
if [[ "${torrent_category,,}" == *"keep"* ]] || [[ "${torrent_category,,}" == *"seed"* ]]; then
    leeching_mode="false"
    echo "[$(date)] MODE: Category matches 'keep/seed'. Switched to SEEDING mode." >> ${log_dir}/qb_upload.log
else
    echo "[$(date)] MODE: Standard mode. Will upload and DELETE." >> ${log_dir}/qb_upload.log
fi

# --- 3. 脚本正文  ---

if [ ! -d ${log_dir} ]
then
        mkdir -p ${log_dir}
fi

# 统一的日志文件
LOG_FILE="${log_dir}/qb_upload.log"

# 检查 qb 版本
version=$(echo $qb_version | grep -P -o "([0-9]\.){2}[0-9]" | sed s/\\.//g)
if [ ${version} -gt 404 ]; then qb_v="1"; else qb_v="2"; fi


# 函数：登录
function qb_login(){
    echo "[$(date)] SCRIPT: Logging in..." >> ${LOG_FILE}
    if [ ${qb_v} == "1" ]; then
        cookie=$(curl -i --silent --header "Referer: ${qb_web_url}" --data "username=${qb_username}&password=${qb_password}" "${qb_web_url}/api/v2/auth/login" | grep -P -o 'SID=\S{32}')
    else
        cookie=$(curl -i --silent --header "Referer: ${qb_web_url}" --data "username=${qb_username}&password=${qb_password}" "${qb_web_url}/login" | grep -P -o 'SID=\S{32}')
    fi

    if [ -n "${cookie}" ]; then
        echo "[$(date)] SCRIPT: Login SUCCESS." >> ${LOG_FILE}
    else
        echo "[$(date)] SCRIPT: Login FAILED." >> ${LOG_FILE}
    fi
}

# 函数：API操作
function qb_api_actions(){
    # 1. 添加标签
    echo "[$(date)] SCRIPT: Adding tag '${auto_del_flag}'" >> ${LOG_FILE}
    if [ ${qb_v} == "1" ]; then
        curl --silent -X POST -d "hashes=${file_hash}&tags=${auto_del_flag}" "${qb_web_url}/api/v2/torrents/addTags" --cookie "${cookie}"
    else
        curl --silent -X POST -d "hashes=${file_hash}&category=${auto_del_flag}" "${qb_web_url}/command/setCategory" --cookie ${cookie}
    fi

    # 2. 根据 leeching_mode 决定是否删除
    if [ ${leeching_mode} == "true" ]; then
        echo "[$(date)] SCRIPT: leeching_mode=true. Deleting local files & torrent..." >> ${LOG_FILE}
        curl --silent -X POST -d "hashes=${file_hash}&deleteFiles=true" "${qb_web_url}/api/v2/torrents/delete" --cookie ${cookie}
    else
        echo "[$(date)] SCRIPT: leeching_mode=false. Keeping local files for seeding." >> ${LOG_FILE}
    fi
}

# 函数：Rclone Copy
function rclone_copy(){
    echo "[$(date)] RCLONE: Starting 'rclone copy' for ${type}: ${content_dir}" >> ${LOG_FILE}
    if [ ${type} == "file" ]; then
        # 运行 rclone，函数将返回 rclone 的退出代码
        /usr/bin/rclone copy --transfers ${rclone_parallel} --log-file ${LOG_FILE} "${content_dir}" ${rclone_dest}:/${rclone_path}/
    elif [ ${type} == "dir" ]; then
        # 运行 rclone，函数将返回 rclone 的退出代码
        /usr/bin/rclone copy --transfers ${rclone_parallel} --log-file ${LOG_FILE} "${content_dir}"/ ${rclone_dest}:/${rclone_path}/"${torrent_name}"/
    fi
}

# --- 4. 主逻辑  ---

echo "-----------------------------------" >> ${LOG_FILE}
echo "[$(date)] SCRIPT: Triggered for ${torrent_name} (Hash: ${file_hash})" >> ${LOG_FILE}

# 判断类型
if [ -f "${content_dir}" ]; then
   type="file"
elif [ -d "${content_dir}" ]; then
   type="dir"
else
   echo "[$(date)] SCRIPT: ERROR! Path not found: ${content_dir}" >> ${LOG_FILE}
   exit 1
fi

# 步骤 1: 尝试 Rclone Copy
rclone_copy

# 步骤 2: 检查 Rclone 的退出代码 ($?)
if [ $? -eq 0 ]; then
    # Rclone 成功 (退出代码 0)
    echo "[$(date)] RCLONE: SUCCESS. Proceeding to QB API actions." >> ${LOG_FILE}

    # 步骤 3: 只有在 Rclone 成功后，才执行登录和 API 操作
    qb_login
    if [ -n "${cookie}" ]; then
        qb_api_actions
    fi
else
    # Rclone 失败 (退出代码非 0)
    echo "[$(date)] RCLONE: FAILED! (Exit code: $?). Aborting API actions." >> ${LOG_FILE}
    echo "[$(date)] SCRIPT: Local files are KEPT. Rclone will retry on next run (if configured)." >> ${LOG_FILE}
    # 脚本退出，不执行任何 qB API 操作，保留本地文件
fi

echo "[$(date)] SCRIPT: Finished." >> ${LOG_FILE}
echo "-----------------------------------" >> ${LOG_FILE}