Add max depth configuration (#23)

Author: João Monte

README.md

@@ -15,6 +15,9 @@ vscode extension for generate call graph in [graphviz dot language](https://www.
 4. Click `save dot file` or `save as svg` in the bottom left corner to save the graph
 5. Add `.callgraphignore` file in your project root directory to ignore some files or folders in workspace (the syntax is the same as `.gitignore`)
 
+## Extra configuration
+- **Max depth**: You can set the max depth at the `Call-graph: Max Depth` setting in the editor settings session. Note that it applies for both incoming and outgoing call graphs.
+
 ## How it works
 It depends `vscode.provideOutgoingCalls` and `vscode.provideIncomingCalls` built-in commands( the same with `Show Call Hierarchy` command, not available for some language server ).
 

package.json

@@ -50,6 +50,11 @@
                     "type": "string",
                     "default": "${workspace}/.callgraphignore",
                     "description": "the file that specific paths should ignore"
+                },
+                "call-graph.maxDepth": {
+                    "type": "number",
+                    "default": 0,
+                    "description": "the graph max depth. Set to 0 for unlimited depth"
                 }
             }
         }

src/call.ts

@@ -12,11 +12,15 @@ async function getCallNode(
     ignore: (item: vscode.CallHierarchyItem) => boolean,
     outgoing: Boolean = true
 ) {
+    const maxDepth = vscode.workspace
+        .getConfiguration()
+        .get<number>('call-graph.maxDepth') || 0
     const command = outgoing
         ? 'vscode.provideOutgoingCalls'
         : 'vscode.provideIncomingCalls'
     const nodes = new Set<CallHierarchyNode>()
-    const insertNode = async (node: CallHierarchyNode) => {
+    const insertNode = async (node: CallHierarchyNode, depth=0) => {
+        if (maxDepth > 0 && depth >= maxDepth) return
         output.appendLine('resolve: ' + node.item.name)
         nodes.add(node)
         const calls:
@@ -45,7 +49,7 @@ async function getCallNode(
             if (isSkip) continue
             const child = { item: next, children: [] }
             node.children.push(child)
-            await insertNode(child)
+            await insertNode(child, depth + 1)
         }
     }
     const graph = { item: entryItem, children: [] as CallHierarchyNode[] }