Initial commit - Event Planner application

node_modules/@mikro-orm/sql/plugin/index.js

@@ -0,0 +1,50 @@
+import {
+  SelectQueryNode as SelectQueryNodeClass,
+  InsertQueryNode as InsertQueryNodeClass,
+  UpdateQueryNode as UpdateQueryNodeClass,
+  DeleteQueryNode as DeleteQueryNodeClass,
+} from 'kysely';
+import { MikroTransformer } from './transformer.js';
+/** Kysely plugin that transforms queries and results to use MikroORM entity/property naming conventions. */
+export class MikroKyselyPlugin {
+  static #queryNodeCache = new WeakMap();
+  #transformer;
+  #options;
+  constructor(em, options = {}) {
+    this.#options = options;
+    this.#transformer = new MikroTransformer(em, options);
+  }
+  transformQuery(args) {
+    this.#transformer.reset();
+    const result = this.#transformer.transformNode(args.node, args.queryId);
+    // Cache the entity map if it is one we can process (for use in transformResult)
+    if (
+      SelectQueryNodeClass.is(args.node) ||
+      InsertQueryNodeClass.is(args.node) ||
+      UpdateQueryNodeClass.is(args.node) ||
+      DeleteQueryNodeClass.is(args.node)
+    ) {
+      // clone the entityMap because the transformer's internal map will be cleared and reused by the next query
+      const entityMap = new Map(this.#transformer.getOutputEntityMap());
+      MikroKyselyPlugin.#queryNodeCache.set(args.queryId, { entityMap });
+    }
+    return result;
+  }
+  async transformResult(args) {
+    // Only transform results if columnNamingStrategy is 'property' or convertValues is true
+    if (this.#options.columnNamingStrategy !== 'property' && !this.#options.convertValues) {
+      return args.result;
+    }
+    // Retrieve the cached query node and metadata
+    const cache = MikroKyselyPlugin.#queryNodeCache.get(args.queryId);
+    if (!cache) {
+      return args.result;
+    }
+    // Transform the result rows using the transformer
+    const transformedRows = this.#transformer.transformResult(args.result.rows ?? [], cache.entityMap);
+    return {
+      ...args.result,
+      rows: transformedRows ?? [],
+    };
+  }
+}