84 lines
2 KiB
Rust
84 lines
2 KiB
Rust
use crate::sql::comment::mightbespace;
|
|
use crate::sql::error::Error::ParserError;
|
|
use crate::sql::error::IResult;
|
|
use nom::bytes::complete::take_while;
|
|
use nom::bytes::complete::take_while_m_n;
|
|
use nom::character::complete::char;
|
|
use nom::character::is_alphanumeric;
|
|
use nom::multi::many1;
|
|
use nom::Err::Error;
|
|
use std::ops::RangeBounds;
|
|
|
|
pub fn colons(i: &str) -> IResult<&str, ()> {
|
|
let (i, _) = mightbespace(i)?;
|
|
let (i, _) = many1(char(';'))(i)?;
|
|
let (i, _) = mightbespace(i)?;
|
|
Ok((i, ()))
|
|
}
|
|
|
|
pub fn commas(i: &str) -> IResult<&str, ()> {
|
|
let (i, _) = mightbespace(i)?;
|
|
let (i, _) = char(',')(i)?;
|
|
let (i, _) = mightbespace(i)?;
|
|
Ok((i, ()))
|
|
}
|
|
|
|
#[inline]
|
|
pub fn is_digit(chr: char) -> bool {
|
|
(0x30..=0x39).contains(&(chr as u8))
|
|
}
|
|
|
|
#[inline]
|
|
pub fn val_char(chr: char) -> bool {
|
|
is_alphanumeric(chr as u8) || chr == '_'
|
|
}
|
|
|
|
#[inline]
|
|
pub fn escape(s: &str, f: &dyn Fn(char) -> bool, c: &str) -> String {
|
|
for x in s.chars() {
|
|
if !f(x) {
|
|
return format!("{}{}{}", c, s, c);
|
|
}
|
|
}
|
|
s.to_owned()
|
|
}
|
|
|
|
pub fn take_u32(i: &str) -> IResult<&str, u32> {
|
|
let (i, v) = take_while(is_digit)(i)?;
|
|
match v.parse::<u32>() {
|
|
Ok(v) => Ok((i, v)),
|
|
_ => Err(Error(ParserError(i))),
|
|
}
|
|
}
|
|
|
|
pub fn take_u64(i: &str) -> IResult<&str, u64> {
|
|
let (i, v) = take_while(is_digit)(i)?;
|
|
match v.parse::<u64>() {
|
|
Ok(v) => Ok((i, v)),
|
|
_ => Err(Error(ParserError(i))),
|
|
}
|
|
}
|
|
|
|
pub fn take_usize(i: &str) -> IResult<&str, usize> {
|
|
let (i, v) = take_while(is_digit)(i)?;
|
|
match v.parse::<usize>() {
|
|
Ok(v) => Ok((i, v)),
|
|
_ => Err(Error(ParserError(i))),
|
|
}
|
|
}
|
|
|
|
pub fn take_digits(i: &str, n: usize) -> IResult<&str, u32> {
|
|
let (i, v) = take_while_m_n(n, n, is_digit)(i)?;
|
|
match v.parse::<u32>() {
|
|
Ok(v) => Ok((i, v)),
|
|
_ => Err(Error(ParserError(i))),
|
|
}
|
|
}
|
|
|
|
pub fn take_digits_range(i: &str, n: usize, range: impl RangeBounds<u32>) -> IResult<&str, u32> {
|
|
let (i, v) = take_while_m_n(n, n, is_digit)(i)?;
|
|
match v.parse::<u32>() {
|
|
Ok(v) if range.contains(&v) => Ok((i, v)),
|
|
_ => Err(Error(ParserError(i))),
|
|
}
|
|
}
|