mseep-patche 1.0.1__tar.gz

mseep_patche-1.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 jingfelix
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mseep_patche-1.0.1/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.1
+Name: mseep-patche
+Version: 1.0.1
+Summary: Modern Patch in Python
+Author-Email: mseep <support@skydeck.ai>
+License: MIT
+Requires-Python: >=3.10
+Requires-Dist: typer[all]>=0.9.0
+Requires-Dist: pydantic>=2.5.3
+Requires-Dist: pydantic-settings>=2.2.1
+Requires-Dist: whatthepatch-pydantic==1.0.6a3
+Description-Content-Type: text/plain
+
+Package managed by MseeP.ai

mseep_patche-1.0.1/pyproject.toml

@@ -0,0 +1,45 @@
+[project]
+name = "mseep-patche"
+version = "1.0.1"
+description = "Modern Patch in Python"
+authors = [
+    { name = "mseep", email = "support@skydeck.ai" },
+]
+dependencies = [
+    "typer[all]>=0.9.0",
+    "pydantic>=2.5.3",
+    "pydantic-settings>=2.2.1",
+    "whatthepatch-pydantic==1.0.6a3",
+]
+requires-python = ">=3.10"
+
+[project.readme]
+content-type = "text/plain"
+text = "Package managed by MseeP.ai"
+
+[project.license]
+text = "MIT"
+
+[project.scripts]
+patche = "Patche.__init__:app"
+patche-mcp = "Patche.mcp.__init__:app"
+
+[build-system]
+requires = [
+    "pdm-backend",
+]
+build-backend = "pdm.backend"
+
+[tool.pdm]
+distribution = true
+
+[tool.pdm.dev-dependencies]
+dev = [
+    "viztracer>=0.16.3",
+    "whatthepatch>=1.0.7",
+]
+
+[dependency-groups]
+mcp = [
+    "mcp[cli]>=1.6.0",
+]

mseep_patche-1.0.1/src/Patche/__init__.py

@@ -0,0 +1 @@
+from .app import __version__, app

mseep_patche-1.0.1/src/Patche/__main__.py

@@ -0,0 +1,3 @@
+from .app import app
+
+app()

mseep_patche-1.0.1/src/Patche/__version__.py

@@ -0,0 +1,9 @@
+import sys
+
+if sys.version_info >= (3, 10):
+    import importlib.metadata as importlib_metadata
+else:
+    import importlib_metadata
+
+
+__version__ = importlib_metadata.version("Patche")

mseep_patche-1.0.1/src/Patche/app.py

@@ -0,0 +1,32 @@
+import logging
+
+import typer
+from rich.console import Console
+from rich.logging import RichHandler
+
+logging.basicConfig(level=logging.INFO, handlers=[RichHandler()], format="%(message)s")
+logger = logging.getLogger()
+
+from Patche.__version__ import __version__
+from Patche.utils.common import post_executed
+
+app = typer.Typer(result_callback=post_executed, no_args_is_help=True)
+
+
+@app.callback(invoke_without_command=True)
+def callback(verbose: bool = False, version: bool = False):
+    """
+    Entry for public options
+    """
+    if verbose:
+        logger.setLevel(logging.DEBUG)
+
+    if version:
+        console = Console()
+        console.print(f"patche version {__version__}")
+        raise typer.Exit()
+
+
+from Patche.commands.apply import apply
+from Patche.commands.help import show_settings
+from Patche.commands.show import show

mseep_patche-1.0.1/src/Patche/commands/apply.py

@@ -0,0 +1,107 @@
+import os
+from typing import Annotated
+
+import typer
+
+from Patche.app import app, logger
+from Patche.model import File
+from Patche.utils.parse import parse_patch
+from Patche.utils.resolve import apply_change
+
+
+@app.command()
+def apply(
+    patch_path: Annotated[str, typer.Argument(help="Path to the patch file")],
+    reverse: Annotated[
+        bool,
+        typer.Option(
+            "-R",
+            "--reverse",
+            help="Assume patches were created with old and new files swapped.",
+        ),
+    ] = False,
+    fuzz: Annotated[
+        int,
+        typer.Option(
+            "-F", "--fuzz", help="Set the fuzz factor to LINES for inexact matching."
+        ),
+    ] = 0,
+):
+    """
+    Apply a patch to a file.
+    """
+
+    if not os.path.exists(patch_path):
+        logger.error(f"Warning: {patch_path} not found!")
+        raise typer.Exit(code=1)
+
+    if reverse:
+        logger.info("Reversing patch...")
+
+    has_failed = False
+
+    with open(patch_path, mode="r", encoding="utf-8") as (f):
+        diffes = parse_patch(f.read()).diff
+
+        for diff in diffes:
+
+            old_filename = diff.header.old_path
+            new_filename = diff.header.new_path
+            if reverse:
+                old_filename, new_filename = new_filename, old_filename
+
+            logger.debug(f"old_filename: {old_filename} new_filename: {new_filename}")
+
+            if old_filename == "/dev/null":
+                # 根据 diffes 创建新文件
+                try:
+                    assert len(diff.hunks) == 1
+
+                    new_line_list = []
+                    for line in diff.changes:
+
+                        assert line.old is None
+                        new_line_list.append(line)
+
+                    with open(new_filename, mode="w+", encoding="utf-8") as f:
+                        for line in new_line_list:
+                            f.write(line.content + "\n")
+
+                except AssertionError:
+                    logger.error("Failed to create new file: invalid diff!")
+                    raise typer.Exit(code=1)
+
+            elif new_filename == "/dev/null":
+                # 移除 old_filename
+                if os.path.exists(old_filename):
+                    os.remove(old_filename)
+                else:
+                    logger.error(f"{old_filename} not found!")
+                    raise typer.Exit(code=1)
+            else:
+                if os.path.exists(old_filename):
+
+                    logger.info(f"Applying patch to {old_filename}...")
+
+                    new_line_list = File(file_path=old_filename).line_list
+                    apply_result = apply_change(
+                        diff.hunks, new_line_list, reverse=reverse, fuzz=fuzz
+                    )
+                    new_line_list = apply_result.new_line_list
+
+                    # 检查失败数
+                    for failed_hunk in apply_result.failed_hunk_list:
+                        has_failed = True
+                        logger.error(f"Failed hunk: {failed_hunk.index}")
+                else:
+                    logger.error(f"{old_filename} not found!")
+                    raise typer.Exit(code=1)
+
+                # 写入文件
+                if not has_failed:
+                    with open(new_filename, mode="w+", encoding="utf-8") as f:
+                        for line in new_line_list:
+                            if line.status:
+                                f.write(line.content + "\n")
+
+    raise typer.Exit(code=1 if has_failed else 0)

