@codingame/monaco-vscode-chat-service-override 37.0.0 → 37.1.0

package/vscode/src/vs/workbench/contrib/chat/browser/agentSessions/agentHost/agentCustomizationContentExpander.js

@@ -0,0 +1,229 @@
+
+import { isCancellationError } from '@codingame/monaco-vscode-api/vscode/vs/base/common/errors';
+import { extname } from '@codingame/monaco-vscode-api/vscode/vs/base/common/path';
+import { joinPath } from '@codingame/monaco-vscode-api/vscode/vs/base/common/resources';
+import { URI } from '@codingame/monaco-vscode-api/vscode/vs/base/common/uri';
+import { parseFrontMatter } from '@codingame/monaco-vscode-api/vscode/vs/base/common/yaml';
+import { toFileOperationResult, FileOperationResult } from '@codingame/monaco-vscode-api/vscode/vs/platform/files/common/files';
+import { SKILL_FILENAME } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/config/promptFileLocations';
+import { PromptsType } from '@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes';
+
+class AgentCustomizationContentExpander {
+    constructor(fileService, logService) {
+        this.fileService = fileService;
+        this.logService = logService;
+    }
+    async expandPluginContents(pluginUri, groupKey, isBundleItem, source, pluginLabel, token) {
+        const fsRoot = pluginUri;
+        const children = [];
+        try {
+            if (!(await this.fileService.canHandleResource(fsRoot))) {
+                return [];
+            }
+            if (token.isCancellationRequested) {
+                return [];
+            }
+            const dirNames = ["agents", "skills", "commands", "rules"];
+            const promptTypes = [
+                PromptsType.agent,
+                PromptsType.skill,
+                PromptsType.prompt,
+                PromptsType.instructions
+            ];
+            const stats = await Promise.all(( dirNames.map(async name => {
+                const resource = URI.joinPath(fsRoot, name);
+                try {
+                    return await this.fileService.resolve(resource);
+                } catch (err) {
+                    if (!isExpectedFileAccessError(err, token)) {
+                        this.logService.trace(
+                            `[AgentCustomizationContentExpander] Failed to resolve customization directory ${( resource.toString())}: ${err}`
+                        );
+                    }
+                    return undefined;
+                }
+            })));
+            if (token.isCancellationRequested) {
+                return [];
+            }
+            for (let i = 0; i < dirNames.length; i++) {
+                const stat = stats[i];
+                const promptType = promptTypes[i];
+                if (!stat?.isDirectory || !stat.children) {
+                    continue;
+                }
+                if (promptType === PromptsType.skill) {
+                    children.push(...(await this.collectFromSkillDir(
+                        stat.children,
+                        pluginUri,
+                        source,
+                        groupKey,
+                        isBundleItem,
+                        pluginLabel,
+                        token
+                    )));
+                } else {
+                    children.push(...(await this.collectFromRegularDir(
+                        stat.children,
+                        pluginUri,
+                        source,
+                        promptType,
+                        groupKey,
+                        isBundleItem,
+                        pluginLabel,
+                        token
+                    )));
+                }
+            }
+            children.sort((a, b) => `${a.type}:${a.name}`.localeCompare(`${b.type}:${b.name}`));
+        } catch (err) {
+            if (!isExpectedFileAccessError(err, token)) {
+                this.logService.trace(
+                    `[AgentCustomizationContentExpander] Failed to expand plugin ${( pluginUri.toString())}: ${err}`
+                );
+            }
+            return [];
+        }
+        return children;
+    }
+    async collectFromSkillDir(entries, pluginUri, source, groupKey, isBundleItem, pluginLabel, token) {
+        const eligible = [];
+        const readMetaDataPromises = [];
+        for (const child of entries) {
+            if (child.name.startsWith(".")) {
+                continue;
+            }
+            if (!child.isDirectory) {
+                continue;
+            }
+            eligible.push(child);
+            readMetaDataPromises.push(this.readPromptMetadata(joinPath(child.resource, SKILL_FILENAME), token));
+        }
+        const promptMetadata = await Promise.all(readMetaDataPromises);
+        if (token.isCancellationRequested) {
+            return [];
+        }
+        const items = [];
+        for (let i = 0; i < eligible.length; i++) {
+            const child = eligible[i];
+            const meta = promptMetadata[i];
+            if (!meta) {
+                continue;
+            }
+            const uri = joinPath(child.resource, SKILL_FILENAME);
+            const name = meta.name ?? child.name;
+            const description = meta.description;
+            const userInvocable = meta.userInvocable;
+            items.push({
+                uri,
+                type: PromptsType.skill,
+                name: name,
+                description,
+                source,
+                groupKey,
+                extensionId: undefined,
+                pluginUri: isBundleItem ? undefined : pluginUri,
+                pluginLabel: isBundleItem ? undefined : pluginLabel,
+                userInvocable
+            });
+        }
+        return items;
+    }
+    async collectFromRegularDir(
+        entries,
+        pluginUri,
+        source,
+        promptType,
+        groupKey,
+        isBundleItem,
+        pluginLabel,
+        token
+    ) {
+        const eligible = [];
+        for (const child of entries) {
+            if (child.name.startsWith(".")) {
+                continue;
+            }
+            if (child.isDirectory) {
+                continue;
+            }
+            const ext = extname(child.name);
+            if (ext !== ".md" && !(promptType === PromptsType.instructions && ext === ".mdc")) {
+                continue;
+            }
+            eligible.push(child);
+        }
+        const parseMetadata = promptType === PromptsType.agent || promptType === PromptsType.instructions;
+        const promptMetadata = parseMetadata ? await Promise.all(( eligible.map(child => this.readPromptMetadata(child.resource, token)))) : undefined;
+        if (token.isCancellationRequested) {
+            return [];
+        }
+        const items = [];
+        for (let i = 0; i < eligible.length; i++) {
+            const child = eligible[i];
+            const meta = promptMetadata?.[i];
+            items.push({
+                uri: child.resource,
+                type: promptType,
+                name: meta?.name ?? stripPromptFileExtensions(child.name),
+                description: meta?.description,
+                source,
+                groupKey,
+                extensionId: undefined,
+                pluginUri: isBundleItem ? undefined : pluginUri,
+                pluginLabel: isBundleItem ? undefined : pluginLabel,
+                userInvocable: promptType === PromptsType.agent ? meta?.userInvocable : undefined
+            });
+        }
+        return items;
+    }
+    async readPromptMetadata(promptFileUri, token) {
+        if (extname(promptFileUri.path) !== ".md") {
+            return undefined;
+        }
+        try {
+            const content = await this.fileService.readFile(promptFileUri);
+            if (token.isCancellationRequested) {
+                return undefined;
+            }
+            const frontmatter = parseFrontMatter(( content.value.toString()));
+            if (frontmatter) {
+                const name = frontmatter.getStringValue("name");
+                const description = frontmatter.getStringValue("description");
+                const userInvocableStr = frontmatter.getStringValue("user-invocable");
+                const userInvocable = userInvocableStr === "true" ? true : userInvocableStr === "false" ? false : undefined;
+                return {
+                    name,
+                    description,
+                    userInvocable
+                };
+            }
+            return {
+                name: undefined,
+                description: undefined,
+                userInvocable: undefined
+            };
+        } catch (err) {
+            if (!isExpectedFileAccessError(err, token)) {
+                this.logService.trace(
+                    `[AgentCustomizationContentExpander] Failed to read prompt metadata ${( promptFileUri.toString())}: ${err}`
+                );
+            }
+            return undefined;
+        }
+    }
+}
+function isExpectedFileAccessError(error, token) {
+    return token.isCancellationRequested || isCancellationError(error) || toFileOperationResult(error) === FileOperationResult.FILE_NOT_FOUND;
+}
+function stripPromptFileExtensions(filename) {
+    const ext = extname(filename);
+    if (!ext) {
+        return filename;
+    }
+    const stem = filename.slice(0, -ext.length);
+    const dotInStem = stem.lastIndexOf(".");
+    return dotInStem > 0 ? stem.slice(0, dotInStem) : stem;
+}
+
+export { AgentCustomizationContentExpander };

