@geekmidas/constructs 10.0.0-alpha.6 → 10.0.0-alpha.61

package/dist/adaptors/aws.mjs

@@ -1,21 +1,21 @@
-import { __toESM } from "../Construct-BD5neF7w.mjs";
-import "../construct-interface-Z4CpS8tK.mjs";
-import { FunctionBuilder } from "../functions-14lpEFL4.mjs";
-import "../clone-DblgEZZa.mjs";
-import { Endpoint, ResponseBuilder, require_lodash } from "../Endpoint-DCQPMdUX.mjs";
-import { envParserFor } from "../surfaceEnv-C3Ktahea.mjs";
-import { createApiGatewayCookies, publishConstructEvents, publishEvents } from "../lazyAccessors-BoJN3SMd.mjs";
-import { createAuditContext, executeWithAuditTransaction } from "../processAudits-Dmts9H_b.mjs";
+import { o as __toESM } from "../Construct-BO1X-kt9.mjs";
+import { a as publishEvents, i as publishConstructEvents, t as FunctionBuilder } from "../functions-Di6n5ltG.mjs";
+import { i as require_lodash_set, n as ResponseBuilder, t as Endpoint } from "../Endpoint-CMWWrQ7u.mjs";
+import { n as envParserFor } from "../surfaceEnv-wMT5QzgZ.mjs";
+import { t as createApiGatewayCookies } from "../lazyAccessors-CFanua0O.mjs";
+import { n as subscribedEvents, r as subscriberContext, t as runSubscriber } from "../runSubscriber-BYEY_bfA.mjs";
+import { n as executeWithAuditTransaction, t as createAuditContext } from "../processAudits-BpRlPQMW.mjs";
 import { ForbiddenError, UnprocessableEntityError, wrapError } from "@geekmidas/errors";
 import { ServiceDiscovery, runWithRequestContext } from "@geekmidas/services";
-import middy from "@middy/core";
-import { DefaultAuditor } from "@geekmidas/audit";
-import { withAuditableTransaction } from "@geekmidas/audit/kysely";
 import { withRlsContext } from "@geekmidas/db/rls";
 import qs from "qs";
