jaclang 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

jaclang/vendor/pygls/lsprotocol/validators.py

@@ -0,0 +1,47 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    import attrs
+
+INTEGER_MIN_VALUE = -(2**31)
+INTEGER_MAX_VALUE = 2**31 - 1
+
+
+def integer_validator(
+    instance: Any,
+    attribute: "attrs.Attribute[int]",
+    value: Any,
+) -> bool:
+    """Validates that integer value belongs in the range expected by LSP."""
+    if not isinstance(value, int) or not (
+        INTEGER_MIN_VALUE <= value <= INTEGER_MAX_VALUE
+    ):
+        name = attribute.name if hasattr(attribute, "name") else str(attribute)
+        raise ValueError(
+            f"{instance.__class__.__qualname__}.{name} should be in range [{INTEGER_MIN_VALUE}:{INTEGER_MAX_VALUE}], but was {value}."
+        )
+    return True
+
+
+UINTEGER_MIN_VALUE = 0
+UINTEGER_MAX_VALUE = 2**31 - 1
+
+
+def uinteger_validator(
+    instance: Any,
+    attribute: "attrs.Attribute[int]",
+    value: Any,
+) -> bool:
+    """Validates that unsigned integer value belongs in the range expected by LSP."""
+    if not isinstance(value, int) or not (
+        UINTEGER_MIN_VALUE <= value <= UINTEGER_MAX_VALUE
+    ):
+        name = attribute.name if hasattr(attribute, "name") else str(attribute)
+        raise ValueError(
+            f"{instance.__class__.__qualname__}.{name} should be in range [{UINTEGER_MIN_VALUE}:{UINTEGER_MAX_VALUE}], but was {value}."
+        )
+    return True

jaclang/vendor/pygls/progress.py

@@ -0,0 +1,79 @@
+import asyncio
+from concurrent.futures import Future
+from typing import Dict
+
+from jaclang.vendor.pygls.lsprotocol.types import (
+    PROGRESS,
+    WINDOW_WORK_DONE_PROGRESS_CREATE,
+    ProgressParams,
+    ProgressToken,
+    WorkDoneProgressBegin,
+    WorkDoneProgressEnd,
+    WorkDoneProgressReport,
+    WorkDoneProgressCreateParams,
+)
+from jaclang.vendor.pygls.protocol import LanguageServerProtocol
+
+
+class Progress:
+    """A class for working with client's progress bar.
+
+    Attributes:
+        _lsp(LanguageServerProtocol): Language server protocol instance
+        tokens(dict): Holds futures for work done progress tokens that are
+            already registered. These futures will be cancelled if the client
+            sends a cancel work done process notification.
+    """
+
+    def __init__(self, lsp: LanguageServerProtocol) -> None:
+        self._lsp = lsp
+
+        self.tokens: Dict[ProgressToken, Future] = {}
+
+    def _check_token_registered(self, token: ProgressToken) -> None:
+        if token in self.tokens:
+            raise Exception("Token is already registered!")
+
+    def _register_token(self, token: ProgressToken) -> None:
+        self.tokens[token] = Future()
+
+    def create(self, token: ProgressToken, callback=None) -> Future:
+        """Create a server initiated work done progress."""
+        self._check_token_registered(token)
+
+        def on_created(*args, **kwargs):
+            self._register_token(token)
+            if callback is not None:
+                callback(*args, **kwargs)
+
+        return self._lsp.send_request(
+            WINDOW_WORK_DONE_PROGRESS_CREATE,
+            WorkDoneProgressCreateParams(token=token),
+            on_created,
+        )
+
+    async def create_async(self, token: ProgressToken) -> asyncio.Future:
+        """Create a server initiated work done progress."""
+        self._check_token_registered(token)
+
+        result = await self._lsp.send_request_async(
+            WINDOW_WORK_DONE_PROGRESS_CREATE,
+            WorkDoneProgressCreateParams(token=token),
+        )
+        self._register_token(token)
+        return result
+
+    def begin(self, token: ProgressToken, value: WorkDoneProgressBegin) -> None:
+        """Notify beginning of work."""
+        # Register cancellation future for the case of client initiated progress
+        self.tokens.setdefault(token, Future())
+
+        return self._lsp.notify(PROGRESS, ProgressParams(token=token, value=value))
+
+    def report(self, token: ProgressToken, value: WorkDoneProgressReport) -> None:
+        """Notify progress of work."""
+        self._lsp.notify(PROGRESS, ProgressParams(token=token, value=value))
+
+    def end(self, token: ProgressToken, value: WorkDoneProgressEnd) -> None:
+        """Notify end of work."""
+        self._lsp.notify(PROGRESS, ProgressParams(token=token, value=value))