feat: add branch option to CLI for cloning and ingesting specific branches

Author: AbhiRam162105

src/gitingest/cli.py

@@ -18,13 +18,15 @@
 @click.option("--exclude-pattern", "-e", multiple=True, help="Patterns to exclude (space-separated patterns allowed)")
 @click.option("--include-pattern", "-i", multiple=True, help="Patterns to include (space-separated patterns allowed)")
 @click.option("--ignore-file", default=".gitingestignore", help="Path to ignore file (default: .gitingestignore)")
+@click.option("--branch", "-b", default=None, help="Branch to clone and ingest")
 def main(
     source: str,
     output: str | None,
     max_size: int,
     exclude_pattern: tuple[str, ...],
     include_pattern: tuple[str, ...],
     ignore_file: str,
+    branch: str | None,
 ):
     """
     Main entry point for the CLI.
@@ -44,8 +46,10 @@ def main(
         A tuple of patterns to include during the analysis. Only files matching these patterns will be processed.
     ignore_file : str
         Path to the ignore file containing additional patterns to exclude.
+    branch : str | None
+        The branch to clone (optional).
     """
-    asyncio.run(async_main(source, output, max_size, exclude_pattern, include_pattern, ignore_file))
+    asyncio.run(async_main(source, output, max_size, exclude_pattern, include_pattern, ignore_file, branch))
 
 
 async def async_main(
@@ -55,6 +59,7 @@ async def async_main(
     exclude_pattern: tuple[str, ...],
     include_pattern: tuple[str, ...],
     ignore_file: str,
+    branch: str | None,
 ) -> None:
     """
     Analyze a directory or repository and create a text dump of its contents.
@@ -78,6 +83,8 @@ async def async_main(
         A tuple of patterns to include during the analysis. Only files matching these patterns will be processed.
     ignore_file : str
         Path to the ignore file containing additional patterns to exclude.
+    branch : str | None
+        The branch to clone (optional).
 
     Raises
     ------
@@ -101,8 +108,8 @@ async def async_main(
         ignore_patterns = parse_ignore_file(ignore_file_path)
         exclude_patterns.update(ignore_patterns)
 
-        # Perform the ingest operation
-        summary, *_ = await ingest(source, max_size, include_patterns, exclude_patterns, output=output)
+        # Perform the ingest operation with branch support
+        summary, *_ = await ingest(source, max_size, include_patterns, exclude_patterns, branch=branch, output=output)
 
         # Display results
         click.echo(f"Analysis complete! Output written to: {output}")

src/gitingest/repository_ingest.py

@@ -16,6 +16,7 @@ async def ingest(
     max_file_size: int = 10 * 1024 * 1024,  # 10 MB
     include_patterns: set[str] | str | None = None,
     exclude_patterns: set[str] | str | None = None,
+    branch: str | None = None,
     output: str | None = None,
 ) -> tuple[str, str, str]:
     """
@@ -36,6 +37,8 @@ async def ingest(
         Pattern or set of patterns specifying which files to include. If `None`, all files are included.
     exclude_patterns : set[str] | str | None, optional
         Pattern or set of patterns specifying which files to exclude. If `None`, no files are excluded.
+    branch : str | None, optional
+        The branch to clone and ingest. If `None`, the default branch is used.
     output : str | None, optional
         File path where the summary and content should be written. If `None`, the results are not written to a file.
 
@@ -62,6 +65,10 @@ async def ingest(
         )
 
         if parsed_query.url:
+            # Override branch if specified
+            if branch is not None:
+                parsed_query.branch = branch
+
             # Extract relevant fields for CloneConfig
             clone_config = CloneConfig(
                 url=parsed_query.url,