Function parse_query_form

nodedb-sql/src/ddl_ast/parse/copy_to.rs:94–130

`COPY (SELECT ...) TO ' ' [WITH (...)]`

(trimmed: &str)

/// `COPY (SELECT ...) TO '<path>' [WITH (...)]`
fn parse_query_form(trimmed: &str) -> Result<NodedbStatement, SqlError> {
    // Find the matching close-paren for the leading `(`.
    let after_copy = trimmed["COPY ".len()..].trim_start();
    let close = find_matching_paren(after_copy).ok_or_else(|| SqlError::Parse {
        detail: "COPY: unclosed parenthesis in query form".to_string(),
    })?;

    let query = after_copy[1..close].trim().to_string();
    if query.is_empty() {
        return Err(SqlError::Parse {
            detail: "COPY: empty query in query form".to_string(),
        });
    }

    // After the closing paren, expect " TO '<path>'"
    let after_paren = after_copy[close + 1..].trim_start();
    let upper_after = after_paren.to_uppercase();
    if !upper_after.starts_with("TO ") {
        return Err(SqlError::Parse {
            detail: format!(
                "COPY: expected TO after query, got: {}",
                &after_paren[..after_paren.len().min(32)]
            ),
        });
    }

    let after_to = after_paren["TO ".len()..].trim_start();
    let (path, _rest, format, delimiter, header) = parse_path_and_opts(after_to)?;

    Ok(NodedbStatement::Misc(MiscStmt::CopyToFile {
        source: CopyToSource::Query(query),
        path,
        format,
        delimiter,
        header,
    }))
}

type PathAndOpts<'a> = (String, &'a str, Option<CopyFormat>, Option<char>, bool);