feat: support custom references and definitions setting

DTStack · Nov 12, 2024 · 540687d · 540687d
1 parent 3d58985
commit 540687d
Show file tree

Hide file tree

Showing 11 changed files with 124 additions and 49 deletions.
diff --git a/src/languageFeatures.ts b/src/languageFeatures.ts
@@ -1,5 +1,7 @@
 import type { ParseError } from 'dt-sql-parser';
-import { EntityContext } from 'dt-sql-parser/dist/parser/common/entityCollector';
+import {
+	EntityContext,
+} from 'dt-sql-parser/dist/parser/common/entityCollector';
 import { WordPosition } from 'dt-sql-parser/dist/parser/common/textAndWord';
 import * as monaco from 'monaco-editor';
 
@@ -13,7 +15,7 @@ import {
 	MarkerSeverity,
 	Position,
 	Range,
-	Uri
+	Uri,
 } from './fillers/monaco-editor-core';
 import type { LanguageServiceDefaults } from './monaco.contribution';
 
@@ -203,17 +205,24 @@ export class CompletionAdapter<T extends BaseSQLWorker>
 }
 
 export class DefinitionAdapter<T extends BaseSQLWorker> implements languages.DefinitionProvider {
-	constructor(private readonly _worker: WorkerAccessor<T>) {}
+	constructor(
+		private readonly _worker: WorkerAccessor<T>,
+		private readonly _defaults: LanguageServiceDefaults) {}
 	provideDefinition(
 		model: editor.IReadOnlyModel,
-		position: Position
+		position: Position,
+		_token: CancellationToken
 	): languages.ProviderResult<languages.Definition | languages.LocationLink[]> {
 		const resource = model.uri;
 		const lineContent = model.getLineContent(position.lineNumber);
 		if (lineContent.startsWith('--')) return null;
 		return this._worker(resource)
 			.then((worker) => {
-				return worker.getAllEntities(model.getValue());
+				let code = model?.getValue() || '';
+				if (typeof this._defaults.preprocessCode === 'function') {
+					code = this._defaults.preprocessCode(code);
+				}
+				return worker.getAllEntities(code);
 			})
 			.then((entities) => {
 				const word = model.getWordAtPosition(position);
@@ -249,17 +258,26 @@ export class DefinitionAdapter<T extends BaseSQLWorker> implements languages.Def
 }
 
 export class ReferenceAdapter<T extends BaseSQLWorker> implements languages.ReferenceProvider {
-	constructor(private readonly _worker: WorkerAccessor<T>) {}
+	constructor(
+		private readonly _worker: WorkerAccessor<T>,
+		private readonly _defaults: LanguageServiceDefaults
+	) {}
 	provideReferences(
 		model: editor.IReadOnlyModel,
-		position: Position
+		position: Position,
+		_context: languages.ReferenceContext,
+		_token: CancellationToken
 	): languages.ProviderResult<languages.Location[]> {
 		const resource = model.uri;
 		const lineContent = model.getLineContent(position.lineNumber);
-		if (!lineContent.startsWith('CREATE')) return null;
+		if (!lineContent.startsWith('CREATE')) return;
 		return this._worker(resource)
 			.then((worker) => {
-				return worker.getAllEntities(model.getValue());
+				let code = model?.getValue() || '';
+				if (typeof this._defaults.preprocessCode === 'function') {
+					code = this._defaults.preprocessCode(code);
+				}
+				return worker.getAllEntities(model?.getValue());
 			})
 			.then((entities) => {
 				const word = model.getWordAtPosition(position);
@@ -279,7 +297,7 @@ export class ReferenceAdapter<T extends BaseSQLWorker> implements languages.Refe
 						});
 					}
 				});
-				return arr.length ? arr : null;
+				return arr;
 			});
 	}
 }
diff --git a/src/languages/flink/flink.contribution.ts b/src/languages/flink/flink.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.FLINK,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.FLINK, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/hive/hive.contribution.ts b/src/languages/hive/hive.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.HIVE,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.HIVE, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/impala/impala.contribution.ts b/src/languages/impala/impala.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.IMPALA,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.IMPALA, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/mysql/mysql.contribution.ts b/src/languages/mysql/mysql.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.MYSQL,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.MYSQL, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/pgsql/pgsql.contribution.ts b/src/languages/pgsql/pgsql.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.PG,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.PG, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/spark/spark.contribution.ts b/src/languages/spark/spark.contribution.ts
@@ -4,8 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.SPARK,
@@ -16,5 +16,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.SPARK, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/languages/trino/trino.contribution.ts b/src/languages/trino/trino.contribution.ts
@@ -1,6 +1,6 @@
 import { registerLanguage } from '../../_.contribution';
