openrewrite-remote 0.3.5__tar.gz → 0.4.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: openrewrite-remote
3
- Version: 0.3.5
3
+ Version: 0.4.1
4
4
  Summary: Remoting functionality for the OpenRewrite library.
5
5
  License: Apache-2.0
6
6
  Author: Moderne Inc.
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "openrewrite-remote"
3
- version = "0.3.5" # This will be replaced by the GitHub Actions workflow
3
+ version = "0.4.1" # This will be replaced by the GitHub Actions workflow
4
4
  description = "Remoting functionality for the OpenRewrite library."
5
5
  authors = ["Moderne Inc. <support@moderne.io>"]
6
6
  license = "Apache-2.0"
@@ -18,8 +18,8 @@ from rewrite.python import Py
18
18
  from rewrite.python.parser import PythonParserBuilder
19
19
  from rewrite.remote import SenderContext, ReceiverContext, RemotingContext, RemotingMessenger, RemotingMessageType, \
20
20
  RemotePrinterFactory, OK, ParseErrorSender
21
- from rewrite.remote.python.receiver import PythonReceiver
22
- from rewrite.remote.python.sender import PythonSender
21
+ from rewrite.python.remote.receiver import PythonReceiver
22
+ from rewrite.python.remote.sender import PythonSender
23
23
 
24
24
  INACTIVITY_TIMEOUT = 300 # 5 minutes
25
25
  _OK: int = 0
