@geekmidas/constructs 0.3.2 → 0.6.0

package/src/endpoints/TestEndpointAdaptor.ts

@@ -1,7 +1,7 @@
 import type {
-  AuditActor,
-  AuditStorage,
-  AuditableAction,
+	AuditActor,
+	AuditableAction,
+	AuditStorage,
 } from '@geekmidas/audit';
 import { DefaultAuditor } from '@geekmidas/audit';
 import { EnvironmentParser } from '@geekmidas/envkit';
@@ -9,392 +9,411 @@ import { UnauthorizedError } from '@geekmidas/errors';
 import type { EventPublisher } from '@geekmidas/events';
 import type { Logger } from '@geekmidas/logger';
 import type {
-  InferComposableStandardSchema,
-  InferStandardSchema,
+	InferComposableStandardSchema,
+	InferStandardSchema,
 } from '@geekmidas/schema';
 import {
-  type Service,
-  ServiceDiscovery,
-  type ServiceRecord,
+	runWithRequestContext,
+	type Service,
+	ServiceDiscovery,
+	type ServiceRecord,
 } from '@geekmidas/services';
 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,
+	type CookieOptions,
+	Endpoint,
+	type EndpointSchemas,
+	ResponseBuilder,
 } from './Endpoint';
-import type { MappedAudit } from './audit';
 import {
-  type AuditExecutionContext,
-  executeWithAuditTransaction,
+	createCookieHeaderAccessor,
+	createObjectHeaders,
+} from './lazyAccessors';
+import {
+	type AuditExecutionContext,
+	executeWithAuditTransaction,
 } from './processAudits';
 
 export type TestHttpResponse<TBody = any> = {
-  body: TBody;
-  status: number;
-  headers: Record<string, string | string[]>;
+	body: TBody;
+	status: number;
+	headers: Record<string, string | string[]>;
 };
 
 /**
  * Serializes a cookie into a Set-Cookie header string
  */
 function serializeCookie(
-  name: string,
-  value: string,
-  options?: CookieOptions,
+	name: string,
+	value: string,
+	options?: CookieOptions,
 ): string {
-  let cookieString = `${name}=${value}`;
-
-  if (options) {
-    if (options.maxAge !== undefined) {
-      cookieString += `; Max-Age=${options.maxAge}`;
-    }
-    if (options.expires) {
-      cookieString += `; Expires=${options.expires.toUTCString()}`;
-    }
-    if (options.domain) {
-      cookieString += `; Domain=${options.domain}`;
-    }
-    if (options.path) {
-      cookieString += `; Path=${options.path}`;
-    }
-    if (options.httpOnly) {
-      cookieString += '; HttpOnly';
-    }
-    if (options.secure) {
-      cookieString += '; Secure';
-    }
-    if (options.sameSite) {
-      cookieString += `; SameSite=${options.sameSite}`;
-    }
-  }
-
-  return cookieString;
+	let cookieString = `${name}=${value}`;
+
+	if (options) {
+		if (options.maxAge !== undefined) {
+			cookieString += `; Max-Age=${options.maxAge}`;
+		}
+		if (options.expires) {
+			cookieString += `; Expires=${options.expires.toUTCString()}`;
+		}
+		if (options.domain) {
+			cookieString += `; Domain=${options.domain}`;
+		}
+		if (options.path) {
+			cookieString += `; Path=${options.path}`;
+		}
+		if (options.httpOnly) {
+			cookieString += '; HttpOnly';
+		}
+		if (options.secure) {
+			cookieString += '; Secure';
+		}
+		if (options.sameSite) {
+			cookieString += `; SameSite=${options.sameSite}`;
+		}
+	}
+
+	return cookieString;
 }
 
 export class TestEndpointAdaptor<
