spark-sql-language-server 0.0.1-beta.6 → 0.2.0-beta.0

package/out-tsc/schema-registry.js

@@ -1,4 +1,8 @@
 "use strict";
+/* --------------------------------------------------------------------------------------------
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ * ------------------------------------------------------------------------------------------ */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SchemaRegistry = exports.SchemaNodeKind = void 0;
 var SchemaNodeKind;
@@ -9,86 +13,106 @@ var SchemaNodeKind;
     SchemaNodeKind["VIEW"] = "view";
     SchemaNodeKind["COLUMN"] = "column";
     SchemaNodeKind["FUNCTION"] = "function";
-})(SchemaNodeKind = exports.SchemaNodeKind || (exports.SchemaNodeKind = {}));
+})(SchemaNodeKind || (exports.SchemaNodeKind = SchemaNodeKind = {}));
 class SchemaRegistry {
     constructor() {
+        this.defaultDatabases = new Map();
         this._schemaNodes = [];
     }
+    getNodeKey(node) {
+        switch (node.kind) {
+            case SchemaNodeKind.CATALOG:
+                return `${node.kind}:${node.label}`;
+            case SchemaNodeKind.DATABASE:
+                return `${node.kind}:${node.catalog}:${node.label}`;
+            case SchemaNodeKind.TABLE:
+                return `${node.kind}:${node.catalog}:${node.database}:${node.label}`;
+            case SchemaNodeKind.VIEW:
+                return `${node.kind}:${node.catalog}:${node.database}:${node.label}`;
+            case SchemaNodeKind.COLUMN:
+                return `${node.kind}:${node.catalog}:${node.database}:${node.table}:${node.label}`;
+            case SchemaNodeKind.FUNCTION:
+                return `${node.kind}:${node.catalog}:${node.database}:${node.label}`;
+        }
+    }
     getSchemas(kind) {
-        return kind ? this._schemaNodes.filter(n => n.kind === kind) : this._schemaNodes;
+        const nodes = kind ? this._schemaNodes.filter(n => n.kind === kind) : this._schemaNodes;
+        const seen = new Set();
+        return nodes.filter(node => {
+            const key = this.getNodeKey(node);
+            if (seen.has(key)) {
+                return false;
+            }
+            seen.add(key);
+            return true;
+        });
     }
     getSchemaCatalogs() {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.CATALOG);
+        return this.getSchemas(SchemaNodeKind.CATALOG);
     }
     getSchemaDatabases(schema = {}) {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.DATABASE && (schema.catalog === undefined || n.catalog === schema.catalog));
+        const catalog = schema.catalog ?? this.defaultCatalog;
+        return this.getSchemas(SchemaNodeKind.DATABASE).filter(n => catalog === undefined || n.catalog === catalog);
     }
     getSchemaTables(schema = {}) {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.TABLE &&
-            (schema.catalog === undefined || n.catalog === schema.catalog) &&
-            (schema.database === undefined || n.database === schema.database));
+        const catalog = schema.catalog ?? this.defaultCatalog;
+        const database = schema.database ?? (catalog && this.defaultDatabases.get(catalog));
+        return this.getSchemas(SchemaNodeKind.TABLE).filter(n => (catalog === undefined || n.catalog === catalog) && (database === undefined || n.database === database));
     }
     getSchemaColumns(schema = {}) {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.COLUMN &&
-            (schema.catalog === undefined || n.catalog === schema.catalog) &&
-            (schema.database === undefined || n.database === schema.database) &&
+        const catalog = schema.catalog ?? this.defaultCatalog;
+        const database = schema.database ?? (catalog && this.defaultDatabases.get(catalog));
+        return this.getSchemas(SchemaNodeKind.COLUMN).filter(n => (catalog === undefined || n.catalog === catalog) &&
+            (database === undefined || n.database === database) &&
             (schema.table === undefined || n.table === schema.table));
     }
     getSchemaViews(schema = {}) {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.VIEW &&
-            (schema.catalog === undefined || n.catalog === schema.catalog) &&
-            (schema.database === undefined || n.database === schema.database));
-    }
-    getSchemaFunctions(schema = {}) {
-        return this._schemaNodes.filter(n => n.kind === SchemaNodeKind.FUNCTION &&
-            (schema.catalog === undefined || n.catalog === schema.catalog) &&
-            (schema.database === undefined || n.database === schema.database));
+        const catalog = schema.catalog ?? this.defaultCatalog;
+        const database = schema.database ?? (catalog && this.defaultDatabases.get(catalog));
+        return this.getSchemas(SchemaNodeKind.VIEW).filter(n => (catalog === undefined || n.catalog === catalog) && (database === undefined || n.database === database));
     }
     register(registerMode, schemas) {
         if (registerMode === 'set') {
             this._schemaNodes = [];
+            this.defaultCatalog = undefined;
+            this.defaultDatabases = new Map();
         }
         if (schemas.catalogs?.length) {
-            this.registerCatalogs(registerMode, ...schemas.catalogs);
+            this.registerCatalogs(registerMode, schemas.catalogs);
         }
         if (schemas.databases?.length) {
-            this.registerDatabases(registerMode, ...schemas.databases);
+            this.registerDatabases(registerMode, schemas.databases);
         }
         if (schemas.tables?.length) {
-            this.registerTables(registerMode, ...schemas.tables);
+            this.registerTables(registerMode, schemas.tables);
         }
         if (schemas.columns?.length) {
-            this.registerColumns(registerMode, ...schemas.columns);
+            this.registerColumns(registerMode, schemas.columns);
         }
         if (schemas.views?.length) {
-            this.registerViews(registerMode, ...schemas.views);
-        }
-        if (schemas.functions?.length) {
-            this.registerFunctions(registerMode, ...schemas.functions);
+            this.registerViews(registerMode, schemas.views);
         }
     }
