From a1bf2a12a02c1e152e3bcfb6d6594c46d66eff60 Mon Sep 17 00:00:00 2001
From: zkldi <20380519+zkldi@users.noreply.github.com>
Date: Wed, 10 Aug 2022 23:51:38 +0100
Subject: [PATCH] fix: winston logging callback didn't work

---
 server/src/utils/misc.ts | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/server/src/utils/misc.ts b/server/src/utils/misc.ts
index d48b990aa..cdfb1c1b4 100644
--- a/server/src/utils/misc.ts
+++ b/server/src/utils/misc.ts
@@ -265,12 +265,20 @@ export function IsRecord(maybeRecord: unknown): maybeRecord is Record<string, un
  * Wrap a promise in an error handler that exits the process safely, and logs
  * when it completes.
  */
-export function WrapScriptPromise(promise: Promise<unknown>, logger: KtLogger) {
-	promise
-		.then(() => logger.info(`Finished executing.`, () => process.exit(0)))
-		.catch((err: unknown) => {
-			logger.error(`Failed executing.`, { err }, () => {
-				process.exit(1);
-			});
-		});
+export async function WrapScriptPromise(promise: Promise<unknown>, logger: KtLogger) {
+	let code = 0;
+
+	try {
+		await promise;
+
+		logger.info(`Finished executing.`);
+	} catch (err) {
+		logger.error(`Failed executing.`, { err });
+
+		code = 1;
+	}
+
+	logger.end(() => {
+		process.exit(code);
+	});
 }