@technicity/data-service-generator 0.14.5 → 0.16.0

package/dist/generation/generate.d.ts

@@ -1,7 +1,7 @@
 import type { IDialect, ISupplementClientOpts } from "../runtime/IRuntime";
-declare type ISpecialCaseUuidColumn = boolean;
-declare type IIncludeMappedFields = boolean;
-declare type IGenerateInput = {
+type ISpecialCaseUuidColumn = boolean;
+type IIncludeMappedFields = boolean;
+type IGenerateInput = {
     dialect: IDialect;
     database: string;
     user: string;

package/dist/generation/generate.js

@@ -26,7 +26,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.generate = void 0;
+exports.generate = generate;
 const path = __importStar(require("node:path"));
 const fs = __importStar(require("node:fs"));
 const os = __importStar(require("node:os"));
@@ -43,6 +43,7 @@ const json_schema_to_typescript_1 = require("json-schema-to-typescript");
 const getDuplicates_1 = require("../lib/getDuplicates");
 const isNotNullOrUndefined_1 = require("../lib/isNotNullOrUndefined");
 const MySQL_1 = require("../runtime/lib/MySQL");
+const capitalizeFirstLetter_1 = require("../lib/capitalizeFirstLetter");
 // json-schema-to-typescript inlines everything. We don't want that,
 // so use `tsType`, and add imports manually.
 // https://github.com/bcherny/json-schema-to-typescript#custom-schema-properties
@@ -201,7 +202,6 @@ async function generate(input) {
     fse.copySync(tmpBuildOutputPath, sdkOutputPath);
     fse.removeSync(tmpDirPath);
 }
-exports.generate = generate;
 function init(input) {
     const { database, user, password, host, port, server } = input;
     if (dialect === "mysql") {
@@ -256,13 +256,18 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
             .concat(Array.from(set).sort())
             .join(",\n")} } from "./types";`;
     }
-    const src = `import type { IRuntime, TMiddleware, TContext } from "./IRuntime"
+    const src = `import { AsyncLocalStorage } from "node:async_hooks";
+  import type { IRuntime, TMiddleware, TContext, TOnHandler, EventOnHandlerError } from "./IRuntime"
   import { artifacts } from "./artifacts";
 
   ${getTypeImports()}
 
+  const asyncLocalStorage = new AsyncLocalStorage<{ isInOnHandler: boolean }>();
+
   export class SDK {
     runtime: IRuntime;
+    onHandlerMap: Map<string, TOnHandler>;
+    eventTarget: EventTarget;
   
     constructor(opts: {
       runtime: any;
@@ -276,6 +281,8 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
       this.runtime = new opts.runtime(opts.clientOpts, ${supplementClientOpts === true
         ? "{ supplementClientOpts: true, ...otherOpts }"
         : "otherOpts"}, artifacts);
+      this.onHandlerMap = new Map();
+      this.eventTarget = new EventTarget();
     }
 
     $use(middleware: TMiddleware) {
@@ -310,6 +317,8 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
     }) {
       const { dbCall, commit, rollback } = await this.runtime.$startTransaction(input);
       const runtime = this.runtime;
+      const eventTarget = this.eventTarget;
+      const onHandlerMap = this.onHandlerMap;
       return {
         $commit: commit,
         $rollback: rollback,
@@ -345,6 +354,10 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
       }
     }
 
+    onOnHandlerError(cb: (event: EventOnHandlerError<unknown>) => void) {
+      this.eventTarget.addEventListener("error", cb);
+    }
+
     ${(await Promise.all(input.flatMap(async (x) => {
         if (x.kind === "getOne") {
             const findOnes = await getFindOnes(x, specialCaseUuidColumn);
@@ -374,6 +387,15 @@ async function getSDKSource(input, specialCaseUuidColumn, supplementClientOpts,
             return getMethodSourceDeleteList(x, false);
         }
     }))).join("\n\n")}
+
+    ${(await Promise.all(input.flatMap(async (x) => {
+        if (x.kind === "patchOne") {
+            const findOnes = await getFindOnes(x, specialCaseUuidColumn);
+            return getMethodSourceOnHandlerPatchOne(x, findOnes);
+        }
+    })))
+        .filter(Boolean)
+        .join("\n\n")}
   }
 
   ${await Promise.all(Object.entries(artifacts).map(async ([table, tableArtifacts]) => {
@@ -490,6 +512,7 @@ function getMethodSourceGetOne(x, findOnes, isTransaction) {
       artifacts,
       context: param2?.context,
       skipCache: param2?.skipCache,
+      eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
       ${isTransaction ? "dbCall" : ""}
     }
   );
@@ -510,6 +533,7 @@ function getMethodSourceGetList(x, isTransaction) {
         artifacts,
         context: param2?.context,
         skipCache: param2?.skipCache,
+        eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
         ${isTransaction ? "dbCall" : ""}
       }
     );
@@ -530,6 +554,7 @@ function getMethodSourceGetListPaginated(x, isTransaction) {
         artifacts,
         context: param2?.context,
         skipCache: param2?.skipCache,
+        eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
         ${isTransaction ? "dbCall" : ""}
       }
     );
@@ -548,6 +573,10 @@ function getMethodSourcePostOne(x, specialCaseUuidColumn, isTransaction) {
     artifacts,
     fields: param2?.$fields as any,
     context: {...param2?.context, specialCaseUuidColumn: ${JSON.stringify(specialCaseUuidColumn)}},
+    eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
+    onHandler: ${isTransaction ? "onHandlerMap" : "this.onHandlerMap"}.get("${mapKindToAction(x.kind)}-${x.table}"),
+    asyncLocalStorage,
+    sdk: this,
     ${isTransaction ? "dbCall" : ""}
   });
 }`;
@@ -569,10 +598,25 @@ return ${isTransaction ? "runtime" : "this.runtime"}.resolve({
   artifacts,
   fields: param2?.$fields as any,
   context: param2?.context,
+  eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
+  onHandler: ${isTransaction ? "onHandlerMap" : "this.onHandlerMap"}.get("${mapKindToAction(x.kind)}-${x.table}"),
+  asyncLocalStorage,
+  sdk: this,
   ${isTransaction ? "dbCall" : ""}
 });
 }`;
 }
+function getMethodSourceOnHandlerPatchOne(x, findOnes) {
+    return `on${(0, capitalizeFirstLetter_1.capitalizeFirstLetter)(x.methodName)}(handler:
+    (sdk: InstanceType<typeof SDK>, input: { $where: ${findOnes
+        .map((findOne) => `{ ${findOne.name}: ${findOne.type}${findOne.nullable ? " | null" : ""} }`)
+        .join(" | ")}, data: ${x.typeDataName} },
+    output: Partial<${getTypeReturnName(x.table)}>
+  ) => Promise<void>
+): void {
+  this.onHandlerMap.set("${mapKindToAction(x.kind)}-${x.table}", handler);
+}`;
+}
 function getMethodSourcePatchList(x, isTransaction) {
     const param2 = `{ ${keyFields}?: ${x.typeFieldsName}, correlationId?: string, context?: TContext }`;
     return `async ${x.methodName}<T extends ${param2}>(
@@ -588,6 +632,10 @@ function getMethodSourcePatchList(x, isTransaction) {
     artifacts,
     fields: param2?.$fields as any,
     context: param2?.context,
+    eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
+    onHandler: ${isTransaction ? "onHandlerMap" : "this.onHandlerMap"}.get("${mapKindToAction(x.kind)}-${x.table}"),
+    asyncLocalStorage,
+    sdk: this,
     ${isTransaction ? "dbCall" : ""}
   });
 }`;
@@ -605,6 +653,10 @@ function getMethodSourceDeleteOne(x, findOnes, isTransaction) {
     args: { $where: param1 },
     artifacts,
     context: param2?.context,
+    eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
+    onHandler: ${isTransaction ? "onHandlerMap" : "this.onHandlerMap"}.get("${mapKindToAction(x.kind)}-${x.table}"),
+    asyncLocalStorage,
+    sdk: this,
     ${isTransaction ? "dbCall" : ""}
   });
 }`;
