geopackage-core 0.1.2

No-IO core for the OGC GeoPackage format: GeoPackage Binary (GPB) codec, table DDL, RTree trigger definitions, and validation rules
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

//! SQL identifier quoting for SQLite.

use crate::Error;

/// Quote an arbitrary identifier for safe interpolation into SQLite DDL/DML.
///
/// Uses standard SQL double-quoting, doubling embedded quotes. Rejects
/// identifiers containing NUL, which SQLite cannot represent.
pub fn quote(ident: &str) -> Result<String, Error> {
    if ident.is_empty() || ident.contains('\0') {
        return Err(Error::InvalidIdentifier(ident.to_owned()));
    }
    let mut out = String::with_capacity(ident.len() + 2);
    out.push('"');
    for c in ident.chars() {
        if c == '"' {
            out.push('"');
        }
        out.push(c);
    }
    out.push('"');
    Ok(out)
}

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

    #[test]
    fn quoting() {
        assert_eq!(quote("roads").unwrap(), "\"roads\"");
        assert_eq!(quote("we\"ird").unwrap(), "\"we\"\"ird\"");
        quote("").unwrap_err();
        quote("a\0b").unwrap_err();
    }
}