openrewrite-remote 0.12.0__py3-none-any.whl → 0.13.1__py3-none-any.whl

{rewrite/remote → rewrite_remote}/server.py

@@ -1,6 +1,8 @@
+# type: ignore
 import importlib
 import importlib.resources
 import os
+import select
 import socket
 import sys
 import time
@@ -8,58 +10,99 @@ import traceback
 import zipfile
 from io import BytesIO, StringIO
 from pathlib import Path
+from typing import Any
 from typing import Optional
 
 import cbor2
-import select
 from cbor2 import dumps
-from rewrite import ParserInput, InMemoryExecutionContext, ExecutionContext, ParseError
+from rewrite import (
+    ParserInput,
+    InMemoryExecutionContext,
+    ExecutionContext,
+    ParseError,
+    Recipe,
+)
+from rewrite.java import J
+from rewrite.java.remote.receiver import JavaReceiver
+from rewrite.java.remote.sender import JavaSender
 from rewrite.python import Py
 from rewrite.python.parser import PythonParserBuilder
-from rewrite.remote import SenderContext, ReceiverContext, RemotingContext, RemotingMessenger, RemotingMessageType, \
-    RemotePrinterFactory, OK, ParseErrorSender
 from rewrite.python.remote.receiver import PythonReceiver
 from rewrite.python.remote.sender import PythonSender
 
+from rewrite_remote.handlers.hello_world_handler import hello_world_handler
+from rewrite_remote.handlers.recipe_install_handler import recipe_install_handler
+from rewrite_remote.handlers.run_recipe_load_and_visitor_handler import (
+    run_recipe_load_and_visitor_handler,
+)
+from rewrite_remote.handlers.list_projects_handler import list_projects_handler
+
+from rewrite_remote.receiver import ReceiverContext
+from rewrite_remote.remoting import (
+    RemotePrinterFactory,
+    RemotingContext,
+    RemotingMessageType,
+    RemotingMessenger,
+)
+from rewrite_remote.sender import ParseErrorSender, SenderContext
+
 INACTIVITY_TIMEOUT = 300  # 5 minutes
 _OK: int = 0
 _ERROR: int = 1
 
 
-def register_remoting_factories():
-    SenderContext.register(ParseError, lambda: ParseErrorSender())
-    SenderContext.register(Py, lambda: PythonSender())
-    ReceiverContext.register(Py, lambda: PythonReceiver())
+def register_remoting_factories() -> None:
+    SenderContext.register(ParseError, ParseErrorSender)
+    SenderContext.register(Py, PythonSender)
+    ReceiverContext.register(Py, PythonReceiver)
+    SenderContext.register(J, JavaSender)
+    ReceiverContext.register(J, JavaReceiver)
 
 
-def find_free_port():
+def find_free_port() -> Any:
     """Find a free port by using the system to allocate a port for us."""
     with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-        s.bind(('', 0))
+        s.bind(("", 0))
         s.listen(1)
         return s.getsockname()[1]
 
 
 class Server:
+    """
+    A server that listens for incoming connections and processes requests from clients.
+    """
+
     _port: Optional[int]
     _path: Optional[str]
     _remoting_context: RemotingContext
     _messenger: RemotingMessenger
 
-    def __init__(self, port: int = None, path: str = None, timeout: int = INACTIVITY_TIMEOUT):
+    def __init__(
+        self,
+        port: Optional[int] = None,
+        path: Optional[str] = None,
+        timeout: int = INACTIVITY_TIMEOUT,
+    ):
         self._port = port
         self._path = path
         self.timeout = timeout
         self._remoting_context = RemotingContext()
+        self._remoting_context._recipe_factories["test"] = (
+            lambda recipe_options: Recipe()
+        )
         self._messenger = RemotingMessenger(
             self._remoting_context,
             {
-                'parse-python-source': self.parse_python_source,
-                'parse-python-file': self.parse_python_file,
-            }
+                "parse-python-source": self.parse_python_source,
+                "parse-python-file": self.parse_python_file,
+                "hello-world": hello_world_handler,
+                "recipe-install": recipe_install_handler,
+                "run-recipe-load-and-visitor": run_recipe_load_and_visitor_handler,
+                "list-projects": list_projects_handler,
+            },
         )
 
-    def start(self):
+    def start(self) -> None:
         """Start the server and listen for connections on the given port."""
         if self._path:
             if os.path.exists(self._path):
@@ -75,7 +118,7 @@ class Server:
         else:
             with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
                 s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-                s.bind(('localhost', self._port))
+                s.bind(("localhost", self._port))
                 s.listen(5)
                 print(f"Server listening on port {self._port}")
                 last_activity_time = time.time()