@@ -1,176 +0,0 @@
1
- from functools import lru_cache
2
- from operator import attrgetter
3
- from typing import Type, Callable, TypeVar, Optional
4
-
5
- from rewrite.java import *
6
- from rewrite.remote import SenderContext, ReceiverContext
7
-
8
- T = TypeVar('T')
9
-
10
- def receive_container(container: Optional[JContainer[T]], _: Optional[str], ctx: ReceiverContext) -> JContainer[T]:
11
- if container is not None:
12
- container = container.with_before(ctx.receive_node(container.before, receive_space))
13
- container = container.padding.with_elements(ctx.receive_nodes(container.padding.elements, receive_right_padded_tree))
14
- container = container.with_markers(ctx.receive_node(container.markers, ctx.receive_markers))
15
- else:
16
- container = JContainer(
17
- ctx.receive_node(None, receive_space),
18
- ctx.receive_nodes(None, receive_right_padded_tree),
19
- ctx.receive_node(None, ctx.receive_markers)
20
- )
21
- return container
22
-
23
-
24
- def send_container(container: JContainer[T], ctx: SenderContext):
25
- ctx.send_node(container, attrgetter('before'), send_space)
26
- ctx.send_nodes(container, attrgetter('padding.elements'), send_right_padded, lambda t: t.element.id)
27
- ctx.send_node(container, attrgetter('markers'), ctx.send_markers)
28
-
29
-
30
- def send_left_padded(left_padded: JLeftPadded[T], ctx: SenderContext):
31
- ctx.send_node(left_padded, attrgetter('before'), send_space)
32
- if isinstance(left_padded.element, (J, Space)):
33
- ctx.send_node(left_padded, attrgetter('element'), ctx.send_tree)
34
- else:
35
- ctx.send_value(left_padded, attrgetter('element'))
36
- ctx.send_node(left_padded, attrgetter('markers'), ctx.send_markers)
37
-
38
-
39
- def send_right_padded(right_padded: JRightPadded[T], ctx: SenderContext):
40
- if isinstance(right_padded.element, J):
41
- ctx.send_node(right_padded, attrgetter('element'), ctx.send_tree)
42
- elif isinstance(right_padded.element, Space):
43
- ctx.send_node(right_padded, attrgetter('element'), send_space)
44
- else:
45
- ctx.send_value(right_padded, attrgetter('element'))
46
- ctx.send_node(right_padded, attrgetter('after'), send_space)
47
- ctx.send_node(right_padded, attrgetter('markers'), ctx.send_markers)
48
-
49
-
50
- def receive_space(space: Optional[Space], _: Optional[str], ctx: ReceiverContext) -> Space:
51
- if space is not None:
52
- space = space.with_comments(ctx.receive_nodes(space.comments, receive_comment))
53
- space = space.with_whitespace(ctx.receive_value(space.whitespace, str))
54
- else:
55
- space = Space(
56
- ctx.receive_nodes(None, receive_comment),
57
- ctx.receive_value(None, str)
58
- )
59
-
60
- return space
61
-
62
-
63
- def receive_comment(comment: Optional[Comment], _: Optional[str], ctx: ReceiverContext) -> Comment:
64
- if comment:
65
- comment = comment.with_multiline(ctx.receive_value(comment.multiline, bool))
66
- comment = comment.with_text(ctx.receive_value(comment.text, str))
67
- comment = comment.with_suffix(ctx.receive_value(comment.suffix, str))
68
- comment = comment.with_markers(ctx.receive_node(comment.markers, ctx.receive_markers))
69
- else:
70
- comment = TextComment(
71
- ctx.receive_value(None, bool),
72
- ctx.receive_value(None, str),
73
- ctx.receive_value(None, str),
74
- ctx.receive_node(None, ctx.receive_markers)
75
- )
76
- return comment
77
-
78
-
79
- def send_space(space: Space, ctx: SenderContext):
80
- ctx.send_nodes(space, attrgetter('comments'), send_comment, lambda x: x)
81
- ctx.send_value(space, attrgetter('whitespace'))
82
-
83
-
84
- def send_comment(comment: Comment, ctx: SenderContext):
85
- ctx.send_value(comment, attrgetter('multiline'))
86
- ctx.send_value(comment, attrgetter('text'))
87
- ctx.send_value(comment, attrgetter('suffix'))
88
- ctx.send_node(comment, attrgetter('markers'), ctx.send_markers)
89
-
90
-
91
- @lru_cache(maxsize=10)
92
- def left_padded_value_receiver(type_: Type) -> Callable[[Optional[JLeftPadded[T]], Optional[str], ReceiverContext], JLeftPadded[T]]:
93
- def receiver(left_padded: Optional[JLeftPadded[T]], _: Optional[str], ctx: ReceiverContext) -> JLeftPadded[T]:
94
- if left_padded is not None:
95
- left_padded = left_padded.with_before(ctx.receive_node(left_padded.before, receive_space))
96
- left_padded = left_padded.with_element(ctx.receive_value(left_padded.element, type_))
97
- left_padded = left_padded.with_markers(ctx.receive_node(left_padded.markers, ctx.receive_markers))
98
- else:
99
- left_padded = JLeftPadded(
100
- ctx.receive_node(None, receive_space),
101
- ctx.receive_value(None, type_),
102
- ctx.receive_node(None, ctx.receive_markers)
103
- )
104
- return left_padded
105
- return receiver
106
-
107
-
108
- @lru_cache(maxsize=10)
109
- def left_padded_node_receiver(type_: Type) -> Callable[[Optional[JLeftPadded[T]], Optional[str], ReceiverContext], JLeftPadded[T]]:
110
- if type_ is Space:
111
- def space_receiver(left_padded: Optional[JLeftPadded[T]], _: Optional[str], ctx: ReceiverContext) -> JLeftPadded[T]:
112
- if left_padded is not None:
113
- left_padded = left_padded.with_before(ctx.receive_node(left_padded.before, receive_space))
114
- left_padded = left_padded.with_element(ctx.receive_node(left_padded.element, receive_space))
115
- left_padded = left_padded.with_markers(ctx.receive_node(left_padded.markers, ctx.receive_markers))
116
- else:
117
- left_padded = JLeftPadded(
118
- ctx.receive_node(None, receive_space),
119
- ctx.receive_node(None, receive_space),
120
- ctx.receive_node(None, ctx.receive_markers)
121
- )
122
- return left_padded
123
- return space_receiver
124
- else:
125
- raise ValueError("Only Space is supported")
126
-
127
-
128
- def receive_left_padded_tree(left_padded: Optional[JLeftPadded[T]], _: Optional[str], ctx: ReceiverContext) -> JLeftPadded[T]:
129
- if left_padded is not None:
130
- left_padded = left_padded.with_before(ctx.receive_node(left_padded.before, receive_space))
131
- left_padded = left_padded.with_element(ctx.receive_node(left_padded.element, ctx.receive_tree))
132
- left_padded = left_padded.with_markers(ctx.receive_node(left_padded.markers, ctx.receive_markers))
133
- else:
134
- left_padded = JLeftPadded(
135
- ctx.receive_node(None, receive_space),
136
- ctx.receive_node(None, ctx.receive_tree),
137
- ctx.receive_node(None, ctx.receive_markers)
138
- )
139
- return left_padded
140
-
141
-
142
- @lru_cache(maxsize=10)
143
- def right_padded_value_receiver(type_: Type) -> Callable[[Optional[JRightPadded[T]], Optional[str], ReceiverContext], JRightPadded[T]]:
144
- def receiver(right_padded: Optional[JRightPadded[T]], _: Optional[str], ctx: ReceiverContext) -> JRightPadded[T]:
145
- if right_padded is not None:
146
- right_padded = right_padded.with_element(ctx.receive_value(right_padded.element, type_))
147
- right_padded = right_padded.with_after(ctx.receive_node(right_padded.after, receive_space))
148
- right_padded = right_padded.with_markers(ctx.receive_node(right_padded.markers, ctx.receive_markers))
149
- else:
150
- right_padded = JRightPadded(
151
- ctx.receive_value(None, type_),
152
- ctx.receive_node(None, receive_space),
153
- ctx.receive_node(None, ctx.receive_markers)
154
- )
155
- return right_padded
156
- return receiver
157
-
158
-
159
- @lru_cache(maxsize=10)
160
- def right_padded_node_receiver(type_: Type) -> Callable[[Optional[JRightPadded[T]], Optional[str], ReceiverContext], JRightPadded[T]]:
161
- raise ValueError("Not implemented")
162
-
163
-
164
- def receive_right_padded_tree(right_padded: Optional[JRightPadded[T]], _: Optional[str],
165
- ctx: ReceiverContext) -> JRightPadded[T]:
166
- if right_padded is not None:
167
- right_padded = right_padded.with_element(ctx.receive_node(right_padded.element, ctx.receive_tree))
168
- right_padded = right_padded.with_after(ctx.receive_node(right_padded.after, receive_space))
169
- right_padded = right_padded.with_markers(ctx.receive_node(right_padded.markers, ctx.receive_markers))
170
- else:
171
- right_padded = JRightPadded(
172
- ctx.receive_node(None, ctx.receive_tree),
173
- ctx.receive_node(None, receive_space),
174
- ctx.receive_node(None, ctx.receive_markers)
175
- )
176
- return right_padded