From e27f42e620c171c39db8a8e989a9f078fd22b297 Mon Sep 17 00:00:00 2001
From: tada5hi <peter.placzek1996@gmail.com>
Date: Sun, 11 Aug 2024 20:45:15 +0200
Subject: [PATCH] fix: adjust data source cleanup behaviour to jsdoc
 description

---
 src/database/check.ts | 49 +++++++++++++++++--------------------------
 src/database/type.ts  | 11 ++++++++--
 2 files changed, 28 insertions(+), 32 deletions(-)

diff --git a/src/database/check.ts b/src/database/check.ts
index e33c0d52..f880b98b 100644
--- a/src/database/check.ts
+++ b/src/database/check.ts
@@ -1,8 +1,7 @@
+import type { DataSourceOptions } from 'typeorm';
 import { DataSource, MigrationExecutor } from 'typeorm';
 import {
     hasDataSource,
-    setDataSource,
-    unsetDataSource,
     useDataSource,
     useDataSourceOptions,
 } from '../data-source';
@@ -15,39 +14,37 @@ import type { DatabaseCheckContext, DatabaseCheckResult } from './type';
  * @param context
  */
 export async function checkDatabase(context: DatabaseCheckContext = {}) : Promise<DatabaseCheckResult> {
-    context.dataSourceCleanup = context.dataSourceCleanup ?? true;
-
     const result : DatabaseCheckResult = {
         exists: true,
         schema: false,
         migrationsPending: [],
     };
 
-    let { dataSource } = context;
+    let dataSource : DataSource;
+    let dataSourceCleanup : boolean;
 
     if (
-        typeof dataSource === 'undefined' &&
+        typeof context.dataSource === 'undefined' &&
+        typeof context.options === 'undefined' &&
         hasDataSource(context.alias)
     ) {
         // todo: data-source might get initialized here
-        dataSource = await useDataSource(context.alias);
-    }
 
-    const dataSourceExisted = !!dataSource;
-
-    if (typeof dataSource === 'undefined') {
+        dataSource = await useDataSource(context.alias);
+        dataSourceCleanup = false;
+    } else {
+        let dataSourceOptions : DataSourceOptions;
         if (context.options) {
-            dataSource = new DataSource({
-                ...context.options,
-                synchronize: false,
-            });
+            dataSourceOptions = context.options;
         } else {
-            const options = await useDataSourceOptions(context.alias);
-            dataSource = new DataSource({
-                ...options,
-                synchronize: false,
-            });
+            dataSourceOptions = await useDataSourceOptions(context.alias);
         }
+
+        dataSource = new DataSource({
+            ...dataSourceOptions,
+            synchronize: true,
+        });
+        dataSourceCleanup = context.dataSourceCleanup ?? true;
     }
 
     try {
@@ -103,16 +100,8 @@ export async function checkDatabase(context: DatabaseCheckContext = {}) : Promis
 
     await queryRunner.release();
 
-    if (!dataSourceExisted) {
-        if (context.dataSourceCleanup) {
-            await dataSource.destroy();
-
-            if (!context.dataSource) {
-                unsetDataSource(context.alias);
-            }
-        } else {
-            setDataSource(dataSource, context.alias);
-        }
+    if (dataSourceCleanup) {
+        await dataSource.destroy();
     }
 
     return result;
diff --git a/src/database/type.ts b/src/database/type.ts
index 21459d7f..da6c4143 100644
--- a/src/database/type.ts
+++ b/src/database/type.ts
@@ -22,7 +22,14 @@ export type DatabaseBaseContext = {
     initialDatabase?: string,
 };
 
-export type DatabaseCheckContext = Omit<DatabaseBaseContext, 'initialDatabase'> & {
+export type DatabaseCheckContext = {
+    /**
+     * Options for finding the typeorm DataSource.
+     *
+     * Default: undefined
+     */
+    options?: DataSourceOptions,
+
     /**
      * Use alias to access already registered DataSource / DataSourceOptions.
      *
@@ -32,7 +39,7 @@ export type DatabaseCheckContext = Omit<DatabaseBaseContext, 'initialDatabase'>
 
     /**
      * Indicates whether to destroy the data-source
-     * afterwards or not.
+     * afterward or not.
      * If a datasource previously existed, this option will be ignored.
      *
      * default: true