@@ -90,7 +133,9 @@ class Server:
                     except socket.timeout:
                         current_time = time.time()
                         if current_time - last_activity_time >= self.timeout:
-                            print("No new connections for 5 minutes, shutting down server.")
+                            print(
+                                "No new connections for 5 minutes, shutting down server."
+                            )
                             break
 
     def handle_client(self, sock: socket.socket) -> None:
@@ -130,62 +175,77 @@ class Server:
         finally:
             sock.close()
 
-    def parse_python_source(self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext):
+    def parse_python_source(
+        self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext
+    ) -> None:
         remoting_ctx.reset()
         source = cbor2.load(stream)
         ctx = InMemoryExecutionContext()
         ctx.put_message(ExecutionContext.REQUIRE_PRINT_EQUALS_INPUT, False)
-        for cu in PythonParserBuilder().build().parse_inputs(
-                [ParserInput(Path('source.py'), None, True, lambda: StringIO(source))],
-                None, ctx):
+        for cu in (
+            PythonParserBuilder()
+            .build()
+            .parse_inputs(
+                [ParserInput(Path("source.py"), None, True, lambda: StringIO(source))],
+                None,
+                ctx,
+            )
+        ):
             response_stream = BytesIO()
             cbor2.dump(RemotingMessageType.Response, response_stream)
-            cbor2.dump(OK, response_stream)
+            cbor2.dump(_OK, response_stream)
             source_stream = BytesIO()
             remoting_ctx.new_sender_context(source_stream).send_any_tree(cu, None)
             cbor2.dump(source_stream.getvalue(), response_stream)
             sock.sendall(response_stream.getvalue())
 
-    def parse_python_file(self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext):
+    def parse_python_file(
+        self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext
+    ) -> None:
         remoting_ctx.reset()
         path = cbor2.load(stream)
         ctx = InMemoryExecutionContext()
         ctx.put_message(ExecutionContext.REQUIRE_PRINT_EQUALS_INPUT, False)
-        for cu in PythonParserBuilder().build().parse_inputs(
+        for cu in (
+            PythonParserBuilder()
+            .build()
+            .parse_inputs(
                 [ParserInput(Path(path), None, True, lambda: read_file_contents(path))],
-                None, ctx):
+                None,
+                ctx,
+            )
+        ):
             response_stream = BytesIO()
             cbor2.dump(RemotingMessageType.Response, response_stream)
-            cbor2.dump(OK, response_stream)
+            cbor2.dump(_OK, response_stream)
             source_stream = BytesIO()
             remoting_ctx.new_sender_context(source_stream).send_any_tree(cu, None)
             cbor2.dump(source_stream.getvalue(), response_stream)
             sock.sendall(response_stream.getvalue())
 
 
-def read_file_contents(path):
-    with open(path, 'r', newline='', encoding='utf-8') as file:
+def read_file_contents(path: str) -> StringIO:
+    with open(path, "r", newline="", encoding="utf-8") as file:
         return StringIO(file.read())
 
 
-def read_data_from_zip():
+def read_data_from_zip() -> None:
     # Access the resource within the 'your_package.resources' package
     # 'data.zip' is the name of the file included
-    with importlib.resources.open_binary('resources', 'rewrite-remote-java.zip') as f:
+    with importlib.resources.open_binary("resources", "rewrite-remote-java.zip") as f:
         # Open the zip file from the resource file stream
         with zipfile.ZipFile(f) as zip_file:
             # List the contents of the zip file
             print(zip_file.namelist())
             # Read a specific file inside the zip file (if you know the file name within it)
-            with zip_file.open('rewrite-remote-java-0.2.0-SNAPSHOT/bin/rewrite-remote-java') as inner_file:
+            with zip_file.open(
+                "rewrite-remote-java-0.2.0-SNAPSHOT/bin/rewrite-remote-java"
+            ) as inner_file:
                 data = inner_file.read()
-                print(data.decode('utf-8'))
+                print(data.decode("utf-8"))
 
 
-# Example usage
-# read_data_from_zip()
-
-if __name__ == "__main__":
+def main() -> Any:
     sys.setrecursionlimit(2000)
     port = int(sys.argv[1]) if len(sys.argv) > 1 else 54322
     timeout = int(sys.argv[2]) if len(sys.argv) > 2 else INACTIVITY_TIMEOUT
@@ -193,3 +253,7 @@ if __name__ == "__main__":
     Server(port=port, timeout=timeout).start()
     # Server(port=find_free_port()).start()
     # Server(path=tempfile.gettempdir() + '/rewrite-csharp.sock').start()
