migrations_macros/
embed_migrations.rs

Help
1use crate::migrations::migration_directory_from_given_path;
2use migrations_internals::{migrations_directories, version_from_string, TomlMetadata};
3use quote::quote;
4use std::error::Error;
5use std::fs::DirEntry;
6use std::path::Path;
7
8pub fn expand(path: String) -> proc_macro2::TokenStream {
9    let migrations_path_opt = if path.is_empty() {
10        None
11    } else {
12        Some(path.replace('"', ""))
13    };
14    let migrations_expr = migration_directory_from_given_path(migrations_path_opt.as_deref())
15        .unwrap_or_else(|_| {
16            panic!(
17                "Failed to receive migrations dir from {:?}",
18                migrations_path_opt
19            )
20        });
21    let embedded_migrations =
22        migration_literals_from_path(&migrations_expr).expect("Failed to read migration literals");
23
24    quote! {
25        diesel_migrations::EmbeddedMigrations::new(&[#(#embedded_migrations,)*])
26    }
27}
28
29fn migration_literals_from_path(
30    path: &Path,
31) -> Result<Vec<proc_macro2::TokenStream>, Box<dyn Error>> {
32    let mut migrations = migrations_directories(path)?.collect::<Result<Vec<_>, _>>()?;
33
34    migrations.sort_by_key(DirEntry::path);
35
36    Ok(migrations
37        .into_iter()
38        .map(|e| migration_literal_from_path(&e.path()))
39        .collect())
40}
41
42fn migration_literal_from_path(path: &Path) -> proc_macro2::TokenStream {
43    let name = path
44        .file_name()
45        .unwrap_or_else(|| panic!("Can't get file name from path `{:?}`", path))
46        .to_string_lossy();
47    if version_from_string(&name).is_none() {
48        panic!(
49            "Invalid migration directory: the directory's name should be \
50             <timestamp>_<name_of_migration>, and it should contain \
51             up.sql and optionally down.sql."
52        );
53    }
54    let up_sql_path = path.join("up.sql");
55    let up_sql_path = up_sql_path.to_str();
56    let down_sql_path = path.join("down.sql");
57    let metadata = TomlMetadata::read_from_file(&path.join("metadata.toml")).unwrap_or_default();
58    let run_in_transaction = metadata.run_in_transaction;
59
60    let down_sql = match down_sql_path.metadata() {
61        Err(e) if e.kind() == std::io::ErrorKind::NotFound => quote! { None },
62        _ => {
63            let down_sql_path = down_sql_path.to_str();
64            quote! { Some(include_str!(#down_sql_path)) }
65        }
66    };
67
68    quote!(diesel_migrations::EmbeddedMigration::new(
69        include_str!(#up_sql_path),
70        #down_sql,
71        diesel_migrations::EmbeddedName::new(#name),
72        diesel_migrations::TomlMetadataWrapper::new(#run_in_transaction)
73    ))
74}
migrations_macros/embed_migrations.rs

migrations_macros/
embed_migrations.rs
