1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
############################################################################
# Copyright(c) Open Law Library. All rights reserved. #
# See ThirdPartyNotices.txt in the project root for additional notices. #
# #
# Licensed under the Apache License, Version 2.0 (the "License") #
# you may not use this file except in compliance with the License. #
# You may obtain a copy of the License at #
# #
# http: // www.apache.org/licenses/LICENSE-2.0 #
# #
# Unless required by applicable law or agreed to in writing, software #
# distributed under the License is distributed on an "AS IS" BASIS, #
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. #
# See the License for the specific language governing permissions and #
# limitations under the License. #
############################################################################
import re
from typing import Optional
from lsprotocol import types
from pygls.server import LanguageServer
NUMBER = re.compile(r"\d+")
COMMENT = re.compile(r"^#$")
server = LanguageServer(
name="inlay-hint-server",
version="v0.1",
notebook_document_sync=types.NotebookDocumentSyncOptions(
notebook_selector=[
types.NotebookDocumentSyncOptionsNotebookSelectorType2(
cells=[
types.NotebookDocumentSyncOptionsNotebookSelectorType2CellsType(
language="python"
)
]
)
]
),
)
def parse_int(chars: str) -> Optional[int]:
try:
return int(chars)
except Exception:
return None
@server.feature(types.TEXT_DOCUMENT_INLAY_HINT)
def inlay_hints(params: types.InlayHintParams):
items = []
document_uri = params.text_document.uri
document = server.workspace.get_text_document(document_uri)
start_line = params.range.start.line
end_line = params.range.end.line
lines = document.lines[start_line : end_line + 1]
for lineno, line in enumerate(lines):
match = COMMENT.match(line)
if match is not None:
nb = server.workspace.get_notebook_document(cell_uri=document_uri)
if nb is not None:
idx = 0
for idx, cell in enumerate(nb.cells):
if cell.document == document_uri:
break
items.append(
types.InlayHint(
label=f"notebook: {nb.uri}, cell {idx+1}",
kind=types.InlayHintKind.Type,
padding_left=False,
padding_right=True,
position=types.Position(line=lineno, character=match.end()),
)
)
for match in NUMBER.finditer(line):
if not match:
continue
number = parse_int(match.group(0))
if number is None:
continue
binary_num = bin(number).split("b")[1]
items.append(
types.InlayHint(
label=f":{binary_num}",
kind=types.InlayHintKind.Type,
padding_left=False,
padding_right=True,
position=types.Position(line=lineno, character=match.end()),
)
)
return items
@server.feature(types.INLAY_HINT_RESOLVE)
def inlay_hint_resolve(hint: types.InlayHint):
try:
n = int(hint.label[1:], 2)
hint.tooltip = f"Binary representation of the number: {n}"
except Exception:
pass
return hint
if __name__ == "__main__":
server.start_io()
|
