Merge pull request #3 from p-x9/feature/file-tree-depth

Add `--depth` option to `file tree` command
p-x9 · Sep 11, 2023 · 9ee6b50 · 9ee6b50
2 parents b29f7f8 + 71f1d9f
commit 9ee6b50
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -162,18 +162,25 @@ app info
 (lldb) file tree -h
 usage:  tree
        [-h]
+       [-b]
        [-l]
        [--documents]
-       [--tmp TMP]
+       [--tmp]
+       [--depth DEPTH]
+
 optional arguments:
   -h, --help
     show this help message and exit
+  -b, --bundle
+    bundle directory (default: False)
   -l, --library
     library directory (default: False)
   --documents
     documents directory (default: False)
-  --tmp TMP
-    tmp directory (default: None)
+  --tmp
+    tmp directory (default: False)
+  --depth DEPTH
+    Maximum depth to be displayed (default: None)
 ```
 
 #### Example

diff --git a/src/file.py b/src/file.py
@@ -41,6 +41,7 @@ def parse_args(args: list[str]) -> argparse.Namespace:
     tree_command.add_argument("-l", "--library", action="store_true", help="library directory")
     tree_command.add_argument("--documents", action="store_true", help="documents directory")
     tree_command.add_argument("--tmp", action="store_true", help="tmp directory")
+    tree_command.add_argument("--depth", type=int, help="Maximum depth to be displayed")
 
     open_command = subparsers.add_parser("open",
                                          help="Open directory with Finder (Simulator Only)",
@@ -68,16 +69,24 @@ def tree(args: argparse.Namespace, debugger: lldb.SBDebugger, result: lldb.SBCom
     script_ret = subprocess.run(f"cat {dir_name}/swift/file.swift", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
 
     script = script_ret.stdout
+
+    depth = 'nil'
+    if args.depth is not None:
+        try:
+            depth = str(int(args.depth))
+        except ValueError:
+            pass
+
     if args.bundle:
-        script += "listFilesInDirectory(Bundle.main.bundleURL)"
+        script += f"listFilesInDirectory(Bundle.main.bundleURL, depth: {depth})"
     elif args.library:
-        script += "listFilesInDirectory(FileManager.default.urls(for: .libraryDirectory, in: .userDomainMask).first!)"
+        script += f"listFilesInDirectory(FileManager.default.urls(for: .libraryDirectory, in: .userDomainMask).first!, depth: {depth})"
     elif args.documents:
-        script += "listFilesInDirectory(FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!)"
+        script += f"listFilesInDirectory(FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first!, depth: {depth})"
     elif args.tmp:
-        script += "listFilesInDirectory(FileManager.default.temporaryDirectory)"
+        script += f"listFilesInDirectory(FileManager.default.temporaryDirectory, depth: {depth})"
     elif args.path:
-        script += f"listFilesInDirectory(URL(fileURLWithPath: {args.path}))"
+        script += f"listFilesInDirectory(URL(fileURLWithPath: {args.path}), depth: {depth})"
 
     _ = util.exp_script(
         debugger,

diff --git a/src/swift/file.swift b/src/swift/file.swift
@@ -1,11 +1,16 @@
 import Foundation
 
-func listFilesInDirectory(_ directoryURL: URL, indentation: String = "") {
+func listFilesInDirectory(_ directoryURL: URL, indentation: String = "", depth: Int? = nil) {
     if indentation.isEmpty {
         let isDirectory = (try? directoryURL.resourceValues(forKeys: [.isDirectoryKey]))?.isDirectory == true
         print("\(isDirectory ? "📁 " : "📄 ")" + directoryURL.lastPathComponent)
     }
 
+    let currentDepth = indentation.replacingOccurrences(of: "│", with: " ").count / 3
+    if let depth, currentDepth >= depth {
+        return
+    }
+
     do {
         let fileManager = FileManager.default
         let contents = try fileManager.contentsOfDirectory(at: directoryURL, includingPropertiesForKeys: nil, options: [])
@@ -20,7 +25,7 @@ func listFilesInDirectory(_ directoryURL: URL, indentation: String = "") {
                 fileHierarchy += "📁 \(url.lastPathComponent)"
                 print(fileHierarchy)
 
-                listFilesInDirectory(url, indentation: indentation + (isLast ? "   " : "│  "))
+                listFilesInDirectory(url, indentation: indentation + (isLast ? "   " : "│  "), depth: depth)
             } else {
                 fileHierarchy += "📄 \(url.lastPathComponent)"
                 print(fileHierarchy)