@@ -620,6 +672,10 @@ function getMethodSourceDeleteList(x, isTransaction) {
       args: param1,
       artifacts,
       context: param2?.context,
+      eventTarget: ${isTransaction ? "eventTarget" : "this.eventTarget"},
+      onHandler: ${isTransaction ? "onHandlerMap" : "this.onHandlerMap"}.get("${mapKindToAction(x.kind)}-${x.table}"),
+      asyncLocalStorage,
+      sdk: this,
       ${isTransaction ? "dbCall" : ""}
     });
   }`;

package/dist/lib/capitalizeFirstLetter.d.ts

@@ -0,0 +1 @@
+export declare function capitalizeFirstLetter(str: string): string;

package/dist/lib/capitalizeFirstLetter.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.capitalizeFirstLetter = capitalizeFirstLetter;
+function capitalizeFirstLetter(str) {
+    return str.charAt(0).toUpperCase() + str.slice(1);
+}

package/dist/lib/getDuplicates.js

@@ -1,10 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDuplicates = void 0;
+exports.getDuplicates = getDuplicates;
 function getDuplicates(arr) {
     return arr.reduce((agg, col) => {
         agg.filter[col] = agg.filter[col] ? agg.dup.push(col) : 2;
         return agg;
     }, { filter: {}, dup: [] }).dup;
 }
-exports.getDuplicates = getDuplicates;

package/dist/lib/isNotNullOrUndefined.js

@@ -1,8 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isNotNullOrUndefined = void 0;
+exports.isNotNullOrUndefined = isNotNullOrUndefined;
 // https://github.com/Microsoft/TypeScript/issues/16069
 function isNotNullOrUndefined(input) {
     return input != null;
 }
-exports.isNotNullOrUndefined = isNotNullOrUndefined;

package/dist/runtime/Cache.d.ts

@@ -1,7 +1,7 @@
 import Redis, { Cluster } from "ioredis";
 import { TResolveParams } from "./IRuntime";
 import Stats from "./Stats";
-export declare type RedisConfig = {
+export type RedisConfig = {
     host: string;
     port: number;
     tls?: boolean;

package/dist/runtime/IRuntime.d.ts

@@ -1,3 +1,4 @@
+import { AsyncLocalStorage } from "node:async_hooks";
 export interface IRuntime {
     resolve(input: TResolveParams): Promise<any>;
     $queryRaw(sql: string, values?: any[]): Promise<any>;
@@ -9,8 +10,8 @@ export interface IRuntime {
         isolationLevel?: "READ UNCOMMITTED" | "READ COMMITTED" | "REPEATABLE READ" | "SERIALIZABLE";
     }): Promise<TBeginTransactionResult>;
 }
-export declare type TAction = "findUnique" | "findMany" | "findManyPaginated" | "create" | "update" | "updateMany" | "delete" | "deleteMany";
-export declare type TResolveParams = {
+export type TAction = "findUnique" | "findMany" | "findManyPaginated" | "create" | "update" | "updateMany" | "delete" | "deleteMany";
+export type TResolveParams = {
     resource: string;
     action: TAction;
     args?: IArgs;
@@ -20,54 +21,68 @@ export declare type TResolveParams = {
     context?: TContext;
     skipCache?: boolean;
     dbCall?: TDbCall;
-};
-export declare type TContext = {
+    eventTarget: EventTarget;
+    onHandler?: TOnHandler;
+    asyncLocalStorage?: AsyncLocalStorage<{
+        isInOnHandler?: boolean;
+        isTransaction?: boolean;
+    }>;
+    sdk?: unknown;
+};
+export type TOnHandler = (sdk: unknown, input: unknown, output: unknown) => Promise<void>;
+export declare class EventOnHandlerError<T> extends Event {
+    error: T;
+    constructor(message: string, data: ConstructorParameters<typeof Event>[1] & {
+        error: T;
+    });
+}
+export type TContext = {
     [k: string]: any;
 };
-export declare type TMiddleware = (params: TResolveParams, next: (params: TResolveParams) => Promise<any>) => Promise<any>;
-export declare type IDialect = "mysql" | "mssql" | "sqlite";
-export declare type TDbCall = (q: string) => Promise<any>;
-export declare type TFormatQuery = (q: string, values: any[]) => string;
-export declare type TBeginTransaction = () => Promise<TBeginTransactionResult>;
-declare type TBeginTransactionResult = {
+export type TMiddleware = (params: TResolveParams, next: (params: TResolveParams) => Promise<any>) => Promise<any>;
+export type IDialect = "mysql" | "mssql" | "sqlite";
+export type TDbCall = (q: string) => Promise<any>;
+export type TFormatQuery = (q: string, values: any[]) => string;
+export type TBeginTransaction = () => Promise<TBeginTransactionResult>;
+type TBeginTransactionResult = {
     dbCall: (q: string) => Promise<any>;
     commit: () => Promise<void>;
     rollback: () => Promise<void>;
 };
-export declare type ISupplementClientOpts = boolean;
-export declare type IOrderBy = {
+export type ISupplementClientOpts = boolean;
+export type IOrderBy = {
     column: string;
     direction: any;
 }[];
-export declare type IArgs = {
+export type IArgs = {
     [k: string]: any;
 };
-export declare type TSelect = {
+export type TSelect = {
     [k: string]: boolean | {
         $fields?: TSelect;
     };
 };
-declare type IWhere = (table: string, args: IArgs) => string | undefined;
-declare type IASTChildColumn = {
+type IWhere = (table: string, args: IArgs) => string | undefined;
+type IASTChildColumn = {
     type: "column";
     name: string;
     fieldName: string;
     as: string;
     fromOtherTable?: string;
 };
-declare type IASTChildComposite = {
+type IASTChildComposite = {
     type: "composite";
     name: string[];
     fieldName: string;
     as: string;
     fromOtherTable?: string;
 };
-declare type ISqlJoin = (t1: string, t2: string, args: IArgs) => string;
-declare type ISqlBatch = {
+type ISqlJoin = (t1: string, t2: string, args: IArgs) => string;
+type ISqlBatch = {
     thisKey: IASTChildColumn;
     parentKey: IASTChildColumn;
 };
-declare type IJunction = {
+type IJunction = {
     sqlTable: string;
     as: string;
     uniqueKey: string | string[];
@@ -83,7 +98,7 @@ declare type IJunction = {
     where?: IWhere;
     sqlJoins: ISqlJoin[];
 };
-export declare type IGetSQLASTInput = {
+export type IGetSQLASTInput = {
     table: string;
     fieldName: string;
     fields: TSelect;
@@ -100,17 +115,17 @@ export declare type IGetSQLASTInput = {
     dialect: IDialect;
     firstChild?: IASTChildColumn | IASTChildComposite;
 };
-export declare type IRelation = {
+export type IRelation = {
     table: string;
     foreignKey: string;
     referencedTable: string;
     referencedKey: string;
     nullable: boolean;
 };
-declare type IOneToManyOrManyToOne = "one-to-many__many-to-one";
-declare type IManyToMany = "many-to-many";
-declare type IRelationKind = "one-to-many" | "many-to-one";
-export declare type IRelationOneToManyOrManyToOne = {
+type IOneToManyOrManyToOne = "one-to-many__many-to-one";
+type IManyToMany = "many-to-many";
+type IRelationKind = "one-to-many" | "many-to-one";
+export type IRelationOneToManyOrManyToOne = {
     type: IOneToManyOrManyToOne;
     kind: IRelationKind;
     grabMany: boolean;
@@ -119,7 +134,7 @@ export declare type IRelationOneToManyOrManyToOne = {
     relation: IRelation;
     nullable: boolean;
 };
-export declare type IRelationManyToMany = {
+export type IRelationManyToMany = {
     type: IManyToMany;
     grabMany: boolean;
     table: string;
@@ -127,8 +142,8 @@ export declare type IRelationManyToMany = {
     name: string;
     relations: IRelation[];
 };
-export declare type IRelationField = IRelationOneToManyOrManyToOne | IRelationManyToMany;
-export declare type IMappedField = {
+export type IRelationField = IRelationOneToManyOrManyToOne | IRelationManyToMany;
+export type IMappedField = {
     name: string;
     as: string;
     type: string;
@@ -138,7 +153,7 @@ export declare type IMappedField = {
     referencedTable: string;
     referencedKey: string;
 };
-export declare type IArtifacts = {
+export type IArtifacts = {
     [k: string]: {
         table: string;
         primaryKey: string;
@@ -164,8 +179,8 @@ export declare type IArtifacts = {
         } | null;
     };
 };
-export declare type TFieldType = "string" | "boolean" | "number" | "integer";
-export declare type TField = {
+export type TFieldType = "string" | "boolean" | "number" | "integer";
+export type TField = {
     kind: "scalar";
     type: TFieldType;
     name: string;

package/dist/runtime/IRuntime.js

@@ -1,2 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.EventOnHandlerError = void 0;
+class EventOnHandlerError extends Event {
+    constructor(message, data) {
+        super(message, data);
+        this.error = data.error;
+    }
+}
+exports.EventOnHandlerError = EventOnHandlerError;

package/dist/runtime/RuntimeSQLite.d.ts

@@ -1,7 +1,7 @@
 import Database from "better-sqlite3";
 import type { IRuntime, TMiddleware, TResolveParams, IArtifacts, ISupplementClientOpts } from "./IRuntime";
 import { MiddlewareHandler } from "./lib/shared";
-declare type TClientOpts = {
+type TClientOpts = {
     filename: string;
 } & Parameters<typeof Database>[1];
 export declare class RuntimeSQLite implements IRuntime {

package/dist/runtime/RuntimeSQLite.js

@@ -128,7 +128,12 @@ class RuntimeSQLite {
         }
     }
     async rollback() {
-        __classPrivateFieldGet(this, _RuntimeSQLite_db, "f").prepare("ROLLBACK").run();
+        try {
+            __classPrivateFieldGet(this, _RuntimeSQLite_db, "f").prepare("ROLLBACK").run();
+        }
+        catch (e) {
+            console.log(e);
+        }
     }
 }
 exports.RuntimeSQLite = RuntimeSQLite;

package/dist/runtime/Stats.d.ts

@@ -4,9 +4,5 @@ declare class Stats {
     constructor(client: Redis | Cluster);
     updateStats: (ms: number, category: string) => Promise<void>;
 }
-declare namespace timer {
-    type Callback = (time: number, ...params: any[]) => void;
-    type Params<T> = T extends (ms: number, ...params: infer U) => any ? U : never;
-}
 export declare function timer(): <T extends timer.Callback>(callback: T, ...args: timer.Params<T>) => void;
 export default Stats;

package/dist/runtime/Stats.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.timer = void 0;
+exports.timer = timer;
 const perf_hooks_1 = require("perf_hooks");
 class Stats {
     constructor(client) {
@@ -28,5 +28,4 @@ function timer() {
         callback(perf_hooks_1.performance.now() - start, ...args);
     };
 }
-exports.timer = timer;
 exports.default = Stats;

package/dist/runtime/lib/addNullFallbacks.js

@@ -2,7 +2,7 @@
 // If a many-to-one relation's ID field is null, we want the
 // relation field to be null as well, not undefined.
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.addNullFallbacks = void 0;
+exports.addNullFallbacks = addNullFallbacks;
 function addNullFallbacks(sqlAST, data) {
     if (data == null) {
         return;
@@ -30,4 +30,3 @@ function addNullFallbacks(sqlAST, data) {
         }
     }
 }
-exports.addNullFallbacks = addNullFallbacks;

package/dist/runtime/lib/cursor.js

@@ -1,11 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.decodeCursor = exports.encodeCursor = void 0;
+exports.encodeCursor = encodeCursor;
+exports.decodeCursor = decodeCursor;
 function encodeCursor(cursor) {
     return Buffer.from(JSON.stringify(cursor)).toString("base64");
 }
-exports.encodeCursor = encodeCursor;
 function decodeCursor(cursor) {
     return JSON.parse(Buffer.from(cursor, "base64").toString("utf8"));
 }
-exports.decodeCursor = decodeCursor;

package/dist/runtime/lib/getDateTimeStringMySQL.js

@@ -1,8 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDateTimeStringMySQL = void 0;
+exports.getDateTimeStringMySQL = getDateTimeStringMySQL;
 // Assumes input uses zero offset
 function getDateTimeStringMySQL(dateTimeString) {
     return dateTimeString.replace("T", " ").slice(0, 19);
 }
-exports.getDateTimeStringMySQL = getDateTimeStringMySQL;

package/dist/runtime/lib/getOrderBy.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getOrderBy = void 0;
+exports.getOrderBy = getOrderBy;
 // https://gist.github.com/pcattori/2bb645d587e45c9fdbcabf5cef7a7106
 function getOrderBy(args, primaryKey) {
     let out = undefined;
@@ -41,7 +41,6 @@ function getOrderBy(args, primaryKey) {
     }
     return { orderBy: out, flip };
 }
-exports.getOrderBy = getOrderBy;
 function getFlip(args) {
     return args?.$paginate?.last != null;
 }

package/dist/runtime/lib/getSqlAst.js

@@ -26,7 +26,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getSqlAst = void 0;
+exports.getSqlAst = getSqlAst;
 const SqlString = __importStar(require("sqlstring"));
 // @ts-ignore
 const TSqlString = __importStar(require("tsqlstring"));
@@ -220,7 +220,6 @@ function getSqlAst(input) {
         }))
     };
 }
-exports.getSqlAst = getSqlAst;
 function keyToASTChild(key, namespace, fromOtherTable) {
     if (Array.isArray(key)) {
         const clumsyName = toClumsyName(key);

package/dist/runtime/lib/getWhere.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getWhere = void 0;
+exports.getWhere = getWhere;
 const stringifyWhere_1 = require("./stringifyWhere");
 function getWhere(
 // Note: `table` is not escaped in getWhere, so it should be escaped beforehand.
@@ -18,4 +18,3 @@ table, args, dialect, orderBy, rowWithMatchingCursor) {
         rowWithMatchingCursor,
     }) || null);
 }
-exports.getWhere = getWhere;

package/dist/runtime/lib/shared.d.ts

@@ -1,4 +1,4 @@
-import type { IGetSQLASTInput, IArtifacts, IDialect, TDbCall, TFormatQuery, TBeginTransaction, TContext, TMiddleware, TResolveParams } from "../IRuntime";
+import { type IGetSQLASTInput, type IArtifacts, type IDialect, type TDbCall, type TFormatQuery, type TBeginTransaction, type TContext, type TMiddleware, type TResolveParams } from "../IRuntime";
 import Cache from "../Cache";
 export declare function resolve(input: TResolveParams, dbCall: TDbCall, formatQuery: TFormatQuery, beginTransaction: TBeginTransaction, dialect: IDialect, middlewareHandler: MiddlewareHandler<TMiddleware>, context: TContext, cache?: Cache): Promise<any>;
 export declare class MiddlewareHandler<M extends Function> {

package/dist/runtime/lib/shared.js

@@ -26,7 +26,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._prepareWhere = exports.whereNeedsProcessing = exports.postProcess = exports.MiddlewareHandler = exports.resolve = void 0;
+exports.MiddlewareHandler = void 0;
+exports.resolve = resolve;
+exports.postProcess = postProcess;
+exports.whereNeedsProcessing = whereNeedsProcessing;
+exports._prepareWhere = _prepareWhere;
 // @ts-ignore
 // import * as queryAST from "join-monster/dist/query-ast-to-sql-ast";
 // @ts-ignore
@@ -39,6 +43,7 @@ const async_hooks_1 = require("async_hooks");
 const _ = __importStar(require("lodash/fp"));
 const uuid_1 = require("uuid");
 const getSqlAst_1 = require("./getSqlAst");
+const IRuntime_1 = require("../IRuntime");
 const getWhere_1 = require("./getWhere");
 const getDateTimeStringMySQL_1 = require("./getDateTimeStringMySQL");
 const cursor_1 = require("./cursor");
@@ -60,13 +65,40 @@ async function resolve(input, dbCall, formatQuery, beginTransaction, dialect, mi
                 return nextMiddleware(paramsMaybeMutated, consumer);
             }
             const paramsChanged = { ...input, ...params };
-            return _resolve(paramsChanged, dbCall, formatQuery, beginTransaction, dialect, context, cache);
+            const p = _resolve(paramsChanged, dbCall, formatQuery, beginTransaction, dialect, context, cache);
+            const onHandler = input.onHandler;
+            const asyncLocalStorage = input.asyncLocalStorage;
+            if (typeof onHandler === "function" &&
+                asyncLocalStorage != null &&
+                asyncLocalStorage?.getStore()?.isInOnHandler !== true) {
+                return p.then((output) => asyncLocalStorage
+                    .run({ isInOnHandler: true }, async () => onHandler(input.sdk, { ...input.args, data: input.data }, output))
+                    .then(() => output)
+                    .catch((error) => {
+                    input.eventTarget.dispatchEvent(new IRuntime_1.EventOnHandlerError("error", { error }));
+                    return output;
+                }));
+            }
+            return p;
         };
         return resource.runInAsyncScope(() => consumer(params));
     }
-    return _resolve(input, dbCall, formatQuery, beginTransaction, dialect, context, cache);
+    const p = _resolve(input, dbCall, formatQuery, beginTransaction, dialect, context, cache);
+    const onHandler = input.onHandler;
+    const asyncLocalStorage = input.asyncLocalStorage;
+    if (typeof onHandler === "function" &&
+        asyncLocalStorage != null &&
+        asyncLocalStorage?.getStore()?.isInOnHandler !== true) {
+        return p.then((output) => asyncLocalStorage
+            .run({ isInOnHandler: true }, async () => onHandler(input.sdk, { ...input.args, data: input.data }, output))
+            .then(() => output)
+            .catch((error) => {
+            input.eventTarget.dispatchEvent(new IRuntime_1.EventOnHandlerError("error", { error }));
+            return output;
+        }));
+    }
+    return p;
 }
-exports.resolve = resolve;
 function _resolve(input, dbCall, formatQuery, beginTransaction, dialect, context, cache) {
     switch (input.action) {
         case "findMany":
@@ -842,7 +874,6 @@ function postProcess(data, fields, shouldRemoveExtraKeys) {
         removeExtraKeys(data, fields);
     }
 }
-exports.postProcess = postProcess;
 function removeExtraKeys(data, fields) {
     if (data == null || (Array.isArray(data) && data[0] == null)) {
         return;
@@ -938,7 +969,6 @@ function typeCastSqlite(data, fields, table, artifacts) {
 function whereNeedsProcessing(where) {
     return JSON.stringify(where).includes("Uuid");
 }
-exports.whereNeedsProcessing = whereNeedsProcessing;
 async function _prepareWhere(artifacts, table, data, dbCall, formatQuery) {
     const mappedFields = artifacts[table].mappedFields;
     let out = {};
@@ -1034,7 +1064,6 @@ async function _prepareWhere(artifacts, table, data, dbCall, formatQuery) {
     });
     return out;
 }
-exports._prepareWhere = _prepareWhere;
 function getPrepareWhereNotFoundMessage(mappedField, value) {
     return `Not found: unable to map \`${mappedField.referencedTable}\`.\`${mappedField.name}\` to \`${mappedField.referencedTable}\`.\`${mappedField.referencedKey}\` for the value \`${value}\`.`;
 }

