middle_office_system/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();

// 初始化MinIO存储桶
// initializeBuckets().catch(console.error);

/**
 * @swagger
 * tags:
 *   name: Upload
 *   description: 文件上传API
 */

// 配置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个文件
  }
});

/**
 * @swagger
 * /upload/image:
 *   post:
 *     summary: 上传图片
 *     tags: [Upload]
 *     security:
 *       - bearerAuth: []
 *     requestBody:
 *       required: true
 *       content:
 *         multipart/form-data:
 *           schema:
 *             type: object
 *             properties:
 *               file:
 *                 type: string
 *                 format: binary
 *                 description: 要上传的图片文件
 *               type:
 *                 type: string
 *                 enum: [avatar, product, document]
 *                 default: document
 *                 description: 上传文件类型
 *     responses:
 *       200:
 *         description: 图片上传成功
 *         content:
 *           application/json:
 *             schema:
 *               type: object
 *               properties:
 *                 success:
 *                   type: boolean
 *                   example: true
 *                 url:
 *                   type: string
 *                   description: 上传后的文件URL
 *                 filename:
 *                   type: string
 *                   description: 上传后的文件名
 *       400:
 *         description: 请求参数错误
 *       401:
 *         description: 未授权
 *       500:
 *         description: 服务器错误
 */
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 || '文件上传失败'
      });
    }
  });
});

/**
 * @swagger
 * /upload:
 *   post:
 *     summary: 多文件上传接口 (支持MediaUpload组件)
 *     tags: [Upload]
 *     security:
 *       - bearerAuth: []
 *     requestBody:
 *       required: true
 *       content:
 *         multipart/form-data:
 *           schema:
 *             type: object
 *             properties:
 *               files:
 *                 type: array
 *                 items:
 *                   type: string
 *                   format: binary
 *                 description: 要上传的文件列表
 *               type:
 *                 type: string
 *                 enum: [avatar, product, document]
 *                 default: document
 *                 description: 上传文件类型
 *     responses:
 *       200:
 *         description: 文件上传成功
 *         content:
 *           application/json:
 *             schema:
 *               type: object
 *               properties:
 *                 success:
 *                   type: boolean
 *                   example: true
 *                 message:
 *                   type: string
 *                   example: 文件上传成功
 *                 data:
 *                   type: array
 *                   items:
 *                     type: object
 *                     properties:
 *                       filename:
 *                         type: string
 *                       originalname:
 *                         type: string
 *                       mimetype:
 *                         type: string
 *                       size:
 *                         type: integer
 *                       path:
 *                         type: string
 *                       url:
 *                         type: string
 *       400:
 *         description: 请求参数错误
 *       401:
 *         description: 未授权
 *       500:
 *         description: 服务器错误
 * @access Private
 */
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 || '文件上传失败'
      });
    }
  });
});

/**
 * @swagger
 * /upload/single:
 *   post:
 *     summary: 单文件上传接口（兼容性接口）
 *     tags: [Upload]
 *     security:
 *       - bearerAuth: []
 *     requestBody:
 *       required: true
 *       content:
 *         multipart/form-data:
 *           schema:
 *             type: object
 *             properties:
 *               file:
 *                 type: string
 *                 format: binary
 *                 description: 要上传的文件
 *               type:
 *                 type: string
 *                 enum: [avatar, product, document]
 *                 default: document
 *                 description: 上传文件类型
 *     responses:
 *       200:
 *         description: 文件上传成功
 *         content:
 *           application/json:
 *             schema:
 *               type: object
 *               properties:
 *                 success:
 *                   type: boolean
 *                   example: true
 *                 message:
 *                   type: string
 *                   example: 文件上传成功
 *                 url:
 *                   type: string
 *                   description: 上传后的文件URL
 *                 filename:
 *                   type: string
 *                   description: 上传后的文件名
 *                 originalname:
 *                   type: string
 *                   description: 原始文件名
 *                 size:
 *                   type: integer
 *                   description: 文件大小
 *       400:
 *         description: 请求参数错误
 *       401:
 *         description: 未授权
 *       500:
 *         description: 服务器错误
 */
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;