surrealpatch/lib/src/sql/comment.rs

use crate::sql::error::IResult;
use nom::branch::alt;
use nom::bytes::complete::take_until;
use nom::character::complete::char;
use nom::character::complete::multispace0;
use nom::character::complete::multispace1;
use nom::character::complete::not_line_ending;
use nom::multi::many0;
use nom::multi::many1;

pub fn mightbespace(i: &str) -> IResult<&str, ()> {
	let (i, _) = many0(alt((comment, space)))(i)?;
	Ok((i, ()))
}

pub fn shouldbespace(i: &str) -> IResult<&str, ()> {
	let (i, _) = many1(alt((comment, space)))(i)?;
	Ok((i, ()))
}

pub fn comment(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace0(i)?;
	let (i, _) = many1(alt((block, slash, dash, hash)))(i)?;
	let (i, _) = multispace0(i)?;
	Ok((i, ()))
}

pub fn block(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace0(i)?;
	let (i, _) = char('/')(i)?;
	let (i, _) = char('*')(i)?;
	let (i, _) = take_until("*/")(i)?;
	let (i, _) = char('*')(i)?;
	let (i, _) = char('/')(i)?;
	let (i, _) = multispace0(i)?;
	Ok((i, ()))
}

pub fn slash(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace0(i)?;
	let (i, _) = char('/')(i)?;
	let (i, _) = char('/')(i)?;
	let (i, _) = not_line_ending(i)?;
	Ok((i, ()))
}

pub fn dash(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace0(i)?;
	let (i, _) = char('-')(i)?;
	let (i, _) = char('-')(i)?;
	let (i, _) = not_line_ending(i)?;
	Ok((i, ()))
}

pub fn hash(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace0(i)?;
	let (i, _) = char('#')(i)?;
	let (i, _) = not_line_ending(i)?;
	Ok((i, ()))
}

fn space(i: &str) -> IResult<&str, ()> {
	let (i, _) = multispace1(i)?;
	Ok((i, ()))
}

#[cfg(test)]
mod test {
	use crate::sql::parse;

	#[test]
	fn any_whitespace() {
		let sql = "USE /* white space and comment between */ NS test;";
		assert!(parse(sql).is_ok());
	}
}