janito 0.6.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

janito/change/validator.py

@@ -1,251 +1,87 @@
-import ast
-# ...existing code...
-# Remove the process_change_request function if it exists in this file
-# Keep all other existing code
-from pathlib import Path
-from typing import Optional, Tuple, Optional, List, Set
-from rich.console import Console
-from rich.prompt import Confirm
-from rich.panel import Panel
-from rich.columns import Columns
-from rich import box
-
-from janito.common import progress_send_message
-from janito.change.history import save_changes_to_history
-from janito.config import config
-from janito.workspace.scan import collect_files_content
-from .viewer import preview_all_changes
-from janito.workspace.analysis import analyze_workspace_content as show_content_stats
-from .parser import FileChange
-
-from .analysis import analyze_request
-
-def validate_file_operations(changes: List[FileChange], collected_files: Set[Path]) -> Tuple[bool, str]:
-    """Validate file operations against current filesystem state.
-
-    Validates:
-    - Path conflicts:
-        - Parent directory exists and is a directory
-        - No Python module name conflicts (dir vs file)
-    - Text modifications:
-        - Search content required for non-append modifications
-        - Replace content required for append operations
-
-    Args:
-        changes: List of file changes to validate
-        collected_files: Set of files that were found during scanning
-
-    Returns:
-        Tuple[bool, str]: (is_valid, error_message)
-    """
-    for change in changes:
-        # For modify operations, validate text changes
-        if change.operation == ChangeOperation.MODIFY_FILE:
-            for mod in change.text_changes:
-                if not mod.is_delete and not mod.is_append:
-                    if not mod.search_content:
-                        return False, f"Search content required for modification in {change.name}"
-                
-                if mod.is_append and not mod.replace_content:
-                    return False, f"Replace content required for append operation in {change.name}"
-
-        # Check for directory/file conflicts
-        if change.operation == ChangeOperation.CREATE_FILE:
-            parent = change.name.parent
-            if parent.exists() and not parent.is_dir():
-                return False, f"Cannot create file - parent path exists as file: {parent}"
-
-            # Check for Python module conflicts 
-            if change.name.suffix == '.py':
-                module_dir = change.name.with_suffix('')
-                if module_dir.exists() and module_dir.is_dir():
-                    return False, f"Cannot create Python file - directory with same name exists: {module_dir}"
-
-        # Basic rename validation (without existence checks)
-        if change.operation == ChangeOperation.RENAME_FILE:
-            if not change.source or not change.target:
-                return False, "Rename operation requires both source and target paths"
-
-    return True, ""
-from pathlib import Path
-from typing import Tuple, List, Set, Optional
-from .parser import FileChange, ChangeOperation
-
-def validate_python_syntax(code: str, filepath: Path | str) -> Tuple[bool, str]:
-    """Validate Python code syntax using ast parser.
-    
-    Args:
-        code: Python source code to validate
-        filepath: Path or string of the file (used for error messages)
-        
-    Returns:
-        Tuple of (is_valid, error_message)
-        - is_valid: True if syntax is valid
-        - error_message: Empty string if valid, error details if invalid
-    """
-    try:
-        ast.parse(code)
-        return True, ""
-    except SyntaxError as e:
-        # Get detailed error information
-        line_num = e.lineno if e.lineno is not None else 0
-        col_num = e.offset if e.offset is not None else 0
-        line = e.text or ""
-        
-        # Build error message with line pointer
-        pointer = " " * (col_num - 1) + "^" if col_num > 0 else ""
-        error_msg = (
-            f"Syntax error at {filepath}:{line_num}:{col_num}\n"
-            f"{line}\n"
-            f"{pointer}\n"
-            f"Error: {str(e)}"
-        )
-        return False, error_msg
-    except Exception as e:
-        return False, f"Parsing error in {filepath}: {str(e)}"
-
-def validate_change(change: FileChange) -> Tuple[bool, Optional[str]]:
-    """Validate a single FileChange object for structural correctness.
-    
-    Validates:
-    - Required fields (name, operation type)
-    - Operation-specific requirements:
-        - Create/Replace: content is required
-        - Rename: target path is required
-        - Modify: at least one text change required
-    - Text change validations:
-        - Append: replace_content is required
-        - Delete: search_content is required
-        - Replace: both search_content and replace_content required
-        - Prevents duplicate search patterns
-
-    Args:
-        change: FileChange object to validate
-
-    Returns:
-        Tuple[bool, Optional[str]]: (is_valid, error_message)
-    """
-    if not change.name:
-        return False, "File name is required"
-
-    operation = change.operation.name.title().lower()
-    if operation not in ['create_file', 'replace_file', 'remove_file', 'rename_file', 'modify_file', 'move_file']:
-        return False, f"Invalid file operation: {change.operation}"
-
-    if operation in ['rename_file', 'move_file'] and not change.target:
-        return False, "Target file path is required for rename/move operation"
-
-    if operation in ['create_file', 'replace_file']:
-        if not change.content:
-            return False, f"Content is required for {change.operation} operation"
-
-    if operation == 'modify_file':
-        if not change.text_changes:
-            return False, "At least one modification is required for modify operation"
-            
-        # Track search texts to avoid duplicates
-        seen_search_texts = set()
-        for mod in change.text_changes:
-            # Validate append operations
-            if mod.is_append:
-                if not mod.replace_content:
-                    return False, "Replace content required for append operation"
-            # Validate other operations
-            elif not mod.is_delete:
-                if not mod.search_content:
-                    return False, "Search content required for non-append modification"
-                    
-            if mod.search_content:
-                seen_search_texts.add(mod.search_content)
-
-    return True, None
-
-def validate_all_changes(changes: List[FileChange], collected_files: Set[Path]) -> Tuple[bool, Optional[str]]:
-    """Validates all aspects of the requested changes.
-
-    Performs complete validation in two phases:
-    1. Individual change validation:
-        - Structure and content requirements
-        - Operation-specific validations
-        - Text modification validations
-    2. Filesystem state validation:
-        - File existence checks
-        - Path conflict checks
-        - Python module conflict checks
-
-    Args:
-        changes: List of changes to validate
-        collected_files: Set of files found during scanning
-
-    Returns:
-        Tuple[bool, Optional[str]]: (is_valid, error_message)
-        - If valid, error_message will be None
-        - If invalid, error_message will describe the validation failure
-    """
-    # First validate individual changes
-    for change in changes:
-        is_valid, error = validate_change(change)
-        if not is_valid:
-            return False, f"Invalid change for {change.name}: {error}"
-    
-    # Then validate file operations against filesystem
-    is_valid, error = validate_file_operations(changes, collected_files)
-    if not is_valid:
-        return False, error
-        
-    return True, None
-
-def validate_file_operations(changes: List[FileChange], collected_files: Set[Path]) -> Tuple[bool, str]:
-    """Validate file operations against current filesystem state.
-
-    Validates:
-    - File existence for operations that require it:
-        - Modify: file must exist in collected files
-        - Replace: file must exist in collected files
-        - Remove: file must exist in collected files
-    - File non-existence for operations that require it:
-        - Create: file must not exist (unless marked as new)
-        - Rename target: target must not exist
-    - Path conflicts:
-        - Parent directory exists and is a directory
-        - No Python module name conflicts (dir vs file)
-    - Text modifications:
-        - Search content required for non-append modifications
-        - Replace content required for append operations
-
-    Args:
-        changes: List of file changes to validate
-        collected_files: Set of files that were found during scanning, includes state metadata
-
-    Returns:
-        Tuple[bool, str]: (is_valid, error_message)
-    """
-    for change in changes:
-        # For modify operations, validate text changes
-        if change.operation == ChangeOperation.MODIFY_FILE:
-            for mod in change.text_changes:
-                if not mod.is_delete and not mod.is_append:
-                    if not mod.search_content:
-                        return False, f"Search content required for modification in {change.name}"
-                
-                if mod.is_append and not mod.replace_content:
-                    return False, f"Replace content required for append operation in {change.name}"
-
-        # Get file state if available
-        file_state = change.name.name.split(' (')[-1].rstrip(')') if ' (' in str(change.name) else None
-        is_new_file = file_state == 'new' if file_state else False
-
-        # Validate file exists for operations requiring it
-        if change.operation in (ChangeOperation.MODIFY_FILE, ChangeOperation.REPLACE_FILE, ChangeOperation.REMOVE_FILE):
-            if change.name not in collected_files and not is_new_file:
-                return False, f"File not found in scanned files: {change.name}"
-
-
-        # Validate rename/move operations
-        if change.operation in (ChangeOperation.RENAME_FILE, ChangeOperation.MOVE_FILE):
-            if not change.source or not change.target:
-                return False, "Rename/move operation requires both source and target paths"
-            if change.source not in collected_files and not is_new_file:
-                return False, f"Source file not found for rename/move: {change.source}"
-
-    return True, ""
+from pathlib import Path
+from typing import Set
+import ast
+import yaml
+import subprocess
+import sys
+import os
+
+class Validator:
+    def __init__(self, preview_dir: Path):
+        self.preview_dir = preview_dir
+        self.validated_files: Set[Path] = set()
+
+    def validate_python_syntax(self, filepath: Path):
+        """Validate Python file syntax using ast."""
+        try:
+            with open(filepath, 'r', encoding="utf-8") as file:
+                content = file.read()
+            ast.parse(content, filename=str(filepath))
+        except SyntaxError as e:
+            raise ValueError(f"Python syntax error in {filepath}: {e}")
+        except Exception as e:
+            raise ValueError(f"Error validating {filepath}: {e}")
+
+    def validate_files(self, files: Set[Path]):
+        """Validate all modified files."""
+        for filepath in files:
+            full_path = self.preview_dir / filepath
+            if not full_path.exists():
+                raise ValueError(f"File not found after changes: {filepath}")
+                
+            if filepath.suffix == '.py':
+                self.validate_python_syntax(full_path)
+            
+            self.validated_files.add(filepath)
+            from rich import print as rprint
+            rprint(f"[green]✓[/green] Validated [cyan]{filepath}[/cyan]")
+
+    def run_tests(self):
+        """Run tests if configured in janito.yaml."""
+        config_file = self.preview_dir / 'janito.yaml'
+        if not config_file.exists():
+            print("No test configuration found")
+            return
+
+        try:
+            with open(config_file) as f:
+                config = yaml.safe_load(f)
+            
+            test_cmd = config.get('test_cmd')
+            if not test_cmd:
+                print("No test_cmd found in configuration")
+                return
+
+            print(f"Running test command: {test_cmd}")
+            
+            # Save current directory
+            original_dir = Path.cwd()
+            try:
+                # Change to preview directory
+                os.chdir(self.preview_dir)
+                # Run the test command
+                exit_code = os.system(test_cmd)
+            finally:
+                # Restore original directory
+                os.chdir(original_dir)
+            
+            if exit_code != 0:
+                raise ValueError(f"Test command failed with exit code {exit_code}")
+            
+            from rich.panel import Panel
+            from rich import print as rprint
+            
+            # Create a summary panel
+            validated_files_list = "\n".join([f"[cyan]• {f}[/cyan]" for f in sorted(self.validated_files)])
+            summary = Panel(
+                f"[green]✓[/green] All files validated successfully:\n\n{validated_files_list}",
+                title="[bold green]Validation Summary[/bold green]",
+                border_style="green"
+            )
+            rprint("\n" + summary + "\n")
+            print("Tests completed successfully")
+
+        except yaml.YAMLError as e:
+            raise ValueError(f"Error parsing janito.yaml: {e}")
+        except Exception as e:
+            raise ValueError(f"Error running tests: {e}")

