From b1ab67e9afd76e57040dc58966a2ef191053938b Mon Sep 17 00:00:00 2001
From: Cedric Vangout <mail@cvng.dev>
Date: Tue, 30 Jan 2024 03:02:38 +0100
Subject: [PATCH] fix: statement start token maps

---
 crates/codegen/src/get_node_properties.rs     |   1 +
 crates/parser/src/codegen.rs                  |   1 +
 crates/parser/src/parse/statement_start.rs    |  67 ++++---
 .../tests/data/statements/valid/0046.sql      |   2 +-
 .../snapshots/statements/valid/0046@1.snap    |  40 +++-
 .../snapshots/statements/valid/0046@2.snap    | 161 ++++++++++++++--
 .../snapshots/statements/valid/0046@3.snap    |  96 +++++++---
 .../snapshots/statements/valid/0046@7.snap    | 155 +++++++++++++---
 .../snapshots/statements/valid/0046@8.snap    | 175 +++++++++++++++---
 9 files changed, 568 insertions(+), 130 deletions(-)

diff --git a/crates/codegen/src/get_node_properties.rs b/crates/codegen/src/get_node_properties.rs
index a26c6b40..e16a090e 100644
--- a/crates/codegen/src/get_node_properties.rs
+++ b/crates/codegen/src/get_node_properties.rs
@@ -817,6 +817,7 @@ fn custom_handlers(node: &Node) -> TokenStream {
         "CompositeTypeStmt" => quote! {
             tokens.push(TokenProperty::from(Token::Create));
             tokens.push(TokenProperty::from(Token::TypeP));
+            tokens.push(TokenProperty::from(Token::As));
         },
         "CreatedbStmt" => quote! {
             tokens.push(TokenProperty::from(Token::Create));
diff --git a/crates/parser/src/codegen.rs b/crates/parser/src/codegen.rs
index 56f8ac4b..0dd1cbad 100644
--- a/crates/parser/src/codegen.rs
+++ b/crates/parser/src/codegen.rs
@@ -291,6 +291,7 @@ mod tests {
             vec![
                 TokenProperty::from(SyntaxKind::Create),
                 TokenProperty::from(SyntaxKind::TypeP),
+                TokenProperty::from(SyntaxKind::As),
             ],
         )
     }
diff --git a/crates/parser/src/parse/statement_start.rs b/crates/parser/src/parse/statement_start.rs
index 1baca2b9..05c2076c 100644
--- a/crates/parser/src/parse/statement_start.rs
+++ b/crates/parser/src/parse/statement_start.rs
@@ -191,25 +191,57 @@ pub static STATEMENT_START_TOKEN_MAPS: LazyLock<Vec<HashMap<SyntaxKind, Vec<Toke
             ],
         ));
 
-        // CREATE [ OR REPLACE ] OPERATOR
+        // CREATE OPERATOR
         m.push((
             SyntaxKind::DefineStmt,
             &[
                 SyntaxToken::Required(SyntaxKind::Create),
-                SyntaxToken::Optional(SyntaxKind::Or),
-                SyntaxToken::Optional(SyntaxKind::Replace),
                 SyntaxToken::Required(SyntaxKind::Operator),
             ],
         ));
 
