Function try_parse

nodedb-sql/src/ddl_ast/parse/copy_to.rs:17–52

Try to parse a COPY TO statement. Returns `None` if the SQL is not a COPY statement. Returns `Some(Err)` on parse errors. Returns `Some(Ok(CopyToFile {...}))` on success.

(upper: &str, trimmed: &str)

/// Returns `None` if the SQL is not a COPY statement. Returns `Some(Err)` on
/// parse errors. Returns `Some(Ok(CopyToFile {...}))` on success.
pub(super) fn try_parse(upper: &str, trimmed: &str) -> Option<Result<NodedbStatement, SqlError>> {
    if !upper.starts_with("COPY ") {
        return None;
    }

    let after_copy = trimmed["COPY ".len()..].trim_start();
    let upper_after = upper["COPY ".len()..].trim_start();

    // Query form: `COPY (SELECT ...) TO '<path>'` — the `(` starts immediately.
    // Skip FROM/TO prefix checks; parse_copy_to will handle it.
    if after_copy.starts_with('(') {
        // Only claim it if there is a `) TO ` somewhere after the closing paren.
        // We do a quick scan: find the closing paren, then check for TO.
        if find_matching_paren(after_copy).is_some_and(|close| {
            let rest = after_copy[close + 1..].trim_start().to_uppercase();
            rest.starts_with("TO ")
        }) {
            return Some(parse_copy_to(trimmed, after_copy));
        }
        return None;
    }

    // Table form: `COPY <collection> TO '<path>'`.
    // Only handle the TO form; FROM form is handled in copy_from.
    // For the table form, look at the upper-case tokens before the first quoted string.
    let has_to = upper_after.contains(" TO ");
    let has_from = upper_after.contains(" FROM ");
    if has_from && (!has_to || upper_after.find(" FROM ") < upper_after.find(" TO ")) {
        return None;
    }
    if !has_to {
        return None;
    }

    Some(parse_copy_to(trimmed, after_copy))
}

fn parse_copy_to(trimmed: &str, after_copy: &str) -> Result<NodedbStatement, SqlError> {
    // Determine if this is the query form or the table form.