feat: exclude files from source tree

closes feature request #14.

Author: mufeedvh

README.md

@@ -97,6 +97,12 @@ Exclude files using glob patterns:
 code2prompt path/to/codebase --exclude="*.txt,*.md"
 ```
 
+Exclude files/folders from the source tree based on exclude patterns:
+
+```sh
+code2prompt path/to/codebase --exclude="*.npy,*.wav" --exclude-from-tree
+```
+
 Display the token count of the generated prompt:
 
 ```sh

src/main.rs

@@ -39,6 +39,10 @@ struct Cli {
     #[clap(long)]
     include_priority: bool,
 
+    /// Exclude files/folders from the source tree based on exclude patterns
+    #[clap(long)]
+    exclude_from_tree: bool,    
+
     /// Display the token count of the generated prompt
     #[clap(long)]
     tokens: bool,
@@ -97,6 +101,7 @@ fn main() -> Result<()> {
         args.include_priority,
         args.line_number,
         args.relative_paths,
+        args.exclude_from_tree,
     );
 
     let (tree, files) = match create_tree {

src/path.rs

@@ -30,6 +30,7 @@ pub fn traverse_directory(
     include_priority: bool,
     line_number: bool,
     relative_paths: bool,
+    exclude_from_tree: bool,
 ) -> Result<(String, Vec<serde_json::Value>)> {
     // ~~~ Initialization ~~~
     let mut files = Vec::new();
@@ -47,6 +48,12 @@ pub fn traverse_directory(
                 let mut current_tree = &mut root;
                 for component in relative_path.components() {
                     let component_str = component.as_os_str().to_string_lossy().to_string();
+
+                    // Check if the current component should be excluded from the tree
+                    if exclude_from_tree && !should_include_file(path, include, exclude, include_priority) {
+                        break;
+                    }
+
                     current_tree = if let Some(pos) = current_tree
                         .leaves
                         .iter_mut()