dbcrab 0.5.0

Modern REPL-first PostgreSQL client.
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

use reedline::{ValidationResult, Validator};

use crate::sql::split_complete_statements;

pub struct SqlValidator;

impl Validator for SqlValidator {
    fn validate(&self, line: &str) -> ValidationResult {
        if is_complete_sql_input(line) {
            ValidationResult::Complete
        } else {
            ValidationResult::Incomplete
        }
    }
}

pub fn is_complete_sql_input(input: &str) -> bool {
    if input.trim().is_empty() {
        return true;
    }

    let (statements, rest) = split_complete_statements(input);
    !statements.is_empty() && rest.trim().is_empty()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn empty_input_is_complete() {
        // Given
        let input = "   ";

        // When
        let complete = is_complete_sql_input(input);

        // Then
        assert!(complete);
    }

    #[test]
    fn input_without_semicolon_is_incomplete() {
        // Given
        let input = "select 1";

        // When
        let complete = is_complete_sql_input(input);

        // Then
        assert!(!complete);
    }

    #[test]
    fn input_with_trailing_incomplete_statement_is_incomplete() {
        // Given
        let input = "select 1;\nselect 2";

        // When
        let complete = is_complete_sql_input(input);

        // Then
        assert!(!complete);
    }

    #[test]
    fn semicolon_inside_string_does_not_complete_input() {
        // Given
        let input = "select ';'";

        // When
        let complete = is_complete_sql_input(input);

        // Then
        assert!(!complete);
    }

    #[test]
    fn semicolon_terminated_multiline_statement_is_complete() {
        // Given
        let input = "select *\nfrom users\nwhere id = 1;";

        // When
        let complete = is_complete_sql_input(input);

        // Then
        assert!(complete);
    }
}