From 1b0fd4a8b2af3956f941dcb0ec96b0536697201d Mon Sep 17 00:00:00 2001
From: "wan.jie" <wan.jie@oushu.com>
Date: Fri, 26 Jan 2024 15:48:15 +0800
Subject: [PATCH] feat: export "incomplete" settings in completionService

---
 src/_.contribution.ts   | 2 +-
 src/languageFeatures.ts | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/_.contribution.ts b/src/_.contribution.ts
index 6863cb28..18d43b6e 100644
--- a/src/_.contribution.ts
+++ b/src/_.contribution.ts
@@ -166,7 +166,7 @@ export type CompletionService = (
 	position: Position,
 	completionContext: languages.CompletionContext,
 	suggestions: Suggestions | null
-) => Promise<ICompletionItem[]>;
+) => Promise<ICompletionItem[] | { completionItem: ICompletionItem[], incomplete: boolean }>;
 
 export interface LanguageServiceDefaults {
 	readonly languageId: string;
diff --git a/src/languageFeatures.ts b/src/languageFeatures.ts
index fc95604f..f84bebe5 100644
--- a/src/languageFeatures.ts
+++ b/src/languageFeatures.ts
@@ -170,8 +170,8 @@ export class CompletionAdapter<T extends IWorker> implements languages.Completio
 					position.lineNumber,
 					wordInfo.endColumn
 				);
-
-				const completionItems: languages.CompletionItem[] = completions.map((item) => ({
+				const unwrappedCompletions = Array.isArray(completions) ? completions : completions.completionItem
+				const completionItems: languages.CompletionItem[] = unwrappedCompletions.map((item) => ({
 					...item,
 					insertText:
 						item.insertText ??
@@ -183,7 +183,8 @@ export class CompletionAdapter<T extends IWorker> implements languages.Completio
 				}));
 
 				return {
-					suggestions: completionItems
+					suggestions: completionItems,
+					incomplete: Array.isArray(completions) ? undefined : completions.incomplete
 				};
 			});
 	}