feat: add hidden protection to friend add route

Testaustime · Mar 12, 2024 · da61703 · da61703
1 parent 54003c3
commit da61703
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/docs/APISPEC.md b/docs/APISPEC.md
@@ -980,7 +980,8 @@ curl --request POST 'https://api.testaustime.fi/friends/add' \
             "project_name": "My Project",
             "language": "javascript",
             "editor_name": "vscode",
-            "hostname": "mylaptop"
+            "hostname": "mylaptop",
+            "hidden": false
         }
     }
 }
@@ -1035,7 +1036,8 @@ curl --request GET ''https://api.testaustime.fi/friends/list' \
                 "project_name": "My Project",
                 "language": "javascript",
                 "editor_name": "vscode",
-                "hostname": "mylaptop"
+                "hostname": "mylaptop",
+                "hidden": true
             }
         }
     }

diff --git a/src/api/friends.rs b/src/api/friends.rs
@@ -39,8 +39,11 @@ pub async fn add_friend(
                 username: friend.username.clone(),
                 coding_time: db.get_coding_time_steps(friend.id).await,
                 status: heartbeats.get(&friend.id).map(|heartbeat| {
-                    let (inner_heartbeat, start_time, duration) = heartbeat.to_owned();
+                    let (mut inner_heartbeat, start_time, duration) = heartbeat.to_owned();
                     drop(heartbeat);
+                    if inner_heartbeat.hidden == Some(true) {
+                        inner_heartbeat.project_name = Some("".to_string());
+                    }
                     CurrentActivity {
                         started: start_time,
                         duration: duration.num_seconds(),