1 files changed, 29 insertions, 32 deletions

diff --git a/src/lib/userBadgesDatabase.ts b/src/lib/userBadgesDatabase.ts
index 78655cd3..8cf40d5b 100644
--- a/src/lib/userBadgesDatabase.ts
+++ b/src/lib/userBadgesDatabase.ts
@@ -1,57 +1,54 @@
-import { dev } from '$app/environment';
-import Database from 'better-sqlite3';
+import { Database } from 'bun:sqlite';
 
 export interface Badge {
-	post: string;
-	image: string;
+	post?: string;
+	image?: string;
 	description?: string;
 	id?: number;
 }
 
 const database = new Database('./data/due_moe.sqlite3', {
-	verbose: dev ? console.log : undefined
+	create: true
 });
 
-database.exec(`create table if not exists user_badges (
-	id integer primary key, 
-	user_id integer not null, 
-	post text(1000) not null,
-	image text(1000) not null,
-	description text(1000) default null,
-	time timestamp default current_timestamp
-)`);
+database
+	.query(
+		`CREATE TABLE IF NOT EXISTS user_badges (
+	id INTEGER PRIMARY KEY, 
+	user_id INTEGER NOT NULL, 
+	post TEXT(1000) NOT NULL,
+	image TEXT(1000) NOT NULL,
+	description TEXT(1000) DEFAULT NULL,
+	time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+)`
+	)
+	.run();
 
 export const getUserBadges = (userId: number): Badge[] => {
 	return database
-		.prepare('select * from user_badges where user_id = ?')
-		.all(String(userId)) as Badge[];
+		.query('SELECT * FROM user_badges WHERE user_id = $userId')
+		.all({ $userId: String(userId) }) as Badge[];
 };
 
 export const addUserBadge = (userId: number, badge: Badge) => {
 	const { post, image, description } = badge;
-	const statement = database.prepare(`
-    insert into user_badges (user_id, post, image, description) values (?, ?, ?, ?)
-  `);
 
-	statement.run(userId, post, image, description);
-};
-
-export const addUserBadges = (userId: number, badges: Badge[]) => {
-	const statement = database.prepare(`
-		insert into user_badges (user_id, post, image, description) values (?, ?, ?, ?)
-	`);
+	if (post === undefined || image === undefined) {
+		return;
+	}
 
-	for (const badge of badges) {
-		const { post, image, description } = badge;
+	const statement = database.query(
+		`INSERT INTO user_badges (user_id, post, image, description) VALUES (?1, ?2, ?3, ?4);`
+	);
 
-		statement.run(userId, post, image, description);
-	}
+	statement.run([userId, post, image, description]);
 };
 
 export const removeUserBadge = (userId: number, id: number) => {
 	if (!isNaN(id)) {
-		const statement = database.prepare('delete from user_badges where user_id = ? and id = ?');
-
-		statement.run(userId, id);
+		database.query('DELETE FROM user_badges WHERE user_id = $userId AND id = $id').run({
+			$userId: userId,
+			$id: id
+		});
 	}
 };