This commit is contained in:
2026-03-25 21:38:34 +00:00
parent 86c150dff1
commit ba4a702574
10 changed files with 128 additions and 118 deletions
Generated
+29 -92
View File
@@ -563,7 +563,8 @@ dependencies = [
name = "bot" name = "bot"
version = "0.1.0" version = "0.1.0"
dependencies = [ dependencies = [
"sea-orm 2.0.0-rc.37", "migration",
"sea-orm",
"serenity", "serenity",
"tokio", "tokio",
] ]
@@ -1641,6 +1642,15 @@ dependencies = [
"serde_core", "serde_core",
] ]
[[package]]
name = "indoc"
version = "2.0.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "79cf5c93f93228cf8efb3ba362535fb11199ac548a09ce117c9b1adc3030d706"
dependencies = [
"rustversion",
]
[[package]] [[package]]
name = "inherent" name = "inherent"
version = "1.0.13" version = "1.0.13"
@@ -2788,27 +2798,6 @@ dependencies = [
"syn 2.0.117", "syn 2.0.117",
] ]
[[package]]
name = "sea-orm"
version = "1.1.19"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6d945f62558fac19e5988680d2fdf747b734c2dbc6ce2cb81ba33ed8dde5b103"
dependencies = [
"async-stream",
"async-trait",
"derive_more",
"futures-util",
"log",
"ouroboros",
"sea-orm-macros 1.1.19",
"sea-query 0.32.7",
"serde",
"strum 0.26.3",
"thiserror 2.0.18",
"tracing",
"url",
]
[[package]] [[package]]
name = "sea-orm" name = "sea-orm"
version = "2.0.0-rc.37" version = "2.0.0-rc.37"
@@ -2828,14 +2817,14 @@ dependencies = [
"pgvector", "pgvector",
"rust_decimal", "rust_decimal",
"sea-orm-arrow", "sea-orm-arrow",
"sea-orm-macros 2.0.0-rc.37", "sea-orm-macros",
"sea-query 1.0.0-rc.31", "sea-query",
"sea-query-sqlx", "sea-query-sqlx",
"sea-schema 0.17.0-rc.17", "sea-schema",
"serde", "serde",
"serde_json", "serde_json",
"sqlx", "sqlx",
"strum 0.27.2", "strum",
"thiserror 2.0.18", "thiserror 2.0.18",
"time", "time",
"tracing", "tracing",
@@ -2850,40 +2839,30 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5c2eee8405f16c1f337fe3a83389361caea83c928d14dbd666a480407072c365" checksum = "5c2eee8405f16c1f337fe3a83389361caea83c928d14dbd666a480407072c365"
dependencies = [ dependencies = [
"arrow", "arrow",
"sea-query 1.0.0-rc.31", "sea-query",
"thiserror 2.0.18", "thiserror 2.0.18",
] ]
[[package]] [[package]]
name = "sea-orm-cli" name = "sea-orm-cli"
version = "1.1.19" version = "2.0.0-rc.37"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c94492e2ab6c045b4cc38013809ce255d14c3d352c9f0d11e6b920e2adc948ad" checksum = "cd9b34d4c8e615079c04eb7863a429c2d2a8bf9c934eb9eeb580f51f36367124"
dependencies = [ dependencies = [
"chrono", "chrono",
"clap", "clap",
"dotenvy", "dotenvy",
"glob", "glob",
"indoc",
"regex", "regex",
"sea-schema",
"sqlx",
"tokio",
"tracing", "tracing",
"tracing-subscriber", "tracing-subscriber",
"url", "url",
] ]
[[package]]
name = "sea-orm-macros"
version = "1.1.19"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "84c2e64a50a9cc8339f10a27577e10062c7f995488e469f2c95762c5ee847832"
dependencies = [
"heck 0.5.0",
"proc-macro2",
"quote",
"sea-bae",
"syn 2.0.117",
"unicode-ident",
]
[[package]] [[package]]
name = "sea-orm-macros" name = "sea-orm-macros"
version = "2.0.0-rc.37" version = "2.0.0-rc.37"
@@ -2902,31 +2881,20 @@ dependencies = [
[[package]] [[package]]
name = "sea-orm-migration" name = "sea-orm-migration"
version = "1.1.19" version = "2.0.0-rc.37"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7315c0cadb7e60fb17ee2bb282aa27d01911fc2a7e5836ec1d4ac37d19250bb4" checksum = "b3ceb928aac8be83332d34d1fdbc827d43696135a800723ffeb2e0b33b7b495e"
dependencies = [ dependencies = [
"async-trait", "async-trait",
"clap", "clap",
"dotenvy", "dotenvy",
"sea-orm 1.1.19", "sea-orm",
"sea-orm-cli", "sea-orm-cli",
"sea-schema 0.16.2", "sea-schema",
"tracing", "tracing",
"tracing-subscriber", "tracing-subscriber",
] ]
[[package]]
name = "sea-query"
version = "0.32.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8a5d1c518eaf5eda38e5773f902b26ab6d5e9e9e2bb2349ca6c64cf96f80448c"
dependencies = [
"inherent",
"ordered-float",
"sea-query-derive 0.4.3",
]
[[package]] [[package]]
name = "sea-query" name = "sea-query"
version = "1.0.0-rc.31" version = "1.0.0-rc.31"
@@ -2937,26 +2905,12 @@ dependencies = [
"inherent", "inherent",
"ordered-float", "ordered-float",
"rust_decimal", "rust_decimal",
"sea-query-derive 1.0.0-rc.12", "sea-query-derive",
"serde_json", "serde_json",
"time", "time",
"uuid", "uuid",
] ]
[[package]]
name = "sea-query-derive"
version = "0.4.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bae0cbad6ab996955664982739354128c58d16e126114fe88c2a493642502aab"
dependencies = [
"darling",
"heck 0.4.1",
"proc-macro2",
"quote",
"syn 2.0.117",
"thiserror 2.0.18",
]
[[package]] [[package]]
name = "sea-query-derive" name = "sea-query-derive"
version = "1.0.0-rc.12" version = "1.0.0-rc.12"
@@ -2977,21 +2931,10 @@ version = "0.8.0-rc.14"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e4377164b09a11bb692dec6966eb0e6908d63d768defef0be689b39e02cf8544" checksum = "e4377164b09a11bb692dec6966eb0e6908d63d768defef0be689b39e02cf8544"
dependencies = [ dependencies = [
"sea-query 1.0.0-rc.31", "sea-query",
"sqlx", "sqlx",
] ]
[[package]]
name = "sea-schema"
version = "0.16.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2239ff574c04858ca77485f112afea1a15e53135d3097d0c86509cef1def1338"
dependencies = [
"futures",
"sea-query 0.32.7",
"sea-schema-derive",
]
[[package]] [[package]]
name = "sea-schema" name = "sea-schema"
version = "0.17.0-rc.17" version = "0.17.0-rc.17"
@@ -2999,7 +2942,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b363dd21c20fe4d1488819cb2bc7f8d4696c62dd9f39554f97639f54d57dd0ab" checksum = "b363dd21c20fe4d1488819cb2bc7f8d4696c62dd9f39554f97639f54d57dd0ab"
dependencies = [ dependencies = [
"async-trait", "async-trait",
"sea-query 1.0.0-rc.31", "sea-query",
"sea-query-sqlx", "sea-query-sqlx",
"sea-schema-derive", "sea-schema-derive",
"sqlx", "sqlx",
@@ -3496,12 +3439,6 @@ version = "0.11.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
[[package]]
name = "strum"
version = "0.26.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8fec0f0aef304996cf250b31b5a10dee7980c85da9d759361292b8bca5a18f06"
[[package]] [[package]]
name = "strum" name = "strum"
version = "0.27.2" version = "0.27.2"
+1
View File
@@ -11,3 +11,4 @@ name = "discord-server-telemetry"
[workspace.dependencies] [workspace.dependencies]
tokio = { version = "1.21.2", features = ["macros", "rt-multi-thread"] } tokio = { version = "1.21.2", features = ["macros", "rt-multi-thread"] }
sea-orm = { version = "2.0.0-rc", features = [ "sqlx-sqlite", "runtime-tokio-rustls", "macros" ] } sea-orm = { version = "2.0.0-rc", features = [ "sqlx-sqlite", "runtime-tokio-rustls", "macros" ] }
migration = { path = "./crates/migration" }
+1
View File
@@ -8,3 +8,4 @@ default-run = "bot"
serenity = "0.12" serenity = "0.12"
tokio = {workspace = true} tokio = {workspace = true}
sea-orm = {workspace = true} sea-orm = {workspace = true}
migration ={workspace = true}
+2
View File
@@ -1,6 +1,7 @@
use std::env; use std::env;
use std::error::Error; use std::error::Error;
use migration::{Migrator, MigratorTrait};
use sea_orm::{Database, DatabaseConnection}; use sea_orm::{Database, DatabaseConnection};
use serenity::async_trait; use serenity::async_trait;
use serenity::model::channel::Message; use serenity::model::channel::Message;
@@ -25,6 +26,7 @@ async fn main() -> Result<(), Box<dyn Error>> {
let token = env::var("DISCORD_TOKEN").expect("Expected a token in the environment"); let token = env::var("DISCORD_TOKEN").expect("Expected a token in the environment");
let dbc: DatabaseConnection = Database::connect("sqlite::memory:").await?; let dbc: DatabaseConnection = Database::connect("sqlite::memory:").await?;
Migrator::up(&dbc, None).await?;
// Set gateway intents, which decides what events the bot will be notified about // Set gateway intents, which decides what events the bot will be notified about
let intents = GatewayIntents::GUILD_MESSAGES let intents = GatewayIntents::GUILD_MESSAGES
+2 -6
View File
@@ -12,11 +12,7 @@ path = "src/lib.rs"
async-std = { version = "1", features = ["attributes", "tokio1"] } async-std = { version = "1", features = ["attributes", "tokio1"] }
[dependencies.sea-orm-migration] [dependencies.sea-orm-migration]
version = "1.1.0" version = "2.0.0-rc"
features = [ features = [
# Enable at least one `ASYNC_RUNTIME` and `DATABASE_DRIVER` feature if you want to run migration via CLI. "sqlx-sqlite", "runtime-tokio-rustls"
# View the list of supported features at https://www.sea-ql.org/SeaORM/docs/install-and-config/database-and-async-runtime.
# e.g.
# "runtime-tokio-rustls", # `ASYNC_RUNTIME` feature
# "sqlx-postgres", # `DATABASE_DRIVER` feature
] ]
+8 -2
View File
@@ -1,12 +1,18 @@
pub use sea_orm_migration::prelude::*; pub use sea_orm_migration::prelude::*;
mod m20220101_000001_create_table; mod m20260325_145211_servers;
mod m20260325_145717_members;
mod m20260325_162920_messages;
pub struct Migrator; pub struct Migrator;
#[async_trait::async_trait] #[async_trait::async_trait]
impl MigratorTrait for Migrator { impl MigratorTrait for Migrator {
fn migrations() -> Vec<Box<dyn MigrationTrait>> { fn migrations() -> Vec<Box<dyn MigrationTrait>> {
vec![Box::new(m20220101_000001_create_table::Migration)] vec![
Box::new(m20260325_145211_servers::Migration),
Box::new(m20260325_145717_members::Migration),
Box::new(m20260325_162920_messages::Migration),
]
} }
} }
@@ -6,36 +6,30 @@ pub struct Migration;
#[async_trait::async_trait] #[async_trait::async_trait]
impl MigrationTrait for Migration { impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> { async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Replace the sample below with your own migration scripts
todo!();
manager manager
.create_table( .create_table(
Table::create() Table::create()
.table(Post::Table) .table(Server::Table)
.if_not_exists() .if_not_exists()
.col(pk_auto(Post::Id)) .col(pk_auto(Server::Id))
.col(string(Post::Title)) .col(string(Server::IdDiscord))
.col(string(Post::Text)) .col(string(Server::Name))
.to_owned(), .to_owned(),
) )
.await .await
} }
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> { async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Replace the sample below with your own migration scripts
todo!();
manager manager
.drop_table(Table::drop().table(Post::Table).to_owned()) .drop_table(Table::drop().table(Server::Table).to_owned())
.await .await
} }
} }
#[derive(DeriveIden)] #[derive(DeriveIden)]
enum Post { enum Server {
Table, Table,
Id, Id,
Title, IdDiscord,
Text, Name,
} }
@@ -0,0 +1,35 @@
use sea_orm_migration::{prelude::*, schema::*};
#[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(Members::Table)
.if_not_exists()
.col(pk_auto(Members::Id))
.col(integer(Members::IdDiscord))
.col(string(Members::Name))
.to_owned(),
)
.await
}
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.drop_table(Table::drop().table(Members::Table).to_owned())
.await
}
}
#[derive(DeriveIden)]
enum Members {
Table,
Id,
IdDiscord,
Name,
}
@@ -0,0 +1,39 @@
use sea_orm_migration::{prelude::*, schema::*};
#[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(Messages::Table)
.if_not_exists()
.col(pk_auto(Messages::Id))
.col(integer(Messages::IdDiscord))
.col(integer(Messages::IdSender))
.col(integer(Messages::IdContent)) // Change to foreign key
.col(integer(Messages::IdServer))
.to_owned(),
)
.await
}
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.drop_table(Table::drop().table(Messages::Table).to_owned())
.await
}
}
#[derive(DeriveIden)]
enum Messages {
Table,
Id,
IdDiscord,
IdSender,
IdContent,
IdServer,
}
+3 -4
View File
@@ -188,13 +188,12 @@
# Inherit inputs from checks. # Inherit inputs from checks.
checks = self.checks.${system}; checks = self.checks.${system};
# Additional dev-shell environment variables can be set directly DATABASE_URL = "sqlite://./db.sqlite?mode=rwc";
# MY_CUSTOM_DEVELOPMENT_VAR = "something else";
# Extra inputs can be added here; cargo and rustc are provided by default.
packages = [ packages = [
pkgs.cargo-hakari pkgs.cargo-hakari
pkgs.sea-orm-cli pkgs.sea-orm-cli
pkgs.clang
pkgs.mold
]; ];
}; };
} }