feat: better formatting for trees in scripting

src/cli/interactive/mod.rs

@@ -23,7 +23,7 @@ use tokio::sync::Mutex;
 
 use crate::{
     runner::scripting::ScriptingResponse,
-    utils::tree::{tree, TreeData, TreeNode},
+    utils::tree::{tree_data, tree_node, TreeData, TreeNode},
 };
 
 use super::opts::Opts;
@@ -61,8 +61,11 @@ pub async fn main_interactive(opts: Opts) -> Result<()> {
     let rl = Arc::new(Mutex::new(rl));
     let mut engine = Engine::new();
 
-    let tree_module = exported_module!(tree);
+    let tree_module = exported_module!(tree_node);
+    let tree_data_module = exported_module!(tree_data);
     engine.register_global_module(tree_module.into());
+    engine.register_global_module(tree_data_module.into());
+
     engine.build_type::<TreeData>();
     engine.build_type::<ScriptingResponse>();
     let engine = Arc::new(Mutex::new(engine));

src/runner/scripting.rs

@@ -2,7 +2,7 @@ use std::collections::BTreeMap;
 
 use crate::{
     cli::opts::Opts,
-    utils::tree::{tree, TreeData},
+    utils::tree::{tree_data, tree_node, TreeData},
 };
 use color_eyre::eyre::{eyre, Result};
 use colored::Colorize;
@@ -53,9 +53,12 @@ pub async fn run_scripts(
     progress: ProgressBar,
 ) -> Result<()> {
     let mut engine = Engine::new();
-    let tree_module = exported_module!(tree);
+    let tree_module = exported_module!(tree_node);
+    let tree_data_module = exported_module!(tree_data);
 
     engine.register_global_module(tree_module.into());
+    engine.register_global_module(tree_data_module.into());
+
     let mut root_scope = Scope::new();
     root_scope.push("data", data.clone());
     root_scope.push("opts", opts.clone());

src/utils/tree.rs

@@ -7,6 +7,7 @@ use rhai::plugin::*;
 use rhai::{CustomType, TypeBuilder};
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
+use std::fmt::Debug;
 use std::{collections::HashMap, sync::Arc};
 
 use crate::{
@@ -37,7 +38,7 @@ pub struct TreeData {
 }
 
 #[export_module]
-pub mod tree {
+pub mod tree_node {
     #[rhai_fn(get = "children")]
     pub fn children(data: &mut TreeNode<TreeData>) -> Dynamic {
         let mut children = Vec::new();
@@ -51,6 +52,36 @@ pub mod tree {
     pub fn data(data: &mut TreeNode<TreeData>) -> TreeData {
         data.data.clone()
     }
+
+    #[rhai_fn(global)]
+    pub fn to_string(data: &mut TreeNode<TreeData>) -> String {
+        format!(
+            "TreeNode {{ data: {}, children: [{}] }}",
+            tree_data::to_string(&mut data.data),
+            data.children.len()
+        )
+    }
+}
+
+#[export_module]
+pub mod tree_data {
+    #[rhai_fn(global)]
+    pub fn to_string(data: &mut TreeData) -> String {
+        format!(
+            "TreeData {{ url: {}, depth: {}, path: {}, status_code: {}, url_type: {:?}, extra: {:?} }}",
+            data.url,
+            data.depth,
+            data.status_code,
+            data.path,
+            match &data.url_type {
+                UrlType::Directory => "dir",
+                UrlType::File(ext) => ext,
+                UrlType::Unknown => "unknown",
+                UrlType::None => "",
+            },
+            data.extra
+        )
+    }
     #[rhai_fn(get = "response")]
     pub fn get_response(data: &mut TreeData) -> Dynamic {
         if let Some(response) = &data.response {