Merge pull request #22 from octet-stream/fix/references

Author: octet-stream

.changeset/spotty-peaches-hunt.md

@@ -0,0 +1,5 @@
+---
+"better-auth-mikro-orm": patch
+---
+
+Use `getReference` to ensure references loaded correctly

src/utils/adapterUtils.ts

@@ -207,11 +207,38 @@ export function createAdapterUtils(orm: MikroORM): AdapterUtils {
     )
   }
 
+  /**
+   * Normalizes property's raw input value: if property is a reference,
+   * then it wraps value using [`orm.em.getReference`](https://mikro-orm.io/docs/entity-manager#entity-references) method,
+   * to unsure it's correctly persisted.
+   *
+   * Otherwise the value is returned as is.
+   *
+   * @param property - Metadata of the property
+   * @param value - Raw input value
+   */
+  const normalizePropertyValue = (
+    property: EntityProperty,
+    value: unknown
+  ): unknown => {
+    if (
+      !property.targetMeta ||
+      property.kind === ReferenceKind.SCALAR ||
+      property.kind === ReferenceKind.EMBEDDED
+    ) {
+      return value
+    }
+
+    return orm.em.getReference(property.targetMeta.class, value)
+  }
+
   const normalizeInput: AdapterUtils["normalizeInput"] = (metadata, input) => {
     const fields: Record<string, any> = {}
     Object.entries(input).forEach(([key, value]) => {
-      const path = getFieldPath(metadata, key)
-      dset(fields, path, value)
+      const property = getPropertyMetadata(metadata, key)
+      const normalizedValue = normalizePropertyValue(property, value)
+
+      dset(fields, [property.name], normalizedValue)
     })
 
     return fields

tests/node/adapter.test.ts

@@ -50,6 +50,24 @@ suite("create", () => {
     expect(actual.userId).toBe(user.id)
   })
 
+  // https://github.com/octet-stream/better-auth-mikro-orm/issues/18
+  test("with referenced value not presented in Identity Map (issue #18)", async () => {
+    const user = await randomUsers.createAndFlushOne()
+
+    orm.em.clear()
+
+    const actual = await adapter.create<SessionInput, DatabaseSession>({
+      model: "session",
+      data: {
+        token: generateId(),
+        userId: user.id,
+        expiresAt: new Date()
+      }
+    })
+
+    expect(actual.userId).toBe(user.id)
+  })
+
   suite("generateId", () => {
     suite("via database.generateId option", () => {
       test("custom generator", async () => {