133 lines
5.1 KiB
Bash
Executable File
133 lines
5.1 KiB
Bash
Executable File
#!/bin/zsh
|
||
# 双击本文件，或在终端执行：./start.command
|
||
cd "$(dirname "$0")"
|
||
|
||
BACKEND_HOST="127.0.0.1"
|
||
BACKEND_PORT="8800"
|
||
BACKEND_URL="http://${BACKEND_HOST}:${BACKEND_PORT}"
|
||
STUDIO_URL="http://localhost:8900"
|
||
|
||
if [[ ! -d .venv ]]; then
|
||
echo "未找到 .venv，正在创建…"
|
||
python3 -m venv .venv || exit 1
|
||
fi
|
||
|
||
source .venv/bin/activate
|
||
# 每次启动都同步依赖，requirements.txt 有更新时自动补装（已装满时耗时极短）
|
||
pip install -q -r server/requirements.txt || exit 1
|
||
|
||
if [[ ! -f .env ]]; then
|
||
echo "未找到 .env，已从 .env.example 复制，请填入 DEEPSEEK_API_KEY 后再启动。"
|
||
cp .env.example .env
|
||
echo "按回车关闭…"
|
||
read -r
|
||
exit 1
|
||
fi
|
||
|
||
# 数据结构变更由 Alembic 管理；必须先迁移，再启动会读取新字段的服务端。
|
||
if ! alembic upgrade head; then
|
||
echo "⚠️ 数据库迁移失败，停止启动以避免新旧代码与表结构不一致"
|
||
exit 1
|
||
fi
|
||
|
||
# ---- 采集插件（extensions/collector）：启动时构建一次，之后监听代码改动自动重建 ----
|
||
# 产物 .output/chrome-mv3 由 Chrome 以解压方式加载；重建后仍需去 chrome://extensions 点刷新
|
||
COLLECTOR_WATCH_PID=""
|
||
COLLECTOR_STAMP="$(mktemp -u /tmp/osk-collector-stamp.XXXXXX)"
|
||
watch_collector() {
|
||
touch "$COLLECTOR_STAMP"
|
||
while true; do
|
||
sleep 3
|
||
if [[ -n "$(find extensions/collector/src extensions/collector/wxt.config.ts -newer "$COLLECTOR_STAMP" -print 2>/dev/null)" ]]; then
|
||
touch "$COLLECTOR_STAMP"
|
||
echo "🔧 检测到插件代码改动，重新构建…"
|
||
if pnpm -C extensions/collector build >/dev/null 2>&1; then
|
||
echo "✅ 采集插件已重建（$(date +%H:%M:%S)），去 chrome://extensions 点刷新生效"
|
||
else
|
||
echo "⚠️ 采集插件重建失败（$(date +%H:%M:%S)）：可手动跑 pnpm -C extensions/collector build 看报错"
|
||
fi
|
||
fi
|
||
done
|
||
}
|
||
|
||
if [[ ! -d extensions/collector ]]; then
|
||
echo "⚠️ 未找到 extensions/collector，跳过采集插件构建"
|
||
elif ! command -v pnpm >/dev/null 2>&1; then
|
||
echo "⚠️ 未找到 pnpm，本次跳过采集插件构建（可手动：pnpm -C extensions/collector build）"
|
||
elif ! CI=true pnpm -C extensions/collector install --silent; then
|
||
echo "⚠️ 采集插件依赖安装失败，本次跳过构建"
|
||
else
|
||
echo "正在构建采集插件…"
|
||
if pnpm -C extensions/collector build; then
|
||
echo "采集插件已构建：extensions/collector/.output/chrome-mv3"
|
||
else
|
||
echo "⚠️ 采集插件构建失败，继续启动服务（可手动重试：pnpm -C extensions/collector build）"
|
||
fi
|
||
watch_collector &
|
||
COLLECTOR_WATCH_PID=$!
|
||
fi
|
||
echo
|
||
|
||
# ---- 同步并启动前端（studio：Vite 开发服务器，/api 代理到后端 8800）----
|
||
STUDIO_PID=""
|
||
WORKER_SUPERVISOR_PID=""
|
||
cleanup() {
|
||
if [[ -n "$WORKER_SUPERVISOR_PID" ]] && kill -0 "$WORKER_SUPERVISOR_PID" 2>/dev/null; then
|
||
# 先终止 Worker 子进程，再结束守护循环；避免退出 start.command 后残留处理器。
|
||
pkill -P "$WORKER_SUPERVISOR_PID" 2>/dev/null
|
||
kill "$WORKER_SUPERVISOR_PID" 2>/dev/null
|
||
fi
|
||
if [[ -n "$STUDIO_PID" ]] && kill -0 "$STUDIO_PID" 2>/dev/null; then
|
||
kill "$STUDIO_PID" 2>/dev/null
|
||
sleep 0.2
|
||
# 兜底：确保 vite 及其子进程退出
|
||
lsof -ti :8900 | xargs kill 2>/dev/null
|
||
fi
|
||
if [[ -n "$COLLECTOR_WATCH_PID" ]] && kill -0 "$COLLECTOR_WATCH_PID" 2>/dev/null; then
|
||
pkill -P "$COLLECTOR_WATCH_PID" 2>/dev/null # 可能正在跑的 build 子进程
|
||
kill "$COLLECTOR_WATCH_PID" 2>/dev/null
|
||
fi
|
||
rm -f "$COLLECTOR_STAMP" 2>/dev/null
|
||
}
|
||
trap cleanup EXIT
|
||
|
||
# ---- 持久化后台任务 Worker（与 uvicorn 独立，服务重启不会中断正在轮询的模型任务）----
|
||
# Worker 因配置/网络等未捕获错误退出时，2 秒后自动拉起；否则任务会长期停在「排队中」。
|
||
supervise_worker() {
|
||
while true; do
|
||
python server/worker.py
|
||
WORKER_EXIT_CODE=$?
|
||
echo "⚠️ 后台任务 Worker 已退出（code: $WORKER_EXIT_CODE），2 秒后自动重启…"
|
||
sleep 2
|
||
done
|
||
}
|
||
supervise_worker &
|
||
WORKER_SUPERVISOR_PID=$!
|
||
echo "后台任务 Worker 守护进程已启动（PID: $WORKER_SUPERVISOR_PID）"
|
||
|
||
if ! command -v pnpm >/dev/null 2>&1; then
|
||
echo "⚠️ 未找到 pnpm，本次跳过 studio 前端（可手动：cd studio && pnpm dev）"
|
||
# CI=true：node_modules 与 lockfile 不一致时自动重建，不弹确认
|
||
elif CI=true pnpm -C studio install --silent; then
|
||
pnpm -C studio dev & # vite：8900，启动后会自动打开浏览器
|
||
STUDIO_PID=$!
|
||
else
|
||
echo "⚠️ studio 依赖安装失败，本次只启动后端"
|
||
fi
|
||
|
||
echo
|
||
echo "前端访问地址："
|
||
echo " 工具台 v1（web/） $BACKEND_URL/ozonSeller.html"
|
||
echo " 发布工作台（studio/） $STUDIO_URL"
|
||
echo " 后端 API 文档 $BACKEND_URL/docs"
|
||
echo
|
||
echo "按 Ctrl+C 可停止全部服务"
|
||
echo
|
||
|
||
# --app-dir server 让 uvicorn 在 server/ 下找 main:app，工作目录仍是仓库根（.env 在此）
|
||
uvicorn main:app --app-dir server --reload --reload-include '*.yaml' --host "$BACKEND_HOST" --port "$BACKEND_PORT"
|
||
|
||
echo
|
||
echo "服务已停止。按回车关闭窗口…"
|
||
read -r
|