From 197f1aacd2dad264cf0cb415efeb57d1ea08c849 Mon Sep 17 00:00:00 2001
From: Samuel <34555296+Flam3rboy@users.noreply.github.com>
Date: Fri, 17 Mar 2023 19:18:31 +0100
Subject: feat: better-sqlite3 support

---
 src/util/util/Database.ts | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

(limited to 'src/util')

diff --git a/src/util/util/Database.ts b/src/util/util/Database.ts
index 64d7ca14..e625ea9b 100644
--- a/src/util/util/Database.ts
+++ b/src/util/util/Database.ts
@@ -36,9 +36,17 @@ if (!process.env) {
 const dbConnectionString =
 	process.env.DATABASE || path.join(process.cwd(), "database.db");
 
-const DatabaseType = dbConnectionString.includes("://")
+let DatabaseType = dbConnectionString.includes("://")
 	? dbConnectionString.split(":")[0]?.replace("+srv", "")
 	: "sqlite";
+
+if (DatabaseType === "sqlite") {
+	try {
+		require("better-sqlite3");
+		DatabaseType = "better-sqlite3";
+		// eslint-disable-next-line no-empty
+	} catch (error) {}
+}
 const isSqlite = DatabaseType.includes("sqlite");
 
 const DataSourceOptions = new DataSource({
@@ -55,6 +63,7 @@ const DataSourceOptions = new DataSource({
 	supportBigNumbers: true,
 	name: "default",
 	migrations: [path.join(__dirname, "..", "migration", DatabaseType, "*.js")],
+	cache: true,
 });
 
 // Gets the existing database connection
@@ -77,7 +86,13 @@ export async function initDatabase(): Promise<DataSource> {
 	}
 
 	if (!process.env.DB_SYNC) {
-		const supported = ["mysql", "mariadb", "postgres", "sqlite"];
+		const supported = [
+			"mysql",
+			"mariadb",
+			"postgres",
+			"sqlite",
+			"better-sqlite3",
+		];
 		if (!supported.includes(DatabaseType)) {
 			console.log(
 				"[Database]" +
-- 
cgit 1.4.1