selfrelease
69 lines
1.9 KiB
TypeScript
69 lines
1.9 KiB
TypeScript
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);
|
|
}
|
|
}
|