fix database types

OutwalkStudios · May 29, 2024 · e349674 · e349674
1 parent ed64765
commit e349674
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...ges/create-firefly/templates/snippets/custom-database/typescript/define-database.template b/...ges/create-firefly/templates/snippets/custom-database/typescript/define-database.template
@@ -1,8 +1,9 @@
 /* implement a database using the firefly database interface */
 class CustomDatabase extends Database {
 
-    async connect(): Promise<void> {
+    async connect(): Promise<CustomDatabase> {
         super.displayConnectionMessage();
+        return this;
     }
 
     isConnected(): boolean {

diff --git a/packages/firefly/types/index.d.ts b/packages/firefly/types/index.d.ts
@@ -118,7 +118,7 @@ declare module "@outwalk/firefly/mongoose" {
         constructor(options?: { url?: string; } & ConnectOptions);
 
         plugin(plugin: any): void;
-        connect(): Promise<Database>;
+        connect(): Promise<MongooseDatabase>;
         isConnected(): boolean;
 
         static get connection(): Connection;