@@ -8,7 +8,7 @@ import { FlinkSQLReferenceProvider } from './Reference';
88import { FlinkSQLRenameProvider } from './Rename' ;
99import { ATNSimulator } from 'antlr4ts/atn/ATNSimulator'
1010import { Token , ParserErrorListener , RecognitionException , Recognizer } from 'antlr4ts' ;
11- // const sqlFormatter = require('sql-formatter-plus');
11+ const sqlFormatter = require ( 'sql-formatter-plus' ) ;
1212
1313interface Config {
1414 indent : string ;
@@ -24,9 +24,9 @@ const config: Config = {
2424 linesBetweenQueries : 2 ,
2525} ;
2626
27- // const format = (text: string): string => {
28- // return sqlFormatter.format(text, config);
29- // };
27+ const format = ( text : string ) : string => {
28+ return sqlFormatter . format ( text , config ) ;
29+ } ;
3030
3131const selector = 'flink-sql'
3232
@@ -53,15 +53,15 @@ export function activate(context: ExtensionContext) {
5353 } )
5454 ) ;
5555
56- // vscode.languages.registerDocumentRangeFormattingEditProvider(selector, {
57- // provideDocumentRangeFormattingEdits: (
58- // document: vscode.TextDocument,
59- // range: vscode.Range,
60- // options: vscode.FormattingOptions
61- // ): vscode.TextEdit[] => [
62- // vscode.TextEdit.replace(range, format(document.getText(range))),
63- // ],
64- // });
56+ vscode . languages . registerDocumentRangeFormattingEditProvider ( selector , {
57+ provideDocumentRangeFormattingEdits : (
58+ document : vscode . TextDocument ,
59+ range : vscode . Range ,
60+ options : vscode . FormattingOptions
61+ ) : vscode . TextEdit [ ] => [
62+ vscode . TextEdit . replace ( range , format ( document . getText ( range ) ) ) ,
63+ ] ,
64+ } ) ;
6565 context . subscriptions . push ( vscode . languages . registerRenameProvider ( selector , new FlinkSQLRenameProvider ( ) ) ) ;
6666
6767 // 注册插件的其他命令和功能...
0 commit comments