mirror of
https://gitee.com/amber-ce/amber-pm
synced 2026-05-14 02:00:20 +08:00
149 lines
3.7 KiB
Bash
Executable File
149 lines
3.7 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# amber-pm-base-overrider
|
|
# 功能：检查 /opt/apm-dummy/files-to-override 下的文件，并覆盖到所有包含
|
|
# /var/lib/apm/文件夹/files 的 /var/lib/apm/文件夹/files/ace-env 下
|
|
|
|
# 日志函数
|
|
log_warn() { echo -e "[\e[33mWARN\e[0m]: \e[1m$*\e[0m"; }
|
|
log_error() { echo -e "[\e[31mERROR\e[0m]: \e[1m$*\e[0m"; }
|
|
log_info() { echo -e "[\e[96mINFO\e[0m]: \e[1m$*\e[0m"; }
|
|
log_debug() { echo -e "[\e[32mDEBUG\e[0m]: \e[1m$*\e[0m"; }
|
|
|
|
# 配置
|
|
OVERRIDE_SOURCE="/opt/apm-dummy/files-to-override"
|
|
APM_BASE_DIR="/var/lib/apm"
|
|
|
|
# 显示帮助信息
|
|
show_help() {
|
|
cat <<EOF
|
|
amber-pm-base-overrider - APM 基础环境覆盖工具
|
|
|
|
用法:
|
|
$(basename "$0") [选项]
|
|
|
|
选项:
|
|
-h, --help 显示此帮助信息
|
|
-v, --verbose 显示详细输出
|
|
-n, --dry-run 试运行，不实际执行覆盖操作
|
|
|
|
功能:
|
|
检查 /opt/apm-dummy/files-to-override 下的文件，
|
|
并覆盖到所有 /var/lib/apm/<包名>/files/ace-env 目录下。
|
|
EOF
|
|
}
|
|
|
|
# 解析参数
|
|
VERBOSE=false
|
|
DRY_RUN=false
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case $1 in
|
|
-h|--help)
|
|
show_help
|
|
exit 0
|
|
;;
|
|
-v|--verbose)
|
|
VERBOSE=true
|
|
shift
|
|
;;
|
|
-n|--dry-run)
|
|
DRY_RUN=true
|
|
shift
|
|
;;
|
|
*)
|
|
log_error "未知选项: $1"
|
|
show_help
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# 检查源目录是否存在
|
|
if [ ! -d "$OVERRIDE_SOURCE" ]; then
|
|
log_error "源目录不存在: $OVERRIDE_SOURCE"
|
|
exit 1
|
|
fi
|
|
|
|
# 检查源目录是否为空
|
|
if [ -z "$(ls -A "$OVERRIDE_SOURCE" 2>/dev/null)" ]; then
|
|
log_warn "源目录为空: $OVERRIDE_SOURCE"
|
|
exit 0
|
|
fi
|
|
|
|
log_info "开始覆盖操作..."
|
|
log_info "源目录: $OVERRIDE_SOURCE"
|
|
|
|
# 检查 APM 基础目录是否存在
|
|
if [ ! -d "$APM_BASE_DIR" ]; then
|
|
log_error "APM 基础目录不存在: $APM_BASE_DIR"
|
|
exit 1
|
|
fi
|
|
|
|
# 计数器
|
|
OVERRIDE_COUNT=0
|
|
SKIPPED_COUNT=0
|
|
ERROR_COUNT=0
|
|
|
|
# 遍历 /var/lib/apm/ 下的所有子目录
|
|
for pkg_dir in "$APM_BASE_DIR"/*/; do
|
|
# 去掉末尾的斜杠
|
|
pkg_dir="${pkg_dir%/}"
|
|
|
|
# 检查是否是目录
|
|
if [ ! -d "$pkg_dir" ]; then
|
|
continue
|
|
fi
|
|
|
|
pkg_name=$(basename "$pkg_dir")
|
|
files_dir="$pkg_dir/files"
|
|
target_dir="$files_dir/ace-env"
|
|
|
|
# 检查是否存在 files 目录
|
|
if [ ! -d "$files_dir" ]; then
|
|
$VERBOSE && log_debug "跳过 $pkg_name: 不存在 files 目录"
|
|
continue
|
|
fi
|
|
|
|
# 检查是否存在 ace-env 目录
|
|
if [ ! -d "$target_dir" ]; then
|
|
$VERBOSE && log_debug "跳过 $pkg_name: 不存在 ace-env 目录"
|
|
continue
|
|
fi
|
|
|
|
log_info "处理包: $pkg_name -> $target_dir"
|
|
|
|
# 使用 rsync 或 cp 进行覆盖
|
|
if [ "$DRY_RUN" = true ]; then
|
|
log_info "[试运行] 将覆盖文件到: $target_dir"
|
|
find "$OVERRIDE_SOURCE" -type f | while read -r file; do
|
|
rel_path="${file#$OVERRIDE_SOURCE/}"
|
|
target_file="$target_dir/$rel_path"
|
|
echo " 将复制: $rel_path -> $target_file"
|
|
done
|
|
((OVERRIDE_COUNT++))
|
|
else
|
|
# 使用 cp -r 进行递归复制，保留权限
|
|
if cp -rL "$OVERRIDE_SOURCE"/* "$target_dir/" 2>/dev/null; then
|
|
log_info "成功覆盖到: $target_dir"
|
|
((OVERRIDE_COUNT++))
|
|
else
|
|
log_error "覆盖失败: $target_dir"
|
|
((ERROR_COUNT++))
|
|
fi
|
|
fi
|
|
done
|
|
|
|
# 输出统计信息
|
|
log_info "===================================="
|
|
log_info "覆盖操作完成"
|
|
log_info "成功处理: $OVERRIDE_COUNT 个包"
|
|
if [ "$DRY_RUN" = true ]; then
|
|
log_info "[试运行模式] 未实际执行覆盖"
|
|
fi
|
|
if [ $ERROR_COUNT -gt 0 ]; then
|
|
log_error "失败: $ERROR_COUNT 个包"
|
|
fi
|
|
|
|
exit 0
|