refactor(wizard): update collect endpoint

Author: ycdzj

src/wizard/wizard.controller.ts

@@ -6,6 +6,7 @@ import {
   Req,
   Sse,
   UseInterceptors,
+  ValidationPipe,
 } from '@nestjs/common';
 import { WizardService } from 'omniboxd/wizard/wizard.service';
 import { CollectRequestDto } from 'omniboxd/wizard/dto/collect-request.dto';
@@ -21,6 +22,20 @@ import {
 } from 'omniboxd/decorators/validate-share.decorator';
 import { Share } from 'omniboxd/shares/entities/share.entity';
 
+@Controller('api/v1/wizard')
+export class CollectController {
+  constructor(private readonly wizardService: WizardService) {}
+
+  @Post('collect')
+  async collect(
+    @UserId() userId: string,
+    @Body() data: CollectRequestDto,
+    @Body('namespaceId', new ValidationPipe()) namespaceId: string,
+  ): Promise<CollectResponseDto> {
+    return await this.wizardService.collect(namespaceId, userId, data);
+  }
+}
+
 @Controller('api/v1/namespaces/:namespaceId/wizard')
 export class WizardController {
   constructor(private readonly wizardService: WizardService) {}
@@ -95,4 +110,21 @@ export class SharedWizardController {
       'ask',
     );
   }
+
+  @Post('write')
+  @Sse()
+  @CookieAuth({ onAuthFail: 'continue' })
+  @ValidateShare()
+  async write(
+    @ValidatedShare() share: Share,
+    @RequestId() requestId: string,
+    @Body() body: AgentRequestDto,
+  ) {
+    return await this.wizardService.streamService.createShareAgentStream(
+      share,
+      body,
+      requestId,
+      'write',
+    );
+  }
 }

src/wizard/wizard.e2e-spec.ts

@@ -181,7 +181,9 @@ describe('WizardController (e2e)', () => {
 
       // Note: The RequestId decorator returns undefined if header is missing,
       // but the service might handle this gracefully
-      const response = await client.post(`/api/v1/namespaces/${client.namespace.id}/wizard/ask`).send(askData);
+      const response = await client
+        .post(`/api/v1/namespaces/${client.namespace.id}/wizard/ask`)
+        .send(askData);
 
       // The endpoint might still work without X-Request-Id
       expect([
@@ -337,7 +339,9 @@ describe('WizardController (e2e)', () => {
       // or if the proxy path doesn't exist. In a real test environment,
       // you might want to mock the wizard service.
       const response = await client
-        .post(`/api/v1/namespaces/${client.namespace.id}/wizard/test-proxy-path`)
+        .post(
+          `/api/v1/namespaces/${client.namespace.id}/wizard/test-proxy-path`,
+        )
         .send({ test: 'data' });
 
       // The response status depends on the wizard service implementation