collabdocchat 1.2.13 → 2.0.0

package/server/routes/export.js

@@ -0,0 +1,440 @@
+import express from 'express';
+import archiver from 'archiver';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { dirname } from 'path';
+import Message from '../models/Message.js';
+import Document from '../models/Document.js';
+import Task from '../models/Task.js';
+import File from '../models/File.js';
+import Group from '../models/Group.js';
+import User from '../models/User.js';
+import KnowledgeBase from '../models/KnowledgeBase.js';
+import { authenticate, isAdmin } from '../middleware/auth.js';
+import { asyncHandler, ValidationError, NotFoundError, AuthorizationError } from '../middleware/errorHandler.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+const router = express.Router();
+
+// 确保导出目录存在
+const exportsDir = path.join(__dirname, '../../exports');
+if (!fs.existsSync(exportsDir)) {
+  fs.mkdirSync(exportsDir, { recursive: true });
+}
+
+// 导出群组聊天记录（JSON格式）
+router.get('/group/:groupId/messages/json', authenticate, asyncHandler(async (req, res) => {
+  const { startDate, endDate } = req.query;
+
+  // 验证群组
+  const group = await Group.findById(req.params.groupId);
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限
+  const isMember = group.members.some(m => m.toString() === req.user.userId);
+  if (!isMember) {
+    throw new AuthorizationError('您不在该群组中');
+  }
+
+  // 构建查询
+  const query = { group: req.params.groupId };
+  if (startDate || endDate) {
+    query.timestamp = {};
+    if (startDate) query.timestamp.$gte = new Date(startDate);
+    if (endDate) query.timestamp.$lte = new Date(endDate);
+  }
+
+  const messages = await Message.find(query)
+    .populate('sender', 'username email')
+    .sort({ timestamp: 1 });
+
+  // 设置响应头
+  const filename = `messages_${group.name}_${Date.now()}.json`;
+  res.setHeader('Content-Type', 'application/json');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.json({
+    group: {
+      id: group._id,
+      name: group.name,
+      description: group.description
+    },
+    exportDate: new Date().toISOString(),
+    messageCount: messages.length,
+    messages: messages.map(msg => ({
+      id: msg._id,
+      sender: msg.sender ? {
+        id: msg.sender._id,
+        username: msg.sender.username,
+        email: msg.sender.email
+      } : null,
+      content: msg.content,
+      timestamp: msg.timestamp,
+      isRecalled: msg.isRecalled,
+      mentions: msg.mentions,
+      messageType: msg.messageType
+    }))
+  });
+}));
+
+// 导出群组聊天记录（TXT格式）
+router.get('/group/:groupId/messages/txt', authenticate, asyncHandler(async (req, res) => {
+  const { startDate, endDate } = req.query;
+
+  // 验证群组
+  const group = await Group.findById(req.params.groupId);
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限
+  const isMember = group.members.some(m => m.toString() === req.user.userId);
+  if (!isMember) {
+    throw new AuthorizationError('您不在该群组中');
+  }
+
+  // 构建查询
+  const query = { group: req.params.groupId };
+  if (startDate || endDate) {
+    query.timestamp = {};
+    if (startDate) query.timestamp.$gte = new Date(startDate);
+    if (endDate) query.timestamp.$lte = new Date(endDate);
+  }
+
+  const messages = await Message.find(query)
+    .populate('sender', 'username')
+    .sort({ timestamp: 1 });
+
+  // 生成文本内容
+  let content = `群组聊天记录导出\n`;
+  content += `群组名称: ${group.name}\n`;
+  content += `导出时间: ${new Date().toLocaleString('zh-CN')}\n`;
+  content += `消息数量: ${messages.length}\n`;
+  content += `\n${'='.repeat(80)}\n\n`;
+
+  messages.forEach(msg => {
+    const time = new Date(msg.timestamp).toLocaleString('zh-CN');
+    const sender = msg.sender ? msg.sender.username : '未知用户';
+    content += `[${time}] ${sender}:\n${msg.content}\n\n`;
+  });
+
+  // 设置响应头
+  const filename = `messages_${group.name}_${Date.now()}.txt`;
+  res.setHeader('Content-Type', 'text/plain; charset=utf-8');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.send(content);
+}));
+
+// 导出群组文档列表
+router.get('/group/:groupId/documents', authenticate, asyncHandler(async (req, res) => {
+  // 验证群组
+  const group = await Group.findById(req.params.groupId);
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限
+  const isMember = group.members.some(m => m.toString() === req.user.userId);
+  if (!isMember) {
+    throw new AuthorizationError('您不在该群组中');
+  }
+
+  const documents = await Document.find({ group: req.params.groupId })
+    .populate('creator', 'username email')
+    .sort({ createdAt: -1 });
+
+  const filename = `documents_${group.name}_${Date.now()}.json`;
+  res.setHeader('Content-Type', 'application/json');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.json({
+    group: {
+      id: group._id,
+      name: group.name
+    },
+    exportDate: new Date().toISOString(),
+    documentCount: documents.length,
+    documents: documents.map(doc => ({
+      id: doc._id,
+      title: doc.title,
+      content: doc.content,
+      creator: doc.creator ? {
+        id: doc.creator._id,
+        username: doc.creator.username,
+        email: doc.creator.email
+      } : null,
+      createdAt: doc.createdAt,
+      updatedAt: doc.updatedAt
+    }))
+  });
+}));
+
+// 导出群组任务列表
+router.get('/group/:groupId/tasks', authenticate, asyncHandler(async (req, res) => {
+  // 验证群组
+  const group = await Group.findById(req.params.groupId);
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限
+  const isMember = group.members.some(m => m.toString() === req.user.userId);
+  if (!isMember) {
+    throw new AuthorizationError('您不在该群组中');
+  }
+
+  const tasks = await Task.find({ group: req.params.groupId })
+    .populate('creator', 'username email')
+    .populate('assignee', 'username email')
+    .sort({ createdAt: -1 });
+
+  const filename = `tasks_${group.name}_${Date.now()}.json`;
+  res.setHeader('Content-Type', 'application/json');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.json({
+    group: {
+      id: group._id,
+      name: group.name
+    },
+    exportDate: new Date().toISOString(),
+    taskCount: tasks.length,
+    tasks: tasks.map(task => ({
+      id: task._id,
+      title: task.title,
+      description: task.description,
+      status: task.status,
+      priority: task.priority,
+      creator: task.creator ? {
+        id: task.creator._id,
+        username: task.creator.username,
+        email: task.creator.email
+      } : null,
+      assignee: task.assignee ? {
+        id: task.assignee._id,
+        username: task.assignee.username,
+        email: task.assignee.email
+      } : null,
+      dueDate: task.dueDate,
+      createdAt: task.createdAt,
+      updatedAt: task.updatedAt
+    }))
+  });
+}));
+
+// 导出知识库文档
+router.get('/group/:groupId/knowledge', authenticate, asyncHandler(async (req, res) => {
+  // 验证群组
+  const group = await Group.findById(req.params.groupId);
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限
+  const isMember = group.members.some(m => m.toString() === req.user.userId);
+  if (!isMember) {
+    throw new AuthorizationError('您不在该群组中');
+  }
+
+  const knowledgeList = await KnowledgeBase.find({ 
+    group: req.params.groupId,
+    status: 'published'
+  })
+    .populate('author', 'username email')
+    .sort({ createdAt: -1 });
+
+  const filename = `knowledge_${group.name}_${Date.now()}.json`;
+  res.setHeader('Content-Type', 'application/json');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.json({
+    group: {
+      id: group._id,
+      name: group.name
+    },
+    exportDate: new Date().toISOString(),
+    knowledgeCount: knowledgeList.length,
+    knowledge: knowledgeList.map(kb => ({
+      id: kb._id,
+      title: kb.title,
+      content: kb.content,
+      category: kb.category,
+      tags: kb.tags,
+      author: kb.author ? {
+        id: kb.author._id,
+        username: kb.author.username,
+        email: kb.author.email
+      } : null,
+      views: kb.views,
+      likesCount: kb.likes.length,
+      createdAt: kb.createdAt,
+      updatedAt: kb.updatedAt
+    }))
+  });
+}));
+
+// 导出群组完整数据（ZIP压缩包）
+router.get('/group/:groupId/full', authenticate, asyncHandler(async (req, res) => {
+  // 验证群组
+  const group = await Group.findById(req.params.groupId)
+    .populate('admin', 'username email')
+    .populate('members', 'username email');
+  
+  if (!group) {
+    throw new NotFoundError('群组');
+  }
+
+  // 检查权限（只有管理员可以导出完整数据）
+  const isGroupAdmin = group.admin._id.toString() === req.user.userId;
+  if (!isGroupAdmin && req.user.role !== 'admin') {
+    throw new AuthorizationError('只有管理员可以导出完整数据');
+  }
+
+  // 获取所有数据
+  const [messages, documents, tasks, files, knowledgeList] = await Promise.all([
+    Message.find({ group: req.params.groupId }).populate('sender', 'username email'),
+    Document.find({ group: req.params.groupId }).populate('creator', 'username email'),
+    Task.find({ group: req.params.groupId }).populate('creator assignee', 'username email'),
+    File.find({ group: req.params.groupId }).populate('uploader', 'username email'),
+    KnowledgeBase.find({ group: req.params.groupId }).populate('author', 'username email')
+  ]);
+
+  // 创建ZIP文件
+  const filename = `group_${group.name}_${Date.now()}.zip`;
+  const zipPath = path.join(exportsDir, filename);
+
+  res.setHeader('Content-Type', 'application/zip');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  const archive = archiver('zip', {
+    zlib: { level: 9 }
+  });
+
+  archive.on('error', (err) => {
+    throw err;
+  });
+
+  archive.pipe(res);
+
+  // 添加群组信息
+  archive.append(JSON.stringify({
+    id: group._id,
+    name: group.name,
+    description: group.description,
+    admin: group.admin,
+    members: group.members,
+    createdAt: group.createdAt
+  }, null, 2), { name: 'group_info.json' });
+
+  // 添加消息
+  archive.append(JSON.stringify(messages, null, 2), { name: 'messages.json' });
+
+  // 添加文档
+  archive.append(JSON.stringify(documents, null, 2), { name: 'documents.json' });
+
+  // 添加任务
+  archive.append(JSON.stringify(tasks, null, 2), { name: 'tasks.json' });
+
+  // 添加文件列表
+  archive.append(JSON.stringify(files.map(f => ({
+    id: f._id,
+    filename: f.filename,
+    originalName: f.originalName,
+    size: f.size,
+    mimetype: f.mimetype,
+    uploader: f.uploader,
+    createdAt: f.createdAt
+  })), null, 2), { name: 'files.json' });
+
+  // 添加知识库
+  archive.append(JSON.stringify(knowledgeList, null, 2), { name: 'knowledge.json' });
+
+  // 添加README
+  const readme = `# ${group.name} 数据导出
+
+导出时间: ${new Date().toLocaleString('zh-CN')}
+导出人: ${req.user.username}
+
+## 包含内容
+
+- group_info.json: 群组基本信息
+- messages.json: 聊天记录 (${messages.length} 条)
+- documents.json: 文档列表 (${documents.length} 个)
+- tasks.json: 任务列表 (${tasks.length} 个)
+- files.json: 文件列表 (${files.length} 个)
+- knowledge.json: 知识库 (${knowledgeList.length} 个)
+
+## 注意事项
+
+1. 此导出不包含实际的上传文件，仅包含文件元数据
+2. 所有数据均为JSON格式，可使用文本编辑器或JSON查看器打开
+3. 请妥善保管导出数据，避免泄露敏感信息
+`;
+
+  archive.append(readme, { name: 'README.md' });
+
+  await archive.finalize();
+}));
+
+// 导出用户个人数据
+router.get('/user/my-data', authenticate, asyncHandler(async (req, res) => {
+  const user = await User.findById(req.user.userId);
+  if (!user) {
+    throw new NotFoundError('用户');
+  }
+
+  // 获取用户相关数据
+  const [myMessages, myDocuments, myTasks, myFiles, myKnowledge] = await Promise.all([
+    Message.find({ sender: req.user.userId }).populate('group', 'name'),
+    Document.find({ creator: req.user.userId }).populate('group', 'name'),
+    Task.find({ $or: [{ creator: req.user.userId }, { assignee: req.user.userId }] }).populate('group', 'name'),
+    File.find({ uploader: req.user.userId }).populate('group', 'name'),
+    KnowledgeBase.find({ author: req.user.userId }).populate('group', 'name')
+  ]);
+
+  const filename = `my_data_${Date.now()}.json`;
+  res.setHeader('Content-Type', 'application/json');
+  res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(filename)}"`);
+
+  res.json({
+    user: {
+      id: user._id,
+      username: user.username,
+      email: user.email,
+      role: user.role,
+      createdAt: user.createdAt
+    },
+    exportDate: new Date().toISOString(),
+    statistics: {
+      messageCount: myMessages.length,
+      documentCount: myDocuments.length,
+      taskCount: myTasks.length,
+      fileCount: myFiles.length,
+      knowledgeCount: myKnowledge.length
+    },
+    data: {
+      messages: myMessages,
+      documents: myDocuments,
+      tasks: myTasks,
+      files: myFiles.map(f => ({
+        id: f._id,
+        filename: f.filename,
+        originalName: f.originalName,
+        size: f.size,
+        mimetype: f.mimetype,
+        group: f.group,
+        createdAt: f.createdAt
+      })),
+      knowledge: myKnowledge
+    }
+  });
+}));
+
+export default router;
+