mseep_patche-1.0.1/src/Patche/commands/help.py

@@ -0,0 +1,12 @@
+from rich import print
+
+from Patche.app import app
+from Patche.config import settings
+
+
+@app.command("settings")
+def show_settings():
+    """
+    Show current settings
+    """
+    print(settings.model_dump_json())

mseep_patche-1.0.1/src/Patche/commands/show.py

@@ -0,0 +1,41 @@
+import os
+from typing import Annotated
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from Patche.app import app, logger
+from Patche.utils.parse import parse_patch
+
+
+@app.command()
+def show(filename: Annotated[str, typer.Argument(help="Path to the patch file")]):
+    """
+    Show details of a patch file.
+    """
+    if not os.path.exists(filename):
+        logger.error(f"Warning: {filename} not found!")
+        return
+
+    content = ""
+    with open(filename, mode="r", encoding="utf-8") as (f):
+        content = f.read()
+
+    patch = parse_patch(content)
+
+    table = Table(box=None)
+    table.add_column("Field", justify="left", style="cyan")
+    table.add_column("Value", justify="left", style="magenta")
+
+    table.add_row("Patch", filename)
+    table.add_row("Sha", patch.sha)
+    table.add_row("Author", patch.author)
+    table.add_row("Date", (patch.date))
+    table.add_row("Subject", patch.subject)
+
+    for diff in patch.diff:
+        table.add_row("Diff", f"{diff.header.old_path} -> {diff.header.new_path}")
+
+    console = Console()
+    console.print(table)

mseep_patche-1.0.1/src/Patche/config.py

@@ -0,0 +1,24 @@
+import importlib.resources as pkg_resources
+import os
+from functools import lru_cache
+
+from pydantic_settings import BaseSettings
+
+
+@lru_cache()
+def get_settings():
+    _settings = Settings()
+    if not os.path.exists(_settings.Config.env_file):
+        open(_settings.Config.env_file, "w").close()
+
+    return _settings
+
+
+class Settings(BaseSettings):
+    max_diff_lines: int = 3
+
+    class Config:
+        env_file = os.path.join(os.environ.get("HOME"), ".Patche.env")
+
+
+settings = get_settings()

mseep_patche-1.0.1/src/Patche/mcp/__init__.py

@@ -0,0 +1,42 @@
+import asyncio
+import logging
+
+import typer
+from typer import Typer
+
+from Patche.app import logger
+from Patche.mcp.server import serve
+
+app = Typer()
+
+
+@app.command()
+# @app.option(
+#     "--repository",
+#     help="Repository to apply the patch",
+#     default=None,
+# )
+# @app.option(
+#     "--debug",
+#     help="Enable debug mode",
+#     default=False,
+# )
+def main(
+    repository: str = typer.Option(
+        None, "--repository", "-r", help="Repository to apply the patch"
+    ),
+    debug: bool = typer.Option(False, "--debug", "-d", help="Enable debug mode"),
+):
+    """
+    Main entry point for the Patche MCP server.
+    """
+
+    logger.info("Starting Patche MCP server...")
+    if debug:
+        logger.setLevel(logging.DEBUG)
+
+    asyncio.run(serve(repository))
+
+
+if __name__ == "__main__":
+    main()

mseep_patche-1.0.1/src/Patche/mcp/__main__.py

@@ -0,0 +1,3 @@
+from Patche.mcp import main
+
+main()

mseep_patche-1.0.1/src/Patche/mcp/model.py

@@ -0,0 +1,29 @@
+from enum import Enum
+
+from pydantic import BaseModel
+
+
+# 定义 Patche 相关的数据模型
+class PatcheConfig(BaseModel):
+    pass
+
+
+class PatcheList(BaseModel):
+    patche_dir: str
+
+
+class PatcheShow(BaseModel):
+    patch_path: str
+
+
+class PatcheApply(BaseModel):
+    patch_path: str
+    target_dir: str
+    reverse: bool = False
+
+
+class PatcheTools(str, Enum):
+    CONFIG = "patche_config"
+    LIST = "patche_list"
+    SHOW = "patche_show"
+    APPLY = "patche_apply"