diff --git a/refinery_core/src/util.rs b/refinery_core/src/util.rs index 64c4388..1c5e76d 100644 --- a/refinery_core/src/util.rs +++ b/refinery_core/src/util.rs @@ -23,13 +23,13 @@ fn file_stem_re() -> &'static Regex { /// Matches the stem + extension of a SQL migration file. fn file_re_sql() -> &'static Regex { static RE: OnceLock = OnceLock::new(); - RE.get_or_init(|| Regex::new([STEM_RE, r"\.sql$"].concat().as_str()).unwrap()) + RE.get_or_init(|| Regex::new([STEM_RE, r"\.(sql|surql)$"].concat().as_str()).unwrap()) } /// Matches the stem + extension of any migration file. fn file_re_all() -> &'static Regex { static RE: OnceLock = OnceLock::new(); - RE.get_or_init(|| Regex::new([STEM_RE, r"\.(rs|sql)$"].concat().as_str()).unwrap()) + RE.get_or_init(|| Regex::new([STEM_RE, r"\.(rs|sql|surql)$"].concat().as_str()).unwrap()) } /// enum containing the migration types used to search for migrations diff --git a/refinery_macros/src/lib.rs b/refinery_macros/src/lib.rs index cd790cc..71e694c 100644 --- a/refinery_macros/src/lib.rs +++ b/refinery_macros/src/lib.rs @@ -120,7 +120,7 @@ pub fn embed_migrations(input: TokenStream) -> TokenStream { let extension = migration.extension().unwrap(); migration_filenames.push(filename.clone()); - if extension == "sql" { + if extension == "sql" || extension == "surql" { _migrations.push(quote! {(#filename, include_str!(#path).to_string())}); } else if extension == "rs" { let rs_content = fs::read_to_string(&path)