@geekmidas/constructs 0.0.11 → 0.0.12

package/src/endpoints/TestEndpointAdaptor.ts

@@ -1,3 +1,4 @@
+import type { AuditableAction, AuditStorage } from '@geekmidas/audit';
 import { EnvironmentParser } from '@geekmidas/envkit';
 import type { EventPublisher } from '@geekmidas/events';
 import type { Logger } from '@geekmidas/logger';
@@ -13,12 +14,17 @@ import {
 import type { StandardSchemaV1 } from '@standard-schema/spec';
 import { publishConstructEvents } from '../publisher';
 import type { HttpMethod } from '../types';
+import type { MappedAudit } from './audit';
 import {
   type CookieOptions,
   Endpoint,
   type EndpointSchemas,
   ResponseBuilder,
 } from './Endpoint';
+import {
+  createAuditContext,
+  executeWithAuditTransaction,
+} from './processAudits';
 
 export type TestHttpResponse<TBody = any> = {
   body: TBody;
@@ -73,6 +79,12 @@ export class TestEndpointAdaptor<
   TSession = unknown,
   TEventPublisher extends EventPublisher<any> | undefined = undefined,
   TEventPublisherServiceName extends string = string,
+  TAuditStorage extends AuditStorage | undefined = undefined,
+  TAuditStorageServiceName extends string = string,
+  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<
+    string,
+    unknown
+  >,
 > {
   static getDefaultServiceDiscover<
     TRoute extends string,
@@ -84,6 +96,12 @@ export class TestEndpointAdaptor<
     TSession = unknown,
     TEventPublisher extends EventPublisher<any> | undefined = undefined,
     TEventPublisherServiceName extends string = string,
+    TAuditStorage extends AuditStorage | undefined = undefined,
+    TAuditStorageServiceName extends string = string,
+    TAuditAction extends AuditableAction<string, unknown> = AuditableAction<
+      string,
+      unknown
+    >,
   >(
     endpoint: Endpoint<
       TRoute,
@@ -94,7 +112,10 @@ export class TestEndpointAdaptor<
       TLogger,
       TSession,
       TEventPublisher,
-      TEventPublisherServiceName
+      TEventPublisherServiceName,
+      TAuditStorage,
+      TAuditStorageServiceName,
+      TAuditAction
     >,
   ) {
     return ServiceDiscovery.getInstance(
@@ -112,7 +133,10 @@ export class TestEndpointAdaptor<
       TLogger,
       TSession,
       TEventPublisher,
-      TEventPublisherServiceName
+      TEventPublisherServiceName,
+      TAuditStorage,
+      TAuditStorageServiceName,
+      TAuditAction
     >,
     private serviceDiscovery: ServiceDiscovery<
       any,
@@ -149,33 +173,82 @@ export class TestEndpointAdaptor<
       cookie,
     });
 
-    const responseBuilder = new ResponseBuilder();
-    const response = await this.endpoint.handler(
+    // Create audit context if audit storage is configured
+    const auditContext = await createAuditContext(
+      this.endpoint,
+      this.serviceDiscovery,
+      logger,
       {
-        body,
-        query,
-        params,
         session,
-        services: ctx.services,
-        logger,
         header,
         cookie,
-      } as any,
-      responseBuilder,
+        services: ctx.services as Record<string, unknown>,
+      },
     );
 
-    // Check if response has metadata
-    let data = response;
-    let metadata = responseBuilder.getMetadata();
-
-    if (Endpoint.hasMetadata(response)) {
-      data = response.data;
-      metadata = response.metadata;
+    // Warn if declarative audits are configured but no audit storage
+    const audits = this.endpoint.audits as MappedAudit<
+      TAuditAction,
+      TOutSchema
+    >[];
+    if (!auditContext && audits?.length) {
+      logger.warn('No auditor storage service available');
     }
 
-    const output = await this.endpoint.parseOutput(data);
-    ctx.publisher && (await this.serviceDiscovery.register([ctx.publisher]));
+    // Execute handler with automatic audit transaction support
+    const result = await executeWithAuditTransaction(
+      auditContext,
+      async (auditor) => {
+        const responseBuilder = new ResponseBuilder();
+        const response = await this.endpoint.handler(
+          {
+            body,
+            query,
+            params,
+            session,
+            services: ctx.services,
+            logger,
+            header,
+            cookie,
+            auditor,
+          } as any,
+          responseBuilder,
+        );
+
+        // Check if response has metadata
+        let data = response;
+        let metadata = responseBuilder.getMetadata();
 
+        if (Endpoint.hasMetadata(response)) {
+          data = response.data;
+          metadata = response.metadata;
+        }
+
+        const output = await this.endpoint.parseOutput(data);
+
+        return { output, metadata, responseBuilder };
+      },
+      // Process declarative audits after handler (inside transaction)
+      async (result, auditor) => {
+        if (!audits?.length) return;
+
+        for (const audit of audits) {
+          if (audit.when && !audit.when(result.output as any)) {
+            continue;
+          }
+          const payload = audit.payload(result.output as any);
+          const entityId = audit.entityId?.(result.output as any);
+          auditor.audit(audit.type as any, payload as any, {
+            table: audit.table,
+            entityId,
+          });
+        }
+      },
+    );
+
+    const { output, metadata } = result;
+
+    ctx.publisher && (await this.serviceDiscovery.register([ctx.publisher]));
     await publishConstructEvents(this.endpoint, output, this.serviceDiscovery);
 
     // Convert cookies to Set-Cookie headers
@@ -186,7 +259,9 @@ export class TestEndpointAdaptor<
     if (metadata.cookies && metadata.cookies.size > 0) {
       const setCookieValues: string[] = [];
       for (const [name, cookie] of metadata.cookies.entries()) {
-        setCookieValues.push(serializeCookie(name, cookie.value, cookie.options));
+        setCookieValues.push(
+          serializeCookie(name, cookie.value, cookie.options),
+        );
       }
       headers['set-cookie'] = setCookieValues;
     }

package/src/endpoints/__tests__/AmazonApiGatewayV1EndpointAdaptor.events.spec.ts

@@ -69,6 +69,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,
@@ -135,6 +136,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 201,
       getSession: undefined,
       authorize: undefined,
@@ -215,6 +217,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,
@@ -274,6 +277,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,
@@ -324,6 +328,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,
@@ -384,6 +389,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,
@@ -455,6 +461,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 201,
       getSession: undefined,
       authorize: undefined,
@@ -526,6 +533,7 @@ describe('AmazonApiGatewayV1Endpoint Events', () => {
       services: [],
       logger: mockLogger,
       timeout: undefined,
+      memorySize: undefined,
       status: 200,
       getSession: undefined,
       authorize: undefined,

package/src/endpoints/__tests__/AmazonApiGatewayV1EndpointAdaptor.spec.ts

@@ -58,6 +58,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -88,6 +89,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: 201,
         getSession: undefined,
         authorize: undefined,
@@ -129,6 +131,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -163,6 +166,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -194,6 +198,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: 204,
         getSession: undefined,
         authorize: undefined,
@@ -226,6 +231,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -269,6 +275,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -305,6 +312,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [TestService],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -334,6 +342,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: async () => mockSession,
         authorize: undefined,
@@ -366,6 +375,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -402,6 +412,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -458,6 +469,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
           services: [],
           logger: mockLogger,
           timeout: undefined,
+        memorySize: undefined,
           status: undefined,
           getSession: undefined,
           authorize: undefined,
@@ -497,6 +509,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -535,6 +548,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -570,6 +584,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -609,6 +624,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -648,6 +664,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -678,6 +695,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -715,6 +733,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -780,6 +799,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         authorize: undefined,
         description: 'List items with tags',
         getSession: () => ({}),
@@ -837,6 +857,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         authorize: undefined,
         description: 'Search with filters',
         getSession: () => ({}),
@@ -896,6 +917,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -942,6 +964,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -976,6 +999,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -1008,6 +1032,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -1041,6 +1066,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -1078,6 +1104,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -1112,6 +1139,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,
@@ -1143,6 +1171,7 @@ describe('AmazonApiGatewayV1Endpoint', () => {
         services: [],
         logger: mockLogger,
         timeout: undefined,
+        memorySize: undefined,
         status: undefined,
         getSession: undefined,
         authorize: undefined,