#!/usr/bin/env bash
set -euo pipefail

# 快速更新脚本：适合已经完成首次部署后的日常代码更新。
# 只做 git 更新、前端 build、重启用户级服务；不检查/安装系统依赖，不重装 Python/npm 依赖。
# 如果本次改动包含依赖、systemd 模板或部署脚本变化，请使用 scripts/deploy-ubuntu.sh。

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
ENV_FILE="${ROOT_DIR}/.env.deploy"
BACKEND_SERVICE="zk-data-agent-backend"
FRONTEND_SERVICE="zk-data-agent-frontend"
BRANCH="${1:-}"

log() {
	printf "\n\033[1;34m==>\033[0m %s\n" "$1"
}

fail() {
	printf "\033[1;31mERROR\033[0m %s\n" "$1" >&2
	exit 1
}

resolve_npm_bin() {
	if [[ -f "${ENV_FILE}" ]]; then
		# shellcheck disable=SC1090
		source "${ENV_FILE}"
	fi
	if [[ -n "${CLAW_NODE_BIN_DIR:-}" ]]; then
		export PATH="${CLAW_NODE_BIN_DIR}:${PATH}"
	fi
	local npm_bin="${CLAW_NPM_BIN:-$(command -v npm || true)}"
	if [[ -z "${npm_bin}" || ! -x "${npm_bin}" ]]; then
		fail "缺少命令：npm。请先安装 Node.js 和 npm，或执行 bash scripts/deploy-ubuntu.sh 重新记录 Node.js 路径。"
	fi
	local node_bin_dir
	node_bin_dir="$(dirname "${npm_bin}")"
	export PATH="${node_bin_dir}:${PATH}"
	if ! "${npm_bin}" --version >/dev/null 2>&1; then
		fail "npm 不可用：${npm_bin}。请确认同目录下存在 node，或重新安装 Node.js/npm。"
	fi
	printf "%s\n" "${npm_bin}"
}

cd "${ROOT_DIR}"
[[ -d .git ]] || fail "${ROOT_DIR} 不是 git 仓库。"

if [[ -z "${BRANCH}" ]]; then
	BRANCH="$(git branch --show-current)"
fi
[[ -n "${BRANCH}" ]] || fail "无法判断当前分支，请显式传入分支名。"

if [[ -n "$(git status --porcelain)" ]]; then
	fail "工作区有未提交改动。请先处理后再快速更新。"
fi

log "拉取代码：origin/${BRANCH}"
git fetch origin
git checkout "${BRANCH}"
git pull --ff-only origin "${BRANCH}"

log "构建前端"
cd "${ROOT_DIR}/frontend/app"
npm_bin="$(resolve_npm_bin)"
export PATH="$(dirname "${npm_bin}"):${PATH}"
"${npm_bin}" run build

log "重启用户服务"
systemctl --user restart "${BACKEND_SERVICE}" "${FRONTEND_SERVICE}"

log "健康检查"
sleep 1
systemctl --user --no-pager --lines=0 status "${BACKEND_SERVICE}" "${FRONTEND_SERVICE}"

echo
echo "快速更新完成。"