@codingame/monaco-vscode-katex-common 24.1.0

package/vscode/src/vs/workbench/contrib/chat/browser/contrib/chatInputRelatedFilesContrib.d.ts

@@ -0,0 +1,34 @@
+import { Event } from "@codingame/monaco-vscode-api/vscode/vs/base/common/event";
+import { Disposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
+import { ResourceSet } from "@codingame/monaco-vscode-api/vscode/vs/base/common/map";
+import { URI } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
+import { IWorkbenchContribution } from "@codingame/monaco-vscode-api/vscode/vs/workbench/common/contributions";
+import { IChatEditingService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatEditingService.service";
+import { IChatWidgetService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/chat.service";
+export declare class ChatRelatedFilesContribution extends Disposable implements IWorkbenchContribution {
+    private readonly chatEditingService;
+    private readonly chatWidgetService;
+    static readonly ID = "chat.relatedFilesWorkingSet";
+    private readonly chatEditingSessionDisposables;
+    private _currentRelatedFilesRetrievalOperation;
+    constructor(chatEditingService: IChatEditingService, chatWidgetService: IChatWidgetService);
+    private _updateRelatedFileSuggestions;
+    private _handleNewEditingSession;
+    dispose(): void;
+}
+export interface IChatRelatedFile {
+    uri: URI;
+    description: string;
+}
+export declare class ChatRelatedFiles extends Disposable {
+    private readonly _onDidChange;
+    readonly onDidChange: Event<void>;
+    private _removedFiles;
+    get removedFiles(): ResourceSet;
+    private _value;
+    get value(): IChatRelatedFile[];
+    set value(value: IChatRelatedFile[]);
+    remove(uri: URI): void;
+    clearRemovedFiles(): void;
+    clear(): void;
+}

package/vscode/src/vs/workbench/contrib/chat/browser/contrib/chatInputRelatedFilesContrib.js

@@ -0,0 +1,139 @@
+
+import { __decorate, __param } from '@codingame/monaco-vscode-api/external/tslib/tslib.es6';
+import { CancellationToken } from '@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation';
+import { Event, Emitter } from '@codingame/monaco-vscode-api/vscode/vs/base/common/event';
+import { Disposable, DisposableStore } from '@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle';
+import { ResourceMap, ResourceSet } from '@codingame/monaco-vscode-api/vscode/vs/base/common/map';
+import '@codingame/monaco-vscode-api/vscode/vs/base/common/observableInternal/index';
+import { isEqual } from '@codingame/monaco-vscode-api/vscode/vs/base/common/resources';
+import { localize } from '@codingame/monaco-vscode-api/vscode/vs/nls';
+import { IChatEditingService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/chatEditingService.service';
+import { IChatWidgetService } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/chat.service';
+import { autorun } from '@codingame/monaco-vscode-api/vscode/vs/base/common/observableInternal/reactions/autorun';
+
+let ChatRelatedFilesContribution = class ChatRelatedFilesContribution extends Disposable {
+    static { this.ID = 'chat.relatedFilesWorkingSet'; }
+    constructor(chatEditingService, chatWidgetService) {
+        super();
+        this.chatEditingService = chatEditingService;
+        this.chatWidgetService = chatWidgetService;
+        this.chatEditingSessionDisposables = ( new ResourceMap());
+        this._register(autorun((reader) => {
+            const sessions = this.chatEditingService.editingSessionsObs.read(reader);
+            sessions.forEach(session => {
+                const widget = this.chatWidgetService.getWidgetBySessionResource(session.chatSessionResource);
+                if (widget && !( this.chatEditingSessionDisposables.has(session.chatSessionResource))) {
+                    this._handleNewEditingSession(session, widget);
+                }
+            });
+        }));
+    }
+    _updateRelatedFileSuggestions(currentEditingSession, widget) {
+        if (this._currentRelatedFilesRetrievalOperation) {
+            return;
+        }
+        const workingSetEntries = currentEditingSession.entries.get();
+        if (workingSetEntries.length > 0 || widget.attachmentModel.fileAttachments.length === 0) {
+            return;
+        }
+        this._currentRelatedFilesRetrievalOperation = this.chatEditingService.getRelatedFiles(currentEditingSession.chatSessionResource, widget.getInput(), widget.attachmentModel.fileAttachments, CancellationToken.None)
+            .then((files) => {
+            if (!files?.length || !widget.viewModel || !widget.input.relatedFiles) {
+                return;
+            }
+            const currentEditingSession = this.chatEditingService.getEditingSession(widget.viewModel.sessionResource);
+            if (!currentEditingSession || currentEditingSession.entries.get().length) {
+                return;
+            }
+            const existingFiles = ( new ResourceSet(
+                [...widget.attachmentModel.fileAttachments, ...widget.input.relatedFiles.removedFiles]
+            ));
+            if (!existingFiles.size) {
+                return;
+            }
+            const newSuggestions = ( new ResourceMap());
+            for (const group of files) {
+                for (const file of group.files) {
+                    if (newSuggestions.size >= 2) {
+                        break;
+                    }
+                    if (( existingFiles.has(file.uri))) {
+                        continue;
+                    }
+                    newSuggestions.set(file.uri, ( localize(5695, "{0} (Suggested)", file.description)));
+                    existingFiles.add(file.uri);
+                }
+            }
+            widget.input.relatedFiles.value = ( [...newSuggestions.entries()].map(([uri, description]) => ({ uri, description })));
+        })
+            .finally(() => {
+            this._currentRelatedFilesRetrievalOperation = undefined;
+        });
+    }
+    _handleNewEditingSession(currentEditingSession, widget) {
+        const disposableStore = ( new DisposableStore());
+        disposableStore.add(currentEditingSession.onDidDispose(() => {
+            disposableStore.clear();
+        }));
+        this._updateRelatedFileSuggestions(currentEditingSession, widget);
+        const onDebouncedType = Event.debounce(widget.inputEditor.onDidChangeModelContent, () => null, 3000);
+        disposableStore.add(onDebouncedType(() => {
+            this._updateRelatedFileSuggestions(currentEditingSession, widget);
+        }));
+        disposableStore.add(widget.attachmentModel.onDidChange(() => {
+            this._updateRelatedFileSuggestions(currentEditingSession, widget);
+        }));
+        disposableStore.add(currentEditingSession.onDidDispose(() => {
+            disposableStore.dispose();
+        }));
+        disposableStore.add(widget.onDidAcceptInput(() => {
+            widget.input.relatedFiles?.clear();
+            this._updateRelatedFileSuggestions(currentEditingSession, widget);
+        }));
+        this.chatEditingSessionDisposables.set(currentEditingSession.chatSessionResource, disposableStore);
+    }
+    dispose() {
+        for (const store of ( this.chatEditingSessionDisposables.values())) {
+            store.dispose();
+        }
+        super.dispose();
+    }
+};
+ChatRelatedFilesContribution = ( __decorate([
+    ( __param(0, IChatEditingService)),
+    ( __param(1, IChatWidgetService))
+], ChatRelatedFilesContribution));
+class ChatRelatedFiles extends Disposable {
+    constructor() {
+        super(...arguments);
+        this._onDidChange = this._register(( new Emitter()));
+        this.onDidChange = this._onDidChange.event;
+        this._removedFiles = ( new ResourceSet());
+        this._value = [];
+    }
+    get removedFiles() {
+        return this._removedFiles;
+    }
+    get value() {
+        return this._value;
+    }
+    set value(value) {
+        this._value = value;
+        this._onDidChange.fire();
+    }
+    remove(uri) {
+        this._value = this._value.filter(file => !isEqual(file.uri, uri));
+        this._removedFiles.add(uri);
+        this._onDidChange.fire();
+    }
+    clearRemovedFiles() {
+        this._removedFiles.clear();
+    }
+    clear() {
+        this._value = [];
+        this._removedFiles.clear();
+        this._onDidChange.fire();
+    }
+}
+
+export { ChatRelatedFiles, ChatRelatedFilesContribution };