blog/scripts/backup-run.mjs

#!/usr/bin/env node
/**
 * 备份包装器 —— 给「唯一辅仓」加一道失败告警
 * =====================================================================
 * 为什么需要它：
 *   OpenList（中兴 F50）现在是**唯一的异地备份**。而这类「每天自动跑」的任务
 *   最典型的失败模式恰恰是**静默**的 —— F50 被带出门、换了网段、OpenList
 *   容器重启后没起来、WebDAV 口令改过……这些都不会有人主动发现，
 *   直到某天真的要用备份时才发现最近三个月一份都没成功。
 *
 *   CNB 主仓还在，所以这不会立刻致命；但「唯一辅仓悄悄失效」这件事本身
 *   必须被告知。于是加这一层：跑备份 → 失败就发邮件。
 *
 * 它做什么：
 *   1. 读 .workbuddy-backup/openlist-backup.env（KEY=VALUE，# 开头为注释）
 *      —— 只填充**尚未设置**的环境变量，环境变量优先（与 backup-bundle.mjs 一致）
 *   2. 跑 scripts/backup-bundle.mjs，参数原样转发，输出实时透传
 *   3. 退出码非 0 → 用 scripts/send_mail.js 发告警邮件，正文附日志尾部
 *      成功时默认不发（避免每天一封的邮件疲劳）；加 --notify-success 才发
 *   4. **原样返回 backup-bundle.mjs 的退出码**（发信失败不改变它 ——
 *      通知不该掩盖真正的故障，也不该把成功改判成失败）
 *
 * 用法（一般由 scripts/backup-task.cmd 调用，也可手跑）：
 *   node scripts/backup-run.mjs                       # 等价于 backup-bundle.mjs 默认参数
 *   node scripts/backup-run.mjs --verify --keep 7
 *   node scripts/backup-run.mjs --dry --notify-success
 *   node scripts/backup-run.mjs --list
 *
 * 邮件相关环境变量（未配则静默跳过发信，不让通知拖垮备份）：
 *   SMTP_HOST / SMTP_PORT / SMTP_USER / SMTP_PASS / SMTP_TO
 *   未设 SMTP_USER 时会回落到 MAIL_USERNAME（本站既有命名），SMTP_PASS 同理。
 */

import fs from 'node:fs';
import path from 'node:path';
import { spawn } from 'node:child_process';
import { fileURLToPath } from 'node:url';

const HERE = path.dirname(fileURLToPath(import.meta.url));
const REPO = path.resolve(HERE, '..');
const BACKUP_SCRIPT = path.join(HERE, 'backup-bundle.mjs');
const MAIL_SCRIPT = path.join(HERE, 'send_mail.js');
const ENV_FILE = path.join(REPO, '.workbuddy-backup', 'openlist-backup.env');

function log(...a) {
  console.log('[' + new Date().toTimeString().slice(0, 8) + ']', ...a);
}

// ---- 1. 载入 env 文件（只补空缺，环境变量优先） ----
function loadEnvFile(file) {
  let raw;
  try {
    raw = fs.readFileSync(file, 'utf8');
  } catch {
    return { loaded: false, count: 0 };
  }
  let count = 0;
  for (const line of raw.split(/\r?\n/)) {
    const s = line.trim();
    if (!s || s.startsWith('#')) continue;
    const eq = s.indexOf('=');
    if (eq <= 0) continue;
    const k = s.slice(0, eq).trim();
    // 值可能被引号包住 —— 只做去引号这一种规整
    let v = s.slice(eq + 1).trim();
    if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) {
      v = v.slice(1, -1);
    }
    if (process.env[k] === undefined) {
      process.env[k] = v;
      count++;
    }
  }
  return { loaded: true, count };
}

const envInfo = loadEnvFile(ENV_FILE);
log('配置文件:', envInfo.loaded ? `${ENV_FILE}（补入 ${envInfo.count} 项）` : `${ENV_FILE} 不存在，只用环境变量`);

