@opensumi/ide-core-common 3.4.5-next-1730102270.0 → 3.4.5-next-1730119322.0

package/src/remote-service/data-store/store.ts

@@ -1,30 +1,37 @@
 import extend from 'lodash/extend';
 
 import { EventEmitter } from '@opensumi/events';
+import { isUndefined } from '@opensumi/ide-utils';
 
 import { select } from './select';
 
-export interface DataStore<Item> {
-  create(item: Item): Item;
-  find(query: Record<string, any>): Item[] | undefined;
-  size(query: Record<string, any>): number;
-  get(id: string, query?: Record<string, any>): Item | undefined;
-  update(id: string, item: Partial<Item>): void;
-  remove(id: string): void;
-}
+/**
+ * The query looks like:
+ *   { field: value }
+ *   { field: conditions }
+ */
+export type Query = Record<string, any>;
 
-export interface DataStoreEvent<Item> extends Record<string, any> {
+export interface DataStoreEvent<Item> {
   created: [item: Item];
   updated: [oldItem: Item, newItem: Item];
   removed: [item: Item];
+
+  [key: string]: any[];
 }
 
 export interface DataStoreOptions {
   id?: string;
 }
 
-export class InMemoryDataStore<Item> extends EventEmitter<DataStoreEvent<Item>> implements DataStore<Item> {
-  private store = new Map<string, Item>();
+type MarkOptional<T, K extends keyof T> = Omit<T, K> & Partial<T>;
+
+export class InMemoryDataStore<
+  Item extends Record<any, any>,
+  PrimaryKey extends keyof Item = 'id',
+  PrimaryKeyType = Required<Item>[PrimaryKey],
+> extends EventEmitter<DataStoreEvent<Item>> {
+  private store = new Map<PrimaryKeyType, Item>();
   private _uid = 0;
   /**
    * primary key
@@ -36,8 +43,8 @@ export class InMemoryDataStore<Item> extends EventEmitter<DataStoreEvent<Item>>
     this.id = options?.id || 'id';
   }
 
-  create(item: Item): Item {
-    const id = item[this.id] || String(this._uid++);
+  create(item: MarkOptional<Item, PrimaryKey>): Item {
+    const id = item[this.id] || (this._uid++ as PrimaryKeyType);
     const result = extend({}, item, { [this.id]: id }) as Item;
 
     this.store.set(id, result);
@@ -46,29 +53,32 @@ export class InMemoryDataStore<Item> extends EventEmitter<DataStoreEvent<Item>>
     return result;
   }
 
-  find(query: Record<string, any>): Item[] | undefined {
+  find(query?: Query): Item[] | undefined {
+    if (isUndefined(query)) {
+      return Array.from(this.store.values());
+    }
     return select(this.store, query);
   }
 
-  size(query?: Record<string, any>): number {
-    if (!query) {
+  count(query?: Query): number {
+    if (isUndefined(query)) {
       return this.store.size;
     }
 
     return this.find(query)?.length || 0;
   }
 
-  get(id: string): Item | undefined {
+  get(id: PrimaryKeyType): Item | undefined {
     return this.store.get(id);
   }
 
-  has(id: string): boolean {
+  has(id: PrimaryKeyType): boolean {
     return this.store.has(id);
   }
 
-  update(id: string, item: Partial<Item>): void {
+  update(id: PrimaryKeyType, item: Partial<Item>): void {
     const current = this.store.get(id);
-    if (!current) {
+    if (isUndefined(current)) {
       return;
     }
 
@@ -78,7 +88,7 @@ export class InMemoryDataStore<Item> extends EventEmitter<DataStoreEvent<Item>>
     this.store.set(id, result);
   }
 
-  remove(id: string): void {
+  remove(id: PrimaryKeyType): void {
     const item = this.store.get(id);
     if (item) {
       this.emit('removed', item);
@@ -86,4 +96,33 @@ export class InMemoryDataStore<Item> extends EventEmitter<DataStoreEvent<Item>>
 
     this.store.delete(id);
   }
+
+  removeItem(item: Item): void {
+    const id = item[this.id] as PrimaryKeyType;
+    if (isUndefined(id)) {
+      return;
+    }
+
+    this.remove(id);
+  }
+
+  removeAll(query?: Query): void {
+    if (isUndefined(query)) {
+      const items = Array.from(this.store.values());
+      this.store.clear();
+
+      items.forEach((item) => {
+        this.emit('removed', item);
+      });
+      return;
+    }
+
+    const items = this.find(query);
+    if (items) {
+      items.forEach((item) => {
+        this.store.delete(item[this.id]);
+        this.emit('removed', item);
+      });
+    }
+  }
 }

package/src/types/ai-native/index.ts

@@ -349,3 +349,10 @@ export interface IHistoryChatMessage extends IChatMessage {
   requestId?: string;
   replyStartTime?: number;
 }
+
+// ## Code Edits start ##
+export enum ECodeEditsSourceTyping {
+  LinterErrors = 'lint_errors',
+  LineChange = 'line_change',
+}
+// ## Code Edits ends ##

package/src/types/ai-native/reporter.ts

@@ -1,6 +1,8 @@
+import { ECodeEditsSourceTyping } from './index';
+
 export const AI_REPORTER_NAME = 'AI';
 
-export enum AISerivceType {
+export enum AIServiceType {
   Chat = 'chat',
   InlineChat = 'inlineChat',
   CodeAction = 'codeAction',
@@ -12,6 +14,7 @@ export enum AISerivceType {
   Rename = 'rename',
   TerminalAICommand = 'terminalAICommand',
   ProblemFix = 'problemFix',
+  CodeEdits = 'codeEdits',
 }
 
 export enum ActionSourceEnum {
@@ -64,7 +67,7 @@ export enum ActionTypeEnum {
 }
 
 export interface CommonLogInfo {
-  msgType: AISerivceType | string;
+  msgType: AIServiceType | string;
   relationId: string;
   replytime: number;
   success: boolean;
@@ -172,15 +175,23 @@ export interface InlineChatRT extends Partial<CommonLogInfo> {
   runByCodeAction?: boolean;
 }
 
+export interface CodeEditsRT extends Partial<CommonLogInfo> {
+  actionSource?: ECodeEditsSourceTyping;
+  isCancel?: boolean;
+  accept?: boolean;
+  discard?: boolean;
+}
+
 export type ReportInfo =
   | Partial<CommonLogInfo>
-  | ({ type: AISerivceType.Completion } & CompletionRT)
-  | ({ type: AISerivceType.MergeConflict } & MergeConflictRT)
-  | ({ type: AISerivceType.Rename } & RenameRT)
-  | ({ type: AISerivceType.InlineChat } & InlineChatRT)
-  | ({ type: AISerivceType.InlineChatInput } & InlineChatRT)
-  | ({ type: AISerivceType.Chat } & ChatRT)
-  | ({ type: AISerivceType.Agent } & ChatRT);
+  | ({ type: AIServiceType.Completion } & CompletionRT)
+  | ({ type: AIServiceType.MergeConflict } & MergeConflictRT)
+  | ({ type: AIServiceType.Rename } & RenameRT)
+  | ({ type: AIServiceType.InlineChat } & InlineChatRT)
+  | ({ type: AIServiceType.InlineChatInput } & InlineChatRT)
+  | ({ type: AIServiceType.Chat } & ChatRT)
+  | ({ type: AIServiceType.Agent } & ChatRT)
+  | ({ type: AIServiceType.CodeEdits } & CodeEditsRT);
 
 export const IAIReporter = Symbol('IAIReporter');
 

package/src/types/authentication.ts

@@ -106,10 +106,3 @@ export interface SessionRequest {
 export interface SessionRequestInfo {
   [scopes: string]: SessionRequest;
 }
-
-export interface AuthenticationGetSessionOptions {
-  createIfNone: boolean;
-  clearSessionPreference: boolean;
-  forceNewSession?: boolean | { detail: string };
-  silent?: boolean;
-}