janito/change/view/content.py

@@ -0,0 +1,63 @@
+from typing import Optional
+from pathlib import Path
+from rich.syntax import Syntax
+
+
+# Mapping of file extensions to syntax lexer names
+FILE_EXTENSION_MAP = {
+    '.py': 'python',
+    '.js': 'javascript',
+    '.ts': 'typescript',
+    '.html': 'html',
+    '.css': 'css',
+    '.json': 'json',
+    '.md': 'markdown',
+    '.yaml': 'yaml',
+    '.yml': 'yaml',
+    '.sh': 'bash',
+    '.bash': 'bash',
+    '.sql': 'sql',
+    '.xml': 'xml',
+    '.cpp': 'cpp',
+    '.c': 'c',
+    '.h': 'cpp',
+    '.hpp': 'cpp',
+    '.java': 'java',
+    '.go': 'go',
+    '.rs': 'rust',
+}
+
+def get_file_syntax(filepath: Path) -> Optional[str]:
+    """Get syntax lexer name based on file extension.
+    
+    Args:
+        filepath: Path object containing the file path
+        
+    Returns:
+        String containing the syntax lexer name or None if not found
+    """
+    return ext_map.get(filepath.suffix.lower())
+
+def create_content_preview(filepath: Path, content: str, is_new: bool = False) -> Syntax:
+    """Create a preview with syntax highlighting using consistent styling
+
+    Args:
+        filepath: Path to the file being previewed
+        content: Content to preview
+        is_new: Whether this is a new file preview
+
+    Returns:
+        Syntax highlighted content
+    """
+    # Get file info
+    syntax_type = get_file_syntax(filepath)
+
+    # Create syntax highlighted content
+    return Syntax(
+        content,
+        syntax_type or "text",
+        theme="monokai",
+        line_numbers=True,
+        word_wrap=True,
+        tab_size=4
+    )

