112 lines
4.3 KiB
Bash
112 lines
4.3 KiB
Bash
#!/usr/bin/env bash
|
||
set -euo pipefail
|
||
|
||
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||
PKG_ROOT="$ROOT_DIR"
|
||
ENV_EX="$PKG_ROOT/compose/.env.example"
|
||
ENV_OUT="$PKG_ROOT/compose/.env"
|
||
|
||
info(){ echo -e "\033[34m[CONFIG]\033[0m $*"; }
|
||
err(){ echo -e "\033[31m[ERROR]\033[0m $*" >&2; }
|
||
|
||
require(){ local ok=1; for c in "$@"; do command -v "$c" >/dev/null 2>&1 || { err "缺少依赖: $c"; ok=0; }; done; [[ $ok -eq 1 ]]; }
|
||
|
||
# Compose 检测：优先 docker compose（v2），回退 docker-compose（v1）
|
||
require_compose(){
|
||
if docker compose version >/dev/null 2>&1; then return 0; fi
|
||
if command -v docker-compose >/dev/null 2>&1 && docker-compose version >/dev/null 2>&1; then return 0; fi
|
||
err "未检测到 Docker Compose，请安装 docker compose v2 或 docker-compose v1"; exit 1
|
||
}
|
||
|
||
require docker awk sed
|
||
require_compose
|
||
|
||
[[ -f "$ENV_EX" ]] || { err "缺少模板文件: $ENV_EX"; exit 1; }
|
||
|
||
# 磁盘空间检查（简化版，默认 >= 5GB）
|
||
check_disk(){ local p="$1"; local need="${2:-5120}"; local free;
|
||
free=$(df -Pm "$p" 2>/dev/null | awk 'NR==2{print $4+0}')
|
||
if [[ -z "$free" || "$free" -lt "$need" ]]; then
|
||
err "磁盘空间不足: $p 剩余 ${free:-0}MB (<${need}MB)，请清理后再继续"
|
||
fi
|
||
}
|
||
|
||
check_disk "$PKG_ROOT" 5120 || true
|
||
check_disk "/var/lib/docker" 5120 || true
|
||
|
||
cp "$ENV_EX" "$ENV_OUT"
|
||
|
||
# 读取/生成 SWARM_MANAGER_ADDR（ARM 下同样引导 Swarm + cluster-info）
|
||
SWARM_MANAGER_ADDR=${SWARM_MANAGER_ADDR:-}
|
||
if [[ -z "${SWARM_MANAGER_ADDR}" ]]; then
|
||
read -rp "请输入本机管理地址 SWARM_MANAGER_ADDR: " SWARM_MANAGER_ADDR
|
||
fi
|
||
info "SWARM_MANAGER_ADDR=$SWARM_MANAGER_ADDR"
|
||
|
||
# 校验 IP 属于本机网卡
|
||
if ! ip -o addr | awk '{print $4}' | cut -d'/' -f1 | grep -qx "$SWARM_MANAGER_ADDR"; then
|
||
err "SWARM_MANAGER_ADDR 非本机地址: $SWARM_MANAGER_ADDR"; exit 1; fi
|
||
|
||
info "开始分配服务端口（起始=20000，避免系统占用与相互冲突）"
|
||
is_port_used(){ local p="$1"; ss -tulnH 2>/dev/null | awk '{print $5}' | sed 's/.*://g' | grep -qx "$p"; }
|
||
declare -A PRESENT=() CHOSEN=() USED=()
|
||
START_PORT="${START_PORT:-20000}"; cur=$START_PORT
|
||
ORDER=(MASTER_PORT PROMETHEUS_PORT GRAFANA_PORT ALERTMANAGER_PORT \
|
||
WEB_PROXY_PORT_8080 WEB_PROXY_PORT_8081 WEB_PROXY_PORT_8082 \
|
||
WEB_PROXY_PORT_8083 WEB_PROXY_PORT_8084 WEB_PROXY_PORT_8085)
|
||
|
||
# 标记 .env.example 中实际存在的键
|
||
for key in "${ORDER[@]}"; do
|
||
if grep -q "^${key}=" "$ENV_EX"; then PRESENT[$key]=1; fi
|
||
done
|
||
|
||
next_free(){ local p="$1"; while :; do if [[ -n "${USED[$p]:-}" ]] || is_port_used "$p"; then p=$((p+1)); else echo "$p"; return; fi; done; }
|
||
|
||
for key in "${ORDER[@]}"; do
|
||
[[ -z "${PRESENT[$key]:-}" ]] && continue
|
||
p=$(next_free "$cur"); CHOSEN[$key]="$p"; USED[$p]=1; cur=$((p+1))
|
||
done
|
||
|
||
info "端口分配结果：MASTER=${CHOSEN[MASTER_PORT]:-} PROM=${CHOSEN[PROMETHEUS_PORT]:-} GRAFANA=${CHOSEN[GRAFANA_PORT]:-} ALERT=${CHOSEN[ALERTMANAGER_PORT]:-} WEB_PROXY(8080..8085)=${CHOSEN[WEB_PROXY_PORT_8080]:-}/${CHOSEN[WEB_PROXY_PORT_8081]:-}/${CHOSEN[WEB_PROXY_PORT_8082]:-}/${CHOSEN[WEB_PROXY_PORT_8083]:-}/${CHOSEN[WEB_PROXY_PORT_8084]:-}/${CHOSEN[WEB_PROXY_PORT_8085]:-}"
|
||
|
||
# 覆盖端口（按唯一化结果写回）
|
||
for key in "${ORDER[@]}"; do
|
||
val="${CHOSEN[$key]:-}"
|
||
[[ -z "$val" ]] && continue
|
||
sed -i -E "s#^$key=.*#$key=${val}#" "$ENV_OUT"
|
||
done
|
||
|
||
# 覆盖/补充 Overlay 名称
|
||
grep -q '^ARGUS_OVERLAY_NET=' "$ENV_OUT" || echo 'ARGUS_OVERLAY_NET=argus-sys-net' >> "$ENV_OUT"
|
||
|
||
# 覆盖/补充构建账户 UID/GID，避免权限问题
|
||
RUID=$(id -u)
|
||
RGID=$(id -g)
|
||
|
||
if grep -q '^ARGUS_BUILD_UID=' "$ENV_OUT"; then
|
||
sed -i -E "s#^ARGUS_BUILD_UID=.*#ARGUS_BUILD_UID=${RUID}#" "$ENV_OUT"
|
||
else
|
||
echo "ARGUS_BUILD_UID=${RUID}" >> "$ENV_OUT"
|
||
fi
|
||
|
||
if grep -q '^ARGUS_BUILD_GID=' "$ENV_OUT"; then
|
||
sed -i -E "s#^ARGUS_BUILD_GID=.*#ARGUS_BUILD_GID=${RGID}#" "$ENV_OUT"
|
||
else
|
||
echo "ARGUS_BUILD_GID=${RGID}" >> "$ENV_OUT"
|
||
fi
|
||
|
||
# 更新 cluster-info.env 的 SWARM_MANAGER_ADDR（ARM 版本也保留该机制）
|
||
CI="$PKG_ROOT/cluster-info.env"
|
||
if [[ -f "$CI" ]]; then
|
||
if grep -q '^SWARM_MANAGER_ADDR=' "$CI"; then
|
||
sed -i -E "s#^SWARM_MANAGER_ADDR=.*#SWARM_MANAGER_ADDR=${SWARM_MANAGER_ADDR}#" "$CI"
|
||
else
|
||
echo "SWARM_MANAGER_ADDR=${SWARM_MANAGER_ADDR}" >> "$CI"
|
||
fi
|
||
else
|
||
echo "SWARM_MANAGER_ADDR=${SWARM_MANAGER_ADDR}" > "$CI"
|
||
fi
|
||
|
||
info "已生成 compose/.env 并更新 cluster-info.env 的 SWARM_MANAGER_ADDR。"
|
||
info "下一步可执行: scripts/install.sh"
|