Add migration parameters

Issue launchbadge#3178

Added migration parameters in a similar fashion to goose's env var
migration parameters. This still needs some tests and documentation.
I wanted to get a proposal for a solution up for folks to take a look
at.

Author: krlohnes

Cargo.lock

@@ -1,4 +1,5 @@
 use anyhow::{bail, Context};
+use regex::Regex;
 use console::style;
 use std::fs::{self, File};
 use std::io::{Read, Write};

sqlx-cli/src/migration.rs

@@ -67,7 +67,7 @@ log = { version = "0.4.18", default-features = false }
 memchr = { version = "2.4.1", default-features = false }
 once_cell = "1.9.0"
 percent-encoding = "2.1.0"
-regex = { version = "1.5.5", optional = true }
+regex = { version = "1.5.5"}
 serde = { version = "1.0.132", features = ["derive", "rc"], optional = true }
 serde_json = { version = "1.0.73", features = ["raw_value"], optional = true }
 sha2 = { version = "0.10.0", default-features = false, optional = true }
@@ -82,6 +82,7 @@ hashlink = "0.10.0"
 indexmap = "2.0"
 event-listener = "5.2.0"
 hashbrown = "0.15.0"
+subst = "0.3.7"
 
 [dev-dependencies]
 sqlx = { workspace = true, features = ["postgres", "sqlite", "mysql", "migrate", "macros", "time", "uuid"] }

sqlx-core/Cargo.toml

@@ -1,9 +1,13 @@
+use regex::{Captures, Regex};
 use std::borrow::Cow;
+use std::sync::OnceLock;
 
 use sha2::{Digest, Sha384};
 
 use super::MigrationType;
 
+static ENV_SUB_REGEX: OnceLock<Regex> = OnceLock::new();
+
 #[derive(Debug, Clone)]
 pub struct Migration {
     pub version: i64,
@@ -23,12 +27,22 @@ impl Migration {
         no_tx: bool,
     ) -> Self {
         let checksum = Cow::Owned(Vec::from(Sha384::digest(sql.as_bytes()).as_slice()));
-
+        let re = ENV_SUB_REGEX.get_or_init(|| {
+            Regex::new(r"--\s?sqlx envsub on((.|\n|\r)*?)--\s?sqlx envsub off").unwrap()
+        });
+        let envsub_sql = re
+            .replace_all(&sql, |cap: &Captures<'_>| {
+                format!(
+                    "-- sqlx envsub on{}-- sqlx envsub off",
+                    subst::substitute(&cap[1], &subst::Env).unwrap_or(cap[1].to_owned())
+                )
+            })
+            .into_owned();
         Migration {
             version,
             description,
             migration_type,
-            sql,
+            sql: Cow::Owned(envsub_sql),
             checksum,
             no_tx,
         }