ethanaobrien/ew
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add server side friend limits

Browse source
This commit is contained in:
Ethan O'Brien 2024-05-10 18:18:01 -05:00
parent 2ce9aa4868
commit 38a5377b10
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/router/friend.rs
View file

@ -4,6 +4,8 @@ use actix_web::{HttpResponse, HttpRequest};
use crate::router::{userdata, global}; use crate::router::{userdata, global};
use crate::encryption; use crate::encryption;
pub const FRIEND_LIMIT: usize = 40;
pub fn friend(req: HttpRequest, body: String) -> HttpResponse { pub fn friend(req: HttpRequest, body: String) -> HttpResponse {
let key = global::get_login(req.headers(), &body); let key = global::get_login(req.headers(), &body);
let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap(); let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
@ -101,7 +103,7 @@ pub fn request(req: HttpRequest, body: String) -> HttpResponse {
let uid = body["user_id"].as_i64().unwrap(); let uid = body["user_id"].as_i64().unwrap();
if !userdata::friend_request_disabled(uid) { if !userdata::friend_request_disabled(uid) {
if !friends["request_user_id_list"].contains(uid) { if !friends["request_user_id_list"].contains(uid) && friends["request_user_id_list"].len() < FRIEND_LIMIT {
friends["request_user_id_list"].push(uid).unwrap(); friends["request_user_id_list"].push(uid).unwrap();
userdata::save_acc_friends(&key, friends); userdata::save_acc_friends(&key, friends);
} }
@ -126,7 +128,7 @@ pub fn approve(req: HttpRequest, body: String) -> HttpResponse {
let index = friends["pending_user_id_list"].members().into_iter().position(|r| *r.to_string() == uid.to_string()); let index = friends["pending_user_id_list"].members().into_iter().position(|r| *r.to_string() == uid.to_string());
if !index.is_none() { if !index.is_none() {
friends["pending_user_id_list"].array_remove(index.unwrap()); friends["pending_user_id_list"].array_remove(index.unwrap());
if body["approve"].to_string() == "1" && ! friends["friend_user_id_list"].contains(uid) { if body["approve"].to_string() == "1" && ! friends["friend_user_id_list"].contains(uid) && friends["friend_user_id_list"].len() < FRIEND_LIMIT {
friends["friend_user_id_list"].push(uid).unwrap(); friends["friend_user_id_list"].push(uid).unwrap();
} }

4
src/router/userdata/mod.rs
View file

@ -400,7 +400,7 @@ pub fn friend_request(uid: i64, requestor: i64) {
let uid = get_uid(&login_token); let uid = get_uid(&login_token);
let friends = DATABASE.lock_and_select("SELECT friends FROM friends WHERE user_id=?1", params!(uid)); let friends = DATABASE.lock_and_select("SELECT friends FROM friends WHERE user_id=?1", params!(uid));
let mut friends = json::parse(&friends.unwrap()).unwrap(); let mut friends = json::parse(&friends.unwrap()).unwrap();
if !friends["pending_user_id_list"].contains(requestor) { if !friends["pending_user_id_list"].contains(requestor) && friends["pending_user_id_list"].len() < crate::router::friend::FRIEND_LIMIT {
friends["pending_user_id_list"].push(requestor).unwrap(); friends["pending_user_id_list"].push(requestor).unwrap();
DATABASE.lock_and_exec("UPDATE friends SET friends=?1 WHERE user_id=?2", params!(json::stringify(friends), uid)); DATABASE.lock_and_exec("UPDATE friends SET friends=?1 WHERE user_id=?2", params!(json::stringify(friends), uid));
} }
@ -422,7 +422,7 @@ pub fn friend_request_approve(uid: i64, requestor: i64, accepted: bool, key: &st
if !index.is_none() { if !index.is_none() {
friends["request_user_id_list"].array_remove(index.unwrap()); friends["request_user_id_list"].array_remove(index.unwrap());
} }
if accepted && !friends["friend_user_id_list"].contains(requestor) { if accepted && !friends["friend_user_id_list"].contains(requestor) && friends["friend_user_id_list"].len() < crate::router::friend::FRIEND_LIMIT {
friends["friend_user_id_list"].push(requestor).unwrap(); friends["friend_user_id_list"].push(requestor).unwrap();
} }
DATABASE.lock_and_exec("UPDATE friends SET friends=?1 WHERE user_id=?2", params!(json::stringify(friends), uid)); DATABASE.lock_and_exec("UPDATE friends SET friends=?1 WHERE user_id=?2", params!(json::stringify(friends), uid));