@geekmidas/constructs 0.0.12 → 0.0.14

package/dist/Function-DfKsM5Kx.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"Function-DfKsM5Kx.mjs","names":["DEFAULT_LOGGER","defaultServices: TServices","defaultLogger: TLogger","services: S","logger: L","obj: any","schema: T","data: unknown","output: unknown","fn: Fn","type: ConstructType","input?: TInput","outputSchema?: OutSchema","services: TServices","logger: TLogger","publisherService?: Service<TEventPublisherServiceName, TEventPublisher>","events: MappedEvent<TEventPublisher, OutSchema>[]","memorySize?: number","auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>","databaseService?: Service<TDatabaseServiceName, TDatabase>"],"sources":["../src/functions/Function.ts"],"sourcesContent":["import type { AuditStorage, AuditableAction, Auditor } from '@geekmidas/audit';\nimport { UnprocessableEntityError } from '@geekmidas/errors';\nimport type { EventPublisher, MappedEvent } from '@geekmidas/events';\nimport type { Logger } from '@geekmidas/logger';\nimport { ConsoleLogger } from '@geekmidas/logger/console';\nimport type {\n  ComposableStandardSchema,\n  InferComposableStandardSchema,\n  InferStandardSchema,\n} from '@geekmidas/schema';\nimport { parseSchema } from '@geekmidas/schema/parser';\nimport type { Service, ServiceRecord } from '@geekmidas/services';\nimport type { StandardSchemaV1 } from '@standard-schema/spec';\nimport uniqBy from 'lodash.uniqby';\nimport { Construct, ConstructType } from '../Construct';\n\nconst DEFAULT_LOGGER = new ConsoleLogger() as any;\n\nexport class FunctionFactory<\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n> {\n  private defaultServices: TServices;\n  constructor(\n    defaultServices: TServices,\n    private defaultLogger: TLogger = DEFAULT_LOGGER,\n  ) {\n    // Initialize default services\n    this.defaultServices = uniqBy(\n      defaultServices,\n      (s) => s.serviceName,\n    ) as TServices;\n  }\n\n  // Create a new factory with services\n  services<S extends Service[]>(\n    services: S,\n  ): FunctionFactory<[...S, ...TServices], TLogger> {\n    return new FunctionFactory<[...S, ...TServices], TLogger>(\n      [...services, ...this.defaultServices],\n      this.defaultLogger,\n    );\n  }\n\n  logger<L extends Logger>(logger: L): FunctionFactory<TServices, L> {\n    return new FunctionFactory<TServices, L>(this.defaultServices, logger);\n  }\n}\n\nexport class Function<\n  TInput extends ComposableStandardSchema | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  OutSchema extends StandardSchemaV1 | undefined = undefined,\n  TEventPublisher extends EventPublisher<any> | undefined = undefined,\n  TEventPublisherServiceName extends string = string,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditStorageServiceName extends string = string,\n  TDatabase = undefined,\n  TDatabaseServiceName extends string = string,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n  Fn extends FunctionHandler<\n    TInput,\n    TServices,\n    TLogger,\n    OutSchema,\n    TDatabase,\n    TAuditStorage,\n    TAuditAction\n  > = FunctionHandler<\n    TInput,\n    TServices,\n    TLogger,\n    OutSchema,\n    TDatabase,\n    TAuditStorage,\n    TAuditAction\n  >,\n> extends Construct<\n  TLogger,\n  TEventPublisherServiceName,\n  TEventPublisher,\n  OutSchema,\n  TServices,\n  TAuditStorageServiceName,\n  TAuditStorage\n> {\n  __IS_FUNCTION__ = true;\n\n  static isFunction(obj: any): obj is Function<any, any, any, any, any> {\n    return Boolean(\n      obj &&\n        obj.__IS_FUNCTION__ === true &&\n        obj.type === ConstructType.Function,\n    );\n  }\n\n  /**\n   * Validates data against a StandardSchema.\n   *\n   * @param schema - The StandardSchema to validate against\n   * @param data - The data to validate\n   * @returns Validation result with value or issues\n   */\n  static validate<T extends StandardSchemaV1>(schema: T, data: unknown) {\n    return schema['~standard'].validate(data);\n  }\n\n  /**\n   * Parses and validates data against a schema, throwing an error if validation fails.\n   *\n   * @param schema - The StandardSchema to validate against\n   * @param data - The data to parse and validate\n   * @returns The validated data with proper typing\n   * @throws {UnprocessableEntityError} When validation fails\n   */\n  static async parseSchema<T extends StandardSchemaV1>(\n    schema: T,\n    data: unknown,\n  ): Promise<InferStandardSchema<T>> {\n    try {\n      return await parseSchema(schema, data);\n    } catch (issues) {\n      throw new UnprocessableEntityError('Validation failed', issues);\n    }\n  }\n\n  /**\n   * Parses and validates the endpoint output against the output schema.\n   *\n   * @param output - The raw output data to validate\n   * @returns The validated output data\n   * @throws {UnprocessableEntityError} When output validation fails\n   */\n  async parseOutput(output: unknown): Promise<InferStandardSchema<OutSchema>> {\n    return Function.parseSchema(\n      this.outputSchema as StandardSchemaV1,\n      output,\n    ) as Promise<InferStandardSchema<OutSchema>>;\n  }\n\n  constructor(\n    protected readonly fn: Fn,\n    timeout = 30000, // Default timeout of 30 seconds\n    type: ConstructType = ConstructType.Function,\n    public input?: TInput,\n    outputSchema?: OutSchema,\n    services: TServices = [] as unknown as TServices,\n    logger: TLogger = DEFAULT_LOGGER,\n    publisherService?: Service<TEventPublisherServiceName, TEventPublisher>,\n    events: MappedEvent<TEventPublisher, OutSchema>[] = [],\n    memorySize?: number,\n    auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>,\n    public databaseService?: Service<TDatabaseServiceName, TDatabase>,\n  ) {\n    super(\n      type,\n      logger,\n      services,\n      events,\n      publisherService,\n      outputSchema,\n      timeout,\n      memorySize,\n      auditorStorageService,\n    );\n  }\n}\n\nexport type FunctionHandler<\n  TInput extends ComposableStandardSchema | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  OutSchema extends StandardSchemaV1 | undefined = undefined,\n  TDatabase = undefined,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n> = (\n  ctx: FunctionContext<\n    TInput,\n    TServices,\n    TLogger,\n    TDatabase,\n    TAuditStorage,\n    TAuditAction\n  >,\n) => OutSchema extends StandardSchemaV1\n  ? InferStandardSchema<OutSchema> | Promise<InferStandardSchema<OutSchema>>\n  : any | Promise<any>;\n\n/**\n * Conditional type that adds `db` property only when TDatabase is configured.\n */\ntype DatabaseContext<TDatabase> = TDatabase extends undefined\n  ? {}\n  : { db: TDatabase };\n\n/**\n * Conditional auditor context - only present when audit storage is configured.\n */\ntype AuditorContext<\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n  TAuditStorage = undefined,\n> = TAuditStorage extends undefined\n  ? {}\n  : {\n      /**\n       * Auditor instance for recording audit events.\n       * Only present when audit storage is configured on the function.\n       */\n      auditor: Auditor<TAuditAction>;\n    };\n\nexport type FunctionContext<\n  Input extends ComposableStandardSchema | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  TDatabase = undefined,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n> = {\n  services: ServiceRecord<TServices>;\n  logger: TLogger;\n  input: InferComposableStandardSchema<Input>;\n} & DatabaseContext<TDatabase> &\n  AuditorContext<TAuditAction, TAuditStorage>;\n"],"mappings":";;;;;;;AAgBA,MAAMA,mBAAiB,IAAI;AAE3B,IAAa,kBAAb,MAAa,gBAGX;CACA,AAAQ;CACR,YACEC,iBACQC,gBAAyBF,kBACjC;EADQ;AAGR,OAAK,kBAAkB,OACrB,iBACA,CAAC,MAAM,EAAE,YACV;CACF;CAGD,SACEG,UACgD;AAChD,SAAO,IAAI,gBACT,CAAC,GAAG,UAAU,GAAG,KAAK,eAAgB,GACtC,KAAK;CAER;CAED,OAAyBC,QAA0C;AACjE,SAAO,IAAI,gBAA8B,KAAK,iBAAiB;CAChE;AACF;AAED,IAAa,WAAb,MAAa,iBAgCH,UAQR;CACA,kBAAkB;CAElB,OAAO,WAAWC,KAAoD;AACpE,SAAO,QACL,OACE,IAAI,oBAAoB,QACxB,IAAI,SAAS,cAAc,SAC9B;CACF;;;;;;;;CASD,OAAO,SAAqCC,QAAWC,MAAe;AACpE,SAAO,OAAO,aAAa,SAAS,KAAK;CAC1C;;;;;;;;;CAUD,aAAa,YACXD,QACAC,MACiC;AACjC,MAAI;AACF,UAAO,MAAM,YAAY,QAAQ,KAAK;EACvC,SAAQ,QAAQ;AACf,SAAM,IAAI,yBAAyB,qBAAqB;EACzD;CACF;;;;;;;;CASD,MAAM,YAAYC,QAA0D;AAC1E,SAAO,SAAS,YACd,KAAK,cACL,OACD;CACF;CAED,YACqBC,IACnB,UAAU,KACVC,OAAsB,cAAc,UAC7BC,OACPC,cACAC,WAAsB,CAAE,GACxBC,SAAkBd,kBAClBe,kBACAC,SAAoD,CAAE,GACtDC,YACAC,uBACOC,iBACP;AACA,QACE,MACA,QACA,UACA,QACA,kBACA,cACA,SACA,YACA,sBACD;EAvBkB;EAGZ;EAQA;CAaR;AACF"}

