yunrui_asset/backend/utils/autoBackupScheduler.js

/**
 * 自动备份调度器模块
 * 使用 node-cron 实现定时自动备份功能
 */

const cron = require('node-cron');
const path = require('path');
const fs = require('fs');
const { createBackup, createIncrementalBackup, getBackupPath } = require('./backup');
const { uploadToRemote } = require('./remoteBackup');
const { getEnabledTargets, getGlobalSettings } = require('./remoteBackupConfig');

// 全局调度器存储
const schedulers = new Map();

// 备份设置文件路径
const SETTINGS_FILE = path.join(__dirname, '..', 'config', 'auto-backup-settings.json');

// 默认设置
const DEFAULT_SETTINGS = {
  enabled: false,
  cronExpression: '0 2 * * *', // 每天凌晨 2 点
  description: '自动备份',
  backupType: 'full', // 'full' 或 'incremental'
  includeFiles: true,
  compress: true,
  maxCount: 30,
  maxAgeDays: 90,
};

/**
 * 加载备份设置
 */
function loadSettings() {
  try {
    if (fs.existsSync(SETTINGS_FILE)) {
      const content = fs.readFileSync(SETTINGS_FILE, 'utf8');
      return { ...DEFAULT_SETTINGS, ...JSON.parse(content) };
    }
  } catch (error) {
    console.error('加载自动备份设置失败:', error);
  }
  return { ...DEFAULT_SETTINGS };
}

/**
 * 保存备份设置
 */
function saveSettings(settings) {
  try {
    const configDir = path.dirname(SETTINGS_FILE);
    if (!fs.existsSync(configDir)) {
      fs.mkdirSync(configDir, { recursive: true });
    }
    fs.writeFileSync(SETTINGS_FILE, JSON.stringify(settings, null, 2), 'utf8');
    return true;
  } catch (error) {
    console.error('保存自动备份设置失败:', error);
    return false;
  }
}

/**
 * 验证 Cron 表达式
 */
function validateCronExpression(expression) {
  return cron.validate(expression);
}

/**
 * 将中文时间转换为 Cron 表达式
 */
function timeToCron(hour, minute) {
  return `${minute} ${hour} * * *`;
}

/**
 * 创建自动备份任务
 */
function createAutoBackupTask(settings) {
  const { cronExpression, description, backupType, includeFiles, compress, maxCount, maxAgeDays } = settings;

  if (!validateCronExpression(cronExpression)) {
    throw new Error('无效的 Cron 表达式');
  }

  // 如果已有调度器，先停止
  if (schedulers.has('auto-backup')) {
    stopAutoBackup();
  }

  // 创建新的调度器
  const task = cron.schedule(cronExpression, async () => {
    console.log('=== 开始执行自动备份 ===');
    try {
      const timestamp = new Date().toISOString().replace(/[:.]/g, '-').substring(0, 19);
      
      // 根据备份类型选择函数
      const backupFunction = backupType === 'incremental' ? createIncrementalBackup : createBackup;
      
      const result = await backupFunction({
        description: `${description} - ${timestamp}`,
        includeFiles,
        compress,
        autoClean: true,
        maxCount,
        maxAgeDays,
      });

      if (result) {
        console.log('自动备份完成:', result.filename);
        console.log(`备份类型：${result.isIncremental ? '增量备份' : '全量备份'}`);
        
        // 上传到远端
        await uploadToRemoteTargets(result.path, result.filename);
        
        console.log('========================\n');
      } else {
        console.log('无数据变化，跳过备份');
      }
    } catch (error) {
      console.error('自动备份失败:', error);
      console.error('========================\n');
    }
  }, {
    scheduled: true,
    timezone: 'Asia/Shanghai', // 设置时区为中国时区
  });

  schedulers.set('auto-backup', task);
  console.log(`自动备份任务已启动，Cron 表达式：${cronExpression}, 备份类型：${backupType}`);

  return task;
}

/**
 * 启动自动备份
 */
function startAutoBackup(settings = null) {
  if (!settings) {
    settings = loadSettings();
  }

  if (!settings.enabled) {
    console.log('自动备份已禁用');
    return false;
  }

  try {
    createAutoBackupTask(settings);
    return true;
  } catch (error) {
    console.error('启动自动备份失败:', error.message);
    return false;
  }
}

/**
 * 停止自动备份
 */
function stopAutoBackup() {
  if (schedulers.has('auto-backup')) {
    const task = schedulers.get('auto-backup');
    task.stop();
    schedulers.delete('auto-backup');
    console.log('自动备份任务已停止');
    return true;
  }
  return false;
}

/**
 * 获取自动备份状态
 */
