Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
ab2d24a5c7 | ||
|
|
a070ad0434 | ||
|
|
3ab3e5e180 | ||
|
|
6c96cddd4e | ||
|
|
efbd0e0d87 | ||
|
|
ed0b492461 |
@@ -31,6 +31,8 @@ TALLYNOTE_INSTALL_PREFIX=./
|
|||||||
TALLYNOTE_UPDATE_METADATA_URL=https://git.awaioi.com/api/v1/repos/awaioi/TallyNote/releases/latest
|
TALLYNOTE_UPDATE_METADATA_URL=https://git.awaioi.com/api/v1/repos/awaioi/TallyNote/releases/latest
|
||||||
TALLYNOTE_UPDATE_ALLOWED_HOSTS=git.awaioi.com
|
TALLYNOTE_UPDATE_ALLOWED_HOSTS=git.awaioi.com
|
||||||
TALLYNOTE_UPDATE_MAX_MB=512
|
TALLYNOTE_UPDATE_MAX_MB=512
|
||||||
|
# Per-request timeout for update metadata, checksums, signatures, and archives.
|
||||||
|
TALLYNOTE_UPDATE_TIMEOUT_SECONDS=30
|
||||||
# SHA-256 is always required. Detached Ed25519 signatures are optional; set
|
# SHA-256 is always required. Detached Ed25519 signatures are optional; set
|
||||||
# this to true only when a root-managed public key is configured below.
|
# this to true only when a root-managed public key is configured below.
|
||||||
TALLYNOTE_UPDATE_REQUIRE_SIGNATURE=false
|
TALLYNOTE_UPDATE_REQUIRE_SIGNATURE=false
|
||||||
|
|||||||
@@ -42,7 +42,7 @@ pnpm build:next
|
|||||||
|
|
||||||
`build:next` 与 `pnpm build` 一样输出到 `dist/web`,可直接由生产 Fastify 服务提供。
|
`build:next` 与 `pnpm build` 一样输出到 `dist/web`,可直接由生产 Fastify 服务提供。
|
||||||
|
|
||||||
本地开发首次初始化管理员使用 `pnpm admin:init`。生产安装器会在首次安装时提供管理员初始化向导;如果选择稍后创建,执行 `sudo tallynote-admin-init` 即可。也可以使用 `sudo tallynote-admin-init --username admin --display-name 管理员 --generate` 生成一次性临时密码。
|
本地开发首次初始化管理员使用 `pnpm admin:init`。生产安装器会在首次安装时提供管理员初始化向导;如果选择稍后创建,执行 `sudo /usr/local/sbin/tallynote-admin-init` 即可。也可以使用 `sudo /usr/local/sbin/tallynote-admin-init --username admin --display-name 管理员 --generate` 生成一次性临时密码。
|
||||||
|
|
||||||
默认地址为 `http://127.0.0.1:3000`,开发界面为 `http://127.0.0.1:5173`。配置项见 `.env.example`。
|
默认地址为 `http://127.0.0.1:3000`,开发界面为 `http://127.0.0.1:5173`。配置项见 `.env.example`。
|
||||||
|
|
||||||
@@ -62,7 +62,13 @@ curl --proto '=https' --tlsv1.2 -fsSL https://git.awaioi.com/awaioi/TallyNote/ra
|
|||||||
curl --proto '=https' --tlsv1.2 -fsSL https://git.awaioi.com/awaioi/TallyNote/raw/branch/main/install.sh | sudo bash
|
curl --proto '=https' --tlsv1.2 -fsSL https://git.awaioi.com/awaioi/TallyNote/raw/branch/main/install.sh | sudo bash
|
||||||
```
|
```
|
||||||
|
|
||||||
首次安装完成网络配置后,向导会询问是否立即创建管理员。选择创建时,用户名、显示名称和密码都在当前 SSH 终端中输入;选择稍后创建也不会阻塞服务启动,之后执行 `sudo tallynote-admin-init` 即可。升级已有安装时,向导会自动识别现有管理员并跳过创建,不会覆盖账号或账目。
|
首次安装完成网络配置后,向导会询问是否立即创建管理员。选择创建时,用户名、显示名称和密码都在当前 SSH 终端中输入,并会直接回显当前输入内容;密码不会写入安装日志、配置文件或命令行参数。选择稍后创建也不会阻塞服务启动,之后执行 `sudo /usr/local/sbin/tallynote-admin-init` 即可。升级已有安装时,向导会自动识别现有管理员并跳过创建,不会覆盖账号或账目。
|
||||||
|
|
||||||
|
如果账号是在旧版本中用正式密码创建、但仍被标记为“首次登录需要修改密码”,可以在服务器上用当前密码修复标志位(不会更换密码):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
sudo /usr/local/sbin/tallynote-admin-init --mark-password-configured --username <用户名>
|
||||||
|
```
|
||||||
|
|
||||||
监听方式有两个选项:`127.0.0.1` 仅本机访问(默认、更安全),或 `0.0.0.0` 允许通过局域网/公网 IP 访问。安装时可输入自定义端口(直接回车使用默认端口),安装器会检查 TCP 端口是否已被占用;选择 `0.0.0.0` 时会尝试通过 HTTPS 自动获取公网 IPv4,并将 `http://公网IP:端口` 作为默认访问地址,也可以改填域名。不能填写 `http://0.0.0.0:3000`。直连 HTTP 未加密,安装器会要求明确确认,只适合受控网络。绑定域名后应改为 HTTPS 反向代理,设置真实的 `TALLYNOTE_PUBLIC_ORIGIN`、`TALLYNOTE_COOKIE_SECURE=true`、`TALLYNOTE_ALLOW_INSECURE_HTTP=false`,然后执行 `sudo systemctl restart tallynote.service`。服务启动后,安装器会先请求本机 `/health`;只有健康检查通过才会报告安装完成并输出最终访问链接。监听 `127.0.0.1` 时该链接只对服务器本机有效;需要公网或其他设备访问时请选择 `0.0.0.0`。健康检查失败时会输出 systemd 状态和最近日志并回滚本次切换。
|
监听方式有两个选项:`127.0.0.1` 仅本机访问(默认、更安全),或 `0.0.0.0` 允许通过局域网/公网 IP 访问。安装时可输入自定义端口(直接回车使用默认端口),安装器会检查 TCP 端口是否已被占用;选择 `0.0.0.0` 时会尝试通过 HTTPS 自动获取公网 IPv4,并将 `http://公网IP:端口` 作为默认访问地址,也可以改填域名。不能填写 `http://0.0.0.0:3000`。直连 HTTP 未加密,安装器会要求明确确认,只适合受控网络。绑定域名后应改为 HTTPS 反向代理,设置真实的 `TALLYNOTE_PUBLIC_ORIGIN`、`TALLYNOTE_COOKIE_SECURE=true`、`TALLYNOTE_ALLOW_INSECURE_HTTP=false`,然后执行 `sudo systemctl restart tallynote.service`。服务启动后,安装器会先请求本机 `/health`;只有健康检查通过才会报告安装完成并输出最终访问链接。监听 `127.0.0.1` 时该链接只对服务器本机有效;需要公网或其他设备访问时请选择 `0.0.0.0`。健康检查失败时会输出 systemd 状态和最近日志并回滚本次切换。
|
||||||
|
|
||||||
|
|||||||
+9
-7
@@ -230,26 +230,26 @@ run_initial_admin_wizard() {
|
|||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
if (( NON_INTERACTIVE )); then
|
if (( NON_INTERACTIVE )); then
|
||||||
log '非交互模式:跳过管理员初始化;稍后可执行 sudo tallynote-admin-init'
|
log "非交互模式:跳过管理员初始化;稍后可执行 sudo $ADMIN_INIT_PATH"
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
[[ -r "$PROMPT_INPUT" && -w "$PROMPT_OUTPUT" ]] || {
|
[[ -r "$PROMPT_INPUT" && -w "$PROMPT_OUTPUT" ]] || {
|
||||||
log '未检测到交互式终端:跳过管理员初始化;稍后可执行 sudo tallynote-admin-init'
|
log "未检测到交互式终端:跳过管理员初始化;稍后可执行 sudo $ADMIN_INIT_PATH"
|
||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
[[ -x "$ADMIN_INIT_PATH" ]] || die '管理员初始化命令未安装'
|
[[ -x "$ADMIN_INIT_PATH" ]] || die '管理员初始化命令未安装'
|
||||||
|
|
||||||
local status choice
|
local status choice
|
||||||
if ! status=$("$ADMIN_INIT_PATH" --check 2>/dev/null); then
|
if ! status=$("$ADMIN_INIT_PATH" --check 2>/dev/null); then
|
||||||
log '无法检查管理员初始化状态;基础安装已完成,稍后可执行 sudo tallynote-admin-init'
|
log "无法检查管理员初始化状态;基础安装已完成,稍后可执行 sudo $ADMIN_INIT_PATH"
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
[[ "$status" == empty ]] || return 0
|
[[ "$status" == empty ]] || return 0
|
||||||
|
|
||||||
exec 9<"$PROMPT_INPUT" || die '无法打开终端输入;请稍后执行 sudo tallynote-admin-init'
|
exec 9<"$PROMPT_INPUT" || die "无法打开终端输入;请稍后执行 sudo $ADMIN_INIT_PATH"
|
||||||
{
|
{
|
||||||
printf '\n首次安装还差一步:请创建管理员账号。\n'
|
printf '\n首次安装还差一步:请创建管理员账号。\n'
|
||||||
printf '管理员账号用于登录 TallyNote,首次登录后需要设置正式密码。\n'
|
printf '管理员账号用于登录 TallyNote;这里输入的密码会直接作为正式密码。\n'
|
||||||
} > "$PROMPT_OUTPUT"
|
} > "$PROMPT_OUTPUT"
|
||||||
while :; do
|
while :; do
|
||||||
prompt_value '现在创建管理员?输入 yes 继续,其他内容稍后创建' 'yes'
|
prompt_value '现在创建管理员?输入 yes 继续,其他内容稍后创建' 'yes'
|
||||||
@@ -258,7 +258,7 @@ run_initial_admin_wizard() {
|
|||||||
yes|YES|Yes|y|Y) break ;;
|
yes|YES|Yes|y|Y) break ;;
|
||||||
no|NO|No|n|N|'')
|
no|NO|No|n|N|'')
|
||||||
exec 9<&-
|
exec 9<&-
|
||||||
log '已跳过管理员初始化;稍后可执行 sudo tallynote-admin-init'
|
log "已跳过管理员初始化;稍后可执行 sudo $ADMIN_INIT_PATH"
|
||||||
return 0
|
return 0
|
||||||
;;
|
;;
|
||||||
*) printf '请输入 yes 或 no。\n' > "$PROMPT_OUTPUT" ;;
|
*) printf '请输入 yes 或 no。\n' > "$PROMPT_OUTPUT" ;;
|
||||||
@@ -267,7 +267,7 @@ run_initial_admin_wizard() {
|
|||||||
stage '创建首位管理员(密码不会写入安装日志)'
|
stage '创建首位管理员(密码不会写入安装日志)'
|
||||||
if ! "$ADMIN_INIT_PATH" <&9 > "$PROMPT_OUTPUT"; then
|
if ! "$ADMIN_INIT_PATH" <&9 > "$PROMPT_OUTPUT"; then
|
||||||
exec 9<&-
|
exec 9<&-
|
||||||
log '管理员初始化未完成;基础安装已完成,稍后可执行 sudo tallynote-admin-init'
|
log "管理员初始化未完成;基础安装已完成,稍后可执行 sudo $ADMIN_INIT_PATH"
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
exec 9<&-
|
exec 9<&-
|
||||||
@@ -1538,5 +1538,7 @@ main() {
|
|||||||
fi
|
fi
|
||||||
log "访问地址:$access_url"
|
log "访问地址:$access_url"
|
||||||
log '查看服务状态:systemctl status tallynote.service'
|
log '查看服务状态:systemctl status tallynote.service'
|
||||||
|
log "管理员初始化命令:sudo $ADMIN_INIT_PATH"
|
||||||
|
log '如 sudo 找不到该命令,请使用上面输出的绝对路径'
|
||||||
}
|
}
|
||||||
main "$@"
|
main "$@"
|
||||||
|
|||||||
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "tallynote",
|
"name": "tallynote",
|
||||||
"version": "1.2.3",
|
"version": "1.2.9",
|
||||||
"private": true,
|
"private": true,
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"packageManager": "pnpm@9.0.6",
|
"packageManager": "pnpm@9.0.6",
|
||||||
|
|||||||
@@ -10,6 +10,8 @@ DATA_DIR=${TALLYNOTE_DATA_DIR:-/var/lib/tallynote}
|
|||||||
REQUEST_FILE="$DATA_DIR/update-request.json"
|
REQUEST_FILE="$DATA_DIR/update-request.json"
|
||||||
CURRENT_LINK="$PREFIX/current"
|
CURRENT_LINK="$PREFIX/current"
|
||||||
STATE_FILE="$PREFIX/.update-state"
|
STATE_FILE="$PREFIX/.update-state"
|
||||||
|
LOCK_FILE="$PREFIX/.update-runner.lock"
|
||||||
|
RUNNER_LOG="$PREFIX/.update-runner.log"
|
||||||
SERVICE_NAME=${TALLYNOTE_SERVICE_NAME:-tallynote.service}
|
SERVICE_NAME=${TALLYNOTE_SERVICE_NAME:-tallynote.service}
|
||||||
HOST=${TALLYNOTE_HOST:-127.0.0.1}
|
HOST=${TALLYNOTE_HOST:-127.0.0.1}
|
||||||
PORT=${TALLYNOTE_PORT:-3000}
|
PORT=${TALLYNOTE_PORT:-3000}
|
||||||
@@ -19,13 +21,72 @@ if [[ "$HEALTH_HOST" == :: ]]; then HEALTH_HOST=::1; fi
|
|||||||
if [[ "$HEALTH_HOST" == *:* && "$HEALTH_HOST" != \[* ]]; then HEALTH_HOST="[$HEALTH_HOST]"; fi
|
if [[ "$HEALTH_HOST" == *:* && "$HEALTH_HOST" != \[* ]]; then HEALTH_HOST="[$HEALTH_HOST]"; fi
|
||||||
|
|
||||||
die() { printf 'tallynote update runner: %s\n' "$*" >&2; exit 1; }
|
die() { printf 'tallynote update runner: %s\n' "$*" >&2; exit 1; }
|
||||||
|
|
||||||
|
# The runner may exit during any of the checks below. Install its EXIT cleanup
|
||||||
|
# before doing privileged preflight so a partial invocation never leaves a
|
||||||
|
# heartbeat or lock behind.
|
||||||
|
STATE_CREATED=0
|
||||||
|
heartbeat_pid=''
|
||||||
|
heartbeat_owner=$$
|
||||||
|
RUNNER_LOCK_FD=9
|
||||||
|
RUNNER_LOCK_MODE=''
|
||||||
|
stop_heartbeat() {
|
||||||
|
if [[ -n "$heartbeat_pid" ]]; then
|
||||||
|
kill "$heartbeat_pid" 2>/dev/null || true
|
||||||
|
wait "$heartbeat_pid" 2>/dev/null || true
|
||||||
|
heartbeat_pid=''
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap
|
||||||
|
release_runner_lock() {
|
||||||
|
if [[ "$RUNNER_LOCK_MODE" == flock ]]; then
|
||||||
|
flock -u "$RUNNER_LOCK_FD" 2>/dev/null || true
|
||||||
|
eval "exec ${RUNNER_LOCK_FD}>&-" 2>/dev/null || true
|
||||||
|
elif [[ "$RUNNER_LOCK_MODE" == mkdir ]]; then
|
||||||
|
rmdir -- "$LOCK_FILE.d" 2>/dev/null || true
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap
|
||||||
|
early_cleanup() {
|
||||||
|
local result=$?
|
||||||
|
stop_heartbeat
|
||||||
|
if (( result != 0 )); then
|
||||||
|
# A preflight failure happens before the normal phase-specific trap is
|
||||||
|
# installed. Remove only the one-shot request marker; never remove an
|
||||||
|
# existing recovery marker unless this invocation created it.
|
||||||
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
|
if (( STATE_CREATED == 1 )); then rm -f -- "$STATE_FILE" 2>/dev/null || true; fi
|
||||||
|
fi
|
||||||
|
release_runner_lock
|
||||||
|
return "$result"
|
||||||
|
}
|
||||||
|
trap early_cleanup EXIT
|
||||||
|
|
||||||
[[ ${EUID:-$(id -u)} -eq 0 ]] || die 'must run as root'
|
[[ ${EUID:-$(id -u)} -eq 0 ]] || die 'must run as root'
|
||||||
|
[[ -d "$PREFIX" ]] || die 'install prefix is missing'
|
||||||
|
if command -v flock >/dev/null 2>&1; then
|
||||||
|
exec 9>"$LOCK_FILE" || die '无法打开更新运行锁'
|
||||||
|
flock -n "$RUNNER_LOCK_FD" || exit 0
|
||||||
|
RUNNER_LOCK_MODE=flock
|
||||||
|
else
|
||||||
|
# macOS development fixtures do not ship util-linux; retain an atomic lock
|
||||||
|
# fallback there while Linux production uses flock above.
|
||||||
|
mkdir "$LOCK_FILE.d" 2>/dev/null || exit 0
|
||||||
|
RUNNER_LOCK_MODE='mkdir'
|
||||||
|
fi
|
||||||
[[ -f "$REQUEST_FILE" || -f "$STATE_FILE" ]] || exit 0
|
[[ -f "$REQUEST_FILE" || -f "$STATE_FILE" ]] || exit 0
|
||||||
[[ -L "$CURRENT_LINK" ]] || die 'current release link is missing'
|
[[ -L "$CURRENT_LINK" ]] || die 'current release link is missing'
|
||||||
|
|
||||||
old_target=$(readlink -f -- "$CURRENT_LINK")
|
old_target=$(readlink -f -- "$CURRENT_LINK")
|
||||||
[[ "$old_target" == "$PREFIX/releases/"* && -d "$old_target" ]] || die 'current release target is invalid'
|
[[ "$old_target" == "$PREFIX/releases/"* && -d "$old_target" ]] || die 'current release target is invalid'
|
||||||
|
|
||||||
|
# Capture the service state before any download/apply work. The value is
|
||||||
|
# persisted in the recovery marker so a later runner process can restore the
|
||||||
|
# operator's original state after a crash (the service is normally inactive by
|
||||||
|
# the time recovery starts).
|
||||||
|
was_active=0
|
||||||
|
if systemctl is-active --quiet "$SERVICE_NAME"; then was_active=1; fi
|
||||||
|
|
||||||
request_operation='apply'
|
request_operation='apply'
|
||||||
if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
||||||
request_operation=$(sed -n 's/.*"operation"[[:space:]]*:[[:space:]]*"\(download\|apply\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
request_operation=$(sed -n 's/.*"operation"[[:space:]]*:[[:space:]]*"\(download\|apply\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
||||||
@@ -39,15 +100,11 @@ job_id=''
|
|||||||
if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
||||||
job_id=$(sed -n 's/.*"jobId"[[:space:]]*:[[:space:]]*"\([0-9a-f-]*\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
job_id=$(sed -n 's/.*"jobId"[[:space:]]*:[[:space:]]*"\([0-9a-f-]*\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
||||||
fi
|
fi
|
||||||
STATE_CREATED=0
|
|
||||||
heartbeat_pid=''
|
|
||||||
heartbeat_owner=$$
|
|
||||||
|
|
||||||
write_recovery_state() {
|
write_recovery_state() {
|
||||||
local phase=$1 temporary
|
local phase=$1 temporary
|
||||||
temporary="$PREFIX/.update-state-$$-${RANDOM}.tmp"
|
temporary="$PREFIX/.update-state-$$-${RANDOM}.tmp"
|
||||||
[[ ! -e "$temporary" && ! -L "$temporary" ]] || return 1
|
[[ ! -e "$temporary" && ! -L "$temporary" ]] || return 1
|
||||||
printf 'job_id=%s\nold_target=%s\nphase=%s\n' "$job_id" "$old_target" "$phase" > "$temporary"
|
printf 'job_id=%s\nold_target=%s\nphase=%s\ninitial_active=%s\n' "$job_id" "$old_target" "$phase" "$was_active" > "$temporary"
|
||||||
chmod 600 "$temporary"
|
chmod 600 "$temporary"
|
||||||
mv -Tf -- "$temporary" "$STATE_FILE"
|
mv -Tf -- "$temporary" "$STATE_FILE"
|
||||||
STATE_CREATED=1
|
STATE_CREATED=1
|
||||||
@@ -59,14 +116,6 @@ clear_recovery_state() {
|
|||||||
STATE_CREATED=0
|
STATE_CREATED=0
|
||||||
}
|
}
|
||||||
|
|
||||||
stop_heartbeat() {
|
|
||||||
if [[ -n "$heartbeat_pid" ]]; then
|
|
||||||
kill "$heartbeat_pid" 2>/dev/null || true
|
|
||||||
wait "$heartbeat_pid" 2>/dev/null || true
|
|
||||||
heartbeat_pid=''
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
heartbeat() {
|
heartbeat() {
|
||||||
# Keep the lease fresh during long downloads/backups, but stop on a hard
|
# Keep the lease fresh during long downloads/backups, but stop on a hard
|
||||||
# runner kill so an orphaned child cannot keep the recovery marker alive.
|
# runner kill so an orphaned child cannot keep the recovery marker alive.
|
||||||
@@ -86,9 +135,11 @@ start_heartbeat() {
|
|||||||
# This trap covers failures before the normal apply cleanup trap is installed,
|
# This trap covers failures before the normal apply cleanup trap is installed,
|
||||||
# including a missing runtime, an invalid current link, and a failed service
|
# including a missing runtime, an invalid current link, and a failed service
|
||||||
# stop. It deliberately does not remove a pre-existing recovery marker.
|
# stop. It deliberately does not remove a pre-existing recovery marker.
|
||||||
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap
|
||||||
preflight_cleanup() {
|
preflight_cleanup() {
|
||||||
local result=$?
|
local result=$?
|
||||||
stop_heartbeat
|
stop_heartbeat
|
||||||
|
release_runner_lock
|
||||||
if (( result != 0 )); then
|
if (( result != 0 )); then
|
||||||
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
||||||
@@ -97,6 +148,33 @@ preflight_cleanup() {
|
|||||||
}
|
}
|
||||||
trap preflight_cleanup EXIT
|
trap preflight_cleanup EXIT
|
||||||
|
|
||||||
|
DOWNLOAD_TIMEOUT_SECONDS=${TALLYNOTE_UPDATE_DOWNLOAD_TIMEOUT_SECONDS:-${TALLYNOTE_UPDATE_RUNNER_DOWNLOAD_TIMEOUT_SECONDS:-1800}}
|
||||||
|
APPLY_TIMEOUT_SECONDS=${TALLYNOTE_UPDATE_APPLY_TIMEOUT_SECONDS:-${TALLYNOTE_UPDATE_RUNNER_APPLY_TIMEOUT_SECONDS:-1800}}
|
||||||
|
FINALIZE_TIMEOUT_SECONDS=${TALLYNOTE_UPDATE_FINALIZE_TIMEOUT_SECONDS:-${TALLYNOTE_UPDATE_RUNNER_FINALIZE_TIMEOUT_SECONDS:-30}}
|
||||||
|
TIMEOUT_BIN=$(command -v timeout || true)
|
||||||
|
|
||||||
|
run_update_cli() {
|
||||||
|
local node=$1 timeout_seconds=$2 label=$3 result
|
||||||
|
shift 3
|
||||||
|
[[ "$timeout_seconds" =~ ^[1-9][0-9]*$ ]] || die "${label} timeout must be a positive integer"
|
||||||
|
{
|
||||||
|
printf '\n[%s] %s (timeout=%ss)\ncommand:' "$(date -u '+%Y-%m-%dT%H:%M:%SZ')" "$label" "$timeout_seconds"
|
||||||
|
printf ' %q' "$node" "$CURRENT_LINK/dist/server/cli/update.js" "$@"
|
||||||
|
printf '\n'
|
||||||
|
} >>"$RUNNER_LOG"
|
||||||
|
if [[ -n "$TIMEOUT_BIN" ]]; then
|
||||||
|
"$TIMEOUT_BIN" --foreground --signal=TERM --kill-after=10s "${timeout_seconds}s" \
|
||||||
|
"$node" "$CURRENT_LINK/dist/server/cli/update.js" "$@" >>"$RUNNER_LOG" 2>&1
|
||||||
|
result=$?
|
||||||
|
elif "$node" "$CURRENT_LINK/dist/server/cli/update.js" "$@" >>"$RUNNER_LOG" 2>&1; then
|
||||||
|
result=0
|
||||||
|
else
|
||||||
|
result=$?
|
||||||
|
fi
|
||||||
|
printf '[%s] %s exited with status %s\n' "$(date -u '+%Y-%m-%dT%H:%M:%SZ')" "$label" "$result" >>"$RUNNER_LOG"
|
||||||
|
return "$result"
|
||||||
|
}
|
||||||
|
|
||||||
# Downloading is intentionally handled while the main service remains up.
|
# Downloading is intentionally handled while the main service remains up.
|
||||||
# The CLI persists the validated payload under the root-owned workspace and
|
# The CLI persists the validated payload under the root-owned workspace and
|
||||||
# leaves the job staged for a later apply request.
|
# leaves the job staged for a later apply request.
|
||||||
@@ -107,6 +185,7 @@ if [[ "$request_operation" == download ]]; then
|
|||||||
clear_recovery_state || die '无法清理上一次下载状态'
|
clear_recovery_state || die '无法清理上一次下载状态'
|
||||||
fi
|
fi
|
||||||
write_recovery_state download || die '无法写入更新恢复状态'
|
write_recovery_state download || die '无法写入更新恢复状态'
|
||||||
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap
|
||||||
cleanup_download() {
|
cleanup_download() {
|
||||||
local result=$?
|
local result=$?
|
||||||
stop_heartbeat
|
stop_heartbeat
|
||||||
@@ -116,6 +195,7 @@ if [[ "$request_operation" == download ]]; then
|
|||||||
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
fi
|
fi
|
||||||
clear_recovery_state || true
|
clear_recovery_state || true
|
||||||
|
release_runner_lock
|
||||||
return "$result"
|
return "$result"
|
||||||
}
|
}
|
||||||
trap cleanup_download EXIT
|
trap cleanup_download EXIT
|
||||||
@@ -128,7 +208,7 @@ if [[ "$request_operation" == download ]]; then
|
|||||||
cli="$CURRENT_LINK/dist/server/cli/update.js"
|
cli="$CURRENT_LINK/dist/server/cli/update.js"
|
||||||
[[ -f "$cli" ]] || die 'update CLI not found in current release'
|
[[ -f "$cli" ]] || die 'update CLI not found in current release'
|
||||||
set +e
|
set +e
|
||||||
"$node_bin" "$cli" --request-file "$REQUEST_FILE"
|
run_update_cli "$node_bin" "$DOWNLOAD_TIMEOUT_SECONDS" download --request-file "$REQUEST_FILE"
|
||||||
download_result=$?
|
download_result=$?
|
||||||
set -e
|
set -e
|
||||||
if (( download_result != 0 )); then
|
if (( download_result != 0 )); then
|
||||||
@@ -139,7 +219,7 @@ if [[ "$request_operation" == download ]]; then
|
|||||||
download_job_id=$(sed -n 's/.*"jobId"[[:space:]]*:[[:space:]]*"\([0-9a-f-]*\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
download_job_id=$(sed -n 's/.*"jobId"[[:space:]]*:[[:space:]]*"\([0-9a-f-]*\)".*/\1/p' "$REQUEST_FILE" | head -n 1)
|
||||||
if [[ "$download_job_id" =~ ^[0-9a-f-]{36}$ ]]; then
|
if [[ "$download_job_id" =~ ^[0-9a-f-]{36}$ ]]; then
|
||||||
for _ in 1 2 3; do
|
for _ in 1 2 3; do
|
||||||
if "$node_bin" "$cli" --finalize-job "$download_job_id" --finalize-status failed --message '更新下载失败' >/dev/null 2>&1; then break; fi
|
if run_update_cli "$node_bin" "$FINALIZE_TIMEOUT_SECONDS" finalize-download --finalize-job "$download_job_id" --finalize-status failed --message '更新下载失败'; then break; fi
|
||||||
sleep 1
|
sleep 1
|
||||||
done
|
done
|
||||||
fi
|
fi
|
||||||
@@ -150,12 +230,11 @@ if [[ "$request_operation" == download ]]; then
|
|||||||
exit 0
|
exit 0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
was_active=0
|
|
||||||
if systemctl is-active --quiet "$SERVICE_NAME"; then was_active=1; fi
|
|
||||||
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
||||||
restore_initial_service() {
|
restore_initial_service() {
|
||||||
local result=$?
|
local result=$?
|
||||||
stop_heartbeat
|
stop_heartbeat
|
||||||
|
release_runner_lock
|
||||||
if (( result != 0 )); then
|
if (( result != 0 )); then
|
||||||
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
||||||
@@ -175,11 +254,11 @@ finalize_state_job() {
|
|||||||
local node=$1 status=$2 state_job=$3
|
local node=$1 status=$2 state_job=$3
|
||||||
[[ "$state_job" =~ ^[0-9a-f-]{36}$ && -n "$node" ]] || return 1
|
[[ "$state_job" =~ ^[0-9a-f-]{36}$ && -n "$node" ]] || return 1
|
||||||
[[ -f "$CURRENT_LINK/dist/server/cli/update.js" ]] || return 1
|
[[ -f "$CURRENT_LINK/dist/server/cli/update.js" ]] || return 1
|
||||||
"$node" "$CURRENT_LINK/dist/server/cli/update.js" --finalize-job "$state_job" --finalize-status "$status" --message '新版本健康检查失败,已恢复上一版本' >/dev/null 2>&1
|
run_update_cli "$node" "$FINALIZE_TIMEOUT_SECONDS" finalize-recovery --finalize-job "$state_job" --finalize-status "$status" --message '新版本健康检查失败,已恢复上一版本'
|
||||||
}
|
}
|
||||||
|
|
||||||
recover_stale_state() {
|
recover_stale_state() {
|
||||||
local state_job state_old state_phase current_target recovery_node rollback_link state_mode state_uid
|
local state_job state_old state_phase state_initial_active current_target recovery_node rollback_link state_mode state_uid
|
||||||
[[ -f "$STATE_FILE" && ! -L "$STATE_FILE" ]] || die 'update state file is invalid'
|
[[ -f "$STATE_FILE" && ! -L "$STATE_FILE" ]] || die 'update state file is invalid'
|
||||||
state_uid=$(stat -c '%u' "$STATE_FILE" 2>/dev/null || stat -f '%u' "$STATE_FILE")
|
state_uid=$(stat -c '%u' "$STATE_FILE" 2>/dev/null || stat -f '%u' "$STATE_FILE")
|
||||||
state_mode=$(stat -c '%a' "$STATE_FILE" 2>/dev/null || stat -f '%Lp' "$STATE_FILE")
|
state_mode=$(stat -c '%a' "$STATE_FILE" 2>/dev/null || stat -f '%Lp' "$STATE_FILE")
|
||||||
@@ -187,8 +266,16 @@ recover_stale_state() {
|
|||||||
state_job=$(sed -n 's/^job_id=//p' "$STATE_FILE" | head -n 1)
|
state_job=$(sed -n 's/^job_id=//p' "$STATE_FILE" | head -n 1)
|
||||||
state_old=$(sed -n 's/^old_target=//p' "$STATE_FILE" | head -n 1)
|
state_old=$(sed -n 's/^old_target=//p' "$STATE_FILE" | head -n 1)
|
||||||
state_phase=$(sed -n 's/^phase=//p' "$STATE_FILE" | head -n 1)
|
state_phase=$(sed -n 's/^phase=//p' "$STATE_FILE" | head -n 1)
|
||||||
|
state_initial_active=$(sed -n 's/^initial_active=//p' "$STATE_FILE" | head -n 1)
|
||||||
[[ "$state_job" =~ ^[0-9a-f-]{36}$ ]] || die 'update state job id is invalid'
|
[[ "$state_job" =~ ^[0-9a-f-]{36}$ ]] || die 'update state job id is invalid'
|
||||||
[[ "$state_old" == "$PREFIX/releases/"* && -d "$state_old" && ! -L "$state_old" ]] || die 'update state target is invalid'
|
[[ "$state_old" == "$PREFIX/releases/"* && -d "$state_old" && ! -L "$state_old" ]] || die 'update state target is invalid'
|
||||||
|
if [[ -z "$state_initial_active" ]]; then
|
||||||
|
# Markers from older releases did not persist this field. Preserve their
|
||||||
|
# historical conservative behavior instead of rejecting recovery.
|
||||||
|
state_initial_active=0
|
||||||
|
fi
|
||||||
|
[[ "$state_initial_active" == 0 || "$state_initial_active" == 1 ]] || die 'update state initial service state is invalid'
|
||||||
|
was_active=$state_initial_active
|
||||||
current_target=$(readlink -f -- "$CURRENT_LINK" 2>/dev/null || true)
|
current_target=$(readlink -f -- "$CURRENT_LINK" 2>/dev/null || true)
|
||||||
if [[ "$state_phase" == download && "$current_target" == "$state_old" ]]; then
|
if [[ "$state_phase" == download && "$current_target" == "$state_old" ]]; then
|
||||||
# Downloading never changes the active release. If the runner was killed
|
# Downloading never changes the active release. If the runner was killed
|
||||||
@@ -312,7 +399,7 @@ finalize_failed_job() {
|
|||||||
# Give SQLite a moment to release a transient lock before declaring the
|
# Give SQLite a moment to release a transient lock before declaring the
|
||||||
# recovery itself failed.
|
# recovery itself failed.
|
||||||
for _ in 1 2 3; do
|
for _ in 1 2 3; do
|
||||||
if "$old_node" "$CURRENT_LINK/dist/server/cli/update.js" --finalize-job "$job_id" --finalize-status failed --message '新版本健康检查失败,已恢复上一版本' >/dev/null 2>&1; then
|
if run_update_cli "$old_node" "$FINALIZE_TIMEOUT_SECONDS" finalize-failed --finalize-job "$job_id" --finalize-status failed --message '新版本健康检查失败,已恢复上一版本'; then
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
sleep 1
|
sleep 1
|
||||||
@@ -323,7 +410,7 @@ finalize_failed_job() {
|
|||||||
finalize_completed_job() {
|
finalize_completed_job() {
|
||||||
[[ "$job_id" =~ ^[0-9a-f-]{36}$ ]] || return 0
|
[[ "$job_id" =~ ^[0-9a-f-]{36}$ ]] || return 0
|
||||||
[[ -n "$final_node" ]] || return 1
|
[[ -n "$final_node" ]] || return 1
|
||||||
"$final_node" "$CURRENT_LINK/dist/server/cli/update.js" --finalize-job "$job_id" --finalize-status completed >/dev/null 2>&1
|
run_update_cli "$final_node" "$FINALIZE_TIMEOUT_SECONDS" finalize-completed --finalize-job "$job_id" --finalize-status completed
|
||||||
}
|
}
|
||||||
|
|
||||||
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
||||||
@@ -347,6 +434,7 @@ cleanup_after_update() {
|
|||||||
else
|
else
|
||||||
systemctl stop "$SERVICE_NAME" || true
|
systemctl stop "$SERVICE_NAME" || true
|
||||||
fi
|
fi
|
||||||
|
release_runner_lock
|
||||||
return "$result"
|
return "$result"
|
||||||
}
|
}
|
||||||
trap cleanup_after_update EXIT
|
trap cleanup_after_update EXIT
|
||||||
@@ -358,7 +446,7 @@ cli="$CURRENT_LINK/dist/server/cli/update.js"
|
|||||||
[[ -f "$cli" ]] || die 'update CLI not found in current release'
|
[[ -f "$cli" ]] || die 'update CLI not found in current release'
|
||||||
|
|
||||||
set +e
|
set +e
|
||||||
"$node_bin" "$cli" --request-file "$REQUEST_FILE" --defer-completion
|
run_update_cli "$node_bin" "$APPLY_TIMEOUT_SECONDS" apply --request-file "$REQUEST_FILE" --defer-completion
|
||||||
update_result=$?
|
update_result=$?
|
||||||
set -e
|
set -e
|
||||||
if (( update_result != 0 )); then
|
if (( update_result != 0 )); then
|
||||||
|
|||||||
@@ -54,9 +54,11 @@ import {
|
|||||||
validateNewPassword,
|
validateNewPassword,
|
||||||
verifyPassword,
|
verifyPassword,
|
||||||
} from "./security.js";
|
} from "./security.js";
|
||||||
|
import { isNewerVersion } from "./update.js";
|
||||||
import {
|
import {
|
||||||
ACTIVE_UPDATE_STATUSES,
|
ACTIVE_UPDATE_STATUSES,
|
||||||
checkForUpdate,
|
checkForUpdate,
|
||||||
|
currentReleaseVersion,
|
||||||
publicCheckFromCache,
|
publicCheckFromCache,
|
||||||
publicUpdateJob,
|
publicUpdateJob,
|
||||||
reconcileOrphanedUpdateJobs,
|
reconcileOrphanedUpdateJobs,
|
||||||
@@ -1012,6 +1014,15 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
const stagedJobId = input.jobId;
|
const stagedJobId = input.jobId;
|
||||||
const staged = database.sqlite.prepare("SELECT id, status, operation, version, asset_url AS assetUrl, asset_name AS assetName, expected_sha256 AS expectedSha256 FROM update_jobs WHERE id=? AND admin_id=?").get(stagedJobId, request.auth!.admin.id) as { id: string; status: string; operation: string; version: string; assetUrl: string; assetName: string | null; expectedSha256: string | null } | undefined;
|
const staged = database.sqlite.prepare("SELECT id, status, operation, version, asset_url AS assetUrl, asset_name AS assetName, expected_sha256 AS expectedSha256 FROM update_jobs WHERE id=? AND admin_id=?").get(stagedJobId, request.auth!.admin.id) as { id: string; status: string; operation: string; version: string; assetUrl: string; assetName: string | null; expectedSha256: string | null } | undefined;
|
||||||
if (!staged || staged.status !== "staged" || staged.version !== input.version.replace(/^v/i, "")) throw new AppError(409, "UPDATE_NOT_STAGED", "更新任务尚未完成下载");
|
if (!staged || staged.status !== "staged" || staged.version !== input.version.replace(/^v/i, "")) throw new AppError(409, "UPDATE_NOT_STAGED", "更新任务尚未完成下载");
|
||||||
|
// A package may have been downloaded before the host was upgraded by
|
||||||
|
// another path. Never apply a staged archive that is no longer newer
|
||||||
|
// than the release currently serving traffic.
|
||||||
|
const effectiveCurrentVersion = currentReleaseVersion(config) ?? config.appVersion;
|
||||||
|
if (!isNewerVersion(effectiveCurrentVersion, staged.version)) {
|
||||||
|
const now = Date.now();
|
||||||
|
database.sqlite.prepare("UPDATE update_jobs SET status='failed', error_message=?, completed_at=?, updated_at=? WHERE id=? AND status='staged'").run("暂存更新已过期,当前版本无需再次升级", now, now, stagedJobId);
|
||||||
|
throw new AppError(409, "UPDATE_NOT_AVAILABLE", "暂存更新已过期,请重新检查更新");
|
||||||
|
}
|
||||||
if (staged.operation === "apply") throw new AppError(409, "UPDATE_IN_PROGRESS", "更新任务正在处理中,请稍候");
|
if (staged.operation === "apply") throw new AppError(409, "UPDATE_IN_PROGRESS", "更新任务正在处理中,请稍候");
|
||||||
enforceUpdateCooldown(database.sqlite, config, request.auth!.admin.id, "apply", reply);
|
enforceUpdateCooldown(database.sqlite, config, request.auth!.admin.id, "apply", reply);
|
||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import { randomUUID } from "node:crypto";
|
|||||||
import { StringDecoder } from "node:string_decoder";
|
import { StringDecoder } from "node:string_decoder";
|
||||||
import { openDatabase, openDatabaseReadOnly } from "../db/index.js";
|
import { openDatabase, openDatabaseReadOnly } from "../db/index.js";
|
||||||
import { acquireInstanceLock, loadConfig, prepareDataDirectories } from "../config.js";
|
import { acquireInstanceLock, loadConfig, prepareDataDirectories } from "../config.js";
|
||||||
import { hashPassword, normalizeUsername, validateNewPassword, temporaryPassword } from "../security.js";
|
import { hashPassword, normalizeUsername, validateNewPassword, temporaryPassword, verifyPassword } from "../security.js";
|
||||||
import { writeAudit } from "../audit.js";
|
import { writeAudit } from "../audit.js";
|
||||||
|
|
||||||
function arg(name: string): string | undefined {
|
function arg(name: string): string | undefined {
|
||||||
@@ -79,8 +79,13 @@ async function readSecret(prompt: string): Promise<string> {
|
|||||||
return;
|
return;
|
||||||
} else if (character === "\u007f" || character === "\b") {
|
} else if (character === "\u007f" || character === "\b") {
|
||||||
value = value.slice(0, -1);
|
value = value.slice(0, -1);
|
||||||
|
// Keep the credential visible in the SSH terminal as requested.
|
||||||
|
// Redraw the current line so backspace behaves predictably without
|
||||||
|
// putting the value into logs or command arguments.
|
||||||
|
output.write("\r\u001b[2K" + prompt + value);
|
||||||
} else {
|
} else {
|
||||||
value += character;
|
value += character;
|
||||||
|
output.write(character);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -128,6 +133,39 @@ async function main() {
|
|||||||
const release = acquireInstanceLock(config);
|
const release = acquireInstanceLock(config);
|
||||||
const database = openDatabase(config);
|
const database = openDatabase(config);
|
||||||
try {
|
try {
|
||||||
|
const markPasswordConfigured = process.argv.includes("--mark-password-configured");
|
||||||
|
if (markPasswordConfigured) {
|
||||||
|
const username = arg("--username") ?? (await readSecret("用户名: "));
|
||||||
|
const password = await readSecret("当前密码: ");
|
||||||
|
const normalized = normalizeUsername(username);
|
||||||
|
const admin = database.sqlite.prepare(
|
||||||
|
"SELECT id, password_hash, must_change_password, version FROM admins WHERE username_norm = ?",
|
||||||
|
).get(normalized) as { id: string; password_hash: string; must_change_password: number; version: number } | undefined;
|
||||||
|
if (!admin || !(await verifyPassword(admin.password_hash, password))) {
|
||||||
|
throw new Error("用户名或当前密码不正确");
|
||||||
|
}
|
||||||
|
if (!admin.must_change_password) {
|
||||||
|
console.log("该管理员已经可以直接使用当前密码登录。");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const now = Date.now();
|
||||||
|
database.sqlite.transaction(() => {
|
||||||
|
const result = database.sqlite.prepare(
|
||||||
|
"UPDATE admins SET must_change_password=0, auth_version=auth_version+1, version=version+1 WHERE id=? AND version=?",
|
||||||
|
).run(admin.id, admin.version);
|
||||||
|
if (result.changes !== 1) throw new Error("管理员资料已被其他操作更新,请重试");
|
||||||
|
writeAudit(database.sqlite, {
|
||||||
|
requestId: `cli:${randomUUID()}`,
|
||||||
|
actorUsername: "cli",
|
||||||
|
action: "admin.password_policy_cleared",
|
||||||
|
targetType: "admin",
|
||||||
|
targetId: admin.id,
|
||||||
|
after: { username: normalized, mustChangePassword: false, changedAt: now },
|
||||||
|
});
|
||||||
|
})();
|
||||||
|
console.log("已确认当前密码为正式密码,后续登录不再要求修改密码。");
|
||||||
|
return;
|
||||||
|
}
|
||||||
const existing = database.sqlite.prepare("SELECT COUNT(*) AS count FROM admins").get() as { count: number };
|
const existing = database.sqlite.prepare("SELECT COUNT(*) AS count FROM admins").get() as { count: number };
|
||||||
if (existing.count > 0) throw new Error("INITIAL_ADMIN_EXISTS:管理员已经初始化");
|
if (existing.count > 0) throw new Error("INITIAL_ADMIN_EXISTS:管理员已经初始化");
|
||||||
const username = arg("--username") ?? (await readSecret("用户名: "));
|
const username = arg("--username") ?? (await readSecret("用户名: "));
|
||||||
@@ -154,8 +192,16 @@ async function main() {
|
|||||||
database.sqlite.prepare(`
|
database.sqlite.prepare(`
|
||||||
INSERT INTO admins(id, username, username_norm, display_name, password_hash, status,
|
INSERT INTO admins(id, username, username_norm, display_name, password_hash, status,
|
||||||
must_change_password, auth_version, version, created_at)
|
must_change_password, auth_version, version, created_at)
|
||||||
VALUES (?, ?, ?, ?, ?, 'active', 1, 1, 1, ?)
|
VALUES (?, ?, ?, ?, ?, 'active', ?, 1, 1, ?)
|
||||||
`).run(id, username.normalize("NFKC").trim(), normalized, normalizedDisplayName, passwordHash, now);
|
`).run(
|
||||||
|
id,
|
||||||
|
username.normalize("NFKC").trim(),
|
||||||
|
normalized,
|
||||||
|
normalizedDisplayName,
|
||||||
|
passwordHash,
|
||||||
|
generate ? 1 : 0,
|
||||||
|
now,
|
||||||
|
);
|
||||||
writeAudit(database.sqlite, {
|
writeAudit(database.sqlite, {
|
||||||
requestId: `cli:${randomUUID()}`,
|
requestId: `cli:${randomUUID()}`,
|
||||||
actorUsername: "cli",
|
actorUsername: "cli",
|
||||||
|
|||||||
+32
-13
@@ -163,7 +163,10 @@ function writeJob(sqlite: Database.Database | undefined, jobId: string, values:
|
|||||||
requested_at=COALESCE(excluded.requested_at, update_jobs.requested_at),
|
requested_at=COALESCE(excluded.requested_at, update_jobs.requested_at),
|
||||||
started_at=COALESCE(excluded.started_at, update_jobs.started_at),
|
started_at=COALESCE(excluded.started_at, update_jobs.started_at),
|
||||||
operation=excluded.operation,
|
operation=excluded.operation,
|
||||||
status=CASE WHEN update_jobs.status='cancelled' THEN update_jobs.status ELSE excluded.status END,
|
-- Terminal rows are immutable from the runner's ordinary progress
|
||||||
|
-- writes. In particular, a stale/replayed request must not resurrect a
|
||||||
|
-- failed job as queued/downloading/etc.
|
||||||
|
status=CASE WHEN update_jobs.status IN ('cancelled', 'failed', 'completed') THEN update_jobs.status ELSE excluded.status END,
|
||||||
version=excluded.version, platform=excluded.platform,
|
version=excluded.version, platform=excluded.platform,
|
||||||
release_url=COALESCE(excluded.release_url, update_jobs.release_url),
|
release_url=COALESCE(excluded.release_url, update_jobs.release_url),
|
||||||
asset_name=COALESCE(excluded.asset_name, update_jobs.asset_name),
|
asset_name=COALESCE(excluded.asset_name, update_jobs.asset_name),
|
||||||
@@ -176,6 +179,11 @@ function writeJob(sqlite: Database.Database | undefined, jobId: string, values:
|
|||||||
error_message=COALESCE(excluded.error_message, update_jobs.error_message),
|
error_message=COALESCE(excluded.error_message, update_jobs.error_message),
|
||||||
updated_at=excluded.updated_at,
|
updated_at=excluded.updated_at,
|
||||||
completed_at=COALESCE(excluded.completed_at, update_jobs.completed_at)
|
completed_at=COALESCE(excluded.completed_at, update_jobs.completed_at)
|
||||||
|
-- Do not let a delayed runner replay overwrite any field on a terminal
|
||||||
|
-- row. The predicate is part of the same SQLite upsert, so a finalizer
|
||||||
|
-- racing this write still wins atomically instead of leaving a partially
|
||||||
|
-- mutated completed/failed/cancelled record.
|
||||||
|
WHERE update_jobs.status NOT IN ('cancelled', 'failed', 'completed')
|
||||||
`).run(
|
`).run(
|
||||||
jobId,
|
jobId,
|
||||||
values.adminId ?? null,
|
values.adminId ?? null,
|
||||||
@@ -230,6 +238,7 @@ async function resolveRelease(options: UpdateRunOptions, platform: ReturnType<ty
|
|||||||
allowedHosts: options.allowedHosts ?? [],
|
allowedHosts: options.allowedHosts ?? [],
|
||||||
baseUrl: metadataUrl.toString(),
|
baseUrl: metadataUrl.toString(),
|
||||||
maxBytes: options.maxBytes ?? 512 * 1024 * 1024,
|
maxBytes: options.maxBytes ?? 512 * 1024 * 1024,
|
||||||
|
timeoutMs: options.timeoutMs,
|
||||||
publicKey: options.publicKey,
|
publicKey: options.publicKey,
|
||||||
requireSignature: options.requireSignature,
|
requireSignature: options.requireSignature,
|
||||||
});
|
});
|
||||||
@@ -398,19 +407,28 @@ export function finalizeUpdateJob(
|
|||||||
status: "completed" | "failed",
|
status: "completed" | "failed",
|
||||||
message?: string,
|
message?: string,
|
||||||
): void {
|
): void {
|
||||||
const row = sqlite.prepare(`
|
|
||||||
SELECT id, status, version, platform, admin_id AS adminId,
|
|
||||||
request_id AS requestId, session_hash AS sessionHash
|
|
||||||
FROM update_jobs WHERE id=?
|
|
||||||
`).get(jobId) as { id: string; status: UpdateJobStatus; version: string; platform: string; adminId: string | null; requestId: string | null; sessionHash: string | null } | undefined;
|
|
||||||
if (!row) throw new Error("更新任务不存在");
|
|
||||||
const canComplete = row.status === "applying" || row.status === "completed";
|
|
||||||
const canFail = ACTIVE_UPDATE_STATUSES.includes(row.status) || row.status === "completed" || row.status === "failed";
|
|
||||||
if (status === "completed" ? !canComplete : !canFail) throw new Error("更新任务状态不允许完成");
|
|
||||||
const now = Date.now();
|
|
||||||
const safeFailureMessage = status === "failed" ? "新版本健康检查失败,已恢复上一版本" : null;
|
|
||||||
sqlite.transaction(() => {
|
sqlite.transaction(() => {
|
||||||
sqlite.prepare("UPDATE update_jobs SET status=?, error_message=?, completed_at=?, updated_at=? WHERE id=?").run(status, safeFailureMessage, now, now, jobId);
|
const row = sqlite.prepare(`
|
||||||
|
SELECT id, status, version, platform, admin_id AS adminId,
|
||||||
|
request_id AS requestId, session_hash AS sessionHash
|
||||||
|
FROM update_jobs WHERE id=?
|
||||||
|
`).get(jobId) as { id: string; status: UpdateJobStatus; version: string; platform: string; adminId: string | null; requestId: string | null; sessionHash: string | null } | undefined;
|
||||||
|
if (!row) throw new Error("更新任务不存在");
|
||||||
|
// A failed finalization can be retried by the runner. Once it has been
|
||||||
|
// committed, make retries a no-op so the error and audit trail stay stable.
|
||||||
|
if (row.status === status) return;
|
||||||
|
// A completed release is terminal. A delayed recovery process must never
|
||||||
|
// be able to downgrade it to failed after the service was healthy.
|
||||||
|
if (row.status === "completed" && status === "failed") throw new Error("更新任务状态不允许完成");
|
||||||
|
const canComplete = row.status === "applying" || row.status === "completed";
|
||||||
|
const canFail = ACTIVE_UPDATE_STATUSES.includes(row.status) || row.status === "completed" || row.status === "failed";
|
||||||
|
if (status === "completed" ? !canComplete : !canFail) throw new Error("更新任务状态不允许完成");
|
||||||
|
const now = Date.now();
|
||||||
|
const safeFailureMessage = status === "failed"
|
||||||
|
? (message?.trim() ? safeErrorMessage(new Error(message)) : "新版本健康检查失败,已恢复上一版本")
|
||||||
|
: null;
|
||||||
|
const result = sqlite.prepare("UPDATE update_jobs SET status=?, error_message=?, completed_at=?, updated_at=? WHERE id=? AND status=?").run(status, safeFailureMessage, now, now, jobId, row.status);
|
||||||
|
if (result.changes !== 1) return;
|
||||||
writeAudit(sqlite, {
|
writeAudit(sqlite, {
|
||||||
requestId: row.requestId || randomUUID(),
|
requestId: row.requestId || randomUUID(),
|
||||||
actorAdminId: row.adminId,
|
actorAdminId: row.adminId,
|
||||||
@@ -569,6 +587,7 @@ export async function main(config: AppConfig = loadConfig()): Promise<void> {
|
|||||||
...(dataBackupArchive ? { dataBackupArchivePath: dataBackupArchive, dataBackupSource: config.dataDir } : {}),
|
...(dataBackupArchive ? { dataBackupArchivePath: dataBackupArchive, dataBackupSource: config.dataDir } : {}),
|
||||||
...((arg("--backup-dir")) ? { backupDir: arg("--backup-dir") } : {}),
|
...((arg("--backup-dir")) ? { backupDir: arg("--backup-dir") } : {}),
|
||||||
allowedHosts: allowedHosts.length ? allowedHosts : config.updateAllowedHosts,
|
allowedHosts: allowedHosts.length ? allowedHosts : config.updateAllowedHosts,
|
||||||
|
timeoutMs: config.updateTimeoutMs,
|
||||||
maxBytes: config.updateMaxBytes,
|
maxBytes: config.updateMaxBytes,
|
||||||
dataBackupMaxBytes: config.maxTotalBytes,
|
dataBackupMaxBytes: config.maxTotalBytes,
|
||||||
currentVersion: config.appVersion,
|
currentVersion: config.appVersion,
|
||||||
|
|||||||
@@ -147,6 +147,7 @@ export function loadConfig() {
|
|||||||
// as 0700 root:root; development/test callers may override --staging-dir.
|
// as 0700 root:root; development/test callers may override --staging-dir.
|
||||||
updateWorkspaceDir: path.join(installPrefix, ".update-work"),
|
updateWorkspaceDir: path.join(installPrefix, ".update-work"),
|
||||||
updateMaxBytes: integerEnv("TALLYNOTE_UPDATE_MAX_MB", 512) * 1024 * 1024,
|
updateMaxBytes: integerEnv("TALLYNOTE_UPDATE_MAX_MB", 512) * 1024 * 1024,
|
||||||
|
updateTimeoutMs: integerEnv("TALLYNOTE_UPDATE_TIMEOUT_SECONDS", 30) * 1000,
|
||||||
// Update checks hit an external release endpoint. Keep a short local
|
// Update checks hit an external release endpoint. Keep a short local
|
||||||
// cooldown so an authenticated account cannot turn the endpoint into an
|
// cooldown so an authenticated account cannot turn the endpoint into an
|
||||||
// outbound request flood; set to 0 only for controlled test environments.
|
// outbound request flood; set to 0 only for controlled test environments.
|
||||||
|
|||||||
+71
-11
@@ -154,19 +154,19 @@ function signatureAssetFor(metadata: ReleaseMetadata, sums: ReleaseAsset): Relea
|
|||||||
export async function attachSidecarHash(
|
export async function attachSidecarHash(
|
||||||
metadata: ReleaseMetadata,
|
metadata: ReleaseMetadata,
|
||||||
asset: ReleaseAsset,
|
asset: ReleaseAsset,
|
||||||
options: { allowedHosts: readonly string[]; baseUrl: string; maxBytes: number; publicKey?: string | undefined; requireSignature?: boolean | undefined },
|
options: { allowedHosts: readonly string[]; baseUrl: string; maxBytes: number; timeoutMs?: number | undefined; publicKey?: string | undefined; requireSignature?: boolean | undefined },
|
||||||
): Promise<{ asset: ReleaseAsset; signatureVerified: boolean }> {
|
): Promise<{ asset: ReleaseAsset; signatureVerified: boolean }> {
|
||||||
let signatureVerified = false;
|
let signatureVerified = false;
|
||||||
if (asset.sha256 && (!options.publicKey || !options.requireSignature)) return { asset, signatureVerified };
|
if (asset.sha256 && (!options.publicKey || !options.requireSignature)) return { asset, signatureVerified };
|
||||||
const sums = metadata.assets.find((candidate) => /^(?:sha256sums?|checksums?)(?:\.txt)?$/i.test(path.basename(candidate.name)));
|
const sums = metadata.assets.find((candidate) => /^(?:sha256sums?|checksums?)(?:\.txt)?$/i.test(path.basename(candidate.name)));
|
||||||
if (!sums) return { asset, signatureVerified };
|
if (!sums) return { asset, signatureVerified };
|
||||||
try {
|
try {
|
||||||
const content = await fetchReleaseText(sums.url, { allowedHosts: options.allowedHosts, baseUrl: options.baseUrl, maxBytes: Math.min(options.maxBytes, 2 * 1024 * 1024) });
|
const content = await fetchReleaseText(sums.url, { allowedHosts: options.allowedHosts, baseUrl: options.baseUrl, maxBytes: Math.min(options.maxBytes, 2 * 1024 * 1024), timeoutMs: options.timeoutMs });
|
||||||
const sha256 = sha256FromSums(content, asset.name);
|
const sha256 = sha256FromSums(content, asset.name);
|
||||||
if (options.publicKey) {
|
if (options.publicKey) {
|
||||||
const signatureAsset = signatureAssetFor(metadata, sums);
|
const signatureAsset = signatureAssetFor(metadata, sums);
|
||||||
if (signatureAsset) {
|
if (signatureAsset) {
|
||||||
const signature = await fetchReleaseBytes(signatureAsset.url, { allowedHosts: options.allowedHosts, baseUrl: options.baseUrl, maxBytes: 64 * 1024 });
|
const signature = await fetchReleaseBytes(signatureAsset.url, { allowedHosts: options.allowedHosts, baseUrl: options.baseUrl, maxBytes: 64 * 1024, timeoutMs: options.timeoutMs });
|
||||||
signatureVerified = verifyReleaseSignature(content, signature, options.publicKey);
|
signatureVerified = verifyReleaseSignature(content, signature, options.publicKey);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -183,6 +183,7 @@ function policy(config: AppConfig) {
|
|||||||
allowedHosts: config.updateAllowedHosts,
|
allowedHosts: config.updateAllowedHosts,
|
||||||
baseUrl: config.updateMetadataUrl,
|
baseUrl: config.updateMetadataUrl,
|
||||||
maxRedirects: 3,
|
maxRedirects: 3,
|
||||||
|
timeoutMs: config.updateTimeoutMs,
|
||||||
} as const;
|
} as const;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -222,6 +223,7 @@ export async function checkForUpdate(database: Database.Database, config: AppCon
|
|||||||
allowedHosts: config.updateAllowedHosts,
|
allowedHosts: config.updateAllowedHosts,
|
||||||
baseUrl: metadataUrl,
|
baseUrl: metadataUrl,
|
||||||
maxBytes: config.updateMaxBytes,
|
maxBytes: config.updateMaxBytes,
|
||||||
|
timeoutMs: config.updateTimeoutMs,
|
||||||
publicKey: config.updatePublicKey,
|
publicKey: config.updatePublicKey,
|
||||||
requireSignature: config.updateRequireSignature,
|
requireSignature: config.updateRequireSignature,
|
||||||
});
|
});
|
||||||
@@ -426,7 +428,18 @@ function requestJobId(filePath: string): string | null {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function currentReleaseVersion(config: AppConfig): string | null {
|
function recoveryStateJobId(filePath: string): string | null {
|
||||||
|
try {
|
||||||
|
const info = lstatSync(filePath);
|
||||||
|
if (!info.isFile() || info.isSymbolicLink()) return null;
|
||||||
|
const match = /^job_id=([0-9a-f-]{36})$/m.exec(readFileSync(filePath, "utf8"));
|
||||||
|
return match?.[1] ?? null;
|
||||||
|
} catch {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function currentReleaseVersion(config: AppConfig): string | null {
|
||||||
try {
|
try {
|
||||||
const target = realpathSync(config.currentLink);
|
const target = realpathSync(config.currentLink);
|
||||||
const releases = realpathSync(config.releasesDir);
|
const releases = realpathSync(config.releasesDir);
|
||||||
@@ -460,6 +473,12 @@ export function reconcileOrphanedUpdateJobs(database: Database.Database, config:
|
|||||||
const statePresent = stateMtime !== null;
|
const statePresent = stateMtime !== null;
|
||||||
const requestFresh = requestPresent && now - (requestMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
const requestFresh = requestPresent && now - (requestMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
||||||
const stateFresh = statePresent && now - (stateMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
const stateFresh = statePresent && now - (stateMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
||||||
|
// The request marker is the hand-off contract between the web process and
|
||||||
|
// the privileged runner. A queued row with a matching, unexpired marker is
|
||||||
|
// still owned by that hand-off even when the runner has not written its
|
||||||
|
// recovery state yet (for example while systemd is starting it).
|
||||||
|
const requestMarkerJobId = requestPresent ? requestJobId(config.updateRequestPath) : null;
|
||||||
|
const stateMarkerJobId = statePresent ? recoveryStateJobId(statePath) : null;
|
||||||
// A staged download is normally kept for an explicit apply. The one
|
// A staged download is normally kept for an explicit apply. The one
|
||||||
// exception is the hand-off window where the API has already changed the
|
// exception is the hand-off window where the API has already changed the
|
||||||
// operation to `apply` but crashed before writing the request file. That
|
// operation to `apply` but crashed before writing the request file. That
|
||||||
@@ -468,11 +487,50 @@ export function reconcileOrphanedUpdateJobs(database: Database.Database, config:
|
|||||||
let reconciled = 0;
|
let reconciled = 0;
|
||||||
const reconciledIds = new Set<string>();
|
const reconciledIds = new Set<string>();
|
||||||
for (const row of rows) {
|
for (const row of rows) {
|
||||||
|
// A fresh request/state marker means the privileged runner still owns the
|
||||||
|
// hand-off. Do not expire a staged/apply row while the runner is finishing
|
||||||
|
// a successful switch and finalization after a service restart.
|
||||||
|
const matchingFreshRequest = requestMarkerJobId === row.id && requestFresh;
|
||||||
|
const matchingFreshState = stateMarkerJobId === row.id && stateFresh;
|
||||||
|
// A staged archive is actionable only while it is strictly newer than the
|
||||||
|
// release currently serving requests. This can become false when an
|
||||||
|
// administrator upgrades the host by another path (or another operator
|
||||||
|
// completes the same release) before returning to this page. Treat the
|
||||||
|
// archive as an expired terminal task so it cannot keep blocking the
|
||||||
|
// queue or appear as an "apply" action for the current version.
|
||||||
|
const effectiveCurrentVersion = releaseVersion ?? config.appVersion;
|
||||||
|
if (row.status === "staged" && !isNewerVersion(effectiveCurrentVersion, row.version) && !matchingFreshRequest && !matchingFreshState) {
|
||||||
|
const changed = database.transaction(() => {
|
||||||
|
const result = database.prepare(`
|
||||||
|
UPDATE update_jobs
|
||||||
|
SET status='failed', error_message=?, completed_at=?, updated_at=?
|
||||||
|
WHERE id=? AND status='staged'
|
||||||
|
`).run("暂存更新已过期,当前版本无需再次升级", now, now, row.id);
|
||||||
|
if (result.changes !== 1) return false;
|
||||||
|
writeAudit(database, {
|
||||||
|
requestId: row.requestId || randomUUID(),
|
||||||
|
actorAdminId: row.adminId,
|
||||||
|
action: "update.reconciled",
|
||||||
|
targetType: "update",
|
||||||
|
targetId: row.id,
|
||||||
|
outcome: "failure",
|
||||||
|
before: { status: row.status, operation: row.operation, version: row.version },
|
||||||
|
after: { status: "failed", version: row.version, reason: "staged_version_not_newer" },
|
||||||
|
});
|
||||||
|
return true;
|
||||||
|
})();
|
||||||
|
if (changed) {
|
||||||
|
reconciled += 1;
|
||||||
|
reconciledIds.add(row.id);
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
// A request that never gets claimed by the root runner must not remain in
|
// A request that never gets claimed by the root runner must not remain in
|
||||||
// the UI as an endless "queued" task. Once the short hand-off window has
|
// the UI as an endless "queued" task. Once the short hand-off window has
|
||||||
// elapsed and no recovery marker exists, release the queue explicitly;
|
// elapsed and no recovery marker exists, release the queue explicitly;
|
||||||
// a fresh state marker proves that the runner has already claimed it.
|
// a fresh state marker proves that the runner has already claimed it.
|
||||||
if (row.status === "queued" && typeof row.updatedAt === "number" && !stateFresh && now - row.updatedAt >= QUEUED_UPDATE_TIMEOUT_MS) {
|
if (row.status === "queued" && typeof row.updatedAt === "number" && !matchingFreshState && now - row.updatedAt >= QUEUED_UPDATE_TIMEOUT_MS) {
|
||||||
|
if (matchingFreshRequest) continue;
|
||||||
const changed = database.transaction(() => {
|
const changed = database.transaction(() => {
|
||||||
const result = database.prepare(`
|
const result = database.prepare(`
|
||||||
UPDATE update_jobs
|
UPDATE update_jobs
|
||||||
@@ -503,7 +561,7 @@ export function reconcileOrphanedUpdateJobs(database: Database.Database, config:
|
|||||||
// A stale request/state marker therefore no longer protects an orphaned
|
// A stale request/state marker therefore no longer protects an orphaned
|
||||||
// row forever, while a fresh marker remains owned by the runner.
|
// row forever, while a fresh marker remains owned by the runner.
|
||||||
if (row.status === "staged") {
|
if (row.status === "staged") {
|
||||||
if (row.operation !== "apply" || requestFresh || stateFresh) continue;
|
if (row.operation !== "apply" || matchingFreshRequest || matchingFreshState) continue;
|
||||||
const changed = database.transaction(() => {
|
const changed = database.transaction(() => {
|
||||||
const result = database.prepare(`
|
const result = database.prepare(`
|
||||||
UPDATE update_jobs
|
UPDATE update_jobs
|
||||||
@@ -529,7 +587,7 @@ export function reconcileOrphanedUpdateJobs(database: Database.Database, config:
|
|||||||
}
|
}
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
if (requestFresh || stateFresh) continue;
|
if (matchingFreshRequest || matchingFreshState) continue;
|
||||||
const status: "completed" | "failed" = row.status === "applying" && releaseVersion === row.version ? "completed" : "failed";
|
const status: "completed" | "failed" = row.status === "applying" && releaseVersion === row.version ? "completed" : "failed";
|
||||||
const errorMessage = status === "failed" ? "更新任务超时,已释放更新队列" : null;
|
const errorMessage = status === "failed" ? "更新任务超时,已释放更新队列" : null;
|
||||||
const changed = database.transaction(() => {
|
const changed = database.transaction(() => {
|
||||||
@@ -586,15 +644,15 @@ export function cancelUpdateJob(
|
|||||||
jobId?: string,
|
jobId?: string,
|
||||||
): { cancelled: boolean; message?: string } {
|
): { cancelled: boolean; message?: string } {
|
||||||
const job = jobId
|
const job = jobId
|
||||||
? database.prepare("SELECT id, status, operation, version, admin_id AS adminId, download_path AS downloadPath FROM update_jobs WHERE id=?").get(jobId) as { id: string; status: UpdateJobStatus; operation: string; version: string; adminId: string | null; downloadPath: string | null } | undefined
|
? database.prepare("SELECT id, status, operation, version, admin_id AS adminId, download_path AS downloadPath FROM update_jobs WHERE id=? AND admin_id=?").get(jobId, adminId) as { id: string; status: UpdateJobStatus; operation: string; version: string; adminId: string | null; downloadPath: string | null } | undefined
|
||||||
: database.prepare("SELECT id, status, operation, version, admin_id AS adminId, download_path AS downloadPath FROM update_jobs WHERE status IN ('queued', 'downloading') ORDER BY created_at DESC LIMIT 1").get() as { id: string; status: UpdateJobStatus; operation: string; version: string; adminId: string | null; downloadPath: string | null } | undefined;
|
: database.prepare("SELECT id, status, operation, version, admin_id AS adminId, download_path AS downloadPath FROM update_jobs WHERE admin_id=? AND status IN ('queued', 'downloading') ORDER BY created_at DESC LIMIT 1").get(adminId) as { id: string; status: UpdateJobStatus; operation: string; version: string; adminId: string | null; downloadPath: string | null } | undefined;
|
||||||
|
|
||||||
if (!job) return { cancelled: false, message: "当前没有处于等待调度或下载中的更新任务" };
|
if (!job) return { cancelled: false, message: "当前没有处于等待调度或下载中的更新任务" };
|
||||||
if (job.status !== "queued" && job.status !== "downloading") return { cancelled: false, message: "任务已进入就绪或切换阶段,无法取消" };
|
if (job.status !== "queued" && job.status !== "downloading") return { cancelled: false, message: "任务已进入就绪或切换阶段,无法取消" };
|
||||||
|
|
||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
const changed = database.transaction(() => {
|
const changed = database.transaction(() => {
|
||||||
const result = database.prepare("UPDATE update_jobs SET status='cancelled', error_message='已手动取消更新', completed_at=?, updated_at=? WHERE id=? AND status IN ('queued', 'downloading')").run(now, now, job.id);
|
const result = database.prepare("UPDATE update_jobs SET status='cancelled', error_message='已手动取消更新', completed_at=?, updated_at=? WHERE id=? AND admin_id=? AND status IN ('queued', 'downloading')").run(now, now, job.id, adminId);
|
||||||
if (result.changes !== 1) return false;
|
if (result.changes !== 1) return false;
|
||||||
writeAudit(database, {
|
writeAudit(database, {
|
||||||
requestId,
|
requestId,
|
||||||
@@ -610,7 +668,9 @@ export function cancelUpdateJob(
|
|||||||
})();
|
})();
|
||||||
|
|
||||||
if (changed) {
|
if (changed) {
|
||||||
forceRemoveRequest(config.updateRequestPath);
|
// The request marker is shared by the privileged runner. Never remove a
|
||||||
|
// newer/different administrator's request while cancelling this row.
|
||||||
|
if (requestJobId(config.updateRequestPath) === job.id) forceRemoveRequest(config.updateRequestPath);
|
||||||
if (job.downloadPath) {
|
if (job.downloadPath) {
|
||||||
const target = path.isAbsolute(job.downloadPath) ? job.downloadPath : path.join(config.stagingDir, job.downloadPath);
|
const target = path.isAbsolute(job.downloadPath) ? job.downloadPath : path.join(config.stagingDir, job.downloadPath);
|
||||||
import("node:fs/promises").then(({ rm }) => rm(target, { recursive: true, force: true })).catch(() => {});
|
import("node:fs/promises").then(({ rm }) => rm(target, { recursive: true, force: true })).catch(() => {});
|
||||||
|
|||||||
+165
-91
@@ -59,8 +59,22 @@ export type UrlPolicy = {
|
|||||||
/** When allowedHosts is omitted, requests are constrained to this URL's host. */
|
/** When allowedHosts is omitted, requests are constrained to this URL's host. */
|
||||||
baseUrl?: string | URL | undefined;
|
baseUrl?: string | URL | undefined;
|
||||||
maxRedirects?: number | undefined;
|
maxRedirects?: number | undefined;
|
||||||
|
/** Maximum time allowed for one metadata/sidecar/archive request. */
|
||||||
|
timeoutMs?: number | undefined;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
/** Release an unread response body before following a redirect or returning
|
||||||
|
* an error. Undici keeps the underlying connection associated with a body
|
||||||
|
* until it is consumed or cancelled; leaving it open can exhaust sockets when
|
||||||
|
* an update feed repeatedly returns errors or oversized responses. */
|
||||||
|
async function cancelResponseBody(response: Response): Promise<void> {
|
||||||
|
try {
|
||||||
|
await response.body?.cancel();
|
||||||
|
} catch {
|
||||||
|
// The body may already be consumed/closed. Cancellation is best effort.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
function invalidVersion(): never {
|
function invalidVersion(): never {
|
||||||
throw new Error("更新版本号无效");
|
throw new Error("更新版本号无效");
|
||||||
}
|
}
|
||||||
@@ -157,8 +171,37 @@ function metadataError(): Error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const DEFAULT_METADATA_MAX_BYTES = 2 * 1024 * 1024;
|
const DEFAULT_METADATA_MAX_BYTES = 2 * 1024 * 1024;
|
||||||
|
/** Maximum time allowed for one update HTTP request, including its body. */
|
||||||
|
export const DEFAULT_UPDATE_TIMEOUT_MS = 30_000;
|
||||||
export const RELEASE_NOTES_MAX_BYTES = 64 * 1024;
|
export const RELEASE_NOTES_MAX_BYTES = 64 * 1024;
|
||||||
|
|
||||||
|
type UpdateFetchOptions = {
|
||||||
|
fetchImpl?: typeof fetch | undefined;
|
||||||
|
maxBytes?: number | undefined;
|
||||||
|
timeoutMs?: number | undefined;
|
||||||
|
};
|
||||||
|
|
||||||
|
function updateTimeoutMs(options: UpdateFetchOptions): number {
|
||||||
|
if (options.timeoutMs !== undefined) {
|
||||||
|
if (!Number.isSafeInteger(options.timeoutMs) || options.timeoutMs <= 0) throw new Error("更新请求超时配置无效");
|
||||||
|
return options.timeoutMs;
|
||||||
|
}
|
||||||
|
const configuredSeconds = process.env.TALLYNOTE_UPDATE_TIMEOUT_SECONDS;
|
||||||
|
if (configuredSeconds !== undefined && configuredSeconds.trim() !== "") {
|
||||||
|
const seconds = Number(configuredSeconds);
|
||||||
|
if (!Number.isSafeInteger(seconds) || seconds <= 0) throw new Error("TALLYNOTE_UPDATE_TIMEOUT_SECONDS 必须是大于 0 的整数");
|
||||||
|
return seconds * 1000;
|
||||||
|
}
|
||||||
|
return DEFAULT_UPDATE_TIMEOUT_MS;
|
||||||
|
}
|
||||||
|
|
||||||
|
function beginUpdateRequest(options: UpdateFetchOptions): { signal: AbortSignal; clear: () => void } {
|
||||||
|
const controller = new AbortController();
|
||||||
|
const timer = setTimeout(() => controller.abort(), updateTimeoutMs(options));
|
||||||
|
timer.unref?.();
|
||||||
|
return { signal: controller.signal, clear: () => clearTimeout(timer) };
|
||||||
|
}
|
||||||
|
|
||||||
function releaseNotesText(value: unknown): string | undefined {
|
function releaseNotesText(value: unknown): string | undefined {
|
||||||
if (typeof value !== "string" || value.length === 0) return undefined;
|
if (typeof value !== "string" || value.length === 0) return undefined;
|
||||||
// Gitea exposes both Markdown (body/body_html) and releaseNotes depending on
|
// Gitea exposes both Markdown (body/body_html) and releaseNotes depending on
|
||||||
@@ -210,20 +253,29 @@ function releaseResourceUrl(value: string, current: URL, options: UrlPolicy): st
|
|||||||
}
|
}
|
||||||
|
|
||||||
/** Read a fetch body without ever buffering more than the caller's bound. */
|
/** Read a fetch body without ever buffering more than the caller's bound. */
|
||||||
async function readBoundedResponse(response: Response, maxBytes: number, tooLargeMessage: string): Promise<Buffer> {
|
async function readBoundedResponse(response: Response, maxBytes: number, tooLargeMessage: string, signal?: AbortSignal): Promise<Buffer> {
|
||||||
if (!Number.isSafeInteger(maxBytes) || maxBytes <= 0) throw new Error("响应大小限制无效");
|
if (!Number.isSafeInteger(maxBytes) || maxBytes <= 0) throw new Error("响应大小限制无效");
|
||||||
const contentLength = response.headers.get("content-length");
|
const contentLength = response.headers.get("content-length");
|
||||||
if (contentLength !== null) {
|
if (contentLength !== null) {
|
||||||
const declared = Number(contentLength);
|
const declared = Number(contentLength);
|
||||||
if (Number.isFinite(declared) && declared > maxBytes) throw new Error(tooLargeMessage);
|
if (Number.isFinite(declared) && declared > maxBytes) {
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
throw new Error(tooLargeMessage);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if (!response.body) return Buffer.alloc(0);
|
if (!response.body) return Buffer.alloc(0);
|
||||||
const reader = response.body.getReader();
|
const reader = response.body.getReader();
|
||||||
const chunks: Buffer[] = [];
|
const chunks: Buffer[] = [];
|
||||||
let total = 0;
|
let total = 0;
|
||||||
|
let onAbort: (() => void) | undefined;
|
||||||
|
const abort = signal ? new Promise<never>((_, reject) => {
|
||||||
|
onAbort = () => reject(new Error("更新请求超时"));
|
||||||
|
if (signal.aborted) onAbort();
|
||||||
|
else signal.addEventListener("abort", onAbort, { once: true });
|
||||||
|
}) : undefined;
|
||||||
try {
|
try {
|
||||||
for (;;) {
|
for (;;) {
|
||||||
const result = await reader.read();
|
const result = await (abort ? Promise.race([reader.read(), abort]) : reader.read());
|
||||||
if (result.done) break;
|
if (result.done) break;
|
||||||
const chunk = Buffer.from(result.value);
|
const chunk = Buffer.from(result.value);
|
||||||
if (chunk.length > maxBytes - total) {
|
if (chunk.length > maxBytes - total) {
|
||||||
@@ -233,7 +285,11 @@ async function readBoundedResponse(response: Response, maxBytes: number, tooLarg
|
|||||||
total += chunk.length;
|
total += chunk.length;
|
||||||
chunks.push(chunk);
|
chunks.push(chunk);
|
||||||
}
|
}
|
||||||
|
} catch (error) {
|
||||||
|
await reader.cancel().catch(() => undefined);
|
||||||
|
throw error;
|
||||||
} finally {
|
} finally {
|
||||||
|
if (signal && onAbort) signal.removeEventListener("abort", onAbort);
|
||||||
reader.releaseLock();
|
reader.releaseLock();
|
||||||
}
|
}
|
||||||
return Buffer.concat(chunks, total);
|
return Buffer.concat(chunks, total);
|
||||||
@@ -241,84 +297,78 @@ async function readBoundedResponse(response: Response, maxBytes: number, tooLarg
|
|||||||
|
|
||||||
export async function fetchReleaseMetadata(
|
export async function fetchReleaseMetadata(
|
||||||
metadataUrl: string | URL,
|
metadataUrl: string | URL,
|
||||||
options: UrlPolicy & { fetchImpl?: typeof fetch | undefined; maxBytes?: number | undefined } = {},
|
options: UrlPolicy & UpdateFetchOptions = {},
|
||||||
): Promise<ReleaseMetadata> {
|
): Promise<ReleaseMetadata> {
|
||||||
const fetchImpl = options.fetchImpl ?? fetch;
|
const fetchImpl = options.fetchImpl ?? fetch;
|
||||||
let current = validateHttpsUrl(metadataUrl, options);
|
let current = validateHttpsUrl(metadataUrl, options);
|
||||||
const maxRedirects = options.maxRedirects ?? 3;
|
const maxRedirects = options.maxRedirects ?? 3;
|
||||||
let response: Response;
|
let response: Response;
|
||||||
for (let redirects = 0; ; redirects += 1) {
|
for (let redirects = 0; ; redirects += 1) {
|
||||||
|
const request = beginUpdateRequest(options);
|
||||||
try {
|
try {
|
||||||
response = await fetchImpl(current, { method: "GET", redirect: "manual", headers: { accept: "application/json" } });
|
response = await fetchImpl(current, { method: "GET", redirect: "manual", headers: { accept: "application/json" }, signal: request.signal });
|
||||||
} catch {
|
} catch {
|
||||||
|
request.clear();
|
||||||
throw metadataError();
|
throw metadataError();
|
||||||
}
|
}
|
||||||
if (response.status < 300 || response.status >= 400) break;
|
if (response.status < 300 || response.status >= 400) {
|
||||||
|
try {
|
||||||
|
if (response.status < 200 || response.status >= 300) {
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
throw metadataError();
|
||||||
|
}
|
||||||
|
const maxBytes = Math.min(options.maxBytes ?? DEFAULT_METADATA_MAX_BYTES, DEFAULT_METADATA_MAX_BYTES);
|
||||||
|
const body = await readBoundedResponse(response, maxBytes, "更新发布信息过大", request.signal);
|
||||||
|
const payload: unknown = JSON.parse(body.toString("utf8"));
|
||||||
|
if (!payload || typeof payload !== "object") throw metadataError();
|
||||||
|
const item = payload as Record<string, unknown>;
|
||||||
|
const rawVersion = typeof item.version === "string" ? item.version : typeof item.tag_name === "string" ? item.tag_name : typeof item.tagName === "string" ? item.tagName : undefined;
|
||||||
|
if (!rawVersion) throw metadataError();
|
||||||
|
const version = parseSemver(rawVersion);
|
||||||
|
if (typeof item.tag_name === "string" && compareSemver(version, item.tag_name) !== 0) throw metadataError();
|
||||||
|
const assetsRaw = Array.isArray(item.assets) ? item.assets : [];
|
||||||
|
const assets: ReleaseAsset[] = [];
|
||||||
|
for (const raw of assetsRaw) {
|
||||||
|
if (!raw || typeof raw !== "object") continue;
|
||||||
|
const asset = raw as Record<string, unknown>;
|
||||||
|
const name = typeof asset.name === "string" ? asset.name : undefined;
|
||||||
|
const url = typeof asset.url === "string" ? asset.url : typeof asset.browser_download_url === "string" ? asset.browser_download_url : undefined;
|
||||||
|
if (!name || !url) continue;
|
||||||
|
let sha256: string | undefined;
|
||||||
|
const digest = typeof asset.sha256 === "string" ? asset.sha256 : typeof asset.digest === "string" ? asset.digest : undefined;
|
||||||
|
if (digest) {
|
||||||
|
const candidate = digest.replace(/^sha256:/i, "").toLowerCase();
|
||||||
|
if (/^[a-f0-9]{64}$/.test(candidate)) sha256 = candidate;
|
||||||
|
}
|
||||||
|
assets.push({ name, url: releaseResourceUrl(url, current, options), ...(sha256 ? { sha256 } : {}), ...(typeof asset.size === "number" && Number.isSafeInteger(asset.size) && asset.size >= 0 ? { size: asset.size } : {}) });
|
||||||
|
}
|
||||||
|
const notes = releaseNotesText(item.body ?? item.releaseNotes ?? item.release_notes ?? item.body_html);
|
||||||
|
const releaseName = releaseNameText(item.name ?? item.releaseName);
|
||||||
|
let releaseUrl: string | undefined;
|
||||||
|
if (typeof item.html_url === "string" || typeof item.url === "string") {
|
||||||
|
try { releaseUrl = releaseResourceUrl(typeof item.html_url === "string" ? item.html_url : item.url as string, current, options); } catch { /* optional */ }
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
version: `${version.major}.${version.minor}.${version.patch}${version.prerelease.length ? `-${version.prerelease.join(".")}` : ""}${version.build.length ? `+${version.build.join(".")}` : ""}`,
|
||||||
|
...(typeof item.tag_name === "string" ? { tagName: item.tag_name } : {}), ...(releaseName ? { releaseName } : {}), ...(typeof item.published_at === "string" ? { publishedAt: item.published_at } : {}), ...(notes ? { notes } : {}), ...(releaseUrl ? { releaseUrl } : {}), assets,
|
||||||
|
};
|
||||||
|
} catch { throw metadataError(); }
|
||||||
|
finally { request.clear(); }
|
||||||
|
}
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
request.clear();
|
||||||
if (redirects >= maxRedirects) throw metadataError();
|
if (redirects >= maxRedirects) throw metadataError();
|
||||||
const location = response.headers.get("location");
|
const location = response.headers.get("location");
|
||||||
if (!location) throw metadataError();
|
if (!location) throw metadataError();
|
||||||
current = validateHttpsUrl(new URL(location, current), options.baseUrl ? options : { ...options, baseUrl: current });
|
current = validateHttpsUrl(new URL(location, current), options.baseUrl ? options : { ...options, baseUrl: current });
|
||||||
}
|
}
|
||||||
if (response.status < 200 || response.status >= 300) throw metadataError();
|
|
||||||
let payload: unknown;
|
|
||||||
try {
|
|
||||||
const maxBytes = Math.min(options.maxBytes ?? DEFAULT_METADATA_MAX_BYTES, DEFAULT_METADATA_MAX_BYTES);
|
|
||||||
const body = await readBoundedResponse(response, maxBytes, "更新发布信息过大");
|
|
||||||
payload = JSON.parse(body.toString("utf8"));
|
|
||||||
} catch { throw metadataError(); }
|
|
||||||
if (!payload || typeof payload !== "object") throw metadataError();
|
|
||||||
const item = payload as Record<string, unknown>;
|
|
||||||
const rawVersion = typeof item.version === "string" ? item.version : typeof item.tag_name === "string" ? item.tag_name : typeof item.tagName === "string" ? item.tagName : undefined;
|
|
||||||
if (!rawVersion) throw metadataError();
|
|
||||||
const version = parseSemver(rawVersion);
|
|
||||||
if (typeof item.tag_name === "string") {
|
|
||||||
try {
|
|
||||||
if (compareSemver(version, item.tag_name) !== 0) throw metadataError();
|
|
||||||
} catch {
|
|
||||||
throw metadataError();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
const assetsRaw = Array.isArray(item.assets) ? item.assets : [];
|
|
||||||
const assets: ReleaseAsset[] = [];
|
|
||||||
for (const raw of assetsRaw) {
|
|
||||||
if (!raw || typeof raw !== "object") continue;
|
|
||||||
const asset = raw as Record<string, unknown>;
|
|
||||||
const name = typeof asset.name === "string" ? asset.name : undefined;
|
|
||||||
const url = typeof asset.url === "string" ? asset.url : typeof asset.browser_download_url === "string" ? asset.browser_download_url : undefined;
|
|
||||||
if (!name || !url) continue;
|
|
||||||
let sha256: string | undefined;
|
|
||||||
const digest = typeof asset.sha256 === "string" ? asset.sha256 : typeof asset.digest === "string" ? asset.digest : undefined;
|
|
||||||
if (digest) {
|
|
||||||
const candidate = digest.replace(/^sha256:/i, "").toLowerCase();
|
|
||||||
if (/^[a-f0-9]{64}$/.test(candidate)) sha256 = candidate;
|
|
||||||
}
|
|
||||||
assets.push({ name, url: releaseResourceUrl(url, current, options), ...(sha256 ? { sha256 } : {}), ...(typeof asset.size === "number" && Number.isSafeInteger(asset.size) && asset.size >= 0 ? { size: asset.size } : {}) });
|
|
||||||
}
|
|
||||||
const notes = releaseNotesText(item.body ?? item.releaseNotes ?? item.release_notes ?? item.body_html);
|
|
||||||
const releaseName = releaseNameText(item.name ?? item.releaseName);
|
|
||||||
let releaseUrl: string | undefined;
|
|
||||||
if (typeof item.html_url === "string" || typeof item.url === "string") {
|
|
||||||
try {
|
|
||||||
const candidate = typeof item.html_url === "string" ? item.html_url : item.url as string;
|
|
||||||
releaseUrl = releaseResourceUrl(candidate, current, options);
|
|
||||||
} catch { /* omit invalid optional release page URL */ }
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
version: `${version.major}.${version.minor}.${version.patch}${version.prerelease.length ? `-${version.prerelease.join(".")}` : ""}${version.build.length ? `+${version.build.join(".")}` : ""}`,
|
|
||||||
...(typeof item.tag_name === "string" ? { tagName: item.tag_name } : {}),
|
|
||||||
...(releaseName ? { releaseName } : {}),
|
|
||||||
...(typeof item.published_at === "string" ? { publishedAt: item.published_at } : {}),
|
|
||||||
...(notes ? { notes } : {}),
|
|
||||||
...(releaseUrl ? { releaseUrl } : {}),
|
|
||||||
assets,
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Fetch a small text sidecar (for example SHA256SUMS) with the same
|
/** Fetch a small text sidecar (for example SHA256SUMS) with the same
|
||||||
* redirect, HTTPS and host policy used for release metadata. */
|
* redirect, HTTPS and host policy used for release metadata. */
|
||||||
export async function fetchReleaseText(
|
export async function fetchReleaseText(
|
||||||
textUrl: string | URL,
|
textUrl: string | URL,
|
||||||
options: UrlPolicy & { fetchImpl?: typeof fetch | undefined; maxBytes?: number | undefined } = {},
|
options: UrlPolicy & UpdateFetchOptions = {},
|
||||||
): Promise<string> {
|
): Promise<string> {
|
||||||
const fetchImpl = options.fetchImpl ?? fetch;
|
const fetchImpl = options.fetchImpl ?? fetch;
|
||||||
let current = validateHttpsUrl(textUrl, options);
|
let current = validateHttpsUrl(textUrl, options);
|
||||||
@@ -328,27 +378,32 @@ export async function fetchReleaseText(
|
|||||||
const maxRedirects = options.maxRedirects ?? 3;
|
const maxRedirects = options.maxRedirects ?? 3;
|
||||||
let response: Response;
|
let response: Response;
|
||||||
for (let redirects = 0; ; redirects += 1) {
|
for (let redirects = 0; ; redirects += 1) {
|
||||||
|
const request = beginUpdateRequest(options);
|
||||||
try {
|
try {
|
||||||
response = await fetchImpl(current, { method: "GET", redirect: "manual" });
|
response = await fetchImpl(current, { method: "GET", redirect: "manual", signal: request.signal });
|
||||||
} catch {
|
} catch {
|
||||||
|
request.clear();
|
||||||
throw new Error("更新校验文件下载失败");
|
throw new Error("更新校验文件下载失败");
|
||||||
}
|
}
|
||||||
if (response.status < 300 || response.status >= 400) break;
|
if (response.status < 300 || response.status >= 400) {
|
||||||
|
if (response.status < 200 || response.status >= 300) { await cancelResponseBody(response); request.clear(); throw new Error("更新校验文件下载失败"); }
|
||||||
|
const declared = Number(response.headers.get("content-length") ?? 0);
|
||||||
|
const maxBytes = options.maxBytes ?? 1024 * 1024;
|
||||||
|
if (declared > maxBytes) { await cancelResponseBody(response); request.clear(); throw new Error("更新校验文件过大"); }
|
||||||
|
try {
|
||||||
|
return (await readBoundedResponse(response, maxBytes, "更新校验文件过大", request.signal)).toString("utf8");
|
||||||
|
} catch (error) {
|
||||||
|
if (error instanceof Error && error.message === "更新校验文件过大") throw error;
|
||||||
|
throw new Error("更新校验文件下载失败");
|
||||||
|
} finally { request.clear(); }
|
||||||
|
}
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
request.clear();
|
||||||
if (redirects >= maxRedirects) throw new Error("更新校验文件下载失败");
|
if (redirects >= maxRedirects) throw new Error("更新校验文件下载失败");
|
||||||
const location = response.headers.get("location");
|
const location = response.headers.get("location");
|
||||||
if (!location) throw new Error("更新校验文件下载失败");
|
if (!location) throw new Error("更新校验文件下载失败");
|
||||||
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
||||||
}
|
}
|
||||||
if (response.status < 200 || response.status >= 300) throw new Error("更新校验文件下载失败");
|
|
||||||
const declared = Number(response.headers.get("content-length") ?? 0);
|
|
||||||
const maxBytes = options.maxBytes ?? 1024 * 1024;
|
|
||||||
if (declared > maxBytes) throw new Error("更新校验文件过大");
|
|
||||||
try {
|
|
||||||
return (await readBoundedResponse(response, maxBytes, "更新校验文件过大")).toString("utf8");
|
|
||||||
} catch (error) {
|
|
||||||
if (error instanceof Error && error.message === "更新校验文件过大") throw error;
|
|
||||||
throw new Error("更新校验文件下载失败");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Fetch a bounded binary sidecar (for example an Ed25519 detached
|
/** Fetch a bounded binary sidecar (for example an Ed25519 detached
|
||||||
@@ -356,7 +411,7 @@ export async function fetchReleaseText(
|
|||||||
* this separate from fetchReleaseText. */
|
* this separate from fetchReleaseText. */
|
||||||
export async function fetchReleaseBytes(
|
export async function fetchReleaseBytes(
|
||||||
bytesUrl: string | URL,
|
bytesUrl: string | URL,
|
||||||
options: UrlPolicy & { fetchImpl?: typeof fetch | undefined; maxBytes?: number | undefined } = {},
|
options: UrlPolicy & UpdateFetchOptions = {},
|
||||||
): Promise<Buffer> {
|
): Promise<Buffer> {
|
||||||
const fetchImpl = options.fetchImpl ?? fetch;
|
const fetchImpl = options.fetchImpl ?? fetch;
|
||||||
let current = validateHttpsUrl(bytesUrl, options);
|
let current = validateHttpsUrl(bytesUrl, options);
|
||||||
@@ -366,27 +421,32 @@ export async function fetchReleaseBytes(
|
|||||||
const maxRedirects = options.maxRedirects ?? 3;
|
const maxRedirects = options.maxRedirects ?? 3;
|
||||||
let response: Response;
|
let response: Response;
|
||||||
for (let redirects = 0; ; redirects += 1) {
|
for (let redirects = 0; ; redirects += 1) {
|
||||||
|
const request = beginUpdateRequest(options);
|
||||||
try {
|
try {
|
||||||
response = await fetchImpl(current, { method: "GET", redirect: "manual" });
|
response = await fetchImpl(current, { method: "GET", redirect: "manual", signal: request.signal });
|
||||||
} catch {
|
} catch {
|
||||||
|
request.clear();
|
||||||
throw new Error("更新签名下载失败");
|
throw new Error("更新签名下载失败");
|
||||||
}
|
}
|
||||||
if (response.status < 300 || response.status >= 400) break;
|
if (response.status < 300 || response.status >= 400) {
|
||||||
|
if (response.status < 200 || response.status >= 300) { await cancelResponseBody(response); request.clear(); throw new Error("更新签名下载失败"); }
|
||||||
|
const declared = Number(response.headers.get("content-length") ?? 0);
|
||||||
|
const maxBytes = options.maxBytes ?? 64 * 1024;
|
||||||
|
if (declared > maxBytes) { await cancelResponseBody(response); request.clear(); throw new Error("更新签名文件过大"); }
|
||||||
|
try {
|
||||||
|
return await readBoundedResponse(response, maxBytes, "更新签名文件过大", request.signal);
|
||||||
|
} catch (error) {
|
||||||
|
if (error instanceof Error && error.message === "更新签名文件过大") throw error;
|
||||||
|
throw new Error("更新签名下载失败");
|
||||||
|
} finally { request.clear(); }
|
||||||
|
}
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
request.clear();
|
||||||
if (redirects >= maxRedirects) throw new Error("更新签名下载失败");
|
if (redirects >= maxRedirects) throw new Error("更新签名下载失败");
|
||||||
const location = response.headers.get("location");
|
const location = response.headers.get("location");
|
||||||
if (!location) throw new Error("更新签名下载失败");
|
if (!location) throw new Error("更新签名下载失败");
|
||||||
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
||||||
}
|
}
|
||||||
if (response.status < 200 || response.status >= 300) throw new Error("更新签名下载失败");
|
|
||||||
const declared = Number(response.headers.get("content-length") ?? 0);
|
|
||||||
const maxBytes = options.maxBytes ?? 64 * 1024;
|
|
||||||
if (declared > maxBytes) throw new Error("更新签名文件过大");
|
|
||||||
try {
|
|
||||||
return await readBoundedResponse(response, maxBytes, "更新签名文件过大");
|
|
||||||
} catch (error) {
|
|
||||||
if (error instanceof Error && error.message === "更新签名文件过大") throw error;
|
|
||||||
throw new Error("更新签名下载失败");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function selectReleaseAsset(release: ReleaseMetadata, platform = detectPlatform(), runtimeHash?: string): ReleaseAsset | undefined {
|
export function selectReleaseAsset(release: ReleaseMetadata, platform = detectPlatform(), runtimeHash?: string): ReleaseAsset | undefined {
|
||||||
@@ -438,7 +498,7 @@ export async function verifySha256(filePath: string, expected: string): Promise<
|
|||||||
export async function downloadReleaseAsset(
|
export async function downloadReleaseAsset(
|
||||||
url: string | URL,
|
url: string | URL,
|
||||||
destination: string,
|
destination: string,
|
||||||
options: UrlPolicy & { fetchImpl?: typeof fetch | undefined; maxBytes?: number | undefined; onProgress?: ((downloadedBytes: number, totalBytes: number | null) => void) | undefined } = {},
|
options: UrlPolicy & UpdateFetchOptions & { onProgress?: ((downloadedBytes: number, totalBytes: number | null) => void) | undefined } = {},
|
||||||
): Promise<{ size: number; sha256: string }> {
|
): Promise<{ size: number; sha256: string }> {
|
||||||
const fetchImpl = options.fetchImpl ?? fetch;
|
const fetchImpl = options.fetchImpl ?? fetch;
|
||||||
let current = validateHttpsUrl(url, options);
|
let current = validateHttpsUrl(url, options);
|
||||||
@@ -448,22 +508,32 @@ export async function downloadReleaseAsset(
|
|||||||
const maxRedirects = options.maxRedirects ?? 3;
|
const maxRedirects = options.maxRedirects ?? 3;
|
||||||
let response: Response;
|
let response: Response;
|
||||||
for (let redirects = 0; ; redirects += 1) {
|
for (let redirects = 0; ; redirects += 1) {
|
||||||
|
const request = beginUpdateRequest(options);
|
||||||
try {
|
try {
|
||||||
response = await fetchImpl(current, { method: "GET", redirect: "manual" });
|
response = await fetchImpl(current, { method: "GET", redirect: "manual", signal: request.signal });
|
||||||
} catch {
|
} catch {
|
||||||
|
request.clear();
|
||||||
throw new Error("更新文件下载失败");
|
throw new Error("更新文件下载失败");
|
||||||
}
|
}
|
||||||
if (response.status < 300 || response.status >= 400) break;
|
if (response.status < 300 || response.status >= 400) { request.clear(); break; }
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
request.clear();
|
||||||
if (redirects >= maxRedirects) throw new Error("更新文件下载失败");
|
if (redirects >= maxRedirects) throw new Error("更新文件下载失败");
|
||||||
const location = response.headers.get("location");
|
const location = response.headers.get("location");
|
||||||
if (!location) throw new Error("更新文件下载失败");
|
if (!location) throw new Error("更新文件下载失败");
|
||||||
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
current = validateHttpsUrl(new URL(location, current), redirectPolicy);
|
||||||
}
|
}
|
||||||
if (response.status < 200 || response.status >= 300 || !response.body) throw new Error("更新文件下载失败");
|
if (response.status < 200 || response.status >= 300 || !response.body) {
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
throw new Error("更新文件下载失败");
|
||||||
|
}
|
||||||
const declared = Number(response.headers.get("content-length") ?? 0);
|
const declared = Number(response.headers.get("content-length") ?? 0);
|
||||||
const totalBytes = Number.isSafeInteger(declared) && declared > 0 ? declared : null;
|
const totalBytes = Number.isSafeInteger(declared) && declared > 0 ? declared : null;
|
||||||
const maxBytes = options.maxBytes ?? 512 * 1024 * 1024;
|
const maxBytes = options.maxBytes ?? 512 * 1024 * 1024;
|
||||||
if (declared > maxBytes) throw new Error("更新文件超过大小限制");
|
if (declared > maxBytes) {
|
||||||
|
await cancelResponseBody(response);
|
||||||
|
throw new Error("更新文件超过大小限制");
|
||||||
|
}
|
||||||
await mkdir(path.dirname(destination), { recursive: true, mode: 0o700 });
|
await mkdir(path.dirname(destination), { recursive: true, mode: 0o700 });
|
||||||
const temporary = `${destination}.part-${randomUUID()}`;
|
const temporary = `${destination}.part-${randomUUID()}`;
|
||||||
let size = 0;
|
let size = 0;
|
||||||
@@ -475,8 +545,10 @@ export async function downloadReleaseAsset(
|
|||||||
hash.update(chunk);
|
hash.update(chunk);
|
||||||
callback(null, chunk);
|
callback(null, chunk);
|
||||||
} });
|
} });
|
||||||
|
const request = beginUpdateRequest(options);
|
||||||
try {
|
try {
|
||||||
await pipeline(Readable.fromWeb(response.body as import("node:stream/web").ReadableStream), meter, createWriteStream(temporary, { flags: "wx", mode: 0o600 }));
|
const source = Readable.fromWeb(response.body as import("node:stream/web").ReadableStream, { signal: request.signal });
|
||||||
|
await pipeline(source, meter, createWriteStream(temporary, { flags: "wx", mode: 0o600 }));
|
||||||
const fd = await open(temporary, "r");
|
const fd = await open(temporary, "r");
|
||||||
await fd.sync();
|
await fd.sync();
|
||||||
await fd.close();
|
await fd.close();
|
||||||
@@ -484,6 +556,8 @@ export async function downloadReleaseAsset(
|
|||||||
} catch (error) {
|
} catch (error) {
|
||||||
await import("node:fs/promises").then(({ rm }) => rm(temporary, { force: true })).catch(() => undefined);
|
await import("node:fs/promises").then(({ rm }) => rm(temporary, { force: true })).catch(() => undefined);
|
||||||
throw error instanceof Error && error.message.startsWith("更新文件") ? error : new Error("更新文件下载失败");
|
throw error instanceof Error && error.message.startsWith("更新文件") ? error : new Error("更新文件下载失败");
|
||||||
|
} finally {
|
||||||
|
request.clear();
|
||||||
}
|
}
|
||||||
return { size, sha256: hash.digest("hex") };
|
return { size, sha256: hash.digest("hex") };
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
[Unit]
|
[Unit]
|
||||||
Description=TallyNote privileged release updater
|
Description=TallyNote privileged release updater
|
||||||
After=network-online.target
|
|
||||||
Wants=network-online.target
|
|
||||||
|
|
||||||
[Service]
|
[Service]
|
||||||
Type=oneshot
|
Type=oneshot
|
||||||
User=root
|
User=root
|
||||||
@@ -11,10 +8,12 @@ WorkingDirectory=/opt/tallynote/current
|
|||||||
EnvironmentFile=-/etc/tallynote/tallynote.env
|
EnvironmentFile=-/etc/tallynote/tallynote.env
|
||||||
ExecStart=/usr/local/libexec/tallynote-update-runner
|
ExecStart=/usr/local/libexec/tallynote-update-runner
|
||||||
Environment=PATH=/usr/sbin:/usr/bin:/sbin:/bin
|
Environment=PATH=/usr/sbin:/usr/bin:/sbin:/bin
|
||||||
|
# The runner consumes queued requests immediately and applies its own bounded
|
||||||
|
# phase timeouts while keeping full CLI diagnostics in the runner log.
|
||||||
# Downloads, archive validation and data backups can exceed systemd's 90s
|
# Downloads, archive validation and data backups can exceed systemd's 90s
|
||||||
# default start timeout on a slower server. Keep one update job alive long
|
# default start timeout on a slower server. Keep one update job alive long
|
||||||
# enough to finish or reach its own health-check/recovery path.
|
# enough to finish or reach its own health-check/recovery path.
|
||||||
TimeoutStartSec=30min
|
TimeoutStartSec=32min
|
||||||
NoNewPrivileges=true
|
NoNewPrivileges=true
|
||||||
# Keep the updater compatible with the same Node/libuv interface discovery
|
# Keep the updater compatible with the same Node/libuv interface discovery
|
||||||
# path while retaining an explicit socket-family allowlist.
|
# path while retaining an explicit socket-family allowlist.
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ TALLYNOTE_TIMEZONE=Asia/Shanghai
|
|||||||
TALLYNOTE_UPDATE_STRATEGY=systemd
|
TALLYNOTE_UPDATE_STRATEGY=systemd
|
||||||
TALLYNOTE_UPDATE_METADATA_URL=https://git.awaioi.com/api/v1/repos/awaioi/TallyNote/releases/latest
|
TALLYNOTE_UPDATE_METADATA_URL=https://git.awaioi.com/api/v1/repos/awaioi/TallyNote/releases/latest
|
||||||
TALLYNOTE_UPDATE_ALLOWED_HOSTS=git.awaioi.com
|
TALLYNOTE_UPDATE_ALLOWED_HOSTS=git.awaioi.com
|
||||||
|
TALLYNOTE_UPDATE_TIMEOUT_SECONDS=30
|
||||||
TALLYNOTE_UPDATE_REQUIRE_SIGNATURE=false
|
TALLYNOTE_UPDATE_REQUIRE_SIGNATURE=false
|
||||||
TALLYNOTE_UPDATE_CHECK_COOLDOWN_SECONDS=60
|
TALLYNOTE_UPDATE_CHECK_COOLDOWN_SECONDS=60
|
||||||
TALLYNOTE_UPDATE_DOWNLOAD_COOLDOWN_SECONDS=15
|
TALLYNOTE_UPDATE_DOWNLOAD_COOLDOWN_SECONDS=15
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { describe, expect, it } from "vitest";
|
import { describe, expect, it } from "vitest";
|
||||||
import { existsSync, mkdtempSync, readFileSync, rmSync } from "node:fs";
|
import { existsSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
|
||||||
import { spawnSync } from "node:child_process";
|
import { spawnSync } from "node:child_process";
|
||||||
import { tmpdir } from "node:os";
|
import { tmpdir } from "node:os";
|
||||||
import path from "node:path";
|
import path from "node:path";
|
||||||
@@ -8,6 +8,9 @@ import Database from "better-sqlite3";
|
|||||||
const root = path.resolve(process.cwd());
|
const root = path.resolve(process.cwd());
|
||||||
const cli = path.join(root, "server", "cli", "admin-init.ts");
|
const cli = path.join(root, "server", "cli", "admin-init.ts");
|
||||||
const tsx = path.join(root, "node_modules", "tsx", "dist", "cli.mjs");
|
const tsx = path.join(root, "node_modules", "tsx", "dist", "cli.mjs");
|
||||||
|
const ptyHelper = path.join(root, "tests", "helpers", "pty-run.py");
|
||||||
|
const hasPython3 = spawnSync("python3", ["--version"]).status === 0;
|
||||||
|
const ttyTest = hasPython3 ? it : it.skip;
|
||||||
|
|
||||||
function runAdmin(dataDir: string, args: string[]) {
|
function runAdmin(dataDir: string, args: string[]) {
|
||||||
return spawnSync(process.execPath, [tsx, cli, ...args], {
|
return spawnSync(process.execPath, [tsx, cli, ...args], {
|
||||||
@@ -24,6 +27,42 @@ function runAdmin(dataDir: string, args: string[]) {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function testEnv(dataDir: string) {
|
||||||
|
return {
|
||||||
|
...process.env,
|
||||||
|
NODE_ENV: "test",
|
||||||
|
TALLYNOTE_DATA_DIR: dataDir,
|
||||||
|
TALLYNOTE_PUBLIC_ORIGIN: "http://127.0.0.1:3999",
|
||||||
|
TALLYNOTE_COOKIE_SECURE: "false",
|
||||||
|
TALLYNOTE_UPDATE_STRATEGY: "disabled",
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// The CI runner has no `expect` binary. Drive the interactive CLI through a
|
||||||
|
// real pseudo-terminal via a tiny Python pty helper (python3 ships on both
|
||||||
|
// macOS and the Linux CI image). This avoids `expect` (not installed on CI)
|
||||||
|
// and BSD `script` (injects a stray EOT byte from file input, corrupting the
|
||||||
|
// first prompt value). If python3 is unavailable the tests are skipped rather
|
||||||
|
// than failing the build.
|
||||||
|
function runAdminTTY(dataDir: string, args: string[], inputText: string) {
|
||||||
|
const parent = mkdtempSync(path.join(tmpdir(), "tallynote-admin-tty-"));
|
||||||
|
const inputFile = path.join(parent, "input");
|
||||||
|
const exitFile = path.join(parent, "exit-code");
|
||||||
|
writeFileSync(inputFile, inputText);
|
||||||
|
try {
|
||||||
|
const result = spawnSync("python3", [ptyHelper, process.execPath, tsx, cli, ...args], {
|
||||||
|
cwd: root,
|
||||||
|
env: { ...testEnv(dataDir), PTY_STDIN_FILE: inputFile, PTY_EXIT_FILE: exitFile },
|
||||||
|
encoding: "utf8",
|
||||||
|
timeout: 30_000,
|
||||||
|
});
|
||||||
|
const exitCode = existsSync(exitFile) ? Number(readFileSync(exitFile, "utf8")) : null;
|
||||||
|
return { exitCode, output: `${result.stdout}${result.stderr}`, spawnError: result.error };
|
||||||
|
} finally {
|
||||||
|
rmSync(parent, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
describe("生产管理员初始化 CLI", () => {
|
describe("生产管理员初始化 CLI", () => {
|
||||||
it("--check 是只读的,空数据目录不会被创建", () => {
|
it("--check 是只读的,空数据目录不会被创建", () => {
|
||||||
const parent = mkdtempSync(path.join(tmpdir(), "tallynote-admin-check-"));
|
const parent = mkdtempSync(path.join(tmpdir(), "tallynote-admin-check-"));
|
||||||
@@ -85,6 +124,46 @@ describe("生产管理员初始化 CLI", () => {
|
|||||||
}
|
}
|
||||||
}, 15_000);
|
}, 15_000);
|
||||||
|
|
||||||
|
ttyTest("交互式输入正式密码后不会强制首次改密", () => {
|
||||||
|
const dataDir = mkdtempSync(path.join(tmpdir(), "tallynote-admin-init-"));
|
||||||
|
try {
|
||||||
|
const result = runAdminTTY(dataDir, [], "manual-admin\n手动管理员\nStrong-password-2026!\nStrong-password-2026!\n");
|
||||||
|
expect(result.spawnError).toBeUndefined();
|
||||||
|
expect(result.exitCode).toBe(0);
|
||||||
|
expect(result.output).toContain("已创建首位管理员");
|
||||||
|
expect(result.output).toContain("Strong-password-2026!");
|
||||||
|
|
||||||
|
const database = new Database(path.join(dataDir, "tallynote.db"));
|
||||||
|
const admin = database.prepare("SELECT username, must_change_password FROM admins").get() as { username: string; must_change_password: number };
|
||||||
|
expect(admin).toEqual({ username: "manual-admin", must_change_password: 0 });
|
||||||
|
database.close();
|
||||||
|
} finally {
|
||||||
|
rmSync(dataDir, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
}, 30_000);
|
||||||
|
|
||||||
|
ttyTest("可以验证当前密码并清除旧版本遗留的首次改密标志", () => {
|
||||||
|
const dataDir = mkdtempSync(path.join(tmpdir(), "tallynote-admin-init-"));
|
||||||
|
try {
|
||||||
|
const first = runAdmin(dataDir, ["--username", "legacy-admin", "--display-name", "旧版管理员", "--generate"]);
|
||||||
|
expect(first.status).toBe(0);
|
||||||
|
const generated = first.stdout.match(/一次性密码:([^\s]+)/)?.[1];
|
||||||
|
expect(generated).toBeTruthy();
|
||||||
|
|
||||||
|
const result = runAdminTTY(dataDir, ["--mark-password-configured", "--username", "legacy-admin"], `${generated}\n`);
|
||||||
|
expect(result.spawnError).toBeUndefined();
|
||||||
|
expect(result.exitCode).toBe(0);
|
||||||
|
expect(result.output).toContain("已确认当前密码为正式密码");
|
||||||
|
|
||||||
|
const database = new Database(path.join(dataDir, "tallynote.db"));
|
||||||
|
const admin = database.prepare("SELECT must_change_password FROM admins WHERE username_norm='legacy-admin'").get() as { must_change_password: number };
|
||||||
|
expect(admin.must_change_password).toBe(0);
|
||||||
|
database.close();
|
||||||
|
} finally {
|
||||||
|
rmSync(dataDir, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
}, 30_000);
|
||||||
|
|
||||||
it("密码输入不是 TTY 时明确拒绝通过管道传入", () => {
|
it("密码输入不是 TTY 时明确拒绝通过管道传入", () => {
|
||||||
const dataDir = mkdtempSync(path.join(tmpdir(), "tallynote-admin-init-"));
|
const dataDir = mkdtempSync(path.join(tmpdir(), "tallynote-admin-init-"));
|
||||||
try {
|
try {
|
||||||
|
|||||||
Executable
+66
@@ -0,0 +1,66 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""Minimal cross-platform pty driver for the admin-init CLI tests.
|
||||||
|
|
||||||
|
Forks a child on a real pseudo-terminal so the CLI sees a TTY and runs its
|
||||||
|
raw-mode password prompts. Forwards a prepared input file to the child's stdin
|
||||||
|
and copies child output to stdout. Writes the child's exit code to a file so
|
||||||
|
the Node test can read it deterministically.
|
||||||
|
|
||||||
|
Used instead of `expect` (not installed on CI) or BSD `script` (injects a stray
|
||||||
|
EOT byte when stdin is a regular file, corrupting the first prompt value).
|
||||||
|
"""
|
||||||
|
import os
|
||||||
|
import pty
|
||||||
|
import select
|
||||||
|
import sys
|
||||||
|
|
||||||
|
argv = sys.argv[1:]
|
||||||
|
exit_file = os.environ.get("PTY_EXIT_FILE", "")
|
||||||
|
stdin_file = os.environ.get("PTY_STDIN_FILE", "")
|
||||||
|
|
||||||
|
pid, master = pty.fork()
|
||||||
|
if pid == 0:
|
||||||
|
# Child: replace with the target command. argv[0] is an absolute node path.
|
||||||
|
os.execvp(argv[0], argv)
|
||||||
|
os._exit(127)
|
||||||
|
|
||||||
|
in_fd = os.open(stdin_file, os.O_RDONLY) if stdin_file else -1
|
||||||
|
open_stdin = in_fd >= 0
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
fds = [master]
|
||||||
|
if open_stdin:
|
||||||
|
fds.append(in_fd)
|
||||||
|
try:
|
||||||
|
readable, _, _ = select.select(fds, [], [], 30.0)
|
||||||
|
except (OSError, ValueError):
|
||||||
|
break
|
||||||
|
if not readable:
|
||||||
|
break
|
||||||
|
if master in readable:
|
||||||
|
try:
|
||||||
|
data = os.read(master, 4096)
|
||||||
|
except OSError:
|
||||||
|
break
|
||||||
|
if not data:
|
||||||
|
break
|
||||||
|
os.write(1, data)
|
||||||
|
if open_stdin and in_fd in readable:
|
||||||
|
data = os.read(in_fd, 4096)
|
||||||
|
if data:
|
||||||
|
os.write(master, data)
|
||||||
|
else:
|
||||||
|
open_stdin = False
|
||||||
|
os.close(in_fd)
|
||||||
|
finally:
|
||||||
|
try:
|
||||||
|
_, status = os.waitpid(pid, 0)
|
||||||
|
except ChildProcessError:
|
||||||
|
status = 0
|
||||||
|
code = os.waitstatus_to_exitcode(status) if hasattr(os, "waitstatus_to_exitcode") else (status >> 8)
|
||||||
|
if exit_file:
|
||||||
|
try:
|
||||||
|
with open(exit_file, "w") as handle:
|
||||||
|
handle.write(str(code))
|
||||||
|
except OSError:
|
||||||
|
pass
|
||||||
@@ -155,6 +155,53 @@ describe("更新 API", () => {
|
|||||||
expect(checked.json().latest).toMatchObject({ version: "1.3.0", isNewer: true });
|
expect(checked.json().latest).toMatchObject({ version: "1.3.0", isNewer: true });
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it("不会应用已经等于当前版本的暂存更新", async () => {
|
||||||
|
const session = await login("update-staged-current");
|
||||||
|
const admin = database.sqlite.prepare("SELECT id FROM admins WHERE username=?").get("update-staged-current") as { id: string };
|
||||||
|
const now = Date.now();
|
||||||
|
const stagedId = randomUUID();
|
||||||
|
database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(
|
||||||
|
id, admin_id, operation, status, version, platform, release_url,
|
||||||
|
asset_name, asset_url, expected_sha256, actual_sha256, download_path,
|
||||||
|
created_at, updated_at
|
||||||
|
) VALUES (?, ?, 'download', 'staged', ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
|
`).run(
|
||||||
|
stagedId,
|
||||||
|
admin.id,
|
||||||
|
config.appVersion,
|
||||||
|
detectPlatform().target,
|
||||||
|
config.updateMetadataUrl,
|
||||||
|
"current.tar.gz",
|
||||||
|
"https://updates.example/current.tar.gz",
|
||||||
|
"c".repeat(64),
|
||||||
|
"c".repeat(64),
|
||||||
|
path.join(config.dataDir, "staged-current"),
|
||||||
|
now,
|
||||||
|
now,
|
||||||
|
);
|
||||||
|
|
||||||
|
const apply = await app.inject({
|
||||||
|
method: "POST",
|
||||||
|
url: "/api/update/apply",
|
||||||
|
headers: { origin: config.publicOrigin, cookie: session.cookies, "x-csrf-token": session.csrf },
|
||||||
|
payload: { jobId: stagedId, version: config.appVersion, confirm: true },
|
||||||
|
});
|
||||||
|
expect(apply.statusCode).toBe(409);
|
||||||
|
// Reconciliation expires same-version staged jobs before the apply route
|
||||||
|
// can consume them, so the public response is the generic not-staged
|
||||||
|
// conflict while the database records the precise expiry reason.
|
||||||
|
expect(apply.json().error.code).toBe("UPDATE_NOT_STAGED");
|
||||||
|
expect(database.sqlite.prepare("SELECT status, error_message AS errorMessage FROM update_jobs WHERE id=?").get(stagedId)).toEqual({
|
||||||
|
status: "failed",
|
||||||
|
errorMessage: "暂存更新已过期,当前版本无需再次升级",
|
||||||
|
});
|
||||||
|
|
||||||
|
const status = await app.inject({ method: "GET", url: "/api/update/status", headers: { cookie: session.cookies } });
|
||||||
|
expect(status.statusCode).toBe(200);
|
||||||
|
expect(status.json().job).toBeNull();
|
||||||
|
});
|
||||||
|
|
||||||
it("更新任务只对发起管理员可见,并隐藏内部错误详情", async () => {
|
it("更新任务只对发起管理员可见,并隐藏内部错误详情", async () => {
|
||||||
const owner = await login("update-owner");
|
const owner = await login("update-owner");
|
||||||
const other = await login("update-other");
|
const other = await login("update-other");
|
||||||
@@ -176,6 +223,42 @@ describe("更新 API", () => {
|
|||||||
expect(ownDetail.json().job.errorMessage).toBe("更新失败,请查看服务器日志或重试");
|
expect(ownDetail.json().job.errorMessage).toBe("更新失败,请查看服务器日志或重试");
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it("取消任务按管理员隔离,并只删除匹配任务的请求文件", async () => {
|
||||||
|
const owner = await login("cancel-owner");
|
||||||
|
const other = await login("cancel-other");
|
||||||
|
const ownerId = (database.sqlite.prepare("SELECT id FROM admins WHERE username=?").get("cancel-owner") as { id: string }).id;
|
||||||
|
const otherId = (database.sqlite.prepare("SELECT id FROM admins WHERE username=?").get("cancel-other") as { id: string }).id;
|
||||||
|
const now = Date.now();
|
||||||
|
const ownerJobId = randomUUID();
|
||||||
|
const otherJobId = randomUUID();
|
||||||
|
const insert = database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, admin_id, operation, status, version, platform, asset_url, created_at, updated_at)
|
||||||
|
VALUES (?, ?, 'download', 'queued', '1.3.0', ?, 'https://updates.example/update.tar.gz', ?, ?)
|
||||||
|
`);
|
||||||
|
insert.run(ownerJobId, ownerId, detectPlatform().target, now, now);
|
||||||
|
insert.run(otherJobId, otherId, detectPlatform().target, now + 1, now + 1);
|
||||||
|
await import("node:fs/promises").then(({ writeFile }) => writeFile(config.updateRequestPath, JSON.stringify({ jobId: otherJobId }), { encoding: "utf8", mode: 0o600 }));
|
||||||
|
|
||||||
|
const ownerCancel = await app.inject({
|
||||||
|
method: "POST", url: "/api/update/cancel",
|
||||||
|
headers: { origin: config.publicOrigin, cookie: owner.cookies, "x-csrf-token": owner.csrf },
|
||||||
|
payload: { jobId: ownerJobId },
|
||||||
|
});
|
||||||
|
expect(ownerCancel.statusCode).toBe(200);
|
||||||
|
expect((database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(ownerJobId) as { status: string }).status).toBe("cancelled");
|
||||||
|
expect((database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(otherJobId) as { status: string }).status).toBe("queued");
|
||||||
|
expect(existsSync(config.updateRequestPath)).toBe(true);
|
||||||
|
|
||||||
|
const otherCancel = await app.inject({
|
||||||
|
method: "POST", url: "/api/update/cancel",
|
||||||
|
headers: { origin: config.publicOrigin, cookie: other.cookies, "x-csrf-token": other.csrf },
|
||||||
|
payload: {},
|
||||||
|
});
|
||||||
|
expect(otherCancel.statusCode).toBe(200);
|
||||||
|
expect((database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(otherJobId) as { status: string }).status).toBe("cancelled");
|
||||||
|
expect(existsSync(config.updateRequestPath)).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
it("应用前重新校验失败时写入失败审计", async () => {
|
it("应用前重新校验失败时写入失败审计", async () => {
|
||||||
const session = await login("update-audit");
|
const session = await login("update-audit");
|
||||||
globalThis.fetch = (async () => new Response("upstream unavailable", { status: 503 })) as typeof fetch;
|
globalThis.fetch = (async () => new Response("upstream unavailable", { status: 503 })) as typeof fetch;
|
||||||
|
|||||||
+28
-7
@@ -286,8 +286,27 @@ describe("更新安全工具", () => {
|
|||||||
expect(await readFile(path.join(config.currentLink, "dist", "marker"), "utf8")).toBe("new");
|
expect(await readFile(path.join(config.currentLink, "dist", "marker"), "utf8")).toBe("new");
|
||||||
const row = database.sqlite.prepare("SELECT operation, status FROM update_jobs WHERE id=?").get(jobId);
|
const row = database.sqlite.prepare("SELECT operation, status FROM update_jobs WHERE id=?").get(jobId);
|
||||||
expect(row).toEqual({ operation: "apply", status: "applying" });
|
expect(row).toEqual({ operation: "apply", status: "applying" });
|
||||||
finalizeUpdateJob(database.sqlite, jobId, "failed");
|
finalizeUpdateJob(database.sqlite, jobId, "failed", "健康检查失败(自定义)");
|
||||||
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
expect(database.sqlite.prepare("SELECT status, error_message AS errorMessage FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed", errorMessage: "健康检查失败(自定义)" });
|
||||||
|
finalizeUpdateJob(database.sqlite, jobId, "failed", "第二次 finalize 不应覆盖原消息");
|
||||||
|
expect(database.sqlite.prepare("SELECT status, error_message AS errorMessage FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed", errorMessage: "健康检查失败(自定义)" });
|
||||||
|
expect(database.sqlite.prepare("SELECT COUNT(*) AS count FROM audit_events WHERE action='update.failed' AND target_id=?").get(jobId)).toEqual({ count: 1 });
|
||||||
|
const defaultJobId = randomUUID();
|
||||||
|
database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, operation, status, version, platform, asset_url, created_at, updated_at)
|
||||||
|
VALUES (?, 'apply', 'applying', '1.3.0', ?, ?, ?, ?)
|
||||||
|
`).run(defaultJobId, detectPlatform().target, "https://updates.example/" + assetName, now, now);
|
||||||
|
finalizeUpdateJob(database.sqlite, defaultJobId, "failed", "");
|
||||||
|
expect(database.sqlite.prepare("SELECT error_message AS errorMessage FROM update_jobs WHERE id=?").get(defaultJobId)).toEqual({ errorMessage: "新版本健康检查失败,已恢复上一版本" });
|
||||||
|
const completedJobId = randomUUID();
|
||||||
|
database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, operation, status, version, platform, asset_url, created_at, updated_at)
|
||||||
|
VALUES (?, 'apply', 'completed', '1.3.0', ?, ?, ?, ?)
|
||||||
|
`).run(completedJobId, detectPlatform().target, "https://updates.example/" + assetName, now, now);
|
||||||
|
finalizeUpdateJob(database.sqlite, completedJobId, "completed");
|
||||||
|
expect(database.sqlite.prepare("SELECT COUNT(*) AS count FROM audit_events WHERE action='update.completed' AND target_id=?").get(completedJobId)).toEqual({ count: 0 });
|
||||||
|
expect(() => finalizeUpdateJob(database.sqlite, completedJobId, "failed", "不能降级已完成任务")).toThrow("更新任务状态不允许完成");
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(completedJobId)).toEqual({ status: "completed" });
|
||||||
} finally {
|
} finally {
|
||||||
globalThis.fetch = previousFetch;
|
globalThis.fetch = previousFetch;
|
||||||
if (database) database.sqlite.close();
|
if (database) database.sqlite.close();
|
||||||
@@ -385,7 +404,7 @@ describe("更新安全工具", () => {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
it("队列任务有新请求标记时可被重新检查,标记过期后才回收", async () => {
|
it("队列任务有匹配请求标记时保留到租约过期,过期后才回收", async () => {
|
||||||
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-queued-marker-"));
|
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-queued-marker-"));
|
||||||
let database: ReturnType<typeof openDatabase> | undefined;
|
let database: ReturnType<typeof openDatabase> | undefined;
|
||||||
try {
|
try {
|
||||||
@@ -412,12 +431,14 @@ describe("更新安全工具", () => {
|
|||||||
const now = Date.now();
|
const now = Date.now();
|
||||||
await utimes(config.updateRequestPath, new Date(now), new Date(now));
|
await utimes(config.updateRequestPath, new Date(now), new Date(now));
|
||||||
|
|
||||||
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, now)).toBe(1);
|
// The DB row is old, but the request marker is fresh and names this
|
||||||
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
// exact job. Keep it queued while systemd has a chance to consume it.
|
||||||
await expect(stat(config.updateRequestPath)).rejects.toThrow();
|
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, now)).toBe(0);
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "queued" });
|
||||||
|
await expect(stat(config.updateRequestPath)).resolves.toBeTruthy();
|
||||||
|
|
||||||
const expiredNow = now + ORPHANED_UPDATE_TIMEOUT_MS + 1;
|
const expiredNow = now + ORPHANED_UPDATE_TIMEOUT_MS + 1;
|
||||||
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, expiredNow)).toBe(0);
|
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, expiredNow)).toBe(1);
|
||||||
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
||||||
await expect(stat(config.updateRequestPath)).rejects.toThrow();
|
await expect(stat(config.updateRequestPath)).rejects.toThrow();
|
||||||
} finally {
|
} finally {
|
||||||
|
|||||||
@@ -32,9 +32,9 @@ function App() {
|
|||||||
const logoutInFlight = useRef(false);
|
const logoutInFlight = useRef(false);
|
||||||
useDialogAccessibility();
|
useDialogAccessibility();
|
||||||
const notify = useCallback((message: string, kind: "success" | "error" | "info" = "info") => {
|
const notify = useCallback((message: string, kind: "success" | "error" | "info" = "info") => {
|
||||||
// The placement container owns the responsive right inset. Keeping the
|
// Keep notices in the lower-right safe area so they do not compete with
|
||||||
// item offset at zero avoids pushing narrow-screen notices off canvas.
|
// the header controls or obscure the page title.
|
||||||
const options = { content: message, duration: 4200, placement: "top-right" as const, offset: [24, 76] as [number, number], zIndex: 6000 };
|
const options = { content: message, duration: 4200, placement: "bottom-right" as const, offset: [24, 24] as [number, number], zIndex: 6000 };
|
||||||
const show = kind === "success" ? NotificationPlugin.success : kind === "error" ? NotificationPlugin.error : NotificationPlugin.info;
|
const show = kind === "success" ? NotificationPlugin.success : kind === "error" ? NotificationPlugin.error : NotificationPlugin.info;
|
||||||
void show(options);
|
void show(options);
|
||||||
}, []);
|
}, []);
|
||||||
|
|||||||
@@ -248,29 +248,53 @@ export default function UpdatePage({
|
|||||||
const announced = useRef<string | null>(null);
|
const announced = useRef<string | null>(null);
|
||||||
const checkInFlight = useRef(false);
|
const checkInFlight = useRef(false);
|
||||||
const actionInFlight = useRef(false);
|
const actionInFlight = useRef(false);
|
||||||
|
const cancelInFlight = useRef(false);
|
||||||
|
const loadInFlight = useRef(false);
|
||||||
const disconnected = useRef(false);
|
const disconnected = useRef(false);
|
||||||
const recoveredNotice = useRef(false);
|
const recoveredNotice = useRef(false);
|
||||||
|
|
||||||
const info = liveInfo;
|
const info = liveInfo;
|
||||||
|
|
||||||
const load = async () => {
|
const mergeInfo = (incoming: UpdateInfo, preserveJob = false) => {
|
||||||
setLoading(true);
|
setLiveInfo((current) => {
|
||||||
|
if (!current) return incoming;
|
||||||
|
const next = {
|
||||||
|
...current,
|
||||||
|
...incoming,
|
||||||
|
// A check response describes the release, but must not erase the job
|
||||||
|
// that is already being displayed while that check is in flight.
|
||||||
|
job: preserveJob
|
||||||
|
? (current.job ?? incoming.job)
|
||||||
|
: (!Object.prototype.hasOwnProperty.call(incoming, "job") ? current.job : incoming.job),
|
||||||
|
};
|
||||||
|
// Some status/check responses are intentionally partial. Keep fields
|
||||||
|
// from the last successful response when a field is omitted.
|
||||||
|
if (!Object.prototype.hasOwnProperty.call(incoming, "latest")) next.latest = current.latest;
|
||||||
|
if (!Object.prototype.hasOwnProperty.call(incoming, "checkedAt")) next.checkedAt = current.checkedAt;
|
||||||
|
if (!Object.prototype.hasOwnProperty.call(incoming, "strategy")) next.strategy = current.strategy;
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const load = async (showLoading = true): Promise<UpdateInfo | null> => {
|
||||||
|
if (loadInFlight.current) return null;
|
||||||
|
loadInFlight.current = true;
|
||||||
|
if (showLoading) setLoading(true);
|
||||||
setError("");
|
setError("");
|
||||||
try {
|
try {
|
||||||
const res = await api<UpdateInfo>("/api/update/status");
|
const res = await api<UpdateInfo>("/api/update/status");
|
||||||
setLiveInfo(res);
|
mergeInfo(res);
|
||||||
updateInfoCache = res;
|
updateInfoCache = res;
|
||||||
|
return res;
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
setError((e as Error).message);
|
setError((e as Error).message);
|
||||||
|
return null;
|
||||||
} finally {
|
} finally {
|
||||||
setLoading(false);
|
if (showLoading) setLoading(false);
|
||||||
|
loadInFlight.current = false;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
void load();
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
// Keep terminal jobs visible so operators can understand what happened and
|
// Keep terminal jobs visible so operators can understand what happened and
|
||||||
// recover without guessing. The polling effect below only polls active jobs.
|
// recover without guessing. The polling effect below only polls active jobs.
|
||||||
const job = info?.job ?? null;
|
const job = info?.job ?? null;
|
||||||
@@ -293,7 +317,8 @@ export default function UpdatePage({
|
|||||||
return () => window.clearInterval(timer);
|
return () => window.clearInterval(timer);
|
||||||
}, [restartAt]);
|
}, [restartAt]);
|
||||||
|
|
||||||
// Live polling effect: only poll when active job exists
|
// Live polling effect for an active job. Completed responses are refreshed
|
||||||
|
// once so latest/checkedAt/strategy stay current; other terminal states stay visible.
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const shouldPoll = Boolean(
|
const shouldPoll = Boolean(
|
||||||
job && (pollableStatuses.has(job.status) || (job.status === "staged" && job.operation === "apply"))
|
job && (pollableStatuses.has(job.status) || (job.status === "staged" && job.operation === "apply"))
|
||||||
@@ -326,23 +351,53 @@ export default function UpdatePage({
|
|||||||
setReloadReady(true);
|
setReloadReady(true);
|
||||||
notify?.("系统升级完成,请刷新页面", "success");
|
notify?.("系统升级完成,请刷新页面", "success");
|
||||||
}
|
}
|
||||||
if (
|
if (result.job.status === "completed") {
|
||||||
|
void load(false);
|
||||||
|
} else if (
|
||||||
pollableStatuses.has(result.job.status) ||
|
pollableStatuses.has(result.job.status) ||
|
||||||
(result.job.status === "staged" && result.job.operation === "apply")
|
(result.job.status === "staged" && result.job.operation === "apply")
|
||||||
) {
|
) {
|
||||||
schedule(1500);
|
schedule(1500);
|
||||||
}
|
}
|
||||||
} catch {
|
} catch (caught) {
|
||||||
if (disposed) return;
|
if (disposed) return;
|
||||||
|
const status = caught instanceof ApiError ? caught.status : 0;
|
||||||
|
const message = caught instanceof Error ? caught.message : "读取更新任务状态失败";
|
||||||
|
if (status === 404) {
|
||||||
|
setPollError("更新任务已结束,正在刷新状态。");
|
||||||
|
void load(false);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (status === 401) {
|
||||||
|
setPollError("登录已失效,请重新登录。");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (status === 403) {
|
||||||
|
setPollError("没有权限读取更新任务状态。");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
failures += 1;
|
failures += 1;
|
||||||
disconnected.current = true;
|
disconnected.current = status === 0 || status === 408;
|
||||||
recoveredNotice.current = false;
|
recoveredNotice.current = false;
|
||||||
setPollError(
|
if (status === 409) {
|
||||||
`服务正在平滑重启${
|
setPollError(`${message},正在刷新状态。`);
|
||||||
restartSeconds !== null ? `,预计 ${restartSeconds} 秒后恢复` : ",页面正在自动探活重试"
|
void load(false);
|
||||||
}。升级任务仍在后台安全执行。`
|
return;
|
||||||
);
|
}
|
||||||
schedule(Math.min(1500 * 2 ** Math.min(failures, 3), 10_000));
|
if (status === 429) {
|
||||||
|
setPollError(`${message},稍后继续同步。`);
|
||||||
|
} else if (status === 408) {
|
||||||
|
setPollError("读取更新任务状态超时,正在重试。");
|
||||||
|
} else if (status === 0) {
|
||||||
|
setPollError("更新服务连接异常,正在重试。");
|
||||||
|
} else {
|
||||||
|
setPollError(`${message},正在重试。`);
|
||||||
|
}
|
||||||
|
const retryAfter = caught instanceof ApiError && caught.retryAfter
|
||||||
|
? Math.max(1000, caught.retryAfter * 1000)
|
||||||
|
: Math.min(1500 * 2 ** Math.min(failures, 3), 10_000);
|
||||||
|
schedule(retryAfter);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
void poll();
|
void poll();
|
||||||
@@ -352,6 +407,42 @@ export default function UpdatePage({
|
|||||||
};
|
};
|
||||||
}, [job?.id, job?.status, job?.operation, notify]);
|
}, [job?.id, job?.status, job?.operation, notify]);
|
||||||
|
|
||||||
|
// With no visible job, make a low-frequency status request so a task created
|
||||||
|
// elsewhere can still appear without creating a request storm.
|
||||||
|
useEffect(() => {
|
||||||
|
if (job) return;
|
||||||
|
let timer: number | undefined;
|
||||||
|
let disposed = false;
|
||||||
|
const discover = () => {
|
||||||
|
if (disposed || document.visibilityState !== "visible") return;
|
||||||
|
void load(false);
|
||||||
|
};
|
||||||
|
const schedule = () => {
|
||||||
|
if (disposed) return;
|
||||||
|
if (timer !== undefined) window.clearTimeout(timer);
|
||||||
|
timer = window.setTimeout(() => {
|
||||||
|
discover();
|
||||||
|
schedule();
|
||||||
|
}, 5000);
|
||||||
|
};
|
||||||
|
const onVisibilityChange = () => {
|
||||||
|
if (document.visibilityState === "visible") {
|
||||||
|
discover();
|
||||||
|
schedule();
|
||||||
|
} else if (timer !== undefined) {
|
||||||
|
window.clearTimeout(timer);
|
||||||
|
timer = undefined;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
if (document.visibilityState === "visible") schedule();
|
||||||
|
document.addEventListener("visibilitychange", onVisibilityChange);
|
||||||
|
return () => {
|
||||||
|
disposed = true;
|
||||||
|
if (timer !== undefined) window.clearTimeout(timer);
|
||||||
|
document.removeEventListener("visibilitychange", onVisibilityChange);
|
||||||
|
};
|
||||||
|
}, [job?.id, job?.status, job?.operation]);
|
||||||
|
|
||||||
// Check update handler
|
// Check update handler
|
||||||
const check = async () => {
|
const check = async () => {
|
||||||
if (checkInFlight.current) return;
|
if (checkInFlight.current) return;
|
||||||
@@ -359,7 +450,17 @@ export default function UpdatePage({
|
|||||||
setChecking(true);
|
setChecking(true);
|
||||||
setError("");
|
setError("");
|
||||||
try {
|
try {
|
||||||
setLiveInfo(await api<UpdateInfo>("/api/update/check", { method: "POST" }));
|
const result = await api<UpdateInfo>("/api/update/check", { method: "POST" });
|
||||||
|
// The check endpoint returns release metadata but not task state. Read
|
||||||
|
// the status endpoint once more so reconciliation performed before the
|
||||||
|
// check is authoritative: an expired staged task must disappear from
|
||||||
|
// this page immediately instead of surviving until a full refresh.
|
||||||
|
const status = await api<UpdateInfo>("/api/update/status");
|
||||||
|
setLiveInfo((current) => ({
|
||||||
|
...(current ?? result),
|
||||||
|
...result,
|
||||||
|
job: status.job,
|
||||||
|
}));
|
||||||
notify?.("版本检查完成", "info");
|
notify?.("版本检查完成", "info");
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
if (e instanceof ApiError && e.status === 429) {
|
if (e instanceof ApiError && e.status === 429) {
|
||||||
@@ -373,9 +474,28 @@ export default function UpdatePage({
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
let disposed = false;
|
||||||
|
const bootstrap = async () => {
|
||||||
|
const snapshot = await load();
|
||||||
|
if (disposed || !snapshot) return;
|
||||||
|
// Status may be satisfied from the release cache. Refresh it on entry
|
||||||
|
// only when the cached result is absent or older than one minute; this
|
||||||
|
// keeps the page current without turning navigation into a burst of
|
||||||
|
// rate-limited checks.
|
||||||
|
const checkedAt = snapshot.checkedAt || 0;
|
||||||
|
if (!checkedAt || !snapshot.latest || Date.now() - checkedAt > 60_000) await check();
|
||||||
|
};
|
||||||
|
void bootstrap();
|
||||||
|
return () => {
|
||||||
|
disposed = true;
|
||||||
|
};
|
||||||
|
}, []);
|
||||||
|
|
||||||
// Cancel queued job handler
|
// Cancel queued job handler
|
||||||
const cancelJob = async () => {
|
const cancelJob = async () => {
|
||||||
if (cancelling) return;
|
if (cancelInFlight.current || !job?.id) return;
|
||||||
|
cancelInFlight.current = true;
|
||||||
setCancelling(true);
|
setCancelling(true);
|
||||||
try {
|
try {
|
||||||
await api("/api/update/cancel", { method: "POST", body: JSON.stringify({ jobId: job?.id }) });
|
await api("/api/update/cancel", { method: "POST", body: JSON.stringify({ jobId: job?.id }) });
|
||||||
@@ -387,6 +507,7 @@ export default function UpdatePage({
|
|||||||
notify?.((e as Error).message, "error");
|
notify?.((e as Error).message, "error");
|
||||||
} finally {
|
} finally {
|
||||||
setCancelling(false);
|
setCancelling(false);
|
||||||
|
cancelInFlight.current = false;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -443,9 +564,20 @@ export default function UpdatePage({
|
|||||||
|
|
||||||
const latest = info?.latest;
|
const latest = info?.latest;
|
||||||
const notes = notesFor(latest);
|
const notes = notesFor(latest);
|
||||||
|
const hasChecked = Boolean(info?.checkedAt);
|
||||||
|
const releaseState = checking
|
||||||
|
? "checking"
|
||||||
|
: !hasChecked
|
||||||
|
? "unverified"
|
||||||
|
: !latest
|
||||||
|
? "unavailable"
|
||||||
|
: latest.isNewer
|
||||||
|
? latest.compatible && latest.integrityReady ? "available" : "blocked"
|
||||||
|
: "up-to-date";
|
||||||
|
|
||||||
const canDownload = Boolean(
|
const canDownload = Boolean(
|
||||||
info?.strategy === "systemd" &&
|
info?.strategy === "systemd" &&
|
||||||
|
!checking &&
|
||||||
latest?.isNewer &&
|
latest?.isNewer &&
|
||||||
latest.compatible &&
|
latest.compatible &&
|
||||||
latest.integrityReady &&
|
latest.integrityReady &&
|
||||||
@@ -454,14 +586,19 @@ export default function UpdatePage({
|
|||||||
|
|
||||||
const canApply = Boolean(
|
const canApply = Boolean(
|
||||||
info?.strategy === "systemd" &&
|
info?.strategy === "systemd" &&
|
||||||
|
!checking &&
|
||||||
job &&
|
job &&
|
||||||
job.status === "staged" &&
|
job.status === "staged" &&
|
||||||
job.operation === "download"
|
job.operation === "download" &&
|
||||||
|
latest?.isNewer &&
|
||||||
|
latest.version === job.version &&
|
||||||
|
job.version !== info.currentVersion
|
||||||
);
|
);
|
||||||
|
|
||||||
const hasActiveJob = Boolean(
|
const hasActiveJob = Boolean(
|
||||||
job && activeStatuses.has(job.status) && job.status !== "staged"
|
job && activeStatuses.has(job.status) && !(job.status === "staged" && job.operation === "download")
|
||||||
);
|
);
|
||||||
|
const showJobDetails = hasActiveJob || canApply || Boolean(job?.status === "staged" && job.operation === "apply");
|
||||||
|
|
||||||
// Compute current pipeline step index (0: check, 1: download, 2: verify/stage, 3: apply/restart)
|
// Compute current pipeline step index (0: check, 1: download, 2: verify/stage, 3: apply/restart)
|
||||||
const currentStep = useMemo(() => {
|
const currentStep = useMemo(() => {
|
||||||
@@ -487,7 +624,7 @@ export default function UpdatePage({
|
|||||||
if (!job) return 0;
|
if (!job) return 0;
|
||||||
if (job.status === "completed" || job.status === "staged") return 100;
|
if (job.status === "completed" || job.status === "staged") return 100;
|
||||||
if (job.status === "downloading") {
|
if (job.status === "downloading") {
|
||||||
if (!job.sizeBytes || !job.downloadedBytes) return 10;
|
if (!job.sizeBytes || !job.downloadedBytes) return 0;
|
||||||
return Math.min(99, Math.max(1, Math.round((job.downloadedBytes / job.sizeBytes) * 100)));
|
return Math.min(99, Math.max(1, Math.round((job.downloadedBytes / job.sizeBytes) * 100)));
|
||||||
}
|
}
|
||||||
if (job.status === "verifying") return 99;
|
if (job.status === "verifying") return 99;
|
||||||
@@ -518,7 +655,7 @@ export default function UpdatePage({
|
|||||||
subtitle="管理系统版本升级、更新包完整性校验与安全热重启"
|
subtitle="管理系统版本升级、更新包完整性校验与安全热重启"
|
||||||
actions={
|
actions={
|
||||||
<div className="tn-update-page-actions">
|
<div className="tn-update-page-actions">
|
||||||
{hasActiveJob && (
|
{showJobDetails && (
|
||||||
<Button
|
<Button
|
||||||
theme="primary"
|
theme="primary"
|
||||||
variant="base"
|
variant="base"
|
||||||
@@ -589,10 +726,18 @@ export default function UpdatePage({
|
|||||||
<span className="tn-metric-label">当前运行版本</span>
|
<span className="tn-metric-label">当前运行版本</span>
|
||||||
<div className="tn-metric-value">v{info.currentVersion}</div>
|
<div className="tn-metric-value">v{info.currentVersion}</div>
|
||||||
<div className="tn-metric-foot">
|
<div className="tn-metric-foot">
|
||||||
{latest?.isNewer ? (
|
{releaseState === "checking" ? (
|
||||||
<Tag theme="primary" size="small">可更新至 v{latest.version}</Tag>
|
<Tag theme="default" size="small">正在检查更新</Tag>
|
||||||
) : (
|
) : releaseState === "unverified" ? (
|
||||||
|
<Tag theme="default" size="small">尚未检查更新</Tag>
|
||||||
|
) : releaseState === "available" ? (
|
||||||
|
<Tag theme="primary" size="small">可更新至 v{latest?.version}</Tag>
|
||||||
|
) : releaseState === "up-to-date" ? (
|
||||||
<Tag theme="success" size="small">已是最新版本</Tag>
|
<Tag theme="success" size="small">已是最新版本</Tag>
|
||||||
|
) : releaseState === "blocked" ? (
|
||||||
|
<Tag theme="warning" size="small">发现新版本,但暂不可更新</Tag>
|
||||||
|
) : (
|
||||||
|
<Tag theme="default" size="small">暂未获取发布信息</Tag>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
</Surface>
|
</Surface>
|
||||||
@@ -631,8 +776,8 @@ export default function UpdatePage({
|
|||||||
<Surface className="tn-ascii-release-container">
|
<Surface className="tn-ascii-release-container">
|
||||||
<div className="tn-ascii-release-head">
|
<div className="tn-ascii-release-head">
|
||||||
<h3 className="tn-ascii-release-title">发布版本详情</h3>
|
<h3 className="tn-ascii-release-title">发布版本详情</h3>
|
||||||
<Tag theme={latest.isNewer ? "primary" : "success"} variant="light-outline">
|
<Tag theme={releaseState === "available" ? "primary" : releaseState === "up-to-date" ? "success" : releaseState === "blocked" ? "warning" : "default"} variant="light-outline">
|
||||||
{latest.isNewer ? "发现新版本" : "已是最新版本"}
|
{releaseState === "available" ? "发现新版本" : releaseState === "up-to-date" ? "已是最新版本" : releaseState === "blocked" ? "暂不可安全更新" : "尚未检查"}
|
||||||
</Tag>
|
</Tag>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -684,7 +829,7 @@ export default function UpdatePage({
|
|||||||
disabled={actionBusy}
|
disabled={actionBusy}
|
||||||
icon={<Download size={16} />}
|
icon={<Download size={16} />}
|
||||||
>
|
>
|
||||||
立即升级至 v{latest.version}
|
下载更新包
|
||||||
</Button>
|
</Button>
|
||||||
)}
|
)}
|
||||||
{canApply && (
|
{canApply && (
|
||||||
@@ -695,7 +840,7 @@ export default function UpdatePage({
|
|||||||
disabled={actionBusy}
|
disabled={actionBusy}
|
||||||
icon={<Zap size={16} />}
|
icon={<Zap size={16} />}
|
||||||
>
|
>
|
||||||
更新包已就绪,立即应用 (v{latest.version})
|
立即应用并重启 v{latest.version}
|
||||||
</Button>
|
</Button>
|
||||||
)}
|
)}
|
||||||
{hasActiveJob && (
|
{hasActiveJob && (
|
||||||
@@ -708,9 +853,14 @@ export default function UpdatePage({
|
|||||||
查看当前升级进度
|
查看当前升级进度
|
||||||
</Button>
|
</Button>
|
||||||
)}
|
)}
|
||||||
{!latest.isNewer && !hasActiveJob && (
|
{releaseState === "blocked" && !hasActiveJob && !canApply && (
|
||||||
|
<Button theme="default" variant="outline" size="large" onClick={() => void check()} loading={checking} disabled={checking} icon={<RefreshCw size={15} />}>
|
||||||
|
重新检查
|
||||||
|
</Button>
|
||||||
|
)}
|
||||||
|
{releaseState === "up-to-date" && !hasActiveJob && !canApply && (
|
||||||
<Button theme="default" variant="outline" size="large" onClick={() => void check()} icon={<RefreshCw size={15} />}>
|
<Button theme="default" variant="outline" size="large" onClick={() => void check()} icon={<RefreshCw size={15} />}>
|
||||||
检查新版本
|
重新检查
|
||||||
</Button>
|
</Button>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
@@ -718,10 +868,10 @@ export default function UpdatePage({
|
|||||||
) : (
|
) : (
|
||||||
<Surface className="tn-empty-surface">
|
<Surface className="tn-empty-surface">
|
||||||
<div className="tn-empty-content">
|
<div className="tn-empty-content">
|
||||||
<CheckCircle2 size={32} className="text-success" />
|
{releaseState === "unverified" || releaseState === "checking" ? <RefreshCw size={32} className={releaseState === "checking" ? "tn-spin" : "text-secondary"} /> : <AlertCircle size={32} className="text-warning" />}
|
||||||
<p>暂无待更新的版本信息,当前系统已是最新状态。</p>
|
<p>{releaseState === "unverified" || releaseState === "checking" ? "尚未完成版本检查,请先获取官方发布信息。" : "暂时没有可用的发布信息,请稍后重新检查。"}</p>
|
||||||
<Button variant="outline" onClick={() => void check()} icon={<RefreshCw size={15} />}>
|
<Button variant="outline" onClick={() => void check()} loading={checking} disabled={checking} icon={<RefreshCw size={15} />}>
|
||||||
检查新版本
|
{releaseState === "checking" ? "正在检查" : "检查新版本"}
|
||||||
</Button>
|
</Button>
|
||||||
</div>
|
</div>
|
||||||
</Surface>
|
</Surface>
|
||||||
@@ -730,7 +880,7 @@ export default function UpdatePage({
|
|||||||
{latest && notes && (
|
{latest && notes && (
|
||||||
<Surface className="tn-ascii-notes-container">
|
<Surface className="tn-ascii-notes-container">
|
||||||
<div className="tn-ascii-notes-head">
|
<div className="tn-ascii-notes-head">
|
||||||
<h3 className="tn-ascii-notes-title">本次版本更新说明</h3>
|
<h3 className="tn-ascii-notes-title">发布说明</h3>
|
||||||
</div>
|
</div>
|
||||||
<div className="tn-ascii-notes-body">
|
<div className="tn-ascii-notes-body">
|
||||||
<MarkdownNotes value={notes} />
|
<MarkdownNotes value={notes} />
|
||||||
@@ -755,7 +905,7 @@ export default function UpdatePage({
|
|||||||
{/* Unified Single Upgrade Modal (800px width on desktop) */}
|
{/* Unified Single Upgrade Modal (800px width on desktop) */}
|
||||||
<Dialog
|
<Dialog
|
||||||
visible={showUpgradeModal}
|
visible={showUpgradeModal}
|
||||||
header={`系统升级控制台 · v${latest?.version || ""}`}
|
header={`系统升级控制台 · v${latest?.version || job?.version || ""}`}
|
||||||
className="tn-dialog-large"
|
className="tn-dialog-large"
|
||||||
width="820px"
|
width="820px"
|
||||||
footer={null}
|
footer={null}
|
||||||
@@ -887,7 +1037,7 @@ export default function UpdatePage({
|
|||||||
)}
|
)}
|
||||||
|
|
||||||
{/* 场景 C: 校验通过准备就绪 (staged) (Exact ASCII) */}
|
{/* 场景 C: 校验通过准备就绪 (staged) (Exact ASCII) */}
|
||||||
{job?.status === "staged" && (
|
{job?.status === "staged" && job.operation === "download" && canApply && (
|
||||||
<div className="tn-modal-card-box">
|
<div className="tn-modal-card-box">
|
||||||
<div className="tn-modal-card-title" style={{ color: "#2f7d5c", display: "flex", alignItems: "center", gap: 8 }}>
|
<div className="tn-modal-card-title" style={{ color: "#2f7d5c", display: "flex", alignItems: "center", gap: 8 }}>
|
||||||
<CheckCircle2 size={18} />
|
<CheckCircle2 size={18} />
|
||||||
@@ -915,6 +1065,36 @@ export default function UpdatePage({
|
|||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
|
||||||
|
{/* A staged archive can become obsolete when the host or release
|
||||||
|
metadata changes while this page is open. Keep the state visible
|
||||||
|
but remove the apply action; the server will reconcile it on the
|
||||||
|
next status request and the operator can perform a fresh check. */}
|
||||||
|
{job?.status === "staged" && job.operation === "download" && !canApply && (
|
||||||
|
<div className="tn-modal-card-box">
|
||||||
|
<div className="tn-modal-card-title">暂存更新已失效</div>
|
||||||
|
<p className="text-secondary" style={{ fontSize: "13px", margin: "0 0 16px", lineHeight: 1.6 }}>
|
||||||
|
这个更新包已不是当前可安全应用的版本,系统不会重复应用。请重新检查更新以获取最新发布信息。
|
||||||
|
</p>
|
||||||
|
<div className="tn-modal-actions-bar">
|
||||||
|
<Button variant="outline" onClick={() => void check()} loading={checking} disabled={checking}>
|
||||||
|
重新检查
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{job?.status === "staged" && job.operation === "apply" && (
|
||||||
|
<div className="tn-modal-card-box">
|
||||||
|
<div className="tn-modal-card-title">
|
||||||
|
<div style={{ marginBottom: 12 }}><BeamBar width={180} /></div>
|
||||||
|
应用请求已提交,正在等待更新服务接管
|
||||||
|
</div>
|
||||||
|
<p className="text-secondary" style={{ fontSize: "13px", margin: "0 0 16px", lineHeight: 1.6 }}>
|
||||||
|
系统正在准备备份与重启。页面会持续同步服务状态,请不要重复提交。
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* 场景 D: 数据快照备份中或服务重启中 (backing_up / applying) (Exact ASCII) */}
|
{/* 场景 D: 数据快照备份中或服务重启中 (backing_up / applying) (Exact ASCII) */}
|
||||||
{(job?.status === "backing_up" || job?.status === "applying") && (
|
{(job?.status === "backing_up" || job?.status === "applying") && (
|
||||||
<div className="tn-modal-card-box">
|
<div className="tn-modal-card-box">
|
||||||
@@ -976,7 +1156,7 @@ export default function UpdatePage({
|
|||||||
)}
|
)}
|
||||||
|
|
||||||
{/* Footer close button */}
|
{/* Footer close button */}
|
||||||
{job?.status !== "staged" && job?.status !== "completed" && !confirmReadyToDownload && (
|
{!(job?.status === "staged" && canApply) && job?.status !== "completed" && !confirmReadyToDownload && (
|
||||||
<div className="tn-modal-footer-close">
|
<div className="tn-modal-footer-close">
|
||||||
<Button variant="outline" onClick={() => setShowUpgradeModal(false)}>
|
<Button variant="outline" onClick={() => setShowUpgradeModal(false)}>
|
||||||
关闭窗口(后台继续运行)
|
关闭窗口(后台继续运行)
|
||||||
|
|||||||
@@ -115,14 +115,14 @@ input:focus-visible, textarea:focus-visible, select:focus-visible {
|
|||||||
box-sizing: border-box !important;
|
box-sizing: border-box !important;
|
||||||
transition: all 0.22s cubic-bezier(0.16, 1, 0.3, 1) !important;
|
transition: all 0.22s cubic-bezier(0.16, 1, 0.3, 1) !important;
|
||||||
}
|
}
|
||||||
.t-notification__show--top-right {
|
.t-notification__show--bottom-right {
|
||||||
top: 76px !important;
|
bottom: 24px !important;
|
||||||
right: 24px !important;
|
right: 24px !important;
|
||||||
z-index: 6000 !important;
|
z-index: 6000 !important;
|
||||||
}
|
}
|
||||||
@media (max-width: 768px) {
|
@media (max-width: 768px) {
|
||||||
.t-notification__show--top-right {
|
.t-notification__show--bottom-right {
|
||||||
top: 16px !important;
|
bottom: 16px !important;
|
||||||
right: 16px !important;
|
right: 16px !important;
|
||||||
left: 16px !important;
|
left: 16px !important;
|
||||||
width: auto !important;
|
width: auto !important;
|
||||||
|
|||||||
Reference in New Issue
Block a user