Merge pull request #163 from starknet-id/ayush/verify-quest-acheivement

feat : verify quest achievements

src/endpoints/achievements/claim_nft/mod.rs

@@ -0,0 +1 @@
+pub mod quests_achievement;

src/endpoints/achievements/claim_nft/quests_achievement.rs

@@ -0,0 +1,166 @@
+use std::sync::Arc;
+
+use crate::{
+    models::{AppState, VerifyAchievementQuery},
+    utils::{get_error},
+};
+use axum::{
+    extract::{Query, State},
+    http::StatusCode,
+    response::IntoResponse,
+    Json,
+};
+use futures::TryStreamExt;
+use mongodb::bson::{doc, Document};
+use starknet::core::types::FieldElement;
+use starknet::signers::{LocalWallet, SigningKey};
+use crate::models::{Reward};
+use crate::utils::get_nft;
+
+const QUEST_ID: u32 = 25;
+const NFT_LEVEL: u32 = 1;
+
+
+fn get_number_of_quests(id: u32) -> u32 {
+    return match id {
+        23 => 1,
+        24 => 3,
+        25 => 10,
+        26 => 25,
+        27 => 50,
+        _ => 0,
+    };
+}
+
+fn get_task_id(id: u32) -> u32 {
+    return match id {
+        23 => 1,
+        24 => 3,
+        25 => 10,
+        26 => 25,
+        27 => 50,
+        _ => 0,
+    };
+}
+
+pub async fn handler(
+    State(state): State<Arc<AppState>>,
+    Query(query): Query<VerifyAchievementQuery>,
+) -> impl IntoResponse {
+    let addr = query.addr;
+    if addr == FieldElement::ZERO {
+        return get_error("Please connect your wallet first".to_string());
+    }
+
+    let achievement_id = query.id;
+    let quests_threshold = get_number_of_quests(achievement_id);
+
+    // check valid achievement id
+    if !(23..=27).contains(&achievement_id) {
+        return get_error("Invalid achievement id".to_string());
+    }
+
+    let pipeline = vec![
+        doc! {
+            "$match": doc! {
+                "address": addr.to_string()
+            }
+        },
+        doc! {
+            "$lookup": doc! {
+                "from": "tasks",
+                "localField": "task_id",
+                "foreignField": "id",
+                "as": "associatedTask"
+            }
+        },
+        doc! {
+            "$unwind": "$associatedTask"
+        },
+        doc! {
+            "$group": doc! {
+                "_id": "$associatedTask.quest_id",
+                "done": doc! {
+                    "$sum": 1
+                }
+            }
+        },
+        doc! {
+            "$lookup": doc! {
+                "from": "tasks",
+                "localField": "_id",
+                "foreignField": "quest_id",
+                "as": "tasks"
+            }
+        },
+        doc! {
+            "$match": doc! {
+                "$expr": doc! {
+                    "$eq": [
+                        "$done",
+                        doc! {
+                            "$size": "$tasks"
+                        }
+                    ]
+                }
+            }
+        },
+        doc! {
+        "$group": doc! {
+            "_id": null,
+            "count": doc! {
+                "$sum": 1
+            }
+        }
+    },
+        doc! {
+        "$addFields": doc! {
+            "result": doc! {
+                "$cond": doc! {
+                    "if": doc! {
+                        "$gte": [
+                            "$count",
+                            quests_threshold
+                        ]
+                    },
+                    "then": true,
+                    "else": false
+                }
+            }
+        }
+    },
+    ];
+    let tasks_collection = state.db.collection::<Document>("completed_tasks");
+
+    match tasks_collection.aggregate(pipeline, None).await {
+        Ok(mut cursor) => {
+            let mut res = false;
+            while let Some(result) = cursor.try_next().await.unwrap() {
+                res = result.get("result").unwrap().as_bool().unwrap();
+            }
+            if !res {
+                return get_error("User hasn't completed required number of quests".into());
+            }
+
+            let signer = LocalWallet::from(SigningKey::from_secret_scalar(
+                state.conf.nft_contract.private_key,
+            ));
+
+            let task_id = get_task_id(achievement_id);
+
+            let Ok((token_id, sig)) = get_nft(QUEST_ID, task_id, &query.addr, NFT_LEVEL, &signer).await else {
+                return get_error("Signature failed".into());
+            };
+
+
+            let rewards = Reward {
+                task_id,
+                nft_contract: state.conf.nft_contract.address.clone(),
+                token_id: token_id.to_string(),
+                sig: (sig.r, sig.s),
+            };
+            (StatusCode::OK, Json(rewards)).into_response()
+        }
+        Err(_) => get_error("Error querying quests".to_string()),
+    }
+}

src/endpoints/achievements/mod.rs

@@ -9,3 +9,5 @@ pub mod verify_has_domain;
 pub mod verify_seniority;
 pub mod verify_tvl;
 pub mod verify_whitelisted;
+pub mod verify_quests;
+pub mod claim_nft;