rustyroad 1.8.7

Rusty Road is a framework written in Rust that is based on Ruby on Rails. It is designed to provide the familiar conventions and ease of use of Ruby on Rails, while also taking advantage of the performance and efficiency of Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use regex::Regex;

pub(in crate::database::migrations::sql_migration_converter) const IDENT: &str =
    r#"(?:"[^"]+"|`[^`]+`|\[[^\]]+\]|\w+)"#;

pub(in crate::database::migrations::sql_migration_converter) fn qualified() -> String {
    format!(r"{IDENT}(?:\s*\.\s*{IDENT})*")
}

pub(in crate::database::migrations::sql_migration_converter) fn clean(name: &str) -> String {
    name.split('.').map(str::trim).collect::<Vec<_>>().join(".")
}

pub(super) enum Action {
    Add(String),
    Drop(String),
    Constraint(String),
    Raw(String),
}

pub(super) fn classify(action: &str) -> Action {
    let id = format!("({IDENT})");
    let capture = |pattern: String| Regex::new(&pattern).unwrap();
    if let Some(c) = capture(format!(r"(?i)^ADD\s+CONSTRAINT\s+{id}")).captures(action) {
        return Action::Constraint(c[1].to_string());
    }
    if Regex::new(r"(?i)^ADD\s+(PRIMARY\s+KEY|FOREIGN\s+KEY|UNIQUE|CHECK|EXCLUDE)\b")
        .unwrap()
        .is_match(action)
        || Regex::new(r"(?i)^DROP\s+(CONSTRAINT|DEFAULT|NOT)\b")
            .unwrap()
            .is_match(action)
    {
        return Action::Raw(action.to_string());
    }
    let add = capture(format!(
        r"(?i)^ADD\s+(?:COLUMN\s+)?(?:IF\s+NOT\s+EXISTS\s+)?{id}"
    ));
    if let Some(c) = add.captures(action) {
        return Action::Add(clean(&c[1]));
    }
    let drop = capture(format!(
        r"(?i)^DROP\s+(?:COLUMN\s+)?(?:IF\s+EXISTS\s+)?{id}"
    ));
    drop.captures(action).map_or_else(
        || Action::Raw(action.to_string()),
        |c| Action::Drop(clean(&c[1])),
    )
}