fix: integrate SQL migrations into binary

Author: lekoOwO

.github/workflows/ci.yml

@@ -32,7 +32,7 @@ jobs:
         run: |
           cargo install sqlx-cli
           cargo sqlx database create
-          cargo sqlx migrate run
+          cargo sqlx migrate run --source src/migrations
           cargo sqlx prepare
       
       - name: Cache dependencies
@@ -176,13 +176,15 @@ jobs:
             bin: InvitationBot
             toolchain: beta
             cross: false
+            skip-tests: true
 
           - os-name: Linux-x86_64
             runs-on: ubuntu-22.04
             target: x86_64-unknown-linux-musl
             bin: InvitationBot
             toolchain: nightly
             cross: false
+            skip-tests: true
 
     steps:
       - uses: actions/checkout@v4
@@ -226,7 +228,7 @@ jobs:
         run: |
           cargo install sqlx-cli
           cargo sqlx database create
-          cargo sqlx migrate run
+          cargo sqlx migrate run --source src/migrations
           cargo sqlx prepare
 
       - name: Cache dependencies
@@ -306,7 +308,7 @@ jobs:
 
             [ -f "${dir}/${CRATE_NAME}" ] && mv "${dir}/${CRATE_NAME}" "release/${dir_name}"
             [ -f "${dir}/${CRATE_NAME}.exe" ] && mv "${dir}/${CRATE_NAME}.exe" "release/${dir_name}.exe"
-            
+
           done
 
           ls -la release

Dockerfile

@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \
 ENV DATABASE_URL=sqlite:/tmp/bot.db
 RUN cargo install sqlx-cli && \
     sqlx database create && \
-    sqlx migrate run
+    sqlx migrate run --source src/migrations
 
 RUN cargo build --release
 

src/http_server.rs

@@ -173,7 +173,6 @@ mod tests {
     async fn setup_test_app() -> (Router, SqlitePool) {
         let db_url = format!("sqlite:file:{}?mode=memory", Uuid::new_v4());
         let pool = crate::utils::db::create_pool(&db_url).await.unwrap();
-        sqlx::migrate!().run(&pool).await.unwrap();
 
         let config = Arc::new(Config {
             bot: BotConfig {

src/main.rs

@@ -6,6 +6,7 @@ use std::collections::HashMap;
 mod handlers;
 mod http_server;
 mod i18n;
+mod migrations;
 mod slash_commands;
 mod utils;
 

…rations/20240101000000_create_tables.sql …rations/20240101000000_create_tables.sqlmigrations/20240101000000_create_tables.sql renamed to src/migrations/20240101000000_create_tables.sql migrations/20240101000000_create_tables.sql renamed to src/migrations/20240101000000_create_tables.sql

@@ -0,0 +1,36 @@
+use rust_embed::RustEmbed;
+
+#[derive(RustEmbed)]
+#[folder = "src/migrations"]
+pub struct Migrations;
+
+pub fn get_migration(name: &str) -> Option<String> {
+    Migrations::get(name).map(|f| String::from_utf8_lossy(f.data.as_ref()).into_owned())
+}
+
+pub fn get_migrations() -> Vec<String> {
+    Migrations::iter().map(|f| f.to_string()).collect()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_get_migration() {
+        // Test existing migration
+        let migration =
+            get_migration("20240101000000_create_tables.sql").expect("Failed to load migration");
+        assert!(migration.contains("CREATE TABLE IF NOT EXISTS invites"));
+
+        // Test non-existing migration
+        assert!(get_migration("invalid.sql").is_none());
+    }
+
+    #[test]
+    fn test_get_migrations() {
+        let migrations = get_migrations();
+        assert!(!migrations.is_empty());
+        assert!(migrations.contains(&"20240101000000_create_tables.sql".to_string()));
+    }
+}

src/migrations/mod.rs

@@ -2,7 +2,12 @@ use sqlx::types::time::OffsetDateTime;
 type Pool = sqlx::Pool<sqlx::Sqlite>;
 
 pub async fn setup_database(pool: &Pool) -> Result<(), sqlx::Error> {
-    sqlx::migrate!().run(pool).await?;
+    let migrations = crate::migrations::get_migrations();
+    for migration in migrations {
+        if let Some(sql) = crate::migrations::get_migration(migration.as_str()) {
+            sqlx::query(&sql).execute(pool).await?;
+        }
+    }
     Ok(())
 }
 
@@ -195,7 +200,6 @@ mod tests {
     async fn setup_test_db() -> Pool {
         let db_url = format!("sqlite:file:{}?mode=memory", Uuid::new_v4());
         let pool = create_pool(&db_url).await.unwrap();
-        sqlx::migrate!().run(&pool).await.unwrap();
         pool
     }
 

src/utils/db.rs

@@ -10,7 +10,6 @@ impl TestContext {
     pub async fn new() -> Self {
         let db_url = format!("sqlite:file:{}?mode=memory", Uuid::new_v4());
         let pool = crate::utils::db::create_pool(&db_url).await.unwrap();
-        sqlx::migrate!().run(&pool).await.unwrap();
 
         let config = crate::utils::config::Config {
             bot: crate::utils::config::BotConfig {