@codingame/monaco-vscode-chat-service-override 28.0.1 → 28.1.2

package/index.d.ts

@@ -1,2 +1,15 @@
 import { type IEditorOverrideServices } from "@codingame/monaco-vscode-api/vscode/vs/editor/standalone/browser/standaloneServices";
-export default function getServiceOverride(): IEditorOverrideServices;
+import { ChatEntitlement, type IChatSentiment, type IQuotas } from "@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService";
+export interface CustomEntitlement {
+    entitlement: ChatEntitlement;
+    sentiment: IChatSentiment;
+    anonymous: boolean;
+    quotas: IQuotas;
+    previewFeaturesDisabled: boolean;
+}
+export interface ChatServiceOverrideOptions {
+    customEntitlement?: CustomEntitlement;
+}
+export default function getServiceOverride({ customEntitlement }?: ChatServiceOverrideOptions): IEditorOverrideServices;
+export { ChatEntitlement };
+export type { IChatSentiment, IQuotas };

package/index.js

@@ -32,6 +32,7 @@ import { IChatMarkdownAnchorService } from '@codingame/monaco-vscode-api/vscode/
 import { ChatMarkdownAnchorService } from './vscode/src/vs/workbench/contrib/chat/browser/widget/chatContentParts/chatMarkdownAnchorService.js';
 import { ChatEditingService } from './vscode/src/vs/workbench/contrib/chat/browser/chatEditing/chatEditingServiceImpl.js';
 import { ChatEntitlementService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService';
+export { ChatEntitlement } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService';
 import { PromptsService } from './vscode/src/vs/workbench/contrib/chat/common/promptSyntax/service/promptsServiceImpl.js';
 import { IChatEntitlementService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/services/chat/common/chatEntitlementService.service';
 import { IPromptsService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService.service';
@@ -104,13 +105,40 @@ import { IChatDebugService } from '@codingame/monaco-vscode-api/vscode/vs/workbe
 import { ChatDebugServiceImpl } from './vscode/src/vs/workbench/contrib/chat/common/chatDebugServiceImpl.js';
 import { IChatResponseResourceFileSystemProvider } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/widget/chatResponseResourceFileSystemProvider.service';
 import { ChatResponseResourceFileSystemProvider } from './vscode/src/vs/workbench/contrib/chat/common/widget/chatResponseResourceFileSystemProvider.js';
+import { Event } from '@codingame/monaco-vscode-api/vscode/vs/base/common/event';
+
 import './vscode/src/vs/workbench/contrib/chat/browser/chat.contribution.js';
 import './vscode/src/vs/workbench/contrib/terminalContrib/chatAgentTools/browser/terminal.chatAgentTools.contribution.js';
 import './vscode/src/vs/workbench/contrib/terminalContrib/chat/browser/terminal.chat.contribution.js';
 import './vscode/src/vs/workbench/contrib/inlineChat/browser/inlineChat.contribution.js';
 import './vscode/src/vs/workbench/contrib/remoteCodingAgents/browser/remoteCodingAgents.contribution.js';
+import { constObservable } from '@codingame/monaco-vscode-api/vscode/vs/base/common/observableInternal/observables/constObservable';
 
-function getServiceOverride() {
+class CustomEntitlementService {
+    constructor(customEntitlement) {
+        this.customEntitlement = customEntitlement;
+        this.onDidChangeEntitlement = Event.None;
+        this.entitlement = this.customEntitlement.entitlement;
+        this.entitlementObs = constObservable(this.entitlement);
+        this.previewFeaturesDisabled = this.customEntitlement.previewFeaturesDisabled;
+        this.organisations = undefined;
+        this.isInternal = true;
+        this.sku = undefined;
+        this.copilotTrackingId = undefined;
+        this.onDidChangeQuotaExceeded = Event.None;
+        this.onDidChangeQuotaRemaining = Event.None;
+        this.quotas = this.customEntitlement.quotas;
+        this.onDidChangeSentiment = Event.None;
+        this.sentiment = this.customEntitlement.sentiment;
+        this.sentimentObs = constObservable(this.sentiment);
+        this.onDidChangeAnonymous = Event.None;
+        this.anonymous = this.customEntitlement.anonymous;
+        this.anonymousObs = constObservable(this.anonymous);
+    }
+    markAnonymousRateLimited() { }
+    async update() { }
+}
+function getServiceOverride({ customEntitlement } = {}) {
     return {
         [IChatService.toString()]: new SyncDescriptor(ChatService, [], true),
         [IChatWidgetService.toString()]: new SyncDescriptor(ChatWidgetService, [], true),
@@ -131,7 +159,9 @@ function getServiceOverride() {
         [IChatEditingService.toString()]: new SyncDescriptor(ChatEditingService, [], true),
         [IChatTransferService.toString()]: new SyncDescriptor(ChatTransferService, [], true),
         [IChatMarkdownAnchorService.toString()]: new SyncDescriptor(ChatMarkdownAnchorService, [], true),
-        [IChatEntitlementService.toString()]: new SyncDescriptor(ChatEntitlementService, [], true),
+        [IChatEntitlementService.toString()]: customEntitlement != null
+            ? new SyncDescriptor(CustomEntitlementService, [customEntitlement], true)
+            : new SyncDescriptor(ChatEntitlementService, [], true),
         [IPromptsService.toString()]: new SyncDescriptor(PromptsService, [], true),
         [IChatStatusItemService.toString()]: new SyncDescriptor(ChatStatusItemService, [], true),
         [IChatContextPickService.toString()]: new SyncDescriptor(ChatContextPickService, [], true),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codingame/monaco-vscode-chat-service-override",
-  "version": "28.0.1",
+  "version": "28.1.2",
   "private": false,
   "description": "VSCode public API plugged on the monaco editor - chat service-override",
   "keywords": [],
@@ -15,10 +15,10 @@
   },
   "type": "module",
   "dependencies": {
-    "@codingame/monaco-vscode-api": "28.0.1",
-    "@codingame/monaco-vscode-katex-common": "28.0.1",
-    "@codingame/monaco-vscode-xterm-addons-common": "28.0.1",
-    "@codingame/monaco-vscode-xterm-common": "28.0.1"
+    "@codingame/monaco-vscode-api": "28.1.2",
+    "@codingame/monaco-vscode-katex-common": "28.1.2",
+    "@codingame/monaco-vscode-xterm-addons-common": "28.1.2",
+    "@codingame/monaco-vscode-xterm-common": "28.1.2"
   },
   "main": "index.js",
   "module": "index.js",

package/vscode/src/vs/workbench/contrib/chat/browser/actions/chatAgentRecommendationActions.js

@@ -126,7 +126,7 @@ function waitForCommandRegistration(commandId) {
         return Promise.resolve();
     }
     return ( new Promise(resolve => {
-        const timer = ( new TimeoutTimer());
+        const timer = new TimeoutTimer();
         const listener = CommandsRegistry.onDidRegisterCommand(id => {
             if (id === commandId) {
                 listener.dispose();

package/vscode/src/vs/workbench/contrib/chat/browser/actions/codeBlockOperations.js

@@ -384,7 +384,7 @@ let ApplyCodeBlockOperation = class ApplyCodeBlockOperation {
         };
     }
     getTextEdits(codeBlock, chatSessionResource, token) {
-        return ( new AsyncIterableProducer(async executor => {
+        return new AsyncIterableProducer(async executor => {
             const request = {
                 codeBlocks: [codeBlock],
                 chatSessionResource
@@ -399,10 +399,10 @@ let ApplyCodeBlockOperation = class ApplyCodeBlockOperation {
             if (result?.errorMessage) {
                 executor.reject(( new Error(result.errorMessage)));
             }
-        }));
+        });
     }
     getNotebookEdits(codeBlock, chatSessionResource, token) {
-        return ( new AsyncIterableProducer(async executor => {
+        return new AsyncIterableProducer(async executor => {
             const request = {
                 codeBlocks: [codeBlock],
                 chatSessionResource,
@@ -420,7 +420,7 @@ let ApplyCodeBlockOperation = class ApplyCodeBlockOperation {
             if (result?.errorMessage) {
                 executor.reject(( new Error(result.errorMessage)));
             }
-        }));
+        });
     }
     async waitForFirstElement(iterable) {
         const iterator = iterable[Symbol.asyncIterator]();