spark-sql-language-server 0.0.1-beta.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +30 -0
- package/out/sparksql-server-worker.js +2 -0
- package/out/sparksql-server-worker.js.map +1 -0
- package/out-tsc/assets/built-in-functions.d.ts +9 -0
- package/out-tsc/assets/built-in-functions.js +3996 -0
- package/out-tsc/assets/built-in-functions.js.map +1 -0
- package/out-tsc/auto-completion.d.ts +4 -0
- package/out-tsc/auto-completion.js +74 -0
- package/out-tsc/auto-completion.js.map +1 -0
- package/out-tsc/constant.d.ts +2 -0
- package/out-tsc/constant.js +6 -0
- package/out-tsc/constant.js.map +1 -0
- package/out-tsc/index.d.ts +3 -0
- package/out-tsc/index.js +20 -0
- package/out-tsc/index.js.map +1 -0
- package/out-tsc/lib/SparkSqlLexer.d.ts +421 -0
- package/out-tsc/lib/SparkSqlLexer.js +2501 -0
- package/out-tsc/lib/SparkSqlLexer.js.map +1 -0
- package/out-tsc/lib/SparkSqlParser.d.ts +5888 -0
- package/out-tsc/lib/SparkSqlParser.js +32320 -0
- package/out-tsc/lib/SparkSqlParser.js.map +1 -0
- package/out-tsc/lib/SparkSqlParserListener.d.ts +1089 -0
- package/out-tsc/lib/SparkSqlParserListener.js +3 -0
- package/out-tsc/lib/SparkSqlParserListener.js.map +1 -0
- package/out-tsc/lib/SparkSqlParserVisitor.d.ts +727 -0
- package/out-tsc/lib/SparkSqlParserVisitor.js +3 -0
- package/out-tsc/lib/SparkSqlParserVisitor.js.map +1 -0
- package/out-tsc/listeners/parse-error.listener.d.ts +11 -0
- package/out-tsc/listeners/parse-error.listener.js +48 -0
- package/out-tsc/listeners/parse-error.listener.js.map +1 -0
- package/out-tsc/listeners/statement.listener.d.ts +11 -0
- package/out-tsc/listeners/statement.listener.js +44 -0
- package/out-tsc/listeners/statement.listener.js.map +1 -0
- package/out-tsc/lsp-server.d.ts +15 -0
- package/out-tsc/lsp-server.js +81 -0
- package/out-tsc/lsp-server.js.map +1 -0
- package/out-tsc/parsing-warehouse.d.ts +19 -0
- package/out-tsc/parsing-warehouse.js +63 -0
- package/out-tsc/parsing-warehouse.js.map +1 -0
- package/out-tsc/protocol-translation.d.ts +6 -0
- package/out-tsc/protocol-translation.js +128 -0
- package/out-tsc/protocol-translation.js.map +1 -0
- package/out-tsc/public-apis.d.ts +1 -0
- package/out-tsc/public-apis.js +6 -0
- package/out-tsc/public-apis.js.map +1 -0
- package/out-tsc/server-worker.d.ts +1 -0
- package/out-tsc/server-worker.js +126 -0
- package/out-tsc/server-worker.js.map +1 -0
- package/package.json +39 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"SparkSqlParserVisitor.js","sourceRoot":"","sources":["../../src/lib/SparkSqlParserVisitor.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { ANTLRErrorListener, Recognizer, Token } from 'antlr4ts';
|
|
2
|
+
import { RecognitionException } from 'antlr4ts/RecognitionException';
|
|
3
|
+
import { Diagnostic } from 'vscode-languageserver';
|
|
4
|
+
import { TextDocument } from 'vscode-languageserver-textdocument';
|
|
5
|
+
export declare class ParseErrorListener implements ANTLRErrorListener<Token> {
|
|
6
|
+
private document;
|
|
7
|
+
private _errors;
|
|
8
|
+
constructor(document: TextDocument);
|
|
9
|
+
getErrors(): Diagnostic[];
|
|
10
|
+
syntaxError(recognizer: Recognizer<Token, any>, offendingSymbol: Token | undefined, line: number, charPositionInLine: number, _: string, e: RecognitionException | undefined): void;
|
|
11
|
+
}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ParseErrorListener = void 0;
|
|
4
|
+
const SparkSqlParser_1 = require("../lib/SparkSqlParser");
|
|
5
|
+
const protocol_translation_1 = require("../protocol-translation");
|
|
6
|
+
class ParseErrorListener {
|
|
7
|
+
constructor(document) {
|
|
8
|
+
this.document = document;
|
|
9
|
+
this._errors = [];
|
|
10
|
+
}
|
|
11
|
+
getErrors() {
|
|
12
|
+
return this._errors;
|
|
13
|
+
}
|
|
14
|
+
syntaxError(recognizer, offendingSymbol, line, charPositionInLine, _, e) {
|
|
15
|
+
const suggestions = e?.expectedTokens
|
|
16
|
+
?.toArray()
|
|
17
|
+
.map(t => {
|
|
18
|
+
return SparkSqlParser_1.SparkSqlParser.VOCABULARY.getDisplayName(t).replace(/'/g, '') || '';
|
|
19
|
+
})
|
|
20
|
+
.filter(w => !!w) || [];
|
|
21
|
+
const toMessage = (word) => {
|
|
22
|
+
let message = `SQL parse failed. A problem is encountered around "${word}" at line ${line}, column ${charPositionInLine + 1}. `;
|
|
23
|
+
if (suggestions.length) {
|
|
24
|
+
message += `Do you mean:\n ${suggestions.map(s => `"${s}"`).join(', ')}`;
|
|
25
|
+
}
|
|
26
|
+
else {
|
|
27
|
+
message += `There are no viable suggestions.`;
|
|
28
|
+
}
|
|
29
|
+
return message;
|
|
30
|
+
};
|
|
31
|
+
if (offendingSymbol?.text !== undefined) {
|
|
32
|
+
this._errors.push({
|
|
33
|
+
message: toMessage(offendingSymbol.text),
|
|
34
|
+
severity: (0, protocol_translation_1.toDiagnosticSeverity)('error'),
|
|
35
|
+
source: 'Language Server',
|
|
36
|
+
range: {
|
|
37
|
+
start: this.document.positionAt(offendingSymbol.startIndex),
|
|
38
|
+
end: this.document.positionAt(offendingSymbol.stopIndex + 1)
|
|
39
|
+
},
|
|
40
|
+
data: {
|
|
41
|
+
suggestions
|
|
42
|
+
}
|
|
43
|
+
});
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
exports.ParseErrorListener = ParseErrorListener;
|
|
48
|
+
//# sourceMappingURL=parse-error.listener.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"parse-error.listener.js","sourceRoot":"","sources":["../../src/listeners/parse-error.listener.ts"],"names":[],"mappings":";;;AASA,0DAAuD;AACvD,kEAA+D;AAE/D,MAAa,kBAAkB;IAG7B,YAAoB,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;QAFlC,YAAO,GAAiB,EAAE,CAAC;IAEU,CAAC;IAE9C,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,WAAW,CACT,UAAkC,EAClC,eAAkC,EAClC,IAAY,EACZ,kBAA0B,EAC1B,CAAS,EACT,CAAmC;QAEnC,MAAM,WAAW,GACf,CAAC,EAAE,cAAc;YACf,EAAE,OAAO,EAAE;aACV,GAAG,CAAC,CAAC,CAAC,EAAE;YACP,OAAO,+BAAc,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7E,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAE5B,MAAM,SAAS,GAAG,CAAC,IAAY,EAAU,EAAE;YACzC,IAAI,OAAO,GAAG,sDAAsD,IAAI,aAAa,IAAI,YACvF,kBAAkB,GAAG,CACvB,IAAI,CAAC;YACL,IAAI,WAAW,CAAC,MAAM,EAAE;gBACtB,OAAO,IAAI,mBAAmB,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;aAC3E;iBAAM;gBACL,OAAO,IAAI,kCAAkC,CAAC;aAC/C;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;QACF,IAAI,eAAe,EAAE,IAAI,KAAK,SAAS,EAAE;YACvC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;gBAChB,OAAO,EAAE,SAAS,CAAC,eAAe,CAAC,IAAI,CAAC;gBACxC,QAAQ,EAAE,IAAA,2CAAoB,EAAC,OAAO,CAAC;gBACvC,MAAM,EAAE,iBAAiB;gBACzB,KAAK,EAAE;oBACL,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,eAAe,CAAC,UAAU,CAAC;oBAC3D,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,eAAe,CAAC,SAAS,GAAG,CAAC,CAAC;iBAC7D;gBACD,IAAI,EAAE;oBACJ,WAAW;iBACZ;aACF,CAAC,CAAC;SACJ;IACH,CAAC;CACF;AAnDD,gDAmDC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import * as lsp from 'vscode-languageserver';
|
|
2
|
+
import { TextDocument } from 'vscode-languageserver-textdocument';
|
|
3
|
+
import { StatementForFoldingContext } from '../lib/SparkSqlParser';
|
|
4
|
+
import { SparkSqlParserListener } from '../lib/SparkSqlParserListener';
|
|
5
|
+
export declare class StatementListener implements SparkSqlParserListener {
|
|
6
|
+
private document;
|
|
7
|
+
private ranges;
|
|
8
|
+
constructor(document: TextDocument);
|
|
9
|
+
getRanges(): lsp.Range[];
|
|
10
|
+
enterStatementForFolding(ctx: StatementForFoldingContext): void;
|
|
11
|
+
}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
19
|
+
if (mod && mod.__esModule) return mod;
|
|
20
|
+
var result = {};
|
|
21
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
22
|
+
__setModuleDefault(result, mod);
|
|
23
|
+
return result;
|
|
24
|
+
};
|
|
25
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
26
|
+
exports.StatementListener = void 0;
|
|
27
|
+
const lsp = __importStar(require("vscode-languageserver"));
|
|
28
|
+
class StatementListener {
|
|
29
|
+
constructor(document) {
|
|
30
|
+
this.document = document;
|
|
31
|
+
this.ranges = [];
|
|
32
|
+
}
|
|
33
|
+
getRanges() {
|
|
34
|
+
return this.ranges;
|
|
35
|
+
}
|
|
36
|
+
enterStatementForFolding(ctx) {
|
|
37
|
+
const startOffset = ctx.start.startIndex;
|
|
38
|
+
const stopOffset = ctx.stop ? ctx.stop.stopIndex + 1 : startOffset + ctx.text.length;
|
|
39
|
+
const range = lsp.Range.create(this.document.positionAt(ctx.start.startIndex), this.document.positionAt(stopOffset));
|
|
40
|
+
this.ranges.push(range);
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
exports.StatementListener = StatementListener;
|
|
44
|
+
//# sourceMappingURL=statement.listener.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"statement.listener.js","sourceRoot":"","sources":["../../src/listeners/statement.listener.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,2DAA6C;AAM7C,MAAa,iBAAiB;IAG5B,YAAoB,QAAsB;QAAtB,aAAQ,GAAR,QAAQ,CAAc;QAFlC,WAAM,GAAgB,EAAE,CAAC;IAEY,CAAC;IAE9C,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,wBAAwB,CAAC,GAA+B;QACtD,MAAM,WAAW,GAAG,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC;QACzC,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC;QACrF,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAC5B,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EAC9C,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,CACrC,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;CACF;AAlBD,8CAkBC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { Token } from 'antlr4ts/Token';
|
|
2
|
+
import { CompletionItem, Diagnostic } from 'vscode-languageserver';
|
|
3
|
+
import * as lsp from 'vscode-languageserver';
|
|
4
|
+
import { Position, TextDocument, TextEdit } from 'vscode-languageserver-textdocument';
|
|
5
|
+
import { ParsingWarehouse } from './parsing-warehouse';
|
|
6
|
+
export declare class LSPServer {
|
|
7
|
+
private readonly _warehouse;
|
|
8
|
+
constructor();
|
|
9
|
+
get warehouse(): ParsingWarehouse;
|
|
10
|
+
doValidation(document: TextDocument): Diagnostic[];
|
|
11
|
+
doCompletion(document: TextDocument, position: Position): CompletionItem[];
|
|
12
|
+
doFormatting(document: TextDocument, options: lsp.FormattingOptions): TextEdit[];
|
|
13
|
+
doFoldingRanges(document: TextDocument): lsp.FoldingRange[];
|
|
14
|
+
getTokenByPosition(document: TextDocument, position: lsp.Position): Token | undefined;
|
|
15
|
+
}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
19
|
+
if (mod && mod.__esModule) return mod;
|
|
20
|
+
var result = {};
|
|
21
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
22
|
+
__setModuleDefault(result, mod);
|
|
23
|
+
return result;
|
|
24
|
+
};
|
|
25
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
26
|
+
exports.LSPServer = void 0;
|
|
27
|
+
const tree_1 = require("antlr4ts/tree");
|
|
28
|
+
const sql_formatter_1 = require("sql-formatter");
|
|
29
|
+
const lsp = __importStar(require("vscode-languageserver"));
|
|
30
|
+
const auto_completion_1 = require("./auto-completion");
|
|
31
|
+
const statement_listener_1 = require("./listeners/statement.listener");
|
|
32
|
+
const parsing_warehouse_1 = require("./parsing-warehouse");
|
|
33
|
+
const protocol_translation_1 = require("./protocol-translation");
|
|
34
|
+
class LSPServer {
|
|
35
|
+
constructor() {
|
|
36
|
+
this._warehouse = new parsing_warehouse_1.ParsingWarehouse();
|
|
37
|
+
}
|
|
38
|
+
get warehouse() {
|
|
39
|
+
return this._warehouse;
|
|
40
|
+
}
|
|
41
|
+
doValidation(document) {
|
|
42
|
+
const parsingValue = this._warehouse.getParsingValue(document);
|
|
43
|
+
return parsingValue.errors;
|
|
44
|
+
}
|
|
45
|
+
doCompletion(document, position) {
|
|
46
|
+
const parsingValue = this._warehouse.getParsingValue(document);
|
|
47
|
+
return (0, auto_completion_1.handleAutoCompletion)(document, position, parsingValue);
|
|
48
|
+
}
|
|
49
|
+
doFormatting(document, options) {
|
|
50
|
+
const parsingValue = this._warehouse.getParsingValue(document);
|
|
51
|
+
try {
|
|
52
|
+
const formattedCode = (0, sql_formatter_1.formatDialect)(parsingValue.origin, {
|
|
53
|
+
dialect: sql_formatter_1.spark,
|
|
54
|
+
tabWidth: 2,
|
|
55
|
+
useTabs: false,
|
|
56
|
+
keywordCase: 'preserve',
|
|
57
|
+
linesBetweenQueries: 1,
|
|
58
|
+
newlineBeforeSemicolon: true
|
|
59
|
+
});
|
|
60
|
+
const textEdit = lsp.TextEdit.replace(lsp.Range.create(0, 0, document.lineCount, 0), formattedCode);
|
|
61
|
+
return [textEdit];
|
|
62
|
+
}
|
|
63
|
+
catch (e) {
|
|
64
|
+
return [];
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
doFoldingRanges(document) {
|
|
68
|
+
const parsingValue = this._warehouse.getParsingValue(document);
|
|
69
|
+
const listener = new statement_listener_1.StatementListener(document);
|
|
70
|
+
tree_1.ParseTreeWalker.DEFAULT.walk(listener, parsingValue.tree);
|
|
71
|
+
const ranges = listener.getRanges();
|
|
72
|
+
return ranges.map(range => (0, protocol_translation_1.toFoldingRange)(range));
|
|
73
|
+
}
|
|
74
|
+
getTokenByPosition(document, position) {
|
|
75
|
+
const parsingValue = this._warehouse.getParsingValue(document);
|
|
76
|
+
const offset = document.offsetAt(position);
|
|
77
|
+
return parsingValue.tokenStream.getTokens().find(t => t.startIndex <= offset && offset <= t.stopIndex + 1);
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
exports.LSPServer = LSPServer;
|
|
81
|
+
//# sourceMappingURL=lsp-server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lsp-server.js","sourceRoot":"","sources":["../src/lsp-server.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,wCAAgD;AAChD,iDAAqD;AAErD,2DAA6C;AAG7C,uDAAyD;AAEzD,uEAAmE;AACnE,2DAAuD;AACvD,iEAAwD;AAExD,MAAa,SAAS;IAGpB;QACE,IAAI,CAAC,UAAU,GAAG,IAAI,oCAAgB,EAAE,CAAC;IAC3C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,YAAY,CAAC,QAAsB;QACjC,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/D,OAAO,YAAY,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,QAAsB,EAAE,QAAkB;QACrD,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/D,OAAO,IAAA,sCAAoB,EAAC,QAAQ,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;IAChE,CAAC;IAED,YAAY,CAAC,QAAsB,EAAE,OAA8B;QACjE,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/D,IAAI;YAEF,MAAM,aAAa,GAAG,IAAA,6BAAa,EAAC,YAAY,CAAC,MAAM,EAAE;gBACvD,OAAO,EAAE,qBAAK;gBACd,QAAQ,EAAE,CAAC;gBACX,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,UAAU;gBACvB,mBAAmB,EAAE,CAAC;gBACtB,sBAAsB,EAAE,IAAI;aAC7B,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAiB,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;YAIlH,OAAO,CAAC,QAAQ,CAAC,CAAC;SACnB;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,EAAE,CAAC;SACX;IACH,CAAC;IAED,eAAe,CAAC,QAAsB;QACpC,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAE/D,MAAM,QAAQ,GAAG,IAAI,sCAAiB,CAAC,QAAQ,CAAC,CAAC;QACjD,sBAAe,CAAC,OAAO,CAAC,IAAI,CAAyB,QAAQ,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;QACpC,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAA,qCAAc,EAAC,KAAK,CAAC,CAAC,CAAC;IACpD,CAAC;IAED,kBAAkB,CAAC,QAAsB,EAAE,QAAsB;QAC/D,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC3C,OAAO,YAAY,CAAC,WAAW,CAAC,SAAS,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,MAAM,IAAI,MAAM,IAAI,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;IAC7G,CAAC;CACF;AA1DD,8BA0DC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { CommonTokenStream } from 'antlr4ts';
|
|
2
|
+
import * as lsp from 'vscode-languageserver';
|
|
3
|
+
import { Range, TextDocument } from 'vscode-languageserver-textdocument';
|
|
4
|
+
import { ProgramContext, SparkSqlParser } from './lib/SparkSqlParser';
|
|
5
|
+
export interface ParseResult {
|
|
6
|
+
origin: string;
|
|
7
|
+
tokenStream: CommonTokenStream;
|
|
8
|
+
parser: SparkSqlParser;
|
|
9
|
+
tree: ProgramContext;
|
|
10
|
+
errors: lsp.Diagnostic[];
|
|
11
|
+
}
|
|
12
|
+
export declare class ParsingWarehouse {
|
|
13
|
+
private _warehouse;
|
|
14
|
+
setParsingValue(document: TextDocument): void;
|
|
15
|
+
parseDocument(document: TextDocument, range?: Range): ParseResult;
|
|
16
|
+
hasParsingValue(document: TextDocument): boolean;
|
|
17
|
+
getParsingValue(document: TextDocument): ParseResult;
|
|
18
|
+
removeParsingValue(documentUri: string): void;
|
|
19
|
+
}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.ParsingWarehouse = void 0;
|
|
4
|
+
const antlr4ts_1 = require("antlr4ts");
|
|
5
|
+
const atn_1 = require("antlr4ts/atn");
|
|
6
|
+
const SparkSqlLexer_1 = require("./lib/SparkSqlLexer");
|
|
7
|
+
const SparkSqlParser_1 = require("./lib/SparkSqlParser");
|
|
8
|
+
const parse_error_listener_1 = require("./listeners/parse-error.listener");
|
|
9
|
+
class ParsingWarehouse {
|
|
10
|
+
constructor() {
|
|
11
|
+
this._warehouse = new Map();
|
|
12
|
+
}
|
|
13
|
+
setParsingValue(document) {
|
|
14
|
+
const parseResult = this.parseDocument(document);
|
|
15
|
+
this._warehouse.set(document.uri, {
|
|
16
|
+
...parseResult
|
|
17
|
+
});
|
|
18
|
+
}
|
|
19
|
+
parseDocument(document, range) {
|
|
20
|
+
const text = range ? document.getText(range) : document.getText();
|
|
21
|
+
const charStream = antlr4ts_1.CharStreams.fromString(text.toUpperCase());
|
|
22
|
+
const lexer = new SparkSqlLexer_1.SparkSqlLexer(charStream);
|
|
23
|
+
const tokenStream = new antlr4ts_1.CommonTokenStream(lexer);
|
|
24
|
+
const parser = new SparkSqlParser_1.SparkSqlParser(tokenStream);
|
|
25
|
+
const errorListener = new parse_error_listener_1.ParseErrorListener(document);
|
|
26
|
+
lexer.removeErrorListeners();
|
|
27
|
+
parser.removeErrorListeners();
|
|
28
|
+
let tree;
|
|
29
|
+
parser.errorHandler = new antlr4ts_1.BailErrorStrategy();
|
|
30
|
+
parser.interpreter.setPredictionMode(atn_1.PredictionMode.SLL);
|
|
31
|
+
try {
|
|
32
|
+
tree = parser.program();
|
|
33
|
+
}
|
|
34
|
+
catch (e) {
|
|
35
|
+
parser.reset();
|
|
36
|
+
parser.errorHandler = new antlr4ts_1.DefaultErrorStrategy();
|
|
37
|
+
parser.addErrorListener(errorListener);
|
|
38
|
+
parser.interpreter.setPredictionMode(atn_1.PredictionMode.LL);
|
|
39
|
+
tree = parser.program();
|
|
40
|
+
}
|
|
41
|
+
return {
|
|
42
|
+
origin: text,
|
|
43
|
+
tokenStream,
|
|
44
|
+
parser,
|
|
45
|
+
tree,
|
|
46
|
+
errors: errorListener.getErrors()
|
|
47
|
+
};
|
|
48
|
+
}
|
|
49
|
+
hasParsingValue(document) {
|
|
50
|
+
return this._warehouse.has(document.uri);
|
|
51
|
+
}
|
|
52
|
+
getParsingValue(document) {
|
|
53
|
+
if (!this._warehouse.has(document.uri)) {
|
|
54
|
+
this.setParsingValue(document);
|
|
55
|
+
}
|
|
56
|
+
return this._warehouse.get(document.uri);
|
|
57
|
+
}
|
|
58
|
+
removeParsingValue(documentUri) {
|
|
59
|
+
this._warehouse.delete(documentUri);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
exports.ParsingWarehouse = ParsingWarehouse;
|
|
63
|
+
//# sourceMappingURL=parsing-warehouse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"parsing-warehouse.js","sourceRoot":"","sources":["../src/parsing-warehouse.ts"],"names":[],"mappings":";;;AAIA,uCAAmG;AACnG,sCAA8C;AAI9C,uDAAoD;AACpD,yDAAsE;AACtE,2EAAsE;AAUtE,MAAa,gBAAgB;IAA7B;QACU,eAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAyDtD,CAAC;IAvDC,eAAe,CAAC,QAAsB;QACpC,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QACjD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE;YAChC,GAAG,WAAW;SACf,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,QAAsB,EAAE,KAAa;QACjD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QAClE,MAAM,UAAU,GAAG,sBAAW,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,IAAI,6BAAa,CAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,WAAW,GAAG,IAAI,4BAAiB,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,+BAAc,CAAC,WAAW,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,IAAI,yCAAkB,CAAC,QAAQ,CAAC,CAAC;QACvD,KAAK,CAAC,oBAAoB,EAAE,CAAC;QAC7B,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAC9B,IAAI,IAAoB,CAAC;QACzB,MAAM,CAAC,YAAY,GAAG,IAAI,4BAAiB,EAAE,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,iBAAiB,CAAC,oBAAc,CAAC,GAAG,CAAC,CAAC;QACzD,IAAI;YAEF,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;SACzB;QAAC,OAAO,CAAC,EAAE;YAEV,MAAM,CAAC,KAAK,EAAE,CAAC;YAEf,MAAM,CAAC,YAAY,GAAG,IAAI,+BAAoB,EAAE,CAAC;YACjD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAEvC,MAAM,CAAC,WAAW,CAAC,iBAAiB,CAAC,oBAAc,CAAC,EAAE,CAAC,CAAC;YACxD,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;SACzB;QACD,OAAO;YACL,MAAM,EAAE,IAAI;YACZ,WAAW;YACX,MAAM;YACN,IAAI;YACJ,MAAM,EAAE,aAAa,CAAC,SAAS,EAAE;SAClC,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,QAAsB;QACpC,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,eAAe,CAAC,QAAsB;QACpC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YACtC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;SAChC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAE,CAAC;IAC5C,CAAC;IAED,kBAAkB,CAAC,WAAmB;QACpC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IACtC,CAAC;CACF;AA1DD,4CA0DC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { CompletionItem, DiagnosticSeverity } from 'vscode-languageserver';
|
|
2
|
+
import * as lsp from 'vscode-languageserver';
|
|
3
|
+
import { MarkupContent } from 'vscode-languageserver-types';
|
|
4
|
+
export declare function toDiagnosticSeverity(category: string): DiagnosticSeverity;
|
|
5
|
+
export declare function toCompletionItem(type: 'catalog' | 'database' | 'table' | 'column' | 'view' | 'function' | 'built-in', label: string, insertText: string, detail: string, documentation?: string | MarkupContent): CompletionItem;
|
|
6
|
+
export declare function toFoldingRange(range: lsp.Range, kind?: string): lsp.FoldingRange;
|
|
@@ -0,0 +1,128 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || function (mod) {
|
|
19
|
+
if (mod && mod.__esModule) return mod;
|
|
20
|
+
var result = {};
|
|
21
|
+
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
|
|
22
|
+
__setModuleDefault(result, mod);
|
|
23
|
+
return result;
|
|
24
|
+
};
|
|
25
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
26
|
+
exports.toFoldingRange = exports.toCompletionItem = exports.toDiagnosticSeverity = void 0;
|
|
27
|
+
const vscode_languageserver_1 = require("vscode-languageserver");
|
|
28
|
+
const lsp = __importStar(require("vscode-languageserver"));
|
|
29
|
+
function toDiagnosticSeverity(category) {
|
|
30
|
+
switch (category) {
|
|
31
|
+
case 'error':
|
|
32
|
+
return vscode_languageserver_1.DiagnosticSeverity.Error;
|
|
33
|
+
case 'warning':
|
|
34
|
+
return vscode_languageserver_1.DiagnosticSeverity.Warning;
|
|
35
|
+
case 'suggestion':
|
|
36
|
+
return vscode_languageserver_1.DiagnosticSeverity.Hint;
|
|
37
|
+
default:
|
|
38
|
+
return vscode_languageserver_1.DiagnosticSeverity.Error;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
exports.toDiagnosticSeverity = toDiagnosticSeverity;
|
|
42
|
+
function toCompletionItem(type, label, insertText, detail, documentation) {
|
|
43
|
+
switch (type) {
|
|
44
|
+
case 'catalog':
|
|
45
|
+
return {
|
|
46
|
+
kind: lsp.CompletionItemKind.Class,
|
|
47
|
+
label,
|
|
48
|
+
insertText,
|
|
49
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
50
|
+
sortText: `AAA${label}`,
|
|
51
|
+
documentation,
|
|
52
|
+
detail,
|
|
53
|
+
data: 'catalog'
|
|
54
|
+
};
|
|
55
|
+
case 'database':
|
|
56
|
+
return {
|
|
57
|
+
kind: lsp.CompletionItemKind.Enum,
|
|
58
|
+
label,
|
|
59
|
+
insertText,
|
|
60
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
61
|
+
sortText: `BAA${label}`,
|
|
62
|
+
documentation,
|
|
63
|
+
detail,
|
|
64
|
+
data: 'database'
|
|
65
|
+
};
|
|
66
|
+
case 'table':
|
|
67
|
+
return {
|
|
68
|
+
kind: lsp.CompletionItemKind.EnumMember,
|
|
69
|
+
label,
|
|
70
|
+
insertText,
|
|
71
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
72
|
+
sortText: `CBA${label}`,
|
|
73
|
+
documentation,
|
|
74
|
+
detail,
|
|
75
|
+
data: 'table'
|
|
76
|
+
};
|
|
77
|
+
case 'column':
|
|
78
|
+
return {
|
|
79
|
+
kind: lsp.CompletionItemKind.Field,
|
|
80
|
+
label,
|
|
81
|
+
insertText,
|
|
82
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
83
|
+
sortText: `DAA${label}`,
|
|
84
|
+
documentation,
|
|
85
|
+
detail,
|
|
86
|
+
data: 'column'
|
|
87
|
+
};
|
|
88
|
+
case 'view':
|
|
89
|
+
return {
|
|
90
|
+
kind: lsp.CompletionItemKind.EnumMember,
|
|
91
|
+
label,
|
|
92
|
+
insertText,
|
|
93
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
94
|
+
sortText: `CAA${label}`,
|
|
95
|
+
documentation,
|
|
96
|
+
detail,
|
|
97
|
+
data: 'view'
|
|
98
|
+
};
|
|
99
|
+
case 'function':
|
|
100
|
+
return {
|
|
101
|
+
kind: lsp.CompletionItemKind.Property,
|
|
102
|
+
label,
|
|
103
|
+
insertText: `\`${insertText}\``,
|
|
104
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
105
|
+
sortText: `EAA${label}`,
|
|
106
|
+
documentation,
|
|
107
|
+
detail,
|
|
108
|
+
data: 'function'
|
|
109
|
+
};
|
|
110
|
+
case 'built-in':
|
|
111
|
+
return {
|
|
112
|
+
kind: lsp.CompletionItemKind.Function,
|
|
113
|
+
label,
|
|
114
|
+
insertText,
|
|
115
|
+
insertTextFormat: lsp.InsertTextFormat.Snippet,
|
|
116
|
+
sortText: `FAA${label}`,
|
|
117
|
+
documentation,
|
|
118
|
+
detail,
|
|
119
|
+
data: 'function'
|
|
120
|
+
};
|
|
121
|
+
}
|
|
122
|
+
}
|
|
123
|
+
exports.toCompletionItem = toCompletionItem;
|
|
124
|
+
function toFoldingRange(range, kind = lsp.FoldingRangeKind.Region) {
|
|
125
|
+
return lsp.FoldingRange.create(range.start.line, range.end.line, range.start.character, range.end.character, kind);
|
|
126
|
+
}
|
|
127
|
+
exports.toFoldingRange = toFoldingRange;
|
|
128
|
+
//# sourceMappingURL=protocol-translation.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"protocol-translation.js","sourceRoot":"","sources":["../src/protocol-translation.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAKA,iEAA2E;AAC3E,2DAA6C;AAG7C,SAAgB,oBAAoB,CAAC,QAAgB;IACnD,QAAQ,QAAQ,EAAE;QAChB,KAAK,OAAO;YACV,OAAO,0CAAkB,CAAC,KAAK,CAAC;QAClC,KAAK,SAAS;YACZ,OAAO,0CAAkB,CAAC,OAAO,CAAC;QACpC,KAAK,YAAY;YACf,OAAO,0CAAkB,CAAC,IAAI,CAAC;QACjC;YACE,OAAO,0CAAkB,CAAC,KAAK,CAAC;KACnC;AACH,CAAC;AAXD,oDAWC;AAED,SAAgB,gBAAgB,CAC9B,IAAoF,EACpF,KAAa,EACb,UAAkB,EAClB,MAAc,EACd,aAAsC;IAEtC,QAAQ,IAAI,EAAE;QACZ,KAAK,SAAS;YACZ,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,KAAK;gBAClC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,SAAS;aAChB,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,IAAI;gBACjC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,UAAU;aACjB,CAAC;QACJ,KAAK,OAAO;YACV,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,UAAU;gBACvC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,OAAO;aACd,CAAC;QACJ,KAAK,QAAQ;YACX,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,KAAK;gBAClC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,QAAQ;aACf,CAAC;QACJ,KAAK,MAAM;YACT,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,UAAU;gBACvC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,MAAM;aACb,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,QAAQ;gBACrC,KAAK;gBACL,UAAU,EAAE,KAAK,UAAU,IAAI;gBAC/B,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,UAAU;aACjB,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,kBAAkB,CAAC,QAAQ;gBACrC,KAAK;gBACL,UAAU;gBACV,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,CAAC,OAAO;gBAC9C,QAAQ,EAAE,MAAM,KAAK,EAAE;gBACvB,aAAa;gBACb,MAAM;gBACN,IAAI,EAAE,UAAU;aACjB,CAAC;KACL;AACH,CAAC;AAtFD,4CAsFC;AAED,SAAgB,cAAc,CAAC,KAAgB,EAAE,IAAI,GAAG,GAAG,CAAC,gBAAgB,CAAC,MAAM;IACjF,OAAO,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;AACrH,CAAC;AAFD,wCAEC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export { LSPServer as SparkSQLServer } from './lsp-server';
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.SparkSQLServer = void 0;
|
|
4
|
+
var lsp_server_1 = require("./lsp-server");
|
|
5
|
+
Object.defineProperty(exports, "SparkSQLServer", { enumerable: true, get: function () { return lsp_server_1.LSPServer; } });
|
|
6
|
+
//# sourceMappingURL=public-apis.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"public-apis.js","sourceRoot":"","sources":["../src/public-apis.ts"],"names":[],"mappings":";;;AAAA,2CAA2D;AAAlD,4GAAA,SAAS,OAAkB"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
const vscode_languageserver_1 = require("vscode-languageserver");
|
|
4
|
+
const browser_1 = require("vscode-languageserver/browser");
|
|
5
|
+
const vscode_languageserver_textdocument_1 = require("vscode-languageserver-textdocument");
|
|
6
|
+
const built_in_functions_1 = require("./assets/built-in-functions");
|
|
7
|
+
const constant_1 = require("./constant");
|
|
8
|
+
const lsp_server_1 = require("./lsp-server");
|
|
9
|
+
const messageReader = new browser_1.BrowserMessageReader(self);
|
|
10
|
+
const messageWriter = new browser_1.BrowserMessageWriter(self);
|
|
11
|
+
const connection = (0, browser_1.createConnection)(messageReader, messageWriter);
|
|
12
|
+
connection.onInitialize(() => {
|
|
13
|
+
const capabilities = {
|
|
14
|
+
textDocumentSync: vscode_languageserver_1.TextDocumentSyncKind.Incremental,
|
|
15
|
+
completionProvider: {
|
|
16
|
+
triggerCharacters: '.1234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'.split(''),
|
|
17
|
+
resolveProvider: true
|
|
18
|
+
},
|
|
19
|
+
executeCommandProvider: {
|
|
20
|
+
commands: [constant_1.VALIDATION_ON_COMMAND]
|
|
21
|
+
},
|
|
22
|
+
documentFormattingProvider: true,
|
|
23
|
+
hoverProvider: true,
|
|
24
|
+
foldingRangeProvider: true
|
|
25
|
+
};
|
|
26
|
+
return { capabilities };
|
|
27
|
+
});
|
|
28
|
+
const documents = new vscode_languageserver_1.TextDocuments(vscode_languageserver_textdocument_1.TextDocument);
|
|
29
|
+
const lspServer = new lsp_server_1.LSPServer();
|
|
30
|
+
let validationOn = false;
|
|
31
|
+
documents.listen(connection);
|
|
32
|
+
documents.onDidClose(e => {
|
|
33
|
+
lspServer.warehouse.removeParsingValue(e.document.uri);
|
|
34
|
+
});
|
|
35
|
+
documents.onDidChangeContent(change => {
|
|
36
|
+
if (!lspServer.warehouse.hasParsingValue(change.document)) {
|
|
37
|
+
lspServer.warehouse.setParsingValue(change.document);
|
|
38
|
+
}
|
|
39
|
+
else {
|
|
40
|
+
const parseValueCache = lspServer.warehouse.getParsingValue(change.document);
|
|
41
|
+
if (change.document.getText() === parseValueCache.origin) {
|
|
42
|
+
return;
|
|
43
|
+
}
|
|
44
|
+
lspServer.warehouse.setParsingValue(change.document);
|
|
45
|
+
}
|
|
46
|
+
if (validationOn) {
|
|
47
|
+
validateTextDocument(change.document);
|
|
48
|
+
}
|
|
49
|
+
});
|
|
50
|
+
function validateTextDocument(textDocument) {
|
|
51
|
+
const diagnostics = lspServer.doValidation(textDocument);
|
|
52
|
+
connection.sendDiagnostics({ uri: textDocument.uri, diagnostics }).then();
|
|
53
|
+
}
|
|
54
|
+
connection.onCompletion((params) => {
|
|
55
|
+
const { textDocument, position } = params;
|
|
56
|
+
const document = documents.get(textDocument.uri);
|
|
57
|
+
if (!document) {
|
|
58
|
+
console.error('Language Server: TextDocument not found.');
|
|
59
|
+
return [];
|
|
60
|
+
}
|
|
61
|
+
return lspServer.doCompletion(document, position);
|
|
62
|
+
});
|
|
63
|
+
connection.onCompletionResolve((item) => {
|
|
64
|
+
if (!item.detail) {
|
|
65
|
+
item.detail = 'keyword';
|
|
66
|
+
}
|
|
67
|
+
return item;
|
|
68
|
+
});
|
|
69
|
+
connection.onRequest(vscode_languageserver_1.DocumentDiagnosticRequest.type, params => {
|
|
70
|
+
const textDocument = documents.get(params.textDocument.uri);
|
|
71
|
+
if (!textDocument) {
|
|
72
|
+
console.error('Language Server: TextDocument not found.');
|
|
73
|
+
return { kind: 'full', items: [] };
|
|
74
|
+
}
|
|
75
|
+
const items = lspServer.doValidation(textDocument);
|
|
76
|
+
return { kind: 'full', items };
|
|
77
|
+
});
|
|
78
|
+
connection.onExecuteCommand(params => {
|
|
79
|
+
if (params.command === constant_1.VALIDATION_ON_COMMAND) {
|
|
80
|
+
validationOn = !!params.arguments?.[0];
|
|
81
|
+
if (!validationOn) {
|
|
82
|
+
documents.all().forEach(d => {
|
|
83
|
+
connection.sendDiagnostics({ uri: d.uri, diagnostics: [] }).then();
|
|
84
|
+
});
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
});
|
|
88
|
+
connection.onDocumentFormatting((params) => {
|
|
89
|
+
const document = documents.get(params.textDocument.uri);
|
|
90
|
+
if (!document) {
|
|
91
|
+
console.error('Language Server: TextDocument not found.');
|
|
92
|
+
return [];
|
|
93
|
+
}
|
|
94
|
+
return lspServer.doFormatting(document, params.options);
|
|
95
|
+
});
|
|
96
|
+
connection.onHover((params) => {
|
|
97
|
+
const document = documents.get(params.textDocument.uri);
|
|
98
|
+
const contents = [];
|
|
99
|
+
if (!document) {
|
|
100
|
+
console.error('Language Server: TextDocument not found.');
|
|
101
|
+
return { contents };
|
|
102
|
+
}
|
|
103
|
+
const token = lspServer.getTokenByPosition(document, params.position);
|
|
104
|
+
if (token?.text === undefined) {
|
|
105
|
+
return { contents };
|
|
106
|
+
}
|
|
107
|
+
const functionItem = built_in_functions_1.BuiltInFunctions.find(f => token.text.localeCompare(f.expr.replace('()', ''), undefined, { sensitivity: 'accent' }) === 0);
|
|
108
|
+
if (functionItem) {
|
|
109
|
+
contents.push(`**System (Built-in) Function**`);
|
|
110
|
+
contents.push(functionItem.description.value);
|
|
111
|
+
contents.push(`For more information, please refer to [Spark Document](https://spark.apache.org/docs/latest/sql-ref-functions-builtin.html).`);
|
|
112
|
+
}
|
|
113
|
+
return {
|
|
114
|
+
contents
|
|
115
|
+
};
|
|
116
|
+
});
|
|
117
|
+
connection.onFoldingRanges((params) => {
|
|
118
|
+
const document = documents.get(params.textDocument.uri);
|
|
119
|
+
if (!document) {
|
|
120
|
+
console.error('Language Server: TextDocument not found.');
|
|
121
|
+
return [];
|
|
122
|
+
}
|
|
123
|
+
return lspServer.doFoldingRanges(document);
|
|
124
|
+
});
|
|
125
|
+
connection.listen();
|
|
126
|
+
//# sourceMappingURL=server-worker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server-worker.js","sourceRoot":"","sources":["../src/server-worker.ts"],"names":[],"mappings":";;AAIA,iEAW+B;AAC/B,2DAA6G;AAC7G,2FAA4E;AAE5E,oEAA+D;AAC/D,yCAAmD;AACnD,6CAAyC;AAIzC,MAAM,aAAa,GAAG,IAAI,8BAAoB,CAAC,IAAI,CAAC,CAAC;AACrD,MAAM,aAAa,GAAG,IAAI,8BAAoB,CAAC,IAAI,CAAC,CAAC;AAErD,MAAM,UAAU,GAAG,IAAA,0BAAgB,EAAC,aAAa,EAAE,aAAa,CAAC,CAAC;AAIlE,UAAU,CAAC,YAAY,CAAC,GAAqB,EAAE;IAC7C,MAAM,YAAY,GAAuB;QACvC,gBAAgB,EAAE,4CAAoB,CAAC,WAAW;QAClD,kBAAkB,EAAE;YAClB,iBAAiB,EAAE,iEAAiE,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9F,eAAe,EAAE,IAAI;SACtB;QACD,sBAAsB,EAAE;YACtB,QAAQ,EAAE,CAAC,gCAAqB,CAAC;SAClC;QACD,0BAA0B,EAAE,IAAI;QAChC,aAAa,EAAE,IAAI;QACnB,oBAAoB,EAAE,IAAI;KAC3B,CAAC;IACF,OAAO,EAAE,YAAY,EAAE,CAAC;AAC1B,CAAC,CAAC,CAAC;AAGH,MAAM,SAAS,GAAG,IAAI,qCAAa,CAAC,iDAAY,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,IAAI,sBAAS,EAAE,CAAC;AAClC,IAAI,YAAY,GAAG,KAAK,CAAC;AAEzB,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;AAG7B,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;IACvB,SAAS,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;AACzD,CAAC,CAAC,CAAC;AAIH,SAAS,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE;IACpC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE;QACzD,SAAS,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;KACtD;SAAM;QACL,MAAM,eAAe,GAAG,SAAS,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,eAAe,CAAC,MAAM,EAAE;YACxD,OAAO;SACR;QACD,SAAS,CAAC,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;KACtD;IACD,IAAI,YAAY,EAAE;QAChB,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;KACvC;AACH,CAAC,CAAC,CAAC;AAEH,SAAS,oBAAoB,CAAC,YAA0B;IACtD,MAAM,WAAW,GAAG,SAAS,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;IAGzD,UAAU,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,YAAY,CAAC,GAAG,EAAE,WAAW,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;AAC5E,CAAC;AAGD,UAAU,CAAC,YAAY,CAAC,CAAC,MAAM,EAAoB,EAAE;IACnD,MAAM,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;IAC1C,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACjD,IAAI,CAAC,QAAQ,EAAE;QAEb,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,OAAO,EAAE,CAAC;KACX;IACD,OAAO,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;AACpD,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,mBAAmB,CAAC,CAAC,IAAI,EAAkB,EAAE;IACtD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;QAChB,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;KACzB;IACD,OAAO,IAAI,CAAC;AACd,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,SAAS,CAAC,iDAAyB,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE;IAC5D,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAC5D,IAAI,CAAC,YAAY,EAAE;QACjB,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;KACpC;IACD,MAAM,KAAK,GAAG,SAAS,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;IACnD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAA8B,CAAC;AAC7D,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE;IACnC,IAAI,MAAM,CAAC,OAAO,KAAK,gCAAqB,EAAE;QAC5C,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,CAAC,YAAY,EAAE;YACjB,SAAS,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;gBAC1B,UAAU,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACrE,CAAC,CAAC,CAAC;SACJ;KACF;AACH,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,oBAAoB,CAAC,CAAC,MAAM,EAAc,EAAE;IACrD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACxD,IAAI,CAAC,QAAQ,EAAE;QACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,OAAO,EAAE,CAAC;KACX;IACD,OAAO,SAAS,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;AAC1D,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,OAAO,CAAC,CAAC,MAAM,EAAS,EAAE;IACnC,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACxD,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,IAAI,CAAC,QAAQ,EAAE;QACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,OAAO,EAAE,QAAQ,EAAE,CAAC;KACrB;IACD,MAAM,KAAK,GAAG,SAAS,CAAC,kBAAkB,CAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IAEtE,IAAI,KAAK,EAAE,IAAI,KAAK,SAAS,EAAE;QAC7B,OAAO,EAAE,QAAQ,EAAE,CAAC;KACrB;IAED,MAAM,YAAY,GAAG,qCAAgB,CAAC,IAAI,CACxC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAK,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,CACrG,CAAC;IACF,IAAI,YAAY,EAAE;QAChB,QAAQ,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAChD,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC9C,QAAQ,CAAC,IAAI,CACX,8HAA8H,CAC/H,CAAC;KACH;IACD,OAAO;QACL,QAAQ;KACT,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,UAAU,CAAC,eAAe,CAAC,CAAC,MAAM,EAAkB,EAAE;IACpD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACxD,IAAI,CAAC,QAAQ,EAAE;QACb,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC1D,OAAO,EAAE,CAAC;KACX;IACD,OAAO,SAAS,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;AAC7C,CAAC,CAAC,CAAC;AAGH,UAAU,CAAC,MAAM,EAAE,CAAC"}
|