Merge pull request #108 from starknet-id/feat/disable_achievements

feat: enable/disable achievements
lfglabs-dev · Oct 25, 2023 · f573cac · f573cac
2 parents 66c0f95 + 65efbee
commit f573cac
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/src/endpoints/achievements/fetch.rs b/src/endpoints/achievements/fetch.rs
@@ -56,6 +56,7 @@ pub async fn handler(
             "category_desc": "$desc",
             "category_img_url": "$img_url",
             "category_type": "$type",
+            "category_disabled": "$disabled",
             "achievements": {
               "id": "$achievement.id",
               "name": "$achievement.name",
@@ -82,7 +83,13 @@ pub async fn handler(
         },
         doc! {
           "$group": {
-            "_id": { "category_name": "$category_name", "category_desc": "$category_desc", "category_img_url": "$category_img_url", "category_type": "$category_type" },
+            "_id": {
+              "category_name": "$category_name",
+              "category_desc": "$category_desc",
+              "category_img_url": "$category_img_url",
+              "category_type": "$category_type",
+              "category_disabled": "$category_disabled",
+            },
             "achievements": { "$push": "$achievements" }
           }
         },
@@ -92,6 +99,7 @@ pub async fn handler(
             "category_desc": "$_id.category_desc",
             "category_img_url": "$_id.category_img_url",
             "category_type": "$_id.category_type",
+            "category_disabled": "$_id.category_disabled",
             "achievements": 1,
             "_id": 0
           }
@@ -105,7 +113,9 @@ pub async fn handler(
                 match result {
                     Ok(document) => {
                         if let Ok(achievement) = from_document::<UserAchievements>(document) {
-                            achievements.push(achievement);
+                            if !achievement.category_disabled {
+                                achievements.push(achievement);
+                            }
                         }
                     }
                     _ => continue,

diff --git a/src/models.rs b/src/models.rs
@@ -106,13 +106,20 @@ pub_struct!(Debug, Serialize, Deserialize; AchievementCategoryDocument {
     img_url: String,
 });
 
-pub_struct!(Debug, Serialize, Deserialize; UserAchievements {
+#[derive(Debug, Serialize, Deserialize)]
+pub struct UserAchievements {
     category_name: String,
     category_desc: String,
     category_img_url: String,
     category_type: String,
+    #[serde(default = "default_category_disabled")]
+    pub category_disabled: bool,
     achievements: Vec<UserAchievement>,
-});
+}
+
+pub fn default_category_disabled() -> bool {
+    false
+}
 
 pub_struct!(Debug, Serialize, Deserialize; UserAchievement {
     id: u32,