janito/change/{viewer → view}/diff.py

@@ -1,43 +1,44 @@
-from typing import List, Tuple
-from difflib import SequenceMatcher
-
-def find_common_sections(search_lines: List[str], replace_lines: List[str]) -> Tuple[List[str], List[str], List[str], List[str], List[str]]:
-    """Find common sections between search and replace content"""
-    # Find common lines from top
-    common_top = []
-    for s, r in zip(search_lines, replace_lines):
-        if s == r:
-            common_top.append(s)
-        else:
-            break
-
-    # Find common lines from bottom
-    search_remaining = search_lines[len(common_top):]
-    replace_remaining = replace_lines[len(common_top):]
-
-    common_bottom = []
-    for s, r in zip(reversed(search_remaining), reversed(replace_remaining)):
-        if s == r:
-            common_bottom.insert(0, s)
-        else:
-            break
-
-    # Get the unique middle sections
-    search_middle = search_remaining[:-len(common_bottom)] if common_bottom else search_remaining
-    replace_middle = replace_remaining[:-len(common_bottom)] if common_bottom else replace_remaining
-
-    return common_top, search_middle, replace_middle, common_bottom, search_lines
-
-def get_line_similarity(line1: str, line2: str) -> float:
-    """Calculate similarity ratio between two lines"""
-    return SequenceMatcher(None, line1, line2).ratio()
-
-def find_similar_lines(deleted_lines: List[str], added_lines: List[str], similarity_threshold: float = 0.5) -> List[Tuple[int, int, float]]:
-    """Find similar lines between deleted and added content"""
-    similar_pairs = []
-    for i, del_line in enumerate(deleted_lines):
-        for j, add_line in enumerate(added_lines):
-            similarity = get_line_similarity(del_line, add_line)
-            if similarity >= similarity_threshold:
-                similar_pairs.append((i, j, similarity))
-    return similar_pairs
+from typing import List, Tuple
+from difflib import SequenceMatcher
+
+def find_common_sections(search_lines: List[str], replace_lines: List[str]) -> Tuple[List[str], List[str], List[str], List[str], List[str]]:
+    """Find common sections between search and replace content"""
+    # Find common lines from top
+    common_top = []
+    for s, r in zip(search_lines, replace_lines):
+        if s == r:
+            common_top.append(s)
+        else:
+            break
+
+    # Find common lines from bottom
+    search_remaining = search_lines[len(common_top):]
+    replace_remaining = replace_lines[len(common_top):]
+
+    common_bottom = []
+    for s, r in zip(reversed(search_remaining), reversed(replace_remaining)):
+        if s == r:
+            common_bottom.insert(0, s)
+        else:
+            break
+
+    # Get the unique middle sections
+    search_middle = search_remaining[:-len(common_bottom)] if common_bottom else search_remaining
+    replace_middle = replace_remaining[:-len(common_bottom)] if common_bottom else replace_remaining
+
+    return common_top, search_middle, replace_middle, common_bottom, search_lines
+
+
+def find_similar_lines(deleted_lines: List[str], added_lines: List[str], similarity_threshold: float = 0.5) -> List[Tuple[int, int, float]]:
+    """Find similar lines between deleted and added content"""
+    similar_pairs = []
+    for i, del_line in enumerate(deleted_lines):
+        for j, add_line in enumerate(added_lines):
+            similarity = get_line_similarity(del_line, add_line)
+            if similarity >= similarity_threshold:
+                similar_pairs.append((i, j, similarity))
+    return similar_pairs
+
+def get_line_similarity(line1: str, line2: str) -> float:
+    """Calculate similarity ratio between two lines"""
+    return SequenceMatcher(None, line1, line2).ratio()