wafer-lsp 0.1.13__py3-none-any.whl

wafer_lsp/services/position_service.py

@@ -0,0 +1,77 @@
+import ast
+
+from lsprotocol.types import Position
+
+
+class PositionService:
+
+    def get_word_at_position(self, content: str, position: Position) -> str:
+        lines = content.split("\n")
+        if position.line >= len(lines):
+            return ""
+
+        line = lines[position.line]
+        if position.character >= len(line):
+            return ""
+
+        start = position.character
+        while start > 0 and (line[start - 1].isalnum() or line[start - 1] == "_" or line[start - 1] == "."):
+            start -= 1
+
+        end = position.character
+        while end < len(line) and (line[end].isalnum() or line[end] == "_" or line[end] == "."):
+            end += 1
+
+        return line[start:end]
+
+    def get_decorator_at_position(self, content: str, position: Position) -> tuple[str, int] | None:
+        lines = content.split("\n")
+        if position.line >= len(lines):
+            return None
+
+        line = lines[position.line]
+        line_stripped = line.strip()
+
+        if not line_stripped.startswith("@"):
+            return None
+
+        at_pos = line.find("@")
+        if at_pos != -1 and position.character < at_pos:
+            return None
+
+        try:
+            tree = ast.parse(line_stripped, mode="eval")
+            if isinstance(tree.body, ast.Attribute):
+                if isinstance(tree.body.value, ast.Name):
+                    decorator_name = f"{tree.body.value.id}.{tree.body.attr}"
+                    for i in range(position.line + 1, len(lines)):
+                        next_line = lines[i].strip()
+                        if next_line.startswith("def ") or next_line.startswith("class "):
+                            return (decorator_name, i)
+            elif isinstance(tree.body, ast.Name):
+                decorator_name = tree.body.id
+                for i in range(position.line + 1, len(lines)):
+                    next_line = lines[i].strip()
+                    if next_line.startswith("def ") or next_line.startswith("class "):
+                        return (decorator_name, i)
+        except SyntaxError:
+            pass
+
+        if "@cute.kernel" in line_stripped:
+            decorator_name = "cute.kernel"
+            for i in range(position.line + 1, len(lines)):
+                next_line = lines[i].strip()
+                if next_line.startswith("def ") or next_line.startswith("class "):
+                    return (decorator_name, i)
+        elif "@cute.struct" in line_stripped:
+            decorator_name = "cute.struct"
+            for i in range(position.line + 1, len(lines)):
+                next_line = lines[i].strip()
+                if next_line.startswith("def ") or next_line.startswith("class "):
+                    return (decorator_name, i)
+
+        return None
+
+
+def create_position_service() -> PositionService:
+    return PositionService()

wafer_lsp/utils/lsp_helpers.py

@@ -0,0 +1,79 @@
+import ast
+
+from lsprotocol.types import Position, Range
+
+
+def get_word_at_position(content: str, position: Position) -> str:
+    lines = content.split("\n")
+    if position.line >= len(lines):
+        return ""
+
+    line = lines[position.line]
+    if position.character >= len(line):
+        return ""
+
+    start = position.character
+    while start > 0 and (line[start - 1].isalnum() or line[start - 1] == "_" or line[start - 1] == "."):
+        start -= 1
+
+    end = position.character
+    while end < len(line) and (line[end].isalnum() or line[end] == "_" or line[end] == "."):
+        end += 1
+
+    return line[start:end]
+
+
+def get_decorator_at_position(content: str, position: Position) -> tuple[str, int] | None:
+    lines = content.split("\n")
+    if position.line >= len(lines):
+        return None
+
+    line = lines[position.line]
+    line_stripped = line.strip()
+
+    if not line_stripped.startswith("@"):
+        return None
+
+    at_pos = line.find("@")
+    if at_pos != -1 and position.character < at_pos:
+        return None
+
+    try:
+        tree = ast.parse(line_stripped, mode="eval")
+        if isinstance(tree.body, ast.Attribute):
+            if isinstance(tree.body.value, ast.Name):
+                decorator_name = f"{tree.body.value.id}.{tree.body.attr}"
+                for i in range(position.line + 1, len(lines)):
+                    next_line = lines[i].strip()
+                    if next_line.startswith("def ") or next_line.startswith("class "):
+                        return (decorator_name, i)
+        elif isinstance(tree.body, ast.Name):
+            decorator_name = tree.body.id
+            for i in range(position.line + 1, len(lines)):
+                next_line = lines[i].strip()
+                if next_line.startswith("def ") or next_line.startswith("class "):
+                    return (decorator_name, i)
+    except SyntaxError:
+        pass
+
+    if "@cute.kernel" in line_stripped:
+        decorator_name = "cute.kernel"
+        for i in range(position.line + 1, len(lines)):
+            next_line = lines[i].strip()
+            if next_line.startswith("def ") or next_line.startswith("class "):
+                return (decorator_name, i)
+    elif "@cute.struct" in line_stripped:
+        decorator_name = "cute.struct"
+        for i in range(position.line + 1, len(lines)):
+            next_line = lines[i].strip()
+            if next_line.startswith("def ") or next_line.startswith("class "):
+                return (decorator_name, i)
+
+    return None
+
+
+def create_range(start_line: int, start_char: int, end_line: int, end_char: int) -> Range:
+    return Range(
+        start=Position(line=start_line, character=start_char),
+        end=Position(line=end_line, character=end_char),
+    )

