fix: Generating message.id if not set

decentralized-identity · Mar 23, 2020 · fabf5f6 · fabf5f6
1 parent 5933e50
commit fabf5f6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/daf-core/package.json b/packages/daf-core/package.json
@@ -11,7 +11,8 @@
     "blakejs": "^1.1.0",
     "debug": "^4.1.1",
     "events": "^3.0.0",
-    "typeorm": "^0.2.24"
+    "typeorm": "^0.2.24",
+    "uuid": "^7.0.2"
   },
   "devDependencies": {
     "@types/debug": "^4.1.5",

diff --git a/packages/daf-core/src/entities/message.ts b/packages/daf-core/src/entities/message.ts
@@ -8,10 +8,12 @@ import {
   JoinTable,
   CreateDateColumn,
   UpdateDateColumn,
+  BeforeInsert,
 } from 'typeorm'
 import { Identity } from './identity'
 import { Presentation } from './presentation'
 import { Credential } from './credential'
+import { v4 as uuidv4 } from 'uuid'
 
 export interface MetaData {
   type: string
@@ -30,6 +32,13 @@ export class Message extends BaseEntity {
     }
   }
 
+  @BeforeInsert()
+  generateId() {
+    if (!this.id) {
+      this.id = uuidv4()
+    }
+  }
+
   @PrimaryColumn()
   id: string