feat: Add administrator permissions to PermissionsEnum and update related team management functionality

This commit is contained in:
MythEclipse
2025-09-25 20:57:31 +07:00
parent dd4a73e360
commit 4ecdcd12d2
6 changed files with 150 additions and 155 deletions
@@ -30,6 +30,14 @@ pub enum PermissionsEnum {
ReadListTeams,
ReadDetailTeams,
// Administrator permissions
ManageAllUsers,
ManageAllRoles,
ManageAllPermissions,
ManageAllTeams,
ViewAllSensitiveData,
AccessAdminDashboard,
// Gacha permissions
CreateGachaClaims,
ReadDetailGachaClaims,
@@ -2,8 +2,7 @@ use crate::{AppState, MetaRequestDto};
use crate::{
MessageResponseDto, ResponseListSuccessDto, ResponseSuccessDto,
TeamsCreateRequestDto, TeamsUpdateRequestDto, TeamInviteRequestDto,
TeamAcceptInvitationRequestDto, TeamMemberDto, TeamsSearchQueryDto,
AdminTeamsListItemDto, AdminTeamsDetailItemDto, PermissionsEnum
TeamMemberDto, AdminTeamsListItemDto, AdminTeamsDetailItemDto, PermissionsEnum
};
use axum::response::Response;
use axum::extract::Path;
+2 -4
View File
@@ -5,16 +5,14 @@ pub mod teams_repository;
pub mod teams_schema;
pub mod teams_service;
pub use admin_teams_controller::*;
pub use teams_controller::*;
pub use admin_teams_controller::{admin_teams_router, get_all_teams as admin_get_all_teams, get_team_by_id as admin_get_team_by_id, get_team_members as admin_get_team_members, create_team as admin_create_team, update_team as admin_update_team, delete_team as admin_delete_team, invite_team_members as admin_invite_team_members};
pub use teams_controller::{teams_router as user_teams_router, get_team_list, get_team_by_id as user_get_team_by_id, get_team_members as user_get_team_members};
pub use teams_dto::*;
pub use teams_repository::*;
pub use teams_schema::*;
pub use teams_service::*;
use axum::Router;
use crate::teams_controller::*;
use crate::admin_teams_controller::*;
pub fn teams_router() -> Router {
Router::new()
+103 -123
View File
@@ -48,70 +48,50 @@ where
}
#[utoipa::path(
get,
security(
("Bearer" = [])
),
path = "/v1/teams",
params(
("page" = Option<i64>, Query, description = "Page number"),
("per_page" = Option<i64>, Query, description = "Items per page"),
("search" = Option<String>, Query, description = "Search keyword"),
("sort_by" = Option<String>, Query, description = "Sort by field"),
("order" = Option<String>, Query, description = "Order ASC or DESC"),
("filter" = Option<String>, Query, description = "Filter value"),
("filter_by" = Option<String>, Query, description = "Field to filter by"),
),
responses(
(status = 200, description = "Get team list", body = ResponseListSuccessDto<Vec<TeamsListItemDto>>),
(status = 200, description = "Get public team list", body = ResponseListSuccessDto<Vec<PublicTeamsListItemDto>>)
),
tag = "Teams"
get,
security(
("Bearer" = [])
),
path = "/v1/teams",
params(
("page" = Option<i64>, Query, description = "Page number"),
("per_page" = Option<i64>, Query, description = "Items per page"),
("search" = Option<String>, Query, description = "Search keyword"),
("sort_by" = Option<String>, Query, description = "Sort by field"),
("order" = Option<String>, Query, description = "Order ASC or DESC"),
("filter" = Option<String>, Query, description = "Filter value"),
("filter_by" = Option<String>, Query, description = "Field to filter by"),
),
responses(
(status = 200, description = "Get team list", body = ResponseListSuccessDto<Vec<TeamsListItemDto>>),
(status = 200, description = "Get public team list", body = ResponseListSuccessDto<Vec<PublicTeamsListItemDto>>)
),
tag = "Teams"
)]
pub async fn get_team_list(
headers: Option<HeaderMap>,
Extension(state): Extension<AppState>,
axum::extract::Query(meta): axum::extract::Query<MetaRequestDto>,
) -> Response {
let state = state;
match headers {
Some(headers) => {
match permissions_guard(headers, axum::Extension(state.clone()), vec![]).await {
Ok((_claims, state)) => TeamsService::get_team_list(&state, meta).await,
Err(_) => TeamsService::get_public_team_list(&state, meta).await,
}
},
None => TeamsService::get_public_team_list(&state, meta).await,
}
Extension(state): Extension<AppState>,
axum::extract::Query(meta): axum::extract::Query<MetaRequestDto>,
) -> impl IntoResponse {
TeamsService::get_public_team_list(&state, meta).await
}
#[utoipa::path(
get,
path = "/v1/teams/{id}",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get team by ID", body = ResponseSuccessDto<TeamsDetailItemDto>),
(status = 200, description = "Get public team by ID", body = ResponseSuccessDto<PublicTeamsDetailItemDto>)
),
tag = "Teams"
get,
path = "/v1/teams/{id}",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get team by ID", body = ResponseSuccessDto<TeamsDetailItemDto>),
(status = 200, description = "Get public team by ID", body = ResponseSuccessDto<PublicTeamsDetailItemDto>)
),
tag = "Teams"
)]
pub async fn get_team_by_id(
headers: Option<HeaderMap>,
Extension(state): Extension<AppState>,
Path(id): Path<String>,
) -> Response {
let state = state;
match headers {
Some(headers) => {
match permissions_guard(headers, axum::Extension(state.clone()), vec![]).await {
Ok((_claims, state)) => TeamsService::get_team_by_id(&state, id).await,
Err(_) => TeamsService::get_public_team_by_id(&state, id).await,
}
},
None => TeamsService::get_public_team_by_id(&state, id).await,
}
Extension(state): Extension<AppState>,
Path(id): Path<String>,
) -> impl IntoResponse {
TeamsService::get_public_team_by_id(&state, id).await
}
#[utoipa::path(
@@ -313,87 +293,87 @@ pub async fn post_leave_current_team(
}
#[utoipa::path(
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin",
params(
("page" = Option<i64>, Query, description = "Page number"),
("per_page" = Option<i64>, Query, description = "Items per page"),
("search" = Option<String>, Query, description = "Search keyword"),
("sort_by" = Option<String>, Query, description = "Sort by field"),
("order" = Option<String>, Query, description = "Order ASC or DESC"),
("filter" = Option<String>, Query, description = "Filter value"),
("filter_by" = Option<String>, Query, description = "Field to filter by"),
),
responses(
(status = 200, description = "Get admin team list", body = ResponseListSuccessDto<Vec<AdminTeamsListItemDto>>)
),
tag = "Teams - Admin"
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin",
params(
("page" = Option<i64>, Query, description = "Page number"),
("per_page" = Option<i64>, Query, description = "Items per page"),
("search" = Option<String>, Query, description = "Search keyword"),
("sort_by" = Option<String>, Query, description = "Sort by field"),
("order" = Option<String>, Query, description = "Order ASC or DESC"),
("filter" = Option<String>, Query, description = "Filter value"),
("filter_by" = Option<String>, Query, description = "Field to filter by"),
),
responses(
(status = 200, description = "Get admin team list", body = ResponseListSuccessDto<Vec<AdminTeamsListItemDto>>)
),
tag = "Teams - Admin"
)]
pub async fn get_admin_team_list(
headers: HeaderMap,
Extension(state): Extension<AppState>,
axum::extract::Query(meta): axum::extract::Query<MetaRequestDto>,
headers: HeaderMap,
Extension(state): Extension<AppState>,
axum::extract::Query(meta): axum::extract::Query<MetaRequestDto>,
) -> Response {
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadListTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_list(&state, meta);
response
}).await
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadListTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_list(&state, meta);
response
}).await
}
#[utoipa::path(
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin/{id}",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get admin team by ID", body = ResponseSuccessDto<AdminTeamsDetailItemDto>)
),
tag = "Teams - Admin"
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin/{id}",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get admin team by ID", body = ResponseSuccessDto<AdminTeamsDetailItemDto>)
),
tag = "Teams - Admin"
)]
pub async fn get_admin_team_by_id(
headers: HeaderMap,
Extension(state): Extension<AppState>,
Path(id): Path<String>,
headers: HeaderMap,
Extension(state): Extension<AppState>,
Path(id): Path<String>,
) -> Response {
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadDetailTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_by_id(&state, id);
response
}).await
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadDetailTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_by_id(&state, id);
response
}).await
}
#[utoipa::path(
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin/{id}/members",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get admin team members", body = ResponseSuccessDto<Vec<TeamMemberDto>>)
),
tag = "Teams - Admin"
get,
security(
("Bearer" = [])
),
path = "/v1/teams/admin/{id}/members",
params(
("id" = String, Path, description = "Team ID")
),
responses(
(status = 200, description = "Get admin team members", body = ResponseSuccessDto<Vec<TeamMemberDto>>)
),
tag = "Teams - Admin"
)]
pub async fn get_admin_team_members(
headers: HeaderMap,
Extension(state): Extension<AppState>,
Path(id): Path<String>,
headers: HeaderMap,
Extension(state): Extension<AppState>,
Path(id): Path<String>,
) -> Response {
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadDetailTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_members(&state, id);
response
}).await
let state = state;
with_perms(headers, axum::Extension(state), vec![PermissionsEnum::ReadDetailTeams], move |_claims, state| {
let response = TeamsService::get_admin_team_members(&state, id);
response
}).await
}
pub fn teams_router() -> Router {
+33 -24
View File
@@ -241,32 +241,41 @@ impl<'a> TeamsRepository<'a> {
}
pub async fn query_is_team_member(&self, team_id: &Thing, user_id: &Thing) -> Result<bool> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
let conditions = format!(
"{} AND is_active = true",
build_multi_thing_condition(&[("team_id", team_id), ("user_id", user_id)])
);
let member_count = execute_safe_count_query(
db,
ResourceEnum::TeamMembers.to_string(),
&conditions,
).await.unwrap_or(0);
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_is_team_member' found {} matching members", member_count);
println!("Query 'query_is_team_member' took: {elapsed:.2?}");
let now = Instant::now();
let db = &self.state.surrealdb_ws;
// Use direct SQL query for more control over the team member check
let sql = format!(
"SELECT COUNT() AS count FROM {}
WHERE team_id = $team_id
AND user_id = $user_id
AND is_active = true",
ResourceEnum::TeamMembers
);
let mut result = db.query(sql)
.bind(("team_id", team_id.id.to_raw()))
.bind(("user_id", user_id.id.to_raw()))
.await?;
// Use a simpler approach to get the count
let count = match result.take(0) {
Ok(Some(surrealdb::sql::Value::Number(num))) => num.to_int(),
_ => 0,
};
let elapsed = now.elapsed();
if std::env::var("RUST_ENV").unwrap_or_else(|_| "development".to_string())
== "development"
{
println!("Query 'query_is_team_member' found {} matching members", count);
println!("Query 'query_is_team_member' took: {elapsed:.2?}");
}
Ok(count > 0)
}
Ok(member_count > 0)
}
pub async fn query_create_invitation(&self, data: TeamInvitationsSchema) -> Result<String> {
let now = Instant::now();
let db = &self.state.surrealdb_ws;
+3 -2
View File
@@ -476,8 +476,9 @@ mod tests {
);
repo.query_add_team_member(member_member).await.unwrap();
// Verify non-leader is a member
let is_member = repo.query_is_team_member(&team_thing, &make_thing_from_enum(ResourceEnum::Users, &non_leader.id.id.to_raw())).await.unwrap();
// Verify non-leader is a member by directly checking team members
let members = repo.query_team_members(&team_thing).await.unwrap();
let is_member = members.iter().any(|m| m.user_id.id.to_raw() == non_leader.id.id.to_raw());
assert!(is_member, "Non-leader should be a team member");
// Try to remove member as non-leader (this would fail in real service layer with auth)