python-color-math 0.2.1__tar.gz → 0.2.2__tar.gz

{python_color_math-0.2.1/python_color_math.egg-info → python_color_math-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-color-math
-Version: 0.2.1
+Version: 0.2.2
 Summary: A fast, local semantic LaTeX colorizer for Markdown, Jupyter, and LaTeX math equations.
 Author: Aditya
 License: MIT License
@@ -232,7 +232,7 @@ Integrate `color-math` into your git workflow using [pre-commit](https://pre-com
 ```yaml
 repos:
   - repo: https://github.com/36ty-blip/color-math
-    rev: v0.2.1
+    rev: v0.2.2
     hooks:
       # Formatter: colorizes modified Markdown, LaTeX, and Jupyter files in-place
       - id: color-math

{python_color_math-0.2.1 → python_color_math-0.2.2}/README.md

@@ -175,7 +175,7 @@ Integrate `color-math` into your git workflow using [pre-commit](https://pre-com
 ```yaml
 repos:
   - repo: https://github.com/36ty-blip/color-math
-    rev: v0.2.1
+    rev: v0.2.2
     hooks:
       # Formatter: colorizes modified Markdown, LaTeX, and Jupyter files in-place
       - id: color-math

{python_color_math-0.2.1 → python_color_math-0.2.2}/color_math/__init__.py

@@ -5,7 +5,7 @@ from .config import ColorMathOptions
 from .converters.block import convert_text
 from .undo import uncolor_text
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 
 __all__ = [
     "__version__",

{python_color_math-0.2.1 → python_color_math-0.2.2}/color_math/main.py

@@ -32,7 +32,7 @@ from .parsers.math_parser import describe_math_blocks
 from .self_test import run_self_test
 from .tutorial import run_tutorial
 
-VERSION = "0.2.1"
+VERSION = "0.2.2"
 SUPPORTED_EXTENSIONS = {".md", ".markdown", ".qmd", ".ipynb", ".tex", ".latex"}
 DEFAULT_EXCLUDES = {
     ".git",

{python_color_math-0.2.1 → python_color_math-0.2.2}/color_math/mcp_server.py

@@ -149,22 +149,30 @@ def create_mcp_server() -> MCPServer:
             normalize_braces=normalize_braces,
         )
 
-        clean_expr = expression.strip()
-        had_delimiters = clean_expr.startswith("$$") and clean_expr.endswith("$$")
-        block_text = clean_expr if had_delimiters else f"$${clean_expr}$$"
+        try:
+            clean_expr = expression.strip()
+            had_delimiters = clean_expr.startswith("$$") and clean_expr.endswith("$$")
+            block_text = clean_expr if had_delimiters else f"$${clean_expr}$$"
 
-        colored = convert_math_block(block_text, pal, opts)
-        if not had_delimiters and colored.startswith("$$") and colored.endswith("$$"):
-            colored_output = colored[2:-2]
-        else:
-            colored_output = colored
+            colored = convert_math_block(block_text, pal, opts)
+            if not had_delimiters and colored.startswith("$$") and colored.endswith("$$"):
+                colored_output = colored[2:-2]
+            else:
+                colored_output = colored
 
-        return {
-            "original": expression,
-            "colorized": colored_output,
-            "changed": colored_output != expression,
-            "theme": theme,
-        }
+            return {
+                "original": expression,
+                "colorized": colored_output,
+                "changed": colored_output != expression,
+                "theme": theme,
+            }
+        except Exception as err:
+            return {
+                "original": expression,
+                "error": str(err),
+                "changed": False,
+                "theme": theme,
+            }
 
     @server.tool(
         name="colorize_text",
@@ -199,16 +207,25 @@ def create_mcp_server() -> MCPServer:
             normalize_braces=normalize_braces,
         )
 
-        fmt = format_name if format_name in FORMATS else "markdown"
-        result = transform_document(text, fmt, undo=False, palette=pal, options=opts)
-        return {
-            "format": fmt,
-            "theme": theme,
-            "original_length": len(text),
-            "colorized_length": len(result),
-            "changed": result != text,
-            "result": result,
-        }
+        try:
+            fmt = format_name if format_name in FORMATS else "markdown"
+            result = transform_document(text, fmt, undo=False, palette=pal, options=opts)
+            return {
+                "format": fmt,
+                "theme": theme,
+                "original_length": len(text),
+                "colorized_length": len(result),
+                "changed": result != text,
+                "result": result,
+            }
+        except Exception as err:
+            return {
+                "format": format_name,
+                "theme": theme,
+                "error": str(err),
+                "changed": False,
+                "result": text,
+            }
 
     @server.tool(
         name="uncolor_text",
@@ -221,15 +238,23 @@ def create_mcp_server() -> MCPServer:
         text: str,
         format_name: str = "markdown",
     ) -> dict[str, Any]:
-        fmt = format_name if format_name in FORMATS else "markdown"
-        result = transform_document(text, fmt, undo=True)
-        return {
-            "format": fmt,
-            "original_length": len(text),
-            "cleaned_length": len(result),
-            "changed": result != text,
-            "result": result,
-        }
+        try:
+            fmt = format_name if format_name in FORMATS else "markdown"
+            result = transform_document(text, fmt, undo=True)
+            return {
+                "format": fmt,
+                "original_length": len(text),
+                "cleaned_length": len(result),
+                "changed": result != text,
+                "result": result,
+            }
+        except Exception as err:
+            return {
+                "format": format_name,
+                "error": str(err),
+                "changed": False,
+                "result": text,
+            }
 
     @server.tool(
         name="process_file",
@@ -271,47 +296,56 @@ def create_mcp_server() -> MCPServer:
             normalize_braces=normalize_braces,
         )
 
-        fmt = detect_format(path)
-        original_text, raw_bytes = read_utf8(path)
-        transformed = transform_document(
-            original_text, fmt, undo=undo, palette=pal, options=opts
-        )
+        try:
+            fmt = detect_format(path)
+            original_text, raw_bytes = read_utf8(path)
+            transformed = transform_document(
+                original_text, fmt, undo=undo, palette=pal, options=opts
+            )
 
-        changed = transformed != original_text
-        diff_output = ""
-        if changed:
-            diff_lines = list(
-                difflib.unified_diff(
-                    original_text.splitlines(keepends=True),
-                    transformed.splitlines(keepends=True),
-                    fromfile=str(path),
-                    tofile=str(path),
+            changed = transformed != original_text
+            diff_output = ""
+            if changed:
+                diff_lines = list(
+                    difflib.unified_diff(
+                        original_text.splitlines(keepends=True),
+                        transformed.splitlines(keepends=True),
+                        fromfile=str(path),
+                        tofile=str(path),
+                    )
                 )
-            )
-            diff_output = "".join(diff_lines[:100])  # limit diff length
+                diff_output = "".join(diff_lines[:100])  # limit diff length
 
-        written = False
-        if changed and in_place:
-            new_bytes = encode_utf8(transformed, raw_bytes)
-            written = replace_bytes(path, new_bytes, raw_bytes)
+            written = False
+            if changed and in_place:
+                new_bytes = encode_utf8(transformed, raw_bytes)
+                written = replace_bytes(path, new_bytes, raw_bytes)
 
-        return {
-            "file_path": str(path),
-            "format": fmt,
-            "action": "undo" if undo else "colorize",
-            "changed": changed,
-            "written_to_disk": written,
-            "diff": diff_output if changed else None,
-            "message": (
-                f"File successfully updated on disk: {path.name}"
-                if written
-                else (
-                    f"Dry run complete: changes detected in {path.name}"
-                    if changed
-                    else f"No changes needed: {path.name} is already up to date."
-                )
-            ),
-        }
+            return {
+                "file_path": str(path),
+                "format": fmt,
+                "action": "undo" if undo else "colorize",
+                "changed": changed,
+                "written_to_disk": written,
+                "diff": diff_output if changed else None,
+                "message": (
+                    f"File successfully updated on disk: {path.name}"
+                    if written
+                    else (
+                        f"Dry run complete: changes detected in {path.name}"
+                        if changed
+                        else f"No changes needed: {path.name} is already up to date."
+                    )
+                ),
+            }
+        except Exception as err:
+            return {
+                "file_path": str(path),
+                "error": str(err),
+                "changed": False,
+                "written_to_disk": False,
+                "message": f"Error processing {path.name}: {err}",
+            }
 
     @server.tool(
         name="scan_vault",
@@ -423,3 +457,8 @@ def run_mcp_server(transport: str = "stdio", **kwargs: Any) -> None:
         server.run(transport="streamable-http", **kwargs)
     else:
         raise ValueError(f"Unsupported transport: {transport}. Expected 'stdio', 'sse', or 'streamable-http'.")
+
+
+if __name__ == "__main__":
+    run_mcp_server()
+

{python_color_math-0.2.1 → python_color_math-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-color-math"
-version = "0.2.1"
+version = "0.2.2"
 description = "A fast, local semantic LaTeX colorizer for Markdown, Jupyter, and LaTeX math equations."
 readme = "README.md"
 requires-python = ">=3.10"

{python_color_math-0.2.1 → python_color_math-0.2.2/python_color_math.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-color-math
-Version: 0.2.1
+Version: 0.2.2
 Summary: A fast, local semantic LaTeX colorizer for Markdown, Jupyter, and LaTeX math equations.
 Author: Aditya
 License: MIT License
@@ -232,7 +232,7 @@ Integrate `color-math` into your git workflow using [pre-commit](https://pre-com
 ```yaml
 repos:
   - repo: https://github.com/36ty-blip/color-math
-    rev: v0.2.1
+    rev: v0.2.2
     hooks:
       # Formatter: colorizes modified Markdown, LaTeX, and Jupyter files in-place
       - id: color-math

{python_color_math-0.2.1 → python_color_math-0.2.2}/tests/test_mcp_hardcore.py

@@ -406,6 +406,30 @@ $$"""
         self.assertIn("-", colorized)
         self.assertIn(r"\end{cases}", colorized)
 
+    # -------------------------------------------------------------
+    # 13. Adversarial MCP Error Resilience (Malformed Input, Binary Files)
+    # -------------------------------------------------------------
+    def test_adversarial_error_resilience(self) -> None:
+        """Verify that malformed inputs and binary files return structured error dicts without crashing."""
+        # 1. Malformed Jupyter notebook in colorize_text
+        bad_ipynb = "{\n  \"cells\": [ this is not valid json! \n"
+        res_color = self._call("colorize_text", {"text": bad_ipynb, "format_name": "jupyter"})
+        self.assertFalse(res_color.get("changed", True))
+        self.assertIn("error", res_color)
+
+        # 2. Malformed Jupyter notebook in uncolor_text
+        res_uncolor = self._call("uncolor_text", {"text": bad_ipynb, "format_name": "jupyter"})
+        self.assertFalse(res_uncolor.get("changed", True))
+        self.assertIn("error", res_uncolor)
+
+        # 3. Binary file in process_file
+        with tempfile.TemporaryDirectory() as tmpdir:
+            bin_file = Path(tmpdir) / "corrupt.md"
+            bin_file.write_bytes(b"\xff\xfe\x00\x00\xaa\xbb\xcc")
+            res_proc = self._call("process_file", {"file_path": str(bin_file)})
+            self.assertFalse(res_proc.get("changed", True))
+            self.assertIn("error", res_proc)
+
 
 if __name__ == "__main__":
     unittest.main()