reposnap 0.7.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

reposnap/controllers/project_controller.py

@@ -19,7 +19,13 @@ class ProjectController:
             ]
             self.input_paths = []
             for p in input_paths:
-                candidate = (self.root_dir / p).resolve()
+                if p.is_absolute():
+                    # Handle absolute paths - use as-is but verify they're under root_dir
+                    candidate = p.resolve()
+                else:
+                    # Handle relative paths - join with root_dir
+                    candidate = (self.root_dir / p).resolve()
+
                 if candidate.exists():
                     try:
                         rel = candidate.relative_to(self.root_dir)
@@ -31,7 +37,7 @@ class ProjectController:
                         )
                 else:
                     self.logger.warning(
-                        f"Path {p} does not exist relative to repository root {self.root_dir}."
+                        f"Path {p} does not exist or is not under repository root {self.root_dir}."
                     )
             self.output_file: Path = (
                 Path(args.output).resolve()
@@ -48,6 +54,8 @@ class ProjectController:
                 args.exclude if hasattr(args, "exclude") else []
             )
             self.changes_only: bool = getattr(args, "changes", False)
+            self.contains: List[str] = getattr(args, "contains", [])
+            self.contains_case: bool = getattr(args, "contains_case", False)
         else:
             self.args = None
             self.input_paths = []
@@ -56,6 +64,8 @@ class ProjectController:
             self.include_patterns = []
             self.exclude_patterns = []
             self.changes_only = False
+            self.contains = []
+            self.contains_case = False
         self.file_tree: Optional[FileTree] = None
         self.gitignore_patterns: List[str] = []
         if self.root_dir:
@@ -110,6 +120,58 @@ class ProjectController:
             files = [f for f in files if not spec_exc.match_file(f.as_posix())]
         return files
 
+    def _apply_content_filter(self, files: List[Path]) -> List[Path]:
+        """
+        Filter files based on content substring matching.
+
+        Args:
+            files: List of relative file paths to filter
+
+        Returns:
+            Filtered list of files that contain at least one of the patterns
+            specified in self.contains. Returns original list if no patterns
+            are specified.
+
+        Note:
+            Uses case-insensitive matching by default unless self.contains_case
+            is True. Skips binary files and files larger than 5MB for performance.
+        """
+        if not self.contains:
+            return files
+
+        from reposnap.core.content_search import filter_files_by_content
+
+        initial_count = len(files)
+        ignore_case = not self.contains_case
+
+        self.logger.debug(
+            f"Applying content filter with patterns: {self.contains}, "
+            f"ignore_case: {ignore_case}"
+        )
+
+        # Convert relative paths to absolute for content search
+        absolute_paths = [self.root_dir / file_path for file_path in files]
+        filtered_absolute = filter_files_by_content(
+            absolute_paths, self.contains, ignore_case
+        )
+
+        # Convert back to relative paths
+        filtered_files = []
+        for abs_path in filtered_absolute:
+            try:
+                rel_path = abs_path.relative_to(self.root_dir)
+                filtered_files.append(rel_path)
+            except ValueError:
+                continue
+
+        kept_count = len(filtered_files)
+        self.logger.info(
+            f"Applied content filter (kept {kept_count} / {initial_count})"
+        )
+        self.logger.debug(f"Files kept after content filter: {filtered_files}")
+
+        return filtered_files
+
     def collect_file_tree(self) -> None:
         if self.changes_only:
             self.logger.info("Collecting uncommitted files from Git repository.")
@@ -147,6 +209,8 @@ class ProjectController:
                         continue
         all_files = self._apply_include_exclude(all_files)
         self.logger.debug(f"All files after applying include/exclude: {all_files}")
+        all_files = self._apply_content_filter(all_files)
+        self.logger.debug(f"All files after applying content filter: {all_files}")
         if self.input_paths:
             trees = []
             for input_path in self.input_paths:

