lora_parser/parser/
mod.rs1use crate::errors::ParseError;
24use lora_ast::*;
25use pest::iterators::Pair;
26use pest::Parser;
27
28mod clauses;
29mod expressions;
30mod limits;
31mod literals;
32mod patterns;
33mod query;
34mod schema;
35mod util;
36
37#[cfg(test)]
38mod tests;
39
40use clauses::lower_standalone_call;
41use query::lower_regular_query;
42use schema::lower_schema_command;
43use util::{pair_span, unexpected_rule};
44
45#[derive(pest_derive::Parser)]
46#[grammar = "cypher.pest"]
47pub(super) struct LoraParser;
48
49pub fn parse_query(input: &str) -> Result<Document, ParseError> {
50 limits::reset_budgets();
51 limits::check_nesting(input)?;
52 let mut pairs = LoraParser::parse(Rule::query, input)
53 .map_err(|e| ParseError::new(e.to_string(), 0, input.len()))?;
54
55 let pair = pairs
56 .next()
57 .ok_or_else(|| ParseError::new("expected query", 0, input.len()))?;
58
59 lower_query(pair)
60}
61
62fn lower_query(pair: Pair<Rule>) -> Result<Document, ParseError> {
63 let span = pair_span(&pair);
64 let mut inner = pair.into_inner();
65
66 let stmt = inner
67 .find(|p| p.as_rule() == Rule::statement)
68 .ok_or_else(|| ParseError::new("expected statement", span.start, span.end))?;
69
70 Ok(Document {
71 statement: lower_statement(stmt)?,
72 span,
73 })
74}
75
76fn lower_statement(pair: Pair<Rule>) -> Result<Statement, ParseError> {
77 let inner = util::single_inner(pair)?;
78 match inner.as_rule() {
79 Rule::regular_query => Ok(Statement::Query(Query::Regular(lower_regular_query(
80 inner,
81 )?))),
82 Rule::standalone_call => Ok(Statement::Query(Query::StandaloneCall(
83 lower_standalone_call(inner)?,
84 ))),
85 Rule::schema_command => Ok(Statement::Schema(lower_schema_command(inner)?)),
86 _ => Err(unexpected_rule("statement", inner)),
87 }
88}