ignore gitignore files content

repo_to_text/main.py

@@ -1,12 +1,37 @@
 import os
 import subprocess
+import pathspec
 
-def get_tree_structure(path='.') -> str:
+def get_tree_structure(path='.', gitignore_spec=None) -> str:
+    # Run the tree command and get its output
     result = subprocess.run(['tree', path], stdout=subprocess.PIPE)
-    return result.stdout.decode('utf-8')
+    tree_output = result.stdout.decode('utf-8')
+
+    if not gitignore_spec:
+        return tree_output
+
+    # Filter the tree output to exclude files in .gitignore (excluding .gitignore itself)
+    filtered_lines = []
+    for line in tree_output.splitlines():
+        parts = line.split()
+        if not any(gitignore_spec.match_file(os.path.join(path, part)) for part in parts if part != '.gitignore'):
+            filtered_lines.append(line)
+    return '\n'.join(filtered_lines)
+
+def load_gitignore(path='.'):
+    gitignore_path = os.path.join(path, '.gitignore')
+    if os.path.exists(gitignore_path):
+        with open(gitignore_path, 'r') as f:
+            return pathspec.PathSpec.from_lines('gitwildmatch', f)
+    return None
+
+def is_ignored_file(file_path: str) -> bool:
+    # Check if the file is part of .git or should be ignored according to .gitignore rules
+    return '.git' in file_path.split(os.sep) or file_path.endswith('.gitignore')
 
 def save_repo_to_text(path='.') -> None:
-    tree_structure: str = get_tree_structure(path)
+    gitignore_spec = load_gitignore(path)
+    tree_structure: str = get_tree_structure(path, gitignore_spec)
     with open('repo_structure.txt', 'w') as file:
         file.write(tree_structure + '\n')
 
@@ -14,6 +39,11 @@ def save_repo_to_text(path='.') -> None:
             for filename in files:
                 file_path: str = os.path.join(root, filename)
                 relative_path: str = os.path.relpath(file_path, path)
+                
+                # Check if the file should be ignored
+                if is_ignored_file(file_path) or (gitignore_spec and gitignore_spec.match_file(file_path)):
+                    continue
+                
                 file.write(f'\n{relative_path}\n')
                 file.write('```\n')
                 try:
@@ -27,4 +57,4 @@ def main() -> None:
     save_repo_to_text()
 
 if __name__ == '__main__':
-    main()
+    main()