feat(badges): categories

1 files changed, 18 insertions, 7 deletions

diff --git a/src/lib/Database/badges.ts b/src/lib/Database/badges.ts
index 1f3267e9..9819e10e 100644
--- a/src/lib/Database/badges.ts
+++ b/src/lib/Database/badges.ts
@@ -7,6 +7,7 @@ export interface Badge {
 	description?: string;
 	id?: number;
 	time?: string;
+	category?: string;
 }
 
 const database = new Database('./data/due_moe.sqlite3', {
@@ -21,6 +22,7 @@ database
 	post TEXT(1000) NOT NULL,
 	image TEXT(1000) NOT NULL,
 	description TEXT(1000) DEFAULT NULL,
+	category TEXT(1000) DEFAULT NULL,
 	time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
 )`
 	)
@@ -39,20 +41,22 @@ export const getUserBadges = (userId: number): Badge[] => {
 };
 
 export const addUserBadge = (userId: number, badge: Badge) => {
-	const { post, image, description, time } = badge;
+	const { post, image, description, time, category } = badge;
 
 	if (post === undefined || image === undefined) return;
 
 	if (time) {
 		database
 			.query(
-				`INSERT INTO user_badges (user_id, post, image, description, time) VALUES (?1, ?2, ?3, ?4, ?5);`
+				`INSERT INTO user_badges (user_id, post, image, description, time, category) VALUES (?1, ?2, ?3, ?4, ?5, ?6);`
 			)
-			.run(userId.toString(), post, image, description || null, time);
+			.run(userId.toString(), post, image, description || null, time, category || null);
 	} else {
 		database
-			.query(`INSERT INTO user_badges (user_id, post, image, description) VALUES (?1, ?2, ?3, ?4);`)
-			.run(userId.toString(), post, image, description || null);
+			.query(
+				`INSERT INTO user_badges (user_id, post, image, description, category) VALUES (?1, ?2, ?3, ?4, ?5);`
+			)
+			.run(userId.toString(), post, image, description || null, category || null);
 	}
 };
 
@@ -70,7 +74,14 @@ export const updateUserBadge = (userId: number, id: number, badge: Badge) => {
 
 	database
 		.query(
-			'UPDATE user_badges SET post = ?1, image = ?2, description = ?3 WHERE id = ?4 AND user_id = ?5'
+			'UPDATE user_badges SET post = ?1, image = ?2, description = ?3, category = ?6 WHERE id = ?4 AND user_id = ?5'
 		)
-		.run(badge.post || null, badge.image || null, badge.description || null, id, userId);
+		.run(
+			badge.post || null,
+			badge.image || null,
+			badge.description || null,
+			id,
+			userId,
+			badge.category || null
+		);
 };