From 4ddf4eab79c47cf3f9e6cea60c69ebef631ba1fc Mon Sep 17 00:00:00 2001 From: Fuwn Date: Sat, 30 Dec 2023 00:33:14 -0800 Subject: feat(badges): manual time set --- src/lib/Database/badges.ts | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) (limited to 'src/lib/Database') 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) => { -- cgit v1.2.3