feat: support intelligent completions always visible (#4039)

Author: Ricbet

packages/ai-native/src/browser/contrib/inline-completions/inline-completions.handler.ts

@@ -11,7 +11,7 @@ import {
   Sequencer,
   runWhenIdle,
 } from '@opensumi/ide-core-common';
-import { EditorSelectionChangeEvent, IEditor } from '@opensumi/ide-editor/lib/browser';
+import { EditorSelectionChangeEvent } from '@opensumi/ide-editor/lib/browser';
 import { ICodeEditor, InlineCompletions, Position, Range } from '@opensumi/ide-monaco';
 import { monacoApi } from '@opensumi/ide-monaco/lib/browser/monaco-api';
 import { InlineCompletionContextKeys } from '@opensumi/monaco-editor-core/esm/vs/editor/contrib/inlineCompletions/browser/inlineCompletionContextKeys';

packages/ai-native/src/browser/contrib/intelligent-completions/intelligent-completions.controller.ts

@@ -5,10 +5,14 @@ import {
   IAICompletionOption,
   IDisposable,
   IntelligentCompletionsRegistryToken,
+  isDefined,
   runWhenIdle,
 } from '@opensumi/ide-core-common';
 import { ICodeEditor, ICursorPositionChangedEvent, IRange, ITextModel, Range } from '@opensumi/ide-monaco';
 import { empty } from '@opensumi/ide-utils/lib/strings';
+import { IObservable, autorun } from '@opensumi/monaco-editor-core/esm/vs/base/common/observable';
+import { GhostTextWidget } from '@opensumi/monaco-editor-core/esm/vs/editor/contrib/inlineCompletions/browser/ghostTextWidget';
+import { InlineCompletionsController } from '@opensumi/monaco-editor-core/esm/vs/editor/contrib/inlineCompletions/browser/inlineCompletionsController';
 
 import { AINativeContextKey } from '../../contextkey/ai-native.contextkey.service';
 import { REWRITE_DECORATION_INLINE_ADD, RewriteWidget } from '../../widget/rewrite/rewrite-widget';
@@ -25,6 +29,7 @@ import { IIntelligentCompletionsResult } from './intelligent-completions';
 import { IntelligentCompletionsRegistry } from './intelligent-completions.feature.registry';
 import { MultiLineDecorationModel } from './multi-line.decoration';
 
+
 export class IntelligentCompletionsController extends BaseAIMonacoEditorController {
   public static readonly ID = 'editor.contrib.ai.intelligent.completions';
 
@@ -36,20 +41,22 @@ export class IntelligentCompletionsController extends BaseAIMonacoEditorControll
     return this.injector.get(IntelligentCompletionsRegistryToken);
   }
 
-  private multiLineDecorationModel: MultiLineDecorationModel;
-  private additionsDeletionsDecorationModel: AdditionsDeletionsDecorationModel;
-
-  private aiNativeContextKey: AINativeContextKey;
-
   private get model(): ITextModel {
     return this.monacoEditor.getModel()!;
   }
 
+  private multiLineDecorationModel: MultiLineDecorationModel;
+  private additionsDeletionsDecorationModel: AdditionsDeletionsDecorationModel;
+  private aiNativeContextKey: AINativeContextKey;
   private rewriteWidget: RewriteWidget | null;
   private whenMultiLineEditsVisibleDisposable: Disposable;
 
   public mount(): IDisposable {
     this.whenMultiLineEditsVisibleDisposable = new Disposable();
+    this.multiLineDecorationModel = new MultiLineDecorationModel(this.monacoEditor);
+    this.additionsDeletionsDecorationModel = new AdditionsDeletionsDecorationModel(this.monacoEditor);
+    this.aiNativeContextKey = this.injector.get(AINativeContextKey, [this.monacoEditor.contextKeyService]);
+
     return this.registerFeature(this.monacoEditor);
   }
 
@@ -66,6 +73,12 @@ export class IntelligentCompletionsController extends BaseAIMonacoEditorControll
       return;
     }
 
+    // 如果上一次补全结果还在，则不重复请求
+    const isVisible = this.aiNativeContextKey.multiLineEditsIsVisible.get();
+    if (isVisible) {
+      return;
+    }
+
     const position = this.monacoEditor.getPosition()!;
     const intelligentCompletionModel = await provider(this.monacoEditor, position, bean, this.token);
 
@@ -81,7 +94,7 @@ export class IntelligentCompletionsController extends BaseAIMonacoEditorControll
   }
 
   private applyInlineDecorations(completionModel: IIntelligentCompletionsResult) {
-    const { items } = completionModel;
+    const { items, alwaysVisible } = completionModel;
 
     const position = this.monacoEditor.getPosition()!;
     const model = this.monacoEditor.getModel();
@@ -160,6 +173,27 @@ export class IntelligentCompletionsController extends BaseAIMonacoEditorControll
         }
       }),
     );
+
+    if (isDefined(alwaysVisible) && alwaysVisible) {
+      const inlineCompletionsController = InlineCompletionsController.get(this.monacoEditor);
+      if (inlineCompletionsController) {
+        /**
+         * https://github.com/microsoft/vscode/blob/1.88.1/src/vs/editor/contrib/inlineCompletions/browser/inlineCompletionsController.ts#L67
+         * 当 _ghostTextWidgets 发生变化时，说明是由下拉补全引起的 “内联补全” 字符出现，此时需要将其销毁
+         */
+        this.whenMultiLineEditsVisibleDisposable.addDispose(
+          autorun((reader) => {
+            const ghostTextWidgetsObservable = inlineCompletionsController['_ghostTextWidgets'] as IObservable<
+              readonly GhostTextWidget[]
+            >;
+            ghostTextWidgetsObservable.read(reader);
+
+            const ghostTextWidget = ghostTextWidgetsObservable.get()[0];
+            ghostTextWidget?.dispose();
+          }),
+        );
+      }
+    }
   }
 
   private async renderRewriteWidget(
@@ -241,10 +275,6 @@ export class IntelligentCompletionsController extends BaseAIMonacoEditorControll
   }
 
   public registerFeature(monacoEditor: ICodeEditor): IDisposable {
-    this.multiLineDecorationModel = new MultiLineDecorationModel(monacoEditor);
-    this.additionsDeletionsDecorationModel = new AdditionsDeletionsDecorationModel(monacoEditor);
-    this.aiNativeContextKey = this.injector.get(AINativeContextKey, [monacoEditor.contextKeyService]);
-
     this.addDispose(
       Event.any<any>(
         monacoEditor.onDidChangeCursorPosition,

packages/startup/entry/sample-modules/ai-native/ai-native.contribution.ts

@@ -494,6 +494,7 @@ export class AINativeContribution implements AINativeCoreContribution {
             },
           ],
           enableMultiLine: true,
+          alwaysVisible: true,
         };
       } catch (error) {
         if (error.name === 'AbortError') {