reposnap/core/content_search.py

@@ -0,0 +1,104 @@
+# src/reposnap/core/content_search.py
+
+"""
+Private content search helpers for substring matching in files.
+
+This module provides stateless utility functions for searching file contents.
+It is intended for internal use by the project controller and should not be
+imported directly by external consumers.
+"""
+
+import logging
+from pathlib import Path
+from typing import List
+
+
+logger = logging.getLogger(__name__)
+
+# Configuration constants
+MAX_FILE_SIZE = 5 * 1024 * 1024  # 5 MiB
+BINARY_CHECK_SIZE = 1024  # First 1KB to check for binary content
+
+
+def file_matches(path: Path, patterns: List[str], ignore_case: bool = True) -> bool:
+    """
+    Check if a file contains any of the given patterns.
+
+    Args:
+        path: Path to the file to search
+        patterns: List of substring patterns to search for
+        ignore_case: Whether to perform case-insensitive matching (default: True)
+
+    Returns:
+        True if file contains any pattern, False otherwise
+
+    Note:
+        Uses streaming read with utf-8 encoding and error handling for binary files.
+        Returns False if file cannot be read as text or if file is too large/binary.
+    """
+    if not patterns:
+        return True
+
+    if not path.is_file():
+        return False
+
+    # Check file size - skip files larger than MAX_FILE_SIZE
+    try:
+        file_size = path.stat().st_size
+        if file_size > MAX_FILE_SIZE:
+            logger.debug(f"Skipping large file {path} ({file_size} bytes)")
+            return False
+    except OSError as e:
+        logger.debug(f"Could not stat file {path}: {e}")
+        return False
+
+    # Check for binary content in first KB
+    try:
+        with path.open("rb") as f:
+            first_chunk = f.read(BINARY_CHECK_SIZE)
+            if b"\0" in first_chunk:
+                logger.debug(f"Skipping binary file {path}")
+                return False
+    except Exception as e:
+        logger.debug(f"Could not read file {path} for binary check: {e}")
+        return False
+
+    # Pre-compute search patterns (case-normalized if needed)
+    search_patterns = [p.lower() if ignore_case else p for p in patterns]
+
+    try:
+        with path.open("r", encoding="utf-8", errors="ignore") as f:
+            for line in f:
+                search_line = line.lower() if ignore_case else line
+
+                if any(pattern in search_line for pattern in search_patterns):
+                    return True
+        return False
+    except Exception as e:
+        logger.debug(f"Could not read file {path} for content search: {e}")
+        return False
+
+
+def filter_files_by_content(
+    files: List[Path], patterns: List[str], ignore_case: bool = True
+) -> List[Path]:
+    """
+    Filter a list of files to only include those containing the given patterns.
+
+    Args:
+        files: List of file paths to filter
+        patterns: List of substring patterns to search for
+        ignore_case: Whether to perform case-insensitive matching (default: True)
+
+    Returns:
+        Filtered list of files that contain at least one pattern
+    """
+    if not patterns:
+        return files
+
+    matched_files = []
+    for file_path in files:
+        if file_matches(file_path, patterns, ignore_case):
+            matched_files.append(file_path)
+
+    return matched_files

reposnap/interfaces/cli.py

@@ -46,6 +46,18 @@ def main():
         action="store_true",
         help="Use only files that are added/modified/untracked/stashed but not yet committed.",
     )
+    parser.add_argument(
+        "-S",
+        "--contains",
+        nargs="+",
+        default=[],
+        help="Only include files whose contents contain these substrings",
+    )
+    parser.add_argument(
+        "--contains-case",
+        action="store_true",
+        help="Make --contains case-sensitive",
+    )
 
     args = parser.parse_args()
 

