@nest-arch/tui 0.2.8 → 0.4.0

package/dist/templates/base/.env.example.hbs

@@ -13,7 +13,7 @@ POSTGRES_URL=postgresql://postgres:postgres@localhost:5432/{{projectNameKebab}}_
 MYSQL_URL=mysql://root:mysql@localhost:3306/{{projectNameKebab}}_db
 {{/if}}
 {{#if hasSqlite}}
-SQLITE_PATH=file:./src/database/dev.db
+SQLITE_PATH=./src/database/dev.db
 {{/if}}
 {{#if hasMongodb}}
 MONGODB_URL=mongodb://localhost:27017/{{projectNameKebab}}_db
@@ -24,11 +24,13 @@ MONGODB_URL=mongodb://localhost:27017/{{projectNameKebab}}_db
 GRAPHQL_PLAYGROUND=true
 
 {{/if}}
-{{#if (and (eq auth "passport") hasDatabase)}}
+{{#if (eq auth "passport")}}
 # JWT
 JWT_ACCESS_SECRET=change-me-to-a-random-secret-at-least-32-characters-long
+{{#if hasDatabase}}
 JWT_REFRESH_SECRET=change-me-to-a-different-random-secret-at-least-32-characters
 {{/if}}
+{{/if}}
 {{#if (or (eq auth "better-auth") (eq auth "better-auth:stateless"))}}
 # Better Auth
 BETTER_AUTH_SECRET=change-me-to-a-random-secret-at-least-32-characters-long

package/dist/templates/base/.env.hbs

@@ -11,15 +11,17 @@ POSTGRES_URL=postgresql://postgres:postgres@localhost:5432/{{projectNameKebab}}_
 MYSQL_URL=mysql://root:mysql@localhost:3306/{{projectNameKebab}}_db
 {{/if}}
 {{#if hasSqlite}}
-SQLITE_PATH=file:./src/database/dev.db
+SQLITE_PATH=./src/database/dev.db
 {{/if}}
 {{#if hasMongodb}}
 MONGODB_URL=mongodb://localhost:27017/{{projectNameKebab}}_db
 {{/if}}
-{{#if (and (eq auth "passport") hasDatabase)}}
+{{#if (eq auth "passport")}}
 JWT_ACCESS_SECRET=change-me-to-a-random-secret-at-least-32-characters-long
+{{#if hasDatabase}}
 JWT_REFRESH_SECRET=change-me-to-a-different-random-secret-at-least-32-characters
 {{/if}}
+{{/if}}
 {{#if (or (eq auth "better-auth") (eq auth "better-auth:stateless"))}}
 BETTER_AUTH_SECRET=change-me-to-a-random-secret-at-least-32-characters-long
 BETTER_AUTH_URL=http://localhost:3000

package/dist/templates/base/.gitignore.hbs

@@ -43,6 +43,7 @@ lerna-debug.log*
 
 # dotenv environment variable files
 .env
+.env.test
 .env.development.local
 .env.test.local
 .env.production.local
@@ -59,4 +60,4 @@ pids
 *.pid.lock
 
 # Diagnostic reports (https://nodejs.org/api/report.html)
-report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json
+report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json

package/dist/templates/base/.yarnrc.yml.hbs

@@ -1,6 +1 @@
 nodeLinker: node-modules
-catalog:
-  "@types/node": ^22.13.14
-  dotenv: ^17.2.2
-  typescript: ^6.0.2
-  zod: ^4.4.3

package/dist/templates/base/README.md.hbs

@@ -79,12 +79,18 @@ This project uses **TypeORM** as the ORM. Available commands:
 {{#if (or (eq auth "better-auth") (eq auth "better-auth:stateless"))}}
 ## Auth Setup
 
-This project uses **Better Auth**. Generate the database schema by running:
+This project uses **Better Auth** with database-backed sessions. Set `BETTER_AUTH_SECRET` and `BETTER_AUTH_URL` in `.env` before running the generated scripts.
 
-```bash
-{{#if isPnpm}}pnpm dlx auth@latest generate --config src/auth/auth.ts --output src/database/orm/prisma/schema.prisma -y && pnpm run prisma:format{{else if isYarn}}yarn dlx auth@latest generate --config src/auth/auth.ts --output src/database/orm/prisma/schema.prisma -y && yarn run prisma:format{{else if isBun}}bun x auth@latest generate --config src/auth/auth.ts --output src/database/orm/prisma/schema.prisma -y && bun run prisma:format{{else}}npx auth@latest generate --config src/auth/auth.ts --output src/database/orm/prisma/schema.prisma -y && npm run prisma:format{{/if}}
-```
-See https://better-auth.com/docs/adapters/prisma for more details.
+Run `auth:generate` with your package manager. The script uses the installed CLI version.
+{{#if usePrisma}}
+Then run `prisma:generate`, `prisma:migrate --name auth`, and `build`. For deployments, apply the committed migrations with `prisma:migrate:deploy`. Auth models live in `src/database/orm/prisma/auth.prisma`; keep application models in `schema.prisma` or other files in that directory. `auth:generate` replaces only the auth-owned file (using `auth-schema.prisma` as CLI output). Inspect that diff after changing plugins.
+{{else if useDrizzle}}
+Then run `db:generate`, `db:migrate`, and `build`. Auth owns `src/database/orm/drizzle/auth-schema.ts`; application tables remain in `schema.ts`. Commit generated migration files. SQLite requires no Docker service.
+{{else}}
+Then run `auth:migrate` and `build`.
+{{/if}}
+
+The runtime auth module receives the database client owned by Nest. CLI schema discovery uses separate configuration. `GET /api/auth-profile` is a protected example for the Prisma and Drizzle profiles.
 
 ### Usage
 
@@ -185,3 +191,25 @@ import { AppRouterClass as AppRouter } from "./app.router.js";
 export class AppModule {}
 ```
 {{/if}}
+
+{{#if isPassportAuth}}
+## Passport authentication
+
+Set a random `JWT_ACCESS_SECRET` of at least 32 characters. Configure `JWT_ISSUER`, `JWT_AUDIENCE`, and `JWT_KEY_ID` consistently across instances. Apply Prisma migrations before starting.
+
+Register with `POST /api/auth/register` and sign in with `POST /api/auth/login`, using `{ "email": "user@example.com", "password": "a-long-unique-password" }`. Send the returned access token in `Authorization: Bearer <token>` to `GET /api/auth/profile`. Access tokens expire after 15 minutes. Passwords are hashed with bcrypt; token payloads contain no password or role supplied by the client.
+
+Refresh with `POST /api/auth/refresh` and `{ "refreshToken": "..." }`. Refresh tokens rotate on every use and expire seven days after login. Reusing a consumed token revokes the entire login session, including access tokens. Concurrent refresh requests therefore require client-side serialization. `POST /api/auth/logout` takes the same body and revokes that login session. Tokens are returned in JSON; no cookie transport is generated.
+
+Only hashes of refresh tokens are stored. Schedule deletion of expired refresh-token families according to your retention policy. Secret replacement invalidates existing access tokens; overlapping signing keys are not implemented in this baseline.
+{{/if}}
+{{#if hasGraphql}}
+## GraphQL
+
+Send queries to `POST /graphql`. `{ hello }` is always available. With the Todo example, `todos`, `createTodo`, `updateTodo`, and `deleteTodo` operate on PostgreSQL. The first profile is unauthenticated; it does not include subscriptions or federation.
+{{/if}}
+{{#if hasMongodb}}
+## MongoDB
+
+Set `MONGODB_URL` including the database name. The native client connects and pings MongoDB during startup and closes with the Nest application. No Prisma migration commands are required. Todo identifiers are serialized ObjectId strings; malformed IDs return 400 and missing mutation targets return 404. Listing returns at most 100 items in identifier order. This baseline performs single-document writes and does not promise multi-document transactions.
+{{/if}}

package/dist/templates/base/nest-cli.json.hbs

@@ -3,6 +3,7 @@
   "collection": "@nestjs/schematics",
   "sourceRoot": "src",
   "compilerOptions": {
-    "deleteOutDir": true
+    "deleteOutDir": true{{#if (includes microservices "grpc")}},
+    "assets": ["proto/**/*"]{{/if}}
   }
 }

package/dist/templates/base/pnpm-workspace.yaml.hbs

@@ -1,4 +1,8 @@
 allowBuilds:
+{{#if useDrizzle}}
+  esbuild: true
+{{/if}}
+  protobufjs: true
   "@nestjs/core": true
 {{#if swagger}}
   "@scarf/scarf": true
@@ -9,6 +13,11 @@ allowBuilds:
 {{#if usePrisma}}
   prisma: true
   "@prisma/engines": true
+  {{#if isPrisma8}}
+  esbuild: true
+  msgpackr-extract: true
+  workerd: true
+  {{/if}}
 {{/if}}
 {{#if (and (eq auth "passport") hasDatabase)}}
   bcrypt: true

package/dist/templates/base/prisma.config.ts.hbs

@@ -1,11 +1,37 @@
-import "dotenv/config";
-import { defineConfig, env } from "prisma/config";
+{{#if isMonorepo}}
+import { env } from "@repo/env/api-gateway";
+{{else}}
+import { env } from "./src/config/env.js";
+{{/if}}
+{{#if isPrisma8}}
+import { definePrismaConfig } from "prisma/config";
+{{#if hasMongodb}}
+import { defineConfig as ormConfig } from "@prisma/orm-mongo/config";
+{{else}}
+import { defineConfig as ormConfig } from "@prisma/orm-postgres/config";
+{{/if}}
+
+export default definePrismaConfig({
+  skills: {
+    agents: ["claude", "cursor", "agents", "devin"],
+  },
+  orm: ormConfig({
+    contract: "./src/prisma/contract.prisma",
+    db: {
+      connection: {{#if hasMongodb}}env.MONGODB_URL{{else}}env.POSTGRES_URL{{/if}},
+    },
+  }),
+});
+{{else}}
+import { defineConfig } from "prisma/config";
 
 export default defineConfig({
-  schema: "src/database/orm/prisma/schema.prisma",
+  schema: "src/database/orm/prisma{{#unless isBetterAuth}}/schema.prisma{{/unless}}",
   datasource: {
-    {{#if hasPostgres}}url: env("POSTGRES_URL"),{{/if}}
-    {{#if hasMysql}}url: env("MYSQL_URL"),{{/if}}
-    {{#if hasSqlite}}url: env("SQLITE_PATH"),{{/if}}
+    {{#if hasPostgres}}url: env.POSTGRES_URL,{{/if}}
+    {{#if hasMysql}}url: env.MYSQL_URL,{{/if}}
+    {{#if hasSqlite}}url: env.SQLITE_PATH,{{/if}}
+    {{#if hasMongodb}}url: env.MONGODB_URL,{{/if}}
   },
 });
+{{/if}}

package/dist/templates/base/src/app.controller.spec.ts.hbs

@@ -1,22 +1,15 @@
 // oxlint-disable vitest/require-mock-type-parameters
 import type { TestingModule } from "@nestjs/testing";
 import { Test } from "@nestjs/testing";
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeEach, describe, expect, it } from "vitest";
 
 import { AppController } from "./app.controller.js";
 import { AppService } from "./app.service.js";
 {{#if (or hasDatabase hasOrm)}}
-import type { ISqlAdapter } from "./database/adapters/interfaces/db-adapter.interface.js";
 import { DbService } from "./database/db.service.js";
 {{/if}}
 
-{{#if (or hasDatabase hasOrm)}}
-const mockSqlAdapter: ISqlAdapter = {
-  execute: vi.fn().mockResolvedValue(null),
-  getClient: vi.fn().mockReturnValue({}),
-  query: vi.fn().mockResolvedValue([]),
-};
-{{/if}}
+
 
 describe(AppController, () => {
   let appController: AppController;
@@ -27,8 +20,7 @@ describe(AppController, () => {
       providers: [
         AppService,
         {{#if (or hasDatabase hasOrm)}}
-        DbService,
-        { provide: "SQL_ADAPTER", useValue: mockSqlAdapter },
+        { provide: DbService, useValue: {} },
         {{/if}}
       ],
     }).compile();

package/dist/templates/base/src/app.controller.ts.hbs

@@ -19,10 +19,10 @@ export class AppController {
   {{/if}}
   {{#if swagger}}
   @ApiOperation({ summary: "/api" })
-  @ApiResponse({ status: 200, description: "Returns a hello world message" })
+  @ApiResponse({ description: "Returns a hello world message", status: 200 })
   {{/if}}
   @Get()
   getHello(): string {
     return this.appService.getHello();
   }
-}
+}

package/dist/templates/base/src/app.module.ts.hbs

@@ -1,5 +1,5 @@
 import { Module } from "@nestjs/common";
-{{#if (and hasAuth hasDatabase)}}
+{{#if (and isPassportAuth hasDatabase)}}
 import { PassportModule } from "@nestjs/passport";
 {{/if}}
 {{#if (or hasDatabase hasOrm)}}
@@ -9,8 +9,10 @@ import { DbModule } from "./database/db.module.js";
 {{#if shouldImportAuthModule}}
 import { AuthModule } from "./auth/auth.module.js";
 {{/if}}
-{{#if (and hasAuth hasDatabase)}}
+{{#if (and isPassportAuth hasDatabase)}}
+{{#if hasGraphql}}
 import { GqlAuthGuard } from "./auth/guards/gql-auth.guard.js";
+{{/if}}
 import { RestAuthGuard } from "./auth/guards/rest-auth.guard.js";
 {{/if}}
 {{#if hasGraphql}}
@@ -55,7 +57,7 @@ import { AppService } from "./app.service.js";
     {{#if shouldImportAuthModule}}
     AuthModule,
     {{/if}}
-    {{#if (and hasAuth hasDatabase)}}
+    {{#if (and isPassportAuth hasDatabase)}}
     PassportModule.register({ defaultStrategy: "jwt" }),
     {{/if}}
     {{#if hasGraphql}}
@@ -73,8 +75,10 @@ import { AppService } from "./app.service.js";
   ],
   providers: [
     AppService,
-    {{#if (and hasAuth hasDatabase)}}
+    {{#if (and isPassportAuth hasDatabase)}}
+    {{#if hasGraphql}}
     GqlAuthGuard,
+    {{/if}}
     RestAuthGuard,
     {{/if}}
     {{#if hasTrpc}}

package/dist/templates/base/src/app.router.ts.hbs

@@ -1,5 +1,5 @@
-import { Router, Query, UseMiddlewares, Ctx } from "nestjs-trpc";
-import z from "zod";
+import { Router, Query{{#if hasTrpcAndAuth}}, UseMiddlewares, Ctx{{/if}} } from "nestjs-trpc";
+import { z } from "zod";
 {{#if hasTrpcAndAuth}}
 import type { UserSession } from "./auth/auth.js";
 import { AuthTRPCMiddleware } from "./libs/middlewares/auth.trpc.middleware.js";

package/dist/templates/base/src/database/adapters/interfaces/db-adapter.interface.ts.hbs

@@ -7,8 +7,15 @@ export interface ISqlAdapter<TClient = unknown> {
 {{/if}}
 
 {{#if hasMongodb}}
+{{#if useRawSql}}
+import type { Collection, Document } from "mongodb";
+{{/if}}
 export interface IMongoAdapter<TClient = unknown> {
+{{#if useRawSql}}
+  collection<T extends Document = Document>(name: string): Collection<T>;
+{{else}}
   collection<T = unknown>(name: string): T;
+{{/if}}
   getClient(): TClient;
 }
 {{/if}}

package/dist/templates/base/src/database/adapters/mongo/prisma-mongo.adapter.ts.hbs

@@ -1,13 +1,36 @@
 import { Injectable, Inject } from '@nestjs/common';
 import { IMongoAdapter } from '../interfaces/db-adapter.interface.js';
+{{#if isPrisma8}}
+{{#if isMonorepo}}
+import { db } from '../../../prisma/db.js';
+{{else}}
+import { db } from '../../prisma/db.js';
+{{/if}}
+{{else}}
 {{#if isMonorepo}}
 import type { PrismaClient } from '../../../database/orm/prisma/generated/client.js';
 {{else}}
 import type { PrismaClient } from '../../orm/prisma/generated/client.js';
 {{/if}}
+{{/if}}
 
 @Injectable()
-export class PrismaMongoAdapter implements IMongoAdapter<PrismaClient> {
+export class PrismaMongoAdapter implements IMongoAdapter<any> {
+{{#if isPrisma8}}
+  readonly client = db;
+
+  collection<T = unknown>(name: string): T {
+    return ((this.client as any).orm as any)?.[name] as T;
+  }
+
+  async runCommand(command: unknown): Promise<unknown> {
+    return (this.client as any).query?.(command);
+  }
+
+  getClient() {
+    return this.client;
+  }
+{{else}}
   constructor(
     @Inject('PRISMA_MONGO_CLIENT') private readonly prisma: PrismaClient
   ) {}
@@ -22,4 +45,5 @@ export class PrismaMongoAdapter implements IMongoAdapter<PrismaClient> {
   getClient(): PrismaClient {
     return this.prisma;
   }
+{{/if}}
 }

package/dist/templates/base/src/database/adapters/mongo/raw-mongo.adapter.ts.hbs

@@ -1,17 +1,22 @@
-import { Injectable, Inject } from '@nestjs/common';
-import { IMongoAdapter } from '../interfaces/db-adapter.interface.js';
+import { Injectable, Inject, type OnModuleInit, type OnModuleDestroy } from "@nestjs/common";
+import { MongoClient, type Collection, type Db, type Document } from "mongodb";
+import type { IMongoAdapter } from "../interfaces/db-adapter.interface.js";
 
 @Injectable()
-export class RawMongoAdapter implements IMongoAdapter {
-  constructor(
-    @Inject('MONGODB_CLIENT') private readonly db: any
-  ) {}
-
-  collection(name: string): any {
-    return this.db.collection(name);
+export class RawMongoAdapter implements IMongoAdapter<Db>, OnModuleInit, OnModuleDestroy {
+  constructor(@Inject("MONGODB_CLIENT") private readonly client: MongoClient) {}
+  collection<T extends Document = Document>(name: string): Collection<T> {
+    return this.client.db().collection<T>(name);
   }
-
-  getClient(): any {
-    return this.db;
+  getClient(): Db { return this.client.db(); }
+  async onModuleInit(): Promise<void> {
+    try {
+      await this.client.connect();
+      await this.client.db().command({ ping: 1 });
+    } catch (error) {
+      await this.client.close();
+      throw error;
+    }
   }
+  async onModuleDestroy(): Promise<void> { await this.client.close(); }
 }

package/dist/templates/base/src/database/adapters/sql/drizzle-sql.adapter.ts.hbs

@@ -21,4 +21,13 @@ export class DrizzleSqlAdapter<TClient = unknown> implements ISqlAdapter<TClient
   getClient(): TClient {
     return this.db;
   }
+
+  async onModuleDestroy(): Promise<void> {
+    const client = (this.db as any).$client;
+    if (typeof client?.end === 'function') {
+      await client.end();
+    } else if (typeof client?.close === 'function') {
+      await client.close();
+    }
+  }
 }

package/dist/templates/base/src/database/adapters/sql/prisma-sql.adapter.ts.hbs

@@ -1,4 +1,3 @@
-// oxlint-disable typescript/no-explicit-any require-await
 import { Injectable, Inject } from '@nestjs/common';
 import { ISqlAdapter } from '../interfaces/db-adapter.interface.js';
 {{#if isMonorepo}}
@@ -15,15 +14,19 @@ export class PrismaSqlAdapter implements ISqlAdapter<PrismaClient> {
     @Inject('PRISMA_SQL_CLIENT') private readonly prisma: PrismaClient
   ) {}
 
-  async query<T = any>(sql: string, params: any[] = []): Promise<T[]> {
-    return this.prisma.$queryRawUnsafe(sql, ...params);
+  query<T = unknown>(sql: string, params: unknown[] = []): Promise<T[]> {
+    return this.prisma.$queryRawUnsafe<T[]>(sql, ...params);
   }
 
-  async execute(sql: string, params: any[] = []): Promise<any> {
+  execute(sql: string, params: unknown[] = []): Promise<number> {
     return this.prisma.$executeRawUnsafe(sql, ...params);
   }
 
   getClient(): PrismaClient {
     return this.prisma;
   }
+
+  async onModuleDestroy(): Promise<void> {
+    await this.prisma.$disconnect();
+  }
 }

package/dist/templates/base/src/database/adapters/sql/raw-sql.adapter.ts.hbs

@@ -1,3 +1,31 @@
+{{#if hasPostgres}}
+import { Inject, Injectable } from "@nestjs/common";
+import type { Pool } from "pg";
+
+import type { ISqlAdapter } from "../interfaces/db-adapter.interface.js";
+
+@Injectable()
+export class RawSqlAdapter implements ISqlAdapter<Pool> {
+  constructor(@Inject("SQL_CLIENT") private readonly db: Pool) {}
+
+  async query<T = unknown>(queryString: string, params: unknown[] = []): Promise<T[]> {
+    const result = await this.db.query(queryString, params);
+    return result.rows as T[];
+  }
+
+  execute(queryString: string, params: unknown[] = []): Promise<unknown> {
+    return this.db.query(queryString, params);
+  }
+
+  getClient(): Pool {
+    return this.db;
+  }
+
+  async onModuleDestroy(): Promise<void> {
+    await this.db.end();
+  }
+}
+{{else}}
 import { Injectable, Inject } from '@nestjs/common';
 import { ISqlAdapter } from '../interfaces/db-adapter.interface.js';
 
@@ -46,4 +74,13 @@ export class RawSqlAdapter implements ISqlAdapter {
   getClient(): any {
     return this.db;
   }
+
+  async onModuleDestroy(): Promise<void> {
+    if (typeof this.db.end === 'function') {
+      await this.db.end();
+    } else if (typeof this.db.close === 'function') {
+      await this.db.close();
+    }
+  }
 }
+{{/if}}

package/dist/templates/base/src/database/adapters/sql/typeorm-sql.adapter.ts.hbs

@@ -1,4 +1,3 @@
-// oxlint-disable typescript/no-explicit-any require-await
 import { Injectable, Inject } from '@nestjs/common';
 import type { DataSource } from 'typeorm';
 import type { ISqlAdapter } from '../interfaces/db-adapter.interface.js';
@@ -9,15 +8,21 @@ export class TypeormSqlAdapter implements ISqlAdapter<DataSource> {
     @Inject('TYPEORM_DATA_SOURCE') private readonly dataSource: DataSource
   ) {}
 
-  async query<T = any>(sql: string, params: any[] = []): Promise<T[]> {
+  query<T = unknown>(sql: string, params: unknown[] = []): Promise<T[]> {
     return this.dataSource.query(sql, params);
   }
 
-  async execute(sql: string, params: any[] = []): Promise<any> {
+  execute(sql: string, params: unknown[] = []): Promise<unknown> {
     return this.dataSource.query(sql, params);
   }
 
   getClient(): DataSource {
     return this.dataSource;
   }
+
+  async onModuleDestroy(): Promise<void> {
+    if (this.dataSource.isInitialized) {
+      await this.dataSource.destroy();
+    }
+  }
 }

package/dist/templates/base/src/database/db.module.ts.hbs

@@ -1,3 +1,15 @@
+{{#if isPrisma8}}
+import { DynamicModule, Global, Module } from "@nestjs/common";
+import { DbService } from "./db.service.js";
+
+@Global()
+@Module({})
+export class DbModule {
+  static forRoot(_options?: unknown): DynamicModule {
+    return { exports: [DbService], module: DbModule, providers: [DbService] };
+  }
+}
+{{else}}
 import { DynamicModule, Module, Global, Provider } from '@nestjs/common';
 import { DbService } from './db.service.js';
 
@@ -87,7 +99,8 @@ export interface DbModuleOptions {
 @Module({})
 export class DbModule {
   static forRoot(options: DbModuleOptions): DynamicModule {
-    const providers: Provider[] = [DbService];
+    const initialProviders: Provider[] = [DbService];
+    const providers = initialProviders;
 
     {{#if usePrisma}}
       {{#if (or hasPostgres hasMysql hasSqlite)}}
@@ -98,7 +111,10 @@ export class DbModule {
     const prismaSql = new PrismaClient({ adapter: sqlAdapter });
         {{/if}}
         {{#if hasMysql}}
-    const sqlAdapter = new PrismaMariaDb(options.mysqlUrl!);
+    if (!options.mysqlUrl) {
+      throw new Error("A MySQL URL is required to initialize Prisma");
+    }
+    const sqlAdapter = new PrismaMariaDb(options.mysqlUrl);
     const prismaSql = new PrismaClient({ adapter: sqlAdapter });
         {{/if}}
         {{#if hasSqlite}}
@@ -137,39 +153,20 @@ export class DbModule {
     {{/if}}
 
     {{#if useDrizzle}}
-      {{#if hasPostgres}}
-    const pool = new Pool({
-      connectionString: options.postgresUrl,
-    });
-    const drizzleClient = drizzle({ client: pool, schema });
-    providers.push({
-      provide: 'DRIZZLE_CLIENT',
-      useValue: drizzleClient,
-    });
-      {{/if}}
-      {{#if hasMysql}}
-    const pool = mysql.createPool({
-      uri: options.mysqlUrl,
-    });
-    const drizzleClient = drizzle({ client: pool, schema });
-    providers.push({
-      provide: 'DRIZZLE_CLIENT',
-      useValue: drizzleClient,
-    });
-      {{/if}}
-      {{#if hasSqlite}}
-    const sqliteDb = new Database(options.sqlitePath || ':memory:');
-    const drizzleClient = drizzle({ client: sqliteDb, schema });
     providers.push({
       provide: 'DRIZZLE_CLIENT',
-      useValue: drizzleClient,
-    });
-      {{/if}}
-
-    providers.push({
-      provide: 'SQL_ADAPTER',
-      useClass: DrizzleSqlAdapter,
+      useFactory: () => {
+        {{#if hasPostgres}}
+        const client = new Pool({ connectionString: options.postgresUrl });
+        {{else if hasMysql}}
+        const client = mysql.createPool({ uri: options.mysqlUrl });
+        {{else}}
+        const client = new Database(options.sqlitePath || ':memory:');
+        {{/if}}
+        return drizzle({ client, schema });
+      },
     });
+    providers.push({ provide: 'SQL_ADAPTER', useClass: DrizzleSqlAdapter });
     {{/if}}
 
     {{#if useTypeorm}}
@@ -220,11 +217,10 @@ export class DbModule {
       {{/if}}
 
       {{#if hasMongodb}}
-    const mongoClient = new MongoClient(options.mongodbUrl || '');
-    const mongoDb = mongoClient.db();
+    if (!options.mongodbUrl) throw new Error("A MongoDB URL is required");
     providers.push({
       provide: 'MONGODB_CLIENT',
-      useValue: mongoDb,
+      useFactory: () => new MongoClient(options.mongodbUrl!, { serverSelectionTimeoutMS: 5000 }),
     });
     providers.push({
       provide: 'MONGO_ADAPTER',
@@ -240,3 +236,4 @@ export class DbModule {
     };
   }
 }
+{{/if}}