jurong_Intermediate/routes/upload.js

const express = require('express');
const multer = require('multer');
const path = require('path');
const { auth } = require('../middleware/auth');
const { authenticateToken } = require('./auth');
const minioService = require('../services/minioService');
const { initializeBuckets } = require('../config/minio');

const router = express.Router();


// 配置multer内存存储（用于MinIO上传）
const storage = multer.memoryStorage();

// 文件过滤器 - 支持图片和视频
const fileFilter = (req, file, cb) => {
  // 允许图片和视频文件
  if (file.mimetype.startsWith('image/') || file.mimetype.startsWith('video/')) {
    cb(null, true);
  } else {
    cb(new Error('只能上传图片或视频文件'), false);
  }
};

// 单文件上传配置
const upload = multer({
  storage: storage,
  fileFilter: fileFilter,
  limits: {
    fileSize: 5 * 1024 * 1024, // 5MB
    files: 1 // 一次只能上传一个文件
  }
});

// 多文件上传配置
const multiUpload = multer({
  storage: storage,
  fileFilter: fileFilter,
  limits: {
    fileSize: 10 * 1024 * 1024, // 10MB (视频文件更大)
    files: 10 // 最多10个文件
  }
});

router.post('/image', authenticateToken, (req, res) => {
  upload.single('file')(req, res, async (err) => {
    if (err instanceof multer.MulterError) {
      if (err.code === 'LIMIT_FILE_SIZE') {
        return res.status(400).json({
          success: false,
          message: '文件大小不能超过 5MB'
        });
      }
      if (err.code === 'LIMIT_FILE_COUNT') {
        return res.status(400).json({
          success: false,
          message: '一次只能上传一个文件'
        });
      }
      return res.status(400).json({
        success: false,
        message: '文件上传失败：' + err.message
      });
    } else if (err) {
      return res.status(400).json({
        success: false,
        message: err.message
      });
    }

    if (!req.file) {
      return res.status(400).json({
        success: false,
        message: '请选择要上传的文件'
      });
    }

    try {
      // 使用MinIO服务上传文件
      const type = req.body.type || 'document';
      const result = await minioService.uploadFile(
        req.file.buffer,
        req.file.originalname,
        req.file.mimetype,
        type
      );

      res.json({
        success: true,
        message: '文件上传成功',
        data: result.data
      });
    } catch (error) {
      console.error('文件上传到MinIO失败:', error);
      res.status(500).json({
        success: false,
        message: error.message || '文件上传失败'
      });
    }
  });
});


router.post('/', authenticateToken, (req, res) => {
  multiUpload.array('file', 10)(req, res, async (err) => {
    if (err instanceof multer.MulterError) {
      if (err.code === 'LIMIT_FILE_SIZE') {
        return res.status(400).json({
          success: false,
          message: '文件大小不能超过 10MB'
        });
      }
      if (err.code === 'LIMIT_FILE_COUNT') {
        return res.status(400).json({
          success: false,
          message: '一次最多只能上传10个文件'
        });
      }
      return res.status(400).json({
        success: false,
        message: '文件上传失败：' + err.message
      });
    } else if (err) {
      return res.status(400).json({
        success: false,
        message: err.message
      });
    }

    if (!req.files || req.files.length === 0) {
      return res.status(400).json({
        success: false,
        message: '请选择要上传的文件'
      });
    }

    try {
      // 使用MinIO服务上传多个文件
      const type = req.body.type || 'document';
      const files = req.files.map(file => ({
        buffer: file.buffer,
        originalName: file.originalname,
        mimeType: file.mimetype
      }));

      const result = await minioService.uploadMultipleFiles(files, type);

      // 如果只上传了一个文件，返回单文件格式以保持兼容性
      if (result.data.files.length === 1) {
        result.data.files.forEach(element => {
          element.path = '/' + element.path
        });
        res.json({
          success: true,
          message: '文件上传成功',
          data: {
            ...result.data.files[0],
            urls: result.data.urls // 同时提供urls数组格式
          }
        });
      } else {
        // 多文件返回数组格式
        res.json({
          success: true,
          message: `成功上传${result.data.files.length}个文件`,
          data: result.data
        });
      }
    } catch (error) {
      console.error('文件上传到MinIO失败:', error);
      res.status(500).json({
        success: false,
        message: error.message || '文件上传失败'
      });
    }
  });
});

router.post('/single', auth, (req, res) => {
  upload.single('file')(req, res, async (err) => {
    if (err instanceof multer.MulterError) {
      return res.status(400).json({
        success: false,
        message: '文件上传失败：' + err.message
      });
    } else if (err) {
      return res.status(400).json({
        success: false,
        message: err.message
      });
    }

    if (!req.file) {
      return res.status(400).json({ success: false, message: '没有上传文件' });
    }

    try {
      // 使用MinIO服务上传文件
      const type = req.body.type || 'document';
      const result = await minioService.uploadFile(
        req.file.buffer,
        req.file.originalname,
        req.file.mimetype,
        type
      );

      res.json({
        success: true,
        message: '文件上传成功',
        url: result.data.url,
        filename: result.data.filename,
        originalname: result.data.originalname,
        size: result.data.size
      });
    } catch (error) {
      console.error('文件上传到MinIO失败:', error);
      res.status(500).json({
        success: false,
        message: error.message || '文件上传失败'
      });
    }
  });
});

// 错误处理中间件
router.use((error, req, res, next) => {
  if (error instanceof multer.MulterError) {
    if (error.code === 'LIMIT_FILE_SIZE') {
      return res.status(400).json({ success: false, message: '文件大小不能超过10MB' });
    }
    if (error.code === 'LIMIT_FILE_COUNT') {
      return res.status(400).json({ success: false, message: '一次最多只能上传10个文件' });
    }
  }

  if (error.message === '只能上传图片或视频文件') {
    return res.status(400).json({ success: false, message: error.message });
  }

  res.status(500).json({ success: false, message: '上传失败' });
});

module.exports = router;