Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
69a4b482ec | ||
|
|
ee89e04aae |
@@ -30,6 +30,7 @@ jobs:
|
|||||||
pnpm install --frozen-lockfile
|
pnpm install --frozen-lockfile
|
||||||
pnpm check
|
pnpm check
|
||||||
pnpm test
|
pnpm test
|
||||||
|
pnpm test:installer
|
||||||
- name: Build Linux release
|
- name: Build Linux release
|
||||||
run: pnpm release:build "${GITHUB_REF_NAME#v}" ./release
|
run: pnpm release:build "${GITHUB_REF_NAME#v}" ./release
|
||||||
- name: Create and publish Gitea Release
|
- name: Create and publish Gitea Release
|
||||||
|
|||||||
+1
-1
@@ -1395,7 +1395,7 @@ main() {
|
|||||||
unit_tmp=$(mktemp -d)
|
unit_tmp=$(mktemp -d)
|
||||||
sed "s#/opt/tallynote#$PREFIX#g; s#/var/lib/tallynote#$DATA_DIR#g" "$release_dir/systemd/tallynote.service" > "$unit_tmp/tallynote.service"
|
sed "s#/opt/tallynote#$PREFIX#g; s#/var/lib/tallynote#$DATA_DIR#g" "$release_dir/systemd/tallynote.service" > "$unit_tmp/tallynote.service"
|
||||||
sed "s#/opt/tallynote#$PREFIX#g; s#/var/lib/tallynote#$DATA_DIR#g; s#/var/lib/tallynote-backups#$(dirname -- "$DATA_DIR")/tallynote-backups#g" "$release_dir/systemd/tallynote-update.service" > "$unit_tmp/tallynote-update.service"
|
sed "s#/opt/tallynote#$PREFIX#g; s#/var/lib/tallynote#$DATA_DIR#g; s#/var/lib/tallynote-backups#$(dirname -- "$DATA_DIR")/tallynote-backups#g" "$release_dir/systemd/tallynote-update.service" > "$unit_tmp/tallynote-update.service"
|
||||||
sed "s#/var/lib/tallynote#$DATA_DIR#g" "$release_dir/systemd/tallynote-update.path" > "$unit_tmp/tallynote-update.path"
|
sed "s#/opt/tallynote#$PREFIX#g; s#/var/lib/tallynote#$DATA_DIR#g" "$release_dir/systemd/tallynote-update.path" > "$unit_tmp/tallynote-update.path"
|
||||||
sed "s#/opt/tallynote#$PREFIX#g; s#/etc/tallynote#$CONFIG_DIR#g" "$release_dir/bin/tallynote-admin-init" > "$unit_tmp/tallynote-admin-init"
|
sed "s#/opt/tallynote#$PREFIX#g; s#/etc/tallynote#$CONFIG_DIR#g" "$release_dir/bin/tallynote-admin-init" > "$unit_tmp/tallynote-admin-init"
|
||||||
install -o root -g root -m 644 "$unit_tmp/tallynote.service" /etc/systemd/system/tallynote.service
|
install -o root -g root -m 644 "$unit_tmp/tallynote.service" /etc/systemd/system/tallynote.service
|
||||||
install -o root -g root -m 644 "$unit_tmp/tallynote-update.service" /etc/systemd/system/tallynote-update.service
|
install -o root -g root -m 644 "$unit_tmp/tallynote-update.service" /etc/systemd/system/tallynote-update.service
|
||||||
|
|||||||
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "tallynote",
|
"name": "tallynote",
|
||||||
"version": "1.1.13",
|
"version": "1.1.15",
|
||||||
"private": true,
|
"private": true,
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"packageManager": "pnpm@9.0.6",
|
"packageManager": "pnpm@9.0.6",
|
||||||
|
|||||||
@@ -32,16 +32,120 @@ if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
|||||||
[[ "$request_operation" == download || "$request_operation" == apply ]] || request_operation='apply'
|
[[ "$request_operation" == download || "$request_operation" == apply ]] || request_operation='apply'
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Capture the request id before any privileged preflight can fail. The
|
||||||
|
# request file is an application-owned one-shot marker; removing it on an
|
||||||
|
# early runner failure lets the server-side lease reaper release the DB row.
|
||||||
|
job_id=''
|
||||||
|
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)
|
||||||
|
fi
|
||||||
|
STATE_CREATED=0
|
||||||
|
heartbeat_pid=''
|
||||||
|
heartbeat_owner=$$
|
||||||
|
|
||||||
|
write_recovery_state() {
|
||||||
|
local phase=$1 temporary
|
||||||
|
temporary="$PREFIX/.update-state-$$-${RANDOM}.tmp"
|
||||||
|
[[ ! -e "$temporary" && ! -L "$temporary" ]] || return 1
|
||||||
|
printf 'job_id=%s\nold_target=%s\nphase=%s\n' "$job_id" "$old_target" "$phase" > "$temporary"
|
||||||
|
chmod 600 "$temporary"
|
||||||
|
mv -Tf -- "$temporary" "$STATE_FILE"
|
||||||
|
STATE_CREATED=1
|
||||||
|
}
|
||||||
|
|
||||||
|
clear_recovery_state() {
|
||||||
|
[[ ! -L "$STATE_FILE" ]] || return 1
|
||||||
|
rm -f -- "$STATE_FILE"
|
||||||
|
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() {
|
||||||
|
# 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.
|
||||||
|
while kill -0 "$heartbeat_owner" 2>/dev/null; do
|
||||||
|
sleep 10 || exit 0
|
||||||
|
[[ -f "$STATE_FILE" && ! -L "$STATE_FILE" ]] || exit 0
|
||||||
|
touch "$STATE_FILE" 2>/dev/null || exit 0
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
start_heartbeat() {
|
||||||
|
stop_heartbeat
|
||||||
|
heartbeat &
|
||||||
|
heartbeat_pid=$!
|
||||||
|
}
|
||||||
|
|
||||||
|
# This trap covers failures before the normal apply cleanup trap is installed,
|
||||||
|
# including a missing runtime, an invalid current link, and a failed service
|
||||||
|
# stop. It deliberately does not remove a pre-existing recovery marker.
|
||||||
|
preflight_cleanup() {
|
||||||
|
local result=$?
|
||||||
|
stop_heartbeat
|
||||||
|
if (( result != 0 )); then
|
||||||
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
|
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
||||||
|
fi
|
||||||
|
return "$result"
|
||||||
|
}
|
||||||
|
trap preflight_cleanup EXIT
|
||||||
|
|
||||||
# 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.
|
||||||
if [[ "$request_operation" == download ]]; then
|
if [[ "$request_operation" == download ]]; then
|
||||||
|
# A previous download runner may have been interrupted after creating its
|
||||||
|
# marker. Clear only that download marker and retry the idempotent request.
|
||||||
|
if [[ -f "$STATE_FILE" && ! -L "$STATE_FILE" ]] && grep -q '^phase=download$' "$STATE_FILE"; then
|
||||||
|
clear_recovery_state || die '无法清理上一次下载状态'
|
||||||
|
fi
|
||||||
|
write_recovery_state download || die '无法写入更新恢复状态'
|
||||||
|
cleanup_download() {
|
||||||
|
local result=$?
|
||||||
|
stop_heartbeat
|
||||||
|
if (( result != 0 )); then
|
||||||
|
# The CLI normally records failed itself. If it died before opening the
|
||||||
|
# database, the expired marker/request will be reconciled by the app.
|
||||||
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
|
fi
|
||||||
|
clear_recovery_state || true
|
||||||
|
return "$result"
|
||||||
|
}
|
||||||
|
trap cleanup_download EXIT
|
||||||
|
trap 'exit 143' TERM
|
||||||
|
trap 'exit 130' INT
|
||||||
|
start_heartbeat
|
||||||
node_bin="$CURRENT_LINK/runtime/bin/node"
|
node_bin="$CURRENT_LINK/runtime/bin/node"
|
||||||
[[ -x "$node_bin" ]] || node_bin=$(command -v node || true)
|
[[ -x "$node_bin" ]] || node_bin=$(command -v node || true)
|
||||||
[[ -n "$node_bin" ]] || die 'node runtime not found'
|
[[ -n "$node_bin" ]] || die 'node runtime not found'
|
||||||
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'
|
||||||
"$node_bin" "$cli" --request-file "$REQUEST_FILE" || exit $?
|
set +e
|
||||||
|
"$node_bin" "$cli" --request-file "$REQUEST_FILE"
|
||||||
|
download_result=$?
|
||||||
|
set -e
|
||||||
|
if (( download_result != 0 )); then
|
||||||
|
# The CLI normally records failed itself. Retry the explicit finalization
|
||||||
|
# for failures that happen before its catch handler can persist the row,
|
||||||
|
# then remove the one-shot request so a failed download cannot keep the
|
||||||
|
# path unit in a permanently triggered state.
|
||||||
|
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
|
||||||
|
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
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
rm -f -- "$REQUEST_FILE"
|
||||||
|
exit "$download_result"
|
||||||
|
fi
|
||||||
rm -f -- "$REQUEST_FILE"
|
rm -f -- "$REQUEST_FILE"
|
||||||
exit 0
|
exit 0
|
||||||
fi
|
fi
|
||||||
@@ -51,34 +155,21 @@ 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
|
||||||
|
if (( result != 0 )); then
|
||||||
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
|
if (( STATE_CREATED == 1 )); then clear_recovery_state || true; fi
|
||||||
|
fi
|
||||||
if (( was_active )); then systemctl start "$SERVICE_NAME" || true; fi
|
if (( was_active )); then systemctl start "$SERVICE_NAME" || true; fi
|
||||||
return "$result"
|
return "$result"
|
||||||
}
|
}
|
||||||
trap restore_initial_service EXIT
|
trap restore_initial_service EXIT
|
||||||
systemctl stop "$SERVICE_NAME"
|
|
||||||
|
|
||||||
job_id=''
|
|
||||||
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)
|
|
||||||
fi
|
|
||||||
old_node="$CURRENT_LINK/runtime/bin/node"
|
old_node="$CURRENT_LINK/runtime/bin/node"
|
||||||
[[ -x "$old_node" ]] || old_node=$(command -v node || true)
|
[[ -x "$old_node" ]] || old_node=$(command -v node || true)
|
||||||
switched=0
|
|
||||||
handled=0
|
handled=0
|
||||||
|
|
||||||
write_update_state() {
|
write_update_state() { write_recovery_state "$1"; }
|
||||||
local phase=$1 temporary
|
clear_update_state() { clear_recovery_state; }
|
||||||
temporary="$PREFIX/.update-state-$$-${RANDOM}.tmp"
|
|
||||||
[[ ! -e "$temporary" && ! -L "$temporary" ]] || return 1
|
|
||||||
printf 'job_id=%s\nold_target=%s\nphase=%s\n' "$job_id" "$old_target" "$phase" > "$temporary"
|
|
||||||
chmod 600 "$temporary"
|
|
||||||
mv -Tf -- "$temporary" "$STATE_FILE"
|
|
||||||
}
|
|
||||||
|
|
||||||
clear_update_state() {
|
|
||||||
[[ ! -L "$STATE_FILE" ]] || return 1
|
|
||||||
rm -f -- "$STATE_FILE"
|
|
||||||
}
|
|
||||||
|
|
||||||
finalize_state_job() {
|
finalize_state_job() {
|
||||||
local node=$1 status=$2 state_job=$3
|
local node=$1 status=$2 state_job=$3
|
||||||
@@ -112,6 +203,27 @@ recover_stale_state() {
|
|||||||
done
|
done
|
||||||
return 1
|
return 1
|
||||||
fi
|
fi
|
||||||
|
if [[ "$current_target" == "$state_old" ]]; then
|
||||||
|
# The process may have restored the old release before it was killed. In
|
||||||
|
# that case the old link is already safe to serve, but the database row
|
||||||
|
# can still be `applying`; finish it as failed before clearing recovery
|
||||||
|
# markers so the UI does not poll forever.
|
||||||
|
recovery_node="$CURRENT_LINK/runtime/bin/node"
|
||||||
|
[[ -x "$recovery_node" ]] || recovery_node=$(command -v node || true)
|
||||||
|
if finalize_state_job "$recovery_node" failed "$state_job"; then
|
||||||
|
rm -f -- "$REQUEST_FILE" 2>/dev/null || true
|
||||||
|
clear_update_state || true
|
||||||
|
return 11
|
||||||
|
fi
|
||||||
|
# A crash before the CLI created its job row is safe to retry. Preserve
|
||||||
|
# the request while dropping only the stale state marker.
|
||||||
|
if [[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]]; then
|
||||||
|
clear_update_state || true
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
clear_update_state || true
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
if [[ "$current_target" != "$state_old" ]]; then
|
if [[ "$current_target" != "$state_old" ]]; then
|
||||||
rollback_link="$PREFIX/.current-recovery-$$-${RANDOM}.tmp"
|
rollback_link="$PREFIX/.current-recovery-$$-${RANDOM}.tmp"
|
||||||
[[ ! -e "$rollback_link" && ! -L "$rollback_link" ]] || return 1
|
[[ ! -e "$rollback_link" && ! -L "$rollback_link" ]] || return 1
|
||||||
@@ -156,10 +268,26 @@ fi
|
|||||||
|
|
||||||
[[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]] || exit 0
|
[[ -f "$REQUEST_FILE" && ! -L "$REQUEST_FILE" ]] || exit 0
|
||||||
|
|
||||||
|
# Only create the marker for this invocation after any marker from a previous
|
||||||
|
# interrupted run has been reconciled. Otherwise the freshly-created `running`
|
||||||
|
# marker is indistinguishable from stale recovery state and the runner can
|
||||||
|
# finalize its own queued job as failed before the update CLI starts.
|
||||||
|
if [[ ! -e "$STATE_FILE" ]]; then
|
||||||
|
write_recovery_state running || die '无法写入更新恢复状态'
|
||||||
|
fi
|
||||||
|
start_heartbeat
|
||||||
|
if ! systemctl stop "$SERVICE_NAME"; then
|
||||||
|
die '无法停止 TallyNote 服务'
|
||||||
|
fi
|
||||||
|
|
||||||
rollback_current() {
|
rollback_current() {
|
||||||
local current_target rollback_link
|
local current_target rollback_link
|
||||||
current_target=$(readlink -f -- "$CURRENT_LINK" 2>/dev/null || true)
|
current_target=$(readlink -f -- "$CURRENT_LINK" 2>/dev/null || true)
|
||||||
[[ "$current_target" == "$old_target" ]] && return 0
|
if [[ "$current_target" == "$old_target" ]]; then
|
||||||
|
# An earlier failure branch may already have restored the link. Keep the
|
||||||
|
# marker truthful so the EXIT trap can still finalize the job.
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
rollback_link="$PREFIX/.current-rollback-$$-${RANDOM}.tmp"
|
rollback_link="$PREFIX/.current-rollback-$$-${RANDOM}.tmp"
|
||||||
[[ ! -e "$rollback_link" && ! -L "$rollback_link" ]] || return 1
|
[[ ! -e "$rollback_link" && ! -L "$rollback_link" ]] || return 1
|
||||||
ln -s -- "$old_target" "$rollback_link" || return 1
|
ln -s -- "$old_target" "$rollback_link" || return 1
|
||||||
@@ -167,13 +295,20 @@ rollback_current() {
|
|||||||
rm -f -- "$rollback_link" 2>/dev/null || true
|
rm -f -- "$rollback_link" 2>/dev/null || true
|
||||||
return 1
|
return 1
|
||||||
fi
|
fi
|
||||||
switched=0
|
|
||||||
}
|
}
|
||||||
|
|
||||||
finalize_failed_job() {
|
finalize_failed_job() {
|
||||||
[[ "$job_id" =~ ^[0-9a-f-]{36}$ ]] || return 0
|
[[ "$job_id" =~ ^[0-9a-f-]{36}$ ]] || return 0
|
||||||
[[ -n "$old_node" && -f "$CURRENT_LINK/dist/server/cli/update.js" ]] || return 0
|
[[ -n "$old_node" && -f "$CURRENT_LINK/dist/server/cli/update.js" ]] || return 1
|
||||||
"$old_node" "$CURRENT_LINK/dist/server/cli/update.js" --finalize-job "$job_id" --finalize-status failed --message '新版本健康检查失败,已恢复上一版本' >/dev/null 2>&1
|
# Give SQLite a moment to release a transient lock before declaring the
|
||||||
|
# recovery itself failed.
|
||||||
|
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
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
sleep 1
|
||||||
|
done
|
||||||
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
finalize_completed_job() {
|
finalize_completed_job() {
|
||||||
@@ -185,9 +320,13 @@ finalize_completed_job() {
|
|||||||
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
# shellcheck disable=SC2329 # invoked indirectly by the EXIT trap below
|
||||||
cleanup_after_update() {
|
cleanup_after_update() {
|
||||||
local result=$? rollback_ok=1
|
local result=$? rollback_ok=1
|
||||||
|
stop_heartbeat
|
||||||
if (( result != 0 && handled == 0 )); then
|
if (( result != 0 && handled == 0 )); then
|
||||||
if ! rollback_current; then rollback_ok=0; fi
|
if ! rollback_current; then rollback_ok=0; fi
|
||||||
if (( rollback_ok == 1 && switched == 0 )); then
|
# Once the old release is active again, always try to close the job. The
|
||||||
|
# previous marker could remain set when an earlier branch had already
|
||||||
|
# rolled back before entering this EXIT trap, leaving `applying` forever.
|
||||||
|
if (( rollback_ok == 1 )); then
|
||||||
if finalize_failed_job; then
|
if finalize_failed_job; then
|
||||||
rm -f -- "$REQUEST_FILE"
|
rm -f -- "$REQUEST_FILE"
|
||||||
clear_update_state || true
|
clear_update_state || true
|
||||||
@@ -203,7 +342,6 @@ cleanup_after_update() {
|
|||||||
}
|
}
|
||||||
trap cleanup_after_update EXIT
|
trap cleanup_after_update EXIT
|
||||||
|
|
||||||
write_update_state running || exit 1
|
|
||||||
node_bin="$CURRENT_LINK/runtime/bin/node"
|
node_bin="$CURRENT_LINK/runtime/bin/node"
|
||||||
[[ -x "$node_bin" ]] || node_bin=$(command -v node || true)
|
[[ -x "$node_bin" ]] || node_bin=$(command -v node || true)
|
||||||
[[ -n "$node_bin" ]] || die 'node runtime not found'
|
[[ -n "$node_bin" ]] || die 'node runtime not found'
|
||||||
@@ -218,9 +356,6 @@ if (( update_result != 0 )); then
|
|||||||
exit "$update_result"
|
exit "$update_result"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [[ "$(readlink -f -- "$CURRENT_LINK" 2>/dev/null || true)" != "$old_target" ]]; then
|
|
||||||
switched=1
|
|
||||||
fi
|
|
||||||
write_update_state health-check || exit 1
|
write_update_state health-check || exit 1
|
||||||
|
|
||||||
systemctl start "$SERVICE_NAME"
|
systemctl start "$SERVICE_NAME"
|
||||||
|
|||||||
@@ -4,6 +4,13 @@ root=$(cd "$(dirname "$0")/.." && pwd)
|
|||||||
bash -n "$root/install.sh" "$root/scripts/tallynote-update.sh"
|
bash -n "$root/install.sh" "$root/scripts/tallynote-update.sh"
|
||||||
grep -Eq '^RestrictAddressFamilies=AF_UNIX AF_INET AF_INET6 AF_NETLINK$' "$root/systemd/tallynote.service"
|
grep -Eq '^RestrictAddressFamilies=AF_UNIX AF_INET AF_INET6 AF_NETLINK$' "$root/systemd/tallynote.service"
|
||||||
grep -Eq '^RestrictAddressFamilies=AF_UNIX AF_INET AF_INET6 AF_NETLINK$' "$root/systemd/tallynote-update.service"
|
grep -Eq '^RestrictAddressFamilies=AF_UNIX AF_INET AF_INET6 AF_NETLINK$' "$root/systemd/tallynote-update.service"
|
||||||
|
grep -Eq '^PathExists=/opt/tallynote/\.update-state$' "$root/systemd/tallynote-update.path"
|
||||||
|
grep -Eq '^PathChanged=/opt/tallynote/\.update-state$' "$root/systemd/tallynote-update.path"
|
||||||
|
grep -Eq '^PathChanged=/opt/tallynote$' "$root/systemd/tallynote-update.path"
|
||||||
|
if grep -Eq '^ConditionPathExists=' "$root/systemd/tallynote-update.service"; then
|
||||||
|
echo 'update service must not require only the request file' >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
output=$(bash "$root/install.sh" --dry-run --release-base-url https://releases.example.test/releases)
|
output=$(bash "$root/install.sh" --dry-run --release-base-url https://releases.example.test/releases)
|
||||||
grep -q 'dry-run' <<<"$output"
|
grep -q 'dry-run' <<<"$output"
|
||||||
grep -q '\[阶段\] 检查运行环境' <<<"$output"
|
grep -q '\[阶段\] 检查运行环境' <<<"$output"
|
||||||
@@ -153,6 +160,35 @@ bash -c '
|
|||||||
wait_for_service_health 0.0.0.0 3011
|
wait_for_service_health 0.0.0.0 3011
|
||||||
' _ "$installer_lib"
|
' _ "$installer_lib"
|
||||||
|
|
||||||
|
# Exercise the privileged runner's normal apply hand-off with portable command
|
||||||
|
# shims. In particular, the freshly-created running marker must not be treated
|
||||||
|
# as stale state before the update CLI gets a chance to process the request.
|
||||||
|
runner_root="$tmp/runner"
|
||||||
|
runner_prefix="$runner_root/prefix"
|
||||||
|
runner_data="$runner_root/data"
|
||||||
|
runner_tools="$runner_root/tools"
|
||||||
|
mkdir -p "$runner_prefix/releases/1.0.0/runtime/bin" "$runner_prefix/releases/1.0.0/dist/server/cli" "$runner_data" "$runner_tools"
|
||||||
|
ln -s "$runner_prefix/releases/1.0.0" "$runner_prefix/current"
|
||||||
|
printf '%s\n' '{"jobId":"00000000-0000-4000-8000-000000000001","operation":"apply"}' > "$runner_data/update-request.json"
|
||||||
|
printf '%s\n' '#!/usr/bin/env bash' 'printf "%s\\n" "$*" >> "$TALLYNOTE_NODE_TRACE"' 'exit 0' > "$runner_prefix/releases/1.0.0/runtime/bin/node"
|
||||||
|
printf '%s\n' cli > "$runner_prefix/releases/1.0.0/dist/server/cli/update.js"
|
||||||
|
printf '%s\n' '#!/usr/bin/env bash' 'case "${1:-}" in is-active) exit 0;; *) exit 0;; esac' > "$runner_tools/systemctl"
|
||||||
|
printf '%s\n' '#!/usr/bin/env bash' 'if [[ "${1:-}" == "-f" ]]; then shift; [[ "${1:-}" == "--" ]] && shift; /bin/realpath "$1"; else /usr/bin/readlink "$@"; fi' > "$runner_tools/readlink"
|
||||||
|
printf '%s\n' '#!/usr/bin/env bash' 'if [[ "${1:-}" == "-Tf" ]]; then shift; /bin/mv -f "$@"; else /bin/mv "$@"; fi' > "$runner_tools/mv"
|
||||||
|
printf '%s\n' '#!/usr/bin/env bash' 'exit 0' > "$runner_tools/curl"
|
||||||
|
chmod 755 "$runner_prefix/releases/1.0.0/runtime/bin/node" "$runner_tools/systemctl" "$runner_tools/readlink" "$runner_tools/mv" "$runner_tools/curl"
|
||||||
|
runner_script="$runner_root/runner.sh"
|
||||||
|
runner_path="$runner_tools:/usr/sbin:/usr/bin:/sbin:/bin"
|
||||||
|
sed "s#PATH=/usr/sbin:/usr/bin:/sbin:/bin#PATH=$runner_path#" "$root/scripts/tallynote-update-runner.sh" > "$runner_script"
|
||||||
|
chmod 755 "$runner_script"
|
||||||
|
runner_prefix_physical=$(cd "$runner_prefix" && pwd -P)
|
||||||
|
runner_data_physical=$(cd "$runner_data" && pwd -P)
|
||||||
|
env EUID=0 TALLYNOTE_INSTALL_PREFIX="$runner_prefix_physical" TALLYNOTE_DATA_DIR="$runner_data_physical" TALLYNOTE_NODE_TRACE="$runner_root/node.log" bash "$runner_script"
|
||||||
|
grep -q -- '--request-file' "$runner_root/node.log"
|
||||||
|
grep -q -- '--finalize-job' "$runner_root/node.log"
|
||||||
|
[[ ! -e "$runner_data/update-request.json" ]]
|
||||||
|
[[ ! -e "$runner_prefix/.update-state" ]]
|
||||||
|
|
||||||
# A RETURN trap installed by install_release must be cleared while its local
|
# A RETURN trap installed by install_release must be cleared while its local
|
||||||
# temporary variables still exist; otherwise set -u fails at the end of main.
|
# temporary variables still exist; otherwise set -u fails at the end of main.
|
||||||
release_fixture="$tmp/release-fixture"
|
release_fixture="$tmp/release-fixture"
|
||||||
@@ -192,6 +228,17 @@ bash -c '
|
|||||||
set_env_key test value
|
set_env_key test value
|
||||||
' _ "$installer_lib" "$release_archive" "$tmp/install-release"
|
' _ "$installer_lib" "$release_archive" "$tmp/install-release"
|
||||||
|
|
||||||
|
# The installed path unit must watch both the data-directory request and the
|
||||||
|
# release-prefix recovery marker after custom paths are substituted.
|
||||||
|
rendered_path="$tmp/rendered-update.path"
|
||||||
|
sed "s#/opt/tallynote#$tmp/custom-prefix#g; s#/var/lib/tallynote#$tmp/custom-data#g" \
|
||||||
|
"$root/systemd/tallynote-update.path" > "$rendered_path"
|
||||||
|
grep -Fxq "PathExists=$tmp/custom-data/update-request.json" "$rendered_path"
|
||||||
|
grep -Fxq "PathChanged=$tmp/custom-data/update-request.json" "$rendered_path"
|
||||||
|
grep -Fxq "PathExists=$tmp/custom-prefix/.update-state" "$rendered_path"
|
||||||
|
grep -Fxq "PathChanged=$tmp/custom-prefix/.update-state" "$rendered_path"
|
||||||
|
grep -Fxq "PathChanged=$tmp/custom-prefix" "$rendered_path"
|
||||||
|
|
||||||
# The production admin wrapper must load a release-relative runtime, change to
|
# The production admin wrapper must load a release-relative runtime, change to
|
||||||
# the release root, and forward CLI arguments without requiring pnpm.
|
# the release root, and forward CLI arguments without requiring pnpm.
|
||||||
wrapper_prefix="$tmp/wrapper-prefix"
|
wrapper_prefix="$tmp/wrapper-prefix"
|
||||||
|
|||||||
@@ -59,6 +59,7 @@ import {
|
|||||||
checkForUpdate,
|
checkForUpdate,
|
||||||
publicCheckFromCache,
|
publicCheckFromCache,
|
||||||
publicUpdateJob,
|
publicUpdateJob,
|
||||||
|
reconcileOrphanedUpdateJobs,
|
||||||
readCachedRelease,
|
readCachedRelease,
|
||||||
writeUpdateRequest,
|
writeUpdateRequest,
|
||||||
type UpdateRequest,
|
type UpdateRequest,
|
||||||
@@ -939,6 +940,7 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
// Release metadata and task state should never be stored by an upstream
|
// Release metadata and task state should never be stored by an upstream
|
||||||
// proxy or a shared browser cache.
|
// proxy or a shared browser cache.
|
||||||
reply.header("Cache-Control", "no-store");
|
reply.header("Cache-Control", "no-store");
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
const cached = publicCheckFromCache(database.sqlite, config);
|
const cached = publicCheckFromCache(database.sqlite, config);
|
||||||
const row = database.sqlite.prepare(`
|
const row = database.sqlite.prepare(`
|
||||||
SELECT id, operation, status, version, platform, asset_name AS assetName,
|
SELECT id, operation, status, version, platform, asset_name AS assetName,
|
||||||
@@ -956,6 +958,7 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
|
|
||||||
app.post("/api/update/check", { preHandler: guard(database, config), bodyLimit: 32 * 1024 }, async (request, reply) => {
|
app.post("/api/update/check", { preHandler: guard(database, config), bodyLimit: 32 * 1024 }, async (request, reply) => {
|
||||||
try {
|
try {
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
// Disabled/dev installs do not contact a release endpoint, so repeated
|
// Disabled/dev installs do not contact a release endpoint, so repeated
|
||||||
// checks are local status reads and should remain immediately usable.
|
// checks are local status reads and should remain immediately usable.
|
||||||
if (config.updateStrategy !== "disabled") {
|
if (config.updateStrategy !== "disabled") {
|
||||||
@@ -995,6 +998,7 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
let applyAuditRecorded = false;
|
let applyAuditRecorded = false;
|
||||||
let applyAuditTarget: string | undefined;
|
let applyAuditTarget: string | undefined;
|
||||||
try {
|
try {
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
if (config.updateStrategy !== "systemd") {
|
if (config.updateStrategy !== "systemd") {
|
||||||
throw new AppError(503, "UPDATE_NOT_AVAILABLE", "当前安装方式未启用一键更新,请使用命令行更新");
|
throw new AppError(503, "UPDATE_NOT_AVAILABLE", "当前安装方式未启用一键更新,请使用命令行更新");
|
||||||
}
|
}
|
||||||
@@ -1145,6 +1149,7 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
|
|
||||||
app.post("/api/update/download", { preHandler: guard(database, config), bodyLimit: 32 * 1024 }, async (request, reply) => {
|
app.post("/api/update/download", { preHandler: guard(database, config), bodyLimit: 32 * 1024 }, async (request, reply) => {
|
||||||
const input = updateDownloadSchema.parse(request.body);
|
const input = updateDownloadSchema.parse(request.body);
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
if (config.updateStrategy !== "systemd") throw new AppError(503, "UPDATE_NOT_AVAILABLE", "当前安装方式未启用一键更新,请使用命令行更新");
|
if (config.updateStrategy !== "systemd") throw new AppError(503, "UPDATE_NOT_AVAILABLE", "当前安装方式未启用一键更新,请使用命令行更新");
|
||||||
const active = database.sqlite.prepare(`SELECT id FROM update_jobs WHERE status IN (${ACTIVE_UPDATE_STATUSES.map(() => "?").join(",")}) LIMIT 1`).get(...ACTIVE_UPDATE_STATUSES) as { id: string } | undefined;
|
const active = database.sqlite.prepare(`SELECT id FROM update_jobs WHERE status IN (${ACTIVE_UPDATE_STATUSES.map(() => "?").join(",")}) LIMIT 1`).get(...ACTIVE_UPDATE_STATUSES) as { id: string } | undefined;
|
||||||
if (active) throw new AppError(409, "UPDATE_IN_PROGRESS", "已有更新任务正在进行,请等待完成");
|
if (active) throw new AppError(409, "UPDATE_IN_PROGRESS", "已有更新任务正在进行,请等待完成");
|
||||||
@@ -1175,6 +1180,7 @@ export async function buildApp(database: DatabaseContext, config: AppConfig) {
|
|||||||
|
|
||||||
app.get("/api/update/jobs/:id", { preHandler: guard(database, config) }, async (request, reply) => {
|
app.get("/api/update/jobs/:id", { preHandler: guard(database, config) }, async (request, reply) => {
|
||||||
const id = z.string().uuid().parse((request.params as { id: string }).id);
|
const id = z.string().uuid().parse((request.params as { id: string }).id);
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
const row = database.sqlite.prepare(`
|
const row = database.sqlite.prepare(`
|
||||||
SELECT id, operation, status, version, platform, asset_name AS assetName,
|
SELECT id, operation, status, version, platform, asset_name AS assetName,
|
||||||
size_bytes AS sizeBytes, error_message AS errorMessage,
|
size_bytes AS sizeBytes, error_message AS errorMessage,
|
||||||
|
|||||||
+33
-24
@@ -26,7 +26,7 @@ import {
|
|||||||
type ReleaseMetadata,
|
type ReleaseMetadata,
|
||||||
type UrlPolicy,
|
type UrlPolicy,
|
||||||
} from "../update.js";
|
} from "../update.js";
|
||||||
import { attachSidecarHash } from "../update-service.js";
|
import { ACTIVE_UPDATE_STATUSES, attachSidecarHash } from "../update-service.js";
|
||||||
import type { UpdateJobStatus } from "../../shared/contracts.js";
|
import type { UpdateJobStatus } from "../../shared/contracts.js";
|
||||||
|
|
||||||
const updateRequestFileSchema = z.object({
|
const updateRequestFileSchema = z.object({
|
||||||
@@ -153,7 +153,7 @@ function writeJob(sqlite: Database.Database | undefined, jobId: string, values:
|
|||||||
operation, status, version, platform, release_url, asset_name, asset_url,
|
operation, status, version, platform, release_url, asset_name, asset_url,
|
||||||
expected_sha256, actual_sha256, download_path, backup_path, size_bytes, error_message,
|
expected_sha256, actual_sha256, download_path, backup_path, size_bytes, error_message,
|
||||||
created_at, updated_at, completed_at)
|
created_at, updated_at, completed_at)
|
||||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
ON CONFLICT(id) DO UPDATE SET
|
ON CONFLICT(id) DO UPDATE SET
|
||||||
admin_id=COALESCE(excluded.admin_id, update_jobs.admin_id),
|
admin_id=COALESCE(excluded.admin_id, update_jobs.admin_id),
|
||||||
session_hash=COALESCE(excluded.session_hash, update_jobs.session_hash),
|
session_hash=COALESCE(excluded.session_hash, update_jobs.session_hash),
|
||||||
@@ -355,7 +355,9 @@ export function finalizeUpdateJob(
|
|||||||
FROM update_jobs WHERE id=?
|
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;
|
`).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("更新任务不存在");
|
if (!row) throw new Error("更新任务不存在");
|
||||||
if (row.status !== "applying" && row.status !== "completed" && row.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 now = Date.now();
|
||||||
const safeFailureMessage = status === "failed" ? "新版本健康检查失败,已恢复上一版本" : null;
|
const safeFailureMessage = status === "failed" ? "新版本健康检查失败,已恢复上一版本" : null;
|
||||||
sqlite.transaction(() => {
|
sqlite.transaction(() => {
|
||||||
@@ -477,27 +479,34 @@ export async function main(config: AppConfig = loadConfig()): Promise<void> {
|
|||||||
const database = openDatabase(config);
|
const database = openDatabase(config);
|
||||||
try {
|
try {
|
||||||
if (request?.operation === "apply") {
|
if (request?.operation === "apply") {
|
||||||
const staged = database.sqlite.prepare("SELECT download_path AS downloadPath, version FROM update_jobs WHERE id=? AND status='staged' AND operation='apply'").get(request.jobId) as { downloadPath: string | null; version: string } | undefined;
|
const staged = database.sqlite.prepare("SELECT status, operation, download_path AS downloadPath, version FROM update_jobs WHERE id=?").get(request.jobId) as { status: UpdateJobStatus; operation: "download" | "apply"; downloadPath: string | null; version: string } | undefined;
|
||||||
if (!staged?.downloadPath || staged.version !== request.version) throw new Error("更新暂存任务无效");
|
if (staged?.status === "staged" && staged.operation === "apply") {
|
||||||
const root = path.resolve(config.updateWorkspaceDir);
|
if (!staged.downloadPath || staged.version !== request.version) throw new Error("更新暂存任务无效");
|
||||||
const candidate = await validateStagedWorkspacePath(staged.downloadPath, root);
|
const root = path.resolve(config.updateWorkspaceDir);
|
||||||
await applyStagedUpdate({
|
const candidate = await validateStagedWorkspacePath(staged.downloadPath, root);
|
||||||
sqlite: database.sqlite,
|
await applyStagedUpdate({
|
||||||
jobId: request.jobId,
|
sqlite: database.sqlite,
|
||||||
version: request.version,
|
jobId: request.jobId,
|
||||||
stagedPath: candidate,
|
version: request.version,
|
||||||
currentDir,
|
stagedPath: candidate,
|
||||||
currentLink: request.currentLink,
|
currentDir,
|
||||||
releasesDir: request.releasesDir,
|
currentLink: request.currentLink,
|
||||||
workspaceRoot: root,
|
releasesDir: request.releasesDir,
|
||||||
...(backupArchive ? { backupArchivePath: backupArchive } : {}),
|
workspaceRoot: root,
|
||||||
...(dataBackupArchive ? { dataBackupArchivePath: dataBackupArchive } : {}),
|
...(backupArchive ? { backupArchivePath: backupArchive } : {}),
|
||||||
dataBackupSource: config.dataDir,
|
...(dataBackupArchive ? { dataBackupArchivePath: dataBackupArchive } : {}),
|
||||||
maxBytes: config.updateMaxBytes,
|
dataBackupSource: config.dataDir,
|
||||||
dataBackupMaxBytes: config.maxTotalBytes,
|
maxBytes: config.updateMaxBytes,
|
||||||
});
|
dataBackupMaxBytes: config.maxTotalBytes,
|
||||||
console.log(`更新已切换:${request.version}`);
|
});
|
||||||
return;
|
console.log(`更新已切换:${request.version}`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (staged && !(staged.status === "queued" && staged.operation === "apply")) throw new Error("更新任务状态无效");
|
||||||
|
// A direct one-click request starts in queued/apply. Older clients do
|
||||||
|
// not have a separate download step, so fall through to runUpdate,
|
||||||
|
// which downloads, verifies, backs up, and switches the release in one
|
||||||
|
// transaction. A staged request still takes the branch above.
|
||||||
}
|
}
|
||||||
const result = await runUpdate({
|
const result = await runUpdate({
|
||||||
...(effectiveMetadataUrl ? { metadataUrl: effectiveMetadataUrl } : {}),
|
...(effectiveMetadataUrl ? { metadataUrl: effectiveMetadataUrl } : {}),
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import { loadConfig, prepareDataDirectories, acquireInstanceLock } from "./confi
|
|||||||
import { openDatabase } from "./db/index.js";
|
import { openDatabase } from "./db/index.js";
|
||||||
import { buildApp } from "./app.js";
|
import { buildApp } from "./app.js";
|
||||||
import { cleanupOrphanedExports, expireExports, resumeExports } from "./exporter.js";
|
import { cleanupOrphanedExports, expireExports, resumeExports } from "./exporter.js";
|
||||||
|
import { reconcileOrphanedUpdateJobs } from "./update-service.js";
|
||||||
|
|
||||||
const config = loadConfig();
|
const config = loadConfig();
|
||||||
prepareDataDirectories(config);
|
prepareDataDirectories(config);
|
||||||
@@ -18,6 +19,7 @@ async function start() {
|
|||||||
await expireExports(database.sqlite, config);
|
await expireExports(database.sqlite, config);
|
||||||
await cleanupOrphanedExports(database.sqlite, config);
|
await cleanupOrphanedExports(database.sqlite, config);
|
||||||
await resumeExports(database.sqlite, config);
|
await resumeExports(database.sqlite, config);
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
const app = await buildApp(database, config);
|
const app = await buildApp(database, config);
|
||||||
const janitor = setInterval(() => {
|
const janitor = setInterval(() => {
|
||||||
void cleanupStaging(config);
|
void cleanupStaging(config);
|
||||||
@@ -27,6 +29,7 @@ async function start() {
|
|||||||
void processFileDeletions(database.sqlite, config);
|
void processFileDeletions(database.sqlite, config);
|
||||||
void expireExports(database.sqlite, config);
|
void expireExports(database.sqlite, config);
|
||||||
void cleanupOrphanedExports(database.sqlite, config);
|
void cleanupOrphanedExports(database.sqlite, config);
|
||||||
|
reconcileOrphanedUpdateJobs(database.sqlite, config);
|
||||||
}, 60_000);
|
}, 60_000);
|
||||||
const shutdown = async () => {
|
const shutdown = async () => {
|
||||||
clearInterval(janitor);
|
clearInterval(janitor);
|
||||||
|
|||||||
@@ -1,7 +1,9 @@
|
|||||||
|
import { lstatSync, realpathSync, unlinkSync } from "node:fs";
|
||||||
import { chmod, mkdir, rename, writeFile } from "node:fs/promises";
|
import { chmod, mkdir, rename, writeFile } from "node:fs/promises";
|
||||||
import path from "node:path";
|
import path from "node:path";
|
||||||
import { createPublicKey, randomUUID, verify as verifySignature } from "node:crypto";
|
import { createPublicKey, randomUUID, verify as verifySignature } from "node:crypto";
|
||||||
import type Database from "better-sqlite3";
|
import type Database from "better-sqlite3";
|
||||||
|
import { writeAudit } from "./audit.js";
|
||||||
import { AppError } from "./errors.js";
|
import { AppError } from "./errors.js";
|
||||||
import type { AppConfig } from "./config.js";
|
import type { AppConfig } from "./config.js";
|
||||||
import {
|
import {
|
||||||
@@ -30,6 +32,12 @@ export const ACTIVE_UPDATE_STATUSES: readonly UpdateJobStatus[] = [
|
|||||||
"applying",
|
"applying",
|
||||||
];
|
];
|
||||||
|
|
||||||
|
// A queued job normally starts within seconds and an applying job completes
|
||||||
|
// after the service health check. The runner refreshes its recovery marker as
|
||||||
|
// a lease while doing long downloads/backups; only an expired lease permits
|
||||||
|
// the server to reclaim an active row.
|
||||||
|
export const ORPHANED_UPDATE_TIMEOUT_MS = 5 * 60 * 1000;
|
||||||
|
|
||||||
export type CachedRelease = {
|
export type CachedRelease = {
|
||||||
checkedAt: number;
|
checkedAt: number;
|
||||||
metadataUrl: string;
|
metadataUrl: string;
|
||||||
@@ -355,3 +363,129 @@ export function publicUpdateJob(row: Record<string, unknown> | undefined): Recor
|
|||||||
...(row.status === "applying" ? { restartWindowSeconds: 30 } : {}),
|
...(row.status === "applying" ? { restartWindowSeconds: 30 } : {}),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function markerMtime(filePath: string): number | null {
|
||||||
|
try {
|
||||||
|
const info = lstatSync(filePath);
|
||||||
|
return info.isFile() ? info.mtimeMs : null;
|
||||||
|
} catch {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function removeExpiredRequest(filePath: string, now: number): void {
|
||||||
|
try {
|
||||||
|
const info = lstatSync(filePath);
|
||||||
|
if (!info.isFile() && !info.isSymbolicLink()) return;
|
||||||
|
if (now - info.mtimeMs < ORPHANED_UPDATE_TIMEOUT_MS) return;
|
||||||
|
unlinkSync(filePath);
|
||||||
|
} catch {
|
||||||
|
// The root runner may own the marker during a recovery race. The DB
|
||||||
|
// transition below is still enough to release the browser queue.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function currentReleaseVersion(config: AppConfig): string | null {
|
||||||
|
try {
|
||||||
|
const target = realpathSync(config.currentLink);
|
||||||
|
const releases = realpathSync(config.releasesDir);
|
||||||
|
if (!target.startsWith(`${releases}${path.sep}`)) return null;
|
||||||
|
return path.basename(target);
|
||||||
|
} catch {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Release an update row left behind after its privileged runner lease expired.
|
||||||
|
* This is deliberately conservative: staged downloads remain available for an
|
||||||
|
* explicit apply, and a fresh request/state marker means the runner still owns
|
||||||
|
* recovery.
|
||||||
|
*/
|
||||||
|
export function reconcileOrphanedUpdateJobs(database: Database.Database, config: AppConfig, now = Date.now()): number {
|
||||||
|
const placeholders = ACTIVE_UPDATE_STATUSES.map(() => "?").join(",");
|
||||||
|
const rows = database.prepare(`
|
||||||
|
SELECT id, status, operation, version, admin_id AS adminId, request_id AS requestId,
|
||||||
|
updated_at AS updatedAt
|
||||||
|
FROM update_jobs
|
||||||
|
WHERE status IN (${placeholders})
|
||||||
|
ORDER BY updated_at ASC
|
||||||
|
`).all(...ACTIVE_UPDATE_STATUSES) as Array<{ id: string; status: UpdateJobStatus; operation: "download" | "apply"; version: string; adminId: string | null; requestId: string | null; updatedAt: number | null }>;
|
||||||
|
if (rows.length === 0) return 0;
|
||||||
|
const statePath = path.join(config.installPrefix, ".update-state");
|
||||||
|
const requestMtime = markerMtime(config.updateRequestPath);
|
||||||
|
const stateMtime = markerMtime(statePath);
|
||||||
|
const requestPresent = requestMtime !== null;
|
||||||
|
const statePresent = stateMtime !== null;
|
||||||
|
const requestFresh = requestPresent && now - (requestMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
||||||
|
const stateFresh = statePresent && now - (stateMtime ?? 0) < ORPHANED_UPDATE_TIMEOUT_MS;
|
||||||
|
// 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
|
||||||
|
// operation to `apply` but crashed before writing the request file. That
|
||||||
|
// row is still safe to retry and must not block the queue forever.
|
||||||
|
const releaseVersion = currentReleaseVersion(config);
|
||||||
|
let reconciled = 0;
|
||||||
|
for (const row of rows) {
|
||||||
|
if (typeof row.updatedAt !== "number" || now - row.updatedAt < ORPHANED_UPDATE_TIMEOUT_MS) continue;
|
||||||
|
// The runner refreshes the state marker while a download is in flight.
|
||||||
|
// A stale request/state marker therefore no longer protects an orphaned
|
||||||
|
// row forever, while a fresh marker remains owned by the runner.
|
||||||
|
if (row.status === "staged") {
|
||||||
|
if (row.operation !== "apply" || requestFresh || stateFresh) continue;
|
||||||
|
const changed = database.transaction(() => {
|
||||||
|
const result = database.prepare(`
|
||||||
|
UPDATE update_jobs
|
||||||
|
SET operation='download', error_message=NULL, updated_at=?
|
||||||
|
WHERE id=? AND status='staged' AND operation='apply' AND updated_at=?
|
||||||
|
`).run(now, row.id, row.updatedAt);
|
||||||
|
if (result.changes !== 1) return false;
|
||||||
|
writeAudit(database, {
|
||||||
|
requestId: row.requestId || randomUUID(),
|
||||||
|
actorAdminId: row.adminId,
|
||||||
|
action: "update.reconciled",
|
||||||
|
targetType: "update",
|
||||||
|
targetId: row.id,
|
||||||
|
outcome: "success",
|
||||||
|
before: { status: row.status, operation: row.operation, version: row.version },
|
||||||
|
after: { status: "staged", operation: "download", version: row.version, reason: "apply_request_missing" },
|
||||||
|
});
|
||||||
|
return true;
|
||||||
|
})();
|
||||||
|
if (changed) reconciled += 1;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (requestFresh || stateFresh) continue;
|
||||||
|
const status: "completed" | "failed" = row.status === "applying" && releaseVersion === row.version ? "completed" : "failed";
|
||||||
|
const errorMessage = status === "failed" ? "更新任务超时,已释放更新队列" : null;
|
||||||
|
const changed = database.transaction(() => {
|
||||||
|
const result = database.prepare(`
|
||||||
|
UPDATE update_jobs
|
||||||
|
SET status=?, error_message=?, completed_at=?, updated_at=?
|
||||||
|
WHERE id=? AND status=? AND updated_at=?
|
||||||
|
`).run(status, errorMessage, now, now, row.id, row.status, row.updatedAt);
|
||||||
|
if (result.changes !== 1) return false;
|
||||||
|
writeAudit(database, {
|
||||||
|
requestId: row.requestId || randomUUID(),
|
||||||
|
actorAdminId: row.adminId,
|
||||||
|
action: "update.reconciled",
|
||||||
|
targetType: "update",
|
||||||
|
targetId: row.id,
|
||||||
|
outcome: status === "completed" ? "success" : "failure",
|
||||||
|
before: { status: row.status, version: row.version },
|
||||||
|
after: { status, version: row.version, reason: "orphaned_timeout" },
|
||||||
|
});
|
||||||
|
return true;
|
||||||
|
})();
|
||||||
|
if (changed) reconciled += 1;
|
||||||
|
}
|
||||||
|
// Prevent a stale request from being replayed after its DB row has been
|
||||||
|
// marked failed. The path is fixed by the server configuration and the
|
||||||
|
// operation is safe even when a root runner is racing with this call.
|
||||||
|
// A download runner refreshes the state marker while it is still using the
|
||||||
|
// request. Keep the request until that lease also expires; otherwise a
|
||||||
|
// long download can lose its job id and fail to finalize its row.
|
||||||
|
if (!stateFresh && (!requestPresent || (requestMtime !== null && now - requestMtime >= ORPHANED_UPDATE_TIMEOUT_MS))) {
|
||||||
|
removeExpiredRequest(config.updateRequestPath, now);
|
||||||
|
}
|
||||||
|
return reconciled;
|
||||||
|
}
|
||||||
|
|||||||
@@ -4,6 +4,15 @@ Description=Watch for TallyNote release update requests
|
|||||||
[Path]
|
[Path]
|
||||||
PathExists=/var/lib/tallynote/update-request.json
|
PathExists=/var/lib/tallynote/update-request.json
|
||||||
PathChanged=/var/lib/tallynote/update-request.json
|
PathChanged=/var/lib/tallynote/update-request.json
|
||||||
|
# The recovery marker lives beside the release link. Watching it as well
|
||||||
|
# allows systemd to resume reconciliation when the runner is interrupted
|
||||||
|
# after consuming the request but before clearing its state file.
|
||||||
|
PathExists=/opt/tallynote/.update-state
|
||||||
|
PathChanged=/opt/tallynote/.update-state
|
||||||
|
# Keep a directory-level fallback because some systemd/inotify versions skip
|
||||||
|
# dotfiles when watching an individual path. State writes are atomic renames,
|
||||||
|
# so the containing directory changes even when the marker itself is hidden.
|
||||||
|
PathChanged=/opt/tallynote
|
||||||
Unit=tallynote-update.service
|
Unit=tallynote-update.service
|
||||||
|
|
||||||
[Install]
|
[Install]
|
||||||
|
|||||||
@@ -2,7 +2,6 @@
|
|||||||
Description=TallyNote privileged release updater
|
Description=TallyNote privileged release updater
|
||||||
After=network-online.target
|
After=network-online.target
|
||||||
Wants=network-online.target
|
Wants=network-online.target
|
||||||
ConditionPathExists=/var/lib/tallynote/update-request.json
|
|
||||||
|
|
||||||
[Service]
|
[Service]
|
||||||
Type=oneshot
|
Type=oneshot
|
||||||
@@ -12,6 +11,10 @@ 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
|
||||||
|
# Downloads, archive validation and data backups can exceed systemd's 90s
|
||||||
|
# default start timeout on a slower server. Keep one update job alive long
|
||||||
|
# enough to finish or reach its own health-check/recovery path.
|
||||||
|
TimeoutStartSec=30min
|
||||||
NoNewPrivileges=true
|
NoNewPrivileges=true
|
||||||
CapabilityBoundingSet=
|
CapabilityBoundingSet=
|
||||||
AmbientCapabilities=
|
AmbientCapabilities=
|
||||||
|
|||||||
@@ -14,6 +14,8 @@ Environment=PATH=/opt/tallynote/current/runtime/bin:/usr/sbin:/usr/bin:/sbin:/bi
|
|||||||
ExecStart=/opt/tallynote/current/bin/tallynote
|
ExecStart=/opt/tallynote/current/bin/tallynote
|
||||||
Restart=on-failure
|
Restart=on-failure
|
||||||
RestartSec=5s
|
RestartSec=5s
|
||||||
|
# Do not let a wedged Node process hold an update stop forever.
|
||||||
|
TimeoutStopSec=30s
|
||||||
NoNewPrivileges=true
|
NoNewPrivileges=true
|
||||||
PrivateTmp=true
|
PrivateTmp=true
|
||||||
ProtectSystem=strict
|
ProtectSystem=strict
|
||||||
|
|||||||
+170
-5
@@ -1,9 +1,9 @@
|
|||||||
import { afterEach, describe, expect, it } from "vitest";
|
import { afterEach, describe, expect, it } from "vitest";
|
||||||
import { mkdir, readlink, symlink, writeFile, readFile, stat, readdir } from "node:fs/promises";
|
import { mkdir, readlink, symlink, writeFile, readFile, stat, readdir, utimes } from "node:fs/promises";
|
||||||
import { mkdtemp, rm } from "node:fs/promises";
|
import { mkdtemp, rm } from "node:fs/promises";
|
||||||
import { tmpdir } from "node:os";
|
import { tmpdir } from "node:os";
|
||||||
import path from "node:path";
|
import path from "node:path";
|
||||||
import { createHash, generateKeyPairSync, sign } from "node:crypto";
|
import { createHash, generateKeyPairSync, randomUUID, sign } from "node:crypto";
|
||||||
import {
|
import {
|
||||||
atomicSwitchRelease,
|
atomicSwitchRelease,
|
||||||
createSafeArchive,
|
createSafeArchive,
|
||||||
@@ -18,13 +18,13 @@ import {
|
|||||||
selectReleaseAsset,
|
selectReleaseAsset,
|
||||||
validateHttpsUrl,
|
validateHttpsUrl,
|
||||||
} from "../server/update.js";
|
} from "../server/update.js";
|
||||||
import { runUpdate } from "../server/cli/update.js";
|
import { finalizeUpdateJob, runUpdate } from "../server/cli/update.js";
|
||||||
import { validateUpdateRequest } from "../server/cli/update.js";
|
import { validateUpdateRequest } from "../server/cli/update.js";
|
||||||
import { checkForUpdate, verifyReleaseSignature } from "../server/update-service.js";
|
import { checkForUpdate, ORPHANED_UPDATE_TIMEOUT_MS, reconcileOrphanedUpdateJobs, verifyReleaseSignature } from "../server/update-service.js";
|
||||||
import { loadConfig, prepareDataDirectories } from "../server/config.js";
|
import { loadConfig, prepareDataDirectories } from "../server/config.js";
|
||||||
import { openDatabase } from "../server/db/index.js";
|
import { openDatabase } from "../server/db/index.js";
|
||||||
|
|
||||||
const envKeys = ["TALLYNOTE_DATA_DIR", "TALLYNOTE_PUBLIC_ORIGIN", "TALLYNOTE_COOKIE_SECURE", "TALLYNOTE_UPDATE_STRATEGY", "TALLYNOTE_UPDATE_METADATA_URL", "TALLYNOTE_UPDATE_ALLOWED_HOSTS", "TALLYNOTE_UPDATE_REQUIRE_SIGNATURE", "TALLYNOTE_UPDATE_PUBLIC_KEY"];
|
const envKeys = ["TALLYNOTE_DATA_DIR", "TALLYNOTE_INSTALL_PREFIX", "TALLYNOTE_PUBLIC_ORIGIN", "TALLYNOTE_COOKIE_SECURE", "TALLYNOTE_UPDATE_STRATEGY", "TALLYNOTE_UPDATE_METADATA_URL", "TALLYNOTE_UPDATE_ALLOWED_HOSTS", "TALLYNOTE_UPDATE_REQUIRE_SIGNATURE", "TALLYNOTE_UPDATE_PUBLIC_KEY"];
|
||||||
const originalFetch = globalThis.fetch;
|
const originalFetch = globalThis.fetch;
|
||||||
|
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
@@ -207,6 +207,171 @@ describe("更新安全工具", () => {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it("更新器支持旧客户端创建的 queued/apply 直接更新请求", async () => {
|
||||||
|
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-direct-"));
|
||||||
|
const previousFetch = globalThis.fetch;
|
||||||
|
let database: ReturnType<typeof openDatabase> | undefined;
|
||||||
|
try {
|
||||||
|
const dataDir = path.join(root, "data");
|
||||||
|
const installPrefix = path.join(root, "install");
|
||||||
|
process.env.TALLYNOTE_DATA_DIR = dataDir;
|
||||||
|
process.env.TALLYNOTE_INSTALL_PREFIX = installPrefix;
|
||||||
|
process.env.TALLYNOTE_PUBLIC_ORIGIN = "http://127.0.0.1:3998";
|
||||||
|
process.env.TALLYNOTE_COOKIE_SECURE = "false";
|
||||||
|
process.env.TALLYNOTE_UPDATE_STRATEGY = "systemd";
|
||||||
|
process.env.TALLYNOTE_UPDATE_METADATA_URL = "https://updates.example/latest";
|
||||||
|
process.env.TALLYNOTE_UPDATE_ALLOWED_HOSTS = "updates.example";
|
||||||
|
process.env.TALLYNOTE_UPDATE_REQUIRE_SIGNATURE = "false";
|
||||||
|
const config = loadConfig();
|
||||||
|
prepareDataDirectories(config);
|
||||||
|
await mkdir(config.releasesDir, { recursive: true, mode: 0o755 });
|
||||||
|
const oldRelease = path.join(config.releasesDir, config.appVersion);
|
||||||
|
await mkdir(path.join(oldRelease, "dist"), { recursive: true, mode: 0o755 });
|
||||||
|
await writeFile(path.join(oldRelease, "dist", "marker"), "old");
|
||||||
|
await symlink(oldRelease, config.currentLink);
|
||||||
|
|
||||||
|
const source = path.join(root, "source");
|
||||||
|
await mkdir(path.join(source, "dist"), { recursive: true, mode: 0o755 });
|
||||||
|
await writeFile(path.join(source, "dist", "marker"), "new");
|
||||||
|
const archive = path.join(root, "release.tar.gz");
|
||||||
|
await createSafeArchive(source, archive);
|
||||||
|
const bytes = await readFile(archive);
|
||||||
|
const digest = createHash("sha256").update(bytes).digest("hex");
|
||||||
|
const jobId = randomUUID();
|
||||||
|
database = openDatabase(config);
|
||||||
|
const now = Date.now();
|
||||||
|
const assetName = `tallynote-1.2.0-${detectPlatform().target}.tar.gz`;
|
||||||
|
database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, operation, status, version, platform, asset_url,
|
||||||
|
expected_sha256, created_at, updated_at, requested_at)
|
||||||
|
VALUES (?, 'apply', 'queued', '1.2.0', ?, ?, ?, ?, ?, ?)
|
||||||
|
`).run(jobId, detectPlatform().target, "https://updates.example/" + assetName, digest, now, now, now);
|
||||||
|
globalThis.fetch = (async (input: string | URL) => {
|
||||||
|
const url = input.toString();
|
||||||
|
if (url.endsWith("/latest")) return new Response(JSON.stringify({ tag_name: "v1.2.0", assets: [{ name: assetName, browser_download_url: `https://updates.example/${assetName}` }, { name: "SHA256SUMS", browser_download_url: "https://updates.example/SHA256SUMS" }] }));
|
||||||
|
if (url.endsWith("SHA256SUMS")) return new Response(`${digest} ${assetName}\n`);
|
||||||
|
return new Response(bytes, { headers: { "content-length": String(bytes.length) } });
|
||||||
|
}) as typeof fetch;
|
||||||
|
|
||||||
|
await runUpdate({
|
||||||
|
metadataUrl: config.updateMetadataUrl,
|
||||||
|
version: "1.2.0",
|
||||||
|
currentVersion: config.appVersion,
|
||||||
|
currentDir: config.currentLink,
|
||||||
|
stagingDir: path.join(root, "staging"),
|
||||||
|
currentLink: config.currentLink,
|
||||||
|
releasesDir: config.releasesDir,
|
||||||
|
allowedHosts: config.updateAllowedHosts,
|
||||||
|
maxBytes: config.updateMaxBytes,
|
||||||
|
dataBackupMaxBytes: config.maxTotalBytes,
|
||||||
|
deferCompletion: true,
|
||||||
|
operation: "apply",
|
||||||
|
jobId,
|
||||||
|
sqlite: database.sqlite,
|
||||||
|
fetchImpl: globalThis.fetch,
|
||||||
|
});
|
||||||
|
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);
|
||||||
|
expect(row).toEqual({ operation: "apply", status: "applying" });
|
||||||
|
finalizeUpdateJob(database.sqlite, jobId, "failed");
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
||||||
|
} finally {
|
||||||
|
globalThis.fetch = previousFetch;
|
||||||
|
if (database) database.sqlite.close();
|
||||||
|
await rm(root, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it("在请求和恢复标记丢失后收敛孤儿任务,但保留 staged 下载", async () => {
|
||||||
|
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-reconcile-"));
|
||||||
|
let database: ReturnType<typeof openDatabase> | undefined;
|
||||||
|
try {
|
||||||
|
const dataDir = path.join(root, "data");
|
||||||
|
const installPrefix = path.join(root, "install");
|
||||||
|
process.env.TALLYNOTE_DATA_DIR = dataDir;
|
||||||
|
process.env.TALLYNOTE_INSTALL_PREFIX = installPrefix;
|
||||||
|
process.env.TALLYNOTE_PUBLIC_ORIGIN = "http://127.0.0.1:3998";
|
||||||
|
process.env.TALLYNOTE_COOKIE_SECURE = "false";
|
||||||
|
process.env.TALLYNOTE_UPDATE_STRATEGY = "systemd";
|
||||||
|
process.env.TALLYNOTE_UPDATE_METADATA_URL = "https://updates.example/latest";
|
||||||
|
process.env.TALLYNOTE_UPDATE_ALLOWED_HOSTS = "updates.example";
|
||||||
|
process.env.TALLYNOTE_UPDATE_REQUIRE_SIGNATURE = "false";
|
||||||
|
const config = loadConfig();
|
||||||
|
prepareDataDirectories(config);
|
||||||
|
await mkdir(path.join(config.releasesDir, config.appVersion, "dist"), { recursive: true });
|
||||||
|
await symlink(path.join(config.releasesDir, config.appVersion), config.currentLink);
|
||||||
|
database = openDatabase(config);
|
||||||
|
const staleAt = Date.now() - ORPHANED_UPDATE_TIMEOUT_MS - 1;
|
||||||
|
const insert = database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, operation, status, version, platform, asset_url, created_at, updated_at)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
|
`);
|
||||||
|
const queuedId = randomUUID();
|
||||||
|
const applyingId = randomUUID();
|
||||||
|
const stagedId = randomUUID();
|
||||||
|
const stagedApplyId = randomUUID();
|
||||||
|
insert.run(queuedId, "apply", "queued", "1.2.0", "linux-x64", "https://updates.example/queued.tar.gz", staleAt, staleAt);
|
||||||
|
insert.run(applyingId, "apply", "applying", config.appVersion, "linux-x64", "https://updates.example/applying.tar.gz", staleAt, staleAt);
|
||||||
|
insert.run(stagedId, "download", "staged", "1.2.0", "linux-x64", "https://updates.example/staged.tar.gz", staleAt, staleAt);
|
||||||
|
insert.run(stagedApplyId, "apply", "staged", "1.2.0", "linux-x64", "https://updates.example/staged-apply.tar.gz", staleAt, staleAt);
|
||||||
|
const now = Date.now();
|
||||||
|
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, now)).toBe(3);
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(queuedId)).toEqual({ status: "failed" });
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(applyingId)).toEqual({ status: "completed" });
|
||||||
|
expect(database.sqlite.prepare("SELECT status, operation FROM update_jobs WHERE id=?").get(stagedId)).toEqual({ status: "staged", operation: "download" });
|
||||||
|
expect(database.sqlite.prepare("SELECT status, operation FROM update_jobs WHERE id=?").get(stagedApplyId)).toEqual({ status: "staged", operation: "download" });
|
||||||
|
} finally {
|
||||||
|
if (database) database.sqlite.close();
|
||||||
|
await rm(root, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it("下载心跳有效时不回收任务或删除仍在使用的请求文件", async () => {
|
||||||
|
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-heartbeat-"));
|
||||||
|
let database: ReturnType<typeof openDatabase> | undefined;
|
||||||
|
try {
|
||||||
|
const dataDir = path.join(root, "data");
|
||||||
|
const installPrefix = path.join(root, "install");
|
||||||
|
process.env.TALLYNOTE_DATA_DIR = dataDir;
|
||||||
|
process.env.TALLYNOTE_INSTALL_PREFIX = installPrefix;
|
||||||
|
process.env.TALLYNOTE_PUBLIC_ORIGIN = "http://127.0.0.1:3998";
|
||||||
|
process.env.TALLYNOTE_COOKIE_SECURE = "false";
|
||||||
|
process.env.TALLYNOTE_UPDATE_STRATEGY = "systemd";
|
||||||
|
process.env.TALLYNOTE_UPDATE_METADATA_URL = "https://updates.example/latest";
|
||||||
|
process.env.TALLYNOTE_UPDATE_ALLOWED_HOSTS = "updates.example";
|
||||||
|
process.env.TALLYNOTE_UPDATE_REQUIRE_SIGNATURE = "false";
|
||||||
|
const config = loadConfig();
|
||||||
|
prepareDataDirectories(config);
|
||||||
|
await mkdir(path.join(config.releasesDir, config.appVersion, "dist"), { recursive: true });
|
||||||
|
await symlink(path.join(config.releasesDir, config.appVersion), config.currentLink);
|
||||||
|
database = openDatabase(config);
|
||||||
|
const staleAt = Date.now() - ORPHANED_UPDATE_TIMEOUT_MS - 1;
|
||||||
|
const jobId = randomUUID();
|
||||||
|
database.sqlite.prepare(`
|
||||||
|
INSERT INTO update_jobs(id, operation, status, version, platform, asset_url, created_at, updated_at)
|
||||||
|
VALUES (?, 'download', 'downloading', '1.2.0', 'linux-x64', ?, ?, ?)
|
||||||
|
`).run(jobId, "https://updates.example/download.tar.gz", staleAt, staleAt);
|
||||||
|
await writeFile(config.updateRequestPath, JSON.stringify({ jobId, operation: "download" }));
|
||||||
|
const statePath = path.join(config.installPrefix, ".update-state");
|
||||||
|
await writeFile(statePath, `job_id=${jobId}\nold_target=${path.join(config.releasesDir, config.appVersion)}\nphase=download\n`);
|
||||||
|
const now = Date.now();
|
||||||
|
await utimes(config.updateRequestPath, new Date(staleAt), new Date(staleAt));
|
||||||
|
await utimes(statePath, new Date(now), new Date(now));
|
||||||
|
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, now)).toBe(0);
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "downloading" });
|
||||||
|
expect(await stat(config.updateRequestPath)).toBeTruthy();
|
||||||
|
|
||||||
|
const expiredNow = now + ORPHANED_UPDATE_TIMEOUT_MS + 1;
|
||||||
|
await utimes(statePath, new Date(staleAt), new Date(staleAt));
|
||||||
|
expect(reconcileOrphanedUpdateJobs(database.sqlite, config, expiredNow)).toBe(1);
|
||||||
|
expect(database.sqlite.prepare("SELECT status FROM update_jobs WHERE id=?").get(jobId)).toEqual({ status: "failed" });
|
||||||
|
await expect(stat(config.updateRequestPath)).rejects.toThrow();
|
||||||
|
} finally {
|
||||||
|
if (database) database.sqlite.close();
|
||||||
|
await rm(root, { recursive: true, force: true });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
it("流式解包在展开大小上限前拒绝高压缩比归档,并修正发布树权限", async () => {
|
it("流式解包在展开大小上限前拒绝高压缩比归档,并修正发布树权限", async () => {
|
||||||
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-stream-"));
|
const root = await mkdtemp(path.join(tmpdir(), "tallynote-update-stream-"));
|
||||||
try {
|
try {
|
||||||
|
|||||||
Reference in New Issue
Block a user