From e6132809d27164077c2bca7f4cebdcea5e03eff0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Garc=C3=ADa?= <dani1861994@hotmail.com>
Date: Tue, 24 Apr 2018 23:04:10 +0200
Subject: [PATCH] Implemented delete user from org and added checks to the
 already existing org methods

---
 src/api/core/organizations.rs | 68 +++++++++++++++++++++++++++--------
 1 file changed, 54 insertions(+), 14 deletions(-)

diff --git a/src/api/core/organizations.rs b/src/api/core/organizations.rs
index 2a00fcb0..bb35569b 100644
--- a/src/api/core/organizations.rs
+++ b/src/api/core/organizations.rs
@@ -81,7 +81,10 @@ fn get_org_details(data: OrgIdData, headers: Headers, conn: DbConn) -> JsonResul
 
 #[get("/organizations/<org_id>/users")]
 fn get_org_users(org_id: String, headers: Headers, conn: DbConn) -> JsonResult {
-    // TODO Check if user in org
+    let user_org = match UserOrganization::find_by_user_and_org(&headers.user.uuid, &org_id, &conn) {
+        Some(user_org) => user_org,
+        None => err!("User isn't member of organization")
+    }
 
     let users = UserOrganization::find_by_org(&org_id, &conn);
     let users_json: Vec<Value> = users.iter().map(|c| c.to_json_details(&conn)).collect();
@@ -141,32 +144,46 @@ struct InviteData {
 fn send_invite(org_id: String, data: Json<InviteData>, headers: Headers, conn: DbConn) -> EmptyResult {
     let data: InviteData = data.into_inner();
 
-    // TODO Check that user is in org and admin or more
+    let user_org = match UserOrganization::find_by_user_and_org(&headers.user.uuid, &org_id, &conn) {
+        Some(user_org) => user_org,
+        None => err!("User isn't member of organization")
+    };
+
+    if user_org.type_ == UserOrgType::User {
+        err!("Users can't invite other people. Ask an Admin or Owner")
+    }
+
+    if type_ != UserOrgType::User && user_org.type_ != UserOrgType::Owner {
+        err!("Only Owners can invite Admins or Owners")
+    }
 
     for user_opt in data.emails.iter().map(|email| User::find_by_mail(email, &conn)) {
         match user_opt {
             None => err!("User email does not exist"),
             Some(user) => {
-                // TODO Check that user is not already in org
+                match UserOrganization::find_by_user_and_org(&user.uuid, &org_id, &conn) {
+                    Some(_) => err!("User already in organization"),
+                    None => ()
+                }
 
-                let mut user_org = UserOrganization::new(
+                let mut new_user_org = UserOrganization::new(
                     user.uuid, org_id.clone());
 
                 if data.accessAll {
-                    user_org.access_all = data.accessAll;
+                    new_user_org.access_all = data.accessAll;
                 } else {
                     err!("Select collections unimplemented")
                     // TODO create Users_collections
                 }
 
-                user_org.type_ = match data.type_.as_ref() {
+                new_user_org.type_ = match data.type_.as_ref() {
                     "Owner" => UserOrgType::Owner,
                     "Admin" => UserOrgType::Admin,
                     "User" => UserOrgType::User,
                     _ => err!("Invalid type")
                 } as i32;
 
-                user_org.save(&conn);
+                new_user_org.save(&conn);
             }
         }
     }
@@ -176,7 +193,18 @@ fn send_invite(org_id: String, data: Json<InviteData>, headers: Headers, conn: D
 
 #[post("/organizations/<org_id>/users/<user_id>/confirm", data = "<data>")]
 fn confirm_invite(org_id: String, user_id: String, data: Json<Value>, headers: Headers, conn: DbConn) -> EmptyResult {
-    // TODO Check that user is in org and admin or more
+    let user_org = match UserOrganization::find_by_user_and_org(&headers.user.uuid, &org_id, &conn) {
+        Some(user_org) => user_org,
+        None => err!("User isn't member of organization")
+    };
+
+    if user_org.type_ == UserOrgType::User {
+        err!("Users can't confirm other people. Ask an Admin or Owner")
+    }
+
+    if type_ != UserOrgType::User && user_org.type_ != UserOrgType::Owner {
+        err!("Only Owners can confirm Admins or Owners")
+    }
 
     let mut user_org = match UserOrganization::find_by_user_and_org(
         &user_id, &org_id, &conn) {
@@ -201,12 +229,24 @@ fn confirm_invite(org_id: String, user_id: String, data: Json<Value>, headers: H
 
 #[post("/organizations/<org_id>/users/<user_id>/delete")]
 fn delete_user(org_id: String, user_id: String, headers: Headers, conn: DbConn) -> EmptyResult {
-    // TODO Check that user is in org and admin or more
-    // TODO To delete a user you need either:
-    //      - To be yourself
-    //      - To be of a superior type (ex. Owner can delete Admin and User, Admin can delete User)
+    let user_org = match UserOrganization::find_by_user_and_org(&headers.user.uuid, &org_id, &conn) {
+        Some(user_org) => user_org,
+        None => err!("User isn't member of organization")
+    };
 
-    // Delete users_organizations and users_collections from this org
+    if user_org.type_ == UserOrgType::User {
+        err!("Users can't delete other people. Ask an Admin or Owner")
+    }
 
-    unimplemented!();
+    if type_ != UserOrgType::User && user_org.type_ != UserOrgType::Owner {
+        err!("Only Owners can delete Admins or Owners")
+    }
+
+    // TODO Don't delete the last owner
+
+    user_org.delete(&conn);
+
+    // TODO Delete  users_collections from this org
+
+    Ok(())
 }
\ No newline at end of file