gluesql_core/
ast.rs

1mod ast_literal;
2mod data_type;
3mod ddl;
4mod expr;
5mod function;
6mod operator;
7mod query;
8
9pub use {
10    ast_literal::{AstLiteral, DateTimeField, TrimWhereField},
11    data_type::DataType,
12    ddl::*,
13    expr::Expr,
14    function::{Aggregate, AggregateFunction, CountArgExpr, Function},
15    operator::*,
16    query::*,
17};
18
19use {
20    serde::{Deserialize, Serialize},
21    strum_macros::Display,
22};
23
24pub trait ToSql {
25    fn to_sql(&self) -> String;
26}
27
28pub trait ToSqlUnquoted {
29    fn to_sql_unquoted(&self) -> String;
30}
31
32#[derive(PartialEq, Debug, Clone, Eq, Hash, Serialize, Deserialize)]
33pub struct ForeignKey {
34    pub name: String,
35    pub referencing_column_name: String,
36    pub referenced_table_name: String,
37    pub referenced_column_name: String,
38    pub on_delete: ReferentialAction,
39    pub on_update: ReferentialAction,
40}
41
42#[derive(PartialEq, Debug, Clone, Eq, Hash, Serialize, Deserialize, Display)]
43pub enum ReferentialAction {
44    #[strum(to_string = "NO ACTION")]
45    NoAction,
46}
47
48#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
49pub enum Statement {
50    ShowColumns {
51        table_name: String,
52    },
53    /// SELECT, VALUES
54    Query(Query),
55    /// INSERT
56    Insert {
57        /// TABLE
58        table_name: String,
59        /// COLUMNS
60        columns: Vec<String>,
61        /// A SQL query that specifies what to insert
62        source: Query,
63    },
64    /// UPDATE
65    Update {
66        /// TABLE
67        table_name: String,
68        /// Column assignments
69        assignments: Vec<Assignment>,
70        /// WHERE
71        selection: Option<Expr>,
72    },
73    /// DELETE
74    Delete {
75        /// FROM
76        table_name: String,
77        /// WHERE
78        selection: Option<Expr>,
79    },
80    /// CREATE TABLE
81    CreateTable {
82        if_not_exists: bool,
83        /// Table name
84        name: String,
85        /// Optional schema
86        columns: Option<Vec<ColumnDef>>,
87        source: Option<Box<Query>>,
88        engine: Option<String>,
89        foreign_keys: Vec<ForeignKey>,
90        comment: Option<String>,
91    },
92    /// CREATE FUNCTION
93    CreateFunction {
94        or_replace: bool,
95        name: String,
96        /// Optional schema
97        args: Vec<OperateFunctionArg>,
98        return_: Expr,
99    },
100    /// ALTER TABLE
101    AlterTable {
102        /// Table name
103        name: String,
104        operation: AlterTableOperation,
105    },
106    /// DROP TABLE
107    DropTable {
108        /// An optional `IF EXISTS` clause. (Non-standard.)
109        if_exists: bool,
110        /// One or more objects to drop. (ANSI SQL requires exactly one.)
111        names: Vec<String>,
112        /// An optional `CASCADE` clause for dropping dependent constructs.
113        cascade: bool,
114    },
115    /// DROP FUNCTION
116    DropFunction {
117        /// An optional `IF EXISTS` clause. (Non-standard.)
118        if_exists: bool,
119        /// One or more objects to drop. (ANSI SQL requires exactly one.)
120        names: Vec<String>,
121    },
122    /// CREATE INDEX
123    CreateIndex {
124        name: String,
125        table_name: String,
126        column: OrderByExpr,
127    },
128    /// DROP INDEX
129    DropIndex {
130        name: String,
131        table_name: String,
132    },
133    /// START TRANSACTION, BEGIN
134    StartTransaction,
135    /// COMMIT
136    Commit,
137    /// ROLLBACK
138    Rollback,
139    /// SHOW VARIABLE
140    ShowVariable(Variable),
141    ShowIndexes(String),
142}
143
144#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
145pub struct Assignment {
146    pub id: String,
147    pub value: Expr,
148}
149
150#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
151pub enum Variable {
152    Tables,
153    Functions,
154    Version,
155}
156
157impl ToSql for ForeignKey {
158    fn to_sql(&self) -> String {
159        let ForeignKey {
160            referencing_column_name,
161            referenced_table_name,
162            referenced_column_name,
163            name,
164            on_delete,
165            on_update,
166        } = self;
167
168        format!(
169            r#"CONSTRAINT "{name}" FOREIGN KEY ("{referencing_column_name}") REFERENCES "{referenced_table_name}" ("{referenced_column_name}") ON DELETE {on_delete} ON UPDATE {on_update}"#
170        )
171    }
172}
173
174#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
175pub struct Array {
176    pub elem: Vec<Expr>,
177    pub named: bool,
178}
gluesql_core/ast.rs

gluesql_core/
ast.rs