+
+
+if __name__ == "__main__":
+    main()

rewrite_remote/type_utils.py

@@ -0,0 +1,109 @@
+import typing
+from typing import Optional
+from dataclasses import Field
+from functools import lru_cache
+from typing import Any, Type
+
+
+def to_java_type_name_from_value(v: Any) -> Optional[str]:
+    if isinstance(v, bool):
+        return "java.lang.Boolean"
+    elif isinstance(v, int):
+        bit_length = v.bit_length()
+        if bit_length < 32:
+            return "java.lang.Integer"
+        elif bit_length < 64:
+            return "java.lang.Long"
+        else:
+            return "java.math.BigInteger"
+    elif isinstance(v, complex):
+        return "java.lang.String"
+    return to_java_type_name(type(v))
+
+
+@lru_cache(maxsize=None)
+def to_java_type_name(t: typing.Type[Any]) -> Optional[str]:
+    if t == bytes:
+        # FIXME can we map this somehow?
+        return None
+    if t == bool:
+        return "java.lang.Boolean"
+    if t == int:
+        return "java.lang.Integer"
+    if t == str:
+        return "java.lang.String"
+    if t == float:
+        return "java.lang.Double"
+    if t == type(None):
+        return "null"
+    if t.__module__.startswith("rewrite.java.support_types"):
+        if t.__name__ == "Space":
+            return "org.openrewrite.java.tree.Space"
+        if t.__name__ == "Comment":
+            return "org.openrewrite.java.tree.Comment"
+        if t.__name__ == "TextComment":
+            return "org.openrewrite.java.tree.TextComment"
+        if t.__name__ == "JLeftPadded":
+            return "org.openrewrite.java.tree.JLeftPadded"
+        if t.__name__ == "JRightPadded":
+            return "org.openrewrite.java.tree.JRightPadded"
+        if t.__name__ == "JContainer":
+            return "org.openrewrite.java.tree.JContainer"
+    if t.__module__.startswith("rewrite.java.markers"):
+        return "org.openrewrite.java.marker." + t.__qualname__
+    if t.__module__.startswith("rewrite.java.tree"):
+        return "org.openrewrite.java.tree.J$" + t.__qualname__.replace(".", "$")
+    if t.__module__.startswith("rewrite.python.support_types"):
+        if t.__name__ == "PyComment":
+            return "org.openrewrite.python.tree.PyComment"
+        if t.__name__ == "PyLeftPadded":
+            return "org.openrewrite.python.tree.PyLeftPadded"
+        if t.__name__ == "PyRightPadded":
+            return "org.openrewrite.python.tree.PyRightPadded"
+        if t.__name__ == "PyContainer":
+            return "org.openrewrite.python.tree.PyContainer"
+    if t.__module__.startswith("rewrite.python.markers"):
+        return "org.openrewrite.python.marker." + t.__qualname__
+    if t.__module__.startswith("rewrite.python.style"):
+        return "org.openrewrite.python.style." + t.__qualname__
+    if t.__module__.startswith("rewrite.python.tree"):
+        return "org.openrewrite.python.tree.Py$" + t.__qualname__.replace(".", "$")
+    if t.__module__.startswith("rewrite.marker"):
+        if t.__name__ == "ParseExceptionResult":
+            return "org.openrewrite.ParseExceptionResult"
+        return "org.openrewrite.marker." + t.__qualname__.replace(".", "$")
+    if t.__module__ == "rewrite.parser" and t.__name__ == "ParseError":
+        return "org.openrewrite.tree.ParseError"
+    if t.__module__.startswith("rewrite.style"):
+        return "org.openrewrite.style." + t.__qualname__
+    if t.__module__.startswith("rewrite.") and t.__module__.endswith(".tree"):
+        model = t.__module__.split(".")[1]
+        return (
+            "org.openrewrite."
+            + model
+            + ".tree."
+            + model.capitalize()
+            + "$"
+            + t.__qualname__.replace(".", "$")
+        )
+    return t.__module__ + "." + t.__qualname__
+    # raise NotImplementedError("to_java_type_name: " + str(o))
+
+
+@lru_cache(maxsize=None)
+def to_java_field_name(field: Field[Any]) -> str:
+    return __convert_snake_to_camel(field.name[1:])
+
+
+def __convert_snake_to_camel(field_name: str) -> str:
+    # Remove leading underscore
+    if field_name.startswith("_"):
+        field_name = field_name[1:]
+
+    # Convert snake case to camel case
+    components = field_name.split("_")
+    return components[0] + "".join(x.capitalize() for x in components[1:])
+
+
+def get_type(type_name: str) -> Type[Any]:
+    raise NotImplementedError("get_type for: " + type_name)