-  TRoute extends string,
-  TMethod extends HttpMethod,
-  TInput extends EndpointSchemas = {},
-  TOutSchema extends StandardSchemaV1 | undefined = undefined,
-  TServices extends Service[] = [],
-  TLogger extends Logger = Logger,
-  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
-  >,
-  TDatabase = undefined,
-  TDatabaseServiceName extends string = string,
+	TRoute extends string,
+	TMethod extends HttpMethod,
+	TInput extends EndpointSchemas = {},
+	TOutSchema extends StandardSchemaV1 | undefined = undefined,
+	TServices extends Service[] = [],
+	TLogger extends Logger = Logger,
+	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
+	>,
+	TDatabase = undefined,
+	TDatabaseServiceName extends string = string,
 > {
-  static getDefaultServiceDiscover<
-    TRoute extends string,
-    TMethod extends HttpMethod,
-    TInput extends EndpointSchemas = {},
-    TOutSchema extends StandardSchemaV1 | undefined = undefined,
-    TServices extends Service[] = [],
-    TLogger extends Logger = Logger,
-    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
-    >,
-    TDatabase = undefined,
-    TDatabaseServiceName extends string = string,
-  >(
-    endpoint: Endpoint<
-      TRoute,
-      TMethod,
-      TInput,
-      TOutSchema,
-      TServices,
-      TLogger,
-      TSession,
-      TEventPublisher,
-      TEventPublisherServiceName,
-      TAuditStorage,
-      TAuditStorageServiceName,
-      TAuditAction,
-      TDatabase,
-      TDatabaseServiceName
-    >,
-  ) {
-    return ServiceDiscovery.getInstance(
-      endpoint.logger,
-      new EnvironmentParser({}),
-    );
-  }
-  constructor(
-    private readonly endpoint: Endpoint<
-      TRoute,
-      TMethod,
-      TInput,
-      TOutSchema,
-      TServices,
-      TLogger,
-      TSession,
-      TEventPublisher,
-      TEventPublisherServiceName,
-      TAuditStorage,
-      TAuditStorageServiceName,
-      TAuditAction,
-      TDatabase,
-      TDatabaseServiceName
-    >,
-    private serviceDiscovery: ServiceDiscovery<
-      any,
-      any
-    > = TestEndpointAdaptor.getDefaultServiceDiscover(endpoint),
-  ) {}
-
-  async fullRequest(
-    ctx: TestRequestAdaptor<
-      TInput,
-      TServices,
-      TEventPublisher,
-      TEventPublisherServiceName,
-      TAuditStorage,
-      TAuditStorageServiceName,
-      TDatabase,
-      TDatabaseServiceName
-    >,
-  ): Promise<TestHttpResponse<InferStandardSchema<TOutSchema>>> {
-    const body = await this.endpoint.parseInput((ctx as any).body, 'body');
-    const query = await this.endpoint.parseInput((ctx as any).query, 'query');
-    const params = await this.endpoint.parseInput(
-      (ctx as any).params,
-      'params',
-    );
-
-    const header = Endpoint.createHeaders(ctx.headers);
-    const cookie = Endpoint.createCookies(ctx.headers.cookie);
-    const logger = this.endpoint.logger.child({
-      route: this.endpoint.route,
-      host: ctx.headers.host,
-      method: this.endpoint.method,
-    }) as TLogger;
-
-    // Get database from context for session extraction
-    const rawDb = (ctx as any).database as TDatabase;
-
-    const session = await this.endpoint.getSession({
-      logger,
-      services: ctx.services,
-      header,
-      cookie,
-      ...(rawDb !== undefined && { db: rawDb }),
-    } as any);
-
-    // Check authorization
-    const isAuthorized = await this.endpoint.authorize({
-      header,
-      cookie,
-      services: ctx.services,
-      logger,
-      session,
-    });
-
-    if (!isAuthorized) {
-      logger.warn('Unauthorized access attempt');
-      throw new UnauthorizedError(
-        'Unauthorized access to the endpoint',
-        'You do not have permission to access this resource.',
-      );
-    }
-
-    // Create audit context if audit storage is provided
-    // The auditorStorage instance is required when endpoint uses .auditor()
-    const auditorStorage = (ctx as any).auditorStorage as TAuditStorage;
-    let auditContext: AuditExecutionContext<TAuditAction> | undefined;
-
-    if (auditorStorage) {
-      // Extract actor if configured
-      let actor: AuditActor = { id: 'system', type: 'system' };
-      if (this.endpoint.actorExtractor) {
-        try {
-          actor = await this.endpoint.actorExtractor({
-            services: ctx.services as any,
-            session,
-            header,
-            cookie,
-            logger,
-          });
-        } catch (error) {
-          logger.error(error as Error, 'Failed to extract actor for audits');
-        }
-      }
-
-      const auditor = new DefaultAuditor<TAuditAction>({
-        actor,
-        storage: auditorStorage as AuditStorage,
-        metadata: {
-          endpoint: this.endpoint.route,
-          method: this.endpoint.method,
-        },
-      });
-
-      auditContext = { auditor, storage: auditorStorage as AuditStorage };
-    }
-
-    // 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');
-    }
-
-    // Execute handler with automatic audit transaction support
-    const result = await executeWithAuditTransaction(
-      auditContext,
-      async (auditor) => {
-        // Use audit transaction as db if available (when storage has same database)
-        // For testing, the tester controls whether to use transactional auditing
-        const trx = auditor?.getTransaction?.();
-        const db = trx ?? rawDb;
-
-        const responseBuilder = new ResponseBuilder();
-        const response = await this.endpoint.handler(
-          {
-            body,
-            query,
-            params,
-            session,
-            services: ctx.services,
-            logger,
-            header,
-            cookie,
-            auditor,
-            db,
-          } 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,
-          });
-        }
-      },
-      // Pass rawDb so storage can reuse existing transactions
-      { db: rawDb },
-    );
-
-    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
-    const headers: Record<string, string | string[]> = {
-      ...(metadata.headers || {}),
-    };
-
-    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),
-        );
-      }
-      headers['set-cookie'] = setCookieValues;
-    }
-
-    // Return HTTP response format
-    return {
-      body: output,
-      status: metadata.status || 200,
-      headers,
-    };
-  }
-
-  async request(
-    ctx: TestRequestAdaptor<
-      TInput,
-      TServices,
-      TEventPublisher,
-      TEventPublisherServiceName,
-      TAuditStorage,
-      TAuditStorageServiceName,
-      TDatabase,
-      TDatabaseServiceName
-    >,
-  ): Promise<InferStandardSchema<TOutSchema>> {
-    const response = await this.fullRequest(ctx);
-    return response.body;
-  }
+	static getDefaultServiceDiscover<
+		TRoute extends string,
+		TMethod extends HttpMethod,
+		TInput extends EndpointSchemas = {},
+		TOutSchema extends StandardSchemaV1 | undefined = undefined,
+		TServices extends Service[] = [],
+		TLogger extends Logger = Logger,
+		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
+		>,
+		TDatabase = undefined,
+		TDatabaseServiceName extends string = string,
+	>(
+		_endpoint: Endpoint<
+			TRoute,
+			TMethod,
+			TInput,
+			TOutSchema,
+			TServices,
+			TLogger,
+			TSession,
+			TEventPublisher,
+			TEventPublisherServiceName,
+			TAuditStorage,
+			TAuditStorageServiceName,
+			TAuditAction,
+			TDatabase,
+			TDatabaseServiceName
+		>,
+	) {
+		return ServiceDiscovery.getInstance(new EnvironmentParser({}));
+	}
+	constructor(
+		private readonly endpoint: Endpoint<
+			TRoute,
+			TMethod,
+			TInput,
+			TOutSchema,
+			TServices,
+			TLogger,
+			TSession,
+			TEventPublisher,
+			TEventPublisherServiceName,
+			TAuditStorage,
+			TAuditStorageServiceName,
+			TAuditAction,
+			TDatabase,
+			TDatabaseServiceName
+		>,
+		private serviceDiscovery: ServiceDiscovery<
+			any,
+			any
+		> = TestEndpointAdaptor.getDefaultServiceDiscover(endpoint),
+	) {}
+
+	async fullRequest(
+		ctx: TestRequestAdaptor<
+			TInput,
+			TServices,
+			TEventPublisher,
+			TEventPublisherServiceName,
+			TAuditStorage,
+			TAuditStorageServiceName,
+			TDatabase,
+			TDatabaseServiceName
+		>,
+	): Promise<TestHttpResponse<InferStandardSchema<TOutSchema>>> {
+		const body = await this.endpoint.parseInput((ctx as any).body, 'body');
+		const query = await this.endpoint.parseInput((ctx as any).query, 'query');
+		const params = await this.endpoint.parseInput(
+			(ctx as any).params,
+			'params',
+		);
+
+		// Lazy accessors - defer parsing until needed
+		const header = createObjectHeaders(ctx.headers);
+		const cookie = createCookieHeaderAccessor(ctx.headers.cookie);
+
+		// Request context setup
+		const startTime = Date.now();
+		const requestId =
+			(ctx as any).requestId ??
+			ctx.headers['x-request-id'] ??
+			crypto.randomUUID();
+
+		const logger = this.endpoint.logger.child({
+			requestId,
+			route: this.endpoint.route,
+			host: ctx.headers.host,
+			method: this.endpoint.method,
+		}) as TLogger;
+
+		// Get database from context for session extraction
+		const rawDb = (ctx as any).database as TDatabase;
+
+		// Wrap handler execution in request context
+		return runWithRequestContext({ logger, requestId, startTime }, async () => {
+			const session = await this.endpoint.getSession({
+				logger,
+				services: ctx.services,
+				header,
+				cookie,
+				...(rawDb !== undefined && { db: rawDb }),
+			} as any);
+
+			// Check authorization
+			const isAuthorized = await this.endpoint.authorize({
+				header,
+				cookie,
+				services: ctx.services,
+				logger,
+				session,
+			});
+
+			if (!isAuthorized) {
+				logger.warn('Unauthorized access attempt');
+				throw new UnauthorizedError(
+					'Unauthorized access to the endpoint',
+					'You do not have permission to access this resource.',
+				);
+			}
+
+			// Create audit context if audit storage is provided
+			// The auditorStorage instance is required when endpoint uses .auditor()
+			const auditorStorage = (ctx as any).auditorStorage as TAuditStorage;
+			let auditContext: AuditExecutionContext<TAuditAction> | undefined;
+
+			if (auditorStorage) {
+				// Extract actor if configured
+				let actor: AuditActor = { id: 'system', type: 'system' };
+				if (this.endpoint.actorExtractor) {
+					try {
+						actor = await this.endpoint.actorExtractor({
+							services: ctx.services as any,
+							session,
+							header,
+							cookie,
+							logger,
+						});
+					} catch (error) {
+						logger.error(error as Error, 'Failed to extract actor for audits');
+					}
+				}
+
+				const auditor = new DefaultAuditor<TAuditAction>({
+					actor,
+					storage: auditorStorage as AuditStorage,
+					metadata: {
+						endpoint: this.endpoint.route,
+						method: this.endpoint.method,
+					},
+				});
+
+				auditContext = { auditor, storage: auditorStorage as AuditStorage };
+			}
+
+			// 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');
+			}
+
+			// Execute handler with automatic audit transaction support
+			const result = await executeWithAuditTransaction(
+				auditContext,
+				async (auditor) => {
+					// Use audit transaction as db if available (when storage has same database)
+					// For testing, the tester controls whether to use transactional auditing
+					const trx = auditor?.getTransaction?.();
+					const db = trx ?? rawDb;
+
+					const responseBuilder = new ResponseBuilder();
+					const response = await this.endpoint.handler(
+						{
+							body,
+							query,
+							params,
+							session,
+							services: ctx.services,
+							logger,
+							header,
+							cookie,
+							auditor,
+							db,
+						} 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,
+						});
+					}
+				},
+				// Pass rawDb so storage can reuse existing transactions
+				{ db: rawDb },
+			);
+
+			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
+			const headers: Record<string, string | string[]> = {
+				...(metadata.headers || {}),
+			};
+
+			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),
+					);
+				}
+				headers['set-cookie'] = setCookieValues;
+			}
+
+			// Return HTTP response format
+			return {
+				body: output,
+				status: metadata.status || 200,
+				headers,
+			};
+		});
+	}
+
+	async request(
+		ctx: TestRequestAdaptor<
+			TInput,
+			TServices,
+			TEventPublisher,
+			TEventPublisherServiceName,
+			TAuditStorage,
+			TAuditStorageServiceName,
+			TDatabase,
+			TDatabaseServiceName
+		>,
+	): Promise<InferStandardSchema<TOutSchema>> {
+		const response = await this.fullRequest(ctx);
+		return response.body;
+	}
 }
 
 /**
  * Conditional audit storage requirement - required when TAuditStorage is configured
  */
 type AuditStorageRequirement<
