feat(core): add second EM parameter to onCreate and onUpdate call…

…back

Closes #5201

packages/core/src/decorators/Property.ts

@@ -12,6 +12,7 @@ import type {
   EntityKey,
 } from '../typings';
 import type { Type, types } from '../types';
+import type { EntityManager } from '../EntityManager';
 
 export function Property<T extends object>(options: PropertyOptions<T> = {}) {
   return function (target: any, propertyName: string) {
@@ -101,12 +102,12 @@ export type PropertyOptions<Owner> = {
    * Automatically set the property value when entity gets created, executed during flush operation.
    * @param entity
    */
-  onCreate?: (entity: Owner) => any;
+  onCreate?: (entity: Owner, em: EntityManager) => any;
   /**
    * Automatically update the property value every time entity gets updated, executed during flush operation.
    * @param entity
    */
-  onUpdate?: (entity: Owner) => any;
+  onUpdate?: (entity: Owner, em: EntityManager) => any;
   /**
    * Specify default column value for {@link https://mikro-orm.io/docs/schema-generator Schema Generator}.
    * This is a runtime value, assignable to the entity property. (SQL only)

packages/core/src/typings.ts

@@ -446,8 +446,8 @@ export interface EntityProperty<Owner = any, Target = any> {
   getterName?: keyof Owner;
   cascade: Cascade[];
   orphanRemoval?: boolean;
-  onCreate?: (entity: Owner) => any;
-  onUpdate?: (entity: Owner) => any;
+  onCreate?: (entity: Owner, em: EntityManager) => any;
+  onUpdate?: (entity: Owner, em: EntityManager) => any;
   deleteRule?: 'cascade' | 'no action' | 'set null' | 'set default' | AnyString;
   updateRule?: 'cascade' | 'no action' | 'set null' | 'set default' | AnyString;
   strategy?: LoadStrategy;

packages/core/src/unit-of-work/ChangeSetComputer.ts

@@ -5,6 +5,7 @@ import { ChangeSet, ChangeSetType } from './ChangeSet';
 import { helper, type Collection, type EntityValidator } from '../entity';
 import type { Platform } from '../platforms';
 import { ReferenceKind } from '../enums';
+import type { EntityManager } from '../EntityManager';
 
 export class ChangeSetComputer {
 
@@ -14,7 +15,8 @@ export class ChangeSetComputer {
               private readonly collectionUpdates: Set<Collection<AnyEntity>>,
               private readonly metadata: MetadataStorage,
               private readonly platform: Platform,
-              private readonly config: Configuration) {
+              private readonly config: Configuration,
+              private readonly em: EntityManager) {
     this.comparator = this.config.getComparator(this.metadata);
   }
 
@@ -84,12 +86,12 @@ export class ChangeSetComputer {
    */
   private processPropertyInitializers<T>(entity: T, prop: EntityProperty<T>, type: ChangeSetType, map: Map<T, [string, unknown][]>, nested?: boolean): void {
     if (prop.onCreate && type === ChangeSetType.CREATE && entity[prop.name] == null) {
-      entity[prop.name] = prop.onCreate(entity);
+      entity[prop.name] = prop.onCreate(entity, this.em);
     }
 
     if (prop.onUpdate && type === ChangeSetType.UPDATE) {
       const pairs = map.get(entity) ?? [];
-      pairs.push([prop.name, prop.onUpdate(entity)]);
+      pairs.push([prop.name, prop.onUpdate(entity, this.em)]);
       map.set(entity, pairs);
     }
 

packages/core/src/unit-of-work/UnitOfWork.ts

@@ -56,7 +56,7 @@ export class UnitOfWork {
     this.platform = this.em.getPlatform();
     this.eventManager = this.em.getEventManager();
     this.comparator = this.em.getComparator();
-    this.changeSetComputer = new ChangeSetComputer(this.em.getValidator(), this.collectionUpdates, this.metadata, this.platform, this.em.config);
+    this.changeSetComputer = new ChangeSetComputer(this.em.getValidator(), this.collectionUpdates, this.metadata, this.platform, this.em.config, this.em);
     this.changeSetPersister = new ChangeSetPersister(this.em.getDriver(), this.metadata, this.em.config.getHydrator(this.metadata), this.em.getEntityFactory(), this.em.getValidator(), this.em.config);
   }