// clearDigitalHumanData.js
import redis from './redis/index.js';
import initQueue from './redis/initQueue.js';

async function clearDigitalHumanData() {
  try {
    console.log('开始清除数字人相关数据...');
    
    // 1. 清除等待队列
    await redis.del('digitalHuman:runninghub:wait');
    await redis.del('digitalHuman:coze:wait');
    console.log('已清除等待队列');
    
    // 2. 清除所有数字人相关的任务数据
    let cursor = '0';
    do {
      // 调用scan并打印返回结果（便于调试）
      const result = await redis.scan(cursor, {
        MATCH: `${initQueue.prefix}:task:*`,
        COUNT: 100
      });
      // 【核心修复】从scan返回的对象中正确提取cursor和keys
      const newCursor = result.cursor; // 取cursor属性
      const keys = result.keys || [];   // 取keys属性，兜底为空数组
      
      console.log(`当前游标: ${newCursor}, 找到keys数量: ${keys.length}`); // 调试日志
      
      if (keys.length > 0) {
        // 加强过滤，确保只保留有效字符串key
        const validKeys = keys.filter(key => {
          return typeof key === 'string' && key.trim() !== '';
        });
        
        if (validKeys.length > 0) {
          await redis.del(...validKeys);
          console.log(`已清除 ${validKeys.length} 个任务数据`);
        }
      }
      
      cursor = newCursor;
    } while (cursor !== '0');
    
    // 3. 清除轮询队列数据
    cursor = '0';
    do {
      const result = await redis.scan(cursor, {
        MATCH: `${initQueue.prefix}:processPolling:*`,
        COUNT: 100
      });
      // 同样修复解构问题
      const newCursor = result.cursor;
      const keys = result.keys || [];
      
      console.log(`当前游标: ${newCursor}, 找到轮询keys数量: ${keys.length}`);
      
      if (keys.length > 0) {
        const validKeys = keys.filter(key => {
          return typeof key === 'string' && key.trim() !== '';
        });
        
        if (validKeys.length > 0) {
          await redis.del(...validKeys);
          console.log(`已清除 ${validKeys.length} 个轮询队列数据`);
        }
      }
      
      cursor = newCursor;
    } while (cursor !== '0');
    
    // 4. 清除结果队列数据
    await redis.del(initQueue.resultName);
    await redis.del(initQueue.resultList);
    console.log('已清除结果队列数据');
    
    // 5. 清除错误队列数据
    await redis.del(initQueue.errorName);
    await redis.del(initQueue.errorList);
    console.log('已清除错误队列数据');
    
    // 6. 清除回调队列数据
    await redis.del(initQueue.callback);
    console.log('已清除回调队列数据');
    
    // 7. 重置平台信息中的数字人相关计数
    await redis.json.set(initQueue.initInfoKey, '$.platforms.digitalHuman:runninghub.WQtasks', '0');
    await redis.json.set(initQueue.initInfoKey, '$.platforms.digitalHuman:runninghub.PQtasks', '0');
    await redis.json.set(initQueue.initInfoKey, '$.platforms.digitalHuman:coze.WQtasks', '0');
    await redis.json.set(initQueue.initInfoKey, '$.platforms.digitalHuman:coze.PQtasks', '0');
    console.log('已重置平台计数');
    
    console.log('数字人相关数据清除完成！');
  } catch (error) {
    console.error('清除数据时出错:', error);
    process.exit(1); // 异常退出进程
  } finally {
    if (redis.isOpen) {
      await redis.disconnect();
      console.log('Redis 连接已正常关闭');
    }
  }
}

clearDigitalHumanData();