package/dist/{Function-G3JPHMaY.d.mts → Function-V9M9UVHp.d.mts}

@@ -1,6 +1,6 @@
 import { Construct, ConstructType } from "./Construct-ZPqE0vhn.mjs";
 import { Service, ServiceRecord } from "@geekmidas/services";
-import { AuditStorage } from "@geekmidas/audit";
+import { AuditStorage, AuditableAction, Auditor } from "@geekmidas/audit";
 import { EventPublisher, MappedEvent } from "@geekmidas/events";
 import { Logger } from "@geekmidas/logger";
 import { StandardSchemaV1 } from "@standard-schema/spec";
@@ -14,9 +14,10 @@ declare class FunctionFactory<TServices extends Service[] = [], TLogger extends
   services<S extends Service[]>(services: S): FunctionFactory<[...S, ...TServices], TLogger>;
   logger<L extends Logger>(logger: L): FunctionFactory<TServices, L>;
 }
-declare class Function<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, Fn extends FunctionHandler<TInput, TServices, TLogger, OutSchema> = FunctionHandler<TInput, TServices, TLogger, OutSchema>, TEventPublisher extends EventPublisher<any> | undefined = undefined, TEventPublisherServiceName extends string = string, TAuditStorage extends AuditStorage | undefined = undefined, TAuditStorageServiceName extends string = string> extends Construct<TLogger, TEventPublisherServiceName, TEventPublisher, OutSchema, TServices, TAuditStorageServiceName, TAuditStorage> {
+declare class Function<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, 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, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>, Fn extends FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction> = FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>> extends Construct<TLogger, TEventPublisherServiceName, TEventPublisher, OutSchema, TServices, TAuditStorageServiceName, TAuditStorage> {
   protected readonly fn: Fn;
   input?: TInput | undefined;
+  databaseService?: Service<TDatabaseServiceName, TDatabase> | undefined;
   __IS_FUNCTION__: boolean;
   static isFunction(obj: any): obj is Function<any, any, any, any, any>;
   /**
@@ -46,14 +47,30 @@ declare class Function<TInput extends ComposableStandardSchema | undefined = und
   parseOutput(output: unknown): Promise<InferStandardSchema<OutSchema>>;
   constructor(fn: Fn, timeout?: number,
   // Default timeout of 30 seconds
-  type?: ConstructType, input?: TInput | undefined, outputSchema?: OutSchema, services?: TServices, logger?: TLogger, publisherService?: Service<TEventPublisherServiceName, TEventPublisher>, events?: MappedEvent<TEventPublisher, OutSchema>[], memorySize?: number, auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>);
+  type?: ConstructType, input?: TInput | undefined, outputSchema?: OutSchema, services?: TServices, logger?: TLogger, publisherService?: Service<TEventPublisherServiceName, TEventPublisher>, events?: MappedEvent<TEventPublisher, OutSchema>[], memorySize?: number, auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>, databaseService?: Service<TDatabaseServiceName, TDatabase> | undefined);
 }
-type FunctionHandler<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined> = (ctx: FunctionContext<TInput, TServices, TLogger>) => OutSchema extends StandardSchemaV1 ? InferStandardSchema<OutSchema> | Promise<InferStandardSchema<OutSchema>> : any | Promise<any>;
-type FunctionContext<Input extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger> = {
+type FunctionHandler<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, TDatabase = undefined, TAuditStorage extends AuditStorage | undefined = undefined, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> = (ctx: FunctionContext<TInput, TServices, TLogger, TDatabase, TAuditStorage, TAuditAction>) => OutSchema extends StandardSchemaV1 ? InferStandardSchema<OutSchema> | Promise<InferStandardSchema<OutSchema>> : any | Promise<any>;
+/**
+ * Conditional type that adds `db` property only when TDatabase is configured.
+ */
+type DatabaseContext<TDatabase> = TDatabase extends undefined ? {} : {
+  db: TDatabase;
+};
+/**
+ * Conditional auditor context - only present when audit storage is configured.
+ */
+type AuditorContext<TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>, TAuditStorage = undefined> = TAuditStorage extends undefined ? {} : {
+  /**
+   * Auditor instance for recording audit events.
+   * Only present when audit storage is configured on the function.
+   */
+  auditor: Auditor<TAuditAction>;
+};
+type FunctionContext<Input extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, TDatabase = undefined, TAuditStorage extends AuditStorage | undefined = undefined, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> = {
   services: ServiceRecord<TServices>;
   logger: TLogger;
   input: InferComposableStandardSchema<Input>;
-};
+} & DatabaseContext<TDatabase> & AuditorContext<TAuditAction, TAuditStorage>;
 //#endregion
 export { Function, FunctionContext, FunctionFactory, FunctionHandler };
-//# sourceMappingURL=Function-G3JPHMaY.d.mts.map
+//# sourceMappingURL=Function-V9M9UVHp.d.mts.map

package/dist/{Function-6EWabl_X.d.cts → Function-VI1TB3Mh.d.cts}

@@ -1,5 +1,5 @@
 import { Construct, ConstructType } from "./Construct-dI_rgdSp.cjs";
-import { AuditStorage } from "@geekmidas/audit";
+import { AuditStorage, AuditableAction, Auditor } from "@geekmidas/audit";
 import { EventPublisher, MappedEvent } from "@geekmidas/events";
 import { Logger } from "@geekmidas/logger";
 import { Service, ServiceRecord } from "@geekmidas/services";
@@ -14,9 +14,10 @@ declare class FunctionFactory<TServices extends Service[] = [], TLogger extends
   services<S extends Service[]>(services: S): FunctionFactory<[...S, ...TServices], TLogger>;
   logger<L extends Logger>(logger: L): FunctionFactory<TServices, L>;
 }
