feat: support "create schema"

supabase-community · Dec 12, 2023 · b527f75 · b527f75
1 parent 00fd160
commit b527f75
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/crates/codegen/src/get_node_properties.rs b/crates/codegen/src/get_node_properties.rs
@@ -536,6 +536,18 @@ fn custom_handlers(node: &Node) -> TokenStream {
                 tokens.push(TokenProperty::from(Token::As));
             }
         },
+        "CreateSchemaStmt" => quote! {
+            tokens.push(TokenProperty::from(Token::Create));
+            tokens.push(TokenProperty::from(Token::Schema));
+            if n.if_not_exists {
+                tokens.push(TokenProperty::from(Token::IfP));
+                tokens.push(TokenProperty::from(Token::Not));
+                tokens.push(TokenProperty::from(Token::Exists));
+            }
+            if n.authrole.is_some() {
+                tokens.push(TokenProperty::from(Token::Authorization));
+            }
+        },
         _ => quote! {},
     }
 }

diff --git a/crates/parser/src/codegen.rs b/crates/parser/src/codegen.rs
@@ -107,4 +107,21 @@ mod tests {
             ],
         )
     }
+
+    #[test]
+    fn test_create_schema() {
+        test_get_node_properties(
+            "create schema if not exists test authorization joe;",
+            SyntaxKind::CreateSchemaStmt,
+            vec![
+                TokenProperty::from(SyntaxKind::Create),
+                TokenProperty::from(SyntaxKind::Schema),
+                TokenProperty::from(SyntaxKind::IfP),
+                TokenProperty::from(SyntaxKind::Not),
+                TokenProperty::from(SyntaxKind::Exists),
+                TokenProperty::from(SyntaxKind::Authorization),
+                TokenProperty::from("test".to_string()),
+            ],
+        )
+    }
 }