@geekmidas/constructs 10.0.0-alpha.1 → 10.0.0-alpha.100

package/dist/adaptors/aws.cjs

@@ -1,23 +1,249 @@
-const require_chunk = require('../chunk-DWy1uDak.cjs');
-require('../construct-interface-BfjcOD_E.cjs');
-require('../Construct-roId57cU.cjs');
-require('../functions-DukjG8ov.cjs');
-require('../clone-Dy0cR2SZ.cjs');
-const require_Endpoint = require('../Endpoint-BBkV6l0k.cjs');
-const require_surfaceEnv = require('../surfaceEnv-BFXEms2f.cjs');
-const require_publisher = require('../publisher-CqGsXB2X.cjs');
-const require_AWSScheduledFunction = require('../AWSScheduledFunction-qu4ieRZR.cjs');
-const require_processAudits = require('../processAudits-Bc8rz6Ev.cjs');
-const require_lazyAccessors = require('../lazyAccessors-DrJj8NpA.cjs');
-const __geekmidas_errors = require_chunk.__toESM(require("@geekmidas/errors"));
-const __geekmidas_services = require_chunk.__toESM(require("@geekmidas/services"));
-const __middy_core = require_chunk.__toESM(require("@middy/core"));
-const __geekmidas_db_rls = require_chunk.__toESM(require("@geekmidas/db/rls"));
-const qs = require_chunk.__toESM(require("qs"));
-
+Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+const require_rolldown_runtime = require("../rolldown-runtime-y5ARvAq5.cjs");
+const require_tracing = require("../tracing-DPMidF_r.cjs");
+const require_functions = require("../functions-CfzJG9hK.cjs");
+const require_Endpoint = require("../Endpoint-BoUg8ocM.cjs");
+const require_surfaceEnv = require("../surfaceEnv-CNimK76p.cjs");
+const require_lazyAccessors = require("../lazyAccessors-hJKGv8Ay.cjs");
+const require_runSubscriber = require("../runSubscriber-D8nLXIv1.cjs");
+const require_processAudits = require("../processAudits-DasOZjPl.cjs");
+let _geekmidas_events = require("@geekmidas/events");
+let _geekmidas_errors = require("@geekmidas/errors");
+let _geekmidas_services = require("@geekmidas/services");
+let qs = require("qs");
+qs = require_rolldown_runtime.__toESM(qs, 1);
+let _middy_core = require("@middy/core");
+_middy_core = require_rolldown_runtime.__toESM(_middy_core, 1);
+let _geekmidas_audit = require("@geekmidas/audit");
+let _geekmidas_audit_kysely = require("@geekmidas/audit/kysely");
+let _geekmidas_db_rls = require("@geekmidas/db/rls");
+//#region src/functions/FunctionExecutionWrapper.ts
+var FunctionExecutionWrapper = class {
+	envParser;
+	fn;
+	constructor(envParser, fn) {
+		this.envParser = envParser;
+		this.fn = fn;
+	}
+	_logger;
+	get logger() {
+		return this._logger || this.fn.logger;
+	}
+	get serviceDiscovery() {
+		return _geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
+	}
+	getServices() {
+		return this.serviceDiscovery.register(this.fn.services);
+	}
+	async getDatabase() {
+		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;
+		return (await this.serviceDiscovery.register([this.fn.auditorStorageService]))[this.fn.auditorStorageService.serviceName];
+	}
+	/**
+	* Create an auditor instance for the function.
+	* Returns undefined if no auditor storage is configured.
+	*/
+	async createAuditor() {
+		const storage = await this.getAuditStorage();
+		if (!storage) return;
+		return new _geekmidas_audit.DefaultAuditor({
+			actor: {
+				id: "system",
+				type: "system"
+			},
+			storage,
+			metadata: { function: this.fn.type }
+		});
+	}
+	/**
+	* Execute handler with audit transaction support.
+	* If the audit storage has a database (via getDatabase()), wraps execution
+	* in a transaction so audits are atomic with handler's database operations.
+	*
+	* @param handler - The handler function to execute (receives auditor and db)
+	* @returns The handler result
+	*/
+	async executeWithAudit(handler) {
+		const auditor = await this.createAuditor();
+		const storage = await this.getAuditStorage();
+		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 (0, _geekmidas_audit_kysely.withAuditableTransaction)(storageDb, auditor, async (trx) => {
+			return await handler({
+				auditor,
+				db: trx
+			});
+		});
+		const response = await handler({
+			auditor,
+			db: await this.getDatabase()
+		});
+		await auditor.flush();
+		return response;
+	}
+	async getFunctionInput(event) {
+		return await require_functions.FunctionBuilder.parseComposableStandardSchema(event, this.fn.input);
+	}
+	async publishEvents(response) {
+		await require_functions.publishEvents(this.logger, this.serviceDiscovery, this.fn.events, response);
+	}
+	async parseComposableStandardSchema(data, schema) {
+		return require_functions.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) => {
+			(req.event?.logger || this.fn.logger).error(req.error || {}, "Error processing function");
+			throw (0, _geekmidas_errors.wrapError)(req.error);
+		} };
+	}
+	baseInput() {
+		return { before: (_req) => {} };
+	}
+	input() {
+		return { before: async (req) => {
+			try {
+				if (this.fn.input) {
+					const parsedInput = await require_functions.FunctionBuilder.parseComposableStandardSchema(req.event, this.fn.input);
+					req.event.parsedInput = parsedInput;
+				} else req.event.parsedInput = req.event;
+			} catch (error) {
+				this.logger.error({
+					err: error,
+					event: req.event
+				}, "Failed to parse input");
+				throw error;
+			}
+		} };
+	}
+	loggerMiddleware() {
+		return { before: (req) => {
+			this._logger = this.fn.logger.child({
+				fn: {
+					name: req.context.functionName,
+					version: req.context.functionVersion,
+					memory: req.context.memoryLimitInMB
+				},
+				req: { id: req.context.awsRequestId }
+			});
+			req.event.logger = this._logger;
+		} };
+	}
+	services() {
+		return { before: async (req) => {
+			req.event.services = await this.getServices();
+		} };
+	}
+	database() {
+		return { before: async (req) => {
+			req.event.db = await this.getDatabase();
+		} };
+	}
+	auditor() {
+		return {
+			before: async (req) => {
+				req.event.auditor = await this.createAuditor();
+			},
+			after: async (req) => {
+				if (req.event.auditor) {
+					const records = req.event.auditor.getRecords();
+					if (records.length > 0) {
+						this.logger.debug({ auditCount: records.length }, "Flushing function audits");
+						await req.event.auditor.flush();
+					}
+				}
+			}
+		};
+	}
+	events() {
+		return { after: async (req) => {
+			const response = req.response || void 0;
+			await this.publishEvents(response);
+		} };
+	}
+	async _handler(event) {
+		const result = await this.fn.fn({
+			input: event.parsedInput,
+			services: event.services,
+			logger: event.logger,
+			db: event.db,
+			auditor: event.auditor
+		});
+		return await this.fn.parseOutput(result);
+	}
+	get handler() {
+		const handler = this._handler.bind(this);
+		const chain = (0, _middy_core.default)(handler).use(this.loggerMiddleware()).use(this.baseInput()).use(this.error()).use(this.services()).use(this.database()).use(this.auditor()).use(this.input()).use(this.events());
+		const wrappedHandler = async (event, context) => {
+			const startTime = Date.now();
+			const requestId = context.awsRequestId;
+			const logger = this.fn.logger.child({ requestId });
+			return (0, _geekmidas_services.runWithRequestContext)({
+				logger,
+				requestId,
+				startTime,
+				operation: context.functionName
+			}, () => chain(event, context));
+		};
+		return wrappedHandler;
+	}
+};
+//#endregion
+//#region src/crons/AWSScheduledFunction.ts
+/**
+* AWS Lambda adaptor for {@link Cron} constructs.
+*
+* A `Cron` is a {@link import('../functions').Function} that is invoked on a
+* schedule (an EventBridge rule) rather than by an API request. Because of
+* that, the runtime execution pipeline is identical to a regular Lambda
+* function: input parsing, service/database/auditor wiring, event publishing,
+* error handling, and the `runWithRequestContext` wrapper that powers
+* request-scoped logging via `serviceContext.getLogger()`.
+*
+* This adaptor therefore reuses {@link AWSLambdaFunction} wholesale and only
+* narrows the constructor to accept a `Cron`. The schedule expression itself
+* (`cron.schedule`) is **deploy-time** infrastructure consumed by the CLI to
+* provision the EventBridge rule — it is not needed at runtime.
+*
+* @example
+* ```typescript
+* import { AWSScheduledFunction } from '@geekmidas/constructs/crons';
+* import { cancelPastBookings } from '../../../src/crons/cancelPastBookings.js';
+* import { envParser } from '../../../src/env';
+*
+* const adapter = new AWSScheduledFunction(envParser, cancelPastBookings);
+*
+* export const handler = adapter.handler;
+* ```
+*/
+var AWSScheduledFunction = class extends AWSLambdaFunction {
+	constructor(envParser, cron) {
+		super(envParser, cron);
+	}
+};
+//#endregion
 //#region src/endpoints/AmazonApiGatewayEndpointAdaptor.ts
