From 7c224eee2acaee28e453ac0fe6855916491511fe Mon Sep 17 00:00:00 2001
From: Andrew Harper <andrew.harper@veracross.com>
Date: Fri, 11 Apr 2025 17:22:51 -0400
Subject: [PATCH] Add support for `PRINT` statement for SQL Server

---
 src/ast/mod.rs           | 21 ++++++++++++++++++++-
 src/ast/spans.rs         |  1 +
 src/keywords.rs          |  1 +
 src/parser/mod.rs        |  8 ++++++++
 tests/sqlparser_mssql.rs | 17 +++++++++++++++++
 5 files changed, 47 insertions(+), 1 deletion(-)

diff --git a/src/ast/mod.rs b/src/ast/mod.rs
index 4031936e9..ce654dbc4 100644
--- a/src/ast/mod.rs
+++ b/src/ast/mod.rs
@@ -4054,6 +4054,12 @@ pub enum Statement {
         arguments: Vec<Expr>,
         options: Vec<RaisErrorOption>,
     },
+    /// ```sql
+    /// PRINT msg_str | @local_variable | string_expr
+    /// ```
+    ///
+    /// See: <https://learn.microsoft.com/en-us/sql/t-sql/statements/print-transact-sql>
+    Print(PrintStatement),
 }
 
 #[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
@@ -5745,7 +5751,7 @@ impl fmt::Display for Statement {
                 }
                 Ok(())
             }
-
+            Statement::Print(s) => write!(f, "{s}"),
             Statement::List(command) => write!(f, "LIST {command}"),
             Statement::Remove(command) => write!(f, "REMOVE {command}"),
         }
@@ -9211,6 +9217,19 @@ pub enum CopyIntoSnowflakeKind {
     Location,
 }
 
+#[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
+#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
+#[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
+pub struct PrintStatement {
+    pub message: Box<Expr>,
+}
+
+impl fmt::Display for PrintStatement {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        write!(f, "PRINT {}", self.message)
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
diff --git a/src/ast/spans.rs b/src/ast/spans.rs
index 230341518..a241fdf4d 100644
--- a/src/ast/spans.rs
+++ b/src/ast/spans.rs
@@ -519,6 +519,7 @@ impl Spanned for Statement {
             Statement::UNLISTEN { .. } => Span::empty(),
             Statement::RenameTable { .. } => Span::empty(),
             Statement::RaisError { .. } => Span::empty(),
+            Statement::Print { .. } => Span::empty(),
             Statement::List(..) | Statement::Remove(..) => Span::empty(),
         }
     }
diff --git a/src/keywords.rs b/src/keywords.rs
index fb2734096..a5400a5b0 100644
--- a/src/keywords.rs
+++ b/src/keywords.rs
@@ -686,6 +686,7 @@ define_keywords!(
     PRESERVE,
     PREWHERE,
     PRIMARY,
+    PRINT,
     PRIOR,
     PRIVILEGES,
     PROCEDURE,
diff --git a/src/parser/mod.rs b/src/parser/mod.rs
index bb989fef1..f9f146ce0 100644
--- a/src/parser/mod.rs
+++ b/src/parser/mod.rs
@@ -617,6 +617,7 @@ impl<'a> Parser<'a> {
                 }
                 // `COMMENT` is snowflake specific https://docs.snowflake.com/en/sql-reference/sql/comment
                 Keyword::COMMENT if self.dialect.supports_comment_on() => self.parse_comment(),
+                Keyword::PRINT => self.parse_print(),
                 _ => self.expected("an SQL statement", next_token),
             },
             Token::LParen => {
@@ -15058,6 +15059,13 @@ impl<'a> Parser<'a> {
         }
     }
 
+    /// Parse [Statement::Print]
+    fn parse_print(&mut self) -> Result<Statement, ParserError> {
+        Ok(Statement::Print(PrintStatement {
+            message: Box::new(self.parse_expr()?),
+        }))
+    }
+
     /// Consume the parser and return its underlying token buffer
     pub fn into_tokens(self) -> Vec<TokenWithSpan> {
         self.tokens
diff --git a/tests/sqlparser_mssql.rs b/tests/sqlparser_mssql.rs
index 644589b58..2786384b3 100644
--- a/tests/sqlparser_mssql.rs
+++ b/tests/sqlparser_mssql.rs
@@ -2053,3 +2053,20 @@ fn parse_drop_trigger() {
         }
     );
 }
+
+#[test]
+fn parse_print() {
+    let print_string_literal = "PRINT 'Hello, world!'";
+    let print_stmt = ms().verified_stmt(print_string_literal);
+    assert_eq!(
+        print_stmt,
+        Statement::Print(PrintStatement {
+            message: Box::new(Expr::Value(
+                (Value::SingleQuotedString("Hello, world!".to_string())).with_empty_span()
+            )),
+        })
+    );
+
+    let _ = ms().verified_stmt("PRINT N'Hello, ⛄️!'");
+    let _ = ms().verified_stmt("PRINT @my_variable");
+}