rust-sql-organizer 1.0.0

This is a simple rust CLI tool to organize .sql files. This is a useful tool for my work, but I decided to share it as a project example
Documentation 
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use lazy_static::lazy_static;

use crate::sql_file::SqlFile;
use regex::Regex;

type FileFormatterFunc = fn(&SqlFile) -> String;

pub struct FileFormatters {
    formatters: Vec<FileFormatterFunc>,
}

impl FileFormatters {
    pub fn format(&self, sql_file: &SqlFile) -> String {
        self.formatters
            .iter()
            .map(|&fmt| fmt(sql_file))
            .collect::<Vec<String>>()
            .join("\n")
    }

    #[cfg(test)]
    pub fn test_new(formatter: FileFormatterFunc) -> FileFormatters {
        FileFormatters {
            formatters: vec![formatter],
        }
    }
}

pub fn format_name(sql_file: &SqlFile) -> String {
    format!("-- {}", sql_file.get_file_name())
}

pub fn format_split(_: &SqlFile) -> String {
    "\n".to_string()
}

pub fn format_plain_text(sql_file: &SqlFile) -> String {
    sql_file.get_sql_text().to_string()
}

pub fn format_no_comments(sql_file: &SqlFile) -> String {
    REMOVE_COMMENTS_RE
        .replace(sql_file.get_sql_text(), "USE")
        .to_string()
}

pub fn format_endln(sql_file: &SqlFile) -> String {
    format!(
        "--____________________ End of {} ____________________--",
        sql_file.get_file_name()
    )
}

lazy_static! {
    static ref REMOVE_COMMENTS_RE: Regex = Regex::new(r"(?mi)--\s+use").unwrap();
    pub static ref STANDARD_FILE_FORMATTER: FileFormatters = FileFormatters {
        formatters: vec![
            format_name,
            format_split,
            format_split,
            format_plain_text,
            format_split,
            format_split,
            format_endln,
            format_split
        ]
    };
    pub static ref FILE_FORMATTER_NO_COMMENTS: FileFormatters = FileFormatters {
        formatters: vec![
            format_name,
            format_split,
            format_split,
            format_no_comments,
            format_split,
            format_split,
            format_endln,
            format_split
        ]
    };
}

#[cfg(test)]
mod test_file_formatter {
    use crate::sql_file::SqlFile;

    use super::{format_endln, format_name, format_no_comments, format_plain_text, format_split};

    #[test]
    fn test_format_name() {
        assert_eq!(
            format_name(&SqlFile::test_new("test", "SELECT 1;")),
            "-- test"
        )
    }

    #[test]
    fn test_format_split() {
        assert_eq!(format_split(&SqlFile::test_new("test", "SELECT 1;")), "\n")
    }

    #[test]
    fn test_format_plain_text() {
        assert_eq!(
            format_plain_text(&SqlFile::test_new("test", "SELECT 1;")),
            "SELECT 1;"
        )
    }

    #[test]
    fn test_format_no_comments() {
        assert_eq!(
            format_no_comments(&SqlFile::test_new(
                "test",
                "-- use ROLE ACCOUNTADMIN;\nSELECT 1;"
            )),
            "USE ROLE ACCOUNTADMIN;\nSELECT 1;"
        )
    }

    #[test]
    fn test_format_endln() {
        assert_eq!(
            format_endln(&SqlFile::test_new("test", "SELECT 1;")),
            "--____________________ End of test ____________________--"
        )
    }
}