datasources-sqlite 0.1.48

SQLite datasource provider
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

use crate::default_value_converter::DefaultValueConverter;
use crate::default_value_options::DefaultValueOptions;
use crate::type_map::TypeMap;
use datasources::{Result, SpecField};
use regex::Regex;
use std::collections::HashMap;
use std::sync::LazyLock;

pub const QUOTE_LEFT: &str = "\"";
pub const QUOTE_RIGHT: &str = "\"";
pub const PROCEDURE_PAYLOAD_TYPE: &str = "JSON";

pub static TYPES: LazyLock<TypeMap> = LazyLock::new(|| {
    TypeMap::new(HashMap::from([
        ("string".into(), "TEXT".into()),
        ("character".into(), "TEXT".into()),
        ("integer".into(), "INTEGER".into()),
        ("unsignedinteger".into(), "INTEGER".into()),
        ("smallinteger".into(), "INTEGER".into()),
        ("unsignedsmallinteger".into(), "INTEGER".into()),
        ("biginteger".into(), "INTEGER".into()),
        ("unsignedbiginteger".into(), "INTEGER".into()),
        ("float".into(), "REAL".into()),
        ("decimal".into(), "NUMERIC".into()),
        ("boolean".into(), "BOOLEAN".into()),
        ("datetime".into(), "TEXT".into()),
        ("binary".into(), "BLOB".into()),
        ("uuid".into(), "TEXT".into()),
        ("reference".into(), "INTEGER".into()),
    ]))
});

pub fn convert_type(field: &SpecField) -> Result<String> {
    TYPES.convert_field(field)
}

pub fn convert_default(spec_type: &str, text: &str) -> Result<Option<String>> {
    DEFAULTS.convert(spec_type, text)
}

pub static DEFAULTS: LazyLock<DefaultValueConverter> = LazyLock::new(|| {
    DefaultValueConverter::new(DefaultValueOptions {
        new_id: || "(lower(hex(randomblob(4))) || '-' || lower(hex(randomblob(2))) || '-4' || substr(lower(hex(randomblob(2))), 2) || '-' || substr('89ab', abs(random()) % 4 + 1, 1) || substr(lower(hex(randomblob(2))), 2) || '-' || lower(hex(randomblob(6))))".into(),
        now: || "(strftime('%Y-%m-%dT%H:%M:%f', 'now', 'localtime'))".into(),
        utc_now: || "(strftime('%Y-%m-%dT%H:%M:%fZ', 'now'))".into(),
        boolean_on: "1".into(),
        boolean_off: "0".into(),
        hex: |arg| format!("X'{arg}'"),
        hex_reverse: Regex::new(r"^X'(.*)'$").expect("hex"),
        quote: None,
        quote_reverse: None,
    })
});

const KEYWORDS: &[&str] = &[
    "AFTER", "ALTER", "AND", "AS", "AT", "AUTOINCREMENT", "BEFORE", "BEGIN", "BETWEEN", "BLOB",
    "BOOLEAN", "BY", "CASCADE", "CAST", "CONSTRAINT", "CREATE", "DECIMAL", "DEFAULT", "DELETE",
    "DROP", "EACH", "ELSE", "END", "EXISTS", "FALSE", "FLOAT", "FOR", "FOREIGN", "FROM",
    "GENERATED", "IF", "IN", "INDEX", "INNER", "INSERT", "INT", "INTEGER", "INTO", "JOIN", "KEY",
    "NOT", "NULL", "NUMERIC", "ON", "OR", "ORDER", "PRIMARY", "REAL", "REFERENCES", "RETURN",
    "RETURNING", "ROW", "SELECT", "SET", "TABLE", "TEXT", "THEN", "TRIGGER", "TRUE", "UNIQUE",
    "UPDATE", "VALUES", "VARCHAR", "WHEN", "WHERE",
];

static TOKEN: LazyLock<Regex> = LazyLock::new(|| {
    Regex::new(r#"('(?:''|[^'])*')|("(?:""|[^"])*")|(--[^\n]*)|(/\*[\s\S]*?\*/)|(?P<word>[A-Za-z_][A-Za-z0-9_]*)"#)
        .expect("token")
});

pub fn apply_keywords(sql: &str, keyword: &dyn Fn(&str) -> String) -> String {
    TOKEN
        .replace_all(sql, |caps: &regex::Captures| {
            if let Some(ident) = caps.name("word") {
                if KEYWORDS.iter().any(|word| word.eq_ignore_ascii_case(ident.as_str())) {
                    return keyword(ident.as_str());
                }
            }
            caps.get(0).map(|m| m.as_str().to_string()).unwrap_or_default()
        })
        .into_owned()
}