multi-platform-media-manage/server/src/services/taskExecutors.ts

/**
 * 任务执行器注册
 * 在应用启动时注册所有任务类型的执行器
 */
import { Task, TaskResult, TaskProgressUpdate } from '@media-manager/shared';
import { taskQueueService } from './TaskQueueService.js';
import { CommentService } from './CommentService.js';
import { WorkService } from './WorkService.js';
import { AccountService } from './AccountService.js';
import { PublishService } from './PublishService.js';
import { logger } from '../utils/logger.js';

// 创建服务实例
const commentService = new CommentService();
const workService = new WorkService();
const accountService = new AccountService();
const publishService = new PublishService();

type ProgressUpdater = (update: Partial<TaskProgressUpdate>) => void;

/**
 * 同步评论任务执行器
 */
async function syncCommentsExecutor(task: Task, updateProgress: ProgressUpdater): Promise<TaskResult> {
  updateProgress({ progress: 5, currentStep: '连接平台...' });

  const onProgress = (current: number, total: number, workTitle: string) => {
    const progress = total > 0 ? Math.min(90, Math.round((current / total) * 90) + 5) : 50;
    updateProgress({
      progress,
      currentStep: `正在同步: ${workTitle || `作品 ${current}/${total}`}`,
      currentStepIndex: current,
    });
  };

  // 从任务中获取 userId（需要在创建任务时传入）
  const userId = (task as Task & { userId?: number }).userId;
  if (!userId) {
    throw new Error('缺少用户ID');
  }

  const result = await commentService.syncComments(userId, task.accountId, onProgress);

  updateProgress({ progress: 100, currentStep: '同步完成' });

  return {
    success: true,
    message: `同步完成，共同步 ${result.synced} 条评论`,
    data: {
      syncedCount: result.synced,
      accountCount: result.accounts,
    },
  };
}

/**
 * 同步作品任务执行器
 */
async function syncWorksExecutor(task: Task, updateProgress: ProgressUpdater): Promise<TaskResult> {
  updateProgress({ progress: 5, currentStep: '获取作品列表...' });

  const userId = (task as Task & { userId?: number }).userId;
  if (!userId) {
    throw new Error('缺少用户ID');
  }

  const result = await workService.syncWorks(userId, task.accountId);

  updateProgress({ progress: 100, currentStep: '同步完成' });

  return {
    success: true,
    message: `同步完成，共同步 ${result.synced} 个作品`,
    data: {
      syncedCount: result.synced,
      accountCount: result.accounts,
    },
  };
}

/**
 * 同步账号信息任务执行器
 */
async function syncAccountExecutor(task: Task, updateProgress: ProgressUpdater): Promise<TaskResult> {
  updateProgress({ progress: 10, currentStep: '获取账号信息...' });

  const userId = (task as Task & { userId?: number }).userId;
  if (!userId) {
    throw new Error('缺少用户ID');
  }

  if (!task.accountId) {
    throw new Error('缺少账号ID');
  }

  await accountService.refreshAccount(userId, task.accountId);

  updateProgress({ progress: 100, currentStep: '同步完成' });

  return {
    success: true,
    message: '账号信息已更新',
  };
}

/**
 * 发布视频任务执行器
 */
async function publishVideoExecutor(task: Task, updateProgress: ProgressUpdater): Promise<TaskResult> {
  updateProgress({ progress: 5, currentStep: '准备发布...' });

  const userId = (task as Task & { userId?: number }).userId;
  if (!userId) {
    throw new Error('缺少用户ID');
  }

  // 从任务数据中获取发布任务ID
  const taskData = task as Task & { publishTaskId?: number };
  if (!taskData.publishTaskId) {
    throw new Error('缺少发布任务ID');
  }

  // 执行发布任务
  await publishService.executePublishTaskWithProgress(
    taskData.publishTaskId, 
    userId,
    (progress, message) => {
      updateProgress({ progress, currentStep: message });
    }
  );

  updateProgress({ progress: 100, currentStep: '发布完成' });

  return {
    success: true,
    message: '视频发布任务已完成',
  };
}

/**
 * 删除平台作品任务执行器
 */
async function deleteWorkExecutor(task: Task, updateProgress: ProgressUpdater): Promise<TaskResult> {
  updateProgress({ progress: 10, currentStep: '准备删除...' });

  const userId = (task as Task & { userId?: number }).userId;
  if (!userId) {
    throw new Error('缺少用户ID');
  }

  const taskData = task as Task & { workId?: number };
  if (!taskData.workId) {
    throw new Error('缺少作品ID');
  }

  updateProgress({ progress: 30, currentStep: '连接平台删除作品...' });

  // 执行平台删除
  const result = await workService.deletePlatformWork(userId, taskData.workId);

  if (result.success) {
    updateProgress({ progress: 80, currentStep: '删除本地记录...' });
    
    // 平台删除成功后，删除本地记录
    try {
      await workService.deleteWork(userId, taskData.workId);
      logger.info(`Local work ${taskData.workId} deleted after platform deletion`);
    } catch (error) {
      logger.warn(`Failed to delete local work ${taskData.workId}:`, error);
      // 本地删除失败不影响整体结果
    }
  }

  updateProgress({ progress: 100, currentStep: result.success ? '删除完成' : '删除失败' });

  return {
    success: result.success,
    message: result.success ? '作品已从平台删除，本地记录已清理' : (result.errorMessage || '删除失败'),
  };
}

/**
 * 注册所有任务执行器
 */
export function registerTaskExecutors(): void {
  taskQueueService.registerExecutor('sync_comments', syncCommentsExecutor);
  taskQueueService.registerExecutor('sync_works', syncWorksExecutor);
  taskQueueService.registerExecutor('sync_account', syncAccountExecutor);
  taskQueueService.registerExecutor('publish_video', publishVideoExecutor);
  taskQueueService.registerExecutor('delete_work', deleteWorkExecutor);
  
  logger.info('All task executors registered');
}