openrewrite-remote 0.13.4__py3-none-any.whl → 0.14.0__py3-none-any.whl

rewrite_remote/server.py

@@ -2,18 +2,15 @@
 import importlib
 import importlib.resources
 import os
-import select
 import socket
 import sys
 import time
 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,
@@ -22,7 +19,6 @@ from rewrite import (
     ParseError,
     Recipe,
 )
-from rewrite.java import J
 from rewrite.java.remote import *
 from rewrite.python import Py
 from rewrite.python.parser import PythonParserBuilder
@@ -30,12 +26,17 @@ 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.list_projects_handler import list_projects_handler
+from rewrite_remote.handlers.parse_project_sources_handler import (
+    parse_project_sources_handler,
+)
+from rewrite_remote.handlers.project_helper import read_file_contents
+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,
@@ -86,9 +87,7 @@ class Server:
         self._path = path
         self.timeout = timeout
         self._remoting_context = RemotingContext()
-        self._remoting_context._recipe_factories["test"] = (
-            lambda recipe_options: Recipe()
-        )
+        self._remoting_context._recipe_factories["test"] = lambda recipe_options: Recipe()
         self._messenger = RemotingMessenger(
             self._remoting_context,
             {
@@ -98,6 +97,7 @@ class Server:
                 "recipe-install": recipe_install_handler,
                 "run-recipe-load-and-visitor": run_recipe_load_and_visitor_handler,
                 "list-projects": list_projects_handler,
+                "parse-project-sources": parse_project_sources_handler,
             },
         )
 
@@ -132,9 +132,7 @@ 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:
@@ -175,7 +173,10 @@ class Server:
             sock.close()
 
     def parse_python_source(
-        self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext
+        self,
+        stream: BytesIO,
+        sock: socket.socket,
+        remoting_ctx: RemotingContext,
     ) -> None:
         remoting_ctx.reset()
         source = cbor2.load(stream)
@@ -199,7 +200,10 @@ class Server:
             sock.sendall(response_stream.getvalue())
 
     def parse_python_file(
-        self, stream: BytesIO, sock: socket.socket, remoting_ctx: RemotingContext
+        self,
+        stream: BytesIO,
+        sock: socket.socket,
+        remoting_ctx: RemotingContext,
     ) -> None:
         remoting_ctx.reset()
         path = cbor2.load(stream)
@@ -209,7 +213,14 @@ class Server:
             PythonParserBuilder()
             .build()
             .parse_inputs(
-                [ParserInput(Path(path), None, True, lambda: read_file_contents(path))],
+                [
+                    ParserInput(
+                        Path(path),
+                        None,
+                        True,
+                        lambda: read_file_contents(path),
+                    )
+                ],
                 None,
                 ctx,
             )
@@ -223,11 +234,6 @@ class Server:
             sock.sendall(response_stream.getvalue())
 
 
-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() -> None:
     # Access the resource within the 'your_package.resources' package
     # 'data.zip' is the name of the file included

rewrite_remote/type_utils.py

@@ -1,8 +1,9 @@
+import importlib
 import typing
-from typing import Optional
 from dataclasses import Field
 from functools import lru_cache
 from typing import Any, Type
+from typing import Optional
 
 
 def to_java_type_name_from_value(v: Any) -> Optional[str]:
@@ -106,4 +107,7 @@ def __convert_snake_to_camel(field_name: str) -> str:
 
 
 def get_type(type_name: str) -> Type[Any]:
-    raise NotImplementedError("get_type for: " + type_name)
+    # `type_name` will look like `org.openrewrite.java.tree.J$CompilationUnit`
+    parts = type_name.split('.')
+    module = importlib.import_module('rewrite.' + parts[2])
+    return getattr(module, parts[-1].split('$')[1])

openrewrite_remote-0.13.4.dist-info/RECORD

@@ -1,23 +0,0 @@
-rewrite_remote/__init__.py,sha256=uuLrPH--ewvE-5owXbNItXDfjCypMXQgsm-72hO_dtc,286
-rewrite_remote/client.py,sha256=95ZCAtVOngF0ZqqKnOsrweUeGKruf3UKGPXNGTrNyy0,1853
-rewrite_remote/event.py,sha256=texLJD1mcFkpBpiXAa-Rmip0Tgqm2OlBpRPHFZyWcBs,359
-rewrite_remote/receiver.py,sha256=9oIdupOcV9z6eY-c2uFL1yfujbTQTgbXgTPG9Qx_ipc,20008
-rewrite_remote/remote_utils.py,sha256=wUo9WZoldgCLihFJGf6RaE1SufhDiEPCFlX74tcODVM,10552
-rewrite_remote/remoting.py,sha256=s0qVJlKTKhLZw8-9CJB1dJ4sPeKVkwIldI2WYEGULHY,13365
-rewrite_remote/sender.py,sha256=z42hmAIXai_dazQ-DhMwv_fDSi0AA2orUT2juWX2UJg,20056
-rewrite_remote/server.py,sha256=LDk4mUs33DX1s8HMCh3rC5lkpN8luyDu1CLeDQXczlY,9343
-rewrite_remote/type_utils.py,sha256=oVrB0olWFSCqhmg2nTU2wrwiAU7kBCUscjwdHK7gf3Y,4219
-rewrite_remote/handlers/__init__.py,sha256=ED6jHcYiuYpr_0vjGz0zx2lrrmJT9sDJCzIljoDfmlM,65
-rewrite_remote/handlers/handler_helpers.py,sha256=CIAXtlzrsE-z5RmzlXBZihT1rfrRLHFttq8Ni_8AH9U,516
-rewrite_remote/handlers/hello_world_handler.py,sha256=NQScHfCJKofdImBgELL8tXtG_KnVyXj683C3Ae7xcKU,1298
-rewrite_remote/handlers/list_projects_handler.py,sha256=ukkCST7whpCxQ0omCWWw9W27Su3rFSnnEIhAS6T4iOU,2007
-rewrite_remote/handlers/project_helper.py,sha256=_zGzniyEgmBB4k14Oy4Sm7B3Lt1MoXKQLOJxofEELGI,7130
-rewrite_remote/handlers/pypi_manager.py,sha256=z4JJarLkNRztQaSLHCh5GViag7ce7E36ARBDqJLQREc,10269
-rewrite_remote/handlers/recipe_install_handler.py,sha256=D6n5d84pS-HBWNiND-qZt-SfB33uFHV0ntZf2ZQ1Imo,4942
-rewrite_remote/handlers/run_recipe_load_and_visitor_handler.py,sha256=Pppn90g2z0z3U9m4_3a3X9Wu0Qh_pvkhoxWgadjAHRs,4146
-rewrite_remote/handlers/types.py,sha256=5TK_oFp-7iy1iACWisuQJsa_WHTsqcPrV9nQTZ-_xgo,518
-openrewrite_remote-0.13.4.dist-info/METADATA,sha256=jL-ze_dwxBxu6JQADFZO2s_QJLF7IxPJwEQdU1pTLaI,386
-openrewrite_remote-0.13.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-openrewrite_remote-0.13.4.dist-info/entry_points.txt,sha256=SMukuF7TPjQr3IZIcH8f98-_QBCqYSbYXYrVv-5UzRI,69
-openrewrite_remote-0.13.4.dist-info/top_level.txt,sha256=ansTioSZ-62aH3F2L3d1Bua0pJF4GOtgQ1PpG-CzcP0,15
-openrewrite_remote-0.13.4.dist-info/RECORD,,