{reposnap-0.7.0.dist-info → reposnap-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reposnap
-Version: 0.7.0
+Version: 0.8.0
 Summary: Generate a Markdown file with all contents of your project
 Author: agoloborodko
 License-File: LICENSE
@@ -25,6 +25,7 @@ Description-Content-Type: text/markdown
 - **Structure Only Option**: The `--structure-only` flag can be used to generate the Markdown file with just the directory structure, omitting the contents of the files.
 - **Gitignore Support**: Automatically respects `.gitignore` patterns to exclude files and directories.
 - **Include and Exclude Patterns**: Use `--include` and `--exclude` to specify patterns for files and directories to include or exclude.
+- **Content Filtering**: Use `--contains` to filter files based on their content, including only files that contain specific substrings or code patterns.
 - **Changes Only Mode**: Use `-c` or `--changes` to snapshot only uncommitted files (staged, unstaged, untracked, and stashed changes).
 
 ## Installation
@@ -50,7 +51,7 @@ pip install -r requirements.lock
 To use `reposnap` from the command line, run it with the following options:
 
 ```bash
-reposnap [-h] [-o OUTPUT] [--structure-only] [--debug] [-i INCLUDE [INCLUDE ...]] [-e EXCLUDE [EXCLUDE ...]] [-c] paths [paths ...]
+reposnap [-h] [-o OUTPUT] [--structure-only] [--debug] [-i INCLUDE [INCLUDE ...]] [-e EXCLUDE [EXCLUDE ...]] [-c] [-S CONTAINS [CONTAINS ...]] [--contains-case] paths [paths ...]
 ```
 
 - `paths`: One or more paths (files or directories) within the repository whose content and structure should be rendered.
@@ -61,6 +62,8 @@ reposnap [-h] [-o OUTPUT] [--structure-only] [--debug] [-i INCLUDE [INCLUDE ...]
 - `-i, --include`: File/folder patterns to include. For example, `-i "*.py"` includes only Python files.
 - `-e, --exclude`: File/folder patterns to exclude. For example, `-e "*.md"` excludes all Markdown files.
 - `-c, --changes`: Use only files that are added/modified/untracked/stashed but not yet committed.
+- `-S, --contains`: Only include files whose contents contain these substrings. Multiple patterns can be specified.
+- `--contains-case`: Make `--contains` case-sensitive (default is case-insensitive).
 
 #### Pattern Matching
 
@@ -73,6 +76,41 @@ reposnap [-h] [-o OUTPUT] [--structure-only] [--debug] [-i INCLUDE [INCLUDE ...]
   - `-i "*.py"`: Includes only files ending with `.py`.
   - `-e "*.test.*"`: Excludes files with `.test.` in their names.
 
+#### Content Filtering
+
+The `--contains` (or `-S`) flag allows you to filter files based on their content, including only files that contain specific substrings. This is particularly useful for focusing on files that contain certain code patterns, imports, or keywords.
+
+- **Case Sensitivity**: By default, content matching is case-insensitive. Use the `--contains-case` flag to enable case-sensitive matching.
+- **Multiple Patterns**: You can specify multiple patterns, and files containing **any** of the patterns will be included (OR logic).
+- **Performance**: Large files (>5MB) and binary files are automatically skipped for performance and safety reasons.
+
+**Examples**:
+
+1. **Find files containing specific imports**:
+    ```bash
+    reposnap . -S "import logging"
+    ```
+
+2. **Search for multiple patterns (OR logic)**:
+    ```bash
+    reposnap . -S "TODO" "FIXME" "import requests"
+    ```
+
+3. **Case-sensitive content search**:
+    ```bash
+    reposnap . -S "TODO" --contains-case
+    ```
+
+4. **Combine content filtering with other filters**:
+    ```bash
+    reposnap . -S "class " -i "*.py" --structure-only
+    ```
+
+5. **Find files with specific function calls**:
+    ```bash
+    reposnap . -S "logger.error" "raise Exception"
+    ```
+
 #### Only Snapshot Your Current Work
 
 The `-c` or `--changes` flag allows you to generate documentation for only the files that have been modified but not yet committed. This includes:
@@ -147,6 +185,18 @@ This is particularly useful when you want to:
     reposnap . -c
     ```
 
+7. **Find and document files containing specific code patterns**:
+
+    ```bash
+    reposnap . -S "import logging" "logger."
+    ```
+
+8. **Combine content filtering with file type filtering**:
+
+    ```bash
+    reposnap . -S "class " -i "*.py" --structure-only
+    ```
+
 ### Graphical User Interface
 
 `reposnap` also provides a GUI for users who prefer an interactive interface.

{reposnap-0.7.0.dist-info → reposnap-0.8.0.dist-info}/RECORD

@@ -1,19 +1,20 @@
 reposnap/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reposnap/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reposnap/controllers/project_controller.py,sha256=RROOcb_FiEhFc9oshptPL8moH-5lkUrio7Lp0MqBqp0,10599
+reposnap/controllers/project_controller.py,sha256=Kg_Vu7gkWRj5EDdiOwoGlsmaZ1XUsBc_XHIV5yJG4w8,13007
 reposnap/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+reposnap/core/content_search.py,sha256=kPiW5edDpNWgDwQg2GbkBJNweAyq7DHUZgtvajTZHK8,3262
 reposnap/core/file_system.py,sha256=82gwvmgrsWf63paMrIz-Z0eqIjbqt9_-vujdXlJJoFE,1074
 reposnap/core/git_repo.py,sha256=YVIbx-Y_MUbnn5Z4E2XBTJbG7Kawx5aUX2tg6vnocd0,4284
 reposnap/core/markdown_generator.py,sha256=V6uEbxVSbCbxKN9ysTDKsIDvEGBxFutpOpyaZRXZUGw,3747
 reposnap/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-reposnap/interfaces/cli.py,sha256=gL0gauEt_AkuRRr-p5YAeHeUPgvZ59lpZMqsopLjHas,1661
+reposnap/interfaces/cli.py,sha256=gmO49_DeNiqDsCtKxMEgZ8T-UjxDqwZ9m0LR_JsES5Q,1976
 reposnap/interfaces/gui.py,sha256=sTuQxjD1nPa9FpgfzOwi6VDO5QMMtDX-5CiEhbJJcs4,5429
 reposnap/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reposnap/models/file_tree.py,sha256=jGo_SizdFcOiDC1OOMz-tiijRN3iSD7ENh6Xw8S6OL0,3362
 reposnap/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reposnap/utils/path_utils.py,sha256=UrMe5cjspTf-4gjg2lzv6BgLwZ7S_1lLECQvDMDZO9Y,507
-reposnap-0.7.0.dist-info/METADATA,sha256=oJF5qQGPWc6aG2mnupOpiDrbJ1hWbApjDCP0tM-lB2Y,6768
-reposnap-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reposnap-0.7.0.dist-info/entry_points.txt,sha256=o3GyO7bpR0dujPCjsvvZMPv4pXNJlFwD49_pA1r5FOA,102
-reposnap-0.7.0.dist-info/licenses/LICENSE,sha256=Aj7WCYBXi98pvi723HPn4GDRyjxToNWb3PC6j1_lnPk,1069
-reposnap-0.7.0.dist-info/RECORD,,
+reposnap-0.8.0.dist-info/METADATA,sha256=dRmQAHOFy1Q41neUvmC8YAwxRbWFgxQ7Z4MQFiKu6mI,8687
+reposnap-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reposnap-0.8.0.dist-info/entry_points.txt,sha256=o3GyO7bpR0dujPCjsvvZMPv4pXNJlFwD49_pA1r5FOA,102
+reposnap-0.8.0.dist-info/licenses/LICENSE,sha256=Aj7WCYBXi98pvi723HPn4GDRyjxToNWb3PC6j1_lnPk,1069
+reposnap-0.8.0.dist-info/RECORD,,