feat: update models and add unique viewers endpoint

src/endpoints/mod.rs

@@ -13,3 +13,4 @@ pub mod leaderboard;
 pub mod quest_boost;
 pub mod quests;
 pub mod get_boosted_quests;
+pub mod unique_page_visit;

src/endpoints/unique_page_visit.rs

@@ -0,0 +1,46 @@
+use std::net::SocketAddr;
+use crate::{
+    models::{AppState},
+    utils::get_error,
+};
+use axum::{
+    extract::{Query, State, ConnectInfo},
+    http::StatusCode,
+    response::IntoResponse,
+    Json,
+};
+use axum_auto_routes::route;
+use mongodb::bson::doc;
+use serde::Deserialize;
+use std::sync::Arc;
+use chrono::Utc;
+use mongodb::Collection;
+use mongodb::options::UpdateOptions;
+use serde_json::json;
+use crate::models::UniquePageVisit;
+
+#[derive(Deserialize)]
+pub struct GetQuestsQuery {
+    page_id: String,
+}
+
+#[route(get, "/unique_page_visit", crate::endpoints::unique_page_visit)]
+pub async fn handler(
+    State(state): State<Arc<AppState>>,
+    Query(query): Query<GetQuestsQuery>,
+    ConnectInfo(addr): ConnectInfo<SocketAddr>,
+) -> impl IntoResponse {
+    let id = query.page_id;
+    let unique_viewers_collection: Collection<UniquePageVisit> =
+        state.db.collection("unique_viewers");
+    let created_at = Utc::now().timestamp_millis();
+    let filter = doc! { "viewer_ip": addr.to_string(), "viewed_page_id": &id };
+    let update = doc! { "$setOnInsert": { "viewer_ip": addr.to_string(), "viewed_page_id": &id,"timestamp":created_at } };
+    let options = UpdateOptions::builder().upsert(true).build();
+
+    match unique_viewers_collection.update_one(filter, update, options)
+        .await {
+        Ok(_) => (StatusCode::OK, Json(json!({"res": true}))).into_response(),
+        Err(_) => get_error("unable to detect page visit status".to_string()),
+    }
+}

src/main.rs

@@ -64,6 +64,7 @@ async fn main() {
             acc.merge(r.to_router(shared_state.clone()))
         })
         .layer(cors);
+
     let addr = SocketAddr::from(([0, 0, 0, 0], conf.server.port));
     println!("server: listening on http://0.0.0.0:{}", conf.server.port);
     axum::Server::bind(&addr)

src/models.rs

@@ -46,12 +46,14 @@ pub_struct!(Debug, Serialize, Deserialize; QuestDocument {
 pub_struct!(Deserialize; CompletedTasks {
     address: String,
     task_id: u32,
+    timestamp: i64,
 });
 
 #[derive(Debug, Serialize, Deserialize)]
 pub struct CompletedTaskDocument {
     address: String,
     task_id: u32,
+    timestamp: i64,
 }
 
 pub_struct!(Serialize; Reward {
@@ -80,6 +82,12 @@ pub_struct!(Deserialize; VerifyQuizQuery {
     user_answers_list: Vec<Vec<String>>,
 });
 
+pub_struct!(Deserialize; UniquePageVisit {
+    viewer_ip: String,
+    viewed_page_id: String,
+    timestamp: i64,
+});
+
 pub_struct!(Deserialize; AchievementQuery {
     addr: FieldElement,
 });
@@ -92,6 +100,7 @@ pub_struct!(Deserialize; VerifyAchievementQuery {
 pub_struct!(Debug, Serialize, Deserialize; AchievedDocument {
     addr: String,
     achievement_id: u32,
+    timestamp:i64,
 });
 
 pub_struct!(Debug, Serialize, Deserialize; AchievementDocument {

src/utils.rs

@@ -125,8 +125,9 @@ impl CompletedTasksTrait for AppState {
     ) -> Result<UpdateResult, mongodb::error::Error> {
         let completed_tasks_collection: Collection<CompletedTasks> =
             self.db.collection("completed_tasks");
+        let created_at = Utc::now().timestamp_millis();
         let filter = doc! { "address": addr.to_string(), "task_id": task_id };
-        let update = doc! { "$setOnInsert": { "address": addr.to_string(), "task_id": task_id } };
+        let update = doc! { "$setOnInsert": { "address": addr.to_string(), "task_id": task_id , "timestamp":created_at} };
         let options = UpdateOptions::builder().upsert(true).build();
 
         let result = completed_tasks_collection
@@ -240,7 +241,7 @@ impl CompletedTasksTrait for AppState {
                             experience.into(),
                             timestamp,
                         )
-                        .await;
+                            .await;
                     }
                     Err(_e) => {
                         get_error("Error querying quests".to_string());
@@ -292,9 +293,10 @@ impl AchievementsTrait for AppState {
         achievement_id: u32,
     ) -> Result<UpdateResult, mongodb::error::Error> {
         let achieved_collection: Collection<CompletedTasks> = self.db.collection("achieved");
+        let created_at = Utc::now().timestamp_millis();
         let filter = doc! { "addr": addr.to_string(), "achievement_id": achievement_id };
         let update =
-            doc! { "$setOnInsert": { "addr": addr.to_string(), "achievement_id": achievement_id } };
+            doc! { "$setOnInsert": { "addr": addr.to_string(), "achievement_id": achievement_id , "timestamp":created_at } };
         let options = UpdateOptions::builder().upsert(true).build();
 
         let result = achieved_collection
@@ -329,7 +331,7 @@ impl AchievementsTrait for AppState {
                     experience.into(),
                     timestamp,
                 )
-                .await;
+                    .await;
             }
             None => {}
         }