feat: Blacklist table

Signed-off-by: Awiteb <a@4rs.nl>

crates/oxidetalis_entities/src/blacklist.rs

@@ -0,0 +1,56 @@
+// OxideTalis Messaging Protocol homeserver core implementation
+// Copyright (c) 2024 OxideTalis Developers <otmp@4rs.nl>
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+use chrono::Utc;
+use sea_orm::entity::prelude::*;
+
+use super::users::Entity as UserEntity;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "blacklist")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id:             i32,
+    pub user_id:        i32,
+    /// Public key of the target
+    pub target:         String,
+    pub blacklisted_at: chrono::DateTime<Utc>,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "UserEntity",
+        from = "Column::UserId",
+        to = "super::users::Column::Id"
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    UserId,
+}
+
+impl Related<UserEntity> for Entity {
+    fn to() -> RelationDef {
+        Relation::UserId.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/oxidetalis_entities/src/lib.rs

@@ -19,6 +19,7 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 // SOFTWARE.
 
+pub mod blacklist;
 pub mod incoming_chat_requests;
 pub mod outgoing_chat_requests;
 pub mod prelude;

crates/oxidetalis_entities/src/prelude.rs

@@ -33,6 +33,12 @@ pub use sea_orm::{
     SqlErr,
 };
 
+pub use super::blacklist::{
+    ActiveModel as BlacklistActiveModel,
+    Column as BlacklistColumn,
+    Entity as BlacklistEntity,
+    Model as BlacklistModel,
+};
 pub use super::incoming_chat_requests::{
     ActiveModel as InChatRequestsActiveModel,
     Column as InChatRequestsColumn,

crates/oxidetalis_entities/src/users.rs

@@ -21,6 +21,7 @@
 
 use sea_orm::entity::prelude::*;
 
+use super::blacklist::Entity as BlacklistEntity;
 use super::incoming_chat_requests::Entity as InChatRequestsEntity;
 use super::outgoing_chat_requests::Entity as OutChatRequestsEntity;
 
@@ -39,6 +40,8 @@ pub enum Relation {
     InChatRequests,
     #[sea_orm(has_many = "OutChatRequestsEntity")]
     OutChatRequests,
+    #[sea_orm(has_many = "BlacklistEntity")]
+    Blacklist,
 }
 
 impl Related<InChatRequestsEntity> for Entity {
@@ -53,4 +56,10 @@ impl Related<OutChatRequestsEntity> for Entity {
     }
 }
 
+impl Related<BlacklistEntity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Blacklist.def()
+    }
+}
+
 impl ActiveModelBehavior for ActiveModel {}

crates/oxidetalis_migrations/src/create_blacklist_table.rs

@@ -0,0 +1,65 @@
+// OxideTalis Messaging Protocol homeserver core implementation
+// Copyright (c) 2024 OxideTalis Developers <otmp@4rs.nl>
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+use sea_orm_migration::prelude::*;
+
+#[derive(DeriveMigrationName)]
+pub struct Migration;
+
+#[async_trait::async_trait]
+impl MigrationTrait for Migration {
+    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+        manager
+            .create_table(
+                Table::create()
+                    .table(Blacklist::Table)
+                    .if_not_exists()
+                    .col(
+                        ColumnDef::new(Blacklist::Id)
+                            .integer()
+                            .not_null()
+                            .auto_increment()
+                            .primary_key(),
+                    )
+                    .col(ColumnDef::new(Blacklist::UserId).integer().not_null())
+                    .col(ColumnDef::new(Blacklist::Target).string().not_null())
+                    .col(ColumnDef::new(Blacklist::Reason).string_len(400))
+                    .col(
+                        ColumnDef::new(Blacklist::BlacklistedAt)
+                            .timestamp_with_time_zone()
+                            .not_null(),
+                    )
+                    .to_owned(),
+            )
+            .await
+    }
+}
+
+#[derive(DeriveIden)]
+enum Blacklist {
+    Table,
+    Id,
+    UserId,
+    /// Public key of the target
+    Target,
+    Reason,
+    BlacklistedAt,
+}

crates/oxidetalis_migrations/src/lib.rs

@@ -21,6 +21,7 @@
 
 pub use sea_orm_migration::prelude::*;
 
+mod create_blacklist_table;
 mod create_incoming_chat_requests_table;
 mod create_outgoing_chat_requests_table;
 mod create_users_table;
@@ -34,6 +35,7 @@ impl MigratorTrait for Migrator {
             Box::new(create_users_table::Migration),
             Box::new(create_incoming_chat_requests_table::Migration),
             Box::new(create_outgoing_chat_requests_table::Migration),
+            Box::new(create_blacklist_table::Migration),
         ]
     }
 }