ai-test-seam.controller.ts

ai-test-seam.controller.tsapps/api/src/modules/internal/ai-test-seam.controller.tsraw ← internal4,535 B← Back to the app
1import { Body, Controller, HttpCode, Inject, Param, Post, UseGuards } from "@nestjs/common";
2import { ZodValidate } from "../../common/pipes/zod-validation.pipe.js";
3import { InternalServiceGuard } from "../../common/guards/internal-service.guard.js";
4import { AiTestSeamService } from "./ai-test-seam.service.js";
5import { AiUsageLimitsDto, MockAssistantDto, MockDbInvalidDto, MockFailDto, MockGenerateSeamDto, MockUnknownBlockIdDto, type AiUsageLimitsInput, type MockAssistantInput, type MockDbInvalidInput, type MockFailInput, type MockGenerateSeamInput, type MockUnknownBlockIdInput } from "./ai-test-seam.dto.js";
6 
7/**
8 * Test-only AI seam controller (TASK-0122).
9 *
10 *  - `POST /internal/workspaces/:workspaceId/ai/mock-fail` —
11 *    `{status?, deltas?}`: arm a deterministic mid-stream outage on the
12 *    next mock chat call FOR THAT WORKSPACE.
13 *  - `POST /internal/workspaces/:workspaceId/ai/usage-limits` —
14 *    `{aiMonthlyTokens?|null}`: set/clear the monthly token budget.
15 *  - `POST /internal/workspaces/:workspaceId/ai/mock-unknown-id` (TASK-0123)
16 *    — `{unknownId}`: arm the next mock WRITING call to emit a forged
17 *    blockId (the unknown-ID-drop validation seam).
18 *
19 * Gating (two independent switches, both fail-closed) — identical to
20 * `InternalTestSeamController`: `InternalServiceGuard` (shared
21 * `INTERNAL_SERVICE_TOKEN`, constant-time compare, user JWTs rejected) AND
22 * `INTERNAL_TEST_SEAM=1` re-checked in the service.
23 *
24 * EXCLUDED from the global `api/v1` prefix (like the other `/internal/*`
25 * routes — see main.ts): the canonical paths are the ones above.
26 */
27@Controller("internal")
28@UseGuards(InternalServiceGuard)
29export class AiTestSeamController {
30  constructor(@Inject(AiTestSeamService) private readonly seam: AiTestSeamService) {}
31 
32  @Post("workspaces/:workspaceId/ai/mock-fail")
33  @HttpCode(200)
34  armMockOutage(
35    @Param("workspaceId") workspaceId: string,
36    @Body(ZodValidate(MockFailDto)) body: MockFailInput
37  ) {
38    return this.seam.armMockOutage(workspaceId, body);
39  }
40 
41  @Post("workspaces/:workspaceId/ai/usage-limits")
42  @HttpCode(200)
43  setAiUsageLimits(
44    @Param("workspaceId") workspaceId: string,
45    @Body(ZodValidate(AiUsageLimitsDto)) body: AiUsageLimitsInput
46  ) {
47    return this.seam.setAiUsageLimits(workspaceId, body);
48  }
49 
50  /**
51   * TASK-0123 — `POST /internal/workspaces/:workspaceId/ai/mock-unknown-id`
52   * `{unknownId}`: arm the NEXT mock WRITING call for this workspace to emit
53   * a forged blockId (the unknown-ID-drop validation seam).
54   */
55  @Post("workspaces/:workspaceId/ai/mock-unknown-id")
56  @HttpCode(200)
57  armMockUnknownBlockId(
58    @Param("workspaceId") workspaceId: string,
59    @Body(ZodValidate(MockUnknownBlockIdDto)) body: MockUnknownBlockIdInput
60  ) {
61    return this.seam.armMockUnknownBlockId(workspaceId, body);
62  }
63 
64  /**
65   * TASK-0124 — `POST /internal/workspaces/:workspaceId/ai/mock-db-invalid`
66   * `{action?}`: arm the NEXT mock DATABASE-GENERATION call to emit a
67   * deliberately invalid tool call (the "retry-once → 422, no partial write"
68   * seam).
69   */
70  @Post("workspaces/:workspaceId/ai/mock-db-invalid")
71  @HttpCode(200)
72  armMockDbInvalid(
73    @Param("workspaceId") workspaceId: string,
74    @Body(ZodValidate(MockDbInvalidDto)) body: MockDbInvalidInput
75  ) {
76    return this.seam.armMockDbInvalid(workspaceId, body);
77  }
78 
79  /**
80   * TASK-0126 — `POST /internal/workspaces/:workspaceId/ai/mock-generate`
81   * `{failStep?, slowMs?}`: arm the NEXT mock PAGE-GENERATION call to emit a
82   * failing plan step (the executor's rollback path) and/or fetch the plan
83   * slowly (the service's total-timeout → `ai_timeout` path).
84   */
85  /**
86   * T-172…T-175 — `POST /internal/workspaces/:workspaceId/ai/mock-assistant`
87   * `{hostile?, forgeId?}`: arm the mock ASSISTANT calls for this workspace to
88   * emit a deliberately INVALID tool call (retry-once → 422) and/or a forged
89   * candidate id (the no-fabricated-refs drop path).
90   */
91  @Post("workspaces/:workspaceId/ai/mock-assistant")
92  @HttpCode(200)
93  armMockAssistant(
94    @Param("workspaceId") workspaceId: string,
95    @Body(ZodValidate(MockAssistantDto)) body: MockAssistantInput
96  ) {
97    return this.seam.armMockAssistantSeam(workspaceId, body);
98  }
99 
100  @Post("workspaces/:workspaceId/ai/mock-generate")
101  @HttpCode(200)
102  armMockGenerate(
103    @Param("workspaceId") workspaceId: string,
104    @Body(ZodValidate(MockGenerateSeamDto)) body: MockGenerateSeamInput
105  ) {
106    return this.seam.armMockGenerateSeam(workspaceId, body);
107  }
108}
109