-
+import { DefaultAuditor } from "@geekmidas/audit";
+import middy from "@middy/core";
+import { withAuditableTransaction } from "@geekmidas/audit/kysely";
 //#region src/functions/FunctionExecutionWrapper.ts
 var FunctionExecutionWrapper = class {
+	envParser;
+	fn;
 	constructor(envParser, fn) {
 		this.envParser = envParser;
 		this.fn = fn;
@@ -25,25 +25,22 @@ var FunctionExecutionWrapper = class {
 		return this._logger || this.fn.logger;
 	}
 	get serviceDiscovery() {
-		const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-		return serviceDiscovery;
+		return ServiceDiscovery.getInstance(this.envParser);
 	}
 	getServices() {
 		return this.serviceDiscovery.register(this.fn.services);
 	}
 	async getDatabase() {
-		if (!this.fn.databaseService) return void 0;
-		const services = await this.serviceDiscovery.register([this.fn.databaseService]);
-		return services[this.fn.databaseService.serviceName];
+		if (!this.fn.databaseService) return;
+		return (await this.serviceDiscovery.register([this.fn.databaseService]))[this.fn.databaseService.serviceName];
 	}
 	/**
 	* Get the audit storage service if configured.
 	* Returns undefined if no auditor storage is configured.
 	*/
 	async getAuditStorage() {
-		if (!this.fn.auditorStorageService) return void 0;
-		const services = await this.serviceDiscovery.register([this.fn.auditorStorageService]);
-		return services[this.fn.auditorStorageService.serviceName];
+		if (!this.fn.auditorStorageService) return;
+		return (await this.serviceDiscovery.register([this.fn.auditorStorageService]))[this.fn.auditorStorageService.serviceName];
 	}
 	/**
 	* Create an auditor instance for the function.
@@ -51,7 +48,7 @@ var FunctionExecutionWrapper = class {
 	*/
 	async createAuditor() {
 		const storage = await this.getAuditStorage();
-		if (!storage) return void 0;
+		if (!storage) return;
 		return new DefaultAuditor({
 			actor: {
 				id: "system",
@@ -72,51 +69,44 @@ var FunctionExecutionWrapper = class {
 	async executeWithAudit(handler) {
 		const auditor = await this.createAuditor();
 		const storage = await this.getAuditStorage();
-		if (!auditor || !storage) {
-			const db$1 = await this.getDatabase();
-			return handler({ db: db$1 });
-		}
+		if (!auditor || !storage) return handler({ db: await this.getDatabase() });
 		const storageDb = storage.getDatabase?.();
 		const databaseServiceName = this.fn.databaseService?.serviceName;
 		const auditDbServiceName = this.fn.auditorStorageService?.serviceName;
 		if (storageDb && databaseServiceName && auditDbServiceName) return withAuditableTransaction(storageDb, auditor, async (trx) => {
-			const response$1 = await handler({
+			return await handler({
 				auditor,
 				db: trx
 			});
-			return response$1;
 		});
-		const db = await this.getDatabase();
 		const response = await handler({
 			auditor,
-			db
+			db: await this.getDatabase()
 		});
 		await auditor.flush();
 		return response;
 	}
 	async getFunctionInput(event) {
-		const parsedInput = await FunctionBuilder.parseComposableStandardSchema(event, this.fn.input);
-		return parsedInput;
+		return await FunctionBuilder.parseComposableStandardSchema(event, this.fn.input);
 	}
 	async publishEvents(response) {
-		await publishEvents(this.logger, this.serviceDiscovery, this.fn.events, response, this.fn.publisherService);
+		await publishEvents(this.logger, this.serviceDiscovery, this.fn.events, response);
 	}
 	async parseComposableStandardSchema(data, schema) {
 		return FunctionBuilder.parseComposableStandardSchema(data, schema);
 	}
 };
-
 //#endregion
 //#region src/functions/AWSLambdaFunction.ts
 var AWSLambdaFunction = class extends FunctionExecutionWrapper {
+	fn;
 	constructor(envParser, fn) {
 		super(envParser, fn);
 		this.fn = fn;
 	}
 	error() {
 		return { onError: (req) => {
-			const logger = req.event?.logger || this.fn.logger;
-			logger.error(req.error || {}, "Error processing function");
+			(req.event?.logger || this.fn.logger).error(req.error || {}, "Error processing function");
 			throw wrapError(req.error);
 		} };
 	}
@@ -192,8 +182,7 @@ var AWSLambdaFunction = class extends FunctionExecutionWrapper {
 			db: event.db,
 			auditor: event.auditor
 		});
-		const output = await this.fn.parseOutput(result);
-		return output;
+		return await this.fn.parseOutput(result);
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
@@ -205,13 +194,13 @@ var AWSLambdaFunction = class extends FunctionExecutionWrapper {
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation: context.functionName
 			}, () => chain(event, context));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 //#region src/crons/AWSScheduledFunction.ts
 /**
@@ -245,11 +234,11 @@ var AWSScheduledFunction = class extends AWSLambdaFunction {
 		super(envParser, cron);
 	}
 };
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayEndpointAdaptor.ts
-var import_lodash = __toESM(require_lodash(), 1);
+var import_lodash_set = /* @__PURE__ */ __toESM(require_lodash_set(), 1);
 var AmazonApiGatewayEndpoint = class {
+	endpoint;
 	options;
 	/**
 	* The parser this adaptor discovers services with.
@@ -287,8 +276,7 @@ var AmazonApiGatewayEndpoint = class {
 	static decodeBody(body, isBase64Encoded, contentType) {
 		if (!body) return void 0;
 		const raw = isBase64Encoded ? Buffer.from(body, "base64").toString("utf-8") : body;
-		const isJson = !contentType || contentType.toLowerCase().includes("application/json");
-		if (isJson) return JSON.parse(raw);
+		if (!contentType || contentType.toLowerCase().includes("application/json")) return JSON.parse(raw);
 		return raw;
 	}
 	getCookies(e) {
@@ -302,11 +290,11 @@ var AmazonApiGatewayEndpoint = class {
 				const headers = req.event.headers;
 				const header = Endpoint.createHeaders(headers);
 				const cookie = this.getCookies(req.event);
-				(0, import_lodash.default)(req.event, "body", await this.endpoint.parseInput(body, "body"));
-				(0, import_lodash.default)(req.event, "query", await this.endpoint.parseInput(query, "query"));
-				(0, import_lodash.default)(req.event, "params", await this.endpoint.parseInput(params, "params"));
-				(0, import_lodash.default)(req.event, "header", header);
-				(0, import_lodash.default)(req.event, "cookie", cookie);
+				(0, import_lodash_set.default)(req.event, "body", await this.endpoint.parseInput(body, "body"));
+				(0, import_lodash_set.default)(req.event, "query", await this.endpoint.parseInput(query, "query"));
+				(0, import_lodash_set.default)(req.event, "params", await this.endpoint.parseInput(params, "params"));
+				(0, import_lodash_set.default)(req.event, "header", header);
+				(0, import_lodash_set.default)(req.event, "cookie", cookie);
 			} catch (error) {
 				if (error && typeof error === "object" && Array.isArray(error)) throw new UnprocessableEntityError("Validation failed", error);
 				throw error;
@@ -325,9 +313,8 @@ var AmazonApiGatewayEndpoint = class {
 	}
 	services() {
 		return { before: async (req) => {
-			const _logger = req.event.logger;
-			const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-			const services = await serviceDiscovery.register(this.endpoint.services);
+			req.event.logger;
+			const services = await ServiceDiscovery.getInstance(this.envParser).register(this.endpoint.services);
 			req.event.services = services;
 		} };
 	}
@@ -338,7 +325,7 @@ var AmazonApiGatewayEndpoint = class {
 			const header = req.event.header;
 			const cookie = req.event.cookie;
 			const session = req.event.session;
-			const isAuthorized = await this.endpoint.authorize({
+			if (!await this.endpoint.authorize({
 				header,
 				cookie,
 				services,
@@ -348,8 +335,7 @@ var AmazonApiGatewayEndpoint = class {
 				body: req.event.body,
 				query: req.event.query,
 				params: req.event.params
-			});
-			if (!isAuthorized) {
+			})) {
 				logger.warn("Forbidden access attempt");
 				throw new ForbiddenError("Forbidden access to the endpoint", "You do not have permission to access this resource.");
 			}
@@ -358,9 +344,8 @@ var AmazonApiGatewayEndpoint = class {
 	database() {
 		return { before: async (req) => {
 			if (!this.endpoint.databaseService) return;
-			const _logger = req.event.logger;
-			const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-			const db = await serviceDiscovery.register([this.endpoint.databaseService]).then((s) => s[this.endpoint.databaseService?.serviceName]);
+			req.event.logger;
+			const db = await ServiceDiscovery.getInstance(this.envParser).register([this.endpoint.databaseService]).then((s) => s[this.endpoint.databaseService?.serviceName]);
 			req.event.db = db;
 		} };
 	}
@@ -411,9 +396,8 @@ var AmazonApiGatewayEndpoint = class {
 			cookie: event.cookie,
 			logger
 		}) : void 0;
-		const result = await executeWithAuditTransaction(auditContext, async (auditor) => {
-			const sameDatabase = auditContext?.storage?.databaseServiceName && auditContext.storage.databaseServiceName === this.endpoint.databaseService?.serviceName;
-			const baseDb = sameDatabase ? auditor?.getTransaction?.() ?? rawDb : rawDb;
+		const { output, metadata } = await executeWithAuditTransaction(auditContext, async (auditor) => {
+			const baseDb = auditContext?.storage?.databaseServiceName && auditContext.storage.databaseServiceName === this.endpoint.databaseService?.serviceName ? auditor?.getTransaction?.() ?? rawDb : rawDb;
 			const executeHandler = async (db) => {
 				const responseBuilder = new ResponseBuilder();
 				const response = await this.endpoint.handler({
@@ -427,33 +411,31 @@ var AmazonApiGatewayEndpoint = class {
 					...input
 				}, responseBuilder);
 				let data = response;
-				let metadata$1 = responseBuilder.getMetadata();
+				let metadata = responseBuilder.getMetadata();
 				if (Endpoint.hasMetadata(response)) {
 					data = response.data;
-					metadata$1 = response.metadata;
+					metadata = response.metadata;
 				}
-				const output$1 = this.endpoint.outputSchema ? await this.endpoint.parseOutput(data) : void 0;
 				return {
-					output: output$1,
-					metadata: metadata$1,
+					output: this.endpoint.outputSchema ? await this.endpoint.parseOutput(data) : void 0,
+					metadata,
 					responseBuilder
 				};
 			};
 			if (rlsActive && rlsContext && baseDb) return withRlsContext(baseDb, rlsContext, async (trx) => executeHandler(trx), { prefix: this.endpoint.rlsConfig?.prefix });
 			return executeHandler(baseDb);
-		}, async (result$1, auditor) => {
+		}, async (result, auditor) => {
 			if (!audits?.length) return;
 			for (const audit of audits) {
-				if (audit.when && !audit.when(result$1.output)) continue;
-				const payload = audit.payload(result$1.output);
-				const entityId = audit.entityId?.(result$1.output);
+				if (audit.when && !audit.when(result.output)) continue;
+				const payload = audit.payload(result.output);
+				const entityId = audit.entityId?.(result.output);
 				auditor.audit(audit.type, payload, {
 					table: audit.table,
 					entityId
 				});
 			}
 		}, { db: rawDb });
-		const { output, metadata } = result;
 		const isJsonResponse = this.endpoint.responseType === "application/json";
 		const body = output === void 0 ? void 0 : isJsonResponse ? JSON.stringify(output) : typeof output === "string" ? output : String(output);
 		event.__response = output;
@@ -512,16 +494,17 @@ var AmazonApiGatewayEndpoint = class {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.endpoint.logger.child({ requestId });
+			const operation = `${this.endpoint.method} ${this.endpoint.route}`;
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation
 			}, () => chain(event, context));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 //#region src/endpoints/parseQueryParams.ts
 /**
@@ -542,10 +525,10 @@ function parseQueryParams(queryParams) {
 	}
 	return qs.parse(parts.join("&"));
 }
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayV1EndpointAdaptor.ts
 var AmazonApiGatewayV1Endpoint = class extends AmazonApiGatewayEndpoint {
+	endpoint;
 	getInput(e) {
 		const multiValueParams = e.multiValueQueryStringParameters || {};
 		const singleValueParams = e.queryStringParameters || {};
@@ -578,17 +561,16 @@ var AmazonApiGatewayV1Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayV2EndpointAdaptor.ts
 var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
+	endpoint;
 	getCookies(e) {
 		return createApiGatewayCookies(e.cookies, e.headers?.cookie);
 	}
 	getInput(e) {
-		const raw = AmazonApiGatewayEndpoint.decodeBody(e.body, e.isBase64Encoded, e.headers?.["content-type"]);
 		return {
-			body: raw,
+			body: AmazonApiGatewayEndpoint.decodeBody(e.body, e.isBase64Encoded, e.headers?.["content-type"]),
 			query: qs.parse(e.rawQueryString),
 			params: e.pathParameters || {}
 		};
@@ -613,7 +595,6 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/queue/AWSLambdaQueueAdaptor.ts
 /**
@@ -628,6 +609,8 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 * retries only those records.
 */
 var AWSLambdaQueue = class {
+	envParser;
+	queue;
 	_services;
 	constructor(envParser, queue) {
 		this.envParser = envParser;
@@ -642,6 +625,12 @@ var AWSLambdaQueue = class {
 		this._services = this.queue.services.length > 0 ? await serviceDiscovery.register(this.queue.services) : {};
 		return this._services;
 	}
+	/** The queue's database, registered as functions register theirs. */
+	async getDatabase() {
+		const service = this.queue.databaseService;
+		if (!service) return void 0;
+		return (await ServiceDiscovery.getInstance(this.envParser).register([service]))[service.serviceName];
+	}
 	safeJsonParse(value) {
 		try {
 			return JSON.parse(value);
@@ -667,6 +656,7 @@ var AWSLambdaQueue = class {
 			req: { id: context.awsRequestId }
 		});
 		const services = await this.getServices();
+		const db = await this.getDatabase();
 		const schema = this.queue.messageSchema;
 		const batchItemFailures = [];
 		const messages = [];
@@ -694,7 +684,8 @@ var AWSLambdaQueue = class {
 			await this.queue.handler({
 				messages,
 				services,
-				logger
+				logger,
+				db
 			});
 		} catch (error) {
 			logger.error(wrapError(error), "Queue handler failed; retrying batch");
@@ -704,24 +695,25 @@ var AWSLambdaQueue = class {
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
-		return async (event, context) => {
+		return (async (event, context) => {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.queue.logger.child({ requestId });
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation: `queue ${this.queue.name}`
 			}, () => handler(event, context));
-		};
+		});
 	}
 };
-
 //#endregion
 //#region src/subscribers/AWSLambdaSubscriberAdaptor.ts
 var AWSLambdaSubscriber = class {
+	envParser;
+	subscriber;
 	_logger;
-	_services;
 	constructor(envParser, subscriber) {
 		this.envParser = envParser;
 		this.subscriber = subscriber;
@@ -730,19 +722,9 @@ var AWSLambdaSubscriber = class {
 	get logger() {
 		return this._logger;
 	}
-	async getServices() {
-		if (this._services) return this._services;
-		const serviceDiscovery = ServiceDiscovery.getInstance(this.envParser);
-		if (this.subscriber.services.length > 0) {
-			const registered = await serviceDiscovery.register(this.subscriber.services);
-			this._services = registered;
-		} else this._services = {};
-		return this._services;
-	}
 	error() {
 		return { onError: (req) => {
-			const logger = req.event?.logger || this.subscriber.logger;
-			logger.error(req.error || {}, "Error processing subscriber");
+			(req.event?.logger || this.subscriber.logger).error(req.error || {}, "Error processing subscriber");
 			throw wrapError(req.error);
 		} };
 	}
@@ -761,110 +743,26 @@ var AWSLambdaSubscriber = class {
 	}
 	services() {
 		return { before: async (req) => {
-			req.event.services = await this.getServices();
+			const { services, db } = await subscriberContext(this.subscriber, this.envParser);
+			req.event.services = services;
+			req.event.db = db;
 		} };
 	}
 	parseEvents() {
 		return { before: async (req) => {
-			const { logger,...e } = req.event;
-			const rawEvent = e;
+			const { logger, ...rawEvent } = req.event;
 			logger.info({ rawEvent });
-			const events = [];
-			if ("Records" in rawEvent) {
-				if (this.isSQSEvent(rawEvent)) for (const record of rawEvent.Records) try {
-					const event = this.parseSQSRecord(record);
-					if (this.shouldIncludeEvent(event)) events.push(event);
-				} catch (error) {
-					this.logger.error({
-						error,
-						record
-					}, "Failed to parse SQS record");
-				}
-				else if (this.isSNSEvent(rawEvent)) for (const record of rawEvent.Records) try {
-					const event = this.parseSNSRecord(record);
-					if (this.shouldIncludeEvent(event)) events.push(event);
-				} catch (error) {
-					this.logger.error({
-						error,
-						record
-					}, "Failed to parse SNS record");
-				}
-			}
-			req.event.events = events;
+			req.event.events = subscribedEvents(this.subscriber, rawEvent, this.logger);
 		} };
 	}
-	isSQSEvent(event) {
-		const firstRecord = event.Records[0];
-		return "Records" in event && event.Records.length > 0 && firstRecord !== void 0 && "eventSource" in firstRecord && firstRecord.eventSource === "aws:sqs";
-	}
-	isSNSEvent(event) {
-		const firstRecord = event.Records[0];
-		return "Records" in event && event.Records.length > 0 && firstRecord !== void 0 && "EventSource" in firstRecord && firstRecord.EventSource === "aws:sns";
-	}
-	parseSNSRecord(record) {
-		const message = this.safeJsonParse(record.Sns.Message);
-		const messageType = record.Sns.MessageAttributes?.type?.Value;
-		if (message === null) return messageType ? {
-			type: messageType,
-			payload: record.Sns.Message
-		} : record.Sns.Message;
-		const resolvedType = messageType ?? message.type;
-		if (message.type) return message;
-		return resolvedType ? {
-			type: resolvedType,
-			payload: message
-		} : message;
-	}
-	parseSQSRecord(record) {
-		const body = this.safeJsonParse(record.body);
-		if (body === null) return record.body;
-		if (body.Type === "Notification" && body.Message) {
-			const snsMessage = this.safeJsonParse(body.Message);
-			const messageType = body.MessageAttributes?.type?.Value;
-			if (snsMessage === null) return messageType ? {
-				type: messageType,
-				payload: body.Message
-			} : body.Message;
-			if (snsMessage.type) return snsMessage;
-			const resolvedType = messageType ?? snsMessage.type;
-			return resolvedType ? {
-				type: resolvedType,
-				payload: snsMessage
-			} : snsMessage;
-		}
-		return body;
-	}
-	safeJsonParse(value) {
-		try {
-			return JSON.parse(value);
-		} catch {
-			return null;
-		}
-	}
-	shouldIncludeEvent(event) {
-		if (typeof event !== "object" || !event?.type) return true;
-		if (!this.subscriber.subscribedEvents) return true;
-		return this.subscriber.subscribedEvents.includes(event.type);
-	}
 	async _handler(event) {
-		if (event.events.length === 0) {
-			this.logger.info("No subscribed events to process");
-			return { batchItemFailures: [] };
-		}
-		const result = await this.subscriber.handler({
-			events: event.events,
-			services: event.services,
-			logger: event.logger
+		const { events, services, logger, db } = event;
+		return runSubscriber(this.subscriber, {
+			events,
+			services,
+			logger,
+			db
 		});
-		if (this.subscriber.outputSchema && result) {
-			const validationResult = await this.subscriber.outputSchema["~standard"].validate(result);
-			if (validationResult.issues) {
-				this.logger.error({ issues: validationResult.issues }, "Subscriber output validation failed");
-				throw new Error("Subscriber output validation failed");
-			}
-			return validationResult.value;
-		}
-		return result;
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
@@ -873,16 +771,18 @@ var AWSLambdaSubscriber = class {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.subscriber.logger.child({ requestId });
+			const operation = `subscriber ${this.subscriber.topicName ?? context.functionName}`;
 			return runWithRequestContext({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation
 			}, () => chain(event, context));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 export { AWSLambdaFunction, AWSLambdaQueue, AWSLambdaSubscriber, AWSScheduledFunction, AmazonApiGatewayV1Endpoint, AmazonApiGatewayV2Endpoint };
+
 //# sourceMappingURL=aws.mjs.map