119 lines
2.7 KiB
Rust
119 lines
2.7 KiB
Rust
use crate::sql::common::val_char;
|
|
use crate::sql::error::IResult;
|
|
use crate::sql::escape::escape_ident;
|
|
use nom::branch::alt;
|
|
use nom::bytes::complete::escaped;
|
|
use nom::bytes::complete::is_not;
|
|
use nom::bytes::complete::tag;
|
|
use nom::bytes::complete::take_while1;
|
|
use nom::character::complete::char;
|
|
use nom::character::complete::one_of;
|
|
use nom::sequence::delimited;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt;
|
|
use std::ops::Deref;
|
|
use std::str;
|
|
|
|
const BRACKET_L: char = '⟨';
|
|
const BRACKET_R: char = '⟩';
|
|
const BRACKET_END: &str = r#"⟩"#;
|
|
|
|
const BACKTICK: &str = r#"`"#;
|
|
const BACKTICK_ESC: &str = r#"\`"#;
|
|
|
|
#[derive(Clone, Debug, Default, Eq, Ord, PartialEq, PartialOrd, Serialize, Deserialize)]
|
|
pub struct Ident(pub String);
|
|
|
|
impl From<String> for Ident {
|
|
fn from(s: String) -> Self {
|
|
Ident(s)
|
|
}
|
|
}
|
|
|
|
impl From<&str> for Ident {
|
|
fn from(i: &str) -> Ident {
|
|
Ident(String::from(i))
|
|
}
|
|
}
|
|
|
|
impl From<&String> for Ident {
|
|
fn from(i: &String) -> Ident {
|
|
Ident(String::from(i))
|
|
}
|
|
}
|
|
|
|
impl Deref for Ident {
|
|
type Target = String;
|
|
fn deref(&self) -> &Self::Target {
|
|
&self.0
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Ident {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "{}", escape_ident(&self.0))
|
|
}
|
|
}
|
|
|
|
pub fn ident(i: &str) -> IResult<&str, Ident> {
|
|
let (i, v) = ident_raw(i)?;
|
|
Ok((i, Ident::from(v)))
|
|
}
|
|
|
|
pub fn ident_raw(i: &str) -> IResult<&str, String> {
|
|
let (i, v) = alt((ident_default, ident_backtick, ident_brackets))(i)?;
|
|
Ok((i, v))
|
|
}
|
|
|
|
fn ident_default(i: &str) -> IResult<&str, String> {
|
|
let (i, v) = take_while1(val_char)(i)?;
|
|
Ok((i, String::from(v)))
|
|
}
|
|
|
|
fn ident_backtick(i: &str) -> IResult<&str, String> {
|
|
let (i, _) = char('`')(i)?;
|
|
let (i, v) = alt((escaped(is_not(BACKTICK_ESC), '\\', one_of(BACKTICK)), tag("")))(i)?;
|
|
let (i, _) = char('`')(i)?;
|
|
Ok((i, String::from(v).replace(BACKTICK_ESC, BACKTICK)))
|
|
}
|
|
|
|
fn ident_brackets(i: &str) -> IResult<&str, String> {
|
|
let (i, v) = delimited(char(BRACKET_L), is_not(BRACKET_END), char(BRACKET_R))(i)?;
|
|
Ok((i, String::from(v)))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn ident_normal() {
|
|
let sql = "test";
|
|
let res = ident(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!("test", format!("{}", out));
|
|
assert_eq!(out, Ident::from("test"));
|
|
}
|
|
|
|
#[test]
|
|
fn ident_quoted_backtick() {
|
|
let sql = "`test`";
|
|
let res = ident(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!("test", format!("{}", out));
|
|
assert_eq!(out, Ident::from("test"));
|
|
}
|
|
|
|
#[test]
|
|
fn ident_quoted_brackets() {
|
|
let sql = "⟨test⟩";
|
|
let res = ident(sql);
|
|
assert!(res.is_ok());
|
|
let out = res.unwrap().1;
|
|
assert_eq!("test", format!("{}", out));
|
|
assert_eq!(out, Ident::from("test"));
|
|
}
|
|
}
|