entity creation

Cargo.lock

@@ -986,6 +986,10 @@ dependencies = [
  "serde",
 ]
 
+[[package]]
+name = "entities"
+version = "0.1.0"
+
 [[package]]
 name = "equivalent"
 version = "1.0.2"

Cargo.toml

@@ -1,6 +1,6 @@
 [workspace]
 resolver = "3"
-members = ["./crates/bot", "./crates/migration"]
+members = ["./crates/bot", "./crates/migration", "crates/entities"]
 
 [workspace.package]
 version = "0.1.0"
@@ -12,3 +12,4 @@ name = "discord-server-telemetry"
 tokio = { version = "1.21.2", features = ["macros", "rt-multi-thread"] }
 sea-orm = { version = "2.0.0-rc", features = [ "sqlx-sqlite", "runtime-tokio-rustls", "macros" ] }
 migration = { path = "./crates/migration" }
+entities = { path = "crates/entities" }

crates/entities/Cargo.toml

@@ -0,0 +1,6 @@
+[package]
+name = "entities"
+edition = "2024"
+version.workspace = true
+
+[dependencies]

crates/entities/src/content.rs

@@ -0,0 +1,26 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "content")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id: i32,
+    pub content: String,
+    pub r#type: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(has_many = "super::messages::Entity")]
+    Messages,
+}
+
+impl Related<super::messages::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Messages.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/entities/src/lib.rs

@@ -0,0 +1,8 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+pub mod prelude;
+
+pub mod content;
+pub mod members;
+pub mod messages;
+pub mod server;

crates/entities/src/members.rs

@@ -0,0 +1,17 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "members")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id: i32,
+    pub id_discord: i32,
+    pub name: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/entities/src/messages.rs

@@ -0,0 +1,34 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "messages")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id: i32,
+    pub id_discord: i32,
+    pub id_sender: i32,
+    pub id_content: i32,
+    pub id_server: i32,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::content::Entity",
+        from = "Column::IdContent",
+        to = "super::content::Column::Id",
+        on_update = "NoAction",
+        on_delete = "NoAction"
+    )]
+    Content,
+}
+
+impl Related<super::content::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Content.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/entities/src/prelude.rs

@@ -0,0 +1,6 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+pub use super::content::Entity as Content;
+pub use super::members::Entity as Members;
+pub use super::messages::Entity as Messages;
+pub use super::server::Entity as Server;

crates/entities/src/server.rs

@@ -0,0 +1,17 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "server")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub id: i32,
+    pub id_discord: String,
+    pub name: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/migration/src/main.rs

@@ -9,6 +9,6 @@ async fn main() {
     db.execute_unprepared("PRAGMA foreign_keys = ON;")
         .await
         .unwrap();
-    let db_fn = async move |_| Ok((db));
+    let db_fn = async move |_| Ok(db);
     cli::run_cli_with_connection(migration::Migrator, db_fn).await;
 }