migrations_internals/
lib.rs

Help
1// Built-in Lints
2// Clippy lints
3#![allow(
4    clippy::map_unwrap_or,
5    clippy::match_same_arms,
6    clippy::type_complexity
7)]
8#![warn(
9    clippy::unwrap_used,
10    clippy::print_stdout,
11    clippy::mut_mut,
12    clippy::non_ascii_literal,
13    clippy::similar_names,
14    clippy::unicode_not_nfc,
15    clippy::enum_glob_use,
16    clippy::if_not_else,
17    clippy::items_after_statements,
18    clippy::used_underscore_binding,
19    missing_debug_implementations,
20    missing_copy_implementations
21)]
22
23use std::ffi::OsString;
24use std::fs::{DirEntry, File};
25use std::io::Read;
26use std::path::{Path, PathBuf};
27
28#[doc(hidden)]
29#[derive(Debug, serde::Deserialize)]
30#[allow(missing_copy_implementations)]
31pub struct TomlMetadata {
32    #[serde(default)]
33    pub run_in_transaction: bool,
34}
35
36impl Default for TomlMetadata {
37    fn default() -> Self {
38        Self {
39            run_in_transaction: true,
40        }
41    }
42}
43
44impl TomlMetadata {
45    pub const fn new(run_in_transaction: bool) -> Self {
46        Self { run_in_transaction }
47    }
48
49    pub fn read_from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {
50        let mut toml = String::new();
51        let mut file = File::open(path)?;
52        file.read_to_string(&mut toml)?;
53
54        Ok(toml::from_str(&toml)?)
55    }
56}
57
58pub fn search_for_migrations_directory(path: &Path) -> Option<PathBuf> {
59    let migration_path = path.join("migrations");
60    if migration_path.is_dir() {
61        Some(migration_path)
62    } else {
63        path.parent().and_then(search_for_migrations_directory)
64    }
65}
66
67pub fn valid_sql_migration_directory(path: &Path) -> bool {
68    file_names(path).is_ok_and(|files| files.iter().any(|f| f == "up.sql"))
69}
70
71pub fn version_from_string(path: &str) -> Option<String> {
72    path.split('_').next().map(|s| s.replace('-', ""))
73}
74
75fn file_names(path: &Path) -> Result<Vec<OsString>, std::io::Error> {
76    path.read_dir()?
77        .filter_map(|entry| match entry {
78            Ok(entry) if entry.file_name().to_string_lossy().starts_with('.') => None,
79            Ok(entry) => Some(Ok(entry.file_name())),
80            Err(e) => Some(Err(e)),
81        })
82        .collect::<Result<Vec<_>, _>>()
83}
84
85pub fn migrations_directories(
86    path: &'_ Path,
87) -> Result<impl Iterator<Item = Result<DirEntry, std::io::Error>> + '_, std::io::Error> {
88    Ok(path.read_dir()?.filter_map(|entry_res| {
89        entry_res
90            .and_then(|entry| {
91                Ok(
92                    if entry.metadata()?.is_file()
93                        || entry.file_name().to_string_lossy().starts_with('.')
94                    {
95                        None
96                    } else {
97                        Some(entry)
98                    },
99                )
100            })
101            .transpose()
102    }))
103}
migrations_internals/lib.rs

migrations_internals/
lib.rs
