implement collection, org creation hard caps

Author: aecsocket

apps/labrinth/src/routes/v3/collections.rs

@@ -2,6 +2,7 @@ use crate::auth::checks::is_visible_collection;
 use crate::auth::{filter_visible_collections, get_user_from_headers};
 use crate::database::models::{
     collection_item, generate_collection_id, project_item,
+    user_limits::UserLimits,
 };
 use crate::database::redis::RedisPool;
 use crate::file_hosting::{FileHost, FileHostPublicity};
@@ -76,6 +77,11 @@ pub async fn collection_create(
     .await?
     .1;
 
+    let limits = UserLimits::get(&current_user, &client).await?;
+    if limits.current.collections + 1 >= limits.max.collections {
+        return Err(CreateError::LimitReached);
+    }
+
     collection_create_data.validate().map_err(|err| {
         CreateError::InvalidInput(validation_errors_to_string(err, None))
     })?;

apps/labrinth/src/routes/v3/organizations.rs

@@ -7,6 +7,7 @@ use crate::auth::{filter_visible_projects, get_user_from_headers};
 use crate::database::models::team_item::DBTeamMember;
 use crate::database::models::{
     DBOrganization, generate_organization_id, team_item,
+    user_limits::UserLimits,
 };
 use crate::database::redis::RedisPool;
 use crate::file_hosting::{FileHost, FileHostPublicity};
@@ -136,6 +137,11 @@ pub async fn organization_create(
     .await?
     .1;
 
+    let limits = UserLimits::get(&current_user, &pool).await?;
+    if limits.current.organizations + 1 >= limits.max.organizations {
+        return Err(CreateError::LimitReached);
+    }
+
     new_organization.validate().map_err(|err| {
         CreateError::ValidationError(validation_errors_to_string(err, None))
     })?;