feat(badges): manual time set

1 files changed, 13 insertions, 6 deletions

diff --git a/src/lib/Database/badges.ts b/src/lib/Database/badges.ts
index e02fd91b..9f857469 100644
--- a/src/lib/Database/badges.ts
+++ b/src/lib/Database/badges.ts
@@ -5,6 +5,7 @@ export interface Badge {
 	image?: string;
 	description?: string;
 	id?: number;
+	time?: string;
 }
 
 const database = new Database('./data/due_moe.sqlite3', {
@@ -31,15 +32,21 @@ export const getUserBadges = (userId: number): Badge[] => {
 };
 
 export const addUserBadge = (userId: number, badge: Badge) => {
-	const { post, image, description } = badge;
+	const { post, image, description, time } = badge;
 
 	if (post === undefined || image === undefined) return;
 
-	const statement = database.query(
-		`INSERT INTO user_badges (user_id, post, image, description) VALUES (?1, ?2, ?3, ?4);`
-	);
-
-	statement.run(userId.toString(), post, image, description || null);
+	if (time) {
+		database
+			.query(
+				`INSERT INTO user_badges (user_id, post, image, description, time) VALUES (?1, ?2, ?3, ?4, ?5);`
+			)
+			.run(userId.toString(), post, image, description || null, time);
+	} else {
+		database
+			.query(`INSERT INTO user_badges (user_id, post, image, description) VALUES (?1, ?2, ?3, ?4);`)
+			.run(userId.toString(), post, image, description || null);
+	}
 };
 
 export const removeUserBadge = (userId: number, id: number) => {