-
Notifications
You must be signed in to change notification settings - Fork 27
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #201 from starknet-id/ayush/nostra
feat: nostra quest staking
- Loading branch information
Showing
14 changed files
with
288 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
pub mod claimable; | ||
pub mod discord_fw_callback; | ||
pub mod verify_added_liquidity; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,2 @@ | ||
pub mod claimable; | ||
pub mod discord_fw_callback; | ||
pub mod verify_added_liquidity; | ||
pub mod liquidity_quest; | ||
pub mod staking_quest; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,107 @@ | ||
use crate::models::{AppState, CompletedTaskDocument, Reward, RewardResponse}; | ||
use crate::utils::{get_error, get_nft}; | ||
use axum::{ | ||
extract::{Query, State}, | ||
http::StatusCode, | ||
response::IntoResponse, | ||
Json, | ||
}; | ||
use axum_auto_routes::route; | ||
use futures::StreamExt; | ||
use mongodb::bson::doc; | ||
use serde::Deserialize; | ||
use starknet::{ | ||
core::types::FieldElement, | ||
signers::{LocalWallet, SigningKey}, | ||
}; | ||
use std::sync::Arc; | ||
|
||
const QUEST_ID: u32 = 27; | ||
const TASK_IDS: &[u32] = &[132, 133, 134]; | ||
const LAST_TASK: u32 = TASK_IDS[2]; | ||
const NFT_LEVEL: u32 = 39; | ||
|
||
#[derive(Deserialize)] | ||
pub struct ClaimableQuery { | ||
addr: FieldElement, | ||
} | ||
|
||
#[route( | ||
get, | ||
"/quests/nostra/staking_quest/claimable", | ||
crate::endpoints::quests::nostra::staking_quest::claimable | ||
)] | ||
pub async fn handler( | ||
State(state): State<Arc<AppState>>, | ||
Query(query): Query<ClaimableQuery>, | ||
) -> impl IntoResponse { | ||
let collection = state | ||
.db | ||
.collection::<CompletedTaskDocument>("completed_tasks"); | ||
|
||
let pipeline = vec![ | ||
doc! { | ||
"$match": { | ||
"address": &query.addr.to_string(), | ||
"task_id": { "$in": TASK_IDS }, | ||
}, | ||
}, | ||
doc! { | ||
"$lookup": { | ||
"from": "tasks", | ||
"localField": "task_id", | ||
"foreignField": "id", | ||
"as": "task", | ||
}, | ||
}, | ||
doc! { | ||
"$match": { | ||
"task.quest_id": QUEST_ID, | ||
}, | ||
}, | ||
doc! { | ||
"$group": { | ||
"_id": "$address", | ||
"completed_tasks": { "$push": "$task_id" }, | ||
}, | ||
}, | ||
doc! { | ||
"$match": { | ||
"completed_tasks": { "$all": TASK_IDS }, | ||
}, | ||
}, | ||
]; | ||
|
||
let completed_tasks = collection.aggregate(pipeline, None).await; | ||
match completed_tasks { | ||
Ok(mut tasks_cursor) => { | ||
if tasks_cursor.next().await.is_none() { | ||
return get_error("User hasn't completed all tasks".into()); | ||
} | ||
|
||
let signer = LocalWallet::from(SigningKey::from_secret_scalar( | ||
state.conf.nft_contract.private_key, | ||
)); | ||
|
||
let mut rewards = vec![]; | ||
|
||
let Ok((token_id, sig)) = get_nft(QUEST_ID, LAST_TASK, &query.addr, NFT_LEVEL, &signer).await else { | ||
return get_error("Signature failed".into()); | ||
}; | ||
|
||
rewards.push(Reward { | ||
task_id: LAST_TASK, | ||
nft_contract: state.conf.nft_contract.address.clone(), | ||
token_id: token_id.to_string(), | ||
sig: (sig.r, sig.s), | ||
}); | ||
|
||
if rewards.is_empty() { | ||
get_error("No rewards found for this user".into()) | ||
} else { | ||
(StatusCode::OK, Json(RewardResponse { rewards })).into_response() | ||
} | ||
} | ||
Err(_) => get_error("Error querying rewards".into()), | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
pub mod claimable; | ||
pub mod verify_twitter_tw; | ||
pub mod verify_stake; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,79 @@ | ||
use std::sync::Arc; | ||
|
||
use crate::{ | ||
models::{AppState, VerifyQuery}, | ||
utils::{get_error, CompletedTasksTrait}, | ||
}; | ||
use axum::{ | ||
extract::{Query, State}, | ||
http::StatusCode, | ||
response::IntoResponse, | ||
Json, | ||
}; | ||
use axum_auto_routes::route; | ||
use serde_json::json; | ||
use starknet::{ | ||
core::types::{BlockId, BlockTag, FieldElement, FunctionCall}, | ||
macros::selector, | ||
providers::Provider, | ||
}; | ||
|
||
#[route( | ||
get, | ||
"/quests/nostra/staking_quest/verify_stake", | ||
crate::endpoints::quests::nostra::staking_quest::verify_stake | ||
)] | ||
pub async fn handler( | ||
State(state): State<Arc<AppState>>, | ||
Query(query): Query<VerifyQuery>, | ||
) -> impl IntoResponse { | ||
let task_id = 133; | ||
let addr = &query.addr; | ||
let balance_calldata = vec![*addr]; | ||
let balance_result = state | ||
.provider | ||
.call( | ||
FunctionCall { | ||
contract_address: state.conf.quests.nostra.staking_contract, | ||
entry_point_selector: selector!("balance_of"), | ||
calldata: balance_calldata, | ||
}, | ||
BlockId::Tag(BlockTag::Latest), | ||
) | ||
.await; | ||
|
||
let user_balance = match &balance_result { | ||
Ok(result) => result[0], | ||
Err(e) => return get_error(format!("{}", e)), | ||
}; | ||
|
||
if user_balance == FieldElement::ZERO { | ||
return get_error("You didn't stake any STRK.".to_string()); | ||
} | ||
|
||
let call_result = state | ||
.provider | ||
.call( | ||
FunctionCall { | ||
contract_address: state.conf.quests.nostra.staking_contract, | ||
entry_point_selector: selector!("convert_to_assets"), | ||
calldata: balance_result.unwrap().to_vec(), | ||
}, | ||
BlockId::Tag(BlockTag::Latest), | ||
) | ||
.await; | ||
|
||
match call_result { | ||
Ok(result) => { | ||
if result[0] < FieldElement::from_dec_str("10").unwrap() { | ||
get_error("You need to stake atleast 10 STRK".to_string()) | ||
} else { | ||
match state.upsert_completed_task(query.addr, task_id).await { | ||
Ok(_) => (StatusCode::OK, Json(json!({"res": true}))).into_response(), | ||
Err(e) => get_error(format!("{}", e)), | ||
} | ||
} | ||
} | ||
Err(e) => get_error(format!("{}", e)), | ||
} | ||
} |
30 changes: 30 additions & 0 deletions
30
src/endpoints/quests/nostra/staking_quest/verify_twitter_tw.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,30 @@ | ||
use std::sync::Arc; | ||
|
||
use crate::{ | ||
models::{AppState, VerifyQuery}, | ||
utils::{get_error, CompletedTasksTrait}, | ||
}; | ||
use axum::{ | ||
extract::{Query, State}, | ||
http::StatusCode, | ||
response::IntoResponse, | ||
Json, | ||
}; | ||
use axum_auto_routes::route; | ||
use serde_json::json; | ||
|
||
#[route( | ||
get, | ||
"/quests/nostra/staking_quest/verify_twitter_tw", | ||
crate::endpoints::quests::nostra::staking_quest::verify_twitter_tw | ||
)] | ||
pub async fn handler( | ||
State(state): State<Arc<AppState>>, | ||
Query(query): Query<VerifyQuery>, | ||
) -> impl IntoResponse { | ||
let task_id = 134; | ||
match state.upsert_completed_task(query.addr, task_id).await { | ||
Ok(_) => (StatusCode::OK, Json(json!({"res": true}))).into_response(), | ||
Err(e) => get_error(format!("{}", e)), | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters