Allow to embed surql files

Author: manelmontilla

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<Regex> = 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<Regex> = 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

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)