nodedb-sql 0.0.6

SQL parser, planner, and optimizer for NodeDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

//! Parse CREATE/DROP/ALTER/DESCRIBE/SHOW for COLLECTION (and TABLE alias).
//!
//! `DROP COLLECTION` extensions (sqlparser 0.61 does not tokenize
//! these, hence custom-handled upper-case keyword scan):
//! - `PURGE` — hard-delete, skipping the retention window
//! - `CASCADE` — recursively drop dependents
//! - `CASCADE FORCE` — cascade through dynamic-SQL schedules
//!
//! `UNDROP COLLECTION <name>` restores a soft-deleted record.

use super::helpers::{extract_name_after_if_exists, extract_name_after_keyword};
use crate::ddl_ast::statement::NodedbStatement;

pub(super) fn try_parse(upper: &str, parts: &[&str], trimmed: &str) -> Option<NodedbStatement> {
    if upper.starts_with("CREATE COLLECTION ") || upper.starts_with("CREATE TABLE ") {
        let if_not_exists = upper.contains("IF NOT EXISTS");
        let name = extract_name_after_keyword(parts, "COLLECTION")
            .or_else(|| extract_name_after_keyword(parts, "TABLE"))?;
        return Some(NodedbStatement::CreateCollection {
            name,
            if_not_exists,
            raw_sql: trimmed.to_string(),
        });
    }
    if upper.starts_with("UNDROP COLLECTION ") || upper.starts_with("UNDROP TABLE ") {
        let name = extract_name_after_keyword(parts, "COLLECTION")
            .or_else(|| extract_name_after_keyword(parts, "TABLE"))?;
        return Some(NodedbStatement::UndropCollection { name });
    }
    if upper.starts_with("DROP COLLECTION ") || upper.starts_with("DROP TABLE ") {
        let if_exists = upper.contains("IF EXISTS");
        let name = extract_name_after_if_exists(parts, "COLLECTION")
            .or_else(|| extract_name_after_if_exists(parts, "TABLE"))?;
        let purge = upper.contains(" PURGE");
        let cascade = upper.contains(" CASCADE");
        // `CASCADE FORCE` implies `cascade`; recognize both orderings.
        let cascade_force = upper.contains(" CASCADE FORCE") || upper.contains(" FORCE CASCADE");
        return Some(NodedbStatement::DropCollection {
            name,
            if_exists,
            purge,
            cascade: cascade || cascade_force,
            cascade_force,
        });
    }
    if upper.starts_with("ALTER COLLECTION ") || upper.starts_with("ALTER TABLE ") {
        let name = extract_name_after_keyword(parts, "COLLECTION")
            .or_else(|| extract_name_after_keyword(parts, "TABLE"))?;
        return Some(NodedbStatement::AlterCollection {
            name,
            raw_sql: trimmed.to_string(),
        });
    }
    if upper.starts_with("DESCRIBE ") && !upper.starts_with("DESCRIBE SEQUENCE") {
        let name = parts.get(1)?.to_string();
        return Some(NodedbStatement::DescribeCollection { name });
    }
    if upper == "\\D" || upper == "SHOW COLLECTIONS" || upper.starts_with("SHOW COLLECTIONS") {
        return Some(NodedbStatement::ShowCollections);
    }
    None
}