// ---- 2. 参数拆分：--notify-success 归自己，其余原样转发给 backup-bundle.mjs ----
const argv = process.argv.slice(2);
const notifySuccess = argv.includes('--notify-success');
const passThrough = argv.filter((a) => a !== '--notify-success');

function mailConfigured() {
  const user = process.env.SMTP_USER || process.env.MAIL_USERNAME || '';
  const pass = process.env.SMTP_PASS || process.env.MAIL_PASSWORD || '';
  return Boolean(user && pass);
}

// ---- 3. 跑备份，实时透传输出，同时留一份尾部用于邮件正文 ----
log('开始：', process.execPath.split(/[\\/]/).pop(), BACKUP_SCRIPT, passThrough.join(' ') || '(默认参数)');
const t0 = Date.now();
const child = spawn(process.execPath, [BACKUP_SCRIPT, ...passThrough], {
  cwd: REPO,
  env: process.env,
  stdio: ['ignore', 'pipe', 'pipe'],
});

const TAIL_MAX = 6000;
let tail = '';
const keep = (d) => {
  tail += d.toString('utf8');
  if (tail.length > TAIL_MAX) tail = tail.slice(-TAIL_MAX);
};

child.stdout.on('data', (d) => { process.stdout.write(d); keep(d); });
child.stderr.on('data', (d) => { process.stderr.write(d); keep(d); });

const code = await new Promise((resolve) => {
  child.on('close', (c) => resolve(c === null ? 1 : c));
  child.on('error', (e) => {
    console.error('启动备份失败:', e.message);
    keep(`启动备份失败: ${e.message}\n`);
    resolve(1);
  });
});

const secs = ((Date.now() - t0) / 1000).toFixed(1);
log(`结束：exit=${code}  用时 ${secs}s`);

// ---- 4. 发信 ----
// 退出用 process.exitCode 而不是 process.exit()：到这一步 stdout 还可能有
// 未排干的写入，process.exit() 会在 flush 前终止进程（Node 官方文档明确警告过）。
// 这里已经是脚本尾部，设完让 Node 自行结束即可。
if (code === 0 && !notifySuccess) {
  log('备份成功（未配置 --notify-success，不发信）');
  process.exitCode = 0;
} else if (!mailConfigured()) {
  log('未配置 SMTP 凭据，跳过发信。需要时在 openlist-backup.env 里补 SMTP_USER / SMTP_PASS');
  process.exitCode = code;
} else {
  const ok = code === 0;
  const stamp = new Date().toLocaleString('zh-CN', { hour12: false });
  const subject = ok
    ? '【优世界】整仓离线备份成功'
    : `【优世界】★ 整仓离线备份失败（exit=${code}）`;

  const body = [
    `时间：${stamp}`,
    `结果：${ok ? '成功' : '失败'}（exit=${code}）`,
    `用时：${secs}s`,
    `命令：backup-bundle.mjs ${passThrough.join(' ') || '(默认参数)'}`,
    '',
    ok ? '备份已上传到 OpenList 并通过校验。' : '★ 备份没有完成 —— 异地备份这一层目前是失效的。',
    '',
    '常见原因：',
    '  · 中兴 F50 不在线 / 换了网段 → 检查 OPENLIST_URL 是否仍可达',
    '  · OpenList 容器重启后没起来',
    '  · WebDAV 账号或口令改过',
    '  · 仓库体积增长导致超时',
    '',
    '--------- 输出尾部 ---------',
    tail.trim() || '(无输出)',
  ].join('\n');

  log('发送通知邮件…');
  const mail = spawn(process.execPath, [MAIL_SCRIPT], {
    cwd: REPO,
    env: { ...process.env, MAIL_SUBJECT: subject, MAIL_BODY: body },
    stdio: 'inherit',
  });
  const mailCode = await new Promise((r) => mail.on('close', (c) => r(c === null ? 1 : c)));
  log(mailCode === 0 ? '通知邮件已处理' : `通知邮件失败（exit=${mailCode}，不影响备份退出码）`);

  process.exitCode = code;
}