feat: support CreateCastStmt

Author: cvng

crates/codegen/src/get_node_properties.rs

@@ -565,6 +565,39 @@ fn custom_handlers(node: &Node) -> TokenStream {
  tokens.push(TokenProperty::from(Token::As));
  tokens.push(TokenProperty::from(Token::EnumP));
  },
+ "CreateCastStmt" => quote! {
+ tokens.push(TokenProperty::from(Token::Create));
+ tokens.push(TokenProperty::from(Token::Cast));
+ tokens.push(TokenProperty::from(Token::As));
+ if n.inout {
+ tokens.push(TokenProperty::from(Token::With));
+ tokens.push(TokenProperty::from(Token::Inout));
+ }
+ else if n.func.is_some() {
+ tokens.push(TokenProperty::from(Token::With));
+ tokens.push(TokenProperty::from(Token::Function));
+ } else {
+ tokens.push(TokenProperty::from(Token::Without));
+ tokens.push(TokenProperty::from(Token::Function));
+ }
+ match n.context {
+ // Implicit
+ 1 => {
+ tokens.push(TokenProperty::from(Token::As));
+ tokens.push(TokenProperty::from(Token::ImplicitP));
+ },
+ // Assignment
+ 2 => {
+ tokens.push(TokenProperty::from(Token::As));
+ tokens.push(TokenProperty::from(Token::Assignment));
+ },
+ // Plpgsql
+ 3 => {},
+ // Explicit
+ 4 => {},
+ _ => panic!("Unknown CreateCastStmt {:#?}", n.context)
+ }
+ },
  _ => quote! {},
  }
 }

crates/parser/src/codegen.rs

@@ -156,4 +156,21 @@ mod tests {
  ],
  )
  }
+
+ #[test]
+ fn test_create_cast() {
+ test_get_node_properties(
+ "create cast (bigint as int4) with inout as assignment;",
+ SyntaxKind::CreateCastStmt,
+ vec![
+ TokenProperty::from(SyntaxKind::Create),
+ TokenProperty::from(SyntaxKind::Cast),
+ TokenProperty::from(SyntaxKind::As),
+ TokenProperty::from(SyntaxKind::With),
+ TokenProperty::from(SyntaxKind::Inout),
+ TokenProperty::from(SyntaxKind::As),
+ TokenProperty::from(SyntaxKind::Assignment),
+ ],
+ )
+ }
 }

crates/parser/tests/data/statements/valid/0038.sql

@@ -0,0 +1,3 @@
+CREATE CAST (bigint AS int4) WITH FUNCTION int4(bigint) AS ASSIGNMENT;
+CREATE CAST (bigint AS int4) WITHOUT FUNCTION AS IMPLICIT;
+CREATE CAST (bigint AS int4) WITH INOUT AS ASSIGNMENT;