mini-starter/packages/file-module/src/routes/multipart-complete/post.ts

import { createRoute, OpenAPIHono, z } from '@hono/zod-openapi';
import { FileService } from '../../services/file.service';
import { ErrorSchema } from '@d8d/shared-utils';
import { AppDataSource } from '@d8d/shared-utils';
import { AuthContext } from '@d8d/shared-types';
import { authMiddleware } from '@d8d/auth-module/middleware';

// 完成分片上传请求Schema
const CompleteMultipartUploadDto = z.object({
  uploadId: z.string().openapi({
    description: '分片上传ID',
    example: '123e4567-e89b-12d3-a456-426614174000'
  }),
  bucket: z.string().openapi({
    description: '存储桶名称',
    example: 'my-bucket'
  }),
  key: z.string().openapi({
    description: '文件键名',
    example: 'documents/report.pdf'
  }),
  parts: z.array(
    z.object({
      partNumber: z.coerce.number().int().positive().openapi({
        description: '分片序号',
        example: 1
      }),
      etag: z.string().openapi({
        description: '分片ETag值',
        example: 'd41d8cd98f00b204e9800998ecf8427e'
      })
    })
  ).openapi({
    description: '分片信息列表',
    example: [
      { partNumber: 1, etag: 'd41d8cd98f00b204e9800998ecf8427e' },
      { partNumber: 2, etag: '5f4dcc3b5aa765d61d8327deb882cf99' }
    ]
  })
});

// 完成分片上传响应Schema
const CompleteMultipartUploadResponse = z.object({
  fileId: z.number().openapi({
    description: '文件ID',
    example: 123456
  }),
  url: z.string().openapi({
    description: '文件访问URL',
    example: 'https://minio.example.com/my-bucket/documents/report.pdf'
  }),
  host: z.string().openapi({
    description: 'MinIO主机地址',
    example: 'minio.example.com'
  }),
  bucket: z.string().openapi({
    description: '存储桶名称',
    example: 'my-bucket'
  }),
  key: z.string().openapi({
    description: '文件键名',
    example: 'documents/report.pdf'
  }),
  size: z.number().openapi({
    description: '文件大小(字节)',
    example: 102400
  })
});

// 创建完成分片上传路由定义
const completeMultipartUploadRoute = createRoute({
  method: 'post',
  path: '/',
  middleware: [authMiddleware],
  request: {
    body: {
      content: {
        'application/json': { schema: CompleteMultipartUploadDto }
      }
    }
  },
  responses: {
    200: {
      description: '完成分片上传成功',
      content: {
        'application/json': { schema: CompleteMultipartUploadResponse }
      }
    },
    401: {
      description: '未授权访问',
      content: { 'application/json': { schema: ErrorSchema } }
    },
    400: {
      description: '请求参数错误',
      content: { 'application/json': { schema: ErrorSchema } }
    },
    500: {
      description: '服务器错误',
      content: { 'application/json': { schema: ErrorSchema } }
    }
  }
});

// 创建路由实例并实现处理逻辑
const app = new OpenAPIHono<AuthContext>().openapi(completeMultipartUploadRoute, async (c) => {
  try {
    const data = await c.req.json();

    // 初始化FileService
    const fileService = new FileService(AppDataSource);
    const result = await fileService.completeMultipartUpload(data);

    // 构建完整的响应包含host和bucket信息
    const response = {
      ...result,
      host: `${process.env.MINIO_USE_SSL ? 'https' : 'http'}://${process.env.MINIO_ENDPOINT}:${process.env.MINIO_PORT}`,
      bucket: data.bucket
    };

    return c.json(response, 200);
  } catch (error) {
    const message = error instanceof Error ? error.message : '完成分片上传失败';
    return c.json({ code: 500, message }, 500);
  }
});

export default app;