feat: optimize inline diff viewer

packages/ai-native/src/browser/preferences/schema.ts

@@ -1,7 +1,7 @@
 import { AINativeSettingSectionsId, PreferenceSchema } from '@opensumi/ide-core-browser';
 import { localize } from '@opensumi/ide-core-common';
 
-export enum EInlineDifPreviewMode {
+export enum EInlineDiffPreviewMode {
   inlineLive = 'inlineLive',
   sideBySide = 'sideBySide',
 }
@@ -10,12 +10,12 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
   properties: {
     [AINativeSettingSectionsId.InlineDiffPreviewMode]: {
       type: 'string',
-      enum: [EInlineDifPreviewMode.inlineLive, EInlineDifPreviewMode.sideBySide],
+      enum: [EInlineDiffPreviewMode.inlineLive, EInlineDiffPreviewMode.sideBySide],
       enumDescriptions: [
         localize('preference.ai.native.inlineDiff.preview.mode.inlineLive'),
         localize('preference.ai.native.inlineDiff.preview.mode.sideBySide'),
       ],
-      default: EInlineDifPreviewMode.inlineLive,
+      default: EInlineDiffPreviewMode.inlineLive,
     },
     [AINativeSettingSectionsId.InlineChatAutoVisible]: {
       type: 'boolean',

packages/ai-native/src/browser/widget/inline-chat/inline-chat.handler.ts

@@ -28,7 +28,7 @@ import { monacoApi } from '@opensumi/ide-monaco/lib/browser/monaco-api';
 import { ContentWidgetPositionPreference } from '@opensumi/ide-monaco/lib/browser/monaco-exports/editor';
 
 import { CodeActionService } from '../../contrib/code-action/code-action.service';
-import { EInlineDifPreviewMode } from '../../preferences/schema';
+import { EInlineDiffPreviewMode } from '../../preferences/schema';
 import { ERunStrategy } from '../../types';
 import {
   BaseInlineDiffPreviewer,
@@ -310,7 +310,7 @@ export class InlineChatHandler extends Disposable {
     });
   }
 
-  private visibleDiffWidget(
+  visibleDiffWidget(
     monacoEditor: monaco.ICodeEditor,
     options: {
       crossSelection: monaco.Selection;
@@ -325,12 +325,12 @@ export class InlineChatHandler extends Disposable {
     const { crossSelection, chatResponse } = options;
     const { relationId, startTime, isRetry } = reportInfo;
 
-    const inlineDiffMode = this.preferenceService.getValid<EInlineDifPreviewMode>(
+    const inlineDiffMode = this.preferenceService.getValid<EInlineDiffPreviewMode>(
       AINativeSettingSectionsId.InlineDiffPreviewMode,
-      EInlineDifPreviewMode.inlineLive,
+      EInlineDiffPreviewMode.inlineLive,
     );
 
-    if (inlineDiffMode === EInlineDifPreviewMode.sideBySide) {
+    if (inlineDiffMode === EInlineDiffPreviewMode.sideBySide) {
       this.diffPreviewer = this.injector.get(SideBySideInlineDiffWidget, [monacoEditor, crossSelection]);
     } else {
       this.diffPreviewer = this.injector.get(LiveInlineDiffPreviewer, [monacoEditor, crossSelection]);
@@ -441,6 +441,18 @@ export class InlineChatHandler extends Disposable {
     );
   }
 
+  get onPartialEditEvent() {
+    return (this.diffPreviewer as LiveInlineDiffPreviewer).onPartialEditEvent;
+  }
+
+  acceptAllPartialEdits() {
+    this.diffPreviewer.handleAction(EResultKind.ACCEPT);
+  }
+
+  discardAllPartialEdits() {
+    this.diffPreviewer.handleAction(EResultKind.DISCARD);
+  }
+
   private async handleDiffPreviewStrategy(
     monacoEditor: monaco.ICodeEditor,
     strategy: (...arg: any[]) => MaybePromise<ChatResponse | InlineChatController>,

packages/ai-native/src/browser/widget/inline-diff/inline-diff-previewer.ts

@@ -1,5 +1,5 @@
 import { Autowired, INJECTOR_TOKEN, Injectable, Injector } from '@opensumi/di';
-import { Disposable, ErrorResponse, ReplyResponse } from '@opensumi/ide-core-common';
+import { Disposable, ErrorResponse, IDisposable, ReplyResponse } from '@opensumi/ide-core-common';
 import { EOL, ICodeEditor, IPosition, ITextModel, Position, Selection } from '@opensumi/ide-monaco';
 import { ContentWidgetPositionPreference } from '@opensumi/ide-monaco/lib/browser/monaco-exports/editor';
 import { DefaultEndOfLine } from '@opensumi/monaco-editor-core/esm/vs/editor/common/model';
@@ -203,4 +203,7 @@ export class LiveInlineDiffPreviewer extends BaseInlineDiffPreviewer<InlineStrea
     const diffModel = this.node.recompute(EComputerMode.legacy);
     this.node.readyRender(diffModel);
   }
+  get onPartialEditEvent() {
+    return this.node.onPartialEditEvent;
+  }
 }

packages/ai-native/src/browser/widget/inline-stream-diff/inline-stream-diff.handler.tsx

@@ -53,7 +53,7 @@ export class InlineStreamDiffHandler extends Disposable {
   private undoRedoGroup: UndoRedoGroup;
   private partialEditWidgetHandle: (widgets: AcceptPartialEditWidget[]) => void;
 
-  protected readonly _onDidEditChange = new Emitter<void>();
+  protected readonly _onDidEditChange = this.registerDispose(new Emitter<void>());
   public readonly onDidEditChange: Event<void> = this._onDidEditChange.event;
 
   constructor(private readonly monacoEditor: ICodeEditor, private readonly selection: Selection) {
@@ -397,4 +397,8 @@ export class InlineStreamDiffHandler extends Disposable {
     this.pushStackElement();
     this.monacoEditor.focus();
   }
+
+  get onPartialEditEvent() {
+    return this.livePreviewDiffDecorationModel.onPartialEditEvent;
+  }
 }

packages/ai-native/src/browser/widget/inline-stream-diff/live-preview.decoration.tsx

@@ -5,7 +5,7 @@ import ReactDOMClient from 'react-dom/client';
 import { Autowired, INJECTOR_TOKEN, Injectable, Injector } from '@opensumi/di';
 import { KeybindingRegistry, StackingLevel } from '@opensumi/ide-core-browser';
 import { AI_INLINE_DIFF_PARTIAL_EDIT } from '@opensumi/ide-core-browser/lib/ai-native/command';
-import { Disposable, Emitter, Event, isUndefined, uuid } from '@opensumi/ide-core-common';
+import { Disposable, Emitter, Event, IRange, isUndefined, uuid } from '@opensumi/ide-core-common';
 import { ISingleEditOperation } from '@opensumi/ide-editor';
 import { ICodeEditor, IEditorDecorationsCollection, Position, Range, Selection } from '@opensumi/ide-monaco';
 import { ReactInlineContentWidget } from '@opensumi/ide-monaco/lib/browser/ai-native/BaseInlineContentWidget';
@@ -46,6 +46,31 @@ enum EPartialEdit {
   discard = 'discard',
 }
 
+export interface IPartialEditEvent {
+  /**
+   * 总 diff 数
+   */
+  totalPartialEditCount: number;
+  /**
+   * 已采纳数
+   */
+  acceptedPartialEditCount: number;
+
+  /**
+   * 已添加行数
+   */
+  totalAddedLinesCount: number;
+  /**
+   * 已删除行数
+   */
+  totalRemovedLinesCount: number;
+  currentPartialEdit: {
+    type: EPartialEdit;
+    deletedLinesCount: number;
+    addedLinesCount: number;
+  };
+}
+
 interface ITextLinesTokens {
   text: string;
   lineTokens: LineTokens;
@@ -64,10 +89,10 @@ export class AcceptPartialEditWidget extends ReactInlineContentWidget {
   private _id: string;
   private _decorationId: string;
 
-  private readonly _onAccept = new Emitter<void>();
+  private readonly _onAccept = this.registerDispose(new Emitter<void>());
   public readonly onAccept: Event<void> = this._onAccept.event;
 
-  private readonly _onDiscard = new Emitter<void>();
+  private readonly _onDiscard = this.registerDispose(new Emitter<void>());
   public readonly onDiscard: Event<void> = this._onDiscard.event;
 
   positionPreference = [ContentWidgetPositionPreference.EXACT];
@@ -191,6 +216,9 @@ export class LivePreviewDiffDecorationModel extends Disposable {
   @Autowired(InlineStreamDiffService)
   private readonly inlineStreamDiffService: InlineStreamDiffService;
 
+  private readonly _onPartialEditEvent = this.registerDispose(new Emitter<IPartialEditEvent>());
+  public readonly onPartialEditEvent: Event<IPartialEditEvent> = this._onPartialEditEvent.event;
+
   private zoneDec: IEditorDecorationsCollection;
 
   private activeLineDec: IEditorDecorationsCollection;
@@ -432,6 +460,18 @@ export class LivePreviewDiffDecorationModel extends Disposable {
       findRemovedWidget?.resume();
     };
 
+    const event: IPartialEditEvent = {
+      totalPartialEditCount: this.partialEditWidgetList.length,
+      acceptedPartialEditCount: this.partialEditWidgetList.filter((w) => w.isHidden).length,
+      totalAddedLinesCount: 0,
+      totalRemovedLinesCount: 0,
+      currentPartialEdit: {
+        addedLinesCount: 0,
+        deletedLinesCount: 0,
+        type,
+      },
+    };
+
     /**
      * 将 partial widget 的所有操作和代码变更放在单独的 undo/redo 堆栈组里面
      */
@@ -470,6 +510,8 @@ export class LivePreviewDiffDecorationModel extends Disposable {
     }
 
     this.monacoEditor.focus();
+
+    this._onPartialEditEvent.fire(event);
     this._onPartialEditWidgetListChange.fire(this.partialEditWidgetList);
   }