-  TAuditStorage extends AuditStorage | undefined = undefined,
+	TAuditStorage extends AuditStorage | undefined = undefined,
 > = TAuditStorage extends undefined
-  ? {}
-  : {
-      /** Audit storage instance - required when endpoint uses .auditor() */
-      auditorStorage: TAuditStorage;
-    };
+	? {}
+	: {
+			/** Audit storage instance - required when endpoint uses .auditor() */
+			auditorStorage: TAuditStorage;
+		};
 
 /**
  * Conditional database requirement - required when TDatabase is configured
  */
 type DatabaseRequirement<TDatabase = undefined> = TDatabase extends undefined
-  ? {}
-  : {
-      /** Database instance - required when endpoint uses .database() */
-      database: TDatabase;
-    };
+	? {}
+	: {
+			/** Database instance - required when endpoint uses .database() */
+			database: TDatabase;
+		};
 
 export type TestRequestAdaptor<
-  TInput extends EndpointSchemas = {},
-  TServices extends Service[] = [],
-  TEventPublisher extends EventPublisher<any> | undefined = undefined,
-  TEventPublisherServiceName extends string = string,
-  TAuditStorage extends AuditStorage | undefined = undefined,
-  _TAuditStorageServiceName extends string = string,
-  TDatabase = undefined,
-  _TDatabaseServiceName extends string = string,
+	TInput extends EndpointSchemas = {},
+	TServices extends Service[] = [],
+	TEventPublisher extends EventPublisher<any> | undefined = undefined,
+	TEventPublisherServiceName extends string = string,
+	TAuditStorage extends AuditStorage | undefined = undefined,
+	_TAuditStorageServiceName extends string = string,
+	TDatabase = undefined,
+	_TDatabaseServiceName extends string = string,
 > = {
-  services: ServiceRecord<TServices>;
-  headers: Record<string, string>;
-  publisher?: Service<TEventPublisherServiceName, TEventPublisher>;
+	services: ServiceRecord<TServices>;
+	headers: Record<string, string>;
+	publisher?: Service<TEventPublisherServiceName, TEventPublisher>;
 } & InferComposableStandardSchema<TInput> &
-  AuditStorageRequirement<TAuditStorage> &
-  DatabaseRequirement<TDatabase>;
+	AuditStorageRequirement<TAuditStorage> &
+	DatabaseRequirement<TDatabase>;