Implement SHOW CREATE SOURCE <source> (#22)

Author: JLDLaughlin

src/ast/mod.rs

@@ -635,6 +635,8 @@ pub enum Statement {
     },
     /// `SHOW CREATE VIEW <view>`
     ShowCreateView { view_name: ObjectName },
+    /// `SHOW CREATE SOURCE <source>`
+    ShowCreateSource { source_name: ObjectName },
     /// `{ BEGIN [ TRANSACTION | WORK ] | START TRANSACTION } ...`
     StartTransaction { modes: Vec<TransactionMode> },
     /// `SET TRANSACTION ...`
@@ -927,6 +929,10 @@ impl fmt::Display for Statement {
                 f.write_str("SHOW CREATE VIEW ")?;
                 write!(f, "{}", view_name)
             }
+            Statement::ShowCreateSource { source_name } => {
+                f.write_str("SHOW CREATE SOURCE ")?;
+                write!(f, "{}", source_name)
+            }
             Statement::StartTransaction { modes } => {
                 write!(f, "START TRANSACTION")?;
                 if !modes.is_empty() {

src/ast/visit_macro.rs

@@ -526,6 +526,13 @@ macro_rules! make_visitor {
                 visit_show_create_view(self, view_name)
             }
 
+            fn visit_show_create_source(
+                &mut self,
+                source_name: &'ast $($mut)* ObjectName,
+            ) {
+                visit_show_create_source(self, source_name)
+            }
+
             fn visit_show_statement_filter(&mut self, filter: &'ast $($mut)* ShowStatementFilter) {
                 visit_show_statement_filter(self, filter)
             }
@@ -666,6 +673,7 @@ macro_rules! make_visitor {
                     filter,
                 } => visitor.visit_show_columns(*extended, *full, table_name, filter.as_auto_ref()),
                 Statement::ShowCreateView { view_name } => visitor.visit_show_create_view(view_name),
+                Statement::ShowCreateSource { source_name } => visitor.visit_show_create_source(source_name),
                 Statement::StartTransaction { modes } => visitor.visit_start_transaction(modes),
                 Statement::SetTransaction { modes } => visitor.visit_set_transaction(modes),
                 Statement::Commit { chain } => visitor.visit_commit(*chain),
@@ -1571,6 +1579,13 @@ macro_rules! make_visitor {
             visitor.visit_object_name(view_name);
         }
 
+        pub fn visit_show_create_source<'ast, V: $name<'ast> + ?Sized>(
+            visitor: &mut V,
+            source_name: &'ast $($mut)* ObjectName
+        ) {
+            visitor.visit_object_name(source_name);
+        }
+
         pub fn visit_show_statement_filter<'ast, V: $name<'ast> + ?Sized>(
             visitor: &mut V,
             filter: &'ast $($mut)* ShowStatementFilter,

src/parser.rs

@@ -1946,6 +1946,10 @@ impl Parser {
             Ok(Statement::ShowCreateView {
                 view_name: self.parse_object_name()?,
             })
+        } else if self.parse_keywords(vec!["CREATE", "SOURCE"]) {
+            Ok(Statement::ShowCreateSource {
+                source_name: self.parse_object_name()?,
+            })
         } else {
             Ok(Statement::ShowVariable {
                 variable: self.parse_identifier()?,

tests/sqlparser_common.rs

@@ -1945,6 +1945,16 @@ fn parse_show_create_view() {
     )
 }
 
+#[test]
+fn parse_show_create_source() {
+    assert_eq!(
+        verified_stmt("SHOW CREATE SOURCE foo"),
+        Statement::ShowCreateSource {
+            source_name: ObjectName(vec!["foo".into()])
+        }
+    )
+}
+
 #[test]
 fn parse_simple_case_expr() {
     // ANSI calls a CASE expression with an operand "<simple case>"