fix(vscode): support aliased references (#4309)

Author: benfdking

sqlmesh/lsp/reference.py

@@ -55,13 +55,20 @@ def get_model_definitions_for_a_path(
         depends_on = model.depends_on
 
         # Normalize the table reference
-        reference_name = table.sql(dialect=model.dialect)
-        normalized_reference_name = normalize_model_name(
-            reference_name,
-            default_catalog=lint_context.context.default_catalog,
-            dialect=model.dialect,
-        )
-        if normalized_reference_name not in depends_on:
+        unaliased = table.copy()
+        if unaliased.args.get("alias") is not None:
+            unaliased.set("alias", None)
+        reference_name = unaliased.sql(dialect=model.dialect)
+        try:
+            normalized_reference_name = normalize_model_name(
+                reference_name,
+                default_catalog=lint_context.context.default_catalog,
+                dialect=model.dialect,
+            )
+            if normalized_reference_name not in depends_on:
+                continue
+        except Exception:
+            # Skip references that cannot be normalized
             continue
 
         # Get the referenced model uri

tests/lsp/test_reference.py

@@ -46,3 +46,20 @@ def test_reference() -> None:
             referenced_text += read_file[line_num]
         referenced_text += read_file[end_line][:end_character]  # Last line up to end_character
     assert referenced_text == "sushi.customers"
+
+
+@pytest.mark.fast
+def test_reference_with_alias() -> None:
+    context = Context(paths=["examples/sushi"])
+    lsp_context = LSPContext(context)
+
+    waiter_revenue_by_day_uri = next(
+        uri for uri, models in lsp_context.map.items() if "sushi.waiter_revenue_by_day" in models
+    )
+
+    references = get_model_definitions_for_a_path(lsp_context, waiter_revenue_by_day_uri)
+    assert len(references) == 3
+
+    assert references[0].uri.endswith("orders.py")
+    assert references[1].uri.endswith("order_items.py")
+    assert references[2].uri.endswith("items.py")