Hotfix/db (#100)

* fix(db): fix db config file

* fix(db): load ssl config only for digital ocean db

Author: AmiraKhalid04

src/databases/data-source.ts

@@ -15,10 +15,12 @@ import { TweetCategory } from '../tweets/entities/tweet-category.entity';
 import { Chat } from '../chat/entities/chat.entity';
 import { Message } from '../messages/entities/message.entity';
 import { readFileSync } from 'fs';
+
 config({ path: resolve(__dirname, '../../config/.env') });
+
 const config_service = new ConfigService();
 
-export default new DataSource({
+const base_config: any = {
     type: 'postgres',
     host: process.env.POSTGRES_HOST || config_service.get<string>('POSTGRES_HOST'),
     username: process.env.POSTGRES_USERNAME || config_service.get<string>('POSTGRES_USERNAME'),
@@ -28,6 +30,7 @@ export default new DataSource({
         parseInt(process.env.POSTGRES_PORT || '5432') ||
         config_service.get<number>('POSTGRES_PORT') ||
         5432,
+
     entities: [
         User,
         Verification,
@@ -48,10 +51,16 @@ export default new DataSource({
         Chat,
         Message,
     ],
-    ssl: {
-        ca: readFileSync(process.env.DATABASE_CA!).toString(), // Path to the CA certificate
-    },
+
     migrations: ['src/migrations/*{.ts,.js}'],
     synchronize: false,
     uuidExtension: 'pgcrypto',
-});
+};
+
+if (process.env.DATABASE_CA) {
+    base_config.ssl = {
+        ca: readFileSync(process.env.DATABASE_CA).toString(),
+    };
+}
+
+export default new DataSource(base_config);

src/databases/postgresql.module.ts

@@ -1,29 +1,41 @@
 import { Module } from '@nestjs/common';
 import { ConfigModule, ConfigService } from '@nestjs/config';
 import { TypeOrmModule } from '@nestjs/typeorm';
-import { readFileSync, writeFileSync } from 'fs';
+import { existsSync, readFileSync } from 'fs';
 
 @Module({
     imports: [
         ConfigModule,
         TypeOrmModule.forRootAsync({
             imports: [ConfigModule],
             inject: [ConfigService],
-            useFactory: (config_service: ConfigService) => ({
-                type: 'postgres',
-                host: config_service.get<string>('POSTGRES_HOST'),
-                username: config_service.get<string>('POSTGRES_USERNAME'),
-                password: config_service.get<string>('POSTGRES_PASSWORD'),
-                database: config_service.get<string>('POSTGRES_DB'),
-                port: config_service.get<number>('POSTGRES_PORT'),
-                synchronize: false, // Using migrations instead
-                autoLoadEntities: true,
-                // logging: ['query'],
-                // logger: 'advanced-console',
-                ssl: {
-                    ca: readFileSync(process.env.DATABASE_CA!).toString(), // Path to the CA certificate
-                },
-            }),
+            useFactory: (config_service: ConfigService) => {
+                // SSL CONFIG
+                let ssl: any;
+
+                const ca_path = process.env.DATABASE_CA;
+
+                if (ca_path) {
+                    if (existsSync(ca_path)) {
+                        ssl = {
+                            ca: readFileSync(ca_path).toString(),
+                        };
+                    }
+                }
+
+                return {
+                    type: 'postgres',
+                    host: config_service.get<string>('POSTGRES_HOST'),
+                    username: config_service.get<string>('POSTGRES_USERNAME'),
+                    password: config_service.get<string>('POSTGRES_PASSWORD'),
+                    database: config_service.get<string>('POSTGRES_DB'),
+                    port: config_service.get<number>('POSTGRES_PORT'),
+                    synchronize: false, // Using migrations instead
+                    autoLoadEntities: true,
+
+                    ssl,
+                };
+            },
         }),
     ],
     providers: [ConfigService],