From 2f851127cdd034245cac5f62cec8eb0584a427f4 Mon Sep 17 00:00:00 2001
From: TcMits <eddiebrock2001@gmail.com>
Date: Fri, 8 Aug 2025 15:24:05 +0700
Subject: [PATCH] finish PRAGMA

---
 parser/src/ast.rs    |  2 +-
 parser/src/parser.rs | 95 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 96 insertions(+), 1 deletion(-)

diff --git a/parser/src/ast.rs b/parser/src/ast.rs
index 1f1bcb7ae..0b3c8e2f5 100644
--- a/parser/src/ast.rs
+++ b/parser/src/ast.rs
@@ -1075,7 +1075,7 @@ pub enum PragmaBody {
 
 /// `PRAGMA` value
 // https://sqlite.org/syntax/pragma-value.html
-pub type PragmaValue = Expr; // TODO
+pub type PragmaValue = Box<Expr>; // TODO
 
 /// `PRAGMA` value
 // https://sqlite.org/pragma.html
diff --git a/parser/src/parser.rs b/parser/src/parser.rs
index eda696337..bdcb1cd7a 100644
--- a/parser/src/parser.rs
+++ b/parser/src/parser.rs
@@ -5,6 +5,7 @@ use crate::ast::{
     LikeOperator, Limit, Literal, Materialized, Name, NullsOrder, OneSelect, Operator, Over,
     QualifiedName, ResolveType, ResultColumn, Select, SelectBody, SelectTable, SortOrder,
     SortedColumn, Stmt, TransactionType, Type, TypeSize, UnaryOperator, Window, WindowDef, With,
+    PragmaBody, PragmaValue
 };
 use crate::error::Error;
 use crate::lexer::{Lexer, Token};
@@ -498,6 +499,7 @@ impl<'a> Parser<'a> {
             TokenType::TK_ANALYZE,
             TokenType::TK_ATTACH,
             TokenType::TK_DETACH,
+            TokenType::TK_PRAGMA,
             // add more
         ])?;
 
@@ -513,6 +515,7 @@ impl<'a> Parser<'a> {
             TokenType::TK_ANALYZE => self.parse_analyze(),
             TokenType::TK_ATTACH => self.parse_attach(),
             TokenType::TK_DETACH => self.parse_detach(),
+            TokenType::TK_PRAGMA => self.parse_pragma(),
             _ => unreachable!(),
         }
     }
@@ -2561,6 +2564,55 @@ impl<'a> Parser<'a> {
 
         Ok(Stmt::Detach { name: self.parse_expr(0)? })
     }
+
+    fn parse_pragma_value(&mut self) -> Result<PragmaValue, Error> {
+        match self.peek_no_eof()?.token_type.unwrap().fallback_id_if_ok() {
+            TokenType::TK_ON | TokenType::TK_DELETE | TokenType::TK_DEFAULT => {
+                let tok = self.eat_assert(&[TokenType::TK_ON, TokenType::TK_DELETE, TokenType::TK_DEFAULT]);
+                Ok(Box::new(Expr::Literal(Literal::Keyword(from_bytes(tok.value)))))
+            }
+            TokenType::TK_ID |
+            TokenType::TK_STRING |
+            TokenType::TK_INDEXED |
+            TokenType::TK_JOIN_KW  => { Ok(Box::new(Expr::Name(self.parse_nm()))) }
+            _ => {
+                self.parse_signed()
+            }
+        }
+    }
+
+    fn parse_pragma(&mut self) -> Result<Stmt, Error> {
+        self.eat_assert(&[TokenType::TK_PRAGMA]);
+        let name = self.parse_fullname(false)?;
+        match self.peek()? {
+            Some(tok) => match tok.token_type.unwrap() {
+                TokenType::TK_EQ => {
+                    self.eat_assert(&[TokenType::TK_EQ]);
+                    Ok(Stmt::Pragma {
+                        name: name,
+                        body: Some(PragmaBody::Equals(self.parse_pragma_value()?)),
+                    })
+                },
+                TokenType::TK_LP => {
+                    self.eat_assert(&[TokenType::TK_LP]);
+                    let value = self.parse_pragma_value()?;
+                    self.eat_expect(&[TokenType::TK_RP])?;
+                    Ok(Stmt::Pragma {
+                        name: name,
+                        body: Some(PragmaBody::Call(value)),
+                    })
+                },
+                _ => Ok(Stmt::Pragma {
+                    name: name,
+                    body: None,
+                }),
+            },
+            _ => Ok(Stmt::Pragma {
+                name: name,
+                body: None,
+            }),
+        }
+    }
 }
 
 #[cfg(test)]
@@ -7304,6 +7356,49 @@ mod tests {
                     name: Box::new(Expr::Id(Name::Ident("bar".to_owned()))),
                 })],
             ),
+            // parse pragma
+            (
+                b"PRAGMA foreign_keys = ON".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: Some(PragmaBody::Equals(Box::new(Expr::Literal(Literal::Keyword("ON".to_owned()))))),
+                })],
+            ),
+            (
+                b"PRAGMA foreign_keys = DELETE".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: Some(PragmaBody::Equals(Box::new(Expr::Literal(Literal::Keyword("DELETE".to_owned()))))),
+                })],
+            ),
+            (
+                b"PRAGMA foreign_keys = DEFAULT".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: Some(PragmaBody::Equals(Box::new(Expr::Literal(Literal::Keyword("DEFAULT".to_owned()))))),
+                })],
+            ),
+            (
+                b"PRAGMA foreign_keys".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: None,
+                })],
+            ),
+            (
+                b"PRAGMA foreign_keys = 1".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: Some(PragmaBody::Equals(Box::new(Expr::Literal(Literal::Numeric("1".to_owned()))))),
+                })],
+            ),
+            (
+                b"PRAGMA foreign_keys = test".as_slice(),
+                vec![Cmd::Stmt(Stmt::Pragma {
+                    name: QualifiedName { db_name: None, name: Name::Ident("foreign_keys".to_owned()),  alias: None },
+                    body: Some(PragmaBody::Equals(Box::new(Expr::Name(Name::Ident("test".to_owned()))))),
+                })],
+            ),
         ];
 
         for (input, expected) in test_cases {