-var import_lodash = require_chunk.__toESM(require_Endpoint.require_lodash(), 1);
+var import_lodash_set = /* @__PURE__ */ require_rolldown_runtime.__toESM(require_Endpoint.require_lodash_set(), 1);
 var AmazonApiGatewayEndpoint = class {
+	endpoint;
 	options;
 	/**
 	* The parser this adaptor discovers services with.
@@ -37,7 +263,7 @@ var AmazonApiGatewayEndpoint = class {
 	error() {
 		return { onError: (req) => {
 			(req.event.logger || this.endpoint.logger).error(req.error || {}, "Error processing request");
-			const wrappedError = (0, __geekmidas_errors.wrapError)(req.error);
+			const wrappedError = (0, _geekmidas_errors.wrapError)(req.error);
 			req.response = {
 				statusCode: wrappedError.statusCode,
 				body: wrappedError.body
@@ -55,8 +281,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) {
@@ -70,13 +295,13 @@ var AmazonApiGatewayEndpoint = class {
 				const headers = req.event.headers;
 				const header = require_Endpoint.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 __geekmidas_errors.UnprocessableEntityError("Validation failed", error);
+				if (error && typeof error === "object" && Array.isArray(error)) throw new _geekmidas_errors.UnprocessableEntityError("Validation failed", error);
 				throw error;
 			}
 		} };
@@ -93,9 +318,8 @@ var AmazonApiGatewayEndpoint = class {
 	}
 	services() {
 		return { before: async (req) => {
-			const _logger = req.event.logger;
-			const serviceDiscovery = __geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
-			const services = await serviceDiscovery.register(this.endpoint.services);
+			req.event.logger;
+			const services = await _geekmidas_services.ServiceDiscovery.getInstance(this.envParser).register(this.endpoint.services);
 			req.event.services = services;
 		} };
 	}
@@ -106,7 +330,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,
@@ -116,19 +340,17 @@ var AmazonApiGatewayEndpoint = class {
 				body: req.event.body,
 				query: req.event.query,
 				params: req.event.params
-			});
-			if (!isAuthorized) {
+			})) {
 				logger.warn("Forbidden access attempt");
-				throw new __geekmidas_errors.ForbiddenError("Forbidden access to the endpoint", "You do not have permission to access this resource.");
+				throw new _geekmidas_errors.ForbiddenError("Forbidden access to the endpoint", "You do not have permission to access this resource.");
 			}
 		} };
 	}
 	database() {
 		return { before: async (req) => {
 			if (!this.endpoint.databaseService) return;
-			const _logger = req.event.logger;
-			const serviceDiscovery = __geekmidas_services.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 _geekmidas_services.ServiceDiscovery.getInstance(this.envParser).register([this.endpoint.databaseService]).then((s) => s[this.endpoint.databaseService?.serviceName]);
 			req.event.db = db;
 		} };
 	}
@@ -153,15 +375,15 @@ var AmazonApiGatewayEndpoint = class {
 			const statusCode = req.response?.statusCode ?? this.endpoint.status;
 			if (require_Endpoint.Endpoint.isSuccessStatus(statusCode)) {
 				const logger = event.logger;
-				const serviceDiscovery = __geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
-				await require_publisher.publishConstructEvents(this.endpoint, response, serviceDiscovery, logger);
+				const serviceDiscovery = _geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
+				await require_functions.publishConstructEvents(this.endpoint, response, serviceDiscovery, logger);
 			}
 		} };
 	}
 	async _handler(event) {
 		const input = this.endpoint.refineInput(event);
 		const logger = event.logger;
-		const serviceDiscovery = __geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
+		const serviceDiscovery = _geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
 		const auditContext = await require_processAudits.createAuditContext(this.endpoint, serviceDiscovery, logger, {
 			session: event.session,
 			header: event.header,
@@ -179,9 +401,8 @@ var AmazonApiGatewayEndpoint = class {
 			cookie: event.cookie,
 			logger
 		}) : void 0;
-		const result = await require_processAudits.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 require_processAudits.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 require_Endpoint.ResponseBuilder();
 				const response = await this.endpoint.handler({
@@ -195,33 +416,31 @@ var AmazonApiGatewayEndpoint = class {
 					...input
 				}, responseBuilder);
 				let data = response;
-				let metadata$1 = responseBuilder.getMetadata();
+				let metadata = responseBuilder.getMetadata();
 				if (require_Endpoint.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 (0, __geekmidas_db_rls.withRlsContext)(baseDb, rlsContext, async (trx) => executeHandler(trx), { prefix: this.endpoint.rlsConfig?.prefix });
+			if (rlsActive && rlsContext && baseDb) return (0, _geekmidas_db_rls.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;
@@ -272,7 +491,7 @@ var AmazonApiGatewayEndpoint = class {
 	}
 	get handler() {
 		const handler = this._handler.bind(this);
-		let chain = (0, __middy_core.default)(handler).use(this.logger()).use(this.error()).use(this.services()).use(this.input()).use(this.database()).use(this.session()).use(this.authorize()).use(this.events());
+		let chain = (0, _middy_core.default)(handler).use(this.logger()).use(this.error()).use(this.services()).use(this.input()).use(this.database()).use(this.session()).use(this.authorize()).use(this.events());
 		const telemetryMiddleware = this.telemetry();
 		if (telemetryMiddleware) chain = chain.use(telemetryMiddleware);
 		if (this.options.telescope?.middleware) chain = chain.use(this.options.telescope.middleware);
@@ -280,16 +499,17 @@ var AmazonApiGatewayEndpoint = class {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.endpoint.logger.child({ requestId });
-			return (0, __geekmidas_services.runWithRequestContext)({
+			const operation = `${this.endpoint.method} ${this.endpoint.route}`;
+			return (0, _geekmidas_services.runWithRequestContext)({
 				logger,
 				requestId,
-				startTime
+				startTime,
+				operation
 			}, () => chain(event, context));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
 //#region src/endpoints/parseQueryParams.ts
 /**
@@ -310,10 +530,10 @@ function parseQueryParams(queryParams) {
 	}
 	return qs.default.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 || {};
@@ -346,17 +566,16 @@ var AmazonApiGatewayV1Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/endpoints/AmazonApiGatewayV2EndpointAdaptor.ts
 var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
+	endpoint;
 	getCookies(e) {
 		return require_lazyAccessors.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.default.parse(e.rawQueryString),
 			params: e.pathParameters || {}
 		};
@@ -381,7 +600,6 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 		this.endpoint = endpoint;
 	}
 };
-
 //#endregion
 //#region src/queue/AWSLambdaQueueAdaptor.ts
 /**
@@ -396,6 +614,8 @@ var AmazonApiGatewayV2Endpoint = class extends AmazonApiGatewayEndpoint {
 * retries only those records.
 */
 var AWSLambdaQueue = class {
+	envParser;
+	queue;
 	_services;
 	constructor(envParser, queue) {
 		this.envParser = envParser;
@@ -406,10 +626,16 @@ var AWSLambdaQueue = class {
 	}
 	async getServices() {
 		if (this._services) return this._services;
-		const serviceDiscovery = __geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
+		const serviceDiscovery = _geekmidas_services.ServiceDiscovery.getInstance(this.envParser);
 		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 _geekmidas_services.ServiceDiscovery.getInstance(this.envParser).register([service]))[service.serviceName];
+	}
 	safeJsonParse(value) {
 		try {
 			return JSON.parse(value);
@@ -435,6 +661,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 = [];
@@ -452,7 +679,7 @@ var AWSLambdaQueue = class {
 			messages.push(validation.value);
 		} catch (error) {
 			logger.error({
-				error,
+				err: error,
 				messageId: record.messageId
 			}, "Failed to parse SQS record");
 			batchItemFailures.push({ itemIdentifier: record.messageId });
@@ -462,34 +689,39 @@ var AWSLambdaQueue = class {
 			await this.queue.handler({
 				messages,
 				services,
-				logger
+				logger,
+				db
 			});
 		} catch (error) {
-			logger.error((0, __geekmidas_errors.wrapError)(error), "Queue handler failed; retrying batch");
+			logger.error((0, _geekmidas_errors.wrapError)(error), "Queue handler failed; retrying batch");
 			for (const record of event.Records) if (!batchItemFailures.some((f) => f.itemIdentifier === record.messageId)) batchItemFailures.push({ itemIdentifier: record.messageId });
 		}
 		return { batchItemFailures };
 	}
 	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 (0, __geekmidas_services.runWithRequestContext)({
+			return (0, _geekmidas_services.runWithRequestContext)({
 				logger,
 				requestId,
-				startTime
-			}, () => handler(event, context));
-		};
+				startTime,
+				operation: `queue ${this.queue.name}`
+			}, () => (0, _geekmidas_events.consumeBatchTraced)({
+				system: "aws_sqs",
+				destination: this.queue.name
+			}, event.Records.map(require_tracing.recordCarrier), () => 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;
@@ -498,20 +730,10 @@ var AWSLambdaSubscriber = class {
 	get logger() {
 		return this._logger;
 	}
-	async getServices() {
-		if (this._services) return this._services;
-		const serviceDiscovery = __geekmidas_services.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");
-			throw (0, __geekmidas_errors.wrapError)(req.error);
+			(req.event?.logger || this.subscriber.logger).error(req.error || {}, "Error processing subscriber");
+			throw (0, _geekmidas_errors.wrapError)(req.error);
 		} };
 	}
 	loggerMiddleware() {
@@ -529,133 +751,55 @@ var AWSLambdaSubscriber = class {
 	}
 	services() {
 		return { before: async (req) => {
-			req.event.services = await this.getServices();
+			const { services, db } = await require_runSubscriber.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 = require_runSubscriber.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 require_runSubscriber.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);
-		const chain = (0, __middy_core.default)(handler).use(this.loggerMiddleware()).use(this.parseEvents()).use(this.error()).use(this.services());
+		const chain = (0, _middy_core.default)(handler).use(this.loggerMiddleware()).use(this.parseEvents()).use(this.error()).use(this.services());
 		const wrappedHandler = async (event, context) => {
 			const startTime = Date.now();
 			const requestId = context.awsRequestId;
 			const logger = this.subscriber.logger.child({ requestId });
-			return (0, __geekmidas_services.runWithRequestContext)({
+			const operation = `subscriber ${this.subscriber.topicName ?? context.functionName}`;
+			const records = event?.Records ?? [];
+			return (0, _geekmidas_services.runWithRequestContext)({
 				logger,
 				requestId,
-				startTime
-			}, () => chain(event, context));
+				startTime,
+				operation
+			}, () => (0, _geekmidas_events.consumeBatchTraced)({
+				system: records[0]?.Sns ? "aws_sns" : "aws_sqs",
+				destination: this.subscriber.topicName ?? context.functionName
+			}, records.map(require_tracing.recordCarrier), () => chain(event, context)));
 		};
 		return wrappedHandler;
 	}
 };
-
 //#endregion
-exports.AWSLambdaFunction = require_AWSScheduledFunction.AWSLambdaFunction;
+exports.AWSLambdaFunction = AWSLambdaFunction;
 exports.AWSLambdaQueue = AWSLambdaQueue;
 exports.AWSLambdaSubscriber = AWSLambdaSubscriber;
-exports.AWSScheduledFunction = require_AWSScheduledFunction.AWSScheduledFunction;
+exports.AWSScheduledFunction = AWSScheduledFunction;
 exports.AmazonApiGatewayV1Endpoint = AmazonApiGatewayV1Endpoint;
 exports.AmazonApiGatewayV2Endpoint = AmazonApiGatewayV2Endpoint;
+
 //# sourceMappingURL=aws.cjs.map