R524809
23 lines
2.0 KiB
TypeScript
23 lines
2.0 KiB
TypeScript
import { Body, Controller, Get, Headers, Inject, Post, Req, Res, UseGuards } from '@nestjs/common';
|
|
import { z } from 'zod';
|
|
import { collectedProductPayloadSchema } from '@atlas/contracts';
|
|
import { ApiError } from '../common/api-error.js';
|
|
import { Public } from '../common/public.decorator.js';
|
|
import { ExtensionAccessGuard, type ExtensionAuthenticatedRequest } from './extension-auth.guard.js';
|
|
import { ExtensionService } from './extension.service.js';
|
|
|
|
type ResponseLike = { status: (code: number) => ResponseLike };
|
|
|
|
@Controller('extension')
|
|
export class ExtensionController {
|
|
public constructor(@Inject(ExtensionService) private readonly extension: ExtensionService) {}
|
|
@Get('session') @Public() @UseGuards(ExtensionAccessGuard)
|
|
public session(@Req() request: ExtensionAuthenticatedRequest) { return { data: { authenticated: true, workspace: { id: request.extensionAuth.workspaceId, available: true }, can_collect: true } }; }
|
|
@Post('materials') @Public() @UseGuards(ExtensionAccessGuard)
|
|
public async materials(@Req() request: ExtensionAuthenticatedRequest, @Headers('idempotency-key') key: string | undefined, @Body() body: unknown, @Res({ passthrough: true }) response: ResponseLike) {
|
|
if (!key || !z.uuid().safeParse(key).success) throw new ApiError(400, 'IDEMPOTENCY_KEY_REQUIRED', '请求缺少或格式错误的 Idempotency-Key');
|
|
const result = await this.extension.ingest(request.extensionAuth.workspaceId, request.extensionAuth.userId, key, parse(collectedProductPayloadSchema, body)); response.status(result.status); return result.body;
|
|
}
|
|
}
|
|
function parse<T extends { safeParse: (value: unknown) => { success: boolean; data?: unknown; error?: { issues: Array<{ path: PropertyKey[]; message: string }> } } }>(schema: T, value: unknown) { const result = schema.safeParse(value); if (result.success) return result.data as never; throw new ApiError(422, 'VALIDATION_FAILED', '请求参数不符合要求', result.error?.issues.map((item) => ({ field: item.path.join('.'), code: 'INVALID_FIELD', message: item.message }))); }
|