surrealdb-sql 1.1.0

Full type definitions for the SurrealQL query language
Documentation
use super::super::{
	comment::shouldbespace,
	part::{data, output, timeout},
	value::whats,
	IResult,
};
use crate::statements::CreateStatement;
use nom::{
	bytes::complete::tag_no_case,
	combinator::{cut, opt},
	sequence::preceded,
};

pub fn create(i: &str) -> IResult<&str, CreateStatement> {
	let (i, _) = tag_no_case("CREATE")(i)?;
	let (i, only) = opt(preceded(shouldbespace, tag_no_case("ONLY")))(i)?;
	let (i, _) = shouldbespace(i)?;
	let (i, what) = whats(i)?;
	let (i, (data, output, timeout, parallel)) = cut(|i| {
		let (i, data) = opt(preceded(shouldbespace, data))(i)?;
		let (i, output) = opt(preceded(shouldbespace, output))(i)?;
		let (i, timeout) = opt(preceded(shouldbespace, timeout))(i)?;
		let (i, parallel) = opt(preceded(shouldbespace, tag_no_case("PARALLEL")))(i)?;
		Ok((i, (data, output, timeout, parallel)))
	})(i)?;
	Ok((
		i,
		CreateStatement {
			only: only.is_some(),
			what,
			data,
			output,
			timeout,
			parallel: parallel.is_some(),
		},
	))
}

#[cfg(test)]
mod tests {

	use super::*;

	#[test]
	fn create_statement() {
		let sql = "CREATE test";
		let res = create(sql);
		let out = res.unwrap().1;
		assert_eq!("CREATE test", format!("{}", out))
	}
}