15 lines
1019 B
TypeScript
15 lines
1019 B
TypeScript
export type SqlCompletionItemType = "keyword" | "table" | "column" | "snippet" | "function" | "schema" | "variable" | "property" | "text";
|
|
|
|
const COMPLETION_SPACE_BLOCKING_CHARACTERS = new Set([",", ";", ":", ")", "]", "}", "'", '"']);
|
|
|
|
export function appendSqlCompletionSpace(insertText: string, options: { enabled: boolean; itemType: SqlCompletionItemType; nextCharacter?: string }): string {
|
|
if (!options.enabled || options.itemType === "property" || options.itemType === "text" || options.itemType === "schema" || options.itemType === "variable" || options.itemType === "snippet" || options.itemType === "function") return insertText;
|
|
if (!insertText || /\s$/.test(insertText) || insertText.endsWith(".")) return insertText;
|
|
|
|
const nextCharacter = options.nextCharacter ?? "";
|
|
if (/\s/.test(nextCharacter) || COMPLETION_SPACE_BLOCKING_CHARACTERS.has(nextCharacter)) return insertText;
|
|
|
|
// Avoid producing `table AS t,` or `column)` when the cursor is before SQL punctuation.
|
|
return `${insertText} `;
|
|
}
|