Ruby: put AST node locations in a single table

Author: nickrolfe

ruby/extractor/src/extractor.rs

@@ -402,11 +402,12 @@ impl Visitor<'_> {
         match &table.kind {
             EntryKind::Token { kind_id, .. } => {
                 self.trap_writer.add_tuple(
-                    &format!("{}_ast_node_parent", self.language_prefix),
+                    &format!("{}_ast_node_info", self.language_prefix),
                     vec![
                         Arg::Label(id),
                         Arg::Label(parent_id),
                         Arg::Int(parent_index),
+                        Arg::Label(loc),
                     ],
                 );
                 self.trap_writer.add_tuple(
@@ -415,7 +416,6 @@ impl Visitor<'_> {
                         Arg::Label(id),
                         Arg::Int(*kind_id),
                         sliced_source_arg(self.source, node),
-                        Arg::Label(loc),
                     ],
                 );
             }
@@ -425,16 +425,16 @@ impl Visitor<'_> {
             } => {
                 if let Some(args) = self.complex_node(&node, fields, &child_nodes, id) {
                     self.trap_writer.add_tuple(
-                        &format!("{}_ast_node_parent", self.language_prefix),
+                        &format!("{}_ast_node_info", self.language_prefix),
                         vec![
                             Arg::Label(id),
                             Arg::Label(parent_id),
                             Arg::Int(parent_index),
+                            Arg::Label(loc),
                         ],
                     );
                     let mut all_args = vec![Arg::Label(id)];
                     all_args.extend(args);
-                    all_args.push(Arg::Label(loc));
                     self.trap_writer.add_tuple(table_name, all_args);
                 }
             }

ruby/generator/src/main.rs

@@ -233,15 +233,6 @@ fn convert_nodes(
                     });
                 }
 
-                // Finally, the type's defining table also includes the location.
-                main_table.columns.push(dbscheme::Column {
-                    unique: false,
-                    db_type: dbscheme::DbColumnType::Int,
-                    name: "loc",
-                    ql_type: ql::Type::At("location"),
-                    ql_type_is_ref: true,
-                });
-
                 entries.push(dbscheme::Entry::Table(main_table));
             }
             node_types::EntryKind::Token { .. } => {}
@@ -251,18 +242,24 @@ fn convert_nodes(
     (entries, ast_node_members, token_kinds)
 }
 
-/// Creates a dbscheme table entry representing the parent relation for AST nodes.
+/// Creates a dbscheme table specifying the parent node and location for each
+/// AST node.
 ///
 /// # Arguments
-/// - `name` - the name of both the table to create and the node parent type.
+/// - `name` - the name of the table to create.
+/// - `parent_name` - the name of the parent type.
 /// - `ast_node_name` - the name of the node child type.
-fn create_ast_node_parent_table<'a>(name: &'a str, ast_node_name: &'a str) -> dbscheme::Table<'a> {
+fn create_ast_node_info_table<'a>(
+    name: &'a str,
+    parent_name: &'a str,
+    ast_node_name: &'a str,
+) -> dbscheme::Table<'a> {
     dbscheme::Table {
         name,
         columns: vec![
             dbscheme::Column {
                 db_type: dbscheme::DbColumnType::Int,
-                name: "child",
+                name: "node",
                 unique: false,
                 ql_type: ql::Type::At(ast_node_name),
                 ql_type_is_ref: true,
@@ -271,7 +268,7 @@ fn create_ast_node_parent_table<'a>(name: &'a str, ast_node_name: &'a str) -> db
                 db_type: dbscheme::DbColumnType::Int,
                 name: "parent",
                 unique: false,
-                ql_type: ql::Type::At(name),
+                ql_type: ql::Type::At(parent_name),
                 ql_type_is_ref: true,
             },
             dbscheme::Column {
@@ -281,6 +278,13 @@ fn create_ast_node_parent_table<'a>(name: &'a str, ast_node_name: &'a str) -> db
                 ql_type: ql::Type::Int,
                 ql_type_is_ref: true,
             },
+            dbscheme::Column {
+                unique: false,
+                db_type: dbscheme::DbColumnType::Int,
+                name: "loc",
+                ql_type: ql::Type::At("location"),
+                ql_type_is_ref: true,
+            },
         ],
         keysets: Some(vec!["parent", "parent_index"]),
     }
@@ -312,13 +316,6 @@ fn create_tokeninfo<'a>(name: &'a str, type_name: &'a str) -> dbscheme::Table<'a
                 ql_type: ql::Type::String,
                 ql_type_is_ref: true,
             },
-            dbscheme::Column {
-                unique: false,
-                db_type: dbscheme::DbColumnType::Int,
-                name: "loc",
-                ql_type: ql::Type::At("location"),
-                ql_type_is_ref: true,
-            },
         ],
     }
 }
@@ -619,6 +616,7 @@ fn main() -> std::io::Result<()> {
     for language in languages {
         let prefix = node_types::to_snake_case(&language.name);
         let ast_node_name = format!("{}_ast_node", &prefix);
+        let node_info_table_name = format!("{}_ast_node_info", &prefix);
         let ast_node_parent_name = format!("{}_ast_node_parent", &prefix);
         let token_name = format!("{}_token", &prefix);
         let tokeninfo_name = format!("{}_tokeninfo", &prefix);
@@ -641,7 +639,8 @@ fn main() -> std::io::Result<()> {
                     name: &ast_node_parent_name,
                     members: [&ast_node_name, "file"].iter().cloned().collect(),
                 }),
-                dbscheme::Entry::Table(create_ast_node_parent_table(
+                dbscheme::Entry::Table(create_ast_node_info_table(
+                    &node_info_table_name,
                     &ast_node_parent_name,
                     &ast_node_name,
                 )),
@@ -651,7 +650,7 @@ fn main() -> std::io::Result<()> {
         let mut body = vec![
             ql::TopLevel::Class(ql_gen::create_ast_node_class(
                 &ast_node_name,
-                &ast_node_parent_name,
+                &node_info_table_name,
             )),
             ql::TopLevel::Class(ql_gen::create_token_class(&token_name, &tokeninfo_name)),
             ql::TopLevel::Class(ql_gen::create_reserved_word_class(&reserved_word_name)),