Included StructureSafe feature, fixed tokio runtime flavor

Author: myOmikron

Cargo.toml

@@ -43,7 +43,7 @@ regex = { version = "~1" }
 once_cell = { version = "~1" }
 
 # Runtime
-tokio = { version = ">=1.23.1", features = ["macros"], optional = true }
+tokio = { version = ">=1.23.1", features = ["macros", "rt"], optional = true }
 
 # Enum simplification
 strum = { version = "~0.25" }

src/main.rs

@@ -11,7 +11,7 @@ pub mod migrate;
 pub mod squash_migrations;
 pub mod utils;
 
-#[tokio::main]
+#[tokio::main(flavor = "current_thread")]
 async fn main() -> anyhow::Result<()> {
     let cli: Cli = Cli::parse();
 

src/migrate/sql_builder.rs

@@ -114,6 +114,7 @@ pub async fn migration_to_sql<'a>(
                 mysql,
                 postgres,
                 sqlite,
+                ..
             } => match db_impl {
                 DBImpl::SQLite => {
                     execute_statement(tx, sqlite.clone(), Vec::new(), do_log).await?;

src/utils/migrations.rs

@@ -239,14 +239,21 @@ pub fn convert_migrations_to_internal_models(
                 }
                 Ok(())
             }
-            Operation::RawSQL { .. } => Err(anyhow!(
-                r#"RawSQL migration found!
+            Operation::RawSQL { structure_safe, .. } => {
+                if *structure_safe {
+                    Ok(())
+                } else {
+                    Err(anyhow!(
+                        r#"RawSQL migration without StructureSafe flag found!
 
 Can not proceed to generate migrations as the current database state can not be determined anymore!
 You can still write migrations with all available operations yourself.
 
-To use the make-migrations feature again, delete all RawSQL operations."#
-            )),
+To use the make-migrations feature again, check that all RawSQL operations don't change any 
+structure and mark them as StructureSafe or delete all RawSQL operations."#
+                    ))
+                }
+            }
         })
     });
 

src/utils/re.rs

@@ -9,8 +9,8 @@ pub struct Regexes {
 }
 
 pub static RE: Lazy<Regexes> = Lazy::new(|| Regexes {
-    numeric_only: Regex::new(r#"^\d+$"#).unwrap(),
+    numeric_only: Regex::new(r"^\d+$").unwrap(),
     forbidden_character: Regex::new(r#"[^a-zA-Z0-9_]"#).unwrap(),
-    migration_allowed_comment: Regex::new(r#"^\w+$"#).unwrap(),
-    migration_allowed_name: Regex::new(r#"^[0-9]{4}_\w+\.toml$"#).unwrap(),
+    migration_allowed_comment: Regex::new(r"^\w+$").unwrap(),
+    migration_allowed_name: Regex::new(r"^[0-9]{4}_\w+\.toml$").unwrap(),
 });