Implement friend endpoints

src/router/friend.rs

@@ -1,21 +1,147 @@
-use json;
-use json::object;
+use json::{object, array};
 use crate::router::global;
 use actix_web::{HttpResponse, HttpRequest};
-//use crate::router::userdata;
+use crate::router::userdata;
+use crate::encryption;
 
-pub fn friend(_req: HttpRequest, _body: String) -> HttpResponse {
-    /*let blank_header = HeaderValue::from_static("");
+pub fn friend(req: HttpRequest, body: String) -> HttpResponse {
+    let key = global::get_login(req.headers(), &body);
+    let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
+    let friends = userdata::get_acc_friends(&key);
     
-    let key = global::get_login(req.headers());
-    let user = userdata::get_acc(&key, uid);*/
+    let mut rv = array![];
+    
+    let rv_data = if body["status"].as_i32().unwrap() == 3 {
+        friends["friend_user_id_list"].clone()
+    } else if body["status"].as_i32().unwrap() == 2 {
+        friends["pending_user_id_list"].clone()
+    } else if body["status"].as_i32().unwrap() == 1 {
+        friends["request_user_id_list"].clone()
+    } else {
+        array![]
+    };
+    
+    for (_i, uid) in rv_data.members().enumerate() {
+        let mut to_push = global::get_user(uid.as_i64().unwrap());
+        to_push["status"] = body["status"].clone();
+        rv.push(to_push).unwrap();
+    }
     
     let resp = object!{
         "code": 0,
         "server_time": global::timestamp(),
         "data": {
-            "friend_list": [] //todo - pull from userdata
+            "friend_list": rv
         }
     };
     global::send(resp)
 }
+
+pub fn ids(req: HttpRequest) -> HttpResponse {
+    let key = global::get_login(req.headers(), "");
+    let friends = userdata::get_acc_friends(&key);
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": friends
+    };
+    global::send(resp)
+}
+
+pub fn recommend(_req: HttpRequest, _body: String) -> HttpResponse {
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": {
+            friend_list: []
+        }
+    };
+    global::send(resp)
+}
+
+pub fn search(_req: HttpRequest, body: String) -> HttpResponse {
+    let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
+    
+    let uid = body["user_id"].as_i64().unwrap();
+    let user = global::get_user(uid);
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": user
+    };
+    global::send(resp)
+}
+
+pub fn request(req: HttpRequest, body: String) -> HttpResponse {
+    let key = global::get_login(req.headers(), &body);
+    let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
+    let user_id = userdata::get_acc(&key)["user"]["id"].as_i64().unwrap();
+    let mut friends = userdata::get_acc_friends(&key);
+    
+    let uid = body["user_id"].as_i64().unwrap();
+    if !userdata::friend_request_disabled(uid) {
+        if !friends["request_user_id_list"].contains(uid) { 
+            friends["request_user_id_list"].push(uid).unwrap(); 
+            userdata::save_acc_friends(&key, friends);
+        }
+        userdata::friend_request(uid, user_id);
+    }
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": []
+    };
+    global::send(resp)
+}
+
+pub fn approve(req: HttpRequest, body: String) -> HttpResponse {
+    let key = global::get_login(req.headers(), &body);
+    let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
+    let user_id = userdata::get_acc(&key)["user"]["id"].as_i64().unwrap();
+    let mut friends = userdata::get_acc_friends(&key);
+    
+    let uid = body["user_id"].as_i64().unwrap();
+    let index = friends["request_user_id_list"].members().into_iter().position(|r| *r.to_string() == uid.to_string());
+    if !index.is_none() {
+        friends["request_user_id_list"].array_remove(index.unwrap());
+    }
+    if body["approve"].to_string() == "1" && ! friends["friend_user_id_list"].contains(uid) {
+        friends["friend_user_id_list"].push(uid).unwrap();
+    }
+    
+    userdata::friend_request_approve(uid, user_id, body["approve"].to_string() == "1");
+    userdata::save_acc_friends(&key, friends);
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": []
+    };
+    global::send(resp)
+}
+
+pub fn cancel(req: HttpRequest, body: String) -> HttpResponse {
+    let key = global::get_login(req.headers(), &body);
+    let body = json::parse(&encryption::decrypt_packet(&body).unwrap()).unwrap();
+    let user_id = userdata::get_acc(&key)["user"]["id"].as_i64().unwrap();
+    let mut friends = userdata::get_acc_friends(&key);
+    
+    let uid = body["user_id"].as_i64().unwrap();
+    let index = friends["request_user_id_list"].members().into_iter().position(|r| *r.to_string() == uid.to_string());
+    if !index.is_none() {
+        friends["request_user_id_list"].array_remove(index.unwrap());
+    }
+    userdata::friend_request_approve(uid, user_id, false);
+    userdata::save_acc_friends(&key, friends);
+    
+    let resp = object!{
+        "code": 0,
+        "server_time": global::timestamp(),
+        "data": []
+    };
+    global::send(resp)
+}