-import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 import { LanguageIdEnum } from '../../common/constants';
+import { setupLanguageFeatures } from '../../setupLanguageFeatures';
 
 registerLanguage({
 	id: LanguageIdEnum.TRINO,
@@ -11,5 +11,7 @@ registerLanguage({
 
 setupLanguageFeatures(LanguageIdEnum.TRINO, {
 	completionItems: true,
-	diagnostics: true
+	diagnostics: true,
+	references: true,
+	definitions: true,
 });
diff --git a/src/monaco.contribution.ts b/src/monaco.contribution.ts
@@ -1,5 +1,16 @@
-import { languages, Emitter, IEvent, editor, Position, IRange } from './fillers/monaco-editor-core';
-import { EntityContext, Suggestions } from 'dt-sql-parser';
+import {
+	EntityContext,
+	Suggestions,
+} from 'dt-sql-parser';
+
+import {
+	editor,
+	Emitter,
+	IEvent,
+	IRange,
+	languages,
+	Position,
+} from './fillers/monaco-editor-core';
 
 /**
  * A completion item.
@@ -65,7 +76,7 @@ export interface ModeConfiguration {
 	/**
 	 * Defines whether the built-in definitions provider is enabled.
 	 */
-	// readonly definitions?: boolean;
+	readonly definitions?: boolean;
 
 	/**
 	 * Defines whether the built-in rename provider is enabled.
@@ -75,7 +86,7 @@ export interface ModeConfiguration {
 	/**
 	 * Defines whether the built-in references provider is enabled.
 	 */
-	// readonly references?: boolean;
+	readonly references?: boolean;
 }
 
 /**
@@ -171,5 +182,7 @@ export const modeConfigurationDefault: Required<ModeConfiguration> = {
 		completionService: defaultCompletionService,
 		triggerCharacters: ['.', ' ']
 	},
-	diagnostics: true
+	diagnostics: true,
+	definitions: true,
+	references: true
 };
diff --git a/src/setupLanguageFeatures.ts b/src/setupLanguageFeatures.ts
@@ -1,13 +1,16 @@
+import { LanguageIdEnum } from './common/constants';
 import {
-	PreprocessCode,
+	IDisposable,
+	languages,
+} from './fillers/monaco-editor-core';
+import {
+	CompletionOptions,
 	LanguageServiceDefaults,
 	LanguageServiceDefaultsImpl,
-	modeConfigurationDefault,
 	ModeConfiguration,
-	CompletionOptions
+	modeConfigurationDefault,
+	PreprocessCode,
 } from './monaco.contribution';
-import { languages, IDisposable } from './fillers/monaco-editor-core';
-import { LanguageIdEnum } from './common/constants';
 
 export interface FeatureConfiguration {
 	/**
@@ -20,6 +23,14 @@ export interface FeatureConfiguration {
 	 * Defaults to true.
 	 */
 	diagnostics?: boolean;
+	/**
+	 * Defines whether the built-in definitions provider is enabled.
+	 */
+	definitions?: boolean;
+	/**
+	 * Defines whether the built-in references provider is enabled.
+	 */
+	references?: boolean;
 	/**
 	 * Define a function to preprocess code.
 	 * By default, do not something.
@@ -110,13 +121,23 @@ function processConfiguration(
 			? configuration.completionItems!.triggerCharacters
 			: (defaults?.modeConfiguration.completionItems.triggerCharacters ??
 				modeConfigurationDefault.completionItems.triggerCharacters);
+	const references =
+				typeof configuration.references === 'boolean'
+					? configuration.references
+					: (defaults?.modeConfiguration.references ?? modeConfigurationDefault.references);
+	const definitions =
+					typeof configuration.definitions === 'boolean'
+						? configuration.definitions
+						: (defaults?.modeConfiguration.definitions ?? modeConfigurationDefault.definitions);
 
 	return {
 		diagnostics,
 		completionItems: {
 			enable: completionEnable,
 			completionService,
 			triggerCharacters
-		}
+		},
+		references,
+		definitions,
 	};
 }
diff --git a/src/setupLanguageMode.ts b/src/setupLanguageMode.ts
@@ -1,5 +1,9 @@
 import type { BaseSQLWorker } from './baseSQLWorker';
-import { IDisposable, languages, Uri } from './fillers/monaco-editor-core';
+import {
+	IDisposable,
+	languages,
+	Uri,
+} from './fillers/monaco-editor-core';
 import * as languageFeatures from './languageFeatures';
 import { LanguageServiceDefaults } from './monaco.contribution';
 import { WorkerManager } from './workerManager';
@@ -19,7 +23,6 @@ export function setupLanguageMode<T extends BaseSQLWorker>(
 
 	function registerProviders(): void {
 		const { languageId, modeConfiguration } = defaults;
-
 		disposeAll(providers);
 
 		if (modeConfiguration.diagnostics) {
@@ -34,18 +37,24 @@ export function setupLanguageMode<T extends BaseSQLWorker>(
 				)
 			);
 		}
-		providers.push(
-			languages.registerDefinitionProvider(
-				languageId,
-				new languageFeatures.DefinitionAdapter(worker)
-			)
-		);
-		providers.push(
-			languages.registerReferenceProvider(
-				languageId,
-				new languageFeatures.ReferenceAdapter(worker)
-			)
-		);
+
+		if (modeConfiguration.references) {
+			providers.push(
+				languages.registerReferenceProvider(
+					languageId,
+					new languageFeatures.ReferenceAdapter(worker, defaults)
+				)
+			);
+		}
+
+		if (modeConfiguration.definitions) {
+			providers.push(
+				languages.registerDefinitionProvider(
+					languageId,
+					new languageFeatures.DefinitionAdapter(worker, defaults)
+				)
+			);
+		}
 	}
 
 	registerProviders();