trying to implement lsp

[ci skip]

Author: benfdking

examples/sushi/audits/assert_item_price_above_zero.sql

@@ -6,4 +6,4 @@ AUDIT (
 
 SELECT *
 FROM sushi.items
-WHERE price <= 0
+WHERE price <= 0   

examples/sushi/models/top_waiters.sql

@@ -10,18 +10,18 @@ MODEL (
 
 WITH test_macros AS (
   SELECT
-    @ADD_ONE(1) AS lit_two,
+    @ADD_ONE(1) AS  lit_two,
     @MULTIPLY(revenue, 2.0) AS sql_exp,
     @SQL_LITERAL(revenue::TEXT, 'x', 'y', a, "b") AS sql_lit
   FROM sushi.waiter_revenue_by_day
 )
 SELECT
-  waiter_id::INT AS waiter_id,
+  waiter_id::INT AS waiter_id ,
   revenue::DOUBLE AS revenue
 FROM sushi.waiter_revenue_by_day
 WHERE
   event_date = (
-    SELECT
+    SELECT 
       MAX(event_date)
     FROM sushi.waiter_revenue_by_day
   )

examples/sushi/models/waiter_as_customer_by_day.sql

@@ -23,7 +23,7 @@ JINJA_QUERY_BEGIN;
 
 SELECT
   w.waiter_id as waiter_id,
-  wn.name as waiter_name,
+  wn.name as waiter_name ,
   {{ alias(identity(x), 'flag') }},
   w.event_date as event_date
 FROM sushi.waiters AS w

package-lock.json

@@ -121,11 +121,16 @@ web = [
     "lsprotocol",
     "pygls",
 ]
+lsp = [
+    "pygls",
+    "lsprotocol"
+]
 risingwave = ["psycopg2"]
 
 [project.scripts]
 sqlmesh = "sqlmesh.cli.main:cli"
 sqlmesh_cicd = "sqlmesh.cicd.bot:bot"
+sqlmesh_lsp = "sqlmesh.lsp.main:main"
 
 [project.entry-points."airflow.plugins"]
 sqlmesh_airflow = "sqlmesh.schedulers.airflow.plugin:SqlmeshAirflowPlugin"
@@ -149,7 +154,7 @@ fallback_version = "0.0.0"
 local_scheme = "no-local-version"
 
 [tool.setuptools.packages.find]
-include = ["sqlmesh", "sqlmesh.*", "web*"]
+include = ["sqlmesh", "sqlmesh.*", "web*"] 
 
 [tool.setuptools.package-data]
 web = ["client/dist/**"]

pyproject.toml

@@ -778,7 +778,7 @@ def ui(ctx: click.Context, host: str, port: int, mode: str) -> None:
 @cli_analytics
 def lsp(obj: Context) -> None:
     """Start a language server for SQLMesh."""
-    from web.lsp.main import server
+    from lsp.main import server
 
     server.start_io()
 

sqlmesh/cli/main.py

@@ -215,3 +215,8 @@ async def did_change_watched_files(ls: LanguageServer, params: types.DidChangeWa
                     )
                     updated[context.path] = True
 
+def main():
+    server.start_io()
+
+if __name__ == "__main__":
+    main()

web/lsp/main.py sqlmesh/lsp/main.pyweb/lsp/main.py renamed to sqlmesh/lsp/main.py

@@ -9,19 +9,21 @@
   "categories": [
     "Other"
   ],
-  "activationEvents": [],
+  "activationEvents": [
+    "onLanguage:sql"
+  ],
   "main": "./dist/extension.js",
   "contributes": {
     "viewsContainers": {
-    "panel": [
-      {
-        "id": "lineage_view",
-        "title": "Lineage",
-        "icon": "./assets/images/dag.svg"
-      }
-    ]
+      "panel": [
+        {
+          "id": "lineage_view",
+          "title": "Lineage",
+          "icon": "./assets/images/dag.svg"
+        }
+      ]
     },
-   "views": {
+    "views": {
       "lineage_view": [
         {
           "id": "sqlmesh.lineage",