py-wlcli 0.8.0__py3-none-any.whl

wl_cli/commands/format/rust_formatter.py

@@ -0,0 +1,106 @@
+"""
+Rust code formatting utilities.
+"""
+
+import subprocess
+from pathlib import Path
+
+from ...utils.platform_adapter import PlatformAdapter
+from .format_outcome import FormatFailure
+
+
+def format_rust_code(rust_dir: str, env: dict, quiet: bool) -> FormatFailure | None:
+    """Format Rust code if rust directory exists.
+
+    Returns a :class:`FormatFailure` when ``cargo fmt`` did not run or did not
+    succeed, so the caller folds the verdict into ``FormatOutcome`` (#187).
+    Before this the failure was only printed and the format gate stayed green.
+
+    ``kind`` is ``tool_missing`` when ``cargo`` could not even be launched and
+    ``rust`` when it ran but judged the formatting failed.
+    """
+    rust_path = Path(rust_dir)
+    if not rust_path.exists():
+        if not quiet:
+            print("Rust directory not found, skipping Rust code formatting")
+        return None
+
+    # Fix encoding issues on Windows
+    environment = env.copy() if env else {}
+    if PlatformAdapter.is_windows():
+        # Set environment variables to ensure proper UTF-8 handling
+        environment["PYTHONIOENCODING"] = "utf-8"
+        environment["PYTHONLEGACYWINDOWSFSENCODING"] = "1"
+
+    # Save current directory and change to rust directory
+    original_cwd = Path.cwd()
+    try:
+        import os
+
+        os.chdir(rust_dir)
+        _run_format_command(["cargo", "fmt"], environment, quiet, passthrough=not quiet)
+        return None
+    except FileNotFoundError as e:
+        # ``cargo`` is not on PATH: the formatter never ran at all.
+        detail = str(e)
+        failure = FormatFailure(
+            path=rust_path,
+            kind="tool_missing",
+            message=f"cargo could not be launched: {detail}",
+        )
+    except Exception as e:
+        detail = str(e)
+        failure = FormatFailure(
+            path=rust_path,
+            kind="rust",
+            message=f"cargo fmt failed: {detail}",
+        )
+    finally:
+        # Always return to original directory
+        os.chdir(original_cwd)
+
+    if not quiet:
+        print(f"Warning: Failed to format Rust code: {detail}")
+    return failure
+
+
+def _run_format_command(
+    command: list, env: dict, quiet: bool, passthrough: bool = False
+) -> None:
+    """Run a formatting command with proper output handling."""
+    # Prepare environment variables
+    environment = env.copy() if env else {}
+
+    # Fix encoding issues on Windows
+    if PlatformAdapter.is_windows():
+        # Set environment variables to ensure proper UTF-8 handling
+        environment["PYTHONIOENCODING"] = "utf-8"
+        environment["PYTHONLEGACYWINDOWSFSENCODING"] = "1"
+
+    # Check for execution of untrusted input - not applicable here as we're running
+    # trusted formatting tools with trusted paths
+    if quiet and not passthrough:
+        # In quiet mode, capture output to suppress it
+        result = subprocess.run(
+            command,
+            env=environment,
+            capture_output=True,
+            text=True,
+            # Explicitly set encoding for Windows systems
+            encoding="utf-8" if PlatformAdapter.is_windows() else None,
+        )
+        if result.returncode != 0:
+            raise Exception(
+                f"Command failed with return code {result.returncode}: {result.stderr}"
+            )
+    else:
+        # In normal mode or passthrough mode, let the command output directly to stdout/stderr
+        result = subprocess.run(
+            command,
+            env=environment,
+            text=True,  # Ensure we get string output instead of bytes
+            # Explicitly set encoding for Windows systems
+            encoding="utf-8" if PlatformAdapter.is_windows() else None,
+        )
+        if result.returncode != 0:
+            raise Exception(f"Command failed with return code {result.returncode}")

wl_cli/commands/gate/__init__.py

@@ -0,0 +1 @@
+"""Gate command package."""