refactor(be): env validation & parsing

krystxf · Oct 9, 2024 · 6abdc5c · 6abdc5c
1 parent 8567a59
commit 6abdc5c
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 7 deletions.
diff --git a/apps/backend/src/app.module.ts b/apps/backend/src/app.module.ts
@@ -11,22 +11,27 @@ import { GraphQLModule } from "@nestjs/graphql";
 import { ScheduleModule } from "@nestjs/schedule";
 
 import { GRAPHQL_API_ROOT } from "src/constants/graphql.const";
+import { Environment } from "src/enums/environment.enum";
 import { LoggerMiddleware } from "src/middleware/logger.middleware";
 import { DepartureModule } from "src/modules/departure/departure.module";
 import { ImportModule } from "src/modules/import/import.module";
 import { PlatformModule } from "src/modules/platform/platform.module";
 import { PrismaModule } from "src/modules/prisma/prisma.module";
 import { StopModule } from "src/modules/stop/stop.module";
+import { envSchema } from "src/schema/env.schema";
 
 @Module({
     imports: [
         PlatformModule,
         DepartureModule,
-        ImportModule,
+        ...(process.env.NODE_ENV === Environment.TEST ? [] : [ImportModule]),
         StopModule,
 
         PrismaModule,
-        ConfigModule.forRoot(),
+        ConfigModule.forRoot({
+            envFilePath: ".env.local",
+            validate: envSchema.parse,
+        }),
         ScheduleModule.forRoot(),
         CacheModule.register({
             isGlobal: true,

diff --git a/apps/backend/src/enums/environment.enum.ts b/apps/backend/src/enums/environment.enum.ts
@@ -0,0 +1,3 @@
+export enum Environment {
+    TEST = "test",
+}
diff --git a/apps/backend/src/main.ts b/apps/backend/src/main.ts
@@ -13,10 +13,6 @@ import {
 async function bootstrap() {
     const app = await NestFactory.create(AppModule);
 
-    if (process.env.GOLEMIO_API_KEY === undefined) {
-        throw new Error("GOLEMIO_API_KEY is not set");
-    }
-
     const config = new DocumentBuilder()
         .setTitle(SWAGGER_TITLE)
         .setDescription(SWAGGER_DESCRIPTION)
@@ -28,6 +24,6 @@ async function bootstrap() {
         customSiteTitle: SWAGGER_TITLE,
     });
 
-    await app.listen(3001);
+    await app.listen(process.env.PORT ?? 3001);
 }
 bootstrap();
diff --git a/apps/backend/src/modules/prisma/prisma.service.ts b/apps/backend/src/modules/prisma/prisma.service.ts
@@ -3,6 +3,11 @@ import { PrismaClient } from "@prisma/client";
 
 @Injectable()
 export class PrismaService extends PrismaClient implements OnModuleInit {
+    constructor() {
+        super({
+            datasourceUrl: `postgresql://${process.env.POSTGRES_USER}:${process.env.POSTGRES_PASSWORD}@${process.env.DB_HOST}:${process.env.DB_PORT}/${process.env.POSTGRES_DB}?schema=${process.env.DB_SCHEMA}`,
+        });
+    }
     async onModuleInit() {
         await this.$connect();
     }

diff --git a/apps/backend/src/schema/env.schema.ts b/apps/backend/src/schema/env.schema.ts
@@ -0,0 +1,12 @@
+import { z } from "zod";
+
+export const envSchema = z.object({
+    GOLEMIO_API_KEY: z.string(),
+    POSTGRES_USER: z.string(),
+    POSTGRES_PASSWORD: z.string(),
+    POSTGRES_DB: z.string(),
+    DB_HOST: z.string(),
+    DB_PORT: z.coerce.number().int().positive(),
+    DB_SCHEMA: z.string(),
+    PORT: z.coerce.number().int().positive().optional(),
+});