module.sh

#!/bin/bash
#
# ——————————————————————————————————————————————————————————————————————————————————
# __   ___                                    _ _     _   
# \ \ / (_)                             /\   | (_)   | |  
#  \ V / _  __ _  ___  _   _  __ _     /  \  | |_ ___| |_ 
#   > < | |/ _` |/ _ \| | | |/ _` |   / /\ \ | | / __| __|
#  / . \| | (_| | (_) | |_| | (_| |  / ____ \| | \__ \ |_ 
# /_/ \_\_|\__,_|\___/ \__, |\__,_| /_/    \_\_|_|___/\__|
#                       __/ |                             
#                      |___/                              
#
# Copyright (c) 2023 DDSRem <https://blog.ddsrem.com>
#
# This is free software, licensed under the Mit License.
#
# ——————————————————————————————————————————————————————————————————————————————————
#
# https://github.com/duckeaty/update_xiaoya_emby_config_library

Green="\033[32m"
Red="\033[31m"
Yellow='\033[33m'
Font="\033[0m"
INFO="[${Green}INFO${Font}]"
ERROR="[${Red}ERROR${Font}]"
WARN="[${Yellow}WARN${Font}]"
function INFO() {
    echo -e "${INFO} ${1}"
}
function ERROR() {
    echo -e "${ERROR} ${1}"
}
function WARN() {
    echo -e "${WARN} ${1}"
}

emby_config_data=/data/config/data
emby_config_data_new=/data/config_data

function update_policy(){
    clear
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
    INFO "开始获取EMBY用户信息"
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
    USER_URL="${EMBY_URL}/Users?api_key=${EMBY_API}"  
    response=$(curl -s "${USER_URL}")  
    USER_COUNT=$(echo "${response}" | jq '. | length')
    for(( i=0 ; i < $USER_COUNT ; i++ ))
    do
        read -r name <<< "$(echo "${response}" | jq -r ".[$i].Name")"  # 使用read命令读取名字  
        read -r id <<< "$(echo "${response}" | jq -r ".[$i].Id")"  # 使用read命令读取ID
        read -r policy <<< "$(echo "${response}" | jq -r ".[$i].Policy | to_entries | from_entries | tojson")"
        USER_URL_2="${EMBY_URL}/Users/$id/Policy?api_key=${EMBY_API}"
        curl -i -H "Content-Type: application/json" -X POST -d "$policy" "$USER_URL_2"
        INFO "【$name】用户策略更新成功！"
        echo -e ""
        echo -e "——————————————————————————————————————————————————————————————————————————————————"
    done
    INFO "所有用户策略更新成功！"
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
}

function update_config(){
    clear
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
    INFO "小雅EMBY_CONFIG同步"
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
    INFO "开始更新CONFIG"
    docker stop ${EMBY_NAME}
    if [ -f /root/xiaoya_emby_library_user.sql ]; then
        rm -f /root/xiaoya_emby_library_user.sql
    fi
    if [ ! -d ${emby_config_data_new} ]; then
        mkdir -p ${emby_config_data_new}
        cp -rf ${emby_config_data}/* ${emby_config_data_new}/
    fi
    if [ ! -d ${emby_config_data_new}/library_bak ]; then
        mkdir -p ${emby_config_data_new}/library_bak
    fi
    sqlite3 ${emby_config_data_new}/library.db ".dump UserDatas" > /root/xiaoya_emby_library_user.sql
    sqlite3 ${emby_config_data_new}/library.db ".dump ItemExtradata" > /root/xiaoya_emby_library_mediaconfig.sql
    mv -f ${emby_config_data_new}/library.db* ${emby_config_data_new}/library_bak/
    cp -f ${emby_config_data}/library.db ${emby_config_data_new}/
    sqlite3 ${emby_config_data_new}/library.db "DROP TABLE IF EXISTS UserDatas;"
    sqlite3 ${emby_config_data_new}/library.db ".read /root/xiaoya_emby_library_user.sql"
    sqlite3 ${emby_config_data_new}/library.db "DROP TABLE IF EXISTS ItemExtradata;"
    sqlite3 ${emby_config_data_new}/library.db ".read /root/xiaoya_emby_library_mediaconfig.sql"
    chmod 777 ${emby_config_data_new}/library.db*
    docker start ${EMBY_NAME}
    echo -e "——————————————————————————————————————————————————————————————————————————————————"
    INFO "正在重启EMBY..."
    SINCE_TIME=$(date +"%Y-%m-%dT%H:%M:%S")
    CONTAINER_NAME=${EMBY_NAME}
    TARGET_LOG_LINE_SUCCESS="All entry points have started"
    TARGET_LOG_LINE_FAIL="sending all processes the KILL signal and exiting"
    while true; do
        line=$(docker logs --since "$SINCE_TIME" "$CONTAINER_NAME" | tail -n 1)
        echo $line
        if [[ "$line" == *"$TARGET_LOG_LINE_SUCCESS"* ]]; then
            echo -e "——————————————————————————————————————————————————————————————————————————————————"
            INFO "更新CONFIG完成"
            rm -rf ${emby_config_data_new}/library_bak
            break
        elif [[ "$line" == *"$TARGET_LOG_LINE_FAIL"* ]]; then
            echo -e "——————————————————————————————————————————————————————————————————————————————————"
            ERROR "EMBY启动失败"
            INFO "正在恢复数据库并重启EMBY"
            docker stop ${EMBY_NAME}
            rm -f ${emby_config_data_new}/library.db*
            mv -f ${emby_config_data_new}/library_bak/library.db* ${emby_config_data_new}/
            rm -rf ${emby_config_data_new}/library_bak
            docker start ${EMBY_NAME}
            INFO "已恢复数据库"
            break
        fi
        sleep 3
    done
}

$1