package/vscode/src/vs/workbench/contrib/chat/browser/agentSessions/agentHost/agentCustomizationItemProvider.d.ts

@@ -0,0 +1,57 @@
+import { CancellationToken } from "@codingame/monaco-vscode-api/vscode/vs/base/common/cancellation";
+import { Disposable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/lifecycle";
+import { Event } from "@codingame/monaco-vscode-api/vscode/vs/base/common/event";
+import { type IObservable } from "@codingame/monaco-vscode-api/vscode/vs/base/common/observable";
+import { URI } from "@codingame/monaco-vscode-api/vscode/vs/base/common/uri";
+import { type AgentCustomization, type ClientPluginCustomization, PluginCustomization } from "@codingame/monaco-vscode-api/vscode/vs/platform/agentHost/common/state/sessionState";
+import { ILogService } from "@codingame/monaco-vscode-api/vscode/vs/platform/log/common/log.service";
+import { ICustomizationItem, ICustomizationItemAction, ICustomizationItemProvider, ICustomizationSourceFolder } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/customizationHarnessService";
+import { IFileService } from "@codingame/monaco-vscode-api/vscode/vs/platform/files/common/files.service";
+import { PromptsType } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/promptTypes";
+import { IAgentHostCustomizationService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/browser/agentSessions/agentHost/agentHostCustomizationService.service";
+import { type ISyncedCustomizationOrigin } from "./syncedCustomizationBundler.js";
+import { ICustomAgent } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService";
+import { IPromptsService } from "@codingame/monaco-vscode-api/vscode/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService.service";
+export declare class AgentCustomizationItemProvider extends Disposable implements ICustomizationItemProvider {
+    private readonly _connectionAuthority;
+    private readonly _getItemActions;
+    private readonly _resolveSyncedOrigin;
+    private readonly _fileService;
+    private readonly _logService;
+    private readonly _customAgentsService;
+    private readonly _promptsService;
+    private readonly _onDidChange;
+    readonly onDidChange: Event<void>;
+    /** Cache: pluginUri → last expansion (keyed by nonce and label so we re-fetch on content or display-name changes). */
+    private readonly _expansionCache;
+    private readonly _contentExpander;
+    private _draftCustomAgents;
+    private _draftCustomizations;
+    constructor(_connectionAuthority: string, _getItemActions: ((customization: PluginCustomization, clientId: string | undefined) => ICustomizationItemAction[] | undefined) | undefined, _resolveSyncedOrigin: ((syncedUri: URI) => ISyncedCustomizationOrigin | undefined) | undefined, _fileService: IFileService, _logService: ILogService, _customAgentsService: IAgentHostCustomizationService, _promptsService: IPromptsService);
+    setDraftCustomAgents(customAgents: IObservable<readonly AgentCustomization[]>): void;
+    setDraftCustomizations(customizations: IObservable<readonly ClientPluginCustomization[]>): void;
+    private toRemoteUri;
+    private toBadge;
+    private toItem;
+    private toDirectoryItems;
+    private toDirectoryChildItem;
+    provideSourceFolders(sessionResource: URI, type: PromptsType, token: CancellationToken): Promise<readonly ICustomizationSourceFolder[]>;
+    provideCustomAgents(sessionResource: URI): Promise<readonly ICustomAgent[]>;
+    provideChatSessionCustomizations(sessionResource: URI, token: CancellationToken): Promise<ICustomizationItem[]>;
+    private getCustomAgents;
+    private getCustomizations;
+    /**
+     * Rewrites a bundle child item to reflect the original source location of
+     * the flattened file, when it can be recovered from the synthetic bundle's
+     * reverse map. The synced (in-memory) URI is replaced with the real local
+     * URI so the item points at its true origin, and the source/extension/plugin
+     * metadata is restored. Returns the item unchanged when no origin is known.
+     */
+    private _applySyncedOrigin;
+    /**
+     * Reads a plugin's directory contents through the agent-host
+     * filesystem provider and returns one {@link ICustomizationItem} per
+     * supported file (agents/skills/instructions/prompts).
+     */
+    private _expandPluginContents;
+}