package/dist/runtime/lib/stringifyWhere.d.ts

@@ -1,8 +1,8 @@
 import type { IOrderBy, IDialect } from "../IRuntime";
-declare type IWhere = {
+type IWhere = {
     [k: string]: any;
 };
-declare type IArgs = {
+type IArgs = {
     [k: string]: any;
 };
 export declare function stringifyWhere(input: {

package/dist/runtime/lib/stringifyWhere.js

@@ -23,7 +23,9 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getEscape = exports.getEscapeId = exports.stringifyWhere = void 0;
+exports.stringifyWhere = stringifyWhere;
+exports.getEscapeId = getEscapeId;
+exports.getEscape = getEscape;
 const _ = __importStar(require("lodash/fp"));
 const MySqlString = __importStar(require("sqlstring"));
 // @ts-expect-error
@@ -49,7 +51,6 @@ function stringifyWhere(input) {
     }
     return result;
 }
-exports.stringifyWhere = stringifyWhere;
 function _stringifyWhere(where, table, escapeId, escape, result) {
     if (Object.prototype.hasOwnProperty.call(where, "$and")) {
         if (Object.keys(where).length !== 1) {
@@ -239,7 +240,6 @@ function getEscapeId(dialect) {
     }
     throw new Error("Unsupported dialect: " + dialect);
 }
-exports.getEscapeId = getEscapeId;
 function getEscape(dialect) {
     if (dialect === "mysql") {
         return MySqlString.escape.bind(MySqlString);
@@ -252,4 +252,3 @@ function getEscape(dialect) {
     }
     throw new Error("Unsupported dialect: " + dialect);
 }
-exports.getEscape = getEscape;

package/dist/runtime/lib/typeCastMSSQL.d.ts

@@ -1,3 +1,3 @@
-declare type TValueCast = (value: any) => any;
+type TValueCast = (value: any) => any;
 export declare function typeCastMSSQL(customTypeCast?: (meta: any, mssql: any) => TValueCast): (result: any, meta: any) => any;
 export {};

package/dist/runtime/lib/typeCastMSSQL.js

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.typeCastMSSQL = void 0;
+exports.typeCastMSSQL = typeCastMSSQL;
 const mssql = __importStar(require("mssql"));
 // TODO: see https://github.com/tediousjs/node-mssql/pull/1171
 function typeCastMSSQL(customTypeCast) {
@@ -60,4 +60,3 @@ function typeCastMSSQL(customTypeCast) {
         });
     };
 }
-exports.typeCastMSSQL = typeCastMSSQL;

package/dist/runtime/lib/utility.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.flatten = exports.mapAsync = void 0;
+exports.mapAsync = mapAsync;
+exports.flatten = flatten;
 /**
  * Eagerly resolve and map input values then concatinate.
  */
@@ -8,8 +9,6 @@ async function mapAsync(source, mapFunction) {
     const output = (await source).map(mapFunction);
     return Promise.all(flatten(output)).then(flatten);
 }
-exports.mapAsync = mapAsync;
 function flatten(array) {
     return [].concat(...array);
 }
-exports.flatten = flatten;

package/dist/traverseFieldArgs.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.traverseFieldArgs = void 0;
+exports.traverseFieldArgs = traverseFieldArgs;
 function traverseFieldArgs(fields, cb) {
     const values = Object.values(fields);
     for (let x of values) {
@@ -15,4 +15,3 @@ function traverseFieldArgs(fields, cb) {
         traverseFieldArgs(fields, cb);
     }
 }
-exports.traverseFieldArgs = traverseFieldArgs;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@technicity/data-service-generator",
-  "version": "0.14.5",
+  "version": "0.16.0",
   "main": "./dist/index.js",
   "files": [
     "dist"
@@ -47,7 +47,7 @@
     "jest": "^29.4.3",
     "sinon": "12.0.1",
     "testcontainers": "^9.1.3",
-    "typescript": "4.6.4"
+    "typescript": "5.5.2"
   },
   "resolutions": {
     "xml2js": "0.5.0",