This commit is contained in:
2026-07-13 01:38:36 +02:00
parent ede38cc042
commit 09cf5f37fb
16 changed files with 747 additions and 644 deletions
+309 -322
View File
@@ -7,7 +7,7 @@ pub struct Migration;
impl MigrationTrait for Migration { impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> { async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// --------------------------------------------------------------------- // ---------------------------------------------------------------------
// Tables principales // Utilisateurs
// --------------------------------------------------------------------- // ---------------------------------------------------------------------
manager manager
@@ -51,6 +51,10 @@ impl MigrationTrait for Migration {
) )
.await?; .await?;
// ---------------------------------------------------------------------
// Serveurs
// ---------------------------------------------------------------------
manager manager
.create_table( .create_table(
Table::create() Table::create()
@@ -83,10 +87,149 @@ impl MigrationTrait for Migration {
.default(false), .default(false),
) )
.col(ColumnDef::new(Alias::new("owner_id")).uuid().null()) .col(ColumnDef::new(Alias::new("owner_id")).uuid().null())
.foreign_key(
ForeignKey::create()
.name("fk_server_owner")
.from(Alias::new("server"), Alias::new("owner_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::SetNull),
)
.to_owned(), .to_owned(),
) )
.await?; .await?;
manager
.create_table(
Table::create()
.table(Alias::new("server_user"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("username")).string().null())
.col(
ColumnDef::new(Alias::new("joined_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.col(
ColumnDef::new(Alias::new("updated_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_server_user_server")
.from(Alias::new("server_user"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_server_user_user")
.from(Alias::new("server_user"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.index(
Index::create()
.name("uq_server_user")
.col(Alias::new("server_id"))
.col(Alias::new("user_id"))
.unique(),
)
.to_owned(),
)
.await?;
// ---------------------------------------------------------------------
// Rôles
// ---------------------------------------------------------------------
manager
.create_table(
Table::create()
.table(Alias::new("role"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("name")).string().not_null())
.col(
ColumnDef::new(Alias::new("is_default"))
.boolean()
.not_null()
.default(false),
)
.col(
ColumnDef::new(Alias::new("created_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_role_server")
.from(Alias::new("role"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.index(
Index::create()
.name("uq_role_server_name")
.col(Alias::new("server_id"))
.col(Alias::new("name"))
.unique(),
)
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Alias::new("role_user"))
.if_not_exists()
.col(ColumnDef::new(Alias::new("role_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.primary_key(
Index::create()
.col(Alias::new("role_id"))
.col(Alias::new("user_id")),
)
.foreign_key(
ForeignKey::create()
.name("fk_role_user_role")
.from(Alias::new("role_user"), Alias::new("role_id"))
.to(Alias::new("role"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_role_user_user")
.from(Alias::new("role_user"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.to_owned(),
)
.await?;
// ---------------------------------------------------------------------
// Catégories et canaux
// ---------------------------------------------------------------------
manager manager
.create_table( .create_table(
Table::create() Table::create()
@@ -123,8 +266,7 @@ impl MigrationTrait for Migration {
.name("fk_category_server") .name("fk_category_server")
.from(Alias::new("category"), Alias::new("server_id")) .from(Alias::new("category"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id")) .to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade) .on_delete(ForeignKeyAction::Cascade),
.on_update(ForeignKeyAction::Cascade),
) )
.to_owned(), .to_owned(),
) )
@@ -186,7 +328,61 @@ impl MigrationTrait for Migration {
.await?; .await?;
// --------------------------------------------------------------------- // ---------------------------------------------------------------------
// Contenu et relations utilisateurs // Membres des canaux
// ---------------------------------------------------------------------
manager
.create_table(
Table::create()
.table(Alias::new("channel_user"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("channel_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("role"))
.string()
.not_null()
.default("member"),
)
.col(
ColumnDef::new(Alias::new("joined_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_channel_user_channel")
.from(Alias::new("channel_user"), Alias::new("channel_id"))
.to(Alias::new("channel"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_channel_user_user")
.from(Alias::new("channel_user"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.index(
Index::create()
.name("uq_channel_user")
.col(Alias::new("channel_id"))
.col(Alias::new("user_id"))
.unique(),
)
.to_owned(),
)
.await?;
// ---------------------------------------------------------------------
// Messages et pièces jointes
// --------------------------------------------------------------------- // ---------------------------------------------------------------------
manager manager
@@ -276,10 +472,14 @@ impl MigrationTrait for Migration {
) )
.await?; .await?;
// ---------------------------------------------------------------------
// Permissions
// ---------------------------------------------------------------------
manager manager
.create_table( .create_table(
Table::create() Table::create()
.table(Alias::new("server_user")) .table(Alias::new("server_role_permission"))
.if_not_exists() .if_not_exists()
.col( .col(
ColumnDef::new(Alias::new("id")) ColumnDef::new(Alias::new("id"))
@@ -288,34 +488,37 @@ impl MigrationTrait for Migration {
.primary_key(), .primary_key(),
) )
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null()) .col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null()) .col(ColumnDef::new(Alias::new("role_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("username")).string().null())
.col( .col(
ColumnDef::new(Alias::new("joined_at")) ColumnDef::new(Alias::new("permission"))
.timestamp_with_time_zone() .big_integer()
.not_null() .not_null()
.default(Expr::current_timestamp()), .default(0),
)
.col(
ColumnDef::new(Alias::new("updated_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
) )
.foreign_key( .foreign_key(
ForeignKey::create() ForeignKey::create()
.name("fk_server_user_server") .name("fk_server_role_permission_server")
.from(Alias::new("server_user"), Alias::new("server_id")) .from(
Alias::new("server_role_permission"),
Alias::new("server_id"),
)
.to(Alias::new("server"), Alias::new("id")) .to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade), .on_delete(ForeignKeyAction::Cascade),
) )
.foreign_key( .foreign_key(
ForeignKey::create() ForeignKey::create()
.name("fk_server_user_user") .name("fk_server_role_permission_role")
.from(Alias::new("server_user"), Alias::new("user_id")) .from(Alias::new("server_role_permission"), Alias::new("role_id"))
.to(Alias::new("user"), Alias::new("id")) .to(Alias::new("role"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade), .on_delete(ForeignKeyAction::Cascade),
) )
.index(
Index::create()
.name("uq_server_role_permission")
.col(Alias::new("server_id"))
.col(Alias::new("role_id"))
.unique(),
)
.to_owned(), .to_owned(),
) )
.await?; .await?;
@@ -323,7 +526,54 @@ impl MigrationTrait for Migration {
manager manager
.create_table( .create_table(
Table::create() Table::create()
.table(Alias::new("channel_user")) .table(Alias::new("channel_role_permission"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("channel_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("role_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("permission"))
.big_integer()
.not_null()
.default(0),
)
.foreign_key(
ForeignKey::create()
.name("fk_channel_role_permission_channel")
.from(
Alias::new("channel_role_permission"),
Alias::new("channel_id"),
)
.to(Alias::new("channel"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_channel_role_permission_role")
.from(Alias::new("channel_role_permission"), Alias::new("role_id"))
.to(Alias::new("role"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.index(
Index::create()
.name("uq_channel_role_permission")
.col(Alias::new("channel_id"))
.col(Alias::new("role_id"))
.unique(),
)
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Alias::new("channel_user_permission"))
.if_not_exists() .if_not_exists()
.col( .col(
ColumnDef::new(Alias::new("id")) ColumnDef::new(Alias::new("id"))
@@ -334,226 +584,34 @@ impl MigrationTrait for Migration {
.col(ColumnDef::new(Alias::new("channel_id")).uuid().not_null()) .col(ColumnDef::new(Alias::new("channel_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null()) .col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.col( .col(
ColumnDef::new(Alias::new("role")) ColumnDef::new(Alias::new("permission"))
.string() .big_integer()
.not_null() .not_null()
.default("member"), .default(0),
)
.col(
ColumnDef::new(Alias::new("joined_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
) )
.foreign_key( .foreign_key(
ForeignKey::create() ForeignKey::create()
.name("fk_channel_user_channel") .name("fk_channel_user_permission_channel")
.from(Alias::new("channel_user"), Alias::new("channel_id")) .from(
Alias::new("channel_user_permission"),
Alias::new("channel_id"),
)
.to(Alias::new("channel"), Alias::new("id")) .to(Alias::new("channel"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade), .on_delete(ForeignKeyAction::Cascade),
) )
.foreign_key( .foreign_key(
ForeignKey::create() ForeignKey::create()
.name("fk_channel_user_user") .name("fk_channel_user_permission_user")
.from(Alias::new("channel_user"), Alias::new("user_id")) .from(Alias::new("channel_user_permission"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id")) .to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade), .on_delete(ForeignKeyAction::Cascade),
) )
.to_owned(), .index(
)
.await?;
// ---------------------------------------------------------------------
// Groupes et attributions
// ---------------------------------------------------------------------
manager
.create_table(
Table::create()
.table(Alias::new("group"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("name")).string().not_null())
.col(
ColumnDef::new(Alias::new("is_default"))
.boolean()
.not_null()
.default(false),
)
.col(
ColumnDef::new(Alias::new("created_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_group_server")
.from(Alias::new("group"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Alias::new("group_member"))
.if_not_exists()
.col(ColumnDef::new(Alias::new("group_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.primary_key(
Index::create() Index::create()
.col(Alias::new("group_id")) .name("uq_channel_user_permission")
.col(Alias::new("user_id")), .col(Alias::new("channel_id"))
) .col(Alias::new("user_id"))
.foreign_key( .unique(),
ForeignKey::create()
.name("fk_group_member_group")
.from(Alias::new("group_member"), Alias::new("group_id"))
.to(Alias::new("group"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_group_member_user")
.from(Alias::new("group_member"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.to_owned(),
)
.await?;
// ---------------------------------------------------------------------
// Permissions sources
// ---------------------------------------------------------------------
manager
.create_table(
Table::create()
.table(Alias::new("group_permission"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("group_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("scope_type"))
.integer()
.not_null(),
)
.col(ColumnDef::new(Alias::new("scope_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("server_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("channel_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("voice_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("created_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_group_permission_group")
.from(Alias::new("group_permission"), Alias::new("group_id"))
.to(Alias::new("group"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_group_permission_server")
.from(Alias::new("group_permission"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.to_owned(),
)
.await?;
manager
.create_table(
Table::create()
.table(Alias::new("user_permission"))
.if_not_exists()
.col(
ColumnDef::new(Alias::new("id"))
.uuid()
.not_null()
.primary_key(),
)
.col(ColumnDef::new(Alias::new("user_id")).uuid().not_null())
.col(ColumnDef::new(Alias::new("server_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("scope_type"))
.integer()
.not_null(),
)
.col(ColumnDef::new(Alias::new("scope_id")).uuid().not_null())
.col(
ColumnDef::new(Alias::new("server_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("channel_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("voice_permissions"))
.big_integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(Alias::new("created_at"))
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.foreign_key(
ForeignKey::create()
.name("fk_user_permission_user")
.from(Alias::new("user_permission"), Alias::new("user_id"))
.to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
)
.foreign_key(
ForeignKey::create()
.name("fk_user_permission_server")
.from(Alias::new("user_permission"), Alias::new("server_id"))
.to(Alias::new("server"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade),
) )
.to_owned(), .to_owned(),
) )
@@ -597,6 +655,7 @@ impl MigrationTrait for Migration {
.primary_key( .primary_key(
Index::create() Index::create()
.col(Alias::new("user_id")) .col(Alias::new("user_id"))
.col(Alias::new("server_id"))
.col(Alias::new("scope_type")) .col(Alias::new("scope_type"))
.col(Alias::new("resource_id")), .col(Alias::new("resource_id")),
) )
@@ -607,92 +666,13 @@ impl MigrationTrait for Migration {
.to(Alias::new("user"), Alias::new("id")) .to(Alias::new("user"), Alias::new("id"))
.on_delete(ForeignKeyAction::Cascade), .on_delete(ForeignKeyAction::Cascade),
) )
.to_owned(), .foreign_key(
) ForeignKey::create()
.await?; .name("fk_computed_permission_server")
.from(Alias::new("computed_permission"), Alias::new("server_id"))
// --------------------------------------------------------------------- .to(Alias::new("server"), Alias::new("id"))
// Index .on_delete(ForeignKeyAction::Cascade),
// --------------------------------------------------------------------- )
let indexes = [
("idx_category_server_id", "category", "server_id"),
("idx_channel_server_id", "channel", "server_id"),
("idx_channel_category_id", "channel", "category_id"),
("idx_message_channel_id", "message", "channel_id"),
("idx_message_user_id", "message", "user_id"),
("idx_attachment_message_id", "attachment", "message_id"),
("idx_server_user_server_id", "server_user", "server_id"),
("idx_server_user_user_id", "server_user", "user_id"),
("idx_channel_user_channel_id", "channel_user", "channel_id"),
("idx_channel_user_user_id", "channel_user", "user_id"),
("idx_group_server_id", "group", "server_id"),
("idx_group_member_user_id", "group_member", "user_id"),
];
for (name, table, column) in indexes {
manager
.create_index(
Index::create()
.name(name)
.table(Alias::new(table))
.col(Alias::new(column))
.to_owned(),
)
.await?;
}
manager
.create_index(
Index::create()
.name("idx_group_permission_group_scope")
.table(Alias::new("group_permission"))
.col(Alias::new("group_id"))
.col(Alias::new("scope_type"))
.col(Alias::new("scope_id"))
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx_group_permission_server")
.table(Alias::new("group_permission"))
.col(Alias::new("server_id"))
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx_user_permission_user_scope")
.table(Alias::new("user_permission"))
.col(Alias::new("user_id"))
.col(Alias::new("scope_type"))
.col(Alias::new("scope_id"))
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx_user_permission_server")
.table(Alias::new("user_permission"))
.col(Alias::new("server_id"))
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx_computed_permission_resource")
.table(Alias::new("computed_permission"))
.col(Alias::new("scope_type"))
.col(Alias::new("resource_id"))
.to_owned(), .to_owned(),
) )
.await?; .await?;
@@ -701,25 +681,32 @@ impl MigrationTrait for Migration {
} }
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> { async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Les tables dépendantes doivent être supprimées avant leurs parents.
let tables = [ let tables = [
"computed_permission", "computed_permission",
"user_permission", "channel_user_permission",
"group_permission", "channel_role_permission",
"group_member", "server_role_permission",
"channel_user",
"server_user",
"attachment", "attachment",
"message", "message",
"channel_user",
"channel", "channel",
"category", "category",
"group", "role_user",
"role",
"server_user",
"server", "server",
"user", "user",
]; ];
for table in tables { for table in tables {
manager manager
.drop_table(Table::drop().table(Alias::new(table)).to_owned()) .drop_table(
Table::drop()
.table(Alias::new(table))
.if_exists()
.to_owned(),
)
.await?; .await?;
} }
+65
View File
@@ -0,0 +1,65 @@
//! SeaORM Entity pour les permissions d'un rôle dans un canal.
use crate::permissions::ChannelPermission;
use sea_orm::entity::prelude::*;
use sea_orm::prelude::async_trait::async_trait;
use sea_orm::{NotSet, Set};
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "channel_role_permission")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
pub channel_id: Uuid,
pub role_id: Uuid,
/// Bitmask des permissions accordées au rôle dans ce canal.
pub permission: i64,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::channel::Entity",
from = "Column::ChannelId",
to = "super::channel::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Channel,
#[sea_orm(
belongs_to = "super::role::Entity",
from = "Column::RoleId",
to = "super::role::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Role,
}
impl Related<super::channel::Entity> for Entity {
fn to() -> RelationDef {
Relation::Channel.def()
}
}
impl Related<super::role::Entity> for Entity {
fn to() -> RelationDef {
Relation::Role.def()
}
}
#[async_trait]
impl ActiveModelBehavior for ActiveModel {
fn new() -> Self {
Self {
id: Set(Uuid::new_v4()),
channel_id: NotSet,
role_id: NotSet,
permission: Set(ChannelPermission::empty().bits() as i64),
}
}
}
+65
View File
@@ -0,0 +1,65 @@
//! SeaORM Entity pour les permissions directes d'un utilisateur dans un canal.
use crate::permissions::ChannelPermission;
use sea_orm::entity::prelude::*;
use sea_orm::prelude::async_trait::async_trait;
use sea_orm::{NotSet, Set};
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "channel_user_permission")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
pub channel_id: Uuid,
pub user_id: Uuid,
/// Bitmask des permissions accordées directement à l'utilisateur dans ce canal.
pub permission: i64,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::channel::Entity",
from = "Column::ChannelId",
to = "super::channel::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Channel,
#[sea_orm(
belongs_to = "super::user::Entity",
from = "Column::UserId",
to = "super::user::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
User,
}
impl Related<super::channel::Entity> for Entity {
fn to() -> RelationDef {
Relation::Channel.def()
}
}
impl Related<super::user::Entity> for Entity {
fn to() -> RelationDef {
Relation::User.def()
}
}
#[async_trait]
impl ActiveModelBehavior for ActiveModel {
fn new() -> Self {
Self {
id: Set(Uuid::new_v4()),
channel_id: NotSet,
user_id: NotSet,
permission: Set(ChannelPermission::empty().bits() as i64),
}
}
}
+19 -3
View File
@@ -1,10 +1,12 @@
//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19 //! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
use async_trait::async_trait;
/// Permet de cache les permissions des utilisateurs pour éviter de recalculer les permissions à chaque fois qu'une requête est faite. /// Permet de cache les permissions des utilisateurs pour éviter de recalculer les permissions à chaque fois qu'une requête est faite.
use sea_orm::entity::prelude::*; use sea_orm::entity::prelude::*;
use sea_orm::{NotSet, Set};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use utoipa::ToSchema; use utoipa::ToSchema;
#[derive( #[derive(
Debug, Clone, Copy, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize, ToSchema, Debug, Clone, Copy, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize, ToSchema,
)] )]
@@ -21,6 +23,8 @@ pub enum PermissionScopeType {
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "computed_permission")] #[sea_orm(table_name = "computed_permission")]
pub struct Model { pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
/// L'utilisateur à qui appartiennent ces permissions /// L'utilisateur à qui appartiennent ces permissions
#[sea_orm(primary_key, auto_increment = false)] #[sea_orm(primary_key, auto_increment = false)]
pub user_id: Uuid, pub user_id: Uuid,
@@ -35,7 +39,6 @@ pub struct Model {
/// Cache des permissions serveur (stocké en i64 pour SQL, utilisé en u64) /// Cache des permissions serveur (stocké en i64 pour SQL, utilisé en u64)
pub server_permissions: i64, pub server_permissions: i64,
pub channel_permissions: i64, pub channel_permissions: i64,
pub voice_permissions: i64,
} }
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)] #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
@@ -56,4 +59,17 @@ impl Related<super::user::Entity> for Entity {
} }
} }
impl ActiveModelBehavior for ActiveModel {} #[async_trait]
impl ActiveModelBehavior for ActiveModel {
fn new() -> Self {
Self {
id: Set(Uuid::new_v4()),
user_id: NotSet,
server_id: NotSet,
scope_type: NotSet,
resource_id: NotSet,
server_permissions: Set(0),
channel_permissions: Set(0),
}
}
}
-67
View File
@@ -1,67 +0,0 @@
//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
use sea_orm::entity::prelude::*;
use serde::{Deserialize, Serialize};
use utoipa::ToSchema;
#[derive(
Debug, Clone, Copy, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize, ToSchema,
)]
#[sea_orm(rs_type = "i32", db_type = "Integer")]
pub enum PermissionScopeType {
#[sea_orm(num_value = 0)]
Server,
#[sea_orm(num_value = 1)]
Category,
#[sea_orm(num_value = 2)]
Channel,
}
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "group_permission")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
pub group_id: Uuid,
pub server_id: Uuid,
// L'ID du scope (soit un Uuid d'une Category, soit l'Uuid d'un Channel, soit l'Uuid du Server)
pub scope_type: PermissionScopeType,
pub scope_id: Uuid,
pub server_permissions: i64,
pub channel_permissions: i64,
pub voice_permissions: i64,
pub created_at: DateTimeUtc,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::group::Entity",
from = "Column::GroupId",
to = "super::group::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Group,
#[sea_orm(
belongs_to = "super::server::Entity",
from = "Column::ServerId",
to = "super::server::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Server,
}
impl Related<super::group::Entity> for Entity {
fn to() -> RelationDef {
Relation::Group.def()
}
}
impl Related<super::server::Entity> for Entity {
fn to() -> RelationDef {
Relation::Server.def()
}
}
impl ActiveModelBehavior for ActiveModel {}
+6 -4
View File
@@ -5,13 +5,15 @@ pub mod prelude;
pub mod attachment; pub mod attachment;
pub mod category; pub mod category;
pub mod channel; pub mod channel;
pub mod channel_role_permission;
pub mod channel_user; pub mod channel_user;
pub mod channel_user_permission;
pub mod computed_permission; pub mod computed_permission;
pub mod group;
pub mod group_member;
pub mod group_permission;
pub mod message; pub mod message;
pub mod role;
pub mod role_user;
pub mod server; pub mod server;
pub mod server_role_permission;
pub mod server_user; pub mod server_user;
pub mod server_user_permission;
pub mod user; pub mod user;
pub mod user_permission;
+4 -4
View File
@@ -5,11 +5,11 @@ pub use super::category::Entity as Category;
pub use super::channel::Entity as Channel; pub use super::channel::Entity as Channel;
pub use super::channel_user::Entity as ChannelUser; pub use super::channel_user::Entity as ChannelUser;
pub use super::computed_permission::Entity as ComputedPermission; pub use super::computed_permission::Entity as ComputedPermission;
pub use super::group::Entity as Group;
pub use super::group_member::Entity as GroupMember;
pub use super::group_permission::Entity as GroupPermission;
pub use super::message::Entity as Message; pub use super::message::Entity as Message;
pub use super::role::Entity as Group;
pub use super::role_user::Entity as GroupMember;
pub use super::server::Entity as Server; pub use super::server::Entity as Server;
pub use super::server_role_permission::Entity as ServerRolePermission;
pub use super::server_user::Entity as ServerUser; pub use super::server_user::Entity as ServerUser;
pub use super::server_user_permission::Entity as ServerUserPermission;
pub use super::user::Entity as User; pub use super::user::Entity as User;
pub use super::user_permission::Entity as UserPermission;
+5 -5
View File
@@ -5,7 +5,7 @@ use sea_orm::prelude::async_trait::async_trait;
use sea_orm::Set; use sea_orm::Set;
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "group")] #[sea_orm(table_name = "role")]
pub struct Model { pub struct Model {
#[sea_orm(primary_key, auto_increment = false)] #[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid, pub id: Uuid,
@@ -25,8 +25,8 @@ pub enum Relation {
on_delete = "Cascade" on_delete = "Cascade"
)] )]
Server, Server,
#[sea_orm(has_many = "super::group_member::Entity")] #[sea_orm(has_many = "super::role_user::Entity")]
GroupMember, RoleUser,
} }
impl Related<super::server::Entity> for Entity { impl Related<super::server::Entity> for Entity {
@@ -35,9 +35,9 @@ impl Related<super::server::Entity> for Entity {
} }
} }
impl Related<super::group_member::Entity> for Entity { impl Related<super::role_user::Entity> for Entity {
fn to() -> RelationDef { fn to() -> RelationDef {
Relation::GroupMember.def() Relation::RoleUser.def()
} }
} }
@@ -3,10 +3,10 @@
use sea_orm::entity::prelude::*; use sea_orm::entity::prelude::*;
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "group_member")] #[sea_orm(table_name = "role_user")]
pub struct Model { pub struct Model {
#[sea_orm(primary_key, auto_increment = false)] #[sea_orm(primary_key, auto_increment = false)]
pub group_id: Uuid, pub role_id: Uuid,
#[sea_orm(primary_key, auto_increment = false)] #[sea_orm(primary_key, auto_increment = false)]
pub user_id: Uuid, pub user_id: Uuid,
} }
@@ -14,9 +14,9 @@ pub struct Model {
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)] #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation { pub enum Relation {
#[sea_orm( #[sea_orm(
belongs_to = "super::group::Entity", belongs_to = "super::role::Entity",
from = "Column::GroupId", from = "Column::RoleId",
to = "super::group::Column::Id", to = "super::role::Column::Id",
on_update = "NoAction", on_update = "NoAction",
on_delete = "Cascade" on_delete = "Cascade"
)] )]
@@ -31,7 +31,7 @@ pub enum Relation {
User, User,
} }
impl Related<super::group::Entity> for Entity { impl Related<super::role::Entity> for Entity {
fn to() -> RelationDef { fn to() -> RelationDef {
Relation::Group.def() Relation::Group.def()
} }
+65
View File
@@ -0,0 +1,65 @@
use crate::permissions::ServerPermission;
use sea_orm::entity::prelude::*;
use sea_orm::prelude::async_trait::async_trait;
use sea_orm::{NotSet, Set};
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "server_group_permission")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
#[sea_orm(primary_key, auto_increment = false)]
pub server_id: Uuid,
#[sea_orm(primary_key, auto_increment = false)]
pub role_id: Uuid,
/// Bitmask des permissions accordées directement à l'utilisateur.
pub permission: i64,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {
#[sea_orm(
belongs_to = "super::server::Entity",
from = "Column::ServerId",
to = "super::server::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Server,
#[sea_orm(
belongs_to = "super::role::Entity",
from = "Column::RoleId",
to = "super::role::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
Role,
}
impl Related<super::server::Entity> for Entity {
fn to() -> RelationDef {
Relation::Server.def()
}
}
impl Related<super::role::Entity> for Entity {
fn to() -> RelationDef {
Relation::Role.def()
}
}
#[async_trait]
impl ActiveModelBehavior for ActiveModel {
fn new() -> Self {
Self {
id: Set(Uuid::new_v4()),
server_id: NotSet,
role_id: NotSet,
permission: Set(ServerPermission::empty().bits() as i64),
}
}
}
@@ -1,47 +1,26 @@
//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19 use crate::permissions::ServerPermission;
use sea_orm::entity::prelude::*; use sea_orm::entity::prelude::*;
use serde::{Deserialize, Serialize}; use sea_orm::prelude::async_trait::async_trait;
use utoipa::ToSchema; use sea_orm::{NotSet, Set};
#[derive(
Debug, Clone, Copy, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize, ToSchema,
)]
#[sea_orm(rs_type = "i32", db_type = "Integer")]
pub enum PermissionScopeType {
#[sea_orm(num_value = 0)]
Server,
#[sea_orm(num_value = 1)]
Category,
#[sea_orm(num_value = 2)]
Channel,
}
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
#[sea_orm(table_name = "user_permission")] #[sea_orm(table_name = "server_user_permission")]
pub struct Model { pub struct Model {
#[sea_orm(primary_key, auto_increment = false)] #[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid, pub id: Uuid,
pub user_id: Uuid,
#[sea_orm(primary_key, auto_increment = false)]
pub server_id: Uuid, pub server_id: Uuid,
// L'ID du scope (soit un Uuid d'une Category, soit l'Uuid d'un Channel, soit l'Uuid du Server)
pub scope_type: PermissionScopeType, #[sea_orm(primary_key, auto_increment = false)]
pub scope_id: Uuid, pub user_id: Uuid,
pub server_permissions: i64,
pub channel_permissions: i64, /// Bitmask des permissions accordées directement à l'utilisateur.
pub voice_permissions: i64, pub permission: i64,
pub created_at: DateTimeUtc,
} }
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)] #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation { pub enum Relation {
#[sea_orm(
belongs_to = "super::user::Entity",
from = "Column::UserId",
to = "super::user::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
User,
#[sea_orm( #[sea_orm(
belongs_to = "super::server::Entity", belongs_to = "super::server::Entity",
from = "Column::ServerId", from = "Column::ServerId",
@@ -50,12 +29,15 @@ pub enum Relation {
on_delete = "Cascade" on_delete = "Cascade"
)] )]
Server, Server,
}
impl Related<super::user::Entity> for Entity { #[sea_orm(
fn to() -> RelationDef { belongs_to = "super::user::Entity",
Relation::User.def() from = "Column::UserId",
} to = "super::user::Column::Id",
on_update = "NoAction",
on_delete = "Cascade"
)]
User,
} }
impl Related<super::server::Entity> for Entity { impl Related<super::server::Entity> for Entity {
@@ -64,4 +46,20 @@ impl Related<super::server::Entity> for Entity {
} }
} }
impl ActiveModelBehavior for ActiveModel {} impl Related<super::user::Entity> for Entity {
fn to() -> RelationDef {
Relation::User.def()
}
}
#[async_trait]
impl ActiveModelBehavior for ActiveModel {
fn new() -> Self {
Self {
id: Set(Uuid::new_v4()),
server_id: NotSet,
user_id: NotSet,
permission: Set(ServerPermission::empty().bits() as i64),
}
}
}
+13 -47
View File
@@ -37,6 +37,7 @@ bitflags! {
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default, Serialize, Deserialize)] #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default, Serialize, Deserialize)]
#[serde(transparent)] #[serde(transparent)]
pub struct ChannelPermission: u64 { pub struct ChannelPermission: u64 {
// Permission communes / texte
/// Voir le canal et son contenu. /// Voir le canal et son contenu.
const READ_CHANNEL = 1 << 0; const READ_CHANNEL = 1 << 0;
@@ -66,37 +67,16 @@ bitflags! {
/// Épingler ou désépingler des messages. /// Épingler ou désépingler des messages.
const MANAGE_MESSAGES = 1 << 10; const MANAGE_MESSAGES = 1 << 10;
}
}
bitflags! { // Permissions vocales (30-45 réservées)
/// Permissions applicables aux canaux vocaux. const JOIN_VOICE = 1 << 30;
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default, Serialize, Deserialize)] const SPEAK = 1 << 31;
#[serde(transparent)] const STREAM = 1 << 32;
pub struct VoicePermission: u64 { const MUTE_SELF = 1 << 33;
/// Rejoindre un canal vocal. const MUTE_OTHERS = 1 << 34;
const JOIN_CHANNEL = 1 << 0; const MOVE_OTHERS = 1 << 35;
const DISCONNECT_OTHERS = 1 << 36;
/// Parler dans un canal vocal. const MANAGE_VOICE_CHANNEL = 1 << 37;
const SPEAK = 1 << 1;
/// Utiliser sa caméra ou partager son écran.
const STREAM = 1 << 2;
/// Couper son propre microphone.
const MUTE_SELF = 1 << 3;
/// Couper le microphone d'un autre membre.
const MUTE_OTHERS = 1 << 4;
/// Déplacer un membre vers un autre canal vocal.
const MOVE_OTHERS = 1 << 6;
/// Expulser un membre d'un canal vocal.
const DISCONNECT_OTHERS = 1 << 7;
/// Modifier les paramètres du canal vocal.
const MANAGE_VOICE_CHANNEL = 1 << 8;
} }
} }
@@ -105,20 +85,11 @@ bitflags! {
pub struct PermissionSet { pub struct PermissionSet {
pub server: ServerPermission, pub server: ServerPermission,
pub channel: ChannelPermission, pub channel: ChannelPermission,
pub voice: VoicePermission,
} }
impl PermissionSet { impl PermissionSet {
pub const fn new( pub const fn new(server: ServerPermission, channel: ChannelPermission) -> Self {
server: ServerPermission, Self { server, channel }
channel: ChannelPermission,
voice: VoicePermission,
) -> Self {
Self {
server,
channel,
voice,
}
} }
/// Permissions par défaut accordées à un membre standard. /// Permissions par défaut accordées à un membre standard.
@@ -131,15 +102,10 @@ impl PermissionSet {
| ChannelPermission::DELETE_OWN_MESSAGE.bits() | ChannelPermission::DELETE_OWN_MESSAGE.bits()
| ChannelPermission::ADD_REACTIONS.bits(), | ChannelPermission::ADD_REACTIONS.bits(),
), ),
VoicePermission::from_bits_retain(
VoicePermission::JOIN_CHANNEL.bits() | VoicePermission::SPEAK.bits(),
),
); );
/// Vérifie si l'ensemble contient la permission demandée. /// Vérifie si l'ensemble contient la permission demandée.
pub const fn contains(&self, required: Self) -> bool { pub const fn contains(&self, required: Self) -> bool {
self.server.contains(required.server) self.server.contains(required.server) && self.channel.contains(required.channel)
&& self.channel.contains(required.channel)
&& self.voice.contains(required.voice)
} }
} }
+134 -128
View File
@@ -1,18 +1,17 @@
use crate::models::{ use crate::models::{
category, channel, computed_permission, group, group_member, group_permission, server, user, channel, channel_role_permission, channel_user_permission, computed_permission, role_user,
user_permission, server_role_permission, server_user,
}; };
use crate::permissions::{ChannelPermission, ServerPermission};
use crate::repositories::{AnyResult, RepositoryContext}; use crate::repositories::{AnyResult, RepositoryContext};
// use sea_orm::prelude::*;
use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, QueryFilter, Set, TransactionTrait};
use std::collections::HashMap;
use crate::models::channel::ChannelType; use sea_orm::{ColumnTrait, EntityTrait, QueryFilter, QuerySelect, Set, TransactionTrait};
use crate::models::computed_permission::PermissionScopeType; use std::collections::HashMap;
use crate::permissions::{ChannelPermission, ServerPermission, VoicePermission};
use std::sync::Arc; use std::sync::Arc;
use uuid::Uuid; use uuid::Uuid;
use crate::models::computed_permission::PermissionScopeType;
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub struct ComputedPermissionRepository { pub struct ComputedPermissionRepository {
pub context: Arc<RepositoryContext>, pub context: Arc<RepositoryContext>,
@@ -20,171 +19,178 @@ pub struct ComputedPermissionRepository {
impl ComputedPermissionRepository { impl ComputedPermissionRepository {
pub async fn get_all(&self) -> AnyResult<Vec<computed_permission::Model>> { pub async fn get_all(&self) -> AnyResult<Vec<computed_permission::Model>> {
let result = computed_permission::Entity::find() Ok(computed_permission::Entity::find()
.all(&self.context.db) .all(&self.context.db)
.await?; .await?)
Ok(result)
} }
/// Recalcule le cache de permissions pour tous les utilisateurs du serveur.
pub async fn full_sync_server(&self, server_id: Uuid) -> AnyResult<()> { pub async fn full_sync_server(&self, server_id: Uuid) -> AnyResult<()> {
let user_ids = server_user::Entity::find()
.filter(server_user::Column::ServerId.eq(server_id))
.select_only()
.column(server_user::Column::UserId)
.into_tuple::<Uuid>()
.all(&self.context.db)
.await?;
for user_id in user_ids {
self.full_sync_user(user_id, server_id).await?;
}
Ok(()) Ok(())
} }
/// Recalcule le cache de permissions d'un utilisateur sur un serveur.
///
/// Les permissions effectives sont composées de :
///
/// - permissions serveur accordées aux rôles de l'utilisateur ;
/// - permissions de canal accordées aux rôles de l'utilisateur ;
/// - permissions directes de l'utilisateur dans les canaux.
pub async fn full_sync_user(&self, user_id: Uuid, server_id: Uuid) -> AnyResult<()> { pub async fn full_sync_user(&self, user_id: Uuid, server_id: Uuid) -> AnyResult<()> {
// 1. Récupérer la structure (tous les salons + catégories du serveur) // ---------------------------------------------------------------------
// 2. Récupérer les sources de vérité (Groupes de l'user + Perms individuelles) // Rôles de l'utilisateur
// 3. Calculer pour l'UUID du serveur (Pseudo-salon pour le Kick/Ban) // ---------------------------------------------------------------------
// 4. Pour chaque salon, calculer la fusion des bits
// 5. Tout enregistrer en une seule transaction dans `computed_permission`
// Récupération du serveur let role_ids = role_user::Entity::find()
let server = server::Entity::find_by_id(server_id) .filter(role_user::Column::UserId.eq(user_id))
.one(&self.context.db) .select_only()
.await?; .column(role_user::Column::RoleId)
.into_tuple::<Uuid>()
// Récupération de l'utilisateur
let user = user::Entity::find_by_id(user_id)
.one(&self.context.db)
.await?;
// Récupération des groupes de l'utilisateur
let groups = group::Entity::find()
.filter(group::Column::ServerId.eq(server_id))
.filter(group_member::Column::UserId.eq(user_id))
.all(&self.context.db) .all(&self.context.db)
.await?; .await?;
// Récupération de toutes les catégories du serveur // ---------------------------------------------------------------------
let categories = category::Entity::find() // Permissions serveur des rôles
.filter(category::Column::ServerId.eq(server_id)) // ---------------------------------------------------------------------
.all(&self.context.db)
.await?; let mut server_permissions = ServerPermission::empty();
if !role_ids.is_empty() {
let role_permissions = server_role_permission::Entity::find()
.filter(server_role_permission::Column::ServerId.eq(server_id))
.filter(server_role_permission::Column::RoleId.is_in(role_ids.clone()))
.all(&self.context.db)
.await?;
for permission in role_permissions {
server_permissions |=
ServerPermission::from_bits_retain(permission.permission as u64);
}
}
// ---------------------------------------------------------------------
// Canaux du serveur
// ---------------------------------------------------------------------
// Récupération de toutes les channels du serveur
let channels = channel::Entity::find() let channels = channel::Entity::find()
.filter(channel::Column::ServerId.eq(server_id)) .filter(channel::Column::ServerId.eq(server_id))
.all(&self.context.db) .all(&self.context.db)
.await?; .await?;
// retrieve permissions let channel_ids: Vec<Uuid> = channels.iter().map(|channel| channel.id).collect();
let user_permissions = user_permission::Entity::find()
.filter(user_permission::Column::UserId.eq(user_id))
.filter(user_permission::Column::ServerId.eq(server_id))
.all(&self.context.db)
.await?;
let group_permissions = group_permission::Entity::find() // ---------------------------------------------------------------------
.filter(group_permission::Column::ServerId.eq(server_id)) // Permissions de rôles pour tous les canaux
.filter(group_member::Column::UserId.eq(user_id)) // ---------------------------------------------------------------------
.all(&self.context.db)
.await?;
// computed existing permissions let role_channel_permissions = if role_ids.is_empty() || channel_ids.is_empty() {
struct VirtualComputedPermission { Vec::new()
server_permissions: i64, } else {
channel_permissions: i64, channel_role_permission::Entity::find()
voice_permissions: i64, .filter(channel_role_permission::Column::ChannelId.is_in(channel_ids.clone()))
} .filter(channel_role_permission::Column::RoleId.is_in(role_ids))
let mut all_permissions: HashMap<Uuid, Vec<VirtualComputedPermission>> = HashMap::new(); .all(&self.context.db)
.await?
};
user_permissions.into_iter().for_each(|user_permission| { let mut permissions_by_channel: HashMap<Uuid, ChannelPermission> = HashMap::new();
let permission = VirtualComputedPermission {
server_permissions: user_permission.server_permissions,
channel_permissions: user_permission.channel_permissions,
voice_permissions: user_permission.voice_permissions,
};
all_permissions for permission in role_channel_permissions {
.entry(user_permission.scope_id) // vérifier que la clé existe dans la HashMap permissions_by_channel
.or_default() // appel Vec::new() automatiquement .entry(permission.channel_id)
.push(permission); // insert la permission dans le Vec
});
group_permissions.into_iter().for_each(|group_permission| {
let permission = VirtualComputedPermission {
server_permissions: group_permission.server_permissions,
channel_permissions: group_permission.channel_permissions,
voice_permissions: group_permission.voice_permissions,
};
all_permissions
.entry(group_permission.scope_id)
.or_default() .or_default()
.push(permission); .insert(ChannelPermission::from_bits_retain(
}); permission.permission as u64,
));
// Compute permissions
let mut computed_permissions: Vec<computed_permission::ActiveModel> = Vec::new();
// server
let mut server_permissions = ServerPermission::empty();
match all_permissions.get(&server_id) {
Some(perm) => {
for permission in perm {
server_permissions |=
ServerPermission::from_bits_retain(permission.server_permissions as u64);
}
computed_permissions.push(computed_permission::ActiveModel {
user_id: Set(user_id),
server_id: Set(server_id),
scope_type: Set(PermissionScopeType::Server),
resource_id: Set(server_id),
server_permissions: Set(server_permissions.bits() as i64),
channel_permissions: Set(ChannelPermission::empty().bits() as i64),
voice_permissions: Set(ChannelPermission::empty().bits() as i64),
});
}
None => {}
} }
// Category // ---------------------------------------------------------------------
// todo : à faire // Permissions directes de l'utilisateur pour tous les canaux
// ---------------------------------------------------------------------
let user_channel_permissions = if channel_ids.is_empty() {
Vec::new()
} else {
channel_user_permission::Entity::find()
.filter(channel_user_permission::Column::UserId.eq(user_id))
.filter(channel_user_permission::Column::ChannelId.is_in(channel_ids))
.all(&self.context.db)
.await?
};
for permission in user_channel_permissions {
permissions_by_channel
.entry(permission.channel_id)
.or_default()
.insert(ChannelPermission::from_bits_retain(
permission.permission as u64,
));
}
// ---------------------------------------------------------------------
// Construction du cache
// ---------------------------------------------------------------------
let mut computed_permissions = Vec::with_capacity(channels.len().saturating_add(1));
// Permissions au niveau serveur.
computed_permissions.push(computed_permission::ActiveModel {
user_id: Set(user_id),
server_id: Set(server_id),
scope_type: Set(PermissionScopeType::Server),
resource_id: Set(server_id),
server_permissions: Set(server_permissions.bits() as i64),
..Default::default()
});
// Permissions au niveau canal.
for channel in channels {
let channel_permissions = permissions_by_channel
.remove(&channel.id)
.unwrap_or_else(ChannelPermission::empty);
// channels
channels.into_iter().for_each(|channel| {
let mut channel_permissions = ChannelPermission::empty();
let mut voice_permissions = VoicePermission::empty();
match all_permissions.get(&channel.id) {
Some(perm) => {
for permission in perm {
channel_permissions |= ChannelPermission::from_bits_retain(
permission.channel_permissions as u64,
);
if channel.channel_type == ChannelType::Voice {
voice_permissions |= VoicePermission::from_bits_retain(
permission.voice_permissions as u64,
);
}
}
}
None => {}
}
computed_permissions.push(computed_permission::ActiveModel { computed_permissions.push(computed_permission::ActiveModel {
user_id: Set(user_id), user_id: Set(user_id),
server_id: Set(server_id), server_id: Set(server_id),
scope_type: Set(PermissionScopeType::Channel), scope_type: Set(PermissionScopeType::Channel),
resource_id: Set(channel.id), resource_id: Set(channel.id),
server_permissions: Set(ServerPermission::empty().bits() as i64),
channel_permissions: Set(channel_permissions.bits() as i64), channel_permissions: Set(channel_permissions.bits() as i64),
voice_permissions: Set(voice_permissions.bits() as i64), ..Default::default()
}); });
}); }
// ---------------------------------------------------------------------
// Remplacement atomique du cache
// ---------------------------------------------------------------------
// apply in the DB
self.context self.context
.db .db
.transaction::<_, (), anyhow::Error>(|txn| { .transaction::<_, (), anyhow::Error>(|transaction| {
Box::pin(async move { Box::pin(async move {
computed_permission::Entity::delete_many() computed_permission::Entity::delete_many()
.filter(computed_permission::Column::UserId.eq(user_id)) .filter(computed_permission::Column::UserId.eq(user_id))
.filter(computed_permission::Column::ServerId.eq(server_id)) .filter(computed_permission::Column::ServerId.eq(server_id))
.exec(txn) .exec(transaction)
.await?; .await?;
if !computed_permissions.is_empty() { if !computed_permissions.is_empty() {
computed_permission::Entity::insert_many(computed_permissions) computed_permission::Entity::insert_many(computed_permissions)
.exec(txn) .exec(transaction)
.await?; .await?;
} }
Ok(()) Ok(())
}) })
}) })
+11 -11
View File
@@ -1,4 +1,4 @@
use crate::models::group; use crate::models::role;
use crate::repositories::{AnyResult, RepositoryContext}; use crate::repositories::{AnyResult, RepositoryContext};
use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, QueryFilter}; use sea_orm::{ActiveModelTrait, ColumnTrait, EntityTrait, QueryFilter};
use std::sync::Arc; use std::sync::Arc;
@@ -10,35 +10,35 @@ pub struct GroupRepository {
} }
impl GroupRepository { impl GroupRepository {
pub async fn get_all_by_server(&self, server_id: Uuid) -> AnyResult<Vec<group::Model>> { pub async fn get_all_by_server(&self, server_id: Uuid) -> AnyResult<Vec<role::Model>> {
Ok(group::Entity::find() Ok(role::Entity::find()
.filter(group::Column::ServerId.eq(server_id)) .filter(role::Column::ServerId.eq(server_id))
.all(&self.context.db) .all(&self.context.db)
.await?) .await?)
} }
pub async fn get_all(&self) -> AnyResult<Vec<group::Model>> { pub async fn get_all(&self) -> AnyResult<Vec<role::Model>> {
Ok(group::Entity::find().all(&self.context.db).await?) Ok(role::Entity::find().all(&self.context.db).await?)
} }
pub async fn get_by_id(&self, id: Uuid) -> AnyResult<Option<group::Model>> { pub async fn get_by_id(&self, id: Uuid) -> AnyResult<Option<role::Model>> {
Ok(group::Entity::find_by_id(id).one(&self.context.db).await?) Ok(role::Entity::find_by_id(id).one(&self.context.db).await?)
} }
pub async fn create(&self, active: group::ActiveModel) -> AnyResult<group::Model> { pub async fn create(&self, active: role::ActiveModel) -> AnyResult<role::Model> {
let group = active.insert(&self.context.db).await?; let group = active.insert(&self.context.db).await?;
self.context.events.emit("group_created", group.clone()); self.context.events.emit("group_created", group.clone());
Ok(group) Ok(group)
} }
pub async fn update(&self, active: group::ActiveModel) -> AnyResult<group::Model> { pub async fn update(&self, active: role::ActiveModel) -> AnyResult<role::Model> {
let group = active.update(&self.context.db).await?; let group = active.update(&self.context.db).await?;
self.context.events.emit("group_updated", group.clone()); self.context.events.emit("group_updated", group.clone());
Ok(group) Ok(group)
} }
pub async fn delete(&self, id: Uuid) -> AnyResult<bool> { pub async fn delete(&self, id: Uuid) -> AnyResult<bool> {
let res = group::Entity::delete_by_id(id) let res = role::Entity::delete_by_id(id)
.exec(&self.context.db) .exec(&self.context.db)
.await?; .await?;
self.context.events.emit("group_deleted", id); self.context.events.emit("group_deleted", id);
+2 -2
View File
@@ -1,6 +1,6 @@
use super::types::{ServerExplorerItem, ServerTree}; use super::types::{ServerExplorerItem, ServerTree};
use super::{AnyResult, RepositoryContext}; use super::{AnyResult, RepositoryContext};
use crate::models::{category, channel, group, server, server_user}; use crate::models::{category, channel, role, server, server_user};
use sea_orm::prelude::*; use sea_orm::prelude::*;
use sea_orm::{ActiveModelTrait, Set}; use sea_orm::{ActiveModelTrait, Set};
@@ -38,7 +38,7 @@ impl ServerRepository {
let server = active.insert(&self.context.db).await?; let server = active.insert(&self.context.db).await?;
// Créer le groupe par défaut pour le serveur // Créer le groupe par défaut pour le serveur
let default_group = group::ActiveModel { let default_group = role::ActiveModel {
server_id: Set(server.id), server_id: Set(server.id),
name: Set("Member".to_string()), name: Set("Member".to_string()),
is_default: Set(true), is_default: Set(true),
+6 -6
View File
@@ -1,9 +1,9 @@
use crate::models::group; use crate::models::role;
use crate::routes::group::dto::{CreateGroupRequest, GroupResponse, UpdateGroupRequest}; use crate::routes::group::dto::{CreateGroupRequest, GroupResponse, UpdateGroupRequest};
use sea_orm::Set; use sea_orm::Set;
use uuid::Uuid; use uuid::Uuid;
pub fn group_model_to_group_response(model: group::Model) -> GroupResponse { pub fn group_model_to_group_response(model: role::Model) -> GroupResponse {
GroupResponse { GroupResponse {
id: model.id, id: model.id,
server_id: model.server_id, server_id: model.server_id,
@@ -13,8 +13,8 @@ pub fn group_model_to_group_response(model: group::Model) -> GroupResponse {
} }
} }
pub fn create_request_to_am(req: CreateGroupRequest) -> group::ActiveModel { pub fn create_request_to_am(req: CreateGroupRequest) -> role::ActiveModel {
group::ActiveModel { role::ActiveModel {
id: Set(Uuid::new_v4()), id: Set(Uuid::new_v4()),
server_id: Set(req.server_id), server_id: Set(req.server_id),
name: Set(req.name), name: Set(req.name),
@@ -27,8 +27,8 @@ pub fn update_request_to_am(
id: Uuid, id: Uuid,
server_id: Uuid, server_id: Uuid,
req: UpdateGroupRequest, req: UpdateGroupRequest,
) -> group::ActiveModel { ) -> role::ActiveModel {
group::ActiveModel { role::ActiveModel {
id: Set(id), id: Set(id),
server_id: Set(server_id), server_id: Set(server_id),
name: Set(req.name), name: Set(req.name),