sqlmesh 0.194.2.dev11__py3-none-any.whl → 0.195.1.dev1__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.194.2.dev11'
-__version_tuple__ = version_tuple = (0, 194, 2, 'dev11')
+__version__ = version = '0.195.1.dev1'
+__version_tuple__ = version_tuple = (0, 195, 1, 'dev1')

sqlmesh/lsp/main.py

@@ -52,6 +52,7 @@ from sqlmesh.lsp.reference import (
     get_references,
     get_all_references,
 )
+from sqlmesh.lsp.rename import prepare_rename, rename_symbol, get_document_highlights
 from sqlmesh.lsp.uri import URI
 from web.server.api.endpoints.lineage import column_lineage, model_lineage
 from web.server.api.endpoints.models import get_models
@@ -435,6 +436,59 @@ class SQLMeshLanguageServer:
                 ls.show_message(f"Error getting locations: {e}", types.MessageType.Error)
                 return None
 
+        @self.server.feature(types.TEXT_DOCUMENT_PREPARE_RENAME)
+        def prepare_rename_handler(
+            ls: LanguageServer, params: types.PrepareRenameParams
+        ) -> t.Optional[types.PrepareRenameResult]:
+            """Prepare for rename operation by checking if the symbol can be renamed."""
+            try:
+                uri = URI(params.text_document.uri)
+                self._ensure_context_for_document(uri)
+                if self.lsp_context is None:
+                    raise RuntimeError(f"No context found for document: {uri}")
+
+                result = prepare_rename(self.lsp_context, uri, params.position)
+                return result
+            except Exception as e:
+                ls.log_trace(f"Error preparing rename: {e}")
+                return None
+
+        @self.server.feature(types.TEXT_DOCUMENT_RENAME)
+        def rename_handler(
+            ls: LanguageServer, params: types.RenameParams
+        ) -> t.Optional[types.WorkspaceEdit]:
+            """Perform rename operation on the symbol at the given position."""
+            try:
+                uri = URI(params.text_document.uri)
+                self._ensure_context_for_document(uri)
+                if self.lsp_context is None:
+                    raise RuntimeError(f"No context found for document: {uri}")
+
+                workspace_edit = rename_symbol(
+                    self.lsp_context, uri, params.position, params.new_name
+                )
+                return workspace_edit
+            except Exception as e:
+                ls.show_message(f"Error performing rename: {e}", types.MessageType.Error)
+                return None
+
+        @self.server.feature(types.TEXT_DOCUMENT_DOCUMENT_HIGHLIGHT)
+        def document_highlight_handler(
+            ls: LanguageServer, params: types.DocumentHighlightParams
+        ) -> t.Optional[t.List[types.DocumentHighlight]]:
+            """Highlight all occurrences of the symbol at the given position."""
+            try:
+                uri = URI(params.text_document.uri)
+                self._ensure_context_for_document(uri)
+                if self.lsp_context is None:
+                    raise RuntimeError(f"No context found for document: {uri}")
+
+                highlights = get_document_highlights(self.lsp_context, uri, params.position)
+                return highlights
+            except Exception as e:
+                ls.log_trace(f"Error getting document highlights: {e}")
+                return None
+
         @self.server.feature(types.TEXT_DOCUMENT_DIAGNOSTIC)
         def diagnostic(
             ls: LanguageServer, params: types.DocumentDiagnosticParams

sqlmesh/lsp/rename.py

@@ -0,0 +1,137 @@
+import typing as t
+from lsprotocol.types import (
+    Position,
+    TextEdit,
+    WorkspaceEdit,
+    PrepareRenameResult_Type1,
+    DocumentHighlight,
+    DocumentHighlightKind,
+)
+
+from sqlmesh.lsp.context import LSPContext
+from sqlmesh.lsp.reference import (
+    _position_within_range,
+    get_cte_references,
+    LSPCteReference,
+)
+from sqlmesh.lsp.uri import URI
+
+
+def prepare_rename(
+    lsp_context: LSPContext, document_uri: URI, position: Position
+) -> t.Optional[PrepareRenameResult_Type1]:
+    """
+    Prepare for rename operation by checking if the symbol at the position can be renamed.
+
+    Args:
+        lsp_context: The LSP context
+        document_uri: The URI of the document
+        position: The position in the document
+
+    Returns:
+        PrepareRenameResult if the symbol can be renamed, None otherwise
+    """
+    # Check if there's a CTE at this position
+    cte_references = get_cte_references(lsp_context, document_uri, position)
+    if cte_references:
+        # Find the target CTE definition to get its range
+        target_range = None
+        for ref in cte_references:
+            # Check if cursor is on a CTE usage
+            if _position_within_range(position, ref.range):
+                target_range = ref.target_range
+                break
+            # Check if cursor is on the CTE definition
+            elif _position_within_range(position, ref.target_range):
+                target_range = ref.target_range
+                break
+        if target_range:
+            return PrepareRenameResult_Type1(range=target_range, placeholder="cte_name")
+
+    # For now, only CTEs are supported
+    return None
+
+
+def rename_symbol(
+    lsp_context: LSPContext, document_uri: URI, position: Position, new_name: str
+) -> t.Optional[WorkspaceEdit]:
+    """
+    Perform rename operation on the symbol at the given position.
+
+    Args:
+        lsp_context: The LSP context
+        document_uri: The URI of the document
+        position: The position in the document
+        new_name: The new name for the symbol
+
+    Returns:
+        WorkspaceEdit with the changes, or None if no symbol to rename
+    """
+    # Check if there's a CTE at this position
+    cte_references = get_cte_references(lsp_context, document_uri, position)
+    if cte_references:
+        return _rename_cte(cte_references, new_name)
+
+    # For now, only CTEs are supported
+    return None
+
+
+def _rename_cte(cte_references: t.List[LSPCteReference], new_name: str) -> WorkspaceEdit:
+    """
+    Create a WorkspaceEdit for renaming a CTE.
+
+    Args:
+        cte_references: List of CTE references (definition and usages)
+        new_name: The new name for the CTE
+
+    Returns:
+        WorkspaceEdit with the text edits for renaming the CTE
+    """
+    changes: t.Dict[str, t.List[TextEdit]] = {}
+
+    for ref in cte_references:
+        uri = ref.uri
+        if uri not in changes:
+            changes[uri] = []
+
+        # Create a text edit for this reference
+        text_edit = TextEdit(range=ref.range, new_text=new_name)
+        changes[uri].append(text_edit)
+
+    return WorkspaceEdit(changes=changes)
+
+
+def get_document_highlights(
+    lsp_context: LSPContext, document_uri: URI, position: Position
+) -> t.Optional[t.List[DocumentHighlight]]:
+    """
+    Get document highlights for all occurrences of the symbol at the given position.
+
+    This function finds all occurrences of a symbol (CTE) within the current document
+    and returns them as DocumentHighlight objects for "Change All Occurrences" feature.
+
+    Args:
+        lsp_context: The LSP context
+        document_uri: The URI of the document
+        position: The position in the document to find highlights for
+
+    Returns:
+        List of DocumentHighlight objects or None if no symbol found
+    """
+    # Check if there's a CTE at this position
+    cte_references = get_cte_references(lsp_context, document_uri, position)
+    if cte_references:
+        highlights = []
+        for ref in cte_references:
+            # Determine the highlight kind based on whether it's a definition or usage
+            kind = (
+                DocumentHighlightKind.Write
+                if ref.range == ref.target_range
+                else DocumentHighlightKind.Read
+            )
+
+            highlights.append(DocumentHighlight(range=ref.range, kind=kind))
+        return highlights
+
+    # For now, only CTEs are supported
+    return None

{sqlmesh-0.194.2.dev11.dist-info → sqlmesh-0.195.1.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh
-Version: 0.194.2.dev11
+Version: 0.195.1.dev1
 Summary: Next-generation data transformation framework
 Author-email: "TobikoData Inc." <engineering@tobikodata.com>
 License:                                  Apache License

{sqlmesh-0.194.2.dev11.dist-info → sqlmesh-0.195.1.dev1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=QjwaamJIdW7voXoS880yAx7tHm-Im4_mqGA-XJgafos,7631
-sqlmesh/_version.py,sha256=MNQTsCYy-w7J4BscaAPLUKS1LdzZ-X95OrTOYfZBTr8,530
+sqlmesh/_version.py,sha256=KKraTIapru6kL78lgWzBs83rYHvwFHv7dGSun2AsNMA,528
 sqlmesh/magics.py,sha256=d7udR_sieKWzp_YXNFPA2O173xDCqk6u9CIKB7Skb2c,41206
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -172,8 +172,9 @@ sqlmesh/lsp/custom.py,sha256=IJ70li8d5RbOFDrjHfsI2UNOiOp-WB22lrgpdJ7lVSg,2942
 sqlmesh/lsp/description.py,sha256=PjguwyVw5YeYSgT0JMhkf5U1Jry-sDPLvQ2caIR9JPQ,856
 sqlmesh/lsp/helpers.py,sha256=EFc1u3-b7kSv5-tNwmKUDxId72RCLDBnN2lLTgRSzzQ,1020
 sqlmesh/lsp/hints.py,sha256=j-blccBUil94yRoDjWfkSM0HqMmsKD6hP2TaWdWWVk4,3734
-sqlmesh/lsp/main.py,sha256=EanNAS5ArBRc2oYsS3jhL46pfzXLn78xwSW-9R4RQ3o,33731
+sqlmesh/lsp/main.py,sha256=8Lr9Yufq4JpfacVmQpyu2XOoWaswxJur8Kmsn15DpuA,36295
 sqlmesh/lsp/reference.py,sha256=NyfpK12TouDuy5-G7KOhvifkJItMfrakKwnMoU6SiVk,32088
+sqlmesh/lsp/rename.py,sha256=399v1vyVRHHxvbCHVcuS4Pl8K2N-Z60JdZ6e1znOI4s,4397
 sqlmesh/lsp/uri.py,sha256=mvSLbZGmMTHPpliThOtVWiFcHnfJzp1w39q9l-xD2SI,802
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/migrations/v0001_init.py,sha256=h8FwMYJj9_O8yGpk12guaSaGwX5A3we2GiKeHrXi15o,2048
@@ -282,7 +283,7 @@ sqlmesh/utils/pydantic.py,sha256=I898jAFqa3qU6lG2uPpRebVFncOUo8ZrLWM_R3K3DHU,116
 sqlmesh/utils/rich.py,sha256=BpB47G0-JcU41LkGkeGKroexj4NOV5fox4-GOIXPgIc,3545
 sqlmesh/utils/windows.py,sha256=r7Yin4WEwbvH7sKA3hOJtRsclmscNuBeCklCfVPvtQs,611
 sqlmesh/utils/yaml.py,sha256=np-WVcwmwsKghCgAj1OBBMsK1SXSRn4wR-H37cGiFr8,2282
-sqlmesh-0.194.2.dev11.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.195.1.dev1.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=w71qWSLmp-xwmVJO8m41k8Rtni3iZlF0hmhiTr-NeCI,5472
 web/client/dist/assets/Audits-sYYNn2gO.js,sha256=zSAKkif2kU0JuPH05xjNYITNPmsbDlvLKUc2-orr8AY,1041
@@ -400,8 +401,8 @@ web/server/api/endpoints/models.py,sha256=ud-lRwriyI_OYg9tZchrBgeiXELiRTLNEodraf
 web/server/api/endpoints/modules.py,sha256=8hqqgonGay_mJmpCw0IdbjsPhWlQH2VLdKAqha-myac,468
 web/server/api/endpoints/plan.py,sha256=OYuQ_4dgbhG9ibTcX1a8mvYgH3ANxw9Fhbb4wg6QWqc,9222
 web/server/api/endpoints/table_diff.py,sha256=SCy3-pr5x6bEEiOaZGCDbiHQhpgxwmnz3Acqa2Ie6jI,2090
-sqlmesh-0.194.2.dev11.dist-info/METADATA,sha256=oq74Uag6ssiJ86MvYiKIxlOyrCeQLWxuEJI98oMSQ54,26691
-sqlmesh-0.194.2.dev11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sqlmesh-0.194.2.dev11.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.194.2.dev11.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.194.2.dev11.dist-info/RECORD,,
+sqlmesh-0.195.1.dev1.dist-info/METADATA,sha256=RrH8OpiPoLmMtc0v556MegQfFOwgiaRyfUKRxOxXWzI,26690
+sqlmesh-0.195.1.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sqlmesh-0.195.1.dev1.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.195.1.dev1.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.195.1.dev1.dist-info/RECORD,,