init: AI培训与智能巡检系统

2026-06-16 00:55:20 +08:00
commit c55598494b
201 changed files with 53131 additions and 0 deletions
@@ -0,0 +1,68 @@
+import {
+  BadRequestException,
+  Body,
+  Controller,
+  Delete,
+  Get,
+  Param,
+  ParseIntPipe,
+  Patch,
+  Post,
+  UploadedFile,
+  UseGuards,
+  UseInterceptors,
+} from '@nestjs/common';
+import { FileInterceptor } from '@nestjs/platform-express';
+import { memoryStorage } from 'multer';
+import { IdentifyService } from './identify.service';
+import { CurrentUser, JwtAuthGuard, type JwtPayload } from '../auth/roles';
+
+const ALLOWED = /\.(jpe?g|png|webp|gif)$/i;
+
+@Controller('api')
+@UseGuards(JwtAuthGuard)
+export class IdentifyController {
+  constructor(private readonly svc: IdentifyService) {}
+
+  /** AI 识车：上传一张照片，调用通义千问视觉模型识别并持久化（命中哈希缓存则复用）。*/
+  @Post('identify')
+  @UseInterceptors(
+    FileInterceptor('file', {
+      storage: memoryStorage(),
+      limits: { fileSize: 8 * 1024 * 1024 },
+      fileFilter: (_req, file, cb) =>
+        ALLOWED.test(file.originalname) || /^image\//.test(file.mimetype)
+          ? cb(null, true)
+          : cb(new BadRequestException('仅支持图片文件'), false),
+    }),
+  )
+  async identify(
+    @UploadedFile() file: Express.Multer.File,
+    @CurrentUser() user: JwtPayload,
+  ) {
+    if (!file) throw new BadRequestException('未收到图片');
+    return this.svc.identifyAndSave(user.sub, file.buffer, file.mimetype);
+  }
+
+  /** 当前用户的识别历史。*/
+  @Get('identifications')
+  list(@CurrentUser() user: JwtPayload) {
+    return this.svc.list(user.sub);
+  }
+
+  /** 修改备注（人工纠正/标注）。*/
+  @Patch('identifications/:id')
+  update(
+    @Param('id', ParseIntPipe) id: number,
+    @Body('note') note: string,
+    @CurrentUser() user: JwtPayload,
+  ) {
+    return this.svc.updateNote(user.sub, id, note ?? '');
+  }
+
+  /** 删除一条识别历史。*/
+  @Delete('identifications/:id')
+  remove(@Param('id', ParseIntPipe) id: number, @CurrentUser() user: JwtPayload) {
+    return this.svc.remove(user.sub, id);
+  }
+}