-declare class Function<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, Fn extends FunctionHandler<TInput, TServices, TLogger, OutSchema> = FunctionHandler<TInput, TServices, TLogger, OutSchema>, TEventPublisher extends EventPublisher<any> | undefined = undefined, TEventPublisherServiceName extends string = string, TAuditStorage extends AuditStorage | undefined = undefined, TAuditStorageServiceName extends string = string> extends Construct<TLogger, TEventPublisherServiceName, TEventPublisher, OutSchema, TServices, TAuditStorageServiceName, TAuditStorage> {
+declare class Function<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, 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, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>, Fn extends FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction> = FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>> extends Construct<TLogger, TEventPublisherServiceName, TEventPublisher, OutSchema, TServices, TAuditStorageServiceName, TAuditStorage> {
   protected readonly fn: Fn;
   input?: TInput | undefined;
+  databaseService?: Service<TDatabaseServiceName, TDatabase> | undefined;
   __IS_FUNCTION__: boolean;
   static isFunction(obj: any): obj is Function<any, any, any, any, any>;
   /**
@@ -46,14 +47,30 @@ declare class Function<TInput extends ComposableStandardSchema | undefined = und
   parseOutput(output: unknown): Promise<InferStandardSchema<OutSchema>>;
   constructor(fn: Fn, timeout?: number,
   // Default timeout of 30 seconds
-  type?: ConstructType, input?: TInput | undefined, outputSchema?: OutSchema, services?: TServices, logger?: TLogger, publisherService?: Service<TEventPublisherServiceName, TEventPublisher>, events?: MappedEvent<TEventPublisher, OutSchema>[], memorySize?: number, auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>);
+  type?: ConstructType, input?: TInput | undefined, outputSchema?: OutSchema, services?: TServices, logger?: TLogger, publisherService?: Service<TEventPublisherServiceName, TEventPublisher>, events?: MappedEvent<TEventPublisher, OutSchema>[], memorySize?: number, auditorStorageService?: Service<TAuditStorageServiceName, TAuditStorage>, databaseService?: Service<TDatabaseServiceName, TDatabase> | undefined);
 }
-type FunctionHandler<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined> = (ctx: FunctionContext<TInput, TServices, TLogger>) => OutSchema extends StandardSchemaV1 ? InferStandardSchema<OutSchema> | Promise<InferStandardSchema<OutSchema>> : any | Promise<any>;
-type FunctionContext<Input extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger> = {
+type FunctionHandler<TInput extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, OutSchema extends StandardSchemaV1 | undefined = undefined, TDatabase = undefined, TAuditStorage extends AuditStorage | undefined = undefined, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> = (ctx: FunctionContext<TInput, TServices, TLogger, TDatabase, TAuditStorage, TAuditAction>) => OutSchema extends StandardSchemaV1 ? InferStandardSchema<OutSchema> | Promise<InferStandardSchema<OutSchema>> : any | Promise<any>;
+/**
+ * Conditional type that adds `db` property only when TDatabase is configured.
+ */
+type DatabaseContext<TDatabase> = TDatabase extends undefined ? {} : {
+  db: TDatabase;
+};
+/**
+ * Conditional auditor context - only present when audit storage is configured.
+ */
+type AuditorContext<TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>, TAuditStorage = undefined> = TAuditStorage extends undefined ? {} : {
+  /**
+   * Auditor instance for recording audit events.
+   * Only present when audit storage is configured on the function.
+   */
+  auditor: Auditor<TAuditAction>;
+};
+type FunctionContext<Input extends ComposableStandardSchema | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, TDatabase = undefined, TAuditStorage extends AuditStorage | undefined = undefined, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> = {
   services: ServiceRecord<TServices>;
   logger: TLogger;
   input: InferComposableStandardSchema<Input>;
-};
+} & DatabaseContext<TDatabase> & AuditorContext<TAuditAction, TAuditStorage>;
 //#endregion
 export { Function, FunctionContext, FunctionFactory, FunctionHandler };
-//# sourceMappingURL=Function-6EWabl_X.d.cts.map
+//# sourceMappingURL=Function-VI1TB3Mh.d.cts.map

package/dist/{FunctionBuilder-CMhLQ4dt.mjs → FunctionBuilder-CVT7bG2o.mjs}

@@ -1,6 +1,6 @@
 import { ConstructType } from "./Construct-LWeB1rSQ.mjs";
-import { Function } from "./Function-COnc-tWM.mjs";
-import { BaseFunctionBuilder } from "./BaseFunctionBuilder-CT7p10K1.mjs";
+import { Function } from "./Function-DfKsM5Kx.mjs";
+import { BaseFunctionBuilder } from "./BaseFunctionBuilder-B5gkW0Kt.mjs";
 import { ConsoleLogger } from "@geekmidas/logger/console";
 import uniqBy from "lodash.uniqby";
 
@@ -44,13 +44,29 @@ var FunctionBuilder = class extends BaseFunctionBuilder {
 		this._auditorStorage = storage;
 		return this;
 	}
+	/**
+	* Set the audit action types for this function.
+	* This provides type-safety for the auditor in the handler context.
+	*/
+	actions() {
+		return this;
+	}
+	/**
+	* Set the database service for this function.
+	* The database will be available in the handler context as `db`.
+	*/
+	database(service) {
+		this._databaseService = service;
+		return this;
+	}
 	handle(fn) {
-		const func = new Function(fn, this._timeout, this.type, this.inputSchema, this.outputSchema, this._services, this._logger, this._publisher, this._events, this._memorySize, this._auditorStorage);
+		const func = new Function(fn, this._timeout, this.type, this.inputSchema, this.outputSchema, this._services, this._logger, this._publisher, this._events, this._memorySize, this._auditorStorage, this._databaseService);
 		this._services = [];
 		this._logger = DEFAULT_LOGGER$1;
 		this._events = [];
 		this._publisher = void 0;
 		this._auditorStorage = void 0;
+		this._databaseService = void 0;
 		this.inputSchema = void 0;
 		this.outputSchema = void 0;
 		this._timeout = void 0;
@@ -61,4 +77,4 @@ var FunctionBuilder = class extends BaseFunctionBuilder {
 
 //#endregion
 export { FunctionBuilder };
-//# sourceMappingURL=FunctionBuilder-CMhLQ4dt.mjs.map
+//# sourceMappingURL=FunctionBuilder-CVT7bG2o.mjs.map

package/dist/FunctionBuilder-CVT7bG2o.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"FunctionBuilder-CVT7bG2o.mjs","names":["DEFAULT_LOGGER","timeout: number","memorySize: number","schema: T","services: T","logger: T","publisher: Service<TName, T>","storage: Service<TName, T>","service: Service<TName, T>","fn: FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >"],"sources":["../src/functions/FunctionBuilder.ts"],"sourcesContent":["import type { AuditStorage, AuditableAction } from '@geekmidas/audit';\nimport type { EventPublisher } from '@geekmidas/events';\nimport type { Logger } from '@geekmidas/logger';\nimport { ConsoleLogger } from '@geekmidas/logger/console';\nimport type { ComposableStandardSchema } from '@geekmidas/schema';\nimport type { Service } from '@geekmidas/services';\nimport type { StandardSchemaV1 } from '@standard-schema/spec';\nimport uniqBy from 'lodash.uniqby';\nimport { ConstructType } from '../Construct';\nimport { BaseFunctionBuilder } from './BaseFunctionBuilder';\nimport { Function, type FunctionHandler } from './Function';\n\nconst DEFAULT_LOGGER = new ConsoleLogger() as any;\n\nexport class FunctionBuilder<\n  TInput extends ComposableStandardSchema,\n  OutSchema extends StandardSchemaV1 | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  TEventPublisher extends EventPublisher<any> | undefined = undefined,\n  TEventPublisherServiceName extends string = string,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditStorageServiceName extends string = string,\n  TDatabase = undefined,\n  TDatabaseServiceName extends string = string,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n> extends BaseFunctionBuilder<\n  TInput,\n  OutSchema,\n  TServices,\n  TLogger,\n  TEventPublisher,\n  TEventPublisherServiceName,\n  TAuditStorage,\n  TAuditStorageServiceName,\n  TDatabase,\n  TDatabaseServiceName\n> {\n  protected _memorySize?: number;\n\n  constructor(public type = ConstructType.Function) {\n    super(type);\n  }\n\n  timeout(timeout: number): this {\n    this._timeout = timeout;\n    return this;\n  }\n\n  memorySize(memorySize: number): this {\n    this._memorySize = memorySize;\n    return this;\n  }\n\n  output<T extends StandardSchemaV1>(\n    schema: T,\n  ): FunctionBuilder<\n    TInput,\n    T,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this.outputSchema = schema as unknown as OutSchema;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      T,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  input<T extends ComposableStandardSchema>(\n    schema: T,\n  ): FunctionBuilder<\n    T,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this.inputSchema = schema as unknown as TInput;\n\n    return this as unknown as FunctionBuilder<\n      T,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  services<T extends Service[]>(\n    services: T,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    [...TServices, ...T],\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._services = uniqBy(\n      [...this._services, ...services],\n      (s) => s.serviceName,\n    ) as TServices;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      [...TServices, ...T],\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  logger<T extends Logger>(\n    logger: T,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    T,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._logger = logger as unknown as TLogger;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      T,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  publisher<T extends EventPublisher<any>, TName extends string>(\n    publisher: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    T,\n    TName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._publisher = publisher as unknown as Service<\n      TEventPublisherServiceName,\n      TEventPublisher\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      T,\n      TName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  auditor<T extends AuditStorage, TName extends string>(\n    storage: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    T,\n    TName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._auditorStorage = storage as unknown as Service<\n      TAuditStorageServiceName,\n      TAuditStorage\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      T,\n      TName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  /**\n   * Set the audit action types for this function.\n   * This provides type-safety for the auditor in the handler context.\n   */\n  actions<T extends AuditableAction<string, unknown>>(): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    T\n  > {\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      T\n    >;\n  }\n\n  /**\n   * Set the database service for this function.\n   * The database will be available in the handler context as `db`.\n   */\n  database<T, TName extends string>(\n    service: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    T,\n    TName,\n    TAuditAction\n  > {\n    this._databaseService = service as unknown as Service<\n      TDatabaseServiceName,\n      TDatabase\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      T,\n      TName,\n      TAuditAction\n    >;\n  }\n\n  handle(\n    fn: FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >,\n  ): Function<\n    TInput,\n    TServices,\n    TLogger,\n    OutSchema,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction,\n    FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >\n  > {\n    const func = new Function(\n      fn,\n      this._timeout,\n      this.type,\n      this.inputSchema,\n      this.outputSchema,\n      this._services,\n      this._logger,\n      this._publisher,\n      this._events,\n      this._memorySize,\n      this._auditorStorage,\n      this._databaseService,\n    );\n\n    // Reset builder state after creating the function to prevent pollution\n    this._services = [] as Service[] as TServices;\n    this._logger = DEFAULT_LOGGER;\n    this._events = [];\n    this._publisher = undefined;\n    this._auditorStorage = undefined;\n    this._databaseService = undefined;\n    this.inputSchema = undefined;\n    this.outputSchema = undefined;\n    this._timeout = undefined;\n    this._memorySize = undefined;\n\n    return func;\n  }\n}\n"],"mappings":";;;;;;;AAYA,MAAMA,mBAAiB,IAAI;AAE3B,IAAa,kBAAb,cAeU,oBAWR;CACA,AAAU;CAEV,YAAmB,OAAO,cAAc,UAAU;AAChD,QAAM,KAAK;EADM;CAElB;CAED,QAAQC,SAAuB;AAC7B,OAAK,WAAW;AAChB,SAAO;CACR;CAED,WAAWC,YAA0B;AACnC,OAAK,cAAc;AACnB,SAAO;CACR;CAED,OACEC,QAaA;AACA,OAAK,eAAe;AAEpB,SAAO;CAaR;CAED,MACEA,QAaA;AACA,OAAK,cAAc;AAEnB,SAAO;CAaR;CAED,SACEC,UAaA;AACA,OAAK,YAAY,OACf,CAAC,GAAG,KAAK,WAAW,GAAG,QAAS,GAChC,CAAC,MAAM,EAAE,YACV;AAED,SAAO;CAaR;CAED,OACEC,QAaA;AACA,OAAK,UAAU;AAEf,SAAO;CAaR;CAED,UACEC,WAaA;AACA,OAAK,aAAa;AAKlB,SAAO;CAaR;CAED,QACEC,SAaA;AACA,OAAK,kBAAkB;AAKvB,SAAO;CAaR;;;;;CAMD,UAYE;AACA,SAAO;CAaR;;;;;CAMD,SACEC,SAaA;AACA,OAAK,mBAAmB;AAKxB,SAAO;CAaR;CAED,OACEC,IA8BA;EACA,MAAM,OAAO,IAAI,SACf,IACA,KAAK,UACL,KAAK,MACL,KAAK,aACL,KAAK,cACL,KAAK,WACL,KAAK,SACL,KAAK,YACL,KAAK,SACL,KAAK,aACL,KAAK,iBACL,KAAK;AAIP,OAAK,YAAY,CAAE;AACnB,OAAK,UAAUT;AACf,OAAK,UAAU,CAAE;AACjB,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AAEL,SAAO;CACR;AACF"}

package/dist/{FunctionBuilder-B3fpp3hA.d.cts → FunctionBuilder-CjVEFTYC.d.cts}

@@ -1,7 +1,7 @@
 import { ConstructType } from "./Construct-dI_rgdSp.cjs";
-import { Function, FunctionHandler } from "./Function-6EWabl_X.cjs";
-import { BaseFunctionBuilder } from "./BaseFunctionBuilder-Cf0op65o.cjs";
-import { AuditStorage } from "@geekmidas/audit";
+import { Function, FunctionHandler } from "./Function-VI1TB3Mh.cjs";
+import { BaseFunctionBuilder } from "./BaseFunctionBuilder-DUZMbEr3.cjs";
+import { AuditStorage, AuditableAction } from "@geekmidas/audit";
 import { EventPublisher } from "@geekmidas/events";
 import { Logger } from "@geekmidas/logger";
 import { Service } from "@geekmidas/services";
@@ -9,20 +9,30 @@ import { StandardSchemaV1 } from "@standard-schema/spec";
 import { ComposableStandardSchema } from "@geekmidas/schema";
 
 //#region src/functions/FunctionBuilder.d.ts
-declare class FunctionBuilder<TInput extends ComposableStandardSchema, OutSchema extends StandardSchemaV1 | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, TEventPublisher extends EventPublisher<any> | undefined = undefined, TEventPublisherServiceName extends string = string, TAuditStorage extends AuditStorage | undefined = undefined, TAuditStorageServiceName extends string = string> extends BaseFunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName> {
+declare class FunctionBuilder<TInput extends ComposableStandardSchema, OutSchema extends StandardSchemaV1 | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, 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, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> extends BaseFunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName> {
   type: ConstructType;
   protected _memorySize?: number;
   constructor(type?: ConstructType);
   timeout(timeout: number): this;
   memorySize(memorySize: number): this;
-  output<T extends StandardSchemaV1>(schema: T): FunctionBuilder<TInput, T, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  input<T extends ComposableStandardSchema>(schema: T): FunctionBuilder<T, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  services<T extends Service[]>(services: T): FunctionBuilder<TInput, OutSchema, [...TServices, ...T], TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  logger<T extends Logger>(logger: T): FunctionBuilder<TInput, OutSchema, TServices, T, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  publisher<T extends EventPublisher<any>, TName extends string>(publisher: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, T, TName, TAuditStorage, TAuditStorageServiceName>;
-  auditor<T extends AuditStorage, TName extends string>(storage: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, T, TName>;
-  handle(fn: FunctionHandler<TInput, TServices, TLogger, OutSchema>): Function<TInput, TServices, TLogger, OutSchema, FunctionHandler<TInput, TServices, TLogger, OutSchema>, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
+  output<T extends StandardSchemaV1>(schema: T): FunctionBuilder<TInput, T, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  input<T extends ComposableStandardSchema>(schema: T): FunctionBuilder<T, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  services<T extends Service[]>(services: T): FunctionBuilder<TInput, OutSchema, [...TServices, ...T], TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  logger<T extends Logger>(logger: T): FunctionBuilder<TInput, OutSchema, TServices, T, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  publisher<T extends EventPublisher<any>, TName extends string>(publisher: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, T, TName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  auditor<T extends AuditStorage, TName extends string>(storage: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, T, TName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  /**
+   * Set the audit action types for this function.
+   * This provides type-safety for the auditor in the handler context.
+   */
+  actions<T extends AuditableAction<string, unknown>>(): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, T>;
+  /**
+   * Set the database service for this function.
+   * The database will be available in the handler context as `db`.
+   */
+  database<T, TName extends string>(service: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, T, TName, TAuditAction>;
+  handle(fn: FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>): Function<TInput, TServices, TLogger, OutSchema, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction, FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>>;
 }
 //#endregion
 export { FunctionBuilder };
-//# sourceMappingURL=FunctionBuilder-B3fpp3hA.d.cts.map
+//# sourceMappingURL=FunctionBuilder-CjVEFTYC.d.cts.map

package/dist/{FunctionBuilder-ByaB_LQ4.d.mts → FunctionBuilder-D1ofSeMd.d.mts}

@@ -1,28 +1,38 @@
 import { ConstructType } from "./Construct-ZPqE0vhn.mjs";
-import { Function, FunctionHandler } from "./Function-G3JPHMaY.mjs";
-import { BaseFunctionBuilder } from "./BaseFunctionBuilder-DilipY1y.mjs";
+import { Function, FunctionHandler } from "./Function-V9M9UVHp.mjs";
+import { BaseFunctionBuilder } from "./BaseFunctionBuilder-CbDnPZpD.mjs";
 import { Service } from "@geekmidas/services";
-import { AuditStorage } from "@geekmidas/audit";
+import { AuditStorage, AuditableAction } from "@geekmidas/audit";
 import { EventPublisher } from "@geekmidas/events";
 import { Logger } from "@geekmidas/logger";
 import { StandardSchemaV1 } from "@standard-schema/spec";
 import { ComposableStandardSchema } from "@geekmidas/schema";
 
 //#region src/functions/FunctionBuilder.d.ts
-declare class FunctionBuilder<TInput extends ComposableStandardSchema, OutSchema extends StandardSchemaV1 | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, TEventPublisher extends EventPublisher<any> | undefined = undefined, TEventPublisherServiceName extends string = string, TAuditStorage extends AuditStorage | undefined = undefined, TAuditStorageServiceName extends string = string> extends BaseFunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName> {
+declare class FunctionBuilder<TInput extends ComposableStandardSchema, OutSchema extends StandardSchemaV1 | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, 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, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> extends BaseFunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName> {
   type: ConstructType;
   protected _memorySize?: number;
   constructor(type?: ConstructType);
   timeout(timeout: number): this;
   memorySize(memorySize: number): this;
-  output<T extends StandardSchemaV1>(schema: T): FunctionBuilder<TInput, T, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  input<T extends ComposableStandardSchema>(schema: T): FunctionBuilder<T, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  services<T extends Service[]>(services: T): FunctionBuilder<TInput, OutSchema, [...TServices, ...T], TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  logger<T extends Logger>(logger: T): FunctionBuilder<TInput, OutSchema, TServices, T, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
-  publisher<T extends EventPublisher<any>, TName extends string>(publisher: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, T, TName, TAuditStorage, TAuditStorageServiceName>;
-  auditor<T extends AuditStorage, TName extends string>(storage: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, T, TName>;
-  handle(fn: FunctionHandler<TInput, TServices, TLogger, OutSchema>): Function<TInput, TServices, TLogger, OutSchema, FunctionHandler<TInput, TServices, TLogger, OutSchema>, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName>;
+  output<T extends StandardSchemaV1>(schema: T): FunctionBuilder<TInput, T, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  input<T extends ComposableStandardSchema>(schema: T): FunctionBuilder<T, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  services<T extends Service[]>(services: T): FunctionBuilder<TInput, OutSchema, [...TServices, ...T], TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  logger<T extends Logger>(logger: T): FunctionBuilder<TInput, OutSchema, TServices, T, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  publisher<T extends EventPublisher<any>, TName extends string>(publisher: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, T, TName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  auditor<T extends AuditStorage, TName extends string>(storage: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, T, TName, TDatabase, TDatabaseServiceName, TAuditAction>;
+  /**
+   * Set the audit action types for this function.
+   * This provides type-safety for the auditor in the handler context.
+   */
+  actions<T extends AuditableAction<string, unknown>>(): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, T>;
+  /**
+   * Set the database service for this function.
+   * The database will be available in the handler context as `db`.
+   */
+  database<T, TName extends string>(service: Service<TName, T>): FunctionBuilder<TInput, OutSchema, TServices, TLogger, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, T, TName, TAuditAction>;
+  handle(fn: FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>): Function<TInput, TServices, TLogger, OutSchema, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction, FunctionHandler<TInput, TServices, TLogger, OutSchema, TDatabase, TAuditStorage, TAuditAction>>;
 }
 //#endregion
 export { FunctionBuilder };
-//# sourceMappingURL=FunctionBuilder-ByaB_LQ4.d.mts.map
+//# sourceMappingURL=FunctionBuilder-D1ofSeMd.d.mts.map

package/dist/{FunctionBuilder-_hMwZUof.cjs → FunctionBuilder-DXvG_XD-.cjs}

@@ -1,7 +1,7 @@
 const require_chunk = require('./chunk-CUT6urMc.cjs');
 const require_Construct = require('./Construct-BYSPikVm.cjs');
-const require_Function = require('./Function-CO-s2pB8.cjs');
-const require_BaseFunctionBuilder = require('./BaseFunctionBuilder-B8rT07QR.cjs');
+const require_Function = require('./Function-DagDbeXo.cjs');
+const require_BaseFunctionBuilder = require('./BaseFunctionBuilder-C5Se7pdL.cjs');
 const __geekmidas_logger_console = require_chunk.__toESM(require("@geekmidas/logger/console"));
 const lodash_uniqby = require_chunk.__toESM(require("lodash.uniqby"));
 
@@ -45,13 +45,29 @@ var FunctionBuilder = class extends require_BaseFunctionBuilder.BaseFunctionBuil
 		this._auditorStorage = storage;
 		return this;
 	}
+	/**
+	* Set the audit action types for this function.
+	* This provides type-safety for the auditor in the handler context.
+	*/
+	actions() {
+		return this;
+	}
+	/**
+	* Set the database service for this function.
+	* The database will be available in the handler context as `db`.
+	*/
+	database(service) {
+		this._databaseService = service;
+		return this;
+	}
 	handle(fn) {
-		const func = new require_Function.Function(fn, this._timeout, this.type, this.inputSchema, this.outputSchema, this._services, this._logger, this._publisher, this._events, this._memorySize, this._auditorStorage);
+		const func = new require_Function.Function(fn, this._timeout, this.type, this.inputSchema, this.outputSchema, this._services, this._logger, this._publisher, this._events, this._memorySize, this._auditorStorage, this._databaseService);
 		this._services = [];
 		this._logger = DEFAULT_LOGGER;
 		this._events = [];
 		this._publisher = void 0;
 		this._auditorStorage = void 0;
+		this._databaseService = void 0;
 		this.inputSchema = void 0;
 		this.outputSchema = void 0;
 		this._timeout = void 0;
@@ -67,4 +83,4 @@ Object.defineProperty(exports, 'FunctionBuilder', {
     return FunctionBuilder;
   }
 });
-//# sourceMappingURL=FunctionBuilder-_hMwZUof.cjs.map
+//# sourceMappingURL=FunctionBuilder-DXvG_XD-.cjs.map

package/dist/FunctionBuilder-DXvG_XD-.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"FunctionBuilder-DXvG_XD-.cjs","names":["ConsoleLogger","BaseFunctionBuilder","ConstructType","timeout: number","memorySize: number","schema: T","services: T","logger: T","publisher: Service<TName, T>","storage: Service<TName, T>","service: Service<TName, T>","fn: FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >","Function"],"sources":["../src/functions/FunctionBuilder.ts"],"sourcesContent":["import type { AuditStorage, AuditableAction } from '@geekmidas/audit';\nimport type { EventPublisher } from '@geekmidas/events';\nimport type { Logger } from '@geekmidas/logger';\nimport { ConsoleLogger } from '@geekmidas/logger/console';\nimport type { ComposableStandardSchema } from '@geekmidas/schema';\nimport type { Service } from '@geekmidas/services';\nimport type { StandardSchemaV1 } from '@standard-schema/spec';\nimport uniqBy from 'lodash.uniqby';\nimport { ConstructType } from '../Construct';\nimport { BaseFunctionBuilder } from './BaseFunctionBuilder';\nimport { Function, type FunctionHandler } from './Function';\n\nconst DEFAULT_LOGGER = new ConsoleLogger() as any;\n\nexport class FunctionBuilder<\n  TInput extends ComposableStandardSchema,\n  OutSchema extends StandardSchemaV1 | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  TEventPublisher extends EventPublisher<any> | undefined = undefined,\n  TEventPublisherServiceName extends string = string,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditStorageServiceName extends string = string,\n  TDatabase = undefined,\n  TDatabaseServiceName extends string = string,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n> extends BaseFunctionBuilder<\n  TInput,\n  OutSchema,\n  TServices,\n  TLogger,\n  TEventPublisher,\n  TEventPublisherServiceName,\n  TAuditStorage,\n  TAuditStorageServiceName,\n  TDatabase,\n  TDatabaseServiceName\n> {\n  protected _memorySize?: number;\n\n  constructor(public type = ConstructType.Function) {\n    super(type);\n  }\n\n  timeout(timeout: number): this {\n    this._timeout = timeout;\n    return this;\n  }\n\n  memorySize(memorySize: number): this {\n    this._memorySize = memorySize;\n    return this;\n  }\n\n  output<T extends StandardSchemaV1>(\n    schema: T,\n  ): FunctionBuilder<\n    TInput,\n    T,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this.outputSchema = schema as unknown as OutSchema;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      T,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  input<T extends ComposableStandardSchema>(\n    schema: T,\n  ): FunctionBuilder<\n    T,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this.inputSchema = schema as unknown as TInput;\n\n    return this as unknown as FunctionBuilder<\n      T,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  services<T extends Service[]>(\n    services: T,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    [...TServices, ...T],\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._services = uniqBy(\n      [...this._services, ...services],\n      (s) => s.serviceName,\n    ) as TServices;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      [...TServices, ...T],\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  logger<T extends Logger>(\n    logger: T,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    T,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._logger = logger as unknown as TLogger;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      T,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  publisher<T extends EventPublisher<any>, TName extends string>(\n    publisher: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    T,\n    TName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._publisher = publisher as unknown as Service<\n      TEventPublisherServiceName,\n      TEventPublisher\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      T,\n      TName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  auditor<T extends AuditStorage, TName extends string>(\n    storage: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    T,\n    TName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction\n  > {\n    this._auditorStorage = storage as unknown as Service<\n      TAuditStorageServiceName,\n      TAuditStorage\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      T,\n      TName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction\n    >;\n  }\n\n  /**\n   * Set the audit action types for this function.\n   * This provides type-safety for the auditor in the handler context.\n   */\n  actions<T extends AuditableAction<string, unknown>>(): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    T\n  > {\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      T\n    >;\n  }\n\n  /**\n   * Set the database service for this function.\n   * The database will be available in the handler context as `db`.\n   */\n  database<T, TName extends string>(\n    service: Service<TName, T>,\n  ): FunctionBuilder<\n    TInput,\n    OutSchema,\n    TServices,\n    TLogger,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    T,\n    TName,\n    TAuditAction\n  > {\n    this._databaseService = service as unknown as Service<\n      TDatabaseServiceName,\n      TDatabase\n    >;\n\n    return this as unknown as FunctionBuilder<\n      TInput,\n      OutSchema,\n      TServices,\n      TLogger,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      T,\n      TName,\n      TAuditAction\n    >;\n  }\n\n  handle(\n    fn: FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >,\n  ): Function<\n    TInput,\n    TServices,\n    TLogger,\n    OutSchema,\n    TEventPublisher,\n    TEventPublisherServiceName,\n    TAuditStorage,\n    TAuditStorageServiceName,\n    TDatabase,\n    TDatabaseServiceName,\n    TAuditAction,\n    FunctionHandler<\n      TInput,\n      TServices,\n      TLogger,\n      OutSchema,\n      TDatabase,\n      TAuditStorage,\n      TAuditAction\n    >\n  > {\n    const func = new Function(\n      fn,\n      this._timeout,\n      this.type,\n      this.inputSchema,\n      this.outputSchema,\n      this._services,\n      this._logger,\n      this._publisher,\n      this._events,\n      this._memorySize,\n      this._auditorStorage,\n      this._databaseService,\n    );\n\n    // Reset builder state after creating the function to prevent pollution\n    this._services = [] as Service[] as TServices;\n    this._logger = DEFAULT_LOGGER;\n    this._events = [];\n    this._publisher = undefined;\n    this._auditorStorage = undefined;\n    this._databaseService = undefined;\n    this.inputSchema = undefined;\n    this.outputSchema = undefined;\n    this._timeout = undefined;\n    this._memorySize = undefined;\n\n    return func;\n  }\n}\n"],"mappings":";;;;;;;;AAYA,MAAM,iBAAiB,IAAIA;AAE3B,IAAa,kBAAb,cAeUC,gDAWR;CACA,AAAU;CAEV,YAAmB,OAAOC,gCAAc,UAAU;AAChD,QAAM,KAAK;EADM;CAElB;CAED,QAAQC,SAAuB;AAC7B,OAAK,WAAW;AAChB,SAAO;CACR;CAED,WAAWC,YAA0B;AACnC,OAAK,cAAc;AACnB,SAAO;CACR;CAED,OACEC,QAaA;AACA,OAAK,eAAe;AAEpB,SAAO;CAaR;CAED,MACEA,QAaA;AACA,OAAK,cAAc;AAEnB,SAAO;CAaR;CAED,SACEC,UAaA;AACA,OAAK,YAAY,2BACf,CAAC,GAAG,KAAK,WAAW,GAAG,QAAS,GAChC,CAAC,MAAM,EAAE,YACV;AAED,SAAO;CAaR;CAED,OACEC,QAaA;AACA,OAAK,UAAU;AAEf,SAAO;CAaR;CAED,UACEC,WAaA;AACA,OAAK,aAAa;AAKlB,SAAO;CAaR;CAED,QACEC,SAaA;AACA,OAAK,kBAAkB;AAKvB,SAAO;CAaR;;;;;CAMD,UAYE;AACA,SAAO;CAaR;;;;;CAMD,SACEC,SAaA;AACA,OAAK,mBAAmB;AAKxB,SAAO;CAaR;CAED,OACEC,IA8BA;EACA,MAAM,OAAO,IAAIC,0BACf,IACA,KAAK,UACL,KAAK,MACL,KAAK,aACL,KAAK,cACL,KAAK,WACL,KAAK,SACL,KAAK,YACL,KAAK,SACL,KAAK,aACL,KAAK,iBACL,KAAK;AAIP,OAAK,YAAY,CAAE;AACnB,OAAK,UAAU;AACf,OAAK,UAAU,CAAE;AACjB,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AACL,OAAK;AAEL,SAAO;CACR;AACF"}

package/dist/FunctionExecutionWrapper-Bubnr0zA.mjs

@@ -0,0 +1,101 @@
+import { publishEvents } from "./publisher-Bw4770Hi.mjs";
+import { FunctionBuilder } from "./FunctionBuilder-CVT7bG2o.mjs";
+import { ServiceDiscovery } from "@geekmidas/services";
+import { DefaultAuditor } from "@geekmidas/audit";
+import { withAuditableTransaction } from "@geekmidas/audit/kysely";
+
+//#region src/functions/FunctionExecutionWrapper.ts
+var FunctionExecutionWrapper = class {
+	constructor(envParser, fn) {
+		this.envParser = envParser;
+		this.fn = fn;
+	}
+	_logger;
+	get logger() {
+		return this._logger || this.fn.logger;
+	}
+	get serviceDiscovery() {
+		const serviceDiscovery = ServiceDiscovery.getInstance(this.logger, this.envParser);
+		return serviceDiscovery;
+	}
+	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];
+	}
+	/**
+	* 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];
+	}
+	/**
+	* 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 void 0;
+		return new 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) {
+			const db$1 = await this.getDatabase();
+			return handler({ db: db$1 });
+		}
+		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({
+				auditor,
+				db: trx
+			});
+			return response$1;
+		});
+		const db = await this.getDatabase();
+		const response = await handler({
+			auditor,
+			db
+		});
+		await auditor.flush();
+		return response;
+	}
+	async getFunctionInput(event) {
+		const parsedInput = await FunctionBuilder.parseComposableStandardSchema(event, this.fn.input);
+		return parsedInput;
+	}
+	async publishEvents(response) {
+		await publishEvents(this.logger, this.serviceDiscovery, this.fn.events, response, this.fn.publisherService);
+	}
+	async parseComposableStandardSchema(data, schema) {
+		return FunctionBuilder.parseComposableStandardSchema(data, schema);
+	}
+};
+
+//#endregion
+export { FunctionExecutionWrapper };
+//# sourceMappingURL=FunctionExecutionWrapper-Bubnr0zA.mjs.map

package/dist/FunctionExecutionWrapper-Bubnr0zA.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"FunctionExecutionWrapper-Bubnr0zA.mjs","names":["envParser: EnvironmentParser<{}>","fn: Function<\n      TInput,\n      TServices,\n      TLogger,\n      TOutSchema,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction,\n      FunctionHandler<\n        TInput,\n        TServices,\n        TLogger,\n        TOutSchema,\n        TDatabase,\n        TAuditStorage,\n        TAuditAction\n      >\n    >","handler: (ctx: {\n      auditor?: Auditor<TAuditAction>;\n      db?: TDatabase;\n    }) => Promise<T>","db","response","event: TEvent","response: InferStandardSchema<TOutSchema>","data: unknown","schema: T"],"sources":["../src/functions/FunctionExecutionWrapper.ts"],"sourcesContent":["import type { AuditStorage, AuditableAction, Auditor } from '@geekmidas/audit';\nimport { DefaultAuditor } from '@geekmidas/audit';\nimport { withAuditableTransaction } from '@geekmidas/audit/kysely';\nimport type { EnvironmentParser } from '@geekmidas/envkit';\nimport type { EventPublisher } from '@geekmidas/events';\nimport type { Logger } from '@geekmidas/logger';\nimport type {\n  ComposableStandardSchema,\n  InferComposableStandardSchema,\n  InferStandardSchema,\n} from '@geekmidas/schema';\nimport {\n  type Service,\n  ServiceDiscovery,\n  type ServiceRecord,\n} from '@geekmidas/services';\nimport type { StandardSchemaV1 } from '@standard-schema/spec';\nimport { publishEvents } from '../publisher';\nimport type { Function, FunctionHandler } from './Function';\nimport { FunctionBuilder } from './FunctionBuilder';\n\nexport abstract class FunctionExecutionWrapper<\n  TInput extends ComposableStandardSchema | undefined = undefined,\n  TOutSchema extends StandardSchemaV1 | undefined = undefined,\n  TServices extends Service[] = [],\n  TLogger extends Logger = Logger,\n  TEventPublisher extends EventPublisher<any> | undefined = undefined,\n  TEventPublisherServiceName extends string = string,\n  TAuditStorage extends AuditStorage | undefined = undefined,\n  TAuditStorageServiceName extends string = string,\n  TDatabase = undefined,\n  TDatabaseServiceName extends string = string,\n  TAuditAction extends AuditableAction<string, unknown> = AuditableAction<\n    string,\n    unknown\n  >,\n> {\n  constructor(\n    protected envParser: EnvironmentParser<{}>,\n    protected readonly fn: Function<\n      TInput,\n      TServices,\n      TLogger,\n      TOutSchema,\n      TEventPublisher,\n      TEventPublisherServiceName,\n      TAuditStorage,\n      TAuditStorageServiceName,\n      TDatabase,\n      TDatabaseServiceName,\n      TAuditAction,\n      FunctionHandler<\n        TInput,\n        TServices,\n        TLogger,\n        TOutSchema,\n        TDatabase,\n        TAuditStorage,\n        TAuditAction\n      >\n    >,\n  ) {}\n\n  protected _logger?: TLogger;\n\n  get logger(): TLogger {\n    return this._logger || this.fn.logger;\n  }\n\n  get serviceDiscovery(): ServiceDiscovery<ServiceRecord<TServices>, Logger> {\n    const serviceDiscovery = ServiceDiscovery.getInstance<\n      ServiceRecord<TServices>,\n      TLogger\n    >(this.logger, this.envParser);\n\n    return serviceDiscovery;\n  }\n\n  getServices(): Promise<ServiceRecord<TServices>> {\n    return this.serviceDiscovery.register(this.fn.services);\n  }\n\n  async getDatabase(): Promise<TDatabase | undefined> {\n    if (!this.fn.databaseService) {\n      return undefined;\n    }\n\n    const services = await this.serviceDiscovery.register([\n      this.fn.databaseService,\n    ]);\n\n    return services[\n      this.fn.databaseService.serviceName as keyof typeof services\n    ] as TDatabase;\n  }\n\n  /**\n   * Get the audit storage service if configured.\n   * Returns undefined if no auditor storage is configured.\n   */\n  async getAuditStorage(): Promise<TAuditStorage | undefined> {\n    if (!this.fn.auditorStorageService) {\n      return undefined;\n    }\n\n    const services = await this.serviceDiscovery.register([\n      this.fn.auditorStorageService,\n    ]);\n\n    return services[\n      this.fn.auditorStorageService.serviceName as keyof typeof services\n    ] as TAuditStorage;\n  }\n\n  /**\n   * Create an auditor instance for the function.\n   * Returns undefined if no auditor storage is configured.\n   */\n  async createAuditor(): Promise<Auditor<TAuditAction> | undefined> {\n    const storage = await this.getAuditStorage();\n    if (!storage) {\n      return undefined;\n    }\n\n    return new DefaultAuditor<TAuditAction>({\n      actor: { id: 'system', type: 'system' },\n      storage: storage as AuditStorage,\n      metadata: {\n        function: this.fn.type,\n      },\n    });\n  }\n\n  /**\n   * Execute handler with audit transaction support.\n   * If the audit storage has a database (via getDatabase()), wraps execution\n   * in a transaction so audits are atomic with handler's database operations.\n   *\n   * @param handler - The handler function to execute (receives auditor and db)\n   * @returns The handler result\n   */\n  async executeWithAudit<T>(\n    handler: (ctx: {\n      auditor?: Auditor<TAuditAction>;\n      db?: TDatabase;\n    }) => Promise<T>,\n  ): Promise<T> {\n    const auditor = await this.createAuditor();\n    const storage = await this.getAuditStorage();\n\n    // No audit context - just run handler with regular db\n    if (!auditor || !storage) {\n      const db = await this.getDatabase();\n      return handler({ db });\n    }\n\n    // Check if storage has a database and db service names match\n    const storageDb = (storage as AuditStorage).getDatabase?.();\n    const databaseServiceName = this.fn.databaseService?.serviceName;\n    const auditDbServiceName = this.fn.auditorStorageService?.serviceName;\n\n    // If the audit storage has a database and we're using the same database service\n    // (or the audit storage provides the database), use transactional execution\n    if (storageDb && databaseServiceName && auditDbServiceName) {\n      return withAuditableTransaction(\n        storageDb as any,\n        auditor as any,\n        async (trx) => {\n          // Use transaction as db\n          const response = await handler({\n            auditor,\n            db: trx as TDatabase,\n          });\n          // Audits are flushed by withAuditableTransaction before commit\n          return response;\n        },\n      );\n    }\n\n    // No database on storage or service names don't match - run handler and flush audits after\n    const db = await this.getDatabase();\n    const response = await handler({ auditor, db });\n\n    // Flush audits (no transaction)\n    await auditor.flush();\n\n    return response;\n  }\n\n  async getFunctionInput<TEvent>(\n    event: TEvent,\n  ): Promise<InferComposableStandardSchema<TInput>> {\n    const parsedInput = await FunctionBuilder.parseComposableStandardSchema(\n      event,\n      this.fn.input,\n    );\n\n    return parsedInput as InferComposableStandardSchema<TInput>;\n  }\n\n  async publishEvents(response: InferStandardSchema<TOutSchema>) {\n    await publishEvents(\n      this.logger,\n      this.serviceDiscovery,\n      this.fn.events,\n      response,\n      this.fn.publisherService,\n    );\n  }\n\n  async parseComposableStandardSchema<T extends ComposableStandardSchema>(\n    data: unknown,\n    schema: T,\n  ): Promise<InferComposableStandardSchema<T>> {\n    return FunctionBuilder.parseComposableStandardSchema(data, schema);\n  }\n}\n"],"mappings":";;;;;;;AAqBA,IAAsB,2BAAtB,MAeE;CACA,YACYA,WACSC,IAsBnB;EAvBU;EACS;CAsBjB;CAEJ,AAAU;CAEV,IAAI,SAAkB;AACpB,SAAO,KAAK,WAAW,KAAK,GAAG;CAChC;CAED,IAAI,mBAAuE;EACzE,MAAM,mBAAmB,iBAAiB,YAGxC,KAAK,QAAQ,KAAK,UAAU;AAE9B,SAAO;CACR;CAED,cAAiD;AAC/C,SAAO,KAAK,iBAAiB,SAAS,KAAK,GAAG,SAAS;CACxD;CAED,MAAM,cAA8C;AAClD,OAAK,KAAK,GAAG,gBACX;EAGF,MAAM,WAAW,MAAM,KAAK,iBAAiB,SAAS,CACpD,KAAK,GAAG,eACT,EAAC;AAEF,SAAO,SACL,KAAK,GAAG,gBAAgB;CAE3B;;;;;CAMD,MAAM,kBAAsD;AAC1D,OAAK,KAAK,GAAG,sBACX;EAGF,MAAM,WAAW,MAAM,KAAK,iBAAiB,SAAS,CACpD,KAAK,GAAG,qBACT,EAAC;AAEF,SAAO,SACL,KAAK,GAAG,sBAAsB;CAEjC;;;;;CAMD,MAAM,gBAA4D;EAChE,MAAM,UAAU,MAAM,KAAK,iBAAiB;AAC5C,OAAK,QACH;AAGF,SAAO,IAAI,eAA6B;GACtC,OAAO;IAAE,IAAI;IAAU,MAAM;GAAU;GAC9B;GACT,UAAU,EACR,UAAU,KAAK,GAAG,KACnB;EACF;CACF;;;;;;;;;CAUD,MAAM,iBACJC,SAIY;EACZ,MAAM,UAAU,MAAM,KAAK,eAAe;EAC1C,MAAM,UAAU,MAAM,KAAK,iBAAiB;AAG5C,OAAK,YAAY,SAAS;GACxB,MAAMC,OAAK,MAAM,KAAK,aAAa;AACnC,UAAO,QAAQ,EAAE,SAAI,EAAC;EACvB;EAGD,MAAM,YAAY,AAAC,QAAyB,eAAe;EAC3D,MAAM,sBAAsB,KAAK,GAAG,iBAAiB;EACrD,MAAM,qBAAqB,KAAK,GAAG,uBAAuB;AAI1D,MAAI,aAAa,uBAAuB,mBACtC,QAAO,yBACL,WACA,SACA,OAAO,QAAQ;GAEb,MAAMC,aAAW,MAAM,QAAQ;IAC7B;IACA,IAAI;GACL,EAAC;AAEF,UAAOA;EACR,EACF;EAIH,MAAM,KAAK,MAAM,KAAK,aAAa;EACnC,MAAM,WAAW,MAAM,QAAQ;GAAE;GAAS;EAAI,EAAC;AAG/C,QAAM,QAAQ,OAAO;AAErB,SAAO;CACR;CAED,MAAM,iBACJC,OACgD;EAChD,MAAM,cAAc,MAAM,gBAAgB,8BACxC,OACA,KAAK,GAAG,MACT;AAED,SAAO;CACR;CAED,MAAM,cAAcC,UAA2C;AAC7D,QAAM,cACJ,KAAK,QACL,KAAK,kBACL,KAAK,GAAG,QACR,UACA,KAAK,GAAG,iBACT;CACF;CAED,MAAM,8BACJC,MACAC,QAC2C;AAC3C,SAAO,gBAAgB,8BAA8B,MAAM,OAAO;CACnE;AACF"}

package/dist/FunctionExecutionWrapper-CwtwYozd.d.cts

@@ -0,0 +1,48 @@
+import { Function, FunctionHandler } from "./Function-VI1TB3Mh.cjs";
+import { AuditStorage, AuditableAction, Auditor } from "@geekmidas/audit";
+import { EventPublisher } from "@geekmidas/events";
+import { Logger } from "@geekmidas/logger";
+import { Service, ServiceDiscovery, ServiceRecord } from "@geekmidas/services";
+import { StandardSchemaV1 } from "@standard-schema/spec";
+import { ComposableStandardSchema, InferComposableStandardSchema, InferStandardSchema } from "@geekmidas/schema";
+import { EnvironmentParser } from "@geekmidas/envkit";
+
+//#region src/functions/FunctionExecutionWrapper.d.ts
+declare abstract class FunctionExecutionWrapper<TInput extends ComposableStandardSchema | undefined = undefined, TOutSchema extends StandardSchemaV1 | undefined = undefined, TServices extends Service[] = [], TLogger extends Logger = Logger, 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, TAuditAction extends AuditableAction<string, unknown> = AuditableAction<string, unknown>> {
+  protected envParser: EnvironmentParser<{}>;
+  protected readonly fn: Function<TInput, TServices, TLogger, TOutSchema, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction, FunctionHandler<TInput, TServices, TLogger, TOutSchema, TDatabase, TAuditStorage, TAuditAction>>;
+  constructor(envParser: EnvironmentParser<{}>, fn: Function<TInput, TServices, TLogger, TOutSchema, TEventPublisher, TEventPublisherServiceName, TAuditStorage, TAuditStorageServiceName, TDatabase, TDatabaseServiceName, TAuditAction, FunctionHandler<TInput, TServices, TLogger, TOutSchema, TDatabase, TAuditStorage, TAuditAction>>);
+  protected _logger?: TLogger;
+  get logger(): TLogger;
+  get serviceDiscovery(): ServiceDiscovery<ServiceRecord<TServices>, Logger>;
+  getServices(): Promise<ServiceRecord<TServices>>;
+  getDatabase(): Promise<TDatabase | undefined>;
+  /**
+   * Get the audit storage service if configured.
+   * Returns undefined if no auditor storage is configured.
+   */
+  getAuditStorage(): Promise<TAuditStorage | undefined>;
+  /**
+   * Create an auditor instance for the function.
+   * Returns undefined if no auditor storage is configured.
+   */
+  createAuditor(): Promise<Auditor<TAuditAction> | undefined>;
+  /**
+   * 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
+   */
+  executeWithAudit<T>(handler: (ctx: {
+    auditor?: Auditor<TAuditAction>;
+    db?: TDatabase;
+  }) => Promise<T>): Promise<T>;
+  getFunctionInput<TEvent>(event: TEvent): Promise<InferComposableStandardSchema<TInput>>;
+  publishEvents(response: InferStandardSchema<TOutSchema>): Promise<void>;
+  parseComposableStandardSchema<T extends ComposableStandardSchema>(data: unknown, schema: T): Promise<InferComposableStandardSchema<T>>;
+}
+//#endregion
+export { FunctionExecutionWrapper };
+//# sourceMappingURL=FunctionExecutionWrapper-CwtwYozd.d.cts.map