Function try_parse_array_statement

nodedb-sql/src/parser/array_stmt/parse.rs:20–66

Top-level entry. Returns `Ok(None)` if the SQL doesn't start with an array statement keyword sequence.

(sql: &str)

/// Top-level entry. Returns `Ok(None)` if the SQL doesn't start with an
/// array statement keyword sequence.
pub fn try_parse_array_statement(sql: &str) -> Result<Option<ArrayStatement>> {
    let trimmed = sql.trim_start();
    let upper: String = trimmed
        .chars()
        .take(40)
        .collect::<String>()
        .to_ascii_uppercase();

    if upper.starts_with("CREATE ARRAY ") || upper == "CREATE ARRAY" {
        let toks = tokenize(trimmed)?;
        let mut p = Parser::new(&toks);
        p.expect_kw("CREATE")?;
        p.expect_kw("ARRAY")?;
        return Ok(Some(ArrayStatement::Create(p.parse_create()?)));
    }
    if upper.starts_with("DROP ARRAY ") || upper == "DROP ARRAY" {
        let toks = tokenize(trimmed)?;
        let mut p = Parser::new(&toks);
        p.expect_kw("DROP")?;
        p.expect_kw("ARRAY")?;
        return Ok(Some(ArrayStatement::Drop(p.parse_drop()?)));
    }
    if upper.starts_with("INSERT INTO ARRAY ") {
        let toks = tokenize(trimmed)?;
        let mut p = Parser::new(&toks);
        p.expect_kw("INSERT")?;
        p.expect_kw("INTO")?;
        p.expect_kw("ARRAY")?;
        return Ok(Some(ArrayStatement::Insert(p.parse_insert()?)));
    }
    if upper.starts_with("DELETE FROM ARRAY ") {
        let toks = tokenize(trimmed)?;
        let mut p = Parser::new(&toks);
        p.expect_kw("DELETE")?;
        p.expect_kw("FROM")?;
        p.expect_kw("ARRAY")?;
        return Ok(Some(ArrayStatement::Delete(p.parse_delete()?)));
    }
    if upper.starts_with("ALTER ARRAY ") || upper == "ALTER ARRAY" {
        let toks = tokenize(trimmed)?;
        let mut p = Parser::new(&toks);
        p.expect_kw("ALTER")?;
        p.expect_kw("ARRAY")?;
        return Ok(Some(ArrayStatement::Alter(p.parse_alter()?)));
    }
    Ok(None)
}

#[cfg(test)]
mod tests {