Access root and jump-to nodes from file's in Lua

Author: dcreager

stack-graphs/src/lua.rs

@@ -139,6 +139,14 @@
 //!
 //! Adds a new internal scope node to this file.
 //!
+//! #### `jump_to_node`
+//!
+//! ``` lua
+//! local node = file:jump_to_node()
+//! ```
+//!
+//! Returns the root node of the graph containing this file.
+//!
 //! #### `pop_scoped_symbol_node`
 //!
 //! ``` lua
@@ -185,6 +193,14 @@
 //! Adds a new definition node to this file.  `symbol` must be a string, or an instance that can be
 //! converted to a string via its `tostring` method.
 //!
+//! #### `root_node`
+//!
+//! ``` lua
+//! local node = file:root_node()
+//! ```
+//!
+//! Returns the root node of the graph containing this file.
+//!
 //! #### `scoped_definition_node`
 //!
 //! ``` lua
@@ -494,6 +510,14 @@ impl UserData for Handle<File> {
             Ok(node_ud)
         });
 
+        methods.add_function("jump_to_node", |l, file_ud: AnyUserData| {
+            let graph_ud = file_ud.user_value::<AnyUserData>()?;
+            let node = StackGraph::jump_to_node();
+            let node_ud = l.create_userdata(node)?;
+            node_ud.set_user_value(graph_ud)?;
+            Ok(node_ud)
+        });
+
         methods.add_function(
             "pop_scoped_symbol_node",
             |l, (file_ud, symbol): (AnyUserData, String)| {
@@ -599,6 +623,14 @@ impl UserData for Handle<File> {
             },
         );
 
+        methods.add_function("root_node", |l, file_ud: AnyUserData| {
+            let graph_ud = file_ud.user_value::<AnyUserData>()?;
+            let node = StackGraph::root_node();
+            let node_ud = l.create_userdata(node)?;
+            node_ud.set_user_value(graph_ud)?;
+            Ok(node_ud)
+        });
+
         methods.add_function(
             "scoped_definition_node",
             |l, (file_ud, symbol): (AnyUserData, String)| {

stack-graphs/tests/it/lua.rs

@@ -258,6 +258,8 @@ fn can_create_all_node_types_from_lua() -> Result<(), anyhow::Error> {
               local root = graph:root_node()
               local jump_to = graph:jump_to_node()
               local file = graph:file("test.py")
+              local file_root = file:root_node()
+              local file_jump_to = file:jump_to_node()
               local drop_scopes = file:drop_scopes_node()
               local exported = file:exported_scope_node()
               local internal = file:internal_scope_node()