175 lines
4.1 KiB
Bash
175 lines
4.1 KiB
Bash
#!/usr/bin/env bash
|
||
|
||
set -Eeuo pipefail
|
||
|
||
REPO_ARCHIVE_URL="https://gitea.tohub.top/Share/AI_English/archive/main.tar.gz"
|
||
TARGET_DIR="/root/data/docker_data/Nginx/html/english/ai"
|
||
PRESERVED_NAME="data"
|
||
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[0;33m'
|
||
BLUE='\033[0;34m'
|
||
CYAN='\033[0;36m'
|
||
NC='\033[0m'
|
||
|
||
print_info() {
|
||
echo -e "${BLUE}[INFO]${NC} $1"
|
||
}
|
||
|
||
print_success() {
|
||
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||
}
|
||
|
||
print_warning() {
|
||
echo -e "${YELLOW}[WARNING]${NC} $1"
|
||
}
|
||
|
||
print_error() {
|
||
echo -e "${RED}[ERROR]${NC} $1" >&2
|
||
}
|
||
|
||
check_root() {
|
||
if [[ ${EUID} -ne 0 ]]; then
|
||
print_error "此脚本需要 root 权限运行"
|
||
print_info "请切换到 root 用户，或使用 sudo bash 执行脚本"
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
check_dependencies() {
|
||
local command_name
|
||
for command_name in curl tar find; do
|
||
if ! command -v "$command_name" >/dev/null 2>&1; then
|
||
print_error "缺少必要命令: $command_name"
|
||
exit 1
|
||
fi
|
||
done
|
||
}
|
||
|
||
prepare_target() {
|
||
if [[ -e "$TARGET_DIR" && ! -d "$TARGET_DIR" ]]; then
|
||
print_error "目标路径存在，但不是文件夹: $TARGET_DIR"
|
||
exit 1
|
||
fi
|
||
|
||
mkdir -p "$TARGET_DIR"
|
||
}
|
||
|
||
clear_target() {
|
||
prepare_target
|
||
print_info "正在清空目标目录（保留 $TARGET_DIR/$PRESERVED_NAME/）..."
|
||
|
||
find "$TARGET_DIR" -mindepth 1 -maxdepth 1 ! -name "$PRESERVED_NAME" \
|
||
-exec rm -rf -- {} +
|
||
|
||
print_success "目标目录已清空，$PRESERVED_NAME/ 中的内容未改动"
|
||
}
|
||
|
||
deploy() (
|
||
local temp_dir archive_file source_dir
|
||
temp_dir=$(mktemp -d)
|
||
archive_file="$temp_dir/repository.tar.gz"
|
||
trap 'rm -rf -- "$temp_dir"' EXIT
|
||
|
||
print_info "正在从 Gitea 下载最新版本..."
|
||
if ! curl -fL --retry 3 --connect-timeout 15 "$REPO_ARCHIVE_URL" -o "$archive_file"; then
|
||
print_error "仓库下载失败，请检查网络连接后重试"
|
||
return 1
|
||
fi
|
||
|
||
print_info "正在解压并验证部署文件..."
|
||
if ! tar -xzf "$archive_file" -C "$temp_dir"; then
|
||
print_error "仓库压缩包解压失败"
|
||
return 1
|
||
fi
|
||
|
||
source_dir=$(find "$temp_dir" -mindepth 2 -maxdepth 2 -type d -name src -print -quit)
|
||
if [[ -z "$source_dir" || ! -f "$source_dir/index.html" || ! -d "$source_dir/js" || ! -d "$source_dir/styles" ]]; then
|
||
print_error "仓库中缺少 src/index.html、src/js 或 src/styles，已取消部署"
|
||
return 1
|
||
fi
|
||
|
||
clear_target
|
||
|
||
print_info "正在拷贝网站文件到 $TARGET_DIR ..."
|
||
cp -a "$source_dir/index.html" "$source_dir/js" "$source_dir/styles" "$TARGET_DIR/"
|
||
|
||
print_success "部署完成"
|
||
print_info "已部署: index.html、js/、styles/"
|
||
print_info "已保留: $TARGET_DIR/$PRESERVED_NAME/"
|
||
)
|
||
|
||
confirm_clear() {
|
||
local answer
|
||
print_warning "将清空 $TARGET_DIR 中除 $PRESERVED_NAME/ 外的所有内容"
|
||
read -r -p "确认继续？[y/N]: " answer
|
||
if [[ "$answer" =~ ^[Yy]$ ]]; then
|
||
clear_target
|
||
else
|
||
print_info "已取消清空"
|
||
fi
|
||
}
|
||
|
||
show_menu() {
|
||
clear 2>/dev/null || true
|
||
echo -e "${CYAN}============================================${NC}"
|
||
echo " AI English 一键部署脚本"
|
||
echo -e "${CYAN}============================================${NC}"
|
||
echo -e "${GREEN}1.${NC} 下载并部署最新版本"
|
||
echo -e "${GREEN}2.${NC} 清空网站文件（保留 data/）"
|
||
echo -e "${GREEN}0.${NC} 退出"
|
||
echo ""
|
||
}
|
||
|
||
run_menu() {
|
||
local choice
|
||
while true; do
|
||
show_menu
|
||
read -r -p "请选择操作 [0-2]: " choice
|
||
case "$choice" in
|
||
1)
|
||
# 部署失败时回到菜单，而不是因 set -e 直接退出脚本
|
||
deploy || print_warning "部署失败，可稍后重试"
|
||
read -r -p "按 Enter 键继续..."
|
||
;;
|
||
2)
|
||
confirm_clear
|
||
read -r -p "按 Enter 键继续..."
|
||
;;
|
||
0)
|
||
print_info "退出脚本"
|
||
return 0
|
||
;;
|
||
*)
|
||
print_error "无效选择，请重新输入"
|
||
sleep 1
|
||
;;
|
||
esac
|
||
done
|
||
}
|
||
|
||
main() {
|
||
check_root
|
||
check_dependencies
|
||
|
||
case "${1:-menu}" in
|
||
deploy)
|
||
deploy
|
||
;;
|
||
clear)
|
||
confirm_clear
|
||
;;
|
||
menu)
|
||
run_menu
|
||
;;
|
||
*)
|
||
print_error "未知参数: $1"
|
||
echo "用法: $0 [deploy|clear|menu]" >&2
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
main "$@"
|