sushen339
298 lines
8.3 KiB
Bash
298 lines
8.3 KiB
Bash
#!/bin/bash
|
||
# AutoUpdateJdCookie.sh - 京东Cookie自动更新工具安装脚本
|
||
|
||
set -euo pipefail
|
||
|
||
# ============================================
|
||
# 配置变量
|
||
# ============================================
|
||
readonly SCRIPT_NAME="AutoUpdateJdCookie Installer"
|
||
readonly REPO_URL="https://gh-proxy.net/github.com/icepage/AutoUpdateJdCookie.git"
|
||
readonly PROJECT_DIR="AutoUpdateJdCookie"
|
||
PYTHON_CMD=""
|
||
LOG_FILE="AutoUpdateJdCookie_install_$(date +%Y%m%d_%H%M%S).log"
|
||
readonly LOG_FILE
|
||
readonly COLOR_RED='\033[0;31m'
|
||
readonly COLOR_GREEN='\033[0;32m'
|
||
readonly COLOR_YELLOW='\033[1;33m'
|
||
readonly COLOR_BLUE='\033[0;34m'
|
||
readonly COLOR_RESET='\033[0m'
|
||
|
||
# ============================================
|
||
# 工具函数
|
||
# ============================================
|
||
|
||
log_info() {
|
||
echo -e "${COLOR_BLUE}[INFO]${COLOR_RESET} $(date '+%Y-%m-%d %H:%M:%S') - $*" | tee -a "$LOG_FILE"
|
||
}
|
||
|
||
log_success() {
|
||
echo -e "${COLOR_GREEN}[SUCCESS]${COLOR_RESET} $(date '+%Y-%m-%d %H:%M:%S') - $*" | tee -a "$LOG_FILE"
|
||
}
|
||
|
||
log_warning() {
|
||
echo -e "${COLOR_YELLOW}[WARNING]${COLOR_RESET} $(date '+%Y-%m-%d %H:%M:%S') - $*" | tee -a "$LOG_FILE"
|
||
}
|
||
|
||
log_error() {
|
||
echo -e "${COLOR_RED}[ERROR]${COLOR_RESET} $(date '+%Y-%m-%d %H:%M:%S') - $*" | tee -a "$LOG_FILE" >&2
|
||
}
|
||
|
||
run_with_progress() {
|
||
local description=$1
|
||
local command=$2
|
||
local log_file=$3
|
||
|
||
eval "$command" >> "$log_file" 2>&1 &
|
||
local pid=$!
|
||
|
||
local spin='⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏'
|
||
local i=0
|
||
|
||
echo -n "$description: "
|
||
while kill -0 $pid 2>/dev/null; do
|
||
i=$(( (i+1) % 10 ))
|
||
printf "\r%s: %b%s%b 处理中..." "$description" "${COLOR_BLUE}" "${spin:$i:1}" "${COLOR_RESET}"
|
||
sleep 0.1
|
||
done
|
||
|
||
wait $pid
|
||
local exit_code=$?
|
||
|
||
if [ $exit_code -eq 0 ]; then
|
||
printf "\r%s: %b✓%b 已完成 \n" "$description" "${COLOR_GREEN}" "${COLOR_RESET}"
|
||
else
|
||
printf "\r%s: %b✗%b 失败 \n" "$description" "${COLOR_RED}" "${COLOR_RESET}"
|
||
fi
|
||
|
||
return $exit_code
|
||
}
|
||
|
||
command_exists() {
|
||
command -v "$1" >/dev/null 2>&1
|
||
}
|
||
|
||
check_result() {
|
||
local exit_code=$?
|
||
if [ $exit_code -ne 0 ]; then
|
||
log_error "$1 失败"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# ============================================
|
||
# 安装函数
|
||
# ============================================
|
||
|
||
detect_python_version() {
|
||
log_info "检测系统 Python 版本..."
|
||
|
||
local python_candidates=("python3.13" "python3.12" "python3.11" "python3.10" "python3.9" "python3")
|
||
|
||
for py_cmd in "${python_candidates[@]}"; do
|
||
if command_exists "$py_cmd"; then
|
||
local py_version
|
||
py_version=$($py_cmd --version 2>&1 | grep -oP '\d+\.\d+')
|
||
|
||
if [[ $(echo "$py_version >= 3.8" | bc -l 2>/dev/null || echo "1") -eq 1 ]]; then
|
||
PYTHON_CMD="$py_cmd"
|
||
log_success "检测到 Python: $py_cmd (版本 $py_version)"
|
||
return 0
|
||
fi
|
||
fi
|
||
done
|
||
|
||
log_error "未找到可用的 Python 3.8+ 版本"
|
||
exit 1
|
||
}
|
||
|
||
check_system_dependencies() {
|
||
log_info "检查系统依赖..."
|
||
|
||
if ! command_exists apt; then
|
||
log_error "此脚本仅支持基于 apt 的系统（Debian/Ubuntu）"
|
||
exit 1
|
||
fi
|
||
|
||
if [ "$EUID" -ne 0 ]; then
|
||
log_warning "建议使用 root 权限运行此脚本"
|
||
read -r -p "是否继续？[y/N] " response
|
||
case "$response" in
|
||
[yY][eE][sS]|[yY])
|
||
log_info "继续执行..."
|
||
;;
|
||
*)
|
||
log_info "已取消安装"
|
||
exit 1
|
||
;;
|
||
esac
|
||
fi
|
||
|
||
log_success "系统依赖检查完成"
|
||
}
|
||
|
||
install_system_packages() {
|
||
log_info "开始安装系统包..."
|
||
|
||
echo ""
|
||
run_with_progress "📦 更新 APT 包列表" "apt update" "$LOG_FILE" || {
|
||
log_error "更新包列表失败"
|
||
exit 1
|
||
}
|
||
|
||
local base_packages=("git" "python3-pip")
|
||
|
||
echo ""
|
||
run_with_progress "📦 安装系统包 (${base_packages[*]})" "apt install -y ${base_packages[*]}" "$LOG_FILE"
|
||
check_result "基础系统包安装"
|
||
|
||
log_success "系统包安装完成"
|
||
}
|
||
|
||
clone_repository() {
|
||
log_info "开始克隆代码仓库..."
|
||
|
||
if [ -d "$PROJECT_DIR" ]; then
|
||
log_warning "项目目录 $PROJECT_DIR 已存在"
|
||
read -r -p "是否删除并重新克隆？[y/N] " response
|
||
case "$response" in
|
||
[yY][eE][sS]|[yY])
|
||
log_info "正在删除旧目录..."
|
||
rm -rf "$PROJECT_DIR"
|
||
;;
|
||
*)
|
||
log_warning "跳过克隆步骤，使用现有项目目录"
|
||
return 0
|
||
;;
|
||
esac
|
||
fi
|
||
|
||
echo ""
|
||
run_with_progress "📥 克隆仓库" "git clone \"$REPO_URL\" \"$PROJECT_DIR\"" "$LOG_FILE"
|
||
check_result "克隆代码仓库"
|
||
log_success "代码仓库克隆完成"
|
||
}
|
||
|
||
install_python_dependencies() {
|
||
log_info "开始安装 Python 依赖..."
|
||
|
||
cd "$PROJECT_DIR" || {
|
||
log_error "无法进入项目目录: $PROJECT_DIR"
|
||
exit 1
|
||
}
|
||
|
||
log_info "使用系统级安装模式"
|
||
|
||
echo ""
|
||
run_with_progress "🔧 升级 pip" "$PYTHON_CMD -m pip install --upgrade pip --break-system-packages" "$LOG_FILE"
|
||
check_result "升级 pip"
|
||
|
||
if [ -f "requirements.txt" ]; then
|
||
echo ""
|
||
run_with_progress "📦 安装 Python 依赖" "$PYTHON_CMD -m pip install -r requirements.txt --break-system-packages" "$LOG_FILE"
|
||
check_result "安装 Python 依赖"
|
||
else
|
||
log_warning "未找到 requirements.txt，跳过依赖安装"
|
||
fi
|
||
|
||
echo ""
|
||
run_with_progress "🔧 安装 OpenCV" "$PYTHON_CMD -m pip install opencv-python --break-system-packages" "$LOG_FILE"
|
||
check_result "安装 opencv-python"
|
||
|
||
log_success "Python 依赖安装完成"
|
||
}
|
||
|
||
install_playwright() {
|
||
log_info "开始安装 Playwright 和浏览器..."
|
||
|
||
echo ""
|
||
run_with_progress "🌐 安装 Playwright 系统依赖" "playwright install-deps" "../$LOG_FILE"
|
||
check_result "安装 Playwright 系统依赖"
|
||
|
||
echo ""
|
||
run_with_progress "🌐 安装 Chromium 浏览器" "playwright install chromium" "../$LOG_FILE"
|
||
check_result "安装 Chromium 浏览器"
|
||
|
||
log_success "Playwright 安装完成"
|
||
}
|
||
|
||
generate_config() {
|
||
log_info "开始生成配置文件..."
|
||
|
||
if [ ! -f "make_config.py" ]; then
|
||
log_error "make_config.py 文件不存在"
|
||
exit 1
|
||
fi
|
||
|
||
if [ -f "config.py" ]; then
|
||
log_warning "配置文件 config.py 已存在"
|
||
read -r -p "是否重新生成配置？[y/N] " response
|
||
case "$response" in
|
||
[yY][eE][sS]|[yY])
|
||
log_info "将重新生成配置文件"
|
||
;;
|
||
*)
|
||
log_warning "跳过配置生成步骤"
|
||
return 0
|
||
;;
|
||
esac
|
||
fi
|
||
|
||
echo ""
|
||
echo "============================================"
|
||
echo "⚙️ 开始配置向导（需要交互式输入）"
|
||
echo "============================================"
|
||
echo ""
|
||
|
||
$PYTHON_CMD make_config.py 2>&1 | tee -a "../$LOG_FILE"
|
||
|
||
if [ "${PIPESTATUS[0]}" -eq 0 ]; then
|
||
echo ""
|
||
log_success "配置文件生成完成"
|
||
else
|
||
echo ""
|
||
log_error "生成配置文件失败"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
show_post_install_info() {
|
||
echo ""
|
||
echo "============================================"
|
||
log_success "$SCRIPT_NAME 安装完成！"
|
||
echo "============================================"
|
||
echo ""
|
||
echo "项目目录: $(pwd)"
|
||
echo "Python 版本: $PYTHON_CMD"
|
||
echo "日志文件: ../$LOG_FILE"
|
||
echo ""
|
||
echo "使用说明："
|
||
echo "1. 进入项目目录: cd $PROJECT_DIR"
|
||
echo "2. 单次运行: python main.py"
|
||
echo "3. 常驻进程: python schedule_main.py"
|
||
echo ""
|
||
echo ""
|
||
echo "============================================"
|
||
}
|
||
|
||
# ============================================
|
||
# 主函数
|
||
# ============================================
|
||
main() {
|
||
log_info "开始执行 $SCRIPT_NAME..."
|
||
log_info "日志文件: $LOG_FILE"
|
||
|
||
check_system_dependencies
|
||
install_system_packages
|
||
detect_python_version
|
||
clone_repository
|
||
install_python_dependencies
|
||
install_playwright
|
||
generate_config
|
||
show_post_install_info
|
||
|
||
log_success "所有安装步骤完成！"
|
||
}
|
||
|
||
main "$@"
|