wafer_lsp-0.1.13.dist-info/METADATA

@@ -0,0 +1,60 @@
+Metadata-Version: 2.4
+Name: wafer-lsp
+Version: 0.1.13
+Summary: Language Server Protocol server for GPU programming languages
+Author-email: Wafer <support@wafer.ai>
+License: MIT
+Keywords: c++,cuda,cutedsl,gpu,language-server,lsp
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Text Editors :: Integrated Development Environments (IDE)
+Requires-Python: >=3.12
+Requires-Dist: lsprotocol>=2024.0.0
+Requires-Dist: pygls>=1.0.0
+Requires-Dist: wafer-core
+Provides-Extra: dev
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Wafer LSP
+
+Language Server Protocol server for CuTeDSL (Python GPU programming).
+
+**Beta Feature**: Currently only available when Beta Mode is enabled in VS Code settings.
+
+## Features
+
+- **Hover Information**: Shows kernel and layout information with compiler analysis when hovering over CuTeDSL code
+
+## Installation
+
+```bash
+pip install wafer-lsp
+```
+
+## Usage
+
+### VS Code Extension
+
+The LSP server is integrated into the `wevin-extension` VS Code extension. It starts automatically when Beta Mode is enabled.
+
+### Standalone
+
+For Neovim or other editors:
+
+```bash
+python -m wafer_lsp
+```
+
+## Supported Languages
+
+- **CuTeDSL**: Python files with `@cute.kernel` decorators
+
+## Architecture
+
+The LSP server uses a modular language registry system. Currently supports CuTeDSL only. See `languages/README.md` for details on adding more languages.

wafer_lsp-0.1.13.dist-info/RECORD

