fix(service): add this context to mongodb migration functions

ngageoint · Sep 23, 2024 · 422e6d5 · 422e6d5
1 parent e6988d7
commit 422e6d5
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/service/src/@types/mongodb-migrations/index.d.ts b/service/src/@types/mongodb-migrations/index.d.ts
@@ -2,7 +2,7 @@
 
 declare module '@ngageoint/mongodb-migrations' {
 
-  import { MongoClientOptions } from 'mongodb'
+  import { MongoClientOptions, Db } from 'mongodb'
   export { MongoClientOptions } from 'mongodb'
 
   export class Migrator {
@@ -74,10 +74,15 @@ declare module '@ngageoint/mongodb-migrations' {
     options?: MongoClientOptions
   }
 
+  export type MigrationContext = {
+    db: Db
+    log: Console
+  }
+
   export interface Migration {
     id: string
-    up?: (callback: (err: any) => any) => any
-    down?: (callback: (err: any) => any) => any
+    up?: (this: MigrationContext, callback: (err: any) => any) => any
+    down?: (this: MigrationContext, callback: (err: any) => any) => any
   }
 
   export interface MigrationResult {