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

use connection_string::JdbcString;
use std::{borrow::Cow, collections::BTreeMap};

use crate::datamodel_connector::Flavour;

pub fn set_config_dir<'a>(flavour: Flavour, config_dir: &std::path::Path, url: &'a str) -> Cow<'a, str> {
    match flavour {
        Flavour::Sqlserver => set_config_dir_mssql(config_dir, url),
        Flavour::Sqlite => set_config_dir_sqlite(config_dir, url),
        _ => set_config_dir_default(config_dir, url),
    }
}

fn set_config_dir_default<'a>(config_dir: &std::path::Path, url: &'a str) -> Cow<'a, str> {
    let set_root = |path: &str| {
        let path = std::path::Path::new(path);

        if path.is_relative() {
            Some(config_dir.join(path).to_str().map(ToString::to_string).unwrap())
        } else {
            None
        }
    };

    let mut url = match url::Url::parse(url) {
        Ok(url) => url,
        Err(_) => return Cow::from(url), // bail
    };

    let mut params: BTreeMap<String, String> = url.query_pairs().map(|(k, v)| (k.to_string(), v.to_string())).collect();

    url.query_pairs_mut().clear();

    // Only for PostgreSQL + MySQL
    if let Some(path) = params.get("sslcert").map(|s| s.as_str()).and_then(set_root) {
        params.insert("sslcert".into(), path);
    }

    // Only for PostgreSQL + MySQL
    if let Some(path) = params.get("sslidentity").map(|s| s.as_str()).and_then(set_root) {
        params.insert("sslidentity".into(), path);
    }

    // Only for MongoDB
    if let Some(path) = params.get("tlsCAFile").map(|s| s.as_str()).and_then(set_root) {
        params.insert("tlsCAFile".into(), path);
    }

    for (k, v) in params.into_iter() {
        url.query_pairs_mut().append_pair(&k, &v);
    }

    url.to_string().into()
}

pub fn set_config_dir_mssql<'a>(config_dir: &std::path::Path, url: &'a str) -> Cow<'a, str> {
    let mut jdbc: JdbcString = match format!("jdbc:{url}").parse() {
        Ok(jdbc) => jdbc,
        _ => return Cow::from(url),
    };

    let set_root = |path: String| {
        let path = std::path::Path::new(&path);

        if path.is_relative() {
            Some(config_dir.join(path).to_str().map(ToString::to_string).unwrap())
        } else {
            Some(path.to_str().unwrap().to_string())
        }
    };

    let props = jdbc.properties_mut();

    let cert_path = props.remove("trustservercertificateca").and_then(set_root);

    if let Some(path) = cert_path {
        props.insert("trustServerCertificateCA".to_owned(), path);
    }

    let final_connection_string = format!("{jdbc}").replace("jdbc:sqlserver://", "sqlserver://");

    Cow::Owned(final_connection_string)
}

pub fn set_config_dir_sqlite<'a>(config_dir: &std::path::Path, url: &'a str) -> Cow<'a, str> {
    let set_root = |path: &str| {
        let path = std::path::Path::new(path);

        if path.is_relative() {
            Some(config_dir.join(path).to_str().map(ToString::to_string).unwrap())
        } else {
            None
        }
    };

    if let Some(path) = set_root(url.trim_start_matches("file:")) {
        return Cow::Owned(format!("file:{path}"));
    };

    Cow::Borrowed(url)
}