@@ -0,0 +1,44 @@
+wafer_lsp/__init__.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+wafer_lsp/__main__.py,sha256=6J_zJGlexw3ASzI-MRpCzMwzaq_hTdIe12JWg97pLRQ,112
+wafer_lsp/server.py,sha256=ZGFXghuPdKJH7aLhrYvOiKp6zfOdtoqWavdtDwg8Bk4,1938
+wafer_lsp/analyzers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wafer_lsp/analyzers/compiler_integration.py,sha256=SLyIZXk52nKP1b04s9Wo77cRLFn5auc4jGRo59Vl9Wc,341
+wafer_lsp/analyzers/docs_index.py,sha256=hrrBgPzIM_mJewCzHyw-2YEWdIchuB3SAMXI8PhtDq0,1309
+wafer_lsp/handlers/__init__.py,sha256=tPFuceRrlCkyW7N5BSRZJY7IMx9Etb6D5c54pZZHJ-w,926
+wafer_lsp/handlers/code_action.py,sha256=PTWs4X3CkfprMmzqC3cg7uEWdfivNNH9thU4E-c6ztQ,1411
+wafer_lsp/handlers/code_lens.py,sha256=mhdJtbS2LMNHGOirJbXUfatqhqMBess5wndfBBOLnok,1301
+wafer_lsp/handlers/completion.py,sha256=qz36GIJXu2fG6K-FdbFNGCCTYSpyV5A6-WOHHNne7QI,142
+wafer_lsp/handlers/diagnostics.py,sha256=zSaBYdZCqBK2Nb2WhCyQfKDATf8nsj6jMv-K1lvxnQc,1137
+wafer_lsp/handlers/document_symbol.py,sha256=H9rehZlbJKZ-2ueyvTYvz0_-0xfiGaAEPXSWbp76JUk,6465
+wafer_lsp/handlers/hip_diagnostics.py,sha256=YFaSBuwBb7DJNWhBQpwVzYDJvA0KmMV5X9aEEk4X8eM,10923
+wafer_lsp/handlers/hover.py,sha256=F3vweiSHcFlF66qvneJkX02hTq3iaxfXK2h9EkufL7w,8788
+wafer_lsp/handlers/inlay_hint.py,sha256=HW-AFeWps2VbEGPMTAG9KHyiT1s2JS81F71BXq7rwoA,7765
+wafer_lsp/handlers/semantic_tokens.py,sha256=JNJetP7Flh6yxNFzkGM3Ho9D4HaHsVSmoXdgcJnJtZM,7447
+wafer_lsp/handlers/workspace_symbol.py,sha256=YGSkf6doChh1BjVezEQFgIsI2NsQHhOjLchoF5bfBK4,2934
+wafer_lsp/languages/README.md,sha256=hLBqOBvKSqeCrNH9Wrr6XAHa8eBnd81IargPDurodU8,5527
+wafer_lsp/languages/__init__.py,sha256=5cXQkPRsh4JLyZd2QubvTxvuip10695Hb2zjyGF6Frc,472
+wafer_lsp/languages/converter.py,sha256=LPKJfkUV-rNXyJg7Tr5AZ1AlREKH4ulfjDHWlUNvyXU,2487
+wafer_lsp/languages/detector.py,sha256=7FDQZlnfnkI7DCeySzecIarYnNLINlIRgRkCj7L3SQY,4235
+wafer_lsp/languages/parser_manager.py,sha256=RBYS0w2wIRkKlrEDYvMEGKbsTkOSEpfUB3fjH_9X0Hk,945
+wafer_lsp/languages/registry.py,sha256=Ka9WDURUniqUAZDpdrQtRTLLzS36n4hq2dIwqj_AFdc,3703
+wafer_lsp/languages/types.py,sha256=c22sDECraXHvazqJSF7WC1ntFNyf9Mbx60EXA9jrbWk,624
+wafer_lsp/parsers/__init__.py,sha256=yh41cxz8Siuv0doHXnhuSY-JPoN82XtqDiOrbJAGeEg,709
+wafer_lsp/parsers/base_parser.py,sha256=RQjk5QOHrS398CrUGP2ZtSsisSl6Kh0KgWLPssSv8ss,176
+wafer_lsp/parsers/cuda_parser.py,sha256=X2EX8gFcZCK8-lSsedby1-vSgYsl2CHLxRxzwr5yIlI,2621
+wafer_lsp/parsers/cutedsl_parser.py,sha256=1r2gdAuwFrp4fQQNEQCRpkwtfunus-hq4h92j2MZyFE,3711
+wafer_lsp/parsers/hip_parser.py,sha256=nJdwsyTWxdeRJkLMHzUC_4px2fWvtwxd5FgdqOvIXAU,26465
+wafer_lsp/services/__init__.py,sha256=lMMuCBSMfKQeTlUnSo82tXuMGN9rmeoTmCfUigZReL0,1191
+wafer_lsp/services/analysis_service.py,sha256=J6jcg0YQZh9iQYyOXO9bF5NqaJDfnIKhwssCLNcUJjA,619
+wafer_lsp/services/docs_service.py,sha256=ltjkPuQ4DUmuc95AZVfQ6GSWgpjKqv_6-Luh2CFPocQ,1393
+wafer_lsp/services/document_service.py,sha256=gdQ-leIUSK7tiffFsi4UsseIyBx2TO_WIlBPFh57GU8,559
+wafer_lsp/services/hip_docs.py,sha256=mmbFy89Jpj3KUXLO_0M73f0zrkiIw6omtqwCva0Y7Jw,40442
+wafer_lsp/services/hip_hover_service.py,sha256=zbnjdsU6jlCIJh38EDM0sGnLr8I7VKbLABaULJxcXwE,14112
+wafer_lsp/services/hover_service.py,sha256=Zby2lhMO4US2IVTAVuU-ntkRKWrrR7HEjWdlGFSRQj4,9743
+wafer_lsp/services/language_registry_service.py,sha256=qvMR9oI1pUrSv09Q-WRF5-sEu2Nma2fZeUhdydIWaqg,864
+wafer_lsp/services/position_service.py,sha256=Nx-_AuBEr_yUXjAYPfJ_wmeGM_Gy5vbIUW163GqblCU,2831
+wafer_lsp/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wafer_lsp/utils/lsp_helpers.py,sha256=pXYvm_gJiJ_TAwwyGxhqvWJQx6CBecYK1c6tVY7gUF8,2739
+wafer_lsp-0.1.13.dist-info/METADATA,sha256=F0NGET532pP9BssFunp3FRIqUw_yvaIH_qsFmxWJ6eA,1742
+wafer_lsp-0.1.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wafer_lsp-0.1.13.dist-info/entry_points.txt,sha256=XHcM0QZy6GzCYY2ajejqL0tZ2X6y6zYWP0Lg1egxGW4,54
+wafer_lsp-0.1.13.dist-info/RECORD,,

wafer_lsp-0.1.13.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

wafer_lsp-0.1.13.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+wafer-lsp = wafer_lsp.__main__:main