openrewrite_remote-0.12.0.dist-info/METADATA

@@ -1,17 +0,0 @@
-Metadata-Version: 2.1
-Name: openrewrite-remote
-Version: 0.12.0
-Summary: Remoting functionality for the OpenRewrite library.
-License: Moderne, Inc. Commercial License
-Author: Moderne Inc.
-Author-email: support@moderne.io
-Requires-Python: >=3.9
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: cbor2 (>=5.6.4,<6.0.0)
-Requires-Dist: openrewrite

openrewrite_remote-0.12.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-rewrite/remote/__init__.py,sha256=e8QQGZNZpUFcqo1NGQRAJb0il0_Zq0R5CrtM0i5DF9w,245
-rewrite/remote/client.py,sha256=LBCh8msEv04wex7kenXVwI11i7JB4YvcnSIbf3oJ3pg,1608
-rewrite/remote/event.py,sha256=9AL5z6D6_GUBUPrNb9WJCHv3RJkID4k2vgiArwIJtmQ,354
-rewrite/remote/receiver.py,sha256=YLu6YhLCj6aNTeRsHdRFIeC7FY1rp4vGAA5PRze3JO0,15586
-rewrite/remote/remote_utils.py,sha256=wZhsDEV3qnayy3M98r6M-NMxGeowQ1qW7rFSJR0cY24,10145
-rewrite/remote/remoting.py,sha256=30tsczUenwEikpTXZE1ghNU8YTl4JU5LoXTgkzDsPgs,13091
-rewrite/remote/sender.py,sha256=miE7aKTa3RLPoZgYBc0sNnaHlpC7huJuK3Lbp8Kqqo8,18433
-rewrite/remote/server.py,sha256=k2fuGPR402EYOeNkYDILKq3aRTt0B0I8HZ4PySFURxs,7840
-rewrite/remote/type_utils.py,sha256=XG3enI43oQF6Bq6c9eaU9bFpBTyL9JJ2qKuwmNCpyHQ,4073
-openrewrite_remote-0.12.0.dist-info/METADATA,sha256=Zzo_3xQaeCiPZ4U8chh8PFHTsoytPhnMtkJ1v7Zpedw,659
-openrewrite_remote-0.12.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-openrewrite_remote-0.12.0.dist-info/RECORD,,

rewrite/remote/__init__.py

@@ -1,7 +0,0 @@
-__path__ = __import__('pkgutil').extend_path(__path__, __name__)
-
-from .receiver import *
-from .sender import *
-from .remoting import *
-
-__all__ = [name for name in dir() if not name.startswith('_') and not isinstance(globals()[name], TypeVar)]

rewrite/remote/client.py

@@ -1,40 +0,0 @@
-import socket
-import tempfile
-from pathlib import Path
-
-import rewrite.java.tree as j
-import rewrite.python.tree as py
-from rewrite import Markers
-from rewrite.python import Py
-from rewrite.java import Space, JavaType
-
-from rewrite import random_id, Cursor, PrintOutputCapture
-from rewrite.remote import RemotingContext, RemotePrinterFactory, SenderContext, ReceiverContext
-from rewrite.remote.python.receiver import PythonReceiver
-from rewrite.remote.python.sender import PythonSender
-
-SenderContext.register(Py, lambda: PythonSender())
-ReceiverContext.register(Py, lambda: PythonReceiver())
-
-# Path to the Unix domain socket
-SOCKET_PATH = tempfile.gettempdir() + '/rewrite-java.sock'
-
-# Create a Unix domain socket
-client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-
-# Connect the socket to the path where the server is listening
-client.connect(('localhost', 65432))
-print(f'Connected to {SOCKET_PATH}')
-
-try:
-    remoting = RemotingContext()
-    remoting.connect(client)
-    RemotePrinterFactory(remoting.client).set_current()
-
-    literal = j.Literal(random_id(), Space.SINGLE_SPACE, Markers.EMPTY, True, 'True', None, JavaType.Primitive())
-    assert_ = py.AssertStatement(random_id(), Space.EMPTY, Markers.EMPTY, [j.JRightPadded(literal, Space.EMPTY, Markers.EMPTY)])
-    cu = py.CompilationUnit(random_id(), Space.EMPTY, Markers.EMPTY, Path('/foo.py'), None, None, False, None, [], [j.JRightPadded(assert_, Space.EMPTY, Markers.EMPTY)], Space.EMPTY)
-    printed = cu.print(Cursor(None, Cursor.ROOT_VALUE), PrintOutputCapture(0))
-    assert printed == 'assert True'
-finally:
-    client.close()

