caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix the handling of new mod files. (#2064)

Browse Source 
Co-authored-by: ryankopf <git@ryankopf.com>
This commit is contained in:
Ryan Kopf 2024-02-28 14:03:05 -06:00 committed by GitHub
parent 94ff8881d0
commit 8025199499
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
sea-orm-cli/src/commands/migrate.rs
View File

@ -226,7 +226,11 @@ fn update_migrator(migration_name: &str, migration_dir: &str) -> Result<(), Box<
// find existing mod declarations, add new line
let mod_regex = Regex::new(r"mod\s+(?P<name>m\d{8}_\d{6}_\w+);")?;
let mods: Vec<_> = mod_regex.captures_iter(&migrator_content).collect();
let mods_end = mods.last().unwrap().get(0).unwrap().end() + 1;
let mods_end = if let Some(last_match) = mods.last() {
last_match.get(0).unwrap().end() + 1
} else {
migrator_content.len()
};
updated_migrator_content.insert_str(mods_end, format!("mod {migration_name};\n").as_str());
// build new vector from declared migration modules