Add depth flag to NFT CLI

Author: mischnic

turbopack/crates/turbopack-nft/src/main.rs

@@ -31,6 +31,9 @@ pub struct Arguments {
 
     #[clap(long)]
     pub show_issues: bool,
+
+    #[clap(long)]
+    pub depth: Option<usize>,
 }
 
 #[global_allocator]
@@ -95,6 +98,7 @@ async fn main_inner(args: Arguments) -> Result<()> {
             args.entry.into(),
             args.graph,
             args.show_issues,
+            args.depth,
         )
         .await?;
         Ok(Default::default())

turbopack/crates/turbopack-nft/src/nft.rs

@@ -1,4 +1,4 @@
-use std::{collections::HashSet, env::current_dir, path::PathBuf};
+use std::{collections::HashSet, env::current_dir, path::PathBuf, usize};
 
 use anyhow::Result;
 use turbo_rcstr::{RcStr, rcstr};
@@ -28,8 +28,9 @@ pub async fn node_file_trace(
     input: RcStr,
     graph: bool,
     show_issues: bool,
+    max_depth: Option<usize>,
 ) -> Result<()> {
-    let op = node_file_trace_operation(project_root.clone(), input.clone(), graph);
+    let op = node_file_trace_operation(project_root.clone(), input.clone(), graph, max_depth);
     let result = op.resolve_strongly_consistent().await?;
 
     if show_issues {
@@ -58,6 +59,7 @@ async fn node_file_trace_operation(
     project_root: RcStr,
     input: RcStr,
     graph: bool,
+    max_depth: Option<usize>,
 ) -> Result<Vc<Vec<RcStr>>> {
     let workspace_fs: Vc<Box<dyn FileSystem>> = Vc::upcast(DiskFileSystem::new(
         rcstr!("workspace"),
@@ -98,7 +100,7 @@ async fn node_file_trace_operation(
             ..Default::default()
         }
         .cell(),
-        Layer::new(rcstr!("test")),
+        Layer::new(rcstr!("externals-tracing")),
     );
     let module = module_asset_context
         .process(Vc::upcast(source), ReferenceType::Undefined)
@@ -107,7 +109,7 @@ async fn node_file_trace_operation(
     let asset = TracedAsset::new(module).to_resolved().await?;
 
     Ok(Vc::cell(if graph {
-        to_graph(ResolvedVc::upcast(asset)).await?
+        to_graph(ResolvedVc::upcast(asset), max_depth.unwrap_or(usize::MAX)).await?
     } else {
         to_list(ResolvedVc::upcast(asset)).await?
     }))
@@ -126,7 +128,7 @@ async fn to_list(asset: ResolvedVc<Box<dyn OutputAsset>>) -> Result<Vec<RcStr>>
     Ok(assets)
 }
 
-async fn to_graph(asset: ResolvedVc<Box<dyn OutputAsset>>) -> Result<Vec<RcStr>> {
+async fn to_graph(asset: ResolvedVc<Box<dyn OutputAsset>>, max_depth: usize) -> Result<Vec<RcStr>> {
     let mut visited = HashSet::new();
     let mut queue = Vec::new();
     queue.push((0, asset));
@@ -142,12 +144,19 @@ async fn to_graph(asset: ResolvedVc<Box<dyn OutputAsset>>) -> Result<Vec<RcStr>>
             for &asset in references.iter().rev() {
                 queue.push((depth + 1, asset));
             }
-            result.push(format!("{}{}", indent, asset.path().to_string().await?).into());
+            if depth <= max_depth {
+                result.push(format!("{}{}", indent, asset.path().to_string().await?).into());
+            }
         } else if references.is_empty() {
-            result.push(format!("{}{} *", indent, asset.path().to_string().await?).into());
-        } else {
+            if depth <= max_depth {
+                result.push(format!("{}{} *", indent, asset.path().to_string().await?).into());
+            }
+        } else if depth <= max_depth {
             result.push(format!("{}{} *...", indent, asset.path().to_string().await?).into());
         }
     }
+    result.push("".into());
+    result.push("*    : revisited".into());
+    result.push("*... : revisited and references were already printed".into());
     Ok(result)
 }