-    registerCatalogs(registerMode, ...catalogs) {
+    registerCatalogs(registerMode, catalogs) {
         catalogs.forEach(c => {
             const node = {
                 kind: SchemaNodeKind.CATALOG,
                 label: c.label,
                 description: c.description
             };
-            if (registerMode === 'set') {
+            const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.CATALOG && n.label === c.label);
+            if (!duplicate) {
                 this._schemaNodes.push(node);
             }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.CATALOG && n.label === c.label);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
-                else {
-                    duplicate.description = c.description;
-                }
+            else if (registerMode === 'patch') {
+                duplicate.description = c.description;
+            }
+            if (c.default) {
+                this.defaultCatalog = c.label;
             }
         });
     }
-    registerDatabases(registerMode, ...databases) {
+    registerDatabases(registerMode, databases) {
         databases.forEach(d => {
             const node = {
                 kind: SchemaNodeKind.DATABASE,
@@ -96,21 +120,19 @@ class SchemaRegistry {
                 label: d.label,
                 description: d.description
             };
-            if (registerMode === 'set') {
+            const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.DATABASE && n.label === d.label && n.catalog === d.catalog);
+            if (!duplicate) {
                 this._schemaNodes.push(node);
             }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.DATABASE && n.label === d.label && n.catalog === d.catalog);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
-                else {
-                    duplicate.description = d.description;
-                }
+            else if (registerMode === 'patch') {
+                duplicate.description = d.description;
+            }
+            if (d.default) {
+                this.defaultDatabases.set(d.catalog, d.label);
             }
         });
     }
-    registerTables(registerMode, ...tables) {
+    registerTables(registerMode, tables) {
         tables.forEach(t => {
             const node = {
                 kind: SchemaNodeKind.TABLE,
@@ -119,24 +141,16 @@ class SchemaRegistry {
                 label: t.label,
                 description: t.description
             };
-            if (registerMode === 'set') {
+            const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.TABLE && n.label === t.label && n.catalog === t.catalog && n.database === t.database);
+            if (!duplicate) {
                 this._schemaNodes.push(node);
             }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.TABLE &&
-                    n.label === t.label &&
-                    n.catalog === t.catalog &&
-                    n.database === t.database);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
-                else {
-                    duplicate.description = t.description;
-                }
+            else if (registerMode === 'patch') {
+                duplicate.description = t.description;
             }
         });
     }
-    registerColumns(registerMode, ...columns) {
+    registerColumns(registerMode, columns) {
         columns.forEach(c => {
             const node = {
                 kind: SchemaNodeKind.COLUMN,
@@ -146,25 +160,20 @@ class SchemaRegistry {
                 label: c.label,
                 description: c.description
             };
-            if (registerMode === 'set') {
+            const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.COLUMN &&
+                n.label === c.label &&
+                n.catalog === c.catalog &&
+                n.database === c.database &&
+                n.table === c.table);
+            if (!duplicate) {
                 this._schemaNodes.push(node);
             }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.COLUMN &&
-                    n.label === c.label &&
-                    n.catalog === c.catalog &&
-                    n.database === c.database &&
-                    n.table === c.table);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
-                else {
-                    duplicate.description = c.description;
-                }
+            else if (registerMode === 'patch') {
+                duplicate.description = c.description;
             }
         });
     }
-    registerViews(registerMode, ...views) {
+    registerViews(registerMode, views) {
         views.forEach(v => {
             const node = {
                 kind: SchemaNodeKind.VIEW,
@@ -173,40 +182,12 @@ class SchemaRegistry {
                 label: v.label,
                 description: v.description
             };
-            if (registerMode === 'set') {
-                this._schemaNodes.push(node);
-            }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.VIEW &&
-                    n.label === v.label &&
-                    n.catalog === v.catalog &&
-                    n.database === v.database);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
-            }
-        });
-    }
-    registerFunctions(registerMode, ...functions) {
-        functions.forEach(f => {
-            const node = {
-                kind: SchemaNodeKind.FUNCTION,
-                catalog: f.catalog,
-                database: f.database,
-                label: f.label,
-                description: f.description
-            };
-            if (registerMode === 'set') {
+            const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.VIEW && n.label === v.label && n.catalog === v.catalog && n.database === v.database);
+            if (!duplicate) {
                 this._schemaNodes.push(node);
             }
-            else {
-                const duplicate = this._schemaNodes.find(n => n.kind === SchemaNodeKind.FUNCTION &&
-                    n.label === f.label &&
-                    n.catalog === f.catalog &&
-                    n.database === f.database);
-                if (!duplicate) {
-                    this._schemaNodes.push(node);
-                }
+            else if (registerMode === 'patch') {
+                duplicate.description = v.description;
             }
         });
     }

package/out-tsc/schema-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"schema-registry.js","sourceRoot":"","sources":["../src/schema-registry.ts"],"names":[],"mappings":";;;AAKA,IAAY,cAOX;AAPD,WAAY,cAAc;IACxB,qCAAmB,CAAA;IACnB,uCAAqB,CAAA;IACrB,iCAAe,CAAA;IACf,+BAAa,CAAA;IACb,mCAAiB,CAAA;IACjB,uCAAqB,CAAA;AACvB,CAAC,EAPW,cAAc,GAAd,sBAAc,KAAd,sBAAc,QAOzB;AA+FD,MAAa,cAAc;IAGzB;QACE,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;IACzB,CAAC;IAED,UAAU,CAAC,IAAqB;QAC9B,OAAO,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;IACnF,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,OAAO,CAAoB,CAAC;IAC7F,CAAC;IAED,kBAAkB,CAAC,SAA+B,EAAE;QAClD,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,CACtF,CAAC;IACxB,CAAC;IAED,eAAe,CAAC,SAAkD,EAAE;QAClE,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,KAAK;YAC/B,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC;YAC9D,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,CAAC,CACnD,CAAC;IACrB,CAAC;IAED,gBAAgB,CAAC,SAAkE,EAAE;QACnF,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,MAAM;YAChC,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC;YAC9D,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,CAAC;YACjE,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CACpC,CAAC;IAC3B,CAAC;IAED,cAAc,CAAC,SAAkD,EAAE;QACjE,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,IAAI;YAC9B,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC;YAC9D,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,CAAC,CACpD,CAAC;IACpB,CAAC;IAED,kBAAkB,CAAC,SAAkD,EAAE;QACrE,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAC7B,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ;YAClC,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC;YAC9D,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,MAAM,CAAC,QAAQ,CAAC,CAChD,CAAC;IACxB,CAAC;IAED,QAAQ,CAAC,YAAgC,EAAE,OAA8B;QACvE,IAAI,YAAY,KAAK,KAAK,EAAE;YAC1B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;SACxB;QACD,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE;YAC5B,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;SAC1D;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE;YAC7B,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;SAC5D;QACD,IAAI,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE;YAC1B,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;SACtD;QACD,IAAI,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE;YAC3B,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;SACxD;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE;YACzB,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;SACpD;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE;YAC7B,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;SAC5D;IACH,CAAC;IAEO,gBAAgB,CAAC,YAAgC,EAAE,GAAG,QAAiC;QAC7F,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACnB,MAAM,IAAI,GAAkB;gBAC1B,IAAI,EAAE,cAAc,CAAC,OAAO;gBAC5B,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;gBACxG,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;qBAAM;oBACL,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;iBACvC;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,YAAgC,EAAE,GAAG,SAAmC;QAChG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACpB,MAAM,IAAI,GAAmB;gBAC3B,IAAI,EAAE,cAAc,CAAC,QAAQ;gBAC7B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO,CAC1F,CAAC;gBACF,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;qBAAM;oBACL,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;iBACvC;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,YAAgC,EAAE,GAAG,MAA6B;QACvF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACjB,MAAM,IAAI,GAAgB;gBACxB,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,KAAK;oBAC/B,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;oBACnB,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;oBACvB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,CAC5B,CAAC;gBACF,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;qBAAM;oBACL,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;iBACvC;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,YAAgC,EAAE,GAAG,OAA+B;QAC1F,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAClB,MAAM,IAAI,GAAsB;gBAC9B,IAAI,EAAE,cAAc,CAAC,MAAM;gBAC3B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,MAAM;oBAChC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;oBACnB,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;oBACvB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ;oBACzB,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CACtB,CAAC;gBACF,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;qBAAM;oBACL,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;iBACvC;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,YAAgC,EAAE,GAAG,KAA2B;QACpF,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAChB,MAAM,IAAI,GAAe;gBACvB,IAAI,EAAE,cAAc,CAAC,IAAI;gBACzB,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,IAAI;oBAC9B,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;oBACnB,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;oBACvB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,CAC5B,CAAC;gBACF,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,YAAgC,EAAE,GAAG,SAAmC;QAChG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACpB,MAAM,IAAI,GAAmB;gBAC3B,IAAI,EAAE,cAAc,CAAC,QAAQ;gBAC7B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,EAAE;gBAC1B,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;aAC9B;iBAAM;gBACL,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ;oBAClC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;oBACnB,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;oBACvB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,CAC5B,CAAC;gBACF,IAAI,CAAC,SAAS,EAAE;oBACd,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC9B;aACF;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA1OD,wCA0OC"}
+{"version":3,"file":"schema-registry.js","sourceRoot":"","sources":["../src/schema-registry.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,IAAY,cAOX;AAPD,WAAY,cAAc;IACxB,qCAAmB,CAAA;IACnB,uCAAqB,CAAA;IACrB,iCAAe,CAAA;IACf,+BAAa,CAAA;IACb,mCAAiB,CAAA;IACjB,uCAAqB,CAAA;AACvB,CAAC,EAPW,cAAc,8BAAd,cAAc,QAOzB;AAmGD,MAAa,cAAc;IAKzB;QAFQ,qBAAgB,GAAwB,IAAI,GAAG,EAAE,CAAC;QAGxD,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;IACzB,CAAC;IAEO,UAAU,CAAC,IAAgB;QACjC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,cAAc,CAAC,OAAO;gBACzB,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACtC,KAAK,cAAc,CAAC,QAAQ;gBAC1B,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACtD,KAAK,cAAc,CAAC,KAAK;gBACvB,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACvE,KAAK,cAAc,CAAC,IAAI;gBACtB,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACvE,KAAK,cAAc,CAAC,MAAM;gBACxB,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACrF,KAAK,cAAc,CAAC,QAAQ;gBAC1B,OAAO,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACzE,CAAC;IACH,CAAC;IAED,UAAU,CAAC,IAAqB;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;QACxF,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAC/B,OAAO,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YACzB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAClC,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClB,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,OAAO,CAAoB,CAAC;IACpE,CAAC;IAED,kBAAkB,CAAC,SAA+B,EAAE;QAClD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QAEtD,OAAQ,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAsB,CAAC,MAAM,CAC1E,CAAC,CAAC,EAAE,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CACpD,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,SAAkD,EAAE;QAClE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QAEpF,OAAQ,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAmB,CAAC,MAAM,CACpE,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAC7G,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,SAAkE,EAAE;QACnF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QACpF,OAAQ,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,MAAM,CAAyB,CAAC,MAAM,CAC3E,CAAC,CAAC,EAAE,CACF,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC;YAChD,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC;YACnD,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CAC3D,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,SAAkD,EAAE;QACjE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,cAAc,CAAC;QACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;QACpF,OAAQ,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,IAAI,CAAkB,CAAC,MAAM,CAClE,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAC7G,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,YAAgC,EAAE,OAA8B;QACvE,IAAI,YAAY,KAAK,KAAK,EAAE,CAAC;YAC3B,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;YAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACpD,CAAC;QACD,IAAI,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC;YAC7B,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YAC9B,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1D,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YAC1B,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,YAAgC,EAAE,QAAiC;QAC1F,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACnB,MAAM,IAAI,GAAkB;gBAC1B,IAAI,EAAE,cAAc,CAAC,OAAO;gBAC5B,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC;YACxG,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;gBACpC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACxC,CAAC;YACD,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;gBACd,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC;YAChC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,YAAgC,EAAE,SAAmC;QAC7F,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACpB,MAAM,IAAI,GAAmB;gBAC3B,IAAI,EAAE,cAAc,CAAC,QAAQ;gBAC7B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,QAAQ,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO,CAC1F,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;gBACpC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACxC,CAAC;YACD,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;gBACd,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,YAAgC,EAAE,MAA6B;QACpF,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACjB,MAAM,IAAI,GAAgB;gBACxB,IAAI,EAAE,cAAc,CAAC,KAAK;gBAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,CACjH,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;gBACpC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,YAAgC,EAAE,OAA+B;QACvF,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAClB,MAAM,IAAI,GAAsB;gBAC9B,IAAI,EAAE,cAAc,CAAC,MAAM;gBAC3B,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,MAAM;gBAChC,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK;gBACnB,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;gBACvB,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ;gBACzB,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,CACtB,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;gBACpC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,YAAgC,EAAE,KAA2B;QACjF,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAChB,MAAM,IAAI,GAAe;gBACvB,IAAI,EAAE,cAAc,CAAC,IAAI;gBACzB,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;gBACpB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CACtC,CAAC,CAAC,EAAE,CACF,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,IAAI,IAAI,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,QAAQ,CAChH,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,YAAY,KAAK,OAAO,EAAE,CAAC;gBACpC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAlND,wCAkNC"}

package/out-tsc/server-worker.d.ts

@@ -1 +1,2 @@
 export {};
+//# sourceMappingURL=server-worker.d.ts.map

package/out-tsc/server-worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server-worker.d.ts","sourceRoot":"","sources":["../src/server-worker.ts"],"names":[],"mappings":""}

package/out-tsc/server-worker.js

@@ -1,34 +1,63 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+/* --------------------------------------------------------------------------------------------
+ * Copyright (c) Microsoft Corporation. All rights reserved.
+ * Licensed under the MIT License. See License.txt in the project root for license information.
+ * ------------------------------------------------------------------------------------------ */
 const vscode_languageserver_1 = require("vscode-languageserver");
 const browser_1 = require("vscode-languageserver/browser");
 const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
-const built_in_functions_1 = require("./assets/built-in-functions");
-const constant_1 = require("./constant");
-const execute_commands_1 = require("./execute-commands");
+const constants_1 = require("./constants");
+const execute_command_1 = require("./execute-command");
 const lsp_server_1 = require("./lsp-server");
+const protocol_translation_1 = require("./protocol-translation");
+const typings_1 = require("./typings");
+/* browser specific setup code */
 const messageReader = new browser_1.BrowserMessageReader(self);
 const messageWriter = new browser_1.BrowserMessageWriter(self);
 const connection = (0, browser_1.createConnection)(messageReader, messageWriter);
 connection.onInitialize(() => {
     const capabilities = {
         textDocumentSync: vscode_languageserver_1.TextDocumentSyncKind.Incremental,
-        completionProvider: {
-            triggerCharacters: '.1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'.split(''),
-            resolveProvider: true
-        },
         executeCommandProvider: {
-            commands: [constant_1.VALIDATION_ON_COMMAND, constant_1.REGISTER_SCHEMAS_COMMAND]
+            commands: [
+                constants_1.EXTRACT_SQL_STRUCTURE_COMMAND,
+                constants_1.REGISTER_SCHEMAS_COMMAND,
+                constants_1.PREVIEW_SCRIPT_COMMAND,
+                constants_1.EXTRACT_SCHEMA_CONTEXTS_COMMAND,
+                constants_1.COLLECT_TOKENS_COMMAND,
+                constants_1.COLUMN_LEVEL_LINEAGE_COMMAND
+            ]
         },
-        documentFormattingProvider: true,
+        referencesProvider: true,
         hoverProvider: true,
-        foldingRangeProvider: true
+        foldingRangeProvider: true,
+        renameProvider: true,
+        codeActionProvider: true,
+        documentFormattingProvider: true,
+        completionProvider: {
+            triggerCharacters: '.1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'.split(''),
+            resolveProvider: true
+        }
     };
     return { capabilities };
 });
 const documents = new vscode_languageserver_1.TextDocuments(vscode_languageserver_textdocument_1.TextDocument);
 const lspServer = new lsp_server_1.LSPServer();
 let validationOn = false;
+// 折叠配置（默认值）
+let foldingOptions = { level: 'all', cteDependencyChain: false };
+// 监听配置变更
+connection.onDidChangeConfiguration(change => {
+    const settings = change.settings?.['lsp-web-extension-spark-sql'];
+    if (settings?.folding) {
+        foldingOptions = {
+            level: settings.folding.level || 'all',
+            cteDependencyChain: settings.folding.cteDependencyChain || false
+        };
+        console.log('[LSP] Folding options updated:', foldingOptions);
+    }
+});
 documents.listen(connection);
 documents.onDidClose(e => {
     lspServer.warehouse.removeParsingValue(e.document.uri);
@@ -44,14 +73,19 @@ documents.onDidChangeContent(change => {
         }
         lspServer.warehouse.setParsingValue(change.document);
     }
-    if (validationOn) {
-        validateTextDocument(change.document);
+});
+connection.onDidChangeWatchedFiles(() => {
+    // Monitored files have change in VSCode
+});
+connection.onRequest(vscode_languageserver_1.DocumentDiagnosticRequest.type, params => {
+    const textDocument = documents.get(params.textDocument.uri);
+    if (!textDocument) {
+        console.error('Language Server: TextDocument not found.');
+        return { kind: 'full', items: [] };
     }
+    const items = lspServer.doValidation(textDocument);
+    return { kind: 'full', items };
 });
-function validateTextDocument(textDocument) {
-    const diagnostics = lspServer.doValidation(textDocument);
-    connection.sendDiagnostics({ uri: textDocument.uri, diagnostics }).then();
-}
 connection.onCompletion((params) => {
     const { textDocument, position } = params;
     const document = documents.get(textDocument.uri);
@@ -59,35 +93,49 @@ connection.onCompletion((params) => {
         console.error('Language Server: TextDocument not found.');
         return [];
     }
-    return lspServer.doCompletion(document, position);
+    return lspServer.doCompletion(document, position).completionItems;
 });
 connection.onCompletionResolve((item) => {
     if (!item.detail) {
-        item.detail = 'keyword';
+        item.detail = 'reserved keyword';
     }
     return item;
 });
-connection.onRequest(vscode_languageserver_1.DocumentDiagnosticRequest.type, params => {
-    const textDocument = documents.get(params.textDocument.uri);
-    if (!textDocument) {
-        console.error('Language Server: TextDocument not found.');
-        return { kind: 'full', items: [] };
+connection.onCodeAction((params) => {
+    const actions = [];
+    const diagnostics = params.context.diagnostics;
+    if (!diagnostics.length) {
+        return [];
     }
-    const items = lspServer.doValidation(textDocument);
-    return { kind: 'full', items };
-});
-connection.onExecuteCommand(params => {
-    if (params.command === constant_1.VALIDATION_ON_COMMAND) {
-        validationOn = !!params.arguments?.[0];
-        if (!validationOn) {
-            documents.all().forEach(d => {
-                connection.sendDiagnostics({ uri: d.uri, diagnostics: [] }).then();
-            });
-        }
+    const document = documents.get(params.textDocument.uri);
+    if (!document) {
+        console.error('Language Server: TextDocument not found.');
+        return [];
     }
-    else {
-        (0, execute_commands_1.handleExecuteCommand)(documents, lspServer, params);
+    const kinds = params.context.only;
+    if (kinds?.includes(vscode_languageserver_1.CodeActionKind.QuickFix)) {
+        diagnostics.forEach(diagnostic => {
+            const quickFixes = diagnostic.data?.suggestions
+                .filter((s) => s !== 'EOF')
+                .map((newText) => ({
+                title: `Cannot find name '${document.getText(diagnostic.range)}'. Do you mean '${newText}'?`,
+                kind: vscode_languageserver_1.CodeActionKind.QuickFix,
+                diagnostics: [diagnostic],
+                edit: {
+                    changes: {
+                        [params.textDocument.uri]: [
+                            {
+                                newText,
+                                range: diagnostic.range
+                            }
+                        ]
+                    }
+                }
+            })) || [];
+            actions.push(...quickFixes);
+        });
     }
+    return actions;
 });
 connection.onDocumentFormatting((params) => {
     const document = documents.get(params.textDocument.uri);
@@ -95,7 +143,42 @@ connection.onDocumentFormatting((params) => {
         console.error('Language Server: TextDocument not found.');
         return [];
     }
-    return lspServer.doFormatting(document, params.options);
+    // Merge default options with client-provided options (consistent with formatSQL)
+    const formattingOptions = {
+        tabSize: params.options?.tabSize ?? 2,
+        insertSpaces: params.options?.insertSpaces ?? true,
+        maxParenthesisLength: params.options?.maxParenthesisLength ?? typings_1.DEFAULT_MAX_PARENTHESIS_LENGTH
+    };
+    return lspServer.doFormatting(document, formattingOptions);
+});
+connection.onReferences((params) => {
+    const document = documents.get(params.textDocument.uri);
+    if (!document) {
+        console.error('Language Server: TextDocument not found.');
+        return [];
+    }
+    const semanticToken = lspServer
+        .collectTokens(document)
+        .semanticTokens.find(t => (0, protocol_translation_1.positionInRange)(document, params.position, t.range));
+    if (!semanticToken) {
+        return [];
+    }
+    let references = lspServer.doReferences(document, params.position).map(reference => {
+        return {
+            uri: document.uri,
+            range: reference
+        };
+    });
+    documents
+        .all()
+        .filter(doc => doc.uri !== params.textDocument.uri)
+        .forEach(doc => {
+        const semanticTokens = lspServer.collectTokens(doc).semanticTokens;
+        references = references.concat(semanticTokens
+            .filter(t => (0, lsp_server_1.isSemanticTokenReference)(t, semanticToken))
+            .map(t => ({ uri: doc.uri, range: t.range })));
+    });
+    return references;
 });
 connection.onHover((params) => {
     const document = documents.get(params.textDocument.uri);
@@ -108,12 +191,7 @@ connection.onHover((params) => {
     if (token?.text === undefined) {
         return { contents };
     }
-    const functionItem = built_in_functions_1.BuiltInFunctions.find(f => token.text.localeCompare(f.expr.replace('()', ''), undefined, { sensitivity: 'accent' }) === 0);
-    if (functionItem) {
-        contents.push(`**System (Built-in) Function**`);
-        contents.push(functionItem.description.value);
-        contents.push(`For more information, please refer to [Spark Document](https://spark.apache.org/docs/latest/sql-ref-functions-builtin.html).`);
-    }
+    // TODO: Add Spark SQL function documentation
     return {
         contents
     };
@@ -124,7 +202,44 @@ connection.onFoldingRanges((params) => {
         console.error('Language Server: TextDocument not found.');
         return [];
     }
-    return lspServer.doFoldingRanges(document);
+    return lspServer.doFoldingRanges(document, foldingOptions);
+});
+connection.onRenameRequest((params) => {
+    const document = documents.get(params.textDocument.uri);
+    if (!document) {
+        console.error('Language Server: TextDocument not found.');
+        return;
+    }
+    const semanticToken = lspServer
+        .collectTokens(document)
+        .semanticTokens.find(t => (0, protocol_translation_1.positionInRange)(document, params.position, t.range));
+    if (!semanticToken) {
+        return;
+    }
+    const workspaceEdit = { changes: {} };
+    documents.all().forEach(doc => {
+        const tokens = lspServer.collectTokens(doc).semanticTokens.filter(t => (0, lsp_server_1.isSemanticTokenReference)(t, semanticToken));
+        workspaceEdit.changes[doc.uri] = tokens.map(t => ({
+            newText: params.newName,
+            range: t.range
+        }));
+    });
+    return workspaceEdit;
+});
+connection.onExecuteCommand(params => {
+    if (params.command === constants_1.VALIDATION_ON_COMMAND) {
+        validationOn = !!params.arguments?.[0];
+        if (!validationOn) {
+            documents.all().forEach(d => {
+                connection.sendDiagnostics({ uri: d.uri, diagnostics: [] }).then();
+            });
+        }
+    }
+    else {
+        // 其他操作
+        (0, execute_command_1.handleExecuteCommand)(documents, lspServer, params);
+    }
 });
+// Listen on the connection
 connection.listen();
 //# sourceMappingURL=server-worker.js.map