Add support for more create table options after AS

Author: yoavcloud

src/ast/helpers/stmt_create_table.rs

@@ -383,6 +383,16 @@ impl CreateTableBuilder {
         self
     }
 
+    /// Returns true if information on the structure of the table
+    /// to be created was provided to the builder. If not, the
+    /// statement is invalid.
+    pub fn has_schema_info(&self) -> bool {
+        !self.columns.is_empty()
+            || self.query.is_some()
+            || self.like.is_some()
+            || self.clone.is_some()
+    }
+
     pub fn build(self) -> Statement {
         Statement::CreateTable(CreateTable {
             or_replace: self.or_replace,

src/dialect/snowflake.rs

@@ -473,12 +473,10 @@ pub fn parse_create_table(
                 Keyword::CLONE => {
                     let clone = parser.parse_object_name(false).ok();
                     builder = builder.clone_clause(clone);
-                    break;
                 }
                 Keyword::LIKE => {
                     let like = parser.parse_object_name(false).ok();
                     builder = builder.like(like);
-                    break;
                 }
                 Keyword::CLUSTER => {
                     parser.expect_keyword_is(Keyword::BY)?;
@@ -604,7 +602,7 @@ pub fn parse_create_table(
                 builder = builder.columns(columns).constraints(constraints);
             }
             Token::EOF => {
-                if builder.columns.is_empty() && builder.query.is_none() {
+                if !builder.has_schema_info() {
                     return Err(ParserError::ParserError(
                         "unexpected end of input".to_string(),
                     ));
@@ -613,7 +611,7 @@ pub fn parse_create_table(
                 break;
             }
             Token::SemiColon => {
-                if builder.columns.is_empty() && builder.query.is_none() {
+                if !builder.has_schema_info() {
                     return Err(ParserError::ParserError(
                         "unexpected end of input".to_string(),
                     ));

tests/sqlparser_snowflake.rs

@@ -996,13 +996,18 @@ fn test_snowflake_create_iceberg_table_without_location() {
 }
 
 #[test]
-fn test_snowflake_create_table_as() {
-    // Test additional options after AS (query)
+fn test_snowflake_create_table_trailing_options() {
     snowflake()
         .parse_sql_statements(
             "CREATE TEMP TABLE dst AS (SELECT * FROM src) ON COMMIT PRESERVE ROWS",
         )
         .unwrap();
+    snowflake()
+        .parse_sql_statements("CREATE TEMP TABLE tbl LIKE customers ON COMMIT PRESERVE ROWS;")
+        .unwrap();
+    snowflake()
+        .parse_sql_statements("CREATE TEMP TABLE tbl CLONE customers ON COMMIT PRESERVE ROWS;")
+        .unwrap();
 }
 
 #[test]