rewrite/remote/type_utils.py

@@ -1,101 +0,0 @@
-import typing
-from dataclasses import Field
-from functools import lru_cache
-from typing import Any, Type
-
-
-def to_java_type_name_from_value(v: Any) -> str:
-    if isinstance(v, bool):
-        return 'java.lang.Boolean'
-    elif isinstance(v, int):
-        bit_length = v.bit_length()
-        if  bit_length < 32:
-            return 'java.lang.Integer'
-        elif bit_length < 64:
-            return 'java.lang.Long'
-        else:
-            return 'java.math.BigInteger'
-    elif isinstance(v, complex):
-        return 'java.lang.String'
-    return to_java_type_name(type(v))
-
-
-@lru_cache(maxsize=None)
-def to_java_type_name(t: typing.Type) -> str:
-    if t == bytes:
-        # FIXME can we map this somehow?
-        return None
-    if t == bool:
-        return 'java.lang.Boolean'
-    if t == int:
-        return 'java.lang.Integer'
-    if t == str:
-        return 'java.lang.String'
-    if t == float:
-        return 'java.lang.Double'
-    if t == type(None):
-        return 'null'
-    if t.__module__.startswith('rewrite.java.support_types'):
-        if t.__name__ == 'Space':
-            return 'org.openrewrite.java.tree.Space'
-        if t.__name__ == 'Comment':
-            return 'org.openrewrite.java.tree.Comment'
-        if t.__name__ == 'TextComment':
-            return 'org.openrewrite.java.tree.TextComment'
-        if t.__name__ == 'JLeftPadded':
-            return 'org.openrewrite.java.tree.JLeftPadded'
-        if t.__name__ == 'JRightPadded':
-            return 'org.openrewrite.java.tree.JRightPadded'
-        if t.__name__ == 'JContainer':
-            return 'org.openrewrite.java.tree.JContainer'
-    if t.__module__.startswith('rewrite.java.markers'):
-        return 'org.openrewrite.java.marker.' + t.__qualname__
-    if t.__module__.startswith('rewrite.java.tree'):
-        return 'org.openrewrite.java.tree.J$' + t.__qualname__.replace('.', '$')
-    if t.__module__.startswith('rewrite.python.support_types'):
-        if t.__name__ == 'PyComment':
-            return 'org.openrewrite.python.tree.PyComment'
-        if t.__name__ == 'PyLeftPadded':
-            return 'org.openrewrite.python.tree.PyLeftPadded'
-        if t.__name__ == 'PyRightPadded':
-            return 'org.openrewrite.python.tree.PyRightPadded'
-        if t.__name__ == 'PyContainer':
-            return 'org.openrewrite.python.tree.PyContainer'
-    if t.__module__.startswith('rewrite.python.markers'):
-        return 'org.openrewrite.python.marker.' + t.__qualname__
-    if t.__module__.startswith('rewrite.python.style'):
-        return 'org.openrewrite.python.style.' + t.__qualname__
-    if t.__module__.startswith('rewrite.python.tree'):
-        return 'org.openrewrite.python.tree.Py$' + t.__qualname__.replace('.', '$')
-    if t.__module__.startswith('rewrite.marker'):
-        if t.__name__ == 'ParseExceptionResult':
-            return 'org.openrewrite.ParseExceptionResult'
-        return 'org.openrewrite.marker.' + t.__qualname__.replace('.', '$')
-    if t.__module__ == 'rewrite.parser' and t.__name__ == 'ParseError':
-        return 'org.openrewrite.tree.ParseError'
-    if t.__module__.startswith('rewrite.style'):
-        return 'org.openrewrite.style.' + t.__qualname__
-    if t.__module__.startswith('rewrite.') and t.__module__.endswith('.tree'):
-        model = t.__module__.split('.')[1]
-        return 'org.openrewrite.' + model + '.tree.' + model.capitalize() + '$' + t.__qualname__.replace('.', '$')
-    return t.__module__ + '.' + t.__qualname__
-    # raise NotImplementedError("to_java_type_name: " + str(o))
-
-
-@lru_cache(maxsize=None)
-def to_java_field_name(field: Field) -> str:
-    return __convert_snake_to_camel(field.name[1:])
-
-
-def __convert_snake_to_camel(field_name: str) -> str:
-    # Remove leading underscore
-    if field_name.startswith('_'):
-        field_name = field_name[1:]
-
-    # Convert snake case to camel case
-    components = field_name.split('_')
-    return components[0] + ''.join(x.capitalize() for x in components[1:])
-
-
-def get_type(type_name: str) -> Type:
-    raise NotImplementedError("get_type for: " + type_name)