From c0e4dc886eaba03485ea4aaac37eaffb311aeb74 Mon Sep 17 00:00:00 2001 From: Fuwn Date: Tue, 2 Jan 2024 21:47:28 -0800 Subject: feat(badges): categories --- src/lib/Database/badges.ts | 25 ++++++++++++++++++------- 1 file changed, 18 insertions(+), 7 deletions(-) (limited to 'src/lib/Database') 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 + ); }; -- cgit v1.2.3