sql_parser: create trigger statement parser implemented

Author: Almaz-KG

src/ast/create.rs

@@ -1,8 +1,10 @@
 use std::fmt::Display;
 
+use crate::ParsingError;
+
 use super::{
     ColumnDefinition, ConflictClause, DeleteStatement, Expression, ForeignKeyClause, Identifier,
-    IndexedColumn, InsertStatement, SelectStatement, UpdateStatement,
+    IndexedColumn, InsertStatement, SelectStatement, Statement, UpdateStatement,
 };
 
 /// An AST for [CREATE VIRTUAL TABLE](https://www.sqlite.org/lang_createvtab.html) SQL statement.
@@ -110,7 +112,7 @@ impl Display for TableOption {
 }
 
 /// An AST for [CREATE TRIGGER](https://www.sqlite.org/lang_createtrigger.html) SQL statement.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub struct CreateTriggerStatement {
     pub temporary: bool,
 
@@ -129,7 +131,7 @@ pub struct CreateTriggerStatement {
     pub trigger_statements: Vec<TriggerStatement>,
 }
 
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub enum TriggerPreCondition {
     Before,
 
@@ -138,21 +140,31 @@ pub enum TriggerPreCondition {
     InsteadOf,
 }
 
-#[derive(Debug, PartialEq)]
+impl Display for TriggerPreCondition {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            TriggerPreCondition::Before => write!(f, "BEFORE"),
+            TriggerPreCondition::After => write!(f, "AFTER"),
+            TriggerPreCondition::InsteadOf => write!(f, "INSTEAD OF"),
+        }
+    }
+}
+
+#[derive(Debug, PartialEq, Clone)]
 pub struct TriggerEvent {
     pub event_type: TriggerEventType,
 
     pub table_name: Identifier,
 }
 
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub enum TriggerEventType {
     Delete,
     Insert,
     Update(Option<Vec<Identifier>>),
 }
 
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub enum TriggerStatement {
     Update(UpdateStatement),
 
@@ -162,3 +174,20 @@ pub enum TriggerStatement {
 
     Select(SelectStatement),
 }
+
+impl TryFrom<Statement> for TriggerStatement {
+    type Error = ParsingError;
+
+    fn try_from(statement: Statement) -> Result<Self, Self::Error> {
+        match statement {
+            Statement::Update(update_statement) => Ok(TriggerStatement::Update(update_statement)),
+            Statement::Insert(insert_statement) => Ok(TriggerStatement::Insert(insert_statement)),
+            Statement::Delete(delete_statement) => Ok(TriggerStatement::Delete(delete_statement)),
+            Statement::Select(select_statement) => Ok(TriggerStatement::Select(select_statement)),
+            _ => Err(ParsingError::UnexpectedParsingState(format!(
+                "Expected UPDATE, INSERT, DELETE or SELECT statement, got: {:?}",
+                statement
+            ))),
+        }
+    }
+}

src/ast/insert.rs

@@ -4,7 +4,7 @@ use super::{
 };
 
 /// An AST for [INSERT](https://www.sqlite.org/lang_insert.html) SQL statement.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub struct InsertStatement {
     pub conflict_clause: ConflictClause,
 
@@ -22,7 +22,7 @@ pub struct InsertStatement {
 }
 
 /// The values to insert into the table.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub enum InsertValues {
     /// The values to insert into the table.
     Values(Vec<Vec<Expression>>),
@@ -35,15 +35,15 @@ pub enum InsertValues {
 }
 
 /// The upsert clause to update the table with the new values.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub struct UpsertClause {
     pub conflict_target: Option<UpsertConflictTarget>,
     pub action: UpsertAction,
 }
 
 /// The [conflict target](https://www.sqlite.org/lang_upsert.html#upsert_conflict_target)
 /// to update the table with the new values.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub struct UpsertConflictTarget {
     pub columns: Vec<IndexedColumn>,
     pub where_clause: Option<Box<Expression>>,
@@ -58,7 +58,7 @@ pub struct IndexedColumn {
 }
 
 /// The [action](https://www.sqlite.org/lang_upsert.html#upsert_action) to take when a conflict occurs.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub enum UpsertAction {
     /// The action to take when a conflict occurs.
     Nothing,
@@ -68,7 +68,7 @@ pub enum UpsertAction {
 }
 
 /// The [update set](https://www.sqlite.org/lang_upsert.html) action to take when a conflict occurs.
-#[derive(Debug, PartialEq)]
+#[derive(Debug, PartialEq, Clone)]
 pub struct UpsertUpdate {
     pub set_clauses: Vec<SetClause>,
     pub where_clause: Option<Box<Expression>>,

src/parser/create/create_table.rs

@@ -152,7 +152,7 @@ impl<'a> CreateTableStatementParser for Parser<'a> {
             Keyword::Foreign => {
                 self.consume_as_keyword(Keyword::Foreign)?;
                 self.consume_as_keyword(Keyword::Key)?;
-                let columns = self.parse_columns()?;
+                let columns = self.parse_columns_names()?;
                 let foreign_key_clause = self.parse_foreign_key_clause()?;
                 Ok(TableConstraintType::ForeignKey(columns, foreign_key_clause))
             }