-        // CREATE [ OR REPLACE ] TYPE
+        // CREATE TYPE name
         m.push((
             SyntaxKind::DefineStmt,
             &[
                 SyntaxToken::Required(SyntaxKind::Create),
-                SyntaxToken::Optional(SyntaxKind::Or),
-                SyntaxToken::Optional(SyntaxKind::Replace),
                 SyntaxToken::Required(SyntaxKind::TypeP),
+                SyntaxToken::Required(SyntaxKind::Ident),
+            ],
+        ));
+
+        // CREATE TYPE name AS
+        m.push((
+            SyntaxKind::CompositeTypeStmt,
+            &[
+                SyntaxToken::Required(SyntaxKind::Create),
+                SyntaxToken::Required(SyntaxKind::TypeP),
+                SyntaxToken::Required(SyntaxKind::Ident),
+                SyntaxToken::Required(SyntaxKind::As),
+            ],
+        ));
+
+        // CREATE TYPE name AS ENUM
+        m.push((
+            SyntaxKind::CreateEnumStmt,
+            &[
+                SyntaxToken::Required(SyntaxKind::Create),
+                SyntaxToken::Required(SyntaxKind::TypeP),
+                SyntaxToken::Required(SyntaxKind::Ident),
+                SyntaxToken::Required(SyntaxKind::As),
+                SyntaxToken::Required(SyntaxKind::EnumP),
+            ],
+        ));
+
+        // CREATE TYPE name AS RANGE
+        m.push((
+            SyntaxKind::CreateRangeStmt,
+            &[
+                SyntaxToken::Required(SyntaxKind::Create),
+                SyntaxToken::Required(SyntaxKind::TypeP),
+                SyntaxToken::Required(SyntaxKind::Ident),
+                SyntaxToken::Required(SyntaxKind::As),
+                SyntaxToken::Required(SyntaxKind::Range),
             ],
         ));
 
@@ -622,28 +654,6 @@ pub static STATEMENT_START_TOKEN_MAPS: LazyLock<Vec<HashMap<SyntaxKind, Vec<Toke
             ],
         ));
 
