Live test fixes (#807)

* fixes

* changes

src/routes/v3/project_creation.rs

@@ -612,11 +612,14 @@ async fn project_create_inner(
             additional_categories.extend(ids.values());
         }
 
+        // Should only be owner if not attached to an organization
+        let is_owner = project_create_data.organization_id.is_none();
+
         let team = models::team_item::TeamBuilder {
             members: vec![models::team_item::TeamMemberBuilder {
                 user_id: current_user.id.into(),
                 role: crate::models::teams::OWNER_ROLE.to_owned(),
-                is_owner: true,
+                is_owner,
                 // Allow all permissions for project creator, even if attached to a project
                 permissions: ProjectPermissions::all(),
                 organization_permissions: None,

src/routes/v3/projects.rs

@@ -54,6 +54,7 @@ pub fn config(cfg: &mut web::ServiceConfig) {
             .route("{id}/gallery", web::delete().to(delete_gallery_item))
             .route("{id}/follow", web::post().to(project_follow))
             .route("{id}/follow", web::delete().to(project_unfollow))
+            .route("{id}/organization", web::get().to(project_get_organization))
             .service(
                 web::scope("{project_id}")
                     .route(
@@ -2184,3 +2185,38 @@ pub async fn project_unfollow(
         ))
     }
 }
+
+pub async fn project_get_organization(
+    req: HttpRequest,
+    info: web::Path<(String,)>,
+    pool: web::Data<PgPool>,
+    redis: web::Data<RedisPool>,
+    session_queue: web::Data<AuthQueue>,
+) -> Result<HttpResponse, ApiError> {
+    let user = get_user_from_headers(&req, &**pool, &redis, &session_queue, None)
+        .await?
+        .1;
+    let string = info.into_inner().0;
+
+    let result = db_models::Project::get(&string, &**pool, &redis)
+        .await?
+        .ok_or_else(|| {
+            ApiError::InvalidInput("The specified project does not exist!".to_string())
+        })?;
+
+    if is_authorized(&result.inner, &Some(user), &pool).await? {
+        Err(ApiError::InvalidInput(
+            "The specified project does not exist!".to_string(),
+        ))
+    } else if let Some(organization_id) = result.inner.organization_id {
+        let organization = db_models::Organization::get_id(organization_id, &**pool, &redis)
+            .await?
+            .ok_or_else(|| {
+                ApiError::InvalidInput("The specified organization does not exist!".to_string())
+            })?;
+
+        Ok(HttpResponse::Ok().json(organization))
+    } else {
+        Err(ApiError::NotFound)
+    }
+}

src/routes/v3/teams.rs

@@ -62,28 +62,14 @@ pub async fn team_members_get_project(
         if !is_authorized(&project.inner, &current_user, &pool).await? {
             return Err(ApiError::NotFound);
         }
-        let mut members_data =
+        let members_data =
             TeamMember::get_from_team_full(project.inner.team_id, &**pool, &redis).await?;
-        let mut member_user_ids = members_data.iter().map(|x| x.user_id).collect::<Vec<_>>();
-
-        // Adds the organization's team members to the list of members, if the project is associated with an organization
-        if let Some(oid) = project.inner.organization_id {
-            let organization_data = Organization::get_id(oid, &**pool, &redis).await?;
-            if let Some(organization_data) = organization_data {
-                let org_team =
-                    TeamMember::get_from_team_full(organization_data.team_id, &**pool, &redis)
-                        .await?;
-                for member in org_team {
-                    if !member_user_ids.contains(&member.user_id) {
-                        member_user_ids.push(member.user_id);
-                        members_data.push(member);
-                    }
-                }
-            }
-        }
-
-        let users =
-            crate::database::models::User::get_many_ids(&member_user_ids, &**pool, &redis).await?;
+        let users = crate::database::models::User::get_many_ids(
+            &members_data.iter().map(|x| x.user_id).collect::<Vec<_>>(),
+            &**pool,
+            &redis,
+        )
+        .await?;
 
         let user_id = current_user.as_ref().map(|x| x.id.into());
 
@@ -94,6 +80,7 @@ pub async fn team_members_get_project(
                     .find(|x| x.user_id == user.id.into() && x.accepted)
             })
             .is_some();
+
         let team_members: Vec<_> = members_data
             .into_iter()
             .filter(|x| {
@@ -109,6 +96,7 @@ pub async fn team_members_get_project(
                 })
             })
             .collect();
+
         Ok(HttpResponse::Ok().json(team_members))
     } else {
         Err(ApiError::NotFound)