Implements #3.

Author: fabiomolinar

README.md

@@ -40,6 +40,10 @@ Use the `--no-headers` flag to remove the file path headers above each code bloc
 
 `git2prompt --no-headers rust-lang/rust-by-example`
 
+Sometimes you only need a single folder from a repository (instead of downloading the entire repo and ignoring most files). Use the `--folder` flag to restrict processing to a single directory:
+
+`git2prompt rust-lang/rust-by-example -f src`
+
 ## **Filtering**
 
 **git2prompt** automatically ignores certain common file types and directories to keep the output clean.

src/lib.rs

@@ -18,6 +18,7 @@ pub async fn process_github_urls(
     no_headers: bool,
     merge_files: bool,
     ignore_file: Option<PathBuf>,
+    folder: Option<String>,
 ) -> Result<Vec<PathBuf>, String> {
     println!(
         "Library received URLs: {:?}, no_headers: {}, merge_files: {}",
@@ -38,8 +39,9 @@ pub async fn process_github_urls(
         .map(|url| {
             let repository = Repository::new(&download_dir, url);
             let ignore_patterns = Arc::clone(&ignore_patterns);
+            let folder = folder.clone();
             tokio::spawn(async move {
-                process_single_repository(repository, no_headers, merge_files, ignore_patterns)
+                process_single_repository(repository, no_headers, merge_files, ignore_patterns, folder)
                     .await
             })
         })

src/main.rs

@@ -27,6 +27,10 @@ struct Args {
         default_value = ".git2promptignore"
     )]
     ignore_file: PathBuf,
+
+    /// Download and process only a specific folder within the repository
+    #[clap(short, long, value_name = "FOLDER PATH")]
+    folder: Option<String>,
 }
 
 #[tokio::main]
@@ -37,13 +41,17 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     println!("Repositories to process: {:?}", args.urls);
     println!("No file headers: {}", args.no_headers);
     println!("Merge into a single output file: {}", args.merge_files);
+    println!("Ignore file path: {:?}", args.ignore_file);
+    println!("Folder to process: {:?}", args.folder);
+    println!("----------------------------------------");
 
     // Call the library function to process the URLs, wrapping the ignore file path in Some
     match process_github_urls(
         args.urls,
         args.no_headers,
         args.merge_files,
         Some(args.ignore_file),
+        args.folder,
     )
     .await
     {

src/processing.rs

@@ -12,6 +12,7 @@ pub async fn process_single_repository(
     no_headers: bool,
     merge_files: bool,
     ignore_patterns: Arc<Vec<String>>,
+    folder: Option<String>,
 ) -> Result<Repository, String> {
     println!(
         "Preparing to clone {} to {:?}",
@@ -24,7 +25,7 @@ pub async fn process_single_repository(
     );
 
     let content =
-        process_repository_files(&repository.path, no_headers, merge_files, &ignore_patterns)
+        process_repository_files(&repository.path, no_headers, merge_files, &ignore_patterns, folder.as_deref())
             .await?;
     repository.content = Some(content);
 
@@ -37,10 +38,21 @@ pub async fn process_repository_files(
     no_headers: bool,
     merge_files: bool,
     ignore_patterns: &[String],
+    folder: Option<&str>,
 ) -> Result<String, String> {
     let mut combined_content = String::new();
 
-    for entry in WalkDir::new(repo_path).into_iter().filter_map(|e| e.ok()) {
+    let base_path = if let Some(folder) = folder {
+        repo_path.join(folder)
+    } else {
+        repo_path.to_path_buf()
+    };
+
+    if !base_path.exists() {
+        return Err(format!("Specified folder {:?} not found in repo", base_path));
+    }
+
+    for entry in WalkDir::new(&base_path).into_iter().filter_map(|e| e.ok()) {
         let path = entry.path();
 
         if is_valid_file(path, repo_path, ignore_patterns) {

tests/integration_tests.rs

@@ -53,15 +53,15 @@ async fn test_process_repository_files() -> Result<(), Box<dyn std::error::Error
     let src_main_path = PathBuf::from("src").join("main.rs");
     let readme_path = PathBuf::from("README.md");
 
-    let content_with_headers = process_repository_files(&test_repo_path, false, false, &Vec::new())
+    let content_with_headers = process_repository_files(&test_repo_path, false, false, &Vec::new(), None)
         .await
         .unwrap();
     assert!(content_with_headers.contains(&format!("## File: {}", src_main_path.display())));
     assert!(content_with_headers.contains("fn main() { println!(\"Hello\"); }"));
     assert!(content_with_headers.contains(&format!("## File: {}", readme_path.display())));
     assert!(content_with_headers.contains("# Test Repo"));
 
-    let content_no_headers = process_repository_files(&test_repo_path, true, false, &Vec::new())
+    let content_no_headers = process_repository_files(&test_repo_path, true, false, &Vec::new(), None)
         .await
         .unwrap();
     assert!(!content_no_headers.contains(&format!("## File: {}", src_main_path.display())));
@@ -70,7 +70,7 @@ async fn test_process_repository_files() -> Result<(), Box<dyn std::error::Error
     assert!(content_no_headers.contains("# Test Repo"));
 
     let content_with_headers_merged =
-        process_repository_files(&test_repo_path, false, true, &Vec::new())
+        process_repository_files(&test_repo_path, false, true, &Vec::new(), None)
             .await
             .unwrap();
     assert!(
@@ -81,7 +81,7 @@ async fn test_process_repository_files() -> Result<(), Box<dyn std::error::Error
     assert!(content_with_headers_merged.contains("# Test Repo"));
 
     let content_no_headers_merged =
-        process_repository_files(&test_repo_path, true, true, &Vec::new())
+        process_repository_files(&test_repo_path, true, true, &Vec::new(), None)
             .await
             .unwrap();
     assert!(!content_no_headers_merged.contains(&format!("### File: {}", src_main_path.display())));
@@ -126,7 +126,7 @@ async fn test_ignore_patterns_cross_platform() -> Result<(), Box<dyn std::error:
     ];
 
     let content =
-        processing::process_repository_files(&test_repo_path, true, true, &ignore_patterns).await?;
+        processing::process_repository_files(&test_repo_path, true, true, &ignore_patterns, None).await?;
 
     // Ignored files should not be in the output
     assert!(!content.contains("secret.txt"));