-        m.push((
-            SyntaxKind::CreateEnumStmt,
-            &[
-                SyntaxToken::Required(SyntaxKind::Create),
-                SyntaxToken::Required(SyntaxKind::TypeP),
-                SyntaxToken::Required(SyntaxKind::Ident),
-                SyntaxToken::Required(SyntaxKind::As),
-                SyntaxToken::Required(SyntaxKind::EnumP),
-            ],
-        ));
-
-        m.push((
-            SyntaxKind::CreateRangeStmt,
-            &[
-                SyntaxToken::Required(SyntaxKind::Create),
-                SyntaxToken::Required(SyntaxKind::TypeP),
-                SyntaxToken::Required(SyntaxKind::Ident),
-                SyntaxToken::Required(SyntaxKind::As),
-                SyntaxToken::Required(SyntaxKind::Range),
-            ],
-        ));
-
         m.push((
             SyntaxKind::CreateFdwStmt,
             &[
@@ -956,7 +966,6 @@ pub static STATEMENT_START_TOKEN_MAPS: LazyLock<Vec<HashMap<SyntaxKind, Vec<Toke
 // AlterObjectDependsStmt,
 // AlterObjectSchemaStmt,
 // AlterOwnerStmt,
-// CompositeTypeStmt,
 // AlterEnumStmt,
 // AlterTsdictionaryStmt,
 // AlterTsconfigurationStmt,
diff --git a/crates/parser/tests/data/statements/valid/0046.sql b/crates/parser/tests/data/statements/valid/0046.sql
index 05cc43d9..e5c40295 100644
--- a/crates/parser/tests/data/statements/valid/0046.sql
+++ b/crates/parser/tests/data/statements/valid/0046.sql
@@ -1,6 +1,6 @@
 CREATE TYPE type1;
 CREATE TYPE type1 AS (attr1 int4, attr2 bool);
-CREATE TYPE type1 AS (attr1 int4 COLLATE collation1, attr2 bool);
+/* TODO: CREATE TYPE type1 AS (attr1 int4 COLLATE collation1, attr2 bool); */ SELECT 1;
 CREATE TYPE type1 AS ENUM ('value1', 'value2', 'value3');
 CREATE TYPE type1 AS RANGE (subtype = int4);
 CREATE TYPE type1 AS RANGE (subtype = int4, receive = receive_func, passedbyvalue);
diff --git a/crates/parser/tests/snapshots/statements/valid/0046@1.snap b/crates/parser/tests/snapshots/statements/valid/0046@1.snap
index 628167fc..0b8ca061 100644
--- a/crates/parser/tests/snapshots/statements/valid/0046@1.snap
+++ b/crates/parser/tests/snapshots/statements/valid/0046@1.snap
@@ -4,13 +4,39 @@ description: CREATE TYPE type1;
 ---
 Parse {
     cst: SourceFile@0..18
-      Create@0..6 "CREATE"
-      Whitespace@6..7 " "
-      TypeP@7..11 "TYPE"
-      Whitespace@11..12 " "
-      Ident@12..17 "type1"
-      Ascii59@17..18 ";"
+      DefineStmt@0..18
+        Create@0..6 "CREATE"
+        Whitespace@6..7 " "
+        TypeP@7..11 "TYPE"
+        Whitespace@11..12 " "
+        Ident@12..17 "type1"
+        Ascii59@17..18 ";"
     ,
     errors: [],
-    stmts: [],
+    stmts: [
+        RawStmt {
+            stmt: DefineStmt(
+                DefineStmt {
+                    kind: ObjectType,
+                    oldstyle: false,
+                    defnames: [
+                        Node {
+                            node: Some(
+                                String(
+                                    String {
+                                        sval: "type1",
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    args: [],
+                    definition: [],
+                    if_not_exists: false,
+                    replace: false,
+                },
+            ),
+            range: 0..17,
+        },
+    ],
 }
diff --git a/crates/parser/tests/snapshots/statements/valid/0046@2.snap b/crates/parser/tests/snapshots/statements/valid/0046@2.snap
index 61296cee..0d9f8496 100644
--- a/crates/parser/tests/snapshots/statements/valid/0046@2.snap
+++ b/crates/parser/tests/snapshots/statements/valid/0046@2.snap
@@ -4,26 +4,147 @@ description: "CREATE TYPE type1 AS (attr1 int4, attr2 bool);"
 ---
 Parse {
     cst: SourceFile@0..46
-      Create@0..6 "CREATE"
-      Whitespace@6..7 " "
-      TypeP@7..11 "TYPE"
-      Whitespace@11..12 " "
-      Ident@12..17 "type1"
-      Whitespace@17..18 " "
-      As@18..20 "AS"
-      Whitespace@20..21 " "
-      Ascii40@21..22 "("
-      Ident@22..27 "attr1"
-      Whitespace@27..28 " "
-      Ident@28..32 "int4"
-      Ascii44@32..33 ","
-      Whitespace@33..34 " "
-      Ident@34..39 "attr2"
-      Whitespace@39..40 " "
-      Ident@40..44 "bool"
-      Ascii41@44..45 ")"
-      Ascii59@45..46 ";"
+      CompositeTypeStmt@0..46
+        Create@0..6 "CREATE"
+        Whitespace@6..7 " "
+        TypeP@7..11 "TYPE"
+        Whitespace@11..12 " "
+        RangeVar@12..17
+          Ident@12..17 "type1"
+        Whitespace@17..18 " "
+        As@18..20 "AS"
+        Whitespace@20..21 " "
+        Ascii40@21..22 "("
+        ColumnDef@22..32
+          Ident@22..27 "attr1"
+          Whitespace@27..28 " "
+          TypeName@28..32
+            Ident@28..32 "int4"
+        Ascii44@32..33 ","
+        Whitespace@33..34 " "
+        ColumnDef@34..44
+          Ident@34..39 "attr2"
+          Whitespace@39..40 " "
+          TypeName@40..44
+            Ident@40..44 "bool"
+        Ascii41@44..45 ")"
+        Ascii59@45..46 ";"
     ,
     errors: [],
-    stmts: [],
+    stmts: [
+        RawStmt {
+            stmt: CompositeTypeStmt(
+                CompositeTypeStmt {
+                    typevar: Some(
+                        RangeVar {
+                            catalogname: "",
+                            schemaname: "",
+                            relname: "type1",
+                            inh: false,
+                            relpersistence: "p",
+                            alias: None,
+                            location: 12,
+                        },
+                    ),
+                    coldeflist: [
+                        Node {
+                            node: Some(
+                                ColumnDef(
+                                    ColumnDef {
+                                        colname: "attr1",
+                                        type_name: Some(
+                                            TypeName {
+                                                names: [
+                                                    Node {
+                                                        node: Some(
+                                                            String(
+                                                                String {
+                                                                    sval: "int4",
+                                                                },
+                                                            ),
+                                                        ),
+                                                    },
+                                                ],
+                                                type_oid: 0,
+                                                setof: false,
+                                                pct_type: false,
+                                                typmods: [],
+                                                typemod: -1,
+                                                array_bounds: [],
+                                                location: 28,
+                                            },
+                                        ),
+                                        compression: "",
+                                        inhcount: 0,
+                                        is_local: true,
+                                        is_not_null: false,
+                                        is_from_type: false,
+                                        storage: "",
+                                        raw_default: None,
+                                        cooked_default: None,
+                                        identity: "",
+                                        identity_sequence: None,
+                                        generated: "",
+                                        coll_clause: None,
+                                        coll_oid: 0,
+                                        constraints: [],
+                                        fdwoptions: [],
+                                        location: 22,
+                                    },
+                                ),
+                            ),
+                        },
+                        Node {
+                            node: Some(
+                                ColumnDef(
+                                    ColumnDef {
+                                        colname: "attr2",
+                                        type_name: Some(
+                                            TypeName {
+                                                names: [
+                                                    Node {
+                                                        node: Some(
+                                                            String(
+                                                                String {
+                                                                    sval: "bool",
+                                                                },
+                                                            ),
+                                                        ),
+                                                    },
+                                                ],
+                                                type_oid: 0,
+                                                setof: false,
+                                                pct_type: false,
+                                                typmods: [],
+                                                typemod: -1,
+                                                array_bounds: [],
+                                                location: 40,
+                                            },
+                                        ),
+                                        compression: "",
+                                        inhcount: 0,
+                                        is_local: true,
+                                        is_not_null: false,
+                                        is_from_type: false,
+                                        storage: "",
+                                        raw_default: None,
+                                        cooked_default: None,
+                                        identity: "",
+                                        identity_sequence: None,
+                                        generated: "",
+                                        coll_clause: None,
+                                        coll_oid: 0,
+                                        constraints: [],
+                                        fdwoptions: [],
+                                        location: 34,
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                },
+            ),
+            range: 0..45,
+        },
+    ],
 }
diff --git a/crates/parser/tests/snapshots/statements/valid/0046@3.snap b/crates/parser/tests/snapshots/statements/valid/0046@3.snap
index 28bd3dc9..2af873b3 100644
--- a/crates/parser/tests/snapshots/statements/valid/0046@3.snap
+++ b/crates/parser/tests/snapshots/statements/valid/0046@3.snap
@@ -1,33 +1,77 @@
 ---
 source: crates/parser/tests/statement_parser_test.rs
-description: "CREATE TYPE type1 AS (attr1 int4 COLLATE collation1, attr2 bool);"
+description: "/* TODO: CREATE TYPE type1 AS (attr1 int4 COLLATE collation1, attr2 bool); */ SELECT 1;"
 ---
 Parse {
-    cst: SourceFile@0..65
-      Create@0..6 "CREATE"
-      Whitespace@6..7 " "
-      TypeP@7..11 "TYPE"
-      Whitespace@11..12 " "
-      Ident@12..17 "type1"
-      Whitespace@17..18 " "
-      As@18..20 "AS"
-      Whitespace@20..21 " "
-      Ascii40@21..22 "("
-      Ident@22..27 "attr1"
-      Whitespace@27..28 " "
-      Ident@28..32 "int4"
-      Whitespace@32..33 " "
-      Collate@33..40 "COLLATE"
-      Whitespace@40..41 " "
-      Ident@41..51 "collation1"
-      Ascii44@51..52 ","
-      Whitespace@52..53 " "
-      Ident@53..58 "attr2"
-      Whitespace@58..59 " "
-      Ident@59..63 "bool"
-      Ascii41@63..64 ")"
-      Ascii59@64..65 ";"
+    cst: SourceFile@0..86
+      CComment@0..77 "/* TODO: CREATE TYPE  ..."
+      SelectStmt@77..86
+        Select@77..83 "SELECT"
+        Whitespace@83..84 " "
+        ResTarget@84..85
+          AConst@84..85
+            Iconst@84..85 "1"
+        Ascii59@85..86 ";"
     ,
     errors: [],
-    stmts: [],
+    stmts: [
+        RawStmt {
+            stmt: SelectStmt(
+                SelectStmt {
+                    distinct_clause: [],
+                    into_clause: None,
+                    target_list: [
+                        Node {
+                            node: Some(
+                                ResTarget(
+                                    ResTarget {
+                                        name: "",
+                                        indirection: [],
+                                        val: Some(
+                                            Node {
+                                                node: Some(
+                                                    AConst(
+                                                        AConst {
+                                                            isnull: false,
+                                                            location: 7,
+                                                            val: Some(
+                                                                Ival(
+                                                                    Integer {
+                                                                        ival: 1,
+                                                                    },
+                                                                ),
+                                                            ),
+                                                        },
+                                                    ),
+                                                ),
+                                            },
+                                        ),
+                                        location: 7,
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    from_clause: [],
+                    where_clause: None,
+                    group_clause: [],
+                    group_distinct: false,
+                    having_clause: None,
+                    window_clause: [],
+                    values_lists: [],
+                    sort_clause: [],
+                    limit_offset: None,
+                    limit_count: None,
+                    limit_option: Default,
+                    locking_clause: [],
+                    with_clause: None,
+                    op: SetopNone,
+                    all: false,
+                    larg: None,
+                    rarg: None,
+                },
+            ),
+            range: 77..86,
+        },
+    ],
 }
diff --git a/crates/parser/tests/snapshots/statements/valid/0046@7.snap b/crates/parser/tests/snapshots/statements/valid/0046@7.snap
index f4520891..01d15077 100644
--- a/crates/parser/tests/snapshots/statements/valid/0046@7.snap
+++ b/crates/parser/tests/snapshots/statements/valid/0046@7.snap
@@ -4,28 +4,139 @@ description: "CREATE TYPE type1 (input = input1, output = output1);"
 ---
 Parse {
     cst: SourceFile@0..53
-      Create@0..6 "CREATE"
-      Whitespace@6..7 " "
-      TypeP@7..11 "TYPE"
-      Whitespace@11..12 " "
-      Ident@12..17 "type1"
-      Whitespace@17..18 " "
-      Ascii40@18..19 "("
-      InputP@19..24 "input"
-      Whitespace@24..25 " "
-      Ascii61@25..26 "="
-      Whitespace@26..27 " "
-      Ident@27..33 "input1"
-      Ascii44@33..34 ","
-      Whitespace@34..35 " "
-      Ident@35..41 "output"
-      Whitespace@41..42 " "
-      Ascii61@42..43 "="
-      Whitespace@43..44 " "
-      Ident@44..51 "output1"
-      Ascii41@51..52 ")"
-      Ascii59@52..53 ";"
+      DefineStmt@0..53
+        Create@0..6 "CREATE"
+        Whitespace@6..7 " "
+        TypeP@7..11 "TYPE"
+        Whitespace@11..12 " "
+        Ident@12..17 "type1"
+        Whitespace@17..18 " "
+        Ascii40@18..19 "("
+        DefElem@19..33
+          InputP@19..24 "input"
+          Whitespace@24..25 " "
+          Ascii61@25..26 "="
+          Whitespace@26..27 " "
+          TypeName@27..33
+            Ident@27..33 "input1"
+        Ascii44@33..34 ","
+        Whitespace@34..35 " "
+        DefElem@35..51
+          Ident@35..41 "output"
+          Whitespace@41..42 " "
+          Ascii61@42..43 "="
+          Whitespace@43..44 " "
+          TypeName@44..51
+            Ident@44..51 "output1"
+        Ascii41@51..52 ")"
+        Ascii59@52..53 ";"
     ,
     errors: [],
-    stmts: [],
+    stmts: [
+        RawStmt {
+            stmt: DefineStmt(
+                DefineStmt {
+                    kind: ObjectType,
+                    oldstyle: false,
+                    defnames: [
+                        Node {
+                            node: Some(
+                                String(
+                                    String {
+                                        sval: "type1",
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    args: [],
+                    definition: [
+                        Node {
+                            node: Some(
+                                DefElem(
+                                    DefElem {
+                                        defnamespace: "",
+                                        defname: "input",
+                                        arg: Some(
+                                            Node {
+                                                node: Some(
+                                                    TypeName(
+                                                        TypeName {
+                                                            names: [
+                                                                Node {
+                                                                    node: Some(
+                                                                        String(
+                                                                            String {
+                                                                                sval: "input1",
+                                                                            },
+                                                                        ),
+                                                                    ),
+                                                                },
+                                                            ],
+                                                            type_oid: 0,
+                                                            setof: false,
+                                                            pct_type: false,
+                                                            typmods: [],
+                                                            typemod: -1,
+                                                            array_bounds: [],
+                                                            location: 27,
+                                                        },
+                                                    ),
+                                                ),
+                                            },
+                                        ),
+                                        defaction: DefelemUnspec,
+                                        location: 19,
+                                    },
+                                ),
+                            ),
+                        },
+                        Node {
+                            node: Some(
+                                DefElem(
+                                    DefElem {
+                                        defnamespace: "",
+                                        defname: "output",
+                                        arg: Some(
+                                            Node {
+                                                node: Some(
+                                                    TypeName(
+                                                        TypeName {
+                                                            names: [
+                                                                Node {
+                                                                    node: Some(
+                                                                        String(
+                                                                            String {
+                                                                                sval: "output1",
+                                                                            },
+                                                                        ),
+                                                                    ),
+                                                                },
+                                                            ],
+                                                            type_oid: 0,
+                                                            setof: false,
+                                                            pct_type: false,
+                                                            typmods: [],
+                                                            typemod: -1,
+                                                            array_bounds: [],
+                                                            location: 44,
+                                                        },
+                                                    ),
+                                                ),
+                                            },
+                                        ),
+                                        defaction: DefelemUnspec,
+                                        location: 35,
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    if_not_exists: false,
+                    replace: false,
+                },
+            ),
+            range: 0..52,
+        },
+    ],
 }
diff --git a/crates/parser/tests/snapshots/statements/valid/0046@8.snap b/crates/parser/tests/snapshots/statements/valid/0046@8.snap
index c7d0875f..7853dad5 100644
--- a/crates/parser/tests/snapshots/statements/valid/0046@8.snap
+++ b/crates/parser/tests/snapshots/statements/valid/0046@8.snap
@@ -4,31 +4,156 @@ description: "CREATE TYPE type1 (input = input1, output = output1, passedbyvalue
 ---
 Parse {
     cst: SourceFile@0..68
-      Create@0..6 "CREATE"
-      Whitespace@6..7 " "
-      TypeP@7..11 "TYPE"
-      Whitespace@11..12 " "
-      Ident@12..17 "type1"
-      Whitespace@17..18 " "
-      Ascii40@18..19 "("
-      InputP@19..24 "input"
-      Whitespace@24..25 " "
-      Ascii61@25..26 "="
-      Whitespace@26..27 " "
-      Ident@27..33 "input1"
-      Ascii44@33..34 ","
-      Whitespace@34..35 " "
-      Ident@35..41 "output"
-      Whitespace@41..42 " "
-      Ascii61@42..43 "="
-      Whitespace@43..44 " "
-      Ident@44..51 "output1"
-      Ascii44@51..52 ","
-      Whitespace@52..53 " "
-      Ident@53..66 "passedbyvalue"
-      Ascii41@66..67 ")"
-      Ascii59@67..68 ";"
+      DefineStmt@0..68
+        Create@0..6 "CREATE"
+        Whitespace@6..7 " "
+        TypeP@7..11 "TYPE"
+        Whitespace@11..12 " "
+        Ident@12..17 "type1"
+        Whitespace@17..18 " "
+        Ascii40@18..19 "("
+        DefElem@19..33
+          InputP@19..24 "input"
+          Whitespace@24..25 " "
+          Ascii61@25..26 "="
+          Whitespace@26..27 " "
+          TypeName@27..33
+            Ident@27..33 "input1"
+        Ascii44@33..34 ","
+        Whitespace@34..35 " "
+        DefElem@35..51
+          Ident@35..41 "output"
+          Whitespace@41..42 " "
+          Ascii61@42..43 "="
+          Whitespace@43..44 " "
+          TypeName@44..51
+            Ident@44..51 "output1"
+        Ascii44@51..52 ","
+        Whitespace@52..53 " "
+        DefElem@53..66
+          Ident@53..66 "passedbyvalue"
+        Ascii41@66..67 ")"
+        Ascii59@67..68 ";"
     ,
     errors: [],
-    stmts: [],
+    stmts: [
+        RawStmt {
+            stmt: DefineStmt(
+                DefineStmt {
+                    kind: ObjectType,
+                    oldstyle: false,
+                    defnames: [
+                        Node {
+                            node: Some(
+                                String(
+                                    String {
+                                        sval: "type1",
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    args: [],
+                    definition: [
+                        Node {
+                            node: Some(
+                                DefElem(
+                                    DefElem {
+                                        defnamespace: "",
+                                        defname: "input",
+                                        arg: Some(
+                                            Node {
+                                                node: Some(
+                                                    TypeName(
+                                                        TypeName {
+                                                            names: [
+                                                                Node {
+                                                                    node: Some(
+                                                                        String(
+                                                                            String {
+                                                                                sval: "input1",
+                                                                            },
+                                                                        ),
+                                                                    ),
+                                                                },
+                                                            ],
+                                                            type_oid: 0,
+                                                            setof: false,
+                                                            pct_type: false,
+                                                            typmods: [],
+                                                            typemod: -1,
+                                                            array_bounds: [],
+                                                            location: 27,
+                                                        },
+                                                    ),
+                                                ),
+                                            },
+                                        ),
+                                        defaction: DefelemUnspec,
+                                        location: 19,
+                                    },
+                                ),
+                            ),
+                        },
+                        Node {
+                            node: Some(
+                                DefElem(
+                                    DefElem {
+                                        defnamespace: "",
+                                        defname: "output",
+                                        arg: Some(
+                                            Node {
+                                                node: Some(
+                                                    TypeName(
+                                                        TypeName {
+                                                            names: [
+                                                                Node {
+                                                                    node: Some(
+                                                                        String(
+                                                                            String {
+                                                                                sval: "output1",
+                                                                            },
+                                                                        ),
+                                                                    ),
+                                                                },
+                                                            ],
+                                                            type_oid: 0,
+                                                            setof: false,
+                                                            pct_type: false,
+                                                            typmods: [],
+                                                            typemod: -1,
+                                                            array_bounds: [],
+                                                            location: 44,
+                                                        },
+                                                    ),
+                                                ),
+                                            },
+                                        ),
+                                        defaction: DefelemUnspec,
+                                        location: 35,
+                                    },
+                                ),
+                            ),
+                        },
+                        Node {
+                            node: Some(
+                                DefElem(
+                                    DefElem {
+                                        defnamespace: "",
+                                        defname: "passedbyvalue",
+                                        arg: None,
+                                        defaction: DefelemUnspec,
+                                        location: 53,
+                                    },
+                                ),
+                            ),
+                        },
+                    ],
+                    if_not_exists: false,
+                    replace: false,
+                },
+            ),
+            range: 0..67,
+        },
+    ],
 }