function getAutoBackupStatus() {
  const settings = loadSettings();
  const isActive = schedulers.has('auto-backup');

  // 计算下次执行时间
  let nextRun = null;
  if (isActive && settings.enabled) {
    // 简单计算下次执行时间（基于当前时间和 Cron 表达式）
    const now = new Date();
    const [minute, hour] = settings.cronExpression.split(' ').slice(0, 2);
    
    const next = new Date(now);
    next.setHours(parseInt(hour), parseInt(minute), 0, 0);
    
    if (next <= now) {
      next.setDate(next.getDate() + 1);
    }
    
    nextRun = next.toISOString();
  }

  return {
    enabled: settings.enabled,
    isActive,
    cronExpression: settings.cronExpression,
    description: settings.description,
    backupType: settings.backupType || 'full',
    includeFiles: settings.includeFiles,
    compress: settings.compress,
    maxCount: settings.maxCount,
    maxAgeDays: settings.maxAgeDays,
    nextRun,
  };
}

/**
 * 更新自动备份设置
 */
function updateAutoBackupSettings(newSettings) {
  const currentSettings = loadSettings();
  const updatedSettings = { ...currentSettings, ...newSettings };

  // 如果提供了小时和分钟，转换为 Cron 表达式
  if (newSettings.hour !== undefined && newSettings.minute !== undefined) {
    updatedSettings.cronExpression = timeToCron(newSettings.hour, newSettings.minute);
    delete updatedSettings.hour;
    delete updatedSettings.minute;
  }

  // 验证 Cron 表达式
  if (!validateCronExpression(updatedSettings.cronExpression)) {
    throw new Error('无效的 Cron 表达式');
  }

  // 保存设置
  if (saveSettings(updatedSettings)) {
    // 如果启用了自动备份，重新启动调度器
    if (updatedSettings.enabled) {
      startAutoBackup(updatedSettings);
    } else {
      stopAutoBackup();
    }
    return true;
  }

  return false;
}

/**
 * 上传备份到所有启用的远端目标
 */
async function uploadToRemoteTargets(localFilePath, filename) {
  const globalSettings = getGlobalSettings();
  
  if (!globalSettings.enabled || !globalSettings.uploadAfterBackup) {
    console.log('远端备份已禁用');
    return [];
  }
  
  const enabledTargets = getEnabledTargets();
  
  if (enabledTargets.length === 0) {
    console.log('没有启用的远端备份目标');
    return [];
  }
  
  const uploadResults = [];
  
  for (const target of enabledTargets) {
    try {
      console.log(`开始上传到目标：${target.name} (${target.protocol})`);
      
      const remotePath = `${target.prefix || 'backups/'}${filename}`;
      
      const result = await uploadToRemote(target, localFilePath, remotePath);
      
      uploadResults.push({
        targetId: target.id,
        targetName: target.name,
        protocol: target.protocol,
        success: true,
        ...result,
      });
      
      console.log(`上传到 ${target.name} 成功`);
    } catch (error) {
      console.error(`上传到 ${target.name} 失败:`, error.message);
      uploadResults.push({
        targetId: target.id,
        targetName: target.name,
        protocol: target.protocol,
        success: false,
        error: error.message,
      });
    }
  }
  
  // 检查是否需要删除本地文件
  const settings = getGlobalSettings();
  if (settings.deleteLocalAfterUpload && uploadResults.every(r => r.success)) {
    try {
      fs.unlinkSync(localFilePath);
      console.log('本地备份文件已删除');
    } catch (error) {
      console.error('删除本地备份文件失败:', error.message);
    }
  }
  
  return uploadResults;
}

/**
 * 立即执行一次备份
 */
async function executeBackupNow(options = {}) {
  console.log('=== 手动触发备份 ===');
  try {
    const settings = loadSettings();
    const result = await createBackup({
      description: options.description || '手动备份',
      includeFiles: options.includeFiles !== undefined ? options.includeFiles : settings.includeFiles,
      compress: options.compress !== undefined ? options.compress : settings.compress,
      autoClean: true,
      maxCount: settings.maxCount,
      maxAgeDays: settings.maxAgeDays,
    });

    console.log('手动备份完成:', result.filename);
    
    // 上传到远端
    const uploadResults = await uploadToRemoteTargets(result.path, result.filename);
    
    console.log('====================\n');
    return { 
      success: true, 
      result,
      remoteUploads: uploadResults,
    };
  } catch (error) {
    console.error('手动备份失败:', error);
    console.error('====================\n');
    return { success: false, error: error.message };
  }
}

/**
 * 初始化自动备份（服务器启动时调用）
 */
function initAutoBackup() {
  console.log('初始化自动备份...');
  const settings = loadSettings();
  
  if (settings.enabled) {
    startAutoBackup(settings);
  } else {
    console.log('自动备份当前为禁用状态');
  }

  return getAutoBackupStatus();
}

module.exports = {
  loadSettings,
  saveSettings,
  validateCronExpression,
  timeToCron,
  startAutoBackup,
  stopAutoBackup,
  getAutoBackupStatus,
  updateAutoBackupSettings,
  executeBackupNow,
  initAutoBackup,
};