mirror of
https://github.com/MemeMeow-Studio/MemeMeow.git
synced 2026-10-18 13:42:36 +08:00
infstellar
332 lines
11 KiB
Bash
Executable File
332 lines
11 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# MemeMeow 的 Compose 全栈生命周期入口；不再启动宿主机进程或 tmux 窗口。
|
||
|
||
set -Eeuo pipefail
|
||
|
||
ROOT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
|
||
COMPOSE_FILE="${MEMEMEOW_COMPOSE_FILE:-$ROOT_DIR/docker-compose.yml}"
|
||
if [[ "$COMPOSE_FILE" != /* ]]; then
|
||
COMPOSE_FILE="$ROOT_DIR/$COMPOSE_FILE"
|
||
fi
|
||
|
||
SESSION_NAME="mememeow"
|
||
BACKEND_PORT="8275"
|
||
FRONTEND_PORT="5275"
|
||
VITE_HOST="${MEMEMEOW_VITE_HOST:-0.0.0.0}"
|
||
VITE_PORT="${MEMEMEOW_VITE_PORT:-$FRONTEND_PORT}"
|
||
LEGACY_BACKEND_PORT="8000"
|
||
LEGACY_FRONTEND_PORT="5173"
|
||
LOG_TAIL="${MEMEMEOW_LOG_TAIL:-200}"
|
||
START_TIMEOUT_SECONDS="${MEMEMEOW_START_TIMEOUT_SECONDS:-180}"
|
||
if [[ ! "$LOG_TAIL" =~ ^[0-9]+$ ]]; then
|
||
LOG_TAIL=200
|
||
fi
|
||
if [[ ! "$START_TIMEOUT_SECONDS" =~ ^[1-9][0-9]*$ ]]; then
|
||
START_TIMEOUT_SECONDS=180
|
||
fi
|
||
|
||
source "$ROOT_DIR/scripts/runtime-identity.sh"
|
||
|
||
print_usage() {
|
||
# 统一说明 Compose 运维命令和独立 Vite 开发模式，避免混淆两种运行入口。
|
||
printf '用法: %s [--vite|start|stop|status|logs [服务名] [Compose 日志参数...]]\n' "$0"
|
||
}
|
||
|
||
compose() {
|
||
# 优先使用当前进程的 Docker 权限；权限尚未刷新时通过 docker 组临时执行。
|
||
if docker info >/dev/null 2>&1; then
|
||
docker compose -f "$COMPOSE_FILE" "$@"
|
||
return
|
||
fi
|
||
|
||
if ! command -v sg >/dev/null 2>&1 || ! getent group docker >/dev/null 2>&1; then
|
||
echo "Docker daemon 不可访问；请确认 Docker 正在运行，并重新登录 docker 组。" >&2
|
||
return 1
|
||
fi
|
||
|
||
local command_line argument quoted
|
||
printf -v command_line 'docker compose -f %q' "$COMPOSE_FILE"
|
||
for argument in "$@"; do
|
||
printf -v quoted '%q' "$argument"
|
||
command_line+=" $quoted"
|
||
done
|
||
sg docker -c "$command_line"
|
||
}
|
||
|
||
require_compose() {
|
||
# 启动任何服务前校验 CLI、Compose 文件和插值配置，失败时不触碰旧进程。
|
||
configure_runtime_identity
|
||
if ! command -v docker >/dev/null 2>&1; then
|
||
echo "未找到 docker；请先安装 Docker Engine/CLI。" >&2
|
||
return 1
|
||
fi
|
||
if ! command -v curl >/dev/null 2>&1; then
|
||
echo "未找到 curl；start.sh 需要它执行本机 API 健康检查。" >&2
|
||
return 1
|
||
fi
|
||
if [[ ! -f "$COMPOSE_FILE" ]]; then
|
||
echo "未找到 Compose 文件: $COMPOSE_FILE" >&2
|
||
return 1
|
||
fi
|
||
if ! docker compose version >/dev/null 2>&1; then
|
||
echo "当前 Docker CLI 未提供 docker compose 子命令。" >&2
|
||
return 1
|
||
fi
|
||
if ! compose --profile app config --quiet; then
|
||
echo "Compose 配置校验失败，请检查 .env 和 $COMPOSE_FILE。" >&2
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
listen_pids() {
|
||
# 只查询 TCP 监听进程，避免误杀连接到应用端口的无关客户端。
|
||
local port="$1"
|
||
if command -v lsof >/dev/null 2>&1; then
|
||
lsof -t -nP -iTCP:"$port" -sTCP:LISTEN 2>/dev/null || true
|
||
fi
|
||
}
|
||
|
||
stop_port_processes() {
|
||
# 只处理本项目固定的旧宿主端口；Docker 代理由 Compose 自己管理。
|
||
local port="$1" pids pid deadline remaining
|
||
pids="$(listen_pids "$port")"
|
||
if [[ -z "$pids" ]]; then
|
||
if ! command -v lsof >/dev/null 2>&1 && command -v fuser >/dev/null 2>&1; then
|
||
fuser -k "${port}/tcp" >/dev/null 2>&1 || true
|
||
fi
|
||
return 0
|
||
fi
|
||
|
||
while read -r pid; do
|
||
[[ "$pid" =~ ^[0-9]+$ ]] || continue
|
||
kill "$pid" 2>/dev/null || true
|
||
done <<< "$pids"
|
||
|
||
deadline=$((SECONDS + 5))
|
||
while (( SECONDS < deadline )); do
|
||
remaining="$(listen_pids "$port")"
|
||
[[ -z "$remaining" ]] && return 0
|
||
sleep 0.2
|
||
done
|
||
|
||
# 旧开发服务器若忽略 SIGTERM，最后只对仍占用固定端口的 PID 强制收束。
|
||
remaining="$(listen_pids "$port")"
|
||
while read -r pid; do
|
||
[[ "$pid" =~ ^[0-9]+$ ]] || continue
|
||
kill -KILL "$pid" 2>/dev/null || true
|
||
done <<< "$remaining"
|
||
}
|
||
|
||
stop_tmux_session() {
|
||
# 清理迁移前遗留的 tmux 会话；没有 tmux 或会话时视为幂等成功。
|
||
if command -v tmux >/dev/null 2>&1 && tmux has-session -t "$SESSION_NAME" 2>/dev/null; then
|
||
tmux kill-session -t "$SESSION_NAME"
|
||
fi
|
||
}
|
||
|
||
compose_service_running() {
|
||
# 判断 Compose 是否已经拥有 API 端口，从而避免启动时误杀 Compose 的 Docker 代理。
|
||
local services
|
||
services="$(compose --profile app ps --status running --services 2>/dev/null || true)"
|
||
grep -Fxq "${1}" <<< "$services"
|
||
}
|
||
|
||
cleanup_legacy_instances() {
|
||
# 启动前清理旧 tmux 和宿主开发端口；正在运行的 Compose API 保持幂等不被误杀。
|
||
stop_tmux_session
|
||
stop_port_processes "$FRONTEND_PORT"
|
||
stop_port_processes "$LEGACY_BACKEND_PORT"
|
||
stop_port_processes "$LEGACY_FRONTEND_PORT"
|
||
if ! compose_service_running mememeow; then
|
||
stop_port_processes "$BACKEND_PORT"
|
||
fi
|
||
}
|
||
|
||
require_vite() {
|
||
# 启动前端开发服务器前检查 Node/npm 与前端工程，避免误触 Compose 或静默失败。
|
||
if ! command -v node >/dev/null 2>&1; then
|
||
echo "未找到 node；--vite 需要安装 Node.js。" >&2
|
||
return 1
|
||
fi
|
||
if ! command -v npm >/dev/null 2>&1; then
|
||
echo "未找到 npm；--vite 需要安装 npm。" >&2
|
||
return 1
|
||
fi
|
||
if [[ ! -f "$ROOT_DIR/frontend/package.json" ]]; then
|
||
echo "未找到前端工程: $ROOT_DIR/frontend/package.json" >&2
|
||
return 1
|
||
fi
|
||
if [[ ! "$VITE_PORT" =~ ^[1-9][0-9]*$ ]]; then
|
||
echo "MEMEMEOW_VITE_PORT 必须是正整数。" >&2
|
||
return 1
|
||
fi
|
||
}
|
||
|
||
start_vite() {
|
||
# 保持 Compose 后端独立运行，只在前台启动 Vite 以提供热更新开发入口。
|
||
require_vite
|
||
stop_tmux_session
|
||
stop_port_processes "$VITE_PORT"
|
||
if [[ ! -d "$ROOT_DIR/frontend/node_modules" ]]; then
|
||
echo "未找到前端依赖，正在运行 npm install。"
|
||
(cd "$ROOT_DIR/frontend" && npm install)
|
||
fi
|
||
printf 'Vite 开发服务器启动中: http://%s:%s\n' "$VITE_HOST" "$VITE_PORT"
|
||
printf 'API 代理目标: http://127.0.0.1:%s\n' "$BACKEND_PORT"
|
||
cd "$ROOT_DIR/frontend"
|
||
exec npm run dev -- --host "$VITE_HOST" --port "$VITE_PORT"
|
||
}
|
||
|
||
compose_snapshot() {
|
||
# 返回可脚本解析的服务状态快照，异常时交给启动诊断统一处理。
|
||
compose --profile app ps --all --format '{{.Service}}|{{.State}}|{{.Health}}|{{.ExitCode}}' 2>/dev/null || true
|
||
}
|
||
|
||
service_line() {
|
||
# 从 Compose 快照中提取单个服务，避免依赖容器自动生成的名称。
|
||
local snapshot="$1" service="$2"
|
||
awk -F'|' -v service="$service" '$1 == service { print; exit }' <<< "$snapshot"
|
||
}
|
||
|
||
stack_ready() {
|
||
# 检查数据库、Agent、视觉和 API 的运行/健康状态，并由 HTTP 探活确认 API 已接收请求。
|
||
local snapshot="$1" service line state health exit_code
|
||
for service in postgres mememeow-agent-runtime mememeow-visual; do
|
||
line="$(service_line "$snapshot" "$service")"
|
||
IFS='|' read -r _ state health exit_code <<< "$line"
|
||
[[ "$state" == "running" && "$health" == "healthy" ]] || return 1
|
||
done
|
||
|
||
line="$(service_line "$snapshot" mememeow-auto-rename)"
|
||
IFS='|' read -r _ state health exit_code <<< "$line"
|
||
[[ "$state" == "running" ]] || return 1
|
||
|
||
line="$(service_line "$snapshot" mememeow)"
|
||
IFS='|' read -r _ state health exit_code <<< "$line"
|
||
[[ "$state" == "running" ]] || return 1
|
||
[[ -z "$health" || "$health" == "healthy" ]] || return 1
|
||
curl --fail --silent --show-error --max-time 5 -o /dev/null "http://127.0.0.1:${BACKEND_PORT}/health"
|
||
}
|
||
|
||
wait_for_stack() {
|
||
# 在固定期限内等待 Compose healthcheck 与 API HTTP 同时就绪，超时保留诊断现场。
|
||
local attempts=$(( (START_TIMEOUT_SECONDS + 1) / 2 )) attempt snapshot line state exit_code
|
||
(( attempts > 0 )) || attempts=1
|
||
for ((attempt = 1; attempt <= attempts; attempt++)); do
|
||
snapshot="$(compose_snapshot)"
|
||
for service in runtime-init db-init; do
|
||
line="$(service_line "$snapshot" "$service")"
|
||
if [[ -n "$line" ]]; then
|
||
IFS='|' read -r _ state _ exit_code <<< "$line"
|
||
if [[ "$state" == "exited" && "$exit_code" != "0" ]]; then
|
||
echo "${service} 初始化容器失败（退出码 ${exit_code}）。" >&2
|
||
return 1
|
||
fi
|
||
fi
|
||
done
|
||
if stack_ready "$snapshot"; then
|
||
return 0
|
||
fi
|
||
sleep 2
|
||
done
|
||
echo "Compose 全栈在 ${START_TIMEOUT_SECONDS} 秒内未达到健康状态。" >&2
|
||
return 1
|
||
}
|
||
|
||
print_diagnostics() {
|
||
# 启动失败时同时展示服务状态和有限日志，便于定位镜像、迁移或健康检查问题。
|
||
echo "--- docker compose ps --all ---" >&2
|
||
compose --profile app ps --all >&2 || true
|
||
echo "--- docker compose logs (tail=${LOG_TAIL}) ---" >&2
|
||
compose --profile app logs --tail="$LOG_TAIL" >&2 || true
|
||
}
|
||
|
||
check_endpoint() {
|
||
# 只验证 HTTP 状态，不输出 /config 响应内容，避免运维日志携带敏感配置。
|
||
local path="$1"
|
||
curl --fail --silent --show-error --max-time 10 -o /dev/null "http://127.0.0.1:${BACKEND_PORT}${path}"
|
||
}
|
||
|
||
start_stack() {
|
||
# 构建并启动包含数据库、迁移、Agent、视觉和 API 的完整 Compose 应用。
|
||
require_compose
|
||
cleanup_legacy_instances
|
||
if ! compose --profile app up -d --build; then
|
||
echo "Compose 全栈启动失败。" >&2
|
||
print_diagnostics
|
||
return 1
|
||
fi
|
||
if ! wait_for_stack; then
|
||
print_diagnostics
|
||
return 1
|
||
fi
|
||
local endpoint
|
||
for endpoint in / /health /config; do
|
||
if ! check_endpoint "$endpoint"; then
|
||
echo "API 端点检查失败: http://127.0.0.1:${BACKEND_PORT}${endpoint}" >&2
|
||
print_diagnostics
|
||
return 1
|
||
fi
|
||
done
|
||
compose --profile app ps --all
|
||
printf 'MemeMeow Compose 全栈已启动。\n'
|
||
printf 'API: http://127.0.0.1:%s\n' "$BACKEND_PORT"
|
||
printf '状态: %s status\n' "$0"
|
||
printf '日志: %s logs\n' "$0"
|
||
}
|
||
|
||
stop_stack() {
|
||
# 先停止 Compose 服务，再收束遗留宿主端口；named volume 始终保留。
|
||
require_compose
|
||
if ! compose --profile app stop; then
|
||
echo "Compose 服务停止失败。" >&2
|
||
compose --profile app ps --all >&2 || true
|
||
return 1
|
||
fi
|
||
stop_tmux_session
|
||
stop_port_processes "$BACKEND_PORT"
|
||
stop_port_processes "$FRONTEND_PORT"
|
||
stop_port_processes "$LEGACY_BACKEND_PORT"
|
||
stop_port_processes "$LEGACY_FRONTEND_PORT"
|
||
echo "MemeMeow Compose 服务已停止，数据库、图片和 named volume 未删除。"
|
||
}
|
||
|
||
show_status() {
|
||
# 只读取 Compose 状态，不启动、停止或修改任何服务与 volume。
|
||
require_compose
|
||
compose --profile app ps --all
|
||
}
|
||
|
||
show_logs() {
|
||
# 默认显示最近日志，并把额外参数透传给 Compose 以支持服务筛选或 --follow。
|
||
require_compose
|
||
shift
|
||
compose --profile app logs --tail="$LOG_TAIL" "$@"
|
||
}
|
||
|
||
case "${1:-start}" in
|
||
--vite)
|
||
[[ "$#" -eq 1 ]] || { print_usage >&2; exit 2; }
|
||
start_vite
|
||
;;
|
||
start)
|
||
[[ "$#" -eq 1 ]] || { print_usage >&2; exit 2; }
|
||
start_stack
|
||
;;
|
||
stop)
|
||
[[ "$#" -eq 1 ]] || { print_usage >&2; exit 2; }
|
||
stop_stack
|
||
;;
|
||
status)
|
||
[[ "$#" -eq 1 ]] || { print_usage >&2; exit 2; }
|
||
show_status
|
||
;;
|
||
logs)
|
||
show_logs "$@"
|
||
;;
|
||
*)
|
||
print_usage >&2
|
||
exit 2
|
||
;;
|
||
esac
|