Atlas › Test

chatService.test|title=ChatService loadRemoteSession progress streaming contributed session read-only state is preserved on the chat model|occurrence=1

Exact test identity: mocha:v1|namespace=vscode@05c208e9e28d8c1c723fa08f85e2b7a96092e8e5|file=vs/workbench/contrib/chat/test/common/chatService/chatService.test|title=ChatService loadRemoteSession progress streaming contributed session read-only state is preserved on the chat model|occurrence=1

Package
mocha:v1|namespace=vscode@05c208e9e28d8c1c723fa08f85e2b7a96092e8e5|file=vs/workbench/contrib/chat/test/common/chatService
Suite / test hierarchy
chatService.test|title=ChatService loadRemoteSession progress streaming contributed session read-only state is preserved on the chat model|occurrence=1
Test
chatService.test|title=ChatService loadRemoteSession progress streaming contributed session read-only state is preserved on the chat model|occurrence=1
Introduced at
chatService.test|title=ChatService loadRemoteSession progress streaming contributed session read-only state is preserved on the chat model|occurrence=1 Frontier kind: Test frontier
Covered ranges
6106
Covered lines
71779
Covered files
309

Covered source

Expand a file to inspect source; the > gutter marks covered lines.

src/vs/platform/mcp/common/modelContextProtocol.ts 3277 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- modelContextProtocol.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /* eslint-disable local/code-no-unexternalized-strings */
7 >
8 > //#region proposals
9 > /**
10 > * MCP protocol proposals.
11 > * - Proposals here MUST have an MCP PR linked to them
12 > * - Proposals here are subject to change and SHALL be removed when
13 > * the upstream MCP PR is merged or closed.
14 > */
15 > export namespace MCP {
16 >
17 > // Nothing, yet
18 >
19 > }
20 >
21 > //#endregion
22 >
23 > /**
24 > * Schema updated from the Model Context Protocol repository at
25 > * https://github.com/modelcontextprotocol/specification/tree/main/schema
26 > *
27 > * ⚠️ Do not edit within `namespace` manually except to update schema versions ⚠️
28 > */
29 > export namespace MCP {
30 > /* JSON-RPC types */
31 >
32 > /**
33 > * Refers to any valid JSON-RPC object that can be decoded off the wire, or encoded to be sent.
34 > *
35 > * @category JSON-RPC
36 > */
37 > export type JSONRPCMessage =
38 > | JSONRPCRequest
39 > | JSONRPCNotification
40 > | JSONRPCResponse;
41 >
42 > /** @internal */
43 > export const LATEST_PROTOCOL_VERSION = "2025-11-25";
44 > /** @internal */
45 > export const JSONRPC_VERSION = "2.0";
46 >
47 > /**
48 > * Represents the contents of a `_meta` field, which clients and servers use to attach additional metadata to their interactions.
49 > *
50 > * Certain key names are reserved by MCP for protocol-level metadata; implementations MUST NOT make assumptions about values at these keys. Additionally, specific schema definitions may reserve particular names for purpose-specific metadata, as declared in those definitions.
51 > *
52 > * Valid keys have two segments:
53 > *
54 > * **Prefix:**
55 > * - Optional - if specified, MUST be a series of _labels_ separated by dots (`.`), followed by a slash (`/`).
56 > * - Labels MUST start with a letter and end with a letter or digit. Interior characters may be letters, digits, or hyphens (`-`).
57 > * - Any prefix consisting of zero or more labels, followed by `modelcontextprotocol` or `mcp`, followed by any label, is **reserved** for MCP use. For example: `modelcontextprotocol.io/`, `mcp.dev/`, `api.modelcontextprotocol.org/`, and `tools.mcp.com/` are all reserved.
58 > *
59 > * **Name:**
60 > * - Unless empty, MUST start and end with an alphanumeric character (`[a-z0-9A-Z]`).
61 > * - Interior characters may be alphanumeric, hyphens (`-`), underscores (`_`), or dots (`.`).
62 > *
63 > * @see [General fields: `_meta`](/specification/draft/basic/index#meta) for more details.
64 > * @category Common Types
65 > */
66 > export type MetaObject = Record<string, unknown>;
67 >
68 > /**
69 > * Extends {@link MetaObject} with additional request-specific fields. All key naming rules from `MetaObject` apply.
70 > *
71 > * @see {@link MetaObject} for key naming rules and reserved prefixes.
72 > * @see [General fields: `_meta`](/specification/draft/basic/index#meta) for more details.
73 > * @category Common Types
74 > */
75 > export interface RequestMetaObject extends MetaObject {
76 > /**
77 > * If specified, the caller is requesting out-of-band progress notifications for this request (as represented by {@link ProgressNotification | notifications/progress}). The value of this parameter is an opaque token that will be attached to any subsequent notifications. The receiver is not obligated to provide these notifications.
78 > */
79 > progressToken?: ProgressToken;
80 > }
81 >
82 > /**
83 > * A progress token, used to associate progress notifications with the original request.
84 > *
85 > * @category Common Types
86 > */
87 > export type ProgressToken = string | number;
88 >
89 > /**
90 > * An opaque token used to represent a cursor for pagination.
91 > *
92 > * @category Common Types
93 > */
94 > export type Cursor = string;
95 >
96 > /**
97 > * Common params for any task-augmented request.
98 > *
99 > * @internal
100 > */
101 > export interface TaskAugmentedRequestParams extends RequestParams {
102 > /**
103 > * If specified, the caller is requesting task-augmented execution for this request.
104 > * The request will return a {@link CreateTaskResult} immediately, and the actual result can be
105 > * retrieved later via {@link GetTaskPayloadRequest | tasks/result}.
106 > *
107 > * Task augmentation is subject to capability negotiation - receivers MUST declare support
108 > * for task augmentation of specific request types in their capabilities.
109 > */
110 > task?: TaskMetadata;
111 > }
112 >
113 > /**
114 > * Common params for any request.
115 > *
116 > * @category Common Types
117 > */
118 > export interface RequestParams {
119 > _meta?: RequestMetaObject;
120 > }
121 >
122 > /** @internal */
123 > export interface Request {
124 > method: string;
125 > // Allow unofficial extensions of `Request.params` without impacting `RequestParams`.
126 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
127 > params?: { [key: string]: any };
128 > }
129 >
130 > /**
131 > * Common params for any notification.
132 > *
133 > * @category Common Types
134 > */
135 > export interface NotificationParams {
136 > _meta?: MetaObject;
137 > }
138 >
139 > /** @internal */
140 > export interface Notification {
141 > method: string;
142 > // Allow unofficial extensions of `Notification.params` without impacting `NotificationParams`.
143 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
144 > params?: { [key: string]: any };
145 > }
146 >
147 > /**
148 > * Common result fields.
149 > *
150 > * @category Common Types
151 > */
152 > export interface Result {
153 > _meta?: MetaObject;
154 > [key: string]: unknown;
155 > }
156 >
157 > /**
158 > * @category Errors
159 > */
160 > export interface Error {
161 > /**
162 > * The error type that occurred.
163 > */
164 > code: number;
165 > /**
166 > * A short description of the error. The message SHOULD be limited to a concise single sentence.
167 > */
168 > message: string;
169 > /**
170 > * Additional information about the error. The value of this member is defined by the sender (e.g. detailed error information, nested errors etc.).
171 > */
172 > data?: unknown;
173 > }
174 >
175 > /**
176 > * A uniquely identifying ID for a request in JSON-RPC.
177 > *
178 > * @category Common Types
179 > */
180 > export type RequestId = string | number;
181 >
182 > /**
183 > * A request that expects a response.
184 > *
185 > * @category JSON-RPC
186 > */
187 > export interface JSONRPCRequest extends Request {
188 > jsonrpc: typeof JSONRPC_VERSION;
189 > id: RequestId;
190 > }
191 >
192 > /**
193 > * A notification which does not expect a response.
194 > *
195 > * @category JSON-RPC
196 > */
197 > export interface JSONRPCNotification extends Notification {
198 > jsonrpc: typeof JSONRPC_VERSION;
199 > }
200 >
201 > /**
202 > * A successful (non-error) response to a request.
203 > *
204 > * @category JSON-RPC
205 > */
206 > export interface JSONRPCResultResponse {
207 > jsonrpc: typeof JSONRPC_VERSION;
208 > id: RequestId;
209 > result: Result;
210 > }
211 >
212 > /**
213 > * A response to a request that indicates an error occurred.
214 > *
215 > * @category JSON-RPC
216 > */
217 > export interface JSONRPCErrorResponse {
218 > jsonrpc: typeof JSONRPC_VERSION;
219 > id?: RequestId;
220 > error: Error;
221 > }
222 >
223 > /**
224 > * A response to a request, containing either the result or error.
225 > *
226 > * @category JSON-RPC
227 > */
228 > export type JSONRPCResponse = JSONRPCResultResponse | JSONRPCErrorResponse;
229 >
230 > // Standard JSON-RPC error codes
231 > export const PARSE_ERROR = -32700;
232 > export const INVALID_REQUEST = -32600;
233 > export const METHOD_NOT_FOUND = -32601;
234 > export const INVALID_PARAMS = -32602;
235 > export const INTERNAL_ERROR = -32603;
236 >
237 > /**
238 > * A JSON-RPC error indicating that invalid JSON was received by the server. This error is returned when the server cannot parse the JSON text of a message.
239 > *
240 > * @see {@link https://www.jsonrpc.org/specification#error_object | JSON-RPC 2.0 Error Object}
241 > *
242 > * @example Invalid JSON
243 > * {@includeCode ./examples/ParseError/invalid-json.json}
244 > *
245 > * @category Errors
246 > */
247 > export interface ParseError extends Error {
248 > code: typeof PARSE_ERROR;
249 > }
250 >
251 > /**
252 > * A JSON-RPC error indicating that the request is not a valid request object. This error is returned when the message structure does not conform to the JSON-RPC 2.0 specification requirements for a request (e.g., missing required fields like `jsonrpc` or `method`, or using invalid types for these fields).
253 > *
254 > * @see {@link https://www.jsonrpc.org/specification#error_object | JSON-RPC 2.0 Error Object}
255 > *
256 > * @category Errors
257 > */
258 > export interface InvalidRequestError extends Error {
259 > code: typeof INVALID_REQUEST;
260 > }
261 >
262 > /**
263 > * A JSON-RPC error indicating that the requested method does not exist or is not available.
264 > *
265 > * In MCP, this error is returned when a request is made for a method that requires a capability that has not been declared. This can occur in either direction:
266 > *
267 > * - A server returning this error when the client requests a capability it doesn't support (e.g., requesting completions when the `completions` capability was not advertised)
268 > * - A client returning this error when the server requests a capability it doesn't support (e.g., requesting roots when the client did not declare the `roots` capability)
269 > *
270 > * @see {@link https://www.jsonrpc.org/specification#error_object | JSON-RPC 2.0 Error Object}
271 > *
272 > * @example Roots not supported
273 > * {@includeCode ./examples/MethodNotFoundError/roots-not-supported.json}
274 > *
275 > * @category Errors
276 > */
277 > export interface MethodNotFoundError extends Error {
278 > code: typeof METHOD_NOT_FOUND;
279 > }
280 >
281 > /**
282 > * A JSON-RPC error indicating that the method parameters are invalid or malformed.
283 > *
284 > * In MCP, this error is returned in various contexts when request parameters fail validation:
285 > *
286 > * - **Tools**: Unknown tool name or invalid tool arguments
287 > * - **Prompts**: Unknown prompt name or missing required arguments
288 > * - **Pagination**: Invalid or expired cursor values
289 > * - **Logging**: Invalid log level
290 > * - **Tasks**: Invalid or nonexistent task ID, invalid cursor, or attempting to cancel a task already in a terminal status
291 > * - **Elicitation**: Server requests an elicitation mode not declared in client capabilities
292 > * - **Sampling**: Missing tool result or tool results mixed with other content
293 > *
294 > * @see {@link https://www.jsonrpc.org/specification#error_object | JSON-RPC 2.0 Error Object}
295 > *
296 > * @example Unknown tool
297 > * {@includeCode ./examples/InvalidParamsError/unknown-tool.json}
298 > *
299 > * @example Invalid tool arguments
300 > * {@includeCode ./examples/InvalidParamsError/invalid-tool-arguments.json}
301 > *
302 > * @example Unknown prompt
303 > * {@includeCode ./examples/InvalidParamsError/unknown-prompt.json}
304 > *
305 > * @example Invalid cursor
306 > * {@includeCode ./examples/InvalidParamsError/invalid-cursor.json}
307 > *
308 > * @category Errors
309 > */
310 > export interface InvalidParamsError extends Error {
311 > code: typeof INVALID_PARAMS;
312 > }
313 >
314 > /**
315 > * A JSON-RPC error indicating that an internal error occurred on the receiver. This error is returned when the receiver encounters an unexpected condition that prevents it from fulfilling the request.
316 > *
317 > * @see {@link https://www.jsonrpc.org/specification#error_object | JSON-RPC 2.0 Error Object}
318 > *
319 > * @example Unexpected error
320 > * {@includeCode ./examples/InternalError/unexpected-error.json}
321 > *
322 > * @category Errors
323 > */
324 > export interface InternalError extends Error {
325 > code: typeof INTERNAL_ERROR;
326 > }
327 >
328 > // Implementation-specific JSON-RPC error codes [-32000, -32099]
329 > /** @internal */
330 > export const URL_ELICITATION_REQUIRED = -32042;
331 >
332 > /**
333 > * An error response that indicates that the server requires the client to provide additional information via an elicitation request.
334 > *
335 > * @example Authorization required
336 > * {@includeCode ./examples/URLElicitationRequiredError/authorization-required.json}
337 > *
338 > * @internal
339 > */
340 > export interface URLElicitationRequiredError extends Omit<
341 > JSONRPCErrorResponse,
342 > "error"
343 > > {
344 > error: Error & {
345 > code: typeof URL_ELICITATION_REQUIRED;
346 > data: {
347 > elicitations: ElicitRequestURLParams[];
348 > [key: string]: unknown;
349 > };
350 > };
351 > }
352 >
353 > /* Empty result */
354 > /**
355 > * A result that indicates success but carries no data.
356 > *
357 > * @category Common Types
358 > */
359 > export type EmptyResult = Result;
360 >
361 > /* Cancellation */
362 > /**
363 > * Parameters for a `notifications/cancelled` notification.
364 > *
365 > * @example User-requested cancellation
366 > * {@includeCode ./examples/CancelledNotificationParams/user-requested-cancellation.json}
367 > *
368 > * @category `notifications/cancelled`
369 > */
370 > export interface CancelledNotificationParams extends NotificationParams {
371 > /**
372 > * The ID of the request to cancel.
373 > *
374 > * This MUST correspond to the ID of a request previously issued in the same direction.
375 > * This MUST be provided for cancelling non-task requests.
376 > * This MUST NOT be used for cancelling tasks (use the {@link CancelTaskRequest | tasks/cancel} request instead).
377 > */
378 > requestId?: RequestId;
379 >
380 > /**
381 > * An optional string describing the reason for the cancellation. This MAY be logged or presented to the user.
382 > */
383 > reason?: string;
384 > }
385 >
386 > /**
387 > * This notification can be sent by either side to indicate that it is cancelling a previously-issued request.
388 > *
389 > * The request SHOULD still be in-flight, but due to communication latency, it is always possible that this notification MAY arrive after the request has already finished.
390 > *
391 > * This notification indicates that the result will be unused, so any associated processing SHOULD cease.
392 > *
393 > * A client MUST NOT attempt to cancel its `initialize` request.
394 > *
395 > * For task cancellation, use the {@link CancelTaskRequest | tasks/cancel} request instead of this notification.
396 > *
397 > * @example User-requested cancellation
398 > * {@includeCode ./examples/CancelledNotification/user-requested-cancellation.json}
399 > *
400 > * @category `notifications/cancelled`
401 > */
402 > export interface CancelledNotification extends JSONRPCNotification {
403 > method: "notifications/cancelled";
404 > params: CancelledNotificationParams;
405 > }
406 >
407 > /* Initialization */
408 > /**
409 > * Parameters for an `initialize` request.
410 > *
411 > * @example Full client capabilities
412 > * {@includeCode ./examples/InitializeRequestParams/full-client-capabilities.json}
413 > *
414 > * @category `initialize`
415 > */
416 > export interface InitializeRequestParams extends RequestParams {
417 > /**
418 > * The latest version of the Model Context Protocol that the client supports. The client MAY decide to support older versions as well.
419 > */
420 > protocolVersion: string;
421 > capabilities: ClientCapabilities;
422 > clientInfo: Implementation;
423 > }
424 >
425 > /**
426 > * This request is sent from the client to the server when it first connects, asking it to begin initialization.
427 > *
428 > * @example Initialize request
429 > * {@includeCode ./examples/InitializeRequest/initialize-request.json}
430 > *
431 > * @category `initialize`
432 > */
433 > export interface InitializeRequest extends JSONRPCRequest {
434 > method: "initialize";
435 > params: InitializeRequestParams;
436 > }
437 >
438 > /**
439 > * The result returned by the server for an {@link InitializeRequest | initialize} request.
440 > *
441 > * @example Full server capabilities
442 > * {@includeCode ./examples/InitializeResult/full-server-capabilities.json}
443 > *
444 > * @category `initialize`
445 > */
446 > export interface InitializeResult extends Result {
447 > /**
448 > * The version of the Model Context Protocol that the server wants to use. This may not match the version that the client requested. If the client cannot support this version, it MUST disconnect.
449 > */
450 > protocolVersion: string;
451 > capabilities: ServerCapabilities;
452 > serverInfo: Implementation;
453 >
454 > /**
455 > * Instructions describing how to use the server and its features.
456 > *
457 > * This can be used by clients to improve the LLM's understanding of available tools, resources, etc. It can be thought of like a "hint" to the model. For example, this information MAY be added to the system prompt.
458 > */
459 > instructions?: string;
460 > }
461 >
462 > /**
463 > * A successful response from the server for a {@link InitializeRequest | initialize} request.
464 > *
465 > * @example Initialize result response
466 > * {@includeCode ./examples/InitializeResultResponse/initialize-result-response.json}
467 > *
468 > * @category `initialize`
469 > */
470 > export interface InitializeResultResponse extends JSONRPCResultResponse {
471 > result: InitializeResult;
472 > }
473 >
474 > /**
475 > * This notification is sent from the client to the server after initialization has finished.
476 > *
477 > * @example Initialized notification
478 > * {@includeCode ./examples/InitializedNotification/initialized-notification.json}
479 > *
480 > * @category `notifications/initialized`
481 > */
482 > export interface InitializedNotification extends JSONRPCNotification {
483 > method: "notifications/initialized";
484 > params?: NotificationParams;
485 > }
486 >
487 > /**
488 > * Capabilities a client may support. Known capabilities are defined here, in this schema, but this is not a closed set: any client can define its own, additional capabilities.
489 > *
490 > * @category `initialize`
491 > */
492 > export interface ClientCapabilities {
493 > /**
494 > * Experimental, non-standard capabilities that the client supports.
495 > */
496 > experimental?: { [key: string]: object };
497 > /**
498 > * Present if the client supports listing roots.
499 > *
500 > * @example Roots - minimum baseline support
501 > * {@includeCode ./examples/ClientCapabilities/roots-minimum-baseline-support.json}
502 > *
503 > * @example Roots - list changed notifications
504 > * {@includeCode ./examples/ClientCapabilities/roots-list-changed-notifications.json}
505 > */
506 > roots?: {
507 > /**
508 > * Whether the client supports notifications for changes to the roots list.
509 > */
510 > listChanged?: boolean;
511 > };
512 > /**
513 > * Present if the client supports sampling from an LLM.
514 > *
515 > * @example Sampling - minimum baseline support
516 > * {@includeCode ./examples/ClientCapabilities/sampling-minimum-baseline-support.json}
517 > *
518 > * @example Sampling - tool use support
519 > * {@includeCode ./examples/ClientCapabilities/sampling-tool-use-support.json}
520 > *
521 > * @example Sampling - context inclusion support (soft-deprecated)
522 > * {@includeCode ./examples/ClientCapabilities/sampling-context-inclusion-support-soft-deprecated.json}
523 > */
524 > sampling?: {
525 > /**
526 > * Whether the client supports context inclusion via `includeContext` parameter.
527 > * If not declared, servers SHOULD only use `includeContext: "none"` (or omit it).
528 > */
529 > context?: object;
530 > /**
531 > * Whether the client supports tool use via `tools` and `toolChoice` parameters.
532 > */
533 > tools?: object;
534 > };
535 > /**
536 > * Present if the client supports elicitation from the server.
537 > *
538 > * @example Elicitation - form and URL mode support
539 > * {@includeCode ./examples/ClientCapabilities/elicitation-form-and-url-mode-support.json}
540 > *
541 > * @example Elicitation - form mode only (implicit)
542 > * {@includeCode ./examples/ClientCapabilities/elicitation-form-only-implicit.json}
543 > */
544 > elicitation?: { form?: object; url?: object };
545 >
546 > /**
547 > * Present if the client supports task-augmented requests.
548 > */
549 > tasks?: {
550 > /**
551 > * Whether this client supports {@link ListTasksRequest | tasks/list}.
552 > */
553 > list?: object;
554 > /**
555 > * Whether this client supports {@link CancelTaskRequest | tasks/cancel}.
556 > */
557 > cancel?: object;
558 > /**
559 > * Specifies which request types can be augmented with tasks.
560 > */
561 > requests?: {
562 > /**
563 > * Task support for sampling-related requests.
564 > */
565 > sampling?: {
566 > /**
567 > * Whether the client supports task-augmented `sampling/createMessage` requests.
568 > */
569 > createMessage?: object;
570 > };
571 > /**
572 > * Task support for elicitation-related requests.
573 > */
574 > elicitation?: {
575 > /**
576 > * Whether the client supports task-augmented {@link ElicitRequest | elicitation/create} requests.
577 > */
578 > create?: object;
579 > };
580 > };
581 > };
582 > /**
583 > * Optional MCP extensions that the client supports. Keys are extension identifiers
584 > * (e.g., "io.modelcontextprotocol/oauth-client-credentials"), and values are
585 > * per-extension settings objects. An empty object indicates support with no settings.
586 > *
587 > * @example Extensions - UI extension with MIME type support
588 > * {@includeCode ./examples/ClientCapabilities/extensions-ui-mime-types.json}
589 > */
590 > extensions?: { [key: string]: object };
591 > }
592 >
593 > /**
594 > * Capabilities that a server may support. Known capabilities are defined here, in this schema, but this is not a closed set: any server can define its own, additional capabilities.
595 > *
596 > * @category `initialize`
597 > */
598 > export interface ServerCapabilities {
599 > /**
600 > * Experimental, non-standard capabilities that the server supports.
601 > */
602 > experimental?: { [key: string]: object };
603 > /**
604 > * Present if the server supports sending log messages to the client.
605 > *
606 > * @example Logging - minimum baseline support
607 > * {@includeCode ./examples/ServerCapabilities/logging-minimum-baseline-support.json}
608 > */
609 > logging?: object;
610 > /**
611 > * Present if the server supports argument autocompletion suggestions.
612 > *
613 > * @example Completions - minimum baseline support
614 > * {@includeCode ./examples/ServerCapabilities/completions-minimum-baseline-support.json}
615 > */
616 > completions?: object;
617 > /**
618 > * Present if the server offers any prompt templates.
619 > *
620 > * @example Prompts - minimum baseline support
621 > * {@includeCode ./examples/ServerCapabilities/prompts-minimum-baseline-support.json}
622 > *
623 > * @example Prompts - list changed notifications
624 > * {@includeCode ./examples/ServerCapabilities/prompts-list-changed-notifications.json}
625 > */
626 > prompts?: {
627 > /**
628 > * Whether this server supports notifications for changes to the prompt list.
629 > */
630 > listChanged?: boolean;
631 > };
632 > /**
633 > * Present if the server offers any resources to read.
634 > *
635 > * @example Resources - minimum baseline support
636 > * {@includeCode ./examples/ServerCapabilities/resources-minimum-baseline-support.json}
637 > *
638 > * @example Resources - subscription to individual resource updates (only)
639 > * {@includeCode ./examples/ServerCapabilities/resources-subscription-to-individual-resource-updates-only.json}
640 > *
641 > * @example Resources - list changed notifications (only)
642 > * {@includeCode ./examples/ServerCapabilities/resources-list-changed-notifications-only.json}
643 > *
644 > * @example Resources - all notifications
645 > * {@includeCode ./examples/ServerCapabilities/resources-all-notifications.json}
646 > */
647 > resources?: {
648 > /**
649 > * Whether this server supports subscribing to resource updates.
650 > */
651 > subscribe?: boolean;
652 > /**
653 > * Whether this server supports notifications for changes to the resource list.
654 > */
655 > listChanged?: boolean;
656 > };
657 > /**
658 > * Present if the server offers any tools to call.
659 > *
660 > * @example Tools - minimum baseline support
661 > * {@includeCode ./examples/ServerCapabilities/tools-minimum-baseline-support.json}
662 > *
663 > * @example Tools - list changed notifications
664 > * {@includeCode ./examples/ServerCapabilities/tools-list-changed-notifications.json}
665 > */
666 > tools?: {
667 > /**
668 > * Whether this server supports notifications for changes to the tool list.
669 > */
670 > listChanged?: boolean;
671 > };
672 > /**
673 > * Present if the server supports task-augmented requests.
674 > */
675 > tasks?: {
676 > /**
677 > * Whether this server supports {@link ListTasksRequest | tasks/list}.
678 > */
679 > list?: object;
680 > /**
681 > * Whether this server supports {@link CancelTaskRequest | tasks/cancel}.
682 > */
683 > cancel?: object;
684 > /**
685 > * Specifies which request types can be augmented with tasks.
686 > */
687 > requests?: {
688 > /**
689 > * Task support for tool-related requests.
690 > */
691 > tools?: {
692 > /**
693 > * Whether the server supports task-augmented {@link CallToolRequest | tools/call} requests.
694 > */
695 > call?: object;
696 > };
697 > };
698 > };
699 > /**
700 > * Optional MCP extensions that the server supports. Keys are extension identifiers
701 > * (e.g., "io.modelcontextprotocol/apps"), and values are per-extension settings
702 > * objects. An empty object indicates support with no settings.
703 > *
704 > * @example Extensions - UI extension support
705 > * {@includeCode ./examples/ServerCapabilities/extensions-ui.json}
706 > */
707 > extensions?: { [key: string]: object };
708 > }
709 >
710 > /**
711 > * An optionally-sized icon that can be displayed in a user interface.
712 > *
713 > * @category Common Types
714 > */
715 > export interface Icon {
716 > /**
717 > * A standard URI pointing to an icon resource. May be an HTTP/HTTPS URL or a
718 > * `data:` URI with Base64-encoded image data.
719 > *
720 > * Consumers SHOULD take steps to ensure URLs serving icons are from the
721 > * same domain as the client/server or a trusted domain.
722 > *
723 > * Consumers SHOULD take appropriate precautions when consuming SVGs as they can contain
724 > * executable JavaScript.
725 > *
726 > * @format uri
727 > */
728 > src: string;
729 >
730 > /**
731 > * Optional MIME type override if the source MIME type is missing or generic.
732 > * For example: `"image/png"`, `"image/jpeg"`, or `"image/svg+xml"`.
733 > */
734 > mimeType?: string;
735 >
736 > /**
737 > * Optional array of strings that specify sizes at which the icon can be used.
738 > * Each string should be in WxH format (e.g., `"48x48"`, `"96x96"`) or `"any"` for scalable formats like SVG.
739 > *
740 > * If not provided, the client should assume that the icon can be used at any size.
741 > */
742 > sizes?: string[];
743 >
744 > /**
745 > * Optional specifier for the theme this icon is designed for. `"light"` indicates
746 > * the icon is designed to be used with a light background, and `"dark"` indicates
747 > * the icon is designed to be used with a dark background.
748 > *
749 > * If not provided, the client should assume the icon can be used with any theme.
750 > */
751 > theme?: "light" | "dark";
752 > }
753 >
754 > /**
755 > * Base interface to add `icons` property.
756 > *
757 > * @internal
758 > */
759 > export interface Icons {
760 > /**
761 > * Optional set of sized icons that the client can display in a user interface.
762 > *
763 > * Clients that support rendering icons MUST support at least the following MIME types:
764 > * - `image/png` - PNG images (safe, universal compatibility)
765 > * - `image/jpeg` (and `image/jpg`) - JPEG images (safe, universal compatibility)
766 > *
767 > * Clients that support rendering icons SHOULD also support:
768 > * - `image/svg+xml` - SVG images (scalable but requires security precautions)
769 > * - `image/webp` - WebP images (modern, efficient format)
770 > */
771 > icons?: Icon[];
772 > }
773 >
774 > /**
775 > * Base interface for metadata with name (identifier) and title (display name) properties.
776 > *
777 > * @internal
778 > */
779 > export interface BaseMetadata {
780 > /**
781 > * Intended for programmatic or logical use, but used as a display name in past specs or fallback (if title isn't present).
782 > */
783 > name: string;
784 >
785 > /**
786 > * Intended for UI and end-user contexts - optimized to be human-readable and easily understood,
787 > * even by those unfamiliar with domain-specific terminology.
788 > *
789 > * If not provided, the name should be used for display (except for {@link Tool},
790 > * where `annotations.title` should be given precedence over using `name`,
791 > * if present).
792 > */
793 > title?: string;
794 > }
795 >
796 > /**
797 > * Describes the MCP implementation.
798 > *
799 > * @category `initialize`
800 > */
801 > export interface Implementation extends BaseMetadata, Icons {
802 > /**
803 > * The version of this implementation.
804 > */
805 > version: string;
806 >
807 > /**
808 > * An optional human-readable description of what this implementation does.
809 > *
810 > * This can be used by clients or servers to provide context about their purpose
811 > * and capabilities. For example, a server might describe the types of resources
812 > * or tools it provides, while a client might describe its intended use case.
813 > */
814 > description?: string;
815 >
816 > /**
817 > * An optional URL of the website for this implementation.
818 > *
819 > * @format uri
820 > */
821 > websiteUrl?: string;
822 > }
823 >
824 > /* Ping */
825 > /**
826 > * A ping, issued by either the server or the client, to check that the other party is still alive. The receiver must promptly respond, or else may be disconnected.
827 > *
828 > * @example Ping request
829 > * {@includeCode ./examples/PingRequest/ping-request.json}
830 > *
831 > * @category `ping`
832 > */
833 > export interface PingRequest extends JSONRPCRequest {
834 > method: "ping";
835 > params?: RequestParams;
836 > }
837 >
838 > /**
839 > * A successful response for a {@link PingRequest | ping} request.
840 > *
841 > * @example Ping result response
842 > * {@includeCode ./examples/PingResultResponse/ping-result-response.json}
843 > *
844 > * @category `ping`
845 > */
846 > export interface PingResultResponse extends JSONRPCResultResponse {
847 > result: EmptyResult;
848 > }
849 >
850 > /* Progress notifications */
851 >
852 > /**
853 > * Parameters for a {@link ProgressNotification | notifications/progress} notification.
854 > *
855 > * @example Progress message
856 > * {@includeCode ./examples/ProgressNotificationParams/progress-message.json}
857 > *
858 > * @category `notifications/progress`
859 > */
860 > export interface ProgressNotificationParams extends NotificationParams {
861 > /**
862 > * The progress token which was given in the initial request, used to associate this notification with the request that is proceeding.
863 > */
864 > progressToken: ProgressToken;
865 > /**
866 > * The progress thus far. This should increase every time progress is made, even if the total is unknown.
867 > *
868 > * @TJS-type number
869 > */
870 > progress: number;
871 > /**
872 > * Total number of items to process (or total progress required), if known.
873 > *
874 > * @TJS-type number
875 > */
876 > total?: number;
877 > /**
878 > * An optional message describing the current progress.
879 > */
880 > message?: string;
881 > }
882 >
883 > /**
884 > * An out-of-band notification used to inform the receiver of a progress update for a long-running request.
885 > *
886 > * @example Progress message
887 > * {@includeCode ./examples/ProgressNotification/progress-message.json}
888 > *
889 > * @category `notifications/progress`
890 > */
891 > export interface ProgressNotification extends JSONRPCNotification {
892 > method: "notifications/progress";
893 > params: ProgressNotificationParams;
894 > }
895 >
896 > /* Pagination */
897 > /**
898 > * Common params for paginated requests.
899 > *
900 > * @example List request with cursor
901 > * {@includeCode ./examples/PaginatedRequestParams/list-with-cursor.json}
902 > *
903 > * @category Common Types
904 > */
905 > export interface PaginatedRequestParams extends RequestParams {
906 > /**
907 > * An opaque token representing the current pagination position.
908 > * If provided, the server should return results starting after this cursor.
909 > */
910 > cursor?: Cursor;
911 > }
912 >
913 > /** @internal */
914 > export interface PaginatedRequest extends JSONRPCRequest {
915 > params?: PaginatedRequestParams;
916 > }
917 >
918 > /** @internal */
919 > export interface PaginatedResult extends Result {
920 > /**
921 > * An opaque token representing the pagination position after the last returned result.
922 > * If present, there may be more results available.
923 > */
924 > nextCursor?: Cursor;
925 > }
926 >
927 > /* Resources */
928 > /**
929 > * Sent from the client to request a list of resources the server has.
930 > *
931 > * @example List resources request
932 > * {@includeCode ./examples/ListResourcesRequest/list-resources-request.json}
933 > *
934 > * @category `resources/list`
935 > */
936 > export interface ListResourcesRequest extends PaginatedRequest {
937 > method: "resources/list";
938 > }
939 >
940 > /**
941 > * The result returned by the server for a {@link ListResourcesRequest | resources/list} request.
942 > *
943 > * @example Resources list with cursor
944 > * {@includeCode ./examples/ListResourcesResult/resources-list-with-cursor.json}
945 > *
946 > * @category `resources/list`
947 > */
948 > export interface ListResourcesResult extends PaginatedResult {
949 > resources: Resource[];
950 > }
951 >
952 > /**
953 > * A successful response from the server for a {@link ListResourcesRequest | resources/list} request.
954 > *
955 > * @example List resources result response
956 > * {@includeCode ./examples/ListResourcesResultResponse/list-resources-result-response.json}
957 > *
958 > * @category `resources/list`
959 > */
960 > export interface ListResourcesResultResponse extends JSONRPCResultResponse {
961 > result: ListResourcesResult;
962 > }
963 >
964 > /**
965 > * Sent from the client to request a list of resource templates the server has.
966 > *
967 > * @example List resource templates request
968 > * {@includeCode ./examples/ListResourceTemplatesRequest/list-resource-templates-request.json}
969 > *
970 > * @category `resources/templates/list`
971 > */
972 > export interface ListResourceTemplatesRequest extends PaginatedRequest {
973 > method: "resources/templates/list";
974 > }
975 >
976 > /**
977 > * The result returned by the server for a {@link ListResourceTemplatesRequest | resources/templates/list} request.
978 > *
979 > * @example Resource templates list
980 > * {@includeCode ./examples/ListResourceTemplatesResult/resource-templates-list.json}
981 > *
982 > * @category `resources/templates/list`
983 > */
984 > export interface ListResourceTemplatesResult extends PaginatedResult {
985 > resourceTemplates: ResourceTemplate[];
986 > }
987 >
988 > /**
989 > * A successful response from the server for a {@link ListResourceTemplatesRequest | resources/templates/list} request.
990 > *
991 > * @example List resource templates result response
992 > * {@includeCode ./examples/ListResourceTemplatesResultResponse/list-resource-templates-result-response.json}
993 > *
994 > * @category `resources/templates/list`
995 > */
996 > export interface ListResourceTemplatesResultResponse extends JSONRPCResultResponse {
997 > result: ListResourceTemplatesResult;
998 > }
999 >
1000 > /**
1001 > * Common params for resource-related requests.
1002 > *
1003 > * @internal
1004 > */
1005 > export interface ResourceRequestParams extends RequestParams {
1006 > /**
1007 > * The URI of the resource. The URI can use any protocol; it is up to the server how to interpret it.
1008 > *
1009 > * @format uri
1010 > */
1011 > uri: string;
1012 > }
1013 >
1014 > /**
1015 > * Parameters for a `resources/read` request.
1016 > *
1017 > * @category `resources/read`
1018 > */
1019 > export interface ReadResourceRequestParams extends ResourceRequestParams { }
1020 >
1021 > /**
1022 > * Sent from the client to the server, to read a specific resource URI.
1023 > *
1024 > * @example Read resource request
1025 > * {@includeCode ./examples/ReadResourceRequest/read-resource-request.json}
1026 > *
1027 > * @category `resources/read`
1028 > */
1029 > export interface ReadResourceRequest extends JSONRPCRequest {
1030 > method: "resources/read";
1031 > params: ReadResourceRequestParams;
1032 > }
1033 >
1034 > /**
1035 > * The result returned by the server for a {@link ReadResourceRequest | resources/read} request.
1036 > *
1037 > * @example File resource contents
1038 > * {@includeCode ./examples/ReadResourceResult/file-resource-contents.json}
1039 > *
1040 > * @category `resources/read`
1041 > */
1042 > export interface ReadResourceResult extends Result {
1043 > contents: (TextResourceContents | BlobResourceContents)[];
1044 > }
1045 >
1046 > /**
1047 > * A successful response from the server for a {@link ReadResourceRequest | resources/read} request.
1048 > *
1049 > * @example Read resource result response
1050 > * {@includeCode ./examples/ReadResourceResultResponse/read-resource-result-response.json}
1051 > *
1052 > * @category `resources/read`
1053 > */
1054 > export interface ReadResourceResultResponse extends JSONRPCResultResponse {
1055 > result: ReadResourceResult;
1056 > }
1057 >
1058 > /**
1059 > * An optional notification from the server to the client, informing it that the list of resources it can read from has changed. This may be issued by servers without any previous subscription from the client.
1060 > *
1061 > * @example Resources list changed
1062 > * {@includeCode ./examples/ResourceListChangedNotification/resources-list-changed.json}
1063 > *
1064 > * @category `notifications/resources/list_changed`
1065 > */
1066 > export interface ResourceListChangedNotification extends JSONRPCNotification {
1067 > method: "notifications/resources/list_changed";
1068 > params?: NotificationParams;
1069 > }
1070 >
1071 > /**
1072 > * Parameters for a `resources/subscribe` request.
1073 > *
1074 > * @example Subscribe to file resource
1075 > * {@includeCode ./examples/SubscribeRequestParams/subscribe-to-file-resource.json}
1076 > *
1077 > * @category `resources/subscribe`
1078 > */
1079 > export interface SubscribeRequestParams extends ResourceRequestParams { }
1080 >
1081 > /**
1082 > * Sent from the client to request {@link ResourceUpdatedNotification | resources/updated} notifications from the server whenever a particular resource changes.
1083 > *
1084 > * @example Subscribe request
1085 > * {@includeCode ./examples/SubscribeRequest/subscribe-request.json}
1086 > *
1087 > * @category `resources/subscribe`
1088 > */
1089 > export interface SubscribeRequest extends JSONRPCRequest {
1090 > method: "resources/subscribe";
1091 > params: SubscribeRequestParams;
1092 > }
1093 >
1094 > /**
1095 > * A successful response from the server for a {@link SubscribeRequest | resources/subscribe} request.
1096 > *
1097 > * @example Subscribe result response
1098 > * {@includeCode ./examples/SubscribeResultResponse/subscribe-result-response.json}
1099 > *
1100 > * @category `resources/subscribe`
1101 > */
1102 > export interface SubscribeResultResponse extends JSONRPCResultResponse {
1103 > result: EmptyResult;
1104 > }
1105 >
1106 > /**
1107 > * Parameters for a `resources/unsubscribe` request.
1108 > *
1109 > * @category `resources/unsubscribe`
1110 > */
1111 > export interface UnsubscribeRequestParams extends ResourceRequestParams { }
1112 >
1113 > /**
1114 > * Sent from the client to request cancellation of {@link ResourceUpdatedNotification | resources/updated} notifications from the server. This should follow a previous {@link SubscribeRequest | resources/subscribe} request.
1115 > *
1116 > * @example Unsubscribe request
1117 > * {@includeCode ./examples/UnsubscribeRequest/unsubscribe-request.json}
1118 > *
1119 > * @category `resources/unsubscribe`
1120 > */
1121 > export interface UnsubscribeRequest extends JSONRPCRequest {
1122 > method: "resources/unsubscribe";
1123 > params: UnsubscribeRequestParams;
1124 > }
1125 >
1126 > /**
1127 > * A successful response from the server for a {@link UnsubscribeRequest | resources/unsubscribe} request.
1128 > *
1129 > * @example Unsubscribe result response
1130 > * {@includeCode ./examples/UnsubscribeResultResponse/unsubscribe-result-response.json}
1131 > *
1132 > * @category `resources/unsubscribe`
1133 > */
1134 > export interface UnsubscribeResultResponse extends JSONRPCResultResponse {
1135 > result: EmptyResult;
1136 > }
1137 >
1138 > /**
1139 > * Parameters for a `notifications/resources/updated` notification.
1140 > *
1141 > * @example File resource updated
1142 > * {@includeCode ./examples/ResourceUpdatedNotificationParams/file-resource-updated.json}
1143 > *
1144 > * @category `notifications/resources/updated`
1145 > */
1146 > export interface ResourceUpdatedNotificationParams extends NotificationParams {
1147 > /**
1148 > * The URI of the resource that has been updated. This might be a sub-resource of the one that the client actually subscribed to.
1149 > *
1150 > * @format uri
1151 > */
1152 > uri: string;
1153 > }
1154 >
1155 > /**
1156 > * A notification from the server to the client, informing it that a resource has changed and may need to be read again. This should only be sent if the client previously sent a {@link SubscribeRequest | resources/subscribe} request.
1157 > *
1158 > * @example File resource updated notification
1159 > * {@includeCode ./examples/ResourceUpdatedNotification/file-resource-updated-notification.json}
1160 > *
1161 > * @category `notifications/resources/updated`
1162 > */
1163 > export interface ResourceUpdatedNotification extends JSONRPCNotification {
1164 > method: "notifications/resources/updated";
1165 > params: ResourceUpdatedNotificationParams;
1166 > }
1167 >
1168 > /**
1169 > * A known resource that the server is capable of reading.
1170 > *
1171 > * @example File resource with annotations
1172 > * {@includeCode ./examples/Resource/file-resource-with-annotations.json}
1173 > *
1174 > * @category `resources/list`
1175 > */
1176 > export interface Resource extends BaseMetadata, Icons {
1177 > /**
1178 > * The URI of this resource.
1179 > *
1180 > * @format uri
1181 > */
1182 > uri: string;
1183 >
1184 > /**
1185 > * A description of what this resource represents.
1186 > *
1187 > * This can be used by clients to improve the LLM's understanding of available resources. It can be thought of like a "hint" to the model.
1188 > */
1189 > description?: string;
1190 >
1191 > /**
1192 > * The MIME type of this resource, if known.
1193 > */
1194 > mimeType?: string;
1195 >
1196 > /**
1197 > * Optional annotations for the client.
1198 > */
1199 > annotations?: Annotations;
1200 >
1201 > /**
1202 > * The size of the raw resource content, in bytes (i.e., before base64 encoding or any tokenization), if known.
1203 > *
1204 > * This can be used by Hosts to display file sizes and estimate context window usage.
1205 > */
1206 > size?: number;
1207 >
1208 > _meta?: MetaObject;
1209 > }
1210 >
1211 > /**
1212 > * A template description for resources available on the server.
1213 > *
1214 > * @category `resources/templates/list`
1215 > */
1216 > export interface ResourceTemplate extends BaseMetadata, Icons {
1217 > /**
1218 > * A URI template (according to RFC 6570) that can be used to construct resource URIs.
1219 > *
1220 > * @format uri-template
1221 > */
1222 > uriTemplate: string;
1223 >
1224 > /**
1225 > * A description of what this template is for.
1226 > *
1227 > * This can be used by clients to improve the LLM's understanding of available resources. It can be thought of like a "hint" to the model.
1228 > */
1229 > description?: string;
1230 >
1231 > /**
1232 > * The MIME type for all resources that match this template. This should only be included if all resources matching this template have the same type.
1233 > */
1234 > mimeType?: string;
1235 >
1236 > /**
1237 > * Optional annotations for the client.
1238 > */
1239 > annotations?: Annotations;
1240 >
1241 > _meta?: MetaObject;
1242 > }
1243 >
1244 > /**
1245 > * The contents of a specific resource or sub-resource.
1246 > *
1247 > * @internal
1248 > */
1249 > export interface ResourceContents {
1250 > /**
1251 > * The URI of this resource.
1252 > *
1253 > * @format uri
1254 > */
1255 > uri: string;
1256 > /**
1257 > * The MIME type of this resource, if known.
1258 > */
1259 > mimeType?: string;
1260 >
1261 > _meta?: MetaObject;
1262 > }
1263 >
1264 > /**
1265 > * @example Text file contents
1266 > * {@includeCode ./examples/TextResourceContents/text-file-contents.json}
1267 > *
1268 > * @category Content
1269 > */
1270 > export interface TextResourceContents extends ResourceContents {
1271 > /**
1272 > * The text of the item. This must only be set if the item can actually be represented as text (not binary data).
1273 > */
1274 > text: string;
1275 > }
1276 >
1277 > /**
1278 > * @example Image file contents
1279 > * {@includeCode ./examples/BlobResourceContents/image-file-contents.json}
1280 > *
1281 > * @category Content
1282 > */
1283 > export interface BlobResourceContents extends ResourceContents {
1284 > /**
1285 > * A base64-encoded string representing the binary data of the item.
1286 > *
1287 > * @format byte
1288 > */
1289 > blob: string;
1290 > }
1291 >
1292 > /* Prompts */
1293 > /**
1294 > * Sent from the client to request a list of prompts and prompt templates the server has.
1295 > *
1296 > * @example List prompts request
1297 > * {@includeCode ./examples/ListPromptsRequest/list-prompts-request.json}
1298 > *
1299 > * @category `prompts/list`
1300 > */
1301 > export interface ListPromptsRequest extends PaginatedRequest {
1302 > method: "prompts/list";
1303 > }
1304 >
1305 > /**
1306 > * The result returned by the server for a {@link ListPromptsRequest | prompts/list} request.
1307 > *
1308 > * @example Prompts list with cursor
1309 > * {@includeCode ./examples/ListPromptsResult/prompts-list-with-cursor.json}
1310 > *
1311 > * @category `prompts/list`
1312 > */
1313 > export interface ListPromptsResult extends PaginatedResult {
1314 > prompts: Prompt[];
1315 > }
1316 >
1317 > /**
1318 > * A successful response from the server for a {@link ListPromptsRequest | prompts/list} request.
1319 > *
1320 > * @example List prompts result response
1321 > * {@includeCode ./examples/ListPromptsResultResponse/list-prompts-result-response.json}
1322 > *
1323 > * @category `prompts/list`
1324 > */
1325 > export interface ListPromptsResultResponse extends JSONRPCResultResponse {
1326 > result: ListPromptsResult;
1327 > }
1328 >
1329 > /**
1330 > * Parameters for a `prompts/get` request.
1331 > *
1332 > * @example Get code review prompt
1333 > * {@includeCode ./examples/GetPromptRequestParams/get-code-review-prompt.json}
1334 > *
1335 > * @category `prompts/get`
1336 > */
1337 > export interface GetPromptRequestParams extends RequestParams {
1338 > /**
1339 > * The name of the prompt or prompt template.
1340 > */
1341 > name: string;
1342 > /**
1343 > * Arguments to use for templating the prompt.
1344 > */
1345 > arguments?: { [key: string]: string };
1346 > }
1347 >
1348 > /**
1349 > * Used by the client to get a prompt provided by the server.
1350 > *
1351 > * @example Get prompt request
1352 > * {@includeCode ./examples/GetPromptRequest/get-prompt-request.json}
1353 > *
1354 > * @category `prompts/get`
1355 > */
1356 > export interface GetPromptRequest extends JSONRPCRequest {
1357 > method: "prompts/get";
1358 > params: GetPromptRequestParams;
1359 > }
1360 >
1361 > /**
1362 > * The result returned by the server for a {@link GetPromptRequest | prompts/get} request.
1363 > *
1364 > * @example Code review prompt
1365 > * {@includeCode ./examples/GetPromptResult/code-review-prompt.json}
1366 > *
1367 > * @category `prompts/get`
1368 > */
1369 > export interface GetPromptResult extends Result {
1370 > /**
1371 > * An optional description for the prompt.
1372 > */
1373 > description?: string;
1374 > messages: PromptMessage[];
1375 > }
1376 >
1377 > /**
1378 > * A successful response from the server for a {@link GetPromptRequest | prompts/get} request.
1379 > *
1380 > * @example Get prompt result response
1381 > * {@includeCode ./examples/GetPromptResultResponse/get-prompt-result-response.json}
1382 > *
1383 > * @category `prompts/get`
1384 > */
1385 > export interface GetPromptResultResponse extends JSONRPCResultResponse {
1386 > result: GetPromptResult;
1387 > }
1388 >
1389 > /**
1390 > * A prompt or prompt template that the server offers.
1391 > *
1392 > * @category `prompts/list`
1393 > */
1394 > export interface Prompt extends BaseMetadata, Icons {
1395 > /**
1396 > * An optional description of what this prompt provides
1397 > */
1398 > description?: string;
1399 >
1400 > /**
1401 > * A list of arguments to use for templating the prompt.
1402 > */
1403 > arguments?: PromptArgument[];
1404 >
1405 > _meta?: MetaObject;
1406 > }
1407 >
1408 > /**
1409 > * Describes an argument that a prompt can accept.
1410 > *
1411 > * @category `prompts/list`
1412 > */
1413 > export interface PromptArgument extends BaseMetadata {
1414 > /**
1415 > * A human-readable description of the argument.
1416 > */
1417 > description?: string;
1418 > /**
1419 > * Whether this argument must be provided.
1420 > */
1421 > required?: boolean;
1422 > }
1423 >
1424 > /**
1425 > * The sender or recipient of messages and data in a conversation.
1426 > *
1427 > * @category Common Types
1428 > */
1429 > export type Role = "user" | "assistant";
1430 >
1431 > /**
1432 > * Describes a message returned as part of a prompt.
1433 > *
1434 > * This is similar to {@link SamplingMessage}, but also supports the embedding of
1435 > * resources from the MCP server.
1436 > *
1437 > * @category `prompts/get`
1438 > */
1439 > export interface PromptMessage {
1440 > role: Role;
1441 > content: ContentBlock;
1442 > }
1443 >
1444 > /**
1445 > * A resource that the server is capable of reading, included in a prompt or tool call result.
1446 > *
1447 > * Note: resource links returned by tools are not guaranteed to appear in the results of {@link ListResourcesRequest | resources/list} requests.
1448 > *
1449 > * @example File resource link
1450 > * {@includeCode ./examples/ResourceLink/file-resource-link.json}
1451 > *
1452 > * @category Content
1453 > */
1454 > export interface ResourceLink extends Resource {
1455 > type: "resource_link";
1456 > }
1457 >
1458 > /**
1459 > * The contents of a resource, embedded into a prompt or tool call result.
1460 > *
1461 > * It is up to the client how best to render embedded resources for the benefit
1462 > * of the LLM and/or the user.
1463 > *
1464 > * @example Embedded file resource with annotations
1465 > * {@includeCode ./examples/EmbeddedResource/embedded-file-resource-with-annotations.json}
1466 > *
1467 > * @category Content
1468 > */
1469 > export interface EmbeddedResource {
1470 > type: "resource";
1471 > resource: TextResourceContents | BlobResourceContents;
1472 >
1473 > /**
1474 > * Optional annotations for the client.
1475 > */
1476 > annotations?: Annotations;
1477 >
1478 > _meta?: MetaObject;
1479 > }
1480 > /**
1481 > * An optional notification from the server to the client, informing it that the list of prompts it offers has changed. This may be issued by servers without any previous subscription from the client.
1482 > *
1483 > * @example Prompts list changed
1484 > * {@includeCode ./examples/PromptListChangedNotification/prompts-list-changed.json}
1485 > *
1486 > * @category `notifications/prompts/list_changed`
1487 > */
1488 > export interface PromptListChangedNotification extends JSONRPCNotification {
1489 > method: "notifications/prompts/list_changed";
1490 > params?: NotificationParams;
1491 > }
1492 >
1493 > /* Tools */
1494 > /**
1495 > * Sent from the client to request a list of tools the server has.
1496 > *
1497 > * @example List tools request
1498 > * {@includeCode ./examples/ListToolsRequest/list-tools-request.json}
1499 > *
1500 > * @category `tools/list`
1501 > */
1502 > export interface ListToolsRequest extends PaginatedRequest {
1503 > method: "tools/list";
1504 > }
1505 >
1506 > /**
1507 > * The result returned by the server for a {@link ListToolsRequest | tools/list} request.
1508 > *
1509 > * @example Tools list with cursor
1510 > * {@includeCode ./examples/ListToolsResult/tools-list-with-cursor.json}
1511 > *
1512 > * @category `tools/list`
1513 > */
1514 > export interface ListToolsResult extends PaginatedResult {
1515 > tools: Tool[];
1516 > }
1517 >
1518 > /**
1519 > * A successful response from the server for a {@link ListToolsRequest | tools/list} request.
1520 > *
1521 > * @example List tools result response
1522 > * {@includeCode ./examples/ListToolsResultResponse/list-tools-result-response.json}
1523 > *
1524 > * @category `tools/list`
1525 > */
1526 > export interface ListToolsResultResponse extends JSONRPCResultResponse {
1527 > result: ListToolsResult;
1528 > }
1529 >
1530 > /**
1531 > * The result returned by the server for a {@link CallToolRequest | tools/call} request.
1532 > *
1533 > * @example Result with unstructured text
1534 > * {@includeCode ./examples/CallToolResult/result-with-unstructured-text.json}
1535 > *
1536 > * @example Result with structured content
1537 > * {@includeCode ./examples/CallToolResult/result-with-structured-content.json}
1538 > *
1539 > * @example Invalid tool input error
1540 > * {@includeCode ./examples/CallToolResult/invalid-tool-input-error.json}
1541 > *
1542 > * @category `tools/call`
1543 > */
1544 > export interface CallToolResult extends Result {
1545 > /**
1546 > * A list of content objects that represent the unstructured result of the tool call.
1547 > */
1548 > content: ContentBlock[];
1549 >
1550 > /**
1551 > * An optional JSON object that represents the structured result of the tool call.
1552 > */
1553 > structuredContent?: { [key: string]: unknown };
1554 >
1555 > /**
1556 > * Whether the tool call ended in an error.
1557 > *
1558 > * If not set, this is assumed to be false (the call was successful).
1559 > *
1560 > * Any errors that originate from the tool SHOULD be reported inside the result
1561 > * object, with `isError` set to true, _not_ as an MCP protocol-level error
1562 > * response. Otherwise, the LLM would not be able to see that an error occurred
1563 > * and self-correct.
1564 > *
1565 > * However, any errors in _finding_ the tool, an error indicating that the
1566 > * server does not support tool calls, or any other exceptional conditions,
1567 > * should be reported as an MCP error response.
1568 > */
1569 > isError?: boolean;
1570 > }
1571 >
1572 > /**
1573 > * A successful response from the server for a {@link CallToolRequest | tools/call} request.
1574 > *
1575 > * @example Call tool result response
1576 > * {@includeCode ./examples/CallToolResultResponse/call-tool-result-response.json}
1577 > *
1578 > * @category `tools/call`
1579 > */
1580 > export interface CallToolResultResponse extends JSONRPCResultResponse {
1581 > result: CallToolResult;
1582 > }
1583 >
1584 > /**
1585 > * Parameters for a `tools/call` request.
1586 > *
1587 > * @example `get_weather` tool call params
1588 > * {@includeCode ./examples/CallToolRequestParams/get-weather-tool-call-params.json}
1589 > *
1590 > * @example Tool call params with progress token
1591 > * {@includeCode ./examples/CallToolRequestParams/tool-call-params-with-progress-token.json}
1592 > *
1593 > * @category `tools/call`
1594 > */
1595 > export interface CallToolRequestParams extends TaskAugmentedRequestParams {
1596 > /**
1597 > * The name of the tool.
1598 > */
1599 > name: string;
1600 > /**
1601 > * Arguments to use for the tool call.
1602 > */
1603 > arguments?: { [key: string]: unknown };
1604 > }
1605 >
1606 > /**
1607 > * Used by the client to invoke a tool provided by the server.
1608 > *
1609 > * @example Call tool request
1610 > * {@includeCode ./examples/CallToolRequest/call-tool-request.json}
1611 > *
1612 > * @category `tools/call`
1613 > */
1614 > export interface CallToolRequest extends JSONRPCRequest {
1615 > method: "tools/call";
1616 > params: CallToolRequestParams;
1617 > }
1618 >
1619 > /**
1620 > * An optional notification from the server to the client, informing it that the list of tools it offers has changed. This may be issued by servers without any previous subscription from the client.
1621 > *
1622 > * @example Tools list changed
1623 > * {@includeCode ./examples/ToolListChangedNotification/tools-list-changed.json}
1624 > *
1625 > * @category `notifications/tools/list_changed`
1626 > */
1627 > export interface ToolListChangedNotification extends JSONRPCNotification {
1628 > method: "notifications/tools/list_changed";
1629 > params?: NotificationParams;
1630 > }
1631 >
1632 > /**
1633 > * Additional properties describing a {@link Tool} to clients.
1634 > *
1635 > * NOTE: all properties in `ToolAnnotations` are **hints**.
1636 > * They are not guaranteed to provide a faithful description of
1637 > * tool behavior (including descriptive properties like `title`).
1638 > *
1639 > * Clients should never make tool use decisions based on `ToolAnnotations`
1640 > * received from untrusted servers.
1641 > *
1642 > * @category `tools/list`
1643 > */
1644 > export interface ToolAnnotations {
1645 > /**
1646 > * A human-readable title for the tool.
1647 > */
1648 > title?: string;
1649 >
1650 > /**
1651 > * If true, the tool does not modify its environment.
1652 > *
1653 > * Default: false
1654 > */
1655 > readOnlyHint?: boolean;
1656 >
1657 > /**
1658 > * If true, the tool may perform destructive updates to its environment.
1659 > * If false, the tool performs only additive updates.
1660 > *
1661 > * (This property is meaningful only when `readOnlyHint == false`)
1662 > *
1663 > * Default: true
1664 > */
1665 > destructiveHint?: boolean;
1666 >
1667 > /**
1668 > * If true, calling the tool repeatedly with the same arguments
1669 > * will have no additional effect on its environment.
1670 > *
1671 > * (This property is meaningful only when `readOnlyHint == false`)
1672 > *
1673 > * Default: false
1674 > */
1675 > idempotentHint?: boolean;
1676 >
1677 > /**
1678 > * If true, this tool may interact with an "open world" of external
1679 > * entities. If false, the tool's domain of interaction is closed.
1680 > * For example, the world of a web search tool is open, whereas that
1681 > * of a memory tool is not.
1682 > *
1683 > * Default: true
1684 > */
1685 > openWorldHint?: boolean;
1686 > }
1687 >
1688 > /**
1689 > * Execution-related properties for a tool.
1690 > *
1691 > * @category `tools/list`
1692 > */
1693 > export interface ToolExecution {
1694 > /**
1695 > * Indicates whether this tool supports task-augmented execution.
1696 > * This allows clients to handle long-running operations through polling
1697 > * the task system.
1698 > *
1699 > * - `"forbidden"`: Tool does not support task-augmented execution (default when absent)
1700 > * - `"optional"`: Tool may support task-augmented execution
1701 > * - `"required"`: Tool requires task-augmented execution
1702 > *
1703 > * Default: `"forbidden"`
1704 > */
1705 > taskSupport?: "forbidden" | "optional" | "required";
1706 > }
1707 >
1708 > /**
1709 > * Definition for a tool the client can call.
1710 > *
1711 > * @example With default 2020-12 input schema
1712 > * {@includeCode ./examples/Tool/with-default-2020-12-input-schema.json}
1713 > *
1714 > * @example With explicit draft-07 input schema
1715 > * {@includeCode ./examples/Tool/with-explicit-draft-07-input-schema.json}
1716 > *
1717 > * @example With no parameters
1718 > * {@includeCode ./examples/Tool/with-no-parameters.json}
1719 > *
1720 > * @example With output schema for structured content
1721 > * {@includeCode ./examples/Tool/with-output-schema-for-structured-content.json}
1722 > *
1723 > * @category `tools/list`
1724 > */
1725 > export interface Tool extends BaseMetadata, Icons {
1726 > /**
1727 > * A human-readable description of the tool.
1728 > *
1729 > * This can be used by clients to improve the LLM's understanding of available tools. It can be thought of like a "hint" to the model.
1730 > */
1731 > description?: string;
1732 >
1733 > /**
1734 > * A JSON Schema object defining the expected parameters for the tool.
1735 > */
1736 > inputSchema: {
1737 > $schema?: string;
1738 > type: "object";
1739 > properties?: { [key: string]: object };
1740 > required?: string[];
1741 > };
1742 >
1743 > /**
1744 > * Execution-related properties for this tool.
1745 > */
1746 > execution?: ToolExecution;
1747 >
1748 > /**
1749 > * An optional JSON Schema object defining the structure of the tool's output returned in
1750 > * the structuredContent field of a {@link CallToolResult}.
1751 > *
1752 > * Defaults to JSON Schema 2020-12 when no explicit `$schema` is provided.
1753 > * Currently restricted to `type: "object"` at the root level.
1754 > */
1755 > outputSchema?: {
1756 > $schema?: string;
1757 > type: "object";
1758 > properties?: { [key: string]: object };
1759 > required?: string[];
1760 > };
1761 >
1762 > /**
1763 > * Optional additional tool information.
1764 > *
1765 > * Display name precedence order is: `title`, `annotations.title`, then `name`.
1766 > */
1767 > annotations?: ToolAnnotations;
1768 >
1769 > _meta?: MetaObject;
1770 > }
1771 >
1772 > /* Tasks */
1773 >
1774 > /**
1775 > * The status of a task.
1776 > *
1777 > * @category `tasks`
1778 > */
1779 > export type TaskStatus =
1780 > | "working" // The request is currently being processed
1781 > | "input_required" // The task is waiting for input (e.g., elicitation or sampling)
1782 > | "completed" // The request completed successfully and results are available
1783 > | "failed" // The associated request did not complete successfully. For tool calls specifically, this includes cases where the tool call result has `isError` set to true.
1784 > | "cancelled"; // The request was cancelled before completion
1785 >
1786 > /**
1787 > * Metadata for augmenting a request with task execution.
1788 > * Include this in the `task` field of the request parameters.
1789 > *
1790 > * @category `tasks`
1791 > */
1792 > export interface TaskMetadata {
1793 > /**
1794 > * Requested duration in milliseconds to retain task from creation.
1795 > */
1796 > ttl?: number;
1797 > }
1798 >
1799 > /**
1800 > * Metadata for associating messages with a task.
1801 > * Include this in the `_meta` field under the key `io.modelcontextprotocol/related-task`.
1802 > *
1803 > * @category `tasks`
1804 > */
1805 > export interface RelatedTaskMetadata {
1806 > /**
1807 > * The task identifier this message is associated with.
1808 > */
1809 > taskId: string;
1810 > }
1811 >
1812 > /**
1813 > * Data associated with a task.
1814 > *
1815 > * @category `tasks`
1816 > */
1817 > export interface Task {
1818 > /**
1819 > * The task identifier.
1820 > */
1821 > taskId: string;
1822 >
1823 > /**
1824 > * Current task state.
1825 > */
1826 > status: TaskStatus;
1827 >
1828 > /**
1829 > * Optional human-readable message describing the current task state.
1830 > * This can provide context for any status, including:
1831 > * - Reasons for "cancelled" status
1832 > * - Summaries for "completed" status
1833 > * - Diagnostic information for "failed" status (e.g., error details, what went wrong)
1834 > */
1835 > statusMessage?: string;
1836 >
1837 > /**
1838 > * ISO 8601 timestamp when the task was created.
1839 > */
1840 > createdAt: string;
1841 >
1842 > /**
1843 > * ISO 8601 timestamp when the task was last updated.
1844 > */
1845 > lastUpdatedAt: string;
1846 >
1847 > /**
1848 > * Actual retention duration from creation in milliseconds, null for unlimited.
1849 > */
1850 > ttl: number | null;
1851 >
1852 > /**
1853 > * Suggested polling interval in milliseconds.
1854 > */
1855 > pollInterval?: number;
1856 > }
1857 >
1858 > /**
1859 > * The result returned for a task-augmented request.
1860 > *
1861 > * @category `tasks`
1862 > */
1863 > export interface CreateTaskResult extends Result {
1864 > task: Task;
1865 > }
1866 >
1867 > /**
1868 > * A successful response for a task-augmented request.
1869 > *
1870 > * @category `tasks`
1871 > */
1872 > export interface CreateTaskResultResponse extends JSONRPCResultResponse {
1873 > result: CreateTaskResult;
1874 > }
1875 >
1876 > /**
1877 > * A request to retrieve the state of a task.
1878 > *
1879 > * @category `tasks/get`
1880 > */
1881 > export interface GetTaskRequest extends JSONRPCRequest {
1882 > method: "tasks/get";
1883 > params: {
1884 > /**
1885 > * The task identifier to query.
1886 > */
1887 > taskId: string;
1888 > };
1889 > }
1890 >
1891 > /**
1892 > * The result returned for a {@link GetTaskRequest | tasks/get} request.
1893 > *
1894 > * @category `tasks/get`
1895 > */
1896 > export type GetTaskResult = Result & Task;
1897 >
1898 > /**
1899 > * A successful response for a {@link GetTaskRequest | tasks/get} request.
1900 > *
1901 > * @category `tasks/get`
1902 > */
1903 > export interface GetTaskResultResponse extends JSONRPCResultResponse {
1904 > result: GetTaskResult;
1905 > }
1906 >
1907 > /**
1908 > * A request to retrieve the result of a completed task.
1909 > *
1910 > * @category `tasks/result`
1911 > */
1912 > export interface GetTaskPayloadRequest extends JSONRPCRequest {
1913 > method: "tasks/result";
1914 > params: {
1915 > /**
1916 > * The task identifier to retrieve results for.
1917 > */
1918 > taskId: string;
1919 > };
1920 > }
1921 >
1922 > /**
1923 > * The result returned for a {@link GetTaskPayloadRequest | tasks/result} request.
1924 > * The structure matches the result type of the original request.
1925 > * For example, a {@link CallToolRequest | tools/call} task would return the {@link CallToolResult} structure.
1926 > *
1927 > * @category `tasks/result`
1928 > */
1929 > export interface GetTaskPayloadResult extends Result {
1930 > [key: string]: unknown;
1931 > }
1932 >
1933 > /**
1934 > * A successful response for a {@link GetTaskPayloadRequest | tasks/result} request.
1935 > *
1936 > * @category `tasks/result`
1937 > */
1938 > export interface GetTaskPayloadResultResponse extends JSONRPCResultResponse {
1939 > result: GetTaskPayloadResult;
1940 > }
1941 >
1942 > /**
1943 > * A request to cancel a task.
1944 > *
1945 > * @category `tasks/cancel`
1946 > */
1947 > export interface CancelTaskRequest extends JSONRPCRequest {
1948 > method: "tasks/cancel";
1949 > params: {
1950 > /**
1951 > * The task identifier to cancel.
1952 > */
1953 > taskId: string;
1954 > };
1955 > }
1956 >
1957 > /**
1958 > * The result returned for a {@link CancelTaskRequest | tasks/cancel} request.
1959 > *
1960 > * @category `tasks/cancel`
1961 > */
1962 > export type CancelTaskResult = Result & Task;
1963 >
1964 > /**
1965 > * A successful response for a {@link CancelTaskRequest | tasks/cancel} request.
1966 > *
1967 > * @category `tasks/cancel`
1968 > */
1969 > export interface CancelTaskResultResponse extends JSONRPCResultResponse {
1970 > result: CancelTaskResult;
1971 > }
1972 >
1973 > /**
1974 > * A request to retrieve a list of tasks.
1975 > *
1976 > * @category `tasks/list`
1977 > */
1978 > export interface ListTasksRequest extends PaginatedRequest {
1979 > method: "tasks/list";
1980 > }
1981 >
1982 > /**
1983 > * The result returned for a {@link ListTasksRequest | tasks/list} request.
1984 > *
1985 > * @category `tasks/list`
1986 > */
1987 > export interface ListTasksResult extends PaginatedResult {
1988 > tasks: Task[];
1989 > }
1990 >
1991 > /**
1992 > * A successful response for a {@link ListTasksRequest | tasks/list} request.
1993 > *
1994 > * @category `tasks/list`
1995 > */
1996 > export interface ListTasksResultResponse extends JSONRPCResultResponse {
1997 > result: ListTasksResult;
1998 > }
1999 >
2000 > /**
2001 > * Parameters for a `notifications/tasks/status` notification.
2002 > *
2003 > * @category `notifications/tasks/status`
2004 > */
2005 > export type TaskStatusNotificationParams = NotificationParams & Task;
2006 >
2007 > /**
2008 > * An optional notification from the receiver to the requestor, informing them that a task's status has changed. Receivers are not required to send these notifications.
2009 > *
2010 > * @category `notifications/tasks/status`
2011 > */
2012 > export interface TaskStatusNotification extends JSONRPCNotification {
2013 > method: "notifications/tasks/status";
2014 > params: TaskStatusNotificationParams;
2015 > }
2016 >
2017 > /* Logging */
2018 >
2019 > /**
2020 > * Parameters for a `logging/setLevel` request.
2021 > *
2022 > * @example Set log level to "info"
2023 > * {@includeCode ./examples/SetLevelRequestParams/set-log-level-to-info.json}
2024 > *
2025 > * @category `logging/setLevel`
2026 > */
2027 > export interface SetLevelRequestParams extends RequestParams {
2028 > /**
2029 > * The level of logging that the client wants to receive from the server. The server should send all logs at this level and higher (i.e., more severe) to the client as {@link LoggingMessageNotification | notifications/message}.
2030 > */
2031 > level: LoggingLevel;
2032 > }
2033 >
2034 > /**
2035 > * A request from the client to the server, to enable or adjust logging.
2036 > *
2037 > * @example Set logging level request
2038 > * {@includeCode ./examples/SetLevelRequest/set-logging-level-request.json}
2039 > *
2040 > * @category `logging/setLevel`
2041 > */
2042 > export interface SetLevelRequest extends JSONRPCRequest {
2043 > method: "logging/setLevel";
2044 > params: SetLevelRequestParams;
2045 > }
2046 >
2047 > /**
2048 > * A successful response from the server for a {@link SetLevelRequest | logging/setLevel} request.
2049 > *
2050 > * @example Set logging level result response
2051 > * {@includeCode ./examples/SetLevelResultResponse/set-logging-level-result-response.json}
2052 > *
2053 > * @category `logging/setLevel`
2054 > */
2055 > export interface SetLevelResultResponse extends JSONRPCResultResponse {
2056 > result: EmptyResult;
2057 > }
2058 >
2059 > /**
2060 > * Parameters for a `notifications/message` notification.
2061 > *
2062 > * @example Log database connection failed
2063 > * {@includeCode ./examples/LoggingMessageNotificationParams/log-database-connection-failed.json}
2064 > *
2065 > * @category `notifications/message`
2066 > */
2067 > export interface LoggingMessageNotificationParams extends NotificationParams {
2068 > /**
2069 > * The severity of this log message.
2070 > */
2071 > level: LoggingLevel;
2072 > /**
2073 > * An optional name of the logger issuing this message.
2074 > */
2075 > logger?: string;
2076 > /**
2077 > * The data to be logged, such as a string message or an object. Any JSON serializable type is allowed here.
2078 > */
2079 > data: unknown;
2080 > }
2081 >
2082 > /**
2083 > * JSONRPCNotification of a log message passed from server to client. If no `logging/setLevel` request has been sent from the client, the server MAY decide which messages to send automatically.
2084 > *
2085 > * @example Log database connection failed
2086 > * {@includeCode ./examples/LoggingMessageNotification/log-database-connection-failed.json}
2087 > *
2088 > * @category `notifications/message`
2089 > */
2090 > export interface LoggingMessageNotification extends JSONRPCNotification {
2091 > method: "notifications/message";
2092 > params: LoggingMessageNotificationParams;
2093 > }
2094 >
2095 > /**
2096 > * The severity of a log message.
2097 > *
2098 > * These map to syslog message severities, as specified in RFC-5424:
2099 > * https://datatracker.ietf.org/doc/html/rfc5424#section-6.2.1
2100 > *
2101 > * @category Common Types
2102 > */
2103 > export type LoggingLevel =
2104 > | "debug"
2105 > | "info"
2106 > | "notice"
2107 > | "warning"
2108 > | "error"
2109 > | "critical"
2110 > | "alert"
2111 > | "emergency";
2112 >
2113 > /* Sampling */
2114 > /**
2115 > * Parameters for a `sampling/createMessage` request.
2116 > *
2117 > * @example Basic request
2118 > * {@includeCode ./examples/CreateMessageRequestParams/basic-request.json}
2119 > *
2120 > * @example Request with tools
2121 > * {@includeCode ./examples/CreateMessageRequestParams/request-with-tools.json}
2122 > *
2123 > * @example Follow-up request with tool results
2124 > * {@includeCode ./examples/CreateMessageRequestParams/follow-up-with-tool-results.json}
2125 > *
2126 > * @category `sampling/createMessage`
2127 > */
2128 > export interface CreateMessageRequestParams extends TaskAugmentedRequestParams {
2129 > messages: SamplingMessage[];
2130 > /**
2131 > * The server's preferences for which model to select. The client MAY ignore these preferences.
2132 > */
2133 > modelPreferences?: ModelPreferences;
2134 > /**
2135 > * An optional system prompt the server wants to use for sampling. The client MAY modify or omit this prompt.
2136 > */
2137 > systemPrompt?: string;
2138 > /**
2139 > * A request to include context from one or more MCP servers (including the caller), to be attached to the prompt.
2140 > * The client MAY ignore this request.
2141 > *
2142 > * Default is `"none"`. Values `"thisServer"` and `"allServers"` are soft-deprecated. Servers SHOULD only use these values if the client
2143 > * declares {@link ClientCapabilities.sampling.context}. These values may be removed in future spec releases.
2144 > */
2145 > includeContext?: "none" | "thisServer" | "allServers";
2146 > /**
2147 > * @TJS-type number
2148 > */
2149 > temperature?: number;
2150 > /**
2151 > * The requested maximum number of tokens to sample (to prevent runaway completions).
2152 > *
2153 > * The client MAY choose to sample fewer tokens than the requested maximum.
2154 > */
2155 > maxTokens: number;
2156 > stopSequences?: string[];
2157 > /**
2158 > * Optional metadata to pass through to the LLM provider. The format of this metadata is provider-specific.
2159 > */
2160 > metadata?: object;
2161 > /**
2162 > * Tools that the model may use during generation.
2163 > * The client MUST return an error if this field is provided but {@link ClientCapabilities.sampling.tools} is not declared.
2164 > */
2165 > tools?: Tool[];
2166 > /**
2167 > * Controls how the model uses tools.
2168 > * The client MUST return an error if this field is provided but {@link ClientCapabilities.sampling.tools} is not declared.
2169 > * Default is `{ mode: "auto" }`.
2170 > */
2171 > toolChoice?: ToolChoice;
2172 > }
2173 >
2174 > /**
2175 > * Controls tool selection behavior for sampling requests.
2176 > *
2177 > * @category `sampling/createMessage`
2178 > */
2179 > export interface ToolChoice {
2180 > /**
2181 > * Controls the tool use ability of the model:
2182 > * - `"auto"`: Model decides whether to use tools (default)
2183 > * - `"required"`: Model MUST use at least one tool before completing
2184 > * - `"none"`: Model MUST NOT use any tools
2185 > */
2186 > mode?: "auto" | "required" | "none";
2187 > }
2188 >
2189 > /**
2190 > * A request from the server to sample an LLM via the client. The client has full discretion over which model to select. The client should also inform the user before beginning sampling, to allow them to inspect the request (human in the loop) and decide whether to approve it.
2191 > *
2192 > * @example Sampling request
2193 > * {@includeCode ./examples/CreateMessageRequest/sampling-request.json}
2194 > *
2195 > * @category `sampling/createMessage`
2196 > */
2197 > export interface CreateMessageRequest extends JSONRPCRequest {
2198 > method: "sampling/createMessage";
2199 > params: CreateMessageRequestParams;
2200 > }
2201 >
2202 > /**
2203 > * The result returned by the client for a {@link CreateMessageRequest | sampling/createMessage} request.
2204 > * The client should inform the user before returning the sampled message, to allow them
2205 > * to inspect the response (human in the loop) and decide whether to allow the server to see it.
2206 > *
2207 > * @example Text response
2208 > * {@includeCode ./examples/CreateMessageResult/text-response.json}
2209 > *
2210 > * @example Tool use response
2211 > * {@includeCode ./examples/CreateMessageResult/tool-use-response.json}
2212 > *
2213 > * @example Final response after tool use
2214 > * {@includeCode ./examples/CreateMessageResult/final-response.json}
2215 > *
2216 > * @category `sampling/createMessage`
2217 > */
2218 > export interface CreateMessageResult extends Result, SamplingMessage {
2219 > /**
2220 > * The name of the model that generated the message.
2221 > */
2222 > model: string;
2223 >
2224 > /**
2225 > * The reason why sampling stopped, if known.
2226 > *
2227 > * Standard values:
2228 > * - `"endTurn"`: Natural end of the assistant's turn
2229 > * - `"stopSequence"`: A stop sequence was encountered
2230 > * - `"maxTokens"`: Maximum token limit was reached
2231 > * - `"toolUse"`: The model wants to use one or more tools
2232 > *
2233 > * This field is an open string to allow for provider-specific stop reasons.
2234 > */
2235 > stopReason?: "endTurn" | "stopSequence" | "maxTokens" | "toolUse" | string;
2236 > }
2237 >
2238 > /**
2239 > * A successful response from the client for a {@link CreateMessageRequest | sampling/createMessage} request.
2240 > *
2241 > * @example Sampling result response
2242 > * {@includeCode ./examples/CreateMessageResultResponse/sampling-result-response.json}
2243 > *
2244 > * @category `sampling/createMessage`
2245 > */
2246 > export interface CreateMessageResultResponse extends JSONRPCResultResponse {
2247 > result: CreateMessageResult;
2248 > }
2249 >
2250 > /**
2251 > * Describes a message issued to or received from an LLM API.
2252 > *
2253 > * @example Single content block
2254 > * {@includeCode ./examples/SamplingMessage/single-content-block.json}
2255 > *
2256 > * @example Multiple content blocks
2257 > * {@includeCode ./examples/SamplingMessage/multiple-content-blocks.json}
2258 > *
2259 > * @category `sampling/createMessage`
2260 > */
2261 > export interface SamplingMessage {
2262 > role: Role;
2263 > content: SamplingMessageContentBlock | SamplingMessageContentBlock[];
2264 > _meta?: MetaObject;
2265 > }
2266 >
2267 > /**
2268 > * @category `sampling/createMessage`
2269 > */
2270 > export type SamplingMessageContentBlock =
2271 > | TextContent
2272 > | ImageContent
2273 > | AudioContent
2274 > | ToolUseContent
2275 > | ToolResultContent;
2276 >
2277 > /**
2278 > * Optional annotations for the client. The client can use annotations to inform how objects are used or displayed
2279 > *
2280 > * @category Common Types
2281 > */
2282 > export interface Annotations {
2283 > /**
2284 > * Describes who the intended audience of this object or data is.
2285 > *
2286 > * It can include multiple entries to indicate content useful for multiple audiences (e.g., `["user", "assistant"]`).
2287 > */
2288 > audience?: Role[];
2289 >
2290 > /**
2291 > * Describes how important this data is for operating the server.
2292 > *
2293 > * A value of 1 means "most important," and indicates that the data is
2294 > * effectively required, while 0 means "least important," and indicates that
2295 > * the data is entirely optional.
2296 > *
2297 > * @TJS-type number
2298 > * @minimum 0
2299 > * @maximum 1
2300 > */
2301 > priority?: number;
2302 >
2303 > /**
2304 > * The moment the resource was last modified, as an ISO 8601 formatted string.
2305 > *
2306 > * Should be an ISO 8601 formatted string (e.g., "2025-01-12T15:00:58Z").
2307 > *
2308 > * Examples: last activity timestamp in an open file, timestamp when the resource
2309 > * was attached, etc.
2310 > */
2311 > lastModified?: string;
2312 > }
2313 >
2314 > /**
2315 > * @category Content
2316 > */
2317 > export type ContentBlock =
2318 > | TextContent
2319 > | ImageContent
2320 > | AudioContent
2321 > | ResourceLink
2322 > | EmbeddedResource;
2323 >
2324 > /**
2325 > * Text provided to or from an LLM.
2326 > *
2327 > * @example Text content
2328 > * {@includeCode ./examples/TextContent/text-content.json}
2329 > *
2330 > * @category Content
2331 > */
2332 > export interface TextContent {
2333 > type: "text";
2334 >
2335 > /**
2336 > * The text content of the message.
2337 > */
2338 > text: string;
2339 >
2340 > /**
2341 > * Optional annotations for the client.
2342 > */
2343 > annotations?: Annotations;
2344 >
2345 > _meta?: MetaObject;
2346 > }
2347 >
2348 > /**
2349 > * An image provided to or from an LLM.
2350 > *
2351 > * @example `image/png` content with annotations
2352 > * {@includeCode ./examples/ImageContent/image-png-content-with-annotations.json}
2353 > *
2354 > * @category Content
2355 > */
2356 > export interface ImageContent {
2357 > type: "image";
2358 >
2359 > /**
2360 > * The base64-encoded image data.
2361 > *
2362 > * @format byte
2363 > */
2364 > data: string;
2365 >
2366 > /**
2367 > * The MIME type of the image. Different providers may support different image types.
2368 > */
2369 > mimeType: string;
2370 >
2371 > /**
2372 > * Optional annotations for the client.
2373 > */
2374 > annotations?: Annotations;
2375 >
2376 > _meta?: MetaObject;
2377 > }
2378 >
2379 > /**
2380 > * Audio provided to or from an LLM.
2381 > *
2382 > * @example `audio/wav` content
2383 > * {@includeCode ./examples/AudioContent/audio-wav-content.json}
2384 > *
2385 > * @category Content
2386 > */
2387 > export interface AudioContent {
2388 > type: "audio";
2389 >
2390 > /**
2391 > * The base64-encoded audio data.
2392 > *
2393 > * @format byte
2394 > */
2395 > data: string;
2396 >
2397 > /**
2398 > * The MIME type of the audio. Different providers may support different audio types.
2399 > */
2400 > mimeType: string;
2401 >
2402 > /**
2403 > * Optional annotations for the client.
2404 > */
2405 > annotations?: Annotations;
2406 >
2407 > _meta?: MetaObject;
2408 > }
2409 >
2410 > /**
2411 > * A request from the assistant to call a tool.
2412 > *
2413 > * @example `get_weather` tool use
2414 > * {@includeCode ./examples/ToolUseContent/get-weather-tool-use.json}
2415 > *
2416 > * @category `sampling/createMessage`
2417 > */
2418 > export interface ToolUseContent {
2419 > type: "tool_use";
2420 >
2421 > /**
2422 > * A unique identifier for this tool use.
2423 > *
2424 > * This ID is used to match tool results to their corresponding tool uses.
2425 > */
2426 > id: string;
2427 >
2428 > /**
2429 > * The name of the tool to call.
2430 > */
2431 > name: string;
2432 >
2433 > /**
2434 > * The arguments to pass to the tool, conforming to the tool's input schema.
2435 > */
2436 > input: { [key: string]: unknown };
2437 >
2438 > /**
2439 > * Optional metadata about the tool use. Clients SHOULD preserve this field when
2440 > * including tool uses in subsequent sampling requests to enable caching optimizations.
2441 > */
2442 > _meta?: MetaObject;
2443 > }
2444 >
2445 > /**
2446 > * The result of a tool use, provided by the user back to the assistant.
2447 > *
2448 > * @example `get_weather` tool result
2449 > * {@includeCode ./examples/ToolResultContent/get-weather-tool-result.json}
2450 > *
2451 > * @category `sampling/createMessage`
2452 > */
2453 > export interface ToolResultContent {
2454 > type: "tool_result";
2455 >
2456 > /**
2457 > * The ID of the tool use this result corresponds to.
2458 > *
2459 > * This MUST match the ID from a previous {@link ToolUseContent}.
2460 > */
2461 > toolUseId: string;
2462 >
2463 > /**
2464 > * The unstructured result content of the tool use.
2465 > *
2466 > * This has the same format as {@link CallToolResult.content} and can include text, images,
2467 > * audio, resource links, and embedded resources.
2468 > */
2469 > content: ContentBlock[];
2470 >
2471 > /**
2472 > * An optional structured result object.
2473 > *
2474 > * If the tool defined an {@link Tool.outputSchema}, this SHOULD conform to that schema.
2475 > */
2476 > structuredContent?: { [key: string]: unknown };
2477 >
2478 > /**
2479 > * Whether the tool use resulted in an error.
2480 > *
2481 > * If true, the content typically describes the error that occurred.
2482 > * Default: false
2483 > */
2484 > isError?: boolean;
2485 >
2486 > /**
2487 > * Optional metadata about the tool result. Clients SHOULD preserve this field when
2488 > * including tool results in subsequent sampling requests to enable caching optimizations.
2489 > */
2490 > _meta?: MetaObject;
2491 > }
2492 >
2493 > /**
2494 > * The server's preferences for model selection, requested of the client during sampling.
2495 > *
2496 > * Because LLMs can vary along multiple dimensions, choosing the "best" model is
2497 > * rarely straightforward. Different models excel in different areas-some are
2498 > * faster but less capable, others are more capable but more expensive, and so
2499 > * on. This interface allows servers to express their priorities across multiple
2500 > * dimensions to help clients make an appropriate selection for their use case.
2501 > *
2502 > * These preferences are always advisory. The client MAY ignore them. It is also
2503 > * up to the client to decide how to interpret these preferences and how to
2504 > * balance them against other considerations.
2505 > *
2506 > * @example With hints and priorities
2507 > * {@includeCode ./examples/ModelPreferences/with-hints-and-priorities.json}
2508 > *
2509 > * @category `sampling/createMessage`
2510 > */
2511 > export interface ModelPreferences {
2512 > /**
2513 > * Optional hints to use for model selection.
2514 > *
2515 > * If multiple hints are specified, the client MUST evaluate them in order
2516 > * (such that the first match is taken).
2517 > *
2518 > * The client SHOULD prioritize these hints over the numeric priorities, but
2519 > * MAY still use the priorities to select from ambiguous matches.
2520 > */
2521 > hints?: ModelHint[];
2522 >
2523 > /**
2524 > * How much to prioritize cost when selecting a model. A value of 0 means cost
2525 > * is not important, while a value of 1 means cost is the most important
2526 > * factor.
2527 > *
2528 > * @TJS-type number
2529 > * @minimum 0
2530 > * @maximum 1
2531 > */
2532 > costPriority?: number;
2533 >
2534 > /**
2535 > * How much to prioritize sampling speed (latency) when selecting a model. A
2536 > * value of 0 means speed is not important, while a value of 1 means speed is
2537 > * the most important factor.
2538 > *
2539 > * @TJS-type number
2540 > * @minimum 0
2541 > * @maximum 1
2542 > */
2543 > speedPriority?: number;
2544 >
2545 > /**
2546 > * How much to prioritize intelligence and capabilities when selecting a
2547 > * model. A value of 0 means intelligence is not important, while a value of 1
2548 > * means intelligence is the most important factor.
2549 > *
2550 > * @TJS-type number
2551 > * @minimum 0
2552 > * @maximum 1
2553 > */
2554 > intelligencePriority?: number;
2555 > }
2556 >
2557 > /**
2558 > * Hints to use for model selection.
2559 > *
2560 > * Keys not declared here are currently left unspecified by the spec and are up
2561 > * to the client to interpret.
2562 > *
2563 > * @category `sampling/createMessage`
2564 > */
2565 > export interface ModelHint {
2566 > /**
2567 > * A hint for a model name.
2568 > *
2569 > * The client SHOULD treat this as a substring of a model name; for example:
2570 > * - `claude-3-5-sonnet` should match `claude-3-5-sonnet-20241022`
2571 > * - `sonnet` should match `claude-3-5-sonnet-20241022`, `claude-3-sonnet-20240229`, etc.
2572 > * - `claude` should match any Claude model
2573 > *
2574 > * The client MAY also map the string to a different provider's model name or a different model family, as long as it fills a similar niche; for example:
2575 > * - `gemini-1.5-flash` could match `claude-3-haiku-20240307`
2576 > */
2577 > name?: string;
2578 > }
2579 >
2580 > /* Autocomplete */
2581 > /**
2582 > * Parameters for a `completion/complete` request.
2583 > *
2584 > * @category `completion/complete`
2585 > *
2586 > * @example Prompt argument completion
2587 > * {@includeCode ./examples/CompleteRequestParams/prompt-argument-completion.json}
2588 > *
2589 > * @example Prompt argument completion with context
2590 > * {@includeCode ./examples/CompleteRequestParams/prompt-argument-completion-with-context.json}
2591 > */
2592 > export interface CompleteRequestParams extends RequestParams {
2593 > ref: PromptReference | ResourceTemplateReference;
2594 > /**
2595 > * The argument's information
2596 > */
2597 > argument: {
2598 > /**
2599 > * The name of the argument
2600 > */
2601 > name: string;
2602 > /**
2603 > * The value of the argument to use for completion matching.
2604 > */
2605 > value: string;
2606 > };
2607 >
2608 > /**
2609 > * Additional, optional context for completions
2610 > */
2611 > context?: {
2612 > /**
2613 > * Previously-resolved variables in a URI template or prompt.
2614 > */
2615 > arguments?: { [key: string]: string };
2616 > };
2617 > }
2618 >
2619 > /**
2620 > * A request from the client to the server, to ask for completion options.
2621 > *
2622 > * @example Completion request
2623 > * {@includeCode ./examples/CompleteRequest/completion-request.json}
2624 > *
2625 > * @category `completion/complete`
2626 > */
2627 > export interface CompleteRequest extends JSONRPCRequest {
2628 > method: "completion/complete";
2629 > params: CompleteRequestParams;
2630 > }
2631 >
2632 > /**
2633 > * The result returned by the server for a {@link CompleteRequest | completion/complete} request.
2634 > *
2635 > * @category `completion/complete`
2636 > *
2637 > * @example Single completion value
2638 > * {@includeCode ./examples/CompleteResult/single-completion-value.json}
2639 > *
2640 > * @example Multiple completion values with more available
2641 > * {@includeCode ./examples/CompleteResult/multiple-completion-values-with-more-available.json}
2642 > */
2643 > export interface CompleteResult extends Result {
2644 > completion: {
2645 > /**
2646 > * An array of completion values. Must not exceed 100 items.
2647 > */
2648 > values: string[];
2649 > /**
2650 > * The total number of completion options available. This can exceed the number of values actually sent in the response.
2651 > */
2652 > total?: number;
2653 > /**
2654 > * Indicates whether there are additional completion options beyond those provided in the current response, even if the exact total is unknown.
2655 > */
2656 > hasMore?: boolean;
2657 > };
2658 > }
2659 >
2660 > /**
2661 > * A successful response from the server for a {@link CompleteRequest | completion/complete} request.
2662 > *
2663 > * @example Completion result response
2664 > * {@includeCode ./examples/CompleteResultResponse/completion-result-response.json}
2665 > *
2666 > * @category `completion/complete`
2667 > */
2668 > export interface CompleteResultResponse extends JSONRPCResultResponse {
2669 > result: CompleteResult;
2670 > }
2671 >
2672 > /**
2673 > * A reference to a resource or resource template definition.
2674 > *
2675 > * @category `completion/complete`
2676 > */
2677 > export interface ResourceTemplateReference {
2678 > type: "ref/resource";
2679 > /**
2680 > * The URI or URI template of the resource.
2681 > *
2682 > * @format uri-template
2683 > */
2684 > uri: string;
2685 > }
2686 >
2687 > /**
2688 > * Identifies a prompt.
2689 > *
2690 > * @category `completion/complete`
2691 > */
2692 > export interface PromptReference extends BaseMetadata {
2693 > type: "ref/prompt";
2694 > }
2695 >
2696 > /* Roots */
2697 > /**
2698 > * Sent from the server to request a list of root URIs from the client. Roots allow
2699 > * servers to ask for specific directories or files to operate on. A common example
2700 > * for roots is providing a set of repositories or directories a server should operate
2701 > * on.
2702 > *
2703 > * This request is typically used when the server needs to understand the file system
2704 > * structure or access specific locations that the client has permission to read from.
2705 > *
2706 > * @example List roots request
2707 > * {@includeCode ./examples/ListRootsRequest/list-roots-request.json}
2708 > *
2709 > * @category `roots/list`
2710 > */
2711 > export interface ListRootsRequest extends JSONRPCRequest {
2712 > method: "roots/list";
2713 > params?: RequestParams;
2714 > }
2715 >
2716 > /**
2717 > * The result returned by the client for a {@link ListRootsRequest | roots/list} request.
2718 > * This result contains an array of {@link Root} objects, each representing a root directory
2719 > * or file that the server can operate on.
2720 > *
2721 > * @example Single root directory
2722 > * {@includeCode ./examples/ListRootsResult/single-root-directory.json}
2723 > *
2724 > * @example Multiple root directories
2725 > * {@includeCode ./examples/ListRootsResult/multiple-root-directories.json}
2726 > *
2727 > * @category `roots/list`
2728 > */
2729 > export interface ListRootsResult extends Result {
2730 > roots: Root[];
2731 > }
2732 >
2733 > /**
2734 > * A successful response from the client for a {@link ListRootsRequest | roots/list} request.
2735 > *
2736 > * @example List roots result response
2737 > * {@includeCode ./examples/ListRootsResultResponse/list-roots-result-response.json}
2738 > *
2739 > * @category `roots/list`
2740 > */
2741 > export interface ListRootsResultResponse extends JSONRPCResultResponse {
2742 > result: ListRootsResult;
2743 > }
2744 >
2745 > /**
2746 > * Represents a root directory or file that the server can operate on.
2747 > *
2748 > * @example Project directory root
2749 > * {@includeCode ./examples/Root/project-directory.json}
2750 > *
2751 > * @category `roots/list`
2752 > */
2753 > export interface Root {
2754 > /**
2755 > * The URI identifying the root. This *must* start with `file://` for now.
2756 > * This restriction may be relaxed in future versions of the protocol to allow
2757 > * other URI schemes.
2758 > *
2759 > * @format uri
2760 > */
2761 > uri: string;
2762 > /**
2763 > * An optional name for the root. This can be used to provide a human-readable
2764 > * identifier for the root, which may be useful for display purposes or for
2765 > * referencing the root in other parts of the application.
2766 > */
2767 > name?: string;
2768 >
2769 > _meta?: MetaObject;
2770 > }
2771 >
2772 > /**
2773 > * A notification from the client to the server, informing it that the list of roots has changed.
2774 > * This notification should be sent whenever the client adds, removes, or modifies any root.
2775 > * The server should then request an updated list of roots using the {@link ListRootsRequest}.
2776 > *
2777 > * @example Roots list changed
2778 > * {@includeCode ./examples/RootsListChangedNotification/roots-list-changed.json}
2779 > *
2780 > * @category `notifications/roots/list_changed`
2781 > */
2782 > export interface RootsListChangedNotification extends JSONRPCNotification {
2783 > method: "notifications/roots/list_changed";
2784 > params?: NotificationParams;
2785 > }
2786 >
2787 > /**
2788 > * The parameters for a request to elicit non-sensitive information from the user via a form in the client.
2789 > *
2790 > * @example Elicit single field
2791 > * {@includeCode ./examples/ElicitRequestFormParams/elicit-single-field.json}
2792 > *
2793 > * @example Elicit multiple fields
2794 > * {@includeCode ./examples/ElicitRequestFormParams/elicit-multiple-fields.json}
2795 > *
2796 > * @category `elicitation/create`
2797 > */
2798 > export interface ElicitRequestFormParams extends TaskAugmentedRequestParams {
2799 > /**
2800 > * The elicitation mode.
2801 > */
2802 > mode?: "form";
2803 >
2804 > /**
2805 > * The message to present to the user describing what information is being requested.
2806 > */
2807 > message: string;
2808 >
2809 > /**
2810 > * A restricted subset of JSON Schema.
2811 > * Only top-level properties are allowed, without nesting.
2812 > */
2813 > requestedSchema: {
2814 > $schema?: string;
2815 > type: "object";
2816 > properties: {
2817 > [key: string]: PrimitiveSchemaDefinition;
2818 > };
2819 > required?: string[];
2820 > };
2821 > }
2822 >
2823 > /**
2824 > * The parameters for a request to elicit information from the user via a URL in the client.
2825 > *
2826 > * @example Elicit sensitive data
2827 > * {@includeCode ./examples/ElicitRequestURLParams/elicit-sensitive-data.json}
2828 > *
2829 > * @category `elicitation/create`
2830 > */
2831 > export interface ElicitRequestURLParams extends TaskAugmentedRequestParams {
2832 > /**
2833 > * The elicitation mode.
2834 > */
2835 > mode: "url";
2836 >
2837 > /**
2838 > * The message to present to the user explaining why the interaction is needed.
2839 > */
2840 > message: string;
2841 >
2842 > /**
2843 > * The ID of the elicitation, which must be unique within the context of the server.
2844 > * The client MUST treat this ID as an opaque value.
2845 > */
2846 > elicitationId: string;
2847 >
2848 > /**
2849 > * The URL that the user should navigate to.
2850 > *
2851 > * @format uri
2852 > */
2853 > url: string;
2854 > }
2855 >
2856 > /**
2857 > * The parameters for a request to elicit additional information from the user via the client.
2858 > *
2859 > * @category `elicitation/create`
2860 > */
2861 > export type ElicitRequestParams =
2862 > | ElicitRequestFormParams
2863 > | ElicitRequestURLParams;
2864 >
2865 > /**
2866 > * A request from the server to elicit additional information from the user via the client.
2867 > *
2868 > * @example Elicitation request
2869 > * {@includeCode ./examples/ElicitRequest/elicitation-request.json}
2870 > *
2871 > * @category `elicitation/create`
2872 > */
2873 > export interface ElicitRequest extends JSONRPCRequest {
2874 > method: "elicitation/create";
2875 > params: ElicitRequestParams;
2876 > }
2877 >
2878 > /**
2879 > * Restricted schema definitions that only allow primitive types
2880 > * without nested objects or arrays.
2881 > *
2882 > * @category `elicitation/create`
2883 > */
2884 > export type PrimitiveSchemaDefinition =
2885 > | StringSchema
2886 > | NumberSchema
2887 > | BooleanSchema
2888 > | EnumSchema;
2889 >
2890 > /**
2891 > * @example Email input schema
2892 > * {@includeCode ./examples/StringSchema/email-input-schema.json}
2893 > *
2894 > * @category `elicitation/create`
2895 > */
2896 > export interface StringSchema {
2897 > type: "string";
2898 > title?: string;
2899 > description?: string;
2900 > minLength?: number;
2901 > maxLength?: number;
2902 > format?: "email" | "uri" | "date" | "date-time";
2903 > default?: string;
2904 > }
2905 >
2906 > /**
2907 > * @example Number input schema
2908 > * {@includeCode ./examples/NumberSchema/number-input-schema.json}
2909 > *
2910 > * @category `elicitation/create`
2911 > */
2912 > export interface NumberSchema {
2913 > type: "number" | "integer";
2914 > title?: string;
2915 > description?: string;
2916 > minimum?: number;
2917 > maximum?: number;
2918 > default?: number;
2919 > }
2920 >
2921 > /**
2922 > * @example Boolean input schema
2923 > * {@includeCode ./examples/BooleanSchema/boolean-input-schema.json}
2924 > *
2925 > * @category `elicitation/create`
2926 > */
2927 > export interface BooleanSchema {
2928 > type: "boolean";
2929 > title?: string;
2930 > description?: string;
2931 > default?: boolean;
2932 > }
2933 >
2934 > /**
2935 > * Schema for single-selection enumeration without display titles for options.
2936 > *
2937 > * @example Color select schema
2938 > * {@includeCode ./examples/UntitledSingleSelectEnumSchema/color-select-schema.json}
2939 > *
2940 > * @category `elicitation/create`
2941 > */
2942 > export interface UntitledSingleSelectEnumSchema {
2943 > type: "string";
2944 > /**
2945 > * Optional title for the enum field.
2946 > */
2947 > title?: string;
2948 > /**
2949 > * Optional description for the enum field.
2950 > */
2951 > description?: string;
2952 > /**
2953 > * Array of enum values to choose from.
2954 > */
2955 > enum: string[];
2956 > /**
2957 > * Optional default value.
2958 > */
2959 > default?: string;
2960 > }
2961 >
2962 > /**
2963 > * Schema for single-selection enumeration with display titles for each option.
2964 > *
2965 > * @example Titled color select schema
2966 > * {@includeCode ./examples/TitledSingleSelectEnumSchema/titled-color-select-schema.json}
2967 > *
2968 > * @category `elicitation/create`
2969 > */
2970 > export interface TitledSingleSelectEnumSchema {
2971 > type: "string";
2972 > /**
2973 > * Optional title for the enum field.
2974 > */
2975 > title?: string;
2976 > /**
2977 > * Optional description for the enum field.
2978 > */
2979 > description?: string;
2980 > /**
2981 > * Array of enum options with values and display labels.
2982 > */
2983 > oneOf: Array<{
2984 > /**
2985 > * The enum value.
2986 > */
2987 > const: string;
2988 > /**
2989 > * Display label for this option.
2990 > */
2991 > title: string;
2992 > }>;
2993 > /**
2994 > * Optional default value.
2995 > */
2996 > default?: string;
2997 > }
2998 >
2999 > /**
3000 > * @category `elicitation/create`
3001 > */
3002 > // Combined single selection enumeration
3003 > export type SingleSelectEnumSchema =
3004 > | UntitledSingleSelectEnumSchema
3005 > | TitledSingleSelectEnumSchema;
3006 >
3007 > /**
3008 > * Schema for multiple-selection enumeration without display titles for options.
3009 > *
3010 > * @example Color multi-select schema
3011 > * {@includeCode ./examples/UntitledMultiSelectEnumSchema/color-multi-select-schema.json}
3012 > *
3013 > * @category `elicitation/create`
3014 > */
3015 > export interface UntitledMultiSelectEnumSchema {
3016 > type: "array";
3017 > /**
3018 > * Optional title for the enum field.
3019 > */
3020 > title?: string;
3021 > /**
3022 > * Optional description for the enum field.
3023 > */
3024 > description?: string;
3025 > /**
3026 > * Minimum number of items to select.
3027 > */
3028 > minItems?: number;
3029 > /**
3030 > * Maximum number of items to select.
3031 > */
3032 > maxItems?: number;
3033 > /**
3034 > * Schema for the array items.
3035 > */
3036 > items: {
3037 > type: "string";
3038 > /**
3039 > * Array of enum values to choose from.
3040 > */
3041 > enum: string[];
3042 > };
3043 > /**
3044 > * Optional default value.
3045 > */
3046 > default?: string[];
3047 > }
3048 >
3049 > /**
3050 > * Schema for multiple-selection enumeration with display titles for each option.
3051 > *
3052 > * @example Titled color multi-select schema
3053 > * {@includeCode ./examples/TitledMultiSelectEnumSchema/titled-color-multi-select-schema.json}
3054 > *
3055 > * @category `elicitation/create`
3056 > */
3057 > export interface TitledMultiSelectEnumSchema {
3058 > type: "array";
3059 > /**
3060 > * Optional title for the enum field.
3061 > */
3062 > title?: string;
3063 > /**
3064 > * Optional description for the enum field.
3065 > */
3066 > description?: string;
3067 > /**
3068 > * Minimum number of items to select.
3069 > */
3070 > minItems?: number;
3071 > /**
3072 > * Maximum number of items to select.
3073 > */
3074 > maxItems?: number;
3075 > /**
3076 > * Schema for array items with enum options and display labels.
3077 > */
3078 > items: {
3079 > /**
3080 > * Array of enum options with values and display labels.
3081 > */
3082 > anyOf: Array<{
3083 > /**
3084 > * The constant enum value.
3085 > */
3086 > const: string;
3087 > /**
3088 > * Display title for this option.
3089 > */
3090 > title: string;
3091 > }>;
3092 > };
3093 > /**
3094 > * Optional default value.
3095 > */
3096 > default?: string[];
3097 > }
3098 >
3099 > /**
3100 > * @category `elicitation/create`
3101 > */
3102 > // Combined multiple selection enumeration
3103 > export type MultiSelectEnumSchema =
3104 > | UntitledMultiSelectEnumSchema
3105 > | TitledMultiSelectEnumSchema;
3106 >
3107 > /**
3108 > * Use {@link TitledSingleSelectEnumSchema} instead.
3109 > * This interface will be removed in a future version.
3110 > *
3111 > * @category `elicitation/create`
3112 > */
3113 > export interface LegacyTitledEnumSchema {
3114 > type: "string";
3115 > title?: string;
3116 > description?: string;
3117 > enum: string[];
3118 > /**
3119 > * (Legacy) Display names for enum values.
3120 > * Non-standard according to JSON schema 2020-12.
3121 > */
3122 > enumNames?: string[];
3123 > default?: string;
3124 > }
3125 >
3126 > /**
3127 > * @category `elicitation/create`
3128 > */
3129 > // Union type for all enum schemas
3130 > export type EnumSchema =
3131 > | SingleSelectEnumSchema
3132 > | MultiSelectEnumSchema
3133 > | LegacyTitledEnumSchema;
3134 >
3135 > /**
3136 > * The result returned by the client for an {@link ElicitRequest | elicitation/create} request.
3137 > *
3138 > * @example Input single field
3139 > * {@includeCode ./examples/ElicitResult/input-single-field.json}
3140 > *
3141 > * @example Input multiple fields
3142 > * {@includeCode ./examples/ElicitResult/input-multiple-fields.json}
3143 > *
3144 > * @example Accept URL mode (no content)
3145 > * {@includeCode ./examples/ElicitResult/accept-url-mode-no-content.json}
3146 > *
3147 > * @category `elicitation/create`
3148 > */
3149 > export interface ElicitResult extends Result {
3150 > /**
3151 > * The user action in response to the elicitation.
3152 > * - `"accept"`: User submitted the form/confirmed the action
3153 > * - `"decline"`: User explicitly declined the action
3154 > * - `"cancel"`: User dismissed without making an explicit choice
3155 > */
3156 > action: "accept" | "decline" | "cancel";
3157 >
3158 > /**
3159 > * The submitted form data, only present when action is `"accept"` and mode was `"form"`.
3160 > * Contains values matching the requested schema.
3161 > * Omitted for out-of-band mode responses.
3162 > */
3163 > content?: { [key: string]: string | number | boolean | string[] };
3164 > }
3165 >
3166 > /**
3167 > * A successful response from the client for a {@link ElicitRequest | elicitation/create} request.
3168 > *
3169 > * @example Elicitation result response
3170 > * {@includeCode ./examples/ElicitResultResponse/elicitation-result-response.json}
3171 > *
3172 > * @category `elicitation/create`
3173 > */
3174 > export interface ElicitResultResponse extends JSONRPCResultResponse {
3175 > result: ElicitResult;
3176 > }
3177 >
3178 > /**
3179 > * An optional notification from the server to the client, informing it of a completion of a out-of-band elicitation request.
3180 > *
3181 > * @example Elicitation complete
3182 > * {@includeCode ./examples/ElicitationCompleteNotification/elicitation-complete.json}
3183 > *
3184 > * @category `notifications/elicitation/complete`
3185 > */
3186 > export interface ElicitationCompleteNotification extends JSONRPCNotification {
3187 > method: "notifications/elicitation/complete";
3188 > params: {
3189 > /**
3190 > * The ID of the elicitation that completed.
3191 > */
3192 > elicitationId: string;
3193 > };
3194 > }
3195 >
3196 > /* Client messages */
3197 > /** @internal */
3198 > export type ClientRequest =
3199 > | PingRequest
3200 > | InitializeRequest
3201 > | CompleteRequest
3202 > | SetLevelRequest
3203 > | GetPromptRequest
3204 > | ListPromptsRequest
3205 > | ListResourcesRequest
3206 > | ListResourceTemplatesRequest
3207 > | ReadResourceRequest
3208 > | SubscribeRequest
3209 > | UnsubscribeRequest
3210 > | CallToolRequest
3211 > | ListToolsRequest
3212 > | GetTaskRequest
3213 > | GetTaskPayloadRequest
3214 > | ListTasksRequest
3215 > | CancelTaskRequest;
3216 >
3217 > /** @internal */
3218 > export type ClientNotification =
3219 > | CancelledNotification
3220 > | ProgressNotification
3221 > | InitializedNotification
3222 > | RootsListChangedNotification
3223 > | TaskStatusNotification;
3224 >
3225 > /** @internal */
3226 > export type ClientResult =
3227 > | EmptyResult
3228 > | CreateMessageResult
3229 > | ListRootsResult
3230 > | ElicitResult
3231 > | GetTaskResult
3232 > | GetTaskPayloadResult
3233 > | ListTasksResult
3234 > | CancelTaskResult;
3235 >
3236 > /* Server messages */
3237 > /** @internal */
3238 > export type ServerRequest =
3239 > | PingRequest
3240 > | CreateMessageRequest
3241 > | ListRootsRequest
3242 > | ElicitRequest
3243 > | GetTaskRequest
3244 > | GetTaskPayloadRequest
3245 > | ListTasksRequest
3246 > | CancelTaskRequest;
3247 >
3248 > /** @internal */
3249 > export type ServerNotification =
3250 > | CancelledNotification
3251 > | ProgressNotification
3252 > | LoggingMessageNotification
3253 > | ResourceUpdatedNotification
3254 > | ResourceListChangedNotification
3255 > | ToolListChangedNotification
3256 > | PromptListChangedNotification
3257 > | ElicitationCompleteNotification
3258 > | TaskStatusNotification;
3259 >
3260 > /** @internal */
3261 > export type ServerResult =
3262 > | EmptyResult
3263 > | InitializeResult
3264 > | CompleteResult
3265 > | GetPromptResult
3266 > | ListPromptsResult
3267 > | ListResourceTemplatesResult
3268 > | ListResourcesResult
3269 > | ReadResourceResult
3270 > | CallToolResult
3271 > | CreateTaskResult
3272 > | ListToolsResult
3273 > | GetTaskResult
3274 > | GetTaskPayloadResult
3275 > | ListTasksResult
3276 > | CancelTaskResult;
3277 > }
src/vs/editor/common/languages.ts 2475 covered LOC · 27 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- languages.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer } from '../../base/common/buffer.js';
7 > import { CancellationToken } from '../../base/common/cancellation.js';
8 > import { Codicon } from '../../base/common/codicons.js';
9 > import { Color } from '../../base/common/color.js';
10 > import { IReadonlyVSDataTransfer } from '../../base/common/dataTransfer.js';
11 > import { Event } from '../../base/common/event.js';
12 > import { HierarchicalKind } from '../../base/common/hierarchicalKind.js';
13 > import { IMarkdownString } from '../../base/common/htmlContent.js';
14 > import { IDisposable } from '../../base/common/lifecycle.js';
15 > import { ThemeIcon } from '../../base/common/themables.js';
16 > import { URI, UriComponents } from '../../base/common/uri.js';
17 > import { EditOperation, ISingleEditOperation } from './core/editOperation.js';
18 > import { IPosition, Position } from './core/position.js';
19 > import { IRange, Range } from './core/range.js';
20 > import { Selection } from './core/selection.js';
21 > import { LanguageId } from './encodedTokenAttributes.js';
22 > import { LanguageSelector } from './languageSelector.js';
23 > import * as model from './model.js';
24 > import { TokenizationRegistry as TokenizationRegistryImpl } from './tokenizationRegistry.js';
25 > import { ContiguousMultilineTokens } from './tokens/contiguousMultilineTokens.js';
26 > import { localize } from '../../nls.js';
27 > import { ExtensionIdentifier } from '../../platform/extensions/common/extensions.js';
28 > import { IMarkerData } from '../../platform/markers/common/markers.js';
29 > import { EditDeltaInfo } from './textModelEditSource.js';
30 > import { FontTokensUpdate } from './textModelEvents.js';
31 >
32 > /**
33 > * @internal
34 > */
35 > export interface ILanguageIdCodec {
36 > encodeLanguageId(languageId: string): LanguageId;
37 > decodeLanguageId(languageId: LanguageId): string;
38 > }
39 >
40 > export class Token {
41 > _tokenBrand: void = undefined;
42 >
43 > constructor(
44 public readonly offset: number,
45 public readonly type: string,
47 ) {
48 }
50 > public toString(): string {
51 return '(' + this.offset + ', ' + this.type + ')';
52 }
53 > } languages.ts
54 >
55 > /**
56 > * @internal
57 > */
58 > export class TokenizationResult {
59 > _tokenizationResultBrand: void = undefined;
60 >
61 > constructor(
62 public readonly tokens: Token[],
63 public readonly endState: IState,
64 ) {
65 }
66 > } languages.ts
67 >
68 > /**
69 > * @internal
70 > */
71 > export interface IFontToken {
72 > readonly startIndex: number;
73 > readonly endIndex: number;
74 > readonly fontFamily: string | null;
75 > readonly fontSizeMultiplier: number | null;
76 > readonly lineHeightMultiplier: number | null;
77 > }
78 >
79 > /**
80 > * @internal
81 > */
82 > export class EncodedTokenizationResult {
83 > _encodedTokenizationResultBrand: void = undefined;
84 >
85 > constructor(
86 /**
87 * The tokens in binary format. Each token occupies two array indices. For token i:
95 ) {
96 }
97 > } languages.ts
98 >
99 > export interface SyntaxNode {
100 > startIndex: number;
101 > endIndex: number;
102 > startPosition: IPosition;
103 > endPosition: IPosition;
104 > }
105 >
106 > export interface QueryCapture {
107 > name: string;
108 > text?: string;
109 > node: SyntaxNode;
110 > encodedLanguageId: number;
111 > }
112 >
113 > /**
114 > * @internal
115 > */
116 > export interface ITokenizationSupport {
117 > /**
118 > * If true, the background tokenizer will only be used to verify tokens against the default background tokenizer.
119 > * Used for debugging.
120 > */
121 > readonly backgroundTokenizerShouldOnlyVerifyTokens?: boolean;
122 >
123 > getInitialState(): IState;
124 >
125 > tokenize(line: string, hasEOL: boolean, state: IState): TokenizationResult;
126 >
127 > tokenizeEncoded(line: string, hasEOL: boolean, state: IState): EncodedTokenizationResult;
128 >
129 > /**
130 > * Can be/return undefined if default background tokenization should be used.
131 > */
132 > createBackgroundTokenizer?(textModel: model.ITextModel, store: IBackgroundTokenizationStore): IBackgroundTokenizer | undefined;
133 > }
134 >
135 > /**
136 > * @internal
137 > */
138 > export interface IBackgroundTokenizer extends IDisposable {
139 > /**
140 > * Instructs the background tokenizer to set the tokens for the given range again.
141 > *
142 > * This might be necessary if the renderer overwrote those tokens with heuristically computed ones for some viewport,
143 > * when the change does not even propagate to that viewport.
144 > */
145 > requestTokens(startLineNumber: number, endLineNumberExclusive: number): void;
146 >
147 > reportMismatchingTokens?(lineNumber: number): void;
148 > }
149 >
150 > /**
151 > * @internal
152 > */
153 > export interface IBackgroundTokenizationStore {
154 > setTokens(tokens: ContiguousMultilineTokens[]): void;
155 >
156 > setFontInfo(changes: FontTokensUpdate): void;
157 >
158 > setEndState(lineNumber: number, state: IState): void;
159 >
160 > /**
161 > * Should be called to indicate that the background tokenization has finished for now.
162 > * (This triggers bracket pair colorization to re-parse the bracket pairs with token information)
163 > */
164 > backgroundTokenizationFinished(): void;
165 > }
166 >
167 > /**
168 > * The state of the tokenizer between two lines.
169 > * It is useful to store flags such as in multiline comment, etc.
170 > * The model will clone the previous line's state and pass it in to tokenize the next line.
171 > */
172 > export interface IState {
173 > clone(): IState;
174 > equals(other: IState): boolean;
175 > }
176 >
177 > /**
178 > * A provider result represents the values a provider, like the {@link HoverProvider},
179 > * may return. For once this is the actual result type `T`, like `Hover`, or a thenable that resolves
180 > * to that type `T`. In addition, `null` and `undefined` can be returned - either directly or from a
181 > * thenable.
182 > */
183 > export type ProviderResult<T> = T | undefined | null | Thenable<T | undefined | null>;
184 >
185 > /**
186 > * A hover represents additional information for a symbol or word. Hovers are
187 > * rendered in a tooltip-like widget.
188 > */
189 > export interface Hover {
190 > /**
191 > * The contents of this hover.
192 > */
193 > contents: IMarkdownString[];
194 >
195 > /**
196 > * The range to which this hover applies. When missing, the
197 > * editor will use the range at the current position or the
198 > * current position itself.
199 > */
200 > range?: IRange;
201 >
202 > /**
203 > * Can increase the verbosity of the hover
204 > */
205 > canIncreaseVerbosity?: boolean;
206 >
207 > /**
208 > * Can decrease the verbosity of the hover
209 > */
210 > canDecreaseVerbosity?: boolean;
211 > }
212 >
213 > /**
214 > * The hover provider interface defines the contract between extensions and
215 > * the [hover](https://code.visualstudio.com/docs/editor/intellisense)-feature.
216 > */
217 > export interface HoverProvider<THover = Hover> {
218 > /**
219 > * Provide a hover for the given position, context and document. Multiple hovers at the same
220 > * position will be merged by the editor. A hover can have a range which defaults
221 > * to the word range at the position when omitted.
222 > */
223 > provideHover(model: model.ITextModel, position: Position, token: CancellationToken, context?: HoverContext<THover>): ProviderResult<THover>;
224 > }
225 >
226 > export interface HoverContext<THover = Hover> {
227 > /**
228 > * Hover verbosity request
229 > */
230 > verbosityRequest?: HoverVerbosityRequest<THover>;
231 > }
232 >
233 > export interface HoverVerbosityRequest<THover = Hover> {
234 > /**
235 > * The delta by which to increase/decrease the hover verbosity level
236 > */
237 > verbosityDelta: number;
238 > /**
239 > * The previous hover for the same position
240 > */
241 > previousHover: THover;
242 > }
243 >
244 > export enum HoverVerbosityAction {
245 > /**
246 > * Increase the verbosity of the hover
247 > */
248 > Increase,
249 > /**
250 > * Decrease the verbosity of the hover
251 > */
252 > Decrease
253 > }
254 >
255 > /**
256 > * An evaluatable expression represents additional information for an expression in a document. Evaluatable expressions are
257 > * evaluated by a debugger or runtime and their result is rendered in a tooltip-like widget.
258 > * @internal
259 > */
260 > export interface EvaluatableExpression {
261 > /**
262 > * The range to which this expression applies.
263 > */
264 > range: IRange;
265 > /**
266 > * This expression overrides the expression extracted from the range.
267 > */
268 > expression?: string;
269 > }
270 >
271 >
272 > /**
273 > * The evaluatable expression provider interface defines the contract between extensions and
274 > * the debug hover.
275 > * @internal
276 > */
277 > export interface EvaluatableExpressionProvider {
278 > /**
279 > * Provide a hover for the given position and document. Multiple hovers at the same
280 > * position will be merged by the editor. A hover can have a range which defaults
281 > * to the word range at the position when omitted.
282 > */
283 > provideEvaluatableExpression(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<EvaluatableExpression>;
284 > }
285 >
286 > /**
287 > * A value-object that contains contextual information when requesting inline values from a InlineValuesProvider.
288 > * @internal
289 > */
290 > export interface InlineValueContext {
291 > frameId: number;
292 > stoppedLocation: Range;
293 > }
294 >
295 > /**
296 > * Provide inline value as text.
297 > * @internal
298 > */
299 > export interface InlineValueText {
300 > type: 'text';
301 > range: IRange;
302 > text: string;
303 > }
304 >
305 > /**
306 > * Provide inline value through a variable lookup.
307 > * @internal
308 > */
309 > export interface InlineValueVariableLookup {
310 > type: 'variable';
311 > range: IRange;
312 > variableName?: string;
313 > caseSensitiveLookup: boolean;
314 > }
315 >
316 > /**
317 > * Provide inline value through an expression evaluation.
318 > * @internal
319 > */
320 > export interface InlineValueExpression {
321 > type: 'expression';
322 > range: IRange;
323 > expression?: string;
324 > }
325 >
326 > /**
327 > * Inline value information can be provided by different means:
328 > * - directly as a text value (class InlineValueText).
329 > * - as a name to use for a variable lookup (class InlineValueVariableLookup)
330 > * - as an evaluatable expression (class InlineValueEvaluatableExpression)
331 > * The InlineValue types combines all inline value types into one type.
332 > * @internal
333 > */
334 > export type InlineValue = InlineValueText | InlineValueVariableLookup | InlineValueExpression;
335 >
336 > /**
337 > * The inline values provider interface defines the contract between extensions and
338 > * the debugger's inline values feature.
339 > * @internal
340 > */
341 > export interface InlineValuesProvider {
342 > /**
343 > */
344 > onDidChangeInlineValues?: Event<void> | undefined;
345 > /**
346 > * Provide the "inline values" for the given range and document. Multiple hovers at the same
347 > * position will be merged by the editor. A hover can have a range which defaults
348 > * to the word range at the position when omitted.
349 > */
350 > provideInlineValues(model: model.ITextModel, viewPort: Range, context: InlineValueContext, token: CancellationToken): ProviderResult<InlineValue[]>;
351 > }
352 >
353 > export const enum CompletionItemKind {
354 > Method,
355 > Function,
356 > Constructor,
357 > Field,
358 > Variable,
359 > Class,
360 > Struct,
361 > Interface,
362 > Module,
363 > Property,
364 > Event,
365 > Operator,
366 > Unit,
367 > Value,
368 > Constant,
369 > Enum,
370 > EnumMember,
371 > Keyword,
372 > Text,
373 > Color,
374 > File,
375 > Reference,
376 > Customcolor,
377 > Folder,
378 > TypeParameter,
379 > User,
380 > Issue,
381 > Tool,
382 > Snippet, // <- highest value (used for compare!)
383 > }
384 >
385 > /**
386 > * @internal
387 > */
388 > export namespace CompletionItemKinds {
389 >
390 > const byKind = new Map<CompletionItemKind, ThemeIcon>();
391 > byKind.set(CompletionItemKind.Method, Codicon.symbolMethod);
392 > byKind.set(CompletionItemKind.Function, Codicon.symbolFunction);
393 > byKind.set(CompletionItemKind.Constructor, Codicon.symbolConstructor);
394 > byKind.set(CompletionItemKind.Field, Codicon.symbolField);
395 > byKind.set(CompletionItemKind.Variable, Codicon.symbolVariable);
396 > byKind.set(CompletionItemKind.Class, Codicon.symbolClass);
397 > byKind.set(CompletionItemKind.Struct, Codicon.symbolStruct);
398 > byKind.set(CompletionItemKind.Interface, Codicon.symbolInterface);
399 > byKind.set(CompletionItemKind.Module, Codicon.symbolModule);
400 > byKind.set(CompletionItemKind.Property, Codicon.symbolProperty);
401 > byKind.set(CompletionItemKind.Event, Codicon.symbolEvent);
402 > byKind.set(CompletionItemKind.Operator, Codicon.symbolOperator);
403 > byKind.set(CompletionItemKind.Unit, Codicon.symbolUnit);
404 > byKind.set(CompletionItemKind.Value, Codicon.symbolValue);
405 > byKind.set(CompletionItemKind.Enum, Codicon.symbolEnum);
406 > byKind.set(CompletionItemKind.Constant, Codicon.symbolConstant);
407 > byKind.set(CompletionItemKind.EnumMember, Codicon.symbolEnumMember);
408 > byKind.set(CompletionItemKind.Keyword, Codicon.symbolKeyword);
409 > byKind.set(CompletionItemKind.Snippet, Codicon.symbolSnippet);
410 > byKind.set(CompletionItemKind.Text, Codicon.symbolText);
411 > byKind.set(CompletionItemKind.Color, Codicon.symbolColor);
412 > byKind.set(CompletionItemKind.File, Codicon.symbolFile);
413 > byKind.set(CompletionItemKind.Reference, Codicon.symbolReference);
414 > byKind.set(CompletionItemKind.Customcolor, Codicon.symbolCustomColor);
415 > byKind.set(CompletionItemKind.Folder, Codicon.symbolFolder);
416 > byKind.set(CompletionItemKind.TypeParameter, Codicon.symbolTypeParameter);
417 > byKind.set(CompletionItemKind.User, Codicon.account);
418 > byKind.set(CompletionItemKind.Issue, Codicon.issues);
419 > byKind.set(CompletionItemKind.Tool, Codicon.tools);
420 >
421 > /**
422 > * @internal
423 > */
424 > export function toIcon(kind: CompletionItemKind): ThemeIcon {
425 let codicon = byKind.get(kind);
426 if (!codicon) {
430 return codicon;
431 }
432 > languages.ts
433 > /**
434 > * @internal
435 > */
436 > export function toLabel(kind: CompletionItemKind): string {
437 switch (kind) {
438 case CompletionItemKind.Method: return localize('suggestWidget.kind.method', 'Method');
468 }
469 }
470 > languages.ts
471 > const data = new Map<string, CompletionItemKind>();
472 > data.set('method', CompletionItemKind.Method);
473 > data.set('function', CompletionItemKind.Function);
474 > data.set('constructor', CompletionItemKind.Constructor);
475 > data.set('field', CompletionItemKind.Field);
476 > data.set('variable', CompletionItemKind.Variable);
477 > data.set('class', CompletionItemKind.Class);
478 > data.set('struct', CompletionItemKind.Struct);
479 > data.set('interface', CompletionItemKind.Interface);
480 > data.set('module', CompletionItemKind.Module);
481 > data.set('property', CompletionItemKind.Property);
482 > data.set('event', CompletionItemKind.Event);
483 > data.set('operator', CompletionItemKind.Operator);
484 > data.set('unit', CompletionItemKind.Unit);
485 > data.set('value', CompletionItemKind.Value);
486 > data.set('constant', CompletionItemKind.Constant);
487 > data.set('enum', CompletionItemKind.Enum);
488 > data.set('enum-member', CompletionItemKind.EnumMember);
489 > data.set('enumMember', CompletionItemKind.EnumMember);
490 > data.set('keyword', CompletionItemKind.Keyword);
491 > data.set('snippet', CompletionItemKind.Snippet);
492 > data.set('text', CompletionItemKind.Text);
493 > data.set('color', CompletionItemKind.Color);
494 > data.set('file', CompletionItemKind.File);
495 > data.set('reference', CompletionItemKind.Reference);
496 > data.set('customcolor', CompletionItemKind.Customcolor);
497 > data.set('folder', CompletionItemKind.Folder);
498 > data.set('type-parameter', CompletionItemKind.TypeParameter);
499 > data.set('typeParameter', CompletionItemKind.TypeParameter);
500 > data.set('account', CompletionItemKind.User);
501 > data.set('issue', CompletionItemKind.Issue);
502 > data.set('tool', CompletionItemKind.Tool);
503 >
504 > /**
505 > * @internal
506 > */
507 > export function fromString(value: string): CompletionItemKind;
508 > /**
509 > * @internal
510 > */
511 > export function fromString(value: string, strict: true): CompletionItemKind | undefined;
512 > /**
513 > * @internal
514 > */
515 > export function fromString(value: string, strict?: boolean): CompletionItemKind | undefined {
516 let res = data.get(value);
517 if (typeof res === 'undefined' && !strict) {
520 return res;
521 }
522 > } languages.ts
523 >
524 > export interface CompletionItemLabel {
525 > label: string;
526 > detail?: string;
527 > description?: string;
528 > }
529 >
530 > export const enum CompletionItemTag {
531 > Deprecated = 1
532 > }
533 >
534 > export const enum CompletionItemInsertTextRule {
535 > None = 0,
536 >
537 > /**
538 > * Adjust whitespace/indentation of multiline insert texts to
539 > * match the current line indentation.
540 > */
541 > KeepWhitespace = 0b001,
542 >
543 > /**
544 > * `insertText` is a snippet.
545 > */
546 > InsertAsSnippet = 0b100,
547 > }
548 >
549 > export interface CompletionItemRanges {
550 > insert: IRange;
551 > replace: IRange;
552 > }
553 >
554 > /**
555 > * A completion item represents a text snippet that is
556 > * proposed to complete text that is being typed.
557 > */
558 > export interface CompletionItem {
559 > /**
560 > * The label of this completion item. By default
561 > * this is also the text that is inserted when selecting
562 > * this completion.
563 > */
564 > label: string | CompletionItemLabel;
565 > /**
566 > * The kind of this completion item. Based on the kind
567 > * an icon is chosen by the editor.
568 > */
569 > kind: CompletionItemKind;
570 > /**
571 > * A modifier to the `kind` which affect how the item
572 > * is rendered, e.g. Deprecated is rendered with a strikeout
573 > */
574 > tags?: ReadonlyArray<CompletionItemTag>;
575 > /**
576 > * A human-readable string with additional information
577 > * about this item, like type or symbol information.
578 > */
579 > detail?: string;
580 > /**
581 > * A human-readable string that represents a doc-comment.
582 > */
583 > documentation?: string | IMarkdownString;
584 > /**
585 > * A string that should be used when comparing this item
586 > * with other items. When `falsy` the {@link CompletionItem.label label}
587 > * is used.
588 > */
589 > sortText?: string;
590 > /**
591 > * A string that should be used when filtering a set of
592 > * completion items. When `falsy` the {@link CompletionItem.label label}
593 > * is used.
594 > */
595 > filterText?: string;
596 > /**
597 > * Select this item when showing. *Note* that only one completion item can be selected and
598 > * that the editor decides which item that is. The rule is that the *first* item of those
599 > * that match best is selected.
600 > */
601 > preselect?: boolean;
602 > /**
603 > * A string or snippet that should be inserted in a document when selecting
604 > * this completion.
605 > */
606 > insertText: string;
607 > /**
608 > * Additional rules (as bitmask) that should be applied when inserting
609 > * this completion.
610 > */
611 > insertTextRules?: CompletionItemInsertTextRule;
612 > /**
613 > * A range of text that should be replaced by this completion item.
614 > *
615 > * *Note:* The range must be a {@link Range.isSingleLine single line} and it must
616 > * {@link Range.contains contain} the position at which completion has been {@link CompletionItemProvider.provideCompletionItems requested}.
617 > */
618 > range: IRange | CompletionItemRanges;
619 > /**
620 > * An optional set of characters that when pressed while this completion is active will accept it first and
621 > * then type that character. *Note* that all commit characters should have `length=1` and that superfluous
622 > * characters will be ignored.
623 > */
624 > commitCharacters?: string[];
625 > /**
626 > * An optional array of additional text edits that are applied when
627 > * selecting this completion. Edits must not overlap with the main edit
628 > * nor with themselves.
629 > */
630 > additionalTextEdits?: ISingleEditOperation[];
631 > /**
632 > * A command that should be run upon acceptance of this item.
633 > */
634 > command?: Command;
635 > /**
636 > * A command that should be run upon acceptance of this item.
637 > */
638 > action?: Command;
639 > /**
640 > * @internal
641 > */
642 > extensionId?: ExtensionIdentifier;
643 >
644 > /**
645 > * @internal
646 > */
647 > _id?: [number, number];
648 > }
649 >
650 > export interface CompletionList {
651 > suggestions: CompletionItem[];
652 > incomplete?: boolean;
653 > dispose?(): void;
654 >
655 > /**
656 > * @internal
657 > */
658 > duration?: number;
659 > }
660 >
661 > /**
662 > * Info provided on partial acceptance.
663 > */
664 > export interface PartialAcceptInfo {
665 > kind: PartialAcceptTriggerKind;
666 > acceptedLength: number;
667 > }
668 >
669 > /**
670 > * How a partial acceptance was triggered.
671 > */
672 > export const enum PartialAcceptTriggerKind {
673 > Word = 0,
674 > Line = 1,
675 > Suggest = 2,
676 > }
677 >
678 > /**
679 > * How a suggest provider was triggered.
680 > */
681 > export const enum CompletionTriggerKind {
682 > Invoke = 0,
683 > TriggerCharacter = 1,
684 > TriggerForIncompleteCompletions = 2
685 > }
686 > /**
687 > * Contains additional information about the context in which
688 > * {@link CompletionItemProvider.provideCompletionItems completion provider} is triggered.
689 > */
690 > export interface CompletionContext {
691 > /**
692 > * How the completion was triggered.
693 > */
694 > triggerKind: CompletionTriggerKind;
695 > /**
696 > * Character that triggered the completion item provider.
697 > *
698 > * `undefined` if provider was not triggered by a character.
699 > */
700 > triggerCharacter?: string;
701 > }
702 > /**
703 > * The completion item provider interface defines the contract between extensions and
704 > * the [IntelliSense](https://code.visualstudio.com/docs/editor/intellisense).
705 > *
706 > * When computing *complete* completion items is expensive, providers can optionally implement
707 > * the `resolveCompletionItem`-function. In that case it is enough to return completion
708 > * items with a {@link CompletionItem.label label} from the
709 > * {@link CompletionItemProvider.provideCompletionItems provideCompletionItems}-function. Subsequently,
710 > * when a completion item is shown in the UI and gains focus this provider is asked to resolve
711 > * the item, like adding {@link CompletionItem.documentation doc-comment} or {@link CompletionItem.detail details}.
712 > */
713 > export interface CompletionItemProvider {
714 >
715 > /**
716 > * Used to identify completions in the (debug) UI and telemetry. This isn't the extension identifier because extensions
717 > * often contribute multiple completion item providers.
718 > *
719 > * @internal
720 > */
721 > _debugDisplayName: string;
722 >
723 > triggerCharacters?: string[];
724 > /**
725 > * Provide completion items for the given position and document.
726 > */
727 > provideCompletionItems(model: model.ITextModel, position: Position, context: CompletionContext, token: CancellationToken): ProviderResult<CompletionList>;
728 >
729 > /**
730 > * Given a completion item fill in more data, like {@link CompletionItem.documentation doc-comment}
731 > * or {@link CompletionItem.detail details}.
732 > *
733 > * The editor will only resolve a completion item once.
734 > */
735 > resolveCompletionItem?(item: CompletionItem, token: CancellationToken): ProviderResult<CompletionItem>;
736 > }
737 >
738 > /**
739 > * How an {@link InlineCompletionsProvider inline completion provider} was triggered.
740 > */
741 > export enum InlineCompletionTriggerKind {
742 > /**
743 > * Completion was triggered automatically while editing.
744 > * It is sufficient to return a single completion item in this case.
745 > */
746 > Automatic = 0,
747 >
748 > /**
749 > * Completion was triggered explicitly by a user gesture.
750 > * Return multiple completion items to enable cycling through them.
751 > */
752 > Explicit = 1,
753 > }
754 >
755 > /**
756 > * Arbitrary data that the provider can pass when firing {@link InlineCompletionsProvider.onDidChangeInlineCompletions}.
757 > * This data is passed back to the provider in {@link InlineCompletionContext.changeHint}.
758 > */
759 > export interface IInlineCompletionChangeHint {
760 > /**
761 > * Arbitrary data that the provider can use to identify what triggered the change.
762 > * This data must be JSON serializable.
763 > */
764 > readonly data?: unknown;
765 > }
766 >
767 > export interface InlineCompletionContext {
768 >
769 > /**
770 > * How the completion was triggered.
771 > */
772 > readonly triggerKind: InlineCompletionTriggerKind;
773 > readonly selectedSuggestionInfo: SelectedSuggestionInfo | undefined;
774 > /**
775 > * @experimental
776 > * @internal
777 > */
778 > readonly userPrompt?: string | undefined;
779 > /**
780 > * @experimental
781 > * @internal
782 > */
783 > readonly requestUuid: string;
784 >
785 > readonly includeInlineEdits: boolean;
786 > readonly includeInlineCompletions: boolean;
787 > readonly requestIssuedDateTime: number;
788 > readonly earliestShownDateTime: number;
789 >
790 > /**
791 > * The change hint that was passed to {@link InlineCompletionsProvider.onDidChangeInlineCompletions}.
792 > * Only set if this request was triggered by such an event.
793 > */
794 > readonly changeHint?: IInlineCompletionChangeHint;
795 > }
796 >
797 > export interface IInlineCompletionModelInfo {
798 > models: IInlineCompletionModel[];
799 > currentModelId: string;
800 > }
801 >
802 > export interface IInlineCompletionModel {
803 > name: string;
804 > id: string;
805 > }
806 >
807 > export interface IInlineCompletionProviderOption {
808 > readonly id: string;
809 > readonly label: string;
810 > readonly values: readonly IInlineCompletionProviderOptionValue[];
811 > readonly currentValueId: string;
812 > }
813 >
814 > export interface IInlineCompletionProviderOptionValue {
815 > readonly id: string;
816 > readonly label: string;
817 > }
818 >
819 > export class SelectedSuggestionInfo {
820 > constructor(
821 public readonly range: IRange,
822 public readonly text: string,
825 ) {
826 }
827 > languages.ts
828 > public equals(other: SelectedSuggestionInfo): boolean {
829 return Range.lift(this.range).equalsRange(other.range)
830 && this.text === other.text
832 && this.isSnippetText === other.isSnippetText;
833 }
834 > } languages.ts
835 >
836 > export interface InlineCompletion {
837 > /**
838 > * The text to insert.
839 > * If the text contains a line break, the range must end at the end of a line.
840 > * If existing text should be replaced, the existing text must be a prefix of the text to insert.
841 > *
842 > * The text can also be a snippet. In that case, a preview with default parameters is shown.
843 > * When accepting the suggestion, the full snippet is inserted.
844 > */
845 > readonly insertText: string | { snippet: string } | undefined;
846 >
847 > /**
848 > * The range to replace.
849 > * Must begin and end on the same line.
850 > * Refers to the current document or `uri` if provided.
851 > */
852 > readonly range?: IRange;
853 >
854 > /**
855 > * An optional array of additional text edits that are applied when
856 > * selecting this completion. Edits must not overlap with the main edit
857 > * nor with themselves.
858 > * Refers to the current document or `uri` if provided.
859 > */
860 > readonly additionalTextEdits?: ISingleEditOperation[];
861 >
862 > /**
863 > * The file for which the edit applies to.
864 > */
865 > readonly uri?: UriComponents;
866 >
867 > /**
868 > * A command that is run upon acceptance of this item.
869 > */
870 > readonly command?: Command;
871 >
872 > readonly gutterMenuLinkAction?: Command;
873 >
874 > /**
875 > * Is called the first time an inline completion is shown.
876 > * @deprecated. Use `onDidShow` of the provider instead.
877 > */
878 > readonly shownCommand?: Command;
879 >
880 > /**
881 > * If set to `true`, unopened closing brackets are removed and unclosed opening brackets are closed.
882 > * Defaults to `false`.
883 > */
884 > readonly completeBracketPairs?: boolean;
885 >
886 > readonly isInlineEdit?: boolean;
887 > readonly showInlineEditMenu?: boolean;
888 >
889 > /** Only show the inline suggestion when the cursor is in the showRange. */
890 > readonly showRange?: IRange;
891 >
892 > readonly warning?: InlineCompletionWarning;
893 >
894 > readonly hint?: IInlineCompletionHint;
895 >
896 > readonly supportsRename?: boolean;
897 >
898 > /**
899 > * Used for telemetry.
900 > */
901 > readonly correlationId?: string | undefined;
902 >
903 > readonly jumpToPosition?: IPosition;
904 >
905 > readonly doNotLog?: boolean;
906 > }
907 >
908 > export interface InlineCompletionWarning {
909 > message: IMarkdownString | string;
910 > icon?: IconPath;
911 > }
912 >
913 > export enum InlineCompletionHintStyle {
914 > Code = 1,
915 > Label = 2
916 > }
917 >
918 > export interface IInlineCompletionHint {
919 > /** Refers to the current document. */
920 > range: IRange;
921 > style: InlineCompletionHintStyle;
922 > content: string;
923 > }
924 >
925 > // TODO: add `| URI | { light: URI; dark: URI }`.
926 > export type IconPath = ThemeIcon;
927 >
928 > export interface InlineCompletions<TItem extends InlineCompletion = InlineCompletion> {
929 > readonly items: readonly TItem[];
930 > /**
931 > * A list of commands associated with the inline completions of this list.
932 > */
933 > readonly commands?: InlineCompletionCommand[];
934 >
935 > readonly suppressSuggestions?: boolean | undefined;
936 >
937 > /**
938 > * When set and the user types a suggestion without derivating from it, the inline suggestion is not updated.
939 > */
940 > readonly enableForwardStability?: boolean | undefined;
941 > }
942 >
943 > export type InlineCompletionCommand = { command: Command; icon?: ThemeIcon };
944 >
945 > export type InlineCompletionProviderGroupId = string;
946 >
947 > export interface InlineCompletionsProvider<T extends InlineCompletions = InlineCompletions> {
948 > provideInlineCompletions(model: model.ITextModel, position: Position, context: InlineCompletionContext, token: CancellationToken): ProviderResult<T>;
949 >
950 > /**
951 > * Will be called when an item is shown.
952 > * @param updatedInsertText Is useful to understand bracket completion.
953 > */
954 > handleItemDidShow?(completions: T, item: T['items'][number], updatedInsertText: string, editDeltaInfo: EditDeltaInfo): void;
955 >
956 > /**
957 > * Will be called when an item is partially accepted. TODO: also handle full acceptance here!
958 > * @param acceptedCharacters Deprecated. Use `info.acceptedCharacters` instead.
959 > */
960 > handlePartialAccept?(completions: T, item: T['items'][number], acceptedCharacters: number, info: PartialAcceptInfo): void;
961 >
962 > /**
963 > * @deprecated Use `handleEndOfLifetime` instead.
964 > */
965 > handleRejection?(completions: T, item: T['items'][number]): void;
966 >
967 > /**
968 > * Is called when an inline completion item is no longer being used.
969 > * Provides a reason of why it is not used anymore.
970 > */
971 > handleEndOfLifetime?(completions: T, item: T['items'][number], reason: InlineCompletionEndOfLifeReason<T['items'][number]>, lifetimeSummary: LifetimeSummary): void;
972 >
973 > /**
974 > * Will be called when a completions list is no longer in use and can be garbage-collected.
975 > */
976 > disposeInlineCompletions(completions: T, reason: InlineCompletionsDisposeReason): void;
977 >
978 > /**
979 > * Fired when the provider wants to trigger a new completion request.
980 > * The event can pass a {@link IInlineCompletionChangeHint} which will be
981 > * included in the {@link InlineCompletionContext} of the subsequent request.
982 > */
983 > onDidChangeInlineCompletions?: Event<IInlineCompletionChangeHint | void>;
984 >
985 > /**
986 > * Only used for {@link yieldsToGroupIds}.
987 > * Multiple providers can have the same group id.
988 > */
989 > groupId?: InlineCompletionProviderGroupId;
990 >
991 > /** @internal */
992 > providerId?: ProviderId;
993 >
994 > /**
995 > * Returns a list of preferred provider {@link groupId}s.
996 > * The current provider is only requested for completions if no provider with a preferred group id returned a result.
997 > */
998 > yieldsToGroupIds?: InlineCompletionProviderGroupId[];
999 >
1000 > excludesGroupIds?: InlineCompletionProviderGroupId[];
1001 >
1002 > displayName?: string;
1003 >
1004 > debounceDelayMs?: number;
1005 >
1006 > modelInfo?: IInlineCompletionModelInfo;
1007 > onDidModelInfoChange?: Event<void>;
1008 > setModelId?(modelId: string): Promise<void>;
1009 >
1010 > providerOptions?: readonly IInlineCompletionProviderOption[];
1011 > onDidProviderOptionsChange?: Event<void>;
1012 > setProviderOption?(optionId: string, valueId: string): Promise<void>;
1013 >
1014 > toString?(): string;
1015 > }
1016 >
1017 >
1018 > /** @internal */
1019 > export class ProviderId {
1020 > public static fromExtensionId(extensionId: string | undefined): ProviderId {
1021 > return new ProviderId(extensionId, undefined, undefined);
1022 > }
1023 >
1024 > constructor(
1025 public readonly extensionId: string | undefined,
1026 public readonly extensionVersion: string | undefined,
1028 ) {
1029 }
1030 > languages.ts
1031 > toString(): string {
1032 let result = '';
1033 if (this.extensionId) {
1045 return result;
1046 }
1047 > languages.ts
1048 > toStringWithoutVersion(): string {
1049 let result = '';
1050 if (this.extensionId) {
1056 return result;
1057 }
1058 > } languages.ts
1059 >
1060 > /** @internal */
1061 > export class VersionedExtensionId {
1062 > public static tryCreate(extensionId: string | undefined, version: string | undefined): VersionedExtensionId | undefined {
1063 > if (!extensionId || !version) {
1064 > return undefined;
1065 > }
1066 > return new VersionedExtensionId(extensionId, version);
1067 > }
1068 >
1069 > constructor(
1070 public readonly extensionId: string,
1071 public readonly version: string,
1072 ) { }
1073 > languages.ts
1074 > toString(): string {
1075 return `${this.extensionId}@${this.version}`;
1076 }
1077 > } languages.ts
1078 >
1079 > export type InlineCompletionsDisposeReason = { kind: 'lostRace' | 'tokenCancellation' | 'other' | 'empty' | 'notTaken' };
1080 >
1081 > export enum InlineCompletionEndOfLifeReasonKind {
1082 > Accepted = 0,
1083 > Rejected = 1,
1084 > Ignored = 2,
1085 > }
1086 >
1087 > export type InlineCompletionEndOfLifeReason<TInlineCompletion = InlineCompletion> = {
1088 > kind: InlineCompletionEndOfLifeReasonKind.Accepted; // User did an explicit action to accept
1089 > alternativeAction: boolean; // Whether the user performed an alternative action.
1090 > } | {
1091 > kind: InlineCompletionEndOfLifeReasonKind.Rejected; // User did an explicit action to reject
1092 > } | {
1093 > kind: InlineCompletionEndOfLifeReasonKind.Ignored;
1094 > supersededBy?: TInlineCompletion;
1095 > userTypingDisagreed: boolean;
1096 > };
1097 >
1098 > export type LifetimeSummary = {
1099 > requestUuid: string;
1100 > correlationId: string | undefined;
1101 > partiallyAccepted: number;
1102 > partiallyAcceptedCountSinceOriginal: number;
1103 > partiallyAcceptedRatioSinceOriginal: number;
1104 > partiallyAcceptedCharactersSinceOriginal: number;
1105 > shown: boolean;
1106 > shownDuration: number;
1107 > shownDurationUncollapsed: number;
1108 > timeUntilShown: number | undefined;
1109 > timeUntilActuallyShown: number | undefined;
1110 > timeUntilProviderRequest: number;
1111 > timeUntilProviderResponse: number;
1112 > notShownReason: string | undefined;
1113 > editorType: string;
1114 > viewKind: string | undefined;
1115 > preceeded: boolean;
1116 > languageId: string;
1117 > requestReason: string;
1118 > performanceMarkers?: string;
1119 > cursorColumnDistance?: number;
1120 > cursorLineDistance?: number;
1121 > lineCountOriginal?: number;
1122 > lineCountModified?: number;
1123 > characterCountOriginal?: number;
1124 > characterCountModified?: number;
1125 > disjointReplacements?: number;
1126 > sameShapeReplacements?: boolean;
1127 > typingInterval: number;
1128 > typingIntervalCharacterCount: number;
1129 > selectedSuggestionInfo: boolean;
1130 > availableProviders: string;
1131 > skuPlan: string | undefined;
1132 > skuType: string | undefined;
1133 > renameCreated: boolean | undefined;
1134 > renameDuration: number | undefined;
1135 > renameTimedOut: boolean | undefined;
1136 > renameDroppedOtherEdits: number | undefined;
1137 > renameDroppedRenameEdits: number | undefined;
1138 > editKind: string | undefined;
1139 > longDistanceHintVisible?: boolean;
1140 > longDistanceHintDistance?: number;
1141 > isForAnotherDocument?: boolean;
1142 > };
1143 >
1144 > export interface CodeAction {
1145 > title: string;
1146 > command?: Command;
1147 > edit?: WorkspaceEdit;
1148 > diagnostics?: IMarkerData[];
1149 > kind?: string;
1150 > isPreferred?: boolean;
1151 > isAI?: boolean;
1152 > disabled?: string;
1153 > ranges?: IRange[];
1154 > }
1155 >
1156 > export const enum CodeActionTriggerType {
1157 > Invoke = 1,
1158 > Auto = 2,
1159 > }
1160 >
1161 > /**
1162 > * @internal
1163 > */
1164 > export interface CodeActionContext {
1165 > only?: string;
1166 > trigger: CodeActionTriggerType;
1167 > }
1168 >
1169 > export interface CodeActionList extends IDisposable {
1170 > readonly actions: ReadonlyArray<CodeAction>;
1171 > }
1172 >
1173 > /**
1174 > * The code action interface defines the contract between extensions and
1175 > * the [light bulb](https://code.visualstudio.com/docs/editor/editingevolved#_code-action) feature.
1176 > * @internal
1177 > */
1178 > export interface CodeActionProvider {
1179 >
1180 > displayName?: string;
1181 >
1182 > extensionId?: string;
1183 >
1184 > /**
1185 > * Provide commands for the given document and range.
1186 > */
1187 > provideCodeActions(model: model.ITextModel, range: Range | Selection, context: CodeActionContext, token: CancellationToken): ProviderResult<CodeActionList>;
1188 >
1189 > /**
1190 > * Given a code action fill in the edit. Will only invoked when missing.
1191 > */
1192 > resolveCodeAction?(codeAction: CodeAction, token: CancellationToken): ProviderResult<CodeAction>;
1193 >
1194 > /**
1195 > * Optional list of CodeActionKinds that this provider returns.
1196 > */
1197 > readonly providedCodeActionKinds?: ReadonlyArray<string>;
1198 >
1199 > readonly documentation?: ReadonlyArray<{ readonly kind: string; readonly command: Command }>;
1200 >
1201 > /**
1202 > * @internal
1203 > */
1204 > _getAdditionalMenuItems?(context: CodeActionContext, actions: readonly CodeAction[]): Command[];
1205 > }
1206 >
1207 > /**
1208 > * @internal
1209 > */
1210 > export interface DocumentPasteEdit {
1211 > readonly title: string;
1212 > readonly kind: HierarchicalKind;
1213 > readonly handledMimeType?: string;
1214 > yieldTo?: readonly DropYieldTo[];
1215 > insertText: string | { readonly snippet: string };
1216 > additionalEdit?: WorkspaceEdit;
1217 > }
1218 >
1219 > /**
1220 > * @internal
1221 > */
1222 > export enum DocumentPasteTriggerKind {
1223 > Automatic = 0,
1224 > PasteAs = 1,
1225 > }
1226 >
1227 > /**
1228 > * @internal
1229 > */
1230 > export interface DocumentPasteContext {
1231 > readonly only?: HierarchicalKind;
1232 > readonly triggerKind: DocumentPasteTriggerKind;
1233 > }
1234 >
1235 > /**
1236 > * @internal
1237 > */
1238 > export interface DocumentPasteEditsSession {
1239 > edits: readonly DocumentPasteEdit[];
1240 > dispose(): void;
1241 > }
1242 >
1243 > /**
1244 > * @internal
1245 > */
1246 > export interface DocumentPasteEditProvider {
1247 > readonly id?: string;
1248 > readonly copyMimeTypes: readonly string[];
1249 > readonly pasteMimeTypes: readonly string[];
1250 > readonly providedPasteEditKinds: readonly HierarchicalKind[];
1251 >
1252 > prepareDocumentPaste?(model: model.ITextModel, ranges: readonly IRange[], dataTransfer: IReadonlyVSDataTransfer, token: CancellationToken): Promise<undefined | IReadonlyVSDataTransfer>;
1253 >
1254 > provideDocumentPasteEdits?(model: model.ITextModel, ranges: readonly IRange[], dataTransfer: IReadonlyVSDataTransfer, context: DocumentPasteContext, token: CancellationToken): Promise<DocumentPasteEditsSession | undefined>;
1255 >
1256 > resolveDocumentPasteEdit?(edit: DocumentPasteEdit, token: CancellationToken): Promise<DocumentPasteEdit>;
1257 > }
1258 >
1259 > /**
1260 > * Represents a parameter of a callable-signature. A parameter can
1261 > * have a label and a doc-comment.
1262 > */
1263 > export interface ParameterInformation {
1264 > /**
1265 > * The label of this signature. Will be shown in
1266 > * the UI.
1267 > */
1268 > label: string | [number, number];
1269 > /**
1270 > * The human-readable doc-comment of this signature. Will be shown
1271 > * in the UI but can be omitted.
1272 > */
1273 > documentation?: string | IMarkdownString;
1274 > }
1275 > /**
1276 > * Represents the signature of something callable. A signature
1277 > * can have a label, like a function-name, a doc-comment, and
1278 > * a set of parameters.
1279 > */
1280 > export interface SignatureInformation {
1281 > /**
1282 > * The label of this signature. Will be shown in
1283 > * the UI.
1284 > */
1285 > label: string;
1286 > /**
1287 > * The human-readable doc-comment of this signature. Will be shown
1288 > * in the UI but can be omitted.
1289 > */
1290 > documentation?: string | IMarkdownString;
1291 > /**
1292 > * The parameters of this signature.
1293 > */
1294 > parameters: ParameterInformation[];
1295 > /**
1296 > * Index of the active parameter.
1297 > *
1298 > * If provided, this is used in place of `SignatureHelp.activeSignature`.
1299 > */
1300 > activeParameter?: number;
1301 > }
1302 > /**
1303 > * Signature help represents the signature of something
1304 > * callable. There can be multiple signatures but only one
1305 > * active and only one active parameter.
1306 > */
1307 > export interface SignatureHelp {
1308 > /**
1309 > * One or more signatures.
1310 > */
1311 > signatures: SignatureInformation[];
1312 > /**
1313 > * The active signature.
1314 > */
1315 > activeSignature: number;
1316 > /**
1317 > * The active parameter of the active signature.
1318 > */
1319 > activeParameter: number;
1320 > }
1321 >
1322 > export interface SignatureHelpResult extends IDisposable {
1323 > value: SignatureHelp;
1324 > }
1325 >
1326 > export enum SignatureHelpTriggerKind {
1327 > Invoke = 1,
1328 > TriggerCharacter = 2,
1329 > ContentChange = 3,
1330 > }
1331 >
1332 > export interface SignatureHelpContext {
1333 > readonly triggerKind: SignatureHelpTriggerKind;
1334 > readonly triggerCharacter?: string;
1335 > readonly isRetrigger: boolean;
1336 > readonly activeSignatureHelp?: SignatureHelp;
1337 > }
1338 >
1339 > /**
1340 > * The signature help provider interface defines the contract between extensions and
1341 > * the [parameter hints](https://code.visualstudio.com/docs/editor/intellisense)-feature.
1342 > */
1343 > export interface SignatureHelpProvider {
1344 >
1345 > readonly signatureHelpTriggerCharacters?: ReadonlyArray<string>;
1346 > readonly signatureHelpRetriggerCharacters?: ReadonlyArray<string>;
1347 >
1348 > /**
1349 > * Provide help for the signature at the given position and document.
1350 > */
1351 > provideSignatureHelp(model: model.ITextModel, position: Position, token: CancellationToken, context: SignatureHelpContext): ProviderResult<SignatureHelpResult>;
1352 > }
1353 >
1354 > /**
1355 > * A document highlight kind.
1356 > */
1357 > export enum DocumentHighlightKind {
1358 > /**
1359 > * A textual occurrence.
1360 > */
1361 > Text,
1362 > /**
1363 > * Read-access of a symbol, like reading a variable.
1364 > */
1365 > Read,
1366 > /**
1367 > * Write-access of a symbol, like writing to a variable.
1368 > */
1369 > Write
1370 > }
1371 > /**
1372 > * A document highlight is a range inside a text document which deserves
1373 > * special attention. Usually a document highlight is visualized by changing
1374 > * the background color of its range.
1375 > */
1376 > export interface DocumentHighlight {
1377 > /**
1378 > * The range this highlight applies to.
1379 > */
1380 > range: IRange;
1381 > /**
1382 > * The highlight kind, default is {@link DocumentHighlightKind.Text text}.
1383 > */
1384 > kind?: DocumentHighlightKind;
1385 > }
1386 >
1387 > /**
1388 > * Represents a set of document highlights for a specific URI.
1389 > */
1390 > export interface MultiDocumentHighlight {
1391 > /**
1392 > * The URI of the document that the highlights belong to.
1393 > */
1394 > uri: URI;
1395 >
1396 > /**
1397 > * The set of highlights for the document.
1398 > */
1399 > highlights: DocumentHighlight[];
1400 > }
1401 >
1402 > /**
1403 > * The document highlight provider interface defines the contract between extensions and
1404 > * the word-highlight-feature.
1405 > */
1406 > export interface DocumentHighlightProvider {
1407 > /**
1408 > * Provide a set of document highlights, like all occurrences of a variable or
1409 > * all exit-points of a function.
1410 > */
1411 > provideDocumentHighlights(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<DocumentHighlight[]>;
1412 > }
1413 >
1414 > /**
1415 > * A provider that can provide document highlights across multiple documents.
1416 > */
1417 > export interface MultiDocumentHighlightProvider {
1418 > readonly selector: LanguageSelector;
1419 >
1420 > /**
1421 > * Provide a Map of URI --> document highlights, like all occurrences of a variable or
1422 > * all exit-points of a function.
1423 > *
1424 > * Used in cases such as split view, notebooks, etc. where there can be multiple documents
1425 > * with shared symbols.
1426 > *
1427 > * @param primaryModel The primary text model.
1428 > * @param position The position at which to provide document highlights.
1429 > * @param otherModels The other text models to search for document highlights.
1430 > * @param token A cancellation token.
1431 > * @returns A map of URI to document highlights.
1432 > */
1433 > provideMultiDocumentHighlights(primaryModel: model.ITextModel, position: Position, otherModels: model.ITextModel[], token: CancellationToken): ProviderResult<Map<URI, DocumentHighlight[]>>;
1434 > }
1435 >
1436 > /**
1437 > * The linked editing range provider interface defines the contract between extensions and
1438 > * the linked editing feature.
1439 > */
1440 > export interface LinkedEditingRangeProvider {
1441 >
1442 > /**
1443 > * Provide a list of ranges that can be edited together.
1444 > */
1445 > provideLinkedEditingRanges(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<LinkedEditingRanges>;
1446 > }
1447 >
1448 > /**
1449 > * Represents a list of ranges that can be edited together along with a word pattern to describe valid contents.
1450 > */
1451 > export interface LinkedEditingRanges {
1452 > /**
1453 > * A list of ranges that can be edited together. The ranges must have
1454 > * identical length and text content. The ranges cannot overlap
1455 > */
1456 > ranges: IRange[];
1457 >
1458 > /**
1459 > * An optional word pattern that describes valid contents for the given ranges.
1460 > * If no pattern is provided, the language configuration's word pattern will be used.
1461 > */
1462 > wordPattern?: RegExp;
1463 > }
1464 >
1465 > /**
1466 > * Value-object that contains additional information when
1467 > * requesting references.
1468 > */
1469 > export interface ReferenceContext {
1470 > /**
1471 > * Include the declaration of the current symbol.
1472 > */
1473 > includeDeclaration: boolean;
1474 > }
1475 > /**
1476 > * The reference provider interface defines the contract between extensions and
1477 > * the [find references](https://code.visualstudio.com/docs/editor/editingevolved#_peek)-feature.
1478 > */
1479 > export interface ReferenceProvider {
1480 > /**
1481 > * Provide a set of project-wide references for the given position and document.
1482 > */
1483 > provideReferences(model: model.ITextModel, position: Position, context: ReferenceContext, token: CancellationToken): ProviderResult<Location[]>;
1484 > }
1485 >
1486 > /**
1487 > * Represents a location inside a resource, such as a line
1488 > * inside a text file.
1489 > */
1490 > export interface Location {
1491 > /**
1492 > * The resource identifier of this location.
1493 > */
1494 > uri: URI;
1495 > /**
1496 > * The document range of this locations.
1497 > */
1498 > range: IRange;
1499 > }
1500 >
1501 > export interface LocationLink {
1502 > /**
1503 > * A range to select where this link originates from.
1504 > */
1505 > originSelectionRange?: IRange;
1506 >
1507 > /**
1508 > * The target uri this link points to.
1509 > */
1510 > uri: URI;
1511 >
1512 > /**
1513 > * The full range this link points to.
1514 > */
1515 > range: IRange;
1516 >
1517 > /**
1518 > * A range to select this link points to. Must be contained
1519 > * in `LocationLink.range`.
1520 > */
1521 > targetSelectionRange?: IRange;
1522 > }
1523 >
1524 > /**
1525 > * @internal
1526 > */
1527 > export function isLocationLink(thing: unknown): thing is LocationLink {
1528 return !!thing
1529 && URI.isUri((thing as LocationLink).uri)
1531 && (Range.isIRange((thing as LocationLink).originSelectionRange) || Range.isIRange((thing as LocationLink).targetSelectionRange));
1532 }
1533 > languages.ts
1534 > /**
1535 > * @internal
1536 > */
1537 > export function isLocation(thing: unknown): thing is Location {
1538 return !!thing
1539 && URI.isUri((thing as Location).uri)
1540 && Range.isIRange((thing as Location).range);
1541 }
1542 > languages.ts
1543 >
1544 > export type Definition = Location | Location[] | LocationLink[];
1545 >
1546 > /**
1547 > * The definition provider interface defines the contract between extensions and
1548 > * the [go to definition](https://code.visualstudio.com/docs/editor/editingevolved#_go-to-definition)
1549 > * and peek definition features.
1550 > */
1551 > export interface DefinitionProvider {
1552 > /**
1553 > * Provide the definition of the symbol at the given position and document.
1554 > */
1555 > provideDefinition(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<Definition | LocationLink[]>;
1556 > }
1557 >
1558 > /**
1559 > * The definition provider interface defines the contract between extensions and
1560 > * the [go to definition](https://code.visualstudio.com/docs/editor/editingevolved#_go-to-definition)
1561 > * and peek definition features.
1562 > */
1563 > export interface DeclarationProvider {
1564 > /**
1565 > * Provide the declaration of the symbol at the given position and document.
1566 > */
1567 > provideDeclaration(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<Definition | LocationLink[]>;
1568 > }
1569 >
1570 > /**
1571 > * The implementation provider interface defines the contract between extensions and
1572 > * the go to implementation feature.
1573 > */
1574 > export interface ImplementationProvider {
1575 > /**
1576 > * Provide the implementation of the symbol at the given position and document.
1577 > */
1578 > provideImplementation(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<Definition | LocationLink[]>;
1579 > }
1580 >
1581 > /**
1582 > * The type definition provider interface defines the contract between extensions and
1583 > * the go to type definition feature.
1584 > */
1585 > export interface TypeDefinitionProvider {
1586 > /**
1587 > * Provide the type definition of the symbol at the given position and document.
1588 > */
1589 > provideTypeDefinition(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<Definition | LocationLink[]>;
1590 > }
1591 >
1592 > /**
1593 > * A symbol kind.
1594 > */
1595 > export const enum SymbolKind {
1596 > File = 0,
1597 > Module = 1,
1598 > Namespace = 2,
1599 > Package = 3,
1600 > Class = 4,
1601 > Method = 5,
1602 > Property = 6,
1603 > Field = 7,
1604 > Constructor = 8,
1605 > Enum = 9,
1606 > Interface = 10,
1607 > Function = 11,
1608 > Variable = 12,
1609 > Constant = 13,
1610 > String = 14,
1611 > Number = 15,
1612 > Boolean = 16,
1613 > Array = 17,
1614 > Object = 18,
1615 > Key = 19,
1616 > Null = 20,
1617 > EnumMember = 21,
1618 > Struct = 22,
1619 > Event = 23,
1620 > Operator = 24,
1621 > TypeParameter = 25
1622 > }
1623 >
1624 > /**
1625 > * @internal
1626 > */
1627 > export const symbolKindNames: { [symbol: number]: string } = {
1628 > [SymbolKind.Array]: localize('Array', "array"),
1629 > [SymbolKind.Boolean]: localize('Boolean', "boolean"),
1630 > [SymbolKind.Class]: localize('Class', "class"),
1631 > [SymbolKind.Constant]: localize('Constant', "constant"),
1632 > [SymbolKind.Constructor]: localize('Constructor', "constructor"),
1633 > [SymbolKind.Enum]: localize('Enum', "enumeration"),
1634 > [SymbolKind.EnumMember]: localize('EnumMember', "enumeration member"),
1635 > [SymbolKind.Event]: localize('Event', "event"),
1636 > [SymbolKind.Field]: localize('Field', "field"),
1637 > [SymbolKind.File]: localize('File', "file"),
1638 > [SymbolKind.Function]: localize('Function', "function"),
1639 > [SymbolKind.Interface]: localize('Interface', "interface"),
1640 > [SymbolKind.Key]: localize('Key', "key"),
1641 > [SymbolKind.Method]: localize('Method', "method"),
1642 > [SymbolKind.Module]: localize('Module', "module"),
1643 > [SymbolKind.Namespace]: localize('Namespace', "namespace"),
1644 > [SymbolKind.Null]: localize('Null', "null"),
1645 > [SymbolKind.Number]: localize('Number', "number"),
1646 > [SymbolKind.Object]: localize('Object', "object"),
1647 > [SymbolKind.Operator]: localize('Operator', "operator"),
1648 > [SymbolKind.Package]: localize('Package', "package"),
1649 > [SymbolKind.Property]: localize('Property', "property"),
1650 > [SymbolKind.String]: localize('String', "string"),
1651 > [SymbolKind.Struct]: localize('Struct', "struct"),
1652 > [SymbolKind.TypeParameter]: localize('TypeParameter', "type parameter"),
1653 > [SymbolKind.Variable]: localize('Variable', "variable"),
1654 > };
1655 >
1656 > /**
1657 > * @internal
1658 > */
1659 > export function getAriaLabelForSymbol(symbolName: string, kind: SymbolKind): string {
1660 return localize('symbolAriaLabel', '{0} ({1})', symbolName, symbolKindNames[kind]);
1661 }
1662 > languages.ts
1663 > export const enum SymbolTag {
1664 > Deprecated = 1,
1665 > }
1666 >
1667 > /**
1668 > * @internal
1669 > */
1670 > export namespace SymbolKinds {
1671 >
1672 > const byKind = new Map<SymbolKind, ThemeIcon>();
1673 > byKind.set(SymbolKind.File, Codicon.symbolFile);
1674 > byKind.set(SymbolKind.Module, Codicon.symbolModule);
1675 > byKind.set(SymbolKind.Namespace, Codicon.symbolNamespace);
1676 > byKind.set(SymbolKind.Package, Codicon.symbolPackage);
1677 > byKind.set(SymbolKind.Class, Codicon.symbolClass);
1678 > byKind.set(SymbolKind.Method, Codicon.symbolMethod);
1679 > byKind.set(SymbolKind.Property, Codicon.symbolProperty);
1680 > byKind.set(SymbolKind.Field, Codicon.symbolField);
1681 > byKind.set(SymbolKind.Constructor, Codicon.symbolConstructor);
1682 > byKind.set(SymbolKind.Enum, Codicon.symbolEnum);
1683 > byKind.set(SymbolKind.Interface, Codicon.symbolInterface);
1684 > byKind.set(SymbolKind.Function, Codicon.symbolFunction);
1685 > byKind.set(SymbolKind.Variable, Codicon.symbolVariable);
1686 > byKind.set(SymbolKind.Constant, Codicon.symbolConstant);
1687 > byKind.set(SymbolKind.String, Codicon.symbolString);
1688 > byKind.set(SymbolKind.Number, Codicon.symbolNumber);
1689 > byKind.set(SymbolKind.Boolean, Codicon.symbolBoolean);
1690 > byKind.set(SymbolKind.Array, Codicon.symbolArray);
1691 > byKind.set(SymbolKind.Object, Codicon.symbolObject);
1692 > byKind.set(SymbolKind.Key, Codicon.symbolKey);
1693 > byKind.set(SymbolKind.Null, Codicon.symbolNull);
1694 > byKind.set(SymbolKind.EnumMember, Codicon.symbolEnumMember);
1695 > byKind.set(SymbolKind.Struct, Codicon.symbolStruct);
1696 > byKind.set(SymbolKind.Event, Codicon.symbolEvent);
1697 > byKind.set(SymbolKind.Operator, Codicon.symbolOperator);
1698 > byKind.set(SymbolKind.TypeParameter, Codicon.symbolTypeParameter);
1699 > /**
1700 > * @internal
1701 > */
1702 > export function toIcon(kind: SymbolKind): ThemeIcon {
1703 let icon = byKind.get(kind);
1704 if (!icon) {
1708 return icon;
1709 }
1710 > languages.ts
1711 > const byCompletionKind = new Map<SymbolKind, CompletionItemKind>();
1712 > byCompletionKind.set(SymbolKind.File, CompletionItemKind.File);
1713 > byCompletionKind.set(SymbolKind.Module, CompletionItemKind.Module);
1714 > byCompletionKind.set(SymbolKind.Namespace, CompletionItemKind.Module);
1715 > byCompletionKind.set(SymbolKind.Package, CompletionItemKind.Module);
1716 > byCompletionKind.set(SymbolKind.Class, CompletionItemKind.Class);
1717 > byCompletionKind.set(SymbolKind.Method, CompletionItemKind.Method);
1718 > byCompletionKind.set(SymbolKind.Property, CompletionItemKind.Property);
1719 > byCompletionKind.set(SymbolKind.Field, CompletionItemKind.Field);
1720 > byCompletionKind.set(SymbolKind.Constructor, CompletionItemKind.Constructor);
1721 > byCompletionKind.set(SymbolKind.Enum, CompletionItemKind.Enum);
1722 > byCompletionKind.set(SymbolKind.Interface, CompletionItemKind.Interface);
1723 > byCompletionKind.set(SymbolKind.Function, CompletionItemKind.Function);
1724 > byCompletionKind.set(SymbolKind.Variable, CompletionItemKind.Variable);
1725 > byCompletionKind.set(SymbolKind.Constant, CompletionItemKind.Constant);
1726 > byCompletionKind.set(SymbolKind.String, CompletionItemKind.Text);
1727 > byCompletionKind.set(SymbolKind.Number, CompletionItemKind.Value);
1728 > byCompletionKind.set(SymbolKind.Boolean, CompletionItemKind.Value);
1729 > byCompletionKind.set(SymbolKind.Array, CompletionItemKind.Value);
1730 > byCompletionKind.set(SymbolKind.Object, CompletionItemKind.Value);
1731 > byCompletionKind.set(SymbolKind.Key, CompletionItemKind.Keyword);
1732 > byCompletionKind.set(SymbolKind.Null, CompletionItemKind.Value);
1733 > byCompletionKind.set(SymbolKind.EnumMember, CompletionItemKind.EnumMember);
1734 > byCompletionKind.set(SymbolKind.Struct, CompletionItemKind.Struct);
1735 > byCompletionKind.set(SymbolKind.Event, CompletionItemKind.Event);
1736 > byCompletionKind.set(SymbolKind.Operator, CompletionItemKind.Operator);
1737 > byCompletionKind.set(SymbolKind.TypeParameter, CompletionItemKind.TypeParameter);
1738 > /**
1739 > * @internal
1740 > */
1741 > export function toCompletionKind(kind: SymbolKind): CompletionItemKind {
1742 let completionKind = byCompletionKind.get(kind);
1743 if (completionKind === undefined) {
1747 return completionKind;
1748 }
1749 > } languages.ts
1750 >
1751 > export interface DocumentSymbol {
1752 > name: string;
1753 > detail: string;
1754 > kind: SymbolKind;
1755 > tags: ReadonlyArray<SymbolTag>;
1756 > containerName?: string;
1757 > range: IRange;
1758 > selectionRange: IRange;
1759 > children?: DocumentSymbol[];
1760 > }
1761 >
1762 > /**
1763 > * The document symbol provider interface defines the contract between extensions and
1764 > * the [go to symbol](https://code.visualstudio.com/docs/editor/editingevolved#_go-to-symbol)-feature.
1765 > */
1766 > export interface DocumentSymbolProvider {
1767 >
1768 > displayName?: string;
1769 >
1770 > /**
1771 > * Provide symbol information for the given document.
1772 > */
1773 > provideDocumentSymbols(model: model.ITextModel, token: CancellationToken): ProviderResult<DocumentSymbol[]>;
1774 > }
1775 >
1776 > export interface TextEdit {
1777 > range: IRange;
1778 > text: string;
1779 > eol?: model.EndOfLineSequence;
1780 > }
1781 >
1782 > /** @internal */
1783 > export abstract class TextEdit {
1784 > static asEditOperation(edit: TextEdit): ISingleEditOperation {
1785 const range = Range.lift(edit.range);
1786 return range.isEmpty()
1788 : EditOperation.replace(range, edit.text);
1789 }
1790 > static isTextEdit(thing: unknown): thing is TextEdit { languages.ts
1791 const possibleTextEdit = thing as TextEdit;
1792 return typeof possibleTextEdit.text === 'string' && Range.isIRange(possibleTextEdit.range);
1793 }
1794 > } languages.ts
1795 >
1796 > /**
1797 > * Interface used to format a model
1798 > */
1799 > export interface FormattingOptions {
1800 > /**
1801 > * Size of a tab in spaces.
1802 > */
1803 > tabSize: number;
1804 > /**
1805 > * Prefer spaces over tabs.
1806 > */
1807 > insertSpaces: boolean;
1808 > }
1809 > /**
1810 > * The document formatting provider interface defines the contract between extensions and
1811 > * the formatting-feature.
1812 > */
1813 > export interface DocumentFormattingEditProvider {
1814 >
1815 > /**
1816 > * @internal
1817 > */
1818 > readonly extensionId?: ExtensionIdentifier;
1819 >
1820 > readonly displayName?: string;
1821 >
1822 > /**
1823 > * Provide formatting edits for a whole document.
1824 > */
1825 > provideDocumentFormattingEdits(model: model.ITextModel, options: FormattingOptions, token: CancellationToken): ProviderResult<TextEdit[]>;
1826 > }
1827 > /**
1828 > * The document formatting provider interface defines the contract between extensions and
1829 > * the formatting-feature.
1830 > */
1831 > export interface DocumentRangeFormattingEditProvider {
1832 > /**
1833 > * @internal
1834 > */
1835 > readonly extensionId?: ExtensionIdentifier;
1836 >
1837 > readonly displayName?: string;
1838 >
1839 > /**
1840 > * Provide formatting edits for a range in a document.
1841 > *
1842 > * The given range is a hint and providers can decide to format a smaller
1843 > * or larger range. Often this is done by adjusting the start and end
1844 > * of the range to full syntax nodes.
1845 > */
1846 > provideDocumentRangeFormattingEdits(model: model.ITextModel, range: Range, options: FormattingOptions, token: CancellationToken): ProviderResult<TextEdit[]>;
1847 >
1848 > provideDocumentRangesFormattingEdits?(model: model.ITextModel, ranges: Range[], options: FormattingOptions, token: CancellationToken): ProviderResult<TextEdit[]>;
1849 > }
1850 > /**
1851 > * The document formatting provider interface defines the contract between extensions and
1852 > * the formatting-feature.
1853 > */
1854 > export interface OnTypeFormattingEditProvider {
1855 >
1856 >
1857 > /**
1858 > * @internal
1859 > */
1860 > readonly extensionId?: ExtensionIdentifier;
1861 >
1862 > autoFormatTriggerCharacters: string[];
1863 >
1864 > /**
1865 > * Provide formatting edits after a character has been typed.
1866 > *
1867 > * The given position and character should hint to the provider
1868 > * what range the position to expand to, like find the matching `{`
1869 > * when `}` has been entered.
1870 > */
1871 > provideOnTypeFormattingEdits(model: model.ITextModel, position: Position, ch: string, options: FormattingOptions, token: CancellationToken): ProviderResult<TextEdit[]>;
1872 > }
1873 >
1874 > /**
1875 > * @internal
1876 > */
1877 > export interface IInplaceReplaceSupportResult {
1878 > value: string;
1879 > range: IRange;
1880 > }
1881 >
1882 > /**
1883 > * A link inside the editor.
1884 > */
1885 > export interface ILink {
1886 > range: IRange;
1887 > url?: URI | string;
1888 > tooltip?: string;
1889 > }
1890 >
1891 > export interface ILinksList {
1892 > links: ILink[];
1893 > dispose?(): void;
1894 > }
1895 > /**
1896 > * A provider of links.
1897 > */
1898 > export interface LinkProvider {
1899 > provideLinks(model: model.ITextModel, token: CancellationToken): ProviderResult<ILinksList>;
1900 > resolveLink?: (link: ILink, token: CancellationToken) => ProviderResult<ILink>;
1901 > }
1902 >
1903 > /**
1904 > * A color in RGBA format.
1905 > */
1906 > export interface IColor {
1907 >
1908 > /**
1909 > * The red component in the range [0-1].
1910 > */
1911 > readonly red: number;
1912 >
1913 > /**
1914 > * The green component in the range [0-1].
1915 > */
1916 > readonly green: number;
1917 >
1918 > /**
1919 > * The blue component in the range [0-1].
1920 > */
1921 > readonly blue: number;
1922 >
1923 > /**
1924 > * The alpha component in the range [0-1].
1925 > */
1926 > readonly alpha: number;
1927 > }
1928 >
1929 > /**
1930 > * String representations for a color
1931 > */
1932 > export interface IColorPresentation {
1933 > /**
1934 > * The label of this color presentation. It will be shown on the color
1935 > * picker header. By default this is also the text that is inserted when selecting
1936 > * this color presentation.
1937 > */
1938 > label: string;
1939 > /**
1940 > * An {@link TextEdit edit} which is applied to a document when selecting
1941 > * this presentation for the color.
1942 > */
1943 > textEdit?: TextEdit;
1944 > /**
1945 > * An optional array of additional {@link TextEdit text edits} that are applied when
1946 > * selecting this color presentation.
1947 > */
1948 > additionalTextEdits?: TextEdit[];
1949 > }
1950 >
1951 > /**
1952 > * A color range is a range in a text model which represents a color.
1953 > */
1954 > export interface IColorInformation {
1955 >
1956 > /**
1957 > * The range within the model.
1958 > */
1959 > range: IRange;
1960 >
1961 > /**
1962 > * The color represented in this range.
1963 > */
1964 > color: IColor;
1965 > }
1966 >
1967 > /**
1968 > * A provider of colors for editor models.
1969 > */
1970 > export interface DocumentColorProvider {
1971 > /**
1972 > * Provides the color ranges for a specific model.
1973 > */
1974 > provideDocumentColors(model: model.ITextModel, token: CancellationToken): ProviderResult<IColorInformation[]>;
1975 > /**
1976 > * Provide the string representations for a color.
1977 > */
1978 > provideColorPresentations(model: model.ITextModel, colorInfo: IColorInformation, token: CancellationToken): ProviderResult<IColorPresentation[]>;
1979 > }
1980 >
1981 > export interface SelectionRange {
1982 > range: IRange;
1983 > }
1984 >
1985 > export interface SelectionRangeProvider {
1986 > /**
1987 > * Provide ranges that should be selected from the given position.
1988 > */
1989 > provideSelectionRanges(model: model.ITextModel, positions: Position[], token: CancellationToken): ProviderResult<SelectionRange[][]>;
1990 > }
1991 >
1992 > export interface FoldingContext {
1993 > }
1994 > /**
1995 > * A provider of folding ranges for editor models.
1996 > */
1997 > export interface FoldingRangeProvider {
1998 >
1999 > /**
2000 > * @internal
2001 > */
2002 > readonly id?: string;
2003 >
2004 > /**
2005 > * An optional event to signal that the folding ranges from this provider have changed.
2006 > */
2007 > onDidChange?: Event<this>;
2008 >
2009 > /**
2010 > * Provides the folding ranges for a specific model.
2011 > */
2012 > provideFoldingRanges(model: model.ITextModel, context: FoldingContext, token: CancellationToken): ProviderResult<FoldingRange[]>;
2013 > }
2014 >
2015 > export interface FoldingRange {
2016 >
2017 > /**
2018 > * The one-based start line of the range to fold. The folded area starts after the line's last character.
2019 > */
2020 > start: number;
2021 >
2022 > /**
2023 > * The one-based end line of the range to fold. The folded area ends with the line's last character.
2024 > */
2025 > end: number;
2026 >
2027 > /**
2028 > * Describes the {@link FoldingRangeKind Kind} of the folding range such as {@link FoldingRangeKind.Comment Comment} or
2029 > * {@link FoldingRangeKind.Region Region}. The kind is used to categorize folding ranges and used by commands
2030 > * like 'Fold all comments'. See
2031 > * {@link FoldingRangeKind} for an enumeration of standardized kinds.
2032 > */
2033 > kind?: FoldingRangeKind;
2034 > }
2035 > export class FoldingRangeKind {
2036 > /**
2037 > * Kind for folding range representing a comment. The value of the kind is 'comment'.
2038 > */
2039 > static readonly Comment = new FoldingRangeKind('comment');
2040 > /**
2041 > * Kind for folding range representing a import. The value of the kind is 'imports'.
2042 > */
2043 > static readonly Imports = new FoldingRangeKind('imports');
2044 > /**
2045 > * Kind for folding range representing regions (for example marked by `#region`, `#endregion`).
2046 > * The value of the kind is 'region'.
2047 > */
2048 > static readonly Region = new FoldingRangeKind('region');
2049 >
2050 > /**
2051 > * Returns a {@link FoldingRangeKind} for the given value.
2052 > *
2053 > * @param value of the kind.
2054 > */
2055 > static fromValue(value: string) {
2056 switch (value) {
2057 case 'comment': return FoldingRangeKind.Comment;
2061 return new FoldingRangeKind(value);
2062 }
2063 > languages.ts
2064 > /**
2065 > * Creates a new {@link FoldingRangeKind}.
2066 > *
2067 > * @param value of the kind.
2068 > */
2069 > public constructor(public value: string) {
2070 > }
2071 > }
2072 >
2073 >
2074 > export interface WorkspaceEditMetadata {
2075 > needsConfirmation: boolean;
2076 > label: string;
2077 > description?: string;
2078 > /**
2079 > * @internal
2080 > */
2081 > iconPath?: ThemeIcon | URI | { light: URI; dark: URI };
2082 > }
2083 >
2084 > export interface WorkspaceFileEditOptions {
2085 > overwrite?: boolean;
2086 > ignoreIfNotExists?: boolean;
2087 > ignoreIfExists?: boolean;
2088 > recursive?: boolean;
2089 > copy?: boolean;
2090 > folder?: boolean;
2091 > skipTrashBin?: boolean;
2092 > maxSize?: number;
2093 >
2094 > /**
2095 > * @internal
2096 > */
2097 > contents?: Promise<VSBuffer>;
2098 > }
2099 >
2100 > export interface IWorkspaceFileEdit {
2101 > oldResource?: URI;
2102 > newResource?: URI;
2103 > options?: WorkspaceFileEditOptions;
2104 > metadata?: WorkspaceEditMetadata;
2105 > }
2106 >
2107 > export interface IWorkspaceTextEdit {
2108 > resource: URI;
2109 > textEdit: TextEdit & { insertAsSnippet?: boolean; keepWhitespace?: boolean };
2110 > versionId: number | undefined;
2111 > metadata?: WorkspaceEditMetadata;
2112 > }
2113 >
2114 > export interface WorkspaceEdit {
2115 > edits: Array<IWorkspaceTextEdit | IWorkspaceFileEdit | ICustomEdit>;
2116 > }
2117 >
2118 > export interface ICustomEdit {
2119 > readonly resource: URI;
2120 > readonly metadata?: WorkspaceEditMetadata;
2121 > undo(): Promise<void> | void;
2122 > redo(): Promise<void> | void;
2123 > }
2124 >
2125 > export interface Rejection {
2126 > rejectReason?: string;
2127 > }
2128 > export interface RenameLocation {
2129 > range: IRange;
2130 > text: string;
2131 > }
2132 >
2133 > export interface RenameProvider {
2134 > provideRenameEdits(model: model.ITextModel, position: Position, newName: string, token: CancellationToken): ProviderResult<WorkspaceEdit & Rejection>;
2135 > resolveRenameLocation?(model: model.ITextModel, position: Position, token: CancellationToken): ProviderResult<RenameLocation & Rejection>;
2136 > }
2137 >
2138 > export enum NewSymbolNameTag {
2139 > AIGenerated = 1
2140 > }
2141 >
2142 > export enum NewSymbolNameTriggerKind {
2143 > Invoke = 0,
2144 > Automatic = 1,
2145 > }
2146 >
2147 > export interface NewSymbolName {
2148 > readonly newSymbolName: string;
2149 > readonly tags?: readonly NewSymbolNameTag[];
2150 > }
2151 >
2152 > export interface NewSymbolNamesProvider {
2153 > supportsAutomaticNewSymbolNamesTriggerKind?: Promise<boolean | undefined>;
2154 > provideNewSymbolNames(model: model.ITextModel, range: IRange, triggerKind: NewSymbolNameTriggerKind, token: CancellationToken): ProviderResult<NewSymbolName[]>;
2155 > }
2156 >
2157 > export interface Command {
2158 > id: string;
2159 > title: string;
2160 > tooltip?: string;
2161 > arguments?: unknown[];
2162 > }
2163 >
2164 > /**
2165 > * @internal
2166 > */
2167 > export namespace Command {
2168 >
2169 > /**
2170 > * @internal
2171 > */
2172 > export function is(obj: unknown): obj is Command {
2173 if (!obj || typeof obj !== 'object') {
2174 return false;
2177 typeof (<Command>obj).title === 'string';
2178 }
2179 > } languages.ts
2180 >
2181 > /**
2182 > * @internal
2183 > */
2184 > export interface CommentThreadTemplate {
2185 > controllerHandle: number;
2186 > label: string;
2187 > acceptInputCommand?: Command;
2188 > additionalCommands?: Command[];
2189 > deleteCommand?: Command;
2190 > }
2191 >
2192 > /**
2193 > * @internal
2194 > */
2195 > export interface CommentInfo<T = IRange> {
2196 > extensionId?: string;
2197 > threads: CommentThread<T>[];
2198 > pendingCommentThreads?: PendingCommentThread[];
2199 > commentingRanges: CommentingRanges;
2200 > }
2201 >
2202 >
2203 > /**
2204 > * @internal
2205 > */
2206 > export interface CommentingRangeResourceHint {
2207 > schemes: readonly string[];
2208 > }
2209 >
2210 > /**
2211 > * @internal
2212 > */
2213 > export enum CommentThreadCollapsibleState {
2214 > /**
2215 > * Determines an item is collapsed
2216 > */
2217 > Collapsed = 0,
2218 > /**
2219 > * Determines an item is expanded
2220 > */
2221 > Expanded = 1
2222 > }
2223 >
2224 > /**
2225 > * @internal
2226 > */
2227 > export enum CommentThreadState {
2228 > Unresolved = 0,
2229 > Resolved = 1
2230 > }
2231 >
2232 > /**
2233 > * @internal
2234 > */
2235 > export enum CommentThreadApplicability {
2236 > Current = 0,
2237 > Outdated = 1
2238 > }
2239 >
2240 > /**
2241 > * @internal
2242 > */
2243 > export interface CommentWidget {
2244 > commentThread: CommentThread;
2245 > comment?: Comment;
2246 > input: string;
2247 > readonly onDidChangeInput: Event<string>;
2248 > }
2249 >
2250 > /**
2251 > * @internal
2252 > */
2253 > export interface CommentInput {
2254 > value: string;
2255 > uri: URI;
2256 > }
2257 >
2258 > export interface CommentThreadRevealOptions {
2259 > preserveFocus: boolean;
2260 > focusReply: boolean;
2261 > }
2262 >
2263 > /**
2264 > * @internal
2265 > */
2266 > export interface CommentThread<T = IRange> {
2267 > isDocumentCommentThread(): this is CommentThread<IRange>;
2268 > commentThreadHandle: number;
2269 > controllerHandle: number;
2270 > extensionId?: string;
2271 > threadId: string;
2272 > resource: string | null;
2273 > range: T | undefined;
2274 > label: string | undefined;
2275 > contextValue: string | undefined;
2276 > comments: ReadonlyArray<Comment> | undefined;
2277 > readonly onDidChangeComments: Event<readonly Comment[] | undefined>;
2278 > collapsibleState?: CommentThreadCollapsibleState;
2279 > initialCollapsibleState?: CommentThreadCollapsibleState;
2280 > readonly onDidChangeInitialCollapsibleState: Event<CommentThreadCollapsibleState | undefined>;
2281 > state?: CommentThreadState;
2282 > applicability?: CommentThreadApplicability;
2283 > canReply: boolean | CommentAuthorInformation;
2284 > input?: CommentInput;
2285 > readonly onDidChangeInput: Event<CommentInput | undefined>;
2286 > readonly onDidChangeLabel: Event<string | undefined>;
2287 > readonly onDidChangeCollapsibleState: Event<CommentThreadCollapsibleState | undefined>;
2288 > readonly onDidChangeState: Event<CommentThreadState | undefined>;
2289 > readonly onDidChangeCanReply: Event<boolean>;
2290 > isDisposed: boolean;
2291 > isTemplate: boolean;
2292 > }
2293 >
2294 > /**
2295 > * @internal
2296 > */
2297 > export interface AddedCommentThread<T = IRange> extends CommentThread<T> {
2298 > editorId?: string;
2299 > }
2300 >
2301 > /**
2302 > * @internal
2303 > */
2304 >
2305 > export interface CommentingRanges {
2306 > readonly resource: URI;
2307 > ranges: IRange[];
2308 > fileComments: boolean;
2309 > }
2310 >
2311 > export interface CommentAuthorInformation {
2312 > name: string;
2313 > iconPath?: UriComponents;
2314 >
2315 > }
2316 >
2317 > /**
2318 > * @internal
2319 > */
2320 > export interface CommentReaction {
2321 > readonly label?: string;
2322 > readonly iconPath?: UriComponents;
2323 > readonly count?: number;
2324 > readonly hasReacted?: boolean;
2325 > readonly canEdit?: boolean;
2326 > readonly reactors?: readonly string[];
2327 > }
2328 >
2329 > /**
2330 > * @internal
2331 > */
2332 > export interface CommentOptions {
2333 > /**
2334 > * An optional string to show on the comment input box when it's collapsed.
2335 > */
2336 > prompt?: string;
2337 >
2338 > /**
2339 > * An optional string to show as placeholder in the comment input box when it's focused.
2340 > */
2341 > placeHolder?: string;
2342 > }
2343 >
2344 > /**
2345 > * @internal
2346 > */
2347 > export enum CommentMode {
2348 > Editing = 0,
2349 > Preview = 1
2350 > }
2351 >
2352 > /**
2353 > * @internal
2354 > */
2355 > export enum CommentState {
2356 > Published = 0,
2357 > Draft = 1
2358 > }
2359 >
2360 > /**
2361 > * @internal
2362 > */
2363 > export interface Comment {
2364 > readonly uniqueIdInThread: number;
2365 > readonly body: string | IMarkdownString;
2366 > readonly userName: string;
2367 > readonly userIconPath?: UriComponents;
2368 > readonly contextValue?: string;
2369 > readonly commentReactions?: CommentReaction[];
2370 > readonly label?: string;
2371 > readonly mode?: CommentMode;
2372 > readonly state?: CommentState;
2373 > readonly timestamp?: string;
2374 > }
2375 >
2376 > export interface PendingCommentThread {
2377 > range: IRange | undefined;
2378 > uri: URI;
2379 > uniqueOwner: string;
2380 > isReply: boolean;
2381 > comment: PendingComment;
2382 > }
2383 >
2384 > export interface PendingComment {
2385 > body: string;
2386 > cursor: IPosition;
2387 > }
2388 >
2389 > /**
2390 > * @internal
2391 > */
2392 > export interface CommentThreadChangedEvent<T> {
2393 > /**
2394 > * Pending comment threads.
2395 > */
2396 > readonly pending: PendingCommentThread[];
2397 >
2398 > /**
2399 > * Added comment threads.
2400 > */
2401 > readonly added: AddedCommentThread<T>[];
2402 >
2403 > /**
2404 > * Removed comment threads.
2405 > */
2406 > readonly removed: CommentThread<T>[];
2407 >
2408 > /**
2409 > * Changed comment threads.
2410 > */
2411 > readonly changed: CommentThread<T>[];
2412 > }
2413 >
2414 > export interface CodeLens {
2415 > range: IRange;
2416 > id?: string;
2417 > command?: Command;
2418 > }
2419 >
2420 > export interface CodeLensList {
2421 > readonly lenses: readonly CodeLens[];
2422 > dispose?(): void;
2423 > }
2424 >
2425 > export interface CodeLensProvider {
2426 > onDidChange?: Event<this>;
2427 > provideCodeLenses(model: model.ITextModel, token: CancellationToken): ProviderResult<CodeLensList>;
2428 > resolveCodeLens?(model: model.ITextModel, codeLens: CodeLens, token: CancellationToken): ProviderResult<CodeLens>;
2429 > }
2430 >
2431 >
2432 > export enum InlayHintKind {
2433 > Type = 1,
2434 > Parameter = 2,
2435 > }
2436 >
2437 > export interface InlayHintLabelPart {
2438 > label: string;
2439 > tooltip?: string | IMarkdownString;
2440 > // collapsible?: boolean;
2441 > command?: Command;
2442 > location?: Location;
2443 > }
2444 >
2445 > export interface InlayHint {
2446 > label: string | InlayHintLabelPart[];
2447 > tooltip?: string | IMarkdownString;
2448 > textEdits?: TextEdit[];
2449 > position: IPosition;
2450 > kind?: InlayHintKind;
2451 > paddingLeft?: boolean;
2452 > paddingRight?: boolean;
2453 > }
2454 >
2455 > export interface InlayHintList {
2456 > hints: InlayHint[];
2457 > dispose(): void;
2458 > }
2459 >
2460 > export interface InlayHintsProvider {
2461 > displayName?: string;
2462 > onDidChangeInlayHints?: Event<void>;
2463 > provideInlayHints(model: model.ITextModel, range: Range, token: CancellationToken): ProviderResult<InlayHintList>;
2464 > resolveInlayHint?(hint: InlayHint, token: CancellationToken): ProviderResult<InlayHint>;
2465 > }
2466 >
2467 > export interface SemanticTokensLegend {
2468 > readonly tokenTypes: string[];
2469 > readonly tokenModifiers: string[];
2470 > }
2471 >
2472 > export interface SemanticTokens {
2473 > readonly resultId?: string;
2474 > readonly data: Uint32Array;
2475 > }
2476 >
2477 > export interface SemanticTokensEdit {
2478 > readonly start: number;
2479 > readonly deleteCount: number;
2480 > readonly data?: Uint32Array;
2481 > }
2482 >
2483 > export interface SemanticTokensEdits {
2484 > readonly resultId?: string;
2485 > readonly edits: SemanticTokensEdit[];
2486 > }
2487 >
2488 > export interface DocumentSemanticTokensProvider {
2489 > readonly onDidChange?: Event<void>;
2490 > getLegend(): SemanticTokensLegend;
2491 > provideDocumentSemanticTokens(model: model.ITextModel, lastResultId: string | null, token: CancellationToken): ProviderResult<SemanticTokens | SemanticTokensEdits>;
2492 > releaseDocumentSemanticTokens(resultId: string | undefined): void;
2493 > }
2494 >
2495 > export interface DocumentRangeSemanticTokensProvider {
2496 > readonly onDidChange?: Event<void>;
2497 > getLegend(): SemanticTokensLegend;
2498 > provideDocumentRangeSemanticTokens(model: model.ITextModel, range: Range, token: CancellationToken): ProviderResult<SemanticTokens>;
2499 > }
2500 >
2501 > /**
2502 > * @internal
2503 > */
2504 > export interface ITokenizationSupportChangedEvent {
2505 > changedLanguages: string[];
2506 > changedColorMap: boolean;
2507 > }
2508 >
2509 > /**
2510 > * @internal
2511 > */
2512 > export interface ILazyTokenizationSupport<TSupport> {
2513 > get tokenizationSupport(): Promise<TSupport | null>;
2514 > }
2515 >
2516 > /**
2517 > * @internal
2518 > */
2519 > export class LazyTokenizationSupport<TSupport = ITokenizationSupport> implements IDisposable, ILazyTokenizationSupport<TSupport> {
2520 > private _tokenizationSupport: Promise<TSupport & IDisposable | null> | null = null;
2521 >
2522 > constructor(private readonly createSupport: () => Promise<TSupport & IDisposable | null>) {
2523 }
2524 > languages.ts
2525 > dispose(): void {
2526 if (this._tokenizationSupport) {
2527 this._tokenizationSupport.then((support) => {
2532 }
2533 }
2534 > languages.ts
2535 > get tokenizationSupport(): Promise<TSupport | null> {
2536 if (!this._tokenizationSupport) {
2537 this._tokenizationSupport = this.createSupport();
2539 return this._tokenizationSupport;
2540 }
2541 > } languages.ts
2542 >
2543 > /**
2544 > * @internal
2545 > */
2546 > export interface ITokenizationRegistry<TSupport> {
2547 >
2548 > /**
2549 > * An event triggered when:
2550 > * - a tokenization support is registered, unregistered or changed.
2551 > * - the color map is changed.
2552 > */
2553 > readonly onDidChange: Event<ITokenizationSupportChangedEvent>;
2554 >
2555 > /**
2556 > * Fire a change event for a language.
2557 > * This is useful for languages that embed other languages.
2558 > */
2559 > handleChange(languageIds: string[]): void;
2560 >
2561 > /**
2562 > * Register a tokenization support.
2563 > */
2564 > register(languageId: string, support: TSupport): IDisposable;
2565 >
2566 > /**
2567 > * Register a tokenization support factory.
2568 > */
2569 > registerFactory(languageId: string, factory: ILazyTokenizationSupport<TSupport>): IDisposable;
2570 >
2571 > /**
2572 > * Get or create the tokenization support for a language.
2573 > * Returns `null` if not found.
2574 > */
2575 > getOrCreate(languageId: string): Promise<TSupport | null>;
2576 >
2577 > /**
2578 > * Get the tokenization support for a language.
2579 > * Returns `null` if not found.
2580 > */
2581 > get(languageId: string): TSupport | null;
2582 >
2583 > /**
2584 > * Returns false if a factory is still pending.
2585 > */
2586 > isResolved(languageId: string): boolean;
2587 >
2588 > /**
2589 > * Set the new color map that all tokens will use in their ColorId binary encoded bits for foreground and background.
2590 > */
2591 > setColorMap(colorMap: Color[]): void;
2592 >
2593 > getColorMap(): Color[] | null;
2594 >
2595 > getDefaultBackground(): Color | null;
2596 > }
2597 >
2598 > /**
2599 > * @internal
2600 > */
2601 > export const TokenizationRegistry: ITokenizationRegistry<ITokenizationSupport> = new TokenizationRegistryImpl();
2602 >
2603 > /**
2604 > * @internal
2605 > */
2606 > export enum ExternalUriOpenerPriority {
2607 > None = 0,
2608 > Option = 1,
2609 > Default = 2,
2610 > Preferred = 3,
2611 > }
2612 >
2613 > /**
2614 > * @internal
2615 > */
2616 > export type DropYieldTo = { readonly kind: HierarchicalKind } | { readonly mimeType: string };
2617 >
2618 > /**
2619 > * @internal
2620 > */
2621 > export interface DocumentDropEdit {
2622 > readonly title: string;
2623 > readonly kind: HierarchicalKind | undefined;
2624 > readonly handledMimeType?: string;
2625 > readonly yieldTo?: readonly DropYieldTo[];
2626 > insertText: string | { readonly snippet: string };
2627 > additionalEdit?: WorkspaceEdit;
2628 > }
2629 >
2630 > /**
2631 > * @internal
2632 > */
2633 > export interface DocumentDropEditsSession {
2634 > edits: readonly DocumentDropEdit[];
2635 > dispose(): void;
2636 > }
2637 >
2638 > /**
2639 > * @internal
2640 > */
2641 > export interface DocumentDropEditProvider {
2642 > readonly id?: string;
2643 > readonly dropMimeTypes?: readonly string[];
2644 > readonly providedDropEditKinds?: readonly HierarchicalKind[];
2645 >
2646 > provideDocumentDropEdits(model: model.ITextModel, position: IPosition, dataTransfer: IReadonlyVSDataTransfer, token: CancellationToken): ProviderResult<DocumentDropEditsSession>;
2647 > resolveDocumentDropEdit?(edit: DocumentDropEdit, token: CancellationToken): Promise<DocumentDropEdit>;
2648 > }
src/vs/workbench/contrib/chat/common/chatService/chatService.ts 1847 covered LOC · 27 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IAction } from '../../../../../base/common/actions.js';
7 > import { DeferredPromise } from '../../../../../base/common/async.js';
8 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
9 > import { IStringDictionary } from '../../../../../base/common/collections.js';
10 > import { Event } from '../../../../../base/common/event.js';
11 > import { IMarkdownString } from '../../../../../base/common/htmlContent.js';
12 > import { DisposableStore, IReference } from '../../../../../base/common/lifecycle.js';
13 > import { autorun, autorunSelfDisposable, IObservable, IReader } from '../../../../../base/common/observable.js';
14 > import { ThemeIcon } from '../../../../../base/common/themables.js';
15 > import { hasKey } from '../../../../../base/common/types.js';
16 > import { URI, UriComponents } from '../../../../../base/common/uri.js';
17 > import { IRange, Range } from '../../../../../editor/common/core/range.js';
18 > import { ISelection } from '../../../../../editor/common/core/selection.js';
19 > import { Command, Location, TextEdit } from '../../../../../editor/common/languages.js';
20 > import { FileType } from '../../../../../platform/files/common/files.js';
21 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
22 > import { IAutostartResult } from '../../../mcp/common/mcpTypes.js';
23 > import { ICellEditOperation } from '../../../notebook/common/notebookCommon.js';
24 > import { IWorkspaceSymbol } from '../../../search/common/search.js';
25 > import { IChatRequestVariableEntry } from '../attachments/chatVariableEntries.js';
26 > import { IChatRequestVariableValue } from '../attachments/chatVariables.js';
27 > import { ReadonlyChatSessionOptionsMap } from '../chatSessionsService.js';
28 > import { ChatAgentLocation, ChatModeKind } from '../constants.js';
29 > import { IChatEditingSession } from '../editing/chatEditingService.js';
30 > import { IChatModel, IChatRequestModeInfo, IChatRequestModel, IChatRequestVariableData, IChatResponseModel, IExportableChatData, ISerializableChatData } from '../model/chatModel.js';
31 > import type { IChatModelReferenceDebugSnapshot } from '../model/chatModelStore.js';
32 > import { IChatAgentCommand, IChatAgentData, IChatAgentResult, UserSelectedTools } from '../participants/chatAgents.js';
33 > import { HookTypeValue } from '../promptSyntax/hookTypes.js';
34 > import { IParsedChatRequest } from '../requestParser/chatParserTypes.js';
35 > import { IChatParserContext } from '../requestParser/chatRequestParser.js';
36 > import { IPreparedToolInvocation, IToolConfirmationMessages, IToolResult, IToolResultInputOutputDetails, ToolDataSource } from '../tools/languageModelToolsService.js';
37 > import { ConfirmationOptionKind, type McpOAuthClient } from '../../../../../platform/agentHost/common/state/protocol/state.js';
38 >
39 > export interface IChatRequest {
40 > message: string;
41 > variables: Record<string, IChatRequestVariableValue[]>;
42 > }
43 >
44 > export enum ChatErrorLevel {
45 > Info = 0,
46 > Warning = 1,
47 > Error = 2
48 > }
49 >
50 > export interface IChatResponseErrorDetailsConfirmationButton {
51 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
52 > data: any;
53 > label: string;
54 > isSecondary?: boolean;
55 > }
56 >
57 > export interface IChatResponseErrorDetails {
58 > message: string;
59 > responseIsIncomplete?: boolean;
60 > responseIsFiltered?: boolean;
61 > responseIsRedacted?: boolean;
62 > isQuotaExceeded?: boolean;
63 > isRateLimited?: boolean;
64 > /**
65 > * If true, the error is an expected operational condition (e.g. user-actionable
66 > * configuration, network connectivity, missing dependency) and should not be
67 > * logged as a `chatAgentError` telemetry event.
68 > */
69 > isExpectedError?: boolean;
70 > level?: ChatErrorLevel;
71 > confirmationButtons?: IChatResponseErrorDetailsConfirmationButton[];
72 > code?: string;
73 > }
74 >
75 > export interface IChatResponseProgressFileTreeData {
76 > label: string;
77 > uri: URI;
78 > type?: FileType;
79 > children?: IChatResponseProgressFileTreeData[];
80 > }
81 >
82 > export type IDocumentContext = {
83 > uri: URI;
84 > version: number;
85 > ranges: IRange[];
86 > };
87 >
88 > export function isIDocumentContext(obj: unknown): obj is IDocumentContext {
89 return (
90 !!obj &&
95 );
96 }
98 > export interface IChatUsedContext {
99 > documents: IDocumentContext[];
100 > kind: 'usedContext';
101 > }
102 >
103 > export function isIUsedContext(obj: unknown): obj is IChatUsedContext {
104 return (
105 !!obj &&
110 );
111 }
113 > export interface IChatContentVariableReference {
114 > variableName: string;
115 > value?: URI | Location;
116 > }
117 >
118 > export function isChatContentVariableReference(obj: unknown): obj is IChatContentVariableReference {
119 return !!obj &&
120 typeof obj === 'object' &&
121 typeof (obj as IChatContentVariableReference).variableName === 'string';
122 }
124 > export enum ChatResponseReferencePartStatusKind {
125 > Complete = 1,
126 > Partial = 2,
127 > Omitted = 3
128 > }
129 >
130 > export enum ChatResponseClearToPreviousToolInvocationReason {
131 > NoReason = 0,
132 > FilteredContentRetry = 1,
133 > CopyrightContentRetry = 2,
134 > }
135 >
136 > export interface IChatContentReference {
137 > reference: URI | Location | IChatContentVariableReference | string;
138 > iconPath?: ThemeIcon | { light: URI; dark?: URI };
139 > options?: {
140 > status?: { description: string; kind: ChatResponseReferencePartStatusKind };
141 > diffMeta?: { added: number; removed: number };
142 > originalUri?: URI;
143 > /** Overrides the reference URI when opening the modified side of a diff. */
144 > modifiedUri?: URI;
145 > isDeletion?: boolean;
146 > };
147 > kind: 'reference';
148 > }
149 >
150 > export interface IChatCodeCitation {
151 > value: URI;
152 > license: string;
153 > snippet: string;
154 > kind: 'codeCitation';
155 > }
156 >
157 > export interface IChatUsagePromptTokenDetail {
158 > category: string;
159 > label: string;
160 > percentageOfPrompt: number;
161 > }
162 >
163 > export interface IChatUsage {
164 > promptTokens: number;
165 > completionTokens: number;
166 > outputBuffer?: number;
167 > promptTokenDetails?: readonly IChatUsagePromptTokenDetail[];
168 > copilotCredits?: number;
169 > /**
170 > * The language-model ID that actually served the request. Set when a
171 > * meta-model (e.g. "auto") routes to a concrete model so consumers
172 > * can look up the real model's metadata (context window size, etc.).
173 > */
174 > actualModelId?: string;
175 > kind: 'usage';
176 > }
177 >
178 > /**
179 > * Formats a copilot credit value for display.
180 > */
181 > export function formatCopilotCredits(credits: number): string {
182 return parseFloat(credits.toFixed(1)).toString();
183 }
185 > export interface IChatContentInlineReference {
186 > resolveId?: string;
187 > inlineReference: URI | Location | IWorkspaceSymbol;
188 > name?: string;
189 > kind: 'inlineReference';
190 > }
191 >
192 > export interface IChatMarkdownContent {
193 > kind: 'markdownContent';
194 > content: IMarkdownString;
195 > inlineReferences?: Record<string, IChatContentInlineReference>;
196 > }
197 >
198 > export interface IChatTreeData {
199 > treeData: IChatResponseProgressFileTreeData;
200 > kind: 'treeData';
201 > }
202 > export interface IMultiDiffResource {
203 > originalUri?: URI;
204 > modifiedUri?: URI;
205 > goToFileUri?: URI;
206 > added?: number;
207 > removed?: number;
208 > }
209 >
210 > export interface IChatMultiDiffInnerData {
211 > title: string;
212 > resources: IMultiDiffResource[];
213 > }
214 >
215 > export interface IChatMultiDiffData {
216 > multiDiffData: IChatMultiDiffInnerData | IObservable<IChatMultiDiffInnerData>;
217 > kind: 'multiDiffData';
218 > collapsed?: boolean;
219 > readOnly?: boolean;
220 > toJSON(): IChatMultiDiffDataSerialized;
221 > }
222 >
223 > export interface IChatMultiDiffDataSerialized {
224 > multiDiffData: IChatMultiDiffInnerData;
225 > kind: 'multiDiffData';
226 > collapsed?: boolean;
227 > readOnly?: boolean;
228 > }
229 >
230 > export class ChatMultiDiffData implements IChatMultiDiffData {
231 > public readonly kind = 'multiDiffData';
232 > public readonly collapsed?: boolean | undefined;
233 > public readonly readOnly?: boolean | undefined;
234 > public readonly multiDiffData: IChatMultiDiffData['multiDiffData'];
235 >
236 > constructor(opts: {
237 multiDiffData: IChatMultiDiffInnerData | IObservable<IChatMultiDiffInnerData>;
238 collapsed?: boolean;
243 this.multiDiffData = opts.multiDiffData;
244 }
246 > toJSON(): IChatMultiDiffDataSerialized {
247 return {
248 kind: this.kind,
252 };
253 }
254 > } chatService.ts
255 >
256 > export interface IChatProgressMessage {
257 > content: IMarkdownString;
258 > kind: 'progressMessage';
259 > shimmer?: boolean;
260 > }
261 >
262 > export interface IChatSystemNotificationPart {
263 > content: IMarkdownString;
264 > kind: 'systemNotification';
265 > }
266 >
267 > export interface IChatTask extends IChatTaskDto {
268 > deferred: DeferredPromise<string | void>;
269 > progress: (IChatWarningMessage | IChatContentReference)[];
270 > readonly onDidAddProgress: Event<IChatWarningMessage | IChatContentReference>;
271 > add(progress: IChatWarningMessage | IChatContentReference): void;
272 >
273 > complete: (result: string | void) => void;
274 > task: () => Promise<string | void>;
275 > isSettled: () => boolean;
276 > toJSON(): IChatTaskSerialized;
277 > }
278 >
279 > export interface IChatUndoStop {
280 > kind: 'undoStop';
281 > id: string;
282 > }
283 >
284 > export interface IChatExternalEditsDto {
285 > kind: 'externalEdits';
286 > undoStopId: string;
287 > start: boolean; /** true=start, false=stop */
288 > resources: UriComponents[];
289 > /**
290 > * When present, these URIs are read instead of the `resources` URIs
291 > * (by-index) when capturing file snapshots. Used by the agent host
292 > * to provide before/after content from the remote filesystem
293 > * or from stored snapshots.
294 > */
295 > contentFor?: UriComponents[];
296 > }
297 >
298 > export interface IChatTaskDto {
299 > content: IMarkdownString;
300 > kind: 'progressTask';
301 > }
302 >
303 > export interface IChatTaskSerialized {
304 > content: IMarkdownString;
305 > progress: (IChatWarningMessage | IChatContentReference)[];
306 > kind: 'progressTaskSerialized';
307 > }
308 >
309 > export interface IChatTaskResult {
310 > content: IMarkdownString | void;
311 > kind: 'progressTaskResult';
312 > }
313 >
314 > export interface IChatWarningMessage {
315 > content: IMarkdownString;
316 > kind: 'warning';
317 > }
318 >
319 > export interface IChatInfoMessage {
320 > content: IMarkdownString;
321 > kind: 'info';
322 > }
323 >
324 > export interface IChatAgentVulnerabilityDetails {
325 > title: string;
326 > description: string;
327 > }
328 >
329 > export interface IChatResponseCodeblockUriPart {
330 > kind: 'codeblockUri';
331 > uri: URI;
332 > isEdit?: boolean;
333 > undoStopId?: string;
334 > subAgentInvocationId?: string;
335 > }
336 >
337 > export interface IChatAgentMarkdownContentWithVulnerability {
338 > content: IMarkdownString;
339 > vulnerabilities: IChatAgentVulnerabilityDetails[];
340 > kind: 'markdownVuln';
341 > }
342 >
343 > export interface IChatCommandButton {
344 > command: Command;
345 > kind: 'command';
346 > additionalCommands?: Command[]; // rendered as secondary buttons
347 > }
348 >
349 > export interface IChatMoveMessage {
350 > uri: URI;
351 > range: IRange;
352 > kind: 'move';
353 > }
354 >
355 > export interface IChatTextEdit {
356 > uri: URI;
357 > edits: TextEdit[];
358 > kind: 'textEdit';
359 > done?: boolean;
360 > isExternalEdit?: boolean;
361 > }
362 >
363 > export interface IChatClearToPreviousToolInvocation {
364 > kind: 'clearToPreviousToolInvocation';
365 > reason: ChatResponseClearToPreviousToolInvocationReason;
366 > }
367 >
368 > export interface IChatNotebookEdit {
369 > uri: URI;
370 > edits: ICellEditOperation[];
371 > kind: 'notebookEdit';
372 > done?: boolean;
373 > isExternalEdit?: boolean;
374 > }
375 >
376 > export interface IChatWorkspaceFileEdit {
377 > oldResource?: URI;
378 > newResource?: URI;
379 > }
380 >
381 > export interface IChatWorkspaceEdit {
382 > kind: 'workspaceEdit';
383 > edits: IChatWorkspaceFileEdit[];
384 > }
385 >
386 > /**
387 > * The kind of file operation an {@link IChatExternalEdit} represents.
388 > */
389 > export type ChatExternalEditKind = 'create' | 'delete' | 'rename' | 'edit';
390 >
391 > /**
392 > * A summary of a file edit that has been performed externally (i.e. by an
393 > * agent or tool outside of chat's own editing pipeline). Carries everything
394 > * needed to render a static "edit pill" without round-tripping through
395 > * {@link IChatEditingSession} for diff computation: the producer already
396 > * knows the URIs and diff stats up-front.
397 > */
398 > export interface IChatExternalEdit {
399 > kind: 'externalEdit';
400 > /** The resulting file URI (after-URI for create/edit/rename, before-URI for delete). */
401 > uri: URI;
402 > /** The kind of file operation. */
403 > editKind: ChatExternalEditKind;
404 > /** For renames, the file URI before the operation. */
405 > originalUri?: URI;
406 > /** URI from which the "before" content can be read (for diff viewing). Absent for creates. */
407 > beforeContentUri?: URI;
408 > /** URI from which the "after" content can be read (for diff viewing). Absent for deletes. */
409 > afterContentUri?: URI;
410 > /** Pre-computed diff display metadata. */
411 > diff?: { added: number; removed: number };
412 > /** Optional undo-stop id (typically the tool call id) for grouping. */
413 > undoStopId?: string;
414 > }
415 >
416 > export interface IChatConfirmation {
417 > title: string;
418 > message: string | IMarkdownString;
419 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
420 > data: any;
421 > buttons?: string[];
422 > isUsed?: boolean;
423 > kind: 'confirmation';
424 > }
425 >
426 > /**
427 > * Validation rules for a question in a question carousel.
428 > */
429 > export interface IChatQuestionValidation {
430 > minLength?: number;
431 > maxLength?: number;
432 > format?: 'email' | 'uri' | 'date' | 'date-time';
433 > minimum?: number;
434 > maximum?: number;
435 > isInteger?: boolean;
436 > }
437 >
438 > /**
439 > * Represents an individual question in a question carousel.
440 > */
441 > export interface IChatQuestion {
442 > id: string;
443 > type: 'text' | 'singleSelect' | 'multiSelect';
444 > title: string;
445 > message?: string | IMarkdownString;
446 > description?: string;
447 > options?: { id: string; label: string; value: string }[];
448 > defaultValue?: string | string[];
449 > allowFreeformInput?: boolean;
450 > required?: boolean;
451 > validation?: IChatQuestionValidation;
452 > detailedMessage?: string | IMarkdownString;
453 > }
454 >
455 > /** Answer shape for a single-select question. */
456 > export interface IChatSingleSelectAnswer {
457 > selectedValue?: string;
458 > freeformValue?: string;
459 > }
460 >
461 > /** Answer shape for a multi-select question. */
462 > export interface IChatMultiSelectAnswer {
463 > selectedValues: string[];
464 > freeformValue?: string;
465 > }
466 >
467 > /** Union of all possible answer values in a question carousel. */
468 > export type IChatQuestionAnswerValue = string | IChatSingleSelectAnswer | IChatMultiSelectAnswer;
469 >
470 > /** Record mapping question IDs to their typed answer values. */
471 > export type IChatQuestionAnswers = Record<string, IChatQuestionAnswerValue>;
472 >
473 > /**
474 > * A carousel for presenting multiple questions inline in the chat response.
475 > * Users can navigate between questions and submit their answers.
476 > */
477 > export interface IChatQuestionCarousel {
478 > questions: IChatQuestion[];
479 > allowSkip: boolean;
480 > /** Unique identifier for resolving the carousel answers back to the extension */
481 > resolveId?: string;
482 > /** Storage for collected answers when user submits */
483 > data?: IChatQuestionAnswers;
484 > /** Whether the carousel has been submitted/skipped */
485 > isUsed?: boolean;
486 > /** True when accepted/answered outside the carousel UI (e.g. via voice) without structured answers. */
487 > answeredExternally?: boolean;
488 > /** Top-level message shown above the questions (e.g. from MCP elicitation message) */
489 > message?: string | IMarkdownString;
490 > /** Source attribution (e.g. MCP server) */
491 > source?: ToolDataSource;
492 > /** Terminal ID when the carousel was triggered by a terminal needing input */
493 > terminalId?: string;
494 > kind: 'questionCarousel';
495 > }
496 >
497 > export const enum ElicitationState {
498 > Pending = 'pending',
499 > Accepted = 'accepted',
500 > Rejected = 'rejected',
501 > }
502 >
503 > export interface IChatElicitationRequest {
504 > kind: 'elicitation2'; // '2' because initially serialized data used the same kind
505 > title: string | IMarkdownString;
506 > message: string | IMarkdownString;
507 > acceptButtonLabel: string;
508 > rejectButtonLabel: string | undefined;
509 > subtitle?: string | IMarkdownString;
510 > source?: ToolDataSource;
511 > state: IObservable<ElicitationState>;
512 > acceptedResult?: Record<string, unknown>;
513 > moreActions?: IAction[];
514 > riskAssessment?: { toolId: string; parameters: unknown };
515 > accept(value: IAction | true): Promise<void>;
516 > reject?: () => Promise<void>;
517 > isHidden?: IObservable<boolean>;
518 > hide?(): void;
519 > toJSON(): IChatElicitationRequestSerialized;
520 > }
521 >
522 > export interface IChatElicitationRequestSerialized {
523 > kind: 'elicitationSerialized';
524 > title: string | IMarkdownString;
525 > message: string | IMarkdownString;
526 > subtitle: string | IMarkdownString | undefined;
527 > source: ToolDataSource | undefined;
528 > state: ElicitationState.Accepted | ElicitationState.Rejected;
529 > isHidden: boolean;
530 > acceptedResult?: Record<string, unknown>;
531 > }
532 >
533 > export interface IChatThinkingPart {
534 > kind: 'thinking';
535 > value?: string | string[];
536 > id?: string;
537 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
538 > metadata?: { readonly [key: string]: any };
539 > generatedTitle?: string;
540 > /** Elapsed reasoning time in milliseconds, persisted so the duration survives reload. */
541 > reasoningDurationMs?: number;
542 > }
543 >
544 > /**
545 > * A progress part representing an auto-mode model routing resolution.
546 > * Shown as a collapsible widget in the chat stream: collapsed displays
547 > * "Routed to <model>", expanded shows routing details and confidence.
548 > */
549 > export interface IChatAutoModeResolutionPart {
550 > kind: 'autoModeResolution';
551 > /** The model ID that was selected by the router */
552 > resolvedModel: string;
553 > /** The user-facing display name of the resolved model */
554 > resolvedModelName: string;
555 > /** The router's classification label */
556 > predictedLabel: 'needs_reasoning' | 'no_reasoning' | 'fallback';
557 > /** Confidence score (0-1) from the router */
558 > confidence: number;
559 > }
560 >
561 > /**
562 > * A progress part representing the execution result of a hook.
563 > * Aligned with the hook output JSON structure: { stopReason, systemMessage, hookSpecificOutput }.
564 > * If {@link stopReason} is set, the hook blocked/denied the operation.
565 > */
566 > export interface IChatHookPart {
567 > kind: 'hook';
568 > /** The type of hook that was executed */
569 > hookType: HookTypeValue;
570 > /** If set, the hook blocked processing. This message is shown to the user. */
571 > stopReason?: string;
572 > /** Warning/system message from the hook, shown to the user */
573 > systemMessage?: string;
574 > /** Display name of the tool that was affected by the hook */
575 > toolDisplayName?: string;
576 > metadata?: { readonly [key: string]: unknown };
577 > /** If set, this hook was executed within a subagent invocation and should be grouped with it. */
578 > subAgentInvocationId?: string;
579 > }
580 >
581 > export interface IChatTerminalToolInvocationData {
582 > kind: 'terminal';
583 > commandLine: {
584 > original: string;
585 > userEdited?: string;
586 > toolEdited?: string;
587 > // command to show in the chat UI (potentially different from what is actually run in the terminal)
588 > forDisplay?: string;
589 > // isSandboxWrapped boolean to run in the terminal (potentially different from original command)
590 > isSandboxWrapped?: boolean;
591 > };
592 > /**
593 > * LM-generated intention describing why the command is being run, shown
594 > * above the command in the terminal tool card. Set by the Agent Host; the
595 > * built-in terminal tool leaves this unset.
596 > */
597 > intention?: string;
598 > /** The working directory URI for the terminal */
599 > cwd?: UriComponents;
600 > /**
601 > * Pre-computed confirmation display data (localization must happen at source).
602 > * Contains the command line to show in confirmation (potentially without cd prefix)
603 > * and the formatted cwd label if a cd prefix was extracted.
604 > */
605 > confirmation?: {
606 > /** The command line to display in the confirmation editor */
607 > commandLine: string;
608 > /** The formatted cwd label to show in title (if cd was extracted) */
609 > cwdLabel?: string;
610 > /** The cd prefix to prepend back when user edits */
611 > cdPrefix?: string;
612 > };
613 > /**
614 > * Overrides to apply to the presentation of the tool call only, but not actually change the
615 > * command that gets run. For example, python -c "print('hello')" can be presented as just
616 > * the Python code with Python syntax highlighting.
617 > */
618 > presentationOverrides?: {
619 > /** The command line to display in the UI */
620 > commandLine: string;
621 > /** The language for syntax highlighting */
622 > language?: string;
623 > };
624 > /** Message for model recommending the use of an alternative tool */
625 > alternativeRecommendation?: string;
626 > language: string;
627 > terminalToolSessionId?: string;
628 > /** False for output-only data that must not create a workbench terminal instance. */
629 > isPty?: boolean;
630 > /** The predefined command ID that will be used for this terminal command */
631 > terminalCommandId?: string;
632 > /** Whether the terminal command was started as a background execution */
633 > isBackground?: boolean;
634 > /** Whether the command was explicitly approved to run outside the sandbox */
635 > requestUnsandboxedExecution?: boolean;
636 > /** The model-provided reason for requesting sandbox bypass */
637 > requestUnsandboxedExecutionReason?: string;
638 > /** Whether the terminal command was approved to run sandboxed with unrestricted network access */
639 > requestAllowNetwork?: boolean;
640 > /** The model-provided reason for requesting unrestricted network access within the sandbox */
641 > requestAllowNetworkReason?: string;
642 > /** Serialized URI for the command that was executed in the terminal */
643 > terminalCommandUri?: UriComponents;
644 > /** Serialized output of the executed command */
645 > terminalCommandOutput?: {
646 > text: string;
647 > truncated?: boolean;
648 > lineCount?: number;
649 > };
650 > /** Stored theme colors at execution time to style detached output */
651 > terminalTheme?: {
652 > background?: string;
653 > foreground?: string;
654 > };
655 > /** Stored command state to restore decorations after reload */
656 > terminalCommandState?: {
657 > exitCode?: number;
658 > timestamp?: number;
659 > duration?: number;
660 > };
661 > /** Whether the user chose to continue in background for this tool invocation */
662 > didContinueInBackground?: boolean;
663 > autoApproveInfo?: IMarkdownString;
664 > /** Names of missing sandbox dependencies that the user may choose to install */
665 > missingSandboxDependencies?: string[];
666 > /** Approved repair actions that may make an installed but unusable sandbox dependency work. */
667 > sandboxRemediations?: string[];
668 > /** User-visible reason a sandbox prerequisite cannot be repaired automatically. */
669 > sandboxPrerequisiteFailure?: string;
670 > }
671 >
672 > /**
673 > * @deprecated This is the old API shape, we should support this for a while before removing it so
674 > * we don't break existing chats
675 > */
676 > export interface ILegacyChatTerminalToolInvocationData {
677 > kind: 'terminal';
678 > command: string;
679 > language: string;
680 > }
681 >
682 > export function isLegacyChatTerminalToolInvocationData(data: unknown): data is ILegacyChatTerminalToolInvocationData {
683 return !!data && typeof data === 'object' && 'command' in data && 'language' in data;
684 }
686 > /**
687 > * Routing information for an MCP App's webview sub-RPCs. The kind
688 > * determines where `tools/call`, `resources/read`,
689 > * `sampling/createMessage`, etc. are sent:
690 > *
691 > * - `local`: resolves the MCP server via {@link IMcpService} from a
692 > * `serverDefinitionId` + `collectionId`. Used for locally-configured
693 > * MCP servers whose state lives in the workbench.
694 > * - `agentHost`: routes through {@link IAgentHostService.handleMcpRequest}
695 > * on an AHP `mcp://` side channel. Used for MCP servers owned by an
696 > * agent host (e.g. Copilot CLI).
697 > */
698 > export type ChatMcpAppData =
699 > | {
700 > kind: 'local';
701 > /** URI of the UI resource for rendering (e.g., "ui://weather-server/dashboard") */
702 > resourceUri: string;
703 > /** Reference to the server definition for reconnection */
704 > serverDefinitionId: string;
705 > /** Reference to the collection containing the server */
706 > collectionId: string;
707 > }
708 > | {
709 > kind: 'agentHost';
710 > /** URI of the UI resource for rendering (e.g., "ui://weather-server/dashboard") */
711 > resourceUri: string;
712 > /** AHP `mcp://` channel URI for the originating server. */
713 > channel: string;
714 > /**
715 > * Stable identifier for the originating server, used as the
716 > * additional key when computing the webview origin. Typically the
717 > * AHP customization id. For top-level (bare) MCP servers this id
718 > * is currently session-scoped, so see {@link ChatMcpAppModel} for
719 > * how it avoids growing persistent application storage on every
720 > * new session.
721 > */
722 > serverId: string;
723 > };
724 >
725 > export interface IChatToolInputInvocationData {
726 > kind: 'input';
727 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
728 > rawInput: any;
729 > /** Optional MCP App UI metadata for rendering during and after tool execution */
730 > mcpAppData?: ChatMcpAppData;
731 > }
732 >
733 > export const enum ToolConfirmKind {
734 > Denied,
735 > ConfirmationNotNeeded,
736 > Setting,
737 > LmServicePerTool,
738 > UserAction,
739 > Skipped
740 > }
741 >
742 > export type ConfirmedReason =
743 > | { type: ToolConfirmKind.Denied }
744 > | { type: ToolConfirmKind.ConfirmationNotNeeded; reason?: string | IMarkdownString }
745 > | { type: ToolConfirmKind.Setting; id: string }
746 > | { type: ToolConfirmKind.LmServicePerTool; scope: 'session' | 'workspace' | 'profile' }
747 > | { type: ToolConfirmKind.UserAction; selectedButton?: string; selectedButtonKind?: ConfirmationOptionKind }
748 > | { type: ToolConfirmKind.Skipped };
749 >
750 > /**
751 > * Active-only controls for a tool call executing on another connected client.
752 > */
753 > export interface IChatToolInvocationOtherClientData {
754 > readonly cancel: () => void;
755 > }
756 >
757 > export interface IChatToolInvocation {
758 > readonly presentation: IPreparedToolInvocation['presentation'];
759 > readonly toolSpecificData?: IChatTerminalToolInvocationData | ILegacyChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatPullRequestContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatToolResourcesInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfigurationData | IChatAutomationConfiguredData;
760 > /** Active-only metadata that is omitted when the invocation is serialized. */
761 > readonly otherClientToolCall?: IChatToolInvocationOtherClientData;
762 > /**
763 > * Observable that tracks the `kind` of `toolSpecificData`. Used by the
764 > * tool invocation part to re-render when the kind changes (e.g. from
765 > * `'input'` to `'terminal'` when terminal content arrives).
766 > */
767 > readonly toolSpecificDataKind: IObservable<string | undefined>;
768 > readonly originMessage: string | IMarkdownString | undefined;
769 > readonly invocationMessage: string | IMarkdownString;
770 > readonly pastTenseMessage: string | IMarkdownString | undefined;
771 > readonly source: ToolDataSource;
772 > readonly toolId: string;
773 > readonly toolCallId: string;
774 > readonly subAgentInvocationId?: string;
775 > readonly icon?: ThemeIcon;
776 > readonly state: IObservable<IChatToolInvocation.State>;
777 > generatedTitle?: string;
778 > isAttachedToThinking: boolean;
779 >
780 > kind: 'toolInvocation';
781 >
782 > toJSON(): IChatToolInvocationSerialized;
783 > }
784 >
785 > export namespace IChatToolInvocation {
786 > export const enum StateKind {
787 > /** Tool call is streaming partial input from the LM */
788 > Streaming,
789 > WaitingForConfirmation,
790 > Executing,
791 > WaitingForPostApproval,
792 > Completed,
793 > Cancelled,
794 > WaitingForAuthentication,
795 > }
796 >
797 > interface IChatToolInvocationStateBase {
798 > type: StateKind;
799 > }
800 >
801 > export interface IChatToolInvocationStreamingState extends IChatToolInvocationStateBase {
802 > type: StateKind.Streaming;
803 > /** Observable partial input from the LM stream */
804 > readonly partialInput: IObservable<unknown>;
805 > /** Custom invocation message from handleToolStream */
806 > readonly streamingMessage: IObservable<string | IMarkdownString | undefined>;
807 > }
808 >
809 > /** Properties available after streaming is complete */
810 > interface IChatToolInvocationPostStreamState {
811 > readonly parameters: unknown;
812 > readonly confirmationMessages?: IToolConfirmationMessages;
813 > }
814 >
815 > interface IChatToolInvocationWaitingForConfirmationState extends IChatToolInvocationStateBase, IChatToolInvocationPostStreamState {
816 > type: StateKind.WaitingForConfirmation;
817 > confirm(reason: ConfirmedReason): void;
818 > }
819 >
820 > interface IChatToolInvocationPostConfirmState extends IChatToolInvocationPostStreamState {
821 > confirmed: ConfirmedReason;
822 > }
823 >
824 > interface IChatToolInvocationExecutingState extends IChatToolInvocationStateBase, IChatToolInvocationPostConfirmState {
825 > type: StateKind.Executing;
826 > progress: IObservable<{ message?: string | IMarkdownString; progress: number | undefined }>;
827 > }
828 >
829 > export interface IChatToolInvocationWaitingForAuthenticationState extends IChatToolInvocationStateBase, IChatToolInvocationPostConfirmState {
830 > type: StateKind.WaitingForAuthentication;
831 > readonly server: IChatMcpAuthenticationRequiredServer;
832 > cancel(): void;
833 > }
834 >
835 > interface IChatToolInvocationPostExecuteState extends IChatToolInvocationPostConfirmState {
836 > resultDetails: IToolResult['toolResultDetails'];
837 > }
838 >
839 > interface IChatToolWaitingForPostApprovalState extends IChatToolInvocationStateBase, IChatToolInvocationPostExecuteState {
840 > type: StateKind.WaitingForPostApproval;
841 > confirm(reason: ConfirmedReason): void;
842 > contentForModel: IToolResult['content'];
843 > }
844 >
845 > interface IChatToolInvocationCompleteState extends IChatToolInvocationStateBase, IChatToolInvocationPostExecuteState {
846 > type: StateKind.Completed;
847 > postConfirmed: ConfirmedReason | undefined;
848 > contentForModel: IToolResult['content'];
849 > }
850 >
851 > interface IChatToolInvocationCancelledState extends IChatToolInvocationStateBase, IChatToolInvocationPostStreamState {
852 > type: StateKind.Cancelled;
853 > reason: ToolConfirmKind.Denied | ToolConfirmKind.Skipped;
854 > /** Optional message explaining why the tool was cancelled (e.g., from hook denial) */
855 > reasonMessage?: string | IMarkdownString;
856 > }
857 >
858 > export type State =
859 > | IChatToolInvocationStreamingState
860 > | IChatToolInvocationWaitingForConfirmationState
861 > | IChatToolInvocationExecutingState
862 > | IChatToolInvocationWaitingForAuthenticationState
863 > | IChatToolWaitingForPostApprovalState
864 > | IChatToolInvocationCompleteState
865 > | IChatToolInvocationCancelledState;
866 >
867 > export function executionConfirmedOrDenied(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): ConfirmedReason | undefined {
868 if (invocation.kind === 'toolInvocationSerialized') {
869 if (invocation.isConfirmed === undefined || typeof invocation.isConfirmed === 'boolean') {
883 return state.confirmed;
884 }
886 > export function awaitConfirmation(invocation: IChatToolInvocation, token?: CancellationToken): Promise<ConfirmedReason> {
887 const reason = executionConfirmedOrDenied(invocation);
888 if (reason) {
909 });
910 }
912 > function postApprovalConfirmedOrDenied(invocation: IChatToolInvocation, reader?: IReader): ConfirmedReason | undefined {
913 const state = invocation.state.read(reader);
914 if (state.type === StateKind.Completed) {
921 return undefined;
922 }
924 > export function confirmWith(invocation: IChatToolInvocation | undefined, reason: ConfirmedReason) {
925 const state = invocation?.state.get();
926 if (state?.type === StateKind.WaitingForConfirmation || state?.type === StateKind.WaitingForPostApproval) {
930 return false;
931 }
933 > export function awaitPostConfirmation(invocation: IChatToolInvocation, token?: CancellationToken): Promise<ConfirmedReason> {
934 const reason = postApprovalConfirmedOrDenied(invocation);
935 if (reason) {
956 });
957 }
959 > export function resultDetails(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader) {
960 if (invocation.kind === 'toolInvocationSerialized') {
961 return invocation.resultDetails;
969 return undefined;
970 }
972 > export function isComplete(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): boolean {
973 if (invocation.kind === 'toolInvocationSerialized') {
974 return true; // always cancelled or complete
978 return state.type === StateKind.Completed || state.type === StateKind.Cancelled;
979 }
981 > export function isEffectivelyHidden(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): boolean {
982 if (invocation.presentation === 'hidden') {
983 return true;
988 return false;
989 }
991 > export function isStreaming(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): boolean {
992 if (invocation.kind === 'toolInvocationSerialized') {
993 return false;
997 return state.type === StateKind.Streaming;
998 }
1000 > /**
1001 > * Get parameters from invocation. Returns undefined during streaming state.
1002 > */
1003 > export function getParameters(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): unknown | undefined {
1004 if (invocation.kind === 'toolInvocationSerialized') {
1005 return undefined; // serialized invocations don't store parameters
1013 return state.parameters;
1014 }
1016 > /**
1017 > * Get confirmation messages from invocation. Returns undefined during streaming state.
1018 > */
1019 > export function getConfirmationMessages(invocation: IChatToolInvocation | IChatToolInvocationSerialized, reader?: IReader): IToolConfirmationMessages | undefined {
1020 if (invocation.kind === 'toolInvocationSerialized') {
1021 return undefined; // serialized invocations don't store confirmation messages
1029 return state.confirmationMessages;
1030 }
1031 > } chatService.ts
1032 >
1033 >
1034 > export interface IToolResultOutputDetailsSerialized {
1035 > output: {
1036 > type: 'data';
1037 > mimeType: string;
1038 > base64Data: string;
1039 > };
1040 > }
1041 >
1042 > /**
1043 > * This is a IChatToolInvocation that has been serialized, like after window reload, so it is no longer an active tool invocation.
1044 > */
1045 > export interface IChatToolInvocationSerialized {
1046 > presentation: IPreparedToolInvocation['presentation'];
1047 > toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatPullRequestContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatToolResourcesInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfiguredData;
1048 > invocationMessage: string | IMarkdownString;
1049 > originMessage: string | IMarkdownString | undefined;
1050 > pastTenseMessage: string | IMarkdownString | undefined;
1051 > resultDetails?: Array<URI | Location> | IToolResultInputOutputDetails | IToolResultOutputDetailsSerialized;
1052 > /** boolean used by pre-1.104 versions */
1053 > isConfirmed: ConfirmedReason | boolean | undefined;
1054 > isComplete: boolean;
1055 > toolCallId: string;
1056 > toolId: string;
1057 > readonly icon?: undefined;
1058 > source: ToolDataSource | undefined; // undefined on pre-1.104 versions
1059 > readonly subAgentInvocationId?: string;
1060 > generatedTitle?: string;
1061 > isAttachedToThinking?: boolean;
1062 > kind: 'toolInvocationSerialized';
1063 > }
1064 >
1065 > export interface IChatExtensionsContent {
1066 > extensions: string[];
1067 > kind: 'extensions';
1068 > }
1069 >
1070 > export interface IChatPullRequestContent {
1071 > /**
1072 > * @deprecated use `command` instead
1073 > */
1074 > uri?: URI;
1075 > command: Command;
1076 > title: string;
1077 > description: string;
1078 > author: string;
1079 > linkTag: string;
1080 > kind: 'pullRequest';
1081 > }
1082 >
1083 > export interface IChatSubagentToolInvocationData {
1084 > kind: 'subagent';
1085 > isActive?: boolean;
1086 > description?: string;
1087 > agentName?: string;
1088 > prompt?: string;
1089 > result?: string;
1090 > modelName?: string;
1091 > credits?: number;
1092 > /** Millisecond timestamp when the subagent's first turn started. */
1093 > startedAt?: number;
1094 > /** Final elapsed duration in milliseconds. Set when the subagent stops. */
1095 > duration?: number;
1096 > /**
1097 > * Resource (URI string) of the subagent's own chat, when the subagent runs as
1098 > * a distinct chat (e.g. an agent host worker chat). Used to offer an "Open
1099 > * chat" link that reveals the subagent's read-only chat. Undefined when the
1100 > * subagent has no separately-openable chat. A string (not a `URI`) so it stays
1101 > * serializable across the extension host protocol.
1102 > */
1103 > chatResource?: string;
1104 > }
1105 >
1106 > /**
1107 > * Progress type for external tool invocation updates from extensions.
1108 > * When isComplete is false, creates or updates a tool invocation.
1109 > * When isComplete is true, completes an existing tool invocation.
1110 > */
1111 > export interface IChatExternalToolInvocationUpdate {
1112 > kind: 'externalToolInvocationUpdate';
1113 > toolCallId: string;
1114 > toolName: string;
1115 > isComplete: boolean;
1116 > errorMessage?: string;
1117 > invocationMessage?: string | IMarkdownString;
1118 > pastTenseMessage?: string | IMarkdownString;
1119 > toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatModifiedFilesConfirmationData;
1120 > subagentInvocationId?: string;
1121 > resultDetails?: IToolResultInputOutputDetails;
1122 > }
1123 >
1124 > export interface IChatTodoListContent {
1125 > kind: 'todoList';
1126 > todoList: Array<{
1127 > id: string;
1128 > title: string;
1129 > status: 'not-started' | 'in-progress' | 'completed';
1130 > }>;
1131 > }
1132 >
1133 > export interface IChatSimpleToolInvocationData {
1134 > kind: 'simpleToolInvocation';
1135 > input: string;
1136 > output: string;
1137 > }
1138 >
1139 > export interface IChatSearchToolInvocationData {
1140 > readonly kind: 'search';
1141 > }
1142 >
1143 > export interface IChatToolResourcesInvocationData {
1144 > readonly kind: 'resources';
1145 > readonly values: Array<URI | Location>;
1146 > }
1147 >
1148 > /**
1149 > * Tool-specific data for a completed `create_session` / `create_chat`
1150 > * agent-host tool call. Carries a clickable link so the renderer can show a
1151 > * deterministic confirmation + "open" button instead of relying on the model
1152 > * to echo a markdown link.
1153 > */
1154 > export interface IChatSessionCreatedData {
1155 > readonly kind: 'sessionCreated';
1156 > /** The `agent-host-session://` link that opens the created/owning session. */
1157 > readonly openLink: string;
1158 > /** Label for the button (e.g. the session title / prompt). */
1159 > readonly label: string;
1160 > /** Whether this is a `create_chat` result (vs `create_session`); selects the pill icon. */
1161 > readonly isChat?: boolean;
1162 > }
1163 >
1164 > /**
1165 > * Tool-specific data for a completed automation create or update. The stable
1166 > * automation ID lets the renderer open the Automations editor at the affected
1167 > * entry without relying on model-authored prose.
1168 > */
1169 > export interface IChatAutomationConfiguredData {
1170 > readonly kind: 'automationConfigured';
1171 > readonly automationId: string;
1172 > readonly automationName: string;
1173 > readonly operation: 'created' | 'updated';
1174 > }
1175 >
1176 > /**
1177 > * Transient preparation data used to guard an automation update across the
1178 > * confirmation boundary. It is omitted from serialized chat invocations.
1179 > */
1180 > export interface IChatAutomationConfigurationData {
1181 > readonly kind: 'automationConfiguration';
1182 > readonly expectedAutomationId: string;
1183 > readonly expectedEditableState: string;
1184 > }
1185 >
1186 > export interface IChatModifiedFilesConfirmationData {
1187 > readonly kind: 'modifiedFilesConfirmation';
1188 > readonly options: readonly string[];
1189 > readonly modifiedFiles: readonly {
1190 > readonly uri: UriComponents;
1191 > /** The pending file operation represented by this entry. */
1192 > readonly editKind?: ChatExternalEditKind;
1193 > readonly originalUri?: UriComponents;
1194 > /**
1195 > * Optional URI to read the modified (after) content from for the diff
1196 > * view. When absent, {@link uri} is used as the modified side.
1197 > */
1198 > readonly modifiedContentUri?: UriComponents;
1199 > /**
1200 > * Optional URI to read the original (before) content from for the diff
1201 > * view. When absent, {@link originalUri} is used as the original side.
1202 > */
1203 > readonly originalContentUri?: UriComponents;
1204 > readonly insertions?: number;
1205 > readonly deletions?: number;
1206 > readonly title?: string;
1207 > readonly description?: string;
1208 > }[];
1209 > }
1210 >
1211 > /**
1212 > * Confirmation data for the agent host `viewUnreviewedComments` tool. The
1213 > * comments themselves are not carried here: the renderer fetches them (and
1214 > * performs reveal/delete/accept actions) through commands registered by the
1215 > * agent feedback feature, so this workbench/chat layer stays decoupled from the
1216 > * `vs/sessions` feedback model. The renderer resolves the owning session from
1217 > * its render context. Only the confirmation button labels are needed up front.
1218 > */
1219 > export interface IChatAgentFeedbackReviewConfirmationData {
1220 > readonly kind: 'agentFeedbackReviewConfirmation';
1221 > /** Confirmation button labels (first is the primary/approve action). */
1222 > readonly options: readonly string[];
1223 > }
1224 >
1225 > /**
1226 > * A single unreviewed comment shown in the {@link IChatAgentFeedbackReviewConfirmationData}
1227 > * confirmation. Produced by {@link AgentFeedbackReviewCommandId.GetComments}.
1228 > */
1229 > export interface IChatAgentFeedbackReviewComment {
1230 > readonly id: string;
1231 > /** Localized origin label, e.g. "PR Review" or "Agent Review"; absent for user-authored comments. */
1232 > readonly kindLabel?: string;
1233 > /** The comment body. */
1234 > readonly text: string;
1235 > /** The file the comment is anchored to. */
1236 > readonly fileUri: UriComponents;
1237 > }
1238 >
1239 > /**
1240 > * Command ids the agent feedback review confirmation renderer (workbench/chat)
1241 > * uses to fetch unreviewed comments and apply the user's selection. They are
1242 > * implemented by the agent feedback feature in `vs/sessions`, keeping the chat
1243 > * layer decoupled from the feedback model. Most take the owning session resource
1244 > * (`UriComponents`) as their first argument; {@link AgentFeedbackReviewCommandId.RevealAt}
1245 > * instead resolves the session from the file resource so a rendered tool call
1246 > * can link to a comment without knowing the session URI.
1247 > */
1248 > export const enum AgentFeedbackReviewCommandId {
1249 > /** `(sessionResource)` -> `IChatAgentFeedbackReviewComment[]` (the `created` reviewable comments). */
1250 > GetComments = '_agentFeedbackReview.getComments',
1251 > /** `(sessionResource, commentId)` -> opens the file and reveals the comment. */
1252 > Reveal = '_agentFeedbackReview.reveal',
1253 > /** `(resourceUri, range)` -> resolves the owning session and reveals the comment at that file range. */
1254 > RevealAt = '_agentFeedbackReview.revealAt',
1255 > /** `(sessionResource, commentId)` -> deletes the comment entirely. */
1256 > Delete = '_agentFeedbackReview.delete',
1257 > /** `(sessionResource, commentIds)` -> accepts (reveals) the given comments. */
1258 > Accept = '_agentFeedbackReview.accept',
1259 > }
1260 >
1261 > export interface IChatMcpServersStarting {
1262 > readonly kind: 'mcpServersStarting';
1263 > readonly state?: IObservable<IAutostartResult>; // not hydrated when serialized
1264 > didStartServerIds?: string[];
1265 > toJSON(): IChatMcpServersStartingSerialized;
1266 > }
1267 >
1268 > export interface IChatMcpServersStartingSerialized {
1269 > readonly kind: 'mcpServersStarting';
1270 > readonly state?: undefined;
1271 > didStartServerIds?: string[];
1272 > }
1273 >
1274 > export interface IChatMcpAuthenticationRequired {
1275 > readonly kind: 'mcpAuthenticationRequired';
1276 > readonly sessionResource: UriComponents;
1277 > readonly servers: IObservable<readonly IChatMcpAuthenticationRequiredServer[]>;
1278 > isUsed: boolean;
1279 > }
1280 >
1281 > export interface IChatMcpAuthenticationRequiredServer {
1282 > readonly id: string;
1283 > readonly name: string;
1284 > readonly resource: string;
1285 > readonly oauthClient?: McpOAuthClient;
1286 > readonly authorizationServers?: readonly string[];
1287 > readonly supportedScopes?: readonly string[];
1288 > readonly requiredScopes?: readonly string[];
1289 > readonly reason?: string;
1290 > }
1291 >
1292 > /**
1293 > * Surfaced by agent-host sessions when one or more MCP servers are still in the
1294 > * {@link McpServerStatus.Starting starting} state a noticeable time after a
1295 > * turn began without any content arriving from the host. The part lists the
1296 > * servers still starting and updates dynamically via {@link servers}: it hides
1297 > * itself (by emptying the observable) once every server has started, content
1298 > * starts being received, or the turn ends — whichever happens first.
1299 > *
1300 > * Unlike {@link IChatMcpServersStarting} (used by the in-process MCP autostart
1301 > * flow), this is a lightweight progress hint with no interactive affordance
1302 > * (there is no "Skip" button).
1303 > */
1304 > export interface IChatMcpServersStartingSlow {
1305 > readonly kind: 'mcpServersStartingSlow';
1306 > readonly sessionResource: UriComponents;
1307 > readonly servers: IObservable<readonly IChatMcpStartingServer[]>;
1308 > }
1309 >
1310 > export interface IChatMcpStartingServer {
1311 > readonly id: string;
1312 > readonly name: string;
1313 > }
1314 >
1315 > export interface IChatDisabledClaudeHooksPart {
1316 > readonly kind: 'disabledClaudeHooks';
1317 > }
1318 >
1319 > /** A single approval option shown in the plan review dropdown button. */
1320 > export interface IChatPlanApprovalAction {
1321 > /**
1322 > * Stable identifier for matching the chosen action programmatically.
1323 > * Unlike `label` this is not localized, so callers should compare
1324 > * against `IChatPlanReviewResult.actionId` rather than `action`.
1325 > * Optional for backwards-compatibility; omit for one-off actions
1326 > * where the localized label is the only intended identifier.
1327 > */
1328 > id?: string;
1329 > label: string;
1330 > description?: string;
1331 > default?: boolean;
1332 > /** When set to 'autopilot', a confirmation dialog is shown before proceeding. */
1333 > permissionLevel?: 'autopilot';
1334 > }
1335 >
1336 > /** The result of reviewing a plan. */
1337 > export interface IChatPlanReviewResult {
1338 > /** The chosen action's localized `label`. */
1339 > action?: string;
1340 > /** The chosen action's stable `id`, if it had one. Prefer this over
1341 > * `action` for programmatic comparisons. */
1342 > actionId?: string;
1343 > rejected: boolean;
1344 > /** Combined feedback string sent to the agent (overall comment + inline
1345 > * comments, joined and formatted as markdown). */
1346 > feedback?: string;
1347 > /** Display-only: the overall textarea comment, kept separate from
1348 > * `feedbackInlineMarkdown` so the chat transcript can render the two
1349 > * parts differently. Falls back to `feedback` when unset. */
1350 > feedbackOverall?: string;
1351 > /** Display-only: pre-formatted markdown listing the inline comments
1352 > * (heading + bullets). See `feedbackOverall`. */
1353 > feedbackInlineMarkdown?: string;
1354 > }
1355 >
1356 > /**
1357 > * A plan review widget. Presents a title, markdown plan content, an optional
1358 > * link to edit the backing plan file, a dropdown of approval actions, a reject
1359 > * button and an optional feedback textarea.
1360 > */
1361 > export interface IChatPlanReview {
1362 > kind: 'planReview';
1363 > /** Title to display in the widget header. */
1364 > title: string;
1365 > /** Markdown content rendered in the body (plan summary or contents). */
1366 > content: string;
1367 > /** Selectable approval actions. Displayed as a dropdown primary button. */
1368 > actions: IChatPlanApprovalAction[];
1369 > /** Whether to show the additional feedback textarea. */
1370 > canProvideFeedback: boolean;
1371 > /** Optional URI to the underlying plan file. An Edit button opens it. */
1372 > planUri?: UriComponents;
1373 > /** Unique identifier for resolving the review back to the extension. */
1374 > resolveId?: string;
1375 > /** Stored result once the user has responded. */
1376 > data?: IChatPlanReviewResult;
1377 > /** Whether the widget has been responded to. */
1378 > isUsed?: boolean;
1379 > /** Source attribution. */
1380 > source?: ToolDataSource;
1381 > }
1382 >
1383 > export class ChatMcpServersStarting implements IChatMcpServersStarting {
1384 > public readonly kind = 'mcpServersStarting';
1385 >
1386 > public didStartServerIds?: string[] = [];
1387 >
1388 > public get isEmpty() {
1389 > const s = this.state.get();
1390 > return !s.working && s.serversRequiringInteraction.length === 0;
1391 > }
1392 >
1393 > constructor(public readonly state: IObservable<IAutostartResult>) { }
1394 >
1395 > wait() {
1396 return new Promise<IAutostartResult>(resolve => {
1397 autorunSelfDisposable(reader => {
1404 });
1405 }
1407 > toJSON(): IChatMcpServersStartingSerialized {
1408 return { kind: 'mcpServersStarting', didStartServerIds: this.didStartServerIds };
1409 }
1410 > } chatService.ts
1411 >
1412 > export type IChatProgress =
1413 > | IChatMarkdownContent
1414 > | IChatAgentMarkdownContentWithVulnerability
1415 > | IChatUsage
1416 > | IChatTreeData
1417 > | IChatMultiDiffData
1418 > | IChatMultiDiffDataSerialized
1419 > | IChatUsedContext
1420 > | IChatContentReference
1421 > | IChatContentInlineReference
1422 > | IChatCodeCitation
1423 > | IChatProgressMessage
1424 > | IChatSystemNotificationPart
1425 > | IChatTask
1426 > | IChatTaskResult
1427 > | IChatCommandButton
1428 > | IChatWarningMessage
1429 > | IChatInfoMessage
1430 > | IChatTextEdit
1431 > | IChatNotebookEdit
1432 > | IChatWorkspaceEdit
1433 > | IChatExternalEdit
1434 > | IChatMoveMessage
1435 > | IChatResponseCodeblockUriPart
1436 > | IChatConfirmation
1437 > | IChatQuestionCarousel
1438 > | IChatPlanReview
1439 > | IChatClearToPreviousToolInvocation
1440 > | IChatToolInvocation
1441 > | IChatToolInvocationSerialized
1442 > | IChatExtensionsContent
1443 > | IChatPullRequestContent
1444 > | IChatUndoStop
1445 > | IChatThinkingPart
1446 > | IChatTaskSerialized
1447 > | IChatElicitationRequest
1448 > | IChatElicitationRequestSerialized
1449 > | IChatMcpServersStarting
1450 > | IChatMcpServersStartingSerialized
1451 > | IChatMcpAuthenticationRequired
1452 > | IChatMcpServersStartingSlow
1453 > | IChatHookPart
1454 > | IChatExternalToolInvocationUpdate
1455 > | IChatDisabledClaudeHooksPart
1456 > | IChatAutoModeResolutionPart;
1457 >
1458 > export interface IChatFollowup {
1459 > kind: 'reply';
1460 > message: string;
1461 > agentId: string;
1462 > subCommand?: string;
1463 > title?: string;
1464 > tooltip?: string;
1465 > }
1466 >
1467 > export function isChatFollowup(obj: unknown): obj is IChatFollowup {
1468 return (
1469 !!obj &&
1473 );
1474 }
1476 > export enum ChatAgentVoteDirection {
1477 > Down = 0,
1478 > Up = 1
1479 > }
1480 >
1481 > export interface IChatVoteAction {
1482 > kind: 'vote';
1483 > direction: ChatAgentVoteDirection;
1484 > }
1485 >
1486 > export enum ChatCopyKind {
1487 > // Keyboard shortcut or context menu
1488 > Action = 1,
1489 > Toolbar = 2
1490 > }
1491 >
1492 > export interface IChatCopyAction {
1493 > kind: 'copy';
1494 > codeBlockIndex: number;
1495 > copyKind: ChatCopyKind;
1496 > copiedCharacters: number;
1497 > totalCharacters: number;
1498 > copiedText: string;
1499 > totalLines: number;
1500 > copiedLines: number;
1501 > modelId: string;
1502 > languageId?: string;
1503 > }
1504 >
1505 > export interface IChatInsertAction {
1506 > kind: 'insert';
1507 > codeBlockIndex: number;
1508 > totalCharacters: number;
1509 > totalLines: number;
1510 > languageId?: string;
1511 > modelId: string;
1512 > newFile?: boolean;
1513 > }
1514 >
1515 > export interface IChatApplyAction {
1516 > kind: 'apply';
1517 > codeBlockIndex: number;
1518 > totalCharacters: number;
1519 > totalLines: number;
1520 > languageId?: string;
1521 > modelId: string;
1522 > newFile?: boolean;
1523 > codeMapper?: string;
1524 > editsProposed: boolean;
1525 > }
1526 >
1527 >
1528 > export interface IChatTerminalAction {
1529 > kind: 'runInTerminal';
1530 > codeBlockIndex: number;
1531 > languageId?: string;
1532 > }
1533 >
1534 > export interface IChatCommandAction {
1535 > kind: 'command';
1536 > commandButton: IChatCommandButton;
1537 > }
1538 >
1539 > export interface IChatFollowupAction {
1540 > kind: 'followUp';
1541 > followup: IChatFollowup;
1542 > }
1543 >
1544 > export interface IChatBugReportAction {
1545 > kind: 'bug';
1546 > }
1547 >
1548 > export interface IChatInlineChatCodeAction {
1549 > kind: 'inlineChat';
1550 > action: 'accepted' | 'discarded';
1551 > }
1552 >
1553 >
1554 > export interface IChatEditingSessionAction {
1555 > kind: 'chatEditingSessionAction';
1556 > uri: URI;
1557 > hasRemainingEdits: boolean;
1558 > outcome: 'accepted' | 'rejected' | 'userModified';
1559 > }
1560 >
1561 > export interface IChatEditingHunkAction {
1562 > kind: 'chatEditingHunkAction';
1563 > uri: URI;
1564 > lineCount: number;
1565 > linesAdded: number;
1566 > linesRemoved: number;
1567 > outcome: 'accepted' | 'rejected';
1568 > hasRemainingEdits: boolean;
1569 > modeId?: string;
1570 > modelId?: string;
1571 > languageId?: string;
1572 > }
1573 >
1574 > export type ChatUserAction = IChatVoteAction | IChatCopyAction | IChatInsertAction | IChatApplyAction | IChatTerminalAction | IChatCommandAction | IChatFollowupAction | IChatBugReportAction | IChatInlineChatCodeAction | IChatEditingSessionAction | IChatEditingHunkAction;
1575 >
1576 > export interface IChatUserActionEvent {
1577 > action: ChatUserAction;
1578 > agentId: string | undefined;
1579 > command: string | undefined;
1580 > sessionResource: URI;
1581 > requestId: string;
1582 > result: IChatAgentResult | undefined;
1583 > modelId?: string | undefined;
1584 > modeId?: string | undefined;
1585 > }
1586 >
1587 > export interface IChatDynamicRequest {
1588 > /**
1589 > * The message that will be displayed in the UI
1590 > */
1591 > message: string;
1592 >
1593 > /**
1594 > * Any extra metadata/context that will go to the provider.
1595 > */
1596 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
1597 > metadata?: any;
1598 > }
1599 >
1600 > export interface IChatCompleteResponse {
1601 > message: string | ReadonlyArray<IChatProgress>;
1602 > result?: IChatAgentResult;
1603 > followups?: IChatFollowup[];
1604 > }
1605 >
1606 > export interface IChatSessionStats {
1607 > readonly fileCount: number;
1608 > readonly added: number;
1609 > readonly removed: number;
1610 > }
1611 >
1612 > export type IChatSessionTiming = {
1613 > /**
1614 > * Timestamp when the session was created in milliseconds elapsed since January 1, 1970 00:00:00 UTC.
1615 > */
1616 > readonly created: number;
1617 >
1618 > /**
1619 > * Timestamp when the most recent request started in milliseconds elapsed since January 1, 1970 00:00:00 UTC.
1620 > *
1621 > * Should be undefined if no requests have been made yet.
1622 > */
1623 > readonly lastRequestStarted: number | undefined;
1624 >
1625 > /**
1626 > * Timestamp when the most recent request completed in milliseconds elapsed since January 1, 1970 00:00:00 UTC.
1627 > *
1628 > * Should be undefined if the most recent request is still in progress or if no requests have been made yet.
1629 > */
1630 > readonly lastRequestEnded: number | undefined;
1631 > };
1632 >
1633 > interface ILegacyChatSessionTiming {
1634 > readonly startTime: number;
1635 > readonly endTime?: number;
1636 > }
1637 >
1638 > export function convertLegacyChatSessionTiming(timing: IChatSessionTiming | ILegacyChatSessionTiming): IChatSessionTiming {
1639 if (hasKey(timing, { created: true })) {
1640 return timing;
1646 };
1647 }
1649 > export const enum ResponseModelState {
1650 > Pending,
1651 > Complete,
1652 > Cancelled,
1653 > Failed,
1654 > NeedsInput,
1655 > }
1656 >
1657 > export interface IChatDetail {
1658 > sessionResource: URI;
1659 > title: string;
1660 > lastMessageDate: number;
1661 > // Also support old timing format for backwards compatibility with persisted data
1662 > timing: IChatSessionTiming | ILegacyChatSessionTiming;
1663 > isActive: boolean;
1664 > stats?: IChatSessionStats;
1665 > lastResponseState: ResponseModelState;
1666 > /**
1667 > * The working directory URI associated with this session.
1668 > * Only populated in the sessions/agents window context.
1669 > */
1670 > workingDirectory?: URI;
1671 > }
1672 >
1673 > export interface IChatProviderInfo {
1674 > id: string;
1675 > }
1676 >
1677 > export interface IChatSendRequestResponseState {
1678 > responseCreatedPromise: Promise<IChatResponseModel>;
1679 > responseCompletePromise: Promise<void>;
1680 > }
1681 >
1682 > export interface IChatSendRequestData extends IChatSendRequestResponseState {
1683 > agent: IChatAgentData;
1684 > slashCommand?: IChatAgentCommand;
1685 > }
1686 >
1687 > /**
1688 > * Result of a sendRequest call - a discriminated union of possible outcomes.
1689 > */
1690 > export type ChatSendResult =
1691 > | ChatSendResultRejected
1692 > | ChatSendResultSent
1693 > | ChatSendResultQueued;
1694 >
1695 > export interface ChatSendResultRejected {
1696 > readonly kind: 'rejected';
1697 > readonly reason: string;
1698 > /** Set when the session was replaced before the request was rejected (e.g. untitled -> read-only contributed session). */
1699 > readonly newSessionResource?: URI;
1700 > }
1701 >
1702 > export interface ChatSendResultSent {
1703 > readonly kind: 'sent';
1704 > readonly data: IChatSendRequestData;
1705 > /** Set when the session was replaced by a new one (e.g. untitled -> real contributed session). */
1706 > readonly newSessionResource?: URI;
1707 > }
1708 >
1709 > export interface ChatSendResultQueued {
1710 > readonly kind: 'queued';
1711 > /**
1712 > * Promise that resolves when the queued message is actually processed.
1713 > * Will resolve to a 'sent' or 'rejected' result.
1714 > */
1715 > readonly deferred: Promise<ChatSendResult>;
1716 > }
1717 >
1718 > export namespace ChatSendResult {
1719 > export function isSent(result: ChatSendResult): result is ChatSendResultSent {
1720 return result.kind === 'sent';
1721 }
1723 > export function isRejected(result: ChatSendResult): result is ChatSendResultRejected {
1724 return result.kind === 'rejected';
1725 }
1727 > export function isQueued(result: ChatSendResult): result is ChatSendResultQueued {
1728 return result.kind === 'queued';
1729 }
1731 > /** Assertion function for tests - asserts that the result is a sent result */
1732 > export function assertSent(result: ChatSendResult): asserts result is ChatSendResultSent {
1733 if (result.kind !== 'sent') {
1734 throw new Error(`Expected ChatSendResult to be 'sent', but was '${result.kind}'`);
1735 }
1736 }
1737 > } chatService.ts
1738 >
1739 > export interface IChatEditorLocationData {
1740 > type: ChatAgentLocation.EditorInline;
1741 > id: string;
1742 > document: URI;
1743 > selection: ISelection;
1744 > wholeRange: IRange;
1745 > }
1746 >
1747 > export interface IChatNotebookLocationData {
1748 > type: ChatAgentLocation.Notebook;
1749 > sessionInputUri: URI;
1750 > }
1751 >
1752 > export interface IChatTerminalLocationData {
1753 > type: ChatAgentLocation.Terminal;
1754 > // TBD
1755 > }
1756 >
1757 > export type IChatLocationData = IChatEditorLocationData | IChatNotebookLocationData | IChatTerminalLocationData;
1758 >
1759 > /**
1760 > * The kind of queue request.
1761 > */
1762 > export const enum ChatRequestQueueKind {
1763 > /** Request is queued to be sent after current request completes */
1764 > Queued = 'queued',
1765 > /** Request is queued and signals the active request to yield */
1766 > Steering = 'steering'
1767 > }
1768 >
1769 > /**
1770 > * A queued or steering message that was authored outside of this client, e.g.
1771 > * by another window connected to the same server-managed session.
1772 > */
1773 > export interface IRemotePendingRequest {
1774 > /** Stable id of the message, as known by the server. */
1775 > readonly id: string;
1776 > readonly kind: ChatRequestQueueKind;
1777 > /** The raw message text. */
1778 > readonly message: string;
1779 > readonly variableData?: IChatRequestVariableData;
1780 > readonly timestamp?: number;
1781 > }
1782 >
1783 > export interface IChatSendRequestOptions {
1784 > modeInfo?: IChatRequestModeInfo;
1785 > userSelectedModelId?: string;
1786 > /**
1787 > * The configuration (e.g. context size, thinking effort) for the selected
1788 > * model as scoped to the requesting editor. When set, it takes precedence
1789 > * over the global per-model configuration so the value sent matches what the
1790 > * editor displays. See issue #320393.
1791 > */
1792 > userSelectedModelConfiguration?: IStringDictionary<unknown>;
1793 > userSelectedTools?: IObservable<UserSelectedTools>;
1794 > location?: ChatAgentLocation;
1795 > locationData?: IChatLocationData;
1796 > parserContext?: IChatParserContext;
1797 > attempt?: number;
1798 > noCommandDetection?: boolean;
1799 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
1800 > acceptedConfirmationData?: any[];
1801 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
1802 > rejectedConfirmationData?: any[];
1803 > attachedContext?: IChatRequestVariableEntry[];
1804 > resolvedVariables?: IChatRequestVariableEntry[];
1805 > agentHostSessionConfig?: Record<string, unknown>;
1806 >
1807 > /** The target agent ID can be specified with this property instead of using @ in 'message' */
1808 > agentId?: string;
1809 > /** agentId, but will not add a @ name to the request */
1810 > agentIdSilent?: string;
1811 > slashCommand?: string;
1812 >
1813 > /**
1814 > * The label of the confirmation action that was selected.
1815 > */
1816 > confirmation?: string;
1817 >
1818 > /**
1819 > * When set, queues this message to be sent after the current request completes.
1820 > * If Steering, also sets yieldRequested on any active request to signal it should wrap up.
1821 > */
1822 > queue?: ChatRequestQueueKind;
1823 >
1824 > /**
1825 > * When true, the queued request will not be processed immediately even if no request is active.
1826 > * The request stays in the queue until `processPendingRequests` is called explicitly.
1827 > */
1828 > pauseQueue?: boolean;
1829 >
1830 > /**
1831 > * When true, the request is rendered as a compact tool-progress-style line
1832 > * instead of a full user message bubble. Used for system-initiated notifications
1833 > * such as terminal command completion.
1834 > */
1835 > isSystemInitiated?: boolean;
1836 >
1837 > /**
1838 > * Display label for system-initiated requests. When set, the request row renders
1839 > * this label as a compact progress-style message instead of the full request text.
1840 > */
1841 > systemInitiatedLabel?: string;
1842 >
1843 > /**
1844 > * Structured terminal execution ID for system-initiated terminal notifications.
1845 > * This avoids parsing IDs from request text when tools need to correlate
1846 > * terminal prompts with follow-up actions.
1847 > */
1848 > terminalExecutionId?: string;
1849 >
1850 > /**
1851 > * When set, the chat service will collect automatic instructions
1852 > * (for example `.instructions.md` files and skills) asynchronously after showing
1853 > * the request in the UI, rather than blocking the UI on collection.
1854 > */
1855 > instructionContext?: {
1856 > modeKind: ChatModeKind;
1857 > enabledTools?: UserSelectedTools;
1858 > enabledSubAgents?: readonly string[];
1859 > };
1860 > }
1861 >
1862 > export type IChatModelReference = IReference<IChatModel>;
1863 >
1864 > export const IChatService = createDecorator<IChatService>('IChatService');
1865 >
1866 > export interface IChatService {
1867 > _serviceBrand: undefined;
1868 > transferredSessionResource: URI | undefined;
1869 >
1870 > readonly onDidSubmitRequest: Event<{ readonly chatSessionResource: URI; readonly message?: IParsedChatRequest }>;
1871 >
1872 > readonly onDidCreateModel: Event<IChatModel>;
1873 >
1874 > /**
1875 > * An observable containing all live chat models.
1876 > */
1877 > readonly chatModels: IObservable<Iterable<IChatModel>>;
1878 >
1879 > readonly editingSessions: readonly IChatEditingSession[];
1880 >
1881 > isEnabled(location: ChatAgentLocation): boolean;
1882 >
1883 > hasSessions(): boolean;
1884 >
1885 > /**
1886 > * Starts a new chat session at the given location.
1887 > *
1888 > * @returns A reference to the session's model.
1889 > */
1890 > startNewLocalSession(location: ChatAgentLocation, options?: IChatSessionStartOptions): IChatModelReference;
1891 >
1892 > /**
1893 > * Get an active session without holding a reference to it.
1894 > *
1895 > * @returns The session's model, or undefined if no active session exists.
1896 > */
1897 > getSession(sessionResource: URI): IChatModel | undefined;
1898 >
1899 > /**
1900 > * Acquire a reference to an active session.
1901 > *
1902 > * @returns A reference to the session's model or undefined if there is no active session for the given resource.
1903 > */
1904 > acquireExistingSession(sessionResource: URI, debugOwner?: string): IChatModelReference | undefined;
1905 >
1906 > /**
1907 > * Tries to acquire an existing a chat session for the resource. If no session exists, tries to load one for the given
1908 > * session resource and location. This may load the session from an external provider.
1909 > *
1910 > * @returns A reference to the session's model, or undefined if the session could not be loaded
1911 > */
1912 > acquireOrLoadSession(sessionResource: URI, location: ChatAgentLocation, token: CancellationToken, debugOwner?: string): Promise<IChatModelReference | undefined>;
1913 >
1914 > /**
1915 > * Loads a session from exported chat data
1916 > */
1917 > loadSessionFromData(data: IExportableChatData | ISerializableChatData, debugOwner?: string): IChatModelReference;
1918 >
1919 > getChatModelReferenceDebugInfo(): IChatModelReferenceDebugSnapshot;
1920 >
1921 > /**
1922 > * Sends a chat request for the given session.
1923 > * @returns A result indicating whether the request was sent, queued, or rejected.
1924 > */
1925 > sendRequest(sessionResource: URI, message: string, options?: IChatSendRequestOptions): Promise<ChatSendResult>;
1926 >
1927 > getSessionTitle(sessionResource: URI): string | undefined;
1928 > setSessionTitle(sessionResource: URI, title: string): void;
1929 >
1930 > appendProgress(request: IChatRequestModel, progress: IChatProgress): void;
1931 > resendRequest(request: IChatRequestModel, options?: IChatSendRequestOptions): Promise<void>;
1932 > adoptRequest(sessionResource: URI, request: IChatRequestModel): Promise<void>;
1933 > removeRequest(sessionResource: URI, requestId: string): Promise<void>;
1934 > cancelCurrentRequestForSession(sessionResource: URI, source?: string): Promise<void>;
1935 > /**
1936 > * Migrates all in-flight and queued pending requests from one session to another.
1937 > * Cancels the in-flight request on the original session, removes queued requests,
1938 > * and re-sends them all on the target session preserving their original send options.
1939 > */
1940 > migrateRequests(originalResource: URI, targetResource: URI): void;
1941 > /**
1942 > * Sets yieldRequested on the active request for the given session.
1943 > */
1944 > setYieldRequested(sessionResource: URI): void;
1945 > /**
1946 > * Removes a pending request from the session's queue.
1947 > */
1948 > removePendingRequest(sessionResource: URI, requestId: string): void;
1949 > /**
1950 > * Sets the pending requests for a session, allowing for deletions/reordering.
1951 > * Adding new requests should go through sendRequest with the queue option.
1952 > */
1953 > setPendingRequests(sessionResource: URI, requests: readonly { requestId: string; kind: ChatRequestQueueKind }[]): void;
1954 > /**
1955 > * Atomically reconciles the pending queue with messages authored by another client.
1956 > * Preserves remote ids and existing matching requests, and no-ops when already equal.
1957 > */
1958 > syncPendingRequestsFromRemote(sessionResource: URI, requests: readonly IRemotePendingRequest[]): void;
1959 > /**
1960 > * Ensures pending requests for the session are processing. If restoring from
1961 > * storage or after an error, pending requests may be present without an
1962 > * active chat message 'loop' happening. THis triggers the loop to happen
1963 > * as needed. Idempotent, safe to call at any time.
1964 > */
1965 > processPendingRequests(sessionResource: URI): void;
1966 > /**
1967 > * Cancels the in-flight request and immediately sends a single pending
1968 > * (queued or steering) request. Local sessions move it to the front and
1969 > * dequeue it; server-managed (agent host) sessions re-send it as a normal
1970 > * turn, since the server does not drain its queue on cancellation.
1971 > */
1972 > sendPendingRequestImmediately(sessionResource: URI, requestId: string): Promise<void>;
1973 > addCompleteRequest(sessionResource: URI, message: IParsedChatRequest | string, variableData: IChatRequestVariableData | undefined, attempt: number | undefined, response: IChatCompleteResponse): void;
1974 > setChatSessionTitle(sessionResource: URI, title: string): void;
1975 > getLocalSessionHistory(): Promise<IChatDetail[]>;
1976 > clearAllHistoryEntries(): Promise<void>;
1977 > removeHistoryEntry(sessionResource: URI): Promise<void>;
1978 > getChatStorageFolder(): URI;
1979 > logChatIndex(): void;
1980 > getLiveSessionItems(): Promise<IChatDetail[]>;
1981 > getHistorySessionItems(): Promise<IChatDetail[]>;
1982 > getMetadataForSession(sessionResource: URI): Promise<IChatDetail | undefined>;
1983 >
1984 > readonly onDidPerformUserAction: Event<IChatUserActionEvent>;
1985 > notifyUserAction(event: IChatUserActionEvent): void;
1986 >
1987 > readonly onDidReceiveQuestionCarouselAnswer: Event<{ requestId: string; resolveId: string; answers: IChatQuestionAnswers | undefined }>;
1988 > notifyQuestionCarouselAnswer(requestId: string, resolveId: string, answers: IChatQuestionAnswers | undefined): void;
1989 >
1990 > readonly onDidDisposeSession: Event<{ readonly sessionResources: readonly URI[]; readonly reason: 'cleared' }>;
1991 >
1992 > transferChatSession(transferredSessionResource: URI, toWorkspace: URI): Promise<void>;
1993 >
1994 > activateDefaultAgent(location: ChatAgentLocation): Promise<void>;
1995 >
1996 > readonly requestInProgressObs: IObservable<boolean>;
1997 >
1998 > /**
1999 > * For tests only!
2000 > */
2001 > setSaveModelsEnabled(enabled: boolean): void;
2002 >
2003 > /**
2004 > * For tests only!
2005 > */
2006 > waitForModelDisposals(): Promise<void>;
2007 > }
2008 >
2009 > export interface IChatSessionContext {
2010 > readonly initialSessionOptions?: ReadonlyChatSessionOptionsMap;
2011 > }
2012 >
2013 > export const KEYWORD_ACTIVIATION_SETTING_ID = 'accessibility.voice.keywordActivation';
2014 >
2015 > export interface IChatSessionStartOptions {
2016 > canUseTools?: boolean;
2017 > disableBackgroundKeepAlive?: boolean;
2018 > debugOwner?: string;
2019 > }
2020 >
2021 > export const ChatStopCancellationNoopEventName = 'chat.stopCancellationNoop';
2022 >
2023 > export type ChatStopCancellationNoopEvent = {
2024 > source: string;
2025 > reason: 'noWidget' | 'noViewModel' | 'noPendingRequest' | 'requestAlreadyCanceled' | 'requestIdUnavailable';
2026 > requestInProgress: 'true' | 'false' | 'unknown';
2027 > pendingRequests: number;
2028 > sessionScheme?: string;
2029 > lastRequestId?: string;
2030 > chatSessionId?: string;
2031 > };
2032 >
2033 > export type ChatStopCancellationNoopClassification = {
2034 > source: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The layer where stop cancellation no-op occurred.' };
2035 > reason: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The no-op reason when stop cancellation did not dispatch fully.' };
2036 > requestInProgress: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether request-in-progress was true, false, or unknown at no-op time.' };
2037 > pendingRequests: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The number of queued pending requests at no-op time when known.'; isMeasurement: true };
2038 > sessionScheme?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The URI scheme of the session resource (e.g. vscodeLocalChatSession vs remote).' };
2039 > lastRequestId?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the last request in the session, for correlating with tool invocations.' };
2040 > chatSessionId?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The chat session ID.' };
2041 > owner: 'roblourens';
2042 > comment: 'Tracks possible no-op stop cancellation paths.';
2043 > };
2044 >
2045 > export const ChatPendingRequestChangeEventName = 'chat.pendingRequestChange';
2046 >
2047 > export type ChatPendingRequestChangeEvent = {
2048 > action: 'add' | 'remove' | 'notCancelable';
2049 > source: string;
2050 > requestId?: string;
2051 > chatSessionId?: string;
2052 > };
2053 >
2054 > export type ChatPendingRequestChangeClassification = {
2055 > action: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether a pending request was added or removed.' };
2056 > source: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The method that triggered the pending request change.' };
2057 > requestId?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The request ID associated with the pending request change.' };
2058 > chatSessionId?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The chat session ID.' };
2059 > owner: 'roblourens';
2060 > comment: 'Tracks pending request lifecycle changes in the chat service.';
2061 > };
src/vs/workbench/contrib/chat/common/model/chatModel.ts 1802 covered LOC · 249 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatModel.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { asArray } from '../../../../../base/common/arrays.js';
7 > import { softAssertNever } from '../../../../../base/common/assert.js';
8 > import { VSBuffer, decodeHex, encodeHex } from '../../../../../base/common/buffer.js';
9 > import { IStringDictionary } from '../../../../../base/common/collections.js';
10 > import { BugIndicatingError } from '../../../../../base/common/errors.js';
11 > import { Emitter, Event } from '../../../../../base/common/event.js';
12 > import { IMarkdownString, MarkdownString, isMarkdownString } from '../../../../../base/common/htmlContent.js';
13 > import { Disposable, DisposableStore, IDisposable, MutableDisposable } from '../../../../../base/common/lifecycle.js';
14 > import { ResourceMap } from '../../../../../base/common/map.js';
15 > import { revive } from '../../../../../base/common/marshalling.js';
16 > import { Schemas } from '../../../../../base/common/network.js';
17 > import { equals } from '../../../../../base/common/objects.js';
18 > import { IObservable, autorun, constObservable, derived, observableFromEvent, observableSignalFromEvent, observableValue, observableValueOpts, registerAutorunSelfDisposable } from '../../../../../base/common/observable.js';
19 > import { basename, isEqual } from '../../../../../base/common/resources.js';
20 > import { hasKey, WithDefinedProps } from '../../../../../base/common/types.js';
21 > import { URI, UriDto } from '../../../../../base/common/uri.js';
22 > import { generateUuid } from '../../../../../base/common/uuid.js';
23 > import { IRange } from '../../../../../editor/common/core/range.js';
24 > import { OffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
25 > import { ISelection } from '../../../../../editor/common/core/selection.js';
26 > import { TextEdit } from '../../../../../editor/common/languages.js';
27 > import { EditSuggestionId } from '../../../../../editor/common/textModelEditSource.js';
28 > import { localize } from '../../../../../nls.js';
29 > import { canLog, ILogService, LogLevel } from '../../../../../platform/log/common/log.js';
30 > import { CellUri, ICellEditOperation } from '../../../notebook/common/notebookCommon.js';
31 > import { ChatRequestToolReferenceEntry, IChatRequestVariableEntry, isImplicitVariableEntry, isStringImplicitContextValue, isStringVariableEntry } from '../attachments/chatVariableEntries.js';
32 > import { migrateLegacyTerminalToolSpecificData } from '../chat.js';
33 > import { ChatPerfMark, markChat } from '../chatPerf.js';
34 > import { ChatAgentVoteDirection, ChatRequestQueueKind, ChatResponseClearToPreviousToolInvocationReason, ElicitationState, IChatAgentMarkdownContentWithVulnerability, IChatAutoModeResolutionPart, IChatClearToPreviousToolInvocation, IChatCodeCitation, IChatCommandButton, IChatConfirmation, IChatContentInlineReference, IChatContentReference, IChatDisabledClaudeHooksPart, IChatEditingSessionAction, IChatElicitationRequest, IChatElicitationRequestSerialized, IChatExternalEdit, IChatExternalToolInvocationUpdate, IChatExtensionsContent, IChatFollowup, IChatHookPart, IChatInfoMessage, IChatLocationData, IChatMarkdownContent, IChatMcpAuthenticationRequired, IChatMcpServersStarting, IChatMcpServersStartingSerialized, IChatMcpServersStartingSlow, IChatModelReference, IChatMultiDiffData, IChatMultiDiffDataSerialized, IChatNotebookEdit, IChatPlanReview, IChatProgress, IChatProgressMessage, IChatPullRequestContent, IChatQuestionCarousel, IChatResponseCodeblockUriPart, IChatResponseProgressFileTreeData, IChatSendRequestOptions, IChatService, IChatSessionTiming, IChatSystemNotificationPart, IChatTask, IChatTaskSerialized, IChatTextEdit, IChatThinkingPart, IChatToolInvocation, IChatToolInvocationSerialized, IChatTreeData, IChatUndoStop, IChatUsage, IChatUsagePromptTokenDetail, IChatUsedContext, IChatWarningMessage, IChatWorkspaceEdit, ResponseModelState, ToolConfirmKind, isIUsedContext } from '../chatService/chatService.js';
35 > import { ChatAgentLocation, ChatModeKind, ChatPermissionLevel } from '../constants.js';
36 > import { ChatToolInvocation } from './chatProgressTypes/chatToolInvocation.js';
37 > import { ChatPlanReviewData } from './chatProgressTypes/chatPlanReviewData.js';
38 > import { ChatQuestionCarouselData } from './chatProgressTypes/chatQuestionCarouselData.js';
39 > import { ToolDataSource, IToolData } from '../tools/languageModelToolsService.js';
40 > import { IChatEditingService, IChatEditingSession, ModifiedFileEntryState } from '../editing/chatEditingService.js';
41 > import { ILanguageModelChatMetadata, ILanguageModelChatMetadataAndIdentifier } from '../languageModels.js';
42 > import { IChatAgentCommand, IChatAgentData, IChatAgentResult, IChatAgentService, UserSelectedTools, reviveSerializedAgent } from '../participants/chatAgents.js';
43 > import { ChatRequestTextPart, IParsedChatRequest, reviveParsedChatRequest } from '../requestParser/chatParserTypes.js';
44 > import { chatSessionResourceToId, LocalChatSessionUri } from './chatUri.js';
45 > import { ObjectMutationLog } from './objectMutationLog.js';
46 >
47 >
48 > /**
49 > * Represents a queued chat request waiting to be processed.
50 > */
51 > export interface IChatPendingRequest {
52 > readonly request: IChatRequestModel;
53 > readonly kind: ChatRequestQueueKind;
54 > /**
55 > * The options that were passed to sendRequest when this request was queued.
56 > * userSelectedTools is snapshotted to a static observable at queue time.
57 > */
58 > readonly sendOptions: IChatSendRequestOptions;
59 > }
60 >
61 > /**
62 > * Serializable version of IChatSendRequestOptions for pending requests.
63 > * Excludes observables and non-serializable fields.
64 > */
65 > export interface ISerializableSendOptions {
66 > modeInfo?: IChatRequestModeInfo;
67 > userSelectedModelId?: string;
68 > userSelectedModelConfiguration?: IStringDictionary<unknown>;
69 > /** Static snapshot of user-selected tools (not an observable) */
70 > userSelectedTools?: UserSelectedTools;
71 > location?: ChatAgentLocation;
72 > locationData?: IChatLocationData;
73 > attempt?: number;
74 > noCommandDetection?: boolean;
75 > agentId?: string;
76 > agentIdSilent?: string;
77 > slashCommand?: string;
78 > confirmation?: string;
79 > isSystemInitiated?: boolean;
80 > systemInitiatedLabel?: string;
81 > terminalExecutionId?: string;
82 > }
83 >
84 > /**
85 > * Serializable representation of a pending chat request.
86 > */
87 > export interface ISerializablePendingRequestData {
88 > id: string;
89 > request: ISerializableChatRequestData;
90 > kind: ChatRequestQueueKind;
91 > sendOptions: ISerializableSendOptions;
92 > }
93 >
94 > export const CHAT_ATTACHABLE_IMAGE_MIME_TYPES: Record<string, string> = {
95 > png: 'image/png',
96 > jpg: 'image/jpeg',
97 > jpeg: 'image/jpeg',
98 > gif: 'image/gif',
99 > webp: 'image/webp',
100 > };
101 >
102 > export function getAttachableImageExtension(mimeType: string): string | undefined {
103 return Object.entries(CHAT_ATTACHABLE_IMAGE_MIME_TYPES).find(([_, value]) => value === mimeType)?.[0];
104 }
105 > chatModel.ts
106 > export interface IChatRequestVariableData {
107 > variables: readonly IChatRequestVariableEntry[];
108 > }
109 >
110 > export namespace IChatRequestVariableData {
111 > export function toExport(data: IChatRequestVariableData): IChatRequestVariableData {
112 return { variables: data.variables.map(IChatRequestVariableEntry.toExport) };
113 }
114 > } chatModel.ts
115 >
116 > export interface IChatRequestModel {
117 > readonly id: string;
118 > readonly timestamp: number;
119 > readonly requestTimestamp: number | undefined;
120 > readonly version: number;
121 > readonly modeInfo?: IChatRequestModeInfo;
122 > readonly session: IChatModel;
123 > readonly message: IParsedChatRequest;
124 > readonly attempt: number;
125 > readonly variableData: IChatRequestVariableData;
126 > readonly confirmation?: string;
127 > readonly locationData?: IChatLocationData;
128 > readonly attachedContext?: IChatRequestVariableEntry[];
129 > readonly isCompleteAddedRequest: boolean;
130 > readonly isTerminalCommand: boolean;
131 > readonly response?: IChatResponseModel;
132 > readonly editedFileEvents?: IChatAgentEditedFileEvent[];
133 > shouldBeRemovedOnSend: IChatRequestDisablement | undefined;
134 > readonly shouldBeBlocked: IObservable<boolean>;
135 > setShouldBeBlocked(value: boolean): void;
136 > readonly modelId?: string;
137 > readonly userSelectedTools?: UserSelectedTools;
138 > readonly isSystemInitiated?: boolean;
139 > readonly systemInitiatedLabel?: string;
140 > readonly terminalExecutionId?: string;
141 > }
142 >
143 > export interface ICodeBlockInfo {
144 > readonly suggestionId: EditSuggestionId;
145 > }
146 >
147 > export interface IChatTextEditGroupState {
148 > sha1: string;
149 > applied: number;
150 > }
151 >
152 > export interface IChatTextEditGroup {
153 > uri: URI;
154 > edits: TextEdit[][];
155 > state?: IChatTextEditGroupState;
156 > kind: 'textEditGroup';
157 > done: boolean | undefined;
158 > isExternalEdit?: boolean;
159 > }
160 >
161 > export function isCellTextEditOperation(value: unknown): value is ICellTextEditOperation {
162 const candidate = value as ICellTextEditOperation;
163 return !!candidate && !!candidate.edit && !!candidate.uri && URI.isUri(candidate.uri);
164 }
165 > chatModel.ts
166 > export function isCellTextEditOperationArray(value: ICellTextEditOperation[] | ICellEditOperation[]): value is ICellTextEditOperation[] {
167 return value.some(isCellTextEditOperation);
168 }
169 > chatModel.ts
170 > export interface ICellTextEditOperation {
171 > edit: TextEdit;
172 > uri: URI;
173 > }
174 >
175 > export interface IChatNotebookEditGroup {
176 > uri: URI;
177 > edits: (ICellTextEditOperation[] | ICellEditOperation[])[];
178 > state?: IChatTextEditGroupState;
179 > kind: 'notebookEditGroup';
180 > done: boolean | undefined;
181 > isExternalEdit?: boolean;
182 > }
183 >
184 > /**
185 > * Progress kinds that are included in the history of a response.
186 > * Excludes "internal" types that are included in history.
187 > */
188 > export type IChatProgressHistoryResponseContent =
189 > | IChatMarkdownContent
190 > | IChatAgentMarkdownContentWithVulnerability
191 > | IChatResponseCodeblockUriPart
192 > | IChatTreeData
193 > | IChatMultiDiffDataSerialized
194 > | IChatContentInlineReference
195 > | IChatProgressMessage
196 > | IChatSystemNotificationPart
197 > | IChatCommandButton
198 > | IChatWarningMessage
199 > | IChatInfoMessage
200 > | IChatTask
201 > | IChatTaskSerialized
202 > | IChatTextEditGroup
203 > | IChatNotebookEditGroup
204 > | IChatConfirmation
205 > | IChatQuestionCarousel
206 > | IChatPlanReview
207 > | IChatExtensionsContent
208 > | IChatThinkingPart
209 > | IChatHookPart
210 > | IChatPullRequestContent
211 > | IChatWorkspaceEdit
212 > | IChatExternalEdit
213 > | IChatAutoModeResolutionPart;
214 >
215 > /**
216 > * "Normal" progress kinds that are rendered as parts of the stream of content.
217 > */
218 > export type IChatProgressResponseContent =
219 > | IChatProgressHistoryResponseContent
220 > | IChatToolInvocation
221 > | IChatToolInvocationSerialized
222 > | IChatMultiDiffData
223 > | IChatUndoStop
224 > | IChatElicitationRequest
225 > | IChatElicitationRequestSerialized
226 > | IChatClearToPreviousToolInvocation
227 > | IChatMcpServersStarting
228 > | IChatMcpServersStartingSerialized
229 > | IChatMcpAuthenticationRequired
230 > | IChatMcpServersStartingSlow
231 > | IChatDisabledClaudeHooksPart;
232 >
233 > export type IChatProgressResponseContentSerialized = Exclude<IChatProgressResponseContent,
234 > | IChatToolInvocation
235 > | IChatElicitationRequest
236 > | IChatTask
237 > | IChatMultiDiffData
238 > | IChatMcpServersStarting
239 > | IChatMcpAuthenticationRequired
240 > | IChatMcpServersStartingSlow
241 > | IChatDisabledClaudeHooksPart
242 > >;
243 >
244 > const nonHistoryKinds = new Set(['toolInvocation', 'toolInvocationSerialized', 'undoStop']);
245 function isChatProgressHistoryResponseContent(content: IChatProgressResponseContent): content is IChatProgressHistoryResponseContent {
246 return !nonHistoryKinds.has(content.kind);
247 }
248 > chatModel.ts
249 > export function toChatHistoryContent(content: ReadonlyArray<IChatProgressResponseContent>): IChatProgressHistoryResponseContent[] {
250 return content.filter(isChatProgressHistoryResponseContent);
251 }
252 > chatModel.ts
253 > export type IChatProgressRenderableResponseContent = Exclude<IChatProgressResponseContent, IChatContentInlineReference | IChatAgentMarkdownContentWithVulnerability | IChatResponseCodeblockUriPart>;
254 >
255 > export interface IResponse {
256 > readonly value: ReadonlyArray<IChatProgressResponseContent>;
257 > getMarkdown(): string;
258 > getFinalResponse(): string;
259 > toString(): string;
260 > }
261 >
262 > export interface IChatResponseModel {
263 > readonly onDidChange: Event<ChatResponseModelChangeReason>;
264 > readonly id: string;
265 > readonly requestId: string;
266 > readonly request: IChatRequestModel | undefined;
267 > readonly username: string;
268 > readonly session: IChatModel;
269 > readonly agent?: IChatAgentData;
270 > readonly usedContext: IChatUsedContext | undefined;
271 > readonly contentReferences: ReadonlyArray<IChatContentReference>;
272 > readonly codeCitations: ReadonlyArray<IChatCodeCitation>;
273 > readonly progressMessages: ReadonlyArray<IChatProgressMessage>;
274 > readonly slashCommand?: IChatAgentCommand;
275 > readonly agentOrSlashCommandDetected: boolean;
276 > /** View of the response shown to the user, may have parts omitted from undo stops. */
277 > readonly response: IResponse;
278 > /** Entire response from the model. */
279 > readonly entireResponse: IResponse;
280 > /** Milliseconds timestamp when this chat response was created. */
281 > readonly timestamp: number;
282 > /** Milliseconds timestamp when this chat response was completed or cancelled. */
283 > readonly completedAt?: number;
284 > /** Known completion timestamp for display. Undefined for legacy responses whose completion time was synthesized during restore. */
285 > readonly completionTimestamp?: number;
286 > /** The state of this response */
287 > readonly state: ResponseModelState;
288 > /** @internal */
289 > readonly stateT: ResponseModelStateT;
290 > /**
291 > * Adjusted millisecond timestamp that excludes the duration during which
292 > * the model was pending user confirmation. `Date.now() - confirmationAdjustedTimestamp`
293 > * will return the amount of time the response was busy generating content.
294 > * This is updated only when `isPendingConfirmation` changes state.
295 > */
296 > readonly confirmationAdjustedTimestamp: IObservable<number>;
297 > readonly isComplete: boolean;
298 > readonly isCanceled: boolean;
299 > readonly isPendingConfirmation: IObservable<{ startedWaitingAt: number; detail?: string } | undefined>;
300 > readonly isInProgress: IObservable<boolean>;
301 > /**
302 > * True whenever this response has not reached a terminal state yet.
303 > * Unlike {@link isInProgress}, this remains true during tool confirmations,
304 > * elicitations, and any other intermediate state.
305 > */
306 > readonly isIncomplete: IObservable<boolean>;
307 > readonly shouldBeRemovedOnSend: IChatRequestDisablement | undefined;
308 > readonly shouldBeBlocked: IObservable<boolean>;
309 > readonly isCompleteAddedRequest: boolean;
310 > /** A stale response is one that has been persisted and rehydrated, so e.g. Commands that have their arguments stored in the EH are gone. */
311 > readonly isStale: boolean;
312 > readonly vote: ChatAgentVoteDirection | undefined;
313 > readonly followups?: IChatFollowup[] | undefined;
314 > readonly result?: IChatAgentResult;
315 > readonly usage?: IChatUsage;
316 > readonly usageObs: IObservable<IChatUsage | undefined>;
317 > readonly completionTokenCount: number | undefined;
318 > readonly completionTokenCountObs: IObservable<number | undefined>;
319 > /** Elapsed generation time in ms (excluding confirmation waits). Set on completion and serialized. */
320 > readonly elapsedMs: number | undefined;
321 > readonly codeBlockInfos: ICodeBlockInfo[] | undefined;
322 >
323 > initializeCodeBlockInfos(codeBlockInfo: ICodeBlockInfo[]): void;
324 > addUndoStop(undoStop: IChatUndoStop): void;
325 > setVote(vote: ChatAgentVoteDirection): void;
326 > setUsage(usage: IChatUsage): void;
327 > setElapsedMs(elapsedMs: number): void;
328 > setEditApplied(edit: IChatTextEditGroup, editCount: number): boolean;
329 > resolveInlineReference(resolveId: string, resolvedReference: IChatContentInlineReference): boolean;
330 > updateContent(progress: IChatProgressResponseContent | IChatTextEdit | IChatNotebookEdit | IChatTask | IChatExternalToolInvocationUpdate, quiet?: boolean): void;
331 > /**
332 > * Adopts any partially-undo {@link response} as the {@link entireResponse}.
333 > * Only valid when {@link isComplete}. This is needed because otherwise an
334 > * undone and then diverged state would start showing old data because the
335 > * undo stops would no longer exist in the model.
336 > */
337 > finalizeUndoState(): void;
338 > }
339 >
340 > export type ChatResponseModelChangeReason =
341 > | { reason: 'other' }
342 > | { reason: 'completedRequest' }
343 > | { reason: 'undoStop'; id: string };
344 >
345 > export const defaultChatResponseModelChangeReason: ChatResponseModelChangeReason = { reason: 'other' };
346 >
347 > export interface IChatRequestModeInfo {
348 > kind: ChatModeKind | undefined; // is undefined in case of telemetryModeId === 'applyCodeBlock'
349 > isBuiltin: boolean;
350 > modeInstructions: IChatRequestModeInstructions | undefined;
351 > telemetryModeId: 'ask' | 'agent' | 'edit' | 'custom' | 'applyCodeBlock' | undefined;
352 > telemetryModeName?: string;
353 > applyCodeBlockSuggestionId: EditSuggestionId | undefined;
354 > permissionLevel?: ChatPermissionLevel;
355 > }
356 >
357 > export interface IChatRequestModeInstructions {
358 > readonly uri?: URI;
359 > readonly name: string;
360 > readonly content: string;
361 > readonly toolReferences: readonly ChatRequestToolReferenceEntry[];
362 > readonly allowedSubagents?: readonly string[];
363 > readonly metadata?: Record<string, boolean | string | number>;
364 > readonly isBuiltin?: boolean;
365 > }
366 >
367 > export interface IChatRequestModelParameters {
368 > session: ChatModel;
369 > message: IParsedChatRequest;
370 > variableData: IChatRequestVariableData;
371 > timestamp?: number;
372 > fallbackTimestamp?: number;
373 > attempt?: number;
374 > modeInfo?: IChatRequestModeInfo;
375 > confirmation?: string;
376 > locationData?: IChatLocationData;
377 > attachedContext?: IChatRequestVariableEntry[];
378 > isCompleteAddedRequest?: boolean;
379 > modelId?: string;
380 > restoredId?: string;
381 > editedFileEvents?: IChatAgentEditedFileEvent[];
382 > userSelectedTools?: UserSelectedTools;
383 > isSystemInitiated?: boolean;
384 > systemInitiatedLabel?: string;
385 > terminalExecutionId?: string;
386 > /** Whether this request runs as a terminal command (agent host `!` prefix). */
387 > isTerminalCommand?: boolean;
388 > }
389 >
390 > export class ChatRequestModel implements IChatRequestModel {
391 > public readonly id: string;
392 > public response: ChatResponseModel | undefined;
393 > public shouldBeRemovedOnSend: IChatRequestDisablement | undefined;
394 > public readonly timestamp: number;
395 > public readonly requestTimestamp: number | undefined;
396 > public readonly message: IParsedChatRequest;
397 > public readonly isCompleteAddedRequest: boolean;
398 > public readonly modelId?: string;
399 > public readonly modeInfo?: IChatRequestModeInfo;
400 > public readonly userSelectedTools?: UserSelectedTools;
401 > public readonly isSystemInitiated?: boolean;
402 > public readonly systemInitiatedLabel?: string;
403 > public readonly terminalExecutionId?: string;
404 > public readonly isTerminalCommand: boolean;
405 >
406 > private readonly _shouldBeBlocked = observableValue<boolean>(this, false);
407 > public get shouldBeBlocked(): IObservable<boolean> {
408 > return this._shouldBeBlocked;
409 > }
410 >
411 > public setShouldBeBlocked(value: boolean): void {
412 this._shouldBeBlocked.set(value, undefined);
413 }
414 > chatModel.ts
415 > private _session: ChatModel;
416 > private readonly _attempt: number;
417 > private _variableData: IChatRequestVariableData;
418 > private readonly _confirmation?: string;
419 > private readonly _locationData?: IChatLocationData;
420 > private readonly _attachedContext?: IChatRequestVariableEntry[];
421 > private readonly _editedFileEvents?: IChatAgentEditedFileEvent[];
422 >
423 > public get session(): ChatModel {
424 return this._session;
425 }
426 > chatModel.ts
427 > public get attempt(): number {
428 return this._attempt;
429 }
430 > chatModel.ts
431 > public get variableData(): IChatRequestVariableData {
432 return this._variableData;
433 }
434 > chatModel.ts
435 > public set variableData(v: IChatRequestVariableData) {
436 this._version++;
437 this._variableData = v;
438 }
439 > chatModel.ts
440 > public get confirmation(): string | undefined {
441 return this._confirmation;
442 }
443 > chatModel.ts
444 > public get locationData(): IChatLocationData | undefined {
445 return this._locationData;
446 }
447 > chatModel.ts
448 > public get attachedContext(): IChatRequestVariableEntry[] | undefined {
449 return this._attachedContext;
450 }
451 > chatModel.ts
452 > public get editedFileEvents(): IChatAgentEditedFileEvent[] | undefined {
453 return this._editedFileEvents;
454 }
455 > chatModel.ts
456 > private _version = 0;
457 > public get version(): number {
458 return this._version;
459 }
460 > chatModel.ts
461 > constructor(params: IChatRequestModelParameters) {
462 > this._session = params.session; chatModel.ts
463 > this.message = params.message;
464 > this._variableData = params.variableData;
465 > this.requestTimestamp = params.timestamp;
466 > this.timestamp = params.timestamp ?? params.fallbackTimestamp ?? Date.now();
467 > this._attempt = params.attempt ?? 0;
468 > this.modeInfo = params.modeInfo;
469 > this._confirmation = params.confirmation;
470 > this._locationData = params.locationData;
471 > this._attachedContext = params.attachedContext;
472 > this.isCompleteAddedRequest = params.isCompleteAddedRequest ?? false;
473 > this.modelId = params.modelId;
474 > this.id = params.restoredId ?? 'request_' + generateUuid();
475 > this._editedFileEvents = params.editedFileEvents;
476 > this.userSelectedTools = params.userSelectedTools;
477 > this.isSystemInitiated = params.isSystemInitiated;
478 > this.systemInitiatedLabel = params.systemInitiatedLabel;
479 > this.terminalExecutionId = params.terminalExecutionId;
480 > this.isTerminalCommand = params.isTerminalCommand ?? false;
481 > }
482 > chatModel.ts
483 > adoptTo(session: ChatModel) {
484 this._session = session;
485 }
486 > } chatModel.ts
487 >
488 > class AbstractResponse implements IResponse {
489 > protected _responseParts: IChatProgressResponseContent[];
490 >
491 > /**
492 > * A stringified representation of response data which might be presented to a screenreader or used when copying a response.
493 > * Computed lazily on demand to avoid expensive string rebuilding during streaming.
494 > */
495 > private _responseRepr: string | undefined;
496 >
497 > /**
498 > * Just the markdown content of the response, used for determining the rendering rate of markdown.
499 > * Computed lazily on demand to avoid expensive string rebuilding during streaming.
500 > */
501 > private _markdownContent: string | undefined;
502 >
503 > get value(): IChatProgressResponseContent[] {
504 > return this._responseParts; chatModel.ts
505 > }
506 > chatModel.ts
507 > constructor(value: IChatProgressResponseContent[]) {
508 > this._responseParts = value; chatModel.ts
509 > }
510 > chatModel.ts
511 > toString(): string {
512 if (this._responseRepr === undefined) {
513 this._responseRepr = this.computeRepr();
515 return this._responseRepr;
516 }
517 > chatModel.ts
518 > /**
519 > * _Just_ the content of markdown parts in the response
520 > */
521 > getMarkdown(): string {
522 if (this._markdownContent === undefined) {
523 this._markdownContent = this.computeMarkdownContent();
525 return this._markdownContent;
526 }
527 > chatModel.ts
528 > /**
529 > * The trailing contiguous markdown/inline-reference content of the response,
530 > * skipping any trailing tool calls or empty markdown parts.
531 > */
532 > getFinalResponse(): string {
533 const parts = this._responseParts;
534 // Walk backwards to find where the last contiguous markdown block starts.
577 return segments.join('');
578 }
579 > chatModel.ts
580 > /**
581 > * Invalidate cached representations so they are recomputed on next access.
582 > */
583 > protected _invalidateRepr() {
584 > this._responseRepr = undefined; chatModel.ts
585 > this._markdownContent = undefined;
586 > }
587 > chatModel.ts
588 > private computeMarkdownContent(): string {
589 const segments: string[] = [];
590 for (const part of this._responseParts) {
599 return segments.join('');
600 }
601 > chatModel.ts
602 > protected computeRepr(): string {
603 return this.partsToRepr(this._responseParts);
604 }
605 > chatModel.ts
606 > private partsToRepr(parts: readonly IChatProgressResponseContent[]): string {
607 const blocks: string[] = [];
608 let currentBlockSegments: string[] = [];
702 return blocks.join('\n\n');
703 }
704 > chatModel.ts
705 > private inlineRefToRepr(part: IChatContentInlineReference) {
706 if ('uri' in part.inlineReference) {
707 return this.uriToRepr(part.inlineReference.uri);
712 : this.uriToRepr(part.inlineReference);
713 }
714 > chatModel.ts
715 > private getToolInvocationText(toolInvocation: IChatToolInvocation | IChatToolInvocationSerialized): { text: string; isBlock?: boolean } {
716 const getTerminalDisplayInput = (terminalData: ReturnType<typeof migrateLegacyTerminalToolSpecificData>) => terminalData.presentationOverrides?.commandLine
717 ?? terminalData.commandLine.forDisplay
763 return { text, isBlock: true };
764 }
765 > chatModel.ts
766 > private uriToRepr(uri: URI): string {
767 if (uri.scheme === Schemas.http || uri.scheme === Schemas.https) {
768 return uri.toString(false);
771 return basename(uri);
772 }
773 > } chatModel.ts
774 >
775 > /** A view of a subset of a response */
776 > class ResponseView extends AbstractResponse {
777 > constructor(
778 _response: IResponse,
779 public readonly undoStop: string,
789 super(idx === -1 ? _response.value.slice() : _response.value.slice(0, idx));
790 }
791 > } chatModel.ts
792 >
793 > export class Response extends AbstractResponse implements IDisposable {
794 > private readonly _store = new DisposableStore();
795 > private _onDidChangeValue = this._store.add(new Emitter<void>());
796 > private _activeReasoning: { part: IChatThinkingPart; startedAt: number } | undefined;
797 > public get onDidChangeValue() {
798 > return this._onDidChangeValue.event;
799 > }
800 >
801 > private _citations: IChatCodeCitation[] = [];
802 >
803 >
804 > constructor(value: IMarkdownString | ReadonlyArray<SerializedChatResponsePart>) {
805 > super(asArray(value).map((v) => ( chatModel.ts
806 'kind' in v ? v :
807 isMarkdownString(v) ? { content: v, kind: 'markdownContent' } satisfies IChatMarkdownContent :
808 { kind: 'treeData', treeData: v }
809 > ))); chatModel.ts
810 > }
811 > chatModel.ts
812 > dispose(): void {
813 > this._store.dispose(); chatModel.ts
814 > }
815 > chatModel.ts
816 >
817 > clear(): void {
818 > this.finalizeReasoningDuration(); chatModel.ts
819 > this._responseParts = [];
820 > this._contentChanged(true);
821 > }
822 > chatModel.ts
823 > clearToPreviousToolInvocation(message?: string): void {
824 this.finalizeReasoningDuration();
825 // look through the response parts and find the last tool invocation, then slice the response parts to that point
842 this._contentChanged(true);
843 }
844 > chatModel.ts
845 > updateContent(progress: IChatProgressResponseContent | IChatTextEdit | IChatNotebookEdit | IChatTask | IChatExternalToolInvocationUpdate, quiet?: boolean): void {
846 if (progress.kind !== 'thinking') {
847 this.finalizeReasoningDuration();
968 }
969 }
970 > chatModel.ts
971 > /**
972 > * Persists the duration of the active reasoning interval.
973 > */
974 > finalizeReasoningDuration(): void {
975 > if (!this._activeReasoning) { chatModel.ts
976 > return; chatModel.ts
977 > }
978
979 this._activeReasoning.part.reasoningDurationMs = Math.max(0, Date.now() - this._activeReasoning.startedAt);
980 this._activeReasoning = undefined;
981 > } chatModel.ts
982 > chatModel.ts
983 > public addCitation(citation: IChatCodeCitation) {
984 this._citations.push(citation);
985 this._contentChanged();
986 }
987 > chatModel.ts
988 > public resolveInlineReference(resolveId: string, resolvedReference: IChatContentInlineReference): boolean {
989 for (let i = 0; i < this._responseParts.length; i++) {
990 const current = this._responseParts[i];
1004 return false;
1005 }
1006 > chatModel.ts
1007 > private _mergeOrPushTextEditGroup(uri: URI, edits: TextEdit[], done: boolean | undefined, isExternalEdit: boolean | undefined): void {
1008 for (const candidate of this._responseParts) {
1009 if (candidate.kind === 'textEditGroup' && !candidate.done && isEqual(candidate.uri, uri)) {
1015 this._responseParts.push({ kind: 'textEditGroup', uri, edits: [edits], done, isExternalEdit });
1016 }
1017 > chatModel.ts
1018 > private _mergeOrPushNotebookEditGroup(uri: URI, edits: ICellTextEditOperation[] | ICellEditOperation[], done: boolean | undefined, isExternalEdit: boolean | undefined): void {
1019 for (const candidate of this._responseParts) {
1020 if (candidate.kind === 'notebookEditGroup' && !candidate.done && isEqual(candidate.uri, uri)) {
1026 this._responseParts.push({ kind: 'notebookEditGroup', uri, edits: [edits], done, isExternalEdit });
1027 }
1028 > chatModel.ts
1029 > private _handleExternalToolInvocationUpdate(progress: IChatExternalToolInvocationUpdate): void {
1030 // Look for existing invocation in the response parts
1031 const existingInvocation = this._responseParts.findLast(
1085 this._responseParts.push(invocation);
1086 }
1087 > chatModel.ts
1088 > protected override computeRepr(): string {
1089 let repr = super.computeRepr();
1090 if (this._citations.length) {
1093 return repr;
1094 }
1095 > chatModel.ts
1096 > private _contentChanged(quiet?: boolean) {
1097 > this._invalidateRepr(); chatModel.ts
1098 > if (!quiet) {
1099 this._onDidChangeValue.fire();
1100 }
1101 > } chatModel.ts
1102 > } chatModel.ts
1103 >
1104 > export interface IChatResponseModelParameters {
1105 > responseContent: IMarkdownString | ReadonlyArray<SerializedChatResponsePart>;
1106 > session: ChatModel;
1107 > agent?: IChatAgentData;
1108 > slashCommand?: IChatAgentCommand;
1109 > requestId: string;
1110 > timestamp?: number;
1111 > vote?: ChatAgentVoteDirection;
1112 > result?: IChatAgentResult;
1113 > followups?: ReadonlyArray<IChatFollowup>;
1114 > isCompleteAddedRequest?: boolean;
1115 > shouldBeRemovedOnSend?: IChatRequestDisablement;
1116 > shouldBeBlocked?: boolean;
1117 > restoredId?: string;
1118 > modelState?: ResponseModelStateT;
1119 > completionTimestamp?: number | null;
1120 > timeSpentWaiting?: number;
1121 > elapsedMs?: number;
1122 > /**
1123 > * undefined means it will be set later.
1124 > */
1125 > codeBlockInfos: ICodeBlockInfo[] | undefined;
1126 > }
1127 >
1128 > export type ResponseModelStateT =
1129 > | { value: ResponseModelState.Pending }
1130 > | { value: ResponseModelState.NeedsInput }
1131 > | { value: ResponseModelState.Complete | ResponseModelState.Cancelled | ResponseModelState.Failed; completedAt: number };
1132 >
1133 > export class ChatResponseModel extends Disposable implements IChatResponseModel {
1134 > private readonly _onDidChange = this._register(new Emitter<ChatResponseModelChangeReason>());
1135 > readonly onDidChange = this._onDidChange.event;
1136 >
1137 > public readonly id: string;
1138 > public readonly requestId: string;
1139 > private _session: ChatModel;
1140 > private _agent: IChatAgentData | undefined;
1141 > private _slashCommand: IChatAgentCommand | undefined;
1142 > private _modelState = observableValue<ResponseModelStateT>(this, { value: ResponseModelState.Pending });
1143 > private _vote?: ChatAgentVoteDirection;
1144 > private _result?: IChatAgentResult;
1145 > private readonly _usageObs = observableValue<IChatUsage | undefined>(this, undefined);
1146 > private _parentUsage: IChatUsage | undefined;
1147 > private readonly _subagentCopilotCredits = new Map<string, number>();
1148 > private readonly _completionTokenCountObs = observableValue<number | undefined>(this, undefined);
1149 > private _shouldBeRemovedOnSend: IChatRequestDisablement | undefined;
1150 > public readonly isCompleteAddedRequest: boolean;
1151 > private readonly _shouldBeBlocked = observableValue<boolean>(this, false);
1152 > private readonly _timestamp: number;
1153 > private _completionTimestamp: number | undefined;
1154 > private _timeSpentWaitingAccumulator: number;
1155 > private _elapsedMs: number | undefined;
1156 >
1157 > public confirmationAdjustedTimestamp: IObservable<number>;
1158 >
1159 > public get shouldBeBlocked(): IObservable<boolean> {
1160 > return this._shouldBeBlocked;
1161 > }
1162 >
1163 > public get request(): IChatRequestModel | undefined {
1164 return this.session.getRequests().find(r => r.id === this.requestId);
1165 }
1166 > chatModel.ts
1167 > public get session() {
1168 return this._session;
1169 }
1170 > chatModel.ts
1171 > public get shouldBeRemovedOnSend() {
1172 > return this._shouldBeRemovedOnSend; chatModel.ts
1173 > }
1174 > chatModel.ts
1175 > public get isComplete(): boolean {
1176 > return this._modelState.get().value !== ResponseModelState.Pending && this._modelState.get().value !== ResponseModelState.NeedsInput; chatModel.ts
1177 > }
1178 > chatModel.ts
1179 > public get timestamp(): number {
1180 return this._timestamp;
1181 }
1182 > chatModel.ts
1183 > public set shouldBeRemovedOnSend(disablement: IChatRequestDisablement | undefined) {
1184 if (this._shouldBeRemovedOnSend === disablement) {
1185 return;
1189 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1190 }
1191 > chatModel.ts
1192 > public get isCanceled(): boolean {
1193 return this._modelState.get().value === ResponseModelState.Cancelled;
1194 }
1195 > chatModel.ts
1196 > public get completedAt(): number | undefined {
1197 const state = this._modelState.get();
1198 if (state.value === ResponseModelState.Complete || state.value === ResponseModelState.Cancelled || state.value === ResponseModelState.Failed) {
1201 return undefined;
1202 }
1203 > chatModel.ts
1204 > public get completionTimestamp(): number | undefined {
1205 return this._completionTimestamp;
1206 }
1207 > chatModel.ts
1208 > public get state(): ResponseModelState {
1209 const state = this._modelState.get().value;
1210 if (state === ResponseModelState.Complete && !!this._result?.errorDetails && this.result?.errorDetails?.code !== 'canceled') {
1215 return state;
1216 }
1217 > chatModel.ts
1218 > public get stateT(): ResponseModelStateT {
1219 return this._modelState.get();
1220 }
1221 > chatModel.ts
1222 > public get vote(): ChatAgentVoteDirection | undefined {
1223 return this._vote;
1224 }
1225 > chatModel.ts
1226 > public get followups(): IChatFollowup[] | undefined {
1227 return this._followups;
1228 }
1229 > chatModel.ts
1230 > private _response: Response;
1231 > private _finalizedResponse?: IResponse;
1232 > public get entireResponse(): IResponse {
1233 return this._finalizedResponse || this._response;
1234 }
1235 > chatModel.ts
1236 > public get result(): IChatAgentResult | undefined {
1237 return this._result;
1238 }
1239 > chatModel.ts
1240 > public get usage(): IChatUsage | undefined {
1241 return this._usageObs.get();
1242 }
1243 > chatModel.ts
1244 > public get usageObs(): IObservable<IChatUsage | undefined> {
1245 return this._usageObs;
1246 }
1247 > chatModel.ts
1248 > public get completionTokenCount(): number | undefined {
1249 return this._completionTokenCountObs.get();
1250 }
1251 > chatModel.ts
1252 > public get completionTokenCountObs(): IObservable<number | undefined> {
1253 return this._completionTokenCountObs;
1254 }
1255 > chatModel.ts
1256 > public get elapsedMs(): number | undefined {
1257 return this._elapsedMs;
1258 }
1259 > chatModel.ts
1260 > public get username(): string {
1261 return this.session.responderUsername;
1262 }
1263 > chatModel.ts
1264 > private _followups?: IChatFollowup[];
1265 >
1266 > public get agent(): IChatAgentData | undefined {
1267 return this._agent;
1268 }
1269 > chatModel.ts
1270 > public get slashCommand(): IChatAgentCommand | undefined {
1271 return this._slashCommand;
1272 }
1273 > chatModel.ts
1274 > private _agentOrSlashCommandDetected: boolean | undefined;
1275 > public get agentOrSlashCommandDetected(): boolean {
1276 return this._agentOrSlashCommandDetected ?? false;
1277 }
1278 > chatModel.ts
1279 > private _usedContext: IChatUsedContext | undefined;
1280 > public get usedContext(): IChatUsedContext | undefined {
1281 return this._usedContext;
1282 }
1283 > chatModel.ts
1284 > private readonly _contentReferences: IChatContentReference[] = [];
1285 > public get contentReferences(): ReadonlyArray<IChatContentReference> {
1286 return Array.from(this._contentReferences);
1287 }
1288 > chatModel.ts
1289 > private readonly _codeCitations: IChatCodeCitation[] = [];
1290 > public get codeCitations(): ReadonlyArray<IChatCodeCitation> {
1291 return this._codeCitations;
1292 }
1293 > chatModel.ts
1294 > private readonly _progressMessages: IChatProgressMessage[] = [];
1295 > public get progressMessages(): ReadonlyArray<IChatProgressMessage> {
1296 return this._progressMessages;
1297 }
1298 > chatModel.ts
1299 > private _isStale: boolean = false;
1300 > public get isStale(): boolean {
1301 return this._isStale;
1302 }
1303 > chatModel.ts
1304 >
1305 > readonly isPendingConfirmation: IObservable<{ startedWaitingAt: number; detail?: string } | undefined>;
1306 >
1307 > readonly isInProgress: IObservable<boolean>;
1308 >
1309 > /**
1310 > * True whenever this response has not reached a terminal state yet.
1311 > * Unlike {@link isInProgress}, this remains true during tool confirmations,
1312 > * elicitations, and any other intermediate state. It only becomes false when
1313 > * the response completes, is cancelled, or fails.
1314 > */
1315 > readonly isIncomplete: IObservable<boolean>;
1316 >
1317 > private _responseView?: ResponseView;
1318 > public get response(): IResponse {
1319 const undoStop = this._shouldBeRemovedOnSend?.afterUndoStop;
1320 if (!undoStop) {
1328 return this._responseView;
1329 }
1330 > chatModel.ts
1331 > private _codeBlockInfos: ICodeBlockInfo[] | undefined;
1332 > public get codeBlockInfos(): ICodeBlockInfo[] | undefined {
1333 return this._codeBlockInfos;
1334 }
1335 > chatModel.ts
1336 > constructor(params: IChatResponseModelParameters) {
1337 > super(); chatModel.ts
1338 >
1339 > this._session = params.session;
1340 > this._agent = params.agent;
1341 > this._slashCommand = params.slashCommand;
1342 > this.requestId = params.requestId;
1343 > this._timestamp = params.timestamp || Date.now();
1344 > if (params.modelState) {
1345 this._modelState.set(params.modelState, undefined);
1346 }
1347 > this._completionTimestamp = params.completionTimestamp === null chatModel.ts
1348 ? undefined
1349 > : params.completionTimestamp ?? (params.modelState && 'completedAt' in params.modelState ? params.modelState.completedAt : undefined); chatModel.ts
1350 > this._timeSpentWaitingAccumulator = params.timeSpentWaiting || 0;
1351 > this._elapsedMs = params.elapsedMs;
1352 > this._vote = params.vote;
1353 > this._result = params.result;
1354 > this._followups = params.followups ? [...params.followups] : undefined;
1355 > this.isCompleteAddedRequest = params.isCompleteAddedRequest ?? false;
1356 > this._shouldBeRemovedOnSend = params.shouldBeRemovedOnSend;
1357 > this._shouldBeBlocked.set(params.shouldBeBlocked ?? false, undefined);
1358 >
1359 > // If we are creating a response with some existing content, consider it stale
1360 > this._isStale = Array.isArray(params.responseContent) && (params.responseContent.length !== 0 || isMarkdownString(params.responseContent) && params.responseContent.value.length !== 0);
1361 >
1362 > this._response = this._register(new Response(params.responseContent));
1363 > this._codeBlockInfos = params.codeBlockInfos ? [...params.codeBlockInfos] : undefined;
1364 >
1365 > const signal = observableSignalFromEvent(this, this.onDidChange);
1366 >
1367 > const _pendingInfo = signal.map((_value, r): string | undefined => {
1368 > signal.read(r);
1369 >
1370 > for (const part of this._response.value) {
1371 if (part.kind === 'toolInvocation') {
1372 const state = part.state.read(r);
1396 }
1397 }
1398 > chatModel.ts
1399 > return undefined;
1400 > });
1401 >
1402 > const _startedWaitingAt = _pendingInfo.map(p => !!p).map(p => p ? Date.now() : undefined);
1403 > this.isPendingConfirmation = _startedWaitingAt.map((waiting, r) => waiting ? { startedWaitingAt: waiting, detail: _pendingInfo.read(r) } : undefined);
1404 >
1405 > this.isInProgress = signal.map((_value, r) => {
1406 >
1407 > signal.read(r);
1408 >
1409 > return !_pendingInfo.read(r)
1410 > && !this.shouldBeRemovedOnSend
1411 > && (this._modelState.read(r).value === ResponseModelState.Pending || this._modelState.read(r).value === ResponseModelState.NeedsInput);
1412 > });
1413 >
1414 > this.isIncomplete = this._modelState.map(state => {
1415 return state.value === ResponseModelState.Pending || state.value === ResponseModelState.NeedsInput;
1416 > }); chatModel.ts
1417 >
1418 > this._register(this._response.onDidChangeValue(() => this._onDidChange.fire(defaultChatResponseModelChangeReason)));
1419 > this.id = params.restoredId ?? 'response_' + generateUuid();
1420 >
1421 > let lastStartedWaitingAt: number | undefined = undefined;
1422 > this.confirmationAdjustedTimestamp = derived(reader => {
1423 > const pending = this.isPendingConfirmation.read(reader);
1424 > if (pending) {
1425 this._modelState.set({ value: ResponseModelState.NeedsInput }, undefined);
1426 if (!lastStartedWaitingAt) {
1427 lastStartedWaitingAt = pending.startedWaitingAt;
1428 }
1429 > } else if (lastStartedWaitingAt) { chatModel.ts
1430 // Restore state to Pending if it was set to NeedsInput by this observable
1431 if (this._modelState.read(reader).value === ResponseModelState.NeedsInput) {
1435 lastStartedWaitingAt = undefined;
1436 }
1437 > chatModel.ts
1438 > return this._timestamp + this._timeSpentWaitingAccumulator;
1439 > }).recomputeInitiallyAndOnChange(this._store);
1440 > }
1441 > chatModel.ts
1442 > initializeCodeBlockInfos(codeBlockInfo: ICodeBlockInfo[]): void {
1443 if (this._codeBlockInfos) {
1444 throw new BugIndicatingError('Code block infos have already been initialized');
1446 this._codeBlockInfos = [...codeBlockInfo];
1447 }
1448 > chatModel.ts
1449 > setBlockedState(isBlocked: boolean): void {
1450 this._shouldBeBlocked.set(isBlocked, undefined);
1451 }
1452 > chatModel.ts
1453 > /**
1454 > * Apply a progress update to the actual response content.
1455 > */
1456 > updateContent(responsePart: IChatProgressResponseContent | IChatTextEdit | IChatNotebookEdit | IChatExternalToolInvocationUpdate, quiet?: boolean) {
1457 this._response.updateContent(responsePart, quiet);
1458 }
1459 > chatModel.ts
1460 > resolveInlineReference(resolveId: string, resolvedReference: IChatContentInlineReference): boolean {
1461 return this._response.resolveInlineReference(resolveId, resolvedReference);
1462 }
1463 > chatModel.ts
1464 > /**
1465 > * Adds an undo stop at the current position in the stream.
1466 > */
1467 > addUndoStop(undoStop: IChatUndoStop) {
1468 this._onDidChange.fire({ reason: 'undoStop', id: undoStop.id });
1469 this._response.updateContent(undoStop, true);
1470 }
1471 > chatModel.ts
1472 > /**
1473 > * Apply one of the progress updates that are not part of the actual response content.
1474 > */
1475 > applyReference(progress: IChatUsedContext | IChatContentReference) {
1476 if (progress.kind === 'usedContext') {
1477 this._usedContext = progress;
1481 }
1482 }
1483 > chatModel.ts
1484 > applyCodeCitation(progress: IChatCodeCitation) {
1485 this._codeCitations.push(progress);
1486 this._response.addCitation(progress);
1487 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1488 }
1489 > chatModel.ts
1490 > setAgent(agent: IChatAgentData, slashCommand?: IChatAgentCommand) {
1491 this._agent = agent;
1492 this._slashCommand = slashCommand;
1494 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1495 }
1496 > chatModel.ts
1497 > setResult(result: IChatAgentResult): void {
1498 // If already cancelled, discard error details from late-arriving agent responses.
1499 if (this.isCanceled && result.errorDetails) {
1505 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1506 }
1507 > chatModel.ts
1508 > setUsage(usage: IChatUsage): void {
1509 this._parentUsage = usage;
1510 this._setUsage(this._withSubagentCopilotCredits(usage), true);
1511 }
1512 > chatModel.ts
1513 > setSubagentCopilotCredits(subagentCallId: string, copilotCredits: number): void {
1514 const currentCredits = this._subagentCopilotCredits.get(subagentCallId);
1515 if (!Number.isFinite(copilotCredits) || copilotCredits < 0 || (currentCredits !== undefined && copilotCredits <= currentCredits)) {
1520 this._setUsage(this._withSubagentCopilotCredits(usage), false);
1521 }
1522 > chatModel.ts
1523 > private _withSubagentCopilotCredits(usage: IChatUsage): IChatUsage {
1524 let subagentCopilotCredits = 0;
1525 for (const credits of this._subagentCopilotCredits.values()) {
1530 : { ...usage, copilotCredits: (usage.copilotCredits ?? 0) + subagentCopilotCredits };
1531 }
1532 > chatModel.ts
1533 > private _setUsage(usage: IChatUsage, countCompletionTokens: boolean): void {
1534 if (this.isSameUsage(usage)) {
1535 return;
1543 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1544 }
1545 > chatModel.ts
1546 > setElapsedMs(elapsedMs: number): void {
1547 this._elapsedMs = Math.max(0, elapsedMs);
1548 }
1549 > chatModel.ts
1550 > private isSameUsage(usage: IChatUsage): boolean {
1551 const currentUsage = this._usageObs.get();
1552 return !!currentUsage
1557 && equals(currentUsage.promptTokenDetails, usage.promptTokenDetails);
1558 }
1559 > chatModel.ts
1560 > complete(completedAt = Date.now()): void {
1561 this._complete(completedAt, completedAt);
1562 }
1563 > chatModel.ts
1564 > completeWithoutTimestamp(): void {
1565 > this._complete(Date.now(), undefined); chatModel.ts
1566 > }
1567 > chatModel.ts
1568 > private _complete(completedAt: number, completionTimestamp: number | undefined): void {
1569 > // No-op if it's already complete chatModel.ts
1570 > if (this.isComplete) {
1571 return;
1572 }
1573 > if (this._result?.errorDetails?.responseIsRedacted) { chatModel.ts
1574 this._response.clear();
1575 }
1576 > this._response.finalizeReasoningDuration(); chatModel.ts
1577 >
1578 > // Compute elapsed generation time before setting terminal state
1579 > this._elapsedMs ??= Math.max(0, completedAt - this.confirmationAdjustedTimestamp.get());
1580 >
1581 > // Canceled sessions can be considered 'Complete'
1582 > const state = !!this._result?.errorDetails && this._result.errorDetails.code !== 'canceled' ? ResponseModelState.Failed : ResponseModelState.Complete; chatModel.ts
1583 > this._completionTimestamp = completionTimestamp;
1584 > this._modelState.set({ value: state, completedAt }, undefined);
1585 > this._onDidChange.fire({ reason: 'completedRequest' });
1586 > }
1587 > chatModel.ts
1588 > cancel(): void {
1589 this._response.finalizeReasoningDuration();
1590 // Transition any tool invocations that are still streaming partial
1608 this._onDidChange.fire({ reason: 'completedRequest' });
1609 }
1610 > chatModel.ts
1611 > setFollowups(followups: IChatFollowup[] | undefined): void {
1612 this._followups = followups;
1613 this._onDidChange.fire(defaultChatResponseModelChangeReason); // Fire so that command followups get rendered on the row
1614 }
1615 > chatModel.ts
1616 > setVote(vote: ChatAgentVoteDirection): void {
1617 this._vote = vote;
1618 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1619 }
1620 > chatModel.ts
1621 > setEditApplied(edit: IChatTextEditGroup, editCount: number): boolean {
1622 if (!this.response.value.includes(edit)) {
1623 return false;
1630 return true;
1631 }
1632 > chatModel.ts
1633 > adoptTo(session: ChatModel) {
1634 this._session = session;
1635 this._onDidChange.fire(defaultChatResponseModelChangeReason);
1636 }
1637 > chatModel.ts
1638 >
1639 > finalizeUndoState(): void {
1640 this._finalizedResponse = this.response;
1641 this._responseView = undefined;
1642 this._shouldBeRemovedOnSend = undefined;
1643 }
1644 > chatModel.ts
1645 > override dispose(): void {
1646 > super.dispose(); chatModel.ts
1647 > this._response.clear();
1648 > if (this._codeBlockInfos) {
1649 this._codeBlockInfos.length = 0;
1650 }
1651 > } chatModel.ts
1652 > chatModel.ts
1653 > toJSON(): Omit<ISerializableChatResponseData, 'timestamp'> {
1654 const modelState = this._modelState.get();
1655 const pendingConfirmation = this.isPendingConfirmation.get();
1676 } satisfies WithDefinedProps<Omit<ISerializableChatResponseData, 'timestamp'>>;
1677 }
1678 > } chatModel.ts
1679 >
1680 >
1681 > export interface IChatRequestDisablement {
1682 > requestId: string;
1683 > afterUndoStop?: string;
1684 > }
1685 >
1686 > /**
1687 > * Information about a chat request that needs user input to continue.
1688 > */
1689 > export interface IChatRequestNeedsInputInfo {
1690 > /** The chat session title */
1691 > readonly title: string;
1692 > /** Optional detail message, e.g., "<toolname> needs approval to run." */
1693 > readonly detail?: string;
1694 > }
1695 >
1696 > export interface IChatModel extends IDisposable {
1697 > readonly onDidDispose: Event<void>;
1698 > readonly onDidChange: Event<IChatChangeEvent>;
1699 >
1700 > readonly sessionResource: URI;
1701 > /** @deprecated Use {@link sessionResource} instead */
1702 > readonly sessionId: string;
1703 >
1704 > /** Milliseconds timestamp this chat model was created. */
1705 > readonly timestamp: number;
1706 > readonly lastMessageDate: number;
1707 > readonly timing: IChatSessionTiming;
1708 > readonly initialLocation: ChatAgentLocation;
1709 > readonly title: string;
1710 > readonly hasCustomTitle: boolean;
1711 > readonly responderUsername: string;
1712 > /** True whenever a request is currently running */
1713 > readonly requestInProgress: IObservable<boolean>;
1714 > /** True whenever the last request has not reached a terminal state, regardless of intermediate states like tool calls or elicitations */
1715 > readonly hasActiveRequest: IObservable<boolean>;
1716 > /** Provides session information when a request needs user interaction to continue */
1717 > readonly requestNeedsInput: IObservable<IChatRequestNeedsInputInfo | undefined>;
1718 > readonly isReadOnly: IObservable<boolean>;
1719 > readonly inputPlaceholder?: string;
1720 > readonly editingSession?: IChatEditingSession | undefined;
1721 > readonly checkpoint: IChatRequestModel | undefined;
1722 > startEditingSession(isGlobalEditingSession?: boolean, transferFromSession?: IChatEditingSession): void;
1723 > /** Input model for managing input state */
1724 > readonly inputModel: IInputModel;
1725 > readonly hasRequests: boolean;
1726 > readonly lastRequest: IChatRequestModel | undefined;
1727 > /** Whether this model will be kept alive while it is running or has edits */
1728 > readonly willKeepAlive: boolean;
1729 > readonly lastRequestObs: IObservable<IChatRequestModel | undefined>;
1730 > /** Total copilot credits consumed across all turns in this session. */
1731 > readonly sessionCost: number;
1732 > getRequests(): IChatRequestModel[];
1733 > setCheckpoint(requestId: string | undefined): void;
1734 >
1735 > toExport(): IExportableChatData;
1736 > toJSON(): ISerializableChatData;
1737 >
1738 > readonly repoData: IExportableRepoData | undefined;
1739 > setRepoData(data: IExportableRepoData | undefined): void;
1740 >
1741 > /**
1742 > * The working directory URI associated with this session.
1743 > * Only set in the sessions/agents window context.
1744 > */
1745 > readonly workingDirectory: URI | undefined;
1746 > setWorkingDirectory(uri: URI | undefined): void;
1747 >
1748 > readonly onDidChangePendingRequests: Event<void>;
1749 > getPendingRequests(): readonly IChatPendingRequest[];
1750 > }
1751 >
1752 > export interface ISerializableChatsData {
1753 > [sessionId: string]: ISerializableChatData;
1754 > }
1755 >
1756 > export type ISerializableChatAgentData = UriDto<IChatAgentData>;
1757 >
1758 > interface ISerializableChatResponseData {
1759 > responseId?: string;
1760 > result?: IChatAgentResult; // Optional for backcompat
1761 > responseMarkdownInfo?: ISerializableMarkdownInfo[];
1762 > followups?: ReadonlyArray<IChatFollowup>;
1763 > modelState?: ResponseModelStateT;
1764 > vote?: ChatAgentVoteDirection;
1765 > timestamp?: number;
1766 > responseTimestamp?: number;
1767 > slashCommand?: IChatAgentCommand;
1768 > /** For backward compat: should be optional */
1769 > usedContext?: IChatUsedContext;
1770 > contentReferences?: ReadonlyArray<IChatContentReference>;
1771 > codeCitations?: ReadonlyArray<IChatCodeCitation>;
1772 > timeSpentWaiting?: number;
1773 > promptTokens?: number;
1774 > completionTokens?: number;
1775 > outputBuffer?: number;
1776 > promptTokenDetails?: readonly IChatUsagePromptTokenDetail[];
1777 > copilotCredits?: number;
1778 > elapsedMs?: number;
1779 > }
1780 >
1781 > export type SerializedChatResponsePart = IMarkdownString | IChatResponseProgressFileTreeData | IChatContentInlineReference | IChatAgentMarkdownContentWithVulnerability | IChatThinkingPart | IChatProgressResponseContentSerialized | IChatQuestionCarousel | IChatPlanReview | IChatDisabledClaudeHooksPart;
1782 >
1783 > export interface ISerializableChatRequestData extends ISerializableChatResponseData {
1784 > requestId: string;
1785 > message: string | IParsedChatRequest; // string => old format
1786 > /** Is really like "prompt data". This is the message in the format in which the agent gets it + variable values. */
1787 > variableData: IChatRequestVariableData;
1788 > response: ReadonlyArray<SerializedChatResponsePart> | undefined;
1789 >
1790 > /**Old, persisted name for shouldBeRemovedOnSend */
1791 > isHidden?: boolean;
1792 > shouldBeRemovedOnSend?: IChatRequestDisablement;
1793 > agent?: ISerializableChatAgentData;
1794 > // responseErrorDetails: IChatResponseErrorDetails | undefined;
1795 > /** @deprecated modelState is used instead now */
1796 > isCanceled?: boolean;
1797 > timestamp?: number;
1798 > confirmation?: string;
1799 > editedFileEvents?: IChatAgentEditedFileEvent[];
1800 > modelId?: string;
1801 > modeInfo?: IChatRequestModeInfo;
1802 > isSystemInitiated?: boolean;
1803 > systemInitiatedLabel?: string;
1804 > terminalExecutionId?: string;
1805 > }
1806 >
1807 > export interface ISerializableMarkdownInfo {
1808 > readonly suggestionId: EditSuggestionId;
1809 > }
1810 >
1811 > /**
1812 > * Repository state captured for chat session export.
1813 > * Enables reproducing the workspace state by cloning, checking out the commit, and applying diffs.
1814 > */
1815 > export interface IExportableRepoData {
1816 > /**
1817 > * Classification of the workspace's version control state.
1818 > * - `remote-git`: Git repo with a configured remote URL
1819 > * - `local-git`: Git repo without any remote (local only)
1820 > * - `plain-folder`: Not a git repository
1821 > */
1822 > workspaceType: 'remote-git' | 'local-git' | 'plain-folder';
1823 >
1824 > /**
1825 > * Sync status between local and remote.
1826 > * - `synced`: Local HEAD matches remote tracking branch (fully pushed)
1827 > * - `unpushed`: Local has commits not pushed to the remote tracking branch
1828 > * - `unpublished`: Local branch has no remote tracking branch configured
1829 > * - `local-only`: No remote configured (local git repo only)
1830 > * - `no-git`: Not a git repository
1831 > */
1832 > syncStatus: 'synced' | 'unpushed' | 'unpublished' | 'local-only' | 'no-git';
1833 >
1834 > /**
1835 > * Remote URL of the repository (e.g., https://github.com/org/repo.git).
1836 > * Undefined if no remote is configured.
1837 > */
1838 > remoteUrl?: string;
1839 >
1840 > /**
1841 > * Vendor/host of the remote repository.
1842 > * Undefined if no remote is configured.
1843 > */
1844 > remoteVendor?: 'github' | 'ado' | 'other';
1845 >
1846 > /**
1847 > * Remote tracking branch for the current branch (e.g., "origin/feature/my-work").
1848 > * Undefined if branch is unpublished or no remote.
1849 > */
1850 > remoteTrackingBranch?: string;
1851 >
1852 > /**
1853 > * Default remote branch used as base for unpublished branches (e.g., "origin/main").
1854 > * Helpful for computing merge-base when branch has no tracking.
1855 > */
1856 > remoteBaseBranch?: string;
1857 >
1858 > /**
1859 > * Commit hash of the remote tracking branch HEAD.
1860 > * Undefined if branch has no remote tracking branch.
1861 > */
1862 > remoteHeadCommit?: string;
1863 >
1864 > /**
1865 > * Name of the current local branch (e.g., "feature/my-work").
1866 > */
1867 > localBranch?: string;
1868 >
1869 > /**
1870 > * Commit hash of the local HEAD when captured.
1871 > */
1872 > localHeadCommit?: string;
1873 >
1874 > /**
1875 > * Working tree diffs (uncommitted changes).
1876 > */
1877 > diffs?: IExportableRepoDiff[];
1878 >
1879 > /**
1880 > * Status of the diffs collection.
1881 > * - `included`: Diffs were successfully captured and included
1882 > * - `tooManyChanges`: Diffs skipped because >100 files changed (degenerate case like mass renames)
1883 > * - `tooLarge`: Diffs skipped because total size exceeded 900KB
1884 > * - `trimmedForStorage`: Diffs were trimmed to save storage (older session)
1885 > * - `noChanges`: No working tree changes detected
1886 > * - `notCaptured`: Diffs not captured (default/undefined case)
1887 > */
1888 > diffsStatus?: 'included' | 'tooManyChanges' | 'tooLarge' | 'trimmedForStorage' | 'noChanges' | 'notCaptured';
1889 >
1890 > /**
1891 > * Number of changed files detected, even if diffs were not included.
1892 > */
1893 > changedFileCount?: number;
1894 > }
1895 >
1896 > /**
1897 > * A file change exported as a unified diff patch compatible with `git apply`.
1898 > */
1899 > export interface IExportableRepoDiff {
1900 > relativePath: string;
1901 > changeType: 'added' | 'modified' | 'deleted' | 'renamed';
1902 > oldRelativePath?: string;
1903 > unifiedDiff?: string;
1904 > status: string;
1905 > }
1906 >
1907 > export interface IExportableChatData {
1908 > initialLocation: ChatAgentLocation | undefined;
1909 > requests: ISerializableChatRequestData[];
1910 > responderUsername: string;
1911 > }
1912 >
1913 > /*
1914 > NOTE: every time the serialized data format is updated, we need to create a new interface, because we may need to handle any old data format when parsing.
1915 > */
1916 >
1917 > export interface ISerializableChatData1 extends IExportableChatData {
1918 > sessionId: string;
1919 > creationDate: number;
1920 > }
1921 >
1922 > export interface ISerializableChatData2 extends ISerializableChatData1 {
1923 > version: 2;
1924 > computedTitle: string | undefined;
1925 > }
1926 >
1927 > export interface ISerializableChatData3 extends Omit<ISerializableChatData2, 'version' | 'computedTitle'> {
1928 > version: 3;
1929 > customTitle: string | undefined;
1930 > /** Whether the session had pending edits when it was stored. */
1931 > hasPendingEdits?: boolean;
1932 > /** Current draft input state (added later, fully backwards compatible) */
1933 > inputState?: ISerializableChatModelInputState;
1934 > repoData?: IExportableRepoData;
1935 > /** Pending requests that were queued but not yet processed */
1936 > pendingRequests?: ISerializablePendingRequestData[];
1937 > /** The working directory URI associated with this session (sessions/agents window). */
1938 > workingDirectory?: string;
1939 > }
1940 >
1941 > /**
1942 > * Input model for managing chat input state independently from the chat model.
1943 > * This keeps display logic separated from the core chat model.
1944 > *
1945 > * The input model:
1946 > * - Manages the current draft state (text, attachments, mode, model selection, cursor/selection)
1947 > * - Provides an observable interface for reactive UI updates
1948 > * - Automatically persists through the chat model's serialization
1949 > * - Enables bidirectional sync between the UI (ChatInputPart) and the model
1950 > * - Uses `undefined` state to indicate no persisted state (new/empty chat)
1951 > *
1952 > * This architecture ensures that:
1953 > * - Input state is preserved when moving chats between editor/sidebar/window
1954 > * - No manual state transfer is needed when switching contexts
1955 > * - The UI stays in sync with the persisted state
1956 > * - New chats use UI defaults (persisted preferences) instead of hardcoded values
1957 > */
1958 > export interface IInputModel {
1959 > /** Observable for current input state (undefined for new/uninitialized chats) */
1960 > readonly state: IObservable<IChatModelInputState | undefined>;
1961 >
1962 > /** Update the input state (partial update) */
1963 > setState(state: Partial<IChatModelInputState>): void;
1964 >
1965 > /** Clear input state (after sending or clearing) */
1966 > clearState(): void;
1967 >
1968 > /** Serializes the state */
1969 > toJSON(): ISerializableChatModelInputState | undefined;
1970 > }
1971 >
1972 > /**
1973 > * Represents the current state of the chat input that hasn't been sent yet.
1974 > * This is the "draft" state that should be preserved across sessions.
1975 > */
1976 > export interface IChatModelInputState {
1977 > /** Current attachments in the input */
1978 > attachments: readonly IChatRequestVariableEntry[];
1979 >
1980 > /** Currently selected chat mode */
1981 > mode: {
1982 > /** Mode ID (e.g., 'ask', 'edit', 'agent', or custom mode ID) */
1983 > id: string;
1984 > /** Mode kind for builtin modes */
1985 > kind: ChatModeKind | undefined;
1986 > };
1987 >
1988 > /** Currently selected language model, if any */
1989 > selectedModel: ILanguageModelChatMetadataAndIdentifier | undefined;
1990 >
1991 > /**
1992 > * Configuration (e.g. context size, thinking effort) for the selected
1993 > * model, captured so it can be restored alongside the model when the
1994 > * session is reopened.
1995 > */
1996 > modelConfiguration?: IStringDictionary<unknown>;
1997 >
1998 > /** Current input text */
1999 > inputText: string;
2000 >
2001 > /** Current selection ranges */
2002 > selections: ISelection[];
2003 >
2004 > /** Current permission level for tool auto-approval */
2005 > permissionLevel?: ChatPermissionLevel;
2006 >
2007 > /** Contributed stored state */
2008 > contrib: Record<string, unknown>;
2009 > }
2010 >
2011 > /**
2012 > * Serializable version of IChatModelInputState
2013 > */
2014 > export interface ISerializableChatModelInputState {
2015 > attachments: readonly IChatRequestVariableEntry[];
2016 > mode: {
2017 > id: string;
2018 > kind: ChatModeKind | undefined;
2019 > };
2020 > selectedModel: {
2021 > identifier: string;
2022 > metadata: ILanguageModelChatMetadata;
2023 > /**
2024 > * Configuration (e.g. context size, thinking effort) for the selected
2025 > * model, captured so it can be restored alongside the model when the
2026 > * session is reopened.
2027 > */
2028 > modelConfiguration?: IStringDictionary<unknown>;
2029 > } | undefined;
2030 > inputText: string;
2031 > selections: ISelection[];
2032 > permissionLevel?: ChatPermissionLevel;
2033 > contrib: Record<string, unknown>;
2034 > }
2035 >
2036 > /**
2037 > * Legacy shape of {@link ISerializableChatModelInputState} as persisted by older
2038 > * versions, where the selected model's configuration was stored as a sibling
2039 > * `modelConfiguration` field instead of nested inside `selectedModel`. Retained
2040 > * so sessions serialized in the old format can still be read.
2041 > */
2042 > interface ILegacySerializableChatModelInputState extends ISerializableChatModelInputState {
2043 > modelConfiguration?: IStringDictionary<unknown>;
2044 > }
2045 >
2046 > /**
2047 > * Chat data that has been parsed and normalized to the current format.
2048 > */
2049 > export type ISerializableChatData = ISerializableChatData3;
2050 >
2051 > export type IChatDataSerializerLog = ObjectMutationLog<IChatModel, ISerializableChatData>;
2052 >
2053 > export interface ISerializedChatDataReference {
2054 > value: ISerializableChatData | IExportableChatData;
2055 > serializer: IChatDataSerializerLog;
2056 > }
2057 >
2058 > /**
2059 > * Chat data that has been loaded but not normalized, and could be any format
2060 > */
2061 > export type ISerializableChatDataIn = ISerializableChatData1 | ISerializableChatData2 | ISerializableChatData3;
2062 >
2063 > /**
2064 > * Normalize chat data from storage to the current format.
2065 > * TODO- ChatModel#_deserialize and reviveSerializedAgent also still do some normalization and maybe that should be done in here too.
2066 > */
2067 > export function normalizeSerializableChatData(raw: ISerializableChatDataIn): ISerializableChatData {
2068 normalizeOldFields(raw);
2069
2086 return raw;
2087 }
2088 > chatModel.ts
2089 function normalizeOldFields(raw: ISerializableChatDataIn): void {
2090 // Fill in fields that very old chat data may be missing
2102 }
2103 }
2104 > chatModel.ts
2105 function getLastYearDate(): number {
2106 const lastYearDate = new Date();
2108 return lastYearDate.getTime();
2109 }
2110 > chatModel.ts
2111 > export function isExportableSessionData(obj: unknown): obj is IExportableChatData {
2112 > return !!obj && chatModel.ts
2113 Array.isArray((obj as IExportableChatData).requests) &&
2114 typeof (obj as IExportableChatData).responderUsername === 'string';
2115 > } chatModel.ts
2116 > chatModel.ts
2117 > export function isSerializableSessionData(obj: unknown): obj is ISerializableChatData {
2118 const data = obj as ISerializableChatData;
2119 return isExportableSessionData(obj) &&
2124 );
2125 }
2126 > chatModel.ts
2127 > export type IChatChangeEvent =
2128 > | IChatInitEvent
2129 > | IChatAddRequestEvent | IChatChangedRequestEvent | IChatRemoveRequestEvent
2130 > | IChatAddResponseEvent
2131 > | IChatSetAgentEvent
2132 > | IChatMoveEvent
2133 > | IChatSetHiddenEvent
2134 > | IChatCompletedRequestEvent
2135 > | IChatSetCustomTitleEvent
2136 > ;
2137 >
2138 > export interface IChatAddRequestEvent {
2139 > kind: 'addRequest';
2140 > request: IChatRequestModel;
2141 > }
2142 >
2143 > export interface IChatChangedRequestEvent {
2144 > kind: 'changedRequest';
2145 > request: IChatRequestModel;
2146 > }
2147 >
2148 > export interface IChatCompletedRequestEvent {
2149 > kind: 'completedRequest';
2150 > request: IChatRequestModel;
2151 > }
2152 >
2153 > export interface IChatAddResponseEvent {
2154 > kind: 'addResponse';
2155 > response: IChatResponseModel;
2156 > }
2157 >
2158 > export const enum ChatRequestRemovalReason {
2159 > /**
2160 > * "Normal" remove
2161 > */
2162 > Removal,
2163 >
2164 > /**
2165 > * Removed because the request will be resent
2166 > */
2167 > Resend,
2168 >
2169 > /**
2170 > * Remove because the request is moving to another model
2171 > */
2172 > Adoption
2173 > }
2174 >
2175 > export interface IChatRemoveRequestEvent {
2176 > kind: 'removeRequest';
2177 > requestId: string;
2178 > responseId?: string;
2179 > reason: ChatRequestRemovalReason;
2180 > }
2181 >
2182 > export interface IChatSetHiddenEvent {
2183 > kind: 'setHidden';
2184 > }
2185 >
2186 > export interface IChatMoveEvent {
2187 > kind: 'move';
2188 > target: URI;
2189 > range: IRange;
2190 > }
2191 >
2192 > export interface IChatSetAgentEvent {
2193 > kind: 'setAgent';
2194 > agent: IChatAgentData;
2195 > command?: IChatAgentCommand;
2196 > }
2197 >
2198 > export interface IChatSetCustomTitleEvent {
2199 > kind: 'setCustomTitle';
2200 > title: string;
2201 > }
2202 >
2203 > export interface IChatInitEvent {
2204 > kind: 'initialize';
2205 > }
2206 >
2207 > /**
2208 > * Internal implementation of IInputModel
2209 > */
2210 > class InputModel implements IInputModel {
2211 > private readonly _state: ReturnType<typeof observableValue<IChatModelInputState | undefined>>;
2212 > readonly state: IObservable<IChatModelInputState | undefined>;
2213 >
2214 > constructor(initialState: IChatModelInputState | undefined, private readonly logger: ILogService, private readonly sessionId: string) {
2215 > this._state = observableValueOpts({ debugName: 'inputModelState', equalsFn: equals }, initialState); chatModel.ts
2216 > this.state = this._state;
2217 > }
2218 > chatModel.ts
2219 > setState(state: Partial<IChatModelInputState>): void {
2220 const current = this._state.get();
2221 _logChangesToStateModel(state, current, this.logger, this.sessionId);
2232 }, undefined);
2233 }
2234 > chatModel.ts
2235 > clearState(): void {
2236 this._state.set(undefined, undefined);
2237 }
2238 > chatModel.ts
2239 > toJSON(): ISerializableChatModelInputState | undefined {
2240 const value = this.state.get();
2241 if (!value) {
2269 };
2270 }
2271 > } chatModel.ts
2272 >
2273 > export class ChatModel extends Disposable implements IChatModel {
2274 > static getDefaultTitle(requests: (ISerializableChatRequestData | IChatRequestModel)[]): string {
2275 > const firstRequestMessage = requests.at(0)?.message ?? '';
2276 > const message = typeof firstRequestMessage === 'string' ?
2277 > firstRequestMessage : chatModel.ts
2278 > firstRequestMessage.text; chatModel.ts
2279 > return message.split('\n')[0].substring(0, 200); chatModel.ts
2280 > }
2281 >
2282 > private readonly _onDidDispose = this._register(new Emitter<void>());
2283 > readonly onDidDispose = this._onDidDispose.event;
2284 >
2285 > private readonly _onDidChange = this._register(new Emitter<IChatChangeEvent>());
2286 > readonly onDidChange = this._onDidChange.event;
2287 >
2288 > private readonly _pendingRequests: IChatPendingRequest[] = [];
2289 > private readonly _onDidChangePendingRequests = this._register(new Emitter<void>());
2290 > readonly onDidChangePendingRequests = this._onDidChangePendingRequests.event;
2291 >
2292 > private _requests: ChatRequestModel[];
2293 >
2294 > private _repoData: IExportableRepoData | undefined;
2295 > public get repoData(): IExportableRepoData | undefined {
2296 return this._repoData;
2297 }
2298 > public setRepoData(data: IExportableRepoData | undefined): void { chatModel.ts
2299 this._repoData = data;
2300 }
2301 > chatModel.ts
2302 > private _workingDirectory: URI | undefined;
2303 > public get workingDirectory(): URI | undefined {
2304 return this._workingDirectory;
2305 }
2306 > public setWorkingDirectory(uri: URI | undefined): void { chatModel.ts
2307 this._workingDirectory = uri;
2308 }
2309 > chatModel.ts
2310 > getPendingRequests(): readonly IChatPendingRequest[] {
2311 return this._pendingRequests;
2312 }
2313 > chatModel.ts
2314 > setPendingRequests(requests: readonly { requestId: string; kind: ChatRequestQueueKind }[]): void {
2315 const existingMap = new Map(this._pendingRequests.map(p => [p.request.id, p]));
2316 const newPending: IChatPendingRequest[] = [];
2326 this._onDidChangePendingRequests.fire();
2327 }
2328 > chatModel.ts
2329 > /**
2330 > * @internal Used by ChatService to atomically replace the pending request queue.
2331 > */
2332 > replacePendingRequests(requests: readonly IChatPendingRequest[]): void {
2333 if (this._pendingRequests.length === requests.length && requests.every((request, index) => this._pendingRequests[index] === request)) {
2334 return;
2338 this._onDidChangePendingRequests.fire();
2339 }
2340 > chatModel.ts
2341 > /**
2342 > * @internal Used by ChatService to add a request to the queue.
2343 > * Steering messages are placed before queued messages.
2344 > */
2345 > addPendingRequest(request: ChatRequestModel, kind: ChatRequestQueueKind, sendOptions: IChatSendRequestOptions): IChatPendingRequest {
2346 const pendingRequest: IChatPendingRequest = {
2347 request,
2369 return pendingRequest;
2370 }
2371 > chatModel.ts
2372 > /**
2373 > * @internal Used by ChatService to remove a pending request
2374 > */
2375 > removePendingRequest(id: string): void {
2376 const index = this._pendingRequests.findIndex(r => r.request.id === id);
2377 if (index !== -1) {
2380 }
2381 }
2382 > chatModel.ts
2383 > /**
2384 > * @internal Used by ChatService to dequeue the next pending request
2385 > */
2386 > dequeuePendingRequest(): IChatPendingRequest | undefined {
2387 const request = this._pendingRequests.shift();
2388 if (request) {
2391 return request;
2392 }
2393 > chatModel.ts
2394 > /**
2395 > * @internal Used by ChatService to dequeue all consecutive steering requests at the front of the queue.
2396 > * Returns an empty array if the first pending request is not a steering request.
2397 > */
2398 > dequeueAllSteeringRequests(): IChatPendingRequest[] {
2399 const steeringRequests: IChatPendingRequest[] = [];
2400 while (this._pendingRequests.at(0)?.kind === ChatRequestQueueKind.Steering) {
2406 return steeringRequests;
2407 }
2408 > chatModel.ts
2409 > /**
2410 > * @internal Used by ChatService to clear all pending requests
2411 > */
2412 > clearPendingRequests(): void {
2413 if (this._pendingRequests.length > 0) {
2414 this._pendingRequests.length = 0;
2416 }
2417 }
2418 > chatModel.ts
2419 > readonly lastRequestObs: IObservable<IChatRequestModel | undefined>;
2420 >
2421 > // TODO to be clear, this is not the same as the id from the session object, which belongs to the provider.
2422 > // It's easier to be able to identify this model before its async initialization is complete
2423 > private readonly _sessionId: string;
2424 > /** @deprecated Use {@link sessionResource} instead */
2425 > get sessionId(): string {
2426 return this._sessionId;
2427 }
2428 > chatModel.ts
2429 > private readonly _sessionResource: URI;
2430 > get sessionResource(): URI {
2431 > return this._sessionResource; chatModel.ts
2432 > }
2433 > chatModel.ts
2434 > readonly requestInProgress: IObservable<boolean>;
2435 > readonly hasActiveRequest: IObservable<boolean>;
2436 > readonly requestNeedsInput: IObservable<IChatRequestNeedsInputInfo | undefined>;
2437 > readonly isReadOnly: IObservable<boolean>;
2438 >
2439 > /** Input model for managing input state */
2440 > readonly inputModel: InputModel;
2441 >
2442 > get hasRequests(): boolean {
2443 return this._requests.length > 0;
2444 }
2445 > chatModel.ts
2446 > get lastRequest(): ChatRequestModel | undefined {
2447 return this._requests.at(-1);
2448 }
2449 > chatModel.ts
2450 > get sessionCost(): number {
2451 let totalCredits = 0;
2452 for (const request of this._requests) {
2458 return totalCredits;
2459 }
2460 > chatModel.ts
2461 > private _timestamp: number;
2462 > get timestamp(): number {
2463 return this._timestamp;
2464 }
2465 > chatModel.ts
2466 > get timing(): IChatSessionTiming {
2467 const lastRequest = this._requests.at(-1);
2468 const lastResponse = lastRequest?.response;
2475 };
2476 }
2477 > chatModel.ts
2478 > get lastMessageDate(): number {
2479 return this._requests.at(-1)?.timestamp ?? this._timestamp;
2480 }
2481 > chatModel.ts
2482 > private get _defaultAgent() {
2483 return this.chatAgentService.getDefaultAgent(ChatAgentLocation.Chat, ChatModeKind.Ask);
2484 }
2485 > chatModel.ts
2486 > private readonly _initialResponderUsername: string | undefined;
2487 > get responderUsername(): string {
2488 return this._defaultAgent?.fullName ??
2489 this._initialResponderUsername ?? '';
2490 }
2491 > chatModel.ts
2492 > private _isImported = false;
2493 > get isImported(): boolean {
2494 return this._isImported;
2495 }
2496 > chatModel.ts
2497 > private _isDeleted = false;
2498 > get isDeleted(): boolean {
2499 return this._isDeleted;
2500 }
2501 > markDeleted(): void { chatModel.ts
2502 this._isDeleted = true;
2503 }
2504 > chatModel.ts
2505 > private _customTitle: string | undefined;
2506 > get customTitle(): string | undefined {
2507 return this._customTitle;
2508 }
2509 > chatModel.ts
2510 > get title(): string {
2511 return this._customTitle || ChatModel.getDefaultTitle(this._requests);
2512 }
2513 > chatModel.ts
2514 > get hasCustomTitle(): boolean {
2515 return this._customTitle !== undefined;
2516 }
2517 > chatModel.ts
2518 > private _editingSession: IChatEditingSession | undefined;
2519 >
2520 > get editingSession(): IChatEditingSession | undefined {
2521 > return this._editingSession; chatModel.ts
2522 > }
2523 > chatModel.ts
2524 > private readonly _initialLocation: ChatAgentLocation;
2525 > get initialLocation(): ChatAgentLocation {
2526 > return this._initialLocation; chatModel.ts
2527 > }
2528 > chatModel.ts
2529 > private readonly _canUseTools: boolean = true;
2530 > get canUseTools(): boolean {
2531 return this._canUseTools;
2532 }
2533 > chatModel.ts
2534 > private _disableBackgroundKeepAlive: boolean;
2535 > get willKeepAlive(): boolean {
2536 return !this._disableBackgroundKeepAlive;
2537 }
2538 > chatModel.ts
2539 > public dataSerializer?: IChatDataSerializerLog;
2540 >
2541 > constructor(
2542 > dataRef: ISerializedChatDataReference | undefined, chatModel.ts
2543 > initialModelProps: { initialLocation: ChatAgentLocation; canUseTools: boolean; inputState?: ISerializableChatModelInputState; resource?: URI; disableBackgroundKeepAlive?: boolean; isReadOnly?: IObservable<boolean> },
2544 > @ILogService private readonly logService: ILogService,
2545 > @IChatAgentService private readonly chatAgentService: IChatAgentService,
2546 > @IChatEditingService private readonly chatEditingService: IChatEditingService,
2547 > @IChatService private readonly chatService: IChatService,
2548 > ) {
2549 > super();
2550 >
2551 > const initialData = dataRef?.value;
2552 > const isValidExportedData = isExportableSessionData(initialData);
2553 > const isValidFullData = isValidExportedData && isSerializableSessionData(initialData);
2554 > if (initialData && !isValidExportedData) {
2555 this.logService.warn(`ChatModel#constructor: Loaded malformed session data: ${JSON.stringify(initialData)}`);
2556 }
2557 > chatModel.ts
2558 > this._isImported = !!initialData && isValidExportedData && !isValidFullData;
2559 >
2560 > // Set the session resource and id
2561 > if (initialModelProps.resource) {
2562 > // prefer using the provided resource if provided chatModel.ts
2563 > this._sessionId = chatSessionResourceToId(initialModelProps.resource);
2564 > this._sessionResource = initialModelProps.resource;
2565 > } else if (isValidFullData) { chatModel.ts
2566 // Otherwise use the serialized id. This is only valid for local chat sessions
2567 this._sessionId = initialData.sessionId;
2573 this._sessionResource = LocalChatSessionUri.forSession(this._sessionId);
2574 }
2575 > chatModel.ts
2576 > this._disableBackgroundKeepAlive = initialModelProps.disableBackgroundKeepAlive ?? false;
2577 >
2578 > this._timestamp = (isValidFullData && initialData.creationDate) || Date.now();
2579 > this._requests = initialData ? this._deserialize(initialData) : [];
2580 > this._customTitle = isValidFullData ? initialData.customTitle : undefined;
2581 >
2582 > // Initialize input model from serialized data (undefined for new chats)
2583 > const serializedInputState = initialModelProps.inputState || (isValidFullData && initialData.inputState ? initialData.inputState : undefined);
2584 > this.inputModel = new InputModel(serializedInputState && {
2585 attachments: (serializedInputState.attachments ?? []).map(IChatRequestVariableEntry.fromExport),
2586 mode: serializedInputState.mode,
2594 selections: serializedInputState.selections,
2595 permissionLevel: serializedInputState.permissionLevel,
2596 > }, this.logService, this._sessionId); chatModel.ts
2597 >
2598 > this.dataSerializer = dataRef?.serializer;
2599 > this._initialResponderUsername = initialData?.responderUsername;
2600 >
2601 > this._repoData = isValidFullData && initialData.repoData ? initialData.repoData : undefined;
2602 >
2603 > this._workingDirectory = isValidFullData && initialData.workingDirectory ? URI.parse(initialData.workingDirectory) : undefined;
2604 >
2605 > // Hydrate pending requests from serialized data
2606 > if (isValidFullData && initialData.pendingRequests) {
2607 this._pendingRequests = this._deserializePendingRequests(initialData.pendingRequests);
2608 }
2609 > chatModel.ts
2610 > this._initialLocation = initialData?.initialLocation ?? initialModelProps.initialLocation;
2611 >
2612 > this._canUseTools = initialModelProps.canUseTools;
2613 > this.isReadOnly = initialModelProps.isReadOnly ?? constObservable(false);
2614 >
2615 > this.lastRequestObs = observableFromEvent(this, this.onDidChange, () => this._requests.at(-1));
2616 >
2617 > this._register(autorun(reader => {
2618 > const request = this.lastRequestObs.read(reader);
2619 > if (!request?.response) {
2620 > return; chatModel.ts
2621 > }
2622 > chatModel.ts
2623 > reader.store.add(request.response.onDidChange(async ev => {
2624 > if (!this._editingSession || ev.reason !== 'completedRequest') { chatModel.ts
2625 return;
2626 }
2627 > chatModel.ts
2628 > this._onDidChange.fire({ kind: 'completedRequest', request });
2629 > })); chatModel.ts
2630 > })); chatModel.ts
2631 >
2632 > this.requestInProgress = this.lastRequestObs.map((request, r) => {
2633 > return request?.response?.isInProgress.read(r) ?? false;
2634 > });
2635 >
2636 > this.hasActiveRequest = this.lastRequestObs.map((request, r) => {
2637 return request?.response?.isIncomplete.read(r) ?? false;
2638 > }); chatModel.ts
2639 >
2640 > this.requestNeedsInput = this.lastRequestObs.map((request, r) => {
2641 > const pendingInfo = request?.response?.isPendingConfirmation.read(r);
2642 > if (!pendingInfo) {
2643 > return undefined;
2644 > }
2645 return {
2646 title: this.title,
2647 detail: pendingInfo.detail,
2648 };
2649 > }); chatModel.ts
2650 >
2651 > // Retain a reference to itself when a request is in progress, so the ChatModel stays alive in the background
2652 > // only while running a request. TODO also keep it alive for 5min or so so we don't have to dispose/restore too often?
2653 > if (this.initialLocation === ChatAgentLocation.Chat && !initialModelProps.disableBackgroundKeepAlive) {
2654 > const selfRef = this._register(new MutableDisposable<IChatModelReference>());
2655 > this._register(autorun(r => {
2656 > const inProgress = this.requestInProgress.read(r);
2657 > const needsInput = this.requestNeedsInput.read(r);
2658 > const shouldStayAlive = inProgress || !!needsInput;
2659 > if (shouldStayAlive && !selfRef.value) {
2660 > selfRef.value = chatService.acquireExistingSession(this._sessionResource, 'ChatModel#requestInProgressKeepAlive'); chatModel.ts
2661 > } else if (!shouldStayAlive && selfRef.value) { chatModel.ts
2662 selfRef.clear();
2663 }
2664 > })); chatModel.ts
2665 > }
2666 > }
2667 > chatModel.ts
2668 > startEditingSession(isGlobalEditingSession?: boolean, transferFromSession?: IChatEditingSession): void {
2669 > const session = this._editingSession ??= this._register( chatModel.ts
2670 > transferFromSession
2671 ? this.chatEditingService.transferEditingSession(this, transferFromSession)
2672 > : isGlobalEditingSession chatModel.ts
2673 > ? this.chatEditingService.startOrContinueGlobalEditingSession(this)
2674 : this.chatEditingService.createEditingSession(this)
2675 > ); chatModel.ts
2676 >
2677 > if (!this._disableBackgroundKeepAlive) {
2678 > // todo@connor4312: hold onto a reference so background sessions don't
2679 > // trigger early disposal. This will be cleaned up with the globalization of edits.
2680 > const selfRef = this._register(new MutableDisposable<IChatModelReference>());
2681 > this._register(autorun(r => {
2682 > const hasModified = session.entries.read(r).some(e => e.state.read(r) === ModifiedFileEntryState.Modified);
2683 > if (hasModified && !selfRef.value) {
2684 selfRef.value = this.chatService.acquireExistingSession(this._sessionResource, 'ChatModel#modifiedEditsKeepAlive');
2685 > } else if (!hasModified && selfRef.value) { chatModel.ts
2686 selfRef.clear();
2687 }
2688 > })); chatModel.ts
2689 > }
2690 >
2691 > this._register(autorun(reader => {
2692 > this._setDisabledRequests(session.requestDisablement.read(reader));
2693 > }));
2694 > }
2695 > chatModel.ts
2696 > private currentEditedFileEvents = new ResourceMap<IChatAgentEditedFileEvent>();
2697 > notifyEditingAction(action: IChatEditingSessionAction): void {
2698 const state = action.outcome === 'accepted' ? ChatRequestEditedFileEventKind.Keep :
2699 action.outcome === 'rejected' ? ChatRequestEditedFileEventKind.Undo :
2707 }
2708 }
2709 > chatModel.ts
2710 > private _deserialize(obj: IExportableChatData | ISerializedChatDataReference): ChatRequestModel[] {
2711 const requests = hasKey(obj, { serializer: true }) ? obj.value.requests : obj.requests;
2712 if (!Array.isArray(requests)) {
2722 }
2723 }
2724 > chatModel.ts
2725 > private _deserializeRequest(raw: ISerializableChatRequestData): ChatRequestModel {
2726 const parsedRequest =
2727 typeof raw.message === 'string'
2813 return request;
2814 }
2815 > chatModel.ts
2816 > private reviveVariableData(raw: IChatRequestVariableData): IChatRequestVariableData {
2817 const variableData = raw && Array.isArray(raw.variables)
2818 ? raw :
2823 return variableData;
2824 }
2825 > chatModel.ts
2826 > private getParsedRequestFromString(message: string): IParsedChatRequest {
2827 // TODO These offsets won't be used, but chat replies need to go through the parser as well
2828 const parts = [new ChatRequestTextPart(new OffsetRange(0, message.length), { startColumn: 1, startLineNumber: 1, endColumn: 1, endLineNumber: 1 }, message)];
2832 };
2833 }
2834 > chatModel.ts
2835 > /**
2836 > * Hydrates pending requests from serialized data.
2837 > * For each serialized pending request, finds the matching request model and adds it to the pending queue.
2838 > */
2839 > private _deserializePendingRequests(pendingRequests: ISerializablePendingRequestData[]): IChatPendingRequest[] {
2840 try {
2841 return pendingRequests.map(pending => ({
2855 }
2856 }
2857 > chatModel.ts
2858 >
2859 >
2860 > getRequests(): ChatRequestModel[] {
2861 > return this._requests; chatModel.ts
2862 > }
2863 > chatModel.ts
2864 > resetCheckpoint(): void {
2865 for (const request of this._requests) {
2866 request.setShouldBeBlocked(false);
2870 }
2871 }
2872 > chatModel.ts
2873 > setCheckpoint(requestId: string | undefined) {
2874 let checkpoint: ChatRequestModel | undefined;
2875 let checkpointIndex = -1;
2910 this._checkpoint = checkpoint;
2911 }
2912 > chatModel.ts
2913 > private _checkpoint: ChatRequestModel | undefined = undefined;
2914 > public get checkpoint() {
2915 return this._checkpoint;
2916 }
2917 > chatModel.ts
2918 > private _setDisabledRequests(requestIds: IChatRequestDisablement[]) {
2919 > this._requests.forEach((request) => { chatModel.ts
2920 const shouldBeRemovedOnSend = requestIds.find(r => r.requestId === request.id);
2921 request.shouldBeRemovedOnSend = shouldBeRemovedOnSend;
2923 request.response.shouldBeRemovedOnSend = shouldBeRemovedOnSend;
2924 }
2925 > }); chatModel.ts
2926 >
2927 > this._onDidChange.fire({ kind: 'setHidden' });
2928 > }
2929 > chatModel.ts
2930 > addRequest(
2931 > message: IParsedChatRequest, chatModel.ts
2932 > variableData: IChatRequestVariableData,
2933 > attempt: number,
2934 > modeInfo?: IChatRequestModeInfo,
2935 > chatAgent?: IChatAgentData,
2936 > slashCommand?: IChatAgentCommand,
2937 > confirmation?: string,
2938 > locationData?: IChatLocationData,
2939 > attachments?: IChatRequestVariableEntry[],
2940 > isCompleteAddedRequest?: boolean,
2941 > modelId?: string,
2942 > userSelectedTools?: UserSelectedTools,
2943 > id?: string,
2944 > isSystemInitiated?: boolean,
2945 > systemInitiatedLabel?: string,
2946 > terminalExecutionId?: string,
2947 > isTerminalCommand?: boolean,
2948 > timestamp?: number | null,
2949 > ): ChatRequestModel {
2950 > const editedFileEvents = [...this.currentEditedFileEvents.values()];
2951 > this.currentEditedFileEvents.clear();
2952 > const requestTimestamp = timestamp === undefined
2953 ? Date.now()
2954 > : typeof timestamp === 'number' && Number.isFinite(timestamp) && timestamp > 0 chatModel.ts
2955 ? timestamp
2956 > : undefined; chatModel.ts
2957 > const request = new ChatRequestModel({ chatModel.ts
2958 > restoredId: id,
2959 > session: this,
2960 > message,
2961 > variableData,
2962 > timestamp: requestTimestamp,
2963 > fallbackTimestamp: this._timestamp,
2964 > attempt,
2965 > modeInfo,
2966 > confirmation,
2967 > locationData,
2968 > attachedContext: attachments,
2969 > isCompleteAddedRequest,
2970 > modelId,
2971 > editedFileEvents: editedFileEvents.length ? editedFileEvents : undefined,
2972 > userSelectedTools,
2973 > isSystemInitiated,
2974 > systemInitiatedLabel,
2975 > terminalExecutionId,
2976 > isTerminalCommand,
2977 > });
2978 > request.response = new ChatResponseModel({
2979 > responseContent: [],
2980 > session: this,
2981 > agent: chatAgent,
2982 > slashCommand,
2983 > requestId: request.id,
2984 > isCompleteAddedRequest,
2985 > codeBlockInfos: undefined,
2986 > });
2987 >
2988 > this._requests.push(request);
2989 > markChat(this.sessionResource, ChatPerfMark.RequestUiUpdated);
2990 > this._onDidChange.fire({ kind: 'addRequest', request });
2991 > return request;
2992 > }
2993 > chatModel.ts
2994 > public setCustomTitle(title: string): void {
2995 this._customTitle = title;
2996 this._onDidChange.fire({ kind: 'setCustomTitle', title });
2997 }
2998 > chatModel.ts
2999 > updateRequest(request: ChatRequestModel, variableData: IChatRequestVariableData) {
3000 request.variableData = variableData;
3001 this._onDidChange.fire({ kind: 'changedRequest', request });
3002 }
3003 > chatModel.ts
3004 > adoptRequest(request: ChatRequestModel): void {
3005 // this doesn't use `removeRequest` because it must not dispose the request object
3006 const oldOwner = request.session;
3020 this._onDidChange.fire({ kind: 'addRequest', request });
3021 }
3022 > chatModel.ts
3023 > acceptResponseProgress(request: ChatRequestModel, progress: IChatProgress, quiet?: boolean): void {
3024 if (!request.response) {
3025 request.response = new ChatResponseModel({
3053 }
3054 }
3055 > chatModel.ts
3056 > removeRequest(id: string, reason: ChatRequestRemovalReason = ChatRequestRemovalReason.Removal): void {
3057 const index = this._requests.findIndex(request => request.id === id);
3058 const request = this._requests[index];
3064 }
3065 }
3066 > chatModel.ts
3067 > cancelRequest(request: ChatRequestModel): void {
3068 if (request.response) {
3069 request.response.cancel();
3070 }
3071 }
3072 > chatModel.ts
3073 > setResponse(request: ChatRequestModel, result: IChatAgentResult): void {
3074 if (!request.response) {
3075 request.response = new ChatResponseModel({
3083 request.response.setResult(result);
3084 }
3085 > chatModel.ts
3086 > setFollowups(request: ChatRequestModel, followups: IChatFollowup[] | undefined): void {
3087 if (!request.response) {
3088 // Maybe something went wrong?
3091 request.response.setFollowups(followups);
3092 }
3093 > chatModel.ts
3094 > setResponseModel(request: ChatRequestModel, response: ChatResponseModel): void {
3095 request.response = response;
3096 this._onDidChange.fire({ kind: 'addResponse', response });
3097 }
3098 > chatModel.ts
3099 > toExport(): IExportableChatData {
3100 return {
3101 responderUsername: this.responderUsername,
3142 };
3143 }
3144 > chatModel.ts
3145 > toJSON(): ISerializableChatData {
3146 return {
3147 version: 3,
3154 };
3155 }
3156 > chatModel.ts
3157 > override dispose() {
3158 > this._requests.forEach(r => r.response?.dispose()); chatModel.ts
3159 > this._onDidDispose.fire();
3160 >
3161 > super.dispose();
3162 >
3163 > // Null out heavy fields to break retention chains. Even after disposal,
3164 > // stale references (closures, cached templates, etc.) may prevent GC
3165 > // from collecting this object. Clearing these fields ensures the
3166 > // conversation data, serialization snapshot, and editing session are
3167 > // freed regardless.
3168 > this._requests.length = 0;
3169 > this.dataSerializer = undefined;
3170 > this._editingSession = undefined;
3171 > }
3172 > } chatModel.ts
3173 >
3174 > export function updateRanges(variableData: IChatRequestVariableData, diff: number): IChatRequestVariableData {
3175 return {
3176 variables: variableData.variables.map(v => ({
3183 };
3184 }
3185 > chatModel.ts
3186 > export function canMergeMarkdownStrings(md1: IMarkdownString, md2: IMarkdownString): boolean {
3187 if (md1.baseUri && md2.baseUri) {
3188 const baseUriEquals = md1.baseUri.scheme === md2.baseUri.scheme
3202 md1.supportThemeIcons === md2.supportThemeIcons;
3203 }
3204 > chatModel.ts
3205 > export function appendMarkdownString(md1: IMarkdownString, md2: IMarkdownString | string): IMarkdownString {
3206 const appendedValue = typeof md2 === 'string' ? md2 : md2.value;
3207 return {
3213 };
3214 }
3215 > chatModel.ts
3216 > export function getCodeCitationsMessage(citations: ReadonlyArray<IChatCodeCitation>): string {
3217 if (citations.length === 0) {
3218 return '';
3225 return label;
3226 }
3227 > chatModel.ts
3228 > /**
3229 > * Converts IChatSendRequestOptions to a serializable format by extracting only
3230 > * serializable fields and converting observables to static values.
3231 > */
3232 > export function serializeSendOptions(options: IChatSendRequestOptions): ISerializableSendOptions {
3233 return {
3234 modeInfo: options.modeInfo,
3249 };
3250 }
3251 > chatModel.ts
3252 > export enum ChatRequestEditedFileEventKind {
3253 > Keep = 1,
3254 > Undo = 2,
3255 > UserModification = 3,
3256 > }
3257 >
3258 > export interface IChatAgentEditedFileEvent {
3259 > readonly uri: URI;
3260 > readonly eventKind: ChatRequestEditedFileEventKind;
3261 > }
3262 >
3263 > /** URI for a resource embedded in a chat request/response */
3264 > export namespace ChatResponseResource {
3265 > export const scheme = 'vscode-chat-response-resource';
3266 >
3267 > export function createUri(sessionResource: URI, toolCallId: string, index: number, basename?: string): URI {
3268 return URI.from({
3269 scheme: ChatResponseResource.scheme,
3272 });
3273 }
3274 > chatModel.ts
3275 > export function parseUri(uri: URI): undefined | { sessionResource: URI; toolCallId: string; index: number } {
3276 if (uri.scheme !== ChatResponseResource.scheme) {
3277 return undefined;
3305 };
3306 }
3307 > } chatModel.ts
3308 >
3309 function _logChangesToStateModel(newState: Partial<IChatModelInputState> | undefined, oldState: Partial<IChatModelInputState> | undefined, logger: ILogService, sessionId: string) {
3310 if (!canLog(logger.getLevel(), LogLevel.Debug) || newState?.selectedModel?.identifier === oldState?.selectedModel?.identifier) {
3315 logger.debug(message);
3316 }
3317 > chatModel.ts
3318 > export function logChangesToStateModel(model: IInputModel | undefined, message: string, newState: Partial<IChatModelInputState> | undefined, oldState: Partial<IChatModelInputState> | undefined, logger: ILogService) {
3319 > if (!canLog(logger.getLevel(), LogLevel.Debug)) { chatModel.ts
3320 > return;
3321 > }
3322 message = [message,
3323 > `model.selectedModel: ${model?.state.get()?.selectedModel?.identifier}`, chatModel.ts
3324 > `new state: ${newState?.selectedModel?.identifier}`,
3325 > `old state: ${oldState?.selectedModel?.identifier}`,
3326 > new Error().stack
3327 > ].join(', ');
3328 >
3329 > logger.debug(`[ChatModelChanged] Chat Model Changed,${message}`);
3330 > }
src/vs/platform/agentHost/common/state/protocol/channels-chat/state.ts 1557 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { ModelSelection } from '../channels-root/state.js';
10 > import type { AgentSelection, McpAuthRequirement, SessionStatus } from '../channels-session/state.js';
11 > import type { ContentRef, ErrorInfo, FileEdit, StringOrMarkdown, TextRange, TextSelection, URI, UsageInfo } from '../common/state.js';
12 >
13 > // ─── Chat State ──────────────────────────────────────────────────────────────
14 >
15 > /**
16 > * Full state for a single chat, loaded when a client subscribes to the chat's
17 > * URI.
18 > *
19 > * The lightweight catalog representation of a chat is {@link ChatSummary},
20 > * carried in {@link SessionState.chats | `SessionState.chats`}. `ChatState`
21 > * **denormalizes** every {@link ChatSummary} field directly onto itself so
22 > * subscribers receive one flat object instead of having to merge a nested
23 > * `summary` sub-object. Producers MUST keep the two representations
24 > * consistent: any change to the inlined fields below SHOULD also be
25 > * announced on the parent session via the matching
26 > * {@link SessionChatUpdatedAction | `session/chatUpdated`} action.
27 > *
28 > * @category Chat State
29 > */
30 > export interface ChatState {
31 > // ── Summary fields (denormalized from ChatSummary) ─────────────────
32 > /** Chat URI */
33 > resource: URI;
34 > /** Chat title */
35 > title: string;
36 > /** Current chat status (reuses SessionStatus shape) */
37 > status: SessionStatus;
38 > /** Human-readable description of what the chat is currently doing */
39 > activity?: string;
40 > /** Last modification timestamp (ISO 8601, e.g. `"2025-03-10T18:42:03.123Z"`) */
41 > modifiedAt: string;
42 > /** How this chat came into existence */
43 > origin?: ChatOrigin;
44 > /**
45 > * How the user can interact with this chat. See {@link ChatInteractivity}.
46 > *
47 > * Supports agent-team patterns where worker chats are read-only or hidden.
48 > * Absence defaults to {@link ChatInteractivity.Full} for backward
49 > * compatibility.
50 > */
51 > interactivity?: ChatInteractivity;
52 > /**
53 > * The subset of the session's
54 > * {@link SessionState.workingDirectories | `workingDirectories`} that this
55 > * chat's agent has tool access to. Every entry MUST be present in the owning
56 > * session's `workingDirectories`; servers MUST reject a
57 > * `chat/workingDirectorySet` action that violates this constraint.
58 > *
59 > * When absent, the chat inherits the full session set. When present but empty
60 > * (not recommended), the chat has no working-directory tool access at all.
61 > *
62 > * Dispatch `chat/workingDirectorySet` / `chat/workingDirectoryRemoved` to
63 > * update the subset on a running chat.
64 > */
65 > workingDirectories?: URI[];
66 > /**
67 > * The chat's primary working directory — the distinguished root this chat is
68 > * centered on (e.g. the agent's process root for this chat, the default
69 > * location for relative paths). MUST be one of this chat's effective working
70 > * directories ({@link workingDirectories}, or the session's set when that is
71 > * absent). Present when the agent advertises
72 > * {@link MultipleWorkingDirectoriesCapability.requiresPrimary}.
73 > *
74 > * **Read-only and fixed at creation.** It is set from
75 > * {@link CreateChatParams.primaryWorkingDirectory} (or, for the session's
76 > * default chat, {@link CreateSessionParams.primaryWorkingDirectory}) and does
77 > * not change over the chat's lifetime — there is no action to mutate it, and
78 > * it does not participate in `session/chatUpdated`.
79 > */
80 > primaryWorkingDirectory?: URI;
81 >
82 > // ── Conversation contents ──────────────────────────────────────────
83 > /** Completed turns */
84 > turns: Turn[];
85 > /**
86 > * Cursor for loading older completed turns into this chat state.
87 > *
88 > * Presence means `turns` is a tail window and more historical turns are
89 > * available. Pass this opaque cursor to `fetchTurns`; the host MUST insert
90 > * the loaded turns into state and update or clear this cursor before
91 > * responding. Absence means the state contains all retained turns.
92 > */
93 > turnsNextCursor?: string;
94 > /** Currently in-progress turn */
95 > activeTurn?: ActiveTurn;
96 > /** Message to inject into the current turn at a convenient point */
97 > steeringMessage?: PendingMessage;
98 > /** Messages to send automatically as new turns after the current turn finishes */
99 > queuedMessages?: PendingMessage[];
100 > /**
101 > * The user's in-progress draft input for this chat — the message they are
102 > * composing but have not sent yet, including its
103 > * {@link Message.model | model} / {@link Message.agent | agent} selection
104 > * and attachments.
105 > *
106 > * Clients MAY periodically sync their local input state into this field so
107 > * a draft survives reloads and is visible to other clients viewing the same
108 > * chat. Eager syncing is **not** required — clients SHOULD debounce and MAY
109 > * sync only at convenient points. When presenting input UI for an existing
110 > * chat, clients SHOULD use any `draft` to initialize their input state.
111 > * Cleared (set to `undefined`) once the message is sent.
112 > */
113 > draft?: Message;
114 > /**
115 > * Additional provider-specific metadata for this chat.
116 > */
117 > _meta?: Record<string, unknown>;
118 > }
119 >
120 > /**
121 > * Lightweight catalog entry for a chat, carried in
122 > * {@link SessionState.chats | `SessionState.chats`}. The full conversation
123 > * lives in {@link ChatState}, which inlines (denormalizes) every field below.
124 > *
125 > * @category Chat State
126 > */
127 > export interface ChatSummary {
128 > /** Chat URI */
129 > resource: URI;
130 > /** Chat title */
131 > title: string;
132 > /** Current chat status (reuses SessionStatus shape) */
133 > status: SessionStatus;
134 > /** Human-readable description of what the chat is currently doing */
135 > activity?: string;
136 > /** Last modification timestamp (ISO 8601, e.g. `"2025-03-10T18:42:03.123Z"`) */
137 > modifiedAt: string;
138 > /** How this chat came into existence */
139 > origin?: ChatOrigin;
140 > /**
141 > * How the user can interact with this chat. See {@link ChatInteractivity}.
142 > *
143 > * Supports agent-team patterns where worker chats are read-only or hidden.
144 > * Absence defaults to {@link ChatInteractivity.Full} for backward
145 > * compatibility.
146 > */
147 > interactivity?: ChatInteractivity;
148 > /**
149 > * The subset of the session's working directories this chat uses.
150 > * See {@link ChatState.workingDirectories} for the full semantics.
151 > */
152 > workingDirectories?: URI[];
153 > /**
154 > * The chat's primary working directory.
155 > * See {@link ChatState.primaryWorkingDirectory} for the full semantics.
156 > */
157 > primaryWorkingDirectory?: URI;
158 > }
159 >
160 > /**
161 > * Discriminant for {@link ChatOrigin} — how a chat came into existence.
162 > *
163 > * @category Chat State
164 > */
165 > export const enum ChatOriginKind {
166 > /** User created the chat explicitly (e.g. via the host UI). */
167 > User = 'user',
168 > /** Forked from an existing chat at a specific turn. */
169 > Fork = 'fork',
170 > /** Created as an independent side conversation from a specific turn. */
171 > SideChat = 'sideChat',
172 > /** Spawned by a tool call running in another chat (e.g. a sub-agent delegation). */
173 > Tool = 'tool',
174 > }
175 >
176 > /**
177 > * Immutable selected-text snapshot captured when a side chat is created.
178 > *
179 > * The host records this exact text when it accepts `createChat`; later changes
180 > * to the source chat do not alter it.
181 > *
182 > * @category Chat State
183 > */
184 > export interface SideChatSelection {
185 > /**
186 > * Exact selected-text snapshot captured at `createChat` acceptance.
187 > *
188 > * MUST be non-empty.
189 > */
190 > text: string;
191 > /**
192 > * Optional provenance for the response part that contained {@link text} when
193 > * the host took the snapshot.
194 > *
195 > * Advisory only: this is not a live range or offset and MUST NOT be used to
196 > * recompute `text`.
197 > */
198 > responsePartId?: string;
199 > }
200 >
201 > /**
202 > * How a chat came into existence. Clients MAY use it to render
203 > * contextual UI (parent indicators, fork markers, "spawned by tool" badges).
204 > *
205 > * Fork and side-chat origins both carry a stable top-level `turnId` alongside
206 > * their discriminated `kind` value instead of snapshotting whether that turn
207 > * was active or historical at creation time. Consumers resolve the identifier
208 > * against the
209 > * source chat's current `activeTurn` or retained `turns` as needed.
210 > *
211 > * When a host accepts side-chat creation from the source chat's current active
212 > * turn, it snapshots the retained history plus that turn's current user
213 > * message and any partial assistant response already available. Later
214 > * source-turn deltas do not retroactively change the created side chat's
215 > * starting context, and once the source turn completes it is still referenced
216 > * by the same `turnId`. Side-chat origins MAY also retain an immutable
217 > * {@link SideChatSelection | selected-text snapshot} captured at acceptance
218 > * time; any `responsePartId` there is provenance only, not a range.
219 > *
220 > * The `tool` variant records a tool-spawned worker from the worker's side: its
221 > * `chat`/`toolCallId` identify the spawning tool call in the parent chat. This
222 > * is the canonical record of the spawn relationship. The same edge is surfaced
223 > * from the parent's side by {@link ToolResultSubagentContent}, whose `resource`
224 > * is this chat's URI; hosts MUST keep the two consistent.
225 > *
226 > * @category Chat State
227 > */
228 > export type ChatOrigin =
229 > | { kind: ChatOriginKind.User }
230 > | { kind: ChatOriginKind.Fork; chat: URI; turnId: string }
231 > | { kind: ChatOriginKind.SideChat; chat: URI; turnId: string; selection?: SideChatSelection }
232 > | { kind: ChatOriginKind.Tool; chat: URI; toolCallId: string };
233 >
234 > /**
235 > * How a user can interact with a chat.
236 > *
237 > * - `Full` — user can send messages and watch (default when absent)
238 > * - `ReadOnly` — user can watch but not send messages (e.g. agent team workers)
239 > * - `Hidden` — internal worker not shown in UI at all
240 > *
241 > * Supports the agent-team pattern where a lead chat is fully interactive and
242 > * worker chats are read-only (visible for observability) or hidden (internal
243 > * implementation detail). The harness sets this based on the chat's role;
244 > * the UI uses it to show appropriate controls.
245 > *
246 > * @category Chat State
247 > */
248 > export const enum ChatInteractivity {
249 > /** User can send messages and watch (default when absent) */
250 > Full = 'full',
251 > /** User can watch but not send messages */
252 > ReadOnly = 'read-only',
253 > /** Internal worker not shown in UI at all */
254 > Hidden = 'hidden',
255 > }
256 >
257 > // ─── Pending Message Types ───────────────────────────────────────────────────
258 >
259 > /**
260 > * Discriminant for pending message kinds.
261 > *
262 > * @category Pending Message Types
263 > */
264 > export const enum PendingMessageKind {
265 > /** Injected into the current turn at a convenient point */
266 > Steering = 'steering',
267 > /** Sent automatically as a new turn after the current turn finishes */
268 > Queued = 'queued',
269 > }
270 >
271 > /**
272 > * A message queued for future delivery to the agent.
273 > *
274 > * Steering messages are injected into the current turn mid-flight.
275 > * Queued messages are automatically started as new turns after the
276 > * current turn naturally finishes.
277 > *
278 > * @category Pending Message Types
279 > */
280 > export interface PendingMessage {
281 > /** Unique identifier for this pending message */
282 > id: string;
283 > /** The message that will start the next turn */
284 > message: Message;
285 > }
286 >
287 >
288 > // ─── Chat Input Types ────────────────────────────────────────────────────
289 >
290 > /**
291 > * How a client completed an input request.
292 > *
293 > * @category Chat Input Types
294 > */
295 > export const enum ChatInputResponseKind {
296 > Accept = 'accept',
297 > Decline = 'decline',
298 > Cancel = 'cancel',
299 > }
300 >
301 > /**
302 > * Question/input control kind.
303 > *
304 > * @category Chat Input Types
305 > */
306 > export const enum ChatInputQuestionKind {
307 > Text = 'text',
308 > Number = 'number',
309 > Integer = 'integer',
310 > Boolean = 'boolean',
311 > SingleSelect = 'single-select',
312 > MultiSelect = 'multi-select',
313 > }
314 >
315 > /**
316 > * A choice in a select-style question.
317 > *
318 > * @category Chat Input Types
319 > */
320 > export interface ChatInputOption {
321 > /** Stable option identifier; for MCP enum values this is the enum string */
322 > id: string;
323 > /** Display label */
324 > label: string;
325 > /** Optional secondary text */
326 > description?: string;
327 > /** Whether this option is the recommended/default choice */
328 > recommended?: boolean;
329 > }
330 >
331 > interface ChatInputQuestionBase {
332 > /** Stable question identifier used as the key in `answers` */
333 > id: string;
334 > /** Short display title */
335 > title?: string;
336 > /** Prompt shown to the user */
337 > message: string;
338 > /** Whether the user must answer this question to accept the request */
339 > required?: boolean;
340 > }
341 >
342 > /** Text question within a chat input request. */
343 > export interface ChatInputTextQuestion extends ChatInputQuestionBase {
344 > kind: ChatInputQuestionKind.Text;
345 > /** Format hint for text questions, such as `email`, `uri`, `date`, or `date-time` */
346 > format?: string;
347 > /** Minimum string length */
348 > min?: number;
349 > /** Maximum string length */
350 > max?: number;
351 > /** Default text */
352 > defaultValue?: string;
353 > }
354 >
355 > /** Numeric question within a chat input request. */
356 > export interface ChatInputNumberQuestion extends ChatInputQuestionBase {
357 > kind: ChatInputQuestionKind.Number | ChatInputQuestionKind.Integer;
358 > /**
359 > * Minimum value
360 > * @format float
361 > */
362 > min?: number;
363 > /**
364 > * Maximum value
365 > * @format float
366 > */
367 > max?: number;
368 > /**
369 > * Default numeric value
370 > * @format float
371 > */
372 > defaultValue?: number;
373 > }
374 >
375 > /** Boolean question within a chat input request. */
376 > export interface ChatInputBooleanQuestion extends ChatInputQuestionBase {
377 > kind: ChatInputQuestionKind.Boolean;
378 > /** Default boolean value */
379 > defaultValue?: boolean;
380 > }
381 >
382 > /** Single-select question within a chat input request. */
383 > export interface ChatInputSingleSelectQuestion extends ChatInputQuestionBase {
384 > kind: ChatInputQuestionKind.SingleSelect;
385 > /** Options the user may select from */
386 > options: ChatInputOption[];
387 > /** Whether the user may enter text instead of selecting an option */
388 > allowFreeformInput?: boolean;
389 > }
390 >
391 > /** Multi-select question within a chat input request. */
392 > export interface ChatInputMultiSelectQuestion extends ChatInputQuestionBase {
393 > kind: ChatInputQuestionKind.MultiSelect;
394 > /** Options the user may select from */
395 > options: ChatInputOption[];
396 > /** Whether the user may enter text in addition to selecting options */
397 > allowFreeformInput?: boolean;
398 > /** Minimum selected item count */
399 > min?: number;
400 > /** Maximum selected item count */
401 > max?: number;
402 > }
403 >
404 > /**
405 > * One question within a chat input request.
406 > *
407 > * @category Chat Input Types
408 > */
409 > export type ChatInputQuestion = ChatInputTextQuestion
410 > | ChatInputNumberQuestion
411 > | ChatInputBooleanQuestion
412 > | ChatInputSingleSelectQuestion
413 > | ChatInputMultiSelectQuestion;
414 >
415 > /**
416 > * The request payload carried by an {@link InputRequestResponsePart}.
417 > *
418 > * The server creates or replaces the containing response part with
419 > * `chat/inputRequested`. Clients sync drafts with `chat/inputAnswerChanged`
420 > * and submit responses with `chat/inputCompleted`.
421 > *
422 > * @category Chat Input Types
423 > */
424 > export interface ChatInputRequest {
425 > /** Stable request identifier */
426 > id: string;
427 > /** Display message for the request as a whole */
428 > message?: string;
429 > /** URL the user should review or open, for URL-style elicitations */
430 > url?: URI;
431 > /** Ordered questions to ask the user */
432 > questions?: ChatInputQuestion[];
433 > /** Current draft or submitted answers, keyed by question ID */
434 > answers?: Record<string, ChatInputAnswer>;
435 > }
436 >
437 > /**
438 > * Answer value kind.
439 > *
440 > * @category Chat Input Types
441 > */
442 > export const enum ChatInputAnswerValueKind {
443 > Text = 'text',
444 > Number = 'number',
445 > Boolean = 'boolean',
446 > Selected = 'selected',
447 > SelectedMany = 'selected-many',
448 > }
449 >
450 > /**
451 > * Value captured for one answer.
452 > *
453 > * @category Chat Input Types
454 > */
455 > export interface ChatInputTextAnswerValue {
456 > kind: ChatInputAnswerValueKind.Text;
457 > value: string;
458 > }
459 >
460 > export interface ChatInputNumberAnswerValue {
461 > kind: ChatInputAnswerValueKind.Number;
462 > /** @format float */
463 > value: number;
464 > }
465 >
466 > export interface ChatInputBooleanAnswerValue {
467 > kind: ChatInputAnswerValueKind.Boolean;
468 > value: boolean;
469 > }
470 >
471 > export interface ChatInputSelectedAnswerValue {
472 > kind: ChatInputAnswerValueKind.Selected;
473 > value: string;
474 > /** Free-form text entered instead of selecting an option */
475 > freeformValues?: string[];
476 > }
477 >
478 > export interface ChatInputSelectedManyAnswerValue {
479 > kind: ChatInputAnswerValueKind.SelectedMany;
480 > value: string[];
481 > /** Free-form text entered in addition to selected options */
482 > freeformValues?: string[];
483 > }
484 >
485 > export type ChatInputAnswerValue = ChatInputTextAnswerValue
486 > | ChatInputNumberAnswerValue
487 > | ChatInputBooleanAnswerValue
488 > | ChatInputSelectedAnswerValue
489 > | ChatInputSelectedManyAnswerValue;
490 >
491 > export interface ChatInputAnswered {
492 > /** Answer state */
493 > state: ChatInputAnswerState.Draft | ChatInputAnswerState.Submitted;
494 > /** Answer value */
495 > value: ChatInputAnswerValue;
496 > }
497 >
498 > export interface ChatInputSkipped {
499 > /** Answer state */
500 > state: ChatInputAnswerState.Skipped;
501 > /** Free-form reason or value captured while skipping, if any */
502 > freeformValues?: string[];
503 > }
504 >
505 > /**
506 > * Answer lifecycle state.
507 > *
508 > * @category Chat Input Types
509 > */
510 > export const enum ChatInputAnswerState {
511 > Draft = 'draft',
512 > Submitted = 'submitted',
513 > Skipped = 'skipped',
514 > }
515 >
516 > /**
517 > * Draft, submitted, or skipped answer for one question.
518 > *
519 > * @category Chat Input Types
520 > */
521 > export type ChatInputAnswer = ChatInputAnswered | ChatInputSkipped;
522 >
523 >
524 > // ─── Turn Types ──────────────────────────────────────────────────────────────
525 >
526 > /**
527 > * How a turn ended.
528 > *
529 > * @category Turn Types
530 > */
531 > export const enum TurnState {
532 > Complete = 'complete',
533 > Cancelled = 'cancelled',
534 > Error = 'error',
535 > }
536 >
537 > /**
538 > * Discriminant for {@link MessageAttachment} variants.
539 > *
540 > * @category Turn Types
541 > */
542 > export const enum MessageAttachmentKind {
543 > /** A simple, opaque attachment whose representation is described by the producer. */
544 > Simple = 'simple',
545 > /** An attachment whose data is embedded inline as a base64 string. */
546 > EmbeddedResource = 'embeddedResource',
547 > /** An attachment that references a resource by URI. */
548 > Resource = 'resource',
549 > /** An attachment that references annotations on an annotations channel. */
550 > Annotations = 'annotations',
551 > /** An attachment that references a bounded transcript from another chat. */
552 > Chat = 'chat',
553 > }
554 >
555 > /**
556 > * A completed request/response cycle.
557 > *
558 > * @category Turn Types
559 > */
560 > export interface Turn {
561 > /** Turn identifier */
562 > id: string;
563 > /** ISO 8601 timestamp when this turn started. */
564 > startedAt?: string;
565 > /** Turn duration in milliseconds. */
566 > duration?: number;
567 > /** The message that initiated the turn */
568 > message: Message;
569 > /**
570 > * All response content in stream order: text, tool calls, reasoning, and content refs.
571 > *
572 > * Consumers should derive display text by concatenating markdown parts,
573 > * and find tool calls by filtering for `ToolCall` parts.
574 > */
575 > responseParts: ResponsePart[];
576 > /** Token usage info */
577 > usage: UsageInfo | undefined;
578 > /** How the turn ended */
579 > state: TurnState;
580 > /** Error details if state is `'error'` */
581 > error?: ErrorInfo;
582 > }
583 >
584 > /**
585 > * An in-progress turn — the assistant is actively streaming.
586 > *
587 > * @category Turn Types
588 > */
589 > export interface ActiveTurn {
590 > /** Turn identifier */
591 > id: string;
592 > /** ISO 8601 timestamp when this turn started. */
593 > startedAt: string;
594 > /** The message that initiated the turn */
595 > message: Message;
596 > /**
597 > * All response content in stream order: text, tool calls, reasoning, and content refs.
598 > *
599 > * Tool call parts include `pendingPermissions` when permissions are awaiting user approval.
600 > */
601 > responseParts: ResponsePart[];
602 > /** Token usage info */
603 > usage: UsageInfo | undefined;
604 > }
605 >
606 > /**
607 > * Discriminant for {@link MessageOrigin} — identifies who produced a message.
608 > *
609 > * @category Turn Types
610 > */
611 > export enum MessageKind {
612 > /** Sent directly by the user. */
613 > User = 'user',
614 > /**
615 > * Produced by the agent itself rather than the user — for example, an agent
616 > * that seeds the first message of a chat it spawned.
617 > */
618 > Agent = 'agent',
619 > /**
620 > * Produced by a tool rather than the user — for example, a tool that spawns a
621 > * worker chat whose first message carries a seed prompt.
622 > */
623 > Tool = 'tool',
624 > /** A system-generated notification rather than a direct user message. */
625 > SystemNotification = 'systemNotification',
626 > }
627 >
628 > /**
629 > * Identifies the origin of a {@link Message} — who produced it. For the message
630 > * that initiates a turn ({@link Turn.message}), this is also the origin of the
631 > * turn; for steering or queued messages it is just the origin of that message.
632 > *
633 > * @category Turn Types
634 > */
635 > export interface MessageOrigin {
636 > /** The kind of actor that produced the message. */
637 > kind: MessageKind;
638 > }
639 >
640 > /**
641 > * A message that initiates or steers a turn. Messages can originate from the
642 > * user, the agent, a tool, or be system-generated (see {@link MessageOrigin}).
643 > *
644 > * Attachments MAY be referenced inside {@link Message.text} via their
645 > * {@link MessageAttachmentBase.range} field. Attachments without a range are
646 > * still associated with the message but do not correspond to a specific span
647 > * in the text.
648 > *
649 > * @category Turn Types
650 > */
651 > export interface Message {
652 > /** Message text */
653 > text: string;
654 > /** The origin of the message */
655 > origin: MessageOrigin;
656 > /** File/selection attachments */
657 > attachments?: MessageAttachment[];
658 > /**
659 > * The model this message was, or will be, sent with.
660 > *
661 > * For historic user/agent messages this records the model actually used, so
662 > * a client editing or resending the message can retain that selection. For a
663 > * {@link ChatState.draft | draft} it carries the model the user picked for
664 > * the message they are composing. Absent means the agent host's default
665 > * model applies.
666 > */
667 > model?: ModelSelection;
668 > /**
669 > * The custom agent this message was, or will be, sent with.
670 > *
671 > * For historic messages this records the agent actually used; for a
672 > * {@link ChatState.draft | draft} it carries the agent the user picked.
673 > * Absent means no custom agent — the provider's default behavior applies.
674 > */
675 > agent?: AgentSelection;
676 > /**
677 > * Additional provider-specific metadata for this message.
678 > *
679 > * Clients MAY look for well-known keys here to provide enhanced UI, and
680 > * agent hosts MAY use it to carry context that does not fit any other
681 > * field. Mirrors the MCP `_meta` convention.
682 > */
683 > _meta?: Record<string, unknown>;
684 > }
685 >
686 > /**
687 > * Common fields shared by all {@link MessageAttachment} variants.
688 > *
689 > * @category Turn Types
690 > */
691 > export interface MessageAttachmentBase {
692 > /**
693 > * A human-readable label for the attachment (e.g. the filename of a file
694 > * attachment). Used for display in UI.
695 > */
696 > label: string;
697 >
698 > /**
699 > * If defined, the range in {@link Message.text} that references this
700 > * attachment. This is a text range, not a byte range.
701 > */
702 > range?: TextRange;
703 >
704 > /**
705 > * Advisory display hint for clients rendering this attachment. Recognized
706 > * values include:
707 > *
708 > * - `'image'`: the attachment is an image
709 > * - `'document'`: the attachment is a textual document
710 > * - `'symbol'`: the attachment is a code symbol (e.g. a function or class)
711 > * - `'directory'`: the attachment is a folder
712 > * - `'selection'`: the attachment is a selection within a document
713 > *
714 > * Implementations MAY provide additional values; clients SHOULD fall back
715 > * to a reasonable default when an unknown value is encountered.
716 > */
717 > displayKind?: string;
718 >
719 > /**
720 > * Additional implementation-defined metadata for the attachment.
721 > *
722 > * If the attachment was produced by the `completions` command, the client
723 > * MUST preserve every property of `_meta` originally returned by the agent
724 > * host when sending the user message containing the accepted completion.
725 > */
726 > _meta?: Record<string, unknown>;
727 > }
728 >
729 > /**
730 > * A simple, opaque attachment whose model representation is described by
731 > * the producer.
732 > *
733 > * @category Turn Types
734 > */
735 > export interface SimpleMessageAttachment extends MessageAttachmentBase {
736 > /** Discriminant */
737 > type: MessageAttachmentKind.Simple;
738 >
739 > /**
740 > * Representation of the attachment as it should be shown to the model.
741 > *
742 > * If the attachment was produced by the client, this property MUST be
743 > * defined so the agent host can correctly interpret the attachment. This
744 > * property MAY be omitted when the attachment originated from a
745 > * `completions` response.
746 > */
747 > modelRepresentation?: string;
748 > }
749 >
750 > /**
751 > * An attachment whose data is embedded inline as a base64 string.
752 > *
753 > * Use this for small binary payloads (e.g. a pasted image) that should be
754 > * delivered with the user message itself rather than fetched separately.
755 > *
756 > * @category Turn Types
757 > */
758 > export interface MessageEmbeddedResourceAttachment extends MessageAttachmentBase {
759 > /** Discriminant */
760 > type: MessageAttachmentKind.EmbeddedResource;
761 > /** Base64-encoded binary data */
762 > data: string;
763 > /** Content MIME type (e.g. `"image/png"`, `"application/pdf"`) */
764 > contentType: string;
765 > /**
766 > * Optional selection within the attached textual resource.
767 > *
768 > * Only meaningful for textual resources.
769 > */
770 > selection?: TextSelection;
771 > }
772 >
773 > /**
774 > * An attachment that references a resource by URI. The content is not
775 > * delivered inline; consumers can fetch it via `resourceRead` when needed.
776 > *
777 > * @category Turn Types
778 > */
779 > export interface MessageResourceAttachment extends MessageAttachmentBase, ContentRef {
780 > /** Discriminant */
781 > type: MessageAttachmentKind.Resource;
782 > /**
783 > * Optional selection within the referenced textual resource.
784 > *
785 > * Only meaningful for textual resources.
786 > */
787 > selection?: TextSelection;
788 > }
789 >
790 > /**
791 > * An attachment that references annotations on a session's annotations
792 > * channel (see {@link AnnotationsState}).
793 > *
794 > * When {@link annotationIds} is omitted the attachment references every
795 > * annotation on the channel; when present it references only the listed
796 > * {@link Annotation.id | annotation ids}.
797 > *
798 > * @category Turn Types
799 > */
800 > export interface MessageAnnotationsAttachment extends MessageAttachmentBase {
801 > /** Discriminant */
802 > type: MessageAttachmentKind.Annotations;
803 > /**
804 > * The annotations channel URI (typically `ahp-session:/<uuid>/annotations`).
805 > * Matches {@link AnnotationsSummary.resource}.
806 > */
807 > resource: URI;
808 > /**
809 > * Specific {@link Annotation.id | annotation ids} to reference. When
810 > * omitted, the attachment references all annotations on the channel.
811 > */
812 > annotationIds?: string[];
813 > }
814 >
815 > /**
816 > * An attachment that references a chat transcript through a fixed completed
817 > * turn.
818 > *
819 > * The referenced chat MUST belong to the same session as the message's chat.
820 > * The host resolves the transcript from its first retained turn through
821 > * `endTurn`, inclusive, when accepting the message. Later turns do not
822 > * change the context represented by an already-sent attachment.
823 > *
824 > * Hosts MUST NOT recursively expand chat attachments found inside the
825 > * referenced transcript. Clients SHOULD keep rendering `label` if the
826 > * referenced chat is later pruned, and treat opening `resource` as best-effort.
827 > *
828 > * @category Turn Types
829 > */
830 > export interface MessageChatAttachment extends MessageAttachmentBase {
831 > /** Discriminant */
832 > type: MessageAttachmentKind.Chat;
833 > /** URI of the referenced chat. */
834 > resource: URI;
835 > /** Last completed turn included in the referenced transcript. */
836 > endTurn: string;
837 > }
838 >
839 > /**
840 > * An attachment associated with a {@link Message}.
841 > *
842 > * @category Turn Types
843 > */
844 > export type MessageAttachment =
845 > | SimpleMessageAttachment
846 > | MessageEmbeddedResourceAttachment
847 > | MessageResourceAttachment
848 > | MessageAnnotationsAttachment
849 > | MessageChatAttachment;
850 >
851 > // ─── Response Parts ──────────────────────────────────────────────────────────
852 >
853 > /**
854 > * Discriminant for response part types.
855 > *
856 > * @category Response Parts
857 > */
858 > export const enum ResponsePartKind {
859 > Markdown = 'markdown',
860 > ContentRef = 'contentRef',
861 > ToolCall = 'toolCall',
862 > Reasoning = 'reasoning',
863 > SystemNotification = 'systemNotification',
864 > InputRequest = 'inputRequest',
865 > }
866 >
867 > /**
868 > * @category Response Parts
869 > */
870 > export interface MarkdownResponsePart {
871 > /** Discriminant */
872 > kind: ResponsePartKind.Markdown;
873 > /** Part identifier, used by `chat/delta` to target this part for content appends */
874 > id: string;
875 > /** Markdown content */
876 > content: string;
877 > }
878 >
879 > /**
880 > * A content part that's a reference to large content stored outside the state tree.
881 > *
882 > * @category Response Parts
883 > */
884 > export interface ResourceReponsePart extends ContentRef {
885 > /** Discriminant */
886 > kind: ResponsePartKind.ContentRef;
887 > }
888 >
889 > /**
890 > * A tool call represented as a response part.
891 > *
892 > * Tool calls are part of the response stream, interleaved with text and
893 > * reasoning. The `toolCall.toolCallId` serves as the part identifier for
894 > * actions that target this part.
895 > *
896 > * @category Response Parts
897 > */
898 > export interface ToolCallResponsePart {
899 > /** Discriminant */
900 > kind: ResponsePartKind.ToolCall;
901 > /** Full tool call lifecycle state */
902 > toolCall: ToolCallState;
903 > }
904 >
905 > /**
906 > * Reasoning/thinking content from the model.
907 > *
908 > * @category Response Parts
909 > */
910 > export interface ReasoningResponsePart {
911 > /** Discriminant */
912 > kind: ResponsePartKind.Reasoning;
913 > /** Part identifier, used by `chat/reasoning` to target this part for content appends */
914 > id: string;
915 > /** Accumulated reasoning text */
916 > content: string;
917 > }
918 >
919 > /**
920 > * @category Response Parts
921 > */
922 > export type ResponsePart =
923 > | MarkdownResponsePart
924 > | ResourceReponsePart
925 > | ToolCallResponsePart
926 > | ReasoningResponsePart
927 > | SystemNotificationResponsePart
928 > | InputRequestResponsePart;
929 >
930 > /**
931 > * A live or resolved input request (elicitation) in the turn response stream.
932 > *
933 > * The server inserts the part with `chat/inputRequested`. While
934 > * {@link response} is absent, clients can update answer drafts with
935 > * `chat/inputAnswerChanged` and submit a response with `chat/inputCompleted`.
936 > * Completion updates this part in place so its stream position is stable and
937 > * the full interaction remains durable and backfillable via `fetchTurns`.
938 > *
939 > * If the turn ends without a submitted response, the unresolved part remains
940 > * in the completed turn transcript with {@link response} absent.
941 > *
942 > * @category Response Parts
943 > */
944 > export interface InputRequestResponsePart {
945 > /** Discriminant */
946 > kind: ResponsePartKind.InputRequest;
947 > /**
948 > * The request, carrying its `id`, `message`, `url`, `questions`, and current
949 > * draft or submitted `answers`.
950 > */
951 > request: ChatInputRequest;
952 > /**
953 > * How the request was resolved. Absent until a client submits `accept`,
954 > * `decline`, or `cancel` with `chat/inputCompleted`.
955 > */
956 > response?: ChatInputResponseKind;
957 > }
958 >
959 > /**
960 > * A system notification surfaced as part of the response stream.
961 > *
962 > * System notifications are messages authored by the agent harness
963 > * that need to be visible to both the agent (for situational awareness) and
964 > * the user (for transcript continuity). Examples include "background subagent
965 > * X completed" or "task Y was cancelled".
966 > *
967 > * @category Response Parts
968 > */
969 > export interface SystemNotificationResponsePart {
970 > /** Discriminant */
971 > kind: ResponsePartKind.SystemNotification;
972 > /** The text of the system notification */
973 > content: StringOrMarkdown;
974 > /**
975 > * Additional provider-specific metadata for this notification.
976 > *
977 > * A host MAY attach a machine-readable descriptor of what triggered the
978 > * notification so clients can categorize, icon, group, filter, or localize
979 > * it without parsing `content`. Clients MAY look for well-known keys here to
980 > * provide enhanced UI, and MUST render coherently from `content` alone when
981 > * `_meta` is absent or unrecognized.
982 > */
983 > _meta?: Record<string, unknown>;
984 > }
985 >
986 >
987 > // ─── Tool Call Types ─────────────────────────────────────────────────────────
988 >
989 > /**
990 > * Status of a tool call in the lifecycle state machine.
991 > *
992 > * @category Tool Call Types
993 > */
994 > export const enum ToolCallStatus {
995 > Streaming = 'streaming',
996 > PendingConfirmation = 'pending-confirmation',
997 > Running = 'running',
998 > /**
999 > * Running paused because the MCP server backing this call needs
1000 > * authentication (typically step-up auth for insufficient scope,
1001 > * surfacing mid-execution). See {@link ToolCallAuthRequiredState}.
1002 > */
1003 > AuthRequired = 'auth-required',
1004 > PendingResultConfirmation = 'pending-result-confirmation',
1005 > Completed = 'completed',
1006 > Cancelled = 'cancelled',
1007 > }
1008 >
1009 > /**
1010 > * How a tool call was confirmed for execution.
1011 > *
1012 > * - `NotNeeded` — No confirmation required (auto-approved)
1013 > * - `UserAction` — User explicitly approved
1014 > * - `Setting` — Approved by a persistent user setting
1015 > *
1016 > * @category Tool Call Types
1017 > */
1018 > export const enum ToolCallConfirmationReason {
1019 > NotNeeded = 'not-needed',
1020 > UserAction = 'user-action',
1021 > Setting = 'setting',
1022 > }
1023 >
1024 > /**
1025 > * Identifies a model judge as the source of a confirmation requirement.
1026 > *
1027 > * @category Tool Call Types
1028 > */
1029 > export const enum ToolCallRiskAssessmentKind {
1030 > Judge = 'judge',
1031 > }
1032 >
1033 > /**
1034 > * Lifecycle status of an asynchronous model-judge confirmation decision.
1035 > *
1036 > * @category Tool Call Types
1037 > */
1038 > export const enum ToolCallRiskAssessmentStatus {
1039 > Loading = 'loading',
1040 > Complete = 'complete',
1041 > }
1042 >
1043 > interface ToolCallRiskAssessmentBase {
1044 > kind: ToolCallRiskAssessmentKind;
1045 > }
1046 >
1047 > /**
1048 > * The model judge is still evaluating the tool call.
1049 > *
1050 > * @category Tool Call Types
1051 > */
1052 > export interface ToolCallRiskAssessmentLoadingState extends ToolCallRiskAssessmentBase {
1053 > status: ToolCallRiskAssessmentStatus.Loading;
1054 > }
1055 >
1056 > /**
1057 > * The model judge has completed its evaluation.
1058 > *
1059 > * @category Tool Call Types
1060 > */
1061 > export interface ToolCallRiskAssessmentCompleteState extends ToolCallRiskAssessmentBase {
1062 > status: ToolCallRiskAssessmentStatus.Complete;
1063 > reason: StringOrMarkdown;
1064 > /**
1065 > * The judge's normalized safety score, where `0` is unsafe and `1` is safe.
1066 > * @format float
1067 > */
1068 > safety: number;
1069 > }
1070 >
1071 > export type ToolCallRiskAssessment =
1072 > | ToolCallRiskAssessmentLoadingState
1073 > | ToolCallRiskAssessmentCompleteState;
1074 >
1075 > /**
1076 > * Why a tool call was cancelled.
1077 > *
1078 > * @category Tool Call Types
1079 > */
1080 > export const enum ToolCallCancellationReason {
1081 > Denied = 'denied',
1082 > Skipped = 'skipped',
1083 > ResultDenied = 'result-denied',
1084 > }
1085 >
1086 > /**
1087 > * Whether a confirmation option represents an approval or denial action.
1088 > *
1089 > * @category Tool Call Types
1090 > */
1091 > export const enum ConfirmationOptionKind {
1092 > Approve = 'approve',
1093 > Deny = 'deny',
1094 > }
1095 >
1096 > /**
1097 > * A confirmation option that the server offers for a tool call awaiting
1098 > * approval. Allows richer choices beyond simple approve/deny — for example,
1099 > * "Approve in this Session" or "Deny with reason."
1100 > *
1101 > * @category Tool Call Types
1102 > */
1103 > export interface ConfirmationOption {
1104 > /** Unique identifier for the option, returned in the confirmed action */
1105 > id: string;
1106 > /** Human-readable label displayed to the user */
1107 > label: string;
1108 > /** Whether this option represents an approval or denial */
1109 > kind: ConfirmationOptionKind;
1110 > /**
1111 > * Logical group number for visual categorisation.
1112 > *
1113 > * Clients SHOULD display options in the order they are defined and MAY
1114 > * use differing group numbers to insert dividers between logical clusters
1115 > * of options.
1116 > */
1117 > group?: number;
1118 > }
1119 >
1120 > export const enum ToolCallContributorKind {
1121 > Client = 'client',
1122 > MCP = 'mcp',
1123 > }
1124 >
1125 > export interface ToolCallClientContributor {
1126 > kind: ToolCallContributorKind.Client;
1127 > /**
1128 > * If this tool is provided by a client, the `clientId` of the owning client.
1129 > * Absent for server-side tools.
1130 > *
1131 > * When set, the identified client is responsible for executing the tool and
1132 > * dispatching `chat/toolCallComplete` with the result.
1133 > */
1134 > clientId: string;
1135 > }
1136 >
1137 > export interface ToolCallMcpContributor {
1138 > kind: ToolCallContributorKind.MCP;
1139 > /**
1140 > * Customization ID of the corresponding MCP server in {@link SessionState.customizations}.
1141 > */
1142 > customizationId: string;
1143 > }
1144 >
1145 > export type ToolCallContributor = ToolCallClientContributor | ToolCallMcpContributor;
1146 >
1147 > /**
1148 > * Metadata common to all tool call states.
1149 > *
1150 > * @category Tool Call Types
1151 > * @remarks
1152 > * Fields like `toolName` carry agent-specific identifiers on the wire despite the
1153 > * agent-agnostic design principle. These exist for debugging and logging purposes.
1154 > * A future version may move these to a separate diagnostic channel or namespace them
1155 > * more clearly.
1156 > */
1157 > interface ToolCallBase {
1158 > /** Unique tool call identifier */
1159 > toolCallId: string;
1160 > /** Internal tool name (for debugging/logging) */
1161 > toolName: string;
1162 > /** Human-readable tool name */
1163 > displayName: string;
1164 > /** Human-readable description of what the tool invocation intends to do */
1165 > intention?: string;
1166 > /**
1167 > * Reference to the contributor of the tool being called.
1168 > */
1169 > contributor?: ToolCallContributor;
1170 > /**
1171 > * Additional provider-specific metadata for this tool call.
1172 > *
1173 > * This MAY include a `ui` field corresponding to the MCP Apps (SEP-1865)
1174 > * `McpUiToolMeta` found in MCP tool calls, which may be used in combination
1175 > * with the {@link contributor} to serve MCP Apps.
1176 > */
1177 > _meta?: Record<string, unknown>;
1178 > }
1179 >
1180 > /**
1181 > * Properties available once tool call parameters are fully received.
1182 > *
1183 > * @category Tool Call Types
1184 > */
1185 > interface ToolCallParameterFields {
1186 > /** Message describing what the tool will do */
1187 > invocationMessage: StringOrMarkdown;
1188 > /** Raw tool input */
1189 > toolInput?: string;
1190 > }
1191 >
1192 > /**
1193 > * Tool execution result details, available after execution completes.
1194 > *
1195 > * @category Tool Call Types
1196 > */
1197 > export interface ToolCallResult {
1198 > /** Whether the tool succeeded */
1199 > success: boolean;
1200 > /** Past-tense description of what the tool did */
1201 > pastTenseMessage: StringOrMarkdown;
1202 > /**
1203 > * Unstructured result content blocks.
1204 > *
1205 > * This mirrors the `content` field of MCP `CallToolResult`.
1206 > */
1207 > content?: ToolResultContent[];
1208 > /**
1209 > * Optional structured result object.
1210 > *
1211 > * This mirrors the `structuredContent` field of MCP `CallToolResult`.
1212 > */
1213 > structuredContent?: Record<string, unknown>;
1214 > /** Error details if the tool failed */
1215 > error?: { message: string; code?: string };
1216 > }
1217 >
1218 > /**
1219 > * LM is streaming the tool call parameters.
1220 > *
1221 > * @category Tool Call Types
1222 > */
1223 > export interface ToolCallStreamingState extends ToolCallBase {
1224 > status: ToolCallStatus.Streaming;
1225 > /** Partial parameters accumulated so far */
1226 > partialInput?: string;
1227 > /** Progress message shown while parameters are streaming */
1228 > invocationMessage?: StringOrMarkdown;
1229 > }
1230 >
1231 > /**
1232 > * Parameters are complete, or a running tool requires re-confirmation
1233 > * (e.g. a mid-execution permission check).
1234 > *
1235 > * @category Tool Call Types
1236 > */
1237 > export interface ToolCallPendingConfirmationState extends ToolCallBase, ToolCallParameterFields {
1238 > status: ToolCallStatus.PendingConfirmation;
1239 > /** Short title for the confirmation prompt (e.g. `"Run in terminal"`, `"Write file"`) */
1240 > confirmationTitle?: StringOrMarkdown;
1241 > /** Risk assessment that informed the confirmation requirement. */
1242 > riskAssessment?: ToolCallRiskAssessment;
1243 > /** File edits that this tool call will perform, for preview before confirmation */
1244 > edits?: { items: FileEdit[] };
1245 > /** Whether the agent host allows the client to edit the tool's input parameters before confirming */
1246 > editable?: boolean;
1247 > /**
1248 > * Options the server offers for this confirmation. When present, the client
1249 > * SHOULD render these instead of a plain approve/deny UI. Each option
1250 > * belongs to a {@link ConfirmationOptionGroup} so the client can still
1251 > * categorise the choices.
1252 > */
1253 > options?: ConfirmationOption[];
1254 > }
1255 >
1256 > /**
1257 > * Fields present on every tool call state that exists **after** confirmation
1258 > * has been resolved: {@link ToolCallRunningState}, {@link ToolCallAuthRequiredState},
1259 > * {@link ToolCallPendingResultConfirmationState}, and {@link ToolCallCompletedState}.
1260 > * `ToolCallPendingConfirmationState` (not yet confirmed) and
1261 > * `ToolCallCancelledState` (the denial path — never ran) don't satisfy this
1262 > * invariant, so they keep their own `selectedOption` field independently
1263 > * rather than extending this one.
1264 > *
1265 > * @category Tool Call Types
1266 > */
1267 > interface ToolCallPostConfirmationFields {
1268 > /** How the tool was confirmed for execution */
1269 > confirmed: ToolCallConfirmationReason;
1270 > /** The confirmation option the user selected, if confirmation options were provided */
1271 > selectedOption?: ConfirmationOption;
1272 > }
1273 >
1274 > /**
1275 > * Tool is actively executing.
1276 > *
1277 > * @category Tool Call Types
1278 > */
1279 > export interface ToolCallRunningState extends ToolCallBase, ToolCallParameterFields, ToolCallPostConfirmationFields {
1280 > status: ToolCallStatus.Running;
1281 > /**
1282 > * Partial content produced while the tool is still executing.
1283 > *
1284 > * For example, a terminal content block lets clients subscribe to live
1285 > * output before the tool completes.
1286 > */
1287 > content?: ToolResultContent[];
1288 > }
1289 >
1290 > /**
1291 > * A running tool call is paused because the MCP server backing it needs
1292 > * authentication — most commonly {@link McpAuthRequirement.reason |
1293 > * `insufficientScope`} step-up auth triggered by the `tools/call` request
1294 > * itself. Only ever reached from {@link ToolCallRunningState}, and normally
1295 > * returns there once authenticated: `running` → `auth-required` → `running`
1296 > * → …. A client MAY instead cancel the invocation without authenticating by
1297 > * dispatching a `chat/toolCallComplete` with a **failed** result, always
1298 > * moving straight to {@link ToolCallCompletedState} —
1299 > * `requiresResultConfirmation` is ignored on this path, so it can never
1300 > * enter {@link ToolCallPendingResultConfirmationState}. A **successful**
1301 > * result dispatched from this state is invalid and MUST be rejected/ignored
1302 > * as a no-op by the reducer, since execution never resumed after the
1303 > * challenge.
1304 > *
1305 > * This is the tool-call-level counterpart to
1306 > * {@link McpServerAuthRequiredState} — that state means the MCP *server*
1307 > * cannot serve any request; this one means *this specific invocation* is
1308 > * waiting on the same kind of challenge. The two are dispatched
1309 > * independently and MAY be true at the same time, or not: an
1310 > * `insufficientScope` challenge triggered by a single tool call, for
1311 > * example, need not block the whole server.
1312 > *
1313 > * Because the challenge is always resolved by pushing a token via the
1314 > * existing `authenticate` command, this state can only originate from a
1315 > * tool call {@link ToolCallContributorKind.MCP | contributed by an MCP
1316 > * server} — `contributor` is narrowed accordingly (unlike the optional,
1317 > * multi-kind `contributor` on other tool call states).
1318 > *
1319 > * @category Tool Call Types
1320 > */
1321 > export interface ToolCallAuthRequiredState extends ToolCallBase, ToolCallParameterFields, ToolCallPostConfirmationFields {
1322 > status: ToolCallStatus.AuthRequired;
1323 > /** The MCP server that contributed this tool call — always MCP, never a client tool. */
1324 > contributor: ToolCallMcpContributor;
1325 > /** The authentication challenge blocking this invocation. */
1326 > auth: McpAuthRequirement;
1327 > /** Partial content produced before the call paused for authentication. */
1328 > content?: ToolResultContent[];
1329 > }
1330 >
1331 > /**
1332 > * Tool finished executing, waiting for client to approve the result.
1333 > *
1334 > * @category Tool Call Types
1335 > */
1336 > export interface ToolCallPendingResultConfirmationState extends ToolCallBase, ToolCallParameterFields, ToolCallResult, ToolCallPostConfirmationFields {
1337 > status: ToolCallStatus.PendingResultConfirmation;
1338 > }
1339 >
1340 > /**
1341 > * Tool completed successfully or with an error.
1342 > *
1343 > * @category Tool Call Types
1344 > */
1345 > export interface ToolCallCompletedState extends ToolCallBase, ToolCallParameterFields, ToolCallResult, ToolCallPostConfirmationFields {
1346 > status: ToolCallStatus.Completed;
1347 > }
1348 >
1349 > /**
1350 > * Tool call was cancelled before execution.
1351 > *
1352 > * @category Tool Call Types
1353 > */
1354 > export interface ToolCallCancelledState extends ToolCallBase, ToolCallParameterFields {
1355 > status: ToolCallStatus.Cancelled;
1356 > /** Why the tool was cancelled */
1357 > reason: ToolCallCancellationReason;
1358 > /** Optional message explaining the cancellation */
1359 > reasonMessage?: StringOrMarkdown;
1360 > /** What the user suggested doing instead */
1361 > userSuggestion?: Message;
1362 > /** The confirmation option the user selected, if confirmation options were provided */
1363 > selectedOption?: ConfirmationOption;
1364 > }
1365 >
1366 > /**
1367 > * Discriminated union of all tool call lifecycle states.
1368 > *
1369 > * See the [state model guide](/guide/state-model.html#tool-call-lifecycle)
1370 > * for the full state machine diagram.
1371 > *
1372 > * @category Tool Call Types
1373 > */
1374 > export type ToolCallState =
1375 > | ToolCallStreamingState
1376 > | ToolCallPendingConfirmationState
1377 > | ToolCallRunningState
1378 > | ToolCallAuthRequiredState
1379 > | ToolCallPendingResultConfirmationState
1380 > | ToolCallCompletedState
1381 > | ToolCallCancelledState;
1382 >
1383 > /**
1384 > * The two tool-call states that block on a client confirmation: parameter
1385 > * confirmation before execution ({@link ToolCallPendingConfirmationState}) and
1386 > * result confirmation after execution
1387 > * ({@link ToolCallPendingResultConfirmationState}).
1388 > *
1389 > * {@link ToolCallAuthRequiredState} is intentionally **not** part of this
1390 > * union: it doesn't block on a `chat/toolCallConfirmed`-style client
1391 > * decision, it blocks on the client completing an OAuth flow and calling
1392 > * `authenticate`. See {@link SessionToolAuthenticationRequest} for its
1393 > * session-level surfacing.
1394 > *
1395 > * Surfaced at the session level by {@link SessionToolConfirmationRequest}.
1396 > *
1397 > * @category Tool Call Types
1398 > */
1399 > export type ToolCallConfirmationState =
1400 > | ToolCallPendingConfirmationState
1401 > | ToolCallPendingResultConfirmationState;
1402 >
1403 >
1404 > // ─── Tool Result Content ─────────────────────────────────────────────────────
1405 >
1406 > /**
1407 > * Discriminant for tool result content types.
1408 > *
1409 > * @category Tool Result Content
1410 > */
1411 > export const enum ToolResultContentType {
1412 > Text = 'text',
1413 > EmbeddedResource = 'embeddedResource',
1414 > Resource = 'resource',
1415 > FileEdit = 'fileEdit',
1416 > Terminal = 'terminal',
1417 > Subagent = 'subagent',
1418 > }
1419 >
1420 > /**
1421 > * Text content in a tool result.
1422 > *
1423 > * Mirrors MCP `TextContent`.
1424 > *
1425 > * @category Tool Result Content
1426 > */
1427 > export interface ToolResultTextContent {
1428 > type: ToolResultContentType.Text;
1429 > /** The text content */
1430 > text: string;
1431 > }
1432 >
1433 > /**
1434 > * Base64-encoded binary content embedded in a tool result.
1435 > *
1436 > * Mirrors MCP `EmbeddedResource` for inline binary data.
1437 > *
1438 > * @category Tool Result Content
1439 > */
1440 > export interface ToolResultEmbeddedResourceContent {
1441 > type: ToolResultContentType.EmbeddedResource;
1442 > /** Base64-encoded data */
1443 > data: string;
1444 > /** Content type (e.g. `"image/png"`, `"application/pdf"`) */
1445 > contentType: string;
1446 > }
1447 >
1448 > /**
1449 > * A reference to a resource stored outside the tool result.
1450 > *
1451 > * Wraps {@link ContentRef} for lazy-loading large results.
1452 > *
1453 > * @category Tool Result Content
1454 > */
1455 > export interface ToolResultResourceContent extends ContentRef {
1456 > type: ToolResultContentType.Resource;
1457 > }
1458 >
1459 > /**
1460 > * Describes a file modification performed by a tool.
1461 > *
1462 > * @category Tool Result Content
1463 > */
1464 > export interface ToolResultFileEditContent extends FileEdit {
1465 > type: ToolResultContentType.FileEdit;
1466 > }
1467 >
1468 > /**
1469 > * A reference to a terminal whose output is relevant to this tool result.
1470 > *
1471 > * Clients can subscribe to the terminal's URI to stream its output in real
1472 > * time, providing live feedback while a tool is executing.
1473 > *
1474 > * When the command exits, {@link result} is filled in on the completed
1475 > * result, retaining the outcome for clients that did not subscribe. This
1476 > * records the command's exit, not the terminal's — the terminal may keep
1477 > * running afterwards.
1478 > *
1479 > * @category Tool Result Content
1480 > */
1481 > export interface ToolResultTerminalContent {
1482 > type: ToolResultContentType.Terminal;
1483 > /** Terminal URI (subscribable for full terminal state) */
1484 > resource: URI;
1485 > /** Display title for the terminal content */
1486 > title: string;
1487 > /**
1488 > * Whether this terminal-style resource is backed by a pseudoterminal.
1489 > * When `false`, output is plain text and clients do not need to parse
1490 > * VT sequences.
1491 > */
1492 > isPty?: boolean;
1493 > /** Outcome of the command, present once it has exited. */
1494 > result?: TerminalCommandResult;
1495 > }
1496 >
1497 > /**
1498 > * Outcome of a command run in a terminal-style tool, filled in on
1499 > * {@link ToolResultTerminalContent.result} once the command exits.
1500 > *
1501 > * @category Tool Result Content
1502 > */
1503 > export interface TerminalCommandResult {
1504 > /** Exit code from the completed command, if reported by the runtime */
1505 > exitCode?: number;
1506 > /**
1507 > * Preview of the command's output, for clients that are not subscribed
1508 > * to the terminal or that arrive after it is disposed. When `isPty` is
1509 > * `true` the preview may contain VT sequences; when `false` it is plain
1510 > * text.
1511 > */
1512 > preview?: string;
1513 > /** Whether `preview` is known to be incomplete or truncated */
1514 > truncated?: boolean;
1515 > }
1516 >
1517 > /**
1518 > * A reference, embedded in a tool result, to a worker chat spawned by the tool
1519 > * call (a sub-agent delegation), referenced by a chat URI (`ahp-chat:/...`).
1520 > *
1521 > * This is the spawning tool call's forward view of the worker. The worker chat
1522 > * records the same edge in reverse via its {@link ChatOrigin} (`kind: 'tool'`),
1523 > * whose `toolCallId` identifies the tool call that emitted this content.
1524 > *
1525 > * @category Tool Result Content
1526 > */
1527 > export interface ToolResultSubagentContent {
1528 > type: ToolResultContentType.Subagent;
1529 > /** Worker chat URI (subscribable for full chat state) */
1530 > resource: URI;
1531 > /** Display title for the subagent */
1532 > title: string;
1533 > /** Internal agent name */
1534 > agentName?: string;
1535 > /** Human-readable description of the subagent's task */
1536 > description?: string;
1537 > }
1538 >
1539 > /**
1540 > * Content block in a tool result.
1541 > *
1542 > * Mirrors the content blocks in MCP `CallToolResult.content`, plus
1543 > * `ToolResultResourceContent` for lazy-loading large results,
1544 > * `ToolResultFileEditContent` for file edit diffs,
1545 > * `ToolResultTerminalContent` for live terminal output and
1546 > * command completion metadata, and
1547 > * `ToolResultSubagentContent` for tool-spawned worker chats (AHP extensions).
1548 > *
1549 > * @category Tool Result Content
1550 > */
1551 > export type ToolResultContent =
1552 > | ToolResultTextContent
1553 > | ToolResultEmbeddedResourceContent
1554 > | ToolResultResourceContent
1555 > | ToolResultFileEditContent
1556 > | ToolResultTerminalContent
1557 > | ToolResultSubagentContent;
src/vs/workbench/common/editor.ts 1397 covered LOC · 36 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- editor.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../nls.js';
7 > import { Event } from '../../base/common/event.js';
8 > import { DeepRequiredNonNullable, assertReturnsDefined } from '../../base/common/types.js';
9 > import { URI } from '../../base/common/uri.js';
10 > import { Disposable, IDisposable, toDisposable } from '../../base/common/lifecycle.js';
11 > import { ICodeEditorViewState, IDiffEditor, IDiffEditorViewState, IEditor, IEditorViewState } from '../../editor/common/editorCommon.js';
12 > import { IEditorOptions, IResourceEditorInput, ITextResourceEditorInput, IBaseTextResourceEditorInput, IBaseUntypedEditorInput, ITextEditorOptions } from '../../platform/editor/common/editor.js';
13 > import type { EditorInput } from './editor/editorInput.js';
14 > import { IInstantiationService, IConstructorSignature, ServicesAccessor, BrandedService } from '../../platform/instantiation/common/instantiation.js';
15 > import { IContextKeyService } from '../../platform/contextkey/common/contextkey.js';
16 > import { Registry } from '../../platform/registry/common/platform.js';
17 > import { IEncodingSupport, ILanguageSupport } from '../services/textfile/common/textfiles.js';
18 > import { IEditorGroup } from '../services/editor/common/editorGroupsService.js';
19 > import { ICompositeControl, IComposite } from './composite.js';
20 > import { FileType, IFileReadLimits, IFileService } from '../../platform/files/common/files.js';
21 > import { IPathData } from '../../platform/window/common/window.js';
22 > import { IExtUri } from '../../base/common/resources.js';
23 > import { Schemas } from '../../base/common/network.js';
24 > import { IEditorService } from '../services/editor/common/editorService.js';
25 > import { ILogService } from '../../platform/log/common/log.js';
26 > import { IErrorWithActions, createErrorWithActions, isErrorWithActions } from '../../base/common/errorMessage.js';
27 > import { IAction, toAction } from '../../base/common/actions.js';
28 > import Severity from '../../base/common/severity.js';
29 > import { IPreferencesService } from '../services/preferences/common/preferences.js';
30 > import { IReadonlyEditorGroupModel } from './editor/editorGroupModel.js';
31 >
32 > // Static values for editor contributions
33 > export const EditorExtensions = {
34 > EditorPane: 'workbench.contributions.editors',
35 > EditorFactory: 'workbench.contributions.editor.inputFactories'
36 > };
37 >
38 > // Static information regarding the text editor
39 > export const DEFAULT_EDITOR_ASSOCIATION = {
40 > id: 'default',
41 > displayName: localize('promptOpenWith.defaultEditor.displayName', "Text Editor"),
42 > providerDisplayName: localize('builtinProviderDisplayName', "Built-in")
43 > };
44 >
45 > /**
46 > * Side by side editor id.
47 > */
48 > export const SIDE_BY_SIDE_EDITOR_ID = 'workbench.editor.sidebysideEditor';
49 >
50 > /**
51 > * Text diff editor id.
52 > */
53 > export const TEXT_DIFF_EDITOR_ID = 'workbench.editors.textDiffEditor';
54 >
55 > /**
56 > * Binary diff editor id.
57 > */
58 > export const BINARY_DIFF_EDITOR_ID = 'workbench.editors.binaryResourceDiffEditor';
59 >
60 > export interface IEditorDescriptor<T extends IEditorPane> {
61 >
62 > /**
63 > * The unique type identifier of the editor. All instances
64 > * of the same `IEditorPane` should have the same type
65 > * identifier.
66 > */
67 > readonly typeId: string;
68 >
69 > /**
70 > * The display name of the editor.
71 > */
72 > readonly name: string;
73 >
74 > /**
75 > * Instantiates the editor pane using the provided services.
76 > */
77 > instantiate(instantiationService: IInstantiationService, group: IEditorGroup): T;
78 >
79 > /**
80 > * Whether the descriptor is for the provided editor pane.
81 > */
82 > describes(editorPane: T): boolean;
83 > }
84 >
85 > /**
86 > * Declares that an editor hosts the full-width group header (rendered by the
87 > * editor group below the tab bar, using the group's configured header menus).
88 > */
89 > export interface IEditorHeaderActions {
90 > /** Editor-scoped instantiation service so the header toolbars' `when` clauses see the editor's context. */
91 > readonly instantiationService: IInstantiationService;
92 > }
93 >
94 > /**
95 > * The editor pane is the container for workbench editors.
96 > */
97 > export interface IEditorPane extends IComposite {
98 >
99 > /**
100 > * An event to notify when the `IEditorControl` in this
101 > * editor pane changes.
102 > *
103 > * This can be used for editor panes that are a compound
104 > * of multiple editor controls to signal that the active
105 > * editor control has changed when the user clicks around.
106 > */
107 > readonly onDidChangeControl: Event<void>;
108 >
109 > /**
110 > * An optional event to notify when the selection inside the editor
111 > * pane changed in case the editor has a selection concept.
112 > *
113 > * For example, in a text editor pane, the selection changes whenever
114 > * the cursor is set to a new location.
115 > */
116 > readonly onDidChangeSelection?: Event<IEditorPaneSelectionChangeEvent>;
117 >
118 > /**
119 > * An optional event to notify when the editor inside the pane scrolled
120 > */
121 > readonly onDidChangeScroll?: Event<void>;
122 >
123 > /**
124 > * The assigned input of this editor.
125 > */
126 > readonly input: EditorInput | undefined;
127 >
128 > /**
129 > * The assigned options of the editor.
130 > */
131 > readonly options: IEditorOptions | undefined;
132 >
133 > /**
134 > * The assigned group this editor is showing in.
135 > */
136 > readonly group: IEditorGroup;
137 >
138 > /**
139 > * The minimum width of this editor.
140 > */
141 > readonly minimumWidth: number;
142 >
143 > /**
144 > * The maximum width of this editor.
145 > */
146 > readonly maximumWidth: number;
147 >
148 > /**
149 > * The minimum height of this editor.
150 > */
151 > readonly minimumHeight: number;
152 >
153 > /**
154 > * The maximum height of this editor.
155 > */
156 > readonly maximumHeight: number;
157 >
158 > /**
159 > * An event to notify whenever minimum/maximum width/height changes.
160 > */
161 > readonly onDidChangeSizeConstraints: Event<{ width: number; height: number } | undefined>;
162 >
163 > /**
164 > * The context key service for this editor. Should be overridden by
165 > * editors that have their own ScopedContextKeyService
166 > */
167 > readonly scopedContextKeyService: IContextKeyService | undefined;
168 >
169 > /**
170 > * Returns the underlying control of this editor. Callers need to cast
171 > * the control to a specific instance as needed, e.g. by using the
172 > * `isCodeEditor` helper method to access the text code editor.
173 > *
174 > * Use the `onDidChangeControl` event to track whenever the control
175 > * changes.
176 > */
177 > getControl(): IEditorControl | undefined;
178 >
179 > /**
180 > * Returns the current view state of the editor if any.
181 > *
182 > * This method is optional to override for the editor pane
183 > * and should only be overridden when the pane can deal with
184 > * `IEditorOptions.viewState` to be applied when opening.
185 > */
186 > getViewState(): object | undefined;
187 >
188 > /**
189 > * An optional method to declare that this editor hosts the full-width group
190 > * header (rendered by the editor group below the tab bar using the group's
191 > * configured header menus), providing the editor-scoped instantiation service
192 > * so the header actions' `when` clauses evaluate in the editor's context.
193 > * Return `undefined` for no header (the default).
194 > */
195 > getHeaderActions?(): IEditorHeaderActions | undefined;
196 >
197 > /**
198 > * An optional method to return the current selection in
199 > * the editor pane in case the editor pane has a selection
200 > * concept.
201 > *
202 > * Clients of this method will typically react to the
203 > * `onDidChangeSelection` event to receive the current
204 > * selection as needed.
205 > */
206 > getSelection?(): IEditorPaneSelection | undefined;
207 >
208 > /**
209 > * An optional method to return the current scroll position
210 > * of an editor inside the pane.
211 > *
212 > * Clients of this method will typically react to the
213 > * `onDidChangeScroll` event to receive the current
214 > * scroll position as needed.
215 > */
216 > getScrollPosition?(): IEditorPaneScrollPosition;
217 >
218 > /**
219 > * An optional method to set the current scroll position
220 > * of an editor inside the pane.
221 > */
222 > setScrollPosition?(scrollPosition: IEditorPaneScrollPosition): void;
223 >
224 > /**
225 > * Finds out if this editor is visible or not.
226 > */
227 > isVisible(): boolean;
228 > }
229 >
230 > export interface IEditorPaneSelectionChangeEvent {
231 >
232 > /**
233 > * More details for how the selection was made.
234 > */
235 > reason: EditorPaneSelectionChangeReason;
236 > }
237 >
238 > export const enum EditorPaneSelectionChangeReason {
239 >
240 > /**
241 > * The selection was changed as a result of a programmatic
242 > * method invocation.
243 > *
244 > * For a text editor pane, this for example can be a selection
245 > * being restored from previous view state automatically.
246 > */
247 > PROGRAMMATIC = 1,
248 >
249 > /**
250 > * The selection was changed by the user.
251 > *
252 > * This typically means the user changed the selection
253 > * with mouse or keyboard.
254 > */
255 > USER,
256 >
257 > /**
258 > * The selection was changed as a result of editing in
259 > * the editor pane.
260 > *
261 > * For a text editor pane, this for example can be typing
262 > * in the text of the editor pane.
263 > */
264 > EDIT,
265 >
266 > /**
267 > * The selection was changed as a result of a navigation
268 > * action.
269 > *
270 > * For a text editor pane, this for example can be a result
271 > * of selecting an entry from a text outline view.
272 > */
273 > NAVIGATION,
274 >
275 > /**
276 > * The selection was changed as a result of a jump action
277 > * from within the editor pane.
278 > *
279 > * For a text editor pane, this for example can be a result
280 > * of invoking "Go to definition" from a symbol.
281 > */
282 > JUMP
283 > }
284 >
285 > export interface IEditorPaneSelection {
286 >
287 > /**
288 > * Asks to compare this selection to another selection.
289 > */
290 > compare(otherSelection: IEditorPaneSelection): EditorPaneSelectionCompareResult;
291 >
292 > /**
293 > * Asks to massage the provided `options` in a way
294 > * that the selection can be restored when the editor
295 > * is opened again.
296 > *
297 > * For a text editor this means to apply the selected
298 > * line and column as text editor options.
299 > */
300 > restore(options: IEditorOptions): IEditorOptions;
301 >
302 > /**
303 > * Only used for logging to print more info about the selection.
304 > */
305 > log?(): string;
306 > }
307 >
308 > export const enum EditorPaneSelectionCompareResult {
309 >
310 > /**
311 > * The selections are identical.
312 > */
313 > IDENTICAL = 1,
314 >
315 > /**
316 > * The selections are similar.
317 > *
318 > * For a text editor this can mean that the one
319 > * selection is in close proximity to the other
320 > * selection.
321 > *
322 > * Upstream clients may decide in this case to
323 > * not treat the selection different from the
324 > * previous one because it is not distinct enough.
325 > */
326 > SIMILAR = 2,
327 >
328 > /**
329 > * The selections are entirely different.
330 > */
331 > DIFFERENT = 3
332 > }
333 >
334 > export interface IEditorPaneWithSelection extends IEditorPane {
335 >
336 > readonly onDidChangeSelection: Event<IEditorPaneSelectionChangeEvent>;
337 >
338 > getSelection(): IEditorPaneSelection | undefined;
339 > }
340 >
341 > export function isEditorPaneWithSelection(editorPane: IEditorPane | undefined): editorPane is IEditorPaneWithSelection {
342 const candidate = editorPane as IEditorPaneWithSelection | undefined;
343
344 return !!candidate && typeof candidate.getSelection === 'function' && !!candidate.onDidChangeSelection;
345 }
346 > editor.ts
347 > export interface IEditorPaneWithScrolling extends IEditorPane {
348 >
349 > readonly onDidChangeScroll: Event<void>;
350 >
351 > getScrollPosition(): IEditorPaneScrollPosition;
352 >
353 > setScrollPosition(position: IEditorPaneScrollPosition): void;
354 > }
355 >
356 > export function isEditorPaneWithScrolling(editorPane: IEditorPane | undefined): editorPane is IEditorPaneWithScrolling {
357 const candidate = editorPane as IEditorPaneWithScrolling | undefined;
358
359 return !!candidate && typeof candidate.getScrollPosition === 'function' && typeof candidate.setScrollPosition === 'function' && !!candidate.onDidChangeScroll;
360 }
361 > editor.ts
362 > /**
363 > * Scroll position of a pane
364 > */
365 > export interface IEditorPaneScrollPosition {
366 > readonly scrollTop: number;
367 > readonly scrollLeft?: number;
368 > }
369 >
370 > /**
371 > * Try to retrieve the view state for the editor pane that
372 > * has the provided editor input opened, if at all.
373 > *
374 > * This method will return `undefined` if the editor input
375 > * is not visible in any of the opened editor panes.
376 > */
377 > export function findViewStateForEditor(input: EditorInput, group: GroupIdentifier, editorService: IEditorService): object | undefined {
378 for (const editorPane of editorService.visibleEditorPanes) {
379 if (editorPane.group.id === group && input.matches(editorPane.input)) {
384 return undefined;
385 }
386 > editor.ts
387 > /**
388 > * Overrides `IEditorPane` where `input` and `group` are known to be set.
389 > */
390 > export interface IVisibleEditorPane extends IEditorPane {
391 > readonly input: EditorInput;
392 > }
393 >
394 > /**
395 > * The text editor pane is the container for workbench text editors.
396 > */
397 > export interface ITextEditorPane extends IEditorPane {
398 >
399 > /**
400 > * Returns the underlying text editor widget of this editor.
401 > */
402 > getControl(): IEditor | undefined;
403 > }
404 >
405 > /**
406 > * The text editor pane is the container for workbench text diff editors.
407 > */
408 > export interface ITextDiffEditorPane extends IEditorPane {
409 >
410 > /**
411 > * Returns the underlying text diff editor widget of this editor.
412 > */
413 > getControl(): IDiffEditor | undefined;
414 > }
415 >
416 > /**
417 > * Marker interface for the control inside an editor pane. Callers
418 > * have to cast the control to work with it, e.g. via methods
419 > * such as `isCodeEditor(control)`.
420 > */
421 > export interface IEditorControl extends ICompositeControl { }
422 >
423 > export interface IFileEditorFactory {
424 >
425 > /**
426 > * The type identifier of the file editor.
427 > */
428 > typeId: string;
429 >
430 > /**
431 > * Creates new editor capable of showing files.
432 > */
433 > createFileEditor(resource: URI, preferredResource: URI | undefined, preferredName: string | undefined, preferredDescription: string | undefined, preferredEncoding: string | undefined, preferredLanguageId: string | undefined, preferredContents: string | undefined, instantiationService: IInstantiationService): IFileEditorInput;
434 >
435 > /**
436 > * Check if the provided object is a file editor.
437 > */
438 > isFileEditor(obj: unknown): obj is IFileEditorInput;
439 > }
440 >
441 > export interface IEditorFactoryRegistry {
442 >
443 > /**
444 > * Registers the file editor factory to use for file editors.
445 > */
446 > registerFileEditorFactory(factory: IFileEditorFactory): void;
447 >
448 > /**
449 > * Returns the file editor factory to use for file editors.
450 > */
451 > getFileEditorFactory(): IFileEditorFactory;
452 >
453 > /**
454 > * Registers a editor serializer for the given editor to the registry.
455 > * An editor serializer is capable of serializing and deserializing editor
456 > * from string data.
457 > *
458 > * @param editorTypeId the type identifier of the editor
459 > * @param serializer the editor serializer for serialization/deserialization
460 > */
461 > registerEditorSerializer<Services extends BrandedService[]>(editorTypeId: string, ctor: { new(...Services: Services): IEditorSerializer }): IDisposable;
462 >
463 > /**
464 > * Returns the editor serializer for the given editor.
465 > */
466 > getEditorSerializer(editor: EditorInput): IEditorSerializer | undefined;
467 > getEditorSerializer(editorTypeId: string): IEditorSerializer | undefined;
468 >
469 > /**
470 > * Starts the registry by providing the required services.
471 > */
472 > start(accessor: ServicesAccessor): void;
473 > }
474 >
475 > export interface IEditorSerializer {
476 >
477 > /**
478 > * Determines whether the given editor can be serialized by the serializer.
479 > */
480 > canSerialize(editor: EditorInput): boolean;
481 >
482 > /**
483 > * Returns a string representation of the provided editor that contains enough information
484 > * to deserialize back to the original editor from the deserialize() method.
485 > */
486 > serialize(editor: EditorInput): string | undefined;
487 >
488 > /**
489 > * Returns an editor from the provided serialized form of the editor. This form matches
490 > * the value returned from the serialize() method.
491 > */
492 > deserialize(instantiationService: IInstantiationService, serializedEditor: string): EditorInput | undefined;
493 > }
494 >
495 > export interface IUntitledTextResourceEditorInput extends IBaseTextResourceEditorInput {
496 >
497 > /**
498 > * Optional resource for the untitled editor. Depending on the value, the editor:
499 > * - should get a unique name if `undefined` (for example `Untitled-1`)
500 > * - should use the resource directly if the scheme is `untitled:`
501 > * - should change the scheme to `untitled:` otherwise and assume an associated path
502 > *
503 > * Untitled editors with associated path behave slightly different from other untitled
504 > * editors:
505 > * - they are dirty right when opening
506 > * - they will not ask for a file path when saving but use the associated path
507 > */
508 > readonly resource: URI | undefined;
509 > }
510 >
511 > /**
512 > * A resource side by side editor input shows 2 editors side by side but
513 > * without highlighting any differences.
514 > *
515 > * Note: both sides will be resolved as editor individually. As such, it is
516 > * possible to show 2 different editors side by side.
517 > *
518 > * @see {@link IResourceDiffEditorInput} for a variant that compares 2 editors.
519 > */
520 > export interface IResourceSideBySideEditorInput extends IBaseUntypedEditorInput {
521 >
522 > /**
523 > * The right hand side editor to open inside a side-by-side editor.
524 > */
525 > readonly primary: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
526 >
527 > /**
528 > * The left hand side editor to open inside a side-by-side editor.
529 > */
530 > readonly secondary: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
531 > }
532 >
533 > /**
534 > * A resource diff editor input compares 2 editors side by side
535 > * highlighting the differences.
536 > *
537 > * Note: both sides must be resolvable to the same editor, or
538 > * a text based presentation will be used as fallback.
539 > */
540 > export interface IResourceDiffEditorInput extends IBaseUntypedEditorInput {
541 >
542 > /**
543 > * The left hand side editor to open inside a diff editor.
544 > */
545 > readonly original: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
546 >
547 > /**
548 > * The right hand side editor to open inside a diff editor.
549 > */
550 > readonly modified: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
551 > }
552 >
553 > export interface ITextResourceDiffEditorInput extends IBaseTextResourceEditorInput {
554 >
555 > /**
556 > * The left hand side text editor to open inside a diff editor.
557 > */
558 > readonly original: Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
559 >
560 > /**
561 > * The right hand side text editor to open inside a diff editor.
562 > */
563 > readonly modified: Omit<ITextResourceEditorInput, 'options'> | Omit<IUntitledTextResourceEditorInput, 'options'>;
564 > }
565 >
566 > /**
567 > * A resource list diff editor input compares multiple resources side by side
568 > * highlighting the differences.
569 > */
570 > export interface IResourceMultiDiffEditorInput extends IBaseUntypedEditorInput {
571 > /**
572 > * A unique identifier of this multi diff editor input.
573 > * If a second multi diff editor with the same uri is opened, the existing one is revealed instead (even if the resources list is different!).
574 > */
575 > readonly multiDiffSource?: URI;
576 >
577 > /**
578 > * The list of resources to compare.
579 > * If not set, the resources are dynamically derived from the {@link multiDiffSource}.
580 > */
581 > readonly resources?: IMultiDiffEditorResource[];
582 >
583 > /**
584 > * Whether the editor should be serialized and stored for subsequent sessions.
585 > */
586 > readonly isTransient?: boolean;
587 > }
588 >
589 > export interface IMultiDiffEditorResource extends IResourceDiffEditorInput {
590 > readonly goToFileResource?: URI;
591 > }
592 > export type IResourceMergeEditorInputSide = (Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'>) & { detail?: string };
593 >
594 > /**
595 > * A resource merge editor input compares multiple editors
596 > * highlighting the differences for merging.
597 > *
598 > * Note: all sides must be resolvable to the same editor, or
599 > * a text based presentation will be used as fallback.
600 > */
601 > export interface IResourceMergeEditorInput extends IBaseUntypedEditorInput {
602 >
603 > /**
604 > * The one changed version of the file.
605 > */
606 > readonly input1: IResourceMergeEditorInputSide;
607 >
608 > /**
609 > * The second changed version of the file.
610 > */
611 > readonly input2: IResourceMergeEditorInputSide;
612 >
613 > /**
614 > * The base common ancestor of the file to merge.
615 > */
616 > readonly base: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'>;
617 >
618 > /**
619 > * The resulting output of the merge.
620 > */
621 > readonly result: Omit<IResourceEditorInput, 'options'> | Omit<ITextResourceEditorInput, 'options'>;
622 > }
623 >
624 > export function isResourceEditorInput(editor: unknown): editor is IResourceEditorInput {
625 if (isEditorInput(editor)) {
626 return false; // make sure to not accidentally match on typed editor inputs
631 return URI.isUri(candidate?.resource);
632 }
633 > editor.ts
634 > export function isResourceDiffEditorInput(editor: unknown): editor is IResourceDiffEditorInput {
635 if (isEditorInput(editor)) {
636 return false; // make sure to not accidentally match on typed editor inputs
641 return candidate?.original !== undefined && candidate.modified !== undefined;
642 }
643 > editor.ts
644 > export function isResourceMultiDiffEditorInput(editor: unknown): editor is IResourceMultiDiffEditorInput {
645 if (isEditorInput(editor)) {
646 return false; // make sure to not accidentally match on typed editor inputs
657 return !!candidate.resources || !!candidate.multiDiffSource;
658 }
659 > editor.ts
660 > export function isResourceSideBySideEditorInput(editor: unknown): editor is IResourceSideBySideEditorInput {
661 if (isEditorInput(editor)) {
662 return false; // make sure to not accidentally match on typed editor inputs
671 return candidate?.primary !== undefined && candidate.secondary !== undefined;
672 }
673 > editor.ts
674 > export function isUntitledResourceEditorInput(editor: unknown): editor is IUntitledTextResourceEditorInput {
675 if (isEditorInput(editor)) {
676 return false; // make sure to not accidentally match on typed editor inputs
684 return candidate.resource === undefined || candidate.resource.scheme === Schemas.untitled || candidate.forceUntitled === true;
685 }
686 > editor.ts
687 > export function isResourceMergeEditorInput(editor: unknown): editor is IResourceMergeEditorInput {
688 if (isEditorInput(editor)) {
689 return false; // make sure to not accidentally match on typed editor inputs
694 return URI.isUri(candidate?.base?.resource) && URI.isUri(candidate?.input1?.resource) && URI.isUri(candidate?.input2?.resource) && URI.isUri(candidate?.result?.resource);
695 }
696 > editor.ts
697 > export const enum Verbosity {
698 > SHORT,
699 > MEDIUM,
700 > LONG
701 > }
702 >
703 > export const enum SaveReason {
704 >
705 > /**
706 > * Explicit user gesture.
707 > */
708 > EXPLICIT = 1,
709 >
710 > /**
711 > * Auto save after a timeout.
712 > */
713 > AUTO = 2,
714 >
715 > /**
716 > * Auto save after editor focus change.
717 > */
718 > FOCUS_CHANGE = 3,
719 >
720 > /**
721 > * Auto save after window change.
722 > */
723 > WINDOW_CHANGE = 4
724 > }
725 >
726 > export type SaveSource = string;
727 >
728 > interface ISaveSourceDescriptor {
729 > source: SaveSource;
730 > label: string;
731 > }
732 >
733 > class SaveSourceFactory {
734 >
735 > private readonly mapIdToSaveSource = new Map<SaveSource, ISaveSourceDescriptor>();
736 >
737 > /**
738 > * Registers a `SaveSource` with an identifier and label
739 > * to the registry so that it can be used in save operations.
740 > */
741 > registerSource(id: string, label: string): SaveSource {
742 let sourceDescriptor = this.mapIdToSaveSource.get(id);
743 if (!sourceDescriptor) {
748 return sourceDescriptor.source;
749 }
750 > editor.ts
751 > getSourceLabel(source: SaveSource): string {
752 return this.mapIdToSaveSource.get(source)?.label ?? source;
753 }
754 > } editor.ts
755 >
756 > export const SaveSourceRegistry = new SaveSourceFactory();
757 >
758 > export interface ISaveOptions {
759 >
760 > /**
761 > * An indicator how the save operation was triggered.
762 > */
763 > reason?: SaveReason;
764 >
765 > /**
766 > * An indicator about the source of the save operation.
767 > *
768 > * Must use `SaveSourceRegistry.registerSource()` to obtain.
769 > */
770 > readonly source?: SaveSource;
771 >
772 > /**
773 > * Forces to save the contents of the working copy
774 > * again even if the working copy is not dirty.
775 > */
776 > readonly force?: boolean;
777 >
778 > /**
779 > * Instructs the save operation to skip any save participants.
780 > */
781 > readonly skipSaveParticipants?: boolean;
782 >
783 > /**
784 > * A hint as to which file systems should be available for saving.
785 > */
786 > readonly availableFileSystems?: string[];
787 > }
788 >
789 > export interface IRevertOptions {
790 >
791 > /**
792 > * Forces to load the contents of the working copy
793 > * again even if the working copy is not dirty.
794 > */
795 > readonly force?: boolean;
796 >
797 > /**
798 > * A soft revert will clear dirty state of a working copy
799 > * but will not attempt to load it from its persisted state.
800 > *
801 > * This option may be used in scenarios where an editor is
802 > * closed and where we do not require to load the contents.
803 > */
804 > readonly soft?: boolean;
805 > }
806 >
807 > export interface IMoveResult {
808 > editor: EditorInput | IUntypedEditorInput;
809 > options?: IEditorOptions;
810 > }
811 >
812 > export const enum EditorInputCapabilities {
813 >
814 > /**
815 > * Signals no specific capability for the input.
816 > */
817 > None = 0,
818 >
819 > /**
820 > * Signals that the input is readonly.
821 > */
822 > Readonly = 1 << 1,
823 >
824 > /**
825 > * Signals that the input is untitled.
826 > */
827 > Untitled = 1 << 2,
828 >
829 > /**
830 > * Signals that the input can only be shown in one group
831 > * and not be split into multiple groups.
832 > */
833 > Singleton = 1 << 3,
834 >
835 > /**
836 > * Signals that the input requires workspace trust.
837 > */
838 > RequiresTrust = 1 << 4,
839 >
840 > /**
841 > * Signals that the editor can split into 2 in the same
842 > * editor group.
843 > */
844 > CanSplitInGroup = 1 << 5,
845 >
846 > /**
847 > * Signals that the editor wants its description to be
848 > * visible when presented to the user. By default, a UI
849 > * component may decide to hide the description portion
850 > * for brevity.
851 > */
852 > ForceDescription = 1 << 6,
853 >
854 > /**
855 > * Signals that the editor supports dropping into the
856 > * editor by holding shift.
857 > */
858 > CanDropIntoEditor = 1 << 7,
859 >
860 > /**
861 > * Signals that the editor is composed of multiple editors
862 > * within.
863 > */
864 > MultipleEditors = 1 << 8,
865 >
866 > /**
867 > * Signals that the editor cannot be in a dirty state
868 > * and may still have unsaved changes
869 > */
870 > Scratchpad = 1 << 9,
871 >
872 > /**
873 > * Signals that the editor should be revealed when being
874 > * opened if it is already opened in any editor group.
875 > */
876 > ForceReveal = 1 << 10,
877 >
878 > /**
879 > * Signals that the editor must be opened in a modal editor
880 > * part. This is honored unless the user has explicitly opted
881 > * out of modal editors via `workbench.editor.useModal: 'off'`.
882 > */
883 > RequiresModal = 1 << 11,
884 >
885 > /**
886 > * Signals that the editor is exempt from the opened editors
887 > * limit (`workbench.editor.limit`): it never counts towards the
888 > * limit and is never auto-closed to satisfy it.
889 > */
890 > ExcludeFromEditorLimit = 1 << 12
891 > }
892 >
893 > export type IUntypedEditorInput = IResourceEditorInput | ITextResourceEditorInput | IUntitledTextResourceEditorInput | IResourceDiffEditorInput | IResourceMultiDiffEditorInput | IResourceSideBySideEditorInput | IResourceMergeEditorInput;
894 >
895 > export abstract class AbstractEditorInput extends Disposable {
896 > // Marker class for implementing `isEditorInput`
897 > }
898 >
899 > export function isEditorInput(editor: unknown): editor is EditorInput {
900 return editor instanceof AbstractEditorInput;
901 }
902 > editor.ts
903 > export interface EditorInputWithPreferredResource {
904 >
905 > /**
906 > * An editor may provide an additional preferred resource alongside
907 > * the `resource` property. While the `resource` property serves as
908 > * unique identifier of the editor that should be used whenever we
909 > * compare to other editors, the `preferredResource` should be used
910 > * in places where e.g. the resource is shown to the user.
911 > *
912 > * For example: on Windows and macOS, the same URI with different
913 > * casing may point to the same file. The editor may chose to
914 > * "normalize" the URIs so that only one editor opens for different
915 > * URIs. But when displaying the editor label to the user, the
916 > * preferred URI should be used.
917 > *
918 > * Not all editors have a `preferredResource`. The `EditorResourceAccessor`
919 > * utility can be used to always get the right resource without having
920 > * to do instanceof checks.
921 > */
922 > readonly preferredResource: URI;
923 > }
924 >
925 function isEditorInputWithPreferredResource(editor: unknown): editor is EditorInputWithPreferredResource {
926 const candidate = editor as EditorInputWithPreferredResource | undefined;
928 return URI.isUri(candidate?.preferredResource);
929 }
930 > editor.ts
931 > export interface ISideBySideEditorInput extends EditorInput {
932 >
933 > /**
934 > * The primary editor input is shown on the right hand side.
935 > */
936 > primary: EditorInput;
937 >
938 > /**
939 > * The secondary editor input is shown on the left hand side.
940 > */
941 > secondary: EditorInput;
942 > }
943 >
944 > export function isSideBySideEditorInput(editor: unknown): editor is ISideBySideEditorInput {
945 const candidate = editor as ISideBySideEditorInput | undefined;
946
947 return isEditorInput(candidate?.primary) && isEditorInput(candidate?.secondary);
948 }
949 > editor.ts
950 > export interface IDiffEditorInput extends EditorInput {
951 >
952 > /**
953 > * The modified (primary) editor input is shown on the right hand side.
954 > */
955 > modified: EditorInput;
956 >
957 > /**
958 > * The original (secondary) editor input is shown on the left hand side.
959 > */
960 > original: EditorInput;
961 > }
962 >
963 > export function isDiffEditorInput(editor: unknown): editor is IDiffEditorInput {
964 const candidate = editor as IDiffEditorInput | undefined;
965
966 return isEditorInput(candidate?.modified) && isEditorInput(candidate?.original);
967 }
968 > editor.ts
969 > export interface IUntypedFileEditorInput extends ITextResourceEditorInput {
970 >
971 > /**
972 > * A marker to create a `IFileEditorInput` from this untyped input.
973 > */
974 > forceFile: true;
975 > }
976 >
977 > /**
978 > * This is a tagging interface to declare an editor input being capable of dealing with files. It is only used in the editor registry
979 > * to register this kind of input to the platform.
980 > */
981 > export interface IFileEditorInput extends EditorInput, IEncodingSupport, ILanguageSupport, EditorInputWithPreferredResource {
982 >
983 > /**
984 > * Gets the resource this file input is about. This will always be the
985 > * canonical form of the resource, so it may differ from the original
986 > * resource that was provided to create the input. Use `preferredResource`
987 > * for the form as it was created.
988 > */
989 > readonly resource: URI;
990 >
991 > /**
992 > * Sets the preferred resource to use for this file input.
993 > */
994 > setPreferredResource(preferredResource: URI): void;
995 >
996 > /**
997 > * Sets the preferred name to use for this file input.
998 > *
999 > * Note: for certain file schemes the input may decide to ignore this
1000 > * name and use our standard naming. Specifically for schemes we own,
1001 > * we do not let others override the name.
1002 > */
1003 > setPreferredName(name: string): void;
1004 >
1005 > /**
1006 > * Sets the preferred description to use for this file input.
1007 > *
1008 > * Note: for certain file schemes the input may decide to ignore this
1009 > * description and use our standard naming. Specifically for schemes we own,
1010 > * we do not let others override the description.
1011 > */
1012 > setPreferredDescription(description: string): void;
1013 >
1014 > /**
1015 > * Sets the preferred encoding to use for this file input.
1016 > */
1017 > setPreferredEncoding(encoding: string): void;
1018 >
1019 > /**
1020 > * Sets the preferred language id to use for this file input.
1021 > */
1022 > setPreferredLanguageId(languageId: string): void;
1023 >
1024 > /**
1025 > * Sets the preferred contents to use for this file input.
1026 > */
1027 > setPreferredContents(contents: string): void;
1028 >
1029 > /**
1030 > * Forces this file input to open as binary instead of text.
1031 > */
1032 > setForceOpenAsBinary(): void;
1033 >
1034 > /**
1035 > * Figure out if the file input has been resolved or not.
1036 > */
1037 > isResolved(): boolean;
1038 > }
1039 >
1040 > export interface IFileLimitedEditorInputOptions extends IEditorOptions {
1041 >
1042 > /**
1043 > * If provided, the size of the file will be checked against the limits
1044 > * and an error will be thrown if any limit is exceeded.
1045 > */
1046 > readonly limits?: IFileReadLimits;
1047 > }
1048 >
1049 > export interface IFileEditorInputOptions extends ITextEditorOptions, IFileLimitedEditorInputOptions { }
1050 >
1051 > export function createTooLargeFileError(group: IEditorGroup, input: EditorInput, options: IEditorOptions | undefined, message: string, preferencesService: IPreferencesService): Error {
1052 return createEditorOpenError(message, [
1053 toAction({
1073 });
1074 }
1075 > editor.ts
1076 > export interface EditorInputWithOptions {
1077 > editor: EditorInput;
1078 > options?: IEditorOptions;
1079 > }
1080 >
1081 > export interface EditorInputWithOptionsAndGroup extends EditorInputWithOptions {
1082 > group: IEditorGroup;
1083 > }
1084 >
1085 > export function isEditorInputWithOptions(editor: unknown): editor is EditorInputWithOptions {
1086 const candidate = editor as EditorInputWithOptions | undefined;
1087
1088 return isEditorInput(candidate?.editor);
1089 }
1090 > editor.ts
1091 > export function isEditorInputWithOptionsAndGroup(editor: unknown): editor is EditorInputWithOptionsAndGroup {
1092 const candidate = editor as EditorInputWithOptionsAndGroup | undefined;
1093
1094 return isEditorInputWithOptions(editor) && candidate?.group !== undefined;
1095 }
1096 > editor.ts
1097 > /**
1098 > * Context passed into `EditorPane#setInput` to give additional
1099 > * context information around why the editor was opened.
1100 > */
1101 > export interface IEditorOpenContext {
1102 >
1103 > /**
1104 > * An indicator if the editor input is new for the group the editor is in.
1105 > * An editor is new for a group if it was not part of the group before and
1106 > * otherwise was already opened in the group and just became the active editor.
1107 > *
1108 > * This hint can e.g. be used to decide whether to restore view state or not.
1109 > */
1110 > newInGroup?: boolean;
1111 > }
1112 >
1113 > export interface IEditorIdentifier {
1114 > groupId: GroupIdentifier;
1115 > editor: EditorInput;
1116 > }
1117 >
1118 > export function isEditorIdentifier(identifier: unknown): identifier is IEditorIdentifier {
1119 const candidate = identifier as IEditorIdentifier | undefined;
1120
1121 return typeof candidate?.groupId === 'number' && isEditorInput(candidate.editor);
1122 }
1123 > editor.ts
1124 > /**
1125 > * The editor commands context is used for editor commands (e.g. in the editor title)
1126 > * and we must ensure that the context is serializable because it potentially travels
1127 > * to the extension host!
1128 > */
1129 > export interface IEditorCommandsContext {
1130 > groupId: GroupIdentifier;
1131 > editorIndex?: number;
1132 >
1133 > preserveFocus?: boolean;
1134 > }
1135 >
1136 > export function isEditorCommandsContext(context: unknown): context is IEditorCommandsContext {
1137 const candidate = context as IEditorCommandsContext | undefined;
1138
1139 return typeof candidate?.groupId === 'number';
1140 }
1141 > editor.ts
1142 > /**
1143 > * More information around why an editor was closed in the model.
1144 > */
1145 > export enum EditorCloseContext {
1146 >
1147 > /**
1148 > * No specific context for closing (e.g. explicit user gesture).
1149 > */
1150 > UNKNOWN,
1151 >
1152 > /**
1153 > * The editor closed because it was replaced with another editor.
1154 > * This can either happen via explicit replace call or when an
1155 > * editor is in preview mode and another editor opens.
1156 > */
1157 > REPLACE,
1158 >
1159 > /**
1160 > * The editor closed as a result of moving it to another group.
1161 > */
1162 > MOVE,
1163 >
1164 > /**
1165 > * The editor closed because another editor turned into preview
1166 > * and this used to be the preview editor before.
1167 > */
1168 > UNPIN
1169 > }
1170 >
1171 > export interface IEditorCloseEvent extends IEditorIdentifier {
1172 >
1173 > /**
1174 > * More information around why the editor was closed.
1175 > */
1176 > readonly context: EditorCloseContext;
1177 >
1178 > /**
1179 > * The index of the editor before closing.
1180 > */
1181 > readonly index: number;
1182 >
1183 > /**
1184 > * Whether the editor was sticky or not.
1185 > */
1186 > readonly sticky: boolean;
1187 > }
1188 >
1189 > export interface IActiveEditorChangeEvent {
1190 >
1191 > /**
1192 > * The new active editor or `undefined` if the group is empty.
1193 > */
1194 > editor: EditorInput | undefined;
1195 >
1196 > /**
1197 > * Indicates whether the editor change is the result of an explicit
1198 > * user action (`true`) or happened automatically as a side effect
1199 > * (e.g. the chat agent opening files it has edited).
1200 > *
1201 > * When omitted, callers should treat the change as explicit.
1202 > */
1203 > isExplicit?: boolean;
1204 > }
1205 >
1206 > export interface IEditorWillMoveEvent extends IEditorIdentifier {
1207 >
1208 > /**
1209 > * The target group of the move operation.
1210 > */
1211 > readonly target: GroupIdentifier;
1212 > }
1213 >
1214 > export interface IEditorWillOpenEvent extends IEditorIdentifier { }
1215 >
1216 > export interface IWillInstantiateEditorPaneEvent {
1217 >
1218 > /**
1219 > * @see {@link IEditorDescriptor.typeId}
1220 > */
1221 > readonly typeId: string;
1222 > }
1223 >
1224 > export type GroupIdentifier = number;
1225 >
1226 > export const enum GroupModelChangeKind {
1227 >
1228 > /* Group Changes */
1229 > GROUP_ACTIVE,
1230 > GROUP_INDEX,
1231 > GROUP_LABEL,
1232 > GROUP_LOCKED,
1233 >
1234 > /* Editors Change */
1235 > EDITORS_SELECTION,
1236 >
1237 > /* Editor Changes */
1238 > EDITOR_OPEN,
1239 > EDITOR_CLOSE,
1240 > EDITOR_MOVE,
1241 > EDITOR_ACTIVE,
1242 > EDITOR_LABEL,
1243 > EDITOR_CAPABILITIES,
1244 > EDITOR_PIN,
1245 > EDITOR_TRANSIENT,
1246 > EDITOR_STICKY,
1247 > EDITOR_DIRTY,
1248 > EDITOR_WILL_DISPOSE
1249 > }
1250 >
1251 > export interface IWorkbenchEditorConfiguration {
1252 > workbench?: {
1253 > editor?: IEditorPartConfiguration;
1254 > iconTheme?: string;
1255 > };
1256 > }
1257 >
1258 > interface IEditorPartLimitConfiguration {
1259 > enabled?: boolean;
1260 > excludeDirty?: boolean;
1261 > value?: number;
1262 > perEditorGroup?: boolean;
1263 > }
1264 >
1265 > export interface IEditorPartLimitOptions extends Required<IEditorPartLimitConfiguration> { }
1266 >
1267 > interface IEditorPartDecorationsConfiguration {
1268 > badges?: boolean;
1269 > colors?: boolean;
1270 > }
1271 >
1272 > export interface IEditorPartDecorationOptions extends Required<IEditorPartDecorationsConfiguration> { }
1273 >
1274 > interface IEditorPartConfiguration {
1275 > showTabs?: 'multiple' | 'single' | 'none';
1276 > wrapTabs?: boolean;
1277 > scrollToSwitchTabs?: boolean;
1278 > highlightModifiedTabs?: boolean;
1279 > tabActionLocation?: 'left' | 'right';
1280 > tabActionCloseVisibility?: boolean;
1281 > tabActionUnpinVisibility?: boolean;
1282 > showTabIndex?: boolean;
1283 > alwaysShowEditorActions?: boolean;
1284 > tabSizing?: 'fit' | 'shrink' | 'fixed';
1285 > tabSizingFixedMinWidth?: number;
1286 > tabSizingFixedMaxWidth?: number;
1287 > pinnedTabSizing?: 'normal' | 'compact' | 'shrink';
1288 > pinnedTabsOnSeparateRow?: boolean;
1289 > tabHeight?: 'default' | 'compact';
1290 > preventPinnedEditorClose?: PreventPinnedEditorClose;
1291 > titleScrollbarSizing?: 'default' | 'large';
1292 > titleScrollbarVisibility?: 'auto' | 'visible' | 'hidden';
1293 > focusRecentEditorAfterClose?: boolean;
1294 > showIcons?: boolean;
1295 > enablePreview?: boolean;
1296 > enablePreviewFromQuickOpen?: boolean;
1297 > enablePreviewFromCodeNavigation?: boolean;
1298 > closeOnFileDelete?: boolean;
1299 > openPositioning?: 'left' | 'right' | 'first' | 'last';
1300 > openSideBySideDirection?: 'right' | 'down';
1301 > closeEmptyGroups?: boolean;
1302 > autoLockGroups?: Set<string>;
1303 > revealIfOpen?: boolean;
1304 > swipeToNavigate?: boolean;
1305 > mouseBackForwardToNavigate?: boolean;
1306 > labelFormat?: 'default' | 'short' | 'medium' | 'long';
1307 > restoreViewState?: boolean;
1308 > splitInGroupLayout?: 'vertical' | 'horizontal';
1309 > splitSizing?: 'auto' | 'split' | 'distribute';
1310 > splitOnDragAndDrop?: boolean;
1311 > allowDropIntoGroup?: boolean;
1312 > dragToOpenWindow?: boolean;
1313 > centeredLayoutFixedWidth?: boolean;
1314 > doubleClickTabToToggleEditorGroupSizes?: 'maximize' | 'expand' | 'off';
1315 > editorActionsLocation?: 'default' | 'titleBar' | 'hidden';
1316 > limit?: IEditorPartLimitConfiguration;
1317 > decorations?: IEditorPartDecorationsConfiguration;
1318 > }
1319 >
1320 > export interface IEditorPartOptions extends DeepRequiredNonNullable<IEditorPartConfiguration> {
1321 > hasIcons: boolean;
1322 > }
1323 >
1324 > export interface IEditorPartOptionsChangeEvent {
1325 > oldPartOptions: IEditorPartOptions;
1326 > newPartOptions: IEditorPartOptions;
1327 > }
1328 >
1329 > export enum SideBySideEditor {
1330 > PRIMARY = 1,
1331 > SECONDARY = 2,
1332 > BOTH = 3,
1333 > ANY = 4
1334 > }
1335 >
1336 > export interface IFindEditorOptions {
1337 >
1338 > /**
1339 > * Whether to consider any or both side by side editor as matching.
1340 > * By default, side by side editors will not be considered
1341 > * as matching, even if the editor is opened in one of the sides.
1342 > */
1343 > supportSideBySide?: SideBySideEditor.PRIMARY | SideBySideEditor.SECONDARY | SideBySideEditor.ANY;
1344 >
1345 > /**
1346 > * The order in which to consider editors for finding.
1347 > */
1348 > order?: EditorsOrder;
1349 > }
1350 >
1351 > export interface IMatchEditorOptions {
1352 >
1353 > /**
1354 > * Whether to consider a side by side editor as matching.
1355 > * By default, side by side editors will not be considered
1356 > * as matching, even if the editor is opened in one of the sides.
1357 > */
1358 > supportSideBySide?: SideBySideEditor.ANY | SideBySideEditor.BOTH;
1359 >
1360 > /**
1361 > * Only consider an editor to match when the
1362 > * `candidate === editor` but not when
1363 > * `candidate.matches(editor)`.
1364 > */
1365 > strictEquals?: boolean;
1366 > }
1367 >
1368 > export interface IEditorResourceAccessorOptions {
1369 >
1370 > /**
1371 > * Allows to access the `resource(s)` of side by side editors. If not
1372 > * specified, a `resource` for a side by side editor will always be
1373 > * `undefined`.
1374 > */
1375 > supportSideBySide?: SideBySideEditor;
1376 >
1377 > /**
1378 > * Allows to filter the scheme to consider. A resource scheme that does
1379 > * not match a filter will not be considered.
1380 > */
1381 > filterByScheme?: string | string[];
1382 > }
1383 >
1384 > class EditorResourceAccessorImpl {
1385 >
1386 > /**
1387 > * The original URI of an editor is the URI that was used originally to open
1388 > * the editor and should be used whenever the URI is presented to the user,
1389 > * e.g. as a label together with utility methods such as `ResourceLabel` or
1390 > * `ILabelService` that can turn this original URI into the best form for
1391 > * presenting.
1392 > *
1393 > * In contrast, the canonical URI (#getCanonicalUri) may be different and should
1394 > * be used whenever the URI is used to e.g. compare with other editors or when
1395 > * caching certain data based on the URI.
1396 > *
1397 > * For example: on Windows and macOS, the same file URI with different casing may
1398 > * point to the same file. The editor may chose to "normalize" the URI into a canonical
1399 > * form so that only one editor opens for same file URIs with different casing. As
1400 > * such, the original URI and the canonical URI can be different.
1401 > */
1402 > getOriginalUri(editor: EditorInput | IUntypedEditorInput | undefined | null): URI | undefined;
1403 > getOriginalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options: IEditorResourceAccessorOptions & { supportSideBySide?: SideBySideEditor.PRIMARY | SideBySideEditor.SECONDARY | SideBySideEditor.ANY }): URI | undefined;
1404 > getOriginalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options: IEditorResourceAccessorOptions & { supportSideBySide: SideBySideEditor.BOTH }): URI | { primary?: URI; secondary?: URI } | undefined;
1405 > getOriginalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options?: IEditorResourceAccessorOptions): URI | { primary?: URI; secondary?: URI } | undefined;
1406 > getOriginalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options?: IEditorResourceAccessorOptions): URI | { primary?: URI; secondary?: URI } | undefined {
1407 if (!editor) {
1408 return undefined;
1443 return this.filterUri(originalResource, options.filterByScheme);
1444 }
1445 > editor.ts
1446 > private getSideEditors(editor: EditorInput | IUntypedEditorInput): { primary: EditorInput | IUntypedEditorInput | undefined; secondary: EditorInput | IUntypedEditorInput | undefined } {
1447 if (isSideBySideEditorInput(editor) || isResourceSideBySideEditorInput(editor)) {
1448 return { primary: editor.primary, secondary: editor.secondary };
1455 return { primary: undefined, secondary: undefined };
1456 }
1457 > editor.ts
1458 > /**
1459 > * The canonical URI of an editor is the true unique identifier of the editor
1460 > * and should be used whenever the URI is used e.g. to compare with other
1461 > * editors or when caching certain data based on the URI.
1462 > *
1463 > * In contrast, the original URI (#getOriginalUri) may be different and should
1464 > * be used whenever the URI is presented to the user, e.g. as a label.
1465 > *
1466 > * For example: on Windows and macOS, the same file URI with different casing may
1467 > * point to the same file. The editor may chose to "normalize" the URI into a canonical
1468 > * form so that only one editor opens for same file URIs with different casing. As
1469 > * such, the original URI and the canonical URI can be different.
1470 > */
1471 > getCanonicalUri(editor: EditorInput | IUntypedEditorInput | undefined | null): URI | undefined;
1472 > getCanonicalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options: IEditorResourceAccessorOptions & { supportSideBySide?: SideBySideEditor.PRIMARY | SideBySideEditor.SECONDARY | SideBySideEditor.ANY }): URI | undefined;
1473 > getCanonicalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options: IEditorResourceAccessorOptions & { supportSideBySide: SideBySideEditor.BOTH }): URI | { primary?: URI; secondary?: URI } | undefined;
1474 > getCanonicalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options?: IEditorResourceAccessorOptions): URI | { primary?: URI; secondary?: URI } | undefined;
1475 > getCanonicalUri(editor: EditorInput | IUntypedEditorInput | undefined | null, options?: IEditorResourceAccessorOptions): URI | { primary?: URI; secondary?: URI } | undefined {
1476 if (!editor) {
1477 return undefined;
1512 return this.filterUri(canonicalResource, options.filterByScheme);
1513 }
1514 > editor.ts
1515 > private filterUri(resource: URI, filter: string | string[]): URI | undefined {
1516
1517 // Multiple scheme filter
1531 return undefined;
1532 }
1533 > } editor.ts
1534 >
1535 > export type PreventPinnedEditorClose = 'keyboardAndMouse' | 'keyboard' | 'mouse' | 'never' | undefined;
1536 >
1537 > export enum EditorCloseMethod {
1538 > UNKNOWN,
1539 > KEYBOARD,
1540 > MOUSE
1541 > }
1542 >
1543 > export function preventEditorClose(group: IEditorGroup | IReadonlyEditorGroupModel, editor: EditorInput, method: EditorCloseMethod, configuration: IEditorPartConfiguration): boolean {
1544 if (!group.isSticky(editor)) {
1545 return false; // only interested in sticky editors
1554 return false;
1555 }
1556 > editor.ts
1557 > export const EditorResourceAccessor = new EditorResourceAccessorImpl();
1558 >
1559 > export const enum CloseDirection {
1560 > LEFT,
1561 > RIGHT
1562 > }
1563 >
1564 > export interface IEditorMemento<T> {
1565 >
1566 > saveEditorState(group: IEditorGroup, resource: URI, state: T): void;
1567 > saveEditorState(group: IEditorGroup, editor: EditorInput, state: T): void;
1568 >
1569 > loadEditorState(group: IEditorGroup, resource: URI): T | undefined;
1570 > loadEditorState(group: IEditorGroup, editor: EditorInput): T | undefined;
1571 >
1572 > clearEditorState(resource: URI, group?: IEditorGroup): void;
1573 > clearEditorState(editor: EditorInput, group?: IEditorGroup): void;
1574 >
1575 > clearEditorStateOnDispose(resource: URI, editor: EditorInput): void;
1576 >
1577 > moveEditorState(source: URI, target: URI, comparer: IExtUri): void;
1578 > }
1579 >
1580 > class EditorFactoryRegistry implements IEditorFactoryRegistry {
1581 > private instantiationService: IInstantiationService | undefined;
1582 >
1583 > private fileEditorFactory: IFileEditorFactory | undefined;
1584 >
1585 > private readonly editorSerializerConstructors = new Map<string /* Type ID */, IConstructorSignature<IEditorSerializer>>();
1586 > private readonly editorSerializerInstances = new Map<string /* Type ID */, IEditorSerializer>();
1587 >
1588 > start(accessor: ServicesAccessor): void {
1589 const instantiationService = this.instantiationService = accessor.get(IInstantiationService);
1590
1595 this.editorSerializerConstructors.clear();
1596 }
1597 > editor.ts
1598 > private createEditorSerializer(editorTypeId: string, ctor: IConstructorSignature<IEditorSerializer>, instantiationService: IInstantiationService): void {
1599 const instance = instantiationService.createInstance(ctor);
1600 this.editorSerializerInstances.set(editorTypeId, instance);
1601 }
1602 > editor.ts
1603 > registerFileEditorFactory(factory: IFileEditorFactory): void {
1604 if (this.fileEditorFactory) {
1605 throw new Error('Can only register one file editor factory.');
1608 this.fileEditorFactory = factory;
1609 }
1610 > editor.ts
1611 > getFileEditorFactory(): IFileEditorFactory {
1612 return assertReturnsDefined(this.fileEditorFactory);
1613 }
1614 > editor.ts
1615 > registerEditorSerializer(editorTypeId: string, ctor: IConstructorSignature<IEditorSerializer>): IDisposable {
1616 if (this.editorSerializerConstructors.has(editorTypeId) || this.editorSerializerInstances.has(editorTypeId)) {
1617 throw new Error(`A editor serializer with type ID '${editorTypeId}' was already registered.`);
1629 });
1630 }
1631 > editor.ts
1632 > getEditorSerializer(editor: EditorInput): IEditorSerializer | undefined;
1633 > getEditorSerializer(editorTypeId: string): IEditorSerializer | undefined;
1634 > getEditorSerializer(arg1: string | EditorInput): IEditorSerializer | undefined {
1635 return this.editorSerializerInstances.get(typeof arg1 === 'string' ? arg1 : arg1.typeId);
1636 }
1637 > } editor.ts
1638 >
1639 > Registry.add(EditorExtensions.EditorFactory, new EditorFactoryRegistry());
1640 >
1641 export async function pathsToEditors(paths: IPathData[] | undefined, fileService: IFileService, logService: ILogService): Promise<ReadonlyArray<IResourceEditorInput | IUntitledTextResourceEditorInput | undefined>> {
1642 if (!paths?.length) {
1691 }));
1692 }
1693 > editor.ts
1694 > export const enum EditorsOrder {
1695 >
1696 > /**
1697 > * Editors sorted by most recent activity (most recent active first)
1698 > */
1699 > MOST_RECENTLY_ACTIVE,
1700 >
1701 > /**
1702 > * Editors sorted by sequential order
1703 > */
1704 > SEQUENTIAL
1705 > }
1706 >
1707 > export function isTextEditorViewState(candidate: unknown): candidate is IEditorViewState {
1708 const viewState = candidate as IEditorViewState | undefined;
1709 if (!viewState) {
1720 return !!(codeEditorViewState.contributionsState && codeEditorViewState.viewState && Array.isArray(codeEditorViewState.cursorState));
1721 }
1722 > editor.ts
1723 > export interface IEditorOpenErrorOptions {
1724 >
1725 > /**
1726 > * If set to true, the message will be taken
1727 > * from the error message entirely and not be
1728 > * composed with more text.
1729 > */
1730 > forceMessage?: boolean;
1731 >
1732 > /**
1733 > * If set, will override the severity of the error.
1734 > */
1735 > forceSeverity?: Severity;
1736 >
1737 > /**
1738 > * If set to true, the error may be shown in a dialog
1739 > * to the user if the editor opening was triggered by
1740 > * user action. Otherwise and by default, the error will
1741 > * be shown as place holder in the editor area.
1742 > */
1743 > allowDialog?: boolean;
1744 > }
1745 >
1746 > export interface IEditorOpenError extends IErrorWithActions, IEditorOpenErrorOptions { }
1747 >
1748 > export function isEditorOpenError(obj: unknown): obj is IEditorOpenError {
1749 return isErrorWithActions(obj);
1750 }
1751 > editor.ts
1752 > export function createEditorOpenError(messageOrError: string | Error, actions: IAction[], options?: IEditorOpenErrorOptions): IEditorOpenError {
1753 const error: IEditorOpenError = createErrorWithActions(messageOrError, actions);
1754
1759 return error;
1760 }
1761 > editor.ts
1762 > export interface IToolbarActions {
1763 > readonly primary: IAction[];
1764 > readonly secondary: IAction[];
1765 > }
src/vs/platform/files/common/files.ts 1360 covered LOC · 44 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- files.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer, VSBufferReadable, VSBufferReadableStream } from '../../../base/common/buffer.js';
7 > import { CancellationToken } from '../../../base/common/cancellation.js';
8 > import { Event } from '../../../base/common/event.js';
9 > import { IExpression, IRelativePattern } from '../../../base/common/glob.js';
10 > import { IDisposable } from '../../../base/common/lifecycle.js';
11 > import { TernarySearchTree } from '../../../base/common/ternarySearchTree.js';
12 > import { sep } from '../../../base/common/path.js';
13 > import { ReadableStreamEvents } from '../../../base/common/stream.js';
14 > import { startsWithIgnoreCase } from '../../../base/common/strings.js';
15 > import { isNumber } from '../../../base/common/types.js';
16 > import { URI } from '../../../base/common/uri.js';
17 > import { localize } from '../../../nls.js';
18 > import { createDecorator } from '../../instantiation/common/instantiation.js';
19 > import { isWeb } from '../../../base/common/platform.js';
20 > import { Schemas } from '../../../base/common/network.js';
21 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
22 > import { Lazy } from '../../../base/common/lazy.js';
23 >
24 > //#region file service & providers
25 >
26 > export const IFileService = createDecorator<IFileService>('fileService');
27 >
28 > export interface IFileService {
29 >
30 > readonly _serviceBrand: undefined;
31 >
32 > /**
33 > * An event that is fired when a file system provider is added or removed
34 > */
35 > readonly onDidChangeFileSystemProviderRegistrations: Event<IFileSystemProviderRegistrationEvent>;
36 >
37 > /**
38 > * An event that is fired when a registered file system provider changes its capabilities.
39 > */
40 > readonly onDidChangeFileSystemProviderCapabilities: Event<IFileSystemProviderCapabilitiesChangeEvent>;
41 >
42 > /**
43 > * An event that is fired when a file system provider is about to be activated. Listeners
44 > * can join this event with a long running promise to help in the activation process.
45 > */
46 > readonly onWillActivateFileSystemProvider: Event<IFileSystemProviderActivationEvent>;
47 >
48 > /**
49 > * Registers a file system provider for a certain scheme.
50 > */
51 > registerProvider(scheme: string, provider: IFileSystemProvider): IDisposable;
52 >
53 > /**
54 > * Returns a file system provider for a certain scheme.
55 > */
56 > getProvider(scheme: string): IFileSystemProvider | undefined;
57 >
58 > /**
59 > * Tries to activate a provider with the given scheme.
60 > */
61 > activateProvider(scheme: string): Promise<void>;
62 >
63 > /**
64 > * Checks if this file service can handle the given resource by
65 > * first activating any extension that wants to be activated
66 > * on the provided resource scheme to include extensions that
67 > * contribute file system providers for the given resource.
68 > */
69 > canHandleResource(resource: URI): Promise<boolean>;
70 >
71 > /**
72 > * Checks if the file service has a registered provider for the
73 > * provided resource.
74 > *
75 > * Note: this does NOT account for contributed providers from
76 > * extensions that have not been activated yet. To include those,
77 > * consider to call `await fileService.canHandleResource(resource)`.
78 > */
79 > hasProvider(resource: URI): boolean;
80 >
81 > /**
82 > * Checks if the provider for the provided resource has the provided file system capability.
83 > */
84 > hasCapability(resource: URI, capability: FileSystemProviderCapabilities): boolean;
85 >
86 > /**
87 > * List the schemes and capabilities for registered file system providers
88 > */
89 > listCapabilities(): Iterable<{ scheme: string; capabilities: FileSystemProviderCapabilities }>;
90 >
91 > /**
92 > * Allows to listen for file changes. The event will fire for every file within the opened workspace
93 > * (if any) as well as all files that have been watched explicitly using the #watch() API.
94 > */
95 > readonly onDidFilesChange: Event<FileChangesEvent>;
96 >
97 > /**
98 > * An event that is fired upon successful completion of a certain file operation.
99 > */
100 > readonly onDidRunOperation: Event<FileOperationEvent>;
101 >
102 > /**
103 > * Resolve the properties of a file/folder identified by the resource. For a folder, children
104 > * information is resolved as well depending on the provided options. Use `stat()` method if
105 > * you do not need children information.
106 > *
107 > * If the optional parameter "resolveTo" is specified in options, the stat service is asked
108 > * to provide a stat object that should contain the full graph of folders up to all of the
109 > * target resources.
110 > *
111 > * If the optional parameter "resolveSingleChildDescendants" is specified in options,
112 > * the stat service is asked to automatically resolve child folders that only
113 > * contain a single element.
114 > *
115 > * If the optional parameter "resolveMetadata" is specified in options,
116 > * the stat will contain metadata information such as size, mtime and etag.
117 > */
118 > resolve(resource: URI, options: IResolveMetadataFileOptions): Promise<IFileStatWithMetadata>;
119 > resolve(resource: URI, options?: IResolveFileOptions): Promise<IFileStat>;
120 >
121 > /**
122 > * Same as `resolve()` but supports resolving multiple resources in parallel.
123 > *
124 > * If one of the resolve targets fails to resolve returns a fake `IFileStat` instead of
125 > * making the whole call fail.
126 > */
127 > resolveAll(toResolve: { resource: URI; options: IResolveMetadataFileOptions }[]): Promise<IFileStatResult[]>;
128 > resolveAll(toResolve: { resource: URI; options?: IResolveFileOptions }[]): Promise<IFileStatResult[]>;
129 >
130 > /**
131 > * Same as `resolve()` but without resolving the children of a folder if the
132 > * resource is pointing to a folder.
133 > */
134 > stat(resource: URI): Promise<IFileStatWithPartialMetadata>;
135 >
136 > /**
137 > * Attempts to resolve the real path of the provided resource. The real path can be
138 > * different from the resource path for example when it is a symlink.
139 > *
140 > * Will return `undefined` if the real path cannot be resolved.
141 > */
142 > realpath(resource: URI): Promise<URI | undefined>;
143 >
144 > /**
145 > * Finds out if a file/folder identified by the resource exists.
146 > */
147 > exists(resource: URI): Promise<boolean>;
148 >
149 > /**
150 > * Read the contents of the provided resource unbuffered.
151 > */
152 > readFile(resource: URI, options?: IReadFileOptions, token?: CancellationToken): Promise<IFileContent>;
153 >
154 > /**
155 > * Read the contents of the provided resource buffered as stream.
156 > */
157 > readFileStream(resource: URI, options?: IReadFileStreamOptions, token?: CancellationToken): Promise<IFileStreamContent>;
158 >
159 > /**
160 > * Updates the content replacing its previous value.
161 > * If `options.append` is true, appends content to the end of the file instead.
162 > *
163 > * Emits a `FileOperation.WRITE` file operation event when successful.
164 > */
165 > writeFile(resource: URI, bufferOrReadableOrStream: VSBuffer | VSBufferReadable | VSBufferReadableStream, options?: IWriteFileOptions): Promise<IFileStatWithMetadata>;
166 >
167 > /**
168 > * Moves the file/folder to a new path identified by the resource.
169 > *
170 > * The optional parameter overwrite can be set to replace an existing file at the location.
171 > *
172 > * Emits a `FileOperation.MOVE` file operation event when successful.
173 > */
174 > move(source: URI, target: URI, overwrite?: boolean): Promise<IFileStatWithMetadata>;
175 >
176 > /**
177 > * Find out if a move operation is possible given the arguments. No changes on disk will
178 > * be performed. Returns an Error if the operation cannot be done.
179 > */
180 > canMove(source: URI, target: URI, overwrite?: boolean): Promise<Error | true>;
181 >
182 > /**
183 > * Copies the file/folder to a path identified by the resource. A folder is copied
184 > * recursively.
185 > *
186 > * Emits a `FileOperation.COPY` file operation event when successful.
187 > */
188 > copy(source: URI, target: URI, overwrite?: boolean): Promise<IFileStatWithMetadata>;
189 >
190 > /**
191 > * Find out if a copy operation is possible given the arguments. No changes on disk will
192 > * be performed. Returns an Error if the operation cannot be done.
193 > */
194 > canCopy(source: URI, target: URI, overwrite?: boolean): Promise<Error | true>;
195 >
196 > /**
197 > * Clones a file to a path identified by the resource. Folders are not supported.
198 > *
199 > * If the target path exists, it will be overwritten.
200 > */
201 > cloneFile(source: URI, target: URI): Promise<void>;
202 >
203 > /**
204 > * Creates a new file with the given path and optional contents. The returned promise
205 > * will have the stat model object as a result.
206 > *
207 > * The optional parameter content can be used as value to fill into the new file.
208 > *
209 > * Emits a `FileOperation.CREATE` file operation event when successful.
210 > */
211 > createFile(resource: URI, bufferOrReadableOrStream?: VSBuffer | VSBufferReadable | VSBufferReadableStream, options?: ICreateFileOptions): Promise<IFileStatWithMetadata>;
212 >
213 > /**
214 > * Find out if a file create operation is possible given the arguments. No changes on disk will
215 > * be performed. Returns an Error if the operation cannot be done.
216 > */
217 > canCreateFile(resource: URI, options?: ICreateFileOptions): Promise<Error | true>;
218 >
219 > /**
220 > * Creates a new folder with the given path. The returned promise
221 > * will have the stat model object as a result.
222 > *
223 > * Emits a `FileOperation.CREATE` file operation event when successful.
224 > */
225 > createFolder(resource: URI): Promise<IFileStatWithMetadata>;
226 >
227 > /**
228 > * Deletes the provided file. The optional useTrash parameter allows to
229 > * move the file to trash. The optional recursive parameter allows to delete
230 > * non-empty folders recursively.
231 > *
232 > * Emits a `FileOperation.DELETE` file operation event when successful.
233 > */
234 > del(resource: URI, options?: Partial<IFileDeleteOptions>): Promise<void>;
235 >
236 > /**
237 > * Find out if a delete operation is possible given the arguments. No changes on disk will
238 > * be performed. Returns an Error if the operation cannot be done.
239 > */
240 > canDelete(resource: URI, options?: Partial<IFileDeleteOptions>): Promise<Error | true>;
241 >
242 > /**
243 > * An event that signals an error when watching for file changes.
244 > */
245 > readonly onDidWatchError: Event<Error>;
246 >
247 > /**
248 > * Allows to start a watcher that reports file/folder change events on the provided resource.
249 > *
250 > * The watcher runs correlated and thus, file events will be reported on the returned
251 > * `IFileSystemWatcher` and not on the generic `IFileService.onDidFilesChange` event.
252 > *
253 > * Note: only non-recursive file watching supports event correlation for now.
254 > */
255 > createWatcher(resource: URI, options: IWatchOptionsWithoutCorrelation & { recursive: false }): IFileSystemWatcher;
256 >
257 > /**
258 > * Allows to start a watcher that reports file/folder change events on the provided resource.
259 > *
260 > * The watcher runs uncorrelated and thus will report all events from `IFileService.onDidFilesChange`.
261 > * This means, most listeners in the application will receive your events. It is encouraged to
262 > * use correlated watchers (via `IWatchOptionsWithCorrelation`) to limit events to your listener.
263 > */
264 > watch(resource: URI, options?: IWatchOptionsWithoutCorrelation): IDisposable;
265 >
266 > /**
267 > * Frees up any resources occupied by this service.
268 > */
269 > dispose(): void;
270 > }
271 >
272 > export interface IFileOverwriteOptions {
273 >
274 > /**
275 > * Set to `true` to overwrite a file if it exists. Will
276 > * throw an error otherwise if the file does exist.
277 > */
278 > readonly overwrite: boolean;
279 > }
280 >
281 > export interface IFileUnlockOptions {
282 >
283 > /**
284 > * Set to `true` to try to remove any write locks the file might
285 > * have. A file that is write locked will throw an error for any
286 > * attempt to write to unless `unlock: true` is provided.
287 > */
288 > readonly unlock: boolean;
289 > }
290 >
291 > export interface IFileAtomicReadOptions {
292 >
293 > /**
294 > * The optional `atomic` flag can be used to make sure
295 > * the `readFile` method is not running in parallel with
296 > * any `write` operations in the same process.
297 > *
298 > * Typically you should not need to use this flag but if
299 > * for example you are quickly reading a file right after
300 > * a file event occurred and the file changes a lot, there
301 > * is a chance that a read returns an empty or partial file
302 > * because a pending write has not finished yet.
303 > *
304 > * Note: this does not prevent the file from being written
305 > * to from a different process. If you need such atomic
306 > * operations, you better use a real database as storage.
307 > */
308 > readonly atomic: boolean;
309 > }
310 >
311 > export interface IFileAtomicOptions {
312 >
313 > /**
314 > * The postfix is used to create a temporary file based
315 > * on the original resource. The resulting temporary
316 > * file will be in the same folder as the resource and
317 > * have `postfix` appended to the resource name.
318 > *
319 > * Example: given a file resource `file:///some/path/foo.txt`
320 > * and a postfix `.vsctmp`, the temporary file will be
321 > * created as `file:///some/path/foo.txt.vsctmp`.
322 > */
323 > readonly postfix: string;
324 > }
325 >
326 > export interface IFileAtomicWriteOptions {
327 >
328 > /**
329 > * The optional `atomic` flag can be used to make sure
330 > * the `writeFile` method updates the target file atomically
331 > * by first writing to a temporary file in the same folder
332 > * and then renaming it over the target.
333 > */
334 > readonly atomic: IFileAtomicOptions | false;
335 > }
336 >
337 > export interface IFileAtomicDeleteOptions {
338 >
339 > /**
340 > * The optional `atomic` flag can be used to make sure
341 > * the `delete` method deletes the target atomically by
342 > * first renaming it to a temporary resource in the same
343 > * folder and then deleting it.
344 > */
345 > readonly atomic: IFileAtomicOptions | false;
346 > }
347 >
348 > export interface IFileReadLimits {
349 >
350 > /**
351 > * If the file exceeds the given size, an error of kind
352 > * `FILE_TOO_LARGE` will be thrown.
353 > */
354 > size?: number;
355 > }
356 >
357 > export interface IFileReadStreamOptions {
358 >
359 > /**
360 > * Is an integer specifying where to begin reading from in the file. If position is undefined,
361 > * data will be read from the current file position.
362 > */
363 > readonly position?: number;
364 >
365 > /**
366 > * Is an integer specifying how many bytes to read from the file. By default, all bytes
367 > * will be read.
368 > */
369 > readonly length?: number;
370 >
371 > /**
372 > * If provided, the size of the file will be checked against the limits
373 > * and an error will be thrown if any limit is exceeded.
374 > */
375 > readonly limits?: IFileReadLimits;
376 > }
377 >
378 > export interface IFileWriteOptions extends IFileOverwriteOptions, IFileUnlockOptions, IFileAtomicWriteOptions {
379 >
380 > /**
381 > * Set to `true` to create a file when it does not exist. Will
382 > * throw an error otherwise if the file does not exist.
383 > */
384 > readonly create: boolean;
385 >
386 > /**
387 > * Set to `true` to append content to the end of the file. Implies `create: true`,
388 > * and set only when the corresponding `FileAppend` capability is defined.
389 > */
390 > readonly append?: boolean;
391 > }
392 >
393 > export type IFileOpenOptions = IFileOpenForReadOptions | IFileOpenForWriteOptions;
394 >
395 > export function isFileOpenForWriteOptions(options: IFileOpenOptions): options is IFileOpenForWriteOptions {
396 return options.create === true;
397 }
398 > files.ts
399 > export interface IFileOpenForReadOptions {
400 >
401 > /**
402 > * A hint that the file should be opened for reading only.
403 > */
404 > readonly create: false;
405 > }
406 >
407 > export interface IFileOpenForWriteOptions extends IFileUnlockOptions {
408 >
409 > /**
410 > * A hint that the file should be opened for reading and writing.
411 > */
412 > readonly create: true;
413 >
414 > /**
415 > * Open the file in append mode. This will write data to the
416 > * end of the file.
417 > */
418 > readonly append?: boolean;
419 > }
420 >
421 > export interface IFileDeleteOptions {
422 >
423 > /**
424 > * Set to `true` to recursively delete any children of the file. This
425 > * only applies to folders and can lead to an error unless provided
426 > * if the folder is not empty.
427 > */
428 > readonly recursive: boolean;
429 >
430 > /**
431 > * Set to `true` to attempt to move the file to trash
432 > * instead of deleting it permanently from disk.
433 > *
434 > * This option maybe not be supported on all providers.
435 > */
436 > readonly useTrash: boolean;
437 >
438 > /**
439 > * The optional `atomic` flag can be used to make sure
440 > * the `delete` method deletes the target atomically by
441 > * first renaming it to a temporary resource in the same
442 > * folder and then deleting it.
443 > *
444 > * This option maybe not be supported on all providers.
445 > */
446 > readonly atomic: IFileAtomicOptions | false;
447 > }
448 >
449 > export enum FileType {
450 >
451 > /**
452 > * File is unknown (neither file, directory nor symbolic link).
453 > */
454 > Unknown = 0,
455 >
456 > /**
457 > * File is a normal file.
458 > */
459 > File = 1,
460 >
461 > /**
462 > * File is a directory.
463 > */
464 > Directory = 2,
465 >
466 > /**
467 > * File is a symbolic link.
468 > *
469 > * Note: even when the file is a symbolic link, you can test for
470 > * `FileType.File` and `FileType.Directory` to know the type of
471 > * the target the link points to.
472 > */
473 > SymbolicLink = 64
474 > }
475 >
476 > export enum FilePermission {
477 >
478 > /**
479 > * File is readonly. Components like editors should not
480 > * offer to edit the contents.
481 > */
482 > Readonly = 1,
483 >
484 > /**
485 > * File is locked. Components like editors should offer
486 > * to edit the contents and ask the user upon saving to
487 > * remove the lock.
488 > */
489 > Locked = 2,
490 >
491 > /**
492 > * File is executable. Relevant for Unix-like systems where
493 > * the executable bit determines if a file can be run.
494 > */
495 > Executable = 4
496 > }
497 >
498 > export interface IStat {
499 >
500 > /**
501 > * The file type.
502 > */
503 > readonly type: FileType;
504 >
505 > /**
506 > * The last modification date represented as millis from unix epoch.
507 > */
508 > readonly mtime: number;
509 >
510 > /**
511 > * The creation date represented as millis from unix epoch.
512 > */
513 > readonly ctime: number;
514 >
515 > /**
516 > * The size of the file in bytes.
517 > */
518 > readonly size: number;
519 >
520 > /**
521 > * The file permissions.
522 > */
523 > readonly permissions?: FilePermission;
524 > }
525 >
526 > export interface IWatchOptionsWithoutCorrelation {
527 >
528 > /**
529 > * Set to `true` to watch for changes recursively in a folder
530 > * and all of its children.
531 > */
532 > recursive: boolean;
533 >
534 > /**
535 > * A set of glob patterns or paths to exclude from watching.
536 > * Paths can be relative or absolute and when relative are
537 > * resolved against the watched folder. Glob patterns are
538 > * always matched relative to the watched folder.
539 > */
540 > excludes: string[];
541 >
542 > /**
543 > * An optional set of glob patterns or paths to include for
544 > * watching. If not provided, all paths are considered for
545 > * events.
546 > * Paths can be relative or absolute and when relative are
547 > * resolved against the watched folder. Glob patterns are
548 > * always matched relative to the watched folder.
549 > */
550 > includes?: Array<string | IRelativePattern>;
551 >
552 > /**
553 > * If provided, allows to filter the events that the watcher should consider
554 > * for emitting. If not provided, all events are emitted.
555 > *
556 > * For example, to emit added and updated events, set to:
557 > * `FileChangeFilter.ADDED | FileChangeFilter.UPDATED`.
558 > */
559 > filter?: FileChangeFilter;
560 > }
561 >
562 > export interface IWatchOptions extends IWatchOptionsWithoutCorrelation {
563 >
564 > /**
565 > * If provided, file change events from the watcher that
566 > * are a result of this watch request will carry the same
567 > * id.
568 > */
569 > readonly correlationId?: number;
570 > }
571 >
572 > export const enum FileChangeFilter {
573 > UPDATED = 1 << 1,
574 > ADDED = 1 << 2,
575 > DELETED = 1 << 3
576 > }
577 >
578 > export interface IWatchOptionsWithCorrelation extends IWatchOptions {
579 > readonly correlationId: number;
580 > }
581 >
582 > export interface IFileSystemWatcher extends IDisposable {
583 >
584 > /**
585 > * An event which fires on file/folder change only for changes
586 > * that correlate to the watch request with matching correlation
587 > * identifier.
588 > */
589 > readonly onDidChange: Event<FileChangesEvent>;
590 > }
591 >
592 > export function isFileSystemWatcher(thing: unknown): thing is IFileSystemWatcher {
593 const candidate = thing as IFileSystemWatcher | undefined;
594
595 return !!candidate && typeof candidate.onDidChange === 'function';
596 }
597 > files.ts
598 > export const enum FileSystemProviderCapabilities {
599 >
600 > /**
601 > * No capabilities.
602 > */
603 > None = 0,
604 >
605 > /**
606 > * Provider supports unbuffered read/write.
607 > */
608 > FileReadWrite = 1 << 1,
609 >
610 > /**
611 > * Provider supports open/read/write/close low level file operations.
612 > */
613 > FileOpenReadWriteClose = 1 << 2,
614 >
615 > /**
616 > * Provider supports stream based reading.
617 > */
618 > FileReadStream = 1 << 4,
619 >
620 > /**
621 > * Provider supports copy operation.
622 > */
623 > FileFolderCopy = 1 << 3,
624 >
625 > /**
626 > * Provider is path case sensitive.
627 > */
628 > PathCaseSensitive = 1 << 10,
629 >
630 > /**
631 > * All files of the provider are readonly.
632 > */
633 > Readonly = 1 << 11,
634 >
635 > /**
636 > * Provider supports to delete via trash.
637 > */
638 > Trash = 1 << 12,
639 >
640 > /**
641 > * Provider support to unlock files for writing.
642 > */
643 > FileWriteUnlock = 1 << 13,
644 >
645 > /**
646 > * Provider support to read files atomically. This implies the
647 > * provider provides the `FileReadWrite` capability too.
648 > */
649 > FileAtomicRead = 1 << 14,
650 >
651 > /**
652 > * Provider support to write files atomically. This implies the
653 > * provider provides the `FileReadWrite` capability too.
654 > */
655 > FileAtomicWrite = 1 << 15,
656 >
657 > /**
658 > * Provider support to delete atomically.
659 > */
660 > FileAtomicDelete = 1 << 16,
661 >
662 > /**
663 > * Provider support to clone files atomically.
664 > */
665 > FileClone = 1 << 17,
666 >
667 > /**
668 > * Provider support to resolve real paths.
669 > */
670 > FileRealpath = 1 << 18,
671 >
672 > /**
673 > * Provider support to append to files.
674 > */
675 > FileAppend = 1 << 19
676 > }
677 >
678 > export interface IFileSystemProvider {
679 >
680 > readonly capabilities: FileSystemProviderCapabilities;
681 > readonly onDidChangeCapabilities: Event<void>;
682 >
683 > readonly onDidChangeFile: Event<readonly IFileChange[]>;
684 > readonly onDidWatchError?: Event<string>;
685 > watch(resource: URI, opts: IWatchOptions): IDisposable;
686 >
687 > stat(resource: URI): Promise<IStat>;
688 > mkdir(resource: URI): Promise<void>;
689 > readdir(resource: URI): Promise<[string, FileType][]>;
690 > delete(resource: URI, opts: IFileDeleteOptions): Promise<void>;
691 >
692 > rename(from: URI, to: URI, opts: IFileOverwriteOptions): Promise<void>;
693 > copy?(from: URI, to: URI, opts: IFileOverwriteOptions): Promise<void>;
694 >
695 > readFile?(resource: URI): Promise<Uint8Array>;
696 > writeFile?(resource: URI, content: Uint8Array, opts: IFileWriteOptions): Promise<void>;
697 >
698 > readFileStream?(resource: URI, opts: IFileReadStreamOptions, token: CancellationToken): ReadableStreamEvents<Uint8Array>;
699 >
700 > open?(resource: URI, opts: IFileOpenOptions): Promise<number>;
701 > close?(fd: number): Promise<void>;
702 > read?(fd: number, pos: number, data: Uint8Array, offset: number, length: number): Promise<number>;
703 > write?(fd: number, pos: number, data: Uint8Array, offset: number, length: number): Promise<number>;
704 >
705 > cloneFile?(from: URI, to: URI): Promise<void>;
706 > }
707 >
708 > export interface IFileSystemProviderWithFileReadWriteCapability extends IFileSystemProvider {
709 > readFile(resource: URI): Promise<Uint8Array>;
710 > writeFile(resource: URI, content: Uint8Array, opts: IFileWriteOptions): Promise<void>;
711 > }
712 >
713 > export function hasReadWriteCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileReadWriteCapability {
714 return !!(provider.capabilities & FileSystemProviderCapabilities.FileReadWrite);
715 }
716 > files.ts
717 > export function hasFileAppendCapability(provider: IFileSystemProvider): boolean {
718 return !!(provider.capabilities & FileSystemProviderCapabilities.FileAppend);
719 }
720 > files.ts
721 > export interface IFileSystemProviderWithFileFolderCopyCapability extends IFileSystemProvider {
722 > copy(from: URI, to: URI, opts: IFileOverwriteOptions): Promise<void>;
723 > }
724 >
725 > export function hasFileFolderCopyCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileFolderCopyCapability {
726 return !!(provider.capabilities & FileSystemProviderCapabilities.FileFolderCopy);
727 }
728 > files.ts
729 > export interface IFileSystemProviderWithFileCloneCapability extends IFileSystemProvider {
730 > cloneFile(from: URI, to: URI): Promise<void>;
731 > }
732 >
733 > export function hasFileCloneCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileCloneCapability {
734 return !!(provider.capabilities & FileSystemProviderCapabilities.FileClone);
735 }
736 > files.ts
737 > export interface IFileSystemProviderWithFileRealpathCapability extends IFileSystemProvider {
738 > realpath(resource: URI): Promise<string>;
739 > }
740 >
741 > export function hasFileRealpathCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileRealpathCapability {
742 return !!(provider.capabilities & FileSystemProviderCapabilities.FileRealpath);
743 }
744 > files.ts
745 > export interface IFileSystemProviderWithOpenReadWriteCloseCapability extends IFileSystemProvider {
746 > open(resource: URI, opts: IFileOpenOptions): Promise<number>;
747 > close(fd: number): Promise<void>;
748 > read(fd: number, pos: number, data: Uint8Array, offset: number, length: number): Promise<number>;
749 > write(fd: number, pos: number, data: Uint8Array, offset: number, length: number): Promise<number>;
750 > }
751 >
752 > export function hasOpenReadWriteCloseCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithOpenReadWriteCloseCapability {
753 return !!(provider.capabilities & FileSystemProviderCapabilities.FileOpenReadWriteClose);
754 }
755 > files.ts
756 > export interface IFileSystemProviderWithFileReadStreamCapability extends IFileSystemProvider {
757 > readFileStream(resource: URI, opts: IFileReadStreamOptions, token: CancellationToken): ReadableStreamEvents<Uint8Array>;
758 > }
759 >
760 > export function hasFileReadStreamCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileReadStreamCapability {
761 return !!(provider.capabilities & FileSystemProviderCapabilities.FileReadStream);
762 }
763 > files.ts
764 > export interface IFileSystemProviderWithFileAtomicReadCapability extends IFileSystemProvider {
765 > readFile(resource: URI, opts?: IFileAtomicReadOptions): Promise<Uint8Array>;
766 > enforceAtomicReadFile?(resource: URI): boolean;
767 > }
768 >
769 > export function hasFileAtomicReadCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileAtomicReadCapability {
770 if (!hasReadWriteCapability(provider)) {
771 return false; // we require the `FileReadWrite` capability too
774 return !!(provider.capabilities & FileSystemProviderCapabilities.FileAtomicRead);
775 }
776 > files.ts
777 > export interface IFileSystemProviderWithFileAtomicWriteCapability extends IFileSystemProvider {
778 > writeFile(resource: URI, contents: Uint8Array, opts?: IFileAtomicWriteOptions): Promise<void>;
779 > enforceAtomicWriteFile?(resource: URI): IFileAtomicOptions | false;
780 > }
781 >
782 > export function hasFileAtomicWriteCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileAtomicWriteCapability {
783 if (!hasReadWriteCapability(provider)) {
784 return false; // we require the `FileReadWrite` capability too
787 return !!(provider.capabilities & FileSystemProviderCapabilities.FileAtomicWrite);
788 }
789 > files.ts
790 > export interface IFileSystemProviderWithFileAtomicDeleteCapability extends IFileSystemProvider {
791 > delete(resource: URI, opts: IFileAtomicDeleteOptions): Promise<void>;
792 > enforceAtomicDelete?(resource: URI): IFileAtomicOptions | false;
793 > }
794 >
795 > export function hasFileAtomicDeleteCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithFileAtomicDeleteCapability {
796 return !!(provider.capabilities & FileSystemProviderCapabilities.FileAtomicDelete);
797 }
798 > files.ts
799 > export interface IFileSystemProviderWithReadonlyCapability extends IFileSystemProvider {
800 >
801 > readonly capabilities: FileSystemProviderCapabilities.Readonly & FileSystemProviderCapabilities;
802 >
803 > /**
804 > * An optional message to show in the UI to explain why the file system is readonly.
805 > */
806 > readonly readOnlyMessage?: IMarkdownString;
807 > }
808 >
809 > export function hasReadonlyCapability(provider: IFileSystemProvider): provider is IFileSystemProviderWithReadonlyCapability {
810 return !!(provider.capabilities & FileSystemProviderCapabilities.Readonly);
811 }
812 > files.ts
813 > export enum FileSystemProviderErrorCode {
814 > FileExists = 'EntryExists',
815 > FileNotFound = 'EntryNotFound',
816 > FileNotADirectory = 'EntryNotADirectory',
817 > FileIsADirectory = 'EntryIsADirectory',
818 > FileExceedsStorageQuota = 'EntryExceedsStorageQuota',
819 > FileTooLarge = 'EntryTooLarge',
820 > FileWriteLocked = 'EntryWriteLocked',
821 > NoPermissions = 'NoPermissions',
822 > Unavailable = 'Unavailable',
823 > Unknown = 'Unknown'
824 > }
825 >
826 > export interface IFileSystemProviderError extends Error {
827 > readonly name: string;
828 > readonly code: FileSystemProviderErrorCode;
829 > }
830 >
831 > export class FileSystemProviderError extends Error implements IFileSystemProviderError {
832 >
833 > static create(error: Error | string, code: FileSystemProviderErrorCode): FileSystemProviderError {
834 > const providerError = new FileSystemProviderError(error.toString(), code);
835 > markAsFileSystemProviderError(providerError, code);
836 >
837 > return providerError;
838 > }
839 >
840 > private constructor(message: string, readonly code: FileSystemProviderErrorCode) {
841 super(message);
842 }
843 > } files.ts
844 >
845 > export function createFileSystemProviderError(error: Error | string, code: FileSystemProviderErrorCode): FileSystemProviderError {
846 return FileSystemProviderError.create(error, code);
847 }
848 > files.ts
849 > export function ensureFileSystemProviderError(error?: Error): Error {
850 if (!error) {
851 return createFileSystemProviderError(localize('unknownError', "Unknown Error"), FileSystemProviderErrorCode.Unknown); // https://github.com/microsoft/vscode/issues/72798
854 return error;
855 }
856 > files.ts
857 > export function markAsFileSystemProviderError(error: Error, code: FileSystemProviderErrorCode): Error {
858 error.name = code ? `${code} (FileSystemError)` : `FileSystemError`;
859
860 return error;
861 }
862 > files.ts
863 > export function toFileSystemProviderErrorCode(error: Error | undefined | null): FileSystemProviderErrorCode {
864
865 // Guard against abuse
893 return FileSystemProviderErrorCode.Unknown;
894 }
895 > files.ts
896 > export function toFileOperationResult(error: Error): FileOperationResult {
897
898 // FileSystemProviderError comes with the result already
921 }
922 }
923 > files.ts
924 > export interface IFileSystemProviderRegistrationEvent {
925 > readonly added: boolean;
926 > readonly scheme: string;
927 > readonly provider?: IFileSystemProvider;
928 > }
929 >
930 > export interface IFileSystemProviderCapabilitiesChangeEvent {
931 > readonly provider: IFileSystemProvider;
932 > readonly scheme: string;
933 > }
934 >
935 > export interface IFileSystemProviderActivationEvent {
936 > readonly scheme: string;
937 > join(promise: Promise<void>): void;
938 > }
939 >
940 > export const enum FileOperation {
941 > CREATE,
942 > DELETE,
943 > MOVE,
944 > COPY,
945 > WRITE
946 > }
947 >
948 > export interface IFileOperationEvent {
949 >
950 > readonly resource: URI;
951 > readonly operation: FileOperation;
952 >
953 > isOperation(operation: FileOperation.DELETE | FileOperation.WRITE): boolean;
954 > isOperation(operation: FileOperation.CREATE | FileOperation.MOVE | FileOperation.COPY): this is IFileOperationEventWithMetadata;
955 > }
956 >
957 > export interface IFileOperationEventWithMetadata extends IFileOperationEvent {
958 > readonly target: IFileStatWithMetadata;
959 > }
960 >
961 > export class FileOperationEvent implements IFileOperationEvent {
962 >
963 > constructor(resource: URI, operation: FileOperation.DELETE | FileOperation.WRITE);
964 > constructor(resource: URI, operation: FileOperation.CREATE | FileOperation.MOVE | FileOperation.COPY, target: IFileStatWithMetadata);
965 > constructor(readonly resource: URI, readonly operation: FileOperation, readonly target?: IFileStatWithMetadata) { }
966 >
967 > isOperation(operation: FileOperation.DELETE | FileOperation.WRITE): boolean;
968 > isOperation(operation: FileOperation.CREATE | FileOperation.MOVE | FileOperation.COPY): this is IFileOperationEventWithMetadata;
969 > isOperation(operation: FileOperation): boolean {
970 return this.operation === operation;
971 }
972 > } files.ts
973 >
974 > /**
975 > * Possible changes that can occur to a file.
976 > */
977 > export const enum FileChangeType {
978 > UPDATED,
979 > ADDED,
980 > DELETED
981 > }
982 >
983 > /**
984 > * Identifies a single change in a file.
985 > */
986 > export interface IFileChange {
987 >
988 > /**
989 > * The type of change that occurred to the file.
990 > */
991 > type: FileChangeType;
992 >
993 > /**
994 > * The unified resource identifier of the file that changed.
995 > */
996 > readonly resource: URI;
997 >
998 > /**
999 > * If provided when starting the file watcher, the correlation
1000 > * identifier will match the original file watching request as
1001 > * a way to identify the original component that is interested
1002 > * in the change.
1003 > */
1004 > readonly cId?: number;
1005 > }
1006 >
1007 > export class FileChangesEvent {
1008 >
1009 > private static readonly MIXED_CORRELATION = null;
1010 >
1011 > private readonly correlationId: number | undefined | typeof FileChangesEvent.MIXED_CORRELATION = undefined;
1012 >
1013 > constructor(changes: readonly IFileChange[], private readonly ignorePathCasing: boolean) {
1014 for (const change of changes) {
1015
1043 }
1044 }
1045 > files.ts
1046 > private readonly added = new Lazy(() => {
1047 > const added = TernarySearchTree.forUris<boolean>(() => this.ignorePathCasing); files.ts
1048 > added.fill(this.rawAdded.map(resource => [resource, true]));
1049 >
1050 > return added;
1051 > }); files.ts
1052 >
1053 > private readonly updated = new Lazy(() => {
1054 > const updated = TernarySearchTree.forUris<boolean>(() => this.ignorePathCasing); files.ts
1055 > updated.fill(this.rawUpdated.map(resource => [resource, true]));
1056 >
1057 > return updated;
1058 > }); files.ts
1059 >
1060 > private readonly deleted = new Lazy(() => {
1061 > const deleted = TernarySearchTree.forUris<boolean>(() => this.ignorePathCasing); files.ts
1062 > deleted.fill(this.rawDeleted.map(resource => [resource, true]));
1063 >
1064 > return deleted;
1065 > }); files.ts
1066 >
1067 > /**
1068 > * Find out if the file change events match the provided resource.
1069 > *
1070 > * Note: when passing `FileChangeType.DELETED`, we consider a match
1071 > * also when the parent of the resource got deleted.
1072 > */
1073 > contains(resource: URI, ...types: FileChangeType[]): boolean {
1074 return this.doContains(resource, { includeChildren: false }, ...types);
1075 }
1076 > files.ts
1077 > /**
1078 > * Find out if the file change events either match the provided
1079 > * resource, or contain a child of this resource.
1080 > */
1081 > affects(resource: URI, ...types: FileChangeType[]): boolean {
1082 return this.doContains(resource, { includeChildren: true }, ...types);
1083 }
1084 > files.ts
1085 > private doContains(resource: URI, options: { includeChildren: boolean }, ...types: FileChangeType[]): boolean {
1086 if (!resource) {
1087 return false;
1125 return false;
1126 }
1127 > files.ts
1128 > /**
1129 > * Returns if this event contains added files.
1130 > */
1131 > gotAdded(): boolean {
1132 return this.rawAdded.length > 0;
1133 }
1134 > files.ts
1135 > /**
1136 > * Returns if this event contains deleted files.
1137 > */
1138 > gotDeleted(): boolean {
1139 return this.rawDeleted.length > 0;
1140 }
1141 > files.ts
1142 > /**
1143 > * Returns if this event contains updated files.
1144 > */
1145 > gotUpdated(): boolean {
1146 return this.rawUpdated.length > 0;
1147 }
1148 > files.ts
1149 > /**
1150 > * Returns if this event contains changes that correlate to the
1151 > * provided `correlationId`.
1152 > *
1153 > * File change event correlation is an advanced watch feature that
1154 > * allows to identify from which watch request the events originate
1155 > * from. This correlation allows to route events specifically
1156 > * only to the requestor and not emit them to all listeners.
1157 > */
1158 > correlates(correlationId: number): boolean {
1159 return this.correlationId === correlationId;
1160 }
1161 > files.ts
1162 > /**
1163 > * Figure out if the event contains changes that correlate to one
1164 > * correlation identifier.
1165 > *
1166 > * File change event correlation is an advanced watch feature that
1167 > * allows to identify from which watch request the events originate
1168 > * from. This correlation allows to route events specifically
1169 > * only to the requestor and not emit them to all listeners.
1170 > */
1171 > hasCorrelation(): boolean {
1172 return typeof this.correlationId === 'number';
1173 }
1174 > files.ts
1175 > /**
1176 > * @deprecated use the `contains` or `affects` method to efficiently find
1177 > * out if the event relates to a given resource. these methods ensure:
1178 > * - that there is no expensive lookup needed (by using a `TernarySearchTree`)
1179 > * - correctly handles `FileChangeType.DELETED` events
1180 > */
1181 > readonly rawAdded: URI[] = [];
1182 >
1183 > /**
1184 > * @deprecated use the `contains` or `affects` method to efficiently find
1185 > * out if the event relates to a given resource. these methods ensure:
1186 > * - that there is no expensive lookup needed (by using a `TernarySearchTree`)
1187 > * - correctly handles `FileChangeType.DELETED` events
1188 > */
1189 > readonly rawUpdated: URI[] = [];
1190 >
1191 > /**
1192 > * @deprecated use the `contains` or `affects` method to efficiently find
1193 > * out if the event relates to a given resource. these methods ensure:
1194 > * - that there is no expensive lookup needed (by using a `TernarySearchTree`)
1195 > * - correctly handles `FileChangeType.DELETED` events
1196 > */
1197 > readonly rawDeleted: URI[] = [];
1198 > }
1199 >
1200 > export function isParent(path: string, candidate: string, ignoreCase?: boolean): boolean {
1201 if (!path || !candidate || path === candidate) {
1202 return false;
1217 return path.indexOf(candidate) === 0;
1218 }
1219 > files.ts
1220 > export interface IBaseFileStat {
1221 >
1222 > /**
1223 > * The unified resource identifier of this file or folder.
1224 > */
1225 > readonly resource: URI;
1226 >
1227 > /**
1228 > * The name which is the last segment
1229 > * of the {{path}}.
1230 > */
1231 > readonly name: string;
1232 >
1233 > /**
1234 > * The size of the file.
1235 > *
1236 > * The value may or may not be resolved as
1237 > * it is optional.
1238 > */
1239 > readonly size?: number;
1240 >
1241 > /**
1242 > * The last modification date represented as millis from unix epoch.
1243 > *
1244 > * The value may or may not be resolved as
1245 > * it is optional.
1246 > */
1247 > readonly mtime?: number;
1248 >
1249 > /**
1250 > * The creation date represented as millis from unix epoch.
1251 > *
1252 > * The value may or may not be resolved as
1253 > * it is optional.
1254 > */
1255 > readonly ctime?: number;
1256 >
1257 > /**
1258 > * A unique identifier that represents the
1259 > * current state of the file or directory.
1260 > *
1261 > * The value may or may not be resolved as
1262 > * it is optional.
1263 > */
1264 > readonly etag?: string;
1265 >
1266 > /**
1267 > * File is readonly. Components like editors should not
1268 > * offer to edit the contents.
1269 > */
1270 > readonly readonly?: boolean;
1271 >
1272 > /**
1273 > * File is locked. Components like editors should offer
1274 > * to edit the contents and ask the user upon saving to
1275 > * remove the lock.
1276 > */
1277 > readonly locked?: boolean;
1278 >
1279 > /**
1280 > * File is executable. Relevant for Unix-like systems where
1281 > * the executable bit determines if a file can be run.
1282 > */
1283 > readonly executable?: boolean;
1284 > }
1285 >
1286 > export interface IBaseFileStatWithMetadata extends Required<IBaseFileStat> { }
1287 >
1288 > /**
1289 > * A file resource with meta information and resolved children if any.
1290 > */
1291 > export interface IFileStat extends IBaseFileStat {
1292 >
1293 > /**
1294 > * The resource is a file.
1295 > */
1296 > readonly isFile: boolean;
1297 >
1298 > /**
1299 > * The resource is a directory.
1300 > */
1301 > readonly isDirectory: boolean;
1302 >
1303 > /**
1304 > * The resource is a symbolic link. Note: even when the
1305 > * file is a symbolic link, you can test for `FileType.File`
1306 > * and `FileType.Directory` to know the type of the target
1307 > * the link points to.
1308 > */
1309 > readonly isSymbolicLink: boolean;
1310 >
1311 > /**
1312 > * The children of the file stat or undefined if none.
1313 > */
1314 > children: IFileStat[] | undefined;
1315 > }
1316 >
1317 > export interface IFileStatWithMetadata extends IFileStat, IBaseFileStatWithMetadata {
1318 > readonly mtime: number;
1319 > readonly ctime: number;
1320 > readonly etag: string;
1321 > readonly size: number;
1322 > readonly readonly: boolean;
1323 > readonly locked: boolean;
1324 > readonly executable: boolean;
1325 > readonly children: IFileStatWithMetadata[] | undefined;
1326 > }
1327 >
1328 > export interface IFileStatResult {
1329 > readonly stat?: IFileStat;
1330 > readonly success: boolean;
1331 > }
1332 >
1333 > export interface IFileStatResultWithMetadata extends IFileStatResult {
1334 > readonly stat?: IFileStatWithMetadata;
1335 > }
1336 >
1337 > export interface IFileStatWithPartialMetadata extends Omit<IFileStatWithMetadata, 'children'> { }
1338 >
1339 > export interface IFileContent extends IBaseFileStatWithMetadata {
1340 >
1341 > /**
1342 > * The content of a file as buffer.
1343 > */
1344 > readonly value: VSBuffer;
1345 > }
1346 >
1347 > export interface IFileStreamContent extends IBaseFileStatWithMetadata {
1348 >
1349 > /**
1350 > * The content of a file as stream.
1351 > */
1352 > readonly value: VSBufferReadableStream;
1353 > }
1354 >
1355 > export interface IBaseReadFileOptions extends IFileReadStreamOptions {
1356 >
1357 > /**
1358 > * The optional etag parameter allows to return early from resolving the resource if
1359 > * the contents on disk match the etag. This prevents accumulated reading of resources
1360 > * that have been read already with the same etag.
1361 > * It is the task of the caller to makes sure to handle this error case from the promise.
1362 > */
1363 > readonly etag?: string;
1364 > }
1365 >
1366 > export interface IReadFileStreamOptions extends IBaseReadFileOptions { }
1367 >
1368 > export interface IReadFileOptions extends IBaseReadFileOptions {
1369 >
1370 > /**
1371 > * The optional `atomic` flag can be used to make sure
1372 > * the `readFile` method is not running in parallel with
1373 > * any `write` operations in the same process.
1374 > *
1375 > * Typically you should not need to use this flag but if
1376 > * for example you are quickly reading a file right after
1377 > * a file event occurred and the file changes a lot, there
1378 > * is a chance that a read returns an empty or partial file
1379 > * because a pending write has not finished yet.
1380 > *
1381 > * Note: this does not prevent the file from being written
1382 > * to from a different process. If you need such atomic
1383 > * operations, you better use a real database as storage.
1384 > */
1385 > readonly atomic?: boolean;
1386 > }
1387 >
1388 > export interface IWriteFileOptions {
1389 >
1390 > /**
1391 > * The last known modification time of the file. This can be used to prevent dirty writes.
1392 > */
1393 > readonly mtime?: number;
1394 >
1395 > /**
1396 > * The etag of the file. This can be used to prevent dirty writes.
1397 > */
1398 > readonly etag?: string;
1399 >
1400 > /**
1401 > * Whether to attempt to unlock a file before writing.
1402 > */
1403 > readonly unlock?: boolean;
1404 >
1405 > /**
1406 > * The optional `atomic` flag can be used to make sure
1407 > * the `writeFile` method updates the target file atomically
1408 > * by first writing to a temporary file in the same folder
1409 > * and then renaming it over the target.
1410 > */
1411 > readonly atomic?: IFileAtomicOptions | false;
1412 >
1413 > /**
1414 > * If set to true, will append to the end of the file instead of
1415 > * replacing its contents. Will create the file if it doesn't exist.
1416 > */
1417 > readonly append?: boolean;
1418 > }
1419 >
1420 > export interface IResolveFileOptions {
1421 >
1422 > /**
1423 > * Automatically continue resolving children of a directory until the provided resources
1424 > * are found.
1425 > */
1426 > readonly resolveTo?: readonly URI[];
1427 >
1428 > /**
1429 > * Automatically continue resolving children of a directory if the number of children is 1.
1430 > */
1431 > readonly resolveSingleChildDescendants?: boolean;
1432 >
1433 > /**
1434 > * Will resolve mtime, ctime, size and etag of files if enabled. This can have a negative impact
1435 > * on performance and thus should only be used when these values are required.
1436 > */
1437 > readonly resolveMetadata?: boolean;
1438 > }
1439 >
1440 > export interface IResolveMetadataFileOptions extends IResolveFileOptions {
1441 > readonly resolveMetadata: true;
1442 > }
1443 >
1444 > export interface ICreateFileOptions {
1445 >
1446 > /**
1447 > * Overwrite the file to create if it already exists on disk. Otherwise
1448 > * an error will be thrown (FILE_MODIFIED_SINCE).
1449 > */
1450 > readonly overwrite?: boolean;
1451 > }
1452 >
1453 > export class FileOperationError extends Error {
1454 > constructor(
1455 message: string,
1456 readonly fileOperationResult: FileOperationResult,
1459 super(message);
1460 }
1461 > } files.ts
1462 >
1463 > export class TooLargeFileOperationError extends FileOperationError {
1464 > constructor(
1465 message: string,
1466 override readonly fileOperationResult: FileOperationResult.FILE_TOO_LARGE,
1470 super(message, fileOperationResult, options);
1471 }
1472 > } files.ts
1473 >
1474 > export class NotModifiedSinceFileOperationError extends FileOperationError {
1475 >
1476 > constructor(
1477 message: string,
1478 readonly stat: IFileStatWithMetadata,
1481 super(message, FileOperationResult.FILE_NOT_MODIFIED_SINCE, options);
1482 }
1483 > } files.ts
1484 >
1485 > export const enum FileOperationResult {
1486 > FILE_IS_DIRECTORY,
1487 > FILE_NOT_FOUND,
1488 > FILE_NOT_MODIFIED_SINCE,
1489 > FILE_MODIFIED_SINCE,
1490 > FILE_MOVE_CONFLICT,
1491 > FILE_WRITE_LOCKED,
1492 > FILE_PERMISSION_DENIED,
1493 > FILE_TOO_LARGE,
1494 > FILE_INVALID_PATH,
1495 > FILE_NOT_DIRECTORY,
1496 > FILE_OTHER_ERROR
1497 > }
1498 >
1499 > //#endregion
1500 >
1501 > //#region Settings
1502 >
1503 > export const AutoSaveConfiguration = {
1504 > OFF: 'off',
1505 > AFTER_DELAY: 'afterDelay',
1506 > ON_FOCUS_CHANGE: 'onFocusChange',
1507 > ON_WINDOW_CHANGE: 'onWindowChange'
1508 > };
1509 >
1510 > export const HotExitConfiguration = {
1511 > OFF: 'off',
1512 > ON_EXIT: 'onExit',
1513 > ON_EXIT_AND_WINDOW_CLOSE: 'onExitAndWindowClose'
1514 > };
1515 >
1516 > export const FILES_ASSOCIATIONS_CONFIG = 'files.associations';
1517 > export const FILES_EXCLUDE_CONFIG = 'files.exclude';
1518 > export const FILES_READONLY_INCLUDE_CONFIG = 'files.readonlyInclude';
1519 > export const FILES_READONLY_EXCLUDE_CONFIG = 'files.readonlyExclude';
1520 > export const FILES_READONLY_FROM_PERMISSIONS_CONFIG = 'files.readonlyFromPermissions';
1521 >
1522 > export interface IGlobPatterns {
1523 > [filepattern: string]: boolean;
1524 > }
1525 >
1526 > export interface IFilesConfiguration {
1527 > files?: IFilesConfigurationNode;
1528 > }
1529 >
1530 > export interface IFilesConfigurationNode {
1531 > associations: { [filepattern: string]: string };
1532 > exclude: IExpression;
1533 > watcherExclude: IGlobPatterns;
1534 > watcherInclude: string[];
1535 > encoding: string;
1536 > autoGuessEncoding: boolean;
1537 > candidateGuessEncodings: string[];
1538 > defaultLanguage: string;
1539 > trimTrailingWhitespace: boolean;
1540 > autoSave: string;
1541 > autoSaveDelay: number;
1542 > autoSaveWorkspaceFilesOnly: boolean;
1543 > autoSaveWhenNoErrors: boolean;
1544 > eol: string;
1545 > enableTrash: boolean;
1546 > hotExit: string;
1547 > saveConflictResolution: 'askUser' | 'overwriteFileOnDisk';
1548 > readonlyInclude: IGlobPatterns;
1549 > readonlyExclude: IGlobPatterns;
1550 > readonlyFromPermissions: boolean;
1551 > }
1552 >
1553 > //#endregion
1554 >
1555 > //#region Utilities
1556 >
1557 > export enum FileKind {
1558 > FILE,
1559 > FOLDER,
1560 > ROOT_FOLDER
1561 > }
1562 >
1563 > /**
1564 > * A hint to disable etag checking for reading/writing.
1565 > */
1566 > export const ETAG_DISABLED = '';
1567 >
1568 > export function etag(stat: { mtime: number; size: number }): string;
1569 > export function etag(stat: { mtime: number | undefined; size: number | undefined }): string | undefined;
1570 > export function etag(stat: { mtime: number | undefined; size: number | undefined }): string | undefined {
1571 if (typeof stat.size !== 'number' || typeof stat.mtime !== 'number') {
1572 return undefined;
1575 return stat.mtime.toString(29) + stat.size.toString(31);
1576 }
1577 > files.ts
1578 export async function whenProviderRegistered(file: URI, fileService: IFileService): Promise<void> {
1579 if (fileService.hasProvider(URI.from({ scheme: file.scheme }))) {
1590 });
1591 }
1592 > files.ts
1593 > /**
1594 > * Helper to format a raw byte size into a human readable label.
1595 > */
1596 > export class ByteSize {
1597 >
1598 > static readonly KB = 1024;
1599 > static readonly MB = ByteSize.KB * ByteSize.KB;
1600 > static readonly GB = ByteSize.MB * ByteSize.KB;
1601 > static readonly TB = ByteSize.GB * ByteSize.KB;
1602 >
1603 > static formatSize(size: number): string {
1604 if (!isNumber(size)) {
1605 size = 0;
1624 return localize('sizeTB', "{0}TB", (size / ByteSize.TB).toFixed(2));
1625 }
1626 > } files.ts
1627 >
1628 > // File limits
1629 >
1630 > export function getLargeFileConfirmationLimit(remoteAuthority?: string): number;
1631 > export function getLargeFileConfirmationLimit(uri?: URI): number;
1632 > export function getLargeFileConfirmationLimit(arg?: string | URI): number {
1633 const isRemote = typeof arg === 'string' || arg?.scheme === Schemas.vscodeRemote;
1634 const isLocal = typeof arg !== 'string' && arg?.scheme === Schemas.file;
1655 return 1024 * ByteSize.MB;
1656 }
1657 > files.ts
1658 > //#endregion
src/vs/platform/agentHost/common/state/protocol/channels-session/state.ts 1351 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { Changeset } from '../channels-changeset/state.js';
10 > import type { AnnotationsSummary } from '../channels-annotations/state.js';
11 > import type { ChatSummary, ChatInputRequest, ToolCallConfirmationState, ToolCallState, ToolCallAuthRequiredState } from '../channels-chat/state.js';
12 > import type { ConfigPropertySchema, ErrorInfo, Icon, ProtectedResourceMetadata, TextRange, URI } from '../common/state.js';
13 >
14 > // ─── Session State ───────────────────────────────────────────────────────────
15 >
16 > /**
17 > * Session initialization state.
18 > *
19 > * @category Session State
20 > */
21 > export const enum SessionLifecycle {
22 > Creating = 'creating',
23 > Ready = 'ready',
24 > CreationFailed = 'creationFailed',
25 > }
26 >
27 > /**
28 > * Bitset of summary-level session status flags.
29 > *
30 > * Use bitwise checks instead of equality for non-terminal activity. For example,
31 > * `status & SessionStatus.InProgress` matches both ordinary in-progress turns
32 > * and turns that are paused waiting for input.
33 > *
34 > * @category Session State
35 > */
36 > export const enum SessionStatus {
37 > /** Session is idle — no turn is active. */
38 > Idle = 1,
39 > /** Session ended with an error. */
40 > Error = 1 << 1,
41 > /** A turn is actively streaming. */
42 > InProgress = 1 << 3,
43 > /** A turn is in progress but blocked waiting for user input or tool confirmation. */
44 > InputNeeded = (1 << 3) | (1 << 4),
45 > /** The client has viewed this session since its last modification. */
46 > IsRead = 1 << 5,
47 > /** The session has been archived by the client. */
48 > IsArchived = 1 << 6,
49 > }
50 >
51 > /**
52 > * Metadata shared between the full {@link SessionState} (delivered when a
53 > * client subscribes to a session's URI) and the lightweight
54 > * {@link SessionSummary} (carried in the root-channel session catalog).
55 > *
56 > * These fields describe the session at a glance and appear in both places.
57 > * `SessionState` owns the authoritative values for a subscribed session;
58 > * `SessionSummary` mirrors them into the catalog so clients that only render a
59 > * session list don't have to subscribe to every session URI. The host keeps
60 > * the catalog in sync via `root/sessionSummaryChanged`.
61 > *
62 > * @category Session State
63 > */
64 > export interface SessionMetadata {
65 > /** Agent provider ID */
66 > provider: string;
67 > /** Session title */
68 > title: string;
69 > /** Current session status */
70 > status: SessionStatus;
71 > /** Human-readable description of what the session is currently doing */
72 > activity?: string;
73 > /** Server-owned project for this session */
74 > project?: ProjectInfo;
75 > /**
76 > * The working directories the session's agent has tool access to, as
77 > * maintained by the `session/workingDirectorySet` /
78 > * `session/workingDirectoryRemoved` actions. Directories are **equal peers** —
79 > * the session has no primary. Individual chats MAY restrict to a subset via
80 > * {@link ChatSummary.workingDirectories | their own `workingDirectories`} and
81 > * designate one of their own directories as primary (see
82 > * {@link ChatState.primaryWorkingDirectory}); a chat that sets no subset
83 > * operates against this full set.
84 > */
85 > workingDirectories?: URI[];
86 > /**
87 > * Lightweight summary of this session's inline annotations channel
88 > * (`ahp-session:/<uuid>/annotations`). Surfaced so badge UI can render
89 > * annotation / entry counts without subscribing. Absent when the session
90 > * does not expose an annotations channel.
91 > */
92 > annotations?: AnnotationsSummary;
93 > }
94 >
95 > /**
96 > * Full state for a single session, loaded when a client subscribes to the session's URI.
97 > *
98 > * Inlines (denormalizes) every {@link SessionMetadata} field directly onto
99 > * itself so subscribers receive one flat object instead of a nested summary.
100 > * The lightweight catalog representation is {@link SessionSummary}, surfaced on
101 > * the root channel; the host keeps the two in sync via
102 > * `root/sessionSummaryChanged`.
103 > *
104 > * @category Session State
105 > */
106 > export interface SessionState extends SessionMetadata {
107 > /** Session initialization state */
108 > lifecycle: SessionLifecycle;
109 > /** Error details if creation failed */
110 > creationError?: ErrorInfo;
111 > /** Tools provided by the server (agent host) for this session */
112 > serverTools?: ToolDefinition[];
113 > /**
114 > * The clients currently providing tools and interactive capabilities to this
115 > * session. If multiple tools or customizations are provided by the same
116 > * active client, an agent host MAY deduplicate them when exposed to a model,
117 > * with a preference given to the client that started the turn.
118 > *
119 > * Membership is host-managed: clients add (or refresh) themselves with
120 > * `session/activeClientSet`, and the host removes them with
121 > * `session/activeClientRemoved` when they unsubscribe, disconnect without
122 > * reconnecting in time, or reconnect without resubscribing to the session.
123 > */
124 > activeClients: SessionActiveClient[];
125 > /** Catalog of chats in this session. */
126 > chats: ChatSummary[];
127 > /**
128 > * The chat that receives input when the user addresses the session without
129 > * selecting a specific chat. This is a UI routing hint, not a hierarchy
130 > * marker — chats remain equal peers at the protocol level. Hosts MAY change
131 > * this over the session's lifetime.
132 > */
133 > defaultChat?: URI;
134 > /** Session configuration schema and current values */
135 > config?: SessionConfigState;
136 > /**
137 > * Top-level customizations active in this session.
138 > *
139 > * Always one of the {@link Customization} variants:
140 > *
141 > * - Container customizations ({@link PluginCustomization},
142 > * {@link DirectoryCustomization}) whose children — agents, skills,
143 > * prompts, rules, hooks, MCP servers — live in each container's
144 > * {@link ContainerCustomizationBase.children | `children`} array.
145 > * - Top-level {@link McpServerCustomization} entries the host
146 > * surfaces directly (for example a globally-configured MCP server
147 > * that isn't bundled in a plugin or directory). MCP servers may
148 > * also appear as children of a container.
149 > *
150 > * Client-published plugins arrive via
151 > * {@link SessionActiveClient.customizations | `activeClients[].customizations`}
152 > * and the host propagates them into this list (typically with the
153 > * container's `clientId` set and `children` populated). Clients
154 > * publish in container shape only; bare MCP servers at the top level
155 > * are server-originated.
156 > */
157 > customizations?: Customization[];
158 > /**
159 > * Catalogue of changesets the server can produce for this session. Each
160 > * entry advertises a subscribable view of file changes (uncommitted,
161 > * session-wide, per-turn, etc.) and the URI template the client expands
162 > * before subscribing. See {@link Changeset} for the full shape and
163 > * {@link /guide/changesets | Changesets} for an overview of the model.
164 > */
165 > changesets?: Changeset[];
166 > /**
167 > * Outstanding input the session is blocked on, aggregated across every chat
168 > * so a client can discover and answer it from the session channel alone,
169 > * without subscribing to individual chats.
170 > *
171 > * Each entry is self-sufficient: it carries the owning chat's URI plus every
172 > * identifier the client needs to respond. A client answers by dispatching the
173 > * ordinary `chat/*` action to that chat's channel — see
174 > * {@link SessionInputRequest} for the per-variant response path. A present,
175 > * non-empty list implies {@link SessionStatus.InputNeeded} on
176 > * {@link SessionSummary.status}.
177 > *
178 > * Host-managed: the host upserts entries with `session/inputNeededSet` as
179 > * chats raise requests and removes them with `session/inputNeededRemoved`
180 > * once the underlying request resolves.
181 > */
182 > inputNeeded?: SessionInputRequest[];
183 > /**
184 > * Additional provider-specific metadata for this session.
185 > *
186 > * Clients MAY look for well-known keys here to provide enhanced UI.
187 > * For example, a `git` key may provide extra git metadata about the session's
188 > * working directories.
189 > */
190 > _meta?: Record<string, unknown>;
191 > }
192 >
193 > /**
194 > * A client currently providing tools and interactive capabilities to a session.
195 > *
196 > * A session MAY have several active clients at once; entries in
197 > * {@link SessionState.activeClients} are keyed by `clientId`. The server SHOULD
198 > * automatically remove an active client when that client disconnects.
199 > *
200 > * @category Session State
201 > */
202 > export interface SessionActiveClient {
203 > /** Client identifier (matches `clientId` from `initialize`) */
204 > clientId: string;
205 > /** Human-readable client name (e.g. `"VS Code"`) */
206 > displayName?: string;
207 > /** Tools this client provides to the session */
208 > tools: ToolDefinition[];
209 > /**
210 > * Plugin customizations this client contributes to the session.
211 > *
212 > * Clients publish in [Open Plugins](https://open-plugins.com/) format
213 > * — i.e. always container-shaped plugins. They MAY synthesize virtual
214 > * plugins in memory and rely on the host to expand them into concrete
215 > * children inside {@link SessionState.customizations}.
216 > */
217 > customizations?: ClientPluginCustomization[];
218 > }
219 >
220 > // ─── Session Input Requests ──────────────────────────────────────────────────
221 >
222 > /**
223 > * Discriminant for the kinds of outstanding input a session can surface in
224 > * {@link SessionState.inputNeeded}.
225 > *
226 > * This is a general/typological union (not a lifecycle), so the discriminant is
227 > * a `*Kind`.
228 > *
229 > * @category Session Input Types
230 > */
231 > export const enum SessionInputRequestKind {
232 > /** A user-facing elicitation mirrored from an unresolved chat response part. */
233 > ChatInput = 'chatInput',
234 > /** A tool call awaiting parameter- or result-confirmation. */
235 > ToolConfirmation = 'toolConfirmation',
236 > /** A running tool the session wants an active client to execute. */
237 > ToolClientExecution = 'toolClientExecution',
238 > /** A tool call blocked on MCP authentication mid-execution. */
239 > ToolAuthentication = 'toolAuthentication',
240 > }
241 >
242 > /**
243 > * Fields common to every {@link SessionInputRequest} variant.
244 > *
245 > * @category Session Input Types
246 > */
247 > interface SessionInputRequestBase {
248 > /**
249 > * Stable key for this entry, unique within the session's
250 > * {@link SessionState.inputNeeded} list. The host derives it however it likes
251 > * (for example from the chat URI plus the underlying request or tool-call
252 > * id); consumers MUST treat it as opaque. It is the key for the
253 > * `session/inputNeededSet` / `session/inputNeededRemoved` upsert convention.
254 > */
255 > id: string;
256 > /**
257 > * The chat the underlying request lives in. This is the channel a client
258 > * dispatches its response to — it does not need to have subscribed to that
259 > * chat first.
260 > */
261 > chat: URI;
262 > }
263 >
264 > /**
265 > * A user-input elicitation surfaced at the session level, mirroring the request
266 > * from an unresolved {@link InputRequestResponsePart} in the owning chat.
267 > *
268 > * Respond by dispatching `chat/inputCompleted` (or syncing drafts with
269 > * `chat/inputAnswerChanged`) to {@link SessionInputRequestBase.chat | `chat`},
270 > * keyed by {@link ChatInputRequest.id | `request.id`}.
271 > *
272 > * @category Session Input Types
273 > */
274 > export interface SessionChatInputRequest extends SessionInputRequestBase {
275 > kind: SessionInputRequestKind.ChatInput;
276 > /** The mirrored chat input request. */
277 > request: ChatInputRequest;
278 > }
279 >
280 > /**
281 > * A tool call blocked on confirmation — either parameter confirmation before
282 > * execution or result confirmation after — surfaced at the session level.
283 > *
284 > * Respond by dispatching `chat/toolCallConfirmed` (for
285 > * {@link ToolCallPendingConfirmationState}) or `chat/toolCallResultConfirmed`
286 > * (for {@link ToolCallPendingResultConfirmationState}) to
287 > * {@link SessionInputRequestBase.chat | `chat`}, keyed by `turnId` and
288 > * `toolCall.toolCallId`.
289 > *
290 > * @category Session Input Types
291 > */
292 > export interface SessionToolConfirmationRequest extends SessionInputRequestBase {
293 > kind: SessionInputRequestKind.ToolConfirmation;
294 > /** The turn the tool call belongs to. */
295 > turnId: string;
296 > /** The tool call awaiting confirmation. */
297 > toolCall: ToolCallConfirmationState;
298 > }
299 >
300 > /**
301 > * A running tool whose execution is delegated to an active client. Surfaced so
302 > * a client that provides the tool can pick up the work without subscribing to
303 > * the owning chat.
304 > *
305 > * The {@link toolCall} is always a {@link ToolCallRunningState} (a
306 > * {@link ToolCallState} in `running` status) whose
307 > * {@link ToolCallRunningState.contributor | `contributor`} is a client
308 > * {@link ToolCallClientContributor} whose `clientId` matches the denormalized
309 > * {@link clientId} here. Execute and report the result by dispatching
310 > * `chat/toolCallComplete` (and optionally streaming with
311 > * `chat/toolCallContentChanged`) to {@link SessionInputRequestBase.chat |
312 > * `chat`}, keyed by `turnId` and `toolCall.toolCallId`.
313 > *
314 > * @category Session Input Types
315 > */
316 > export interface SessionToolClientExecutionRequest extends SessionInputRequestBase {
317 > kind: SessionInputRequestKind.ToolClientExecution;
318 > /** The turn the tool call belongs to. */
319 > turnId: string;
320 > /**
321 > * The `clientId` expected to execute the tool. Matches the `clientId` of the
322 > * tool call's client {@link ToolCallContributor}.
323 > */
324 > clientId: string;
325 > /**
326 > * The running tool call the session wants the owning client to execute. The
327 > * host only ever populates this with a {@link ToolCallRunningState} (i.e. a
328 > * {@link ToolCallState} in `running` status).
329 > */
330 > toolCall: ToolCallState;
331 > }
332 >
333 > /**
334 > * A tool call blocked on MCP authentication mid-execution, surfaced at the
335 > * session level.
336 > *
337 > * The {@link toolCall} is always a {@link ToolCallAuthRequiredState} (a
338 > * {@link ToolCallState} in `auth-required` status). Unlike
339 > * {@link SessionToolConfirmationRequest}, this is **not** answered by
340 > * dispatching a `chat/*` action directly: the client obtains a token for
341 > * {@link ToolCallAuthRequiredState.auth | `toolCall.auth`}`.resource` and
342 > * pushes it via the existing `authenticate` command (see
343 > * {@link /specification/authentication | Authentication}). The host resumes
344 > * the tool call and dispatches `chat/toolCallAuthResolved` once the token is
345 > * accepted, at which point it also removes this entry with
346 > * `session/inputNeededRemoved`.
347 > *
348 > * @category Session Input Types
349 > */
350 > export interface SessionToolAuthenticationRequest extends SessionInputRequestBase {
351 > kind: SessionInputRequestKind.ToolAuthentication;
352 > /** The turn the tool call belongs to. */
353 > turnId: string;
354 > /** The tool call awaiting authentication. */
355 > toolCall: ToolCallAuthRequiredState;
356 > }
357 >
358 > /**
359 > * One outstanding piece of input a session is blocked on, aggregated across all
360 > * chats in {@link SessionState.inputNeeded}.
361 > *
362 > * Each entry is self-sufficient: it carries the owning
363 > * {@link SessionInputRequestBase.chat | `chat`} URI plus every identifier needed
364 > * to construct the response, so a client can answer by dispatching the ordinary
365 > * `chat/*` action (`chat/inputCompleted`, `chat/toolCallConfirmed`,
366 > * `chat/toolCallComplete`, …) to that chat's channel **without having subscribed
367 > * to the chat** — except {@link SessionToolAuthenticationRequest}, which is
368 > * resolved via the `authenticate` command instead. The host removes the entry
369 > * with `session/inputNeededRemoved` once the underlying request resolves.
370 > *
371 > * @category Session Input Types
372 > */
373 > export type SessionInputRequest =
374 > | SessionChatInputRequest
375 > | SessionToolConfirmationRequest
376 > | SessionToolClientExecutionRequest
377 > | SessionToolAuthenticationRequest;
378 >
379 > /**
380 > * Server-owned project metadata for a session.
381 > *
382 > * @category Session State
383 > */
384 > export interface ProjectInfo {
385 > /** Project URI */
386 > uri: URI;
387 > /** Human-readable project name */
388 > displayName: string;
389 > }
390 >
391 > /**
392 > * Lightweight catalog entry summarizing one session. Surfaced via
393 > * {@link RootChannelCommands.listSessions | `root/listSessions`} and
394 > * `root/sessionAdded`/`root/sessionSummaryChanged` notifications.
395 > *
396 > * **Aggregation across chats.** Once a session contains more than one chat,
397 > * several `SessionSummary` fields are derived from the underlying
398 > * {@link SessionState.chats | chat catalog}. Producers SHOULD follow these
399 > * rules so clients that only consume the session summary (e.g. a session
400 > * list) still see meaningful state:
401 > *
402 > * - `status`: take the activity bits (`Idle` / `InProgress` / `InputNeeded` /
403 > * `Error` — bits 0–4) from the
404 > * {@link SessionState.defaultChat | default chat} when present, else from
405 > * the most recently modified chat. **Promote** `InputNeeded` whenever any
406 > * chat in the session needs input, and **promote** `Error` whenever any
407 > * chat is in an error state — both override the default-chat bits. The
408 > * orthogonal flag bits (`IsRead`, `IsArchived`) remain session-scoped.
409 > * - `activity`: mirror the activity string of the default chat, or of the
410 > * chat currently driving the promoted status bits when a non-default chat
411 > * wins (e.g. the chat that raised `InputNeeded`).
412 > * - `modifiedAt`: the max of all chats' `modifiedAt`.
413 > * - `workingDirectories`: the session-level set. Individual chats MAY restrict
414 > * to a subset via {@link ChatSummary.workingDirectories}; aggregating these
415 > * up is meaningless and SHOULD NOT be attempted.
416 > * - `changes`: optional roll-up across all chats. Producers MAY sum the
417 > * per-chat changeset stats or report the most expensive chat's stats —
418 > * whichever is cheaper for the host to compute.
419 > *
420 > * Sessions with a single chat trivially satisfy all of the above (the chat's
421 > * values pass through unchanged). The rules only matter once a session
422 > * carries multiple chats.
423 > *
424 > * @category Session State
425 > */
426 > export interface SessionSummary extends SessionMetadata {
427 > /** Session URI */
428 > resource: URI;
429 > /** Creation timestamp (ISO 8601, e.g. `"2025-03-10T18:42:03.123Z"`) */
430 > createdAt: string;
431 > /** Last modification timestamp (ISO 8601, e.g. `"2025-03-10T18:42:03.123Z"`) */
432 > modifiedAt: string;
433 > /**
434 > * Aggregate summary of file changes associated with this session. Servers
435 > * may populate this to give clients a quick at-a-glance view of the
436 > * session's footprint (e.g., for list rendering) without requiring the
437 > * client to subscribe to a changeset.
438 > */
439 > changes?: ChangesSummary;
440 > /**
441 > * Lightweight server-defined metadata clients may use for the session
442 > * presentation. The protocol does not interpret these values; producers
443 > * SHOULD keep the payload small because summaries appear in session lists
444 > * and session notifications.
445 > */
446 > _meta?: Record<string, unknown>;
447 > }
448 >
449 > /**
450 > * Aggregate counts describing the file changes associated with a session.
451 > *
452 > * All fields are optional so servers can populate only the metrics they
453 > * cheaply have available.
454 > *
455 > * @category Session State
456 > */
457 > export interface ChangesSummary {
458 > /** Total number of inserted lines across all changed files. */
459 > additions?: number;
460 > /** Total number of deleted lines across all changed files. */
461 > deletions?: number;
462 > /** Number of files that have changes. */
463 > files?: number;
464 > }
465 >
466 > // ─── Agent Selection ─────────────────────────────────────────────────────────
467 >
468 > /**
469 > * A selected custom agent for a session.
470 > *
471 > * The `uri` identifies a specific custom agent (matching an
472 > * {@link AgentCustomization.uri | `AgentCustomization.uri`} exposed via
473 > * the session's effective customizations). Consumers resolve the agent's
474 > * display name by looking up `uri` in the session's customization tree.
475 > *
476 > * A message with no `agent` selected uses the provider's default behavior.
477 > *
478 > * @category Session State
479 > */
480 > export interface AgentSelection {
481 > /** Stable agent URI (matches an {@link AgentCustomization.uri}). */
482 > uri: URI;
483 > }
484 >
485 > // ─── Session Config Types ────────────────────────────────────────────────────
486 >
487 > /**
488 > * A session configuration property descriptor.
489 > *
490 > * Extends the generic {@link ConfigPropertySchema} with session-specific
491 > * display extensions.
492 > *
493 > * @category Session Config Types
494 > */
495 > export interface SessionConfigPropertySchema extends ConfigPropertySchema {
496 > /**
497 > * Display extension: when `true`, the full set of allowed values is too large
498 > * to enumerate statically. The client SHOULD use `sessionConfigCompletions`
499 > * to fetch matching values based on user input. Any values in `enum` are
500 > * seed/recent values for initial display.
501 > */
502 > enumDynamic?: boolean;
503 > /** When `true`, the user may change this property after session creation */
504 > sessionMutable?: boolean;
505 > }
506 >
507 > /**
508 > * A JSON Schema object describing available session configuration metadata.
509 > *
510 > * @category Session Config Types
511 > */
512 > export interface SessionConfigSchema {
513 > /** JSON Schema: always `'object'` */
514 > type: 'object';
515 > /** JSON Schema: property descriptors keyed by property id */
516 > properties: Record<string, SessionConfigPropertySchema>;
517 > /** JSON Schema: list of required property ids */
518 > required?: string[];
519 > }
520 >
521 > /**
522 > * Live session configuration metadata.
523 > *
524 > * The schema describes the available configuration properties and the values
525 > * contain the current value for each resolved property.
526 > *
527 > * @category Session Config Types
528 > */
529 > export interface SessionConfigState {
530 > /** JSON Schema describing available configuration properties */
531 > schema: SessionConfigSchema;
532 > /** Current configuration values */
533 > values: Record<string, unknown>;
534 > }
535 >
536 > // ─── Tool Definition Types ───────────────────────────────────────────────────
537 >
538 > /**
539 > * Describes a tool available in a session, provided by either the server or the active client.
540 > *
541 > * @category Tool Definition Types
542 > */
543 > export interface ToolDefinition {
544 > /** Unique tool identifier */
545 > name: string;
546 > /** Human-readable display name */
547 > title?: string;
548 > /** Description of what the tool does */
549 > description?: string;
550 > /**
551 > * JSON Schema defining the expected input parameters.
552 > *
553 > * Optional because client-provided tools may not have formal schemas.
554 > * Mirrors MCP `Tool.inputSchema`.
555 > */
556 > inputSchema?: {
557 > type: 'object';
558 > properties?: Record<string, object>;
559 > required?: string[];
560 > };
561 > /**
562 > * JSON Schema defining the structure of the tool's output.
563 > *
564 > * Mirrors MCP `Tool.outputSchema`.
565 > */
566 > outputSchema?: {
567 > type: 'object';
568 > properties?: Record<string, object>;
569 > required?: string[];
570 > };
571 > /** Behavioral hints about the tool. All properties are advisory. */
572 > annotations?: ToolAnnotations;
573 > /**
574 > * Additional provider-specific metadata.
575 > *
576 > * Mirrors the MCP `_meta` convention.
577 > */
578 > _meta?: Record<string, unknown>;
579 > }
580 >
581 > /**
582 > * Behavioral hints about a tool. All properties are advisory and not
583 > * guaranteed to faithfully describe tool behavior.
584 > *
585 > * Mirrors MCP `ToolAnnotations` from the Model Context Protocol specification.
586 > *
587 > * @category Tool Definition Types
588 > */
589 > export interface ToolAnnotations {
590 > /** Alternate human-readable title */
591 > title?: string;
592 > /** Tool does not modify its environment (default: false) */
593 > readOnlyHint?: boolean;
594 > /** Tool may perform destructive updates (default: true) */
595 > destructiveHint?: boolean;
596 > /** Repeated calls with the same arguments have no additional effect (default: false) */
597 > idempotentHint?: boolean;
598 > /** Tool may interact with external entities (default: true) */
599 > openWorldHint?: boolean;
600 > }
601 >
602 > // ─── Customization Types ─────────────────────────────────────────────────────
603 >
604 > /**
605 > * Discriminant for the kind of customization.
606 > *
607 > * Top-level entries in {@link SessionState.customizations} and
608 > * {@link AgentInfo.customizations} are either container customizations
609 > * ({@link CustomizationType.Plugin | `Plugin`} or
610 > * {@link CustomizationType.Directory | `Directory`}) or
611 > * {@link CustomizationType.McpServer | `McpServer`} entries surfaced
612 > * directly by the host. The remaining types appear only as children of
613 > * a container.
614 > *
615 > * @category Customization Types
616 > */
617 > export const enum CustomizationType {
618 > Plugin = 'plugin',
619 > Directory = 'directory',
620 > Agent = 'agent',
621 > Skill = 'skill',
622 > Prompt = 'prompt',
623 > Rule = 'rule',
624 > Hook = 'hook',
625 > McpServer = 'mcpServer',
626 > }
627 >
628 > /**
629 > * Customization types that appear as children of a
630 > * {@link PluginCustomization} or {@link DirectoryCustomization}.
631 > *
632 > * @category Customization Types
633 > */
634 > export type ChildCustomizationType =
635 > | CustomizationType.Agent
636 > | CustomizationType.Skill
637 > | CustomizationType.Prompt
638 > | CustomizationType.Rule
639 > | CustomizationType.Hook
640 > | CustomizationType.McpServer;
641 >
642 > /**
643 > * Fields shared by every customization variant.
644 > *
645 > * @category Customization Types
646 > */
647 > interface CustomizationBase {
648 > /**
649 > * Session-unique opaque identifier. Used by every action that targets a
650 > * specific customization. Minted by whoever publishes the customization
651 > * (typically the agent host).
652 > */
653 > id: string;
654 > /**
655 > * Source URI for this customization. A plugin URL, a file URI, or a
656 > * directory URI.
657 > *
658 > * For declarations that live inside a larger file — e.g. an MCP
659 > * server declared inline in a `plugins.json` manifest — `uri` points
660 > * to the containing file and {@link CustomizationBase.range | `range`}
661 > * narrows it to the declaration's span.
662 > */
663 > uri: URI;
664 > /** Human-readable name. */
665 > name: string;
666 > /** Icons for UI display. */
667 > icons?: Icon[];
668 > /**
669 > * Optional span within {@link CustomizationBase.uri | `uri`} when this
670 > * customization is a subset of a larger file (for example, one entry
671 > * in an inline `mcpServers` block of a `plugins.json` manifest).
672 > * Absent when the customization covers the whole resource.
673 > */
674 > range?: TextRange;
675 > /**
676 > * Additional provider-specific metadata for this customization.
677 > *
678 > * Mirrors the MCP `_meta` convention. Optional and opaque to the
679 > * protocol; producers and consumers agree on its contents
680 > * out-of-band.
681 > */
682 > _meta?: Record<string, unknown>;
683 > }
684 >
685 > /**
686 > * Discriminant values for {@link CustomizationLoadState}.
687 > *
688 > * @category Customization Types
689 > */
690 > export const enum CustomizationLoadStatus {
691 > Loading = 'loading',
692 > Loaded = 'loaded',
693 > Degraded = 'degraded',
694 > Error = 'error',
695 > }
696 >
697 > /**
698 > * Container is being loaded by the host.
699 > *
700 > * @category Customization Types
701 > */
702 > export interface CustomizationLoadingState {
703 > kind: CustomizationLoadStatus.Loading;
704 > }
705 >
706 > /**
707 > * Container loaded successfully.
708 > *
709 > * @category Customization Types
710 > */
711 > export interface CustomizationLoadedState {
712 > kind: CustomizationLoadStatus.Loaded;
713 > }
714 >
715 > /**
716 > * Container partially loaded but has warnings.
717 > *
718 > * @category Customization Types
719 > */
720 > export interface CustomizationDegradedState {
721 > kind: CustomizationLoadStatus.Degraded;
722 > /** Human-readable description of the warning. */
723 > message: string;
724 > }
725 >
726 > /**
727 > * Container failed to load.
728 > *
729 > * @category Customization Types
730 > */
731 > export interface CustomizationErrorState {
732 > kind: CustomizationLoadStatus.Error;
733 > /** Human-readable error message. */
734 > message: string;
735 > }
736 >
737 > /**
738 > * Discriminated load state for a container customization
739 > * ({@link PluginCustomization} or {@link DirectoryCustomization}).
740 > *
741 > * @category Customization Types
742 > */
743 > export type CustomizationLoadState =
744 > | CustomizationLoadingState
745 > | CustomizationLoadedState
746 > | CustomizationDegradedState
747 > | CustomizationErrorState;
748 >
749 > /**
750 > * Fields shared by container customizations.
751 > *
752 > * @category Customization Types
753 > */
754 > interface ContainerCustomizationBase extends CustomizationBase {
755 > /** Whether this container is currently enabled. */
756 > enabled: boolean;
757 > /**
758 > * `clientId` of the client that contributed this container. Absent for
759 > * server-originated entries.
760 > */
761 > clientId?: string;
762 > /**
763 > * Host-reported load state. Absent means the host has not yet reported
764 > * a load state for this container.
765 > */
766 > load?: CustomizationLoadState;
767 > /**
768 > * Children discovered inside this container.
769 > *
770 > * Absent means the host has not parsed this container yet. An empty
771 > * array means the host parsed the container and it contributes
772 > * nothing.
773 > */
774 > children?: ChildCustomization[];
775 > }
776 >
777 > /**
778 > * An [Open Plugins](https://open-plugins.com/) plugin.
779 > *
780 > * @category Customization Types
781 > */
782 > export interface PluginCustomization extends ContainerCustomizationBase {
783 > type: CustomizationType.Plugin;
784 > /**
785 > * Version of the plugin, sourced from the
786 > * [Open Plugins](https://open-plugins.com/) manifest's optional
787 > * `version` field (semver, e.g. `"1.2.0"`). Absent when the manifest
788 > * declares no version — the field is optional there — or the source
789 > * has no version concept. Provenance / display only: the host neither
790 > * parses nor enforces it.
791 > */
792 > version?: string;
793 > }
794 >
795 > /**
796 > * A {@link PluginCustomization} as published by a client. Extends the
797 > * server-facing shape with an opaque `nonce` so the host can detect when
798 > * the client's view of a plugin has changed and re-parse only as needed.
799 > *
800 > * Clients SHOULD include a `nonce`. Server-side fields like
801 > * {@link ContainerCustomizationBase.children | `children`} and
802 > * {@link ContainerCustomizationBase.load | `load`} are typically left
803 > * absent on publication and populated by the host when the resolved
804 > * plugin appears in {@link SessionState.customizations}.
805 > *
806 > * @category Customization Types
807 > */
808 > export interface ClientPluginCustomization extends PluginCustomization {
809 > /** Opaque version token used by the host to detect changes. */
810 > nonce?: string;
811 > }
812 >
813 > /**
814 > * A directory the host watches for this session.
815 > *
816 > * Presence in the customization list signals that the host may discover
817 > * customizations from this directory. When `writable` is `true`, clients
818 > * MAY persist new customizations into the directory using
819 > * [`resourceWrite`](/reference/common#resourcewrite); the host will
820 > * then surface the resulting child via the customization actions.
821 > *
822 > * The directory may not yet exist on disk.
823 > *
824 > * @category Customization Types
825 > */
826 > export interface DirectoryCustomization extends ContainerCustomizationBase {
827 > type: CustomizationType.Directory;
828 > /** Which child customization type this directory holds. */
829 > contents: ChildCustomizationType;
830 > /** Whether clients may write into this directory. */
831 > writable: boolean;
832 > }
833 >
834 > /**
835 > * Fields shared by the leaf child customizations that live inside a
836 > * container — {@link AgentCustomization}, {@link SkillCustomization},
837 > * {@link PromptCustomization}, {@link RuleCustomization}, and
838 > * {@link HookCustomization}.
839 > *
840 > * {@link McpServerCustomization} is also a child but does not extend this
841 > * base: it always carries an explicit {@link McpServerCustomization.enabled}
842 > * because it can appear as a top-level customization too.
843 > *
844 > * @category Customization Types
845 > */
846 > interface ChildCustomizationBase extends CustomizationBase {
847 > /**
848 > * Whether this child is individually enabled. Absent means enabled, so a
849 > * producer only needs to set it to surface a child that exists but is
850 > * turned off on its own.
851 > *
852 > * This flag is independent of the parent container's: the **effective**
853 > * enabled state of a child is
854 > * `container.enabled && (child.enabled ?? true)`, so a disabled container
855 > * disables every child regardless of each child's own flag.
856 > *
857 > * A child is turned on or off by id with
858 > * {@link SessionCustomizationToggledAction | `session/customizationToggled`}.
859 > */
860 > enabled?: boolean;
861 > }
862 >
863 > /**
864 > * A custom agent contributed by a plugin or directory.
865 > *
866 > * Mirrors the [Open Plugins agent](https://open-plugins.com/agent-builders/components/agents)
867 > * format: a markdown file with YAML frontmatter, where the body is the
868 > * agent's system prompt.
869 > *
870 > * @category Customization Types
871 > */
872 > export interface AgentCustomization extends ChildCustomizationBase {
873 > type: CustomizationType.Agent;
874 > /**
875 > * Short description of what the agent specializes in and when to
876 > * invoke it. Sourced from the agent file's frontmatter `description`.
877 > */
878 > description?: string;
879 > /**
880 > * Model the agent is pinned to, sourced from the agent file's
881 > * frontmatter `model`. Absent means the agent inherits the session's
882 > * default model.
883 > */
884 > model?: string;
885 > /**
886 > * Allowlist of tool names the agent is scoped to, sourced from the
887 > * agent file's frontmatter `tools`. A non-empty list restricts the
888 > * agent to exactly those tools. Absent — or an empty list — imposes no
889 > * restriction beyond the session default: the agent may use any
890 > * available tool. Producers express "no restriction" by omitting the
891 > * field rather than sending an empty array, so an empty list carries no
892 > * meaning distinct from absence.
893 > */
894 > tools?: string[];
895 > /**
896 > * When `true`, the agent will not auto-delegate to this custom agent
897 > * as a sub-agent; it can only be selected by the user. Absent or
898 > * `false` means the agent may delegate to it.
899 > */
900 > disableModelInvocation?: boolean;
901 > /**
902 > * When `true`, the user cannot select this custom agent (for example,
903 > * in a picker); it remains available for the agent to auto-delegate
904 > * to. Absent or `false` means the user may select it.
905 > */
906 > disableUserInvocation?: boolean;
907 > }
908 >
909 > /**
910 > * A skill contributed by a plugin or directory.
911 > *
912 > * Covers both [Open Plugins skill formats](https://open-plugins.com/agent-builders/components/skills)
913 > * — the `skills/` directory layout (one subdirectory per skill, each with
914 > * a `SKILL.md`) and the flatter `commands/` directory of slash-command
915 > * skills.
916 > *
917 > * @category Customization Types
918 > */
919 > export interface SkillCustomization extends ChildCustomizationBase {
920 > type: CustomizationType.Skill;
921 > /**
922 > * Short description used for help text and auto-invocation matching.
923 > * Sourced from the skill's frontmatter `description`.
924 > */
925 > description?: string;
926 > /**
927 > * When `true`, only the user can invoke this skill — the agent will not
928 > * auto-invoke it. Sourced from the command skill's frontmatter
929 > * `disable-model-invocation` flag.
930 > */
931 > disableModelInvocation?: boolean;
932 > /**
933 > * When `true`, the user cannot directly invoke this skill (for example,
934 > * as a slash command); it remains available for the agent to
935 > * auto-invoke. Absent or `false` means the user may invoke it.
936 > */
937 > disableUserInvocation?: boolean;
938 > }
939 >
940 > /**
941 > * A prompt contributed by a plugin or directory.
942 > *
943 > * @category Customization Types
944 > */
945 > export interface PromptCustomization extends ChildCustomizationBase {
946 > type: CustomizationType.Prompt;
947 > /** Short description of what the prompt does. */
948 > description?: string;
949 > }
950 >
951 > /**
952 > * A rule contributed by a plugin or directory.
953 > *
954 > * Mirrors the [Open Plugins rule](https://open-plugins.com/agent-builders/components/rules)
955 > * format: a markdown file (e.g. `.mdc`) whose body is injected into
956 > * context while the rule is active. This type also covers tool-specific
957 > * "instruction" formats (e.g. VS Code Copilot's
958 > * `.github/instructions/*.md`), which differ only in naming — they
959 > * share the same semantics of `description`, optional always-on
960 > * activation, and optional glob scoping.
961 > *
962 > * @category Customization Types
963 > */
964 > export interface RuleCustomization extends ChildCustomizationBase {
965 > type: CustomizationType.Rule;
966 > /**
967 > * Description of what the rule enforces.
968 > */
969 > description?: string;
970 > /**
971 > * When `true`, the rule is always active (subject to `globs` if any).
972 > * When `false` or absent, the agent or user decides whether to apply
973 > * the rule.
974 > */
975 > alwaysApply?: boolean;
976 > /**
977 > * Glob patterns the rule applies to. When present, the rule is only
978 > * active for matching files.
979 > */
980 > globs?: string[];
981 > }
982 >
983 > /**
984 > * A hook manifest contributed by a plugin or directory.
985 > *
986 > * @category Customization Types
987 > */
988 > export interface HookCustomization extends ChildCustomizationBase {
989 > type: CustomizationType.Hook;
990 > }
991 >
992 > /**
993 > * An MCP server contributed by a plugin or directory.
994 > *
995 > * When the server is declared inline in the containing plugin manifest,
996 > * `uri` points at the manifest file and
997 > * {@link CustomizationBase.range | `range`} narrows it to the
998 > * declaration's span.
999 > *
1000 > * The MCP server customization also reflects its current status.
1001 > *
1002 > * @category Customization Types
1003 > */
1004 > export interface McpServerCustomization extends CustomizationBase {
1005 > type: CustomizationType.McpServer;
1006 > /**
1007 > * Whether this MCP server is currently enabled.
1008 > */
1009 > enabled: boolean;
1010 > /**
1011 > * Current lifecycle state of the MCP server.
1012 > */
1013 > state: McpServerState;
1014 > /**
1015 > * An `mcp://`-protocol channel the client uses to side-channel traffic
1016 > * into the upstream MCP server itself. The channel is NOT a fresh raw MCP
1017 > * connection: it piggybacks on the AHP transport
1018 > * and skips the MCP `initialize` sequence.
1019 > *
1020 > * The agent host MAY only serve a subset of MCP on this
1021 > * channel; the served subset is described by domain-specific
1022 > * capabilities such as those in
1023 > * {@link McpServerCustomizationApps.capabilities}.
1024 > *
1025 > * The channel URI SHOULD be stable across the server's lifetime, but
1026 > * the agent host MAY change it (for example across a restart) and
1027 > * MAY only expose it while the server is in
1028 > * {@link McpServerStatus.Ready | `Ready`}. Absence means no
1029 > * side-channel is currently available.
1030 > */
1031 > channel?: URI;
1032 > /**
1033 > * MCP App support. This property SHOULD be advertised for MCP servers
1034 > * which support apps.
1035 > */
1036 > mcpApp?: McpServerCustomizationApps;
1037 > }
1038 >
1039 > /**
1040 > * Information from the agent host needed to render MCP Apps served
1041 > * by this MCP server.
1042 > *
1043 > * @category MCP Server State
1044 > */
1045 > export interface McpServerCustomizationApps {
1046 > /**
1047 > * The subset of MCP App
1048 > * [`HostCapabilities`](https://github.com/modelcontextprotocol/ext-apps/blob/main/specification/draft/apps.mdx)
1049 > * the AHP host can satisfy for Views backed by this server. The
1050 > * client feeds these straight through into the `hostCapabilities` of
1051 > * the `ui/initialize` response delivered to the View.
1052 > */
1053 > capabilities: AhpMcpUiHostCapabilities;
1054 > }
1055 >
1056 > /**
1057 > * The subset of MCP App
1058 > * [`HostCapabilities`](https://github.com/modelcontextprotocol/ext-apps/blob/main/specification/draft/apps.mdx)
1059 > * an AHP host can derive from the upstream MCP server (and from AHP's own
1060 > * forwarding plumbing). Advertised on
1061 > * {@link McpServerCustomizationApps.capabilities} so clients can pass it
1062 > * through into the `hostCapabilities` of the `ui/initialize` response
1063 > * delivered to an MCP App View.
1064 > *
1065 > * Field names mirror the MCP Apps spec exactly, so the AHP-side producer
1066 > * can pass them straight through into the `hostCapabilities` of the
1067 > * `ui/initialize` response delivered to the View.
1068 > *
1069 > * Capabilities outside this set (`openLinks`, `downloadFile`, `sandbox`,
1070 > * `experimental`) are decided locally by whichever AHP client renders the
1071 > * View and are NOT part of this AHP-level advertisement — only the
1072 > * server-derived subset is.
1073 > *
1074 > * An agent host MUST only advertise a capability when it actually accepts the
1075 > * corresponding methods/notifications on the `mcp://` channel:
1076 > *
1077 > * - {@link serverTools}: host proxies `tools/list` and `tools/call` to
1078 > * the MCP server. When `listChanged` is `true`, the host also forwards
1079 > * `notifications/tools/list_changed`.
1080 > * - {@link serverResources}: host proxies `resources/read`,
1081 > * `resources/list`, and `resources/templates/list` to the MCP server.
1082 > * When `listChanged` is `true`, the host also forwards
1083 > * `notifications/resources/list_changed`.
1084 > * - {@link logging}: host accepts `notifications/message` log entries
1085 > * from the App and forwards them via `mcpNotification` (and forwards
1086 > * `logging/setLevel` calls to the server).
1087 > * - {@link sampling}: host serves `sampling/createMessage` via
1088 > * `mcpMethodCall`. When `sampling.tools` is present, the host also
1089 > * accepts SEP-1577 `tools` / `toolChoice` / `tool_use` content blocks
1090 > * inside `CreateMessageRequest`.
1091 > *
1092 > * @category MCP Server State
1093 > * @see {@link https://github.com/modelcontextprotocol/ext-apps/blob/main/specification/draft/apps.mdx | MCP Apps spec (SEP-1865)}
1094 > */
1095 > export interface AhpMcpUiHostCapabilities {
1096 > /** Producer proxies the MCP `tools/*` methods to the upstream server. */
1097 > serverTools?: {
1098 > /** Producer forwards `notifications/tools/list_changed` from the server. */
1099 > listChanged?: boolean;
1100 > };
1101 > /** Producer proxies the MCP `resources/*` methods to the upstream server. */
1102 > serverResources?: {
1103 > /** Producer forwards `notifications/resources/list_changed` from the server. */
1104 > listChanged?: boolean;
1105 > };
1106 > /** Producer accepts `notifications/message` log entries from the App via `mcpNotification`. */
1107 > logging?: Record<string, never>;
1108 > /** Producer serves `sampling/createMessage` via `mcpMethodCall`. */
1109 > sampling?: {
1110 > /**
1111 > * Producer accepts SEP-1577 `tools` / `toolChoice` / `tool_use` content
1112 > * blocks inside `CreateMessageRequest`.
1113 > */
1114 > tools?: Record<string, never>;
1115 > };
1116 > }
1117 >
1118 > /**
1119 > * Child customizations that live inside a {@link PluginCustomization} or
1120 > * {@link DirectoryCustomization}.
1121 > *
1122 > * @category Customization Types
1123 > */
1124 > export type ChildCustomization =
1125 > | AgentCustomization
1126 > | SkillCustomization
1127 > | PromptCustomization
1128 > | RuleCustomization
1129 > | HookCustomization
1130 > | McpServerCustomization;
1131 >
1132 > /**
1133 > * A top-level customization active in a session. Either a container
1134 > * ({@link PluginCustomization} or {@link DirectoryCustomization}) whose
1135 > * leaf customizations live in its
1136 > * {@link ContainerCustomizationBase.children | `children`} array, or a
1137 > * bare {@link McpServerCustomization} surfaced directly by the host.
1138 > *
1139 > * @category Customization Types
1140 > */
1141 > export type Customization =
1142 > | PluginCustomization
1143 > | DirectoryCustomization
1144 > | McpServerCustomization;
1145 >
1146 >
1147 > // ─── MCP Server State ────────────────────────────────────────────────────────
1148 >
1149 > /**
1150 > * Discriminant for the {@link McpServerState} union.
1151 > *
1152 > * @category MCP Server State
1153 > */
1154 > export const enum McpServerStatus {
1155 > /** Server has been registered but is not yet running. */
1156 > Starting = 'starting',
1157 > /** Server is running and serving requests. */
1158 > Ready = 'ready',
1159 > /**
1160 > * Server is reachable but requires additional authentication before it
1161 > * can start, or before it can serve a particular request. Carries the
1162 > * RFC 9728 Protected Resource Metadata the client needs to obtain a
1163 > * token; the client then pushes the token via the existing
1164 > * `authenticate` command.
1165 > */
1166 > AuthRequired = 'authRequired',
1167 > /** Server failed to start, crashed, or otherwise transitioned to a fatal error. */
1168 > Error = 'error',
1169 > /** Server has been shut down. */
1170 > Stopped = 'stopped',
1171 > }
1172 >
1173 > /**
1174 > * Why an MCP server is currently in the {@link McpServerStatus.AuthRequired}
1175 > * state. Mirrors the three failure modes defined by the
1176 > * [MCP authorization spec](https://modelcontextprotocol.io/specification/2025-11-25/basic/authorization.md).
1177 > *
1178 > * @category MCP Server State
1179 > */
1180 > export const enum McpAuthRequiredReason {
1181 > /** No token has been provided yet (HTTP 401, no prior token). */
1182 > Required = 'required',
1183 > /** A previously valid token expired or was revoked (HTTP 401). */
1184 > Expired = 'expired',
1185 > /**
1186 > * Step-up auth: a token is present but its scopes are insufficient for
1187 > * the requested operation (HTTP 403 with
1188 > * `WWW-Authenticate: Bearer error="insufficient_scope"`).
1189 > *
1190 > * Unlike {@link Required} and {@link Expired} — which typically surface
1191 > * before any tool work is in flight — `InsufficientScope` is almost
1192 > * always triggered by an MCP request issued mid-turn (a `tools/call`,
1193 > * `resources/read`, etc.). The host SHOULD pair the
1194 > * {@link McpServerAuthRequiredState} transition with
1195 > * {@link SessionStatus.InputNeeded} on
1196 > * {@link SessionSummary.status | the session} so the activity becomes
1197 > * visible at the session-summary level, and clients SHOULD watch for
1198 > * this kind on any
1199 > * {@link McpServerCustomization | MCP server} backing a running tool
1200 > * call so they can present an explicit "grant more access" affordance
1201 > * tied to the blocked tool call.
1202 > */
1203 > InsufficientScope = 'insufficientScope',
1204 > }
1205 >
1206 > /**
1207 > * Server is registered with the host but has not yet started.
1208 > *
1209 > * @category MCP Server State
1210 > */
1211 > export interface McpServerStartingState {
1212 > kind: McpServerStatus.Starting;
1213 > }
1214 >
1215 > /**
1216 > * Server is running and serving requests.
1217 > *
1218 > * @category MCP Server State
1219 > */
1220 > export interface McpServerReadyState {
1221 > kind: McpServerStatus.Ready;
1222 > }
1223 >
1224 > /**
1225 > * A pre-registered OAuth client that clients use instead of dynamic client
1226 > * registration when resolving an MCP authentication challenge.
1227 > *
1228 > * @category MCP Server State
1229 > */
1230 > export interface McpOAuthClient {
1231 > /** OAuth client identifier registered with the authorization server. */
1232 > clientId: string;
1233 > /**
1234 > * OAuth client secret for a confidential client. Absence means the client is
1235 > * public and uses a secretless flow such as authorization code with PKCE.
1236 > */
1237 > clientSecret?: string;
1238 > }
1239 >
1240 > /**
1241 > * Reusable MCP authentication challenge — the RFC 9728 discovery info a
1242 > * client needs to obtain a token and push it via the `authenticate` command.
1243 > * Deliberately carries **no token**: this describes what is being asked for,
1244 > * never the ****** itself.
1245 > *
1246 > * Shared by two independent state machines that describe the same OAuth
1247 > * challenge from different vantage points:
1248 > *
1249 > * - {@link McpServerAuthRequiredState} — the MCP server itself cannot serve
1250 > * *any* request until the client authenticates.
1251 > * - {@link ToolCallAuthRequiredState} — a specific in-flight tool call is
1252 > * paused pending authentication (typically
1253 > * {@link McpAuthRequiredReason.InsufficientScope} step-up auth
1254 > * mid-execution). The server state and the tool-call state remain
1255 > * separate on purpose: the server saying "I need auth" and a tool
1256 > * invocation saying "I am waiting on that auth" are different facts that
1257 > * can be true independently.
1258 > *
1259 > * @category MCP Server State
1260 > */
1261 > export interface McpAuthRequirement {
1262 > /** Why authentication is required. */
1263 > reason: McpAuthRequiredReason;
1264 > /**
1265 > * Pre-registered OAuth client to use for authorization. When present, clients
1266 > * MUST use these credentials instead of dynamic client registration.
1267 > */
1268 > oauthClient?: McpOAuthClient;
1269 > /**
1270 > * RFC 9728 Protected Resource Metadata. The `resource` field is the
1271 > * canonical MCP server URI per RFC 8707, used as the OAuth `resource`
1272 > * indicator. `authorization_servers` is REQUIRED by the MCP
1273 > * authorization spec.
1274 > */
1275 > resource: ProtectedResourceMetadata;
1276 > /**
1277 > * Scopes required for the current challenge, parsed from the
1278 > * `WWW-Authenticate: ******"…"` header (or `scopes_supported`
1279 > * fallback). Authoritative for the next authorization request — clients
1280 > * MUST NOT assume any subset/superset relationship to
1281 > * `resource.scopes_supported`.
1282 > */
1283 > requiredScopes?: string[];
1284 > /** Human-readable hint, typically from the OAuth `error_description`. */
1285 > description?: string;
1286 > }
1287 >
1288 > /**
1289 > * Server is reachable but cannot serve requests until the client
1290 > * authenticates. Mirrors the discovery flow defined by
1291 > * [RFC 9728](https://datatracker.ietf.org/doc/html/rfc9728)
1292 > * (Protected Resource Metadata) and the OAuth 2.1 / RFC 6750 challenge
1293 > * semantics required by the MCP authorization spec.
1294 > *
1295 > * Clients react to this state by calling the existing `authenticate`
1296 > * command with the {@link ProtectedResourceMetadata.resource | resource}
1297 > * carried here. There is **no** `notify/authRequired` notification for
1298 > * MCP servers — the action stream is the single source of truth.
1299 > *
1300 > * When the transition is triggered by a request issued during a turn
1301 > * — most commonly
1302 > * {@link McpAuthRequiredReason.InsufficientScope | `InsufficientScope`}
1303 > * surfacing mid-tool-call — the host SHOULD also raise
1304 > * {@link SessionStatus.InputNeeded} on the session so the block is
1305 > * visible at the summary level. Clients SHOULD watch this status on
1306 > * any MCP server backing a running tool call and surface an explicit
1307 > * affordance (e.g. a "grant additional access" prompt) tied to that
1308 > * tool call, rather than relying on the user to notice the
1309 > * customization’s status badge.
1310 > *
1311 > * @category MCP Server State
1312 > */
1313 > export interface McpServerAuthRequiredState extends McpAuthRequirement {
1314 > kind: McpServerStatus.AuthRequired;
1315 > }
1316 >
1317 > /**
1318 > * Server failed to start, crashed, or otherwise transitioned to a
1319 > * non-recoverable error. Use {@link McpServerStatus.AuthRequired}
1320 > * for authentication failures.
1321 > *
1322 > * @category MCP Server State
1323 > */
1324 > export interface McpServerErrorState {
1325 > kind: McpServerStatus.Error;
1326 > /** Error details. */
1327 > error: ErrorInfo;
1328 > }
1329 >
1330 > /**
1331 > * Server has been shut down. The host MAY remove the server from the
1332 > * session entirely shortly after this state.
1333 > *
1334 > * @category MCP Server State
1335 > */
1336 > export interface McpServerStoppedState {
1337 > kind: McpServerStatus.Stopped;
1338 > }
1339 >
1340 > /**
1341 > * Discriminated union of all MCP server lifecycle states.
1342 > * Discriminated by `kind` (a {@link McpServerStatus} value).
1343 > *
1344 > * @category MCP Server State
1345 > */
1346 > export type McpServerState =
1347 > | McpServerStartingState
1348 > | McpServerReadyState
1349 > | McpServerAuthRequiredState
1350 > | McpServerErrorState
1351 > | McpServerStoppedState;
src/vs/base/common/async.ts 1292 covered LOC · 221 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- async.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken, CancellationTokenSource } from './cancellation.js';
7 > import { BugIndicatingError, CancellationError, isCancellationError } from './errors.js';
8 > import { Emitter, Event } from './event.js';
9 > import { Disposable, DisposableMap, DisposableStore, IDisposable, isDisposable, MutableDisposable, toDisposable } from './lifecycle.js';
10 > import { extUri as defaultExtUri, IExtUri } from './resources.js';
11 > import { URI } from './uri.js';
12 > import { setTimeout0 } from './platform.js';
13 > import { MicrotaskDelay } from './symbols.js';
14 > import { Lazy } from './lazy.js';
15 >
16 > export function isThenable<T>(obj: unknown): obj is Promise<T> {
17 return !!obj && typeof (obj as unknown as Promise<T>).then === 'function';
18 }
19 > async.ts
20 > export interface CancelablePromise<T> extends Promise<T> {
21 > cancel(): void;
22 > }
23 >
24 > /**
25 > * Returns a promise that can be cancelled using the provided cancellation token.
26 > *
27 > * @remarks When cancellation is requested, the promise will be rejected with a {@link CancellationError}.
28 > * If the promise resolves to a disposable object, it will be automatically disposed when cancellation
29 > * is requested.
30 > *
31 > * @param callback A function that accepts a cancellation token and returns a promise
32 > * @returns A promise that can be cancelled
33 > */
34 > export function createCancelablePromise<T>(callback: (token: CancellationToken) => Promise<T>): CancelablePromise<T> {
35 const source = new CancellationTokenSource();
36
80 };
81 }
82 > async.ts
83 > /**
84 > * Returns a promise that resolves with `undefined` as soon as the passed token is cancelled.
85 > * @see {@link raceCancellationError}
86 > */
87 > export function raceCancellation<T>(promise: Promise<T>, token: CancellationToken): Promise<T | undefined>;
88 >
89 > /**
90 > * Returns a promise that resolves with `defaultValue` as soon as the passed token is cancelled.
91 > * @see {@link raceCancellationError}
92 > */
93 > export function raceCancellation<T>(promise: Promise<T>, token: CancellationToken, defaultValue: T): Promise<T>;
94 >
95 > export function raceCancellation<T>(promise: Promise<T>, token: CancellationToken, defaultValue?: T): Promise<T | undefined> {
96 return new Promise((resolve, reject) => {
97 const ref = token.onCancellationRequested(() => {
102 });
103 }
104 > async.ts
105 > /**
106 > * Returns a promise that rejects with an {@CancellationError} as soon as the passed token is cancelled.
107 > * @see {@link raceCancellation}
108 > */
109 > export function raceCancellationError<T>(promise: Promise<T>, token: CancellationToken): Promise<T> {
110 > return new Promise((resolve, reject) => { async.ts
111 > const ref = token.onCancellationRequested(() => {
112 ref.dispose();
113 reject(new CancellationError());
114 > }); async.ts
115 > promise.then(resolve, reject).finally(() => ref.dispose());
116 > });
117 > }
118 > async.ts
119 > export function rejectIfNotCanceled(err: unknown): undefined {
120 if (isCancellationError(err)) {
121 return undefined;
123 return Promise.reject(err) as never;
124 }
125 > async.ts
126 > /**
127 > * Wraps a cancellable promise such that it is no cancellable. Can be used to
128 > * avoid issues with shared promises that would normally be returned as
129 > * cancellable to consumers.
130 > */
131 > export function notCancellablePromise<T>(promise: CancelablePromise<T>): Promise<T> {
132 return new Promise<T>((resolve, reject) => {
133 promise.then(resolve, reject);
134 });
135 }
136 > async.ts
137 > /**
138 > * Returns as soon as one of the promises resolves or rejects and cancels remaining promises
139 > */
140 > export function raceCancellablePromises<T>(cancellablePromises: (CancelablePromise<T> | Promise<T>)[]): CancelablePromise<T> {
141 let resolvedPromiseIndex = -1;
142 const promises = cancellablePromises.map((promise, index) => promise.then(result => { resolvedPromiseIndex = index; return result; }));
154 return promise;
155 }
156 > async.ts
157 > export function raceTimeout<T>(promise: Promise<T>, timeout: number, onTimeout?: () => void): Promise<T | undefined> {
158 let promiseResolve: ((value: T | undefined) => void) | undefined = undefined;
159
168 ]);
169 }
170 > async.ts
171 > export function asPromise<T>(callback: () => T | Thenable<T>): Promise<T> {
172 return new Promise<T>((resolve, reject) => {
173 const item = callback();
179 });
180 }
181 > async.ts
182 > /**
183 > * Creates and returns a new promise, plus its `resolve` and `reject` callbacks.
184 > *
185 > * Replace with standardized [`Promise.withResolvers`](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Promise/withResolvers) once it is supported
186 > */
187 > export function promiseWithResolvers<T>(): { promise: Promise<T>; resolve: (value: T | PromiseLike<T>) => void; reject: (err?: any) => void } {
188 let resolve: (value: T | PromiseLike<T>) => void;
189 let reject: (reason?: any) => void;
194 return { promise, resolve: resolve!, reject: reject! };
195 }
196 > async.ts
197 > export interface ITask<T> {
198 > (): T;
199 > }
200 >
201 > export interface ICancellableTask<T> {
202 > (token: CancellationToken): T;
203 > }
204 >
205 > /**
206 > * A helper to prevent accumulation of sequential async tasks.
207 > *
208 > * Imagine a mail man with the sole task of delivering letters. As soon as
209 > * a letter submitted for delivery, he drives to the destination, delivers it
210 > * and returns to his base. Imagine that during the trip, N more letters were submitted.
211 > * When the mail man returns, he picks those N letters and delivers them all in a
212 > * single trip. Even though N+1 submissions occurred, only 2 deliveries were made.
213 > *
214 > * The throttler implements this via the queue() method, by providing it a task
215 > * factory. Following the example:
216 > *
217 > * const throttler = new Throttler();
218 > * const letters = [];
219 > *
220 > * function deliver() {
221 > * const lettersToDeliver = letters;
222 > * letters = [];
223 > * return makeTheTrip(lettersToDeliver);
224 > * }
225 > *
226 > * function onLetterReceived(l) {
227 > * letters.push(l);
228 > * throttler.queue(deliver);
229 > * }
230 > */
231 > export class Throttler implements IDisposable {
232 >
233 > private activePromise: Promise<any> | null;
234 > private queuedPromise: Promise<any> | null;
235 > private queuedPromiseFactory: ICancellableTask<Promise<any>> | null;
236 > private cancellationTokenSource: CancellationTokenSource;
237 >
238 > constructor() {
239 > this.activePromise = null; async.ts
240 > this.queuedPromise = null;
241 > this.queuedPromiseFactory = null;
242 >
243 > this.cancellationTokenSource = new CancellationTokenSource();
244 > }
245 > async.ts
246 > queue<T>(promiseFactory: ICancellableTask<Promise<T>>): Promise<T> {
247 if (this.cancellationTokenSource.token.isCancellationRequested) {
248 return Promise.reject(new Error('Throttler is disposed'));
288 });
289 }
290 > async.ts
291 > dispose(): void {
292 > this.cancellationTokenSource.cancel(); async.ts
293 > }
294 > } async.ts
295 >
296 > export class Sequencer {
297 > async.ts
298 > private current: Promise<unknown> = Promise.resolve(null);
299 > async.ts
300 > queue<T>(promiseTask: ITask<Promise<T>>): Promise<T> {
301 > return this.current = this.current.then(() => promiseTask(), () => promiseTask()); async.ts
302 > }
303 > } async.ts
304 >
305 > /**
306 > * A {@link Throttler} per key. Calls for the same key coalesce (only the most
307 > * recently queued task runs after the active one settles); calls for different
308 > * keys are independent. Idle keys are cleaned up automatically.
309 > */
310 > export class ThrottlerByKey<TKey> implements IDisposable {
311
312 private readonly throttlers = new Map<TKey, { throttler: Throttler; count: number }>();
313 > async.ts
314 > queue<T>(key: TKey, task: ITask<Promise<T>>): Promise<T> {
315 let entry = this.throttlers.get(key);
316 if (!entry) {
327 });
328 }
329 > async.ts
330 > dispose(): void {
331 for (const { throttler } of this.throttlers.values()) {
332 throttler.dispose();
334 this.throttlers.clear();
335 }
336 > } async.ts
337 >
338 > export class SequencerByKey<TKey> {
339
340 private promiseMap = new Map<TKey, Promise<unknown>>();
341 > async.ts
342 > queue<T>(key: TKey, promiseTask: ITask<Promise<T>>): Promise<T> {
343 const runningPromise = this.promiseMap.get(key) ?? Promise.resolve();
344 const newPromise = runningPromise
353 return newPromise;
354 }
355 > async.ts
356 > peek(key: TKey): Promise<unknown> | undefined {
357 return this.promiseMap.get(key) || undefined;
358 }
359 > async.ts
360 > keys(): IterableIterator<TKey> {
361 return this.promiseMap.keys();
362 }
363 > } async.ts
364 >
365 > interface IScheduledLater extends IDisposable {
366 > isTriggered(): boolean;
367 > }
368 >
369 > const timeoutDeferred = (timeout: number, fn: () => void): IScheduledLater => {
370 let scheduled = true;
371 const handle = setTimeout(() => {
381 };
382 };
383 > async.ts
384 > const microtaskDeferred = (fn: () => void): IScheduledLater => {
385 let scheduled = true;
386 queueMicrotask(() => {
396 };
397 };
398 > async.ts
399 > /**
400 > * A helper to delay (debounce) execution of a task that is being requested often.
401 > *
402 > * Following the throttler, now imagine the mail man wants to optimize the number of
403 > * trips proactively. The trip itself can be long, so he decides not to make the trip
404 > * as soon as a letter is submitted. Instead he waits a while, in case more
405 > * letters are submitted. After said waiting period, if no letters were submitted, he
406 > * decides to make the trip. Imagine that N more letters were submitted after the first
407 > * one, all within a short period of time between each other. Even though N+1
408 > * submissions occurred, only 1 delivery was made.
409 > *
410 > * The delayer offers this behavior via the trigger() method, into which both the task
411 > * to be executed and the waiting period (delay) must be passed in as arguments. Following
412 > * the example:
413 > *
414 > * const delayer = new Delayer(WAITING_PERIOD);
415 > * const letters = [];
416 > *
417 > * function letterReceived(l) {
418 > * letters.push(l);
419 > * delayer.trigger(() => { return makeTheTrip(); });
420 > * }
421 > */
422 > export class Delayer<T> implements IDisposable {
423 >
424 > private deferred: IScheduledLater | null;
425 > private completionPromise: Promise<any> | null;
426 > private doResolve: ((value?: any | Promise<any>) => void) | null;
427 > private doReject: ((err: unknown) => void) | null;
428 > private task: ITask<T | Promise<T>> | null;
429 >
430 > constructor(public defaultDelay: number | typeof MicrotaskDelay) {
431 > this.deferred = null; async.ts
432 > this.completionPromise = null;
433 > this.doResolve = null;
434 > this.doReject = null;
435 > this.task = null;
436 > }
437 > async.ts
438 > trigger(task: ITask<T | Promise<T>>, delay = this.defaultDelay): Promise<T> {
439 this.task = task;
440 this.cancelTimeout();
465 return this.completionPromise;
466 }
467 > async.ts
468 > isTriggered(): boolean {
469 return !!this.deferred?.isTriggered();
470 }
471 > async.ts
472 > cancel(): void {
473 > this.cancelTimeout(); async.ts
474 >
475 > if (this.completionPromise) {
476 this.doReject?.(new CancellationError());
477 this.completionPromise = null;
478 }
479 > } async.ts
480 > async.ts
481 > private cancelTimeout(): void {
482 > this.deferred?.dispose(); async.ts
483 > this.deferred = null;
484 > }
485 > async.ts
486 > dispose(): void {
487 > this.cancel(); async.ts
488 > }
489 > } async.ts
490 >
491 > /**
492 > * A helper to delay execution of a task that is being requested often, while
493 > * preventing accumulation of consecutive executions, while the task runs.
494 > *
495 > * The mail man is clever and waits for a certain amount of time, before going
496 > * out to deliver letters. While the mail man is going out, more letters arrive
497 > * and can only be delivered once he is back. Once he is back the mail man will
498 > * do one more trip to deliver the letters that have accumulated while he was out.
499 > */
500 > export class ThrottledDelayer<T> {
501 >
502 > private delayer: Delayer<Promise<T>>;
503 > private throttler: Throttler;
504 >
505 > constructor(defaultDelay: number) {
506 > this.delayer = new Delayer(defaultDelay); async.ts
507 > this.throttler = new Throttler();
508 > }
509 > async.ts
510 > trigger(promiseFactory: ICancellableTask<Promise<T>>, delay?: number): Promise<T> {
511 return this.delayer.trigger(() => this.throttler.queue(promiseFactory), delay) as unknown as Promise<T>;
512 }
513 > async.ts
514 > isTriggered(): boolean {
515 return this.delayer.isTriggered();
516 }
517 > async.ts
518 > cancel(): void {
519 this.delayer.cancel();
520 }
521 > async.ts
522 > dispose(): void {
523 > this.delayer.dispose(); async.ts
524 > this.throttler.dispose();
525 > }
526 > } async.ts
527 >
528 > /**
529 > * A barrier that is initially closed and then becomes opened permanently.
530 > */
531 > export class Barrier {
532 > private _isOpen: boolean;
533 > private _promise: Promise<boolean>;
534 > private _completePromise!: (v: boolean) => void;
535 >
536 > constructor() {
537 this._isOpen = false;
538 this._promise = new Promise<boolean>((c, e) => {
540 });
541 }
542 > async.ts
543 > isOpen(): boolean {
544 return this._isOpen;
545 }
546 > async.ts
547 > open(): void {
548 this._isOpen = true;
549 this._completePromise(true);
550 }
551 > async.ts
552 > wait(): Promise<boolean> {
553 return this._promise;
554 }
555 > } async.ts
556 >
557 > /**
558 > * A barrier that is initially closed and then becomes opened permanently after a certain period of
559 > * time or when open is called explicitly
560 > */
561 > export class AutoOpenBarrier extends Barrier {
562 >
563 > private readonly _timeout: Timeout;
564 >
565 > constructor(autoOpenTimeMs: number) {
566 super();
567 this._timeout = setTimeout(() => this.open(), autoOpenTimeMs);
568 }
569 > async.ts
570 > override open(): void {
571 clearTimeout(this._timeout);
572 super.open();
573 }
574 > } async.ts
575 >
576 > export function timeout(millis: number): CancelablePromise<void>;
577 > export function timeout(millis: number, token: CancellationToken): Promise<void>;
578 > export function timeout(millis: number, token?: CancellationToken): CancelablePromise<void> | Promise<void> {
579 if (!token) {
580 return createCancelablePromise(token => timeout(millis, token));
593 });
594 }
595 > async.ts
596 > /**
597 > * Creates a timeout that can be disposed using its returned value.
598 > * @param handler The timeout handler.
599 > * @param timeout An optional timeout in milliseconds.
600 > * @param store An optional {@link DisposableStore} that will have the timeout disposable managed automatically.
601 > *
602 > * @example
603 > * const store = new DisposableStore;
604 > * // Call the timeout after 1000ms at which point it will be automatically
605 > * // evicted from the store.
606 > * const timeoutDisposable = disposableTimeout(() => {}, 1000, store);
607 > *
608 > * if (foo) {
609 > * // Cancel the timeout and evict it from store.
610 > * timeoutDisposable.dispose();
611 > * }
612 > */
613 > export function disposableTimeout(handler: () => void, timeout = 0, store?: DisposableStore): IDisposable {
614 const timer = setTimeout(() => {
615 handler();
625 return disposable;
626 }
627 > async.ts
628 > /**
629 > * The largest delay (in milliseconds) a single `setTimeout` can represent.
630 > * Larger values overflow its internal 32-bit signed integer and fire (almost)
631 > * immediately instead of waiting.
632 > */
633 > export const MAX_TIMEOUT_DELAY = 2 ** 31 - 1; // ~24.8 days
634 >
635 > /**
636 > * Like {@link disposableTimeout}, but supports delays larger than
637 > * {@link MAX_TIMEOUT_DELAY} (~24.8 days), which a single `setTimeout` cannot
638 > * represent. The wait is split into chunks and re-armed until the target time is
639 > * reached, so the handler fires at approximately `Date.now() + timeout`.
640 > *
641 > * Note: like `setTimeout`, firing is best-effort and may drift across system
642 > * sleep or wall-clock changes; do not rely on it for precise scheduling.
643 > *
644 > * @param handler The timeout handler.
645 > * @param timeout The timeout in milliseconds. May exceed {@link MAX_TIMEOUT_DELAY}.
646 > * @param store An optional {@link DisposableStore} that will have the timeout disposable managed automatically.
647 > */
648 > export function disposableLongTimeout(handler: () => void, timeout: number, store?: DisposableStore): IDisposable {
649 const target = Date.now() + timeout;
650 let timer: Timeout;
671 return disposable;
672 }
673 > async.ts
674 > /**
675 > * Runs the provided list of promise factories in sequential order. The returned
676 > * promise will complete to an array of results from each promise.
677 > */
678 >
679 > export function sequence<T>(promiseFactories: ITask<Promise<T>>[]): Promise<T[]> {
680 const results: T[] = [];
681 let index = 0;
701 return Promise.resolve(null).then(thenHandler);
702 }
703 > async.ts
704 > export function first<T>(promiseFactories: ITask<Promise<T>>[], shouldStop: (t: T) => boolean = t => !!t, defaultValue: T | null = null): Promise<T | null> {
705 let index = 0;
706 const len = promiseFactories.length;
725 return loop();
726 }
727 > async.ts
728 > /**
729 > * Returns the result of the first promise that matches the "shouldStop",
730 > * running all promises in parallel. Supports cancelable promises.
731 > */
732 > export function firstParallel<T>(promiseList: Promise<T>[], shouldStop?: (t: T) => boolean, defaultValue?: T | null): Promise<T | null>;
733 > export function firstParallel<T, R extends T>(promiseList: Promise<T>[], shouldStop: (t: T) => t is R, defaultValue?: R | null): Promise<R | null>;
734 > export function firstParallel<T>(promiseList: Promise<T>[], shouldStop: (t: T) => boolean = t => !!t, defaultValue: T | null = null) {
735 if (promiseList.length === 0) {
736 return Promise.resolve(defaultValue);
764 });
765 }
766 > async.ts
767 > interface ILimitedTaskFactory<T> {
768 > factory: ITask<Promise<T>>;
769 > c: (value: T | Promise<T>) => void;
770 > e: (error?: unknown) => void;
771 > }
772 >
773 > export interface ILimiter<T> {
774 >
775 > readonly size: number;
776 >
777 > queue(factory: ITask<Promise<T>>): Promise<T>;
778 >
779 > clear(): void;
780 > }
781 >
782 > /**
783 > * A helper to queue N promises and run them all with a max degree of parallelism. The helper
784 > * ensures that at any time no more than M promises are running at the same time.
785 > */
786 > export class Limiter<T> implements ILimiter<T> {
787 >
788 > private _size = 0;
789 > private _isDisposed = false;
790 > private runningPromises: number;
791 > private readonly maxDegreeOfParalellism: number;
792 > private readonly outstandingPromises: ILimitedTaskFactory<T>[];
793 > private readonly _onDrained: Emitter<void>;
794 >
795 > constructor(maxDegreeOfParalellism: number) {
796 this.maxDegreeOfParalellism = maxDegreeOfParalellism;
797 this.outstandingPromises = [];
799 this._onDrained = new Emitter<void>();
800 }
801 > async.ts
802 > /**
803 > *
804 > * @returns A promise that resolved when all work is done (onDrained) or when
805 > * there is nothing to do
806 > */
807 > whenIdle(): Promise<void> {
808 return this.size > 0
809 ? Event.toPromise(this.onDrained)
810 : Promise.resolve();
811 }
812 > async.ts
813 > get onDrained(): Event<void> {
814 return this._onDrained.event;
815 }
816 > async.ts
817 > get size(): number {
818 return this._size;
819 }
820 > async.ts
821 > queue(factory: ITask<Promise<T>>): Promise<T> {
822 if (this._isDisposed) {
823 throw new Error('Object has been disposed');
830 });
831 }
832 > async.ts
833 > private consume(): void {
834 while (this.outstandingPromises.length && this.runningPromises < this.maxDegreeOfParalellism) {
835 const iLimitedTask = this.outstandingPromises.shift()!;
841 }
842 }
843 > async.ts
844 > private consumed(): void {
845 if (this._isDisposed) {
846 return;
855 }
856 }
857 > async.ts
858 > clear(): void {
859 if (this._isDisposed) {
860 throw new Error('Object has been disposed');
863 this._size = this.runningPromises;
864 }
865 > async.ts
866 > dispose(): void {
867 this._isDisposed = true;
868 this.outstandingPromises.length = 0; // stop further processing
870 this._onDrained.dispose();
871 }
872 > } async.ts
873 >
874 > /**
875 > * A queue is handles one promise at a time and guarantees that at any time only one promise is executing.
876 > */
877 > export class Queue<T> extends Limiter<T> {
878 >
879 > constructor() {
880 super(1);
881 }
882 > } async.ts
883 >
884 > /**
885 > * Same as `Queue`, ensures that only 1 task is executed at the same time. The difference to `Queue` is that
886 > * there is only 1 task about to be scheduled next. As such, calling `queue` while a task is executing will
887 > * replace the currently queued task until it executes.
888 > *
889 > * As such, the returned promise may not be from the factory that is passed in but from the next factory that
890 > * is running after having called `queue`.
891 > */
892 > export class LimitedQueue {
893
894 private readonly sequentializer = new TaskSequentializer();
895
896 private tasks = 0;
897 > async.ts
898 > queue(factory: ITask<Promise<void>>): Promise<void> {
899 if (!this.sequentializer.isRunning()) {
900 return this.sequentializer.run(this.tasks++, factory());
905 });
906 }
907 > } async.ts
908 >
909 > /**
910 > * A helper to organize queues per resource. The ResourceQueue makes sure to manage queues per resource
911 > * by disposing them once the queue is empty.
912 > */
913 > export class ResourceQueue implements IDisposable {
914
915 private readonly queues = new Map<string, Queue<void>>();
919 private drainListeners: DisposableMap<number> | undefined = undefined;
920 private drainListenerCount = 0;
921 > async.ts
922 > async whenDrained(): Promise<void> {
923 if (this.isDrained()) {
924 return;
930 return promise.p;
931 }
932 > async.ts
933 > private isDrained(): boolean {
934 for (const [, queue] of this.queues) {
935 if (queue.size > 0) {
940 return true;
941 }
942 > async.ts
943 > queueSize(resource: URI, extUri: IExtUri = defaultExtUri): number {
944 const key = extUri.getComparisonKey(resource);
945
946 return this.queues.get(key)?.size ?? 0;
947 }
948 > async.ts
949 > queueFor(resource: URI, factory: ITask<Promise<void>>, extUri: IExtUri = defaultExtUri): Promise<void> {
950 const key = extUri.getComparisonKey(resource);
951
977 return queue.queue(factory);
978 }
979 > async.ts
980 > private onDidQueueDrain(): void {
981 if (!this.isDrained()) {
982 return; // not done yet
985 this.releaseDrainers();
986 }
987 > async.ts
988 > private releaseDrainers(): void {
989 for (const drainer of this.drainers) {
990 drainer.complete();
993 this.drainers.clear();
994 }
995 > async.ts
996 > dispose(): void {
997 for (const [, queue] of this.queues) {
998 queue.dispose();
1011 this.drainListeners?.dispose();
1012 }
1013 > } async.ts
1014 >
1015 > export type Task<T = void> = () => (Promise<T> | T);
1016 >
1017 > /**
1018 > * Wrap a type in an optional promise. This can be useful to avoid the runtime
1019 > * overhead of creating a promise.
1020 > */
1021 > export type MaybePromise<T> = Promise<T> | T;
1022 >
1023 > /**
1024 > * Processes tasks in the order they were scheduled.
1025 > */
1026 > export class TaskQueue {
1027 private _runningTask: Task<any> | undefined = undefined;
1028 private _pendingTasks: { task: Task<any>; deferred: DeferredPromise<any>; setUndefinedWhenCleared: boolean }[] = [];
1029 > async.ts
1030 > /**
1031 > * Waits for the current and pending tasks to finish, then runs and awaits the given task.
1032 > * If the task is skipped because of clearPending, the promise is rejected with a CancellationError.
1033 > */
1034 > public schedule<T>(task: Task<T>): Promise<T> {
1035 const deferred = new DeferredPromise<T>();
1036 this._pendingTasks.push({ task, deferred, setUndefinedWhenCleared: false });
1038 return deferred.p;
1039 }
1040 > async.ts
1041 > /**
1042 > * Waits for the current and pending tasks to finish, then runs and awaits the given task.
1043 > * If the task is skipped because of clearPending, the promise is resolved with undefined.
1044 > */
1045 > public scheduleSkipIfCleared<T>(task: Task<T>): Promise<T | undefined> {
1046 const deferred = new DeferredPromise<T>();
1047 this._pendingTasks.push({ task, deferred, setUndefinedWhenCleared: true });
1049 return deferred.p;
1050 }
1051 > async.ts
1052 > private _runIfNotRunning(): void {
1053 if (this._runningTask === undefined) {
1054 this._processQueue();
1055 }
1056 }
1057 > async.ts
1058 > private async _processQueue(): Promise<void> {
1059 if (this._pendingTasks.length === 0) {
1060 return;
1082 }
1083 }
1084 > async.ts
1085 > /**
1086 > * Clears all pending tasks. Does not cancel the currently running task.
1087 > */
1088 > public clearPending(): void {
1089 const tasks = this._pendingTasks;
1090 this._pendingTasks = [];
1097 }
1098 }
1099 > } async.ts
1100 >
1101 > export class TimeoutTimer implements IDisposable {
1102 > private _token: Timeout | undefined;
1103 > private _isDisposed = false;
1104 >
1105 > constructor();
1106 > constructor(runner: () => void, timeout: number);
1107 > constructor(runner?: () => void, timeout?: number) {
1108 this._token = undefined;
1109
1112 }
1113 }
1114 > async.ts
1115 > dispose(): void {
1116 this.cancel();
1117 this._isDisposed = true;
1118 }
1119 > async.ts
1120 > cancel(): void {
1121 if (this._token !== undefined) {
1122 clearTimeout(this._token);
1124 }
1125 }
1126 > async.ts
1127 > cancelAndSet(runner: () => void, timeout: number): void {
1128 if (this._isDisposed) {
1129 throw new BugIndicatingError(`Calling 'cancelAndSet' on a disposed TimeoutTimer`);
1136 }, timeout);
1137 }
1138 > async.ts
1139 > setIfNotSet(runner: () => void, timeout: number): void {
1140 if (this._isDisposed) {
1141 throw new BugIndicatingError(`Calling 'setIfNotSet' on a disposed TimeoutTimer`);
1151 }, timeout);
1152 }
1153 > } async.ts
1154 >
1155 > export class IntervalTimer implements IDisposable {
1156
1157 private disposable: IDisposable | undefined = undefined;
1158 private isDisposed = false;
1159 > async.ts
1160 > cancel(): void {
1161 this.disposable?.dispose();
1162 this.disposable = undefined;
1163 }
1164 > async.ts
1165 > cancelAndSet(runner: () => void, interval: number, context = globalThis): void {
1166 if (this.isDisposed) {
1167 throw new BugIndicatingError(`Calling 'cancelAndSet' on a disposed IntervalTimer`);
1178 });
1179 }
1180 > async.ts
1181 > dispose(): void {
1182 this.cancel();
1183 this.isDisposed = true;
1184 }
1185 > } async.ts
1186 >
1187 > export class RunOnceScheduler<Runner extends (...args: any[]) => any = () => any> implements IDisposable {
1188 >
1189 > protected runner: Runner | null;
1190 >
1191 > private timeoutToken: Timeout | undefined;
1192 > private timeout: number;
1193 > private timeoutHandler: () => void;
1194 >
1195 > constructor(runner: Runner, delay: number) {
1196 > this.timeoutToken = undefined; async.ts
1197 > this.runner = runner;
1198 > this.timeout = delay;
1199 > this.timeoutHandler = this.onTimeout.bind(this);
1200 > }
1201 > async.ts
1202 > /**
1203 > * Dispose RunOnceScheduler
1204 > */
1205 > dispose(): void {
1206 > this.cancel(); async.ts
1207 > this.runner = null;
1208 > }
1209 > async.ts
1210 > /**
1211 > * Cancel current scheduled runner (if any).
1212 > */
1213 > cancel(): void {
1214 > if (this.isScheduled()) { async.ts
1215 clearTimeout(this.timeoutToken);
1216 this.timeoutToken = undefined;
1217 }
1218 > } async.ts
1219 > async.ts
1220 > /**
1221 > * Cancel previous runner (if any) & schedule a new runner.
1222 > */
1223 > schedule(delay = this.timeout): void {
1224 > this.cancel(); async.ts
1225 > this.timeoutToken = setTimeout(this.timeoutHandler, delay);
1226 > }
1227 > async.ts
1228 > get delay(): number {
1229 return this.timeout;
1230 }
1231 > async.ts
1232 > set delay(value: number) {
1233 this.timeout = value;
1234 }
1235 > async.ts
1236 > /**
1237 > * Returns true if scheduled.
1238 > */
1239 > isScheduled(): boolean {
1240 > return this.timeoutToken !== undefined; async.ts
1241 > }
1242 > async.ts
1243 > flush(): void {
1244 if (this.isScheduled()) {
1245 this.cancel();
1247 }
1248 }
1249 > async.ts
1250 > private onTimeout() {
1251 this.timeoutToken = undefined;
1252 if (this.runner) {
1254 }
1255 }
1256 > async.ts
1257 > protected doRun(): void {
1258 this.runner?.();
1259 }
1260 > } async.ts
1261 >
1262 > /**
1263 > * Same as `RunOnceScheduler`, but doesn't count the time spent in sleep mode.
1264 > * > **NOTE**: Only offers 1s resolution.
1265 > *
1266 > * When calling `setTimeout` with 3hrs, and putting the computer immediately to sleep
1267 > * for 8hrs, `setTimeout` will fire **as soon as the computer wakes from sleep**. But
1268 > * this scheduler will execute 3hrs **after waking the computer from sleep**.
1269 > */
1270 > export class ProcessTimeRunOnceScheduler {
1271 >
1272 > private runner: (() => void) | null;
1273 > private timeout: number;
1274 >
1275 > private counter: number;
1276 > private intervalToken: Timeout | undefined;
1277 > private intervalHandler: () => void;
1278 >
1279 > constructor(runner: () => void, delay: number) {
1280 if (delay % 1000 !== 0) {
1281 console.warn(`ProcessTimeRunOnceScheduler resolution is 1s, ${delay}ms is not a multiple of 1000ms.`);
1287 this.intervalHandler = this.onInterval.bind(this);
1288 }
1289 > async.ts
1290 > dispose(): void {
1291 this.cancel();
1292 this.runner = null;
1293 }
1294 > async.ts
1295 > cancel(): void {
1296 if (this.isScheduled()) {
1297 clearInterval(this.intervalToken);
1299 }
1300 }
1301 > async.ts
1302 > /**
1303 > * Cancel previous runner (if any) & schedule a new runner.
1304 > */
1305 > schedule(delay = this.timeout): void {
1306 if (delay % 1000 !== 0) {
1307 console.warn(`ProcessTimeRunOnceScheduler resolution is 1s, ${delay}ms is not a multiple of 1000ms.`);
1311 this.intervalToken = setInterval(this.intervalHandler, 1000);
1312 }
1313 > async.ts
1314 > /**
1315 > * Returns true if scheduled.
1316 > */
1317 > isScheduled(): boolean {
1318 return this.intervalToken !== undefined;
1319 }
1320 > async.ts
1321 > private onInterval() {
1322 this.counter--;
1323 if (this.counter > 0) {
1331 this.runner?.();
1332 }
1333 > } async.ts
1334 >
1335 > export class RunOnceWorker<T> extends RunOnceScheduler<(units: T[]) => void> {
1336 >
1337 > private units: T[] = [];
1338 >
1339 > constructor(runner: (units: T[]) => void, timeout: number) {
1340 super(runner, timeout);
1341 }
1342 > async.ts
1343 > work(unit: T): void {
1344 this.units.push(unit);
1345
1348 }
1349 }
1350 > async.ts
1351 > protected override doRun(): void {
1352 const units = this.units;
1353 this.units = [];
1355 this.runner?.(units);
1356 }
1357 > async.ts
1358 > override dispose(): void {
1359 this.units = [];
1360
1361 super.dispose();
1362 }
1363 > } async.ts
1364 >
1365 > export interface IThrottledWorkerOptions {
1366 >
1367 > /**
1368 > * maximum of units the worker will pass onto handler at once
1369 > */
1370 > maxWorkChunkSize: number;
1371 >
1372 > /**
1373 > * maximum of units the worker will keep in memory for processing
1374 > */
1375 > maxBufferedWork: number | undefined;
1376 >
1377 > /**
1378 > * delay before processing the next round of chunks when chunk size exceeds limits
1379 > */
1380 > throttleDelay: number;
1381 >
1382 > /**
1383 > * When enabled will guarantee that two distinct calls to `work()` are not executed
1384 > * without throttle delay between them.
1385 > * Otherwise if the worker isn't currently throttling it will execute work immediately.
1386 > */
1387 > waitThrottleDelayBetweenWorkUnits?: boolean;
1388 > }
1389 >
1390 > /**
1391 > * The `ThrottledWorker` will accept units of work `T`
1392 > * to handle. The contract is:
1393 > * * there is a maximum of units the worker can handle at once (via `maxWorkChunkSize`)
1394 > * * there is a maximum of units the worker will keep in memory for processing (via `maxBufferedWork`)
1395 > * * after having handled `maxWorkChunkSize` units, the worker needs to rest (via `throttleDelay`)
1396 > */
1397 > export class ThrottledWorker<T> extends Disposable {
1398 >
1399 > private readonly pendingWork: T[] = [];
1400 >
1401 > private readonly throttler = this._register(new MutableDisposable<RunOnceScheduler>());
1402 > private disposed = false;
1403 > private lastExecutionTime = 0;
1404 >
1405 > constructor(
1406 private options: IThrottledWorkerOptions,
1407 private readonly handler: (units: T[]) => void
1409 super();
1410 }
1411 > async.ts
1412 > /**
1413 > * The number of work units that are pending to be processed.
1414 > */
1415 > get pending(): number { return this.pendingWork.length; }
1416 >
1417 > /**
1418 > * Add units to be worked on. Use `pending` to figure out
1419 > * how many units are not yet processed after this method
1420 > * was called.
1421 > *
1422 > * @returns whether the work was accepted or not. If the
1423 > * worker is disposed, it will not accept any more work.
1424 > * If the number of pending units would become larger
1425 > * than `maxPendingWork`, more work will also not be accepted.
1426 > */
1427 > work(units: readonly T[]): boolean {
1428 if (this.disposed) {
1429 return false; // work not accepted: disposed
1469 return true; // work accepted
1470 }
1471 > async.ts
1472 > private doWork(): void {
1473 this.lastExecutionTime = Date.now();
1474
1481 }
1482 }
1483 > async.ts
1484 > private scheduleThrottler(delay = this.options.throttleDelay): void {
1485 this.throttler.value = new RunOnceScheduler(() => {
1486 this.throttler.clear();
1490 this.throttler.value.schedule();
1491 }
1492 > async.ts
1493 > override dispose(): void {
1494 super.dispose();
1495
1497 this.disposed = true;
1498 }
1499 > } async.ts
1500 >
1501 > //#region -- run on idle tricks ------------
1502 >
1503 > export interface IdleDeadline {
1504 > readonly didTimeout: boolean;
1505 > timeRemaining(): number;
1506 > }
1507 >
1508 > type IdleApi = Pick<typeof globalThis, 'requestIdleCallback' | 'cancelIdleCallback'>;
1509 >
1510 >
1511 > /**
1512 > * Execute the callback the next time the browser is idle, returning an
1513 > * {@link IDisposable} that will cancel the callback when disposed. This wraps
1514 > * [requestIdleCallback] so it will fallback to [setTimeout] if the environment
1515 > * doesn't support it.
1516 > *
1517 > * @param callback The callback to run when idle, this includes an
1518 > * [IdleDeadline] that provides the time alloted for the idle callback by the
1519 > * browser. Not respecting this deadline will result in a degraded user
1520 > * experience.
1521 > * @param timeout A timeout at which point to queue no longer wait for an idle
1522 > * callback but queue it on the regular event loop (like setTimeout). Typically
1523 > * this should not be used.
1524 > *
1525 > * [IdleDeadline]: https://developer.mozilla.org/en-US/docs/Web/API/IdleDeadline
1526 > * [requestIdleCallback]: https://developer.mozilla.org/en-US/docs/Web/API/Window/requestIdleCallback
1527 > * [setTimeout]: https://developer.mozilla.org/en-US/docs/Web/API/Window/setTimeout
1528 > *
1529 > * **Note** that there is `dom.ts#runWhenWindowIdle` which is better suited when running inside a browser
1530 > * context
1531 > */
1532 > export let runWhenGlobalIdle: (callback: (idle: IdleDeadline) => void, timeout?: number) => IDisposable;
1533 >
1534 > export let _runWhenIdle: (targetWindow: IdleApi, callback: (idle: IdleDeadline) => void, timeout?: number) => IDisposable;
1535 >
1536 > (function () {
1537 > const safeGlobal: any = globalThis;
1538 > if (typeof safeGlobal.requestIdleCallback !== 'function' || typeof safeGlobal.cancelIdleCallback !== 'function') {
1539 > _runWhenIdle = (_targetWindow, runner, timeout?) => {
1540 setTimeout0(() => {
1541 if (disposed) {
1561 };
1562 };
1563 > } else { async.ts
1564 _runWhenIdle = (targetWindow: typeof safeGlobal, runner, timeout?) => {
1565 const handle: number = targetWindow.requestIdleCallback(runner, typeof timeout === 'number' ? { timeout } : undefined);
1576 };
1577 }
1578 > runWhenGlobalIdle = (runner, timeout) => _runWhenIdle(globalThis, runner, timeout); async.ts
1579 > })();
1580 >
1581 > export function installFakeRunWhenIdle(fakeImpl: typeof _runWhenIdle): IDisposable {
1582 const origRunWhenIdle = _runWhenIdle;
1583 const origRunWhenGlobalIdle = runWhenGlobalIdle;
1589 });
1590 }
1591 > async.ts
1592 > export abstract class AbstractIdleValue<T> {
1593 >
1594 > private readonly _executor: () => void;
1595 > private readonly _handle: IDisposable;
1596 >
1597 > private _didRun: boolean = false;
1598 > private _value?: T;
1599 > private _error: unknown;
1600 >
1601 > constructor(targetWindow: IdleApi, executor: () => T) {
1602 this._executor = () => {
1603 try {
1611 this._handle = _runWhenIdle(targetWindow, () => this._executor());
1612 }
1613 > async.ts
1614 > dispose(): void {
1615 this._handle.dispose();
1616 }
1617 > async.ts
1618 > get value(): T {
1619 if (!this._didRun) {
1620 this._handle.dispose();
1626 return this._value!;
1627 }
1628 > async.ts
1629 > get isInitialized(): boolean {
1630 return this._didRun;
1631 }
1632 > } async.ts
1633 >
1634 > /**
1635 > * An `IdleValue` that always uses the current window (which might be throttled or inactive)
1636 > *
1637 > * **Note** that there is `dom.ts#WindowIdleValue` which is better suited when running inside a browser
1638 > * context
1639 > */
1640 > export class GlobalIdleValue<T> extends AbstractIdleValue<T> {
1641 >
1642 > constructor(executor: () => T) {
1643 super(globalThis, executor);
1644 }
1645 > } async.ts
1646 >
1647 > //#endregion
1648 >
1649 export async function retry<T>(task: ITask<Promise<T>>, delay: number, retries: number): Promise<T> {
1650 let lastError: Error | undefined;
1662 throw lastError;
1663 }
1664 > async.ts
1665 > //#region Task Sequentializer
1666 >
1667 > interface IRunningTask {
1668 > readonly taskId: number;
1669 > readonly cancel: () => void;
1670 > readonly promise: Promise<void>;
1671 > }
1672 >
1673 > interface IQueuedTask {
1674 > readonly promise: Promise<void>;
1675 > readonly promiseResolve: () => void;
1676 > readonly promiseReject: (error: Error) => void;
1677 > run: ITask<Promise<void>>;
1678 > }
1679 >
1680 > export interface ITaskSequentializerWithRunningTask {
1681 > readonly running: Promise<void>;
1682 > }
1683 >
1684 > export interface ITaskSequentializerWithQueuedTask {
1685 > readonly queued: IQueuedTask;
1686 > }
1687 >
1688 > /**
1689 > * @deprecated use `LimitedQueue` instead for an easier to use API
1690 > */
1691 > export class TaskSequentializer {
1692 >
1693 > private _running?: IRunningTask;
1694 > private _queued?: IQueuedTask;
1695 >
1696 > isRunning(taskId?: number): this is ITaskSequentializerWithRunningTask {
1697 if (typeof taskId === 'number') {
1698 return this._running?.taskId === taskId;
1701 return !!this._running;
1702 }
1703 > async.ts
1704 > get running(): Promise<void> | undefined {
1705 return this._running?.promise;
1706 }
1707 > async.ts
1708 > cancelRunning(): void {
1709 this._running?.cancel();
1710 }
1711 > async.ts
1712 > run(taskId: number, promise: Promise<void>, onCancel?: () => void,): Promise<void> {
1713 this._running = { taskId, cancel: () => onCancel?.(), promise };
1714
1717 return promise;
1718 }
1719 > async.ts
1720 > private doneRunning(taskId: number): void {
1721 if (this._running && taskId === this._running.taskId) {
1722
1728 }
1729 }
1730 > async.ts
1731 > private runQueued(): void {
1732 if (this._queued) {
1733 const queued = this._queued;
1738 }
1739 }
1740 > async.ts
1741 > /**
1742 > * Note: the promise to schedule as next run MUST itself call `run`.
1743 > * Otherwise, this sequentializer will report `false` for `isRunning`
1744 > * even when this task is running. Missing this detail means that
1745 > * suddenly multiple tasks will run in parallel.
1746 > */
1747 > queue(run: ITask<Promise<void>>): Promise<void> {
1748
1749 // this is our first queued task, so we create associated promise with it
1767 return this._queued.promise;
1768 }
1769 > async.ts
1770 > hasQueued(): this is ITaskSequentializerWithQueuedTask {
1771 return !!this._queued;
1772 }
1773 > async.ts
1774 > async join(): Promise<void> {
1775 return this._queued?.promise ?? this._running?.promise;
1776 }
1777 > } async.ts
1778 >
1779 > //#endregion
1780 >
1781 > //#region
1782 >
1783 > /**
1784 > * The `IntervalCounter` allows to count the number
1785 > * of calls to `increment()` over a duration of
1786 > * `interval`. This utility can be used to conditionally
1787 > * throttle a frequent task when a certain threshold
1788 > * is reached.
1789 > */
1790 > export class IntervalCounter {
1791 >
1792 > private lastIncrementTime = 0;
1793 >
1794 > private value = 0;
1795 >
1796 > constructor(private readonly interval: number, private readonly nowFn = () => Date.now()) { }
1797 >
1798 > increment(): number {
1799 const now = this.nowFn();
1800
1810 return this.value;
1811 }
1812 > } async.ts
1813 >
1814 > //#endregion
1815 >
1816 > //#region
1817 >
1818 > export type ValueCallback<T = unknown> = (value: T | Promise<T>) => void;
1819 >
1820 > const enum DeferredOutcome {
1821 > Resolved,
1822 > Rejected
1823 > }
1824 >
1825 > /**
1826 > * Creates a promise whose resolution or rejection can be controlled imperatively.
1827 > */
1828 > export class DeferredPromise<T> {
1829 >
1830 > public static fromPromise<T>(promise: Promise<T>): DeferredPromise<T> {
1831 const deferred = new DeferredPromise<T>();
1832 deferred.settleWith(promise);
1833 return deferred;
1834 }
1835 > async.ts
1836 > private completeCallback!: ValueCallback<T>;
1837 > private errorCallback!: (err: unknown) => void;
1838 > private outcome?: { outcome: DeferredOutcome.Rejected; value: unknown } | { outcome: DeferredOutcome.Resolved; value: T };
1839 >
1840 > public get isRejected() {
1841 return this.outcome?.outcome === DeferredOutcome.Rejected;
1842 }
1843 > async.ts
1844 > public get isResolved() {
1845 return this.outcome?.outcome === DeferredOutcome.Resolved;
1846 }
1847 > async.ts
1848 > public get isSettled() {
1849 return !!this.outcome;
1850 }
1851 > async.ts
1852 > public get value() {
1853 return this.outcome?.outcome === DeferredOutcome.Resolved ? this.outcome?.value : undefined;
1854 }
1855 > async.ts
1856 > public readonly p: Promise<T>;
1857 >
1858 > constructor() {
1859 this.p = new Promise<T>((c, e) => {
1860 this.completeCallback = c;
1862 });
1863 }
1864 > async.ts
1865 > public complete(value: T) {
1866 if (this.isSettled) {
1867 return Promise.resolve();
1874 });
1875 }
1876 > async.ts
1877 > public error(err: unknown) {
1878 if (this.isSettled) {
1879 return Promise.resolve();
1886 });
1887 }
1888 > async.ts
1889 > public settleWith(promise: Promise<T>): Promise<void> {
1890 return promise.then(
1891 value => this.complete(value),
1893 );
1894 }
1895 > async.ts
1896 > public cancel() {
1897 return this.error(new CancellationError());
1898 }
1899 > } async.ts
1900 >
1901 > //#endregion
1902 >
1903 > //#region Promises
1904 >
1905 > export namespace Promises {
1906 >
1907 > /**
1908 > * A drop-in replacement for `Promise.all` with the only difference
1909 > * that the method awaits every promise to either fulfill or reject.
1910 > *
1911 > * Similar to `Promise.all`, only the first error will be returned
1912 > * if any.
1913 > */
1914 > export async function settled<T>(promises: Promise<T>[]): Promise<T[]> {
1915 let firstError: Error | undefined = undefined;
1916
1929 return result as unknown as T[]; // cast is needed and protected by the `throw` above
1930 }
1931 > async.ts
1932 > /**
1933 > * A helper to create a new `Promise<T>` with a body that is a promise
1934 > * itself. By default, an error that raises from the async body will
1935 > * end up as a unhandled rejection, so this utility properly awaits the
1936 > * body and rejects the promise as a normal promise does without async
1937 > * body.
1938 > *
1939 > * This method should only be used in rare cases where otherwise `async`
1940 > * cannot be used (e.g. when callbacks are involved that require this).
1941 > */
1942 > export function withAsyncBody<T, E = Error>(bodyFn: (resolve: (value: T) => unknown, reject: (error: E) => unknown) => Promise<unknown>): Promise<T> {
1943 // eslint-disable-next-line no-async-promise-executor
1944 return new Promise<T>(async (resolve, reject) => {
1950 });
1951 }
1952 > } async.ts
1953 >
1954 > export class StatefulPromise<T> {
1955 > private _value: T | undefined = undefined;
1956 > get value(): T | undefined { return this._value; }
1957 >
1958 > private _error: unknown = undefined;
1959 > get error(): unknown { return this._error; }
1960 >
1961 > private _isResolved = false;
1962 > get isResolved() { return this._isResolved; }
1963 >
1964 > public readonly promise: Promise<T>;
1965 >
1966 > constructor(promise: Promise<T>) {
1967 this.promise = promise.then(
1968 value => {
1978 );
1979 }
1980 > async.ts
1981 > /**
1982 > * Returns the resolved value.
1983 > * Throws if the promise is not resolved yet.
1984 > */
1985 > public requireValue(): T {
1986 if (!this._isResolved) {
1987 throw new BugIndicatingError('Promise is not resolved yet');
1992 return this._value!;
1993 }
1994 > } async.ts
1995 >
1996 > export class LazyStatefulPromise<T> {
1997 > private readonly _promise = new Lazy(() => new StatefulPromise(this._compute()));
1998 >
1999 > constructor(
2000 private readonly _compute: () => Promise<T>,
2001 ) { }
2002 > async.ts
2003 > /**
2004 > * Returns the resolved value.
2005 > * Throws if the promise is not resolved yet.
2006 > */
2007 > public requireValue(): T {
2008 return this._promise.value.requireValue();
2009 }
2010 > async.ts
2011 > /**
2012 > * Returns the promise (and triggers a computation of the promise if not yet done so).
2013 > */
2014 > public getPromise(): Promise<T> {
2015 return this._promise.value.promise;
2016 }
2017 > async.ts
2018 > /**
2019 > * Reads the current value without triggering a computation of the promise.
2020 > */
2021 > public get currentValue(): T | undefined {
2022 return this._promise.rawValue?.value;
2023 }
2024 > } async.ts
2025 >
2026 > //#endregion
2027 >
2028 > //#region
2029 >
2030 > const enum AsyncIterableSourceState {
2031 > Initial,
2032 > DoneOK,
2033 > DoneError,
2034 > }
2035 >
2036 > /**
2037 > * An object that allows to emit async values asynchronously or bring the iterable to an error state using `reject()`.
2038 > * This emitter is valid only for the duration of the executor (until the promise returned by the executor settles).
2039 > */
2040 > export interface AsyncIterableEmitter<T> {
2041 > /**
2042 > * The value will be appended at the end.
2043 > *
2044 > * **NOTE** If `reject()` has already been called, this method has no effect.
2045 > */
2046 > emitOne(value: T): void;
2047 > /**
2048 > * The values will be appended at the end.
2049 > *
2050 > * **NOTE** If `reject()` has already been called, this method has no effect.
2051 > */
2052 > emitMany(values: T[]): void;
2053 > /**
2054 > * Writing an error will permanently invalidate this iterable.
2055 > * The current users will receive an error thrown, as will all future users.
2056 > *
2057 > * **NOTE** If `reject()` have already been called, this method has no effect.
2058 > */
2059 > reject(error: Error): void;
2060 > }
2061 >
2062 > /**
2063 > * An executor for the `AsyncIterableObject` that has access to an emitter.
2064 > */
2065 > export interface AsyncIterableExecutor<T> {
2066 > /**
2067 > * @param emitter An object that allows to emit async values valid only for the duration of the executor.
2068 > */
2069 > (emitter: AsyncIterableEmitter<T>): unknown | Promise<unknown>;
2070 > }
2071 >
2072 > /**
2073 > * A rich implementation for an `AsyncIterable<T>`.
2074 > */
2075 > export class AsyncIterableObject<T> implements AsyncIterable<T> {
2076 >
2077 > public static fromArray<T>(items: T[]): AsyncIterableObject<T> {
2078 > return new AsyncIterableObject<T>((writer) => {
2079 > writer.emitMany(items);
2080 > });
2081 > }
2082 >
2083 > public static fromPromise<T>(promise: Promise<T[]>): AsyncIterableObject<T> {
2084 return new AsyncIterableObject<T>(async (emitter) => {
2085 emitter.emitMany(await promise);
2086 });
2087 }
2088 > async.ts
2089 > public static fromPromisesResolveOrder<T>(promises: Promise<T>[]): AsyncIterableObject<T> {
2090 return new AsyncIterableObject<T>(async (emitter) => {
2091 await Promise.all(promises.map(async (p) => emitter.emitOne(await p)));
2092 });
2093 }
2094 > async.ts
2095 > public static merge<T>(iterables: AsyncIterable<T>[]): AsyncIterableObject<T> {
2096 return new AsyncIterableObject(async (emitter) => {
2097 await Promise.all(iterables.map(async (iterable) => {
2102 });
2103 }
2104 > async.ts
2105 > public static EMPTY = AsyncIterableObject.fromArray<any>([]);
2106 >
2107 > private _state: AsyncIterableSourceState;
2108 > private _results: T[];
2109 > private _error: Error | null;
2110 > private readonly _onReturn?: () => void | Promise<void>;
2111 > private readonly _onStateChanged: Emitter<void>;
2112 >
2113 > constructor(executor: AsyncIterableExecutor<T>, onReturn?: () => void | Promise<void>) {
2114 > this._state = AsyncIterableSourceState.Initial;
2115 > this._results = [];
2116 > this._error = null;
2117 > this._onReturn = onReturn;
2118 > this._onStateChanged = new Emitter<void>();
2119 >
2120 > queueMicrotask(async () => {
2121 > const writer: AsyncIterableEmitter<T> = {
2122 > emitOne: (item) => this.emitOne(item),
2123 > emitMany: (items) => this.emitMany(items),
2124 > reject: (error) => this.reject(error)
2125 > };
2126 > try {
2127 > await Promise.resolve(executor(writer));
2128 > this.resolve();
2129 > } catch (err) {
2130 this.reject(err);
2131 > } finally { async.ts
2132 > // The executor has settled; emitting afterwards must be a no-op per the
2133 > // documented "no effect after resolve()/reject()" contract (see emitOne).
2134 > writer.emitOne = () => { };
2135 > writer.emitMany = () => { };
2136 > writer.reject = () => { };
2137 > }
2138 > });
2139 > }
2140 >
2141 > [Symbol.asyncIterator](): AsyncIterator<T, undefined, undefined> {
2142 let i = 0;
2143 return {
2162 };
2163 }
2164 > async.ts
2165 > public static map<T, R>(iterable: AsyncIterable<T>, mapFn: (item: T) => R): AsyncIterableObject<R> {
2166 return new AsyncIterableObject<R>(async (emitter) => {
2167 for await (const item of iterable) {
2170 });
2171 }
2172 > async.ts
2173 > public map<R>(mapFn: (item: T) => R): AsyncIterableObject<R> {
2174 return AsyncIterableObject.map(this, mapFn);
2175 }
2176 > async.ts
2177 > public static filter<T>(iterable: AsyncIterable<T>, filterFn: (item: T) => boolean): AsyncIterableObject<T> {
2178 return new AsyncIterableObject<T>(async (emitter) => {
2179 for await (const item of iterable) {
2184 });
2185 }
2186 > async.ts
2187 > public filter<T2 extends T>(filterFn: (item: T) => item is T2): AsyncIterableObject<T2>;
2188 > public filter(filterFn: (item: T) => boolean): AsyncIterableObject<T>;
2189 > public filter(filterFn: (item: T) => boolean): AsyncIterableObject<T> {
2190 return AsyncIterableObject.filter(this, filterFn);
2191 }
2192 > async.ts
2193 > public static coalesce<T>(iterable: AsyncIterable<T | undefined | null>): AsyncIterableObject<T> {
2194 return <AsyncIterableObject<T>>AsyncIterableObject.filter(iterable, item => !!item);
2195 }
2196 > async.ts
2197 > public coalesce(): AsyncIterableObject<NonNullable<T>> {
2198 return AsyncIterableObject.coalesce(this) as AsyncIterableObject<NonNullable<T>>;
2199 }
2200 > async.ts
2201 > public static async toPromise<T>(iterable: AsyncIterable<T>): Promise<T[]> {
2202 const result: T[] = [];
2203 for await (const item of iterable) {
2206 return result;
2207 }
2208 > async.ts
2209 > public toPromise(): Promise<T[]> {
2210 return AsyncIterableObject.toPromise(this);
2211 }
2212 > async.ts
2213 > /**
2214 > * The value will be appended at the end.
2215 > *
2216 > * **NOTE** If `resolve()` or `reject()` have already been called, this method has no effect.
2217 > */
2218 > private emitOne(value: T): void {
2219 if (this._state !== AsyncIterableSourceState.Initial) {
2220 return;
2225 this._onStateChanged.fire();
2226 }
2227 > async.ts
2228 > /**
2229 > * The values will be appended at the end.
2230 > *
2231 > * **NOTE** If `resolve()` or `reject()` have already been called, this method has no effect.
2232 > */
2233 > private emitMany(values: T[]): void {
2234 > if (this._state !== AsyncIterableSourceState.Initial) {
2235 return;
2236 }
2237 > // it is important to add new values at the end, async.ts
2238 > // as we may have iterators already running on the array
2239 > this._results = this._results.concat(values);
2240 > this._onStateChanged.fire();
2241 > }
2242 >
2243 > /**
2244 > * Calling `resolve()` will mark the result array as complete.
2245 > *
2246 > * **NOTE** `resolve()` must be called, otherwise all consumers of this iterable will hang indefinitely, similar to a non-resolved promise.
2247 > * **NOTE** If `resolve()` or `reject()` have already been called, this method has no effect.
2248 > */
2249 > private resolve(): void {
2250 > if (this._state !== AsyncIterableSourceState.Initial) {
2251 return;
2252 }
2253 > this._state = AsyncIterableSourceState.DoneOK; async.ts
2254 > this._onStateChanged.fire();
2255 > }
2256 >
2257 > /**
2258 > * Writing an error will permanently invalidate this iterable.
2259 > * The current users will receive an error thrown, as will all future users.
2260 > *
2261 > * **NOTE** If `resolve()` or `reject()` have already been called, this method has no effect.
2262 > */
2263 > private reject(error: Error) {
2264 if (this._state !== AsyncIterableSourceState.Initial) {
2265 return;
2269 this._onStateChanged.fire();
2270 }
2271 > } async.ts
2272 >
2273 >
2274 > export function createCancelableAsyncIterableProducer<T>(callback: (token: CancellationToken) => AsyncIterable<T>): CancelableAsyncIterableProducer<T> {
2275 const source = new CancellationTokenSource();
2276 const innerIterable = callback(source.token);
2299 });
2300 }
2301 > async.ts
2302 > export class AsyncIterableSource<T> {
2303 >
2304 > private readonly _deferred = new DeferredPromise<void>();
2305 > private readonly _asyncIterable: AsyncIterableObject<T>;
2306 >
2307 > private _errorFn: (error: Error) => void;
2308 > private _emitOneFn: (item: T) => void;
2309 > private _emitManyFn: (item: T[]) => void;
2310 >
2311 > /**
2312 > *
2313 > * @param onReturn A function that will be called when consuming the async iterable
2314 > * has finished by the consumer, e.g the for-await-loop has be existed (break, return) early.
2315 > * This is NOT called when resolving this source by its owner.
2316 > */
2317 > constructor(onReturn?: () => Promise<void> | void) {
2318 this._asyncIterable = new AsyncIterableObject(emitter => {
2319
2354 };
2355 }
2356 > async.ts
2357 > get asyncIterable(): AsyncIterableObject<T> {
2358 return this._asyncIterable;
2359 }
2360 > async.ts
2361 > resolve(): void {
2362 this._deferred.complete();
2363 }
2364 > async.ts
2365 > reject(error: Error): void {
2366 this._errorFn(error);
2367 this._deferred.complete();
2368 }
2369 > async.ts
2370 > emitOne(item: T): void {
2371 this._emitOneFn(item);
2372 }
2373 > async.ts
2374 > emitMany(items: T[]) {
2375 this._emitManyFn(items);
2376 }
2377 > } async.ts
2378 >
2379 > export function cancellableIterable<T>(iterableOrIterator: AsyncIterator<T> | AsyncIterable<T>, token: CancellationToken): AsyncIterableIterator<T> {
2380 const iterator = Symbol.asyncIterator in iterableOrIterator ? iterableOrIterator[Symbol.asyncIterator]() : iterableOrIterator;
2381
2395 };
2396 }
2397 > async.ts
2398 > type ProducerConsumerValue<T> = {
2399 > ok: true;
2400 > value: T;
2401 > } | {
2402 > ok: false;
2403 > error: Error;
2404 > };
2405 >
2406 > class ProducerConsumer<T> {
2407 > private readonly _unsatisfiedConsumers: DeferredPromise<T>[] = [];
2408 > private readonly _unconsumedValues: ProducerConsumerValue<T>[] = [];
2409 > private _finalValue: ProducerConsumerValue<T> | undefined;
2410 >
2411 > public get hasFinalValue(): boolean {
2412 > return !!this._finalValue;
2413 > }
2414 >
2415 > produce(value: ProducerConsumerValue<T>): void {
2416 this._ensureNoFinalValue();
2417 if (this._unsatisfiedConsumers.length > 0) {
2422 }
2423 }
2424 > async.ts
2425 > produceFinal(value: ProducerConsumerValue<T>): void {
2426 > this._ensureNoFinalValue();
2427 > this._finalValue = value;
2428 > for (const deferred of this._unsatisfiedConsumers) {
2429 this._resolveOrRejectDeferred(deferred, value);
2430 }
2431 > this._unsatisfiedConsumers.length = 0; async.ts
2432 > }
2433 >
2434 > private _ensureNoFinalValue(): void {
2435 > if (this._finalValue) {
2436 throw new BugIndicatingError('ProducerConsumer: cannot produce after final value has been set');
2437 }
2438 > } async.ts
2439 >
2440 > private _resolveOrRejectDeferred(deferred: DeferredPromise<T>, value: ProducerConsumerValue<T>): void {
2441 if (value.ok) {
2442 deferred.complete(value.value);
2445 }
2446 }
2447 > async.ts
2448 > consume(): Promise<T> {
2449 if (this._unconsumedValues.length > 0 || this._finalValue) {
2450 const value = this._unconsumedValues.length > 0 ? this._unconsumedValues.shift()! : this._finalValue!;
2460 }
2461 }
2462 > } async.ts
2463 >
2464 > /**
2465 > * Important difference to AsyncIterableObject:
2466 > * If it is iterated two times, the second iterator will not see the values emitted by the first iterator.
2467 > */
2468 > export class AsyncIterableProducer<T> implements AsyncIterable<T> {
2469 > private readonly _producerConsumer = new ProducerConsumer<IteratorResult<T>>();
2470 >
2471 > constructor(executor: AsyncIterableExecutor<T>, private readonly _onReturn?: () => void) {
2472 > queueMicrotask(async () => {
2473 > const p = executor({
2474 > emitOne: value => this._producerConsumer.produce({ ok: true, value: { done: false, value: value } }),
2475 > emitMany: values => {
2476 > for (const value of values) {
2477 this._producerConsumer.produce({ ok: true, value: { done: false, value: value } });
2478 }
2479 > }, async.ts
2480 > reject: error => this._finishError(error),
2481 > });
2482 >
2483 > if (!this._producerConsumer.hasFinalValue) {
2484 > try {
2485 > await p;
2486 > this._finishOk();
2487 > } catch (error) {
2488 this._finishError(error);
2489 }
2490 > } async.ts
2491 > });
2492 > }
2493 >
2494 > public static fromArray<T>(items: T[]): AsyncIterableProducer<T> {
2495 > return new AsyncIterableProducer<T>((writer) => {
2496 > writer.emitMany(items);
2497 > });
2498 > }
2499 >
2500 > public static fromPromise<T>(promise: Promise<T[]>): AsyncIterableProducer<T> {
2501 return new AsyncIterableProducer<T>(async (emitter) => {
2502 emitter.emitMany(await promise);
2503 });
2504 }
2505 > async.ts
2506 > public static fromPromisesResolveOrder<T>(promises: Promise<T>[]): AsyncIterableProducer<T> {
2507 return new AsyncIterableProducer<T>(async (emitter) => {
2508 await Promise.all(promises.map(async (p) => emitter.emitOne(await p)));
2509 });
2510 }
2511 > async.ts
2512 > public static merge<T>(iterables: AsyncIterable<T>[]): AsyncIterableProducer<T> {
2513 return new AsyncIterableProducer(async (emitter) => {
2514 await Promise.all(iterables.map(async (iterable) => {
2519 });
2520 }
2521 > async.ts
2522 > public static EMPTY = AsyncIterableProducer.fromArray<any>([]);
2523 >
2524 > public static map<T, R>(iterable: AsyncIterable<T>, mapFn: (item: T) => R): AsyncIterableProducer<R> {
2525 return new AsyncIterableProducer<R>(async (emitter) => {
2526 for await (const item of iterable) {
2529 });
2530 }
2531 > async.ts
2532 > public static tee<T>(iterable: AsyncIterable<T>): [AsyncIterableProducer<T>, AsyncIterableProducer<T>] {
2533 let emitter1: AsyncIterableEmitter<T> | undefined;
2534 let emitter2: AsyncIterableEmitter<T> | undefined;
2565 return [p1, p2];
2566 }
2567 > async.ts
2568 > public map<R>(mapFn: (item: T) => R): AsyncIterableProducer<R> {
2569 return AsyncIterableProducer.map(this, mapFn);
2570 }
2571 > async.ts
2572 > public static coalesce<T>(iterable: AsyncIterable<T | undefined | null>): AsyncIterableProducer<T> {
2573 return <AsyncIterableProducer<T>>AsyncIterableProducer.filter(iterable, item => !!item);
2574 }
2575 > async.ts
2576 > public coalesce(): AsyncIterableProducer<NonNullable<T>> {
2577 return AsyncIterableProducer.coalesce(this) as AsyncIterableProducer<NonNullable<T>>;
2578 }
2579 > async.ts
2580 > public static filter<T>(iterable: AsyncIterable<T>, filterFn: (item: T) => boolean): AsyncIterableProducer<T> {
2581 return new AsyncIterableProducer<T>(async (emitter) => {
2582 for await (const item of iterable) {
2587 });
2588 }
2589 > async.ts
2590 > public filter<T2 extends T>(filterFn: (item: T) => item is T2): AsyncIterableProducer<T2>;
2591 > public filter(filterFn: (item: T) => boolean): AsyncIterableProducer<T>;
2592 > public filter(filterFn: (item: T) => boolean): AsyncIterableProducer<T> {
2593 return AsyncIterableProducer.filter(this, filterFn);
2594 }
2595 > async.ts
2596 > private _finishOk(): void {
2597 > if (!this._producerConsumer.hasFinalValue) {
2598 > this._producerConsumer.produceFinal({ ok: true, value: { done: true, value: undefined } });
2599 > }
2600 > }
2601 >
2602 > private _finishError(error: Error): void {
2603 if (!this._producerConsumer.hasFinalValue) {
2604 this._producerConsumer.produceFinal({ ok: false, error: error });
2606 // Warning: this can cause to dropped errors.
2607 }
2608 > async.ts
2609 > private readonly _iterator: AsyncIterator<T, void, void> = {
2610 > next: () => this._producerConsumer.consume(),
2611 > return: () => {
2612 this._onReturn?.();
2613 return Promise.resolve({ done: true, value: undefined });
2614 },
2615 > throw: async (e) => { async.ts
2616 this._finishError(e);
2617 return { done: true, value: undefined };
2618 },
2619 > }; async.ts
2620 >
2621 > [Symbol.asyncIterator](): AsyncIterator<T, void, void> {
2622 return this._iterator;
2623 }
2624 > } async.ts
2625 >
2626 > export class CancelableAsyncIterableProducer<T> extends AsyncIterableProducer<T> {
2627 > constructor(
2628 private readonly _source: CancellationTokenSource,
2629 executor: AsyncIterableExecutor<T>
2631 super(executor);
2632 }
2633 > async.ts
2634 > cancel(): void {
2635 this._source.cancel();
2636 }
2637 > } async.ts
2638 >
2639 > //#endregion
2640 >
2641 > export const AsyncReaderEndOfStream = Symbol('AsyncReaderEndOfStream');
2642 >
2643 > export class AsyncReader<T> {
2644 > private _buffer: T[] = [];
2645 > private _atEnd = false;
2646 >
2647 > public get endOfStream(): boolean { return this._buffer.length === 0 && this._atEnd; }
2648 > private _extendBufferPromise: Promise<void> | undefined;
2649 >
2650 > constructor(
2651 private readonly _source: AsyncIterator<T>
2652 ) {
2653 }
2654 > async.ts
2655 > public async read(): Promise<T | typeof AsyncReaderEndOfStream> {
2656 if (this._buffer.length === 0 && !this._atEnd) {
2657 await this._extendBuffer();
2662 return this._buffer.shift()!;
2663 }
2664 > async.ts
2665 > public async readWhile(predicate: (value: T) => boolean, callback: (element: T) => unknown): Promise<void> {
2666 do {
2667 const piece = await this.peek();
2676 } while (true);
2677 }
2678 > async.ts
2679 > public readBufferedOrThrow(): T | typeof AsyncReaderEndOfStream {
2680 const value = this.peekBufferedOrThrow();
2681 this._buffer.shift();
2682 return value;
2683 }
2684 > async.ts
2685 > public async consumeToEnd(): Promise<void> {
2686 while (!this.endOfStream) {
2687 await this.read();
2688 }
2689 }
2690 > async.ts
2691 > public async peek(): Promise<T | typeof AsyncReaderEndOfStream> {
2692 if (this._buffer.length === 0 && !this._atEnd) {
2693 await this._extendBuffer();
2698 return this._buffer[0];
2699 }
2700 > async.ts
2701 > public peekBufferedOrThrow(): T | typeof AsyncReaderEndOfStream {
2702 if (this._buffer.length === 0) {
2703 if (this._atEnd) {
2709 return this._buffer[0];
2710 }
2711 > async.ts
2712 > public async peekTimeout(timeoutMs: number): Promise<T | typeof AsyncReaderEndOfStream | undefined> {
2713 if (this._buffer.length === 0 && !this._atEnd) {
2714 await raceTimeout(this._extendBuffer(), timeoutMs);
2722 return this._buffer[0];
2723 }
2724 > async.ts
2725 > private _extendBuffer(): Promise<void> {
2726 if (this._atEnd) {
2727 return Promise.resolve();
2742 return this._extendBufferPromise;
2743 }
2744 > } async.ts
2745 >
2746 > export function createTimeout(ms: number, cb: () => void): IDisposable {
2747 const t = setTimeout(cb, ms);
2748 return toDisposable(() => clearTimeout(t));
src/vs/platform/quickinput/common/quickInput.ts 1261 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- quickInput.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../base/common/cancellation.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { createDecorator } from '../../instantiation/common/instantiation.js';
9 > import { IQuickAccessController } from './quickAccess.js';
10 > import { IMatch } from '../../../base/common/filters.js';
11 > import { IItemAccessor } from '../../../base/common/fuzzyScorer.js';
12 > import { ResolvedKeybinding } from '../../../base/common/keybindings.js';
13 > import { IDisposable } from '../../../base/common/lifecycle.js';
14 > import { Schemas } from '../../../base/common/network.js';
15 > import { IObservable } from '../../../base/common/observable.js';
16 > import Severity from '../../../base/common/severity.js';
17 > import { URI } from '../../../base/common/uri.js';
18 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
19 >
20 > export interface IQuickItemHighlights {
21 > label?: IMatch[];
22 > description?: IMatch[];
23 > }
24 >
25 > export interface IQuickPickItemHighlights extends IQuickItemHighlights {
26 > detail?: IMatch[];
27 > }
28 >
29 > export type QuickPickItem = IQuickPickSeparator | IQuickPickItem;
30 >
31 > /**
32 > * Base properties for a quick pick and quick tree item.
33 > */
34 > export interface IQuickItem {
35 > id?: string;
36 > label: string;
37 > ariaLabel?: string;
38 > description?: string;
39 > /**
40 > * Whether the item is displayed in italics.
41 > */
42 > italic?: boolean;
43 > /**
44 > * Whether the item is displayed with a strikethrough.
45 > */
46 > strikethrough?: boolean;
47 > /**
48 > * Icon classes to be passed on as `IIconLabelValueOptions`
49 > * to the underlying `IconLabel` widget.
50 > */
51 > iconClasses?: readonly string[];
52 > iconPath?: { dark: URI; light?: URI };
53 > /**
54 > * Icon class to be assigned to the quick item container
55 > * directly.
56 > */
57 > iconClass?: string;
58 > highlights?: IQuickItemHighlights;
59 > buttons?: readonly IQuickInputButton[];
60 > /**
61 > * Used when we're in multi-select mode. Renders a disabled checkbox.
62 > */
63 > disabled?: boolean;
64 > }
65 >
66 > /**
67 > * Represents a quick pick item used in the quick pick UI.
68 > */
69 > export interface IQuickPickItem extends IQuickItem {
70 > /**
71 > * The type of the quick pick item. Used to distinguish between 'item' and 'separator'
72 > */
73 > type?: 'item';
74 > /**
75 > * The detail text of the quick pick item. Shown as the second line.
76 > */
77 > detail?: string;
78 > /**
79 > * The tooltip for the quick pick item.
80 > */
81 > tooltip?: string | IMarkdownString;
82 > highlights?: IQuickPickItemHighlights;
83 > /**
84 > * Allows to show a keybinding next to the item to indicate
85 > * how the item can be triggered outside of the picker using
86 > * keyboard shortcut.
87 > */
88 > keybinding?: ResolvedKeybinding;
89 > /**
90 > * Whether the item is picked by default when the Quick Pick is shown.
91 > */
92 > picked?: boolean;
93 > /**
94 > * Whether the item is always shown in the Quick Pick regardless of filtering.
95 > */
96 > alwaysShow?: boolean;
97 > /**
98 > * Defaults to true with `IQuickPick.canSelectMany`, can be false to disable picks for a single item
99 > */
100 > pickable?: boolean;
101 > }
102 >
103 > export interface IQuickPickSeparator {
104 > /**
105 > * The type of the quick pick item. Used to distinguish between 'item' and 'separator'
106 > */
107 > type: 'separator';
108 > id?: string;
109 > label?: string;
110 > description?: string;
111 > ariaLabel?: string;
112 > buttons?: readonly IQuickInputButton[];
113 > tooltip?: string | IMarkdownString;
114 > }
115 >
116 > export interface IKeyMods {
117 > readonly ctrlCmd: boolean;
118 > readonly alt: boolean;
119 > readonly shift: boolean;
120 > }
121 >
122 > export function isKeyModified(keyMods: IKeyMods): boolean {
123 return keyMods.ctrlCmd || keyMods.alt || keyMods.shift;
124 }
126 > export const NO_KEY_MODS: IKeyMods = { ctrlCmd: false, alt: false, shift: false };
127 >
128 > export interface IQuickNavigateConfiguration {
129 > keybindings: readonly ResolvedKeybinding[];
130 > }
131 >
132 > export interface IPickOptions<T extends IQuickPickItem> {
133 >
134 > /**
135 > * an optional string to show as the title of the quick input
136 > */
137 > title?: string;
138 >
139 > /**
140 > * the value to prefill in the input box
141 > */
142 > value?: string;
143 >
144 > /**
145 > * an optional string to show as placeholder in the input box to guide the user what she picks on
146 > */
147 > placeHolder?: string;
148 >
149 > /**
150 > * the text to display underneath the input box
151 > */
152 > prompt?: string;
153 >
154 > /**
155 > * an optional flag to include the description when filtering the picks
156 > */
157 > matchOnDescription?: boolean;
158 >
159 > /**
160 > * an optional flag to include the detail when filtering the picks
161 > */
162 > matchOnDetail?: boolean;
163 >
164 > /**
165 > * an optional flag to filter the picks based on label. Defaults to true.
166 > */
167 > matchOnLabel?: boolean;
168 >
169 > /**
170 > * an optional flag to sort the picks based by the label.
171 > */
172 > sortByLabel?: boolean;
173 >
174 > /**
175 > * an optional flag to not close the picker on focus lost
176 > */
177 > ignoreFocusLost?: boolean;
178 >
179 > /**
180 > * an optional flag to make this picker multi-select
181 > */
182 > canPickMany?: boolean;
183 >
184 > /**
185 > * enables quick navigate in the picker to open an element without typing
186 > */
187 > quickNavigate?: IQuickNavigateConfiguration;
188 >
189 > /**
190 > * Hides the input box from the picker UI. This is typically used
191 > * in combination with quick-navigation where no search UI should
192 > * be presented.
193 > */
194 > hideInput?: boolean;
195 >
196 > /**
197 > * a context key to set when this picker is active
198 > */
199 > contextKey?: string;
200 >
201 > /**
202 > * an optional property for the item to focus initially.
203 > */
204 > activeItem?: Promise<T> | T;
205 >
206 > /**
207 > * an optional anchor for the picker
208 > */
209 > anchor?: unknown /* HTMLElement */ | { x: number; y: number };
210 >
211 > /**
212 > * Placement of the quick input relative to {@link anchor}.
213 > * `'overlay'` positions the input box directly on top of the anchor (which must be an HTMLElement)
214 > * and auto-sizes its width to match. Defaults to `'above'`.
215 > */
216 > anchorPosition?: 'above' | 'overlay';
217 >
218 > onKeyMods?: (keyMods: IKeyMods) => void;
219 > onDidFocus?: (entry: T) => void;
220 > onDidTriggerItemButton?: (context: IQuickPickItemButtonContext<T>) => void;
221 > onDidTriggerSeparatorButton?: (context: IQuickPickSeparatorButtonEvent) => void;
222 > }
223 >
224 > export interface IInputOptions {
225 >
226 > /**
227 > * an optional string to show as the title of the quick input
228 > */
229 > title?: string;
230 >
231 > /**
232 > * the value to prefill in the input box
233 > */
234 > value?: string;
235 >
236 > /**
237 > * the selection of value, default to the whole prefilled value
238 > */
239 > valueSelection?: readonly [number, number];
240 >
241 > /**
242 > * the text to display underneath the input box
243 > */
244 > prompt?: string;
245 >
246 > /**
247 > * an optional string to show as placeholder in the input box to guide the user what to type
248 > */
249 > placeHolder?: string;
250 >
251 > /**
252 > * Controls if a password input is shown. Password input hides the typed text.
253 > */
254 > password?: boolean;
255 >
256 > /**
257 > * an optional flag to not close the input on focus lost
258 > */
259 > ignoreFocusLost?: boolean;
260 >
261 > /**
262 > * an optional function that is used to validate user input.
263 > */
264 > validateInput?: (input: string) => Promise<string | null | undefined | { content: string; severity: Severity }>;
265 > }
266 >
267 > export enum QuickInputHideReason {
268 >
269 > /**
270 > * Focus moved away from the quick input.
271 > */
272 > Blur = 1,
273 >
274 > /**
275 > * An explicit user gesture, e.g. pressing Escape key.
276 > */
277 > Gesture,
278 >
279 > /**
280 > * Anything else.
281 > */
282 > Other
283 > }
284 >
285 > export interface IQuickInputHideEvent {
286 > reason: QuickInputHideReason;
287 > }
288 >
289 > /**
290 > * A collection of the different types of QuickInput
291 > */
292 > export const enum QuickInputType {
293 > QuickPick = 'quickPick',
294 > InputBox = 'inputBox',
295 > QuickWidget = 'quickWidget',
296 > QuickTree = 'quickTree'
297 > }
298 >
299 > /**
300 > * Represents a quick input control that allows users to make selections or provide input quickly.
301 > */
302 > export interface IQuickInput extends IDisposable {
303 >
304 > /**
305 > * The type of the quick input.
306 > */
307 > readonly type: QuickInputType;
308 >
309 > /**
310 > * An event that is fired when the quick input is hidden.
311 > */
312 > readonly onDidHide: Event<IQuickInputHideEvent>;
313 >
314 > /**
315 > * An event that is fired when the quick input will be hidden.
316 > */
317 > readonly onWillHide: Event<IQuickInputHideEvent>;
318 >
319 > /**
320 > * An event that is fired when the quick input is disposed.
321 > */
322 > readonly onDispose: Event<void>;
323 >
324 > /**
325 > * The title of the quick input.
326 > */
327 > title: string | undefined;
328 >
329 > /**
330 > * The description of the quick input. This is rendered right below the input box.
331 > */
332 > description: string | undefined;
333 >
334 > /**
335 > * The current step of the quick input rendered in the titlebar.
336 > */
337 > step: number | undefined;
338 >
339 > /**
340 > * The total number of steps in the quick input rendered in the titlebar.
341 > */
342 > totalSteps: number | undefined;
343 >
344 > /**
345 > * The buttons displayed in the quick input titlebar.
346 > */
347 > buttons: ReadonlyArray<IQuickInputButton>;
348 >
349 > /**
350 > * An event that is fired when a button in the quick input is triggered.
351 > */
352 > readonly onDidTriggerButton: Event<IQuickInputButton>;
353 >
354 > /**
355 > * Indicates whether the input is enabled.
356 > */
357 > enabled: boolean;
358 >
359 > /**
360 > * The context key associated with the quick input.
361 > */
362 > contextKey: string | undefined;
363 >
364 > /**
365 > * Indicates whether the quick input is busy. Renders a progress bar if true.
366 > */
367 > busy: boolean;
368 >
369 > /**
370 > * Indicates whether the quick input should be hidden when it loses focus.
371 > */
372 > ignoreFocusOut: boolean;
373 >
374 > /**
375 > * An optional anchor for the quick input.
376 > */
377 > anchor?: unknown /* HTMLElement */ | { x: number; y: number };
378 >
379 > /**
380 > * Placement of the quick input relative to {@link anchor}.
381 > * `'overlay'` positions the input box directly on top of the anchor (which must be an HTMLElement)
382 > * and auto-sizes its width to match. Defaults to `'above'`.
383 > */
384 > anchorPosition?: 'above' | 'overlay';
385 >
386 > /**
387 > * Shows the quick input.
388 > */
389 > show(): void;
390 >
391 > /**
392 > * Hides the quick input.
393 > */
394 > hide(): void;
395 >
396 > /**
397 > * Notifies that the quick input has been hidden.
398 > * @param reason The reason why the quick input was hidden.
399 > */
400 > didHide(reason?: QuickInputHideReason): void;
401 >
402 > /**
403 > * Notifies that the quick input will be hidden.
404 > * @param reason The reason why the quick input will be hidden.
405 > */
406 > willHide(reason?: QuickInputHideReason): void;
407 > }
408 >
409 > export interface IQuickWidget extends IQuickInput {
410 >
411 > /**
412 > * The type of the quick input.
413 > */
414 > readonly type: QuickInputType.QuickWidget;
415 >
416 > /**
417 > * A HTML element that will be rendered inside the quick input.
418 > */
419 > widget: unknown /* HTMLElement */ | undefined;
420 > }
421 >
422 > export interface IQuickPickWillAcceptEvent {
423 >
424 > /**
425 > * Allows to disable the default accept handling
426 > * of the picker. If `veto` is called, the picker
427 > * will not trigger the `onDidAccept` event.
428 > */
429 > veto(): void;
430 > }
431 >
432 > export interface IQuickPickDidAcceptEvent {
433 >
434 > /**
435 > * Signals if the picker item is to be accepted
436 > * in the background while keeping the picker open.
437 > */
438 > inBackground: boolean;
439 > }
440 >
441 > /**
442 > * Represents the activation behavior for items in a quick input. This means which item will be
443 > * "active" (aka focused).
444 > */
445 > export enum ItemActivation {
446 > /**
447 > * No item will be active.
448 > */
449 > NONE,
450 > /**
451 > * First item will be active.
452 > */
453 > FIRST,
454 > /**
455 > * Second item will be active.
456 > */
457 > SECOND,
458 > /**
459 > * Last item will be active.
460 > */
461 > LAST
462 > }
463 >
464 > /**
465 > * Represents the focus options for a quick pick.
466 > */
467 > export enum QuickPickFocus {
468 > /**
469 > * Focus the first item in the list.
470 > */
471 > First = 1,
472 > /**
473 > * Focus the second item in the list.
474 > */
475 > Second,
476 > /**
477 > * Focus the last item in the list.
478 > */
479 > Last,
480 > /**
481 > * Focus the next item in the list.
482 > */
483 > Next,
484 > /**
485 > * Focus the previous item in the list.
486 > */
487 > Previous,
488 > /**
489 > * Focus the next page in the list.
490 > */
491 > NextPage,
492 > /**
493 > * Focus the previous page in the list.
494 > */
495 > PreviousPage,
496 > /**
497 > * Focus the first item under the next separator.
498 > */
499 > NextSeparator,
500 > /**
501 > * Focus the first item under the current separator.
502 > */
503 > PreviousSeparator
504 > }
505 >
506 > /**
507 > * Represents a quick pick control that allows the user to select an item from a list of options.
508 > */
509 > export interface IQuickPick<T extends IQuickPickItem, O extends { useSeparators: boolean } = { useSeparators: false }> extends IQuickInput {
510 >
511 > /**
512 > * The type of the quick input.
513 > */
514 > readonly type: QuickInputType.QuickPick;
515 >
516 > /**
517 > * The current value of the quick pick input.
518 > */
519 > value: string;
520 >
521 > /**
522 > * A method that allows to massage the value used for filtering, e.g, to remove certain parts.
523 > * @param value The value to be filtered.
524 > * @returns The filtered value.
525 > */
526 > filterValue: (value: string) => string;
527 >
528 > /**
529 > * The ARIA label for the quick pick input.
530 > */
531 > ariaLabel: string | undefined;
532 >
533 > /**
534 > * The placeholder text for the quick pick input.
535 > */
536 > placeholder: string | undefined;
537 >
538 > /**
539 > * Text shown below the quick pick input.
540 > */
541 > prompt: string | undefined;
542 >
543 > /**
544 > * An event that is fired when the value of the quick pick input changes.
545 > */
546 > readonly onDidChangeValue: Event<string>;
547 >
548 > /**
549 > * An event that is fired when the quick pick is about to accept the selected item.
550 > */
551 > readonly onWillAccept: Event<IQuickPickWillAcceptEvent>;
552 >
553 > /**
554 > * An event that is fired when the quick pick has accepted the selected item.
555 > */
556 > readonly onDidAccept: Event<IQuickPickDidAcceptEvent>;
557 >
558 > /**
559 > * If enabled, the `onDidAccept` event will be fired when pressing the arrow-right key to accept the selected item without closing the picker.
560 > */
561 > canAcceptInBackground: boolean;
562 >
563 > /**
564 > * The OK button state. It can be a boolean value or the string 'default'.
565 > */
566 > ok: boolean | 'default';
567 >
568 > /**
569 > * The OK button label.
570 > */
571 > okLabel: string | undefined;
572 >
573 > /**
574 > * An event that is fired when the custom button is triggered. The custom button is a button with text rendered to the right of the input.
575 > */
576 > readonly onDidCustom: Event<void>;
577 >
578 > /**
579 > * Whether to show the custom button. The custom button is a button with text rendered to the right of the input.
580 > */
581 > customButton: boolean;
582 >
583 > /**
584 > * The label for the custom button. The custom button is a button with text rendered to the right of the input.
585 > */
586 > customLabel: string | undefined;
587 >
588 > /**
589 > * The hover text for the custom button. The custom button is a button with text rendered to the right of the input.
590 > */
591 > customHover: string | undefined;
592 >
593 > /**
594 > * Whether the custom button should be rendered as a secondary button.
595 > */
596 > customButtonSecondary?: boolean;
597 >
598 > /**
599 > * An event that is fired when an item button is triggered.
600 > */
601 > readonly onDidTriggerItemButton: Event<IQuickPickItemButtonEvent<T>>;
602 >
603 > /**
604 > * An event that is fired when a separator button is triggered.
605 > */
606 > readonly onDidTriggerSeparatorButton: Event<IQuickPickSeparatorButtonEvent>;
607 >
608 > /**
609 > * The items to be displayed in the quick pick.
610 > */
611 > items: O extends { useSeparators: true } ? ReadonlyArray<T | IQuickPickSeparator> : ReadonlyArray<T>;
612 >
613 > /**
614 > * Whether multiple items can be selected. If so, checkboxes will be rendered.
615 > */
616 > canSelectMany: boolean;
617 >
618 > /**
619 > * Whether to match on the description of the items.
620 > */
621 > matchOnDescription: boolean;
622 >
623 > /**
624 > * Whether to match on the detail of the items.
625 > */
626 > matchOnDetail: boolean;
627 >
628 > /**
629 > * Whether to match on the label of the items.
630 > */
631 > matchOnLabel: boolean;
632 >
633 > /**
634 > * The mode to filter the label with. It can be 'fuzzy' or 'contiguous'. Defaults to 'fuzzy'.
635 > */
636 > matchOnLabelMode: 'fuzzy' | 'contiguous';
637 >
638 > /**
639 > * Whether to sort the items by label.
640 > */
641 > sortByLabel: boolean;
642 >
643 > /**
644 > * Whether to keep the scroll position when the quick pick input is updated.
645 > */
646 > keepScrollPosition: boolean;
647 >
648 > /**
649 > * The configuration for quick navigation.
650 > */
651 > quickNavigate: IQuickNavigateConfiguration | undefined;
652 >
653 > /**
654 > * The currently active items.
655 > */
656 > activeItems: ReadonlyArray<T>;
657 >
658 > /**
659 > * An event that is fired when the active items change.
660 > */
661 > readonly onDidChangeActive: Event<T[]>;
662 >
663 > /**
664 > * The item activation behavior for the next time `items` is set. Item activation means which
665 > * item is "active" (aka focused) when the quick pick is opened or when `items` is set.
666 > */
667 > itemActivation: ItemActivation;
668 >
669 > /**
670 > * The currently selected items.
671 > */
672 > selectedItems: ReadonlyArray<T>;
673 >
674 > /**
675 > * An event that is fired when the selected items change.
676 > */
677 > readonly onDidChangeSelection: Event<T[]>;
678 >
679 > /**
680 > * The key modifiers.
681 > */
682 > readonly keyMods: IKeyMods;
683 >
684 > /**
685 > * The selection range for the value in the input.
686 > */
687 > valueSelection: Readonly<[number, number]> | undefined;
688 >
689 > /**
690 > * The validation message for the quick pick. This is rendered below the input.
691 > */
692 > validationMessage: string | undefined;
693 >
694 > /**
695 > * The severity of the validation message.
696 > */
697 > severity: Severity;
698 >
699 > /**
700 > * Checks if the quick pick input has focus.
701 > * @returns `true` if the quick pick input has focus, `false` otherwise.
702 > */
703 > inputHasFocus(): boolean;
704 >
705 > /**
706 > * Focuses on the quick pick input.
707 > */
708 > focusOnInput(): void;
709 >
710 > /**
711 > * Hides the input box from the picker UI. This is typically used in combination with quick-navigation where no search UI should be presented.
712 > */
713 > hideInput: boolean;
714 >
715 > /**
716 > * Controls whether the count for the items should be shown.
717 > */
718 > hideCountBadge: boolean;
719 >
720 > /**
721 > * Whether to hide the "Check All" checkbox.
722 > */
723 > hideCheckAll: boolean;
724 >
725 > /**
726 > * Focus a particular item in the list. Used internally for keyboard navigation.
727 > * @param focus The focus behavior.
728 > */
729 > focus(focus: QuickPickFocus): void;
730 >
731 > /**
732 > * Programmatically accepts an item. Used internally for keyboard navigation.
733 > * @param inBackground Whether you are accepting an item in the background and keeping the picker open.
734 > */
735 > accept(inBackground?: boolean): void;
736 > }
737 >
738 > /**
739 > * Represents an input box in a quick input dialog.
740 > */
741 > export interface IInputBox extends IQuickInput {
742 >
743 > /**
744 > * The type of the quick input.
745 > */
746 > readonly type: QuickInputType.InputBox;
747 >
748 > /**
749 > * Value shown in the input box.
750 > */
751 > value: string;
752 >
753 > /**
754 > * Provide start and end values to be selected in the input box.
755 > */
756 > valueSelection: Readonly<[number, number]> | undefined;
757 >
758 > /**
759 > * Value shown as example for input.
760 > */
761 > placeholder: string | undefined;
762 >
763 > /**
764 > * Determines if the input value should be hidden while typing.
765 > */
766 > password: boolean;
767 >
768 > /**
769 > * Event called when the input value changes.
770 > */
771 > readonly onDidChangeValue: Event<string>;
772 >
773 > /**
774 > * Event called when the user submits the input.
775 > */
776 > readonly onDidAccept: Event<void>;
777 >
778 > /**
779 > * Text show below the input box.
780 > */
781 > prompt: string | undefined;
782 >
783 > /**
784 > * An optional validation message indicating a problem with the current input value.
785 > * Returning undefined clears the validation message.
786 > */
787 > validationMessage: string | undefined;
788 >
789 > /**
790 > * Severity of the input validation message.
791 > */
792 > severity: Severity;
793 >
794 > /**
795 > * Programmatically accepts an item. Used internally for keyboard navigation.
796 > */
797 > accept(): void;
798 > }
799 >
800 > export enum QuickInputButtonLocation {
801 > /**
802 > * In the title bar.
803 > */
804 > Title = 1,
805 >
806 > /**
807 > * To the right of the input box.
808 > */
809 > Inline = 2,
810 >
811 > /**
812 > * At the far end inside the input box.
813 > * Used by the public API to create toggles.
814 > */
815 > Input = 3,
816 > }
817 >
818 > /**
819 > * Represents a button in the quick input UI.
820 > */
821 > export interface IQuickInputButton {
822 > /**
823 > * The path to the icon for the button.
824 > * Either `iconPath` or `iconClass` is required.
825 > */
826 > iconPath?: { dark: URI; light?: URI };
827 > /**
828 > * The CSS class for the icon of the button.
829 > * Either `iconPath` or `iconClass` is required.
830 > */
831 > iconClass?: string;
832 > /**
833 > * The tooltip text for the button.
834 > */
835 > tooltip?: string;
836 > /**
837 > * Whether to always show the button.
838 > * By default, buttons are only visible when hovering over them with the mouse.
839 > */
840 > alwaysVisible?: boolean;
841 > /**
842 > * Where the button should be rendered. The default is {@link QuickInputButtonLocation.Title}.
843 > * @note This property is ignored if the button was added to a QuickPickItem.
844 > */
845 > location?: QuickInputButtonLocation;
846 > /**
847 > * When present, indicates that the button is a toggle button that can be checked or unchecked.
848 > * The `checked` property indicates the current state of the toggle and will be updated
849 > * when the button is clicked.
850 > */
851 > readonly toggle?: { checked: boolean };
852 > /**
853 > * Optional label for the button. When used with secondary actions, this label appears in the overflow menu.
854 > */
855 > label?: string;
856 > /**
857 > * When true, the button will be rendered as a secondary action in the toolbar overflow menu.
858 > * By default, buttons are rendered as primary actions.
859 > * @note This does not currently apply to buttons in the Input location
860 > */
861 > secondary?: boolean;
862 > }
863 >
864 > export interface IQuickInputButtonWithToggle extends IQuickInputButton {
865 > readonly toggle: { checked: boolean };
866 > }
867 >
868 > /**
869 > * Represents an event that occurs when a button associated with a quick pick item is clicked.
870 > * @template T - The type of the quick pick item.
871 > */
872 > export interface IQuickPickItemButtonEvent<T extends IQuickPickItem> {
873 > /**
874 > * The button that was clicked.
875 > */
876 > button: IQuickInputButton;
877 > /**
878 > * The quick pick item associated with the button.
879 > */
880 > item: T;
881 > }
882 >
883 > /**
884 > * Represents an event that occurs when a separator button is clicked in a quick pick.
885 > */
886 > export interface IQuickPickSeparatorButtonEvent {
887 > /**
888 > * The button that was clicked.
889 > */
890 > button: IQuickInputButton;
891 > /**
892 > * The separator associated with the button.
893 > */
894 > separator: IQuickPickSeparator;
895 > }
896 >
897 > /**
898 > * Represents a context for a button associated with a quick pick item.
899 > * @template T - The type of the quick pick item.
900 > */
901 > export interface IQuickPickItemButtonContext<T extends IQuickPickItem> extends IQuickPickItemButtonEvent<T> {
902 > /**
903 > * Removes the associated item from the quick pick.
904 > */
905 > removeItem(): void;
906 > }
907 >
908 > export type QuickPickInput<T = IQuickPickItem> = T | IQuickPickSeparator;
909 >
910 >
911 > //#region Fuzzy Scorer Support
912 >
913 > export type IQuickPickItemWithResource = IQuickPickItem & { resource?: URI };
914 >
915 > export class QuickPickItemScorerAccessor implements IItemAccessor<IQuickPickItemWithResource> {
916 >
917 > constructor(private options?: { skipDescription?: boolean; skipPath?: boolean }) { }
918 >
919 > getItemLabel(entry: IQuickPickItemWithResource): string {
920 return entry.label;
921 }
923 > getItemDescription(entry: IQuickPickItemWithResource): string | undefined {
924 if (this.options?.skipDescription) {
925 return undefined;
928 return entry.description;
929 }
931 > getItemPath(entry: IQuickPickItemWithResource): string | undefined {
932 if (this.options?.skipPath) {
933 return undefined;
940 return entry.resource?.path;
941 }
942 > } quickInput.ts
943 >
944 > export const quickPickItemScorerAccessor = new QuickPickItemScorerAccessor();
945 >
946 > //#endregion
947 >
948 > export const IQuickInputService = createDecorator<IQuickInputService>('quickInputService');
949 >
950 > export type Omit<T, K extends keyof T> = Pick<T, Exclude<keyof T, K>>;
951 >
952 > export type QuickInputAlignment = 'top' | 'center' | 'custom';
953 >
954 > export interface IQuickInputService {
955 >
956 > readonly _serviceBrand: undefined;
957 >
958 > /**
959 > * Provides access to the back button in quick input.
960 > */
961 > readonly backButton: IQuickInputButton;
962 >
963 > /**
964 > * Provides access to the quick access providers.
965 > */
966 > readonly quickAccess: IQuickAccessController;
967 >
968 > /**
969 > * Allows to register on the event that quick input is showing.
970 > */
971 > readonly onShow: Event<void>;
972 >
973 > /**
974 > * Allows to register on the event that quick input is hiding.
975 > */
976 > readonly onHide: Event<void>;
977 >
978 > /**
979 > * The current alignment of the quick input widget.
980 > */
981 > readonly alignment: IObservable<QuickInputAlignment>;
982 >
983 > /**
984 > * Opens the quick input box for selecting items and returns a promise
985 > * with the user selected item(s) if any.
986 > */
987 > pick<T extends IQuickPickItem>(picks: Promise<QuickPickInput<T>[]> | QuickPickInput<T>[], options?: IPickOptions<T> & { canPickMany: true }, token?: CancellationToken): Promise<T[] | undefined>;
988 > pick<T extends IQuickPickItem>(picks: Promise<QuickPickInput<T>[]> | QuickPickInput<T>[], options?: IPickOptions<T> & { canPickMany: false }, token?: CancellationToken): Promise<T | undefined>;
989 > pick<T extends IQuickPickItem>(picks: Promise<QuickPickInput<T>[]> | QuickPickInput<T>[], options?: Omit<IPickOptions<T>, 'canPickMany'>, token?: CancellationToken): Promise<T | undefined>;
990 >
991 > /**
992 > * Opens the quick input box for text input and returns a promise with the user typed value if any.
993 > */
994 > input(options?: IInputOptions, token?: CancellationToken): Promise<string | undefined>;
995 >
996 > /**
997 > * Provides raw access to the quick pick controller.
998 > */
999 > createQuickPick<T extends IQuickPickItem>(options: { useSeparators: true }): IQuickPick<T, { useSeparators: true }>;
1000 > createQuickPick<T extends IQuickPickItem>(options?: { useSeparators: boolean }): IQuickPick<T, { useSeparators: false }>;
1001 >
1002 > /**
1003 > * Provides raw access to the input box controller.
1004 > */
1005 > createInputBox(): IInputBox;
1006 >
1007 > /**
1008 > * Provides raw access to the quick widget controller.
1009 > */
1010 > createQuickWidget(): IQuickWidget;
1011 >
1012 > /**
1013 > * Provides raw access to the quick tree controller.
1014 > * @template T The type of items in the quick tree.
1015 > */
1016 > createQuickTree<T extends IQuickTreeItem>(): IQuickTree<T>;
1017 >
1018 > /**
1019 > * Moves focus into quick input.
1020 > */
1021 > focus(): void;
1022 >
1023 > /**
1024 > * Toggle the checked state of the selected item.
1025 > */
1026 > toggle(): void;
1027 >
1028 > /**
1029 > * Navigate inside the opened quick input list.
1030 > */
1031 > navigate(next: boolean, quickNavigate?: IQuickNavigateConfiguration): void;
1032 >
1033 > /**
1034 > * Navigate back in a multi-step quick input.
1035 > */
1036 > back(): Promise<void>;
1037 >
1038 > /**
1039 > * Accept the selected item.
1040 > *
1041 > * @param keyMods allows to override the state of key
1042 > * modifiers that should be present when invoking.
1043 > */
1044 > accept(keyMods?: IKeyMods): Promise<void>;
1045 >
1046 > /**
1047 > * Cancels quick input and closes it.
1048 > */
1049 > cancel(reason?: QuickInputHideReason): Promise<void>;
1050 >
1051 > /**
1052 > * Toggles hover for the current quick input item
1053 > */
1054 > toggleHover(): void;
1055 >
1056 > /**
1057 > * The current quick pick that is visible. Undefined if none is open.
1058 > */
1059 > currentQuickInput: IQuickInput | undefined;
1060 >
1061 > /**
1062 > * Set the alignment of the quick input.
1063 > * @param alignment either a preset or a custom alignment
1064 > */
1065 > setAlignment(alignment: 'top' | 'center' | { top: number; left: number }): void;
1066 > }
1067 >
1068 > //#region Quick Tree
1069 >
1070 > /**
1071 > * Represents a quick tree control that displays hierarchical data with checkboxes.
1072 > */
1073 > export interface IQuickTree<T extends IQuickTreeItem> extends IQuickInput {
1074 >
1075 > /**
1076 > * The type of the quick input.
1077 > */
1078 > readonly type: QuickInputType.QuickTree;
1079 >
1080 > /**
1081 > * The current value of the quick tree filter input.
1082 > */
1083 > value: string;
1084 >
1085 > /**
1086 > * The ARIA label for the quick tree input.
1087 > */
1088 > ariaLabel: string | undefined;
1089 >
1090 > /**
1091 > * The placeholder text for the quick tree filter input.
1092 > */
1093 > placeholder: string | undefined;
1094 >
1095 > /**
1096 > * An event that is fired when the filter value changes.
1097 > */
1098 > readonly onDidChangeValue: Event<string>;
1099 >
1100 > /**
1101 > * An event that is fired when the quick tree has accepted the selected items.
1102 > */
1103 > readonly onDidAccept: Event<void>;
1104 >
1105 > /**
1106 > * Whether to match on the description of the items.
1107 > */
1108 > matchOnDescription: boolean;
1109 >
1110 > /**
1111 > * Whether to match on the label of the items.
1112 > */
1113 > matchOnLabel: boolean;
1114 >
1115 > /**
1116 > * Whether to sort the items by label. Defaults to true.
1117 > */
1118 > sortByLabel: boolean;
1119 >
1120 > /**
1121 > * The currently active items.
1122 > */
1123 > activeItems: ReadonlyArray<T>;
1124 >
1125 > /**
1126 > * The validation message for the quick pick. This is rendered below the input.
1127 > */
1128 > validationMessage: string | undefined;
1129 >
1130 > /**
1131 > * The severity of the validation message.
1132 > */
1133 > severity: Severity;
1134 >
1135 > /**
1136 > * The items currently displayed in the quick tree.
1137 > * @note modifications to this array directly will not cause updates.
1138 > */
1139 > readonly itemTree: ReadonlyArray<Readonly<T>>;
1140 >
1141 > /**
1142 > * The currently selected leaf items.
1143 > */
1144 > readonly checkedLeafItems: ReadonlyArray<T>;
1145 >
1146 > /**
1147 > * Get the parent element of the element passed in
1148 > * @param element
1149 > */
1150 > getParent(element: T): T | undefined;
1151 >
1152 > /**
1153 > * An event that is fired when the active items change.
1154 > */
1155 > readonly onDidChangeActive: Event<ReadonlyArray<T>>;
1156 >
1157 > /**
1158 > * An event that is fired when the selected items change.
1159 > */
1160 > readonly onDidChangeCheckedLeafItems: Event<ReadonlyArray<T>>;
1161 >
1162 > /**
1163 > * An event that is fired when the checkbox state of an item changes.
1164 > */
1165 > readonly onDidChangeCheckboxState: Event<T>;
1166 >
1167 > /**
1168 > * An event that is fired when an item button is triggered.
1169 > */
1170 > readonly onDidTriggerItemButton: Event<IQuickTreeItemButtonEvent<T>>;
1171 >
1172 > /**
1173 > * Sets the items to be displayed in the quick tree.
1174 > * @param itemTree The items to display.
1175 > */
1176 > setItemTree(itemTree: T[]): void;
1177 >
1178 > /**
1179 > * Expands an item.
1180 > * @param element The item to expand.
1181 > */
1182 > expand(element: T): void;
1183 >
1184 > /**
1185 > * Collapses an item.
1186 > * @param element The item to collapse.
1187 > */
1188 > collapse(element: T): void;
1189 >
1190 > /**
1191 > * Checks if an item is collapsed.
1192 > * @param element The item to check.
1193 > * @returns True if the item is collapsed.
1194 > */
1195 > isCollapsed(element: T): boolean;
1196 >
1197 > /**
1198 > * Focuses on the tree input.
1199 > */
1200 > focusOnInput(): void;
1201 >
1202 > /**
1203 > * Reveals and focuses a specific item in the tree.
1204 > * @param element The item to reveal and focus.
1205 > */
1206 > reveal(element: T): void;
1207 >
1208 > /**
1209 > * Focus a particular item in the list. Used internally for keyboard navigation.
1210 > * @param focus The focus behavior.
1211 > */
1212 > focus(focus: QuickPickFocus): void;
1213 >
1214 > /**
1215 > * Programmatically accepts an item. Used internally for keyboard navigation.
1216 > * @param inBackground Whether you are accepting an item in the background and keeping the picker open.
1217 > */
1218 > accept(inBackground?: boolean): void;
1219 > }
1220 >
1221 > /**
1222 > * Represents a tree item in the quick tree.
1223 > */
1224 > export interface IQuickTreeItem extends IQuickItem {
1225 > /**
1226 > * The checked state of the item. Can be true, false, or 'mixed' for tri-state.
1227 > * When canSelectMany is false, this is ignored and the item is treated as a single selection.
1228 > * When canSelectMany is true, this indicates the checkbox state of the item.
1229 > * If undefined, the item is unchecked by default.
1230 > */
1231 > checked?: boolean | 'mixed';
1232 >
1233 > /**
1234 > * The collapsible state of the tree item. Defaults to 'Expanded' if children are present.
1235 > */
1236 > collapsed?: boolean;
1237 >
1238 > /**
1239 > * The children of this tree item.
1240 > */
1241 > children?: readonly IQuickTreeItem[];
1242 >
1243 > /**
1244 > * Defaults to true, can be false to disable picks for a single item.
1245 > * When false, the item is not selectable and does not respond to mouse/keyboard activation.
1246 > */
1247 > pickable?: boolean;
1248 > }
1249 >
1250 > /**
1251 > * Represents an event that occurs when the checkbox state of a tree item changes.
1252 > * @template T - The type of the tree item.
1253 > */
1254 > export interface IQuickTreeCheckboxEvent<T extends IQuickTreeItem> {
1255 > /**
1256 > * The tree item whose checkbox state changed.
1257 > */
1258 > item: T;
1259 >
1260 > /**
1261 > * The new checked state.
1262 > */
1263 > checked: boolean | 'mixed';
1264 > }
1265 >
1266 > /**
1267 > * Represents an event that occurs when a button associated with a quick tree item is clicked.
1268 > * @template T - The type of the quick tree item.
1269 > */
1270 > export interface IQuickTreeItemButtonEvent<T extends IQuickTreeItem> {
1271 > /**
1272 > * The button that was clicked.
1273 > */
1274 > button: IQuickInputButton;
1275 > /**
1276 > * The quick tree item associated with the button.
1277 > */
1278 > item: T;
1279 > }
1280 >
1281 > //#endregion
src/vs/platform/agentHost/common/state/protocol/common/commands.ts 1071 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI, Snapshot } from './state.js';
10 > import type { ActionEnvelope, StateAction } from './actions.js';
11 > import type { TelemetryCapabilities } from '../channels-otlp/state.js';
12 >
13 > // ─── BaseParams ──────────────────────────────────────────────────────────────
14 >
15 > /**
16 > * Base shape every command's params extends.
17 > *
18 > * `channel` identifies the channel the command targets, mirroring the
19 > * `channel` field on every protocol notification. For commands that operate
20 > * on a specific channel (a session, terminal, or changeset), `channel` is
21 > * that channel's URI. For commands that are connection-level rather than
22 > * channel-scoped (e.g. {@link InitializeParams | `initialize`},
23 > * {@link PingParams | `ping`}, {@link ListSessionsParams | `listSessions`},
24 > * the `resource*` filesystem commands, and {@link AuthenticateParams |
25 > * `authenticate`}), the params type narrows `channel` to the literal
26 > * root URI `'ahp-root://'`.
27 > *
28 > * This invariant lets implementations route every incoming message —
29 > * request, response, or notification — by inspecting `params.channel`
30 > * without needing to know the per-method param shape.
31 > *
32 > * @category Commands
33 > */
34 > export interface BaseParams {
35 > /** Channel URI this command targets. */
36 > channel: URI;
37 > }
38 >
39 > // ─── Pagination ──────────────────────────────────────────────────────────────
40 >
41 > /**
42 > * Cursor-based pagination inputs, mixed into the params of any list command
43 > * that can page a large result set (e.g. {@link ListSessionsParams |
44 > * `listSessions`}). The paired output is {@link PaginatedResult}.
45 > *
46 > * Pagination is **opaque and cursor-based**, mirroring the shape `fetchTurns`
47 > * already uses for chat history: the server owns the ordering and keyset, and
48 > * the client walks pages by echoing the cursor from the previous
49 > * {@link PaginatedResult.nextCursor} back on the next request.
50 > *
51 > * The contract every paginated command shares:
52 > *
53 > * - To fetch the first page, omit `cursor`. Supply `limit` to bound the page.
54 > * - If the result carries a {@link PaginatedResult.nextCursor}, more entries
55 > * exist — pass it back as `cursor` to fetch the following page. A missing
56 > * `nextCursor` signals the end of the collection.
57 > * - Cursors are **server-defined and opaque**: clients MUST NOT parse, modify,
58 > * or persist them across connections. An unrecognised cursor SHOULD be
59 > * rejected with an `InvalidParams` error.
60 > * - Pagination is **fully additive**: a client that omits `limit`/`cursor` and
61 > * ignores `nextCursor` sees the pre-pagination behaviour (subject to any
62 > * server-imposed cap), and a server that does not paginate ignores the inputs
63 > * and returns everything in a single page.
64 > *
65 > * @category Commands
66 > */
67 > export interface PaginatedParams {
68 > /**
69 > * Maximum number of entries to return in this page. The server SHOULD respect
70 > * this bound but MAY return fewer entries and MAY impose its own upper cap.
71 > * Omit to let the server choose the page size.
72 > */
73 > limit?: number;
74 > /**
75 > * Opaque pagination cursor from a previous {@link PaginatedResult.nextCursor}.
76 > * Omit to fetch the first page. Cursors are server-defined and MUST be treated
77 > * as opaque — do not parse, modify, or persist them across connections. An
78 > * unrecognised cursor SHOULD be rejected with an `InvalidParams` error.
79 > */
80 > cursor?: string;
81 > }
82 >
83 > /**
84 > * Cursor-based pagination output, extended by the result of any list command
85 > * that can page a large result set (e.g. {@link ListSessionsResult |
86 > * `listSessions`}). See {@link PaginatedParams} for the full pagination
87 > * contract shared by every paginated command.
88 > *
89 > * @category Commands
90 > */
91 > export interface PaginatedResult {
92 > /**
93 > * Opaque cursor for the next page. Present when more entries exist beyond the
94 > * returned page; absent signals the end of the collection. Pass it back as
95 > * {@link PaginatedParams.cursor} to fetch the following page.
96 > */
97 > nextCursor?: string;
98 > }
99 >
100 > // ─── initialize ──────────────────────────────────────────────────────────────
101 >
102 > /**
103 > * Identifies a protocol implementation — the software (and build) on one end
104 > * of the connection, as distinct from the {@link AgentInfo | agent persona} it
105 > * hosts. Carried as {@link InitializeParams.clientInfo | `clientInfo`} on the
106 > * client side and {@link InitializeResult.serverInfo | `serverInfo`} on the
107 > * server side, mirroring LSP's `clientInfo`/`serverInfo` and MCP's
108 > * `Implementation`.
109 > *
110 > * This is **informational only**: it exists for logging, telemetry, an
111 > * about/status affordance, and — as a last resort — a known-issue workaround
112 > * for a specific buggy build. It is **not** a feature-detection mechanism.
113 > * Feature availability stays with the capability model
114 > * ({@link ClientCapabilities} and the various `*.capabilities` declarations);
115 > * implementations SHOULD NOT gate protocol behaviour on parsing
116 > * {@link Implementation.version | `version`}.
117 > *
118 > * @category Commands
119 > */
120 > export interface Implementation {
121 > /** Implementation name, e.g. a product or package identifier. */
122 > name: string;
123 > /**
124 > * Implementation version. A [SemVer](https://semver.org) string is
125 > * recommended but not required.
126 > */
127 > version?: string;
128 > /** Optional human-readable display name. */
129 > title?: string;
130 > }
131 >
132 > /**
133 > * Establishes a new connection and negotiates the protocol version.
134 > * This MUST be the first message sent by the client.
135 > *
136 > * @category Commands
137 > * @method initialize
138 > * @direction Client → Server
139 > * @messageType Request
140 > * @version 1
141 > * @see {@link /specification/lifecycle | Lifecycle} for the full handshake flow.
142 > */
143 > export interface InitializeParams extends BaseParams {
144 > channel: 'ahp-root://';
145 > /**
146 > * Protocol versions the client is willing to speak, ordered from most
147 > * preferred to least preferred. Each entry is a [SemVer](https://semver.org)
148 > * `MAJOR.MINOR.PATCH` string (e.g. `"0.1.0"`).
149 > *
150 > * The server selects one entry and returns it as `InitializeResult.protocolVersion`.
151 > * If the server cannot speak any of the offered versions, it MUST return
152 > * error code `-32005` (`UnsupportedProtocolVersion`).
153 > */
154 > protocolVersions: string[];
155 > /** Unique client identifier */
156 > clientId: string;
157 > /**
158 > * Optional identity of the client implementation (name and version).
159 > * Informational only — see {@link Implementation} for how it may and may not
160 > * be used. Distinct from {@link InitializeParams.clientId | `clientId`},
161 > * which is an opaque per-connection identifier used for reconnection, not a
162 > * human-readable implementation name.
163 > */
164 > clientInfo?: Implementation;
165 > /** URIs to subscribe to during handshake */
166 > initialSubscriptions?: URI[];
167 > /**
168 > * IETF BCP 47 language tag indicating the client's preferred locale
169 > * (e.g. `"en-US"`, `"ja"`). The server SHOULD use this to localise
170 > * user-facing strings such as confirmation option labels.
171 > */
172 > locale?: string;
173 > /**
174 > * Optional client capability declarations.
175 > *
176 > * Servers SHOULD only advertise features whose corresponding client
177 > * capability is set here. Absent means "not declared" — the server
178 > * MUST assume the client does not support the feature.
179 > */
180 > capabilities?: ClientCapabilities;
181 > }
182 >
183 > /**
184 > * Optional capabilities a client declares during `initialize`.
185 > *
186 > * Each field is a presence flag: an empty object `{}` means "supported",
187 > * absence means "not supported". Sub-fields on individual capabilities
188 > * are reserved for future per-capability options.
189 > *
190 > * @category Commands
191 > */
192 > export interface ClientCapabilities {
193 > /**
194 > * Client can render
195 > * [MCP Apps](https://github.com/modelcontextprotocol/ext-apps) — i.e.
196 > * it can host the View sandbox, run the `ui/*` protocol against it,
197 > * and forward `mcp://`-channel traffic on the App's behalf.
198 > *
199 > * Hosts SHOULD only populate
200 > * {@link McpServerCustomization.mcpApp | `McpServerCustomization.mcpApp`}
201 > * (and expose the corresponding
202 > * {@link McpServerCustomization.channel | `mcp://` channel}) when this
203 > * capability is declared. Clients that omit it MUST treat
204 > * App-bearing tool calls as ordinary MCP tool calls.
205 > */
206 > mcpApps?: Record<string, never>;
207 > }
208 >
209 > /**
210 > * Result of the `initialize` command.
211 > *
212 > * `protocolVersion` is the version the server has selected from the client's
213 > * `protocolVersions` list. The client and server MUST use this version for
214 > * the rest of the connection. If the server cannot speak any of the offered
215 > * versions it MUST return error code `-32005` (`UnsupportedProtocolVersion`)
216 > * instead of a result.
217 > */
218 > export interface InitializeResult {
219 > /**
220 > * Protocol version selected by the server. MUST be one of the entries in
221 > * `InitializeParams.protocolVersions`. Formatted as a [SemVer](https://semver.org)
222 > * `MAJOR.MINOR.PATCH` string (e.g. `"0.1.0"`).
223 > */
224 > protocolVersion: string;
225 > /** Current server sequence number */
226 > serverSeq: number;
227 > /**
228 > * Optional identity of the server implementation (name and version).
229 > * Informational only — see {@link Implementation} for how it may and may not
230 > * be used. Whereas {@link InitializeResult.protocolVersion | `protocolVersion`}
231 > * identifies the negotiated protocol, `serverInfo` identifies the host
232 > * software behind it.
233 > */
234 > serverInfo?: Implementation;
235 > /** Snapshots for each `initialSubscriptions` URI */
236 > snapshots: Snapshot[];
237 > /** Suggested default directory for remote filesystem browsing */
238 > defaultDirectory?: URI;
239 > /**
240 > * Characters that, when typed in a {@link Message} input, SHOULD cause
241 > * the client to issue a `completions` request with
242 > * {@link CompletionItemKind.UserMessage}. Typically includes characters like
243 > * `'@'` or `'/'`.
244 > */
245 > completionTriggerCharacters?: string[];
246 > /**
247 > * Prefix that the host recognizes at the start of a user {@link Message.text}
248 > * as a shorthand for executing the remainder as a terminal command. Currently
249 > * the standardized convention is `"!"`; absence means the host does not
250 > * support command prefixes.
251 > */
252 > terminalCommandPrefix?: string;
253 > /**
254 > * OTLP telemetry channels the host emits, if any. Each populated field is
255 > * either a literal `ahp-otlp:` channel URI or an RFC 6570 URI template a
256 > * client expands before subscribing (currently only the `logs` channel
257 > * defines a template variable, `{level}`, for subscriber-side severity
258 > * filtering). Clients MAY ignore signals they cannot process.
259 > *
260 > * @see {@link /specification/telemetry-channel | Telemetry Channel}
261 > */
262 > telemetry?: TelemetryCapabilities;
263 > }
264 >
265 > // ─── ping ────────────────────────────────────────────────────────────────────
266 >
267 > /**
268 > * Verifies that the AHP connection is still alive and keeps it from being
269 > * closed by idle-timeout intermediaries (proxies, load balancers, etc.).
270 > *
271 > * The server MUST respond regardless of whether the client has completed
272 > * `initialize` or holds any subscriptions. Ping carries no payload in either
273 > * direction; the response itself is the signal.
274 > *
275 > * @category Commands
276 > * @method ping
277 > * @direction Client → Server
278 > * @messageType Request
279 > * @version 1
280 > */
281 > export interface PingParams extends BaseParams {
282 > channel: 'ahp-root://';
283 > }
284 >
285 > // ─── reconnect ───────────────────────────────────────────────────────────────
286 >
287 > /**
288 > * Discriminant for reconnect result types.
289 > *
290 > * @category Commands
291 > */
292 > export const enum ReconnectResultType {
293 > Replay = 'replay',
294 > Snapshot = 'snapshot',
295 > }
296 >
297 > /**
298 > * Re-establishes a dropped connection. The server replays missed actions or
299 > * provides fresh snapshots.
300 > *
301 > * @category Commands
302 > * @method reconnect
303 > * @direction Client → Server
304 > * @messageType Request
305 > * @version 1
306 > * @see {@link /specification/lifecycle | Lifecycle} for details.
307 > */
308 > export interface ReconnectParams extends BaseParams {
309 > channel: 'ahp-root://';
310 > /** Client identifier from the original connection */
311 > clientId: string;
312 > /** Last `serverSeq` the client received */
313 > lastSeenServerSeq: number;
314 > /** URIs the client was subscribed to */
315 > subscriptions: URI[];
316 > }
317 >
318 > /**
319 > * Reconnect result when the server can replay from the requested sequence.
320 > *
321 > * The server MUST include all replayed data in the response.
322 > */
323 > export interface ReconnectReplayResult {
324 > /** Discriminant */
325 > type: ReconnectResultType.Replay;
326 > /** Missed action envelopes since `lastSeenServerSeq` */
327 > actions: ActionEnvelope[];
328 > /**
329 > * URIs from `ReconnectParams.subscriptions` that the server cannot resume.
330 > * This includes resources that no longer exist (e.g. disposed sessions or
331 > * terminals) as well as resources the client is no longer permitted to
332 > * observe. Clients SHOULD drop these from their local subscription set.
333 > */
334 > missing: URI[];
335 > }
336 >
337 > /**
338 > * Reconnect result when the gap exceeds the replay buffer.
339 > */
340 > export interface ReconnectSnapshotResult {
341 > /** Discriminant */
342 > type: ReconnectResultType.Snapshot;
343 > /** Fresh snapshots for each subscription */
344 > snapshots: Snapshot[];
345 > }
346 >
347 > /** Result of the `reconnect` command. */
348 > export type ReconnectResult = ReconnectReplayResult | ReconnectSnapshotResult;
349 >
350 > // ─── subscribe ───────────────────────────────────────────────────────────────
351 >
352 > /**
353 > * Subscribe to a URI-identified channel.
354 > *
355 > * A channel MAY have state associated with it (e.g. root, sessions,
356 > * terminals) or be stateless (pure pub/sub for streaming data). For
357 > * state-bearing channels the result includes a snapshot; for stateless
358 > * channels `snapshot` is omitted.
359 > *
360 > * @category Commands
361 > * @method subscribe
362 > * @direction Client → Server
363 > * @messageType Request
364 > * @version 1
365 > * @see {@link /specification/subscriptions | Subscriptions}
366 > */
367 > export interface SubscribeParams extends BaseParams {
368 > /**
369 > * Optional delivery preferences for this subscription.
370 > *
371 > * Servers MAY use these preferences to buffer and coalesce high-frequency
372 > * updates while preserving the same reduced state. Omit this field for the
373 > * server's default delivery behavior.
374 > */
375 > delivery?: SubscriptionDeliveryOptions;
376 > /**
377 > * Optional client-requested shape for the returned snapshot.
378 > *
379 > * Servers that do not understand a requested view ignore it and return their
380 > * default snapshot. Clients MUST tolerate receiving more state than requested.
381 > */
382 > view?: SubscribeView;
383 > }
384 >
385 > /**
386 > * Optional client-requested shape for a subscription snapshot.
387 > *
388 > * @category Commands
389 > */
390 > export interface SubscribeView {
391 > /**
392 > * Advisory number of most-recent completed turns to expose in a chat
393 > * snapshot.
394 > *
395 > * Servers MAY return more or fewer turns than requested. When omitted, the
396 > * host MUST return all retained turns. When older turns remain available, the
397 > * returned {@link ChatState} carries `turnsNextCursor`; clients pass that
398 > * cursor to `fetchTurns` to ask the host to page more turns into the chat
399 > * state.
400 > */
401 > turns?: number;
402 > }
403 >
404 > /**
405 > * Advisory delivery preferences for a single subscription.
406 > *
407 > * @category Commands
408 > */
409 > export interface SubscriptionDeliveryOptions {
410 > /**
411 > * Maximum time, in milliseconds, that the server may intentionally delay
412 > * delivery while buffering/coalescing updates for this subscription.
413 > *
414 > * A value of `0` requests immediate delivery with no intentional coalescing.
415 > */
416 > maxLatencyMs?: number;
417 > }
418 >
419 > /**
420 > * Result of the `subscribe` command.
421 > *
422 > * `snapshot` is present when the subscribed channel has associated state, and
423 > * absent for stateless channels.
424 > */
425 > export interface SubscribeResult {
426 > /** Snapshot of the subscribed channel's state (omitted for stateless channels) */
427 > snapshot?: Snapshot;
428 > }
429 >
430 > // ─── unsubscribe ─────────────────────────────────────────────────────────────
431 >
432 > /**
433 > * Stop receiving updates for a channel.
434 > *
435 > * @category Commands
436 > * @method unsubscribe
437 > * @direction Client → Server
438 > * @messageType Notification
439 > * @version 1
440 > * @see {@link /specification/subscriptions | Subscriptions}
441 > */
442 > export interface UnsubscribeParams {
443 > /** Channel URI to unsubscribe from */
444 > channel: URI;
445 > }
446 >
447 > // ─── dispatchAction ──────────────────────────────────────────────────────────
448 >
449 > /**
450 > * Fire-and-forget action dispatch (write-ahead). The client applies actions
451 > * optimistically to local state and the server echoes them back as an
452 > * {@link ActionEnvelope} once accepted.
453 > *
454 > * The client → server method is named `dispatchAction`; the server's reply
455 > * arrives on the server → client `action` notification (params:
456 > * {@link ActionEnvelope}).
457 > *
458 > * @category Commands
459 > * @method dispatchAction
460 > * @direction Client → Server
461 > * @messageType Notification
462 > * @version 1
463 > * @see {@link /guide/actions | Actions} for the full list of client-dispatchable actions.
464 > */
465 > export interface DispatchActionParams {
466 > /** Channel URI this action targets */
467 > channel: URI;
468 > /** Client sequence number */
469 > clientSeq: number;
470 > /** The action to dispatch */
471 > action: StateAction;
472 > }
473 >
474 > // ─── resourceRead ────────────────────────────────────────────────────────
475 >
476 > /**
477 > * Encoding of fetched content data.
478 > *
479 > * @category Commands
480 > */
481 > export const enum ContentEncoding {
482 > Base64 = 'base64',
483 > Utf8 = 'utf-8',
484 > }
485 >
486 > /**
487 > * Reads the content of a resource by URI.
488 > *
489 > * Content references keep the state tree small by storing large data (images,
490 > * long tool outputs) by reference rather than inline.
491 > *
492 > * Binary content (images, etc.) MUST use `base64` encoding. Text content MAY
493 > * use `utf-8` encoding.
494 > *
495 > * Like all `resource*` methods, `resourceRead` is symmetrical and MAY be
496 > * sent in either direction. Hosts use it to fetch content from a
497 > * client-published URI (e.g. `virtual://my-client/...` plugins); clients
498 > * use it to read host-side files. The receiver enforces access via the
499 > * same permission/`resourceRequest` flow regardless of which peer initiated.
500 > *
501 > * @category Commands
502 > * @method resourceRead
503 > * @direction Client ↔ Server
504 > * @messageType Request
505 > * @version 1
506 > * @throws `NotFound` (`-32008`) if the URI does not exist.
507 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to read the URI.
508 > * @example
509 > * ```jsonc
510 > * // Client → Server
511 > * { "jsonrpc": "2.0", "id": 10, "method": "resourceRead",
512 > * "params": { "uri": "ahp-session:/<uuid>/content/img-1" } }
513 > *
514 > * // Server → Client
515 > * { "jsonrpc": "2.0", "id": 10, "result": {
516 > * "data": "iVBORw0KGgo...",
517 > * "encoding": "base64",
518 > * "contentType": "image/png"
519 > * }}
520 > * ```
521 > */
522 > export interface ResourceReadParams extends BaseParams {
523 > channel: 'ahp-root://';
524 > /** Content URI from a `ContentRef` */
525 > uri: string;
526 > /** Preferred encoding for the returned data (default: server-chosen) */
527 > encoding?: ContentEncoding;
528 > }
529 >
530 > /**
531 > * Result of the `resourceRead` command.
532 > *
533 > * The server SHOULD honor the `encoding` requested in the params. If the
534 > * server cannot provide the requested encoding, it MUST fall back to either
535 > * `base64` or `utf-8`.
536 > */
537 > export interface ResourceReadResult {
538 > /** Content encoded as a string */
539 > data: string;
540 > /** How `data` is encoded */
541 > encoding: ContentEncoding;
542 > /** Content type (e.g. `"image/png"`, `"text/plain"`) */
543 > contentType?: string;
544 > }
545 >
546 > // ─── resourceWrite ───────────────────────────────────────────────────────────
547 >
548 > /**
549 > * How {@link ResourceWriteParams.data} is placed within the target file.
550 > *
551 > * Each mode interprets {@link ResourceWriteParams.position} differently:
552 > *
553 > * - `truncate` (default): rooted at the **start** of the file. The file is
554 > * truncated at `position` (0 by default) and `data` is written from that
555 > * offset, so the resulting file is `existing[0..position] + data`. With
556 > * `position` omitted this is a full overwrite.
557 > * - `append`: rooted at the **end** of the file. `position` counts bytes
558 > * backwards from EOF, so `position: 0` (the default) writes at EOF —
559 > * POSIX append — and `position: 5` inserts `data` 5 bytes before the
560 > * current EOF, shifting those trailing 5 bytes after the inserted region.
561 > * The server MUST evaluate the effective EOF and write atomically with
562 > * respect to other appenders so concurrent `append` writes do not
563 > * clobber each other.
564 > * - `insert`: rooted at the **start** of the file. `position` (0 by default)
565 > * is the byte offset at which `data` is spliced in; bytes at or after
566 > * `position` are shifted right by `data.length`. `insert` always grows
567 > * the file — use `truncate` to overwrite bytes in place.
568 > *
569 > * @category Commands
570 > */
571 > export const enum ResourceWriteMode {
572 > Truncate = 'truncate',
573 > Append = 'append',
574 > Insert = 'insert',
575 > }
576 >
577 > /**
578 > * Writes content to a file on the server's filesystem.
579 > *
580 > * Binary content (images, etc.) MUST use `base64` encoding. Text content MAY
581 > * use `utf-8` encoding.
582 > *
583 > * If the file does not exist, it is created. If the file already exists, the
584 > * effect on existing bytes depends on {@link ResourceWriteParams.mode}:
585 > * `truncate` (default) overwrites from the chosen offset onward, `append`
586 > * preserves all existing bytes and adds `data` at a position rooted at EOF,
587 > * and `insert` preserves all existing bytes and splices `data` in at an
588 > * offset rooted at the start of the file.
589 > *
590 > * Like all `resource*` methods, `resourceWrite` is symmetrical and MAY be
591 > * sent in either direction.
592 > *
593 > * @category Commands
594 > * @method resourceWrite
595 > * @direction Client ↔ Server
596 > * @messageType Request
597 > * @version 1
598 > * @throws `NotFound` (`-32008`) if the parent directory does not exist.
599 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to write to the path.
600 > * @throws `AlreadyExists` (`-32010`) if `createOnly` is set and the file already exists.
601 > * @throws `Conflict` (`-32011`) if `ifMatch` is set and the current `etag` does not match.
602 > * @example
603 > * ```jsonc
604 > * // Client → Server
605 > * { "jsonrpc": "2.0", "id": 11, "method": "resourceWrite",
606 > * "params": { "uri": "file:///workspace/hello.txt", "data": "SGVsbG8=",
607 > * "encoding": "base64", "contentType": "text/plain" } }
608 > *
609 > * // Server → Client
610 > * { "jsonrpc": "2.0", "id": 11, "result": {} }
611 > * ```
612 > */
613 > export interface ResourceWriteParams extends BaseParams {
614 > channel: 'ahp-root://';
615 > /** Target file URI on the server filesystem */
616 > uri: URI;
617 > /** Content encoded as a string */
618 > data: string;
619 > /** How `data` is encoded */
620 > encoding: ContentEncoding;
621 > /** Content type (e.g. `"text/plain"`, `"image/png"`) */
622 > contentType?: string;
623 > /**
624 > * If `true`, the server MUST fail if the file already exists instead of
625 > * overwriting it. Useful for safe creation of new files.
626 > */
627 > createOnly?: boolean;
628 > /**
629 > * How `data` is placed within the target file. Defaults to `'truncate'`
630 > * (full overwrite) when omitted. See {@link ResourceWriteMode} for the
631 > * meaning of each mode and how it interprets {@link position}.
632 > */
633 > mode?: ResourceWriteMode;
634 > /**
635 > * Byte offset interpreted according to {@link mode}. Defaults to `0`.
636 > * - `truncate`: offset from the start of the file at which to truncate
637 > * before writing.
638 > * - `append`: bytes back from EOF at which to insert `data`.
639 > * - `insert`: offset from the start of the file at which to splice in
640 > * `data`.
641 > */
642 > position?: number;
643 > /**
644 > * Optimistic-concurrency token previously returned by
645 > * {@link ResourceResolveResult.etag}. When set, the server MUST fail with
646 > * `Conflict` if the current `etag` does not match — preventing lost
647 > * updates between a `resourceResolve` and a subsequent `resourceWrite`.
648 > */
649 > ifMatch?: string;
650 > }
651 >
652 > /**
653 > * Result of the `resourceWrite` command.
654 > *
655 > * An empty object on success.
656 > */
657 > export interface ResourceWriteResult {
658 > }
659 >
660 > // ─── resourceList ────────────────────────────────────────────────────────
661 >
662 > /**
663 > * Lists directory entries at a file URI on the server's filesystem.
664 > *
665 > * This is intended for remote folder pickers and similar UI that needs to let
666 > * users navigate the server's local filesystem.
667 > *
668 > * The server MUST return success only if the target exists and is a directory.
669 > * If the target does not exist, is not a directory, or cannot be accessed, the
670 > * server MUST return a JSON-RPC error.
671 > *
672 > * Like all `resource*` methods, `resourceList` is symmetrical and MAY be
673 > * sent in either direction.
674 > *
675 > * @category Commands
676 > * @method resourceList
677 > * @direction Client ↔ Server
678 > * @messageType Request
679 > * @version 1
680 > * @throws `NotFound` (`-32008`) if the directory does not exist.
681 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to browse the directory.
682 > */
683 > export interface ResourceListParams extends BaseParams {
684 > channel: 'ahp-root://';
685 > /** Directory URI on the server filesystem */
686 > uri: URI;
687 > }
688 >
689 > /**
690 > * Directory entry returned by `resourceList`.
691 > */
692 > export interface DirectoryEntry {
693 > /** Base name of the entry */
694 > name: string;
695 > /** Whether the entry is a file or directory */
696 > type: 'file' | 'directory';
697 > }
698 >
699 > /**
700 > * Result of the `resourceList` command.
701 > */
702 > export interface ResourceListResult {
703 > /** Entries directly contained in the requested directory */
704 > entries: DirectoryEntry[];
705 > }
706 >
707 > // ─── resourceCopy ────────────────────────────────────────────────────────────
708 >
709 > /**
710 > * Copies a resource from one URI to another on the server's filesystem.
711 > *
712 > * If the destination already exists, it is overwritten unless `failIfExists`
713 > * is set.
714 > *
715 > * Like all `resource*` methods, `resourceCopy` is symmetrical and MAY be
716 > * sent in either direction.
717 > *
718 > * @category Commands
719 > * @method resourceCopy
720 > * @direction Client ↔ Server
721 > * @messageType Request
722 > * @version 1
723 > * @throws `NotFound` (`-32008`) if the source does not exist.
724 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to read the source or write to the destination.
725 > * @throws `AlreadyExists` (`-32010`) if `failIfExists` is set and the destination already exists.
726 > */
727 > export interface ResourceCopyParams extends BaseParams {
728 > channel: 'ahp-root://';
729 > /** Source URI to copy from */
730 > source: URI;
731 > /** Destination URI to copy to */
732 > destination: URI;
733 > /**
734 > * If `true`, the server MUST fail if the destination already exists instead
735 > * of overwriting it.
736 > */
737 > failIfExists?: boolean;
738 > }
739 >
740 > /**
741 > * Result of the `resourceCopy` command.
742 > *
743 > * An empty object on success.
744 > */
745 > export interface ResourceCopyResult {
746 > }
747 >
748 > // ─── resourceDelete ──────────────────────────────────────────────────────────
749 >
750 > /**
751 > * Deletes a resource at a URI on the server's filesystem.
752 > *
753 > * Like all `resource*` methods, `resourceDelete` is symmetrical and MAY be
754 > * sent in either direction.
755 > *
756 > * @category Commands
757 > * @method resourceDelete
758 > * @direction Client ↔ Server
759 > * @messageType Request
760 > * @version 1
761 > * @throws `NotFound` (`-32008`) if the resource does not exist.
762 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to delete the resource.
763 > */
764 > export interface ResourceDeleteParams extends BaseParams {
765 > channel: 'ahp-root://';
766 > /** URI of the resource to delete */
767 > uri: URI;
768 > /**
769 > * If `true` and the target is a directory, delete it and all its contents
770 > * recursively. If `false` (default), deleting a non-empty directory MUST fail.
771 > */
772 > recursive?: boolean;
773 > }
774 >
775 > /**
776 > * Result of the `resourceDelete` command.
777 > *
778 > * An empty object on success.
779 > */
780 > export interface ResourceDeleteResult {
781 > }
782 >
783 > // ─── resourceRequest ─────────────────────────────────────────────────────────
784 >
785 > /**
786 > * Requests permission to access a resource on the receiver's filesystem.
787 > *
788 > * `resourceRequest` is symmetrical and MAY be sent in either direction: a
789 > * client asks the server to grant access to a server-side resource, or a
790 > * server asks the client to grant access to a client-side resource. The
791 > * receiver decides whether to allow, deny, or prompt the user for the
792 > * requested access.
793 > *
794 > * If the receiver denies access, it MUST respond with `PermissionDenied`
795 > * (-32009). The error data MAY include a `ResourceRequestParams` value
796 > * describing the access the caller would need to be granted for the
797 > * operation to succeed; see `PermissionDeniedErrorData` in
798 > * `types/errors.ts`.
799 > *
800 > * After a successful `resourceRequest`, the caller MAY use the corresponding
801 > * `resource*` commands (e.g. `resourceRead`, `resourceWrite`) to perform the
802 > * operation. Receivers MAY rescind access at any time by returning
803 > * `PermissionDenied` on subsequent operations.
804 > *
805 > * Either `read`, `write`, or both SHOULD be set to `true`. A request with
806 > * neither flag set is treated as `read: true` by receivers.
807 > *
808 > * @category Commands
809 > * @method resourceRequest
810 > * @direction Client ↔ Server
811 > * @messageType Request
812 > * @version 1
813 > * @throws `PermissionDenied` (`-32009`) if access is denied.
814 > */
815 > export interface ResourceRequestParams extends BaseParams {
816 > channel: 'ahp-root://';
817 > /**
818 > * Resource URI being requested. Typically a `file:` URI on the receiver's
819 > * filesystem, but any URI scheme that the receiver mediates access to is
820 > * allowed.
821 > */
822 > uri: URI;
823 > /** Whether the caller needs read access to the resource. */
824 > read?: boolean;
825 > /** Whether the caller needs write access to the resource. */
826 > write?: boolean;
827 > }
828 >
829 > /**
830 > * Result of the `resourceRequest` command.
831 > *
832 > * An empty object on success.
833 > */
834 > export interface ResourceRequestResult {
835 > }
836 >
837 > // ─── resourceMove ────────────────────────────────────────────────────────────
838 >
839 > /**
840 > * Moves (renames) a resource from one URI to another on the server's filesystem.
841 > *
842 > * If the destination already exists, it is overwritten unless `failIfExists`
843 > * is set.
844 > *
845 > * Like all `resource*` methods, `resourceMove` is symmetrical and MAY be
846 > * sent in either direction.
847 > *
848 > * @category Commands
849 > * @method resourceMove
850 > * @direction Client ↔ Server
851 > * @messageType Request
852 > * @version 1
853 > * @throws `NotFound` (`-32008`) if the source does not exist.
854 > * @throws `PermissionDenied` (`-32009`) if the client is not permitted to move the resource.
855 > * @throws `AlreadyExists` (`-32010`) if `failIfExists` is set and the destination already exists.
856 > */
857 > export interface ResourceMoveParams extends BaseParams {
858 > channel: 'ahp-root://';
859 > /** Source URI to move from */
860 > source: URI;
861 > /** Destination URI to move to */
862 > destination: URI;
863 > /**
864 > * If `true`, the server MUST fail if the destination already exists instead
865 > * of overwriting it.
866 > */
867 > failIfExists?: boolean;
868 > }
869 >
870 > /**
871 > * Result of the `resourceMove` command.
872 > *
873 > * An empty object on success.
874 > */
875 > export interface ResourceMoveResult {
876 > }
877 >
878 > // ─── resourceResolve ─────────────────────────────────────────────────────────
879 >
880 > /**
881 > * Discriminant for {@link ResourceResolveResult.type}.
882 > *
883 > * @category Commands
884 > */
885 > export const enum ResourceType {
886 > File = 'file',
887 > Directory = 'directory',
888 > Symlink = 'symlink',
889 > }
890 >
891 > /**
892 > * Resolves a resource — the combination of POSIX `stat` and `realpath`.
893 > *
894 > * `resourceResolve` returns metadata about the resource together with its
895 > * canonical URI after symlink resolution. Use this in place of any
896 > * `resourceExists` shim: a missing resource MUST surface as a `NotFound`
897 > * JSON-RPC error rather than a success with a sentinel value. Callers that
898 > * truly need a boolean check should attempt `resourceResolve` and treat
899 > * `NotFound` as "does not exist".
900 > *
901 > * Like all `resource*` methods, `resourceResolve` is symmetrical and MAY be
902 > * sent in either direction.
903 > *
904 > * @category Commands
905 > * @method resourceResolve
906 > * @direction Client ↔ Server
907 > * @messageType Request
908 > * @version 1
909 > * @throws `NotFound` (`-32008`) if the resource does not exist.
910 > * @throws `PermissionDenied` (`-32009`) if the caller is not permitted to stat the URI.
911 > * @example
912 > * ```jsonc
913 > * // Client → Server
914 > * { "jsonrpc": "2.0", "id": 20, "method": "resourceResolve",
915 > * "params": { "channel": "ahp-root://", "uri": "file:///workspace/hello.txt" } }
916 > *
917 > * // Server → Client
918 > * { "jsonrpc": "2.0", "id": 20, "result": {
919 > * "uri": "file:///workspace/hello.txt",
920 > * "type": "file",
921 > * "size": 5,
922 > * "mtime": "2026-01-15T12:34:56.789Z",
923 > * "etag": "W/\"5-abc123\""
924 > * }}
925 > * ```
926 > */
927 > export interface ResourceResolveParams extends BaseParams {
928 > channel: 'ahp-root://';
929 > /** URI to resolve */
930 > uri: URI;
931 > /**
932 > * When `true` (default), follow symlinks and report the metadata of the
933 > * link target — and set `uri` in the result to the canonical (realpath)
934 > * URI. When `false`, stat the link itself (lstat semantics) and report
935 > * `type: 'symlink'`.
936 > */
937 > followSymlinks?: boolean;
938 > }
939 >
940 > /**
941 > * Result of the `resourceResolve` command.
942 > */
943 > export interface ResourceResolveResult {
944 > /**
945 > * Canonical URI after symlink resolution. Equal to the requested URI when
946 > * `followSymlinks` is `false` or the URI does not traverse a symlink.
947 > */
948 > uri: URI;
949 > /** Resource kind. */
950 > type: ResourceType;
951 > /**
952 > * Size in bytes. Omitted for directories when the provider cannot
953 > * cheaply compute it.
954 > */
955 > size?: number;
956 > /** Last-modified time in ISO 8601 format, when known. */
957 > mtime?: string;
958 > /** Creation time in ISO 8601 format, when known. */
959 > ctime?: string;
960 > /** Sniffed MIME type, when known (e.g. `"text/plain"`, `"image/png"`). */
961 > contentType?: string;
962 > /**
963 > * Opaque per-provider version token. When present, pass it as
964 > * {@link ResourceWriteParams.ifMatch} on a subsequent `resourceWrite` to
965 > * detect concurrent modifications.
966 > */
967 > etag?: string;
968 > }
969 >
970 > // ─── resourceMkdir ───────────────────────────────────────────────────────────
971 >
972 > /**
973 > * Creates a directory on the server's filesystem with `mkdir -p` semantics.
974 > *
975 > * The server MUST create any missing parent directories. Creating a
976 > * directory that already exists is a no-op success. If `uri` already
977 > * exists but is **not** a directory, the server MUST fail with
978 > * `AlreadyExists`.
979 > *
980 > * Like all `resource*` methods, `resourceMkdir` is symmetrical and MAY be
981 > * sent in either direction.
982 > *
983 > * @category Commands
984 > * @method resourceMkdir
985 > * @direction Client ↔ Server
986 > * @messageType Request
987 > * @version 1
988 > * @throws `PermissionDenied` (`-32009`) if the caller is not permitted to create the directory.
989 > * @throws `AlreadyExists` (`-32010`) if `uri` already exists as a non-directory.
990 > */
991 > export interface ResourceMkdirParams extends BaseParams {
992 > channel: 'ahp-root://';
993 > /** Directory URI to create (parents created as needed). */
994 > uri: URI;
995 > }
996 >
997 > /**
998 > * Result of the `resourceMkdir` command.
999 > *
1000 > * An empty object on success.
1001 > */
1002 > export interface ResourceMkdirResult {
1003 > }
1004 >
1005 > // ─── authenticate ────────────────────────────────────────────────────────────
1006 >
1007 > /**
1008 > * Pushes a ****** for a protected resource. The `resource` field MUST
1009 > * match a protected-resource identifier the client has discovered from the
1010 > * server — whether declared statically in `AgentInfo.protectedResources`,
1011 > * or discovered dynamically from a live `McpServerAuthRequiredState.resource`
1012 > * or `ToolCallAuthRequiredState.auth.resource` (both surfaced only once the
1013 > * corresponding MCP server or tool call actually challenges for auth).
1014 > * Servers MUST accept any `resource` value they have themselves advertised
1015 > * through one of these three mechanisms.
1016 > *
1017 > * Tokens are delivered using [RFC 6750](https://datatracker.ietf.org/doc/html/rfc6750)
1018 > * (****** Usage) semantics. The client obtains the token from the
1019 > * authorization server(s) listed in the resource's metadata and pushes it
1020 > * to the server via this command.
1021 > *
1022 > * @category Commands
1023 > * @method authenticate
1024 > * @direction Client → Server
1025 > * @messageType Request
1026 > * @version 1
1027 > * @see {@link /specification/authentication | Authentication}
1028 > * @example
1029 > * ```jsonc
1030 > * // Client → Server
1031 > * { "jsonrpc": "2.0", "id": 3, "method": "authenticate",
1032 > * "params": { "channel": "ahp-root://", "resource": "https://api.github.com", "token": "gho_xxxx" } }
1033 > *
1034 > * // Server → Client (success)
1035 > * { "jsonrpc": "2.0", "id": 3, "result": {} }
1036 > *
1037 > * // Server → Client (failure — invalid token)
1038 > * { "jsonrpc": "2.0", "id": 3, "error": { "code": -32007, "message": "Invalid token" } }
1039 > * ```
1040 > */
1041 > export interface AuthenticateParams extends BaseParams {
1042 > channel: 'ahp-root://';
1043 > /**
1044 > * The protected resource identifier. MUST match a `resource` value the
1045 > * server has advertised — via `ProtectedResourceMetadata` in
1046 > * `AgentInfo.protectedResources`, or via a live
1047 > * `McpServerAuthRequiredState.resource` / `ToolCallAuthRequiredState.auth.resource`.
1048 > */
1049 > resource: string;
1050 > /** ****** obtained from the resource's authorization server */
1051 > token: string;
1052 > /**
1053 > * OAuth scopes the token grants, when known. Lets the server determine
1054 > * whether a specific challenge — e.g. the `requiredScopes` on a live
1055 > * `McpServerAuthRequiredState` or `ToolCallAuthRequiredState.auth` — is
1056 > * satisfied without decoding the (opaque, server-specific) token itself.
1057 > * Omit when the client doesn't track granted scopes separately from the
1058 > * token.
1059 > */
1060 > scopes?: string[];
1061 > }
1062 >
1063 > /**
1064 > * Result of the `authenticate` command.
1065 > *
1066 > * An empty object on success. If the token is invalid or the resource is
1067 > * unrecognized, the server MUST return a JSON-RPC error (e.g. `AuthRequired`
1068 > * `-32007` or `InvalidParams` `-32602`).
1069 > */
1070 > export interface AuthenticateResult {
1071 > }
src/vs/platform/contextkey/common/contextkey.ts 1045 covered LOC · 304 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- contextkey.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from '../../../base/common/charCode.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { isChrome, isEdge, isFirefox, isLinux, isMacintosh, isSafari, isWeb, isWindows } from '../../../base/common/platform.js';
9 > import { isFalsyOrWhitespace } from '../../../base/common/strings.js';
10 > import { Scanner, LexingError, Token, TokenType } from './scanner.js';
11 > import { createDecorator } from '../../instantiation/common/instantiation.js';
12 > import { localize } from '../../../nls.js';
13 > import { IDisposable } from '../../../base/common/lifecycle.js';
14 > import { illegalArgument } from '../../../base/common/errors.js';
15 >
16 > const CONSTANT_VALUES = new Map<string, boolean>();
17 > CONSTANT_VALUES.set('false', false);
18 > CONSTANT_VALUES.set('true', true);
19 > CONSTANT_VALUES.set('isMac', isMacintosh);
20 > CONSTANT_VALUES.set('isLinux', isLinux);
21 > CONSTANT_VALUES.set('isWindows', isWindows);
22 > CONSTANT_VALUES.set('isWeb', isWeb);
23 > CONSTANT_VALUES.set('isMacNative', isMacintosh && !isWeb);
24 > CONSTANT_VALUES.set('isEdge', isEdge);
25 > CONSTANT_VALUES.set('isFirefox', isFirefox);
26 > CONSTANT_VALUES.set('isChrome', isChrome);
27 > CONSTANT_VALUES.set('isSafari', isSafari);
28 >
29 > /** allow register constant context keys that are known only after startup; requires running `substituteConstants` on the context key - https://github.com/microsoft/vscode/issues/174218#issuecomment-1437972127 */
30 > export function setConstant(key: string, value: boolean) {
31 if (CONSTANT_VALUES.get(key) !== undefined) { throw illegalArgument('contextkey.setConstant(k, v) invoked with already set constant `k`'); }
32
33 CONSTANT_VALUES.set(key, value);
34 }
36 > const hasOwnProperty = Object.prototype.hasOwnProperty;
37 >
38 > export const enum ContextKeyExprType {
39 > False = 0,
40 > True = 1,
41 > Defined = 2,
42 > Not = 3,
43 > Equals = 4,
44 > NotEquals = 5,
45 > And = 6,
46 > Regex = 7,
47 > NotRegex = 8,
48 > Or = 9,
49 > In = 10,
50 > NotIn = 11,
51 > Greater = 12,
52 > GreaterEquals = 13,
53 > Smaller = 14,
54 > SmallerEquals = 15,
55 > }
56 >
57 > export interface IContextKeyExprMapper {
58 > mapDefined(key: string): ContextKeyExpression;
59 > mapNot(key: string): ContextKeyExpression;
60 > mapEquals(key: string, value: any): ContextKeyExpression;
61 > mapNotEquals(key: string, value: any): ContextKeyExpression;
62 > mapGreater(key: string, value: any): ContextKeyExpression;
63 > mapGreaterEquals(key: string, value: any): ContextKeyExpression;
64 > mapSmaller(key: string, value: any): ContextKeyExpression;
65 > mapSmallerEquals(key: string, value: any): ContextKeyExpression;
66 > mapRegex(key: string, regexp: RegExp | null): ContextKeyRegexExpr;
67 > mapIn(key: string, valueKey: string): ContextKeyInExpr;
68 > mapNotIn(key: string, valueKey: string): ContextKeyNotInExpr;
69 > }
70 >
71 > export interface IContextKeyExpression {
72 > cmp(other: ContextKeyExpression): number;
73 > equals(other: ContextKeyExpression): boolean;
74 > substituteConstants(): ContextKeyExpression | undefined;
75 > evaluate(context: IContext): boolean;
76 > serialize(): string;
77 > keys(): string[];
78 > map(mapFnc: IContextKeyExprMapper): ContextKeyExpression;
79 > negate(): ContextKeyExpression;
80 >
81 > }
82 >
83 > export type ContextKeyExpression = (
84 > ContextKeyFalseExpr | ContextKeyTrueExpr | ContextKeyDefinedExpr | ContextKeyNotExpr
85 > | ContextKeyEqualsExpr | ContextKeyNotEqualsExpr | ContextKeyRegexExpr
86 > | ContextKeyNotRegexExpr | ContextKeyAndExpr | ContextKeyOrExpr | ContextKeyInExpr
87 > | ContextKeyNotInExpr | ContextKeyGreaterExpr | ContextKeyGreaterEqualsExpr
88 > | ContextKeySmallerExpr | ContextKeySmallerEqualsExpr
89 > );
90 >
91 >
92 > /*
93 >
94 > Syntax grammar:
95 >
96 > ```ebnf
97 >
98 > expression ::= or
99 >
100 > or ::= and { '||' and }*
101 >
102 > and ::= term { '&&' term }*
103 >
104 > term ::=
105 > | '!' (KEY | true | false | parenthesized)
106 > | primary
107 >
108 > primary ::=
109 > | 'true'
110 > | 'false'
111 > | parenthesized
112 > | KEY '=~' REGEX
113 > | KEY [ ('==' | '!=' | '<' | '<=' | '>' | '>=' | 'not' 'in' | 'in') value ]
114 >
115 > parenthesized ::=
116 > | '(' expression ')'
117 >
118 > value ::=
119 > | 'true'
120 > | 'false'
121 > | 'in' // we support `in` as a value because there's an extension that uses it, ie "when": "languageId == in"
122 > | VALUE // matched by the same regex as KEY; consider putting the value in single quotes if it's a string (e.g., with spaces)
123 > | SINGLE_QUOTED_STR
124 > | EMPTY_STR // this allows "when": "foo == " which's used by existing extensions
125 >
126 > ```
127 > */
128 >
129 > export type ParserConfig = {
130 > /**
131 > * with this option enabled, the parser can recover from regex parsing errors, e.g., unescaped slashes: `/src//` is accepted as `/src\//` would be
132 > */
133 > regexParsingWithErrorRecovery: boolean;
134 > };
135 >
136 > const defaultConfig: ParserConfig = {
137 > regexParsingWithErrorRecovery: true
138 > };
139 >
140 > export type ParsingError = {
141 > message: string;
142 > offset: number;
143 > lexeme: string;
144 > additionalInfo?: string;
145 > };
146 >
147 > const errorEmptyString = localize('contextkey.parser.error.emptyString', "Empty context key expression");
148 > const hintEmptyString = localize('contextkey.parser.error.emptyString.hint', "Did you forget to write an expression? You can also put 'false' or 'true' to always evaluate to false or true, respectively.");
149 > const errorNoInAfterNot = localize('contextkey.parser.error.noInAfterNot', "'in' after 'not'.");
150 > const errorClosingParenthesis = localize('contextkey.parser.error.closingParenthesis', "closing parenthesis ')'");
151 > const errorUnexpectedToken = localize('contextkey.parser.error.unexpectedToken', "Unexpected token");
152 > const hintUnexpectedToken = localize('contextkey.parser.error.unexpectedToken.hint', "Did you forget to put && or || before the token?");
153 > const errorUnexpectedEOF = localize('contextkey.parser.error.unexpectedEOF', "Unexpected end of expression");
154 > const hintUnexpectedEOF = localize('contextkey.parser.error.unexpectedEOF.hint', "Did you forget to put a context key?");
155 >
156 > /**
157 > * A parser for context key expressions.
158 > *
159 > * Example:
160 > * ```ts
161 > * const parser = new Parser();
162 > * const expr = parser.parse('foo == "bar" && baz == true');
163 > *
164 > * if (expr === undefined) {
165 > * // there were lexing or parsing errors
166 > * // process lexing errors with `parser.lexingErrors`
167 > * // process parsing errors with `parser.parsingErrors`
168 > * } else {
169 > * // expr is a valid expression
170 > * }
171 > * ```
172 > */
173 > export class Parser {
174 > // Note: this doesn't produce an exact syntax tree but a normalized one
175 > // ContextKeyExpression's that we use as AST nodes do not expose constructors that do not normalize
176 >
177 > private static _parseError = new Error();
178 >
179 > // lifetime note: `_scanner` lives as long as the parser does, i.e., is not reset between calls to `parse`
180 > private readonly _scanner = new Scanner();
181 >
182 > // lifetime note: `_tokens`, `_current`, and `_parsingErrors` must be reset between calls to `parse`
183 > private _tokens: Token[] = [];
184 > private _current = 0; // invariant: 0 <= this._current < this._tokens.length ; any incrementation of this value must first call `_isAtEnd`
185 > private _parsingErrors: ParsingError[] = [];
186 >
187 > get lexingErrors(): Readonly<LexingError[]> {
188 return this._scanner.errors;
189 }
191 > get parsingErrors(): Readonly<ParsingError[]> {
192 return this._parsingErrors;
193 }
195 > constructor(private readonly _config: ParserConfig = defaultConfig) {
196 > }
197 >
198 > /**
199 > * Parse a context key expression.
200 > *
201 > * @param input the expression to parse
202 > * @returns the parsed expression or `undefined` if there's an error - call `lexingErrors` and `parsingErrors` to see the errors
203 > */
204 > parse(input: string): ContextKeyExpression | undefined {
205
206 if (input === '') {
231 }
232 }
234 > private _expr(): ContextKeyExpression | undefined {
235 return this._or();
236 }
238 > private _or(): ContextKeyExpression | undefined {
239 const expr = [this._and()];
240
246 return expr.length === 1 ? expr[0] : ContextKeyExpr.or(...expr);
247 }
249 > private _and(): ContextKeyExpression | undefined {
250 const expr = [this._term()];
251
257 return expr.length === 1 ? expr[0] : ContextKeyExpr.and(...expr);
258 }
260 > private _term(): ContextKeyExpression | undefined {
261 if (this._matchOne(TokenType.Neg)) {
262 const peek = this._peek();
283 return this._primary();
284 }
286 > private _primary(): ContextKeyExpression | undefined {
287
288 const peek = this._peek();
498 }
499 }
501 > private _value(): string {
502 const token = this._peek();
503 switch (token.type) {
521 }
522 }
524 > private _flagsGYRe = /g|y/g;
525 > private _removeFlagsGY(flags: string): string {
526 return flags.replaceAll(this._flagsGYRe, '');
527 }
529 > // careful: this can throw if current token is the initial one (ie index = 0)
530 > private _previous() {
531 return this._tokens[this._current - 1];
532 }
534 > private _matchOne(token: TokenType) {
535 if (this._check(token)) {
536 this._advance();
540 return false;
541 }
543 > private _advance() {
544 if (!this._isAtEnd()) {
545 this._current++;
547 return this._previous();
548 }
550 > private _consume(type: TokenType, message: string) {
551 if (this._check(type)) {
552 return this._advance();
555 throw this._errExpectedButGot(message, this._peek());
556 }
558 > private _errExpectedButGot(expected: string, got: Token, additionalInfo?: string) {
559 const message = localize('contextkey.parser.error.expectedButGot', "Expected: {0}\nReceived: '{1}'.", expected, Scanner.getLexeme(got));
560 const offset = got.offset;
563 return Parser._parseError;
564 }
566 > private _check(type: TokenType) {
567 return this._peek().type === type;
568 }
570 > private _peek() {
571 return this._tokens[this._current];
572 }
574 > private _isAtEnd() {
575 return this._peek().type === TokenType.EOF;
576 }
577 > } contextkey.ts
578 >
579 > export abstract class ContextKeyExpr {
580 >
581 > public static false(): ContextKeyExpression {
582 return ContextKeyFalseExpr.INSTANCE;
583 }
584 > public static true(): ContextKeyExpression { contextkey.ts
585 return ContextKeyTrueExpr.INSTANCE;
586 }
587 > public static has(key: string): ContextKeyExpression { contextkey.ts
588 > return ContextKeyDefinedExpr.create(key); contextkey.ts
589 > }
590 > public static equals(key: string, value: any): ContextKeyExpression { contextkey.ts
591 return ContextKeyEqualsExpr.create(key, value);
592 }
593 > public static notEquals(key: string, value: any): ContextKeyExpression { contextkey.ts
594 return ContextKeyNotEqualsExpr.create(key, value);
595 }
596 > public static regex(key: string, value: RegExp): ContextKeyExpression { contextkey.ts
597 > return ContextKeyRegexExpr.create(key, value); contextkey.ts
598 > }
599 > public static in(key: string, value: string): ContextKeyExpression { contextkey.ts
600 return ContextKeyInExpr.create(key, value);
601 }
602 > public static notIn(key: string, value: string): ContextKeyExpression { contextkey.ts
603 return ContextKeyNotInExpr.create(key, value);
604 }
605 > public static not(key: string): ContextKeyExpression { contextkey.ts
606 return ContextKeyNotExpr.create(key);
607 }
608 > public static and(...expr: Array<ContextKeyExpression | undefined | null>): ContextKeyExpression | undefined { contextkey.ts
609 > return ContextKeyAndExpr.create(expr, null, true); contextkey.ts
610 > }
611 > public static or(...expr: Array<ContextKeyExpression | undefined | null>): ContextKeyExpression | undefined { contextkey.ts
612 > return ContextKeyOrExpr.create(expr, null, true); contextkey.ts
613 > }
614 > public static greater(key: string, value: number): ContextKeyExpression { contextkey.ts
615 return ContextKeyGreaterExpr.create(key, value);
616 }
617 > public static greaterEquals(key: string, value: number): ContextKeyExpression { contextkey.ts
618 return ContextKeyGreaterEqualsExpr.create(key, value);
619 }
620 > public static smaller(key: string, value: number): ContextKeyExpression { contextkey.ts
621 return ContextKeySmallerExpr.create(key, value);
622 }
623 > public static smallerEquals(key: string, value: number): ContextKeyExpression { contextkey.ts
624 return ContextKeySmallerEqualsExpr.create(key, value);
625 }
627 > private static _parser = new Parser({ regexParsingWithErrorRecovery: false });
628 > public static deserialize(serialized: string | null | undefined): ContextKeyExpression | undefined {
629 if (serialized === undefined || serialized === null) { // an empty string needs to be handled by the parser to get a corresponding parsing error reported
630 return undefined;
634 return expr;
635 }
637 > }
638 >
639 >
640 > export function validateWhenClauses(whenClauses: string[]): any {
641
642 const parser = new Parser({ regexParsingWithErrorRecovery: false }); // we run with no recovery to guide users to use correct regexes
664 });
665 }
667 > export function expressionsAreEqualWithConstantSubstitution(a: ContextKeyExpression | null | undefined, b: ContextKeyExpression | null | undefined): boolean {
668 const aExpr = a ? a.substituteConstants() : undefined;
669 const bExpr = b ? b.substituteConstants() : undefined;
676 return aExpr.equals(bExpr);
677 }
679 > function cmp(a: ContextKeyExpression, b: ContextKeyExpression): number { contextkey.ts
680 > return a.cmp(b);
681 > }
683 > export class ContextKeyFalseExpr implements IContextKeyExpression {
684 > public static INSTANCE = new ContextKeyFalseExpr();
685 >
686 > public readonly type = ContextKeyExprType.False;
687 >
688 > protected constructor() {
689 > }
690 >
691 > public cmp(other: ContextKeyExpression): number {
692 return this.type - other.type;
693 }
695 > public equals(other: ContextKeyExpression): boolean {
696 return (other.type === this.type);
697 }
699 > public substituteConstants(): ContextKeyExpression | undefined {
700 return this;
701 }
703 > public evaluate(context: IContext): boolean {
704 return false;
705 }
707 > public serialize(): string {
708 return 'false';
709 }
711 > public keys(): string[] {
712 return [];
713 }
715 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
716 return this;
717 }
719 > public negate(): ContextKeyExpression {
720 return ContextKeyTrueExpr.INSTANCE;
721 }
722 > } contextkey.ts
723 >
724 > export class ContextKeyTrueExpr implements IContextKeyExpression {
725 > public static INSTANCE = new ContextKeyTrueExpr();
726 >
727 > public readonly type = ContextKeyExprType.True;
728 >
729 > protected constructor() {
730 > }
731 >
732 > public cmp(other: ContextKeyExpression): number {
733 return this.type - other.type;
734 }
736 > public equals(other: ContextKeyExpression): boolean {
737 return (other.type === this.type);
738 }
740 > public substituteConstants(): ContextKeyExpression | undefined {
741 return this;
742 }
744 > public evaluate(context: IContext): boolean {
745 return true;
746 }
748 > public serialize(): string {
749 return 'true';
750 }
752 > public keys(): string[] {
753 return [];
754 }
756 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
757 return this;
758 }
760 > public negate(): ContextKeyExpression {
761 return ContextKeyFalseExpr.INSTANCE;
762 }
763 > } contextkey.ts
764 >
765 > export class ContextKeyDefinedExpr implements IContextKeyExpression {
766 > public static create(key: string, negated: ContextKeyExpression | null = null): ContextKeyExpression {
767 > const constantValue = CONSTANT_VALUES.get(key);
768 > if (typeof constantValue === 'boolean') {
769 return constantValue ? ContextKeyTrueExpr.INSTANCE : ContextKeyFalseExpr.INSTANCE;
770 }
771 > return new ContextKeyDefinedExpr(key, negated); contextkey.ts
772 > }
773 >
774 > public readonly type = ContextKeyExprType.Defined;
775 >
776 > protected constructor(
777 > readonly key: string, contextkey.ts
778 > private negated: ContextKeyExpression | null
779 > ) {
780 > }
782 > public cmp(other: ContextKeyExpression): number {
783 > if (other.type !== this.type) { contextkey.ts
784 > return this.type - other.type; contextkey.ts
785 > }
786 return cmp1(this.key, other.key);
787 > } contextkey.ts
789 > public equals(other: ContextKeyExpression): boolean {
790 > if (other.type === this.type) { contextkey.ts
791 return (this.key === other.key);
792 }
793 > return false; contextkey.ts
794 > } contextkey.ts
796 > public substituteConstants(): ContextKeyExpression | undefined {
797 const constantValue = CONSTANT_VALUES.get(this.key);
798 if (typeof constantValue === 'boolean') {
801 return this;
802 }
804 > public evaluate(context: IContext): boolean {
805 return (!!context.getValue(this.key));
806 }
808 > public serialize(): string {
809 return this.key;
810 }
812 > public keys(): string[] {
813 return [this.key];
814 }
816 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
817 return mapFnc.mapDefined(this.key);
818 }
820 > public negate(): ContextKeyExpression {
821 > if (!this.negated) { contextkey.ts
822 > this.negated = ContextKeyNotExpr.create(this.key, this);
823 > }
824 > return this.negated;
825 > }
826 > } contextkey.ts
827 >
828 > export class ContextKeyEqualsExpr implements IContextKeyExpression {
829 >
830 > public static create(key: string, value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
831 > if (typeof value === 'boolean') {
832 > return (value ? ContextKeyDefinedExpr.create(key, negated) : ContextKeyNotExpr.create(key, negated)); contextkey.ts
833 > }
834 > const constantValue = CONSTANT_VALUES.get(key); contextkey.ts
835 > if (typeof constantValue === 'boolean') {
836 const trueValue = constantValue ? 'true' : 'false';
837 return (value === trueValue ? ContextKeyTrueExpr.INSTANCE : ContextKeyFalseExpr.INSTANCE);
838 }
839 > return new ContextKeyEqualsExpr(key, value, negated); contextkey.ts
840 > } contextkey.ts
841 >
842 > public readonly type = ContextKeyExprType.Equals;
843 >
844 > private constructor(
845 > private readonly key: string, contextkey.ts
846 > private readonly value: any,
847 > private negated: ContextKeyExpression | null
848 > ) {
849 > }
851 > public cmp(other: ContextKeyExpression): number {
852 > if (other.type !== this.type) { contextkey.ts
853 return this.type - other.type;
854 }
855 > return cmp2(this.key, this.value, other.key, other.value); contextkey.ts
856 > } contextkey.ts
858 > public equals(other: ContextKeyExpression): boolean {
859 > if (other.type === this.type) { contextkey.ts
860 > return (this.key === other.key && this.value === other.value); contextkey.ts
861 > }
862 return false;
863 > } contextkey.ts
865 > public substituteConstants(): ContextKeyExpression | undefined {
866 const constantValue = CONSTANT_VALUES.get(this.key);
867 if (typeof constantValue === 'boolean') {
871 return this;
872 }
874 > public evaluate(context: IContext): boolean {
875 // Intentional ==
876 // eslint-disable-next-line eqeqeq
877 return (context.getValue(this.key) == this.value);
878 }
880 > public serialize(): string {
881 return `${this.key} == '${this.value}'`;
882 }
884 > public keys(): string[] {
885 return [this.key];
886 }
888 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
889 return mapFnc.mapEquals(this.key, this.value);
890 }
892 > public negate(): ContextKeyExpression {
893 > if (!this.negated) { contextkey.ts
894 > this.negated = ContextKeyNotEqualsExpr.create(this.key, this.value, this);
895 > }
896 > return this.negated;
897 > }
898 > } contextkey.ts
899 >
900 > export class ContextKeyInExpr implements IContextKeyExpression {
901 >
902 > public static create(key: string, valueKey: string): ContextKeyInExpr {
903 > return new ContextKeyInExpr(key, valueKey);
904 > }
905 >
906 > public readonly type = ContextKeyExprType.In;
907 > private negated: ContextKeyExpression | null = null;
908 >
909 > private constructor(
910 private readonly key: string,
911 private readonly valueKey: string,
912 ) {
913 }
915 > public cmp(other: ContextKeyExpression): number {
916 if (other.type !== this.type) {
917 return this.type - other.type;
919 return cmp2(this.key, this.valueKey, other.key, other.valueKey);
920 }
922 > public equals(other: ContextKeyExpression): boolean {
923 if (other.type === this.type) {
924 return (this.key === other.key && this.valueKey === other.valueKey);
926 return false;
927 }
929 > public substituteConstants(): ContextKeyExpression | undefined {
930 return this;
931 }
933 > public evaluate(context: IContext): boolean {
934 const source = context.getValue(this.valueKey);
935
964 return false;
965 }
967 > public serialize(): string {
968 return `${this.key} in '${this.valueKey}'`;
969 }
971 > public keys(): string[] {
972 return [this.key, this.valueKey];
973 }
975 > public map(mapFnc: IContextKeyExprMapper): ContextKeyInExpr {
976 return mapFnc.mapIn(this.key, this.valueKey);
977 }
979 > public negate(): ContextKeyExpression {
980 if (!this.negated) {
981 this.negated = ContextKeyNotInExpr.create(this.key, this.valueKey);
983 return this.negated;
984 }
985 > } contextkey.ts
986 >
987 > export class ContextKeyNotInExpr implements IContextKeyExpression {
988 >
989 > public static create(key: string, valueKey: string): ContextKeyNotInExpr {
990 > return new ContextKeyNotInExpr(key, valueKey);
991 > }
992 >
993 > public readonly type = ContextKeyExprType.NotIn;
994 >
995 > private readonly _negated: ContextKeyInExpr;
996 >
997 > private constructor(
998 private readonly key: string,
999 private readonly valueKey: string,
1001 this._negated = ContextKeyInExpr.create(key, valueKey);
1002 }
1003 > contextkey.ts
1004 > public cmp(other: ContextKeyExpression): number {
1005 if (other.type !== this.type) {
1006 return this.type - other.type;
1008 return this._negated.cmp(other._negated);
1009 }
1010 > contextkey.ts
1011 > public equals(other: ContextKeyExpression): boolean {
1012 if (other.type === this.type) {
1013 return this._negated.equals(other._negated);
1015 return false;
1016 }
1017 > contextkey.ts
1018 > public substituteConstants(): ContextKeyExpression | undefined {
1019 return this;
1020 }
1021 > contextkey.ts
1022 > public evaluate(context: IContext): boolean {
1023 return !this._negated.evaluate(context);
1024 }
1025 > contextkey.ts
1026 > public serialize(): string {
1027 return `${this.key} not in '${this.valueKey}'`;
1028 }
1029 > contextkey.ts
1030 > public keys(): string[] {
1031 return this._negated.keys();
1032 }
1033 > contextkey.ts
1034 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1035 return mapFnc.mapNotIn(this.key, this.valueKey);
1036 }
1037 > contextkey.ts
1038 > public negate(): ContextKeyExpression {
1039 return this._negated;
1040 }
1041 > } contextkey.ts
1042 >
1043 > export class ContextKeyNotEqualsExpr implements IContextKeyExpression {
1044 >
1045 > public static create(key: string, value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1046 > if (typeof value === 'boolean') {
1047 if (value) {
1048 return ContextKeyNotExpr.create(key, negated);
1050 return ContextKeyDefinedExpr.create(key, negated);
1051 }
1052 > const constantValue = CONSTANT_VALUES.get(key); contextkey.ts
1053 > if (typeof constantValue === 'boolean') {
1054 const falseValue = constantValue ? 'true' : 'false';
1055 return (value === falseValue ? ContextKeyFalseExpr.INSTANCE : ContextKeyTrueExpr.INSTANCE);
1056 }
1057 > return new ContextKeyNotEqualsExpr(key, value, negated); contextkey.ts
1058 > } contextkey.ts
1059 >
1060 > public readonly type = ContextKeyExprType.NotEquals;
1061 >
1062 > private constructor(
1063 > private readonly key: string, contextkey.ts
1064 > private readonly value: any,
1065 > private negated: ContextKeyExpression | null
1066 > ) {
1067 > }
1068 > contextkey.ts
1069 > public cmp(other: ContextKeyExpression): number {
1070 if (other.type !== this.type) {
1071 return this.type - other.type;
1073 return cmp2(this.key, this.value, other.key, other.value);
1074 }
1075 > contextkey.ts
1076 > public equals(other: ContextKeyExpression): boolean {
1077 > if (other.type === this.type) { contextkey.ts
1078 return (this.key === other.key && this.value === other.value);
1079 }
1080 > return false; contextkey.ts
1081 > }
1082 > contextkey.ts
1083 > public substituteConstants(): ContextKeyExpression | undefined {
1084 const constantValue = CONSTANT_VALUES.get(this.key);
1085 if (typeof constantValue === 'boolean') {
1089 return this;
1090 }
1091 > contextkey.ts
1092 > public evaluate(context: IContext): boolean {
1093 // Intentional !=
1094 // eslint-disable-next-line eqeqeq
1095 return (context.getValue(this.key) != this.value);
1096 }
1097 > contextkey.ts
1098 > public serialize(): string {
1099 return `${this.key} != '${this.value}'`;
1100 }
1101 > contextkey.ts
1102 > public keys(): string[] {
1103 return [this.key];
1104 }
1105 > contextkey.ts
1106 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1107 return mapFnc.mapNotEquals(this.key, this.value);
1108 }
1109 > contextkey.ts
1110 > public negate(): ContextKeyExpression {
1111 if (!this.negated) {
1112 this.negated = ContextKeyEqualsExpr.create(this.key, this.value, this);
1114 return this.negated;
1115 }
1116 > } contextkey.ts
1117 >
1118 > export class ContextKeyNotExpr implements IContextKeyExpression {
1119 >
1120 > public static create(key: string, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1121 > const constantValue = CONSTANT_VALUES.get(key);
1122 > if (typeof constantValue === 'boolean') {
1123 > return (constantValue ? ContextKeyFalseExpr.INSTANCE : ContextKeyTrueExpr.INSTANCE); contextkey.ts
1124 > }
1125 > return new ContextKeyNotExpr(key, negated); contextkey.ts
1126 > }
1127 >
1128 > public readonly type = ContextKeyExprType.Not;
1129 >
1130 > private constructor(
1131 > private readonly key: string, contextkey.ts
1132 > private negated: ContextKeyExpression | null
1133 > ) {
1134 > }
1135 > contextkey.ts
1136 > public cmp(other: ContextKeyExpression): number {
1137 > if (other.type !== this.type) { contextkey.ts
1138 return this.type - other.type;
1139 }
1140 > return cmp1(this.key, other.key); contextkey.ts
1141 > } contextkey.ts
1142 > contextkey.ts
1143 > public equals(other: ContextKeyExpression): boolean {
1144 > if (other.type === this.type) { contextkey.ts
1145 > return (this.key === other.key); contextkey.ts
1146 > }
1147 return false;
1148 > } contextkey.ts
1149 > contextkey.ts
1150 > public substituteConstants(): ContextKeyExpression | undefined {
1151 const constantValue = CONSTANT_VALUES.get(this.key);
1152 if (typeof constantValue === 'boolean') {
1155 return this;
1156 }
1157 > contextkey.ts
1158 > public evaluate(context: IContext): boolean {
1159 return (!context.getValue(this.key));
1160 }
1161 > contextkey.ts
1162 > public serialize(): string {
1163 return `!${this.key}`;
1164 }
1165 > contextkey.ts
1166 > public keys(): string[] {
1167 return [this.key];
1168 }
1169 > contextkey.ts
1170 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1171 return mapFnc.mapNot(this.key);
1172 }
1173 > contextkey.ts
1174 > public negate(): ContextKeyExpression {
1175 > if (!this.negated) { contextkey.ts
1176 this.negated = ContextKeyDefinedExpr.create(this.key, this);
1177 }
1178 > return this.negated; contextkey.ts
1179 > }
1180 > } contextkey.ts
1181 >
1182 function withFloatOrStr<T extends ContextKeyExpression>(value: any, callback: (value: number | string) => T): T | ContextKeyFalseExpr {
1183 if (typeof value === 'string') {
1192 return ContextKeyFalseExpr.INSTANCE;
1193 }
1194 > contextkey.ts
1195 > export class ContextKeyGreaterExpr implements IContextKeyExpression {
1196 >
1197 > public static create(key: string, _value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1198 > return withFloatOrStr(_value, (value) => new ContextKeyGreaterExpr(key, value, negated));
1199 > }
1200 >
1201 > public readonly type = ContextKeyExprType.Greater;
1202 >
1203 > private constructor(
1204 private readonly key: string,
1205 private readonly value: number | string,
1206 private negated: ContextKeyExpression | null
1207 ) { }
1208 > contextkey.ts
1209 > public cmp(other: ContextKeyExpression): number {
1210 if (other.type !== this.type) {
1211 return this.type - other.type;
1213 return cmp2(this.key, this.value, other.key, other.value);
1214 }
1215 > contextkey.ts
1216 > public equals(other: ContextKeyExpression): boolean {
1217 if (other.type === this.type) {
1218 return (this.key === other.key && this.value === other.value);
1220 return false;
1221 }
1222 > contextkey.ts
1223 > public substituteConstants(): ContextKeyExpression | undefined {
1224 return this;
1225 }
1226 > contextkey.ts
1227 > public evaluate(context: IContext): boolean {
1228 if (typeof this.value === 'string') {
1229 return false;
1231 return (parseFloat(context.getValue<any>(this.key)) > this.value);
1232 }
1233 > contextkey.ts
1234 > public serialize(): string {
1235 return `${this.key} > ${this.value}`;
1236 }
1237 > contextkey.ts
1238 > public keys(): string[] {
1239 return [this.key];
1240 }
1241 > contextkey.ts
1242 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1243 return mapFnc.mapGreater(this.key, this.value);
1244 }
1245 > contextkey.ts
1246 > public negate(): ContextKeyExpression {
1247 if (!this.negated) {
1248 this.negated = ContextKeySmallerEqualsExpr.create(this.key, this.value, this);
1250 return this.negated;
1251 }
1252 > } contextkey.ts
1253 >
1254 > export class ContextKeyGreaterEqualsExpr implements IContextKeyExpression {
1255 >
1256 > public static create(key: string, _value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1257 > return withFloatOrStr(_value, (value) => new ContextKeyGreaterEqualsExpr(key, value, negated));
1258 > }
1259 >
1260 > public readonly type = ContextKeyExprType.GreaterEquals;
1261 >
1262 > private constructor(
1263 private readonly key: string,
1264 private readonly value: number | string,
1265 private negated: ContextKeyExpression | null
1266 ) { }
1267 > contextkey.ts
1268 > public cmp(other: ContextKeyExpression): number {
1269 if (other.type !== this.type) {
1270 return this.type - other.type;
1272 return cmp2(this.key, this.value, other.key, other.value);
1273 }
1274 > contextkey.ts
1275 > public equals(other: ContextKeyExpression): boolean {
1276 if (other.type === this.type) {
1277 return (this.key === other.key && this.value === other.value);
1279 return false;
1280 }
1281 > contextkey.ts
1282 > public substituteConstants(): ContextKeyExpression | undefined {
1283 return this;
1284 }
1285 > contextkey.ts
1286 > public evaluate(context: IContext): boolean {
1287 if (typeof this.value === 'string') {
1288 return false;
1290 return (parseFloat(context.getValue<any>(this.key)) >= this.value);
1291 }
1292 > contextkey.ts
1293 > public serialize(): string {
1294 return `${this.key} >= ${this.value}`;
1295 }
1296 > contextkey.ts
1297 > public keys(): string[] {
1298 return [this.key];
1299 }
1300 > contextkey.ts
1301 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1302 return mapFnc.mapGreaterEquals(this.key, this.value);
1303 }
1304 > contextkey.ts
1305 > public negate(): ContextKeyExpression {
1306 if (!this.negated) {
1307 this.negated = ContextKeySmallerExpr.create(this.key, this.value, this);
1309 return this.negated;
1310 }
1311 > } contextkey.ts
1312 >
1313 > export class ContextKeySmallerExpr implements IContextKeyExpression {
1314 >
1315 > public static create(key: string, _value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1316 > return withFloatOrStr(_value, (value) => new ContextKeySmallerExpr(key, value, negated));
1317 > }
1318 >
1319 > public readonly type = ContextKeyExprType.Smaller;
1320 >
1321 > private constructor(
1322 private readonly key: string,
1323 private readonly value: number | string,
1325 ) {
1326 }
1327 > contextkey.ts
1328 > public cmp(other: ContextKeyExpression): number {
1329 if (other.type !== this.type) {
1330 return this.type - other.type;
1332 return cmp2(this.key, this.value, other.key, other.value);
1333 }
1334 > contextkey.ts
1335 > public equals(other: ContextKeyExpression): boolean {
1336 if (other.type === this.type) {
1337 return (this.key === other.key && this.value === other.value);
1339 return false;
1340 }
1341 > contextkey.ts
1342 > public substituteConstants(): ContextKeyExpression | undefined {
1343 return this;
1344 }
1345 > contextkey.ts
1346 > public evaluate(context: IContext): boolean {
1347 if (typeof this.value === 'string') {
1348 return false;
1350 return (parseFloat(context.getValue<any>(this.key)) < this.value);
1351 }
1352 > contextkey.ts
1353 > public serialize(): string {
1354 return `${this.key} < ${this.value}`;
1355 }
1356 > contextkey.ts
1357 > public keys(): string[] {
1358 return [this.key];
1359 }
1360 > contextkey.ts
1361 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1362 return mapFnc.mapSmaller(this.key, this.value);
1363 }
1364 > contextkey.ts
1365 > public negate(): ContextKeyExpression {
1366 if (!this.negated) {
1367 this.negated = ContextKeyGreaterEqualsExpr.create(this.key, this.value, this);
1369 return this.negated;
1370 }
1371 > } contextkey.ts
1372 >
1373 > export class ContextKeySmallerEqualsExpr implements IContextKeyExpression {
1374 >
1375 > public static create(key: string, _value: any, negated: ContextKeyExpression | null = null): ContextKeyExpression {
1376 > return withFloatOrStr(_value, (value) => new ContextKeySmallerEqualsExpr(key, value, negated));
1377 > }
1378 >
1379 > public readonly type = ContextKeyExprType.SmallerEquals;
1380 >
1381 > private constructor(
1382 private readonly key: string,
1383 private readonly value: number | string,
1385 ) {
1386 }
1387 > contextkey.ts
1388 > public cmp(other: ContextKeyExpression): number {
1389 if (other.type !== this.type) {
1390 return this.type - other.type;
1392 return cmp2(this.key, this.value, other.key, other.value);
1393 }
1394 > contextkey.ts
1395 > public equals(other: ContextKeyExpression): boolean {
1396 if (other.type === this.type) {
1397 return (this.key === other.key && this.value === other.value);
1399 return false;
1400 }
1401 > contextkey.ts
1402 > public substituteConstants(): ContextKeyExpression | undefined {
1403 return this;
1404 }
1405 > contextkey.ts
1406 > public evaluate(context: IContext): boolean {
1407 if (typeof this.value === 'string') {
1408 return false;
1410 return (parseFloat(context.getValue<any>(this.key)) <= this.value);
1411 }
1412 > contextkey.ts
1413 > public serialize(): string {
1414 return `${this.key} <= ${this.value}`;
1415 }
1416 > contextkey.ts
1417 > public keys(): string[] {
1418 return [this.key];
1419 }
1420 > contextkey.ts
1421 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1422 return mapFnc.mapSmallerEquals(this.key, this.value);
1423 }
1424 > contextkey.ts
1425 > public negate(): ContextKeyExpression {
1426 if (!this.negated) {
1427 this.negated = ContextKeyGreaterExpr.create(this.key, this.value, this);
1429 return this.negated;
1430 }
1431 > } contextkey.ts
1432 >
1433 > export class ContextKeyRegexExpr implements IContextKeyExpression {
1434 >
1435 > public static create(key: string, regexp: RegExp | null): ContextKeyRegexExpr {
1436 > return new ContextKeyRegexExpr(key, regexp);
1437 > }
1438 >
1439 > public readonly type = ContextKeyExprType.Regex;
1440 > private negated: ContextKeyExpression | null = null;
1441 >
1442 > private constructor(
1443 > private readonly key: string, contextkey.ts
1444 > private readonly regexp: RegExp | null
1445 > ) {
1446 > //
1447 > }
1448 > contextkey.ts
1449 > public cmp(other: ContextKeyExpression): number {
1450 > if (other.type !== this.type) { contextkey.ts
1451 return this.type - other.type;
1452 }
1453 > if (this.key < other.key) { contextkey.ts
1454 return -1;
1455 }
1456 > if (this.key > other.key) { contextkey.ts
1457 return 1;
1458 }
1459 > const thisSource = this.regexp ? this.regexp.source : ''; contextkey.ts
1460 > const otherSource = other.regexp ? other.regexp.source : '';
1461 > if (thisSource < otherSource) {
1462 return -1;
1463 }
1464 > if (thisSource > otherSource) { contextkey.ts
1465 > return 1;
1466 > }
1467 return 0;
1468 > } contextkey.ts
1469 > contextkey.ts
1470 > public equals(other: ContextKeyExpression): boolean {
1471 > if (other.type === this.type) { contextkey.ts
1472 > const thisSource = this.regexp ? this.regexp.source : '';
1473 > const otherSource = other.regexp ? other.regexp.source : '';
1474 > return (this.key === other.key && thisSource === otherSource);
1475 > }
1476 return false;
1477 > } contextkey.ts
1478 > contextkey.ts
1479 > public substituteConstants(): ContextKeyExpression | undefined {
1480 return this;
1481 }
1482 > contextkey.ts
1483 > public evaluate(context: IContext): boolean {
1484 const value = context.getValue<any>(this.key);
1485 return this.regexp ? this.regexp.test(value) : false;
1486 }
1487 > contextkey.ts
1488 > public serialize(): string {
1489 const value = this.regexp
1490 ? `/${this.regexp.source}/${this.regexp.flags}`
1492 return `${this.key} =~ ${value}`;
1493 }
1494 > contextkey.ts
1495 > public keys(): string[] {
1496 return [this.key];
1497 }
1498 > contextkey.ts
1499 > public map(mapFnc: IContextKeyExprMapper): ContextKeyRegexExpr {
1500 return mapFnc.mapRegex(this.key, this.regexp);
1501 }
1502 > contextkey.ts
1503 > public negate(): ContextKeyExpression {
1504 > if (!this.negated) { contextkey.ts
1505 > this.negated = ContextKeyNotRegexExpr.create(this);
1506 > }
1507 > return this.negated;
1508 > }
1509 > } contextkey.ts
1510 >
1511 > export class ContextKeyNotRegexExpr implements IContextKeyExpression {
1512 >
1513 > public static create(actual: ContextKeyRegexExpr): ContextKeyExpression {
1514 > return new ContextKeyNotRegexExpr(actual);
1515 > }
1516 >
1517 > public readonly type = ContextKeyExprType.NotRegex;
1518 >
1519 > private constructor(private readonly _actual: ContextKeyRegexExpr) {
1520 > // contextkey.ts
1521 > }
1522 > contextkey.ts
1523 > public cmp(other: ContextKeyExpression): number {
1524 if (other.type !== this.type) {
1525 return this.type - other.type;
1527 return this._actual.cmp(other._actual);
1528 }
1529 > contextkey.ts
1530 > public equals(other: ContextKeyExpression): boolean {
1531 > if (other.type === this.type) { contextkey.ts
1532 return this._actual.equals(other._actual);
1533 }
1534 > return false; contextkey.ts
1535 > }
1536 > contextkey.ts
1537 > public substituteConstants(): ContextKeyExpression | undefined {
1538 return this;
1539 }
1540 > contextkey.ts
1541 > public evaluate(context: IContext): boolean {
1542 return !this._actual.evaluate(context);
1543 }
1544 > contextkey.ts
1545 > public serialize(): string {
1546 return `!(${this._actual.serialize()})`;
1547 }
1548 > contextkey.ts
1549 > public keys(): string[] {
1550 return this._actual.keys();
1551 }
1552 > contextkey.ts
1553 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1554 return new ContextKeyNotRegexExpr(this._actual.map(mapFnc));
1555 }
1556 > contextkey.ts
1557 > public negate(): ContextKeyExpression {
1558 return this._actual;
1559 }
1560 > } contextkey.ts
1561 >
1562 > /**
1563 > * @returns the same instance if nothing changed.
1564 > */
1565 function eliminateConstantsInArray(arr: ContextKeyExpression[]): (ContextKeyExpression | undefined)[] {
1566 // Allocate array only if there is a difference
1591 return newArr;
1592 }
1593 > contextkey.ts
1594 > export class ContextKeyAndExpr implements IContextKeyExpression {
1595 >
1596 > public static create(_expr: ReadonlyArray<ContextKeyExpression | null | undefined>, negated: ContextKeyExpression | null, extraRedundantCheck: boolean): ContextKeyExpression | undefined {
1597 > return ContextKeyAndExpr._normalizeArr(_expr, negated, extraRedundantCheck);
1598 > }
1599 >
1600 > public readonly type = ContextKeyExprType.And;
1601 >
1602 > private constructor(
1603 > public readonly expr: ContextKeyExpression[], contextkey.ts
1604 > private negated: ContextKeyExpression | null
1605 > ) {
1606 > }
1607 > contextkey.ts
1608 > public cmp(other: ContextKeyExpression): number {
1609 if (other.type !== this.type) {
1610 return this.type - other.type;
1624 return 0;
1625 }
1626 > contextkey.ts
1627 > public equals(other: ContextKeyExpression): boolean {
1628 if (other.type === this.type) {
1629 if (this.expr.length !== other.expr.length) {
1639 return false;
1640 }
1641 > contextkey.ts
1642 > public substituteConstants(): ContextKeyExpression | undefined {
1643 const exprArr = eliminateConstantsInArray(this.expr);
1644 if (exprArr === this.expr) {
1648 return ContextKeyAndExpr.create(exprArr, this.negated, false);
1649 }
1650 > contextkey.ts
1651 > public evaluate(context: IContext): boolean {
1652 for (let i = 0, len = this.expr.length; i < len; i++) {
1653 if (!this.expr[i].evaluate(context)) {
1657 return true;
1658 }
1659 > contextkey.ts
1660 > private static _normalizeArr(arr: ReadonlyArray<ContextKeyExpression | null | undefined>, negated: ContextKeyExpression | null, extraRedundantCheck: boolean): ContextKeyExpression | undefined {
1661 > const expr: ContextKeyExpression[] = []; contextkey.ts
1662 > let hasTrue = false;
1663 >
1664 > for (const e of arr) {
1665 > if (!e) {
1666 continue;
1667 }
1668 > contextkey.ts
1669 > if (e.type === ContextKeyExprType.True) {
1670 // anything && true ==> anything
1671 hasTrue = true;
1672 continue;
1673 }
1674 > contextkey.ts
1675 > if (e.type === ContextKeyExprType.False) {
1676 // anything && false ==> false
1677 return ContextKeyFalseExpr.INSTANCE;
1678 }
1679 > contextkey.ts
1680 > if (e.type === ContextKeyExprType.And) {
1681 expr.push(...e.expr);
1682 continue;
1683 }
1684 > contextkey.ts
1685 > expr.push(e);
1686 > }
1687 >
1688 > if (expr.length === 0 && hasTrue) {
1689 return ContextKeyTrueExpr.INSTANCE;
1690 }
1691 > contextkey.ts
1692 > if (expr.length === 0) {
1693 return undefined;
1694 }
1695 > contextkey.ts
1696 > if (expr.length === 1) {
1697 return expr[0];
1698 }
1699 > contextkey.ts
1700 > expr.sort(cmp);
1701 >
1702 > // eliminate duplicate terms
1703 > for (let i = 1; i < expr.length; i++) {
1704 > if (expr[i - 1].equals(expr[i])) {
1705 expr.splice(i, 1);
1706 i--;
1707 }
1708 > } contextkey.ts
1709 >
1710 > if (expr.length === 1) {
1711 return expr[0];
1712 }
1713 > contextkey.ts
1714 > // We must distribute any OR expression because we don't support parens
1715 > // OR extensions will be at the end (due to sorting rules)
1716 > while (expr.length > 1) {
1717 > const lastElement = expr[expr.length - 1];
1718 > if (lastElement.type !== ContextKeyExprType.Or) {
1719 > break;
1720 > }
1721 // pop the last element
1722 expr.pop();
1738 expr.sort(cmp);
1739 }
1740 > } contextkey.ts
1741 >
1742 > if (expr.length === 1) {
1743 return expr[0];
1744 }
1745 > contextkey.ts
1746 > // resolve false AND expressions
1747 > if (extraRedundantCheck) {
1748 > for (let i = 0; i < expr.length; i++) {
1749 > for (let j = i + 1; j < expr.length; j++) {
1750 > if (expr[i].negate().equals(expr[j])) {
1751 // A && !A case
1752 return ContextKeyFalseExpr.INSTANCE;
1753 }
1754 > } contextkey.ts
1755 > } contextkey.ts
1756 >
1757 > if (expr.length === 1) {
1758 return expr[0];
1759 }
1760 > } contextkey.ts
1761 > contextkey.ts
1762 > return new ContextKeyAndExpr(expr, negated);
1763 > } contextkey.ts
1764 > contextkey.ts
1765 > public serialize(): string {
1766 return this.expr.map(e => e.serialize()).join(' && ');
1767 }
1768 > contextkey.ts
1769 > public keys(): string[] {
1770 const result: string[] = [];
1771 for (const expr of this.expr) {
1774 return result;
1775 }
1776 > contextkey.ts
1777 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1778 return new ContextKeyAndExpr(this.expr.map(expr => expr.map(mapFnc)), null);
1779 }
1780 > contextkey.ts
1781 > public negate(): ContextKeyExpression {
1782 if (!this.negated) {
1783 const result: ContextKeyExpression[] = [];
1789 return this.negated;
1790 }
1791 > } contextkey.ts
1792 >
1793 > export class ContextKeyOrExpr implements IContextKeyExpression {
1794 >
1795 > public static create(_expr: ReadonlyArray<ContextKeyExpression | null | undefined>, negated: ContextKeyExpression | null, extraRedundantCheck: boolean): ContextKeyExpression | undefined {
1796 > return ContextKeyOrExpr._normalizeArr(_expr, negated, extraRedundantCheck);
1797 > }
1798 >
1799 > public readonly type = ContextKeyExprType.Or;
1800 >
1801 > private constructor(
1802 > public readonly expr: ContextKeyExpression[], contextkey.ts
1803 > private negated: ContextKeyExpression | null
1804 > ) {
1805 > }
1806 > contextkey.ts
1807 > public cmp(other: ContextKeyExpression): number {
1808 if (other.type !== this.type) {
1809 return this.type - other.type;
1823 return 0;
1824 }
1825 > contextkey.ts
1826 > public equals(other: ContextKeyExpression): boolean {
1827 if (other.type === this.type) {
1828 if (this.expr.length !== other.expr.length) {
1838 return false;
1839 }
1840 > contextkey.ts
1841 > public substituteConstants(): ContextKeyExpression | undefined {
1842 const exprArr = eliminateConstantsInArray(this.expr);
1843 if (exprArr === this.expr) {
1847 return ContextKeyOrExpr.create(exprArr, this.negated, false);
1848 }
1849 > contextkey.ts
1850 > public evaluate(context: IContext): boolean {
1851 for (let i = 0, len = this.expr.length; i < len; i++) {
1852 if (this.expr[i].evaluate(context)) {
1856 return false;
1857 }
1858 > contextkey.ts
1859 > private static _normalizeArr(arr: ReadonlyArray<ContextKeyExpression | null | undefined>, negated: ContextKeyExpression | null, extraRedundantCheck: boolean): ContextKeyExpression | undefined {
1860 > let expr: ContextKeyExpression[] = []; contextkey.ts
1861 > let hasFalse = false;
1862 >
1863 > if (arr) {
1864 > for (let i = 0, len = arr.length; i < len; i++) {
1865 > const e = arr[i];
1866 > if (!e) {
1867 continue;
1868 }
1869 > contextkey.ts
1870 > if (e.type === ContextKeyExprType.False) {
1871 // anything || false ==> anything
1872 hasFalse = true;
1873 continue;
1874 }
1875 > contextkey.ts
1876 > if (e.type === ContextKeyExprType.True) {
1877 > // anything || true ==> true contextkey.ts
1878 > return ContextKeyTrueExpr.INSTANCE;
1879 > }
1880 > contextkey.ts
1881 > if (e.type === ContextKeyExprType.Or) {
1882 expr = expr.concat(e.expr);
1883 continue;
1884 }
1885 > contextkey.ts
1886 > expr.push(e);
1887 > }
1888 >
1889 > if (expr.length === 0 && hasFalse) {
1890 return ContextKeyFalseExpr.INSTANCE;
1891 }
1892 > contextkey.ts
1893 > expr.sort(cmp);
1894 > }
1895 >
1896 > if (expr.length === 0) {
1897 return undefined;
1898 }
1899 > contextkey.ts
1900 > if (expr.length === 1) {
1901 return expr[0];
1902 }
1903 > contextkey.ts
1904 > // eliminate duplicate terms
1905 > for (let i = 1; i < expr.length; i++) {
1906 > if (expr[i - 1].equals(expr[i])) {
1907 expr.splice(i, 1);
1908 i--;
1909 }
1910 > } contextkey.ts
1911 >
1912 > if (expr.length === 1) {
1913 return expr[0];
1914 }
1915 > contextkey.ts
1916 > // resolve true OR expressions
1917 > if (extraRedundantCheck) {
1918 > for (let i = 0; i < expr.length; i++) {
1919 > for (let j = i + 1; j < expr.length; j++) {
1920 > if (expr[i].negate().equals(expr[j])) {
1921 // A || !A case
1922 return ContextKeyTrueExpr.INSTANCE;
1923 }
1924 > } contextkey.ts
1925 > } contextkey.ts
1926 >
1927 > if (expr.length === 1) {
1928 return expr[0];
1929 }
1930 > } contextkey.ts
1931 > contextkey.ts
1932 > return new ContextKeyOrExpr(expr, negated);
1933 > } contextkey.ts
1934 > contextkey.ts
1935 > public serialize(): string {
1936 return this.expr.map(e => e.serialize()).join(' || ');
1937 }
1938 > contextkey.ts
1939 > public keys(): string[] {
1940 const result: string[] = [];
1941 for (const expr of this.expr) {
1944 return result;
1945 }
1946 > contextkey.ts
1947 > public map(mapFnc: IContextKeyExprMapper): ContextKeyExpression {
1948 return new ContextKeyOrExpr(this.expr.map(expr => expr.map(mapFnc)), null);
1949 }
1950 > contextkey.ts
1951 > public negate(): ContextKeyExpression {
1952 if (!this.negated) {
1953 const result: ContextKeyExpression[] = [];
1976 return this.negated;
1977 }
1978 > } contextkey.ts
1979 >
1980 > export interface ContextKeyInfo {
1981 > readonly key: string;
1982 > readonly type?: string;
1983 > readonly description?: string;
1984 > }
1985 >
1986 > export class RawContextKey<T extends ContextKeyValue> extends ContextKeyDefinedExpr {
1987 >
1988 > private static _info: ContextKeyInfo[] = [];
1989 >
1990 > static all(): IterableIterator<ContextKeyInfo> {
1991 return RawContextKey._info.values();
1992 }
1993 > contextkey.ts
1994 > private readonly _defaultValue: T | undefined;
1995 >
1996 > constructor(key: string, defaultValue: T | undefined, metaOrHide?: string | true | { type: string; description: string }) {
1997 > super(key, null); contextkey.ts
1998 > this._defaultValue = defaultValue;
1999 >
2000 > // collect all context keys into a central place
2001 > if (typeof metaOrHide === 'object') {
2002 > RawContextKey._info.push({ ...metaOrHide, key }); contextkey.ts
2003 > } else if (metaOrHide !== true) { contextkey.ts
2004 > RawContextKey._info.push({ key, description: metaOrHide, type: defaultValue !== null && defaultValue !== undefined ? typeof defaultValue : undefined }); contextkey.ts
2005 > }
2006 > } contextkey.ts
2007 > contextkey.ts
2008 > public bindTo(target: IContextKeyService): IContextKey<T> {
2009 > return target.createKey(this.key, this._defaultValue); contextkey.ts
2010 > }
2011 > contextkey.ts
2012 > public getValue(target: IContextKeyService): T | undefined {
2013 return target.getContextKeyValue<T>(this.key);
2014 }
2015 > contextkey.ts
2016 > public toNegated(): ContextKeyExpression {
2017 > return this.negate(); contextkey.ts
2018 > }
2019 > contextkey.ts
2020 > public isEqualTo(value: any): ContextKeyExpression {
2021 > return ContextKeyEqualsExpr.create(this.key, value); contextkey.ts
2022 > }
2023 > contextkey.ts
2024 > public notEqualsTo(value: any): ContextKeyExpression {
2025 > return ContextKeyNotEqualsExpr.create(this.key, value); contextkey.ts
2026 > }
2027 > contextkey.ts
2028 > public greater(value: any): ContextKeyExpression {
2029 return ContextKeyGreaterExpr.create(this.key, value);
2030 }
2031 > } contextkey.ts
2032 >
2033 > export type ContextKeyValue = null | undefined | boolean | number | string
2034 > | Array<null | undefined | boolean | number | string>
2035 > | Record<string, null | undefined | boolean | number | string>;
2036 >
2037 > export interface IContext {
2038 > getValue<T extends ContextKeyValue = ContextKeyValue>(key: string): T | undefined;
2039 > }
2040 >
2041 > export interface IContextKey<T extends ContextKeyValue = ContextKeyValue> {
2042 > set(value: T): void;
2043 > reset(): void;
2044 > get(): T | undefined;
2045 > }
2046 >
2047 > export interface IContextKeyServiceTarget {
2048 > parentElement: IContextKeyServiceTarget | null;
2049 > setAttribute(attr: string, value: string): void;
2050 > removeAttribute(attr: string): void;
2051 > hasAttribute(attr: string): boolean;
2052 > getAttribute(attr: string): string | null;
2053 > }
2054 >
2055 > export const IContextKeyService = createDecorator<IContextKeyService>('contextKeyService');
2056 >
2057 > export interface IReadableSet<T> {
2058 > has(value: T): boolean;
2059 > }
2060 >
2061 > export interface IContextKeyChangeEvent {
2062 > affectsSome(keys: IReadableSet<string>): boolean;
2063 > allKeysContainedIn(keys: IReadableSet<string>): boolean;
2064 > }
2065 >
2066 > export type IScopedContextKeyService = IContextKeyService & IDisposable;
2067 >
2068 > export interface IContextKeyService {
2069 > readonly _serviceBrand: undefined;
2070 >
2071 > readonly onDidChangeContext: Event<IContextKeyChangeEvent>;
2072 > bufferChangeEvents(callback: Function): void;
2073 >
2074 > createKey<T extends ContextKeyValue>(key: string, defaultValue: T | undefined): IContextKey<T>;
2075 > contextMatchesRules(rules: ContextKeyExpression | undefined): boolean;
2076 > getContextKeyValue<T>(key: string): T | undefined;
2077 >
2078 > createScoped(target: IContextKeyServiceTarget): IScopedContextKeyService;
2079 > createOverlay(overlay: Iterable<[string, any]>): IContextKeyService;
2080 > getContext(target: IContextKeyServiceTarget | null): IContext;
2081 >
2082 > updateParent(parentContextKeyService: IContextKeyService): void;
2083 > }
2084 >
2085 > function cmp1(key1: string, key2: string): number { contextkey.ts
2086 > if (key1 < key2) {
2087 > return -1; contextkey.ts
2088 > }
2089 > if (key1 > key2) { contextkey.ts
2090 > return 1;
2091 > }
2092 return 0;
2093 }
2094 > contextkey.ts
2095 > function cmp2(key1: string, value1: any, key2: string, value2: any): number { contextkey.ts
2096 > if (key1 < key2) {
2097 return -1;
2098 }
2099 > if (key1 > key2) { contextkey.ts
2100 return 1;
2101 }
2102 > if (value1 < value2) { contextkey.ts
2103 > return -1; contextkey.ts
2104 > }
2105 if (value1 > value2) {
2106 return 1;
2108 return 0;
2109 }
2110 > contextkey.ts
2111 > /**
2112 > * Returns true if it is provable `p` implies `q`.
2113 > */
2114 > export function implies(p: ContextKeyExpression, q: ContextKeyExpression): boolean {
2115
2116 if (p.type === ContextKeyExprType.False || q.type === ContextKeyExprType.True) {
2152 return p.equals(q);
2153 }
2154 > contextkey.ts
2155 > /**
2156 > * Returns true if all elements in `p` are also present in `q`.
2157 > * The two arrays are assumed to be sorted
2158 > */
2159 function allElementsIncluded(p: ContextKeyExpression[], q: ContextKeyExpression[]): boolean {
2160 let pIndex = 0;
2175 return (pIndex === p.length);
2176 }
2177 > contextkey.ts
2178 function getTerminals(node: ContextKeyExpression) {
2179 if (node.type === ContextKeyExprType.Or) {
src/vs/workbench/contrib/chat/common/languageModels.ts 986 covered LOC · 93 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- languageModels.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { SequencerByKey, timeout } from '../../../../base/common/async.js';
7 > import { VSBuffer } from '../../../../base/common/buffer.js';
8 > import { CancellationToken } from '../../../../base/common/cancellation.js';
9 > import { IStringDictionary } from '../../../../base/common/collections.js';
10 > import { CancellationError, getErrorMessage, isCancellationError } from '../../../../base/common/errors.js';
11 > import { Emitter, Event } from '../../../../base/common/event.js';
12 > import { hash } from '../../../../base/common/hash.js';
13 > import { Iterable } from '../../../../base/common/iterator.js';
14 > import { IJSONSchema, TypeFromJsonSchema } from '../../../../base/common/jsonSchema.js';
15 > import { DisposableStore, IDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
16 > import { IObservable, observableValue } from '../../../../base/common/observable.js';
17 > import { equals } from '../../../../base/common/objects.js';
18 > import Severity from '../../../../base/common/severity.js';
19 > import { format, isFalsyOrWhitespace } from '../../../../base/common/strings.js';
20 > import { ThemeIcon } from '../../../../base/common/themables.js';
21 > import { IAction, SubmenuAction } from '../../../../base/common/actions.js';
22 > import { isObject, isString } from '../../../../base/common/types.js';
23 > import { Schemas } from '../../../../base/common/network.js';
24 > import { URI } from '../../../../base/common/uri.js';
25 > import { generateUuid } from '../../../../base/common/uuid.js';
26 > import { localize } from '../../../../nls.js';
27 > import { ContextKeyExpr, IContextKey, IContextKeyService } from '../../../../platform/contextkey/common/contextkey.js';
28 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
29 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
30 > import { ILogService } from '../../../../platform/log/common/log.js';
31 > import { INotificationService, NeverShowAgainScope } from '../../../../platform/notification/common/notification.js';
32 > import { IOpenerService } from '../../../../platform/opener/common/opener.js';
33 > import { IProductService } from '../../../../platform/product/common/productService.js';
34 > import { asJson, IRequestService } from '../../../../platform/request/common/request.js';
35 > import { IQuickInputService, IQuickPickItem, QuickInputHideReason } from '../../../../platform/quickinput/common/quickInput.js';
36 > import { ISecretStorageService } from '../../../../platform/secrets/common/secrets.js';
37 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
38 > import { ITelemetryService } from '../../../../platform/telemetry/common/telemetry.js';
39 > import { IExtensionService } from '../../../services/extensions/common/extensions.js';
40 > import { ExtensionsRegistry } from '../../../services/extensions/common/extensionsRegistry.js';
41 > import { ChatContextKeys } from './actions/chatContextKeys.js';
42 > import { ChatAgentLocation } from './constants.js';
43 > import { ILanguageModelsProviderGroup, ILanguageModelsConfigurationService } from './languageModelsConfiguration.js';
44 >
45 > /**
46 > * Vendor id used for the built-in GitHub Copilot language model provider. Treated as the default
47 > * vendor across the chat stack (see `ILanguageModelProviderDescriptor.isDefault`).
48 > */
49 > export const COPILOT_VENDOR_ID = 'copilot';
50 >
51 > /** Whether a missing model is conclusively absent from a vendor's live model list. Empty Copilot results remain transient while token-backed discovery completes. */
52 > export function isLanguageModelVendorAbsenceConclusive(vendor: string, hasLiveModels: boolean, hasResolved: boolean): boolean {
53 return hasLiveModels || (hasResolved && vendor !== COPILOT_VENDOR_ID);
54 }
56 > /**
57 > * Vendor ids of the BYOK language-model providers that ship in-built with the GitHub Copilot Chat
58 > * extension. Each provider's vendor id is `providerName.toLowerCase()` (see
59 > * `extensions/copilot/src/extension/byok/vscode-node/*Provider.ts`). This list is intentionally
60 > * hardcoded: the in-built provider set is stable and known ahead of time, which lets us report these
61 > * providers by name while bucketing every other (third-party) provider as `3p-extension`.
62 > */
63 > const BUILT_IN_BYOK_VENDOR_IDS = new Set<string>([
64 > 'openai',
65 > 'anthropic',
66 > 'gemini',
67 > 'ollama',
68 > 'openrouter',
69 > 'azure',
70 > 'xai',
71 > 'customoai',
72 > 'customendpoint',
73 > ]);
74 >
75 > /**
76 > * Bucket reported for any non-Copilot provider that is not an in-built BYOK provider, i.e. a model
77 > * contributed by a third-party extension. We never report the third-party vendor id directly to avoid
78 > * logging potentially identifying values.
79 > */
80 > export const THIRD_PARTY_PROVIDER_TELEMETRY_NAME = '3p-extension';
81 >
82 > const BUILT_IN_BYOK_EXTENSION_IDS = [
83 > 'github.copilot-chat',
84 > 'github.copilot',
85 > ];
86 >
87 > /**
88 > * Normalizes a non-Copilot model vendor into a non-identifying provider name suitable for telemetry:
89 > * the in-built BYOK vendor id (e.g. `openai`, `ollama`) when contributed by the built-in Copilot
90 > * extensions, or {@link THIRD_PARTY_PROVIDER_TELEMETRY_NAME} otherwise. Returns `undefined` for the
91 > * first-party Copilot vendor (or no vendor) so callers skip logging first-party usage.
92 > */
93 > export function getByokProviderTelemetryName(vendor: string | undefined, extension: ExtensionIdentifier | undefined): string | undefined {
94 if (!vendor || vendor === COPILOT_VENDOR_ID) {
95 return undefined;
100 return THIRD_PARTY_PROVIDER_TELEMETRY_NAME;
101 }
103 > export const enum ChatMessageRole {
104 > System,
105 > User,
106 > Assistant,
107 > }
108 >
109 > export enum LanguageModelPartAudience {
110 > Assistant = 0,
111 > User = 1,
112 > Extension = 2,
113 > }
114 >
115 > export interface IChatMessageTextPart {
116 > type: 'text';
117 > value: string;
118 > audience?: LanguageModelPartAudience[];
119 > }
120 >
121 > export interface IChatMessageImagePart {
122 > type: 'image_url';
123 > value: IChatImageURLPart;
124 > }
125 >
126 > export interface IChatMessageThinkingPart {
127 > type: 'thinking';
128 > value: string | string[];
129 > id?: string;
130 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
131 > metadata?: { readonly [key: string]: any };
132 > }
133 >
134 > export interface IChatMessageDataPart {
135 > type: 'data';
136 > mimeType: string;
137 > data: VSBuffer;
138 > audience?: LanguageModelPartAudience[];
139 > }
140 >
141 > export interface IChatImageURLPart {
142 > /**
143 > * The image's MIME type (e.g., "image/png", "image/jpeg").
144 > */
145 > mimeType: ChatImageMimeType;
146 >
147 > /**
148 > * The raw binary data of the image, encoded as a Uint8Array. Note: do not use base64 encoding. Maximum image size is 5MB.
149 > */
150 > data: VSBuffer;
151 > }
152 >
153 > /**
154 > * Enum for supported image MIME types.
155 > */
156 > export enum ChatImageMimeType {
157 > PNG = 'image/png',
158 > JPEG = 'image/jpeg',
159 > GIF = 'image/gif',
160 > WEBP = 'image/webp',
161 > BMP = 'image/bmp',
162 > }
163 >
164 > /**
165 > * Specifies the detail level of the image.
166 > */
167 > export enum ImageDetailLevel {
168 > Low = 'low',
169 > High = 'high'
170 > }
171 >
172 >
173 > export interface IChatMessageToolResultPart {
174 > type: 'tool_result';
175 > toolCallId: string;
176 > value: (IChatResponseTextPart | IChatResponsePromptTsxPart | IChatResponseDataPart)[];
177 > isError?: boolean;
178 > }
179 >
180 > export type IChatMessagePart = IChatMessageTextPart | IChatMessageToolResultPart | IChatResponseToolUsePart | IChatMessageImagePart | IChatMessageDataPart | IChatMessageThinkingPart;
181 >
182 > export interface IChatMessage {
183 > readonly name?: string | undefined;
184 > readonly role: ChatMessageRole;
185 > readonly content: IChatMessagePart[];
186 > }
187 >
188 > export interface IChatResponseTextPart {
189 > type: 'text';
190 > value: string;
191 > audience?: LanguageModelPartAudience[];
192 > }
193 >
194 > export interface IChatResponsePromptTsxPart {
195 > type: 'prompt_tsx';
196 > value: unknown;
197 > }
198 >
199 > export interface IChatResponseDataPart {
200 > type: 'data';
201 > mimeType: string;
202 > data: VSBuffer;
203 > audience?: LanguageModelPartAudience[];
204 > }
205 >
206 > export interface IChatResponseToolUsePart {
207 > type: 'tool_use';
208 > name: string;
209 > toolCallId: string;
210 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
211 > parameters: any;
212 > }
213 >
214 > export interface IChatResponseThinkingPart {
215 > type: 'thinking';
216 > value: string | string[];
217 > id?: string;
218 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
219 > metadata?: { readonly [key: string]: any };
220 > }
221 >
222 > export interface IChatResponsePullRequestPart {
223 > type: 'pullRequest';
224 > uri: URI;
225 > title: string;
226 > description: string;
227 > author: string;
228 > linkTag: string;
229 > }
230 >
231 > export type IChatResponsePart = IChatResponseTextPart | IChatResponseToolUsePart | IChatResponseDataPart | IChatResponseThinkingPart;
232 >
233 > export type IExtendedChatResponsePart = IChatResponsePullRequestPart;
234 >
235 > export interface ILanguageModelConfigurationSchema extends IJSONSchema {
236 > properties?: {
237 > [key: string]: IJSONSchema & {
238 > /** When set to `'navigation'`, the property is shown as a primary action in the model picker. */
239 > group?: string;
240 > /** Labels for enum values. If provided, these are shown instead of the raw enum values. */
241 > enumItemLabels?: string[];
242 > };
243 > };
244 > }
245 >
246 > export interface ILanguageModelChatMetadata {
247 > readonly extension: ExtensionIdentifier;
248 >
249 > readonly name: string;
250 > readonly id: string;
251 > readonly vendor: string;
252 > readonly version: string;
253 > readonly tooltip?: string;
254 > readonly detail?: string;
255 > readonly multiplierNumeric?: number;
256 > readonly isBYOK?: boolean;
257 > readonly pricing?: string;
258 > readonly inputCost?: number;
259 > readonly cacheCost?: number;
260 > readonly cacheWriteCost?: number;
261 > readonly outputCost?: number;
262 > readonly longContextInputCost?: number;
263 > readonly longContextCacheCost?: number;
264 > readonly longContextCacheWriteCost?: number;
265 > readonly longContextOutputCost?: number;
266 > readonly priceCategory?: string;
267 > readonly category?: string;
268 > readonly family: string;
269 > readonly maxInputTokens: number;
270 > readonly maxOutputTokens: number;
271 >
272 > readonly isDefaultForLocation: { [K in ChatAgentLocation]?: boolean };
273 > readonly isUserSelectable?: boolean;
274 > readonly statusIcon?: ThemeIcon;
275 > readonly auth?: {
276 > readonly providerLabel: string;
277 > readonly accountLabel?: string;
278 > };
279 > readonly capabilities?: {
280 > readonly vision?: boolean;
281 > readonly toolCalling?: boolean;
282 > readonly agentMode?: boolean;
283 > readonly editTools?: ReadonlyArray<string>;
284 > };
285 > /**
286 > * When set, this model is only shown in the model picker for the specified chat session type.
287 > * Models with this property are excluded from the general model picker and only appear
288 > * when the user is in a session matching this type.
289 > */
290 > readonly targetChatSessionType?: string;
291 > /**
292 > * Optional grouping hint for the model picker. When set, the picker buckets this model
293 > * under a sub-group within its vendor, identified by this vendor id — e.g. agent-host models,
294 > * which all share one vendor, grouped by their upstream provider — instead of a single
295 > * vendor-wide bucket. The display name is resolved from the vendor registry
296 > * ({@link ILanguageModelsService.getVendors}), the same source used for every other vendor.
297 > * Presentation-only; it does not affect model selection or routing.
298 > */
299 > readonly modelGroup?: { readonly id: string };
300 > /**
301 > * For an agent-host copy of an extension-provided BYOK model, the identifier the
302 > * original model is registered under in the renderer's LM service
303 > * (`toModelIdentifier(vendor, group, id)` — `<vendor>/<group>/<id>` or `<vendor>/<id>`).
304 > * This is exactly the id the "Manage Models" view keys visibility by; it is carried
305 > * across the agent-host bridge and surfaced here so the model picker can honour the
306 > * model's visibility toggle. Absent for native agent-host models and non-agent-host
307 > * models.
308 > */
309 > readonly byokModelIdentifier?: string;
310 > /**
311 > * An optional JSON schema describing the per-model configuration options.
312 > * Used to validate user-provided per-model configuration in `chatLanguageModels.json`.
313 > */
314 > readonly configurationSchema?: ILanguageModelConfigurationSchema;
315 > /**
316 > * Optional warning text to display in the model picker hover as a warning banner.
317 > * The keys are warning categories (e.g. "data_retention") and the values are markdown strings.
318 > */
319 > readonly warningText?: IStringDictionary<string>;
320 > /**
321 > * Optional promotional information for this model. Positive discounts surface
322 > * promotional UI; non-positive discounts only feature the model in the picker.
323 > */
324 > readonly promo?: {
325 > readonly id: string;
326 > readonly discountPercent: number;
327 > readonly endsAt: string;
328 > readonly message: string;
329 > };
330 > }
331 >
332 > export namespace ILanguageModelChatMetadata {
333 > export function suitableForAgentMode(metadata: ILanguageModelChatMetadata): boolean {
334 const supportsToolsAgent = typeof metadata.capabilities?.agentMode === 'undefined' || metadata.capabilities.agentMode;
335 return supportsToolsAgent && !!metadata.capabilities?.toolCalling;
336 }
338 > export function asQualifiedName(metadata: ILanguageModelChatMetadata): string {
339 return `${metadata.name} (${metadata.vendor})`;
340 }
342 > export function matchesQualifiedName(name: string, metadata: ILanguageModelChatMetadata): boolean {
343 if (metadata.vendor === COPILOT_VENDOR_ID && name === metadata.name) {
344 return true;
346 return name === asQualifiedName(metadata);
347 }
349 > export function hasPromoDiscount(metadata: ILanguageModelChatMetadata): metadata is ILanguageModelChatMetadata & { readonly promo: NonNullable<ILanguageModelChatMetadata['promo']> } {
350 return !!metadata.promo && metadata.promo.discountPercent > 0;
351 }
353 > /**
354 > * Documentation link explaining how Auto model selection works.
355 > * NOTE: Also defined in extensions/copilot/src/extension/conversation/common/languageModelAccess.ts — keep in sync.
356 > */
357 > export const autoModelSelectionDocsUrl = 'https://docs.github.com/en/copilot/concepts/models/auto-model-selection';
358 >
359 > /**
360 > * Builds the shared description shown for the Auto model, rendered as Markdown
361 > * (it contains a "Learn More" link). The discount sentence is only included
362 > * when a positive discount is provided.
363 > *
364 > * @param discountPercent Whole-number percentage (e.g. `10` for 10%). When
365 > * omitted or not positive, the discount sentence is left out entirely.
366 > */
367 > export function getAutoModelDescription(discountPercent?: number): string {
368 const base = localize('autoModel.description', "Auto routes based on your task and real-time system health and model performance.");
369 const learnMore = localize('autoModel.learnMore', "[Learn More]({0})", autoModelSelectionDocsUrl);
374 return `${base} ${learnMore}`;
375 }
377 > /**
378 > * The "Manage Models" identifier that an agent-host copy of an extension-provided
379 > * BYOK model is toggled under, or `undefined` when the model is not such a copy.
380 > *
381 > * Agent-host BYOK models make a round trip that rewrites their id (the node agent host
382 > * re-advertises the extension model under the agent-host vendor). Their original LM
383 > * service identifier — `toModelIdentifier(vendor, group, id)`, i.e. `<vendor>/<group>/<id>`
384 > * or `<vendor>/<id>`, which is what the Manage Models view stores when hiding the model —
385 > * is carried across the bridge and surfaced on {@link ILanguageModelChatMetadata.byokModelIdentifier}.
386 > * This returns it, so callers can match the copy against the user's visibility toggles.
387 > *
388 > * Returns `undefined` for models that are not agent-host BYOK copies (native harness
389 > * models and non-agent-host models), which are matched by their own identifier instead.
390 > */
391 > export function getAgentHostByokManageModelsIdentifier(metadata: ILanguageModelChatMetadata): string | undefined {
392 return metadata.byokModelIdentifier;
393 }
395 >
396 > export interface ILanguageModelChatResponse {
397 > stream: AsyncIterable<IChatResponsePart | IChatResponsePart[]>;
398 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
399 > result: Promise<any>;
400 > }
401 >
402 export async function getTextResponseFromStream(response: ILanguageModelChatResponse): Promise<string> {
403 let responseText = '';
429 }
430 }
432 > export interface ILanguageModelChatProvider {
433 > readonly onDidChange: Event<void>;
434 > provideLanguageModelChatInfo(options: ILanguageModelChatInfoOptions, token: CancellationToken): Promise<ILanguageModelChatMetadataAndIdentifier[]>;
435 > sendChatRequest(modelId: string, messages: IChatMessage[], from: ExtensionIdentifier | undefined, options: ILanguageModelChatRequestOptions, token: CancellationToken): Promise<ILanguageModelChatResponse>;
436 > provideTokenCount(modelId: string, message: string | IChatMessage, token: CancellationToken): Promise<number>;
437 > }
438 >
439 > export interface ILanguageModelChat {
440 > metadata: ILanguageModelChatMetadata;
441 > sendChatRequest(messages: IChatMessage[], from: ExtensionIdentifier | undefined, options: ILanguageModelChatRequestOptions, token: CancellationToken): Promise<ILanguageModelChatResponse>;
442 > provideTokenCount(message: string | IChatMessage, token: CancellationToken): Promise<number>;
443 > }
444 >
445 > export interface ILanguageModelChatSelector {
446 > readonly name?: string;
447 > readonly id?: string;
448 > readonly vendor?: string;
449 > readonly version?: string;
450 > readonly family?: string;
451 > readonly tokens?: number;
452 > readonly extension?: ExtensionIdentifier;
453 > }
454 >
455 >
456 > export function isILanguageModelChatSelector(value: unknown): value is ILanguageModelChatSelector {
457 if (typeof value !== 'object' || value === null) {
458 return false;
469 );
470 }
472 > export const ILanguageModelsService = createDecorator<ILanguageModelsService>('ILanguageModelsService');
473 >
474 > export interface ILanguageModelChatMetadataAndIdentifier {
475 > metadata: ILanguageModelChatMetadata;
476 > identifier: string;
477 > }
478 >
479 > export interface ILanguageModelChatInfoOptions {
480 > readonly group?: string;
481 > readonly silent: boolean;
482 > readonly configuration?: IStringDictionary<unknown>;
483 > }
484 >
485 > export interface ILanguageModelChatRequestOptions {
486 > readonly modelOptions?: IStringDictionary<unknown>;
487 > readonly configuration?: IStringDictionary<unknown>;
488 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
489 > readonly [name: string]: any;
490 > }
491 >
492 > export interface ILanguageModelsGroup {
493 > readonly group?: ILanguageModelsProviderGroup;
494 > readonly modelIdentifiers: string[];
495 > readonly status?: {
496 > readonly message: string;
497 > readonly severity: Severity;
498 > };
499 > }
500 >
501 > export interface ILanguageModelsService {
502 >
503 > readonly _serviceBrand: undefined;
504 >
505 > readonly onDidChangeLanguageModelVendors: Event<readonly string[]>;
506 > readonly onDidChangeLanguageModels: Event<string>;
507 >
508 > getLanguageModelIds(): string[];
509 >
510 > getVendors(): ILanguageModelProviderDescriptor[];
511 >
512 > lookupLanguageModel(modelId: string): ILanguageModelChatMetadata | undefined;
513 >
514 > /**
515 > * Find a model by its qualified name. The qualified name is what is used in prompt and agent files and is in the format "Model Name (Vendor)".
516 > */
517 > lookupLanguageModelByQualifiedName(qualifiedName: string): ILanguageModelChatMetadataAndIdentifier | undefined;
518 >
519 > getLanguageModelGroups(vendor: string): ILanguageModelsGroup[];
520 >
521 > /**
522 > * Returns true if the given vendor's provider has completed at least one
523 > * model resolution since registration. A `false` result indicates the
524 > * vendor is still in a startup/reload race where its model list isn't yet
525 > * authoritative — callers can fall back to a cached list in that case.
526 > */
527 > hasResolvedVendor(vendor: string): boolean;
528 >
529 > /**
530 > * Given a selector, returns a list of model identifiers
531 > * @param selector The selector to lookup for language models. If the selector is empty, all language models are returned.
532 > */
533 > selectLanguageModels(selector: ILanguageModelChatSelector): Promise<string[]>;
534 >
535 > registerLanguageModelProvider(vendor: string, provider: ILanguageModelChatProvider): IDisposable;
536 >
537 > deltaLanguageModelChatProviderDescriptors(added: IUserFriendlyLanguageModel[], removed: IUserFriendlyLanguageModel[]): void;
538 >
539 > sendChatRequest(modelId: string, from: ExtensionIdentifier | undefined, messages: IChatMessage[], options: ILanguageModelChatRequestOptions, token: CancellationToken): Promise<ILanguageModelChatResponse>;
540 >
541 > computeTokenLength(modelId: string, message: string | IChatMessage, token: CancellationToken): Promise<number>;
542 >
543 > /**
544 > * Returns the resolved per-model configuration for the given model identifier.
545 > * Includes schema defaults with user overrides applied on top.
546 > * Returns undefined if the model has no configuration schema and no user config.
547 > */
548 > getModelConfiguration(modelId: string): IStringDictionary<unknown> | undefined;
549 >
550 > /**
551 > * Updates the per-model configuration for the given model.
552 > * Merges the provided values into the existing configuration.
553 > */
554 > setModelConfiguration(modelId: string, values: IStringDictionary<unknown>): Promise<void>;
555 >
556 > /**
557 > * Returns actions for configuring the given model based on its configuration schema.
558 > * For enum properties, returns submenu actions with checkable values.
559 > * Returns an empty array if the model has no configuration schema.
560 > */
561 > getModelConfigurationActions(modelId: string): IAction[];
562 >
563 > addLanguageModelsProviderGroup(name: string, vendorId: string, configuration: IStringDictionary<unknown> | undefined): Promise<void>;
564 >
565 > removeLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void>;
566 >
567 > configureLanguageModelsProviderGroup(vendorId: string, name?: string): Promise<void>;
568 >
569 > renameLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void>;
570 >
571 > updateLanguageModelsProviderGroupApiKey(vendorId: string, providerGroupName: string): Promise<void>;
572 >
573 > addLanguageModelsProviderGroupModel(vendorId: string, providerGroupName: string): Promise<void>;
574 >
575 > openLanguageModelsProviderGroupSettings(vendorId: string, providerGroupName: string): Promise<void>;
576 >
577 > /**
578 > * Opens the language models configuration file and navigates to
579 > * or creates the per-model configuration for the given model.
580 > */
581 > configureModel(modelId: string): Promise<void>;
582 >
583 > migrateLanguageModelsProviderGroup(languageModelsProviderGroup: ILanguageModelsProviderGroup): Promise<void>;
584 >
585 > /**
586 > * Returns the most recently used model identifiers, ordered by most-recent-first.
587 > * @param maxCount Maximum number of entries to return (default 7).
588 > */
589 > getRecentlyUsedModelIds(): string[];
590 >
591 > /**
592 > * Records that a model was used, updating the recently used list.
593 > */
594 > addToRecentlyUsedList(modelIdentifier: string): void;
595 >
596 > /**
597 > * Clears the recently used model list.
598 > */
599 > clearRecentlyUsedList(): void;
600 >
601 > /**
602 > * Returns the pinned model identifiers, in the order they were pinned.
603 > */
604 > getPinnedModelIds(): string[];
605 >
606 > /**
607 > * Pins a model so it appears in the pinned section of the model picker.
608 > */
609 > pinModel(modelIdentifier: string): void;
610 >
611 > /**
612 > * Unpins a model, removing it from the pinned section.
613 > */
614 > unpinModel(modelIdentifier: string): void;
615 >
616 > /**
617 > * Returns whether the given model is pinned.
618 > */
619 > isModelPinned(modelIdentifier: string): boolean;
620 >
621 > /**
622 > * Fires when the pinned models list changes.
623 > */
624 > readonly onDidChangePinnedModels: Event<void>;
625 >
626 > /**
627 > * Returns whether the given model is hidden from the chat model picker.
628 > */
629 > isModelHidden(modelIdentifier: string): boolean;
630 >
631 > /**
632 > * Returns whether every resolved model in the given (vendor, groupName)
633 > * bucket is hidden from the chat model picker.
634 > */
635 > isGroupHidden(vendor: string, groupName: string): boolean;
636 >
637 > /**
638 > * Hide or show a single model in the chat model picker.
639 > */
640 > setModelHidden(modelIdentifier: string, hidden: boolean): void;
641 >
642 > /**
643 > * Hide or show every model in a (vendor, groupName) bucket.
644 > */
645 > setGroupHidden(vendor: string, groupName: string, hidden: boolean): void;
646 >
647 > /**
648 > * Returns the persisted per-model hidden identifiers.
649 > */
650 > getHiddenModelIds(): string[];
651 >
652 > /**
653 > * Fires when any model or group visibility state changes.
654 > */
655 > readonly onDidChangeModelVisibility: Event<void>;
656 >
657 > /**
658 > * Returns the models from the control manifest,
659 > * separated into free and paid tiers.
660 > */
661 > getModelsControlManifest(): IModelsControlManifest;
662 >
663 > /**
664 > * Fires when models control manifest changes.
665 > */
666 > readonly onDidChangeModelsControlManifest: Event<IModelsControlManifest>;
667 >
668 > /**
669 > * Observable map of restricted chat participant names to allowed extension publisher/IDs.
670 > * Fetched from the chat control manifest.
671 > */
672 > readonly restrictedChatParticipants: IObservable<{ [name: string]: string[] }>;
673 > }
674 >
675 > export interface IModelControlEntry {
676 > readonly label: string;
677 > readonly featured?: boolean;
678 > readonly minVSCodeVersion?: string;
679 > readonly exists: boolean;
680 > }
681 >
682 > export interface IModelsControlManifest {
683 > readonly free: IStringDictionary<IModelControlEntry>;
684 > readonly paid: IStringDictionary<IModelControlEntry>;
685 > }
686 >
687 > const languageModelChatProviderType = {
688 > type: 'object',
689 > required: ['vendor', 'displayName'],
690 > properties: {
691 > vendor: {
692 > type: 'string',
693 > description: localize('vscode.extension.contributes.languageModels.vendor', "A globally unique vendor of language model chat provider.")
694 > },
695 > displayName: {
696 > type: 'string',
697 > description: localize('vscode.extension.contributes.languageModels.displayName', "The display name of the language model chat provider.")
698 > },
699 > configuration: {
700 > type: 'object',
701 > description: localize('vscode.extension.contributes.languageModels.configuration', "Configuration options for the language model chat provider."),
702 > anyOf: [
703 > {
704 > $ref: 'http://json-schema.org/draft-07/schema#'
705 > },
706 > {
707 > properties: {
708 > properties: {
709 > type: 'object',
710 > additionalProperties: {
711 > $ref: 'http://json-schema.org/draft-07/schema#',
712 > properties: {
713 > secret: {
714 > type: 'boolean',
715 > description: localize('vscode.extension.contributes.languageModels.configuration.secret', "Whether the property is a secret.")
716 > }
717 > }
718 > }
719 > },
720 > additionalProperties: {
721 > $ref: 'http://json-schema.org/draft-07/schema#',
722 > properties: {
723 > secret: {
724 > type: 'boolean',
725 > description: localize('vscode.extension.contributes.languageModels.configuration.secret', "Whether the property is a secret.")
726 > }
727 > }
728 > }
729 > }
730 > }
731 > ]
732 >
733 > },
734 > managementCommand: {
735 > type: 'string',
736 > description: localize('vscode.extension.contributes.languageModels.managementCommand', "A command to manage the language model chat provider, e.g. 'Manage Copilot models'. This is used in the chat model picker. If not provided, a gear icon is not rendered during vendor selection."),
737 > deprecated: true,
738 > deprecationMessage: localize('vscode.extension.contributes.languageModels.managementCommand.deprecated', "The managementCommand property is deprecated and will be removed in a future release. Use the new configuration property instead.")
739 > },
740 > deprecation: {
741 > type: 'object',
742 > description: localize('vscode.extension.contributes.languageModels.deprecation', "Marks this language model chat provider as deprecated. When set, the Manage Models view renders the provider with a link pointing to a replacement."),
743 > properties: {
744 > link: {
745 > type: 'string',
746 > description: localize('vscode.extension.contributes.languageModels.deprecation.link', "A URL opened when the user clicks the deprecation link shown next to the provider name. Use a 'vscode:extension/<publisher>.<name>' URI to open a replacement extension in the Extensions view.")
747 > }
748 > }
749 > },
750 > when: {
751 > type: 'string',
752 > description: localize('vscode.extension.contributes.languageModels.when', "Condition which must be true to show this language model chat provider in the Manage Models list.")
753 > }
754 > }
755 > } as const satisfies IJSONSchema;
756 >
757 > export type IUserFriendlyLanguageModel = Omit<TypeFromJsonSchema<typeof languageModelChatProviderType>, 'deprecation'> & {
758 > /**
759 > * Marks a provider as deprecated. The Manage Models view renders a link
760 > * (pointing to a replacement, e.g. a `vscode:extension/<publisher>.<name>` URI)
761 > * next to the provider name. Optional so existing provider descriptors are unaffected.
762 > */
763 > readonly deprecation?: { readonly link?: string };
764 > };
765 >
766 > export interface ILanguageModelProviderDescriptor extends IUserFriendlyLanguageModel {
767 > readonly isDefault: boolean;
768 > }
769 >
770 > /**
771 > * Resolves a provider `deprecation.link` for opening inside the current build. Contributions point
772 > * at the replacement extension with a stable `vscode:extension/<id>` URI, but the URL service only
773 > * routes URIs whose scheme matches this build's `urlProtocol` (e.g. `code-oss`, `vscode-insiders`).
774 > * The `vscode:` scheme is therefore rewritten to the current protocol so the extensions URL handler
775 > * opens the extension; without this the opener falls back to treating the URI as a (non-existent)
776 > * file resource and fails. Other schemes (http(s), command) are returned unchanged.
777 > */
778 > export function resolveProviderDeprecationLink(link: string, urlProtocol: string | undefined): URI {
779 const uri = URI.parse(link);
780 return uri.scheme === Schemas.vscode && urlProtocol ? uri.with({ scheme: urlProtocol }) : uri;
781 }
783 > export const languageModelChatProviderExtensionPoint = ExtensionsRegistry.registerExtensionPoint<IUserFriendlyLanguageModel | IUserFriendlyLanguageModel[]>({
784 > extensionPoint: 'languageModelChatProviders',
785 > jsonSchema: {
786 > description: localize('vscode.extension.contributes.languageModelChatProviders', "Contribute language model chat providers of a specific vendor."),
787 > oneOf: [
788 > languageModelChatProviderType,
789 > {
790 > type: 'array',
791 > items: languageModelChatProviderType
792 > }
793 > ]
794 > },
795 > activationEventsGenerator: function* (contribs: readonly IUserFriendlyLanguageModel[]) {
796 for (const contrib of contribs) {
797 yield `onLanguageModelChatProvider:${contrib.vendor}`;
798 }
799 }
800 > }); languageModels.ts
801 >
802 > const CHAT_MODEL_RECENTLY_USED_STORAGE_KEY = 'chatModelRecentlyUsed';
803 > const CHAT_MODEL_PINNED_STORAGE_KEY = 'chatModelPinned';
804 > const CHAT_MODEL_VISIBILITY_STORAGE_KEY = 'chatModelVisibility';
805 >
806 > /**
807 > * The identifier for the Auto model which dynamically routes to the best backend.
808 > * Auto should never appear in user-curated lists (MRU, pinned).
809 > */
810 > const AUTO_MODEL_IDENTIFIER = 'copilot/auto';
811 >
812 > export function isAutoLanguageModel(model: ILanguageModelChatMetadataAndIdentifier | undefined): boolean {
813 return model?.metadata.id === 'auto' || model?.identifier === AUTO_MODEL_IDENTIFIER;
814 }
816 > const CHAT_PARTICIPANT_NAME_REGISTRY_STORAGE_KEY = 'chat.participantNameRegistry';
817 > const CHAT_MODELS_CONTROL_STORAGE_KEY = 'chat.modelsControl';
818 >
819 > interface IChatControlResponse {
820 > readonly version: number;
821 > readonly restrictedChatParticipants: { [name: string]: string[] };
822 > readonly models?: {
823 > readonly free?: Record<string, { readonly label: string; readonly featured?: boolean }>;
824 > readonly paid?: Record<string, { readonly label: string; readonly featured?: boolean; readonly minVSCodeVersion?: string }>;
825 > };
826 > }
827 >
828 > /**
829 > * Builds the per-model configuration submenu actions from a model's
830 > * {@link ILanguageModelConfigurationSchema}. The current value is read from
831 > * `currentConfig` and selections are routed through `setValue`, allowing the
832 > * caller to decide whether changes apply globally or to a per-editor override.
833 > */
834 > export function createModelConfigurationActions(
835 schema: ILanguageModelConfigurationSchema | undefined,
836 currentConfig: IStringDictionary<unknown>,
873 return actions;
874 }
876 > export class LanguageModelsService implements ILanguageModelsService {
877 >
878 > private static SECRET_KEY_PREFIX = 'chat.lm.secret.';
879 > private static SECRET_INPUT = '${input:{0}}';
880 >
881 > readonly _serviceBrand: undefined;
882 >
883 > private readonly _store = new DisposableStore();
884 >
885 > private readonly _providers = new Map<string, ILanguageModelChatProvider>();
886 > private readonly _vendors = new Map<string, ILanguageModelProviderDescriptor>();
887 >
888 > /** Vendors for which a deprecation notice has already been shown this session. */
889 > private readonly _deprecationNoticeShownVendors = new Set<string>();
890 >
891 > private readonly _onDidChangeLanguageModelVendors = this._store.add(new Emitter<string[]>());
892 > readonly onDidChangeLanguageModelVendors = this._onDidChangeLanguageModelVendors.event;
893 >
894 > private readonly _modelsGroups = new Map<string, ILanguageModelsGroup[]>();
895 > private readonly _modelCache = new Map<string, ILanguageModelChatMetadata>();
896 > private readonly _resolveLMSequencer = new SequencerByKey<string>();
897 > private readonly _modelConfigurations = new Map<string, IStringDictionary<unknown>>();
898 > private readonly _hasUserSelectableModels: IContextKey<boolean>;
899 > private readonly _hasNonCopilotUserSelectableModels: IContextKey<boolean>;
900 >
901 > private readonly _onLanguageModelChange = this._store.add(new Emitter<string>());
902 > readonly onDidChangeLanguageModels: Event<string> = this._onLanguageModelChange.event;
903 >
904 > private _recentlyUsedModelIds: string[] = [];
905 > private _pinnedModelIds: string[] = [];
906 >
907 > private _hiddenModelIds = new Set<string>();
908 >
909 > private readonly _onDidChangeModelsControlManifest = this._store.add(new Emitter<IModelsControlManifest>());
910 > readonly onDidChangeModelsControlManifest = this._onDidChangeModelsControlManifest.event;
911 >
912 > private readonly _onDidChangePinnedModels = this._store.add(new Emitter<void>());
913 > readonly onDidChangePinnedModels = this._onDidChangePinnedModels.event;
914 >
915 > private readonly _onDidChangeModelVisibility = this._store.add(new Emitter<void>());
916 > readonly onDidChangeModelVisibility = this._onDidChangeModelVisibility.event;
917 >
918 > private _modelsControlManifest: IModelsControlManifest = { free: {}, paid: {} };
919 > private _modelsControlRawResponse: IChatControlResponse['models'] | undefined;
920 >
921 > private _chatControlUrl: string | undefined;
922 > private _chatControlDisposed = false;
923 >
924 > private readonly _restrictedChatParticipants = observableValue<{ [name: string]: string[] }>(this, Object.create(null));
925 > readonly restrictedChatParticipants: IObservable<{ [name: string]: string[] }> = this._restrictedChatParticipants;
926 >
927 > constructor(
928 @IExtensionService private readonly _extensionService: IExtensionService,
929 @ILogService private readonly _logService: ILogService,
996 }));
997 }
999 > deltaLanguageModelChatProviderDescriptors(added: IUserFriendlyLanguageModel[], removed: IUserFriendlyLanguageModel[]): void {
1000 const addedVendorIds: string[] = [];
1001 const removedVendorIds: string[] = [];
1051 }
1052 }
1054 > private async _onDidChangeLanguageModelGroups(changedGroups: readonly ILanguageModelsProviderGroup[]): Promise<void> {
1055 const changedVendors = new Set(changedGroups.map(g => g.vendor));
1056 await Promise.all(Array.from(changedVendors).map(vendor => this._resolveAllLanguageModels(vendor, true)));
1057 }
1059 > getVendors(): ILanguageModelProviderDescriptor[] {
1060 return Array.from(this._vendors.values())
1061 .filter(vendor => {
1067 });
1068 }
1070 > getLanguageModelIds(): string[] {
1071 return Array.from(this._modelCache.keys());
1072 }
1074 > lookupLanguageModel(modelIdentifier: string): ILanguageModelChatMetadata | undefined {
1075 return this._modelCache.get(modelIdentifier);
1076 }
1078 > lookupLanguageModelByQualifiedName(referenceName: string): ILanguageModelChatMetadataAndIdentifier | undefined {
1079 for (const [identifier, model] of this._modelCache.entries()) {
1080 if (ILanguageModelChatMetadata.matchesQualifiedName(referenceName, model)) {
1084 return undefined;
1085 }
1087 > private async _resolveAllLanguageModels(vendorId: string, silent: boolean): Promise<void> {
1088
1089 const vendor = this._vendors.get(vendorId);
1247 });
1248 }
1250 > private _hasGroupStructureChanged(oldGroups: readonly ILanguageModelsGroup[], newGroups: readonly ILanguageModelsGroup[]): boolean {
1251 if (oldGroups.length !== newGroups.length) {
1252 return true;
1265 return false;
1266 }
1268 > getLanguageModelGroups(vendor: string): ILanguageModelsGroup[] {
1269 return this._modelsGroups.get(vendor) ?? [];
1270 }
1272 > hasResolvedVendor(vendor: string): boolean {
1273 return this._modelsGroups.has(vendor);
1274 }
1276 > async selectLanguageModels(selector: ILanguageModelChatSelector): Promise<string[]> {
1277
1278 if (selector.vendor) {
1298 return result;
1299 }
1301 > registerLanguageModelProvider(vendor: string, provider: ILanguageModelChatProvider): IDisposable {
1302 this._logService.trace('[LM] registering language model provider', vendor, provider);
1303
1323 });
1324 }
1326 > async sendChatRequest(modelId: string, from: ExtensionIdentifier | undefined, messages: IChatMessage[], options: ILanguageModelChatRequestOptions, token: CancellationToken): Promise<ILanguageModelChatResponse> {
1327 const metadata = this._modelCache.get(modelId);
1328 const provider = this._providers.get(metadata?.vendor || '');
1338 return provider.sendChatRequest(modelId, messages, from, mergedOptions, token);
1339 }
1341 > /**
1342 > * When a chat request is made against a deprecated provider (one that contributes a
1343 > * `deprecation.link`), prompt the user once per session to install the replacement
1344 > * extension. The notification can be dismissed, and offers a "Don't Show Again" choice that
1345 > * is persisted across sessions via the notification service's `neverShowAgain` support.
1346 > */
1347 > private _maybeShowProviderDeprecationNotice(metadata: ILanguageModelChatMetadata): void {
1348 const vendor = this._vendors.get(metadata.vendor);
1349 const link = vendor?.deprecation?.link;
1369 );
1370 }
1372 > /**
1373 > * Reports which in-built BYOK provider (or third-party extension) backs a model request. First-party
1374 > * Copilot models are intentionally not reported here (see {@link getByokProviderTelemetryName}).
1375 > */
1376 > private _logProviderUsageTelemetry(metadata: ILanguageModelChatMetadata | undefined): void {
1377 const provider = getByokProviderTelemetryName(metadata?.vendor, metadata?.extension);
1378 if (!provider) {
1394 });
1395 }
1397 > private _resolveModelConfigurationWithDefaults(modelId: string, metadata: ILanguageModelChatMetadata | undefined): IStringDictionary<unknown> | undefined {
1398 const userConfig = this._modelConfigurations.get(modelId);
1399 const schema = metadata?.configurationSchema;
1420 return { ...defaults, ...userConfig };
1421 }
1423 > computeTokenLength(modelId: string, message: string | IChatMessage, token: CancellationToken): Promise<number> {
1424 const model = this._modelCache.get(modelId);
1425 if (!model) {
1432 return provider.provideTokenCount(modelId, message, token);
1433 }
1435 > getModelConfiguration(modelId: string): IStringDictionary<unknown> | undefined {
1436 const metadata = this._modelCache.get(modelId);
1437 return this._resolveModelConfigurationWithDefaults(modelId, metadata);
1438 }
1440 > async setModelConfiguration(modelId: string, values: IStringDictionary<unknown>): Promise<void> {
1441 const metadata = this._modelCache.get(modelId);
1442 if (!metadata) {
1528 this._onLanguageModelChange.fire(metadata.vendor);
1529 }
1531 > getModelConfigurationActions(modelId: string): IAction[] {
1532 const metadata = this._modelCache.get(modelId);
1533 const currentConfig = this._modelConfigurations.get(modelId) ?? {};
1538 );
1539 }
1541 > async configureLanguageModelsProviderGroup(vendorId: string, providerGroupName?: string): Promise<void> {
1542
1543 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1583 }
1584 }
1586 > async renameLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void> {
1587 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1588 if (!vendor) {
1603 await this._languageModelsConfigurationService.updateLanguageModelsProviderGroup(existing, { ...existing, name });
1604 }
1606 > async updateLanguageModelsProviderGroupApiKey(vendorId: string, providerGroupName: string): Promise<void> {
1607 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1608 const schema = vendor?.configuration as IJSONSchema | undefined;
1638 }
1639 }
1641 > async addLanguageModelsProviderGroupModel(vendorId: string, providerGroupName: string): Promise<void> {
1642 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1643 const schema = vendor?.configuration as IJSONSchema | undefined;
1664 });
1665 }
1667 > async openLanguageModelsProviderGroupSettings(vendorId: string, providerGroupName: string): Promise<void> {
1668 const group = this._languageModelsConfigurationService.getLanguageModelsProviderGroups().find(group => group.vendor === vendorId && group.name === providerGroupName);
1669 if (!group) {
1673 await this._languageModelsConfigurationService.configureLanguageModels({ group });
1674 }
1676 > async configureModel(modelId: string): Promise<void> {
1677 const metadata = this._modelCache.get(modelId);
1678 if (!metadata || !metadata.configurationSchema) {
1708 await this._languageModelsConfigurationService.configureLanguageModels({ group, snippet });
1709 }
1711 > private _getModelConfigurationSnippet(modelId: string, schema: ILanguageModelConfigurationSchema): string {
1712 const properties: string[] = [];
1713 if (schema.properties) {
1730 return `"settings": {\n\t\t"${modelId}": ${modelContent}\n\t}`;
1731 }
1733 > async addLanguageModelsProviderGroup(name: string, vendorId: string, configuration: IStringDictionary<unknown> | undefined): Promise<void> {
1734 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1735 if (!vendor) {
1740 await this._languageModelsConfigurationService.addLanguageModelsProviderGroup(languageModelProviderGroup);
1741 }
1743 > async removeLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void> {
1744 const vendor = this.getVendors().find(({ vendor }) => vendor === vendorId);
1745 if (!vendor) {
1757 await this._languageModelsConfigurationService.removeLanguageModelsProviderGroup(existing);
1758 }
1760 > private requireConfiguring(schema: IJSONSchema): boolean {
1761 if (schema.additionalProperties) {
1762 return true;
1772 return false;
1773 }
1775 > private getSnippetForFirstUnconfiguredProperty(configuration: IStringDictionary<unknown>, schema: IJSONSchema): string | undefined {
1776 if (!schema.properties) {
1777 return undefined;
1788 return undefined;
1789 }
1791 > private getSnippetForProperty(property: string, propertySchema: IJSONSchema): string | undefined {
1792 const bodyText = this.getDefaultSnippetBodyText(propertySchema);
1793 return bodyText ? `"${property}": ${bodyText}` : undefined;
1794 }
1796 > private getSnippetForArrayItem(propertySchema: IJSONSchema): string | undefined {
1797 return this.getDefaultSnippetBodyText(propertySchema, true);
1798 }
1800 > private getDefaultSnippetBodyText(propertySchema: IJSONSchema, arrayItem = false): string | undefined {
1801 const snippet = propertySchema.defaultSnippets?.[0];
1802 if (!snippet) {
1813 return bodyText.replace(/"(\^[^"]*)"/g, (_, value) => value.substring(1));
1814 }
1816 > private async promptForName(languageModelProviderGroups: readonly ILanguageModelsProviderGroup[], vendor: IUserFriendlyLanguageModel, existing: ILanguageModelsProviderGroup | undefined): Promise<string | undefined> {
1817 let providerGroupName = existing?.name;
1818 if (!providerGroupName) {
1861 return result;
1862 }
1864 > private async promptForConfiguration(groupName: string, configuration: IJSONSchema, existing: IStringDictionary<unknown> | undefined): Promise<IStringDictionary<unknown> | undefined> {
1865 if (!configuration.properties) {
1866 return;
1880 return result;
1881 }
1883 > private async promptForValue(groupName: string, property: string, propertySchema: IJSONSchema | undefined, required: boolean, existing: IStringDictionary<unknown> | undefined): Promise<unknown | undefined> {
1884 if (!propertySchema) {
1885 return undefined;
1909 return value;
1910 }
1912 > private canPromptForProperty(propertySchema: IJSONSchema | undefined): boolean {
1913 if (!propertySchema || typeof propertySchema === 'boolean') {
1914 return false;
1925 return false;
1926 }
1928 > private getDescriptionPlaintext(propertySchema: IJSONSchema): string | undefined {
1929 if (propertySchema.description) {
1930 return propertySchema.description;
1942 .replace(/\[([^\]]+)\]\([^)]+\)/g, '$1');
1943 }
1945 > private async promptForArray(groupName: string, property: string, propertySchema: IJSONSchema): Promise<string[] | undefined> {
1946 if (!propertySchema.items || Array.isArray(propertySchema.items) || !propertySchema.items.enum) {
1947 return undefined;
1971 }
1972 }
1974 > private async promptForEnum(groupName: string, property: string, propertySchema: IJSONSchema & { enumItemLabels?: string[] }, existing: IStringDictionary<unknown> | undefined): Promise<string | undefined> {
1975 const values = propertySchema.enum;
1976 if (!Array.isArray(values) || values.length === 0) {
2014 }
2015 }
2017 > private async promptForInput(groupName: string, property: string, propertySchema: IJSONSchema, required: boolean, existing: IStringDictionary<unknown> | undefined): Promise<string | number | boolean | undefined> {
2018 const disposables = new DisposableStore();
2019 try {
2090 }
2091 }
2093 > private encodeSecretKey(property: string): string {
2094 return format(LanguageModelsService.SECRET_INPUT, property);
2095 }
2097 > private decodeSecretKey(secretInput: unknown): string | undefined {
2098 if (!isString(secretInput)) {
2099 return undefined;
2101 return secretInput.substring(secretInput.indexOf(':') + 1, secretInput.length - 1);
2102 }
2104 > private _clearModelCache(vendor: string): Map<string, ILanguageModelChatMetadata> {
2105 const removed = new Map<string, ILanguageModelChatMetadata>();
2106 for (const [id, model] of this._modelCache.entries()) {
2112 return removed;
2113 }
2115 > private _clearModelConfigurations(vendor: string): void {
2116 for (const [id] of this._modelConfigurations) {
2117 if (this._modelCache.get(id)?.vendor === vendor || id.startsWith(`${vendor}/`)) {
2120 }
2121 }
2123 > private async _resolveConfiguration(group: ILanguageModelsProviderGroup, schema: IJSONSchema | undefined): Promise<IStringDictionary<unknown>> {
2124 if (!schema) {
2125 return {};
2141 return result;
2142 }
2144 > private async _resolveLanguageModelProviderGroup(name: string, vendor: string, configuration: IStringDictionary<unknown> | undefined, schema: IJSONSchema | undefined): Promise<ILanguageModelsProviderGroup> {
2145 if (!schema) {
2146 return { name, vendor };
2160 return { name, vendor, ...result };
2161 }
2163 > private async _deleteSecretsInConfiguration(group: ILanguageModelsProviderGroup, schema: IJSONSchema | undefined): Promise<void> {
2164 if (!schema) {
2165 return;
2177 }
2178 }
2180 > async migrateLanguageModelsProviderGroup(languageModelsProviderGroup: ILanguageModelsProviderGroup): Promise<void> {
2181 const { vendor, name, ...configuration } = languageModelsProviderGroup;
2182 if (!this._vendors.get(vendor)) {
2194 await this.addLanguageModelsProviderGroup(name, vendor, configuration);
2195 }
2197 > //#region Recently used models
2198 >
2199 > private _readRecentlyUsedModels(): string[] {
2200 return this._storageService.getObject<string[]>(CHAT_MODEL_RECENTLY_USED_STORAGE_KEY, StorageScope.PROFILE, []);
2201 }
2203 > private _saveRecentlyUsedModels(): void {
2204 this._storageService.store(CHAT_MODEL_RECENTLY_USED_STORAGE_KEY, this._recentlyUsedModelIds, StorageScope.PROFILE, StorageTarget.USER);
2205 }
2207 > getRecentlyUsedModelIds(): string[] {
2208 // Filter to only include models that still exist in the cache
2209 return this._recentlyUsedModelIds
2211 .slice(0, 4);
2212 }
2214 > addToRecentlyUsedList(modelIdentifier: string): void {
2215 if (modelIdentifier === AUTO_MODEL_IDENTIFIER) {
2216 return;
2230 this._saveRecentlyUsedModels();
2231 }
2233 > clearRecentlyUsedList(): void {
2234 this._recentlyUsedModelIds = [];
2235 this._saveRecentlyUsedModels();
2236 }
2238 > //#endregion
2239 >
2240 > //#region Pinned models
2241 >
2242 > private _readPinnedModels(): string[] {
2243 return this._storageService.getObject<string[]>(CHAT_MODEL_PINNED_STORAGE_KEY, StorageScope.PROFILE, []);
2244 }
2246 > private _savePinnedModels(): void {
2247 this._storageService.store(CHAT_MODEL_PINNED_STORAGE_KEY, this._pinnedModelIds, StorageScope.PROFILE, StorageTarget.USER);
2248 }
2250 > getPinnedModelIds(): string[] {
2251 return this._pinnedModelIds.filter(id => id !== AUTO_MODEL_IDENTIFIER && this._modelCache.has(id));
2252 }
2254 > pinModel(modelIdentifier: string): void {
2255 if (modelIdentifier === AUTO_MODEL_IDENTIFIER || this._pinnedModelIds.includes(modelIdentifier)) {
2256 return;
2260 this._onDidChangePinnedModels.fire();
2261 }
2263 > unpinModel(modelIdentifier: string): void {
2264 const index = this._pinnedModelIds.indexOf(modelIdentifier);
2265 if (index === -1) {
2270 this._onDidChangePinnedModels.fire();
2271 }
2273 > isModelPinned(modelIdentifier: string): boolean {
2274 return modelIdentifier !== AUTO_MODEL_IDENTIFIER && this._pinnedModelIds.includes(modelIdentifier);
2275 }
2277 > //#endregion
2278 >
2279 > //#region Model visibility
2280 >
2281 > private _getGroupNameForVendor(vendor: string): string {
2282 return this._vendors.get(vendor)?.displayName ?? vendor;
2283 }
2285 > private _getModelIdsInGroup(vendor: string, groupName: string): string[] {
2286 const vendorGroups = this._modelsGroups.get(vendor);
2287 if (!vendorGroups) {
2311 return result;
2312 }
2314 > private _readVisibility(): void {
2315 const raw = this._storageService.getObject<{ hiddenModels?: string[] }>(CHAT_MODEL_VISIBILITY_STORAGE_KEY, StorageScope.PROFILE, {});
2316 this._hiddenModelIds = new Set(Array.isArray(raw?.hiddenModels) ? raw.hiddenModels : []);
2317 }
2319 > private _saveVisibility(): void {
2320 this._storageService.store(
2321 CHAT_MODEL_VISIBILITY_STORAGE_KEY,
2325 );
2326 }
2328 > isGroupHidden(vendor: string, groupName: string): boolean {
2329 const modelIds = this._getModelIdsInGroup(vendor, groupName);
2330 return modelIds.length > 0 && modelIds.every(id => this._hiddenModelIds.has(id));
2331 }
2333 > isModelHidden(modelIdentifier: string): boolean {
2334 return this._hiddenModelIds.has(modelIdentifier);
2335 }
2337 > setGroupHidden(vendor: string, groupName: string, hidden: boolean): void {
2338 let changed = false;
2339 const modelIds = this._getModelIdsInGroup(vendor, groupName);
2353 }
2354 }
2356 > setModelHidden(modelIdentifier: string, hidden: boolean): void {
2357 let changed = false;
2358 if (hidden) {
2369 }
2370 }
2372 > getHiddenModelIds(): string[] {
2373 return Array.from(this._hiddenModelIds);
2374 }
2376 > //#endregion
2377 >
2378 > //#region Models control manifest
2379 >
2380 > getModelsControlManifest(): IModelsControlManifest {
2381 return this._modelsControlManifest;
2382 }
2384 > private _setModelsControlManifest(response: IChatControlResponse['models']): void {
2385 this._modelsControlRawResponse = response;
2386 this._refreshModelsControlManifest();
2387 }
2389 > private _refreshModelsControlManifest(): void {
2390 const response = this._modelsControlRawResponse;
2391 const free: IStringDictionary<IModelControlEntry> = {};
2415 this._onDidChangeModelsControlManifest.fire(this._modelsControlManifest);
2416 }
2418 > //#region Chat control data
2419 > private _initChatControlData(): void {
2420 this._chatControlUrl = this._productService.chatParticipantRegistry;
2421 if (!this._chatControlUrl) {
2444 this._refreshChatControlData();
2445 }
2447 > private _refreshChatControlData(): void {
2448 if (this._chatControlDisposed) {
2449 return;
2455 .then(() => this._refreshChatControlData());
2456 }
2458 > private async _fetchChatControlData(): Promise<void> {
2459 this._logService.trace('[LM] Fetching chat control data from', this._chatControlUrl);
2460
2499 }
2500 }
2502 > //#endregion
2503 >
2504 > dispose() {
2505 this._chatControlDisposed = true;
2506 this._store.dispose();
2507 this._providers.clear();
2508 }
2510 > }
src/vs/base/common/event.ts 981 covered LOC · 137 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- event.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancelablePromise } from './async.js';
7 > import { CancellationToken } from './cancellation.js';
8 > import { diffSets } from './collections.js';
9 > import { onUnexpectedError } from './errors.js';
10 > import { createSingleCallFunction } from './functional.js';
11 > import { combinedDisposable, Disposable, DisposableMap, DisposableStore, IDisposable, toDisposable } from './lifecycle.js';
12 > import { LinkedList } from './linkedList.js';
13 > import { IObservable, IObservableWithChange, IObserver } from './observable.js';
14 > import { env } from './process.js';
15 > import { StopWatch } from './stopwatch.js';
16 > import { MicrotaskDelay } from './symbols.js';
17 >
18 >
19 > // -----------------------------------------------------------------------------------------------------------------------
20 > // Uncomment the next line to print warnings whenever an emitter with listeners is disposed. That is a sign of code smell.
21 > // -----------------------------------------------------------------------------------------------------------------------
22 > const _enableDisposeWithListenerWarning = false
23 > // || Boolean("TRUE") // causes a linter warning so that it cannot be pushed
24 > ;
25 >
26 >
27 > // -----------------------------------------------------------------------------------------------------------------------
28 > // Uncomment the next line to print warnings whenever a snapshotted event is used repeatedly without cleanup.
29 > // See https://github.com/microsoft/vscode/issues/142851
30 > // -----------------------------------------------------------------------------------------------------------------------
31 > const _enableSnapshotPotentialLeakWarning = false
32 > // || Boolean("TRUE") // causes a linter warning so that it cannot be pushed
33 > ;
34 >
35 >
36 > const _bufferLeakWarnCountThreshold = 100;
37 > const _bufferLeakWarnTimeThreshold = 60_000; // 1 minute
38 >
39 function _isBufferLeakWarningEnabled(): boolean {
40 return !!env['VSCODE_DEV'];
41 }
42 > event.ts
43 > /**
44 > * An event with zero or one parameters that can be subscribed to. The event is a function itself.
45 > */
46 > export interface Event<T> {
47 > (listener: (e: T) => unknown, thisArgs?: any, disposables?: IDisposable[] | DisposableStore): IDisposable;
48 > }
49 >
50 > export namespace Event {
51 > export const None: Event<any> = () => Disposable.None;
52 >
53 > function _addLeakageTraceLogic(options: EmitterOptions) {
54 if (_enableSnapshotPotentialLeakWarning) {
55 const { onDidAddListener: origListenerDidAdd } = options;
65 }
66 }
67 > event.ts
68 > /**
69 > * Given an event, returns another event which debounces calls and defers the listeners to a later task via a shared
70 > * `setTimeout`. The event is converted into a signal (`Event<void>`) to avoid additional object creation as a
71 > * result of merging events and to try prevent race conditions that could arise when using related deferred and
72 > * non-deferred events.
73 > *
74 > * This is useful for deferring non-critical work (eg. general UI updates) to ensure it does not block critical work
75 > * (eg. latency of keypress to text rendered).
76 > *
77 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
78 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
79 > * returned event causes this utility to leak a listener on the original event.
80 > *
81 > * @param event The event source for the new event.
82 > * @param flushOnListenerRemove Whether to fire all debounced events when a listener is removed. If this is not
83 > * specified, some events could go missing. Use this if it's important that all events are processed, even if the
84 > * listener gets disposed before the debounced event fires.
85 > * @param disposable A disposable store to add the new EventEmitter to.
86 > */
87 > export function defer(event: Event<unknown>, flushOnListenerRemove?: boolean, disposable?: DisposableStore): Event<void> {
88 return debounce<unknown, void>(event, () => void 0, 0, undefined, flushOnListenerRemove ?? true, undefined, disposable);
89 }
90 > event.ts
91 > /**
92 > * Given an event, returns another event which only fires once.
93 > *
94 > * @param event The event source for the new event.
95 > */
96 > export function once<T>(event: Event<T>): Event<T> {
97 return (listener, thisArgs = null, disposables?) => {
98 // we need this, in case the event fires during the listener call
118 };
119 }
120 > event.ts
121 > /**
122 > * Given an event, returns another event which only fires once, and only when the condition is met.
123 > *
124 > * @param event The event source for the new event.
125 > */
126 > export function onceIf<T>(event: Event<T>, condition: (e: T) => boolean): Event<T> {
127 return Event.once(Event.filter(event, condition));
128 }
129 > event.ts
130 > /**
131 > * Maps an event of one type into an event of another type using a mapping function, similar to how
132 > * `Array.prototype.map` works.
133 > *
134 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
135 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
136 > * returned event causes this utility to leak a listener on the original event.
137 > *
138 > * @param event The event source for the new event.
139 > * @param map The mapping function.
140 > * @param disposable A disposable store to add the new EventEmitter to.
141 > */
142 > export function map<I, O>(event: Event<I>, map: (i: I) => O, disposable?: DisposableStore): Event<O> {
143 return snapshot((listener, thisArgs = null, disposables?) => event(i => listener.call(thisArgs, map(i)), null, disposables), disposable);
144 }
145 > event.ts
146 > /**
147 > * Wraps an event in another event that performs some function on the event object before firing.
148 > *
149 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
150 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
151 > * returned event causes this utility to leak a listener on the original event.
152 > *
153 > * @param event The event source for the new event.
154 > * @param each The function to perform on the event object.
155 > * @param disposable A disposable store to add the new EventEmitter to.
156 > */
157 > export function forEach<I>(event: Event<I>, each: (i: I) => void, disposable?: DisposableStore): Event<I> {
158 return snapshot((listener, thisArgs = null, disposables?) => event(i => { each(i); listener.call(thisArgs, i); }, null, disposables), disposable);
159 }
160 > event.ts
161 > /**
162 > * Wraps an event in another event that fires only when some condition is met.
163 > *
164 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
165 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
166 > * returned event causes this utility to leak a listener on the original event.
167 > *
168 > * @param event The event source for the new event.
169 > * @param filter The filter function that defines the condition. The event will fire for the object if this function
170 > * returns true.
171 > * @param disposable A disposable store to add the new EventEmitter to.
172 > */
173 > export function filter<T, U>(event: Event<T | U>, filter: (e: T | U) => e is T, disposable?: DisposableStore): Event<T>;
174 > export function filter<T>(event: Event<T>, filter: (e: T) => boolean, disposable?: DisposableStore): Event<T>;
175 > export function filter<T, R>(event: Event<T | R>, filter: (e: T | R) => e is R, disposable?: DisposableStore): Event<R>;
176 > export function filter<T>(event: Event<T>, filter: (e: T) => boolean, disposable?: DisposableStore): Event<T> {
177 return snapshot((listener, thisArgs = null, disposables?) => event(e => filter(e) && listener.call(thisArgs, e), null, disposables), disposable);
178 }
179 > event.ts
180 > /**
181 > * Given an event, returns the same event but typed as `Event<void>`.
182 > */
183 > export function signal<T>(event: Event<T>): Event<void> {
184 return event as Event<any> as Event<void>;
185 }
186 > event.ts
187 > /**
188 > * Given a collection of events, returns a single event which emits whenever any of the provided events emit.
189 > */
190 > export function any<T>(...events: Event<T>[]): Event<T>;
191 > export function any(...events: Event<any>[]): Event<void>;
192 > export function any<T>(...events: Event<T>[]): Event<T> {
193 return (listener, thisArgs = null, disposables?) => {
194 const disposable = combinedDisposable(...events.map(event => event(e => listener.call(thisArgs, e))));
196 };
197 }
198 > event.ts
199 > /**
200 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
201 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
202 > * returned event causes this utility to leak a listener on the original event.
203 > */
204 > export function reduce<I, O>(event: Event<I>, merge: (last: O | undefined, event: I) => O, initial?: O, disposable?: DisposableStore): Event<O> {
205 let output: O | undefined = initial;
206
210 }, disposable);
211 }
212 > event.ts
213 > function snapshot<T>(event: Event<T>, disposable: DisposableStore | undefined): Event<T> {
214 let listener: IDisposable | undefined;
215
233 return emitter.event;
234 }
235 > event.ts
236 > /**
237 > * Adds the IDisposable to the store if it's set, and returns it. Useful to
238 > * Event function implementation.
239 > */
240 > function addAndReturnDisposable<T extends IDisposable>(d: T, store: DisposableStore | IDisposable[] | undefined): T {
241 if (store instanceof Array) {
242 store.push(d);
246 return d;
247 }
248 > event.ts
249 > /**
250 > * Given an event, creates a new emitter that event that will debounce events based on {@link delay} and give an
251 > * array event object of all events that fired.
252 > *
253 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
254 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
255 > * returned event causes this utility to leak a listener on the original event.
256 > *
257 > * @param event The original event to debounce.
258 > * @param merge A function that reduces all events into a single event.
259 > * @param delay The number of milliseconds to debounce.
260 > * @param leading Whether to fire a leading event without debouncing.
261 > * @param flushOnListenerRemove Whether to fire all debounced events when a listener is removed. If this is not
262 > * specified, some events could go missing. Use this if it's important that all events are processed, even if the
263 > * listener gets disposed before the debounced event fires.
264 > * @param leakWarningThreshold See {@link EmitterOptions.leakWarningThreshold}.
265 > * @param disposable A disposable store to register the debounce emitter to.
266 > */
267 > export function debounce<T>(event: Event<T>, merge: (last: T | undefined, event: T) => T, delay?: number | typeof MicrotaskDelay, leading?: boolean, flushOnListenerRemove?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event<T>;
268 > export function debounce<I, O>(event: Event<I>, merge: (last: O | undefined, event: I) => O, delay?: number | typeof MicrotaskDelay, leading?: boolean, flushOnListenerRemove?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event<O>;
269 > export function debounce<I, O>(event: Event<I>, merge: (last: O | undefined, event: I) => O, delay: number | typeof MicrotaskDelay = 100, leading = false, flushOnListenerRemove = false, leakWarningThreshold?: number, disposable?: DisposableStore): Event<O> {
270 let subscription: IDisposable;
271 let output: O | undefined = undefined;
330 return emitter.event;
331 }
332 > event.ts
333 > /**
334 > * Debounces an event, firing after some delay (default=0) with an array of all event original objects.
335 > *
336 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
337 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
338 > * returned event causes this utility to leak a listener on the original event.
339 > *
340 > * @param event The event source for the new event.
341 > * @param delay The number of milliseconds to debounce.
342 > * @param flushOnListenerRemove Whether to fire all debounced events when a listener is removed. If this is not
343 > * specified, some events could go missing. Use this if it's important that all events are processed, even if the
344 > * listener gets disposed before the debounced event fires.
345 > * @param disposable A disposable store to add the new EventEmitter to.
346 > */
347 > export function accumulate<T>(event: Event<T>, delay: number | typeof MicrotaskDelay = 0, flushOnListenerRemove?: boolean, disposable?: DisposableStore): Event<T[]> {
348 return Event.debounce<T, T[]>(event, (last, e) => {
349 if (!last) {
354 }, delay, undefined, flushOnListenerRemove ?? true, undefined, disposable);
355 }
356 > event.ts
357 > /**
358 > * Throttles an event, ensuring the event is fired at most once during the specified delay period.
359 > * Unlike debounce, throttle will fire immediately on the leading edge and/or after the delay on the trailing edge.
360 > *
361 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
362 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
363 > * returned event causes this utility to leak a listener on the original event.
364 > *
365 > * @param event The event source for the new event.
366 > * @param merge An accumulator function that merges events if multiple occur during the throttle period.
367 > * @param delay The number of milliseconds to throttle.
368 > * @param leading Whether to fire on the leading edge (immediately on first event).
369 > * @param trailing Whether to fire on the trailing edge (after delay with the last value).
370 > * @param leakWarningThreshold See {@link EmitterOptions.leakWarningThreshold}.
371 > * @param disposable A disposable store to register the throttle emitter to.
372 > */
373 > export function throttle<T>(event: Event<T>, merge: (last: T | undefined, event: T) => T, delay?: number | typeof MicrotaskDelay, leading?: boolean, trailing?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event<T>;
374 > export function throttle<I, O>(event: Event<I>, merge: (last: O | undefined, event: I) => O, delay?: number | typeof MicrotaskDelay, leading?: boolean, trailing?: boolean, leakWarningThreshold?: number, disposable?: DisposableStore): Event<O>;
375 > export function throttle<I, O>(event: Event<I>, merge: (last: O | undefined, event: I) => O, delay: number | typeof MicrotaskDelay = 100, leading = true, trailing = true, leakWarningThreshold?: number, disposable?: DisposableStore): Event<O> {
376 let subscription: IDisposable;
377 let output: O | undefined = undefined;
437 return emitter.event;
438 }
439 > event.ts
440 > /**
441 > * Filters an event such that some condition is _not_ met more than once in a row, effectively ensuring duplicate
442 > * event objects from different sources do not fire the same event object.
443 > *
444 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
445 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
446 > * returned event causes this utility to leak a listener on the original event.
447 > *
448 > * @param event The event source for the new event.
449 > * @param equals The equality condition.
450 > * @param disposable A disposable store to add the new EventEmitter to.
451 > *
452 > * @example
453 > * ```
454 > * // Fire only one time when a single window is opened or focused
455 > * Event.latch(Event.any(onDidOpenWindow, onDidFocusWindow))
456 > * ```
457 > */
458 > export function latch<T>(event: Event<T>, equals: (a: T, b: T) => boolean = (a, b) => a === b, disposable?: DisposableStore): Event<T> {
459 let firstCall = true;
460 let cache: T;
467 }, disposable);
468 }
469 > event.ts
470 > /**
471 > * Splits an event whose parameter is a union type into 2 separate events for each type in the union.
472 > *
473 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
474 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
475 > * returned event causes this utility to leak a listener on the original event.
476 > *
477 > * @example
478 > * ```
479 > * const event = new EventEmitter<number | undefined>().event;
480 > * const [numberEvent, undefinedEvent] = Event.split(event, isUndefined);
481 > * ```
482 > *
483 > * @param event The event source for the new event.
484 > * @param isT A function that determines what event is of the first type.
485 > * @param disposable A disposable store to add the new EventEmitter to.
486 > */
487 > export function split<T, U>(event: Event<T | U>, isT: (e: T | U) => e is T, disposable?: DisposableStore): [Event<T>, Event<U>] {
488 return [
489 Event.filter(event, isT, disposable),
491 ];
492 }
493 > event.ts
494 > /**
495 > * Buffers an event until it has a listener attached.
496 > *
497 > * *NOTE* that this function returns an `Event` and it MUST be called with a `DisposableStore` whenever the returned
498 > * event is accessible to "third parties", e.g the event is a public property. Otherwise a leaked listener on the
499 > * returned event causes this utility to leak a listener on the original event.
500 > *
501 > * @param event The event source for the new event.
502 > * @param debugName A name for this buffer, used in leak detection warnings.
503 > * @param flushAfterTimeout Determines whether to flush the buffer after a timeout immediately or after a
504 > * `setTimeout` when the first event listener is added.
505 > * @param _buffer Internal: A source event array used for tests.
506 > *
507 > * @example
508 > * ```
509 > * // Start accumulating events, when the first listener is attached, flush
510 > * // the event after a timeout such that multiple listeners attached before
511 > * // the timeout would receive the event
512 > * this.onInstallExtension = Event.buffer(service.onInstallExtension, 'onInstallExtension', true);
513 > * ```
514 > */
515 > export function buffer<T>(event: Event<T>, debugName: string, flushAfterTimeout = false, _buffer: T[] = [], disposable?: DisposableStore): Event<T> {
516 let buffer: T[] | null = _buffer.slice();
517
600 return emitter.event;
601 }
602 > /** event.ts
603 > * Wraps the event in an {@link IChainableEvent}, allowing a more functional programming style.
604 > *
605 > * @example
606 > * ```
607 > * // Normal
608 > * const onEnterPressNormal = Event.filter(
609 > * Event.map(onKeyPress.event, e => new StandardKeyboardEvent(e)),
610 > * e.keyCode === KeyCode.Enter
611 > * ).event;
612 > *
613 > * // Using chain
614 > * const onEnterPressChain = Event.chain(onKeyPress.event, $ => $
615 > * .map(e => new StandardKeyboardEvent(e))
616 > * .filter(e => e.keyCode === KeyCode.Enter)
617 > * );
618 > * ```
619 > */
620 > export function chain<T, R>(event: Event<T>, sythensize: ($: IChainableSythensis<T>) => IChainableSythensis<R>): Event<R> {
621 const fn: Event<R> = (listener, thisArgs, disposables) => {
622 const cs = sythensize(new ChainableSynthesis()) as ChainableSynthesis;
631 return fn;
632 }
633 > event.ts
634 > const HaltChainable = Symbol('HaltChainable');
635 >
636 > class ChainableSynthesis implements IChainableSythensis<any> {
637 private readonly steps: ((input: any) => unknown)[] = [];
638 > event.ts
639 > map<O>(fn: (i: any) => O): this {
640 this.steps.push(fn);
641 return this;
642 }
643 > event.ts
644 > forEach(fn: (i: any) => void): this {
645 this.steps.push(v => {
646 fn(v);
649 return this;
650 }
651 > event.ts
652 > filter(fn: (e: any) => boolean): this {
653 this.steps.push(v => fn(v) ? v : HaltChainable);
654 return this;
655 }
656 > event.ts
657 > reduce<R>(merge: (last: R | undefined, event: any) => R, initial?: R | undefined): this {
658 let last = initial;
659 this.steps.push(v => {
663 return this;
664 }
665 > event.ts
666 > latch(equals: (a: any, b: any) => boolean = (a, b) => a === b): ChainableSynthesis {
667 let firstCall = true;
668 let cache: any;
676 return this;
677 }
678 > event.ts
679 > public evaluate(value: any) {
680 for (const step of this.steps) {
681 value = step(value);
687 return value;
688 }
689 > } event.ts
690 >
691 > export interface IChainableSythensis<T> {
692 > map<O>(fn: (i: T) => O): IChainableSythensis<O>;
693 > forEach(fn: (i: T) => void): IChainableSythensis<T>;
694 > filter<R extends T>(fn: (e: T) => e is R): IChainableSythensis<R>;
695 > filter(fn: (e: T) => boolean): IChainableSythensis<T>;
696 > reduce<R>(merge: (last: R, event: T) => R, initial: R): IChainableSythensis<R>;
697 > reduce<R>(merge: (last: R | undefined, event: T) => R): IChainableSythensis<R>;
698 > latch(equals?: (a: T, b: T) => boolean): IChainableSythensis<T>;
699 > }
700 >
701 > export interface NodeEventEmitter {
702 > on(event: string | symbol, listener: Function): unknown;
703 > removeListener(event: string | symbol, listener: Function): unknown;
704 > }
705 >
706 > /**
707 > * Creates an {@link Event} from a node event emitter.
708 > */
709 > export function fromNodeEventEmitter<T>(emitter: NodeEventEmitter, eventName: string, map: (...args: any[]) => T = id => id): Event<T> {
710 const fn = (...args: unknown[]) => result.fire(map(...args));
711 const onFirstListenerAdd = () => emitter.on(eventName, fn);
715 return result.event;
716 }
717 > event.ts
718 > export interface DOMEventEmitter {
719 > addEventListener(event: string | symbol, listener: Function): void;
720 > removeEventListener(event: string | symbol, listener: Function): void;
721 > }
722 >
723 > /**
724 > * Creates an {@link Event} from a DOM event emitter.
725 > */
726 > export function fromDOMEventEmitter<T>(emitter: DOMEventEmitter, eventName: string, map: (...args: any[]) => T = id => id): Event<T> {
727 const fn = (...args: unknown[]) => result.fire(map(...args));
728 const onFirstListenerAdd = () => emitter.addEventListener(eventName, fn);
732 return result.event;
733 }
734 > event.ts
735 > /**
736 > * Creates a promise out of an event, using the {@link Event.once} helper.
737 > */
738 > export function toPromise<T>(event: Event<T>, disposables?: IDisposable[] | DisposableStore): CancelablePromise<T> {
739 let cancelRef: () => void;
740 let listener: IDisposable;
756 return promise;
757 }
758 > event.ts
759 > /**
760 > * A convenience function for forwarding an event to another emitter which
761 > * improves readability.
762 > *
763 > * This is similar to {@link Relay} but allows instantiating and forwarding
764 > * on a single line and also allows for multiple source events.
765 > * @param from The event to forward.
766 > * @param to The emitter to forward the event to.
767 > * @example
768 > * Event.forward(event, emitter);
769 > * // equivalent to
770 > * event(e => emitter.fire(e));
771 > * // equivalent to
772 > * event(emitter.fire, emitter);
773 > */
774 > export function forward<T>(from: Event<T>, to: Emitter<T>): IDisposable {
775 return from(e => to.fire(e));
776 }
777 > event.ts
778 > /**
779 > * Adds a listener to an event and calls the listener immediately with undefined as the event object.
780 > *
781 > * @example
782 > * ```
783 > * // Initialize the UI and update it when dataChangeEvent fires
784 > * runAndSubscribe(dataChangeEvent, () => this._updateUI());
785 > * ```
786 > */
787 > export function runAndSubscribe<T>(event: Event<T>, handler: (e: T) => unknown, initial: T): IDisposable;
788 > export function runAndSubscribe<T>(event: Event<T>, handler: (e: T | undefined) => unknown): IDisposable;
789 > export function runAndSubscribe<T>(event: Event<T>, handler: (e: T | undefined) => unknown, initial?: T): IDisposable {
790 handler(initial);
791 return event(e => handler(e));
792 }
793 > event.ts
794 > class EmitterObserver<T> implements IObserver {
795 >
796 > readonly emitter: Emitter<T>;
797 >
798 > private _counter = 0;
799 > private _hasChanged = false;
800 >
801 > constructor(readonly _observable: IObservable<T>, store: DisposableStore | undefined) {
802 const options: EmitterOptions = {
803 onWillAddFirstListener: () => {
819 }
820 }
821 > event.ts
822 > beginUpdate<T>(_observable: IObservable<T>): void {
823 // assert(_observable === this.obs);
824 this._counter++;
825 }
826 > event.ts
827 > handlePossibleChange<T>(_observable: IObservable<T>): void {
828 // assert(_observable === this.obs);
829 }
830 > event.ts
831 > handleChange<T, TChange>(_observable: IObservableWithChange<T, TChange>, _change: TChange): void {
832 // assert(_observable === this.obs);
833 this._hasChanged = true;
834 }
835 > event.ts
836 > endUpdate<T>(_observable: IObservable<T>): void {
837 // assert(_observable === this.obs);
838 this._counter--;
845 }
846 }
847 > } event.ts
848 >
849 > /**
850 > * Creates an event emitter that is fired when the observable changes.
851 > * Each listeners subscribes to the emitter.
852 > */
853 > export function fromObservable<T>(obs: IObservable<T>, store?: DisposableStore): Event<T> {
854 const observer = new EmitterObserver(obs, store);
855 return observer.emitter.event;
856 }
857 > event.ts
858 > /**
859 > * Each listener is attached to the observable directly.
860 > */
861 > export function fromObservableLight(observable: IObservable<unknown>): Event<void> {
862 return (listener, thisArgs, disposables) => {
863 let count = 0;
897 };
898 }
899 > } event.ts
900 >
901 > export interface EmitterOptions {
902 > /**
903 > * Optional function that's called *before* the very first listener is added
904 > */
905 > onWillAddFirstListener?: Function;
906 > /**
907 > * Optional function that's called *after* the very first listener is added
908 > */
909 > onDidAddFirstListener?: Function;
910 > /**
911 > * Optional function that's called after a listener is added
912 > */
913 > onDidAddListener?: Function;
914 > /**
915 > * Optional function that's called *after* remove the very last listener
916 > */
917 > onDidRemoveLastListener?: Function;
918 > /**
919 > * Optional function that's called *before* a listener is removed
920 > */
921 > onWillRemoveListener?: Function;
922 > /**
923 > * Optional function that's called when a listener throws an error. Defaults to
924 > * {@link onUnexpectedError}
925 > */
926 > onListenerError?: (e: any) => void;
927 > /**
928 > * Number of listeners that are allowed before assuming a leak. Default to
929 > * a globally configured value
930 > *
931 > * @see setGlobalLeakWarningThreshold
932 > */
933 > leakWarningThreshold?: number;
934 > /**
935 > * Human-readable name for the emitter, included in leak warning error
936 > * messages to help identify which emitter is leaking in telemetry.
937 > */
938 > leakWarningName?: string;
939 > /**
940 > * Pass in a delivery queue, which is useful for ensuring
941 > * in order event delivery across multiple emitters.
942 > */
943 > deliveryQueue?: EventDeliveryQueue;
944 >
945 > /** ONLY enable this during development */
946 > _profName?: string;
947 > }
948 >
949 >
950 > export class EventProfiling {
951 >
952 > static readonly all = new Set<EventProfiling>();
953 >
954 > private static _idPool = 0;
955 >
956 > readonly name: string;
957 > public listenerCount: number = 0;
958 > public invocationCount = 0;
959 > public elapsedOverall = 0;
960 > public durations: number[] = [];
961 >
962 > private _stopWatch?: StopWatch;
963 >
964 > constructor(name: string) {
965 this.name = `${name}_${EventProfiling._idPool++}`;
966 EventProfiling.all.add(this);
967 }
968 > event.ts
969 > start(listenerCount: number): void {
970 this._stopWatch = new StopWatch();
971 this.listenerCount = listenerCount;
972 }
973 > event.ts
974 > stop(): void {
975 if (this._stopWatch) {
976 const elapsed = this._stopWatch.elapsed();
981 }
982 }
983 > } event.ts
984 >
985 > let _globalLeakWarningThreshold = -1;
986 > export function setGlobalLeakWarningThreshold(n: number): IDisposable {
987 const oldValue = _globalLeakWarningThreshold;
988 _globalLeakWarningThreshold = n;
993 };
994 }
995 > event.ts
996 > class LeakageMonitor {
997 >
998 > private static _idPool = 1;
999 >
1000 > private _stacks: Map<string, number> | undefined;
1001 > private _warnCountdown: number = 0;
1002 >
1003 > constructor(
1004 private readonly _errorHandler: (err: Error) => void,
1005 readonly threshold: number,
1006 readonly name: string = (LeakageMonitor._idPool++).toString(16).padStart(3, '0')
1007 ) { }
1008 > event.ts
1009 > dispose(): void {
1010 this._stacks?.clear();
1011 }
1012 > event.ts
1013 > check(stack: Stacktrace, listenerCount: number): undefined | (() => void) {
1014
1015 const threshold = this.threshold;
1046 };
1047 }
1048 > event.ts
1049 > getMostFrequentStack(): [string, number] | undefined {
1050 if (!this._stacks) {
1051 return undefined;
1061 return topStack;
1062 }
1063 > } event.ts
1064 >
1065 > class Stacktrace {
1066 >
1067 > static create() {
1068 > const err = new Error();
1069 > return new Stacktrace(err.stack ?? '');
1070 > }
1071 >
1072 > private constructor(readonly value: string) { }
1073 >
1074 > print() {
1075 console.warn(this.value.split('\n').slice(2).join('\n'));
1076 }
1077 > } event.ts
1078 >
1079 > // error that is logged when going over the configured listener threshold
1080 > export class ListenerLeakError extends Error {
1081 > readonly kind: string;
1082 > readonly listenerCount: number;
1083 > /**
1084 > * The detailed message including listener count and most frequent stack.
1085 > * Available locally for debugging but intentionally not used as the error
1086 > * `message`. When `emitterName` is provided, errors group by emitter name
1087 > * and kind in telemetry; otherwise they group by kind alone.
1088 > */
1089 > readonly details: string;
1090 > constructor(kind: 'dominated' | 'popular', details: string, stack: string, listenerCount: number, emitterName?: string) {
1091 super(emitterName
1092 ? `[${emitterName}] potential listener LEAK detected, ${kind}`
1098 this.stack = stack;
1099 }
1100 > event.ts
1101 > static is(err: unknown): err is ListenerLeakError {
1102 return err instanceof ListenerLeakError
1103 || (err instanceof Error && typeof (err as Error & { kind: unknown; listenerCount: unknown }).kind === 'string' && typeof (err as Error & { kind: unknown; listenerCount: unknown }).listenerCount === 'number');
1104 }
1105 > } event.ts
1106 >
1107 > // SEVERE error that is logged when having gone way over the configured listener
1108 > // threshold so that the emitter refuses to accept more listeners
1109 > export class ListenerRefusalError extends ListenerLeakError {
1110 > constructor(kind: 'dominated' | 'popular', details: string, stack: string, listenerCount: number, emitterName?: string) {
1111 super(kind, details, stack, listenerCount, emitterName);
1112 this.name = 'ListenerRefusalError';
1113 }
1114 > } event.ts
1115 >
1116 > let id = 0;
1117 > class UniqueContainer<T> {
1118 > stack?: Stacktrace;
1119 > public id = id++;
1120 > constructor(public readonly value: T) { }
1121 > }
1122 > const compactionThreshold = 2;
1123 >
1124 > type ListenerContainer<T> = UniqueContainer<(data: T) => void>;
1125 > type ListenerOrListeners<T> = (ListenerContainer<T> | undefined)[] | ListenerContainer<T>;
1126 >
1127 > const forEachListener = <T>(listeners: ListenerOrListeners<T>, fn: (c: ListenerContainer<T>) => void) => {
1128 if (listeners instanceof UniqueContainer) {
1129 fn(listeners);
1137 }
1138 };
1139 > event.ts
1140 > /**
1141 > * The Emitter can be used to expose an Event to the public
1142 > * to fire it from the insides.
1143 > * Sample:
1144 > class Document {
1145 >
1146 > private readonly _onDidChange = new Emitter<(value:string)=>any>();
1147 >
1148 > public onDidChange = this._onDidChange.event;
1149 >
1150 > // getter-style
1151 > // get onDidChange(): Event<(value:string)=>any> {
1152 > // return this._onDidChange.event;
1153 > // }
1154 >
1155 > private _doIt() {
1156 > //...
1157 > this._onDidChange.fire(value);
1158 > }
1159 > }
1160 > */
1161 > export class Emitter<T> {
1162 >
1163 > private readonly _options?: EmitterOptions;
1164 > private readonly _leakageMon?: LeakageMonitor;
1165 > private readonly _perfMon?: EventProfiling;
1166 > private _disposed?: true;
1167 > private _event?: Event<T>;
1168 >
1169 > /**
1170 > * A listener, or list of listeners. A single listener is the most common
1171 > * for event emitters (#185789), so we optimize that special case to avoid
1172 > * wrapping it in an array (just like Node.js itself.)
1173 > *
1174 > * A list of listeners never 'downgrades' back to a plain function if
1175 > * listeners are removed, for two reasons:
1176 > *
1177 > * 1. That's complicated (especially with the deliveryQueue)
1178 > * 2. A listener with >1 listener is likely to have >1 listener again at
1179 > * some point, and swapping between arrays and functions may[citation needed]
1180 > * introduce unnecessary work and garbage.
1181 > *
1182 > * The array listeners can be 'sparse', to avoid reallocating the array
1183 > * whenever any listener is added or removed. If more than `1 / compactionThreshold`
1184 > * of the array is empty, only then is it resized.
1185 > */
1186 > protected _listeners?: ListenerOrListeners<T>;
1187 >
1188 > /**
1189 > * Always to be defined if _listeners is an array. It's no longer a true
1190 > * queue, but holds the dispatching 'state'. If `fire()` is called on an
1191 > * emitter, any work left in the _deliveryQueue is finished first.
1192 > */
1193 > private _deliveryQueue?: EventDeliveryQueuePrivate;
1194 > protected _size = 0;
1195 >
1196 > constructor(options?: EmitterOptions) {
1197 > this._options = options; event.ts
1198 > this._leakageMon = (_globalLeakWarningThreshold > 0 || this._options?.leakWarningThreshold)
1199 ? new LeakageMonitor(options?.onListenerError ?? onUnexpectedError, this._options?.leakWarningThreshold ?? _globalLeakWarningThreshold, this._options?.leakWarningName) :
1200 > undefined; event.ts
1201 > this._perfMon = this._options?._profName ? new EventProfiling(this._options._profName) : undefined;
1202 > this._deliveryQueue = this._options?.deliveryQueue as EventDeliveryQueuePrivate | undefined;
1203 > }
1204 > event.ts
1205 > dispose() {
1206 > if (!this._disposed) { event.ts
1207 > this._disposed = true;
1208 >
1209 > // It is bad to have listeners at the time of disposing an emitter, it is worst to have listeners keep the emitter
1210 > // alive via the reference that's embedded in their disposables. Therefore we loop over all remaining listeners and
1211 > // unset their subscriptions/disposables. Looping and blaming remaining listeners is done on next tick because the
1212 > // the following programming pattern is very popular:
1213 > //
1214 > // const someModel = this._disposables.add(new ModelObject()); // (1) create and register model
1215 > // this._disposables.add(someModel.onDidChange(() => { ... }); // (2) subscribe and register model-event listener
1216 > // ...later...
1217 > // this._disposables.dispose(); disposes (1) then (2): don't warn after (1) but after the "overall dispose" is done
1218 >
1219 > if (this._deliveryQueue?.current === this) {
1220 this._deliveryQueue.reset();
1221 }
1222 > if (this._listeners) { event.ts
1223 > if (_enableDisposeWithListenerWarning) { event.ts
1224 const listeners = this._listeners;
1225 queueMicrotask(() => {
1227 });
1228 }
1229 > event.ts
1230 > this._listeners = undefined;
1231 > this._size = 0;
1232 > }
1233 > this._options?.onDidRemoveLastListener?.(); event.ts
1234 > this._leakageMon?.dispose();
1235 > }
1236 > }
1237 > event.ts
1238 > /**
1239 > * For the public to allow to subscribe
1240 > * to events from this Emitter
1241 > */
1242 > get event(): Event<T> {
1243 > this._event ??= (callback: (e: T) => unknown, thisArgs?: any, disposables?: IDisposable[] | DisposableStore) => { event.ts
1244 > if (this._leakageMon && this._size > this._leakageMon.threshold ** 2) { event.ts
1245 const message = `[${this._leakageMon.name}] REFUSES to accept new listeners because it exceeded its threshold by far (${this._size} vs ${this._leakageMon.threshold})`;
1246 console.warn(message);
1254 return Disposable.None;
1255 }
1256 > event.ts
1257 > if (this._disposed) {
1258 // todo: should we warn if a listener is added to a disposed emitter? This happens often
1259 return Disposable.None;
1260 }
1261 > event.ts
1262 > if (thisArgs) {
1263 callback = callback.bind(thisArgs);
1264 }
1265 > event.ts
1266 > const contained = new UniqueContainer(callback);
1267 >
1268 > let removeMonitor: Function | undefined;
1269 > let stack: Stacktrace | undefined;
1270 > if (this._leakageMon && this._size >= Math.ceil(this._leakageMon.threshold * 0.2)) {
1271 // check and record this emitter for potential leakage
1272 contained.stack = Stacktrace.create();
1273 removeMonitor = this._leakageMon.check(contained.stack, this._size + 1);
1274 }
1275 > event.ts
1276 > if (_enableDisposeWithListenerWarning) {
1277 contained.stack = stack ?? Stacktrace.create();
1278 }
1279 > event.ts
1280 > if (!this._listeners) {
1281 > this._options?.onWillAddFirstListener?.(this);
1282 > this._listeners = contained;
1283 > this._options?.onDidAddFirstListener?.(this);
1284 > } else if (this._listeners instanceof UniqueContainer) {
1285 > this._deliveryQueue ??= new EventDeliveryQueuePrivate(); event.ts
1286 > this._listeners = [this._listeners, contained];
1287 > } else {
1288 this._listeners.push(contained);
1289 }
1290 > this._options?.onDidAddListener?.(this); event.ts
1291 >
1292 > this._size++;
1293 >
1294 >
1295 > const result = toDisposable(() => {
1296 > removeMonitor?.(); event.ts
1297 > this._removeListener(contained);
1298 > }); event.ts
1299 > addToDisposables(result, disposables);
1300 >
1301 > return result;
1302 > };
1303 > event.ts
1304 > return this._event;
1305 > }
1306 > event.ts
1307 > private _removeListener(listener: ListenerContainer<T>) {
1308 > this._options?.onWillRemoveListener?.(this); event.ts
1309 >
1310 > if (!this._listeners) {
1311 > return; // expected if a listener gets disposed event.ts
1312 > }
1313 > event.ts
1314 > if (this._size === 1) {
1315 > this._listeners = undefined; event.ts
1316 > this._options?.onDidRemoveLastListener?.(this);
1317 > this._size = 0;
1318 > return;
1319 > }
1320
1321 // size > 1 which requires that listeners be a list:
1348 listeners.length = n;
1349 }
1350 > } event.ts
1351 > event.ts
1352 > private _deliver(listener: undefined | UniqueContainer<(value: T) => void>, value: T) {
1353 > if (!listener) { event.ts
1354 return;
1355 }
1356 > event.ts
1357 > const errorHandler = this._options?.onListenerError || onUnexpectedError;
1358 > if (!errorHandler) {
1359 listener.value(value);
1360 return;
1361 }
1362 > event.ts
1363 > try {
1364 > listener.value(value);
1365 > } catch (e) {
1366 errorHandler(e);
1367 }
1368 > } event.ts
1369 > event.ts
1370 > /** Delivers items in the queue. Assumes the queue is ready to go. */
1371 > private _deliverQueue(dq: EventDeliveryQueuePrivate) {
1372 > const listeners = dq.current!._listeners! as (ListenerContainer<T> | undefined)[]; event.ts
1373 > while (dq.i < dq.end) {
1374 > // important: dq.i is incremented before calling deliver() because it might reenter deliverQueue()
1375 > this._deliver(listeners[dq.i++], dq.value as T);
1376 > }
1377 > dq.reset();
1378 > }
1379 > event.ts
1380 > /**
1381 > * To be kept private to fire an event to
1382 > * subscribers
1383 > */
1384 > fire(event: T): void {
1385 > if (this._deliveryQueue?.current) { event.ts
1386 this._deliverQueue(this._deliveryQueue);
1387 this._perfMon?.stop(); // last fire() will have starting perfmon, stop it before starting the next dispatch
1388 }
1389 > event.ts
1390 > this._perfMon?.start(this._size);
1391 >
1392 > if (!this._listeners) {
1393 > // no-op event.ts
1394 > } else if (this._listeners instanceof UniqueContainer) { event.ts
1395 > this._deliver(this._listeners, event); event.ts
1396 > } else { event.ts
1397 > const dq = this._deliveryQueue!; event.ts
1398 > dq.enqueue(this, event, this._listeners.length);
1399 > this._deliverQueue(dq);
1400 > }
1401 > event.ts
1402 > this._perfMon?.stop();
1403 > }
1404 > event.ts
1405 > hasListeners(): boolean {
1406 return this._size > 0;
1407 }
1408 > } event.ts
1409 >
1410 > export interface EventDeliveryQueue {
1411 > _isEventDeliveryQueue: true;
1412 > }
1413 >
1414 > export const createEventDeliveryQueue = (): EventDeliveryQueue => new EventDeliveryQueuePrivate();
1415 >
1416 > class EventDeliveryQueuePrivate implements EventDeliveryQueue { event.ts
1417 > declare _isEventDeliveryQueue: true;
1418 >
1419 > /**
1420 > * Index in current's listener list.
1421 > */
1422 > public i = -1;
1423 >
1424 > /**
1425 > * The last index in the listener's list to deliver.
1426 > */
1427 > public end = 0;
1428 > event.ts
1429 > /**
1430 > * Emitter currently being dispatched on. Emitter._listeners is always an array.
1431 > */
1432 > public current?: Emitter<any>;
1433 > /**
1434 > * Currently emitting value. Defined whenever `current` is.
1435 > */
1436 > public value?: unknown;
1437 >
1438 > public enqueue<T>(emitter: Emitter<T>, value: T, end: number) {
1439 > this.i = 0; event.ts
1440 > this.end = end;
1441 > this.current = emitter;
1442 > this.value = value;
1443 > }
1444 > event.ts
1445 > public reset() {
1446 > this.i = this.end; // force any current emission loop to stop, mainly for during dispose event.ts
1447 > this.current = undefined;
1448 > this.value = undefined;
1449 > }
1450 > } event.ts
1451 >
1452 > export interface IWaitUntil {
1453 > token: CancellationToken;
1454 > waitUntil(thenable: Promise<unknown>): void;
1455 > }
1456 >
1457 > export type IWaitUntilData<T> = Omit<Omit<T, 'waitUntil'>, 'token'>;
1458 >
1459 > export class AsyncEmitter<T extends IWaitUntil> extends Emitter<T> {
1460 >
1461 > private _asyncDeliveryQueue?: LinkedList<[(ev: T) => void, IWaitUntilData<T>]>;
1462 >
1463 > async fireAsync(data: IWaitUntilData<T>, token: CancellationToken, promiseJoin?: (p: Promise<unknown>, listener: Function) => Promise<unknown>): Promise<void> {
1464 if (!this._listeners) {
1465 return;
1512 }
1513 }
1514 > } event.ts
1515 >
1516 >
1517 > export class PauseableEmitter<T> extends Emitter<T> {
1518 >
1519 > private _isPaused = 0;
1520 > protected _eventQueue = new LinkedList<T>();
1521 > private _mergeFn?: (input: T[]) => T;
1522 >
1523 > public get isPaused(): boolean {
1524 > return this._isPaused !== 0;
1525 > }
1526 >
1527 > constructor(options?: EmitterOptions & { merge?: (input: T[]) => T }) {
1528 > super(options); event.ts
1529 > this._mergeFn = options?.merge;
1530 > }
1531 > event.ts
1532 > pause(): void {
1533 this._isPaused++;
1534 }
1535 > event.ts
1536 > resume(): void {
1537 if (this._isPaused !== 0 && --this._isPaused === 0) {
1538 if (this._mergeFn) {
1554 }
1555 }
1556 > event.ts
1557 > override fire(event: T): void {
1558 if (this._size) {
1559 if (this._isPaused !== 0) {
1564 }
1565 }
1566 > } event.ts
1567 >
1568 > export class DebounceEmitter<T> extends PauseableEmitter<T> {
1569 >
1570 > private readonly _delay: number;
1571 > private _handle: Timeout | undefined;
1572 >
1573 > constructor(options: EmitterOptions & { merge: (input: T[]) => T; delay?: number }) {
1574 super(options);
1575 this._delay = options.delay ?? 100;
1576 }
1577 > event.ts
1578 > override fire(event: T): void {
1579 if (!this._handle) {
1580 this.pause();
1586 super.fire(event);
1587 }
1588 > } event.ts
1589 >
1590 > /**
1591 > * An emitter which queue all events and then process them at the
1592 > * end of the event loop.
1593 > */
1594 > export class MicrotaskEmitter<T> extends Emitter<T> {
1595 > private _queuedEvents: T[] = [];
1596 > private _mergeFn?: (input: T[]) => T;
1597 >
1598 > constructor(options?: EmitterOptions & { merge?: (input: T[]) => T }) {
1599 super(options);
1600 this._mergeFn = options?.merge;
1601 }
1602 > override fire(event: T): void { event.ts
1603
1604 if (!this.hasListeners()) {
1618 }
1619 }
1620 > } event.ts
1621 >
1622 > /**
1623 > * An event emitter that multiplexes many events into a single event.
1624 > *
1625 > * @example Listen to the `onData` event of all `Thing`s, dynamically adding and removing `Thing`s
1626 > * to the multiplexer as needed.
1627 > *
1628 > * ```typescript
1629 > * const anythingDataMultiplexer = new EventMultiplexer<{ data: string }>();
1630 > *
1631 > * const thingListeners = DisposableMap<Thing, IDisposable>();
1632 > *
1633 > * thingService.onDidAddThing(thing => {
1634 > * thingListeners.set(thing, anythingDataMultiplexer.add(thing.onData);
1635 > * });
1636 > * thingService.onDidRemoveThing(thing => {
1637 > * thingListeners.deleteAndDispose(thing);
1638 > * });
1639 > *
1640 > * anythingDataMultiplexer.event(e => {
1641 > * console.log('Something fired data ' + e.data)
1642 > * });
1643 > * ```
1644 > */
1645 > export class EventMultiplexer<T> implements IDisposable {
1646 >
1647 > private readonly emitter: Emitter<T>;
1648 > private hasListeners = false;
1649 > private events: { event: Event<T>; listener: IDisposable | null }[] = [];
1650 >
1651 > constructor() {
1652 this.emitter = new Emitter<T>({
1653 onWillAddFirstListener: () => this.onFirstListenerAdd(),
1655 });
1656 }
1657 > event.ts
1658 > get event(): Event<T> {
1659 return this.emitter.event;
1660 }
1661 > event.ts
1662 > add(event: Event<T>): IDisposable {
1663 const e = { event: event, listener: null };
1664 this.events.push(e);
1679 return toDisposable(createSingleCallFunction(dispose));
1680 }
1681 > event.ts
1682 > private onFirstListenerAdd(): void {
1683 this.hasListeners = true;
1684 this.events.forEach(e => this.hook(e));
1685 }
1686 > event.ts
1687 > private onLastListenerRemove(): void {
1688 this.hasListeners = false;
1689 this.events.forEach(e => this.unhook(e));
1690 }
1691 > event.ts
1692 > private hook(e: { event: Event<T>; listener: IDisposable | null }): void {
1693 e.listener = e.event(r => this.emitter.fire(r));
1694 }
1695 > event.ts
1696 > private unhook(e: { event: Event<T>; listener: IDisposable | null }): void {
1697 e.listener?.dispose();
1698 e.listener = null;
1699 }
1700 > event.ts
1701 > dispose(): void {
1702 this.emitter.dispose();
1703
1707 this.events = [];
1708 }
1709 > } event.ts
1710 >
1711 > export interface IDynamicListEventMultiplexer<TEventType> extends IDisposable {
1712 > readonly event: Event<TEventType>;
1713 > }
1714 > export class DynamicListEventMultiplexer<TItem, TEventType> implements IDynamicListEventMultiplexer<TEventType> {
1715 > private readonly _store = new DisposableStore();
1716 >
1717 > readonly event: Event<TEventType>;
1718 >
1719 > constructor(
1720 items: TItem[],
1721 onAddItem: Event<TItem>,
1747 this.event = multiplexer.event;
1748 }
1749 > event.ts
1750 > dispose() {
1751 this._store.dispose();
1752 }
1753 > } event.ts
1754 >
1755 > /**
1756 > * The EventBufferer is useful in situations in which you want
1757 > * to delay firing your events during some code.
1758 > * You can wrap that code and be sure that the event will not
1759 > * be fired during that wrap.
1760 > *
1761 > * ```
1762 > * const emitter: Emitter;
1763 > * const delayer = new EventDelayer();
1764 > * const delayedEvent = delayer.wrapEvent(emitter.event);
1765 > *
1766 > * delayedEvent(console.log);
1767 > *
1768 > * delayer.bufferEvents(() => {
1769 > * emitter.fire(); // event will not be fired yet
1770 > * });
1771 > *
1772 > * // event will only be fired at this point
1773 > * ```
1774 > */
1775 > export class EventBufferer {
1776
1777 private data: { buffers: Function[] }[] = [];
1778 > event.ts
1779 > wrapEvent<T>(event: Event<T>): Event<T>;
1780 > wrapEvent<T>(event: Event<T>, reduce: (last: T | undefined, event: T) => T): Event<T>;
1781 > wrapEvent<T, O>(event: Event<T>, reduce: (last: O | undefined, event: T) => O, initial: O): Event<O>;
1782 > wrapEvent<T, O>(event: Event<T>, reduce?: (last: T | O | undefined, event: T) => T | O, initial?: O): Event<O | T> {
1783 return (listener, thisArgs?, disposables?) => {
1784 return event(i => {
1832 };
1833 }
1834 > event.ts
1835 > bufferEvents<R = void>(fn: () => R): R {
1836 const data = { buffers: new Array<Function>() };
1837 this.data.push(data);
1841 return r;
1842 }
1843 > } event.ts
1844 >
1845 > /**
1846 > * A Relay is an event forwarder which functions as a replugabble event pipe.
1847 > * Once created, you can connect an input event to it and it will simply forward
1848 > * events from that input event through its own `event` property. The `input`
1849 > * can be changed at any point in time.
1850 > */
1851 > export class Relay<T> implements IDisposable {
1852
1853 private listening = false;
1867
1868 readonly event: Event<T> = this.emitter.event;
1869 > event.ts
1870 > set input(event: Event<T>) {
1871 this.inputEvent = event;
1872
1876 }
1877 }
1878 > event.ts
1879 > dispose() {
1880 this.inputEventListener.dispose();
1881 this.emitter.dispose();
1882 }
1883 > } event.ts
1884 >
1885 > export interface IValueWithChangeEvent<T> {
1886 > readonly onDidChange: Event<void>;
1887 > get value(): T;
1888 > }
1889 >
1890 > export class ValueWithChangeEvent<T> implements IValueWithChangeEvent<T> {
1891 > public static const<T>(value: T): IValueWithChangeEvent<T> {
1892 > return new ConstValueWithChangeEvent(value);
1893 > }
1894 >
1895 > private readonly _onDidChange = new Emitter<void>();
1896 > readonly onDidChange: Event<void> = this._onDidChange.event;
1897 >
1898 > constructor(private _value: T) { }
1899 >
1900 > get value(): T {
1901 return this._value;
1902 }
1903 > event.ts
1904 > set value(value: T) {
1905 if (value !== this._value) {
1906 this._value = value;
1908 }
1909 }
1910 > } event.ts
1911 >
1912 > class ConstValueWithChangeEvent<T> implements IValueWithChangeEvent<T> {
1913 > public readonly onDidChange: Event<void> = Event.None;
1914 >
1915 > constructor(readonly value: T) { }
1916 > }
1917 >
1918 > /**
1919 > * @param handleItem Is called for each item in the set (but only the first time the item is seen in the set).
1920 > * The returned disposable is disposed if the item is no longer in the set.
1921 > */
1922 > export function trackSetChanges<T>(getData: () => ReadonlySet<T>, onDidChangeData: Event<unknown>, handleItem: (d: T) => IDisposable): IDisposable {
1923 const map = new DisposableMap<T, IDisposable>();
1924 let oldData = new Set(getData());
1942 return store;
1943 }
1944 > event.ts
1945 >
1946 > function addToDisposables(result: IDisposable, disposables: DisposableStore | IDisposable[] | undefined) { event.ts
1947 > if (disposables instanceof DisposableStore) {
1948 disposables.add(result);
1949 > } else if (Array.isArray(disposables)) { event.ts
1950 disposables.push(result);
1951 }
1952 > } event.ts
1953 > event.ts
1954 function disposeAndRemove(result: IDisposable, disposables: DisposableStore | IDisposable[] | undefined) {
1955 if (disposables instanceof DisposableStore) {
src/vs/workbench/contrib/mcp/common/mcpTypes.ts 935 covered LOC · 31 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { equals as arraysEqual } from '../../../../base/common/arrays.js';
7 > import { assertNever } from '../../../../base/common/assert.js';
8 > import { decodeHex, encodeHex, VSBuffer } from '../../../../base/common/buffer.js';
9 > import { CancellationToken } from '../../../../base/common/cancellation.js';
10 > import { Event } from '../../../../base/common/event.js';
11 > import { IMarkdownString } from '../../../../base/common/htmlContent.js';
12 > import { Disposable, IDisposable } from '../../../../base/common/lifecycle.js';
13 > import { equals as objectsEqual } from '../../../../base/common/objects.js';
14 > import { IObservable, ObservableMap } from '../../../../base/common/observable.js';
15 > import { IIterativePager } from '../../../../base/common/paging.js';
16 > import Severity from '../../../../base/common/severity.js';
17 > import { URI, UriComponents } from '../../../../base/common/uri.js';
18 > import { Location } from '../../../../editor/common/languages.js';
19 > import { localize } from '../../../../nls.js';
20 > import { ConfigurationTarget } from '../../../../platform/configuration/common/configuration.js';
21 > import { RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
22 > import { IEditorOptions } from '../../../../platform/editor/common/editor.js';
23 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
24 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
25 > import { McpGalleryManifestStatus } from '../../../../platform/mcp/common/mcpGalleryManifest.js';
26 > import { IGalleryMcpServer, IGalleryMcpServerConfiguration, IInstallableMcpServer, IQueryOptions } from '../../../../platform/mcp/common/mcpManagement.js';
27 > import { IMcpDevModeConfig, IMcpSandboxConfiguration, IMcpServerConfiguration } from '../../../../platform/mcp/common/mcpPlatformTypes.js';
28 > import { StorageScope } from '../../../../platform/storage/common/storage.js';
29 > import { IWorkspaceFolder, IWorkspaceFolderData } from '../../../../platform/workspace/common/workspace.js';
30 > import { IWorkbenchLocalMcpServer, IWorkbencMcpServerInstallOptions, WORKSPACE_FOLDER_CONFIG_ID_PREFIX } from '../../../services/mcp/common/mcpWorkbenchManagementService.js';
31 > import { ContributionEnablementState, IEnablementModel } from '../../chat/common/enablement.js';
32 > import { ToolProgress } from '../../chat/common/tools/languageModelToolsService.js';
33 > import { IMcpServerSamplingConfiguration } from './mcpConfiguration.js';
34 > import { McpServerRequestHandler } from './mcpServerRequestHandler.js';
35 > import { MCP } from './modelContextProtocol.js';
36 > import { UriTemplate } from '../../../../base/common/uriTemplate.js';
37 >
38 > export const extensionMcpCollectionPrefix = 'ext.';
39 >
40 > /**
41 > * Prefix of the collection id used for MCP servers configured via the various
42 > * `mcp.json`-style config files (user, remote user, workspace, and
43 > * `.vscode/mcp.json` workspace-folder configs). The suffix is the
44 > * {@link IMcpConfigPath.id} of the originating config path.
45 > */
46 > export const MCP_CONFIGURATION_COLLECTION_ID_PREFIX = 'mcp.config.';
47 >
48 > export function extensionPrefixedIdentifier(identifier: ExtensionIdentifier, id: string): string {
49 return ExtensionIdentifier.toKey(identifier) + '/' + id;
50 }
52 > /**
53 > * An McpCollection contains McpServers. There may be multiple collections for
54 > * different locations servers are discovered.
55 > */
56 > export interface McpCollectionDefinition {
57 > /** Origin authority from which this collection was discovered. */
58 > readonly remoteAuthority: string | null;
59 > /** Globally-unique, stable ID for this definition */
60 > readonly id: string;
61 > /** Human-readable label for the definition */
62 > readonly label: string;
63 > /** Definitions this collection contains. */
64 > readonly serverDefinitions: IObservable<readonly McpServerDefinition[]>;
65 > /**
66 > * Trust behavior of the servers. `Trusted` means it will run without a prompt, always.
67 > * `TrustedOnNonce` means it will run without a prompt as long as the nonce matches.
68 > */
69 > readonly trustBehavior: McpServerTrust.Kind.Trusted | McpServerTrust.Kind.TrustedOnNonce;
70 > /** Scope where associated collection info should be stored. */
71 > readonly scope: StorageScope;
72 > /** Configuration target where configuration related to this server should be stored. */
73 > readonly configTarget: ConfigurationTarget;
74 > /** Root-level sandbox settings from the mcp config file. */
75 > readonly sandbox?: IMcpSandboxConfiguration;
76 >
77 > /** Resolves a server definition. If present, always called before a server starts. */
78 > resolveServerLanch?(definition: McpServerDefinition): Promise<McpServerLaunch | undefined>;
79 >
80 > /** For lazy-loaded collections only: */
81 > readonly lazy?: {
82 > /** True if `serverDefinitions` were loaded from the cache */
83 > isCached: boolean;
84 > /** Triggers a load of the real server definition, which should be pushed to the IMcpRegistry. If not this definition will be removed. */
85 > load(): Promise<void>;
86 > /** Called after `load()` if the extension is not found. */
87 > removed?(): void;
88 > };
89 >
90 > readonly source?: IWorkbenchMcpServer | ExtensionIdentifier;
91 >
92 > /** Sort order of the collection. Lower values have higher priority. */
93 > readonly order: number;
94 >
95 > readonly presentation?: {
96 > /** Place where this collection is configured, used in workspace trust prompts and "show config" */
97 > readonly origin?: URI;
98 > };
99 > }
100 >
101 > export const enum McpCollectionSortOrder {
102 > WorkspaceFolder = 0,
103 > Workspace = 100,
104 > User = 200,
105 > Extension = 300,
106 > Plugin = 350,
107 > Filesystem = 400,
108 >
109 > RemoteBoost = -50,
110 > }
111 >
112 > export namespace McpCollectionDefinition {
113 > export interface FromExtHost {
114 > readonly id: string;
115 > readonly label: string;
116 > readonly isTrustedByDefault: boolean;
117 > readonly scope: StorageScope;
118 > readonly canResolveLaunch: boolean;
119 > readonly extensionId: string;
120 > readonly configTarget: ConfigurationTarget;
121 > }
122 >
123 > export function equals(a: McpCollectionDefinition, b: McpCollectionDefinition): boolean {
124 return a.id === b.id
125 && a.remoteAuthority === b.remoteAuthority
128 && objectsEqual(a.sandbox, b.sandbox);
129 }
130 > mcpTypes.ts
131 > /**
132 > * Returns `true` when the collection was discovered from the workspace (its
133 > * config target is the workspace or a workspace folder). This is
134 > * intentionally based on the config target and not the storage scope:
135 > * extension-contributed collections use a workspace storage scope but are
136 > * configured at the user level, so they are not workspace-discovered.
137 > */
138 > export function isWorkspaceDiscovered(collection: McpCollectionDefinition): boolean {
139 return collection.configTarget === ConfigurationTarget.WORKSPACE
140 || collection.configTarget === ConfigurationTarget.WORKSPACE_FOLDER;
141 }
142 > mcpTypes.ts
143 > /**
144 > * Returns `true` when the collection originates from a `.vscode/mcp.json`
145 > * workspace-folder config, identified by its collection id prefix (the
146 > * shared `mcp.config.` prefix plus the workspace-folder config id).
147 > */
148 > export function isVscodeMcpJson(collection: McpCollectionDefinition): boolean {
149 return collection.id.startsWith(`${MCP_CONFIGURATION_COLLECTION_ID_PREFIX}${WORKSPACE_FOLDER_CONFIG_ID_PREFIX}`);
150 }
151 > } mcpTypes.ts
152 >
153 > export interface McpServerDefinition {
154 > /** Globally-unique, stable ID for this definition */
155 > readonly id: string;
156 > /** Human-readable label for the definition */
157 > readonly label: string;
158 > /** Descriptor defining how the configuration should be launched. */
159 > readonly launch: McpServerLaunch;
160 > /** Explicit roots. If undefined, all workspace folders. */
161 > readonly roots?: URI[] | undefined;
162 > /** If set, allows configuration variables to be resolved in the {@link launch} with the given context */
163 > readonly variableReplacement?: McpServerDefinitionVariableReplacement;
164 > /** Nonce used for caching the server. Changing the nonce will indicate that tools need to be refreshed. */
165 > readonly cacheNonce: string;
166 > /** Dev mode configuration for the server */
167 > readonly devMode?: IMcpDevModeConfig;
168 > /** Static description of server tools/data, used to hydrate the cache. */
169 > readonly staticMetadata?: McpServerStaticMetadata;
170 > /** Indicates if the sandbox is enabled for this server. */
171 > readonly sandboxEnabled?: boolean;
172 >
173 >
174 > readonly presentation?: {
175 > /** Sort order of the definition. */
176 > readonly order?: number;
177 > /** Place where this server is configured, used in workspace trust prompts and "show config" */
178 > readonly origin?: Location;
179 > };
180 > }
181 >
182 > export const enum McpServerStaticToolAvailability {
183 > /** Tool is expected to be present as soon as the server is started. */
184 > Initial,
185 > /** Tool may be present later. */
186 > Dynamic,
187 > }
188 >
189 > export interface McpServerStaticMetadata {
190 > tools?: { availability: McpServerStaticToolAvailability; definition: MCP.Tool }[];
191 > instructions?: string;
192 > capabilities?: MCP.ServerCapabilities;
193 > serverInfo?: MCP.Implementation;
194 > }
195 >
196 > export namespace McpServerDefinition {
197 > export interface Serialized {
198 > readonly id: string;
199 > readonly label: string;
200 > readonly cacheNonce: string;
201 > readonly launch: McpServerLaunch.Serialized;
202 > readonly variableReplacement?: McpServerDefinitionVariableReplacement.Serialized;
203 > readonly staticMetadata?: McpServerStaticMetadata;
204 > readonly sandboxEnabled?: boolean;
205 > }
206 >
207 > export function toSerialized(def: McpServerDefinition): McpServerDefinition.Serialized {
208 return def;
209 }
210 > mcpTypes.ts
211 > export function fromSerialized(def: McpServerDefinition.Serialized): McpServerDefinition {
212 return {
213 id: def.id,
220 };
221 }
222 > mcpTypes.ts
223 > export function equals(a: McpServerDefinition, b: McpServerDefinition): boolean {
224 return a.id === b.id
225 && a.label === b.label
233
234 }
235 > } mcpTypes.ts
236 >
237 >
238 > export interface McpServerDefinitionVariableReplacement {
239 > section?: string; // e.g. 'mcp'
240 > folder?: IWorkspaceFolderData;
241 > target: ConfigurationTarget;
242 > }
243 >
244 > export namespace McpServerDefinitionVariableReplacement {
245 > export interface Serialized {
246 > target: ConfigurationTarget;
247 > section?: string;
248 > folder?: { name: string; index: number; uri: UriComponents };
249 > }
250 >
251 > export function toSerialized(def: McpServerDefinitionVariableReplacement): McpServerDefinitionVariableReplacement.Serialized {
252 return def;
253 }
254 > mcpTypes.ts
255 > export function fromSerialized(def: McpServerDefinitionVariableReplacement.Serialized): McpServerDefinitionVariableReplacement {
256 return {
257 section: def.section,
260 };
261 }
262 > } mcpTypes.ts
263 >
264 > /** An observable of the auto-starting servers. When 'starting' is empty, the operation is complete. */
265 > export interface IAutostartResult {
266 > working: boolean;
267 > starting: McpDefinitionReference[];
268 > serversRequiringInteraction: Array<McpDefinitionReference & { errorMessage?: string }>;
269 > }
270 >
271 > export namespace IAutostartResult {
272 > export const Empty: IAutostartResult = { working: false, starting: [], serversRequiringInteraction: [] };
273 > }
274 >
275 > export interface IMcpService {
276 > _serviceBrand: undefined;
277 > readonly servers: IObservable<readonly IMcpServer[]>;
278 >
279 > /** The enablement model for MCP servers. */
280 > readonly enablementModel: IEnablementModel;
281 >
282 > /** Resets the cached tools. */
283 > resetCaches(): void;
284 >
285 > /** Resets trusted MCP servers. */
286 > resetTrust(): void;
287 >
288 > /** Set if there are extensions that register MCP servers that have never been activated. */
289 > readonly lazyCollectionState: IObservable<{ state: LazyCollectionState; collections: McpCollectionDefinition[] }>;
290 >
291 > /** Auto-starts pending servers based on user settings. */
292 > autostart(token?: CancellationToken): IObservable<IAutostartResult>;
293 >
294 > /** Cancels any current autostart @internal */
295 > cancelAutostart(): void;
296 >
297 > /** Activates extension-providing MCP servers that have not yet been discovered. */
298 > activateCollections(): Promise<void>;
299 > }
300 >
301 > export const enum LazyCollectionState {
302 > HasUnknown,
303 > LoadingUnknown,
304 > AllKnown,
305 > }
306 >
307 > export const IMcpService = createDecorator<IMcpService>('IMcpService');
308 >
309 > export interface McpCollectionReference {
310 > id: string;
311 > label: string;
312 > order: number;
313 > presentation?: McpCollectionDefinition['presentation'];
314 > }
315 >
316 > export interface McpDefinitionReference {
317 > id: string;
318 > label: string;
319 > }
320 >
321 > export class McpStartServerInteraction {
322 /** @internal */
323 public readonly participants = new ObservableMap</* server definition ID */ string, { s: 'unknown' | 'resolved' } | { s: 'waiting'; definition: McpServerDefinition; collection: McpCollectionDefinition }>();
324 > choice?: Promise<string[] | undefined>; mcpTypes.ts
325 > }
326 >
327 > export interface IMcpServerStartOpts {
328 > /**
329 > * Automatically trust if changed. This should ONLY be set for afforances that
330 > * ensure the user sees the config before it gets started (e.g. code lenses)
331 > */
332 > autoTrustChanges?: boolean;
333 > /**
334 > * When to trigger the trust prompt.
335 > * - only-new: only prompt for servers that are not previously explicitly untrusted (default)
336 > * - all-untrusted: prompt for all servers that are not trusted
337 > * - never: don't prompt, fail silently when trying to start an untrusted server
338 > */
339 > promptType?: 'only-new' | 'all-untrusted' | 'never';
340 > /** True if th servre should be launched with debugging. */
341 > debug?: boolean;
342 > /** Correlate multiple interactions such that any trust prompts are presented in combination. */
343 > interaction?: McpStartServerInteraction;
344 > /**
345 > * If true, throw an error if any user interaction would be required during startup.
346 > * This includes variable resolution, trust prompts, and authentication prompts.
347 > */
348 > errorOnUserInteraction?: boolean;
349 > }
350 >
351 > export namespace McpServerTrust {
352 > export const enum Kind {
353 > /** The server is trusted */
354 > Trusted,
355 > /** The server is trusted as long as its nonce matches */
356 > TrustedOnNonce,
357 > /** The server trust was denied. */
358 > Untrusted,
359 > /** The server is not yet trusted or untrusted. */
360 > Unknown,
361 > }
362 > }
363 >
364 > export interface IMcpServer extends IDisposable {
365 > readonly collection: McpCollectionReference;
366 > readonly definition: McpDefinitionReference;
367 > readonly enablement: IObservable<ContributionEnablementState>;
368 > readonly connection: IObservable<IMcpServerConnection | undefined>;
369 > readonly connectionState: IObservable<McpConnectionState>;
370 > readonly serverMetadata: IObservable<{
371 > serverName?: string;
372 > serverInstructions?: string;
373 > icons: IMcpIcons;
374 > } | undefined>;
375 >
376 > /**
377 > * Full definition as it exists in the MCP registry. Unlike the references
378 > * in `collection` and `definition`, this may change over time.
379 > */
380 > readDefinitions(): IObservable<{ server: McpServerDefinition | undefined; collection: McpCollectionDefinition | undefined }>;
381 >
382 > showOutput(preserveFocus?: boolean): Promise<void>;
383 > /**
384 > * Starts the server and returns its resulting state. One of:
385 > * - Running, if all good
386 > * - Error, if the server failed to start
387 > * - Stopped, if the server was disposed or the user cancelled the launch
388 > */
389 > start(opts?: IMcpServerStartOpts): Promise<McpConnectionState>;
390 > stop(): Promise<void>;
391 >
392 > readonly cacheState: IObservable<McpServerCacheState>;
393 > readonly tools: IObservable<readonly IMcpTool[]>;
394 > readonly prompts: IObservable<readonly IMcpPrompt[]>;
395 > readonly capabilities: IObservable<McpCapability | undefined>;
396 >
397 > /**
398 > * Lists all resources on the server.
399 > */
400 > resources(token?: CancellationToken): AsyncIterable<IMcpResource[]>;
401 >
402 > /**
403 > * List resource templates on the server.
404 > */
405 > resourceTemplates(token?: CancellationToken): Promise<IMcpResourceTemplate[]>;
406 > }
407 >
408 > /**
409 > * A representation of an MCP resource. The `uri` is namespaced to VS Code and
410 > * can be used in filesystem APIs.
411 > */
412 > export interface IMcpResource {
413 > /** Identifier for the file in VS Code and operable with filesystem API */
414 > readonly uri: URI;
415 > /** Identifier of the file as given from the MCP server. */
416 > readonly mcpUri: string;
417 > readonly name: string;
418 > readonly title?: string;
419 > readonly description?: string;
420 > readonly mimeType?: string;
421 > readonly sizeInBytes?: number;
422 > readonly icons: IMcpIcons;
423 > }
424 >
425 > export interface IMcpResourceTemplate {
426 > readonly name: string;
427 > readonly title?: string;
428 > readonly description?: string;
429 > readonly mimeType?: string;
430 > readonly template: UriTemplate;
431 > readonly icons: IMcpIcons;
432 >
433 > /** Gets string completions for the given template part. */
434 > complete(templatePart: string, prefix: string, alreadyResolved: Record<string, string | string[]>, token: CancellationToken): Promise<string[]>;
435 >
436 > /** Gets the resolved URI from template parts. */
437 > resolveURI(vars: Record<string, unknown>): URI;
438 > }
439 >
440 > export const isMcpResourceTemplate = (obj: IMcpResource | IMcpResourceTemplate): obj is IMcpResourceTemplate => {
441 return (obj as IMcpResourceTemplate).template !== undefined;
442 };
443 > export const isMcpResource = (obj: IMcpResource | IMcpResourceTemplate): obj is IMcpResource => { mcpTypes.ts
444 return (obj as IMcpResource).mcpUri !== undefined;
445 };
446 > mcpTypes.ts
447 > export const enum McpServerCacheState {
448 > /** Tools have not been read before */
449 > Unknown,
450 > /** Tools were read from the cache */
451 > Cached,
452 > /** Tools were read from the cache or live, but they may be outdated. */
453 > Outdated,
454 > /** Tools are refreshing for the first time */
455 > RefreshingFromUnknown,
456 > /** Tools are refreshing and the current tools are cached */
457 > RefreshingFromCached,
458 > /** Tool state is live, server is connected */
459 > Live,
460 > }
461 >
462 > export interface IMcpPrompt {
463 > readonly id: string;
464 > readonly name: string;
465 > readonly title?: string;
466 > readonly description?: string;
467 > readonly arguments: readonly MCP.PromptArgument[];
468 >
469 > /** Gets string completions for the given prompt part. */
470 > complete(argument: string, prefix: string, alreadyResolved: Record<string, string>, token: CancellationToken): Promise<string[]>;
471 >
472 > resolve(args: Record<string, string | undefined>, token?: CancellationToken): Promise<IMcpPromptMessage[]>;
473 > }
474 >
475 > export const mcpPromptReplaceSpecialChars = (s: string) => s.replace(/[^a-z0-9_.-]/gi, '_');
476 >
477 > export const mcpPromptPrefix = (definition: McpDefinitionReference) =>
478 `/mcp.` + mcpPromptReplaceSpecialChars(definition.label);
479 > mcpTypes.ts
480 > export interface IMcpPromptMessage extends MCP.PromptMessage { }
481 >
482 > export interface IMcpToolCallContext {
483 > chatSessionResource: URI | undefined;
484 > chatRequestId?: string;
485 > /**
486 > * Optional W3C trace context `traceparent` value to forward to the MCP server
487 > * via `_meta.traceparent` on the JSON-RPC `tools/call` request (MCP SEP-414).
488 > */
489 > traceparent?: string;
490 > /** Optional W3C trace context `tracestate` value paired with {@link traceparent}. */
491 > tracestate?: string;
492 > }
493 >
494 > /**
495 > * Visibility of an MCP tool, based on the MCP Apps `_meta.ui.visibility` field.
496 > * @see https://github.com/anthropics/mcp/blob/main/apps.md
497 > */
498 > export const enum McpToolVisibility {
499 > /** Tool is visible to and callable by the language model */
500 > Model = 1 << 0,
501 > /** Tool is callable by the MCP App UI */
502 > App = 1 << 1,
503 > }
504 >
505 > /**
506 > * Serializable data for MCP App UI rendering.
507 > * This contains all the information needed to render an MCP App webview.
508 > *
509 > * The transport for the App's sub-RPCs (`tools/call`, `resources/read`,
510 > * `sampling/createMessage`, …) is determined by the discriminator:
511 > *
512 > * - `local`: resolves the MCP server via {@link IMcpService} from
513 > * `serverDefinitionId` + `collectionId`. Used for locally-configured
514 > * MCP servers.
515 > * - `agentHost`: routes through {@link IAgentHostService.handleMcpRequest}
516 > * on the AHP `mcp://` side `channel`. Used for MCP servers owned by
517 > * an agent host.
518 > */
519 > export type IMcpToolCallUIData =
520 > | {
521 > readonly kind: 'local';
522 > /** URI of the UI resource for rendering (e.g., "ui://weather-server/dashboard") */
523 > readonly resourceUri: string;
524 > /** Reference to the server definition for reconnection */
525 > readonly serverDefinitionId: string;
526 > /** Reference to the collection containing the server */
527 > readonly collectionId: string;
528 > }
529 > | {
530 > readonly kind: 'agentHost';
531 > /** URI of the UI resource for rendering (e.g., "ui://weather-server/dashboard") */
532 > readonly resourceUri: string;
533 > /** AHP `mcp://` channel URI for the originating server. */
534 > readonly channel: string;
535 > /** Stable identifier for the originating server (used as webview origin key). */
536 > readonly serverId: string;
537 > };
538 >
539 > export interface IMcpTool {
540 >
541 > readonly id: string;
542 > /** Name for #referencing in chat */
543 > readonly referenceName: string;
544 > readonly icons: IMcpIcons;
545 > readonly definition: MCP.Tool;
546 > /** Visibility of the tool (Model, App, or both). Defaults to Model | App. */
547 > readonly visibility: McpToolVisibility;
548 > /** Optional UI resource URI for MCP App rendering */
549 > readonly uiResourceUri?: string;
550 >
551 > /**
552 > * Calls a tool
553 > * @throws {@link MpcResponseError} if the tool fails to execute
554 > * @throws {@link McpConnectionFailedError} if the connection to the server fails
555 > */
556 > call(params: Record<string, unknown>, context?: IMcpToolCallContext, token?: CancellationToken): Promise<MCP.CallToolResult>;
557 >
558 > /**
559 > * Identical to {@link call}, but reports progress.
560 > */
561 > callWithProgress(params: Record<string, unknown>, progress: ToolProgress, context?: IMcpToolCallContext, token?: CancellationToken): Promise<MCP.CallToolResult>;
562 > }
563 >
564 > export const enum McpServerTransportType {
565 > /** A command-line MCP server communicating over standard in/out */
566 > Stdio = 1 << 0,
567 > /** An MCP server that uses Server-Sent Events */
568 > HTTP = 1 << 1,
569 > }
570 >
571 > /**
572 > * MCP server launched on the command line which communicated over stdio.
573 > * https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#stdio
574 > */
575 > export interface McpServerTransportStdio {
576 > readonly type: McpServerTransportType.Stdio;
577 > readonly cwd: string | undefined;
578 > readonly command: string;
579 > readonly args: readonly string[];
580 > readonly env: Record<string, string | number | null>;
581 > readonly envFile: string | undefined;
582 > readonly sandbox: IMcpSandboxConfiguration | undefined;
583 > }
584 >
585 > export interface McpServerTransportHTTPAuthentication {
586 > /**
587 > * Authentication provider ID to use to get a session for the initial MCP server connection.
588 > */
589 > readonly providerId: string;
590 > /**
591 > * Scopes to use to get a session for the initial MCP server connection.
592 > */
593 > readonly scopes: string[];
594 > }
595 >
596 > export interface McpServerTransportHTTPOAuth {
597 > readonly clientId?: string;
598 > /**
599 > * (Preview) When true, the MCP server uses enterprise-managed authentication via the configured
600 > * SSO issuer (see `mcp.enterpriseManagedAuth.idp`). Tokens are obtained through OAuth Identity
601 > * Assertion Authorization Grant (ID-JAG) so that, after a one-time sign-in, subsequent enterprise-managed
602 > * servers connect silently.
603 > */
604 > readonly enterpriseManaged?: boolean;
605 > }
606 >
607 > /**
608 > * Returns the secret-storage key under which an MCP server OAuth client secret is stored.
609 > * Scoped by the MCP server URL AND the OAuth client_id so that two servers sharing the same
610 > * client_id string (e.g. against different authorization servers) cannot clobber each other's
611 > * secret, and so the key is stable across mcp.json configurations that happen to share a label
612 > * (e.g. user mcp.json vs. workspace mcp.json). Set by the "Set Client Secret" code lens in
613 > * mcp.json and read at authentication time so that client secrets are never stored in
614 > * plain-text config files.
615 > */
616 > export function mcpOAuthClientSecretStorageKey(mcpServerUrl: string, clientId: string): string {
617 return `mcp.oauth.clientSecret:${mcpServerUrl}:${clientId}`;
618 }
619 > mcpTypes.ts
620 > /**
621 > * MCP server launched on the command line which communicated over SSE or Streamable HTTP.
622 > * https://spec.modelcontextprotocol.io/specification/2024-11-05/basic/transports/#http-with-sse
623 > * https://modelcontextprotocol.io/specification/2025-03-26/basic/transports#streamable-http
624 > */
625 > export interface McpServerTransportHTTP {
626 > readonly type: McpServerTransportType.HTTP;
627 > readonly uri: URI;
628 > readonly headers: [string, string][];
629 > readonly oauth?: McpServerTransportHTTPOAuth;
630 > /**
631 > * @deprecated this was originally used for step-auth auth but a different approach was used instead
632 > * so it's effectively dead code.
633 > */
634 > readonly authentication?: McpServerTransportHTTPAuthentication;
635 > }
636 >
637 > export type McpServerLaunch =
638 > | McpServerTransportStdio
639 > | McpServerTransportHTTP;
640 >
641 > export namespace McpServerLaunch {
642 > export type Serialized =
643 > | { type: McpServerTransportType.HTTP; uri: UriComponents; headers: [string, string][]; oauth?: McpServerTransportHTTPOAuth; authentication?: McpServerTransportHTTPAuthentication }
644 > | { type: McpServerTransportType.Stdio; cwd: string | undefined; command: string; args: readonly string[]; env: Record<string, string | number | null>; envFile: string | undefined; sandbox: IMcpSandboxConfiguration | undefined };
645 >
646 > export function toSerialized(launch: McpServerLaunch): McpServerLaunch.Serialized {
647 return launch;
648 }
649 > mcpTypes.ts
650 > export function fromSerialized(launch: McpServerLaunch.Serialized): McpServerLaunch {
651 switch (launch.type) {
652 case McpServerTransportType.HTTP:
664 }
665 }
666 > mcpTypes.ts
667 > export async function hash(launch: McpServerLaunch): Promise<string> {
668 const nonce = await crypto.subtle.digest('SHA-256', new TextEncoder().encode(JSON.stringify(launch)));
669 return encodeHex(VSBuffer.wrap(new Uint8Array(nonce)));
670 }
671 > } mcpTypes.ts
672 >
673 > /**
674 > * An instance that manages a connection to an MCP server. It can be started,
675 > * stopped, and restarted. Once started and in a running state, it will
676 > * eventually build a {@link IMcpServerConnection.handler}.
677 > */
678 > export interface IMcpPotentialSandboxBlock {
679 > readonly kind: 'network' | 'filesystem';
680 > readonly message: string;
681 > readonly host?: string;
682 > readonly path?: string;
683 > }
684 >
685 > export interface IMcpServerConnection extends IDisposable {
686 > readonly definition: McpServerDefinition;
687 > readonly state: IObservable<McpConnectionState>;
688 > readonly handler: IObservable<McpServerRequestHandler | undefined>;
689 > readonly onPotentialSandboxBlock: Event<IMcpPotentialSandboxBlock>;
690 >
691 > /**
692 > * Resolved launch definition. Might not match the `definition.launch` due to
693 > * resolution logic in extension-provided MCPs.
694 > */
695 > readonly launchDefinition: McpServerLaunch;
696 >
697 > /**
698 > * Starts the server if it's stopped. Returns a promise that resolves once
699 > * server exits a 'starting' state.
700 > */
701 > start(methods: IMcpClientMethods): Promise<McpConnectionState>;
702 >
703 > /**
704 > * Stops the server.
705 > */
706 > stop(): Promise<void>;
707 > }
708 >
709 > /** Client methods whose implementations are passed through the server connection. */
710 > export interface IMcpClientMethods {
711 > /** Handler for `sampling/createMessage` */
712 > createMessageRequestHandler?(req: MCP.CreateMessageRequest['params'], token?: CancellationToken): Promise<MCP.CreateMessageResult>;
713 > /** Handler for `elicitation/create` */
714 > elicitationRequestHandler?(req: MCP.ElicitRequest['params'], token?: CancellationToken): Promise<MCP.ElicitResult>;
715 > }
716 >
717 > /**
718 > * McpConnectionState is the state of the underlying connection and is
719 > * communicated e.g. from the extension host to the renderer.
720 > */
721 > export namespace McpConnectionState {
722 > export const enum Kind {
723 > Stopped,
724 > Starting,
725 > Running,
726 > Error,
727 > }
728 >
729 > export const toString = (s: McpConnectionState): string => {
730 switch (s.state) {
731 case Kind.Stopped:
741 }
742 };
743 > mcpTypes.ts
744 > export const toKindString = (s: McpConnectionState.Kind): string => {
745 switch (s) {
746 case Kind.Stopped:
756 }
757 };
758 > mcpTypes.ts
759 > /** Returns if the MCP state is one where starting a new server is valid */
760 > export const canBeStarted = (s: Kind) => s === Kind.Error || s === Kind.Stopped;
761 >
762 > /** Gets whether the state is a running state. */
763 > export const isRunning = (s: McpConnectionState) => !canBeStarted(s.state);
764 >
765 > export interface Stopped {
766 > readonly state: Kind.Stopped;
767 > readonly reason?: 'needs-user-interaction';
768 > }
769 >
770 > export interface Starting {
771 > readonly state: Kind.Starting;
772 > }
773 >
774 > export interface Running {
775 > readonly state: Kind.Running;
776 > }
777 >
778 > export interface Error {
779 > readonly state: Kind.Error;
780 > readonly code?: string;
781 > readonly shouldRetry?: boolean;
782 > readonly message: string;
783 > }
784 > }
785 >
786 > export type McpConnectionState =
787 > | McpConnectionState.Stopped
788 > | McpConnectionState.Starting
789 > | McpConnectionState.Running
790 > | McpConnectionState.Error;
791 >
792 > export class MpcResponseError extends Error {
793 > constructor(message: string, public readonly code: number, public readonly data: unknown) {
794 super(`MPC ${code}: ${message}`);
795 }
796 > } mcpTypes.ts
797 >
798 > export class McpConnectionFailedError extends Error { }
799 >
800 > export class UserInteractionRequiredError extends Error {
801 > private static readonly prefix = 'User interaction required: ';
802 >
803 > public static is(error: Error): boolean {
804 return error.message.startsWith(this.prefix);
805 }
806 > mcpTypes.ts
807 > constructor(public readonly reason: string) {
808 super(`${UserInteractionRequiredError.prefix}${reason}`);
809 }
810 > } mcpTypes.ts
811 >
812 > export interface IMcpConfigPath {
813 > id: string;
814 > key: 'userLocalValue' | 'userRemoteValue' | 'workspaceValue' | 'workspaceFolderValue';
815 > label: string;
816 > scope: StorageScope;
817 > target: ConfigurationTarget;
818 > order: number;
819 > remoteAuthority?: string;
820 > uri: URI | undefined;
821 > section?: string[];
822 > workspaceFolder?: IWorkspaceFolder;
823 > }
824 >
825 > export interface IMcpServerContainer extends IDisposable {
826 > mcpServer: IWorkbenchMcpServer | null;
827 > update(): void;
828 > }
829 >
830 > export interface IMcpServerEditorOptions extends IEditorOptions {
831 > tab?: McpServerEditorTab;
832 > sideByside?: boolean;
833 > }
834 >
835 > export const enum McpServerEnablementState {
836 > Disabled,
837 > DisabledByAccess,
838 > DisabledProfile,
839 > DisabledWorkspace,
840 > Enabled,
841 > }
842 >
843 > export const enum McpServerInstallState {
844 > Installing,
845 > Installed,
846 > Uninstalling,
847 > Uninstalled
848 > }
849 >
850 > export const enum McpServerEditorTab {
851 > Readme = 'readme',
852 > Manifest = 'manifest',
853 > Configuration = 'configuration',
854 > }
855 >
856 > export type McpServerEnablementStatus = {
857 > readonly state: McpServerEnablementState;
858 > readonly message?: {
859 > readonly severity: Severity;
860 > readonly text: IMarkdownString;
861 > };
862 > };
863 >
864 > export interface IWorkbenchMcpServer {
865 > readonly gallery: IGalleryMcpServer | undefined;
866 > readonly local: IWorkbenchLocalMcpServer | undefined;
867 > readonly installable: IInstallableMcpServer | undefined;
868 > readonly installState: McpServerInstallState;
869 > readonly runtimeStatus: McpServerEnablementStatus | undefined;
870 > readonly id: string;
871 > readonly name: string;
872 > readonly label: string;
873 > readonly description: string;
874 > readonly icon?: {
875 > readonly dark: string;
876 > readonly light: string;
877 > };
878 > readonly codicon?: string;
879 > readonly publisherUrl?: string;
880 > readonly publisherDisplayName?: string;
881 > readonly starsCount?: number;
882 > readonly license?: string;
883 > readonly repository?: string;
884 > readonly config?: IMcpServerConfiguration | undefined;
885 > readonly readmeUrl?: URI;
886 > getReadme(token: CancellationToken): Promise<string>;
887 > getManifest(token: CancellationToken): Promise<IGalleryMcpServerConfiguration>;
888 > }
889 >
890 > export const IMcpWorkbenchService = createDecorator<IMcpWorkbenchService>('IMcpWorkbenchService');
891 > export interface IMcpWorkbenchService {
892 > readonly _serviceBrand: undefined;
893 > readonly onChange: Event<IWorkbenchMcpServer | undefined>;
894 > readonly onReset: Event<void>;
895 > readonly local: readonly IWorkbenchMcpServer[];
896 > getEnabledLocalMcpServers(): IWorkbenchLocalMcpServer[];
897 > queryLocal(): Promise<IWorkbenchMcpServer[]>;
898 > queryGallery(options?: IQueryOptions, token?: CancellationToken): Promise<IIterativePager<IWorkbenchMcpServer>>;
899 > canInstall(mcpServer: IWorkbenchMcpServer): true | IMarkdownString;
900 > install(server: IWorkbenchMcpServer, installOptions?: IWorkbencMcpServerInstallOptions): Promise<IWorkbenchMcpServer>;
901 > uninstall(mcpServer: IWorkbenchMcpServer): Promise<void>;
902 > getMcpConfigPath(arg: IWorkbenchLocalMcpServer): IMcpConfigPath | undefined;
903 > getMcpConfigPath(arg: URI): Promise<IMcpConfigPath | undefined>;
904 > openSearch(searchValue: string, preserveFocus?: boolean): Promise<void>;
905 > open(extension: IWorkbenchMcpServer | string, options?: IMcpServerEditorOptions): Promise<void>;
906 > }
907 >
908 > export class McpServerContainers extends Disposable {
909 > constructor(
910 private readonly containers: IMcpServerContainer[],
911 @IMcpWorkbenchService mcpWorkbenchService: IMcpWorkbenchService
914 this._register(mcpWorkbenchService.onChange(this.update, this));
915 }
916 > mcpTypes.ts
917 > set mcpServer(extension: IWorkbenchMcpServer | null) {
918 this.containers.forEach(c => c.mcpServer = extension);
919 }
920 > mcpTypes.ts
921 > update(server: IWorkbenchMcpServer | undefined): void {
922 for (const container of this.containers) {
923 if (server && container.mcpServer) {
930 }
931 }
932 > } mcpTypes.ts
933 >
934 > export const McpServersGalleryStatusContext = new RawContextKey<string>('mcpServersGalleryStatus', McpGalleryManifestStatus.Unavailable);
935 > export const HasInstalledMcpServersContext = new RawContextKey<boolean>('hasInstalledMcpServers', true);
936 > export const InstalledMcpServersViewId = 'workbench.views.mcp.installed';
937 >
938 > export namespace McpResourceURI {
939 > export const scheme = 'mcp-resource';
940 >
941 > // Random placeholder for empty authorities, otherwise they're represente as
942 > // `scheme//path/here` in the URI which would get normalized to `scheme/path/here`.
943 > const emptyAuthorityPlaceholder = 'dylo78gyp'; // chosen by a fair dice roll. Guaranteed to be random.
944 >
945 > export function fromServer(def: McpDefinitionReference, resourceURI: URI | string): URI {
946 if (typeof resourceURI === 'string') {
947 resourceURI = URI.parse(resourceURI);
953 });
954 }
955 > mcpTypes.ts
956 > export function toServer(uri: URI | string): { definitionId: string; resourceURL: URL } {
957 if (typeof uri === 'string') {
958 uri = URI.parse(uri);
978 };
979 }
980 > mcpTypes.ts
981 > }
982 >
983 > /** Warning: this enum is cached in `mcpServer.ts` and all changes MUST only be additive. */
984 > export const enum McpCapability {
985 > Logging = 1 << 0,
986 > Completions = 1 << 1,
987 > Prompts = 1 << 2,
988 > PromptsListChanged = 1 << 3,
989 > Resources = 1 << 4,
990 > ResourcesSubscribe = 1 << 5,
991 > ResourcesListChanged = 1 << 6,
992 > Tools = 1 << 7,
993 > ToolsListChanged = 1 << 8,
994 > }
995 >
996 > export interface ISamplingOptions {
997 > server: IMcpServer;
998 > isDuringToolCall: boolean;
999 > params: MCP.CreateMessageRequest['params'];
1000 > }
1001 >
1002 > export interface ISamplingResult {
1003 > sample: MCP.CreateMessageResult;
1004 > }
1005 >
1006 > export interface IMcpSamplingService {
1007 > _serviceBrand: undefined;
1008 >
1009 > sample(opts: ISamplingOptions, token?: CancellationToken): Promise<ISamplingResult>;
1010 >
1011 > /** Whether MCP sampling logs are available for this server */
1012 > hasLogs(server: IMcpServer): boolean;
1013 > /** Gets a text report of the MCP server's sampling usage */
1014 > getLogText(server: IMcpServer): string;
1015 >
1016 > getConfig(server: IMcpServer): IMcpServerSamplingConfiguration;
1017 > updateConfig(server: IMcpServer, mutate: (r: IMcpServerSamplingConfiguration) => unknown): Promise<IMcpServerSamplingConfiguration>;
1018 > }
1019 >
1020 > export const IMcpSamplingService = createDecorator<IMcpSamplingService>('IMcpServerSampling');
1021 >
1022 > export class McpError extends Error {
1023 > public static methodNotFound(method: string) {
1024 > return new McpError(MCP.METHOD_NOT_FOUND, `Method not found: ${method}`);
1025 > }
1026 >
1027 > public static notAllowed() {
1028 return new McpError(-32000, 'The user has denied permission to call this method.');
1029 }
1030 > mcpTypes.ts
1031 > public static unknown(e: Error) {
1032 const mcpError = new McpError(MCP.INTERNAL_ERROR, `Unknown error: ${e.stack}`);
1033 mcpError.cause = e;
1034 return mcpError;
1035 }
1036 > mcpTypes.ts
1037 > constructor(
1038 public readonly code: number,
1039 message: string,
1042 super(message);
1043 }
1044 > } mcpTypes.ts
1045 >
1046 > export const enum McpToolName {
1047 > Prefix = 'mcp_',
1048 > MaxPrefixLen = 18,
1049 > MaxLength = 64,
1050 > }
1051 >
1052 >
1053 > export interface IMcpElicitationService {
1054 > _serviceBrand: undefined;
1055 >
1056 > /**
1057 > * Elicits a response from the user. The `context` is optional and can be used
1058 > * to provide additional information about the request.
1059 > *
1060 > * @param context Context for the elicitation, e.g. chat session ID.
1061 > * @param elicitation Request to elicit a response.
1062 > * @returns A promise that resolves to an {@link ElicitationResult}.
1063 > */
1064 > elicit(server: IMcpServer, context: IMcpToolCallContext | undefined, elicitation: MCP.ElicitRequest['params'], token: CancellationToken): Promise<ElicitResult>;
1065 > }
1066 >
1067 > export const enum ElicitationKind {
1068 > Form,
1069 > URL,
1070 > }
1071 >
1072 > export interface IUrlModeElicitResult extends IDisposable {
1073 > kind: ElicitationKind.URL;
1074 > value: MCP.ElicitResult;
1075 > /**
1076 > * Waits until the server tells us the elicitation is completed before resolving.
1077 > * Rejects with a CancellationError if the server stops before elicitation is
1078 > * complete, or if the token is cancelled.
1079 > */
1080 > wait: Promise<void>;
1081 > }
1082 >
1083 > export interface IFormModeElicitResult extends IDisposable {
1084 > kind: ElicitationKind.Form;
1085 > value: MCP.ElicitResult;
1086 > }
1087 >
1088 > export type ElicitResult = IUrlModeElicitResult | IFormModeElicitResult;
1089 >
1090 > export const IMcpElicitationService = createDecorator<IMcpElicitationService>('IMcpElicitationService');
1091 >
1092 > export const McpToolResourceLinkMimeType = 'application/vnd.code.resource-link';
1093 >
1094 > export interface IMcpToolResourceLinkContents {
1095 > uri: UriComponents;
1096 > underlyingMimeType?: string;
1097 > }
1098 >
1099 > export interface IMcpIcons {
1100 > /** Gets the image URI appropriate to the approximate display size */
1101 > getUrl(size: number): { dark: URI; light?: URI } | undefined;
1102 > }
src/vs/workbench/contrib/notebook/common/notebookCommon.ts 925 covered LOC · 26 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- notebookCommon.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer } from '../../../../base/common/buffer.js';
7 > import { CancellationToken } from '../../../../base/common/cancellation.js';
8 > import { IDiffResult } from '../../../../base/common/diff/diff.js';
9 > import { Event } from '../../../../base/common/event.js';
10 > import * as glob from '../../../../base/common/glob.js';
11 > import { IMarkdownString } from '../../../../base/common/htmlContent.js';
12 > import { Iterable } from '../../../../base/common/iterator.js';
13 > import { IDisposable } from '../../../../base/common/lifecycle.js';
14 > import { Mimes } from '../../../../base/common/mime.js';
15 > import { Schemas } from '../../../../base/common/network.js';
16 > import { basename } from '../../../../base/common/path.js';
17 > import { isWindows } from '../../../../base/common/platform.js';
18 > import { ISplice } from '../../../../base/common/sequence.js';
19 > import { ThemeColor } from '../../../../base/common/themables.js';
20 > import { URI, UriComponents } from '../../../../base/common/uri.js';
21 > import { Range } from '../../../../editor/common/core/range.js';
22 > import * as editorCommon from '../../../../editor/common/editorCommon.js';
23 > import { Command, WorkspaceEditMetadata } from '../../../../editor/common/languages.js';
24 > import { IReadonlyTextBuffer, ITextModel } from '../../../../editor/common/model.js';
25 > import { IAccessibilityInformation } from '../../../../platform/accessibility/common/accessibility.js';
26 > import { RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
27 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
28 > import { IFileReadLimits } from '../../../../platform/files/common/files.js';
29 > import { UndoRedoGroup } from '../../../../platform/undoRedo/common/undoRedo.js';
30 > import { IRevertOptions, ISaveOptions, IUntypedEditorInput } from '../../../common/editor.js';
31 > import { NotebookTextModel } from './model/notebookTextModel.js';
32 > import { ICellExecutionError } from './notebookExecutionStateService.js';
33 > import { INotebookTextModelLike } from './notebookKernelService.js';
34 > import { ICellRange } from './notebookRange.js';
35 > import { RegisteredEditorPriority } from '../../../services/editor/common/editorResolverService.js';
36 > import { generateMetadataUri, generate as generateUri, extractCellOutputDetails, parseMetadataUri, parse as parseUri } from '../../../services/notebook/common/notebookDocumentService.js';
37 > import { IWorkingCopyBackupMeta, IWorkingCopySaveEvent } from '../../../services/workingCopy/common/workingCopy.js';
38 > import { SnapshotContext } from '../../../services/workingCopy/common/fileWorkingCopy.js';
39 >
40 > export const NOTEBOOK_EDITOR_ID = 'workbench.editor.notebook';
41 > export const NOTEBOOK_DIFF_EDITOR_ID = 'workbench.editor.notebookTextDiffEditor';
42 > export const NOTEBOOK_MULTI_DIFF_EDITOR_ID = 'workbench.editor.notebookMultiTextDiffEditor';
43 > export const INTERACTIVE_WINDOW_EDITOR_ID = 'workbench.editor.interactive';
44 > export const REPL_EDITOR_ID = 'workbench.editor.repl';
45 > export const NOTEBOOK_OUTPUT_EDITOR_ID = 'workbench.editor.notebookOutputEditor';
46 >
47 > export const EXECUTE_REPL_COMMAND_ID = 'replNotebook.input.execute';
48 >
49 > export enum CellKind {
50 > Markup = 1,
51 > Code = 2
52 > }
53 >
54 > export const NOTEBOOK_DISPLAY_ORDER: readonly string[] = [
55 > 'application/json',
56 > 'application/javascript',
57 > 'text/html',
58 > 'image/svg+xml',
59 > Mimes.latex,
60 > Mimes.markdown,
61 > 'image/png',
62 > 'image/jpeg',
63 > Mimes.text
64 > ];
65 >
66 > export const ACCESSIBLE_NOTEBOOK_DISPLAY_ORDER: readonly string[] = [
67 > Mimes.latex,
68 > Mimes.markdown,
69 > 'application/json',
70 > 'text/html',
71 > 'image/svg+xml',
72 > 'image/png',
73 > 'image/jpeg',
74 > Mimes.text,
75 > ];
76 >
77 > /**
78 > * A mapping of extension IDs who contain renderers, to notebook ids who they
79 > * should be treated as the same in the renderer selection logic. This is used
80 > * to prefer the 1st party Jupyter renderers even though they're in a separate
81 > * extension, for instance. See #136247.
82 > */
83 > export const RENDERER_EQUIVALENT_EXTENSIONS: ReadonlyMap<string, ReadonlySet<string>> = new Map([
84 > ['ms-toolsai.jupyter', new Set(['jupyter-notebook', 'interactive'])],
85 > ['ms-toolsai.jupyter-renderers', new Set(['jupyter-notebook', 'interactive'])],
86 > ]);
87 >
88 > export const RENDERER_NOT_AVAILABLE = '_notAvailable';
89 >
90 > export type ContributedNotebookRendererEntrypoint = string | { readonly extends: string; readonly path: string };
91 >
92 > export enum NotebookRunState {
93 > Running = 1,
94 > Idle = 2
95 > }
96 >
97 > export type NotebookDocumentMetadata = Record<string, unknown>;
98 >
99 > export enum NotebookCellExecutionState {
100 > Unconfirmed = 1,
101 > Pending = 2,
102 > Executing = 3
103 > }
104 > export enum NotebookExecutionState {
105 > Unconfirmed = 1,
106 > Pending = 2,
107 > Executing = 3
108 > }
109 >
110 > export interface INotebookCellPreviousExecutionResult {
111 > executionOrder?: number;
112 > success?: boolean;
113 > duration?: number;
114 > }
115 >
116 > export interface NotebookCellMetadata {
117 > /**
118 > * custom metadata
119 > */
120 > [key: string]: unknown;
121 > }
122 >
123 > export interface NotebookCellInternalMetadata {
124 > /**
125 > * Used only for diffing of Notebooks.
126 > * This is not persisted and generally useful only when diffing two notebooks.
127 > * Useful only after we've manually matched a few cells together so we know which cells are matching.
128 > */
129 > internalId?: string;
130 > executionId?: string;
131 > executionOrder?: number;
132 > lastRunSuccess?: boolean;
133 > runStartTime?: number;
134 > runStartTimeAdjustment?: number;
135 > runEndTime?: number;
136 > renderDuration?: { [key: string]: number };
137 > error?: ICellExecutionError;
138 > }
139 >
140 > export interface NotebookCellCollapseState {
141 > inputCollapsed?: boolean;
142 > outputCollapsed?: boolean;
143 > }
144 >
145 > export interface NotebookCellDefaultCollapseConfig {
146 > codeCell?: NotebookCellCollapseState;
147 > markupCell?: NotebookCellCollapseState;
148 > }
149 >
150 > export type InteractiveWindowCollapseCodeCells = 'always' | 'never' | 'fromEditor';
151 >
152 > export type TransientCellMetadata = { readonly [K in keyof NotebookCellMetadata]?: boolean };
153 > export type CellContentMetadata = { readonly [K in keyof NotebookCellMetadata]?: boolean };
154 > export type TransientDocumentMetadata = { readonly [K in keyof NotebookDocumentMetadata]?: boolean };
155 >
156 > export interface TransientOptions {
157 > readonly transientOutputs: boolean;
158 > readonly transientCellMetadata: TransientCellMetadata;
159 > readonly transientDocumentMetadata: TransientDocumentMetadata;
160 > readonly cellContentMetadata: CellContentMetadata;
161 > }
162 >
163 > /** Note: enum values are used for sorting */
164 > export const enum NotebookRendererMatch {
165 > /** Renderer has a hard dependency on an available kernel */
166 > WithHardKernelDependency = 0,
167 > /** Renderer works better with an available kernel */
168 > WithOptionalKernelDependency = 1,
169 > /** Renderer is kernel-agnostic */
170 > Pure = 2,
171 > /** Renderer is for a different mimeType or has a hard dependency which is unsatisfied */
172 > Never = 3,
173 > }
174 >
175 > /**
176 > * Renderer messaging requirement. While this allows for 'optional' messaging,
177 > * VS Code effectively treats it the same as true right now. "Partial
178 > * activation" of extensions is a very tricky problem, which could allow
179 > * solving this. But for now, optional is mostly only honored for aznb.
180 > */
181 > export const enum RendererMessagingSpec {
182 > Always = 'always',
183 > Never = 'never',
184 > Optional = 'optional',
185 > }
186 >
187 > export type NotebookRendererEntrypoint = { readonly extends: string | undefined; readonly path: URI };
188 >
189 > export interface INotebookRendererInfo {
190 > readonly id: string;
191 > readonly displayName: string;
192 > readonly entrypoint: NotebookRendererEntrypoint;
193 > readonly extensionLocation: URI;
194 > readonly extensionId: ExtensionIdentifier;
195 > readonly messaging: RendererMessagingSpec;
196 >
197 > readonly mimeTypes: readonly string[];
198 >
199 > readonly isBuiltin: boolean;
200 >
201 > matchesWithoutKernel(mimeType: string): NotebookRendererMatch;
202 > matches(mimeType: string, kernelProvides: ReadonlyArray<string>): NotebookRendererMatch;
203 > }
204 >
205 > export interface INotebookStaticPreloadInfo {
206 > readonly type: string;
207 > readonly entrypoint: URI;
208 > readonly extensionLocation: URI;
209 > readonly localResourceRoots: readonly URI[];
210 > }
211 >
212 > export interface IOrderedMimeType {
213 > mimeType: string;
214 > rendererId: string;
215 > isTrusted: boolean;
216 > }
217 >
218 > export interface IOutputItemDto {
219 > readonly mime: string;
220 > readonly data: VSBuffer;
221 > }
222 >
223 > export interface IOutputDto {
224 > outputs: IOutputItemDto[];
225 > outputId: string;
226 > metadata?: Record<string, any>;
227 > }
228 >
229 > export interface ICellOutput {
230 > readonly versionId: number;
231 > outputs: IOutputItemDto[];
232 > metadata?: Record<string, any>;
233 > outputId: string;
234 > /**
235 > * Alternative output id that's reused when the output is updated.
236 > */
237 > alternativeOutputId: string;
238 > readonly onDidChangeData: Event<void>;
239 > replaceData(items: IOutputDto): void;
240 > appendData(items: IOutputItemDto[]): void;
241 > appendedSinceVersion(versionId: number, mime: string): VSBuffer | undefined;
242 > asDto(): IOutputDto;
243 > bumpVersion(): void;
244 > dispose(): void;
245 > }
246 >
247 > export interface CellInternalMetadataChangedEvent {
248 > readonly lastRunSuccessChanged?: boolean;
249 > }
250 >
251 > export interface INotebookDocumentMetadataTextModel {
252 > /**
253 > * Notebook Metadata Uri.
254 > */
255 > readonly uri: URI;
256 > /**
257 > * Triggered when the Notebook Metadata changes.
258 > */
259 > readonly onDidChange: Event<void>;
260 > readonly metadata: Readonly<NotebookDocumentMetadata>;
261 > readonly textBuffer: IReadonlyTextBuffer;
262 > /**
263 > * Text representation of the Notebook Metadata
264 > */
265 > getValue(): string;
266 > getHash(): string;
267 > }
268 >
269 > export interface ICell {
270 > readonly uri: URI;
271 > handle: number;
272 > language: string;
273 > cellKind: CellKind;
274 > outputs: ICellOutput[];
275 > metadata: NotebookCellMetadata;
276 > internalMetadata: NotebookCellInternalMetadata;
277 > getHashValue(): number;
278 > textBuffer: IReadonlyTextBuffer;
279 > textModel?: ITextModel;
280 > readonly onDidChangeTextModel: Event<void>;
281 > getValue(): string;
282 > readonly onDidChangeOutputs?: Event<NotebookCellOutputsSplice>;
283 > readonly onDidChangeOutputItems?: Event<void>;
284 > readonly onDidChangeLanguage: Event<string>;
285 > readonly onDidChangeMetadata: Event<void>;
286 > readonly onDidChangeInternalMetadata: Event<CellInternalMetadataChangedEvent>;
287 > }
288 >
289 > export interface INotebookSnapshotOptions {
290 > context: SnapshotContext;
291 > outputSizeLimit: number;
292 > transientOptions?: TransientOptions;
293 > }
294 >
295 > export interface INotebookTextModel extends INotebookTextModelLike, IDisposable {
296 > readonly notebookType: string;
297 > readonly viewType: string;
298 > metadata: NotebookDocumentMetadata;
299 > readonly transientOptions: TransientOptions;
300 > readonly uri: URI;
301 > readonly versionId: number;
302 > readonly length: number;
303 > readonly cells: readonly ICell[];
304 > reset(cells: ICellDto2[], metadata: NotebookDocumentMetadata, transientOptions: TransientOptions): void;
305 > createSnapshot(options: INotebookSnapshotOptions): NotebookData;
306 > restoreSnapshot(snapshot: NotebookData, transientOptions?: TransientOptions): void;
307 > applyEdits(rawEdits: ICellEditOperation[], synchronous: boolean, beginSelectionState: ISelectionState | undefined, endSelectionsComputer: () => ISelectionState | undefined, undoRedoGroup: UndoRedoGroup | undefined, computeUndoRedo?: boolean): boolean;
308 > readonly onDidChangeContent: Event<NotebookTextModelChangedEvent>;
309 > readonly onWillDispose: Event<void>;
310 > }
311 >
312 > export type NotebookCellTextModelSplice<T> = [
313 > start: number,
314 > deleteCount: number,
315 > newItems: T[]
316 > ];
317 >
318 > export type NotebookCellOutputsSplice = {
319 > start: number /* start */;
320 > deleteCount: number /* delete count */;
321 > newOutputs: ICellOutput[];
322 > };
323 >
324 > export interface IMainCellDto {
325 > handle: number;
326 > url: string;
327 > source: string[];
328 > eol: string;
329 > versionId: number;
330 > language: string;
331 > cellKind: CellKind;
332 > outputs: IOutputDto[];
333 > metadata?: NotebookCellMetadata;
334 > internalMetadata?: NotebookCellInternalMetadata;
335 > }
336 >
337 > export enum NotebookCellsChangeType {
338 > ModelChange = 1,
339 > Move = 2,
340 > ChangeCellLanguage = 5,
341 > Initialize = 6,
342 > ChangeCellMetadata = 7,
343 > Output = 8,
344 > OutputItem = 9,
345 > ChangeCellContent = 10,
346 > ChangeDocumentMetadata = 11,
347 > ChangeCellInternalMetadata = 12,
348 > ChangeCellMime = 13,
349 > Unknown = 100
350 > }
351 >
352 > export interface NotebookCellsInitializeEvent<T> {
353 > readonly kind: NotebookCellsChangeType.Initialize;
354 > readonly changes: NotebookCellTextModelSplice<T>[];
355 > }
356 >
357 > export interface NotebookCellContentChangeEvent {
358 > readonly kind: NotebookCellsChangeType.ChangeCellContent;
359 > readonly index: number;
360 > }
361 >
362 > export interface NotebookCellsModelChangedEvent<T> {
363 > readonly kind: NotebookCellsChangeType.ModelChange;
364 > readonly changes: NotebookCellTextModelSplice<T>[];
365 > }
366 >
367 > export interface NotebookCellsModelMoveEvent<T> {
368 > readonly kind: NotebookCellsChangeType.Move;
369 > readonly index: number;
370 > readonly length: number;
371 > readonly newIdx: number;
372 > readonly cells: T[];
373 > }
374 >
375 > export interface NotebookOutputChangedEvent {
376 > readonly kind: NotebookCellsChangeType.Output;
377 > readonly index: number;
378 > readonly outputs: IOutputDto[];
379 > readonly append: boolean;
380 > }
381 >
382 > export interface NotebookOutputItemChangedEvent {
383 > readonly kind: NotebookCellsChangeType.OutputItem;
384 > readonly index: number;
385 > readonly outputId: string;
386 > readonly outputItems: IOutputItemDto[];
387 > readonly append: boolean;
388 > }
389 >
390 > export interface NotebookCellsChangeLanguageEvent {
391 > readonly kind: NotebookCellsChangeType.ChangeCellLanguage;
392 > readonly index: number;
393 > readonly language: string;
394 > }
395 >
396 > export interface NotebookCellsChangeMimeEvent {
397 > readonly kind: NotebookCellsChangeType.ChangeCellMime;
398 > readonly index: number;
399 > readonly mime: string | undefined;
400 > }
401 >
402 > export interface NotebookCellsChangeMetadataEvent {
403 > readonly kind: NotebookCellsChangeType.ChangeCellMetadata;
404 > readonly index: number;
405 > readonly metadata: NotebookCellMetadata;
406 > }
407 >
408 > export interface NotebookCellsChangeInternalMetadataEvent {
409 > readonly kind: NotebookCellsChangeType.ChangeCellInternalMetadata;
410 > readonly index: number;
411 > readonly internalMetadata: NotebookCellInternalMetadata;
412 > }
413 >
414 > export interface NotebookDocumentChangeMetadataEvent {
415 > readonly kind: NotebookCellsChangeType.ChangeDocumentMetadata;
416 > readonly metadata: NotebookDocumentMetadata;
417 > }
418 >
419 > export interface NotebookDocumentUnknownChangeEvent {
420 > readonly kind: NotebookCellsChangeType.Unknown;
421 > }
422 >
423 > export type NotebookRawContentEventDto = NotebookCellsInitializeEvent<IMainCellDto> | NotebookDocumentChangeMetadataEvent | NotebookCellContentChangeEvent | NotebookCellsModelChangedEvent<IMainCellDto> | NotebookCellsModelMoveEvent<IMainCellDto> | NotebookOutputChangedEvent | NotebookOutputItemChangedEvent | NotebookCellsChangeLanguageEvent | NotebookCellsChangeMimeEvent | NotebookCellsChangeMetadataEvent | NotebookCellsChangeInternalMetadataEvent | NotebookDocumentUnknownChangeEvent;
424 >
425 > export type NotebookCellsChangedEventDto = {
426 > readonly rawEvents: NotebookRawContentEventDto[];
427 > readonly versionId: number;
428 > };
429 >
430 > export type NotebookRawContentEvent = (NotebookCellsInitializeEvent<ICell> | NotebookDocumentChangeMetadataEvent | NotebookCellContentChangeEvent | NotebookCellsModelChangedEvent<ICell> | NotebookCellsModelMoveEvent<ICell> | NotebookOutputChangedEvent | NotebookOutputItemChangedEvent | NotebookCellsChangeLanguageEvent | NotebookCellsChangeMimeEvent | NotebookCellsChangeMetadataEvent | NotebookCellsChangeInternalMetadataEvent | NotebookDocumentUnknownChangeEvent) & { transient: boolean };
431 >
432 > export enum SelectionStateType {
433 > Handle = 0,
434 > Index = 1
435 > }
436 >
437 > export interface ISelectionHandleState {
438 > kind: SelectionStateType.Handle;
439 > primary: number | null;
440 > selections: number[];
441 > }
442 >
443 > export interface ISelectionIndexState {
444 > kind: SelectionStateType.Index;
445 > focus: ICellRange;
446 > selections: ICellRange[];
447 > }
448 >
449 > export type ISelectionState = ISelectionHandleState | ISelectionIndexState;
450 >
451 > export type NotebookTextModelChangedEvent = {
452 > readonly rawEvents: NotebookRawContentEvent[];
453 > readonly versionId: number;
454 > readonly synchronous: boolean | undefined;
455 > readonly endSelectionState: ISelectionState | undefined;
456 > };
457 >
458 > export type NotebookTextModelWillAddRemoveEvent = {
459 > readonly rawEvent: NotebookCellsModelChangedEvent<ICell>;
460 > };
461 >
462 > export const enum CellEditType {
463 > Replace = 1,
464 > Output = 2,
465 > Metadata = 3,
466 > CellLanguage = 4,
467 > DocumentMetadata = 5,
468 > Move = 6,
469 > OutputItems = 7,
470 > PartialMetadata = 8,
471 > PartialInternalMetadata = 9,
472 > }
473 >
474 > export interface ICellDto2 {
475 > source: string;
476 > language: string;
477 > mime: string | undefined;
478 > cellKind: CellKind;
479 > outputs: IOutputDto[];
480 > metadata?: NotebookCellMetadata;
481 > internalMetadata?: NotebookCellInternalMetadata;
482 > collapseState?: NotebookCellCollapseState;
483 > }
484 >
485 > export interface ICellReplaceEdit {
486 > editType: CellEditType.Replace;
487 > index: number;
488 > count: number;
489 > cells: ICellDto2[];
490 > }
491 >
492 > export interface ICellOutputEdit {
493 > editType: CellEditType.Output;
494 > index: number;
495 > outputs: IOutputDto[];
496 > append?: boolean;
497 > }
498 >
499 > export interface ICellOutputEditByHandle {
500 > editType: CellEditType.Output;
501 > handle: number;
502 > outputs: IOutputDto[];
503 > append?: boolean;
504 > }
505 >
506 > export interface ICellOutputItemEdit {
507 > editType: CellEditType.OutputItems;
508 > outputId: string;
509 > items: IOutputItemDto[];
510 > append?: boolean;
511 > }
512 >
513 > export interface ICellMetadataEdit {
514 > editType: CellEditType.Metadata;
515 > index: number;
516 > metadata: NotebookCellMetadata;
517 > }
518 >
519 > // These types are nullable because we need to use 'null' on the EH side so it is JSON-stringified
520 > export type NullablePartialNotebookCellMetadata = {
521 > [Key in keyof Partial<NotebookCellMetadata>]: NotebookCellMetadata[Key] | null
522 > };
523 >
524 > export interface ICellPartialMetadataEdit {
525 > editType: CellEditType.PartialMetadata;
526 > index: number;
527 > metadata: NullablePartialNotebookCellMetadata;
528 > }
529 >
530 > export interface ICellPartialMetadataEditByHandle {
531 > editType: CellEditType.PartialMetadata;
532 > handle: number;
533 > metadata: NullablePartialNotebookCellMetadata;
534 > }
535 >
536 > export type NullablePartialNotebookCellInternalMetadata = {
537 > [Key in keyof Partial<NotebookCellInternalMetadata>]: NotebookCellInternalMetadata[Key] | null
538 > };
539 > export interface ICellPartialInternalMetadataEdit {
540 > editType: CellEditType.PartialInternalMetadata;
541 > index: number;
542 > internalMetadata: NullablePartialNotebookCellInternalMetadata;
543 > }
544 >
545 > export interface ICellPartialInternalMetadataEditByHandle {
546 > editType: CellEditType.PartialInternalMetadata;
547 > handle: number;
548 > internalMetadata: NullablePartialNotebookCellInternalMetadata;
549 > }
550 >
551 > export interface ICellLanguageEdit {
552 > editType: CellEditType.CellLanguage;
553 > index: number;
554 > language: string;
555 > }
556 >
557 > export interface IDocumentMetadataEdit {
558 > editType: CellEditType.DocumentMetadata;
559 > metadata: NotebookDocumentMetadata;
560 > }
561 >
562 > export interface ICellMoveEdit {
563 > editType: CellEditType.Move;
564 > index: number;
565 > length: number;
566 > newIdx: number;
567 > }
568 >
569 > export type IImmediateCellEditOperation = ICellOutputEditByHandle | ICellPartialMetadataEditByHandle | ICellOutputItemEdit | ICellPartialInternalMetadataEdit | ICellPartialInternalMetadataEditByHandle | ICellPartialMetadataEdit;
570 > export type ICellEditOperation = IImmediateCellEditOperation | ICellReplaceEdit | ICellOutputEdit | ICellMetadataEdit | ICellPartialMetadataEdit | ICellPartialInternalMetadataEdit | IDocumentMetadataEdit | ICellMoveEdit | ICellOutputItemEdit | ICellLanguageEdit;
571 >
572 >
573 > export interface IWorkspaceNotebookCellEdit {
574 > metadata?: WorkspaceEditMetadata;
575 > resource: URI;
576 > notebookVersionId: number | undefined;
577 > cellEdit: ICellPartialMetadataEdit | IDocumentMetadataEdit | ICellReplaceEdit;
578 > }
579 >
580 > export interface IWorkspaceNotebookCellEditDto {
581 > metadata?: WorkspaceEditMetadata;
582 > resource: URI;
583 > notebookVersionId: number | undefined;
584 > cellEdit: ICellPartialMetadataEdit | IDocumentMetadataEdit | ICellReplaceEdit;
585 > }
586 >
587 > export interface NotebookData {
588 > readonly cells: ICellDto2[];
589 > readonly metadata: NotebookDocumentMetadata;
590 > }
591 >
592 >
593 > export interface INotebookContributionData {
594 > extension?: ExtensionIdentifier;
595 > providerDisplayName: string;
596 > displayName: string;
597 > filenamePattern: (string | glob.IRelativePattern | INotebookExclusiveDocumentFilter)[];
598 > priority?: RegisteredEditorPriority;
599 > }
600 >
601 > export namespace NotebookMetadataUri {
602 > export const scheme = Schemas.vscodeNotebookMetadata;
603 > export function generate(notebook: URI): URI {
604 return generateMetadataUri(notebook);
605 }
606 > export function parse(metadata: URI): URI | undefined { notebookCommon.ts
607 return parseMetadataUri(metadata);
608 }
610 >
611 > export namespace CellUri {
612 > export const scheme = Schemas.vscodeNotebookCell;
613 > export function generate(notebook: URI, handle: number): URI {
614 return generateUri(notebook, handle);
615 }
617 > export function parse(cell: URI): { notebook: URI; handle: number } | undefined {
618 return parseUri(cell);
619 }
621 > /**
622 > * Generates a URI for a cell output in a notebook using the output ID.
623 > * Used when URI should be opened as text in the editor.
624 > */
625 > export function generateCellOutputUriWithId(notebook: URI, outputId?: string) {
626 return notebook.with({
627 scheme: Schemas.vscodeNotebookCellOutput,
633 });
634 }
635 > /** notebookCommon.ts
636 > * Generates a URI for a cell output in a notebook using the output index.
637 > * Used when URI should be opened in notebook editor.
638 > */
639 > export function generateCellOutputUriWithIndex(notebook: URI, cellUri: URI, outputIndex: number): URI {
640 return notebook.with({
641 scheme: Schemas.vscodeNotebookCellOutput,
647 });
648 }
650 > export function generateOutputEditorUri(notebook: URI, cellId: string, cellIndex: number, outputId: string, outputIndex: number): URI {
651 return notebook.with({
652 scheme: Schemas.vscodeNotebookCellOutput,
660 });
661 }
663 > export function parseCellOutputUri(uri: URI): { notebook: URI; openIn: string; outputId?: string; cellFragment?: string; outputIndex?: number; cellHandle?: number; cellIndex?: number } | undefined {
664 return extractCellOutputDetails(uri);
665 }
667 > export function generateCellPropertyUri(notebook: URI, handle: number, scheme: string): URI {
668 return CellUri.generate(notebook, handle).with({ scheme: scheme });
669 }
671 > export function parseCellPropertyUri(uri: URI, propertyScheme: string) {
672 if (uri.scheme !== propertyScheme) {
673 return undefined;
676 return CellUri.parse(uri.with({ scheme: scheme }));
677 }
679 >
680 > const normalizeSlashes = (str: string) => isWindows ? str.replace(/\//g, '\\') : str;
681 >
682 > interface IMimeTypeWithMatcher {
683 > pattern: string;
684 > matches: glob.ParsedPattern;
685 > }
686 >
687 > export class MimeTypeDisplayOrder {
688 > private readonly order: IMimeTypeWithMatcher[];
689 >
690 > constructor(
691 initialValue: readonly string[] = [],
692 private readonly defaultOrder = NOTEBOOK_DISPLAY_ORDER,
697 }));
698 }
700 > /**
701 > * Returns a sorted array of the input mimeTypes.
702 > */
703 > public sort(mimeTypes: Iterable<string>): string[] {
704 const remaining = new Map(Iterable.map(mimeTypes, m => [m, normalizeSlashes(m)]));
705 let sorted: string[] = [];
723 return sorted;
724 }
726 > /**
727 > * Records that the user selected the given mimetype over the other
728 > * possible mimeTypes, prioritizing it for future reference.
729 > */
730 > public prioritize(chosenMimetype: string, otherMimeTypes: readonly string[]) {
731 const chosenIndex = this.findIndex(chosenMimetype);
732 if (chosenIndex === -1) {
747 }
748 }
750 > /**
751 > * Gets an array of in-order mimetype preferences.
752 > */
753 > public toArray() {
754 return this.order.map(o => o.pattern);
755 }
757 > private findIndex(mimeType: string, maxIndex = this.order.length) {
758 const normalized = normalizeSlashes(mimeType);
759 for (let i = 0; i < maxIndex; i++) {
765 return -1;
766 }
768 >
769 > interface IMutableSplice<T> extends ISplice<T> {
770 > readonly toInsert: T[];
771 > deleteCount: number;
772 > }
773 >
774 > export function diff<T>(before: T[], after: T[], contains: (a: T) => boolean, equal: (a: T, b: T) => boolean = (a: T, b: T) => a === b): ISplice<T>[] {
775 const result: IMutableSplice<T>[] = [];
776
827 return result;
828 }
830 > export interface ICellEditorViewState {
831 > selections: editorCommon.ICursorState[];
832 > }
833 >
834 > export const NOTEBOOK_EDITOR_CURSOR_BOUNDARY = new RawContextKey<'none' | 'top' | 'bottom' | 'both'>('notebookEditorCursorAtBoundary', 'none');
835 >
836 > export const NOTEBOOK_EDITOR_CURSOR_LINE_BOUNDARY = new RawContextKey<'none' | 'start' | 'end' | 'both'>('notebookEditorCursorAtLineBoundary', 'none');
837 >
838 > export interface INotebookLoadOptions {
839 > /**
840 > * Go to disk bypassing any cache of the model if any.
841 > */
842 > forceReadFromFile?: boolean;
843 > /**
844 > * If provided, the size of the file will be checked against the limits
845 > * and an error will be thrown if any limit is exceeded.
846 > */
847 > readonly limits?: IFileReadLimits;
848 > }
849 >
850 > export type NotebookEditorModelCreationOptions = {
851 > limits?: IFileReadLimits;
852 > scratchpad?: boolean;
853 > viewType?: string;
854 > };
855 >
856 > export interface IResolvedNotebookEditorModel extends INotebookEditorModel {
857 > notebook: NotebookTextModel;
858 > }
859 >
860 > export interface INotebookEditorModel extends IDisposable {
861 > readonly onDidChangeDirty: Event<void>;
862 > readonly onDidSave: Event<IWorkingCopySaveEvent>;
863 > readonly onDidChangeOrphaned: Event<void>;
864 > readonly onDidChangeReadonly: Event<void>;
865 > readonly onDidRevertUntitled: Event<void>;
866 > readonly resource: URI;
867 > readonly viewType: string;
868 > readonly notebook: INotebookTextModel | undefined;
869 > readonly hasErrorState: boolean;
870 > isResolved(): boolean;
871 > isDirty(): boolean;
872 > isModified(): boolean;
873 > isReadonly(): boolean | IMarkdownString;
874 > isOrphaned(): boolean;
875 > hasAssociatedFilePath(): boolean;
876 > load(options?: INotebookLoadOptions): Promise<IResolvedNotebookEditorModel>;
877 > save(options?: ISaveOptions): Promise<boolean>;
878 > saveAs(target: URI): Promise<IUntypedEditorInput | undefined>;
879 > revert(options?: IRevertOptions): Promise<void>;
880 > }
881 >
882 > export interface INotebookDiffEditorModel extends IDisposable {
883 > original: { notebook: NotebookTextModel; resource: URI; viewType: string };
884 > modified: { notebook: NotebookTextModel; resource: URI; viewType: string };
885 > }
886 >
887 > export interface NotebookDocumentBackupData extends IWorkingCopyBackupMeta {
888 > readonly viewType: string;
889 > readonly backupId?: string;
890 > readonly mtime?: number;
891 > }
892 >
893 > export enum NotebookEditorPriority {
894 > default = 'default',
895 > option = 'option',
896 > }
897 >
898 > export interface INotebookFindOptions {
899 > regex?: boolean;
900 > wholeWord?: boolean;
901 > caseSensitive?: boolean;
902 > wordSeparators?: string;
903 > includeMarkupInput?: boolean;
904 > includeMarkupPreview?: boolean;
905 > includeCodeInput?: boolean;
906 > includeOutput?: boolean;
907 > findScope?: INotebookFindScope;
908 > }
909 >
910 > export interface INotebookFindScope {
911 > findScopeType: NotebookFindScopeType;
912 > selectedCellRanges?: ICellRange[];
913 > selectedTextRanges?: Range[];
914 > }
915 >
916 > export enum NotebookFindScopeType {
917 > Cells = 'cells',
918 > Text = 'text',
919 > None = 'none'
920 > }
921 >
922 > export interface INotebookExclusiveDocumentFilter {
923 > include?: string | glob.IRelativePattern;
924 > exclude?: string | glob.IRelativePattern;
925 > }
926 >
927 > export interface INotebookDocumentFilter {
928 > viewType?: string | string[];
929 > filenamePattern?: string | glob.IRelativePattern | INotebookExclusiveDocumentFilter;
930 > }
931 >
932 > //TODO@rebornix test
933 >
934 > export function isDocumentExcludePattern(filenamePattern: string | glob.IRelativePattern | INotebookExclusiveDocumentFilter): filenamePattern is { include: string | glob.IRelativePattern; exclude: string | glob.IRelativePattern } {
935 const arg = filenamePattern as INotebookExclusiveDocumentFilter;
936
942 return false;
943 }
944 > export function notebookDocumentFilterMatch(filter: INotebookDocumentFilter, viewType: string, resource: URI): boolean { notebookCommon.ts
945 if (Array.isArray(filter.viewType) && filter.viewType.indexOf(viewType) >= 0) {
946 return true;
967 return false;
968 }
970 > export interface INotebookCellStatusBarItemProvider {
971 > viewType: string;
972 > onDidChangeStatusBarItems?: Event<void>;
973 > provideCellStatusBarItems(uri: URI, index: number, token: CancellationToken): Promise<INotebookCellStatusBarItemList | undefined>;
974 > }
975 >
976 >
977 > export interface INotebookDiffResult {
978 > cellsDiff: IDiffResult;
979 > metadataChanged: boolean;
980 > }
981 >
982 > export interface INotebookCellStatusBarItem {
983 > readonly alignment: CellStatusbarAlignment;
984 > readonly priority?: number;
985 > readonly text: string;
986 > readonly color?: string | ThemeColor;
987 > readonly backgroundColor?: string | ThemeColor;
988 > readonly tooltip?: string | IMarkdownString;
989 > readonly command?: string | Command;
990 > readonly accessibilityInformation?: IAccessibilityInformation;
991 > readonly opacity?: string;
992 > readonly onlyShowWhenActive?: boolean;
993 > }
994 >
995 > export interface INotebookCellStatusBarItemList {
996 > items: INotebookCellStatusBarItem[];
997 > dispose?(): void;
998 > }
999 >
1000 > export type ShowCellStatusBarType = 'hidden' | 'visible' | 'visibleAfterExecute';
1001 > export const NotebookSetting = {
1002 > displayOrder: 'notebook.displayOrder',
1003 > cellToolbarLocation: 'notebook.cellToolbarLocation',
1004 > cellToolbarVisibility: 'notebook.cellToolbarVisibility',
1005 > showCellStatusBar: 'notebook.showCellStatusBar',
1006 > cellExecutionTimeVerbosity: 'notebook.cellExecutionTimeVerbosity',
1007 > textDiffEditorPreview: 'notebook.diff.enablePreview',
1008 > diffOverviewRuler: 'notebook.diff.overviewRuler',
1009 > experimentalInsertToolbarAlignment: 'notebook.experimental.insertToolbarAlignment',
1010 > compactView: 'notebook.compactView',
1011 > focusIndicator: 'notebook.cellFocusIndicator',
1012 > insertToolbarLocation: 'notebook.insertToolbarLocation',
1013 > globalToolbar: 'notebook.globalToolbar',
1014 > stickyScrollEnabled: 'notebook.stickyScroll.enabled',
1015 > stickyScrollMode: 'notebook.stickyScroll.mode',
1016 > undoRedoPerCell: 'notebook.undoRedoPerCell',
1017 > consolidatedOutputButton: 'notebook.consolidatedOutputButton',
1018 > openOutputInPreviewEditor: 'notebook.output.openInPreviewEditor.enabled',
1019 > showFoldingControls: 'notebook.showFoldingControls',
1020 > dragAndDropEnabled: 'notebook.dragAndDropEnabled',
1021 > cellEditorOptionsCustomizations: 'notebook.editorOptionsCustomizations',
1022 > consolidatedRunButton: 'notebook.consolidatedRunButton',
1023 > openGettingStarted: 'notebook.experimental.openGettingStarted',
1024 > globalToolbarShowLabel: 'notebook.globalToolbarShowLabel',
1025 > markupFontSize: 'notebook.markup.fontSize',
1026 > markdownLineHeight: 'notebook.markdown.lineHeight',
1027 > interactiveWindowCollapseCodeCells: 'interactiveWindow.collapseCellInputCode',
1028 > outputScrolling: 'notebook.output.scrolling',
1029 > textOutputLineLimit: 'notebook.output.textLineLimit',
1030 > LinkifyOutputFilePaths: 'notebook.output.linkifyFilePaths',
1031 > minimalErrorRendering: 'notebook.output.minimalErrorRendering',
1032 > formatOnSave: 'notebook.formatOnSave.enabled',
1033 > insertFinalNewline: 'notebook.insertFinalNewline',
1034 > defaultFormatter: 'notebook.defaultFormatter',
1035 > formatOnCellExecution: 'notebook.formatOnCellExecution',
1036 > codeActionsOnSave: 'notebook.codeActionsOnSave',
1037 > outputWordWrap: 'notebook.output.wordWrap',
1038 > outputLineHeight: 'notebook.output.lineHeight',
1039 > outputFontSize: 'notebook.output.fontSize',
1040 > outputFontFamily: 'notebook.output.fontFamily',
1041 > findFilters: 'notebook.find.filters',
1042 > logging: 'notebook.logging',
1043 > confirmDeleteRunningCell: 'notebook.confirmDeleteRunningCell',
1044 > remoteSaving: 'notebook.experimental.remoteSave',
1045 > gotoSymbolsAllSymbols: 'notebook.gotoSymbols.showAllSymbols',
1046 > outlineShowMarkdownHeadersOnly: 'notebook.outline.showMarkdownHeadersOnly',
1047 > outlineShowCodeCells: 'notebook.outline.showCodeCells',
1048 > outlineShowCodeCellSymbols: 'notebook.outline.showCodeCellSymbols',
1049 > breadcrumbsShowCodeCells: 'notebook.breadcrumbs.showCodeCells',
1050 > scrollToRevealCell: 'notebook.scrolling.revealNextCellOnExecute',
1051 > cellChat: 'notebook.experimental.cellChat',
1052 > cellGenerate: 'notebook.experimental.generate',
1053 > notebookVariablesView: 'notebook.variablesView',
1054 > notebookInlineValues: 'notebook.inlineValues',
1055 > InteractiveWindowPromptToSave: 'interactiveWindow.promptToSaveOnClose',
1056 > cellFailureDiagnostics: 'notebook.cellFailureDiagnostics',
1057 > outputBackupSizeLimit: 'notebook.backup.sizeLimit',
1058 > multiCursor: 'notebook.multiCursor.enabled',
1059 > markupFontFamily: 'notebook.markup.fontFamily',
1060 > } as const;
1061 >
1062 > export const enum CellStatusbarAlignment {
1063 > Left = 1,
1064 > Right = 2
1065 > }
1066 >
1067 > export class NotebookWorkingCopyTypeIdentifier {
1068 >
1069 > private static _prefix = 'notebook/';
1070 >
1071 > static create(notebookType: string, viewType?: string): string {
1072 return `${NotebookWorkingCopyTypeIdentifier._prefix}${notebookType}/${viewType ?? notebookType}`;
1073 }
1075 > static parse(candidate: string): { notebookType: string; viewType: string } | undefined {
1076 if (candidate.startsWith(NotebookWorkingCopyTypeIdentifier._prefix)) {
1077 const split = candidate.substring(NotebookWorkingCopyTypeIdentifier._prefix.length).split('/');
1082 return undefined;
1083 }
1085 >
1086 > export interface NotebookExtensionDescription {
1087 > readonly id: ExtensionIdentifier;
1088 > readonly location: UriComponents | undefined;
1089 > }
1090 >
1091 > const textDecoder = new TextDecoder();
1092 >
1093 > /**
1094 > * Given a stream of individual stdout outputs, this function will return the compressed lines, escaping some of the common terminal escape codes.
1095 > * E.g. some terminal escape codes would result in the previous line getting cleared, such if we had 3 lines and
1096 > * last line contained such a code, then the result string would be just the first two lines.
1097 > * @returns a single VSBuffer with the concatenated and compressed data, and whether any compression was done.
1098 > */
1099 > export function compressOutputItemStreams(outputs: Uint8Array[]) {
1100 const buffers: Uint8Array[] = [];
1101 let startAppending = false;
1115 return { data, didCompression };
1116 }
1118 > export const MOVE_CURSOR_1_LINE_COMMAND = `${String.fromCharCode(27)}[A`;
1119 > const MOVE_CURSOR_1_LINE_COMMAND_BYTES = MOVE_CURSOR_1_LINE_COMMAND.split('').map(c => c.charCodeAt(0));
1120 > const LINE_FEED = 10;
1121 function compressStreamBuffer(streams: Uint8Array[]) {
1122 let didCompress = false;
1143 return didCompress;
1144 }
1146 >
1147 >
1148 > /**
1149 > * Took this from jupyter/notebook
1150 > * https://github.com/jupyter/notebook/blob/b8b66332e2023e83d2ee04f83d8814f567e01a4e/notebook/static/base/js/utils.js
1151 > * Remove characters that are overridden by backspace characters
1152 > */
1153 function fixBackspace(txt: string) {
1154 let tmp = txt;
1160 return txt;
1161 }
1163 > /**
1164 > * Remove chunks that should be overridden by the effect of carriage return characters
1165 > * From https://github.com/jupyter/notebook/blob/master/notebook/static/base/js/utils.js
1166 > */
1167 function fixCarriageReturn(txt: string) {
1168 txt = txt.replace(/\r+\n/gm, '\n'); // \r followed by \n --> newline
1175 return txt;
1176 }
1178 > const BACKSPACE_CHARACTER = '\b'.charCodeAt(0);
1179 > const CARRIAGE_RETURN_CHARACTER = '\r'.charCodeAt(0);
1180 function formatStreamText(buffer: VSBuffer): VSBuffer {
1181 // We have special handling for backspace and carriage return characters.
1187 return VSBuffer.fromString(fixCarriageReturn(fixBackspace(textDecoder.decode(buffer.buffer))));
1188 }
1190 > export interface INotebookKernelSourceAction {
1191 > readonly label: string;
1192 > readonly description?: string;
1193 > readonly detail?: string;
1194 > readonly command?: string | Command;
1195 > readonly documentation?: UriComponents | string;
1196 > }
src/vs/workbench/contrib/chat/common/chatSessionsService.ts 910 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatSessionsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../base/common/cancellation.js';
7 > import { Event } from '../../../../base/common/event.js';
8 > import { IMarkdownString } from '../../../../base/common/htmlContent.js';
9 > import { IDisposable } from '../../../../base/common/lifecycle.js';
10 > import { IObservable } from '../../../../base/common/observable.js';
11 > import { ThemeIcon } from '../../../../base/common/themables.js';
12 > import { URI } from '../../../../base/common/uri.js';
13 > import { IPosition } from '../../../../editor/common/core/position.js';
14 > import { isRemoteAgentHostSessionType } from '../../../../platform/agentHost/common/agentHostSessionType.js';
15 > import { createDecorator, ServicesAccessor } from '../../../../platform/instantiation/common/instantiation.js';
16 > import { Registry } from '../../../../platform/registry/common/platform.js';
17 > import { LOCAL_AGENT_HOST_SCHEME_PREFIX } from '../../../../platform/agentHost/common/agentHostConnectionsService.js';
18 > import { IChatAgentAttachmentCapabilities, IChatAgentRequest } from './participants/chatAgents.js';
19 > import { IChatEditingSession } from './editing/chatEditingService.js';
20 > import { IChatRequestModeInstructions, IChatRequestVariableData, ISerializableChatModelInputState } from './model/chatModel.js';
21 > import { IChatProgress, IChatResponseErrorDetails, IChatSessionTiming } from './chatService/chatService.js';
22 > import { Target } from './promptSyntax/promptTypes.js';
23 >
24 > export const enum ChatSessionsExtensions {
25 > AsyncActivation = 'workbench.contrib.chatSessions.asyncActivation'
26 > }
27 >
28 > export interface IAsyncChatSessionActivationContribution {
29 > matchSessionType(sessionType: string): boolean;
30 > waitForActivation(accessor: ServicesAccessor, sessionType: string): Promise<boolean>;
31 > }
32 >
33 > export interface IAsyncChatSessionActivationRegistry {
34 > register(contribution: IAsyncChatSessionActivationContribution): IDisposable;
35 > getActivators(sessionType: string): readonly IAsyncChatSessionActivationContribution[];
36 > }
37 >
38 > class AsyncChatSessionActivationRegistry implements IAsyncChatSessionActivationRegistry {
39 > private readonly _contributions = new Set<IAsyncChatSessionActivationContribution>();
40 >
41 > register(contribution: IAsyncChatSessionActivationContribution): IDisposable {
42 this._contributions.add(contribution);
43 return {
45 };
46 }
48 > getActivators(sessionType: string): readonly IAsyncChatSessionActivationContribution[] {
49 return Array.from(this._contributions).filter(contribution => contribution.matchSessionType(sessionType));
50 }
52 >
53 > Registry.add(ChatSessionsExtensions.AsyncActivation, new AsyncChatSessionActivationRegistry());
54 >
55 > export const enum ChatSessionStatus {
56 > Failed = 0,
57 > Completed = 1,
58 > InProgress = 2,
59 > NeedsInput = 3
60 > }
61 >
62 > export interface IChatSessionCommandContribution {
63 > readonly name: string;
64 > readonly description: string;
65 > readonly when?: string;
66 > }
67 >
68 > export interface IChatSessionProviderOptionModelMetadata {
69 > readonly name: string;
70 > readonly id: string;
71 > readonly vendor?: string;
72 > readonly version?: string;
73 > readonly family?: string;
74 > readonly tooltip?: string;
75 > readonly pricing?: string;
76 > readonly multiplierNumeric?: number;
77 > readonly inputCost?: number;
78 > readonly outputCost?: number;
79 > readonly cacheCost?: number;
80 > readonly cacheWriteCost?: number;
81 > readonly longContextInputCost?: number;
82 > readonly longContextOutputCost?: number;
83 > readonly longContextCacheCost?: number;
84 > readonly longContextCacheWriteCost?: number;
85 > readonly priceCategory?: string;
86 > readonly promo?: {
87 > readonly id: string;
88 > readonly discountPercent: number;
89 > readonly endsAt: string;
90 > readonly message: string;
91 > };
92 > readonly maxInputTokens?: number;
93 > readonly maxOutputTokens?: number;
94 > readonly capabilities?: {
95 > readonly vision?: boolean;
96 > readonly toolCalling?: boolean;
97 > };
98 > }
99 >
100 > export interface IChatSessionProviderOptionItem {
101 > readonly id: string;
102 > readonly name: string;
103 > readonly description?: string;
104 > readonly detail?: string;
105 > readonly locked?: boolean;
106 > readonly icon?: ThemeIcon;
107 > readonly default?: boolean;
108 > readonly slashCommand?: string;
109 > readonly tooltip?: string;
110 > readonly modelMetadata?: IChatSessionProviderOptionModelMetadata;
111 > // [key: string]: any;
112 > }
113 >
114 > export interface IChatSessionProviderOptionGroupCommand {
115 > readonly command: string;
116 > readonly title: string;
117 > readonly tooltip?: string;
118 > readonly arguments?: readonly unknown[];
119 > }
120 >
121 > export interface IChatSessionProviderOptionGroup {
122 > readonly id: string;
123 > readonly name: string;
124 > readonly description?: string;
125 > readonly detail?: string;
126 > readonly selected?: IChatSessionProviderOptionItem;
127 > readonly items: readonly IChatSessionProviderOptionItem[];
128 > /**
129 > * A context key expression that controls visibility of this option group picker.
130 > * When specified, the picker is only visible when the expression evaluates to true.
131 > * The expression can reference other option group values via `chatSessionOption.<groupId>`.
132 > * Example: `"chatSessionOption.models == 'gpt-4'"`
133 > */
134 > readonly when?: string;
135 > readonly icon?: ThemeIcon;
136 > /**
137 > * Custom commands to show in the option group's picker UI.
138 > * These will be shown in a separate section at the end of the picker.
139 > */
140 > readonly commands?: readonly IChatSessionProviderOptionGroupCommand[];
141 > /**
142 > * Optional kind hint that controls how the group is presented.
143 > * - `'permissions'`: the group's items are surfaced inside the chat permission picker
144 > * instead of being rendered as a standalone picker. At most one group per provider
145 > * may use this kind; if multiple are declared, the first one (in declaration order)
146 > * wins. The group has no UI of its own — it is invisible when the permission
147 > * picker is hidden by its own `when` clauses.
148 > */
149 > readonly kind?: 'permissions';
150 > }
151 >
152 > export interface IChatSessionsExtensionPoint {
153 > readonly type: string;
154 > readonly name: string;
155 > readonly displayName: string;
156 > readonly description: string;
157 > readonly when?: string;
158 > readonly icon?: string | { light: string; dark: string };
159 > readonly order?: number;
160 > readonly alternativeIds?: string[];
161 > readonly welcomeTitle?: string;
162 > readonly welcomeMessage?: string;
163 > readonly welcomeTips?: string;
164 > readonly inputPlaceholder?: string;
165 > readonly capabilities?: IChatAgentAttachmentCapabilities;
166 > readonly commands?: IChatSessionCommandContribution[];
167 > readonly canDelegate?: boolean;
168 > readonly isReadOnly?: boolean;
169 > /**
170 > * When set, the chat session will show a filtered mode picker with custom agents
171 > * that have a matching `target` property. This enables contributed chat sessions
172 > * to reuse the standard agent/mode dropdown with filtered custom agents.
173 > * Custom agents without a `target` property are also shown in all filtered lists
174 > */
175 > readonly customAgentTarget?: Target;
176 > readonly requiresCustomModels?: boolean;
177 > /**
178 > * Whether this session type supports the synthetic "Auto" model fallback.
179 > * Defaults to true. When false and no models are available, the picker
180 > * shows a "No models available" state instead of "Auto".
181 > *
182 > * This is distinct from {@link requiresCustomModels}, which only controls
183 > * whether the picker is filtered to the session's own model pool — a
184 > * session can own a custom pool yet still support Auto (e.g. the Copilot
185 > * CLI agent host).
186 > */
187 > readonly supportsAutoModel?: boolean;
188 > /**
189 > * Logical Agent Host provider ID for Agent Host-backed chat sessions.
190 > * For example, both local `agent-host-copilotcli` and remote
191 > * `remote-{authority}-copilotcli` sessions use `copilotcli`.
192 > */
193 > readonly agentHostProviderId?: string;
194 > /**
195 > * Whether this type needs a GitHub Copilot account and so is unusable until the user signs in. Set by
196 > * Copilot-backed types (Copilot CLI / agent host, cloud agent) where BYOK isn't supported. Defaults to false, so
197 > * third-party types that don't depend on Copilot stay usable while signed out.
198 > */
199 > readonly requiresCopilotSignIn?: boolean;
200 > /**
201 > * When false, the delegation picker is hidden for this session type.
202 > * Defaults to true.
203 > */
204 > readonly supportsDelegation?: boolean;
205 > /**
206 > * Decides whether to automatically attach instruction files to chat requests
207 > * for this session type. Defaults to false when not specified.
208 > */
209 > readonly autoAttachReferences?: boolean;
210 > }
211 >
212 > export interface IChatSessionItem {
213 > readonly resource: URI;
214 > readonly label: string;
215 > readonly iconPath?: ThemeIcon;
216 > readonly badge?: string | IMarkdownString;
217 > readonly description?: string | IMarkdownString;
218 > readonly status?: ChatSessionStatus;
219 > readonly tooltip?: string | IMarkdownString;
220 > readonly timing: IChatSessionTiming;
221 > readonly changes?: {
222 > readonly files: number;
223 > readonly insertions: number;
224 > readonly deletions: number;
225 > } | readonly IChatSessionFileChange[] | readonly IChatSessionFileChange2[];
226 > readonly archived?: boolean;
227 > readonly metadata?: IChatSessionItemMetadata;
228 > /**
229 > * Resource identifier the item was previously known by. When set, host-stored
230 > * per-resource state (archive, pin, read) recorded under that URI is adopted
231 > * forward onto {@link resource} on first state read, and the legacy entry is
232 > * removed. Scheme must match {@link resource}'s scheme; otherwise ignored.
233 > */
234 > readonly legacyResource?: URI;
235 > }
236 >
237 > export interface IChatSessionItemMetadata {
238 > //#region Changes metadata (for sessions window)
239 > readonly repositoryPath?: string;
240 > readonly workingDirectoryPath?: string;
241 > readonly firstCheckpointRef?: string;
242 > readonly lastCheckpointRef?: string;
243 > readonly worktreePath?: string;
244 > readonly uncommittedChanges?: number;
245 > readonly baseRefOid?: string;
246 > readonly headRefOid?: string;
247 > readonly branchName?: string;
248 > readonly branch?: string;
249 > readonly baseBranchName?: string;
250 > readonly baseBranch?: string;
251 > readonly baseBranchProtected?: boolean;
252 > readonly hasGitHubRemote?: boolean;
253 > readonly upstreamBranchName?: string;
254 > readonly incomingChanges?: number;
255 > readonly outgoingChanges?: number;
256 > //#endregion
257 >
258 > readonly [key: string]: unknown;
259 > }
260 >
261 > export interface IChatSessionFileChange {
262 > readonly modifiedUri: URI;
263 > readonly originalUri?: URI;
264 > readonly insertions: number;
265 > readonly deletions: number;
266 > readonly reviewed?: boolean;
267 > }
268 >
269 > export interface IChatSessionFileChange2 {
270 > readonly uri: URI;
271 > readonly originalUri?: URI;
272 > readonly modifiedUri?: URI;
273 > readonly insertions: number;
274 > readonly deletions: number;
275 > readonly reviewed?: boolean;
276 > }
277 >
278 > export type IChatSessionHistoryItem = {
279 > id?: string;
280 > type: 'request';
281 > prompt: string;
282 > participant: string;
283 > command?: string;
284 > variableData?: IChatRequestVariableData;
285 > modelId?: string;
286 > timestamp?: number;
287 > modeInstructions?: IChatRequestModeInstructions;
288 > isSystemInitiated?: boolean;
289 > systemInitiatedLabel?: string;
290 > isTerminalRequest?: boolean;
291 > } | {
292 > type: 'response';
293 > parts: IChatProgress[];
294 > participant: string;
295 > details?: string;
296 > elapsedMs?: number;
297 > completedAt?: number;
298 > /**
299 > * Error details for a failed response. Rendered as a proper chat error
300 > * (including the quota-exceeded upgrade affordance), mirroring the live
301 > * agent result's `errorDetails`.
302 > */
303 > errorDetails?: IChatResponseErrorDetails;
304 > };
305 >
306 > export type IChatSessionRequestHistoryItem = Extract<IChatSessionHistoryItem, { type: 'request' }>;
307 >
308 > export interface IChatSessionServerRequest {
309 > readonly prompt: string;
310 > readonly variableData?: IChatRequestVariableData;
311 > readonly timestamp?: number;
312 > readonly isSystemInitiated?: boolean;
313 > readonly systemInitiatedLabel?: string;
314 > readonly isTerminalRequest?: boolean;
315 > }
316 >
317 > /**
318 > * Whether `text` runs as a terminal command for the given command `prefix`
319 > * (e.g. `!`) — it starts with the prefix and has a non-empty command after it.
320 > * Mirrors the agent host's bang parser, where a lone `!` (or `!` followed only
321 > * by whitespace) is forwarded to the agent rather than executed.
322 > */
323 > export function isTerminalCommandPrompt(text: string, prefix: string | undefined): boolean {
324 return !!prefix && text.startsWith(prefix) && text.slice(prefix.length).trim().length > 0;
325 }
327 > /**
328 > * A set of well-known session types
329 > */
330 > export namespace SessionType {
331 > export const CopilotCLI = 'copilotcli';
332 > export const CopilotCloud = 'copilot-cloud-agent';
333 > export const Local = 'local';
334 > export const ClaudeCode = 'claude-code';
335 > export const Codex = 'openai-codex';
336 > export const Growth = 'copilot-growth';
337 > export const AgentHostCopilot = 'agent-host-copilotcli';
338 > export const AgentHostClaude = 'agent-host-claude';
339 > export const AgentHostCodex = 'agent-host-codex';
340 > }
341 >
342 > /**
343 > * Returns whether the given session type is a local agent host target.
344 > */
345 > export function isLocalAgentHostTarget(target: string): boolean {
346 > return target === SessionType.AgentHostCopilot || chatSessionsService.ts
347 > target.startsWith(LOCAL_AGENT_HOST_SCHEME_PREFIX); chatSessionsService.ts
350 > /**
351 > * Returns whether the given session type is a remote agent host target.
352 > *
353 > * Note: The `remote-` prefix convention is established by
354 > * `RemoteAgentHostContribution` which generates session types as
355 > * `remote-{sanitizedAddress}-{provider}`. If future remote providers that
356 > * are NOT agent hosts need a different prefix, this function must be updated.
357 > */
358 > export function isRemoteAgentHostTarget(target: string): boolean {
359 > return isRemoteAgentHostSessionType(target); chatSessionsService.ts
360 > }
362 > /**
363 > * Returns whether the given session type is an agent host target.
364 > * Matches the local agent host (`agent-host-*`) and remote agent hosts (`remote-*`).
365 > */
366 > export function isAgentHostTarget(target: string): boolean {
367 > return isLocalAgentHostTarget(target) || isRemoteAgentHostTarget(target); chatSessionsService.ts
368 > }
370 > /**
371 > * The session type used for local agent chat sessions.
372 > */
373 > export const localChatSessionType = SessionType.Local;
374 >
375 > export interface IChatSession extends IDisposable {
376 > readonly onWillDispose: Event<void>;
377 >
378 > readonly sessionResource: URI;
379 >
380 > readonly title?: string;
381 >
382 > readonly history: readonly IChatSessionHistoryItem[];
383 >
384 >
385 > readonly options?: ReadonlyChatSessionOptionsMap;
386 >
387 > readonly progressObs?: IObservable<IChatProgress[]>;
388 > readonly isCompleteObs?: IObservable<boolean>;
389 > readonly isReadOnly?: IObservable<boolean>;
390 > readonly interruptActiveResponseCallback?: () => Promise<boolean>;
391 >
392 > /**
393 > * Event fired when the server initiates a new request (e.g. from a consumed
394 > * queued message). The consumer should create a new request+response pair in
395 > * the model and prepare to receive progress via {@link progressObs}.
396 > */
397 > readonly onDidStartServerRequest?: Event<IChatSessionServerRequest>;
398 >
399 > /**
400 > * Editing session transferred from a previously-untitled chat session in `onDidCommitChatSessionItem`.
401 > */
402 > transferredState?: {
403 > readonly editingSession: IChatEditingSession | undefined;
404 > readonly inputState: ISerializableChatModelInputState | undefined;
405 > };
406 >
407 > requestHandler?: (
408 > request: IChatAgentRequest,
409 > progress: (progress: IChatProgress[]) => void,
410 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
411 > history: any[], // TODO: Nail down types
412 > token: CancellationToken
413 > ) => Promise<void>;
414 >
415 > /**
416 > * Forks the session from the given request point.
417 > * @param request The request history item to fork from, or undefined to fork from the end.
418 > * @param token Cancellation token.
419 > * @returns The forked session item. The promise is rejected if forking fails.
420 > */
421 > forkSession?: (request: IChatSessionRequestHistoryItem | undefined, token: CancellationToken) => Promise<IChatSessionItem>;
422 >
423 > /**
424 > * Renames the session.
425 > * @param title The new title for the session.
426 > * @param token Cancellation token.
427 > * @returns A promise that resolves once the rename has been dispatched. The promise is rejected if renaming fails.
428 > */
429 > renameSession?: (title: string, token: CancellationToken) => Promise<void>;
430 > }
431 >
432 > export interface IChatSessionContentProvider {
433 > provideChatSessionContent(sessionResource: URI, token: CancellationToken): Promise<IChatSession>;
434 >
435 > /** Resolves a parsed response Markdown URI before it is sanitized and rendered. */
436 > resolveChatResponseUri?(sessionResource: URI, href: string, kind: 'link' | 'image'): string;
437 >
438 > /**
439 > * Optional. Compute completion items for an input being composed in this
440 > * session. Returning `undefined` lets the workbench fall back to its
441 > * default in-process completion providers.
442 > */
443 > provideChatInputCompletions?(sessionResource: URI, params: IChatInputCompletionsParams, token: CancellationToken): Promise<IChatInputCompletionsResult | undefined>;
444 >
445 > /**
446 > * Optional. Trigger characters that, when typed in the chat input,
447 > * SHOULD cause the workbench to issue a `provideChatInputCompletions`
448 > * request. Used to register a Monaco completion provider scoped to
449 > * sessions handled by this content provider.
450 > */
451 > provideChatInputCompletionTriggerCharacters?(): Promise<readonly string[]>;
452 > }
453 >
454 > /**
455 > * Inputs for {@link IChatSessionContentProvider.provideChatInputCompletions}
456 > * and {@link IChatSessionsService.provideChatInputCompletions}.
457 > */
458 > export interface IChatInputCompletionsParams {
459 > /**
460 > * The complete text of the input being completed (e.g. the user message
461 > * the user is currently composing).
462 > */
463 > readonly text: string;
464 > /**
465 > * The character offset within {@link text} at which the completion is
466 > * requested, measured in UTF-16 code units. MUST satisfy
467 > * `0 <= offset <= text.length`.
468 > */
469 > readonly offset: number;
470 > }
471 >
472 > /**
473 > * A neutral completion-item shape returned by
474 > * {@link IChatSessionContentProvider.provideChatInputCompletions}. The
475 > * workbench-side completion glue maps these into Monaco completion items
476 > * and the corresponding chat-input attachment.
477 > */
478 > export interface IChatInputCompletionItem {
479 > /** Text inserted into the input when this item is accepted. */
480 > readonly insertText: string;
481 > /**
482 > * Optional display label shown in the completion picker. When omitted, the
483 > * workbench displays {@link insertText}. Set this when the inserted text
484 > * differs from the label — e.g. an action item that inserts nothing
485 > * (`insertText: ''`) but should still be shown to the user.
486 > */
487 > readonly label?: string;
488 > /**
489 > * Half-open range `[start, end)` in the *current* input text that
490 > * {@link insertText} replaces. Positions use 1-based `lineNumber` and
491 > * `column` to match Monaco. When omitted, the workbench replaces the
492 > * word at the cursor.
493 > */
494 > readonly start?: IPosition;
495 > readonly end?: IPosition;
496 > /** Attachment associated with the item. */
497 > readonly attachment: IChatInputCompletionResourceAttachment | IChatInputCompletionCommandAttachment | IChatInputCompletionSkillAttachment;
498 > }
499 >
500 > /**
501 > * Resource attachment associated with a completion item. The workbench
502 > * adds it to the input's variable model when the item is accepted.
503 > */
504 > export interface IChatInputCompletionResourceAttachment {
505 > readonly kind: 'resource';
506 > readonly uri: URI;
507 > readonly displayName?: string;
508 > readonly isDirectory?: boolean;
509 > /**
510 > * Implementation-defined metadata that MUST be preserved by the
511 > * workbench when the accepted completion is sent back as part of a
512 > * user message attachment.
513 > */
514 > readonly _meta?: Record<string, unknown>;
515 > }
516 >
517 > /**
518 > * Command attachment associated with a completion item.
519 > */
520 > export interface IChatInputCompletionCommandAttachment {
521 > readonly kind: 'command';
522 > readonly command: string;
523 > readonly description: string;
524 > /**
525 > * Implementation-defined metadata that MUST be preserved by the
526 > * workbench when the accepted completion is sent back as part of a
527 > * user message attachment.
528 > */
529 > readonly _meta?: Record<string, unknown>;
530 > }
531 >
532 > /**
533 > * Skill attachment associated with a completion item. The workbench
534 > * adds it to the input's variable model when the item is accepted.
535 > */
536 > export interface IChatInputCompletionSkillAttachment {
537 > readonly kind: 'skill';
538 > readonly uri: URI;
539 > readonly displayName?: string;
540 > readonly description?: string;
541 > /**
542 > * Implementation-defined metadata that MUST be preserved by the
543 > * workbench when the accepted completion is sent back as part of a
544 > * user message attachment.
545 > */
546 > readonly _meta?: Record<string, unknown>;
547 > }
548 >
549 > /**
550 > * Result of {@link IChatSessionContentProvider.provideChatInputCompletions}.
551 > */
552 > export interface IChatInputCompletionsResult {
553 > readonly items: readonly IChatInputCompletionItem[];
554 > }
555 >
556 > export interface IChatNewSessionRequest {
557 > readonly prompt: string;
558 > readonly command?: string;
559 >
560 > readonly initialSessionOptions?: ReadonlyChatSessionOptionsMap;
561 >
562 > /**
563 > * The chat-input session resource the user was typing into when this
564 > * request was issued. Set when the chat infrastructure is rewriting an
565 > * untitled session URI to a real one on first send. Controllers can use
566 > * this to bridge any pre-creation state they tracked under the old URI
567 > * (e.g. provisional agent-host sessions) to the new resource that the
568 > * controller returns.
569 > */
570 > readonly untitledResource?: URI;
571 > }
572 >
573 > export interface IChatSessionItemsDelta {
574 > readonly addedOrUpdated?: readonly IChatSessionItem[];
575 > readonly removed?: readonly URI[];
576 > }
577 >
578 > export interface IChatSessionItemController {
579 >
580 > readonly onDidChangeChatSessionItems: Event<IChatSessionItemsDelta>;
581 >
582 > get items(): readonly IChatSessionItem[];
583 >
584 > refresh(token: CancellationToken): Promise<void>;
585 >
586 > newChatSessionItem?(request: IChatNewSessionRequest, token: CancellationToken): Promise<IChatSessionItem | undefined>;
587 >
588 > getNewChatSessionInputState?(sessionResource: URI, token: CancellationToken): Promise<readonly IChatSessionProviderOptionGroup[] | undefined>;
589 >
590 > resolveChatSessionItem?(resource: URI, token: CancellationToken): Promise<IChatSessionItem | undefined>;
591 >
592 > /**
593 > * Permanently delete the session identified by `resource`. Implementations should tear down any backend state for
594 > * the session. The controller is expected to fire an `onDidChangeChatSessionItems` event with the removed resource
595 > * as a result of the deletion.
596 > */
597 > deleteChatSessionItem?(resource: URI, token: CancellationToken): Promise<void>;
598 >
599 > /**
600 > * Set the authoritative archived state for the session identified by `resource`.
601 > */
602 > setChatSessionItemArchived?(resource: URI, archived: boolean): void;
603 > }
604 >
605 > export interface IChatSessionOptionsChangeEvent {
606 > readonly sessionResource: URI;
607 > readonly updates: ReadonlyMap<string, string | IChatSessionProviderOptionItem | undefined>;
608 > }
609 >
610 > export type ResolvedChatSessionsExtensionPoint = Omit<IChatSessionsExtensionPoint, 'icon'> & {
611 > readonly icon: ThemeIcon | URI | undefined;
612 > };
613 >
614 > /**
615 > * Session options as key-value pairs.
616 > *
617 > * Keys correspond to option group IDs (e.g., 'models', 'subagents') and values are either the selected option item IDs (string) or full option items (for locked state).
618 > */
619 > export type ChatSessionOptionsMap = Map<string, string | IChatSessionProviderOptionItem>;
620 >
621 > export namespace ChatSessionOptionsMap {
622 > export function fromRecord(obj: { [key: string]: string | IChatSessionProviderOptionItem }): ChatSessionOptionsMap {
623 return new Map(Object.entries(obj));
624 }
626 > export function toRecord(map: ReadonlyChatSessionOptionsMap): Record<string, string | IChatSessionProviderOptionItem> {
627 const record: Record<string, string | IChatSessionProviderOptionItem> = Object.create(null);
628 const entries = ensureIterable(map);
632 return record;
633 }
635 > export function toStrValueArray(map: ReadonlyChatSessionOptionsMap | undefined): Array<{ optionId: string; value: string }> | undefined {
636 if (!map) {
637 return undefined;
640 return Array.from(entries, ([optionId, value]) => ({ optionId, value: typeof value === 'string' ? value : value.id }));
641 }
643 > /**
644 > * Ensures the input is iterable. If a plain object is passed (e.g. due to
645 > * serialization across process boundaries losing the Map prototype), it is
646 > * converted to Map entries on the fly.
647 > */
648 > function ensureIterable(map: ReadonlyChatSessionOptionsMap): Iterable<[string, string | IChatSessionProviderOptionItem]> {
649 if (map instanceof Map) {
650 return map;
653 return Object.entries(map as unknown as Record<string, string | IChatSessionProviderOptionItem>);
654 }
656 >
657 > /**
658 > * Readonly version of {@link ChatSessionOptionsMap}
659 > */
660 > export type ReadonlyChatSessionOptionsMap = ReadonlyMap<string, string | IChatSessionProviderOptionItem>;
661 >
662 > export interface IChatSessionCustomizationItem {
663 > readonly label: string;
664 > readonly description?: string;
665 > readonly uri: URI;
666 > readonly storageLocation: number;
667 > readonly icon?: ThemeIcon;
668 > }
669 >
670 > export interface IChatSessionCustomizationItemGroup {
671 > readonly id: string;
672 > readonly items: IChatSessionCustomizationItem[];
673 > readonly commands?: readonly { readonly id: string; readonly title: string; readonly arguments?: readonly unknown[] }[];
674 > readonly itemCommands?: readonly { readonly id: string; readonly title: string; readonly arguments?: readonly unknown[] }[];
675 > }
676 >
677 > export interface IChatSessionCustomizationsProvider {
678 > readonly onDidChangeCustomizations: Event<void>;
679 > provideCustomizations(token: CancellationToken): Promise<IChatSessionCustomizationItemGroup[] | undefined>;
680 > }
681 >
682 >
683 > export interface IChatSessionCommitEvent {
684 > /** The original (untitled) session resource. */
685 > readonly original: URI;
686 > /** The committed (real) session resource. */
687 > readonly committed: URI;
688 > }
689 >
690 > export const IChatSessionsService = createDecorator<IChatSessionsService>('chatSessionsService');
691 >
692 > export interface IChatSessionsService {
693 > readonly _serviceBrand: undefined;
694 >
695 > // #region Chat session item provider support
696 > readonly onDidChangeItemsProviders: Event<{ readonly chatSessionType: string }>;
697 > readonly onDidChangeSessionItems: Event<IChatSessionItemsDelta>;
698 >
699 > /**
700 > * Fired when an untitled session is committed (URI swapped to a real resource)
701 > * after the first turn completes.
702 > */
703 > readonly onDidCommitSession: Event<IChatSessionCommitEvent>;
704 >
705 > readonly onDidChangeAvailability: Event<void>;
706 > readonly onDidChangeInProgress: Event<void>;
707 >
708 > getChatSessionContribution(chatSessionType: string): ResolvedChatSessionsExtensionPoint | undefined;
709 > getAllChatSessionContributions(): ResolvedChatSessionsExtensionPoint[];
710 >
711 > /**
712 > * Programmatically register a chat session contribution (for internal session types
713 > * that don't go through the extension point).
714 > */
715 > registerChatSessionContribution(contribution: IChatSessionsExtensionPoint): IDisposable;
716 >
717 > registerChatSessionItemController(chatSessionType: string, controller: IChatSessionItemController): IDisposable;
718 > getRegisteredChatSessionItemProviders(): readonly string[];
719 > activateChatSessionItemProvider(chatSessionType: string): Promise<void>;
720 >
721 > /**
722 > * Get the list of current chat session items grouped by session type.
723 > *
724 > * @param providerTypeFilter If specified, only returns items from the given providers. If undefined, returns items from all providers.
725 > *
726 > * @returns An async iterable that produces the list of session items for each provider. The order is not guaranteed. Some provider may take a long time to resolve.
727 > */
728 > getChatSessionItems(providerTypeFilter: readonly string[] | undefined, token: CancellationToken): AsyncIterable<{ readonly chatSessionType: string; readonly items: readonly IChatSessionItem[] }>;
729 >
730 > /**
731 > * Forces the controllers to refresh their session items, optionally filtered by provider type.
732 > */
733 > refreshChatSessionItems(providerTypeFilter: readonly string[] | undefined, token: CancellationToken): Promise<void>;
734 >
735 > /** @deprecated Use `getChatSessionItems` */
736 > getInProgress(): { chatSessionType: string; count: number }[];
737 >
738 > /**
739 > * Lazily resolves a chat session item, filling in expensive details like timing, changes, and badge.
740 > * Returns the resolved item, or undefined if no resolve handler is available.
741 > */
742 > resolveChatSessionItem(chatSessionType: string, resource: URI, token: CancellationToken): Promise<IChatSessionItem | undefined>;
743 >
744 > /**
745 > * Whether the registered item controller owns archived state for the session.
746 > */
747 > canSetChatSessionItemArchived(sessionResource: URI): boolean;
748 >
749 > /**
750 > * Sets archived state by delegating to the registered item controller.
751 > */
752 > setChatSessionItemArchived(sessionResource: URI, archived: boolean): void;
753 >
754 > // #endregion
755 >
756 > // #region Content provider support
757 > readonly onDidChangeContentProviderSchemes: Event<{ readonly added: string[]; readonly removed: string[] }>;
758 >
759 > getContentProviderSchemes(): string[];
760 >
761 > registerChatSessionContentProvider(scheme: string, provider: IChatSessionContentProvider): IDisposable;
762 > canResolveChatSession(sessionType: string): Promise<boolean>;
763 > getOrCreateChatSession(sessionResource: URI, token: CancellationToken): Promise<IChatSession>;
764 > /** Resolves a parsed response Markdown URI through its session content provider. */
765 > resolveChatResponseUri(sessionResource: URI, href: string, kind: 'link' | 'image'): string;
766 >
767 > /**
768 > * Compute completion items for an input being composed in the chat
769 > * session identified by `sessionResource`. Delegates to the registered
770 > * {@link IChatSessionContentProvider} for the session, if it implements
771 > * {@link IChatSessionContentProvider.provideChatInputCompletions}.
772 > * Returns `undefined` when no provider is available, in which case the
773 > * workbench's default in-process providers should be used.
774 > */
775 > provideChatInputCompletions(sessionResource: URI, params: IChatInputCompletionsParams, token: CancellationToken): Promise<IChatInputCompletionsResult | undefined>;
776 >
777 > /**
778 > * Trigger characters announced by the content provider for the given
779 > * session type. Used to dynamically register Monaco completion
780 > * providers per content-provider scheme. Returns `undefined` when the
781 > * scheme has no content provider, or `[]` when the provider does not
782 > * announce any trigger characters.
783 > */
784 > getChatInputCompletionTriggerCharacters(sessionType: string): Promise<readonly string[] | undefined>;
785 >
786 > getSessionOptions(sessionResource: URI): ReadonlyChatSessionOptionsMap | undefined;
787 > getSessionOption(sessionResource: URI, optionId: string): string | IChatSessionProviderOptionItem | undefined;
788 > setSessionOption(sessionResource: URI, optionId: string, value: string | IChatSessionProviderOptionItem): boolean;
789 > updateSessionOptions(sessionResource: URI, updates: ReadonlyChatSessionOptionsMap): boolean;
790 >
791 > /**
792 > * Fired when options for a chat session change.
793 > */
794 > readonly onDidChangeSessionOptions: Event<IChatSessionOptionsChangeEvent>;
795 >
796 > /**
797 > * Get the capabilities for a specific session type
798 > */
799 > getCapabilitiesForSessionType(chatSessionType: string): IChatAgentAttachmentCapabilities | undefined;
800 >
801 > /**
802 > * Get the customAgentTarget for a specific session type.
803 > * When the Target is not `Target.Undefined`, the mode picker should show filtered custom agents matching this target.
804 > */
805 > getCustomAgentTargetForSessionType(chatSessionType: string): Target;
806 >
807 > /**
808 > * Returns whether the session type requires custom models. When true, the model picker should show filtered custom models.
809 > */
810 > requiresCustomModelsForSessionType(chatSessionType: string): boolean;
811 >
812 > /**
813 > * Returns whether the session type supports the synthetic "Auto" model
814 > * fallback. The built-in local chat always supports it; contributed session
815 > * types default to `false` unless they set `supportsAutoModel`. When false
816 > * and no models are available, the picker shows a "No models available"
817 > * state instead of "Auto".
818 > */
819 > supportsAutoModelForSessionType(chatSessionType: string): boolean;
820 >
821 > /**
822 > * Whether the session type needs a Copilot account and so is unusable until the user signs in (BYOK isn't
823 > * supported). Defaults to false, so third-party types stay usable while signed out.
824 > */
825 > requiresCopilotSignInForSessionType(chatSessionType: string): boolean;
826 >
827 > /**
828 > * Returns whether the session type supports delegation.
829 > * Defaults to true when not explicitly set.
830 > */
831 > supportsDelegationForSessionType(chatSessionType: string): boolean;
832 >
833 > /**
834 > * Returns whether the loaded session supports forking conversations.
835 > */
836 > sessionSupportsFork(sessionResource: URI): boolean;
837 >
838 > /**
839 > * Forks a contributed chat session from the given request point.
840 > * @param sessionResource The session resource to fork.
841 > * @param request The request history item to fork from, or undefined to fork from the end.
842 > * @param token Cancellation token.
843 > * @returns The forked session item, or undefined if forking failed.
844 > */
845 > forkChatSession(sessionResource: URI, request: IChatSessionRequestHistoryItem | undefined, token: CancellationToken): Promise<IChatSessionItem>;
846 >
847 > /**
848 > * Returns whether the loaded session supports renaming.
849 > */
850 > sessionSupportsRename(sessionResource: URI): boolean;
851 >
852 > /**
853 > * Renames a contributed chat session.
854 > * @param sessionResource The session resource to rename.
855 > * @param title The new title for the session.
856 > * @param token Cancellation token.
857 > */
858 > renameChatSession(sessionResource: URI, title: string, token: CancellationToken): Promise<void>;
859 >
860 > readonly onDidChangeOptionGroups: Event<string>;
861 >
862 > getOptionGroupsForSessionType(chatSessionType: string): IChatSessionProviderOptionGroup[] | undefined;
863 > setOptionGroupsForSessionType(chatSessionType: string, handle: number, optionGroups?: readonly IChatSessionProviderOptionGroup[]): void;
864 >
865 > /**
866 > * Get the default options for new sessions of this type, derived from option groups'
867 > * `selected` or `default` items.
868 > */
869 > getNewChatSessionInputState(chatSessionType: string, sessionResource: URI): Promise<readonly IChatSessionProviderOptionGroup[] | undefined>;
870 >
871 > /**
872 > * Creates a new chat session item using the controller's newChatSessionItemHandler.
873 > * Returns undefined if the controller doesn't have a handler or if no controller is registered.
874 > */
875 > createNewChatSessionItem(chatSessionType: string, request: IChatNewSessionRequest, token: CancellationToken): Promise<IChatSessionItem | undefined>;
876 >
877 > /**
878 > * Permanently deletes a chat session item by delegating to the registered controller's `deleteChatSessionItem`
879 > * handler. Throws if the controller does not implement `deleteChatSessionItem`.
880 > */
881 > deleteChatSessionItem(sessionResource: URI, token: CancellationToken): Promise<void>;
882 >
883 > /**
884 > * Records the inverse `real → untitled` alias so option lookups for the real
885 > * session resolve to the untitled session's entry (e.g. {@link updateSessionOptions}).
886 > *
887 > * Call this BEFORE the real session loads, and never remove it — the real
888 > * session keeps reading its options through this alias even after the untitled
889 > * model is disposed. (Only the forward mapping is cleared, via
890 > * {@link clearMaterializedSessionResource}.) Publishing the forward mapping is a
891 > * separate step; see {@link setMaterializedSessionResource}.
892 > */
893 > registerSessionResourceAlias(untitledResource: URI, realResource: URI): void;
894 >
895 > /**
896 > * Records the forward `untitled → real` mapping (read via
897 > * {@link getMaterializedSessionResource}) so a late send still addressed to the
898 > * untitled resource re-targets the real session. Call this only AFTER the real
899 > * session has loaded.
900 > *
901 > * Kept separate from {@link registerSessionResourceAlias} on purpose: the
902 > * inverse alias must exist BEFORE the load (for option lookups), but this
903 > * forward mapping must appear only AFTER the real session exists — published
904 > * earlier, a failed or still-loading session would be re-targeted before it
905 > * exists (a later send would throw "Unknown session").
906 > */
907 > setMaterializedSessionResource(untitledResource: URI, realResource: URI): void;
908 >
909 > /**
910 > * Returns the real session resource that `untitledResource` materialized
911 > * into (via {@link setMaterializedSessionResource}), or `undefined` if it has
912 > * not materialized or the mapping was already cleared.
913 > */
914 > getMaterializedSessionResource(untitledResource: URI): URI | undefined;
915 >
916 > /**
917 > * Clears the forward `untitled → real` mapping for `sessionResource` (passed
918 > * either the untitled key or the real value), so {@link getMaterializedSessionResource}
919 > * stops re-targeting once the session is disposed. Does NOT remove the inverse
920 > * alias, which is intentionally permanent (see {@link registerSessionResourceAlias}).
921 > */
922 > clearMaterializedSessionResource(sessionResource: URI): void;
923 >
924 > /**
925 > * Fires {@link onDidCommitSession} to notify listeners that an untitled
926 > * session has been committed with a real resource URI.
927 > */
928 > fireSessionCommitted(original: URI, committed: URI): void;
929 >
930 > // #region Customizations provider support
931 > readonly onDidChangeCustomizations: Event<{ readonly chatSessionType: string }>;
932 > registerCustomizationsProvider(chatSessionType: string, provider: IChatSessionCustomizationsProvider): IDisposable;
933 > hasCustomizationsProvider(chatSessionType: string): boolean;
934 > getCustomizations(chatSessionType: string, token: CancellationToken): Promise<IChatSessionCustomizationItemGroup[] | undefined>;
935 > // #endregion
936 > }
937 >
938 > export function isSessionInProgressStatus(state: ChatSessionStatus): boolean {
939 return state === ChatSessionStatus.InProgress || state === ChatSessionStatus.NeedsInput;
940 }
942 > export function isIChatSessionFileChange2(obj: unknown): obj is IChatSessionFileChange2 {
943 const candidate = obj as IChatSessionFileChange2;
944 return candidate && candidate.uri instanceof URI && typeof candidate.insertions === 'number' && typeof candidate.deletions === 'number';
src/vs/platform/agentHost/common/state/sessionState.ts 822 covered LOC · 57 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- sessionState.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // Immutable state types for the sessions process protocol.
7 > // See protocol.md for the full design rationale.
8 > //
9 > // Most types are imported from the auto-generated protocol layer
10 > // (synced from the agent-host-protocol repo). This file adds VS Code-specific
11 > // helpers and re-exports.
12 >
13 > import { decodeBase64, encodeBase64, VSBuffer } from '../../../../base/common/buffer.js';
14 > import { hasKey, type Mutable } from '../../../../base/common/types.js';
15 > import { URI as ResourceURI } from '../../../../base/common/uri.js';
16 > import type { IProductService } from '../../../product/common/productService.js';
17 > import { readToolCallMeta } from '../meta/agentToolCallMeta.js';
18 > import {
19 > ResponsePartKind,
20 > SessionStatus,
21 > ToolCallStatus,
22 > SessionLifecycle,
23 > TerminalState,
24 > ToolResultContentType,
25 > ToolResultFileEditContent,
26 > ChatOriginKind,
27 > ChatInteractivity,
28 > type ActiveTurn,
29 > type ChangesetState,
30 > type ChatState,
31 > type ChatSummary,
32 > type PendingMessage,
33 > type Turn,
34 > type AnnotationsState,
35 > type URI as ProtocolURI,
36 > type RootState,
37 > type SessionState,
38 > type SessionSummary,
39 > type TextRange,
40 > type ToolCallCancelledState,
41 > type ToolCallCompletedState,
42 > type ToolCallResult,
43 > type ToolCallState,
44 > type ToolResultContent,
45 > type ToolResultSubagentContent,
46 > type ToolResultTextContent,
47 > type UsageInfo,
48 > type Message,
49 > } from './protocol/state.js';
50 >
51 > // Re-export everything from the protocol state module
52 > export {
53 > ChangesetOperationScope, ChangesetOperationStatus, ChangesetStatus, CustomizationLoadStatus,
54 > CustomizationType, MessageAttachmentKind, MessageKind,
55 > PendingMessageKind,
56 > PolicyState,
57 > ResponsePartKind,
58 > ChatInputAnswerState as SessionInputAnswerState,
59 > ChatInputAnswerValueKind as SessionInputAnswerValueKind,
60 > ChatInputQuestionKind as SessionInputQuestionKind,
61 > ChatInputResponseKind as SessionInputResponseKind,
62 > ChatInteractivity,
63 > ChatOriginKind,
64 > SessionLifecycle,
65 > SessionStatus, ToolCallCancellationReason, ToolCallConfirmationReason, ToolCallContributorKind, ToolCallRiskAssessmentKind, ToolCallRiskAssessmentStatus, ToolCallStatus,
66 > ToolResultContentType,
67 > TurnState, type ActiveTurn, type AgentCustomization, type AgentCapabilities, type AgentInfo, type AgentSelection, type Annotation, type AnnotationEntry, type AnnotationsState, type AnnotationsSummary, type Changeset, type ChangesetFile,
68 > type ChangesetOperation, type ChangesetState, type ChatState, type ChatSummary, type ChatOrigin, type ChildCustomization, type ClientPluginCustomization, type ConfigPropertySchema,
69 > type ConfigSchema,
70 > type ContentRef, type Customization, type CustomizationDegradedState,
71 > type CustomizationErrorState, type CustomizationLoadedState, type CustomizationLoadingState, type CustomizationLoadState, type DirectoryCustomization, type ErrorInfo, type HookCustomization, type FileEdit as ISessionFileDiff, type ToolResultEmbeddedResourceContent as IToolResultBinaryContent, type MarkdownResponsePart, type McpServerCustomization, type MessageAttachment,
72 > type MessageResourceAttachment, type MessageEmbeddedResourceAttachment, type MessageAnnotationsAttachment, type ModelSelection, type PendingMessage, type PluginCustomization, type ProjectInfo, type PromptCustomization, type ReasoningResponsePart,
73 > type ResponsePart,
74 > type RootState, type RuleCustomization, type SessionActiveClient,
75 > type SessionConfigState, type ChatInputAnswer as SessionInputAnswer,
76 > type ChatInputOption as SessionInputOption, type ChatInputQuestion as SessionInputQuestion, type ChatInputRequest as SessionInputRequest, type SessionModelInfo,
77 > type SessionState,
78 > type SessionSummary, type SkillCustomization, type Snapshot, type StringOrMarkdown, type TerminalState, type TextRange,
79 > type ToolAnnotations,
80 > type ToolCallCancelledState,
81 > type ToolCallCompletedState,
82 > type ToolCallPendingConfirmationState,
83 > type ToolCallPendingResultConfirmationState,
84 > type ToolCallResponsePart,
85 > type ToolCallResult,
86 > type ToolCallRiskAssessment,
87 > type ToolCallRiskAssessmentCompleteState,
88 > type ToolCallRiskAssessmentLoadingState,
89 > type ToolCallRunningState,
90 > type ToolCallState,
91 > type ToolCallStreamingState,
92 > type ToolCallContributor,
93 > type ToolDefinition, type ToolResultContent,
94 > type ToolResultFileEditContent,
95 > type TerminalCommandResult,
96 > type ToolResultSubagentContent,
97 > type ToolResultTerminalContent,
98 > type ToolResultTextContent,
99 > type Turn, type URI, type UsageInfo,
100 > type Message
101 > } from './protocol/state.js';
102 >
103 > /**
104 > * Well-known keys that may appear on {@link UsageInfo._meta}.
105 > * Clients MAY read these to provide enhanced UI (e.g. credit cost display).
106 > */
107 > export interface UsageInfoMeta {
108 > /** Per-turn credit cost reported by the backend. */
109 > cost?: number;
110 > /** The concrete model selected by Copilot Auto and the routing explanation. */
111 > autoModeResolved?: IAutoModeResolvedInfo;
112 > /** Copilot-specific usage breakdown, including nano-AIU totals. */
113 > copilotUsage?: {
114 > totalNanoAiu?: number;
115 > [key: string]: unknown;
116 > };
117 > /**
118 > * Per-category account quota snapshots reported by the backend on the
119 > * model-call usage event, keyed by quota type (e.g. `chat`,
120 > * `premium_interactions`). Clients MAY use these to keep the account quota
121 > * UI current without a separate quota fetch.
122 > */
123 > quotaSnapshots?: {
124 > [quotaType: string]: {
125 > readonly isUnlimitedEntitlement?: boolean;
126 > readonly entitlementRequests?: number;
127 > readonly usedRequests?: number;
128 > readonly remainingPercentage?: number;
129 > readonly overage?: number;
130 > readonly overageAllowedWithExhaustedQuota?: boolean;
131 > /** ISO 8601 date when the quota resets, if applicable. */
132 > readonly resetDate?: string;
133 > } | undefined;
134 > };
135 > /**
136 > * Per-source context-window attribution breakdown reported by the SDK's
137 > * `session.rpc.metadata.getContextAttribution()`. Populated asynchronously
138 > * after each usage event and piped to the context-usage widget as
139 > * `promptTokenDetails`.
140 > */
141 > contextAttribution?: IContextAttributionData;
142 > [key: string]: unknown;
143 > }
144 >
145 > export interface IAutoModeResolvedInfo {
146 > readonly chosenModel: string;
147 > readonly reasoningBucket?: 'low' | 'medium' | 'high';
148 > readonly categoryScores?: Readonly<Record<string, number | undefined>>;
149 > readonly predictedLabel?: string;
150 > readonly confidence?: number;
151 > readonly candidateModels?: readonly string[];
152 > }
153 >
154 > /**
155 > * Mirrors the SDK's `SessionContextAttribution` shape — a flat list of
156 > * per-source entries describing what occupies the session's context window.
157 > */
158 > export interface IContextAttributionData {
159 > readonly totalTokens: number;
160 > readonly entries: readonly IContextAttributionEntry[];
161 > readonly compactions: { readonly count: number };
162 > }
163 >
164 > export interface IContextAttributionEntry {
165 > readonly kind: string;
166 > readonly id: string;
167 > readonly label: string;
168 > readonly tokens: number;
169 > readonly parentId?: string;
170 > readonly attributes?: Readonly<Record<string, string | undefined>>;
171 > }
172 >
173 > type AccountQuotaSnapshot = NonNullable<NonNullable<UsageInfoMeta['quotaSnapshots']>[string]>;
174 >
175 function readAccountQuotaSnapshot(value: unknown): AccountQuotaSnapshot | undefined {
176 if (!value || typeof value !== 'object' || Array.isArray(value)) {
188 return snapshot;
189 }
191 > /**
192 > * Reads the well-known {@link UsageInfoMeta} keys from a usage report's open
193 > * `_meta` bag, ignoring unrelated provider-specific keys and validating each
194 > * field's type. Always read {@link UsageInfo._meta} through this helper rather
195 > * than casting the bag to {@link UsageInfoMeta}, so a malformed or partial bag
196 > * degrades to absent fields instead of producing values of the wrong runtime
197 > * type. Returns an empty object when the bag is absent.
198 > */
199 > export function readUsageInfoMeta(usage: UsageInfo | undefined): UsageInfoMeta {
200 const meta = usage?._meta;
201 if (!meta) {
227 return result;
228 }
230 function readAutoModeResolvedInfo(value: unknown): IAutoModeResolvedInfo | undefined {
231 if (!value || typeof value !== 'object' || Array.isArray(value)) {
258 return result;
259 }
261 function readContextAttribution(value: unknown): IContextAttributionData | undefined {
262 if (!value || typeof value !== 'object' || Array.isArray(value)) {
295 return { totalTokens: raw['totalTokens'] as number, entries, compactions };
296 }
298 function filterStringAttributes(raw: Record<string, unknown>): Record<string, string | undefined> {
299 const result: Record<string, string | undefined> = {};
305 return result;
306 }
308 > export {
309 > ChangesetOperationTargetKind, type ChangesetOperationFollowUp, type ChangesetOperationTarget
310 > } from './protocol/commands.js';
311 >
312 > // Canonical chat-input type names (the protocol renamed the former
313 > // `SessionInput*` types to `ChatInput*` when input requests moved onto the
314 > // chat channel). Re-exported here so consumers can import them from the glue
315 > // layer alongside the legacy `SessionInput*` aliases above.
316 > export {
317 > ChatInputAnswerState,
318 > ChatInputAnswerValueKind,
319 > ChatInputQuestionKind,
320 > ChatInputResponseKind,
321 > type ChatInputAnswer,
322 > type ChatInputOption,
323 > type ChatInputQuestion,
324 > type ChatInputRequest,
325 > type InputRequestResponsePart,
326 > } from './protocol/state.js';
327 >
328 > // ---- File edit kind ---------------------------------------------------------
329 >
330 > /**
331 > * The kind of file edit operation. Derived from the presence/absence of
332 > * `before`/`after` in {@link ToolResultFileEditContent}.
333 > */
334 > export const enum FileEditKind {
335 > /** Content edit (same file URI, different content). */
336 > Edit = 'edit',
337 > /** File creation (no before state). */
338 > Create = 'create',
339 > /** File deletion (no after state). */
340 > Delete = 'delete',
341 > /** File rename/move (different before and after URIs). */
342 > Rename = 'rename',
343 > }
344 >
345 > // ---- Well-known URIs --------------------------------------------------------
346 >
347 > /** URI for the root state subscription. */
348 > export const ROOT_STATE_URI = 'ahp-root://';
349 >
350 > /** Scheme used by {@link ROOT_STATE_URI}. */
351 > export const AHP_ROOT_SCHEME = 'ahp-root';
352 >
353 > /** Scheme used by resource-watch channel URIs (`ahp-resource-watch:/<encoded>`). */
354 > export const AHP_RESOURCE_WATCH_SCHEME = 'ahp-resource-watch';
355 >
356 > /**
357 > * Encode a resource-watch descriptor into its canonical channel URI. The
358 > * descriptor is serialised into the URI path so the receiver can recover
359 > * the watch parameters without any server-side bookkeeping — subscribe is
360 > * the only point where state is materialised (an `IFileService` watcher
361 > * is attached on the first subscriber and held through a grace window
362 > * after the last drops).
363 > */
364 > export function buildResourceWatchChannelUri(descriptor: {
365 readonly root: string;
366 readonly recursive?: boolean;
380 return `${AHP_RESOURCE_WATCH_SCHEME}://r/${json}`;
381 }
383 > /**
384 > * Inverse of {@link buildResourceWatchChannelUri}. Returns `undefined` if
385 > * `uri` is not a well-formed `ahp-resource-watch:` URI — callers should
386 > * surface that as a not-found error to the client.
387 > */
388 > export function parseResourceWatchChannelUri(uri: string): {
389 root: string;
390 recursive: boolean;
421 }
422 }
424 > /** Returns `true` when `uri` identifies a resource-watch channel. */
425 > export function isAhpResourceWatchChannel(uri: string): boolean {
426 try {
427 return ResourceURI.parse(uri).scheme === AHP_RESOURCE_WATCH_SCHEME;
430 }
431 }
433 > /**
434 > * Returns `true` when `uri` identifies the root channel, regardless of
435 > * whether the caller passes the canonical wire form (`'ahp-root://'`) or a
436 > * variant that has been round-tripped through the workbench {@link URI} class
437 > * (which normalizes the authority-less form to `'ahp-root:'`). Always prefer
438 > * this helper over a direct `=== ROOT_STATE_URI` comparison so the two
439 > * spellings stay interchangeable.
440 > */
441 > export function isAhpRootChannel(uri: string): boolean {
442 if (uri === ROOT_STATE_URI) {
443 return true;
449 }
450 }
452 > /**
453 > * Mints a session-unique opaque id for a customization, derived from its
454 > * source URI and (when present) its `range` within the source. Plugins MAY
455 > * declare multiple children (e.g. MCP servers, hooks) inside the same
456 > * manifest file; including the range disambiguates them without an extra
457 > * mapping table.
458 > *
459 > * The range is appended as a reserved `#range=` query-style suffix; any
460 > * existing `#` in the URI is percent-encoded first so a source URI that
461 > * already contains a fragment cannot collide with a ranged id.
462 > */
463 > export function customizationId(uri: string, range?: TextRange): string {
464 if (!range) {
465 return uri;
468 return `${safeUri}#range=${range.start.line}:${range.start.character}-${range.end.line}:${range.end.character}`;
469 }
471 > // ---- VS Code-specific derived types -----------------------------------------
472 >
473 > /**
474 > * A tool call in a terminal state, stored in completed turns.
475 > */
476 > export type ICompletedToolCall = ToolCallCompletedState | ToolCallCancelledState;
477 >
478 > /**
479 > * Derived status type for the tool call lifecycle.
480 > */
481 > export type ToolCallStatusString = ToolCallState['status'];
482 >
483 > // ---- Tool output helper -----------------------------------------------------
484 >
485 > /**
486 > * Extracts a plain-text tool output string from a tool call result's `content`
487 > * array. Joins all text-type content parts into a single string.
488 > *
489 > * Returns `undefined` if there are no text content parts.
490 > */
491 > export function getToolOutputText(result: ToolCallResult): string | undefined {
492 if (!result.content || result.content.length === 0) {
493 return undefined;
504 return textParts.map(p => p.text).join('\n');
505 }
507 > /**
508 > * Extracts file edit content entries from a tool call result's `content` array.
509 > * Returns an empty array if there are no file edit content parts.
510 > */
511 > export function getToolFileEdits(result: ToolCallResult): ToolResultFileEditContent[] {
512 if (!result.content || result.content.length === 0) {
513 return [];
521 return edits;
522 }
524 > /**
525 > * Extracts the first subagent content entry from a tool call's `content` array.
526 > * Works with both completed tool call results and running tool call states.
527 > * Returns `undefined` if there are no subagent content parts.
528 > */
529 > export function getToolSubagentContent(result: { content?: readonly ToolResultContent[] }): ToolResultSubagentContent | undefined {
530 if (!result.content || result.content.length === 0) {
531 return undefined;
538 return undefined;
539 }
541 > // ---- Subagent URI helpers ---------------------------------------------------
542 >
543 > const SUBAGENT_URI_SEGMENT = 'subagent';
544 > const SUBAGENT_URI_MARKER = `/${SUBAGENT_URI_SEGMENT}/`;
545 > const SUBAGENT_URI_PATH_REGEX = /^(?<parentPath>.+)\/subagent\/(?<toolCallId>.+)$/;
546 >
547 function asResourceUri(uri: ProtocolURI | ResourceURI): ResourceURI {
548 return typeof uri === 'string' ? ResourceURI.parse(uri) : uri;
549 }
551 function getSubagentBasePath(parentSession: ProtocolURI | ResourceURI): { parent: ResourceURI; path: string } {
552 const parent = asResourceUri(parentSession);
554 return { parent, path: `${parentPath}${SUBAGENT_URI_MARKER}` };
555 }
557 > /**
558 > * Builds a subagent session URI from a parent session URI and tool call ID.
559 > * Convention: `{parentSessionUri}/subagent/{toolCallId}`
560 > */
561 > export function buildSubagentSessionUri(parentSession: ProtocolURI | ResourceURI, toolCallId: string): string {
562 const { parent, path } = getSubagentBasePath(parentSession);
563 return parent.with({ path: `${path}${toolCallId}` }).toString();
564 }
566 > /**
567 > * Parses a subagent session URI into its parent session URI and tool call ID.
568 > * Returns `undefined` if the URI does not follow the subagent convention.
569 > */
570 > export function parseSubagentSessionUri(uri: ProtocolURI | ResourceURI): { parentSession: ResourceURI; toolCallId: string } | undefined {
571 const resource = asResourceUri(uri);
572 const match = SUBAGENT_URI_PATH_REGEX.exec(resource.path);
579 };
580 }
582 > /**
583 > * Returns whether a session URI represents a subagent session.
584 > */
585 > export function isSubagentSession(uri: ProtocolURI | ResourceURI): boolean {
586 return parseSubagentSessionUri(uri) !== undefined;
587 }
589 > /**
590 > * Builds the string prefix used by the state manager for cached subagent sessions.
591 > */
592 > export function buildSubagentSessionUriPrefix(parentSession: ProtocolURI | ResourceURI): string {
593 const { parent, path } = getSubagentBasePath(parentSession);
594 return parent.with({ path }).toString();
595 }
597 > // ---- Factory helpers --------------------------------------------------------
598 >
599 > export function createRootState(): RootState {
600 return {
601 agents: [],
603 };
604 }
606 > /**
607 > * Creates the initial flat {@link SessionState} for a session from its
608 > * root-channel {@link SessionSummary} catalog entry. Session metadata
609 > * ({@link SessionMetadata}) — and the shared `_meta` bag — are inlined directly
610 > * onto the state.
611 > */
612 > export function createSessionState(summary: SessionSummary): SessionState {
613 const state: SessionState = {
614 provider: summary.provider,
627 return state;
628 }
630 > /**
631 > * Creates an empty {@link ChatState} for a chat. The summary fields are
632 > * denormalized onto the chat state per the protocol contract; callers pass
633 > * the chat's catalog summary and this seeds an empty conversation.
634 > */
635 > export function createChatState(summary: ChatSummary): ChatState {
636 return {
637 resource: summary.resource,
648 };
649 }
651 > /**
652 > * Derives the default-chat {@link ChatSummary} for a session from its
653 > * {@link SessionSummary}. The default chat inherits the session's title,
654 > * status, activity and working directory, and is marked as a
655 > * {@link ChatOriginKind.User | user-originated} chat. Both the session and
656 > * chat `modifiedAt` are ISO-8601 strings, so it is carried over directly.
657 > */
658 > export function createDefaultChatSummary(session: SessionSummary, chatUri: ProtocolURI): ChatSummary {
659 const summary: ChatSummary = {
660 resource: chatUri,
675 return summary;
676 }
678 > /** Activity bits (0-4) of {@link SessionStatus}; the high bits carry orthogonal flags (IsRead / IsArchived). */
679 > const STATUS_ACTIVITY_MASK = (1 << 5) - 1;
680 >
681 > /** Whether the active turn has a `PendingConfirmation` tool call auto-approved by the session's bypass setting. */
682 function hasAutoApprovedPendingConfirmation(state: ChatState): boolean {
683 return !!state.activeTurn?.responseParts.some(part =>
687 );
688 }
690 > /** Whether the chat is genuinely blocked on user input (an open input request, an auth-required tool, or a non-auto-approved confirmation gate). */
691 function chatAwaitsUserInput(state: ChatState): boolean {
692 return !!state.activeTurn?.responseParts.some(part => {
708 });
709 }
711 > /**
712 > * Projects a chat's status for session-summary aggregation, demoting an
713 > * `InputNeeded` back to `InProgress` only when it is caused solely by an
714 > * auto-approved confirmation — otherwise a session with bypass approvals flashes
715 > * "input needed" in the sessions list while an auto-approved tool runs.
716 > */
717 function chatSummaryStatus(state: ChatState): SessionStatus {
718 const status = state.status;
728 return status;
729 }
731 > /**
732 > * Derives a {@link ChatSummary} from a fully-populated {@link ChatState} by
733 > * projecting out the denormalized summary fields. Used to keep the parent
734 > * session's `chats` catalog in sync with a chat's denormalized state.
735 > */
736 > export function chatSummaryFromState(state: ChatState): ChatSummary {
737 const summary: ChatSummary = {
738 resource: state.resource,
748 return summary;
749 }
751 > /**
752 > * The effective interactivity of a chat given its session's archived state.
753 > *
754 > * `interactivity` is the general read-only mechanism (e.g. subagent worker
755 > * chats are `ReadOnly`). An archived session is read-only too, so its
756 > * interactive chats are downgraded to `ReadOnly`. `Hidden` chats stay hidden —
757 > * archiving only downgrades `Full` chats. Absent interactivity defaults to
758 > * `Full` for backward compatibility.
759 > *
760 > * The host uses this to enforce read-only turns off a single signal
761 > * ({@link isChatReadOnly}) rather than special-casing archived; the same rule
762 > * is mirrored client-side to hide the composer.
763 > */
764 > export function effectiveChatInteractivity(interactivity: ChatInteractivity | undefined, sessionArchived: boolean): ChatInteractivity {
765 if (interactivity === ChatInteractivity.Hidden) {
766 return ChatInteractivity.Hidden;
771 return interactivity ?? ChatInteractivity.Full;
772 }
774 > /**
775 > * Whether a chat rejects user-dispatched turns, given its own interactivity and
776 > * its session's archived state. `true` for `ReadOnly` chats (including archived
777 > * sessions' interactive chats). See {@link effectiveChatInteractivity}.
778 > */
779 > export function isChatReadOnly(interactivity: ChatInteractivity | undefined, sessionArchived: boolean): boolean {
780 return effectiveChatInteractivity(interactivity, sessionArchived) === ChatInteractivity.ReadOnly;
781 }
783 > export function createActiveTurn(id: string, message: Message, startedAt: string): ActiveTurn {
784 return {
785 id,
790 };
791 }
793 > export const enum StateComponents {
794 > Root,
795 > Session,
796 > Chat,
797 > Terminal,
798 > Changeset,
799 > Annotations,
800 > }
801 >
802 > export type ComponentToState = {
803 > [StateComponents.Root]: RootState;
804 > [StateComponents.Session]: SessionState;
805 > [StateComponents.Chat]: ChatState;
806 > [StateComponents.Terminal]: TerminalState;
807 > [StateComponents.Changeset]: ChangesetState;
808 > [StateComponents.Annotations]: AnnotationsState;
809 > };
810 >
811 > // ---- Default chat URI helpers ----------------------------------------------
812 >
813 > /** Scheme used by chat channel URIs (`ahp-chat://...`). */
814 > export const AHP_CHAT_SCHEME = 'ahp-chat';
815 >
816 > /** Chat id of the default chat that every session owns. */
817 > export const DEFAULT_CHAT_ID = 'default';
818 >
819 > /**
820 > * Derives the deterministic channel URI for a chat within a session. Every chat
821 > * — the default chat and any additional peer chats — encodes its owning session
822 > * URI into the path so producers and consumers can recover the session without a
823 > * lookup table (see {@link parseChatUri}). The chat id is carried in the URI
824 > * authority.
825 > *
826 > * `ahp-chat://<chatId>/<base64(sessionUri)>`
827 > */
828 > export function buildChatUri(sessionUri: ProtocolURI | ResourceURI, chatId: string): string {
829 const session = typeof sessionUri === 'string' ? sessionUri : sessionUri.toString();
830 const encoded = encodeBase64(VSBuffer.fromString(session), false, true);
831 return `${AHP_CHAT_SCHEME}://${chatId}/${encoded}`;
832 }
834 > /**
835 > * Derives the deterministic default-chat channel URI for a session. While the
836 > * protocol allows a session to contain many chats, every session always owns a
837 > * default chat whose URI is derived from the owning session URI so producers and
838 > * consumers can compute it without a lookup table.
839 > *
840 > * The session URI is encoded into the path so {@link parseChatUri} can recover
841 > * it.
842 > */
843 > export function buildDefaultChatUri(sessionUri: ProtocolURI | ResourceURI): string {
844 return buildChatUri(sessionUri, DEFAULT_CHAT_ID);
845 }
847 > const SUBAGENT_CHAT_ID = 'subagent';
848 >
849 > export function isSubagentChatUri(uri: ProtocolURI | ResourceURI): boolean {
850 const parsed = typeof uri === 'string' ? ResourceURI.parse(uri) : uri;
851 return parsed.scheme === AHP_CHAT_SCHEME && parsed.authority === SUBAGENT_CHAT_ID;
852 }
854 > export function buildSubagentChatUri(sessionUri: ProtocolURI | ResourceURI, toolCallId: string): string {
855 const session = typeof sessionUri === 'string' ? sessionUri : sessionUri.toString();
856 const encoded = encodeBase64(VSBuffer.fromString(session), false, true);
857 return `${AHP_CHAT_SCHEME}://${SUBAGENT_CHAT_ID}/${encoded}/${encodeURIComponent(toolCallId)}`;
858 }
860 > /**
861 > * Inverse of {@link buildChatUri}: recovers the owning session URI and chat id
862 > * from any chat channel URI. Returns `undefined` when `uri` is not a well-formed
863 > * chat URI.
864 > */
865 > export function parseChatUri(uri: ProtocolURI | ResourceURI): { session: string; chatId: string } | undefined {
866 let parsed: ResourceURI;
867 try {
891 }
892 }
894 > /**
895 > * Inverse of {@link buildDefaultChatUri}: recovers the owning session URI from a
896 > * chat channel URI. Returns `undefined` when `uri` is not a well-formed chat URI.
897 > * Accepts any chat URI (default or additional) so callers that only need the
898 > * parent session can use it uniformly.
899 > */
900 > export function parseDefaultChatUri(uri: ProtocolURI | ResourceURI): string | undefined {
901 return parseChatUri(uri)?.session;
902 }
904 > export function parseRequiredSessionUriFromChatUri(uri: ProtocolURI | ResourceURI): string {
905 const session = parseDefaultChatUri(uri);
906 if (session === undefined) {
909 return session;
910 }
912 > /** Returns `true` when `uri` is the default chat of its session. */
913 > export function isDefaultChatUri(uri: ProtocolURI | ResourceURI): boolean {
914 return parseChatUri(uri)?.chatId === DEFAULT_CHAT_ID;
915 }
917 > /**
918 > * Resolves a feature-level `(session, chat)` pair to the single chat URI used by
919 > * the agent session/chat surface. A session always owns a DEFAULT chat addressed
920 > * by the session URI itself; additional (peer) chats are addressed by their own
921 > * chat channel URIs. This is the one place default-chat resolution lives so
922 > * agents never re-derive "is this the default chat?".
923 > */
924 > export function resolveChatUri(session: ResourceURI, chat: ResourceURI): ResourceURI {
925 return isDefaultChatUri(chat) ? session : chat;
926 }
928 > /** Returns `true` when `uri` identifies a chat channel. */
929 > export function isAhpChatChannel(uri: string): boolean {
930 try {
931 return ResourceURI.parse(uri).scheme === AHP_CHAT_SCHEME;
934 }
935 }
937 > // ---- Session + default-chat composite --------------------------------------
938 >
939 > /**
940 > * A single chat's effective session context: the shared {@link SessionState}
941 > * (working directories, active clients, config, customizations/MCP scope, …)
942 > * resolved for one chat and merged with that chat's conversation contents.
943 > *
944 > * The protocol moved turns and pending state off the session and onto a
945 > * per-chat channel, and lets a chat override the session's working directories
946 > * with a subset (e.g. {@link ChatState.workingDirectories}) and carry its own
947 > * read-only {@link ChatState.primaryWorkingDirectory | primary} (fixed at chat
948 > * creation — the session has no primary). This composite recombines the session
949 > * with one of its chats — default or peer — so consumers read the chat's
950 > * effective context and conversation through one object without walking back to
951 > * the session to re-derive shared state. The {@link ISessionWithDefaultChat.workingDirectories}
952 > * carry the chat's *effective* working directories (its own subset override when
953 > * present, else the session's full set); {@link ISessionWithDefaultChat.primaryWorkingDirectory}
954 > * is the chat's own primary.
955 > */
956 > export interface ISessionWithDefaultChat extends SessionState {
957 > /** The chat's read-only primary working directory (fixed at chat creation). */
958 > primaryWorkingDirectory?: ProtocolURI;
959 > /** Completed turns of this chat. */
960 > turns: Turn[];
961 > /** Currently in-progress turn of this chat. */
962 > activeTurn?: ActiveTurn;
963 > /** Steering message pending on this chat. */
964 > steeringMessage?: PendingMessage;
965 > /** Queued messages pending on this chat. */
966 > queuedMessages?: PendingMessage[];
967 > /** Draft input of this chat. */
968 > draft?: Message;
969 > }
970 >
971 > /**
972 > * Projects a {@link SessionState} and one of its {@link ChatState | chats}
973 > * (default or peer) into that chat's {@link ISessionWithDefaultChat | effective
974 > * session context}. Per-chat overrides (the working-directories subset and the
975 > * chat's own primary) are layered over the session defaults, and the
976 > * conversation fields are taken from the chat. When the chat state is absent
977 > * (e.g. not yet hydrated) the conversation fields default to empty and the
978 > * session defaults apply.
979 > */
980 > export function mergeSessionWithDefaultChat(session: SessionState, chat: ChatState | undefined): ISessionWithDefaultChat {
981 return {
982 ...session,
990 };
991 }
993 > /**
994 > * Resolves the active turn of a session's default chat, if any.
995 > */
996 > export function getActiveTurn(chat: ChatState | undefined): ActiveTurn | undefined {
997 return chat?.activeTurn;
998 }
1000 > /**
1001 > * Resolves the default chat's catalog summary from a session, if present.
1002 > */
1003 > export function getDefaultChat(session: SessionState): ChatSummary | undefined {
1004 if (session.defaultChat !== undefined) {
1005 const match = session.chats.find(c => c.resource === session.defaultChat);
1010 return session.chats[0];
1011 }
1013 > // ---- SessionMeta accessors -------------------------------------------------
1014 >
1015 > /**
1016 > * VS Code-side alias for the protocol's open `_meta` property bag on
1017 > * {@link SessionState}. Keys SHOULD be namespaced (e.g. `git`, `vscode.foo`)
1018 > * to avoid collisions; values MUST be JSON-serializable.
1019 > */
1020 > export type SessionMeta = Record<string, unknown>;
1021 >
1022 > /**
1023 > * VS Code-side alias for the protocol's open `_meta` property bag on
1024 > * {@link SessionSummary}. Keys SHOULD be namespaced (e.g. `git`, `vscode.foo`)
1025 > * to avoid collisions; values MUST be JSON-serializable.
1026 > */
1027 > export type SessionSummaryMeta = Record<string, unknown>;
1028 >
1029 > /**
1030 > * Reserved key under {@link SessionMeta} for the well-known git-state
1031 > * payload. Value at this key, when present, MUST be shaped like
1032 > * {@link ISessionGitState}. This is a VS Code-specific convention layered
1033 > * on top of the protocol's generic `_meta` bag — the protocol itself does
1034 > * not know about git state.
1035 > */
1036 > export const SESSION_META_GIT_KEY = 'git';
1037 >
1038 > /**
1039 > * Reserved key under {@link SessionMeta} for the well-known GitHub-state
1040 > * payload. Value at this key, when present, MUST be shaped like
1041 > * {@link ISessionGitHubState}. This is a VS Code-specific convention layered
1042 > * on top of the protocol's generic `_meta` bag — the protocol itself does
1043 > * not know about GitHub state.
1044 > */
1045 > export const SESSION_META_GITHUB_KEY = 'github';
1046 >
1047 > export const SESSION_META_PROMPT_CACHE_KEY = 'vscode.promptCache';
1048 >
1049 > /** Latest known prompt-cache state for the model active in an agent session. */
1050 > export interface ISessionPromptCacheState {
1051 > readonly modelId: string;
1052 > readonly cacheExpiresAt: string;
1053 > }
1054 >
1055 > /** Reads the latest known prompt-cache state from session metadata. */
1056 > export function readSessionPromptCacheState(meta: SessionMeta | undefined): ISessionPromptCacheState | undefined {
1057 const value = meta?.[SESSION_META_PROMPT_CACHE_KEY];
1058 if (!value || typeof value !== 'object' || Array.isArray(value)) {
1064 : undefined;
1065 }
1067 > /** Returns session metadata with the prompt-cache slot updated or removed. */
1068 > export function withSessionPromptCacheState(meta: SessionMeta | undefined, promptCache: ISessionPromptCacheState | undefined): SessionMeta | undefined {
1069 const next: SessionMeta = { ...meta };
1070 if (promptCache) {
1075 return Object.keys(next).length > 0 ? next : undefined;
1076 }
1078 > /**
1079 > * Git state of a session's working directory, carried under
1080 > * {@link SessionMeta} at {@link SESSION_META_GIT_KEY}. Used by clients to
1081 > * drive source-control affordances (e.g. PR/merge buttons in the Agents
1082 > * app).
1083 > *
1084 > * All fields are optional — agents that do not track a particular field
1085 > * should omit it rather than send a placeholder, so clients can distinguish
1086 > * "unknown" from "known to be zero".
1087 > */
1088 > export interface ISessionGitState {
1089 > /** Whether the working directory has a `github.com` git remote. */
1090 > readonly hasGitHubRemote?: boolean;
1091 > /** Current branch name. */
1092 > readonly branchName?: string;
1093 > /** Base branch the work targets (e.g. `main`). */
1094 > readonly baseBranchName?: string;
1095 > /** Upstream tracking branch (e.g. `origin/feature`). */
1096 > readonly upstreamBranchName?: string;
1097 > /** Number of commits the upstream branch has ahead of the local branch. */
1098 > readonly incomingChanges?: number;
1099 > /** Number of commits the local branch has ahead of the upstream branch. */
1100 > readonly outgoingChanges?: number;
1101 > /** Number of files with uncommitted changes. */
1102 > readonly uncommittedChanges?: number;
1103 > /** GitHub repository owner parsed from the working copy's GitHub remote (preferring `origin`, falling back to the first GitHub remote). */
1104 > readonly githubOwner?: string;
1105 > /** GitHub repository name parsed from the working copy's GitHub remote (preferring `origin`, falling back to the first GitHub remote). */
1106 > readonly githubRepo?: string;
1107 > }
1108 >
1109 > /**
1110 > * GitHub state of a session, carried under {@link SessionMeta} at
1111 > * {@link SESSION_META_GITHUB_KEY}. Used by clients to drive GitHub-specific
1112 > * affordances (e.g. PR/merge buttons in the Agents app).
1113 > *
1114 > * All fields are optional — agents that do not track a particular field
1115 > * should omit it rather than send a placeholder, so clients can distinguish
1116 > * "unknown" from "known to be zero".
1117 > */
1118 > export interface ISessionGitHubState {
1119 > /** The owner of the GitHub repository. */
1120 > readonly owner?: string;
1121 > /** The name of the GitHub repository. */
1122 > readonly repo?: string;
1123 > /** The URL of the GitHub pull request. */
1124 > readonly pullRequestUrl?: string;
1125 > }
1126 >
1127 > /**
1128 > * Reads the well-known git-state payload from {@link SessionMeta}, if
1129 > * present. Returns `undefined` when the meta bag is absent or the value at
1130 > * the git key is not a plain object (e.g. an array or a primitive).
1131 > * Individual fields with wrong types are silently dropped so partial state
1132 > * still propagates.
1133 > *
1134 > * Unlike the other typed readers, this takes the raw {@link SessionMeta} value
1135 > * rather than its parent {@link SessionState}: the sessions provider stores and
1136 > * reads a detached meta snapshot without retaining the owning state.
1137 > */
1138 > export function readSessionGitState(meta: SessionMeta | undefined): ISessionGitState | undefined {
1139 const value = meta?.[SESSION_META_GIT_KEY];
1140 if (!value || typeof value !== 'object' || Array.isArray(value)) {
1164 return result;
1165 }
1167 > /**
1168 > * Returns a new {@link SessionMeta} with the git-state payload set to
1169 > * `gitState`, or with the git slot removed if `gitState` is `undefined`.
1170 > * Returns `undefined` if the result would be empty.
1171 > */
1172 > export function withSessionGitState(meta: SessionMeta | undefined, gitState: ISessionGitState | undefined): SessionMeta | undefined {
1173 const next: { [key: string]: unknown } = { ...meta };
1174 if (gitState !== undefined) {
1179 return Object.keys(next).length > 0 ? next : undefined;
1180 }
1182 > /**
1183 > * Reads the well-known GitHub state payload from {@link SessionSummaryMeta}, if
1184 > * present. Returns `undefined` when the meta bag is absent or the value at the
1185 > * GitHub key is not a plain object (e.g. an array or a primitive).
1186 > * Individual fields with wrong types are silently dropped so partial state
1187 > * still propagates.
1188 > *
1189 > * Unlike the other typed readers, this takes the raw {@link SessionSummaryMeta}
1190 > * value rather than its parent {@link SessionState}: the sessions provider stores and
1191 > * reads a detached meta snapshot without retaining the owning state.
1192 > */
1193 > export function readSessionGitHubState(meta: SessionSummaryMeta | undefined): ISessionGitHubState | undefined {
1194 const value = meta?.[SESSION_META_GITHUB_KEY];
1195 if (!value || typeof value !== 'object' || Array.isArray(value)) {
1208 return result;
1209 }
1211 > /**
1212 > * Returns a new {@link SessionSummaryMeta} with the GitHub-state payload set to
1213 > * `gitHubState`, or with the GitHub slot removed if `gitHubState` is `undefined`.
1214 > * Returns `undefined` if the result would be empty.
1215 > */
1216 > export function withSessionGitHubState(meta: SessionSummaryMeta | undefined, gitHubState: ISessionGitHubState | undefined): SessionSummaryMeta | undefined {
1217 const next: { [key: string]: unknown } = { ...meta };
1218 if (gitHubState !== undefined) {
1223 return Object.keys(next).length > 0 ? next : undefined;
1224 }
1226 > /**
1227 > * Reserved key under {@link SessionSummaryMeta} recording how deeply a session
1228 > * was spawned via the `create_session` host tool (0 for a top-level, user-created
1229 > * session). Used to bound recursive session creation. VS Code-specific convention
1230 > * layered on top of the protocol's generic `_meta` bag.
1231 > */
1232 > export const SESSION_META_SPAWN_DEPTH_KEY = 'agentHost/sessionSpawnDepth';
1233 >
1234 > /**
1235 > * Reads the `create_session` spawn depth from a {@link SessionSummaryMeta} bag,
1236 > * returning `0` when the key is absent or not a finite number.
1237 > */
1238 > export function readSessionSpawnDepth(meta: SessionSummaryMeta | undefined): number {
1239 const value = meta?.[SESSION_META_SPAWN_DEPTH_KEY];
1240 return typeof value === 'number' && Number.isFinite(value) ? value : 0;
1241 }
1243 > /**
1244 > * Returns a new {@link SessionSummaryMeta} with the `create_session` spawn depth
1245 > * set to `depth`, preserving any other keys in the bag.
1246 > */
1247 > export function withSessionSpawnDepth(meta: SessionSummaryMeta | undefined, depth: number): SessionSummaryMeta {
1248 return { ...meta, [SESSION_META_SPAWN_DEPTH_KEY]: depth };
1249 }
1251 > /**
1252 > * Reserved key under {@link SessionSummaryMeta} marking a session as
1253 > * workspace-less: a session with no workspace/folder binding (surfaced in the
1254 > * UI as a "Quick Chat"). Carried on the summary bag (not the full state) so
1255 > * clients can group/style such sessions in session lists without subscribing to
1256 > * full session state. VS Code-specific convention layered on the protocol's
1257 > * generic `_meta` bag.
1258 > */
1259 > export const SESSION_META_WORKSPACELESS_KEY = 'workspaceless';
1260 >
1261 > /**
1262 > * Session-database metadata key recording whether a session is workspace-less (a
1263 > * workspace-less chat). Owned by the AH service: `AgentService` writes it centrally at
1264 > * create/materialize and overlays it onto every agent's summary `_meta` in
1265 > * `listSessions`; agents only read it (e.g. to pick the workspace-less system prompt
1266 > * on resume) and never persist it themselves.
1267 > */
1268 > export const AH_META_WORKSPACELESS_DB_KEY = 'agentHost.workspaceless';
1269 >
1270 > /**
1271 > * Session-database metadata key recording whether a session is archived. Written by
1272 > * the AH orchestrator (`AgentSideEffects` on `SessionIsArchivedChanged`) and read by
1273 > * both the orchestrator (`AgentService` restore/list) and agents (e.g. `CopilotAgent`
1274 > * decides whether to recreate a missing worktree vs. resume read-only for history).
1275 > * {@link AH_META_IS_DONE_DB_KEY} is the legacy name kept for sessions persisted before
1276 > * the rename; readers fall back to it when {@link AH_META_IS_ARCHIVED_DB_KEY} is absent.
1277 > */
1278 > export const AH_META_IS_ARCHIVED_DB_KEY = 'isArchived';
1279 >
1280 > /** Legacy metadata key for the archived flag; see {@link AH_META_IS_ARCHIVED_DB_KEY}. */
1281 > export const AH_META_IS_DONE_DB_KEY = 'isDone';
1282 >
1283 > /**
1284 > * Reads the workspace-less marker from {@link SessionSummaryMeta}. Returns
1285 > * `true` only when the well-known key is present and set to boolean `true`.
1286 > */
1287 > export function readSessionWorkspaceless(meta: SessionSummaryMeta | undefined): boolean {
1288 return meta?.[SESSION_META_WORKSPACELESS_KEY] === true;
1289 }
1291 > /**
1292 > * Returns a new {@link SessionSummaryMeta} with the workspace-less marker set,
1293 > * or with the slot removed when `workspaceless` is `false`. Returns `undefined`
1294 > * if the result would be empty.
1295 > */
1296 > export function withSessionWorkspaceless(meta: SessionSummaryMeta | undefined, workspaceless: boolean): SessionSummaryMeta | undefined {
1297 const next: { [key: string]: unknown } = { ...meta };
1298 if (workspaceless) {
1303 return Object.keys(next).length > 0 ? next : undefined;
1304 }
1306 > // ---- RootState _meta accessors ---------------------------------------------
1307 >
1308 > /**
1309 > * VS Code-side alias for the protocol's open `_meta` property bag on
1310 > * {@link RootState}. Keys SHOULD be namespaced to avoid collisions; values MUST
1311 > * be JSON-serializable.
1312 > */
1313 > export type RootMeta = Record<string, unknown>;
1314 >
1315 > /**
1316 > * Reserved key under {@link RootMeta} for the well-known host-build payload.
1317 > * Value at this key, when present, MUST be shaped like {@link IHostBuildInfo}.
1318 > * This is a VS Code-specific convention layered on top of the protocol's
1319 > * generic `_meta` bag — the protocol itself does not know about build info.
1320 > */
1321 > export const ROOT_META_HOST_BUILD_KEY = 'hostBuild';
1322 >
1323 > /**
1324 > * Build information about the program hosting the agent host (the VS Code CLI),
1325 > * carried under {@link RootMeta} at {@link ROOT_META_HOST_BUILD_KEY}. Lets a
1326 > * client see which build is hosting it — useful when inspecting the output of a
1327 > * remote agent host.
1328 > *
1329 > * All fields except {@link version} are optional — a build that does not track
1330 > * a particular field should omit it.
1331 > */
1332 > export interface IHostBuildInfo {
1333 > /** Product version (e.g. `1.96.0`). */
1334 > readonly version: string;
1335 > /** Commit SHA of the build, if known. */
1336 > readonly commit?: string;
1337 > /** Build date (ISO 8601), if known. */
1338 > readonly date?: string;
1339 > /** Release quality (e.g. `stable`, `insider`), if known. */
1340 > readonly quality?: string;
1341 > }
1342 >
1343 > /**
1344 > * Derives {@link IHostBuildInfo} from the host's {@link IProductService}.
1345 > */
1346 > export function hostBuildInfoFromProduct(productService: IProductService): IHostBuildInfo {
1347 return {
1348 version: productService.version,
1352 };
1353 }
1355 > /**
1356 > * Reads the well-known host-build payload from {@link RootMeta}, if present.
1357 > * Returns `undefined` when the meta bag is absent or the value at the host-build
1358 > * key is not a plain object with a string `version`. Optional fields with wrong
1359 > * types are silently dropped.
1360 > */
1361 > export function readHostBuildInfo(state: RootState | undefined): IHostBuildInfo | undefined {
1362 const meta = state?._meta;
1363 const value = meta?.[ROOT_META_HOST_BUILD_KEY];
1377 return result;
1378 }
1380 > /**
1381 > * Returns a new {@link RootMeta} with the host-build payload set to
1382 > * `buildInfo`, or with the slot removed if `buildInfo` is `undefined`. Returns
1383 > * `undefined` if the result would be empty.
1384 > */
1385 > export function withHostBuildInfo(meta: RootMeta | undefined, buildInfo: IHostBuildInfo | undefined): RootMeta | undefined {
1386 const next: { [key: string]: unknown } = { ...meta };
1387 if (buildInfo !== undefined) {
1392 return Object.keys(next).length > 0 ? next : undefined;
1393 }
1395 > /**
1396 > * Formats {@link IHostBuildInfo} as a short single-line human-readable string,
1397 > * e.g. `1.96.0 (commit abc1234, 2024-01-02T03:04:05Z, insider)`.
1398 > */
1399 > export function formatHostBuildInfo(info: IHostBuildInfo): string {
1400 const details: string[] = [];
1401 if (info.commit) { details.push(`commit ${info.commit}`); }
src/vs/base/common/keyCodes.ts 792 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- keyCodes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * Virtual Key Codes, the value does not hold any inherent meaning.
8 > * Inspired somewhat from https://msdn.microsoft.com/en-us/library/windows/desktop/dd375731(v=vs.85).aspx
9 > * But these are "more general", as they should work across browsers & OS`s.
10 > */
11 > export const enum KeyCode {
12 > DependsOnKbLayout = -1,
13 >
14 > /**
15 > * Placed first to cover the 0 value of the enum.
16 > */
17 > Unknown = 0,
18 >
19 > Backspace,
20 > Tab,
21 > Enter,
22 > Shift,
23 > Ctrl,
24 > Alt,
25 > PauseBreak,
26 > CapsLock,
27 > Escape,
28 > Space,
29 > PageUp,
30 > PageDown,
31 > End,
32 > Home,
33 > LeftArrow,
34 > UpArrow,
35 > RightArrow,
36 > DownArrow,
37 > Insert,
38 > Delete,
39 >
40 > Digit0,
41 > Digit1,
42 > Digit2,
43 > Digit3,
44 > Digit4,
45 > Digit5,
46 > Digit6,
47 > Digit7,
48 > Digit8,
49 > Digit9,
50 >
51 > KeyA,
52 > KeyB,
53 > KeyC,
54 > KeyD,
55 > KeyE,
56 > KeyF,
57 > KeyG,
58 > KeyH,
59 > KeyI,
60 > KeyJ,
61 > KeyK,
62 > KeyL,
63 > KeyM,
64 > KeyN,
65 > KeyO,
66 > KeyP,
67 > KeyQ,
68 > KeyR,
69 > KeyS,
70 > KeyT,
71 > KeyU,
72 > KeyV,
73 > KeyW,
74 > KeyX,
75 > KeyY,
76 > KeyZ,
77 >
78 > Meta,
79 > ContextMenu,
80 >
81 > F1,
82 > F2,
83 > F3,
84 > F4,
85 > F5,
86 > F6,
87 > F7,
88 > F8,
89 > F9,
90 > F10,
91 > F11,
92 > F12,
93 > F13,
94 > F14,
95 > F15,
96 > F16,
97 > F17,
98 > F18,
99 > F19,
100 > F20,
101 > F21,
102 > F22,
103 > F23,
104 > F24,
105 >
106 > NumLock,
107 > ScrollLock,
108 >
109 > /**
110 > * Used for miscellaneous characters; it can vary by keyboard.
111 > * For the US standard keyboard, the ';:' key
112 > */
113 > Semicolon,
114 > /**
115 > * For any country/region, the '+' key
116 > * For the US standard keyboard, the '=+' key
117 > */
118 > Equal,
119 > /**
120 > * For any country/region, the ',' key
121 > * For the US standard keyboard, the ',<' key
122 > */
123 > Comma,
124 > /**
125 > * For any country/region, the '-' key
126 > * For the US standard keyboard, the '-_' key
127 > */
128 > Minus,
129 > /**
130 > * For any country/region, the '.' key
131 > * For the US standard keyboard, the '.>' key
132 > */
133 > Period,
134 > /**
135 > * Used for miscellaneous characters; it can vary by keyboard.
136 > * For the US standard keyboard, the '/?' key
137 > */
138 > Slash,
139 > /**
140 > * Used for miscellaneous characters; it can vary by keyboard.
141 > * For the US standard keyboard, the '`~' key
142 > */
143 > Backquote,
144 > /**
145 > * Used for miscellaneous characters; it can vary by keyboard.
146 > * For the US standard keyboard, the '[{' key
147 > */
148 > BracketLeft,
149 > /**
150 > * Used for miscellaneous characters; it can vary by keyboard.
151 > * For the US standard keyboard, the '\|' key
152 > */
153 > Backslash,
154 > /**
155 > * Used for miscellaneous characters; it can vary by keyboard.
156 > * For the US standard keyboard, the ']}' key
157 > */
158 > BracketRight,
159 > /**
160 > * Used for miscellaneous characters; it can vary by keyboard.
161 > * For the US standard keyboard, the ''"' key
162 > */
163 > Quote,
164 > /**
165 > * Used for miscellaneous characters; it can vary by keyboard.
166 > */
167 > OEM_8,
168 > /**
169 > * Either the angle bracket key or the backslash key on the RT 102-key keyboard.
170 > */
171 > IntlBackslash,
172 >
173 > Numpad0, // VK_NUMPAD0, 0x60, Numeric keypad 0 key
174 > Numpad1, // VK_NUMPAD1, 0x61, Numeric keypad 1 key
175 > Numpad2, // VK_NUMPAD2, 0x62, Numeric keypad 2 key
176 > Numpad3, // VK_NUMPAD3, 0x63, Numeric keypad 3 key
177 > Numpad4, // VK_NUMPAD4, 0x64, Numeric keypad 4 key
178 > Numpad5, // VK_NUMPAD5, 0x65, Numeric keypad 5 key
179 > Numpad6, // VK_NUMPAD6, 0x66, Numeric keypad 6 key
180 > Numpad7, // VK_NUMPAD7, 0x67, Numeric keypad 7 key
181 > Numpad8, // VK_NUMPAD8, 0x68, Numeric keypad 8 key
182 > Numpad9, // VK_NUMPAD9, 0x69, Numeric keypad 9 key
183 >
184 > NumpadMultiply, // VK_MULTIPLY, 0x6A, Multiply key
185 > NumpadAdd, // VK_ADD, 0x6B, Add key
186 > NUMPAD_SEPARATOR, // VK_SEPARATOR, 0x6C, Separator key
187 > NumpadSubtract, // VK_SUBTRACT, 0x6D, Subtract key
188 > NumpadDecimal, // VK_DECIMAL, 0x6E, Decimal key
189 > NumpadDivide, // VK_DIVIDE, 0x6F,
190 >
191 > /**
192 > * Cover all key codes when IME is processing input.
193 > */
194 > KEY_IN_COMPOSITION,
195 >
196 > ABNT_C1, // Brazilian (ABNT) Keyboard
197 > ABNT_C2, // Brazilian (ABNT) Keyboard
198 >
199 > AudioVolumeMute,
200 > AudioVolumeUp,
201 > AudioVolumeDown,
202 >
203 > BrowserSearch,
204 > BrowserHome,
205 > BrowserBack,
206 > BrowserForward,
207 >
208 > MediaTrackNext,
209 > MediaTrackPrevious,
210 > MediaStop,
211 > MediaPlayPause,
212 > LaunchMediaPlayer,
213 > LaunchMail,
214 > LaunchApp2,
215 >
216 > /**
217 > * VK_CLEAR, 0x0C, CLEAR key
218 > */
219 > Clear,
220 >
221 > /**
222 > * Placed last to cover the length of the enum.
223 > * Please do not depend on this value!
224 > */
225 > MAX_VALUE
226 > }
227 >
228 > /**
229 > * keyboardEvent.code
230 > */
231 > export const enum ScanCode {
232 > DependsOnKbLayout = -1,
233 > None,
234 > Hyper,
235 > Super,
236 > Fn,
237 > FnLock,
238 > Suspend,
239 > Resume,
240 > Turbo,
241 > Sleep,
242 > WakeUp,
243 > KeyA,
244 > KeyB,
245 > KeyC,
246 > KeyD,
247 > KeyE,
248 > KeyF,
249 > KeyG,
250 > KeyH,
251 > KeyI,
252 > KeyJ,
253 > KeyK,
254 > KeyL,
255 > KeyM,
256 > KeyN,
257 > KeyO,
258 > KeyP,
259 > KeyQ,
260 > KeyR,
261 > KeyS,
262 > KeyT,
263 > KeyU,
264 > KeyV,
265 > KeyW,
266 > KeyX,
267 > KeyY,
268 > KeyZ,
269 > Digit1,
270 > Digit2,
271 > Digit3,
272 > Digit4,
273 > Digit5,
274 > Digit6,
275 > Digit7,
276 > Digit8,
277 > Digit9,
278 > Digit0,
279 > Enter,
280 > Escape,
281 > Backspace,
282 > Tab,
283 > Space,
284 > Minus,
285 > Equal,
286 > BracketLeft,
287 > BracketRight,
288 > Backslash,
289 > IntlHash,
290 > Semicolon,
291 > Quote,
292 > Backquote,
293 > Comma,
294 > Period,
295 > Slash,
296 > CapsLock,
297 > F1,
298 > F2,
299 > F3,
300 > F4,
301 > F5,
302 > F6,
303 > F7,
304 > F8,
305 > F9,
306 > F10,
307 > F11,
308 > F12,
309 > PrintScreen,
310 > ScrollLock,
311 > Pause,
312 > Insert,
313 > Home,
314 > PageUp,
315 > Delete,
316 > End,
317 > PageDown,
318 > ArrowRight,
319 > ArrowLeft,
320 > ArrowDown,
321 > ArrowUp,
322 > NumLock,
323 > NumpadDivide,
324 > NumpadMultiply,
325 > NumpadSubtract,
326 > NumpadAdd,
327 > NumpadEnter,
328 > Numpad1,
329 > Numpad2,
330 > Numpad3,
331 > Numpad4,
332 > Numpad5,
333 > Numpad6,
334 > Numpad7,
335 > Numpad8,
336 > Numpad9,
337 > Numpad0,
338 > NumpadDecimal,
339 > IntlBackslash,
340 > ContextMenu,
341 > Power,
342 > NumpadEqual,
343 > F13,
344 > F14,
345 > F15,
346 > F16,
347 > F17,
348 > F18,
349 > F19,
350 > F20,
351 > F21,
352 > F22,
353 > F23,
354 > F24,
355 > Open,
356 > Help,
357 > Select,
358 > Again,
359 > Undo,
360 > Cut,
361 > Copy,
362 > Paste,
363 > Find,
364 > AudioVolumeMute,
365 > AudioVolumeUp,
366 > AudioVolumeDown,
367 > NumpadComma,
368 > IntlRo,
369 > KanaMode,
370 > IntlYen,
371 > Convert,
372 > NonConvert,
373 > Lang1,
374 > Lang2,
375 > Lang3,
376 > Lang4,
377 > Lang5,
378 > Abort,
379 > Props,
380 > NumpadParenLeft,
381 > NumpadParenRight,
382 > NumpadBackspace,
383 > NumpadMemoryStore,
384 > NumpadMemoryRecall,
385 > NumpadMemoryClear,
386 > NumpadMemoryAdd,
387 > NumpadMemorySubtract,
388 > NumpadClear,
389 > NumpadClearEntry,
390 > ControlLeft,
391 > ShiftLeft,
392 > AltLeft,
393 > MetaLeft,
394 > ControlRight,
395 > ShiftRight,
396 > AltRight,
397 > MetaRight,
398 > BrightnessUp,
399 > BrightnessDown,
400 > MediaPlay,
401 > MediaRecord,
402 > MediaFastForward,
403 > MediaRewind,
404 > MediaTrackNext,
405 > MediaTrackPrevious,
406 > MediaStop,
407 > Eject,
408 > MediaPlayPause,
409 > MediaSelect,
410 > LaunchMail,
411 > LaunchApp2,
412 > LaunchApp1,
413 > SelectTask,
414 > LaunchScreenSaver,
415 > BrowserSearch,
416 > BrowserHome,
417 > BrowserBack,
418 > BrowserForward,
419 > BrowserStop,
420 > BrowserRefresh,
421 > BrowserFavorites,
422 > ZoomToggle,
423 > MailReply,
424 > MailForward,
425 > MailSend,
426 >
427 > MAX_VALUE
428 > }
429 >
430 > class KeyCodeStrMap {
431 >
432 > public _keyCodeToStr: string[];
433 > public _strToKeyCode: { [str: string]: KeyCode };
434 >
435 > constructor() {
436 > this._keyCodeToStr = [];
437 > this._strToKeyCode = Object.create(null);
438 > }
439 >
440 > define(keyCode: KeyCode, str: string): void {
441 > this._keyCodeToStr[keyCode] = str;
442 > this._strToKeyCode[str.toLowerCase()] = keyCode;
443 > }
444 >
445 > keyCodeToStr(keyCode: KeyCode): string {
446 return this._keyCodeToStr[keyCode];
447 }
448 > keyCodes.ts
449 > strToKeyCode(str: string): KeyCode {
450 return this._strToKeyCode[str.toLowerCase()] || KeyCode.Unknown;
451 }
452 > } keyCodes.ts
453 >
454 > const uiMap = new KeyCodeStrMap();
455 > const userSettingsUSMap = new KeyCodeStrMap();
456 > const userSettingsGeneralMap = new KeyCodeStrMap();
457 > export const EVENT_KEY_CODE_MAP: { [keyCode: number]: KeyCode } = new Array(230);
458 > export const NATIVE_WINDOWS_KEY_CODE_TO_KEY_CODE: { [nativeKeyCode: string]: KeyCode } = {};
459 > export const SCAN_CODE_STR_TO_EVENT_KEY_CODE: { [scanCodeStr: string]: number } = {};
460 > const scanCodeIntToStr: string[] = [];
461 > const scanCodeStrToInt: { [code: string]: number } = Object.create(null);
462 > const scanCodeLowerCaseStrToInt: { [code: string]: number } = Object.create(null);
463 >
464 > export const ScanCodeUtils = {
465 > lowerCaseToEnum: (scanCode: string) => scanCodeLowerCaseStrToInt[scanCode] || ScanCode.None,
466 > toEnum: (scanCode: string) => scanCodeStrToInt[scanCode] || ScanCode.None,
467 > toString: (scanCode: ScanCode) => scanCodeIntToStr[scanCode] || 'None'
468 > };
469 >
470 > /**
471 > * -1 if a ScanCode => KeyCode mapping depends on kb layout.
472 > */
473 > export const IMMUTABLE_CODE_TO_KEY_CODE: KeyCode[] = [];
474 >
475 > /**
476 > * -1 if a KeyCode => ScanCode mapping depends on kb layout.
477 > */
478 > export const IMMUTABLE_KEY_CODE_TO_CODE: ScanCode[] = [];
479 >
480 > for (let i = 0; i <= ScanCode.MAX_VALUE; i++) {
481 > IMMUTABLE_CODE_TO_KEY_CODE[i] = KeyCode.DependsOnKbLayout;
482 > }
483 >
484 > for (let i = 0; i <= KeyCode.MAX_VALUE; i++) {
485 > IMMUTABLE_KEY_CODE_TO_CODE[i] = ScanCode.DependsOnKbLayout;
486 > }
487 >
488 > (function () {
489 >
490 > // See https://msdn.microsoft.com/en-us/library/windows/desktop/dd375731(v=vs.85).aspx
491 > // See https://github.com/microsoft/node-native-keymap/blob/88c0b0e5/deps/chromium/keyboard_codes_win.h
492 >
493 > const empty = '';
494 > type IMappingEntry = [0 | 1, ScanCode, string, KeyCode, string, number, string, string, string];
495 > const mappings: IMappingEntry[] = [
496 > // immutable, scanCode, scanCodeStr, keyCode, keyCodeStr, eventKeyCode, vkey, usUserSettingsLabel, generalUserSettingsLabel
497 > [1, ScanCode.None, 'None', KeyCode.Unknown, 'unknown', 0, 'VK_UNKNOWN', empty, empty],
498 > [1, ScanCode.Hyper, 'Hyper', KeyCode.Unknown, empty, 0, empty, empty, empty],
499 > [1, ScanCode.Super, 'Super', KeyCode.Unknown, empty, 0, empty, empty, empty],
500 > [1, ScanCode.Fn, 'Fn', KeyCode.Unknown, empty, 0, empty, empty, empty],
501 > [1, ScanCode.FnLock, 'FnLock', KeyCode.Unknown, empty, 0, empty, empty, empty],
502 > [1, ScanCode.Suspend, 'Suspend', KeyCode.Unknown, empty, 0, empty, empty, empty],
503 > [1, ScanCode.Resume, 'Resume', KeyCode.Unknown, empty, 0, empty, empty, empty],
504 > [1, ScanCode.Turbo, 'Turbo', KeyCode.Unknown, empty, 0, empty, empty, empty],
505 > [1, ScanCode.Sleep, 'Sleep', KeyCode.Unknown, empty, 0, 'VK_SLEEP', empty, empty],
506 > [1, ScanCode.WakeUp, 'WakeUp', KeyCode.Unknown, empty, 0, empty, empty, empty],
507 > [0, ScanCode.KeyA, 'KeyA', KeyCode.KeyA, 'A', 65, 'VK_A', empty, empty],
508 > [0, ScanCode.KeyB, 'KeyB', KeyCode.KeyB, 'B', 66, 'VK_B', empty, empty],
509 > [0, ScanCode.KeyC, 'KeyC', KeyCode.KeyC, 'C', 67, 'VK_C', empty, empty],
510 > [0, ScanCode.KeyD, 'KeyD', KeyCode.KeyD, 'D', 68, 'VK_D', empty, empty],
511 > [0, ScanCode.KeyE, 'KeyE', KeyCode.KeyE, 'E', 69, 'VK_E', empty, empty],
512 > [0, ScanCode.KeyF, 'KeyF', KeyCode.KeyF, 'F', 70, 'VK_F', empty, empty],
513 > [0, ScanCode.KeyG, 'KeyG', KeyCode.KeyG, 'G', 71, 'VK_G', empty, empty],
514 > [0, ScanCode.KeyH, 'KeyH', KeyCode.KeyH, 'H', 72, 'VK_H', empty, empty],
515 > [0, ScanCode.KeyI, 'KeyI', KeyCode.KeyI, 'I', 73, 'VK_I', empty, empty],
516 > [0, ScanCode.KeyJ, 'KeyJ', KeyCode.KeyJ, 'J', 74, 'VK_J', empty, empty],
517 > [0, ScanCode.KeyK, 'KeyK', KeyCode.KeyK, 'K', 75, 'VK_K', empty, empty],
518 > [0, ScanCode.KeyL, 'KeyL', KeyCode.KeyL, 'L', 76, 'VK_L', empty, empty],
519 > [0, ScanCode.KeyM, 'KeyM', KeyCode.KeyM, 'M', 77, 'VK_M', empty, empty],
520 > [0, ScanCode.KeyN, 'KeyN', KeyCode.KeyN, 'N', 78, 'VK_N', empty, empty],
521 > [0, ScanCode.KeyO, 'KeyO', KeyCode.KeyO, 'O', 79, 'VK_O', empty, empty],
522 > [0, ScanCode.KeyP, 'KeyP', KeyCode.KeyP, 'P', 80, 'VK_P', empty, empty],
523 > [0, ScanCode.KeyQ, 'KeyQ', KeyCode.KeyQ, 'Q', 81, 'VK_Q', empty, empty],
524 > [0, ScanCode.KeyR, 'KeyR', KeyCode.KeyR, 'R', 82, 'VK_R', empty, empty],
525 > [0, ScanCode.KeyS, 'KeyS', KeyCode.KeyS, 'S', 83, 'VK_S', empty, empty],
526 > [0, ScanCode.KeyT, 'KeyT', KeyCode.KeyT, 'T', 84, 'VK_T', empty, empty],
527 > [0, ScanCode.KeyU, 'KeyU', KeyCode.KeyU, 'U', 85, 'VK_U', empty, empty],
528 > [0, ScanCode.KeyV, 'KeyV', KeyCode.KeyV, 'V', 86, 'VK_V', empty, empty],
529 > [0, ScanCode.KeyW, 'KeyW', KeyCode.KeyW, 'W', 87, 'VK_W', empty, empty],
530 > [0, ScanCode.KeyX, 'KeyX', KeyCode.KeyX, 'X', 88, 'VK_X', empty, empty],
531 > [0, ScanCode.KeyY, 'KeyY', KeyCode.KeyY, 'Y', 89, 'VK_Y', empty, empty],
532 > [0, ScanCode.KeyZ, 'KeyZ', KeyCode.KeyZ, 'Z', 90, 'VK_Z', empty, empty],
533 > [0, ScanCode.Digit1, 'Digit1', KeyCode.Digit1, '1', 49, 'VK_1', empty, empty],
534 > [0, ScanCode.Digit2, 'Digit2', KeyCode.Digit2, '2', 50, 'VK_2', empty, empty],
535 > [0, ScanCode.Digit3, 'Digit3', KeyCode.Digit3, '3', 51, 'VK_3', empty, empty],
536 > [0, ScanCode.Digit4, 'Digit4', KeyCode.Digit4, '4', 52, 'VK_4', empty, empty],
537 > [0, ScanCode.Digit5, 'Digit5', KeyCode.Digit5, '5', 53, 'VK_5', empty, empty],
538 > [0, ScanCode.Digit6, 'Digit6', KeyCode.Digit6, '6', 54, 'VK_6', empty, empty],
539 > [0, ScanCode.Digit7, 'Digit7', KeyCode.Digit7, '7', 55, 'VK_7', empty, empty],
540 > [0, ScanCode.Digit8, 'Digit8', KeyCode.Digit8, '8', 56, 'VK_8', empty, empty],
541 > [0, ScanCode.Digit9, 'Digit9', KeyCode.Digit9, '9', 57, 'VK_9', empty, empty],
542 > [0, ScanCode.Digit0, 'Digit0', KeyCode.Digit0, '0', 48, 'VK_0', empty, empty],
543 > [1, ScanCode.Enter, 'Enter', KeyCode.Enter, 'Enter', 13, 'VK_RETURN', empty, empty],
544 > [1, ScanCode.Escape, 'Escape', KeyCode.Escape, 'Escape', 27, 'VK_ESCAPE', empty, empty],
545 > [1, ScanCode.Backspace, 'Backspace', KeyCode.Backspace, 'Backspace', 8, 'VK_BACK', empty, empty],
546 > [1, ScanCode.Tab, 'Tab', KeyCode.Tab, 'Tab', 9, 'VK_TAB', empty, empty],
547 > [1, ScanCode.Space, 'Space', KeyCode.Space, 'Space', 32, 'VK_SPACE', empty, empty],
548 > [0, ScanCode.Minus, 'Minus', KeyCode.Minus, '-', 189, 'VK_OEM_MINUS', '-', 'OEM_MINUS'],
549 > [0, ScanCode.Equal, 'Equal', KeyCode.Equal, '=', 187, 'VK_OEM_PLUS', '=', 'OEM_PLUS'],
550 > [0, ScanCode.BracketLeft, 'BracketLeft', KeyCode.BracketLeft, '[', 219, 'VK_OEM_4', '[', 'OEM_4'],
551 > [0, ScanCode.BracketRight, 'BracketRight', KeyCode.BracketRight, ']', 221, 'VK_OEM_6', ']', 'OEM_6'],
552 > [0, ScanCode.Backslash, 'Backslash', KeyCode.Backslash, '\\', 220, 'VK_OEM_5', '\\', 'OEM_5'],
553 > [0, ScanCode.IntlHash, 'IntlHash', KeyCode.Unknown, empty, 0, empty, empty, empty], // has been dropped from the w3c spec
554 > [0, ScanCode.Semicolon, 'Semicolon', KeyCode.Semicolon, ';', 186, 'VK_OEM_1', ';', 'OEM_1'],
555 > [0, ScanCode.Quote, 'Quote', KeyCode.Quote, '\'', 222, 'VK_OEM_7', '\'', 'OEM_7'],
556 > [0, ScanCode.Backquote, 'Backquote', KeyCode.Backquote, '`', 192, 'VK_OEM_3', '`', 'OEM_3'],
557 > [0, ScanCode.Comma, 'Comma', KeyCode.Comma, ',', 188, 'VK_OEM_COMMA', ',', 'OEM_COMMA'],
558 > [0, ScanCode.Period, 'Period', KeyCode.Period, '.', 190, 'VK_OEM_PERIOD', '.', 'OEM_PERIOD'],
559 > [0, ScanCode.Slash, 'Slash', KeyCode.Slash, '/', 191, 'VK_OEM_2', '/', 'OEM_2'],
560 > [1, ScanCode.CapsLock, 'CapsLock', KeyCode.CapsLock, 'CapsLock', 20, 'VK_CAPITAL', empty, empty],
561 > [1, ScanCode.F1, 'F1', KeyCode.F1, 'F1', 112, 'VK_F1', empty, empty],
562 > [1, ScanCode.F2, 'F2', KeyCode.F2, 'F2', 113, 'VK_F2', empty, empty],
563 > [1, ScanCode.F3, 'F3', KeyCode.F3, 'F3', 114, 'VK_F3', empty, empty],
564 > [1, ScanCode.F4, 'F4', KeyCode.F4, 'F4', 115, 'VK_F4', empty, empty],
565 > [1, ScanCode.F5, 'F5', KeyCode.F5, 'F5', 116, 'VK_F5', empty, empty],
566 > [1, ScanCode.F6, 'F6', KeyCode.F6, 'F6', 117, 'VK_F6', empty, empty],
567 > [1, ScanCode.F7, 'F7', KeyCode.F7, 'F7', 118, 'VK_F7', empty, empty],
568 > [1, ScanCode.F8, 'F8', KeyCode.F8, 'F8', 119, 'VK_F8', empty, empty],
569 > [1, ScanCode.F9, 'F9', KeyCode.F9, 'F9', 120, 'VK_F9', empty, empty],
570 > [1, ScanCode.F10, 'F10', KeyCode.F10, 'F10', 121, 'VK_F10', empty, empty],
571 > [1, ScanCode.F11, 'F11', KeyCode.F11, 'F11', 122, 'VK_F11', empty, empty],
572 > [1, ScanCode.F12, 'F12', KeyCode.F12, 'F12', 123, 'VK_F12', empty, empty],
573 > [1, ScanCode.PrintScreen, 'PrintScreen', KeyCode.Unknown, empty, 0, empty, empty, empty],
574 > [1, ScanCode.ScrollLock, 'ScrollLock', KeyCode.ScrollLock, 'ScrollLock', 145, 'VK_SCROLL', empty, empty],
575 > [1, ScanCode.Pause, 'Pause', KeyCode.PauseBreak, 'PauseBreak', 19, 'VK_PAUSE', empty, empty],
576 > [1, ScanCode.Insert, 'Insert', KeyCode.Insert, 'Insert', 45, 'VK_INSERT', empty, empty],
577 > [1, ScanCode.Home, 'Home', KeyCode.Home, 'Home', 36, 'VK_HOME', empty, empty],
578 > [1, ScanCode.PageUp, 'PageUp', KeyCode.PageUp, 'PageUp', 33, 'VK_PRIOR', empty, empty],
579 > [1, ScanCode.Delete, 'Delete', KeyCode.Delete, 'Del', 46, 'VK_DELETE', 'Delete', empty],
580 > [1, ScanCode.End, 'End', KeyCode.End, 'End', 35, 'VK_END', empty, empty],
581 > [1, ScanCode.PageDown, 'PageDown', KeyCode.PageDown, 'PageDown', 34, 'VK_NEXT', empty, empty],
582 > [1, ScanCode.ArrowRight, 'ArrowRight', KeyCode.RightArrow, 'RightArrow', 39, 'VK_RIGHT', 'Right', empty],
583 > [1, ScanCode.ArrowLeft, 'ArrowLeft', KeyCode.LeftArrow, 'LeftArrow', 37, 'VK_LEFT', 'Left', empty],
584 > [1, ScanCode.ArrowDown, 'ArrowDown', KeyCode.DownArrow, 'DownArrow', 40, 'VK_DOWN', 'Down', empty],
585 > [1, ScanCode.ArrowUp, 'ArrowUp', KeyCode.UpArrow, 'UpArrow', 38, 'VK_UP', 'Up', empty],
586 > [1, ScanCode.NumLock, 'NumLock', KeyCode.NumLock, 'NumLock', 144, 'VK_NUMLOCK', empty, empty],
587 > [1, ScanCode.NumpadDivide, 'NumpadDivide', KeyCode.NumpadDivide, 'NumPad_Divide', 111, 'VK_DIVIDE', empty, empty],
588 > [1, ScanCode.NumpadMultiply, 'NumpadMultiply', KeyCode.NumpadMultiply, 'NumPad_Multiply', 106, 'VK_MULTIPLY', empty, empty],
589 > [1, ScanCode.NumpadSubtract, 'NumpadSubtract', KeyCode.NumpadSubtract, 'NumPad_Subtract', 109, 'VK_SUBTRACT', empty, empty],
590 > [1, ScanCode.NumpadAdd, 'NumpadAdd', KeyCode.NumpadAdd, 'NumPad_Add', 107, 'VK_ADD', empty, empty],
591 > [1, ScanCode.NumpadEnter, 'NumpadEnter', KeyCode.Enter, empty, 0, empty, empty, empty],
592 > [1, ScanCode.Numpad1, 'Numpad1', KeyCode.Numpad1, 'NumPad1', 97, 'VK_NUMPAD1', empty, empty],
593 > [1, ScanCode.Numpad2, 'Numpad2', KeyCode.Numpad2, 'NumPad2', 98, 'VK_NUMPAD2', empty, empty],
594 > [1, ScanCode.Numpad3, 'Numpad3', KeyCode.Numpad3, 'NumPad3', 99, 'VK_NUMPAD3', empty, empty],
595 > [1, ScanCode.Numpad4, 'Numpad4', KeyCode.Numpad4, 'NumPad4', 100, 'VK_NUMPAD4', empty, empty],
596 > [1, ScanCode.Numpad5, 'Numpad5', KeyCode.Numpad5, 'NumPad5', 101, 'VK_NUMPAD5', empty, empty],
597 > [1, ScanCode.Numpad6, 'Numpad6', KeyCode.Numpad6, 'NumPad6', 102, 'VK_NUMPAD6', empty, empty],
598 > [1, ScanCode.Numpad7, 'Numpad7', KeyCode.Numpad7, 'NumPad7', 103, 'VK_NUMPAD7', empty, empty],
599 > [1, ScanCode.Numpad8, 'Numpad8', KeyCode.Numpad8, 'NumPad8', 104, 'VK_NUMPAD8', empty, empty],
600 > [1, ScanCode.Numpad9, 'Numpad9', KeyCode.Numpad9, 'NumPad9', 105, 'VK_NUMPAD9', empty, empty],
601 > [1, ScanCode.Numpad0, 'Numpad0', KeyCode.Numpad0, 'NumPad0', 96, 'VK_NUMPAD0', empty, empty],
602 > [1, ScanCode.NumpadDecimal, 'NumpadDecimal', KeyCode.NumpadDecimal, 'NumPad_Decimal', 110, 'VK_DECIMAL', empty, empty],
603 > [0, ScanCode.IntlBackslash, 'IntlBackslash', KeyCode.IntlBackslash, 'OEM_102', 226, 'VK_OEM_102', empty, empty],
604 > [1, ScanCode.ContextMenu, 'ContextMenu', KeyCode.ContextMenu, 'ContextMenu', 93, empty, empty, empty],
605 > [1, ScanCode.Power, 'Power', KeyCode.Unknown, empty, 0, empty, empty, empty],
606 > [1, ScanCode.NumpadEqual, 'NumpadEqual', KeyCode.Unknown, empty, 0, empty, empty, empty],
607 > [1, ScanCode.F13, 'F13', KeyCode.F13, 'F13', 124, 'VK_F13', empty, empty],
608 > [1, ScanCode.F14, 'F14', KeyCode.F14, 'F14', 125, 'VK_F14', empty, empty],
609 > [1, ScanCode.F15, 'F15', KeyCode.F15, 'F15', 126, 'VK_F15', empty, empty],
610 > [1, ScanCode.F16, 'F16', KeyCode.F16, 'F16', 127, 'VK_F16', empty, empty],
611 > [1, ScanCode.F17, 'F17', KeyCode.F17, 'F17', 128, 'VK_F17', empty, empty],
612 > [1, ScanCode.F18, 'F18', KeyCode.F18, 'F18', 129, 'VK_F18', empty, empty],
613 > [1, ScanCode.F19, 'F19', KeyCode.F19, 'F19', 130, 'VK_F19', empty, empty],
614 > [1, ScanCode.F20, 'F20', KeyCode.F20, 'F20', 131, 'VK_F20', empty, empty],
615 > [1, ScanCode.F21, 'F21', KeyCode.F21, 'F21', 132, 'VK_F21', empty, empty],
616 > [1, ScanCode.F22, 'F22', KeyCode.F22, 'F22', 133, 'VK_F22', empty, empty],
617 > [1, ScanCode.F23, 'F23', KeyCode.F23, 'F23', 134, 'VK_F23', empty, empty],
618 > [1, ScanCode.F24, 'F24', KeyCode.F24, 'F24', 135, 'VK_F24', empty, empty],
619 > [1, ScanCode.Open, 'Open', KeyCode.Unknown, empty, 0, empty, empty, empty],
620 > [1, ScanCode.Help, 'Help', KeyCode.Unknown, empty, 0, empty, empty, empty],
621 > [1, ScanCode.Select, 'Select', KeyCode.Unknown, empty, 0, empty, empty, empty],
622 > [1, ScanCode.Again, 'Again', KeyCode.Unknown, empty, 0, empty, empty, empty],
623 > [1, ScanCode.Undo, 'Undo', KeyCode.Unknown, empty, 0, empty, empty, empty],
624 > [1, ScanCode.Cut, 'Cut', KeyCode.Unknown, empty, 0, empty, empty, empty],
625 > [1, ScanCode.Copy, 'Copy', KeyCode.Unknown, empty, 0, empty, empty, empty],
626 > [1, ScanCode.Paste, 'Paste', KeyCode.Unknown, empty, 0, empty, empty, empty],
627 > [1, ScanCode.Find, 'Find', KeyCode.Unknown, empty, 0, empty, empty, empty],
628 > [1, ScanCode.AudioVolumeMute, 'AudioVolumeMute', KeyCode.AudioVolumeMute, 'AudioVolumeMute', 173, 'VK_VOLUME_MUTE', empty, empty],
629 > [1, ScanCode.AudioVolumeUp, 'AudioVolumeUp', KeyCode.AudioVolumeUp, 'AudioVolumeUp', 175, 'VK_VOLUME_UP', empty, empty],
630 > [1, ScanCode.AudioVolumeDown, 'AudioVolumeDown', KeyCode.AudioVolumeDown, 'AudioVolumeDown', 174, 'VK_VOLUME_DOWN', empty, empty],
631 > [1, ScanCode.NumpadComma, 'NumpadComma', KeyCode.NUMPAD_SEPARATOR, 'NumPad_Separator', 108, 'VK_SEPARATOR', empty, empty],
632 > [0, ScanCode.IntlRo, 'IntlRo', KeyCode.ABNT_C1, 'ABNT_C1', 193, 'VK_ABNT_C1', empty, empty],
633 > [1, ScanCode.KanaMode, 'KanaMode', KeyCode.Unknown, empty, 0, empty, empty, empty],
634 > [0, ScanCode.IntlYen, 'IntlYen', KeyCode.Unknown, empty, 0, empty, empty, empty],
635 > [1, ScanCode.Convert, 'Convert', KeyCode.Unknown, empty, 0, empty, empty, empty],
636 > [1, ScanCode.NonConvert, 'NonConvert', KeyCode.Unknown, empty, 0, empty, empty, empty],
637 > [1, ScanCode.Lang1, 'Lang1', KeyCode.Unknown, empty, 0, empty, empty, empty],
638 > [1, ScanCode.Lang2, 'Lang2', KeyCode.Unknown, empty, 0, empty, empty, empty],
639 > [1, ScanCode.Lang3, 'Lang3', KeyCode.Unknown, empty, 0, empty, empty, empty],
640 > [1, ScanCode.Lang4, 'Lang4', KeyCode.Unknown, empty, 0, empty, empty, empty],
641 > [1, ScanCode.Lang5, 'Lang5', KeyCode.Unknown, empty, 0, empty, empty, empty],
642 > [1, ScanCode.Abort, 'Abort', KeyCode.Unknown, empty, 0, empty, empty, empty],
643 > [1, ScanCode.Props, 'Props', KeyCode.Unknown, empty, 0, empty, empty, empty],
644 > [1, ScanCode.NumpadParenLeft, 'NumpadParenLeft', KeyCode.Unknown, empty, 0, empty, empty, empty],
645 > [1, ScanCode.NumpadParenRight, 'NumpadParenRight', KeyCode.Unknown, empty, 0, empty, empty, empty],
646 > [1, ScanCode.NumpadBackspace, 'NumpadBackspace', KeyCode.Unknown, empty, 0, empty, empty, empty],
647 > [1, ScanCode.NumpadMemoryStore, 'NumpadMemoryStore', KeyCode.Unknown, empty, 0, empty, empty, empty],
648 > [1, ScanCode.NumpadMemoryRecall, 'NumpadMemoryRecall', KeyCode.Unknown, empty, 0, empty, empty, empty],
649 > [1, ScanCode.NumpadMemoryClear, 'NumpadMemoryClear', KeyCode.Unknown, empty, 0, empty, empty, empty],
650 > [1, ScanCode.NumpadMemoryAdd, 'NumpadMemoryAdd', KeyCode.Unknown, empty, 0, empty, empty, empty],
651 > [1, ScanCode.NumpadMemorySubtract, 'NumpadMemorySubtract', KeyCode.Unknown, empty, 0, empty, empty, empty],
652 > [1, ScanCode.NumpadClear, 'NumpadClear', KeyCode.Clear, 'Clear', 12, 'VK_CLEAR', empty, empty],
653 > [1, ScanCode.NumpadClearEntry, 'NumpadClearEntry', KeyCode.Unknown, empty, 0, empty, empty, empty],
654 > [1, ScanCode.None, empty, KeyCode.Ctrl, 'Ctrl', 17, 'VK_CONTROL', empty, empty],
655 > [1, ScanCode.None, empty, KeyCode.Shift, 'Shift', 16, 'VK_SHIFT', empty, empty],
656 > [1, ScanCode.None, empty, KeyCode.Alt, 'Alt', 18, 'VK_MENU', empty, empty],
657 > [1, ScanCode.None, empty, KeyCode.Meta, 'Meta', 91, 'VK_COMMAND', empty, empty],
658 > [1, ScanCode.ControlLeft, 'ControlLeft', KeyCode.Ctrl, empty, 0, 'VK_LCONTROL', empty, empty],
659 > [1, ScanCode.ShiftLeft, 'ShiftLeft', KeyCode.Shift, empty, 0, 'VK_LSHIFT', empty, empty],
660 > [1, ScanCode.AltLeft, 'AltLeft', KeyCode.Alt, empty, 0, 'VK_LMENU', empty, empty],
661 > [1, ScanCode.MetaLeft, 'MetaLeft', KeyCode.Meta, empty, 0, 'VK_LWIN', empty, empty],
662 > [1, ScanCode.ControlRight, 'ControlRight', KeyCode.Ctrl, empty, 0, 'VK_RCONTROL', empty, empty],
663 > [1, ScanCode.ShiftRight, 'ShiftRight', KeyCode.Shift, empty, 0, 'VK_RSHIFT', empty, empty],
664 > [1, ScanCode.AltRight, 'AltRight', KeyCode.Alt, empty, 0, 'VK_RMENU', empty, empty],
665 > [1, ScanCode.MetaRight, 'MetaRight', KeyCode.Meta, empty, 0, 'VK_RWIN', empty, empty],
666 > [1, ScanCode.BrightnessUp, 'BrightnessUp', KeyCode.Unknown, empty, 0, empty, empty, empty],
667 > [1, ScanCode.BrightnessDown, 'BrightnessDown', KeyCode.Unknown, empty, 0, empty, empty, empty],
668 > [1, ScanCode.MediaPlay, 'MediaPlay', KeyCode.Unknown, empty, 0, empty, empty, empty],
669 > [1, ScanCode.MediaRecord, 'MediaRecord', KeyCode.Unknown, empty, 0, empty, empty, empty],
670 > [1, ScanCode.MediaFastForward, 'MediaFastForward', KeyCode.Unknown, empty, 0, empty, empty, empty],
671 > [1, ScanCode.MediaRewind, 'MediaRewind', KeyCode.Unknown, empty, 0, empty, empty, empty],
672 > [1, ScanCode.MediaTrackNext, 'MediaTrackNext', KeyCode.MediaTrackNext, 'MediaTrackNext', 176, 'VK_MEDIA_NEXT_TRACK', empty, empty],
673 > [1, ScanCode.MediaTrackPrevious, 'MediaTrackPrevious', KeyCode.MediaTrackPrevious, 'MediaTrackPrevious', 177, 'VK_MEDIA_PREV_TRACK', empty, empty],
674 > [1, ScanCode.MediaStop, 'MediaStop', KeyCode.MediaStop, 'MediaStop', 178, 'VK_MEDIA_STOP', empty, empty],
675 > [1, ScanCode.Eject, 'Eject', KeyCode.Unknown, empty, 0, empty, empty, empty],
676 > [1, ScanCode.MediaPlayPause, 'MediaPlayPause', KeyCode.MediaPlayPause, 'MediaPlayPause', 179, 'VK_MEDIA_PLAY_PAUSE', empty, empty],
677 > [1, ScanCode.MediaSelect, 'MediaSelect', KeyCode.LaunchMediaPlayer, 'LaunchMediaPlayer', 181, 'VK_MEDIA_LAUNCH_MEDIA_SELECT', empty, empty],
678 > [1, ScanCode.LaunchMail, 'LaunchMail', KeyCode.LaunchMail, 'LaunchMail', 180, 'VK_MEDIA_LAUNCH_MAIL', empty, empty],
679 > [1, ScanCode.LaunchApp2, 'LaunchApp2', KeyCode.LaunchApp2, 'LaunchApp2', 183, 'VK_MEDIA_LAUNCH_APP2', empty, empty],
680 > [1, ScanCode.LaunchApp1, 'LaunchApp1', KeyCode.Unknown, empty, 0, 'VK_MEDIA_LAUNCH_APP1', empty, empty],
681 > [1, ScanCode.SelectTask, 'SelectTask', KeyCode.Unknown, empty, 0, empty, empty, empty],
682 > [1, ScanCode.LaunchScreenSaver, 'LaunchScreenSaver', KeyCode.Unknown, empty, 0, empty, empty, empty],
683 > [1, ScanCode.BrowserSearch, 'BrowserSearch', KeyCode.BrowserSearch, 'BrowserSearch', 170, 'VK_BROWSER_SEARCH', empty, empty],
684 > [1, ScanCode.BrowserHome, 'BrowserHome', KeyCode.BrowserHome, 'BrowserHome', 172, 'VK_BROWSER_HOME', empty, empty],
685 > [1, ScanCode.BrowserBack, 'BrowserBack', KeyCode.BrowserBack, 'BrowserBack', 166, 'VK_BROWSER_BACK', empty, empty],
686 > [1, ScanCode.BrowserForward, 'BrowserForward', KeyCode.BrowserForward, 'BrowserForward', 167, 'VK_BROWSER_FORWARD', empty, empty],
687 > [1, ScanCode.BrowserStop, 'BrowserStop', KeyCode.Unknown, empty, 0, 'VK_BROWSER_STOP', empty, empty],
688 > [1, ScanCode.BrowserRefresh, 'BrowserRefresh', KeyCode.Unknown, empty, 0, 'VK_BROWSER_REFRESH', empty, empty],
689 > [1, ScanCode.BrowserFavorites, 'BrowserFavorites', KeyCode.Unknown, empty, 0, 'VK_BROWSER_FAVORITES', empty, empty],
690 > [1, ScanCode.ZoomToggle, 'ZoomToggle', KeyCode.Unknown, empty, 0, empty, empty, empty],
691 > [1, ScanCode.MailReply, 'MailReply', KeyCode.Unknown, empty, 0, empty, empty, empty],
692 > [1, ScanCode.MailForward, 'MailForward', KeyCode.Unknown, empty, 0, empty, empty, empty],
693 > [1, ScanCode.MailSend, 'MailSend', KeyCode.Unknown, empty, 0, empty, empty, empty],
694 >
695 > // See https://lists.w3.org/Archives/Public/www-dom/2010JulSep/att-0182/keyCode-spec.html
696 > // If an Input Method Editor is processing key input and the event is keydown, return 229.
697 > [1, ScanCode.None, empty, KeyCode.KEY_IN_COMPOSITION, 'KeyInComposition', 229, empty, empty, empty],
698 > [1, ScanCode.None, empty, KeyCode.ABNT_C2, 'ABNT_C2', 194, 'VK_ABNT_C2', empty, empty],
699 > [1, ScanCode.None, empty, KeyCode.OEM_8, 'OEM_8', 223, 'VK_OEM_8', empty, empty],
700 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_KANA', empty, empty],
701 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_HANGUL', empty, empty],
702 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_JUNJA', empty, empty],
703 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_FINAL', empty, empty],
704 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_HANJA', empty, empty],
705 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_KANJI', empty, empty],
706 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_CONVERT', empty, empty],
707 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_NONCONVERT', empty, empty],
708 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_ACCEPT', empty, empty],
709 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_MODECHANGE', empty, empty],
710 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_SELECT', empty, empty],
711 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_PRINT', empty, empty],
712 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_EXECUTE', empty, empty],
713 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_SNAPSHOT', empty, empty],
714 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_HELP', empty, empty],
715 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_APPS', empty, empty],
716 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_PROCESSKEY', empty, empty],
717 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_PACKET', empty, empty],
718 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_DBE_SBCSCHAR', empty, empty],
719 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_DBE_DBCSCHAR', empty, empty],
720 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_ATTN', empty, empty],
721 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_CRSEL', empty, empty],
722 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_EXSEL', empty, empty],
723 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_EREOF', empty, empty],
724 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_PLAY', empty, empty],
725 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_ZOOM', empty, empty],
726 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_NONAME', empty, empty],
727 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_PA1', empty, empty],
728 > [1, ScanCode.None, empty, KeyCode.Unknown, empty, 0, 'VK_OEM_CLEAR', empty, empty],
729 > ];
730 >
731 > const seenKeyCode: boolean[] = [];
732 > const seenScanCode: boolean[] = [];
733 > for (const mapping of mappings) {
734 > const [immutable, scanCode, scanCodeStr, keyCode, keyCodeStr, eventKeyCode, vkey, usUserSettingsLabel, generalUserSettingsLabel] = mapping;
735 > if (!seenScanCode[scanCode]) {
736 > seenScanCode[scanCode] = true;
737 > scanCodeIntToStr[scanCode] = scanCodeStr;
738 > scanCodeStrToInt[scanCodeStr] = scanCode;
739 > scanCodeLowerCaseStrToInt[scanCodeStr.toLowerCase()] = scanCode;
740 > if (immutable) {
741 > IMMUTABLE_CODE_TO_KEY_CODE[scanCode] = keyCode;
742 > if ((keyCode !== KeyCode.Unknown) && (keyCode !== KeyCode.Enter) && !isModifierKey(keyCode)) {
743 > IMMUTABLE_KEY_CODE_TO_CODE[keyCode] = scanCode;
744 > }
745 > }
746 > }
747 > if (!seenKeyCode[keyCode]) {
748 > seenKeyCode[keyCode] = true;
749 > if (!keyCodeStr) {
750 throw new Error(`String representation missing for key code ${keyCode} around scan code ${scanCodeStr}`);
751 }
752 > uiMap.define(keyCode, keyCodeStr); keyCodes.ts
753 > userSettingsUSMap.define(keyCode, usUserSettingsLabel || keyCodeStr);
754 > userSettingsGeneralMap.define(keyCode, generalUserSettingsLabel || usUserSettingsLabel || keyCodeStr);
755 > }
756 > if (eventKeyCode) {
757 > EVENT_KEY_CODE_MAP[eventKeyCode] = keyCode;
758 > }
759 > if (scanCodeStr) {
760 > SCAN_CODE_STR_TO_EVENT_KEY_CODE[scanCodeStr] = eventKeyCode;
761 > }
762 > if (vkey) {
763 > NATIVE_WINDOWS_KEY_CODE_TO_KEY_CODE[vkey] = keyCode;
764 > }
765 > }
766 > // Manually added due to the exclusion above (due to duplication with NumpadEnter)
767 > IMMUTABLE_KEY_CODE_TO_CODE[KeyCode.Enter] = ScanCode.Enter;
768 >
769 > })();
770 >
771 > export namespace KeyCodeUtils {
772 > export function toString(keyCode: KeyCode): string {
773 return uiMap.keyCodeToStr(keyCode);
774 }
775 > export function fromString(key: string): KeyCode { keyCodes.ts
776 return uiMap.strToKeyCode(key);
777 }
778 > keyCodes.ts
779 > export function toUserSettingsUS(keyCode: KeyCode): string {
780 return userSettingsUSMap.keyCodeToStr(keyCode);
781 }
782 > export function toUserSettingsGeneral(keyCode: KeyCode): string { keyCodes.ts
783 return userSettingsGeneralMap.keyCodeToStr(keyCode);
784 }
785 > export function fromUserSettings(key: string): KeyCode { keyCodes.ts
786 return userSettingsUSMap.strToKeyCode(key) || userSettingsGeneralMap.strToKeyCode(key);
787 }
788 > keyCodes.ts
789 > export function toElectronAccelerator(keyCode: KeyCode): string | null {
790 if (keyCode >= KeyCode.Numpad0 && keyCode <= KeyCode.NumpadDivide) {
791 // [Electron Accelerators] Electron is able to parse numpad keys, but unfortunately it
815 return uiMap.keyCodeToStr(keyCode);
816 }
817 > } keyCodes.ts
818 >
819 > export const enum KeyMod {
820 > CtrlCmd = (1 << 11) >>> 0,
821 > Shift = (1 << 10) >>> 0,
822 > Alt = (1 << 9) >>> 0,
823 > WinCtrl = (1 << 8) >>> 0,
824 > }
825 >
826 > export function KeyChord(firstPart: number, secondPart: number): number {
827 const chordPart = ((secondPart & 0x0000FFFF) << 16) >>> 0;
828 return (firstPart | chordPart) >>> 0;
829 }
830 > keyCodes.ts
831 > export function isModifierKey(keyCode: KeyCode): boolean {
832 > return (
833 > keyCode === KeyCode.Ctrl
834 > || keyCode === KeyCode.Shift
835 > || keyCode === KeyCode.Alt
836 > || keyCode === KeyCode.Meta
837 > );
838 > }
src/vs/base/common/codiconsLibrary.ts 758 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- codiconsLibrary.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { register } from './codiconsUtil.js';
6 >
7 >
8 > // This file is automatically generated by (microsoft/vscode-codicons)/scripts/export-to-ts.js
9 > // Please don't edit it, as your changes will be overwritten.
10 > // Instead, add mappings to codiconsDerived in codicons.ts.
11 > export const codiconsLibrary = {
12 > add: register('add', 0xea60),
13 > plus: register('plus', 0xea60),
14 > gistNew: register('gist-new', 0xea60),
15 > repoCreate: register('repo-create', 0xea60),
16 > lightbulb: register('lightbulb', 0xea61),
17 > lightBulb: register('light-bulb', 0xea61),
18 > repo: register('repo', 0xea62),
19 > repoDelete: register('repo-delete', 0xea62),
20 > gistFork: register('gist-fork', 0xea63),
21 > repoForked: register('repo-forked', 0xea63),
22 > gitPullRequest: register('git-pull-request', 0xea64),
23 > gitPullRequestAbandoned: register('git-pull-request-abandoned', 0xea64),
24 > recordKeys: register('record-keys', 0xea65),
25 > keyboard: register('keyboard', 0xea65),
26 > tag: register('tag', 0xea66),
27 > gitPullRequestLabel: register('git-pull-request-label', 0xea66),
28 > tagAdd: register('tag-add', 0xea66),
29 > tagRemove: register('tag-remove', 0xea66),
30 > person: register('person', 0xea67),
31 > personFollow: register('person-follow', 0xea67),
32 > personOutline: register('person-outline', 0xea67),
33 > personFilled: register('person-filled', 0xea67),
34 > sourceControl: register('source-control', 0xea68),
35 > mirror: register('mirror', 0xea69),
36 > mirrorPublic: register('mirror-public', 0xea69),
37 > star: register('star', 0xea6a),
38 > starAdd: register('star-add', 0xea6a),
39 > starDelete: register('star-delete', 0xea6a),
40 > starEmpty: register('star-empty', 0xea6a),
41 > comment: register('comment', 0xea6b),
42 > commentAdd: register('comment-add', 0xea6b),
43 > alert: register('alert', 0xea6c),
44 > warning: register('warning', 0xea6c),
45 > search: register('search', 0xea6d),
46 > searchSave: register('search-save', 0xea6d),
47 > logOut: register('log-out', 0xea6e),
48 > signOut: register('sign-out', 0xea6e),
49 > logIn: register('log-in', 0xea6f),
50 > signIn: register('sign-in', 0xea6f),
51 > eye: register('eye', 0xea70),
52 > eyeUnwatch: register('eye-unwatch', 0xea70),
53 > eyeWatch: register('eye-watch', 0xea70),
54 > circleFilled: register('circle-filled', 0xea71),
55 > primitiveDot: register('primitive-dot', 0xea71),
56 > closeDirty: register('close-dirty', 0xea71),
57 > debugBreakpoint: register('debug-breakpoint', 0xea71),
58 > debugBreakpointDisabled: register('debug-breakpoint-disabled', 0xea71),
59 > debugHint: register('debug-hint', 0xea71),
60 > terminalDecorationSuccess: register('terminal-decoration-success', 0xea71),
61 > primitiveSquare: register('primitive-square', 0xea72),
62 > edit: register('edit', 0xea73),
63 > pencil: register('pencil', 0xea73),
64 > info: register('info', 0xea74),
65 > issueOpened: register('issue-opened', 0xea74),
66 > gistPrivate: register('gist-private', 0xea75),
67 > gitForkPrivate: register('git-fork-private', 0xea75),
68 > lock: register('lock', 0xea75),
69 > mirrorPrivate: register('mirror-private', 0xea75),
70 > close: register('close', 0xea76),
71 > removeClose: register('remove-close', 0xea76),
72 > x: register('x', 0xea76),
73 > repoSync: register('repo-sync', 0xea77),
74 > sync: register('sync', 0xea77),
75 > clone: register('clone', 0xea78),
76 > desktopDownload: register('desktop-download', 0xea78),
77 > beaker: register('beaker', 0xea79),
78 > microscope: register('microscope', 0xea79),
79 > vm: register('vm', 0xea7a),
80 > deviceDesktop: register('device-desktop', 0xea7a),
81 > file: register('file', 0xea7b),
82 > more: register('more', 0xea7c),
83 > ellipsis: register('ellipsis', 0xea7c),
84 > kebabHorizontal: register('kebab-horizontal', 0xea7c),
85 > mailReply: register('mail-reply', 0xea7d),
86 > reply: register('reply', 0xea7d),
87 > organization: register('organization', 0xea7e),
88 > organizationFilled: register('organization-filled', 0xea7e),
89 > organizationOutline: register('organization-outline', 0xea7e),
90 > newFile: register('new-file', 0xea7f),
91 > fileAdd: register('file-add', 0xea7f),
92 > newFolder: register('new-folder', 0xea80),
93 > fileDirectoryCreate: register('file-directory-create', 0xea80),
94 > trash: register('trash', 0xea81),
95 > trashcan: register('trashcan', 0xea81),
96 > history: register('history', 0xea82),
97 > clock: register('clock', 0xea82),
98 > folder: register('folder', 0xea83),
99 > fileDirectory: register('file-directory', 0xea83),
100 > symbolFolder: register('symbol-folder', 0xea83),
101 > logoGithub: register('logo-github', 0xea84),
102 > markGithub: register('mark-github', 0xea84),
103 > github: register('github', 0xea84),
104 > terminal: register('terminal', 0xea85),
105 > console: register('console', 0xea85),
106 > repl: register('repl', 0xea85),
107 > zap: register('zap', 0xea86),
108 > symbolEvent: register('symbol-event', 0xea86),
109 > error: register('error', 0xea87),
110 > stop: register('stop', 0xea87),
111 > variable: register('variable', 0xea88),
112 > symbolVariable: register('symbol-variable', 0xea88),
113 > array: register('array', 0xea8a),
114 > symbolArray: register('symbol-array', 0xea8a),
115 > symbolModule: register('symbol-module', 0xea8b),
116 > symbolPackage: register('symbol-package', 0xea8b),
117 > symbolNamespace: register('symbol-namespace', 0xea8b),
118 > symbolObject: register('symbol-object', 0xea8b),
119 > symbolMethod: register('symbol-method', 0xea8c),
120 > symbolFunction: register('symbol-function', 0xea8c),
121 > symbolConstructor: register('symbol-constructor', 0xea8c),
122 > symbolBoolean: register('symbol-boolean', 0xea8f),
123 > symbolNull: register('symbol-null', 0xea8f),
124 > symbolNumeric: register('symbol-numeric', 0xea90),
125 > symbolNumber: register('symbol-number', 0xea90),
126 > symbolStructure: register('symbol-structure', 0xea91),
127 > symbolStruct: register('symbol-struct', 0xea91),
128 > symbolParameter: register('symbol-parameter', 0xea92),
129 > symbolTypeParameter: register('symbol-type-parameter', 0xea92),
130 > symbolKey: register('symbol-key', 0xea93),
131 > symbolText: register('symbol-text', 0xea93),
132 > symbolReference: register('symbol-reference', 0xea94),
133 > goToFile: register('go-to-file', 0xea94),
134 > symbolEnum: register('symbol-enum', 0xea95),
135 > symbolValue: register('symbol-value', 0xea95),
136 > symbolRuler: register('symbol-ruler', 0xea96),
137 > symbolUnit: register('symbol-unit', 0xea96),
138 > activateBreakpoints: register('activate-breakpoints', 0xea97),
139 > archive: register('archive', 0xea98),
140 > arrowBoth: register('arrow-both', 0xea99),
141 > arrowDown: register('arrow-down', 0xea9a),
142 > arrowLeft: register('arrow-left', 0xea9b),
143 > arrowRight: register('arrow-right', 0xea9c),
144 > arrowSmallDown: register('arrow-small-down', 0xea9d),
145 > arrowSmallLeft: register('arrow-small-left', 0xea9e),
146 > arrowSmallRight: register('arrow-small-right', 0xea9f),
147 > arrowSmallUp: register('arrow-small-up', 0xeaa0),
148 > arrowUp: register('arrow-up', 0xeaa1),
149 > bell: register('bell', 0xeaa2),
150 > bold: register('bold', 0xeaa3),
151 > book: register('book', 0xeaa4),
152 > bookmark: register('bookmark', 0xeaa5),
153 > debugBreakpointConditionalUnverified: register('debug-breakpoint-conditional-unverified', 0xeaa6),
154 > debugBreakpointConditional: register('debug-breakpoint-conditional', 0xeaa7),
155 > debugBreakpointConditionalDisabled: register('debug-breakpoint-conditional-disabled', 0xeaa7),
156 > debugBreakpointDataUnverified: register('debug-breakpoint-data-unverified', 0xeaa8),
157 > debugBreakpointData: register('debug-breakpoint-data', 0xeaa9),
158 > debugBreakpointDataDisabled: register('debug-breakpoint-data-disabled', 0xeaa9),
159 > debugBreakpointLogUnverified: register('debug-breakpoint-log-unverified', 0xeaaa),
160 > debugBreakpointLog: register('debug-breakpoint-log', 0xeaab),
161 > debugBreakpointLogDisabled: register('debug-breakpoint-log-disabled', 0xeaab),
162 > briefcase: register('briefcase', 0xeaac),
163 > broadcast: register('broadcast', 0xeaad),
164 > browser: register('browser', 0xeaae),
165 > bug: register('bug', 0xeaaf),
166 > calendar: register('calendar', 0xeab0),
167 > caseSensitive: register('case-sensitive', 0xeab1),
168 > check: register('check', 0xeab2),
169 > checklist: register('checklist', 0xeab3),
170 > chevronDown: register('chevron-down', 0xeab4),
171 > chevronLeft: register('chevron-left', 0xeab5),
172 > chevronRight: register('chevron-right', 0xeab6),
173 > chevronUp: register('chevron-up', 0xeab7),
174 > chromeClose: register('chrome-close', 0xeab8),
175 > chromeMaximize: register('chrome-maximize', 0xeab9),
176 > chromeMinimize: register('chrome-minimize', 0xeaba),
177 > chromeRestore: register('chrome-restore', 0xeabb),
178 > circleOutline: register('circle-outline', 0xeabc),
179 > circle: register('circle', 0xeabc),
180 > debugBreakpointUnverified: register('debug-breakpoint-unverified', 0xeabc),
181 > terminalDecorationIncomplete: register('terminal-decoration-incomplete', 0xeabc),
182 > circleSlash: register('circle-slash', 0xeabd),
183 > circuitBoard: register('circuit-board', 0xeabe),
184 > clearAll: register('clear-all', 0xeabf),
185 > clippy: register('clippy', 0xeac0),
186 > closeAll: register('close-all', 0xeac1),
187 > cloudDownload: register('cloud-download', 0xeac2),
188 > cloudUpload: register('cloud-upload', 0xeac3),
189 > code: register('code', 0xeac4),
190 > collapseAll: register('collapse-all', 0xeac5),
191 > colorMode: register('color-mode', 0xeac6),
192 > commentDiscussion: register('comment-discussion', 0xeac7),
193 > creditCard: register('credit-card', 0xeac9),
194 > dash: register('dash', 0xeacc),
195 > dashboard: register('dashboard', 0xeacd),
196 > database: register('database', 0xeace),
197 > debugContinue: register('debug-continue', 0xeacf),
198 > debugDisconnect: register('debug-disconnect', 0xead0),
199 > debugPause: register('debug-pause', 0xead1),
200 > debugRestart: register('debug-restart', 0xead2),
201 > debugStart: register('debug-start', 0xead3),
202 > debugStepInto: register('debug-step-into', 0xead4),
203 > debugStepOut: register('debug-step-out', 0xead5),
204 > debugStepOver: register('debug-step-over', 0xead6),
205 > debugStop: register('debug-stop', 0xead7),
206 > debug: register('debug', 0xead8),
207 > deviceCameraVideo: register('device-camera-video', 0xead9),
208 > deviceCamera: register('device-camera', 0xeada),
209 > deviceMobile: register('device-mobile', 0xeadb),
210 > diffAdded: register('diff-added', 0xeadc),
211 > diffIgnored: register('diff-ignored', 0xeadd),
212 > diffModified: register('diff-modified', 0xeade),
213 > diffRemoved: register('diff-removed', 0xeadf),
214 > diffRenamed: register('diff-renamed', 0xeae0),
215 > diff: register('diff', 0xeae1),
216 > diffSidebyside: register('diff-sidebyside', 0xeae1),
217 > discard: register('discard', 0xeae2),
218 > editorLayout: register('editor-layout', 0xeae3),
219 > emptyWindow: register('empty-window', 0xeae4),
220 > exclude: register('exclude', 0xeae5),
221 > extensions: register('extensions', 0xeae6),
222 > eyeClosed: register('eye-closed', 0xeae7),
223 > fileBinary: register('file-binary', 0xeae8),
224 > fileCode: register('file-code', 0xeae9),
225 > fileMedia: register('file-media', 0xeaea),
226 > filePdf: register('file-pdf', 0xeaeb),
227 > fileSubmodule: register('file-submodule', 0xeaec),
228 > fileSymlinkDirectory: register('file-symlink-directory', 0xeaed),
229 > fileSymlinkFile: register('file-symlink-file', 0xeaee),
230 > fileZip: register('file-zip', 0xeaef),
231 > files: register('files', 0xeaf0),
232 > filter: register('filter', 0xeaf1),
233 > flame: register('flame', 0xeaf2),
234 > foldDown: register('fold-down', 0xeaf3),
235 > foldUp: register('fold-up', 0xeaf4),
236 > fold: register('fold', 0xeaf5),
237 > folderActive: register('folder-active', 0xeaf6),
238 > folderOpened: register('folder-opened', 0xeaf7),
239 > gear: register('gear', 0xeaf8),
240 > gift: register('gift', 0xeaf9),
241 > gistSecret: register('gist-secret', 0xeafa),
242 > gist: register('gist', 0xeafb),
243 > gitCommit: register('git-commit', 0xeafc),
244 > gitCompare: register('git-compare', 0xeafd),
245 > compareChanges: register('compare-changes', 0xeafd),
246 > gitMerge: register('git-merge', 0xeafe),
247 > githubAction: register('github-action', 0xeaff),
248 > githubAlt: register('github-alt', 0xeb00),
249 > globe: register('globe', 0xeb01),
250 > grabber: register('grabber', 0xeb02),
251 > graph: register('graph', 0xeb03),
252 > gripper: register('gripper', 0xeb04),
253 > heart: register('heart', 0xeb05),
254 > home: register('home', 0xeb06),
255 > horizontalRule: register('horizontal-rule', 0xeb07),
256 > hubot: register('hubot', 0xeb08),
257 > inbox: register('inbox', 0xeb09),
258 > issueReopened: register('issue-reopened', 0xeb0b),
259 > issues: register('issues', 0xeb0c),
260 > italic: register('italic', 0xeb0d),
261 > jersey: register('jersey', 0xeb0e),
262 > json: register('json', 0xeb0f),
263 > bracket: register('bracket', 0xeb0f),
264 > kebabVertical: register('kebab-vertical', 0xeb10),
265 > key: register('key', 0xeb11),
266 > law: register('law', 0xeb12),
267 > lightbulbAutofix: register('lightbulb-autofix', 0xeb13),
268 > linkExternal: register('link-external', 0xeb14),
269 > link: register('link', 0xeb15),
270 > listOrdered: register('list-ordered', 0xeb16),
271 > listUnordered: register('list-unordered', 0xeb17),
272 > liveShare: register('live-share', 0xeb18),
273 > loading: register('loading', 0xeb19),
274 > location: register('location', 0xeb1a),
275 > mailRead: register('mail-read', 0xeb1b),
276 > mail: register('mail', 0xeb1c),
277 > markdown: register('markdown', 0xeb1d),
278 > megaphone: register('megaphone', 0xeb1e),
279 > mention: register('mention', 0xeb1f),
280 > milestone: register('milestone', 0xeb20),
281 > gitPullRequestMilestone: register('git-pull-request-milestone', 0xeb20),
282 > mortarBoard: register('mortar-board', 0xeb21),
283 > move: register('move', 0xeb22),
284 > multipleWindows: register('multiple-windows', 0xeb23),
285 > mute: register('mute', 0xeb24),
286 > noNewline: register('no-newline', 0xeb25),
287 > note: register('note', 0xeb26),
288 > octoface: register('octoface', 0xeb27),
289 > openPreview: register('open-preview', 0xeb28),
290 > package: register('package', 0xeb29),
291 > paintcan: register('paintcan', 0xeb2a),
292 > pin: register('pin', 0xeb2b),
293 > play: register('play', 0xeb2c),
294 > run: register('run', 0xeb2c),
295 > plug: register('plug', 0xeb2d),
296 > preserveCase: register('preserve-case', 0xeb2e),
297 > preview: register('preview', 0xeb2f),
298 > project: register('project', 0xeb30),
299 > pulse: register('pulse', 0xeb31),
300 > question: register('question', 0xeb32),
301 > quote: register('quote', 0xeb33),
302 > radioTower: register('radio-tower', 0xeb34),
303 > reactions: register('reactions', 0xeb35),
304 > references: register('references', 0xeb36),
305 > refresh: register('refresh', 0xeb37),
306 > regex: register('regex', 0xeb38),
307 > remoteExplorer: register('remote-explorer', 0xeb39),
308 > remote: register('remote', 0xeb3a),
309 > remove: register('remove', 0xeb3b),
310 > replaceAll: register('replace-all', 0xeb3c),
311 > replace: register('replace', 0xeb3d),
312 > repoClone: register('repo-clone', 0xeb3e),
313 > repoForcePush: register('repo-force-push', 0xeb3f),
314 > repoPull: register('repo-pull', 0xeb40),
315 > repoPush: register('repo-push', 0xeb41),
316 > report: register('report', 0xeb42),
317 > requestChanges: register('request-changes', 0xeb43),
318 > rocket: register('rocket', 0xeb44),
319 > rootFolderOpened: register('root-folder-opened', 0xeb45),
320 > rootFolder: register('root-folder', 0xeb46),
321 > rss: register('rss', 0xeb47),
322 > ruby: register('ruby', 0xeb48),
323 > saveAll: register('save-all', 0xeb49),
324 > saveAs: register('save-as', 0xeb4a),
325 > save: register('save', 0xeb4b),
326 > screenFull: register('screen-full', 0xeb4c),
327 > screenNormal: register('screen-normal', 0xeb4d),
328 > searchStop: register('search-stop', 0xeb4e),
329 > server: register('server', 0xeb50),
330 > settingsGear: register('settings-gear', 0xeb51),
331 > settings: register('settings', 0xeb52),
332 > shield: register('shield', 0xeb53),
333 > smiley: register('smiley', 0xeb54),
334 > sortPrecedence: register('sort-precedence', 0xeb55),
335 > splitHorizontal: register('split-horizontal', 0xeb56),
336 > splitVertical: register('split-vertical', 0xeb57),
337 > squirrel: register('squirrel', 0xeb58),
338 > starFull: register('star-full', 0xeb59),
339 > starHalf: register('star-half', 0xeb5a),
340 > symbolClass: register('symbol-class', 0xeb5b),
341 > symbolColor: register('symbol-color', 0xeb5c),
342 > symbolConstant: register('symbol-constant', 0xeb5d),
343 > symbolEnumMember: register('symbol-enum-member', 0xeb5e),
344 > symbolField: register('symbol-field', 0xeb5f),
345 > symbolFile: register('symbol-file', 0xeb60),
346 > symbolInterface: register('symbol-interface', 0xeb61),
347 > symbolKeyword: register('symbol-keyword', 0xeb62),
348 > symbolMisc: register('symbol-misc', 0xeb63),
349 > symbolOperator: register('symbol-operator', 0xeb64),
350 > symbolProperty: register('symbol-property', 0xeb65),
351 > wrench: register('wrench', 0xeb65),
352 > wrenchSubaction: register('wrench-subaction', 0xeb65),
353 > symbolSnippet: register('symbol-snippet', 0xeb66),
354 > tasklist: register('tasklist', 0xeb67),
355 > telescope: register('telescope', 0xeb68),
356 > textSize: register('text-size', 0xeb69),
357 > threeBars: register('three-bars', 0xeb6a),
358 > thumbsdown: register('thumbsdown', 0xeb6b),
359 > thumbsup: register('thumbsup', 0xeb6c),
360 > tools: register('tools', 0xeb6d),
361 > triangleDown: register('triangle-down', 0xeb6e),
362 > triangleLeft: register('triangle-left', 0xeb6f),
363 > triangleRight: register('triangle-right', 0xeb70),
364 > triangleUp: register('triangle-up', 0xeb71),
365 > twitter: register('twitter', 0xeb72),
366 > unfold: register('unfold', 0xeb73),
367 > unlock: register('unlock', 0xeb74),
368 > unmute: register('unmute', 0xeb75),
369 > unverified: register('unverified', 0xeb76),
370 > verified: register('verified', 0xeb77),
371 > versions: register('versions', 0xeb78),
372 > vmActive: register('vm-active', 0xeb79),
373 > vmOutline: register('vm-outline', 0xeb7a),
374 > vmRunning: register('vm-running', 0xeb7b),
375 > watch: register('watch', 0xeb7c),
376 > whitespace: register('whitespace', 0xeb7d),
377 > wholeWord: register('whole-word', 0xeb7e),
378 > window: register('window', 0xeb7f),
379 > wordWrap: register('word-wrap', 0xeb80),
380 > zoomIn: register('zoom-in', 0xeb81),
381 > zoomOut: register('zoom-out', 0xeb82),
382 > listFilter: register('list-filter', 0xeb83),
383 > listFlat: register('list-flat', 0xeb84),
384 > listSelection: register('list-selection', 0xeb85),
385 > selection: register('selection', 0xeb85),
386 > listTree: register('list-tree', 0xeb86),
387 > debugBreakpointFunctionUnverified: register('debug-breakpoint-function-unverified', 0xeb87),
388 > debugBreakpointFunction: register('debug-breakpoint-function', 0xeb88),
389 > debugBreakpointFunctionDisabled: register('debug-breakpoint-function-disabled', 0xeb88),
390 > debugStackframeActive: register('debug-stackframe-active', 0xeb89),
391 > circleSmallFilled: register('circle-small-filled', 0xeb8a),
392 > debugStackframeDot: register('debug-stackframe-dot', 0xeb8a),
393 > terminalDecorationMark: register('terminal-decoration-mark', 0xeb8a),
394 > debugStackframe: register('debug-stackframe', 0xeb8b),
395 > debugStackframeFocused: register('debug-stackframe-focused', 0xeb8b),
396 > debugBreakpointUnsupported: register('debug-breakpoint-unsupported', 0xeb8c),
397 > symbolString: register('symbol-string', 0xeb8d),
398 > debugReverseContinue: register('debug-reverse-continue', 0xeb8e),
399 > debugStepBack: register('debug-step-back', 0xeb8f),
400 > debugRestartFrame: register('debug-restart-frame', 0xeb90),
401 > debugAlt: register('debug-alt', 0xeb91),
402 > callIncoming: register('call-incoming', 0xeb92),
403 > callOutgoing: register('call-outgoing', 0xeb93),
404 > menu: register('menu', 0xeb94),
405 > expandAll: register('expand-all', 0xeb95),
406 > feedback: register('feedback', 0xeb96),
407 > gitPullRequestReviewer: register('git-pull-request-reviewer', 0xeb96),
408 > groupByRefType: register('group-by-ref-type', 0xeb97),
409 > ungroupByRefType: register('ungroup-by-ref-type', 0xeb98),
410 > account: register('account', 0xeb99),
411 > gitPullRequestAssignee: register('git-pull-request-assignee', 0xeb99),
412 > bellDot: register('bell-dot', 0xeb9a),
413 > debugConsole: register('debug-console', 0xeb9b),
414 > library: register('library', 0xeb9c),
415 > output: register('output', 0xeb9d),
416 > runAll: register('run-all', 0xeb9e),
417 > syncIgnored: register('sync-ignored', 0xeb9f),
418 > pinned: register('pinned', 0xeba0),
419 > githubInverted: register('github-inverted', 0xeba1),
420 > serverProcess: register('server-process', 0xeba2),
421 > serverEnvironment: register('server-environment', 0xeba3),
422 > pass: register('pass', 0xeba4),
423 > issueClosed: register('issue-closed', 0xeba4),
424 > stopCircle: register('stop-circle', 0xeba5),
425 > playCircle: register('play-circle', 0xeba6),
426 > record: register('record', 0xeba7),
427 > debugAltSmall: register('debug-alt-small', 0xeba8),
428 > vmConnect: register('vm-connect', 0xeba9),
429 > cloud: register('cloud', 0xebaa),
430 > merge: register('merge', 0xebab),
431 > export: register('export', 0xebac),
432 > graphLeft: register('graph-left', 0xebad),
433 > magnet: register('magnet', 0xebae),
434 > notebook: register('notebook', 0xebaf),
435 > redo: register('redo', 0xebb0),
436 > checkAll: register('check-all', 0xebb1),
437 > pinnedDirty: register('pinned-dirty', 0xebb2),
438 > passFilled: register('pass-filled', 0xebb3),
439 > circleLargeFilled: register('circle-large-filled', 0xebb4),
440 > circleLarge: register('circle-large', 0xebb5),
441 > circleLargeOutline: register('circle-large-outline', 0xebb5),
442 > combine: register('combine', 0xebb6),
443 > gather: register('gather', 0xebb6),
444 > table: register('table', 0xebb7),
445 > variableGroup: register('variable-group', 0xebb8),
446 > typeHierarchy: register('type-hierarchy', 0xebb9),
447 > typeHierarchySub: register('type-hierarchy-sub', 0xebba),
448 > typeHierarchySuper: register('type-hierarchy-super', 0xebbb),
449 > gitPullRequestCreate: register('git-pull-request-create', 0xebbc),
450 > runAbove: register('run-above', 0xebbd),
451 > runBelow: register('run-below', 0xebbe),
452 > notebookTemplate: register('notebook-template', 0xebbf),
453 > debugRerun: register('debug-rerun', 0xebc0),
454 > workspaceTrusted: register('workspace-trusted', 0xebc1),
455 > workspaceUntrusted: register('workspace-untrusted', 0xebc2),
456 > workspaceUnknown: register('workspace-unknown', 0xebc3),
457 > terminalCmd: register('terminal-cmd', 0xebc4),
458 > terminalDebian: register('terminal-debian', 0xebc5),
459 > terminalLinux: register('terminal-linux', 0xebc6),
460 > terminalPowershell: register('terminal-powershell', 0xebc7),
461 > terminalTmux: register('terminal-tmux', 0xebc8),
462 > terminalUbuntu: register('terminal-ubuntu', 0xebc9),
463 > terminalBash: register('terminal-bash', 0xebca),
464 > arrowSwap: register('arrow-swap', 0xebcb),
465 > copy: register('copy', 0xebcc),
466 > personAdd: register('person-add', 0xebcd),
467 > filterFilled: register('filter-filled', 0xebce),
468 > wand: register('wand', 0xebcf),
469 > debugLineByLine: register('debug-line-by-line', 0xebd0),
470 > inspect: register('inspect', 0xebd1),
471 > layers: register('layers', 0xebd2),
472 > layersDot: register('layers-dot', 0xebd3),
473 > layersActive: register('layers-active', 0xebd4),
474 > compass: register('compass', 0xebd5),
475 > compassDot: register('compass-dot', 0xebd6),
476 > compassActive: register('compass-active', 0xebd7),
477 > azure: register('azure', 0xebd8),
478 > issueDraft: register('issue-draft', 0xebd9),
479 > gitPullRequestClosed: register('git-pull-request-closed', 0xebda),
480 > gitPullRequestDraft: register('git-pull-request-draft', 0xebdb),
481 > debugAll: register('debug-all', 0xebdc),
482 > debugCoverage: register('debug-coverage', 0xebdd),
483 > runErrors: register('run-errors', 0xebde),
484 > folderLibrary: register('folder-library', 0xebdf),
485 > debugContinueSmall: register('debug-continue-small', 0xebe0),
486 > beakerStop: register('beaker-stop', 0xebe1),
487 > graphLine: register('graph-line', 0xebe2),
488 > graphScatter: register('graph-scatter', 0xebe3),
489 > pieChart: register('pie-chart', 0xebe4),
490 > bracketDot: register('bracket-dot', 0xebe5),
491 > bracketError: register('bracket-error', 0xebe6),
492 > lockSmall: register('lock-small', 0xebe7),
493 > azureDevops: register('azure-devops', 0xebe8),
494 > verifiedFilled: register('verified-filled', 0xebe9),
495 > newline: register('newline', 0xebea),
496 > layout: register('layout', 0xebeb),
497 > layoutActivitybarLeft: register('layout-activitybar-left', 0xebec),
498 > layoutActivitybarRight: register('layout-activitybar-right', 0xebed),
499 > layoutPanelLeft: register('layout-panel-left', 0xebee),
500 > layoutPanelCenter: register('layout-panel-center', 0xebef),
501 > layoutPanelJustify: register('layout-panel-justify', 0xebf0),
502 > layoutPanelRight: register('layout-panel-right', 0xebf1),
503 > layoutPanel: register('layout-panel', 0xebf2),
504 > layoutSidebarLeft: register('layout-sidebar-left', 0xebf3),
505 > layoutSidebarRight: register('layout-sidebar-right', 0xebf4),
506 > layoutStatusbar: register('layout-statusbar', 0xebf5),
507 > layoutMenubar: register('layout-menubar', 0xebf6),
508 > layoutCentered: register('layout-centered', 0xebf7),
509 > target: register('target', 0xebf8),
510 > indent: register('indent', 0xebf9),
511 > recordSmall: register('record-small', 0xebfa),
512 > errorSmall: register('error-small', 0xebfb),
513 > terminalDecorationError: register('terminal-decoration-error', 0xebfb),
514 > arrowCircleDown: register('arrow-circle-down', 0xebfc),
515 > arrowCircleLeft: register('arrow-circle-left', 0xebfd),
516 > arrowCircleRight: register('arrow-circle-right', 0xebfe),
517 > arrowCircleUp: register('arrow-circle-up', 0xebff),
518 > layoutSidebarRightOff: register('layout-sidebar-right-off', 0xec00),
519 > layoutPanelOff: register('layout-panel-off', 0xec01),
520 > layoutSidebarLeftOff: register('layout-sidebar-left-off', 0xec02),
521 > blank: register('blank', 0xec03),
522 > heartFilled: register('heart-filled', 0xec04),
523 > map: register('map', 0xec05),
524 > mapHorizontal: register('map-horizontal', 0xec05),
525 > foldHorizontal: register('fold-horizontal', 0xec05),
526 > mapFilled: register('map-filled', 0xec06),
527 > mapHorizontalFilled: register('map-horizontal-filled', 0xec06),
528 > foldHorizontalFilled: register('fold-horizontal-filled', 0xec06),
529 > circleSmall: register('circle-small', 0xec07),
530 > bellSlash: register('bell-slash', 0xec08),
531 > bellSlashDot: register('bell-slash-dot', 0xec09),
532 > commentUnresolved: register('comment-unresolved', 0xec0a),
533 > gitPullRequestGoToChanges: register('git-pull-request-go-to-changes', 0xec0b),
534 > gitPullRequestNewChanges: register('git-pull-request-new-changes', 0xec0c),
535 > searchFuzzy: register('search-fuzzy', 0xec0d),
536 > commentDraft: register('comment-draft', 0xec0e),
537 > send: register('send', 0xec0f),
538 > sparkle: register('sparkle', 0xec10),
539 > insert: register('insert', 0xec11),
540 > mic: register('mic', 0xec12),
541 > thumbsdownFilled: register('thumbsdown-filled', 0xec13),
542 > thumbsupFilled: register('thumbsup-filled', 0xec14),
543 > coffee: register('coffee', 0xec15),
544 > snake: register('snake', 0xec16),
545 > game: register('game', 0xec17),
546 > vr: register('vr', 0xec18),
547 > chip: register('chip', 0xec19),
548 > piano: register('piano', 0xec1a),
549 > music: register('music', 0xec1b),
550 > micFilled: register('mic-filled', 0xec1c),
551 > repoFetch: register('repo-fetch', 0xec1d),
552 > copilot: register('copilot', 0xec1e),
553 > lightbulbSparkle: register('lightbulb-sparkle', 0xec1f),
554 > robot: register('robot', 0xec20),
555 > sparkleFilled: register('sparkle-filled', 0xec21),
556 > diffSingle: register('diff-single', 0xec22),
557 > diffMultiple: register('diff-multiple', 0xec23),
558 > surroundWith: register('surround-with', 0xec24),
559 > share: register('share', 0xec25),
560 > gitStash: register('git-stash', 0xec26),
561 > gitStashApply: register('git-stash-apply', 0xec27),
562 > gitStashPop: register('git-stash-pop', 0xec28),
563 > vscode: register('vscode', 0xec29),
564 > vscodeInsiders: register('vscode-insiders', 0xec2a),
565 > codeOss: register('code-oss', 0xec2b),
566 > runCoverage: register('run-coverage', 0xec2c),
567 > runAllCoverage: register('run-all-coverage', 0xec2d),
568 > coverage: register('coverage', 0xec2e),
569 > githubProject: register('github-project', 0xec2f),
570 > mapVertical: register('map-vertical', 0xec30),
571 > foldVertical: register('fold-vertical', 0xec30),
572 > mapVerticalFilled: register('map-vertical-filled', 0xec31),
573 > foldVerticalFilled: register('fold-vertical-filled', 0xec31),
574 > goToSearch: register('go-to-search', 0xec32),
575 > percentage: register('percentage', 0xec33),
576 > sortPercentage: register('sort-percentage', 0xec33),
577 > attach: register('attach', 0xec34),
578 > goToEditingSession: register('go-to-editing-session', 0xec35),
579 > editSession: register('edit-session', 0xec36),
580 > codeReview: register('code-review', 0xec37),
581 > copilotWarning: register('copilot-warning', 0xec38),
582 > python: register('python', 0xec39),
583 > copilotLarge: register('copilot-large', 0xec3a),
584 > copilotWarningLarge: register('copilot-warning-large', 0xec3b),
585 > keyboardTab: register('keyboard-tab', 0xec3c),
586 > copilotBlocked: register('copilot-blocked', 0xec3d),
587 > copilotNotConnected: register('copilot-not-connected', 0xec3e),
588 > flag: register('flag', 0xec3f),
589 > lightbulbEmpty: register('lightbulb-empty', 0xec40),
590 > symbolMethodArrow: register('symbol-method-arrow', 0xec41),
591 > copilotUnavailable: register('copilot-unavailable', 0xec42),
592 > repoPinned: register('repo-pinned', 0xec43),
593 > keyboardTabAbove: register('keyboard-tab-above', 0xec44),
594 > keyboardTabBelow: register('keyboard-tab-below', 0xec45),
595 > gitPullRequestDone: register('git-pull-request-done', 0xec46),
596 > mcp: register('mcp', 0xec47),
597 > extensionsLarge: register('extensions-large', 0xec48),
598 > layoutPanelDock: register('layout-panel-dock', 0xec49),
599 > layoutSidebarLeftDock: register('layout-sidebar-left-dock', 0xec4a),
600 > layoutSidebarRightDock: register('layout-sidebar-right-dock', 0xec4b),
601 > copilotInProgress: register('copilot-in-progress', 0xec4c),
602 > copilotError: register('copilot-error', 0xec4d),
603 > copilotSuccess: register('copilot-success', 0xec4e),
604 > chatSparkle: register('chat-sparkle', 0xec4f),
605 > searchSparkle: register('search-sparkle', 0xec50),
606 > editSparkle: register('edit-sparkle', 0xec51),
607 > copilotSnooze: register('copilot-snooze', 0xec52),
608 > sendToRemoteAgent: register('send-to-remote-agent', 0xec53),
609 > commentDiscussionSparkle: register('comment-discussion-sparkle', 0xec54),
610 > chatSparkleWarning: register('chat-sparkle-warning', 0xec55),
611 > chatSparkleError: register('chat-sparkle-error', 0xec56),
612 > collection: register('collection', 0xec57),
613 > newCollection: register('new-collection', 0xec58),
614 > thinking: register('thinking', 0xec59),
615 > build: register('build', 0xec5a),
616 > commentDiscussionQuote: register('comment-discussion-quote', 0xec5b),
617 > cursor: register('cursor', 0xec5c),
618 > eraser: register('eraser', 0xec5d),
619 > fileText: register('file-text', 0xec5e),
620 > quotes: register('quotes', 0xec60),
621 > rename: register('rename', 0xec61),
622 > runWithDeps: register('run-with-deps', 0xec62),
623 > debugConnected: register('debug-connected', 0xec63),
624 > strikethrough: register('strikethrough', 0xec64),
625 > openInProduct: register('open-in-product', 0xec65),
626 > indexZero: register('index-zero', 0xec66),
627 > agent: register('agent', 0xec67),
628 > editCode: register('edit-code', 0xec68),
629 > repoSelected: register('repo-selected', 0xec69),
630 > skip: register('skip', 0xec6a),
631 > mergeInto: register('merge-into', 0xec6b),
632 > gitBranchChanges: register('git-branch-changes', 0xec6c),
633 > gitBranchStagedChanges: register('git-branch-staged-changes', 0xec6d),
634 > gitBranchConflicts: register('git-branch-conflicts', 0xec6e),
635 > gitBranch: register('git-branch', 0xec6f),
636 > gitBranchCreate: register('git-branch-create', 0xec6f),
637 > gitBranchDelete: register('git-branch-delete', 0xec6f),
638 > searchLarge: register('search-large', 0xec70),
639 > terminalGitBash: register('terminal-git-bash', 0xec71),
640 > windowActive: register('window-active', 0xec72),
641 > forward: register('forward', 0xec73),
642 > download: register('download', 0xec74),
643 > clockface: register('clockface', 0xec75),
644 > unarchive: register('unarchive', 0xec76),
645 > sessionInProgress: register('session-in-progress', 0xec77),
646 > collectionSmall: register('collection-small', 0xec78),
647 > vmSmall: register('vm-small', 0xec79),
648 > cloudSmall: register('cloud-small', 0xec7a),
649 > addSmall: register('add-small', 0xec7b),
650 > removeSmall: register('remove-small', 0xec7c),
651 > worktreeSmall: register('worktree-small', 0xec7d),
652 > worktree: register('worktree', 0xec7e),
653 > screenCut: register('screen-cut', 0xec7f),
654 > ask: register('ask', 0xec80),
655 > openai: register('openai', 0xec81),
656 > claude: register('claude', 0xec82),
657 > openInWindow: register('open-in-window', 0xec83),
658 > newSession: register('new-session', 0xec84),
659 > terminalSecure: register('terminal-secure', 0xec85),
660 > chatImport: register('chat-import', 0xec86),
661 > chatExport: register('chat-export', 0xec87),
662 > shareWindow: register('share-window', 0xec88),
663 > circleSlashCompact: register('circle-slash-compact', 0xec89),
664 > copilotCompact: register('copilot-compact', 0xec8a),
665 > folderOpenedCompact: register('folder-opened-compact', 0xec8b),
666 > folderCompact: register('folder-compact', 0xec8c),
667 > gearCompact: register('gear-compact', 0xec8d),
668 > gitBranchCompact: register('git-branch-compact', 0xec8e),
669 > libraryCompact: register('library-compact', 0xec8f),
670 > recordKeysCompact: register('record-keys-compact', 0xec90),
671 > remoteCompact: register('remote-compact', 0xec91),
672 > repoForkedCompact: register('repo-forked-compact', 0xec92),
673 > repoCompact: register('repo-compact', 0xec93),
674 > shieldCompact: register('shield-compact', 0xec94),
675 > sparkleCompact: register('sparkle-compact', 0xec95),
676 > symbolColorCompact: register('symbol-color-compact', 0xec96),
677 > windowCompact: register('window-compact', 0xec97),
678 > errorCompact: register('error-compact', 0xec98),
679 > warningCompact: register('warning-compact', 0xec99),
680 > passCompact: register('pass-compact', 0xec9a),
681 > important: register('important', 0xec9b),
682 > importantCompact: register('important-compact', 0xec9c),
683 > rocketCompact: register('rocket-compact', 0xec9d),
684 > unpin: register('unpin', 0xec9e),
685 > addCompact: register('add-compact', 0xec9f),
686 > attachCompact: register('attach-compact', 0xeca0),
687 > beakerCompact: register('beaker-compact', 0xeca1),
688 > checkCompact: register('check-compact', 0xeca2),
689 > checklistCompact: register('checklist-compact', 0xeca3),
690 > chevronDownCompact: register('chevron-down-compact', 0xeca4),
691 > chevronLeftCompact: register('chevron-left-compact', 0xeca5),
692 > chevronRightCompact: register('chevron-right-compact', 0xeca6),
693 > chevronUpCompact: register('chevron-up-compact', 0xeca7),
694 > circleFilledCompact: register('circle-filled-compact', 0xeca8),
695 > circleSmallFilledCompact: register('circle-small-filled-compact', 0xeca9),
696 > closeCompact: register('close-compact', 0xecaa),
697 > collapseAllCompact: register('collapse-all-compact', 0xecab),
698 > commentCompact: register('comment-compact', 0xecac),
699 > commentUnresolvedCompact: register('comment-unresolved-compact', 0xecad),
700 > debugConnectedCompact: register('debug-connected-compact', 0xecae),
701 > debugDisconnectCompact: register('debug-disconnect-compact', 0xecaf),
702 > editCompact: register('edit-compact', 0xecb0),
703 > fileMediaCompact: register('file-media-compact', 0xecb1),
704 > gitFetch: register('git-fetch', 0xecb2),
705 > lightbulbCompact: register('lightbulb-compact', 0xecb3),
706 > loadingCompact: register('loading-compact', 0xecb4),
707 > passFilledCompact: register('pass-filled-compact', 0xecb5),
708 > projectCompact: register('project-compact', 0xecb6),
709 > refreshCompact: register('refresh-compact', 0xecb7),
710 > searchCompact: register('search-compact', 0xecb8),
711 > sessionInProgressCompact: register('session-in-progress-compact', 0xecb9),
712 > syncCompact: register('sync-compact', 0xecba),
713 > terminalCompact: register('terminal-compact', 0xecbb),
714 > vmPending: register('vm-pending', 0xecbc),
715 > worktreeCompact: register('worktree-compact', 0xecbd),
716 > developerTools: register('developer-tools', 0xecbe),
717 > cloudCompact: register('cloud-compact', 0xecbf),
718 > agentCompact: register('agent-compact', 0xecc0),
719 > askCompact: register('ask-compact', 0xecc1),
720 > settingsCompact: register('settings-compact', 0xecc2),
721 > vmCompact: register('vm-compact', 0xecc3),
722 > runCompact: register('run-compact', 0xecc4),
723 > gitPullRequestComment: register('git-pull-request-comment', 0xecc5),
724 > gitPullRequestError: register('git-pull-request-error', 0xecc6),
725 > rightPanelHide: register('right-panel-hide', 0xecc7),
726 > rightPanelShow: register('right-panel-show', 0xecc8),
727 > vscodeInsidersOutline: register('vscode-insiders-outline', 0xecc9),
728 > vscodeOutline: register('vscode-outline', 0xecca),
729 > voiceMode: register('voice-mode', 0xeccb),
730 > voiceModeCompact: register('voice-mode-compact', 0xeccc),
731 > micDownload: register('mic-download', 0xeccd),
732 > micDownloadCompact: register('mic-download-compact', 0xecce),
733 > voiceModeDownload: register('voice-mode-download', 0xeccf),
734 > voiceModeDownloadCompact: register('voice-mode-download-compact', 0xecd0),
735 > googleGemini: register('google-gemini', 0xecd1),
736 > kimi: register('kimi', 0xecd2),
737 > microsoft: register('microsoft', 0xecd3),
738 > fish1Happy: register('fish1-happy', 0xecd4),
739 > fish1Neutral: register('fish1-neutral', 0xecd5),
740 > fish1Sad: register('fish1-sad', 0xecd6),
741 > fish1VerySad: register('fish1-very-sad', 0xecd7),
742 > fish2Happy: register('fish2-happy', 0xecd8),
743 > fish2Neutral: register('fish2-neutral', 0xecd9),
744 > fish2Sad: register('fish2-sad', 0xecda),
745 > fish2VerySad: register('fish2-very-sad', 0xecdb),
746 > fish3Happy: register('fish3-happy', 0xecdc),
747 > fish3Neutral: register('fish3-neutral', 0xecdd),
748 > fish3Sad: register('fish3-sad', 0xecde),
749 > fish3VerySad: register('fish3-very-sad', 0xecdf),
750 > fish4Happy: register('fish4-happy', 0xece0),
751 > fish4Neutral: register('fish4-neutral', 0xece1),
752 > fish4Sad: register('fish4-sad', 0xece2),
753 > fish4VerySad: register('fish4-very-sad', 0xece3),
754 > personVoice: register('person-voice', 0xece4),
755 > personVoiceCompact: register('person-voice-compact', 0xece5),
756 > personVoiceFilled: register('person-voice-filled', 0xece6),
757 > personVoiceFilledCompact: register('person-voice-filled-compact', 0xece7),
758 > } as const;
src/vs/workbench/contrib/chat/common/promptSyntax/service/promptsService.ts 733 covered LOC · 9 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../../base/common/cancellation.js';
7 > import { Event } from '../../../../../../base/common/event.js';
8 > import { IDisposable } from '../../../../../../base/common/lifecycle.js';
9 > import { URI } from '../../../../../../base/common/uri.js';
10 > import { ITextModel } from '../../../../../../editor/common/model.js';
11 > import { ExtensionIdentifier, IExtensionDescription } from '../../../../../../platform/extensions/common/extensions.js';
12 > import { createDecorator } from '../../../../../../platform/instantiation/common/instantiation.js';
13 > import { IChatModeInstructions, IVariableReference } from '../../chatModes.js';
14 > import { PromptFileSource, PromptsType, Target } from '../promptTypes.js';
15 > import { IHandOff, ParsedPromptFile } from '../promptFileParser.js';
16 > import { ResourceSet } from '../../../../../../base/common/map.js';
17 > import { IResolvedPromptSourceFolder } from '../config/promptFileLocations.js';
18 > import { ChatRequestHooks } from '../hookSchema.js';
19 > import { isEqual } from '../../../../../../base/common/resources.js';
20 >
21 > /**
22 > * A single structured debug detail entry from the instructions context computer.
23 > */
24 > export interface InstructionsCollectionDebugEntry {
25 > readonly category: 'applying' | 'skipped' | 'referenced' | 'skill' | 'custom-agent' | 'hook';
26 > readonly name: string;
27 > readonly uri?: URI;
28 > readonly reason?: string;
29 > }
30 >
31 > export type InstructionsCollectionEvent = {
32 > applyingInstructionsCount: number;
33 > referencedInstructionsCount: number;
34 > agentInstructionsCount: number;
35 > listedInstructionsCount: number;
36 > totalInstructionsCount: number;
37 > claudeRulesCount: number;
38 > claudeMdCount: number;
39 > claudeAgentsCount: number;
40 > };
41 >
42 > /**
43 > * Debug-only information collected alongside {@link InstructionsCollectionEvent}.
44 > * This data is used for debug logging and is not sent as telemetry.
45 > */
46 > export type InstructionsCollectionDebugInfo = {
47 > /** Per-file detail entries for debug logging. */
48 > debugDetails: InstructionsCollectionDebugEntry[];
49 > /** Total wall-clock time of the collect() call in milliseconds. */
50 > durationInMillis: number;
51 > };
52 >
53 > export function newInstructionsCollectionEvent(): InstructionsCollectionEvent {
54 return { applyingInstructionsCount: 0, referencedInstructionsCount: 0, agentInstructionsCount: 0, listedInstructionsCount: 0, totalInstructionsCount: 0, claudeRulesCount: 0, claudeMdCount: 0, claudeAgentsCount: 0 };
55 }
57 > export function newInstructionsCollectionDebugInfo(): InstructionsCollectionDebugInfo {
58 return { debugDetails: [], durationInMillis: 0 };
59 }
61 > /**
62 > * Activation events for prompt file providers.
63 > */
64 > export const CUSTOM_AGENT_PROVIDER_ACTIVATION_EVENT = 'onCustomAgentProvider';
65 > export const INSTRUCTIONS_PROVIDER_ACTIVATION_EVENT = 'onInstructionsProvider';
66 > export const PROMPT_FILE_PROVIDER_ACTIVATION_EVENT = 'onPromptFileProvider';
67 > export const SKILL_PROVIDER_ACTIVATION_EVENT = 'onSkillProvider';
68 >
69 > /**
70 > * Context for querying prompt files.
71 > */
72 > export interface IPromptFileContext { }
73 >
74 > /**
75 > * Represents a prompt file resource from an external provider.
76 > */
77 > export interface IPromptFileResource {
78 > /**
79 > * The URI to the agent or prompt resource file.
80 > */
81 > readonly uri: URI;
82 > /**
83 > * Optional externally provided prompt command name.
84 > */
85 > readonly name?: string;
86 > /**
87 > * Optional externally provided prompt command description.
88 > */
89 > readonly description?: string;
90 > /**
91 > * Optional condition that must evaluate to true for this resource to be offered.
92 > */
93 > readonly when?: string;
94 > /**
95 > * Optional session types that describe when this resource should be offered.
96 > */
97 > readonly sessionTypes?: readonly string[];
98 > }
99 >
100 > /**
101 > * Returns whether a customization can be used in the provided chat session type.
102 > */
103 > export function matchesSessionType(sessionTypes: readonly string[] | undefined, currentSessionType: string | undefined): boolean {
104 return sessionTypes === undefined || currentSessionType === undefined || sessionTypes.includes(currentSessionType);
105 }
107 > /**
108 > * Provides prompt services.
109 > */
110 > export const IPromptsService = createDecorator<IPromptsService>('IPromptsService');
111 >
112 > /**
113 > * Where the prompt is stored.
114 > */
115 > export enum PromptsStorage {
116 > local = 'local',
117 > user = 'user',
118 > extension = 'extension',
119 > plugin = 'plugin',
120 > builtIn = 'builtin',
121 > }
122 >
123 > /**
124 > * Represents a prompt path with its type.
125 > * This is used for both prompt files and prompt source folders.
126 > */
127 > export type IPromptPath = IExtensionPromptPath | ILocalPromptPath | IUserPromptPath | IPluginPromptPath | IBuiltinPromptPath;
128 >
129 >
130 > export interface IPromptPathBase {
131 > /**
132 > * URI of the prompt.
133 > */
134 > readonly uri: URI;
135 >
136 > /**
137 > * Storage of the prompt.
138 > */
139 > readonly storage: PromptsStorage;
140 >
141 > /**
142 > * Type of the prompt (e.g. 'prompt' or 'instructions').
143 > */
144 > readonly type: PromptsType;
145 >
146 > /**
147 > * Identifier of the contributing extension (only when storage === PromptsStorage.extension).
148 > */
149 > readonly extension?: IExtensionDescription;
150 >
151 > /**
152 > * Identifier of the contributing plugin (only when storage === PromptsStorage.plugin).
153 > */
154 > readonly pluginUri?: URI;
155 >
156 > /**
157 > * Human-readable name of the contributing plugin, used for plugin-scoped slash command names.
158 > */
159 > readonly pluginLabel?: string;
160 >
161 > /**
162 > * The source that produced this prompt path.
163 > */
164 > readonly source?: PromptFileSource;
165 >
166 > readonly name?: string;
167 >
168 > readonly description?: string;
169 >
170 > /**
171 > * Optional session types that describe when this resource should be offered.
172 > */
173 > readonly sessionTypes?: readonly string[];
174 > }
175 >
176 > export interface IExtensionPromptPath extends IPromptPathBase {
177 > readonly storage: PromptsStorage.extension;
178 > readonly extension: IExtensionDescription;
179 > readonly source: PromptFileSource.ExtensionContribution | PromptFileSource.ExtensionAPI;
180 > readonly name?: string;
181 > readonly description?: string;
182 > readonly when?: string;
183 > }
184 >
185 > export function isExtensionPromptPath(obj: IPromptPath): obj is IExtensionPromptPath {
186 return obj.storage === PromptsStorage.extension;
187 }
189 > export interface ILocalPromptPath extends IPromptPathBase {
190 > readonly storage: PromptsStorage.local;
191 > }
192 > export interface IUserPromptPath extends IPromptPathBase {
193 > readonly storage: PromptsStorage.user;
194 > }
195 >
196 > export interface IPluginPromptPath extends IPromptPathBase {
197 > readonly storage: PromptsStorage.plugin;
198 > readonly pluginUri: URI;
199 > readonly source: PromptFileSource.Plugin;
200 > }
201 >
202 > /**
203 > * Prompt path for built-in prompts bundled with the application (e.g. skills
204 > * shipped with the Agents app). These are read-only and provided by
205 > * {@link IPromptsService.listPromptFiles}/`listPromptFilesForStorage`.
206 > */
207 > export interface IBuiltinPromptPath extends IPromptPathBase {
208 > readonly storage: PromptsStorage.builtIn;
209 > }
210 >
211 > export function isBuiltinPromptPath(obj: IPromptPath): obj is IBuiltinPromptPath {
212 return obj.storage === PromptsStorage.builtIn;
213 }
215 > export type IAgentSource = {
216 > readonly storage: PromptsStorage.extension;
217 > readonly extensionId: ExtensionIdentifier;
218 > } | {
219 > readonly storage: PromptsStorage.local | PromptsStorage.user | PromptsStorage.builtIn;
220 > } | {
221 > readonly storage: PromptsStorage.plugin;
222 > readonly pluginUri: URI;
223 > };
224 >
225 > export namespace IAgentSource {
226 > export function fromPromptPath(promptPath: IPromptPath): IAgentSource {
227 if (promptPath.storage === PromptsStorage.extension) {
228 return { storage: PromptsStorage.extension, extensionId: promptPath.extension.identifier };
233 }
234 }
236 > export function isEquals(a: IAgentSource | undefined, b: IAgentSource | undefined): boolean {
237 if (a === b) {
238 return true;
251 return true;
252 }
254 >
255 > /**
256 > * The visibility/availability of an agent.
257 > * - 'all': available as custom agent in picker AND can be used as subagent
258 > * - 'user': only available in the custom agent picker
259 > * - 'agent': only usable as subagent by the subagent tool
260 > * - 'hidden': neither in picker nor usable as subagent
261 > */
262 > export type ICustomAgentVisibility = {
263 > readonly userInvocable: boolean;
264 > readonly agentInvocable: boolean;
265 > };
266 >
267 > export function isCustomAgentVisibility(obj: unknown): obj is ICustomAgentVisibility {
268 if (typeof obj !== 'object' || obj === null) {
269 return false;
272 return typeof v.userInvocable === 'boolean' && typeof v.agentInvocable === 'boolean';
273 }
275 > export interface ICustomAgent {
276 >
277 > readonly id: string;
278 > /**
279 > * URI of a custom agent file.
280 > */
281 > readonly uri: URI;
282 >
283 > /**
284 > * Name of the custom agent as used in prompt files or contexts
285 > */
286 > readonly name: string;
287 >
288 > /**
289 > * Description of the agent
290 > */
291 > readonly description?: string;
292 >
293 > /**
294 > * Tools metadata in the prompt header.
295 > */
296 > readonly tools?: readonly string[];
297 >
298 > /**
299 > * Model metadata in the prompt header.
300 > */
301 > readonly model?: readonly string[];
302 >
303 > /**
304 > * Argument hint metadata in the prompt header that describes what inputs the agent expects or supports.
305 > */
306 > readonly argumentHint?: string;
307 >
308 > /**
309 > * Target of the agent: Copilot, VSCode, Claude, or undefined if not specified.
310 > */
311 > readonly target: Target;
312 >
313 > /**
314 > * What visibility the agent has (user invocable, subagent invocable).
315 > */
316 > readonly visibility: ICustomAgentVisibility;
317 >
318 > /**
319 > * Contents of the custom agent file body and other agent instructions.
320 > */
321 > readonly agentInstructions: IChatModeInstructions;
322 >
323 > /**
324 > * Hand-offs defined in the custom agent file.
325 > */
326 > readonly handOffs?: readonly IHandOff[];
327 >
328 > /**
329 > * List of subagent names that can be used by the agent.
330 > * If empty, no subagents are available. If ['*'] or undefined, all agents can be used.
331 > */
332 > readonly agents?: readonly string[];
333 >
334 > /**
335 > * Lifecycle hooks scoped to this subagent.
336 > */
337 > readonly hooks?: ChatRequestHooks;
338 >
339 > /**
340 > * Where the agent was loaded from.
341 > */
342 > readonly source: IAgentSource;
343 >
344 > /**
345 > * Optional session types that describe when this agent should be offered.
346 > */
347 > readonly sessionTypes?: readonly string[];
348 >
349 > /**
350 > * Whether this agent is enabled. Disabled agents are included in the list
351 > * but should not be offered to users or used in automated flows.
352 > */
353 > readonly enabled: boolean;
354 > }
355 >
356 > export interface IAgentInstructions {
357 > readonly content: string;
358 > readonly toolReferences: readonly IVariableReference[];
359 > readonly metadata?: Record<string, boolean | string | number>;
360 > }
361 >
362 > export interface IChatPromptSlashCommand {
363 > readonly uri: URI;
364 > readonly name: string;
365 > readonly type: PromptsType;
366 > readonly storage: PromptsStorage;
367 > readonly source?: PromptFileSource;
368 > readonly description?: string;
369 > readonly argumentHint?: string;
370 > readonly userInvocable: boolean;
371 > readonly extension?: IExtensionDescription;
372 > readonly pluginUri?: URI;
373 > readonly pluginLabel?: string;
374 > /**
375 > * Optional session types that describe when this slash command should be offered.
376 > */
377 > readonly sessionTypes?: readonly string[];
378 > }
379 >
380 > export interface IResolvedChatPromptSlashCommand extends IChatPromptSlashCommand {
381 > readonly parsedPromptFile: ParsedPromptFile;
382 > }
383 >
384 >
385 > /**
386 > * A fully resolved instruction file with parsed header metadata and provenance information.
387 > */
388 > export interface IInstructionFile {
389 > /**
390 > * URI of the instruction file.
391 > */
392 > readonly uri: URI;
393 > /**
394 > * Name as listed in the instruction file header or derived from the file name
395 > */
396 > readonly name: string;
397 > /**
398 > * Description as listed in the instruction file header. Used to load the instruction on-demand and for display in the UI.
399 > */
400 > readonly description: string | undefined;
401 > /**
402 > * Storage of the prompt.
403 > */
404 > readonly storage: PromptsStorage;
405 > /**
406 > * The "applyTo" pattern (or `paths` when in a Claude rules file) from the instruction file header.
407 > * Describes when this instruction file should be applied.
408 > */
409 > readonly pattern: string | undefined;
410 > /**
411 > * Identifier of the contributing extension (only when storage === PromptsStorage.extension).
412 > */
413 > readonly extension?: IExtensionDescription;
414 >
415 > /**
416 > * Identifier of the contributing plugin (only when storage === PromptsStorage.plugin).
417 > */
418 > readonly pluginUri?: URI;
419 >
420 > /**
421 > * The source that produced this prompt path.
422 > */
423 > readonly source?: PromptFileSource;
424 >
425 > /**
426 > * Optional session types that describe when this instruction should be offered.
427 > */
428 > readonly sessionTypes?: readonly string[];
429 > }
430 >
431 > /**
432 > * Supply-chain metadata describing where a skill originated.
433 > */
434 > export interface IAgentSkill {
435 > readonly uri: URI;
436 > readonly storage: PromptsStorage;
437 > readonly name: string;
438 > readonly description: string | undefined;
439 > /**
440 > * If true, the skill should not be automatically loaded by the agent.
441 > * Use for workflows you want to trigger manually with /name.
442 > */
443 > readonly disableModelInvocation: boolean;
444 > /**
445 > * If false, the skill is hidden from the / menu.
446 > * Use for background knowledge users shouldn't invoke directly.
447 > */
448 > readonly userInvocable: boolean;
449 > /**
450 > * Optional plugin URI describing where this skill originated.
451 > */
452 > readonly pluginUri?: URI;
453 > /**
454 > * Optional plugin display name describing where this skill originated.
455 > */
456 > readonly pluginLabel?: string;
457 > /**
458 > * Optional extension metadata describing where this skill originated.
459 > */
460 > readonly extension?: IExtensionDescription;
461 > /**
462 > * Optional session types that describe when this skill should be offered.
463 > */
464 > readonly sessionTypes?: readonly string[];
465 > }
466 >
467 > /**
468 > * Type of agent instruction file.
469 > */
470 > export enum AgentInstructionFileType {
471 > agentsMd = 'agentsMd',
472 > claudeMd = 'claudeMd',
473 > copilotInstructionsMd = 'copilotInstructionsMd',
474 > }
475 >
476 > /**
477 > * Represents a resolved agent instruction file with its real path for duplicate detection.
478 > * Used by listAgentInstructions to filter out symlinks pointing to the same file.
479 > */
480 > export interface IAgentInstructionFile {
481 > readonly uri: URI;
482 > /**
483 > * The real path of the file, if it is a symlink.
484 > */
485 > readonly realPath: URI | undefined;
486 > readonly type: AgentInstructionFileType;
487 > }
488 >
489 > export interface Logger {
490 > logInfo(message: string): void;
491 > }
492 >
493 > /**
494 > * Reason why a prompt file was skipped during discovery.
495 > */
496 > export type PromptFileSkipReason =
497 > | 'missing-name'
498 > | 'missing-description'
499 > | 'name-mismatch'
500 > | 'duplicate-name'
501 > | 'parse-error'
502 > | 'disabled'
503 > | 'all-hooks-disabled'
504 > | 'claude-hooks-disabled'
505 > | 'workspace-untrusted';
506 >
507 > /**
508 > * Result of discovering a single prompt file.
509 > */
510 > export interface IPromptFileDiscoveryResult {
511 > readonly status: 'loaded' | 'skipped';
512 > readonly skipReason?: PromptFileSkipReason;
513 > /** Error message if parse-error */
514 > readonly errorMessage?: string;
515 > /** For duplicates, the URI of the file that took precedence */
516 > readonly duplicateOf?: URI;
517 > /** Prompt path for the discovered file. */
518 > readonly promptPath: IPromptPath;
519 > /** Whether the skill is user-invocable in the / menu (set user-invocable: false to hide it) */
520 > readonly userInvocable?: boolean;
521 > /** If true, the skill won't be automatically loaded by the agent (disable-model-invocation: true) */
522 > readonly disableModelInvocation?: boolean;
523 > }
524 >
525 > /**
526 > * Diagnostic information about a source folder that was searched during discovery.
527 > */
528 > export interface IPromptSourceFolderResult {
529 > readonly uri: URI;
530 > readonly storage: PromptsStorage;
531 > }
532 >
533 > /**
534 > * Summary of prompt file discovery for a specific type.
535 > */
536 > export interface IPromptDiscoveryInfo {
537 > readonly type: PromptsType;
538 > readonly files: readonly IPromptFileDiscoveryResult[];
539 > /** Time in milliseconds required to compute this discovery result. */
540 > readonly durationInMillis: number;
541 > /** Source folders that were searched */
542 > readonly sourceFolders?: readonly IPromptSourceFolderResult[];
543 > }
544 >
545 > /**
546 > * Discovery result for a slash command file, including the parsed prompt file.
547 > */
548 > export interface ISlashCommandDiscoveryResult extends IPromptFileDiscoveryResult {
549 > readonly userInvocable?: boolean;
550 > readonly argumentHint?: string;
551 > }
552 >
553 > /**
554 > * Summary of slash command discovery, including parsed prompt files.
555 > */
556 > export interface ISlashCommandDiscoveryInfo extends IPromptDiscoveryInfo {
557 > readonly files: readonly ISlashCommandDiscoveryResult[];
558 > }
559 >
560 > /**
561 > * Discovery result for an instruction file, including the resolved applyTo metadata.
562 > */
563 > export interface IInstructionDiscoveryResult extends IPromptFileDiscoveryResult {
564 > readonly pattern?: string;
565 > }
566 >
567 > /**
568 > * Summary of instruction discovery, including resolved metadata.
569 > */
570 > export interface IInstructionDiscoveryInfo extends IPromptDiscoveryInfo {
571 > readonly files: readonly IInstructionDiscoveryResult[];
572 > }
573 >
574 > /**
575 > * Discovery result for an agent file, including the fully resolved agent.
576 > */
577 > export interface IAgentDiscoveryResult extends IPromptFileDiscoveryResult {
578 > readonly agent?: ICustomAgent;
579 > }
580 >
581 > /**
582 > * Summary of agent discovery, including resolved agents.
583 > */
584 > export interface IAgentDiscoveryInfo extends IPromptDiscoveryInfo {
585 > readonly files: readonly IAgentDiscoveryResult[];
586 > }
587 >
588 > export interface IConfiguredHooksInfo {
589 > readonly hooks: ChatRequestHooks;
590 > readonly hasDisabledClaudeHooks: boolean;
591 > }
592 >
593 > /**
594 > * Summary of hook discovery, including the resolved hooks info.
595 > */
596 > export interface IHookDiscoveryInfo extends IPromptDiscoveryInfo {
597 > readonly hooksInfo: IConfiguredHooksInfo | undefined;
598 > }
599 >
600 > /**
601 > * Provides prompt services.
602 > */
603 > export interface IPromptsService extends IDisposable {
604 > readonly _serviceBrand: undefined;
605 >
606 > /**
607 > * The parsed prompt file for the provided text model.
608 > * @param textModel Returns the parsed prompt file.
609 > */
610 > getParsedPromptFile(textModel: ITextModel): ParsedPromptFile;
611 >
612 > /**
613 > * List all available prompt files.
614 > */
615 > listPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IPromptPath[]>;
616 >
617 > /**
618 > * List all available prompt files.
619 > */
620 > listPromptFilesForStorage(type: PromptsType, storage: PromptsStorage, token: CancellationToken): Promise<readonly IPromptPath[]>;
621 >
622 > /**
623 > * Get a list of prompt source folders based on the provided prompt type.
624 > */
625 > getSourceFolders(type: PromptsType): Promise<readonly IPromptPath[]>;
626 >
627 > /**
628 > * Get a list of resolved prompt source folders with full metadata.
629 > * This includes displayPath, isDefault, and storage information.
630 > * Used for diagnostics and config-info displays.
631 > */
632 > getResolvedSourceFolders(type: PromptsType): Promise<readonly IResolvedPromptSourceFolder[]>;
633 >
634 > /**
635 > * Validates if the provided command name is a valid prompt slash command.
636 > */
637 > isValidSlashCommandName(name: string): boolean;
638 >
639 > /**
640 > * Synchronously checks whether `name` matches a discovered prompt slash command.
641 > * Backed by a cache that is populated lazily on the first call and refreshed on
642 > * subsequent {@link onDidChangeSlashCommands} firings, so the very first call after
643 > * service creation may return `false` for known commands until the first discovery
644 > * completes.
645 > */
646 > hasPromptSlashCommand(name: string): boolean;
647 >
648 > /**
649 > * Gets the prompt file for a slash command.
650 > */
651 > resolvePromptSlashCommand(command: string, sessionType: string | undefined, token: CancellationToken): Promise<IResolvedChatPromptSlashCommand | undefined>;
652 >
653 > /**
654 > * Event that is triggered when the slash command to ParsedPromptFile cache is updated.
655 > * Event handlers can use {@link resolvePromptSlashCommand} to retrieve the latest data.
656 > */
657 > readonly onDidChangeSlashCommands: Event<void>;
658 >
659 > /**
660 > * Returns a prompt command if the command name is valid.
661 > */
662 > getPromptSlashCommands(token: CancellationToken): Promise<readonly IChatPromptSlashCommand[]>;
663 >
664 > /**
665 > * Returns the prompt command name for the given URI.
666 > */
667 > getPromptSlashCommandName(uri: URI, token: CancellationToken): Promise<string>;
668 >
669 > /**
670 > * Event that is triggered when the list of custom agents changes.
671 > */
672 > readonly onDidChangeCustomAgents: Event<void>;
673 >
674 > /**
675 > * Event that is triggered when the list of instruction files changes.
676 > */
677 > readonly onDidChangeInstructions: Event<void>;
678 >
679 > /**
680 > * Event that is triggered when the list of agent instruction files changes.
681 > */
682 > readonly onDidChangeAgentInstructions: Event<void>;
683 >
684 > /**
685 > * Finds all available custom agents
686 > */
687 > getCustomAgents(token: CancellationToken): Promise<readonly ICustomAgent[]>;
688 >
689 > /**
690 > * Parses the provided URI
691 > * @param uris
692 > */
693 > parseNew(uri: URI, token: CancellationToken): Promise<ParsedPromptFile>;
694 >
695 > /**
696 > * Internal: register a contributed file. Returns a disposable that removes the contribution.
697 > * Not intended for extension authors; used by contribution point handler.
698 > */
699 > registerContributedFile(type: PromptsType, uri: URI, extension: IExtensionDescription, name: string | undefined, description: string | undefined, when?: string, sessionTypes?: readonly string[]): IDisposable;
700 >
701 >
702 > getPromptLocationLabel(promptPath: IPromptPath): string;
703 >
704 > /**
705 > * Gets list of AGENTS.md files, including optionally nested ones from subfolders.
706 > */
707 > listNestedAgentMDs(token: CancellationToken): Promise<IAgentInstructionFile[]>;
708 >
709 > /**
710 > * Gets combined list of agent instruction files (AGENTS.md, CLAUDE.md, copilot-instructions.md).
711 > * Combines results from listAgentMDs (non-nested), listClaudeMDs, and listCopilotInstructionsMDs.
712 > */
713 > listAgentInstructions(token: CancellationToken, logger?: Logger): Promise<IAgentInstructionFile[]>;
714 >
715 > /**
716 > * For a chat mode file URI, return the name of the agent file that it should use.
717 > * @param oldURI
718 > */
719 > getAgentFileURIFromModeFile(oldURI: URI): URI | undefined;
720 >
721 > /**
722 > * Returns the list of disabled prompt file URIs for a given type. By default no prompt files are disabled.
723 > */
724 > getDisabledPromptFiles(type: PromptsType): ResourceSet;
725 >
726 > /**
727 > * Persists the set of disabled prompt file URIs for the given type.
728 > */
729 > setDisabledPromptFiles(type: PromptsType, uris: ResourceSet): void;
730 >
731 > /**
732 > * Registers a prompt file provider that can provide prompt files for repositories.
733 > * @param extension The extension registering the provider.
734 > * @param type The type of contribution.
735 > * @param provider The provider implementation with optional change event.
736 > * @returns A disposable that unregisters the provider when disposed.
737 > */
738 > registerPromptFileProvider(extension: IExtensionDescription, type: PromptsType, provider: {
739 > onDidChangePromptFiles?: Event<void>;
740 > providePromptFiles: (context: IPromptFileContext, token: CancellationToken) => Promise<IPromptFileResource[] | undefined>;
741 > }): IDisposable;
742 >
743 > /**
744 > * Gets list of agent skills files.
745 > */
746 > findAgentSkills(token: CancellationToken): Promise<IAgentSkill[] | undefined>;
747 >
748 > /**
749 > * Event that is triggered when the list of skills changes.
750 > */
751 > readonly onDidChangeSkills: Event<void>;
752 >
753 > /**
754 > * Event that is triggered when the effective hook availability or configuration changes.
755 > */
756 > readonly onDidChangeHooks: Event<void>;
757 >
758 > /**
759 > * Gets all hooks collected from hooks.json files.
760 > * The result is cached and invalidated when the effective hook availability or configuration changes.
761 > */
762 > getHooks(token: CancellationToken): Promise<IConfiguredHooksInfo | undefined>;
763 >
764 > /**
765 > * Gets all instruction files
766 > */
767 > getInstructionFiles(token: CancellationToken): Promise<readonly IInstructionFile[]>;
768 >
769 > /**
770 > * Returns the cached discovery info for the given prompt type.
771 > */
772 > getDiscoveryInfo(type: PromptsType, token: CancellationToken): Promise<IPromptDiscoveryInfo>;
773 > }
src/vs/workbench/contrib/chat/common/chatService/chatServiceImpl.ts 662 covered LOC · 151 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatServiceImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DeferredPromise, raceCancellationError, raceTimeout } from '../../../../../base/common/async.js';
7 > import { CancellationToken, CancellationTokenSource } from '../../../../../base/common/cancellation.js';
8 > import { IStringDictionary } from '../../../../../base/common/collections.js';
9 > import { toErrorMessage } from '../../../../../base/common/errorMessage.js';
10 > import { BugIndicatingError, ErrorNoTelemetry } from '../../../../../base/common/errors.js';
11 > import { Emitter, Event } from '../../../../../base/common/event.js';
12 > import { MarkdownString } from '../../../../../base/common/htmlContent.js';
13 > import { Iterable } from '../../../../../base/common/iterator.js';
14 > import { Disposable, DisposableResourceMap, DisposableStore, IDisposable, MutableDisposable } from '../../../../../base/common/lifecycle.js';
15 > import { ResourceMap } from '../../../../../base/common/map.js';
16 > import { revive } from '../../../../../base/common/marshalling.js';
17 > import { equals } from '../../../../../base/common/objects.js';
18 > import { autorun, derived, IObservable, ISettableObservable, observableValue } from '../../../../../base/common/observable.js';
19 > import { isEqual } from '../../../../../base/common/resources.js';
20 > import { StopWatch } from '../../../../../base/common/stopwatch.js';
21 > import { isDefined } from '../../../../../base/common/types.js';
22 > import { URI } from '../../../../../base/common/uri.js';
23 > import { generateUuid } from '../../../../../base/common/uuid.js';
24 > import { OffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
25 > import { localize } from '../../../../../nls.js';
26 > import { IConfigurationService } from '../../../../../platform/configuration/common/configuration.js';
27 > import { IInstantiationService } from '../../../../../platform/instantiation/common/instantiation.js';
28 > import { ILogService } from '../../../../../platform/log/common/log.js';
29 > import { Progress } from '../../../../../platform/progress/common/progress.js';
30 > import { IStorageService, StorageScope, StorageTarget } from '../../../../../platform/storage/common/storage.js';
31 > import { ITelemetryService } from '../../../../../platform/telemetry/common/telemetry.js';
32 > import { IWorkspaceContextService } from '../../../../../platform/workspace/common/workspace.js';
33 > import { IExtensionService } from '../../../../services/extensions/common/extensions.js';
34 > import { IChatEntitlementService } from '../../../../services/chat/common/chatEntitlementService.js';
35 > import { IChatDebugService } from '../chatDebugService.js';
36 > import { IMcpService } from '../../../mcp/common/mcpTypes.js';
37 > import { awaitStatsForSession } from '../chat.js';
38 > import { ChatPerfMark, clearChatMarks, markChat } from '../chatPerf.js';
39 > import { IChatAgentAttachmentCapabilities, IChatAgentCommand, IChatAgentData, IChatAgentHistoryEntry, IChatAgentRequest, IChatAgentResult, IChatAgentService } from '../participants/chatAgents.js';
40 > import { chatEditingSessionIsReady } from '../editing/chatEditingService.js';
41 > import { ChatModel, ChatRequestModel, ChatRequestRemovalReason, IChatModel, IChatPendingRequest, IChatRequestModel, IChatRequestModeInfo, IChatRequestVariableData, IChatResponseModel, IExportableChatData, ISerializableChatData, ISerializableChatDataIn, ISerializableChatsData, ISerializedChatDataReference, normalizeSerializableChatData, toChatHistoryContent, updateRanges, ISerializableChatModelInputState, logChangesToStateModel } from '../model/chatModel.js';
42 > import { ChatModelStore, IStartSessionProps } from '../model/chatModelStore.js';
43 > import { chatAgentLeader, ChatRequestAgentPart, ChatRequestAgentSubcommandPart, ChatRequestSlashCommandPart, ChatRequestTextPart, chatSubcommandLeader, getPromptText, IParsedChatRequest } from '../requestParser/chatParserTypes.js';
44 > import { ChatRequestParser } from '../requestParser/chatRequestParser.js';
45 > import { ChatMcpServersStarting, ChatPendingRequestChangeClassification, ChatPendingRequestChangeEvent, ChatPendingRequestChangeEventName, ChatRequestQueueKind, ChatSendResult, ChatSendResultQueued, ChatSendResultSent, ChatStopCancellationNoopClassification, ChatStopCancellationNoopEvent, ChatStopCancellationNoopEventName, IChatCompleteResponse, IChatDetail, IChatFollowup, IChatModelReference, IChatProgress, IChatQuestionAnswers, IChatSendRequestOptions, IChatSendRequestResponseState, IChatService, IChatSessionStartOptions, IChatUserActionEvent, IRemotePendingRequest, ResponseModelState } from './chatService.js';
46 > import { ChatRequestTelemetry, ChatServiceTelemetry } from './chatServiceTelemetry.js';
47 > import { IChatSessionsService, isAgentHostTarget, isTerminalCommandPrompt, localChatSessionType } from '../chatSessionsService.js';
48 > import { ChatSessionStore, IChatSessionEntryMetadata } from '../model/chatSessionStore.js';
49 > import { IChatSlashCommandService } from '../participants/chatSlashCommands.js';
50 > import { IChatTransferService } from '../model/chatTransferService.js';
51 > import { chatSessionResourceToId, getChatSessionType, isUntitledChatSession, LocalChatSessionUri } from '../model/chatUri.js';
52 > import { ChatRequestVariableSet, IChatRequestVariableEntry, isExplicitFileOrImageVariableEntry, isPromptTextVariableEntry } from '../attachments/chatVariableEntries.js';
53 > import { IDynamicVariable } from '../attachments/chatVariables.js';
54 > import { ChatAgentLocation, ChatConfiguration, ChatModeKind } from '../constants.js';
55 > import { ChatMessageRole, IChatMessage, ILanguageModelsService } from '../languageModels.js';
56 > import { ILanguageModelToolsService, ToolAndToolSetEnablementMap } from '../tools/languageModelToolsService.js';
57 > import { ChatSessionOperationLog } from '../model/chatSessionOperationLog.js';
58 > import { IPromptsService } from '../promptSyntax/service/promptsService.js';
59 > import { AGENT_DEBUG_LOG_FILE_LOGGING_ENABLED_SETTING, TROUBLESHOOT_COMMAND_NAME, TROUBLESHOOT_SKILL_PATH, COPILOT_SKILL_URI_SCHEME } from '../promptSyntax/promptTypes.js';
60 > import { ChatRequestHooks, mergeHooks } from '../promptSyntax/hookSchema.js';
61 > import { ComputeAutomaticInstructions } from '../promptSyntax/computeAutomaticInstructions.js';
62 > import { findLast } from '../../../../../base/common/arraysFind.js';
63 > import { ChatMode } from '../chatModes.js';
64 >
65 > const serializedChatKey = 'interactive.sessions';
66 >
67 > /**
68 > * True when the user has typed text or attached non-trivial context to the input
69 > * but not yet sent it. Used to decide whether an external session needs metadata
70 > * persisted on dispose so the draft survives switching sessions.
71 > */
72 > function hasDraftInput(model: ChatModel): boolean { chatServiceImpl.ts
73 > const state = model.inputModel.state.get();
74 > if (!state) {
75 > return false; chatServiceImpl.ts
76 > }
77 if (state.inputText.trim().length > 0) {
78 return true;
80 return state.attachments.length > 0;
81 }
83 > class CancellableRequest implements IDisposable {
84 > private readonly _yieldRequested: ISettableObservable<boolean> = observableValue(this, false);
85 >
86 > get yieldRequested(): IObservable<boolean> {
87 > return this._yieldRequested;
88 > }
89 >
90 >
91 > constructor(
92 public readonly cancellationTokenSource: CancellationTokenSource,
93 public requestId: string | undefined,
96 @ILanguageModelToolsService private readonly toolsService: ILanguageModelToolsService
97 ) { }
99 > dispose() {
100 if (this.requestId) {
101 this.toolsService.cancelToolCallsForRequest(this.requestId);
103 this.cancellationTokenSource.dispose();
104 }
106 > cancel() {
107 if (this.requestId) {
108 this.toolsService.cancelToolCallsForRequest(this.requestId);
111 this.cancellationTokenSource.cancel();
112 }
114 > setYieldRequested(): void {
115 this._yieldRequested.set(true, undefined);
116 }
118 > resetYieldRequested(): void {
119 this._yieldRequested.set(false, undefined);
120 }
122 >
123 > const EMPTY_REFERENCES: ReadonlyArray<IDynamicVariable> = Object.freeze([]);
124 > const EMPTY_TOOL_ENABLEMENT_MAP: ToolAndToolSetEnablementMap = ToolAndToolSetEnablementMap.fromEntries([]);
125 >
126 > /**
127 > * Preserve the picker state from `stateToApply`, only recovering a custom agent mode from
128 > * `savedState` when the applied state fell back to the default Agent.
129 > *
130 > * `stateToApply` is the input state about to be applied to the session being restored (an
131 > * agent-host transferred draft, or the saved draft as a fallback). Its `selectedModel` is the
132 > * authoritative model selection.
133 > * `savedState` is only used for `mode`: prefer its custom agent over the plain default Agent, but
134 > * never override a different explicit mode already present in `stateToApply`.
135 > */
136 > export function backfillRestoredPickerState(
137 > stateToApply: ISerializableChatModelInputState | undefined, chatServiceImpl.ts
138 > savedState: ISerializableChatModelInputState | undefined,
139 > defaultAgentModeId: string,
140 > ): ISerializableChatModelInputState | undefined {
141 > if (!stateToApply || !savedState) {
142 > return stateToApply; chatServiceImpl.ts
143 > }
144 const mode = (stateToApply.mode.id === defaultAgentModeId && savedState.mode.id !== defaultAgentModeId)
145 ? savedState.mode
146 : stateToApply.mode;
147 > if (mode === stateToApply.mode) { chatServiceImpl.ts
148 return stateToApply;
149 }
150 return { ...stateToApply, mode };
151 }
153 > /**
154 > * Recover the selected model on a transferred input state when it was dropped during a cold
155 > * handoff.
156 > *
157 > * At cold restore an agent-host transferred draft can arrive without its `selectedModel` (the live
158 > * model list is not loaded yet, so the model resolved to `undefined`). Fall back to the model
159 > * derived from the session's request history so the picker restores the last-used model instead of
160 > * Auto. The history-derived model carries full metadata (including `targetChatSessionType`), so the
161 > * input part can wait for the model pool and apply it once it loads. An explicit model already
162 > * present on `transferredState` is never overridden.
163 > */
164 > export function backfillTransferredModel(
165 transferredState: ISerializableChatModelInputState | undefined,
166 historyModel: ISerializableChatModelInputState['selectedModel'],
171 return { ...transferredState, selectedModel: historyModel };
172 }
174 > export class ChatService extends Disposable implements IChatService {
175 > declare _serviceBrand: undefined;
176 >
177 > private readonly _sessionModels: ChatModelStore;
178 > private readonly _pendingRequests = this._register(new DisposableResourceMap<CancellableRequest>());
179 > private readonly _queuedRequestDeferreds = new Map<string, DeferredPromise<ChatSendResult>>();
180 > /** Pending requests that are synthetic streamed-turn trackers (not real in-flight requests). */
181 > private readonly _syntheticPendingRequests = new WeakSet<CancellableRequest>();
182 >
183 > /**
184 > * In-flight untitled→real materializations, keyed by the original untitled
185 > * chat session resource. A first send to an untitled contributed session
186 > * stores the promise that resolves to the newly minted real resource (or
187 > * `undefined` on failure). A concurrent second send for the same untitled
188 > * resource awaits this instead of materializing a second real session.
189 > *
190 > * The committed (settled) untitled→real mapping is owned by
191 > * {@link IChatSessionsService} (published via `setMaterializedSessionResource`
192 > * and read via `getMaterializedSessionResource`); this map only tracks the
193 > * transient in-flight serialization.
194 > */
195 > private readonly _inFlightUntitledMaterializations = new ResourceMap<Promise<URI | undefined>>();
196 > private _saveModelsEnabled = true;
197 >
198 > private _transferredSessionResource: URI | undefined;
199 > public get transferredSessionResource(): URI | undefined {
200 > return this._transferredSessionResource;
201 > }
202 >
203 > private readonly _onDidSubmitRequest = this._register(new Emitter<{ readonly chatSessionResource: URI; readonly message?: IParsedChatRequest }>());
204 > public readonly onDidSubmitRequest = this._onDidSubmitRequest.event;
205 >
206 > public get onDidCreateModel() { return this._sessionModels.onDidCreateModel; }
207 >
208 > private readonly _onDidPerformUserAction = this._register(new Emitter<IChatUserActionEvent>());
209 > public readonly onDidPerformUserAction: Event<IChatUserActionEvent> = this._onDidPerformUserAction.event;
210 >
211 > private readonly _onDidReceiveQuestionCarouselAnswer = this._register(new Emitter<{ requestId: string; resolveId: string; answers: IChatQuestionAnswers | undefined }>());
212 > public readonly onDidReceiveQuestionCarouselAnswer = this._onDidReceiveQuestionCarouselAnswer.event;
213 >
214 > private readonly _onDidDisposeSession = this._register(new Emitter<{ readonly sessionResources: URI[]; reason: 'cleared' }>());
215 > public readonly onDidDisposeSession = this._onDidDisposeSession.event;
216 >
217 > private readonly _sessionFollowupCancelTokens = this._register(new DisposableResourceMap<CancellationTokenSource>());
218 > private readonly _chatServiceTelemetry: ChatServiceTelemetry;
219 > private readonly _chatSessionStore: ChatSessionStore;
220 >
221 > readonly requestInProgressObs: IObservable<boolean>;
222 >
223 > readonly chatModels: IObservable<Iterable<IChatModel>>;
224 >
225 > /**
226 > * For test use only
227 > */
228 > setSaveModelsEnabled(enabled: boolean): void {
229 this._saveModelsEnabled = enabled;
230 }
232 > /**
233 > * For test use only
234 > */
235 > waitForModelDisposals(): Promise<void> {
236 > return this._sessionModels.waitForModelDisposals(); chatServiceImpl.ts
237 > }
239 > private get isEmptyWindow(): boolean {
240 > const workspace = this.workspaceContextService.getWorkspace(); chatServiceImpl.ts
241 > return !workspace.configuration && workspace.folders.length === 0;
242 > }
244 > constructor(
245 > @IStorageService private readonly storageService: IStorageService, chatServiceImpl.ts
246 > @ILogService private readonly logService: ILogService,
247 > @ITelemetryService private readonly telemetryService: ITelemetryService,
248 > @IExtensionService private readonly extensionService: IExtensionService,
249 > @IInstantiationService private readonly instantiationService: IInstantiationService,
250 > @IWorkspaceContextService private readonly workspaceContextService: IWorkspaceContextService,
251 > @IChatSlashCommandService private readonly chatSlashCommandService: IChatSlashCommandService,
252 > @IChatAgentService private readonly chatAgentService: IChatAgentService,
253 > @IConfigurationService private readonly configurationService: IConfigurationService,
254 > @IChatTransferService private readonly chatTransferService: IChatTransferService,
255 > @IChatSessionsService private readonly chatSessionService: IChatSessionsService,
256 > @IMcpService private readonly mcpService: IMcpService,
257 > @IPromptsService private readonly promptsService: IPromptsService,
258 > @IChatEntitlementService private readonly chatEntitlementService: IChatEntitlementService,
259 > @ILanguageModelsService private readonly languageModelsService: ILanguageModelsService,
260 > @IChatDebugService private readonly chatDebugService: IChatDebugService,
261 > ) {
262 > super();
263 >
264 > this._sessionModels = this._register(instantiationService.createInstance(ChatModelStore, {
265 > createModel: (props: IStartSessionProps) => this._startSession(props),
266 > willDisposeModel: async (model: ChatModel) => {
267 > const localSessionId = LocalChatSessionUri.parseLocalSessionId(model.sessionResource); chatServiceImpl.ts
268 > if (localSessionId && this.shouldStoreSession(model)) {
269 // Always preserve sessions that have custom titles, even if empty
270 if (model.getRequests().length === 0 && !model.customTitle) {
275 await this._chatSessionStore.storeSessions([model]);
276 }
277 > } else if (!localSessionId && (model.getRequests().length > 0 || hasDraftInput(model))) { chatServiceImpl.ts
278 logChangesToStateModel(model.inputModel, `disposing external session ${model.sessionResource} with requests or draft input, storing metadata to storage`, undefined, undefined, this.logService);
279 // External sessions: persist metadata when there are requests, OR when the
281 await this._chatSessionStore.storeSessionsMetadataOnly([model]);
282 }
284 > })); chatServiceImpl.ts
285 > this._register(this._sessionModels.onDidDisposeModel(model => {
286 clearChatMarks(model.sessionResource);
287 this.chatDebugService.endSession(model.sessionResource);
290 this.chatSessionService.clearMaterializedSessionResource(model.sessionResource);
291 this._onDidDisposeSession.fire({ sessionResources: [model.sessionResource], reason: 'cleared' });
292 > })); chatServiceImpl.ts
293 >
294 > this._chatServiceTelemetry = this.instantiationService.createInstance(ChatServiceTelemetry);
295 > this._chatSessionStore = this._register(this.instantiationService.createInstance(ChatSessionStore));
296 > this._chatSessionStore.migrateDataIfNeeded(() => this.migrateData());
297 >
298 > const transferredData = this._chatSessionStore.getTransferredSessionData();
299 > if (transferredData) {
300 this.trace('constructor', `Transferred session ${transferredData}`);
301 this._transferredSessionResource = transferredData;
302 }
304 > this._register(storageService.onWillSaveState(() => this.saveState()));
305 >
306 > this.chatModels = derived(this, reader => [...this._sessionModels.observable.read(reader).values()]);
307 >
308 > this.requestInProgressObs = derived(reader => {
309 const models = this._sessionModels.observable.read(reader).values();
310 return Iterable.some(models, model => model.requestInProgress.read(reader));
311 > }); chatServiceImpl.ts
312 > }
314 > public get editingSessions() {
315 return [...this._sessionModels.values()].map(v => v.editingSession).filter(isDefined);
316 }
318 > isEnabled(location: ChatAgentLocation): boolean {
319 return this.chatAgentService.getContributedDefaultAgent(location) !== undefined;
320 }
322 > private migrateData(): ISerializableChatsData | undefined {
323 > const sessionData = this.storageService.get(serializedChatKey, this.isEmptyWindow ? StorageScope.APPLICATION : StorageScope.WORKSPACE, ''); chatServiceImpl.ts
324 > if (sessionData) {
325 const persistedSessions = this.deserializeChats(sessionData);
326 const countsForLog = Object.keys(persistedSessions).length;
331 return persistedSessions;
332 }
334 > return;
335 > }
337 > private saveState(): void {
338 if (!this._saveModelsEnabled) {
339 return;
358 this._chatSessionStore.storeSessionsMetadataOnly(liveNonLocalChats);
359 }
361 > /**
362 > * Only persist local sessions from chat that are not imported.
363 > */
364 > private shouldStoreSession(session: ChatModel): boolean {
365 if (session.isDeleted) {
366 return false;
371 return session.initialLocation === ChatAgentLocation.Chat && !session.isImported;
372 }
374 > notifyUserAction(action: IChatUserActionEvent): void {
375 this._chatServiceTelemetry.notifyUserAction(action);
376 this._onDidPerformUserAction.fire(action);
382 }
383 }
385 > notifyQuestionCarouselAnswer(requestId: string, resolveId: string, answers: IChatQuestionAnswers | undefined): void {
386 this._onDidReceiveQuestionCarouselAnswer.fire({ requestId, resolveId, answers });
387 }
389 > async setChatSessionTitle(sessionResource: URI, title: string): Promise<void> {
390 const model = this._sessionModels.get(sessionResource);
391 if (model) {
401 }
402 }
404 > private trace(method: string, message?: string): void {
405 > if (message) { chatServiceImpl.ts
406 > this.logService.trace(`ChatService#${method}: ${message}`);
407 > } else {
408 this.logService.trace(`ChatService#${method}`);
409 }
412 > private info(method: string, message?: string): void {
413 if (message) {
414 this.logService.info(`ChatService#${method}: ${message}`);
417 }
418 }
420 > private error(method: string, message: string): void {
421 this.logService.error(`ChatService#${method} ${message}`);
422 }
424 > private deserializeChats(sessionData: string): ISerializableChatsData {
425 try {
426 const arrayOfSessions: ISerializableChatDataIn[] = revive(JSON.parse(sessionData)); // Revive serialized URIs in session data
453 }
454 }
456 > /**
457 > * Returns an array of chat details for all persisted chat sessions that have at least one request.
458 > * Chat sessions that have already been loaded into the chat view are excluded from the result.
459 > * Imported chat sessions are also excluded from the result.
460 > * TODO this is only used by the old "show chats" command which can be removed when the pre-agents view
461 > * options are removed.
462 > */
463 > async getLocalSessionHistory(): Promise<IChatDetail[]> {
464 const liveSessionItems = await this.getLiveSessionItems();
465 const historySessionItems = await this.getHistorySessionItems();
467 return [...liveSessionItems, ...historySessionItems];
468 }
470 > /**
471 > * Returns an array of chat details for all local live chat sessions.
472 > */
473 > async getLiveSessionItems(): Promise<IChatDetail[]> {
474 return await Promise.all(Array.from(this._sessionModels.values())
475 .filter(session => this.shouldBeInHistory(session))
476 .map(chatModelToChatDetail));
477 }
479 > /**
480 > * Returns an array of chat details for all local chat sessions in history (not currently loaded).
481 > */
482 > async getHistorySessionItems(): Promise<IChatDetail[]> {
483 const index = await this._chatSessionStore.getIndex();
484 return Object.values(index)
496 });
497 }
499 > async getMetadataForSession(sessionResource: URI): Promise<IChatDetail | undefined> {
500 const index = await this._chatSessionStore.getIndex();
501 const metadata: IChatSessionEntryMetadata | undefined = index[sessionResource.toString()];
512 return undefined;
513 }
515 > private shouldBeInHistory(entry: ChatModel): boolean {
516 return !entry.isImported && !entry.isDeleted && !!LocalChatSessionUri.parseLocalSessionId(entry.sessionResource) && entry.initialLocation === ChatAgentLocation.Chat;
517 }
519 > async removeHistoryEntry(sessionResource: URI): Promise<void> {
520 await this._chatSessionStore.deleteSession(this.toLocalSessionId(sessionResource));
521 const model = this._sessionModels.get(sessionResource);
525 this._onDidDisposeSession.fire({ sessionResources: [sessionResource], reason: 'cleared' });
526 }
528 > async clearAllHistoryEntries(): Promise<void> {
529 await this._chatSessionStore.clearAllSessions();
530 }
532 > startNewLocalSession(location: ChatAgentLocation, options?: IChatSessionStartOptions): IChatModelReference {
533 this.trace('startNewLocalSession');
534 const sessionResource = LocalChatSessionUri.forSession(generateUuid());
541 }, options?.debugOwner ?? 'ChatService#startNewLocalSession');
542 }
544 > private _startSession(props: IStartSessionProps): ChatModel {
545 > const { initialData, location, sessionResource, canUseTools, transferEditingSession, disableBackgroundKeepAlive, inputState, isReadOnly } = props; chatServiceImpl.ts
546 > const model = this.instantiationService.createInstance(ChatModel, initialData, { initialLocation: location, canUseTools, resource: sessionResource, disableBackgroundKeepAlive, inputState, isReadOnly });
547 > if (location === ChatAgentLocation.Chat) {
548 > model.startEditingSession(true, transferEditingSession);
549 > }
550 >
551 > this.initializeSession(model);
552 > return model;
553 > }
555 > private initializeSession(model: ChatModel): void {
556 > this.trace('initializeSession', `Initialize session ${model.sessionResource}`); chatServiceImpl.ts
557 >
558 > // Activate the default extension provided agent but do not wait
559 > // for it to be ready so that the session can be used immediately
560 > // without having to wait for the agent to be ready.
561 > this.activateDefaultAgent(model.initialLocation).catch(e => this.logService.error(e));
562 > }
564 > async activateDefaultAgent(location: ChatAgentLocation): Promise<void> {
565 > await this.extensionService.whenInstalledExtensionsRegistered(); chatServiceImpl.ts
566 >
567 > const defaultAgentData = this.chatAgentService.getContributedDefaultAgent(location) ?? this.chatAgentService.getContributedDefaultAgent(ChatAgentLocation.Chat);
568 > if (!defaultAgentData) {
569 throw new ErrorNoTelemetry('No default agent contributed');
570 }
572 > // Await activation of the extension provided agent
573 > // Using `activateById` as workaround for the issue
574 > // https://github.com/microsoft/vscode/issues/250590
575 > if (!defaultAgentData.isCore) {
576 > await this.extensionService.activateById(defaultAgentData.extensionId, {
577 > activationEvent: `onChatParticipant:${defaultAgentData.id}`,
578 > extensionId: defaultAgentData.extensionId,
579 > startup: false
580 > });
581 > }
582 >
583 > const defaultAgent = this.chatAgentService.getActivatedAgents().find(agent => agent.id === defaultAgentData.id);
584 > if (!defaultAgent) {
585 throw new ErrorNoTelemetry('No default agent registered');
586 }
589 > getSession(sessionResource: URI): IChatModel | undefined {
590 return this._sessionModels.get(sessionResource);
591 }
593 > acquireExistingSession(sessionResource: URI, debugOwner?: string): IChatModelReference | undefined {
594 > return this._sessionModels.acquireExisting(sessionResource, debugOwner ?? 'ChatService#acquireExistingSession'); chatServiceImpl.ts
595 > }
597 > getChatModelReferenceDebugInfo() {
598 return this._sessionModels.getReferenceDebugSnapshot();
599 }
601 > private async acquireOrRestoreLocalSession(sessionResource: URI, debugOwner?: string): Promise<IChatModelReference | undefined> {
602 this.trace('acquireOrRestoreSession', `${sessionResource}`);
603 const existingRef = this.acquireExistingSession(sessionResource, debugOwner);
630 return sessionRef;
631 }
633 > // There are some cases where this returns a real string. What happens if it doesn't?
634 > // This had titles restored from the index, so just return titles from index instead, sync.
635 > getSessionTitle(sessionResource: URI): string | undefined {
636 const sessionId = LocalChatSessionUri.parseLocalSessionId(sessionResource);
637 if (!sessionId) {
642 this._chatSessionStore.getMetadataForSessionSync(sessionResource)?.title;
643 }
645 > loadSessionFromData(data: IExportableChatData | ISerializableChatData, debugOwner?: string): IChatModelReference {
646 const sessionId = (data as ISerializableChatData).sessionId ?? generateUuid();
647 const sessionResource = LocalChatSessionUri.forSession(sessionId);
653 }, debugOwner ?? 'ChatService#loadSessionFromData');
654 }
656 > async acquireOrLoadSession(sessionResource: URI, location: ChatAgentLocation, token: CancellationToken, debugOwner?: string): Promise<IChatModelReference | undefined> {
657 > if (LocalChatSessionUri.isLocalSession(sessionResource)) { chatServiceImpl.ts
658 return this.acquireOrRestoreLocalSession(sessionResource, debugOwner);
659 > } else { chatServiceImpl.ts
660 > return this.loadRemoteSession(sessionResource, location, token, debugOwner); chatServiceImpl.ts
661 > }
664 > private async loadRemoteSession(sessionResource: URI, location: ChatAgentLocation, token: CancellationToken, debugOwner?: string): Promise<IChatModelReference | undefined> {
665 > // Check if session already exists before resolving the provider, chatServiceImpl.ts
666 > // so we can return a cached model even if the provider was unregistered.
667 > {
668 > const existingRef = this.acquireExistingSession(sessionResource, debugOwner);
669 > if (existingRef) {
670 return existingRef;
671 }
673 >
674 > if (!await raceCancellationError(this.chatSessionService.canResolveChatSession(getChatSessionType(sessionResource)), token)) {
675 return undefined;
676 }
678 > const providedSession = await this.chatSessionService.getOrCreateChatSession(sessionResource, token);
679 >
680 > // Make sure we haven't created this in the meantime
681 > {
682 > const existingRef = this.acquireExistingSession(sessionResource, debugOwner);
683 > if (existingRef) {
684 return existingRef;
685 }
687 > const chatSessionType = getChatSessionType(sessionResource); chatServiceImpl.ts
688 > const modelId = findLast(providedSession.history.filter(m => m.type === 'request'), req => req.modelId)?.modelId;
689 > const agentUri = findLast(providedSession.history.filter(m => m.type === 'request'), req => req.modeInstructions?.uri)?.modeInstructions?.uri; chatServiceImpl.ts
690 > const storedMetadata = this._chatSessionStore.getMetadataForSessionSync(sessionResource);
691 > const storedPermissionLevel = storedMetadata?.permissionLevel;
692 > const storedInputState = storedMetadata?.inputState;
693 > let initialData: ISerializedChatDataReference | undefined = undefined;
694 > let historySelectedModel: string | undefined = undefined;
695 > let historyDerivedModel: ISerializableChatModelInputState['selectedModel'] = undefined;
696 > if ((modelId || agentUri)) {
697 const mode: ISerializableChatModelInputState['mode'] = agentUri ? { kind: ChatModeKind.Agent, id: agentUri.toString() } : { kind: ChatModeKind.Agent, id: ChatMode.Agent.id };
698 const modelMetadata = modelId ? this.languageModelsService.lookupLanguageModel(modelId) : undefined;
747 };
748 }
750 > // Contributed sessions do not use UI tools.
751 > // Prefer (in order): a transferred draft, the persisted draft from metadata,
752 > // otherwise let the constructor fall back to initialData.value.inputState.
753 > // When restoring the persisted draft we keep the unsent text/selections/mode but
754 > // deliberately drop its persisted selectedModel identifier (it can be stale or belong
755 > // to a different model pool) in favour of the model derived from the session's request
756 > // history. The user's per-model configuration (thinking effort, context window) is
757 > // carried over onto that history-derived model above when the ids match. When no
758 > // history model is available the model is left undefined so the input part resolves
759 > // it via its own selection logic.
760 > const restoredDraft: ISerializableChatModelInputState | undefined = storedInputState
761 ? { ...storedInputState, selectedModel: historyDerivedModel }
762 > : undefined; chatServiceImpl.ts
763 > // At cold restore the agent-host transferred draft can drop the user's per-session picker chatServiceImpl.ts
764 > // selections (model/mode); restore them from the session's own saved `storedInputState`
765 > // (mode, via {@link backfillRestoredPickerState}) and from the history-derived model
766 > // (via {@link backfillTransferredModel}). The persisted draft already contains
767 > // `historyDerivedModel`, so only a transferred draft needs this backfill.
768 > const transferredInputState = providedSession.transferredState?.inputState;
769 > const stateToApply = transferredInputState
770 ? backfillTransferredModel(transferredInputState, historyDerivedModel)
771 > : restoredDraft; chatServiceImpl.ts
772 > const inputState = backfillRestoredPickerState(stateToApply, storedInputState, ChatMode.Agent.id); chatServiceImpl.ts
773 > const modelRef = this._sessionModels.acquireOrCreate({
774 > initialData,
775 > location,
776 > sessionResource: sessionResource,
777 > canUseTools: false,
778 > transferEditingSession: providedSession.transferredState?.editingSession,
779 > inputState,
780 > isReadOnly: providedSession.isReadOnly,
781 > }, debugOwner ?? 'ChatService#loadRemoteSession');
782 >
783 > logChangesToStateModel(modelRef.object.inputModel, `loadRemoteSession inputState source: session=${sessionResource.toString()}, chatSessionType=${chatSessionType}, historyModelId=${modelId}, agentUri=${agentUri?.toString()}, historySelectedModel=${historySelectedModel}, transferredSelectedModel=${providedSession.transferredState?.inputState?.selectedModel?.identifier}, storedSelectedModel=${storedInputState?.selectedModel?.identifier}, finalSelectedModel=${modelRef.object.inputModel.state.get()?.selectedModel?.identifier}, hasTransferredInputState=${!!providedSession.transferredState?.inputState}, hasStoredInputState=${!!storedInputState}, hasInitialData=${!!initialData}`, modelRef.object.inputModel.state.get(), undefined, this.logService);
784 >
785 > // Restore permission level from metadata even when initialData was not constructed
786 > // and no inputState carried it through.
787 > if (storedPermissionLevel && !initialData && !storedInputState) {
788 modelRef.object.inputModel.setState({ permissionLevel: storedPermissionLevel });
789 }
791 > if (providedSession.title) {
792 modelRef.object.setCustomTitle(providedSession.title);
793 }
795 > const model = modelRef.object;
796 > const disposables = new DisposableStore();
797 > disposables.add(modelRef.object.onDidDispose(() => {
798 > disposables.dispose();
799 > providedSession.dispose();
800 > }));
801 >
802 > const isAgentHostSession = isAgentHostTarget(chatSessionType);
803 > const requestParser = isAgentHostSession ? this.instantiationService.createInstance(ChatRequestParser) : undefined; chatServiceImpl.ts
804 > const parseAgentHostHistoryPrompt = (text: string, agent: IChatAgentData | undefined): IParsedChatRequest => {
805 > if (requestParser) { chatServiceImpl.ts
806 > try {
807 > const attachmentCapabilities = this.getAttachmentCapabilitiesForParser(chatSessionType, agent);
808 > const parsed = requestParser.parseChatRequestWithReferences(
809 > EMPTY_REFERENCES,
810 > EMPTY_TOOL_ENABLEMENT_MAP,
811 > text,
812 > location,
813 > { sessionType: chatSessionType, forcedAgent: agent, attachmentCapabilities },
814 > );
815 > if (parsed.parts.length > 0) {
816 > return parsed;
817 > }
818 > } catch (e) {
819 this.logService.warn(`ChatService#loadRemoteSession: failed to re-parse historical prompt for ${chatSessionType}`, e);
820 }
822 return {
823 text,
828 )]
829 };
832 > let lastRequest: ChatRequestModel | undefined;
833 > let lastResponseCompletedAt: number | undefined;
834 > const completeLastResponse = () => {
835 > if (Number.isFinite(lastResponseCompletedAt)) { chatServiceImpl.ts
836 lastRequest?.response?.complete(lastResponseCompletedAt);
837 > } else { chatServiceImpl.ts
838 > lastRequest?.response?.completeWithoutTimestamp(); chatServiceImpl.ts
839 > }
840 > lastResponseCompletedAt = undefined; chatServiceImpl.ts
841 > };
842 > for (const message of providedSession.history) { chatServiceImpl.ts
843 > if (message.type === 'request') { chatServiceImpl.ts
844 > if (lastRequest) {
845 completeLastResponse();
846 }
848 > const requestText = message.prompt;
849 > const agent =
850 > message.participant
851 > ? this.chatAgentService.getAgent(message.participant) // TODO(jospicer): Remove and always hardcode?
852 : this.chatAgentService.getAgent(chatSessionType);
853 > const parsedRequest = parseAgentHostHistoryPrompt(requestText, agent); chatServiceImpl.ts
854 > const modeInfo = message.modeInstructions ? {
855 kind: ChatModeKind.Agent,
856 isBuiltin: message.modeInstructions.isBuiltin ?? false,
858 telemetryModeId: 'custom',
859 applyCodeBlockSuggestionId: undefined,
860 > } satisfies IChatRequestModeInfo : undefined; chatServiceImpl.ts
861 > lastRequest = model.addRequest(parsedRequest,
862 > message.variableData ?? { variables: [] },
863 > 0, // attempt
864 > modeInfo,
865 > agent,
866 > undefined, // slashCommand
867 > undefined, // confirmation
868 > undefined, // locationData
869 > undefined, // attachments
870 > false, // Do not treat as requests completed, else edit pills won't show.
871 > message.modelId,
872 > undefined,
873 > message.id,
874 > message.isSystemInitiated,
875 > message.systemInitiatedLabel,
876 > undefined, // terminalExecutionId
877 > message.isTerminalRequest,
878 > message.timestamp ?? null,
879 > );
880 > } else {
881 // response
882 if (lastRequest) {
896 }
897 }
900 > // Set up progress streaming and cancellation for contributed sessions.
901 > // This handles both the initial in-flight response (from session load)
902 > // and any subsequent server-initiated turns (e.g. consumed queued messages).
903 > const hasProgressStreaming = providedSession.progressObs && providedSession.interruptActiveResponseCallback;
904 > if (hasProgressStreaming) { chatServiceImpl.ts
905 let lastProgressLength = 0;
906
1040 }
1041 }));
1042 > } else { chatServiceImpl.ts
1043 > if (providedSession.isCompleteObs?.get()) { chatServiceImpl.ts
1044 completeLastResponse();
1045 }
1047 > this.telemetryService.publicLog2<ChatPendingRequestChangeEvent, ChatPendingRequestChangeClassification>(ChatPendingRequestChangeEventName, { action: 'notCancelable', source: 'remoteSession', chatSessionId: chatSessionResourceToId(model.sessionResource) });
1048 > if (lastRequest && model.editingSession) {
1049 > // wait for timeline to load so that a 'changes' part is added when the response completes chatServiceImpl.ts
1050 > await chatEditingSessionIsReady(model.editingSession);
1051 > completeLastResponse();
1052 > }
1055 > return modelRef;
1058 > async resendRequest(request: IChatRequestModel, options?: IChatSendRequestOptions): Promise<void> {
1059 const model = this._sessionModels.get(request.session.sessionResource);
1060 if (!model && model !== request.session) {
1085 await this._sendRequestAsync(model, model.sessionResource, request.message, attempt, enableCommandDetection, defaultAgent, location, resendOptions).responseCompletePromise;
1086 }
1088 > private queuePendingRequest(model: ChatModel, sessionResource: URI, request: string, options: IChatSendRequestOptions): ChatSendResultQueued {
1089 const location = options.location ?? model.initialLocation;
1090 const parsedRequest = this.parseChatRequest(sessionResource, request, location, options);
1116 return { kind: 'queued', deferred: deferred.p };
1117 }
1119 > async sendRequest(sessionResource: URI, request: string, options?: IChatSendRequestOptions): Promise<ChatSendResult> {
1120 > this.trace('sendRequest', `sessionResource: ${sessionResource.toString()}, message: ${request.substring(0, 20)}${request.length > 20 ? '[...]' : ''}}`); chatServiceImpl.ts
1121 >
1122 > const hasExplicitFileOrImageAttachment = [...(options?.attachedContext ?? []), ...(options?.resolvedVariables ?? [])].some(isExplicitFileOrImageVariableEntry);
1123 > if (!request.trim() && !hasExplicitFileOrImageAttachment && !options?.slashCommand && !options?.agentId && !options?.agentIdSilent) {
1124 this.trace('sendRequest', 'Rejected empty message');
1125 return { kind: 'rejected', reason: 'Empty message' };
1126 }
1128 > let newSessionResource: URI | undefined;
1129 >
1130 > // A late send may arrive on a stale untitled resource after it already
1131 > // materialized into a real session but before the UI swapped to the real
1132 > // resource. Re-target to the real resource so we don't materialize a
1133 > // second session, and report it as a new session so the caller swaps its
1134 > // UI from the untitled resource to the real one (mirroring the first send).
1135 > const materializedReal = this.chatSessionService.getMaterializedSessionResource(sessionResource);
1136 > if (materializedReal) {
1137 sessionResource = materializedReal;
1138 newSessionResource = materializedReal;
1139 }
1141 > let model = this._sessionModels.get(sessionResource);
1142 > if (!model) {
1143 throw new Error(`Unknown session: ${sessionResource}`);
1144 }
1145 > if (model.isReadOnly.get()) { chatServiceImpl.ts
1146 > return { chatServiceImpl.ts
1147 > kind: 'rejected',
1148 > reason: 'Session is read-only',
1149 > ...(newSessionResource ? { newSessionResource } : {}),
1150 > };
1151 > }
1152
1153 // Internally blank widgets use special sessions with an untitled- path.
1156 // and serialized per untitled resource — see
1157 // `_materializeUntitledSession`) before processing the request.
1158 > if (!model.hasRequests && isUntitledChatSession(sessionResource) && getChatSessionType(sessionResource) !== localChatSessionType) { chatServiceImpl.ts
1159 const materialized = await this._materializeUntitledSession(sessionResource, request, options, model);
1160 if (materialized) {
1170 const hasPendingRequest = this._pendingRequests.has(sessionResource);
1171
1172 > if (options?.queue) { chatServiceImpl.ts
1173 const queued = this.queuePendingRequest(model, sessionResource, request, options);
1174 if (!options.pauseQueue) {
1193 }
1194
1195 > const location = options?.location ?? model.initialLocation; chatServiceImpl.ts
1196 > const attempt = options?.attempt ?? 0;
1197 > const defaultAgent = this.chatAgentService.getDefaultAgent(location, options?.modeInfo?.kind);
1198 > if (!defaultAgent) {
1199 this.logService.warn('sendRequest', `No default agent for location ${location}`);
1200 return { kind: 'rejected', reason: 'No default agent available' };
1202
1203 const parsedRequest = this.parseChatRequest(sessionResource, request, location, options);
1204 > const silentAgent = options?.agentIdSilent ? this.chatAgentService.getAgent(options.agentIdSilent) : undefined; chatServiceImpl.ts
1205 > const agent = silentAgent ?? parsedRequest.parts.find((r): r is ChatRequestAgentPart => r instanceof ChatRequestAgentPart)?.agent ?? defaultAgent;
1206 > const agentSlashCommandPart = parsedRequest.parts.find((r): r is ChatRequestAgentSubcommandPart => r instanceof ChatRequestAgentSubcommandPart);
1207 >
1208 > // This method is only returning whether the request was accepted - don't block on the actual request
1209 > return {
1210 > kind: 'sent',
1211 > newSessionResource,
1212 > data: {
1213 > ...this._sendRequestAsync(model, sessionResource, parsedRequest, attempt, !options?.noCommandDetection, silentAgent ?? defaultAgent, location, options),
1214 > agent,
1215 > slashCommand: agentSlashCommandPart?.command,
1216 > },
1217 > };
1218 > }
1220 > /**
1221 > * Converts an untitled contributed chat session into its real session on the
1222 > * first send and returns the real model/resource so the caller can re-target
1223 > * the request. Serialized per untitled resource: a first send stores an
1224 > * in-flight promise, and a concurrent second send awaits it and converges on
1225 > * the same real session (where the caller's pending-request check then rejects
1226 > * the duplicate) instead of minting a second real session.
1227 > *
1228 > * Returns `undefined` when no conversion happened — either there is no
1229 > * `newChatSessionItem` handler / the handler declined, or a concurrent
1230 > * materialization failed — in which case the caller keeps using the untitled
1231 > * session (the original behavior).
1232 > */
1233 > private async _materializeUntitledSession(untitledResource: URI, request: string, options: IChatSendRequestOptions | undefined, untitledModel: ChatModel): Promise<{ model: ChatModel; sessionResource: URI; newSessionResource: URI } | undefined> {
1234 const inFlight = this._inFlightUntitledMaterializations.get(untitledResource);
1235 if (inFlight) {
1316 }
1317 }
1319 > private getAttachmentCapabilitiesForParser(chatSessionType: string, agent: IChatAgentData | undefined): IChatAgentAttachmentCapabilities | undefined {
1320 > return this.chatSessionService.getCapabilitiesForSessionType(chatSessionType) ?? agent?.capabilities; chatServiceImpl.ts
1321 > }
1323 > private parseChatRequest(sessionResource: URI, request: string, location: ChatAgentLocation, options: IChatSendRequestOptions | undefined): IParsedChatRequest {
1324 let parserContext = options?.parserContext;
1325 let contextAgent = parserContext?.forcedAgent ?? parserContext?.selectedAgent;
1351 return parsedRequest;
1352 }
1354 > private refreshFollowupsCancellationToken(sessionResource: URI): CancellationToken {
1355 this._sessionFollowupCancelTokens.get(sessionResource)?.cancel();
1356 const newTokenSource = new CancellationTokenSource();
1359 return newTokenSource.token;
1360 }
1362 > private _sendRequestAsync(model: ChatModel, sessionResource: URI, parsedRequest: IParsedChatRequest, attempt: number, enableCommandDetection: boolean, defaultAgent: IChatAgentData, location: ChatAgentLocation, options?: IChatSendRequestOptions): IChatSendRequestResponseState {
1363 const followupsCancelToken = this.refreshFollowupsCancellationToken(sessionResource);
1364 let request: ChatRequestModel | undefined;
1825 };
1826 }
1828 > processPendingRequests(sessionResource: URI): void {
1829 const model = this._sessionModels.get(sessionResource);
1830 if (model && !this._pendingRequests.has(sessionResource)) {
1832 }
1833 }
1835 > /**
1836 > * Returns true if the session is backed by an agent host server, which
1837 > * controls queued-message dequeuing on the server side.
1838 > */
1839 > private _isServerManagedQueue(sessionResource: URI): boolean {
1840 return getChatSessionType(sessionResource).startsWith('agent-host-');
1841 }
1843 > /**
1844 > * Process the next pending request from the model's queue, if any.
1845 > * Called after a request completes to continue processing queued requests.
1846 > * Multiple consecutive steering requests are combined into a single request.
1847 > */
1848 > private processNextPendingRequest(model: ChatModel): void {
1849 // Agent host sessions delegate queue management to the server.
1850 // The server dispatches ChatTurnStarted with queuedMessageId when
1948 }
1949 }
1951 > private generateInitialChatTitleIfNeeded(model: ChatModel, request: IChatAgentRequest, defaultAgent: IChatAgentData, token: CancellationToken): void {
1952 // Generate a title only for the first request, and only via the default agent.
1953 // Use a single-entry history based on the current request (no full chat history).
1969 void generate();
1970 }
1972 > private prepareContext(attachedContextVariables: IChatRequestVariableEntry[] | undefined): IChatRequestVariableEntry[] {
1973 attachedContextVariables ??= [];
1974
1990 return attachedContextVariables;
1991 }
1993 > private getHistoryEntriesFromModel(requests: IChatRequestModel[], location: ChatAgentLocation, forAgentId: string): IChatAgentHistoryEntry[] {
1994 const history: IChatAgentHistoryEntry[] = [];
1995 const agent = this.chatAgentService.getAgent(forAgentId);
2027 return history;
2028 }
2030 > async removeRequest(sessionResource: URI, requestId: string): Promise<void> {
2031 const model = this._sessionModels.get(sessionResource);
2032 if (!model) {
2043 model.removeRequest(requestId);
2044 }
2046 > async adoptRequest(sessionResource: URI, request: IChatRequestModel) {
2047 if (!(request instanceof ChatRequestModel)) {
2048 throw new TypeError('Can only adopt requests of type ChatRequestModel');
2066 }
2067 }
2069 > async addCompleteRequest(sessionResource: URI, message: IParsedChatRequest | string, variableData: IChatRequestVariableData | undefined, attempt: number | undefined, response: IChatCompleteResponse): Promise<void> {
2070 this.trace('addCompleteRequest', `message: ${message}`);
2071
2093 request.response?.complete();
2094 }
2096 > async cancelCurrentRequestForSession(sessionResource: URI, source?: string): Promise<void> {
2097 this.trace('cancelCurrentRequestForSession', `session: ${sessionResource}`);
2098 const pendingRequest = this._pendingRequests.get(sessionResource);
2126 }
2127 }
2129 > setYieldRequested(sessionResource: URI): void {
2130 const pendingRequest = this._pendingRequests.get(sessionResource);
2131 if (pendingRequest) {
2133 }
2134 }
2136 > migrateRequests(originalResource: URI, targetResource: URI): void {
2137 const model = this._sessionModels.get(originalResource);
2138 if (!model) {
2159 }
2160 }
2162 > removePendingRequest(sessionResource: URI, requestId: string): void {
2163 const model = this._sessionModels.get(sessionResource) as ChatModel | undefined;
2164 if (model) {
2180 }
2181 }
2183 > setPendingRequests(sessionResource: URI, requests: readonly { requestId: string; kind: ChatRequestQueueKind }[]): void {
2184 const model = this._sessionModels.get(sessionResource) as ChatModel | undefined;
2185 if (model) {
2187 }
2188 }
2190 > syncPendingRequestsFromRemote(sessionResource: URI, requests: readonly IRemotePendingRequest[]): void {
2191 const model = this._sessionModels.get(sessionResource) as ChatModel | undefined;
2192 if (!model) {
2236 }
2237 }
2239 > async sendPendingRequestImmediately(sessionResource: URI, requestId: string): Promise<void> {
2240 const model = this._sessionModels.get(sessionResource) as ChatModel | undefined;
2241 if (!model) {
2288 this.processPendingRequests(sessionResource);
2289 }
2291 > public hasSessions(): boolean {
2292 return this._chatSessionStore.hasSessions();
2293 }
2295 > async transferChatSession(transferredSessionResource: URI, toWorkspace: URI): Promise<void> {
2296 if (!LocalChatSessionUri.isLocalSession(transferredSessionResource)) {
2297 throw new Error(`Can only transfer local chat sessions. Invalid session: ${transferredSessionResource}`);
2315 this.trace('transferChatSession', `Transferred session ${model.sessionResource} to workspace ${toWorkspace.toString()}`);
2316 }
2318 > getChatStorageFolder(): URI {
2319 return this._chatSessionStore.getChatStorageFolder();
2320 }
2322 > logChatIndex(): void {
2323 this._chatSessionStore.logIndex();
2324 }
2326 > setSessionTitle(sessionResource: URI, title: string): void {
2327 this._sessionModels.get(sessionResource)?.setCustomTitle(title);
2328 }
2330 > appendProgress(request: IChatRequestModel, progress: IChatProgress): void {
2331 const model = this._sessionModels.get(request.session.sessionResource);
2332 if (!(request instanceof ChatRequestModel)) {
2336 model?.acceptResponseProgress(request, progress);
2337 }
2339 > private toLocalSessionId(sessionResource: URI) {
2340 const localSessionId = LocalChatSessionUri.parseLocalSessionId(sessionResource);
2341 if (!localSessionId) {
2344 return localSessionId;
2345 }
2347 >
2348 export async function chatModelToChatDetail(model: IChatModel): Promise<IChatDetail> {
2349 const title = model.title || localize('newChat', "New Chat");
src/vs/platform/extensionManagement/common/extensionManagement.ts 660 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensionManagement.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../base/common/cancellation.js';
7 > import { IStringDictionary } from '../../../base/common/collections.js';
8 > import { Event } from '../../../base/common/event.js';
9 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
10 > import { IPager } from '../../../base/common/paging.js';
11 > import { Platform } from '../../../base/common/platform.js';
12 > import { PolicyCategory } from '../../../base/common/policy.js';
13 > import { URI } from '../../../base/common/uri.js';
14 > import { localize, localize2 } from '../../../nls.js';
15 > import { ConfigurationScope, Extensions, IConfigurationRegistry } from '../../configuration/common/configurationRegistry.js';
16 > import { ExtensionType, IExtension, IExtensionManifest, TargetPlatform } from '../../extensions/common/extensions.js';
17 > import { FileOperationError, FileOperationResult, IFileService, IFileStat } from '../../files/common/files.js';
18 > import { createDecorator } from '../../instantiation/common/instantiation.js';
19 > import { Registry } from '../../registry/common/platform.js';
20 > import { IExtensionGalleryManifest } from './extensionGalleryManifest.js';
21 >
22 > export const EXTENSION_IDENTIFIER_PATTERN = '^([a-z0-9A-Z][a-z0-9-A-Z]*)\\.([a-z0-9A-Z][a-z0-9-A-Z]*)$';
23 > export const EXTENSION_IDENTIFIER_REGEX = new RegExp(EXTENSION_IDENTIFIER_PATTERN);
24 > export const WEB_EXTENSION_TAG = '__web_extension';
25 > export const LANGUAGE_MODEL_CHAT_PROVIDER_EXTENSION_TAG = 'language-models';
26 > export const EXTENSION_INSTALL_SKIP_WALKTHROUGH_CONTEXT = 'skipWalkthrough';
27 > export const EXTENSION_INSTALL_SKIP_PUBLISHER_TRUST_CONTEXT = 'skipPublisherTrust';
28 > export const EXTENSION_INSTALL_SOURCE_CONTEXT = 'extensionInstallSource';
29 > export const EXTENSION_INSTALL_DEP_PACK_CONTEXT = 'dependecyOrPackExtensionInstall';
30 > export const EXTENSION_INSTALL_CLIENT_TARGET_PLATFORM_CONTEXT = 'clientTargetPlatform';
31 >
32 > export const enum ExtensionInstallSource {
33 > COMMAND = 'command',
34 > SETTINGS_SYNC = 'settingsSync',
35 > }
36 >
37 > export interface IProductVersion {
38 > readonly version: string;
39 > readonly date?: string;
40 > }
41 >
42 > export function TargetPlatformToString(targetPlatform: TargetPlatform) {
43 switch (targetPlatform) {
44 case TargetPlatform.WIN32_X64: return 'Windows 64 bit';
62 }
63 }
65 > export function toTargetPlatform(targetPlatform: string): TargetPlatform {
66 switch (targetPlatform) {
67 case TargetPlatform.WIN32_X64: return TargetPlatform.WIN32_X64;
84 }
85 }
87 > export function getTargetPlatform(platform: Platform | 'alpine', arch: string | undefined): TargetPlatform {
88 switch (platform) {
89 case Platform.Windows:
129 }
130 }
132 > export function isNotWebExtensionInWebTargetPlatform(allTargetPlatforms: TargetPlatform[], productTargetPlatform: TargetPlatform): boolean {
133 // Not a web extension in web target platform
134 return productTargetPlatform === TargetPlatform.WEB && !allTargetPlatforms.includes(TargetPlatform.WEB);
135 }
137 > export function isTargetPlatformCompatible(extensionTargetPlatform: TargetPlatform, allTargetPlatforms: TargetPlatform[], productTargetPlatform: TargetPlatform): boolean {
138 // Not compatible when extension is not a web extension in web target platform
139 if (isNotWebExtensionInWebTargetPlatform(allTargetPlatforms, productTargetPlatform)) {
163 return false;
164 }
166 > export interface IGalleryExtensionProperties {
167 > dependencies?: string[];
168 > extensionPack?: string[];
169 > engine?: string;
170 > enabledApiProposals?: string[];
171 > localizedLanguages?: string[];
172 > targetPlatform: TargetPlatform;
173 > isPreReleaseVersion: boolean;
174 > executesCode?: boolean;
175 > }
176 >
177 > export interface IGalleryExtensionAsset {
178 > uri: string;
179 > fallbackUri: string;
180 > }
181 >
182 > export interface IGalleryExtensionAssets {
183 > manifest: IGalleryExtensionAsset | null;
184 > readme: IGalleryExtensionAsset | null;
185 > changelog: IGalleryExtensionAsset | null;
186 > license: IGalleryExtensionAsset | null;
187 > repository: IGalleryExtensionAsset | null;
188 > download: IGalleryExtensionAsset;
189 > icon: IGalleryExtensionAsset | null;
190 > signature: IGalleryExtensionAsset | null;
191 > coreTranslations: [string, IGalleryExtensionAsset][];
192 > }
193 >
194 > export function isIExtensionIdentifier(obj: unknown): obj is IExtensionIdentifier {
195 const thing = obj as IExtensionIdentifier | undefined;
196 return !!thing
199 && (!thing.uuid || typeof thing.uuid === 'string');
200 }
202 > export interface IExtensionIdentifier {
203 > id: string;
204 > uuid?: string;
205 > }
206 >
207 > export interface IGalleryExtensionIdentifier extends IExtensionIdentifier {
208 > uuid: string;
209 > }
210 >
211 > export interface IGalleryExtensionVersion {
212 > version: string;
213 > date: string;
214 > isPreReleaseVersion: boolean;
215 > targetPlatforms: TargetPlatform[];
216 > }
217 >
218 > export interface IGalleryExtension {
219 > type: 'gallery';
220 > name: string;
221 > identifier: IGalleryExtensionIdentifier;
222 > version: string;
223 > displayName: string;
224 > publisherId: string;
225 > publisher: string;
226 > publisherDisplayName: string;
227 > publisherDomain?: { link: string; verified: boolean };
228 > publisherLink?: string;
229 > publisherSponsorLink?: string;
230 > description: string;
231 > installCount: number;
232 > rating: number;
233 > ratingCount: number;
234 > categories: readonly string[];
235 > tags: readonly string[];
236 > releaseDate: number;
237 > lastUpdated: number;
238 > preview: boolean;
239 > private: boolean;
240 > hasPreReleaseVersion: boolean;
241 > hasReleaseVersion: boolean;
242 > isSigned: boolean;
243 > allTargetPlatforms: TargetPlatform[];
244 > assets: IGalleryExtensionAssets;
245 > properties: IGalleryExtensionProperties;
246 > detailsLink?: string;
247 > ratingLink?: string;
248 > supportLink?: string;
249 > telemetryData?: IStringDictionary<unknown>;
250 > queryContext?: IStringDictionary<unknown>;
251 > }
252 >
253 > export type InstallSource = 'gallery' | 'vsix' | 'resource';
254 >
255 > export interface IGalleryMetadata {
256 > id: string;
257 > publisherId: string;
258 > private: boolean;
259 > publisherDisplayName: string;
260 > isPreReleaseVersion: boolean;
261 > targetPlatform?: TargetPlatform;
262 > }
263 >
264 > export type Metadata = Partial<IGalleryMetadata & {
265 > isApplicationScoped: boolean;
266 > isMachineScoped: boolean;
267 > isBuiltin: boolean;
268 > isSystem: boolean;
269 > updated: boolean;
270 > preRelease: boolean;
271 > hasPreReleaseVersion: boolean;
272 > installedTimestamp: number;
273 > pinned: boolean;
274 > source: InstallSource;
275 > size: number;
276 > }>;
277 >
278 > export interface ILocalExtension extends IExtension {
279 > isWorkspaceScoped: boolean;
280 > isMachineScoped: boolean;
281 > isApplicationScoped: boolean;
282 > publisherId: string | null;
283 > installedTimestamp?: number;
284 > isPreReleaseVersion: boolean;
285 > hasPreReleaseVersion: boolean;
286 > private: boolean;
287 > preRelease: boolean;
288 > updated: boolean;
289 > pinned: boolean;
290 > forceAutoUpdate: boolean;
291 > source: InstallSource;
292 > size: number;
293 > }
294 >
295 > export const enum SortBy {
296 > NoneOrRelevance = 'NoneOrRelevance',
297 > LastUpdatedDate = 'LastUpdatedDate',
298 > Title = 'Title',
299 > PublisherName = 'PublisherName',
300 > InstallCount = 'InstallCount',
301 > PublishedDate = 'PublishedDate',
302 > AverageRating = 'AverageRating',
303 > WeightedRating = 'WeightedRating'
304 > }
305 >
306 > export const enum SortOrder {
307 > Default = 0,
308 > Ascending = 1,
309 > Descending = 2
310 > }
311 >
312 > export const enum FilterType {
313 > Category = 'Category',
314 > ExtensionId = 'ExtensionId',
315 > ExtensionName = 'ExtensionName',
316 > ExcludeWithFlags = 'ExcludeWithFlags',
317 > Featured = 'Featured',
318 > SearchText = 'SearchText',
319 > Tag = 'Tag',
320 > Target = 'Target',
321 > }
322 >
323 > export interface IQueryOptions {
324 > text?: string;
325 > exclude?: string[];
326 > pageSize?: number;
327 > sortBy?: SortBy;
328 > sortOrder?: SortOrder;
329 > source?: string;
330 > includePreRelease?: boolean;
331 > productVersion?: IProductVersion;
332 > }
333 >
334 > export const enum StatisticType {
335 > Install = 'install',
336 > Uninstall = 'uninstall'
337 > }
338 >
339 > export interface IDeprecationInfo {
340 > readonly disallowInstall?: boolean;
341 > readonly extension?: {
342 > readonly id: string;
343 > readonly displayName: string;
344 > readonly autoMigrate?: {
345 > readonly storage: boolean;
346 > readonly donotDisable?: boolean;
347 > };
348 > readonly preRelease?: boolean;
349 > };
350 > readonly settings?: readonly string[];
351 > readonly additionalInfo?: string;
352 > }
353 >
354 > export interface ISearchPrefferedResults {
355 > readonly query?: string;
356 > readonly preferredResults?: string[];
357 > }
358 >
359 > export type MaliciousExtensionInfo = {
360 > readonly extensionOrPublisher: IExtensionIdentifier | string;
361 > readonly learnMoreLink?: string;
362 > };
363 >
364 > export interface IExtensionsControlManifest {
365 > readonly malicious: ReadonlyArray<MaliciousExtensionInfo>;
366 > readonly deprecated: IStringDictionary<IDeprecationInfo>;
367 > readonly search: ISearchPrefferedResults[];
368 > readonly autoUpdate?: IStringDictionary<string>;
369 > }
370 >
371 > export const enum InstallOperation {
372 > None = 1,
373 > Install,
374 > Update,
375 > Migrate,
376 > }
377 >
378 > export interface ITranslation {
379 > contents: { [key: string]: {} };
380 > }
381 >
382 > export interface IExtensionInfo extends IExtensionIdentifier {
383 > version?: string;
384 > preRelease?: boolean;
385 > hasPreRelease?: boolean;
386 > currentVersion?: string;
387 > }
388 >
389 > export interface IExtensionQueryOptions {
390 > targetPlatform?: TargetPlatform;
391 > productVersion?: IProductVersion;
392 > compatible?: boolean;
393 > queryAllVersions?: boolean;
394 > source?: string;
395 > }
396 >
397 > export interface IExtensionGalleryCapabilities {
398 > readonly query: {
399 > readonly sortBy: readonly SortBy[];
400 > readonly filters: readonly FilterType[];
401 > };
402 > readonly allRepositorySigned: boolean;
403 > }
404 >
405 > export const IExtensionGalleryService = createDecorator<IExtensionGalleryService>('extensionGalleryService');
406 >
407 > /**
408 > * Service to interact with the Visual Studio Code Marketplace to get extensions.
409 > * @throws Error if the Marketplace is not enabled or not reachable.
410 > */
411 > export interface IExtensionGalleryService {
412 > readonly _serviceBrand: undefined;
413 > isEnabled(): boolean;
414 > query(options: IQueryOptions, token: CancellationToken): Promise<IPager<IGalleryExtension>>;
415 > getExtensions(extensionInfos: ReadonlyArray<IExtensionInfo>, token: CancellationToken): Promise<IGalleryExtension[]>;
416 > getExtensions(extensionInfos: ReadonlyArray<IExtensionInfo>, options: IExtensionQueryOptions, token: CancellationToken): Promise<IGalleryExtension[]>;
417 > isExtensionCompatible(extension: IGalleryExtension, includePreRelease: boolean, targetPlatform: TargetPlatform, productVersion?: IProductVersion): Promise<boolean>;
418 > getCompatibleExtension(extension: IGalleryExtension, includePreRelease: boolean, targetPlatform: TargetPlatform, productVersion?: IProductVersion): Promise<IGalleryExtension | null>;
419 > getAllCompatibleVersions(extensionIdentifier: IExtensionIdentifier, includePreRelease: boolean, targetPlatform: TargetPlatform): Promise<IGalleryExtensionVersion[]>;
420 > getAllVersions(extensionIdentifier: IExtensionIdentifier): Promise<IGalleryExtensionVersion[]>;
421 > download(extension: IGalleryExtension, location: URI, operation: InstallOperation): Promise<void>;
422 > downloadSignatureArchive(extension: IGalleryExtension, location: URI): Promise<void>;
423 > reportStatistic(publisher: string, name: string, version: string, type: StatisticType): Promise<void>;
424 > getReadme(extension: IGalleryExtension, token: CancellationToken): Promise<string>;
425 > getManifest(extension: IGalleryExtension, token: CancellationToken): Promise<IExtensionManifest | null>;
426 > getChangelog(extension: IGalleryExtension, token: CancellationToken): Promise<string>;
427 > getCoreTranslation(extension: IGalleryExtension, languageId: string): Promise<ITranslation | null>;
428 > getExtensionsControlManifest(): Promise<IExtensionsControlManifest>;
429 > }
430 >
431 > export interface InstallExtensionEvent {
432 > readonly identifier: IExtensionIdentifier;
433 > readonly source: URI | IGalleryExtension;
434 > readonly profileLocation: URI;
435 > readonly applicationScoped?: boolean;
436 > readonly workspaceScoped?: boolean;
437 > }
438 >
439 > export interface InstallExtensionResult {
440 > readonly identifier: IExtensionIdentifier;
441 > readonly operation: InstallOperation;
442 > readonly source?: URI | IGalleryExtension;
443 > readonly local?: ILocalExtension;
444 > readonly error?: Error;
445 > readonly context?: IStringDictionary<unknown>;
446 > readonly profileLocation: URI;
447 > readonly applicationScoped?: boolean;
448 > readonly workspaceScoped?: boolean;
449 > }
450 >
451 > export interface UninstallExtensionEvent {
452 > readonly identifier: IExtensionIdentifier;
453 > readonly profileLocation: URI;
454 > readonly applicationScoped?: boolean;
455 > readonly workspaceScoped?: boolean;
456 > }
457 >
458 > export interface DidUninstallExtensionEvent {
459 > readonly identifier: IExtensionIdentifier;
460 > readonly error?: string;
461 > readonly profileLocation: URI;
462 > readonly applicationScoped?: boolean;
463 > readonly workspaceScoped?: boolean;
464 > }
465 >
466 > export interface DidUpdateExtensionMetadata {
467 > readonly profileLocation: URI;
468 > readonly local: ILocalExtension;
469 > }
470 >
471 > export const enum ExtensionGalleryErrorCode {
472 > Timeout = 'Timeout',
473 > Cancelled = 'Cancelled',
474 > ClientError = 'ClientError',
475 > ServerError = 'ServerError',
476 > Failed = 'Failed',
477 > DownloadFailedWriting = 'DownloadFailedWriting',
478 > Offline = 'Offline',
479 > }
480 >
481 > export class ExtensionGalleryError extends Error {
482 > constructor(message: string, readonly code: ExtensionGalleryErrorCode) {
483 super(message);
484 this.name = code;
485 }
487 >
488 > export const enum ExtensionManagementErrorCode {
489 > NotFound = 'NotFound',
490 > Unsupported = 'Unsupported',
491 > Deprecated = 'Deprecated',
492 > Malicious = 'Malicious',
493 > Incompatible = 'Incompatible',
494 > IncompatibleApi = 'IncompatibleApi',
495 > IncompatibleTargetPlatform = 'IncompatibleTargetPlatform',
496 > ReleaseVersionNotFound = 'ReleaseVersionNotFound',
497 > Invalid = 'Invalid',
498 > Download = 'Download',
499 > DownloadSignature = 'DownloadSignature',
500 > DownloadFailedWriting = ExtensionGalleryErrorCode.DownloadFailedWriting,
501 > UpdateMetadata = 'UpdateMetadata',
502 > Extract = 'Extract',
503 > Scanning = 'Scanning',
504 > ScanningExtension = 'ScanningExtension',
505 > ReadRemoved = 'ReadRemoved',
506 > UnsetRemoved = 'UnsetRemoved',
507 > Delete = 'Delete',
508 > Rename = 'Rename',
509 > IntializeDefaultProfile = 'IntializeDefaultProfile',
510 > AddToProfile = 'AddToProfile',
511 > InstalledExtensionNotFound = 'InstalledExtensionNotFound',
512 > PostInstall = 'PostInstall',
513 > CorruptZip = 'CorruptZip',
514 > IncompleteZip = 'IncompleteZip',
515 > PackageNotSigned = 'PackageNotSigned',
516 > SignatureVerificationInternal = 'SignatureVerificationInternal',
517 > SignatureVerificationFailed = 'SignatureVerificationFailed',
518 > NotAllowed = 'NotAllowed',
519 > Gallery = 'Gallery',
520 > Cancelled = 'Cancelled',
521 > Unknown = 'Unknown',
522 > Internal = 'Internal',
523 > }
524 >
525 > export enum ExtensionSignatureVerificationCode {
526 > 'NotSigned' = 'NotSigned',
527 > 'Success' = 'Success',
528 > 'RequiredArgumentMissing' = 'RequiredArgumentMissing', // A required argument is missing.
529 > 'InvalidArgument' = 'InvalidArgument', // An argument is invalid.
530 > 'PackageIsUnreadable' = 'PackageIsUnreadable', // The extension package is unreadable.
531 > 'UnhandledException' = 'UnhandledException', // An unhandled exception occurred.
532 > 'SignatureManifestIsMissing' = 'SignatureManifestIsMissing', // The extension is missing a signature manifest file (.signature.manifest).
533 > 'SignatureManifestIsUnreadable' = 'SignatureManifestIsUnreadable', // The signature manifest is unreadable.
534 > 'SignatureIsMissing' = 'SignatureIsMissing', // The extension is missing a signature file (.signature.p7s).
535 > 'SignatureIsUnreadable' = 'SignatureIsUnreadable', // The signature is unreadable.
536 > 'CertificateIsUnreadable' = 'CertificateIsUnreadable', // The certificate is unreadable.
537 > 'SignatureArchiveIsUnreadable' = 'SignatureArchiveIsUnreadable',
538 > 'FileAlreadyExists' = 'FileAlreadyExists', // The output file already exists.
539 > 'SignatureArchiveIsInvalidZip' = 'SignatureArchiveIsInvalidZip',
540 > 'SignatureArchiveHasSameSignatureFile' = 'SignatureArchiveHasSameSignatureFile', // The signature archive has the same signature file.
541 > 'PackageIntegrityCheckFailed' = 'PackageIntegrityCheckFailed', // The package integrity check failed.
542 > 'SignatureIsInvalid' = 'SignatureIsInvalid', // The extension has an invalid signature file (.signature.p7s).
543 > 'SignatureManifestIsInvalid' = 'SignatureManifestIsInvalid', // The extension has an invalid signature manifest file (.signature.manifest).
544 > 'SignatureIntegrityCheckFailed' = 'SignatureIntegrityCheckFailed', // The extension's signature integrity check failed. Extension integrity is suspect.
545 > 'EntryIsMissing' = 'EntryIsMissing', // An entry referenced in the signature manifest was not found in the extension.
546 > 'EntryIsTampered' = 'EntryIsTampered', // The integrity check for an entry referenced in the signature manifest failed.
547 > 'Untrusted' = 'Untrusted', // An X.509 certificate in the extension signature is untrusted.
548 > 'CertificateRevoked' = 'CertificateRevoked', // An X.509 certificate in the extension signature has been revoked.
549 > 'SignatureIsNotValid' = 'SignatureIsNotValid', // The extension signature is invalid.
550 > 'UnknownError' = 'UnknownError', // An unknown error occurred.
551 > 'PackageIsInvalidZip' = 'PackageIsInvalidZip', // The extension package is not valid ZIP format.
552 > 'SignatureArchiveHasTooManyEntries' = 'SignatureArchiveHasTooManyEntries', // The signature archive has too many entries.
553 > }
554 >
555 > export class ExtensionManagementError extends Error {
556 > constructor(message: string, readonly code: ExtensionManagementErrorCode) {
557 super(message);
558 this.name = code;
559 }
561 >
562 > export interface InstallExtensionSummary {
563 > failed: {
564 > id: string;
565 > installOptions: InstallOptions;
566 > }[];
567 > }
568 >
569 > export type InstallOptions = {
570 > isBuiltin?: boolean;
571 > isWorkspaceScoped?: boolean;
572 > isMachineScoped?: boolean;
573 > isApplicationScoped?: boolean;
574 > pinned?: boolean;
575 > donotIncludePackAndDependencies?: boolean;
576 > installGivenVersion?: boolean;
577 > preRelease?: boolean;
578 > installPreReleaseVersion?: boolean;
579 > donotVerifySignature?: boolean;
580 > operation?: InstallOperation;
581 > profileLocation?: URI;
582 > productVersion?: IProductVersion;
583 > keepExisting?: boolean;
584 > downloadExtensionsLocally?: boolean;
585 > /**
586 > * Context passed through to InstallExtensionResult
587 > */
588 > context?: IStringDictionary<unknown>;
589 > };
590 >
591 > export type UninstallOptions = {
592 > readonly profileLocation?: URI;
593 > readonly donotIncludePack?: boolean;
594 > readonly donotCheckDependents?: boolean;
595 > readonly versionOnly?: boolean;
596 > readonly remove?: boolean;
597 > };
598 >
599 > export interface IExtensionManagementParticipant {
600 > postInstall(local: ILocalExtension, source: URI | IGalleryExtension, options: InstallOptions, token: CancellationToken): Promise<void>;
601 > postUninstall(local: ILocalExtension, options: UninstallOptions, token: CancellationToken): Promise<void>;
602 > }
603 >
604 > export type InstallExtensionInfo = { readonly extension: IGalleryExtension; readonly options: InstallOptions };
605 > export type UninstallExtensionInfo = { readonly extension: ILocalExtension; readonly options?: UninstallOptions };
606 >
607 > export const IExtensionManagementService = createDecorator<IExtensionManagementService>('extensionManagementService');
608 > export interface IExtensionManagementService {
609 > readonly _serviceBrand: undefined;
610 >
611 > readonly preferPreReleases: boolean;
612 >
613 > onInstallExtension: Event<InstallExtensionEvent>;
614 > onDidInstallExtensions: Event<readonly InstallExtensionResult[]>;
615 > onUninstallExtension: Event<UninstallExtensionEvent>;
616 > onDidUninstallExtension: Event<DidUninstallExtensionEvent>;
617 > onDidUpdateExtensionMetadata: Event<DidUpdateExtensionMetadata>;
618 >
619 > zip(extension: ILocalExtension): Promise<URI>;
620 > getManifest(vsix: URI): Promise<IExtensionManifest>;
621 > install(vsix: URI, options?: InstallOptions): Promise<ILocalExtension>;
622 > canInstall(extension: IGalleryExtension): Promise<true | IMarkdownString>;
623 > installFromGallery(extension: IGalleryExtension, options?: InstallOptions): Promise<ILocalExtension>;
624 > installGalleryExtensions(extensions: InstallExtensionInfo[]): Promise<InstallExtensionResult[]>;
625 > installFromLocation(location: URI, profileLocation: URI): Promise<ILocalExtension>;
626 > installExtensionsFromProfile(extensions: IExtensionIdentifier[], fromProfileLocation: URI, toProfileLocation: URI): Promise<ILocalExtension[]>;
627 > uninstall(extension: ILocalExtension, options?: UninstallOptions): Promise<void>;
628 > uninstallExtensions(extensions: UninstallExtensionInfo[]): Promise<void>;
629 > toggleApplicationScope(extension: ILocalExtension, fromProfileLocation: URI): Promise<ILocalExtension>;
630 > getInstalled(type?: ExtensionType, profileLocation?: URI, productVersion?: IProductVersion, language?: string): Promise<ILocalExtension[]>;
631 > getExtensionsControlManifest(): Promise<IExtensionsControlManifest>;
632 > copyExtensions(fromProfileLocation: URI, toProfileLocation: URI): Promise<void>;
633 > updateMetadata(local: ILocalExtension, metadata: Partial<Metadata>, profileLocation: URI): Promise<ILocalExtension>;
634 > resetPinnedStateForAllUserExtensions(pinned: boolean): Promise<void>;
635 >
636 > download(extension: IGalleryExtension, operation: InstallOperation, donotVerifySignature: boolean): Promise<URI>;
637 >
638 > registerParticipant(pariticipant: IExtensionManagementParticipant): void;
639 > getTargetPlatform(): Promise<TargetPlatform>;
640 >
641 > cleanUp(): Promise<void>;
642 > }
643 >
644 > export const DISABLED_EXTENSIONS_STORAGE_PATH = 'extensionsIdentifiers/disabled';
645 > export const ENABLED_EXTENSIONS_STORAGE_PATH = 'extensionsIdentifiers/enabled';
646 > export const IGlobalExtensionEnablementService = createDecorator<IGlobalExtensionEnablementService>('IGlobalExtensionEnablementService');
647 >
648 > export interface IGlobalExtensionEnablementService {
649 > readonly _serviceBrand: undefined;
650 > readonly onDidChangeEnablement: Event<{ readonly extensions: IExtensionIdentifier[]; readonly source?: string }>;
651 >
652 > getDisabledExtensions(): IExtensionIdentifier[];
653 > enableExtension(extension: IExtensionIdentifier, source?: string): Promise<boolean>;
654 > disableExtension(extension: IExtensionIdentifier, source?: string): Promise<boolean>;
655 >
656 > }
657 >
658 > export type IConfigBasedExtensionTip = {
659 > readonly extensionId: string;
660 > readonly extensionName: string;
661 > readonly isExtensionPack: boolean;
662 > readonly configName: string;
663 > readonly important: boolean;
664 > readonly whenNotInstalled?: string[];
665 > };
666 >
667 > export type IExecutableBasedExtensionTip = {
668 > readonly extensionId: string;
669 > readonly extensionName: string;
670 > readonly isExtensionPack: boolean;
671 > readonly exeName: string;
672 > readonly exeFriendlyName: string;
673 > readonly windowsPath?: string;
674 > readonly whenNotInstalled?: string[];
675 > };
676 >
677 > export const IExtensionTipsService = createDecorator<IExtensionTipsService>('IExtensionTipsService');
678 > export interface IExtensionTipsService {
679 > readonly _serviceBrand: undefined;
680 >
681 > getConfigBasedTips(folder: URI): Promise<IConfigBasedExtensionTip[]>;
682 > getImportantExecutableBasedTips(): Promise<IExecutableBasedExtensionTip[]>;
683 > getOtherExecutableBasedTips(): Promise<IExecutableBasedExtensionTip[]>;
684 > }
685 >
686 > export type AllowedExtensionsConfigValueType = IStringDictionary<boolean | string | string[]>;
687 >
688 > export const IAllowedExtensionsService = createDecorator<IAllowedExtensionsService>('IAllowedExtensionsService');
689 > export interface IAllowedExtensionsService {
690 > readonly _serviceBrand: undefined;
691 >
692 > readonly allowedExtensionsConfigValue: AllowedExtensionsConfigValueType | undefined;
693 > readonly onDidChangeAllowedExtensionsConfigValue: Event<void>;
694 >
695 > isAllowed(extension: IGalleryExtension | IExtension): true | IMarkdownString;
696 > isAllowed(extension: { id: string; publisherDisplayName: string | undefined; version?: string; prerelease?: boolean; targetPlatform?: TargetPlatform }): true | IMarkdownString;
697 > }
698 >
699 export async function computeSize(location: URI, fileService: IFileService): Promise<number> {
700 let stat: IFileStat;
713 return stat.size ?? 0;
714 }
716 > export const ExtensionsLocalizedLabel = localize2('extensions', "Extensions");
717 > export const PreferencesLocalizedLabel = localize2('preferences', 'Preferences');
718 > export const AllowedExtensionsConfigKey = 'extensions.allowed';
719 > export const VerifyExtensionSignatureConfigKey = 'extensions.verifySignature';
720 > export const ExtensionRequestsTimeoutConfigKey = 'extensions.requestTimeout';
721 >
722 > Registry.as<IConfigurationRegistry>(Extensions.Configuration)
723 > .registerConfiguration({
724 > id: 'extensions',
725 > order: 30,
726 > title: localize('extensionsConfigurationTitle', "Extensions"),
727 > type: 'object',
728 > properties: {
729 > [AllowedExtensionsConfigKey]: {
730 > // Note: Type is set only to object because to support policies generation during build time, where single type is expected.
731 > type: 'object',
732 > markdownDescription: localize('extensions.allowed', "Specify a list of extensions that are allowed to use. This helps maintain a secure and consistent development environment by restricting the use of unauthorized extensions. For more information on how to configure this setting, please visit the [Configure Allowed Extensions](https://aka.ms/vscode/enterprise/extensions/allowed) section."),
733 > default: '*',
734 > defaultSnippets: [{
735 > body: {},
736 > description: localize('extensions.allowed.none', "No extensions are allowed."),
737 > }, {
738 > body: {
739 > '*': true
740 > },
741 > description: localize('extensions.allowed.all', "All extensions are allowed."),
742 > }],
743 > scope: ConfigurationScope.APPLICATION,
744 > policy: {
745 > name: 'AllowedExtensions',
746 > category: PolicyCategory.Extensions,
747 > minimumVersion: '1.96',
748 > localization: {
749 > description: {
750 > key: 'extensions.allowed.policy',
751 > value: localize('extensions.allowed.policy', "Specify a list of extensions that are allowed to use. This helps maintain a secure and consistent development environment by restricting the use of unauthorized extensions. More information: https://aka.ms/vscode/enterprise/extensions/allowed"),
752 > }
753 > }
754 > },
755 > additionalProperties: false,
756 > patternProperties: {
757 > '([a-z0-9A-Z][a-z0-9-A-Z]*)\\.([a-z0-9A-Z][a-z0-9-A-Z]*)$': {
758 > anyOf: [
759 > {
760 > type: ['boolean', 'string'],
761 > enum: [true, false, 'stable'],
762 > description: localize('extensions.allow.description', "Allow or disallow the extension."),
763 > enumDescriptions: [
764 > localize('extensions.allowed.enable.desc', "Extension is allowed."),
765 > localize('extensions.allowed.disable.desc', "Extension is not allowed."),
766 > localize('extensions.allowed.disable.stable.desc', "Allow only stable versions of the extension."),
767 > ],
768 > },
769 > {
770 > type: 'array',
771 > items: {
772 > type: 'string',
773 > },
774 > description: localize('extensions.allow.version.description', "Allow or disallow specific versions of the extension. To specifcy a platform specific version, use the format `[email protected]`, e.g. `[email protected]`. Supported platforms are `win32-x64`, `win32-arm64`, `linux-x64`, `linux-arm64`, `linux-armhf`, `alpine-x64`, `alpine-arm64`, `darwin-x64`, `darwin-arm64`"),
775 > },
776 > ]
777 > },
778 > '([a-z0-9A-Z][a-z0-9-A-Z]*)$': {
779 > type: ['boolean', 'string'],
780 > enum: [true, false, 'stable'],
781 > description: localize('extension.publisher.allow.description', "Allow or disallow all extensions from the publisher."),
782 > enumDescriptions: [
783 > localize('extensions.publisher.allowed.enable.desc', "All extensions from the publisher are allowed."),
784 > localize('extensions.publisher.allowed.disable.desc', "All extensions from the publisher are not allowed."),
785 > localize('extensions.publisher.allowed.disable.stable.desc', "Allow only stable versions of the extensions from the publisher."),
786 > ],
787 > },
788 > '\\*': {
789 > type: 'boolean',
790 > enum: [true, false],
791 > description: localize('extensions.allow.all.description', "Allow or disallow all extensions."),
792 > enumDescriptions: [
793 > localize('extensions.allow.all.enable', "Allow all extensions."),
794 > localize('extensions.allow.all.disable', "Disallow all extensions.")
795 > ],
796 > }
797 > }
798 > }
799 > }
800 > });
801 >
802 > export function shouldRequireRepositorySignatureFor(isPrivate: boolean, galleryManifest: IExtensionGalleryManifest | null): boolean {
803 if (isPrivate) {
804 return galleryManifest?.capabilities.signing?.allPrivateRepositorySigned === true;
src/vs/workbench/services/chat/common/chatEntitlementService.ts 651 covered LOC · 52 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatEntitlementService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import product from '../../../../platform/product/common/product.js';
7 > import { Barrier } from '../../../../base/common/async.js';
8 > import { CancellationToken, CancellationTokenSource } from '../../../../base/common/cancellation.js';
9 > import { Emitter, Event } from '../../../../base/common/event.js';
10 > import { Lazy } from '../../../../base/common/lazy.js';
11 > import { Disposable, MutableDisposable } from '../../../../base/common/lifecycle.js';
12 > import { IRequestContext } from '../../../../base/parts/request/common/request.js';
13 > import { localize } from '../../../../nls.js';
14 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
15 > import { IContextKey, IContextKeyService, RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
16 > import { IDialogService } from '../../../../platform/dialogs/common/dialogs.js';
17 > import { createDecorator, IInstantiationService } from '../../../../platform/instantiation/common/instantiation.js';
18 > import { ILogService, LogLevel } from '../../../../platform/log/common/log.js';
19 > import { IProductService } from '../../../../platform/product/common/productService.js';
20 > import { asText, IRequestService } from '../../../../platform/request/common/request.js';
21 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
22 > import { ITelemetryService, TelemetryLevel } from '../../../../platform/telemetry/common/telemetry.js';
23 > import { AuthenticationSession, IAuthenticationService } from '../../authentication/common/authentication.js';
24 > import { IOpenerService } from '../../../../platform/opener/common/opener.js';
25 > import { URI } from '../../../../base/common/uri.js';
26 > import Severity from '../../../../base/common/severity.js';
27 > import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.js';
28 > import { isWeb } from '../../../../base/common/platform.js';
29 > import { ILifecycleService } from '../../lifecycle/common/lifecycle.js';
30 > import { Mutable } from '../../../../base/common/types.js';
31 > import { InstantiationType, registerSingleton } from '../../../../platform/instantiation/common/extensions.js';
32 > import { IObservable, observableFromEvent } from '../../../../base/common/observable.js';
33 > import { IDefaultAccountService } from '../../../../platform/defaultAccount/common/defaultAccount.js';
34 > import { IDefaultAccount, IEntitlementsData } from '../../../../base/common/defaultAccount.js';
35 >
36 > export namespace ChatEntitlementContextKeys {
37 >
38 > export const Setup = {
39 > hidden: new RawContextKey<boolean>('chatSetupHidden', false, true), // True when chat setup is explicitly hidden.
40 > installed: new RawContextKey<boolean>('chatSetupInstalled', false, true), // True when the chat extension is installed and enabled.
41 > disabled: new RawContextKey<boolean>('chatSetupDisabled', false, true), // True when the chat extension is disabled due to any other reason than workspace trust.
42 > disabledInWorkspace: new RawContextKey<boolean>('chatSetupDisabledInWorkspace', false, true), // True when chat is disabled at the workspace level via settings.
43 > untrusted: new RawContextKey<boolean>('chatSetupUntrusted', false, true), // True when the chat extension is disabled due to workspace trust.
44 > later: new RawContextKey<boolean>('chatSetupLater', false, true), // True when the user wants to finish setup later.
45 > registered: new RawContextKey<boolean>('chatSetupRegistered', false, true), // True when the user has registered as Free or Pro user.
46 > completed: new RawContextKey<boolean>('chatSetupCompleted', false, true) // True when the user has completed the setup flow, regardless of the outcome.
47 > };
48 >
49 > export const Entitlement = {
50 > signedOut: new RawContextKey<boolean>('chatEntitlementSignedOut', false, true), // True when user is signed out.
51 > canSignUp: new RawContextKey<boolean>('chatPlanCanSignUp', false, true), // True when user can sign up to be a chat free user.
52 >
53 > planFree: new RawContextKey<boolean>('chatPlanFree', false, true), // True when user is a chat free user.
54 > planPro: new RawContextKey<boolean>('chatPlanPro', false, true), // True when user is a chat pro user.
55 > planEdu: new RawContextKey<boolean>('chatPlanEdu', false, true), // True when user is a chat edu user.
56 > planProPlus: new RawContextKey<boolean>('chatPlanProPlus', false, true), // True when user is a chat pro plus user.
57 > planMax: new RawContextKey<boolean>('chatPlanMax', false, true), // True when user is a chat max user.
58 > planBusiness: new RawContextKey<boolean>('chatPlanBusiness', false, true), // True when user is a chat business user.
59 > planEnterprise: new RawContextKey<boolean>('chatPlanEnterprise', false, true), // True when user is a chat enterprise user.
60 >
61 > organisations: new RawContextKey<string[]>('chatEntitlementOrganisations', undefined, true), // The organizations the user belongs to.
62 > internal: new RawContextKey<boolean>('chatEntitlementInternal', false, true), // True when user belongs to internal organisation.
63 > sku: new RawContextKey<string>('chatEntitlementSku', undefined, true), // The SKU of the user.
64 > };
65 >
66 > export const chatQuotaExceeded = new RawContextKey<boolean>('chatQuotaExceeded', false, true);
67 > export const completionsQuotaExceeded = new RawContextKey<boolean>('completionsQuotaExceeded', false, true);
68 >
69 > export const chatAnonymous = new RawContextKey<boolean>('chatAnonymous', false, true);
70 >
71 > export const clientByokEnabled = new RawContextKey<boolean>('github.copilot.clientByokEnabled', true, true);
72 >
73 > export const hasByokModels = new RawContextKey<boolean>('github.copilot.hasByokModels', false, true);
74 > }
75 >
76 > export const IChatEntitlementService = createDecorator<IChatEntitlementService>('chatEntitlementService');
77 >
78 > export enum ChatEntitlement {
79 > /** Signed out */
80 > Unknown = 1,
81 > /** Signed in but not yet resolved */
82 > Unresolved = 2,
83 > /** Signed in and entitled to Free */
84 > Available = 3,
85 > /** Signed in but not entitled to Free */
86 > Unavailable = 4,
87 > /** Signed-up to Free */
88 > Free = 5,
89 > /** Signed-up to EDU */
90 > EDU = 10,
91 > /** Signed-up to Pro */
92 > Pro = 6,
93 > /** Signed-up to Pro Plus */
94 > ProPlus = 7,
95 > /** Signed-up to Business */
96 > Business = 8,
97 > /** Signed-up to Enterprise */
98 > Enterprise = 9,
99 > /** Signed-up to Max */
100 > Max = 11,
101 > }
102 >
103 > export interface IChatSentiment {
104 >
105 > /**
106 > * Whether the user has completed the setup flow or not, regardless of the outcome
107 > */
108 > completed?: boolean;
109 >
110 > /**
111 > * User has Chat installed.
112 > */
113 > installed?: boolean;
114 >
115 > /**
116 > * User signals no intent in using Chat.
117 > *
118 > * Note: in contrast to `disabled`, this should not only disable
119 > * Chat but also hide all of its UI.
120 > */
121 > hidden?: boolean;
122 >
123 > /**
124 > * User signals intent to disable Chat.
125 > *
126 > * Note: in contrast to `hidden`, this should not hide
127 > * Chat but but disable its functionality.
128 > */
129 > disabled?: boolean;
130 >
131 > /**
132 > * Chat is disabled at the workspace level
133 > *
134 > * Note: in contrast to `hidden` (which hides all UI globally),
135 > * this only disables Chat in the current workspace while
136 > * keeping its UI visible so the user can re-enable it.
137 > */
138 > disabledInWorkspace?: boolean;
139 >
140 > /**
141 > * Chat is disabled due to missing workspace trust.
142 > *
143 > * Note: even though this disables Chat, we want to treat it
144 > * different from the `disabled` state that is by explicit
145 > * user choice.
146 > */
147 > untrusted?: boolean;
148 >
149 > /**
150 > * User signals intent to use Chat later.
151 > */
152 > later?: boolean;
153 >
154 > /**
155 > * User has registered as Free or Pro user.
156 > */
157 > registered?: boolean;
158 > }
159 >
160 > /**
161 > * The inputs needed to decide whether Chat still requires the user to run setup
162 > * (sign in / sign up / trust / enable) before it can service a request.
163 > */
164 > export interface IChatSetupRequirement {
165 > /** Whether the setup flow has been completed (any outcome). */
166 > readonly completed: boolean;
167 > /** Whether the chat extension is disabled for a reason other than trust. */
168 > readonly disabled: boolean;
169 > /** Whether the chat extension is disabled because the workspace is untrusted. */
170 > readonly untrusted: boolean;
171 > /** The user's last known or resolved entitlement. */
172 > readonly entitlement: ChatEntitlement;
173 > /** Whether anonymous (signed-out) Chat access is enabled. */
174 > readonly anonymous: boolean;
175 > /** Whether BYOK models are available. */
176 > readonly hasByokModels: boolean;
177 > }
178 >
179 > /**
180 > * Single source of truth for whether Chat still requires setup before it can
181 > * service a request. Shared by the setup agent (which routes a sent message
182 > * through setup) and the model picker (which surfaces a "Sign in to use Copilot"
183 > * state instead of a misleading lone "Auto"). BYOK models and anonymous access
184 > * intentionally satisfy the entitlement-based checks so those flows keep working.
185 > */
186 > export function chatRequiresSetup(context: IChatSetupRequirement): boolean {
187 return (
188 (!context.completed && !context.hasByokModels) || // Setup not completed (unless BYOK models are available)
197 );
198 }
200 > export interface IChatEntitlementService {
201 >
202 > _serviceBrand: undefined;
203 >
204 > readonly onDidChangeEntitlement: Event<void>;
205 >
206 > readonly entitlement: ChatEntitlement;
207 > readonly entitlementObs: IObservable<ChatEntitlement>;
208 >
209 > readonly clientByokEnabled: boolean;
210 > readonly hasByokModels: boolean;
211 >
212 > readonly organisations: string[] | undefined;
213 > readonly isInternal: boolean;
214 > readonly sku: string | undefined;
215 > readonly copilotTrackingId: string | undefined;
216 >
217 > readonly onDidChangeQuotaExceeded: Event<void>;
218 > readonly onDidChangeQuotaRemaining: Event<void>;
219 > readonly onDidChangeUsageBasedBilling: Event<void>;
220 >
221 > readonly quotas: IQuotas;
222 >
223 > readonly onDidChangeSentiment: Event<void>;
224 >
225 > readonly sentiment: IChatSentiment;
226 > readonly sentimentObs: IObservable<IChatSentiment>;
227 >
228 > // TODO@bpasero eventually this will become enabled by default
229 > // and in that case we only need to check on entitlements change
230 > // between `unknown` and any other entitlement.
231 > readonly onDidChangeAnonymous: Event<void>;
232 > readonly anonymous: boolean;
233 > readonly anonymousObs: IObservable<boolean>;
234 >
235 > acceptQuotas(quotas: IQuotas): void;
236 >
237 > /**
238 > * Clear all quota state.
239 > */
240 > clearQuotas(): void;
241 >
242 > markAnonymousRateLimited(): void;
243 >
244 > /**
245 > * Mark the chat setup flow as completed.
246 > */
247 > markSetupCompleted(): void;
248 >
249 > /**
250 > * Force the hidden state on or off, overriding the normal entitlement logic.
251 > * Used by the account policy gate to hide all AI features when the gate is
252 > * active and unsatisfied.
253 > */
254 > setForceHidden(hidden: boolean): void;
255 >
256 > update(token: CancellationToken): Promise<void>;
257 > }
258 >
259 > //#region Helper Functions
260 >
261 > /**
262 > * Checks the chat entitlements to see if the user falls into the paid category
263 > * @param chatEntitlement The chat entitlement to check
264 > * @returns Whether or not they are a paid user
265 > */
266 > export function isProUser(chatEntitlement: ChatEntitlement): boolean {
267 return chatEntitlement === ChatEntitlement.EDU ||
268 chatEntitlement === ChatEntitlement.Pro ||
272 chatEntitlement === ChatEntitlement.Enterprise;
273 }
275 > /**
276 > * Gets the full plan name for the given chat entitlement
277 > * @param chatEntitlement The chat entitlement to get the plan name for
278 > * @returns The localized full plan name (e.g., "Copilot Pro", "Copilot Free")
279 > */
280 > export function getChatPlanName(chatEntitlement: ChatEntitlement): string {
281 switch (chatEntitlement) {
282 case ChatEntitlement.EDU:
296 }
297 }
299 > //#region Service Implementation
300 >
301 > const defaultChatAgent = {
302 > upgradePlanUrl: product.defaultChatAgent?.upgradePlanUrl ?? '',
303 > providerUriSetting: product.defaultChatAgent?.providerUriSetting ?? '',
304 > entitlementSignupLimitedUrl: product.defaultChatAgent?.entitlementSignupLimitedUrl ?? '',
305 > chatQuotaExceededContext: product.defaultChatAgent?.chatQuotaExceededContext ?? '',
306 > completionsQuotaExceededContext: product.defaultChatAgent?.completionsQuotaExceededContext ?? ''
307 > };
308 >
309 > interface IChatQuotasAccessor {
310 > clearQuotas(): void;
311 > acceptQuotas(quotas: IQuotas): void;
312 > }
313 >
314 > const CHAT_ALLOW_ANONYMOUS_CONFIGURATION_KEY = 'chat.allowAnonymousAccess';
315 >
316 function isAnonymous(configurationService: IConfigurationService, entitlement: ChatEntitlement, sentiment: IChatSentiment): boolean {
317 if (configurationService.getValue(CHAT_ALLOW_ANONYMOUS_CONFIGURATION_KEY) !== true) {
329 return true;
330 }
332 > type ChatEntitlementClassification = {
333 > owner: 'bpasero';
334 > comment: 'Provides insight into chat entitlements.';
335 > chatHidden: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether chat is hidden or not.' };
336 > chatEntitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The current chat entitlement of the user.' };
337 > chatAnonymous: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user is anonymously using chat.' };
338 > chatRegistered: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user is registered for chat.' };
339 > chatDisabled: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether chat is disabled or not.' };
340 > };
341 > type ChatEntitlementEvent = {
342 > chatHidden: boolean;
343 > chatEntitlement: ChatEntitlement;
344 > chatAnonymous: boolean;
345 > chatRegistered: boolean;
346 > chatDisabled: boolean;
347 > };
348 >
349 function logChatEntitlements(state: IChatEntitlementContextState, configurationService: IConfigurationService, telemetryService: ITelemetryService): void {
350 telemetryService.publicLog2<ChatEntitlementEvent, ChatEntitlementClassification>('chatEntitlements', {
356 });
357 }
359 > type ChatAdditionalSpendConfigurationClassification = {
360 > owner: 'pwang347';
361 > comment: 'Tracks when a user enables or disables additional spend.';
362 > enabled: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether additional spend is now enabled or disabled.' };
363 > entitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The current chat entitlement of the user.' };
364 > };
365 > type ChatAdditionalSpendConfigurationEvent = {
366 > enabled: boolean;
367 > entitlement: ChatEntitlement;
368 > };
369 >
370 > type ChatAdditionalSpendActiveClassification = {
371 > owner: 'pwang347';
372 > comment: 'Tracks when a user enters additional spend (included quota exhausted while additional spend is enabled).';
373 > entitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The current chat entitlement of the user.' };
374 > additionalUsageCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The number of additional spend interactions used so far.' };
375 > };
376 > type ChatAdditionalSpendActiveEvent = {
377 > entitlement: ChatEntitlement;
378 > additionalUsageCount: number;
379 > };
380 >
381 > export class ChatEntitlementService extends Disposable implements IChatEntitlementService {
382 >
383 > declare _serviceBrand: undefined;
384 >
385 > private static readonly CACHED_UBB_STORAGE_KEY = 'chat.usageBasedBilling';
386 >
387 > readonly context: Lazy<ChatEntitlementContext> | undefined;
388 > readonly requests: Lazy<ChatEntitlementRequests> | undefined;
389 >
390 > constructor(
391 @IInstantiationService instantiationService: IInstantiationService,
392 @IProductService productService: IProductService,
467 this.registerListeners();
468 }
470 > //#region --- Entitlements
471 >
472 > readonly onDidChangeEntitlement: Event<void>;
473 > readonly entitlementObs: IObservable<ChatEntitlement>;
474 >
475 > get entitlement(): ChatEntitlement {
476 if (this.contextKeyService.getContextKeyValue<boolean>(ChatEntitlementContextKeys.Entitlement.planEdu.key) === true) {
477 return ChatEntitlement.EDU;
496 return ChatEntitlement.Unresolved;
497 }
499 > get isInternal(): boolean {
500 return this.contextKeyService.getContextKeyValue<boolean>(ChatEntitlementContextKeys.Entitlement.internal.key) === true;
501 }
503 > get organisations(): string[] | undefined {
504 return this.contextKeyService.getContextKeyValue<string[]>(ChatEntitlementContextKeys.Entitlement.organisations.key);
505 }
507 > get sku(): string | undefined {
508 return this.contextKeyService.getContextKeyValue<string>(ChatEntitlementContextKeys.Entitlement.sku.key);
509 }
511 > get copilotTrackingId(): string | undefined {
512 return this.context?.value.state.copilotTrackingId;
513 }
515 > get clientByokEnabled(): boolean {
516 return this.contextKeyService.getContextKeyValue<boolean>('github.copilot.clientByokEnabled') === true;
517 }
519 > get hasByokModels(): boolean {
520 return this.contextKeyService.getContextKeyValue<boolean>('github.copilot.hasByokModels') === true;
521 }
523 > //#endregion
524 >
525 > //#region --- Quotas
526 >
527 > private readonly _onDidChangeQuotaExceeded = this._register(new Emitter<void>());
528 > readonly onDidChangeQuotaExceeded = this._onDidChangeQuotaExceeded.event;
529 >
530 > private readonly _onDidChangeQuotaRemaining = this._register(new Emitter<void>());
531 > readonly onDidChangeQuotaRemaining = this._onDidChangeQuotaRemaining.event;
532 >
533 > private readonly _onDidChangeUsageBasedBilling = this._register(new Emitter<void>());
534 > readonly onDidChangeUsageBasedBilling = this._onDidChangeUsageBasedBilling.event;
535 >
536 > private _quotas: IQuotas;
537 > private quotaCopilotTrackingId: string | undefined;
538 > get quotas() { return this._quotas; }
539 >
540 > private readonly chatQuotaExceededContextKey: IContextKey<boolean>;
541 > private readonly completionsQuotaExceededContextKey: IContextKey<boolean>;
542 >
543 > private ExtensionQuotaContextKeys = {
544 > chatQuotaExceeded: defaultChatAgent.chatQuotaExceededContext,
545 > completionsQuotaExceeded: defaultChatAgent.completionsQuotaExceededContext,
546 > };
547 >
548 > private registerListeners(): void {
549 const quotaExceededSet = new Set([this.ExtensionQuotaContextKeys.chatQuotaExceeded, this.ExtensionQuotaContextKeys.completionsQuotaExceeded]);
550
585 this._register(this.onDidChangeSentiment(() => updateAnonymousUsage()));
586 }
588 > acceptQuotas(incomingQuotas: IQuotas): void {
589 const oldQuota = this._quotas;
590 const cachedQuota = this.quotaCopilotTrackingId === this.copilotTrackingId ? oldQuota : {};
649 }
650 }
652 > private compareQuotas(oldQuota: IQuotaSnapshot | undefined, newQuota: IQuotaSnapshot | undefined): { changed: { exceeded: boolean; remaining: boolean } } {
653 return {
654 changed: {
659 };
660 }
662 > clearQuotas(): void {
663 this.acceptQuotas({});
664 }
666 > private updateContextKeys(): void {
667 const chatExhausted = this._quotas.chat?.percentRemaining === 0;
668 const premiumChatExhausted = this._quotas.premiumChat?.unlimited
677 this.completionsQuotaExceededContextKey.set(this._quotas.completions?.percentRemaining === 0);
678 }
680 > //#endregion
681 >
682 > //#region --- Sentiment
683 >
684 > readonly onDidChangeSentiment: Event<void>;
685 > readonly sentimentObs: IObservable<IChatSentiment>;
686 >
687 > get sentiment(): IChatSentiment {
688 return {
689 completed: this.contextKeyService.getContextKeyValue<boolean>(ChatEntitlementContextKeys.Setup.completed.key) === true,
697 };
698 }
700 > //#endregion
701 >
702 > //region --- Anonymous
703 >
704 > private readonly anonymousContextKey: IContextKey<boolean>;
705 >
706 > private readonly _onDidChangeAnonymous = this._register(new Emitter<void>());
707 > readonly onDidChangeAnonymous = this._onDidChangeAnonymous.event;
708 >
709 > readonly anonymousObs = observableFromEvent(this.onDidChangeAnonymous, () => this.anonymous);
710 >
711 > get anonymous(): boolean {
712 return isAnonymous(this.configurationService, this.entitlement, this.sentiment);
713 }
715 > //#endregion
716 >
717 > markAnonymousRateLimited(): void {
718 if (!this.anonymous) {
719 return;
723 this._onDidChangeQuotaExceeded.fire();
724 }
726 > markSetupCompleted(): void {
727 this.context?.value.update({ completed: true });
728 }
730 > setForceHidden(hidden: boolean): void {
731 if (this.context) {
732 this.context.value.setForceHidden(hidden);
737 }
738 }
740 > async update(token: CancellationToken): Promise<void> {
741 await this.requests?.value.forceResolveEntitlement(token);
742 }
744 >
745 > //#endregion
746 >
747 > //#region Chat Entitlement Request Service
748 >
749 > type EntitlementClassification = {
750 > tid: { classification: 'EndUserPseudonymizedInformation'; purpose: 'BusinessInsight'; comment: 'The anonymized analytics id returned by the service'; endpoint: 'GoogleAnalyticsId' };
751 > entitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Flag indicating the chat entitlement state' };
752 > sku: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The SKU of the chat entitlement' };
753 > quotaChatUnlimited: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user has unlimited chat requests' };
754 > quotaChatHasQuota: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user currently has chat quota available' };
755 > quotaChatEntitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The raw chat quota entitlement count' };
756 > quotaPremiumChat: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The percentage of premium chat requests remaining for the user' };
757 > quotaPremiumChatUnlimited: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user has unlimited premium chat requests' };
758 > quotaPremiumChatHasQuota: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user currently has premium chat quota available' };
759 > quotaPremiumChatEntitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The raw premium chat quota entitlement count' };
760 > quotaCompletions: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The percentage of completions remaining for the user' };
761 > quotaCompletionsUnlimited: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user has unlimited completions' };
762 > quotaCompletionsHasQuota: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user currently has completions quota available' };
763 > quotaCompletionsEntitlement: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The raw completions quota entitlement count' };
764 > quotaResetDate: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The date the quota will reset' };
765 > usageBasedBilling: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user is on usage-based billing' };
766 > additionalUsageEnabled: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether overage / additional spend is enabled' };
767 > additionalUsageCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'The number of overage interactions used' };
768 > canUpgradePlan: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user is eligible to upgrade their plan' };
769 > owner: 'bpasero';
770 > comment: 'Reporting chat entitlements';
771 > };
772 >
773 > type EntitlementEvent = {
774 > entitlement: ChatEntitlement;
775 > tid: string;
776 > sku: string | undefined;
777 > quotaChatUnlimited: boolean | undefined;
778 > quotaChatHasQuota: boolean | undefined;
779 > quotaChatEntitlement: number | undefined;
780 > quotaPremiumChat: number | undefined;
781 > quotaPremiumChatUnlimited: boolean | undefined;
782 > quotaPremiumChatHasQuota: boolean | undefined;
783 > quotaPremiumChatEntitlement: number | undefined;
784 > quotaCompletions: number | undefined;
785 > quotaCompletionsUnlimited: boolean | undefined;
786 > quotaCompletionsHasQuota: boolean | undefined;
787 > quotaCompletionsEntitlement: number | undefined;
788 > quotaResetDate: string | undefined;
789 > usageBasedBilling: boolean | undefined;
790 > additionalUsageEnabled: boolean | undefined;
791 > additionalUsageCount: number | undefined;
792 > canUpgradePlan: boolean | undefined;
793 > };
794 >
795 > interface IEntitlements {
796 > readonly entitlement: ChatEntitlement;
797 > readonly organisations?: string[];
798 > readonly sku?: string;
799 > readonly copilotTrackingId?: string;
800 > readonly quotas?: IQuotas;
801 > }
802 >
803 > export interface IQuotaSnapshot {
804 > readonly percentRemaining: number;
805 > readonly unlimited: boolean;
806 > readonly hasQuota?: boolean;
807 > readonly resetAt?: number;
808 > readonly usageBasedBilling?: boolean;
809 > readonly entitlement?: number;
810 > readonly quotaRemaining?: number;
811 > readonly creditsUsed?: number;
812 > }
813 >
814 > export interface IRateLimitSnapshot {
815 > readonly percentRemaining: number;
816 > readonly unlimited: boolean;
817 > readonly resetDate?: string;
818 > }
819 >
820 > interface IQuotas {
821 > readonly resetDate?: string;
822 > readonly resetDateHasTime?: boolean;
823 >
824 > readonly usageBasedBilling?: boolean;
825 > readonly canUpgradePlan?: boolean;
826 >
827 > readonly chat?: IQuotaSnapshot;
828 > readonly completions?: IQuotaSnapshot;
829 > readonly premiumChat?: IQuotaSnapshot;
830 > readonly additionalUsageEnabled?: boolean;
831 > readonly additionalUsageCount?: number;
832 > readonly additionalUsageEntitlement?: number;
833 >
834 > readonly sessionRateLimit?: IRateLimitSnapshot;
835 > readonly weeklyRateLimit?: IRateLimitSnapshot;
836 > }
837 >
838 function mergeDefinedSnapshot<T extends object>(previous: T | undefined, current: T): T {
839 const result = { ...previous, ...current };
845 return result;
846 }
848 > export function parseQuotas(entitlementsData: IEntitlementsData): IQuotas {
849 const quotas: Mutable<IQuotas> = {
850 resetDate: entitlementsData.quota_reset_date_utc ?? entitlementsData.quota_reset_date ?? entitlementsData.limited_user_reset_date,
919 return quotas;
920 }
922 > export class ChatEntitlementRequests extends Disposable {
923 >
924 > private state: IEntitlements;
925 >
926 > private pendingResolveCts = new CancellationTokenSource();
927 >
928 > constructor(
929 private readonly context: ChatEntitlementContext,
930 private readonly chatQuotasAccessor: IChatQuotasAccessor,
946 this.resolve();
947 }
949 > private registerListeners(): void {
950 this._register(this.defaultAccountService.onDidChangeDefaultAccount(() => this.resolve()));
951
959 }));
960 }
962 > private async resolve(): Promise<void> {
963 this.pendingResolveCts.dispose(true);
964 const cts = this.pendingResolveCts = new CancellationTokenSource();
989 }
990 }
992 > private async resolveEntitlement(defaultAccount: IDefaultAccount, token: CancellationToken): Promise<IEntitlements | undefined> {
993 const entitlements = await this.doResolveEntitlement(defaultAccount, token);
994 if (typeof entitlements?.entitlement === 'number' && !token.isCancellationRequested) {
997 return entitlements;
998 }
1000 > private async doResolveEntitlement(defaultAccount: IDefaultAccount, token: CancellationToken): Promise<IEntitlements | undefined> {
1001 if (token.isCancellationRequested) {
1002 return undefined;
1065 return entitlements;
1066 }
1068 > private toQuotas(entitlementsData: IEntitlementsData): IQuotas {
1069 return parseQuotas(entitlementsData);
1070 }
1072 > private async request(url: string, type: 'GET', body: undefined, sessions: AuthenticationSession[], token: CancellationToken, callSite: string): Promise<IRequestContext | undefined>;
1073 > private async request(url: string, type: 'POST', body: object, sessions: AuthenticationSession[], token: CancellationToken, callSite: string): Promise<IRequestContext | undefined>;
1074 > private async request(url: string, type: 'GET' | 'POST', body: object | undefined, sessions: AuthenticationSession[], token: CancellationToken, callSite: string): Promise<IRequestContext | undefined> {
1075 let lastRequest: IRequestContext | undefined;
1076
1108 return lastRequest;
1109 }
1111 > private update(state: IEntitlements): void {
1112 this.state = state;
1113
1118 }
1119 }
1121 > async forceResolveEntitlement(token = CancellationToken.None): Promise<IEntitlements | undefined> {
1122 const defaultAccount = await this.defaultAccountService.refresh({ forceRefresh: true });
1123 if (!defaultAccount) {
1127 return this.resolveEntitlement(defaultAccount, token);
1128 }
1130 > async signUpFree(): Promise<true /* signed up */ | false /* already signed up */ | { errorCode: number } /* error */ | undefined /* no session */> {
1131 const sessions = await this.getSessions();
1132 if (sessions.length === 0) {
1135 return this.doSignUpFree(sessions);
1136 }
1138 > private async doSignUpFree(sessions: AuthenticationSession[]): Promise<true /* signed up */ | false /* already signed up */ | { errorCode: number } /* error */> {
1139 const body = {
1140 restricted_telemetry: this.telemetryService.telemetryLevel === TelemetryLevel.NONE ? 'disabled' : 'enabled',
1194 return Boolean(parsedResult?.subscribed);
1195 }
1197 > private async getSessions(): Promise<AuthenticationSession[]> {
1198 const defaultAccount = await this.defaultAccountService.getDefaultAccount();
1199 if (defaultAccount) {
1206 return [...(await this.authenticationService.getSessions(this.defaultAccountService.getDefaultAccountAuthenticationProvider().id))];
1207 }
1209 > private async onUnknownSignUpError(detail: string, logMessage: string): Promise<boolean> {
1210 this.logService.error(logMessage);
1211
1223 return false;
1224 }
1226 > private onUnprocessableSignUpError(logMessage: string, logDetails: string): void {
1227 this.logService.error(logMessage);
1228
1245 }
1246 }
1248 > async signIn(options?: { useSocialProvider?: string; additionalScopes?: readonly string[] }): Promise<{ defaultAccount?: IDefaultAccount; entitlements?: IEntitlements }> {
1249 const defaultAccount = await this.defaultAccountService.signIn({
1250 additionalScopes: options?.additionalScopes,
1259 return { defaultAccount, entitlements };
1260 }
1262 > override dispose(): void {
1263 this.pendingResolveCts.dispose(true);
1264
1265 super.dispose();
1266 }
1268 >
1269 > //#endregion
1270 >
1271 > //#region Context
1272 >
1273 > export interface IChatEntitlementContextState extends IChatSentiment {
1274 >
1275 > /**
1276 > * Users last known or resolved entitlement.
1277 > */
1278 > entitlement: ChatEntitlement;
1279 >
1280 > /**
1281 > * User's last known or resolved raw SKU type.
1282 > */
1283 > sku: string | undefined;
1284 >
1285 > /**
1286 > * User's last known or resolved organisations.
1287 > */
1288 > organisations: string[] | undefined;
1289 >
1290 > /**
1291 > * User's Copilot tracking ID from the entitlement API.
1292 > */
1293 > copilotTrackingId: string | undefined;
1294 > }
1295 >
1296 > export class ChatEntitlementContext extends Disposable {
1297 >
1298 > private static readonly CHAT_ENTITLEMENT_CONTEXT_STORAGE_KEY = 'chat.setupContext';
1299 > private static readonly CHAT_ENTITLEMENT_CONTEXT_MIGRATED_STORAGE_KEY = 'chat.setupContext.migrated.v1';
1300 >
1301 > private static readonly CHAT_DISABLED_CONFIGURATION_KEY = 'chat.disableAIFeatures';
1302 >
1303 > private readonly canSignUpContextKey: IContextKey<boolean>;
1304 > private readonly signedOutContextKey: IContextKey<boolean>;
1305 >
1306 > private readonly freeContextKey: IContextKey<boolean>;
1307 > private readonly eduContextKey: IContextKey<boolean>;
1308 > private readonly proContextKey: IContextKey<boolean>;
1309 > private readonly proPlusContextKey: IContextKey<boolean>;
1310 > private readonly maxContextKey: IContextKey<boolean>;
1311 > private readonly businessContextKey: IContextKey<boolean>;
1312 > private readonly enterpriseContextKey: IContextKey<boolean>;
1313 >
1314 > private readonly organisationsContextKey: IContextKey<string[] | undefined>;
1315 > private readonly isInternalContextKey: IContextKey<boolean>;
1316 > private readonly skuContextKey: IContextKey<string | undefined>;
1317 >
1318 > private readonly completedContext: IContextKey<boolean>;
1319 > private readonly hiddenContext: IContextKey<boolean>;
1320 > private readonly disabledInWorkspaceContext: IContextKey<boolean>;
1321 > private readonly laterContext: IContextKey<boolean>;
1322 > private readonly installedContext: IContextKey<boolean>;
1323 > private readonly disabledContext: IContextKey<boolean>;
1324 > private readonly untrustedContext: IContextKey<boolean>;
1325 > private readonly registeredContext: IContextKey<boolean>;
1326 >
1327 > private _state: IChatEntitlementContextState;
1328 > private suspendedState: IChatEntitlementContextState | undefined = undefined;
1329 > get state(): IChatEntitlementContextState { return this.withConfiguration(this.suspendedState ?? this._state); }
1330 >
1331 > private readonly _onDidChange = this._register(new Emitter<void>());
1332 > readonly onDidChange = this._onDidChange.event;
1333 >
1334 > private updateBarrier: Barrier | undefined = undefined;
1335 >
1336 > constructor(
1337 @IContextKeyService contextKeyService: IContextKeyService,
1338 @IStorageService private readonly storageService: IStorageService,
1387 this.registerListeners();
1388 }
1390 > private registerListeners(): void {
1391 this._register(this.configurationService.onDidChangeConfiguration(e => {
1392 if (e.affectsConfiguration(ChatEntitlementContext.CHAT_DISABLED_CONFIGURATION_KEY)) {
1395 }));
1396 }
1398 > private _forceHidden = false;
1399 >
1400 > private withConfiguration(state: IChatEntitlementContextState): IChatEntitlementContextState {
1401 if (this._forceHidden || this.configurationService.getValue(ChatEntitlementContext.CHAT_DISABLED_CONFIGURATION_KEY) === true) {
1402 return {
1408 return state;
1409 }
1411 > setForceHidden(hidden: boolean): void {
1412 if (this._forceHidden !== hidden) {
1413 this._forceHidden = hidden;
1415 }
1416 }
1418 > update(context: { installed: boolean; disabled: boolean; untrusted: boolean; disabledInWorkspace: boolean }): Promise<void>;
1419 > update(context: { completed: true }): Promise<void>;
1420 > update(context: { hidden: false }): Promise<void>; // legacy UI state from before we had a setting to hide, keep around to still support users who used this
1421 > update(context: { later: boolean }): Promise<void>;
1422 > update(context: { entitlement: ChatEntitlement; organisations: string[] | undefined; sku: string | undefined; copilotTrackingId: string | undefined }): Promise<void>;
1423 > async update(context: { completed?: boolean; installed?: boolean; disabled?: boolean; untrusted?: boolean; disabledInWorkspace?: boolean; hidden?: false; later?: boolean; entitlement?: ChatEntitlement; organisations?: string[]; sku?: string; copilotTrackingId?: string }): Promise<void> {
1424 this.logService.trace(`[chat entitlement context] update(): ${JSON.stringify(context)}`);
1425
1477 return this.updateContext();
1478 }
1480 > private async updateContext(): Promise<void> {
1481 await this.updateBarrier?.wait();
1482
1483 this.updateContextSync();
1484 }
1486 > private updateContextSync(): void {
1487 const state = this.withConfiguration(this._state);
1488
1516 this._onDidChange.fire();
1517 }
1519 > suspend(): void {
1520 this.suspendedState = { ...this._state };
1521 this.updateBarrier = new Barrier();
1522 }
1524 > resume(): void {
1525 this.suspendedState = undefined;
1526 this.updateBarrier?.open();
1527 this.updateBarrier = undefined;
1528 }
1530 >
1531 > //#endregion
1532 >
1533 > registerSingleton(IChatEntitlementService, ChatEntitlementService, InstantiationType.Eager /* To ensure context keys are set asap */);
src/vs/workbench/services/extensions/common/extensionsRegistry.ts 641 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensionsRegistry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as nls from '../../../../nls.js';
7 > import { onUnexpectedError } from '../../../../base/common/errors.js';
8 > import { IJSONSchema } from '../../../../base/common/jsonSchema.js';
9 > import Severity from '../../../../base/common/severity.js';
10 > import { EXTENSION_IDENTIFIER_PATTERN } from '../../../../platform/extensionManagement/common/extensionManagement.js';
11 > import { Extensions, IJSONContributionRegistry } from '../../../../platform/jsonschemas/common/jsonContributionRegistry.js';
12 > import { Registry } from '../../../../platform/registry/common/platform.js';
13 > import { IMessage } from './extensions.js';
14 > import { IExtensionDescription, EXTENSION_CATEGORIES, ExtensionIdentifierSet } from '../../../../platform/extensions/common/extensions.js';
15 > import { ExtensionKind } from '../../../../platform/environment/common/environment.js';
16 > import { productSchemaId } from '../../../../platform/product/common/productService.js';
17 > import { ImplicitActivationEvents, IActivationEventsGenerator } from '../../../../platform/extensionManagement/common/implicitActivationEvents.js';
18 > import { IDisposable } from '../../../../base/common/lifecycle.js';
19 > import { allApiProposals } from '../../../../platform/extensions/common/extensionsApiProposals.js';
20 >
21 > const schemaRegistry = Registry.as<IJSONContributionRegistry>(Extensions.JSONContribution);
22 >
23 > export class ExtensionMessageCollector {
24 >
25 > private readonly _messageHandler: (msg: IMessage) => void;
26 > private readonly _extension: IExtensionDescription;
27 > private readonly _extensionPointId: string;
28 >
29 > constructor(
30 messageHandler: (msg: IMessage) => void,
31 extension: IExtensionDescription,
36 this._extensionPointId = extensionPointId;
37 }
39 > private _msg(type: Severity, message: string): void {
40 this._messageHandler({
41 type: type,
45 });
46 }
48 > public error(message: string): void {
49 this._msg(Severity.Error, message);
50 }
52 > public warn(message: string): void {
53 this._msg(Severity.Warning, message);
54 }
56 > public info(message: string): void {
57 this._msg(Severity.Info, message);
58 }
60 >
61 > export interface IExtensionPointUser<T> {
62 > description: IExtensionDescription;
63 > value: T;
64 > collector: ExtensionMessageCollector;
65 > }
66 >
67 > export type IExtensionPointHandler<T> = (extensions: readonly IExtensionPointUser<T>[], delta: ExtensionPointUserDelta<T>) => void;
68 >
69 > export interface IExtensionPoint<T> {
70 > readonly name: string;
71 > setHandler(handler: IExtensionPointHandler<T>): IDisposable;
72 > readonly defaultExtensionKind: ExtensionKind[] | undefined;
73 > readonly canHandleResolver?: boolean;
74 > }
75 >
76 > export class ExtensionPointUserDelta<T> {
77 >
78 > private static _toSet<T>(arr: readonly IExtensionPointUser<T>[]): ExtensionIdentifierSet {
79 > const result = new ExtensionIdentifierSet();
80 > for (let i = 0, len = arr.length; i < len; i++) {
81 > result.add(arr[i].description.identifier);
82 > }
83 > return result;
84 > }
85 >
86 > public static compute<T>(previous: readonly IExtensionPointUser<T>[] | null, current: readonly IExtensionPointUser<T>[]): ExtensionPointUserDelta<T> {
87 if (!previous || !previous.length) {
88 return new ExtensionPointUserDelta<T>(current, []);
100 return new ExtensionPointUserDelta<T>(added, removed);
101 }
103 > constructor(
104 public readonly added: readonly IExtensionPointUser<T>[],
105 public readonly removed: readonly IExtensionPointUser<T>[],
106 ) { }
108 >
109 > export class ExtensionPoint<T> implements IExtensionPoint<T> {
110 >
111 > public readonly name: string;
112 > public readonly defaultExtensionKind: ExtensionKind[] | undefined;
113 > public readonly canHandleResolver?: boolean;
114 >
115 > private _handler: IExtensionPointHandler<T> | null;
116 > private _users: IExtensionPointUser<T>[] | null;
117 > private _delta: ExtensionPointUserDelta<T> | null;
118 >
119 > constructor(name: string, defaultExtensionKind: ExtensionKind[] | undefined, canHandleResolver?: boolean) {
120 > this.name = name; extensionsRegistry.ts
121 > this.defaultExtensionKind = defaultExtensionKind;
122 > this.canHandleResolver = canHandleResolver;
123 > this._handler = null;
124 > this._users = null;
125 > this._delta = null;
126 > }
128 > setHandler(handler: IExtensionPointHandler<T>): IDisposable {
129 if (this._handler !== null) {
130 throw new Error('Handler already set!');
139 };
140 }
142 > acceptUsers(users: IExtensionPointUser<T>[]): void {
143 this._delta = ExtensionPointUserDelta.compute(this._users, users);
144 this._users = users;
145 this._handle();
146 }
148 > private _handle(): void {
149 if (this._handler === null || this._users === null || this._delta === null) {
150 return;
157 }
158 }
160 >
161 > const extensionKindSchema: IJSONSchema = {
162 > type: 'string',
163 > enum: [
164 > 'ui',
165 > 'workspace'
166 > ],
167 > enumDescriptions: [
168 > nls.localize('ui', "UI extension kind. In a remote window, such extensions are enabled only when available on the local machine."),
169 > nls.localize('workspace', "Workspace extension kind. In a remote window, such extensions are enabled only when available on the remote."),
170 > ],
171 > };
172 >
173 > const schemaId = 'vscode://schemas/vscode-extensions';
174 > export const schema: IJSONSchema = {
175 > properties: {
176 > engines: {
177 > type: 'object',
178 > description: nls.localize('vscode.extension.engines', "Engine compatibility."),
179 > properties: {
180 > 'vscode': {
181 > type: 'string',
182 > description: nls.localize('vscode.extension.engines.vscode', 'For VS Code extensions, specifies the VS Code version that the extension is compatible with. Cannot be *. For example: ^1.105.0 indicates compatibility with a minimum VS Code version of 1.105.0.'),
183 > default: '^1.105.0',
184 > }
185 > }
186 > },
187 > publisher: {
188 > description: nls.localize('vscode.extension.publisher', 'The publisher of the VS Code extension.'),
189 > type: 'string'
190 > },
191 > displayName: {
192 > description: nls.localize('vscode.extension.displayName', 'The display name for the extension used in the VS Code gallery.'),
193 > type: 'string'
194 > },
195 > categories: {
196 > description: nls.localize('vscode.extension.categories', 'The categories used by the VS Code gallery to categorize the extension.'),
197 > type: 'array',
198 > uniqueItems: true,
199 > items: {
200 > oneOf: [{
201 > type: 'string',
202 > enum: EXTENSION_CATEGORIES,
203 > },
204 > {
205 > type: 'string',
206 > const: 'Languages',
207 > deprecationMessage: nls.localize('vscode.extension.category.languages.deprecated', 'Use \'Programming Languages\' instead'),
208 > }]
209 > }
210 > },
211 > galleryBanner: {
212 > type: 'object',
213 > description: nls.localize('vscode.extension.galleryBanner', 'Banner used in the VS Code marketplace.'),
214 > properties: {
215 > color: {
216 > description: nls.localize('vscode.extension.galleryBanner.color', 'The banner color on the VS Code marketplace page header.'),
217 > type: 'string'
218 > },
219 > theme: {
220 > description: nls.localize('vscode.extension.galleryBanner.theme', 'The color theme for the font used in the banner.'),
221 > type: 'string',
222 > enum: ['dark', 'light']
223 > }
224 > }
225 > },
226 > contributes: {
227 > description: nls.localize('vscode.extension.contributes', 'All contributions of the VS Code extension represented by this package.'),
228 > type: 'object',
229 > // eslint-disable-next-line local/code-no-any-casts
230 > properties: {
231 > // extensions will fill in
232 > } as any as { [key: string]: any },
233 > default: {}
234 > },
235 > preview: {
236 > type: 'boolean',
237 > description: nls.localize('vscode.extension.preview', 'Sets the extension to be flagged as a Preview in the Marketplace.'),
238 > },
239 > enableProposedApi: {
240 > type: 'boolean',
241 > deprecationMessage: nls.localize('vscode.extension.enableProposedApi.deprecated', 'Use `enabledApiProposals` instead.'),
242 > },
243 > enabledApiProposals: {
244 > markdownDescription: nls.localize('vscode.extension.enabledApiProposals', 'Enable API proposals to try them out. Only valid **during development**. Extensions **cannot be published** with this property. For more details visit: https://code.visualstudio.com/api/advanced-topics/using-proposed-api'),
245 > type: 'array',
246 > uniqueItems: true,
247 > items: {
248 > type: 'string',
249 > enum: Object.keys(allApiProposals).map(proposalName => proposalName),
250 > markdownEnumDescriptions: Object.values(allApiProposals).map(value => value.proposal)
251 > }
252 > },
253 > api: {
254 > markdownDescription: nls.localize('vscode.extension.api', 'Describe the API provided by this extension. For more details visit: https://code.visualstudio.com/api/advanced-topics/remote-extensions#handling-dependencies-with-remote-extensions'),
255 > type: 'string',
256 > enum: ['none'],
257 > enumDescriptions: [
258 > nls.localize('vscode.extension.api.none', "Give up entirely the ability to export any APIs. This allows other extensions that depend on this extension to run in a separate extension host process or in a remote machine.")
259 > ]
260 > },
261 > activationEvents: {
262 > description: nls.localize('vscode.extension.activationEvents', 'Activation events for the VS Code extension.'),
263 > type: 'array',
264 > items: {
265 > type: 'string',
266 > defaultSnippets: [
267 > {
268 > label: 'onWebviewPanel',
269 > description: nls.localize('vscode.extension.activationEvents.onWebviewPanel', 'An activation event emitted when a webview is loaded of a certain viewType'),
270 > body: 'onWebviewPanel:viewType'
271 > },
272 > {
273 > label: 'onLanguage',
274 > description: nls.localize('vscode.extension.activationEvents.onLanguage', 'An activation event emitted whenever a file that resolves to the specified language gets opened.'),
275 > body: 'onLanguage:${1:languageId}'
276 > },
277 > {
278 > label: 'onCommand',
279 > description: nls.localize('vscode.extension.activationEvents.onCommand', 'An activation event emitted whenever the specified command gets invoked.'),
280 > body: 'onCommand:${2:commandId}'
281 > },
282 > {
283 > label: 'onDebug',
284 > description: nls.localize('vscode.extension.activationEvents.onDebug', 'An activation event emitted whenever a user is about to start debugging or about to setup debug configurations.'),
285 > body: 'onDebug'
286 > },
287 > {
288 > label: 'onDebugInitialConfigurations',
289 > description: nls.localize('vscode.extension.activationEvents.onDebugInitialConfigurations', 'An activation event emitted whenever a "launch.json" needs to be created (and all provideDebugConfigurations methods need to be called).'),
290 > body: 'onDebugInitialConfigurations'
291 > },
292 > {
293 > label: 'onDebugDynamicConfigurations',
294 > description: nls.localize('vscode.extension.activationEvents.onDebugDynamicConfigurations', 'An activation event emitted whenever a list of all debug configurations needs to be created (and all provideDebugConfigurations methods for the "dynamic" scope need to be called).'),
295 > body: 'onDebugDynamicConfigurations'
296 > },
297 > {
298 > label: 'onDebugResolve',
299 > description: nls.localize('vscode.extension.activationEvents.onDebugResolve', 'An activation event emitted whenever a debug session with the specific type is about to be launched (and a corresponding resolveDebugConfiguration method needs to be called).'),
300 > body: 'onDebugResolve:${6:type}'
301 > },
302 > {
303 > label: 'onDebugAdapterProtocolTracker',
304 > description: nls.localize('vscode.extension.activationEvents.onDebugAdapterProtocolTracker', 'An activation event emitted whenever a debug session with the specific type is about to be launched and a debug protocol tracker might be needed.'),
305 > body: 'onDebugAdapterProtocolTracker:${6:type}'
306 > },
307 > {
308 > label: 'workspaceContains',
309 > description: nls.localize('vscode.extension.activationEvents.workspaceContains', 'An activation event emitted whenever a folder is opened that contains at least a file matching the specified glob pattern.'),
310 > body: 'workspaceContains:${4:filePattern}'
311 > },
312 > {
313 > label: 'onStartupFinished',
314 > description: nls.localize('vscode.extension.activationEvents.onStartupFinished', 'An activation event emitted after the start-up finished (after all `*` activated extensions have finished activating).'),
315 > body: 'onStartupFinished'
316 > },
317 > {
318 > label: 'onTaskType',
319 > description: nls.localize('vscode.extension.activationEvents.onTaskType', 'An activation event emitted whenever tasks of a certain type need to be listed or resolved.'),
320 > body: 'onTaskType:${1:taskType}'
321 > },
322 > {
323 > label: 'onFileSystem',
324 > description: nls.localize('vscode.extension.activationEvents.onFileSystem', 'An activation event emitted whenever a file or folder is accessed with the given scheme.'),
325 > body: 'onFileSystem:${1:scheme}'
326 > },
327 > {
328 > label: 'onEditSession',
329 > description: nls.localize('vscode.extension.activationEvents.onEditSession', 'An activation event emitted whenever an edit session is accessed with the given scheme.'),
330 > body: 'onEditSession:${1:scheme}'
331 > },
332 > {
333 > label: 'onSearch',
334 > description: nls.localize('vscode.extension.activationEvents.onSearch', 'An activation event emitted whenever a search is started in the folder with the given scheme.'),
335 > body: 'onSearch:${7:scheme}'
336 > },
337 > {
338 > label: 'onView',
339 > body: 'onView:${5:viewId}',
340 > description: nls.localize('vscode.extension.activationEvents.onView', 'An activation event emitted whenever the specified view is expanded.'),
341 > },
342 > {
343 > label: 'onUri',
344 > body: 'onUri',
345 > description: nls.localize('vscode.extension.activationEvents.onUri', 'An activation event emitted whenever a system-wide Uri directed towards this extension is open.'),
346 > },
347 > {
348 > label: 'onOpenExternalUri',
349 > body: 'onOpenExternalUri',
350 > description: nls.localize('vscode.extension.activationEvents.onOpenExternalUri', 'An activation event emitted whenever a external uri (such as an http or https link) is being opened.'),
351 > },
352 > {
353 > label: 'onCustomEditor',
354 > body: 'onCustomEditor:${9:viewType}',
355 > description: nls.localize('vscode.extension.activationEvents.onCustomEditor', 'An activation event emitted whenever the specified custom editor becomes visible.'),
356 > },
357 > {
358 > label: 'onNotebook',
359 > body: 'onNotebook:${1:type}',
360 > description: nls.localize('vscode.extension.activationEvents.onNotebook', 'An activation event emitted whenever the specified notebook document is opened.'),
361 > },
362 > {
363 > label: 'onAuthenticationRequest',
364 > body: 'onAuthenticationRequest:${11:authenticationProviderId}',
365 > description: nls.localize('vscode.extension.activationEvents.onAuthenticationRequest', 'An activation event emitted whenever sessions are requested from the specified authentication provider.')
366 > },
367 > {
368 > label: 'onRenderer',
369 > description: nls.localize('vscode.extension.activationEvents.onRenderer', 'An activation event emitted whenever a notebook output renderer is used.'),
370 > body: 'onRenderer:${11:rendererId}'
371 > },
372 > {
373 > label: 'onTerminalProfile',
374 > body: 'onTerminalProfile:${1:terminalId}',
375 > description: nls.localize('vscode.extension.activationEvents.onTerminalProfile', 'An activation event emitted when a specific terminal profile is launched.'),
376 > },
377 > {
378 > label: 'onTerminalQuickFixRequest',
379 > body: 'onTerminalQuickFixRequest:${1:quickFixId}',
380 > description: nls.localize('vscode.extension.activationEvents.onTerminalQuickFixRequest', 'An activation event emitted when a command matches the selector associated with this ID'),
381 > },
382 > {
383 > label: 'onWalkthrough',
384 > body: 'onWalkthrough:${1:walkthroughID}',
385 > description: nls.localize('vscode.extension.activationEvents.onWalkthrough', 'An activation event emitted when a specified walkthrough is opened.'),
386 > },
387 > {
388 > label: 'onIssueReporterOpened',
389 > body: 'onIssueReporterOpened',
390 > description: nls.localize('vscode.extension.activationEvents.onIssueReporterOpened', 'An activation event emitted when the issue reporter is opened.'),
391 > },
392 > {
393 > label: 'onChatParticipant',
394 > body: 'onChatParticipant:${1:participantId}',
395 > description: nls.localize('vscode.extension.activationEvents.onChatParticipant', 'An activation event emitted when the specified chat participant is invoked.'),
396 > },
397 > {
398 > label: 'onChatContextProvider',
399 > body: 'onChatContextProvider:${1:contextProviderId}',
400 > description: nls.localize('vscode.extension.activationEvents.onChatContextProvider', 'An activation event emitted when the specified chat context provider is invoked.'),
401 > },
402 > {
403 > label: 'onLanguageModelChatProvider',
404 > body: 'onLanguageModelChatProvider:${1:vendor}',
405 > description: nls.localize('vscode.extension.activationEvents.onLanguageModelChatProvider', 'An activation event emitted when a chat model provider for the given vendor is requested.'),
406 > },
407 > {
408 > label: 'onLanguageModelTool',
409 > body: 'onLanguageModelTool:${1:toolId}',
410 > description: nls.localize('vscode.extension.activationEvents.onLanguageModelTool', 'An activation event emitted when the specified language model tool is invoked.'),
411 > },
412 > {
413 > label: 'onTerminal',
414 > body: 'onTerminal:{1:shellType}',
415 > description: nls.localize('vscode.extension.activationEvents.onTerminal', 'An activation event emitted when a terminal of the given shell type is opened.'),
416 > },
417 > {
418 > label: 'onTerminalShellIntegration',
419 > body: 'onTerminalShellIntegration:${1:shellType}',
420 > description: nls.localize('vscode.extension.activationEvents.onTerminalShellIntegration', 'An activation event emitted when terminal shell integration is activated for the given shell type.'),
421 > },
422 > {
423 > label: 'onMcpCollection',
424 > description: nls.localize('vscode.extension.activationEvents.onMcpCollection', 'An activation event emitted whenever a tool from the MCP server is requested.'),
425 > body: 'onMcpCollection:${2:collectionId}',
426 > },
427 > {
428 > label: '*',
429 > description: nls.localize('vscode.extension.activationEvents.star', 'An activation event emitted on VS Code startup. To ensure a great end user experience, please use this activation event in your extension only when no other activation events combination works in your use-case.'),
430 > body: '*'
431 > }
432 > ],
433 > }
434 > },
435 > badges: {
436 > type: 'array',
437 > description: nls.localize('vscode.extension.badges', 'Array of badges to display in the sidebar of the Marketplace\'s extension page.'),
438 > items: {
439 > type: 'object',
440 > required: ['url', 'href', 'description'],
441 > properties: {
442 > url: {
443 > type: 'string',
444 > description: nls.localize('vscode.extension.badges.url', 'Badge image URL.')
445 > },
446 > href: {
447 > type: 'string',
448 > description: nls.localize('vscode.extension.badges.href', 'Badge link.')
449 > },
450 > description: {
451 > type: 'string',
452 > description: nls.localize('vscode.extension.badges.description', 'Badge description.')
453 > }
454 > }
455 > }
456 > },
457 > markdown: {
458 > type: 'string',
459 > description: nls.localize('vscode.extension.markdown', "Controls the Markdown rendering engine used in the Marketplace. Either github (default) or standard."),
460 > enum: ['github', 'standard'],
461 > default: 'github'
462 > },
463 > qna: {
464 > default: 'marketplace',
465 > description: nls.localize('vscode.extension.qna', "Controls the Q&A link in the Marketplace. Set to marketplace to enable the default Marketplace Q & A site. Set to a string to provide the URL of a custom Q & A site. Set to false to disable Q & A altogether."),
466 > anyOf: [
467 > {
468 > type: ['string', 'boolean'],
469 > enum: ['marketplace', false]
470 > },
471 > {
472 > type: 'string'
473 > }
474 > ]
475 > },
476 > extensionDependencies: {
477 > description: nls.localize('vscode.extension.extensionDependencies', 'Dependencies to other extensions. The identifier of an extension is always ${publisher}.${name}. For example: vscode.csharp.'),
478 > type: 'array',
479 > uniqueItems: true,
480 > items: {
481 > type: 'string',
482 > pattern: EXTENSION_IDENTIFIER_PATTERN
483 > }
484 > },
485 > extensionAffinity: {
486 > description: nls.localize('vscode.extension.extensionAffinity', 'Extensions that this extension should be colocated with in the same extension host process if possible. The identifier of an extension is always ${publisher}.${name}. For example: vscode.git.'),
487 > type: 'array',
488 > uniqueItems: true,
489 > items: {
490 > type: 'string',
491 > pattern: EXTENSION_IDENTIFIER_PATTERN
492 > }
493 > },
494 > extensionPack: {
495 > description: nls.localize('vscode.extension.contributes.extensionPack', "A set of extensions that can be installed together. The identifier of an extension is always ${publisher}.${name}. For example: vscode.csharp."),
496 > type: 'array',
497 > uniqueItems: true,
498 > items: {
499 > type: 'string',
500 > pattern: EXTENSION_IDENTIFIER_PATTERN
501 > }
502 > },
503 > extensionKind: {
504 > description: nls.localize('extensionKind', "Define the kind of an extension. `ui` extensions are installed and run on the local machine while `workspace` extensions run on the remote."),
505 > type: 'array',
506 > items: extensionKindSchema,
507 > default: ['workspace'],
508 > defaultSnippets: [
509 > {
510 > body: ['ui'],
511 > description: nls.localize('extensionKind.ui', "Define an extension which can run only on the local machine when connected to remote window.")
512 > },
513 > {
514 > body: ['workspace'],
515 > description: nls.localize('extensionKind.workspace', "Define an extension which can run only on the remote machine when connected remote window.")
516 > },
517 > {
518 > body: ['ui', 'workspace'],
519 > description: nls.localize('extensionKind.ui-workspace', "Define an extension which can run on either side, with a preference towards running on the local machine.")
520 > },
521 > {
522 > body: ['workspace', 'ui'],
523 > description: nls.localize('extensionKind.workspace-ui', "Define an extension which can run on either side, with a preference towards running on the remote machine.")
524 > },
525 > {
526 > body: [],
527 > description: nls.localize('extensionKind.empty', "Define an extension which cannot run in a remote context, neither on the local, nor on the remote machine.")
528 > }
529 > ]
530 > },
531 > capabilities: {
532 > description: nls.localize('vscode.extension.capabilities', "Declare the set of supported capabilities by the extension."),
533 > type: 'object',
534 > properties: {
535 > virtualWorkspaces: {
536 > description: nls.localize('vscode.extension.capabilities.virtualWorkspaces', "Declares whether the extension should be enabled in virtual workspaces. A virtual workspace is a workspace which is not backed by any on-disk resources. When false, this extension will be automatically disabled in virtual workspaces. Default is true."),
537 > type: ['boolean', 'object'],
538 > defaultSnippets: [
539 > { label: 'limited', body: { supported: '${1:limited}', description: '${2}' } },
540 > { label: 'false', body: { supported: false, description: '${2}' } },
541 > ],
542 > default: true.valueOf,
543 > properties: {
544 > supported: {
545 > markdownDescription: nls.localize('vscode.extension.capabilities.virtualWorkspaces.supported', "Declares the level of support for virtual workspaces by the extension."),
546 > type: ['string', 'boolean'],
547 > enum: ['limited', true, false],
548 > enumDescriptions: [
549 > nls.localize('vscode.extension.capabilities.virtualWorkspaces.supported.limited', "The extension will be enabled in virtual workspaces with some functionality disabled."),
550 > nls.localize('vscode.extension.capabilities.virtualWorkspaces.supported.true', "The extension will be enabled in virtual workspaces with all functionality enabled."),
551 > nls.localize('vscode.extension.capabilities.virtualWorkspaces.supported.false', "The extension will not be enabled in virtual workspaces."),
552 > ]
553 > },
554 > description: {
555 > type: 'string',
556 > markdownDescription: nls.localize('vscode.extension.capabilities.virtualWorkspaces.description', "A description of how virtual workspaces affects the extensions behavior and why it is needed. This only applies when `supported` is not `true`."),
557 > }
558 > }
559 > },
560 > untrustedWorkspaces: {
561 > description: nls.localize('vscode.extension.capabilities.untrustedWorkspaces', 'Declares how the extension should be handled in untrusted workspaces.'),
562 > type: 'object',
563 > required: ['supported'],
564 > defaultSnippets: [
565 > { body: { supported: '${1:limited}', description: '${2}' } },
566 > ],
567 > properties: {
568 > supported: {
569 > markdownDescription: nls.localize('vscode.extension.capabilities.untrustedWorkspaces.supported', "Declares the level of support for untrusted workspaces by the extension."),
570 > type: ['string', 'boolean'],
571 > enum: ['limited', true, false],
572 > enumDescriptions: [
573 > nls.localize('vscode.extension.capabilities.untrustedWorkspaces.supported.limited', "The extension will be enabled in untrusted workspaces with some functionality disabled."),
574 > nls.localize('vscode.extension.capabilities.untrustedWorkspaces.supported.true', "The extension will be enabled in untrusted workspaces with all functionality enabled."),
575 > nls.localize('vscode.extension.capabilities.untrustedWorkspaces.supported.false', "The extension will not be enabled in untrusted workspaces."),
576 > ]
577 > },
578 > restrictedConfigurations: {
579 > description: nls.localize('vscode.extension.capabilities.untrustedWorkspaces.restrictedConfigurations', "A list of configuration keys contributed by the extension that should not use workspace values in untrusted workspaces."),
580 > type: 'array',
581 > items: {
582 > type: 'string'
583 > }
584 > },
585 > description: {
586 > type: 'string',
587 > markdownDescription: nls.localize('vscode.extension.capabilities.untrustedWorkspaces.description', "A description of how workspace trust affects the extensions behavior and why it is needed. This only applies when `supported` is not `true`."),
588 > }
589 > }
590 > }
591 > }
592 > },
593 > sponsor: {
594 > description: nls.localize('vscode.extension.contributes.sponsor', "Specify the location from where users can sponsor your extension."),
595 > type: 'object',
596 > defaultSnippets: [
597 > { body: { url: '${1:https:}' } },
598 > ],
599 > properties: {
600 > 'url': {
601 > description: nls.localize('vscode.extension.contributes.sponsor.url', "URL from where users can sponsor your extension. It must be a valid URL with a HTTP or HTTPS protocol. Example value: https://github.com/sponsors/nvaccess"),
602 > type: 'string',
603 > }
604 > }
605 > },
606 > scripts: {
607 > type: 'object',
608 > properties: {
609 > 'vscode:prepublish': {
610 > description: nls.localize('vscode.extension.scripts.prepublish', 'Script executed before the package is published as a VS Code extension.'),
611 > type: 'string'
612 > },
613 > 'vscode:uninstall': {
614 > description: nls.localize('vscode.extension.scripts.uninstall', 'Uninstall hook for VS Code extension. Script that gets executed when the extension is completely uninstalled from VS Code which is when VS Code is restarted (shutdown and start) after the extension is uninstalled. Only Node scripts are supported.'),
615 > type: 'string'
616 > }
617 > }
618 > },
619 > icon: {
620 > type: 'string',
621 > description: nls.localize('vscode.extension.icon', 'The path to a 128x128 pixel icon.')
622 > },
623 > l10n: {
624 > type: 'string',
625 > description: nls.localize({
626 > key: 'vscode.extension.l10n',
627 > comment: [
628 > '{Locked="bundle.l10n._locale_.json"}',
629 > '{Locked="vscode.l10n API"}'
630 > ]
631 > }, 'The relative path to a folder containing localization (bundle.l10n.*.json) files. Must be specified if you are using the vscode.l10n API.')
632 > },
633 > pricing: {
634 > type: 'string',
635 > markdownDescription: nls.localize('vscode.extension.pricing', 'The pricing information for the extension. Can be Free (default) or Trial. For more details visit: https://code.visualstudio.com/api/working-with-extensions/publishing-extension#extension-pricing-label'),
636 > enum: ['Free', 'Trial'],
637 > default: 'Free'
638 > }
639 > }
640 > };
641 >
642 > export type removeArray<T> = T extends Array<infer X> ? X : T;
643 >
644 > export interface IExtensionPointDescriptor<T> {
645 > extensionPoint: string;
646 > deps?: IExtensionPoint<unknown>[];
647 > jsonSchema: IJSONSchema;
648 > defaultExtensionKind?: ExtensionKind[];
649 > canHandleResolver?: boolean;
650 > /**
651 > * A function which runs before the extension point has been validated and which
652 > * should collect automatic activation events from the contribution.
653 > */
654 > activationEventsGenerator?: IActivationEventsGenerator<removeArray<T>>;
655 > }
656 >
657 > export class ExtensionsRegistryImpl {
658 >
659 > private readonly _extensionPoints = new Map<string, ExtensionPoint<any>>();
660 >
661 > public registerExtensionPoint<T>(desc: IExtensionPointDescriptor<T>): IExtensionPoint<T> {
662 > if (this._extensionPoints.has(desc.extensionPoint)) { extensionsRegistry.ts
663 throw new Error('Duplicate extension point: ' + desc.extensionPoint);
664 }
665 > const result = new ExtensionPoint<T>(desc.extensionPoint, desc.defaultExtensionKind, desc.canHandleResolver); extensionsRegistry.ts
666 > this._extensionPoints.set(desc.extensionPoint, result);
667 > if (desc.activationEventsGenerator) {
668 > ImplicitActivationEvents.register(desc.extensionPoint, desc.activationEventsGenerator); extensionsRegistry.ts
669 > }
671 > schema.properties!['contributes'].properties![desc.extensionPoint] = desc.jsonSchema;
672 > schemaRegistry.registerSchema(schemaId, schema);
673 >
674 > return result;
675 > }
677 > public getExtensionPoints(): ExtensionPoint<unknown>[] {
678 return Array.from(this._extensionPoints.values());
679 }
681 >
682 > const PRExtensions = {
683 > ExtensionsRegistry: 'ExtensionsRegistry'
684 > };
685 > Registry.add(PRExtensions.ExtensionsRegistry, new ExtensionsRegistryImpl());
686 > export const ExtensionsRegistry: ExtensionsRegistryImpl = Registry.as(PRExtensions.ExtensionsRegistry);
687 >
688 > schemaRegistry.registerSchema(schemaId, schema);
689 >
690 >
691 > schemaRegistry.registerSchema(productSchemaId, {
692 > properties: {
693 > extensionEnabledApiProposals: {
694 > description: nls.localize('product.extensionEnabledApiProposals', "API proposals that the respective extensions can freely use."),
695 > type: 'object',
696 > properties: {},
697 > additionalProperties: {
698 > anyOf: [{
699 > type: 'array',
700 > uniqueItems: true,
701 > items: {
702 > type: 'string',
703 > enum: Object.keys(allApiProposals),
704 > markdownEnumDescriptions: Object.values(allApiProposals).map(value => value.proposal)
705 > }
706 > }]
707 > }
708 > }
709 > }
710 > });
src/vs/platform/configuration/common/configurationRegistry.ts 638 covered LOC · 98 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- configurationRegistry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { distinct } from '../../../base/common/arrays.js';
7 > import { IStringDictionary } from '../../../base/common/collections.js';
8 > import { Emitter, Event } from '../../../base/common/event.js';
9 > import { IJSONSchema } from '../../../base/common/jsonSchema.js';
10 > import * as types from '../../../base/common/types.js';
11 > import * as nls from '../../../nls.js';
12 > import { getLanguageTagSettingPlainKey } from './configuration.js';
13 > import { Extensions as JSONExtensions, IJSONContributionRegistry } from '../../jsonschemas/common/jsonContributionRegistry.js';
14 > import { Registry } from '../../registry/common/platform.js';
15 > import { IPolicy, IPolicyReference, PolicyName } from '../../../base/common/policy.js';
16 > import { Disposable } from '../../../base/common/lifecycle.js';
17 > import product from '../../product/common/product.js';
18 >
19 > export enum EditPresentationTypes {
20 > Multiline = 'multilineText',
21 > Singleline = 'singlelineText'
22 > }
23 >
24 > export const Extensions = {
25 > Configuration: 'base.contributions.configuration'
26 > };
27 >
28 > export interface IConfigurationDelta {
29 > removedDefaults?: IConfigurationDefaults[];
30 > removedConfigurations?: IConfigurationNode[];
31 > addedDefaults?: IConfigurationDefaults[];
32 > addedConfigurations?: IConfigurationNode[];
33 > }
34 >
35 > export interface IConfigurationRegistry {
36 >
37 > /**
38 > * Register a configuration to the registry.
39 > */
40 > registerConfiguration(configuration: IConfigurationNode): IConfigurationNode;
41 >
42 > /**
43 > * Register multiple configurations to the registry.
44 > */
45 > registerConfigurations(configurations: IConfigurationNode[], validate?: boolean): void;
46 >
47 > /**
48 > * Deregister multiple configurations from the registry.
49 > */
50 > deregisterConfigurations(configurations: IConfigurationNode[]): void;
51 >
52 > /**
53 > * update the configuration registry by
54 > * - registering the configurations to add
55 > * - dereigstering the configurations to remove
56 > */
57 > updateConfigurations(configurations: { add: IConfigurationNode[]; remove: IConfigurationNode[] }): void;
58 >
59 > /**
60 > * Register multiple default configurations to the registry.
61 > */
62 > registerDefaultConfigurations(defaultConfigurations: IConfigurationDefaults[]): void;
63 >
64 > /**
65 > * Deregister multiple default configurations from the registry.
66 > */
67 > deregisterDefaultConfigurations(defaultConfigurations: IConfigurationDefaults[]): void;
68 >
69 > /**
70 > * Bulk update of the configuration registry (default and configurations, remove and add)
71 > * @param delta
72 > */
73 > deltaConfiguration(delta: IConfigurationDelta): void;
74 >
75 > /**
76 > * Return the registered default configurations
77 > */
78 > getRegisteredDefaultConfigurations(): IConfigurationDefaults[];
79 >
80 > /**
81 > * Return the registered configuration defaults overrides
82 > */
83 > getConfigurationDefaultsOverrides(): Map<string, IConfigurationDefaultOverrideValue>;
84 >
85 > /**
86 > * Signal that the schema of a configuration setting has changes. It is currently only supported to change enumeration values.
87 > * Property or default value changes are not allowed.
88 > */
89 > notifyConfigurationSchemaUpdated(...configurations: IConfigurationNode[]): void;
90 >
91 > /**
92 > * Event that fires whenever a configuration has been
93 > * registered.
94 > */
95 > readonly onDidSchemaChange: Event<void>;
96 >
97 > /**
98 > * Event that fires whenever a configuration has been
99 > * registered.
100 > */
101 > readonly onDidUpdateConfiguration: Event<{ properties: ReadonlySet<string>; defaultsOverrides?: boolean }>;
102 >
103 > /**
104 > * Returns all configuration nodes contributed to this registry.
105 > */
106 > getConfigurations(): IConfigurationNode[];
107 >
108 > /**
109 > * Returns all configurations settings of all configuration nodes contributed to this registry.
110 > */
111 > getConfigurationProperties(): IStringDictionary<IRegisteredConfigurationPropertySchema>;
112 >
113 > /**
114 > * Returns the owning setting key per policy name (at most one owner per name).
115 > */
116 > getPolicyConfigurations(): Map<PolicyName, string>;
117 >
118 > /**
119 > * Returns the referencing setting keys per policy name.
120 > */
121 > getPolicyReferenceConfigurations(): Map<PolicyName, Set<string>>;
122 >
123 > /**
124 > * Returns all excluded configurations settings of all configuration nodes contributed to this registry.
125 > */
126 > getExcludedConfigurationProperties(): IStringDictionary<IRegisteredConfigurationPropertySchema>;
127 >
128 > /**
129 > * Register the identifiers for editor configurations
130 > */
131 > registerOverrideIdentifiers(identifiers: string[]): void;
132 > }
133 >
134 > export const enum ConfigurationScope {
135 > /**
136 > * Application specific configuration, which can be configured only in default profile user settings.
137 > */
138 > APPLICATION = 1,
139 > /**
140 > * Machine specific configuration, which can be configured only in local and remote user settings.
141 > */
142 > MACHINE,
143 > /**
144 > * An application machine specific configuration, which can be configured only in default profile user settings and remote user settings.
145 > */
146 > APPLICATION_MACHINE,
147 > /**
148 > * Window specific configuration, which can be configured in the user or workspace settings.
149 > */
150 > WINDOW,
151 > /**
152 > * Resource specific configuration, which can be configured in the user, workspace or folder settings.
153 > */
154 > RESOURCE,
155 > /**
156 > * Resource specific configuration that can be configured in language specific settings
157 > */
158 > LANGUAGE_OVERRIDABLE,
159 > /**
160 > * Machine specific configuration that can also be configured in workspace or folder settings.
161 > */
162 > MACHINE_OVERRIDABLE,
163 > }
164 >
165 >
166 > export interface IConfigurationPropertySchema extends IJSONSchema {
167 >
168 > scope?: ConfigurationScope;
169 >
170 > /**
171 > * When restricted, value of this configuration will be read only from trusted sources.
172 > * For eg., If the workspace is not trusted, then the value of this configuration is not read from workspace settings file.
173 > */
174 > restricted?: boolean;
175 >
176 > /**
177 > * When `false` this property is excluded from the registry. Default is to include.
178 > */
179 > included?: boolean;
180 >
181 > /**
182 > * List of tags associated to the property.
183 > * - A tag can be used for filtering
184 > * - Use `experimental` tag for marking the setting as experimental.
185 > */
186 > tags?: string[];
187 >
188 > /**
189 > * When enabled this setting is ignored during sync and user can override this.
190 > */
191 > ignoreSync?: boolean;
192 >
193 > /**
194 > * When enabled this setting is ignored during sync and user cannot override this.
195 > */
196 > disallowSyncIgnore?: boolean;
197 >
198 > /**
199 > * Disallow extensions to contribute configuration default value for this setting.
200 > */
201 > disallowConfigurationDefault?: boolean;
202 >
203 > /**
204 > * Labels for enumeration items
205 > */
206 > enumItemLabels?: string[];
207 >
208 > /**
209 > * Optional keywords used for search purposes.
210 > */
211 > keywords?: string[];
212 >
213 > /**
214 > * When specified, controls the presentation format of string settings.
215 > * Otherwise, the presentation format defaults to `singleline`.
216 > */
217 > editPresentation?: EditPresentationTypes;
218 >
219 > /**
220 > * When specified, gives an order number for the setting
221 > * within the settings editor. Otherwise, the setting is placed at the end.
222 > */
223 > order?: number;
224 >
225 > /**
226 > * When specified, this setting's value can always be overwritten by
227 > * a system-wide policy. Exactly one setting may *own* a given policy name.
228 > */
229 > policy?: IPolicy;
230 >
231 > /**
232 > * When specified, this setting is governed by a policy owned by another setting.
233 > * A setting must not declare both `policy` and `policyReference`.
234 > * The type must match the owning setting (enforced when exporting the policy catalog).
235 > */
236 > policyReference?: IPolicyReference;
237 >
238 > /**
239 > * When specified, this setting's default value can always be overwritten by
240 > * an experiment.
241 > */
242 > experiment?: {
243 > /**
244 > * The mode of the experiment.
245 > * - `startup`: The setting value is updated to the experiment value only on startup.
246 > * - `auto`: The setting value is updated to the experiment value automatically (whenever the experiment value changes).
247 > */
248 > mode: 'startup' | 'auto';
249 >
250 > /**
251 > * The name of the experiment. By default, this is `config.${settingId}`
252 > */
253 > name?: string;
254 > };
255 >
256 > /**
257 > * When specified, provides configuration overrides for the Agents window.
258 > */
259 > agentsWindow?: {
260 > /**
261 > * Override default value for this setting in the Agents window.
262 > */
263 > default?: unknown;
264 >
265 > /**
266 > * When `true`, this setting is read-only in the Agents window
267 > * and cannot be changed by the user.
268 > */
269 > readOnly?: boolean;
270 > };
271 > }
272 >
273 > export interface IExtensionInfo {
274 > id: string;
275 > displayName?: string;
276 > }
277 >
278 > export interface IConfigurationNode {
279 > id?: string;
280 > order?: number;
281 > type?: string | string[];
282 > title?: string;
283 > description?: string;
284 > properties?: IStringDictionary<IConfigurationPropertySchema>;
285 > allOf?: IConfigurationNode[];
286 > scope?: ConfigurationScope;
287 > extensionInfo?: IExtensionInfo;
288 > restrictedProperties?: string[];
289 > }
290 >
291 > export type ConfigurationDefaultSource = IExtensionInfo | string;
292 >
293 > export function isConfigurationDefaultSourceEquals(a: ConfigurationDefaultSource | undefined, b: ConfigurationDefaultSource | undefined): boolean {
294 if (a === b) {
295 return true;
303 return a.id === b.id;
304 }
306 > export type ConfigurationDefaultValueSource = ConfigurationDefaultSource | Map<string, ConfigurationDefaultSource>;
307 >
308 > export interface IConfigurationDefaults {
309 > overrides: IStringDictionary<unknown>;
310 > source?: ConfigurationDefaultSource;
311 > donotCache?: boolean;
312 > preventExperimentOverride?: boolean;
313 > }
314 >
315 > export type IRegisteredConfigurationPropertySchema = IConfigurationPropertySchema & {
316 > section?: {
317 > id?: string;
318 > title?: string;
319 > order?: number;
320 > extensionInfo?: IExtensionInfo;
321 > };
322 > defaultDefaultValue?: unknown;
323 > source?: ConfigurationDefaultSource; // Source of the Property
324 > defaultValueSource?: ConfigurationDefaultValueSource; // Source of the Default Value
325 > };
326 >
327 > export interface IConfigurationDefaultOverride {
328 > readonly value: unknown;
329 > readonly source?: ConfigurationDefaultSource; // Source of the default override
330 > }
331 >
332 > export interface IConfigurationDefaultOverrideValue {
333 > readonly value: unknown;
334 > readonly source?: ConfigurationDefaultValueSource;
335 > }
336 >
337 > export const allSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
338 > export const applicationSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
339 > export const applicationMachineSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
340 > export const machineSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
341 > export const machineOverridableSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
342 > export const windowSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
343 > export const resourceSettings: { properties: IStringDictionary<IConfigurationPropertySchema>; patternProperties: IStringDictionary<IConfigurationPropertySchema> } = { properties: {}, patternProperties: {} };
344 >
345 > export const resourceLanguageSettingsSchemaId = 'vscode://schemas/settings/resourceLanguage';
346 > export const configurationDefaultsSchemaId = 'vscode://schemas/settings/configurationDefaults';
347 >
348 > const contributionRegistry = Registry.as<IJSONContributionRegistry>(JSONExtensions.JSONContribution);
349 >
350 > class ConfigurationRegistry extends Disposable implements IConfigurationRegistry {
351 >
352 > private readonly registeredConfigurationDefaults: IConfigurationDefaults[] = [];
353 > private readonly configurationDefaultsOverrides: Map<string, { configurationDefaultOverrides: IConfigurationDefaultOverride[]; configurationDefaultOverrideValue?: IConfigurationDefaultOverrideValue }>;
354 > private readonly defaultLanguageConfigurationOverridesNode: IConfigurationNode;
355 > private readonly configurationContributors: IConfigurationNode[];
356 > private readonly configurationProperties: IStringDictionary<IRegisteredConfigurationPropertySchema>;
357 > private readonly policyConfigurations: Map<PolicyName, string>;
358 > private readonly policyReferenceConfigurations: Map<PolicyName, Set<string>>;
359 > private readonly excludedConfigurationProperties: IStringDictionary<IRegisteredConfigurationPropertySchema>;
360 > private readonly resourceLanguageSettingsSchema: IJSONSchema;
361 > private readonly overrideIdentifiers = new Set<string>();
362 >
363 > private readonly _onDidSchemaChange = this._register(new Emitter<void>());
364 > readonly onDidSchemaChange: Event<void> = this._onDidSchemaChange.event;
365 >
366 > private readonly _onDidUpdateConfiguration = this._register(new Emitter<{ properties: ReadonlySet<string>; defaultsOverrides?: boolean }>());
367 > readonly onDidUpdateConfiguration = this._onDidUpdateConfiguration.event;
368 >
369 > constructor() {
370 > super();
371 > this.configurationDefaultsOverrides = new Map();
372 > this.defaultLanguageConfigurationOverridesNode = {
373 > id: 'defaultOverrides',
374 > title: nls.localize('defaultLanguageConfigurationOverrides.title', "Default Language Configuration Overrides"),
375 > properties: {}
376 > };
377 > this.configurationContributors = [this.defaultLanguageConfigurationOverridesNode];
378 > this.resourceLanguageSettingsSchema = {
379 > properties: {},
380 > patternProperties: {},
381 > additionalProperties: true,
382 > allowTrailingCommas: true,
383 > allowComments: true
384 > };
385 > this.configurationProperties = {};
386 > this.policyConfigurations = new Map<PolicyName, string>();
387 > this.policyReferenceConfigurations = new Map<PolicyName, Set<string>>();
388 > this.excludedConfigurationProperties = {};
389 >
390 > contributionRegistry.registerSchema(resourceLanguageSettingsSchemaId, this.resourceLanguageSettingsSchema);
391 > this.registerOverridePropertyPatternKey();
392 > }
393 >
394 > public registerConfiguration(configuration: IConfigurationNode, validate: boolean = true): IConfigurationNode {
395 > this.registerConfigurations([configuration], validate); configurationRegistry.ts
396 > return configuration;
397 > }
399 > public registerConfigurations(configurations: IConfigurationNode[], validate: boolean = true): void {
400 > const properties = new Set<string>(); configurationRegistry.ts
401 > this.doRegisterConfigurations(configurations, validate, properties);
402 >
403 > contributionRegistry.registerSchema(resourceLanguageSettingsSchemaId, this.resourceLanguageSettingsSchema);
404 > this._onDidSchemaChange.fire();
405 > this._onDidUpdateConfiguration.fire({ properties });
406 > }
408 > public deregisterConfigurations(configurations: IConfigurationNode[]): void {
409 const properties = new Set<string>();
410 this.doDeregisterConfigurations(configurations, properties);
414 this._onDidUpdateConfiguration.fire({ properties });
415 }
417 > public updateConfigurations({ add, remove }: { add: IConfigurationNode[]; remove: IConfigurationNode[] }): void {
418 > const properties = new Set<string>(); configurationRegistry.ts
419 > this.doDeregisterConfigurations(remove, properties);
420 > this.doRegisterConfigurations(add, false, properties);
421 >
422 > contributionRegistry.registerSchema(resourceLanguageSettingsSchemaId, this.resourceLanguageSettingsSchema);
423 > this._onDidSchemaChange.fire();
424 > this._onDidUpdateConfiguration.fire({ properties });
425 > }
427 > public registerDefaultConfigurations(configurationDefaults: IConfigurationDefaults[]): void {
428 const properties = new Set<string>();
429 this.doRegisterDefaultConfigurations(configurationDefaults, properties);
431 this._onDidUpdateConfiguration.fire({ properties, defaultsOverrides: true });
432 }
434 > private doRegisterDefaultConfigurations(configurationDefaults: IConfigurationDefaults[], bucket: Set<string>) {
435
436 this.registeredConfigurationDefaults.push(...configurationDefaults);
480 this.doRegisterOverrideIdentifiers(overrideIdentifiers);
481 }
483 > public deregisterDefaultConfigurations(defaultConfigurations: IConfigurationDefaults[]): void {
484 const properties = new Set<string>();
485 this.doDeregisterDefaultConfigurations(defaultConfigurations, properties);
487 this._onDidUpdateConfiguration.fire({ properties, defaultsOverrides: true });
488 }
490 > private doDeregisterDefaultConfigurations(defaultConfigurations: IConfigurationDefaults[], bucket: Set<string>): void {
491 for (const defaultConfiguration of defaultConfigurations) {
492 const index = this.registeredConfigurationDefaults.indexOf(defaultConfiguration);
544 this.updateOverridePropertyPatternKey();
545 }
547 > private updateDefaultOverrideProperty(key: string, newDefaultOverride: IConfigurationDefaultOverrideValue, source: ConfigurationDefaultSource | undefined): void {
548 const property: IRegisteredConfigurationPropertySchema = {
549 section: {
564 this.defaultLanguageConfigurationOverridesNode.properties![key] = property;
565 }
567 > private mergeDefaultConfigurationsForOverrideIdentifier(overrideIdentifier: string, configurationValueObject: IStringDictionary<unknown>, valueSource: ConfigurationDefaultSource | undefined, existingDefaultOverride: IConfigurationDefaultOverrideValue | undefined): IConfigurationDefaultOverrideValue | undefined {
568 const defaultValue = existingDefaultOverride?.value || {};
569 const source = existingDefaultOverride?.source ?? new Map<string, ConfigurationDefaultSource>();
605 return { value: defaultValue, source };
606 }
608 > private mergeDefaultConfigurationsForConfigurationProperty(propertyKey: string, value: unknown, valuesSource: ConfigurationDefaultSource | undefined, existingDefaultOverride: IConfigurationDefaultOverrideValue | undefined): IConfigurationDefaultOverrideValue | undefined {
609 const property = this.configurationProperties[propertyKey];
610 const existingDefaultValue = existingDefaultOverride?.value ?? property?.defaultDefaultValue;
637 return { value, source };
638 }
640 > public deltaConfiguration(delta: IConfigurationDelta): void {
641 // defaults: remove
642 let defaultsOverrides = false;
662 this._onDidUpdateConfiguration.fire({ properties, defaultsOverrides });
663 }
665 > public notifyConfigurationSchemaUpdated(...configurations: IConfigurationNode[]) {
666 this._onDidSchemaChange.fire();
667 }
669 > public registerOverrideIdentifiers(overrideIdentifiers: string[]): void {
670 this.doRegisterOverrideIdentifiers(overrideIdentifiers);
671 this._onDidSchemaChange.fire();
672 }
674 > private doRegisterOverrideIdentifiers(overrideIdentifiers: string[]) {
675 for (const overrideIdentifier of overrideIdentifiers) {
676 this.overrideIdentifiers.add(overrideIdentifier);
678 this.updateOverridePropertyPatternKey();
679 }
681 > private doRegisterConfigurations(configurations: IConfigurationNode[], validate: boolean, bucket: Set<string>): void {
683 > configurations.forEach(configuration => {
684 >
685 > this.validateAndRegisterProperties(configuration, validate, configuration.extensionInfo, configuration.restrictedProperties, undefined, bucket);
686 >
687 > this.configurationContributors.push(configuration);
688 > this.registerJSONConfiguration(configuration);
689 > });
690 > }
692 > private doDeregisterConfigurations(configurations: IConfigurationNode[], bucket: Set<string>): void {
694 > const deregisterConfiguration = (configuration: IConfigurationNode) => {
695 if (configuration.properties) {
696 for (const key in configuration.properties) {
715 configuration.allOf?.forEach(node => deregisterConfiguration(node));
716 };
717 > for (const configuration of configurations) { configurationRegistry.ts
718 deregisterConfiguration(configuration);
719 const index = this.configurationContributors.indexOf(configuration);
722 }
723 }
726 > private validateAndRegisterProperties(configuration: IConfigurationNode, validate: boolean = true, extensionInfo: IExtensionInfo | undefined, restrictedProperties: string[] | undefined, scope: ConfigurationScope = ConfigurationScope.WINDOW, bucket: Set<string>): void {
727 > scope = types.isUndefinedOrNull(configuration.scope) ? scope : configuration.scope; configurationRegistry.ts
728 > const properties = configuration.properties;
729 > if (properties) {
730 > for (const key in properties) {
731 > const property: IRegisteredConfigurationPropertySchema = properties[key];
732 > property.section = {
733 > id: configuration.id,
734 > title: configuration.title,
735 > order: configuration.order,
736 > extensionInfo: configuration.extensionInfo
737 > };
738 > if (validate && validateProperty(key, property, extensionInfo?.id)) {
739 delete properties[key];
740 continue;
741 }
743 > property.source = extensionInfo;
744 >
745 > // update default value
746 > property.defaultDefaultValue = properties[key].default;
747 > this.updatePropertyDefaultValue(key, property);
748 >
749 > // update scope
750 > if (OVERRIDE_PROPERTY_REGEX.test(key)) {
751 property.scope = undefined; // No scope for overridable properties `[${identifier}]`
752 > } else { configurationRegistry.ts
753 > property.scope = types.isUndefinedOrNull(property.scope) ? scope : property.scope;
754 > property.restricted = types.isUndefinedOrNull(property.restricted) ? !!restrictedProperties?.includes(key) : property.restricted;
755 > }
756 >
757 > if (property.experiment) {
758 > if (!property.tags?.some(tag => tag.toLowerCase() === 'onexp')) { configurationRegistry.ts
759 > property.tags = property.tags ?? [];
760 > property.tags.push('onExP');
761 > }
762 > } else if (property.tags?.some(tag => tag.toLowerCase() === 'onexp')) { configurationRegistry.ts
763 console.error(`Invalid tag 'onExP' found for property '${key}'. Please use 'experiment' property instead.`);
764 property.experiment = { mode: 'startup' };
765 }
767 > const excluded = properties[key].hasOwnProperty('included') && !properties[key].included;
768 > const policyName = properties[key].policy?.name; configurationRegistry.ts
769 > const policyReferenceName = properties[key].policyReference?.name;
770 >
771 > if (excluded) {
772 this.excludedConfigurationProperties[key] = properties[key];
773 if (policyName) {
780 }
781 delete properties[key];
782 > } else { configurationRegistry.ts
783 > bucket.add(key);
784 > if (policyName) {
785 > this.policyConfigurations.set(policyName, key); configurationRegistry.ts
786 > }
787 > if (policyReferenceName) { configurationRegistry.ts
788 this.addPolicyReferenceConfiguration(policyReferenceName, key);
789 }
790 > this.configurationProperties[key] = properties[key]; configurationRegistry.ts
791 > if (!properties[key].deprecationMessage && properties[key].markdownDeprecationMessage) {
792 // If not set, default deprecationMessage to the markdown source
793 properties[key].deprecationMessage = properties[key].markdownDeprecationMessage;
794 }
797 >
798 > }
799 > }
800 > const subNodes = configuration.allOf;
801 > if (subNodes) {
802 for (const node of subNodes) {
803 this.validateAndRegisterProperties(node, validate, extensionInfo, restrictedProperties, scope, bucket);
804 }
805 }
808 > private addPolicyReferenceConfiguration(policyName: PolicyName, key: string): void {
809 let keys = this.policyReferenceConfigurations.get(policyName);
810 if (!keys) {
814 keys.add(key);
815 }
817 > // Only for tests
818 > getConfigurations(): IConfigurationNode[] {
819 return this.configurationContributors;
820 }
822 > getConfigurationProperties(): IStringDictionary<IRegisteredConfigurationPropertySchema> {
823 > return this.configurationProperties; configurationRegistry.ts
824 > }
826 > getPolicyConfigurations(): Map<PolicyName, string> {
827 > return this.policyConfigurations; configurationRegistry.ts
828 > }
830 > getPolicyReferenceConfigurations(): Map<PolicyName, Set<string>> {
831 return this.policyReferenceConfigurations;
832 }
834 > getExcludedConfigurationProperties(): IStringDictionary<IRegisteredConfigurationPropertySchema> {
835 return this.excludedConfigurationProperties;
836 }
838 > getRegisteredDefaultConfigurations(): IConfigurationDefaults[] {
839 return [...this.registeredConfigurationDefaults];
840 }
842 > getConfigurationDefaultsOverrides(): Map<string, IConfigurationDefaultOverrideValue> {
843 const configurationDefaultsOverrides = new Map<string, IConfigurationDefaultOverrideValue>();
844 for (const [key, value] of this.configurationDefaultsOverrides) {
849 return configurationDefaultsOverrides;
850 }
852 > private registerJSONConfiguration(configuration: IConfigurationNode) {
853 > const register = (configuration: IConfigurationNode) => { configurationRegistry.ts
854 > const properties = configuration.properties;
855 > if (properties) {
856 > for (const key in properties) {
857 > this.updateSchema(key, properties[key]); configurationRegistry.ts
858 > }
860 > const subNodes = configuration.allOf;
861 > subNodes?.forEach(register);
862 > };
863 > register(configuration);
864 > }
866 > private updateSchema(key: string, property: IConfigurationPropertySchema): void {
867 > allSettings.properties[key] = property; configurationRegistry.ts
868 > switch (property.scope) {
869 > case ConfigurationScope.APPLICATION:
870 > applicationSettings.properties[key] = property; configurationRegistry.ts
871 > break;
872 > case ConfigurationScope.MACHINE: configurationRegistry.ts
873 > machineSettings.properties[key] = property; configurationRegistry.ts
874 > break;
875 > case ConfigurationScope.APPLICATION_MACHINE: configurationRegistry.ts
876 applicationMachineSettings.properties[key] = property;
877 break;
878 > case ConfigurationScope.MACHINE_OVERRIDABLE: configurationRegistry.ts
879 machineOverridableSettings.properties[key] = property;
880 break;
881 > case ConfigurationScope.WINDOW: configurationRegistry.ts
882 windowSettings.properties[key] = property;
883 break;
884 > case ConfigurationScope.RESOURCE: configurationRegistry.ts
885 resourceSettings.properties[key] = property;
886 break;
887 > case ConfigurationScope.LANGUAGE_OVERRIDABLE: configurationRegistry.ts
888 resourceSettings.properties[key] = property;
889 this.resourceLanguageSettingsSchema.properties![key] = property;
890 break;
892 > }
894 > private removeFromSchema(key: string, property: IConfigurationPropertySchema): void {
895 delete allSettings.properties[key];
896 switch (property.scope) {
917 }
918 }
920 > private updateOverridePropertyPatternKey(): void {
921 for (const overrideIdentifier of this.overrideIdentifiers.values()) {
922 const overrideIdentifierProperty = `[${overrideIdentifier}]`;
937 }
938 }
940 > private registerOverridePropertyPatternKey(): void {
941 > const resourceLanguagePropertiesSchema: IJSONSchema = {
942 > type: 'object',
943 > description: nls.localize('overrideSettings.defaultDescription', "Configure editor settings to be overridden for a language."),
944 > errorMessage: nls.localize('overrideSettings.errorMessage', "This setting does not support per-language configuration."),
945 > $ref: resourceLanguageSettingsSchemaId,
946 > };
947 > allSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
948 > applicationSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
949 > applicationMachineSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
950 > machineSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
951 > machineOverridableSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
952 > windowSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
953 > resourceSettings.patternProperties[OVERRIDE_PROPERTY_PATTERN] = resourceLanguagePropertiesSchema;
954 > this._onDidSchemaChange.fire();
955 > }
956 >
957 > private updatePropertyDefaultValue(key: string, property: IRegisteredConfigurationPropertySchema): void {
958 > const configurationdefaultOverride = this.configurationDefaultsOverrides.get(key)?.configurationDefaultOverrideValue; configurationRegistry.ts
959 > let defaultValue = undefined;
960 > let defaultSource = undefined;
961 > if (configurationdefaultOverride
962 && (!property.disallowConfigurationDefault || !configurationdefaultOverride.source) // Prevent overriding the default value if the property is disallowed to be overridden by configuration defaults from extensions
964 defaultValue = configurationdefaultOverride.value;
965 defaultSource = configurationdefaultOverride.source;
966 }
967 > if (types.isUndefined(defaultValue)) { configurationRegistry.ts
968 > defaultValue = property.defaultDefaultValue; configurationRegistry.ts
969 > defaultSource = undefined;
970 > }
971 > if (types.isUndefined(defaultValue)) { configurationRegistry.ts
972 > defaultValue = getDefaultValue(property.type); configurationRegistry.ts
973 > }
974 > property.default = defaultValue; configurationRegistry.ts
975 > property.defaultValueSource = defaultSource;
976 > }
978 >
979 > const OVERRIDE_IDENTIFIER_PATTERN = `\\[([^\\]]+)\\]`;
980 > const OVERRIDE_IDENTIFIER_REGEX = new RegExp(OVERRIDE_IDENTIFIER_PATTERN, 'g');
981 > export const OVERRIDE_PROPERTY_PATTERN = `^(${OVERRIDE_IDENTIFIER_PATTERN})+$`;
982 > export const OVERRIDE_PROPERTY_REGEX = new RegExp(OVERRIDE_PROPERTY_PATTERN);
983 >
984 > export function overrideIdentifiersFromKey(key: string): string[] {
985 const identifiers: string[] = [];
986 if (OVERRIDE_PROPERTY_REGEX.test(key)) {
996 return distinct(identifiers);
997 }
999 > export function keyFromOverrideIdentifiers(overrideIdentifiers: string[]): string {
1000 return overrideIdentifiers.reduce((result, overrideIdentifier) => `${result}[${overrideIdentifier}]`, '');
1001 }
1003 > export function getDefaultValue(type: string | string[] | undefined) {
1004 > const t = Array.isArray(type) ? type[0] : <string>type; configurationRegistry.ts
1005 > switch (t) {
1006 > case 'boolean':
1007 return false;
1008 > case 'integer': configurationRegistry.ts
1009 > case 'number':
1010 return 0;
1011 > case 'string': configurationRegistry.ts
1012 > return ''; configurationRegistry.ts
1013 > case 'array': configurationRegistry.ts
1014 > return []; configurationRegistry.ts
1015 > case 'object': configurationRegistry.ts
1016 return {};
1017 > default: configurationRegistry.ts
1018 return null;
1020 > }
1022 > const configurationRegistry = new ConfigurationRegistry();
1023 > Registry.add(Extensions.Configuration, configurationRegistry);
1024 >
1025 > export function validateProperty(property: string, schema: IRegisteredConfigurationPropertySchema, extensionId?: string): string | null {
1026 > if (!property.trim()) { configurationRegistry.ts
1027 return nls.localize('config.property.empty', "Cannot register an empty property");
1028 }
1029 > if (OVERRIDE_PROPERTY_REGEX.test(property)) { configurationRegistry.ts
1030 return nls.localize('config.property.languageDefault', "Cannot register '{0}'. This matches property pattern '\\\\[.*\\\\]$' for describing language specific editor settings. Use 'configurationDefaults' contribution.", property);
1031 }
1032 > if (configurationRegistry.getConfigurationProperties()[property] !== undefined && (!extensionId || !EXTENSION_UNIFICATION_EXTENSION_IDS.has(extensionId.toLowerCase()))) { configurationRegistry.ts
1033 return nls.localize('config.property.duplicate', "Cannot register '{0}'. This property is already registered.", property);
1034 }
1035 > if (schema.policy && schema.policyReference) { configurationRegistry.ts
1036 return nls.localize('config.policy.bothPolicyAndReference', "Cannot register '{0}'. A setting must not declare both 'policy' and 'policyReference'.", property);
1037 }
1038 > if (schema.policy?.name && configurationRegistry.getPolicyConfigurations().get(schema.policy?.name) !== undefined) { configurationRegistry.ts
1039 return nls.localize('config.policy.duplicate', "Cannot register '{0}'. The associated policy {1} is already registered with {2}. To attach another setting to the same policy, use 'policyReference'.", property, schema.policy?.name, configurationRegistry.getPolicyConfigurations().get(schema.policy?.name));
1040 }
1041 > return null; configurationRegistry.ts
1042 > }
1044 > export function getScopes(): [string, ConfigurationScope | undefined][] {
1045 const scopes: [string, ConfigurationScope | undefined][] = [];
1046 const configurationProperties = configurationRegistry.getConfigurationProperties();
1052 return scopes;
1053 }
1055 > export function getAllConfigurationProperties(configurationNode: IConfigurationNode[]): IStringDictionary<IRegisteredConfigurationPropertySchema> {
1056 const result: IStringDictionary<IRegisteredConfigurationPropertySchema> = {};
1057 for (const configuration of configurationNode) {
1068 return result;
1069 }
1071 > export function parseScope(scope: string): ConfigurationScope {
1072 switch (scope) {
1073 case 'application':
1085 }
1086 }
1088 > // Used for extension unification. Should be removed when complete.
1089 > export const EXTENSION_UNIFICATION_EXTENSION_IDS: Set<string> = new Set(product.defaultChatAgent ? [product.defaultChatAgent.extensionId, product.defaultChatAgent.chatExtensionId].map(id => id.toLowerCase()) : []);
src/vs/workbench/contrib/chat/common/tools/languageModelToolsService.ts 621 covered LOC · 39 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- languageModelToolsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Separator } from '../../../../../base/common/actions.js';
7 > import { VSBuffer } from '../../../../../base/common/buffer.js';
8 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
9 > import { Event } from '../../../../../base/common/event.js';
10 > import { IMarkdownString } from '../../../../../base/common/htmlContent.js';
11 > import { Iterable } from '../../../../../base/common/iterator.js';
12 > import { IJSONSchema } from '../../../../../base/common/jsonSchema.js';
13 > import { Disposable, IDisposable, toDisposable } from '../../../../../base/common/lifecycle.js';
14 > import { Schemas } from '../../../../../base/common/network.js';
15 > import { derived, IObservable, IReader, ITransaction, ObservableSet } from '../../../../../base/common/observable.js';
16 > import { ThemeIcon } from '../../../../../base/common/themables.js';
17 > import { URI } from '../../../../../base/common/uri.js';
18 > import { Location } from '../../../../../editor/common/languages.js';
19 > import { localize } from '../../../../../nls.js';
20 > import { ConfirmationOption } from '../../../../../platform/agentHost/common/state/protocol/state.js';
21 > import { ContextKeyExpression, IContextKeyService } from '../../../../../platform/contextkey/common/contextkey.js';
22 > import { ExtensionIdentifier } from '../../../../../platform/extensions/common/extensions.js';
23 > import { ByteSize } from '../../../../../platform/files/common/files.js';
24 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
25 > import { IProgress } from '../../../../../platform/progress/common/progress.js';
26 > import { ChatRequestToolReferenceEntry } from '../attachments/chatVariableEntries.js';
27 > import { IVariableReference } from '../chatModes.js';
28 > import { ConfirmedReason, IChatAgentFeedbackReviewConfirmationData, IChatAutomationConfigurationData, IChatAutomationConfiguredData, IChatExtensionsContent, IChatModifiedFilesConfirmationData, IChatSearchToolInvocationData, IChatSessionCreatedData, IChatSimpleToolInvocationData, IChatSubagentToolInvocationData, IChatTodoListContent, IChatToolInputInvocationData, IChatToolInvocation, type IChatTerminalToolInvocationData } from '../chatService/chatService.js';
29 > import { ILanguageModelChatMetadata, LanguageModelPartAudience } from '../languageModels.js';
30 > import { UserSelectedTools } from '../participants/chatAgents.js';
31 > import { PromptElementJSON, stringifyPromptElementJSON } from './promptTsxTypes.js';
32 >
33 > /**
34 > * Selector for matching language models by vendor, family, version, or id.
35 > * Used to filter tools to specific models or model families.
36 > */
37 > export interface ILanguageModelChatSelector {
38 > readonly vendor?: string;
39 > readonly family?: string;
40 > readonly version?: string;
41 > readonly id?: string;
42 > }
43 >
44 > export interface IToolData {
45 > readonly id: string;
46 > readonly source: ToolDataSource;
47 > readonly toolReferenceName?: string;
48 > readonly legacyToolReferenceFullNames?: readonly string[];
49 > readonly icon?: { dark: URI; light?: URI } | ThemeIcon;
50 > readonly when?: ContextKeyExpression;
51 > readonly tags?: readonly string[];
52 > readonly displayName: string;
53 > readonly userDescription?: string;
54 > readonly modelDescription: string;
55 > readonly inputSchema?: IJSONSchema;
56 > readonly canBeReferencedInPrompt?: boolean;
57 > /**
58 > * True if the tool runs in the (possibly remote) workspace, false if it runs
59 > * on the host, undefined if known.
60 > */
61 > readonly runsInWorkspace?: boolean;
62 > readonly alwaysDisplayInputOutput?: boolean;
63 > /** True if this tool might ask for pre-approval */
64 > readonly canRequestPreApproval?: boolean;
65 > /** True if this tool might ask for post-approval */
66 > readonly canRequestPostApproval?: boolean;
67 > /**
68 > * Model selectors that this tool is available for.
69 > * If defined, the tool is only available when the selected model matches one of the selectors.
70 > */
71 > readonly models?: readonly ILanguageModelChatSelector[];
72 > }
73 >
74 > /**
75 > * Check if a tool matches the given model metadata based on the tool's `models` selectors.
76 > * If the tool has no `models` defined, it matches all models.
77 > * If model is undefined, model-specific filtering is skipped (tool is included).
78 > */
79 > export function toolMatchesModel(toolData: IToolData, model: ILanguageModelChatMetadata | undefined): boolean {
80 // If no model selectors are defined, the tool is available for all models
81 if (!toolData.models || toolData.models.length === 0) {
94 );
95 }
97 > export interface IToolProgressStep {
98 > readonly message: string | IMarkdownString | undefined;
99 > /** 0-1 progress of the tool call */
100 > readonly progress?: number;
101 > }
102 >
103 > export type ToolProgress = IProgress<IToolProgressStep>;
104 >
105 > export type ToolDataSource =
106 > | {
107 > type: 'extension';
108 > label: string;
109 > extensionId: ExtensionIdentifier;
110 > }
111 > | {
112 > type: 'mcp';
113 > label: string;
114 > serverLabel: string | undefined;
115 > instructions: string | undefined;
116 > collectionId: string;
117 > definitionId: string;
118 > }
119 > | {
120 > type: 'user';
121 > label: string;
122 > file: URI;
123 > }
124 > | {
125 > type: 'internal';
126 > label: string;
127 > } | {
128 > type: 'external';
129 > label: string;
130 > };
131 >
132 > export namespace ToolDataSource {
133 >
134 > export const Internal: ToolDataSource = { type: 'internal', label: 'Built-In' };
135 >
136 > /** External tools may not be contributed or invoked, but may be invoked externally and described in an IChatToolInvocationSerialized */
137 > export const External: ToolDataSource = { type: 'external', label: 'External' };
138 >
139 > export function toKey(source: ToolDataSource): string {
140 switch (source.type) {
141 case 'extension': return `extension:${source.extensionId.value}`;
146 }
147 }
149 > export function equals(a: ToolDataSource, b: ToolDataSource): boolean {
150 return toKey(a) === toKey(b);
151 }
153 > export function classify(source: ToolDataSource): { readonly ordinal: number; readonly label: string } {
154 if (source.type === 'internal') {
155 return { ordinal: 1, label: localize('builtin', 'Built-In') };
162 }
163 }
165 >
166 > /**
167 > * Pre-tool-use hook result passed from the extension when the hook was executed externally.
168 > */
169 > export interface IExternalPreToolUseHookResult {
170 > permissionDecision?: 'allow' | 'deny' | 'ask';
171 > permissionDecisionReason?: string;
172 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
173 > updatedInput?: Record<string, any>;
174 > }
175 >
176 > export interface IToolInvocation {
177 > callId: string;
178 > toolId: string;
179 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
180 > parameters: Record<string, any>;
181 > tokenBudget?: number;
182 > context: IToolInvocationContext | undefined;
183 > chatRequestId?: string;
184 > chatInteractionId?: string;
185 > /**
186 > * Optional tool call ID from the chat stream, used to correlate with pending streaming tool calls.
187 > */
188 > chatStreamToolCallId?: string;
189 > /**
190 > * Lets us add some nicer UI to toolcalls that came from a sub-agent, but in the long run, this should probably just be rendered in a similar way to thinking text + tool call groups
191 > */
192 > subAgentInvocationId?: string;
193 > toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfigurationData | IChatAutomationConfiguredData;
194 > modelId?: string;
195 > userSelectedTools?: UserSelectedTools;
196 > /** The label of the custom button selected by the user during confirmation, if custom buttons were used. */
197 > selectedCustomButton?: string;
198 > /** Pre-tool-use hook result passed from the extension, if the hook was already executed externally. */
199 > preToolUseResult?: IExternalPreToolUseHookResult;
200 > /**
201 > * A confirmation reason resolved out-of-band by the caller (e.g. the agent
202 > * host, which decides auto-approval server-side). When set, the invocation
203 > * is treated as already auto-approved and transitions straight to executing
204 > * without ever entering the `WaitingForConfirmation` state. This avoids a
205 > * transient "needs input" flicker in surfaces (like the sessions list) that
206 > * observe pending confirmations, for tool calls that will be auto-approved
207 > * anyway.
208 > */
209 > preApproved?: ConfirmedReason;
210 > /**
211 > * Optional W3C trace context `traceparent` value identifying the parent distributed
212 > * tracing span for this tool invocation. Forwarded to MCP tool implementations as
213 > * `_meta.traceparent` (MCP SEP-414).
214 > */
215 > traceparent?: string;
216 > /** Optional W3C trace context `tracestate` value paired with {@link traceparent}. */
217 > tracestate?: string;
218 > }
219 >
220 > export interface IToolInvocationContext {
221 > readonly sessionResource: URI;
222 > /**
223 > * The working directory URI associated with this session.
224 > * Only set in the agents window context where each session can
225 > * have its own working directory that differs from the workspace folders.
226 > */
227 > readonly workingDirectory?: URI;
228 > }
229 >
230 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
231 > export function isToolInvocationContext(obj: any): obj is IToolInvocationContext {
232 return obj !== null && typeof obj === 'object' && URI.isUri(obj.sessionResource);
233 }
235 > export interface IToolInvocationPreparationContext {
236 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
237 > parameters: any;
238 > toolCallId: string;
239 > chatRequestId?: string;
240 > chatSessionResource: URI | undefined;
241 > chatInteractionId?: string;
242 > modelId?: string;
243 > /** If set, tells the tool that it should include confirmation messages. */
244 > forceConfirmationReason?: string;
245 > /**
246 > * The working directory URI for the session, if set.
247 > * Used by tools to resolve relative paths and check file access.
248 > */
249 > workingDirectory?: URI;
250 > }
251 >
252 > export type ToolInputOutputBase = {
253 > /** Mimetype of the value, optional */
254 > mimeType?: string;
255 > /** URI of the resource on the MCP server. */
256 > uri?: URI;
257 > /** If true, this part came in as a resource reference rather than direct data. */
258 > asResource?: boolean;
259 > /** Audience of the data part */
260 > audience?: LanguageModelPartAudience[];
261 > };
262 >
263 > export type ToolInputOutputEmbedded = ToolInputOutputBase & {
264 > type: 'embed';
265 > value: string;
266 > /** If true, value is text. If false or not given, value is base64 */
267 > isText?: boolean;
268 > };
269 >
270 > export type ToolInputOutputReference = ToolInputOutputBase & { type: 'ref'; uri: URI };
271 >
272 > export interface IToolResultInputOutputDetails {
273 > readonly input: string;
274 > /** Language identifier for syntax highlighting the input. Defaults to 'json'. */
275 > readonly inputLanguage?: string;
276 > readonly output: (ToolInputOutputEmbedded | ToolInputOutputReference)[];
277 > readonly isError?: boolean;
278 > /** Raw MCP tool result for MCP App UI rendering */
279 > readonly mcpOutput?: unknown;
280 > }
281 >
282 > export interface IToolResultOutputDetails {
283 > readonly output: { type: 'data'; mimeType: string; value: VSBuffer };
284 > }
285 >
286 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
287 > export function isToolResultInputOutputDetails(obj: any): obj is IToolResultInputOutputDetails {
288 return typeof obj === 'object' && typeof obj?.input === 'string' && (typeof obj?.output === 'string' || Array.isArray(obj?.output));
289 }
291 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
292 > export function isToolResultOutputDetails(obj: any): obj is IToolResultOutputDetails {
293 return typeof obj === 'object' && typeof obj?.output === 'object' && typeof obj?.output?.mimeType === 'string' && obj?.output?.type === 'data';
294 }
296 > export interface IToolResult {
297 > content: (IToolResultPromptTsxPart | IToolResultTextPart | IToolResultDataPart)[];
298 > toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfiguredData;
299 > toolResultMessage?: string | IMarkdownString;
300 > toolResultDetails?: Array<URI | Location> | IToolResultInputOutputDetails | IToolResultOutputDetails;
301 > toolResultError?: string | boolean;
302 > toolMetadata?: unknown;
303 > /** Whether to ask the user to confirm these tool results. Overrides {@link IToolConfirmationMessages.confirmResults}. */
304 > confirmResults?: boolean;
305 > }
306 >
307 > export function toolContentToA11yString(part: IToolResult['content']) {
308 return part.map(p => {
309 switch (p.kind) {
317 }).join(', ');
318 }
320 > export function toolResultHasBuffers(result: IToolResult): boolean {
321 return result.content.some(part => part.kind === 'data');
322 }
324 > export interface IToolResultPromptTsxPart {
325 > kind: 'promptTsx';
326 > value: unknown;
327 > }
328 >
329 > export function stringifyPromptTsxPart(part: IToolResultPromptTsxPart): string {
330 return stringifyPromptElementJSON(part.value as PromptElementJSON);
331 }
333 > export interface IToolResultTextPart {
334 > kind: 'text';
335 > value: string;
336 > audience?: LanguageModelPartAudience[];
337 > title?: string;
338 > }
339 >
340 > export interface IToolResultDataPart {
341 > kind: 'data';
342 > value: {
343 > mimeType: string;
344 > data: VSBuffer;
345 > };
346 > audience?: LanguageModelPartAudience[];
347 > title?: string;
348 > }
349 >
350 > export type IToolApprovalReason =
351 > | { readonly status: 'loading' }
352 > | {
353 > readonly status: 'complete';
354 > readonly explanation: string | IMarkdownString;
355 > readonly safety: number;
356 > };
357 >
358 > export interface IToolConfirmationMessages {
359 > /** Title for the confirmation. If set, the user will be asked to confirm execution of the tool */
360 > title?: string | IMarkdownString;
361 > /** MUST be set if `title` is also set */
362 > message?: string | IMarkdownString;
363 > disclaimer?: string | IMarkdownString;
364 > /** Model-provided assessment of whether automatic approval is safe. */
365 > approvalReason?: IToolApprovalReason;
366 > /** Whether this confirmation is eligible for automatic approval. */
367 > allowAutoConfirm?: boolean;
368 > terminalCustomActions?: ToolConfirmationAction[];
369 > /** If true, confirmation will be requested after the tool executes and before results are sent to the model */
370 > confirmResults?: boolean;
371 > /** If title is not set (no confirmation needed), this reason will be shown to explain why confirmation was not needed */
372 > confirmationNotNeededReason?: string | IMarkdownString;
373 > /** Custom options to display instead of the default Allow/Skip buttons. */
374 > customOptions?: ConfirmationOption[];
375 > /** When set, shows an additional approval option to approve this particular combination of tool and arguments */
376 > approveCombination?: {
377 > /** Human-readable label for the approval option */
378 > label: string | IMarkdownString;
379 > /** Precomputed SHA-256 key for the combination (set during tool preparation) */
380 > key: string;
381 > /** String representation of the arguments for this combination */
382 > arguments?: string;
383 > };
384 > }
385 >
386 > export interface IToolConfirmationAction {
387 > label: string;
388 > disabled?: boolean;
389 > tooltip?: string;
390 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
391 > data: any;
392 > }
393 >
394 > export type ToolConfirmationAction = IToolConfirmationAction | Separator;
395 >
396 > export enum ToolInvocationPresentation {
397 > Hidden = 'hidden',
398 > HiddenAfterComplete = 'hiddenAfterComplete'
399 > }
400 >
401 > export interface IToolInvocationStreamContext {
402 > toolCallId: string;
403 > rawInput: unknown;
404 > chatRequestId?: string;
405 > chatSessionResource?: URI;
406 > chatInteractionId?: string;
407 > }
408 >
409 > export interface IStreamedToolInvocation {
410 > invocationMessage?: string | IMarkdownString;
411 > }
412 >
413 > export interface IPreparedToolInvocation {
414 > invocationMessage?: string | IMarkdownString;
415 > pastTenseMessage?: string | IMarkdownString;
416 > originMessage?: string | IMarkdownString;
417 > confirmationMessages?: IToolConfirmationMessages;
418 > presentation?: ToolInvocationPresentation;
419 > icon?: ThemeIcon;
420 > toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfigurationData | IChatAutomationConfiguredData;
421 > }
422 >
423 > export interface IToolImpl {
424 > invoke(invocation: IToolInvocation, countTokens: CountTokensCallback, progress: ToolProgress, token: CancellationToken): Promise<IToolResult>;
425 > prepareToolInvocation?(context: IToolInvocationPreparationContext, token: CancellationToken): Promise<IPreparedToolInvocation | undefined>;
426 > handleToolStream?(context: IToolInvocationStreamContext, token: CancellationToken): Promise<IStreamedToolInvocation | undefined>;
427 > }
428 >
429 > export interface IToolSet {
430 > readonly id: string;
431 > readonly referenceName: string;
432 > readonly icon: ThemeIcon;
433 > readonly source: ToolDataSource;
434 > readonly description?: string;
435 > /** A longer, human-readable description of what the tool set is for, shown as a subtitle in the Chat Customizations "Tools" section. */
436 > readonly detail?: string;
437 > readonly legacyFullNames?: string[];
438 > /** When true, this tool set is deprecated: it is hidden from the Chat Customizations "Tools" section and these groupings will be removed when the Local harness is dropped. */
439 > readonly deprecated?: boolean;
440 > /** When true, this tool set is hidden from the chat tools picker (e.g. a customizations-only grouping). */
441 > readonly hiddenInToolsPicker?: boolean;
442 >
443 > getTools(r?: IReader): Iterable<IToolData>;
444 > }
445 >
446 >
447 >
448 > /**
449 > * Maps tools and tool sets to their enablement state. Use a class to control creation of the map and ensure
450 > * that it is not mutated after creation.
451 > */
452 > export class ToolAndToolSetEnablementMap implements Iterable<[IToolData | IToolSet, boolean]> {
453 >
454 > static fromEntries(entries: Iterable<[IToolData | IToolSet, boolean]>): ToolAndToolSetEnablementMap {
455 > return new ToolAndToolSetEnablementMap(new Map(entries));
456 > }
457 >
458 > static fromMap(map: Map<IToolData | IToolSet, boolean>): ToolAndToolSetEnablementMap {
459 return new ToolAndToolSetEnablementMap(new Map(map));
460 }
462 > private constructor(private readonly _map: Map<IToolData | IToolSet, boolean>) {
465 > [Symbol.iterator](): IterableIterator<[IToolData | IToolSet, boolean]> {
466 > return this._map[Symbol.iterator](); languageModelToolsService.ts
467 > }
469 > public get(toolOrToolSet: IToolData | IToolSet): boolean | undefined {
470 return this._map.get(toolOrToolSet);
471 }
473 > public has(toolOrToolSet: IToolData | IToolSet): boolean {
474 return this._map.has(toolOrToolSet);
475 }
477 > public get size(): number {
478 return this._map.size;
479 }
481 > public entries(): IterableIterator<[IToolData | IToolSet, boolean]> {
482 return this._map.entries();
483 }
485 >
486 > export function isToolSet(obj: IToolData | IToolSet | undefined): obj is IToolSet {
487 return !!obj && (obj as IToolSet).getTools !== undefined;
488 }
490 > export class ToolSet implements IToolSet {
491 >
492 > protected readonly _tools = new ObservableSet<IToolData>();
493 >
494 > protected readonly _toolSets = new ObservableSet<IToolSet>();
495 >
496 > /**
497 > * A homogenous tool set only contains tools from the same source as the tool set itself
498 > */
499 > readonly isHomogenous: IObservable<boolean>;
500 >
501 > constructor(
502 > readonly id: string, languageModelToolsService.ts
503 > readonly referenceName: string,
504 > readonly icon: ThemeIcon,
505 > readonly source: ToolDataSource,
506 > readonly description: string | undefined,
507 > readonly detail: string | undefined,
508 > readonly legacyFullNames: string[] | undefined,
509 > readonly deprecated: boolean | undefined,
510 > readonly hiddenInToolsPicker: boolean | undefined,
511 > private readonly _contextKeyService: IContextKeyService,
512 > ) {
513 >
514 > this.isHomogenous = derived(r => {
515 return !Iterable.some(this._tools.observable.read(r), tool => !ToolDataSource.equals(tool.source, this.source))
516 && !Iterable.some(this._toolSets.observable.read(r), toolSet => !ToolDataSource.equals(toolSet.source, this.source));
518 > }
520 > addTool(data: IToolData, tx?: ITransaction): IDisposable {
521 this._tools.add(data, tx);
522 return toDisposable(() => {
524 });
525 }
527 > addToolSet(toolSet: IToolSet, tx?: ITransaction): IDisposable {
528 if (toolSet === this) {
529 return Disposable.None;
534 });
535 }
537 > getTools(r?: IReader): Iterable<IToolData> {
538 return Iterable.concat(
539 Iterable.filter(this._tools.observable.read(r), toolData => this._contextKeyService.contextMatchesRules(toolData.when)),
541 );
542 }
544 >
545 > export class ToolSetForModel {
546 > public get id() {
547 > return this._toolSet.id;
548 > }
549 >
550 > public get referenceName() {
551 return this._toolSet.referenceName;
552 }
554 > public get icon() {
555 return this._toolSet.icon;
556 }
558 > public get source() {
559 return this._toolSet.source;
560 }
562 > public get description() {
563 return this._toolSet.description;
564 }
566 > public get detail() {
567 return this._toolSet.detail;
568 }
570 > public get legacyFullNames() {
571 return this._toolSet.legacyFullNames;
572 }
574 > public get deprecated() {
575 return this._toolSet.deprecated;
576 }
578 > public get hiddenInToolsPicker() {
579 return this._toolSet.hiddenInToolsPicker;
580 }
582 > constructor(
583 private readonly _toolSet: IToolSet,
584 private readonly model: ILanguageModelChatMetadata | undefined,
585 private readonly toolFilter?: (toolData: IToolData) => boolean,
586 ) { }
588 > public getTools(r?: IReader): Iterable<IToolData> {
589 return Iterable.filter(this._toolSet.getTools(r), toolData => toolMatchesModel(toolData, this.model) && (!this.toolFilter || this.toolFilter(toolData)));
590 }
592 >
593 >
594 > export interface IBeginToolCallOptions {
595 > toolCallId: string;
596 > toolId: string;
597 > chatRequestId?: string;
598 > sessionResource?: URI;
599 > subagentInvocationId?: string;
600 > /**
601 > * Create the streaming invocation even when the tool does not
602 > * implement `handleToolStream`. Used by callers that need a
603 > * `ChatToolInvocation` handle to observe state transitions (e.g.
604 > * confirmation) before invoking the tool.
605 > */
606 > force?: boolean;
607 > }
608 >
609 > export interface IToolInvokedEvent {
610 > readonly toolId: string;
611 > readonly sessionResource: URI | undefined;
612 > readonly requestId: string | undefined;
613 > readonly subagentInvocationId: string | undefined;
614 > }
615 >
616 > export const ILanguageModelToolsService = createDecorator<ILanguageModelToolsService>('ILanguageModelToolsService');
617 >
618 > export type CountTokensCallback = (input: string, token: CancellationToken) => Promise<number>;
619 >
620 > export interface ILanguageModelToolsService {
621 > _serviceBrand: undefined;
622 > readonly vscodeToolSet: ToolSet;
623 > readonly executeToolSet: ToolSet;
624 > readonly readToolSet: ToolSet;
625 > readonly agentToolSet: ToolSet;
626 > readonly onDidChangeTools: Event<void>;
627 > readonly onDidPrepareToolCallBecomeUnresponsive: Event<{ readonly sessionResource: URI; readonly toolData: IToolData }>;
628 > readonly onDidInvokeTool: Event<IToolInvokedEvent>;
629 > registerToolData(toolData: IToolData): IDisposable;
630 > registerToolImplementation(id: string, tool: IToolImpl): IDisposable;
631 > registerTool(toolData: IToolData, tool: IToolImpl): IDisposable;
632 >
633 > /**
634 > * Get all tools currently enabled (matching `when` clauses and model).
635 > * @param model The language model metadata to filter tools by. If undefined, model-specific filtering is skipped.
636 > */
637 > getTools(model: ILanguageModelChatMetadata | undefined): Iterable<IToolData>;
638 >
639 > /**
640 > * Creats an observable of enabled tools in the context. Note the observable
641 > * should be created and reused, not created per reader, for example:
642 > *
643 > * ```
644 > * const toolsObs = toolsService.observeTools(model);
645 > * autorun(reader => {
646 > * const tools = toolsObs.read(reader);
647 > * ...
648 > * });
649 > * ```
650 > * @param model The language model metadata to filter tools by. If undefined, model-specific filtering is skipped.
651 > */
652 > observeTools(model: ILanguageModelChatMetadata | undefined): IObservable<readonly IToolData[]>;
653 >
654 > /**
655 > * Get all registered tools regardless of enablement state.
656 > * Use this for configuration UIs, completions, etc. where all tools should be visible.
657 > */
658 > getAllToolsIncludingDisabled(): Iterable<IToolData>;
659 >
660 > /**
661 > * Get a tool by its ID. Does not check when clauses.
662 > */
663 > getTool(id: string): IToolData | undefined;
664 >
665 > /**
666 > * Get a tool by its reference name. Does not check when clauses.
667 > */
668 > getToolByName(name: string): IToolData | undefined;
669 >
670 > /**
671 > * Begin a tool call in the streaming phase.
672 > * Creates a ChatToolInvocation in the Streaming state and appends it to the chat.
673 > * Returns the invocation so it can be looked up later when invokeTool is called.
674 > */
675 > beginToolCall(options: IBeginToolCallOptions): IChatToolInvocation | undefined;
676 >
677 > /**
678 > * Update the streaming state of a pending tool call.
679 > * Calls the tool's handleToolStream method to get a custom invocation message.
680 > */
681 > updateToolStream(toolCallId: string, partialInput: unknown, token: CancellationToken): Promise<void>;
682 >
683 > invokeTool(invocation: IToolInvocation, countTokens: CountTokensCallback, token: CancellationToken): Promise<IToolResult>;
684 > cancelToolCallsForRequest(requestId: string): void;
685 > /** Flush any pending tool updates to the extension hosts. */
686 > flushToolUpdates(): void;
687 >
688 > readonly toolSets: IObservable<Iterable<IToolSet>>;
689 > getToolSetsForModel(model: ILanguageModelChatMetadata | undefined, reader?: IReader): Iterable<IToolSet>;
690 > getToolSet(id: string): IToolSet | undefined;
691 > getToolSetByName(name: string): IToolSet | undefined;
692 > createToolSet(source: ToolDataSource, id: string, referenceName: string, options?: { icon?: ThemeIcon; description?: string; detail?: string; legacyFullNames?: string[]; deprecated?: boolean; hiddenInToolsPicker?: boolean }): ToolSet & IDisposable;
693 >
694 > // tool names in prompt and agent files ('full reference names')
695 > getFullReferenceNames(): Iterable<string>;
696 > getFullReferenceName(tool: IToolData | IToolSet, toolSet?: IToolSet): string;
697 > getFullReferenceNameMap(): Map<IToolData | IToolSet, string>;
698 > getToolByFullReferenceName(fullReferenceName: string): IToolData | IToolSet | undefined;
699 > getDeprecatedFullReferenceNames(): Map<string, Set<string>>;
700 >
701 > /**
702 > * Gets the enablement maps based on the given set of references.
703 > * @param fullReferenceNames The full reference names of the tools and tool sets to enable.
704 > * @param model Optional language model metadata to filter tools by.
705 > * If undefined is passed, all tools will be returned, even if normally disabled.
706 > */
707 > toToolAndToolSetEnablementMap(fullReferenceNames: readonly string[], model: ILanguageModelChatMetadata | undefined): ToolAndToolSetEnablementMap;
708 >
709 > toFullReferenceNames(map: ToolAndToolSetEnablementMap): string[];
710 > toToolReferences(variableReferences: readonly IVariableReference[]): ChatRequestToolReferenceEntry[];
711 > }
712 >
713 >
714 > export function createToolInputUri(toolCallId: string): URI {
715 return URI.from({ scheme: Schemas.inMemory, path: `/lm/tool/${toolCallId}/tool_input.json` });
716 }
718 > export function createToolSchemaUri(toolOrId: IToolData | string): URI {
719 if (typeof toolOrId !== 'string') {
720 toolOrId = toolOrId.id;
722 return URI.from({ scheme: Schemas.vscode, authority: 'schemas', path: `/lm/tool/${toolOrId}` });
723 }
725 > export namespace SpecedToolAliases {
726 > export const execute = 'execute';
727 > export const edit = 'edit';
728 > export const search = 'search';
729 > export const agent = 'agent';
730 > export const read = 'read';
731 > export const web = 'web';
732 > export const todo = 'todo';
733 > }
734 >
735 > export namespace VSCodeToolReference {
736 > export const runSubagent = 'runSubagent';
737 > export const vscode = 'vscode';
738 >
739 > }
src/vs/base/common/lifecycle.ts 607 covered LOC · 146 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- lifecycle.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { compareBy, numberComparator } from './arrays.js';
7 > import { groupBy } from './collections.js';
8 > import { SetMap, ResourceMap } from './map.js';
9 > import { URI } from './uri.js';
10 > import { createSingleCallFunction } from './functional.js';
11 > import { Iterable } from './iterator.js';
12 > import { BugIndicatingError, onUnexpectedError } from './errors.js';
13 >
14 > // #region Disposable Tracking
15 >
16 > /**
17 > * Enables logging of potentially leaked disposables.
18 > *
19 > * A disposable is considered leaked if it is not disposed or not registered as the child of
20 > * another disposable. This tracking is very simple an only works for classes that either
21 > * extend Disposable or use a DisposableStore. This means there are a lot of false positives.
22 > */
23 > const TRACK_DISPOSABLES = false;
24 > let disposableTracker: IDisposableTracker | null = null;
25 >
26 > export interface IDisposableTracker {
27 > /**
28 > * Is called on construction of a disposable.
29 > */
30 > trackDisposable(disposable: IDisposable): void;
31 >
32 > /**
33 > * Is called when a disposable is registered as child of another disposable (e.g. {@link DisposableStore}).
34 > * If parent is `null`, the disposable is removed from its former parent.
35 > */
36 > setParent(child: IDisposable, parent: IDisposable | null): void;
37 >
38 > /**
39 > * Is called after a disposable is disposed.
40 > */
41 > markAsDisposed(disposable: IDisposable): void;
42 >
43 > /**
44 > * Indicates that the given object is a singleton which does not need to be disposed.
45 > */
46 > markAsSingleton(disposable: IDisposable): void;
47 > }
48 >
49 > export class GCBasedDisposableTracker implements IDisposableTracker {
50
51 private readonly _registry = new FinalizationRegistry<string>(heldValue => {
52 console.warn(`[LEAKED DISPOSABLE] ${heldValue}`);
53 });
55 > trackDisposable(disposable: IDisposable): void {
56 const stack = new Error('CREATED via:').stack!;
57 this._registry.register(disposable, stack, disposable);
58 }
60 > setParent(child: IDisposable, parent: IDisposable | null): void {
61 if (parent) {
62 this._registry.unregister(child);
65 }
66 }
68 > markAsDisposed(disposable: IDisposable): void {
69 this._registry.unregister(disposable);
70 }
72 > markAsSingleton(disposable: IDisposable): void {
73 this._registry.unregister(disposable);
74 }
75 > } lifecycle.ts
76 >
77 > export interface DisposableInfo {
78 > value: IDisposable;
79 > source: string | null;
80 > parent: IDisposable | null;
81 > isSingleton: boolean;
82 > idx: number;
83 > }
84 >
85 > export class DisposableTracker implements IDisposableTracker {
86 > private static idx = 0; lifecycle.ts
87 >
88 > private readonly livingDisposables = new Map<IDisposable, DisposableInfo>();
90 > private getDisposableData(d: IDisposable): DisposableInfo {
91 > let val = this.livingDisposables.get(d); lifecycle.ts
92 > if (!val) {
93 > val = { parent: null, source: null, isSingleton: false, value: d, idx: DisposableTracker.idx++ };
94 > this.livingDisposables.set(d, val);
95 > }
96 > return val;
97 > }
99 > trackDisposable(d: IDisposable): void {
100 > const data = this.getDisposableData(d); lifecycle.ts
101 > if (!data.source) {
102 > data.source =
103 > new Error().stack!;
104 > }
105 > }
106 > lifecycle.ts
107 > setParent(child: IDisposable, parent: IDisposable | null): void {
108 > const data = this.getDisposableData(child); lifecycle.ts
109 > data.parent = parent;
110 > }
111 > lifecycle.ts
112 > markAsDisposed(x: IDisposable): void {
113 > this.livingDisposables.delete(x); lifecycle.ts
114 > }
115 > lifecycle.ts
116 > markAsSingleton(disposable: IDisposable): void {
117 this.getDisposableData(disposable).isSingleton = true;
118 }
119 > lifecycle.ts
120 > private getRootParent(data: DisposableInfo, cache: Map<DisposableInfo, DisposableInfo>): DisposableInfo {
121 const cacheValue = cache.get(data);
122 if (cacheValue) {
128 return result;
129 }
130 > lifecycle.ts
131 > getTrackedDisposables(): IDisposable[] {
132 const rootParentCache = new Map<DisposableInfo, DisposableInfo>();
133
138 return leaking;
139 }
140 > lifecycle.ts
141 > computeLeakingDisposables(maxReported = 10, preComputedLeaks?: DisposableInfo[]): { leaks: DisposableInfo[]; details: string } | undefined {
142 > let uncoveredLeakingObjs: DisposableInfo[] | undefined; lifecycle.ts
143 > if (preComputedLeaks) {
144 uncoveredLeakingObjs = preComputedLeaks;
145 > } else { lifecycle.ts
146 > const rootParentCache = new Map<DisposableInfo, DisposableInfo>();
147 >
148 > const leakingObjects = [...this.livingDisposables.values()]
149 > .filter((info) => info.source !== null && !this.getRootParent(info, rootParentCache).isSingleton);
150 >
151 > if (leakingObjects.length === 0) {
152 > return; lifecycle.ts
153 > }
154 const leakingObjsSet = new Set(leakingObjects.map(o => o.value));
155
162 throw new Error('There are cyclic diposable chains!');
163 }
164 > } lifecycle.ts
165
166 if (!uncoveredLeakingObjs) {
224
225 return { leaks: uncoveredLeakingObjs, details: message };
226 > } lifecycle.ts
227 > } lifecycle.ts
228 >
229 > export function setDisposableTracker(tracker: IDisposableTracker | null): void {
230 > disposableTracker = tracker; lifecycle.ts
231 > }
232 > lifecycle.ts
233 > if (TRACK_DISPOSABLES) {
234 const __is_disposable_tracked__ = '__is_disposable_tracked__';
235 setDisposableTracker(new class implements IDisposableTracker {
268 });
269 }
270 > lifecycle.ts
271 > export function trackDisposable<T extends IDisposable>(x: T): T {
272 > disposableTracker?.trackDisposable(x); lifecycle.ts
273 > return x;
274 > }
275 > lifecycle.ts
276 > export function markAsDisposed(disposable: IDisposable): void {
277 > disposableTracker?.markAsDisposed(disposable); lifecycle.ts
278 > }
279 > lifecycle.ts
280 > function setParentOfDisposable(child: IDisposable, parent: IDisposable | null): void { lifecycle.ts
281 > disposableTracker?.setParent(child, parent);
282 > }
283 > lifecycle.ts
284 function setParentOfDisposables(children: IDisposable[], parent: IDisposable | null): void {
285 if (!disposableTracker) {
290 }
291 }
292 > lifecycle.ts
293 > /**
294 > * Indicates that the given object is a singleton which does not need to be disposed.
295 > */
296 > export function markAsSingleton<T extends IDisposable>(singleton: T): T {
297 disposableTracker?.markAsSingleton(singleton);
298 return singleton;
299 }
300 > lifecycle.ts
301 > // #endregion
302 >
303 > /**
304 > * An object that performs a cleanup operation when `.dispose()` is called.
305 > *
306 > * Some examples of how disposables are used:
307 > *
308 > * - An event listener that removes itself when `.dispose()` is called.
309 > * - A resource such as a file system watcher that cleans up the resource when `.dispose()` is called.
310 > * - The return value from registering a provider. When `.dispose()` is called, the provider is unregistered.
311 > */
312 > export interface IDisposable {
313 > dispose(): void;
314 > }
315 >
316 > /**
317 > * Check if `thing` is {@link IDisposable disposable}.
318 > */
319 > export function isDisposable<E>(thing: E): thing is E & IDisposable {
320 // eslint-disable-next-line local/code-no-any-casts
321 return typeof thing === 'object' && thing !== null && typeof (<IDisposable><any>thing).dispose === 'function' && (<IDisposable><any>thing).dispose.length === 0;
322 }
323 > lifecycle.ts
324 > /**
325 > * Disposes of the value(s) passed in.
326 > */
327 > export function dispose<T extends IDisposable>(disposable: T): T;
328 > export function dispose<T extends IDisposable>(disposable: T | undefined): T | undefined;
329 > export function dispose<T extends IDisposable, A extends Iterable<T> = Iterable<T>>(disposables: A): A;
330 > export function dispose<T extends IDisposable>(disposables: Array<T>): Array<T>;
331 > export function dispose<T extends IDisposable>(disposables: ReadonlyArray<T>): ReadonlyArray<T>;
332 > export function dispose<T extends IDisposable>(arg: T | Iterable<T> | undefined): any {
333 > if (Iterable.is(arg)) { lifecycle.ts
334 > const errors: any[] = []; lifecycle.ts
335 >
336 > for (const d of arg) {
337 > if (d) { lifecycle.ts
338 > try {
339 > d.dispose();
340 > } catch (e) {
341 errors.push(e);
342 }
343 > } lifecycle.ts
344 > }
345 > lifecycle.ts
346 > if (errors.length === 1) {
347 throw errors[0];
348 > } else if (errors.length > 1) { lifecycle.ts
349 throw new AggregateError(errors, 'Encountered errors while disposing of store');
350 }
351 > lifecycle.ts
352 > return Array.isArray(arg) ? [] : arg; lifecycle.ts
353 > } else if (arg) { lifecycle.ts
354 arg.dispose();
355 return arg;
356 }
357 > } lifecycle.ts
358 > lifecycle.ts
359 > export function disposeIfDisposable<T extends IDisposable | object>(disposables: Array<T>): Array<T> {
360 for (const d of disposables) {
361 if (isDisposable(d)) {
365 return [];
366 }
367 > lifecycle.ts
368 > /**
369 > * Combine multiple disposable values into a single {@link IDisposable}.
370 > */
371 > export function combinedDisposable(...disposables: IDisposable[]): IDisposable {
372 const parent = toDisposable(() => dispose(disposables));
373 setParentOfDisposables(disposables, parent);
374 return parent;
375 }
376 > lifecycle.ts
377 > class FunctionDisposable implements IDisposable {
378 > private _isDisposed: boolean;
379 > private readonly _fn: () => void;
380 >
381 > constructor(fn: () => void) {
382 > this._isDisposed = false; lifecycle.ts
383 > this._fn = fn;
384 > trackDisposable(this);
385 > }
386 > lifecycle.ts
387 > dispose() {
388 > if (this._isDisposed) { lifecycle.ts
389 return;
390 }
391 > if (!this._fn) { lifecycle.ts
392 throw new Error(`Unbound disposable context: Need to use an arrow function to preserve the value of this`);
393 }
394 > this._isDisposed = true; lifecycle.ts
395 > markAsDisposed(this);
396 > this._fn();
397 > }
398 > } lifecycle.ts
399 >
400 > /**
401 > * Turn a function that implements dispose into an {@link IDisposable}.
402 > *
403 > * @param fn Clean up function, guaranteed to be called only **once**.
404 > */
405 > export function toDisposable(fn: () => void): IDisposable {
406 > return new FunctionDisposable(fn); lifecycle.ts
407 > }
408 > lifecycle.ts
409 > /**
410 > * Manages a collection of disposable values.
411 > *
412 > * This is the preferred way to manage multiple disposables. A `DisposableStore` is safer to work with than an
413 > * `IDisposable[]` as it considers edge cases, such as registering the same value multiple times or adding an item to a
414 > * store that has already been disposed of.
415 > */
416 > export class DisposableStore implements IDisposable {
417 >
418 > static DISABLE_DISPOSED_WARNING = false;
419 >
420 > private readonly _toDispose = new Set<IDisposable>();
421 > private _isDisposed = false;
422 >
423 > constructor() {
424 > trackDisposable(this); lifecycle.ts
425 > }
426 > lifecycle.ts
427 > /**
428 > * Dispose of all registered disposables and mark this object as disposed.
429 > *
430 > * Any future disposables added to this object will be disposed of on `add`.
431 > */
432 > public dispose(): void {
433 > if (this._isDisposed) { lifecycle.ts
434 > return; lifecycle.ts
435 > }
436 > lifecycle.ts
437 > markAsDisposed(this);
438 > this._isDisposed = true;
439 > this.clear();
440 > }
441 > lifecycle.ts
442 > /**
443 > * @return `true` if this object has been disposed of.
444 > */
445 > public get isDisposed(): boolean {
446 return this._isDisposed;
447 }
448 > lifecycle.ts
449 > /**
450 > * Dispose of all registered disposables but do not mark this object as disposed.
451 > */
452 > public clear(): void {
453 > if (this._toDispose.size === 0) { lifecycle.ts
454 > return; lifecycle.ts
455 > }
456 > lifecycle.ts
457 > try {
458 > dispose(this._toDispose);
459 > } finally {
460 > this._toDispose.clear();
461 > }
462 > } lifecycle.ts
463 > lifecycle.ts
464 > /**
465 > * Add a new {@link IDisposable disposable} to the collection.
466 > */
467 > public add<T extends IDisposable>(o: T): T {
468 > if (!o || o === Disposable.None) { lifecycle.ts
469 > return o; lifecycle.ts
470 > }
471 > if ((o as unknown as DisposableStore) === this) { lifecycle.ts
472 throw new Error('Cannot register a disposable on itself!');
473 }
474 > lifecycle.ts
475 > setParentOfDisposable(o, this);
476 > if (this._isDisposed) {
477 if (!DisposableStore.DISABLE_DISPOSED_WARNING) {
478 console.warn(new Error('Trying to add a disposable to a DisposableStore that has already been disposed of. The added object will be leaked!').stack);
479 }
480 > } else { lifecycle.ts
481 > this._toDispose.add(o);
482 > }
483 >
484 > return o;
485 > } lifecycle.ts
486 > lifecycle.ts
487 > /**
488 > * Deletes a disposable from store and disposes of it. This will not throw or warn and proceed to dispose the
489 > * disposable even when the disposable is not part in the store.
490 > */
491 > public delete<T extends IDisposable>(o: T): void {
492 if (!o) {
493 return;
499 o.dispose();
500 }
501 > lifecycle.ts
502 > /**
503 > * Deletes the value from the store, but does not dispose it.
504 > */
505 > public deleteAndLeak<T extends IDisposable>(o: T): void {
506 if (!o) {
507 return;
511 }
512 }
513 > lifecycle.ts
514 > public assertNotDisposed(): void {
515 if (this._isDisposed) {
516 onUnexpectedError(new BugIndicatingError('Object disposed'));
517 }
518 }
519 > } lifecycle.ts
520 >
521 > /**
522 > * Abstract base class for a {@link IDisposable disposable} object.
523 > *
524 > * Subclasses can {@linkcode _register} disposables that will be automatically cleaned up when this object is disposed of.
525 > */
526 > export abstract class Disposable implements IDisposable {
527 >
528 > /**
529 > * A disposable that does nothing when it is disposed of.
530 > *
531 > * TODO: This should not be a static property.
532 > */
533 > static readonly None = Object.freeze<IDisposable>({ dispose() { } });
534 >
535 > protected readonly _store = new DisposableStore();
536 >
537 > constructor() {
538 > trackDisposable(this); lifecycle.ts
539 > setParentOfDisposable(this._store, this);
540 > }
541 > lifecycle.ts
542 > public dispose(): void {
543 > markAsDisposed(this); lifecycle.ts
544 >
545 > this._store.dispose();
546 > }
547 > lifecycle.ts
548 > /**
549 > * Adds `o` to the collection of disposables managed by this object.
550 > */
551 > protected _register<T extends IDisposable>(o: T): T {
552 > if ((o as unknown as Disposable) === this) { lifecycle.ts
553 throw new Error('Cannot register a disposable on itself!');
554 }
555 > return this._store.add(o); lifecycle.ts
556 > }
557 > } lifecycle.ts
558 >
559 > /**
560 > * Manages the lifecycle of a disposable value that may be changed.
561 > *
562 > * This ensures that when the disposable value is changed, the previously held disposable is disposed of. You can
563 > * also register a `MutableDisposable` on a `Disposable` to ensure it is automatically cleaned up.
564 > */
565 > export class MutableDisposable<T extends IDisposable> implements IDisposable {
566 > private _value?: T;
567 > private _isDisposed = false;
568 >
569 > constructor() {
570 > trackDisposable(this); lifecycle.ts
571 > }
572 > lifecycle.ts
573 > /**
574 > * Get the currently held disposable value, or `undefined` if this MutableDisposable has been disposed
575 > */
576 > get value(): T | undefined {
577 > return this._isDisposed ? undefined : this._value; lifecycle.ts
578 > }
579 > lifecycle.ts
580 > /**
581 > * Set a new disposable value.
582 > *
583 > * Behaviour:
584 > * - If the MutableDisposable has been disposed, the setter is a no-op.
585 > * - If the new value is strictly equal to the current value, the setter is a no-op.
586 > * - Otherwise the previous value (if any) is disposed and the new value is stored.
587 > *
588 > * Related helpers:
589 > * - clear() resets the value to `undefined` (and disposes the previous value).
590 > * - clearAndLeak() returns the old value without disposing it and removes its parent.
591 > */
592 > set value(value: T | undefined) {
593 > if (this._isDisposed || value === this._value) { lifecycle.ts
594 > return; lifecycle.ts
595 > }
596
597 > this._value?.dispose(); lifecycle.ts
598 > if (value) {
599 setParentOfDisposable(value, this);
600 }
601 this._value = value;
602 > } lifecycle.ts
603 > lifecycle.ts
604 > /**
605 > * Resets the stored value and disposed of the previously stored value.
606 > */
607 > clear(): void {
608 this.value = undefined;
609 }
610 > lifecycle.ts
611 > dispose(): void {
612 > this._isDisposed = true; lifecycle.ts
613 > markAsDisposed(this);
614 > this._value?.dispose();
615 > this._value = undefined;
616 > }
617 > lifecycle.ts
618 > /**
619 > * Clears the value, but does not dispose it.
620 > * The old value is returned.
621 > */
622 > clearAndLeak(): T | undefined {
623 const oldValue = this._value;
624 this._value = undefined;
628 return oldValue;
629 }
630 > } lifecycle.ts
631 >
632 > /**
633 > * Manages the lifecycle of a disposable value that may be changed like {@link MutableDisposable}, but the value must
634 > * exist and cannot be undefined.
635 > */
636 > export class MandatoryMutableDisposable<T extends IDisposable> implements IDisposable {
637 > private readonly _disposable = new MutableDisposable<T>();
638 > private _isDisposed = false;
639 >
640 > constructor(initialValue: T) {
641 this._disposable.value = initialValue;
642 }
643 > lifecycle.ts
644 > get value(): T {
645 return this._disposable.value!;
646 }
647 > lifecycle.ts
648 > set value(value: T) {
649 if (this._isDisposed || value === this._disposable.value) {
650 return;
652 this._disposable.value = value;
653 }
654 > lifecycle.ts
655 > dispose() {
656 this._isDisposed = true;
657 this._disposable.dispose();
658 }
659 > } lifecycle.ts
660 >
661 > export class RefCountedDisposable {
662 >
663 > private _counter: number = 1;
664 >
665 > constructor(
666 private readonly _disposable: IDisposable,
667 ) { }
668 > lifecycle.ts
669 > acquire() {
670 this._counter++;
671 return this;
672 }
673 > lifecycle.ts
674 > release() {
675 if (--this._counter === 0) {
676 this._disposable.dispose();
678 return this;
679 }
680 > } lifecycle.ts
681 >
682 > export interface IReference<T> extends IDisposable {
683 > readonly object: T;
684 > }
685 >
686 > export abstract class ReferenceCollection<T> {
687 > lifecycle.ts
688 > private readonly references: Map<string, { readonly object: T; counter: number }> = new Map();
689 > lifecycle.ts
690 > acquire(key: string, ...args: unknown[]): IReference<T> {
691 > let reference = this.references.get(key); lifecycle.ts
692 >
693 > if (!reference) {
694 > reference = { counter: 0, object: this.createReferencedObject(key, ...args) };
695 > this.references.set(key, reference);
696 > }
697 >
698 > const { object } = reference;
699 > const dispose = createSingleCallFunction(() => {
700 > if (--reference.counter === 0) { lifecycle.ts
701 > this.destroyReferencedObject(key, reference.object);
702 > this.references.delete(key);
703 > }
704 > }); lifecycle.ts
705 >
706 > reference.counter++;
707 >
708 > return { object, dispose };
709 > }
710 > lifecycle.ts
711 > protected abstract createReferencedObject(key: string, ...args: unknown[]): T;
712 > protected abstract destroyReferencedObject(key: string, object: T): void;
713 > }
714 >
715 > /**
716 > * Unwraps a reference collection of promised values. Makes sure
717 > * references are disposed whenever promises get rejected.
718 > */
719 > export class AsyncReferenceCollection<T> {
720 >
721 > constructor(private referenceCollection: ReferenceCollection<Promise<T>>) { }
722 >
723 > async acquire(key: string, ...args: unknown[]): Promise<IReference<T>> {
724 const ref = this.referenceCollection.acquire(key, ...args);
725
736 }
737 }
738 > } lifecycle.ts
739 >
740 > export class ImmortalReference<T> implements IReference<T> {
741 > constructor(public object: T) { }
742 > dispose(): void { /* noop */ }
743 > }
744 >
745 > export function disposeOnReturn(fn: (store: DisposableStore) => void): void {
746 const store = new DisposableStore();
747 try {
751 }
752 }
753 > lifecycle.ts
754 > /**
755 > * A map the manages the lifecycle of the values that it stores.
756 > */
757 > export class DisposableMap<K, V extends IDisposable = IDisposable> implements IDisposable {
758 >
759 > private readonly _store: Map<K, V>;
760 > private _isDisposed = false;
761 >
762 > constructor(store: Map<K, V> = new Map<K, V>()) {
763 > this._store = store; lifecycle.ts
764 > trackDisposable(this);
765 > }
766 > lifecycle.ts
767 > /**
768 > * Disposes of all stored values and mark this object as disposed.
769 > *
770 > * Trying to use this object after it has been disposed of is an error.
771 > */
772 > dispose(): void {
773 > markAsDisposed(this); lifecycle.ts
774 > this._isDisposed = true;
775 > this.clearAndDisposeAll();
776 > }
777 > lifecycle.ts
778 > /**
779 > * Disposes of all stored values and clear the map, but DO NOT mark this object as disposed.
780 > */
781 > clearAndDisposeAll(): void {
782 > if (!this._store.size) { lifecycle.ts
783 > return; lifecycle.ts
784 > }
785
786 try {
789 this._store.clear();
790 }
791 > } lifecycle.ts
792 > lifecycle.ts
793 > has(key: K): boolean {
794 return this._store.has(key);
795 }
796 > lifecycle.ts
797 > get size(): number {
798 return this._store.size;
799 }
800 > lifecycle.ts
801 > get(key: K): V | undefined {
802 return this._store.get(key);
803 }
804 > lifecycle.ts
805 > set(key: K, value: V, skipDisposeOnOverwrite = false): void {
806 if (this._isDisposed) {
807 console.warn(new Error('Trying to add a disposable to a DisposableMap that has already been disposed of. The added object will be leaked!').stack);
815 setParentOfDisposable(value, this);
816 }
817 > lifecycle.ts
818 > /**
819 > * Delete the value stored for `key` from this map and also dispose of it.
820 > */
821 > deleteAndDispose(key: K): void {
822 this._store.get(key)?.dispose();
823 this._store.delete(key);
824 }
825 > lifecycle.ts
826 > /**
827 > * Delete the value stored for `key` from this map but return it. The caller is
828 > * responsible for disposing of the value.
829 > */
830 > deleteAndLeak(key: K): V | undefined {
831 const value = this._store.get(key);
832 if (value) {
836 return value;
837 }
838 > lifecycle.ts
839 > keys(): IterableIterator<K> {
840 return this._store.keys();
841 }
842 > lifecycle.ts
843 > values(): IterableIterator<V> {
844 return this._store.values();
845 }
846 > lifecycle.ts
847 > [Symbol.iterator](): IterableIterator<[K, V]> {
848 return this._store[Symbol.iterator]();
849 }
850 > } lifecycle.ts
851 >
852 > /**
853 > * A set that manages the lifecycle of the values that it stores.
854 > */
855 > export class DisposableSet<V extends IDisposable = IDisposable> implements IDisposable {
856 >
857 > private readonly _store: Set<V>;
858 > private _isDisposed = false;
859 >
860 > constructor(store: Set<V> = new Set<V>()) {
861 this._store = store;
862 trackDisposable(this);
863 }
864 > lifecycle.ts
865 > /**
866 > * Disposes of all stored values and mark this object as disposed.
867 > *
868 > * Trying to use this object after it has been disposed of is an error.
869 > */
870 > dispose(): void {
871 markAsDisposed(this);
872 this._isDisposed = true;
873 this.clearAndDisposeAll();
874 }
875 > lifecycle.ts
876 > /**
877 > * Disposes of all stored values and clear the set, but DO NOT mark this object as disposed.
878 > */
879 > clearAndDisposeAll(): void {
880 if (!this._store.size) {
881 return;
888 }
889 }
890 > lifecycle.ts
891 > has(value: V): boolean {
892 return this._store.has(value);
893 }
894 > lifecycle.ts
895 > get size(): number {
896 return this._store.size;
897 }
898 > lifecycle.ts
899 > add(value: V): void {
900 if (this._isDisposed) {
901 console.warn(new Error('Trying to add a disposable to a DisposableSet that has already been disposed of. The added object will be leaked!').stack);
905 setParentOfDisposable(value, this);
906 }
907 > lifecycle.ts
908 > /**
909 > * Delete the value from this set and also dispose of it.
910 > */
911 > deleteAndDispose(value: V): void {
912 if (this._store.delete(value)) {
913 value.dispose();
914 }
915 }
916 > lifecycle.ts
917 > /**
918 > * Delete the value from this set but return it. The caller is
919 > * responsible for disposing of the value.
920 > */
921 > deleteAndLeak(value: V): V | undefined {
922 if (this._store.delete(value)) {
923 setParentOfDisposable(value, null);
926 return undefined;
927 }
928 > lifecycle.ts
929 > values(): IterableIterator<V> {
930 return this._store.values();
931 }
932 > lifecycle.ts
933 > [Symbol.iterator](): IterableIterator<V> {
934 return this._store[Symbol.iterator]();
935 }
936 > } lifecycle.ts
937 >
938 > /**
939 > * Call `then` on a Promise, unless the returned disposable is disposed.
940 > */
941 > export function thenIfNotDisposed<T>(promise: Promise<T>, then: (result: T) => void): IDisposable {
942 let disposed = false;
943 promise.then(result => {
951 });
952 }
953 > lifecycle.ts
954 > /**
955 > * Call `then` on a promise that resolves to a {@link IDisposable}, then either register the
956 > * disposable or register it to the {@link DisposableStore}, depending on whether the store is
957 > * disposed or not.
958 > */
959 > export function thenRegisterOrDispose<T extends IDisposable>(promise: Promise<T>, store: DisposableStore): Promise<T> {
960 return promise.then(disposable => {
961 if (store.isDisposed) {
967 });
968 }
969 > lifecycle.ts
970 > export class DisposableResourceMap<V extends IDisposable = IDisposable> extends DisposableMap<URI, V> {
971 > constructor() {
972 > super(new ResourceMap()); lifecycle.ts
973 > }
974 > } lifecycle.ts
src/vs/workbench/services/search/common/searchExtTypes.ts 560 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- searchExtTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../base/common/cancellation.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { IProgress } from '../../../../platform/progress/common/progress.js';
9 >
10 > export class Position {
11 > constructor(readonly line: number, readonly character: number) { }
12 >
13 > isBefore(other: Position): boolean { return false; }
14 > isBeforeOrEqual(other: Position): boolean { return false; }
15 > isAfter(other: Position): boolean { return false; }
16 > isAfterOrEqual(other: Position): boolean { return false; }
17 > isEqual(other: Position): boolean { return false; }
18 > compareTo(other: Position): number { return 0; }
19 > translate(lineDelta?: number, characterDelta?: number): Position;
20 > translate(change: { lineDelta?: number; characterDelta?: number }): Position;
21 > translate(_?: any, _2?: any): Position { return new Position(0, 0); }
22 > with(line?: number, character?: number): Position;
23 > with(change: { line?: number; character?: number }): Position;
24 > with(_: any): Position { return new Position(0, 0); }
25 > }
26 >
27 > export class Range {
28 > readonly start: Position;
29 > readonly end: Position;
30 >
31 > constructor(startLine: number, startCol: number, endLine: number, endCol: number) {
32 this.start = new Position(startLine, startCol);
33 this.end = new Position(endLine, endCol);
34 }
36 > isEmpty = false;
37 > isSingleLine = false;
38 > contains(positionOrRange: Position | Range): boolean { return false; }
39 > isEqual(other: Range): boolean { return false; }
40 > intersection(range: Range): Range | undefined { return undefined; }
41 > union(other: Range): Range { return new Range(0, 0, 0, 0); }
42 >
43 > with(start?: Position, end?: Position): Range;
44 > with(change: { start?: Position; end?: Position }): Range;
45 > with(_: any): Range { return new Range(0, 0, 0, 0); }
46 > }
47 >
48 > export type ProviderResult<T> = T | undefined | null | Thenable<T | undefined | null>;
49 >
50 > /**
51 > * A relative pattern is a helper to construct glob patterns that are matched
52 > * relatively to a base path. The base path can either be an absolute file path
53 > * or a [workspace folder](#WorkspaceFolder).
54 > */
55 > export interface RelativePattern {
56 >
57 > /**
58 > * A base file path to which this pattern will be matched against relatively. The
59 > * file path must be absolute, should not have any trailing path separators and
60 > * not include any relative segments (`.` or `..`).
61 > */
62 > baseUri: URI;
63 >
64 > /**
65 > * A file glob pattern like `*.{ts,js}` that will be matched on file paths
66 > * relative to the base path.
67 > *
68 > * Example: Given a base of `/home/work/folder` and a file path of `/home/work/folder/index.js`,
69 > * the file glob pattern will match on `index.js`.
70 > */
71 > pattern: string;
72 > }
73 >
74 > /**
75 > * A file glob pattern to match file paths against. This can either be a glob pattern string
76 > * (like `** /*.{ts,js}` without space before / or `*.{ts,js}`) or a [relative pattern](#RelativePattern).
77 > *
78 > * Glob patterns can have the following syntax:
79 > * * `*` to match zero or more characters in a path segment
80 > * * `?` to match on one character in a path segment
81 > * * `**` to match any number of path segments, including none
82 > * * `{}` to group conditions (e.g. `** /*.{ts,js}` without space before / matches all TypeScript and JavaScript files)
83 > * * `[]` to declare a range of characters to match in a path segment (e.g., `example.[0-9]` to match on `example.0`, `example.1`, …)
84 > * * `[!...]` to negate a range of characters to match in a path segment (e.g., `example.[!0-9]` to match on `example.a`, `example.b`, but not `example.0`)
85 > *
86 > * Note: a backslash (`\`) is not valid within a glob pattern. If you have an existing file
87 > * path to match against, consider to use the [relative pattern](#RelativePattern) support
88 > * that takes care of converting any backslash into slash. Otherwise, make sure to convert
89 > * any backslash to slash when creating the glob pattern.
90 > */
91 > export type GlobPattern = string | RelativePattern;
92 >
93 > /**
94 > * The parameters of a query for text search.
95 > */
96 > export interface TextSearchQuery2 {
97 > /**
98 > * The text pattern to search for.
99 > */
100 > pattern: string;
101 >
102 > /**
103 > * Whether or not `pattern` should match multiple lines of text.
104 > */
105 > isMultiline?: boolean;
106 >
107 > /**
108 > * Whether or not `pattern` should be interpreted as a regular expression.
109 > */
110 > isRegExp?: boolean;
111 >
112 > /**
113 > * Whether or not the search should be case-sensitive.
114 > */
115 > isCaseSensitive?: boolean;
116 >
117 > /**
118 > * Whether or not to search for whole word matches only.
119 > */
120 > isWordMatch?: boolean;
121 > }
122 >
123 >
124 > export interface TextSearchProviderFolderOptions {
125 > /**
126 > * The root folder to search within.
127 > */
128 > folder: URI;
129 >
130 > /**
131 > * Files that match an `includes` glob pattern should be included in the search.
132 > */
133 > includes: string[];
134 >
135 > /**
136 > * Files that match an `excludes` glob pattern should be excluded from the search.
137 > */
138 > excludes: GlobPattern[];
139 >
140 > /**
141 > * Whether to ignore case for glob patterns.
142 > */
143 > ignoreGlobCase?: boolean;
144 >
145 > /**
146 > * Whether symlinks should be followed while searching.
147 > * For more info, see the setting description for `search.followSymlinks`.
148 > */
149 > followSymlinks: boolean;
150 >
151 > /**
152 > * Which file locations we should look for ignore (.gitignore or .ignore) files to respect.
153 > */
154 > useIgnoreFiles: {
155 > /**
156 > * Use ignore files at the current workspace root.
157 > */
158 > local: boolean;
159 > /**
160 > * Use ignore files at the parent directory. If set, `local` in {@link TextSearchProviderFolderOptions.useIgnoreFiles} should also be `true`.
161 > */
162 > parent: boolean;
163 > /**
164 > * Use global ignore files. If set, `local` in {@link TextSearchProviderFolderOptions.useIgnoreFiles} should also be `true`.
165 > */
166 > global: boolean;
167 > };
168 >
169 > /**
170 > * Interpret files using this encoding.
171 > * See the vscode setting `"files.encoding"`
172 > */
173 > encoding: string;
174 > }
175 >
176 > /**
177 > * Options that apply to text search.
178 > */
179 > export interface TextSearchProviderOptions {
180 >
181 > folderOptions: TextSearchProviderFolderOptions[];
182 >
183 > /**
184 > * The maximum number of results to be returned.
185 > */
186 > maxResults: number;
187 >
188 > /**
189 > * Options to specify the size of the result text preview.
190 > */
191 > previewOptions: {
192 > /**
193 > * The maximum number of lines in the preview.
194 > * Only search providers that support multiline search will ever return more than one line in the match.
195 > * Defaults to 100.
196 > */
197 > matchLines: number;
198 >
199 > /**
200 > * The maximum number of characters included per line.
201 > * Defaults to 10000.
202 > */
203 > charsPerLine: number;
204 > };
205 >
206 > /**
207 > * Exclude files larger than `maxFileSize` in bytes.
208 > */
209 > maxFileSize: number | undefined;
210 >
211 >
212 > /**
213 > * Number of lines of context to include before and after each match.
214 > */
215 > surroundingContext: number;
216 > }
217 >
218 >
219 > /**
220 > * Information collected when text search is complete.
221 > */
222 > export interface TextSearchComplete2 {
223 > /**
224 > * Whether the search hit the limit on the maximum number of search results.
225 > * `maxResults` on [`TextSearchOptions`](#TextSearchOptions) specifies the max number of results.
226 > * - If exactly that number of matches exist, this should be false.
227 > * - If `maxResults` matches are returned and more exist, this should be true.
228 > * - If search hits an internal limit which is less than `maxResults`, this should be true.
229 > */
230 > limitHit?: boolean;
231 > }
232 >
233 > export interface FileSearchProviderFolderOptions {
234 > /**
235 > * The root folder to search within.
236 > */
237 > folder: URI;
238 >
239 > /**
240 > * Files that match an `includes` glob pattern should be included in the search.
241 > */
242 > includes: string[];
243 >
244 > /**
245 > * Files that match an `excludes` glob pattern should be excluded from the search.
246 > */
247 > excludes: GlobPattern[];
248 >
249 > /**
250 > * Whether symlinks should be followed while searching.
251 > * For more info, see the setting description for `search.followSymlinks`.
252 > */
253 > followSymlinks: boolean;
254 >
255 > /**
256 > * Which file locations we should look for ignore (.gitignore or .ignore) files to respect.
257 > */
258 > useIgnoreFiles: {
259 > /**
260 > * Use ignore files at the current workspace root.
261 > */
262 > local: boolean;
263 > /**
264 > * Use ignore files at the parent directory. If set, {@link FileSearchProviderOptions.useIgnoreFiles.local} should also be `true`.
265 > */
266 > parent: boolean;
267 > /**
268 > * Use global ignore files. If set, {@link FileSearchProviderOptions.useIgnoreFiles.local} should also be `true`.
269 > */
270 > global: boolean;
271 > };
272 > }
273 >
274 > /**
275 > * Options that apply to file search.
276 > */
277 > export interface FileSearchProviderOptions {
278 > folderOptions: FileSearchProviderFolderOptions[];
279 >
280 > /**
281 > * An object with a lifespan that matches the session's lifespan. If the provider chooses to, this object can be used as the key for a cache,
282 > * and searches with the same session object can search the same cache. When the token is cancelled, the session is complete and the cache can be cleared.
283 > */
284 > session: unknown;
285 >
286 > /**
287 > * The maximum number of results to be returned.
288 > */
289 > maxResults: number;
290 > }
291 >
292 > /**
293 > * The main match information for a {@link TextSearchResult2}.
294 > */
295 > export class TextSearchMatch2 {
296 > /**
297 > * @param uri The uri for the matching document.
298 > * @param ranges The ranges associated with this match.
299 > * @param previewText The text that is used to preview the match. The highlighted range in `previewText` is specified in `ranges`.
300 > */
301 > constructor(
302 public uri: URI,
303 public ranges: { sourceRange: Range; previewRange: Range }[],
304 public previewText: string) { }
306 > }
307 >
308 > /**
309 > * The potential context information for a {@link TextSearchResult2}.
310 > */
311 > export class TextSearchContext2 {
312 > /**
313 > * @param uri The uri for the matching document.
314 > * @param text The line of context text.
315 > * @param lineNumber The line number of this line of context.
316 > */
317 > constructor(
318 public uri: URI,
319 public text: string,
320 public lineNumber: number) { }
322 >
323 > /**
324 > /**
325 > * Keyword suggestion for AI search.
326 > */
327 > export class AISearchKeyword {
328 > /**
329 > * @param keyword The keyword associated with the search.
330 > */
331 > constructor(public keyword: string) { }
332 > }
333 >
334 > /**
335 > * A result payload for a text search, pertaining to matches within a single file.
336 > */
337 > export type TextSearchResult2 = TextSearchMatch2 | TextSearchContext2;
338 >
339 > /**
340 > * A result payload for an AI search.
341 > * This can be a {@link TextSearchMatch2 match} or a {@link AISearchKeyword keyword}.
342 > * The result can be a match or a keyword.
343 > */
344 > export type AISearchResult = TextSearchResult2 | AISearchKeyword;
345 >
346 > /**
347 > * A FileSearchProvider provides search results for files in the given folder that match a query string. It can be invoked by quickaccess or other extensions.
348 > *
349 > * A FileSearchProvider is the more powerful of two ways to implement file search in VS Code. Use a FileSearchProvider if you wish to search within a folder for
350 > * all files that match the user's query.
351 > *
352 > * The FileSearchProvider will be invoked on every keypress in quickaccess. When `workspace.findFiles` is called, it will be invoked with an empty query string,
353 > * and in that case, every file in the folder should be returned.
354 > */
355 > export interface FileSearchProvider2 {
356 > /**
357 > * Provide the set of files that match a certain file path pattern.
358 > * @param query The parameters for this query.
359 > * @param options A set of options to consider while searching files.
360 > * @param progress A progress callback that must be invoked for all results.
361 > * @param token A cancellation token.
362 > */
363 > provideFileSearchResults(pattern: string, options: FileSearchProviderOptions, token: CancellationToken): ProviderResult<URI[]>;
364 > }
365 >
366 > /**
367 > * A TextSearchProvider provides search results for text results inside files in the workspace.
368 > */
369 > export interface TextSearchProvider2 {
370 > /**
371 > * Provide results that match the given text pattern.
372 > * @param query The parameters for this query.
373 > * @param options A set of options to consider while searching.
374 > * @param progress A progress callback that must be invoked for all results.
375 > * @param token A cancellation token.
376 > */
377 > provideTextSearchResults(query: TextSearchQuery2, options: TextSearchProviderOptions, progress: IProgress<TextSearchResult2>, token: CancellationToken): ProviderResult<TextSearchComplete2>;
378 > }
379 >
380 > /**
381 > * Information collected when text search is complete.
382 > */
383 > export interface TextSearchComplete2 {
384 > /**
385 > * Whether the search hit the limit on the maximum number of search results.
386 > * `maxResults` on {@linkcode TextSearchOptions} specifies the max number of results.
387 > * - If exactly that number of matches exist, this should be false.
388 > * - If `maxResults` matches are returned and more exist, this should be true.
389 > * - If search hits an internal limit which is less than `maxResults`, this should be true.
390 > */
391 > limitHit?: boolean;
392 >
393 > /**
394 > * Additional information regarding the state of the completed search.
395 > *
396 > * Messages with "Information" style support links in markdown syntax:
397 > * - Click to [run a command](command:workbench.action.OpenQuickPick)
398 > * - Click to [open a website](https://aka.ms)
399 > *
400 > * Commands may optionally return { triggerSearch: true } to signal to the editor that the original search should run be again.
401 > */
402 > message?: TextSearchCompleteMessage2[];
403 > }
404 >
405 > /**
406 > * A message regarding a completed search.
407 > */
408 > export interface TextSearchCompleteMessage2 {
409 > /**
410 > * Markdown text of the message.
411 > */
412 > text: string;
413 > /**
414 > * Whether the source of the message is trusted, command links are disabled for untrusted message sources.
415 > * Messaged are untrusted by default.
416 > */
417 > trusted?: boolean;
418 > /**
419 > * The message type, this affects how the message will be rendered.
420 > */
421 > type: TextSearchCompleteMessageType;
422 > }
423 >
424 >
425 > /**
426 > * A FileSearchProvider provides search results for files in the given folder that match a query string. It can be invoked by quickaccess or other extensions.
427 > *
428 > * A FileSearchProvider is the more powerful of two ways to implement file search in VS Code. Use a FileSearchProvider if you wish to search within a folder for
429 > * all files that match the user's query.
430 > *
431 > * The FileSearchProvider will be invoked on every keypress in quickaccess. When `workspace.findFiles` is called, it will be invoked with an empty query string,
432 > * and in that case, every file in the folder should be returned.
433 > */
434 > export interface FileSearchProvider2 {
435 > /**
436 > * Provide the set of files that match a certain file path pattern.
437 > * @param query The parameters for this query.
438 > * @param options A set of options to consider while searching files.
439 > * @param progress A progress callback that must be invoked for all results.
440 > * @param token A cancellation token.
441 > */
442 > provideFileSearchResults(pattern: string, options: FileSearchProviderOptions, token: CancellationToken): ProviderResult<URI[]>;
443 > }
444 >
445 > /**
446 > * A TextSearchProvider provides search results for text results inside files in the workspace.
447 > */
448 > export interface TextSearchProvider2 {
449 > /**
450 > * Provide results that match the given text pattern.
451 > * @param query The parameters for this query.
452 > * @param options A set of options to consider while searching.
453 > * @param progress A progress callback that must be invoked for all results.
454 > * @param token A cancellation token.
455 > */
456 > provideTextSearchResults(query: TextSearchQuery2, options: TextSearchProviderOptions, progress: IProgress<TextSearchResult2>, token: CancellationToken): ProviderResult<TextSearchComplete2>;
457 > }
458 >
459 > /**
460 > * Information collected when text search is complete.
461 > */
462 > export interface TextSearchComplete2 {
463 > /**
464 > * Whether the search hit the limit on the maximum number of search results.
465 > * `maxResults` on {@link TextSearchOptions} specifies the max number of results.
466 > * - If exactly that number of matches exist, this should be false.
467 > * - If `maxResults` matches are returned and more exist, this should be true.
468 > * - If search hits an internal limit which is less than `maxResults`, this should be true.
469 > */
470 > limitHit?: boolean;
471 >
472 > /**
473 > * Additional information regarding the state of the completed search.
474 > *
475 > * Messages with "Information" style support links in markdown syntax:
476 > * - Click to [run a command](command:workbench.action.OpenQuickPick)
477 > * - Click to [open a website](https://aka.ms)
478 > *
479 > * Commands may optionally return { triggerSearch: true } to signal to the editor that the original search should run be again.
480 > */
481 > message?: TextSearchCompleteMessage2[];
482 > }
483 >
484 > /**
485 > * A message regarding a completed search.
486 > */
487 > export interface TextSearchCompleteMessage2 {
488 > /**
489 > * Markdown text of the message.
490 > */
491 > text: string;
492 > /**
493 > * Whether the source of the message is trusted, command links are disabled for untrusted message sources.
494 > * Messaged are untrusted by default.
495 > */
496 > trusted?: boolean;
497 > /**
498 > * The message type, this affects how the message will be rendered.
499 > */
500 > type: TextSearchCompleteMessageType;
501 > }
502 >
503 > /**
504 > * Options for following search.exclude and files.exclude settings.
505 > */
506 > export enum ExcludeSettingOptions {
507 > /*
508 > * Don't use any exclude settings.
509 > */
510 > None = 1,
511 > /*
512 > * Use:
513 > * - files.exclude setting
514 > */
515 > FilesExclude = 2,
516 > /*
517 > * Use:
518 > * - files.exclude setting
519 > * - search.exclude setting
520 > */
521 > SearchAndFilesExclude = 3
522 > }
523 >
524 > export enum TextSearchCompleteMessageType {
525 > Information = 1,
526 > Warning = 2,
527 > }
528 >
529 >
530 > /**
531 > * A message regarding a completed search.
532 > */
533 > export interface TextSearchCompleteMessage {
534 > /**
535 > * Markdown text of the message.
536 > */
537 > text: string;
538 > /**
539 > * Whether the source of the message is trusted, command links are disabled for untrusted message sources.
540 > */
541 > trusted?: boolean;
542 > /**
543 > * The message type, this affects how the message will be rendered.
544 > */
545 > type: TextSearchCompleteMessageType;
546 > }
547 >
548 >
549 > /**
550 > * An AITextSearchProvider provides additional AI text search results in the workspace.
551 > */
552 > export interface AITextSearchProvider {
553 >
554 > /**
555 > * The name of the AI searcher. Will be displayed as `{name} Results` in the Search View.
556 > */
557 > readonly name?: string;
558 >
559 > /**
560 > * WARNING: VERY EXPERIMENTAL.
561 > *
562 > * Provide results that match the given text pattern.
563 > * @param query The parameter for this query.
564 > * @param options A set of options to consider while searching.
565 > * @param progress A progress callback that must be invoked for all results.
566 > * @param token A cancellation token.
567 > */
568 > provideAITextSearchResults(query: string, options: TextSearchProviderOptions, progress: IProgress<TextSearchResult2>, token: CancellationToken): ProviderResult<TextSearchComplete2>;
569 > }
src/vs/workbench/services/search/common/search.ts 559 covered LOC · 34 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- search.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { mapArrayOrNot } from '../../../../base/common/arrays.js';
7 > import { CancellationToken } from '../../../../base/common/cancellation.js';
8 > import * as glob from '../../../../base/common/glob.js';
9 > import { IDisposable } from '../../../../base/common/lifecycle.js';
10 > import * as objects from '../../../../base/common/objects.js';
11 > import * as extpath from '../../../../base/common/extpath.js';
12 > import { fuzzyContains, getNLines } from '../../../../base/common/strings.js';
13 > import { URI, UriComponents } from '../../../../base/common/uri.js';
14 > import { IFilesConfiguration } from '../../../../platform/files/common/files.js';
15 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
16 > import { ITelemetryData } from '../../../../platform/telemetry/common/telemetry.js';
17 > import { Event } from '../../../../base/common/event.js';
18 > import * as paths from '../../../../base/common/path.js';
19 > import { isCancellationError } from '../../../../base/common/errors.js';
20 > import { AISearchKeyword, GlobPattern, TextSearchCompleteMessageType } from './searchExtTypes.js';
21 > import { isThenable } from '../../../../base/common/async.js';
22 > import { ResourceSet } from '../../../../base/common/map.js';
23 >
24 > export { TextSearchCompleteMessageType };
25 >
26 > export const VIEWLET_ID = 'workbench.view.search';
27 > export const PANEL_ID = 'workbench.panel.search';
28 > export const VIEW_ID = 'workbench.view.search';
29 > export const SEARCH_RESULT_LANGUAGE_ID = 'search-result';
30 >
31 > export const SEARCH_EXCLUDE_CONFIG = 'search.exclude';
32 > export const DEFAULT_MAX_SEARCH_RESULTS = 20000;
33 >
34 > // Warning: this pattern is used in the search editor to detect offsets. If you
35 > // change this, also change the search-result built-in extension
36 > const SEARCH_ELIDED_PREFIX = '⟪ ';
37 > const SEARCH_ELIDED_SUFFIX = ' characters skipped ⟫';
38 > const SEARCH_ELIDED_MIN_LEN = (SEARCH_ELIDED_PREFIX.length + SEARCH_ELIDED_SUFFIX.length + 5) * 2;
39 >
40 > export const ISearchService = createDecorator<ISearchService>('searchService');
41 >
42 > /**
43 > * A service that enables to search for files or with in files.
44 > */
45 > export interface ISearchService {
46 > readonly _serviceBrand: undefined;
47 > textSearch(query: ITextQuery, token?: CancellationToken, onProgress?: (result: ISearchProgressItem) => void): Promise<ISearchComplete>;
48 > aiTextSearch(query: IAITextQuery, token?: CancellationToken, onProgress?: (result: ISearchProgressItem) => void): Promise<ISearchComplete>;
49 > getAIName(): Promise<string | undefined>;
50 > textSearchSplitSyncAsync(query: ITextQuery, token?: CancellationToken | undefined, onProgress?: ((result: ISearchProgressItem) => void) | undefined, notebookFilesToIgnore?: ResourceSet, asyncNotebookFilesToIgnore?: Promise<ResourceSet>): { syncResults: ISearchComplete; asyncResults: Promise<ISearchComplete> };
51 > fileSearch(query: IFileQuery, token?: CancellationToken): Promise<ISearchComplete>;
52 > schemeHasFileSearchProvider(scheme: string): boolean;
53 > clearCache(cacheKey: string): Promise<void>;
54 > registerSearchResultProvider(scheme: string, type: SearchProviderType, provider: ISearchResultProvider): IDisposable;
55 > }
56 >
57 > /**
58 > * TODO@roblou - split text from file search entirely, or share code in a more natural way.
59 > */
60 > export const enum SearchProviderType {
61 > file,
62 > text,
63 > aiText
64 > }
65 >
66 > export interface ISearchResultProvider {
67 > getAIName(): Promise<string | undefined>;
68 > textSearch(query: ITextQuery, onProgress?: (p: ISearchProgressItem) => void, token?: CancellationToken): Promise<ISearchComplete>;
69 > fileSearch(query: IFileQuery, token?: CancellationToken): Promise<ISearchComplete>;
70 > clearCache(cacheKey: string): Promise<void>;
71 > }
72 >
73 >
74 > export interface ExcludeGlobPattern<U extends UriComponents = URI> {
75 > folder?: U;
76 > pattern: glob.IExpression;
77 > }
78 >
79 > export interface IFolderQuery<U extends UriComponents = URI> {
80 > folder: U;
81 > folderName?: string;
82 > excludePattern?: ExcludeGlobPattern<U>[];
83 > includePattern?: glob.IExpression;
84 > ignoreGlobCase?: boolean;
85 > fileEncoding?: string;
86 > disregardIgnoreFiles?: boolean;
87 > disregardGlobalIgnoreFiles?: boolean;
88 > disregardParentIgnoreFiles?: boolean;
89 > ignoreSymlinks?: boolean;
90 > }
91 >
92 > export interface ICommonQueryProps<U extends UriComponents> {
93 > /** For telemetry - indicates what is triggering the source */
94 > _reason?: string;
95 >
96 > folderQueries: IFolderQuery<U>[];
97 > // The include pattern for files that gets passed into ripgrep.
98 > // Note that this will override any ignore files if applicable.
99 > includePattern?: glob.IExpression;
100 > excludePattern?: glob.IExpression;
101 > ignoreGlobCase?: boolean;
102 > extraFileResources?: U[];
103 >
104 > onlyOpenEditors?: boolean;
105 >
106 > maxResults?: number;
107 > usingSearchPaths?: boolean;
108 > onlyFileScheme?: boolean;
109 > }
110 >
111 > export interface IFileQueryProps<U extends UriComponents> extends ICommonQueryProps<U> {
112 > type: QueryType.File;
113 > filePattern?: string;
114 >
115 > // when walking through the tree to find the result, don't use the filePattern to fuzzy match.
116 > // Instead, should use glob matching.
117 > shouldGlobMatchFilePattern?: boolean;
118 >
119 > /**
120 > * If true no results will be returned. Instead `limitHit` will indicate if at least one result exists or not.
121 > * Currently does not work with queries including a 'siblings clause'.
122 > */
123 > exists?: boolean;
124 > sortByScore?: boolean;
125 > cacheKey?: string;
126 > }
127 >
128 > export interface ITextQueryProps<U extends UriComponents> extends ICommonQueryProps<U> {
129 > type: QueryType.Text;
130 > contentPattern: IPatternInfo;
131 >
132 > previewOptions?: ITextSearchPreviewOptions;
133 > maxFileSize?: number;
134 > surroundingContext?: number;
135 >
136 > userDisabledExcludesAndIgnoreFiles?: boolean;
137 > }
138 >
139 > export interface IAITextQueryProps<U extends UriComponents> extends ICommonQueryProps<U> {
140 > type: QueryType.aiText;
141 > contentPattern: string;
142 >
143 > previewOptions?: ITextSearchPreviewOptions;
144 > maxFileSize?: number;
145 > surroundingContext?: number;
146 >
147 > userDisabledExcludesAndIgnoreFiles?: boolean;
148 > }
149 >
150 > export type IFileQuery = IFileQueryProps<URI>;
151 > export type IRawFileQuery = IFileQueryProps<UriComponents>;
152 > export type ITextQuery = ITextQueryProps<URI>;
153 > export type IRawTextQuery = ITextQueryProps<UriComponents>;
154 > export type IAITextQuery = IAITextQueryProps<URI>;
155 > export type IRawAITextQuery = IAITextQueryProps<UriComponents>;
156 >
157 > export type IRawQuery = IRawTextQuery | IRawFileQuery | IRawAITextQuery;
158 > export type ISearchQuery = ITextQuery | IFileQuery | IAITextQuery;
159 > export type ITextSearchQuery = ITextQuery | IAITextQuery;
160 >
161 > export const enum QueryType {
162 > File = 1,
163 > Text = 2,
164 > aiText = 3
165 > }
166 >
167 > /* __GDPR__FRAGMENT__
168 > "IPatternInfo" : {
169 > "isRegExp": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
170 > "isWordMatch": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
171 > "wordSeparators": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
172 > "isMultiline": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
173 > "isCaseSensitive": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
174 > "isSmartCase": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true }
175 > }
176 > */
177 > export interface IPatternInfo {
178 > pattern: string;
179 > isRegExp?: boolean;
180 > isWordMatch?: boolean;
181 > wordSeparators?: string;
182 > isMultiline?: boolean;
183 > isUnicode?: boolean;
184 > isCaseSensitive?: boolean;
185 > notebookInfo?: INotebookPatternInfo;
186 > }
187 >
188 > export interface INotebookPatternInfo {
189 > isInNotebookMarkdownInput?: boolean;
190 > isInNotebookMarkdownPreview?: boolean;
191 > isInNotebookCellInput?: boolean;
192 > isInNotebookCellOutput?: boolean;
193 > }
194 >
195 > export interface IFileMatch<U extends UriComponents = URI> {
196 > resource: U;
197 > results?: ITextSearchResult<U>[];
198 > }
199 >
200 > export type IRawFileMatch2 = IFileMatch<UriComponents>;
201 >
202 > export interface ITextSearchPreviewOptions {
203 > matchLines: number;
204 > charsPerLine: number;
205 > }
206 >
207 > export interface ISearchRange {
208 > readonly startLineNumber: number;
209 > readonly startColumn: number;
210 > readonly endLineNumber: number;
211 > readonly endColumn: number;
212 > }
213 >
214 > export interface ITextSearchMatch<U extends UriComponents = URI> {
215 > uri?: U;
216 > rangeLocations: SearchRangeSetPairing[];
217 > previewText: string;
218 > webviewIndex?: number;
219 > cellFragment?: string;
220 > }
221 >
222 > export interface ITextSearchContext<U extends UriComponents = URI> {
223 > uri?: U;
224 > text: string;
225 > lineNumber: number;
226 > }
227 >
228 > export type ITextSearchResult<U extends UriComponents = URI> = ITextSearchMatch<U> | ITextSearchContext<U>;
229 >
230 > export function resultIsMatch(result: ITextSearchResult): result is ITextSearchMatch {
231 return !!(<ITextSearchMatch>result).rangeLocations && !!(<ITextSearchMatch>result).previewText;
232 }
233 > search.ts
234 > export interface IProgressMessage {
235 > message: string;
236 > }
237 >
238 > export type ISearchProgressItem = IFileMatch | IProgressMessage | AISearchKeyword;
239 >
240 > export function isFileMatch(p: ISearchProgressItem): p is IFileMatch {
241 return !!(<IFileMatch>p).resource;
242 }
243 > search.ts
244 > export function isAIKeyword(p: ISearchProgressItem): p is AISearchKeyword {
245 return !!(<AISearchKeyword>p).keyword;
246 }
247 > search.ts
248 > export function isProgressMessage(p: ISearchProgressItem | ISerializedSearchProgressItem): p is IProgressMessage {
249 return !!(p as IProgressMessage).message;
250 }
251 > search.ts
252 > export interface ITextSearchCompleteMessage {
253 > text: string;
254 > type: TextSearchCompleteMessageType;
255 > trusted?: boolean;
256 > }
257 >
258 > export interface ISearchCompleteStats {
259 > limitHit?: boolean;
260 > messages: ITextSearchCompleteMessage[];
261 > stats?: IFileSearchStats | ITextSearchStats;
262 > }
263 >
264 > export interface ISearchComplete extends ISearchCompleteStats {
265 > results: IFileMatch[];
266 > exit?: SearchCompletionExitCode;
267 > aiKeywords?: AISearchKeyword[];
268 > }
269 >
270 > export const enum SearchCompletionExitCode {
271 > Normal,
272 > NewSearchStarted
273 > }
274 >
275 > export interface ITextSearchStats {
276 > type: 'textSearchProvider' | 'searchProcess' | 'aiTextSearchProvider';
277 > }
278 >
279 > export interface IFileSearchStats {
280 > fromCache: boolean;
281 > detailStats: ISearchEngineStats | ICachedSearchStats | IFileSearchProviderStats;
282 >
283 > resultCount: number;
284 > type: 'fileSearchProvider' | 'searchProcess';
285 > sortingTime?: number;
286 > }
287 >
288 > export interface ICachedSearchStats {
289 > cacheWasResolved: boolean;
290 > cacheLookupTime: number;
291 > cacheFilterTime: number;
292 > cacheEntryCount: number;
293 > }
294 >
295 > export interface ISearchEngineStats {
296 > fileWalkTime: number;
297 > directoriesWalked: number;
298 > filesWalked: number;
299 > cmdTime: number;
300 > cmdResultCount?: number;
301 > }
302 >
303 > export interface IFileSearchProviderStats {
304 > providerTime: number;
305 > postProcessTime: number;
306 > }
307 >
308 > export class FileMatch implements IFileMatch {
309 > results: ITextSearchResult[] = [];
310 > constructor(public resource: URI) {
311 // empty
312 }
313 > } search.ts
314 >
315 > export interface SearchRangeSetPairing {
316 > source: ISearchRange;
317 > preview: ISearchRange;
318 > }
319 >
320 > export class TextSearchMatch implements ITextSearchMatch {
321 > rangeLocations: SearchRangeSetPairing[] = [];
322 > previewText: string;
323 > webviewIndex?: number;
324 >
325 > constructor(text: string, ranges: ISearchRange | ISearchRange[], previewOptions?: ITextSearchPreviewOptions, webviewIndex?: number) {
326 this.webviewIndex = webviewIndex;
327
371 }
372 }
373 > } search.ts
374 >
375 function isSingleLineRangeList(ranges: ISearchRange[]): boolean {
376 const line = ranges[0].startLineNumber;
383 return true;
384 }
385 > search.ts
386 > export class SearchRange implements ISearchRange {
387 > startLineNumber: number;
388 > startColumn: number;
389 > endLineNumber: number;
390 > endColumn: number;
391 >
392 > constructor(startLineNumber: number, startColumn: number, endLineNumber: number, endColumn: number) {
393 this.startLineNumber = startLineNumber;
394 this.startColumn = startColumn;
396 this.endColumn = endColumn;
397 }
398 > } search.ts
399 >
400 > export class OneLineRange extends SearchRange {
401 > constructor(lineNumber: number, startColumn: number, endColumn: number) {
402 super(lineNumber, startColumn, lineNumber, endColumn);
403 }
404 > } search.ts
405 >
406 > export const enum ViewMode {
407 > List = 'list',
408 > Tree = 'tree'
409 > }
410 >
411 > export const enum SearchSortOrder {
412 > Default = 'default',
413 > FileNames = 'fileNames',
414 > Type = 'type',
415 > Modified = 'modified',
416 > CountDescending = 'countDescending',
417 > CountAscending = 'countAscending'
418 > }
419 >
420 > export const enum SemanticSearchBehavior {
421 > Auto = 'auto',
422 > Manual = 'manual',
423 > RunOnEmpty = 'runOnEmpty',
424 > }
425 >
426 > export interface ISearchConfigurationProperties {
427 > exclude: glob.IExpression;
428 > /**
429 > * Use ignore file for file search.
430 > */
431 > useIgnoreFiles: boolean;
432 > useGlobalIgnoreFiles: boolean;
433 > useParentIgnoreFiles: boolean;
434 > followSymlinks: boolean;
435 > smartCase: boolean;
436 > globalFindClipboard: boolean;
437 > useReplacePreview: boolean;
438 > showLineNumbers: boolean;
439 > actionsPosition: 'auto' | 'right';
440 > maxResults: number | null;
441 > collapseResults: 'auto' | 'alwaysCollapse' | 'alwaysExpand';
442 > searchOnType: boolean;
443 > seedOnFocus: boolean;
444 > seedWithNearestWord: boolean;
445 > searchOnTypeDebouncePeriod: number;
446 > mode: 'view' | 'reuseEditor' | 'newEditor';
447 > searchEditor: {
448 > doubleClickBehaviour: 'selectWord' | 'goToLocation' | 'openLocationToSide';
449 > singleClickBehaviour: 'default' | 'peekDefinition';
450 > reusePriorSearchConfiguration: boolean;
451 > defaultNumberOfContextLines: number | null;
452 > focusResultsOnSearch: boolean;
453 > experimental: {};
454 > };
455 > sortOrder: SearchSortOrder;
456 > decorations: {
457 > colors: boolean;
458 > badges: boolean;
459 > };
460 > quickAccess: {
461 > preserveInput: boolean;
462 > };
463 > defaultViewMode: ViewMode;
464 > experimental: {
465 > closedNotebookRichContentResults: boolean;
466 > };
467 > searchView: {
468 > semanticSearchBehavior: string;
469 > keywordSuggestions: boolean;
470 > };
471 > }
472 >
473 > export interface ISearchConfiguration extends IFilesConfiguration {
474 > search?: ISearchConfigurationProperties;
475 > editor: {
476 > wordSeparators: string;
477 > };
478 > }
479 >
480 > export function getExcludes(configuration: ISearchConfiguration, includeSearchExcludes = true): glob.IExpression | undefined {
481 const fileExcludes = configuration && configuration.files && configuration.files.exclude;
482 const searchExcludes = includeSearchExcludes && configuration && configuration.search && configuration.search.exclude;
497 return allExcludes;
498 }
499 > search.ts
500 > export function pathIncludedInQuery(queryProps: ICommonQueryProps<URI>, fsPath: string): boolean {
501 const globOptions = queryProps.ignoreGlobCase ? { ignoreCase: true } : undefined;
502 if (queryProps.excludePattern && glob.match(queryProps.excludePattern, fsPath, globOptions)) {
527 return true;
528 }
529 > search.ts
530 > export enum SearchErrorCode {
531 > unknownEncoding = 1,
532 > regexParseError,
533 > globParseError,
534 > invalidLiteral,
535 > rgProcessError,
536 > other,
537 > canceled
538 > }
539 >
540 > export class SearchError extends Error {
541 > constructor(message: string, readonly code?: SearchErrorCode) {
542 super(message);
543 }
544 > } search.ts
545 >
546 > export function deserializeSearchError(error: Error): SearchError {
547 const errorMsg = error.message;
548
558 }
559 }
560 > search.ts
561 > export function serializeSearchError(searchError: SearchError): Error {
562 const details = { message: searchError.message, code: searchError.code };
563 return new Error(JSON.stringify(details));
564 }
565 > export interface ITelemetryEvent { search.ts
566 > eventName: string;
567 > data: ITelemetryData;
568 > }
569 >
570 > export interface IRawSearchService {
571 > fileSearch(search: IRawFileQuery): Event<ISerializedSearchProgressItem | ISerializedSearchComplete>;
572 > textSearch(search: IRawTextQuery): Event<ISerializedSearchProgressItem | ISerializedSearchComplete>;
573 > clearCache(cacheKey: string): Promise<void>;
574 > }
575 >
576 > export interface IRawFileMatch {
577 > base?: string;
578 > /**
579 > * The path of the file relative to the containing `base` folder.
580 > * This path is exactly as it appears on the filesystem.
581 > */
582 > relativePath: string;
583 > /**
584 > * This path is transformed for search purposes. For example, this could be
585 > * the `relativePath` with the workspace folder name prepended. This way the
586 > * search algorithm would also match against the name of the containing folder.
587 > *
588 > * If not given, the search algorithm should use `relativePath`.
589 > */
590 > searchPath: string | undefined;
591 > }
592 >
593 > export interface ISearchEngine<T> {
594 > search: (onResult: (matches: T) => void, onProgress: (progress: IProgressMessage) => void, done: (error: Error | null, complete: ISearchEngineSuccess) => void) => void;
595 > cancel: () => void;
596 > }
597 >
598 > export interface ISerializedSearchSuccess {
599 > type: 'success';
600 > limitHit: boolean;
601 > messages: ITextSearchCompleteMessage[];
602 > stats?: IFileSearchStats | ITextSearchStats;
603 > }
604 >
605 > export interface ISearchEngineSuccess {
606 > limitHit: boolean;
607 > messages: ITextSearchCompleteMessage[];
608 > stats: ISearchEngineStats;
609 > }
610 >
611 > export interface ISerializedSearchError {
612 > type: 'error';
613 > error: {
614 > message: string;
615 > stack: string;
616 > };
617 > }
618 >
619 > export type ISerializedSearchComplete = ISerializedSearchSuccess | ISerializedSearchError;
620 >
621 > export function isSerializedSearchComplete(arg: ISerializedSearchProgressItem | ISerializedSearchComplete): arg is ISerializedSearchComplete {
622 // eslint-disable-next-line local/code-no-any-casts
623 if ((arg as any).type === 'error') {
630 }
631 }
632 > search.ts
633 > export function isSerializedSearchSuccess(arg: ISerializedSearchComplete): arg is ISerializedSearchSuccess {
634 return arg.type === 'success';
635 }
636 > search.ts
637 > export function isSerializedFileMatch(arg: ISerializedSearchProgressItem): arg is ISerializedFileMatch {
638 return !!(<ISerializedFileMatch>arg).path;
639 }
640 > search.ts
641 > const filePatternIgnoreCaseOptions = { ignoreCase: true };
642 >
643 > export function isFilePatternMatch(candidate: IRawFileMatch, filePatternToUse: string, fuzzy = true, ignoreCase?: boolean): boolean {
644 const pathToMatch = candidate.searchPath ? candidate.searchPath : candidate.relativePath;
645 return fuzzy ?
647 glob.match(filePatternToUse, pathToMatch, ignoreCase ? filePatternIgnoreCaseOptions : undefined);
648 }
649 > search.ts
650 > export interface ISerializedFileMatch {
651 > path: string;
652 > results?: ITextSearchResult[];
653 > numMatches?: number;
654 > }
655 >
656 > // Type of the possible values for progress calls from the engine
657 > export type ISerializedSearchProgressItem = ISerializedFileMatch | ISerializedFileMatch[] | IProgressMessage;
658 > export type IFileSearchProgressItem = IRawFileMatch | IRawFileMatch[] | IProgressMessage;
659 >
660 >
661 > export class SerializableFileMatch implements ISerializedFileMatch {
662 > path: string;
663 > results: ITextSearchMatch[];
664 >
665 > constructor(path: string) {
666 this.path = path;
667 this.results = [];
668 }
669 > search.ts
670 > addMatch(match: ITextSearchMatch): void {
671 this.results.push(match);
672 }
673 > search.ts
674 > serialize(): ISerializedFileMatch {
675 return {
676 path: this.path,
679 };
680 }
681 > } search.ts
682 >
683 > /**
684 > * Computes the patterns that the provider handles. Discards sibling clauses and 'false' patterns
685 > */
686 > export function resolvePatternsForProvider(globalPattern: glob.IExpression | undefined, folderPattern: glob.IExpression | undefined): string[] {
687 const merged = {
688 ...(globalPattern || {}),
696 });
697 }
698 > search.ts
699 > export class QueryGlobTester {
700 >
701 > private _excludeExpression: glob.IExpression[]; // TODO: evaluate globs based on baseURI of pattern
702 > private _parsedExcludeExpression: glob.ParsedExpression[];
703 >
704 > private _parsedIncludeExpression: glob.ParsedExpression | null = null;
705 >
706 > constructor(config: ISearchQuery, folderQuery: IFolderQuery) {
707 const globOptions = config.ignoreGlobCase || folderQuery.ignoreGlobCase ? { ignoreCase: true } : undefined;
708
739 }
740 }
741 > search.ts
742 > private _evalParsedExcludeExpression(testPath: string, basename: string | undefined, hasSibling?: (name: string) => boolean): string | null {
743 // todo: less hacky way of evaluating sync vs async sibling clauses
744 let result: string | null = null;
756 return result;
757 }
758 > search.ts
759 >
760 > matchesExcludesSync(testPath: string, basename?: string, hasSibling?: (name: string) => boolean): boolean {
761 if (this._parsedExcludeExpression && this._evalParsedExcludeExpression(testPath, basename, hasSibling)) {
762 return true;
765 return false;
766 }
767 > search.ts
768 > /**
769 > * Guaranteed sync - siblingsFn should not return a promise.
770 > */
771 > includedInQuerySync(testPath: string, basename?: string, hasSibling?: (name: string) => boolean): boolean {
772 if (this._parsedExcludeExpression && this._evalParsedExcludeExpression(testPath, basename, hasSibling)) {
773 return false;
780 return true;
781 }
782 > search.ts
783 > /**
784 > * Evaluating the exclude expression is only async if it includes sibling clauses. As an optimization, avoid doing anything with Promises
785 > * unless the expression is async.
786 > */
787 > includedInQuery(testPath: string, basename?: string, hasSibling?: (name: string) => boolean | Promise<boolean>): Promise<boolean> | boolean {
788
789 const isIncluded = () => {
811
812 }
813 > search.ts
814 > hasSiblingExcludeClauses(): boolean {
815 return this._excludeExpression.reduce((prev, curr) => hasSiblingClauses(curr) || prev, false);
816 }
817 > } search.ts
818 >
819 function hasSiblingClauses(pattern: glob.IExpression): boolean {
820 for (const key in pattern) {
826 return false;
827 }
828 > search.ts
829 > export function hasSiblingPromiseFn(siblingsFn?: () => Promise<string[]>) {
830 if (!siblingsFn) {
831 return undefined;
841 };
842 }
843 > search.ts
844 > export function hasSiblingFn(siblingsFn?: () => string[]) {
845 if (!siblingsFn) {
846 return undefined;
856 };
857 }
858 > search.ts
859 function listToMap(list: string[]) {
860 const map: Record<string, true> = {};
864 return map;
865 }
866 > search.ts
867 > export function excludeToGlobPattern(excludesForFolder: { baseUri?: URI | undefined; patterns: string[] }[]): GlobPattern[] {
868 return excludesForFolder.flatMap(exclude => exclude.patterns.map(pattern => {
869 return exclude.baseUri ?
874 }));
875 }
876 > search.ts
877 > export const DEFAULT_TEXT_SEARCH_PREVIEW_OPTIONS = {
878 > matchLines: 100,
879 > charsPerLine: 10000
880 > };
src/vs/workbench/contrib/chat/common/participants/chatAgents.ts 548 covered LOC · 74 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatAgents.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { findLast } from '../../../../../base/common/arraysFind.js';
7 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
8 > import { IStringDictionary } from '../../../../../base/common/collections.js';
9 > import { Emitter, Event } from '../../../../../base/common/event.js';
10 > import { IMarkdownString } from '../../../../../base/common/htmlContent.js';
11 > import { Iterable } from '../../../../../base/common/iterator.js';
12 > import { Disposable, IDisposable, toDisposable } from '../../../../../base/common/lifecycle.js';
13 > import { revive, Revived } from '../../../../../base/common/marshalling.js';
14 > import { IObservable } from '../../../../../base/common/observable.js';
15 > import { equalsIgnoreCase } from '../../../../../base/common/strings.js';
16 > import { ThemeIcon } from '../../../../../base/common/themables.js';
17 > import { URI } from '../../../../../base/common/uri.js';
18 > import { Command } from '../../../../../editor/common/languages.js';
19 > import { IConfigurationService } from '../../../../../platform/configuration/common/configuration.js';
20 > import { ContextKeyExpr, IContextKey, IContextKeyService } from '../../../../../platform/contextkey/common/contextkey.js';
21 > import { ExtensionIdentifier } from '../../../../../platform/extensions/common/extensions.js';
22 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
23 > import { ChatContextKeys } from '../actions/chatContextKeys.js';
24 > import { IChatAgentEditedFileEvent, IChatProgressHistoryResponseContent, IChatRequestModeInstructions, IChatRequestVariableData, ISerializableChatAgentData } from '../model/chatModel.js';
25 > import { ChatRequestHooks } from '../promptSyntax/hookSchema.js';
26 > import { IRawChatCommandContribution } from './chatParticipantContribTypes.js';
27 > import { IChatFollowup, IChatLocationData, IChatProgress, IChatResponseErrorDetails, IChatTaskDto } from '../chatService/chatService.js';
28 > import { ChatAgentLocation, ChatConfiguration, ChatModeKind, ChatPermissionLevel } from '../constants.js';
29 > import { ILanguageModelsService } from '../languageModels.js';
30 > import { ChatPerfMark, markChat } from '../chatPerf.js';
31 >
32 > //#region agent service, commands etc
33 >
34 > export interface IChatAgentHistoryEntry {
35 > request: IChatAgentRequest;
36 > response: ReadonlyArray<IChatProgressHistoryResponseContent | IChatTaskDto>;
37 > result: IChatAgentResult;
38 > }
39 >
40 > export interface IChatAgentAttachmentCapabilities {
41 > supportsFileAttachments?: boolean;
42 > supportsToolAttachments?: boolean;
43 > supportsMCPAttachments?: boolean;
44 > supportsImageAttachments?: boolean;
45 > supportsSearchResultAttachments?: boolean;
46 > supportsInstructionAttachments?: boolean;
47 > supportsSourceControlAttachments?: boolean;
48 > supportsProblemAttachments?: boolean;
49 > supportsSymbolAttachments?: boolean;
50 > supportsTerminalAttachments?: boolean;
51 > supportsPromptAttachments?: boolean;
52 > supportsHandOffs?: boolean;
53 > supportsCheckpoints?: boolean;
54 > /**
55 > * The prefix (e.g. `!`) that marks a message in this
56 > * session type as a terminal command rather than a message to the agent.
57 > * Undefined when the session type has no terminal command support.
58 > */
59 > terminalCommandPrefix?: string;
60 > }
61 >
62 > export interface IChatAgentData {
63 > id: string;
64 > name: string;
65 > fullName?: string;
66 > description?: string;
67 > /** This is string, not ContextKeyExpression, because dealing with serializing/deserializing is hard and need a better pattern for this */
68 > when?: string;
69 > extensionId: ExtensionIdentifier;
70 > extensionVersion: string | undefined;
71 > extensionPublisherId: string;
72 > /** This is the extension publisher id, or, in the case of a dynamically registered participant (remote agent), whatever publisher name we have for it */
73 > publisherDisplayName?: string;
74 > extensionDisplayName: string;
75 > /** The agent invoked when no agent is specified */
76 > isDefault?: boolean;
77 > /** This agent is not contributed in package.json, but is registered dynamically */
78 > isDynamic?: boolean;
79 > /** This agent is contributed from core and not from an extension */
80 > isCore?: boolean;
81 > canAccessPreviousChatHistory?: boolean;
82 > metadata: IChatAgentMetadata;
83 > slashCommands: IChatAgentCommand[];
84 > locations: ChatAgentLocation[];
85 > /** This is only relevant for isDefault agents. Others should have all modes available. */
86 > modes: ChatModeKind[];
87 > disambiguation: { category: string; description: string; examples: string[] }[];
88 > capabilities?: IChatAgentAttachmentCapabilities;
89 > }
90 >
91 > export interface IChatWelcomeMessageContent {
92 > icon: ThemeIcon;
93 > title: string;
94 > message: IMarkdownString;
95 > }
96 >
97 > export interface IChatAgentImplementation {
98 > invoke(request: IChatAgentRequest, progress: (parts: IChatProgress[]) => void, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatAgentResult>;
99 > setRequestTools?(requestId: string, tools: UserSelectedTools): void;
100 > setYieldRequested?(requestId: string, value: boolean): void;
101 > provideFollowups?(request: IChatAgentRequest, result: IChatAgentResult, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatFollowup[]>;
102 > provideChatTitle?: (history: IChatAgentHistoryEntry[], token: CancellationToken) => Promise<string | undefined>;
103 > provideChatSummary?: (history: IChatAgentHistoryEntry[], token: CancellationToken) => Promise<string | undefined>;
104 > }
105 >
106 > export interface IChatParticipantDetectionResult {
107 > participant: string;
108 > command?: string;
109 > }
110 >
111 > export interface IChatParticipantMetadata {
112 > participant: string;
113 > command?: string;
114 > disambiguation: { category: string; description: string; examples: string[] }[];
115 > }
116 >
117 > export interface IChatParticipantDetectionProvider {
118 > provideParticipantDetection(request: IChatAgentRequest, history: IChatAgentHistoryEntry[], options: { location: ChatAgentLocation; participants: IChatParticipantMetadata[] }, token: CancellationToken): Promise<IChatParticipantDetectionResult | null | undefined>;
119 > }
120 >
121 > export type IChatAgent = IChatAgentData & IChatAgentImplementation;
122 >
123 > export interface IChatAgentCommand extends IRawChatCommandContribution {
124 > followupPlaceholder?: string;
125 > }
126 >
127 > export interface IChatAgentMetadata {
128 > helpTextPrefix?: string | IMarkdownString;
129 > helpTextPostfix?: string | IMarkdownString;
130 > icon?: URI;
131 > iconDark?: URI;
132 > themeIcon?: ThemeIcon;
133 > sampleRequest?: string;
134 > supportIssueReporting?: boolean;
135 > followupPlaceholder?: string;
136 > isSticky?: boolean;
137 > additionalWelcomeMessage?: string | IMarkdownString;
138 > }
139 >
140 > export type UserSelectedTools = Record<string, boolean>;
141 >
142 >
143 > export interface IChatAgentRequest {
144 > sessionResource: URI;
145 > requestId: string;
146 > agentId: string;
147 > command?: string;
148 > message: string;
149 > attempt?: number;
150 > enableCommandDetection?: boolean;
151 > isParticipantDetected?: boolean;
152 > variables: IChatRequestVariableData;
153 > location: ChatAgentLocation;
154 > locationData?: Revived<IChatLocationData>;
155 > acceptedConfirmationData?: unknown[];
156 > rejectedConfirmationData?: unknown[];
157 > agentHostSessionConfig?: Record<string, unknown>;
158 > userSelectedModelId?: string;
159 > modelConfiguration?: IStringDictionary<unknown>;
160 > userSelectedTools?: UserSelectedTools;
161 > modeInstructions?: IChatRequestModeInstructions;
162 > editedFileEvents?: IChatAgentEditedFileEvent[];
163 > /**
164 > * The working directory URI for the session, if set.
165 > * In the agents window, each session can have its own working directory
166 > * that differs from the current workspace folders.
167 > */
168 > workingDirectory?: URI;
169 > /**
170 > * Collected hooks configuration for this request.
171 > * Contains all hooks defined in hooks .json files, organized by hook type.
172 > */
173 > hooks?: ChatRequestHooks;
174 > /**
175 > * Whether any hooks are enabled for this request.
176 > */
177 > hasHooksEnabled?: boolean;
178 > /**
179 > * The permission level for tool auto-approval in this request.
180 > * - `'autoApprove'`: Auto-approve all tool calls and retry on errors.
181 > * - `'autopilot'`: Everything autoApprove does plus continues until the task is done.
182 > */
183 > permissionLevel?: ChatPermissionLevel;
184 > /**
185 > * Unique ID for the subagent invocation, used to group tool calls from the same subagent run together.
186 > */
187 > subAgentInvocationId?: string;
188 > /**
189 > * Display name of the subagent that is invoking this request.
190 > */
191 > subAgentName?: string;
192 > /**
193 > * The request ID of the parent request that invoked this subagent.
194 > */
195 > parentRequestId?: string;
196 >
197 > /**
198 > * When true, this request was initiated by the system rather than the user.
199 > */
200 > isSystemInitiated?: boolean;
201 > }
202 >
203 > export interface IChatQuestion {
204 > readonly prompt: string;
205 > readonly participant?: string;
206 > readonly command?: string;
207 > }
208 >
209 > export interface IChatAgentResultTimings {
210 > firstProgress?: number;
211 > totalElapsed: number;
212 > }
213 >
214 > export interface IChatAgentResult {
215 > errorDetails?: IChatResponseErrorDetails;
216 > timings?: IChatAgentResultTimings;
217 > /** Extra properties that the agent can use to identify a result */
218 > readonly metadata?: { readonly [key: string]: unknown };
219 > readonly details?: string;
220 > nextQuestion?: IChatQuestion;
221 > }
222 >
223 > export const IChatAgentService = createDecorator<IChatAgentService>('chatAgentService');
224 >
225 > interface IChatAgentEntry {
226 > data: IChatAgentData;
227 > impl?: IChatAgentImplementation;
228 > }
229 >
230 > export interface IChatAgentCompletionItem {
231 > id: string;
232 > name?: string;
233 > fullName?: string;
234 > icon?: ThemeIcon;
235 > value: unknown;
236 > command?: Command;
237 > }
238 >
239 > export interface IChatAgentInvocationEvent {
240 > readonly agentId: string;
241 > readonly request: Readonly<IChatAgentRequest>;
242 > }
243 >
244 > export interface IChatAgentService {
245 > _serviceBrand: undefined;
246 > /**
247 > * undefined when an agent was removed
248 > */
249 > readonly onDidChangeAgents: Event<IChatAgent | undefined>;
250 > readonly onWillInvokeAgent: Event<IChatAgentInvocationEvent>;
251 > readonly hasToolsAgent: boolean;
252 > registerAgent(id: string, data: IChatAgentData): IDisposable;
253 > registerAgentImplementation(id: string, agent: IChatAgentImplementation): IDisposable;
254 > registerDynamicAgent(data: IChatAgentData, agentImpl: IChatAgentImplementation): IDisposable;
255 > registerAgentCompletionProvider(id: string, provider: (query: string, token: CancellationToken) => Promise<IChatAgentCompletionItem[]>): IDisposable;
256 > getAgentCompletionItems(id: string, query: string, token: CancellationToken): Promise<IChatAgentCompletionItem[]>;
257 > registerChatParticipantDetectionProvider(handle: number, provider: IChatParticipantDetectionProvider): IDisposable;
258 > detectAgentOrCommand(request: IChatAgentRequest, history: IChatAgentHistoryEntry[], options: { location: ChatAgentLocation }, token: CancellationToken): Promise<{ agent: IChatAgentData; command?: IChatAgentCommand } | undefined>;
259 > hasChatParticipantDetectionProviders(): boolean;
260 > invokeAgent(agent: string, request: IChatAgentRequest, progress: (parts: IChatProgress[]) => void, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatAgentResult>;
261 > setRequestTools(agent: string, requestId: string, tools: UserSelectedTools): void;
262 > setYieldRequested(agent: string, requestId: string, value: boolean): void;
263 > getFollowups(id: string, request: IChatAgentRequest, result: IChatAgentResult, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatFollowup[]>;
264 > getChatTitle(id: string, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<string | undefined>;
265 > getChatSummary(id: string, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<string | undefined>;
266 > getAgent(id: string, includeDisabled?: boolean): IChatAgentData | undefined;
267 > getAgentByFullyQualifiedId(id: string): IChatAgentData | undefined;
268 > getAgents(): IChatAgentData[];
269 > getActivatedAgents(): Array<IChatAgent>;
270 > getAgentsByName(name: string): IChatAgentData[];
271 > agentHasDupeName(id: string): boolean;
272 >
273 > /**
274 > * Get the default agent (only if activated)
275 > */
276 > getDefaultAgent(location: ChatAgentLocation, mode?: ChatModeKind): IChatAgent | undefined;
277 >
278 > /**
279 > * Get the default agent data that has been contributed (may not be activated yet)
280 > */
281 > getContributedDefaultAgent(location: ChatAgentLocation): IChatAgentData | undefined;
282 > updateAgent(id: string, updateMetadata: IChatAgentMetadata): void;
283 > }
284 >
285 > export class ChatAgentService extends Disposable implements IChatAgentService {
286 >
287 > public static readonly AGENT_LEADER = '@';
288 >
289 > declare _serviceBrand: undefined;
290 >
291 > private _agents = new Map<string, IChatAgentEntry>();
292 >
293 > private readonly _onDidChangeAgents = this._register(new Emitter<IChatAgent | undefined>());
294 > readonly onDidChangeAgents: Event<IChatAgent | undefined> = this._onDidChangeAgents.event;
295 > private readonly _onWillInvokeAgent = this._register(new Emitter<IChatAgentInvocationEvent>());
296 > readonly onWillInvokeAgent: Event<IChatAgentInvocationEvent> = this._onWillInvokeAgent.event;
297 >
298 > private readonly _agentsContextKeys = new Set<string>();
299 > private readonly _hasDefaultAgent: IContextKey<boolean>;
300 > private readonly _extensionAgentRegistered: IContextKey<boolean>;
301 > private readonly _defaultAgentRegistered: IContextKey<boolean>;
302 > private _hasToolsAgent = false;
303 >
304 > private _chatParticipantDetectionProviders = new Map<number, IChatParticipantDetectionProvider>();
305 >
306 > constructor(
307 > @IContextKeyService private readonly contextKeyService: IContextKeyService, chatAgents.ts
308 > @IConfigurationService private readonly configurationService: IConfigurationService,
309 > ) {
310 > super();
311 > this._hasDefaultAgent = ChatContextKeys.enabled.bindTo(this.contextKeyService);
312 > this._extensionAgentRegistered = ChatContextKeys.extensionParticipantRegistered.bindTo(this.contextKeyService);
313 > this._defaultAgentRegistered = ChatContextKeys.panelParticipantRegistered.bindTo(this.contextKeyService);
314 > this._register(contextKeyService.onDidChangeContext((e) => {
315 if (e.affectsSome(this._agentsContextKeys)) {
316 this._updateContextKeys();
317 }
318 > })); chatAgents.ts
319 > }
321 > registerAgent(id: string, data: IChatAgentData): IDisposable {
322 > const existingAgent = this.getAgent(id); chatAgents.ts
323 > if (existingAgent) {
324 throw new Error(`Agent already registered: ${JSON.stringify(id)}`);
325 }
327 > const that = this;
328 > const commands = data.slashCommands;
329 > data = {
330 > ...data,
331 > get slashCommands() {
332 return commands.filter(c => !c.when || that.contextKeyService.contextMatchesRules(ContextKeyExpr.deserialize(c.when)));
333 }
334 > }; chatAgents.ts
335 > const entry = { data };
336 > this._agents.set(id, entry);
337 > this._updateAgentsContextKeys();
338 > this._updateContextKeys();
339 > this._onDidChangeAgents.fire(undefined);
340 >
341 > return toDisposable(() => {
342 > this._agents.delete(id);
343 > this._updateAgentsContextKeys();
344 > this._updateContextKeys();
345 > this._onDidChangeAgents.fire(undefined);
346 > });
347 > }
349 > private _updateAgentsContextKeys(): void {
350 > // Update the set of context keys used by all agents chatAgents.ts
351 > this._agentsContextKeys.clear();
352 > for (const agent of this._agents.values()) {
353 > if (agent.data.when) {
354 const expr = ContextKeyExpr.deserialize(agent.data.when);
355 for (const key of expr?.keys() || []) {
357 }
358 }
359 > } chatAgents.ts
360 > }
362 > private _updateContextKeys(): void {
363 > let extensionAgentRegistered = false; chatAgents.ts
364 > let defaultAgentRegistered = false;
365 > let toolsAgentRegistered = false;
366 > for (const agent of this.getAgents()) {
367 > if (agent.isDefault) { chatAgents.ts
368 > if (!agent.isCore) { chatAgents.ts
369 > extensionAgentRegistered = true;
370 > }
371 > if (agent.id === 'chat.setup' || agent.id === 'github.copilot.editsAgent') {
372 // TODO@roblourens firing the event below probably isn't necessary but leave it alone for now
373 toolsAgentRegistered = true;
374 > } else { chatAgents.ts
375 > defaultAgentRegistered = true;
376 > }
377 > }
378 > } chatAgents.ts
379 > this._defaultAgentRegistered.set(defaultAgentRegistered); chatAgents.ts
380 > this._extensionAgentRegistered.set(extensionAgentRegistered);
381 > if (toolsAgentRegistered !== this._hasToolsAgent) {
382 this._hasToolsAgent = toolsAgentRegistered;
383 this._onDidChangeAgents.fire(this.getDefaultAgent(ChatAgentLocation.Chat, ChatModeKind.Agent));
384 }
385 > } chatAgents.ts
387 > registerAgentImplementation(id: string, agentImpl: IChatAgentImplementation): IDisposable {
388 > const entry = this._agents.get(id); chatAgents.ts
389 > if (!entry) {
390 throw new Error(`Unknown agent: ${JSON.stringify(id)}`);
391 }
393 > if (entry.impl) {
394 throw new Error(`Agent already has implementation: ${JSON.stringify(id)}`);
395 }
397 > if (entry.data.isDefault) {
398 > this._hasDefaultAgent.set(true); chatAgents.ts
399 > }
401 > entry.impl = agentImpl;
402 > this._onDidChangeAgents.fire(new MergedChatAgent(entry.data, agentImpl));
403 >
404 > return toDisposable(() => {
405 > entry.impl = undefined;
406 > this._onDidChangeAgents.fire(undefined);
407 >
408 > if (entry.data.isDefault) {
409 > this._hasDefaultAgent.set(Iterable.some(this._agents.values(), agent => agent.data.isDefault && !!agent.impl)); chatAgents.ts
410 > }
411 > }); chatAgents.ts
412 > } chatAgents.ts
414 > registerDynamicAgent(data: IChatAgentData, agentImpl: IChatAgentImplementation): IDisposable {
415 data.isDynamic = true;
416 const agent = { data, impl: agentImpl };
423 });
424 }
426 > private _agentCompletionProviders = new Map<string, (query: string, token: CancellationToken) => Promise<IChatAgentCompletionItem[]>>();
427 >
428 > registerAgentCompletionProvider(id: string, provider: (query: string, token: CancellationToken) => Promise<IChatAgentCompletionItem[]>) {
429 this._agentCompletionProviders.set(id, provider);
430 return {
432 };
433 }
435 > async getAgentCompletionItems(id: string, query: string, token: CancellationToken) {
436 return await this._agentCompletionProviders.get(id)?.(query, token) ?? [];
437 }
439 > updateAgent(id: string, updateMetadata: IChatAgentMetadata): void {
440 > const agent = this._agents.get(id); chatAgents.ts
441 > if (!agent?.impl) {
442 throw new Error(`No activated agent with id ${JSON.stringify(id)} registered`);
443 }
444 > agent.data.metadata = { ...agent.data.metadata, ...updateMetadata }; chatAgents.ts
445 > this._onDidChangeAgents.fire(new MergedChatAgent(agent.data, agent.impl));
446 > }
448 > getDefaultAgent(location: ChatAgentLocation, mode: ChatModeKind = ChatModeKind.Ask): IChatAgent | undefined {
449 return this._preferExtensionAgent(this.getActivatedAgents().filter(a => {
450 if (mode && !a.modes.includes(mode)) {
455 }));
456 }
458 > public get hasToolsAgent(): boolean {
459 // The chat participant enablement is just based on this setting. Don't wait for the extension to be loaded.
460 return !!this.configurationService.getValue(ChatConfiguration.AgentEnabled);
461 }
463 > getContributedDefaultAgent(location: ChatAgentLocation): IChatAgentData | undefined {
464 > return this._preferExtensionAgent(this.getAgents().filter(a => !!a.isDefault && a.locations.includes(location))); chatAgents.ts
465 > }
467 > private _preferExtensionAgent<T extends IChatAgentData>(agents: T[]): T | undefined {
468 > // We potentially have multiple agents on the same location, chatAgents.ts
469 > // contributed from core and from extensions.
470 > // This method will prefer the last extensions provided agent
471 > // falling back to the last core agent if no extension agent is found.
472 > return findLast(agents, agent => !agent.isCore) ?? agents.at(-1);
473 > }
475 > getAgent(id: string, includeDisabled = false): IChatAgentData | undefined {
476 > if (!this._agentIsEnabled(id) && !includeDisabled) { chatAgents.ts
477 return;
478 }
480 > return this._agents.get(id)?.data;
481 > }
483 > private _agentIsEnabled(idOrAgent: string | IChatAgentEntry): boolean {
484 > const entry = typeof idOrAgent === 'string' ? this._agents.get(idOrAgent) : idOrAgent; chatAgents.ts
485 > return !entry?.data.when || this.contextKeyService.contextMatchesRules(ContextKeyExpr.deserialize(entry.data.when));
486 > }
488 > getAgentByFullyQualifiedId(id: string): IChatAgentData | undefined {
489 const agent = Iterable.find(this._agents.values(), a => getFullyQualifiedId(a.data) === id)?.data;
490 if (agent && !this._agentIsEnabled(agent.id)) {
494 return agent;
495 }
497 > /**
498 > * Returns all agent datas that exist- static registered and dynamic ones.
499 > */
500 > getAgents(): IChatAgentData[] {
501 > return Array.from(this._agents.values()) chatAgents.ts
502 > .map(entry => entry.data)
503 > .filter(a => this._agentIsEnabled(a.id));
504 > }
506 > getActivatedAgents(): IChatAgent[] {
507 > return Array.from(this._agents.values()) chatAgents.ts
508 > .filter(a => !!a.impl)
509 > .filter(a => this._agentIsEnabled(a.data.id))
510 > .map(a => new MergedChatAgent(a.data, a.impl!));
511 > }
513 > getAgentsByName(name: string): IChatAgentData[] {
514 return this._preferExtensionAgents(this.getAgents().filter(a => a.name === name));
515 }
517 > private _preferExtensionAgents<T extends IChatAgentData>(agents: T[]): T[] {
518 // We potentially have multiple agents on the same location,
519 // contributed from core and from extensions.
523 return extensionAgents.length > 0 ? extensionAgents : agents;
524 }
526 > agentHasDupeName(id: string): boolean {
527 const agent = this.getAgent(id);
528 if (!agent) {
533 .filter(a => a.extensionId.value !== agent.extensionId.value).length > 0;
534 }
536 > async invokeAgent(id: string, request: IChatAgentRequest, progress: (parts: IChatProgress[]) => void, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatAgentResult> {
537 markChat(request.sessionResource, ChatPerfMark.AgentWillInvoke);
538 const data = this._agents.get(id);
546 return result;
547 }
549 > setRequestTools(id: string, requestId: string, tools: UserSelectedTools): void {
550 const data = this._agents.get(id);
551 if (!data?.impl) {
555 data.impl.setRequestTools?.(requestId, tools);
556 }
558 > setYieldRequested(id: string, requestId: string, value: boolean): void {
559 const data = this._agents.get(id);
560 if (!data?.impl) {
564 data.impl.setYieldRequested?.(requestId, value);
565 }
567 > async getFollowups(id: string, request: IChatAgentRequest, result: IChatAgentResult, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatFollowup[]> {
568 const data = this._agents.get(id);
569 if (!data?.impl?.provideFollowups) {
573 return data.impl.provideFollowups(request, result, history, token);
574 }
576 > async getChatTitle(id: string, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<string | undefined> {
577 const data = this._agents.get(id);
578 if (!data?.impl?.provideChatTitle) {
582 return data.impl.provideChatTitle(history, token);
583 }
585 > async getChatSummary(id: string, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<string | undefined> {
586 const data = this._agents.get(id);
587 if (!data?.impl?.provideChatSummary) {
591 return data.impl.provideChatSummary(history, token);
592 }
594 > registerChatParticipantDetectionProvider(handle: number, provider: IChatParticipantDetectionProvider) {
595 this._chatParticipantDetectionProviders.set(handle, provider);
596 return toDisposable(() => {
598 });
599 }
601 > hasChatParticipantDetectionProviders() {
602 return this._chatParticipantDetectionProviders.size > 0;
603 }
605 > async detectAgentOrCommand(request: IChatAgentRequest, history: IChatAgentHistoryEntry[], options: { location: ChatAgentLocation }, token: CancellationToken): Promise<{ agent: IChatAgentData; command?: IChatAgentCommand } | undefined> {
606 // TODO@joyceerhl should we have a selector to be able to narrow down which provider to use
607 const provider = Iterable.first(this._chatParticipantDetectionProviders.values());
643 return { agent, command };
644 }
645 > } chatAgents.ts
646 >
647 > export class MergedChatAgent implements IChatAgent {
648 > constructor(
649 > private readonly data: IChatAgentData, chatAgents.ts
650 > private readonly impl: IChatAgentImplementation
651 > ) { }
652 > when?: string | undefined; chatAgents.ts
653 > publisherDisplayName?: string | undefined;
654 > isDynamic?: boolean | undefined;
655 >
656 > get id(): string { return this.data.id; }
657 > get name(): string { return this.data.name ?? ''; }
658 > get fullName(): string { return this.data.fullName ?? ''; }
659 > get description(): string { return this.data.description ?? ''; }
660 > get extensionId(): ExtensionIdentifier { return this.data.extensionId; }
661 > get extensionVersion(): string | undefined { return this.data.extensionVersion; }
662 > get extensionPublisherId(): string { return this.data.extensionPublisherId; }
663 > get extensionPublisherDisplayName() { return this.data.publisherDisplayName; }
664 > get extensionDisplayName(): string { return this.data.extensionDisplayName; }
665 > get isDefault(): boolean | undefined { return this.data.isDefault; }
666 > get isCore(): boolean | undefined { return this.data.isCore; }
667 > get metadata(): IChatAgentMetadata { return this.data.metadata; }
668 > get slashCommands(): IChatAgentCommand[] { return this.data.slashCommands; }
669 > get locations(): ChatAgentLocation[] { return this.data.locations; }
670 > get modes(): ChatModeKind[] { return this.data.modes; }
671 > get disambiguation(): { category: string; description: string; examples: string[] }[] { return this.data.disambiguation; }
672 >
673 > async invoke(request: IChatAgentRequest, progress: (parts: IChatProgress[]) => void, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatAgentResult> {
674 return this.impl.invoke(request, progress, history, token);
675 }
677 > setRequestTools(requestId: string, tools: UserSelectedTools): void {
678 this.impl.setRequestTools?.(requestId, tools);
679 }
681 > setYieldRequested(requestId: string, value: boolean): void {
682 this.impl.setYieldRequested?.(requestId, value);
683 }
685 > async provideFollowups(request: IChatAgentRequest, result: IChatAgentResult, history: IChatAgentHistoryEntry[], token: CancellationToken): Promise<IChatFollowup[]> {
686 if (this.impl.provideFollowups) {
687 return this.impl.provideFollowups(request, result, history, token);
690 return [];
691 }
693 > toJSON(): IChatAgentData {
694 return this.data;
695 }
696 > } chatAgents.ts
697 >
698 > export const IChatAgentNameService = createDecorator<IChatAgentNameService>('chatAgentNameService');
699 >
700 > export interface IChatAgentNameService {
701 > _serviceBrand: undefined;
702 > getAgentNameRestriction(chatAgentData: IChatAgentData): boolean;
703 > }
704 >
705 > export class ChatAgentNameService implements IChatAgentNameService {
706 >
707 > declare _serviceBrand: undefined;
708 >
709 > constructor(
710 @ILanguageModelsService private readonly languageModelsService: ILanguageModelsService,
711 ) {
712 }
714 > /**
715 > * Returns true if the agent is allowed to use this name
716 > */
717 > getAgentNameRestriction(chatAgentData: IChatAgentData): boolean {
718 if (chatAgentData.isCore) {
719 return true; // core agents are always allowed to use any name
725 return nameAllowed && fullNameAllowed;
726 }
728 > private checkAgentNameRestriction(name: string, chatAgentData: IChatAgentData): IObservable<boolean> {
729 // Registry is a map of name to an array of extension publisher IDs or extension IDs that are allowed to use it.
730 // Look up the list of extensions that are allowed to use this name
738 });
739 }
740 > } chatAgents.ts
741 >
742 > export function getFullyQualifiedId(chatAgentData: IChatAgentData): string {
743 return `${chatAgentData.extensionId.value}.${chatAgentData.id}`;
744 }
746 > /**
747 > * There was a period where serialized chat agent data used 'id' instead of 'name'.
748 > * Don't copy this pattern, serialized data going forward should be versioned with strict interfaces.
749 > */
750 > interface IOldSerializedChatAgentData extends Omit<ISerializableChatAgentData, 'name'> {
751 > id: string;
752 > extensionPublisher?: string;
753 > }
754 >
755 function isSerializableChatAgentData(obj: ISerializableChatAgentData | IOldSerializedChatAgentData): obj is ISerializableChatAgentData {
756 return (obj as ISerializableChatAgentData).name !== undefined;
757 }
759 > export function reviveSerializedAgent(raw: ISerializableChatAgentData | IOldSerializedChatAgentData): IChatAgentData {
760 const normalized: ISerializableChatAgentData = isSerializableChatAgentData(raw) ?
761 raw :
src/vs/platform/extensions/common/extensionsApiProposals.ts 539 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensionsApiProposals.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // THIS IS A GENERATED FILE. DO NOT EDIT DIRECTLY.
7 >
8 > const _allApiProposals = {
9 > activeComment: {
10 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.activeComment.d.ts',
11 > },
12 > agentEditorComments: {
13 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.agentEditorComments.d.ts',
14 > },
15 > agentSessionsWorkspace: {
16 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.agentSessionsWorkspace.d.ts',
17 > },
18 > agentsWindowConfiguration: {
19 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.agentsWindowConfiguration.d.ts',
20 > },
21 > aiRelatedInformation: {
22 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.aiRelatedInformation.d.ts',
23 > },
24 > aiSettingsSearch: {
25 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.aiSettingsSearch.d.ts',
26 > },
27 > aiTextSearchProvider: {
28 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.aiTextSearchProvider.d.ts',
29 > },
30 > authIssuers: {
31 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authIssuers.d.ts',
32 > },
33 > authLearnMore: {
34 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authLearnMore.d.ts',
35 > },
36 > authProviderSpecific: {
37 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authProviderSpecific.d.ts',
38 > },
39 > authSession: {
40 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authSession.d.ts',
41 > },
42 > authSessionAudience: {
43 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authSessionAudience.d.ts',
44 > },
45 > authenticationChallenges: {
46 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.authenticationChallenges.d.ts',
47 > },
48 > browser: {
49 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.browser.d.ts',
50 > },
51 > canonicalUriProvider: {
52 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.canonicalUriProvider.d.ts',
53 > },
54 > chatContextProvider: {
55 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatContextProvider.d.ts',
56 > },
57 > chatDebug: {
58 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatDebug.d.ts',
59 > },
60 > chatHooks: {
61 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatHooks.d.ts',
62 > },
63 > chatInputNotification: {
64 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatInputNotification.d.ts',
65 > },
66 > chatOutputRenderer: {
67 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatOutputRenderer.d.ts',
68 > },
69 > chatParticipantAdditions: {
70 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatParticipantAdditions.d.ts',
71 > },
72 > chatParticipantPrivate: {
73 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatParticipantPrivate.d.ts',
74 > },
75 > chatPromptFiles: {
76 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatPromptFiles.d.ts',
77 > },
78 > chatProvider: {
79 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatProvider.d.ts',
80 > },
81 > chatReferenceBinaryData: {
82 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatReferenceBinaryData.d.ts',
83 > },
84 > chatReferenceDiagnostic: {
85 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatReferenceDiagnostic.d.ts',
86 > },
87 > chatSessionCustomizationProvider: {
88 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatSessionCustomizationProvider.d.ts',
89 > },
90 > chatSessionsProvider: {
91 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatSessionsProvider.d.ts',
92 > },
93 > chatStatusItem: {
94 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatStatusItem.d.ts',
95 > },
96 > chatTab: {
97 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.chatTab.d.ts',
98 > },
99 > codeActionAI: {
100 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.codeActionAI.d.ts',
101 > },
102 > codeActionRanges: {
103 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.codeActionRanges.d.ts',
104 > },
105 > codiconDecoration: {
106 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.codiconDecoration.d.ts',
107 > },
108 > commentReactor: {
109 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.commentReactor.d.ts',
110 > },
111 > commentReveal: {
112 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.commentReveal.d.ts',
113 > },
114 > commentThreadApplicability: {
115 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.commentThreadApplicability.d.ts',
116 > },
117 > commentingRangeHint: {
118 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.commentingRangeHint.d.ts',
119 > },
120 > commentsDraftState: {
121 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.commentsDraftState.d.ts',
122 > },
123 > contribAccessibilityHelpContent: {
124 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribAccessibilityHelpContent.d.ts',
125 > },
126 > contribChatEditorInlineGutterMenu: {
127 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribChatEditorInlineGutterMenu.d.ts',
128 > },
129 > contribCommentEditorActionsMenu: {
130 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribCommentEditorActionsMenu.d.ts',
131 > },
132 > contribCommentPeekContext: {
133 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribCommentPeekContext.d.ts',
134 > },
135 > contribCommentThreadAdditionalMenu: {
136 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribCommentThreadAdditionalMenu.d.ts',
137 > },
138 > contribCommentsViewThreadMenus: {
139 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribCommentsViewThreadMenus.d.ts',
140 > },
141 > contribDebugCreateConfiguration: {
142 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribDebugCreateConfiguration.d.ts',
143 > },
144 > contribDiffEditorGutterToolBarMenus: {
145 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribDiffEditorGutterToolBarMenus.d.ts',
146 > },
147 > contribEditSessions: {
148 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribEditSessions.d.ts',
149 > },
150 > contribEditorContentMenu: {
151 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribEditorContentMenu.d.ts',
152 > },
153 > contribLabelFormatterWorkspaceTooltip: {
154 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribLabelFormatterWorkspaceTooltip.d.ts',
155 > },
156 > contribLanguageModelToolSets: {
157 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribLanguageModelToolSets.d.ts',
158 > },
159 > contribMenuBarHome: {
160 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribMenuBarHome.d.ts',
161 > },
162 > contribMergeEditorMenus: {
163 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribMergeEditorMenus.d.ts',
164 > },
165 > contribMultiDiffEditorMenus: {
166 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribMultiDiffEditorMenus.d.ts',
167 > },
168 > contribNotebookStaticPreloads: {
169 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribNotebookStaticPreloads.d.ts',
170 > },
171 > contribRemoteHelp: {
172 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribRemoteHelp.d.ts',
173 > },
174 > contribShareMenu: {
175 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribShareMenu.d.ts',
176 > },
177 > contribSourceControlArtifactGroupMenu: {
178 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlArtifactGroupMenu.d.ts',
179 > },
180 > contribSourceControlArtifactMenu: {
181 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlArtifactMenu.d.ts',
182 > },
183 > contribSourceControlHistoryItemMenu: {
184 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlHistoryItemMenu.d.ts',
185 > },
186 > contribSourceControlHistoryTitleMenu: {
187 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlHistoryTitleMenu.d.ts',
188 > },
189 > contribSourceControlInputBoxMenu: {
190 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlInputBoxMenu.d.ts',
191 > },
192 > contribSourceControlTitleMenu: {
193 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribSourceControlTitleMenu.d.ts',
194 > },
195 > contribStatusBarItems: {
196 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribStatusBarItems.d.ts',
197 > },
198 > contribViewContainerTitle: {
199 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribViewContainerTitle.d.ts',
200 > },
201 > contribViewsRemote: {
202 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribViewsRemote.d.ts',
203 > },
204 > contribViewsWelcome: {
205 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.contribViewsWelcome.d.ts',
206 > },
207 > css: {
208 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.css.d.ts',
209 > },
210 > customEditorDiffs: {
211 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.customEditorDiffs.d.ts',
212 > },
213 > customEditorMove: {
214 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.customEditorMove.d.ts',
215 > },
216 > customEditorPriority: {
217 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.customEditorPriority.d.ts',
218 > },
219 > dataChannels: {
220 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.dataChannels.d.ts',
221 > },
222 > debugVisualization: {
223 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.debugVisualization.d.ts',
224 > },
225 > defaultChatParticipant: {
226 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.defaultChatParticipant.d.ts',
227 > },
228 > devDeviceId: {
229 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.devDeviceId.d.ts',
230 > },
231 > diffCommand: {
232 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.diffCommand.d.ts',
233 > },
234 > diffContentOptions: {
235 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.diffContentOptions.d.ts',
236 > },
237 > documentDiff: {
238 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.documentDiff.d.ts',
239 > },
240 > documentFiltersExclusive: {
241 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.documentFiltersExclusive.d.ts',
242 > },
243 > documentSyntaxHighlighting: {
244 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.documentSyntaxHighlighting.d.ts',
245 > },
246 > editSessionIdentityProvider: {
247 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.editSessionIdentityProvider.d.ts',
248 > },
249 > editorHoverVerbosityLevel: {
250 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.editorHoverVerbosityLevel.d.ts',
251 > },
252 > editorInsets: {
253 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.editorInsets.d.ts',
254 > },
255 > embeddings: {
256 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.embeddings.d.ts',
257 > },
258 > envIsConnectionMetered: {
259 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.envIsConnectionMetered.d.ts',
260 > },
261 > environmentPower: {
262 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.environmentPower.d.ts',
263 > },
264 > extensionAffinity: {
265 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.extensionAffinity.d.ts',
266 > },
267 > extensionRuntime: {
268 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.extensionRuntime.d.ts',
269 > },
270 > extensionsAny: {
271 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.extensionsAny.d.ts',
272 > },
273 > externalUriOpener: {
274 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.externalUriOpener.d.ts',
275 > },
276 > fileSearchProvider: {
277 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.fileSearchProvider.d.ts',
278 > },
279 > fileSearchProvider2: {
280 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.fileSearchProvider2.d.ts',
281 > },
282 > findFiles2: {
283 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.findFiles2.d.ts',
284 > },
285 > findTextInFiles: {
286 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.findTextInFiles.d.ts',
287 > },
288 > findTextInFiles2: {
289 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.findTextInFiles2.d.ts',
290 > },
291 > fsChunks: {
292 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.fsChunks.d.ts',
293 > },
294 > inlineCompletionsAdditions: {
295 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.inlineCompletionsAdditions.d.ts',
296 > },
297 > interactive: {
298 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.interactive.d.ts',
299 > },
300 > interactiveWindow: {
301 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.interactiveWindow.d.ts',
302 > },
303 > ipc: {
304 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.ipc.d.ts',
305 > },
306 > languageModelCapabilities: {
307 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelCapabilities.d.ts',
308 > },
309 > languageModelPricing: {
310 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelPricing.d.ts',
311 > },
312 > languageModelProxy: {
313 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelProxy.d.ts',
314 > },
315 > languageModelSystem: {
316 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelSystem.d.ts',
317 > },
318 > languageModelThinkingPart: {
319 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelThinkingPart.d.ts',
320 > },
321 > languageModelToolResultAudience: {
322 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelToolResultAudience.d.ts',
323 > },
324 > languageModelToolSupportsModel: {
325 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageModelToolSupportsModel.d.ts',
326 > },
327 > languageStatusText: {
328 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.languageStatusText.d.ts',
329 > },
330 > mappedEditsProvider: {
331 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.mappedEditsProvider.d.ts',
332 > },
333 > markdownAlertSyntax: {
334 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.markdownAlertSyntax.d.ts',
335 > },
336 > mcpServerDefinitions: {
337 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.mcpServerDefinitions.d.ts',
338 > },
339 > mcpToolDefinitions: {
340 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.mcpToolDefinitions.d.ts',
341 > },
342 > multiDocumentHighlightProvider: {
343 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.multiDocumentHighlightProvider.d.ts',
344 > },
345 > nativeWindowHandle: {
346 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.nativeWindowHandle.d.ts',
347 > },
348 > newSymbolNamesProvider: {
349 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.newSymbolNamesProvider.d.ts',
350 > },
351 > notebookCellExecution: {
352 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookCellExecution.d.ts',
353 > },
354 > notebookControllerAffinityHidden: {
355 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookControllerAffinityHidden.d.ts',
356 > },
357 > notebookDeprecated: {
358 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookDeprecated.d.ts',
359 > },
360 > notebookExecution: {
361 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookExecution.d.ts',
362 > },
363 > notebookKernelSource: {
364 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookKernelSource.d.ts',
365 > },
366 > notebookLiveShare: {
367 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookLiveShare.d.ts',
368 > },
369 > notebookMessaging: {
370 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookMessaging.d.ts',
371 > },
372 > notebookMime: {
373 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookMime.d.ts',
374 > },
375 > notebookReplDocument: {
376 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookReplDocument.d.ts',
377 > },
378 > notebookVariableProvider: {
379 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.notebookVariableProvider.d.ts',
380 > },
381 > portsAttributes: {
382 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.portsAttributes.d.ts',
383 > },
384 > profileContentHandlers: {
385 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.profileContentHandlers.d.ts',
386 > },
387 > quickDiffProvider: {
388 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.quickDiffProvider.d.ts',
389 > },
390 > quickPickItemTooltip: {
391 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.quickPickItemTooltip.d.ts',
392 > },
393 > quickPickSortByLabel: {
394 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.quickPickSortByLabel.d.ts',
395 > },
396 > remoteCodingAgents: {
397 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.remoteCodingAgents.d.ts',
398 > },
399 > resolvers: {
400 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.resolvers.d.ts',
401 > },
402 > scmActionButton: {
403 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmActionButton.d.ts',
404 > },
405 > scmArtifactProvider: {
406 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmArtifactProvider.d.ts',
407 > },
408 > scmHistoryProvider: {
409 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmHistoryProvider.d.ts',
410 > },
411 > scmMultiDiffEditor: {
412 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmMultiDiffEditor.d.ts',
413 > },
414 > scmProviderOptions: {
415 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmProviderOptions.d.ts',
416 > },
417 > scmSelectedProvider: {
418 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmSelectedProvider.d.ts',
419 > },
420 > scmTextDocument: {
421 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmTextDocument.d.ts',
422 > },
423 > scmValidation: {
424 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.scmValidation.d.ts',
425 > },
426 > shareProvider: {
427 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.shareProvider.d.ts',
428 > },
429 > speech: {
430 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.speech.d.ts',
431 > },
432 > statusBarItemTooltip: {
433 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.statusBarItemTooltip.d.ts',
434 > },
435 > tabInputMultiDiff: {
436 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.tabInputMultiDiff.d.ts',
437 > },
438 > tabInputTextMerge: {
439 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.tabInputTextMerge.d.ts',
440 > },
441 > taskExecutionTerminal: {
442 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.taskExecutionTerminal.d.ts',
443 > },
444 > taskPresentationGroup: {
445 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.taskPresentationGroup.d.ts',
446 > },
447 > taskProblemMatcherStatus: {
448 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.taskProblemMatcherStatus.d.ts',
449 > },
450 > taskRunOptions: {
451 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.taskRunOptions.d.ts',
452 > },
453 > telemetry: {
454 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.telemetry.d.ts',
455 > },
456 > terminalCompletionProvider: {
457 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalCompletionProvider.d.ts',
458 > },
459 > terminalDataWriteEvent: {
460 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalDataWriteEvent.d.ts',
461 > },
462 > terminalDimensions: {
463 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalDimensions.d.ts',
464 > },
465 > terminalExecuteCommandEvent: {
466 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalExecuteCommandEvent.d.ts',
467 > },
468 > terminalQuickFixProvider: {
469 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalQuickFixProvider.d.ts',
470 > },
471 > terminalSelection: {
472 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalSelection.d.ts',
473 > },
474 > terminalShellEnv: {
475 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalShellEnv.d.ts',
476 > },
477 > terminalTitle: {
478 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.terminalTitle.d.ts',
479 > },
480 > testObserver: {
481 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.testObserver.d.ts',
482 > },
483 > testRelatedCode: {
484 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.testRelatedCode.d.ts',
485 > },
486 > textDocumentChangeReason: {
487 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.textDocumentChangeReason.d.ts',
488 > },
489 > textEditorDiffInformation: {
490 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.textEditorDiffInformation.d.ts',
491 > },
492 > textSearchComplete2: {
493 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.textSearchComplete2.d.ts',
494 > },
495 > textSearchProvider: {
496 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.textSearchProvider.d.ts',
497 > },
498 > textSearchProvider2: {
499 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.textSearchProvider2.d.ts',
500 > },
501 > timeline: {
502 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.timeline.d.ts',
503 > },
504 > tokenInformation: {
505 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.tokenInformation.d.ts',
506 > },
507 > toolInvocationApproveCombination: {
508 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.toolInvocationApproveCombination.d.ts',
509 > },
510 > toolProgress: {
511 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.toolProgress.d.ts',
512 > },
513 > treeItemMarkdownLabel: {
514 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.treeItemMarkdownLabel.d.ts',
515 > },
516 > treeViewActiveItem: {
517 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.treeViewActiveItem.d.ts',
518 > },
519 > treeViewMarkdownMessage: {
520 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.treeViewMarkdownMessage.d.ts',
521 > },
522 > treeViewReveal: {
523 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.treeViewReveal.d.ts',
524 > },
525 > tunnelFactory: {
526 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.tunnelFactory.d.ts',
527 > },
528 > tunnels: {
529 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.tunnels.d.ts',
530 > },
531 > valueSelectionInQuickPick: {
532 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.valueSelectionInQuickPick.d.ts',
533 > },
534 > workspaceTrust: {
535 > proposal: 'https://raw.githubusercontent.com/microsoft/vscode/main/src/vscode-dts/vscode.proposed.workspaceTrust.d.ts',
536 > }
537 > };
538 > export const allApiProposals = Object.freeze<{ [proposalName: string]: Readonly<{ proposal: string }> }>(_allApiProposals);
539 > export type ApiProposalName = keyof typeof _allApiProposals;
src/vs/workbench/test/common/workbenchTestServices.ts 514 covered LOC · 96 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workbenchTestServices.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DeferredPromise, timeout } from '../../../base/common/async.js';
7 > import { bufferToStream, readableToBuffer, VSBuffer, VSBufferReadable } from '../../../base/common/buffer.js';
8 > import { CancellationToken } from '../../../base/common/cancellation.js';
9 > import { Emitter, Event } from '../../../base/common/event.js';
10 > import { Iterable } from '../../../base/common/iterator.js';
11 > import { Disposable, IDisposable, toDisposable } from '../../../base/common/lifecycle.js';
12 > import { ResourceMap, ResourceSet } from '../../../base/common/map.js';
13 > import { Schemas } from '../../../base/common/network.js';
14 > import { observableValue } from '../../../base/common/observable.js';
15 > import { join } from '../../../base/common/path.js';
16 > import { isLinux, isMacintosh } from '../../../base/common/platform.js';
17 > import { basename, isEqual, isEqualOrParent } from '../../../base/common/resources.js';
18 > import { URI } from '../../../base/common/uri.js';
19 > import { ITextResourcePropertiesService } from '../../../editor/common/services/textResourceConfiguration.js';
20 > import { IConfigurationService } from '../../../platform/configuration/common/configuration.js';
21 > import { IResourceEditorInput } from '../../../platform/editor/common/editor.js';
22 > import { FileChangesEvent, FileOperationEvent, FileSystemProviderCapabilities, IBaseFileStat, ICreateFileOptions, IFileContent, IFileService, IFileStat, IFileStatResult, IFileStatWithMetadata, IFileStatWithPartialMetadata, IFileStreamContent, IFileSystemProvider, IFileSystemProviderActivationEvent, IFileSystemProviderCapabilitiesChangeEvent, IFileSystemWatcher, IReadFileOptions, IReadFileStreamOptions, IResolveFileOptions, IResolveMetadataFileOptions, IWatchOptions, IWatchOptionsWithCorrelation, IWriteFileOptions } from '../../../platform/files/common/files.js';
23 > import { AbstractLoggerService, ILogger, LogLevel, NullLogger } from '../../../platform/log/common/log.js';
24 > import { IMarker, IMarkerData, IMarkerService, IResourceMarker, MarkerStatistics } from '../../../platform/markers/common/markers.js';
25 > import product from '../../../platform/product/common/product.js';
26 > import { IProgress, IProgressStep } from '../../../platform/progress/common/progress.js';
27 > import { InMemoryStorageService, WillSaveStateReason } from '../../../platform/storage/common/storage.js';
28 > import { toUserDataProfile } from '../../../platform/userDataProfile/common/userDataProfile.js';
29 > import { ISingleFolderWorkspaceIdentifier, IWorkspace, IWorkspaceContextService, IWorkspaceFolder, IWorkspaceFoldersChangeEvent, IWorkspaceFoldersWillChangeEvent, IWorkspaceIdentifier, WorkbenchState, Workspace } from '../../../platform/workspace/common/workspace.js';
30 > import { IWorkspaceTrustEnablementService, IWorkspaceTrustManagementService, IWorkspaceTrustRequestService, IWorkspaceTrustTransitionParticipant, IWorkspaceTrustUriInfo, ResourceTrustRequestOptions, WorkspaceTrustRequestOptions, WorkspaceTrustUriResponse } from '../../../platform/workspace/common/workspaceTrust.js';
31 > import { TestWorkspace } from '../../../platform/workspace/test/common/testWorkspace.js';
32 > import { GroupIdentifier, IRevertOptions, ISaveOptions, SaveReason } from '../../common/editor.js';
33 > import { EditorInput } from '../../common/editor/editorInput.js';
34 > import { IActivity, IActivityService } from '../../services/activity/common/activity.js';
35 > import { ChatEntitlement, ChatEntitlementContext, IChatEntitlementService } from '../../services/chat/common/chatEntitlementService.js';
36 > import { Lazy } from '../../../base/common/lazy.js';
37 > import { NullExtensionService } from '../../services/extensions/common/extensions.js';
38 > import { IAutoSaveConfiguration, IAutoSaveMode, IFilesConfigurationService } from '../../services/filesConfiguration/common/filesConfigurationService.js';
39 > import { IHistoryService } from '../../services/history/common/history.js';
40 > import { BeforeShutdownErrorEvent, ILifecycleService, InternalBeforeShutdownEvent, LifecyclePhase, ShutdownReason, StartupKind, WillShutdownEvent } from '../../services/lifecycle/common/lifecycle.js';
41 > import { IResourceEncoding } from '../../services/textfile/common/textfiles.js';
42 > import { IUserDataProfileService } from '../../services/userDataProfile/common/userDataProfile.js';
43 > import { IStoredFileWorkingCopySaveEvent } from '../../services/workingCopy/common/storedFileWorkingCopy.js';
44 > import { IWorkingCopy, IWorkingCopyBackup, WorkingCopyCapabilities } from '../../services/workingCopy/common/workingCopy.js';
45 > import { ICopyOperation, ICreateFileOperation, ICreateOperation, IDeleteOperation, IFileOperationUndoRedoInfo, IMoveOperation, IStoredFileWorkingCopySaveParticipant, IStoredFileWorkingCopySaveParticipantContext, IWorkingCopyFileOperationParticipant, IWorkingCopyFileService, WorkingCopyFileEvent } from '../../services/workingCopy/common/workingCopyFileService.js';
46 >
47 > export class TestLoggerService extends AbstractLoggerService {
48 > constructor(logsHome?: URI) {
49 super(LogLevel.Info, logsHome ?? URI.file('tests').with({ scheme: 'vscode-tests' }));
50 }
51 > protected doCreateLogger(): ILogger { return new NullLogger(); } workbenchTestServices.ts
52 > }
53 >
54 > export class TestTextResourcePropertiesService implements ITextResourcePropertiesService {
55 >
56 > declare readonly _serviceBrand: undefined;
57 >
58 > constructor(
59 @IConfigurationService private readonly configurationService: IConfigurationService,
60 ) {
61 }
63 > getEOL(resource: URI, language?: string): string {
64 const eol = this.configurationService.getValue('files.eol', { overrideIdentifier: language, resource });
65 if (eol && typeof eol === 'string' && eol !== 'auto') {
68 return (isLinux || isMacintosh) ? '\n' : '\r\n';
69 }
71 >
72 > export class TestUserDataProfileService implements IUserDataProfileService {
73
74 readonly _serviceBrand: undefined;
75 readonly onDidChangeCurrentProfile = Event.None;
76 readonly currentProfile = toUserDataProfile('test', 'test', URI.file('tests').with({ scheme: 'vscode-tests' }), URI.file('tests').with({ scheme: 'vscode-tests' }));
77 > async updateCurrentProfile(): Promise<void> { } workbenchTestServices.ts
78 > }
79 >
80 > export class TestContextService implements IWorkspaceContextService {
81 >
82 > declare readonly _serviceBrand: undefined;
83 >
84 > private workspace: Workspace;
85 > private options: object;
86 >
87 > private readonly _onDidChangeWorkspaceName: Emitter<void>;
88 > get onDidChangeWorkspaceName(): Event<void> { return this._onDidChangeWorkspaceName.event; }
89 >
90 > private readonly _onWillChangeWorkspaceFolders: Emitter<IWorkspaceFoldersWillChangeEvent>;
91 > get onWillChangeWorkspaceFolders(): Event<IWorkspaceFoldersWillChangeEvent> { return this._onWillChangeWorkspaceFolders.event; }
92 >
93 > private readonly _onDidChangeWorkspaceFolders: Emitter<IWorkspaceFoldersChangeEvent>;
94 > get onDidChangeWorkspaceFolders(): Event<IWorkspaceFoldersChangeEvent> { return this._onDidChangeWorkspaceFolders.event; }
95 >
96 > private readonly _onDidChangeWorkbenchState: Emitter<WorkbenchState>;
97 > get onDidChangeWorkbenchState(): Event<WorkbenchState> { return this._onDidChangeWorkbenchState.event; }
98 >
99 > constructor(workspace = TestWorkspace, options = null) {
100 > this.workspace = workspace; workbenchTestServices.ts
101 > this.options = options || Object.create(null);
102 > this._onDidChangeWorkspaceName = new Emitter<void>();
103 > this._onWillChangeWorkspaceFolders = new Emitter<IWorkspaceFoldersWillChangeEvent>();
104 > this._onDidChangeWorkspaceFolders = new Emitter<IWorkspaceFoldersChangeEvent>();
105 > this._onDidChangeWorkbenchState = new Emitter<WorkbenchState>();
106 > }
108 > getFolders(): IWorkspaceFolder[] {
109 return this.workspace ? this.workspace.folders : [];
110 }
112 > getWorkbenchState(): WorkbenchState {
113 if (this.workspace.configuration) {
114 return WorkbenchState.WORKSPACE;
121 return WorkbenchState.EMPTY;
122 }
124 > hasWorkspaceData(): boolean {
125 return this.getWorkbenchState() !== WorkbenchState.EMPTY;
126 }
128 > getCompleteWorkspace(): Promise<IWorkspace> {
129 return Promise.resolve(this.getWorkspace());
130 }
132 > getWorkspace(): IWorkspace {
133 > return this.workspace; workbenchTestServices.ts
134 > }
136 > getWorkspaceFolder(resource: URI): IWorkspaceFolder | null {
137 return this.workspace.getFolder(resource);
138 }
140 > setWorkspace(workspace: any): void {
141 this.workspace = workspace;
142 }
144 > getOptions() {
145 return this.options;
146 }
148 > updateOptions() { }
149 >
150 > isInsideWorkspace(resource: URI): boolean {
151 if (resource && this.workspace) {
152 return isEqualOrParent(resource, this.workspace.folders[0].uri);
155 return false;
156 }
158 > toResource(workspaceRelativePath: string): URI {
159 return URI.file(join('C:\\', workspaceRelativePath));
160 }
162 > isCurrentWorkspace(workspaceIdOrFolder: IWorkspaceIdentifier | ISingleFolderWorkspaceIdentifier | URI): boolean {
163 return URI.isUri(workspaceIdOrFolder) && isEqual(this.workspace.folders[0].uri, workspaceIdOrFolder);
164 }
166 >
167 > export class TestStorageService extends InMemoryStorageService {
168 >
169 > testEmitWillSaveState(reason: WillSaveStateReason): void {
170 super.emitWillSaveState(reason);
171 }
173 >
174 > export class TestHistoryService implements IHistoryService {
175 >
176 > declare readonly _serviceBrand: undefined;
177 >
178 > constructor(private root?: URI) { }
179 >
180 > async reopenLastClosedEditor(): Promise<void> { }
181 > async goForward(): Promise<void> { }
182 > async goBack(): Promise<void> { }
183 > async goPrevious(): Promise<void> { }
184 > async goLast(): Promise<void> { }
185 > removeFromHistory(_input: EditorInput | IResourceEditorInput): void { }
186 > clear(): void { }
187 > clearRecentlyOpened(): void { }
188 > getHistory(): readonly (EditorInput | IResourceEditorInput)[] { return []; }
189 > async openNextRecentlyUsedEditor(group?: GroupIdentifier): Promise<void> { }
190 > async openPreviouslyUsedEditor(group?: GroupIdentifier): Promise<void> { }
191 > getLastActiveWorkspaceRoot(_schemeFilter: string): URI | undefined { return this.root; }
192 > getLastActiveFile(_schemeFilter: string): URI | undefined { return undefined; }
193 > }
194 >
195 > export class TestWorkingCopy extends Disposable implements IWorkingCopy {
196 >
197 > private readonly _onDidChangeDirty = this._register(new Emitter<void>());
198 > readonly onDidChangeDirty = this._onDidChangeDirty.event;
199 >
200 > private readonly _onDidChangeContent = this._register(new Emitter<void>());
201 > readonly onDidChangeContent = this._onDidChangeContent.event;
202 >
203 > private readonly _onDidSave = this._register(new Emitter<IStoredFileWorkingCopySaveEvent>());
204 > readonly onDidSave = this._onDidSave.event;
205 >
206 > readonly capabilities = WorkingCopyCapabilities.None;
207 >
208 > readonly name;
209 >
210 > private dirty = false;
211 >
212 > constructor(readonly resource: URI, isDirty = false, readonly typeId = 'testWorkingCopyType') {
213 super();
214
216 this.dirty = isDirty;
217 }
219 > setDirty(dirty: boolean): void {
220 if (this.dirty !== dirty) {
221 this.dirty = dirty;
223 }
224 }
226 > setContent(content: string): void {
227 this._onDidChangeContent.fire();
228 }
230 > isDirty(): boolean {
231 return this.dirty;
232 }
234 > isModified(): boolean {
235 return this.isDirty();
236 }
238 > async save(options?: ISaveOptions, stat?: IFileStatWithMetadata): Promise<boolean> {
239 this._onDidSave.fire({ reason: options?.reason ?? SaveReason.EXPLICIT, stat: stat ?? createFileStat(this.resource), source: options?.source });
240
241 return true;
242 }
244 > async revert(options?: IRevertOptions): Promise<void> {
245 this.setDirty(false);
246 }
248 > async backup(token: CancellationToken): Promise<IWorkingCopyBackup> {
249 return {};
250 }
252 >
253 > export function createFileStat(resource: URI, readonly = false, isFile?: boolean, isDirectory?: boolean, isSymbolicLink?: boolean, children?: { resource: URI; isFile?: boolean; isDirectory?: boolean; isSymbolicLink?: boolean; executable?: boolean }[] | undefined, executable?: boolean): IFileStatWithMetadata {
254 return {
255 resource,
268 };
269 }
271 > export class TestWorkingCopyFileService implements IWorkingCopyFileService {
272
273 declare readonly _serviceBrand: undefined;
280
281 readonly hasSaveParticipants = false;
282 > addSaveParticipant(participant: IStoredFileWorkingCopySaveParticipant): IDisposable { return Disposable.None; } workbenchTestServices.ts
283 > async runSaveParticipants(workingCopy: IWorkingCopy, context: IStoredFileWorkingCopySaveParticipantContext, progress: IProgress<IProgressStep>, token: CancellationToken): Promise<void> { }
284 >
285 > async delete(operations: IDeleteOperation[], token: CancellationToken, undoInfo?: IFileOperationUndoRedoInfo): Promise<void> { }
286 >
287 > registerWorkingCopyProvider(provider: (resourceOrFolder: URI) => IWorkingCopy[]): IDisposable { return Disposable.None; }
288 >
289 > getDirty(resource: URI): IWorkingCopy[] { return []; }
290 >
291 > create(operations: ICreateFileOperation[], token: CancellationToken, undoInfo?: IFileOperationUndoRedoInfo): Promise<IFileStatWithMetadata[]> { throw new Error('Method not implemented.'); }
292 > createFolder(operations: ICreateOperation[], token: CancellationToken, undoInfo?: IFileOperationUndoRedoInfo): Promise<IFileStatWithMetadata[]> { throw new Error('Method not implemented.'); }
293 >
294 > move(operations: IMoveOperation[], token: CancellationToken, undoInfo?: IFileOperationUndoRedoInfo): Promise<IFileStatWithMetadata[]> { throw new Error('Method not implemented.'); }
295 >
296 > copy(operations: ICopyOperation[], token: CancellationToken, undoInfo?: IFileOperationUndoRedoInfo): Promise<IFileStatWithMetadata[]> { throw new Error('Method not implemented.'); }
297 > }
298 >
299 > export function mock<T>(): Ctor<T> {
300 > // eslint-disable-next-line local/code-no-any-casts workbenchTestServices.ts
301 > return function () { } as any;
302 > }
304 > export interface Ctor<T> {
305 > new(): T;
306 > }
307 >
308 > export class TestExtensionService extends NullExtensionService { }
309 >
310 > export const TestProductService = { _serviceBrand: undefined, ...product };
311 >
312 > export class TestActivityService implements IActivityService {
313 _serviceBrand: undefined;
314 onDidChangeActivity = Event.None;
315 > getViewContainerActivities(viewContainerId: string): IActivity[] { workbenchTestServices.ts
316 return [];
317 }
318 > getActivity(id: string): IActivity[] { workbenchTestServices.ts
319 return [];
320 }
321 > showViewContainerActivity(viewContainerId: string, badge: IActivity): IDisposable { workbenchTestServices.ts
322 return this;
323 }
324 > showViewActivity(viewId: string, badge: IActivity): IDisposable { workbenchTestServices.ts
325 return this;
326 }
327 > showAccountsActivity(activity: IActivity): IDisposable { workbenchTestServices.ts
328 return this;
329 }
330 > showGlobalActivity(activity: IActivity): IDisposable { workbenchTestServices.ts
331 return this;
332 }
334 > dispose() { }
335 > }
336 >
337 > export const NullFilesConfigurationService = new class implements IFilesConfigurationService {
338 >
339 > _serviceBrand: undefined;
340 >
341 > readonly onDidChangeAutoSaveConfiguration = Event.None;
342 > readonly onDidChangeAutoSaveDisabled = Event.None;
343 > readonly onDidChangeReadonly = Event.None;
344 > readonly onDidChangeFilesAssociation = Event.None;
345 >
346 > readonly isHotExitEnabled = false;
347 > readonly hotExitConfiguration = undefined;
348 >
349 > getAutoSaveConfiguration(): IAutoSaveConfiguration { throw new Error('Method not implemented.'); }
350 > getAutoSaveMode(): IAutoSaveMode { throw new Error('Method not implemented.'); }
351 > hasShortAutoSaveDelay(): boolean { throw new Error('Method not implemented.'); }
352 > toggleAutoSave(): Promise<void> { throw new Error('Method not implemented.'); }
353 > enableAutoSaveAfterShortDelay(resourceOrEditor: URI | EditorInput): IDisposable { throw new Error('Method not implemented.'); }
354 > disableAutoSave(resourceOrEditor: URI | EditorInput): IDisposable { throw new Error('Method not implemented.'); }
355 > isReadonly(resource: URI, stat?: IBaseFileStat | undefined): boolean { return false; }
356 > async updateReadonly(_resource: URI | URI[], _readonly: boolean | 'toggle' | 'reset'): Promise<void> { }
357 > preventSaveConflicts(resource: URI, language?: string | undefined): boolean { throw new Error('Method not implemented.'); }
358 > };
359 >
360 > export class TestWorkspaceTrustEnablementService implements IWorkspaceTrustEnablementService {
361 > _serviceBrand: undefined;
362 >
363 > constructor(private isEnabled: boolean = true) { }
364 >
365 > isWorkspaceTrustEnabled(): boolean {
366 return this.isEnabled;
367 }
369 >
370 > export class TestWorkspaceTrustManagementService extends Disposable implements IWorkspaceTrustManagementService {
371 > _serviceBrand: undefined;
372 >
373 > private _onDidChangeTrust = this._register(new Emitter<boolean>());
374 > onDidChangeTrust = this._onDidChangeTrust.event;
375 >
376 > private _onDidChangeTrustedFolders = this._register(new Emitter<void>());
377 > onDidChangeTrustedFolders = this._onDidChangeTrustedFolders.event;
378 >
379 > private _onDidInitiateWorkspaceTrustRequestOnStartup = this._register(new Emitter<void>());
380 > onDidInitiateWorkspaceTrustRequestOnStartup = this._onDidInitiateWorkspaceTrustRequestOnStartup.event;
381 >
382 >
383 > constructor(
384 private trusted: boolean = true,
385 private trustedUris: ResourceSet = new ResourceSet()
387 super();
388 }
390 > get acceptsOutOfWorkspaceFiles(): boolean {
391 throw new Error('Method not implemented.');
392 }
394 > set acceptsOutOfWorkspaceFiles(value: boolean) {
395 throw new Error('Method not implemented.');
396 }
398 > addWorkspaceTrustTransitionParticipant(participant: IWorkspaceTrustTransitionParticipant): IDisposable {
399 throw new Error('Method not implemented.');
400 }
402 > getTrustedUris(): URI[] {
403 throw new Error('Method not implemented.');
404 }
406 > setParentFolderTrust(trusted: boolean): Promise<void> {
407 throw new Error('Method not implemented.');
408 }
410 > getUriTrustInfo(uri: URI): Promise<IWorkspaceTrustUriInfo> {
411 return Promise.resolve({ trusted: this.trustedUris.has(uri), uri });
412 }
414 > async setTrustedUris(folders: URI[]): Promise<void> {
415 this.trustedUris = new ResourceSet(folders);
416 }
418 > async setUrisTrust(uris: URI[], trusted: boolean): Promise<void> {
419 throw new Error('Method not implemented.');
420 }
422 > canSetParentFolderTrust(): boolean {
423 throw new Error('Method not implemented.');
424 }
426 > canSetWorkspaceTrust(): boolean {
427 throw new Error('Method not implemented.');
428 }
430 > isWorkspaceTrusted(): boolean {
431 return this.trusted;
432 }
434 > isWorkspaceTrustForced(): boolean {
435 return false;
436 }
438 > get workspaceTrustInitialized(): Promise<void> {
439 return Promise.resolve();
440 }
442 > get workspaceResolved(): Promise<void> {
443 return Promise.resolve();
444 }
446 > async setWorkspaceTrust(trusted: boolean): Promise<void> {
447 if (this.trusted !== trusted) {
448 this.trusted = trusted;
450 }
451 }
453 >
454 > export class TestWorkspaceTrustRequestService extends Disposable implements IWorkspaceTrustRequestService {
455 > _serviceBrand: any;
456 >
457 > private readonly _onDidInitiateOpenFilesTrustRequest = this._register(new Emitter<void>());
458 > readonly onDidInitiateOpenFilesTrustRequest = this._onDidInitiateOpenFilesTrustRequest.event;
459 >
460 > private readonly _onDidInitiateResourcesTrustRequest = this._register(new Emitter<ResourceTrustRequestOptions>());
461 > readonly onDidInitiateResourcesTrustRequest = this._onDidInitiateResourcesTrustRequest.event;
462 >
463 > private readonly _onDidInitiateWorkspaceTrustRequest = this._register(new Emitter<WorkspaceTrustRequestOptions>());
464 > readonly onDidInitiateWorkspaceTrustRequest = this._onDidInitiateWorkspaceTrustRequest.event;
465 >
466 > private readonly _onDidInitiateWorkspaceTrustRequestOnStartup = this._register(new Emitter<void>());
467 > readonly onDidInitiateWorkspaceTrustRequestOnStartup = this._onDidInitiateWorkspaceTrustRequestOnStartup.event;
468 >
469 > constructor(private readonly _trusted: boolean) {
470 super();
471 }
474 return WorkspaceTrustUriResponse.Open;
475 };
477 > requestOpenFilesTrust(uris: URI[]): Promise<WorkspaceTrustUriResponse> {
478 return this.requestOpenUrisHandler(uris);
479 }
481 > async completeOpenFilesTrustRequest(result: WorkspaceTrustUriResponse, saveResponse: boolean): Promise<void> {
482 throw new Error('Method not implemented.');
483 }
485 > async completeResourcesTrustRequest(uri: URI, result: WorkspaceTrustUriResponse): Promise<void> {
486 throw new Error('Method not implemented.');
487 }
489 > async requestResourcesTrust(options: ResourceTrustRequestOptions): Promise<boolean | undefined> {
490 return this._trusted;
491 }
493 > cancelWorkspaceTrustRequest(): void {
494 throw new Error('Method not implemented.');
495 }
497 > async completeWorkspaceTrustRequest(trusted?: boolean): Promise<void> {
498 throw new Error('Method not implemented.');
499 }
501 > async requestWorkspaceTrust(options?: WorkspaceTrustRequestOptions): Promise<boolean> {
502 return this._trusted;
503 }
505 > requestWorkspaceTrustOnStartup(): void {
506 throw new Error('Method not implemented.');
507 }
509 >
510 > export class TestMarkerService implements IMarkerService {
511
512 _serviceBrand: undefined;
513
514 onMarkerChanged = Event.None;
516 > getStatistics(): MarkerStatistics { throw new Error('Method not implemented.'); }
517 > changeOne(owner: string, resource: URI, markers: IMarkerData[]): void { }
518 > changeAll(owner: string, data: IResourceMarker[]): void { }
519 > remove(owner: string, resources: URI[]): void { }
520 > read(filter?: { owner?: string | undefined; resource?: URI | undefined; severities?: number | undefined; take?: number | undefined } | undefined): IMarker[] { return []; }
521 > installResourceFilter(resource: URI, reason: string): IDisposable {
522 return { dispose: () => { /* TODO: Implement cleanup logic */ } };
523 }
525 >
526 > export class TestFileService implements IFileService {
528 > declare readonly _serviceBrand: undefined;
529 >
530 > private readonly _onDidFilesChange = new Emitter<FileChangesEvent>();
531 > get onDidFilesChange(): Event<FileChangesEvent> { return this._onDidFilesChange.event; }
532 > fireFileChanges(event: FileChangesEvent): void { this._onDidFilesChange.fire(event); }
533 >
534 > private readonly _onDidRunOperation = new Emitter<FileOperationEvent>();
535 > get onDidRunOperation(): Event<FileOperationEvent> { return this._onDidRunOperation.event; }
536 > fireAfterOperation(event: FileOperationEvent): void { this._onDidRunOperation.fire(event); }
537 >
538 > private readonly _onDidChangeFileSystemProviderCapabilities = new Emitter<IFileSystemProviderCapabilitiesChangeEvent>();
539 > get onDidChangeFileSystemProviderCapabilities(): Event<IFileSystemProviderCapabilitiesChangeEvent> { return this._onDidChangeFileSystemProviderCapabilities.event; }
540 > fireFileSystemProviderCapabilitiesChangeEvent(event: IFileSystemProviderCapabilitiesChangeEvent): void { this._onDidChangeFileSystemProviderCapabilities.fire(event); }
541 >
542 > private _onWillActivateFileSystemProvider = new Emitter<IFileSystemProviderActivationEvent>();
543 > readonly onWillActivateFileSystemProvider = this._onWillActivateFileSystemProvider.event;
544 > readonly onDidWatchError = Event.None;
545 >
546 > protected content = 'Hello Html';
547 > protected lastReadFileUri!: URI;
548 >
549 > readonly = false;
550 >
551 > // Tracking functionality for tests
552 > readonly writeOperations: Array<{ resource: URI; content: string }> = [];
553 > readonly readOperations: Array<{ resource: URI }> = [];
554 >
555 > setContent(content: string): void { this.content = content; }
556 > getContent(): string { return this.content; }
557 > getLastReadFileUri(): URI { return this.lastReadFileUri; }
558 >
559 > // Clear tracking data for tests
560 > clearTracking(): void {
561 this.writeOperations.length = 0;
562 this.readOperations.length = 0;
563 }
565 > resolve(resource: URI, _options: IResolveMetadataFileOptions): Promise<IFileStatWithMetadata>;
566 > resolve(resource: URI, _options?: IResolveFileOptions): Promise<IFileStat>;
567 > async resolve(resource: URI, _options?: IResolveFileOptions): Promise<IFileStat> {
568 return createFileStat(resource, this.readonly);
569 }
571 > stat(resource: URI): Promise<IFileStatWithPartialMetadata> {
572 return this.resolve(resource, { resolveMetadata: true });
573 }
575 > async realpath(resource: URI): Promise<URI> {
576 return resource;
577 }
579 > async resolveAll(toResolve: { resource: URI; options?: IResolveFileOptions }[]): Promise<IFileStatResult[]> {
580 const stats = await Promise.all(toResolve.map(resourceAndOption => this.resolve(resourceAndOption.resource, resourceAndOption.options)));
581
582 return stats.map(stat => ({ stat, success: true }));
583 }
585 > readonly notExistsSet = new ResourceMap<boolean>();
586 >
587 > async exists(_resource: URI): Promise<boolean> { return !this.notExistsSet.has(_resource); }
588 >
589 > readShouldThrowError: Error | undefined = undefined;
590 >
591 > async readFile(resource: URI, options?: IReadFileOptions | undefined): Promise<IFileContent> {
592 if (this.readShouldThrowError) {
593 throw this.readShouldThrowError;
602 };
603 }
605 > async readFileStream(resource: URI, options?: IReadFileStreamOptions | undefined): Promise<IFileStreamContent> {
606 if (this.readShouldThrowError) {
607 throw this.readShouldThrowError;
615 };
616 }
618 > writeShouldThrowError: Error | undefined = undefined;
619 >
620 > async writeFile(resource: URI, bufferOrReadable: VSBuffer | VSBufferReadable, options?: IWriteFileOptions): Promise<IFileStatWithMetadata> {
621 await timeout(0);
622
642 return createFileStat(resource, this.readonly);
643 }
645 > move(_source: URI, _target: URI, _overwrite?: boolean): Promise<IFileStatWithMetadata> { return Promise.resolve(null!); }
646 > copy(_source: URI, _target: URI, _overwrite?: boolean): Promise<IFileStatWithMetadata> { return Promise.resolve(null!); }
647 > async cloneFile(_source: URI, _target: URI): Promise<void> { }
648 > createFile(_resource: URI, _content?: VSBuffer | VSBufferReadable, _options?: ICreateFileOptions): Promise<IFileStatWithMetadata> { return Promise.resolve(null!); }
649 > createFolder(_resource: URI): Promise<IFileStatWithMetadata> { return Promise.resolve(null!); }
650 >
651 > onDidChangeFileSystemProviderRegistrations = Event.None;
652 >
653 > private providers = new Map<string, IFileSystemProvider>();
654 >
655 > registerProvider(scheme: string, provider: IFileSystemProvider) {
656 this.providers.set(scheme, provider);
657
658 return toDisposable(() => this.providers.delete(scheme));
659 }
661 > getProvider(scheme: string) {
662 return this.providers.get(scheme);
663 }
665 > async activateProvider(_scheme: string): Promise<void> {
666 this._onWillActivateFileSystemProvider.fire({ scheme: _scheme, join: () => { } });
667 }
668 > async canHandleResource(resource: URI): Promise<boolean> { return this.hasProvider(resource); } workbenchTestServices.ts
669 > hasProvider(resource: URI): boolean { return resource.scheme === Schemas.file || this.providers.has(resource.scheme); }
670 > listCapabilities() {
671 return [
672 { scheme: Schemas.file, capabilities: FileSystemProviderCapabilities.FileOpenReadWriteClose },
674 ];
675 }
676 > hasCapability(resource: URI, capability: FileSystemProviderCapabilities): boolean { workbenchTestServices.ts
677 if (capability === FileSystemProviderCapabilities.PathCaseSensitive && isLinux) {
678 return true;
683 return !!(provider && (provider.capabilities & capability));
684 }
686 > async del(_resource: URI, _options?: { useTrash?: boolean; recursive?: boolean }): Promise<void> { }
687 >
688 > createWatcher(resource: URI, options: IWatchOptions): IFileSystemWatcher {
689 return {
690 onDidChange: Event.None,
692 };
693 }
695 >
696 > readonly watches: URI[] = [];
697 > watch(_resource: URI, options: IWatchOptionsWithCorrelation): IFileSystemWatcher; workbenchTestServices.ts
698 > watch(_resource: URI): IDisposable;
699 > watch(_resource: URI): IDisposable {
700 this.watches.push(_resource);
701
702 return toDisposable(() => this.watches.splice(this.watches.indexOf(_resource), 1));
703 }
705 > getWriteEncoding(_resource: URI): IResourceEncoding { return { encoding: 'utf8', hasBOM: false }; }
706 > dispose(): void { }
707 >
708 > async canCreateFile(source: URI, options?: ICreateFileOptions): Promise<Error | true> { return true; }
709 > async canMove(source: URI, target: URI, overwrite?: boolean | undefined): Promise<Error | true> { return true; }
710 > async canCopy(source: URI, target: URI, overwrite?: boolean | undefined): Promise<Error | true> { return true; }
711 > async canDelete(resource: URI, options?: { useTrash?: boolean | undefined; recursive?: boolean | undefined } | undefined): Promise<Error | true> { return true; }
712 > }
713 >
714 > /**
715 > * TestFileService with in-memory file storage.
716 > * Use this when your test needs to write files and read them back.
717 > */
718 > export class InMemoryTestFileService extends TestFileService {
720 > private files = new ResourceMap<VSBuffer>();
722 > override clearTracking(): void {
723 super.clearTracking();
724 this.files.clear();
725 }
727 > override async readFile(resource: URI, options?: IReadFileOptions | undefined): Promise<IFileContent> {
728 if (this.readShouldThrowError) {
729 throw this.readShouldThrowError;
747 };
748 }
750 > override async writeFile(resource: URI, bufferOrReadable: VSBuffer | VSBufferReadable, options?: IWriteFileOptions): Promise<IFileStatWithMetadata> {
751 await timeout(0);
752
768 return createFileStat(resource, this.readonly);
769 }
771 > override async del(resource: URI, _options?: { useTrash?: boolean; recursive?: boolean }): Promise<void> {
772 this.files.delete(resource);
773 this.notExistsSet.set(resource, true);
774 }
776 > override async exists(resource: URI): Promise<boolean> {
777 const inMemory = this.files.has(resource);
778 if (inMemory) {
782 return super.exists(resource);
783 }
785 >
786 > export class TestChatEntitlementService implements IChatEntitlementService {
788 > _serviceBrand: undefined;
789 >
790 > context: Lazy<ChatEntitlementContext> | undefined;
791 >
792 > readonly organisations: undefined;
793 > readonly isInternal = false;
794 > readonly sku = undefined;
795 > readonly copilotTrackingId = undefined;
796 >
797 > readonly onDidChangeQuotaExceeded = Event.None;
798 > readonly onDidChangeQuotaRemaining = Event.None;
799 > readonly onDidChangeUsageBasedBilling = Event.None;
800 > readonly quotas = {};
801 >
802 > update(token: CancellationToken): Promise<void> {
803 > throw new Error('Method not implemented.');
804 > }
805 >
806 > readonly onDidChangeSentiment = Event.None;
807 > readonly sentimentObs = observableValue({}, {});
808 > readonly sentiment = {};
809 >
810 > readonly onDidChangeEntitlement = Event.None;
811 > entitlement: ChatEntitlement = ChatEntitlement.Unknown;
812 > readonly entitlementObs = observableValue({}, ChatEntitlement.Unknown);
813 >
814 > readonly anonymous = false;
815 > onDidChangeAnonymous = Event.None;
816 > readonly anonymousObs = observableValue({}, false);
817 >
818 > acceptQuotas(): void { }
819 > clearQuotas(): void { }
820 > markAnonymousRateLimited(): void { }
821 > markSetupCompleted(): void { }
822 > setForceHidden(_hidden: boolean): void { }
823 >
824 > readonly clientByokEnabled = false;
825 > readonly hasByokModels = false;
827 >
828 > export class TestLifecycleService extends Disposable implements ILifecycleService {
829
830 declare readonly _serviceBrand: undefined;
884
885 shutdownJoiners: Promise<void>[] = [];
887 > fireShutdown(reason = ShutdownReason.QUIT): void {
888 this.shutdownJoiners = [];
889
898 });
899 }
901 > fireBeforeShutdown(event: InternalBeforeShutdownEvent): void { this._onBeforeShutdown.fire(event); }
902 >
903 > fireWillShutdown(event: WillShutdownEvent): void { this._onWillShutdown.fire(event); }
904 >
905 > async shutdown(): Promise<void> {
906 this.fireShutdown();
907 }
src/vs/platform/dialogs/common/dialogs.ts 511 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- dialogs.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../base/common/cancellation.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { ThemeIcon } from '../../../base/common/themables.js';
9 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
10 > import { basename } from '../../../base/common/resources.js';
11 > import Severity from '../../../base/common/severity.js';
12 > import { URI } from '../../../base/common/uri.js';
13 > import { localize } from '../../../nls.js';
14 > import { createDecorator } from '../../instantiation/common/instantiation.js';
15 > import { ITelemetryData } from '../../telemetry/common/telemetry.js';
16 >
17 > export interface IDialogArgs {
18 > readonly confirmArgs?: IConfirmDialogArgs;
19 > readonly inputArgs?: IInputDialogArgs;
20 > readonly promptArgs?: IPromptDialogArgs;
21 > }
22 >
23 > export interface IBaseDialogOptions {
24 > readonly type?: Severity | DialogType;
25 >
26 > readonly title?: string;
27 > readonly message: string;
28 > readonly detail?: string;
29 >
30 > readonly checkbox?: ICheckbox;
31 >
32 > /**
33 > * Allows to enforce use of custom dialog even in native environments.
34 > */
35 > readonly custom?: boolean | ICustomDialogOptions;
36 >
37 > /**
38 > * An optional cancellation token that can be used to dismiss the dialog
39 > * programmatically for custom dialog implementations.
40 > *
41 > * When cancelled, the custom dialog resolves as if the cancel button was
42 > * pressed. Native dialog handlers cannot currently be dismissed
43 > * programmatically and ignore this option unless a custom dialog is
44 > * explicitly enforced via the {@link custom} option.
45 > */
46 > readonly token?: CancellationToken;
47 > }
48 >
49 > export interface IConfirmDialogArgs {
50 > readonly confirmation: IConfirmation;
51 > }
52 >
53 > export interface IConfirmation extends IBaseDialogOptions {
54 >
55 > /**
56 > * If not provided, defaults to `Yes`.
57 > */
58 > readonly primaryButton?: string;
59 >
60 > /**
61 > * If not provided, defaults to `Cancel`.
62 > */
63 > readonly cancelButton?: string;
64 > }
65 >
66 > export interface IConfirmationResult extends ICheckboxResult {
67 >
68 > /**
69 > * Will be true if the dialog was confirmed with the primary button pressed.
70 > */
71 > readonly confirmed: boolean;
72 > }
73 >
74 > export interface IInputDialogArgs {
75 > readonly input: IInput;
76 > }
77 >
78 > export interface IInput extends IConfirmation {
79 > readonly inputs: IInputElement[];
80 >
81 > /**
82 > * If not provided, defaults to `Ok`.
83 > */
84 > readonly primaryButton?: string;
85 > }
86 >
87 > export interface IInputElement {
88 > readonly type?: 'text' | 'password';
89 > readonly value?: string;
90 > readonly placeholder?: string;
91 > }
92 >
93 > export interface IInputResult extends IConfirmationResult {
94 >
95 > /**
96 > * Values for the input fields as provided by the user or `undefined` if none.
97 > */
98 > readonly values?: string[];
99 > }
100 >
101 > export interface IPromptDialogArgs {
102 > readonly prompt: IPrompt<unknown>;
103 > }
104 >
105 > export interface IPromptBaseButton<T> {
106 >
107 > /**
108 > * @returns the result of the prompt button will be returned
109 > * as result from the `prompt()` call.
110 > */
111 > run(checkbox: ICheckboxResult): T | Promise<T>;
112 > }
113 >
114 > export interface IPromptButton<T> extends IPromptBaseButton<T> {
115 > readonly label: string;
116 > }
117 >
118 > export interface IPromptCancelButton<T> extends IPromptBaseButton<T> {
119 >
120 > /**
121 > * The cancel button to show in the prompt. Defaults to
122 > * `Cancel` if not provided.
123 > */
124 > readonly label?: string;
125 > }
126 >
127 > export interface IPrompt<T> extends IBaseDialogOptions {
128 >
129 > /**
130 > * The buttons to show in the prompt. Defaults to `OK`
131 > * if no buttons or cancel button is provided.
132 > */
133 > readonly buttons?: IPromptButton<T>[];
134 >
135 > /**
136 > * The cancel button to show in the prompt. Defaults to
137 > * `Cancel` if set to `true`.
138 > */
139 > readonly cancelButton?: IPromptCancelButton<T> | true | string;
140 > }
141 >
142 > export interface IPromptWithCustomCancel<T> extends IPrompt<T> {
143 > readonly cancelButton: IPromptCancelButton<T>;
144 > }
145 >
146 > export interface IPromptWithDefaultCancel<T> extends IPrompt<T> {
147 > readonly cancelButton: true | string;
148 > }
149 >
150 > export interface IPromptResult<T> extends ICheckboxResult {
151 >
152 > /**
153 > * The result of the `IPromptButton` that was pressed or `undefined` if none.
154 > */
155 > readonly result?: T;
156 > }
157 >
158 > export interface IPromptResultWithCancel<T> extends IPromptResult<T> {
159 > readonly result: T;
160 > }
161 >
162 > export interface IAsyncPromptResult<T> extends ICheckboxResult {
163 >
164 > /**
165 > * The result of the `IPromptButton` that was pressed or `undefined` if none.
166 > */
167 > readonly result?: Promise<T>;
168 > }
169 >
170 > export interface IAsyncPromptResultWithCancel<T> extends IAsyncPromptResult<T> {
171 > readonly result: Promise<T>;
172 > }
173 >
174 > export type IDialogResult = IConfirmationResult | IInputResult | IAsyncPromptResult<unknown>;
175 >
176 > export type DialogType = 'none' | 'info' | 'error' | 'question' | 'warning';
177 >
178 > export interface ICheckbox {
179 > readonly label: string;
180 > readonly checked?: boolean;
181 > }
182 >
183 > export interface ICheckboxResult {
184 >
185 > /**
186 > * This will only be defined if the confirmation was created
187 > * with the checkbox option defined.
188 > */
189 > readonly checkboxChecked?: boolean;
190 > }
191 >
192 > export interface IPickAndOpenOptions {
193 > readonly forceNewWindow?: boolean;
194 > defaultUri?: URI;
195 > readonly telemetryExtraData?: ITelemetryData;
196 > availableFileSystems?: string[];
197 > remoteAuthority?: string | null;
198 > }
199 >
200 > export interface FileFilter {
201 > readonly extensions: string[];
202 > readonly name: string;
203 > }
204 >
205 > export interface ISaveDialogOptions {
206 >
207 > /**
208 > * A human-readable string for the dialog title
209 > */
210 > title?: string;
211 >
212 > /**
213 > * The resource the dialog shows when opened.
214 > */
215 > defaultUri?: URI;
216 >
217 > /**
218 > * A set of file filters that are used by the dialog. Each entry is a human readable label,
219 > * like "TypeScript", and an array of extensions.
220 > */
221 > filters?: FileFilter[];
222 >
223 > /**
224 > * A human-readable string for the ok button
225 > */
226 > readonly saveLabel?: { readonly withMnemonic: string; readonly withoutMnemonic: string } | string;
227 >
228 > /**
229 > * Specifies a list of schemas for the file systems the user can save to. If not specified, uses the schema of the defaultURI or, if also not specified,
230 > * the schema of the current window.
231 > */
232 > availableFileSystems?: readonly string[];
233 > }
234 >
235 > export interface IOpenDialogOptions {
236 >
237 > /**
238 > * A human-readable string for the dialog title
239 > */
240 > readonly title?: string;
241 >
242 > /**
243 > * The resource the dialog shows when opened.
244 > */
245 > defaultUri?: URI;
246 >
247 > /**
248 > * A human-readable string for the open button.
249 > */
250 > readonly openLabel?: { readonly withMnemonic: string; readonly withoutMnemonic: string } | string;
251 >
252 > /**
253 > * Allow to select files, defaults to `true`.
254 > */
255 > canSelectFiles?: boolean;
256 >
257 > /**
258 > * Allow to select folders, defaults to `false`.
259 > */
260 > canSelectFolders?: boolean;
261 >
262 > /**
263 > * Allow to select many files or folders.
264 > */
265 > readonly canSelectMany?: boolean;
266 >
267 > /**
268 > * A set of file filters that are used by the dialog. Each entry is a human readable label,
269 > * like "TypeScript", and an array of extensions.
270 > */
271 > filters?: FileFilter[];
272 >
273 > /**
274 > * Specifies a list of schemas for the file systems the user can load from. If not specified, uses the schema of the defaultURI or, if also not available,
275 > * the schema of the current window.
276 > */
277 > availableFileSystems?: readonly string[];
278 > }
279 >
280 > export const IDialogService = createDecorator<IDialogService>('dialogService');
281 >
282 > export interface ICustomDialogOptions {
283 > readonly buttonDetails?: string[];
284 > readonly markdownDetails?: ICustomDialogMarkdown[];
285 > readonly classes?: string[];
286 > readonly icon?: ThemeIcon;
287 > readonly disableCloseAction?: boolean;
288 > }
289 >
290 > export interface ICustomDialogMarkdown {
291 > readonly markdown: IMarkdownString;
292 > readonly classes?: string[];
293 > /** Custom link handler for markdown content, see {@link IContentActionHandler}. Defaults to {@link openLinkFromMarkdown}. */
294 > actionHandler?(link: string): Promise<boolean>;
295 > }
296 >
297 > /**
298 > * A handler to bring up modal dialogs.
299 > */
300 > export interface IDialogHandler {
301 >
302 > /**
303 > * Ask the user for confirmation with a modal dialog.
304 > */
305 > confirm(confirmation: IConfirmation): Promise<IConfirmationResult>;
306 >
307 > /**
308 > * Prompt the user with a modal dialog.
309 > */
310 > prompt<T>(prompt: IPrompt<T>): Promise<IAsyncPromptResult<T>>;
311 >
312 > /**
313 > * Present a modal dialog to the user asking for input.
314 > */
315 > input(input: IInput): Promise<IInputResult>;
316 >
317 > /**
318 > * Present the about dialog to the user.
319 > */
320 > about(title: string, details: string, detailsToCopy: string): Promise<void>;
321 > }
322 >
323 > enum DialogKind {
324 > Confirmation = 1,
325 > Prompt,
326 > Input
327 > }
328 >
329 > export abstract class AbstractDialogHandler implements IDialogHandler {
330 >
331 > protected getConfirmationButtons(dialog: IConfirmation): string[] {
332 return this.getButtons(dialog, DialogKind.Confirmation);
333 }
334 > dialogs.ts
335 > protected getPromptButtons(dialog: IPrompt<unknown>): string[] {
336 return this.getButtons(dialog, DialogKind.Prompt);
337 }
338 > dialogs.ts
339 > protected getInputButtons(dialog: IInput): string[] {
340 return this.getButtons(dialog, DialogKind.Input);
341 }
342 > dialogs.ts
343 > private getButtons(dialog: IConfirmation, kind: DialogKind.Confirmation): string[];
344 > private getButtons(dialog: IPrompt<unknown>, kind: DialogKind.Prompt): string[];
345 > private getButtons(dialog: IInput, kind: DialogKind.Input): string[];
346 > private getButtons(dialog: IConfirmation | IInput | IPrompt<unknown>, kind: DialogKind): string[] {
347
348 // We put buttons in the order of "default" button first and "cancel"
418 return buttons;
419 }
420 > dialogs.ts
421 > protected getDialogType(type: Severity | DialogType | undefined): DialogType | undefined {
422 if (typeof type === 'string') {
423 return type;
430 return undefined;
431 }
432 > dialogs.ts
433 > protected getPromptResult<T>(prompt: IPrompt<T>, buttonIndex: number, checkboxChecked: boolean | undefined): IAsyncPromptResult<T> {
434 const promptButtons: IPromptBaseButton<T>[] = [...(prompt.buttons ?? [])];
435 if (prompt.cancelButton && typeof prompt.cancelButton !== 'string' && typeof prompt.cancelButton !== 'boolean') {
444 return { result, checkboxChecked };
445 }
446 > dialogs.ts
447 > abstract confirm(confirmation: IConfirmation): Promise<IConfirmationResult>;
448 > abstract input(input: IInput): Promise<IInputResult>;
449 > abstract prompt<T>(prompt: IPrompt<T>): Promise<IAsyncPromptResult<T>>;
450 > abstract about(title: string, details: string, detailsToCopy: string): Promise<void>;
451 > }
452 >
453 > /**
454 > * A service to bring up modal dialogs.
455 > *
456 > * Note: use the `INotificationService.prompt()` method for a non-modal way to ask
457 > * the user for input.
458 > */
459 > export interface IDialogService {
460 >
461 > readonly _serviceBrand: undefined;
462 >
463 > /**
464 > * An event that fires when a dialog is about to show.
465 > */
466 > readonly onWillShowDialog: Event<void>;
467 >
468 > /**
469 > * An event that fires when a dialog did show (closed).
470 > */
471 > readonly onDidShowDialog: Event<void>;
472 >
473 > /**
474 > * Ask the user for confirmation with a modal dialog.
475 > */
476 > confirm(confirmation: IConfirmation): Promise<IConfirmationResult>;
477 >
478 > /**
479 > * Prompt the user with a modal dialog. Provides a bit
480 > * more control over the dialog compared to the simpler
481 > * `confirm` method. Specifically, allows to show more
482 > * than 2 buttons and makes it easier to just show a
483 > * message to the user.
484 > *
485 > * @returns a promise that resolves to the `T` result
486 > * from the provided `IPromptButton<T>` or `undefined`.
487 > */
488 > prompt<T>(prompt: IPromptWithCustomCancel<T>): Promise<IPromptResultWithCancel<T>>;
489 > prompt<T>(prompt: IPromptWithDefaultCancel<T>): Promise<IPromptResult<T>>;
490 > prompt<T>(prompt: IPrompt<T>): Promise<IPromptResult<T>>;
491 >
492 > /**
493 > * Present a modal dialog to the user asking for input.
494 > */
495 > input(input: IInput): Promise<IInputResult>;
496 >
497 > /**
498 > * Show a modal info dialog.
499 > */
500 > info(message: string, detail?: string): Promise<void>;
501 >
502 > /**
503 > * Show a modal warning dialog.
504 > */
505 > warn(message: string, detail?: string): Promise<void>;
506 >
507 > /**
508 > * Show a modal error dialog.
509 > */
510 > error(message: string, detail?: string): Promise<void>;
511 >
512 > /**
513 > * Present the about dialog to the user.
514 > */
515 > about(): Promise<void>;
516 > }
517 >
518 > export const IFileDialogService = createDecorator<IFileDialogService>('fileDialogService');
519 >
520 > /**
521 > * A service to bring up file dialogs.
522 > */
523 > export interface IFileDialogService {
524 >
525 > readonly _serviceBrand: undefined;
526 >
527 > /**
528 > * The default path for a new file based on previously used files.
529 > * @param schemeFilter The scheme of the file path. If no filter given, the scheme of the current window is used.
530 > * Falls back to user home in the absence of enough information to find a better URI.
531 > */
532 > defaultFilePath(schemeFilter?: string): Promise<URI>;
533 >
534 > /**
535 > * The default path for a new folder based on previously used folders.
536 > * @param schemeFilter The scheme of the folder path. If no filter given, the scheme of the current window is used.
537 > * Falls back to user home in the absence of enough information to find a better URI.
538 > */
539 > defaultFolderPath(schemeFilter?: string): Promise<URI>;
540 >
541 > /**
542 > * The default path for a new workspace based on previously used workspaces.
543 > * @param schemeFilter The scheme of the workspace path. If no filter given, the scheme of the current window is used.
544 > * Falls back to user home in the absence of enough information to find a better URI.
545 > */
546 > defaultWorkspacePath(schemeFilter?: string): Promise<URI>;
547 >
548 > /**
549 > * Shows a file-folder selection dialog and opens the selected entry.
550 > */
551 > pickFileFolderAndOpen(options: IPickAndOpenOptions): Promise<void>;
552 >
553 > /**
554 > * Shows a file selection dialog and opens the selected entry.
555 > */
556 > pickFileAndOpen(options: IPickAndOpenOptions): Promise<void>;
557 >
558 > /**
559 > * Shows a folder selection dialog and opens the selected entry.
560 > */
561 > pickFolderAndOpen(options: IPickAndOpenOptions): Promise<void>;
562 >
563 > /**
564 > * Shows a workspace selection dialog and opens the selected entry.
565 > */
566 > pickWorkspaceAndOpen(options: IPickAndOpenOptions): Promise<void>;
567 >
568 > /**
569 > * Shows a save file dialog and save the file at the chosen file URI.
570 > */
571 > pickFileToSave(defaultUri: URI, availableFileSystems?: string[]): Promise<URI | undefined>;
572 >
573 > /**
574 > * The preferred folder path to open the dialog at.
575 > * @param schemeFilter The scheme of the file path. If no filter given, the scheme of the current window is used.
576 > * Falls back to user home in the absence of a setting.
577 > */
578 > preferredHome(schemeFilter?: string): Promise<URI>;
579 >
580 > /**
581 > * Shows a save file dialog and returns the chosen file URI.
582 > */
583 > showSaveDialog(options: ISaveDialogOptions): Promise<URI | undefined>;
584 >
585 > /**
586 > * Shows a confirm dialog for saving 1-N files.
587 > */
588 > showSaveConfirm(fileNamesOrResources: (string | URI)[]): Promise<ConfirmResult>;
589 >
590 > /**
591 > * Shows a open file dialog and returns the chosen file URI.
592 > */
593 > showOpenDialog(options: IOpenDialogOptions): Promise<URI[] | undefined>;
594 > }
595 >
596 > export const enum ConfirmResult {
597 > SAVE,
598 > DONT_SAVE,
599 > CANCEL
600 > }
601 >
602 > const MAX_CONFIRM_FILES = 10;
603 > export function getFileNamesMessage(fileNamesOrResources: readonly (string | URI)[]): string {
604 const message: string[] = [];
605 message.push(...fileNamesOrResources.slice(0, MAX_CONFIRM_FILES).map(fileNameOrResource => typeof fileNameOrResource === 'string' ? fileNameOrResource : basename(fileNameOrResource)));
616 return message.join('\n');
617 }
618 > dialogs.ts
619 > export interface INativeOpenDialogOptions {
620 > readonly forceNewWindow?: boolean;
621 >
622 > readonly defaultPath?: string;
623 >
624 > readonly telemetryEventName?: string;
625 > readonly telemetryExtraData?: ITelemetryData;
626 > }
src/vs/workbench/services/authentication/common/authentication.ts 507 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- authentication.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { Event } from '../../../../base/common/event.js';
6 > import { IDisposable } from '../../../../base/common/lifecycle.js';
7 > import { IAuthenticationChallenge, IAuthorizationProtectedResourceMetadata, IAuthorizationServerMetadata } from '../../../../base/common/oauth.js';
8 > import { URI } from '../../../../base/common/uri.js';
9 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
10 >
11 > /**
12 > * Use this if you don't want the onDidChangeSessions event to fire in the extension host
13 > */
14 > export const INTERNAL_AUTH_PROVIDER_PREFIX = '__';
15 >
16 > export interface AuthenticationSessionAccount {
17 > label: string;
18 > id: string;
19 > }
20 >
21 > export interface AuthenticationSession {
22 > id: string;
23 > accessToken: string;
24 > account: AuthenticationSessionAccount;
25 > scopes: ReadonlyArray<string>;
26 > idToken?: string;
27 > }
28 >
29 > export interface AuthenticationSessionsChangeEvent {
30 > added: ReadonlyArray<AuthenticationSession> | undefined;
31 > removed: ReadonlyArray<AuthenticationSession> | undefined;
32 > changed: ReadonlyArray<AuthenticationSession> | undefined;
33 > }
34 >
35 > export interface AuthenticationProviderInformation {
36 > id: string;
37 > label: string;
38 > authorizationServerGlobs?: ReadonlyArray<string>;
39 > }
40 >
41 > /**
42 > * Options for creating an authentication session via the service.
43 > */
44 > export interface IAuthenticationCreateSessionOptions {
45 > activateImmediate?: boolean;
46 > /**
47 > * The account that is being asked about. If this is passed in, the provider should
48 > * attempt to return the sessions that are only related to this account.
49 > */
50 > account?: AuthenticationSessionAccount;
51 > /**
52 > * The authorization server URI to use for this creation request. If passed in, first we validate that
53 > * the provider can use this authorization server, then it is passed down to the auth provider.
54 > */
55 > authorizationServer?: URI;
56 > /**
57 > * When specified, the authentication provider will request a token bound to this resource URI
58 > * (RFC 8707 resource indicator).
59 > */
60 > resource?: string;
61 > /**
62 > * The audience for the requested access token. Primarily used for OAuth Identity Assertion
63 > * Authorization Grant (ID-JAG, defined in `draft-ietf-oauth-identity-assertion-authz-grant` using RFC 8693 token-exchange semantics) flows where the audience identifies the authorization server of the resource that
64 > * will redeem the assertion (typically the resource's authorization server URL). Providers that do not understand audience-bound tokens should
65 > * ignore this option.
66 > */
67 > audience?: string;
68 > /**
69 > * Allows the authentication provider to take in additional parameters.
70 > * It is up to the provider to define what these parameters are and handle them.
71 > * This is useful for passing in additional information that is specific to the provider
72 > * and not part of the standard authentication flow.
73 > */
74 > [key: string]: any;
75 > }
76 >
77 > export interface IAuthenticationWwwAuthenticateRequest {
78 > /**
79 > * The raw WWW-Authenticate header value that triggered this challenge.
80 > * This will be parsed by the authentication provider to extract the necessary
81 > * challenge information.
82 > */
83 > readonly wwwAuthenticate: string;
84 >
85 > /**
86 > * Optional scopes for the session. If not provided, the authentication provider
87 > * may use default scopes or extract them from the challenge.
88 > */
89 > readonly fallbackScopes?: readonly string[];
90 > }
91 >
92 > export function isAuthenticationWwwAuthenticateRequest(obj: unknown): obj is IAuthenticationWwwAuthenticateRequest {
93 return typeof obj === 'object'
94 && obj !== null
96 && (typeof obj.wwwAuthenticate === 'string');
97 }
99 > /**
100 > * Represents constraints for authentication, including challenges and optional scopes.
101 > * This is used when creating or retrieving sessions that must satisfy specific authentication
102 > * requirements from WWW-Authenticate headers.
103 > */
104 > export interface IAuthenticationConstraint {
105 > /**
106 > * Array of authentication challenges parsed from WWW-Authenticate headers.
107 > */
108 > readonly challenges: readonly IAuthenticationChallenge[];
109 >
110 > /**
111 > * Optional scopes for the session. If not provided, the authentication provider
112 > * may extract scopes from the challenges or use default scopes.
113 > */
114 > readonly fallbackScopes?: readonly string[];
115 > }
116 >
117 > /**
118 > * Options for getting authentication sessions via the service.
119 > */
120 > export interface IAuthenticationGetSessionsOptions {
121 > /**
122 > * Whether the provider must avoid user interaction while resolving existing sessions.
123 > */
124 > silent?: boolean;
125 > /**
126 > * The account that is being asked about. If this is passed in, the provider should
127 > * attempt to return the sessions that are only related to this account.
128 > */
129 > account?: AuthenticationSessionAccount;
130 > /**
131 > * The authorization server URI to use for this request. If passed in, first we validate that
132 > * the provider can use this authorization server, then it is passed down to the auth provider.
133 > */
134 > authorizationServer?: URI;
135 > /**
136 > * When specified, the authentication provider will request a token bound to this resource URI
137 > * (RFC 8707 resource indicator).
138 > */
139 > resource?: string;
140 > /**
141 > * The audience for the requested access token. Primarily used for OAuth Identity Assertion
142 > * Authorization Grant (ID-JAG, defined in `draft-ietf-oauth-identity-assertion-authz-grant` using RFC 8693 token-exchange semantics) flows where the audience identifies the authorization server of the resource that
143 > * will redeem the assertion (typically the resource's authorization server URL). Providers that do not understand audience-bound tokens should
144 > * ignore this option.
145 > */
146 > audience?: string;
147 > /**
148 > * Allows the authentication provider to take in additional parameters.
149 > * It is up to the provider to define what these parameters are and handle them.
150 > * This is useful for passing in additional information that is specific to the provider
151 > * and not part of the standard authentication flow.
152 > */
153 > [key: string]: any;
154 > }
155 >
156 > export interface AllowedExtension {
157 > id: string;
158 > name: string;
159 > /**
160 > * If true or undefined, the extension is allowed to use the account
161 > * If false, the extension is not allowed to use the account
162 > * TODO: undefined shouldn't be a valid value, but it is for now
163 > */
164 > allowed?: boolean;
165 > lastUsed?: number;
166 > // If true, this comes from the product.json
167 > trusted?: boolean;
168 > }
169 >
170 > export interface IAuthenticationProviderHostDelegate {
171 > /** Priority for this delegate, delegates are tested in descending priority order */
172 > readonly priority: number;
173 > create(authorizationServer: URI, serverMetadata: IAuthorizationServerMetadata, resource: IAuthorizationProtectedResourceMetadata | undefined, clientId?: string, clientSecret?: string): Promise<string>;
174 > /**
175 > * Creates an XAA (enterprise-managed, ID-JAG) authentication provider for the given SSO issuer.
176 > * The returned string is the provider id.
177 > */
178 > createXaa?(issuer: URI): Promise<string>;
179 > }
180 >
181 > export function getDynamicAuthenticationProviderId(authorizationServer: URI, resource: IAuthorizationProtectedResourceMetadata | undefined): string {
182 return resource ? `${authorizationServer.toString(true)} ${resource.resource}` : authorizationServer.toString(true);
183 }
185 > export const IAuthenticationService = createDecorator<IAuthenticationService>('IAuthenticationService');
186 >
187 > export interface IAuthenticationService {
188 > readonly _serviceBrand: undefined;
189 >
190 > /**
191 > * Fires when an authentication provider has been registered
192 > */
193 > readonly onDidRegisterAuthenticationProvider: Event<AuthenticationProviderInformation>;
194 > /**
195 > * Fires when an authentication provider has been unregistered
196 > */
197 > readonly onDidUnregisterAuthenticationProvider: Event<AuthenticationProviderInformation>;
198 >
199 > /**
200 > * Fires when the list of sessions for a provider has been added, removed or changed
201 > */
202 > readonly onDidChangeSessions: Event<{ providerId: string; label: string; event: AuthenticationSessionsChangeEvent }>;
203 >
204 > /**
205 > * Fires when the list of declaredProviders has changed
206 > */
207 > readonly onDidChangeDeclaredProviders: Event<void>;
208 >
209 > /**
210 > * All providers that have been statically declared by extensions. These may not actually be registered or active yet.
211 > */
212 > readonly declaredProviders: AuthenticationProviderInformation[];
213 >
214 > /**
215 > * Registers that an extension has declared an authentication provider in their package.json
216 > * @param provider The provider information to register
217 > */
218 > registerDeclaredAuthenticationProvider(provider: AuthenticationProviderInformation): void;
219 >
220 > /**
221 > * Unregisters a declared authentication provider
222 > * @param id The id of the provider to unregister
223 > */
224 > unregisterDeclaredAuthenticationProvider(id: string): void;
225 >
226 > /**
227 > * Checks if an authentication provider has been registered
228 > * @param id The id of the provider to check
229 > */
230 > isAuthenticationProviderRegistered(id: string): boolean;
231 >
232 > /**
233 > * Checks if an authentication provider is dynamic
234 > * @param id The id of the provider to check
235 > */
236 > isDynamicAuthenticationProvider(id: string): boolean;
237 >
238 > /**
239 > * Registers an authentication provider
240 > * @param id The id of the provider
241 > * @param provider The implementation of the provider
242 > */
243 > registerAuthenticationProvider(id: string, provider: IAuthenticationProvider): void;
244 >
245 > /**
246 > * Unregisters an authentication provider
247 > * @param id The id of the provider to unregister
248 > */
249 > unregisterAuthenticationProvider(id: string): void;
250 >
251 > /**
252 > * Gets the provider ids of all registered authentication providers
253 > */
254 > getProviderIds(): string[];
255 >
256 > /**
257 > * Gets the provider with the given id.
258 > * @param id The id of the provider to get
259 > * @throws if the provider is not registered
260 > */
261 > getProvider(id: string): IAuthenticationProvider;
262 >
263 > /**
264 > * Gets all accounts that are currently logged in across all sessions
265 > * @param id The id of the provider to ask for accounts
266 > * @returns A promise that resolves to an array of accounts
267 > */
268 > getAccounts(id: string): Promise<ReadonlyArray<AuthenticationSessionAccount>>;
269 >
270 > /**
271 > * Gets all sessions that satisfy the given scopes from the provider with the given id
272 > * @param id The id of the provider to ask for a session
273 > * @param scopes The scopes for the session
274 > * @param options Additional options for getting sessions
275 > * @param activateImmediate If true, the provider should activate immediately if it is not already
276 > */
277 > getSessions(id: string, scopeListOrRequest?: ReadonlyArray<string> | IAuthenticationWwwAuthenticateRequest, options?: IAuthenticationGetSessionsOptions, activateImmediate?: boolean): Promise<ReadonlyArray<AuthenticationSession>>;
278 >
279 > /**
280 > * Creates an AuthenticationSession with the given provider and scopes
281 > * @param providerId The id of the provider
282 > * @param scopes The scopes to request
283 > * @param options Additional options for creating the session
284 > */
285 > createSession(providerId: string, scopeListOrRequest: ReadonlyArray<string> | IAuthenticationWwwAuthenticateRequest, options?: IAuthenticationCreateSessionOptions): Promise<AuthenticationSession>;
286 >
287 > /**
288 > * Removes the session with the given id from the provider with the given id
289 > * @param providerId The id of the provider
290 > * @param sessionId The id of the session to remove
291 > */
292 > removeSession(providerId: string, sessionId: string): Promise<void>;
293 >
294 > /**
295 > * Gets a provider id for a specified authorization server
296 > * @param authorizationServer The authorization server url that this provider is responsible for
297 > * @param resourceServer The resource server URI that should match the provider's resourceServer (if defined)
298 > */
299 > getOrActivateProviderIdForServer(authorizationServer: URI, resourceServer?: URI): Promise<string | undefined>;
300 >
301 > /**
302 > * Allows the ability register a delegate that will be used to start authentication providers
303 > * @param delegate The delegate to register
304 > */
305 > registerAuthenticationProviderHostDelegate(delegate: IAuthenticationProviderHostDelegate): IDisposable;
306 >
307 > /**
308 > * Creates a dynamic authentication provider for the given server metadata
309 > * @param serverMetadata The metadata for the server that is being authenticated against
310 > */
311 > createDynamicAuthenticationProvider(authorizationServer: URI, serverMetadata: IAuthorizationServerMetadata, resourceMetadata: IAuthorizationProtectedResourceMetadata | undefined, clientId?: string, clientSecret?: string): Promise<IAuthenticationProvider | undefined>;
312 >
313 > /**
314 > * Gets or creates a built-in XAA (enterprise-managed, ID-JAG) authentication provider for the given
315 > * SSO issuer. Subsequent calls with the same issuer return the existing provider. The returned id
316 > * can be used with {@link getSessions}/{@link createSession} just like any other provider.
317 > *
318 > * @param issuer The OAuth/OIDC issuer URL (typically read from `mcp.enterpriseManagedAuth.idp`).
319 > */
320 > createOrGetXaaProvider(issuer: URI): Promise<string | undefined>;
321 > }
322 >
323 > export function isAuthenticationSession(thing: unknown): thing is AuthenticationSession {
324 if (typeof thing !== 'object' || !thing) {
325 return false;
349 return true;
350 }
352 > // TODO: Move this into MainThreadAuthentication
353 > export const IAuthenticationExtensionsService = createDecorator<IAuthenticationExtensionsService>('IAuthenticationExtensionsService');
354 > export interface IAuthenticationExtensionsService {
355 > readonly _serviceBrand: undefined;
356 >
357 > /**
358 > * Fires when an account preference for a specific provider has changed for the specified extensions. Does not fire when:
359 > * * An account preference is removed
360 > * * A session preference is changed (because it's deprecated)
361 > * * A session preference is removed (because it's deprecated)
362 > */
363 > readonly onDidChangeAccountPreference: Event<{ extensionIds: string[]; providerId: string }>;
364 > /**
365 > * Returns the accountName (also known as account.label) to pair with `IAuthenticationAccessService` to get the account preference
366 > * @param providerId The authentication provider id
367 > * @param extensionId The extension id to get the preference for
368 > * @returns The accountName of the preference, or undefined if there is no preference set
369 > */
370 > getAccountPreference(extensionId: string, providerId: string): string | undefined;
371 > /**
372 > * Sets the account preference for the given provider and extension
373 > * @param providerId The authentication provider id
374 > * @param extensionId The extension id to set the preference for
375 > * @param account The account to set the preference to
376 > */
377 > updateAccountPreference(extensionId: string, providerId: string, account: AuthenticationSessionAccount): void;
378 > /**
379 > * Removes the account preference for the given provider and extension
380 > * @param providerId The authentication provider id
381 > * @param extensionId The extension id to remove the preference for
382 > */
383 > removeAccountPreference(extensionId: string, providerId: string): void;
384 > /**
385 > * @deprecated Sets the session preference for the given provider and extension
386 > * @param providerId
387 > * @param extensionId
388 > * @param session
389 > */
390 > updateSessionPreference(providerId: string, extensionId: string, session: AuthenticationSession): void;
391 > /**
392 > * @deprecated Gets the session preference for the given provider and extension
393 > * @param providerId
394 > * @param extensionId
395 > * @param scopes
396 > */
397 > getSessionPreference(providerId: string, extensionId: string, scopes: string[]): string | undefined;
398 > /**
399 > * @deprecated Removes the session preference for the given provider and extension
400 > * @param providerId
401 > * @param extensionId
402 > * @param scopes
403 > */
404 > removeSessionPreference(providerId: string, extensionId: string, scopes: string[]): void;
405 > selectSession(providerId: string, extensionId: string, extensionName: string, scopeListOrRequest: ReadonlyArray<string> | IAuthenticationWwwAuthenticateRequest, possibleSessions: readonly AuthenticationSession[]): Promise<AuthenticationSession>;
406 > requestSessionAccess(providerId: string, extensionId: string, extensionName: string, scopeListOrRequest: ReadonlyArray<string> | IAuthenticationWwwAuthenticateRequest, possibleSessions: readonly AuthenticationSession[]): void;
407 > requestNewSession(providerId: string, scopeListOrRequest: ReadonlyArray<string> | IAuthenticationWwwAuthenticateRequest, extensionId: string, extensionName: string): Promise<void>;
408 > updateNewSessionRequests(providerId: string, addedSessions: readonly AuthenticationSession[]): void;
409 > }
410 >
411 > /**
412 > * Options passed to the authentication provider when asking for sessions.
413 > */
414 > export interface IAuthenticationProviderSessionOptions {
415 > /**
416 > * Whether the provider must avoid user interaction while resolving existing sessions.
417 > */
418 > silent?: boolean;
419 > /**
420 > * The account that is being asked about. If this is passed in, the provider should
421 > * attempt to return the sessions that are only related to this account.
422 > */
423 > account?: AuthenticationSessionAccount;
424 > /**
425 > * The authorization server that is being asked about. If this is passed in, the provider should
426 > * attempt to return sessions that are only related to this authorization server.
427 > */
428 > authorizationServer?: URI;
429 > /**
430 > * When specified, the authentication provider will request a token bound to this resource URI
431 > * (RFC 8707 resource indicator).
432 > */
433 > resource?: string;
434 > /**
435 > * The audience for the requested access token. Primarily used for OAuth Identity Assertion
436 > * Authorization Grant (ID-JAG, defined in `draft-ietf-oauth-identity-assertion-authz-grant` using RFC 8693 token-exchange semantics) flows where the audience identifies the authorization server of the resource that
437 > * will redeem the assertion (typically the resource's authorization server URL). Providers that do not understand audience-bound tokens should
438 > * ignore this option.
439 > */
440 > audience?: string;
441 > /**
442 > * Allows the authentication provider to take in additional parameters.
443 > * It is up to the provider to define what these parameters are and handle them.
444 > * This is useful for passing in additional information that is specific to the provider
445 > * and not part of the standard authentication flow.
446 > */
447 > [key: string]: any;
448 > }
449 >
450 > /**
451 > * Represents an authentication provider.
452 > */
453 > export interface IAuthenticationProvider {
454 > /**
455 > * The unique identifier of the authentication provider.
456 > */
457 > readonly id: string;
458 >
459 > /**
460 > * The display label of the authentication provider.
461 > */
462 > readonly label: string;
463 >
464 > /**
465 > * The resource server URI that this provider is responsible for, if any.
466 > * TODO@TylerLeonhardt: Rather than this being added to the provider, it should be passed in to
467 > * getSessions/createSession/etc... this way we can have providers that handle multiple resource servers.
468 > */
469 > readonly resourceServer?: URI;
470 >
471 > /**
472 > * The resolved authorization servers. These can still contain globs, but should be concrete URIs
473 > */
474 > readonly authorizationServers?: ReadonlyArray<URI>;
475 >
476 > /**
477 > * Indicates whether the authentication provider supports multiple accounts.
478 > */
479 > readonly supportsMultipleAccounts: boolean;
480 >
481 > /**
482 > * Optional function to provide a custom confirmation message for authentication prompts.
483 > * If not implemented, the default confirmation messages will be used.
484 > * @param extensionName - The name of the extension requesting authentication.
485 > * @param recreatingSession - Whether this is recreating an existing session.
486 > * @returns A custom confirmation message or undefined to use the default message.
487 > */
488 > readonly confirmation?: (extensionName: string, recreatingSession: boolean) => string | undefined;
489 >
490 > /**
491 > * An {@link Event} which fires when the array of sessions has changed, or data
492 > * within a session has changed.
493 > */
494 > readonly onDidChangeSessions: Event<AuthenticationSessionsChangeEvent>;
495 >
496 > /**
497 > * Retrieves a list of authentication sessions.
498 > * @param scopes - An optional list of scopes. If provided, the sessions returned should match these permissions, otherwise all sessions should be returned.
499 > * @param options - Additional options for getting sessions.
500 > * @returns A promise that resolves to an array of authentication sessions.
501 > */
502 > getSessions(scopes: string[] | undefined, options: IAuthenticationProviderSessionOptions): Promise<readonly AuthenticationSession[]>;
503 >
504 > /**
505 > * Prompts the user to log in.
506 > * If login is successful, the `onDidChangeSessions` event should be fired.
507 > * If login fails, a rejected promise should be returned.
508 > * If the provider does not support multiple accounts, this method should not be called if there is already an existing session matching the provided scopes.
509 > * @param scopes - A list of scopes that the new session should be created with.
510 > * @param options - Additional options for creating the session.
511 > * @returns A promise that resolves to an authentication session.
512 > */
513 > createSession(scopes: string[], options: IAuthenticationProviderSessionOptions): Promise<AuthenticationSession>;
514 >
515 > /**
516 > * Get existing sessions that match the given authentication constraints.
517 > *
518 > * @param constraint The authentication constraint containing challenges and optional scopes
519 > * @param options Options for the session request
520 > * @returns A thenable that resolves to an array of existing authentication sessions
521 > */
522 > getSessionsFromChallenges?(constraint: IAuthenticationConstraint, options: IAuthenticationProviderSessionOptions): Promise<readonly AuthenticationSession[]>;
523 >
524 > /**
525 > * Create a new session based on authentication constraints.
526 > * This is called when no existing session matches the constraint requirements.
527 > *
528 > * @param constraint The authentication constraint containing challenges and optional scopes
529 > * @param options Options for the session creation
530 > * @returns A thenable that resolves to a new authentication session
531 > */
532 > createSessionFromChallenges?(constraint: IAuthenticationConstraint, options: IAuthenticationProviderSessionOptions): Promise<AuthenticationSession>;
533 >
534 > /**
535 > * Removes the session corresponding to the specified session ID.
536 > * If the removal is successful, the `onDidChangeSessions` event should be fired.
537 > * If a session cannot be removed, the provider should reject with an error message.
538 > * @param sessionId - The ID of the session to remove.
539 > */
540 > removeSession(sessionId: string): Promise<void>;
541 > }
src/vs/workbench/contrib/chat/common/attachments/chatVariableEntries.ts 506 covered LOC · 54 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatVariableEntries.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Codicon } from '../../../../../base/common/codicons.js';
7 > import { IMarkdownString } from '../../../../../base/common/htmlContent.js';
8 > import { basename } from '../../../../../base/common/resources.js';
9 > import { ThemeIcon } from '../../../../../base/common/themables.js';
10 > import { URI } from '../../../../../base/common/uri.js';
11 > import { generateUuid } from '../../../../../base/common/uuid.js';
12 > import { IRange } from '../../../../../editor/common/core/range.js';
13 > import { IOffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
14 > import { isLocation, Location, SymbolKind } from '../../../../../editor/common/languages.js';
15 > import { localize } from '../../../../../nls.js';
16 > import { MarkerSeverity, IMarker } from '../../../../../platform/markers/common/markers.js';
17 > import { ISCMHistoryItem } from '../../../scm/common/history.js';
18 > import { IChatContentReference } from '../chatService/chatService.js';
19 > import { IChatRequestVariableValue } from './chatVariables.js';
20 > import { IToolData, IToolSet } from '../tools/languageModelToolsService.js';
21 > import type { ILanguageModelChatMetadata } from '../languageModels.js';
22 > import { decodeBase64, encodeBase64, VSBuffer } from '../../../../../base/common/buffer.js';
23 > import { Mutable } from '../../../../../base/common/types.js';
24 >
25 >
26 > /**
27 > * An icon for a chat context item. Mirrors the `IconPath` type from the extension API:
28 > * either a {@link ThemeIcon theme icon}, a single {@link URI} or separate light/dark {@link URI uris}.
29 > */
30 > export type ChatContextIconPath = ThemeIcon | URI | { light: URI; dark: URI };
31 >
32 > /**
33 > * Type guard for {@link ChatContextIconPath}. Accepts a {@link ThemeIcon theme icon}, a single
34 > * {@link URI} or an object with both `light` and `dark` {@link URI uris}. Rejects `null`, `undefined`
35 > * and partially-specified light/dark objects.
36 > */
37 > export function isChatContextIconPath(value: unknown): value is ChatContextIconPath {
38 if (!value || typeof value !== 'object') {
39 return false;
45 return URI.isUri(asDualPath.light) && URI.isUri(asDualPath.dark);
46 }
48 > /**
49 > * Resolve a {@link ChatContextIconPath} into a value that can be passed to the `iconPath`
50 > * option of an icon label, picking the light or dark uri based on the current theme.
51 > *
52 > * @param iconPath The icon path to resolve.
53 > * @param useDark Whether the current theme is a dark theme.
54 > */
55 > export function resolveChatContextIcon(iconPath: ChatContextIconPath, useDark: boolean): ThemeIcon | URI {
56 if (ThemeIcon.isThemeIcon(iconPath) || URI.isUri(iconPath)) {
57 return iconPath;
59 return useDark ? iconPath.dark : iconPath.light;
60 }
62 > interface IBaseChatRequestVariableEntry {
63 > readonly id: string;
64 > readonly fullName?: string;
65 > readonly icon?: ThemeIcon;
66 > readonly name: string;
67 > readonly modelDescription?: string;
68 >
69 > /**
70 > * The offset-range in the prompt. This means this entry has been explicitly typed out
71 > * by the user.
72 > */
73 > readonly range?: IOffsetRange;
74 > readonly value: IChatRequestVariableValue;
75 > readonly references?: IChatContentReference[];
76 >
77 > /**
78 > * Implementation-defined metadata that providers attach to a variable
79 > * entry. Used to round-trip provider-specific data (e.g. agent-host
80 > * `_meta`) when an entry is sent back to the provider as part of a
81 > * request attachment.
82 > */
83 > readonly _meta?: Record<string, unknown>;
84 >
85 > omittedState?: OmittedState;
86 > }
87 >
88 > export interface IGenericChatRequestVariableEntry extends IBaseChatRequestVariableEntry {
89 > kind: 'generic';
90 > tooltip?: IMarkdownString;
91 > /**
92 > * A provider-supplied icon that may be a {@link ThemeIcon theme icon}, a single uri or light/dark uris.
93 > * Takes precedence over the {@link IBaseChatRequestVariableEntry.icon base theme icon} when rendering.
94 > */
95 > iconPath?: ChatContextIconPath;
96 > }
97 >
98 > export const ChatPasteAttachmentMetadata = {
99 > Kind: 'vscode.chat.attachment.kind',
100 > Language: 'vscode.chat.attachment.language',
101 > FileName: 'vscode.chat.attachment.fileName',
102 > PastedLines: 'vscode.chat.attachment.pastedLines',
103 > } as const;
104 >
105 > export interface IRestorablePasteAttachment {
106 > readonly label: string;
107 > readonly displayKind?: string;
108 > readonly modelRepresentation?: string;
109 > readonly _meta?: Record<string, unknown>;
110 > }
111 >
112 > export const enum AgentHostCompletionReferenceKind {
113 > Skill = 'skill',
114 > Command = 'command',
115 > }
116 >
117 > export interface IAgentHostCompletionVariableValue {
118 > readonly $mid: 'agentHostCompletion';
119 > readonly kind: AgentHostCompletionReferenceKind;
120 > }
121 >
122 function agentHostCompletionVariableValue(kind: AgentHostCompletionReferenceKind): IAgentHostCompletionVariableValue {
123 return { $mid: 'agentHostCompletion', kind };
124 }
126 function agentHostCompletionVariableId(kind: AgentHostCompletionReferenceKind, reference: URI | string): string {
127 switch (kind) {
132 }
133 }
135 > export function toAgentHostCompletionVariableEntry(kind: AgentHostCompletionReferenceKind, name: string, reference: URI | string | undefined, _meta: Record<string, unknown> | undefined): IGenericChatRequestVariableEntry & { value: IAgentHostCompletionVariableValue } {
136 return {
137 kind: 'generic',
142 };
143 }
145 > export function toAgentHostCompletionVariableEntryFromMetadata(kind: AgentHostCompletionReferenceKind, name: string, _meta: Record<string, unknown> | undefined): IGenericChatRequestVariableEntry & { value: IAgentHostCompletionVariableValue } {
146 switch (kind) {
147 case AgentHostCompletionReferenceKind.Skill:
151 }
152 }
154 > export function getAgentHostCompletionReferenceKind(entry: IChatRequestVariableEntry): AgentHostCompletionReferenceKind | undefined {
155 if (entry.kind !== 'generic') {
156 return undefined;
158 return getAgentHostCompletionReferenceKindFromValue(entry.value);
159 }
161 > export function getAgentHostCompletionReferenceKindFromValue(value: IChatRequestVariableValue): AgentHostCompletionReferenceKind | undefined {
162 if (typeof value !== 'object' || value === null) {
163 return undefined;
176 return undefined;
177 }
179 > export function isAgentHostCompletionVariableEntry(entry: IChatRequestVariableEntry): entry is IGenericChatRequestVariableEntry & { value: IAgentHostCompletionVariableValue } {
180 return getAgentHostCompletionReferenceKind(entry) !== undefined;
181 }
183 >
184 > export interface IChatRequestDirectoryEntry extends IBaseChatRequestVariableEntry {
185 > kind: 'directory';
186 > imageCount?: number;
187 > }
188 >
189 > export interface IChatRequestFileEntry extends IBaseChatRequestVariableEntry {
190 > kind: 'file';
191 > }
192 >
193 > export const enum OmittedState {
194 > NotOmitted,
195 > Partial,
196 > Full,
197 > ImageLimitExceeded,
198 > }
199 >
200 > const CLAUDE_MESSAGES_MAX_IMAGES_PER_REQUEST = 20;
201 > const GEMINI_MAX_IMAGES_PER_REQUEST = 10;
202 >
203 > /**
204 > * Returns the image-attachment limit for the selected model.
205 > *
206 > * Claude-family models use a max of 20 (Messages API), Gemini-family models use
207 > * a max of 10. Other models do not have a UI-enforced image count limit.
208 > */
209 > export function getImageAttachmentLimit(model: Pick<ILanguageModelChatMetadata, 'family'> | undefined): number | undefined {
210 if (!model) {
211 return undefined;
223 return undefined;
224 }
226 > export interface IChatRequestToolEntry extends IBaseChatRequestVariableEntry {
227 > readonly kind: 'tool';
228 > }
229 >
230 > export interface IChatRequestToolSetEntry extends IBaseChatRequestVariableEntry {
231 > readonly kind: 'toolset';
232 > readonly value: IChatRequestToolEntry[];
233 > }
234 >
235 > export type ChatRequestToolReferenceEntry = IChatRequestToolEntry | IChatRequestToolSetEntry;
236 >
237 > export interface StringChatContextValue {
238 > value?: string;
239 > name?: string;
240 > modelDescription?: string;
241 > iconPath?: ChatContextIconPath;
242 > uri: URI;
243 > resourceUri?: URI;
244 > tooltip?: IMarkdownString;
245 > /**
246 > * Command ID to execute when this context item is clicked.
247 > */
248 > readonly commandId?: string;
249 > readonly handle: number;
250 > }
251 >
252 > export interface IChatRequestImplicitVariableEntry extends IBaseChatRequestVariableEntry {
253 > readonly kind: 'implicit';
254 > readonly isFile: true;
255 > readonly value: URI | Location | StringChatContextValue | undefined;
256 > readonly uri: URI | undefined;
257 > readonly isSelection: boolean;
258 > enabled: boolean;
259 > }
260 >
261 > export interface IChatRequestStringVariableEntry extends IBaseChatRequestVariableEntry {
262 > readonly kind: 'string';
263 > readonly value: string | undefined;
264 > readonly modelDescription?: string;
265 > readonly iconPath?: ChatContextIconPath;
266 > readonly uri: URI;
267 > readonly resourceUri?: URI;
268 > readonly tooltip?: IMarkdownString;
269 > /**
270 > * Command ID to execute when this context item is clicked.
271 > */
272 > readonly commandId?: string;
273 > readonly handle: number;
274 > }
275 >
276 > export interface IChatRequestWorkspaceVariableEntry extends IBaseChatRequestVariableEntry {
277 > readonly kind: 'workspace';
278 > readonly value: string;
279 > readonly modelDescription?: string;
280 > }
281 >
282 >
283 > export interface IChatRequestPasteVariableEntry extends IBaseChatRequestVariableEntry {
284 > readonly kind: 'paste';
285 > readonly code: string;
286 > readonly language: string;
287 > readonly pastedLines: string;
288 >
289 > // This is only used for old serialized data and should be removed once we no longer support it
290 > readonly fileName: string;
291 >
292 > // This is only undefined on old serialized data
293 > readonly copiedFrom: {
294 > readonly uri: URI;
295 > readonly range: IRange;
296 > } | undefined;
297 > }
298 >
299 > export function toPasteVariableEntry(
300 name: string,
301 code: string,
332 };
333 }
335 > export function restorePasteVariableEntryFromAttachment(attachment: IRestorablePasteAttachment): IChatRequestPasteVariableEntry | undefined {
336 const modelRepresentation = attachment.modelRepresentation;
337 if (typeof modelRepresentation !== 'string' || attachment._meta?.[ChatPasteAttachmentMetadata.Kind] !== 'paste') {
350 });
351 }
353 > export interface ISymbolVariableEntry extends IBaseChatRequestVariableEntry {
354 > readonly kind: 'symbol';
355 > readonly value: Location;
356 > readonly symbolKind: SymbolKind;
357 > }
358 >
359 > export interface ICommandResultVariableEntry extends IBaseChatRequestVariableEntry {
360 > readonly kind: 'command';
361 > }
362 >
363 > export interface IImageVariableEntry extends IBaseChatRequestVariableEntry {
364 > readonly kind: 'image';
365 > readonly isPasted?: boolean;
366 > readonly isURL?: boolean;
367 > readonly mimeType?: string;
368 > }
369 >
370 > export interface INotebookOutputVariableEntry extends IBaseChatRequestVariableEntry {
371 > readonly kind: 'notebookOutput';
372 > readonly outputIndex?: number;
373 > readonly mimeType?: string;
374 > }
375 >
376 > export interface IDiagnosticVariableEntryFilterData {
377 > readonly owner?: string;
378 > readonly problemMessage?: string;
379 > readonly filterUri?: URI;
380 > readonly filterSeverity?: MarkerSeverity;
381 > readonly filterRange?: IRange;
382 > }
383 >
384 >
385 >
386 > export namespace IDiagnosticVariableEntryFilterData {
387 > export const icon = Codicon.error;
388 >
389 > export function fromMarker(marker: IMarker): IDiagnosticVariableEntryFilterData {
390 return {
391 filterUri: marker.resource,
395 };
396 }
398 > export function toEntry(data: IDiagnosticVariableEntryFilterData): IDiagnosticVariableEntry {
399 return {
400 id: id(data),
406 };
407 }
409 > export function id(data: IDiagnosticVariableEntryFilterData) {
410 return [data.filterUri, data.owner, data.filterSeverity, data.filterRange?.startLineNumber, data.filterRange?.startColumn].join(':');
411 }
413 > export function label(data: IDiagnosticVariableEntryFilterData) {
414 const enum TrimThreshold {
415 MaxChars = 30,
436 return labelStr;
437 }
439 >
440 > export interface IDiagnosticVariableEntry extends IBaseChatRequestVariableEntry, IDiagnosticVariableEntryFilterData {
441 > readonly kind: 'diagnostic';
442 > }
443 >
444 > export interface IElementAncestorData {
445 > readonly tagName: string;
446 > readonly id?: string;
447 > readonly classNames?: string[];
448 > }
449 >
450 > export interface IElementVariableEntry extends IBaseChatRequestVariableEntry {
451 > readonly kind: 'element';
452 > readonly value: string;
453 > readonly ancestors?: IElementAncestorData[];
454 > readonly attributes?: Record<string, string>;
455 > readonly computedStyles?: Record<string, string>;
456 > readonly dimensions?: { readonly top: number; readonly left: number; readonly width: number; readonly height: number };
457 > readonly innerText?: string;
458 > }
459 >
460 > export interface IPromptFileVariableEntry extends IBaseChatRequestVariableEntry {
461 > readonly kind: 'promptFile';
462 > readonly value: URI;
463 > readonly isRoot: boolean;
464 > readonly originLabel?: string;
465 > readonly modelDescription: string;
466 > readonly automaticallyAdded: boolean;
467 > readonly toolReferences?: readonly ChatRequestToolReferenceEntry[];
468 > }
469 >
470 > export interface IPromptTextVariableEntry extends IBaseChatRequestVariableEntry {
471 > readonly kind: 'promptText';
472 > readonly value: string;
473 > readonly settingId?: string;
474 > readonly modelDescription: string;
475 > readonly automaticallyAdded: boolean;
476 > readonly toolReferences?: readonly ChatRequestToolReferenceEntry[];
477 > }
478 >
479 > export interface ISCMHistoryItemVariableEntry extends IBaseChatRequestVariableEntry {
480 > readonly kind: 'scmHistoryItem';
481 > readonly value: URI;
482 > readonly historyItem: ISCMHistoryItem;
483 > }
484 >
485 > export interface ISCMHistoryItemChangeVariableEntry extends IBaseChatRequestVariableEntry {
486 > readonly kind: 'scmHistoryItemChange';
487 > readonly value: URI;
488 > readonly historyItem: ISCMHistoryItem;
489 > }
490 >
491 > export interface ISCMHistoryItemChangeRangeVariableEntry extends IBaseChatRequestVariableEntry {
492 > readonly kind: 'scmHistoryItemChangeRange';
493 > readonly value: URI;
494 > readonly historyItemChangeStart: {
495 > readonly uri: URI;
496 > readonly historyItem: ISCMHistoryItem;
497 > };
498 > readonly historyItemChangeEnd: {
499 > readonly uri: URI;
500 > readonly historyItem: ISCMHistoryItem;
501 > };
502 > }
503 >
504 > export interface ITerminalVariableEntry extends IBaseChatRequestVariableEntry {
505 > readonly kind: 'terminalCommand';
506 > readonly value: string;
507 > readonly resource: URI;
508 > readonly command: string;
509 > readonly output?: string;
510 > readonly exitCode?: number;
511 > }
512 >
513 > export interface IDebugVariableEntry extends IBaseChatRequestVariableEntry {
514 > readonly kind: 'debugVariable';
515 > readonly value: string;
516 > readonly expression: string;
517 > readonly type?: string;
518 > }
519 >
520 > export interface IAgentFeedbackVariableEntry extends IBaseChatRequestVariableEntry {
521 > readonly kind: 'agentFeedback';
522 > readonly sessionResource: URI;
523 > /**
524 > * The agent-host annotations channel URI that backs these feedback items
525 > * (each item id is an annotation id on this channel). Set only for
526 > * agent-host sessions; used to emit {@link MessageAnnotationsAttachment}s
527 > * referencing the specific comments on the wire.
528 > */
529 > readonly annotationsResource?: URI;
530 > readonly feedbackItems: ReadonlyArray<{
531 > readonly id: string;
532 > readonly text: string;
533 > readonly resourceUri: URI;
534 > readonly range: IRange;
535 > readonly codeSelection?: string;
536 > readonly diffHunks?: string;
537 > /** When this item was converted from a PR review comment, the original thread ID. */
538 > readonly sourcePRReviewCommentId?: string;
539 > /** Additional replies that belong to the same comment thread as {@link text}. */
540 > readonly replies?: readonly string[];
541 > }>;
542 > }
543 >
544 > export interface IChatRequestDebugEventsVariableEntry extends IBaseChatRequestVariableEntry {
545 > readonly kind: 'debugEvents';
546 > /** Timestamp when the debug events were snapshotted. */
547 > readonly snapshotTime: number;
548 > /** The session resource these debug events belong to. */
549 > readonly sessionResource: URI;
550 > }
551 >
552 > export interface IChatRequestSessionReferenceVariableEntry extends IBaseChatRequestVariableEntry {
553 > readonly kind: 'sessionReference';
554 > readonly value: URI;
555 > }
556 >
557 > export interface IBrowserViewVariableEntry extends IBaseChatRequestVariableEntry {
558 > readonly kind: 'browserView';
559 > readonly value: URI;
560 > readonly browserId: string;
561 > }
562 >
563 > export function isBrowserViewVariableEntry(entry: IChatRequestVariableEntry): entry is IBrowserViewVariableEntry {
564 return entry.kind === 'browserView';
565 }
567 > export type IChatRequestVariableEntry = IGenericChatRequestVariableEntry | IChatRequestImplicitVariableEntry | IChatRequestPasteVariableEntry
568 > | ISymbolVariableEntry | ICommandResultVariableEntry | IDiagnosticVariableEntry | IImageVariableEntry
569 > | IChatRequestToolEntry | IChatRequestToolSetEntry
570 > | IChatRequestDirectoryEntry | IChatRequestFileEntry | INotebookOutputVariableEntry | IElementVariableEntry
571 > | IPromptFileVariableEntry | IPromptTextVariableEntry
572 > | ISCMHistoryItemVariableEntry | ISCMHistoryItemChangeVariableEntry | ISCMHistoryItemChangeRangeVariableEntry | ITerminalVariableEntry
573 > | IChatRequestStringVariableEntry | IChatRequestWorkspaceVariableEntry | IDebugVariableEntry | IAgentFeedbackVariableEntry
574 > | IChatRequestDebugEventsVariableEntry | IChatRequestSessionReferenceVariableEntry | IBrowserViewVariableEntry;
575 >
576 > export namespace IChatRequestVariableEntry {
577 >
578 > /**
579 > * Returns URI of the passed variant entry. Return undefined if not found.
580 > */
581 > export function toUri(entry: IChatRequestVariableEntry): URI | undefined {
582 return URI.isUri(entry.value)
583 ? entry.value
586 : undefined;
587 }
589 > export function toExport(v: IChatRequestVariableEntry): IChatRequestVariableEntry {
590 if (v.value instanceof Uint8Array) {
591 // 'dup' here is needed otherwise TS complains about the narrowed `value` in a spread operation
597 return v;
598 }
600 > export function fromExport(v: IChatRequestVariableEntry): IChatRequestVariableEntry {
601 // Old variables format
602 // eslint-disable-next-line local/code-no-in-operator
623 }
624 }
626 >
627 > export function isImplicitVariableEntry(obj: IChatRequestVariableEntry): obj is IChatRequestImplicitVariableEntry {
628 return obj.kind === 'implicit';
629 }
631 > export function isStringVariableEntry(obj: IChatRequestVariableEntry): obj is IChatRequestStringVariableEntry {
632 return obj.kind === 'string';
633 }
635 > export function isTerminalVariableEntry(obj: IChatRequestVariableEntry): obj is ITerminalVariableEntry {
636 return obj.kind === 'terminalCommand';
637 }
639 > export function isDebugVariableEntry(obj: IChatRequestVariableEntry): obj is IDebugVariableEntry {
640 return obj.kind === 'debugVariable';
641 }
643 > export function isAgentFeedbackVariableEntry(obj: IChatRequestVariableEntry): obj is IAgentFeedbackVariableEntry {
644 return obj.kind === 'agentFeedback';
645 }
647 > export function isPasteVariableEntry(obj: IChatRequestVariableEntry): obj is IChatRequestPasteVariableEntry {
648 return obj.kind === 'paste';
649 }
651 > export function isWorkspaceVariableEntry(obj: IChatRequestVariableEntry): obj is IChatRequestWorkspaceVariableEntry {
652 return obj.kind === 'workspace';
653 }
655 > export function isImageVariableEntry(obj: IChatRequestVariableEntry): obj is IImageVariableEntry {
656 return obj.kind === 'image';
657 }
659 > export function isExplicitFileOrImageVariableEntry(obj: IChatRequestVariableEntry): obj is IChatRequestFileEntry | IChatRequestDirectoryEntry | IImageVariableEntry {
660 return obj.kind === 'file' || obj.kind === 'directory' || obj.kind === 'image';
661 }
663 > export function getExplicitFileOrImageAttachmentSummary(entries: readonly IChatRequestVariableEntry[]): string | undefined {
664 const fileOrImageEntries = entries.filter(isExplicitFileOrImageVariableEntry);
665 if (!fileOrImageEntries.length) {
677 : localize('chat.attachmentSummary.file.many', "Attached {0} files", fileOrImageEntries.length);
678 }
680 > export function isNotebookOutputVariableEntry(obj: IChatRequestVariableEntry): obj is INotebookOutputVariableEntry {
681 return obj.kind === 'notebookOutput';
682 }
684 > export function isElementVariableEntry(obj: IChatRequestVariableEntry): obj is IElementVariableEntry {
685 return obj.kind === 'element';
686 }
688 > export function isDiagnosticsVariableEntry(obj: IChatRequestVariableEntry): obj is IDiagnosticVariableEntry {
689 return obj.kind === 'diagnostic';
690 }
692 > export function isChatRequestFileEntry(obj: IChatRequestVariableEntry): obj is IChatRequestFileEntry {
693 return obj.kind === 'file';
694 }
696 > export function isPromptFileVariableEntry(obj: IChatRequestVariableEntry): obj is IPromptFileVariableEntry {
697 return obj.kind === 'promptFile';
698 }
700 > export function isPromptTextVariableEntry(obj: IChatRequestVariableEntry): obj is IPromptTextVariableEntry {
701 return obj.kind === 'promptText';
702 }
704 > export function isChatRequestVariableEntry(obj: unknown): obj is IChatRequestVariableEntry {
705 const entry = obj as IChatRequestVariableEntry;
706 return typeof entry === 'object' &&
709 typeof entry.name === 'string';
710 }
712 > export function isSCMHistoryItemVariableEntry(obj: IChatRequestVariableEntry): obj is ISCMHistoryItemVariableEntry {
713 return obj.kind === 'scmHistoryItem';
714 }
716 > export function isSCMHistoryItemChangeVariableEntry(obj: IChatRequestVariableEntry): obj is ISCMHistoryItemChangeVariableEntry {
717 return obj.kind === 'scmHistoryItemChange';
718 }
720 > export function isSCMHistoryItemChangeRangeVariableEntry(obj: IChatRequestVariableEntry): obj is ISCMHistoryItemChangeRangeVariableEntry {
721 return obj.kind === 'scmHistoryItemChangeRange';
722 }
724 > export function isStringImplicitContextValue(value: unknown): value is StringChatContextValue {
725 const asStringImplicitContextValue = value as Partial<StringChatContextValue>;
726 return (
736 );
737 }
739 > export enum PromptFileVariableKind {
740 > Instruction = 'vscode.instructions.file.root',
741 > InstructionReference = `vscode.instructions.file.reference`,
742 > PromptFile = 'vscode.prompt.file',
743 > }
744 >
745 > /**
746 > * Utility to convert a {@link uri} to a chat variable entry.
747 > * The `id` of the chat variable can be one of the following:
748 > *
749 > * - `vscode.instructions.file.reference__<URI>`: for all non-root prompt instructions references
750 > * - `vscode.instructions.file.root__<URI>`: for *root* prompt instructions references
751 > * - `vscode.prompt.file__<URI>`: for prompt file references
752 > *
753 > * @param uri A resource URI that points to a prompt instructions file.
754 > * @param kind The kind of the prompt file variable entry.
755 > */
756 > export function toPromptFileVariableEntry(uri: URI, kind: PromptFileVariableKind, originLabel?: string, automaticallyAdded = false, toolReferences?: ChatRequestToolReferenceEntry[]): IPromptFileVariableEntry {
757 // `id` for all `prompt files` starts with the well-defined part that the copilot extension(or other chatbot) can rely on
758 return {
768 };
769 }
771 > enum PromptTextVariableKind {
772 > CustomizationsIndex = 'vscode.customizations.index',
773 > }
774 >
775 > export function toPromptTextVariableEntry(content: string, automaticallyAdded = false, toolReferences?: ChatRequestToolReferenceEntry[]): IPromptTextVariableEntry {
776 return {
777 id: PromptTextVariableKind.CustomizationsIndex,
784 };
785 }
787 > export function toFileVariableEntry(uri: URI, range?: IRange): IChatRequestFileEntry {
788 return {
789 kind: 'file',
793 };
794 }
796 > export function toToolVariableEntry(entry: IToolData, range?: IOffsetRange): IChatRequestToolEntry {
797 return {
798 kind: 'tool',
804 };
805 }
807 > export function toToolSetVariableEntry(entry: IToolSet, range?: IOffsetRange): IChatRequestToolSetEntry {
808 return {
809 kind: 'toolset',
815 };
816 }
818 > export class ChatRequestVariableSet {
819 > private _ids = new Set<string>();
820 > private _entries: IChatRequestVariableEntry[] = [];
821 >
822 > constructor(entries?: IChatRequestVariableEntry[]) {
823 if (entries) {
824 this.add(...entries);
825 }
826 }
828 > public add(...entry: IChatRequestVariableEntry[]): void {
829 for (const e of entry) {
830 if (!this._ids.has(e.id)) {
834 }
835 }
837 > public insertFirst(entry: IChatRequestVariableEntry): void {
838 if (!this._ids.has(entry.id)) {
839 this._ids.add(entry.id);
841 }
842 }
844 > public remove(entry: IChatRequestVariableEntry): void {
845 this._ids.delete(entry.id);
846 this._entries = this._entries.filter(e => e.id !== entry.id);
847 }
849 > public has(entry: IChatRequestVariableEntry): boolean {
850 return this._ids.has(entry.id);
851 }
853 > public asArray(): IChatRequestVariableEntry[] {
854 return this._entries.slice(0); // return a copy
855 }
857 > public get length(): number {
858 return this._entries.length;
859 }
src/vs/base/common/strings.ts 488 covered LOC · 114 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- strings.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { LRUCachedFunction } from './cache.js';
7 > import { CharCode } from './charCode.js';
8 > import { Lazy } from './lazy.js';
9 > import { Constants } from './uint.js';
10 >
11 > export function isFalsyOrWhitespace(str: string | undefined): boolean {
12 if (!str || typeof str !== 'string') {
13 return true;
15 return str.trim().length === 0;
16 }
17 > strings.ts
18 > const _formatRegexp = /{(\d+)}/g;
19 >
20 > /**
21 > * Helper to produce a string with a variable number of arguments. Insert variable segments
22 > * into the string using the {n} notation where N is the index of the argument following the string.
23 > * @param value string to which formatting is applied
24 > * @param args replacements for {n}-entries
25 > */
26 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
27 > export function format(value: string, ...args: any[]): string {
28 if (args.length === 0) {
29 return value;
36 });
37 }
38 > strings.ts
39 > const _format2Regexp = /{([^}]+)}/g;
40 >
41 > /**
42 > * Helper to create a string from a template and a string record.
43 > * Similar to `format` but with objects instead of positional arguments.
44 > */
45 > export function format2(template: string, values: Record<string, unknown>): string {
46 if (Object.keys(values).length === 0) {
47 return template;
49 return template.replace(_format2Regexp, (match, group) => (values[group] ?? match) as string);
50 }
51 > strings.ts
52 > /**
53 > * Encodes the given value so that it can be used as literal value in html attributes.
54 > *
55 > * In other words, computes `$val`, such that `attr` in `<div attr="$val" />` has the runtime value `value`.
56 > * This prevents XSS injection.
57 > */
58 > export function htmlAttributeEncodeValue(value: string): string {
59 return value.replace(/[<>"'&]/g, ch => {
60 switch (ch) {
68 });
69 }
70 > strings.ts
71 > /**
72 > * Converts HTML characters inside the string to use entities instead. Makes the string safe from
73 > * being used e.g. in HTMLElement.innerHTML.
74 > */
75 > export function escape(html: string): string {
76 return html.replace(/[<>&]/g, function (match) {
77 switch (match) {
83 });
84 }
85 > strings.ts
86 > /**
87 > * Escapes regular expression characters in a given string
88 > */
89 > export function escapeRegExpCharacters(value: string): string {
90 return value.replace(/[\\\{\}\*\+\?\|\^\$\.\[\]\(\)]/g, '\\$&');
91 }
92 > strings.ts
93 > /**
94 > * Counts how often `substr` occurs inside `value`.
95 > */
96 > export function count(value: string, substr: string): number {
97 let result = 0;
98 let index = value.indexOf(substr);
103 return result;
104 }
105 > strings.ts
106 > export function truncate(value: string, maxLength: number, suffix = Ellipsis): string {
107 if (value.length <= maxLength) {
108 return value;
111 return `${value.substr(0, maxLength)}${suffix}`;
112 }
113 > strings.ts
114 > export function truncateMiddle(value: string, maxLength: number, suffix = Ellipsis): string {
115 if (value.length <= maxLength) {
116 return value;
122 return `${value.substr(0, prefixLength)}${suffix}${value.substr(value.length - suffixLength)}`;
123 }
124 > strings.ts
125 > /**
126 > * Removes all occurrences of needle from the beginning and end of haystack.
127 > * @param haystack string to trim
128 > * @param needle the thing to trim (default is a blank)
129 > */
130 > export function trim(haystack: string, needle: string = ' '): string {
131 const trimmed = ltrim(haystack, needle);
132 return rtrim(trimmed, needle);
133 }
134 > strings.ts
135 > /**
136 > * Removes all occurrences of needle from the beginning of haystack.
137 > * @param haystack string to trim
138 > * @param needle the thing to trim
139 > */
140 > export function ltrim(haystack: string, needle: string): string {
141 if (!haystack || !needle) {
142 return haystack;
157 return haystack.substring(offset);
158 }
159 > strings.ts
160 > /**
161 > * Removes all occurrences of needle from the end of haystack.
162 > * @param haystack string to trim
163 > * @param needle the thing to trim
164 > */
165 > export function rtrim(haystack: string, needle: string): string {
166 if (!haystack || !needle) {
167 return haystack;
187 return haystack.substring(0, offset);
188 }
189 > strings.ts
190 > export function convertSimple2RegExpPattern(pattern: string): string {
191 return pattern.replace(/[\-\\\{\}\+\?\|\^\$\.\,\[\]\(\)\#\s]/g, '\\$&').replace(/[\*]/g, '.*');
192 }
193 > strings.ts
194 > export interface RegExpOptions {
195 > matchCase?: boolean;
196 > wholeWord?: boolean;
197 > multiline?: boolean;
198 > global?: boolean;
199 > unicode?: boolean;
200 > }
201 >
202 > export function createRegExp(searchString: string, isRegex: boolean, options: RegExpOptions = {}): RegExp {
203 if (!searchString) {
204 throw new Error('Cannot create regex from empty string');
231 return new RegExp(searchString, modifiers);
232 }
233 > strings.ts
234 > export function regExpLeadsToEndlessLoop(regexp: RegExp): boolean {
235 // Exit early if it's one of these special cases which are meant to match
236 // against an empty string
244 return !!(match && regexp.lastIndex === 0);
245 }
246 > strings.ts
247 > export function joinStrings(items: (string | undefined | null | false)[], separator: string): string {
248 return items.filter(item => item !== undefined && item !== null && item !== false).join(separator);
249 }
250 > strings.ts
251 > export function splitLines(str: string): string[] {
252 return str.split(/\r\n|\r|\n/);
253 }
254 > strings.ts
255 > export function splitLinesIncludeSeparators(str: string): string[] {
256 const linesWithSeparators: string[] = [];
257 const splitLinesAndSeparators = str.split(/(\r\n|\r|\n)/);
261 return linesWithSeparators;
262 }
263 > strings.ts
264 > export function indexOfPattern(str: string, re: RegExp) {
265 const match = re.exec(str);
266 if (match) {
269 return -1;
270 }
271 > strings.ts
272 > /**
273 > * Returns first index of the string that is not whitespace.
274 > * If string is empty or contains only whitespaces, returns -1
275 > */
276 > export function firstNonWhitespaceIndex(str: string): number {
277 for (let i = 0, len = str.length; i < len; i++) {
278 const chCode = str.charCodeAt(i);
283 return -1;
284 }
285 > strings.ts
286 > /**
287 > * Returns the leading whitespace of the string.
288 > * If the string contains only whitespaces, returns entire string
289 > */
290 > export function getLeadingWhitespace(str: string, start: number = 0, end: number = str.length): string {
291 for (let i = start; i < end; i++) {
292 const chCode = str.charCodeAt(i);
297 return str.substring(start, end);
298 }
299 > strings.ts
300 > /**
301 > * Returns last index of the string that is not whitespace.
302 > * If string is empty or contains only whitespaces, returns -1
303 > */
304 > export function lastNonWhitespaceIndex(str: string, startIndex: number = str.length - 1): number {
305 for (let i = startIndex; i >= 0; i--) {
306 const chCode = str.charCodeAt(i);
311 return -1;
312 }
313 > strings.ts
314 > export function getIndentationLength(str: string): number {
315 const idx = firstNonWhitespaceIndex(str);
316 if (idx === -1) { return str.length; }
317 return idx;
318 }
319 > strings.ts
320 > /**
321 > * Function that works identically to String.prototype.replace, except, the
322 > * replace function is allowed to be async and return a Promise.
323 > */
324 > export function replaceAsync(str: string, search: RegExp, replacer: (match: string, ...args: unknown[]) => Promise<string>): Promise<string> {
325 const parts: (string | Promise<string>)[] = [];
326
340 return Promise.all(parts).then(p => p.join(''));
341 }
342 > strings.ts
343 > export function compare(a: string, b: string): number {
344 if (a < b) {
345 return -1;
350 }
351 }
352 > strings.ts
353 > export function compareSubstring(a: string, b: string, aStart: number = 0, aEnd: number = a.length, bStart: number = 0, bEnd: number = b.length): number {
354 > for (; aStart < aEnd && bStart < bEnd; aStart++, bStart++) { strings.ts
355 > const codeA = a.charCodeAt(aStart);
356 > const codeB = b.charCodeAt(bStart);
357 > if (codeA < codeB) {
358 return -1;
359 > } else if (codeA > codeB) { strings.ts
360 return 1;
361 }
362 > } strings.ts
363 > const aLen = aEnd - aStart; strings.ts
364 > const bLen = bEnd - bStart;
365 > if (aLen < bLen) {
366 return -1;
367 > } else if (aLen > bLen) { strings.ts
368 return 1;
369 }
370 > return 0; strings.ts
371 > }
372 > strings.ts
373 > export function compareIgnoreCase(a: string, b: string): number {
374 > return compareSubstringIgnoreCase(a, b, 0, a.length, 0, b.length); strings.ts
375 > }
376 > strings.ts
377 > export function compareSubstringIgnoreCase(a: string, b: string, aStart: number = 0, aEnd: number = a.length, bStart: number = 0, bEnd: number = b.length): number {
378 > strings.ts
379 > for (; aStart < aEnd && bStart < bEnd; aStart++, bStart++) {
380 >
381 > let codeA = a.charCodeAt(aStart);
382 > let codeB = b.charCodeAt(bStart);
383 >
384 > if (codeA === codeB) {
385 > // equal strings.ts
386 > continue;
387 > }
388
389 > if (codeA >= 128 || codeB >= 128) { strings.ts
390 // not ASCII letters -> fallback to lower-casing strings
391 return compareSubstring(a.toLowerCase(), b.toLowerCase(), aStart, aEnd, bStart, bEnd);
409 return diff;
410 }
411 > strings.ts
412 > const aLen = aEnd - aStart;
413 > const bLen = bEnd - bStart;
414 >
415 > if (aLen < bLen) {
416 return -1;
417 > } else if (aLen > bLen) { strings.ts
418 return 1;
419 }
420 > strings.ts
421 > return 0;
422 > }
423 > strings.ts
424 > export function isAsciiDigit(code: number): boolean {
425 return code >= CharCode.Digit0 && code <= CharCode.Digit9;
426 }
427 > strings.ts
428 > export function isLowerAsciiLetter(code: number): boolean {
429 return code >= CharCode.a && code <= CharCode.z;
430 }
431 > strings.ts
432 > export function isUpperAsciiLetter(code: number): boolean {
433 return code >= CharCode.A && code <= CharCode.Z;
434 }
435 > strings.ts
436 > export function equalsIgnoreCase(a: string, b: string): boolean {
437 return a.length === b.length && compareSubstringIgnoreCase(a, b) === 0;
438 }
439 > strings.ts
440 > export function equals(a: string | undefined, b: string | undefined, ignoreCase?: boolean): boolean {
441 return a === b || (!!ignoreCase && a !== undefined && b !== undefined && equalsIgnoreCase(a, b));
442 }
443 > strings.ts
444 > export function startsWithIgnoreCase(str: string, candidate: string): boolean {
445 const len = candidate.length;
446 return len <= str.length && compareSubstringIgnoreCase(str, candidate, 0, len) === 0;
447 }
448 > strings.ts
449 > export function endsWithIgnoreCase(str: string, candidate: string): boolean {
450 const len = str.length;
451 const start = len - candidate.length;
452 return start >= 0 && compareSubstringIgnoreCase(str, candidate, start, len) === 0;
453 }
454 > strings.ts
455 > /**
456 > * @returns the length of the common prefix of the two strings.
457 > */
458 > export function commonPrefixLength(a: string, b: string): number {
459
460 const len = Math.min(a.length, b.length);
469 return len;
470 }
471 > strings.ts
472 > /**
473 > * @returns the length of the common suffix of the two strings.
474 > */
475 > export function commonSuffixLength(a: string, b: string): number {
476
477 const len = Math.min(a.length, b.length);
489 return len;
490 }
491 > strings.ts
492 > /**
493 > * See http://en.wikipedia.org/wiki/Surrogate_pair
494 > */
495 > export function isHighSurrogate(charCode: number): boolean {
496 return (0xD800 <= charCode && charCode <= 0xDBFF);
497 }
498 > strings.ts
499 > /**
500 > * See http://en.wikipedia.org/wiki/Surrogate_pair
501 > */
502 > export function isLowSurrogate(charCode: number): boolean {
503 return (0xDC00 <= charCode && charCode <= 0xDFFF);
504 }
505 > strings.ts
506 > /**
507 > * See http://en.wikipedia.org/wiki/Surrogate_pair
508 > */
509 > export function computeCodePoint(highSurrogate: number, lowSurrogate: number): number {
510 return ((highSurrogate - 0xD800) << 10) + (lowSurrogate - 0xDC00) + 0x10000;
511 }
512 > strings.ts
513 > /**
514 > * get the code point that begins at offset `offset`
515 > */
516 > export function getNextCodePoint(str: string, len: number, offset: number): number {
517 const charCode = str.charCodeAt(offset);
518 if (isHighSurrogate(charCode) && offset + 1 < len) {
524 return charCode;
525 }
526 > strings.ts
527 > /**
528 > * get the code point that ends right before offset `offset`
529 > */
530 function getPrevCodePoint(str: string, offset: number): number {
531 const charCode = str.charCodeAt(offset - 1);
538 return charCode;
539 }
540 > strings.ts
541 > export class CodePointIterator {
542 >
543 > private readonly _str: string;
544 > private readonly _len: number;
545 > private _offset: number;
546 >
547 > public get offset(): number {
548 return this._offset;
549 }
550 > strings.ts
551 > constructor(str: string, offset: number = 0) {
552 this._str = str;
553 this._len = str.length;
554 this._offset = offset;
555 }
556 > strings.ts
557 > public setOffset(offset: number): void {
558 this._offset = offset;
559 }
560 > strings.ts
561 > public prevCodePoint(): number {
562 const codePoint = getPrevCodePoint(this._str, this._offset);
563 this._offset -= (codePoint >= Constants.UNICODE_SUPPLEMENTARY_PLANE_BEGIN ? 2 : 1);
564 return codePoint;
565 }
566 > strings.ts
567 > public nextCodePoint(): number {
568 const codePoint = getNextCodePoint(this._str, this._len, this._offset);
569 this._offset += (codePoint >= Constants.UNICODE_SUPPLEMENTARY_PLANE_BEGIN ? 2 : 1);
570 return codePoint;
571 }
572 > strings.ts
573 > public eol(): boolean {
574 return (this._offset >= this._len);
575 }
576 > } strings.ts
577 >
578 > export class GraphemeIterator {
579 >
580 > private readonly _iterator: CodePointIterator;
581 >
582 > public get offset(): number {
583 return this._iterator.offset;
584 }
585 > strings.ts
586 > constructor(str: string, offset: number = 0) {
587 this._iterator = new CodePointIterator(str, offset);
588 }
589 > strings.ts
590 > public nextGraphemeLength(): number {
591 const graphemeBreakTree = GraphemeBreakTree.getInstance();
592 const iterator = this._iterator;
606 return (iterator.offset - initialOffset);
607 }
608 > strings.ts
609 > public prevGraphemeLength(): number {
610 const graphemeBreakTree = GraphemeBreakTree.getInstance();
611 const iterator = this._iterator;
625 return (initialOffset - iterator.offset);
626 }
627 > strings.ts
628 > public eol(): boolean {
629 return this._iterator.eol();
630 }
631 > } strings.ts
632 >
633 > export function nextCharLength(str: string, initialOffset: number): number {
634 const iterator = new GraphemeIterator(str, initialOffset);
635 return iterator.nextGraphemeLength();
636 }
637 > strings.ts
638 > export function prevCharLength(str: string, initialOffset: number): number {
639 const iterator = new GraphemeIterator(str, initialOffset);
640 return iterator.prevGraphemeLength();
641 }
642 > strings.ts
643 > export function getCharContainingOffset(str: string, offset: number): [number, number] {
644 if (offset > 0 && isLowSurrogate(str.charCodeAt(offset))) {
645 offset--;
649 return [startOffset, endOffset];
650 }
651 > strings.ts
652 > export function charCount(str: string): number {
653 const iterator = new GraphemeIterator(str);
654 let length = 0;
659 return length;
660 }
661 > strings.ts
662 > let CONTAINS_RTL: RegExp | undefined = undefined;
663 >
664 function makeContainsRtl() {
665 // Generated using https://github.com/alexdima/unicode-utils/blob/main/rtl-test.js
666 return /(?:[\u05BE\u05C0\u05C3\u05C6\u05D0-\u05F4\u0608\u060B\u060D\u061B-\u064A\u066D-\u066F\u0671-\u06D5\u06E5\u06E6\u06EE\u06EF\u06FA-\u0710\u0712-\u072F\u074D-\u07A5\u07B1-\u07EA\u07F4\u07F5\u07FA\u07FE-\u0815\u081A\u0824\u0828\u0830-\u0858\u085E-\u088E\u08A0-\u08C9\u200F\uFB1D\uFB1F-\uFB28\uFB2A-\uFD3D\uFD50-\uFDC7\uFDF0-\uFDFC\uFE70-\uFEFC]|\uD802[\uDC00-\uDD1B\uDD20-\uDE00\uDE10-\uDE35\uDE40-\uDEE4\uDEEB-\uDF35\uDF40-\uDFFF]|\uD803[\uDC00-\uDD23\uDE80-\uDEA9\uDEAD-\uDF45\uDF51-\uDF81\uDF86-\uDFF6]|\uD83A[\uDC00-\uDCCF\uDD00-\uDD43\uDD4B-\uDFFF]|\uD83B[\uDC00-\uDEBB])/;
667 }
668 > strings.ts
669 > /**
670 > * Returns true if `str` contains any Unicode character that is classified as "R" or "AL".
671 > */
672 > export function containsRTL(str: string): boolean {
673 if (!CONTAINS_RTL) {
674 CONTAINS_RTL = makeContainsRtl();
677 return CONTAINS_RTL.test(str);
678 }
679 > strings.ts
680 > const IS_BASIC_ASCII = /^[\t\n\r\x20-\x7E]*$/;
681 > /**
682 > * Returns true if `str` contains only basic ASCII characters in the range 32 - 126 (including 32 and 126) or \n, \r, \t
683 > */
684 > export function isBasicASCII(str: string): boolean {
685 return IS_BASIC_ASCII.test(str);
686 }
687 > strings.ts
688 > export const UNUSUAL_LINE_TERMINATORS = /[\u2028\u2029]/; // LINE SEPARATOR (LS) or PARAGRAPH SEPARATOR (PS)
689 > /**
690 > * Returns true if `str` contains unusual line terminators, like LS or PS
691 > */
692 > export function containsUnusualLineTerminators(str: string): boolean {
693 return UNUSUAL_LINE_TERMINATORS.test(str);
694 }
695 > strings.ts
696 > export function isFullWidthCharacter(charCode: number): boolean {
697 // Do a cheap trick to better support wrapping of wide characters, treat them as 2 columns
698 // http://jrgraphix.net/research/unicode_blocks.php
741 );
742 }
743 > strings.ts
744 > /**
745 > * A fast function (therefore imprecise) to check if code points are emojis.
746 > * Generated using https://github.com/alexdima/unicode-utils/blob/main/emoji-test.js
747 > */
748 > export function isEmojiImprecise(x: number): boolean {
749 return (
750 (x >= 0x1F1E6 && x <= 0x1F1FF) || (x === 8986) || (x === 8987) || (x === 9200)
755 );
756 }
757 > strings.ts
758 > /**
759 > * Given a string and a max length returns a shorted version. Shorting
760 > * happens at favorable positions - such as whitespace or punctuation characters.
761 > * The return value can be longer than the given value of `n`. Leading whitespace is always trimmed.
762 > */
763 > export function lcut(text: string, n: number, prefix = ''): string {
764 const trimmed = text.trimStart();
765
785 return prefix + trimmed.substring(i).trimStart();
786 }
787 > strings.ts
788 > /**
789 > * Given a string and a max length returns a shortened version keeping the beginning.
790 > * Shortening happens at favorable positions - such as whitespace or punctuation characters.
791 > * Trailing whitespace is always trimmed.
792 > */
793 > export function rcut(text: string, n: number, suffix = ''): string {
794 const trimmed = text.trimEnd();
795
832 return result + suffix;
833 }
834 > strings.ts
835 > // Defacto standard: https://invisible-island.net/xterm/ctlseqs/ctlseqs.html
836 > const CSI_SEQUENCE = /(?:\x1b\[|\x9b)[=?>!]?[\d;:]*["$#'* ]?[a-zA-Z@^`{}|~]/;
837 > const OSC_SEQUENCE = /(?:\x1b\]|\x9d).*?(?:\x1b\\|\x07|\x9c)/;
838 > const ESC_SEQUENCE = /\x1b(?:[ #%\(\)\*\+\-\.\/]?[a-zA-Z0-9\|}~@])/;
839 > const CONTROL_SEQUENCES = new RegExp('(?:' + [
840 > CSI_SEQUENCE.source,
841 > OSC_SEQUENCE.source,
842 > ESC_SEQUENCE.source,
843 > ].join('|') + ')', 'g');
844 >
845 > /** Iterates over parts of a string with CSI sequences */
846 > export function* forAnsiStringParts(str: string) {
847 let last = 0;
848 for (const match of str.matchAll(CONTROL_SEQUENCES)) {
859 }
860 }
861 > strings.ts
862 > /**
863 > * Strips ANSI escape sequences from a string.
864 > * @param str The dastringa stringo strip the ANSI escape sequences from.
865 > *
866 > * @example
867 > * removeAnsiEscapeCodes('\u001b[31mHello, World!\u001b[0m');
868 > * // 'Hello, World!'
869 > */
870 > export function removeAnsiEscapeCodes(str: string): string {
871 if (str) {
872 str = str.replace(CONTROL_SEQUENCES, '');
875 return str;
876 }
877 > strings.ts
878 > const PROMPT_NON_PRINTABLE = /\\\[.*?\\\]/g;
879 >
880 > /**
881 > * Strips ANSI escape sequences from a UNIX-style prompt string (eg. `$PS1`).
882 > * @param str The string to strip the ANSI escape sequences from.
883 > *
884 > * @example
885 > * removeAnsiEscapeCodesFromPrompt('\n\\[\u001b[01;34m\\]\\w\\[\u001b[00m\\]\n\\[\u001b[1;32m\\]> \\[\u001b[0m\\]');
886 > * // '\n\\w\n> '
887 > */
888 > export function removeAnsiEscapeCodesFromPrompt(str: string): string {
889 return removeAnsiEscapeCodes(str).replace(PROMPT_NON_PRINTABLE, '');
890 }
891 > strings.ts
892 >
893 > // -- UTF-8 BOM
894 >
895 > export const UTF8_BOM_CHARACTER = String.fromCharCode(CharCode.UTF8_BOM);
896 >
897 > export function startsWithUTF8BOM(str: string): boolean {
898 return !!(str && str.length > 0 && str.charCodeAt(0) === CharCode.UTF8_BOM);
899 }
900 > strings.ts
901 > export function stripUTF8BOM(str: string): string {
902 return startsWithUTF8BOM(str) ? str.substr(1) : str;
903 }
904 > strings.ts
905 > /**
906 > * Checks if the characters of the provided query string are included in the
907 > * target string. The characters do not have to be contiguous within the string.
908 > */
909 > export function fuzzyContains(target: string, query: string): boolean {
910 if (!target || !query) {
911 return false; // return early if target or query are undefined
936 return true;
937 }
938 > strings.ts
939 > export function containsUppercaseCharacter(target: string, ignoreEscapedChars = false): boolean {
940 if (!target) {
941 return false;
948 return target.toLowerCase() !== target;
949 }
950 > strings.ts
951 > export function uppercaseFirstLetter(str: string): string {
952 return str.charAt(0).toUpperCase() + str.slice(1);
953 }
954 > strings.ts
955 > export function getNLines(str: string, n = 1): string {
956 if (n === 0) {
957 return '';
974 return str.substr(0, idx);
975 }
976 > strings.ts
977 > /**
978 > * Produces 'a'-'z', followed by 'A'-'Z'... followed by 'a'-'z', etc.
979 > */
980 > export function singleLetterHash(n: number): string {
981 const LETTERS_CNT = (CharCode.Z - CharCode.A + 1);
982
989 return String.fromCharCode(CharCode.A + n - LETTERS_CNT);
990 }
991 > strings.ts
992 > //#region Unicode Grapheme Break
993 >
994 > export function getGraphemeBreakType(codePoint: number): GraphemeBreakType {
995 const graphemeBreakTree = GraphemeBreakTree.getInstance();
996 return graphemeBreakTree.getGraphemeBreakType(codePoint);
997 }
998 > strings.ts
999 function breakBetweenGraphemeBreakType(breakTypeA: GraphemeBreakType, breakTypeB: GraphemeBreakType): boolean {
1000 // http://www.unicode.org/reports/tr29/#Grapheme_Cluster_Boundary_Rules
1076 return true;
1077 }
1078 > strings.ts
1079 > export const enum GraphemeBreakType {
1080 > Other = 0,
1081 > Prepend = 1,
1082 > CR = 2,
1083 > LF = 3,
1084 > Control = 4,
1085 > Extend = 5,
1086 > Regional_Indicator = 6,
1087 > SpacingMark = 7,
1088 > L = 8,
1089 > V = 9,
1090 > T = 10,
1091 > LV = 11,
1092 > LVT = 12,
1093 > ZWJ = 13,
1094 > Extended_Pictographic = 14
1095 > }
1096 >
1097 > class GraphemeBreakTree {
1098 >
1099 > private static _INSTANCE: GraphemeBreakTree | null = null;
1100 > public static getInstance(): GraphemeBreakTree {
1101 if (!GraphemeBreakTree._INSTANCE) {
1102 GraphemeBreakTree._INSTANCE = new GraphemeBreakTree();
1104 return GraphemeBreakTree._INSTANCE;
1105 }
1106 > strings.ts
1107 > private readonly _data: number[];
1108 >
1109 > constructor() {
1110 this._data = getGraphemeBreakRawData();
1111 }
1112 > strings.ts
1113 > public getGraphemeBreakType(codePoint: number): GraphemeBreakType {
1114 // !!! Let's make 7bit ASCII a bit faster: 0..31
1115 if (codePoint < 32) {
1145 return GraphemeBreakType.Other;
1146 }
1147 > } strings.ts
1148 >
1149 function getGraphemeBreakRawData(): number[] {
1150 // generated using https://github.com/alexdima/unicode-utils/blob/main/grapheme-break.js
1151 return JSON.parse('[0,0,0,51229,51255,12,44061,44087,12,127462,127487,6,7083,7085,5,47645,47671,12,54813,54839,12,128678,128678,14,3270,3270,5,9919,9923,14,45853,45879,12,49437,49463,12,53021,53047,12,71216,71218,7,128398,128399,14,129360,129374,14,2519,2519,5,4448,4519,9,9742,9742,14,12336,12336,14,44957,44983,12,46749,46775,12,48541,48567,12,50333,50359,12,52125,52151,12,53917,53943,12,69888,69890,5,73018,73018,5,127990,127990,14,128558,128559,14,128759,128760,14,129653,129655,14,2027,2035,5,2891,2892,7,3761,3761,5,6683,6683,5,8293,8293,4,9825,9826,14,9999,9999,14,43452,43453,5,44509,44535,12,45405,45431,12,46301,46327,12,47197,47223,12,48093,48119,12,48989,49015,12,49885,49911,12,50781,50807,12,51677,51703,12,52573,52599,12,53469,53495,12,54365,54391,12,65279,65279,4,70471,70472,7,72145,72147,7,119173,119179,5,127799,127818,14,128240,128244,14,128512,128512,14,128652,128652,14,128721,128722,14,129292,129292,14,129445,129450,14,129734,129743,14,1476,1477,5,2366,2368,7,2750,2752,7,3076,3076,5,3415,3415,5,4141,4144,5,6109,6109,5,6964,6964,5,7394,7400,5,9197,9198,14,9770,9770,14,9877,9877,14,9968,9969,14,10084,10084,14,43052,43052,5,43713,43713,5,44285,44311,12,44733,44759,12,45181,45207,12,45629,45655,12,46077,46103,12,46525,46551,12,46973,46999,12,47421,47447,12,47869,47895,12,48317,48343,12,48765,48791,12,49213,49239,12,49661,49687,12,50109,50135,12,50557,50583,12,51005,51031,12,51453,51479,12,51901,51927,12,52349,52375,12,52797,52823,12,53245,53271,12,53693,53719,12,54141,54167,12,54589,54615,12,55037,55063,12,69506,69509,5,70191,70193,5,70841,70841,7,71463,71467,5,72330,72342,5,94031,94031,5,123628,123631,5,127763,127765,14,127941,127941,14,128043,128062,14,128302,128317,14,128465,128467,14,128539,128539,14,128640,128640,14,128662,128662,14,128703,128703,14,128745,128745,14,129004,129007,14,129329,129330,14,129402,129402,14,129483,129483,14,129686,129704,14,130048,131069,14,173,173,4,1757,1757,1,2200,2207,5,2434,2435,7,2631,2632,5,2817,2817,5,3008,3008,5,3201,3201,5,3387,3388,5,3542,3542,5,3902,3903,7,4190,4192,5,6002,6003,5,6439,6440,5,6765,6770,7,7019,7027,5,7154,7155,7,8205,8205,13,8505,8505,14,9654,9654,14,9757,9757,14,9792,9792,14,9852,9853,14,9890,9894,14,9937,9937,14,9981,9981,14,10035,10036,14,11035,11036,14,42654,42655,5,43346,43347,7,43587,43587,5,44006,44007,7,44173,44199,12,44397,44423,12,44621,44647,12,44845,44871,12,45069,45095,12,45293,45319,12,45517,45543,12,45741,45767,12,45965,45991,12,46189,46215,12,46413,46439,12,46637,46663,12,46861,46887,12,47085,47111,12,47309,47335,12,47533,47559,12,47757,47783,12,47981,48007,12,48205,48231,12,48429,48455,12,48653,48679,12,48877,48903,12,49101,49127,12,49325,49351,12,49549,49575,12,49773,49799,12,49997,50023,12,50221,50247,12,50445,50471,12,50669,50695,12,50893,50919,12,51117,51143,12,51341,51367,12,51565,51591,12,51789,51815,12,52013,52039,12,52237,52263,12,52461,52487,12,52685,52711,12,52909,52935,12,53133,53159,12,53357,53383,12,53581,53607,12,53805,53831,12,54029,54055,12,54253,54279,12,54477,54503,12,54701,54727,12,54925,54951,12,55149,55175,12,68101,68102,5,69762,69762,7,70067,70069,7,70371,70378,5,70720,70721,7,71087,71087,5,71341,71341,5,71995,71996,5,72249,72249,7,72850,72871,5,73109,73109,5,118576,118598,5,121505,121519,5,127245,127247,14,127568,127569,14,127777,127777,14,127872,127891,14,127956,127967,14,128015,128016,14,128110,128172,14,128259,128259,14,128367,128368,14,128424,128424,14,128488,128488,14,128530,128532,14,128550,128551,14,128566,128566,14,128647,128647,14,128656,128656,14,128667,128673,14,128691,128693,14,128715,128715,14,128728,128732,14,128752,128752,14,128765,128767,14,129096,129103,14,129311,129311,14,129344,129349,14,129394,129394,14,129413,129425,14,129466,129471,14,129511,129535,14,129664,129666,14,129719,129722,14,129760,129767,14,917536,917631,5,13,13,2,1160,1161,5,1564,1564,4,1807,1807,1,2085,2087,5,2307,2307,7,2382,2383,7,2497,2500,5,2563,2563,7,2677,2677,5,2763,2764,7,2879,2879,5,2914,2915,5,3021,3021,5,3142,3144,5,3263,3263,5,3285,3286,5,3398,3400,7,3530,3530,5,3633,3633,5,3864,3865,5,3974,3975,5,4155,4156,7,4229,4230,5,5909,5909,7,6078,6085,7,6277,6278,5,6451,6456,7,6744,6750,5,6846,6846,5,6972,6972,5,7074,7077,5,7146,7148,7,7222,7223,5,7416,7417,5,8234,8238,4,8417,8417,5,9000,9000,14,9203,9203,14,9730,9731,14,9748,9749,14,9762,9763,14,9776,9783,14,9800,9811,14,9831,9831,14,9872,9873,14,9882,9882,14,9900,9903,14,9929,9933,14,9941,9960,14,9974,9974,14,9989,9989,14,10006,10006,14,10062,10062,14,10160,10160,14,11647,11647,5,12953,12953,14,43019,43019,5,43232,43249,5,43443,43443,5,43567,43568,7,43696,43696,5,43765,43765,7,44013,44013,5,44117,44143,12,44229,44255,12,44341,44367,12,44453,44479,12,44565,44591,12,44677,44703,12,44789,44815,12,44901,44927,12,45013,45039,12,45125,45151,12,45237,45263,12,45349,45375,12,45461,45487,12,45573,45599,12,45685,45711,12,45797,45823,12,45909,45935,12,46021,46047,12,46133,46159,12,46245,46271,12,46357,46383,12,46469,46495,12,46581,46607,12,46693,46719,12,46805,46831,12,46917,46943,12,47029,47055,12,47141,47167,12,47253,47279,12,47365,47391,12,47477,47503,12,47589,47615,12,47701,47727,12,47813,47839,12,47925,47951,12,48037,48063,12,48149,48175,12,48261,48287,12,48373,48399,12,48485,48511,12,48597,48623,12,48709,48735,12,48821,48847,12,48933,48959,12,49045,49071,12,49157,49183,12,49269,49295,12,49381,49407,12,49493,49519,12,49605,49631,12,49717,49743,12,49829,49855,12,49941,49967,12,50053,50079,12,50165,50191,12,50277,50303,12,50389,50415,12,50501,50527,12,50613,50639,12,50725,50751,12,50837,50863,12,50949,50975,12,51061,51087,12,51173,51199,12,51285,51311,12,51397,51423,12,51509,51535,12,51621,51647,12,51733,51759,12,51845,51871,12,51957,51983,12,52069,52095,12,52181,52207,12,52293,52319,12,52405,52431,12,52517,52543,12,52629,52655,12,52741,52767,12,52853,52879,12,52965,52991,12,53077,53103,12,53189,53215,12,53301,53327,12,53413,53439,12,53525,53551,12,53637,53663,12,53749,53775,12,53861,53887,12,53973,53999,12,54085,54111,12,54197,54223,12,54309,54335,12,54421,54447,12,54533,54559,12,54645,54671,12,54757,54783,12,54869,54895,12,54981,55007,12,55093,55119,12,55243,55291,10,66045,66045,5,68325,68326,5,69688,69702,5,69817,69818,5,69957,69958,7,70089,70092,5,70198,70199,5,70462,70462,5,70502,70508,5,70750,70750,5,70846,70846,7,71100,71101,5,71230,71230,7,71351,71351,5,71737,71738,5,72000,72000,7,72160,72160,5,72273,72278,5,72752,72758,5,72882,72883,5,73031,73031,5,73461,73462,7,94192,94193,7,119149,119149,7,121403,121452,5,122915,122916,5,126980,126980,14,127358,127359,14,127535,127535,14,127759,127759,14,127771,127771,14,127792,127793,14,127825,127867,14,127897,127899,14,127945,127945,14,127985,127986,14,128000,128007,14,128021,128021,14,128066,128100,14,128184,128235,14,128249,128252,14,128266,128276,14,128335,128335,14,128379,128390,14,128407,128419,14,128444,128444,14,128481,128481,14,128499,128499,14,128526,128526,14,128536,128536,14,128543,128543,14,128556,128556,14,128564,128564,14,128577,128580,14,128643,128645,14,128649,128649,14,128654,128654,14,128660,128660,14,128664,128664,14,128675,128675,14,128686,128689,14,128695,128696,14,128705,128709,14,128717,128719,14,128725,128725,14,128736,128741,14,128747,128748,14,128755,128755,14,128762,128762,14,128981,128991,14,129009,129023,14,129160,129167,14,129296,129304,14,129320,129327,14,129340,129342,14,129356,129356,14,129388,129392,14,129399,129400,14,129404,129407,14,129432,129442,14,129454,129455,14,129473,129474,14,129485,129487,14,129648,129651,14,129659,129660,14,129671,129679,14,129709,129711,14,129728,129730,14,129751,129753,14,129776,129782,14,917505,917505,4,917760,917999,5,10,10,3,127,159,4,768,879,5,1471,1471,5,1536,1541,1,1648,1648,5,1767,1768,5,1840,1866,5,2070,2073,5,2137,2139,5,2274,2274,1,2363,2363,7,2377,2380,7,2402,2403,5,2494,2494,5,2507,2508,7,2558,2558,5,2622,2624,7,2641,2641,5,2691,2691,7,2759,2760,5,2786,2787,5,2876,2876,5,2881,2884,5,2901,2902,5,3006,3006,5,3014,3016,7,3072,3072,5,3134,3136,5,3157,3158,5,3260,3260,5,3266,3266,5,3274,3275,7,3328,3329,5,3391,3392,7,3405,3405,5,3457,3457,5,3536,3537,7,3551,3551,5,3636,3642,5,3764,3772,5,3895,3895,5,3967,3967,7,3993,4028,5,4146,4151,5,4182,4183,7,4226,4226,5,4253,4253,5,4957,4959,5,5940,5940,7,6070,6070,7,6087,6088,7,6158,6158,4,6432,6434,5,6448,6449,7,6679,6680,5,6742,6742,5,6754,6754,5,6783,6783,5,6912,6915,5,6966,6970,5,6978,6978,5,7042,7042,7,7080,7081,5,7143,7143,7,7150,7150,7,7212,7219,5,7380,7392,5,7412,7412,5,8203,8203,4,8232,8232,4,8265,8265,14,8400,8412,5,8421,8432,5,8617,8618,14,9167,9167,14,9200,9200,14,9410,9410,14,9723,9726,14,9733,9733,14,9745,9745,14,9752,9752,14,9760,9760,14,9766,9766,14,9774,9774,14,9786,9786,14,9794,9794,14,9823,9823,14,9828,9828,14,9833,9850,14,9855,9855,14,9875,9875,14,9880,9880,14,9885,9887,14,9896,9897,14,9906,9916,14,9926,9927,14,9935,9935,14,9939,9939,14,9962,9962,14,9972,9972,14,9978,9978,14,9986,9986,14,9997,9997,14,10002,10002,14,10017,10017,14,10055,10055,14,10071,10071,14,10133,10135,14,10548,10549,14,11093,11093,14,12330,12333,5,12441,12442,5,42608,42610,5,43010,43010,5,43045,43046,5,43188,43203,7,43302,43309,5,43392,43394,5,43446,43449,5,43493,43493,5,43571,43572,7,43597,43597,7,43703,43704,5,43756,43757,5,44003,44004,7,44009,44010,7,44033,44059,12,44089,44115,12,44145,44171,12,44201,44227,12,44257,44283,12,44313,44339,12,44369,44395,12,44425,44451,12,44481,44507,12,44537,44563,12,44593,44619,12,44649,44675,12,44705,44731,12,44761,44787,12,44817,44843,12,44873,44899,12,44929,44955,12,44985,45011,12,45041,45067,12,45097,45123,12,45153,45179,12,45209,45235,12,45265,45291,12,45321,45347,12,45377,45403,12,45433,45459,12,45489,45515,12,45545,45571,12,45601,45627,12,45657,45683,12,45713,45739,12,45769,45795,12,45825,45851,12,45881,45907,12,45937,45963,12,45993,46019,12,46049,46075,12,46105,46131,12,46161,46187,12,46217,46243,12,46273,46299,12,46329,46355,12,46385,46411,12,46441,46467,12,46497,46523,12,46553,46579,12,46609,46635,12,46665,46691,12,46721,46747,12,46777,46803,12,46833,46859,12,46889,46915,12,46945,46971,12,47001,47027,12,47057,47083,12,47113,47139,12,47169,47195,12,47225,47251,12,47281,47307,12,47337,47363,12,47393,47419,12,47449,47475,12,47505,47531,12,47561,47587,12,47617,47643,12,47673,47699,12,47729,47755,12,47785,47811,12,47841,47867,12,47897,47923,12,47953,47979,12,48009,48035,12,48065,48091,12,48121,48147,12,48177,48203,12,48233,48259,12,48289,48315,12,48345,48371,12,48401,48427,12,48457,48483,12,48513,48539,12,48569,48595,12,48625,48651,12,48681,48707,12,48737,48763,12,48793,48819,12,48849,48875,12,48905,48931,12,48961,48987,12,49017,49043,12,49073,49099,12,49129,49155,12,49185,49211,12,49241,49267,12,49297,49323,12,49353,49379,12,49409,49435,12,49465,49491,12,49521,49547,12,49577,49603,12,49633,49659,12,49689,49715,12,49745,49771,12,49801,49827,12,49857,49883,12,49913,49939,12,49969,49995,12,50025,50051,12,50081,50107,12,50137,50163,12,50193,50219,12,50249,50275,12,50305,50331,12,50361,50387,12,50417,50443,12,50473,50499,12,50529,50555,12,50585,50611,12,50641,50667,12,50697,50723,12,50753,50779,12,50809,50835,12,50865,50891,12,50921,50947,12,50977,51003,12,51033,51059,12,51089,51115,12,51145,51171,12,51201,51227,12,51257,51283,12,51313,51339,12,51369,51395,12,51425,51451,12,51481,51507,12,51537,51563,12,51593,51619,12,51649,51675,12,51705,51731,12,51761,51787,12,51817,51843,12,51873,51899,12,51929,51955,12,51985,52011,12,52041,52067,12,52097,52123,12,52153,52179,12,52209,52235,12,52265,52291,12,52321,52347,12,52377,52403,12,52433,52459,12,52489,52515,12,52545,52571,12,52601,52627,12,52657,52683,12,52713,52739,12,52769,52795,12,52825,52851,12,52881,52907,12,52937,52963,12,52993,53019,12,53049,53075,12,53105,53131,12,53161,53187,12,53217,53243,12,53273,53299,12,53329,53355,12,53385,53411,12,53441,53467,12,53497,53523,12,53553,53579,12,53609,53635,12,53665,53691,12,53721,53747,12,53777,53803,12,53833,53859,12,53889,53915,12,53945,53971,12,54001,54027,12,54057,54083,12,54113,54139,12,54169,54195,12,54225,54251,12,54281,54307,12,54337,54363,12,54393,54419,12,54449,54475,12,54505,54531,12,54561,54587,12,54617,54643,12,54673,54699,12,54729,54755,12,54785,54811,12,54841,54867,12,54897,54923,12,54953,54979,12,55009,55035,12,55065,55091,12,55121,55147,12,55177,55203,12,65024,65039,5,65520,65528,4,66422,66426,5,68152,68154,5,69291,69292,5,69633,69633,5,69747,69748,5,69811,69814,5,69826,69826,5,69932,69932,7,70016,70017,5,70079,70080,7,70095,70095,5,70196,70196,5,70367,70367,5,70402,70403,7,70464,70464,5,70487,70487,5,70709,70711,7,70725,70725,7,70833,70834,7,70843,70844,7,70849,70849,7,71090,71093,5,71103,71104,5,71227,71228,7,71339,71339,5,71344,71349,5,71458,71461,5,71727,71735,5,71985,71989,7,71998,71998,5,72002,72002,7,72154,72155,5,72193,72202,5,72251,72254,5,72281,72283,5,72344,72345,5,72766,72766,7,72874,72880,5,72885,72886,5,73023,73029,5,73104,73105,5,73111,73111,5,92912,92916,5,94095,94098,5,113824,113827,4,119142,119142,7,119155,119162,4,119362,119364,5,121476,121476,5,122888,122904,5,123184,123190,5,125252,125258,5,127183,127183,14,127340,127343,14,127377,127386,14,127491,127503,14,127548,127551,14,127744,127756,14,127761,127761,14,127769,127769,14,127773,127774,14,127780,127788,14,127796,127797,14,127820,127823,14,127869,127869,14,127894,127895,14,127902,127903,14,127943,127943,14,127947,127950,14,127972,127972,14,127988,127988,14,127992,127994,14,128009,128011,14,128019,128019,14,128023,128041,14,128064,128064,14,128102,128107,14,128174,128181,14,128238,128238,14,128246,128247,14,128254,128254,14,128264,128264,14,128278,128299,14,128329,128330,14,128348,128359,14,128371,128377,14,128392,128393,14,128401,128404,14,128421,128421,14,128433,128434,14,128450,128452,14,128476,128478,14,128483,128483,14,128495,128495,14,128506,128506,14,128519,128520,14,128528,128528,14,128534,128534,14,128538,128538,14,128540,128542,14,128544,128549,14,128552,128555,14,128557,128557,14,128560,128563,14,128565,128565,14,128567,128576,14,128581,128591,14,128641,128642,14,128646,128646,14,128648,128648,14,128650,128651,14,128653,128653,14,128655,128655,14,128657,128659,14,128661,128661,14,128663,128663,14,128665,128666,14,128674,128674,14,128676,128677,14,128679,128685,14,128690,128690,14,128694,128694,14,128697,128702,14,128704,128704,14,128710,128714,14,128716,128716,14,128720,128720,14,128723,128724,14,128726,128727,14,128733,128735,14,128742,128744,14,128746,128746,14,128749,128751,14,128753,128754,14,128756,128758,14,128761,128761,14,128763,128764,14,128884,128895,14,128992,129003,14,129008,129008,14,129036,129039,14,129114,129119,14,129198,129279,14,129293,129295,14,129305,129310,14,129312,129319,14,129328,129328,14,129331,129338,14,129343,129343,14,129351,129355,14,129357,129359,14,129375,129387,14,129393,129393,14,129395,129398,14,129401,129401,14,129403,129403,14,129408,129412,14,129426,129431,14,129443,129444,14,129451,129453,14,129456,129465,14,129472,129472,14,129475,129482,14,129484,129484,14,129488,129510,14,129536,129647,14,129652,129652,14,129656,129658,14,129661,129663,14,129667,129670,14,129680,129685,14,129705,129708,14,129712,129718,14,129723,129727,14,129731,129733,14,129744,129750,14,129754,129759,14,129768,129775,14,129783,129791,14,917504,917504,4,917506,917535,4,917632,917759,4,918000,921599,4,0,9,4,11,12,4,14,31,4,169,169,14,174,174,14,1155,1159,5,1425,1469,5,1473,1474,5,1479,1479,5,1552,1562,5,1611,1631,5,1750,1756,5,1759,1764,5,1770,1773,5,1809,1809,5,1958,1968,5,2045,2045,5,2075,2083,5,2089,2093,5,2192,2193,1,2250,2273,5,2275,2306,5,2362,2362,5,2364,2364,5,2369,2376,5,2381,2381,5,2385,2391,5,2433,2433,5,2492,2492,5,2495,2496,7,2503,2504,7,2509,2509,5,2530,2531,5,2561,2562,5,2620,2620,5,2625,2626,5,2635,2637,5,2672,2673,5,2689,2690,5,2748,2748,5,2753,2757,5,2761,2761,7,2765,2765,5,2810,2815,5,2818,2819,7,2878,2878,5,2880,2880,7,2887,2888,7,2893,2893,5,2903,2903,5,2946,2946,5,3007,3007,7,3009,3010,7,3018,3020,7,3031,3031,5,3073,3075,7,3132,3132,5,3137,3140,7,3146,3149,5,3170,3171,5,3202,3203,7,3262,3262,7,3264,3265,7,3267,3268,7,3271,3272,7,3276,3277,5,3298,3299,5,3330,3331,7,3390,3390,5,3393,3396,5,3402,3404,7,3406,3406,1,3426,3427,5,3458,3459,7,3535,3535,5,3538,3540,5,3544,3550,7,3570,3571,7,3635,3635,7,3655,3662,5,3763,3763,7,3784,3789,5,3893,3893,5,3897,3897,5,3953,3966,5,3968,3972,5,3981,3991,5,4038,4038,5,4145,4145,7,4153,4154,5,4157,4158,5,4184,4185,5,4209,4212,5,4228,4228,7,4237,4237,5,4352,4447,8,4520,4607,10,5906,5908,5,5938,5939,5,5970,5971,5,6068,6069,5,6071,6077,5,6086,6086,5,6089,6099,5,6155,6157,5,6159,6159,5,6313,6313,5,6435,6438,7,6441,6443,7,6450,6450,5,6457,6459,5,6681,6682,7,6741,6741,7,6743,6743,7,6752,6752,5,6757,6764,5,6771,6780,5,6832,6845,5,6847,6862,5,6916,6916,7,6965,6965,5,6971,6971,7,6973,6977,7,6979,6980,7,7040,7041,5,7073,7073,7,7078,7079,7,7082,7082,7,7142,7142,5,7144,7145,5,7149,7149,5,7151,7153,5,7204,7211,7,7220,7221,7,7376,7378,5,7393,7393,7,7405,7405,5,7415,7415,7,7616,7679,5,8204,8204,5,8206,8207,4,8233,8233,4,8252,8252,14,8288,8292,4,8294,8303,4,8413,8416,5,8418,8420,5,8482,8482,14,8596,8601,14,8986,8987,14,9096,9096,14,9193,9196,14,9199,9199,14,9201,9202,14,9208,9210,14,9642,9643,14,9664,9664,14,9728,9729,14,9732,9732,14,9735,9741,14,9743,9744,14,9746,9746,14,9750,9751,14,9753,9756,14,9758,9759,14,9761,9761,14,9764,9765,14,9767,9769,14,9771,9773,14,9775,9775,14,9784,9785,14,9787,9791,14,9793,9793,14,9795,9799,14,9812,9822,14,9824,9824,14,9827,9827,14,9829,9830,14,9832,9832,14,9851,9851,14,9854,9854,14,9856,9861,14,9874,9874,14,9876,9876,14,9878,9879,14,9881,9881,14,9883,9884,14,9888,9889,14,9895,9895,14,9898,9899,14,9904,9905,14,9917,9918,14,9924,9925,14,9928,9928,14,9934,9934,14,9936,9936,14,9938,9938,14,9940,9940,14,9961,9961,14,9963,9967,14,9970,9971,14,9973,9973,14,9975,9977,14,9979,9980,14,9982,9985,14,9987,9988,14,9992,9996,14,9998,9998,14,10000,10001,14,10004,10004,14,10013,10013,14,10024,10024,14,10052,10052,14,10060,10060,14,10067,10069,14,10083,10083,14,10085,10087,14,10145,10145,14,10175,10175,14,11013,11015,14,11088,11088,14,11503,11505,5,11744,11775,5,12334,12335,5,12349,12349,14,12951,12951,14,42607,42607,5,42612,42621,5,42736,42737,5,43014,43014,5,43043,43044,7,43047,43047,7,43136,43137,7,43204,43205,5,43263,43263,5,43335,43345,5,43360,43388,8,43395,43395,7,43444,43445,7,43450,43451,7,43454,43456,7,43561,43566,5,43569,43570,5,43573,43574,5,43596,43596,5,43644,43644,5,43698,43700,5,43710,43711,5,43755,43755,7,43758,43759,7,43766,43766,5,44005,44005,5,44008,44008,5,44012,44012,7,44032,44032,11,44060,44060,11,44088,44088,11,44116,44116,11,44144,44144,11,44172,44172,11,44200,44200,11,44228,44228,11,44256,44256,11,44284,44284,11,44312,44312,11,44340,44340,11,44368,44368,11,44396,44396,11,44424,44424,11,44452,44452,11,44480,44480,11,44508,44508,11,44536,44536,11,44564,44564,11,44592,44592,11,44620,44620,11,44648,44648,11,44676,44676,11,44704,44704,11,44732,44732,11,44760,44760,11,44788,44788,11,44816,44816,11,44844,44844,11,44872,44872,11,44900,44900,11,44928,44928,11,44956,44956,11,44984,44984,11,45012,45012,11,45040,45040,11,45068,45068,11,45096,45096,11,45124,45124,11,45152,45152,11,45180,45180,11,45208,45208,11,45236,45236,11,45264,45264,11,45292,45292,11,45320,45320,11,45348,45348,11,45376,45376,11,45404,45404,11,45432,45432,11,45460,45460,11,45488,45488,11,45516,45516,11,45544,45544,11,45572,45572,11,45600,45600,11,45628,45628,11,45656,45656,11,45684,45684,11,45712,45712,11,45740,45740,11,45768,45768,11,45796,45796,11,45824,45824,11,45852,45852,11,45880,45880,11,45908,45908,11,45936,45936,11,45964,45964,11,45992,45992,11,46020,46020,11,46048,46048,11,46076,46076,11,46104,46104,11,46132,46132,11,46160,46160,11,46188,46188,11,46216,46216,11,46244,46244,11,46272,46272,11,46300,46300,11,46328,46328,11,46356,46356,11,46384,46384,11,46412,46412,11,46440,46440,11,46468,46468,11,46496,46496,11,46524,46524,11,46552,46552,11,46580,46580,11,46608,46608,11,46636,46636,11,46664,46664,11,46692,46692,11,46720,46720,11,46748,46748,11,46776,46776,11,46804,46804,11,46832,46832,11,46860,46860,11,46888,46888,11,46916,46916,11,46944,46944,11,46972,46972,11,47000,47000,11,47028,47028,11,47056,47056,11,47084,47084,11,47112,47112,11,47140,47140,11,47168,47168,11,47196,47196,11,47224,47224,11,47252,47252,11,47280,47280,11,47308,47308,11,47336,47336,11,47364,47364,11,47392,47392,11,47420,47420,11,47448,47448,11,47476,47476,11,47504,47504,11,47532,47532,11,47560,47560,11,47588,47588,11,47616,47616,11,47644,47644,11,47672,47672,11,47700,47700,11,47728,47728,11,47756,47756,11,47784,47784,11,47812,47812,11,47840,47840,11,47868,47868,11,47896,47896,11,47924,47924,11,47952,47952,11,47980,47980,11,48008,48008,11,48036,48036,11,48064,48064,11,48092,48092,11,48120,48120,11,48148,48148,11,48176,48176,11,48204,48204,11,48232,48232,11,48260,48260,11,48288,48288,11,48316,48316,11,48344,48344,11,48372,48372,11,48400,48400,11,48428,48428,11,48456,48456,11,48484,48484,11,48512,48512,11,48540,48540,11,48568,48568,11,48596,48596,11,48624,48624,11,48652,48652,11,48680,48680,11,48708,48708,11,48736,48736,11,48764,48764,11,48792,48792,11,48820,48820,11,48848,48848,11,48876,48876,11,48904,48904,11,48932,48932,11,48960,48960,11,48988,48988,11,49016,49016,11,49044,49044,11,49072,49072,11,49100,49100,11,49128,49128,11,49156,49156,11,49184,49184,11,49212,49212,11,49240,49240,11,49268,49268,11,49296,49296,11,49324,49324,11,49352,49352,11,49380,49380,11,49408,49408,11,49436,49436,11,49464,49464,11,49492,49492,11,49520,49520,11,49548,49548,11,49576,49576,11,49604,49604,11,49632,49632,11,49660,49660,11,49688,49688,11,49716,49716,11,49744,49744,11,49772,49772,11,49800,49800,11,49828,49828,11,49856,49856,11,49884,49884,11,49912,49912,11,49940,49940,11,49968,49968,11,49996,49996,11,50024,50024,11,50052,50052,11,50080,50080,11,50108,50108,11,50136,50136,11,50164,50164,11,50192,50192,11,50220,50220,11,50248,50248,11,50276,50276,11,50304,50304,11,50332,50332,11,50360,50360,11,50388,50388,11,50416,50416,11,50444,50444,11,50472,50472,11,50500,50500,11,50528,50528,11,50556,50556,11,50584,50584,11,50612,50612,11,50640,50640,11,50668,50668,11,50696,50696,11,50724,50724,11,50752,50752,11,50780,50780,11,50808,50808,11,50836,50836,11,50864,50864,11,50892,50892,11,50920,50920,11,50948,50948,11,50976,50976,11,51004,51004,11,51032,51032,11,51060,51060,11,51088,51088,11,51116,51116,11,51144,51144,11,51172,51172,11,51200,51200,11,51228,51228,11,51256,51256,11,51284,51284,11,51312,51312,11,51340,51340,11,51368,51368,11,51396,51396,11,51424,51424,11,51452,51452,11,51480,51480,11,51508,51508,11,51536,51536,11,51564,51564,11,51592,51592,11,51620,51620,11,51648,51648,11,51676,51676,11,51704,51704,11,51732,51732,11,51760,51760,11,51788,51788,11,51816,51816,11,51844,51844,11,51872,51872,11,51900,51900,11,51928,51928,11,51956,51956,11,51984,51984,11,52012,52012,11,52040,52040,11,52068,52068,11,52096,52096,11,52124,52124,11,52152,52152,11,52180,52180,11,52208,52208,11,52236,52236,11,52264,52264,11,52292,52292,11,52320,52320,11,52348,52348,11,52376,52376,11,52404,52404,11,52432,52432,11,52460,52460,11,52488,52488,11,52516,52516,11,52544,52544,11,52572,52572,11,52600,52600,11,52628,52628,11,52656,52656,11,52684,52684,11,52712,52712,11,52740,52740,11,52768,52768,11,52796,52796,11,52824,52824,11,52852,52852,11,52880,52880,11,52908,52908,11,52936,52936,11,52964,52964,11,52992,52992,11,53020,53020,11,53048,53048,11,53076,53076,11,53104,53104,11,53132,53132,11,53160,53160,11,53188,53188,11,53216,53216,11,53244,53244,11,53272,53272,11,53300,53300,11,53328,53328,11,53356,53356,11,53384,53384,11,53412,53412,11,53440,53440,11,53468,53468,11,53496,53496,11,53524,53524,11,53552,53552,11,53580,53580,11,53608,53608,11,53636,53636,11,53664,53664,11,53692,53692,11,53720,53720,11,53748,53748,11,53776,53776,11,53804,53804,11,53832,53832,11,53860,53860,11,53888,53888,11,53916,53916,11,53944,53944,11,53972,53972,11,54000,54000,11,54028,54028,11,54056,54056,11,54084,54084,11,54112,54112,11,54140,54140,11,54168,54168,11,54196,54196,11,54224,54224,11,54252,54252,11,54280,54280,11,54308,54308,11,54336,54336,11,54364,54364,11,54392,54392,11,54420,54420,11,54448,54448,11,54476,54476,11,54504,54504,11,54532,54532,11,54560,54560,11,54588,54588,11,54616,54616,11,54644,54644,11,54672,54672,11,54700,54700,11,54728,54728,11,54756,54756,11,54784,54784,11,54812,54812,11,54840,54840,11,54868,54868,11,54896,54896,11,54924,54924,11,54952,54952,11,54980,54980,11,55008,55008,11,55036,55036,11,55064,55064,11,55092,55092,11,55120,55120,11,55148,55148,11,55176,55176,11,55216,55238,9,64286,64286,5,65056,65071,5,65438,65439,5,65529,65531,4,66272,66272,5,68097,68099,5,68108,68111,5,68159,68159,5,68900,68903,5,69446,69456,5,69632,69632,7,69634,69634,7,69744,69744,5,69759,69761,5,69808,69810,7,69815,69816,7,69821,69821,1,69837,69837,1,69927,69931,5,69933,69940,5,70003,70003,5,70018,70018,7,70070,70078,5,70082,70083,1,70094,70094,7,70188,70190,7,70194,70195,7,70197,70197,7,70206,70206,5,70368,70370,7,70400,70401,5,70459,70460,5,70463,70463,7,70465,70468,7,70475,70477,7,70498,70499,7,70512,70516,5,70712,70719,5,70722,70724,5,70726,70726,5,70832,70832,5,70835,70840,5,70842,70842,5,70845,70845,5,70847,70848,5,70850,70851,5,71088,71089,7,71096,71099,7,71102,71102,7,71132,71133,5,71219,71226,5,71229,71229,5,71231,71232,5,71340,71340,7,71342,71343,7,71350,71350,7,71453,71455,5,71462,71462,7,71724,71726,7,71736,71736,7,71984,71984,5,71991,71992,7,71997,71997,7,71999,71999,1,72001,72001,1,72003,72003,5,72148,72151,5,72156,72159,7,72164,72164,7,72243,72248,5,72250,72250,1,72263,72263,5,72279,72280,7,72324,72329,1,72343,72343,7,72751,72751,7,72760,72765,5,72767,72767,5,72873,72873,7,72881,72881,7,72884,72884,7,73009,73014,5,73020,73021,5,73030,73030,1,73098,73102,7,73107,73108,7,73110,73110,7,73459,73460,5,78896,78904,4,92976,92982,5,94033,94087,7,94180,94180,5,113821,113822,5,118528,118573,5,119141,119141,5,119143,119145,5,119150,119154,5,119163,119170,5,119210,119213,5,121344,121398,5,121461,121461,5,121499,121503,5,122880,122886,5,122907,122913,5,122918,122922,5,123566,123566,5,125136,125142,5,126976,126979,14,126981,127182,14,127184,127231,14,127279,127279,14,127344,127345,14,127374,127374,14,127405,127461,14,127489,127490,14,127514,127514,14,127538,127546,14,127561,127567,14,127570,127743,14,127757,127758,14,127760,127760,14,127762,127762,14,127766,127768,14,127770,127770,14,127772,127772,14,127775,127776,14,127778,127779,14,127789,127791,14,127794,127795,14,127798,127798,14,127819,127819,14,127824,127824,14,127868,127868,14,127870,127871,14,127892,127893,14,127896,127896,14,127900,127901,14,127904,127940,14,127942,127942,14,127944,127944,14,127946,127946,14,127951,127955,14,127968,127971,14,127973,127984,14,127987,127987,14,127989,127989,14,127991,127991,14,127995,127999,5,128008,128008,14,128012,128014,14,128017,128018,14,128020,128020,14,128022,128022,14,128042,128042,14,128063,128063,14,128065,128065,14,128101,128101,14,128108,128109,14,128173,128173,14,128182,128183,14,128236,128237,14,128239,128239,14,128245,128245,14,128248,128248,14,128253,128253,14,128255,128258,14,128260,128263,14,128265,128265,14,128277,128277,14,128300,128301,14,128326,128328,14,128331,128334,14,128336,128347,14,128360,128366,14,128369,128370,14,128378,128378,14,128391,128391,14,128394,128397,14,128400,128400,14,128405,128406,14,128420,128420,14,128422,128423,14,128425,128432,14,128435,128443,14,128445,128449,14,128453,128464,14,128468,128475,14,128479,128480,14,128482,128482,14,128484,128487,14,128489,128494,14,128496,128498,14,128500,128505,14,128507,128511,14,128513,128518,14,128521,128525,14,128527,128527,14,128529,128529,14,128533,128533,14,128535,128535,14,128537,128537,14]');
1152 }
1153 > strings.ts
1154 > //#endregion
1155 >
1156 > /**
1157 > * Computes the offset after performing a left delete on the given string,
1158 > * while considering unicode grapheme/emoji rules.
1159 > */
1160 > export function getLeftDeleteOffset(offset: number, str: string): number {
1161 if (offset === 0) {
1162 return 0;
1174 return iterator.offset;
1175 }
1176 > strings.ts
1177 function getOffsetBeforeLastEmojiComponent(initialOffset: number, str: string): number | undefined {
1178 // See https://www.unicode.org/reports/tr51/tr51-14.html#EBNF_and_Regex for the
1210 return resultOffset;
1211 }
1212 > strings.ts
1213 function isEmojiModifier(codePoint: number): boolean {
1214 return 0x1F3FB <= codePoint && codePoint <= 0x1F3FF;
1215 }
1216 > strings.ts
1217 > const enum CodePoint {
1218 > zwj = 0x200D,
1219 >
1220 > /**
1221 > * Variation Selector-16 (VS16)
1222 > */
1223 > emojiVariantSelector = 0xFE0F,
1224 >
1225 > /**
1226 > * Combining Enclosing Keycap
1227 > */
1228 > enclosingKeyCap = 0x20E3,
1229 >
1230 > space = 0x0020,
1231 > }
1232 >
1233 > export const noBreakWhitespace = '\xa0';
1234 >
1235 > export class AmbiguousCharacters {
1236 > private static readonly ambiguousCharacterData = new Lazy<
1237 > Record<
1238 > string | '_common' | '_default',
1239 > /* code point -> ascii code point */ number[]
1240 > >
1241 > >(() => {
1242 // Generated using https://github.com/hediet/vscode-unicode-data
1243 // Stored as key1, value1, key2, value2, ...
1245 '{\"_common\":[8232,32,8233,32,5760,32,8192,32,8193,32,8194,32,8195,32,8196,32,8197,32,8198,32,8200,32,8201,32,8202,32,8287,32,8199,32,8239,32,2042,95,65101,95,65102,95,65103,95,8208,45,8209,45,8210,45,65112,45,1748,45,8259,45,727,45,8722,45,10134,45,11450,45,1549,44,1643,44,184,44,42233,44,894,59,2307,58,2691,58,1417,58,1795,58,1796,58,5868,58,65072,58,6147,58,6153,58,8282,58,1475,58,760,58,42889,58,8758,58,720,58,42237,58,451,33,11601,33,660,63,577,63,2429,63,5038,63,42731,63,119149,46,8228,46,1793,46,1794,46,42510,46,68176,46,1632,46,1776,46,42232,46,1373,96,65287,96,8219,96,1523,96,8242,96,1370,96,8175,96,65344,96,900,96,8189,96,8125,96,8127,96,8190,96,697,96,884,96,712,96,714,96,715,96,756,96,699,96,701,96,700,96,702,96,42892,96,1497,96,2036,96,2037,96,5194,96,5836,96,94033,96,94034,96,65339,91,10088,40,10098,40,12308,40,64830,40,65341,93,10089,41,10099,41,12309,41,64831,41,10100,123,119060,123,10101,125,65342,94,8270,42,1645,42,8727,42,66335,42,5941,47,8257,47,8725,47,8260,47,9585,47,10187,47,10744,47,119354,47,12755,47,12339,47,11462,47,20031,47,12035,47,65340,92,65128,92,8726,92,10189,92,10741,92,10745,92,119311,92,119355,92,12756,92,20022,92,12034,92,42872,38,708,94,710,94,5869,43,10133,43,66203,43,8249,60,10094,60,706,60,119350,60,5176,60,5810,60,5120,61,11840,61,12448,61,42239,61,8250,62,10095,62,707,62,119351,62,5171,62,94015,62,8275,126,732,126,8128,126,8764,126,65372,124,65293,45,118002,50,120784,50,120794,50,120804,50,120814,50,120824,50,130034,50,42842,50,423,50,1000,50,42564,50,5311,50,42735,50,119302,51,118003,51,120785,51,120795,51,120805,51,120815,51,120825,51,130035,51,42923,51,540,51,439,51,42858,51,11468,51,1248,51,94011,51,71882,51,118004,52,120786,52,120796,52,120806,52,120816,52,120826,52,130036,52,5070,52,71855,52,118005,53,120787,53,120797,53,120807,53,120817,53,120827,53,130037,53,444,53,71867,53,118006,54,120788,54,120798,54,120808,54,120818,54,120828,54,130038,54,11474,54,5102,54,71893,54,119314,55,118007,55,120789,55,120799,55,120809,55,120819,55,120829,55,130039,55,66770,55,71878,55,2819,56,2538,56,2666,56,125131,56,118008,56,120790,56,120800,56,120810,56,120820,56,120830,56,130040,56,547,56,546,56,66330,56,2663,57,2920,57,2541,57,3437,57,118009,57,120791,57,120801,57,120811,57,120821,57,120831,57,130041,57,42862,57,11466,57,71884,57,71852,57,71894,57,9082,97,65345,97,119834,97,119886,97,119938,97,119990,97,120042,97,120094,97,120146,97,120198,97,120250,97,120302,97,120354,97,120406,97,120458,97,593,97,945,97,120514,97,120572,97,120630,97,120688,97,120746,97,65313,65,117974,65,119808,65,119860,65,119912,65,119964,65,120016,65,120068,65,120120,65,120172,65,120224,65,120276,65,120328,65,120380,65,120432,65,913,65,120488,65,120546,65,120604,65,120662,65,120720,65,5034,65,5573,65,42222,65,94016,65,66208,65,119835,98,119887,98,119939,98,119991,98,120043,98,120095,98,120147,98,120199,98,120251,98,120303,98,120355,98,120407,98,120459,98,388,98,5071,98,5234,98,5551,98,65314,66,8492,66,117975,66,119809,66,119861,66,119913,66,120017,66,120069,66,120121,66,120173,66,120225,66,120277,66,120329,66,120381,66,120433,66,42932,66,914,66,120489,66,120547,66,120605,66,120663,66,120721,66,5108,66,5623,66,42192,66,66178,66,66209,66,66305,66,65347,99,8573,99,119836,99,119888,99,119940,99,119992,99,120044,99,120096,99,120148,99,120200,99,120252,99,120304,99,120356,99,120408,99,120460,99,7428,99,1010,99,11429,99,43951,99,66621,99,128844,67,71913,67,71922,67,65315,67,8557,67,8450,67,8493,67,117976,67,119810,67,119862,67,119914,67,119966,67,120018,67,120174,67,120226,67,120278,67,120330,67,120382,67,120434,67,1017,67,11428,67,5087,67,42202,67,66210,67,66306,67,66581,67,66844,67,8574,100,8518,100,119837,100,119889,100,119941,100,119993,100,120045,100,120097,100,120149,100,120201,100,120253,100,120305,100,120357,100,120409,100,120461,100,1281,100,5095,100,5231,100,42194,100,8558,68,8517,68,117977,68,119811,68,119863,68,119915,68,119967,68,120019,68,120071,68,120123,68,120175,68,120227,68,120279,68,120331,68,120383,68,120435,68,5024,68,5598,68,5610,68,42195,68,8494,101,65349,101,8495,101,8519,101,119838,101,119890,101,119942,101,120046,101,120098,101,120150,101,120202,101,120254,101,120306,101,120358,101,120410,101,120462,101,43826,101,1213,101,8959,69,65317,69,8496,69,117978,69,119812,69,119864,69,119916,69,120020,69,120072,69,120124,69,120176,69,120228,69,120280,69,120332,69,120384,69,120436,69,917,69,120492,69,120550,69,120608,69,120666,69,120724,69,11577,69,5036,69,42224,69,71846,69,71854,69,66182,69,119839,102,119891,102,119943,102,119995,102,120047,102,120099,102,120151,102,120203,102,120255,102,120307,102,120359,102,120411,102,120463,102,43829,102,42905,102,383,102,7837,102,1412,102,119315,70,8497,70,117979,70,119813,70,119865,70,119917,70,120021,70,120073,70,120125,70,120177,70,120229,70,120281,70,120333,70,120385,70,120437,70,42904,70,988,70,120778,70,5556,70,42205,70,71874,70,71842,70,66183,70,66213,70,66853,70,65351,103,8458,103,119840,103,119892,103,119944,103,120048,103,120100,103,120152,103,120204,103,120256,103,120308,103,120360,103,120412,103,120464,103,609,103,7555,103,397,103,1409,103,117980,71,119814,71,119866,71,119918,71,119970,71,120022,71,120074,71,120126,71,120178,71,120230,71,120282,71,120334,71,120386,71,120438,71,1292,71,5056,71,5107,71,42198,71,65352,104,8462,104,119841,104,119945,104,119997,104,120049,104,120101,104,120153,104,120205,104,120257,104,120309,104,120361,104,120413,104,120465,104,1211,104,1392,104,5058,104,65320,72,8459,72,8460,72,8461,72,117981,72,119815,72,119867,72,119919,72,120023,72,120179,72,120231,72,120283,72,120335,72,120387,72,120439,72,919,72,120494,72,120552,72,120610,72,120668,72,120726,72,11406,72,5051,72,5500,72,42215,72,66255,72,731,105,9075,105,65353,105,8560,105,8505,105,8520,105,119842,105,119894,105,119946,105,119998,105,120050,105,120102,105,120154,105,120206,105,120258,105,120310,105,120362,105,120414,105,120466,105,120484,105,618,105,617,105,953,105,8126,105,890,105,120522,105,120580,105,120638,105,120696,105,120754,105,1110,105,42567,105,1231,105,43893,105,5029,105,71875,105,65354,106,8521,106,119843,106,119895,106,119947,106,119999,106,120051,106,120103,106,120155,106,120207,106,120259,106,120311,106,120363,106,120415,106,120467,106,1011,106,1112,106,65322,74,117983,74,119817,74,119869,74,119921,74,119973,74,120025,74,120077,74,120129,74,120181,74,120233,74,120285,74,120337,74,120389,74,120441,74,42930,74,895,74,1032,74,5035,74,5261,74,42201,74,119844,107,119896,107,119948,107,120000,107,120052,107,120104,107,120156,107,120208,107,120260,107,120312,107,120364,107,120416,107,120468,107,8490,75,65323,75,117984,75,119818,75,119870,75,119922,75,119974,75,120026,75,120078,75,120130,75,120182,75,120234,75,120286,75,120338,75,120390,75,120442,75,922,75,120497,75,120555,75,120613,75,120671,75,120729,75,11412,75,5094,75,5845,75,42199,75,66840,75,1472,108,8739,73,9213,73,65512,73,1633,108,1777,73,66336,108,125127,108,118001,108,120783,73,120793,73,120803,73,120813,73,120823,73,130033,73,65321,73,8544,73,8464,73,8465,73,117982,108,119816,73,119868,73,119920,73,120024,73,120128,73,120180,73,120232,73,120284,73,120336,73,120388,73,120440,73,65356,108,8572,73,8467,108,119845,108,119897,108,119949,108,120001,108,120053,108,120105,73,120157,73,120209,73,120261,73,120313,73,120365,73,120417,73,120469,73,448,73,120496,73,120554,73,120612,73,120670,73,120728,73,11410,73,1030,73,1216,73,1493,108,1503,108,1575,108,126464,108,126592,108,65166,108,65165,108,1994,108,11599,73,5825,73,42226,73,93992,73,66186,124,66313,124,119338,76,8556,76,8466,76,117985,76,119819,76,119871,76,119923,76,120027,76,120079,76,120131,76,120183,76,120235,76,120287,76,120339,76,120391,76,120443,76,11472,76,5086,76,5290,76,42209,76,93974,76,71843,76,71858,76,66587,76,66854,76,65325,77,8559,77,8499,77,117986,77,119820,77,119872,77,119924,77,120028,77,120080,77,120132,77,120184,77,120236,77,120288,77,120340,77,120392,77,120444,77,924,77,120499,77,120557,77,120615,77,120673,77,120731,77,1018,77,11416,77,5047,77,5616,77,5846,77,42207,77,66224,77,66321,77,119847,110,119899,110,119951,110,120003,110,120055,110,120107,110,120159,110,120211,110,120263,110,120315,110,120367,110,120419,110,120471,110,1400,110,1404,110,65326,78,8469,78,117987,78,119821,78,119873,78,119925,78,119977,78,120029,78,120081,78,120185,78,120237,78,120289,78,120341,78,120393,78,120445,78,925,78,120500,78,120558,78,120616,78,120674,78,120732,78,11418,78,42208,78,66835,78,3074,111,3202,111,3330,111,3458,111,2406,111,2662,111,2790,111,3046,111,3174,111,3302,111,3430,111,3664,111,3792,111,4160,111,1637,111,1781,111,65359,111,8500,111,119848,111,119900,111,119952,111,120056,111,120108,111,120160,111,120212,111,120264,111,120316,111,120368,111,120420,111,120472,111,7439,111,7441,111,43837,111,959,111,120528,111,120586,111,120644,111,120702,111,120760,111,963,111,120532,111,120590,111,120648,111,120706,111,120764,111,11423,111,4351,111,1413,111,1505,111,1607,111,126500,111,126564,111,126596,111,65259,111,65260,111,65258,111,65257,111,1726,111,64428,111,64429,111,64427,111,64426,111,1729,111,64424,111,64425,111,64423,111,64422,111,1749,111,3360,111,4125,111,66794,111,71880,111,71895,111,66604,111,1984,79,2534,79,2918,79,12295,79,70864,79,71904,79,118000,79,120782,79,120792,79,120802,79,120812,79,120822,79,130032,79,65327,79,117988,79,119822,79,119874,79,119926,79,119978,79,120030,79,120082,79,120134,79,120186,79,120238,79,120290,79,120342,79,120394,79,120446,79,927,79,120502,79,120560,79,120618,79,120676,79,120734,79,11422,79,1365,79,11604,79,4816,79,2848,79,66754,79,42227,79,71861,79,66194,79,66219,79,66564,79,66838,79,9076,112,65360,112,119849,112,119901,112,119953,112,120005,112,120057,112,120109,112,120161,112,120213,112,120265,112,120317,112,120369,112,120421,112,120473,112,961,112,120530,112,120544,112,120588,112,120602,112,120646,112,120660,112,120704,112,120718,112,120762,112,120776,112,11427,112,65328,80,8473,80,117989,80,119823,80,119875,80,119927,80,119979,80,120031,80,120083,80,120187,80,120239,80,120291,80,120343,80,120395,80,120447,80,929,80,120504,80,120562,80,120620,80,120678,80,120736,80,11426,80,5090,80,5229,80,42193,80,66197,80,119850,113,119902,113,119954,113,120006,113,120058,113,120110,113,120162,113,120214,113,120266,113,120318,113,120370,113,120422,113,120474,113,1307,113,1379,113,1382,113,8474,81,117990,81,119824,81,119876,81,119928,81,119980,81,120032,81,120084,81,120188,81,120240,81,120292,81,120344,81,120396,81,120448,81,11605,81,119851,114,119903,114,119955,114,120007,114,120059,114,120111,114,120163,114,120215,114,120267,114,120319,114,120371,114,120423,114,120475,114,43847,114,43848,114,7462,114,11397,114,43905,114,119318,82,8475,82,8476,82,8477,82,117991,82,119825,82,119877,82,119929,82,120033,82,120189,82,120241,82,120293,82,120345,82,120397,82,120449,82,422,82,5025,82,5074,82,66740,82,5511,82,42211,82,94005,82,65363,115,119852,115,119904,115,119956,115,120008,115,120060,115,120112,115,120164,115,120216,115,120268,115,120320,115,120372,115,120424,115,120476,115,42801,115,445,115,1109,115,43946,115,71873,115,66632,115,65331,83,117992,83,119826,83,119878,83,119930,83,119982,83,120034,83,120086,83,120138,83,120190,83,120242,83,120294,83,120346,83,120398,83,120450,83,1029,83,1359,83,5077,83,5082,83,42210,83,94010,83,66198,83,66592,83,119853,116,119905,116,119957,116,120009,116,120061,116,120113,116,120165,116,120217,116,120269,116,120321,116,120373,116,120425,116,120477,116,8868,84,10201,84,128872,84,65332,84,117993,84,119827,84,119879,84,119931,84,119983,84,120035,84,120087,84,120139,84,120191,84,120243,84,120295,84,120347,84,120399,84,120451,84,932,84,120507,84,120565,84,120623,84,120681,84,120739,84,11430,84,5026,84,42196,84,93962,84,71868,84,66199,84,66225,84,66325,84,119854,117,119906,117,119958,117,120010,117,120062,117,120114,117,120166,117,120218,117,120270,117,120322,117,120374,117,120426,117,120478,117,42911,117,7452,117,43854,117,43858,117,651,117,965,117,120534,117,120592,117,120650,117,120708,117,120766,117,1405,117,66806,117,71896,117,8746,85,8899,85,117994,85,119828,85,119880,85,119932,85,119984,85,120036,85,120088,85,120140,85,120192,85,120244,85,120296,85,120348,85,120400,85,120452,85,1357,85,4608,85,66766,85,5196,85,42228,85,94018,85,71864,85,8744,118,8897,118,65366,118,8564,118,119855,118,119907,118,119959,118,120011,118,120063,118,120115,118,120167,118,120219,118,120271,118,120323,118,120375,118,120427,118,120479,118,7456,118,957,118,120526,118,120584,118,120642,118,120700,118,120758,118,1141,118,1496,118,71430,118,43945,118,71872,118,119309,86,1639,86,1783,86,8548,86,117995,86,119829,86,119881,86,119933,86,119985,86,120037,86,120089,86,120141,86,120193,86,120245,86,120297,86,120349,86,120401,86,120453,86,1140,86,11576,86,5081,86,5167,86,42719,86,42214,86,93960,86,71840,86,66845,86,623,119,119856,119,119908,119,119960,119,120012,119,120064,119,120116,119,120168,119,120220,119,120272,119,120324,119,120376,119,120428,119,120480,119,7457,119,1121,119,1309,119,1377,119,71434,119,71438,119,71439,119,43907,119,71910,87,71919,87,117996,87,119830,87,119882,87,119934,87,119986,87,120038,87,120090,87,120142,87,120194,87,120246,87,120298,87,120350,87,120402,87,120454,87,1308,87,5043,87,5076,87,42218,87,5742,120,10539,120,10540,120,10799,120,65368,120,8569,120,119857,120,119909,120,119961,120,120013,120,120065,120,120117,120,120169,120,120221,120,120273,120,120325,120,120377,120,120429,120,120481,120,5441,120,5501,120,5741,88,9587,88,66338,88,71916,88,65336,88,8553,88,117997,88,119831,88,119883,88,119935,88,119987,88,120039,88,120091,88,120143,88,120195,88,120247,88,120299,88,120351,88,120403,88,120455,88,42931,88,935,88,120510,88,120568,88,120626,88,120684,88,120742,88,11436,88,11613,88,5815,88,42219,88,66192,88,66228,88,66327,88,66855,88,611,121,7564,121,65369,121,119858,121,119910,121,119962,121,120014,121,120066,121,120118,121,120170,121,120222,121,120274,121,120326,121,120378,121,120430,121,120482,121,655,121,7935,121,43866,121,947,121,8509,121,120516,121,120574,121,120632,121,120690,121,120748,121,1199,121,4327,121,71900,121,65337,89,117998,89,119832,89,119884,89,119936,89,119988,89,120040,89,120092,89,120144,89,120196,89,120248,89,120300,89,120352,89,120404,89,120456,89,933,89,978,89,120508,89,120566,89,120624,89,120682,89,120740,89,11432,89,1198,89,5033,89,5053,89,42220,89,94019,89,71844,89,66226,89,119859,122,119911,122,119963,122,120015,122,120067,122,120119,122,120171,122,120223,122,120275,122,120327,122,120379,122,120431,122,120483,122,7458,122,43923,122,71876,122,71909,90,66293,90,65338,90,8484,90,8488,90,117999,90,119833,90,119885,90,119937,90,119989,90,120041,90,120197,90,120249,90,120301,90,120353,90,120405,90,120457,90,918,90,120493,90,120551,90,120609,90,120667,90,120725,90,5059,90,42204,90,71849,90,65282,34,65283,35,65284,36,65285,37,65286,38,65290,42,65291,43,65294,46,65295,47,65296,48,65298,50,65299,51,65300,52,65301,53,65302,54,65303,55,65304,56,65305,57,65308,60,65309,61,65310,62,65312,64,65316,68,65318,70,65319,71,65324,76,65329,81,65330,82,65333,85,65334,86,65335,87,65343,95,65346,98,65348,100,65350,102,65355,107,65357,109,65358,110,65361,113,65362,114,65364,116,65365,117,65367,119,65370,122,65371,123,65373,125,119846,109],\"_default\":[160,32,8211,45,65374,126,8218,44,65306,58,65281,33,8216,96,8217,96,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"cs\":[65374,126,8218,44,65306,58,65281,33,8216,96,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"de\":[65374,126,65306,58,65281,33,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"es\":[8211,45,65374,126,8218,44,65306,58,65281,33,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"fr\":[65374,126,8218,44,65306,58,65281,33,8216,96,8245,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"it\":[160,32,8211,45,65374,126,8218,44,65306,58,65281,33,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"ja\":[8211,45,8218,44,65281,33,8216,96,8245,96,180,96,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65292,44,65297,49,65307,59],\"ko\":[8211,45,65374,126,8218,44,65306,58,65281,33,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"pl\":[65374,126,65306,58,65281,33,8216,96,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"pt-BR\":[65374,126,8218,44,65306,58,65281,33,8216,96,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"qps-ploc\":[160,32,8211,45,65374,126,8218,44,65306,58,65281,33,8216,96,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"ru\":[65374,126,8218,44,65306,58,65281,33,8216,96,8245,96,180,96,12494,47,305,105,921,73,1009,112,215,120,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"tr\":[160,32,8211,45,65374,126,8218,44,65306,58,65281,33,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65288,40,65289,41,65292,44,65297,49,65307,59,65311,63],\"zh-hans\":[160,32,65374,126,8218,44,8245,96,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89,65297,49],\"zh-hant\":[8211,45,65374,126,8218,44,180,96,12494,47,1047,51,1073,54,1072,97,1040,65,1068,98,1042,66,1089,99,1057,67,1077,101,1045,69,1053,72,305,105,1050,75,921,73,1052,77,1086,111,1054,79,1009,112,1088,112,1056,80,1075,114,1058,84,215,120,1093,120,1061,88,1091,121,1059,89]}'
1246 );
1247 > }); strings.ts
1248 >
1249 > private static readonly cache = new LRUCachedFunction<string, AmbiguousCharacters>((localesStr) => {
1250 const locales = localesStr.split(',');
1251
1304
1305 return new AmbiguousCharacters(map);
1306 > }); strings.ts
1307 >
1308 > public static getInstance(locales: Iterable<string>): AmbiguousCharacters {
1309 return AmbiguousCharacters.cache.get(Array.from(locales).join(','));
1310 }
1311 > strings.ts
1312 > private static _locales = new Lazy<string[]>(() =>
1313 Object.keys(AmbiguousCharacters.ambiguousCharacterData.value).filter(
1314 (k) => !k.startsWith('_')
1315 )
1316 > ); strings.ts
1317 > public static getLocales(): string[] {
1318 return AmbiguousCharacters._locales.value;
1319 }
1320 > strings.ts
1321 > private constructor(
1322 private readonly confusableDictionary: Map<number, number>
1323 ) { }
1324 > strings.ts
1325 > public isAmbiguous(codePoint: number): boolean {
1326 return this.confusableDictionary.has(codePoint);
1327 }
1328 > strings.ts
1329 > public containsAmbiguousCharacter(str: string): boolean {
1330 for (let i = 0; i < str.length; i++) {
1331 const codePoint = str.codePointAt(i);
1336 return false;
1337 }
1338 > strings.ts
1339 > /**
1340 > * Returns the non basic ASCII code point that the given code point can be confused,
1341 > * or undefined if such code point does note exist.
1342 > */
1343 > public getPrimaryConfusable(codePoint: number): number | undefined {
1344 return this.confusableDictionary.get(codePoint);
1345 }
1346 > strings.ts
1347 > public getConfusableCodePoints(): ReadonlySet<number> {
1348 return new Set(this.confusableDictionary.keys());
1349 }
1350 > } strings.ts
1351 >
1352 > export class InvisibleCharacters {
1353 > private static getRawData(): Record<string | '_common', number[]> {
1354 // Generated using https://github.com/hediet/vscode-unicode-data
1355 return JSON.parse('{\"_common\":[11,12,13,127,847,1564,4447,4448,6068,6069,6155,6156,6157,6158,7355,7356,8192,8193,8194,8195,8196,8197,8198,8199,8200,8201,8202,8204,8205,8206,8207,8234,8235,8236,8237,8238,8239,8287,8288,8289,8290,8291,8292,8293,8294,8295,8296,8297,8298,8299,8300,8301,8302,8303,10240,12644,65024,65025,65026,65027,65028,65029,65030,65031,65032,65033,65034,65035,65036,65037,65038,65039,65279,65440,65520,65521,65522,65523,65524,65525,65526,65527,65528,65532,78844,119155,119156,119157,119158,119159,119160,119161,119162,917504,917505,917506,917507,917508,917509,917510,917511,917512,917513,917514,917515,917516,917517,917518,917519,917520,917521,917522,917523,917524,917525,917526,917527,917528,917529,917530,917531,917532,917533,917534,917535,917536,917537,917538,917539,917540,917541,917542,917543,917544,917545,917546,917547,917548,917549,917550,917551,917552,917553,917554,917555,917556,917557,917558,917559,917560,917561,917562,917563,917564,917565,917566,917567,917568,917569,917570,917571,917572,917573,917574,917575,917576,917577,917578,917579,917580,917581,917582,917583,917584,917585,917586,917587,917588,917589,917590,917591,917592,917593,917594,917595,917596,917597,917598,917599,917600,917601,917602,917603,917604,917605,917606,917607,917608,917609,917610,917611,917612,917613,917614,917615,917616,917617,917618,917619,917620,917621,917622,917623,917624,917625,917626,917627,917628,917629,917630,917631,917760,917761,917762,917763,917764,917765,917766,917767,917768,917769,917770,917771,917772,917773,917774,917775,917776,917777,917778,917779,917780,917781,917782,917783,917784,917785,917786,917787,917788,917789,917790,917791,917792,917793,917794,917795,917796,917797,917798,917799,917800,917801,917802,917803,917804,917805,917806,917807,917808,917809,917810,917811,917812,917813,917814,917815,917816,917817,917818,917819,917820,917821,917822,917823,917824,917825,917826,917827,917828,917829,917830,917831,917832,917833,917834,917835,917836,917837,917838,917839,917840,917841,917842,917843,917844,917845,917846,917847,917848,917849,917850,917851,917852,917853,917854,917855,917856,917857,917858,917859,917860,917861,917862,917863,917864,917865,917866,917867,917868,917869,917870,917871,917872,917873,917874,917875,917876,917877,917878,917879,917880,917881,917882,917883,917884,917885,917886,917887,917888,917889,917890,917891,917892,917893,917894,917895,917896,917897,917898,917899,917900,917901,917902,917903,917904,917905,917906,917907,917908,917909,917910,917911,917912,917913,917914,917915,917916,917917,917918,917919,917920,917921,917922,917923,917924,917925,917926,917927,917928,917929,917930,917931,917932,917933,917934,917935,917936,917937,917938,917939,917940,917941,917942,917943,917944,917945,917946,917947,917948,917949,917950,917951,917952,917953,917954,917955,917956,917957,917958,917959,917960,917961,917962,917963,917964,917965,917966,917967,917968,917969,917970,917971,917972,917973,917974,917975,917976,917977,917978,917979,917980,917981,917982,917983,917984,917985,917986,917987,917988,917989,917990,917991,917992,917993,917994,917995,917996,917997,917998,917999],\"cs\":[173,8203,12288],\"de\":[173,8203,12288],\"es\":[8203,12288],\"fr\":[173,8203,12288],\"it\":[160,173,12288],\"ja\":[173],\"ko\":[173,12288],\"pl\":[173,8203,12288],\"pt-BR\":[173,8203,12288],\"qps-ploc\":[160,173,8203,12288],\"ru\":[173,12288],\"tr\":[160,173,8203,12288],\"zh-hans\":[160,173,8203,12288],\"zh-hant\":[173,12288]}');
1356 }
1357 > strings.ts
1358 > private static _data: Set<number> | undefined = undefined;
1359 >
1360 > private static getData() {
1361 if (!this._data) {
1362 this._data = new Set([...Object.values(InvisibleCharacters.getRawData())].flat());
1364 return this._data;
1365 }
1366 > strings.ts
1367 > public static isInvisibleCharacter(codePoint: number): boolean {
1368 return InvisibleCharacters.getData().has(codePoint);
1369 }
1370 > strings.ts
1371 > public static containsInvisibleCharacter(str: string): boolean {
1372 for (let i = 0; i < str.length; i++) {
1373 const codePoint = str.codePointAt(i);
1378 return false;
1379 }
1380 > strings.ts
1381 > public static get codePoints(): ReadonlySet<number> {
1382 return InvisibleCharacters.getData();
1383 }
1384 > } strings.ts
1385 >
1386 > export const Ellipsis = '\u2026';
1387 >
1388 > /**
1389 > * Convert a Unicode string to a string in which each 16-bit unit occupies only one byte
1390 > *
1391 > * From https://developer.mozilla.org/en-US/docs/Web/API/WindowOrWorkerGlobalScope/btoa
1392 > */
1393 function toBinary(str: string): string {
1394 const codeUnits = new Uint16Array(str.length);
1403 return binary;
1404 }
1405 > strings.ts
1406 > /**
1407 > * Version of the global `btoa` function that handles multi-byte characters instead
1408 > * of throwing an exception.
1409 > */
1410 >
1411 > export function multibyteAwareBtoa(str: string): string {
1412 return btoa(toBinary(str));
1413 }
src/vs/platform/extensions/common/extensions.ts 486 covered LOC · 25 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensions.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import Severity from '../../../base/common/severity.js';
7 > import * as strings from '../../../base/common/strings.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { ILocalizedString } from '../../action/common/action.js';
10 > import { ExtensionKind } from '../../environment/common/environment.js';
11 > import { createDecorator } from '../../instantiation/common/instantiation.js';
12 > import { getRemoteName } from '../../remote/common/remoteHosts.js';
13 >
14 > export const USER_MANIFEST_CACHE_FILE = 'extensions.user.cache';
15 > export const BUILTIN_MANIFEST_CACHE_FILE = 'extensions.builtin.cache';
16 > export const UNDEFINED_PUBLISHER = 'undefined_publisher';
17 >
18 > export interface ICommand {
19 > command: string;
20 > title: string | ILocalizedString;
21 > category?: string | ILocalizedString;
22 > }
23 >
24 > export interface IDebugger {
25 > label?: string;
26 > type: string;
27 > runtime?: string;
28 > }
29 >
30 > export interface IGrammar {
31 > language?: string;
32 > }
33 >
34 > export interface IJSONValidation {
35 > fileMatch: string | string[];
36 > url: string;
37 > }
38 >
39 > export interface IJSONValidationRegistry {
40 > url: string;
41 > }
42 >
43 > export interface IKeyBinding {
44 > command: string;
45 > key: string;
46 > when?: string;
47 > mac?: string;
48 > linux?: string;
49 > win?: string;
50 > }
51 >
52 > export interface ILanguage {
53 > id: string;
54 > extensions: string[];
55 > aliases: string[];
56 > }
57 >
58 > export interface IMenu {
59 > command: string;
60 > alt?: string;
61 > when?: string;
62 > group?: string;
63 > }
64 >
65 > export interface ISnippet {
66 > language: string;
67 > }
68 >
69 > export interface ITheme {
70 > label: string;
71 > }
72 >
73 > export interface IViewContainer {
74 > id: string;
75 > title: string;
76 > }
77 >
78 > export interface IView {
79 > id: string;
80 > name: string;
81 > }
82 >
83 > export interface IColor {
84 > id: string;
85 > description: string;
86 > defaults: { light: string; dark: string; highContrast: string };
87 > }
88 >
89 > interface IWebviewEditor {
90 > readonly viewType: string;
91 > readonly priority: string;
92 > readonly selector: readonly {
93 > readonly filenamePattern?: string;
94 > }[];
95 > }
96 >
97 > export interface ICodeActionContributionAction {
98 > readonly kind: string;
99 > readonly title: string;
100 > readonly description?: string;
101 > }
102 >
103 > export interface ICodeActionContribution {
104 > readonly languages: readonly string[];
105 > readonly actions: readonly ICodeActionContributionAction[];
106 > }
107 >
108 > export interface IAuthenticationContribution {
109 > readonly id: string;
110 > readonly label: string;
111 > readonly authorizationServerGlobs?: string[];
112 > }
113 >
114 > export interface IWalkthroughStep {
115 > readonly id: string;
116 > readonly title: string;
117 > readonly description: string | undefined;
118 > readonly media:
119 > | { image: string | { dark: string; light: string; hc: string }; altText: string; markdown?: never; svg?: never; video?: never }
120 > | { markdown: string; image?: never; svg?: never; video?: never }
121 > | { svg: string; altText: string; markdown?: never; image?: never; video?: never }
122 > | { video: string | { dark: string; light: string; hc: string }; poster: string | { dark: string; light: string; hc: string }; altText: string; markdown?: never; image?: never; svg?: never };
123 > readonly completionEvents?: string[];
124 > /** @deprecated use `completionEvents: 'onCommand:...'` */
125 > readonly doneOn?: { command: string };
126 > readonly when?: string;
127 > }
128 >
129 > export interface IWalkthrough {
130 > readonly id: string;
131 > readonly title: string;
132 > readonly icon?: string;
133 > readonly description: string;
134 > readonly steps: IWalkthroughStep[];
135 > readonly featuredFor: string[] | undefined;
136 > readonly when?: string;
137 > }
138 >
139 > export interface IStartEntry {
140 > readonly title: string;
141 > readonly description: string;
142 > readonly command: string;
143 > readonly when?: string;
144 > readonly category: 'file' | 'folder' | 'notebook';
145 > }
146 >
147 > export interface INotebookEntry {
148 > readonly type: string;
149 > readonly displayName: string;
150 > }
151 >
152 > export interface INotebookRendererContribution {
153 > readonly id: string;
154 > readonly displayName: string;
155 > readonly mimeTypes: string[];
156 > }
157 >
158 > export interface IDebugVisualizationContribution {
159 > readonly id: string;
160 > readonly when: string;
161 > }
162 >
163 > export interface ITranslation {
164 > id: string;
165 > path: string;
166 > }
167 >
168 > export interface ILocalizationContribution {
169 > languageId: string;
170 > languageName?: string;
171 > localizedLanguageName?: string;
172 > translations: ITranslation[];
173 > minimalTranslations?: { [key: string]: string };
174 > }
175 >
176 > export interface IChatParticipantContribution {
177 > id: string;
178 > name: string;
179 > fullName: string;
180 > description?: string;
181 > isDefault?: boolean;
182 > commands?: { name: string }[];
183 > }
184 >
185 > export interface IToolContribution {
186 > name: string;
187 > displayName: string;
188 > modelDescription: string;
189 > userDescription?: string;
190 > }
191 >
192 > export interface IToolSetContribution {
193 > name: string;
194 > referenceName: string;
195 > description: string;
196 > icon?: string;
197 > tools: string[];
198 > }
199 >
200 > export interface IMcpCollectionContribution {
201 > readonly id: string;
202 > readonly label: string;
203 > readonly when?: string;
204 > }
205 >
206 > export interface IChatFileContribution {
207 > readonly path: string;
208 > readonly name?: string;
209 > readonly description?: string;
210 > readonly when?: string;
211 > readonly sessionTypes?: readonly string[];
212 > }
213 >
214 > export interface IExtensionContributions {
215 > commands?: ICommand[];
216 > configuration?: any;
217 > configurationDefaults?: any;
218 > debuggers?: IDebugger[];
219 > grammars?: IGrammar[];
220 > jsonValidation?: IJSONValidation[];
221 > jsonValidationRegistry?: IJSONValidationRegistry[];
222 > keybindings?: IKeyBinding[];
223 > languages?: ILanguage[];
224 > menus?: { [context: string]: IMenu[] };
225 > snippets?: ISnippet[];
226 > themes?: ITheme[];
227 > iconThemes?: ITheme[];
228 > productIconThemes?: ITheme[];
229 > viewsContainers?: { [location: string]: IViewContainer[] };
230 > views?: { [location: string]: IView[] };
231 > colors?: IColor[];
232 > localizations?: ILocalizationContribution[];
233 > readonly customEditors?: readonly IWebviewEditor[];
234 > readonly codeActions?: readonly ICodeActionContribution[];
235 > authentication?: IAuthenticationContribution[];
236 > walkthroughs?: IWalkthrough[];
237 > startEntries?: IStartEntry[];
238 > readonly notebooks?: INotebookEntry[];
239 > readonly notebookRenderer?: INotebookRendererContribution[];
240 > readonly debugVisualizers?: IDebugVisualizationContribution[];
241 > readonly chatParticipants?: ReadonlyArray<IChatParticipantContribution>;
242 > readonly chatPromptFiles?: ReadonlyArray<IChatFileContribution>;
243 > readonly chatInstructions?: ReadonlyArray<IChatFileContribution>;
244 > readonly chatAgents?: ReadonlyArray<IChatFileContribution>;
245 > readonly chatSkills?: ReadonlyArray<IChatFileContribution>;
246 > readonly chatPlugins?: ReadonlyArray<IChatFileContribution>;
247 > readonly languageModelTools?: ReadonlyArray<IToolContribution>;
248 > readonly languageModelToolSets?: ReadonlyArray<IToolSetContribution>;
249 > readonly mcpServerDefinitionProviders?: ReadonlyArray<IMcpCollectionContribution>;
250 > }
251 >
252 > export interface IExtensionCapabilities {
253 > readonly virtualWorkspaces?: ExtensionVirtualWorkspaceSupport;
254 > readonly untrustedWorkspaces?: ExtensionUntrustedWorkspaceSupport;
255 > }
256 >
257 >
258 > export const ALL_EXTENSION_KINDS: readonly ExtensionKind[] = ['ui', 'workspace', 'web'];
259 >
260 > export type LimitedWorkspaceSupportType = 'limited';
261 > export type ExtensionUntrustedWorkspaceSupportType = boolean | LimitedWorkspaceSupportType;
262 > export type ExtensionUntrustedWorkspaceSupport = { supported: true } | { supported: false; description: string } | { supported: LimitedWorkspaceSupportType; description: string; restrictedConfigurations?: string[] };
263 >
264 > export type ExtensionVirtualWorkspaceSupportType = boolean | LimitedWorkspaceSupportType;
265 > export type ExtensionVirtualWorkspaceSupport = boolean | { supported: true } | { supported: false | LimitedWorkspaceSupportType; description: string };
266 >
267 > export function getWorkspaceSupportTypeMessage(supportType: ExtensionUntrustedWorkspaceSupport | ExtensionVirtualWorkspaceSupport | undefined): string | undefined {
268 if (typeof supportType === 'object' && supportType !== null) {
269 if (supportType.supported !== true) {
273 return undefined;
274 }
276 >
277 > export interface IExtensionIdentifier {
278 > id: string;
279 > uuid?: string;
280 > }
281 >
282 > export const EXTENSION_CATEGORIES = [
283 > 'AI',
284 > 'Azure',
285 > 'Chat',
286 > 'Data Science',
287 > 'Debuggers',
288 > 'Extension Packs',
289 > 'Education',
290 > 'Formatters',
291 > 'Keymaps',
292 > 'Language Packs',
293 > 'Linters',
294 > 'Machine Learning',
295 > 'Notebooks',
296 > 'Programming Languages',
297 > 'SCM Providers',
298 > 'Snippets',
299 > 'Testing',
300 > 'Themes',
301 > 'Visualization',
302 > 'Other',
303 > ];
304 >
305 > export interface IRelaxedExtensionManifest {
306 > name: string;
307 > displayName?: string;
308 > publisher: string;
309 > version: string;
310 > engines: { readonly vscode: string };
311 > description?: string;
312 > main?: string;
313 > type?: string;
314 > browser?: string;
315 > preview?: boolean;
316 > // For now this only supports pointing to l10n bundle files
317 > // but it will be used for package.l10n.json files in the future
318 > l10n?: string;
319 > icon?: string;
320 > categories?: string[];
321 > keywords?: string[];
322 > activationEvents?: readonly string[];
323 > extensionDependencies?: string[];
324 > extensionAffinity?: string[];
325 > extensionPack?: string[];
326 > extensionKind?: ExtensionKind | ExtensionKind[];
327 > contributes?: IExtensionContributions;
328 > repository?: { url: string };
329 > bugs?: { url: string };
330 > originalEnabledApiProposals?: readonly string[];
331 > enabledApiProposals?: readonly string[];
332 > api?: string;
333 > scripts?: { [key: string]: string };
334 > capabilities?: IExtensionCapabilities;
335 > }
336 >
337 > export type IExtensionManifest = Readonly<IRelaxedExtensionManifest>;
338 >
339 > export const enum ExtensionType {
340 > System,
341 > User
342 > }
343 >
344 > export const enum TargetPlatform {
345 > WIN32_X64 = 'win32-x64',
346 > WIN32_ARM64 = 'win32-arm64',
347 >
348 > LINUX_X64 = 'linux-x64',
349 > LINUX_ARM64 = 'linux-arm64',
350 > LINUX_ARMHF = 'linux-armhf',
351 >
352 > ALPINE_X64 = 'alpine-x64',
353 > ALPINE_ARM64 = 'alpine-arm64',
354 >
355 > DARWIN_X64 = 'darwin-x64',
356 > DARWIN_ARM64 = 'darwin-arm64',
357 >
358 > WEB = 'web',
359 >
360 > UNIVERSAL = 'universal',
361 > UNKNOWN = 'unknown',
362 > UNDEFINED = 'undefined',
363 > }
364 >
365 > export interface IExtension {
366 > readonly type: ExtensionType;
367 > readonly isBuiltin: boolean;
368 > readonly identifier: IExtensionIdentifier;
369 > readonly manifest: IExtensionManifest;
370 > readonly location: URI;
371 > readonly targetPlatform: TargetPlatform;
372 > readonly publisherDisplayName?: string;
373 > readonly readmeUrl?: URI;
374 > readonly changelogUrl?: URI;
375 > readonly isValid: boolean;
376 > readonly validations: readonly [Severity, string][];
377 > readonly preRelease: boolean;
378 > }
379 >
380 > /**
381 > * **!Do not construct directly!**
382 > *
383 > * **!Only static methods because it gets serialized!**
384 > *
385 > * This represents the "canonical" version for an extension identifier. Extension ids
386 > * have to be case-insensitive (due to the marketplace), but we must ensure case
387 > * preservation because the extension API is already public at this time.
388 > *
389 > * For example, given an extension with the publisher `"Hello"` and the name `"World"`,
390 > * its canonical extension identifier is `"Hello.World"`. This extension could be
391 > * referenced in some other extension's dependencies using the string `"hello.world"`.
392 > *
393 > * To make matters more complicated, an extension can optionally have an UUID. When two
394 > * extensions have the same UUID, they are considered equal even if their identifier is different.
395 > */
396 > export class ExtensionIdentifier {
397 > public readonly value: string;
398 >
399 > /**
400 > * Do not use directly. This is public to avoid mangling and thus
401 > * allow compatibility between running from source and a built version.
402 > */
403 > readonly _lower: string;
404 >
405 > constructor(value: string) {
406 > this.value = value; extensions.ts
407 > this._lower = value.toLowerCase();
408 > }
410 > public static equals(a: ExtensionIdentifier | string | null | undefined, b: ExtensionIdentifier | string | null | undefined) {
411 if (typeof a === 'undefined' || a === null) {
412 return (typeof b === 'undefined' || b === null);
426 return (a._lower === b._lower);
427 }
429 > /**
430 > * Gives the value by which to index (for equality).
431 > */
432 > public static toKey(id: ExtensionIdentifier | string): string {
433 if (typeof id === 'string') {
434 return id.toLowerCase();
436 return id._lower;
437 }
438 > } extensions.ts
439 >
440 > export class ExtensionIdentifierSet {
441 >
442 > private readonly _set = new Set<string>();
443 >
444 > public get size(): number {
445 > return this._set.size;
446 > }
447 >
448 > constructor(iterable?: Iterable<ExtensionIdentifier | string>) {
449 if (iterable) {
450 for (const value of iterable) {
453 }
454 }
456 > public add(id: ExtensionIdentifier | string): void {
457 this._set.add(ExtensionIdentifier.toKey(id));
458 }
460 > public delete(extensionId: ExtensionIdentifier): boolean {
461 return this._set.delete(ExtensionIdentifier.toKey(extensionId));
462 }
464 > public has(id: ExtensionIdentifier | string): boolean {
465 return this._set.has(ExtensionIdentifier.toKey(id));
466 }
467 > } extensions.ts
468 >
469 > export class ExtensionIdentifierMap<T> {
470
471 private readonly _map = new Map<string, T>();
473 > public clear(): void {
474 this._map.clear();
475 }
477 > public delete(id: ExtensionIdentifier | string): void {
478 this._map.delete(ExtensionIdentifier.toKey(id));
479 }
481 > public get(id: ExtensionIdentifier | string): T | undefined {
482 return this._map.get(ExtensionIdentifier.toKey(id));
483 }
485 > public has(id: ExtensionIdentifier | string): boolean {
486 return this._map.has(ExtensionIdentifier.toKey(id));
487 }
489 > public set(id: ExtensionIdentifier | string, value: T): void {
490 this._map.set(ExtensionIdentifier.toKey(id), value);
491 }
493 > public values(): IterableIterator<T> {
494 return this._map.values();
495 }
497 > forEach(callbackfn: (value: T, key: string, map: Map<string, T>) => void): void {
498 this._map.forEach(callbackfn);
499 }
501 > [Symbol.iterator](): IterableIterator<[string, T]> {
502 return this._map[Symbol.iterator]();
503 }
504 > } extensions.ts
505 >
506 > /**
507 > * An error that is clearly from an extension, identified by the `ExtensionIdentifier`
508 > */
509 > export class ExtensionError extends Error {
510 >
511 > readonly extension: ExtensionIdentifier;
512 >
513 > constructor(extensionIdentifier: ExtensionIdentifier, cause: Error, message?: string) {
514 const detail = message && cause?.message ? `${message}: ${cause.message}` : (message ?? cause?.message);
515 super(`Error in extension ${ExtensionIdentifier.toKey(extensionIdentifier)}: ${detail}`, { cause });
523 }
524 }
525 > } extensions.ts
526 >
527 > export interface IRelaxedExtensionDescription extends IRelaxedExtensionManifest {
528 > id?: string;
529 > identifier: ExtensionIdentifier;
530 > uuid?: string;
531 > publisherDisplayName?: string;
532 > targetPlatform: TargetPlatform;
533 > isBuiltin: boolean;
534 > isUserBuiltin: boolean;
535 > isUnderDevelopment: boolean;
536 > extensionLocation: URI;
537 > preRelease: boolean;
538 > }
539 >
540 > export type IExtensionDescription = Readonly<IRelaxedExtensionDescription>;
541 >
542 > export function isApplicationScopedExtension(manifest: IExtensionManifest): boolean {
543 return isLanguagePackExtension(manifest);
544 }
546 > export function isLanguagePackExtension(manifest: IExtensionManifest): boolean {
547 return manifest.contributes && manifest.contributes.localizations ? manifest.contributes.localizations.length > 0 : false;
548 }
550 > export function isAuthenticationProviderExtension(manifest: IExtensionManifest): boolean {
551 return manifest.contributes && manifest.contributes.authentication ? manifest.contributes.authentication.length > 0 : false;
552 }
554 > export function isResolverExtension(manifest: IExtensionManifest, remoteAuthority: string | undefined): boolean {
555 if (remoteAuthority) {
556 const activationEvent = `onResolveRemoteAuthority:${getRemoteName(remoteAuthority)}`;
559 return false;
560 }
562 > export function parseEnabledApiProposalNames(enabledApiProposals: string[]): string[] {
563 return enabledApiProposals.map(proposal => proposal.split('@')[0]);
564 }
566 > export const IBuiltinExtensionsScannerService = createDecorator<IBuiltinExtensionsScannerService>('IBuiltinExtensionsScannerService');
567 > export interface IBuiltinExtensionsScannerService {
568 > readonly _serviceBrand: undefined;
569 > scanBuiltinExtensions(): Promise<IExtension[]>;
570 > }
src/vs/workbench/services/extensions/common/extensions.ts 484 covered LOC · 24 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensions.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { Disposable, IDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
8 > import { raceTimeout } from '../../../../base/common/async.js';
9 > import Severity from '../../../../base/common/severity.js';
10 > import { URI } from '../../../../base/common/uri.js';
11 > import { IMessagePassingProtocol } from '../../../../base/parts/ipc/common/ipc.js';
12 > import { IAssignmentService } from '../../../../platform/assignment/common/assignment.js';
13 > import { getExtensionId, getGalleryExtensionId } from '../../../../platform/extensionManagement/common/extensionManagementUtil.js';
14 > import { ImplicitActivationEvents } from '../../../../platform/extensionManagement/common/implicitActivationEvents.js';
15 > import { ExtensionIdentifier, ExtensionIdentifierMap, ExtensionIdentifierSet, ExtensionType, IExtension, IExtensionContributions, IExtensionDescription, TargetPlatform } from '../../../../platform/extensions/common/extensions.js';
16 > import { ApiProposalName } from '../../../../platform/extensions/common/extensionsApiProposals.js';
17 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
18 > import { IV8Profile } from '../../../../platform/profiling/common/profiling.js';
19 > import { ExtensionHostKind } from './extensionHostKind.js';
20 > import { IExtensionDescriptionDelta, IExtensionDescriptionSnapshot } from './extensionHostProtocol.js';
21 > import { ExtensionRunningLocation } from './extensionRunningLocation.js';
22 > import { IExtensionPoint } from './extensionsRegistry.js';
23 >
24 > export const nullExtensionDescription = Object.freeze<IExtensionDescription>({
25 > identifier: new ExtensionIdentifier('nullExtensionDescription'),
26 > name: 'Null Extension Description',
27 > version: '0.0.0',
28 > publisher: 'vscode',
29 > engines: { vscode: '' },
30 > extensionLocation: URI.parse('void:location'),
31 > isBuiltin: false,
32 > targetPlatform: TargetPlatform.UNDEFINED,
33 > isUserBuiltin: false,
34 > isUnderDevelopment: false,
35 > preRelease: false,
36 > });
37 >
38 > export type WebWorkerExtHostConfigValue = boolean | 'auto';
39 > export const webWorkerExtHostConfig = 'extensions.webWorker';
40 >
41 > export const IExtensionService = createDecorator<IExtensionService>('extensionService');
42 >
43 > export interface IMessage {
44 > type: Severity;
45 > message: string;
46 > extensionId: ExtensionIdentifier;
47 > extensionPointId: string;
48 > }
49 >
50 > export interface IExtensionsStatus {
51 > id: ExtensionIdentifier;
52 > messages: IMessage[];
53 > activationStarted: boolean;
54 > activationTimes: ActivationTimes | undefined;
55 > runtimeErrors: Error[];
56 > runningLocation: ExtensionRunningLocation | null;
57 > }
58 >
59 > export class MissingExtensionDependency {
60 > constructor(readonly dependency: string) { }
61 > }
62 >
63 > /**
64 > * e.g.
65 > * ```
66 > * {
67 > * startTime: 1511954813493000,
68 > * endTime: 1511954835590000,
69 > * deltas: [ 100, 1500, 123456, 1500, 100000 ],
70 > * ids: [ 'idle', 'self', 'extension1', 'self', 'idle' ]
71 > * }
72 > * ```
73 > */
74 > export interface IExtensionHostProfile {
75 > /**
76 > * Profiling start timestamp in microseconds.
77 > */
78 > startTime: number;
79 > /**
80 > * Profiling end timestamp in microseconds.
81 > */
82 > endTime: number;
83 > /**
84 > * Duration of segment in microseconds.
85 > */
86 > deltas: number[];
87 > /**
88 > * Segment identifier: extension id or one of the four known strings.
89 > */
90 > ids: ProfileSegmentId[];
91 >
92 > /**
93 > * Get the information as a .cpuprofile.
94 > */
95 > data: IV8Profile;
96 >
97 > /**
98 > * Get the aggregated time per segmentId
99 > */
100 > getAggregatedTimes(): Map<ProfileSegmentId, number>;
101 > }
102 >
103 > export const enum ExtensionHostStartup {
104 > /**
105 > * The extension host should be launched immediately and doesn't require a `$startExtensionHost` call.
106 > */
107 > EagerAutoStart = 1,
108 > /**
109 > * The extension host should be launched immediately and needs a `$startExtensionHost` call.
110 > */
111 > EagerManualStart = 2,
112 > /**
113 > * The extension host should be launched lazily and only when it has extensions it needs to host. It doesn't require a `$startExtensionHost` call.
114 > */
115 > LazyAutoStart = 3,
116 > }
117 >
118 > export interface IExtensionInspectInfo {
119 > readonly port: number;
120 > readonly host: string;
121 > readonly devtoolsUrl?: string;
122 > readonly devtoolsLabel?: string;
123 > }
124 >
125 > export interface IExtensionHost {
126 > readonly pid: number | null;
127 > readonly runningLocation: ExtensionRunningLocation;
128 > readonly remoteAuthority: string | null;
129 > readonly startup: ExtensionHostStartup;
130 > /**
131 > * A collection of extensions which includes information about which
132 > * extension will execute or is executing on this extension host.
133 > * **NOTE**: this will reflect extensions correctly only after `start()` resolves.
134 > */
135 > readonly extensions: ExtensionHostExtensions | null;
136 > readonly onExit: Event<[number, string | null]>;
137 >
138 > start(): Promise<IMessagePassingProtocol>;
139 > getInspectPort(): IExtensionInspectInfo | undefined;
140 > enableInspectPort(): Promise<boolean>;
141 > disconnect?(): Promise<void>;
142 > dispose(): void;
143 > }
144 >
145 > export class ExtensionHostExtensions {
146 > private _versionId: number;
147 > private _allExtensions: IExtensionDescription[];
148 > private _myExtensions: ExtensionIdentifier[];
149 > private _myActivationEvents: Set<string> | null;
150 >
151 > public get versionId(): number {
152 return this._versionId;
153 }
155 > public get allExtensions(): IExtensionDescription[] {
156 return this._allExtensions;
157 }
159 > public get myExtensions(): ExtensionIdentifier[] {
160 return this._myExtensions;
161 }
163 > constructor(versionId: number, allExtensions: readonly IExtensionDescription[], myExtensions: ExtensionIdentifier[]) {
164 this._versionId = versionId;
165 this._allExtensions = allExtensions.slice(0);
167 this._myActivationEvents = null;
168 }
170 > toSnapshot(): IExtensionDescriptionSnapshot {
171 return {
172 versionId: this._versionId,
176 };
177 }
179 > public set(versionId: number, allExtensions: IExtensionDescription[], myExtensions: ExtensionIdentifier[]): IExtensionDescriptionDelta {
180 if (this._versionId > versionId) {
181 throw new Error(`ExtensionHostExtensions: invalid versionId ${versionId} (current: ${this._versionId})`);
245 return delta;
246 }
248 > public delta(extensionsDelta: IExtensionDescriptionDelta): IExtensionDescriptionDelta | null {
249 if (this._versionId >= extensionsDelta.versionId) {
250 // ignore older deltas
281 return extensionsDelta;
282 }
284 > public containsExtension(extensionId: ExtensionIdentifier): boolean {
285 for (const myExtensionId of this._myExtensions) {
286 if (ExtensionIdentifier.equals(myExtensionId, extensionId)) {
290 return false;
291 }
293 > public containsActivationEvent(activationEvent: string): boolean {
294 if (!this._myActivationEvents) {
295 this._myActivationEvents = this._readMyActivationEvents();
297 return this._myActivationEvents.has(activationEvent);
298 }
300 > private _readMyActivationEvents(): Set<string> {
301 const result = new Set<string>();
302
314 return result;
315 }
316 > } extensions.ts
317 >
318 function extensionDescriptionArrayToMap(extensions: IExtensionDescription[]): ExtensionIdentifierMap<IExtensionDescription> {
319 const result = new ExtensionIdentifierMap<IExtensionDescription>();
323 return result;
324 }
326 > export function isProposedApiEnabled(extension: IExtensionDescription, proposal: ApiProposalName): boolean {
327 if (!extension.enabledApiProposals) {
328 return false;
339 return enabled;
340 }
342 > export interface IProposedApiUsage {
343 > /**
344 > * The identifier of the extension that attempted to use the proposal.
345 > */
346 > readonly extensionId: string;
347 > /**
348 > * The name of the API proposal that the extension is not allowed to use.
349 > */
350 > readonly proposalName: ApiProposalName;
351 > }
352 >
353 > type ProposedApiUsageReporter = (usage: IProposedApiUsage) => void;
354 >
355 > let _proposedApiUsageReporter: ProposedApiUsageReporter | undefined;
356 > const _reportedProposedApiUsages = new Set<string>();
357 >
358 > /**
359 > * Registers a reporter that is invoked whenever an extension attempts to use a proposed API
360 > * that it has not declared via its `enabledApiProposals`-property. This is used to gather
361 > * telemetry about extensions that rely on proposed API they are not entitled to use.
362 > *
363 > * Each unique extension/proposal combination is reported at most once per session in order to
364 > * avoid flooding telemetry from the (potentially hot) call sites of {@link isProposedApiEnabled}.
365 > */
366 > export function setProposedApiUsageReporter(reporter: ProposedApiUsageReporter): IDisposable {
367 _proposedApiUsageReporter = reporter;
368 return toDisposable(() => {
372 });
373 }
375 function reportDisabledProposedApiUsage(extension: IExtensionDescription, proposal: ApiProposalName): void {
376 const reporter = _proposedApiUsageReporter;
385 reporter({ extensionId: extension.identifier.value, proposalName: proposal });
386 }
388 > type ProposedApiEnabledResolver = (extension: IExtensionDescription, proposal: ApiProposalName) => boolean;
389 >
390 > let _proposedApiEnabledResolver: ProposedApiEnabledResolver | undefined;
391 >
392 > /**
393 > * The name of the experiment ("treatment") that can grant proposed API access to
394 > * extension/proposal combinations that have not declared the proposal themselves.
395 > */
396 > export const enabledApiProposalsFallbackExperimentName = 'extensionEnabledApiProposalsFallback';
397 >
398 > /**
399 > * Experiment value that explicitly blocks all proposals reaching the fallback.
400 > */
401 > export const enabledApiProposalsFallbackNone = 'none';
402 >
403 > /**
404 > * Resolves the value of the {@link enabledApiProposalsFallbackExperimentName}-experiment, or
405 > * `undefined` when it does not apply (non-`stable` quality) or cannot be read in time.
406 > */
407 export async function resolveEnabledApiProposalsFallbackExperiment(assignmentService: IAssignmentService, quality: string | undefined): Promise<string | undefined> {
408 if (quality !== 'stable') {
419 }
420 }
422 > /**
423 > * Enables the {@link enabledApiProposalsFallbackExperimentName}-experiment which can grant proposed
424 > * API access to extension/proposal combinations that have not declared the proposal via their
425 > * `enabledApiProposals`-property. It only takes effect on `stable` builds.
426 > *
427 > * Note that the experiment only applies to extensions that already declare at least one proposal:
428 > * an extension with no `enabledApiProposals` at all is never granted access.
429 > *
430 > * @param value A comma-separated list of `publisher.extension:proposalName` entries. Any combination
431 > * that appears here will have {@link isProposedApiEnabled} return `true` even when the extension has
432 > * not declared that particular proposal. When unset, all proposals are allowed;
433 > * {@link enabledApiProposalsFallbackNone} blocks all proposals that reach the fallback.
434 > * @param quality The product quality. The experiment only takes effect when this is `stable`.
435 > */
436 > export function setEnabledApiProposalsFallbackExperiment(value: string | undefined, quality: string | undefined): IDisposable {
437 if (quality !== 'stable') {
438 return Disposable.None;
463 });
464 }
466 > export function checkProposedApiEnabled(extension: IExtensionDescription, proposal: ApiProposalName): void {
467 if (!isProposedApiEnabled(extension, proposal)) {
468 throw new Error(`Extension '${extension.identifier.value}' CANNOT use API proposal: ${proposal}.\nIts package.json#enabledApiProposals-property declares: ${extension.enabledApiProposals?.join(', ') ?? '[]'} but NOT ${proposal}.\n The missing proposal MUST be added and you must start in extension development mode or use the following command line switch: --enable-proposed-api ${extension.identifier.value}`);
469 }
470 }
472 >
473 > /**
474 > * Extension id or one of the four known program states.
475 > */
476 > export type ProfileSegmentId = string | 'idle' | 'program' | 'gc' | 'self';
477 >
478 > export interface ExtensionActivationReason {
479 > readonly startup: boolean;
480 > readonly extensionId: ExtensionIdentifier;
481 > readonly activationEvent: string;
482 > }
483 >
484 > export class ActivationTimes {
485 > constructor(
486 public readonly codeLoadingTime: number,
487 public readonly activateCallTime: number,
490 ) {
491 }
492 > } extensions.ts
493 >
494 > export class ExtensionPointContribution<T> {
495 > readonly description: IExtensionDescription;
496 > readonly value: T;
497 >
498 > constructor(description: IExtensionDescription, value: T) {
499 this.description = description;
500 this.value = value;
501 }
502 > } extensions.ts
503 >
504 > export interface IWillActivateEvent {
505 > readonly event: string;
506 > readonly activation: Promise<void>;
507 > readonly activationKind: ActivationKind;
508 > }
509 >
510 > export interface IResponsiveStateChangeEvent {
511 > extensionHostKind: ExtensionHostKind;
512 > isResponsive: boolean;
513 > /**
514 > * Return the inspect port or `0`. `0` means inspection is not possible.
515 > */
516 > getInspectListener(tryEnableInspector: boolean): Promise<IExtensionInspectInfo | undefined>;
517 > }
518 >
519 > export const enum ActivationKind {
520 > Normal = 0,
521 > Immediate = 1
522 > }
523 >
524 > export interface WillStopExtensionHostsEvent {
525 >
526 > /**
527 > * A human readable reason for stopping the extension hosts
528 > * that e.g. can be shown in a confirmation dialog to the
529 > * user.
530 > */
531 > readonly reason: string;
532 >
533 > /**
534 > * A flag to indicate if the operation was triggered automatically
535 > */
536 > readonly auto: boolean;
537 >
538 > /**
539 > * Allows to veto the stopping of extension hosts. The veto can be a long running
540 > * operation.
541 > *
542 > * @param reason a human readable reason for vetoing the extension host stop in case
543 > * where the resolved `value: true`.
544 > */
545 > veto(value: boolean | Promise<boolean>, reason: string): void;
546 > }
547 >
548 > export interface IExtensionService {
549 > readonly _serviceBrand: undefined;
550 >
551 > /**
552 > * An event emitted when extensions are registered after their extension points got handled.
553 > *
554 > * This event will also fire on startup to signal the installed extensions.
555 > *
556 > * @returns the extensions that got registered
557 > */
558 > readonly onDidRegisterExtensions: Event<void>;
559 >
560 > /**
561 > * @event
562 > * Fired when extensions status changes.
563 > * The event contains the ids of the extensions that have changed.
564 > */
565 > readonly onDidChangeExtensionsStatus: Event<ExtensionIdentifier[]>;
566 >
567 > /**
568 > * Fired when the available extensions change (i.e. when extensions are added or removed).
569 > */
570 > readonly onDidChangeExtensions: Event<{ readonly added: readonly IExtensionDescription[]; readonly removed: readonly IExtensionDescription[] }>;
571 >
572 > /**
573 > * All registered extensions.
574 > * - List will be empty initially during workbench startup and will be filled with extensions as they are registered
575 > * - Listen to `onDidChangeExtensions` event for any changes to the extensions list. It will change as extensions get registered or de-reigstered.
576 > * - Listen to `onDidRegisterExtensions` event or wait for `whenInstalledExtensionsRegistered` promise to get the initial list of registered extensions.
577 > */
578 > readonly extensions: readonly IExtensionDescription[];
579 >
580 > /**
581 > * An event that is fired when activation happens.
582 > */
583 > readonly onWillActivateByEvent: Event<IWillActivateEvent>;
584 >
585 > /**
586 > * An event that is fired when an extension host changes its
587 > * responsive-state.
588 > */
589 > readonly onDidChangeResponsiveChange: Event<IResponsiveStateChangeEvent>;
590 >
591 > /**
592 > * Fired before stop of extension hosts happens. Allows listeners to veto against the
593 > * stop to prevent it from happening.
594 > */
595 > readonly onWillStop: Event<WillStopExtensionHostsEvent>;
596 >
597 > /**
598 > * Send an activation event and activate interested extensions.
599 > *
600 > * This will wait for the normal startup of the extension host(s).
601 > *
602 > * In extraordinary circumstances, if the activation event needs to activate
603 > * one or more extensions before the normal startup is finished, then you can use
604 > * `ActivationKind.Immediate`. Please do not use this flag unless really necessary
605 > * and you understand all consequences.
606 > */
607 > activateByEvent(activationEvent: string, activationKind?: ActivationKind): Promise<void>;
608 >
609 > /**
610 > * Send an activation ID and activate interested extensions.
611 > *
612 > */
613 > activateById(extensionId: ExtensionIdentifier, reason: ExtensionActivationReason): Promise<void>;
614 >
615 > /**
616 > * Determine if `activateByEvent(activationEvent)` has resolved already.
617 > *
618 > * i.e. the activation event is finished and all interested extensions are already active.
619 > */
620 > activationEventIsDone(activationEvent: string): boolean;
621 >
622 > /**
623 > * An promise that resolves when the installed extensions are registered after
624 > * their extension points got handled.
625 > */
626 > whenInstalledExtensionsRegistered(): Promise<boolean>;
627 >
628 > /**
629 > * Return a specific extension
630 > * @param id An extension id
631 > */
632 > getExtension(id: string): Promise<IExtensionDescription | undefined>;
633 >
634 > /**
635 > * Returns `true` if the given extension can be added. Otherwise `false`.
636 > * @param extension An extension
637 > */
638 > canAddExtension(extension: IExtensionDescription): boolean;
639 >
640 > /**
641 > * Returns `true` if the given extension can be removed. Otherwise `false`.
642 > * @param extension An extension
643 > */
644 > canRemoveExtension(extension: IExtensionDescription): boolean;
645 >
646 > /**
647 > * Read all contributions to an extension point.
648 > */
649 > readExtensionPointContributions<T extends IExtensionContributions[keyof IExtensionContributions]>(extPoint: IExtensionPoint<T>): Promise<ExtensionPointContribution<T>[]>;
650 >
651 > /**
652 > * Get information about extensions status.
653 > */
654 > getExtensionsStatus(): { [id: string]: IExtensionsStatus };
655 >
656 > /**
657 > * Return the inspect ports (if inspection is possible) for extension hosts of kind `extensionHostKind`.
658 > */
659 > getInspectPorts(extensionHostKind: ExtensionHostKind, tryEnableInspector: boolean): Promise<IExtensionInspectInfo[]>;
660 >
661 > /**
662 > * Stops the extension hosts.
663 > *
664 > * @param reason a human readable reason for stopping the extension hosts. This maybe
665 > * can be presented to the user when showing dialogs.
666 > *
667 > * @param auto indicates if the operation was triggered by an automatic action
668 > *
669 > * @returns a promise that resolves to `true` if the extension hosts were stopped, `false`
670 > * if the operation was vetoed by listeners of the `onWillStop` event.
671 > */
672 > stopExtensionHosts(reason: string, auto?: boolean): Promise<boolean>;
673 >
674 > /**
675 > * Starts the extension hosts. If updates are provided, the extension hosts are started with the given updates.
676 > */
677 > startExtensionHosts(updates?: { readonly toAdd: readonly IExtension[]; readonly toRemove: readonly string[] }): Promise<void>;
678 >
679 > /**
680 > * Modify the environment of the remote extension host
681 > * @param env New properties for the remote extension host
682 > */
683 > setRemoteEnvironment(env: { [key: string]: string | null }): Promise<void>;
684 > }
685 >
686 > export interface IInternalExtensionService {
687 > _activateById(extensionId: ExtensionIdentifier, reason: ExtensionActivationReason): Promise<void>;
688 > _onWillActivateExtension(extensionId: ExtensionIdentifier): void;
689 > _onDidActivateExtension(extensionId: ExtensionIdentifier, codeLoadingTime: number, activateCallTime: number, activateResolvedTime: number, activationReason: ExtensionActivationReason): void;
690 > _onDidActivateExtensionError(extensionId: ExtensionIdentifier, error: Error): void;
691 > _onExtensionRuntimeError(extensionId: ExtensionIdentifier, err: Error): void;
692 > }
693 >
694 > export interface ProfileSession {
695 > stop(): Promise<IExtensionHostProfile>;
696 > }
697 >
698 > export function toExtension(extensionDescription: IExtensionDescription): IExtension {
699 return {
700 type: extensionDescription.isBuiltin ? ExtensionType.System : ExtensionType.User,
710 };
711 }
713 > export function toExtensionDescription(extension: IExtension, isUnderDevelopment?: boolean): IExtensionDescription {
714 const id = getExtensionId(extension.manifest.publisher, extension.manifest.name);
715 return {
727 };
728 }
730 >
731 > export class NullExtensionService implements IExtensionService {
732 > declare readonly _serviceBrand: undefined; extensions.ts
733 > readonly onDidRegisterExtensions: Event<void> = Event.None;
734 > readonly onDidChangeExtensionsStatus: Event<ExtensionIdentifier[]> = Event.None;
735 > onDidChangeExtensions = Event.None;
736 > readonly onWillActivateByEvent: Event<IWillActivateEvent> = Event.None;
737 > readonly onDidChangeResponsiveChange: Event<IResponsiveStateChangeEvent> = Event.None;
738 > readonly onWillStop: Event<WillStopExtensionHostsEvent> = Event.None;
739 > readonly extensions = [];
740 > activateByEvent(_activationEvent: string): Promise<void> { return Promise.resolve(undefined); } extensions.ts
741 > activateById(extensionId: ExtensionIdentifier, reason: ExtensionActivationReason): Promise<void> { return Promise.resolve(undefined); }
742 > activationEventIsDone(_activationEvent: string): boolean { return false; }
743 > whenInstalledExtensionsRegistered(): Promise<boolean> { return Promise.resolve(true); }
744 > getExtension() { return Promise.resolve(undefined); }
745 > readExtensionPointContributions<T>(_extPoint: IExtensionPoint<T>): Promise<ExtensionPointContribution<T>[]> { return Promise.resolve(Object.create(null)); }
746 > getExtensionsStatus(): { [id: string]: IExtensionsStatus } { return Object.create(null); }
747 > getInspectPorts(_extensionHostKind: ExtensionHostKind, _tryEnableInspector: boolean): Promise<IExtensionInspectInfo[]> { return Promise.resolve([]); }
748 > async stopExtensionHosts(): Promise<boolean> { return true; }
749 > async startExtensionHosts(): Promise<void> { }
750 > async setRemoteEnvironment(_env: { [key: string]: string | null }): Promise<void> { }
751 > canAddExtension(): boolean { return false; }
752 > canRemoveExtension(): boolean { return false; }
753 > }
src/vs/base/common/uri.ts 477 covered LOC · 102 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- uri.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from './charCode.js';
7 > import { MarshalledId } from './marshallingIds.js';
8 > import * as paths from './path.js';
9 > import { isWindows } from './platform.js';
10 >
11 > const _schemePattern = /^\w[\w\d+.-]*$/;
12 > const _singleSlashStart = /^\//;
13 > const _doubleSlashStart = /^\/\//;
14 >
15 > function _validateUri(ret: URI, _strict?: boolean): void { uri.ts
16 >
17 > // scheme, must be set
18 > if (!ret.scheme && _strict) {
19 throw new Error(`[UriError]: Scheme is missing: {scheme: "", authority: "${ret.authority}", path: "${ret.path}", query: "${ret.query}", fragment: "${ret.fragment}"}`);
20 }
21 > uri.ts
22 > // scheme, https://tools.ietf.org/html/rfc3986#section-3.1
23 > // ALPHA *( ALPHA / DIGIT / "+" / "-" / "." )
24 > if (ret.scheme && !_schemePattern.test(ret.scheme)) { uri.ts
25 const matches = [...ret.scheme.matchAll(/[^\w\d+.-]/gu)];
26 const detail = matches.length > 0
29 throw new Error(`[UriError]: Scheme contains illegal characters.${detail} (len:${ret.scheme.length})`);
30 }
31 > uri.ts
32 > // path, http://tools.ietf.org/html/rfc3986#section-3.3
33 > // If a URI contains an authority component, then the path component
34 > // must either be empty or begin with a slash ("/") character. If a URI
35 > // does not contain an authority component, then the path cannot begin
36 > // with two slash characters ("//").
37 > if (ret.path) {
38 > if (ret.authority) { uri.ts
39 if (!_singleSlashStart.test(ret.path)) {
40 throw new Error('[UriError]: If a URI contains an authority component, then the path component must either be empty or begin with a slash ("/") character');
41 }
42 > } else { uri.ts
43 > if (_doubleSlashStart.test(ret.path)) { uri.ts
44 throw new Error('[UriError]: If a URI does not contain an authority component, then the path cannot begin with two slash characters ("//")');
45 }
46 > } uri.ts
47 > } uri.ts
48 > } uri.ts
49 > uri.ts
50 > // for a while we allowed uris *without* schemes and this is the migration
51 > // for them, e.g. an uri without scheme and without strict-mode warns and falls
52 > // back to the file-scheme. that should cause the least carnage and still be a
53 > // clear warning
54 > function _schemeFix(scheme: string, _strict: boolean): string { uri.ts
55 > if (!scheme && !_strict) {
56 return 'file';
57 }
58 > return scheme; uri.ts
59 > }
60 > uri.ts
61 > // implements a bit of https://tools.ietf.org/html/rfc3986#section-5
62 > function _referenceResolution(scheme: string, path: string): string { uri.ts
63 >
64 > // the slash-character is our 'default base' as we don't
65 > // support constructing URIs relative to other URIs. This
66 > // also means that we alter and potentially break paths.
67 > // see https://tools.ietf.org/html/rfc3986#section-5.1.4
68 > switch (scheme) {
69 > case 'https':
70 > case 'http':
71 > case 'file':
72 > if (!path) { uri.ts
73 path = _slash;
74 > } else if (path[0] !== _slash) { uri.ts
75 path = _slash + path;
76 }
77 > break; uri.ts
78 > } uri.ts
79 > return path;
80 > }
81 > uri.ts
82 > const _empty = '';
83 > const _slash = '/';
84 > const _regexp = /^(([^:/?#]+?):)?(\/\/([^/?#]*))?([^?#]*)(\?([^#]*))?(#(.*))?/;
85 >
86 > /**
87 > * Uniform Resource Identifier (URI) http://tools.ietf.org/html/rfc3986.
88 > * This class is a simple parser which creates the basic component parts
89 > * (http://tools.ietf.org/html/rfc3986#section-3) with minimal validation
90 > * and encoding.
91 > *
92 > * ```txt
93 > * foo://example.com:8042/over/there?name=ferret#nose
94 > * \_/ \______________/\_________/ \_________/ \__/
95 > * | | | | |
96 > * scheme authority path query fragment
97 > * | _____________________|__
98 > * / \ / \
99 > * urn:example:animal:ferret:nose
100 > * ```
101 > */
102 > export class URI implements UriComponents {
103 >
104 > static isUri(thing: unknown): thing is URI {
105 if (thing instanceof URI) {
106 return true;
118 && typeof (<URI>thing).toString === 'function';
119 }
120 > uri.ts
121 > /**
122 > * scheme is the 'http' part of 'http://www.example.com/some/path?query#fragment'.
123 > * The part before the first colon.
124 > */
125 > readonly scheme: string;
126 >
127 > /**
128 > * authority is the 'www.example.com' part of 'http://www.example.com/some/path?query#fragment'.
129 > * The part between the first double slashes and the next slash.
130 > */
131 > readonly authority: string;
132 >
133 > /**
134 > * path is the '/some/path' part of 'http://www.example.com/some/path?query#fragment'.
135 > */
136 > readonly path: string;
137 >
138 > /**
139 > * query is the 'query' part of 'http://www.example.com/some/path?query#fragment'.
140 > */
141 > readonly query: string;
142 >
143 > /**
144 > * fragment is the 'fragment' part of 'http://www.example.com/some/path?query#fragment'.
145 > */
146 > readonly fragment: string;
147 >
148 > /**
149 > * @internal
150 > */
151 > protected constructor(scheme: string, authority?: string, path?: string, query?: string, fragment?: string, _strict?: boolean);
152 >
153 > /**
154 > * @internal
155 > */
156 > protected constructor(components: UriComponents);
157 >
158 > /**
159 > * @internal
160 > */
161 > protected constructor(schemeOrData: string | UriComponents, authority?: string, path?: string, query?: string, fragment?: string, _strict: boolean = false) {
162 > uri.ts
163 > if (typeof schemeOrData === 'object') {
164 this.scheme = schemeOrData.scheme || _empty;
165 this.authority = schemeOrData.authority || _empty;
170 // that creates uri components.
171 // _validateUri(this);
172 > } else { uri.ts
173 > this.scheme = _schemeFix(schemeOrData, _strict);
174 > this.authority = authority || _empty;
175 > this.path = _referenceResolution(this.scheme, path || _empty);
176 > this.query = query || _empty;
177 > this.fragment = fragment || _empty;
178 >
179 > _validateUri(this, _strict);
180 > }
181 > }
182 > uri.ts
183 > // ---- filesystem path -----------------------
184 >
185 > /**
186 > * Returns a string representing the corresponding file system path of this URI.
187 > * Will handle UNC paths, normalizes windows drive letters to lower-case, and uses the
188 > * platform specific path separator.
189 > *
190 > * * Will *not* validate the path for invalid characters and semantics.
191 > * * Will *not* look at the scheme of this URI.
192 > * * The result shall *not* be used for display purposes but for accessing a file on disk.
193 > *
194 > *
195 > * The *difference* to `URI#path` is the use of the platform specific separator and the handling
196 > * of UNC paths. See the below sample of a file-uri with an authority (UNC path).
197 > *
198 > * ```ts
199 > const u = URI.parse('file://server/c$/folder/file.txt')
200 > u.authority === 'server'
201 > u.path === '/shares/c$/file.txt'
202 > u.fsPath === '\\server\c$\folder\file.txt'
203 > ```
204 > *
205 > * Using `URI#path` to read a file (using fs-apis) would not be enough because parts of the path,
206 > * namely the server name, would be missing. Therefore `URI#fsPath` exists - it's sugar to ease working
207 > * with URIs that represent files on disk (`file` scheme).
208 > */
209 > get fsPath(): string {
210 // if (this.scheme !== 'file') {
211 // console.warn(`[UriError] calling fsPath with scheme ${this.scheme}`);
213 return uriToFsPath(this, false);
214 }
215 > uri.ts
216 > // ---- modify to new -------------------------
217 >
218 > with(change: { scheme?: string; authority?: string | null; path?: string | null; query?: string | null; fragment?: string | null }): URI {
219 > uri.ts
220 > if (!change) {
221 return this;
222 }
223 > uri.ts
224 > let { scheme, authority, path, query, fragment } = change;
225 > if (scheme === undefined) {
226 > scheme = this.scheme; uri.ts
227 > } else if (scheme === null) { uri.ts
228 scheme = _empty;
229 }
230 > if (authority === undefined) { uri.ts
231 > authority = this.authority; uri.ts
232 > } else if (authority === null) { uri.ts
233 authority = _empty;
234 }
235 > if (path === undefined) { uri.ts
236 path = this.path;
237 > } else if (path === null) { uri.ts
238 path = _empty;
239 }
240 > if (query === undefined) { uri.ts
241 > query = this.query; uri.ts
242 > } else if (query === null) { uri.ts
243 query = _empty;
244 }
245 > if (fragment === undefined) { uri.ts
246 > fragment = this.fragment; uri.ts
247 > } else if (fragment === null) { uri.ts
248 fragment = _empty;
249 }
250 > uri.ts
251 > if (scheme === this.scheme
252 > && authority === this.authority uri.ts
253 > && path === this.path
254 && query === this.query
255 > && fragment === this.fragment) { uri.ts
256
257 return this;
258 }
259 > uri.ts
260 > return new Uri(scheme, authority, path, query, fragment);
261 > } uri.ts
262 > uri.ts
263 > // ---- parse & validate ------------------------
264 >
265 > /**
266 > * Creates a new URI from a string, e.g. `http://www.example.com/some/path`,
267 > * `file:///usr/home`, or `scheme:with/path`.
268 > *
269 > * @param value A string which represents an URI (see `URI#toString`).
270 > */
271 > static parse(value: string, _strict: boolean = false): URI {
272 > const match = _regexp.exec(value); uri.ts
273 > if (!match) {
274 return new Uri(_empty, _empty, _empty, _empty, _empty);
275 }
276 > return new Uri( uri.ts
277 > match[2] || _empty,
278 > percentDecode(match[4] || _empty),
279 > percentDecode(match[5] || _empty),
280 > percentDecode(match[7] || _empty),
281 > percentDecode(match[9] || _empty),
282 > _strict
283 > );
284 > }
285 > uri.ts
286 > /**
287 > * Creates a new URI from a file system path, e.g. `c:\my\files`,
288 > * `/usr/home`, or `\\server\share\some\path`.
289 > *
290 > * The *difference* between `URI#parse` and `URI#file` is that the latter treats the argument
291 > * as path, not as stringified-uri. E.g. `URI.file(path)` is **not the same as**
292 > * `URI.parse('file://' + path)` because the path might contain characters that are
293 > * interpreted (# and ?). See the following sample:
294 > * ```ts
295 > const good = URI.file('/coding/c#/project1');
296 > good.scheme === 'file';
297 > good.path === '/coding/c#/project1';
298 > good.fragment === '';
299 > const bad = URI.parse('file://' + '/coding/c#/project1');
300 > bad.scheme === 'file';
301 > bad.path === '/coding/c'; // path is now broken
302 > bad.fragment === '/project1';
303 > ```
304 > *
305 > * @param path A file system path (see `URI#fsPath`)
306 > */
307 > static file(path: string): URI {
308 > uri.ts
309 > let authority = _empty;
310 >
311 > // normalize to fwd-slashes on windows,
312 > // on other systems bwd-slashes are valid
313 > // filename character, eg /f\oo/ba\r.txt
314 > if (isWindows) {
315 path = path.replace(/\\/g, _slash);
316 }
317 > uri.ts
318 > // check for authority as used in UNC shares
319 > // or use the path as given
320 > if (path[0] === _slash && path[1] === _slash) {
321 const idx = path.indexOf(_slash, 2);
322 if (idx === -1) {
328 }
329 }
330 > uri.ts
331 > return new Uri('file', authority, path, _empty, _empty);
332 > }
333 > uri.ts
334 > /**
335 > * Creates new URI from uri components.
336 > *
337 > * Unless `strict` is `true` the scheme is defaults to be `file`. This function performs
338 > * validation and should be used for untrusted uri components retrieved from storage,
339 > * user input, command arguments etc
340 > */
341 > static from(components: UriComponents, strict?: boolean): URI {
342 > const result = new Uri( uri.ts
343 > components.scheme,
344 > components.authority,
345 > components.path,
346 > components.query,
347 > components.fragment,
348 > strict
349 > );
350 > return result;
351 > }
352 > uri.ts
353 > /**
354 > * Join a URI path with path fragments and normalizes the resulting path.
355 > *
356 > * @param uri The input URI.
357 > * @param pathFragment The path fragment to add to the URI path.
358 > * @returns The resulting URI.
359 > */
360 > static joinPath(uri: URI, ...pathFragment: string[]): URI {
361 > if (!uri.path) { uri.ts
362 throw new Error(`[UriError]: cannot call joinPath on URI without path: ${uri.toString()}`);
363 }
364 > let newPath: string; uri.ts
365 > if (isWindows && uri.scheme === 'file') {
366 newPath = URI.file(paths.win32.join(uriToFsPath(uri, true), ...pathFragment)).path;
367 > } else { uri.ts
368 > newPath = paths.posix.join(uri.path, ...pathFragment);
369 > }
370 > return uri.with({ path: newPath });
371 > }
372 > uri.ts
373 > // ---- printing/externalize ---------------------------
374 >
375 > /**
376 > * Creates a string representation for this URI. It's guaranteed that calling
377 > * `URI.parse` with the result of this function creates an URI which is equal
378 > * to this URI.
379 > *
380 > * * The result shall *not* be used for display purposes but for externalization or transport.
381 > * * The result will be encoded using the percentage encoding and encoding happens mostly
382 > * ignore the scheme-specific encoding rules.
383 > *
384 > * @param skipEncoding Do not encode the result, default is `false`
385 > */
386 > toString(skipEncoding: boolean = false): string {
387 return _asFormatted(this, skipEncoding);
388 }
389 > uri.ts
390 > toJSON(): UriComponents {
391 return this;
392 }
393 > uri.ts
394 > /**
395 > * A helper function to revive URIs.
396 > *
397 > * **Note** that this function should only be used when receiving URI#toJSON generated data
398 > * and that it doesn't do any validation. Use {@link URI.from} when received "untrusted"
399 > * uri components such as command arguments or data from storage.
400 > *
401 > * @param data The URI components or URI to revive.
402 > * @returns The revived URI or undefined or null.
403 > */
404 > static revive(data: UriComponents | URI): URI;
405 > static revive(data: UriComponents | URI | undefined): URI | undefined;
406 > static revive(data: UriComponents | URI | null): URI | null;
407 > static revive(data: UriComponents | URI | undefined | null): URI | undefined | null;
408 > static revive(data: UriComponents | URI | undefined | null): URI | undefined | null {
409 if (!data) {
410 return data;
418 }
419 }
420 > uri.ts
421 > [Symbol.for('debug.description')]() {
422 return `URI(${this.toString()})`;
423 }
424 > } uri.ts
425 >
426 > export interface UriComponents {
427 > scheme: string;
428 > authority?: string;
429 > path?: string;
430 > query?: string;
431 > fragment?: string;
432 > }
433 >
434 > export function isUriComponents(thing: unknown): thing is UriComponents {
435 if (!thing || typeof thing !== 'object') {
436 return false;
442 && (typeof (<UriComponents>thing).fragment === 'string' || typeof (<UriComponents>thing).fragment === 'undefined');
443 }
444 > uri.ts
445 > interface UriState extends UriComponents {
446 > $mid: MarshalledId.Uri;
447 > external?: string;
448 > fsPath?: string;
449 > _sep?: 1;
450 > }
451 >
452 > const _pathSepMarker = isWindows ? 1 : undefined;
453 >
454 > // This class exists so that URI is compatible with vscode.Uri (API).
455 > class Uri extends URI { uri.ts
456 >
457 > _formatted: string | null = null;
458 > _fsPath: string | null = null;
459 > uri.ts
460 > override get fsPath(): string {
461 if (!this._fsPath) {
462 this._fsPath = uriToFsPath(this, false);
464 return this._fsPath;
465 }
466 > uri.ts
467 > override toString(skipEncoding: boolean = false): string {
468 > if (!skipEncoding) { uri.ts
469 > if (!this._formatted) { uri.ts
470 > this._formatted = _asFormatted(this, false);
471 > }
472 > return this._formatted;
473 > } else { uri.ts
474 // we don't cache that
475 return _asFormatted(this, true);
476 }
477 > } uri.ts
478 > uri.ts
479 > override toJSON(): UriComponents {
480 // eslint-disable-next-line local/code-no-dangerous-type-assertions
481 const res = <UriState>{
512 return res;
513 }
514 > } uri.ts
515 >
516 > // reserved characters: https://tools.ietf.org/html/rfc3986#section-2.2
517 > const encodeTable: { [ch: number]: string } = {
518 > [CharCode.Colon]: '%3A', // gen-delims
519 > [CharCode.Slash]: '%2F',
520 > [CharCode.QuestionMark]: '%3F',
521 > [CharCode.Hash]: '%23',
522 > [CharCode.OpenSquareBracket]: '%5B',
523 > [CharCode.CloseSquareBracket]: '%5D',
524 > [CharCode.AtSign]: '%40',
525 >
526 > [CharCode.ExclamationMark]: '%21', // sub-delims
527 > [CharCode.DollarSign]: '%24',
528 > [CharCode.Ampersand]: '%26',
529 > [CharCode.SingleQuote]: '%27',
530 > [CharCode.OpenParen]: '%28',
531 > [CharCode.CloseParen]: '%29',
532 > [CharCode.Asterisk]: '%2A',
533 > [CharCode.Plus]: '%2B',
534 > [CharCode.Comma]: '%2C',
535 > [CharCode.Semicolon]: '%3B',
536 > [CharCode.Equals]: '%3D',
537 >
538 > [CharCode.Space]: '%20',
539 > };
540 >
541 > function encodeURIComponentFast(uriComponent: string, isPath: boolean, isAuthority: boolean): string { uri.ts
542 > let res: string | undefined = undefined;
543 > let nativeEncodePos = -1;
544 >
545 > for (let pos = 0; pos < uriComponent.length; pos++) {
546 > const code = uriComponent.charCodeAt(pos);
547 >
548 > // unreserved characters: https://tools.ietf.org/html/rfc3986#section-2.3
549 > if (
550 > (code >= CharCode.a && code <= CharCode.z)
551 > || (code >= CharCode.A && code <= CharCode.Z) uri.ts
552 > || (code >= CharCode.Digit0 && code <= CharCode.Digit9)
553 > || code === CharCode.Dash uri.ts
554 > || code === CharCode.Period uri.ts
555 > || code === CharCode.Underline
556 > || code === CharCode.Tilde
557 > || (isPath && code === CharCode.Slash)
558 || (isAuthority && code === CharCode.OpenSquareBracket)
559 || (isAuthority && code === CharCode.CloseSquareBracket)
560 || (isAuthority && code === CharCode.Colon)
561 > ) { uri.ts
562 > // check if we are delaying native encode
563 > if (nativeEncodePos !== -1) {
564 res += encodeURIComponent(uriComponent.substring(nativeEncodePos, pos));
565 nativeEncodePos = -1;
566 }
567 > // check if we write into a new string (by default we try to return the param) uri.ts
568 > if (res !== undefined) {
569 res += uriComponent.charAt(pos);
570 }
571 > uri.ts
572 > } else {
573 // encoding needed, we need to allocate a new string
574 if (res === undefined) {
594 }
595 }
596 > } uri.ts
597 >
598 > if (nativeEncodePos !== -1) {
599 res += encodeURIComponent(uriComponent.substring(nativeEncodePos));
600 }
601 > uri.ts
602 > return res !== undefined ? res : uriComponent;
603 > }
604 > uri.ts
605 function encodeURIComponentMinimal(path: string): string {
606 let res: string | undefined = undefined;
620 return res !== undefined ? res : path;
621 }
622 > uri.ts
623 > /**
624 > * Compute `fsPath` for the given uri
625 > */
626 > export function uriToFsPath(uri: URI, keepDriveLetterCasing: boolean): string {
627
628 let value: string;
650 return value;
651 }
652 > uri.ts
653 > /**
654 > * Create the external version of a uri
655 > */
656 > function _asFormatted(uri: URI, skipEncoding: boolean): string { uri.ts
657 >
658 > const encoder = !skipEncoding
659 > ? encodeURIComponentFast uri.ts
660 : encodeURIComponentMinimal;
661 > uri.ts
662 > let res = '';
663 > let { scheme, authority, path, query, fragment } = uri;
664 > if (scheme) {
665 > res += scheme;
666 > res += ':';
667 > }
668 > if (authority || scheme === 'file') {
669 > res += _slash; uri.ts
670 > res += _slash;
671 > }
672 > if (authority) { uri.ts
673 let idx = authority.indexOf('@');
674 if (idx !== -1) {
697 }
698 }
699 > if (path) { uri.ts
700 > // lower-case windows drive letters in /C:/fff or C:/fff uri.ts
701 > if (path.length >= 3 && path.charCodeAt(0) === CharCode.Slash && path.charCodeAt(2) === CharCode.Colon) {
702 const code = path.charCodeAt(1);
703 if (code >= CharCode.A && code <= CharCode.Z) {
704 path = `/${String.fromCharCode(code + 32)}:${path.substr(3)}`; // "/c:".length === 3
705 }
706 > } else if (path.length >= 2 && path.charCodeAt(1) === CharCode.Colon) { uri.ts
707 const code = path.charCodeAt(0);
708 if (code >= CharCode.A && code <= CharCode.Z) {
710 }
711 }
712 > // encode the rest of the path uri.ts
713 > res += encoder(path, true, false);
714 > }
715 > if (query) { uri.ts
716 res += '?';
717 res += encoder(query, false, false);
718 }
719 > if (fragment) { uri.ts
720 res += '#';
721 res += !skipEncoding ? encodeURIComponentFast(fragment, false, false) : fragment;
722 }
723 > return res; uri.ts
724 > }
725 > uri.ts
726 > // --- decode
727 >
728 function decodeURIComponentGraceful(str: string): string {
729 try {
737 }
738 }
739 > uri.ts
740 > const _rEncodedAsHex = /(%[0-9A-Za-z][0-9A-Za-z])+/g;
741 >
742 > function percentDecode(str: string): string { uri.ts
743 > if (!str.match(_rEncodedAsHex)) {
744 > return str;
745 > }
746 return str.replace(_rEncodedAsHex, (match) => decodeURIComponentGraceful(match));
747 }
748 > uri.ts
749 > /**
750 > * Mapped-type that replaces all occurrences of URI with UriComponents
751 > */
752 > export type UriDto<T> = { [K in keyof T]: T[K] extends URI
753 > ? UriComponents
754 > : UriDto<T[K]> };
src/vs/platform/storage/common/storage.ts 475 covered LOC · 52 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- storage.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Promises, RunOnceScheduler, runWhenGlobalIdle } from '../../../base/common/async.js';
7 > import { Emitter, Event, PauseableEmitter } from '../../../base/common/event.js';
8 > import { Disposable, DisposableStore, dispose, MutableDisposable } from '../../../base/common/lifecycle.js';
9 > import { mark } from '../../../base/common/performance.js';
10 > import { isUndefinedOrNull } from '../../../base/common/types.js';
11 > import { InMemoryStorageDatabase, IStorage, IStorageChangeEvent, Storage, StorageHint, StorageValue } from '../../../base/parts/storage/common/storage.js';
12 > import { createDecorator } from '../../instantiation/common/instantiation.js';
13 > import { isUserDataProfile, IUserDataProfile } from '../../userDataProfile/common/userDataProfile.js';
14 > import { IAnyWorkspaceIdentifier } from '../../workspace/common/workspace.js';
15 >
16 > export const IS_NEW_KEY = '__$__isNewStorageMarker';
17 > export const TARGET_KEY = '__$__targetStorageMarker';
18 >
19 > export const IStorageService = createDecorator<IStorageService>('storageService');
20 >
21 > export enum WillSaveStateReason {
22 >
23 > /**
24 > * No specific reason to save state.
25 > */
26 > NONE,
27 >
28 > /**
29 > * A hint that the workbench is about to shutdown.
30 > */
31 > SHUTDOWN
32 > }
33 >
34 > export interface IWillSaveStateEvent {
35 > readonly reason: WillSaveStateReason;
36 > }
37 >
38 > export interface IStorageEntry {
39 > readonly key: string;
40 > readonly value: StorageValue;
41 > readonly scope: StorageScope;
42 > readonly target: StorageTarget;
43 > }
44 >
45 > export interface IWorkspaceStorageValueChangeEvent extends IStorageValueChangeEvent {
46 > readonly scope: StorageScope.WORKSPACE;
47 > }
48 >
49 > export interface IProfileStorageValueChangeEvent extends IStorageValueChangeEvent {
50 > readonly scope: StorageScope.PROFILE;
51 > }
52 >
53 > export interface IApplicationStorageValueChangeEvent extends IStorageValueChangeEvent {
54 > readonly scope: StorageScope.APPLICATION;
55 > }
56 >
57 > export interface IApplicationSharedStorageValueChangeEvent extends IStorageValueChangeEvent {
58 > readonly scope: StorageScope.APPLICATION_SHARED;
59 > }
60 >
61 > export interface IStorageService {
62 >
63 > readonly _serviceBrand: undefined;
64 >
65 > /**
66 > * Emitted whenever data is updated or deleted on the given
67 > * scope and optional key.
68 > *
69 > * @param scope the `StorageScope` to listen to changes
70 > * @param key the optional key to filter for or all keys of
71 > * the scope if `undefined`
72 > */
73 > onDidChangeValue(scope: StorageScope.WORKSPACE, key: string | undefined, disposable: DisposableStore): Event<IWorkspaceStorageValueChangeEvent>;
74 > onDidChangeValue(scope: StorageScope.PROFILE, key: string | undefined, disposable: DisposableStore): Event<IProfileStorageValueChangeEvent>;
75 > onDidChangeValue(scope: StorageScope.APPLICATION, key: string | undefined, disposable: DisposableStore): Event<IApplicationStorageValueChangeEvent>;
76 > onDidChangeValue(scope: StorageScope.APPLICATION_SHARED, key: string | undefined, disposable: DisposableStore): Event<IApplicationSharedStorageValueChangeEvent>;
77 > onDidChangeValue(scope: StorageScope, key: string | undefined, disposable: DisposableStore): Event<IStorageValueChangeEvent>;
78 >
79 > /**
80 > * Emitted whenever target of a storage entry changes.
81 > */
82 > readonly onDidChangeTarget: Event<IStorageTargetChangeEvent>;
83 >
84 > /**
85 > * Emitted when the storage is about to persist. This is the right time
86 > * to persist data to ensure it is stored before the application shuts
87 > * down.
88 > *
89 > * The will save state event allows to optionally ask for the reason of
90 > * saving the state, e.g. to find out if the state is saved due to a
91 > * shutdown.
92 > *
93 > * Note: this event may be fired many times, not only on shutdown to prevent
94 > * loss of state in situations where the shutdown is not sufficient to
95 > * persist the data properly.
96 > */
97 > readonly onWillSaveState: Event<IWillSaveStateEvent>;
98 >
99 > /**
100 > * Retrieve an element stored with the given key from storage. Use
101 > * the provided `defaultValue` if the element is `null` or `undefined`.
102 > *
103 > * @param scope allows to define the scope of the storage operation
104 > * to either the current workspace only, all workspaces or all profiles.
105 > */
106 > get(key: string, scope: StorageScope, fallbackValue: string): string;
107 > get(key: string, scope: StorageScope, fallbackValue?: string): string | undefined;
108 >
109 > /**
110 > * Retrieve an element stored with the given key from storage. Use
111 > * the provided `defaultValue` if the element is `null` or `undefined`.
112 > * The element will be converted to a `boolean`.
113 > *
114 > * @param scope allows to define the scope of the storage operation
115 > * to either the current workspace only, all workspaces or all profiles.
116 > */
117 > getBoolean(key: string, scope: StorageScope, fallbackValue: boolean): boolean;
118 > getBoolean(key: string, scope: StorageScope, fallbackValue?: boolean): boolean | undefined;
119 >
120 > /**
121 > * Retrieve an element stored with the given key from storage. Use
122 > * the provided `defaultValue` if the element is `null` or `undefined`.
123 > * The element will be converted to a `number` using `parseInt` with a
124 > * base of `10`.
125 > *
126 > * @param scope allows to define the scope of the storage operation
127 > * to either the current workspace only, all workspaces or all profiles.
128 > */
129 > getNumber(key: string, scope: StorageScope, fallbackValue: number): number;
130 > getNumber(key: string, scope: StorageScope, fallbackValue?: number): number | undefined;
131 >
132 > /**
133 > * Retrieve an element stored with the given key from storage. Use
134 > * the provided `defaultValue` if the element is `null` or `undefined`.
135 > * The element will be converted to a `object` using `JSON.parse`.
136 > *
137 > * @param scope allows to define the scope of the storage operation
138 > * to either the current workspace only, all workspaces or all profiles.
139 > */
140 > getObject<T extends object>(key: string, scope: StorageScope, fallbackValue: T): T;
141 > getObject<T extends object>(key: string, scope: StorageScope, fallbackValue?: T): T | undefined;
142 >
143 > /**
144 > * Store a value under the given key to storage. The value will be
145 > * converted to a `string`. Storing either `undefined` or `null` will
146 > * remove the entry under the key.
147 > *
148 > * @param scope allows to define the scope of the storage operation
149 > * to either the current workspace only, all workspaces or all profiles.
150 > *
151 > * @param target allows to define the target of the storage operation
152 > * to either the current machine or user.
153 > */
154 > store(key: string, value: StorageValue, scope: StorageScope, target: StorageTarget): void;
155 >
156 > /**
157 > * Allows to store multiple values in a bulk operation. Events will only
158 > * be emitted when all values have been stored.
159 > *
160 > * @param external a hint to indicate the source of the operation is external,
161 > * such as settings sync or profile changes.
162 > */
163 > storeAll(entries: Array<IStorageEntry>, external: boolean): void;
164 >
165 > /**
166 > * Delete an element stored under the provided key from storage.
167 > *
168 > * The scope argument allows to define the scope of the storage
169 > * operation to either the current workspace only, all workspaces
170 > * or all profiles.
171 > */
172 > remove(key: string, scope: StorageScope): void;
173 >
174 > /**
175 > * Returns all the keys used in the storage for the provided `scope`
176 > * and `target`.
177 > *
178 > * Note: this will NOT return all keys stored in the storage layer.
179 > * Some keys may not have an associated `StorageTarget` and thus
180 > * will be excluded from the results.
181 > *
182 > * @param scope allows to define the scope for the keys
183 > * to either the current workspace only, all workspaces or all profiles.
184 > *
185 > * @param target allows to define the target for the keys
186 > * to either the current machine or user.
187 > */
188 > keys(scope: StorageScope, target: StorageTarget): string[];
189 >
190 > /**
191 > * Log the contents of the storage to the console.
192 > */
193 > log(): void;
194 >
195 > /**
196 > * Returns true if the storage service handles the provided scope.
197 > */
198 > hasScope(scope: IAnyWorkspaceIdentifier | IUserDataProfile): boolean;
199 >
200 > /**
201 > * Switch storage to another workspace or profile. Optionally preserve the
202 > * current data to the new storage.
203 > */
204 > switch(to: IAnyWorkspaceIdentifier | IUserDataProfile, preserveData: boolean): Promise<void>;
205 >
206 > /**
207 > * Whether the storage for the given scope was created during this session or
208 > * existed before.
209 > */
210 > isNew(scope: StorageScope): boolean;
211 >
212 > /**
213 > * Attempts to reduce the DB size via optimization commands if supported.
214 > */
215 > optimize(scope: StorageScope): Promise<void>;
216 >
217 > /**
218 > * Allows to flush state, e.g. in cases where a shutdown is
219 > * imminent. This will send out the `onWillSaveState` to ask
220 > * everyone for latest state.
221 > *
222 > * @returns a `Promise` that can be awaited on when all updates
223 > * to the underlying storage have been flushed.
224 > */
225 > flush(reason?: WillSaveStateReason): Promise<void>;
226 > }
227 >
228 > export const enum StorageScope {
229 >
230 > /**
231 > * The stored data will be scoped to all workspaces across all profiles
232 > * and shared across VS Code and Sessions app.
233 > */
234 > APPLICATION_SHARED = -2,
235 >
236 > /**
237 > * The stored data will be scoped to all workspaces across all profiles.
238 > */
239 > APPLICATION = -1,
240 >
241 > /**
242 > * The stored data will be scoped to all workspaces of the same profile.
243 > */
244 > PROFILE = 0,
245 >
246 > /**
247 > * The stored data will be scoped to the current workspace.
248 > */
249 > WORKSPACE = 1
250 > }
251 >
252 > export const enum StorageTarget {
253 >
254 > /**
255 > * The stored data is user specific and applies across machines.
256 > */
257 > USER,
258 >
259 > /**
260 > * The stored data is machine specific.
261 > */
262 > MACHINE
263 > }
264 >
265 > export interface IStorageValueChangeEvent {
266 >
267 > /**
268 > * The scope for the storage entry that changed
269 > * or was removed.
270 > */
271 > readonly scope: StorageScope;
272 >
273 > /**
274 > * The `key` of the storage entry that was changed
275 > * or was removed.
276 > */
277 > readonly key: string;
278 >
279 > /**
280 > * The `target` can be `undefined` if a key is being
281 > * removed.
282 > */
283 > readonly target: StorageTarget | undefined;
284 >
285 > /**
286 > * A hint how the storage change event was triggered. If
287 > * `true`, the storage change was triggered by an external
288 > * source, such as:
289 > * - another process (for example another window)
290 > * - operations such as settings sync or profiles change
291 > */
292 > readonly external?: boolean;
293 > }
294 >
295 > export interface IStorageTargetChangeEvent {
296 >
297 > /**
298 > * The scope for the target that changed. Listeners
299 > * should use `keys(scope, target)` to get an updated
300 > * list of keys for the given `scope` and `target`.
301 > */
302 > readonly scope: StorageScope;
303 > }
304 >
305 > interface IKeyTargets {
306 > [key: string]: StorageTarget;
307 > }
308 >
309 > export interface IStorageServiceOptions {
310 > readonly flushInterval: number;
311 > }
312 >
313 > export function loadKeyTargets(storage: IStorage): IKeyTargets {
314 const keysRaw = storage.get(TARGET_KEY);
315 if (keysRaw) {
323 return Object.create(null);
324 }
325 > storage.ts
326 > export abstract class AbstractStorageService extends Disposable implements IStorageService {
327 >
328 > declare readonly _serviceBrand: undefined;
329 >
330 > private static DEFAULT_FLUSH_INTERVAL = 60 * 1000; // every minute
331 >
332 > private readonly _onDidChangeValue = this._register(new PauseableEmitter<IStorageValueChangeEvent>());
333 >
334 > private readonly _onDidChangeTarget = this._register(new PauseableEmitter<IStorageTargetChangeEvent>());
335 > readonly onDidChangeTarget = this._onDidChangeTarget.event;
336 >
337 > private readonly _onWillSaveState = this._register(new Emitter<IWillSaveStateEvent>());
338 > readonly onWillSaveState = this._onWillSaveState.event;
339 >
340 > private initializationPromise: Promise<void> | undefined;
341 >
342 > private readonly flushWhenIdleScheduler: RunOnceScheduler;
343 > private readonly runFlushWhenIdle = this._register(new MutableDisposable());
344 >
345 > constructor(options: IStorageServiceOptions = { flushInterval: AbstractStorageService.DEFAULT_FLUSH_INTERVAL }) {
346 > super(); storage.ts
347 >
348 > this.flushWhenIdleScheduler = this._register(new RunOnceScheduler(() => this.doFlushWhenIdle(), options.flushInterval));
349 > }
350 > storage.ts
351 > onDidChangeValue(scope: StorageScope.WORKSPACE, key: string | undefined, disposable: DisposableStore): Event<IWorkspaceStorageValueChangeEvent>;
352 > onDidChangeValue(scope: StorageScope.PROFILE, key: string | undefined, disposable: DisposableStore): Event<IProfileStorageValueChangeEvent>;
353 > onDidChangeValue(scope: StorageScope.APPLICATION, key: string | undefined, disposable: DisposableStore): Event<IApplicationStorageValueChangeEvent>;
354 > onDidChangeValue(scope: StorageScope.APPLICATION_SHARED, key: string | undefined, disposable: DisposableStore): Event<IApplicationSharedStorageValueChangeEvent>;
355 > onDidChangeValue(scope: StorageScope, key: string | undefined, disposable: DisposableStore): Event<IStorageValueChangeEvent> {
356 return Event.filter(this._onDidChangeValue.event, e => e.scope === scope && (key === undefined || e.key === key), disposable);
357 }
358 > storage.ts
359 > private doFlushWhenIdle(): void {
360 this.runFlushWhenIdle.value = runWhenGlobalIdle(() => {
361 if (this.shouldFlushWhenIdle()) {
367 });
368 }
369 > storage.ts
370 > protected shouldFlushWhenIdle(): boolean {
371 return true;
372 }
373 > storage.ts
374 > protected stopFlushWhenIdle(): void {
375 dispose([this.runFlushWhenIdle, this.flushWhenIdleScheduler]);
376 }
377 > storage.ts
378 > initialize(): Promise<void> {
379 if (!this.initializationPromise) {
380 this.initializationPromise = (async () => {
402 return this.initializationPromise;
403 }
404 > storage.ts
405 > protected emitDidChangeValue(scope: StorageScope, event: IStorageChangeEvent): void {
406 const { key, external } = event;
407
434 }
435 }
436 > storage.ts
437 > protected emitWillSaveState(reason: WillSaveStateReason): void {
438 this._onWillSaveState.fire({ reason });
439 }
440 > storage.ts
441 > get(key: string, scope: StorageScope, fallbackValue: string): string;
442 > get(key: string, scope: StorageScope): string | undefined;
443 > get(key: string, scope: StorageScope, fallbackValue?: string): string | undefined {
444 > return this.getStorage(scope)?.get(key, fallbackValue); storage.ts
445 > }
446 > storage.ts
447 > getBoolean(key: string, scope: StorageScope, fallbackValue: boolean): boolean;
448 > getBoolean(key: string, scope: StorageScope): boolean | undefined;
449 > getBoolean(key: string, scope: StorageScope, fallbackValue?: boolean): boolean | undefined {
450 return this.getStorage(scope)?.getBoolean(key, fallbackValue);
451 }
452 > storage.ts
453 > getNumber(key: string, scope: StorageScope, fallbackValue: number): number;
454 > getNumber(key: string, scope: StorageScope): number | undefined;
455 > getNumber(key: string, scope: StorageScope, fallbackValue?: number): number | undefined {
456 return this.getStorage(scope)?.getNumber(key, fallbackValue);
457 }
458 > storage.ts
459 > getObject(key: string, scope: StorageScope, fallbackValue: object): object;
460 > getObject(key: string, scope: StorageScope): object | undefined;
461 > getObject(key: string, scope: StorageScope, fallbackValue?: object): object | undefined {
462 > return this.getStorage(scope)?.getObject(key, fallbackValue); storage.ts
463 > }
464 > storage.ts
465 > storeAll(entries: Array<IStorageEntry>, external: boolean): void {
466 this.withPausedEmitters(() => {
467 for (const entry of entries) {
470 });
471 }
472 > storage.ts
473 > store(key: string, value: StorageValue, scope: StorageScope, target: StorageTarget, external = false): void {
474
475 // We remove the key for undefined/null values
489 });
490 }
491 > storage.ts
492 > remove(key: string, scope: StorageScope, external = false): void {
493
494 // Update our datastructures but send events only after
502 });
503 }
504 > storage.ts
505 > private withPausedEmitters(fn: Function): void {
506
507 // Pause emitters
518 }
519 }
520 > storage.ts
521 > keys(scope: StorageScope, target: StorageTarget): string[] {
522 const keys: string[] = [];
523
532 return keys;
533 }
534 > storage.ts
535 > private updateKeyTarget(key: string, scope: StorageScope, target: StorageTarget | undefined, external = false): void {
536
537 // Add
552 }
553 }
554 > storage.ts
555 > private _workspaceKeyTargets: IKeyTargets | undefined = undefined;
556 > private get workspaceKeyTargets(): IKeyTargets {
557 if (!this._workspaceKeyTargets) {
558 this._workspaceKeyTargets = this.loadKeyTargets(StorageScope.WORKSPACE);
561 return this._workspaceKeyTargets;
562 }
563 > storage.ts
564 > private _profileKeyTargets: IKeyTargets | undefined = undefined;
565 > private get profileKeyTargets(): IKeyTargets {
566 if (!this._profileKeyTargets) {
567 this._profileKeyTargets = this.loadKeyTargets(StorageScope.PROFILE);
570 return this._profileKeyTargets;
571 }
572 > storage.ts
573 > private _applicationKeyTargets: IKeyTargets | undefined = undefined;
574 > private get applicationKeyTargets(): IKeyTargets {
575 if (!this._applicationKeyTargets) {
576 this._applicationKeyTargets = this.loadKeyTargets(StorageScope.APPLICATION);
579 return this._applicationKeyTargets;
580 }
581 > storage.ts
582 > private _applicationSharedKeyTargets: IKeyTargets | undefined = undefined;
583 > private get applicationSharedKeyTargets(): IKeyTargets {
584 if (!this._applicationSharedKeyTargets) {
585 this._applicationSharedKeyTargets = this.loadKeyTargets(StorageScope.APPLICATION_SHARED);
588 return this._applicationSharedKeyTargets;
589 }
590 > storage.ts
591 > private getKeyTargets(scope: StorageScope): IKeyTargets {
592 switch (scope) {
593 case StorageScope.APPLICATION_SHARED:
601 }
602 }
603 > storage.ts
604 > private loadKeyTargets(scope: StorageScope): { [key: string]: StorageTarget } {
605 const storage = this.getStorage(scope);
606
607 return storage ? loadKeyTargets(storage) : Object.create(null);
608 }
609 > storage.ts
610 > isNew(scope: StorageScope): boolean {
611 return this.getBoolean(IS_NEW_KEY, scope) === true;
612 }
613 > storage.ts
614 > async flush(reason = WillSaveStateReason.NONE): Promise<void> {
615
616 // Signal event to collect changes
646 }
647 }
648 > storage.ts
649 > async log(): Promise<void> {
650 const applicationItems = this.getStorage(StorageScope.APPLICATION)?.items ?? new Map<string, string>();
651 const applicationSharedItems = this.getStorage(StorageScope.APPLICATION_SHARED)?.items ?? new Map<string, string>();
664 );
665 }
666 > storage.ts
667 > async optimize(scope: StorageScope): Promise<void> {
668
669 // Await pending data to be flushed to the DB
673 return this.getStorage(scope)?.optimize();
674 }
675 > storage.ts
676 > async switch(to: IAnyWorkspaceIdentifier | IUserDataProfile, preserveData: boolean): Promise<void> {
677
678 // Signal as event so that clients can store data before we switch
685 return this.switchToWorkspace(to, preserveData);
686 }
687 > storage.ts
688 > protected canSwitchProfile(from: IUserDataProfile, to: IUserDataProfile): boolean {
689 if (from.id === to.id) {
690 return false; // both profiles are same
697 return true;
698 }
699 > storage.ts
700 > protected switchData(oldStorage: Map<string, string>, newStorage: IStorage, scope: StorageScope): void {
701 this.withPausedEmitters(() => {
702 // Signal storage keys that have changed
718 });
719 }
720 > storage.ts
721 > // --- abstract
722 >
723 > abstract hasScope(scope: IAnyWorkspaceIdentifier | IUserDataProfile): boolean;
724 >
725 > protected abstract doInitialize(): Promise<void>;
726 >
727 > protected abstract getStorage(scope: StorageScope): IStorage | undefined;
728 >
729 > protected abstract getLogDetails(scope: StorageScope): string | undefined;
730 >
731 > protected abstract switchToProfile(toProfile: IUserDataProfile, preserveData: boolean): Promise<void>;
732 > protected abstract switchToWorkspace(toWorkspace: IAnyWorkspaceIdentifier | IUserDataProfile, preserveData: boolean): Promise<void>;
733 > }
734 >
735 > export function isProfileUsingDefaultStorage(profile: IUserDataProfile): boolean {
736 return profile.isDefault || !!profile.useDefaultFlags?.globalState;
737 }
738 > storage.ts
739 > export class InMemoryStorageService extends AbstractStorageService {
740 >
741 > private readonly applicationStorage = this._register(new Storage(new InMemoryStorageDatabase(), { hint: StorageHint.STORAGE_IN_MEMORY }));
742 > private readonly applicationSharedStorage = this._register(new Storage(new InMemoryStorageDatabase(), { hint: StorageHint.STORAGE_IN_MEMORY }));
743 > private readonly profileStorage = this._register(new Storage(new InMemoryStorageDatabase(), { hint: StorageHint.STORAGE_IN_MEMORY }));
744 > private readonly workspaceStorage = this._register(new Storage(new InMemoryStorageDatabase(), { hint: StorageHint.STORAGE_IN_MEMORY }));
745 >
746 > constructor() {
747 > super(); storage.ts
748 >
749 > this._register(this.workspaceStorage.onDidChangeStorage(e => this.emitDidChangeValue(StorageScope.WORKSPACE, e)));
750 > this._register(this.profileStorage.onDidChangeStorage(e => this.emitDidChangeValue(StorageScope.PROFILE, e)));
751 > this._register(this.applicationStorage.onDidChangeStorage(e => this.emitDidChangeValue(StorageScope.APPLICATION, e)));
752 > this._register(this.applicationSharedStorage.onDidChangeStorage(e => this.emitDidChangeValue(StorageScope.APPLICATION_SHARED, e)));
753 > }
754 > storage.ts
755 > protected getStorage(scope: StorageScope): IStorage {
756 > switch (scope) { storage.ts
757 > case StorageScope.APPLICATION_SHARED:
758 return this.applicationSharedStorage;
759 > case StorageScope.APPLICATION: storage.ts
760 return this.applicationStorage;
761 > case StorageScope.PROFILE: storage.ts
762 > return this.profileStorage; storage.ts
763 > default: storage.ts
764 > return this.workspaceStorage; storage.ts
765 > } storage.ts
766 > }
767 > storage.ts
768 > protected getLogDetails(scope: StorageScope): string | undefined {
769 switch (scope) {
770 case StorageScope.APPLICATION_SHARED:
778 }
779 }
780 > storage.ts
781 > protected async doInitialize(): Promise<void> { }
782 >
783 > protected async switchToProfile(): Promise<void> {
784 // no-op when in-memory
785 }
786 > storage.ts
787 > protected async switchToWorkspace(): Promise<void> {
788 // no-op when in-memory
789 }
790 > storage.ts
791 > protected override shouldFlushWhenIdle(): boolean {
792 return false;
793 }
794 > storage.ts
795 > hasScope(scope: IAnyWorkspaceIdentifier | IUserDataProfile): boolean {
796 return false;
797 }
798 > } storage.ts
799 >
800 export async function logStorage(application: Map<string, string>, applicationShared: Map<string, string>, profile: Map<string, string>, workspace: Map<string, string>, applicationPath: string, applicationSharedPath: string, profilePath: string, workspacePath: string): Promise<void> {
801 const safeParse = (value: string) => {
src/vs/platform/notification/common/notification.ts 464 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- notification.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../../nls.js';
7 > import { IAction } from '../../../base/common/actions.js';
8 > import { Event } from '../../../base/common/event.js';
9 > import BaseSeverity from '../../../base/common/severity.js';
10 > import { createDecorator } from '../../instantiation/common/instantiation.js';
11 >
12 > export import Severity = BaseSeverity;
13 >
14 > export const INotificationService = createDecorator<INotificationService>('notificationService');
15 >
16 > export type NotificationMessage = string | Error;
17 >
18 > export enum NotificationPriority {
19 >
20 > /**
21 > * Default priority: notification will be visible unless do not disturb mode is enabled.
22 > */
23 > DEFAULT,
24 >
25 > /**
26 > * Optional priority: notification might only be visible from the notifications center.
27 > */
28 > OPTIONAL,
29 >
30 > /**
31 > * Silent priority: notification will only be visible from the notifications center.
32 > */
33 > SILENT,
34 >
35 > /**
36 > * Urgent priority: notification will be visible even when do not disturb mode is enabled.
37 > */
38 > URGENT
39 > }
40 >
41 > export interface INotificationProperties {
42 >
43 > /**
44 > * Sticky notifications are not automatically removed after a certain timeout.
45 > *
46 > * Currently, only 2 kinds of notifications are sticky:
47 > * - Error notifications with primary actions
48 > * - Notifications that show progress
49 > */
50 > readonly sticky?: boolean;
51 >
52 > /**
53 > * Allows to override the priority of the notification based on needs.
54 > */
55 > readonly priority?: NotificationPriority;
56 >
57 > /**
58 > * Adds an action to never show the notification again. The choice will be persisted
59 > * such as future requests will not cause the notification to show again.
60 > */
61 > readonly neverShowAgain?: INeverShowAgainOptions;
62 > }
63 >
64 > export enum NeverShowAgainScope {
65 >
66 > /**
67 > * Will never show this notification on the current workspace again.
68 > */
69 > WORKSPACE,
70 >
71 > /**
72 > * Will never show this notification on any workspace of the same
73 > * profile again.
74 > */
75 > PROFILE,
76 >
77 > /**
78 > * Will never show this notification on any workspace across all
79 > * profiles again.
80 > */
81 > APPLICATION
82 > }
83 >
84 > export interface INeverShowAgainOptions {
85 >
86 > /**
87 > * The id is used to persist the selection of not showing the notification again.
88 > */
89 > readonly id: string;
90 >
91 > /**
92 > * By default the action will show up as primary action. Setting this to true will
93 > * make it a secondary action instead.
94 > */
95 > readonly isSecondary?: boolean;
96 >
97 > /**
98 > * Whether to persist the choice in the current workspace or for all workspaces. By
99 > * default it will be persisted for all workspaces across all profiles
100 > * (= `NeverShowAgainScope.APPLICATION`).
101 > */
102 > readonly scope?: NeverShowAgainScope;
103 > }
104 >
105 > export interface INotificationSource {
106 >
107 > /**
108 > * The id of the source.
109 > */
110 > readonly id: string;
111 >
112 > /**
113 > * The label of the source.
114 > */
115 > readonly label: string;
116 > }
117 >
118 > export function isNotificationSource(thing: unknown): thing is INotificationSource {
119 if (thing) {
120 const candidate = thing as INotificationSource;
125 return false;
126 }
128 > export interface INotification extends INotificationProperties {
129 >
130 > /**
131 > * The id of the notification. If provided, will be used to compare
132 > * notifications with others to decide whether a notification is
133 > * duplicate or not.
134 > */
135 > readonly id?: string;
136 >
137 > /**
138 > * The severity of the notification. Either `Info`, `Warning` or `Error`.
139 > */
140 > readonly severity: Severity;
141 >
142 > /**
143 > * The message of the notification. This can either be a `string` or `Error`. Messages
144 > * can optionally include links in the format: `[text](link)`
145 > */
146 > readonly message: NotificationMessage;
147 >
148 > /**
149 > * The source of the notification appears as additional information.
150 > */
151 > readonly source?: string | INotificationSource;
152 >
153 > /**
154 > * Actions to show as part of the notification. Primary actions show up as
155 > * buttons as part of the message and will close the notification once clicked.
156 > *
157 > * Secondary actions are meant to provide additional configuration or context
158 > * for the notification and will show up less prominent. A notification does not
159 > * close automatically when invoking a secondary action.
160 > *
161 > * **Note:** If your intent is to show a message with actions to the user, consider
162 > * the `INotificationService.prompt()` method instead which are optimized for
163 > * this usecase and much easier to use!
164 > */
165 > actions?: INotificationActions;
166 >
167 > /**
168 > * The initial set of progress properties for the notification. To update progress
169 > * later on, access the `INotificationHandle.progress` property.
170 > */
171 > readonly progress?: INotificationProgressProperties;
172 > }
173 >
174 > export interface INotificationActions {
175 >
176 > /**
177 > * Primary actions show up as buttons as part of the message and will close
178 > * the notification once clicked.
179 > *
180 > * Pass `ActionWithMenuAction` for an action that has additional menu actions.
181 > */
182 > readonly primary?: readonly IAction[];
183 >
184 > /**
185 > * Secondary actions are meant to provide additional configuration or context
186 > * for the notification and will show up less prominent. A notification does not
187 > * close automatically when invoking a secondary action.
188 > */
189 > readonly secondary?: readonly IAction[];
190 > }
191 >
192 > export interface INotificationProgressProperties {
193 >
194 > /**
195 > * Causes the progress bar to spin infinitley.
196 > */
197 > readonly infinite?: boolean;
198 >
199 > /**
200 > * Indicate the total amount of work.
201 > */
202 > readonly total?: number;
203 >
204 > /**
205 > * Indicate that a specific chunk of work is done.
206 > */
207 > readonly worked?: number;
208 > }
209 >
210 > export interface INotificationProgress {
211 >
212 > /**
213 > * Causes the progress bar to spin infinitley.
214 > */
215 > infinite(): void;
216 >
217 > /**
218 > * Indicate the total amount of work.
219 > */
220 > total(value: number): void;
221 >
222 > /**
223 > * Indicate that a specific chunk of work is done.
224 > */
225 > worked(value: number): void;
226 >
227 > /**
228 > * Indicate that the long running operation is done.
229 > */
230 > done(): void;
231 > }
232 >
233 > export interface INotificationHandle {
234 >
235 > /**
236 > * Will be fired once the notification is closed.
237 > */
238 > readonly onDidClose: Event<void>;
239 >
240 > /**
241 > * Will be fired whenever the visibility of the notification changes.
242 > * A notification can either be visible as toast or inside the notification
243 > * center if it is visible.
244 > */
245 > readonly onDidChangeVisibility: Event<boolean>;
246 >
247 > /**
248 > * Allows to indicate progress on the notification even after the
249 > * notification is already visible.
250 > */
251 > readonly progress: INotificationProgress;
252 >
253 > /**
254 > * Allows to update the severity of the notification.
255 > */
256 > updateSeverity(severity: Severity): void;
257 >
258 > /**
259 > * Allows to update the message of the notification even after the
260 > * notification is already visible.
261 > */
262 > updateMessage(message: NotificationMessage): void;
263 >
264 > /**
265 > * Allows to update the actions of the notification even after the
266 > * notification is already visible.
267 > */
268 > updateActions(actions?: INotificationActions): void;
269 >
270 > /**
271 > * Hide the notification and remove it from the notification center.
272 > */
273 > close(): void;
274 > }
275 >
276 > export interface IStatusHandle {
277 >
278 > /**
279 > * Hide the status message.
280 > */
281 > close(): void;
282 > }
283 >
284 > interface IBasePromptChoice {
285 >
286 > /**
287 > * Label to show for the choice to the user.
288 > */
289 > readonly label: string;
290 >
291 > /**
292 > * Whether to keep the notification open after the choice was selected
293 > * by the user. By default, will close the notification upon click.
294 > */
295 > readonly keepOpen?: boolean;
296 >
297 > /**
298 > * Triggered when the user selects the choice.
299 > */
300 > run: () => void;
301 > }
302 >
303 > export interface IPromptChoice extends IBasePromptChoice {
304 >
305 > /**
306 > * Primary choices show up as buttons in the notification below the message.
307 > * Secondary choices show up under the gear icon in the header of the notification.
308 > */
309 > readonly isSecondary?: boolean;
310 > }
311 >
312 > export interface IPromptChoiceWithMenu extends IPromptChoice {
313 >
314 > /**
315 > * Additional choices those will be shown in the dropdown menu for this choice.
316 > */
317 > readonly menu: IBasePromptChoice[];
318 >
319 > /**
320 > * Menu is not supported on secondary choices
321 > */
322 > readonly isSecondary: false | undefined;
323 > }
324 >
325 > export interface IPromptOptions extends INotificationProperties {
326 >
327 > /**
328 > * Will be called if the user closed the notification without picking
329 > * any of the provided choices.
330 > */
331 > onCancel?: () => void;
332 > }
333 >
334 > export interface IStatusMessageOptions {
335 >
336 > /**
337 > * An optional timeout after which the status message should show. By default
338 > * the status message will show immediately.
339 > */
340 > readonly showAfter?: number;
341 >
342 > /**
343 > * An optional timeout after which the status message is to be hidden. By default
344 > * the status message will not hide until another status message is displayed.
345 > */
346 > readonly hideAfter?: number;
347 > }
348 >
349 > export enum NotificationsFilter {
350 >
351 > /**
352 > * No filter is enabled.
353 > */
354 > OFF,
355 >
356 > /**
357 > * All notifications are silent except error notifications.
358 > */
359 > ERROR
360 > }
361 >
362 > export interface INotificationSourceFilter extends INotificationSource {
363 > readonly filter: NotificationsFilter;
364 > }
365 >
366 > /**
367 > * A service to bring up notifications and non-modal prompts.
368 > *
369 > * Note: use the `IDialogService` for a modal way to ask the user for input.
370 > */
371 > export interface INotificationService {
372 >
373 > readonly _serviceBrand: undefined;
374 >
375 > /**
376 > * Emitted when the notifications filter changed.
377 > */
378 > readonly onDidChangeFilter: Event<void>;
379 >
380 > /**
381 > * Sets a notification filter either for all notifications
382 > * or for a specific source.
383 > */
384 > setFilter(filter: NotificationsFilter | INotificationSourceFilter): void;
385 >
386 > /**
387 > * Gets the notification filter either for all notifications
388 > * or for a specific source.
389 > */
390 > getFilter(source?: INotificationSource): NotificationsFilter;
391 >
392 > /**
393 > * Returns all filters with their sources.
394 > */
395 > getFilters(): INotificationSourceFilter[];
396 >
397 > /**
398 > * Removes a filter for a specific source.
399 > */
400 > removeFilter(sourceId: string): void;
401 >
402 > /**
403 > * Show the provided notification to the user. The returned `INotificationHandle`
404 > * can be used to control the notification afterwards.
405 > *
406 > * **Note:** If your intent is to show a message with actions to the user, consider
407 > * the `INotificationService.prompt()` method instead which are optimized for
408 > * this usecase and much easier to use!
409 > *
410 > * @returns a handle on the notification to e.g. hide it or update message, buttons, etc.
411 > */
412 > notify(notification: INotification): INotificationHandle;
413 >
414 > /**
415 > * A convenient way of reporting infos. Use the `INotificationService.notify`
416 > * method if you need more control over the notification.
417 > */
418 > info(message: NotificationMessage | NotificationMessage[]): void;
419 >
420 > /**
421 > * A convenient way of reporting warnings. Use the `INotificationService.notify`
422 > * method if you need more control over the notification.
423 > */
424 > warn(message: NotificationMessage | NotificationMessage[]): void;
425 >
426 > /**
427 > * A convenient way of reporting errors. Use the `INotificationService.notify`
428 > * method if you need more control over the notification.
429 > */
430 > error(message: NotificationMessage | NotificationMessage[]): void;
431 >
432 > /**
433 > * Shows a prompt in the notification area with the provided choices. The prompt
434 > * is non-modal. If you want to show a modal dialog instead, use `IDialogService`.
435 > *
436 > * @param severity the severity of the notification. Either `Info`, `Warning` or `Error`.
437 > * @param message the message to show as status.
438 > * @param choices options to be chosen from.
439 > * @param options provides some optional configuration options.
440 > *
441 > * @returns a handle on the notification to e.g. hide it or update message, buttons, etc.
442 > */
443 > prompt(severity: Severity, message: string, choices: (IPromptChoice | IPromptChoiceWithMenu)[], options?: IPromptOptions): INotificationHandle;
444 >
445 > /**
446 > * Shows a status message in the status area with the provided text.
447 > *
448 > * @param message the message to show as status
449 > * @param options provides some optional configuration options
450 > *
451 > * @returns a handle to hide the status message
452 > */
453 > status(message: NotificationMessage, options?: IStatusMessageOptions): IStatusHandle;
454 > }
455 >
456 > export class NoOpNotification implements INotificationHandle {
457
458 readonly progress = new NoOpProgress();
460 readonly onDidClose = Event.None;
461 readonly onDidChangeVisibility = Event.None;
463 > updateSeverity(severity: Severity): void { }
464 > updateMessage(message: NotificationMessage): void { }
465 > updateActions(actions?: INotificationActions): void { }
466 >
467 > close(): void { }
468 > }
469 >
470 > export class NoOpProgress implements INotificationProgress {
471 > infinite(): void { }
472 > done(): void { }
473 > total(value: number): void { }
474 > worked(value: number): void { }
475 > }
476 >
477 > export function withSeverityPrefix(label: string, severity: Severity): string {
478
479 // Add severity prefix to match WCAG 4.1.3 Status
src/vs/workbench/contrib/chat/common/customizationHarnessService.ts 452 covered LOC · 16 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- customizationHarnessService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Codicon } from '../../../../base/common/codicons.js';
7 > import { derived, IObservable, ISettableObservable, observableValue } from '../../../../base/common/observable.js';
8 > import { IDisposable } from '../../../../base/common/lifecycle.js';
9 > import { Emitter, Event } from '../../../../base/common/event.js';
10 > import { ThemeIcon } from '../../../../base/common/themables.js';
11 > import { URI } from '../../../../base/common/uri.js';
12 > import { localize } from '../../../../nls.js';
13 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
14 > import { AICustomizationManagementSection, AICustomizationSource, BUILTIN_STORAGE } from './aiCustomizationWorkspaceService.js';
15 > import { PromptsType } from './promptSyntax/promptTypes.js';
16 > import { AGENT_MD_FILENAME } from './promptSyntax/config/promptFileLocations.js';
17 > import { IAgentSource, IChatPromptSlashCommand, ICustomAgent, IPromptsService, IResolvedChatPromptSlashCommand, matchesSessionType, PromptsStorage } from './promptSyntax/service/promptsService.js';
18 > import { CancellationToken } from '../../../../base/common/cancellation.js';
19 > import { SessionType } from './chatSessionsService.js';
20 > import { CustomAgent } from './promptSyntax/service/promptsServiceImpl.js';
21 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
22 > import { getCanonicalPluginCommandId } from './plugins/agentPluginService.js';
23 > import { getChatSessionType, LocalChatSessionUri } from './model/chatUri.js';
24 >
25 >
26 > export const ICustomizationHarnessService = createDecorator<ICustomizationHarnessService>('customizationHarnessService');
27 >
28 > /**
29 > * Override for a management section's create-button behavior.
30 > */
31 > export interface ISectionOverride {
32 > /**
33 > * Label for the primary button. Required when `commandId` or `rootFile`
34 > * is set. Ignored otherwise (the widget uses its default label).
35 > */
36 > readonly label?: string;
37 > /** When set, the primary button invokes this command (e.g. hooks quick pick). */
38 > readonly commandId?: string;
39 > /** When set, the primary button creates this file at the workspace root. */
40 > readonly rootFile?: string;
41 > /**
42 > * Custom type label for the dropdown workspace/user create actions
43 > * (e.g. "Rule" instead of "Instruction"). When undefined, the
44 > * section's default type label is used.
45 > */
46 > readonly typeLabel?: string;
47 > /**
48 > * Root-level file shortcuts added to the dropdown (e.g. `['AGENTS.md']`).
49 > * Each entry creates a "New {filename}" action that creates the file at
50 > * the workspace root. Harnesses that don't support a file simply omit it.
51 > */
52 > readonly rootFileShortcuts?: readonly string[];
53 > /**
54 > * File extension override for new files created under this section.
55 > * When set, files are created with this extension (e.g. `.md` for
56 > * Claude rules) instead of the default for the prompt type
57 > * (e.g. `.instructions.md`).
58 > */
59 > readonly fileExtension?: string;
60 > }
61 >
62 > export interface ICustomizationItemAction {
63 > readonly id: string;
64 > readonly label: string;
65 > readonly tooltip?: string;
66 > readonly icon?: ThemeIcon;
67 > readonly enabled?: boolean;
68 > run(): void | Promise<void>;
69 > }
70 >
71 > /**
72 > * Describes a single harness option for the UI toggle.
73 > */
74 > export interface IHarnessDescriptor {
75 > /**
76 > * The harness/session-type identifier.
77 > */
78 > readonly id: string;
79 > readonly label: string;
80 > readonly icon: ThemeIcon;
81 > /**
82 > * Management sections that should be hidden when this harness is active.
83 > * For example, Claude does not support prompt files so the Prompts
84 > * section is hidden.
85 > */
86 > readonly hiddenSections?: readonly string[];
87 > /**
88 > * When `true`, the "Generate with AI" sparkle button is hidden and replaced
89 > * with a plain "New X" manual-creation button (like sessions).
90 > */
91 > readonly hideGenerateButton?: boolean;
92 > /**
93 > * Per-section overrides for the create button behavior.
94 > *
95 > * A `commandId` entry replaces the button entirely with a command
96 > * invocation (e.g. Claude hooks → `copilot.claude.hooks`).
97 > *
98 > * A `rootFile` entry makes the primary button create a specific file
99 > * at the workspace root (e.g. Claude instructions → `CLAUDE.md`).
100 > * When combined with `typeLabel`, the dropdown create actions use
101 > * that label instead of the section's default (e.g. "Rule" instead
102 > * of "Instruction").
103 > */
104 > readonly sectionOverrides?: ReadonlyMap<string, ISectionOverride>;
105 > /**
106 > * The chat agent ID that must be registered for this harness to appear.
107 > * When `undefined`, the harness is always available (e.g. Local).
108 > */
109 > readonly requiredAgentId?: string;
110 > /**
111 > * When set, this harness is backed by an extension-contributed provider
112 > * that can supply customization items directly (bypassing promptsService
113 > * discovery and filtering).
114 > */
115 > readonly itemProvider?: ICustomizationItemProvider;
116 > /**
117 > * When `true`, the "Troubleshoot" action is available in item context
118 > * menus. This opens chat with the `/troubleshoot` command pre-filled
119 > * for the selected customization.
120 > */
121 > readonly supportsTroubleshoot?: boolean;
122 > /**
123 > * When set, this harness uses an opt-out sync model where all eligible
124 > * local customizations are synced by default. The UI shows disable
125 > * affordances when this harness is active.
126 > */
127 > readonly syncProvider?: ICustomizationSyncProvider;
128 > /**
129 > * Optional plugin-management actions shown in the Plugins section add menu.
130 > * Harnesses can use these to add environment-specific commands alongside
131 > * the default install-from-source action (for example, configuring plugins on
132 > * a remote agent host). The create action remains a separate toolbar button.
133 > */
134 > readonly pluginActions?: readonly ICustomizationItemAction[];
135 > }
136 >
137 > /**
138 > * Represents a customization item provided by any source.
139 > */
140 > export interface ICustomizationItem {
141 > /** Optional stable identity used by list widgets when URI alone is not unique. */
142 > readonly itemKey?: string;
143 > readonly uri: URI;
144 > readonly type: string;
145 > readonly name: string;
146 > readonly description?: string;
147 > /** Customization source (local, user, extension, plugin, builtin). Set by providers that know the source. */
148 > readonly source: AICustomizationSource;
149 > /** The extension identifier that contributed this customization, if any. */
150 > readonly extensionId: string | undefined;
151 > /** The URI of the plugin that contributed this customization, if any. */
152 > readonly pluginUri: URI | undefined;
153 > /** Human-readable name of the plugin that contributed this customization, if any. */
154 > readonly pluginLabel?: string;
155 > /** Server-reported loading status for this customization. */
156 > readonly status?: 'loading' | 'loaded' | 'degraded' | 'error';
157 > /** Human-readable status detail (e.g. error message or warning). */
158 > readonly statusMessage?: string;
159 > /** Whether this customization is currently enabled. */
160 > readonly enabled?: boolean;
161 > /** When set, items with the same groupKey are displayed under a shared collapsible header. */
162 > readonly groupKey?: string;
163 > /** When set, shows a small inline badge next to the item name (e.g. an applyTo glob pattern). */
164 > readonly badge?: string;
165 > /** Tooltip shown when hovering the badge. */
166 > readonly badgeTooltip?: string;
167 > /**
168 > * Whether this customization item can be invoked by the user.
169 > * Relevant for prompt / skill and custom agents
170 > */
171 > readonly userInvocable?: boolean;
172 > /** Optional inline/context-menu actions specific to this item. */
173 > readonly actions?: readonly ICustomizationItemAction[];
174 > }
175 >
176 > export interface ICustomizationAgentRef {
177 > readonly id: string;
178 >
179 > readonly uri: URI;
180 > /** Agent name (from frontmatter `name`, or file-derived) */
181 > readonly name: string;
182 > /** Optional short description for UI preview (from frontmatter `description`) */
183 > readonly description?: string;
184 > }
185 >
186 > export function isPluginCustomizationItem(item: { readonly type: string }): boolean {
187 return item.type === 'plugin' || item.type === AICustomizationManagementSection.Plugins;
188 }
190 > /**
191 > * Provider interface for extension-contributed harnesses that supply
192 > * customization items directly from their SDK.
193 > */
194 > export interface ICustomizationItemProvider {
195 > /**
196 > * Event that fires when the provider's customizations change.
197 > */
198 > readonly onDidChange: Event<void>;
199 > /**
200 > * Provide the customization items this harness supports.
201 > *
202 > * @param sessionResource URI of the chat session whose
203 > * customizations should be included. Providers that surface
204 > * session-scoped state (e.g. an agent host) should read from
205 > * this session.
206 > */
207 > provideChatSessionCustomizations(sessionResource: URI, token: CancellationToken): Promise<ICustomizationItem[] | undefined>;
208 >
209 > /**
210 > * Provide the custom agents this harness supports.
211 > *
212 > * @param sessionResource URI of the chat session whose
213 > * customizations should be included. Providers that surface
214 > * session-scoped state (e.g. an agent host) should read from
215 > * this session.
216 > */
217 > provideCustomAgents?(sessionResource: URI, token: CancellationToken): Promise<readonly ICustomAgent[]>;
218 >
219 > /**
220 > * Provide the directories where new customization files of the given
221 > * type can be created for this session. The result includes both
222 > * workspace-scoped and user-scoped folders; the caller is responsible
223 > * for partitioning them by storage target.
224 > *
225 > * @param sessionResource URI of the chat session whose
226 > * creation locations should be returned.
227 > */
228 > provideSourceFolders?(sessionResource: URI, type: PromptsType, token: CancellationToken): Promise<readonly ICustomizationSourceFolder[] | undefined>;
229 > }
230 >
231 > /**
232 > * A directory where new customization files of a given type can be created.
233 > */
234 > export interface ICustomizationSourceFolder {
235 > readonly uri: URI;
236 > /** Display label for the picker when multiple folders are offered. */
237 > readonly label: string;
238 > /** Customization source for this folder (typically 'local' or 'user' for writable creation locations). */
239 > readonly source: AICustomizationSource;
240 > }
241 >
242 > /**
243 > * Provider interface for harnesses that use an opt-out sync model.
244 > *
245 > * Every eligible local customization is synced by default; the user
246 > * can disable individual items. The persisted set captures only the
247 > * user's opt-outs.
248 > */
249 > export interface ICustomizationSyncProvider {
250 > readonly onDidChange: Event<void>;
251 > isDisabled(uri: URI): boolean;
252 > setDisabled(uri: URI, disabled: boolean): void;
253 > }
254 >
255 > /**
256 > * Service that manages the active customization harness and provides
257 > * per-type storage source filters based on the selected harness.
258 > *
259 > * The default (core) registration exposes a single "VS Code" harness
260 > * that shows all storage sources. The sessions window overrides this
261 > * to provide CLI-scoped harnesses.
262 > */
263 > export interface ICustomizationHarnessService {
264 > readonly _serviceBrand: undefined;
265 >
266 > /**
267 > * The currently active chat session resource.
268 > */
269 > readonly activeSessionResource: IObservable<URI>;
270 >
271 > /**
272 > * The currently active harness.
273 > */
274 > readonly activeHarness: IObservable<string>;
275 >
276 > /**
277 > * All harnesses available in this window.
278 > * When only one harness is available the UI should hide the toggle.
279 > */
280 > readonly availableHarnesses: IObservable<readonly IHarnessDescriptor[]>;
281 >
282 > /**
283 > * Finds the descriptor of the harness with the given id, or `undefined` if no such harness exists.
284 > * @param sessionType The harness id (sessionType)
285 > */
286 > findHarnessById(sessionType: string): IHarnessDescriptor | undefined;
287 >
288 > /**
289 > * Changes the active session. The new session's type must be present in
290 > * `availableHarnesses`.
291 > */
292 > setActiveSession(sessionResource: URI): void;
293 >
294 > /**
295 > * Returns the descriptor of the currently active harness.
296 > */
297 > getActiveDescriptor(): IHarnessDescriptor;
298 >
299 > /**
300 > * Registers an external harness contributed by an extension.
301 > * The harness appears in the UI toggle alongside static harnesses.
302 > * Returns a disposable that removes the harness when disposed.
303 > */
304 > registerExternalHarness(descriptor: IHarnessDescriptor): IDisposable;
305 >
306 >
307 > /**
308 > * Fires when one of the provided slash commands changes.
309 > */
310 > readonly onDidChangeSlashCommands: Event<{ readonly sessionType: string }>;
311 >
312 > /**
313 > * Fires when one of the provided custom agents changes.
314 > */
315 > readonly onDidChangeCustomAgents: Event<{ readonly sessionType: string }>;
316 >
317 > /**
318 > * Returns the prompt and skill slash commands for the given session type.
319 > * Provider-backed harnesses contribute their own items directly; the default
320 > * VS Code harness falls back to the core prompts service.
321 > *
322 > * @param sessionResource URI of the chat session whose customizations
323 > * should be considered. Forwarded to the underlying
324 > * {@link ICustomizationItemProvider.provideChatSessionCustomizations}.
325 > */
326 > getSlashCommands(sessionResource: URI, token: CancellationToken): Promise<readonly IChatPromptSlashCommand[]>;
327 >
328 > /**
329 > * Returns the custom agents for the given session type.
330 > * Provider-backed harnesses select items via their own provider and resolve
331 > * details via the core prompts service.
332 > *
333 > * @param sessionResource URI of the chat session whose customizations
334 > * should be considered. Forwarded to the underlying
335 > * {@link ICustomizationItemProvider.provideChatSessionCustomizations}.
336 > */
337 > getCustomAgents(sessionResource: URI, token: CancellationToken): Promise<readonly ICustomAgent[]>;
338 >
339 > /**
340 > * Resolves a slash command to its full metadata, including the parsed prompt file for prompt commands.
341 > * Provider-backed harnesses resolve their own items directly; the default VS Code harness falls back to the core prompts service.
342 > *
343 > * @param sessionResource URI of the chat session whose customizations
344 > * should be considered when looking up the slash command.
345 > */
346 > resolvePromptSlashCommand(name: string, sessionResource: URI, token: CancellationToken): Promise<IResolvedChatPromptSlashCommand | undefined>;
347 >
348 > /**
349 > * Returns the best session resource to use for a harness lookup.
350 > * Implementations should prefer the most recently used session for the
351 > * given session type and fall back to an untitled session resource.
352 > */
353 > getSessionResourceForHarness(sessionType: string): URI;
354 > }
355 >
356 > /**
357 > * Minimal slash-command metadata resolved from the active harness.
358 > */
359 > export interface ICustomizationSlashCommand {
360 > readonly uri: URI;
361 > readonly type: PromptsType.prompt | PromptsType.skill;
362 > readonly name: string;
363 > readonly description?: string;
364 > readonly userInvocable: boolean;
365 > readonly sessionTypes?: readonly string[];
366 > }
367 >
368 > // #region Shared descriptor constants
369 >
370 > /**
371 > * Empty descriptor returned when no harness is registered yet.
372 > */
373 > const EMPTY_DESCRIPTOR: IHarnessDescriptor = {
374 > id: '',
375 > label: '',
376 > icon: Codicon.sparkle,
377 > };
378 >
379 >
380 > // #endregion
381 >
382 > // #region Harness descriptor factories
383 >
384 > /**
385 > * Builds the full source list from the base set (local, user, plugin)
386 > * plus any additional sources specific to the window type.
387 > *
388 > * Core passes `[PromptsStorage.extension]`; sessions passes its
389 > * BUILTIN_STORAGE constant.
390 > */
391 >
392 > /**
393 > * Creates a "VS Code" harness descriptor that shows all storage sources
394 > * with no user-root restrictions.
395 > */
396 > export function createVSCodeHarnessDescriptor(): IHarnessDescriptor {
397 return {
398 id: SessionType.Local,
408 };
409 }
411 > // #endregion
412 >
413 > // #region Base implementation
414 >
415 > /**
416 > * Reusable base implementation of {@link ICustomizationHarnessService}.
417 > * Concrete registrations only need to supply the list of harness
418 > * descriptors and a default harness id.
419 > */
420 > export class CustomizationHarnessServiceBase implements ICustomizationHarnessService {
421 > declare readonly _serviceBrand: undefined;
422 > private readonly _onDidChangeSlashCommands = new Emitter<{ readonly sessionType: string }>();
423 > readonly onDidChangeSlashCommands = this._onDidChangeSlashCommands.event;
424 > private readonly _onDidChangeCustomAgents = new Emitter<{ readonly sessionType: string }>();
425 > readonly onDidChangeCustomAgents = this._onDidChangeCustomAgents.event;
426 > private readonly _providerListeners: IDisposable[] = [];
427 > private _isDisposed = false;
428 >
429 > private readonly _activeSessionResource: ISettableObservable<URI>;
430 > readonly activeSessionResource: IObservable<URI>;
431 >
432 > private readonly _activeHarness: IObservable<string>;
433 > readonly activeHarness: IObservable<string>;
434 >
435 > private readonly _staticHarnesses: readonly IHarnessDescriptor[];
436 > private readonly _externalHarnesses: IHarnessDescriptor[] = [];
437 > private readonly _availableHarnesses: ISettableObservable<readonly IHarnessDescriptor[]>;
438 > readonly availableHarnesses: IObservable<readonly IHarnessDescriptor[]>;
439 >
440 > constructor(
441 staticHarnesses: readonly IHarnessDescriptor[],
442 defaultHarness: string,
453 this._rebindProviderListeners();
454 }
456 > private _getAllHarnesses(): readonly IHarnessDescriptor[] {
457 // External harnesses shadow static ones with the same id so that
458 // extension-contributed harnesses can upgrade a built-in entry.
471 this._rebindProviderListeners();
472 }
474 > private _rebindProviderListeners(): void {
475 for (const listener of this._providerListeners) {
476 listener.dispose();
488 }
489 }
491 > dispose(): void {
492 this._isDisposed = true;
493 for (const listener of this._providerListeners) {
498 this._onDidChangeCustomAgents.dispose();
499 }
501 > registerExternalHarness(descriptor: IHarnessDescriptor): IDisposable {
502 this._externalHarnesses.push(descriptor);
503 this._refreshAvailableHarnesses();
515 };
516 }
518 > findHarnessById(id: string): IHarnessDescriptor | undefined {
519 return this._getAllHarnesses().find(h => h.id === id);
520 }
522 > setActiveSession(sessionResource: URI): void {
523 this._activeSessionResource.set(sessionResource, undefined);
524 }
526 > getActiveDescriptor(): IHarnessDescriptor {
527 const activeId = this._activeHarness.get();
528 const descriptor = this.findHarnessById(activeId);
529 return descriptor ?? EMPTY_DESCRIPTOR;
530 }
532 > async getSlashCommands(sessionResource: URI, token: CancellationToken): Promise<readonly IChatPromptSlashCommand[]> {
533 const sessionType = getChatSessionType(sessionResource);
534 const harness = this.findHarnessById(sessionType);
565 return result;
566 }
568 > async getCustomAgents(sessionResource: URI, token: CancellationToken): Promise<readonly ICustomAgent[]> {
569 const sessionType = getChatSessionType(sessionResource);
570 const harness = this.findHarnessById(sessionType);
613 return result;
614 }
616 > public async resolvePromptSlashCommand(name: string, sessionResource: URI, token: CancellationToken): Promise<IResolvedChatPromptSlashCommand | undefined> {
617 const commands = await this.getSlashCommands(sessionResource, token);
618 const command = commands.find(cmd => cmd.name === name);
627 return undefined;
628 }
630 > getSessionResourceForHarness(sessionType: string): URI {
631 if (sessionType === SessionType.Local) {
632 return LocalChatSessionUri.getNewSessionUri();
src/vs/base/common/charCode.ts 450 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- charCode.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // Names from https://blog.codinghorror.com/ascii-pronunciation-rules-for-programmers/
7 >
8 > /**
9 > * An inlined enum containing useful character codes (to be used with String.charCodeAt).
10 > * Please leave the const keyword such that it gets inlined when compiled to JavaScript!
11 > */
12 > export const enum CharCode {
13 > Null = 0,
14 > /**
15 > * The `\b` character.
16 > */
17 > Backspace = 8,
18 > /**
19 > * The `\t` character.
20 > */
21 > Tab = 9,
22 > /**
23 > * The `\n` character.
24 > */
25 > LineFeed = 10,
26 > /**
27 > * The `\r` character.
28 > */
29 > CarriageReturn = 13,
30 > Space = 32,
31 > /**
32 > * The `!` character.
33 > */
34 > ExclamationMark = 33,
35 > /**
36 > * The `"` character.
37 > */
38 > DoubleQuote = 34,
39 > /**
40 > * The `#` character.
41 > */
42 > Hash = 35,
43 > /**
44 > * The `$` character.
45 > */
46 > DollarSign = 36,
47 > /**
48 > * The `%` character.
49 > */
50 > PercentSign = 37,
51 > /**
52 > * The `&` character.
53 > */
54 > Ampersand = 38,
55 > /**
56 > * The `'` character.
57 > */
58 > SingleQuote = 39,
59 > /**
60 > * The `(` character.
61 > */
62 > OpenParen = 40,
63 > /**
64 > * The `)` character.
65 > */
66 > CloseParen = 41,
67 > /**
68 > * The `*` character.
69 > */
70 > Asterisk = 42,
71 > /**
72 > * The `+` character.
73 > */
74 > Plus = 43,
75 > /**
76 > * The `,` character.
77 > */
78 > Comma = 44,
79 > /**
80 > * The `-` character.
81 > */
82 > Dash = 45,
83 > /**
84 > * The `.` character.
85 > */
86 > Period = 46,
87 > /**
88 > * The `/` character.
89 > */
90 > Slash = 47,
91 >
92 > Digit0 = 48,
93 > Digit1 = 49,
94 > Digit2 = 50,
95 > Digit3 = 51,
96 > Digit4 = 52,
97 > Digit5 = 53,
98 > Digit6 = 54,
99 > Digit7 = 55,
100 > Digit8 = 56,
101 > Digit9 = 57,
102 >
103 > /**
104 > * The `:` character.
105 > */
106 > Colon = 58,
107 > /**
108 > * The `;` character.
109 > */
110 > Semicolon = 59,
111 > /**
112 > * The `<` character.
113 > */
114 > LessThan = 60,
115 > /**
116 > * The `=` character.
117 > */
118 > Equals = 61,
119 > /**
120 > * The `>` character.
121 > */
122 > GreaterThan = 62,
123 > /**
124 > * The `?` character.
125 > */
126 > QuestionMark = 63,
127 > /**
128 > * The `@` character.
129 > */
130 > AtSign = 64,
131 >
132 > A = 65,
133 > B = 66,
134 > C = 67,
135 > D = 68,
136 > E = 69,
137 > F = 70,
138 > G = 71,
139 > H = 72,
140 > I = 73,
141 > J = 74,
142 > K = 75,
143 > L = 76,
144 > M = 77,
145 > N = 78,
146 > O = 79,
147 > P = 80,
148 > Q = 81,
149 > R = 82,
150 > S = 83,
151 > T = 84,
152 > U = 85,
153 > V = 86,
154 > W = 87,
155 > X = 88,
156 > Y = 89,
157 > Z = 90,
158 >
159 > /**
160 > * The `[` character.
161 > */
162 > OpenSquareBracket = 91,
163 > /**
164 > * The `\` character.
165 > */
166 > Backslash = 92,
167 > /**
168 > * The `]` character.
169 > */
170 > CloseSquareBracket = 93,
171 > /**
172 > * The `^` character.
173 > */
174 > Caret = 94,
175 > /**
176 > * The `_` character.
177 > */
178 > Underline = 95,
179 > /**
180 > * The ``(`)`` character.
181 > */
182 > BackTick = 96,
183 >
184 > a = 97,
185 > b = 98,
186 > c = 99,
187 > d = 100,
188 > e = 101,
189 > f = 102,
190 > g = 103,
191 > h = 104,
192 > i = 105,
193 > j = 106,
194 > k = 107,
195 > l = 108,
196 > m = 109,
197 > n = 110,
198 > o = 111,
199 > p = 112,
200 > q = 113,
201 > r = 114,
202 > s = 115,
203 > t = 116,
204 > u = 117,
205 > v = 118,
206 > w = 119,
207 > x = 120,
208 > y = 121,
209 > z = 122,
210 >
211 > /**
212 > * The `{` character.
213 > */
214 > OpenCurlyBrace = 123,
215 > /**
216 > * The `|` character.
217 > */
218 > Pipe = 124,
219 > /**
220 > * The `}` character.
221 > */
222 > CloseCurlyBrace = 125,
223 > /**
224 > * The `~` character.
225 > */
226 > Tilde = 126,
227 >
228 > /**
229 > * The &nbsp; (no-break space) character.
230 > * Unicode Character 'NO-BREAK SPACE' (U+00A0)
231 > */
232 > NoBreakSpace = 160,
233 >
234 > U_Combining_Grave_Accent = 0x0300, // U+0300 Combining Grave Accent
235 > U_Combining_Acute_Accent = 0x0301, // U+0301 Combining Acute Accent
236 > U_Combining_Circumflex_Accent = 0x0302, // U+0302 Combining Circumflex Accent
237 > U_Combining_Tilde = 0x0303, // U+0303 Combining Tilde
238 > U_Combining_Macron = 0x0304, // U+0304 Combining Macron
239 > U_Combining_Overline = 0x0305, // U+0305 Combining Overline
240 > U_Combining_Breve = 0x0306, // U+0306 Combining Breve
241 > U_Combining_Dot_Above = 0x0307, // U+0307 Combining Dot Above
242 > U_Combining_Diaeresis = 0x0308, // U+0308 Combining Diaeresis
243 > U_Combining_Hook_Above = 0x0309, // U+0309 Combining Hook Above
244 > U_Combining_Ring_Above = 0x030A, // U+030A Combining Ring Above
245 > U_Combining_Double_Acute_Accent = 0x030B, // U+030B Combining Double Acute Accent
246 > U_Combining_Caron = 0x030C, // U+030C Combining Caron
247 > U_Combining_Vertical_Line_Above = 0x030D, // U+030D Combining Vertical Line Above
248 > U_Combining_Double_Vertical_Line_Above = 0x030E, // U+030E Combining Double Vertical Line Above
249 > U_Combining_Double_Grave_Accent = 0x030F, // U+030F Combining Double Grave Accent
250 > U_Combining_Candrabindu = 0x0310, // U+0310 Combining Candrabindu
251 > U_Combining_Inverted_Breve = 0x0311, // U+0311 Combining Inverted Breve
252 > U_Combining_Turned_Comma_Above = 0x0312, // U+0312 Combining Turned Comma Above
253 > U_Combining_Comma_Above = 0x0313, // U+0313 Combining Comma Above
254 > U_Combining_Reversed_Comma_Above = 0x0314, // U+0314 Combining Reversed Comma Above
255 > U_Combining_Comma_Above_Right = 0x0315, // U+0315 Combining Comma Above Right
256 > U_Combining_Grave_Accent_Below = 0x0316, // U+0316 Combining Grave Accent Below
257 > U_Combining_Acute_Accent_Below = 0x0317, // U+0317 Combining Acute Accent Below
258 > U_Combining_Left_Tack_Below = 0x0318, // U+0318 Combining Left Tack Below
259 > U_Combining_Right_Tack_Below = 0x0319, // U+0319 Combining Right Tack Below
260 > U_Combining_Left_Angle_Above = 0x031A, // U+031A Combining Left Angle Above
261 > U_Combining_Horn = 0x031B, // U+031B Combining Horn
262 > U_Combining_Left_Half_Ring_Below = 0x031C, // U+031C Combining Left Half Ring Below
263 > U_Combining_Up_Tack_Below = 0x031D, // U+031D Combining Up Tack Below
264 > U_Combining_Down_Tack_Below = 0x031E, // U+031E Combining Down Tack Below
265 > U_Combining_Plus_Sign_Below = 0x031F, // U+031F Combining Plus Sign Below
266 > U_Combining_Minus_Sign_Below = 0x0320, // U+0320 Combining Minus Sign Below
267 > U_Combining_Palatalized_Hook_Below = 0x0321, // U+0321 Combining Palatalized Hook Below
268 > U_Combining_Retroflex_Hook_Below = 0x0322, // U+0322 Combining Retroflex Hook Below
269 > U_Combining_Dot_Below = 0x0323, // U+0323 Combining Dot Below
270 > U_Combining_Diaeresis_Below = 0x0324, // U+0324 Combining Diaeresis Below
271 > U_Combining_Ring_Below = 0x0325, // U+0325 Combining Ring Below
272 > U_Combining_Comma_Below = 0x0326, // U+0326 Combining Comma Below
273 > U_Combining_Cedilla = 0x0327, // U+0327 Combining Cedilla
274 > U_Combining_Ogonek = 0x0328, // U+0328 Combining Ogonek
275 > U_Combining_Vertical_Line_Below = 0x0329, // U+0329 Combining Vertical Line Below
276 > U_Combining_Bridge_Below = 0x032A, // U+032A Combining Bridge Below
277 > U_Combining_Inverted_Double_Arch_Below = 0x032B, // U+032B Combining Inverted Double Arch Below
278 > U_Combining_Caron_Below = 0x032C, // U+032C Combining Caron Below
279 > U_Combining_Circumflex_Accent_Below = 0x032D, // U+032D Combining Circumflex Accent Below
280 > U_Combining_Breve_Below = 0x032E, // U+032E Combining Breve Below
281 > U_Combining_Inverted_Breve_Below = 0x032F, // U+032F Combining Inverted Breve Below
282 > U_Combining_Tilde_Below = 0x0330, // U+0330 Combining Tilde Below
283 > U_Combining_Macron_Below = 0x0331, // U+0331 Combining Macron Below
284 > U_Combining_Low_Line = 0x0332, // U+0332 Combining Low Line
285 > U_Combining_Double_Low_Line = 0x0333, // U+0333 Combining Double Low Line
286 > U_Combining_Tilde_Overlay = 0x0334, // U+0334 Combining Tilde Overlay
287 > U_Combining_Short_Stroke_Overlay = 0x0335, // U+0335 Combining Short Stroke Overlay
288 > U_Combining_Long_Stroke_Overlay = 0x0336, // U+0336 Combining Long Stroke Overlay
289 > U_Combining_Short_Solidus_Overlay = 0x0337, // U+0337 Combining Short Solidus Overlay
290 > U_Combining_Long_Solidus_Overlay = 0x0338, // U+0338 Combining Long Solidus Overlay
291 > U_Combining_Right_Half_Ring_Below = 0x0339, // U+0339 Combining Right Half Ring Below
292 > U_Combining_Inverted_Bridge_Below = 0x033A, // U+033A Combining Inverted Bridge Below
293 > U_Combining_Square_Below = 0x033B, // U+033B Combining Square Below
294 > U_Combining_Seagull_Below = 0x033C, // U+033C Combining Seagull Below
295 > U_Combining_X_Above = 0x033D, // U+033D Combining X Above
296 > U_Combining_Vertical_Tilde = 0x033E, // U+033E Combining Vertical Tilde
297 > U_Combining_Double_Overline = 0x033F, // U+033F Combining Double Overline
298 > U_Combining_Grave_Tone_Mark = 0x0340, // U+0340 Combining Grave Tone Mark
299 > U_Combining_Acute_Tone_Mark = 0x0341, // U+0341 Combining Acute Tone Mark
300 > U_Combining_Greek_Perispomeni = 0x0342, // U+0342 Combining Greek Perispomeni
301 > U_Combining_Greek_Koronis = 0x0343, // U+0343 Combining Greek Koronis
302 > U_Combining_Greek_Dialytika_Tonos = 0x0344, // U+0344 Combining Greek Dialytika Tonos
303 > U_Combining_Greek_Ypogegrammeni = 0x0345, // U+0345 Combining Greek Ypogegrammeni
304 > U_Combining_Bridge_Above = 0x0346, // U+0346 Combining Bridge Above
305 > U_Combining_Equals_Sign_Below = 0x0347, // U+0347 Combining Equals Sign Below
306 > U_Combining_Double_Vertical_Line_Below = 0x0348, // U+0348 Combining Double Vertical Line Below
307 > U_Combining_Left_Angle_Below = 0x0349, // U+0349 Combining Left Angle Below
308 > U_Combining_Not_Tilde_Above = 0x034A, // U+034A Combining Not Tilde Above
309 > U_Combining_Homothetic_Above = 0x034B, // U+034B Combining Homothetic Above
310 > U_Combining_Almost_Equal_To_Above = 0x034C, // U+034C Combining Almost Equal To Above
311 > U_Combining_Left_Right_Arrow_Below = 0x034D, // U+034D Combining Left Right Arrow Below
312 > U_Combining_Upwards_Arrow_Below = 0x034E, // U+034E Combining Upwards Arrow Below
313 > U_Combining_Grapheme_Joiner = 0x034F, // U+034F Combining Grapheme Joiner
314 > U_Combining_Right_Arrowhead_Above = 0x0350, // U+0350 Combining Right Arrowhead Above
315 > U_Combining_Left_Half_Ring_Above = 0x0351, // U+0351 Combining Left Half Ring Above
316 > U_Combining_Fermata = 0x0352, // U+0352 Combining Fermata
317 > U_Combining_X_Below = 0x0353, // U+0353 Combining X Below
318 > U_Combining_Left_Arrowhead_Below = 0x0354, // U+0354 Combining Left Arrowhead Below
319 > U_Combining_Right_Arrowhead_Below = 0x0355, // U+0355 Combining Right Arrowhead Below
320 > U_Combining_Right_Arrowhead_And_Up_Arrowhead_Below = 0x0356, // U+0356 Combining Right Arrowhead And Up Arrowhead Below
321 > U_Combining_Right_Half_Ring_Above = 0x0357, // U+0357 Combining Right Half Ring Above
322 > U_Combining_Dot_Above_Right = 0x0358, // U+0358 Combining Dot Above Right
323 > U_Combining_Asterisk_Below = 0x0359, // U+0359 Combining Asterisk Below
324 > U_Combining_Double_Ring_Below = 0x035A, // U+035A Combining Double Ring Below
325 > U_Combining_Zigzag_Above = 0x035B, // U+035B Combining Zigzag Above
326 > U_Combining_Double_Breve_Below = 0x035C, // U+035C Combining Double Breve Below
327 > U_Combining_Double_Breve = 0x035D, // U+035D Combining Double Breve
328 > U_Combining_Double_Macron = 0x035E, // U+035E Combining Double Macron
329 > U_Combining_Double_Macron_Below = 0x035F, // U+035F Combining Double Macron Below
330 > U_Combining_Double_Tilde = 0x0360, // U+0360 Combining Double Tilde
331 > U_Combining_Double_Inverted_Breve = 0x0361, // U+0361 Combining Double Inverted Breve
332 > U_Combining_Double_Rightwards_Arrow_Below = 0x0362, // U+0362 Combining Double Rightwards Arrow Below
333 > U_Combining_Latin_Small_Letter_A = 0x0363, // U+0363 Combining Latin Small Letter A
334 > U_Combining_Latin_Small_Letter_E = 0x0364, // U+0364 Combining Latin Small Letter E
335 > U_Combining_Latin_Small_Letter_I = 0x0365, // U+0365 Combining Latin Small Letter I
336 > U_Combining_Latin_Small_Letter_O = 0x0366, // U+0366 Combining Latin Small Letter O
337 > U_Combining_Latin_Small_Letter_U = 0x0367, // U+0367 Combining Latin Small Letter U
338 > U_Combining_Latin_Small_Letter_C = 0x0368, // U+0368 Combining Latin Small Letter C
339 > U_Combining_Latin_Small_Letter_D = 0x0369, // U+0369 Combining Latin Small Letter D
340 > U_Combining_Latin_Small_Letter_H = 0x036A, // U+036A Combining Latin Small Letter H
341 > U_Combining_Latin_Small_Letter_M = 0x036B, // U+036B Combining Latin Small Letter M
342 > U_Combining_Latin_Small_Letter_R = 0x036C, // U+036C Combining Latin Small Letter R
343 > U_Combining_Latin_Small_Letter_T = 0x036D, // U+036D Combining Latin Small Letter T
344 > U_Combining_Latin_Small_Letter_V = 0x036E, // U+036E Combining Latin Small Letter V
345 > U_Combining_Latin_Small_Letter_X = 0x036F, // U+036F Combining Latin Small Letter X
346 >
347 > /**
348 > * Unicode Character 'LINE SEPARATOR' (U+2028)
349 > * http://www.fileformat.info/info/unicode/char/2028/index.htm
350 > */
351 > LINE_SEPARATOR = 0x2028,
352 > /**
353 > * Unicode Character 'PARAGRAPH SEPARATOR' (U+2029)
354 > * http://www.fileformat.info/info/unicode/char/2029/index.htm
355 > */
356 > PARAGRAPH_SEPARATOR = 0x2029,
357 > /**
358 > * Unicode Character 'NEXT LINE' (U+0085)
359 > * http://www.fileformat.info/info/unicode/char/0085/index.htm
360 > */
361 > NEXT_LINE = 0x0085,
362 >
363 > // http://www.fileformat.info/info/unicode/category/Sk/list.htm
364 > U_CIRCUMFLEX = 0x005E, // U+005E CIRCUMFLEX
365 > U_GRAVE_ACCENT = 0x0060, // U+0060 GRAVE ACCENT
366 > U_DIAERESIS = 0x00A8, // U+00A8 DIAERESIS
367 > U_MACRON = 0x00AF, // U+00AF MACRON
368 > U_ACUTE_ACCENT = 0x00B4, // U+00B4 ACUTE ACCENT
369 > U_CEDILLA = 0x00B8, // U+00B8 CEDILLA
370 > U_MODIFIER_LETTER_LEFT_ARROWHEAD = 0x02C2, // U+02C2 MODIFIER LETTER LEFT ARROWHEAD
371 > U_MODIFIER_LETTER_RIGHT_ARROWHEAD = 0x02C3, // U+02C3 MODIFIER LETTER RIGHT ARROWHEAD
372 > U_MODIFIER_LETTER_UP_ARROWHEAD = 0x02C4, // U+02C4 MODIFIER LETTER UP ARROWHEAD
373 > U_MODIFIER_LETTER_DOWN_ARROWHEAD = 0x02C5, // U+02C5 MODIFIER LETTER DOWN ARROWHEAD
374 > U_MODIFIER_LETTER_CENTRED_RIGHT_HALF_RING = 0x02D2, // U+02D2 MODIFIER LETTER CENTRED RIGHT HALF RING
375 > U_MODIFIER_LETTER_CENTRED_LEFT_HALF_RING = 0x02D3, // U+02D3 MODIFIER LETTER CENTRED LEFT HALF RING
376 > U_MODIFIER_LETTER_UP_TACK = 0x02D4, // U+02D4 MODIFIER LETTER UP TACK
377 > U_MODIFIER_LETTER_DOWN_TACK = 0x02D5, // U+02D5 MODIFIER LETTER DOWN TACK
378 > U_MODIFIER_LETTER_PLUS_SIGN = 0x02D6, // U+02D6 MODIFIER LETTER PLUS SIGN
379 > U_MODIFIER_LETTER_MINUS_SIGN = 0x02D7, // U+02D7 MODIFIER LETTER MINUS SIGN
380 > U_BREVE = 0x02D8, // U+02D8 BREVE
381 > U_DOT_ABOVE = 0x02D9, // U+02D9 DOT ABOVE
382 > U_RING_ABOVE = 0x02DA, // U+02DA RING ABOVE
383 > U_OGONEK = 0x02DB, // U+02DB OGONEK
384 > U_SMALL_TILDE = 0x02DC, // U+02DC SMALL TILDE
385 > U_DOUBLE_ACUTE_ACCENT = 0x02DD, // U+02DD DOUBLE ACUTE ACCENT
386 > U_MODIFIER_LETTER_RHOTIC_HOOK = 0x02DE, // U+02DE MODIFIER LETTER RHOTIC HOOK
387 > U_MODIFIER_LETTER_CROSS_ACCENT = 0x02DF, // U+02DF MODIFIER LETTER CROSS ACCENT
388 > U_MODIFIER_LETTER_EXTRA_HIGH_TONE_BAR = 0x02E5, // U+02E5 MODIFIER LETTER EXTRA-HIGH TONE BAR
389 > U_MODIFIER_LETTER_HIGH_TONE_BAR = 0x02E6, // U+02E6 MODIFIER LETTER HIGH TONE BAR
390 > U_MODIFIER_LETTER_MID_TONE_BAR = 0x02E7, // U+02E7 MODIFIER LETTER MID TONE BAR
391 > U_MODIFIER_LETTER_LOW_TONE_BAR = 0x02E8, // U+02E8 MODIFIER LETTER LOW TONE BAR
392 > U_MODIFIER_LETTER_EXTRA_LOW_TONE_BAR = 0x02E9, // U+02E9 MODIFIER LETTER EXTRA-LOW TONE BAR
393 > U_MODIFIER_LETTER_YIN_DEPARTING_TONE_MARK = 0x02EA, // U+02EA MODIFIER LETTER YIN DEPARTING TONE MARK
394 > U_MODIFIER_LETTER_YANG_DEPARTING_TONE_MARK = 0x02EB, // U+02EB MODIFIER LETTER YANG DEPARTING TONE MARK
395 > U_MODIFIER_LETTER_UNASPIRATED = 0x02ED, // U+02ED MODIFIER LETTER UNASPIRATED
396 > U_MODIFIER_LETTER_LOW_DOWN_ARROWHEAD = 0x02EF, // U+02EF MODIFIER LETTER LOW DOWN ARROWHEAD
397 > U_MODIFIER_LETTER_LOW_UP_ARROWHEAD = 0x02F0, // U+02F0 MODIFIER LETTER LOW UP ARROWHEAD
398 > U_MODIFIER_LETTER_LOW_LEFT_ARROWHEAD = 0x02F1, // U+02F1 MODIFIER LETTER LOW LEFT ARROWHEAD
399 > U_MODIFIER_LETTER_LOW_RIGHT_ARROWHEAD = 0x02F2, // U+02F2 MODIFIER LETTER LOW RIGHT ARROWHEAD
400 > U_MODIFIER_LETTER_LOW_RING = 0x02F3, // U+02F3 MODIFIER LETTER LOW RING
401 > U_MODIFIER_LETTER_MIDDLE_GRAVE_ACCENT = 0x02F4, // U+02F4 MODIFIER LETTER MIDDLE GRAVE ACCENT
402 > U_MODIFIER_LETTER_MIDDLE_DOUBLE_GRAVE_ACCENT = 0x02F5, // U+02F5 MODIFIER LETTER MIDDLE DOUBLE GRAVE ACCENT
403 > U_MODIFIER_LETTER_MIDDLE_DOUBLE_ACUTE_ACCENT = 0x02F6, // U+02F6 MODIFIER LETTER MIDDLE DOUBLE ACUTE ACCENT
404 > U_MODIFIER_LETTER_LOW_TILDE = 0x02F7, // U+02F7 MODIFIER LETTER LOW TILDE
405 > U_MODIFIER_LETTER_RAISED_COLON = 0x02F8, // U+02F8 MODIFIER LETTER RAISED COLON
406 > U_MODIFIER_LETTER_BEGIN_HIGH_TONE = 0x02F9, // U+02F9 MODIFIER LETTER BEGIN HIGH TONE
407 > U_MODIFIER_LETTER_END_HIGH_TONE = 0x02FA, // U+02FA MODIFIER LETTER END HIGH TONE
408 > U_MODIFIER_LETTER_BEGIN_LOW_TONE = 0x02FB, // U+02FB MODIFIER LETTER BEGIN LOW TONE
409 > U_MODIFIER_LETTER_END_LOW_TONE = 0x02FC, // U+02FC MODIFIER LETTER END LOW TONE
410 > U_MODIFIER_LETTER_SHELF = 0x02FD, // U+02FD MODIFIER LETTER SHELF
411 > U_MODIFIER_LETTER_OPEN_SHELF = 0x02FE, // U+02FE MODIFIER LETTER OPEN SHELF
412 > U_MODIFIER_LETTER_LOW_LEFT_ARROW = 0x02FF, // U+02FF MODIFIER LETTER LOW LEFT ARROW
413 > U_GREEK_LOWER_NUMERAL_SIGN = 0x0375, // U+0375 GREEK LOWER NUMERAL SIGN
414 > U_GREEK_TONOS = 0x0384, // U+0384 GREEK TONOS
415 > U_GREEK_DIALYTIKA_TONOS = 0x0385, // U+0385 GREEK DIALYTIKA TONOS
416 > U_GREEK_KORONIS = 0x1FBD, // U+1FBD GREEK KORONIS
417 > U_GREEK_PSILI = 0x1FBF, // U+1FBF GREEK PSILI
418 > U_GREEK_PERISPOMENI = 0x1FC0, // U+1FC0 GREEK PERISPOMENI
419 > U_GREEK_DIALYTIKA_AND_PERISPOMENI = 0x1FC1, // U+1FC1 GREEK DIALYTIKA AND PERISPOMENI
420 > U_GREEK_PSILI_AND_VARIA = 0x1FCD, // U+1FCD GREEK PSILI AND VARIA
421 > U_GREEK_PSILI_AND_OXIA = 0x1FCE, // U+1FCE GREEK PSILI AND OXIA
422 > U_GREEK_PSILI_AND_PERISPOMENI = 0x1FCF, // U+1FCF GREEK PSILI AND PERISPOMENI
423 > U_GREEK_DASIA_AND_VARIA = 0x1FDD, // U+1FDD GREEK DASIA AND VARIA
424 > U_GREEK_DASIA_AND_OXIA = 0x1FDE, // U+1FDE GREEK DASIA AND OXIA
425 > U_GREEK_DASIA_AND_PERISPOMENI = 0x1FDF, // U+1FDF GREEK DASIA AND PERISPOMENI
426 > U_GREEK_DIALYTIKA_AND_VARIA = 0x1FED, // U+1FED GREEK DIALYTIKA AND VARIA
427 > U_GREEK_DIALYTIKA_AND_OXIA = 0x1FEE, // U+1FEE GREEK DIALYTIKA AND OXIA
428 > U_GREEK_VARIA = 0x1FEF, // U+1FEF GREEK VARIA
429 > U_GREEK_OXIA = 0x1FFD, // U+1FFD GREEK OXIA
430 > U_GREEK_DASIA = 0x1FFE, // U+1FFE GREEK DASIA
431 >
432 > U_IDEOGRAPHIC_FULL_STOP = 0x3002, // U+3002 IDEOGRAPHIC FULL STOP
433 > U_LEFT_CORNER_BRACKET = 0x300C, // U+300C LEFT CORNER BRACKET
434 > U_RIGHT_CORNER_BRACKET = 0x300D, // U+300D RIGHT CORNER BRACKET
435 > U_LEFT_BLACK_LENTICULAR_BRACKET = 0x3010, // U+3010 LEFT BLACK LENTICULAR BRACKET
436 > U_RIGHT_BLACK_LENTICULAR_BRACKET = 0x3011, // U+3011 RIGHT BLACK LENTICULAR BRACKET
437 >
438 >
439 > U_OVERLINE = 0x203E, // Unicode Character 'OVERLINE'
440 >
441 > /**
442 > * UTF-8 BOM
443 > * Unicode Character 'ZERO WIDTH NO-BREAK SPACE' (U+FEFF)
444 > * http://www.fileformat.info/info/unicode/char/feff/index.htm
445 > */
446 > UTF8_BOM = 65279,
447 >
448 > U_FULLWIDTH_SEMICOLON = 0xFF1B, // U+FF1B FULLWIDTH SEMICOLON
449 > U_FULLWIDTH_COMMA = 0xFF0C, // U+FF0C FULLWIDTH COMMA
450 > }
src/vs/platform/log/common/log.ts 449 covered LOC · 74 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- log.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as nls from '../../../nls.js';
7 > import { toErrorMessage } from '../../../base/common/errorMessage.js';
8 > import { Emitter, Event } from '../../../base/common/event.js';
9 > import { hash } from '../../../base/common/hash.js';
10 > import { Disposable, IDisposable, toDisposable } from '../../../base/common/lifecycle.js';
11 > import { ResourceMap } from '../../../base/common/map.js';
12 > import { isWindows } from '../../../base/common/platform.js';
13 > import { joinPath } from '../../../base/common/resources.js';
14 > import { Mutable, isNumber, isString } from '../../../base/common/types.js';
15 > import { URI } from '../../../base/common/uri.js';
16 > import { ILocalizedString } from '../../action/common/action.js';
17 > import { RawContextKey } from '../../contextkey/common/contextkey.js';
18 > import { IEnvironmentService } from '../../environment/common/environment.js';
19 > import { createDecorator } from '../../instantiation/common/instantiation.js';
20 >
21 > export const ILogService = createDecorator<ILogService>('logService');
22 > export const ILoggerService = createDecorator<ILoggerService>('loggerService');
23 >
24 function now(): string {
25 return new Date().toISOString();
26 }
27 > log.ts
28 > export function isLogLevel(thing: unknown): thing is LogLevel {
29 return isNumber(thing);
30 }
31 > log.ts
32 > export enum LogLevel {
33 > Off,
34 > Trace,
35 > Debug,
36 > Info,
37 > Warning,
38 > Error
39 > }
40 >
41 > export const DEFAULT_LOG_LEVEL: LogLevel = LogLevel.Info;
42 >
43 > export interface ILogger extends IDisposable {
44 > readonly onDidChangeLogLevel: Event<LogLevel>;
45 > getLevel(): LogLevel;
46 > setLevel(level: LogLevel): void;
47 >
48 > trace(message: string, ...args: unknown[]): void;
49 > debug(message: string, ...args: unknown[]): void;
50 > info(message: string, ...args: unknown[]): void;
51 > warn(message: string, ...args: unknown[]): void;
52 > error(message: string | Error, ...args: unknown[]): void;
53 >
54 > /**
55 > * An operation to flush the contents. Can be synchronous.
56 > */
57 > flush(): void;
58 > }
59 >
60 > export function canLog(loggerLevel: LogLevel, messageLevel: LogLevel): boolean {
61 > return loggerLevel !== LogLevel.Off && loggerLevel <= messageLevel; log.ts
62 > }
63 > log.ts
64 > export function log(logger: ILogger, level: LogLevel, message: string): void {
65 switch (level) {
66 case LogLevel.Trace: logger.trace(message); break;
73 }
74 }
75 > log.ts
76 > type ConsoleMethod = 'debug' | 'error' | 'info' | 'log' | 'warn';
77 > type ConsoleMethodFn = (...args: unknown[]) => void;
78 >
79 > /**
80 > * Flag to enable forwarding of console.* calls to the log service in development.
81 > * This is intended for the use of agents to quickly instrument the code with console.logs
82 > * which will end up in the log service's file outputs.
83 > */
84 > export const isDevConsoleLogForwardingEnabled = false
85 > // || Boolean("true") // done "weirdly" so that a lint warning prevents you from pushing this
86 > ;
87 >
88 > let isConsoleForwarding = false;
89 > let isLogServiceConsoleEcho = false;
90 >
91 function getConsoleMethod(method: ConsoleMethod): ConsoleMethodFn {
92 switch (method) {
98 }
99 }
100 > log.ts
101 function setConsoleMethod(method: ConsoleMethod, fn: ConsoleMethodFn): void {
102 switch (method) {
108 }
109 }
110 > log.ts
111 function logToConsole(method: ConsoleMethod, ...args: unknown[]): void {
112 if (isConsoleForwarding) {
120 }
121 }
122 > log.ts
123 > export function registerDevConsoleLogForwarder(logService: ILogService): IDisposable {
124 const originalConsoleMethods: Record<ConsoleMethod, ConsoleMethodFn> = {
125 debug: console.debug,
177 });
178 }
179 > log.ts
180 > export function format(args: any, verbose: boolean = false): string {
181 let result = '';
182
199 return result;
200 }
201 > log.ts
202 > export type LoggerGroup = {
203 > readonly id: string;
204 > readonly name: string;
205 > };
206 >
207 > export interface ILogService extends ILogger {
208 > readonly _serviceBrand: undefined;
209 > }
210 >
211 > export interface ILoggerOptions {
212 >
213 > /**
214 > * Id of the logger.
215 > */
216 > id?: string;
217 >
218 > /**
219 > * Name of the logger.
220 > */
221 > name?: string;
222 >
223 > /**
224 > * Do not create rotating files if max size exceeds.
225 > */
226 > donotRotate?: boolean;
227 >
228 > /**
229 > * Do not use formatters.
230 > */
231 > donotUseFormatters?: boolean;
232 >
233 > /**
234 > * When to log. Set to `always` to log always.
235 > */
236 > logLevel?: 'always' | LogLevel;
237 >
238 > /**
239 > * Whether the log should be hidden from the user.
240 > */
241 > hidden?: boolean;
242 >
243 > /**
244 > * Condition which must be true to show this logger
245 > */
246 > when?: string;
247 >
248 > /**
249 > * Id of the extension that created this logger.
250 > */
251 > extensionId?: string;
252 >
253 > /**
254 > * Group of the logger.
255 > */
256 > group?: LoggerGroup;
257 > }
258 >
259 > export interface ILoggerResource {
260 > readonly resource: URI;
261 > readonly id: string;
262 > readonly name?: string;
263 > readonly logLevel?: LogLevel;
264 > readonly hidden?: boolean;
265 > readonly when?: string;
266 > readonly extensionId?: string;
267 > readonly group?: LoggerGroup;
268 > }
269 >
270 > export type DidChangeLoggersEvent = {
271 > readonly added: Iterable<ILoggerResource>;
272 > readonly removed: Iterable<ILoggerResource>;
273 > };
274 >
275 > export interface ILoggerService {
276 >
277 > readonly _serviceBrand: undefined;
278 >
279 > /**
280 > * Creates a logger for the given resource, or gets one if it already exists.
281 > *
282 > * This will also register the logger with the logger service.
283 > */
284 > createLogger(resource: URI, options?: ILoggerOptions): ILogger;
285 >
286 > /**
287 > * Creates a logger with the given id in the logs folder, or gets one if it already exists.
288 > *
289 > * This will also register the logger with the logger service.
290 > */
291 > createLogger(id: string, options?: Omit<ILoggerOptions, 'id'>): ILogger;
292 >
293 > /**
294 > * Gets an existing logger, if any.
295 > */
296 > getLogger(resourceOrId: URI | string): ILogger | undefined;
297 >
298 > /**
299 > * An event which fires when the log level of a logger has changed
300 > */
301 > readonly onDidChangeLogLevel: Event<LogLevel | [URI, LogLevel]>;
302 >
303 > /**
304 > * Set default log level.
305 > */
306 > setLogLevel(level: LogLevel): void;
307 >
308 > /**
309 > * Set log level for a logger.
310 > */
311 > setLogLevel(resource: URI, level: LogLevel): void;
312 >
313 > /**
314 > * Get log level for a logger or the default log level.
315 > */
316 > getLogLevel(resource?: URI): LogLevel;
317 >
318 > /**
319 > * An event which fires when the visibility of a logger has changed
320 > */
321 > readonly onDidChangeVisibility: Event<[URI, boolean]>;
322 >
323 > /**
324 > * Set the visibility of a logger.
325 > */
326 > setVisibility(resourceOrId: URI | string, visible: boolean): void;
327 >
328 > /**
329 > * An event which fires when the logger resources are changed
330 > */
331 > readonly onDidChangeLoggers: Event<DidChangeLoggersEvent>;
332 >
333 > /**
334 > * Register a logger with the logger service.
335 > *
336 > * Note that this will not create a logger, but only register it.
337 > *
338 > * Use `createLogger` to create a logger and register it.
339 > *
340 > * Use it when you want to register a logger that is not created by the logger service.
341 > */
342 > registerLogger(resource: ILoggerResource): void;
343 >
344 > /**
345 > * Deregister the logger for the given resource.
346 > */
347 > deregisterLogger(idOrResource: URI | string): void;
348 >
349 > /**
350 > * Get all registered loggers
351 > */
352 > getRegisteredLoggers(): Iterable<ILoggerResource>;
353 >
354 > /**
355 > * Get the registered logger for the given resource.
356 > */
357 > getRegisteredLogger(resource: URI): ILoggerResource | undefined;
358 > }
359 >
360 > export abstract class AbstractLogger extends Disposable implements ILogger {
361
362 private level: LogLevel = DEFAULT_LOG_LEVEL;
363 private readonly _onDidChangeLogLevel: Emitter<LogLevel> = this._register(new Emitter<LogLevel>());
364 > get onDidChangeLogLevel(): Event<LogLevel> { return this._onDidChangeLogLevel.event; } log.ts
365 >
366 > setLevel(level: LogLevel): void {
367 if (this.level !== level) {
368 this.level = level;
370 }
371 }
372 > log.ts
373 > getLevel(): LogLevel {
374 return this.level;
375 }
376 > log.ts
377 > protected checkLogLevel(level: LogLevel): boolean {
378 return canLog(this.level, level);
379 }
380 > log.ts
381 > protected canLog(level: LogLevel): boolean {
382 if (this._store.isDisposed) {
383 return false;
385 return this.checkLogLevel(level);
386 }
387 > log.ts
388 > abstract trace(message: string, ...args: unknown[]): void;
389 > abstract debug(message: string, ...args: unknown[]): void;
390 > abstract info(message: string, ...args: unknown[]): void;
391 > abstract warn(message: string, ...args: unknown[]): void;
392 > abstract error(message: string | Error, ...args: unknown[]): void;
393 > abstract flush(): void;
394 > }
395 >
396 > export abstract class AbstractMessageLogger extends AbstractLogger implements ILogger {
397 >
398 > constructor(private readonly logAlways?: boolean) {
399 super();
400 }
401 > log.ts
402 > protected override checkLogLevel(level: LogLevel): boolean {
403 return this.logAlways || super.checkLogLevel(level);
404 }
405 > log.ts
406 > trace(message: string, ...args: unknown[]): void {
407 if (this.canLog(LogLevel.Trace)) {
408 this.log(LogLevel.Trace, format([message, ...args], true));
409 }
410 }
411 > log.ts
412 > debug(message: string, ...args: unknown[]): void {
413 if (this.canLog(LogLevel.Debug)) {
414 this.log(LogLevel.Debug, format([message, ...args]));
415 }
416 }
417 > log.ts
418 > info(message: string, ...args: unknown[]): void {
419 if (this.canLog(LogLevel.Info)) {
420 this.log(LogLevel.Info, format([message, ...args]));
421 }
422 }
423 > log.ts
424 > warn(message: string, ...args: unknown[]): void {
425 if (this.canLog(LogLevel.Warning)) {
426 this.log(LogLevel.Warning, format([message, ...args]));
427 }
428 }
429 > log.ts
430 > error(message: string | Error, ...args: unknown[]): void {
431 if (this.canLog(LogLevel.Error)) {
432 if (message instanceof Error) {
439 }
440 }
441 > log.ts
442 > flush(): void { }
443 >
444 > protected abstract log(level: LogLevel, message: string): void;
445 > }
446 >
447 >
448 > export class ConsoleMainLogger extends AbstractLogger implements ILogger {
449 >
450 > private useColors: boolean;
451 >
452 > constructor(logLevel: LogLevel = DEFAULT_LOG_LEVEL) {
453 super();
454 this.setLevel(logLevel);
455 this.useColors = !isWindows;
456 }
457 > log.ts
458 > trace(message: string, ...args: unknown[]): void {
459 if (this.canLog(LogLevel.Trace)) {
460 if (this.useColors) {
465 }
466 }
467 > log.ts
468 > debug(message: string, ...args: unknown[]): void {
469 if (this.canLog(LogLevel.Debug)) {
470 if (this.useColors) {
475 }
476 }
477 > log.ts
478 > info(message: string, ...args: unknown[]): void {
479 if (this.canLog(LogLevel.Info)) {
480 if (this.useColors) {
485 }
486 }
487 > log.ts
488 > warn(message: string | Error, ...args: unknown[]): void {
489 if (this.canLog(LogLevel.Warning)) {
490 if (this.useColors) {
495 }
496 }
497 > log.ts
498 > error(message: string, ...args: unknown[]): void {
499 if (this.canLog(LogLevel.Error)) {
500 if (this.useColors) {
505 }
506 }
507 > log.ts
508 > flush(): void {
509 // noop
510 }
511 > log.ts
512 > }
513 >
514 > export class ConsoleLogger extends AbstractLogger implements ILogger {
515 >
516 > constructor(logLevel: LogLevel = DEFAULT_LOG_LEVEL, private readonly useColors: boolean = true) {
517 super();
518 this.setLevel(logLevel);
519 }
520 > log.ts
521 > trace(message: string, ...args: unknown[]): void {
522 if (this.canLog(LogLevel.Trace)) {
523 if (this.useColors) {
528 }
529 }
530 > log.ts
531 > debug(message: string, ...args: unknown[]): void {
532 if (this.canLog(LogLevel.Debug)) {
533 if (this.useColors) {
538 }
539 }
540 > log.ts
541 > info(message: string, ...args: unknown[]): void {
542 if (this.canLog(LogLevel.Info)) {
543 if (this.useColors) {
548 }
549 }
550 > log.ts
551 > warn(message: string | Error, ...args: unknown[]): void {
552 if (this.canLog(LogLevel.Warning)) {
553 if (this.useColors) {
558 }
559 }
560 > log.ts
561 > error(message: string, ...args: unknown[]): void {
562 if (this.canLog(LogLevel.Error)) {
563 if (this.useColors) {
568 }
569 }
570 > log.ts
571 >
572 > flush(): void {
573 // noop
574 }
575 > } log.ts
576 >
577 > export class AdapterLogger extends AbstractLogger implements ILogger {
578 >
579 > constructor(private readonly adapter: { log: (logLevel: LogLevel, args: any[]) => void }, logLevel: LogLevel = DEFAULT_LOG_LEVEL) {
580 super();
581 this.setLevel(logLevel);
582 }
583 > log.ts
584 > trace(message: string, ...args: unknown[]): void {
585 if (this.canLog(LogLevel.Trace)) {
586 this.adapter.log(LogLevel.Trace, [this.extractMessage(message), ...args]);
587 }
588 }
589 > log.ts
590 > debug(message: string, ...args: unknown[]): void {
591 if (this.canLog(LogLevel.Debug)) {
592 this.adapter.log(LogLevel.Debug, [this.extractMessage(message), ...args]);
593 }
594 }
595 > log.ts
596 > info(message: string, ...args: unknown[]): void {
597 if (this.canLog(LogLevel.Info)) {
598 this.adapter.log(LogLevel.Info, [this.extractMessage(message), ...args]);
599 }
600 }
601 > log.ts
602 > warn(message: string | Error, ...args: unknown[]): void {
603 if (this.canLog(LogLevel.Warning)) {
604 this.adapter.log(LogLevel.Warning, [this.extractMessage(message), ...args]);
605 }
606 }
607 > log.ts
608 > error(message: string | Error, ...args: unknown[]): void {
609 if (this.canLog(LogLevel.Error)) {
610 this.adapter.log(LogLevel.Error, [this.extractMessage(message), ...args]);
611 }
612 }
613 > log.ts
614 > private extractMessage(msg: string | Error): string {
615 if (typeof msg === 'string') {
616 return msg;
619 return toErrorMessage(msg, this.canLog(LogLevel.Trace));
620 }
621 > log.ts
622 > flush(): void {
623 // noop
624 }
625 > } log.ts
626 >
627 > export class MultiplexLogger extends AbstractLogger implements ILogger {
628 >
629 > constructor(private readonly loggers: ReadonlyArray<ILogger>) {
630 super();
631 if (loggers.length) {
633 }
634 }
635 > log.ts
636 > override setLevel(level: LogLevel): void {
637 for (const logger of this.loggers) {
638 logger.setLevel(level);
640 super.setLevel(level);
641 }
642 > log.ts
643 > trace(message: string, ...args: unknown[]): void {
644 for (const logger of this.loggers) {
645 logger.trace(message, ...args);
646 }
647 }
648 > log.ts
649 > debug(message: string, ...args: unknown[]): void {
650 for (const logger of this.loggers) {
651 logger.debug(message, ...args);
652 }
653 }
654 > log.ts
655 > info(message: string, ...args: unknown[]): void {
656 for (const logger of this.loggers) {
657 logger.info(message, ...args);
658 }
659 }
660 > log.ts
661 > warn(message: string, ...args: unknown[]): void {
662 for (const logger of this.loggers) {
663 logger.warn(message, ...args);
664 }
665 }
666 > log.ts
667 > error(message: string | Error, ...args: unknown[]): void {
668 for (const logger of this.loggers) {
669 logger.error(message, ...args);
670 }
671 }
672 > log.ts
673 > flush(): void {
674 for (const logger of this.loggers) {
675 logger.flush();
676 }
677 }
678 > log.ts
679 > override dispose(): void {
680 for (const logger of this.loggers) {
681 logger.dispose();
683 super.dispose();
684 }
685 > } log.ts
686 >
687 > type LoggerEntry = { logger: ILogger | undefined; info: Mutable<ILoggerResource> };
688 >
689 > export abstract class AbstractLoggerService extends Disposable implements ILoggerService {
690 >
691 > declare readonly _serviceBrand: undefined;
692 >
693 > private readonly _loggers = new ResourceMap<LoggerEntry>();
694 >
695 > private _onDidChangeLoggers = this._register(new Emitter<{ added: ILoggerResource[]; removed: ILoggerResource[] }>);
696 > readonly onDidChangeLoggers = this._onDidChangeLoggers.event;
697 >
698 > private _onDidChangeLogLevel = this._register(new Emitter<LogLevel | [URI, LogLevel]>);
699 > readonly onDidChangeLogLevel = this._onDidChangeLogLevel.event;
700 >
701 > private _onDidChangeVisibility = this._register(new Emitter<[URI, boolean]>);
702 > readonly onDidChangeVisibility = this._onDidChangeVisibility.event;
703 >
704 > constructor(
705 protected logLevel: LogLevel,
706 private readonly logsHome: URI,
714 }
715 }
716 > log.ts
717 > private getLoggerEntry(resourceOrId: URI | string): LoggerEntry | undefined {
718 if (isString(resourceOrId)) {
719 return [...this._loggers.values()].find(logger => logger.info.id === resourceOrId);
721 return this._loggers.get(resourceOrId);
722 }
723 > log.ts
724 > getLogger(resourceOrId: URI | string): ILogger | undefined {
725 return this.getLoggerEntry(resourceOrId)?.logger;
726 }
727 > log.ts
728 > createLogger(idOrResource: URI | string, options?: ILoggerOptions): ILogger {
729 const resource = this.toResource(idOrResource);
730 const id = isString(idOrResource) ? idOrResource : (options?.id ?? hash(resource.toString()).toString(16));
752 return logger;
753 }
754 > log.ts
755 > protected toResource(idOrResource: string | URI): URI {
756 return isString(idOrResource) ? joinPath(this.logsHome, `${idOrResource.replace(/[\\/:\*\?"<>\|]/g, '')}.log`) : idOrResource;
757 }
758 > log.ts
759 > setLogLevel(logLevel: LogLevel): void;
760 > setLogLevel(resource: URI, logLevel: LogLevel): void;
761 > setLogLevel(arg1: any, arg2?: any): void {
762 if (URI.isUri(arg1)) {
763 const resource = arg1;
780 }
781 }
782 > log.ts
783 > setVisibility(resourceOrId: URI | string, visibility: boolean): void {
784 const logger = this.getLoggerEntry(resourceOrId);
785 if (logger && visibility !== !logger.info.hidden) {
789 }
790 }
791 > log.ts
792 > getLogLevel(resource?: URI): LogLevel {
793 let logLevel;
794 if (resource) {
797 return logLevel ?? this.logLevel;
798 }
799 > log.ts
800 > registerLogger(resource: ILoggerResource): void {
801 const existing = this._loggers.get(resource.resource);
802 if (existing) {
809 }
810 }
811 > log.ts
812 > deregisterLogger(idOrResource: URI | string): void {
813 const resource = this.toResource(idOrResource);
814 const existing = this._loggers.get(resource);
821 }
822 }
823 > log.ts
824 > *getRegisteredLoggers(): Iterable<ILoggerResource> {
825 for (const entry of this._loggers.values()) {
826 yield entry.info;
827 }
828 }
829 > log.ts
830 > getRegisteredLogger(resource: URI): ILoggerResource | undefined {
831 return this._loggers.get(resource)?.info;
832 }
833 > log.ts
834 > override dispose(): void {
835 this._loggers.forEach(logger => logger.logger?.dispose());
836 this._loggers.clear();
837 super.dispose();
838 }
839 > log.ts
840 > protected abstract doCreateLogger(resource: URI, logLevel: LogLevel, options?: ILoggerOptions): ILogger;
841 > }
842 >
843 > export class NullLogger implements ILogger {
844 > readonly onDidChangeLogLevel: Event<LogLevel> = new Emitter<LogLevel>().event; log.ts
845 > setLevel(level: LogLevel): void { } log.ts
846 > getLevel(): LogLevel { return LogLevel.Info; }
847 > trace(message: string, ...args: unknown[]): void { }
848 > debug(message: string, ...args: unknown[]): void { }
849 > info(message: string, ...args: unknown[]): void { }
850 > warn(message: string, ...args: unknown[]): void { }
851 > error(message: string | Error, ...args: unknown[]): void { }
852 > critical(message: string | Error, ...args: unknown[]): void { }
853 > dispose(): void { }
854 > flush(): void { }
855 > }
856 >
857 > export class NullLogService extends NullLogger implements ILogService {
858 > declare readonly _serviceBrand: undefined;
859 > }
860 >
861 > export class NullLoggerService extends AbstractLoggerService {
862 > constructor() {
863 super(LogLevel.Off, URI.parse('log:///log'));
864 }
865 > protected override doCreateLogger(resource: URI, logLevel: LogLevel, options?: ILoggerOptions): ILogger { log.ts
866 return new NullLogger();
867 }
868 > } log.ts
869 >
870 > export function getLogLevel(environmentService: IEnvironmentService): LogLevel {
871 if (environmentService.verbose) {
872 return LogLevel.Trace;
880 return DEFAULT_LOG_LEVEL;
881 }
882 > log.ts
883 > export function LogLevelToString(logLevel: LogLevel): string {
884 > switch (logLevel) {
885 > case LogLevel.Trace: return 'trace';
886 > case LogLevel.Debug: return 'debug';
887 > case LogLevel.Info: return 'info';
888 > case LogLevel.Warning: return 'warn';
889 > case LogLevel.Error: return 'error';
890 > case LogLevel.Off: return 'off';
891 > }
892 > }
893 >
894 > export function LogLevelToLocalizedString(logLevel: LogLevel): ILocalizedString {
895 switch (logLevel) {
896 case LogLevel.Trace: return { original: 'Trace', value: nls.localize('trace', "Trace") };
902 }
903 }
904 > log.ts
905 > export function parseLogLevel(logLevel: string): LogLevel | undefined {
906 switch (logLevel) {
907 case 'trace':
922 return undefined;
923 }
924 > log.ts
925 > // Contexts
926 > export const CONTEXT_LOG_LEVEL = new RawContextKey<string>('logLevel', LogLevelToString(LogLevel.Info));
src/vs/workbench/contrib/chat/common/chatDebugService.ts 435 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatDebugService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { IDisposable } from '../../../../base/common/lifecycle.js';
8 > import { URI } from '../../../../base/common/uri.js';
9 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
10 > import { CancellationToken } from '../../../../base/common/cancellation.js';
11 >
12 > /**
13 > * The severity level of a chat debug log event.
14 > */
15 > export enum ChatDebugLogLevel {
16 > Trace = 0,
17 > Info = 1,
18 > Warning = 2,
19 > Error = 3
20 > }
21 >
22 > /**
23 > * The result of a hook execution.
24 > */
25 > export enum ChatDebugHookResult {
26 > /** The hook executed successfully (exit code 0). */
27 > Success = 0,
28 > /** The hook returned a blocking error (exit code 2). */
29 > Error = 1,
30 > /** The hook returned a non-blocking warning (other non-zero exit codes). */
31 > NonBlockingError = 2
32 > }
33 >
34 > /**
35 > * Common properties shared by all chat debug event types.
36 > */
37 > export interface IChatDebugEventCommon {
38 > readonly id?: string;
39 > readonly sessionResource: URI;
40 > readonly created: Date;
41 > readonly parentEventId?: string;
42 > }
43 >
44 > /**
45 > * A tool call event in the chat debug log.
46 > */
47 > export interface IChatDebugToolCallEvent extends IChatDebugEventCommon {
48 > readonly kind: 'toolCall';
49 > readonly toolName: string;
50 > readonly toolCallId?: string;
51 > readonly input?: string;
52 > readonly output?: string;
53 > readonly result?: 'success' | 'error';
54 > readonly durationInMillis?: number;
55 > }
56 >
57 > /**
58 > * A model turn event representing an LLM request/response.
59 > */
60 > export interface IChatDebugModelTurnEvent extends IChatDebugEventCommon {
61 > readonly kind: 'modelTurn';
62 > readonly model?: string;
63 > readonly requestName?: string;
64 > readonly inputTokens?: number;
65 > readonly outputTokens?: number;
66 > readonly cachedTokens?: number;
67 > readonly totalTokens?: number;
68 > readonly copilotUsageNanoAiu?: number;
69 > readonly durationInMillis?: number;
70 > }
71 >
72 > /**
73 > * A generic log event for unstructured or miscellaneous messages.
74 > */
75 > export interface IChatDebugGenericEvent extends IChatDebugEventCommon {
76 > readonly kind: 'generic';
77 > readonly name: string;
78 > readonly details?: string;
79 > readonly level: ChatDebugLogLevel;
80 > readonly category?: string;
81 > }
82 >
83 > /**
84 > * A subagent invocation event, representing a spawned sub-agent within a session.
85 > */
86 > export interface IChatDebugSubagentInvocationEvent extends IChatDebugEventCommon {
87 > readonly kind: 'subagentInvocation';
88 > readonly agentName: string;
89 > readonly description?: string;
90 > readonly status?: 'running' | 'completed' | 'failed';
91 > readonly durationInMillis?: number;
92 > readonly toolCallCount?: number;
93 > readonly modelTurnCount?: number;
94 > }
95 >
96 > /**
97 > * A named section within a user message or agent response.
98 > */
99 > export interface IChatDebugMessageSection {
100 > readonly name: string;
101 > readonly content: string;
102 > }
103 >
104 > /**
105 > * A user message event, representing the full prompt sent by the user.
106 > */
107 > export interface IChatDebugUserMessageEvent extends IChatDebugEventCommon {
108 > readonly kind: 'userMessage';
109 > readonly message: string;
110 > readonly sections: readonly IChatDebugMessageSection[];
111 > }
112 >
113 > /**
114 > * An agent response event, representing the agent's response.
115 > */
116 > export interface IChatDebugAgentResponseEvent extends IChatDebugEventCommon {
117 > readonly kind: 'agentResponse';
118 > readonly message: string;
119 > readonly sections: readonly IChatDebugMessageSection[];
120 > }
121 >
122 > /**
123 > * Union of all internal chat debug event types.
124 > */
125 > export type IChatDebugEvent = IChatDebugToolCallEvent | IChatDebugModelTurnEvent | IChatDebugGenericEvent | IChatDebugSubagentInvocationEvent | IChatDebugUserMessageEvent | IChatDebugAgentResponseEvent;
126 >
127 > export const IChatDebugService = createDecorator<IChatDebugService>('chatDebugService');
128 >
129 > /**
130 > * Service for collecting and exposing chat debug events.
131 > * Internal components can log events,
132 > * and the debug editor pane can display them.
133 > */
134 > export interface IChatDebugService extends IDisposable {
135 > readonly _serviceBrand: undefined;
136 >
137 > /**
138 > * Fired when a new event is added.
139 > */
140 > readonly onDidAddEvent: Event<IChatDebugEvent>;
141 >
142 > /**
143 > * Fired when provider events are cleared for a session (before re-invoking providers).
144 > */
145 > readonly onDidClearProviderEvents: Event<URI>;
146 >
147 > /**
148 > * Fired when a debug session ends (see {@link endSession}), so providers
149 > * can release any per-session resources (e.g. live file watchers).
150 > */
151 > readonly onDidEndSession: Event<URI>;
152 >
153 > /**
154 > * Log a generic event to the debug service.
155 > */
156 > log(sessionResource: URI, name: string, details?: string, level?: ChatDebugLogLevel, options?: { id?: string; category?: string; parentEventId?: string }): void;
157 >
158 > /**
159 > * Add a typed event to the debug service.
160 > */
161 > addEvent(event: IChatDebugEvent): void;
162 >
163 > /**
164 > * Add an event sourced from an external provider.
165 > * These events are cleared before re-invoking providers to avoid duplicates.
166 > */
167 > addProviderEvent(event: IChatDebugEvent): void;
168 >
169 > /**
170 > * Get all events for a specific session.
171 > */
172 > getEvents(sessionResource?: URI): readonly IChatDebugEvent[];
173 >
174 > /**
175 > * Get all session resources that have logged events.
176 > */
177 > getSessionResources(): readonly URI[];
178 >
179 > /**
180 > * The currently active session resource for debugging.
181 > */
182 > activeSessionResource: URI | undefined;
183 >
184 > /**
185 > * Clear all logged events.
186 > */
187 > clear(): void;
188 >
189 > /**
190 > * Register an external provider that can supply additional debug events.
191 > * This is used by the extension API (ChatDebugLogProvider).
192 > */
193 > registerProvider(provider: IChatDebugLogProvider): IDisposable;
194 >
195 > /**
196 > * Check whether providers have already been invoked for a given session.
197 > */
198 > hasInvokedProviders(sessionResource: URI): boolean;
199 >
200 > /**
201 > * Invoke all registered providers for a given session resource.
202 > * Called when the Debug View is opened to fetch events from extensions.
203 > */
204 > invokeProviders(sessionResource: URI): Promise<void>;
205 >
206 > /**
207 > * End a debug session: cancels any in-flight provider invocation,
208 > * disposes the associated CancellationTokenSource, and removes it.
209 > * Called when the chat session is disposed/archived.
210 > */
211 > endSession(sessionResource: URI): void;
212 >
213 > /**
214 > * Resolve the full details of an event by its id.
215 > * Delegates to the registered provider's resolveChatDebugLogEvent.
216 > */
217 > resolveEvent(eventId: string): Promise<IChatDebugResolvedEventContent | undefined>;
218 >
219 > /**
220 > /**
221 > * Export the debug log for a session via the registered provider.
222 > */
223 > exportLog(sessionResource: URI): Promise<Uint8Array | undefined>;
224 >
225 > /**
226 > * Import a previously exported debug log via the registered provider.
227 > * Returns the session URI for the imported data.
228 > */
229 > importLog(data: Uint8Array): Promise<URI | undefined>;
230 >
231 > /**
232 > * Returns true if the event was logged by VS Code core
233 > * (not sourced from an external provider).
234 > */
235 > isCoreEvent(event: IChatDebugEvent): boolean;
236 >
237 > /**
238 > * Store a human-readable title for an imported session.
239 > */
240 > setImportedSessionTitle(sessionResource: URI, title: string): void;
241 >
242 > /**
243 > * Get the stored title for an imported session, if available.
244 > */
245 > getImportedSessionTitle(sessionResource: URI): string | undefined;
246 >
247 > /**
248 > * Fired when available session resources change (e.g. historical sessions discovered from disk).
249 > */
250 > readonly onDidChangeAvailableSessionResources: Event<void>;
251 >
252 > /**
253 > * Store session resources that have debug log data available on disk.
254 > * Called by the main thread after the extension reports historical sessions.
255 > */
256 > addAvailableSessionResources(resources: readonly { uri: URI; title?: string }[]): void;
257 >
258 > /**
259 > * Get all session resources that have debug log data available,
260 > * including historical sessions persisted on disk by the provider.
261 > * Triggers a lazy fetch from the registered fetcher on first call.
262 > */
263 > getAvailableSessionResources(): readonly URI[];
264 >
265 > /**
266 > * Register a callback that fetches available session resources from a provider.
267 > * Called lazily when `getAvailableSessionResources()` is first invoked. Multiple
268 > * fetchers may be registered (e.g. the extension host and local agent-host
269 > * discovery); each is invoked at most once. Dispose to unregister.
270 > */
271 > registerAvailableSessionsFetcher(fetcher: (token: CancellationToken) => Promise<{ uri: URI; title?: string }[]>): IDisposable;
272 >
273 > /**
274 > * Get the stored title for a historical session discovered from disk.
275 > */
276 > getHistoricalSessionTitle(sessionResource: URI): string | undefined;
277 >
278 > }
279 >
280 > /**
281 > * Plain text content for a resolved debug event.
282 > */
283 > export interface IChatDebugEventTextContent {
284 > readonly kind: 'text';
285 > readonly value: string;
286 > }
287 >
288 > /**
289 > * The status of a file in a file list content.
290 > */
291 > export type ChatDebugFileStatus = 'loaded' | 'skipped';
292 >
293 > /**
294 > * A single file entry in a file list content.
295 > */
296 > export interface IChatDebugFileEntry {
297 > readonly uri: URI;
298 > readonly name?: string;
299 > readonly status: ChatDebugFileStatus;
300 > readonly storage?: string;
301 > readonly extensionId?: string;
302 > readonly skipReason?: string;
303 > readonly errorMessage?: string;
304 > readonly duplicateOf?: URI;
305 > }
306 >
307 > /**
308 > * A source folder entry in a file list content.
309 > */
310 > export interface IChatDebugSourceFolderEntry {
311 > readonly uri: URI;
312 > readonly storage: string;
313 > }
314 >
315 > /**
316 > * Structured file list content for a resolved debug event.
317 > * Contains resolved files and skipped/failed paths for rich rendering.
318 > */
319 > export interface IChatDebugEventFileListContent {
320 > readonly kind: 'fileList';
321 > readonly discoveryType: string;
322 > readonly durationInMillis: number;
323 > readonly files: readonly IChatDebugFileEntry[];
324 > readonly sourceFolders?: readonly IChatDebugSourceFolderEntry[];
325 > }
326 >
327 > /**
328 > * Structured message content for a resolved debug event,
329 > * containing collapsible sections.
330 > */
331 > export interface IChatDebugEventMessageContent {
332 > readonly kind: 'message';
333 > readonly type: 'user' | 'agent';
334 > readonly message: string;
335 > readonly sections: readonly IChatDebugMessageSection[];
336 > }
337 >
338 > /**
339 > * Structured tool call content for a resolved debug event.
340 > * Contains the tool name, status, arguments, and output for rich rendering.
341 > */
342 > export interface IChatDebugEventToolCallContent {
343 > readonly kind: 'toolCall';
344 > readonly toolName: string;
345 > readonly result?: 'success' | 'error';
346 > readonly durationInMillis?: number;
347 > readonly input?: string;
348 > readonly output?: string;
349 > }
350 >
351 > /**
352 > * Structured model turn content for a resolved debug event.
353 > * Contains request metadata, token usage, and timing for rich rendering.
354 > */
355 > export interface IChatDebugEventModelTurnContent {
356 > readonly kind: 'modelTurn';
357 > readonly requestName: string;
358 > readonly model?: string;
359 > readonly status?: string;
360 > readonly durationInMillis?: number;
361 > readonly timeToFirstTokenInMillis?: number;
362 > readonly requestId?: string;
363 > readonly maxInputTokens?: number;
364 > readonly maxOutputTokens?: number;
365 > readonly inputTokens?: number;
366 > readonly outputTokens?: number;
367 > readonly cachedTokens?: number;
368 > readonly totalTokens?: number;
369 > readonly requestOptions?: string;
370 > readonly errorMessage?: string;
371 > readonly sections?: readonly IChatDebugMessageSection[];
372 > }
373 >
374 > /**
375 > * Structured hook execution content for a resolved debug event.
376 > * Contains the hook type, command, input, output, and result for rich rendering.
377 > */
378 > export interface IChatDebugEventHookContent {
379 > readonly kind: 'hook';
380 > readonly hookType: string;
381 > readonly command?: string;
382 > readonly result?: ChatDebugHookResult;
383 > readonly durationInMillis?: number;
384 > readonly input?: string;
385 > readonly output?: string;
386 > readonly exitCode?: number;
387 > readonly errorMessage?: string;
388 > }
389 >
390 > /**
391 > * A single entry in the customization resolution log.
392 > */
393 > export interface IChatDebugCustomizationLogEntry {
394 > readonly category: 'applying' | 'skipped' | 'referenced' | 'skill' | 'custom-agent' | 'hook';
395 > readonly name: string;
396 > readonly uri?: URI;
397 > readonly reason?: string;
398 > }
399 >
400 > /**
401 > * Structured customization summary content for a resolved debug event.
402 > * Contains per-file resolution logs showing how applyTo patterns, agent
403 > * instructions, and referenced files were resolved by the instructions
404 > * context computer.
405 > */
406 > export interface IChatDebugEventCustomizationSummaryContent {
407 > readonly kind: 'customizationSummary';
408 > /** Per-file resolution detail entries. */
409 > readonly resolutionLogs: readonly IChatDebugCustomizationLogEntry[];
410 > /** Total wall-clock time of the collect() call in milliseconds. */
411 > readonly durationInMillis: number;
412 > /** Counts by type for the summary header. */
413 > readonly counts: {
414 > readonly instructions: number;
415 > readonly skills: number;
416 > readonly agents: number;
417 > readonly hooks: number;
418 > readonly skipped: number;
419 > };
420 > }
421 >
422 > /**
423 > * Union of all resolved event content types.
424 > */
425 > export type IChatDebugResolvedEventContent = IChatDebugEventTextContent | IChatDebugEventFileListContent | IChatDebugEventMessageContent | IChatDebugEventToolCallContent | IChatDebugEventModelTurnContent | IChatDebugEventHookContent | IChatDebugEventCustomizationSummaryContent;
426 >
427 > /**
428 > * Provider interface for debug events.
429 > */
430 > export interface IChatDebugLogProvider {
431 > provideChatDebugLog(sessionResource: URI, token: CancellationToken): Promise<IChatDebugEvent[] | undefined>;
432 > resolveChatDebugLogEvent?(eventId: string, token: CancellationToken): Promise<IChatDebugResolvedEventContent | undefined>;
433 > provideChatDebugLogExport?(sessionResource: URI, token: CancellationToken): Promise<Uint8Array | undefined>;
434 > resolveChatDebugLogImport?(data: Uint8Array, token: CancellationToken): Promise<URI | undefined>;
435 > }
src/vs/workbench/contrib/chat/common/editing/chatEditingService.ts 423 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatEditingService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { decodeHex, encodeHex, VSBuffer } from '../../../../../base/common/buffer.js';
7 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
8 > import { CancellationError } from '../../../../../base/common/errors.js';
9 > import { Event } from '../../../../../base/common/event.js';
10 > import { IDisposable } from '../../../../../base/common/lifecycle.js';
11 > import { autorunSelfDisposable, IObservable, IReader } from '../../../../../base/common/observable.js';
12 > import { hasKey } from '../../../../../base/common/types.js';
13 > import { URI } from '../../../../../base/common/uri.js';
14 > import { IDocumentDiff } from '../../../../../editor/common/diff/documentDiffProvider.js';
15 > import { TextEdit } from '../../../../../editor/common/languages.js';
16 > import { ITextModel } from '../../../../../editor/common/model.js';
17 > import { EditSuggestionId } from '../../../../../editor/common/textModelEditSource.js';
18 > import { localize } from '../../../../../nls.js';
19 > import { RawContextKey } from '../../../../../platform/contextkey/common/contextkey.js';
20 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
21 > import { IEditorPane } from '../../../../common/editor.js';
22 > import { ICellEditOperation } from '../../../notebook/common/notebookCommon.js';
23 > import { IChatMultiDiffData, IChatMultiDiffDataSerialized, IChatProgress, IChatWorkspaceEdit } from '../chatService/chatService.js';
24 > import { ChatModel, IChatRequestDisablement, IChatResponseModel } from '../model/chatModel.js';
25 > import { IChatAgentResult } from '../participants/chatAgents.js';
26 >
27 > export const IChatEditingService = createDecorator<IChatEditingService>('chatEditingService');
28 >
29 > export interface IChatEditingSessionProvider {
30 > createEditingSession(chatSessionResource: URI): IChatEditingSession;
31 > }
32 >
33 > export interface IChatEditingService {
34 >
35 > _serviceBrand: undefined;
36 >
37 > startOrContinueGlobalEditingSession(chatModel: ChatModel): IChatEditingSession;
38 >
39 > getEditingSession(chatSessionResource: URI): IChatEditingSession | undefined;
40 >
41 > /**
42 > * All editing sessions, sorted by recency, e.g the last created session comes first.
43 > */
44 > readonly editingSessionsObs: IObservable<readonly IChatEditingSession[]>;
45 >
46 > /**
47 > * Creates a new short lived editing session
48 > */
49 > createEditingSession(chatModel: ChatModel): IChatEditingSession;
50 >
51 > /**
52 > * Creates an editing session with state transferred from the provided session.
53 > */
54 > transferEditingSession(chatModel: ChatModel, session: IChatEditingSession): IChatEditingSession;
55 >
56 > /**
57 > * Registers a provider that creates editing sessions for chat sessions
58 > * with the given URI scheme. When {@link createEditingSession} is called
59 > * for a chat model whose sessionResource matches the scheme, the provider
60 > * is used instead of the default implementation.
61 > */
62 > registerEditingSessionProvider(scheme: string, provider: IChatEditingSessionProvider): IDisposable;
63 > }
64 >
65 > export interface WorkingSetDisplayMetadata {
66 > state: ModifiedFileEntryState;
67 > description?: string;
68 > }
69 >
70 > export interface IStreamingEdits {
71 > pushText(edits: TextEdit[], isLastEdits: boolean): void;
72 > pushNotebookCellText(cell: URI, edits: TextEdit[], isLastEdits: boolean): void;
73 > pushNotebook(edits: ICellEditOperation[], isLastEdits: boolean): void;
74 > /** Marks edits as done, idempotent */
75 > complete(): void;
76 > }
77 >
78 > export interface IModifiedEntryTelemetryInfo {
79 > readonly agentId: string | undefined;
80 > readonly command: string | undefined;
81 > readonly sessionResource: URI;
82 > readonly requestId: string;
83 > readonly result: IChatAgentResult | undefined;
84 > readonly modelId: string | undefined;
85 > readonly modeId: 'ask' | 'edit' | 'agent' | 'custom' | 'applyCodeBlock' | undefined;
86 > readonly applyCodeBlockSuggestionId: EditSuggestionId | undefined;
87 > readonly feature: 'sideBarChat' | 'inlineChat' | undefined;
88 > }
89 >
90 > export interface ISnapshotEntry {
91 > readonly resource: URI;
92 > readonly languageId: string;
93 > readonly snapshotUri: URI;
94 > readonly original: string;
95 > readonly current: string;
96 > readonly state: ModifiedFileEntryState;
97 > telemetryInfo: IModifiedEntryTelemetryInfo;
98 > /** True if this entry represents a deleted file */
99 > readonly isDeleted?: boolean;
100 > }
101 >
102 > export interface IChatEditingSession extends IDisposable {
103 > readonly isGlobalEditingSession: boolean;
104 > readonly supportsKeepUndo: boolean;
105 > readonly chatSessionResource: URI;
106 > readonly onDidDispose: Event<void>;
107 > readonly state: IObservable<ChatEditingSessionState>;
108 > readonly entries: IObservable<readonly IModifiedFileEntry[]>;
109 > /** Requests disabled by undo/redo in the session */
110 > readonly requestDisablement: IObservable<IChatRequestDisablement[]>;
111 >
112 > show(previousChanges?: boolean): Promise<void>;
113 > accept(...uris: URI[]): Promise<void>;
114 > reject(...uris: URI[]): Promise<void>;
115 > getEntry(uri: URI): IModifiedFileEntry | undefined;
116 > readEntry(uri: URI, reader: IReader): IModifiedFileEntry | undefined;
117 >
118 > restoreSnapshot(requestId: string, stopId: string | undefined): Promise<void>;
119 >
120 > /**
121 > * Marks all edits to the given resources as agent edits until
122 > * {@link stopExternalEdits} is called with the same ID. This is used for
123 > * agents that make changes on-disk rather than streaming edits through the
124 > * chat session.
125 > */
126 > startExternalEdits(responseModel: IChatResponseModel, operationId: number, resources: URI[], undoStopId: string, contentFor?: URI[]): Promise<IChatProgress[]>;
127 > stopExternalEdits(responseModel: IChatResponseModel, operationId: number, contentFor?: URI[]): Promise<IChatProgress[]>;
128 >
129 > /**
130 > * Gets the snapshot URI of a file at the request and _after_ changes made in the undo stop.
131 > * @param uri File in the workspace
132 > */
133 > getSnapshotUri(requestId: string, uri: URI, stopId: string | undefined): URI | undefined;
134 >
135 > getSnapshotContents(requestId: string, uri: URI, stopId: string | undefined): Promise<VSBuffer | undefined>;
136 > getSnapshotModel(requestId: string, undoStop: string | undefined, snapshotUri: URI): Promise<ITextModel | null>;
137 >
138 > /**
139 > * Will lead to this object getting disposed
140 > */
141 > stop(clearState?: boolean): Promise<void>;
142 >
143 > /**
144 > * Starts making edits to the resource.
145 > * @param resource URI that's being edited
146 > * @param responseModel The response model making the edits
147 > * @param inUndoStop The undo stop the edits will be grouped in
148 > */
149 > startStreamingEdits(resource: URI, responseModel: IChatResponseModel, inUndoStop: string | undefined): IStreamingEdits;
150 >
151 > /**
152 > * Applies a workspace edit (file deletions, creations, renames).
153 > * @param edit The workspace edit containing file operations
154 > * @param responseModel The response model making the edit
155 > * @param undoStopId The undo stop ID for this edit
156 > */
157 > applyWorkspaceEdit(edit: IChatWorkspaceEdit, responseModel: IChatResponseModel, undoStopId: string): void;
158 >
159 > /**
160 > * Gets the document diff of a change made to a URI between one undo stop and
161 > * the next one.
162 > * @returns The observable or undefined if there is no diff between the stops.
163 > */
164 > getEntryDiffBetweenStops(uri: URI, requestId: string | undefined, stopId: string | undefined): IObservable<IEditSessionEntryDiff | undefined> | undefined;
165 >
166 > /**
167 > * Gets the document diff of a change made to a URI between one request to another one.
168 > * @returns The observable or undefined if there is no diff between the requests.
169 > */
170 > getEntryDiffBetweenRequests(uri: URI, startRequestIs: string, stopRequestId: string): IObservable<IEditSessionEntryDiff | undefined>;
171 >
172 > /**
173 > * Gets the diff of each file modified in this session, comparing the initial
174 > * baseline to the current state.
175 > */
176 > getDiffsForFilesInSession(): IObservable<readonly IEditSessionEntryDiff[]>;
177 >
178 > /**
179 > * Gets the diff of each file modified in the request.
180 > */
181 > getDiffsForFilesInRequest(requestId: string): IObservable<readonly IEditSessionEntryDiff[]>;
182 >
183 > /**
184 > * Whether there are any edits made in the given request.
185 > */
186 > hasEditsInRequest(requestId: string, reader?: IReader): boolean;
187 >
188 > /**
189 > * Gets the aggregated diff stats for all files modified in this session.
190 > */
191 > getDiffForSession(): IObservable<IEditSessionDiffStats>;
192 >
193 > readonly canUndo: IObservable<boolean>;
194 > readonly canRedo: IObservable<boolean>;
195 > undoInteraction(): Promise<void>;
196 > redoInteraction(): Promise<void>;
197 >
198 > /**
199 > * Triggers generation of explanations for all modified files in the session.
200 > */
201 > triggerExplanationGeneration(): Promise<void>;
202 >
203 > /**
204 > * Clears any active explanation generation.
205 > */
206 > clearExplanations(): void;
207 >
208 > /**
209 > * Whether explanations are currently being generated or displayed.
210 > */
211 > hasExplanations(): boolean;
212 > }
213 >
214 > export function chatEditingSessionIsReady(session: IChatEditingSession): Promise<void> {
215 > return new Promise<void>(resolve => { chatEditingService.ts
216 > autorunSelfDisposable(reader => {
217 > const state = session.state.read(reader);
218 > if (state !== ChatEditingSessionState.Initial) {
219 > reader.dispose();
220 > resolve();
221 > }
222 > });
223 > });
224 > }
226 > export function editEntriesToMultiDiffData(entriesObs: IObservable<readonly IEditSessionEntryDiff[]>): IChatMultiDiffData {
227 const multiDiffData = entriesObs.map(entries => ({
228 title: localize('chatMultidiff.autoGenerated', 'Changes to {0} files', entries.length),
249 };
250 }
252 > export function awaitCompleteChatEditingDiff(diff: IObservable<IEditSessionEntryDiff>, token?: CancellationToken): Promise<IEditSessionEntryDiff>;
253 > export function awaitCompleteChatEditingDiff(diff: IObservable<readonly IEditSessionEntryDiff[]>, token?: CancellationToken): Promise<readonly IEditSessionEntryDiff[]>;
254 > export function awaitCompleteChatEditingDiff(diff: IObservable<readonly IEditSessionEntryDiff[] | IEditSessionEntryDiff>, token?: CancellationToken): Promise<readonly IEditSessionEntryDiff[] | IEditSessionEntryDiff> {
255 return new Promise<readonly IEditSessionEntryDiff[] | IEditSessionEntryDiff>((resolve, reject) => {
256 autorunSelfDisposable(reader => {
279 });
280 }
282 > export interface IEditSessionDiffStats {
283 > /** Added data (e.g. line numbers) to show in the UI */
284 > added: number;
285 > /** Removed data (e.g. line numbers) to show in the UI */
286 > removed: number;
287 > }
288 >
289 > export interface IEditSessionEntryDiff extends IEditSessionDiffStats {
290 > /** LHS and RHS of a diff editor, if opened: */
291 > originalURI: URI;
292 > modifiedURI: URI;
293 >
294 > /**
295 > * Optional frozen "after" content for the RHS. When set, this is the exact
296 > * modified-side snapshot the diff represents (e.g. an agent-host per-turn
297 > * checkpoint), as opposed to {@link modifiedURI} which may be the live
298 > * working file and therefore include later changes. Consumers that want the
299 > * changeset's own diff should prefer this when present; {@link modifiedURI}
300 > * remains the file's identity for labels and go-to-file.
301 > *
302 > * Note: distinct from the agent-host checkpoint-ref readability fix (#323932).
303 > * That made the frozen snapshot blobs *readable*; this field carries *which*
304 > * snapshot to diff against so a per-turn review shows only that turn's changes.
305 > */
306 > modifiedSnapshotURI?: URI;
307 >
308 > /** Diff state information: */
309 > quitEarly: boolean;
310 > identical: boolean;
311 >
312 > /** True if nothing else will be added to this diff. */
313 > isFinal: boolean;
314 >
315 > /** True if the diff is currently being computed or updated. */
316 > isBusy: boolean;
317 > }
318 >
319 > export function emptySessionEntryDiff(originalURI: URI, modifiedURI: URI): IEditSessionEntryDiff {
320 return {
321 originalURI,
329 };
330 }
332 > export const enum ModifiedFileEntryState {
333 > Modified,
334 > Accepted,
335 > Rejected,
336 > }
337 >
338 > /**
339 > * Represents a part of a change
340 > */
341 > export interface IModifiedFileEntryChangeHunk {
342 > accept(): Promise<boolean>;
343 > reject(): Promise<boolean>;
344 > }
345 >
346 > export interface IModifiedFileEntryEditorIntegration extends IDisposable {
347 >
348 > /**
349 > * The index of a change
350 > */
351 > currentIndex: IObservable<number>;
352 >
353 > /**
354 > * Reveal the first (`true`) or last (`false`) change
355 > */
356 > reveal(firstOrLast: boolean, preserveFocus?: boolean): void;
357 >
358 > /**
359 > * Go to next change and increate `currentIndex`
360 > * @param wrap When at the last, start over again or not
361 > * @returns If it went next
362 > */
363 > next(wrap: boolean): boolean;
364 >
365 > /**
366 > * @see `next`
367 > */
368 > previous(wrap: boolean): boolean;
369 >
370 > /**
371 > * Enable the accessible diff viewer for this editor
372 > */
373 > enableAccessibleDiffView(): void;
374 >
375 > /**
376 > * Accept the change given or the nearest
377 > * @param change An opaque change object
378 > */
379 > acceptNearestChange(change?: IModifiedFileEntryChangeHunk): Promise<void>;
380 >
381 > /**
382 > * @see `acceptNearestChange`
383 > */
384 > rejectNearestChange(change?: IModifiedFileEntryChangeHunk): Promise<void>;
385 >
386 > /**
387 > * Toggle between diff-editor and normal editor
388 > * @param change An opaque change object
389 > * @param show Optional boolean to control if the diff should show
390 > */
391 > toggleDiff(change: IModifiedFileEntryChangeHunk | undefined, show?: boolean): Promise<void>;
392 > }
393 >
394 > export interface IModifiedFileEntry {
395 > readonly entryId: string;
396 > readonly originalURI: URI;
397 > readonly modifiedURI: URI;
398 > readonly isDeletion?: boolean;
399 >
400 > readonly lastModifyingRequestId: string;
401 >
402 > readonly state: IObservable<ModifiedFileEntryState>;
403 > readonly isCurrentlyBeingModifiedBy: IObservable<{ responseModel: IChatResponseModel; undoStopId: string | undefined } | undefined>;
404 > readonly lastModifyingResponse: IObservable<IChatResponseModel | undefined>;
405 > readonly rewriteRatio: IObservable<number>;
406 >
407 > readonly waitsForLastEdits: IObservable<boolean>;
408 >
409 > accept(): Promise<void>;
410 > reject(): Promise<void>;
411 >
412 > reviewMode: IObservable<boolean>;
413 > autoAcceptController: IObservable<{ total: number; remaining: number; cancel(): void } | undefined>;
414 > enableReviewModeUntilSettled(): void;
415 >
416 > /**
417 > * Number of changes for this file
418 > */
419 > readonly changesCount: IObservable<number>;
420 >
421 > /**
422 > * Diff information for this entry
423 > */
424 > readonly diffInfo?: IObservable<IDocumentDiff>;
425 >
426 > /**
427 > * Number of lines added in this entry.
428 > */
429 > readonly linesAdded?: IObservable<number>;
430 >
431 > /**
432 > * Number of lines removed in this entry
433 > */
434 > readonly linesRemoved?: IObservable<number>;
435 >
436 > getEditorIntegration(editor: IEditorPane): IModifiedFileEntryEditorIntegration;
437 > /**
438 > * Gets the document diff info, waiting for any ongoing promises to flush.
439 > */
440 > getDiffInfo?(): Promise<IDocumentDiff>;
441 > }
442 >
443 > export interface IChatEditingSessionStream {
444 > textEdits(resource: URI, textEdits: TextEdit[], isLastEdits: boolean, responseModel: IChatResponseModel): void;
445 > notebookEdits(resource: URI, edits: ICellEditOperation[], isLastEdits: boolean, responseModel: IChatResponseModel): void;
446 > }
447 >
448 > export const enum ChatEditingSessionState {
449 > Initial = 0,
450 > StreamingEdits = 1,
451 > Idle = 2,
452 > Disposed = 3
453 > }
454 >
455 > export const CHAT_EDITING_MULTI_DIFF_SOURCE_RESOLVER_SCHEME = 'chat-editing-multi-diff-source';
456 >
457 > export const chatEditingWidgetFileStateContextKey = new RawContextKey<ModifiedFileEntryState>('chatEditingWidgetFileState', undefined, localize('chatEditingWidgetFileState', "The current state of the file in the chat editing widget"));
458 > export const chatEditingAgentSupportsReadonlyReferencesContextKey = new RawContextKey<boolean>('chatEditingAgentSupportsReadonlyReferences', undefined, localize('chatEditingAgentSupportsReadonlyReferences', "Whether the chat editing agent supports readonly references (temporary)"));
459 > export const decidedChatEditingResourceContextKey = new RawContextKey<string[]>('decidedChatEditingResource', []);
460 > export const chatEditingResourceContextKey = new RawContextKey<string | undefined>('chatEditingResource', undefined);
461 > export const inChatEditingSessionContextKey = new RawContextKey<boolean | undefined>('inChatEditingSession', undefined);
462 > export const hasUndecidedChatEditingResourceContextKey = new RawContextKey<boolean | undefined>('hasUndecidedChatEditingResource', false);
463 > export const hasAppliedChatEditsContextKey = new RawContextKey<boolean | undefined>('hasAppliedChatEdits', false);
464 > export const applyingChatEditsFailedContextKey = new RawContextKey<boolean | undefined>('applyingChatEditsFailed', false);
465 >
466 > export const chatEditingMaxFileAssignmentName = 'chatEditingSessionFileLimit';
467 > export const defaultChatEditingMaxFileLimit = 10;
468 >
469 > export const enum ChatEditKind {
470 > Created,
471 > Modified,
472 > Deleted,
473 > }
474 >
475 > export interface IChatEditingActionContext {
476 > // The chat session that this editing session is associated with
477 > sessionResource: URI;
478 > }
479 >
480 > export function isChatEditingActionContext(thing: unknown): thing is IChatEditingActionContext {
481 return typeof thing === 'object' && !!thing && hasKey(thing, { sessionResource: true });
482 }
484 > export function getMultiDiffSourceUri(session: IChatEditingSession, showPreviousChanges?: boolean): URI {
485 return URI.from({
486 scheme: CHAT_EDITING_MULTI_DIFF_SOURCE_RESOLVER_SCHEME,
489 });
490 }
492 > export function parseChatMultiDiffUri(uri: URI): { chatSessionResource: URI; showPreviousChanges: boolean } {
493 const chatSessionResource = URI.parse(decodeHex(uri.authority).toString());
494 const showPreviousChanges = uri.query === 'previous';
src/vs/base/common/arrays.ts 410 covered LOC · 76 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- arrays.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { findFirstIdxMonotonousOrArrLen } from './arraysFind.js';
7 > import { CancellationToken } from './cancellation.js';
8 > import { CancellationError } from './errors.js';
9 > import { ISplice } from './sequence.js';
10 >
11 > /**
12 > * Returns the last entry and the initial N-1 entries of the array, as a tuple of [rest, last].
13 > *
14 > * The array must have at least one element.
15 > *
16 > * @param arr The input array
17 > * @returns A tuple of [rest, last] where rest is all but the last element and last is the last element
18 > * @throws Error if the array is empty
19 > */
20 > export function tail<T>(arr: T[]): [T[], T] {
21 if (arr.length === 0) {
22 throw new Error('Invalid tail call');
25 return [arr.slice(0, arr.length - 1), arr[arr.length - 1]];
26 }
27 > arrays.ts
28 > export function equals<T>(one: ReadonlyArray<T> | undefined, other: ReadonlyArray<T> | undefined, itemEquals: (a: T, b: T) => boolean = (a, b) => a === b): boolean {
29 if (one === other) {
30 return true;
47 return true;
48 }
49 > arrays.ts
50 > /**
51 > * Remove the element at `index` by replacing it with the last element. This is faster than `splice`
52 > * but changes the order of the array
53 > */
54 > export function removeFastWithoutKeepingOrder<T>(array: T[], index: number) {
55 const last = array.length - 1;
56 if (index < last) {
59 array.pop();
60 }
61 > arrays.ts
62 > /**
63 > * Performs a binary search algorithm over a sorted array.
64 > *
65 > * @param array The array being searched.
66 > * @param key The value we search for.
67 > * @param comparator A function that takes two array elements and returns zero
68 > * if they are equal, a negative number if the first element precedes the
69 > * second one in the sorting order, or a positive number if the second element
70 > * precedes the first one.
71 > * @return See {@link binarySearch2}
72 > */
73 > export function binarySearch<T>(array: ReadonlyArray<T>, key: T, comparator: (op1: T, op2: T) => number): number {
74 return binarySearch2(array.length, i => comparator(array[i], key));
75 }
76 > arrays.ts
77 > /**
78 > * Performs a binary search algorithm over a sorted collection. Useful for cases
79 > * when we need to perform a binary search over something that isn't actually an
80 > * array, and converting data to an array would defeat the use of binary search
81 > * in the first place.
82 > *
83 > * @param length The collection length.
84 > * @param compareToKey A function that takes an index of an element in the
85 > * collection and returns zero if the value at this index is equal to the
86 > * search key, a negative number if the value precedes the search key in the
87 > * sorting order, or a positive number if the search key precedes the value.
88 > * @return A non-negative index of an element, if found. If not found, the
89 > * result is -(n+1) (or ~n, using bitwise notation), where n is the index
90 > * where the key should be inserted to maintain the sorting order.
91 > */
92 > export function binarySearch2(length: number, compareToKey: (index: number) => number): number {
93 let low = 0,
94 high = length - 1;
107 return -(low + 1);
108 }
109 > arrays.ts
110 > type Compare<T> = (a: T, b: T) => number;
111 >
112 > /**
113 > * Finds the nth smallest element in the array using quickselect algorithm.
114 > * The data does not need to be sorted.
115 > *
116 > * @param nth The zero-based index of the element to find (0 = smallest, 1 = second smallest, etc.)
117 > * @param data The unsorted array
118 > * @param compare A comparator function that defines the sort order
119 > * @returns The nth smallest element
120 > * @throws TypeError if nth is >= data.length
121 > */
122 > export function quickSelect<T>(nth: number, data: T[], compare: Compare<T>): T {
123
124 nth = nth | 0;
152 }
153 }
154 > arrays.ts
155 > export function groupBy<T>(data: ReadonlyArray<T>, compare: (a: T, b: T) => number): T[][] {
156 const result: T[][] = [];
157 let currentGroup: T[] | undefined = undefined;
166 return result;
167 }
168 > arrays.ts
169 > /**
170 > * Splits the given items into a list of (non-empty) groups.
171 > * `shouldBeGrouped` is used to decide if two consecutive items should be in the same group.
172 > * The order of the items is preserved.
173 > */
174 > export function* groupAdjacentBy<T>(items: Iterable<T>, shouldBeGrouped: (item1: T, item2: T) => boolean): Iterable<T[]> {
175 let currentGroup: T[] | undefined;
176 let last: T | undefined;
190 }
191 }
192 > arrays.ts
193 > export function forEachAdjacent<T>(arr: T[], f: (item1: T | undefined, item2: T | undefined) => void): void {
194 for (let i = 0; i <= arr.length; i++) {
195 f(i === 0 ? undefined : arr[i - 1], i === arr.length ? undefined : arr[i]);
196 }
197 }
198 > arrays.ts
199 > export function forEachWithNeighbors<T>(arr: T[], f: (before: T | undefined, element: T, after: T | undefined) => void): void {
200 for (let i = 0; i < arr.length; i++) {
201 f(i === 0 ? undefined : arr[i - 1], arr[i], i + 1 === arr.length ? undefined : arr[i + 1]);
202 }
203 }
204 > arrays.ts
205 > export function concatArrays<T extends any[]>(...arrays: T): T[number][number][] {
206 return [].concat(...arrays);
207 }
208 > arrays.ts
209 > interface IMutableSplice<T> extends ISplice<T> {
210 > readonly toInsert: T[];
211 > deleteCount: number;
212 > }
213 >
214 > /**
215 > * Diffs two *sorted* arrays and computes the splices which apply the diff.
216 > */
217 > export function sortedDiff<T>(before: ReadonlyArray<T>, after: ReadonlyArray<T>, compare: (a: T, b: T) => number): ISplice<T>[] {
218 const result: IMutableSplice<T>[] = [];
219
266 return result;
267 }
268 > arrays.ts
269 > /**
270 > * Takes two *sorted* arrays and computes their delta (removed, added elements).
271 > * Finishes in `Math.min(before.length, after.length)` steps.
272 > */
273 > export function delta<T>(before: ReadonlyArray<T>, after: ReadonlyArray<T>, compare: (a: T, b: T) => number): { removed: T[]; added: T[] } {
274 const splices = sortedDiff(before, after, compare);
275 const removed: T[] = [];
283 return { removed, added };
284 }
285 > arrays.ts
286 > /**
287 > * Returns the top N elements from the array.
288 > *
289 > * Faster than sorting the entire array when the array is a lot larger than N.
290 > *
291 > * @param array The unsorted array.
292 > * @param compare A sort function for the elements.
293 > * @param n The number of elements to return.
294 > * @return The first n elements from array when sorted with compare.
295 > */
296 > export function top<T>(array: ReadonlyArray<T>, compare: (a: T, b: T) => number, n: number): T[] {
297 if (n === 0) {
298 return [];
302 return result;
303 }
304 > arrays.ts
305 > /**
306 > * Asynchronous variant of `top()` allowing for splitting up work in batches between which the event loop can run.
307 > *
308 > * Returns the top N elements from the array.
309 > *
310 > * Faster than sorting the entire array when the array is a lot larger than N.
311 > *
312 > * @param array The unsorted array.
313 > * @param compare A sort function for the elements.
314 > * @param n The number of elements to return.
315 > * @param batch The number of elements to examine before yielding to the event loop.
316 > * @return The first n elements from array when sorted with compare.
317 > */
318 > export function topAsync<T>(array: T[], compare: (a: T, b: T) => number, n: number, batch: number, token?: CancellationToken): Promise<T[]> {
319 if (n === 0) {
320 return Promise.resolve([]);
339 });
340 }
341 > arrays.ts
342 function topStep<T>(array: ReadonlyArray<T>, compare: (a: T, b: T) => number, result: T[], i: number, m: number): void {
343 for (const n = result.length; i < m; i++) {
350 }
351 }
352 > arrays.ts
353 > /**
354 > * @returns New array with all falsy values removed. The original array IS NOT modified.
355 > */
356 > export function coalesce<T>(array: ReadonlyArray<T | undefined | null>): T[] {
357 return array.filter((e): e is T => !!e);
358 }
359 > arrays.ts
360 > /**
361 > * Remove all falsy values from `array`. The original array IS modified.
362 > */
363 > export function coalesceInPlace<T>(array: Array<T | undefined | null>): asserts array is Array<T> {
364 let to = 0;
365 for (let i = 0; i < array.length; i++) {
371 array.length = to;
372 }
373 > arrays.ts
374 > /**
375 > * @deprecated Use `Array.copyWithin` instead
376 > */
377 > export function move(array: unknown[], from: number, to: number): void {
378 array.splice(to, 0, array.splice(from, 1)[0]);
379 }
380 > arrays.ts
381 > /**
382 > * @returns false if the provided object is an array and not empty.
383 > */
384 > export function isFalsyOrEmpty(obj: unknown): boolean {
385 return !Array.isArray(obj) || obj.length === 0;
386 }
387 > arrays.ts
388 > /**
389 > * @returns True if the provided object is an array and has at least one element.
390 > */
391 > export function isNonEmptyArray<T>(obj: T[] | undefined | null): obj is T[];
392 > export function isNonEmptyArray<T>(obj: readonly T[] | undefined | null): obj is readonly T[];
393 > export function isNonEmptyArray<T>(obj: T[] | readonly T[] | undefined | null): obj is T[] | readonly T[] {
394 return Array.isArray(obj) && obj.length > 0;
395 }
396 > arrays.ts
397 > /**
398 > * Removes duplicates from the given array. The optional keyFn allows to specify
399 > * how elements are checked for equality by returning an alternate value for each.
400 > */
401 > export function distinct<T>(array: ReadonlyArray<T>, keyFn: (value: T) => unknown = value => value): T[] {
402 const seen = new Set<any>();
403
411 });
412 }
413 > arrays.ts
414 > export function uniqueFilter<T, R>(keyFn: (t: T) => R): (t: T) => boolean {
415 const seen = new Set<R>();
416
426 };
427 }
428 > arrays.ts
429 > export function commonPrefixLength<T>(one: ReadonlyArray<T>, other: ReadonlyArray<T>, equals: (a: T, b: T) => boolean = (a, b) => a === b): number {
430 let result = 0;
431
436 return result;
437 }
438 > arrays.ts
439 > export function range(to: number): number[];
440 > export function range(from: number, to: number): number[];
441 > export function range(arg: number, to?: number): number[] {
442 let from = typeof to === 'number' ? arg : 0;
443
463 return result;
464 }
465 > arrays.ts
466 > export function index<T>(array: ReadonlyArray<T>, indexer: (t: T) => string): { [key: string]: T };
467 > export function index<T, R>(array: ReadonlyArray<T>, indexer: (t: T) => string, mapper: (t: T) => R): { [key: string]: R };
468 > export function index<T, R>(array: ReadonlyArray<T>, indexer: (t: T) => string, mapper?: (t: T) => R): { [key: string]: R } {
469 return array.reduce((r, t) => {
470 r[indexer(t)] = mapper ? mapper(t) : t;
472 }, Object.create(null));
473 }
474 > arrays.ts
475 > /**
476 > * Inserts an element into an array. Returns a function which, when
477 > * called, will remove that element from the array.
478 > *
479 > * @deprecated In almost all cases, use a `Set<T>` instead.
480 > */
481 > export function insert<T>(array: T[], element: T): () => void {
482 array.push(element);
483
484 return () => remove(array, element);
485 }
486 > arrays.ts
487 > /**
488 > * Removes an element from an array if it can be found.
489 > *
490 > * @deprecated In almost all cases, use a `Set<T>` instead.
491 > */
492 > export function remove<T>(array: T[], element: T): T | undefined {
493 const index = array.indexOf(element);
494 if (index > -1) {
500 return undefined;
501 }
502 > arrays.ts
503 > /**
504 > * Insert `insertArr` inside `target` at `insertIndex`.
505 > * Please don't touch unless you understand https://jsperf.com/inserting-an-array-within-an-array
506 > */
507 > export function arrayInsert<T>(target: T[], insertIndex: number, insertArr: T[]): T[] {
508 const before = target.slice(0, insertIndex);
509 const after = target.slice(insertIndex);
510 return before.concat(insertArr, after);
511 }
512 > arrays.ts
513 > /**
514 > * Uses Fisher-Yates shuffle to shuffle the given array
515 > */
516 > export function shuffle<T>(array: T[], _seed?: number): void {
517 let rand: () => number;
518
536 }
537 }
538 > arrays.ts
539 > /**
540 > * Pushes an element to the start of the array, if found.
541 > */
542 > export function pushToStart<T>(arr: T[], value: T): void {
543 const index = arr.indexOf(value);
544
548 }
549 }
550 > arrays.ts
551 > /**
552 > * Pushes an element to the end of the array, if found.
553 > */
554 > export function pushToEnd<T>(arr: T[], value: T): void {
555 const index = arr.indexOf(value);
556
560 }
561 }
562 > arrays.ts
563 > export function pushMany<T>(arr: T[], items: ReadonlyArray<T>): void {
564 for (const item of items) {
565 arr.push(item);
566 }
567 }
568 > arrays.ts
569 > export function mapArrayOrNot<T, U>(items: T | T[], fn: (_: T) => U): U | U[] {
570 return Array.isArray(items) ?
571 items.map(fn) :
572 fn(items);
573 }
574 > arrays.ts
575 > export function mapFilter<T, U>(array: ReadonlyArray<T>, fn: (t: T) => U | undefined): U[] {
576 const result: U[] = [];
577 for (const item of array) {
583 return result;
584 }
585 > arrays.ts
586 > export function withoutDuplicates<T>(array: ReadonlyArray<T>): T[] {
587 const s = new Set(array);
588 return Array.from(s);
589 }
590 > arrays.ts
591 > export function asArray<T>(x: T | T[]): T[];
592 > export function asArray<T>(x: T | readonly T[]): readonly T[];
593 > export function asArray<T>(x: T | T[]): T[] {
594 > return Array.isArray(x) ? x : [x]; arrays.ts
595 > }
596 > arrays.ts
597 > export function getRandomElement<T>(arr: T[]): T | undefined {
598 return arr[Math.floor(Math.random() * arr.length)];
599 }
600 > arrays.ts
601 > /**
602 > * Insert the new items in the array.
603 > * @param array The original array.
604 > * @param start The zero-based location in the array from which to start inserting elements.
605 > * @param newItems The items to be inserted
606 > */
607 > export function insertInto<T>(array: T[], start: number, newItems: T[]): void {
608 const startIdx = getActualStartIndex(array, start);
609 const originalLength = array.length;
619 }
620 }
621 > arrays.ts
622 > /**
623 > * Removes elements from an array and inserts new elements in their place, returning the deleted elements. Alternative to the native Array.splice method, it
624 > * can only support limited number of items due to the maximum call stack size limit.
625 > * @param array The original array.
626 > * @param start The zero-based location in the array from which to start removing elements.
627 > * @param deleteCount The number of elements to remove.
628 > * @returns An array containing the elements that were deleted.
629 > */
630 > export function splice<T>(array: T[], start: number, deleteCount: number, newItems: T[]): T[] {
631 const index = getActualStartIndex(array, start);
632 let result = array.splice(index, deleteCount);
638 return result;
639 }
640 > arrays.ts
641 > /**
642 > * Determine the actual start index (same logic as the native splice() or slice())
643 > * If greater than the length of the array, start will be set to the length of the array. In this case, no element will be deleted but the method will behave as an adding function, adding as many element as item[n*] provided.
644 > * If negative, it will begin that many elements from the end of the array. (In this case, the origin -1, meaning -n is the index of the nth last element, and is therefore equivalent to the index of array.length - n.) If array.length + start is less than 0, it will begin from index 0.
645 > * @param array The target array.
646 > * @param start The operation index.
647 > */
648 function getActualStartIndex<T>(array: T[], start: number): number {
649 return start < 0 ? Math.max(start + array.length, 0) : Math.min(start, array.length);
650 }
651 > arrays.ts
652 >
653 >
654 > /**
655 > * When comparing two values,
656 > * a negative number indicates that the first value is less than the second,
657 > * a positive number indicates that the first value is greater than the second,
658 > * and zero indicates that neither is the case.
659 > */
660 > export type CompareResult = number;
661 >
662 > export namespace CompareResult {
663 > export function isLessThan(result: CompareResult): boolean {
664 return result < 0;
665 }
666 > arrays.ts
667 > export function isLessThanOrEqual(result: CompareResult): boolean {
668 return result <= 0;
669 }
670 > arrays.ts
671 > export function isGreaterThan(result: CompareResult): boolean {
672 return result > 0;
673 }
674 > arrays.ts
675 > export function isNeitherLessOrGreaterThan(result: CompareResult): boolean {
676 return result === 0;
677 }
678 > arrays.ts
679 > export const greaterThan = 1;
680 > export const lessThan = -1;
681 > export const neitherLessOrGreaterThan = 0;
682 > }
683 >
684 > /**
685 > * A comparator `c` defines a total order `<=` on `T` as following:
686 > * `c(a, b) <= 0` iff `a` <= `b`.
687 > * We also have `c(a, b) == 0` iff `c(b, a) == 0`.
688 > */
689 > export type Comparator<T> = (a: T, b: T) => CompareResult;
690 >
691 > export function compareBy<TItem, TCompareBy>(selector: (item: TItem) => TCompareBy, comparator: Comparator<TCompareBy>): Comparator<TItem> {
692 > return (a, b) => comparator(selector(a), selector(b)); arrays.ts
693 > }
694 > arrays.ts
695 > export function tieBreakComparators<TItem>(...comparators: Comparator<TItem>[]): Comparator<TItem> {
696 > return (item1, item2) => { arrays.ts
697 for (const comparator of comparators) {
698 const result = comparator(item1, item2);
703 return CompareResult.neitherLessOrGreaterThan;
704 };
705 > } arrays.ts
706 > arrays.ts
707 > /**
708 > * The natural order on numbers.
709 > */
710 > export const numberComparator: Comparator<number> = (a, b) => a - b;
711 >
712 > export const booleanComparator: Comparator<boolean> = (a, b) => numberComparator(a ? 1 : 0, b ? 1 : 0);
713 >
714 > export function reverseOrder<TItem>(comparator: Comparator<TItem>): Comparator<TItem> {
715 return (a, b) => -comparator(a, b);
716 }
717 > arrays.ts
718 > /**
719 > * Returns a new comparator that treats `undefined` as the smallest value.
720 > * All other values are compared using the given comparator.
721 > */
722 > export function compareUndefinedSmallest<T>(comparator: Comparator<T>): Comparator<T | undefined> {
723 return (a, b) => {
724 if (a === undefined) {
731 };
732 }
733 > arrays.ts
734 > export class ArrayQueue<T> {
735 > private readonly items: readonly T[];
736 > private firstIdx = 0;
737 > private lastIdx: number;
738 >
739 > /**
740 > * Constructs a queue that is backed by the given array. Runtime is O(1).
741 > */
742 > constructor(items: readonly T[]) {
743 this.items = items;
744 this.lastIdx = this.items.length - 1;
745 }
746 > arrays.ts
747 > get length(): number {
748 return this.lastIdx - this.firstIdx + 1;
749 }
750 > arrays.ts
751 > /**
752 > * Consumes elements from the beginning of the queue as long as the predicate returns true.
753 > * If no elements were consumed, `null` is returned. Has a runtime of O(result.length).
754 > */
755 > takeWhile(predicate: (value: T) => boolean): T[] | null {
756 // P(k) := k <= this.lastIdx && predicate(this.items[k])
757 // Find s := min { k | k >= this.firstIdx && !P(k) } and return this.data[this.firstIdx...s)
765 return result;
766 }
767 > arrays.ts
768 > /**
769 > * Consumes elements from the end of the queue as long as the predicate returns true.
770 > * If no elements were consumed, `null` is returned.
771 > * The result has the same order as the underlying array!
772 > */
773 > takeFromEndWhile(predicate: (value: T) => boolean): T[] | null {
774 // P(k) := this.firstIdx >= k && predicate(this.items[k])
775 // Find s := max { k | k <= this.lastIdx && !P(k) } and return this.data(s...this.lastIdx]
783 return result;
784 }
785 > arrays.ts
786 > peek(): T | undefined {
787 if (this.length === 0) {
788 return undefined;
790 return this.items[this.firstIdx];
791 }
792 > arrays.ts
793 > peekLast(): T | undefined {
794 if (this.length === 0) {
795 return undefined;
797 return this.items[this.lastIdx];
798 }
799 > arrays.ts
800 > dequeue(): T | undefined {
801 const result = this.items[this.firstIdx];
802 this.firstIdx++;
803 return result;
804 }
805 > arrays.ts
806 > removeLast(): T | undefined {
807 const result = this.items[this.lastIdx];
808 this.lastIdx--;
809 return result;
810 }
811 > arrays.ts
812 > takeCount(count: number): T[] {
813 const result = this.items.slice(this.firstIdx, this.firstIdx + count);
814 this.firstIdx += count;
815 return result;
816 }
817 > } arrays.ts
818 >
819 > /**
820 > * This class is faster than an iterator and array for lazy computed data.
821 > */
822 > export class CallbackIterable<T> {
823 > public static readonly empty = new CallbackIterable<never>(_callback => { });
824 >
825 > constructor(
826 > /**
827 > * Calls the callback for every item.
828 > * Stops when the callback returns false.
829 > */
830 > public readonly iterate: (callback: (item: T) => boolean) => void
831 > ) {
832 > }
833 >
834 > forEach(handler: (item: T) => void) {
835 this.iterate(item => { handler(item); return true; });
836 }
837 > arrays.ts
838 > toArray(): T[] {
839 const result: T[] = [];
840 this.iterate(item => { result.push(item); return true; });
841 return result;
842 }
843 > arrays.ts
844 > filter(predicate: (item: T) => boolean): CallbackIterable<T> {
845 return new CallbackIterable(cb => this.iterate(item => predicate(item) ? cb(item) : true));
846 }
847 > arrays.ts
848 > map<TResult>(mapFn: (item: T) => TResult): CallbackIterable<TResult> {
849 return new CallbackIterable<TResult>(cb => this.iterate(item => cb(mapFn(item))));
850 }
851 > arrays.ts
852 > some(predicate: (item: T) => boolean): boolean {
853 let result = false;
854 this.iterate(item => { result = predicate(item); return !result; });
855 return result;
856 }
857 > arrays.ts
858 > findFirst(predicate: (item: T) => boolean): T | undefined {
859 let result: T | undefined;
860 this.iterate(item => {
867 return result;
868 }
869 > arrays.ts
870 > findLast(predicate: (item: T) => boolean): T | undefined {
871 let result: T | undefined;
872 this.iterate(item => {
878 return result;
879 }
880 > arrays.ts
881 > findLastMaxBy(comparator: Comparator<T>): T | undefined {
882 let result: T | undefined;
883 let first = true;
891 return result;
892 }
893 > } arrays.ts
894 >
895 > /**
896 > * Represents a re-arrangement of items in an array.
897 > */
898 > export class Permutation {
899 > constructor(private readonly _indexMap: readonly number[]) { }
900 >
901 > /**
902 > * Returns a permutation that sorts the given array according to the given compare function.
903 > */
904 > public static createSortPermutation<T>(arr: readonly T[], compareFn: (a: T, b: T) => number): Permutation {
905 const sortIndices = Array.from(arr.keys()).sort((index1, index2) => compareFn(arr[index1], arr[index2]));
906 return new Permutation(sortIndices);
907 }
908 > arrays.ts
909 > /**
910 > * Returns a new array with the elements of the given array re-arranged according to this permutation.
911 > */
912 > apply<T>(arr: readonly T[]): T[] {
913 return arr.map((_, index) => arr[this._indexMap[index]]);
914 }
915 > arrays.ts
916 > /**
917 > * Returns a new permutation that undoes the re-arrangement of this permutation.
918 > */
919 > inverse(): Permutation {
920 const inverseIndexMap = this._indexMap.slice();
921 for (let i = 0; i < this._indexMap.length; i++) {
924 return new Permutation(inverseIndexMap);
925 }
926 > } arrays.ts
927 >
928 > /**
929 > * Asynchronous variant of `Array.find()`, returning the first element in
930 > * the array for which the predicate returns true.
931 > *
932 > * This implementation does not bail early and waits for all promises to
933 > * resolve before returning.
934 > */
935 export async function findAsync<T>(array: readonly T[], predicate: (element: T, index: number) => Promise<boolean>): Promise<T | undefined> {
936 const results = await Promise.all(array.map(
940 return results.find(r => r.ok)?.element;
941 }
942 > arrays.ts
943 > export function sum(array: readonly number[]): number {
944 return array.reduce((acc, value) => acc + value, 0);
945 }
946 > arrays.ts
947 > export function sumBy<T>(array: readonly T[], selector: (value: T) => number): number {
948 return array.reduce((acc, value) => acc + selector(value), 0);
949 }
src/vs/platform/agentPlugins/common/pluginParsers.ts 410 covered LOC · 47 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- pluginParsers.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { parse as parseJSONC } from '../../../base/common/json.js';
7 > import { cloneAndChange, equals as objectEquals } from '../../../base/common/objects.js';
8 > import { isAbsolute } from '../../../base/common/path.js';
9 > import { basename, extname, isEqualOrParent, joinPath, normalizePath, isEqual as isURLEquals, dirname } from '../../../base/common/resources.js';
10 > import { escapeRegExpCharacters } from '../../../base/common/strings.js';
11 > import { hasKey, Mutable } from '../../../base/common/types.js';
12 > import { URI } from '../../../base/common/uri.js';
13 > import { IFileService } from '../../files/common/files.js';
14 > import { parseFrontMatter } from '../../../base/common/yaml.js';
15 > import { IMcpRemoteServerConfiguration, IMcpServerConfiguration, IMcpStdioServerConfiguration, McpServerType } from '../../mcp/common/mcpPlatformTypes.js';
16 > import { CustomizationType, McpServerStatus, type AgentCustomization, type HookCustomization, type McpServerCustomization, type RuleCustomization, type SkillCustomization } from '../../agentHost/common/state/protocol/state.js';
17 > import { DEFAULT_MCP_APP } from '../../agentHost/common/state/protocol/mcpAppDefaults.js';
18 > import { customizationId } from '../../agentHost/common/state/sessionState.js';
19 > import { readAgentPluginManifest } from './agentPluginParser.js';
20 >
21 > // ---------------------------------------------------------------------------
22 > // Types
23 > // ---------------------------------------------------------------------------
24 >
25 > /** A single hook command to execute. Platform resolution happens at conversion time. */
26 > export interface IParsedHookCommand {
27 > /** Cross-platform default command. */
28 > readonly command?: string;
29 > /** Windows-specific command. */
30 > readonly windows?: string;
31 > /** Linux-specific command. */
32 > readonly linux?: string;
33 > /** macOS-specific command. */
34 > readonly osx?: string;
35 > /** Working directory. */
36 > readonly cwd?: URI;
37 > /** Environment variables. */
38 > readonly env?: Record<string, string>;
39 > /** Timeout in seconds. */
40 > readonly timeout?: number;
41 > /** URI of the file this hook was defined in. */
42 > readonly sourceUri?: URI;
43 > }
44 >
45 > export namespace IParsedHookCommand {
46 > export function isEquals(a: IParsedHookCommand | undefined, b: IParsedHookCommand | undefined): boolean {
47 if (a === b) {
48 return true;
60 && isURLEquals(a.sourceUri, b.sourceUri);
61 }
63 >
64 > /** A group of hooks for a single lifecycle event. */
65 > export interface IParsedHookGroup {
66 > /** Canonical hook type identifier (e.g. `'SessionStart'`, `'PreToolUse'`). */
67 > readonly type: string;
68 > /** The commands to execute for this hook type. */
69 > readonly commands: readonly IParsedHookCommand[];
70 > /** URI where this hook is defined. */
71 > readonly uri: URI;
72 > /** Original key as it appears in the hook file. */
73 > readonly originalId: string;
74 > /**
75 > * Protocol-level projection of this hook group as a child customization.
76 > * Multiple groups parsed from the same file share the same `customization.id`
77 > * so consumers can dedupe by id when collecting customizations.
78 > */
79 > readonly customization: HookCustomization;
80 > }
81 >
82 > export interface IMcpServerDefinition {
83 > readonly name: string;
84 > readonly configuration: IMcpServerConfiguration;
85 > readonly uri: URI;
86 > /** Protocol-level projection of this MCP server as a child customization. */
87 > readonly customization: McpServerCustomization;
88 > }
89 >
90 > /** A named resource (skill, agent, command, or instruction) within a plugin. */
91 > export interface INamedPluginResource {
92 > readonly uri: URI;
93 > readonly name: string;
94 > /**
95 > * Optional short description, populated for resources whose readers
96 > * parse it from the file's YAML frontmatter (e.g. agents).
97 > */
98 > readonly description?: string;
99 > }
100 >
101 > /** A parsed agent paired with its protocol-level child customization. */
102 > export interface IParsedAgent extends INamedPluginResource {
103 > readonly customization: AgentCustomization;
104 > }
105 >
106 > /** A parsed skill paired with its protocol-level child customization. */
107 > export interface IParsedSkill extends INamedPluginResource {
108 > readonly customization: SkillCustomization;
109 > }
110 >
111 > /** A parsed rule (instruction) paired with its protocol-level child customization. */
112 > export interface IParsedRule extends INamedPluginResource {
113 > readonly customization: RuleCustomization;
114 > }
115 >
116 > /** The result of parsing a single plugin directory. */
117 > export interface IParsedPlugin {
118 > readonly format: PluginFormat;
119 > readonly hooks: readonly IParsedHookGroup[];
120 > readonly mcpServers: readonly IMcpServerDefinition[];
121 > readonly skills: readonly IParsedSkill[];
122 > readonly agents: readonly IParsedAgent[];
123 > readonly instructions: readonly IParsedRule[];
124 > }
125 >
126 > // ---------------------------------------------------------------------------
127 > // Plugin format detection
128 > // ---------------------------------------------------------------------------
129 >
130 > export const enum PluginFormat {
131 > Copilot,
132 > Claude,
133 > OpenPlugin,
134 > AgentPlugin,
135 > }
136 >
137 > export interface IPluginFormatConfig {
138 > readonly format: PluginFormat;
139 > readonly manifestPath: string;
140 > readonly hookConfigPath: string;
141 > readonly componentPaths?: Readonly<Partial<Record<PluginComponent, string | false>>>;
142 > readonly requiresManifest?: boolean;
143 > readonly pluginRootTokens: readonly string[];
144 > readonly pluginRootEnvVars: readonly string[];
145 > /** Parses hooks from a JSON object using the format's conventions. */
146 > parseHooks(hookUri: URI, json: unknown, pluginUri: URI, workspaceRoot: URI | undefined, userHome: URI): IParsedHookGroup[];
147 > }
148 >
149 > export type PluginComponent = 'commands' | 'skills' | 'agents' | 'rules' | 'hooks' | 'mcpServers';
150 >
151 > const COPILOT_FORMAT: IPluginFormatConfig = {
152 > format: PluginFormat.Copilot,
153 > manifestPath: 'plugin.json',
154 > hookConfigPath: 'hooks.json',
155 > pluginRootTokens: ['${PLUGIN_ROOT}', '${CLAUDE_PLUGIN_ROOT}'],
156 > pluginRootEnvVars: ['PLUGIN_ROOT', 'CLAUDE_PLUGIN_ROOT'],
157 > parseHooks(hookUri, json, _pluginUri, workspaceRoot, userHome) {
158 return parseHooksJson(hookUri, json, workspaceRoot, userHome);
159 },
160 > }; pluginParsers.ts
161 >
162 > const CLAUDE_FORMAT: IPluginFormatConfig = {
163 > format: PluginFormat.Claude,
164 > manifestPath: '.claude-plugin/plugin.json',
165 > hookConfigPath: 'hooks/hooks.json',
166 > pluginRootTokens: ['${PLUGIN_ROOT}', '${CLAUDE_PLUGIN_ROOT}'],
167 > pluginRootEnvVars: ['PLUGIN_ROOT', 'CLAUDE_PLUGIN_ROOT'],
168 > parseHooks(hookUri, json, pluginUri, workspaceRoot, userHome) {
169 return interpolateHookPluginRoot(hookUri, json, pluginUri, workspaceRoot, userHome, '${CLAUDE_PLUGIN_ROOT}', 'CLAUDE_PLUGIN_ROOT');
170 },
171 > }; pluginParsers.ts
172 >
173 > const OPEN_PLUGIN_FORMAT: IPluginFormatConfig = {
174 > format: PluginFormat.OpenPlugin,
175 > manifestPath: '.plugin/plugin.json',
176 > hookConfigPath: 'hooks/hooks.json',
177 > pluginRootTokens: ['${PLUGIN_ROOT}', '${CLAUDE_PLUGIN_ROOT}'],
178 > pluginRootEnvVars: ['PLUGIN_ROOT', 'CLAUDE_PLUGIN_ROOT'],
179 > parseHooks(hookUri, json, pluginUri, workspaceRoot, userHome) {
180 return interpolateHookPluginRoot(hookUri, json, pluginUri, workspaceRoot, userHome, '${PLUGIN_ROOT}', 'PLUGIN_ROOT');
181 },
182 > }; pluginParsers.ts
183 >
184 > const AGENT_PLUGIN_FORMAT: IPluginFormatConfig = {
185 > format: PluginFormat.AgentPlugin,
186 > manifestPath: 'plugin.json',
187 > hookConfigPath: '',
188 > componentPaths: {
189 > commands: false,
190 > skills: 'skills',
191 > agents: false,
192 > rules: false,
193 > hooks: false,
194 > mcpServers: 'mcp.json',
195 > },
196 > requiresManifest: true,
197 > pluginRootTokens: [],
198 > pluginRootEnvVars: [],
199 > parseHooks() {
200 return [];
201 },
202 > }; pluginParsers.ts
203 >
204 export async function detectPluginFormat(pluginUri: URI, fileService: IFileService): Promise<IPluginFormatConfig> {
205 if (await readAgentPluginManifest(pluginUri, fileService)) {
217 return COPILOT_FORMAT;
218 }
220 export async function readPluginManifest(pluginUri: URI, format: IPluginFormatConfig, fileService: IFileService): Promise<Record<string, unknown> | undefined> {
221 if (format.format === PluginFormat.AgentPlugin) {
226 return json && typeof json === 'object' && !Array.isArray(json) ? json as Record<string, unknown> : undefined;
227 }
229 > export function getPluginManifestComponent(format: IPluginFormatConfig, component: PluginComponent, manifest: Record<string, unknown> | undefined): unknown {
230 return format.componentPaths && Object.hasOwn(format.componentPaths, component) ? undefined : manifest?.[component];
231 }
233 > export function resolvePluginComponentDirs(
234 pluginUri: URI,
235 format: IPluginFormatConfig,
252 );
253 }
255 > // ---------------------------------------------------------------------------
256 > // Child customization helpers
257 > // ---------------------------------------------------------------------------
258 >
259 > /**
260 > * Mints a child-customization id from a source uri plus an optional opaque
261 > * disambiguator. Used when multiple customizations are declared inline in
262 > * a single file (e.g. two MCP servers in one `.mcp.json`, or two hook
263 > * lifecycle groups in one hook file).
264 > *
265 > * Percent-encodes any pre-existing `#` in the URI before appending the
266 > * disambiguating fragment so the resulting id can never collide with a
267 > * URI that happens to already contain a matching fragment.
268 > */
269 function buildChildId(uri: URI, disambiguator?: string): string {
270 const base = customizationId(uri.toString());
274 return `${base.replace(/#/g, '%23')}#${disambiguator}`;
275 }
277 function makeAgentCustomization(resource: INamedPluginResource): AgentCustomization {
278 const uri = resource.uri.toString();
285 };
286 }
288 function makeSkillCustomization(resource: INamedPluginResource): SkillCustomization {
289 const uri = resource.uri.toString();
296 };
297 }
299 function makeRuleCustomization(resource: INamedPluginResource): RuleCustomization {
300 const uri = resource.uri.toString();
307 };
308 }
310 function makeHookCustomization(hookUri: URI): HookCustomization {
311 return {
316 };
317 }
319 > /**
320 > * Builds the protocol {@link McpServerCustomization} for an MCP server
321 > * declared at `definitionUri` (the manifest / settings / `.mcp.json` file
322 > * the server is defined in). The id is disambiguated by server `name` so
323 > * multiple servers declared in one file get distinct ids, and the entry
324 > * carries {@link DEFAULT_MCP_APP} so MCP App support is advertised
325 > * consistently with every other MCP customization.
326 > *
327 > * The seed state is {@link McpServerStatus.Stopped}: a declared-but-not-yet
328 > * connected server has not been started by any SDK, so it must not claim to
329 > * be {@link McpServerStatus.Starting}. The live state is enriched from the
330 > * SDK's reported status once a session materializes.
331 > */
332 > export function makeMcpServerCustomization(definitionUri: URI, name: string): McpServerCustomization {
333 return {
334 type: CustomizationType.McpServer,
341 };
342 }
344 > // ---------------------------------------------------------------------------
345 > // Component path config
346 > // ---------------------------------------------------------------------------
347 >
348 > export interface IComponentPathConfig {
349 > readonly paths: readonly string[];
350 > readonly exclusive: boolean;
351 > }
352 >
353 > const emptyComponentPathConfig: IComponentPathConfig = { paths: [], exclusive: false };
354 >
355 > /**
356 > * Parses a manifest component path field into a normalized config.
357 > * Supports `undefined`, `string`, `string[]`, and `{ paths: string[], exclusive?: boolean }`.
358 > */
359 > export function parseComponentPathConfig(raw: unknown): IComponentPathConfig {
360 if (raw === undefined || raw === null) {
361 return emptyComponentPathConfig;
389 return emptyComponentPathConfig;
390 }
392 > /**
393 > * Resolves the directories to scan for a given component type, combining
394 > * the default directory with any custom paths from the manifest config.
395 > * Paths that resolve outside the boundary are silently ignored.
396 > * @param boundaryUri The outermost directory that resolved paths must stay within. Defaults to {@link pluginUri}.
397 > */
398 > export function resolveComponentDirs(pluginUri: URI, defaultDir: string, config: IComponentPathConfig, boundaryUri?: URI): readonly URI[] {
399 const boundary = (boundaryUri && isEqualOrParent(pluginUri, boundaryUri)) ? boundaryUri : pluginUri;
400 const dirs: URI[] = [];
410 return dirs;
411 }
413 > // ---------------------------------------------------------------------------
414 > // MCP server helpers
415 > // ---------------------------------------------------------------------------
416 >
417 > /**
418 > * Extracts the MCP server map from a raw JSON value. Accepts both the
419 > * wrapped format `{ mcpServers: { … } }` and the flat format.
420 > */
421 > export function resolveMcpServersMap(raw: unknown): Record<string, unknown> | undefined {
422 if (!raw || typeof raw !== 'object' || Array.isArray(raw)) {
423 return undefined;
428 : obj;
429 }
431 > /**
432 > * Normalizes a raw JSON value into a typed MCP server configuration.
433 > */
434 > export function normalizeMcpServerConfiguration(rawConfig: unknown): IMcpServerConfiguration | undefined {
435 if (!rawConfig || typeof rawConfig !== 'object') {
436 return undefined;
477 return undefined;
478 }
480 > /**
481 > * Characters in a file path that require shell quoting to prevent
482 > * word splitting or interpretation by common shells.
483 > */
484 > const shellUnsafeChars = /[\s&|<>()^;!`"']/;
485 >
486 > /**
487 > * Replaces a plugin-root token in a shell command string with the
488 > * given fsPath, shell-quoting if the path contains special characters.
489 > */
490 > export function shellQuotePluginRootInCommand(command: string, fsPath: string, token: string) {
491 if (!command.includes(token)) {
492 return command;
511 });
512 }
514 > /**
515 > * Replaces plugin-root token references in MCP server definition string fields
516 > * with the plugin root filesystem path.
517 > */
518 > export function interpolateMcpPluginRoot(
519 def: IMcpServerDefinition,
520 fsPath: string,
562 return { name: def.name, configuration: interpolated, uri: def.uri, customization: def.customization };
563 }
565 > /**
566 > * Regex matching bare `${VAR_NAME}` references (uppercase only) that are NOT
567 > * using VS Code's `${env:VAR}` colon-delimited syntax.
568 > */
569 > const BARE_ENV_VAR_RE = /\$\{(?![A-Za-z]+:)([A-Z_][A-Z0-9_]*)\}/g;
570 >
571 > /**
572 > * Converts bare `${VAR}` environment-variable references to VS Code `${env:VAR}` syntax.
573 > */
574 > export function convertBareEnvVarsToVsCodeSyntax(
575 def: IMcpServerDefinition,
576 ): IMcpServerDefinition {
586 });
587 }
589 > // ---------------------------------------------------------------------------
590 > // Hook parsing helpers
591 > // ---------------------------------------------------------------------------
592 >
593 > /**
594 > * Maps known hook type identifiers from all formats (VS Code PascalCase,
595 > * Copilot CLI camelCase, Claude PascalCase) to canonical identifiers.
596 > */
597 > const HOOK_TYPE_MAP: Record<string, string> = {
598 > // PascalCase (VS Code / Claude)
599 > 'SessionStart': 'SessionStart',
600 > 'SessionEnd': 'SessionEnd',
601 > 'UserPromptSubmit': 'UserPromptSubmit',
602 > 'PreToolUse': 'PreToolUse',
603 > 'PostToolUse': 'PostToolUse',
604 > 'PreCompact': 'PreCompact',
605 > 'SubagentStart': 'SubagentStart',
606 > 'SubagentStop': 'SubagentStop',
607 > 'Stop': 'Stop',
608 > 'ErrorOccurred': 'ErrorOccurred',
609 > // camelCase (GitHub Copilot CLI)
610 > 'sessionStart': 'SessionStart',
611 > 'sessionEnd': 'SessionEnd',
612 > 'userPromptSubmitted': 'UserPromptSubmit',
613 > 'preToolUse': 'PreToolUse',
614 > 'postToolUse': 'PostToolUse',
615 > 'agentStop': 'Stop',
616 > 'subagentStop': 'SubagentStop',
617 > 'errorOccurred': 'ErrorOccurred',
618 > };
619 >
620 > /**
621 > * Normalizes a raw hook command object, validating structure and mapping
622 > * legacy `bash`/`powershell` fields to platform-specific overrides.
623 > */
624 function normalizeHookCommand(raw: Record<string, unknown>): IParsedHookCommand | undefined {
625 // Allow omitted type (Claude compatibility) — treat as 'command'
656 };
657 }
659 > /**
660 > * Resolves a raw hook command JSON object into a {@link IParsedHookCommand},
661 > * normalizing fields and resolving the working directory.
662 > */
663 function resolveHookCommand(raw: Record<string, unknown>, workspaceRoot: URI | undefined, userHome: URI): IParsedHookCommand | undefined {
664 const normalized = normalizeHookCommand(raw);
683 return { ...normalized, cwd: cwdUri };
684 }
686 > /**
687 > * Extracts hook commands from an item that may be a direct command object
688 > * or a nested structure with a `matcher` (Claude format).
689 > */
690 function extractHookCommands(item: unknown, workspaceRoot: URI | undefined, userHome: URI): IParsedHookCommand[] {
691 if (!item || typeof item !== 'object') {
717 return commands;
718 }
720 > /**
721 > * Parses hooks from a JSON object (any supported format).
722 > *
723 > * Handles Claude's `disableAllHooks` short-circuit, the `HOOK_TYPE_MAP`
724 > * canonicalization, and the nested `{ matcher, hooks: [...] }` command
725 > * form. Returns one {@link IParsedHookGroup} per recognized lifecycle
726 > * event; all groups parsed from the same file share a single
727 > * {@link IParsedHookGroup.customization} (keyed on `hookUri`), so callers
728 > * that only need the file-level customization can read it off any group.
729 > */
730 > export function parseHooksJson(
731 hookUri: URI,
732 json: unknown,
777 return result;
778 }
780 > /**
781 > * Applies plugin-root token interpolation to hook commands for
782 > * Claude and OpenPlugin formats.
783 > */
784 > export function interpolateHookPluginRoot(
785 hookUri: URI,
786 json: unknown,
834 return parseHooksJson(hookUri, cloneAndChange(json, replacer), workspaceRoot, userHome);
835 }
837 > // ---------------------------------------------------------------------------
838 > // Filesystem helpers
839 > // ---------------------------------------------------------------------------
840 >
841 export async function readJsonFile(uri: URI, fileService: IFileService): Promise<unknown | undefined> {
842 try {
847 }
848 }
850 export async function pathExists(resource: URI, fileService: IFileService): Promise<boolean> {
851 try {
856 }
857 }
859 > // ---------------------------------------------------------------------------
860 > // Component readers
861 > // ---------------------------------------------------------------------------
862 >
863 > const COMMAND_FILE_SUFFIX = '.md';
864 > const RULE_FILE_SUFFIX = '.mdc';
865 > const INSTRUCTION_FILE_SUFFIX = '.instructions.md';
866 >
867 export async function readSkills(
868 pluginRoot: URI,
931 return skills;
932 }
934 export async function readPluginSkills(pluginRoot: URI, dirs: readonly URI[], format: IPluginFormatConfig, fileService: IFileService): Promise<readonly INamedPluginResource[]> {
935 return readSkills(pluginRoot, dirs, fileService, format.format === PluginFormat.AgentPlugin
937 : undefined);
938 }
940 async function isResolvedWithin(root: URI, resource: URI, fileService: IFileService): Promise<boolean> {
941 try {
949 }
950 }
952 export async function readMarkdownComponents(dirs: readonly URI[], fileService: IFileService): Promise<readonly INamedPluginResource[]> {
953 const seen = new Set<string>();
989 return items;
990 }
992 function getInstructionFileName(resource: URI): string | undefined {
993 const fileName = basename(resource);
1001 return undefined;
1002 }
1004 > /**
1005 > * Reads rule/instruction files from plugin `rules` component directories.
1006 > *
1007 > * Open Plugins rules are conventionally `.mdc` files. We also accept
1008 > * `.instructions.md` for compatibility with VS Code-discovered instructions
1009 > * bundled as synthetic plugins.
1010 > */
1011 export async function readInstructionComponents(dirs: readonly URI[], fileService: IFileService): Promise<readonly INamedPluginResource[]> {
1012 const seen = new Set<string>();
1054 return items;
1055 }
1057 > /**
1058 > * Reads `.md` files in agent directories and enriches each entry with
1059 > * the optional `name` / `description` from YAML frontmatter. Falls back
1060 > * to the file-derived name when frontmatter is missing or unreadable.
1061 > */
1062 export async function readAgentComponents(dirs: readonly URI[], fileService: IFileService): Promise<readonly INamedPluginResource[]> {
1063 const files = await readMarkdownComponents(dirs, fileService);
1090 return result;
1091 }
1093 export async function parseAgentFile(uri: URI, fileService: IFileService): Promise<{ name: string; description?: string; userInvocable?: boolean }> {
1094 // Use regex to strip the trailing `.agent.md` or .md before parsing, so we can fall back to a cleaner name if frontmatter is missing or broken.
1105 }
1106 }
1108 export async function parseSkillFile(uri: URI, fileService: IFileService): Promise<{ name: string; description?: string; userInvokable?: boolean }> {
1109 try {
1118 }
1119 }
1121 export async function parseRuleFile(uri: URI, fileService: IFileService): Promise<{ name: string; description?: string; globs?: string[]; alwaysApply?: boolean }> {
1122 const nameFromFile = basename(uri).replace(/(\.instructions)?\.md$/i, '');
1133 }
1134 }
1136 async function readHooks(
1137 pluginUri: URI,
1152 return [];
1153 }
1155 async function readMcpServers(
1156 pluginUri: URI,
1173 return [...merged.values()].sort((a, b) => a.name.localeCompare(b.name));
1174 }
1176 export async function readPluginMcpServers(
1177 pluginUri: URI,
1182 return readMcpServers(pluginUri, paths, format, fileService);
1183 }
1185 > export function parseMcpServerDefinitionMap(
1186 definitionURI: URI,
1187 raw: unknown,
1219 return definitions;
1220 }
1222 > // ---------------------------------------------------------------------------
1223 > // Top-level parse function
1224 > // ---------------------------------------------------------------------------
1225 >
1226 > /**
1227 > * Parses a plugin directory to extract hooks, MCP servers, skills, agents,
1228 > * and instructions.
1229 > * This is the main entry point for the agent host to discover plugin contents.
1230 > */
1231 export async function parsePlugin(
1232 pluginUri: URI,
1292 };
1293 }
1295 > /** Pairs an agent {@link INamedPluginResource} with its protocol-level {@link AgentCustomization}. */
1296 > export function toParsedAgent(resource: INamedPluginResource): IParsedAgent {
1297 return { ...resource, customization: makeAgentCustomization(resource) };
1298 }
1300 > /** Pairs a skill {@link INamedPluginResource} with its protocol-level {@link SkillCustomization}. */
1301 > export function toParsedSkill(resource: INamedPluginResource): IParsedSkill {
1302 return { ...resource, customization: makeSkillCustomization(resource) };
1303 }
1305 function toParsedRule(resource: INamedPluginResource): IParsedRule {
1306 return { ...resource, customization: makeRuleCustomization(resource) };
src/vs/base/common/json.ts 399 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- json.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export const enum ScanError {
7 > None = 0,
8 > UnexpectedEndOfComment = 1,
9 > UnexpectedEndOfString = 2,
10 > UnexpectedEndOfNumber = 3,
11 > InvalidUnicode = 4,
12 > InvalidEscapeCharacter = 5,
13 > InvalidCharacter = 6
14 > }
15 >
16 > export const enum SyntaxKind {
17 > OpenBraceToken = 1,
18 > CloseBraceToken = 2,
19 > OpenBracketToken = 3,
20 > CloseBracketToken = 4,
21 > CommaToken = 5,
22 > ColonToken = 6,
23 > NullKeyword = 7,
24 > TrueKeyword = 8,
25 > FalseKeyword = 9,
26 > StringLiteral = 10,
27 > NumericLiteral = 11,
28 > LineCommentTrivia = 12,
29 > BlockCommentTrivia = 13,
30 > LineBreakTrivia = 14,
31 > Trivia = 15,
32 > Unknown = 16,
33 > EOF = 17
34 > }
35 >
36 > /**
37 > * The scanner object, representing a JSON scanner at a position in the input string.
38 > */
39 > export interface JSONScanner {
40 > /**
41 > * Sets the scan position to a new offset. A call to 'scan' is needed to get the first token.
42 > */
43 > setPosition(pos: number): void;
44 > /**
45 > * Read the next token. Returns the token code.
46 > */
47 > scan(): SyntaxKind;
48 > /**
49 > * Returns the current scan position, which is after the last read token.
50 > */
51 > getPosition(): number;
52 > /**
53 > * Returns the last read token.
54 > */
55 > getToken(): SyntaxKind;
56 > /**
57 > * Returns the last read token value. The value for strings is the decoded string content. For numbers its of type number, for boolean it's true or false.
58 > */
59 > getTokenValue(): string;
60 > /**
61 > * The start offset of the last read token.
62 > */
63 > getTokenOffset(): number;
64 > /**
65 > * The length of the last read token.
66 > */
67 > getTokenLength(): number;
68 > /**
69 > * An error code of the last scan.
70 > */
71 > getTokenError(): ScanError;
72 > }
73 >
74 >
75 >
76 > export interface ParseError {
77 > error: ParseErrorCode;
78 > offset: number;
79 > length: number;
80 > }
81 >
82 > export const enum ParseErrorCode {
83 > InvalidSymbol = 1,
84 > InvalidNumberFormat = 2,
85 > PropertyNameExpected = 3,
86 > ValueExpected = 4,
87 > ColonExpected = 5,
88 > CommaExpected = 6,
89 > CloseBraceExpected = 7,
90 > CloseBracketExpected = 8,
91 > EndOfFileExpected = 9,
92 > InvalidCommentToken = 10,
93 > UnexpectedEndOfComment = 11,
94 > UnexpectedEndOfString = 12,
95 > UnexpectedEndOfNumber = 13,
96 > InvalidUnicode = 14,
97 > InvalidEscapeCharacter = 15,
98 > InvalidCharacter = 16
99 > }
100 >
101 > export type NodeType = 'object' | 'array' | 'property' | 'string' | 'number' | 'boolean' | 'null';
102 >
103 > export interface Node {
104 > readonly type: NodeType;
105 > readonly value?: any;
106 > readonly offset: number;
107 > readonly length: number;
108 > readonly colonOffset?: number;
109 > readonly parent?: Node;
110 > readonly children?: Node[];
111 > }
112 >
113 > export type Segment = string | number;
114 > export type JSONPath = Segment[];
115 >
116 > export interface Location {
117 > /**
118 > * The previous property key or literal value (string, number, boolean or null) or undefined.
119 > */
120 > previousNode?: Node;
121 > /**
122 > * The path describing the location in the JSON document. The path consists of a sequence strings
123 > * representing an object property or numbers for array indices.
124 > */
125 > path: JSONPath;
126 > /**
127 > * Matches the locations path against a pattern consisting of strings (for properties) and numbers (for array indices).
128 > * '*' will match a single segment, of any property name or index.
129 > * '**' will match a sequence of segments or no segment, of any property name or index.
130 > */
131 > matches: (patterns: JSONPath) => boolean;
132 > /**
133 > * If set, the location's offset is at a property key.
134 > */
135 > isAtPropertyKey: boolean;
136 > }
137 >
138 > export interface ParseOptions {
139 > disallowComments?: boolean;
140 > allowTrailingComma?: boolean;
141 > allowEmptyContent?: boolean;
142 > }
143 >
144 > export namespace ParseOptions {
145 > export const DEFAULT = {
146 > allowTrailingComma: true
147 > };
148 > }
149 >
150 > export interface JSONVisitor {
151 > /**
152 > * Invoked when an open brace is encountered and an object is started. The offset and length represent the location of the open brace.
153 > */
154 > onObjectBegin?: (offset: number, length: number) => void;
155 >
156 > /**
157 > * Invoked when a property is encountered. The offset and length represent the location of the property name.
158 > */
159 > onObjectProperty?: (property: string, offset: number, length: number) => void;
160 >
161 > /**
162 > * Invoked when a closing brace is encountered and an object is completed. The offset and length represent the location of the closing brace.
163 > */
164 > onObjectEnd?: (offset: number, length: number) => void;
165 >
166 > /**
167 > * Invoked when an open bracket is encountered. The offset and length represent the location of the open bracket.
168 > */
169 > onArrayBegin?: (offset: number, length: number) => void;
170 >
171 > /**
172 > * Invoked when a closing bracket is encountered. The offset and length represent the location of the closing bracket.
173 > */
174 > onArrayEnd?: (offset: number, length: number) => void;
175 >
176 > /**
177 > * Invoked when a literal value is encountered. The offset and length represent the location of the literal value.
178 > */
179 > onLiteralValue?: (value: any, offset: number, length: number) => void;
180 >
181 > /**
182 > * Invoked when a comma or colon separator is encountered. The offset and length represent the location of the separator.
183 > */
184 > onSeparator?: (character: string, offset: number, length: number) => void;
185 >
186 > /**
187 > * When comments are allowed, invoked when a line or block comment is encountered. The offset and length represent the location of the comment.
188 > */
189 > onComment?: (offset: number, length: number) => void;
190 >
191 > /**
192 > * Invoked on an error.
193 > */
194 > onError?: (error: ParseErrorCode, offset: number, length: number) => void;
195 > }
196 >
197 > /**
198 > * Creates a JSON scanner on the given text.
199 > * If ignoreTrivia is set, whitespaces or comments are ignored.
200 > */
201 > export function createScanner(text: string, ignoreTrivia: boolean = false): JSONScanner {
202
203 let pos = 0;
558 };
559 }
560 > json.ts
561 function isWhitespace(ch: number): boolean {
562 return ch === CharacterCodes.space || ch === CharacterCodes.tab || ch === CharacterCodes.verticalTab || ch === CharacterCodes.formFeed ||
564 ch === CharacterCodes.narrowNoBreakSpace || ch === CharacterCodes.mathematicalSpace || ch === CharacterCodes.ideographicSpace || ch === CharacterCodes.byteOrderMark;
565 }
566 > json.ts
567 function isLineBreak(ch: number): boolean {
568 return ch === CharacterCodes.lineFeed || ch === CharacterCodes.carriageReturn || ch === CharacterCodes.lineSeparator || ch === CharacterCodes.paragraphSeparator;
569 }
570 > json.ts
571 function isDigit(ch: number): boolean {
572 return ch >= CharacterCodes._0 && ch <= CharacterCodes._9;
573 }
574 > json.ts
575 > const enum CharacterCodes {
576 > nullCharacter = 0,
577 > maxAsciiCharacter = 0x7F,
578 >
579 > lineFeed = 0x0A, // \n
580 > carriageReturn = 0x0D, // \r
581 > lineSeparator = 0x2028,
582 > paragraphSeparator = 0x2029,
583 >
584 > // REVIEW: do we need to support this? The scanner doesn't, but our IText does. This seems
585 > // like an odd disparity? (Or maybe it's completely fine for them to be different).
586 > nextLine = 0x0085,
587 >
588 > // Unicode 3.0 space characters
589 > space = 0x0020, // " "
590 > nonBreakingSpace = 0x00A0, //
591 > enQuad = 0x2000,
592 > emQuad = 0x2001,
593 > enSpace = 0x2002,
594 > emSpace = 0x2003,
595 > threePerEmSpace = 0x2004,
596 > fourPerEmSpace = 0x2005,
597 > sixPerEmSpace = 0x2006,
598 > figureSpace = 0x2007,
599 > punctuationSpace = 0x2008,
600 > thinSpace = 0x2009,
601 > hairSpace = 0x200A,
602 > zeroWidthSpace = 0x200B,
603 > narrowNoBreakSpace = 0x202F,
604 > ideographicSpace = 0x3000,
605 > mathematicalSpace = 0x205F,
606 > ogham = 0x1680,
607 >
608 > _ = 0x5F,
609 > $ = 0x24,
610 >
611 > _0 = 0x30,
612 > _1 = 0x31,
613 > _2 = 0x32,
614 > _3 = 0x33,
615 > _4 = 0x34,
616 > _5 = 0x35,
617 > _6 = 0x36,
618 > _7 = 0x37,
619 > _8 = 0x38,
620 > _9 = 0x39,
621 >
622 > a = 0x61,
623 > b = 0x62,
624 > c = 0x63,
625 > d = 0x64,
626 > e = 0x65,
627 > f = 0x66,
628 > g = 0x67,
629 > h = 0x68,
630 > i = 0x69,
631 > j = 0x6A,
632 > k = 0x6B,
633 > l = 0x6C,
634 > m = 0x6D,
635 > n = 0x6E,
636 > o = 0x6F,
637 > p = 0x70,
638 > q = 0x71,
639 > r = 0x72,
640 > s = 0x73,
641 > t = 0x74,
642 > u = 0x75,
643 > v = 0x76,
644 > w = 0x77,
645 > x = 0x78,
646 > y = 0x79,
647 > z = 0x7A,
648 >
649 > A = 0x41,
650 > B = 0x42,
651 > C = 0x43,
652 > D = 0x44,
653 > E = 0x45,
654 > F = 0x46,
655 > G = 0x47,
656 > H = 0x48,
657 > I = 0x49,
658 > J = 0x4A,
659 > K = 0x4B,
660 > L = 0x4C,
661 > M = 0x4D,
662 > N = 0x4E,
663 > O = 0x4F,
664 > P = 0x50,
665 > Q = 0x51,
666 > R = 0x52,
667 > S = 0x53,
668 > T = 0x54,
669 > U = 0x55,
670 > V = 0x56,
671 > W = 0x57,
672 > X = 0x58,
673 > Y = 0x59,
674 > Z = 0x5A,
675 >
676 > ampersand = 0x26, // &
677 > asterisk = 0x2A, // *
678 > at = 0x40, // @
679 > backslash = 0x5C, // \
680 > bar = 0x7C, // |
681 > caret = 0x5E, // ^
682 > closeBrace = 0x7D, // }
683 > closeBracket = 0x5D, // ]
684 > closeParen = 0x29, // )
685 > colon = 0x3A, // :
686 > comma = 0x2C, // ,
687 > dot = 0x2E, // .
688 > doubleQuote = 0x22, // "
689 > equals = 0x3D, // =
690 > exclamation = 0x21, // !
691 > greaterThan = 0x3E, // >
692 > lessThan = 0x3C, // <
693 > minus = 0x2D, // -
694 > openBrace = 0x7B, // {
695 > openBracket = 0x5B, // [
696 > openParen = 0x28, // (
697 > percent = 0x25, // %
698 > plus = 0x2B, // +
699 > question = 0x3F, // ?
700 > semicolon = 0x3B, // ;
701 > singleQuote = 0x27, // '
702 > slash = 0x2F, // /
703 > tilde = 0x7E, // ~
704 >
705 > backspace = 0x08, // \b
706 > formFeed = 0x0C, // \f
707 > byteOrderMark = 0xFEFF,
708 > tab = 0x09, // \t
709 > verticalTab = 0x0B, // \v
710 > }
711 >
712 > interface NodeImpl extends Node {
713 > type: NodeType;
714 > value?: any;
715 > offset: number;
716 > length: number;
717 > colonOffset?: number;
718 > parent?: NodeImpl;
719 > children?: NodeImpl[];
720 > }
721 >
722 > /**
723 > * For a given offset, evaluate the location in the JSON document. Each segment in the location path is either a property name or an array index.
724 > */
725 > export function getLocation(text: string, position: number): Location {
726 const segments: Segment[] = []; // strings or numbers
727 const earlyReturnException = new Object();
838 };
839 }
840 > json.ts
841 >
842 > /**
843 > * Parses the given text and returns the object the JSON content represents. On invalid input, the parser tries to be as fault tolerant as possible, but still return a result.
844 > * Therefore always check the errors list to find out if the input was valid.
845 > */
846 > export function parse(text: string, errors: ParseError[] = [], options: ParseOptions = ParseOptions.DEFAULT): any {
847 let currentProperty: string | null = null;
848 let currentParent: any = [];
889 return currentParent[0];
890 }
891 > json.ts
892 >
893 > /**
894 > * Parses the given text and returns a tree representation the JSON content. On invalid input, the parser tries to be as fault tolerant as possible, but still return a result.
895 > */
896 > export function parseTree(text: string, errors: ParseError[] = [], options: ParseOptions = ParseOptions.DEFAULT): Node {
897 let currentParent: NodeImpl = { type: 'array', offset: -1, length: -1, children: [], parent: undefined }; // artificial root
898
955 return result;
956 }
957 > json.ts
958 > /**
959 > * Finds the node at the given path in a JSON DOM.
960 > */
961 > export function findNodeAtLocation(root: Node, path: JSONPath): Node | undefined {
962 if (!root) {
963 return undefined;
990 return node;
991 }
992 > json.ts
993 > /**
994 > * Gets the JSON path of the given JSON DOM node
995 > */
996 > export function getNodePath(node: Node): JSONPath {
997 if (!node.parent || !node.parent.children) {
998 return [];
1010 return path;
1011 }
1012 > json.ts
1013 > /**
1014 > * Evaluates the JavaScript object of the given JSON DOM node
1015 > */
1016 > export function getNodeValue(node: Node): any {
1017 switch (node.type) {
1018 case 'array':
1038
1039 }
1040 > json.ts
1041 > export function contains(node: Node, offset: number, includeRightBound = false): boolean {
1042 return (offset >= node.offset && offset < (node.offset + node.length)) || includeRightBound && (offset === (node.offset + node.length));
1043 }
1044 > json.ts
1045 > /**
1046 > * Finds the most inner node at the given offset. If includeRightBound is set, also finds nodes that end at the given offset.
1047 > */
1048 > export function findNodeAtOffset(node: Node, offset: number, includeRightBound = false): Node | undefined {
1049 if (contains(node, offset, includeRightBound)) {
1050 const children = node.children;
1062 return undefined;
1063 }
1064 > json.ts
1065 >
1066 > /**
1067 > * Parses the given text and invokes the visitor functions for each object, array and literal reached.
1068 > */
1069 > export function visit(text: string, visitor: JSONVisitor, options: ParseOptions = ParseOptions.DEFAULT): any {
1070
1071 const _scanner = createScanner(text, false);
1308 return true;
1309 }
1310 > json.ts
1311 > export function getNodeType(value: unknown): NodeType {
1312 switch (typeof value) {
1313 case 'boolean': return 'boolean';
src/vs/workbench/contrib/chat/common/promptSyntax/hookSchema.ts 385 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hookSchema.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IJSONSchema } from '../../../../../base/common/jsonSchema.js';
7 > import * as nls from '../../../../../nls.js';
8 > import { URI } from '../../../../../base/common/uri.js';
9 > import { joinPath } from '../../../../../base/common/resources.js';
10 > import { isAbsolute } from '../../../../../base/common/path.js';
11 > import { untildify } from '../../../../../base/common/labels.js';
12 > import { OperatingSystem } from '../../../../../base/common/platform.js';
13 > import { IParsedHookCommand } from '../../../../../platform/agentPlugins/common/pluginParsers.js';
14 > import { HookType, HOOKS_BY_TARGET, HOOK_METADATA } from './hookTypes.js';
15 > import { Target } from './promptTypes.js';
16 > import { IValue, IMapValue } from './promptFileParser.js';
17 >
18 > /**
19 > * A single hook command configuration.
20 > * Extends the platform-layer {@link IParsedHookCommand} with editor-specific
21 > * metadata used for UI display and field highlighting.
22 > */
23 > export interface IHookCommand extends IParsedHookCommand {
24 > readonly type: 'command';
25 > /** Original JSON field name that provided the windows command. */
26 > readonly windowsSource?: 'windows' | 'powershell';
27 > /** Original JSON field name that provided the linux command. */
28 > readonly linuxSource?: 'linux' | 'bash';
29 > /** Original JSON field name that provided the osx command. */
30 > readonly osxSource?: 'osx' | 'bash';
31 > }
32 >
33 > /**
34 > * Collected hooks for a chat request, organized by hook type.
35 > * This is passed to the extension host so it knows what hooks are available.
36 > */
37 > export type ChatRequestHooks = {
38 > readonly [K in HookType]?: readonly IParsedHookCommand[];
39 > };
40 >
41 > export namespace ChatRequestHooks {
42 > export function isEquals(a: ChatRequestHooks | undefined, b: ChatRequestHooks | undefined): boolean {
43 if (a === b) {
44 return true;
63 return true;
64 }
65 > } hookSchema.ts
66 >
67 > /**
68 > * Merges two sets of hooks by concatenating the command arrays for each hook type.
69 > * Additional hooks are appended after the base hooks.
70 > */
71 > export function mergeHooks(base: ChatRequestHooks | undefined, additional: ChatRequestHooks): ChatRequestHooks {
72 if (!base) {
73 return additional;
84 return result as ChatRequestHooks;
85 }
87 > /**
88 > * Descriptions for hook command fields, used by both the JSON schema and the hover provider.
89 > */
90 > export const HOOK_COMMAND_FIELD_DESCRIPTIONS: Record<string, string> = {
91 > type: nls.localize('hook.type', 'Must be "command".'),
92 > command: nls.localize('hook.command', 'The command to execute. This is the default cross-platform command.'),
93 > windows: nls.localize('hook.windows', 'Windows-specific command. If specified and running on Windows, this overrides the "command" field.'),
94 > linux: nls.localize('hook.linux', 'Linux-specific command. If specified and running on Linux, this overrides the "command" field.'),
95 > osx: nls.localize('hook.osx', 'macOS-specific command. If specified and running on macOS, this overrides the "command" field.'),
96 > bash: nls.localize('hook.bash', 'Bash command for Linux and macOS.'),
97 > powershell: nls.localize('hook.powershell', 'PowerShell command for Windows.'),
98 > cwd: nls.localize('hook.cwd', 'Working directory for the script (relative to repository root).'),
99 > env: nls.localize('hook.env', 'Additional environment variables that are merged with the existing environment.'),
100 > timeout: nls.localize('hook.timeout', 'Maximum execution time in seconds (default: 30).'),
101 > timeoutSec: nls.localize('hook.timeoutSec', 'Maximum execution time in seconds (default: 10).'),
102 > };
103 >
104 > /**
105 > * JSON Schema for GitHub Copilot hook configuration files.
106 > * Hooks enable executing custom shell commands at strategic points in an agent's workflow.
107 > */
108 > const vscodeHookCommandSchema: IJSONSchema = {
109 > type: 'object',
110 > additionalProperties: true,
111 > required: ['type'],
112 > anyOf: [
113 > { required: ['command'] },
114 > { required: ['windows'] },
115 > { required: ['linux'] },
116 > { required: ['osx'] },
117 > { required: ['bash'] },
118 > { required: ['powershell'] }
119 > ],
120 > errorMessage: nls.localize('hook.commandRequired', 'At least one of "command", "windows", "linux", or "osx" must be specified.'),
121 > properties: {
122 > type: {
123 > type: 'string',
124 > enum: ['command'],
125 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.type
126 > },
127 > command: {
128 > type: 'string',
129 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.command
130 > },
131 > windows: {
132 > type: 'string',
133 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.windows
134 > },
135 > linux: {
136 > type: 'string',
137 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.linux
138 > },
139 > osx: {
140 > type: 'string',
141 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.osx
142 > },
143 > cwd: {
144 > type: 'string',
145 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.cwd
146 > },
147 > env: {
148 > type: 'object',
149 > additionalProperties: { type: 'string' },
150 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.env
151 > },
152 > timeout: {
153 > type: 'number',
154 > default: 30,
155 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.timeout
156 > }
157 > }
158 > };
159 >
160 > const hookArraySchema: IJSONSchema = {
161 > type: 'array',
162 > items: vscodeHookCommandSchema
163 > };
164 >
165 > /**
166 > * Builds JSON Schema hook properties for a given target by looking up
167 > * the hook keys from HOOKS_BY_TARGET and descriptions from HOOK_METADATA.
168 > */
169 > function buildHookProperties(target: Target, arraySchema: IJSONSchema): Record<string, IJSONSchema> {
170 > return Object.fromEntries(
171 > Object.entries(HOOKS_BY_TARGET[target]).map(([key, hookType]) => [
172 > key,
173 > { ...arraySchema, description: HOOK_METADATA[hookType]?.description }
174 > ])
175 > );
176 > }
177 >
178 > /**
179 > * Hook properties for the VS Code format.
180 > */
181 > const vscodeHookProperties: Record<string, IJSONSchema> = buildHookProperties(Target.VSCode, hookArraySchema);
182 >
183 > /**
184 > * Hook command schema for the Copilot CLI format.
185 > * Adds `bash`, `powershell`, and `timeoutSec` fields alongside the standard ones.
186 > */
187 > const copilotCliHookCommandSchema: IJSONSchema = {
188 > type: 'object',
189 > additionalProperties: true,
190 > required: ['type'],
191 > anyOf: [
192 > { required: ['bash'] },
193 > { required: ['powershell'] }
194 > ],
195 > errorMessage: nls.localize('hook.cliCommandRequired', 'At least one of "bash" or "powershell" must be specified.'),
196 > properties: {
197 > type: {
198 > type: 'string',
199 > enum: ['command'],
200 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.type
201 > },
202 > bash: {
203 > type: 'string',
204 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.bash
205 > },
206 > powershell: {
207 > type: 'string',
208 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.powershell
209 > },
210 > cwd: {
211 > type: 'string',
212 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.cwd
213 > },
214 > env: {
215 > type: 'object',
216 > additionalProperties: { type: 'string' },
217 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.env
218 > },
219 > timeoutSec: {
220 > type: 'number',
221 > default: 10,
222 > description: HOOK_COMMAND_FIELD_DESCRIPTIONS.timeoutSec
223 > }
224 > }
225 > };
226 >
227 > const copilotCliHookArraySchema: IJSONSchema = {
228 > type: 'array',
229 > items: copilotCliHookCommandSchema
230 > };
231 >
232 > /**
233 > * Hook properties for the Copilot CLI format.
234 > */
235 > const copilotCliHookProperties: Record<string, IJSONSchema> = buildHookProperties(Target.GitHubCopilot, copilotCliHookArraySchema);
236 >
237 > export const hookFileSchema: IJSONSchema = {
238 > $schema: 'http://json-schema.org/draft-07/schema#',
239 > type: 'object',
240 > description: nls.localize('hookFile.description', 'GitHub Copilot hook configuration file. Hooks enable executing custom shell commands at strategic points in an agent\'s workflow.'),
241 > additionalProperties: true,
242 > required: ['hooks'],
243 > properties: {
244 > hooks: {
245 > type: 'object',
246 > description: nls.localize('hookFile.hooks', 'Hook definitions organized by type.'),
247 > additionalProperties: true,
248 > }
249 > },
250 > // Conditionally apply PascalCase or camelCase hook properties based on
251 > // whether the file uses the Copilot CLI format (detected by the "version" field).
252 > if: {
253 > required: ['version'],
254 > properties: {
255 > version: { type: 'number' }
256 > }
257 > },
258 > then: {
259 > // Copilot CLI format: camelCase hook names, bash/powershell/timeoutSec fields
260 > properties: {
261 > version: {
262 > type: 'number',
263 > description: nls.localize('hookFile.version', 'Hook configuration format version.'),
264 > },
265 > hooks: {
266 > properties: copilotCliHookProperties
267 > }
268 > }
269 > },
270 > else: {
271 > // VS Code / PascalCase format
272 > properties: {
273 > hooks: {
274 > properties: vscodeHookProperties
275 > }
276 > }
277 > },
278 > defaultSnippets: [
279 > {
280 > label: nls.localize('hookFile.snippet.basic', 'Basic hook configuration'),
281 > description: nls.localize('hookFile.snippet.basic.description', 'A basic hook configuration with common hooks'),
282 > body: {
283 > hooks: {
284 > SessionStart: [
285 > {
286 > type: 'command',
287 > command: '${1:echo "Session started" >> session.log}',
288 > }
289 > ],
290 > PreToolUse: [
291 > {
292 > type: 'command',
293 > command: '${2:./scripts/validate.sh}',
294 > timeout: 15
295 > }
296 > ]
297 > }
298 > }
299 > }
300 > ]
301 > };
302 >
303 > /**
304 > * URI for the hook schema registration.
305 > */
306 > export const HOOK_SCHEMA_URI = 'vscode://schemas/hooks';
307 >
308 > /**
309 > * Normalizes a raw hook type identifier to the canonical HookType enum value.
310 > * Only matches exact enum values. For tool-specific naming conventions (e.g., Claude, Copilot CLI),
311 > * use the corresponding compat module's resolver function.
312 > */
313 > export function toHookType(rawHookTypeId: string): HookType | undefined {
314 if (Object.values(HookType).includes(rawHookTypeId as HookType)) {
315 return rawHookTypeId as HookType;
317 return undefined;
318 }
320 > /**
321 > * Normalizes a raw hook command object, validating structure.
322 > * Maps legacy bash/powershell fields to platform-specific overrides:
323 > * - bash -> linux + osx
324 > * - powershell -> windows
325 > * This is an internal helper - use resolveHookCommand for the full resolution.
326 > */
327 function normalizeHookCommand(raw: Record<string, unknown>): { command?: string; windows?: string; linux?: string; osx?: string; windowsSource?: 'windows' | 'powershell'; linuxSource?: 'linux' | 'bash'; osxSource?: 'osx' | 'bash'; cwd?: string; env?: Record<string, string>; timeout?: number } | undefined {
328 if (raw.type !== 'command') {
364 };
365 }
367 > /**
368 > * Gets a label for the given platform.
369 > */
370 > export function getPlatformLabel(os: OperatingSystem): string {
371 if (os === OperatingSystem.Windows) {
372 return 'Windows';
378 return '';
379 }
381 > /**
382 > * Resolves the effective command for the given platform.
383 > * This applies OS-specific overrides (windows, linux, osx) to get the actual command that will be executed.
384 > * Similar to how launch.json handles platform-specific configurations in debugAdapter.ts.
385 > */
386 > export function resolveEffectiveCommand(hook: IParsedHookCommand, os: OperatingSystem): string | undefined {
387 // Select the platform-specific override based on the OS
388 if (os === OperatingSystem.Windows && hook.windows) {
397 return hook.command;
398 }
400 > /**
401 > * Checks if the hook is using a platform-specific command override.
402 > */
403 > export function isUsingPlatformOverride(hook: IParsedHookCommand, os: OperatingSystem): boolean {
404 if (os === OperatingSystem.Windows && hook.windows) {
405 return true;
411 return false;
412 }
414 > /**
415 > * Gets the source shell type for the effective command on the given platform.
416 > * Returns 'powershell' if the Windows command came from a powershell field,
417 > * 'bash' if the Linux/macOS command came from a bash field,
418 > * or undefined for default shell handling.
419 > */
420 > export function getEffectiveCommandSource(hook: IHookCommand, os: OperatingSystem): 'powershell' | 'bash' | undefined {
421 if (os === OperatingSystem.Windows && hook.windows && hook.windowsSource === 'powershell') {
422 return 'powershell';
428 return undefined;
429 }
431 > /**
432 > * Gets the original JSON field key name for the given platform's command.
433 > * Returns the actual field name from the JSON (e.g., 'bash' instead of 'osx' if bash was used).
434 > * This is used for editor focus to highlight the correct field.
435 > */
436 > export function getEffectiveCommandFieldKey(hook: IHookCommand | IParsedHookCommand, os: OperatingSystem): string {
437 const h = hook as Partial<IHookCommand>;
438 if (os === OperatingSystem.Windows && hook.windows) {
445 return 'command';
446 }
448 > /**
449 > * Formats a hook command for display.
450 > * Resolves OS-specific overrides to show the effective command for the given platform.
451 > * If using a platform-specific override, includes the platform as a prefix badge.
452 > */
453 > export function formatHookCommandLabel(hook: IParsedHookCommand, os: OperatingSystem): string {
454 const command = resolveEffectiveCommand(hook, os);
455 if (!command) {
458 return command;
459 }
461 > /**
462 > * Resolves a raw hook command object to the canonical IHookCommand format.
463 > * Normalizes the command and resolves the cwd path relative to the workspace root.
464 > * @param raw The raw hook command object from JSON
465 > * @param workspaceRootUri The workspace root URI to resolve relative cwd paths against
466 > * @param userHome The user's home directory path for tilde expansion
467 > */
468 > export function resolveHookCommand(raw: Record<string, unknown>, workspaceRootUri: URI | undefined, userHome: string): IHookCommand | undefined {
469 const normalized = normalizeHookCommand(raw);
470 if (!normalized) {
501 };
502 }
504 > /**
505 > * Helper to extract hook commands from an item that could be:
506 > * 1. A direct command object: { type: 'command', command: '...' }
507 > * 2. A nested structure with matcher (Claude style): { matcher: '...', hooks: [{ type: 'command', command: '...' }] }
508 > *
509 > * This allows Copilot format to handle Claude-style entries if pasted.
510 > * Also handles Claude's leniency where 'type' field can be omitted.
511 > */
512 > export function extractHookCommandsFromItem(
513 item: unknown,
514 workspaceRootUri: URI | undefined,
546 return commands;
547 }
549 > /**
550 > * Normalizes a hook command object for resolving.
551 > * Claude format allows omitting the 'type' field, treating it as 'command'.
552 > * This ensures compatibility when Claude-style hooks are pasted into Copilot format.
553 > */
554 function normalizeForResolve(raw: Record<string, unknown>): Record<string, unknown> {
555 // If type is missing or already 'command', ensure it's set to 'command'
559 return raw;
560 }
562 > /**
563 > * Converts an {@link IValue} YAML AST node into a plain JavaScript value
564 > * (string, array, or object) suitable for passing to hook parsing helpers.
565 > */
566 function yamlValueToPlain(value: IValue): unknown {
567 switch (value.type) {
579 }
580 }
582 > /**
583 > * Parses hooks from a subagent's YAML frontmatter `hooks` attribute.
584 > *
585 > * Supports two formats for hook entries:
586 > *
587 > * 1. **Direct command** (our format, without matcher):
588 > * ```yaml
589 > * hooks:
590 > * PreToolUse:
591 > * - type: command
592 > * command: "./scripts/validate.sh"
593 > * ```
594 > *
595 > * 2. **Nested with matcher** (Claude Code format):
596 > * ```yaml
597 > * hooks:
598 > * PreToolUse:
599 > * - matcher: "Bash"
600 > * hooks:
601 > * - type: command
602 > * command: "./scripts/validate.sh"
603 > * ```
604 > *
605 > * @param hooksMap The raw YAML map value from the `hooks` frontmatter attribute.
606 > * @param workspaceRootUri Workspace root for resolving relative `cwd` paths.
607 > * @param userHome User home directory path for tilde expansion.
608 > * @param target The agent's target, used to resolve hook type names correctly.
609 > * @returns Resolved hooks organized by hook type, ready for use in {@link ChatRequestHooks}.
610 > */
611 > export function parseSubagentHooksFromYaml(
612 hooksMap: IMapValue,
613 workspaceRootUri: URI | undefined,
src/vs/platform/policy/common/copilotManagedSettings.ts 373 covered LOC · 20 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- copilotManagedSettings.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { IPolicyData } from '../../../base/common/defaultAccount.js';
8 > import { IExtraKnownMarketplaceEntry, extraKnownMarketplacesToConfigDict } from '../../../base/common/managedSettings.js';
9 > import { IManagedSettingPolicyDefinition, IManagedSettingsPolicyDefinitions, ManagedSettingValue, ManagedSettingsData } from '../../../base/common/policy.js';
10 > import { IStringDictionary } from '../../../base/common/collections.js';
11 > import { isEmptyObject, isObject, isString } from '../../../base/common/types.js';
12 > import { createDecorator } from '../../instantiation/common/instantiation.js';
13 > import { PolicyDefinition } from './policy.js';
14 >
15 > export type { ManagedSettingsData } from '../../../base/common/policy.js';
16 >
17 > export type RawManagedSettingsData = Readonly<Record<string, unknown>>;
18 >
19 > /** Windows registry root for GitHub Copilot policies. */
20 > export const GITHUB_COPILOT_WIN32_REGISTRY_PATH = 'SOFTWARE\\Policies\\GitHubCopilot';
21 >
22 > /** Windows product name passed to the native policy watcher. */
23 > export const GITHUB_COPILOT_WIN32_POLICY_NAME = 'GitHubCopilot';
24 >
25 > /** macOS CFPreferences application ID for GitHub Copilot managed preferences. */
26 > export const GITHUB_COPILOT_MACOS_BUNDLE_ID = 'com.github.copilot';
27 >
28 > /** MDM key for the V0 managed setting. */
29 > export const COPILOT_DISABLE_BYPASS_PERMISSIONS_MODE_KEY = 'permissions.disableBypassPermissionsMode';
30 >
31 > /** Managed-settings key for enterprise plugin enablement (carried as a JSON-encoded `{ [pluginId]: boolean }`). */
32 > export const COPILOT_ENABLED_PLUGINS_KEY = 'enabledPlugins';
33 >
34 > /** Managed-settings key for enterprise marketplaces (carried as a JSON-encoded `{ [name]: url-or-shorthand }`). */
35 > export const COPILOT_EXTRA_MARKETPLACES_KEY = 'extraKnownMarketplaces';
36 >
37 > /** Managed-settings key for the strict-marketplace allowlist (carried as a JSON-encoded array of source entries; absent = no restrictions, `[]` = lockdown). */
38 > export const COPILOT_STRICT_MARKETPLACES_KEY = 'strictKnownMarketplaces';
39 >
40 > /** Managed-settings key for the per-server MCP allowlist (carried as a JSON-encoded array of matcher entries; absent = no allow restriction, `[]` = only servers matching an entry, i.e. block all). */
41 > export const COPILOT_ALLOWED_MCP_SERVERS_KEY = 'allowedMcpServers';
42 >
43 > /** Managed-settings key for the per-server MCP denylist (carried as a JSON-encoded array of matcher entries; deny always takes precedence over allow). */
44 > export const COPILOT_DENIED_MCP_SERVERS_KEY = 'deniedMcpServers';
45 >
46 > /**
47 > * Managed-settings key for the default chat model (carried as a plain string: `auto`, a model
48 > * family name, or a full model id). Nested under `permissions` in the managed-settings schema
49 > * (alongside {@link COPILOT_DISABLE_BYPASS_PERMISSIONS_MODE_KEY}), so it flattens to the dot-path
50 > * `permissions.model` in the normalized bag — the key policy `value()` callbacks must read.
51 > */
52 > export const COPILOT_MODEL_KEY = 'permissions.model';
53 >
54 > /**
55 > * Enterprise OTel managed-settings keys. These are the scalar leaves of the canonical
56 > * `telemetry` block from the cross-client managed-settings schema (see the CLI
57 > * `ManagedTelemetrySettings`); they flatten to dot-path bag keys via
58 > * {@link normalizeManagedSettings}, so no {@link STRUCTURED_MANAGED_SETTINGS} entry is needed.
59 > * The `telemetry.resourceAttributes` and `telemetry.headers` map fields are structured
60 > * ({@link STRUCTURED_MANAGED_SETTINGS} rows carry them as JSON-encoded objects under their nested
61 > * keys); `telemetry.serviceName` is a scalar.
62 > */
63 >
64 > /** Managed-settings key for enterprise OTel enablement. */
65 > export const COPILOT_OTEL_ENABLED_KEY = 'telemetry.enabled';
66 >
67 > /** Managed-settings key for the enterprise OTLP collector endpoint. */
68 > export const COPILOT_OTEL_ENDPOINT_KEY = 'telemetry.endpoint';
69 >
70 > /** Managed-settings key for the enterprise OTLP protocol (`http/json`, `http/protobuf`, or `grpc`). */
71 > export const COPILOT_OTEL_PROTOCOL_KEY = 'telemetry.protocol';
72 >
73 > /** Managed-settings key for enterprise OTel content capture. */
74 > export const COPILOT_OTEL_CAPTURE_CONTENT_KEY = 'telemetry.captureContent';
75 >
76 > /** Managed-settings key that prevents users from enabling OTel content capture themselves. */
77 > export const COPILOT_OTEL_LOCK_CAPTURE_CONTENT_KEY = 'telemetry.lockCaptureContent';
78 >
79 > /** Managed-settings key for the OTel `service.name` resource attribute. */
80 > export const COPILOT_OTEL_SERVICE_NAME_KEY = 'telemetry.serviceName';
81 >
82 > /** Managed-settings key for additional OTel resource attributes (a `{ [k]: string }` map). */
83 > export const COPILOT_OTEL_RESOURCE_ATTRIBUTES_KEY = 'telemetry.resourceAttributes';
84 >
85 > /** Managed-settings key for extra OTLP exporter headers (a `{ [k]: string }` map). */
86 > export const COPILOT_OTEL_HEADERS_KEY = 'telemetry.headers';
87 >
88 > const managedSettingValueCallbacks = new Map<string, (policyData: IPolicyData) => ManagedSettingValue | undefined>();
89 >
90 > /**
91 > * Standard pass-through `value` callback for a managed-settings-driven policy: locks the setting
92 > * to the managed value when the enterprise has set it, and returns `undefined` otherwise so the
93 > * user's own setting falls through. Use for the common case; policies that combine the managed
94 > * value with other conditions (e.g. `chat_preview_features_enabled`) keep a custom callback.
95 > *
96 > * The callback is memoized per key, so repeated calls for the same key return the SAME function
97 > * reference. That reference identity is what lets `isSamePolicyDefinition` skip needless
98 > * re-registration, and memoizing makes the guarantee hold regardless of where the helper is called.
99 > */
100 > export function managedSettingValue(key: string): (policyData: IPolicyData) => ManagedSettingValue | undefined {
101 let callback = managedSettingValueCallbacks.get(key);
102 if (!callback) {
106 return callback;
107 }
109 > let managedModelValueCallback: ((policyData: IPolicyData) => ManagedSettingValue | undefined) | undefined;
110 >
111 > /**
112 > * `value` callback for the default-chat-model managed setting ({@link COPILOT_MODEL_KEY}). Like
113 > * {@link managedSettingValue} it locks the setting to the managed value and otherwise falls through
114 > * to the user's own value, but it additionally trims the string and treats a blank/whitespace-only
115 > * value as "unset" (returns `undefined`) — an admin clearing the field must not lock the setting to
116 > * an empty string. The model-specific normalization lives here, alongside the other managed-settings
117 > * handling, rather than inline at the policy declaration, so every managed-settings control is wired
118 > * the same way.
119 > *
120 > * Memoized (single key) so repeated calls return the SAME function reference, matching the
121 > * reference-identity contract {@link managedSettingValue} relies on for `isSamePolicyDefinition`.
122 > */
123 > export function managedModelValue(): (policyData: IPolicyData) => ManagedSettingValue | undefined {
124 if (!managedModelValueCallback) {
125 managedModelValueCallback = policyData => {
131 return managedModelValueCallback;
132 }
134 > export const INativeManagedSettingsService = createDecorator<INativeManagedSettingsService>('nativeManagedSettingsService');
135 >
136 > export interface INativeManagedSettingsService {
137 > readonly _serviceBrand: undefined;
138 > readonly managedSettings: ManagedSettingsData;
139 > readonly onDidChangeManagedSettings: Event<ManagedSettingsData>;
140 > updatePolicyDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<ManagedSettingsData>;
141 > }
142 >
143 > export class NullNativeManagedSettingsService implements INativeManagedSettingsService {
144 readonly _serviceBrand: undefined;
145 readonly managedSettings: ManagedSettingsData = {};
146 readonly onDidChangeManagedSettings = Event.None;
148 > async updatePolicyDefinitions(): Promise<ManagedSettingsData> { return this.managedSettings; }
149 > }
150 >
151 function flattenManagedSettings(object: unknown): Record<string, string | number | boolean> {
152 const result: Record<string, string | number | boolean> = {};
154 return result;
155 }
157 function flattenManagedSettingsValue(value: unknown, prefix: string | undefined, result: Record<string, string | number | boolean>): void {
158 if (typeof value === 'string' || typeof value === 'number' || typeof value === 'boolean') {
171 }
172 }
174 function isManagedSettingsObject(value: unknown): value is Record<string, unknown> {
175 return typeof value === 'object' && value !== null && !Array.isArray(value);
176 }
178 > /**
179 > * Aggregate the `managedSettings` declarations of every policy definition into a single
180 > * key -> definition map. This is the single source of truth for which Copilot managed-settings
181 > * keys (and their value types) are honored, and it drives both delivery channels: the native
182 > * MDM watcher and the server `managed_settings` endpoint projection.
183 > */
184 > export function collectManagedSettingsDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): IManagedSettingsPolicyDefinitions {
185 const definitions: Record<string, IManagedSettingPolicyDefinition> = {};
186 for (const policyName in policyDefinitions) {
194 return definitions;
195 }
197 > /**
198 > * Whether any policy in `policyDefinitions` declares at least one managed-settings key. Cheap
199 > * existence check (short-circuits) used to decide whether the native MDM watcher is needed at all,
200 > * without aggregating the full {@link collectManagedSettingsDefinitions} map.
201 > */
202 > export function hasManagedSettingsDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): boolean {
203 for (const policyName in policyDefinitions) {
204 const policyManagedSettings = policyDefinitions[policyName].managedSettings;
209 return false;
210 }
212 > /**
213 > * Project a raw managed-settings bag onto the declared schema: keep only keys declared by a
214 > * policy definition whose runtime value matches the declared type. Undeclared keys and
215 > * type-mismatched values are dropped (with an optional warning). Values are validated, never
216 > * coerced, so a key declared as `string` keeps its string value untouched.
217 > *
218 > * This keeps the server endpoint and native MDM delivery aligned on the same
219 > * declaration-driven key set and value types.
220 > */
221 > export function projectManagedSettings(values: ManagedSettingsData, definitions: IManagedSettingsPolicyDefinitions, onWarn?: (msg: string) => void): ManagedSettingsData {
222 const projected: Record<string, ManagedSettingValue> = {};
223 for (const key in definitions) {
234 return projected;
235 }
237 > /**
238 > * A delivery channel that can provide managed settings. Managed settings can be delivered by more
239 > * than one channel, so this names the known sources to give policy evaluation and the Policy
240 > * Diagnostics report one shared vocabulary. Extend this union (and {@link MANAGED_SETTINGS_CHANNELS}
241 > * / {@link pickManagedSettings}) when adding a new channel.
242 > */
243 > export type ManagedSettingsChannel =
244 > /** GitHub `/copilot_internal/managed_settings` endpoint (server-delivered). */
245 > | 'server'
246 > /** Native MDM: OS registry (Windows) / managed preferences (macOS) via `@vscode/policy-watcher`. */
247 > | 'nativeMdm'
248 > /** File on a well-known disk path (`managed-settings.json`). */
249 > | 'file';
250 >
251 > /**
252 > * The source attributed to an effective managed setting (or to the overall report). A
253 > * {@link ManagedSettingsChannel} once a channel has won, or `'none'` when no channel contributes.
254 > */
255 > export type ManagedSettingsSource = ManagedSettingsChannel | 'none';
256 >
257 > /**
258 > * The delivery channels in fixed precedence order (highest first): native MDM → server-delivered →
259 > * file on disk. This single ordered list drives the per-key resolution in {@link pickManagedSettings}
260 > * and is the one place to extend when a new channel is introduced. Rationale for the order: the
261 > * server is harder to bypass than local MDM, and a local file is the most easily tampered with.
262 > */
263 > export const MANAGED_SETTINGS_CHANNELS: readonly ManagedSettingsChannel[] = ['nativeMdm', 'server', 'file'];
264 >
265 > /** A single channel's contribution to a managed-settings key, for provenance in the resolution. */
266 > export interface IManagedSettingsContribution {
267 > /** The channel that supplied this value. */
268 > readonly channel: ManagedSettingsChannel;
269 > /** The value the channel supplied for the key. */
270 > readonly value: ManagedSettingValue;
271 > }
272 >
273 > /** How a single managed-settings key was resolved across the delivery channels. */
274 > export interface IManagedSettingResolution {
275 > /** The effective (winning) value applied for the key. */
276 > readonly value: ManagedSettingValue;
277 > /** The channel whose value won (always the first {@link contributions} entry's channel). */
278 > readonly source: ManagedSettingsChannel;
279 > /** Every channel that supplied this key, in precedence order (winner first, overridden after). */
280 > readonly contributions: readonly IManagedSettingsContribution[];
281 > }
282 >
283 > /** The result of merging managed settings from every delivery channel on a per-key basis. */
284 > export interface IManagedSettingsPick {
285 > /** The effective merged bag: the winning value for each key contributed by any channel. */
286 > readonly values: ManagedSettingsData;
287 > /** Per-key provenance: how each key resolved and which channels were overridden. */
288 > readonly resolutions: ReadonlyMap<string, IManagedSettingResolution>;
289 > /** The channels that supplied at least one *winning* key, in precedence order. */
290 > readonly activeSources: readonly ManagedSettingsChannel[];
291 > }
292 >
293 > /**
294 > * Merge the managed-settings bags from every delivery channel on a **per-key** basis.
295 > *
296 > * Precedence (highest first): native MDM → server-delivered → file on disk. Unlike a single
297 > * authoritative source, the channels *are* merged key-by-key: for each key the highest-precedence
298 > * channel that supplies it wins, but a key that the higher channels never set is still filled in by
299 > * a lower channel. A value an admin locks via native MDM therefore cannot be overwritten by the
300 > * server or a file, while keys those higher channels leave unset remain available to lower ones.
301 > *
302 > * The parameter order matches the precedence so call sites read top-to-bottom. Centralizing the
303 > * resolution here (rather than inlining it at each call site) keeps policy evaluation
304 > * ({@link AccountPolicyService.getPolicyData}) and the Policy Diagnostics report from drifting apart,
305 > * and gives one obvious place to extend when a new channel is introduced. Empty or absent channels
306 > * contribute nothing.
307 > */
308 > export function pickManagedSettings(nativeMdm: ManagedSettingsData | undefined, server: ManagedSettingsData | undefined, file: ManagedSettingsData | undefined): IManagedSettingsPick {
309 const bags: Record<ManagedSettingsChannel, ManagedSettingsData | undefined> = { nativeMdm, server, file };
310
349 };
350 }
352 > // --- File-based managed settings ---
353 >
354 > /** macOS well-known path for file-based managed settings. */
355 > export const MANAGED_SETTINGS_MACOS_FILE_PATH = '/Library/Application Support/GitHubCopilot/managed-settings.json';
356 >
357 > /** Linux well-known path for file-based managed settings. */
358 > export const MANAGED_SETTINGS_LINUX_FILE_PATH = '/etc/github-copilot/managed-settings.json';
359 >
360 > /** Windows directory name under %ProgramFiles% for file-based managed settings. */
361 > export const MANAGED_SETTINGS_WINDOWS_DIR = 'GitHubCopilot';
362 >
363 > /** Managed settings file name. */
364 > export const MANAGED_SETTINGS_FILE_NAME = 'managed-settings.json';
365 >
366 > /**
367 > * Descriptor for a structured (object/array) managed setting: one carried across every delivery
368 > * channel as a canonical JSON string under a single key. This table is the single place that
369 > * knows how to turn a managed-settings schema field into that canonical value, so adding a
370 > * structured key is one row here (plus the policy declaration that reads the bag key).
371 > *
372 > * `key` is both the source field name read from the parsed input and the canonical bag key the
373 > * JSON string is stored under — for structured settings these are identical by contract (a
374 > * structured key's bag name matches the schema field exactly; only scalar settings flatten to a
375 > * differently-shaped dot-path, and those don't go through this table).
376 > */
377 > interface IStructuredManagedSetting {
378 > /** Source field name read from the parsed input, and the canonical bag key the JSON string is stored under. */
379 > readonly key: string;
380 > /**
381 > * Normalize the raw value into the canonical pre-stringify shape an admin authors via native
382 > * MDM. Return `undefined` to omit the key (absent or malformed value). Note an empty array (the
383 > * `strictKnownMarketplaces` lockdown case) is returned as-is, not omitted.
384 > */
385 > readonly encode: (value: unknown, onWarn?: (msg: string) => void) => unknown;
386 > }
387 >
388 > /**
389 > * Encode a managed-settings value into a canonical `{ [k]: string }` map: keeps string values
390 > * as-is and coerces number/boolean values to strings; drops keys with non-primitive values.
391 > * Returns `undefined` for a non-object input so the structured key is omitted.
392 > */
393 function encodeStringMap(value: unknown): Record<string, string> | undefined {
394 if (!isObject(value)) {
408 return out;
409 }
411 > /** Pass an object value through unchanged; omit the key for any non-object value. */
412 function encodeObject(value: unknown): object | undefined {
413 return isObject(value) ? value : undefined;
414 }
416 > /** Pass an array value through unchanged (including an empty array); omit the key otherwise. */
417 function encodeArray(value: unknown): unknown[] | undefined {
418 return Array.isArray(value) ? value : undefined;
419 }
421 > /**
422 > * Encode the schema's `{ [id]: { source } }` marketplace map into the canonical
423 > * `{ [name]: url-or-shorthand }` dict; drops malformed entries (with an optional warning) and omits
424 > * the key when there are none.
425 > */
426 function encodeExtraMarketplaces(value: unknown, onWarn?: (msg: string) => void): Record<string, string> | undefined {
427 return extraKnownMarketplacesToConfigDict(normalizeExtraKnownMarketplaces(value, onWarn));
428 }
430 > const STRUCTURED_MANAGED_SETTINGS: readonly IStructuredManagedSetting[] = [
431 > {
432 > key: COPILOT_ENABLED_PLUGINS_KEY,
433 > encode: encodeObject,
434 > },
435 > {
436 > key: COPILOT_STRICT_MARKETPLACES_KEY,
437 > encode: encodeArray,
438 > },
439 > {
440 > key: COPILOT_ALLOWED_MCP_SERVERS_KEY,
441 > encode: encodeArray,
442 > },
443 > {
444 > key: COPILOT_DENIED_MCP_SERVERS_KEY,
445 > encode: encodeArray,
446 > },
447 > {
448 > key: COPILOT_EXTRA_MARKETPLACES_KEY,
449 > encode: encodeExtraMarketplaces,
450 > },
451 > {
452 > // Nested under `telemetry`; carried as a JSON-encoded `{ [k]: string }` map. Non-string
453 > // primitive values are coerced to strings; non-primitive values are dropped.
454 > key: COPILOT_OTEL_RESOURCE_ATTRIBUTES_KEY,
455 > encode: encodeStringMap,
456 > },
457 > {
458 > // Nested under `telemetry`; carried as a JSON-encoded `{ [k]: string }` map of OTLP headers.
459 > key: COPILOT_OTEL_HEADERS_KEY,
460 > encode: encodeStringMap,
461 > },
462 > ];
463 >
464 > /**
465 > * Read a (possibly nested) dot-separated key from a parsed managed-settings object, e.g.
466 > * `telemetry.resourceAttributes`. Returns `undefined` if any path segment is missing or not an
467 > * object. Single-segment keys behave like a plain property read.
468 > */
469 function readNestedManagedKey(obj: Record<string, unknown>, dottedKey: string): unknown {
470 let current: unknown = obj;
477 return current;
478 }
480 > /**
481 > * Return a copy of `obj` with the (possibly nested) dot-separated key removed, cloning only the
482 > * objects along the touched path so the original (and any shared sub-objects) stay untouched. The
483 > * spread-then-`delete` shape matches a destructuring rest: it copies own enumerable keys (including
484 > * an own `__proto__`) without triggering the inherited `__proto__` setter.
485 > */
486 function withNestedManagedKeyDeleted(obj: Record<string, unknown>, dottedKey: string): Record<string, unknown> {
487 const dot = dottedKey.indexOf('.');
498 return { ...obj, [head]: withNestedManagedKeyDeleted(child as Record<string, unknown>, dottedKey.slice(dot + 1)) };
499 }
501 > /**
502 > * Normalize a parsed managed-settings object (from the server `managed_settings` API, a file on
503 > * disk, or any other source using the managed-settings schema) into the canonical
504 > * `ManagedSettingsData` bag that the policy framework consumes. This is the **single**
505 > * normalization path for all delivery channels, so downstream projection and policy `value()`
506 > * callbacks behave identically regardless of source. It does not enforce the declared
507 > * `managedSettings` schema — dropping undeclared or type-mismatched keys happens later, at
508 > * {@link projectManagedSettings}.
509 > *
510 > * - Scalar leaves (`permissions.*` and any forward-compatible scalar keys) are flattened into
511 > * dot-separated keys.
512 > * - Structured settings (declared in {@link STRUCTURED_MANAGED_SETTINGS}) are carried as canonical
513 > * JSON strings under a single key each — the same shape an admin authors via native MDM.
514 > * `PolicyConfiguration` parses the JSON back into the object-typed setting on read.
515 > * `extraKnownMarketplaces` is normalized from the schema's `{ [id]: { source } }` map to the
516 > * `{ [name]: url-or-shorthand }` dict.
517 > *
518 > * Malformed marketplace entries are dropped (with an optional warning via {@link onWarn}) rather
519 > * than throwing, so a bad enterprise settings file degrades gracefully instead of blocking startup.
520 > */
521 > export function normalizeManagedSettings(parsed: Record<string, unknown>, onWarn?: (msg: string) => void): ManagedSettingsData {
522 // Spread + delete (not for..in + assignment) so the scalar remainder keeps exact `{ ...rest }`
523 // semantics: it never triggers the inherited `__proto__` setter for a source-sent own
540 return result;
541 }
543 > /**
544 > * Normalize the schema's `{ [id]: { source } }` marketplace map into an
545 > * {@link IExtraKnownMarketplaceEntry} array, preserving the marketplace `name`,
546 > * source discriminator, and any `ref`. Malformed or off-spec entries are dropped
547 > * (with an optional warning via {@link onWarn}).
548 > */
549 function normalizeExtraKnownMarketplaces(value: unknown, onWarn?: (msg: string) => void): IExtraKnownMarketplaceEntry[] | undefined {
550 if (!isObject(value)) {
576 return entries;
577 }
579 > export const IFileManagedSettingsService = createDecorator<IFileManagedSettingsService>('fileManagedSettingsService');
580 >
581 > export interface IFileManagedSettingsService {
582 > readonly _serviceBrand: undefined;
583 > readonly rawManagedSettings: RawManagedSettingsData;
584 > readonly managedSettings: ManagedSettingsData;
585 > readonly onDidChangeRawManagedSettings: Event<RawManagedSettingsData>;
586 > readonly onDidChangeManagedSettings: Event<ManagedSettingsData>;
587 > }
588 >
589 > export class NullFileManagedSettingsService implements IFileManagedSettingsService {
590 readonly _serviceBrand: undefined;
591 readonly rawManagedSettings: RawManagedSettingsData = {};
593 readonly onDidChangeRawManagedSettings = Event.None;
594 readonly onDidChangeManagedSettings = Event.None;
src/vs/workbench/contrib/chat/common/chatModes.ts 367 covered LOC · 51 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatModes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken, CancellationTokenSource } from '../../../../base/common/cancellation.js';
7 > import { Emitter, Event } from '../../../../base/common/event.js';
8 > import { Disposable, IDisposable } from '../../../../base/common/lifecycle.js';
9 > import { constObservable, IObservable, ISettableObservable, observableValue, transaction } from '../../../../base/common/observable.js';
10 > import { isUriComponents, URI } from '../../../../base/common/uri.js';
11 > import { IOffsetRange } from '../../../../editor/common/core/ranges/offsetRange.js';
12 > import { localize } from '../../../../nls.js';
13 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
14 > import { IContextKey, IContextKeyService } from '../../../../platform/contextkey/common/contextkey.js';
15 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
16 > import { createDecorator, IInstantiationService } from '../../../../platform/instantiation/common/instantiation.js';
17 > import { ILogService } from '../../../../platform/log/common/log.js';
18 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
19 > import { IChatAgentService } from './participants/chatAgents.js';
20 > import { ChatContextKeys } from './actions/chatContextKeys.js';
21 > import { getChatSessionType, LocalChatSessionUri } from './model/chatUri.js';
22 > import { ChatConfiguration, ChatModeKind } from './constants.js';
23 > import { IHandOff } from './promptSyntax/promptFileParser.js';
24 > import { IAgentSource, ICustomAgent, ICustomAgentVisibility, isCustomAgentVisibility, PromptsStorage } from './promptSyntax/service/promptsService.js';
25 > import { ICustomizationHarnessService } from './customizationHarnessService.js';
26 > import { PromptFileSource, Target } from './promptSyntax/promptTypes.js';
27 > import { ThemeIcon } from '../../../../base/common/themables.js';
28 > import { Codicon } from '../../../../base/common/codicons.js';
29 > import { hash } from '../../../../base/common/hash.js';
30 > import { isString } from '../../../../base/common/types.js';
31 > import { isTarget } from './promptSyntax/languageProviders/promptFileAttributes.js';
32 > import { equals as arraysEqual } from '../../../../base/common/arrays.js';
33 > import { isEqual as isURLEquals } from '../../../../base/common/resources.js';
34 > import { equals as objectEquals } from '../../../../base/common/objects.js';
35 > import { Delayer } from '../../../../base/common/async.js';
36 > import { isCancellationError } from '../../../../base/common/errors.js';
37 >
38 >
39 > export const IChatModeService = createDecorator<IChatModeService>('chatModeService');
40 > export interface IChatModeService {
41 > readonly _serviceBrand: undefined;
42 >
43 > /**
44 > * Returns the chat modes available for the given session resource.
45 > *
46 > * Instances need to be disposed by the caller when no longer needed
47 > */
48 > createModes(sessionResource: URI): IChatModes & IDisposable;
49 >
50 > /**
51 > * Returns the local chat modes after awaiting any in-flight refresh.
52 > */
53 > getLocalModes(): Promise<IChatModes>;
54 > }
55 >
56 > /**
57 > * The set of chat modes available for a particular session type, partitioned
58 > * into builtin and custom modes, with helpers for lookup by id or name.
59 > */
60 > export interface IChatModes {
61 > readonly onDidChange: Event<void>;
62 > readonly builtin: readonly IChatMode[];
63 > readonly custom: readonly IChatMode[];
64 > findModeById(id: string): IChatMode | undefined;
65 > findModeByName(name: string): IChatMode | undefined;
66 >
67 > /**
68 > * Awaits the most recently scheduled update of custom prompt modes.
69 > * After this resolves, {@link custom} reflects the latest data from the
70 > * prompts service.
71 > */
72 > waitForPendingUpdates(): Promise<void>;
73 > }
74 >
75 > class ChatModes extends Disposable implements IChatModes {
76 >
77 > private static readonly CUSTOM_MODES_STORAGE_KEY_PREFIX = 'chat.customModes.';
78 >
79 > private readonly hasCustomModes: IContextKey<boolean>;
80 > private readonly _customModeInstances = new Map<string, CustomChatMode>();
81 > private readonly _storageKey: string;
82 >
83 > private readonly _onDidChange = this._register(new Emitter<void>());
84 > readonly onDidChange = this._onDidChange.event;
85 >
86 > /** Tracks the most recent refresh of custom prompt modes. */
87 > private _pendingRefresh: Promise<void> = Promise.resolve();
88 >
89 > private _refreshCancellationSource: CancellationTokenSource | undefined;
90 > private readonly _refreshThrottler = this._register(new Delayer<void>(100));
91 >
92 > constructor(
93 private readonly sessionResource: URI,
94 @IChatAgentService private readonly chatAgentService: IChatAgentService,
132 }));
133 }
134 > chatModes.ts
135 > get builtin(): readonly IChatMode[] {
136 return this.getBuiltinModes();
137 }
138 > chatModes.ts
139 > get custom(): readonly IChatMode[] {
140 return this.getCustomModes();
141 }
142 > chatModes.ts
143 > findModeById(id: string | ChatModeKind): IChatMode | undefined {
144 return this.getBuiltinModes().find(mode => mode.id === id) ?? this._customModeInstances.get(id);
145 }
146 > chatModes.ts
147 > findModeByName(name: string): IChatMode | undefined {
148 return this.getBuiltinModes().find(mode => mode.name.get() === name) ?? this.getCustomModes().find(mode => mode.name.get() === name || mode.id === name);
149 }
150 > chatModes.ts
151 > waitForPendingUpdates(): Promise<void> {
152 return this._pendingRefresh;
153 }
154 > chatModes.ts
155 > private loadCachedModes(): void {
156 try {
157 const cachedCustomModes = this.storageService.getObject(this._storageKey, StorageScope.WORKSPACE);
163 }
164 }
165 > chatModes.ts
166 > private deserializeCachedModes(cachedCustomModes: unknown): void {
167 if (!Array.isArray(cachedCustomModes)) {
168 this.logService.error('Invalid cached custom modes data: expected array');
205 this.hasCustomModes.set(this._customModeInstances.size > 0);
206 }
207 > chatModes.ts
208 > private saveCachedModes(): void {
209 try {
210 const modesToCache = Array.from(this._customModeInstances.values());
214 }
215 }
216 > chatModes.ts
217 > private triggerRefresh(): Promise<void> {
218 this._refreshCancellationSource?.cancel();
219 this._refreshCancellationSource?.dispose();
230 });
231 }
232 > chatModes.ts
233 > override dispose(): void {
234 this._refreshCancellationSource?.cancel();
235 this._refreshCancellationSource?.dispose();
237 super.dispose();
238 }
239 > chatModes.ts
240 > private async refreshCustomPromptModes(token: CancellationToken): Promise<void> {
241 let hasChanges = false;
242 try {
295 }
296 }
297 > chatModes.ts
298 > private getBuiltinModes(): IChatMode[] {
299 const builtinModes: IChatMode[] = [
300 ChatMode.Ask,
311 return builtinModes;
312 }
313 > chatModes.ts
314 > private getCustomModes(): IChatMode[] {
315 // Show custom modes when agent mode is enabled OR when disabled by policy (to show them in the policy-managed group)
316 return this.chatAgentService.hasToolsAgent || this.isAgentModeDisabledByPolicy() ? Array.from(this._customModeInstances.values()) : [];
317 }
318 > chatModes.ts
319 > private isAgentModeDisabledByPolicy(): boolean {
320 return this.configurationService.inspect<boolean>(ChatConfiguration.AgentEnabled).policyValue === false;
321 }
322 > } chatModes.ts
323 >
324 > export class ChatModeService extends Disposable implements IChatModeService {
325 > declare readonly _serviceBrand: undefined;
326 >
327 > private readonly agentModeDisabledByPolicy: IContextKey<boolean>;
328 > private localMode: Promise<IChatModes> | undefined;
329 >
330 > constructor(
331 @IInstantiationService private readonly instantiationService: IInstantiationService,
332 @IContextKeyService contextKeyService: IContextKeyService,
347 }));
348 }
349 > chatModes.ts
350 > createModes(sessionResource: URI): IChatModes & IDisposable {
351 return this.instantiationService.createInstance(ChatModes, sessionResource);
352 }
353 > chatModes.ts
354 > async getLocalModes(): Promise<IChatModes> {
355 if (!this.localMode) {
356 this.localMode = (async () => {
362 return this.localMode;
363 }
364 > chatModes.ts
365 > private updateAgentModePolicyContextKey(): void {
366 this.agentModeDisabledByPolicy.set(this.isAgentModeDisabledByPolicy());
367 }
368 > chatModes.ts
369 > private isAgentModeDisabledByPolicy(): boolean {
370 return this.configurationService.inspect<boolean>(ChatConfiguration.AgentEnabled).policyValue === false;
371 }
372 > } chatModes.ts
373 >
374 > export interface IChatModeData {
375 > readonly id: string;
376 > readonly name: string;
377 > readonly description?: string;
378 > readonly kind: ChatModeKind;
379 > readonly customTools?: readonly string[];
380 > readonly model?: readonly string[] | string;
381 > readonly argumentHint?: string;
382 > readonly modeInstructions?: IChatModeInstructions;
383 > readonly body?: string; /* deprecated */
384 > readonly handOffs?: readonly IHandOff[];
385 > readonly uri?: URI;
386 > readonly source?: IChatModeSourceData;
387 > readonly target?: Target;
388 > readonly visibility?: ICustomAgentVisibility;
389 > readonly agents?: readonly string[];
390 > readonly sessionTypes?: readonly string[];
391 > readonly infer?: boolean; // deprecated, only available in old cached data
392 > }
393 >
394 > export interface IChatMode {
395 > readonly id: string;
396 > readonly name: IObservable<string>;
397 > readonly label: IObservable<string>;
398 > readonly icon: IObservable<ThemeIcon | undefined>;
399 > readonly description: IObservable<string | undefined>;
400 > readonly isBuiltin: boolean;
401 > readonly kind: ChatModeKind;
402 > readonly customTools?: IObservable<readonly string[] | undefined>;
403 > readonly handOffs?: IObservable<readonly IHandOff[] | undefined>;
404 > readonly model?: IObservable<readonly string[] | undefined>;
405 > readonly argumentHint?: IObservable<string | undefined>;
406 > readonly modeInstructions?: IObservable<IChatModeInstructions>;
407 > readonly uri?: IObservable<URI>;
408 > readonly source?: IAgentSource;
409 > readonly target: IObservable<Target>;
410 > readonly visibility?: IObservable<ICustomAgentVisibility | undefined>;
411 > readonly agents?: IObservable<readonly string[] | undefined>;
412 > readonly sessionTypes?: readonly string[];
413 > }
414 >
415 > export interface IVariableReference {
416 > readonly name: string;
417 > readonly range: IOffsetRange;
418 > }
419 >
420 > export interface IChatModeInstructions {
421 > readonly content: string;
422 > readonly toolReferences: readonly IVariableReference[];
423 > readonly metadata?: Record<string, boolean | string | number>;
424 > }
425 >
426 > export namespace IChatModeInstructions {
427 > export function isEquals(a: IChatModeInstructions | undefined, b: IChatModeInstructions | undefined): boolean {
428 if (a === b) {
429 return true;
436 objectEquals(a.metadata, b.metadata);
437 }
438 > chatModes.ts
439 > }
440 >
441 function isCachedChatModeData(data: unknown): data is IChatModeData {
442 if (typeof data !== 'object' || data === null) {
461 (mode.sessionTypes === undefined || Array.isArray(mode.sessionTypes));
462 }
463 > chatModes.ts
464 > export class CustomChatMode implements IChatMode {
465 > private readonly _nameObservable: ISettableObservable<string>;
466 > private readonly _descriptionObservable: ISettableObservable<string | undefined>;
467 > private readonly _customToolsObservable: ISettableObservable<readonly string[] | undefined>;
468 > private readonly _modeInstructions: ISettableObservable<IChatModeInstructions>;
469 > private readonly _uriObservable: ISettableObservable<URI>;
470 > private readonly _modelObservable: ISettableObservable<readonly string[] | undefined>;
471 > private readonly _argumentHintObservable: ISettableObservable<string | undefined>;
472 > private readonly _handoffsObservable: ISettableObservable<readonly IHandOff[] | undefined>;
473 > private readonly _targetObservable: ISettableObservable<Target>;
474 > private readonly _visibilityObservable: ISettableObservable<ICustomAgentVisibility | undefined>;
475 > private readonly _agentsObservable: ISettableObservable<readonly string[] | undefined>;
476 > private _source: IAgentSource;
477 > private _sessionTypes: readonly string[] | undefined;
478 >
479 > public readonly id: string;
480 >
481 > get name(): IObservable<string> {
482 > return this._nameObservable;
483 > }
484 >
485 > get description(): IObservable<string | undefined> {
486 return this._descriptionObservable;
487 }
488 > chatModes.ts
489 > get icon(): IObservable<ThemeIcon | undefined> {
490 return constObservable(undefined);
491 }
492 > chatModes.ts
493 > public get isBuiltin(): boolean {
494 return isBuiltinChatMode(this);
495 }
496 > chatModes.ts
497 > get customTools(): IObservable<readonly string[] | undefined> {
498 return this._customToolsObservable;
499 }
500 > chatModes.ts
501 > get model(): IObservable<readonly string[] | undefined> {
502 return this._modelObservable;
503 }
504 > chatModes.ts
505 > get argumentHint(): IObservable<string | undefined> {
506 return this._argumentHintObservable;
507 }
508 > chatModes.ts
509 > get modeInstructions(): IObservable<IChatModeInstructions> {
510 return this._modeInstructions;
511 }
512 > chatModes.ts
513 > get uri(): IObservable<URI> {
514 return this._uriObservable;
515 }
516 > chatModes.ts
517 > get label(): IObservable<string> {
518 return this.name;
519 }
520 > chatModes.ts
521 > get handOffs(): IObservable<readonly IHandOff[] | undefined> {
522 return this._handoffsObservable;
523 }
524 > chatModes.ts
525 > get source(): IAgentSource {
526 return this._source;
527 }
528 > chatModes.ts
529 > get target(): IObservable<Target> {
530 return this._targetObservable;
531 }
532 > chatModes.ts
533 > get visibility(): IObservable<ICustomAgentVisibility | undefined> {
534 return this._visibilityObservable;
535 }
536 > chatModes.ts
537 > get agents(): IObservable<readonly string[] | undefined> {
538 return this._agentsObservable;
539 }
540 > chatModes.ts
541 > get sessionTypes(): readonly string[] | undefined {
542 return this._sessionTypes;
543 }
544 > chatModes.ts
545 > public readonly kind = ChatModeKind.Agent;
546 >
547 > constructor(
548 customChatMode: ICustomAgent
549 ) {
563 this._sessionTypes = customChatMode.sessionTypes;
564 }
565 > chatModes.ts
566 > /**
567 > * Updates the underlying data and triggers observable changes
568 > */
569 > updateData(newData: ICustomAgent): boolean {
570 let hasChanges = false;
571
599 return hasChanges;
600 }
601 > chatModes.ts
602 > toJSON(): IChatModeData {
603 return {
604 id: this.id,
619 };
620 }
621 > } chatModes.ts
622 >
623 > type IChatModeSourceData =
624 > | { readonly storage: PromptsStorage.extension; readonly extensionId: string; type?: PromptFileSource.ExtensionContribution | PromptFileSource.ExtensionAPI }
625 > | { readonly storage: PromptsStorage.local | PromptsStorage.user | PromptsStorage.builtIn }
626 > | { readonly storage: PromptsStorage.plugin; readonly pluginUri: URI };
627 >
628 function isChatModeSourceData(value: unknown): value is IChatModeSourceData {
629 if (typeof value !== 'object' || value === null) {
639 return data.storage === PromptsStorage.local || data.storage === PromptsStorage.user || data.storage === PromptsStorage.builtIn;
640 }
641 > chatModes.ts
642 function serializeChatModeSource(source: IAgentSource | undefined): IChatModeSourceData | undefined {
643 if (!source) {
652 return { storage: source.storage };
653 }
654 > chatModes.ts
655 function reviveChatModeSource(data: IChatModeSourceData | undefined): IAgentSource | undefined {
656 if (!data) {
665 return { storage: data.storage };
666 }
667 > chatModes.ts
668 > export class BuiltinChatMode implements IChatMode {
669 > public readonly name: IObservable<string>;
670 > public readonly label: IObservable<string>;
671 > public readonly description: IObservable<string>;
672 > public readonly icon: IObservable<ThemeIcon>;
673 > public readonly target: IObservable<Target>;
674 >
675 > constructor(
676 > public readonly kind: ChatModeKind,
677 > label: string,
678 > description: string,
679 > icon: ThemeIcon,
680 > ) {
681 > this.name = constObservable(kind);
682 > this.label = constObservable(label);
683 > this.description = observableValue('description', description);
684 > this.icon = constObservable(icon);
685 > this.target = constObservable(Target.Undefined);
686 > }
687 >
688 > public get isBuiltin(): boolean {
689 return isBuiltinChatMode(this);
690 }
691 > chatModes.ts
692 > get id(): string {
693 > // Need a differentiator? chatModes.ts
694 > return this.kind;
695 > }
696 > chatModes.ts
697 > /**
698 > * Getters are not json-stringified
699 > */
700 > toJSON(): IChatModeData {
701 return {
702 id: this.id,
706 };
707 }
708 > } chatModes.ts
709 >
710 > export namespace ChatMode {
711 > export const Ask = new BuiltinChatMode(ChatModeKind.Ask, 'Ask', localize('chatDescription', "Explore and understand your code"), Codicon.question);
712 > export const Edit = new BuiltinChatMode(ChatModeKind.Edit, 'Edit', localize('editsDescription', "Edit or refactor selected code"), Codicon.edit);
713 > export const Agent = new BuiltinChatMode(ChatModeKind.Agent, 'Agent', localize('agentDescription', "Describe what to build"), Codicon.agent);
714 > }
715 >
716 > export function isBuiltinChatMode(mode: IChatMode): boolean {
717 return mode.id === ChatMode.Ask.id ||
718 mode.id === ChatMode.Edit.id ||
719 mode.id === ChatMode.Agent.id;
720 }
721 > chatModes.ts
722 > /**
723 > * Returns a telemetry-safe mode name. User/local mode names are hashed
724 > * to avoid leaking PII; builtin and extension mode names are returned as-is.
725 > */
726 > export function getModeNameForTelemetry(mode: IChatMode): string {
727 const modeStorage = mode.source?.storage;
728 if (modeStorage === PromptsStorage.local || modeStorage === PromptsStorage.user) {
731 return mode.name.get();
732 }
733 > chatModes.ts
734 > /**
735 > * Generates a stable identifier for a handoff by combining the target agent
736 > * name with a slugified version of the display label.
737 > *
738 > * Within a single source agent, the combination of `agent` + `label` must be
739 > * unique for IDs to be unambiguous.
740 > *
741 > * @example
742 > * ```
743 > * getHandoffId({ agent: 'agent', label: 'Continue', prompt: '...' })
744 > * // => 'agent:continue'
745 > * ```
746 > */
747 > export function getHandoffId(handoff: IHandOff): string {
748 const slug = handoff.label.toLowerCase().replace(/[^a-z0-9]+/g, '-').replace(/^-|-$/g, '');
749 return `${handoff.agent}:${slug}`;
750 }
751 > chatModes.ts
752 > /**
753 > * Describes a single handoff defined in a custom agent's `.agent.md` file.
754 > */
755 > export interface IHandoffInfo {
756 > /** Stable identifier for programmatic matching (format: `<agent>:<slugified-label>`). */
757 > readonly id: string;
758 > readonly label: string;
759 > readonly agent: string;
760 > readonly prompt: string;
761 > readonly send?: boolean;
762 > readonly showContinueOn?: boolean;
763 > readonly model?: string;
764 > }
765 >
766 > /**
767 > * Describes a custom agent (or built-in mode) and the handoffs it defines.
768 > */
769 > export interface ICustomAgentInfo {
770 > readonly id: string;
771 > readonly name: string;
772 > readonly isBuiltin: boolean;
773 > readonly visibility: {
774 > readonly userInvocable: boolean;
775 > readonly agentInvocable: boolean;
776 > };
777 > readonly handoffs: IHandoffInfo[];
778 > }
779 >
780 > /**
781 > * Builds an array of {@link ICustomAgentInfo} with handoff metadata for the given agents/modes.
782 > *
783 > * @param modes - The set of agents/modes to include. Pass all modes to get a
784 > * complete picture, or a filtered subset to scope the result.
785 > * @returns One entry per agent/mode, each containing the agent's metadata and
786 > * its declared handoffs.
787 > */
788 > export function buildCustomAgentHandoffsInfo(modes: readonly IChatMode[]): ICustomAgentInfo[] {
789 return modes.map(mode => {
790 const handoffs = mode.handOffs?.get() ?? [];
src/vs/workbench/contrib/chat/common/promptSyntax/languageProviders/promptFileAttributes.ts 362 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptFileAttributes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { dirname } from '../../../../../../base/common/resources.js';
7 > import { URI } from '../../../../../../base/common/uri.js';
8 > import { localize } from '../../../../../../nls.js';
9 > import { SpecedToolAliases } from '../../tools/languageModelToolsService.js';
10 > import { CLAUDE_AGENTS_SOURCE_FOLDER, isInClaudeRulesFolder } from '../config/promptFileLocations.js';
11 > import { PromptHeader, PromptHeaderAttributes } from '../promptFileParser.js';
12 > import { PromptsType, Target } from '../promptTypes.js';
13 >
14 > export namespace GithubPromptHeaderAttributes {
15 > export const mcpServers = 'mcp-servers';
16 > export const github = 'github';
17 > }
18 >
19 > export namespace ClaudeHeaderAttributes {
20 > export const disallowedTools = 'disallowedTools';
21 > }
22 >
23 > export function isTarget(value: unknown): value is Target {
24 return value === Target.VSCode || value === Target.GitHubCopilot || value === Target.Claude || value === Target.Undefined;
25 }
27 >
28 > interface IAttributeDefinition {
29 > readonly type: string;
30 > readonly description: string;
31 > readonly defaults?: readonly string[];
32 > readonly items?: readonly { name: string; description?: string }[];
33 > readonly enums?: readonly { name: string; description?: string }[];
34 > }
35 >
36 > const booleanAttributeEnumValues: readonly IValueEntry[] = [
37 > { name: 'true' },
38 > { name: 'false' }
39 > ];
40 >
41 > const targetAttributeEnumValues: readonly IValueEntry[] = [
42 > { name: 'vscode' },
43 > { name: 'github-copilot' },
44 > ];
45 >
46 > // Attribute metadata for prompt files (`*.prompt.md`).
47 > export const promptFileAttributes: Record<string, IAttributeDefinition> = {
48 > [PromptHeaderAttributes.name]: {
49 > type: 'scalar',
50 > description: localize('promptHeader.prompt.name', 'The name of the prompt. This is also the name of the slash command that will run this prompt.'),
51 > },
52 > [PromptHeaderAttributes.description]: {
53 > type: 'scalar',
54 > description: localize('promptHeader.prompt.description', 'The description of the reusable prompt, what it does and when to use it.'),
55 > },
56 > [PromptHeaderAttributes.argumentHint]: {
57 > type: 'scalar',
58 > description: localize('promptHeader.prompt.argumentHint', 'The argument-hint describes what inputs the prompt expects or supports.'),
59 > },
60 > [PromptHeaderAttributes.model]: {
61 > type: 'scalar | sequence',
62 > description: localize('promptHeader.prompt.model', 'The model to use in this prompt. Can also be a list of models. The first available model will be used.'),
63 > },
64 > [PromptHeaderAttributes.tools]: {
65 > type: 'scalar | sequence',
66 > description: localize('promptHeader.prompt.tools', 'The tools to use in this prompt.'),
67 > defaults: ['[]', '[\'search\', \'edit\', \'web\']'],
68 > },
69 > [PromptHeaderAttributes.agent]: {
70 > type: 'scalar',
71 > description: localize('promptHeader.prompt.agent.description', 'The agent to use when running this prompt.'),
72 > },
73 > [PromptHeaderAttributes.mode]: {
74 > type: 'scalar',
75 > description: localize('promptHeader.prompt.agent.description', 'The agent to use when running this prompt.'),
76 > },
77 > };
78 >
79 > // Attribute metadata for instructions files (`*.instructions.md`).
80 > export const instructionAttributes: Record<string, IAttributeDefinition> = {
81 > [PromptHeaderAttributes.name]: {
82 > type: 'scalar',
83 > description: localize('promptHeader.instructions.name', 'The name of the instruction file as shown in the UI. If not set, the name is derived from the file name.'),
84 > },
85 > [PromptHeaderAttributes.description]: {
86 > type: 'scalar',
87 > description: localize('promptHeader.instructions.description', 'The description of the instruction file. It can be used to provide additional context or information about the instructions and is passed to the language model as part of the prompt.'),
88 > },
89 > [PromptHeaderAttributes.applyTo]: {
90 > type: 'scalar',
91 > description: localize('promptHeader.instructions.applyToRange', 'One or more glob pattern (separated by comma) that describe for which files the instructions apply to. Based on these patterns, the file is automatically included in the prompt, when the context contains a file that matches one or more of these patterns. Use `**` when you want this file to always be added.\nExample: `**/*.ts`, `**/*.js`, `client/**`'),
92 > defaults: [
93 > '\'**\'',
94 > '\'**/*.ts, **/*.js\'',
95 > '\'**/*.php\'',
96 > '\'**/*.py\''
97 > ],
98 > },
99 > [PromptHeaderAttributes.excludeAgent]: {
100 > type: 'scalar | sequence',
101 > description: localize('promptHeader.instructions.excludeAgent', 'One or more agents to exclude from using this instruction file.'),
102 > },
103 > };
104 >
105 > // Attribute metadata for custom agent files (`*.agent.md`).
106 > export const customAgentAttributes: Record<string, IAttributeDefinition> = {
107 > [PromptHeaderAttributes.name]: {
108 > type: 'scalar',
109 > description: localize('promptHeader.agent.name', 'The name of the agent as shown in the UI.'),
110 > },
111 > [PromptHeaderAttributes.description]: {
112 > type: 'scalar',
113 > description: localize('promptHeader.agent.description', 'The description of the custom agent, what it does and when to use it.'),
114 > },
115 > [PromptHeaderAttributes.argumentHint]: {
116 > type: 'scalar',
117 > description: localize('promptHeader.agent.argumentHint', 'The argument-hint describes what inputs the custom agent expects or supports.'),
118 > },
119 > [PromptHeaderAttributes.model]: {
120 > type: 'scalar | sequence',
121 > description: localize('promptHeader.agent.model', 'Specify the model that runs this custom agent. Can also be a list of models. The first available model will be used.'),
122 > },
123 > [PromptHeaderAttributes.tools]: {
124 > type: 'scalar | sequence',
125 > description: localize('promptHeader.agent.tools', 'The set of tools that the custom agent has access to.'),
126 > defaults: ['[]', '[search, edit, web]'],
127 > },
128 > [PromptHeaderAttributes.handOffs]: {
129 > type: 'sequence',
130 > description: localize('promptHeader.agent.handoffs', 'Possible handoff actions when the agent has completed its task.'),
131 > },
132 > [PromptHeaderAttributes.target]: {
133 > type: 'scalar',
134 > description: localize('promptHeader.agent.target', 'The target to which the header attributes like tools apply to. Possible values are `github-copilot` and `vscode`.'),
135 > enums: targetAttributeEnumValues,
136 > },
137 > [PromptHeaderAttributes.infer]: {
138 > type: 'scalar',
139 > description: localize('promptHeader.agent.infer', 'Controls visibility of the agent.'),
140 > enums: booleanAttributeEnumValues,
141 > },
142 > [PromptHeaderAttributes.agents]: {
143 > type: 'sequence',
144 > description: localize('promptHeader.agent.agents', 'One or more agents that this agent can use as subagents. Use \'*\' to specify all available agents.'),
145 > defaults: ['["*"]'],
146 > },
147 > [PromptHeaderAttributes.userInvocable]: {
148 > type: 'scalar',
149 > description: localize('promptHeader.agent.userInvocable', 'Whether the agent can be selected and invoked by users in the UI.'),
150 > enums: booleanAttributeEnumValues,
151 > },
152 > [PromptHeaderAttributes.disableModelInvocation]: {
153 > type: 'scalar',
154 > description: localize('promptHeader.agent.disableModelInvocation', 'If true, prevents the agent from being invoked as a subagent.'),
155 > enums: booleanAttributeEnumValues,
156 > },
157 > [PromptHeaderAttributes.advancedOptions]: {
158 > type: 'map',
159 > description: localize('promptHeader.agent.advancedOptions', 'Advanced options for custom agent behavior.'),
160 > },
161 > [GithubPromptHeaderAttributes.github]: {
162 > type: 'map',
163 > description: localize('promptHeader.agent.github', 'GitHub-specific configuration for the agent, such as token permissions.'),
164 > },
165 > [PromptHeaderAttributes.hooks]: {
166 > type: 'map',
167 > description: localize('promptHeader.agent.hooks', 'Lifecycle hooks scoped to this agent. Define hooks that run only while this agent is active.'),
168 > },
169 > };
170 >
171 > // Attribute metadata for skill files (`SKILL.md`).
172 > export const skillAttributes: Record<string, IAttributeDefinition> = {
173 > [PromptHeaderAttributes.name]: {
174 > type: 'scalar',
175 > description: localize('promptHeader.skill.name', 'The name of the skill.'),
176 > },
177 > [PromptHeaderAttributes.description]: {
178 > type: 'scalar',
179 > description: localize('promptHeader.skill.description', 'The description of the skill. The description is added to every request and will be used by the agent to decide when to load the skill.'),
180 > },
181 > [PromptHeaderAttributes.argumentHint]: {
182 > type: 'scalar',
183 > description: localize('promptHeader.skill.argumentHint', 'Hint shown during autocomplete to indicate expected arguments. Example: [issue-number] or [filename] [format]'),
184 > },
185 > [PromptHeaderAttributes.userInvocable]: {
186 > type: 'scalar',
187 > description: localize('promptHeader.skill.userInvocable', 'Set to false to hide from the / menu. Use for background knowledge users should not invoke directly. Default: true.'),
188 > enums: booleanAttributeEnumValues,
189 > },
190 > [PromptHeaderAttributes.disableModelInvocation]: {
191 > type: 'scalar',
192 > description: localize('promptHeader.skill.disableModelInvocation', 'Set to true to prevent the agent from automatically loading this skill. Use for workflows you want to trigger manually with /name. Default: false.'),
193 > enums: booleanAttributeEnumValues,
194 > },
195 > [PromptHeaderAttributes.license]: {
196 > type: 'scalar | map',
197 > description: localize('promptHeader.skill.license', 'License information for the skill.'),
198 > },
199 > [PromptHeaderAttributes.compatibility]: {
200 > type: 'scalar | map',
201 > description: localize('promptHeader.skill.compatibility', 'Compatibility metadata for environments or runtimes.'),
202 > },
203 > [PromptHeaderAttributes.metadata]: {
204 > type: 'map',
205 > description: localize('promptHeader.skill.metadata', 'Additional metadata for the skill.'),
206 > },
207 > [PromptHeaderAttributes.context]: {
208 > type: 'scalar',
209 > description: localize('promptHeader.skill.context', 'Controls how the skill is loaded. Set to \'fork\' to spawn a subagent with the skill instructions instead of returning them inline.'),
210 > enums: [{ name: 'fork', description: localize('promptHeader.skill.context.fork', 'Spawn a subagent with the skill instructions injected as system context.') }],
211 > },
212 > };
213 >
214 > const allAttributeNames: Record<PromptsType, string[]> = {
215 > [PromptsType.prompt]: Object.keys(promptFileAttributes),
216 > [PromptsType.instructions]: Object.keys(instructionAttributes),
217 > [PromptsType.agent]: Object.keys(customAgentAttributes),
218 > [PromptsType.skill]: Object.keys(skillAttributes),
219 > [PromptsType.hook]: [], // hooks are JSON files, not markdown with YAML frontmatter
220 > };
221 > const githubCopilotAgentAttributeNames = [PromptHeaderAttributes.name, PromptHeaderAttributes.description, PromptHeaderAttributes.tools, PromptHeaderAttributes.target, GithubPromptHeaderAttributes.mcpServers, GithubPromptHeaderAttributes.github, PromptHeaderAttributes.infer];
222 > const recommendedAttributeNames: Record<PromptsType, string[]> = {
223 > [PromptsType.prompt]: allAttributeNames[PromptsType.prompt].filter(name => !isNonRecommendedAttribute(name)),
224 > [PromptsType.instructions]: allAttributeNames[PromptsType.instructions].filter(name => !isNonRecommendedAttribute(name)),
225 > [PromptsType.agent]: allAttributeNames[PromptsType.agent].filter(name => !isNonRecommendedAttribute(name)),
226 > [PromptsType.skill]: allAttributeNames[PromptsType.skill].filter(name => !isNonRecommendedAttribute(name)),
227 > [PromptsType.hook]: [], // hooks are JSON files, not markdown with YAML frontmatter
228 > };
229 >
230 > export function getValidAttributeNames(promptType: PromptsType, includeNonRecommended: boolean, target: Target): string[] {
231 if (target === Target.Claude) {
232 if (promptType === PromptsType.instructions) {
241 return includeNonRecommended ? allAttributeNames[promptType] : recommendedAttributeNames[promptType];
242 }
244 > export function isNonRecommendedAttribute(attributeName: string): boolean {
245 > return attributeName === PromptHeaderAttributes.advancedOptions || attributeName === PromptHeaderAttributes.excludeAgent || attributeName === PromptHeaderAttributes.mode || attributeName === PromptHeaderAttributes.infer;
246 > }
247 >
248 > export function getAttributeDefinition(attributeName: string, promptType: PromptsType, target: Target): IAttributeDefinition | undefined {
249 switch (promptType) {
250 case PromptsType.instructions:
266 }
267 }
269 > // The list of tools known to be used by GitHub Copilot custom agents
270 > export const knownGithubCopilotTools = [
271 > { name: SpecedToolAliases.execute, description: localize('githubCopilot.execute', 'Execute commands') },
272 > { name: SpecedToolAliases.read, description: localize('githubCopilot.read', 'Read files') },
273 > { name: SpecedToolAliases.edit, description: localize('githubCopilot.edit', 'Edit files') },
274 > { name: SpecedToolAliases.search, description: localize('githubCopilot.search', 'Search files') },
275 > { name: SpecedToolAliases.agent, description: localize('githubCopilot.agent', 'Use subagents') },
276 > ];
277 >
278 > export interface IValueEntry {
279 > readonly name: string;
280 > readonly description?: string;
281 > }
282 >
283 > export const knownClaudeTools = [
284 > { name: 'Bash', description: localize('claude.bash', 'Execute shell commands'), toolEquivalent: [SpecedToolAliases.execute] },
285 > { name: 'Edit', description: localize('claude.edit', 'Make targeted file edits'), toolEquivalent: ['edit/editNotebook', 'edit/editFiles'] },
286 > { name: 'Glob', description: localize('claude.glob', 'Find files by pattern'), toolEquivalent: ['search/fileSearch'] },
287 > { name: 'Grep', description: localize('claude.grep', 'Search file contents with regex'), toolEquivalent: ['search/textSearch'] },
288 > { name: 'Read', description: localize('claude.read', 'Read file contents'), toolEquivalent: ['read/readFile', 'read/getNotebookSummary'] },
289 > { name: 'Write', description: localize('claude.write', 'Create/overwrite files'), toolEquivalent: ['edit/createDirectory', 'edit/createFile', 'edit/createJupyterNotebook'] },
290 > { name: 'WebFetch', description: localize('claude.webFetch', 'Fetch URL content'), toolEquivalent: [SpecedToolAliases.web] },
291 > { name: 'WebSearch', description: localize('claude.webSearch', 'Perform web searches'), toolEquivalent: [SpecedToolAliases.web] },
292 > { name: 'Task', description: localize('claude.task', 'Run subagents for complex tasks'), toolEquivalent: [SpecedToolAliases.agent] },
293 > { name: 'Skill', description: localize('claude.skill', 'Execute skills'), toolEquivalent: [] },
294 > { name: 'LSP', description: localize('claude.lsp', 'Code intelligence (requires plugin)'), toolEquivalent: [] },
295 > { name: 'NotebookEdit', description: localize('claude.notebookEdit', 'Modify Jupyter notebooks'), toolEquivalent: ['edit/editNotebook'] },
296 > { name: 'AskUserQuestion', description: localize('claude.askUserQuestion', 'Ask multiple-choice questions'), toolEquivalent: ['vscode/askQuestions'] },
297 > { name: 'MCPSearch', description: localize('claude.mcpSearch', 'Searches for MCP tools when tool search is enabled'), toolEquivalent: [] }
298 > ];
299 >
300 > export const knownClaudeModels = [
301 > { name: 'sonnet', description: localize('claude.sonnet', 'Latest Claude Sonnet'), modelEquivalent: 'Claude Sonnet 4.5 (copilot)' },
302 > { name: 'opus', description: localize('claude.opus', 'Latest Claude Opus'), modelEquivalent: 'Claude Opus 4.6 (copilot)' },
303 > { name: 'haiku', description: localize('claude.haiku', 'Latest Claude Haiku, fast for simple tasks'), modelEquivalent: 'Claude Haiku 4.5 (copilot)' },
304 > { name: 'inherit', description: localize('claude.inherit', 'Inherit model from parent agent or prompt'), modelEquivalent: undefined },
305 > ];
306 >
307 > export function mapClaudeModels(claudeModelNames: readonly string[]): readonly string[] {
308 const result = [];
309 for (const name of claudeModelNames) {
315 return result;
316 }
318 > /**
319 > * Maps Claude tool names to their VS Code tool equivalents.
320 > */
321 > export function mapClaudeTools(claudeToolNames: readonly string[]): string[] {
322 const result: string[] = [];
323 for (const name of claudeToolNames) {
329 return result;
330 }
332 > export const claudeAgentAttributes: Record<string, IAttributeDefinition> = {
333 > 'name': {
334 > type: 'scalar',
335 > description: localize('attribute.name', "Unique identifier using lowercase letters and hyphens (required)"),
336 > },
337 > 'description': {
338 > type: 'scalar',
339 > description: localize('attribute.description', "When to delegate to this subagent (required)"),
340 > },
341 > 'tools': {
342 > type: 'sequence',
343 > description: localize('attribute.tools', "Array of tools the subagent can use. Inherits all tools if omitted"),
344 > defaults: ['Read, Edit, Bash'],
345 > items: knownClaudeTools
346 > },
347 > 'disallowedTools': {
348 > type: 'sequence',
349 > description: localize('attribute.disallowedTools', "Tools to deny, removed from inherited or specified list"),
350 > defaults: ['Write, Edit, Bash'],
351 > items: knownClaudeTools
352 > },
353 > 'model': {
354 > type: 'scalar',
355 > description: localize('attribute.model', "Model to use: sonnet, opus, haiku, or inherit. Defaults to inherit."),
356 > defaults: ['sonnet', 'opus', 'haiku', 'inherit'],
357 > enums: knownClaudeModels
358 > },
359 > 'permissionMode': {
360 > type: 'scalar',
361 > description: localize('attribute.permissionMode', "Permission mode: default, acceptEdits, dontAsk, bypassPermissions, or plan."),
362 > defaults: ['default', 'acceptEdits', 'dontAsk', 'bypassPermissions', 'plan'],
363 > enums: [
364 > { name: 'default', description: localize('claude.permissionMode.default', 'Standard behavior: prompts for permission on first use of each tool.') },
365 > { name: 'acceptEdits', description: localize('claude.permissionMode.acceptEdits', 'Automatically accepts file edit permissions for the session.') },
366 > { name: 'plan', description: localize('claude.permissionMode.plan', 'Plan Mode: Claude can analyze but not modify files or execute commands.') },
367 > { name: 'delegate', description: localize('claude.permissionMode.delegate', 'Coordination-only mode for agent team leads. Only available when an agent team is active.') },
368 > { name: 'dontAsk', description: localize('claude.permissionMode.dontAsk', 'Auto-denies tools unless pre-approved via /permissions or permissions.allow rules.') },
369 > { name: 'bypassPermissions', description: localize('claude.permissionMode.bypassPermissions', 'Skips all permission prompts (requires safe environment like containers).') }
370 > ]
371 > },
372 > 'skills': {
373 > type: 'sequence',
374 > description: localize('attribute.skills', "Skills to load into the subagent's context at startup."),
375 > },
376 > 'mcpServers': {
377 > type: 'sequence',
378 > description: localize('attribute.mcpServers', "MCP servers available to this subagent."),
379 > },
380 > 'hooks': {
381 > type: 'object',
382 > description: localize('attribute.hooks', "Lifecycle hooks scoped to this subagent."),
383 > },
384 > 'memory': {
385 > type: 'scalar',
386 > description: localize('attribute.memory', "Persistent memory scope: user, project, or local. Enables cross-session learning."),
387 > defaults: ['user', 'project', 'local'],
388 > enums: [
389 > { name: 'user', description: localize('claude.memory.user', "Remember learnings across all projects.") },
390 > { name: 'project', description: localize('claude.memory.project', "The subagent's knowledge is project-specific and shareable via version control.") },
391 > { name: 'local', description: localize('claude.memory.local', "The subagent's knowledge is project-specific but should not be checked into version control.") }
392 > ]
393 > }
394 > };
395 >
396 > /**
397 > * Attributes supported in Claude rules files (`.claude/rules/*.md`).
398 > * Claude rules use `paths` instead of `applyTo` for glob patterns.
399 > */
400 > export const claudeRulesAttributes: Record<string, IAttributeDefinition> = {
401 > 'description': {
402 > type: 'scalar',
403 > description: localize('attribute.rules.description', "A description of what this rule covers, used to provide context about when it applies."),
404 > },
405 > 'paths': {
406 > type: 'sequence',
407 > description: localize('attribute.rules.paths', "Array of glob patterns that describe for which files the rule applies. Based on these patterns, the file is automatically included in the prompt when the context contains a file that matches.\nExample: `['src/**/*.ts', 'test/**']`"),
408 > },
409 > };
410 >
411 > export function isVSCodeOrDefaultTarget(target: Target): boolean {
412 return target === Target.VSCode || target === Target.Undefined;
413 }
415 > export function getTarget(promptType: PromptsType, header: PromptHeader | URI): Target {
416 const uri = header instanceof URI ? header : header.uri;
417 if (promptType === PromptsType.agent) {
src/vs/platform/workspace/common/workspace.ts 360 covered LOC · 34 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workspace.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../../nls.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { basename, extname } from '../../../base/common/path.js';
9 > import { TernarySearchTree } from '../../../base/common/ternarySearchTree.js';
10 > import { extname as resourceExtname, basenameOrAuthority, joinPath, extUriBiasedIgnorePathCase } from '../../../base/common/resources.js';
11 > import { URI, UriComponents } from '../../../base/common/uri.js';
12 > import { createDecorator } from '../../instantiation/common/instantiation.js';
13 > import { IEnvironmentService } from '../../environment/common/environment.js';
14 > import { Schemas } from '../../../base/common/network.js';
15 >
16 > export const IWorkspaceContextService = createDecorator<IWorkspaceContextService>('contextService');
17 >
18 > export interface IWorkspaceContextService {
19 >
20 > readonly _serviceBrand: undefined;
21 >
22 > /**
23 > * An event which fires on workbench state changes.
24 > */
25 > readonly onDidChangeWorkbenchState: Event<WorkbenchState>;
26 >
27 > /**
28 > * An event which fires on workspace name changes.
29 > */
30 > readonly onDidChangeWorkspaceName: Event<void>;
31 >
32 > /**
33 > * An event which fires before workspace folders change.
34 > */
35 > readonly onWillChangeWorkspaceFolders: Event<IWorkspaceFoldersWillChangeEvent>;
36 >
37 > /**
38 > * An event which fires on workspace folders change.
39 > */
40 > readonly onDidChangeWorkspaceFolders: Event<IWorkspaceFoldersChangeEvent>;
41 >
42 > /**
43 > * Provides access to the complete workspace object.
44 > */
45 > getCompleteWorkspace(): Promise<IWorkspace>;
46 >
47 > /**
48 > * Provides access to the workspace object the window is running with.
49 > * Use `getCompleteWorkspace` to get complete workspace object.
50 > */
51 > getWorkspace(): IWorkspace;
52 >
53 > /**
54 > * Return the state of the workbench.
55 > *
56 > * WorkbenchState.EMPTY - if the workbench was opened with empty window or file
57 > * WorkbenchState.FOLDER - if the workbench was opened with a folder
58 > * WorkbenchState.WORKSPACE - if the workbench was opened with a workspace
59 > */
60 > getWorkbenchState(): WorkbenchState;
61 >
62 > /**
63 > * Returns the folder for the given resource from the workspace.
64 > * Can be null if there is no workspace or the resource is not inside the workspace.
65 > */
66 > getWorkspaceFolder(resource: URI): IWorkspaceFolder | null;
67 >
68 > /**
69 > * Return `true` if the current workspace has the given identifier or root URI otherwise `false`.
70 > */
71 > isCurrentWorkspace(workspaceIdOrFolder: IWorkspaceIdentifier | ISingleFolderWorkspaceIdentifier | URI): boolean;
72 >
73 > /**
74 > * Returns if the provided resource is inside the workspace or not.
75 > */
76 > isInsideWorkspace(resource: URI): boolean;
77 >
78 > /**
79 > * Return `true` if the current workspace has data (e.g. folders or a workspace configuration) that can be sent to the extension host, otherwise `false`.
80 > */
81 > hasWorkspaceData(): boolean;
82 > }
83 >
84 > export interface IResolvedWorkspace extends IWorkspaceIdentifier, IBaseWorkspace {
85 > readonly folders: IWorkspaceFolder[];
86 > }
87 >
88 > export interface IBaseWorkspace {
89 >
90 > /**
91 > * If present, marks the window that opens the workspace
92 > * as a remote window with the given authority.
93 > */
94 > readonly remoteAuthority?: string;
95 >
96 > /**
97 > * Transient workspaces are meant to go away after being used
98 > * once, e.g. a window reload of a transient workspace will
99 > * open an empty window.
100 > *
101 > * See: https://github.com/microsoft/vscode/issues/119695
102 > */
103 > readonly transient?: boolean;
104 > }
105 >
106 > export interface IBaseWorkspaceIdentifier {
107 >
108 > /**
109 > * Every workspace (multi-root, single folder or empty)
110 > * has a unique identifier. It is not possible to open
111 > * a workspace with the same `id` in multiple windows
112 > */
113 > readonly id: string;
114 > }
115 >
116 > /**
117 > * A single folder workspace identifier is a path to a folder + id.
118 > */
119 > export interface ISingleFolderWorkspaceIdentifier extends IBaseWorkspaceIdentifier {
120 >
121 > /**
122 > * Folder path as `URI`.
123 > */
124 > readonly uri: URI;
125 > }
126 >
127 > /**
128 > * A multi-root workspace identifier is a path to a workspace file + id.
129 > */
130 > export interface IWorkspaceIdentifier extends IBaseWorkspaceIdentifier {
131 >
132 > /**
133 > * Workspace config file path as `URI`.
134 > */
135 > configPath: URI;
136 > }
137 >
138 > export interface IEmptyWorkspaceIdentifier extends IBaseWorkspaceIdentifier { }
139 >
140 > export type IAnyWorkspaceIdentifier = IWorkspaceIdentifier | ISingleFolderWorkspaceIdentifier | IEmptyWorkspaceIdentifier;
141 >
142 > export function isSingleFolderWorkspaceIdentifier(obj: unknown): obj is ISingleFolderWorkspaceIdentifier {
143 const singleFolderIdentifier = obj as ISingleFolderWorkspaceIdentifier | undefined;
144
145 return typeof singleFolderIdentifier?.id === 'string' && URI.isUri(singleFolderIdentifier.uri);
146 }
147 > workspace.ts
148 > export function isEmptyWorkspaceIdentifier(obj: unknown): obj is IEmptyWorkspaceIdentifier {
149 const emptyWorkspaceIdentifier = obj as IEmptyWorkspaceIdentifier | undefined;
150 return typeof emptyWorkspaceIdentifier?.id === 'string'
152 && !isWorkspaceIdentifier(obj);
153 }
154 > workspace.ts
155 > export const EXTENSION_DEVELOPMENT_EMPTY_WINDOW_WORKSPACE: IEmptyWorkspaceIdentifier = { id: 'ext-dev' };
156 > export const UNKNOWN_EMPTY_WINDOW_WORKSPACE: IEmptyWorkspaceIdentifier = { id: 'empty-window' };
157 >
158 > export function toWorkspaceIdentifier(workspace: IWorkspace): IAnyWorkspaceIdentifier;
159 > export function toWorkspaceIdentifier(backupPath: string | undefined, isExtensionDevelopment: boolean): IEmptyWorkspaceIdentifier;
160 > export function toWorkspaceIdentifier(arg0: IWorkspace | string | undefined, isExtensionDevelopment?: boolean): IAnyWorkspaceIdentifier {
161
162 // Empty workspace
202 };
203 }
204 > workspace.ts
205 > export function isWorkspaceIdentifier(obj: unknown): obj is IWorkspaceIdentifier {
206 const workspaceIdentifier = obj as IWorkspaceIdentifier | undefined;
207
208 return typeof workspaceIdentifier?.id === 'string' && URI.isUri(workspaceIdentifier.configPath);
209 }
210 > workspace.ts
211 > export interface ISerializedSingleFolderWorkspaceIdentifier extends IBaseWorkspaceIdentifier {
212 > readonly uri: UriComponents;
213 > }
214 >
215 > export interface ISerializedWorkspaceIdentifier extends IBaseWorkspaceIdentifier {
216 > readonly configPath: UriComponents;
217 > }
218 >
219 > export function reviveIdentifier(identifier: undefined): undefined;
220 > export function reviveIdentifier(identifier: ISerializedWorkspaceIdentifier): IWorkspaceIdentifier;
221 > export function reviveIdentifier(identifier: ISerializedSingleFolderWorkspaceIdentifier): ISingleFolderWorkspaceIdentifier;
222 > export function reviveIdentifier(identifier: IEmptyWorkspaceIdentifier): IEmptyWorkspaceIdentifier;
223 > export function reviveIdentifier(identifier: ISerializedWorkspaceIdentifier | ISerializedSingleFolderWorkspaceIdentifier | IEmptyWorkspaceIdentifier | undefined): IAnyWorkspaceIdentifier | undefined;
224 > export function reviveIdentifier(identifier: ISerializedWorkspaceIdentifier | ISerializedSingleFolderWorkspaceIdentifier | IEmptyWorkspaceIdentifier | undefined): IAnyWorkspaceIdentifier | undefined {
225
226 // Single Folder
243 return undefined;
244 }
245 > workspace.ts
246 > export const enum WorkbenchState {
247 > EMPTY = 1,
248 > FOLDER,
249 > WORKSPACE
250 > }
251 >
252 > export interface IWorkspaceFoldersWillChangeEvent {
253 >
254 > readonly changes: IWorkspaceFoldersChangeEvent;
255 > readonly fromCache: boolean;
256 >
257 > join(promise: Promise<void>): void;
258 > }
259 >
260 > export interface IWorkspaceFoldersChangeEvent {
261 > added: IWorkspaceFolder[];
262 > removed: IWorkspaceFolder[];
263 > changed: IWorkspaceFolder[];
264 > }
265 >
266 > export interface IWorkspace {
267 >
268 > /**
269 > * the unique identifier of the workspace.
270 > */
271 > readonly id: string;
272 >
273 > /**
274 > * Folders in the workspace.
275 > */
276 > readonly folders: IWorkspaceFolder[];
277 >
278 > /**
279 > * Transient workspaces are meant to go away after being used
280 > * once, e.g. a window reload of a transient workspace will
281 > * open an empty window.
282 > */
283 > readonly transient?: boolean;
284 >
285 > /**
286 > * the location of the workspace configuration
287 > */
288 > readonly configuration?: URI | null;
289 >
290 > /**
291 > * Optional display name for the workspace.
292 > */
293 > readonly name?: string;
294 >
295 > }
296 >
297 > export function isWorkspace(thing: unknown): thing is IWorkspace {
298 const candidate = thing as IWorkspace | undefined;
299
302 && Array.isArray(candidate.folders));
303 }
304 > workspace.ts
305 > export interface IWorkspaceFolderData {
306 >
307 > /**
308 > * The associated URI for this workspace folder.
309 > */
310 > readonly uri: URI;
311 >
312 > /**
313 > * The name of this workspace folder. Defaults to
314 > * the basename of its [uri-path](#Uri.path)
315 > */
316 > readonly name: string;
317 >
318 > /**
319 > * The ordinal number of this workspace folder.
320 > */
321 > readonly index: number;
322 > }
323 >
324 > export interface IWorkspaceFolder extends IWorkspaceFolderData {
325 >
326 > /**
327 > * Given workspace folder relative path, returns the resource with the absolute path.
328 > */
329 > toResource: (relativePath: string) => URI;
330 > }
331 >
332 > export function isWorkspaceFolder(thing: unknown): thing is IWorkspaceFolder {
333 const candidate = thing as IWorkspaceFolder;
334
338 && typeof candidate.toResource === 'function');
339 }
340 > workspace.ts
341 > export class Workspace implements IWorkspace {
342 >
343 > private foldersMap: TernarySearchTree<URI, WorkspaceFolder>;
344 >
345 > private _folders!: WorkspaceFolder[];
346 > get folders(): WorkspaceFolder[] { return this._folders; }
347 > set folders(folders: WorkspaceFolder[]) {
348 > this._folders = folders; workspace.ts
349 > this.updateFoldersMap();
350 > }
351 > workspace.ts
352 > constructor(
353 > private _id: string, workspace.ts
354 > folders: WorkspaceFolder[],
355 > private _transient: boolean,
356 > private _configuration: URI | null,
357 > private ignorePathCasing: (key: URI) => boolean,
358 > private _workspaceName?: string,
359 > ) {
360 > this.foldersMap = TernarySearchTree.forUris<WorkspaceFolder>(this.ignorePathCasing, () => true);
361 > this.folders = folders;
362 > }
363 > workspace.ts
364 > update(workspace: Workspace) {
365 this._id = workspace.id;
366 this._configuration = workspace.configuration;
370 this.folders = workspace.folders;
371 }
372 > workspace.ts
373 > get id(): string {
374 > return this._id; workspace.ts
375 > }
376 > workspace.ts
377 > get transient(): boolean {
378 return this._transient;
379 }
380 > workspace.ts
381 > get configuration(): URI | null {
382 > return this._configuration; workspace.ts
383 > }
384 > workspace.ts
385 > set configuration(configuration: URI | null) {
386 this._configuration = configuration;
387 }
388 > workspace.ts
389 > get name(): string | undefined {
390 return this._workspaceName;
391 }
392 > workspace.ts
393 > getFolder(resource: URI): IWorkspaceFolder | null {
394 if (!resource) {
395 return null;
398 return this.foldersMap.findSubstr(resource) || null;
399 }
400 > workspace.ts
401 > private updateFoldersMap(): void {
402 > this.foldersMap = TernarySearchTree.forUris<WorkspaceFolder>(this.ignorePathCasing, () => true); workspace.ts
403 > for (const folder of this.folders) {
404 > this.foldersMap.set(folder.uri, folder);
405 > }
406 > }
407 > workspace.ts
408 > toJSON(): IWorkspace {
409 return { id: this.id, folders: this.folders, transient: this.transient, configuration: this.configuration, name: this.name };
410 }
411 > } workspace.ts
412 >
413 > export interface IRawFileWorkspaceFolder {
414 > readonly path: string;
415 > name?: string;
416 > }
417 >
418 > export interface IRawUriWorkspaceFolder {
419 > readonly uri: string;
420 > name?: string;
421 > }
422 >
423 > export class WorkspaceFolder implements IWorkspaceFolder {
424 >
425 > readonly uri: URI;
426 > readonly name: string;
427 > readonly index: number;
428 >
429 > constructor(
430 > data: IWorkspaceFolderData, workspace.ts
431 > /**
432 > * Provides access to the original metadata for this workspace
433 > * folder. This can be different from the metadata provided in
434 > * this class:
435 > * - raw paths can be relative
436 > * - raw paths are not normalized
437 > */
438 > readonly raw?: IRawFileWorkspaceFolder | IRawUriWorkspaceFolder
439 > ) {
440 > this.uri = data.uri;
441 > this.index = data.index;
442 > this.name = data.name;
443 > }
444 > workspace.ts
445 > toResource(relativePath: string): URI {
446 return joinPath(this.uri, relativePath);
447 }
448 > workspace.ts
449 > toJSON(): IWorkspaceFolderData {
450 return { uri: this.uri, name: this.name, index: this.index };
451 }
452 > } workspace.ts
453 >
454 > export function toWorkspaceFolder(resource: URI): WorkspaceFolder {
455 > return new WorkspaceFolder({ uri: resource, index: 0, name: basenameOrAuthority(resource) }, { uri: resource.toString() }); workspace.ts
456 > }
457 > workspace.ts
458 > export const WORKSPACE_EXTENSION = 'code-workspace';
459 > export const WORKSPACE_SUFFIX = `.${WORKSPACE_EXTENSION}`;
460 > export const WORKSPACE_FILTER = [{ name: localize('codeWorkspace', "Code Workspace"), extensions: [WORKSPACE_EXTENSION] }];
461 > export const UNTITLED_WORKSPACE_NAME = 'workspace.json';
462 >
463 > export function isUntitledWorkspace(path: URI, environmentService: IEnvironmentService): boolean {
464 return extUriBiasedIgnorePathCase.isEqualOrParent(path, environmentService.untitledWorkspacesHome);
465 }
466 > workspace.ts
467 > export function isTemporaryWorkspace(workspace: IWorkspace): boolean;
468 > export function isTemporaryWorkspace(path: URI): boolean;
469 > export function isTemporaryWorkspace(arg1: IWorkspace | URI): boolean {
470 let path: URI | null | undefined;
471 if (URI.isUri(arg1)) {
477 return path?.scheme === Schemas.tmp;
478 }
479 > workspace.ts
480 > export const STANDALONE_EDITOR_WORKSPACE_ID = '4064f6ec-cb38-4ad0-af64-ee6467e63c82';
481 > export function isStandaloneEditorWorkspace(workspace: IWorkspace): boolean {
482 return workspace.id === STANDALONE_EDITOR_WORKSPACE_ID;
483 }
484 > workspace.ts
485 > export function isSavedWorkspace(path: URI, environmentService: IEnvironmentService): boolean {
486 return !isUntitledWorkspace(path, environmentService) && !isTemporaryWorkspace(path);
487 }
488 > workspace.ts
489 > export function hasWorkspaceFileExtension(path: string | URI) {
490 const ext = (typeof path === 'string') ? extname(path) : resourceExtname(path);
491
src/vs/base/common/ternarySearchTree.ts 345 covered LOC · 103 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- ternarySearchTree.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { shuffle } from './arrays.js';
7 > import { assert } from './assert.js';
8 > import { CharCode } from './charCode.js';
9 > import { compare, compareIgnoreCase, compareSubstring, compareSubstringIgnoreCase } from './strings.js';
10 > import { URI } from './uri.js';
11 >
12 > export interface IKeyIterator<K> {
13 > reset(key: K): this;
14 > next(): this;
15 >
16 > hasNext(): boolean;
17 > cmp(a: string): number;
18 > value(): string;
19 > }
20 >
21 > export class StringIterator implements IKeyIterator<string> {
22
23 private _value: string = '';
24 private _pos: number = 0;
26 > reset(key: string): this {
27 this._value = key;
28 this._pos = 0;
29 return this;
30 }
32 > next(): this {
33 this._pos += 1;
34 return this;
35 }
37 > hasNext(): boolean {
38 return this._pos < this._value.length - 1;
39 }
41 > cmp(a: string): number {
42 const aCode = a.charCodeAt(0);
43 const thisCode = this._value.charCodeAt(this._pos);
44 return aCode - thisCode;
45 }
47 > value(): string {
48 return this._value[this._pos];
49 }
51 >
52 > export class ConfigKeysIterator implements IKeyIterator<string> {
53 >
54 > private _value!: string;
55 > private _from!: number;
56 > private _to!: number;
57 >
58 > constructor(
59 private readonly _caseSensitive: boolean = true
60 ) { }
62 > reset(key: string): this {
63 this._value = key;
64 this._from = 0;
66 return this.next();
67 }
69 > hasNext(): boolean {
70 return this._to < this._value.length;
71 }
73 > next(): this {
74 // this._data = key.split(/[\\/]/).filter(s => !!s);
75 this._from = this._to;
89 return this;
90 }
92 > cmp(a: string): number {
93 return this._caseSensitive
94 ? compareSubstring(a, this._value, 0, a.length, this._from, this._to)
95 : compareSubstringIgnoreCase(a, this._value, 0, a.length, this._from, this._to);
96 }
98 > value(): string {
99 return this._value.substring(this._from, this._to);
100 }
102 >
103 > export class PathIterator implements IKeyIterator<string> {
104 >
105 > private _value!: string;
106 > private _valueLen!: number;
107 > private _from!: number;
108 > private _to!: number;
109 >
110 > constructor(
111 > private readonly _splitOnBackslash: boolean = true, ternarySearchTree.ts
112 > private readonly _caseSensitive: boolean = true
113 > ) { }
115 > reset(key: string): this {
116 > this._from = 0; ternarySearchTree.ts
117 > this._to = 0;
118 > this._value = key;
119 > this._valueLen = key.length;
120 > for (let pos = key.length - 1; pos >= 0; pos--, this._valueLen--) {
121 > const ch = this._value.charCodeAt(pos);
122 > if (!(ch === CharCode.Slash || this._splitOnBackslash && ch === CharCode.Backslash)) {
123 > break;
124 > }
125 > }
126 >
127 > return this.next();
128 > }
130 > hasNext(): boolean {
131 > return this._to < this._valueLen; ternarySearchTree.ts
132 > }
134 > next(): this {
135 > // this._data = key.split(/[\\/]/).filter(s => !!s); ternarySearchTree.ts
136 > this._from = this._to;
137 > let justSeps = true;
138 > for (; this._to < this._valueLen; this._to++) {
139 > const ch = this._value.charCodeAt(this._to);
140 > if (ch === CharCode.Slash || this._splitOnBackslash && ch === CharCode.Backslash) {
141 > if (justSeps) { ternarySearchTree.ts
142 > this._from++;
143 > } else {
144 break;
145 }
146 > } else { ternarySearchTree.ts
147 > justSeps = false;
148 > }
149 > }
150 > return this;
151 > }
153 > cmp(a: string): number {
154 > return this._caseSensitive ternarySearchTree.ts
155 > ? compareSubstring(a, this._value, 0, a.length, this._from, this._to) ternarySearchTree.ts
156 : compareSubstringIgnoreCase(a, this._value, 0, a.length, this._from, this._to);
159 > value(): string {
160 > return this._value.substring(this._from, this._to); ternarySearchTree.ts
161 > }
163 >
164 > const enum UriIteratorState {
165 > Scheme = 1, Authority = 2, Path = 3, Query = 4, Fragment = 5
166 > }
167 >
168 > export class UriIterator implements IKeyIterator<URI> {
169 >
170 > private _pathIterator!: PathIterator;
171 > private _value!: URI;
172 > private _states: UriIteratorState[] = [];
173 > private _stateIdx: number = 0;
174 >
175 > constructor(
176 > private readonly _ignorePathCasing: (uri: URI) => boolean, ternarySearchTree.ts
177 > private readonly _ignoreQueryAndFragment: (uri: URI) => boolean) { }
179 > reset(key: URI): this {
180 > this._value = key; ternarySearchTree.ts
181 > this._states = [];
182 > if (this._value.scheme) {
183 > this._states.push(UriIteratorState.Scheme);
184 > }
185 > if (this._value.authority) {
186 this._states.push(UriIteratorState.Authority);
187 }
188 > if (this._value.path) { ternarySearchTree.ts
189 > this._pathIterator = new PathIterator(false, !this._ignorePathCasing(key));
190 > this._pathIterator.reset(key.path);
191 > if (this._pathIterator.value()) {
192 > this._states.push(UriIteratorState.Path);
193 > }
194 > }
195 > if (!this._ignoreQueryAndFragment(key)) {
196 if (this._value.query) {
197 this._states.push(UriIteratorState.Query);
201 }
202 }
203 > this._stateIdx = 0; ternarySearchTree.ts
204 > return this;
205 > }
207 > next(): this {
208 > if (this._states[this._stateIdx] === UriIteratorState.Path && this._pathIterator.hasNext()) { ternarySearchTree.ts
209 this._pathIterator.next();
210 > } else { ternarySearchTree.ts
211 > this._stateIdx += 1;
212 > }
213 > return this;
214 > }
216 > hasNext(): boolean {
217 > return (this._states[this._stateIdx] === UriIteratorState.Path && this._pathIterator.hasNext()) ternarySearchTree.ts
218 > || this._stateIdx < this._states.length - 1;
219 > }
221 > cmp(a: string): number {
222 > if (this._states[this._stateIdx] === UriIteratorState.Scheme) { ternarySearchTree.ts
223 > return compareIgnoreCase(a, this._value.scheme);
224 > } else if (this._states[this._stateIdx] === UriIteratorState.Authority) {
225 return compareIgnoreCase(a, this._value.authority);
226 > } else if (this._states[this._stateIdx] === UriIteratorState.Path) { ternarySearchTree.ts
227 > return this._pathIterator.cmp(a); ternarySearchTree.ts
228 > } else if (this._states[this._stateIdx] === UriIteratorState.Query) { ternarySearchTree.ts
229 return compare(a, this._value.query);
230 } else if (this._states[this._stateIdx] === UriIteratorState.Fragment) {
232 }
233 throw new Error();
236 > value(): string {
237 > if (this._states[this._stateIdx] === UriIteratorState.Scheme) { ternarySearchTree.ts
238 > return this._value.scheme;
239 > } else if (this._states[this._stateIdx] === UriIteratorState.Authority) {
240 return this._value.authority;
241 > } else if (this._states[this._stateIdx] === UriIteratorState.Path) { ternarySearchTree.ts
242 > return this._pathIterator.value();
243 > } else if (this._states[this._stateIdx] === UriIteratorState.Query) {
244 return this._value.query;
245 } else if (this._states[this._stateIdx] === UriIteratorState.Fragment) {
247 }
248 throw new Error();
251 >
252 > abstract class Undef {
253 >
254 > static readonly Val: unique symbol = Symbol('undefined_placeholder');
255 >
256 > static wrap<V>(value: V | undefined): V | typeof Undef.Val {
257 > return value === undefined ? Undef.Val : value; ternarySearchTree.ts
258 > }
260 > static unwrap<V>(value: V | typeof Undef.Val): V | undefined {
261 > return value === Undef.Val ? undefined : value; ternarySearchTree.ts
262 > }
264 >
265 > class TernarySearchTreeNode<K, V> { ternarySearchTree.ts
266 > height: number = 1;
267 > segment!: string;
268 > value: V | typeof Undef.Val | undefined = undefined;
269 > key: K | undefined = undefined;
270 > left: TernarySearchTreeNode<K, V> | undefined = undefined;
271 > mid: TernarySearchTreeNode<K, V> | undefined = undefined;
272 > right: TernarySearchTreeNode<K, V> | undefined = undefined;
274 > isEmpty(): boolean {
275 return !this.left && !this.mid && !this.right && this.value === undefined;
276 }
278 > rotateLeft() {
279 const tmp = this.right!;
280 this.right = tmp.left;
284 return tmp;
285 }
287 > rotateRight() {
288 const tmp = this.left!;
289 this.left = tmp.right;
293 return tmp;
294 }
296 > updateHeight() {
297 > this.height = 1 + Math.max(this.heightLeft, this.heightRight); ternarySearchTree.ts
298 > }
300 > balanceFactor() {
301 > return this.heightRight - this.heightLeft; ternarySearchTree.ts
302 > }
304 > get heightLeft() {
305 > return this.left?.height ?? 0; ternarySearchTree.ts
306 > }
308 > get heightRight() {
309 > return this.right?.height ?? 0; ternarySearchTree.ts
310 > }
312 >
313 > const enum Dir {
314 > Left = -1,
315 > Mid = 0,
316 > Right = 1
317 > }
318 >
319 > export class TernarySearchTree<K, V> {
320 >
321 > static forUris<E>(ignorePathCasing: (key: URI) => boolean = () => false, ignoreQueryAndFragment: (key: URI) => boolean = () => false): TernarySearchTree<URI, E> {
322 > return new TernarySearchTree<URI, E>(new UriIterator(ignorePathCasing, ignoreQueryAndFragment)); ternarySearchTree.ts
323 > }
325 > static forPaths<E>(ignorePathCasing = false): TernarySearchTree<string, E> {
326 > return new TernarySearchTree<string, E>(new PathIterator(undefined, !ignorePathCasing)); ternarySearchTree.ts
327 > }
329 > static forStrings<E>(): TernarySearchTree<string, E> {
330 return new TernarySearchTree<string, E>(new StringIterator());
331 }
333 > static forConfigKeys<E>(): TernarySearchTree<string, E> {
334 return new TernarySearchTree<string, E>(new ConfigKeysIterator());
335 }
337 > private _iter: IKeyIterator<K>;
338 > private _root: TernarySearchTreeNode<K, V> | undefined;
339 >
340 > constructor(segments: IKeyIterator<K>) {
341 > this._iter = segments; ternarySearchTree.ts
342 > }
344 > clear(): void {
345 this._root = undefined;
346 }
348 > /**
349 > * Fill the tree with the same value of the given keys
350 > */
351 > fill(element: V, keys: readonly K[]): void;
352 > /**
353 > * Fill the tree with given [key,value]-tuples
354 > */
355 > fill(values: readonly [K, V][]): void;
356 > fill(values: readonly [K, V][] | V, keys?: readonly K[]): void {
357 if (keys) {
358 const arr = keys.slice(0);
369 }
370 }
372 > set(key: K, element: V): V | undefined {
373 > const iter = this._iter.reset(key); ternarySearchTree.ts
374 > let node: TernarySearchTreeNode<K, V>;
375 >
376 > if (!this._root) {
377 > this._root = new TernarySearchTreeNode<K, V>();
378 > this._root.segment = iter.value();
379 > }
380 > const stack: [Dir, TernarySearchTreeNode<K, V>][] = [];
381 >
382 > // find insert_node
383 > node = this._root;
384 > while (true) {
385 > const val = iter.cmp(node.segment);
386 > if (val > 0) {
387 // left
388 if (!node.left) {
393 node = node.left;
394
395 > } else if (val < 0) { ternarySearchTree.ts
396 // right
397 if (!node.right) {
402 node = node.right;
403
404 > } else if (iter.hasNext()) { ternarySearchTree.ts
405 > // mid ternarySearchTree.ts
406 > iter.next();
407 > if (!node.mid) {
408 > node.mid = new TernarySearchTreeNode<K, V>();
409 > node.mid.segment = iter.value();
410 > }
411 > stack.push([Dir.Mid, node]);
412 > node = node.mid;
413 > } else { ternarySearchTree.ts
414 > break;
415 > }
416 > }
417 >
418 > // set value
419 > const oldElement = Undef.unwrap(node.value);
420 > node.value = Undef.wrap(element);
421 > node.key = key;
422 >
423 > // balance
424 > for (let i = stack.length - 1; i >= 0; i--) {
425 > const node = stack[i][1]; ternarySearchTree.ts
426 >
427 > node.updateHeight();
428 > const bf = node.balanceFactor();
429 >
430 > if (bf < -1 || bf > 1) {
431 // needs rotate
432 const d1 = stack[i][0];
502 return node;
503 }
505 > has(key: K): boolean {
506 const node = this._getNode(key);
507 return !(node?.value === undefined && node?.mid === undefined);
508 }
510 > delete(key: K): void {
511 return this._delete(key, false);
512 }
514 > deleteSuperstr(key: K): void {
515 return this._delete(key, true);
516 }
518 > private _delete(key: K, superStr: boolean): void {
519 const iter = this._iter.reset(key);
520 const stack: [Dir, TernarySearchTreeNode<K, V>][] = [];
620 this._root = this._balanceByStack(stack) ?? this._root;
621 }
623 > private _min(node: TernarySearchTreeNode<K, V>, stack: [Dir, TernarySearchTreeNode<K, V>][]): TernarySearchTreeNode<K, V> {
624 while (node.left) {
625 stack.push([Dir.Left, node]);
628 return node;
629 }
631 > private _balanceByStack(stack: [Dir, TernarySearchTreeNode<K, V>][]) {
632
633 for (let i = stack.length - 1; i >= 0; i--) {
679 return undefined;
680 }
682 > findSubstr(key: K): V | undefined {
683 const iter = this._iter.reset(key);
684 let node = this._root;
703 return node && Undef.unwrap(node.value) || candidate;
704 }
706 > findSuperstr(key: K): IterableIterator<[K, V]> | undefined {
707 return this._findSuperstrOrElement(key, false);
708 }
710 > private _findSuperstrOrElement(key: K, allowValue: true): IterableIterator<[K, V]> | V | undefined;
711 > private _findSuperstrOrElement(key: K, allowValue: false): IterableIterator<[K, V]> | undefined;
712 > private _findSuperstrOrElement(key: K, allowValue: boolean): IterableIterator<[K, V]> | V | undefined {
713 const iter = this._iter.reset(key);
714 let node = this._root;
740 return undefined;
741 }
743 > hasElementOrSubtree(key: K): boolean {
744 return this._findSuperstrOrElement(key, true) !== undefined;
745 }
747 > forEach(callback: (value: V, index: K) => unknown): void {
748 for (const [key, value] of this) {
749 callback(value, key);
750 }
751 }
753 > *[Symbol.iterator](): IterableIterator<[K, V]> {
754 yield* this._entries(this._root);
755 }
757 > private _entries(node: TernarySearchTreeNode<K, V> | undefined): IterableIterator<[K, V]> {
758 const result: [K, V][] = [];
759 this._dfsEntries(node, result);
760 return result[Symbol.iterator]();
761 }
763 > private _dfsEntries(node: TernarySearchTreeNode<K, V> | undefined, bucket: [K, V][]) {
764 // DFS
765 if (!node) {
779 }
780 }
782 > // for debug/testing
783 > _isBalanced(): boolean {
784 const nodeIsBalanced = (node: TernarySearchTreeNode<unknown, unknown> | undefined): boolean => {
785 if (!node) {
src/vs/workbench/contrib/chat/common/promptSyntax/service/promptsServiceImpl.ts 344 covered LOC · 66 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptsServiceImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken, CancellationTokenPool } from '../../../../../../base/common/cancellation.js';
7 > import { CancellationError, isCancellationError } from '../../../../../../base/common/errors.js';
8 > import { Emitter, Event } from '../../../../../../base/common/event.js';
9 > import { ParseError, parse as parseJSONC } from '../../../../../../base/common/json.js';
10 > import { getParseErrorMessage } from '../../../../../../base/common/jsonErrorMessages.js';
11 > import { Disposable, DisposableStore, IDisposable, MutableDisposable } from '../../../../../../base/common/lifecycle.js';
12 > import { StopWatch } from '../../../../../../base/common/stopwatch.js';
13 > import { autorun, IReader } from '../../../../../../base/common/observable.js';
14 > import { ResourceMap, ResourceSet } from '../../../../../../base/common/map.js';
15 > import { basename, dirname, isEqual } from '../../../../../../base/common/resources.js';
16 > import { URI } from '../../../../../../base/common/uri.js';
17 > import { OffsetRange } from '../../../../../../editor/common/core/ranges/offsetRange.js';
18 > import { type ITextModel } from '../../../../../../editor/common/model.js';
19 > import { IModelService } from '../../../../../../editor/common/services/model.js';
20 > import { localize } from '../../../../../../nls.js';
21 > import { IConfigurationService } from '../../../../../../platform/configuration/common/configuration.js';
22 > import { IExtensionDescription } from '../../../../../../platform/extensions/common/extensions.js';
23 > import { FileOperationError, FileOperationResult, IFileService } from '../../../../../../platform/files/common/files.js';
24 > import { IInstantiationService } from '../../../../../../platform/instantiation/common/instantiation.js';
25 > import { ILabelService } from '../../../../../../platform/label/common/label.js';
26 > import { ILogService } from '../../../../../../platform/log/common/log.js';
27 > import { IStorageService, StorageScope, StorageTarget } from '../../../../../../platform/storage/common/storage.js';
28 > import { ITelemetryService } from '../../../../../../platform/telemetry/common/telemetry.js';
29 > import { IUserDataProfileService } from '../../../../../services/userDataProfile/common/userDataProfile.js';
30 > import { IVariableReference } from '../../chatModes.js';
31 > import { PromptsConfig } from '../config/config.js';
32 > import { AGENT_MD_FILENAME, CLAUDE_CONFIG_FOLDER, CLAUDE_LOCAL_MD_FILENAME, CLAUDE_MD_FILENAME, COPILOT_CONFIG_FOLDER, COPILOT_CUSTOM_INSTRUCTIONS_FILENAME, getCleanPromptName, getSkillFolderName, GITHUB_CONFIG_FOLDER, IResolvedPromptSourceFolder, isInClaudeRulesFolder } from '../config/promptFileLocations.js';
33 > import { PROMPT_LANGUAGE_ID, PromptFileSource, PromptsType, Target, getPromptsTypeForLanguageId } from '../promptTypes.js';
34 > import { IWorkspaceInstructionFile, PromptFilesLocator } from '../utils/promptFilesLocator.js';
35 > import { evaluateApplyToPattern, PromptFileParser, ParsedPromptFile, PromptHeaderAttributes } from '../promptFileParser.js';
36 > import { IAgentInstructions, IAgentSource, IChatPromptSlashCommand, IConfiguredHooksInfo, ICustomAgent, IExtensionPromptPath, ILocalPromptPath, IPluginPromptPath, IBuiltinPromptPath, IPromptPath, IPromptsService, IAgentSkill, IInstructionDiscoveryInfo, IInstructionDiscoveryResult, IInstructionFile, IUserPromptPath, PromptsStorage, IPromptFileContext, IPromptFileResource, IPromptDiscoveryInfo, IPromptFileDiscoveryResult, IPromptSourceFolderResult, ICustomAgentVisibility, IAgentInstructionFile, AgentInstructionFileType, Logger, ISlashCommandDiscoveryInfo, ISlashCommandDiscoveryResult, IAgentDiscoveryInfo, IAgentDiscoveryResult, IHookDiscoveryInfo, IResolvedChatPromptSlashCommand, matchesSessionType } from './promptsService.js';
37 > import { Delayer, raceCancellationError } from '../../../../../../base/common/async.js';
38 > import { Schemas } from '../../../../../../base/common/network.js';
39 > import { ChatRequestHooks, parseSubagentHooksFromYaml } from '../hookSchema.js';
40 > import { type IParsedHookCommand } from '../../../../../../platform/agentPlugins/common/pluginParsers.js';
41 > import { HookType } from '../hookTypes.js';
42 > import { HookSourceFormat, parseHooksFromFile } from '../hookCompatibility.js';
43 > import { IWorkspaceContextService } from '../../../../../../platform/workspace/common/workspace.js';
44 > import { IWorkspaceTrustManagementService } from '../../../../../../platform/workspace/common/workspaceTrust.js';
45 > import { IPathService } from '../../../../../services/path/common/pathService.js';
46 > import { getTarget, mapClaudeModels, mapClaudeTools } from '../languageProviders/promptFileAttributes.js';
47 > import { getCanonicalPluginCommandId, IAgentPlugin, IAgentPluginService } from '../../plugins/agentPluginService.js';
48 > import { isContributionEnabled } from '../../enablement.js';
49 > import { assertNever } from '../../../../../../base/common/assert.js';
50 > import { ExtensionPromptFileService } from './extensionPromptFileService.js';
51 >
52 > /**
53 > * Provides prompt services.
54 > */
55 > export class PromptsService extends Disposable implements IPromptsService {
56 > public declare readonly _serviceBrand: undefined;
57 >
58 > /**
59 > * Prompt files locator utility.
60 > */
61 > private readonly fileLocator: PromptFilesLocator;
62 >
63 > /**
64 > * Cached agent discovery info.
65 > */
66 > private readonly cachedCustomAgents: CachedPromise<IAgentDiscoveryInfo>;
67 >
68 > /**
69 > * Cached slash command discovery info.
70 > */
71 > private readonly cachedSlashCommands: CachedPromise<ISlashCommandDiscoveryInfo>;
72 >
73 > /**
74 > * Cached hooks. Invalidated when hook files change.
75 > */
76 > private readonly cachedHooks: CachedPromise<IHookDiscoveryInfo>;
77 >
78 > /**
79 > * Cached skill discovery info.
80 > */
81 > private readonly cachedSkills: CachedPromise<IPromptDiscoveryInfo>;
82 >
83 > /**
84 > * Cached instructions.
85 > */
86 > private readonly cachedInstructions: CachedPromise<IInstructionDiscoveryInfo>;
87 > private readonly agentInstructionsWatcher = this._register(new MutableDisposable<IDisposable>());
88 > private readonly _onDidChangeAgentInstructions = this._register(new Emitter<void>({
89 > onWillAddFirstListener: () => {
90 > const store = new DisposableStore(); promptsServiceImpl.ts
91 > const agentInstructionsUpdatedEvent = this.fileLocator.createAgentInstructionsUpdatedEvent();
92 > store.add(agentInstructionsUpdatedEvent);
93 > store.add(agentInstructionsUpdatedEvent.event(() => this._onDidChangeAgentInstructions.fire()));
94 > this.agentInstructionsWatcher.value = store;
95 > },
96 > onDidRemoveLastListener: () => { promptsServiceImpl.ts
97 > this.agentInstructionsWatcher.clear();
98 > }
99 > }));
100 >
101 > /**
102 > * Synchronous mirror of the names exposed by {@link getPromptSlashCommands},
103 > * maintained for {@link hasPromptSlashCommand} so callers (e.g. the chat request
104 > * parser) can disambiguate `<cmd>:<sub>` vs bare `<cmd>` without an async hop.
105 > */
106 > private readonly knownPromptSlashCommandNames = new Set<string>();
107 >
108 > /**
109 > * Cache for parsed prompt files keyed by URI.
110 > * The number in the returned tuple is textModel.getVersionId(), which is an internal VS Code counter that increments every time the text model's content changes.
111 > */
112 > private readonly cachedParsedPromptFromModels = new ResourceMap<[number, ParsedPromptFile]>();
113 >
114 > /**
115 > * Cached file locations commands. Caching only happens if the corresponding `fileLocatorEvents` event is used.
116 > */
117 > private readonly cachedFileLocations: { [key in PromptsType]?: Promise<readonly IPromptPath[]> } = {};
118 >
119 > /**
120 > * Lazily created events that notify listeners when the file locations for a given prompt type change.
121 > * An event is created on demand for each prompt type and can be used by consumers to react to updates
122 > * in the set of prompt files (e.g., when prompt files are added, removed, or modified).
123 > */
124 > private readonly fileLocatorEvents: { [key in PromptsType]?: Event<void> } = {};
125 >
126 >
127 > /**
128 > * Owns the registry of extension-contributed prompt files (both via
129 > * contribution points and via provider API).
130 > */
131 > private readonly extensionPromptFiles: ExtensionPromptFileService;
132 >
133 > private readonly _onDidPluginPromptFilesChange = this._register(new Emitter<PromptsType>());
134 > private readonly _onDidPluginHooksChange = this._register(new Emitter<void>());
135 > private _pluginPromptFilesByType = new Map<PromptsType, readonly IPluginPromptPath[]>();
136 >
137 > constructor(
138 @ILogService public readonly logger: ILogService,
139 @ILabelService private readonly labelService: ILabelService,
260 }));
261 }
263 > private watchPluginPromptFilesForType(
264 type: PromptsType,
265 getItems: (plugin: IAgentPlugin, reader: IReader) => readonly { uri: URI; name: string }[],
291 });
292 }
294 > protected createPromptFilesLocator(): PromptFilesLocator {
295 return this.instantiationService.createInstance(PromptFilesLocator);
296 }
298 > private getFileLocatorEvent(type: PromptsType): Event<void> {
299 let event = this.fileLocatorEvents[type];
300 if (!event) {
306 return event;
307 }
309 > public getParsedPromptFile(textModel: ITextModel): ParsedPromptFile {
310 const cached = this.cachedParsedPromptFromModels.get(textModel.uri);
311 if (cached && cached[0] === textModel.getVersionId()) {
318 return ast;
319 }
321 > public async listPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IPromptPath[]> {
322 let listPromise = this.cachedFileLocations[type];
323 if (!listPromise) {
331 return listPromise;
332 }
334 > private async computeListPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IPromptPath[]> {
335 const prompts = await Promise.all([
336 this.fileLocator.listFiles(type, PromptsStorage.user, token).then(uris => uris.map(uri => ({ uri, storage: PromptsStorage.user, type } satisfies IUserPromptPath))),
343 return prompts.flat();
344 }
346 > /**
347 > * Collects diagnostic information about which source folders were searched for display in the debug panel.
348 > */
349 > private async _collectSourceFolderDiagnostics(type: PromptsType): Promise<IPromptSourceFolderResult[]> {
350 const resolvedFolders = await this.fileLocator.getSourceFoldersInDiscoveryOrder(type);
351 return resolvedFolders.map(folder => ({
354 }));
355 }
357 > /**
358 > * Registers a prompt file provider (CustomAgentProvider, InstructionsProvider, or PromptFileProvider).
359 > * This will be called by the extension host bridge when
360 > * an extension registers a provider via vscode.chat.registerCustomAgentProvider(),
361 > * registerInstructionsProvider(), or registerPromptFileProvider().
362 > */
363 > public registerPromptFileProvider(extension: IExtensionDescription, type: PromptsType, provider: {
364 onDidChangePromptFiles?: Event<void>;
365 providePromptFiles: (context: IPromptFileContext, token: CancellationToken) => Promise<IPromptFileResource[] | undefined>;
367 return this.extensionPromptFiles.registerPromptFileProvider(extension, type, provider);
368 }
370 >
371 > public async listPromptFilesForStorage(type: PromptsType, storage: PromptsStorage, token: CancellationToken): Promise<readonly IPromptPath[]> {
372 let promptPaths: readonly IPromptPath[];
373 switch (storage) {
393 return promptPaths;
394 }
396 > private getExtensionPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IExtensionPromptPath[]> {
397 return this.extensionPromptFiles.getExtensionPromptFiles(type, token);
398 }
400 > /**
401 > * Returns the built-in prompt files of the given type. The base service ships
402 > * no built-in prompts; subclasses (e.g. the Agents app) override this to
403 > * contribute bundled prompts such as built-in skills.
404 > */
405 > protected async getBuiltinPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IBuiltinPromptPath[]> {
406 return [];
407 }
409 > public async getSourceFolders(type: PromptsType): Promise<readonly IPromptPath[]> {
410 const result: IPromptPath[] = [];
411
431 return result;
432 }
434 > public async getResolvedSourceFolders(type: PromptsType): Promise<readonly IResolvedPromptSourceFolder[]> {
435 return this.fileLocator.getResolvedSourceFolders(type);
436 }
438 > // slash prompt commands
439 >
440 > /**
441 > * Emitter for slash commands change events.
442 > */
443 > public get onDidChangeSlashCommands(): Event<void> {
444 return this.cachedSlashCommands.onDidChangePromise;
445 }
447 > public async getPromptSlashCommands(token: CancellationToken): Promise<readonly IChatPromptSlashCommand[]> {
448 const discoveryInfo = await this.cachedSlashCommands.get(token);
449 const result = this.slashCommandsFromDiscoveryInfo(discoveryInfo);
450 return result;
451 }
453 > /**
454 > * Computes discovery info for slash commands, combining prompts and skills.
455 > */
456 > private async computeSlashCommandDiscoveryInfo(token: CancellationToken): Promise<ISlashCommandDiscoveryInfo> {
457 const stopWatch = StopWatch.create(true);
458 const promptFiles = await this.listPromptFiles(PromptsType.prompt, token);
527 return { type: PromptsType.prompt, files, sourceFolders, durationInMillis: stopWatch.elapsed() };
528 }
530 > /**
531 > * Derives IChatPromptSlashCommand[] from cached discovery info.
532 > */
533 > private slashCommandsFromDiscoveryInfo(discoveryInfo: ISlashCommandDiscoveryInfo): readonly IChatPromptSlashCommand[] {
534 const result: IChatPromptSlashCommand[] = [];
535 const seen = new ResourceSet();
554 return result;
555 }
557 > public isValidSlashCommandName(command: string): boolean {
558 return command.match(/^[\p{L}\d_\-\.:]+$/u) !== null;
559 }
561 > public hasPromptSlashCommand(name: string): boolean {
562 if (!this.knownPromptSlashCommandsHydrationStarted) {
563 this.knownPromptSlashCommandsHydrationStarted = true;
567 return this.knownPromptSlashCommandNames.has(name);
568 }
570 > private knownPromptSlashCommandsHydrationStarted = false;
571 >
572 > private refreshKnownPromptSlashCommandNames(): void {
573 this.getPromptSlashCommands(CancellationToken.None).then(commands => {
574 this.knownPromptSlashCommandNames.clear();
578 }, () => { /* discovery failures already logged; sync cache stays as-is */ });
579 }
581 > public async resolvePromptSlashCommand(name: string, sessionType: string | undefined, token: CancellationToken): Promise<IResolvedChatPromptSlashCommand | undefined> {
582 const commands = await this.getPromptSlashCommands(token);
583 const command = commands.find(cmd => cmd.name === name && matchesSessionType(cmd.sessionTypes, sessionType));
590 return undefined;
591 }
593 > private asChatPromptSlashCommand(argumentHint: string | undefined, userInvocable: boolean | undefined, promptPath: IPromptPath): IChatPromptSlashCommand {
594 let name = promptPath.name ?? getCleanPromptName(promptPath.uri);
595 name = name.replace(/[^\p{L}\d_\-\.:]+/gu, '-'); // replace spaces with dashes
609 };
610 }
612 > public async getPromptSlashCommandName(uri: URI, token: CancellationToken): Promise<string> {
613 const slashCommands = await this.getPromptSlashCommands(token);
614 const slashCommand = slashCommands.find(c => isEqual(c.uri, uri));
618 return slashCommand.name;
619 }
621 > // custom agents
622 >
623 > /**
624 > * Emitter for custom agents change events.
625 > */
626 > public get onDidChangeCustomAgents(): Event<void> {
627 return this.cachedCustomAgents.onDidChangePromise;
628 }
630 > public get onDidChangeInstructions(): Event<void> {
631 return this.cachedInstructions.onDidChangePromise;
632 }
634 > public get onDidChangeAgentInstructions(): Event<void> {
635 return this._onDidChangeAgentInstructions.event;
636 }
638 > public async getCustomAgents(token: CancellationToken): Promise<readonly ICustomAgent[]> {
639 const discoveryInfo = await this.cachedCustomAgents.get(token);
640 const result = this.agentsFromDiscoveryInfo(discoveryInfo);
641 return result;
642 }
644 > /**
645 > * Derives ICustomAgent[] from cached discovery info.
646 > */
647 > private agentsFromDiscoveryInfo(discoveryInfo: IAgentDiscoveryInfo): readonly ICustomAgent[] {
648 const result: ICustomAgent[] = [];
649 for (const file of discoveryInfo.files) {
654 return result;
655 }
657 > private async computeAgentDiscoveryInfo(token: CancellationToken): Promise<IAgentDiscoveryInfo> {
658 const stopWatch = StopWatch.create(true);
659 const allAgentFiles = await this.listPromptFiles(PromptsType.agent, token);
714 return { type: PromptsType.agent, files, sourceFolders, durationInMillis: stopWatch.elapsed() };
715 }
717 >
718 > public async parseNew(uri: URI, token: CancellationToken): Promise<ParsedPromptFile> {
719 const model = this.modelService.getModel(uri);
720 if (model) {
728 return new PromptFileParser().parse(uri, fileContent.value.toString());
729 }
731 > public registerContributedFile(type: PromptsType, uri: URI, extension: IExtensionDescription, name?: string, description?: string, when?: string, sessionTypes?: readonly string[]) {
732 return this.extensionPromptFiles.registerContributedFile(type, uri, extension, name, description, when, sessionTypes);
733 }
735 > getPromptLocationLabel(promptPath: IPromptPath): string {
736 switch (promptPath.storage) {
737 case PromptsStorage.local: return this.labelService.getUriLabel(dirname(promptPath.uri), { relative: true });
745 }
746 }
748 > public async listNestedAgentMDs(token: CancellationToken): Promise<IAgentInstructionFile[]> {
749 const useAgentMD = this.configurationService.getValue(PromptsConfig.USE_AGENT_MD);
750 if (!useAgentMD) {
757 return [];
758 }
760 > public async listAgentInstructions(token: CancellationToken, logger: Logger | undefined): Promise<IAgentInstructionFile[]> {
761 const resolvedAgentFiles: IAgentInstructionFile[] = [];
762 const promises: Promise<IAgentInstructionFile[]>[] = [];
822 return result.sort((a, b) => a.uri.toString().localeCompare(b.uri.toString()));
823 }
825 > public getAgentFileURIFromModeFile(oldURI: URI): URI | undefined {
826 return this.fileLocator.getAgentFileURIFromModeFile(oldURI);
827 }
829 > // --- Enabled Prompt Files -----------------------------------------------------------
830 >
831 > private readonly disabledPromptsStorageKeyPrefix = 'chat.disabledPromptFiles.';
832 >
833 > public getDisabledPromptFiles(type: PromptsType): ResourceSet {
834 // Migration: if disabled key absent but legacy enabled key present, convert once.
835 const disabledKey = this.disabledPromptsStorageKeyPrefix + type;
852 return result;
853 }
855 > public setDisabledPromptFiles(type: PromptsType, uris: ResourceSet): void {
856 const disabled = Array.from(uris).map(uri => uri.toJSON());
857 this.storageService.store(this.disabledPromptsStorageKeyPrefix + type, JSON.stringify(disabled), StorageScope.PROFILE, StorageTarget.USER);
863 }
864 }
866 > // Agent skills
867 >
868 > private sanitizeAgentSkillText(text: string): string {
869 // Remove XML tags
870 return text.replace(/<[^>]+>/g, '');
871 }
873 > private truncateAgentSkillName(name: string, uri: URI): string {
874 const MAX_NAME_LENGTH = 64;
875 const sanitized = this.sanitizeAgentSkillText(name);
883 return sanitized;
884 }
886 > private truncateAgentSkillDescription(description: string | undefined, uri: URI): string | undefined {
887 if (!description) {
888 return undefined;
899 return sanitized;
900 }
902 > public get onDidChangeSkills(): Event<void> {
903 return this.cachedSkills.onDidChangePromise;
904 }
906 > public get onDidChangeHooks(): Event<void> {
907 return this.cachedHooks.onDidChangePromise;
908 }
910 > public async findAgentSkills(token: CancellationToken): Promise<IAgentSkill[] | undefined> {
911 const useAgentSkills = this.configurationService.getValue(PromptsConfig.USE_AGENT_SKILLS);
912 if (!useAgentSkills) {
918 return result;
919 }
921 > /**
922 > * Derives IAgentSkill[] from cached discovery info.
923 > */
924 > private skillsFromDiscoveryInfo(discoveryInfo: IPromptDiscoveryInfo): IAgentSkill[] {
925 const result: IAgentSkill[] = [];
926 for (const file of discoveryInfo.files) {
943 return result;
944 }
946 > /**
947 > * Computes the full skill discovery info, including source folders and telemetry.
948 > */
949 > private async computeSkillDiscovery(token: CancellationToken): Promise<IPromptDiscoveryInfo> {
950 const stopWatch = StopWatch.create(true);
951 const files = await this.computeSkillDiscoveryInfo(token);
1047 return { type: PromptsType.skill, files, sourceFolders, durationInMillis: stopWatch.elapsed() };
1048 }
1050 > public async getHooks(token: CancellationToken): Promise<IConfiguredHooksInfo | undefined> {
1051 const discoveryInfo = await this.cachedHooks.get(token);
1052 const result = discoveryInfo.hooksInfo;
1053 return result;
1054 }
1056 > public async getDiscoveryInfo(type: PromptsType, token: CancellationToken): Promise<IPromptDiscoveryInfo> {
1057 switch (type) {
1058 case PromptsType.instructions:
1068 }
1069 }
1071 > public async getInstructionFiles(token: CancellationToken): Promise<readonly IInstructionFile[]> {
1072 const discoveryInfo = await this.cachedInstructions.get(token);
1073 const result = this.instructionsFromDiscoveryInfo(discoveryInfo);
1074 return result;
1075 }
1077 > private instructionsFromDiscoveryInfo(discoveryInfo: IInstructionDiscoveryInfo): IInstructionFile[] {
1078 const result: IInstructionFile[] = [];
1079 for (const file of discoveryInfo.files) {
1094 return result;
1095 }
1097 > private withPromptPathMetadata(promptPath: IPromptPath, name: string | undefined, description: string | undefined): IPromptPath {
1098 return { ...promptPath, name, description };
1099 }
1101 > private async computeInstructionFiles(token: CancellationToken): Promise<IInstructionDiscoveryInfo> {
1102 return await this.getInstructionsDiscoveryInfo(token);
1103 }
1105 > private async computeHooks(token: CancellationToken): Promise<IHookDiscoveryInfo> {
1106 const stopWatch = StopWatch.create(true);
1107 const useChatHooks = this.configurationService.getValue(PromptsConfig.USE_CHAT_HOOKS);
1303 return { type: PromptsType.hook, files, sourceFolders, hooksInfo: { hooks: result, hasDisabledClaudeHooks }, durationInMillis: stopWatch.elapsed() };
1304 }
1306 > /**
1307 > * Precedence used when deduplicating skills that share the same canonical
1308 > * name: workspace > personal > plugin > extension API > extension contribution.
1309 > * Lower numbers win.
1310 > */
1311 > private getSkillPriority(skill: IPromptPath): number {
1312 if (skill.storage === PromptsStorage.local) {
1313 return 0; // workspace
1327 return 5;
1328 }
1330 > /**
1331 > * Returns the discovery results for skill files.
1332 > */
1333 > private async computeSkillDiscoveryInfo(token: CancellationToken): Promise<IPromptFileDiscoveryResult[]> {
1334 const files: IPromptFileDiscoveryResult[] = [];
1335 const seenNames = new Set<string>();
1397 return files;
1398 }
1400 > private async getInstructionsDiscoveryInfo(token: CancellationToken): Promise<IInstructionDiscoveryInfo> {
1401 const stopWatch = StopWatch.create(true);
1402 const files: IInstructionDiscoveryResult[] = [];
1429 return { type: PromptsType.instructions, files, sourceFolders, durationInMillis: stopWatch.elapsed() };
1430 }
1432 >
1433 > // helpers
1434 >
1435 > class CachedPromise<T> extends Disposable {
1436 > private cachedPromise: Promise<T> | undefined = undefined;
1437 > private cachedPool: CancellationTokenPool | undefined = undefined;
1438 > private readonly onDidUpdatePromiseEmitter: Emitter<void>;
1439 >
1440 > constructor(private readonly computeFn: (token: CancellationToken) => Promise<T>, private readonly getEvent: () => Event<void>, private readonly delay: number = 0) {
1441 super();
1442 this.onDidUpdatePromiseEmitter = this._register(new Emitter<void>());
1447 }));
1448 }
1450 > public get onDidChangePromise(): Event<void> {
1451 return this.onDidUpdatePromiseEmitter.event;
1452 }
1454 > public get(token: CancellationToken): Promise<T> {
1455 // If a previous in-flight computation had all of its callers cancel, the pool's
1456 // token will have fired and the computation may have rejected/aborted. A new
1486 return raceCancellationError(this.cachedPromise, token);
1487 }
1489 > public refresh(): void {
1490 this.cachedPromise = undefined;
1491 this.onDidUpdatePromiseEmitter?.fire();
1492 }
1494 >
1495 > interface ModelChangeEvent {
1496 > readonly promptType: PromptsType;
1497 > readonly uri: URI;
1498 > }
1499 >
1500 > class ModelChangeTracker extends Disposable {
1501 >
1502 > private readonly listeners = new ResourceMap<IDisposable>();
1503 > private readonly onDidPromptModelChange: Emitter<ModelChangeEvent>;
1504 >
1505 > public get onDidPromptChange(): Event<ModelChangeEvent> {
1506 > return this.onDidPromptModelChange.event;
1507 > }
1508 >
1509 > constructor(modelService: IModelService) {
1510 super();
1511 this.onDidPromptModelChange = this._register(new Emitter<ModelChangeEvent>());
1540 this._register(modelService.onModelRemoved(model => onRemove(model.getLanguageId(), model.uri)));
1541 }
1543 > public override dispose(): void {
1544 super.dispose();
1545 this.listeners.forEach(listener => listener.dispose());
1546 this.listeners.clear();
1547 }
1549 >
1550 > export namespace CustomAgent {
1551 > export function fromParsedPromptFile(ast: ParsedPromptFile, extra: { name?: string; description?: string; source: IAgentSource; hooks?: ChatRequestHooks; sessionTypes: readonly string[] | undefined; enabled: boolean }): ICustomAgent {
1552 const uri = ast.uri;
1553 const { hooks, sessionTypes, enabled } = extra;
src/vs/base/common/map.ts 343 covered LOC · 109 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- map.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from './uri.js';
7 >
8 > export function getOrSet<K, V>(map: Map<K, V>, key: K, value: V): V {
9 let result = map.get(key);
10 if (result === undefined) {
15 return result;
16 }
17 > map.ts
18 > export function mapToString<K, V>(map: Map<K, V>): string {
19 const entries: string[] = [];
20 map.forEach((value, key) => {
24 return `Map(${map.size}) {${entries.join(', ')}}`;
25 }
26 > map.ts
27 > export function setToString<K>(set: Set<K>): string {
28 const entries: K[] = [];
29 set.forEach(value => {
33 return `Set(${set.size}) {${entries.join(', ')}}`;
34 }
35 > map.ts
36 > interface ResourceMapKeyFn {
37 > (resource: URI): string;
38 > }
39 >
40 > class ResourceMapEntry<T> {
41 > constructor(readonly uri: URI, readonly value: T) { }
42 > }
43 >
44 > function isEntries<T>(arg: ResourceMap<T> | ResourceMapKeyFn | readonly (readonly [URI, T])[] | undefined): arg is readonly (readonly [URI, T])[] { map.ts
45 > return Array.isArray(arg);
46 > }
47 > map.ts
48 > export class ResourceMap<T> implements Map<URI, T> {
49 >
50 > private static readonly defaultToKey = (resource: URI) => resource.toString();
51 >
52 > readonly [Symbol.toStringTag] = 'ResourceMap';
53 >
54 > private readonly map: Map<string, ResourceMapEntry<T>>;
55 > private readonly toKey: ResourceMapKeyFn;
56 >
57 > /**
58 > *
59 > * @param toKey Custom uri identity function, e.g use an existing `IExtUri#getComparison`-util
60 > */
61 > constructor(toKey?: ResourceMapKeyFn);
62 >
63 > /**
64 > *
65 > * @param other Another resource which this maps is created from
66 > * @param toKey Custom uri identity function, e.g use an existing `IExtUri#getComparison`-util
67 > */
68 > constructor(other?: ResourceMap<T>, toKey?: ResourceMapKeyFn);
69 >
70 > /**
71 > *
72 > * @param other Another resource which this maps is created from
73 > * @param toKey Custom uri identity function, e.g use an existing `IExtUri#getComparison`-util
74 > */
75 > constructor(entries?: readonly (readonly [URI, T])[], toKey?: ResourceMapKeyFn);
76 >
77 > constructor(arg?: ResourceMap<T> | ResourceMapKeyFn | readonly (readonly [URI, T])[], toKey?: ResourceMapKeyFn) {
78 > if (arg instanceof ResourceMap) { map.ts
79 this.map = new Map(arg.map);
80 this.toKey = toKey ?? ResourceMap.defaultToKey;
81 > } else if (isEntries(arg)) { map.ts
82 this.map = new Map();
83 this.toKey = toKey ?? ResourceMap.defaultToKey;
86 this.set(resource, value);
87 }
88 > } else { map.ts
89 > this.map = new Map();
90 > this.toKey = arg ?? ResourceMap.defaultToKey;
91 > }
92 > }
93 > map.ts
94 > set(resource: URI, value: T): this {
95 this.map.set(this.toKey(resource), new ResourceMapEntry(resource, value));
96 return this;
97 }
98 > map.ts
99 > get(resource: URI): T | undefined {
100 > return this.map.get(this.toKey(resource))?.value; map.ts
101 > }
102 > map.ts
103 > has(resource: URI): boolean {
104 return this.map.has(this.toKey(resource));
105 }
106 > map.ts
107 > get size(): number {
108 > return this.map.size; map.ts
109 > }
110 > map.ts
111 > clear(): void {
112 > this.map.clear(); map.ts
113 > }
114 > map.ts
115 > delete(resource: URI): boolean {
116 return this.map.delete(this.toKey(resource));
117 }
118 > map.ts
119 > forEach(clb: (value: T, key: URI, map: Map<URI, T>) => void, thisArg?: object): void {
120 if (typeof thisArg !== 'undefined') {
121 clb = clb.bind(thisArg);
125 }
126 }
127 > map.ts
128 > *values(): MapIterator<T> {
129 > for (const entry of this.map.values()) { map.ts
130 yield entry.value;
131 }
132 > } map.ts
133 > map.ts
134 > *keys(): MapIterator<URI> {
135 for (const entry of this.map.values()) {
136 yield entry.uri;
137 }
138 }
139 > map.ts
140 > *entries(): MapIterator<[URI, T]> {
141 for (const entry of this.map.values()) {
142 yield [entry.uri, entry.value];
143 }
144 }
145 > map.ts
146 > *[Symbol.iterator](): MapIterator<[URI, T]> {
147 for (const [, entry] of this.map) {
148 yield [entry.uri, entry.value];
149 }
150 }
151 > } map.ts
152 >
153 > export class ResourceSet implements Set<URI> {
154 >
155 > readonly [Symbol.toStringTag]: string = 'ResourceSet';
156 >
157 > private readonly _map: ResourceMap<URI>;
158 >
159 > constructor(toKey?: ResourceMapKeyFn);
160 > constructor(entries: readonly URI[], toKey?: ResourceMapKeyFn);
161 > constructor(entriesOrKey?: readonly URI[] | ResourceMapKeyFn, toKey?: ResourceMapKeyFn) {
162 if (!entriesOrKey || typeof entriesOrKey === 'function') {
163 this._map = new ResourceMap(entriesOrKey);
167 }
168 }
169 > map.ts
170 >
171 > get size(): number {
172 return this._map.size;
173 }
174 > map.ts
175 > add(value: URI): this {
176 this._map.set(value, value);
177 return this;
178 }
179 > map.ts
180 > clear(): void {
181 this._map.clear();
182 }
183 > map.ts
184 > delete(value: URI): boolean {
185 return this._map.delete(value);
186 }
187 > map.ts
188 > forEach(callbackfn: (value: URI, value2: URI, set: Set<URI>) => void, thisArg?: unknown): void {
189 this._map.forEach((_value, key) => callbackfn.call(thisArg, key, key, this));
190 }
191 > map.ts
192 > has(value: URI): boolean {
193 return this._map.has(value);
194 }
195 > map.ts
196 > entries(): SetIterator<[URI, URI]> {
197 return this._map.entries() as unknown as SetIterator<[URI, URI]>;
198 }
199 > map.ts
200 > keys(): SetIterator<URI> {
201 return this._map.keys() as unknown as SetIterator<URI>;
202 }
203 > map.ts
204 > values(): SetIterator<URI> {
205 return this._map.keys() as unknown as SetIterator<URI>;
206 }
207 > map.ts
208 > [Symbol.iterator](): SetIterator<URI> {
209 return this.keys();
210 }
211 > } map.ts
212 >
213 >
214 > interface Item<K, V> {
215 > previous: Item<K, V> | undefined;
216 > next: Item<K, V> | undefined;
217 > key: K;
218 > value: V;
219 > }
220 >
221 > export const enum Touch {
222 > None = 0,
223 > AsOld = 1,
224 > AsNew = 2
225 > }
226 >
227 > export class LinkedMap<K, V> implements Map<K, V> {
228 >
229 > readonly [Symbol.toStringTag] = 'LinkedMap';
230 >
231 > private _map: Map<K, Item<K, V>>;
232 > private _head: Item<K, V> | undefined;
233 > private _tail: Item<K, V> | undefined;
234 > private _size: number;
235 >
236 > private _state: number;
237 >
238 > constructor() {
239 > this._map = new Map<K, Item<K, V>>(); map.ts
240 > this._head = undefined;
241 > this._tail = undefined;
242 > this._size = 0;
243 > this._state = 0;
244 > }
245 > map.ts
246 > clear(): void {
247 this._map.clear();
248 this._head = undefined;
251 this._state++;
252 }
253 > map.ts
254 > isEmpty(): boolean {
255 return !this._head && !this._tail;
256 }
257 > map.ts
258 > get size(): number {
259 return this._size;
260 }
261 > map.ts
262 > get first(): V | undefined {
263 return this._head?.value;
264 }
265 > map.ts
266 > get last(): V | undefined {
267 return this._tail?.value;
268 }
269 > map.ts
270 > has(key: K): boolean {
271 return this._map.has(key);
272 }
273 > map.ts
274 > get(key: K, touch: Touch = Touch.None): V | undefined {
275 const item = this._map.get(key);
276 if (!item) {
282 return item.value;
283 }
284 > map.ts
285 > set(key: K, value: V, touch: Touch = Touch.None): this {
286 let item = this._map.get(key);
287 if (item) {
311 return this;
312 }
313 > map.ts
314 > delete(key: K): boolean {
315 return !!this.remove(key);
316 }
317 > map.ts
318 > remove(key: K): V | undefined {
319 const item = this._map.get(key);
320 if (!item) {
326 return item.value;
327 }
328 > map.ts
329 > shift(): V | undefined {
330 if (!this._head && !this._tail) {
331 return undefined;
340 return item.value;
341 }
342 > map.ts
343 > forEach(callbackfn: (value: V, key: K, map: Map<K, V>) => void, thisArg?: unknown): void {
344 const state = this._state;
345 let current = this._head;
356 }
357 }
358 > map.ts
359 > keys(): MapIterator<K> {
360 const map = this;
361 const state = this._state;
381 return iterator;
382 }
383 > map.ts
384 > values(): MapIterator<V> {
385 const map = this;
386 const state = this._state;
406 return iterator;
407 }
408 > map.ts
409 > entries(): MapIterator<[K, V]> {
410 const map = this;
411 const state = this._state;
431 return iterator;
432 }
433 > map.ts
434 > [Symbol.iterator](): MapIterator<[K, V]> {
435 return this.entries();
436 }
437 > map.ts
438 > protected trimOld(newSize: number) {
439 if (newSize >= this.size) {
440 return;
458 this._state++;
459 }
460 > map.ts
461 > protected trimNew(newSize: number) {
462 if (newSize >= this.size) {
463 return;
481 this._state++;
482 }
483 > map.ts
484 > private addItemFirst(item: Item<K, V>): void {
485 // First time Insert
486 if (!this._head && !this._tail) {
495 this._state++;
496 }
497 > map.ts
498 > private addItemLast(item: Item<K, V>): void {
499 // First time Insert
500 if (!this._head && !this._tail) {
509 this._state++;
510 }
511 > map.ts
512 > private removeItem(item: Item<K, V>): void {
513 if (item === this._head && item === this._tail) {
514 this._head = undefined;
546 this._state++;
547 }
548 > map.ts
549 > private touch(item: Item<K, V>, touch: Touch): void {
550 if (!this._head || !this._tail) {
551 throw new Error('Invalid list');
608 }
609 }
610 > map.ts
611 > toJSON(): [K, V][] {
612 const data: [K, V][] = [];
613
618 return data;
619 }
620 > map.ts
621 > fromJSON(data: [K, V][]): void {
622 this.clear();
623
626 }
627 }
628 > } map.ts
629 >
630 > abstract class Cache<K, V> extends LinkedMap<K, V> {
631 >
632 > protected _limit: number;
633 > protected _ratio: number;
634 >
635 > constructor(limit: number, ratio: number = 1) {
636 > super(); map.ts
637 > this._limit = limit;
638 > this._ratio = Math.min(Math.max(0, ratio), 1);
639 > }
640 > map.ts
641 > get limit(): number {
642 return this._limit;
643 }
644 > map.ts
645 > set limit(limit: number) {
646 this._limit = limit;
647 this.checkTrim();
648 }
649 > map.ts
650 > get ratio(): number {
651 return this._ratio;
652 }
653 > map.ts
654 > set ratio(ratio: number) {
655 this._ratio = Math.min(Math.max(0, ratio), 1);
656 this.checkTrim();
657 }
658 > map.ts
659 > override get(key: K, touch: Touch = Touch.AsNew): V | undefined {
660 return super.get(key, touch);
661 }
662 > map.ts
663 > peek(key: K): V | undefined {
664 return super.get(key, Touch.None);
665 }
666 > map.ts
667 > override set(key: K, value: V): this {
668 super.set(key, value, Touch.AsNew);
669 return this;
670 }
671 > map.ts
672 > protected checkTrim() {
673 if (this.size > this._limit) {
674 this.trim(Math.round(this._limit * this._ratio));
675 }
676 }
677 > map.ts
678 > protected abstract trim(newSize: number): void;
679 > }
680 >
681 > export class LRUCache<K, V> extends Cache<K, V> {
682 >
683 > constructor(limit: number, ratio: number = 1) {
684 > super(limit, ratio); map.ts
685 > }
686 > map.ts
687 > protected override trim(newSize: number) {
688 this.trimOld(newSize);
689 }
690 > map.ts
691 > override set(key: K, value: V): this {
692 super.set(key, value);
693 this.checkTrim();
694 return this;
695 }
696 > } map.ts
697 >
698 > export class MRUCache<K, V> extends Cache<K, V> {
699 >
700 > constructor(limit: number, ratio: number = 1) {
701 super(limit, ratio);
702 }
703 > map.ts
704 > protected override trim(newSize: number) {
705 this.trimNew(newSize);
706 }
707 > map.ts
708 > override set(key: K, value: V): this {
709 if (this._limit <= this.size && !this.has(key)) {
710 this.trim(Math.round(this._limit * this._ratio) - 1);
714 return this;
715 }
716 > } map.ts
717 >
718 > export class CounterSet<T> {
719
720 private map = new Map<T, number>();
721 > map.ts
722 > add(value: T): CounterSet<T> {
723 this.map.set(value, (this.map.get(value) || 0) + 1);
724 return this;
725 }
726 > map.ts
727 > delete(value: T): boolean {
728 let counter = this.map.get(value) || 0;
729
742 return true;
743 }
744 > map.ts
745 > has(value: T): boolean {
746 return this.map.has(value);
747 }
748 > } map.ts
749 >
750 > /**
751 > * A map that allows access both by keys and values.
752 > * **NOTE**: values need to be unique.
753 > */
754 > export class BidirectionalMap<K, V> {
755 >
756 > private readonly _m1 = new Map<K, V>();
757 > private readonly _m2 = new Map<V, K>();
758 >
759 > constructor(entries?: readonly (readonly [K, V])[]) {
760 if (entries) {
761 for (const [key, value] of entries) {
764 }
765 }
766 > map.ts
767 > clear(): void {
768 this._m1.clear();
769 this._m2.clear();
770 }
771 > map.ts
772 > set(key: K, value: V): void {
773 this._m1.set(key, value);
774 this._m2.set(value, key);
775 }
776 > map.ts
777 > get(key: K): V | undefined {
778 return this._m1.get(key);
779 }
780 > map.ts
781 > getKey(value: V): K | undefined {
782 return this._m2.get(value);
783 }
784 > map.ts
785 > delete(key: K): boolean {
786 const value = this._m1.get(key);
787 if (value === undefined) {
792 return true;
793 }
794 > map.ts
795 > forEach(callbackfn: (value: V, key: K, map: BidirectionalMap<K, V>) => void, thisArg?: unknown): void {
796 this._m1.forEach((value, key) => {
797 callbackfn.call(thisArg, value, key, this);
798 });
799 }
800 > map.ts
801 > keys(): IterableIterator<K> {
802 return this._m1.keys();
803 }
804 > map.ts
805 > values(): IterableIterator<V> {
806 return this._m1.values();
807 }
808 > } map.ts
809 >
810 > export class SetMap<K, V> {
811
812 private map = new Map<K, Set<V>>();
813 > map.ts
814 > add(key: K, value: V): void {
815 let values = this.map.get(key);
816
822 values.add(value);
823 }
824 > map.ts
825 > delete(key: K, value: V): void {
826 const values = this.map.get(key);
827
836 }
837 }
838 > map.ts
839 > forEach(key: K, fn: (value: V) => void): void {
840 const values = this.map.get(key);
841
846 values.forEach(fn);
847 }
848 > map.ts
849 > get(key: K): ReadonlySet<V> {
850 const values = this.map.get(key);
851 if (!values) {
854 return values;
855 }
856 > } map.ts
857 >
858 > export function mapsStrictEqualIgnoreOrder(a: Map<unknown, unknown>, b: Map<unknown, unknown>): boolean {
859 if (a === b) {
860 return true;
879 return true;
880 }
881 > map.ts
882 > /**
883 > * A map that is addressable with an arbitrary number of keys. This is useful in high performance
884 > * scenarios where creating a composite key whenever the data is accessed is too expensive. For
885 > * example for a very hot function, constructing a string like `first-second-third` for every call
886 > * will cause a significant hit to performance.
887 > */
888 > export class NKeyMap<TValue, TKeys extends (string | boolean | number)[]> {
889 private _data: Map<any, any> = new Map();
890 > map.ts
891 > /**
892 > * Sets a value on the map. Note that unlike a standard `Map`, the first argument is the value.
893 > * This is because the spread operator is used for the keys and must be last..
894 > * @param value The value to set.
895 > * @param keys The keys for the value.
896 > */
897 > public set(value: TValue, ...keys: [...TKeys]): void {
898 let currentMap = this._data;
899 for (let i = 0; i < keys.length - 1; i++) {
907 currentMap.set(keys[keys.length - 1], value);
908 }
909 > map.ts
910 > public get(...keys: [...TKeys]): TValue | undefined {
911 let currentMap = this._data;
912 for (let i = 0; i < keys.length - 1; i++) {
919 return currentMap.get(keys[keys.length - 1]);
920 }
921 > map.ts
922 > public delete(...keys: [...TKeys]): boolean {
923 const maps: Map<any, any>[] = [this._data];
924 let currentMap = this._data;
939 return deleted;
940 }
941 > map.ts
942 > public deleteAll(...keys: Partial<TKeys>): boolean {
943 if (keys.length === 0) {
944 const hadData = this._data.size > 0;
964 return deleted;
965 }
966 > map.ts
967 > public clear(): void {
968 this._data.clear();
969 }
970 > map.ts
971 > public *getAll(...keys: Partial<TKeys>): IterableIterator<TValue> {
972 let currentMap = this._data;
973 for (const key of keys) {
980 yield* this._values(currentMap);
981 }
982 > map.ts
983 > public *values(): IterableIterator<TValue> {
984 yield* this._values(this._data);
985 }
986 > map.ts
987 > private *_values(map: Map<any, any>): IterableIterator<TValue> {
988 for (const value of map.values()) {
989 if (value instanceof Map) {
994 }
995 }
996 > map.ts
997 > /**
998 > * Get a textual representation of the map for debugging purposes.
999 > */
1000 > public toString(): string {
1001 const printMap = (map: Map<any, any>, depth: number): string => {
1002 let result = '';
1014 return printMap(this._data, 0);
1015 }
1016 > } map.ts
src/vs/base/common/naturalLanguage/korean.ts 329 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- korean.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 >
8 > /**
9 > * Gets alternative Korean characters for the character code. This will return the ascii
10 > * character code(s) that a Hangul character may have been input with using a qwerty layout.
11 > *
12 > * This only aims to cover modern (not archaic) Hangul syllables.
13 > *
14 > * @param code The character code to get alternate characters for
15 > */
16 > export function getKoreanAltChars(code: number): ArrayLike<number> | undefined {
17 const result = disassembleKorean(code);
18 if (result && result.length > 0) {
21 return undefined;
22 }
23 > korean.ts
24 > let codeBufferLength = 0;
25 > const codeBuffer = new Uint32Array(10);
26 function disassembleKorean(code: number): Uint32Array | undefined {
27 codeBufferLength = 0;
91 return undefined;
92 }
93 > korean.ts
94 function getCodesFromArray(code: number, array: ArrayLike<number>, arrayStartIndex: number): void {
95 // Verify the code is within the array's range
98 }
99 }
100 > korean.ts
101 function addCodesToBuffer(codes: number): void {
102 // NUL is ignored, this is used for archaic characters to avoid using a Map
114 }
115 }
116 > korean.ts
117 > const enum HangulRangeStartCode {
118 > InitialConsonant = 0x1100,
119 > Vowel = 0x1161,
120 > FinalConsonant = 0x11A8,
121 > CompatibilityJamo = 0x3131,
122 > }
123 >
124 > const enum AsciiCode {
125 > NUL = 0,
126 > A = 65,
127 > B = 66,
128 > C = 67,
129 > D = 68,
130 > E = 69,
131 > F = 70,
132 > G = 71,
133 > H = 72,
134 > I = 73,
135 > J = 74,
136 > K = 75,
137 > L = 76,
138 > M = 77,
139 > N = 78,
140 > O = 79,
141 > P = 80,
142 > Q = 81,
143 > R = 82,
144 > S = 83,
145 > T = 84,
146 > U = 85,
147 > V = 86,
148 > W = 87,
149 > X = 88,
150 > Y = 89,
151 > Z = 90,
152 > a = 97,
153 > b = 98,
154 > c = 99,
155 > d = 100,
156 > e = 101,
157 > f = 102,
158 > g = 103,
159 > h = 104,
160 > i = 105,
161 > j = 106,
162 > k = 107,
163 > l = 108,
164 > m = 109,
165 > n = 110,
166 > o = 111,
167 > p = 112,
168 > q = 113,
169 > r = 114,
170 > s = 115,
171 > t = 116,
172 > u = 117,
173 > v = 118,
174 > w = 119,
175 > x = 120,
176 > y = 121,
177 > z = 122,
178 > }
179 >
180 > /**
181 > * Numbers that represent multiple ascii codes. These are precomputed at compile time to reduce
182 > * bundle and runtime overhead.
183 > */
184 > const enum AsciiCodeCombo {
185 > fa = AsciiCode.a << 8 | AsciiCode.f,
186 > fg = AsciiCode.g << 8 | AsciiCode.f,
187 > fq = AsciiCode.q << 8 | AsciiCode.f,
188 > fr = AsciiCode.r << 8 | AsciiCode.f,
189 > ft = AsciiCode.t << 8 | AsciiCode.f,
190 > fv = AsciiCode.v << 8 | AsciiCode.f,
191 > fx = AsciiCode.x << 8 | AsciiCode.f,
192 > hk = AsciiCode.k << 8 | AsciiCode.h,
193 > hl = AsciiCode.l << 8 | AsciiCode.h,
194 > ho = AsciiCode.o << 8 | AsciiCode.h,
195 > ml = AsciiCode.l << 8 | AsciiCode.m,
196 > nj = AsciiCode.j << 8 | AsciiCode.n,
197 > nl = AsciiCode.l << 8 | AsciiCode.n,
198 > np = AsciiCode.p << 8 | AsciiCode.n,
199 > qt = AsciiCode.t << 8 | AsciiCode.q,
200 > rt = AsciiCode.t << 8 | AsciiCode.r,
201 > sg = AsciiCode.g << 8 | AsciiCode.s,
202 > sw = AsciiCode.w << 8 | AsciiCode.s,
203 > }
204 >
205 > /**
206 > * Hangul Jamo - Modern consonants #1
207 > *
208 > * Range U+1100..U+1112
209 > *
210 > * | | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C | D | E | F |
211 > * |--------|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
212 > * | U+110x | ᄀ | ᄁ | ᄂ | ᄃ | ᄄ | ᄅ | ᄆ | ᄇ | ᄈ | ᄉ | ᄊ | ᄋ | ᄌ | ᄍ | ᄎ | ᄏ |
213 > * | U+111x | ᄐ | ᄑ | ᄒ |
214 > */
215 > const modernConsonants = new Uint8Array([
216 > AsciiCode.r, // ㄱ
217 > AsciiCode.R, // ㄲ
218 > AsciiCode.s, // ㄴ
219 > AsciiCode.e, // ㄷ
220 > AsciiCode.E, // ㄸ
221 > AsciiCode.f, // ㄹ
222 > AsciiCode.a, // ㅁ
223 > AsciiCode.q, // ㅂ
224 > AsciiCode.Q, // ㅃ
225 > AsciiCode.t, // ㅅ
226 > AsciiCode.T, // ㅆ
227 > AsciiCode.d, // ㅇ
228 > AsciiCode.w, // ㅈ
229 > AsciiCode.W, // ㅉ
230 > AsciiCode.c, // ㅊ
231 > AsciiCode.z, // ㅋ
232 > AsciiCode.x, // ㅌ
233 > AsciiCode.v, // ㅍ
234 > AsciiCode.g, // ㅎ
235 > ]);
236 >
237 > /**
238 > * Hangul Jamo - Modern Vowels
239 > *
240 > * Range U+1161..U+1175
241 > *
242 > * | | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C | D | E | F |
243 > * |--------|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
244 > * | U+116x | | ᅡ | ᅢ | ᅣ | ᅤ | ᅥ | ᅦ | ᅧ | ᅨ | ᅩ | ᅪ | ᅫ | ᅬ | ᅭ | ᅮ | ᅯ |
245 > * | U+117x | ᅰ | ᅱ | ᅲ | ᅳ | ᅴ | ᅵ |
246 > */
247 > const modernVowels = new Uint16Array([
248 > AsciiCode.k, // -> ㅏ
249 > AsciiCode.o, // -> ㅐ
250 > AsciiCode.i, // -> ㅑ
251 > AsciiCode.O, // -> ㅒ
252 > AsciiCode.j, // -> ㅓ
253 > AsciiCode.p, // -> ㅔ
254 > AsciiCode.u, // -> ㅕ
255 > AsciiCode.P, // -> ㅖ
256 > AsciiCode.h, // -> ㅗ
257 > AsciiCodeCombo.hk, // -> ㅘ
258 > AsciiCodeCombo.ho, // -> ㅙ
259 > AsciiCodeCombo.hl, // -> ㅚ
260 > AsciiCode.y, // -> ㅛ
261 > AsciiCode.n, // -> ㅜ
262 > AsciiCodeCombo.nj, // -> ㅝ
263 > AsciiCodeCombo.np, // -> ㅞ
264 > AsciiCodeCombo.nl, // -> ㅟ
265 > AsciiCode.b, // -> ㅠ
266 > AsciiCode.m, // -> ㅡ
267 > AsciiCodeCombo.ml, // -> ㅢ
268 > AsciiCode.l, // -> ㅣ
269 > ]);
270 >
271 > /**
272 > * Hangul Jamo - Modern Consonants #2
273 > *
274 > * Range U+11A8..U+11C2
275 > *
276 > * | | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C | D | E | F |
277 > * |--------|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
278 > * | U+11Ax | | | | | | | | | ᆨ | ᆩ | ᆪ | ᆫ | ᆬ | ᆭ | ᆮ | ᆯ |
279 > * | U+11Bx | ᆰ | ᆱ | ᆲ | ᆳ | ᆴ | ᆵ | ᆶ | ᆷ | ᆸ | ᆹ | ᆺ | ᆻ | ᆼ | ᆽ | ᆾ | ᆿ |
280 > * | U+11Cx | ᇀ | ᇁ | ᇂ |
281 > */
282 > const modernFinalConsonants = new Uint16Array([
283 > AsciiCode.r, // ㄱ
284 > AsciiCode.R, // ㄲ
285 > AsciiCodeCombo.rt, // ㄳ
286 > AsciiCode.s, // ㄴ
287 > AsciiCodeCombo.sw, // ㄵ
288 > AsciiCodeCombo.sg, // ㄶ
289 > AsciiCode.e, // ㄷ
290 > AsciiCode.f, // ㄹ
291 > AsciiCodeCombo.fr, // ㄺ
292 > AsciiCodeCombo.fa, // ㄻ
293 > AsciiCodeCombo.fq, // ㄼ
294 > AsciiCodeCombo.ft, // ㄽ
295 > AsciiCodeCombo.fx, // ㄾ
296 > AsciiCodeCombo.fv, // ㄿ
297 > AsciiCodeCombo.fg, // ㅀ
298 > AsciiCode.a, // ㅁ
299 > AsciiCode.q, // ㅂ
300 > AsciiCodeCombo.qt, // ㅄ
301 > AsciiCode.t, // ㅅ
302 > AsciiCode.T, // ㅆ
303 > AsciiCode.d, // ㅇ
304 > AsciiCode.w, // ㅈ
305 > AsciiCode.c, // ㅊ
306 > AsciiCode.z, // ㅋ
307 > AsciiCode.x, // ㅌ
308 > AsciiCode.v, // ㅍ
309 > AsciiCode.g, // ㅎ
310 > ]);
311 >
312 > /**
313 > * Hangul Compatibility Jamo
314 > *
315 > * Range U+3131..U+318F
316 > *
317 > * This includes range includes archaic jamo which we don't consider, these are
318 > * given the NUL character code in order to be ignored.
319 > *
320 > * | | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | A | B | C | D | E | F |
321 > * |--------|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
322 > * | U+313x | | ㄱ | ㄲ | ㄳ | ㄴ | ㄵ | ㄶ | ㄷ | ㄸ | ㄹ | ㄺ | ㄻ | ㄼ | ㄽ | ㄾ | ㄿ |
323 > * | U+314x | ㅀ | ㅁ | ㅂ | ㅃ | ㅄ | ㅅ | ㅆ | ㅇ | ㅈ | ㅉ | ㅊ | ㅋ | ㅌ | ㅍ | ㅎ | ㅏ |
324 > * | U+315x | ㅐ | ㅑ | ㅒ | ㅓ | ㅔ | ㅕ | ㅖ | ㅗ | ㅘ | ㅙ | ㅚ | ㅛ | ㅜ | ㅝ | ㅞ | ㅟ |
325 > * | U+316x | ㅠ | ㅡ | ㅢ | ㅣ | HF | ㅥ | ㅦ | ㅧ | ㅨ | ㅩ | ㅪ | ㅫ | ㅬ | ㅭ | ㅮ | ㅯ |
326 > * | U+317x | ㅰ | ㅱ | ㅲ | ㅳ | ㅴ | ㅵ | ㅶ | ㅷ | ㅸ | ㅹ | ㅺ | ㅻ | ㅼ | ㅽ | ㅾ | ㅿ |
327 > * | U+318x | ㆀ | ㆁ | ㆂ | ㆃ | ㆄ | ㆅ | ㆆ | ㆇ | ㆈ | ㆉ | ㆊ | ㆋ | ㆌ | ㆍ | ㆎ |
328 > */
329 > const compatibilityJamo = new Uint16Array([
330 > AsciiCode.r, // ㄱ
331 > AsciiCode.R, // ㄲ
332 > AsciiCodeCombo.rt, // ㄳ
333 > AsciiCode.s, // ㄴ
334 > AsciiCodeCombo.sw, // ㄵ
335 > AsciiCodeCombo.sg, // ㄶ
336 > AsciiCode.e, // ㄷ
337 > AsciiCode.E, // ㄸ
338 > AsciiCode.f, // ㄹ
339 > AsciiCodeCombo.fr, // ㄺ
340 > AsciiCodeCombo.fa, // ㄻ
341 > AsciiCodeCombo.fq, // ㄼ
342 > AsciiCodeCombo.ft, // ㄽ
343 > AsciiCodeCombo.fx, // ㄾ
344 > AsciiCodeCombo.fv, // ㄿ
345 > AsciiCodeCombo.fg, // ㅀ
346 > AsciiCode.a, // ㅁ
347 > AsciiCode.q, // ㅂ
348 > AsciiCode.Q, // ㅃ
349 > AsciiCodeCombo.qt, // ㅄ
350 > AsciiCode.t, // ㅅ
351 > AsciiCode.T, // ㅆ
352 > AsciiCode.d, // ㅇ
353 > AsciiCode.w, // ㅈ
354 > AsciiCode.W, // ㅉ
355 > AsciiCode.c, // ㅊ
356 > AsciiCode.z, // ㅋ
357 > AsciiCode.x, // ㅌ
358 > AsciiCode.v, // ㅍ
359 > AsciiCode.g, // ㅎ
360 > AsciiCode.k, // ㅏ
361 > AsciiCode.o, // ㅐ
362 > AsciiCode.i, // ㅑ
363 > AsciiCode.O, // ㅒ
364 > AsciiCode.j, // ㅓ
365 > AsciiCode.p, // ㅔ
366 > AsciiCode.u, // ㅕ
367 > AsciiCode.P, // ㅖ
368 > AsciiCode.h, // ㅗ
369 > AsciiCodeCombo.hk, // ㅘ
370 > AsciiCodeCombo.ho, // ㅙ
371 > AsciiCodeCombo.hl, // ㅚ
372 > AsciiCode.y, // ㅛ
373 > AsciiCode.n, // ㅜ
374 > AsciiCodeCombo.nj, // ㅝ
375 > AsciiCodeCombo.np, // ㅞ
376 > AsciiCodeCombo.nl, // ㅟ
377 > AsciiCode.b, // ㅠ
378 > AsciiCode.m, // ㅡ
379 > AsciiCodeCombo.ml, // ㅢ
380 > AsciiCode.l, // ㅣ
381 > // HF: Hangul Filler (everything after this is archaic)
382 > // ㅥ
383 > // ㅦ
384 > // ㅧ
385 > // ㅨ
386 > // ㅩ
387 > // ㅪ
388 > // ㅫ
389 > // ㅬ
390 > // ㅮ
391 > // ㅯ
392 > // ㅰ
393 > // ㅱ
394 > // ㅲ
395 > // ㅳ
396 > // ㅴ
397 > // ㅵ
398 > // ㅶ
399 > // ㅷ
400 > // ㅸ
401 > // ㅹ
402 > // ㅺ
403 > // ㅻ
404 > // ㅼ
405 > // ㅽ
406 > // ㅾ
407 > // ㅿ
408 > // ㆀ
409 > // ㆁ
410 > // ㆂ
411 > // ㆃ
412 > // ㆄ
413 > // ㆅ
414 > // ㆆ
415 > // ㆇ
416 > // ㆈ
417 > // ㆉ
418 > // ㆊ
419 > // ㆋ
420 > // ㆌ
421 > // ㆍ
422 > // ㆎ
423 > ]);
src/vs/base/common/stream.ts 325 covered LOC · 29 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- stream.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from './cancellation.js';
7 > import { onUnexpectedError } from './errors.js';
8 > import { DisposableStore, toDisposable } from './lifecycle.js';
9 >
10 > /**
11 > * The payload that flows in readable stream events.
12 > */
13 > export type ReadableStreamEventPayload<T> = T | Error | 'end';
14 >
15 > export interface ReadableStreamEvents<T> {
16 >
17 > /**
18 > * The 'data' event is emitted whenever the stream is
19 > * relinquishing ownership of a chunk of data to a consumer.
20 > *
21 > * NOTE: PLEASE UNDERSTAND THAT ADDING A DATA LISTENER CAN
22 > * TURN THE STREAM INTO FLOWING MODE. IT IS THEREFOR THE
23 > * LAST LISTENER THAT SHOULD BE ADDED AND NOT THE FIRST
24 > *
25 > * Use `listenStream` as a helper method to listen to
26 > * stream events in the right order.
27 > */
28 > on(event: 'data', callback: (data: T) => void): void;
29 >
30 > /**
31 > * Emitted when any error occurs.
32 > */
33 > on(event: 'error', callback: (err: Error) => void): void;
34 >
35 > /**
36 > * The 'end' event is emitted when there is no more data
37 > * to be consumed from the stream. The 'end' event will
38 > * not be emitted unless the data is completely consumed.
39 > */
40 > on(event: 'end', callback: () => void): void;
41 > }
42 >
43 > /**
44 > * A interface that emulates the API shape of a node.js readable
45 > * stream for use in native and web environments.
46 > */
47 > export interface ReadableStream<T> extends ReadableStreamEvents<T> {
48 >
49 > /**
50 > * Stops emitting any events until resume() is called.
51 > */
52 > pause(): void;
53 >
54 > /**
55 > * Starts emitting events again after pause() was called.
56 > */
57 > resume(): void;
58 >
59 > /**
60 > * Destroys the stream and stops emitting any event.
61 > */
62 > destroy(): void;
63 >
64 > /**
65 > * Allows to remove a listener that was previously added.
66 > */
67 > removeListener(event: string, callback: Function): void;
68 > }
69 >
70 > /**
71 > * A interface that emulates the API shape of a node.js readable
72 > * for use in native and web environments.
73 > */
74 > export interface Readable<T> {
75 >
76 > /**
77 > * Read data from the underlying source. Will return
78 > * null to indicate that no more data can be read.
79 > */
80 > read(): T | null;
81 > }
82 >
83 > export function isReadable<T>(obj: unknown): obj is Readable<T> {
84 const candidate = obj as Readable<T> | undefined;
85 if (!candidate) {
89 return typeof candidate.read === 'function';
90 }
91 > stream.ts
92 > /**
93 > * A interface that emulates the API shape of a node.js writeable
94 > * stream for use in native and web environments.
95 > */
96 > export interface WriteableStream<T> extends ReadableStream<T> {
97 >
98 > /**
99 > * Writing data to the stream will trigger the on('data')
100 > * event listener if the stream is flowing and buffer the
101 > * data otherwise until the stream is flowing.
102 > *
103 > * If a `highWaterMark` is configured and writing to the
104 > * stream reaches this mark, a promise will be returned
105 > * that should be awaited on before writing more data.
106 > * Otherwise there is a risk of buffering a large number
107 > * of data chunks without consumer.
108 > */
109 > write(data: T): void | Promise<void>;
110 >
111 > /**
112 > * Signals an error to the consumer of the stream via the
113 > * on('error') handler if the stream is flowing.
114 > *
115 > * NOTE: call `end` to signal that the stream has ended,
116 > * this DOES NOT happen automatically from `error`.
117 > */
118 > error(error: Error): void;
119 >
120 > /**
121 > * Signals the end of the stream to the consumer. If the
122 > * result is provided, will trigger the on('data') event
123 > * listener if the stream is flowing and buffer the data
124 > * otherwise until the stream is flowing.
125 > */
126 > end(result?: T): void;
127 > }
128 >
129 > /**
130 > * A stream that has a buffer already read. Returns the original stream
131 > * that was read as well as the chunks that got read.
132 > *
133 > * The `ended` flag indicates if the stream has been fully consumed.
134 > */
135 > export interface ReadableBufferedStream<T> {
136 >
137 > /**
138 > * The original stream that is being read.
139 > */
140 > stream: ReadableStream<T>;
141 >
142 > /**
143 > * An array of chunks already read from this stream.
144 > */
145 > buffer: T[];
146 >
147 > /**
148 > * Signals if the stream has ended or not. If not, consumers
149 > * should continue to read from the stream until consumed.
150 > */
151 > ended: boolean;
152 > }
153 >
154 > export function isReadableStream<T>(obj: unknown): obj is ReadableStream<T> {
155 const candidate = obj as ReadableStream<T> | undefined;
156 if (!candidate) {
160 return [candidate.on, candidate.pause, candidate.resume, candidate.destroy].every(fn => typeof fn === 'function');
161 }
162 > stream.ts
163 > export function isReadableBufferedStream<T>(obj: unknown): obj is ReadableBufferedStream<T> {
164 const candidate = obj as ReadableBufferedStream<T> | undefined;
165 if (!candidate) {
169 return isReadableStream(candidate.stream) && Array.isArray(candidate.buffer) && typeof candidate.ended === 'boolean';
170 }
171 > stream.ts
172 > export interface IReducer<T, R = T> {
173 > (data: T[]): R;
174 > }
175 >
176 > export interface IDataTransformer<Original, Transformed> {
177 > (data: Original): Transformed;
178 > }
179 >
180 > export interface IErrorTransformer {
181 > (error: Error): Error;
182 > }
183 >
184 > export interface ITransformer<Original, Transformed> {
185 > data: IDataTransformer<Original, Transformed>;
186 > error?: IErrorTransformer;
187 > }
188 >
189 > export function newWriteableStream<T>(reducer: IReducer<T> | null, options?: WriteableStreamOptions): WriteableStream<T> {
190 return new WriteableStreamImpl<T>(reducer, options);
191 }
192 > stream.ts
193 > export interface WriteableStreamOptions {
194 >
195 > /**
196 > * The number of objects to buffer before WriteableStream#write()
197 > * signals back that the buffer is full. Can be used to reduce
198 > * the memory pressure when the stream is not flowing.
199 > */
200 > highWaterMark?: number;
201 > }
202 >
203 > class WriteableStreamImpl<T> implements WriteableStream<T> {
204 >
205 > private readonly state = {
206 > flowing: false,
207 > ended: false,
208 > destroyed: false
209 > };
210 >
211 > private readonly buffer = {
212 > data: [] as T[],
213 > error: [] as Error[]
214 > };
215 >
216 > private readonly listeners = {
217 > data: [] as { (data: T): void }[],
218 > error: [] as { (error: Error): void }[],
219 > end: [] as { (): void }[]
220 > };
221 >
222 > private readonly pendingWritePromises: Function[] = [];
223 >
224 > /**
225 > * @param reducer a function that reduces the buffered data into a single object;
226 > * because some objects can be complex and non-reducible, we also
227 > * allow passing the explicit `null` value to skip the reduce step
228 > * @param options stream options
229 > */
230 > constructor(private reducer: IReducer<T> | null, private options?: WriteableStreamOptions) { }
231 >
232 > pause(): void {
233 if (this.state.destroyed) {
234 return;
237 this.state.flowing = false;
238 }
239 > stream.ts
240 > resume(): void {
241 if (this.state.destroyed) {
242 return;
252 }
253 }
254 > stream.ts
255 > write(data: T): void | Promise<void> {
256 if (this.state.destroyed) {
257 return;
273 }
274 }
275 > stream.ts
276 > error(error: Error): void {
277 if (this.state.destroyed) {
278 return;
289 }
290 }
291 > stream.ts
292 > end(result?: T): void {
293 if (this.state.destroyed) {
294 return;
312 }
313 }
314 > stream.ts
315 > private emitData(data: T): void {
316 this.listeners.data.slice(0).forEach(listener => listener(data)); // slice to avoid listener mutation from delivering event
317 }
318 > stream.ts
319 > private emitError(error: Error): void {
320 if (this.listeners.error.length === 0) {
321 onUnexpectedError(error); // nobody listened to this error so we log it as unexpected
324 }
325 }
326 > stream.ts
327 > private emitEnd(): void {
328 this.listeners.end.slice(0).forEach(listener => listener()); // slice to avoid listener mutation from delivering event
329 }
330 > stream.ts
331 > on(event: 'data', callback: (data: T) => void): void;
332 > on(event: 'error', callback: (err: Error) => void): void;
333 > on(event: 'end', callback: () => void): void;
334 > on(event: 'data' | 'error' | 'end', callback: ((data: T) => void) | ((err: Error) => void) | (() => void)): void {
335 if (this.state.destroyed) {
336 return;
372 }
373 }
374 > stream.ts
375 > removeListener(event: string, callback: Function): void {
376 if (this.state.destroyed) {
377 return;
401 }
402 }
403 > stream.ts
404 > private flowData(): void {
405 // if buffer is empty, nothing to do
406 if (this.buffer.data.length === 0) {
428 pendingWritePromises.forEach(pendingWritePromise => pendingWritePromise());
429 }
430 > stream.ts
431 > private flowErrors(): void {
432 if (this.listeners.error.length > 0) {
433 for (const error of this.buffer.error) {
438 }
439 }
440 > stream.ts
441 > private flowEnd(): boolean {
442 if (this.state.ended) {
443 this.emitEnd();
448 return false;
449 }
450 > stream.ts
451 > destroy(): void {
452 if (!this.state.destroyed) {
453 this.state.destroyed = true;
464 }
465 }
466 > } stream.ts
467 >
468 > /**
469 > * Helper to fully read a T readable into a T.
470 > */
471 > export function consumeReadable<T>(readable: Readable<T>, reducer: IReducer<T>): T {
472 const chunks: T[] = [];
473
479 return reducer(chunks);
480 }
481 > stream.ts
482 > /**
483 > * Helper to read a T readable up to a maximum of chunks. If the limit is
484 > * reached, will return a readable instead to ensure all data can still
485 > * be read.
486 > */
487 > export function peekReadable<T>(readable: Readable<T>, reducer: IReducer<T>, maxChunks: number): T | Readable<T> {
488 const chunks: T[] = [];
489
527 };
528 }
529 > stream.ts
530 > /**
531 > * Helper to fully read a T stream into a T or consuming
532 > * a stream fully, awaiting all the events without caring
533 > * about the data.
534 > */
535 > export function consumeStream<T, R = T>(stream: ReadableStreamEvents<T>, reducer: IReducer<T, R>): Promise<R>;
536 > export function consumeStream(stream: ReadableStreamEvents<unknown>): Promise<undefined>;
537 > export function consumeStream<T, R = T>(stream: ReadableStreamEvents<T>, reducer?: IReducer<T, R>): Promise<R | undefined> {
538 return new Promise((resolve, reject) => {
539 const chunks: T[] = [];
562 });
563 }
564 > stream.ts
565 > export interface IStreamListener<T> {
566 >
567 > /**
568 > * The 'data' event is emitted whenever the stream is
569 > * relinquishing ownership of a chunk of data to a consumer.
570 > */
571 > onData(data: T): void;
572 >
573 > /**
574 > * Emitted when any error occurs.
575 > */
576 > onError(err: Error): void;
577 >
578 > /**
579 > * The 'end' event is emitted when there is no more data
580 > * to be consumed from the stream. The 'end' event will
581 > * not be emitted unless the data is completely consumed.
582 > */
583 > onEnd(): void;
584 > }
585 >
586 > /**
587 > * Helper to listen to all events of a T stream in proper order.
588 > */
589 > export function listenStream<T>(stream: ReadableStreamEvents<T>, listener: IStreamListener<T>, token?: CancellationToken): void {
590
591 stream.on('error', error => {
610 });
611 }
612 > stream.ts
613 > /**
614 > * Helper to peek up to `maxChunks` into a stream. The return type signals if
615 > * the stream has ended or not. If not, caller needs to add a `data` listener
616 > * to continue reading.
617 > */
618 > export function peekStream<T>(stream: ReadableStream<T>, maxChunks: number): Promise<ReadableBufferedStream<T>> {
619 return new Promise((resolve, reject) => {
620 const streamListeners = new DisposableStore();
666 });
667 }
668 > stream.ts
669 > /**
670 > * Helper to create a readable stream from an existing T.
671 > */
672 > export function toStream<T>(t: T, reducer: IReducer<T>): ReadableStream<T> {
673 const stream = newWriteableStream<T>(reducer);
674
677 return stream;
678 }
679 > stream.ts
680 > /**
681 > * Helper to create an empty stream
682 > */
683 > export function emptyStream(): ReadableStream<never> {
684 const stream = newWriteableStream<never>(() => { throw new Error('not supported'); });
685 stream.end();
687 return stream;
688 }
689 > stream.ts
690 > /**
691 > * Helper to convert a T into a Readable<T>.
692 > */
693 > export function toReadable<T>(t: T): Readable<T> {
694 let consumed = false;
695
706 };
707 }
708 > stream.ts
709 > /**
710 > * Helper to transform a readable stream into another stream.
711 > */
712 > export function transform<Original, Transformed>(stream: ReadableStreamEvents<Original>, transformer: ITransformer<Original, Transformed>, reducer: IReducer<Transformed>): ReadableStream<Transformed> {
713 const target = newWriteableStream<Transformed>(reducer);
714
721 return target;
722 }
723 > stream.ts
724 > /**
725 > * Helper to take an existing readable that will
726 > * have a prefix injected to the beginning.
727 > */
728 > export function prefixedReadable<T>(prefix: T, readable: Readable<T>, reducer: IReducer<T>): Readable<T> {
729 let prefixHandled = false;
730
751 };
752 }
753 > stream.ts
754 > /**
755 > * Helper to take an existing stream that will
756 > * have a prefix injected to the beginning.
757 > */
758 > export function prefixedStream<T>(prefix: T, stream: ReadableStream<T>, reducer: IReducer<T>): ReadableStream<T> {
759 let prefixHandled = false;
760
src/vs/platform/agentHost/common/state/protocol/channels-session/commands.ts 323 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI } from '../common/state.js';
10 > import type { BaseParams } from '../common/commands.js';
11 > import type { SessionActiveClient } from './state.js';
12 > import type { MessageAttachment } from '../channels-chat/state.js';
13 >
14 > // ─── createSession ───────────────────────────────────────────────────────────
15 >
16 > /**
17 > * Creates a new session with the specified agent provider.
18 > *
19 > * If the session URI already exists, the server MUST return an error with code
20 > * `-32003` (`SessionAlreadyExists`).
21 > *
22 > * After creation, the client should subscribe to the session URI to receive state
23 > * updates. The server also broadcasts a `root/sessionAdded` notification to all
24 > * clients.
25 > *
26 > * @category Commands
27 > * @method createSession
28 > * @direction Client → Server
29 > * @messageType Request
30 > * @version 1
31 > * @example
32 > * ```jsonc
33 > * // Client → Server
34 > * { "jsonrpc": "2.0", "id": 2, "method": "createSession",
35 > * "params": { "channel": "ahp-session:/<uuid>", "provider": "copilot" } }
36 > *
37 > * // Server → Client (success)
38 > * { "jsonrpc": "2.0", "id": 2, "result": null }
39 > *
40 > * // Server → Client (failure — provider not found)
41 > * { "jsonrpc": "2.0", "id": 2, "error": { "code": -32002, "message": "No agent for provider" } }
42 > *
43 > * // Server → Client (failure — session already exists)
44 > * { "jsonrpc": "2.0", "id": 2, "error": { "code": -32003, "message": "Session already exists" } }
45 > * ```
46 > */
47 > /**
48 > * Identifies a source session and turn to fork from.
49 > *
50 > * When provided in `createSession`, the server populates the new session with
51 > * content from the source session up to and including the response of the
52 > * specified turn.
53 > */
54 > export interface SessionForkSource {
55 > /** URI of the existing session to fork from */
56 > session: URI;
57 > /** Turn ID in the source session; content up to and including this turn's response is copied */
58 > turnId: string;
59 > }
60 >
61 > export interface CreateSessionParams extends BaseParams {
62 > /** Session URI (client-chosen, e.g. `ahp-session:/<uuid>`) */
63 > channel: URI;
64 > /** Agent provider ID */
65 > provider?: string;
66 > /**
67 > * The working directories the session's agent is granted tool access to.
68 > * A session may span multiple directories; they are equal peers except when
69 > * the agent advertises
70 > * {@link MultipleWorkingDirectoriesCapability.requiresPrimary}, in which case
71 > * one of them should be designated the primary via
72 > * {@link primaryWorkingDirectory}.
73 > *
74 > * A client MUST NOT supply more than one entry unless the agent advertises
75 > * {@link AgentCapabilities.multipleWorkingDirectories}; a server without that
76 > * capability treats only the first entry as the session's working directory
77 > * and ignores the rest. Dispatch `session/workingDirectorySet` /
78 > * `session/workingDirectoryRemoved` to change the set after the session has
79 > * started.
80 > *
81 > * Ignored for forked sessions — a fork inherits its working directories
82 > * from the source session identified by `fork`.
83 > */
84 > workingDirectories?: URI[];
85 > /**
86 > * The primary working directory for the session's **default chat**.
87 > *
88 > * A session has no primary of its own — primary is a per-chat notion (see
89 > * {@link ChatState.primaryWorkingDirectory}). But `createSession` implicitly
90 > * creates the session's default chat, and there is no separate `createChat`
91 > * call to carry that chat's create-time fields. This field is therefore the
92 > * only place a client can designate the **default chat's** primary at birth;
93 > * it is copied into that chat's read-only `primaryWorkingDirectory`. For any
94 > * non-default chat, pass {@link CreateChatParams.primaryWorkingDirectory}
95 > * instead.
96 > *
97 > * When set, it MUST be one of {@link workingDirectories}. A client SHOULD
98 > * supply this when the agent advertises
99 > * {@link MultipleWorkingDirectoriesCapability.requiresPrimary}; a host MAY
100 > * reject creation that omits it, or fall back to the first entry of
101 > * `workingDirectories`. Ignored for forked sessions (a fork inherits the
102 > * source session's chats and their primaries).
103 > */
104 > primaryWorkingDirectory?: URI;
105 > /**
106 > * Fork from an existing session. The new session is populated with content
107 > * from the source session up to and including the specified turn's response.
108 > */
109 > fork?: SessionForkSource;
110 > /**
111 > * Agent-specific configuration values collected via `resolveSessionConfig`.
112 > * Keys and values correspond to the schema returned by the server.
113 > */
114 > config?: Record<string, unknown>;
115 > /**
116 > * Eagerly claim an active client role for the new session.
117 > *
118 > * When provided, the server initializes the session with this client as an
119 > * active client, equivalent to dispatching a `session/activeClientSet`
120 > * action immediately after creation. The `clientId` MUST match the
121 > * `clientId` the creating client supplied in `initialize`.
122 > */
123 > activeClient?: SessionActiveClient;
124 > /**
125 > * Opt-in progress token. When set, the client is offering to receive
126 > * `progress` notifications (see `ProgressParams`) for any long-running work
127 > * the server does to bring this session up — most notably the lazy,
128 > * first-use download of the provider's native SDK. The server echoes this
129 > * exact token on every `progress` frame so the client can correlate it to
130 > * this `createSession` call (and the UI awaiting it).
131 > *
132 > * The token MUST be unique across the client's active requests. The server
133 > * MAY ignore it (e.g. when nothing long-running is needed), in which case no
134 > * `progress` notifications are emitted.
135 > */
136 > progressToken?: string;
137 > }
138 >
139 > // ─── disposeSession ──────────────────────────────────────────────────────────
140 >
141 > /**
142 > * Disposes a session and cleans up server-side resources.
143 > *
144 > * The server broadcasts a `root/sessionRemoved` notification to all clients.
145 > *
146 > * @category Commands
147 > * @method disposeSession
148 > * @direction Client → Server
149 > * @messageType Request
150 > * @version 1
151 > */
152 > export interface DisposeSessionParams extends BaseParams { }
153 >
154 > // ─── fetchTurns ──────────────────────────────────────────────────────────────
155 >
156 > /**
157 > * Requests that the host load older historical turns into a chat state.
158 > *
159 > * The command result does not carry turns. Instead, before responding, the host
160 > * MUST dispatch `chat/turnsLoaded` to insert any loaded turns into the chat
161 > * channel's `turns` state, ahead of the already-loaded window, and update or
162 > * clear `turnsNextCursor`.
163 > *
164 > * Before applying any operation that references a turn outside the currently
165 > * loaded window, the host MUST eagerly load enough older turns into state for
166 > * that operation to reduce against valid state.
167 > *
168 > * @category Commands
169 > * @method fetchTurns
170 > * @direction Client → Server
171 > * @messageType Request
172 > * @version 1
173 > * @example
174 > * ```jsonc
175 > * // Client → Server (load the next page indicated by ChatState.turnsNextCursor)
176 > * { "jsonrpc": "2.0", "id": 8, "method": "fetchTurns",
177 > * "params": { "channel": "ahp-chat:/<uuid>", "cursor": "opaque-cursor" } }
178 > *
179 > * // Server updates chat state, then responds
180 > * { "jsonrpc": "2.0", "id": 8, "result": {} }
181 > * ```
182 > */
183 > export interface FetchTurnsParams extends BaseParams {
184 > /** Chat URI */
185 > channel: URI;
186 > /**
187 > * Opaque cursor from `ChatState.turnsNextCursor`.
188 > *
189 > * The host MUST reject unrecognised cursors with `InvalidParams`. Omit only
190 > * when asking the host to opportunistically load its next older page for the
191 > * chat, if any.
192 > */
193 > cursor?: string;
194 > }
195 >
196 > /**
197 > * Result of the `fetchTurns` command.
198 > */
199 > export interface FetchTurnsResult { }
200 >
201 > // ─── completions ─────────────────────────────────────────────────────────────
202 >
203 > /**
204 > * The kind of completion items being requested.
205 > *
206 > * @category Commands
207 > */
208 > export const enum CompletionItemKind {
209 > /**
210 > * Completions for the text of a {@link Message} the user is composing.
211 > * Each returned item carries an attachment that gets associated with the
212 > * message when accepted.
213 > */
214 > UserMessage = 'userMessage',
215 > }
216 >
217 > /**
218 > * Requests completion items for a partially-typed input (e.g. a user message
219 > * the user is currently composing). Used to power `@`-mention pickers,
220 > * file/symbol references, and similar inline-completion experiences.
221 > *
222 > * Servers SHOULD treat this command as best-effort and return promptly. The
223 > * client SHOULD debounce calls to avoid flooding the server with requests on
224 > * every keystroke.
225 > *
226 > * @category Commands
227 > * @method completions
228 > * @direction Client → Server
229 > * @messageType Request
230 > * @version 1
231 > * @example
232 > * ```jsonc
233 > * // User has typed "look at @foo" and the cursor is just after "@foo".
234 > * // Client → Server
235 > * { "jsonrpc": "2.0", "id": 12, "method": "completions",
236 > * "params": { "kind": "userMessage", "channel": "ahp-chat:/<uuid>",
237 > * "text": "look at @foo", "offset": 12 } }
238 > *
239 > * // Server → Client
240 > * { "jsonrpc": "2.0", "id": 12, "result": {
241 > * "items": [
242 > * {
243 > * "insertText": "@foo.ts",
244 > * "rangeStart": 8,
245 > * "rangeEnd": 12,
246 > * "attachment": {
247 > * "type": "resource",
248 > * "label": "foo.ts",
249 > * "displayKind": "document",
250 > * "uri": "file:///workspace/foo.ts"
251 > * }
252 > * }
253 > * ]
254 > * }}
255 > * ```
256 > */
257 > export interface CompletionsParams extends BaseParams {
258 > /** What kind of completion is being requested. */
259 > kind: CompletionItemKind;
260 > /** The chat URI the completion is being requested for. */
261 > channel: URI;
262 > /**
263 > * The complete text of the input being completed (e.g. the full user
264 > * message text typed so far).
265 > */
266 > text: string;
267 > /**
268 > * The character offset within `text` at which the completion is requested,
269 > * measured in UTF-16 code units. MUST satisfy `0 <= offset <= text.length`.
270 > */
271 > offset: number;
272 > }
273 >
274 > /**
275 > * A single completion item returned by the `completions` command.
276 > *
277 > * When the user accepts an item, the client SHOULD:
278 > * 1. Replace the range `[rangeStart, rangeEnd)` in the input with `insertText`
279 > * (or insert `insertText` at the cursor when the range is omitted).
280 > * 2. Associate the item's `attachment` with the resulting {@link Message}.
281 > *
282 > * @category Commands
283 > */
284 > export interface CompletionItem {
285 > /**
286 > * The text inserted into the input when this item is accepted.
287 > */
288 > insertText: string;
289 >
290 > /**
291 > * If defined, the start of the range in the input's `text` that is replaced
292 > * by `insertText`. The range is the half-open interval
293 > * `[rangeStart, rangeEnd)` of character offsets, measured in UTF-16 code
294 > * units.
295 > *
296 > * When omitted, the client SHOULD insert `insertText` at the cursor.
297 > *
298 > * Note: this range refers to positions in the *current* input. The
299 > * attachment's own `rangeStart`/`rangeEnd` (when present) refer to
300 > * positions in the final {@link Message.text} after the item is
301 > * accepted.
302 > */
303 > rangeStart?: number;
304 >
305 > /**
306 > * The end of the range in the input's `text` that is replaced by
307 > * `insertText`. See {@link rangeStart}.
308 > */
309 > rangeEnd?: number;
310 >
311 > /**
312 > * The attachment associated with this completion item.
313 > */
314 > attachment: MessageAttachment;
315 > }
316 >
317 > /**
318 > * Result of the `completions` command.
319 > */
320 > export interface CompletionsResult {
321 > /** The completion items, in the order the server suggests displaying them. */
322 > items: CompletionItem[];
323 > }
src/vs/workbench/contrib/chat/common/constants.ts 306 covered LOC · 18 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- constants.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Schemas } from '../../../../base/common/network.js';
7 > import { IChatSessionsService, isAgentHostTarget, localChatSessionType, SessionType } from './chatSessionsService.js';
8 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
9 > import { IStorageService } from '../../../../platform/storage/common/storage.js';
10 > import { IWorkspace } from '../../../../platform/workspace/common/workspace.js';
11 > import { isVirtualWorkspace } from '../../../../platform/workspace/common/virtualWorkspace.js';
12 > import { ServicesAccessor } from '../../../../platform/instantiation/common/instantiation.js';
13 > import { ContextKeyExpr, RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
14 > import { ChatEntitlementContextKeys } from '../../../services/chat/common/chatEntitlementService.js';
15 > import { IsAuxiliaryWindowContext, IsSessionsWindowContext } from '../../../common/contextkeys.js';
16 > import { URI } from '../../../../base/common/uri.js';
17 > import { generateUuid } from '../../../../base/common/uuid.js';
18 > import { LocalChatSessionUri } from './model/chatUri.js';
19 > import { clearUserSelectedSessionType, getRememberedSessionType, hasPreferredCopilotHarness, storeUserSelectedSessionType } from './chatSessionTypePreference.js';
20 >
21 > export const enum BYOKUtilityModelDefault {
22 > None = 'none',
23 > MainAgent = 'mainAgent',
24 > Copilot = 'copilot',
25 > }
26 >
27 > export enum ChatConfiguration {
28 > AIDisabled = 'chat.disableAIFeatures',
29 > PluginsEnabled = 'chat.plugins.enabled',
30 > PluginLocations = 'chat.pluginLocations',
31 > PluginMarketplaces = 'chat.plugins.marketplaces',
32 > ExtraMarketplaces = 'chat.plugins.extraMarketplaces',
33 > StrictMarketplaces = 'chat.plugins.strictMarketplaces',
34 > EnabledPlugins = 'chat.plugins.enabledPlugins',
35 > AgentEnabled = 'chat.agent.enabled',
36 > PlanAgentDefaultModel = 'chat.planAgent.defaultModel',
37 > ExploreAgentDefaultModel = 'chat.exploreAgent.defaultModel',
38 > UtilityModel = 'chat.utilityModel',
39 > UtilitySmallModel = 'chat.utilitySmallModel',
40 > BYOKUtilityModelDefault = 'chat.byokUtilityModelDefault',
41 > RequestQueueingDefaultAction = 'chat.requestQueuing.defaultAction',
42 > AgentStatusEnabled = 'chat.agentsControl.enabled',
43 > EditorAssociations = 'chat.editorAssociations',
44 > UnifiedAgentsBar = 'chat.unifiedAgentsBar.enabled',
45 > AgentSessionProjectionEnabled = 'chat.agentSessionProjection.enabled',
46 > ExtensionToolsEnabled = 'chat.extensionTools.enabled',
47 > RepoInfoEnabled = 'chat.repoInfo.enabled',
48 > EditRequests = 'chat.editRequests',
49 > InlineReferencesStyle = 'chat.inlineReferences.style',
50 > AutoReply = 'chat.autoReply',
51 > GlobalAutoApprove = 'chat.tools.global.autoApprove',
52 > AutoApproveEdits = 'chat.tools.edits.autoApprove',
53 > AutoApprovedUrls = 'chat.tools.urls.autoApprove',
54 > EligibleForAutoApproval = 'chat.tools.eligibleForAutoApproval',
55 > EnableMath = 'chat.math.enabled',
56 > CheckpointsEnabled = 'chat.checkpoints.enabled',
57 > ThinkingStyle = 'chat.agent.thinkingStyle',
58 > ThinkingGenerateTitles = 'chat.agent.thinking.generateTitles',
59 > TerminalToolsInThinking = 'chat.agent.thinking.terminalTools',
60 > SimpleTerminalCollapsible = 'chat.tools.terminal.simpleCollapsible',
61 > CompressOutputEnabled = 'chat.tools.compressOutput.enabled',
62 > ThinkingPhrases = 'chat.agent.thinking.phrases',
63 > AutoExpandToolFailures = 'chat.tools.autoExpandFailures',
64 > TodosShowWidget = 'chat.tools.todos.showWidget',
65 > NotifyWindowOnConfirmation = 'chat.notifyWindowOnConfirmation',
66 > NotifyWindowOnResponseReceived = 'chat.notifyWindowOnResponseReceived',
67 > ChatViewSessionsEnabled = 'chat.viewSessions.enabled',
68 > SessionSyncEnabled = 'chat.sessionSync.enabled',
69 > SessionSyncExcludeRepositories = 'chat.sessionSync.excludeRepositories',
70 > ChatViewSessionsGrouping = 'chat.viewSessions.grouping',
71 > ChatViewSessionsOrientation = 'chat.viewSessions.orientation',
72 > ChatViewProgressBadgeEnabled = 'chat.viewProgressBadge.enabled',
73 > ChatContextUsageEnabled = 'chat.contextUsage.enabled',
74 > Verbose = 'chat.verbose',
75 > ProgressBorder = 'chat.progressBorder.enabled',
76 > SubagentToolCustomAgents = 'chat.customAgentInSubagent.enabled',
77 > SubagentsAllowInvocationsFromSubagents = 'chat.subagents.allowInvocationsFromSubagents',
78 > ShowCodeBlockProgressAnimation = 'chat.agent.codeBlockProgress',
79 > RestoreLastPanelSession = 'chat.restoreLastPanelSession',
80 > ExitAfterDelegation = 'chat.exitAfterDelegation',
81 > ExplainChangesEnabled = 'chat.editing.explainChanges.enabled',
82 > RevealNextChangeOnResolve = 'chat.editing.revealNextChangeOnResolve',
83 > OpenChangedFileInDiffEditor = 'chat.editing.openChangedFileInDiffEditor',
84 > GrowthNotificationEnabled = 'chat.growthNotification.enabled',
85 > TitleBarSignInEnabled = 'chat.titleBar.signIn.enabled',
86 > TitleBarOpenInAgentsWindowEnabled = 'chat.titleBar.openInAgentsWindow.enabled',
87 >
88 > ChatCustomizationsStructuredPreviewEnabled = 'chat.customizations.structuredPreview.enabled',
89 > ChatCustomizationsPromptMigrationEnabled = 'chat.customizations.promptMigration.enabled',
90 > AutopilotAdvancedEnabled = 'chat.autopilot.advanced.enabled',
91 > PlanReviewInlineEditorEnabled = 'chat.planReview.inlineEditor.enabled',
92 > DefaultPermissionLevel = 'chat.permissions.default',
93 > AssistedPermissionsEnabled = 'chat.assistedPermissions.enabled',
94 > PermissionsSandboxToggleEnabled = 'chat.experimental.permissionsSandboxToggle.enabled',
95 > DefaultConfiguration = 'chat.defaultConfiguration',
96 > DefaultModel = 'chat.defaultModel',
97 > ImageCarouselEnabled = 'imageCarousel.chat.enabled',
98 > ArtifactsEnabled = 'chat.artifacts.enabled',
99 > ArtifactsRulesByMimeType = 'chat.artifacts.rules.byMimeType',
100 > ArtifactsRulesByFilePath = 'chat.artifacts.rules.byFilePath',
101 > ArtifactsRulesByMemoryFilePath = 'chat.artifacts.rules.byMemoryFilePath',
102 > ToolConfirmationCarousel = 'chat.tools.confirmationCarousel.enabled',
103 > ToolRiskAssessmentEnabled = 'chat.tools.riskAssessment.enabled',
104 > ToolRiskAssessmentModel = 'chat.tools.riskAssessment.model',
105 > DefaultNewSessionMode = 'chat.newSession.defaultMode',
106 > CopilotCliHideExtensionHostAgents = 'chat.agents.copilotCli.hideExtensionHost',
107 > EditorPreferCopilotHarness = 'chat.editor.preferCopilotHarness',
108 > DefaultToCopilotHarness = 'chat.defaultToCopilotHarness',
109 > EditorLocalAgentEnabled = 'chat.editor.localAgent.enabled',
110 > CopilotCliHideExtensionHostEditor = 'chat.editor.copilotCli.hideExtensionHost',
111 > AgentsHandoffTipMode = 'chat.agentsHandoffTip.mode',
112 > TurnStatusPills = 'chat.turnStatusPills',
113 >
114 > IncrementalRendering = 'chat.experimental.incrementalRendering.enabled',
115 > IncrementalRenderingStyle = 'chat.experimental.incrementalRendering.animationStyle',
116 > IncrementalRenderingBuffering = 'chat.experimental.incrementalRendering.buffering',
117 >
118 > CollectInstructionsInExtension = 'chat.experimental.collectInstructionsInExtension',
119 > ImplicitContextActiveEditor = 'chat.implicitContext.includeActiveEditor',
120 > }
121 >
122 > /**
123 > * The "kind" of agents for custom agents.
124 > */
125 > export enum ChatModeKind {
126 > Ask = 'ask',
127 > Edit = 'edit',
128 > Agent = 'agent'
129 > }
130 >
131 > /**
132 > * The permission level controlling tool auto-approval behavior.
133 > */
134 > export enum ChatPermissionLevel {
135 > /** Use existing auto-approve settings */
136 > Default = 'default',
137 > /** Delegate approval decisions to a model */
138 > Assisted = 'assisted',
139 > /** Auto-approve all tool calls, auto-retry on error */
140 > AutoApprove = 'autoApprove',
141 > /** Everything AutoApprove does plus an internal stop hook that continues until the task is done */
142 > Autopilot = 'autopilot'
143 > }
144 >
145 > const chatPermissionLevels = new Set<string>(Object.values(ChatPermissionLevel));
146 >
147 > export function isChatPermissionLevel(level: unknown | undefined): level is ChatPermissionLevel {
148 return chatPermissionLevels.has(level as string);
149 }
150 > constants.ts
151 > /**
152 > * Shape of the {@link ChatConfiguration.DefaultConfiguration}
153 > * object setting. Controls the starting `mode` and `approvals` for new agent-host
154 > * sessions (such as Copilot CLI). All properties are optional — a missing property
155 > * falls back to the per-axis default.
156 > */
157 > export type AgentSessionMode = 'interactive' | 'plan' | 'autopilot';
158 >
159 > /** Approval values exposed by the `chat.defaultConfiguration` setting. */
160 > export enum ChatDefaultPermissionLevel {
161 > Default = 'default',
162 > Assisted = 'assisted',
163 > AllowAll = 'allowAll',
164 > }
165 >
166 > export interface IChatDefaultConfiguration {
167 > /** Starting agent mode: `interactive` / `plan` / `autopilot`. */
168 > readonly mode?: AgentSessionMode;
169 > /** Starting approval level: `default` / `assisted` / `allowAll`. */
170 > readonly approvals?: ChatDefaultPermissionLevel;
171 > }
172 >
173 > /** Maps a default-configuration value to the internal Agent Host permission level. */
174 > export function getChatPermissionLevelFromDefaultConfiguration(value: unknown): ChatPermissionLevel | undefined {
175 switch (value) {
176 case ChatDefaultPermissionLevel.Default:
185 }
186 }
187 > constants.ts
188 > /**
189 > * Returns true if the permission level enables auto-approval of all tool calls.
190 > * Both {@link ChatPermissionLevel.AutoApprove} and {@link ChatPermissionLevel.Autopilot} enable auto-approval.
191 > */
192 > export function isAutoApproveLevel(level: ChatPermissionLevel | undefined): boolean {
193 return level === ChatPermissionLevel.AutoApprove || level === ChatPermissionLevel.Autopilot;
194 }
195 > constants.ts
196 > /**
197 > * True for {@link ChatPermissionLevel.Autopilot} only. Unlike {@link isAutoApproveLevel}, this
198 > * excludes {@link ChatPermissionLevel.AutoApprove}, so it can gate Autopilot-only behavior such as
199 > * risk-based skipping of tool calls.
200 > */
201 > export function isAutopilotLevel(level: ChatPermissionLevel | undefined): boolean {
202 return level === ChatPermissionLevel.Autopilot;
203 }
204 > constants.ts
205 > // Thinking display modes for pinned content
206 > export enum ThinkingDisplayMode {
207 > Collapsed = 'collapsed',
208 > CollapsedPreview = 'collapsedPreview',
209 > FixedScrolling = 'fixedScrolling',
210 > }
211 >
212 > export enum CollapsedToolsDisplayMode {
213 > Off = 'off',
214 > WithThinking = 'withThinking',
215 > Always = 'always',
216 > }
217 >
218 > export enum ChatNotificationMode {
219 > Off = 'off',
220 > WindowNotFocused = 'windowNotFocused',
221 > Always = 'always',
222 > }
223 >
224 > export type RawChatParticipantLocation = 'panel' | 'terminal' | 'notebook' | 'editing-session';
225 >
226 > export enum ChatAgentLocation {
227 > /**
228 > * This is chat, whether it's in the sidebar, a chat editor, or quick chat.
229 > * Leaving the values alone as they are in stored data so we don't have to normalize them.
230 > */
231 > Chat = 'panel',
232 > Terminal = 'terminal',
233 > Notebook = 'notebook',
234 > /**
235 > * EditorInline means inline chat in a text editor.
236 > */
237 > EditorInline = 'editor',
238 > }
239 >
240 > export namespace ChatAgentLocation {
241 > export function fromRaw(value: RawChatParticipantLocation | string): ChatAgentLocation {
242 switch (value) {
243 case 'panel': return ChatAgentLocation.Chat;
248 return ChatAgentLocation.Chat;
249 }
250 > } constants.ts
251 >
252 > /**
253 > * List of file schemes that are always unsupported for use in chat
254 > */
255 > const chatAlwaysUnsupportedFileSchemes = new Set([
256 > Schemas.vscodeChatEditor,
257 > Schemas.walkThrough,
258 > Schemas.vscodeLocalChatSession,
259 > Schemas.vscodeSettings,
260 > Schemas.webviewPanel,
261 > Schemas.vscodeUserData,
262 > Schemas.extension,
263 > 'ccreq',
264 > 'openai-codex', // Codex session custom editor scheme
265 > ]);
266 >
267 > export function isSupportedChatFileScheme(accessor: ServicesAccessor, scheme: string): boolean {
268 const chatService = accessor.get(IChatSessionsService);
269
281 return true;
282 }
283 > constants.ts
284 > /**
285 > * Returns the effective default session type for a new chat in the VS Code
286 > * editor window.
287 > *
288 > * Virtual workspaces always default to {@link localChatSessionType}. Otherwise,
289 > * when the agent host is enabled and `chat.defaultToCopilotHarness` is opted in,
290 > * Agent Host Copilot CLI is the default. It falls back to the local harness
291 > * when enabled, or to the first visible non-local provider.
292 > */
293 > export function getComputedDefaultSessionType(
294 configurationService: IConfigurationService,
295 chatSessionsService: Pick<IChatSessionsService, 'getChatSessionContribution' | 'getAllChatSessionContributions'>,
311 return getVisibleNonLocalEditorChatSessionTypes(configurationService, chatSessionsService, workspace)[0] ?? localChatSessionType;
312 }
313 > constants.ts
314 > export function getComputedDefaultSessionResource(
315 configurationService: IConfigurationService,
316 chatSessionsService: Pick<IChatSessionsService, 'getChatSessionContribution' | 'getAllChatSessionContributions'>,
323 : URI.from({ scheme: defaultType, path: `/untitled-${generateUuid()}` });
324 }
325 > constants.ts
326 > export function isRememberedSessionTypeUsable(
327 sessionType: string,
328 configurationService: IConfigurationService,
338 return !!chatSessionsService.getChatSessionContribution(sessionType);
339 }
340 > constants.ts
341 > export interface IDefaultNewChatSessionTypeOptions {
342 > readonly explicitOverride?: string;
343 > readonly currentSessionType?: string;
344 > }
345 >
346 > export interface IResolvedNewChatSessionType {
347 > /** The session type to open for the new chat. */
348 > readonly sessionType: string;
349 > /**
350 > * True when {@link sessionType} is the one-time `chat.editor.preferCopilotHarness`
351 > * swap. The caller must persist the marker (via `markPreferredCopilotHarness`)
352 > * only once it has actually applied this session type, so the migration is not
353 > * consumed by a caller that discards the result.
354 > */
355 > readonly isPreferCopilotHarnessSwap: boolean;
356 > }
357 >
358 > export function getDefaultNewChatSessionType(
359 configurationService: IConfigurationService,
360 chatSessionsService: Pick<IChatSessionsService, 'getChatSessionContribution' | 'getAllChatSessionContributions'>,
379 return getComputedDefaultSessionType(configurationService, chatSessionsService, workspace, agentHostEnabled);
380 }
381 > constants.ts
382 > export function resolveDefaultNewChatSessionType(
383 configurationService: IConfigurationService,
384 chatSessionsService: Pick<IChatSessionsService, 'getChatSessionContribution' | 'getAllChatSessionContributions'>,
413 return { sessionType: getDefaultNewChatSessionType(configurationService, chatSessionsService, storageService, workspace, agentHostEnabled, options), isPreferCopilotHarnessSwap: false };
414 }
415 > constants.ts
416 function getUsableRememberedSessionType(
417 storageService: IStorageService,
423 return remembered && isRememberedSessionTypeUsable(remembered, configurationService, chatSessionsService, workspace) ? remembered : undefined;
424 }
425 > constants.ts
426 > export function getDefaultNewChatSessionResource(
427 configurationService: IConfigurationService,
428 chatSessionsService: Pick<IChatSessionsService, 'getChatSessionContribution' | 'getAllChatSessionContributions'>,
437 : URI.from({ scheme: defaultType, path: `/untitled-${generateUuid()}` });
438 }
439 > constants.ts
440 > export function recordUserSelectedSessionType(
441 storageService: IStorageService,
442 configurationService: IConfigurationService,
452 }
453 }
454 > constants.ts
455 > export function isEditorLocalAgentEnabled(configurationService: IConfigurationService, workspace: IWorkspace): boolean {
456 return isVirtualWorkspace(workspace) || (configurationService.getValue<boolean>(ChatConfiguration.EditorLocalAgentEnabled) ?? true);
457 }
458 > constants.ts
459 > export function isVisibleEditorChatSessionType(
460 sessionType: string,
461 configurationService: IConfigurationService,
473 return !!chatSessionsService.getChatSessionContribution(sessionType);
474 }
475 > constants.ts
476 function getVisibleNonLocalEditorChatSessionTypes(
477 configurationService: IConfigurationService,
487 return Array.from(sessionTypes);
488 }
489 > constants.ts
490 > export const MANAGE_CHAT_COMMAND_ID = 'workbench.action.chat.manage';
491 > export const CHAT_OPEN_AGENT_HOST_CHAT_COMMAND_ID = 'workbench.action.chat.openAgentHostChat';
492 >
493 > export const OPEN_WORKSPACE_IN_AGENTS_WINDOW_COMMAND_ID = 'workbench.action.openWorkspaceInAgentsWindow';
494 > export const OPEN_AGENTS_WINDOW_COMMAND_ID = 'workbench.action.openAgentsWindow';
495 > export const OPEN_AGENTS_WINDOW_PRECONDITION = ContextKeyExpr.and(
496 > ChatEntitlementContextKeys.Setup.hidden.negate(),
497 > ChatEntitlementContextKeys.Setup.disabledInWorkspace.negate(),
498 > IsSessionsWindowContext.negate(),
499 > ContextKeyExpr.has(`config.${ChatConfiguration.AgentEnabled}`),
500 > IsAuxiliaryWindowContext.negate()
501 > );
502 >
503 > export const ChatEditorTitleMaxLength = 30;
504 >
505 > export const CHAT_TERMINAL_OUTPUT_MAX_PREVIEW_LINES = 1000;
506 > export const CONTEXT_MODELS_EDITOR = new RawContextKey<boolean>('inModelsEditor', false);
507 > export const CONTEXT_MODELS_SEARCH_FOCUS = new RawContextKey<boolean>('inModelsSearch', false);
src/vs/base/common/yaml.ts 301 covered LOC · 54 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- yaml.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../nls.js';
7 >
8 > /**
9 > * Parses a simplified YAML-like input from a single string.
10 > * Supports objects, arrays, primitive types (string, number, boolean, null).
11 > * Tracks positions for error reporting and node locations.
12 > *
13 > * Limitations:
14 > * - No anchors or references
15 > * - No complex types (dates, binary)
16 > * - No single pair implicit entries
17 > *
18 > * @param input A string containing the YAML-like input
19 > * @param errors Array to collect parsing errors
20 > * @returns The parsed representation (YamlMapNode, YamlSequenceNode, or YamlScalarNode)
21 > */
22 > export function parse(input: string, errors: YamlParseError[] = [], options: ParseOptions = {}): YamlNode | undefined {
23 const scanner = new YamlScanner(input);
24 const tokens = scanner.scan();
26 return parser.parse();
27 }
28 > yaml.ts
29 > /**
30 > * Helper to parse a Markdown with YAML frontmatter document
31 > * @returns
32 > */
33 > export function parseFrontMatter(input: string, errors: YamlParseError[] = [], options: ParseOptions = {}): MarkdownNode | undefined {
34 const tokens = new YamlScanner(input).scan();
35 if (tokens.length === 0 || tokens[0].type !== TokenType.DocumentStart) {
46 return new MarkdownNode(header, body);
47 }
48 > yaml.ts
49 > export class MarkdownNode {
50 > constructor(public readonly header: YamlNode | undefined, public readonly body: string) {
51 }
52 > yaml.ts
53 > getStringValue(name: string): string | undefined {
54 if (this.header && this.header.type === 'map') {
55 const property = this.header.properties.find(p => p.key.value === name);
60 return undefined;
61 }
62 > yaml.ts
63 > getStringArrayValue(name: string): string[] | undefined {
64 if (this.header && this.header.type === 'map') {
65 const property = this.header.properties.find(p => p.key.value === name);
76 return undefined;
77 }
78 > yaml.ts
79 > getBooleanValue(name: string): boolean | undefined {
80 const value = this.getStringValue(name);
81 if (value === 'true') {
86 return undefined;
87 }
88 > } yaml.ts
89 >
90 >
91 > /**
92 > * Parses a comma-separated list from a scalar node's value into an array of scalars.
93 > * Handles single-quoted and double-quoted items, trimming surrounding whitespace for
94 > * unquoted items. Offsets on each produced scalar node are relative to the original
95 > * document that the input scalar was parsed from.
96 > *
97 > * Internally wraps the scalar value in `[…]` and delegates to the full YAML parser so
98 > * that quoting, whitespace, and escape handling are consistent with the rest of the parser.
99 > *
100 > * @param scalar A scalar node whose value contains a comma-separated list.
101 > */
102 > export function parseCommaSeparatedList(value: string, offset: number = 0): YamlScalarNode[] {
103 // Wrap the value as a YAML flow sequence and parse it.
104 const parsed = parse(`[${value}]`);
118 return items;
119 }
120 > yaml.ts
121 > // -- AST Node Types ----------------------------------------------------------
122 >
123 > export interface YamlScalarNode {
124 > readonly type: 'scalar';
125 > readonly value: string;
126 > readonly rawValue: string;
127 > readonly startOffset: number;
128 > readonly endOffset: number;
129 > readonly format: 'single' | 'double' | 'none' | 'literal' | 'folded';
130 > }
131 >
132 > export interface YamlMapNode {
133 > readonly type: 'map';
134 > readonly properties: { key: YamlScalarNode; value: YamlNode }[];
135 > readonly style: 'block' | 'flow';
136 > readonly startOffset: number;
137 > readonly endOffset: number;
138 > }
139 >
140 > export interface YamlSequenceNode {
141 > readonly type: 'sequence';
142 > readonly items: YamlNode[];
143 > readonly style: 'block' | 'flow';
144 > readonly startOffset: number;
145 > readonly endOffset: number;
146 > }
147 >
148 > export type YamlNode = YamlSequenceNode | YamlMapNode | YamlScalarNode;
149 >
150 > export interface YamlParseError {
151 > readonly message: string;
152 > readonly startOffset: number;
153 > readonly endOffset: number;
154 > readonly code: string;
155 > }
156 >
157 > export interface ParseOptions {
158 > readonly allowDuplicateKeys?: boolean;
159 > }
160 >
161 > // -- Token Types -------------------------------------------------------------
162 >
163 > const enum TokenType {
164 > // Scalar values (unquoted, single-quoted, double-quoted)
165 > Scalar,
166 > // Structural tokens
167 > Colon, // ':'
168 > Dash, // '- '
169 > Comma, // ','
170 > FlowMapStart, // '{'
171 > FlowMapEnd, // '}'
172 > FlowSeqStart, // '['
173 > FlowSeqEnd, // ']'
174 > // Whitespace / structure
175 > Newline,
176 > Indent, // leading whitespace at start of line (carries the indent level)
177 > Comment,
178 > DocumentStart, // '---'
179 > DocumentEnd, // '...'
180 > EOF,
181 > }
182 >
183 > interface Token {
184 > readonly type: TokenType;
185 > readonly startOffset: number;
186 > readonly endOffset: number;
187 > /** For Scalar tokens: the raw text (including quotes). */
188 > readonly rawValue: string;
189 > /** For Scalar tokens: the interpreted string value. */
190 > readonly value: string;
191 > /** For Scalar tokens: quote style. */
192 > readonly format: 'single' | 'double' | 'none' | 'literal' | 'folded';
193 > /** For Indent tokens: the column (number of spaces). */
194 > readonly indent: number;
195 > }
196 >
197 function makeToken(
198 type: TokenType,
211 };
212 }
213 > yaml.ts
214 > // -- Scanner -----------------------------------------------------------------
215 >
216 > class YamlScanner {
217 > private pos = 0;
218 > private readonly tokens: Token[] = [];
219 > // Track flow nesting depth so commas and flow indicators are only special inside flow collections
220 > private flowDepth = 0;
221 > // Track whether we've already seen a block colon on the current line.
222 > // After the first key: value colon, subsequent ': ' on the same line is part of the scalar value.
223 > private seenBlockColon = false;
224 > private seenDocumentStart = 0;
225 >
226 > constructor(private readonly input: string) { }
227 >
228 > scan(maxDocuments = 1): Token[] {
229 while (this.pos < this.input.length) {
230 this.scanLine();
236 return this.tokens;
237 }
238 > yaml.ts
239 > // Scan a single logical line (up to and including the newline character)
240 > private scanLine(): void {
241 this.seenBlockColon = false;
242 // Handle blank lines / lines that are only whitespace
319 this.scanNewline();
320 }
321 > yaml.ts
322 > private scanLineContent(): void {
323 while (this.pos < this.input.length && this.peekChar() !== '\n' && this.peekChar() !== '\r') {
324 this.skipInlineWhitespace();
376 }
377 }
378 > yaml.ts
379 > /** Check if '-' is a block sequence dash (followed by space, newline, or EOF) */
380 > private isBlockDash(): boolean {
381 const next = this.input[this.pos + 1];
382 return next === undefined || next === ' ' || next === '\t' || next === '\n' || next === '\r';
383 }
384 > yaml.ts
385 > /** Check if ':' acts as a mapping value indicator (followed by space, newline, EOF, or flow indicator) */
386 > private isBlockColon(): boolean {
387 // In block context, after the first key-value colon on a line,
388 // subsequent ': ' is part of the scalar value, not a mapping indicator.
394 return false;
395 }
396 > yaml.ts
397 > /** Check if the last non-whitespace token is a JSON-like node (quoted scalar or flow end) */
398 > private lastTokenIsJsonLike(): boolean {
399 for (let i = this.tokens.length - 1; i >= 0; i--) {
400 const t = this.tokens[i];
409 return false;
410 }
411 > yaml.ts
412 > private scanQuotedScalar(quote: '\'' | '"'): void {
413 const start = this.pos;
414 this.pos++; // skip opening quote
567 }));
568 }
569 > yaml.ts
570 > private scanUnquotedScalar(): void {
571 const start = this.pos;
572 let end = this.pos;
598 }));
599 }
600 > yaml.ts
601 > /**
602 > * Check if '|' or '>' at the current position is a block scalar indicator.
603 > * Must be followed by optional indentation/chomping indicators, optional comment, then newline.
604 > */
605 > private isBlockScalarStart(): boolean {
606 let p = this.pos + 1;
607 // Skip optional indentation indicator (digit 1-9) and chomping indicator (+/-)
619 return c === '\n' || c === '\r' || c === '#';
620 }
621 > yaml.ts
622 > /**
623 > * Scan a block scalar (literal '|' or folded '>').
624 > * Parses the header line for indentation indicator and chomping mode,
625 > * then collects all content lines that are indented beyond the detected indentation.
626 > */
627 > private scanBlockScalar(style: '|' | '>'): void {
628 const start = this.pos;
629 this.pos++; // skip '|' or '>'
849 }));
850 }
851 > yaml.ts
852 > /**
853 > * Determine the parent block's indentation level for a block scalar.
854 > * Looks at preceding tokens to find the context:
855 > * - After Colon: the indentation of the line containing the mapping key
856 > * - After Dash: the column of the dash
857 > * - At document level: -1 (allows content at indent 0)
858 > */
859 > private getParentBlockIndent(blockScalarPos: number): number {
860 for (let i = this.tokens.length - 1; i >= 0; i--) {
861 const t = this.tokens[i];
883 return 0;
884 }
885 > yaml.ts
886 > /**
887 > * Get the column (0-based offset from start of line) for a position in the input.
888 > */
889 > private getColumnAt(offset: number): number {
890 let col = 0;
891 let p = offset - 1;
896 return col;
897 }
898 > yaml.ts
899 > private scanComment(): void {
900 const start = this.pos;
901 while (this.pos < this.input.length && this.input[this.pos] !== '\n' && this.input[this.pos] !== '\r') {
907 }));
908 }
909 > yaml.ts
910 > private scanNewline(): void {
911 const start = this.pos;
912 if (this.consumeNewline()) {
914 }
915 }
916 > yaml.ts
917 > private skipInlineWhitespace(): void {
918 while (this.pos < this.input.length) {
919 const ch = this.input[this.pos];
925 }
926 }
927 > yaml.ts
928 > /** Advance past a newline sequence (\r\n, \n, or \r). Returns true if a newline was consumed. */
929 > private consumeNewline(): boolean {
930 if (this.pos >= this.input.length) { return false; }
931 if (this.input[this.pos] === '\r' && this.input[this.pos + 1] === '\n') {
939 return false;
940 }
941 > yaml.ts
942 > private peekChar(): string {
943 return this.input[this.pos];
944 }
945 > } yaml.ts
946 >
947 > // -- Parser ------------------------------------------------------------------
948 >
949 > class YamlParser {
950 > private pos = 0;
951 >
952 > constructor(
953 private readonly tokens: Token[],
954 private readonly input: string,
956 private readonly options: ParseOptions,
957 ) { }
958 > yaml.ts
959 > parse(): YamlNode | undefined {
960 this.skipNewlinesAndComments();
961 // Skip document start marker (---) if present
970 return result;
971 }
972 > yaml.ts
973 > // -- helpers ----------------------------------------------------------
974 >
975 > private currentToken(): Token {
976 return this.tokens[this.pos];
977 }
978 > yaml.ts
979 > private peek(offset = 0): Token {
980 return this.tokens[Math.min(this.pos + offset, this.tokens.length - 1)];
981 }
982 > yaml.ts
983 > private advance(): Token {
984 const t = this.tokens[this.pos];
985 if (t.type !== TokenType.EOF) {
988 return t;
989 }
990 > yaml.ts
991 > private expect(type: TokenType): Token {
992 const t = this.currentToken();
993 if (t.type === type) {
996 return t;
997 }
998 > yaml.ts
999 > private emitError(message: string, startOffset: number, endOffset: number, code: string): void {
1000 this.errors.push({ message, startOffset, endOffset, code });
1001 }
1002 > yaml.ts
1003 > private skipNewlinesAndComments(): void {
1004 while (
1005 this.currentToken().type === TokenType.Newline ||
1010 }
1011 }
1012 > yaml.ts
1013 > /** Returns true if the current Indent token is followed immediately by Newline/Comment/EOF */
1014 > private isFollowedByNewlineOrComment(): boolean {
1015 const next = this.peek(1);
1016 return next.type === TokenType.Newline || next.type === TokenType.Comment || next.type === TokenType.EOF;
1017 }
1018 > yaml.ts
1019 > /**
1020 > * Determines the current indentation level.
1021 > * If the current token is an Indent, returns its indent value.
1022 > * Otherwise returns 0 (token is at column 0).
1023 > */
1024 > private currentIndent(): number {
1025 if (this.currentToken().type === TokenType.Indent) {
1026 return this.currentToken().indent;
1028 return 0;
1029 }
1030 > yaml.ts
1031 > // -- Main parse entry for a value at a given indentation --------------
1032 >
1033 > private parseValue(parentIndent: number): YamlNode | undefined {
1034 this.skipNewlinesAndComments();
1035 const token = this.currentToken();
1065 return undefined;
1066 }
1067 > yaml.ts
1068 > /** Peek past an optional Indent token to see the first content token */
1069 > private peekPastIndent(): Token {
1070 if (this.currentToken().type === TokenType.Indent) {
1071 return this.peek(1);
1073 return this.currentToken();
1074 }
1075 > yaml.ts
1076 > /** Check if tokens at current position look like a mapping entry (key: value) */
1077 > private looksLikeMapping(): boolean {
1078 let offset = 0;
1079 if (this.peek(offset).type === TokenType.Indent) { offset++; }
1084 return false;
1085 }
1086 > yaml.ts
1087 > // -- Scalar ----------------------------------------------------------
1088 >
1089 > private parseScalar(parentIndent: number = -1): YamlScalarNode {
1090 // Skip indent if present
1091 if (this.currentToken().type === TokenType.Indent) {
1100 return this.parsePlainMultiline(token, parentIndent);
1101 }
1102 > yaml.ts
1103 > /**
1104 > * Parse a multiline plain scalar. The first line's token is already consumed.
1105 > * Continuation lines must be indented deeper than `parentIndent`.
1106 > * Line folding rules:
1107 > * - Single line break → space
1108 > * - Each empty line → preserved as \n
1109 > */
1110 > private parsePlainMultiline(firstToken: Token, parentIndent: number): YamlScalarNode {
1111 let value = firstToken.value;
1112 let endOffset = firstToken.endOffset;
1249 };
1250 }
1251 > yaml.ts
1252 > // -- Block mapping ---------------------------------------------------
1253 >
1254 > private parseBlockMapping(baseIndent: number, inlineFirstEntry = false): YamlMapNode {
1255 const startOffset = this.currentToken().startOffset;
1256 const properties: { key: YamlScalarNode; value: YamlNode }[] = [];
1304 return { type: 'map', properties, style: 'block', startOffset, endOffset };
1305 }
1306 > yaml.ts
1307 > private parseMappingEntry(baseIndent: number): { key: YamlScalarNode; value: YamlNode } | undefined {
1308 // Skip indent
1309 if (this.currentToken().type === TokenType.Indent) {
1327 return { key, value };
1328 }
1329 > yaml.ts
1330 > private parseMappingValue(baseIndent: number, colonToken: Token): YamlNode {
1331 // Check if there's a value on the same line after the colon
1332 const next = this.currentToken();
1377 return this.parseValue(baseIndent) ?? this.makeEmptyScalar(colonToken.endOffset);
1378 }
1379 > yaml.ts
1380 > // -- Block sequence --------------------------------------------------
1381 >
1382 > private parseBlockSequence(baseIndent: number): YamlSequenceNode {
1383 const items: YamlNode[] = [];
1384 const startOffset = this.currentToken().startOffset;
1434 return { type: 'sequence', items, style: 'block', startOffset, endOffset };
1435 }
1436 > yaml.ts
1437 > private parseSequenceItemValue(baseIndent: number, dashToken: Token): YamlNode {
1438 const next = this.currentToken();
1439
1482 return this.parseValue(baseIndent) ?? this.makeEmptyScalar(dashToken.endOffset);
1483 }
1484 > yaml.ts
1485 > /** Calculate the start of the line containing the given offset */
1486 > private getLineStart(offset: number): number {
1487 let i = offset - 1;
1488 while (i >= 0 && this.input[i] !== '\n' && this.input[i] !== '\r') {
1491 return i + 1;
1492 }
1493 > yaml.ts
1494 > // -- Flow map --------------------------------------------------------
1495 >
1496 > private parseFlowMap(): YamlMapNode {
1497 const startToken = this.advance(); // consume '{'
1498 const properties: { key: YamlScalarNode; value: YamlNode }[] = [];
1551 };
1552 }
1553 > yaml.ts
1554 > // -- Flow sequence ---------------------------------------------------
1555 >
1556 > private parseFlowSeq(): YamlSequenceNode {
1557 const startToken = this.advance(); // consume '['
1558 const items: YamlNode[] = [];
1598 };
1599 }
1600 > yaml.ts
1601 > /**
1602 > * Parse a scalar inside a flow collection, handling multiline plain scalars.
1603 > * In flow context, plain (unquoted) scalars can span multiple lines;
1604 > * line breaks are folded into spaces.
1605 > */
1606 > private parseFlowScalar(): YamlScalarNode {
1607 const token = this.advance();
1608 // Quoted scalars are complete as-is (scanner handles their multiline folding)
1652 };
1653 }
1654 > yaml.ts
1655 > /** Parse a value in flow context (used after colon in flow mappings/implicit mappings) */
1656 > private parseFlowValue(): YamlNode {
1657 if (this.currentToken().type === TokenType.FlowMapStart) {
1658 return this.parseFlowMap();
1665 }
1666 }
1667 > yaml.ts
1668 > /** Skip whitespace, newlines, and comments inside flow collections */
1669 > private skipFlowWhitespace(): void {
1670 while (true) {
1671 const t = this.currentToken().type;
1677 }
1678 }
1679 > yaml.ts
1680 > private scalarFromToken(token: Token): YamlScalarNode {
1681 return {
1682 type: 'scalar',
1688 };
1689 }
1690 > yaml.ts
1691 > private makeEmptyScalar(offset: number): YamlScalarNode {
1692 return {
1693 type: 'scalar',
src/vs/base/common/types.ts 299 covered LOC · 31 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- types.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { assert } from './assert.js';
7 >
8 > /**
9 > * @returns whether the provided parameter is a JavaScript String or not.
10 > */
11 > export function isString(str: unknown): str is string {
12 > return (typeof str === 'string'); types.ts
13 > }
14 > types.ts
15 > /**
16 > * @returns whether the provided parameter is a JavaScript Array and each element in the array is a string.
17 > */
18 > export function isStringArray(value: unknown): value is string[] {
19 return isArrayOf(value, isString);
20 }
21 > types.ts
22 > /**
23 > * @returns whether the provided parameter is a JavaScript Array and each element in the array satisfies the provided type guard.
24 > */
25 > export function isArrayOf<T>(value: unknown, check: (item: unknown) => item is T): value is T[] {
26 return Array.isArray(value) && value.every(check);
27 }
28 > types.ts
29 > /**
30 > * @returns whether the provided parameter is of type `object` but **not**
31 > * `null`, an `array`, a `regexp`, nor a `date`.
32 > */
33 > export function isObject(obj: unknown): obj is Object {
34 > // The method can't do a type cast since there are type (like strings) which types.ts
35 > // are subclasses of any put not positvely matched by the function. Hence type
36 > // narrowing results in wrong results.
37 > return typeof obj === 'object'
38 > && obj !== null types.ts
39 > && !Array.isArray(obj)
40 > && !(obj instanceof RegExp) types.ts
41 > && !(obj instanceof Date);
42 > } types.ts
43 > types.ts
44 > /**
45 > * @returns whether the provided parameter is of type `Buffer` or Uint8Array dervived type
46 > */
47 > export function isTypedArray(obj: unknown): obj is Object {
48 const TypedArray = Object.getPrototypeOf(Uint8Array);
49 return typeof obj === 'object'
50 && obj instanceof TypedArray;
51 }
52 > types.ts
53 > /**
54 > * In **contrast** to just checking `typeof` this will return `false` for `NaN`.
55 > * @returns whether the provided parameter is a JavaScript Number or not.
56 > */
57 > export function isNumber(obj: unknown): obj is number {
58 return (typeof obj === 'number' && !isNaN(obj));
59 }
60 > types.ts
61 > /**
62 > * @returns whether the provided parameter is an Iterable, casting to the given generic
63 > */
64 > export function isIterable<T>(obj: unknown): obj is Iterable<T> {
65 // eslint-disable-next-line local/code-no-any-casts
66 return !!obj && typeof (obj as any)[Symbol.iterator] === 'function';
67 }
68 > types.ts
69 > /**
70 > * @returns whether the provided parameter is an Iterable, casting to the given generic
71 > */
72 > export function isAsyncIterable<T>(obj: unknown): obj is AsyncIterable<T> {
73 // eslint-disable-next-line local/code-no-any-casts
74 return !!obj && typeof (obj as any)[Symbol.asyncIterator] === 'function';
75 }
76 > types.ts
77 > /**
78 > * @returns whether the provided parameter is a JavaScript Boolean or not.
79 > */
80 > export function isBoolean(obj: unknown): obj is boolean {
81 return (obj === true || obj === false);
82 }
83 > types.ts
84 > /**
85 > * @returns whether the provided parameter is undefined.
86 > */
87 > export function isUndefined(obj: unknown): obj is undefined {
88 > return (typeof obj === 'undefined'); types.ts
89 > }
90 > types.ts
91 > /**
92 > * @returns whether the provided parameter is defined.
93 > */
94 > export function isDefined<T>(arg: T | null | undefined): arg is T {
95 return !isUndefinedOrNull(arg);
96 }
97 > types.ts
98 > /**
99 > * @returns whether the provided parameter is undefined or null.
100 > */
101 > export function isUndefinedOrNull(obj: unknown): obj is undefined | null {
102 > return (isUndefined(obj) || obj === null); types.ts
103 > }
104 > types.ts
105 >
106 > export function assertType(condition: unknown, type?: string): asserts condition {
107 if (!condition) {
108 throw new Error(type ? `Unexpected type, expected '${type}'` : 'Unexpected type');
109 }
110 }
111 > types.ts
112 > /**
113 > * Asserts that the argument passed in is neither undefined nor null.
114 > *
115 > * @see {@link assertDefined} for a similar utility that leverages TS assertion functions to narrow down the type of `arg` to be non-nullable.
116 > */
117 > export function assertReturnsDefined<T>(arg: T | null | undefined): NonNullable<T> {
118 assert(
119 arg !== null && arg !== undefined,
123 return arg;
124 }
125 > types.ts
126 > /**
127 > * Asserts that a provided `value` is `defined` - not `null` or `undefined`,
128 > * throwing an error with the provided error or error message, while also
129 > * narrowing down the type of the `value` to be `NonNullable` using TS
130 > * assertion functions.
131 > *
132 > * @throws if the provided `value` is `null` or `undefined`.
133 > *
134 > * ## Examples
135 > *
136 > * ```typescript
137 > * // an assert with an error message
138 > * assertDefined('some value', 'String constant is not defined o_O.');
139 > *
140 > * // `throws!` the provided error
141 > * assertDefined(null, new Error('Should throw this error.'));
142 > *
143 > * // narrows down the type of `someValue` to be non-nullable
144 > * const someValue: string | undefined | null = blackbox();
145 > * assertDefined(someValue, 'Some value must be defined.');
146 > * console.log(someValue.length); // now type of `someValue` is `string`
147 > * ```
148 > *
149 > * @see {@link assertReturnsDefined} for a similar utility but without assertion.
150 > * @see {@link https://www.typescriptlang.org/docs/handbook/release-notes/typescript-3-7.html#assertion-functions typescript-3-7.html#assertion-functions}
151 > */
152 > export function assertDefined<T>(value: T, error: string | NonNullable<Error>): asserts value is NonNullable<T> {
153 if (value === null || value === undefined) {
154 const errorToThrow = typeof error === 'string' ? new Error(error) : error;
157 }
158 }
159 > types.ts
160 > /**
161 > * Asserts that each argument passed in is neither undefined nor null.
162 > */
163 > export function assertReturnsAllDefined<T1, T2>(t1: T1 | null | undefined, t2: T2 | null | undefined): [T1, T2];
164 > export function assertReturnsAllDefined<T1, T2, T3>(t1: T1 | null | undefined, t2: T2 | null | undefined, t3: T3 | null | undefined): [T1, T2, T3];
165 > export function assertReturnsAllDefined<T1, T2, T3, T4>(t1: T1 | null | undefined, t2: T2 | null | undefined, t3: T3 | null | undefined, t4: T4 | null | undefined): [T1, T2, T3, T4];
166 > export function assertReturnsAllDefined(...args: (unknown | null | undefined)[]): unknown[] {
167 const result = [];
168
179 return result;
180 }
181 > types.ts
182 > /**
183 > * Checks if the provided value is one of the vales in the provided list.
184 > *
185 > * ## Examples
186 > *
187 > * ```typescript
188 > * // note! item type is a `subset of string`
189 > * type TItem = ':' | '.' | '/';
190 > *
191 > * // note! item is type of `string` here
192 > * const item: string = ':';
193 > * // list of the items to check against
194 > * const list: TItem[] = [':', '.'];
195 > *
196 > * // ok
197 > * assert(
198 > * isOneOf(item, list),
199 > * 'Must succeed.',
200 > * );
201 > *
202 > * // `item` is of `TItem` type now
203 > * ```
204 > */
205 > export const isOneOf = <TType, TSubtype extends TType>(
206 value: TType,
207 validValues: readonly TSubtype[],
211 return validValues.includes(<TSubtype>value);
212 };
213 > types.ts
214 > /**
215 > * Compile-time type check of a variable.
216 > */
217 > export function typeCheck<T = never>(_thing: NoInfer<T>): void { }
218 >
219 > const hasOwnProperty = Object.prototype.hasOwnProperty;
220 >
221 > /**
222 > * @returns whether the provided parameter is an empty JavaScript Object or not.
223 > */
224 > export function isEmptyObject(obj: unknown): obj is object {
225 if (!isObject(obj)) {
226 return false;
235 return true;
236 }
237 > types.ts
238 > /**
239 > * @returns whether the provided parameter is a JavaScript Function or not.
240 > */
241 > export function isFunction(obj: unknown): obj is Function {
242 return (typeof obj === 'function');
243 }
244 > types.ts
245 > /**
246 > * @returns whether the provided parameters is are JavaScript Function or not.
247 > */
248 > export function areFunctions(...objects: unknown[]): boolean {
249 return objects.length > 0 && objects.every(isFunction);
250 }
251 > types.ts
252 > export type TypeConstraint = string | Function;
253 >
254 > export function validateConstraints(args: unknown[], constraints: Array<TypeConstraint | undefined>): void {
255 const len = Math.min(args.length, constraints.length);
256 for (let i = 0; i < len; i++) {
258 }
259 }
260 > types.ts
261 > export function validateConstraint(arg: unknown, constraint: TypeConstraint | undefined): void {
262
263 if (isString(constraint)) {
283 }
284 }
285 > types.ts
286 > /**
287 > * Helper type assertion that safely upcasts a type to a supertype.
288 > *
289 > * This can be used to make sure the argument correctly conforms to the subtype while still being able to pass it
290 > * to contexts that expects the supertype.
291 > */
292 > export function upcast<Base, Sub extends Base = Base>(x: Sub): Base {
293 return x;
294 }
295 > types.ts
296 > type AddFirstParameterToFunction<T, TargetFunctionsReturnType, FirstParameter> = T extends (...args: any[]) => TargetFunctionsReturnType ?
297 > // Function: add param to function
298 > (firstArg: FirstParameter, ...args: Parameters<T>) => ReturnType<T> :
299 >
300 > // Else: just leave as is
301 > T;
302 >
303 > /**
304 > * Allows to add a first parameter to functions of a type.
305 > */
306 > export type AddFirstParameterToFunctions<Target, TargetFunctionsReturnType, FirstParameter> = {
307 > // For every property
308 > [K in keyof Target]: AddFirstParameterToFunction<Target[K], TargetFunctionsReturnType, FirstParameter>;
309 > };
310 >
311 > /**
312 > * Given an object with all optional properties, requires at least one to be defined.
313 > * i.e. AtLeastOne<MyObject>;
314 > */
315 > export type AtLeastOne<T, U = { [K in keyof T]: Pick<T, K> }> = Partial<T> & U[keyof U];
316 >
317 > /**
318 > * Only picks the non-optional properties of a type.
319 > */
320 > export type OmitOptional<T> = { [K in keyof T as T[K] extends Required<T>[K] ? K : never]: T[K] };
321 >
322 > /**
323 > * A type that removed readonly-less from all properties of `T`
324 > */
325 > export type Mutable<T> = {
326 > -readonly [P in keyof T]: T[P]
327 > };
328 >
329 > /**
330 > * A type that adds readonly to all properties of T, recursively.
331 > */
332 > export type DeepImmutable<T> = T extends (infer U)[]
333 > ? ReadonlyArray<DeepImmutable<U>>
334 > : T extends ReadonlyArray<infer U>
335 > ? ReadonlyArray<DeepImmutable<U>>
336 > : T extends Map<infer K, infer V>
337 > ? ReadonlyMap<K, DeepImmutable<V>>
338 > : T extends Set<infer U>
339 > ? ReadonlySet<DeepImmutable<U>>
340 > : T extends object
341 > ? {
342 > readonly [K in keyof T]: DeepImmutable<T[K]>;
343 > }
344 > : T;
345 >
346 > /**
347 > * A single object or an array of the objects.
348 > */
349 > export type SingleOrMany<T> = T | T[];
350 >
351 > /**
352 > * Given a `type X = { foo?: string }` checking that an object `satisfies X`
353 > * will ensure each property was explicitly defined, ensuring no properties
354 > * are omitted or forgotten.
355 > */
356 > export type WithDefinedProps<T> = { [K in keyof Required<T>]: T[K] };
357 >
358 >
359 > /**
360 > * A type that recursively makes all properties of `T` required
361 > */
362 > export type DeepRequiredNonNullable<T> = {
363 > [P in keyof T]-?: T[P] extends object ? DeepRequiredNonNullable<T[P]> : Required<NonNullable<T[P]>>;
364 > };
365 >
366 >
367 > /**
368 > * Represents a type that is a partial version of a given type `T`, where all properties are optional and can be deeply nested.
369 > */
370 > export type DeepPartial<T> = {
371 > [P in keyof T]?: T[P] extends object ? DeepPartial<T[P]> : Partial<T[P]>;
372 > };
373 >
374 > /**
375 > * Represents a type that is a partial version of a given type `T`, except a subset.
376 > */
377 > export type PartialExcept<T, K extends keyof T> = Partial<Omit<T, K>> & Pick<T, K>;
378 >
379 >
380 > type KeysOfUnionType<T> = T extends T ? keyof T : never;
381 > type FilterType<T, TTest> = T extends TTest ? T : never;
382 > type MakeOptionalAndTrue<T extends object> = { [K in keyof T]?: true };
383 >
384 > /**
385 > * Type guard that checks if an object has specific keys and narrows the type accordingly.
386 > *
387 > * @param x - The object to check
388 > * @param key - An object with boolean values indicating which keys to check for
389 > * @returns true if all specified keys exist in the object, false otherwise
390 > *
391 > * @example
392 > * ```typescript
393 > * type A = { a: string };
394 > * type B = { b: number };
395 > * const obj: A | B = getObject();
396 > *
397 > * if (hasKey(obj, { a: true })) {
398 > * // obj is now narrowed to type A
399 > * console.log(obj.a);
400 > * }
401 > * ```
402 > */
403 > export function hasKey<T extends object, TKeys extends MakeOptionalAndTrue<T>>(x: T, key: TKeys): x is FilterType<T, { [K in KeysOfUnionType<T> & keyof TKeys]: unknown }> {
404 for (const k in key) {
405 if (!(k in x)) {
src/vs/base/common/observableInternal/observables/derivedImpl.ts 297 covered LOC · 64 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- derivedImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IObservableWithChange, IObserver, IReaderWithStore, ISettableObservable, ITransaction, } from '../base.js';
7 > import { BaseObservable } from './baseObservable.js';
8 > import { DebugNameData } from '../debugName.js';
9 > import { BugIndicatingError, DisposableStore, EqualityComparer, assertFn, onBugIndicatingError } from '../commonFacade/deps.js';
10 > import { getLogger } from '../logging/logging.js';
11 > import { IChangeTracker } from '../changeTracker.js';
12 > import { DebugLocation } from '../debugLocation.js';
13 >
14 > export interface IDerivedReader<TChange = void> extends IReaderWithStore {
15 > /**
16 > * Call this to report a change delta or to force report a change, even if the new value is the same as the old value.
17 > */
18 > reportChange(change: TChange): void;
19 > }
20 >
21 > export const enum DerivedState {
22 > /** Initial state, no previous value, recomputation needed */
23 > initial = 0,
24 >
25 > /**
26 > * A dependency could have changed.
27 > * We need to explicitly ask them if at least one dependency changed.
28 > */
29 > dependenciesMightHaveChanged = 1,
30 >
31 > /**
32 > * A dependency changed and we need to recompute.
33 > * After recomputation, we need to check the previous value to see if we changed as well.
34 > */
35 > stale = 2,
36 >
37 > /**
38 > * No change reported, our cached value is up to date.
39 > */
40 > upToDate = 3,
41 > }
42 >
43 function derivedStateToString(state: DerivedState): string {
44 switch (state) {
50 }
51 }
53 > export class Derived<T, TChangeSummary = any, TChange = void> extends BaseObservable<T, TChange> implements IDerivedReader<TChange>, IObserver {
54 > private _state = DerivedState.initial;
55 > private _value: T | undefined = undefined;
56 > private _updateCount = 0;
57 > private _dependencies = new Set<IObservable<any>>();
58 > private _dependenciesToBeRemoved = new Set<IObservable<any>>();
59 > private _changeSummary: TChangeSummary | undefined = undefined;
60 > private _isUpdating = false;
61 > private _isComputing = false;
62 > private _didReportChange = false;
63 > private _isInBeforeUpdate = false;
64 > private _isReaderValid = false;
65 > private _store: DisposableStore | undefined = undefined;
66 > private _delayedStore: DisposableStore | undefined = undefined;
67 > private _removedObserverToCallEndUpdateOn: Set<IObserver> | null = null;
68 >
69 > public override get debugName(): string {
70 > return this._debugNameData.getDebugName(this) ?? '(anonymous)';
71 > }
72 >
73 > constructor(
74 > public readonly _debugNameData: DebugNameData, derivedImpl.ts
75 > public readonly _computeFn: (reader: IDerivedReader<TChange>, changeSummary: TChangeSummary) => T,
76 > private readonly _changeTracker: IChangeTracker<TChangeSummary> | undefined,
77 > private readonly _handleLastObserverRemoved: (() => void) | undefined = undefined,
78 > private readonly _equalityComparator: EqualityComparer<T>,
79 > debugLocation: DebugLocation,
80 > ) {
81 > super(debugLocation);
82 > this._changeSummary = this._changeTracker?.createChangeSummary(undefined);
83 > }
85 > protected override onLastObserverRemoved(): void {
86 > /** derivedImpl.ts
87 > * We are not tracking changes anymore, thus we have to assume
88 > * that our cache is invalid.
89 > */
90 > this._state = DerivedState.initial;
91 > this._value = undefined;
92 > getLogger()?.handleDerivedCleared(this);
93 > for (const d of this._dependencies) {
94 > d.removeObserver(this); derivedImpl.ts
95 > }
96 > this._dependencies.clear(); derivedImpl.ts
97 >
98 > if (this._store !== undefined) {
99 this._store.dispose();
100 this._store = undefined;
101 }
102 > if (this._delayedStore !== undefined) { derivedImpl.ts
103 this._delayedStore.dispose();
104 this._delayedStore = undefined;
105 }
107 > this._handleLastObserverRemoved?.();
108 > }
110 > public override get(): T {
111 > const checkEnabled = false; // TODO set to true derivedImpl.ts
112 > if (this._isComputing && checkEnabled) {
113 // investigate why this fails in the diff editor!
114 throw new BugIndicatingError('Cyclic deriveds are not supported yet!');
115 }
117 > if (this._observers.size === 0) {
118 let result;
119 // Without observers, we don't know when to clean up stuff.
134 return result;
135
136 > } else { derivedImpl.ts
137 > do { derivedImpl.ts
138 > // We might not get a notification for a dependency that changed while it is updating,
139 > // thus we also have to ask all our depedencies if they changed in this case.
140 > if (this._state === DerivedState.dependenciesMightHaveChanged) {
141 > for (const d of this._dependencies) { derivedImpl.ts
142 > /** might call {@link handleChange} indirectly, which could make us stale */
143 > d.reportChanges();
144 >
145 > if (this._state as DerivedState === DerivedState.stale) {
146 > // The other dependencies will refresh on demand, so early break derivedImpl.ts
147 > break;
148 > }
149 > } derivedImpl.ts
150 > }
152 > // We called report changes of all dependencies.
153 > // If we are still not stale, we can assume to be up to date again.
154 > if (this._state === DerivedState.dependenciesMightHaveChanged) {
155 > this._state = DerivedState.upToDate; derivedImpl.ts
156 > }
158 > if (this._state !== DerivedState.upToDate) {
159 > this._recompute();
160 > }
161 > // In case recomputation changed one of our dependencies, we need to recompute again.
162 > } while (this._state !== DerivedState.upToDate);
163 > return this._value!;
164 > }
165 > } derivedImpl.ts
167 > private _recompute() {
168 > let didChange = false; derivedImpl.ts
169 > this._isComputing = true;
170 > this._didReportChange = false;
171 >
172 > const emptySet = this._dependenciesToBeRemoved;
173 > this._dependenciesToBeRemoved = this._dependencies;
174 > this._dependencies = emptySet;
175 >
176 > try {
177 > const changeSummary = this._changeSummary!;
178 >
179 > this._isReaderValid = true;
180 > if (this._changeTracker) {
181 this._isInBeforeUpdate = true;
182 this._changeTracker.beforeUpdate?.(this, changeSummary);
184 this._changeSummary = this._changeTracker?.createChangeSummary(changeSummary);
185 }
187 > const hadValue = this._state !== DerivedState.initial;
188 > const oldValue = this._value;
189 > this._state = DerivedState.upToDate;
190 >
191 > const delayedStore = this._delayedStore;
192 > if (delayedStore !== undefined) {
193 this._delayedStore = undefined;
194 }
195 > try { derivedImpl.ts
196 > if (this._store !== undefined) {
197 this._store.dispose();
198 this._store = undefined;
199 }
200 > /** might call {@link handleChange} indirectly, which could invalidate us */ derivedImpl.ts
201 > this._value = this._computeFn(this, changeSummary);
202 >
203 > } finally {
204 > this._isReaderValid = false;
205 > // We don't want our observed observables to think that they are (not even temporarily) not being observed.
206 > // Thus, we only unsubscribe from observables that are definitely not read anymore.
207 > for (const o of this._dependenciesToBeRemoved) {
208 o.removeObserver(this);
209 }
210 > this._dependenciesToBeRemoved.clear(); derivedImpl.ts
211 >
212 > if (delayedStore !== undefined) {
213 delayedStore.dispose();
214 }
215 > } derivedImpl.ts
216 >
217 > didChange = this._didReportChange || (hadValue && !(this._equalityComparator(oldValue!, this._value)));
218 >
219 > getLogger()?.handleObservableUpdated(this, {
220 > oldValue,
221 > newValue: this._value,
222 > change: undefined,
223 > didChange,
224 > hadValue,
225 > });
226 > } catch (e) {
227 onBugIndicatingError(e);
228 }
230 > this._isComputing = false;
231 >
232 > if (!this._didReportChange && didChange) {
233 > for (const r of this._observers) { derivedImpl.ts
234 > r.handleChange(this, undefined);
235 > }
236 > } else { derivedImpl.ts
237 > this._didReportChange = false;
238 > }
239 > }
241 > public override toString(): string {
242 return `LazyDerived<${this.debugName}>`;
243 }
245 > // IObserver Implementation
246 >
247 > public beginUpdate<T>(_observable: IObservable<T>): void {
248 > if (this._isUpdating) { derivedImpl.ts
249 throw new BugIndicatingError('Cyclic deriveds are not supported yet!');
250 }
252 > this._updateCount++;
253 > this._isUpdating = true;
254 > try {
255 > const propagateBeginUpdate = this._updateCount === 1;
256 > if (this._state === DerivedState.upToDate) {
257 > this._state = DerivedState.dependenciesMightHaveChanged;
258 > // If we propagate begin update, that will already signal a possible change.
259 > if (!propagateBeginUpdate) {
260 for (const r of this._observers) {
261 r.handlePossibleChange(this);
262 }
263 }
264 > } derivedImpl.ts
265 > if (propagateBeginUpdate) {
266 > for (const r of this._observers) {
267 > r.beginUpdate(this); // This signals a possible change
268 > }
269 > }
270 > } finally {
271 > this._isUpdating = false;
272 > }
273 > }
275 > public endUpdate<T>(_observable: IObservable<T>): void {
276 > this._updateCount--; derivedImpl.ts
277 > if (this._updateCount === 0) {
278 > // End update could change the observer list.
279 > const observers = [...this._observers];
280 > for (const r of observers) {
281 > r.endUpdate(this);
282 > }
283 > if (this._removedObserverToCallEndUpdateOn) {
284 const observers = [...this._removedObserverToCallEndUpdateOn];
285 this._removedObserverToCallEndUpdateOn = null;
288 }
289 }
290 > } derivedImpl.ts
291 > assertFn(() => this._updateCount >= 0);
292 > }
294 > public handlePossibleChange<T>(observable: IObservable<T>): void {
295 // In all other states, observers already know that we might have changed.
296 if (this._state === DerivedState.upToDate && this._dependencies.has(observable) && !this._dependenciesToBeRemoved.has(observable)) {
301 }
302 }
304 > public handleChange<T, TChange>(observable: IObservableWithChange<T, TChange>, change: TChange): void {
305 > if (this._dependencies.has(observable) && !this._dependenciesToBeRemoved.has(observable) || this._isInBeforeUpdate) { derivedImpl.ts
306 > getLogger()?.handleDerivedDependencyChanged(this, observable, change);
307 >
308 > let shouldReact = false;
309 > try {
310 > shouldReact = this._changeTracker ? this._changeTracker.handleChange({
311 changedObservable: observable,
312 change,
313 // eslint-disable-next-line local/code-no-any-casts
314 didChange: (o): this is any => o === observable as any,
315 > }, this._changeSummary!) : true; derivedImpl.ts
316 > } catch (e) {
317 onBugIndicatingError(e);
318 }
320 > const wasUpToDate = this._state === DerivedState.upToDate;
321 > if (shouldReact && (this._state === DerivedState.dependenciesMightHaveChanged || wasUpToDate)) {
322 > this._state = DerivedState.stale;
323 > if (wasUpToDate) {
324 for (const r of this._observers) {
325 r.handlePossibleChange(this);
326 }
327 }
328 > } derivedImpl.ts
329 > }
330 > }
332 > // IReader Implementation
333 >
334 > private _ensureReaderValid(): void {
335 > if (!this._isReaderValid) { throw new BugIndicatingError('The reader object cannot be used outside its compute function!'); } derivedImpl.ts
336 > }
338 > public readObservable<T>(observable: IObservable<T>): T {
339 > this._ensureReaderValid(); derivedImpl.ts
340 >
341 > // Subscribe before getting the value to enable caching
342 > observable.addObserver(this);
343 > /** This might call {@link handleChange} indirectly, which could invalidate us */
344 > const value = observable.get();
345 > // Which is why we only add the observable to the dependencies now.
346 > this._dependencies.add(observable);
347 > this._dependenciesToBeRemoved.delete(observable);
348 > return value;
349 > }
351 > public reportChange(change: TChange): void {
352 this._ensureReaderValid();
353
358 }
359 }
361 > get store(): DisposableStore {
362 this._ensureReaderValid();
363
367 return this._store;
368 }
370 > get delayedStore(): DisposableStore {
371 this._ensureReaderValid();
372
376 return this._delayedStore;
377 }
379 > public override addObserver(observer: IObserver): void {
380 > const shouldCallBeginUpdate = !this._observers.has(observer) && this._updateCount > 0; derivedImpl.ts
381 > super.addObserver(observer);
382 >
383 > if (shouldCallBeginUpdate) {
384 if (!this._removedObserverToCallEndUpdateOn?.delete(observer)) {
385 observer.beginUpdate(this);
386 }
387 }
388 > } derivedImpl.ts
390 > public override removeObserver(observer: IObserver): void {
391 > if (this._observers.has(observer) && this._updateCount > 0) { derivedImpl.ts
392 if (!this._removedObserverToCallEndUpdateOn) {
393 this._removedObserverToCallEndUpdateOn = new Set();
395 this._removedObserverToCallEndUpdateOn.add(observer);
396 }
397 > super.removeObserver(observer); derivedImpl.ts
398 > }
400 > public debugGetState() {
401 return {
402 state: this._state,
408 };
409 }
411 > public debugSetValue(newValue: unknown) {
412 // eslint-disable-next-line local/code-no-any-casts
413 this._value = newValue as any;
414 }
416 > public debugRecompute(): void {
417 this.beginUpdate(this);
418 try {
426 }
427 }
429 > public setValue(newValue: T, tx: ITransaction, change: TChange): void {
430 this._value = newValue;
431 const observers = this._observers;
435 }
436 }
437 > } derivedImpl.ts
438 >
439 >
440 > export class DerivedWithSetter<T, TChangeSummary = any, TOutChanges = any> extends Derived<T, TChangeSummary, TOutChanges> implements ISettableObservable<T, TOutChanges> {
441 > constructor(
442 debugNameData: DebugNameData,
443 computeFn: (reader: IDerivedReader<TOutChanges>, changeSummary: TChangeSummary) => T,
src/vs/base/common/network.ts 292 covered LOC · 20 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- network.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as errors from './errors.js';
7 > import * as platform from './platform.js';
8 > import { equalsIgnoreCase, startsWithIgnoreCase } from './strings.js';
9 > import { URI } from './uri.js';
10 > import * as paths from './path.js';
11 >
12 > export namespace Schemas {
13 >
14 > /**
15 > * A schema that is used for models that exist in memory
16 > * only and that have no correspondence on a server or such.
17 > */
18 > export const inMemory = 'inmemory';
19 >
20 > /**
21 > * A schema that is used for setting files
22 > */
23 > export const vscode = 'vscode';
24 >
25 > /**
26 > * A schema that is used for internal private files
27 > */
28 > export const internal = 'private';
29 >
30 > /**
31 > * A walk-through document.
32 > */
33 > export const walkThrough = 'walkThrough';
34 >
35 > /**
36 > * An embedded code snippet.
37 > */
38 > export const walkThroughSnippet = 'walkThroughSnippet';
39 >
40 > export const http = 'http';
41 >
42 > export const https = 'https';
43 >
44 > export const file = 'file';
45 >
46 > export const mailto = 'mailto';
47 >
48 > export const untitled = 'untitled';
49 >
50 > export const data = 'data';
51 >
52 > export const command = 'command';
53 >
54 > export const vscodeRemote = 'vscode-remote';
55 >
56 > export const vscodeRemoteResource = 'vscode-remote-resource';
57 >
58 > export const vscodeManagedRemoteResource = 'vscode-managed-remote-resource';
59 >
60 > export const vscodeUserData = 'vscode-userdata';
61 >
62 > export const vscodeCustomEditor = 'vscode-custom-editor';
63 >
64 > export const vscodeNotebookCell = 'vscode-notebook-cell';
65 > export const vscodeNotebookCellMetadata = 'vscode-notebook-cell-metadata';
66 > export const vscodeNotebookCellMetadataDiff = 'vscode-notebook-cell-metadata-diff';
67 > export const vscodeNotebookCellOutput = 'vscode-notebook-cell-output';
68 > export const vscodeNotebookCellOutputDiff = 'vscode-notebook-cell-output-diff';
69 > export const vscodeNotebookMetadata = 'vscode-notebook-metadata';
70 > export const vscodeInteractiveInput = 'vscode-interactive-input';
71 >
72 > export const vscodeSettings = 'vscode-settings';
73 >
74 > export const vscodeWorkspaceTrust = 'vscode-workspace-trust';
75 >
76 > export const vscodeTerminal = 'vscode-terminal';
77 >
78 > /** Scheme used for the image carousel editor. */
79 > export const vscodeImageCarousel = 'vscode-image-carousel';
80 >
81 > /** Scheme used for code blocks in chat. */
82 > export const vscodeChatCodeBlock = 'vscode-chat-code-block';
83 >
84 > /** Scheme used for LHS of code compare (aka diff) blocks in chat. */
85 > export const vscodeChatCodeCompareBlock = 'vscode-chat-code-compare-block';
86 >
87 > /** Scheme used for the chat input editor. */
88 > export const vscodeChatEditor = 'vscode-chat-editor';
89 >
90 > /** Scheme used for the chat input part */
91 > export const vscodeChatInput = 'chatSessionInput';
92 >
93 > /** Scheme used for local chat session content */
94 > export const vscodeLocalChatSession = 'vscode-chat-session';
95 >
96 > /**
97 > * Scheme used internally for webviews that aren't linked to a resource (i.e. not custom editors)
98 > */
99 > export const webviewPanel = 'webview-panel';
100 >
101 > /**
102 > * Scheme used for loading the wrapper html and script in webviews.
103 > */
104 > export const vscodeWebview = 'vscode-webview';
105 >
106 > /**
107 > * Scheme used for integrated browser tabs using WebContentsView.
108 > */
109 > export const vscodeBrowser = 'vscode-browser';
110 >
111 > /**
112 > * Scheme used for extension pages
113 > */
114 > export const extension = 'extension';
115 >
116 > /**
117 > * Scheme used as a replacement of `file` scheme to load
118 > * files with our custom protocol handler (desktop only).
119 > */
120 > export const vscodeFileResource = 'vscode-file';
121 >
122 > /**
123 > * Scheme used for temporary resources
124 > */
125 > export const tmp = 'tmp';
126 >
127 > /**
128 > * Scheme used vs live share
129 > */
130 > export const vsls = 'vsls';
131 >
132 > /**
133 > * Scheme used for the Source Control commit input's text document
134 > */
135 > export const vscodeSourceControl = 'vscode-scm';
136 >
137 > /**
138 > * Scheme used for input box for creating comments.
139 > */
140 > export const commentsInput = 'comment';
141 >
142 > /**
143 > * Scheme used for special rendering of settings in the release notes
144 > */
145 > export const codeSetting = 'code-setting';
146 >
147 > /**
148 > * Scheme used for output panel resources
149 > */
150 > export const outputChannel = 'output';
151 >
152 > /**
153 > * Scheme used for the accessible view
154 > */
155 > export const accessibleView = 'accessible-view';
156 >
157 > /**
158 > * Used for snapshots of chat edits
159 > */
160 > export const chatEditingSnapshotScheme = 'chat-editing-snapshot-text-model';
161 > export const chatEditingModel = 'chat-editing-text-model';
162 >
163 > /**
164 > * Used for rendering multidiffs in copilot agent sessions
165 > */
166 > export const copilotPr = 'copilot-pr';
167 > }
168 >
169 > export function matchesScheme(target: URI | string, scheme: string): boolean {
170 if (URI.isUri(target)) {
171 return equalsIgnoreCase(target.scheme, scheme);
174 }
175 }
176 > network.ts
177 > export function matchesSomeScheme(target: URI | string, ...schemes: string[]): boolean {
178 return schemes.some(scheme => matchesScheme(target, scheme));
179 }
180 > network.ts
181 > export const connectionTokenCookieName = 'vscode-tkn';
182 > export const connectionTokenQueryName = 'tkn';
183 >
184 > class RemoteAuthoritiesImpl {
185 > private readonly _hosts: { [authority: string]: string | undefined } = Object.create(null);
186 > private readonly _ports: { [authority: string]: number | undefined } = Object.create(null);
187 > private readonly _connectionTokens: { [authority: string]: string | undefined } = Object.create(null);
188 > private _preferredWebSchema: 'http' | 'https' = 'http';
189 > private _delegate: ((uri: URI) => URI) | null = null;
190 > private _serverRootPath: string = '/';
191 >
192 > setPreferredWebSchema(schema: 'http' | 'https') {
193 this._preferredWebSchema = schema;
194 }
195 > network.ts
196 > setDelegate(delegate: (uri: URI) => URI): void {
197 this._delegate = delegate;
198 }
199 > network.ts
200 > setServerRootPath(product: { quality?: string; commit?: string }, serverBasePath: string | undefined): void {
201 this._serverRootPath = paths.posix.join(serverBasePath ?? '/', getServerProductSegment(product));
202 }
203 > network.ts
204 > getServerRootPath(): string {
205 return this._serverRootPath;
206 }
207 > network.ts
208 > private get _remoteResourcesPath(): string {
209 return paths.posix.join(this._serverRootPath, Schemas.vscodeRemoteResource);
210 }
211 > network.ts
212 > set(authority: string, host: string, port: number): void {
213 this._hosts[authority] = host;
214 this._ports[authority] = port;
215 }
216 > network.ts
217 > setConnectionToken(authority: string, connectionToken: string): void {
218 this._connectionTokens[authority] = connectionToken;
219 }
220 > network.ts
221 > getPreferredWebSchema(): 'http' | 'https' {
222 return this._preferredWebSchema;
223 }
224 > network.ts
225 > rewrite(uri: URI): URI {
226 if (this._delegate) {
227 try {
250 });
251 }
252 > } network.ts
253 >
254 > export const RemoteAuthorities = new RemoteAuthoritiesImpl();
255 >
256 > export function getServerProductSegment(product: { quality?: string; commit?: string }) {
257 return `${product.quality ?? 'oss'}-${product.commit ?? 'dev'}`;
258 }
259 > network.ts
260 > /**
261 > * A string pointing to a path inside the app. It should not begin with ./ or ../
262 > */
263 > export type AppResourcePath = (
264 > `a${string}` | `b${string}` | `c${string}` | `d${string}` | `e${string}` | `f${string}`
265 > | `g${string}` | `h${string}` | `i${string}` | `j${string}` | `k${string}` | `l${string}`
266 > | `m${string}` | `n${string}` | `o${string}` | `p${string}` | `q${string}` | `r${string}`
267 > | `s${string}` | `t${string}` | `u${string}` | `v${string}` | `w${string}` | `x${string}`
268 > | `y${string}` | `z${string}`
269 > );
270 >
271 > export const builtinExtensionsPath: AppResourcePath = 'vs/../../extensions';
272 > export const nodeModulesPath: AppResourcePath = 'vs/../../node_modules';
273 > export const nodeModulesAsarPath: AppResourcePath = 'vs/../../node_modules.asar';
274 > export const nodeModulesAsarUnpackedPath: AppResourcePath = 'vs/../../node_modules.asar.unpacked';
275 >
276 > export const VSCODE_AUTHORITY = 'vscode-app';
277 >
278 > class FileAccessImpl {
279 >
280 > private static readonly FALLBACK_AUTHORITY = VSCODE_AUTHORITY;
281 >
282 > /**
283 > * Returns a URI to use in contexts where the browser is responsible
284 > * for loading (e.g. fetch()) or when used within the DOM.
285 > *
286 > * **Note:** use `dom.ts#asCSSUrl` whenever the URL is to be used in CSS context.
287 > */
288 > asBrowserUri(resourcePath: AppResourcePath | ''): URI {
289 const uri = this.toUri(resourcePath);
290 return this.uriToBrowserUri(uri);
291 }
292 > network.ts
293 > /**
294 > * Returns a URI to use in contexts where the browser is responsible
295 > * for loading (e.g. fetch()) or when used within the DOM.
296 > *
297 > * **Note:** use `dom.ts#asCSSUrl` whenever the URL is to be used in CSS context.
298 > */
299 > uriToBrowserUri(uri: URI): URI {
300 // Handle remote URIs via `RemoteAuthorities`
301 if (uri.scheme === Schemas.vscodeRemote) {
328 return uri;
329 }
330 > network.ts
331 > /**
332 > * Returns the `file` URI to use in contexts where node.js
333 > * is responsible for loading.
334 > */
335 > asFileUri(resourcePath: AppResourcePath | ''): URI {
336 const uri = this.toUri(resourcePath);
337 return this.uriToFileUri(uri);
338 }
339 > network.ts
340 > /**
341 > * Returns the `file` URI to use in contexts where node.js
342 > * is responsible for loading.
343 > */
344 > uriToFileUri(uri: URI): URI {
345 // Only convert the URI if it is `vscode-file:` scheme
346 if (uri.scheme === Schemas.vscodeFileResource) {
358 return uri;
359 }
360 > network.ts
361 > private toUri(uriOrModule: URI | string): URI {
362 if (URI.isUri(uriOrModule)) {
363 return uriOrModule;
379 throw new Error('Cannot determine URI for module id!');
380 }
381 > } network.ts
382 >
383 > export const FileAccess = new FileAccessImpl();
384 >
385 > export const CacheControlheaders: Record<string, string> = Object.freeze({
386 > 'Cache-Control': 'no-cache, no-store'
387 > });
388 >
389 > export const DocumentPolicyheaders: Record<string, string> = Object.freeze({
390 > 'Document-Policy': 'include-js-call-stacks-in-crash-reports'
391 > });
392 >
393 > export namespace COI {
394 >
395 > const coiHeaders = new Map<'3' | '2' | '1' | string, Record<string, string>>([
396 > ['1', { 'Cross-Origin-Opener-Policy': 'same-origin' }],
397 > ['2', { 'Cross-Origin-Embedder-Policy': 'require-corp' }],
398 > ['3', { 'Cross-Origin-Opener-Policy': 'same-origin', 'Cross-Origin-Embedder-Policy': 'require-corp' }],
399 > ]);
400 >
401 > export const CoopAndCoep = Object.freeze(coiHeaders.get('3'));
402 >
403 > const coiSearchParamName = 'vscode-coi';
404 >
405 > /**
406 > * Extract desired headers from `vscode-coi` invocation
407 > */
408 > export function getHeadersFromQuery(url: string | URI | URL): Record<string, string> | undefined {
409 let params: URLSearchParams | undefined;
410 if (typeof url === 'string') {
421 return coiHeaders.get(value);
422 }
423 > network.ts
424 > /**
425 > * Add the `vscode-coi` query attribute based on wanting `COOP` and `COEP`. Will be a noop when `crossOriginIsolated`
426 > * isn't enabled the current context
427 > */
428 > export function addSearchParam(urlOrSearch: URLSearchParams | Record<string, string>, coop: boolean, coep: boolean): void {
429 if (!(globalThis as typeof globalThis & { crossOriginIsolated?: boolean }).crossOriginIsolated) {
430 // depends on the current context being COI
src/vs/workbench/services/lifecycle/common/lifecycle.ts 292 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- lifecycle.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../base/common/cancellation.js';
7 > import { Event } from '../../../../base/common/event.js';
8 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
9 >
10 > export const ILifecycleService = createDecorator<ILifecycleService>('lifecycleService');
11 >
12 > /**
13 > * An event that is send out when the window is about to close. Clients have a chance to veto
14 > * the closing by either calling veto with a boolean "true" directly or with a promise that
15 > * resolves to a boolean. Returning a promise is useful in cases of long running operations
16 > * on shutdown.
17 > *
18 > * Note: It is absolutely important to avoid long running promises if possible. Please try hard
19 > * to return a boolean directly. Returning a promise has quite an impact on the shutdown sequence!
20 > */
21 > export interface BeforeShutdownEvent {
22 >
23 > /**
24 > * The reason why the application will be shutting down.
25 > */
26 > readonly reason: ShutdownReason;
27 >
28 > /**
29 > * Allows to veto the shutdown. The veto can be a long running operation but it
30 > * will block the application from closing.
31 > *
32 > * @param id to identify the veto operation in case it takes very long or never
33 > * completes.
34 > */
35 > veto(value: boolean | Promise<boolean>, id: string): void;
36 > }
37 >
38 > export interface InternalBeforeShutdownEvent extends BeforeShutdownEvent {
39 >
40 > /**
41 > * Allows to set a veto operation to run after all other
42 > * vetos have been handled from the `BeforeShutdownEvent`
43 > *
44 > * This method is hidden from the API because it is intended
45 > * to be only used once internally.
46 > */
47 > finalVeto(vetoFn: () => boolean | Promise<boolean>, id: string): void;
48 > }
49 >
50 > /**
51 > * An event that signals an error happened during `onBeforeShutdown` veto handling.
52 > * In this case the shutdown operation will not proceed because this is an unexpected
53 > * condition that is treated like a veto.
54 > */
55 > export interface BeforeShutdownErrorEvent {
56 >
57 > /**
58 > * The reason why the application is shutting down.
59 > */
60 > readonly reason: ShutdownReason;
61 >
62 > /**
63 > * The error that happened during shutdown handling.
64 > */
65 > readonly error: Error;
66 > }
67 >
68 > export enum WillShutdownJoinerOrder {
69 >
70 > /**
71 > * Joiners to run before the `Last` joiners. This is the default order and best for
72 > * most cases. You can be sure that services are still functional at this point.
73 > */
74 > Default = 1,
75 >
76 > /**
77 > * The joiners to run last. This should ONLY be used in rare cases when you have no
78 > * dependencies to workbench services or state. The workbench may be in a state where
79 > * resources can no longer be accessed or changed.
80 > */
81 > Last
82 > }
83 >
84 > export interface IWillShutdownEventJoiner {
85 > readonly id: string;
86 > readonly label: string;
87 > readonly order?: WillShutdownJoinerOrder;
88 > }
89 >
90 > export interface IWillShutdownEventDefaultJoiner extends IWillShutdownEventJoiner {
91 > readonly order?: WillShutdownJoinerOrder.Default;
92 > }
93 >
94 > export interface IWillShutdownEventLastJoiner extends IWillShutdownEventJoiner {
95 > readonly order: WillShutdownJoinerOrder.Last;
96 > }
97 >
98 > /**
99 > * An event that is send out when the window closes. Clients have a chance to join the closing
100 > * by providing a promise from the join method. Returning a promise is useful in cases of long
101 > * running operations on shutdown.
102 > *
103 > * Note: It is absolutely important to avoid long running promises if possible. Please try hard
104 > * to return a boolean directly. Returning a promise has quite an impact on the shutdown sequence!
105 > */
106 > export interface WillShutdownEvent {
107 >
108 > /**
109 > * The reason why the application is shutting down.
110 > */
111 > readonly reason: ShutdownReason;
112 >
113 > /**
114 > * A token that will signal cancellation when the
115 > * shutdown was forced by the user.
116 > */
117 > readonly token: CancellationToken;
118 >
119 > /**
120 > * Allows to join the shutdown. The promise can be a long running operation but it
121 > * will block the application from closing.
122 > *
123 > * @param promise the promise to join the shutdown event.
124 > * @param joiner to identify the join operation in case it takes very long or never
125 > * completes.
126 > */
127 > join(promise: Promise<void>, joiner: IWillShutdownEventDefaultJoiner): void;
128 >
129 > /**
130 > * Allows to join the shutdown at the end. The promise can be a long running operation but it
131 > * will block the application from closing.
132 > *
133 > * @param promiseFn the promise to join the shutdown event.
134 > * @param joiner to identify the join operation in case it takes very long or never
135 > * completes.
136 > */
137 > join(promiseFn: (() => Promise<void>), joiner: IWillShutdownEventLastJoiner): void;
138 >
139 > /**
140 > * Allows to access the joiners that have not finished joining this event.
141 > */
142 > joiners(): IWillShutdownEventJoiner[];
143 >
144 > /**
145 > * Allows to enforce the shutdown, even when there are
146 > * pending `join` operations to complete.
147 > */
148 > force(): void;
149 > }
150 >
151 > export const enum ShutdownReason {
152 >
153 > /**
154 > * The window is closed.
155 > */
156 > CLOSE = 1,
157 >
158 > /**
159 > * The window closes because the application quits.
160 > */
161 > QUIT,
162 >
163 > /**
164 > * The window is reloaded.
165 > */
166 > RELOAD,
167 >
168 > /**
169 > * The window is loaded into a different workspace context.
170 > */
171 > LOAD
172 > }
173 >
174 > export const enum StartupKind {
175 > NewWindow = 1,
176 > ReloadedWindow = 3,
177 > ReopenedWindow = 4
178 > }
179 >
180 > export function StartupKindToString(startupKind: StartupKind): string {
181 switch (startupKind) {
182 case StartupKind.NewWindow: return 'NewWindow';
185 }
186 }
187 > lifecycle.ts
188 > export const enum LifecyclePhase {
189 >
190 > /**
191 > * The first phase signals that we are about to startup getting ready.
192 > *
193 > * Note: doing work in this phase blocks an editor from showing to
194 > * the user, so please rather consider to use `Restored` phase.
195 > */
196 > Starting = 1,
197 >
198 > /**
199 > * Services are ready and the window is about to restore its UI state.
200 > *
201 > * Note: doing work in this phase blocks an editor from showing to
202 > * the user, so please rather consider to use `Restored` phase.
203 > */
204 > Ready = 2,
205 >
206 > /**
207 > * Views, panels and editors have restored. Editors are given a bit of
208 > * time to restore their contents.
209 > */
210 > Restored = 3,
211 >
212 > /**
213 > * The last phase after views, panels and editors have restored and
214 > * some time has passed (2-5 seconds).
215 > */
216 > Eventually = 4
217 > }
218 >
219 > export function LifecyclePhaseToString(phase: LifecyclePhase): string {
220 switch (phase) {
221 case LifecyclePhase.Starting: return 'Starting';
225 }
226 }
227 > lifecycle.ts
228 > /**
229 > * A lifecycle service informs about lifecycle events of the
230 > * application, such as shutdown.
231 > */
232 > export interface ILifecycleService {
233 >
234 > readonly _serviceBrand: undefined;
235 >
236 > /**
237 > * Value indicates how this window got loaded.
238 > */
239 > readonly startupKind: StartupKind;
240 >
241 > /**
242 > * A flag indicating in what phase of the lifecycle we currently are.
243 > */
244 > phase: LifecyclePhase;
245 >
246 > /**
247 > * Fired before shutdown happens. Allows listeners to veto against the
248 > * shutdown to prevent it from happening.
249 > *
250 > * The event carries a shutdown reason that indicates how the shutdown was triggered.
251 > */
252 > readonly onBeforeShutdown: Event<BeforeShutdownEvent>;
253 >
254 > /**
255 > * Fired when the shutdown was prevented by a component giving veto.
256 > */
257 > readonly onShutdownVeto: Event<void>;
258 >
259 > /**
260 > * Fired when an error happened during `onBeforeShutdown` veto handling.
261 > * In this case the shutdown operation will not proceed because this is
262 > * an unexpected condition that is treated like a veto.
263 > *
264 > * The event carries a shutdown reason that indicates how the shutdown was triggered.
265 > */
266 > readonly onBeforeShutdownError: Event<BeforeShutdownErrorEvent>;
267 >
268 > /**
269 > * Fired when no client is preventing the shutdown from happening (from `onBeforeShutdown`).
270 > *
271 > * This event can be joined with a long running operation via `WillShutdownEvent#join()` to
272 > * handle long running shutdown operations.
273 > *
274 > * The event carries a shutdown reason that indicates how the shutdown was triggered.
275 > */
276 > readonly onWillShutdown: Event<WillShutdownEvent>;
277 >
278 > /**
279 > * A flag indicating that we are about to shutdown without further veto.
280 > */
281 > readonly willShutdown: boolean;
282 >
283 > /**
284 > * Fired when the shutdown is about to happen after long running shutdown operations
285 > * have finished (from `onWillShutdown`).
286 > *
287 > * This event should be used to dispose resources.
288 > */
289 > readonly onDidShutdown: Event<void>;
290 >
291 > /**
292 > * Returns a promise that resolves when a certain lifecycle phase
293 > * has started.
294 > */
295 > when(phase: LifecyclePhase): Promise<void>;
296 >
297 > /**
298 > * Triggers a shutdown of the workbench. Depending on native or web, this can have
299 > * different implementations and behaviour.
300 > *
301 > * **Note:** this should normally not be called. See related methods in `IHostService`
302 > * and `INativeHostService` to close a window or quit the application.
303 > */
304 > shutdown(): Promise<void>;
305 > }
src/vs/workbench/contrib/chat/common/model/chatSessionStore.ts 290 covered LOC · 66 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatSessionStore.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Sequencer } from '../../../../../base/common/async.js';
7 > import { VSBuffer } from '../../../../../base/common/buffer.js';
8 > import { toErrorMessage } from '../../../../../base/common/errorMessage.js';
9 > import { MarkdownString } from '../../../../../base/common/htmlContent.js';
10 > import { Disposable } from '../../../../../base/common/lifecycle.js';
11 > import { revive } from '../../../../../base/common/marshalling.js';
12 > import { isEqual, joinPath } from '../../../../../base/common/resources.js';
13 > import { URI } from '../../../../../base/common/uri.js';
14 > import { localize } from '../../../../../nls.js';
15 > import { IConfigurationService } from '../../../../../platform/configuration/common/configuration.js';
16 > import { IDialogService } from '../../../../../platform/dialogs/common/dialogs.js';
17 > import { IEnvironmentService } from '../../../../../platform/environment/common/environment.js';
18 > import { FileOperationResult, IFileService, toFileOperationResult } from '../../../../../platform/files/common/files.js';
19 > import { ILogService } from '../../../../../platform/log/common/log.js';
20 > import { IOpenerService } from '../../../../../platform/opener/common/opener.js';
21 > import { IStorageService, StorageScope, StorageTarget } from '../../../../../platform/storage/common/storage.js';
22 > import { ITelemetryService } from '../../../../../platform/telemetry/common/telemetry.js';
23 > import { IUserDataProfilesService } from '../../../../../platform/userDataProfile/common/userDataProfile.js';
24 > import { IAnyWorkspaceIdentifier, isEmptyWorkspaceIdentifier, IWorkspaceContextService } from '../../../../../platform/workspace/common/workspace.js';
25 > import { Dto } from '../../../../services/extensions/common/proxyIdentifier.js';
26 > import { ILifecycleService } from '../../../../services/lifecycle/common/lifecycle.js';
27 > import { IWorkspaceEditingService } from '../../../../services/workspaces/common/workspaceEditing.js';
28 > import { awaitStatsForSession } from '../chat.js';
29 > import { IChatSessionStats, IChatSessionTiming, ResponseModelState } from '../chatService/chatService.js';
30 > import { ChatAgentLocation, ChatPermissionLevel } from '../constants.js';
31 > import { ModifiedFileEntryState } from '../editing/chatEditingService.js';
32 > import { ChatModel, ISerializableChatData, ISerializableChatDataIn, ISerializableChatModelInputState, ISerializableChatsData, ISerializedChatDataReference, normalizeSerializableChatData } from './chatModel.js';
33 > import { ChatSessionOperationLog } from './chatSessionOperationLog.js';
34 > import { LocalChatSessionUri } from './chatUri.js';
35 > import { stringifyEntryWithFallback } from './objectMutationLog.js';
36 >
37 > const maxPersistedSessions = 400;
38 >
39 > const ChatIndexStorageKey = 'chat.ChatSessionStore.index';
40 > const ChatTransferIndexStorageKey = 'ChatSessionStore.transferIndex';
41 >
42 > export class ChatSessionStore extends Disposable {
43 > private storageRoot: URI;
44 > private readonly previousEmptyWindowStorageRoot: URI | undefined;
45 > private readonly transferredSessionStorageRoot: URI;
46 >
47 > private readonly storeQueue = new Sequencer();
48 >
49 > private storeTask: Promise<void> | undefined;
50 > private shuttingDown = false;
51 >
52 > constructor(
53 > @IFileService private readonly fileService: IFileService, chatSessionStore.ts
54 > @IEnvironmentService private readonly environmentService: IEnvironmentService,
55 > @ILogService private readonly logService: ILogService,
56 > @IWorkspaceContextService private readonly workspaceContextService: IWorkspaceContextService,
57 > @ITelemetryService private readonly telemetryService: ITelemetryService,
58 > @IStorageService private readonly storageService: IStorageService,
59 > @ILifecycleService private readonly lifecycleService: ILifecycleService,
60 > @IUserDataProfilesService private readonly userDataProfilesService: IUserDataProfilesService,
61 > @IConfigurationService private readonly configurationService: IConfigurationService,
62 > @IWorkspaceEditingService private readonly workspaceEditingService: IWorkspaceEditingService,
63 > @IDialogService private readonly dialogService: IDialogService,
64 > @IOpenerService private readonly openerService: IOpenerService,
65 > ) {
66 > super();
67 >
68 > const workspace = this.workspaceContextService.getWorkspace();
69 > const isEmptyWindow = !workspace.configuration && workspace.folders.length === 0;
70 > const workspaceId = this.workspaceContextService.getWorkspace().id;
71 > this.storageRoot = isEmptyWindow ?
72 joinPath(this.userDataProfilesService.defaultProfile.globalStorageHome, 'emptyWindowChatSessions') :
73 > joinPath(this.environmentService.workspaceStorageHome, workspaceId, 'chatSessions'); chatSessionStore.ts
75 > this.previousEmptyWindowStorageRoot = isEmptyWindow ?
76 joinPath(this.environmentService.workspaceStorageHome, 'no-workspace', 'chatSessions') :
77 > undefined; chatSessionStore.ts
79 > this.transferredSessionStorageRoot = joinPath(this.userDataProfilesService.defaultProfile.globalStorageHome, 'transferredChatSessions');
80 >
81 > // Listen to workspace transitions to migrate chat sessions
82 > this._register(this.workspaceEditingService.onDidEnterWorkspace(event => {
83 const transitionPromise = this.storeQueue.queue(() => this.handleWorkspaceTransition(event.oldWorkspace, event.newWorkspace));
84 event.join(transitionPromise);
86 >
87 > this._register(this.lifecycleService.onWillShutdown(e => {
88 this.shuttingDown = true;
89 if (!this.storeTask) {
95 label: localize('join.chatSessionStore', "Saving chat history")
96 });
98 > }
100 > private async handleWorkspaceTransition(oldWorkspace: IAnyWorkspaceIdentifier, newWorkspace: IAnyWorkspaceIdentifier): Promise<void> {
101 const wasEmptyWindow = isEmptyWorkspaceIdentifier(oldWorkspace);
102 const isNewWorkspaceEmpty = isEmptyWorkspaceIdentifier(newWorkspace);
128 await this.migrateSessionsToNewWorkspace(oldStorageRoot, wasEmptyWindow, isNewWorkspaceEmpty);
129 }
131 > private async migrateSessionsToNewWorkspace(oldStorageRoot: URI, wasEmptyWindow: boolean, isNewWorkspaceEmpty: boolean): Promise<void> {
132 try {
133 // Check if old storage location exists
182 }
183 }
185 > async storeSessions(sessions: ChatModel[]): Promise<void> {
186 if (this.shuttingDown) {
187 // Don't start this task if we missed the chance to block shutdown
204 }
205 }
207 > async storeSessionsMetadataOnly(sessions: ChatModel[]): Promise<void> {
208 if (this.shuttingDown) {
209 // Don't start this task if we missed the chance to block shutdown
225 }
226 }
228 > async storeTransferSession(transferData: IChatTransfer, session: ChatModel): Promise<void> {
229 const index = this.getTransferredSessionIndex();
230 const workspaceKey = transferData.toWorkspace.toString();
262 }
263 }
265 > private getTransferredSessionIndex(): IChatTransferIndex {
266 > try { chatSessionStore.ts
267 > const data: IChatTransferIndex = this.storageService.getObject(ChatTransferIndexStorageKey, StorageScope.PROFILE, {});
268 > return data;
269 > } catch (e) {
270 this.reportError('getTransferredSessionIndex', 'Error reading chat transfer index', e);
271 return {};
272 }
275 > private static readonly TRANSFER_EXPIRATION_MS = 60 * 1000 * 5;
276 >
277 > getTransferredSessionData(): URI | undefined {
278 > try { chatSessionStore.ts
279 > const index = this.getTransferredSessionIndex();
280 > const workspaceFolders = this.workspaceContextService.getWorkspace().folders;
281 > if (workspaceFolders.length !== 1) {
282 // Can only transfer sessions to single-folder workspaces
283 return undefined;
284 }
286 > const workspaceKey = workspaceFolders[0].uri.toString();
287 > const transferredSessionForWorkspace: IChatTransferDto = index[workspaceKey];
288 > if (!transferredSessionForWorkspace) {
289 > return undefined; chatSessionStore.ts
290 > }
291
292 // Check if the transfer has expired
298 }
299 return !!LocalChatSessionUri.parseLocalSessionId(revivedTransferData.sessionResource) && revivedTransferData.sessionResource;
300 > } catch (e) { chatSessionStore.ts
301 this.reportError('getTransferredSession', 'Error getting transferred chat session URI', e);
302 return undefined;
303 }
306 > async readTransferredSession(sessionResource: URI): Promise<ISerializedChatDataReference | undefined> {
307 try {
308 const storageLocation = this.getTransferredSessionStorageLocation(sessionResource);
323 }
324 }
326 > private async cleanupTransferredSession(sessionResource: URI): Promise<void> {
327 try {
328 // Remove from index
344 }
345 }
347 > private _didReportIssue = false;
348 >
349 > private async writeSession(session: ChatModel | ISerializableChatData): Promise<void> {
350 try {
351 const index = this.internalGetIndex();
402 }
403 }
405 > private async writeSessionMetadataOnly(session: ChatModel): Promise<void> {
406 // Only to be used for external sessions
407 if (LocalChatSessionUri.parseLocalSessionId(session.sessionResource)) {
419 }
420 }
422 > private async flushIndex(): Promise<void> {
423 const index = this.internalGetIndex();
424 try {
429 }
430 }
432 > private getIndexStorageScope(): StorageScope {
433 > const workspace = this.workspaceContextService.getWorkspace(); chatSessionStore.ts
434 > const isEmptyWindow = !workspace.configuration && workspace.folders.length === 0;
435 > return isEmptyWindow ? StorageScope.APPLICATION : StorageScope.WORKSPACE;
436 > }
438 > private async trimEntries(): Promise<void> {
439 const index = this.internalGetIndex();
440 const entries = Object.entries(index.entries)
452 }
453 }
455 > private async internalDeleteSession(sessionId: string): Promise<void> {
456 const index = this.internalGetIndex();
457 if (!index.entries[sessionId]) {
474 }
475 }
477 > hasSessions(): boolean {
478 return Object.keys(this.internalGetIndex().entries).length > 0;
479 }
481 > isSessionEmpty(sessionId: string): boolean {
482 const index = this.internalGetIndex();
483 return index.entries[sessionId]?.isEmpty ?? true;
484 }
486 > async deleteSession(sessionId: string): Promise<void> {
487 await this.storeQueue.queue(async () => {
488 await this.internalDeleteSession(sessionId);
490 });
491 }
493 > async clearAllSessions(): Promise<void> {
494 await this.storeQueue.queue(async () => {
495 const index = this.internalGetIndex();
500 });
501 }
503 > public async setSessionTitle(sessionId: string, title: string): Promise<void> {
504 await this.storeQueue.queue(async () => {
505 const index = this.internalGetIndex();
509 });
510 }
512 > private reportError(reasonForTelemetry: string, message: string, error?: Error): void {
513 const fileOperationReason = error && toFileOperationResult(error);
514
537 });
538 }
540 > private indexCache: IChatSessionIndexData | undefined;
541 > private internalGetIndex(): IChatSessionIndexData {
542 > if (this.indexCache) { chatSessionStore.ts
543 return this.indexCache;
544 }
546 > const data = this.storageService.get(ChatIndexStorageKey, this.getIndexStorageScope(), undefined);
547 > if (!data) {
548 > this.indexCache = { version: 1, entries: {} };
549 > return this.indexCache;
550 > }
551
552 try {
559 this.indexCache = { version: 1, entries: {} };
560 }
562 > } catch (e) {
563 // Only if JSON.parse fails
564 this.reportError('invalidIndexJSON', `Index corrupt: ${data}`, e);
579
580 return this.indexCache;
583 > async getIndex(): Promise<IChatSessionIndex> {
584 return this.storeQueue.queue(async () => {
585 return this.internalGetIndex().entries;
586 });
587 }
589 > getMetadataForSessionSync(sessionResource: URI): IChatSessionEntryMetadata | undefined {
590 > const index = this.internalGetIndex(); chatSessionStore.ts
591 > return index.entries[this.getIndexKey(sessionResource)];
592 > }
594 > private getIndexKey(sessionResource: URI): string {
595 > const sessionId = LocalChatSessionUri.parseLocalSessionId(sessionResource); chatSessionStore.ts
596 > return sessionId ?? sessionResource.toString();
597 > }
599 > logIndex(): void {
600 const data = this.storageService.get(ChatIndexStorageKey, this.getIndexStorageScope(), undefined);
601 this.logService.info('ChatSessionStore index: ', data);
602 }
604 > async migrateDataIfNeeded(getInitialData: () => ISerializableChatsData | undefined): Promise<void> {
605 > await this.storeQueue.queue(async () => { chatSessionStore.ts
606 > const data = this.storageService.get(ChatIndexStorageKey, this.getIndexStorageScope(), undefined);
607 > const needsMigrationFromStorageService = !data;
608 > if (needsMigrationFromStorageService) {
609 > const initialData = getInitialData(); chatSessionStore.ts
610 > if (initialData) {
611 await this.migrate(initialData);
612 }
615 > }
617 > private async migrate(initialData: ISerializableChatsData): Promise<void> {
618 const numSessions = Object.keys(initialData).length;
619 this.logService.info(`ChatSessionStore: Migrating ${numSessions} chat sessions from storage service to file system`);
625 await this.flushIndex();
626 }
628 > public async readSession(sessionId: string): Promise<ISerializedChatDataReference | undefined> {
629 return await this.storeQueue.queue(async () => {
630 const storageLocation = this.getStorageLocation(sessionId);
632 });
633 }
635 > private async readSessionFromLocation(flatStorageLocation: URI, logStorageLocation: URI | undefined, sessionId: string): Promise<ISerializedChatDataReference | undefined> {
636 let fromLocation = flatStorageLocation;
637 let rawData: VSBuffer | undefined;
693 }
694 }
696 > private async readSessionFromPreviousLocation(sessionId: string): Promise<VSBuffer | undefined> {
697 let rawData: VSBuffer | undefined;
698
710 return rawData;
711 }
713 > private getStorageLocation(chatSessionId: string): {
714 /** <1.109 flat JSON file */
715 flat: URI;
723 };
724 }
726 > private getTransferredSessionStorageLocation(sessionResource: URI): URI {
727 const sessionId = LocalChatSessionUri.parseLocalSessionId(sessionResource);
728 return joinPath(this.transferredSessionStorageRoot, `${sessionId}.json`);
729 }
731 > /**
732 > * Synchronously update the in-memory index entries for the given sessions
733 > * and flush the index to storage. This ensures the index is persisted
734 > * even when called from a synchronous `onWillSaveState` handler where
735 > * async file-write work would complete after the storage service has
736 > * already flushed.
737 > */
738 > updateAndFlushIndexSync(localSessions: ChatModel[], externalSessions: ChatModel[]): void {
739 const index = this.internalGetIndex();
740 for (const session of localSessions) {
751 }
752 }
754 > public getChatStorageFolder(): URI {
755 return this.storageRoot;
756 }
758 >
759 > export interface IChatSessionEntryMetadata {
760 > sessionId: string;
761 > title: string;
762 > lastMessageDate: number;
763 > timing: IChatSessionTiming;
764 > initialLocation?: ChatAgentLocation;
765 > hasPendingEdits?: boolean;
766 > stats?: IChatSessionStats;
767 > lastResponseState: ResponseModelState;
768 >
769 > /**
770 > * The working directory URI string associated with this session.
771 > * Persisted so it survives window reload in the agents/sessions window.
772 > */
773 > workingDirectory?: string;
774 >
775 > /**
776 > * This only exists because the migrated data from the storage service had empty sessions persisted, and it's impossible to know which ones are
777 > * currently in use. Now, `clearSession` deletes empty sessions, so old ones shouldn't take up space in the store anymore, but we still need to
778 > * filter the old ones out of history.
779 > */
780 > isEmpty?: boolean;
781 >
782 > /**
783 > * Whether this session was loaded from an external provider (eg background/cloud sessions).
784 > */
785 > isExternal?: boolean;
786 >
787 > /**
788 > * The permission level for tool auto-approval, if not default.
789 > */
790 > permissionLevel?: ChatPermissionLevel;
791 >
792 > /**
793 > * Serialized draft input state (text, attachments, mode, selected model, ...) for
794 > * external sessions, so that unsent input is preserved when switching away and
795 > * back. Local sessions instead persist their full state via storeSessions.
796 > */
797 > inputState?: ISerializableChatModelInputState;
798 > }
799 >
800 function isChatSessionEntryMetadata(obj: unknown): obj is IChatSessionEntryMetadata {
801 return (
807 );
808 }
810 > export type IChatSessionIndex = Record<string, IChatSessionEntryMetadata>;
811 >
812 > interface IChatSessionIndexData {
813 > version: 1;
814 > entries: IChatSessionIndex;
815 > }
816 >
817 > // TODO if we update the index version:
818 > // Don't throw away index when moving backwards in VS Code version. Try to recover it. But this scenario is hard.
819 function isChatSessionIndex(data: unknown): data is IChatSessionIndexData {
820 if (typeof data !== 'object' || data === null) {
839 return true;
840 }
842 > /**
843 > * Builds session metadata synchronously from a live ChatModel.
844 > * Used both by {@link updateAndFlushIndexSync} (where async work is not
845 > * possible) and by {@link getSessionMetadata} (which layers on async stats).
846 > */
847 function getSessionMetadataSync(session: ChatModel): IChatSessionEntryMetadata {
848 const title = session.customTitle || session.title;
872 };
873 }
875 async function getSessionMetadata(session: ChatModel | ISerializableChatData): Promise<IChatSessionEntryMetadata> {
876 if (session instanceof ChatModel) {
899 };
900 }
902 > export interface IChatTransfer {
903 > toWorkspace: URI;
904 > sessionResource: URI;
905 > timestampInMilliseconds: number;
906 > }
907 >
908 > export interface IChatTransfer2 extends IChatTransfer {
909 > chat: ISerializableChatData;
910 > }
911 >
912 > type IChatTransferDto = Dto<IChatTransfer>;
913 >
914 > /**
915 > * Map of destination workspace URI to chat transfer data
916 > */
917 > type IChatTransferIndex = Record<string, IChatTransferDto>;
src/vs/workbench/contrib/chat/common/model/objectMutationLog.ts 287 covered LOC · 34 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- objectMutationLog.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { assertNever } from '../../../../../base/common/assert.js';
7 > import { VSBuffer } from '../../../../../base/common/buffer.js';
8 > import { isUndefinedOrNull } from '../../../../../base/common/types.js';
9 >
10 > /**
11 > * Updates an error's message and stack trace with a prefix. In V8 the stack
12 > * string starts with "ErrorName: message\n at …", so we rebuild the header
13 > * after mutating the message.
14 > */
15 function prefixError(e: Error, prefix: string): void {
16 e.message = prefix + e.message;
22 }
23 }
25 > /**
26 > * Prepends a path segment to an error as it unwinds through nested extract
27 > * calls. Each level adds its segment so the final message reads e.g.
28 > * `.responses[2].content: Cannot read property 'x' of undefined`.
29 > */
30 function rethrowWithPathSegment(e: unknown, segment: string | number): never {
31 if (e instanceof Error) {
36 throw e;
37 }
39 > /** IMPORTANT: `Key` comes first. Then we should sort in order of least->most expensive to diff */
40 > const enum TransformKind {
41 > Key,
42 > Primitive,
43 > Array,
44 > Object,
45 > }
46 >
47 > /** Schema entries sorted with key properties first */
48 > export type SchemaEntries = [string, Transform<unknown, unknown>][];
49 >
50 > interface TransformBase<TFrom, TTo> {
51 > readonly kind: TransformKind;
52 > /** Extracts the serializable value from the source object */
53 > extract(from: TFrom): TTo;
54 > }
55 >
56 > /** Transform for primitive values (keys and values) that can be compared for equality */
57 > export interface TransformValue<TFrom, TTo> extends TransformBase<TFrom, TTo> {
58 > readonly kind: TransformKind.Key | TransformKind.Primitive;
59 > /** Compares two serialized values for equality */
60 > equals(a: TTo, b: TTo): boolean;
61 > }
62 >
63 > /** Transform for arrays with an item schema */
64 > export interface TransformArray<TFrom, TTo> extends TransformBase<TFrom, TTo> {
65 > readonly kind: TransformKind.Array;
66 > /** The schema for array items */
67 > readonly itemSchema: TransformObject<unknown, unknown> | TransformValue<unknown, unknown>;
68 > }
69 >
70 > /** Transform for objects with child properties */
71 > export interface TransformObject<TFrom, TTo> extends TransformBase<TFrom, TTo> {
72 > readonly kind: TransformKind.Object;
73 > /** Schema entries sorted with Key properties first */
74 > readonly children: SchemaEntries;
75 > /** Checks if the object is sealed (won't change). */
76 > sealed?(obj: TTo, wasSerialized: boolean): boolean;
77 > }
78 >
79 > export type Transform<TFrom, TTo> =
80 > | TransformValue<TFrom, TTo>
81 > | TransformArray<TFrom, TTo>
82 > | TransformObject<TFrom, TTo>;
83 >
84 > export type Schema<TFrom, TTo> = {
85 > [K in keyof Required<TTo>]: Transform<TFrom, TTo[K]>
86 > };
87 >
88 > /**
89 > * A primitive that will be tracked and compared first. If this is changed, the entire
90 > * object is thrown out and re-stored.
91 > */
92 > export function key<T, R = T>(comparator?: (a: R, b: R) => boolean): TransformValue<T, R> {
93 > return { objectMutationLog.ts
94 > kind: TransformKind.Key,
95 > extract: (from: T) => from as unknown as R,
96 > equals: comparator ?? ((a, b) => a === b),
97 > };
98 > }
100 > /** A value that will be tracked and replaced if the comparator is not equal. */
101 > export function value<T, R extends string | number | boolean | undefined>(): TransformValue<T, R>;
102 > export function value<T, R>(comparator: (a: R, b: R) => boolean): TransformValue<T, R>;
103 > export function value<T, R>(comparator?: (a: R, b: R) => boolean): TransformValue<T, R> {
104 > return { objectMutationLog.ts
105 > kind: TransformKind.Primitive,
106 > extract: (from: T) => {
107 let value = from as unknown as R;
108 // We map the object to JSON for two reasons (a) reduce issues with references to
116 return value;
117 },
118 > equals: comparator ?? ((a, b) => a === b), objectMutationLog.ts
119 > };
120 > }
122 > /** An array that will use the schema to compare items positionally. */
123 > export function array<T, R>(schema: TransformObject<T, R> | TransformValue<T, R>): TransformArray<readonly T[], R[]> {
124 > return { objectMutationLog.ts
125 > kind: TransformKind.Array,
126 > itemSchema: schema,
127 > extract: from => from?.map((item, i) => {
128 try {
129 return schema.extract(item);
132 }
133 }),
135 > }
137 > export interface ObjectOptions<R> {
138 > /**
139 > * Returns true if the object is sealed and will never change again.
140 > * When comparing two sealed objects, only key fields are compared
141 > * (to detect replacement), but other fields are not diffed.
142 > */
143 > sealed?: (obj: R, wasSerialized: boolean) => boolean;
144 > }
145 >
146 > /** An object schema. */
147 > export function object<T, R extends object>(schema: Schema<T, R>, options?: ObjectOptions<R>): TransformObject<T, R> {
148 > // Sort entries with key properties first for fast key checking objectMutationLog.ts
149 > const entries = (Object.entries(schema) as [string, Transform<T, R[keyof R]>][]).sort(([, a], [, b]) => a.kind - b.kind);
150 > return {
151 > kind: TransformKind.Object,
152 > children: entries as SchemaEntries,
153 > sealed: options?.sealed,
154 > extract: (from: T) => {
155 if (isUndefinedOrNull(from)) {
156 return from as unknown as R;
167 return result as R;
168 },
170 > }
172 > /**
173 > * Defines a getter on the object to extract a value, compared with the given schema.
174 > * It should return the value that will get serialized in the resulting log file.
175 > */
176 > export function t<T, O, R>(getter: (obj: T) => O, schema: Transform<O, R>): Transform<T, R> {
177 > return { objectMutationLog.ts
178 > ...schema,
179 > extract: (from: T) => schema.extract(getter(from)),
180 > };
181 > }
183 > /** Shortcut for t(fn, value()) */
184 > export function v<T, R extends string | number | boolean | undefined>(getter: (obj: T) => R): TransformValue<T, R>;
185 > export function v<T, R>(getter: (obj: T) => R, comparator: (a: R, b: R) => boolean): TransformValue<T, R>;
186 > export function v<T, R>(getter: (obj: T) => R, comparator?: (a: R, b: R) => boolean): TransformValue<T, R> {
187 > const inner = value(comparator!); objectMutationLog.ts
188 > return {
189 > ...inner,
190 > extract: (from: T) => inner.extract(getter(from)),
191 > };
192 > }
194 >
195 > const enum EntryKind {
196 > /** Initial complete object state, valid only as the first entry */
197 > Initial = 0,
198 > /** Property update */
199 > Set = 1,
200 > /** Array push/splice. */
201 > Push = 2,
202 > /** Delete a property */
203 > Delete = 3,
204 > }
205 >
206 > type ObjectPath = (string | number)[];
207 >
208 > type Entry =
209 > | { kind: EntryKind.Initial; v: unknown }
210 > /** Update a property of an object, replacing it entirely */
211 > | { kind: EntryKind.Set; k: ObjectPath; v: unknown }
212 > /** Delete a property of an object */
213 > | { kind: EntryKind.Delete; k: ObjectPath }
214 > /** Pushes 0 or more new entries to an array. If `i` is set, everything after that index is removed */
215 > | { kind: EntryKind.Push; k: ObjectPath; v?: unknown[]; i?: number };
216 >
217 > const LF = VSBuffer.fromString('\n');
218 >
219 > /**
220 > * Per-string cap (in UTF-16 code units, matching `string.length`) applied when
221 > * {@link stringifyEntryWithFallback} retries after `JSON.stringify` throws
222 > * `RangeError: Invalid string length` (V8's max string length is ~512 MiB on
223 > * 64-bit). Any single string longer than this is replaced with a marker on
224 > * retry. Generous so it triggers only on outliers.
225 > */
226 > export const PERSIST_ENTRY_MAX_STRING_CHARS = 1 * 1024 * 1024;
227 >
228 > /**
229 > * Total-size budget (sum of `string.length` for tracked strings, in UTF-16
230 > * code units) for the retry of {@link stringifyEntryWithFallback}. Once the
231 > * cumulative tracked size during serialization exceeds this, remaining values
232 > * are replaced with a marker.
233 > *
234 > * This is an approximation: JSON escaping, property keys, and non-string
235 > * payload are not counted, so the actual output may be moderately larger.
236 > * The cap is sized well under V8's max string length to leave ample headroom
237 > * for that overhead.
238 > */
239 > export const PERSIST_ENTRY_MAX_TOTAL_CHARS = 100 * 1024 * 1024;
240 >
241 > const TRUNCATION_MARKER_PREFIX = '[VS Code: value truncated for persistence';
242 > const TRUNCATION_MARKER_TOTAL = `${TRUNCATION_MARKER_PREFIX}; entry exceeded size budget]`;
243 >
244 > /**
245 > * Wraps `JSON.stringify(entry)` with a safety net for the V8 max-string-length
246 > * limit. The common path is a single `JSON.stringify` with zero overhead. If
247 > * stringification throws `RangeError` (the resulting JSON would exceed V8's
248 > * ~512 MiB max string length — see microsoft/vscode#308843), retry with a
249 > * replacer that truncates oversized strings. Extensions sometimes put very
250 > * large content (browser dumps, command output, …) into chat result metadata;
251 > * losing the tail of one such value is dramatically better than losing the
252 > * entire chat session.
253 > */
254 > export function stringifyEntryWithFallback(entry: unknown): string {
255 try {
256 return JSON.stringify(entry);
262 }
263 }
265 > /**
266 > * Deep-clones `value` through JSON with the same V8 max-string-length safety net
267 > * as {@link stringifyEntryWithFallback}. Exported for testing only.
268 > */
269 > export function deepCloneWithFallback<T>(value: T): T {
270 return JSON.parse(stringifyEntryWithFallback(value)) as T;
271 }
273 > /**
274 > * Exported for testing only. Builds the stateful `JSON.stringify` replacer
275 > * used by {@link stringifyEntryWithFallback} on its retry path.
276 > *
277 > * Sizes are tracked in UTF-16 code units (`string.length`); JSON escaping,
278 > * property keys, and non-string payload are not counted.
279 > */
280 > export function makeTruncatingReplacer(maxStringChars: number, maxTotalChars: number): (key: string, value: unknown) => unknown {
281 let total = 0;
282 return (_key, val) => {
297 };
298 }
300 > /**
301 > * An implementation of an append-based mutation logger. Given a `Transform`
302 > * definition of an object, it can recreate it from a file on disk. It is
303 > * then stateful, and given a `write` call it can update the log in a minimal
304 > * way.
305 > */
306 > export class ObjectMutationLog<TFrom, TTo> {
307 > private _previous: TTo | undefined;
308 > private _entryCount = 0;
309 > private _hasPendingWrite = false;
310 > private _pendingPrevious: TTo | undefined;
311 > private _pendingEntryCount = 0;
312 >
313 > constructor(
314 private readonly _transform: Transform<TFrom, TTo>,
315 private readonly _compactAfterEntries = 512,
316 ) { }
318 > /**
319 > * Creates an initial log file from the given object.
320 > */
321 > createInitial(current: TFrom): VSBuffer {
322 return this.createInitialFromSerialized(this._transform.extract(current));
323 }
325 >
326 > /**
327 > * Creates an initial log file from the serialized object.
328 > *
329 > * Unlike {@link write}, this commits state immediately without requiring
330 > * {@link confirmWrite}. This is safe because the returned buffer contains
331 > * a self-contained `Initial` entry — if it fails to persist, no
332 > * incremental entries can be appended to a non-existent file.
333 > */
334 > createInitialFromSerialized(value: TTo): VSBuffer {
335 this._previous = value;
336 this._entryCount = 1;
339 return VSBuffer.fromString(stringifyEntryWithFallback(entry) + '\n');
340 }
342 > /**
343 > * Reads and reconstructs the state from a log file.
344 > */
345 > read(content: VSBuffer): TTo {
346 let state: unknown;
347 let lineCount = 0;
399 return state as TTo;
400 }
402 > /**
403 > * Writes updates to the log. Returns the operation type and data to write.
404 > * The caller **must** invoke {@link confirmWrite} after the data is
405 > * successfully persisted to commit the internal state. Without confirmation,
406 > * the next write is computed against the last confirmed state, and will only
407 > * produce a full initial entry when no confirmed state exists, preventing
408 > * corrupted log files when a write fails.
409 > */
410 > write(current: TFrom): { op: 'append' | 'replace'; data: VSBuffer } {
411 const currentValue = this._transform.extract(current);
412
450 return { op: 'append', data: VSBuffer.fromString(data) };
451 }
453 > /**
454 > * Commits the internal state after a successful write to disk.
455 > */
456 > confirmWrite(): void {
457 if (this._hasPendingWrite) {
458 this._previous = this._pendingPrevious;
461 }
462 }
464 > private _clearPending(): void {
465 this._hasPendingWrite = false;
466 this._pendingPrevious = undefined;
467 this._pendingEntryCount = 0;
468 }
470 > private _applySet(state: unknown, path: ObjectPath, value: unknown): void {
471 if (path.length === 0) {
472 return; // Root replacement handled by caller
480 current[path[path.length - 1]] = value;
481 }
483 > private _applyPush(state: unknown, path: ObjectPath, values: unknown[] | undefined, startIndex: number | undefined): void {
484 let current = state as Record<string | number, unknown>;
485 for (let i = 0; i < path.length - 1; i++) {
500 current[arrayKey] = arr;
501 }
503 > private _diff<T, R>(
504 transform: Transform<T, R>,
505 path: ObjectPath,
531 }
532 }
534 > private _diffObject(
535 children: SchemaEntries,
536 path: ObjectPath,
571 }
572 }
574 > private _diffArray<T, R>(
575 transform: TransformArray<T, R>,
576 path: ObjectPath,
644 }
645 }
647 > private _hasKeyMismatch(children: SchemaEntries, prev: unknown, curr: unknown): boolean {
648 const prevObj = prev as Record<string, unknown> | undefined;
649 const currObj = curr as Record<string, unknown>;
src/vs/platform/mcp/common/mcpManagement.ts 281 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpManagement.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../base/common/cancellation.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
9 > import { IIterativePager } from '../../../base/common/paging.js';
10 > import { URI } from '../../../base/common/uri.js';
11 > import { SortBy, SortOrder } from '../../extensionManagement/common/extensionManagement.js';
12 > import { createDecorator } from '../../instantiation/common/instantiation.js';
13 > import { IMcpServerIdentity } from './allowedMcpServers.js';
14 > import { IMcpSandboxConfiguration, IMcpServerConfiguration, IMcpServerVariable } from './mcpPlatformTypes.js';
15 >
16 > export type InstallSource = 'gallery' | 'local';
17 >
18 > export interface ILocalMcpServer {
19 > readonly name: string;
20 > readonly config: IMcpServerConfiguration;
21 > readonly rootSandbox?: IMcpSandboxConfiguration;
22 > readonly version?: string;
23 > readonly mcpResource: URI;
24 > readonly location?: URI;
25 > readonly displayName?: string;
26 > readonly description?: string;
27 > readonly galleryUrl?: string;
28 > readonly galleryId?: string;
29 > readonly repositoryUrl?: string;
30 > readonly readmeUrl?: URI;
31 > readonly publisher?: string;
32 > readonly publisherDisplayName?: string;
33 > readonly icon?: {
34 > readonly dark: string;
35 > readonly light: string;
36 > };
37 > readonly codicon?: string;
38 > readonly manifest?: IGalleryMcpServerConfiguration;
39 > readonly source: InstallSource;
40 > }
41 >
42 > export interface IMcpServerInput {
43 > readonly description?: string;
44 > readonly isRequired?: boolean;
45 > readonly format?: 'string' | 'number' | 'boolean' | 'filepath';
46 > readonly value?: string;
47 > readonly isSecret?: boolean;
48 > readonly default?: string;
49 > readonly choices?: readonly string[];
50 > }
51 >
52 > export interface IMcpServerVariableInput extends IMcpServerInput {
53 > readonly variables?: Record<string, IMcpServerInput>;
54 > }
55 >
56 > export interface IMcpServerPositionalArgument extends IMcpServerVariableInput {
57 > readonly type: 'positional';
58 > readonly valueHint?: string;
59 > readonly isRepeated?: boolean;
60 > }
61 >
62 > export interface IMcpServerNamedArgument extends IMcpServerVariableInput {
63 > readonly type: 'named';
64 > readonly name: string;
65 > readonly isRepeated?: boolean;
66 > }
67 >
68 > export interface IMcpServerKeyValueInput extends IMcpServerVariableInput {
69 > readonly name: string;
70 > readonly value?: string;
71 > }
72 >
73 > export type IMcpServerArgument = IMcpServerPositionalArgument | IMcpServerNamedArgument;
74 >
75 > export const enum RegistryType {
76 > NODE = 'npm',
77 > PYTHON = 'pypi',
78 > DOCKER = 'oci',
79 > NUGET = 'nuget',
80 > MCPB = 'mcpb',
81 > REMOTE = 'remote'
82 > }
83 >
84 > export const enum TransportType {
85 > STDIO = 'stdio',
86 > STREAMABLE_HTTP = 'streamable-http',
87 > SSE = 'sse'
88 > }
89 >
90 > export interface StdioTransport {
91 > readonly type: TransportType.STDIO;
92 > }
93 >
94 > export interface StreamableHttpTransport {
95 > readonly type: TransportType.STREAMABLE_HTTP;
96 > readonly url: string;
97 > readonly headers?: ReadonlyArray<IMcpServerKeyValueInput>;
98 > }
99 >
100 > export interface SseTransport {
101 > readonly type: TransportType.SSE;
102 > readonly url: string;
103 > readonly headers?: ReadonlyArray<IMcpServerKeyValueInput>;
104 > }
105 >
106 > export type Transport = StdioTransport | StreamableHttpTransport | SseTransport;
107 >
108 > export interface IMcpServerPackage {
109 > readonly registryType: RegistryType;
110 > readonly identifier: string;
111 > readonly transport: Transport;
112 > readonly version?: string;
113 > readonly registryBaseUrl?: string;
114 > readonly fileSha256?: string;
115 > readonly packageArguments?: readonly IMcpServerArgument[];
116 > readonly runtimeHint?: string;
117 > readonly runtimeArguments?: readonly IMcpServerArgument[];
118 > readonly environmentVariables?: ReadonlyArray<IMcpServerKeyValueInput>;
119 > }
120 >
121 > export interface IGalleryMcpServerConfiguration {
122 > readonly packages?: readonly IMcpServerPackage[];
123 > readonly remotes?: ReadonlyArray<SseTransport | StreamableHttpTransport>;
124 > }
125 >
126 > export const enum GalleryMcpServerStatus {
127 > Active = 'active',
128 > Deprecated = 'deprecated'
129 > }
130 >
131 > export interface IGalleryMcpServer {
132 > readonly name: string;
133 > readonly displayName: string;
134 > readonly description: string;
135 > readonly version: string;
136 > readonly isLatest: boolean;
137 > readonly status: GalleryMcpServerStatus;
138 > readonly id?: string;
139 > readonly galleryUrl?: string;
140 > readonly webUrl?: string;
141 > readonly codicon?: string;
142 > readonly icon?: {
143 > readonly dark: string;
144 > readonly light: string;
145 > };
146 > readonly lastUpdated?: number;
147 > readonly publishDate?: number;
148 > readonly repositoryUrl?: string;
149 > readonly configuration: IGalleryMcpServerConfiguration;
150 > readonly readmeUrl?: string;
151 > readonly readme?: string;
152 > readonly publisher: string;
153 > readonly publisherDisplayName?: string;
154 > readonly publisherUrl?: string;
155 > readonly publisherDomain?: { link: string; verified: boolean };
156 > readonly ratingCount?: number;
157 > readonly topics?: readonly string[];
158 > readonly license?: string;
159 > readonly starsCount?: number;
160 > }
161 >
162 > export interface IQueryOptions {
163 > text?: string;
164 > sortBy?: SortBy;
165 > sortOrder?: SortOrder;
166 > }
167 >
168 > export const IMcpGalleryService = createDecorator<IMcpGalleryService>('IMcpGalleryService');
169 > export interface IMcpGalleryService {
170 > readonly _serviceBrand: undefined;
171 > isEnabled(): boolean;
172 > query(options?: IQueryOptions, token?: CancellationToken): Promise<IIterativePager<IGalleryMcpServer>>;
173 > getMcpServersFromGallery(infos: { name: string; id?: string }[]): Promise<IGalleryMcpServer[]>;
174 > getMcpServer(url: string): Promise<IGalleryMcpServer | undefined>;
175 > getReadme(extension: IGalleryMcpServer, token: CancellationToken): Promise<string>;
176 > }
177 >
178 > export interface InstallMcpServerEvent {
179 > readonly name: string;
180 > readonly mcpResource: URI;
181 > readonly source?: IGalleryMcpServer;
182 > }
183 >
184 > export interface InstallMcpServerResult {
185 > readonly name: string;
186 > readonly mcpResource: URI;
187 > readonly source?: IGalleryMcpServer;
188 > readonly local?: ILocalMcpServer;
189 > readonly error?: Error;
190 > }
191 >
192 > export interface UninstallMcpServerEvent {
193 > readonly name: string;
194 > readonly mcpResource: URI;
195 > }
196 >
197 > export interface DidUninstallMcpServerEvent {
198 > readonly name: string;
199 > readonly mcpResource: URI;
200 > readonly error?: string;
201 > }
202 >
203 > export type InstallOptions = {
204 > packageType?: RegistryType;
205 > mcpResource?: URI;
206 > };
207 >
208 > export type UninstallOptions = {
209 > mcpResource?: URI;
210 > };
211 >
212 > export interface IInstallableMcpServer {
213 > readonly name: string;
214 > readonly config: IMcpServerConfiguration;
215 > readonly inputs?: IMcpServerVariable[];
216 > }
217 >
218 > export type McpServerConfiguration = Omit<IInstallableMcpServer, 'name'>;
219 > export interface McpServerConfigurationParseResult {
220 > readonly mcpServerConfiguration: McpServerConfiguration;
221 > readonly notices: string[];
222 > }
223 >
224 > export const IMcpManagementService = createDecorator<IMcpManagementService>('IMcpManagementService');
225 > export interface IMcpManagementService {
226 > readonly _serviceBrand: undefined;
227 > readonly onInstallMcpServer: Event<InstallMcpServerEvent>;
228 > readonly onDidInstallMcpServers: Event<readonly InstallMcpServerResult[]>;
229 > readonly onDidUpdateMcpServers: Event<readonly InstallMcpServerResult[]>;
230 > readonly onUninstallMcpServer: Event<UninstallMcpServerEvent>;
231 > readonly onDidUninstallMcpServer: Event<DidUninstallMcpServerEvent>;
232 > getInstalled(mcpResource?: URI): Promise<ILocalMcpServer[]>;
233 > canInstall(server: IGalleryMcpServer | IInstallableMcpServer): true | IMarkdownString;
234 > install(server: IInstallableMcpServer, options?: InstallOptions): Promise<ILocalMcpServer>;
235 > installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer>;
236 > updateMetadata(local: ILocalMcpServer, server: IGalleryMcpServer, profileLocation?: URI): Promise<ILocalMcpServer>;
237 > uninstall(server: ILocalMcpServer, options?: UninstallOptions): Promise<void>;
238 >
239 > getMcpServerConfigurationFromManifest(manifest: IGalleryMcpServerConfiguration, packageType: RegistryType): McpServerConfigurationParseResult;
240 > }
241 >
242 > export const IAllowedMcpServersService = createDecorator<IAllowedMcpServersService>('IAllowedMcpServersService');
243 > export interface IAllowedMcpServersService {
244 > readonly _serviceBrand: undefined;
245 >
246 > readonly onDidChangeAllowedMcpServers: Event<void>;
247 > isAllowed(mcpServer: IGalleryMcpServer | ILocalMcpServer | IInstallableMcpServer): true | IMarkdownString;
248 >
249 > /**
250 > * Checks whether an MCP server identified by name / remote URL / local command is permitted by
251 > * the `chat.mcp.allowedServers` allowlist (in addition to the `chat.mcp.access` gate). Used by
252 > * the runtime enforcement path, which does not have a gallery/local/installable representation.
253 > */
254 > isServerAllowed(identity: IMcpServerIdentity): true | IMarkdownString;
255 > }
256 >
257 > export const mcpAccessConfig = 'chat.mcp.access';
258 > export const mcpAllowedServersConfig = 'chat.mcp.allowedServers';
259 > export const mcpDeniedServersConfig = 'chat.mcp.deniedServers';
260 > export const mcpGalleryServiceUrlConfig = 'chat.mcp.gallery.serviceUrl';
261 > export const mcpGalleryServiceEnablementConfig = 'chat.mcp.gallery.enabled';
262 > export const mcpAutoStartConfig = 'chat.mcp.autostart';
263 > export const mcpAppsEnabledConfig = 'chat.mcp.apps.enabled';
264 >
265 > export interface IMcpGalleryConfig {
266 > readonly serviceUrl?: string;
267 > readonly enabled?: boolean;
268 > readonly version?: string;
269 > }
270 >
271 > export const enum McpAutoStartValue {
272 > Never = 'never',
273 > OnlyNew = 'onlyNew',
274 > NewAndOutdated = 'newAndOutdated',
275 > }
276 >
277 > export const enum McpAccessValue {
278 > None = 'none',
279 > Registry = 'registry',
280 > All = 'all',
281 > }
src/vs/platform/userDataProfile/common/userDataProfile.ts 281 covered LOC · 29 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- userDataProfile.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { hash } from '../../../base/common/hash.js';
7 > import { Emitter, Event } from '../../../base/common/event.js';
8 > import { Disposable } from '../../../base/common/lifecycle.js';
9 > import { basename, joinPath } from '../../../base/common/resources.js';
10 > import { URI, UriDto } from '../../../base/common/uri.js';
11 > import { localize } from '../../../nls.js';
12 > import { IEnvironmentService } from '../../environment/common/environment.js';
13 > import { FileOperationResult, IFileService, toFileOperationResult } from '../../files/common/files.js';
14 > import { createDecorator } from '../../instantiation/common/instantiation.js';
15 > import { ILogService } from '../../log/common/log.js';
16 > import { IAnyWorkspaceIdentifier, isSingleFolderWorkspaceIdentifier, isWorkspaceIdentifier } from '../../workspace/common/workspace.js';
17 > import { IStringDictionary } from '../../../base/common/collections.js';
18 > import { IUriIdentityService } from '../../uriIdentity/common/uriIdentity.js';
19 > import { Promises } from '../../../base/common/async.js';
20 > import { generateUuid } from '../../../base/common/uuid.js';
21 > import { escapeRegExpCharacters } from '../../../base/common/strings.js';
22 > import { isString, Mutable } from '../../../base/common/types.js';
23 >
24 > export const AGENTS_WINDOW_PROFILE_ID = 'agents';
25 >
26 > const AGENTS_WINDOW_PROFILE_FLAGS: UseDefaultProfileFlags = {
27 > settings: true,
28 > keybindings: true,
29 > prompts: true,
30 > mcp: true,
31 > languageModels: true,
32 > snippets: true,
33 > tasks: true,
34 > extensions: true,
35 > };
36 >
37 > export const enum ProfileResourceType {
38 > Settings = 'settings',
39 > Keybindings = 'keybindings',
40 > Snippets = 'snippets',
41 > Prompts = 'prompts',
42 > Tasks = 'tasks',
43 > Extensions = 'extensions',
44 > GlobalState = 'globalState',
45 > Mcp = 'mcp',
46 > LanguageModels = 'languageModels',
47 > }
48 >
49 > /**
50 > * Flags to indicate whether to use the default profile or not.
51 > */
52 > export type UseDefaultProfileFlags = { [key in ProfileResourceType]?: boolean };
53 > export type ProfileResourceTypeFlags = UseDefaultProfileFlags;
54 > export type SettingValue = string | boolean | number | undefined | null | object;
55 > export type ISettingsDictionary = Record<string, SettingValue>;
56 >
57 > export interface IUserDataProfile {
58 > readonly id: string;
59 > readonly isDefault: boolean;
60 > readonly name: string;
61 > readonly icon?: string;
62 > readonly location: URI;
63 > readonly globalStorageHome: URI;
64 > readonly settingsResource: URI;
65 > readonly keybindingsResource: URI;
66 > readonly tasksResource: URI;
67 > readonly snippetsHome: URI;
68 > readonly promptsHome: URI;
69 > readonly extensionsResource: URI;
70 > readonly mcpResource: URI;
71 > readonly languageModelsResource: URI;
72 > readonly agentPluginsHome: URI;
73 > readonly cacheHome: URI;
74 > readonly useDefaultFlags?: UseDefaultProfileFlags;
75 > readonly isInternal?: boolean;
76 > readonly isTransient?: boolean;
77 > readonly isAgentsWindowProfile?: boolean;
78 > readonly workspaces?: readonly URI[];
79 > }
80 >
81 > export function isUserDataProfile(thing: unknown): thing is IUserDataProfile {
82 const candidate = thing as IUserDataProfile | undefined;
83
99 );
100 }
102 > export interface IParsedUserDataProfileTemplate {
103 > readonly name: string;
104 > readonly icon?: string;
105 > readonly settings?: ISettingsDictionary;
106 > readonly globalState?: IStringDictionary<string>;
107 > }
108 >
109 > export interface ISystemProfileTemplate extends IParsedUserDataProfileTemplate {
110 > readonly id: string;
111 > }
112 >
113 > export type DidChangeProfilesEvent = { readonly added: readonly IUserDataProfile[]; readonly removed: readonly IUserDataProfile[]; readonly updated: readonly IUserDataProfile[]; readonly all: readonly IUserDataProfile[] };
114 >
115 > export type WillCreateProfileEvent = {
116 > profile: IUserDataProfile;
117 > join(promise: Promise<void>): void;
118 > };
119 >
120 > export type WillRemoveProfileEvent = {
121 > profile: IUserDataProfile;
122 > join(promise: Promise<void>): void;
123 > };
124 >
125 > export interface IUserDataProfileOptions {
126 > readonly icon?: string;
127 > readonly useDefaultFlags?: UseDefaultProfileFlags;
128 > readonly transient?: boolean;
129 > readonly workspaces?: readonly URI[];
130 > }
131 >
132 > export interface IUserDataProfileUpdateOptions extends Omit<IUserDataProfileOptions, 'icon'> {
133 > readonly name?: string;
134 > readonly icon?: string | null;
135 > }
136 >
137 > export const IUserDataProfilesService = createDecorator<IUserDataProfilesService>('IUserDataProfilesService');
138 > export interface IUserDataProfilesService {
139 > readonly _serviceBrand: undefined;
140 >
141 > readonly profilesHome: URI;
142 > readonly defaultProfile: IUserDataProfile;
143 >
144 > readonly onDidChangeProfiles: Event<DidChangeProfilesEvent>;
145 > readonly profiles: readonly IUserDataProfile[];
146 >
147 > readonly onDidResetWorkspaces: Event<void>;
148 >
149 > createNamedProfile(name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile>;
150 > createTransientProfile(workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile>;
151 > createProfile(id: string, name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile>;
152 > updateProfile(profile: IUserDataProfile, options?: IUserDataProfileUpdateOptions,): Promise<IUserDataProfile>;
153 > removeProfile(profile: IUserDataProfile): Promise<void>;
154 >
155 > setProfileForWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier, profile: IUserDataProfile): Promise<void>;
156 > resetWorkspaces(): Promise<void>;
157 >
158 > cleanUp(): Promise<void>;
159 > cleanUpTransientProfiles(): Promise<void>;
160 > }
161 >
162 > export function reviveProfile(profile: UriDto<IUserDataProfile>, scheme: string): IUserDataProfile {
163 return {
164 id: profile.id,
185 };
186 }
188 > export function toUserDataProfile(id: string, name: string, location: URI, profilesCacheHome: URI, options?: IUserDataProfileOptions, defaultProfile?: IUserDataProfile): IUserDataProfile {
189 > const isAgentsWindowProfile = id === AGENTS_WINDOW_PROFILE_ID; userDataProfile.ts
190 > return {
191 > id,
192 > name,
193 > location,
194 > isDefault: false,
195 > icon: options?.icon,
196 > globalStorageHome: defaultProfile && options?.useDefaultFlags?.globalState ? defaultProfile.globalStorageHome : joinPath(location, 'globalStorage'),
197 > settingsResource: defaultProfile && options?.useDefaultFlags?.settings ? defaultProfile.settingsResource : joinPath(location, 'settings.json'),
198 > keybindingsResource: defaultProfile && options?.useDefaultFlags?.keybindings ? defaultProfile.keybindingsResource : joinPath(location, 'keybindings.json'),
199 > tasksResource: defaultProfile && options?.useDefaultFlags?.tasks ? defaultProfile.tasksResource : joinPath(location, 'tasks.json'),
200 > snippetsHome: defaultProfile && options?.useDefaultFlags?.snippets ? defaultProfile.snippetsHome : joinPath(location, 'snippets'),
201 > promptsHome: defaultProfile && options?.useDefaultFlags?.prompts ? defaultProfile.promptsHome : joinPath(location, 'prompts'),
202 > extensionsResource: defaultProfile && options?.useDefaultFlags?.extensions ? defaultProfile.extensionsResource : joinPath(location, 'extensions.json'),
203 > mcpResource: defaultProfile && options?.useDefaultFlags?.mcp ? defaultProfile.mcpResource : joinPath(location, 'mcp.json'),
204 > languageModelsResource: defaultProfile && options?.useDefaultFlags?.languageModels ? defaultProfile.languageModelsResource : joinPath(location, 'chatLanguageModels.json'),
205 > agentPluginsHome: defaultProfile ? defaultProfile.agentPluginsHome : joinPath(location, 'agent-plugins'),
206 > cacheHome: joinPath(profilesCacheHome, id),
207 > useDefaultFlags: options?.useDefaultFlags,
208 > isTransient: options?.transient,
209 > isInternal: isAgentsWindowProfile || options?.transient,
210 > isAgentsWindowProfile,
211 > workspaces: options?.workspaces,
212 > };
213 > }
215 > export type UserDataProfilesObject = {
216 > profiles: IUserDataProfile[];
217 > emptyWindows: Map<string, IUserDataProfile>;
218 > };
219 >
220 > export type StoredUserDataProfile = {
221 > name: string;
222 > location: URI;
223 > icon?: string;
224 > useDefaultFlags?: UseDefaultProfileFlags;
225 > };
226 >
227 > export type StoredProfileAssociations = {
228 > workspaces?: IStringDictionary<string>;
229 > emptyWindows?: IStringDictionary<string>;
230 > };
231 >
232 > const SYSTEM_PROFILES_HOME = 'builtin';
233 >
234 > export class UserDataProfilesService extends Disposable implements IUserDataProfilesService {
235 >
236 > readonly _serviceBrand: undefined;
237 >
238 > protected static readonly PROFILES_KEY = 'userDataProfiles';
239 > protected static readonly PROFILE_ASSOCIATIONS_KEY = 'profileAssociations';
240 >
241 > readonly profilesHome: URI;
242 > private readonly profilesCacheHome: URI;
243 >
244 > get defaultProfile(): IUserDataProfile { return this.profiles[0]; }
245 > get profiles(): IUserDataProfile[] { return [...this.profilesObject.profiles, ...this.transientProfilesObject.profiles]; }
246 >
247 > protected readonly _onDidChangeProfiles = this._register(new Emitter<DidChangeProfilesEvent>());
248 > readonly onDidChangeProfiles = this._onDidChangeProfiles.event;
249 >
250 > protected readonly _onWillCreateProfile = this._register(new Emitter<WillCreateProfileEvent>());
251 > readonly onWillCreateProfile = this._onWillCreateProfile.event;
252 >
253 > protected readonly _onWillRemoveProfile = this._register(new Emitter<WillRemoveProfileEvent>());
254 > readonly onWillRemoveProfile = this._onWillRemoveProfile.event;
255 >
256 > private readonly _onDidResetWorkspaces = this._register(new Emitter<void>());
257 > readonly onDidResetWorkspaces = this._onDidResetWorkspaces.event;
258 >
259 > private profileCreationPromises = new Map<string, Promise<IUserDataProfile>>();
260 >
261 > protected readonly transientProfilesObject: UserDataProfilesObject = {
262 > profiles: [],
263 > emptyWindows: new Map()
264 > };
265 >
266 > constructor(
267 @IEnvironmentService protected environmentService: IEnvironmentService,
268 @IFileService protected fileService: IFileService,
274 this.profilesCacheHome = joinPath(this.environmentService.cacheHome, 'CachedProfilesData');
275 }
277 > init(): void {
278 this._profilesObject = undefined;
279 }
281 > protected _profilesObject: UserDataProfilesObject | undefined;
282 > protected get profilesObject(): UserDataProfilesObject {
283 if (!this._profilesObject) {
284 const defaultProfile = this.createDefaultProfile();
336 return this._profilesObject;
337 }
339 > private isInvalidProfile(storedProfile: StoredUserDataProfile): boolean {
340 if (!storedProfile.name) {
341 return true;
349 return false;
350 }
352 > protected createDefaultProfile() {
353 const defaultProfile = toUserDataProfile('__default__profile__', localize('defaultProfile', "Default"), this.environmentService.userRoamingDataHome, this.profilesCacheHome);
354 return { ...defaultProfile, extensionsResource: this.getDefaultProfileExtensionsLocation() ?? defaultProfile.extensionsResource, isDefault: true };
355 }
357 > async createTransientProfile(workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
358 const namePrefix = `Temp`;
359 const nameRegEx = new RegExp(`${escapeRegExpCharacters(namePrefix)}\\s(\\d+)`);
367 return this.createProfile(hash(generateUuid()).toString(16), name, { transient: true }, workspaceIdentifier);
368 }
370 > async createNamedProfile(name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
371 return this.createProfile(hash(generateUuid()).toString(16), name, options, workspaceIdentifier);
372 }
374 > async createProfile(id: string, name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
375 const profile = await this.doCreateProfile(id, name, options, workspaceIdentifier);
376
377 return profile;
378 }
380 > private async doCreateProfile(id: string, name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
381 if (!isString(name) || !name) {
382 throw new Error('Name of the profile is mandatory and must be of type `string`');
428 return profileCreationPromise;
429 }
431 > async updateProfile(profile: IUserDataProfile, options: IUserDataProfileUpdateOptions): Promise<IUserDataProfile> {
432 if (profile.isAgentsWindowProfile) {
433 throw new Error('Cannot update agents window profile');
481 return updatedProfile;
482 }
484 > async removeProfile(profileToRemove: IUserDataProfile): Promise<void> {
485 if (profileToRemove.isDefault) {
486 throw new Error('Cannot remove default profile');
515 }
516 }
518 > async setProfileForWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier, profileToSet: IUserDataProfile): Promise<void> {
519 const profile = this.profiles.find(p => p.id === profileToSet.id);
520 if (!profile) {
534 }
535 }
537 > unsetWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier, transient: boolean = false): void {
538 const workspace = this.getWorkspace(workspaceIdentifier);
539 if (URI.isUri(workspace)) {
547 }
548 }
550 > async resetWorkspaces(): Promise<void> {
551 this.transientProfilesObject.emptyWindows.clear();
552 this.profilesObject.emptyWindows.clear();
557 this._onDidResetWorkspaces.fire();
558 }
560 > async cleanUp(): Promise<void> {
561 try {
562 if (await this.fileService.exists(this.profilesHome)) {
587 }
588 }
590 > async cleanUpTransientProfiles(): Promise<void> {
591 const unAssociatedTransientProfiles = this.transientProfilesObject.profiles.filter(p => !this.isProfileAssociatedToWorkspace(p));
592 await Promise.allSettled(unAssociatedTransientProfiles.map(p => this.removeProfile(p)));
593 }
595 > getProfileForWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier): IUserDataProfile | undefined {
596 const workspace = this.getWorkspace(workspaceIdentifier);
597
604 : (this.profilesObject.emptyWindows.get(workspace) ?? this.transientProfilesObject.emptyWindows.get(workspace));
605 }
607 > protected getWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier): URI | string {
608 if (isSingleFolderWorkspaceIdentifier(workspaceIdentifier)) {
609 return workspaceIdentifier.uri;
614 return workspaceIdentifier.id;
615 }
617 > private isProfileAssociatedToWorkspace(profile: IUserDataProfile): boolean {
618 if (profile.workspaces?.length) {
619 return true;
627 return false;
628 }
630 > private updateProfiles(added: IUserDataProfile[], removed: IUserDataProfile[], updated: IUserDataProfile[], donotTrigger: boolean = false): void {
631 const allProfiles: Mutable<IUserDataProfile>[] = [...this.profiles, ...added];
632
679 }
680 }
682 > protected triggerProfilesChanges(added: IUserDataProfile[], removed: IUserDataProfile[], updated: IUserDataProfile[]) {
683 this._onDidChangeProfiles.fire({ added, removed, updated, all: this.profiles });
684 }
686 > private updateEmptyWindowAssociation(windowId: string, newProfile: IUserDataProfile | undefined, transient: boolean): void {
687 // Force transient if the new profile to associate is transient
688 transient = newProfile?.isTransient ? true : transient;
706 }
707 }
709 > private updateStoredProfiles(profiles: IUserDataProfile[]): void {
710 const storedProfiles: StoredUserDataProfile[] = [];
711 const workspaces: IStringDictionary<string> = {};
739 this._profilesObject = undefined;
740 }
742 > protected getStoredProfiles(): StoredUserDataProfile[] { return []; }
743 > protected saveStoredProfiles(storedProfiles: StoredUserDataProfile[]): void { throw new Error('not implemented'); }
744 >
745 > protected getStoredProfileAssociations(): StoredProfileAssociations { return {}; }
746 > protected saveStoredProfileAssociations(storedProfileAssociations: StoredProfileAssociations): void { throw new Error('not implemented'); }
747 > protected getDefaultProfileExtensionsLocation(): URI | undefined { return undefined; }
748 > }
749 >
750 > export class InMemoryUserDataProfilesService extends UserDataProfilesService {
751 private storedProfiles: StoredUserDataProfile[] = [];
752 protected override getStoredProfiles(): StoredUserDataProfile[] { return this.storedProfiles; }
754
755 private storedProfileAssociations: StoredProfileAssociations = {};
756 > protected override getStoredProfileAssociations(): StoredProfileAssociations { return this.storedProfileAssociations; } userDataProfile.ts
757 > protected override saveStoredProfileAssociations(storedProfileAssociations: StoredProfileAssociations): void { this.storedProfileAssociations = storedProfileAssociations; }
758 > }
src/vs/base/common/path.ts 279 covered LOC · 62 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- path.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // NOTE: VSCode's copy of nodejs path library to be usable in common (non-node) namespace
7 > // Copied from: https://github.com/nodejs/node/commits/v22.15.0/lib/path.js
8 > // Excluding: the change that adds primordials
9 > // (https://github.com/nodejs/node/commit/187a862d221dec42fa9a5c4214e7034d9092792f and others)
10 > // Excluding: the change that adds glob matching
11 > // (https://github.com/nodejs/node/commit/57b8b8e18e5e2007114c63b71bf0baedc01936a6)
12 >
13 > /**
14 > * Copyright Joyent, Inc. and other Node contributors.
15 > *
16 > * Permission is hereby granted, free of charge, to any person obtaining a
17 > * copy of this software and associated documentation files (the
18 > * "Software"), to deal in the Software without restriction, including
19 > * without limitation the rights to use, copy, modify, merge, publish,
20 > * distribute, sublicense, and/or sell copies of the Software, and to permit
21 > * persons to whom the Software is furnished to do so, subject to the
22 > * following conditions:
23 > *
24 > * The above copyright notice and this permission notice shall be included
25 > * in all copies or substantial portions of the Software.
26 > *
27 > * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
28 > * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
29 > * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN
30 > * NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM,
31 > * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
32 > * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
33 > * USE OR OTHER DEALINGS IN THE SOFTWARE.
34 > */
35 >
36 > import * as process from './process.js';
37 >
38 > const CHAR_UPPERCASE_A = 65;/* A */
39 > const CHAR_LOWERCASE_A = 97; /* a */
40 > const CHAR_UPPERCASE_Z = 90; /* Z */
41 > const CHAR_LOWERCASE_Z = 122; /* z */
42 > const CHAR_DOT = 46; /* . */
43 > const CHAR_FORWARD_SLASH = 47; /* / */
44 > const CHAR_BACKWARD_SLASH = 92; /* \ */
45 > const CHAR_COLON = 58; /* : */
46 > const CHAR_QUESTION_MARK = 63; /* ? */
47 >
48 > class ErrorInvalidArgType extends Error {
49 > code: 'ERR_INVALID_ARG_TYPE';
50 > constructor(name: string, expected: string, actual: unknown) {
51 // determiner: 'must be' or 'must not be'
52 let determiner;
66 this.code = 'ERR_INVALID_ARG_TYPE';
67 }
68 > } path.ts
69 >
70 function validateObject(pathObject: object, name: string) {
71 if (pathObject === null || typeof pathObject !== 'object') {
73 }
74 }
75 > path.ts
76 > function validateString(value: string, name: string) { path.ts
77 > if (typeof value !== 'string') {
78 throw new ErrorInvalidArgType(name, 'string', value);
79 }
80 > } path.ts
81 > path.ts
82 > const platformIsWin32 = (process.platform === 'win32');
83 >
84 function isPathSeparator(code: number | undefined) {
85 return code === CHAR_FORWARD_SLASH || code === CHAR_BACKWARD_SLASH;
86 }
87 > path.ts
88 > function isPosixPathSeparator(code: number | undefined) { path.ts
89 > return code === CHAR_FORWARD_SLASH;
90 > }
91 > path.ts
92 function isWindowsDeviceRoot(code: number) {
93 return (code >= CHAR_UPPERCASE_A && code <= CHAR_UPPERCASE_Z) ||
94 (code >= CHAR_LOWERCASE_A && code <= CHAR_LOWERCASE_Z);
95 }
96 > path.ts
97 > // Resolves . and .. elements in a path with directory names
98 > function normalizeString(path: string, allowAboveRoot: boolean, separator: string, isPathSeparator: (code?: number) => boolean) { path.ts
99 > let res = '';
100 > let lastSegmentLength = 0;
101 > let lastSlash = -1;
102 > let dots = 0;
103 > let code = 0;
104 > for (let i = 0; i <= path.length; ++i) {
105 > if (i < path.length) {
106 > code = path.charCodeAt(i);
107 > }
108 > else if (isPathSeparator(code)) {
109 break;
110 }
111 > else { path.ts
112 > code = CHAR_FORWARD_SLASH;
113 > }
114 > path.ts
115 > if (isPathSeparator(code)) {
116 > if (lastSlash === i - 1 || dots === 1) {
117 > // NOOP path.ts
118 > } else if (dots === 2) { path.ts
119 if (res.length < 2 || lastSegmentLength !== 2 ||
120 res.charCodeAt(res.length - 1) !== CHAR_DOT ||
144 lastSegmentLength = 2;
145 }
146 > } else { path.ts
147 > if (res.length > 0) {
148 > res += `${separator}${path.slice(lastSlash + 1, i)}`; path.ts
149 > }
150 > else { path.ts
151 > res = path.slice(lastSlash + 1, i);
152 > }
153 > lastSegmentLength = i - lastSlash - 1;
154 > }
155 > lastSlash = i; path.ts
156 > dots = 0;
157 > } else if (code === CHAR_DOT && dots !== -1) {
158 ++dots;
159 > } else { path.ts
160 > dots = -1;
161 > }
162 > } path.ts
163 > return res;
164 > }
165 > path.ts
166 function formatExt(ext: string): string {
167 return ext ? `${ext[0] === '.' ? '' : '.'}${ext}` : '';
168 }
169 > path.ts
170 function _format(sep: string, pathObject: ParsedPath) {
171 validateObject(pathObject, 'pathObject');
178 return dir === pathObject.root ? `${dir}${base}` : `${dir}${sep}${base}`;
179 }
180 > path.ts
181 > export interface ParsedPath {
182 > root: string;
183 > dir: string;
184 > base: string;
185 > ext: string;
186 > name: string;
187 > }
188 >
189 > export interface IPath {
190 > normalize(path: string): string;
191 > isAbsolute(path: string): boolean;
192 > join(...paths: string[]): string;
193 > resolve(...pathSegments: string[]): string;
194 > relative(from: string, to: string): string;
195 > dirname(path: string): string;
196 > basename(path: string, suffix?: string): string;
197 > extname(path: string): string;
198 > format(pathObject: ParsedPath): string;
199 > parse(path: string): ParsedPath;
200 > toNamespacedPath(path: string): string;
201 > sep: '\\' | '/';
202 > delimiter: string;
203 > win32: IPath | null;
204 > posix: IPath | null;
205 > }
206 >
207 > export const win32: IPath = {
208 > // path.resolve([from ...], to)
209 > resolve(...pathSegments: string[]): string {
210 let resolvedDevice = '';
211 let resolvedTail = '';
343 `${resolvedDevice}${resolvedTail}` || '.';
344 },
345 > path.ts
346 > normalize(path: string): string {
347 validateString(path, 'path');
348 const len = path.length;
450 return isAbsolute ? `${device}\\${tail}` : `${device}${tail}`;
451 },
452 > path.ts
453 > isAbsolute(path: string): boolean {
454 validateString(path, 'path');
455 const len = path.length;
466 isPathSeparator(path.charCodeAt(2)));
467 },
468 > path.ts
469 > join(...paths: string[]): string {
470 if (paths.length === 0) {
471 return '.';
536 return win32.normalize(joined);
537 },
538 > path.ts
539 >
540 > // It will solve the relative path from `from` to `to`, for instance:
541 > // from = 'C:\\orandea\\test\\aaa'
542 > // to = 'C:\\orandea\\impl\\bbb'
543 > // The output of the function should be: '..\\..\\impl\\bbb'
544 > relative(from: string, to: string): string {
545 validateString(from, 'from');
546 validateString(to, 'to');
699 return toOrig.slice(toStart, toEnd);
700 },
701 > path.ts
702 > toNamespacedPath(path: string): string {
703 // Note: this will *probably* throw somewhere.
704 if (typeof path !== 'string' || path.length === 0) {
730 return resolvedPath;
731 },
732 > path.ts
733 > dirname(path: string): string {
734 validateString(path, 'path');
735 const len = path.length;
818 return path.slice(0, end);
819 },
820 > path.ts
821 > basename(path: string, suffix?: string): string {
822 if (suffix !== undefined) {
823 validateString(suffix, 'suffix');
906 return path.slice(start, end);
907 },
908 > path.ts
909 > extname(path: string): string {
910 validateString(path, 'path');
911 let start = 0;
972 return path.slice(startDot, end);
973 },
974 > path.ts
975 > format: _format.bind(null, '\\'),
976 >
977 > parse(path) {
978 validateString(path, 'path');
979
1126 return ret;
1127 },
1128 > path.ts
1129 > sep: '\\',
1130 > delimiter: ';',
1131 > win32: null,
1132 > posix: null
1133 > };
1134 >
1135 > const posixCwd = (() => {
1136 > if (platformIsWin32) {
1137 // Converts Windows' backslash path separators to POSIX forward slashes
1138 // and truncates any drive indicator
1143 };
1144 }
1145 > path.ts
1146 > // We're already on POSIX, no need for any transformations
1147 > return () => process.cwd();
1148 > })();
1149 >
1150 > export const posix: IPath = {
1151 > // path.resolve([from ...], to)
1152 > resolve(...pathSegments: string[]): string {
1153 let resolvedPath = '';
1154 let resolvedAbsolute = false;
1186 return resolvedPath.length > 0 ? resolvedPath : '.';
1187 },
1188 > path.ts
1189 > normalize(path: string): string {
1190 > validateString(path, 'path'); path.ts
1191 >
1192 > if (path.length === 0) {
1193 return '.';
1194 }
1195 > path.ts
1196 > const isAbsolute = path.charCodeAt(0) === CHAR_FORWARD_SLASH;
1197 > const trailingSeparator =
1198 > path.charCodeAt(path.length - 1) === CHAR_FORWARD_SLASH;
1199 >
1200 > // Normalize the path
1201 > path = normalizeString(path, !isAbsolute, '/', isPosixPathSeparator);
1202 >
1203 > if (path.length === 0) {
1204 if (isAbsolute) {
1205 return '/';
1207 return trailingSeparator ? './' : '.';
1208 }
1209 > if (trailingSeparator) { path.ts
1210 path += '/';
1211 }
1212 > path.ts
1213 > return isAbsolute ? `/${path}` : path; path.ts
1214 > },
1215 > path.ts
1216 > isAbsolute(path: string): boolean {
1217 validateString(path, 'path');
1218 return path.length > 0 && path.charCodeAt(0) === CHAR_FORWARD_SLASH;
1219 },
1220 > path.ts
1221 > join(...paths: string[]): string {
1222 > if (paths.length === 0) { path.ts
1223 return '.';
1224 }
1225 > path.ts
1226 > const path = [];
1227 > for (let i = 0; i < paths.length; ++i) {
1228 > const arg = paths[i];
1229 > validateString(arg, 'path');
1230 > if (arg.length > 0) {
1231 > path.push(arg);
1232 > }
1233 > }
1234 >
1235 > if (path.length === 0) {
1236 return '.';
1237 }
1238 > path.ts
1239 > return posix.normalize(path.join('/'));
1240 > },
1241 > path.ts
1242 > relative(from: string, to: string): string {
1243 validateString(from, 'from');
1244 validateString(to, 'to');
1312 return `${out}${to.slice(toStart + lastCommonSep)}`;
1313 },
1314 > path.ts
1315 > toNamespacedPath(path: string): string {
1316 // Non-op on posix systems
1317 return path;
1318 },
1319 > path.ts
1320 > dirname(path: string): string {
1321 validateString(path, 'path');
1322 if (path.length === 0) {
1346 return path.slice(0, end);
1347 },
1348 > path.ts
1349 > basename(path: string, suffix?: string): string {
1350 > if (suffix !== undefined) { path.ts
1351 validateString(suffix, 'suffix');
1352 }
1353 > validateString(path, 'path'); path.ts
1354 >
1355 > let start = 0;
1356 > let end = -1;
1357 > let matchedSlash = true;
1358 > let i;
1359 >
1360 > if (suffix !== undefined && suffix.length > 0 && suffix.length <= path.length) {
1361 if (suffix === path) {
1362 return '';
1405 return path.slice(start, end);
1406 }
1407 > for (i = path.length - 1; i >= 0; --i) { path.ts
1408 > if (path.charCodeAt(i) === CHAR_FORWARD_SLASH) { path.ts
1409 > // If we reached a path separator that was not part of a set of path path.ts
1410 > // separators at the end of the string, stop now
1411 > if (!matchedSlash) {
1412 > start = i + 1;
1413 > break;
1414 > }
1415 > } else if (end === -1) { path.ts
1416 > // We saw the first non-path separator, mark this as the end of our
1417 > // path component
1418 > matchedSlash = false;
1419 > end = i + 1;
1420 > }
1421 > }
1422 > path.ts
1423 > if (end === -1) {
1424 return '';
1425 }
1426 > return path.slice(start, end); path.ts
1427 > }, path.ts
1428 > path.ts
1429 > extname(path: string): string {
1430 validateString(path, 'path');
1431 let startDot = -1;
1480 return path.slice(startDot, end);
1481 },
1482 > path.ts
1483 > format: _format.bind(null, '/'),
1484 >
1485 > parse(path: string): ParsedPath {
1486 validateString(path, 'path');
1487
1565 return ret;
1566 },
1567 > path.ts
1568 > sep: '/',
1569 > delimiter: ':',
1570 > win32: null,
1571 > posix: null
1572 > };
1573 >
1574 > posix.win32 = win32.win32 = win32;
1575 > posix.posix = win32.posix = posix;
1576 >
1577 > export const normalize = (platformIsWin32 ? win32.normalize : posix.normalize);
1578 > export const isAbsolute = (platformIsWin32 ? win32.isAbsolute : posix.isAbsolute);
1579 > export const join = (platformIsWin32 ? win32.join : posix.join);
1580 > export const resolve = (platformIsWin32 ? win32.resolve : posix.resolve);
1581 > export const relative = (platformIsWin32 ? win32.relative : posix.relative);
1582 > export const dirname = (platformIsWin32 ? win32.dirname : posix.dirname);
1583 > export const basename = (platformIsWin32 ? win32.basename : posix.basename);
1584 > export const extname = (platformIsWin32 ? win32.extname : posix.extname);
1585 > export const format = (platformIsWin32 ? win32.format : posix.format);
1586 > export const parse = (platformIsWin32 ? win32.parse : posix.parse);
1587 > export const toNamespacedPath = (platformIsWin32 ? win32.toNamespacedPath : posix.toNamespacedPath);
1588 > export const sep = (platformIsWin32 ? win32.sep : posix.sep);
1589 > export const delimiter = (platformIsWin32 ? win32.delimiter : posix.delimiter);
src/vs/platform/agentHost/common/state/protocol/channels-changeset/state.ts 277 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { StringOrMarkdown, FileEdit, ErrorInfo } from '../common/state.js';
10 >
11 > // ─── Changesets ──────────────────────────────────────────────────────────────
12 >
13 > /**
14 > * Catalogue entry describing one changeset the server can produce for a
15 > * session.
16 > *
17 > * Catalogue entries are intentionally lightweight — just enough to render a
18 > * chip or list row without subscribing. Full per-changeset detail
19 > * ({@link ChangesetState}) lives on the subscribable URI obtained by
20 > * expanding {@link uriTemplate}.
21 > *
22 > * @category Changesets
23 > */
24 > export interface Changeset {
25 > /** Human-readable label, e.g. `"Uncommitted Changes"`. */
26 > label: string;
27 > /**
28 > * RFC 6570 URI template. Clients parse the variables directly out of the
29 > * template using the standard `{name}` syntax — they are not redeclared
30 > * here.
31 > *
32 > * Only the following template shapes are defined by this protocol; any
33 > * other variable name MUST be ignored by clients (there is no
34 > * protocol-defined way to obtain values for unknown variables):
35 > *
36 > * | Variables in template | Meaning |
37 > * | ------------------------------------------- | ------------------------------------------------------------------------------------ |
38 > * | _(none)_ | A static, session-wide changeset. The template is itself a subscribable URI. |
39 > * | `{turnId}` | Per-turn slice. Expand with a `Turn.id` from the session. |
40 > * | `{originalTurnId}` and `{modifiedTurnId}` | Diff between two turns. Both variables MUST be present. |
41 > *
42 > * Future protocol versions MAY add new well-known variables.
43 > */
44 > uriTemplate: string;
45 > /** Optional longer description. */
46 > description?: string;
47 > /**
48 > * Advisory hint describing what kind of changeset this is, so clients can
49 > * group, sort, or render an appropriate icon without parsing
50 > * {@link uriTemplate}. Recognized values include:
51 > *
52 > * - `'session'`: a static, session-wide changeset covering all changes the
53 > * agent has produced in this session.
54 > * - `'branch'`: changes relative to a base branch (e.g. a feature branch
55 > * diffed against `main`).
56 > * - `'uncommitted'`: the workspace's current uncommitted changes.
57 > * - `'turn'`: changes produced by a single turn. Typically paired with a
58 > * `{turnId}` variable in {@link uriTemplate}.
59 > * - `'compare-turns'`: a diff between two turns. Typically paired with
60 > * `{originalTurnId}` and `{modifiedTurnId}` variables in
61 > * {@link uriTemplate}.
62 > *
63 > * Implementations MAY provide additional values; clients SHOULD fall back
64 > * to a reasonable default when an unknown value is encountered.
65 > */
66 > changeKind: string;
67 > /**
68 > * Optional capability declarations for this changeset. Absent (or an empty
69 > * object) means the changeset advertises no optional capabilities.
70 > *
71 > * Because the catalogue entry is delivered up-front on
72 > * {@link ChangesetState | the session's changeset list}, clients can decide
73 > * whether to surface capability-gated UI (such as review checkboxes) without
74 > * first subscribing to the changeset URI. Mirrors the presence-flag
75 > * convention of `ClientCapabilities`.
76 > */
77 > capabilities?: ChangesetCapabilities;
78 > }
79 >
80 > /**
81 > * Optional capabilities a changeset advertises on its catalogue
82 > * {@link Changeset} entry.
83 > *
84 > * Each field is a presence flag: an empty object `{}` means "supported",
85 > * absence means "not supported". Sub-fields on individual capabilities are
86 > * reserved for future per-capability options.
87 > *
88 > * @category Changesets
89 > */
90 > export interface ChangesetCapabilities {
91 > /**
92 > * The changeset supports the per-file **review** workflow. When declared,
93 > * clients MAY surface a GitHub-style "Viewed" toggle per file and dispatch
94 > * {@link ChangesetFilesReviewChangedAction | `changeset/filesReviewChanged`} to
95 > * set each file's {@link ChangesetFile.reviewed} flag. Clients that omit
96 > * handling MUST treat the changeset as non-reviewable.
97 > */
98 > review?: Record<string, never>;
99 > }
100 >
101 > /**
102 > * Computation lifecycle of a {@link ChangesetState}.
103 > *
104 > * @category Changesets
105 > */
106 > export const enum ChangesetStatus {
107 > /** The server is still computing the contents of this changeset. */
108 > Computing = 'computing',
109 > /** The changeset has been fully computed and is up-to-date. */
110 > Ready = 'ready',
111 > /**
112 > * Computation failed. The cause is described by
113 > * {@link ChangesetState.error}.
114 > */
115 > Error = 'error',
116 > }
117 >
118 > /**
119 > * Full state for a single changeset, returned when a client subscribes to
120 > * an expanded changeset URI.
121 > *
122 > * The client already knows the URI it subscribed to, so this state does
123 > * not redundantly carry it (or the catalogue's `id`, `label`, etc.).
124 > * Aggregate counts (`additions`, `deletions`, `files`) are likewise
125 > * omitted: clients trivially compute them from `files[].edit.diff`.
126 > *
127 > * @category Changesets
128 > */
129 > export interface ChangesetState {
130 > /** Computation lifecycle. */
131 > status: ChangesetStatus;
132 > /** Present iff `status === ChangesetStatus.Error`. */
133 > error?: ErrorInfo;
134 > /** Files in this changeset, keyed by {@link ChangesetFile.id}. */
135 > files: ChangesetFile[];
136 > /**
137 > * Operations the client may invoke against this changeset. Omit when no
138 > * operations are available.
139 > */
140 > operations?: ChangesetOperation[];
141 > }
142 >
143 > /**
144 > * One file entry within a {@link ChangesetState}.
145 > *
146 > * @category Changesets
147 > */
148 > export interface ChangesetFile {
149 > /**
150 > * Stable identifier within the changeset. Typically `after.uri`
151 > * (or `before.uri` for deletions).
152 > */
153 > id: string;
154 > /**
155 > * Reuses the existing {@link FileEdit} shape. Clients derive line
156 > * additions, deletions, and rename/create/delete semantics from this.
157 > */
158 > edit: FileEdit;
159 > /**
160 > * Whether a reviewer has marked this file as reviewed (the GitHub-style
161 > * "Viewed" checkbox). Absent is equivalent to `false` — clients MUST treat
162 > * a missing value as not-yet-reviewed.
163 > *
164 > * Requires the changeset to advertise {@link ChangesetCapabilities.review}.
165 > * Clients toggle it by dispatching
166 > * {@link ChangesetFilesReviewChangedAction | `changeset/filesReviewChanged`};
167 > * the server MAY also originate it (e.g. an agent self-reviewing its own
168 > * output).
169 > *
170 > * There is no content version in the protocol, so review is **not** reset
171 > * automatically when a file's contents change under a stable id. The server,
172 > * which is the authority on what changed, resets review explicitly — either
173 > * by re-emitting the file (via {@link ChangesetFileSetAction} or
174 > * {@link ChangesetContentChangedAction}) without `reviewed: true`, or by
175 > * dispatching `changeset/filesReviewChanged` with `reviewed: false`.
176 > */
177 > reviewed?: boolean;
178 > /**
179 > * Server-defined opaque metadata, surfaced to operations and tooling
180 > * but not interpreted by the protocol.
181 > */
182 > _meta?: Record<string, unknown>;
183 > }
184 >
185 > /**
186 > * Execution lifecycle of a {@link ChangesetOperation}.
187 > *
188 > * An operation is invoked imperatively via `invokeChangesetOperation`, but
189 > * its progress and outcome are reflected back into changeset state so that
190 > * every subscriber observes a consistent view (e.g. a spinner on a "Create
191 > * Pull Request" button, or an inline error after a failed "revert").
192 > *
193 > * @category Changesets
194 > */
195 > export const enum ChangesetOperationStatus {
196 > /**
197 > * The operation is ready to be invoked. This is the default when
198 > * {@link ChangesetOperation.status} is omitted.
199 > */
200 > Idle = 'idle',
201 > /** An invocation of this operation is currently in flight. */
202 > Running = 'running',
203 > /**
204 > * The most recent invocation failed. The cause is described by
205 > * {@link ChangesetOperation.error}.
206 > */
207 > Error = 'error',
208 > /**
209 > * The operation is currently disabled and cannot be invoked.
210 > */
211 > Disabled = 'disabled',
212 > }
213 >
214 > /**
215 > * Where a {@link ChangesetOperation} can be invoked.
216 > *
217 > * @category Changesets
218 > */
219 > export const enum ChangesetOperationScope {
220 > /** Applies to the whole changeset. */
221 > Changeset = 'changeset',
222 > /** Applies to a single file within the changeset. */
223 > Resource = 'resource',
224 > /** Applies to a line range within a single file. */
225 > Range = 'range',
226 > }
227 >
228 > /**
229 > * A server-declared invokable verb the client can run against a
230 > * changeset, a file, or a range — `"stage"`, `"revert"`, `"create-pr"`,
231 > * and so on.
232 > *
233 > * The term "operation" is used deliberately to avoid colliding with the
234 > * protocol-level [Actions](/guide/actions) that mutate state.
235 > *
236 > * @category Changesets
237 > */
238 > export interface ChangesetOperation {
239 > /** Stable identifier, unique within this changeset. */
240 > id: string;
241 > /** Human-readable button/menu label. */
242 > label: string;
243 > /** Optional longer description shown on hover or in tooltips. */
244 > description?: string;
245 > /** Where this operation can be invoked. */
246 > scopes: ChangesetOperationScope[];
247 > /**
248 > * Optional confirmation prompt to show before invoking. When present,
249 > * the client MUST display this message to the user (typically in a
250 > * confirmation dialog) and only invoke the operation after the user
251 > * accepts. The presence of this field also signals that the operation
252 > * is destructive — clients SHOULD style the affirmative button
253 > * accordingly (e.g. with a warning colour).
254 > */
255 > confirmation?: StringOrMarkdown;
256 > /** Optional generic icon hint, e.g. `"check"`, `"trash"`. */
257 > icon?: string;
258 > /** Optional group identifier, used to group related operations together. */
259 > group?: string;
260 > /**
261 > * Current execution status. The server sets
262 > * {@link ChangesetOperationStatus.Running | Running} while an invocation
263 > * is in flight, {@link ChangesetOperationStatus.Error | Error} when the
264 > * most recent invocation failed, and
265 > * {@link ChangesetOperationStatus.Idle | Idle} otherwise.
266 > *
267 > * Clients SHOULD reflect this state in the UI — e.g. disabling the
268 > * control or showing a spinner while `Running`, and surfacing
269 > * {@link error} while `Error`.
270 > */
271 > status: ChangesetOperationStatus;
272 > /**
273 > * Cause of failure. Present iff
274 > * `status === ChangesetOperationStatus.Error`; otherwise omitted.
275 > */
276 > error?: ErrorInfo;
277 > }
src/vs/platform/request/common/request.ts 277 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- request.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { streamToBuffer } from '../../../base/common/buffer.js';
7 > import { CancellationToken } from '../../../base/common/cancellation.js';
8 > import { getErrorMessage } from '../../../base/common/errors.js';
9 > import { Emitter, Event } from '../../../base/common/event.js';
10 > import { Disposable } from '../../../base/common/lifecycle.js';
11 > import { IHeaders, IRequestContext, IRequestOptions } from '../../../base/parts/request/common/request.js';
12 > import { localize } from '../../../nls.js';
13 > import { ConfigurationScope, Extensions, IConfigurationNode, IConfigurationRegistry } from '../../configuration/common/configurationRegistry.js';
14 > import { createDecorator } from '../../instantiation/common/instantiation.js';
15 > import { ILogService } from '../../log/common/log.js';
16 > import { Registry } from '../../registry/common/platform.js';
17 >
18 > export const IRequestService = createDecorator<IRequestService>('requestService');
19 >
20 > /**
21 > * Use as the {@link IRequestOptions.callSite} value to prevent
22 > * request telemetry from being emitted. This is needed for
23 > * callers such as the telemetry sender to avoid cyclical calls.
24 > */
25 > export const NO_FETCH_TELEMETRY = 'NO_FETCH_TELEMETRY';
26 >
27 > export interface IRequestCompleteEvent {
28 > readonly callSite: string;
29 > readonly latency: number;
30 > readonly statusCode: number | undefined;
31 > }
32 >
33 > export interface AuthInfo {
34 > isProxy: boolean;
35 > scheme: string;
36 > host: string;
37 > port: number;
38 > realm: string;
39 > attempt: number;
40 > }
41 >
42 > export interface Credentials {
43 > username: string;
44 > password: string;
45 > }
46 >
47 > export interface IRequestService {
48 > readonly _serviceBrand: undefined;
49 >
50 > /**
51 > * Fires when a request completes (successfully or with an error response).
52 > */
53 > readonly onDidCompleteRequest: Event<IRequestCompleteEvent>;
54 >
55 > request(options: IRequestOptions, token: CancellationToken): Promise<IRequestContext>;
56 >
57 > resolveProxy(url: string): Promise<string | undefined>;
58 > lookupAuthorization(authInfo: AuthInfo): Promise<Credentials | undefined>;
59 > lookupKerberosAuthorization(url: string): Promise<string | undefined>;
60 > loadCertificates(): Promise<string[]>;
61 > }
62 >
63 > class LoggableHeaders {
64 >
65 > private headers: IHeaders | undefined;
66 >
67 > constructor(private readonly original: IHeaders) { }
68 >
69 > toJSON(): any {
70 if (!this.headers) {
71 const headers = Object.create(null);
81 return this.headers;
82 }
83 > request.ts
84 > }
85 >
86 > export abstract class AbstractRequestService extends Disposable implements IRequestService {
87 >
88 > declare readonly _serviceBrand: undefined;
89 >
90 > private counter = 0;
91 >
92 > private readonly _onDidCompleteRequest = this._register(new Emitter<IRequestCompleteEvent>());
93 > readonly onDidCompleteRequest = this._onDidCompleteRequest.event;
94 >
95 > constructor(protected readonly logService: ILogService) {
96 super();
97 }
98 > request.ts
99 > protected async logAndRequest(options: IRequestOptions, request: () => Promise<IRequestContext>): Promise<IRequestContext> {
100 const prefix = `#${++this.counter}: ${options.url}`;
101 this.logService.trace(`${prefix} - begin`, options.type, new LoggableHeaders(options.headers ?? {}));
115 }
116 }
117 > request.ts
118 > abstract request(options: IRequestOptions, token: CancellationToken): Promise<IRequestContext>;
119 > abstract resolveProxy(url: string): Promise<string | undefined>;
120 > abstract lookupAuthorization(authInfo: AuthInfo): Promise<Credentials | undefined>;
121 > abstract lookupKerberosAuthorization(url: string): Promise<string | undefined>;
122 > abstract loadCertificates(): Promise<string[]>;
123 > }
124 >
125 > export function isSuccess(context: IRequestContext): boolean {
126 return (context.res.statusCode && context.res.statusCode >= 200 && context.res.statusCode < 300) || context.res.statusCode === 1223;
127 }
128 > request.ts
129 > export function isClientError(context: IRequestContext): boolean {
130 return !!context.res.statusCode && context.res.statusCode >= 400 && context.res.statusCode < 500;
131 }
132 > request.ts
133 > export function isServerError(context: IRequestContext): boolean {
134 return !!context.res.statusCode && context.res.statusCode >= 500 && context.res.statusCode < 600;
135 }
136 > request.ts
137 > /**
138 > * Reads a header value from an {@link IHeaders} map, tolerating array-shaped
139 > * values and case-insensitive lookups.
140 > */
141 > export function readHeader(headers: IHeaders | undefined, name: string): string | undefined {
142 if (!headers) {
143 return undefined;
149 return value;
150 }
151 > request.ts
152 > /**
153 > * Parses the `Retry-After` header as a number of seconds. Returns `undefined`
154 > * if absent or not a finite positive number. The HTTP-date form is not parsed.
155 > */
156 > export function retryAfterFromHeaders(headers: IHeaders | undefined): number | undefined {
157 const value = readHeader(headers, 'retry-after');
158 if (!value) {
162 return Number.isFinite(parsed) && parsed > 0 ? parsed : undefined;
163 }
164 > request.ts
165 > export function hasNoContent(context: IRequestContext): boolean {
166 return context.res.statusCode === 204;
167 }
168 > request.ts
169 export async function asText(context: IRequestContext): Promise<string | null> {
170 if (hasNoContent(context)) {
174 return buffer.toString();
175 }
176 > request.ts
177 export async function asTextOrError(context: IRequestContext): Promise<string | null> {
178 if (!isSuccess(context)) {
181 return asText(context);
182 }
183 > request.ts
184 export async function asJson<T = {}>(context: IRequestContext): Promise<T | null> {
185 if (!isSuccess(context)) {
198 }
199 }
200 > request.ts
201 > export function updateProxyConfigurationsScope(useHostProxy: boolean, useHostProxyDefault: boolean): void {
202 registerProxyConfigurations(useHostProxy, useHostProxyDefault);
203 }
204 > request.ts
205 > export const USER_LOCAL_AND_REMOTE_SETTINGS = [
206 > 'http.proxy',
207 > 'http.proxyStrictSSL',
208 > 'http.proxyKerberosServicePrincipal',
209 > 'http.noProxy',
210 > 'http.proxyAuthorization',
211 > 'http.proxySupport',
212 > 'http.systemCertificates',
213 > 'http.systemCertificatesNode',
214 > 'http.experimental.systemCertificatesV2',
215 > 'http.fetchAdditionalSupport',
216 > 'http.experimental.networkInterfaceCheckInterval',
217 > ];
218 >
219 > export const systemCertificatesNodeDefault = false;
220 >
221 > let proxyConfiguration: IConfigurationNode[] = [];
222 > let previousUseHostProxy: boolean | undefined = undefined;
223 > let previousUseHostProxyDefault: boolean | undefined = undefined;
224 > function registerProxyConfigurations(useHostProxy = true, useHostProxyDefault = true): void {
225 > if (previousUseHostProxy === useHostProxy && previousUseHostProxyDefault === useHostProxyDefault) {
226 return;
227 }
228 > request.ts
229 > previousUseHostProxy = useHostProxy;
230 > previousUseHostProxyDefault = useHostProxyDefault;
231 >
232 > const configurationRegistry = Registry.as<IConfigurationRegistry>(Extensions.Configuration);
233 > const oldProxyConfiguration = proxyConfiguration;
234 > proxyConfiguration = [
235 > {
236 > id: 'http',
237 > order: 15,
238 > title: localize('httpConfigurationTitle', "HTTP"),
239 > type: 'object',
240 > scope: ConfigurationScope.MACHINE,
241 > properties: {
242 > 'http.useLocalProxyConfiguration': {
243 > type: 'boolean',
244 > default: useHostProxyDefault,
245 > markdownDescription: localize('useLocalProxy', "Controls whether in the remote extension host the local proxy configuration should be used. This setting only applies as a remote setting during [remote development](https://aka.ms/vscode-remote)."),
246 > restricted: true
247 > },
248 > }
249 > },
250 > {
251 > id: 'http',
252 > order: 15,
253 > title: localize('httpConfigurationTitle', "HTTP"),
254 > type: 'object',
255 > scope: ConfigurationScope.APPLICATION,
256 > properties: {
257 > 'http.electronFetch': {
258 > type: 'boolean',
259 > default: false,
260 > description: localize('electronFetch', "Controls whether use of Electron's fetch implementation instead of Node.js' should be enabled. All local extensions will get Electron's fetch implementation for the global fetch API."),
261 > restricted: true
262 > },
263 > }
264 > },
265 > {
266 > id: 'http',
267 > order: 15,
268 > title: localize('httpConfigurationTitle', "HTTP"),
269 > type: 'object',
270 > scope: useHostProxy ? ConfigurationScope.APPLICATION : ConfigurationScope.MACHINE,
271 > properties: {
272 > 'http.proxy': {
273 > type: 'string',
274 > pattern: '^(https?|socks|socks4a?|socks5h?)://([^:]*(:[^@]*)?@)?([^:]+|\\[[:0-9a-fA-F]+\\])(:\\d+)?/?$|^$',
275 > markdownDescription: localize('proxy', "The proxy setting to use. If not set, will be inherited from the `http_proxy` and `https_proxy` environment variables. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
276 > restricted: true
277 > },
278 > 'http.proxyStrictSSL': {
279 > type: 'boolean',
280 > default: true,
281 > markdownDescription: localize('strictSSL', "Controls whether the proxy server certificate should be verified against the list of supplied CAs. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
282 > restricted: true
283 > },
284 > 'http.proxyKerberosServicePrincipal': {
285 > type: 'string',
286 > markdownDescription: localize('proxyKerberosServicePrincipal', "Overrides the principal service name for Kerberos authentication with the HTTP proxy. A default based on the proxy hostname is used when this is not set. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
287 > restricted: true
288 > },
289 > 'http.noProxy': {
290 > type: 'array',
291 > items: { type: 'string' },
292 > markdownDescription: localize('noProxy', "Specifies domain names for which proxy settings should be ignored for HTTP/HTTPS requests. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
293 > restricted: true
294 > },
295 > 'http.proxyAuthorization': {
296 > type: ['null', 'string'],
297 > default: null,
298 > markdownDescription: localize('proxyAuthorization', "The value to send as the `Proxy-Authorization` header for every network request. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
299 > restricted: true
300 > },
301 > 'http.proxySupport': {
302 > type: 'string',
303 > enum: ['off', 'on', 'fallback', 'override'],
304 > enumDescriptions: [
305 > localize('proxySupportOff', "Disable proxy support for extensions."),
306 > localize('proxySupportOn', "Enable proxy support for extensions."),
307 > localize('proxySupportFallback', "Enable proxy support for extensions, fall back to request options, when no proxy found."),
308 > localize('proxySupportOverride', "Enable proxy support for extensions, override request options."),
309 > ],
310 > default: 'override',
311 > markdownDescription: localize('proxySupport', "Use the proxy support for extensions. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
312 > restricted: true
313 > },
314 > 'http.systemCertificates': {
315 > type: 'boolean',
316 > default: true,
317 > markdownDescription: localize('systemCertificates', "Controls whether CA certificates should be loaded from the OS. On Windows and macOS, a reload of the window is required after turning this off. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
318 > restricted: true
319 > },
320 > 'http.systemCertificatesNode': {
321 > type: 'boolean',
322 > tags: ['experimental'],
323 > default: systemCertificatesNodeDefault,
324 > markdownDescription: localize('systemCertificatesNode', "Controls whether system certificates should be loaded using Node.js built-in support. Reload the window after changing this setting. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
325 > restricted: true,
326 > experiment: {
327 > mode: 'auto'
328 > }
329 > },
330 > 'http.experimental.systemCertificatesV2': {
331 > type: 'boolean',
332 > tags: ['experimental'],
333 > default: false,
334 > markdownDescription: localize('systemCertificatesV2', "Controls whether experimental loading of CA certificates from the OS should be enabled. This uses a more general approach than the default implementation. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
335 > restricted: true
336 > },
337 > 'http.fetchAdditionalSupport': {
338 > type: 'boolean',
339 > default: true,
340 > markdownDescription: localize('fetchAdditionalSupport', "Controls whether Node.js' fetch implementation should be extended with additional support. Currently proxy support ({1}) and system certificates ({2}) are added when the corresponding settings are enabled. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`', '`#http.proxySupport#`', '`#http.systemCertificates#`'),
341 > restricted: true
342 > },
343 > 'http.webSocketAdditionalSupport': {
344 > type: 'boolean',
345 > default: true,
346 > markdownDescription: localize('webSocketAdditionalSupport', "Controls whether the built-in WebSocket implementation should be extended with additional support. Currently proxy support ({1}) and system certificates ({2}) are added when the corresponding settings are enabled. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`', '`#http.proxySupport#`', '`#http.systemCertificates#`'),
347 > restricted: true
348 > },
349 > 'http.experimental.networkInterfaceCheckInterval': {
350 > type: 'number',
351 > default: 300,
352 > minimum: -1,
353 > tags: ['experimental'],
354 > markdownDescription: localize('networkInterfaceCheckInterval', "Controls the interval in seconds for checking network interface changes to invalidate the proxy cache. Set to -1 to disable. When during [remote development](https://aka.ms/vscode-remote) the {0} setting is disabled this setting can be configured in the local and the remote settings separately.", '`#http.useLocalProxyConfiguration#`'),
355 > restricted: true,
356 > experiment: {
357 > mode: 'auto'
358 > }
359 > }
360 > }
361 > }
362 > ];
363 > configurationRegistry.updateConfigurations({ add: proxyConfiguration, remove: oldProxyConfiguration });
364 > }
365 >
366 > registerProxyConfigurations();
src/vs/workbench/common/editor/editorInput.ts 265 covered LOC · 28 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- editorInput.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter } from '../../../base/common/event.js';
7 > import { URI } from '../../../base/common/uri.js';
8 > import { EditorInputCapabilities, Verbosity, GroupIdentifier, ISaveOptions, IRevertOptions, IMoveResult, IEditorDescriptor, IEditorPane, IUntypedEditorInput, EditorResourceAccessor, AbstractEditorInput, isEditorInput, IEditorIdentifier } from '../editor.js';
9 > import { isEqual } from '../../../base/common/resources.js';
10 > import { ConfirmResult } from '../../../platform/dialogs/common/dialogs.js';
11 > import { IMarkdownString } from '../../../base/common/htmlContent.js';
12 > import { IDisposable } from '../../../base/common/lifecycle.js';
13 > import { ThemeIcon } from '../../../base/common/themables.js';
14 >
15 > export interface IEditorCloseHandler {
16 >
17 > /**
18 > * If `true`, will call into the `confirm` method to ask for confirmation
19 > * before closing the editor.
20 > */
21 > showConfirm(): boolean;
22 >
23 > /**
24 > * Allows an editor to control what should happen when the editor
25 > * (or a list of editor of the same kind) is being closed.
26 > *
27 > * By default a file specific dialog will open if the editor is
28 > * dirty and not in the process of saving.
29 > *
30 > * If the editor is not dealing with files or another condition
31 > * should be used besides dirty state, this method should be
32 > * implemented to show a different dialog.
33 > *
34 > * @param editors All editors of the same kind that are being closed. Should be used
35 > * to show a combined dialog.
36 > */
37 > confirm(editors: ReadonlyArray<IEditorIdentifier>): Promise<ConfirmResult>;
38 > }
39 >
40 > export interface IUntypedEditorOptions {
41 >
42 > /**
43 > * Implementations should try to preserve as much
44 > * view state as possible from the typed input based
45 > * on the group the editor is opened.
46 > */
47 > readonly preserveViewState?: GroupIdentifier;
48 >
49 > /**
50 > * Implementations should preserve the original
51 > * resource of the typed input and not alter
52 > * it.
53 > */
54 > readonly preserveResource?: boolean;
55 > }
56 >
57 > /**
58 > * Editor inputs are lightweight objects that can be passed to the workbench API to open inside the editor part.
59 > * Each editor input is mapped to an editor that is capable of opening it through the Platform facade.
60 > */
61 > export abstract class EditorInput extends AbstractEditorInput {
62
63 protected readonly _onDidChangeDirty = this._register(new Emitter<void>());
86 */
87 readonly onWillDispose = this._onWillDispose.event;
89 > /**
90 > * Optional: subclasses can override to implement
91 > * custom confirmation on close behavior.
92 > */
93 > readonly closeHandler?: IEditorCloseHandler;
94 >
95 > /**
96 > * Unique type identifier for this input. Every editor input of the
97 > * same class should share the same type identifier. The type identifier
98 > * is used for example for serialising/deserialising editor inputs
99 > * via the serialisers of the `EditorInputFactoryRegistry`.
100 > */
101 > abstract get typeId(): string;
102 >
103 > /**
104 > * Returns the optional associated resource of this input.
105 > *
106 > * This resource should be unique for all editors of the same
107 > * kind and input and is often used to identify the editor input among
108 > * others.
109 > *
110 > * **Note:** DO NOT use this property for anything but identity
111 > * checks. DO NOT use this property to present as label to the user.
112 > * Please refer to `EditorResourceAccessor` documentation in that case.
113 > */
114 > abstract get resource(): URI | undefined;
115 >
116 > /**
117 > * Identifies the type of editor this input represents
118 > * This ID is registered with the {@link EditorResolverService} to allow
119 > * for resolving an untyped input to a typed one
120 > */
121 > get editorId(): string | undefined {
122 return undefined;
123 }
125 > /**
126 > * The capabilities of the input.
127 > */
128 > get capabilities(): EditorInputCapabilities {
129 return EditorInputCapabilities.Readonly;
130 }
132 > /**
133 > * Figure out if the input has the provided capability.
134 > */
135 > hasCapability(capability: EditorInputCapabilities): boolean {
136 if (capability === EditorInputCapabilities.None) {
137 return this.capabilities === EditorInputCapabilities.None;
140 return (this.capabilities & capability) !== 0;
141 }
143 > isReadonly(): boolean | IMarkdownString {
144 return this.hasCapability(EditorInputCapabilities.Readonly);
145 }
147 > /**
148 > * Returns the display name of this input.
149 > */
150 > getName(): string {
151 return `Editor ${this.typeId}`;
152 }
154 > /**
155 > * Returns the display description of this input.
156 > */
157 > getDescription(verbosity?: Verbosity): string | undefined {
158 return undefined;
159 }
161 > /**
162 > * Returns the display title of this input.
163 > */
164 > getTitle(verbosity?: Verbosity): string {
165 return this.getName();
166 }
168 > /**
169 > * Returns the extra classes to apply to the label of this input.
170 > */
171 > getLabelExtraClasses(): string[] {
172 return [];
173 }
175 > /**
176 > * Returns the aria label to be read out by a screen reader.
177 > */
178 > getAriaLabel(): string {
179 return this.getTitle(Verbosity.SHORT);
180 }
182 > /**
183 > * Returns the icon which represents this editor input.
184 > * If undefined, the default icon will be used.
185 > */
186 > getIcon(): ThemeIcon | URI | undefined {
187 return undefined;
188 }
190 > /**
191 > * Returns a descriptor suitable for telemetry events.
192 > *
193 > * Subclasses should extend if they can contribute.
194 > */
195 > getTelemetryDescriptor(): { [key: string]: unknown } {
196 /* __GDPR__FRAGMENT__
197 "EditorTelemetryDescriptor" : {
201 return { typeId: this.typeId };
202 }
204 > /**
205 > * Returns if this input is dirty or not.
206 > */
207 > isDirty(): boolean {
208 return false;
209 }
211 > /**
212 > * Returns if the input has unsaved changes.
213 > */
214 > isModified(): boolean {
215 return this.isDirty();
216 }
218 > /**
219 > * Returns if this input is currently being saved or soon to be
220 > * saved. Based on this assumption the editor may for example
221 > * decide to not signal the dirty state to the user assuming that
222 > * the save is scheduled to happen anyway.
223 > */
224 > isSaving(): boolean {
225 return false;
226 }
228 > /**
229 > * Returns a type of `IDisposable` that represents the resolved input.
230 > * Subclasses should override to provide a meaningful model or return
231 > * `null` if the editor does not require a model.
232 > *
233 > * The `options` parameter are passed down from the editor when the
234 > * input is resolved as part of it.
235 > */
236 > async resolve(): Promise<IDisposable | null> {
237 return null;
238 }
240 > /**
241 > * Saves the editor. The provided groupId helps implementors
242 > * to e.g. preserve view state of the editor and re-open it
243 > * in the correct group after saving.
244 > *
245 > * @returns the resulting editor input (typically the same) of
246 > * this operation or `undefined` to indicate that the operation
247 > * failed or was canceled.
248 > */
249 > async save(group: GroupIdentifier, options?: ISaveOptions): Promise<EditorInput | IUntypedEditorInput | undefined> {
250 return this;
251 }
253 > /**
254 > * Saves the editor to a different location. The provided `group`
255 > * helps implementors to e.g. preserve view state of the editor
256 > * and re-open it in the correct group after saving.
257 > *
258 > * @returns the resulting editor input (typically a different one)
259 > * of this operation or `undefined` to indicate that the operation
260 > * failed or was canceled.
261 > */
262 > async saveAs(group: GroupIdentifier, options?: ISaveOptions): Promise<EditorInput | IUntypedEditorInput | undefined> {
263 return this;
264 }
266 > /**
267 > * Reverts this input from the provided group.
268 > */
269 > async revert(group: GroupIdentifier, options?: IRevertOptions): Promise<void> { }
270 >
271 > /**
272 > * Called to determine how to handle a resource that is renamed that matches
273 > * the editors resource (or is a child of).
274 > *
275 > * Implementors are free to not implement this method to signal no intent
276 > * to participate. If an editor is returned though, it will replace the
277 > * current one with that editor and optional options.
278 > */
279 > async rename(group: GroupIdentifier, target: URI): Promise<IMoveResult | undefined> {
280 return undefined;
281 }
283 > /**
284 > * Returns a copy of the current editor input. Used when we can't just reuse the input
285 > */
286 > copy(): EditorInput {
287 return this;
288 }
290 > /**
291 > * Indicates if this editor can be moved to another group. By default
292 > * editors can freely be moved around groups. If an editor cannot be
293 > * moved, a message should be returned to show to the user.
294 > *
295 > * @returns `true` if the editor can be moved to the target group, or
296 > * a string with a message to show to the user if the editor cannot be
297 > * moved.
298 > */
299 > canMove(sourceGroup: GroupIdentifier, targetGroup: GroupIdentifier): true | string {
300 return true;
301 }
303 > /**
304 > * Indicates if this editor can be reopened after being closed. By default
305 > * editors can be reopened. Subclasses can override to prevent this.
306 > *
307 > * @returns `true` if the editor can be reopened after being closed.
308 > */
309 > canReopen(): boolean {
310 return true;
311 }
313 > /**
314 > * Returns if the other object matches this input.
315 > */
316 > matches(otherInput: EditorInput | IUntypedEditorInput): boolean {
317
318 // Typed inputs: via === check
331 return isEqual(this.resource, EditorResourceAccessor.getCanonicalUri(otherInput));
332 }
334 > /**
335 > * If a editor was registered onto multiple editor panes, this method
336 > * will be asked to return the preferred one to use.
337 > *
338 > * @param editorPanes a list of editor pane descriptors that are candidates
339 > * for the editor to open in.
340 > */
341 > prefersEditorPane<T extends IEditorDescriptor<IEditorPane>>(editorPanes: T[]): T | undefined {
342 return editorPanes.at(0);
343 }
345 > /**
346 > * Returns a representation of this typed editor input as untyped
347 > * resource editor input that e.g. can be used to serialize the
348 > * editor input into a form that it can be restored.
349 > *
350 > * May return `undefined` if an untyped representation is not supported.
351 > */
352 > toUntyped(options?: IUntypedEditorOptions): IUntypedEditorInput | undefined {
353 return undefined;
354 }
356 > /**
357 > * Returns if this editor is disposed.
358 > */
359 > isDisposed(): boolean {
360 return this._store.isDisposed;
361 }
363 > override dispose(): void {
364 if (!this.isDisposed()) {
365 this._onWillDispose.fire();
src/vs/workbench/contrib/chat/common/promptSyntax/utils/promptFilesLocator.ts 261 covered LOC · 31 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptFilesLocator.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../../base/common/uri.js';
7 > import { isAbsolute } from '../../../../../../base/common/path.js';
8 > import { ResourceSet } from '../../../../../../base/common/map.js';
9 > import * as nls from '../../../../../../nls.js';
10 > import { FileOperation, FileOperationError, FileOperationResult, IFileService } from '../../../../../../platform/files/common/files.js';
11 > import { getPromptFileLocationsConfigKey, isTildePath, PromptsConfig } from '../config/config.js';
12 > import { basename, dirname, isEqual, isEqualOrParent, joinPath } from '../../../../../../base/common/resources.js';
13 > import { IWorkspaceContextService, IWorkspaceFolder } from '../../../../../../platform/workspace/common/workspace.js';
14 > import { IConfigurationService } from '../../../../../../platform/configuration/common/configuration.js';
15 > import { AGENTS_SOURCE_FOLDER, CLAUDE_CONFIG_FOLDER, COPILOT_CONFIG_FOLDER, GITHUB_CONFIG_FOLDER, getPromptFileExtension, getPromptFileType, LEGACY_MODE_FILE_EXTENSION, getCleanPromptName, AGENT_FILE_EXTENSION, getPromptFileDefaultLocations, SKILL_FILENAME, IPromptSourceFolder, IResolvedPromptSourceFolder } from '../config/promptFileLocations.js';
16 > import { PromptFileSource, PromptsType } from '../promptTypes.js';
17 > import { IWorkbenchEnvironmentService } from '../../../../../services/environment/common/environmentService.js';
18 > import { Schemas } from '../../../../../../base/common/network.js';
19 > import { getExcludes, IFileQuery, ISearchConfiguration, ISearchService, QueryType } from '../../../../../services/search/common/search.js';
20 > import { CancellationToken, CancellationTokenSource } from '../../../../../../base/common/cancellation.js';
21 > import { isCancellationError } from '../../../../../../base/common/errors.js';
22 > import { AgentInstructionFileType, IPromptPath, IAgentInstructionFile, Logger, PromptsStorage } from '../service/promptsService.js';
23 > import { IUserDataProfileService } from '../../../../../services/userDataProfile/common/userDataProfile.js';
24 > import { Emitter, Event } from '../../../../../../base/common/event.js';
25 > import { DisposableStore, toDisposable } from '../../../../../../base/common/lifecycle.js';
26 > import { ILogService } from '../../../../../../platform/log/common/log.js';
27 > import { IPathService } from '../../../../../services/path/common/pathService.js';
28 > import { equalsIgnoreCase } from '../../../../../../base/common/strings.js';
29 > import { IWorkspaceTrustManagementService } from '../../../../../../platform/workspace/common/workspaceTrust.js';
30 > import { AGENT_HOST_SCHEME } from '../../../../../../platform/agentHost/common/agentHostUri.js';
31 >
32 > /**
33 > * Maximum recursion depth when traversing subdirectories for instruction files.
34 > */
35 > const MAX_INSTRUCTIONS_RECURSION_DEPTH = 5;
36 >
37 > export interface IWorkspaceInstructionFile {
38 > readonly fileName: string;
39 > readonly type: AgentInstructionFileType;
40 > }
41 >
42 > /**
43 > * Utility class to locate prompt files.
44 > */
45 > export class PromptFilesLocator {
46 >
47 > private readonly userDataFolder: IResolvedPromptSourceFolder;
48 >
49 > constructor(
50 @IFileService private readonly fileService: IFileService,
51 @IConfigurationService private readonly configService: IConfigurationService,
70 };
71 }
73 > protected getWorkspaceFolders(): readonly IWorkspaceFolder[] {
74 // Agent host workspace folders surface customizations through AHP
75 // (session state + findAgentSkills), not via filesystem scanning.
79 return this.workspaceService.getWorkspace().folders.filter(f => f.uri.scheme !== AGENT_HOST_SCHEME);
80 }
82 > protected getWorkspaceFolder(resource: URI): IWorkspaceFolder | undefined {
83 return this.workspaceService.getWorkspaceFolder(resource) ?? undefined;
84 }
86 > protected onDidChangeWorkspaceFolders(): Event<void> {
87 return Event.map(this.workspaceService.onDidChangeWorkspaceFolders, () => undefined);
88 }
90 > /**
91 > * Returns the configured prompt source folders for the given type.
92 > * Subclasses can override to filter out unsupported sources.
93 > */
94 > protected getPromptSourceFolders(type: PromptsType): IPromptSourceFolder[] {
95 return PromptsConfig.promptSourceFolders(this.configService, type);
96 }
98 > /**
99 > * Returns the default prompt source folders for the given type.
100 > * Subclasses can override to filter out unsupported sources.
101 > */
102 > protected getDefaultSourceFolders(type: PromptsType): readonly IPromptSourceFolder[] {
103 return getPromptFileDefaultLocations(type);
104 }
106 > public async getWorkspaceFolderRoots(includeParents: boolean, logger?: Logger): Promise<URI[]> {
107 const workspaceFolders = this.getWorkspaceFolders();
108 if (includeParents) {
123 return workspaceFolders.map(f => f.uri);
124 }
126 > /**
127 > * Walks up from {@link folderUri} collecting parent folders until a
128 > * repository root (a folder containing `.git`) is found. Returns the
129 > * intermediate parent folders only when a repo root is found; returns
130 > * an empty array when the walk reaches the filesystem root, the user
131 > * home directory, or a folder already present in {@link seen}.
132 > */
133 > private async findParentRepoFolders(folderUri: URI, userHome: URI, seen: ResourceSet, logger?: Logger): Promise<URI[]> {
134 const candidates: URI[] = [];
135 let current = folderUri;
164 return [];
165 }
167 > /**
168 > * List all prompt files from the filesystem.
169 > *
170 > * @returns List of prompt files found in the workspace.
171 > */
172 > public async listFiles(type: PromptsType, storage: PromptsStorage, token: CancellationToken): Promise<readonly URI[]> {
173 if (storage !== PromptsStorage.user && storage !== PromptsStorage.local) {
174 throw new Error(`Unsupported prompt file storage: ${storage}`);
199 return [...paths];
200 }
202 > public createFilesUpdatedEvent(type: PromptsType): { readonly event: Event<void>; dispose: () => void } {
203 const disposables = new DisposableStore();
204 const eventEmitter = disposables.add(new Emitter<void>());
265 return { event: eventEmitter.event, dispose: () => disposables.dispose() };
266 }
268 > public createAgentInstructionsUpdatedEvent(): { readonly event: Event<void>; dispose: () => void } {
269 const disposables = new DisposableStore();
270 const eventEmitter = disposables.add(new Emitter<void>());
372 return { event: eventEmitter.event, dispose: () => disposables.dispose() };
373 }
375 > /**
376 > * Gets the hook source folders for creating new hooks.
377 > * Returns configured hook folders, excluding Claude paths (which are read-only).
378 > */
379 > public async getHookSourceFolders(): Promise<readonly IResolvedPromptSourceFolder[]> {
380 const configuredLocations = this.getPromptSourceFolders(PromptsType.hook);
381
404 return result;
405 }
407 > /**
408 > * Get all possible unambiguous prompt file source folders based on
409 > * the current workspace folder structure.
410 > *
411 > * This method is currently primarily used by the `> Create Prompt`
412 > * command that providers users with the list of destination folders
413 > * for a newly created prompt file. Because such a list cannot contain
414 > * paths that include `glob pattern` in them, we need to process config
415 > * values and try to create a list of clear and unambiguous locations.
416 > *
417 > * @returns List of possible unambiguous prompt file folders.
418 > */
419 > public async getConfigBasedSourceFolders(type: PromptsType): Promise<readonly URI[]> {
420 const configuredLocations = this.getPromptSourceFolders(type);
421 const absoluteLocations = await this.toAbsoluteLocations(type, configuredLocations);
461 return [...result];
462 }
464 > /**
465 > * Gets all resolved source folders for the given prompt type with metadata.
466 > * This method merges configured locations with default locations and resolves them
467 > * to absolute paths, including displayPath and isDefault information.
468 > *
469 > * The returned order prefers workspace (local) folders first, then user folders.
470 > * This is used for UX like the "Create Prompt" command where workspace is preferred.
471 > *
472 > * @param type The type of prompt files.
473 > * @returns List of resolved source folders with metadata.
474 > */
475 > public async getResolvedSourceFolders(type: PromptsType): Promise<readonly IResolvedPromptSourceFolder[]> {
476 const absoluteLocations = await this.getLocalStorageFolders(type);
477
480 return this.dedupeSourceFolders([...localFolders, ...userFolders]);
481 }
483 > /**
484 > * Gets all resolved source folders in the same order that file discovery
485 > * searches them (user folders first, then local/workspace folders).
486 > * This matches the order used by {@link listFiles} and should be used
487 > * for debug/diagnostic output so the displayed order is accurate.
488 > */
489 > public async getSourceFoldersInDiscoveryOrder(type: PromptsType): Promise<readonly IResolvedPromptSourceFolder[]> {
490 const absoluteLocations = await this.getLocalStorageFolders(type);
491 const userFolders = absoluteLocations.filter(loc => loc.storage === PromptsStorage.user);
493 return this.dedupeSourceFolders([...userFolders, ...localFolders]);
494 }
496 > /**
497 > * Gets all local (workspace) storage folders for the given prompt type.
498 > * This merges default folders with configured locations.
499 > */
500 > private async getLocalStorageFolders(type: PromptsType): Promise<readonly IResolvedPromptSourceFolder[]> {
501 const configuredLocations = this.getPromptSourceFolders(type);
502 const defaultFolders = this.getDefaultSourceFolders(type);
514 return absoluteLocations;
515 }
517 > /**
518 > * Deduplicates source folders by URI.
519 > */
520 > private dedupeSourceFolders(folders: readonly IResolvedPromptSourceFolder[]): IResolvedPromptSourceFolder[] {
521 const seen = new ResourceSet();
522 const result: IResolvedPromptSourceFolder[] = [];
529 return result;
530 }
532 > /**
533 > * Converts locations defined in `settings` to absolute filesystem path URIs with metadata.
534 > * This conversion is needed because locations in settings can be relative,
535 > * hence we need to resolve them based on the current workspace folders.
536 > * If userHome is provided, paths starting with `~` will be expanded. Otherwise these paths are ignored.
537 > * Preserves the type and location properties from the source folder definitions.
538 > */
539 > private async toAbsoluteLocations(type: PromptsType, configuredLocations: readonly IPromptSourceFolder[], defaultLocations?: readonly IPromptSourceFolder[]): Promise<IResolvedPromptSourceFolder[]> {
540 const result: IResolvedPromptSourceFolder[] = [];
541 const seen = new ResourceSet();
614 return result;
615 }
617 > /**
618 > * Uses the file service to resolve the provided location and return either the file at the location of files in the directory.
619 > * For instruction folders, this searches recursively (up to {@link MAX_INSTRUCTIONS_RECURSION_DEPTH} levels deep) provided
620 > * the location is not a workspace folder root and does not contain wildcards, to support subdirectories while avoiding
621 > * accidentally broad traversal.
622 > */
623 > private async resolveFilesAtLocation(location: URI, type: PromptsType, token: CancellationToken, depth: number = 0): Promise<URI[]> {
624 if (type === PromptsType.skill) {
625 return this.findAgentSkillsInFolder(location, token);
663 return [];
664 }
666 > /**
667 > * Uses the search service to find all files at the provided location.
668 > * Requires a FileSearchProvider to be available for the folder's scheme.
669 > */
670 > private async searchFilesInLocation(folder: URI, filePattern: string | undefined, token: CancellationToken): Promise<URI[]> {
671 // Check if a FileSearchProvider is available for this scheme
672 if (!this.searchService.schemeHasFileSearchProvider(folder.scheme)) {
703 return [];
704 }
706 > /**
707 > * Gets list of `AGENTS.md` files anywhere in the workspace.
708 > */
709 > public async findAgentMDsInWorkspace(token: CancellationToken): Promise<IAgentInstructionFile[]> {
710 const result = await Promise.all(this.getWorkspaceFolders().map(folder => this.findAgentMDsInFolder(folder.uri, token)));
711 return result.flat(1);
712 }
714 > private async findAgentMDsInFolder(folder: URI, token: CancellationToken): Promise<IAgentInstructionFile[]> {
715 // Check if a FileSearchProvider is available for this scheme
716 if (this.searchService.schemeHasFileSearchProvider(folder.scheme)) {
750 }
751 }
753 > /**
754 > * Recursively traverses a folder using the file service to find AGENTS.md files.
755 > * This is used as a fallback when no FileSearchProvider is available for the scheme.
756 > */
757 > private async findAgentMDsUsingFileService(folder: URI, token: CancellationToken): Promise<IAgentInstructionFile[]> {
758 const result: IAgentInstructionFile[] = [];
759 const agentsMdFileName = 'agents.md';
784 return result;
785 }
787 >
788 >
789 > public async findFilesInRoots(roots: URI[], folder: string | undefined, paths: IWorkspaceInstructionFile[], token: CancellationToken, result: IAgentInstructionFile[] = []): Promise<IAgentInstructionFile[]> {
790 const toResolve = roots.map(root => ({ resource: folder !== undefined ? joinPath(root, folder) : root }));
791 const resolvedRoots = await this.fileService.resolveAll(toResolve);
808 return result;
809 }
811 > public getAgentFileURIFromModeFile(oldURI: URI): URI | undefined {
812 if (oldURI.path.endsWith(LEGACY_MODE_FILE_EXTENSION)) {
813 let newLocation;
822 return undefined;
823 }
825 > private async findAgentSkillsInFolder(uri: URI, token: CancellationToken): Promise<URI[]> {
826 try {
827 const result: URI[] = [];
854 }
855 }
857 > /**
858 > * Searches for skills in all configured locations.
859 > */
860 > public async findAgentSkills(token: CancellationToken): Promise<IPromptPath[]> {
861 const configuredLocations = this.getPromptSourceFolders(PromptsType.skill);
862 const absoluteLocations = await this.toAbsoluteLocations(PromptsType.skill, configuredLocations);
875 return allResults;
876 }
878 >
879 >
880 > /**
881 > * Checks if the provided path contains a glob pattern (* or **).
882 > * Used to detect deprecated glob usage in prompt file locations.
883 > *
884 > * @param path - path to check
885 > * @returns `true` if the path contains `*` or `**`, `false` otherwise
886 > */
887 > export function hasGlobPattern(path: string): boolean {
888 return path.includes('*');
889 }
891 >
892 > /**
893 > * Checks if the provided `pattern` could be a valid glob pattern.
894 > */
895 > export function isValidGlob(pattern: string): boolean {
896 let squareBrackets = false;
897 let squareBracketsCount = 0;
959 return false;
960 }
962 > interface ISearchLocationResult {
963 > readonly searchRoot: URI;
964 > readonly filePattern?: string;
965 > }
966 >
967 > /**
968 > * Resolves the search root and optional file pattern for the provided location.
969 > * For paths with glob patterns, finds the deepest non-glob ancestor directory.
970 > *
971 > * Assumes that the location that is provided has a valid path (is abstract)
972 > *
973 > * ## Examples
974 > *
975 > * ```typescript
976 > * assert.strictDeepEqual(
977 > * resolveSearchLocation(PromptsType.prompt, URI.file('/home/user/{folder1,folder2}/file.md')),
978 > * { searchRoot: URI.file('/home/user'), filePattern: '{folder1,folder2}/file.md' },
979 > * 'Must find correct non-glob search root.',
980 > * );
981 > * ```
982 > */
983 function resolveSearchLocation(type: PromptsType, location: URI): ISearchLocationResult {
984 if (type !== PromptsType.instructions && type !== PromptsType.prompt) {
1008 };
1009 }
1011 >
1012 > /**
1013 > * Regex pattern string for validating paths for all prompt files.
1014 > * Paths only support:
1015 > * - Relative paths: someFolder, ./someFolder
1016 > * - User home paths: ~/folder (only forward slash, not backslash for cross-platform sharing)
1017 > * - Parent relative paths for monorepos: ../folder
1018 > *
1019 > * NOT supported:
1020 > * - Absolute paths (portability issue)
1021 > * - Glob patterns with * or ** (performance issue)
1022 > * - Backslashes (paths should be shareable in repos across platforms)
1023 > * - Tilde without forward slash (e.g., ~abc, ~\folder)
1024 > * - Empty or whitespace-only paths
1025 > *
1026 > * The regex validates:
1027 > * - Not a Windows absolute path (e.g., C:\, C:/)
1028 > * - Not starting with / (Unix absolute path)
1029 > * - No backslashes anywhere (use forward slashes only)
1030 > * - If starts with ~, must be followed by /
1031 > * - No glob pattern characters: * ? [ ] { }
1032 > * - At least one non-whitespace character
1033 > */
1034 > export const VALID_PROMPT_FOLDER_PATTERN = '^(?![A-Za-z]:[\\\\/])(?!/)(?!~(?!/))(?!.*\\\\)(?!.*[*?\\[\\]{}]).*\\S.*$';
1035 > const VALID_PROMPT_FOLDER_REGEX = new RegExp(VALID_PROMPT_FOLDER_PATTERN);
1036 >
1037 > /**
1038 > * Validates if a path is allowed for simplified path configurations.
1039 > * Only forward slashes are supported to ensure paths are shareable across platforms.
1040 > */
1041 > export function isValidPromptFolderPath(path: string): boolean {
1042 return VALID_PROMPT_FOLDER_REGEX.test(path);
1043 }
src/vs/base/common/resources.ts 259 covered LOC · 26 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- resources.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from './charCode.js';
7 > import * as extpath from './extpath.js';
8 > import { Schemas } from './network.js';
9 > import * as paths from './path.js';
10 > import { isLinux, isWindows } from './platform.js';
11 > import { compare as strCompare, equalsIgnoreCase } from './strings.js';
12 > import { URI, uriToFsPath } from './uri.js';
13 >
14 > export function originalFSPath(uri: URI): string {
15 return uriToFsPath(uri, true);
16 }
18 > //#region IExtUri
19 >
20 > export interface IExtUri {
21 >
22 > // --- identity
23 >
24 > /**
25 > * Compares two uris.
26 > *
27 > * @param uri1 Uri
28 > * @param uri2 Uri
29 > * @param ignoreFragment Ignore the fragment (defaults to `false`)
30 > */
31 > compare(uri1: URI, uri2: URI, ignoreFragment?: boolean): number;
32 >
33 > /**
34 > * Tests whether two uris are equal
35 > *
36 > * @param uri1 Uri
37 > * @param uri2 Uri
38 > * @param ignoreFragment Ignore the fragment (defaults to `false`)
39 > */
40 > isEqual(uri1: URI | undefined, uri2: URI | undefined, ignoreFragment?: boolean): boolean;
41 >
42 > /**
43 > * Tests whether a `candidate` URI is a parent or equal of a given `base` URI.
44 > *
45 > * @param base A uri which is "longer" or at least same length as `parentCandidate`
46 > * @param parentCandidate A uri which is "shorter" or up to same length as `base`
47 > * @param ignoreFragment Ignore the fragment (defaults to `false`)
48 > */
49 > isEqualOrParent(base: URI, parentCandidate: URI, ignoreFragment?: boolean): boolean;
50 >
51 > /**
52 > * Creates a key from a resource URI to be used to resource comparison and for resource maps.
53 > * @see {@link ResourceMap}
54 > * @param uri Uri
55 > * @param ignoreFragment Ignore the fragment (defaults to `false`)
56 > */
57 > getComparisonKey(uri: URI, ignoreFragment?: boolean): string;
58 >
59 > /**
60 > * Whether the casing of the path-component of the uri should be ignored.
61 > */
62 > ignorePathCasing(uri: URI): boolean;
63 >
64 > // --- path math
65 >
66 > basenameOrAuthority(resource: URI): string;
67 >
68 > /**
69 > * Returns the basename of the path component of an uri.
70 > * @param resource
71 > */
72 > basename(resource: URI): string;
73 >
74 > /**
75 > * Returns the extension of the path component of an uri.
76 > * @param resource
77 > */
78 > extname(resource: URI): string;
79 > /**
80 > * Return a URI representing the directory of a URI path.
81 > *
82 > * @param resource The input URI.
83 > * @returns The URI representing the directory of the input URI.
84 > */
85 > dirname(resource: URI): URI;
86 > /**
87 > * Join a URI path with path fragments and normalizes the resulting path.
88 > *
89 > * @param resource The input URI.
90 > * @param pathFragment The path fragment to add to the URI path.
91 > * @returns The resulting URI.
92 > */
93 > joinPath(resource: URI, ...pathFragment: string[]): URI;
94 > /**
95 > * Normalizes the path part of a URI: Resolves `.` and `..` elements with directory names.
96 > *
97 > * @param resource The URI to normalize the path.
98 > * @returns The URI with the normalized path.
99 > */
100 > normalizePath(resource: URI): URI;
101 > /**
102 > *
103 > * @param from
104 > * @param to
105 > */
106 > relativePath(from: URI, to: URI): string | undefined;
107 > /**
108 > * Resolves an absolute or relative path against a base URI.
109 > * The path can be relative or absolute posix or a Windows path
110 > */
111 > resolvePath(base: URI, path: string): URI;
112 >
113 > // --- misc
114 >
115 > /**
116 > * Returns true if the URI path is absolute.
117 > */
118 > isAbsolutePath(resource: URI): boolean;
119 > /**
120 > * Tests whether the two authorities are the same
121 > */
122 > isEqualAuthority(a1: string, a2: string): boolean;
123 > /**
124 > * Returns true if the URI path has a trailing path separator
125 > */
126 > hasTrailingPathSeparator(resource: URI, sep?: string): boolean;
127 > /**
128 > * Removes a trailing path separator, if there's one.
129 > * Important: Doesn't remove the first slash, it would make the URI invalid
130 > */
131 > removeTrailingPathSeparator(resource: URI, sep?: string): URI;
132 > /**
133 > * Adds a trailing path separator to the URI if there isn't one already.
134 > * For example, c:\ would be unchanged, but c:\users would become c:\users\
135 > */
136 > addTrailingPathSeparator(resource: URI, sep?: string): URI;
137 > }
138 >
139 > export class ExtUri implements IExtUri {
140 >
141 > constructor(private _ignorePathCasing: (uri: URI) => boolean) { }
142 >
143 > compare(uri1: URI, uri2: URI, ignoreFragment: boolean = false): number {
144 if (uri1 === uri2) {
145 return 0;
147 return strCompare(this.getComparisonKey(uri1, ignoreFragment), this.getComparisonKey(uri2, ignoreFragment));
148 }
149 > resources.ts
150 > isEqual(uri1: URI | undefined, uri2: URI | undefined, ignoreFragment: boolean = false): boolean {
151 if (uri1 === uri2) {
152 return true;
157 return this.getComparisonKey(uri1, ignoreFragment) === this.getComparisonKey(uri2, ignoreFragment);
158 }
159 > resources.ts
160 > getComparisonKey(uri: URI, ignoreFragment: boolean = false): string {
161 return uri.with({
162 path: this._ignorePathCasing(uri) ? uri.path.toLowerCase() : undefined,
164 }).toString();
165 }
166 > resources.ts
167 > ignorePathCasing(uri: URI): boolean {
168 return this._ignorePathCasing(uri);
169 }
170 > resources.ts
171 > isEqualOrParent(base: URI, parentCandidate: URI, ignoreFragment: boolean = false): boolean {
172 if (base.scheme === parentCandidate.scheme) {
173 if (base.scheme === Schemas.file) {
180 return false;
181 }
182 > resources.ts
183 > // --- path math
184 >
185 > joinPath(resource: URI, ...pathFragment: string[]): URI {
186 > return URI.joinPath(resource, ...pathFragment); resources.ts
187 > }
188 > resources.ts
189 > basenameOrAuthority(resource: URI): string {
190 > return basename(resource) || resource.authority; resources.ts
191 > }
192 > resources.ts
193 > basename(resource: URI, suffix?: string): string {
194 > return paths.posix.basename(resource.path, suffix); resources.ts
195 > }
196 > resources.ts
197 > extname(resource: URI): string {
198 return paths.posix.extname(resource.path);
199 }
200 > resources.ts
201 > dirname(resource: URI): URI {
202 if (resource.path.length === 0) {
203 return resource;
217 });
218 }
219 > resources.ts
220 > normalizePath(resource: URI): URI {
221 if (!resource.path.length) {
222 return resource;
232 });
233 }
234 > resources.ts
235 > relativePath(from: URI, to: URI): string | undefined {
236 if (from.scheme !== to.scheme || !isEqualAuthority(from.authority, to.authority)) {
237 return undefined;
257 return paths.posix.relative(fromPath, toPath);
258 }
259 > resources.ts
260 > resolvePath(base: URI, path: string): URI {
261 if (base.scheme === Schemas.file) {
262 const newURI = URI.file(paths.resolve(originalFSPath(base), path));
271 });
272 }
273 > resources.ts
274 > // --- misc
275 >
276 > isAbsolutePath(resource: URI): boolean {
277 return !!resource.path && resource.path[0] === '/';
278 }
279 > resources.ts
280 > isEqualAuthority(a1: string | undefined, a2: string | undefined) {
281 return a1 === a2 || (a1 !== undefined && a2 !== undefined && equalsIgnoreCase(a1, a2));
282 }
283 > resources.ts
284 > hasTrailingPathSeparator(resource: URI, sep: string = paths.sep): boolean {
285 if (resource.scheme === Schemas.file) {
286 const fsp = originalFSPath(resource);
291 }
292 }
293 > resources.ts
294 > removeTrailingPathSeparator(resource: URI, sep: string = paths.sep): URI {
295 // Make sure that the path isn't a drive letter. A trailing separator there is not removable.
296 if (hasTrailingPathSeparator(resource, sep)) {
299 return resource;
300 }
301 > resources.ts
302 > addTrailingPathSeparator(resource: URI, sep: string = paths.sep): URI {
303 let isRootSep: boolean = false;
304 if (resource.scheme === Schemas.file) {
315 return resource;
316 }
317 > } resources.ts
318 >
319 >
320 > /**
321 > * Unbiased utility that takes uris "as they are". This means it can be interchanged with
322 > * uri#toString() usages. The following is true
323 > * ```
324 > * assertEqual(aUri.toString() === bUri.toString(), exturi.isEqual(aUri, bUri))
325 > * ```
326 > */
327 > export const extUri = new ExtUri(() => false);
328 >
329 > /**
330 > * BIASED utility that _mostly_ ignored the case of urs paths. ONLY use this util if you
331 > * understand what you are doing.
332 > *
333 > * This utility is INCOMPATIBLE with `uri.toString()`-usages and both CANNOT be used interchanged.
334 > *
335 > * When dealing with uris from files or documents, `extUri` (the unbiased friend)is sufficient
336 > * because those uris come from a "trustworthy source". When creating unknown uris it's always
337 > * better to use `IUriIdentityService` which exposes an `IExtUri`-instance which knows when path
338 > * casing matters.
339 > */
340 > export const extUriBiasedIgnorePathCase = new ExtUri(uri => {
341 // A file scheme resource is in the same platform as code, so ignore case for non linux platforms
342 // Resource can be from another platform. Lowering the case as an hack. Should come from File system provider
343 return uri.scheme === Schemas.file ? !isLinux : true;
344 });
345 > resources.ts
346 >
347 > /**
348 > * BIASED utility that always ignores the casing of uris paths. ONLY use this util if you
349 > * understand what you are doing.
350 > *
351 > * This utility is INCOMPATIBLE with `uri.toString()`-usages and both CANNOT be used interchanged.
352 > *
353 > * When dealing with uris from files or documents, `extUri` (the unbiased friend)is sufficient
354 > * because those uris come from a "trustworthy source". When creating unknown uris it's always
355 > * better to use `IUriIdentityService` which exposes an `IExtUri`-instance which knows when path
356 > * casing matters.
357 > */
358 > export const extUriIgnorePathCase = new ExtUri(_ => true);
359 >
360 > export const isEqual = extUri.isEqual.bind(extUri);
361 > export const isEqualOrParent = extUri.isEqualOrParent.bind(extUri);
362 > export const getComparisonKey = extUri.getComparisonKey.bind(extUri);
363 > export const basenameOrAuthority = extUri.basenameOrAuthority.bind(extUri);
364 > export const basename = extUri.basename.bind(extUri);
365 > export const extname = extUri.extname.bind(extUri);
366 > export const dirname = extUri.dirname.bind(extUri);
367 > export const joinPath = extUri.joinPath.bind(extUri);
368 > export const normalizePath = extUri.normalizePath.bind(extUri);
369 > export const relativePath = extUri.relativePath.bind(extUri);
370 > export const resolvePath = extUri.resolvePath.bind(extUri);
371 > export const isAbsolutePath = extUri.isAbsolutePath.bind(extUri);
372 > export const isEqualAuthority = extUri.isEqualAuthority.bind(extUri);
373 > export const hasTrailingPathSeparator = extUri.hasTrailingPathSeparator.bind(extUri);
374 > export const removeTrailingPathSeparator = extUri.removeTrailingPathSeparator.bind(extUri);
375 > export const addTrailingPathSeparator = extUri.addTrailingPathSeparator.bind(extUri);
376 >
377 > //#endregion
378 >
379 > export function distinctParents<T>(items: T[], resourceAccessor: (item: T) => URI): T[] {
380 const distinctParents: T[] = [];
381 for (let i = 0; i < items.length; i++) {
396 return distinctParents;
397 }
398 > resources.ts
399 > /**
400 > * Data URI related helpers.
401 > */
402 > export namespace DataUri {
403 >
404 > export const META_DATA_LABEL = 'label';
405 > export const META_DATA_DESCRIPTION = 'description';
406 > export const META_DATA_SIZE = 'size';
407 > export const META_DATA_MIME = 'mime';
408 >
409 > export function parseMetaData(dataUri: URI): Map<string, string> {
410 const metadata = new Map<string, string>();
411
429 return metadata;
430 }
431 > } resources.ts
432 >
433 > export function toLocalResource(resource: URI, authority: string | undefined, localScheme: string): URI {
434 if (authority) {
435 let path = resource.path;
src/vs/editor/common/core/range.ts 255 covered LOC · 44 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- range.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IPosition, Position } from './position.js';
7 >
8 > /**
9 > * A range in the editor. This interface is suitable for serialization.
10 > */
11 > export interface IRange {
12 > /**
13 > * Line number on which the range starts (starts at 1).
14 > */
15 > readonly startLineNumber: number;
16 > /**
17 > * Column on which the range starts in line `startLineNumber` (starts at 1).
18 > */
19 > readonly startColumn: number;
20 > /**
21 > * Line number on which the range ends.
22 > */
23 > readonly endLineNumber: number;
24 > /**
25 > * Column on which the range ends in line `endLineNumber`.
26 > */
27 > readonly endColumn: number;
28 > }
29 >
30 > /**
31 > * A range in the editor. (startLineNumber,startColumn) is <= (endLineNumber,endColumn)
32 > */
33 > export class Range {
34 >
35 > /**
36 > * Line number on which the range starts (starts at 1).
37 > */
38 > public readonly startLineNumber: number;
39 > /**
40 > * Column on which the range starts in line `startLineNumber` (starts at 1).
41 > */
42 > public readonly startColumn: number;
43 > /**
44 > * Line number on which the range ends.
45 > */
46 > public readonly endLineNumber: number;
47 > /**
48 > * Column on which the range ends in line `endLineNumber`.
49 > */
50 > public readonly endColumn: number;
51 >
52 > constructor(startLineNumber: number, startColumn: number, endLineNumber: number, endColumn: number) {
53 > if ((startLineNumber > endLineNumber) || (startLineNumber === endLineNumber && startColumn > endColumn)) { range.ts
54 this.startLineNumber = endLineNumber;
55 this.startColumn = endColumn;
56 this.endLineNumber = startLineNumber;
57 this.endColumn = startColumn;
58 > } else { range.ts
59 > this.startLineNumber = startLineNumber; range.ts
60 > this.startColumn = startColumn;
61 > this.endLineNumber = endLineNumber;
62 > this.endColumn = endColumn;
63 > }
64 > } range.ts
65 > range.ts
66 > /**
67 > * Test if this range is empty.
68 > */
69 > public isEmpty(): boolean {
70 return Range.isEmpty(this);
71 }
72 > range.ts
73 > /**
74 > * Test if `range` is empty.
75 > */
76 > public static isEmpty(range: IRange): boolean {
77 return (range.startLineNumber === range.endLineNumber && range.startColumn === range.endColumn);
78 }
79 > range.ts
80 > /**
81 > * Test if position is in this range. If the position is at the edges, will return true.
82 > */
83 > public containsPosition(position: IPosition): boolean {
84 return Range.containsPosition(this, position);
85 }
86 > range.ts
87 > /**
88 > * Test if `position` is in `range`. If the position is at the edges, will return true.
89 > */
90 > public static containsPosition(range: IRange, position: IPosition): boolean {
91 if (position.lineNumber < range.startLineNumber || position.lineNumber > range.endLineNumber) {
92 return false;
100 return true;
101 }
102 > range.ts
103 > /**
104 > * Test if `position` is in `range`. If the position is at the edges, will return false.
105 > * @internal
106 > */
107 > public static strictContainsPosition(range: IRange, position: IPosition): boolean {
108 if (position.lineNumber < range.startLineNumber || position.lineNumber > range.endLineNumber) {
109 return false;
117 return true;
118 }
119 > range.ts
120 > /**
121 > * Test if range is in this range. If the range is equal to this range, will return true.
122 > */
123 > public containsRange(range: IRange): boolean {
124 return Range.containsRange(this, range);
125 }
126 > range.ts
127 > /**
128 > * Test if `otherRange` is in `range`. If the ranges are equal, will return true.
129 > */
130 > public static containsRange(range: IRange, otherRange: IRange): boolean {
131 if (otherRange.startLineNumber < range.startLineNumber || otherRange.endLineNumber < range.startLineNumber) {
132 return false;
143 return true;
144 }
145 > range.ts
146 > /**
147 > * Test if `range` is strictly in this range. `range` must start after and end before this range for the result to be true.
148 > */
149 > public strictContainsRange(range: IRange): boolean {
150 return Range.strictContainsRange(this, range);
151 }
152 > range.ts
153 > /**
154 > * Test if `otherRange` is strictly in `range` (must start after, and end before). If the ranges are equal, will return false.
155 > */
156 > public static strictContainsRange(range: IRange, otherRange: IRange): boolean {
157 if (otherRange.startLineNumber < range.startLineNumber || otherRange.endLineNumber < range.startLineNumber) {
158 return false;
169 return true;
170 }
171 > range.ts
172 > /**
173 > * A reunion of the two ranges.
174 > * The smallest position will be used as the start point, and the largest one as the end point.
175 > */
176 > public plusRange(range: IRange): Range {
177 return Range.plusRange(this, range);
178 }
179 > range.ts
180 > /**
181 > * A reunion of the two ranges.
182 > * The smallest position will be used as the start point, and the largest one as the end point.
183 > */
184 > public static plusRange(a: IRange, b: IRange): Range {
185 let startLineNumber: number;
186 let startColumn: number;
212 return new Range(startLineNumber, startColumn, endLineNumber, endColumn);
213 }
214 > range.ts
215 > /**
216 > * A intersection of the two ranges.
217 > */
218 > public intersectRanges(range: IRange): Range | null {
219 return Range.intersectRanges(this, range);
220 }
221 > range.ts
222 > /**
223 > * A intersection of the two ranges.
224 > */
225 > public static intersectRanges(a: IRange, b: IRange): Range | null {
226 let resultStartLineNumber = a.startLineNumber;
227 let resultStartColumn = a.startColumn;
256 return new Range(resultStartLineNumber, resultStartColumn, resultEndLineNumber, resultEndColumn);
257 }
258 > range.ts
259 > /**
260 > * Test if this range equals other.
261 > */
262 > public equalsRange(other: IRange | null | undefined): boolean {
263 return Range.equalsRange(this, other);
264 }
265 > range.ts
266 > /**
267 > * Test if range `a` equals `b`.
268 > */
269 > public static equalsRange(a: IRange | null | undefined, b: IRange | null | undefined): boolean {
270 if (!a && !b) {
271 return true;
280 );
281 }
282 > range.ts
283 > /**
284 > * Return the end position (which will be after or equal to the start position)
285 > */
286 > public getEndPosition(): Position {
287 return Range.getEndPosition(this);
288 }
289 > range.ts
290 > /**
291 > * Return the end position (which will be after or equal to the start position)
292 > */
293 > public static getEndPosition(range: IRange): Position {
294 return new Position(range.endLineNumber, range.endColumn);
295 }
296 > range.ts
297 > /**
298 > * Return the start position (which will be before or equal to the end position)
299 > */
300 > public getStartPosition(): Position {
301 return Range.getStartPosition(this);
302 }
303 > range.ts
304 > /**
305 > * Return the start position (which will be before or equal to the end position)
306 > */
307 > public static getStartPosition(range: IRange): Position {
308 return new Position(range.startLineNumber, range.startColumn);
309 }
310 > range.ts
311 > /**
312 > * Transform to a user presentable string representation.
313 > */
314 > public toString(): string {
315 return '[' + this.startLineNumber + ',' + this.startColumn + ' -> ' + this.endLineNumber + ',' + this.endColumn + ']';
316 }
317 > range.ts
318 > /**
319 > * Create a new range using this range's start position, and using endLineNumber and endColumn as the end position.
320 > */
321 > public setEndPosition(endLineNumber: number, endColumn: number): Range {
322 return new Range(this.startLineNumber, this.startColumn, endLineNumber, endColumn);
323 }
324 > range.ts
325 > /**
326 > * Create a new range using this range's end position, and using startLineNumber and startColumn as the start position.
327 > */
328 > public setStartPosition(startLineNumber: number, startColumn: number): Range {
329 return new Range(startLineNumber, startColumn, this.endLineNumber, this.endColumn);
330 }
331 > range.ts
332 > /**
333 > * Create a new empty range using this range's start position.
334 > */
335 > public collapseToStart(): Range {
336 return Range.collapseToStart(this);
337 }
338 > range.ts
339 > /**
340 > * Create a new empty range using this range's start position.
341 > */
342 > public static collapseToStart(range: IRange): Range {
343 return new Range(range.startLineNumber, range.startColumn, range.startLineNumber, range.startColumn);
344 }
345 > range.ts
346 > /**
347 > * Create a new empty range using this range's end position.
348 > */
349 > public collapseToEnd(): Range {
350 return Range.collapseToEnd(this);
351 }
352 > range.ts
353 > /**
354 > * Create a new empty range using this range's end position.
355 > */
356 > public static collapseToEnd(range: IRange): Range {
357 return new Range(range.endLineNumber, range.endColumn, range.endLineNumber, range.endColumn);
358 }
359 > range.ts
360 > /**
361 > * Moves the range by the given amount of lines.
362 > */
363 > public delta(lineCount: number): Range {
364 return new Range(this.startLineNumber + lineCount, this.startColumn, this.endLineNumber + lineCount, this.endColumn);
365 }
366 > range.ts
367 > /**
368 > * Test if this range starts and ends on the same line.
369 > */
370 > public isSingleLine(): boolean {
371 return this.startLineNumber === this.endLineNumber;
372 }
373 > range.ts
374 > // ---
375 >
376 > public static fromPositions(start: IPosition, end: IPosition = start): Range {
377 return new Range(start.lineNumber, start.column, end.lineNumber, end.column);
378 }
379 > range.ts
380 > /**
381 > * Create a `Range` from an `IRange`.
382 > */
383 > public static lift(range: undefined | null): null;
384 > public static lift(range: IRange): Range;
385 > public static lift(range: IRange | undefined | null): Range | null;
386 > public static lift(range: IRange | undefined | null): Range | null {
387 if (!range) {
388 return null;
390 return new Range(range.startLineNumber, range.startColumn, range.endLineNumber, range.endColumn);
391 }
392 > range.ts
393 > /**
394 > * Test if `obj` is an `IRange`.
395 > */
396 > public static isIRange(obj: unknown): obj is IRange {
397 return (
398 !!obj
403 );
404 }
405 > range.ts
406 > /**
407 > * Test if the two ranges are touching in any way.
408 > */
409 > public static areIntersectingOrTouching(a: IRange, b: IRange): boolean {
410 // Check if `a` is before `b`
411 if (a.endLineNumber < b.startLineNumber || (a.endLineNumber === b.startLineNumber && a.endColumn < b.startColumn)) {
421 return true;
422 }
423 > range.ts
424 > /**
425 > * Test if the two ranges are intersecting. If the ranges are touching it returns true.
426 > */
427 > public static areIntersecting(a: IRange, b: IRange): boolean {
428 // Check if `a` is before `b`
429 if (a.endLineNumber < b.startLineNumber || (a.endLineNumber === b.startLineNumber && a.endColumn <= b.startColumn)) {
439 return true;
440 }
441 > range.ts
442 > /**
443 > * Test if the two ranges are intersecting, but not touching at all.
444 > */
445 > public static areOnlyIntersecting(a: IRange, b: IRange): boolean {
446 // Check if `a` is before `b`
447 if (a.endLineNumber < (b.startLineNumber - 1) || (a.endLineNumber === b.startLineNumber && a.endColumn < (b.startColumn - 1))) {
457 return true;
458 }
459 > range.ts
460 > /**
461 > * A function that compares ranges, useful for sorting ranges
462 > * It will first compare ranges on the startPosition and then on the endPosition
463 > */
464 > public static compareRangesUsingStarts(a: IRange | null | undefined, b: IRange | null | undefined): number {
465 if (a && b) {
466 const aStartLineNumber = a.startLineNumber | 0;
490 return aExists - bExists;
491 }
492 > range.ts
493 > /**
494 > * A function that compares ranges, useful for sorting ranges
495 > * It will first compare ranges on the endPosition and then on the startPosition
496 > */
497 > public static compareRangesUsingEnds(a: IRange, b: IRange): number {
498 if (a.endLineNumber === b.endLineNumber) {
499 if (a.endColumn === b.endColumn) {
507 return a.endLineNumber - b.endLineNumber;
508 }
509 > range.ts
510 > /**
511 > * Test if the range spans multiple lines.
512 > */
513 > public static spansMultipleLines(range: IRange): boolean {
514 return range.endLineNumber > range.startLineNumber;
515 }
516 > range.ts
517 > public toJSON(): IRange {
518 return this;
519 }
520 > } range.ts
src/vs/workbench/contrib/chat/common/promptSyntax/config/promptFileLocations.ts 252 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptFileLocations.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../../base/common/uri.js';
7 > import { basename, dirname } from '../../../../../../base/common/resources.js';
8 > import { PromptFileSource, PromptsType } from '../promptTypes.js';
9 > import { PromptsStorage } from '../service/promptsService.js';
10 > import { compareIgnoreCase } from '../../../../../../base/common/strings.js';
11 >
12 > /**
13 > * File extension for the reusable prompt files.
14 > */
15 > export const PROMPT_FILE_EXTENSION = '.prompt.md';
16 >
17 > /**
18 > * File extension for the reusable instruction files.
19 > */
20 > export const INSTRUCTION_FILE_EXTENSION = '.instructions.md';
21 >
22 > /**
23 > * File extension for the modes files.
24 > */
25 > export const LEGACY_MODE_FILE_EXTENSION = '.chatmode.md';
26 >
27 > /**
28 > * File extension for the agent files.
29 > */
30 > export const AGENT_FILE_EXTENSION = '.agent.md';
31 >
32 > /**
33 > * Skill file name (case insensitive).
34 > */
35 > export const SKILL_FILENAME = 'SKILL.md';
36 >
37 > /**
38 > * Check if a filename is a skill file (case insensitive).
39 > */
40 > export function isSkillFilename(filename: string): boolean {
41 return compareIgnoreCase(filename, SKILL_FILENAME) === 0;
42 }
44 > /**
45 > * Regex for valid skill names: lowercase alphanumeric and hyphens only.
46 > */
47 > export const VALID_SKILL_NAME_REGEX = /^[a-z0-9-]+$/;
48 >
49 > /**
50 > * AGENT file name
51 > */
52 > export const AGENT_MD_FILENAME = 'AGENTS.md';
53 >
54 > /**
55 > * Claude file name.
56 > */
57 > export const CLAUDE_MD_FILENAME = 'CLAUDE.md';
58 >
59 > /**
60 > * Claude local file name.
61 > */
62 > export const CLAUDE_LOCAL_MD_FILENAME = 'CLAUDE.local.md';
63 >
64 > /**
65 > * Claude configuration folder name.
66 > */
67 > export const CLAUDE_CONFIG_FOLDER = '.claude';
68 >
69 > /**
70 > * Copilot configuration folder name.
71 > */
72 > export const COPILOT_CONFIG_FOLDER = '.copilot';
73 >
74 > /**
75 > * Copilot custom instructions file name.
76 > */
77 > export const COPILOT_CUSTOM_INSTRUCTIONS_FILENAME = 'copilot-instructions.md';
78 >
79 > /**
80 > * GitHub configuration folder name.
81 > */
82 > export const GITHUB_CONFIG_FOLDER = '.github';
83 >
84 > /**
85 > * Default reusable prompt files source folder.
86 > */
87 > export const PROMPT_DEFAULT_SOURCE_FOLDER = '.github/prompts';
88 >
89 > /**
90 > * Default reusable instructions files source folder.
91 > */
92 > export const INSTRUCTIONS_DEFAULT_SOURCE_FOLDER = '.github/instructions';
93 >
94 > /**
95 > * Default modes source folder.
96 > */
97 > export const LEGACY_MODE_DEFAULT_SOURCE_FOLDER = '.github/chatmodes';
98 >
99 > /**
100 > * Agents folder.
101 > */
102 > export const AGENTS_SOURCE_FOLDER = '.github/agents';
103 >
104 > /**
105 > * Claude agents folder.
106 > */
107 > export const CLAUDE_AGENTS_SOURCE_FOLDER = '.claude/agents';
108 >
109 > /**
110 > * Copilot user agents folder.
111 > */
112 > export const COPILOT_USER_AGENTS_SOURCE_FOLDER = '~/.copilot/agents';
113 >
114 > /**
115 > * Claude rules folder.
116 > */
117 > export const CLAUDE_RULES_SOURCE_FOLDER = '.claude/rules';
118 >
119 > /**
120 > * Hooks folder.
121 > */
122 > export const HOOKS_SOURCE_FOLDER = '.github/hooks';
123 >
124 > /**
125 > * Subset of {@link PromptFileSource} values that can appear on folder-based
126 > * prompt source configurations (excludes extension/plugin-only sources).
127 > */
128 > export type PromptFolderSource = Exclude<PromptFileSource, PromptFileSource.ExtensionContribution | PromptFileSource.ExtensionAPI | PromptFileSource.Plugin>;
129 >
130 > /**
131 > * Prompt source folder path with source and storage type.
132 > */
133 > export interface IPromptSourceFolder {
134 > readonly path: string;
135 > readonly source: PromptFolderSource;
136 > readonly storage: PromptsStorage.local | PromptsStorage.user;
137 > }
138 >
139 > /**
140 > * Resolved prompt folder with source and storage type.
141 > */
142 > export interface IResolvedPromptSourceFolder {
143 > readonly uri: URI;
144 > readonly searchRoot: URI; // matches the URI when no glob pattern is used
145 > readonly filePattern: string | undefined; // the part of the path with the glob pattern, or undefined if no glob pattern is used
146 > readonly source: PromptFolderSource;
147 > readonly storage: PromptsStorage.local | PromptsStorage.user;
148 > /**
149 > * The original path string before resolution (e.g., '~/.copilot/agents' or '.github/agents').
150 > * Used for display purposes.
151 > */
152 > readonly displayPath?: string;
153 > /**
154 > * Whether this is a default location (vs user-configured).
155 > */
156 > readonly isDefault?: boolean;
157 > }
158 >
159 > /**
160 > * All default skill source folders (both workspace and user home).
161 > */
162 > export const DEFAULT_SKILL_SOURCE_FOLDERS: readonly IPromptSourceFolder[] = [
163 > { path: '.agents/skills', source: PromptFileSource.AgentsWorkspace, storage: PromptsStorage.local },
164 > { path: '.github/skills', source: PromptFileSource.GitHubWorkspace, storage: PromptsStorage.local },
165 > { path: '.claude/skills', source: PromptFileSource.ClaudeWorkspace, storage: PromptsStorage.local },
166 > { path: '~/.agents/skills', source: PromptFileSource.AgentsPersonal, storage: PromptsStorage.user },
167 > { path: '~/.copilot/skills', source: PromptFileSource.CopilotPersonal, storage: PromptsStorage.user },
168 > { path: '~/.claude/skills', source: PromptFileSource.ClaudePersonal, storage: PromptsStorage.user },
169 > ];
170 >
171 > /**
172 > * Default instructions source folders.
173 > */
174 > export const DEFAULT_INSTRUCTIONS_SOURCE_FOLDERS: readonly IPromptSourceFolder[] = [
175 > { path: INSTRUCTIONS_DEFAULT_SOURCE_FOLDER, source: PromptFileSource.GitHubWorkspace, storage: PromptsStorage.local },
176 > { path: CLAUDE_RULES_SOURCE_FOLDER, source: PromptFileSource.ClaudeWorkspace, storage: PromptsStorage.local },
177 > { path: '~/.copilot/instructions', source: PromptFileSource.CopilotPersonal, storage: PromptsStorage.user },
178 > { path: '~/' + CLAUDE_RULES_SOURCE_FOLDER, source: PromptFileSource.ClaudePersonal, storage: PromptsStorage.user },
179 > ];
180 >
181 > /**
182 > * Default prompt source folders.
183 > */
184 > export const DEFAULT_PROMPT_SOURCE_FOLDERS: readonly IPromptSourceFolder[] = [
185 > { path: PROMPT_DEFAULT_SOURCE_FOLDER, source: PromptFileSource.GitHubWorkspace, storage: PromptsStorage.local },
186 > ];
187 >
188 > /**
189 > * Default agent source folders.
190 > */
191 > export const DEFAULT_AGENT_SOURCE_FOLDERS: readonly IPromptSourceFolder[] = [
192 > { path: AGENTS_SOURCE_FOLDER, source: PromptFileSource.GitHubWorkspace, storage: PromptsStorage.local },
193 > { path: CLAUDE_AGENTS_SOURCE_FOLDER, source: PromptFileSource.ClaudeWorkspace, storage: PromptsStorage.local },
194 > { path: COPILOT_USER_AGENTS_SOURCE_FOLDER, source: PromptFileSource.CopilotPersonal, storage: PromptsStorage.user },
195 > { path: '~/' + CLAUDE_AGENTS_SOURCE_FOLDER, source: PromptFileSource.ClaudePersonal, storage: PromptsStorage.user },
196 > ];
197 >
198 > /**
199 > * Default hook file paths.
200 > * Entries can be either a directory or a specific file path (.json)
201 > */
202 > export const DEFAULT_HOOK_FILE_PATHS: readonly IPromptSourceFolder[] = [
203 > { path: '.github/hooks', source: PromptFileSource.GitHubWorkspace, storage: PromptsStorage.local },
204 > { path: '.claude/settings.local.json', source: PromptFileSource.ClaudeWorkspaceLocal, storage: PromptsStorage.local },
205 > { path: '.claude/settings.json', source: PromptFileSource.ClaudeWorkspace, storage: PromptsStorage.local },
206 > { path: '~/.copilot/hooks', source: PromptFileSource.CopilotPersonal, storage: PromptsStorage.user },
207 > { path: '~/.claude/settings.json', source: PromptFileSource.ClaudePersonal, storage: PromptsStorage.user },
208 > ];
209 >
210 > /**
211 > * Helper function to check if a file is directly in the .github/agents/ folder (not in subfolders).
212 > */
213 function isInAgentsFolder(fileUri: URI): boolean {
214 const dir = dirname(fileUri).path;
215 return dir.endsWith('/' + AGENTS_SOURCE_FOLDER) || dir.endsWith('/' + CLAUDE_AGENTS_SOURCE_FOLDER) || isInCopilotAgentsFolder(fileUri);
216 }
218 > /**
219 > * Helper function to check if a file is directly in the .claude/agents/ folder.
220 > */
221 > export function isInClaudeAgentsFolder(fileUri: URI): boolean {
222 const dir = dirname(fileUri).path;
223 return dir.endsWith('/' + CLAUDE_AGENTS_SOURCE_FOLDER);
224 }
226 > /**
227 > * Helper function to check if a file is directly in the ~/.copilot/agents/ folder.
228 > */
229 > export function isInCopilotAgentsFolder(fileUri: URI): boolean {
230 const dir = dirname(fileUri).path;
231 return dir.endsWith(COPILOT_USER_AGENTS_SOURCE_FOLDER.substring(1));
232 }
234 > /**
235 > * Helper function to check if a file is inside the .claude/rules/ folder (including subfolders).
236 > * Claude rules files (.md) in this folder are treated as instruction files.
237 > */
238 > export function isInClaudeRulesFolder(fileUri: URI): boolean {
239 const path = fileUri.path;
240 return path.includes('/' + CLAUDE_RULES_SOURCE_FOLDER + '/');
241 }
243 > /**
244 > * Gets the prompt file type from the provided path.
245 > *
246 > * Note: This function assumes the URI is already known to be a prompt file
247 > * (e.g., from a configured prompt source folder). It does not validate that
248 > * arbitrary URIs are prompt files - for example, any .json file will return
249 > * PromptsType.hook regardless of its location.
250 > */
251 > export function getPromptFileType(fileUri: URI): PromptsType | undefined {
252 const filename = basename(fileUri);
253
288 return undefined;
289 }
291 > /**
292 > * Check if provided URI points to a file that with prompt file extension.
293 > */
294 > export function isPromptOrInstructionsFile(fileUri: URI): boolean {
295 return getPromptFileType(fileUri) !== undefined;
296 }
298 > export function getPromptFileExtension(type: PromptsType): string {
299 switch (type) {
300 case PromptsType.instructions:
312 }
313 }
315 > export function getPromptFileDefaultLocations(type: PromptsType): readonly IPromptSourceFolder[] {
316 switch (type) {
317 case PromptsType.instructions:
329 }
330 }
332 > export function getSkillFolderName(fileUri: URI): string {
333 return basename(dirname(fileUri));
334 }
336 > /**
337 > * Gets clean prompt name without file extension.
338 > */
339 > export function getCleanPromptName(fileUri: URI): string {
340 const fileName = basename(fileUri);
341
src/vs/workbench/common/contextkeys.ts 250 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- contextkeys.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DisposableStore } from '../../base/common/lifecycle.js';
7 > import { URI } from '../../base/common/uri.js';
8 > import { localize } from '../../nls.js';
9 > import { IContextKeyService, IContextKey, RawContextKey } from '../../platform/contextkey/common/contextkey.js';
10 > import { basename, dirname, extname, isEqual } from '../../base/common/resources.js';
11 > import { ILanguageService } from '../../editor/common/languages/language.js';
12 > import { IFileService } from '../../platform/files/common/files.js';
13 > import { IModelService } from '../../editor/common/services/model.js';
14 > import { Schemas } from '../../base/common/network.js';
15 > import { EditorInput } from './editor/editorInput.js';
16 > import { IEditorResolverService } from '../services/editor/common/editorResolverService.js';
17 > import { DEFAULT_EDITOR_ASSOCIATION, isDiffEditorInput } from './editor.js';
18 >
19 > //#region < --- Workbench --- >
20 >
21 > export const WorkbenchStateContext = new RawContextKey<string>('workbenchState', undefined, { type: 'string', description: localize('workbenchState', "The kind of workspace opened in the window, either 'empty' (no workspace), 'folder' (single folder) or 'workspace' (multi-root workspace)") });
22 > export const WorkspaceFolderCountContext = new RawContextKey<number>('workspaceFolderCount', 0, localize('workspaceFolderCount', "The number of root folders in the workspace"));
23 >
24 > export const OpenFolderWorkspaceSupportContext = new RawContextKey<boolean>('openFolderWorkspaceSupport', true, true);
25 > export const EnterMultiRootWorkspaceSupportContext = new RawContextKey<boolean>('enterMultiRootWorkspaceSupport', true, true);
26 > export const EmptyWorkspaceSupportContext = new RawContextKey<boolean>('emptyWorkspaceSupport', true, true);
27 >
28 > export const DirtyWorkingCopiesContext = new RawContextKey<boolean>('dirtyWorkingCopies', false, localize('dirtyWorkingCopies', "Whether there are any working copies with unsaved changes"));
29 >
30 > export const RemoteNameContext = new RawContextKey<string>('remoteName', '', localize('remoteName', "The name of the remote the window is connected to or an empty string if not connected to any remote"));
31 >
32 > export const VirtualWorkspaceContext = new RawContextKey<string>('virtualWorkspace', '', localize('virtualWorkspace', "The scheme of the current workspace is from a virtual file system or an empty string."));
33 > export const TemporaryWorkspaceContext = new RawContextKey<boolean>('temporaryWorkspace', false, localize('temporaryWorkspace', "The scheme of the current workspace is from a temporary file system."));
34 >
35 > export const IsSessionsWindowContext = new RawContextKey<boolean>('isSessionsWindow', false, localize('isSessionsWindow', "Whether the current window is a agent sessions window."));
36 >
37 > export const HasWebFileSystemAccess = new RawContextKey<boolean>('hasWebFileSystemAccess', false, true); // Support for FileSystemAccess web APIs (https://wicg.github.io/file-system-access)
38 >
39 > export const EmbedderIdentifierContext = new RawContextKey<string | undefined>('embedderIdentifier', undefined, localize('embedderIdentifier', 'The identifier of the embedder according to the product service, if one is defined'));
40 >
41 > export const InAutomationContext = new RawContextKey<boolean>('inAutomation', false, localize('inAutomation', "Whether VS Code is running under automation/smoke test"));
42 >
43 > //#endregion
44 >
45 > //#region < --- Window --- >
46 >
47 > export const IsMainWindowFullscreenContext = new RawContextKey<boolean>('isFullscreen', false, localize('isFullscreen', "Whether the main window is in fullscreen mode"));
48 > export const IsAuxiliaryWindowFocusedContext = new RawContextKey<boolean>('isAuxiliaryWindowFocusedContext', false, localize('isAuxiliaryWindowFocusedContext', "Whether an auxiliary window is focused"));
49 >
50 > export const IsWindowAlwaysOnTopContext = new RawContextKey<boolean>('isWindowAlwaysOnTop', false, localize('isWindowAlwaysOnTop', "Whether the window is always on top"));
51 >
52 > export const IsAuxiliaryWindowContext = new RawContextKey<boolean>('isAuxiliaryWindow', false, localize('isAuxiliaryWindow', "Window is an auxiliary window"));
53 >
54 >
55 > //#endregion
56 >
57 >
58 > //#region < --- Editor --- >
59 >
60 > // Editor State Context Keys
61 > export const ActiveEditorDirtyContext = new RawContextKey<boolean>('activeEditorIsDirty', false, localize('activeEditorIsDirty', "Whether the active editor has unsaved changes"));
62 > export const ActiveEditorPinnedContext = new RawContextKey<boolean>('activeEditorIsNotPreview', false, localize('activeEditorIsNotPreview', "Whether the active editor is not in preview mode"));
63 > export const ActiveEditorFirstInGroupContext = new RawContextKey<boolean>('activeEditorIsFirstInGroup', false, localize('activeEditorIsFirstInGroup', "Whether the active editor is the first one in its group"));
64 > export const ActiveEditorLastInGroupContext = new RawContextKey<boolean>('activeEditorIsLastInGroup', false, localize('activeEditorIsLastInGroup', "Whether the active editor is the last one in its group"));
65 > export const ActiveEditorStickyContext = new RawContextKey<boolean>('activeEditorIsPinned', false, localize('activeEditorIsPinned', "Whether the active editor is pinned"));
66 > export const ActiveEditorReadonlyContext = new RawContextKey<boolean>('activeEditorIsReadonly', false, localize('activeEditorIsReadonly', "Whether the active editor is read-only"));
67 > export const ActiveCompareEditorCanSwapContext = new RawContextKey<boolean>('activeCompareEditorCanSwap', false, localize('activeCompareEditorCanSwap', "Whether the active compare editor can swap sides"));
68 > export const ActiveEditorCanToggleReadonlyContext = new RawContextKey<boolean>('activeEditorCanToggleReadonly', true, localize('activeEditorCanToggleReadonly', "Whether the active editor can toggle between being read-only or writeable"));
69 > export const ActiveEditorCanRevertContext = new RawContextKey<boolean>('activeEditorCanRevert', false, localize('activeEditorCanRevert', "Whether the active editor can revert"));
70 > export const ActiveEditorCanSplitInGroupContext = new RawContextKey<boolean>('activeEditorCanSplitInGroup', true);
71 >
72 > // Editor Kind Context Keys
73 > export const ActiveEditorContext = new RawContextKey<string | null>('activeEditor', null, { type: 'string', description: localize('activeEditor', "The identifier of the active editor") });
74 > export const ActiveEditorAvailableEditorIdsContext = new RawContextKey<string>('activeEditorAvailableEditorIds', '', localize('activeEditorAvailableEditorIds', "The available editor identifiers that are usable for the active editor"));
75 > export const TextCompareEditorVisibleContext = new RawContextKey<boolean>('textCompareEditorVisible', false, localize('textCompareEditorVisible', "Whether a text compare editor is visible"));
76 > export const TextCompareEditorActiveContext = new RawContextKey<boolean>('textCompareEditorActive', false, localize('textCompareEditorActive', "Whether a text compare editor is active"));
77 > export const SideBySideEditorActiveContext = new RawContextKey<boolean>('sideBySideEditorActive', false, localize('sideBySideEditorActive', "Whether a side by side editor is active"));
78 > export const ActiveCustomEditorDiffCanToggleLayoutContext = new RawContextKey<boolean>('activeCustomEditorDiffCanToggleLayout', false, localize('activeCustomEditorDiffCanToggleLayout', "Whether the active custom editor diff can toggle between inline and side by side layout"));
79 > export const ActiveCustomEditorTextDiffContext = new RawContextKey<boolean>('activeCustomEditorTextDiff', false, localize('activeCustomEditorTextDiff', "Whether the active custom editor diff is backed by text documents"));
80 >
81 > // Editor Group Context Keys
82 > export const EditorGroupEditorsCountContext = new RawContextKey<number>('groupEditorsCount', 0, localize('groupEditorsCount', "The number of opened editor groups"));
83 > export const IsTopRightEditorGroupContext = new RawContextKey<boolean>('isTopRightEditorGroup', false, localize('isTopRightEditorGroup', "Whether the editor group is the top right editor group in the editor part"));
84 > export const ActiveEditorGroupEmptyContext = new RawContextKey<boolean>('activeEditorGroupEmpty', false, localize('activeEditorGroupEmpty', "Whether the active editor group is empty"));
85 > export const ActiveEditorGroupIndexContext = new RawContextKey<number>('activeEditorGroupIndex', 0, localize('activeEditorGroupIndex', "The index of the active editor group"));
86 > export const ActiveEditorGroupLastContext = new RawContextKey<boolean>('activeEditorGroupLast', false, localize('activeEditorGroupLast', "Whether the active editor group is the last group"));
87 > export const ActiveEditorGroupLockedContext = new RawContextKey<boolean>('activeEditorGroupLocked', false, localize('activeEditorGroupLocked', "Whether the active editor group is locked"));
88 > export const MultipleEditorGroupsContext = new RawContextKey<boolean>('multipleEditorGroups', false, localize('multipleEditorGroups', "Whether there are multiple editor groups opened"));
89 > export const SingleEditorGroupsContext = MultipleEditorGroupsContext.toNegated();
90 > export const MultipleEditorsSelectedInGroupContext = new RawContextKey<boolean>('multipleEditorsSelectedInGroup', false, localize('multipleEditorsSelectedInGroup', "Whether multiple editors have been selected in an editor group"));
91 > export const TwoEditorsSelectedInGroupContext = new RawContextKey<boolean>('twoEditorsSelectedInGroup', false, localize('twoEditorsSelectedInGroup', "Whether exactly two editors have been selected in an editor group"));
92 > export const SelectedEditorsInGroupFileOrUntitledResourceContextKey = new RawContextKey<boolean>('SelectedEditorsInGroupFileOrUntitledResourceContextKey', true, localize('SelectedEditorsInGroupFileOrUntitledResourceContextKey', "Whether all selected editors in a group have a file or untitled resource associated"));
93 >
94 > // Editor Part Context Keys
95 > export const EditorPartMultipleEditorGroupsContext = new RawContextKey<boolean>('editorPartMultipleEditorGroups', false, localize('editorPartMultipleEditorGroups', "Whether there are multiple editor groups opened in an editor part"));
96 > export const EditorPartSingleEditorGroupsContext = EditorPartMultipleEditorGroupsContext.toNegated();
97 > export const EditorPartMaximizedEditorGroupContext = new RawContextKey<boolean>('editorPartMaximizedEditorGroup', false, localize('editorPartEditorGroupMaximized', "Editor Part has a maximized group"));
98 >
99 > export const EditorPartModalContext = new RawContextKey<boolean>('editorPartModal', false, localize('editorPartModal', "Whether focus is in a modal editor part"));
100 > export const EditorPartModalVisibleContext = new RawContextKey<boolean>('editorPartModalVisible', false, localize('editorPartModalVisible', "Whether a modal editor part is visible"));
101 > export const EditorPartModalMaximizedContext = new RawContextKey<boolean>('editorPartModalMaximized', false, localize('editorPartModalMaximized', "Whether the modal editor part is maximized"));
102 > export const EditorPartModalNavigationContext = new RawContextKey<boolean>('editorPartModalNavigation', false, localize('editorPartModalNavigation', "Whether the modal editor part has navigation context"));
103 > export const EditorPartModalSidebarContext = new RawContextKey<boolean>('editorPartModalSidebar', false, localize('editorPartModalSidebar', "Whether the modal editor part has a sidebar"));
104 > export const EditorPartModalSidebarVisibleContext = new RawContextKey<boolean>('editorPartModalSidebarVisible', false, localize('editorPartModalSidebarVisible', "Whether the modal editor part sidebar is visible"));
105 >
106 > // Editor Layout Context Keys
107 > export const EditorsVisibleContext = new RawContextKey<boolean>('editorIsOpen', false, localize('editorIsOpen', "Whether an editor is open"));
108 > export const EditorAreaFocusContext = new RawContextKey<boolean>('editorAreaFocus', false, localize('editorAreaFocus', "Whether the editor area (any editor part) has keyboard focus"));
109 > export const InEditorZenModeContext = new RawContextKey<boolean>('inZenMode', false, localize('inZenMode', "Whether Zen mode is enabled"));
110 > export const IsMainEditorCenteredLayoutContext = new RawContextKey<boolean>('isCenteredLayout', false, localize('isMainEditorCenteredLayout', "Whether centered layout is enabled for the main editor"));
111 > export const SplitEditorsVertically = new RawContextKey<boolean>('splitEditorsVertically', false, localize('splitEditorsVertically', "Whether editors split vertically"));
112 > export const MainEditorAreaVisibleContext = new RawContextKey<boolean>('mainEditorAreaVisible', true, localize('mainEditorAreaVisible', "Whether the editor area in the main window is visible"));
113 > export const EditorTabsVisibleContext = new RawContextKey<boolean>('editorTabsVisible', true, localize('editorTabsVisible', "Whether editor tabs are visible"));
114 >
115 > //#endregion
116 >
117 >
118 > //#region < --- Side Bar --- >
119 >
120 > export const SideBarVisibleContext = new RawContextKey<boolean>('sideBarVisible', false, localize('sideBarVisible', "Whether the sidebar is visible"));
121 > export const SidebarFocusContext = new RawContextKey<boolean>('sideBarFocus', false, localize('sideBarFocus', "Whether the sidebar has keyboard focus"));
122 > export const ActiveViewletContext = new RawContextKey<string>('activeViewlet', '', localize('activeViewlet', "The identifier of the active viewlet"));
123 >
124 > //#endregion
125 >
126 >
127 > //#region < --- Status Bar --- >
128 >
129 > export const StatusBarFocused = new RawContextKey<boolean>('statusBarFocused', false, localize('statusBarFocused', "Whether the status bar has keyboard focus"));
130 >
131 > //#endregion
132 >
133 > //#region < --- Title Bar --- >
134 >
135 > export const TitleBarStyleContext = new RawContextKey<string>('titleBarStyle', 'custom', localize('titleBarStyle', "Style of the window title bar"));
136 > export const TitleBarVisibleContext = new RawContextKey<boolean>('titleBarVisible', false, localize('titleBarVisible', "Whether the title bar is visible"));
137 > export const IsCompactTitleBarContext = new RawContextKey<boolean>('isCompactTitleBar', false, localize('isCompactTitleBar', "Title bar is in compact mode"));
138 >
139 > //#endregion
140 >
141 >
142 > //#region < --- Banner --- >
143 >
144 > export const BannerFocused = new RawContextKey<boolean>('bannerFocused', false, localize('bannerFocused', "Whether the banner has keyboard focus"));
145 >
146 > //#endregion
147 >
148 >
149 > //#region < --- Notifications --- >
150 >
151 > export const NotificationFocusedContext = new RawContextKey<boolean>('notificationFocus', true, localize('notificationFocus', "Whether a notification has keyboard focus"));
152 > export const NotificationsCenterVisibleContext = new RawContextKey<boolean>('notificationCenterVisible', false, localize('notificationCenterVisible', "Whether the notifications center is visible"));
153 > export const NotificationsToastsVisibleContext = new RawContextKey<boolean>('notificationToastsVisible', false, localize('notificationToastsVisible', "Whether a notification toast is visible"));
154 >
155 > //#endregion
156 >
157 >
158 > //#region < --- Auxiliary Bar --- >
159 >
160 > export const ActiveAuxiliaryContext = new RawContextKey<string>('activeAuxiliary', '', localize('activeAuxiliary', "The identifier of the active auxiliary panel"));
161 > export const AuxiliaryBarFocusContext = new RawContextKey<boolean>('auxiliaryBarFocus', false, localize('auxiliaryBarFocus', "Whether the auxiliary bar has keyboard focus"));
162 > export const AuxiliaryBarVisibleContext = new RawContextKey<boolean>('auxiliaryBarVisible', false, localize('auxiliaryBarVisible', "Whether the auxiliary bar is visible"));
163 > export const SecondarySideBarVisibleContext = new RawContextKey<boolean>('secondarySideBarVisible', false, localize('secondarySideBarVisible', "Whether the layout surface representing the secondary side bar is visible"));
164 > export const AuxiliaryBarMaximizedContext = new RawContextKey<boolean>('auxiliaryBarMaximized', false, localize('auxiliaryBarMaximized', "Whether the auxiliary bar is maximized"));
165 >
166 > //#endregion
167 >
168 >
169 > //#region < --- Panel --- >
170 >
171 > export const ActivePanelContext = new RawContextKey<string>('activePanel', '', localize('activePanel', "The identifier of the active panel"));
172 > export const PanelFocusContext = new RawContextKey<boolean>('panelFocus', false, localize('panelFocus', "Whether the panel has keyboard focus"));
173 > export const PanelPositionContext = new RawContextKey<string>('panelPosition', 'bottom', localize('panelPosition', "The position of the panel, always 'bottom'"));
174 > export const PanelAlignmentContext = new RawContextKey<string>('panelAlignment', 'center', localize('panelAlignment', "The alignment of the panel, either 'center', 'left', 'right' or 'justify'"));
175 > export const PanelVisibleContext = new RawContextKey<boolean>('panelVisible', false, localize('panelVisible', "Whether the panel is visible"));
176 > export const PanelMaximizedContext = new RawContextKey<boolean>('panelMaximized', false, localize('panelMaximized', "Whether the panel is maximized"));
177 >
178 > //#endregion
179 >
180 >
181 > //#region < --- Views --- >
182 >
183 > export const FocusedViewContext = new RawContextKey<string>('focusedView', '', localize('focusedView', "The identifier of the view that has keyboard focus"));
184 > export function getVisbileViewContextKey(viewId: string): string { return `view.${viewId}.visible`; }
185 >
186 > //#endregion
187 >
188 >
189 > //#region < --- Resources --- >
190 >
191 > abstract class AbstractResourceContextKey {
192 >
193 > // NOTE: DO NOT CHANGE THE DEFAULT VALUE TO ANYTHING BUT
194 > // UNDEFINED! IT IS IMPORTANT THAT DEFAULTS ARE INHERITED
195 > // FROM THE PARENT CONTEXT AND ONLY UNDEFINED DOES THIS
196 >
197 > static readonly Scheme = new RawContextKey<string>('resourceScheme', undefined, { type: 'string', description: localize('resourceScheme', "The scheme of the resource") });
198 > static readonly Filename = new RawContextKey<string>('resourceFilename', undefined, { type: 'string', description: localize('resourceFilename', "The file name of the resource") });
199 > static readonly Dirname = new RawContextKey<string>('resourceDirname', undefined, { type: 'string', description: localize('resourceDirname', "The folder name the resource is contained in") });
200 > static readonly Path = new RawContextKey<string>('resourcePath', undefined, { type: 'string', description: localize('resourcePath', "The full path of the resource") });
201 > static readonly LangId = new RawContextKey<string>('resourceLangId', undefined, { type: 'string', description: localize('resourceLangId', "The language identifier of the resource") });
202 > static readonly Resource = new RawContextKey<string>('resource', undefined, { type: 'URI', description: localize('resource', "The full value of the resource including scheme and path") });
203 > static readonly Extension = new RawContextKey<string>('resourceExtname', undefined, { type: 'string', description: localize('resourceExtname', "The extension name of the resource") });
204 > static readonly HasResource = new RawContextKey<boolean>('resourceSet', undefined, { type: 'boolean', description: localize('resourceSet', "Whether a resource is present or not") });
205 > static readonly IsFileSystemResource = new RawContextKey<boolean>('isFileSystemResource', undefined, { type: 'boolean', description: localize('isFileSystemResource', "Whether the resource is backed by a file system provider") });
206 >
207 > protected _value: URI | undefined;
208 > protected readonly _resourceKey: IContextKey<string | null>;
209 > protected readonly _schemeKey: IContextKey<string | null>;
210 > protected readonly _filenameKey: IContextKey<string | null>;
211 > protected readonly _dirnameKey: IContextKey<string | null>;
212 > protected readonly _pathKey: IContextKey<string | null>;
213 > protected readonly _langIdKey: IContextKey<string | null>;
214 > protected readonly _extensionKey: IContextKey<string | null>;
215 > protected readonly _hasResource: IContextKey<boolean>;
216 > protected readonly _isFileSystemResource: IContextKey<boolean>;
217 >
218 > constructor(
219 @IContextKeyService protected readonly _contextKeyService: IContextKeyService,
220 @IFileService protected readonly _fileService: IFileService,
232 this._isFileSystemResource = AbstractResourceContextKey.IsFileSystemResource.bindTo(this._contextKeyService);
233 }
235 > protected _setLangId(): void {
236 const value = this.get();
237 if (!value) {
242 this._langIdKey.set(langId);
243 }
245 > set(value: URI | null | undefined) {
246 value = value ?? undefined;
247 if (isEqual(this._value, value)) {
261 });
262 }
264 > protected uriToPath(uri: URI): string {
265 if (uri.scheme === Schemas.file) {
266 return uri.fsPath;
268 return uri.path;
269 }
271 > reset(): void {
272 this._value = undefined;
273 this._contextKeyService.bufferChangeEvents(() => {
283 });
284 }
286 > get(): URI | undefined {
287 return this._value;
288 }
289 > } contextkeys.ts
290 >
291 > export class ResourceContextKey extends AbstractResourceContextKey {
292 >
293 > private readonly _disposables = new DisposableStore();
294 >
295 > constructor(
296 @IContextKeyService contextKeyService: IContextKeyService,
297 @IFileService fileService: IFileService,
315 }));
316 }
318 > dispose(): void {
319 this._disposables.dispose();
320 }
321 > } contextkeys.ts
322 >
323 > /**
324 > * This is a version of ResourceContextKey that is not disposable and has no listeners for model change events.
325 > * It will configure itself for the state/presence of a model only when created and not update.
326 > */
327 > export class StaticResourceContextKey extends AbstractResourceContextKey { }
328 >
329 >
330 > //#endregion
331 >
332 > export function applyAvailableEditorIds(contextKey: IContextKey<string>, editor: EditorInput | undefined | null, editorResolverService: IEditorResolverService): void {
333 if (!editor) {
334 contextKey.set('');
339 contextKey.set(editors.join(','));
340 }
342 function getAvailableEditorIds(editor: EditorInput, editorResolverService: IEditorResolverService): string[] {
343 // Non text editor untitled files cannot be easily serialized between
src/vs/platform/theme/common/iconRegistry.ts 247 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- iconRegistry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { RunOnceScheduler } from '../../../base/common/async.js';
7 > import { Codicon } from '../../../base/common/codicons.js';
8 > import { getCodiconFontCharacters } from '../../../base/common/codiconsUtil.js';
9 > import { ThemeIcon, IconIdentifier } from '../../../base/common/themables.js';
10 > import { Emitter, Event } from '../../../base/common/event.js';
11 > import { IJSONSchema, IJSONSchemaMap } from '../../../base/common/jsonSchema.js';
12 > import { isString } from '../../../base/common/types.js';
13 > import { URI } from '../../../base/common/uri.js';
14 > import { localize } from '../../../nls.js';
15 > import { Extensions as JSONExtensions, IJSONContributionRegistry } from '../../jsonschemas/common/jsonContributionRegistry.js';
16 > import * as platform from '../../registry/common/platform.js';
17 > import { Disposable } from '../../../base/common/lifecycle.js';
18 >
19 > // ------ API types
20 >
21 >
22 > // icon registry
23 > export const Extensions = {
24 > IconContribution: 'base.contributions.icons'
25 > };
26 >
27 > export type IconDefaults = ThemeIcon | IconDefinition;
28 >
29 > export interface IconDefinition {
30 > readonly font?: IconFontContribution; // undefined for the default font (codicon)
31 > readonly fontCharacter: string;
32 > }
33 >
34 >
35 > export interface IconContribution {
36 > readonly id: string;
37 > description: string | undefined;
38 > readonly deprecationMessage?: string;
39 > readonly defaults: IconDefaults;
40 > }
41 >
42 > export namespace IconContribution {
43 > export function getDefinition(contribution: IconContribution, registry: IIconRegistry): IconDefinition | undefined {
44 let definition = contribution.defaults;
45 while (ThemeIcon.isThemeIcon(definition)) {
52 return definition;
53 }
55 >
56 > export interface IconFontContribution {
57 > readonly id: string;
58 > readonly definition: IconFontDefinition;
59 > }
60 >
61 > export interface IconFontDefinition {
62 > readonly weight?: string;
63 > readonly style?: string;
64 > readonly src: IconFontSource[];
65 > }
66 >
67 > export namespace IconFontDefinition {
68 > export function toJSONObject(iconFont: IconFontDefinition): any {
69 return {
70 weight: iconFont.weight,
73 };
74 }
75 > export function fromJSONObject(json: any): IconFontDefinition | undefined { iconRegistry.ts
76 const stringOrUndef = (s: any) => isString(s) ? s : undefined;
77 if (json && Array.isArray(json.src) && json.src.every((s: any) => isString(s.format) && isString(s.location))) {
84 return undefined;
85 }
87 >
88 >
89 > export interface IconFontSource {
90 > readonly location: URI;
91 > readonly format: string;
92 > }
93 >
94 > export interface IIconRegistry {
95 >
96 > readonly onDidChange: Event<void>;
97 >
98 > /**
99 > * Register a icon to the registry.
100 > * @param id The icon id
101 > * @param defaults The default values
102 > * @param description The description
103 > */
104 > registerIcon(id: IconIdentifier, defaults: IconDefaults, description?: string): ThemeIcon;
105 >
106 > /**
107 > * Deregister a icon from the registry.
108 > */
109 > deregisterIcon(id: IconIdentifier): void;
110 >
111 > /**
112 > * Get all icon contributions
113 > */
114 > getIcons(): IconContribution[];
115 >
116 > /**
117 > * Get the icon for the given id
118 > */
119 > getIcon(id: IconIdentifier): IconContribution | undefined;
120 >
121 > /**
122 > * JSON schema for an object to assign icon values to one of the icon contributions.
123 > */
124 > getIconSchema(): IJSONSchema;
125 >
126 > /**
127 > * JSON schema to for a reference to a icon contribution.
128 > */
129 > getIconReferenceSchema(): IJSONSchema;
130 >
131 > /**
132 > * Register a icon font to the registry.
133 > * @param id The icon font id
134 > * @param definition The icon font definition
135 > */
136 > registerIconFont(id: string, definition: IconFontDefinition): IconFontDefinition;
137 >
138 > /**
139 > * Deregister an icon font to the registry.
140 > */
141 > deregisterIconFont(id: string): void;
142 >
143 > /**
144 > * Get the icon font for the given id
145 > */
146 > getIconFont(id: string): IconFontDefinition | undefined;
147 > }
148 >
149 > // regexes for validation of font properties
150 >
151 > export const fontIdRegex = /^([\w_-]+)$/;
152 > export const fontStyleRegex = /^(normal|italic|(oblique[ \w\s-]+))$/;
153 > export const fontWeightRegex = /^(normal|bold|lighter|bolder|(\d{0-1000}))$/;
154 > export const fontSizeRegex = /^([\w_.%+-]+)$/;
155 > export const fontFormatRegex = /^woff|woff2|truetype|opentype|embedded-opentype|svg$/;
156 > export const fontColorRegex = /^#[0-9a-fA-F]{0,6}$/;
157 >
158 > export const fontIdErrorMessage = localize('schema.fontId.formatError', 'The font ID must only contain letters, numbers, underscores and dashes.');
159 >
160 > class IconRegistry extends Disposable implements IIconRegistry {
161 >
162 > private readonly _onDidChange = this._register(new Emitter<void>());
163 > readonly onDidChange: Event<void> = this._onDidChange.event;
164 >
165 > private iconsById: { [key: string]: IconContribution };
166 > private iconSchema: IJSONSchema & { properties: IJSONSchemaMap } = {
167 > definitions: {
168 > icons: {
169 > type: 'object',
170 > properties: {
171 > fontId: { type: 'string', description: localize('iconDefinition.fontId', 'The id of the font to use. If not set, the font that is defined first is used.'), pattern: fontIdRegex.source, patternErrorMessage: fontIdErrorMessage },
172 > fontCharacter: { type: 'string', description: localize('iconDefinition.fontCharacter', 'The font character associated with the icon definition.') }
173 > },
174 > additionalProperties: false,
175 > defaultSnippets: [{ body: { fontCharacter: '\\\\e030' } }]
176 > }
177 > },
178 > type: 'object',
179 > properties: {}
180 > };
181 > private iconReferenceSchema: IJSONSchema & { enum: string[]; enumDescriptions: string[] } = { type: 'string', pattern: `^${ThemeIcon.iconNameExpression}$`, enum: [], enumDescriptions: [] };
182 >
183 > private iconFontsById: { [key: string]: IconFontDefinition };
184 >
185 > constructor() {
186 > super();
187 > this.iconsById = {};
188 > this.iconFontsById = {};
189 > }
190 >
191 > public registerIcon(id: string, defaults: IconDefaults, description?: string, deprecationMessage?: string): ThemeIcon {
192 > const existing = this.iconsById[id];
193 > if (existing) {
194 if (description && !existing.description) {
195 existing.description = description;
203 return existing;
204 }
205 > const iconContribution: IconContribution = { id, description, defaults, deprecationMessage }; iconRegistry.ts
206 > this.iconsById[id] = iconContribution;
207 > const propertySchema: IJSONSchema = { $ref: '#/definitions/icons' };
208 > if (deprecationMessage) {
209 propertySchema.deprecationMessage = deprecationMessage;
210 }
211 > if (description) { iconRegistry.ts
212 > propertySchema.markdownDescription = `${description}: $(${id})`;
213 > }
214 > this.iconSchema.properties[id] = propertySchema;
215 > this.iconReferenceSchema.enum.push(id);
216 > this.iconReferenceSchema.enumDescriptions.push(description || '');
217 >
218 > this._onDidChange.fire();
219 > return { id };
220 > }
221 >
222 >
223 > public deregisterIcon(id: string): void {
224 delete this.iconsById[id];
225 delete this.iconSchema.properties[id];
231 this._onDidChange.fire();
232 }
234 > public getIcons(): IconContribution[] {
235 return Object.keys(this.iconsById).map(id => this.iconsById[id]);
236 }
238 > public getIcon(id: string): IconContribution | undefined {
239 return this.iconsById[id];
240 }
242 > public getIconSchema(): IJSONSchema {
243 > return this.iconSchema;
244 > }
245 >
246 > public getIconReferenceSchema(): IJSONSchema {
247 return this.iconReferenceSchema;
248 }
250 > public registerIconFont(id: string, definition: IconFontDefinition): IconFontDefinition {
251 const existing = this.iconFontsById[id];
252 if (existing) {
257 return definition;
258 }
260 > public deregisterIconFont(id: string): void {
261 delete this.iconFontsById[id];
262 }
264 > public getIconFont(id: string): IconFontDefinition | undefined {
265 return this.iconFontsById[id];
266 }
268 > public override toString() {
269 const sorter = (i1: IconContribution, i2: IconContribution) => {
270 return i1.id.localeCompare(i2.id);
297 return reference.join('\n');
298 }
300 > }
301 >
302 > const iconRegistry = new IconRegistry();
303 > platform.Registry.add(Extensions.IconContribution, iconRegistry);
304 >
305 > export function registerIcon(id: string, defaults: IconDefaults, description: string, deprecationMessage?: string): ThemeIcon {
306 > return iconRegistry.registerIcon(id, defaults, description, deprecationMessage);
307 > }
308 >
309 > export function getIconRegistry(): IIconRegistry {
310 return iconRegistry;
311 }
313 > function initialize() {
314 > const codiconFontCharacters = getCodiconFontCharacters();
315 > for (const icon in codiconFontCharacters) {
316 > const fontCharacter = '\\' + codiconFontCharacters[icon].toString(16);
317 > iconRegistry.registerIcon(icon, { fontCharacter });
318 > }
319 > }
320 > initialize();
321 >
322 > export const iconsSchemaId = 'vscode://schemas/icons';
323 >
324 > const schemaRegistry = platform.Registry.as<IJSONContributionRegistry>(JSONExtensions.JSONContribution);
325 > schemaRegistry.registerSchema(iconsSchemaId, iconRegistry.getIconSchema());
326 >
327 > const delayer = new RunOnceScheduler(() => schemaRegistry.notifySchemaChanged(iconsSchemaId), 200);
328 > iconRegistry.onDidChange(() => {
329 > if (!delayer.isScheduled()) {
330 > delayer.schedule();
331 > }
332 > });
333 >
334 > //setTimeout(_ => console.log(iconRegistry.toString()), 5000);
335 >
336 >
337 > // common icons
338 >
339 > export const widgetClose = registerIcon('widget-close', Codicon.close, localize('widgetClose', 'Icon for the close action in widgets.'));
340 >
341 > export const gotoPreviousLocation = registerIcon('goto-previous-location', Codicon.arrowUp, localize('previousChangeIcon', 'Icon for goto previous editor location.'));
342 > export const gotoNextLocation = registerIcon('goto-next-location', Codicon.arrowDown, localize('nextChangeIcon', 'Icon for goto next editor location.'));
343 >
344 > export const syncing = ThemeIcon.modify(Codicon.sync, 'spin');
345 > export const spinningLoading = ThemeIcon.modify(Codicon.loading, 'spin');
src/vs/platform/agentHost/common/state/protocol/channels-root/state.ts 244 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { ConfigSchema, JsonPrimitive, ProtectedResourceMetadata } from '../common/state.js';
10 > import type { TerminalInfo } from '../channels-terminal/state.js';
11 > import type { Customization } from '../channels-session/state.js';
12 >
13 > // ─── Root State ──────────────────────────────────────────────────────────────
14 >
15 > /**
16 > * Policy configuration state for a model.
17 > *
18 > * @category Root State
19 > */
20 > export const enum PolicyState {
21 > Enabled = 'enabled',
22 > Disabled = 'disabled',
23 > Unconfigured = 'unconfigured',
24 > }
25 >
26 > /**
27 > * Global state shared with every client subscribed to `ahp-root://`.
28 > *
29 > * @category Root State
30 > */
31 > export interface RootState {
32 > /** Available agent backends and their models */
33 > agents: AgentInfo[];
34 > /** Number of active (non-disposed) sessions on the server */
35 > activeSessions?: number;
36 > /** Known terminals on the server. Subscribe to individual terminal URIs for full state. */
37 > terminals?: TerminalInfo[];
38 > /** Agent host configuration schema and current values */
39 > config?: RootConfigState;
40 > /**
41 > * Additional implementation-defined metadata about the agent host itself.
42 > *
43 > * Clients MAY look for well-known keys here to provide enhanced UI.
44 > */
45 > _meta?: Record<string, unknown>;
46 > }
47 >
48 > /**
49 > * @category Root State
50 > */
51 > export interface AgentInfo {
52 > /** Agent provider ID (e.g. `'copilot'`) */
53 > provider: string;
54 > /** Human-readable name */
55 > displayName: string;
56 > /** Description string */
57 > description: string;
58 > /** Available models for this agent */
59 > models: SessionModelInfo[];
60 > /**
61 > * Protected resources this agent requires authentication for.
62 > *
63 > * Each entry describes an OAuth 2.0 protected resource using
64 > * [RFC 9728](https://datatracker.ietf.org/doc/html/rfc9728) semantics.
65 > * Clients should obtain tokens from the declared `authorization_servers`
66 > * and push them via the `authenticate` command before creating sessions
67 > * with this agent.
68 > *
69 > * @see {@link /specification/authentication | Authentication}
70 > */
71 > protectedResources?: ProtectedResourceMetadata[];
72 > /**
73 > * Customizations associated with this agent.
74 > *
75 > * Either container customizations —
76 > * {@link PluginCustomization | `PluginCustomization`} entries the agent
77 > * bundles, plus {@link DirectoryCustomization | `DirectoryCustomization`}
78 > * entries it watches in any workspace it's used with — or top-level
79 > * {@link McpServerCustomization | `McpServerCustomization`} entries
80 > * the agent host declares directly. When a session is created with
81 > * this agent, these entries are augmented (e.g. directory URIs are
82 > * resolved against the workspace, children are parsed) and propagated
83 > * into the session's `customizations` list.
84 > */
85 > customizations?: Customization[];
86 > /**
87 > * Static capabilities the agent advertises about itself. Clients use these
88 > * to gate features (multi-chat, fork) instead of switching on the provider
89 > * id.
90 > */
91 > capabilities?: AgentCapabilities;
92 > }
93 >
94 > /**
95 > * Static capabilities an {@link AgentInfo} advertises. Modelled after MCP
96 > * capabilities: each field is opt-in and its presence (an empty object `{}`)
97 > * signals support, while absence means the feature is unsupported and the
98 > * corresponding client commands MUST NOT be used. Sub-fields carry
99 > * per-capability options.
100 > *
101 > * @category Root State
102 > */
103 > export interface AgentCapabilities {
104 > /**
105 > * The agent can host more than one concurrent chat per session. When absent,
106 > * clients MUST NOT call `createChat` to open chats beyond the default one the
107 > * session starts with. An empty object `{}` advertises multi-chat without
108 > * source-based creation; set {@link MultipleChatsCapability.fork} or
109 > * {@link MultipleChatsCapability.sideChat} to allow the corresponding mode.
110 > */
111 > multipleChats?: MultipleChatsCapability;
112 > /**
113 > * The session's agent can be granted tool access to more than one working
114 > * directory. The directories are treated as equal peers except where the
115 > * agent advertises {@link MultipleWorkingDirectoriesCapability.requiresPrimary}
116 > * (some backends need one directory designated as a primary root).
117 > *
118 > * When absent, clients MUST NOT mutate a session's or chat's working-directory
119 > * set and MUST NOT set more than one entry in
120 > * {@link CreateSessionParams.workingDirectories}.
121 > */
122 > multipleWorkingDirectories?: MultipleWorkingDirectoriesCapability;
123 > }
124 >
125 > /**
126 > * Options for the {@link AgentCapabilities.multipleChats} capability.
127 > *
128 > * @category Root State
129 > */
130 > export interface MultipleChatsCapability {
131 > /**
132 > * The agent can fork a chat from a specific turn. When absent or `false`,
133 > * clients MUST NOT pass a {@link ChatSource} with `kind: "fork"` to
134 > * `createChat`.
135 > * Forking always implies multi-chat support.
136 > */
137 > fork?: boolean;
138 > /**
139 > * The agent can create a side chat from a specific turn. When absent or
140 > * `false`, clients MUST NOT pass a {@link ChatSource} with
141 > * `kind: "sideChat"` to `createChat`.
142 > *
143 > * A side chat receives the source turn as context without copying the source
144 > * transcript into its own visible history. The source is identified by a
145 > * stable `turnId`, which the host resolves against the source chat's current
146 > * `activeTurn` or retained history. When it names the current active turn,
147 > * the host snapshots the available partial assistant response at creation
148 > * time. Side-chat support always implies multi-chat support.
149 > */
150 > sideChat?: boolean;
151 > }
152 >
153 > /**
154 > * Options for the {@link AgentCapabilities.multipleWorkingDirectories} capability.
155 > *
156 > * @category Root State
157 > */
158 > export interface MultipleWorkingDirectoriesCapability {
159 > /**
160 > * The agent requires each chat to designate one of its working directories as
161 > * the **primary** — a distinguished root the chat is centered on (e.g. the
162 > * agent's process root for that chat, the default location for relative
163 > * paths). Primary is a **per-chat** notion, fixed at chat creation. When
164 > * `true`, a client SHOULD supply {@link CreateChatParams.primaryWorkingDirectory}
165 > * (and {@link CreateSessionParams.primaryWorkingDirectory}, which seeds the
166 > * session's default chat); a host MAY reject creation that omits it, or fall
167 > * back to the first entry of the chat's working directories. The chosen
168 > * primary is reported (read-only) on {@link ChatState.primaryWorkingDirectory}.
169 > *
170 > * When absent or `false`, the agent has no primary — all directories are
171 > * equal peers and clients need not designate one.
172 > */
173 > requiresPrimary?: boolean;
174 > }
175 >
176 > /**
177 > * @category Root State
178 > */
179 > export interface SessionModelInfo {
180 > /** Model identifier */
181 > id: string;
182 > /** Provider this model belongs to */
183 > provider: string;
184 > /** Human-readable model name */
185 > name: string;
186 > /** Maximum context window size */
187 > maxContextWindow?: number;
188 > /** Maximum number of output tokens the model can generate */
189 > maxOutputTokens?: number;
190 > /** Maximum number of prompt (input) tokens the model accepts */
191 > maxPromptTokens?: number;
192 > /** Whether the model supports vision */
193 > supportsVision?: boolean;
194 > /** Policy configuration state */
195 > policyState?: PolicyState;
196 > /**
197 > * Configuration schema describing model-specific options (e.g. thinking
198 > * level). Clients present this as a form and pass the resolved values in
199 > * {@link ModelSelection.config} when creating or changing sessions.
200 > */
201 > configSchema?: ConfigSchema;
202 > /**
203 > * Additional provider-specific metadata for this model.
204 > *
205 > * Clients MAY look for well-known keys here to provide enhanced UI.
206 > * For example, a `pricing` key may carry model pricing metadata.
207 > */
208 > _meta?: Record<string, unknown>;
209 > }
210 >
211 > /**
212 > * A model selection: the chosen model ID together with any model-specific
213 > * configuration values whose keys correspond to the model's
214 > * {@link SessionModelInfo.configSchema}.
215 > *
216 > * @category Root State
217 > */
218 > export interface ModelSelection {
219 > /** Model identifier */
220 > id: string;
221 > /**
222 > * Model-specific configuration values. Values are JSON primitives: most
223 > * pickers produce strings, but some (e.g. a numeric context-size picker)
224 > * produce numbers or booleans, which are carried through as-is.
225 > */
226 > config?: Record<string, JsonPrimitive>;
227 > }
228 >
229 > // ─── Root Config Types ───────────────────────────────────────────────────────
230 >
231 > /**
232 > * Live agent-host configuration metadata.
233 > *
234 > * The schema describes the available configuration properties and the values
235 > * contain the current value for each resolved property.
236 > *
237 > * @category Root State
238 > */
239 > export interface RootConfigState {
240 > /** JSON Schema describing available configuration properties */
241 > schema: ConfigSchema;
242 > /** Current configuration values */
243 > values: Record<string, unknown>;
244 > }
src/vs/base/parts/storage/common/storage.ts 237 covered LOC · 42 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- storage.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ThrottledDelayer } from '../../../common/async.js';
7 > import { Event, PauseableEmitter } from '../../../common/event.js';
8 > import { Disposable, IDisposable } from '../../../common/lifecycle.js';
9 > import { parse, stringify } from '../../../common/marshalling.js';
10 > import { isObject, isUndefined, isUndefinedOrNull } from '../../../common/types.js';
11 >
12 > export enum StorageHint {
13 >
14 > // A hint to the storage that the storage
15 > // does not exist on disk yet. This allows
16 > // the storage library to improve startup
17 > // time by not checking the storage for data.
18 > STORAGE_DOES_NOT_EXIST,
19 >
20 > // A hint to the storage that the storage
21 > // is backed by an in-memory storage.
22 > STORAGE_IN_MEMORY
23 > }
24 >
25 > export interface IStorageOptions {
26 > readonly hint?: StorageHint;
27 > }
28 >
29 > export interface IUpdateRequest {
30 > readonly insert?: Map<string, string>;
31 > readonly delete?: Set<string>;
32 > }
33 >
34 > export interface IStorageItemsChangeEvent {
35 > readonly changed?: Map<string, string>;
36 > readonly deleted?: Set<string>;
37 > }
38 >
39 > export function isStorageItemsChangeEvent(thing: unknown): thing is IStorageItemsChangeEvent {
40 const candidate = thing as IStorageItemsChangeEvent | undefined;
41
42 return candidate?.changed instanceof Map || candidate?.deleted instanceof Set;
43 }
44 > storage.ts
45 > export interface IStorageDatabase {
46 >
47 > readonly onDidChangeItemsExternal: Event<IStorageItemsChangeEvent>;
48 >
49 > getItems(): Promise<Map<string, string>>;
50 > updateItems(request: IUpdateRequest): Promise<void>;
51 >
52 > optimize(): Promise<void>;
53 >
54 > close(recovery?: () => Map<string, string>): Promise<void>;
55 > }
56 >
57 > export interface IStorageChangeEvent {
58 >
59 > /**
60 > * The `key` of the storage entry that was changed
61 > * or was removed.
62 > */
63 > readonly key: string;
64 >
65 > /**
66 > * A hint how the storage change event was triggered. If
67 > * `true`, the storage change was triggered by an external
68 > * source, such as:
69 > * - another process (for example another window)
70 > * - operations such as settings sync or profiles change
71 > */
72 > readonly external?: boolean;
73 > }
74 >
75 > export type StorageValue = string | boolean | number | undefined | null | object;
76 >
77 > export interface IStorage extends IDisposable {
78 >
79 > readonly onDidChangeStorage: Event<IStorageChangeEvent>;
80 >
81 > readonly items: Map<string, string>;
82 > readonly size: number;
83 >
84 > init(): Promise<void>;
85 >
86 > get(key: string, fallbackValue: string): string;
87 > get(key: string, fallbackValue?: string): string | undefined;
88 >
89 > getBoolean(key: string, fallbackValue: boolean): boolean;
90 > getBoolean(key: string, fallbackValue?: boolean): boolean | undefined;
91 >
92 > getNumber(key: string, fallbackValue: number): number;
93 > getNumber(key: string, fallbackValue?: number): number | undefined;
94 >
95 > getObject<T extends object>(key: string, fallbackValue: T): T;
96 > getObject<T extends object>(key: string, fallbackValue?: T): T | undefined;
97 >
98 > set(key: string, value: StorageValue, external?: boolean): Promise<void>;
99 > delete(key: string, external?: boolean): Promise<void>;
100 >
101 > flush(delay?: number): Promise<void>;
102 > whenFlushed(): Promise<void>;
103 >
104 > optimize(): Promise<void>;
105 >
106 > close(): Promise<void>;
107 > }
108 >
109 > export enum StorageState {
110 > None,
111 > Initialized,
112 > Closed
113 > }
114 >
115 > export class Storage extends Disposable implements IStorage {
116 >
117 > private static readonly DEFAULT_FLUSH_DELAY = 100;
118 >
119 > private readonly _onDidChangeStorage = this._register(new PauseableEmitter<IStorageChangeEvent>());
120 > readonly onDidChangeStorage = this._onDidChangeStorage.event;
121 >
122 > private state = StorageState.None;
123 >
124 > private cache = new Map<string, string>();
125 >
126 > private readonly flushDelayer = this._register(new ThrottledDelayer<void>(Storage.DEFAULT_FLUSH_DELAY));
127 >
128 > private pendingDeletes = new Set<string>();
129 > private pendingInserts = new Map<string, string>();
130 >
131 > private pendingClose: Promise<void> | undefined = undefined;
132 >
133 > private readonly whenFlushedCallbacks: Function[] = [];
134 >
135 > constructor(
136 > protected readonly database: IStorageDatabase, storage.ts
137 > private readonly options: IStorageOptions = Object.create(null)
138 > ) {
139 > super();
140 >
141 > this.registerListeners();
142 > }
143 > storage.ts
144 > private registerListeners(): void {
145 > this._register(this.database.onDidChangeItemsExternal(e => this.onDidChangeItemsExternal(e))); storage.ts
146 > }
147 > storage.ts
148 > private onDidChangeItemsExternal(e: IStorageItemsChangeEvent): void {
149 this._onDidChangeStorage.pause();
150
161 }
162 }
163 > storage.ts
164 > private acceptExternal(key: string, value: string | undefined): void {
165 if (this.state === StorageState.Closed) {
166 return; // Return early if we are already closed
188 }
189 }
190 > storage.ts
191 > get items(): Map<string, string> {
192 return this.cache;
193 }
194 > storage.ts
195 > get size(): number {
196 return this.cache.size;
197 }
198 > storage.ts
199 > async init(): Promise<void> {
200 if (this.state !== StorageState.None) {
201 return; // either closed or already initialized
213 this.cache = await this.database.getItems();
214 }
215 > storage.ts
216 > get(key: string, fallbackValue: string): string;
217 > get(key: string, fallbackValue?: string): string | undefined;
218 > get(key: string, fallbackValue?: string): string | undefined {
219 > const value = this.cache.get(key); storage.ts
220 >
221 > if (isUndefinedOrNull(value)) {
222 > return fallbackValue; storage.ts
223 > }
224
225 return value;
226 > } storage.ts
227 > storage.ts
228 > getBoolean(key: string, fallbackValue: boolean): boolean;
229 > getBoolean(key: string, fallbackValue?: boolean): boolean | undefined;
230 > getBoolean(key: string, fallbackValue?: boolean): boolean | undefined {
231 const value = this.get(key);
232
237 return value === 'true';
238 }
239 > storage.ts
240 > getNumber(key: string, fallbackValue: number): number;
241 > getNumber(key: string, fallbackValue?: number): number | undefined;
242 > getNumber(key: string, fallbackValue?: number): number | undefined {
243 const value = this.get(key);
244
249 return parseInt(value, 10);
250 }
251 > storage.ts
252 > getObject(key: string, fallbackValue: object): object;
253 > getObject(key: string, fallbackValue?: object | undefined): object | undefined;
254 > getObject(key: string, fallbackValue?: object): object | undefined {
255 > const value = this.get(key); storage.ts
256 >
257 > if (isUndefinedOrNull(value)) {
258 > return fallbackValue; storage.ts
259 > }
260
261 return parse(value);
262 > } storage.ts
263 > storage.ts
264 > async set(key: string, value: string | boolean | number | null | undefined | object, external = false): Promise<void> {
265 if (this.state === StorageState.Closed) {
266 return; // Return early if we are already closed
292 return this.doFlush();
293 }
294 > storage.ts
295 > async delete(key: string, external = false): Promise<void> {
296 if (this.state === StorageState.Closed) {
297 return; // Return early if we are already closed
316 return this.doFlush();
317 }
318 > storage.ts
319 > async optimize(): Promise<void> {
320 if (this.state === StorageState.Closed) {
321 return; // Return early if we are already closed
328 return this.database.optimize();
329 }
330 > storage.ts
331 > async close(): Promise<void> {
332 if (!this.pendingClose) {
333 this.pendingClose = this.doClose();
336 return this.pendingClose;
337 }
338 > storage.ts
339 > private async doClose(): Promise<void> {
340
341 // Update state
356 await this.database.close(() => this.cache);
357 }
358 > storage.ts
359 > private get hasPending() {
360 return this.pendingInserts.size > 0 || this.pendingDeletes.size > 0;
361 }
362 > storage.ts
363 > private async flushPending(): Promise<void> {
364 if (!this.hasPending) {
365 return; // return early if nothing to do
383 });
384 }
385 > storage.ts
386 > async flush(delay?: number): Promise<void> {
387 if (
388 this.state === StorageState.Closed || // Return early if we are already closed
394 return this.doFlush(delay);
395 }
396 > storage.ts
397 > private async doFlush(delay?: number): Promise<void> {
398 if (this.options.hint === StorageHint.STORAGE_IN_MEMORY) {
399 return this.flushPending(); // return early if in-memory
402 return this.flushDelayer.trigger(() => this.flushPending(), delay);
403 }
404 > storage.ts
405 > async whenFlushed(): Promise<void> {
406 if (!this.hasPending) {
407 return; // return early if nothing to do
410 return new Promise(resolve => this.whenFlushedCallbacks.push(resolve));
411 }
412 > storage.ts
413 > isInMemory(): boolean {
414 return this.options.hint === StorageHint.STORAGE_IN_MEMORY;
415 }
416 > } storage.ts
417 >
418 > export class InMemoryStorageDatabase implements IStorageDatabase {
419 > storage.ts
420 > readonly onDidChangeItemsExternal = Event.None;
421 >
422 > private readonly items = new Map<string, string>();
423 > storage.ts
424 > async getItems(): Promise<Map<string, string>> {
425 return this.items;
426 }
427 > storage.ts
428 > async updateItems(request: IUpdateRequest): Promise<void> {
429 request.insert?.forEach((value, key) => this.items.set(key, value));
430
431 request.delete?.forEach(key => this.items.delete(key));
432 }
433 > storage.ts
434 > async optimize(): Promise<void> { }
435 > async close(): Promise<void> { }
436 > }
437 >
438 >
439 > export const MIGRATED_KEY = '__$__migratedStorageMarker';
440 >
441 > export class MigratingStorage extends Storage {
442
443 private migratedKeys: Set<string> = new Set();
444 private fallbackStorage: IStorage | undefined = undefined;
445 private isFallbackStorageReadonly: boolean = false;
446 > storage.ts
447 > override async init(): Promise<void> {
448 await super.init();
449
451 this.migratedKeys = this.loadMigratedKeys();
452 }
453 > storage.ts
454 > public setFallbackStorage(storage: IStorage, isReadonly: boolean): void {
455 this.fallbackStorage = storage;
456 this.isFallbackStorageReadonly = isReadonly;
457 }
458 > storage.ts
459 > private static readonly INTERNAL_KEY_PREFIX = '__$__';
460 >
461 > override get(key: string, fallbackValue: string): string;
462 > override get(key: string, fallbackValue?: string): string | undefined;
463 > override get(key: string, fallbackValue?: string): string | undefined {
464 if (!key.startsWith(MigratingStorage.INTERNAL_KEY_PREFIX) && !this.migratedKeys.has(key) && isUndefined(super.get(key))) {
465 // Check fallback storage and auto-migrate on hit.
480 return super.get(key, fallbackValue);
481 }
482 > storage.ts
483 > private loadMigratedKeys(): Set<string> {
484 const raw = super.get(MIGRATED_KEY);
485 if (raw) {
492 return new Set();
493 }
494 > storage.ts
495 > private persistMigratedKeys(): void {
496 this.set(MIGRATED_KEY, JSON.stringify([...this.migratedKeys]));
497 }
498 > } storage.ts
499
src/vs/nls.ts 224 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- nls.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export function getNLSMessages(): string[] {
7 return globalThis._VSCODE_NLS_MESSAGES;
8 }
9 > nls.ts
10 > export function getNLSLanguage(): string | undefined {
11 > return globalThis._VSCODE_NLS_LANGUAGE;
12 > }
13 >
14 > declare const document: { location?: { hash?: string } } | undefined;
15 > const isPseudo = getNLSLanguage() === 'pseudo' || (typeof document !== 'undefined' && document.location && typeof document.location.hash === 'string' && document.location.hash.indexOf('pseudo=true') >= 0);
16 >
17 > export interface ILocalizeInfo {
18 > key: string;
19 > comment: string[];
20 > }
21 >
22 > export interface ILocalizedString {
23 > original: string;
24 > value: string;
25 > }
26 >
27 > function _format(message: string, args: (string | number | boolean | undefined | null)[]): string { nls.ts
28 > let result: string;
29 >
30 > if (args.length === 0) {
31 > result = message; nls.ts
32 > } else { nls.ts
33 > result = message.replace(/\{(\d+)\}/g, (match, rest) => { nls.ts
34 > const index = rest[0];
35 > const arg = args[index];
36 > let result = match;
37 > if (typeof arg === 'string') {
38 > result = arg; nls.ts
39 > } else if (typeof arg === 'number' || typeof arg === 'boolean' || arg === void 0 || arg === null) { nls.ts
40 result = String(arg);
41 }
42 > return result; nls.ts
43 > });
44 > }
45 > nls.ts
46 > if (isPseudo) {
47 // FF3B and FF3D is the Unicode zenkaku representation for [ and ]
48 result = '\uFF3B' + result.replace(/[aouei]/g, '$&$&') + '\uFF3D';
49 }
50 > nls.ts
51 > return result;
52 > }
53 > nls.ts
54 > /**
55 > * Marks a string to be localized. Returns the localized string.
56 > *
57 > * @param info The {@linkcode ILocalizeInfo} which describes the id and comments associated with the localized string.
58 > * @param message The string to localize
59 > * @param args The arguments to the string
60 > *
61 > * @note `message` can contain `{n}` notation where it is replaced by the nth value in `...args`
62 > * @example `localize({ key: 'sayHello', comment: ['Welcomes user'] }, 'hello {0}', name)`
63 > *
64 > * @returns string The localized string.
65 > */
66 > export function localize(info: ILocalizeInfo, message: string, ...args: (string | number | boolean | undefined | null)[]): string;
67 >
68 > /**
69 > * Marks a string to be localized. Returns the localized string.
70 > *
71 > * @param key The key to use for localizing the string
72 > * @param message The string to localize
73 > * @param args The arguments to the string
74 > *
75 > * @note `message` can contain `{n}` notation where it is replaced by the nth value in `...args`
76 > * @example For example, `localize('sayHello', 'hello {0}', name)`
77 > *
78 > * @returns string The localized string.
79 > */
80 > export function localize(key: string, message: string, ...args: (string | number | boolean | undefined | null)[]): string;
81 >
82 > /**
83 > * @skipMangle
84 > */
85 > export function localize(data: ILocalizeInfo | string /* | number when built */, message: string /* | null when built */, ...args: (string | number | boolean | undefined | null)[]): string {
86 > if (typeof data === 'number') { nls.ts
87 return _format(lookupMessage(data, message), args);
88 }
89 > return _format(message, args); nls.ts
90 > }
91 > nls.ts
92 > /**
93 > * Only used when built: Looks up the message in the global NLS table.
94 > * This table is being made available as a global through bootstrapping
95 > * depending on the target context.
96 > */
97 function lookupMessage(index: number, fallback: string | null): string {
98 const message = getNLSMessages()?.[index];
105 return message;
106 }
107 > nls.ts
108 > /**
109 > * Marks a string to be localized. Returns an {@linkcode ILocalizedString}
110 > * which contains the localized string and the original string.
111 > *
112 > * @param info The {@linkcode ILocalizeInfo} which describes the id and comments associated with the localized string.
113 > * @param message The string to localize
114 > * @param args The arguments to the string
115 > *
116 > * @note `message` can contain `{n}` notation where it is replaced by the nth value in `...args`
117 > * @example `localize2({ key: 'sayHello', comment: ['Welcomes user'] }, 'hello {0}', name)`
118 > *
119 > * @returns ILocalizedString which contains the localized string and the original string.
120 > */
121 > export function localize2(info: ILocalizeInfo, message: string, ...args: (string | number | boolean | undefined | null)[]): ILocalizedString;
122 >
123 > /**
124 > * Marks a string to be localized. Returns an {@linkcode ILocalizedString}
125 > * which contains the localized string and the original string.
126 > *
127 > * @param key The key to use for localizing the string
128 > * @param message The string to localize
129 > * @param args The arguments to the string
130 > *
131 > * @note `message` can contain `{n}` notation where it is replaced by the nth value in `...args`
132 > * @example `localize('sayHello', 'hello {0}', name)`
133 > *
134 > * @returns ILocalizedString which contains the localized string and the original string.
135 > */
136 > export function localize2(key: string, message: string, ...args: (string | number | boolean | undefined | null)[]): ILocalizedString;
137 >
138 > /**
139 > * @skipMangle
140 > */
141 > export function localize2(data: ILocalizeInfo | string /* | number when built */, originalMessage: string, ...args: (string | number | boolean | undefined | null)[]): ILocalizedString {
142 > let message: string; nls.ts
143 > if (typeof data === 'number') {
144 message = lookupMessage(data, originalMessage);
145 > } else { nls.ts
146 > message = originalMessage;
147 > }
148 >
149 > const value = _format(message, args);
150 >
151 > return {
152 > value,
153 > original: originalMessage === message ? value : _format(originalMessage, args)
154 > };
155 > }
156 > nls.ts
157 > export interface INLSLanguagePackConfiguration {
158 >
159 > /**
160 > * The path to the translations config file that contains pointers to
161 > * all message bundles for `main` and extensions.
162 > */
163 > readonly translationsConfigFile: string;
164 >
165 > /**
166 > * The path to the file containing the translations for this language
167 > * pack as flat string array.
168 > */
169 > readonly messagesFile: string;
170 >
171 > /**
172 > * The path to the file that can be used to signal a corrupt language
173 > * pack, for example when reading the `messagesFile` fails. This will
174 > * instruct the application to re-create the cache on next startup.
175 > */
176 > readonly corruptMarkerFile: string;
177 > }
178 >
179 > export interface INLSConfiguration {
180 >
181 > /**
182 > * Locale as defined in `argv.json` or `app.getLocale()`.
183 > */
184 > readonly userLocale: string;
185 >
186 > /**
187 > * Locale as defined by the OS (e.g. `app.getPreferredSystemLanguages()`).
188 > */
189 > readonly osLocale: string;
190 >
191 > /**
192 > * The actual language of the UI that ends up being used considering `userLocale`
193 > * and `osLocale`.
194 > */
195 > readonly resolvedLanguage: string;
196 >
197 > /**
198 > * Defined if a language pack is used that is not the
199 > * default english language pack. This requires a language
200 > * pack to be installed as extension.
201 > */
202 > readonly languagePack?: INLSLanguagePackConfiguration;
203 >
204 > /**
205 > * The path to the file containing the default english messages
206 > * as flat string array. The file is only present in built
207 > * versions of the application.
208 > */
209 > readonly defaultMessagesFile: string;
210 >
211 > /**
212 > * Below properties are deprecated and only there to continue support
213 > * for `vscode-nls` module that depends on them.
214 > * Refs https://github.com/microsoft/vscode-nls/blob/main/src/node/main.ts#L36-L46
215 > */
216 > /** @deprecated */
217 > readonly locale: string;
218 > /** @deprecated */
219 > readonly availableLanguages: Record<string, string>;
220 > /** @deprecated */
221 > readonly _languagePackSupport?: boolean;
222 > /** @deprecated */
223 > readonly _languagePackId?: string;
224 > /** @deprecated */
225 > readonly _translationsConfigFile?: string;
226 > /** @deprecated */
227 > readonly _cacheRoot?: string;
228 > /** @deprecated */
229 > readonly _resolvedLanguagePackCoreLocation?: string;
230 > /** @deprecated */
231 > readonly _corruptedFile?: string;
232 > }
233 >
234 > export interface ILanguagePack {
235 > readonly hash: string;
236 > readonly label: string | undefined;
237 > readonly extensions: {
238 > readonly extensionIdentifier: { readonly id: string; readonly uuid?: string };
239 > readonly version: string;
240 > }[];
241 > readonly translations: Record<string, string | undefined>;
242 > }
243 >
244 > export type ILanguagePacks = Record<string, ILanguagePack | undefined>;
src/vs/base/common/observableInternal/reactions/autorunImpl.ts 221 covered LOC · 54 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- autorunImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IObservableWithChange, IObserver, IReaderWithStore } from '../base.js';
7 > import { DebugNameData } from '../debugName.js';
8 > import { assertFn, BugIndicatingError, DisposableStore, IDisposable, markAsDisposed, onBugIndicatingError, trackDisposable } from '../commonFacade/deps.js';
9 > import { getLogger } from '../logging/logging.js';
10 > import { IChangeTracker } from '../changeTracker.js';
11 > import { DebugLocation } from '../debugLocation.js';
12 >
13 > export const enum AutorunState {
14 > /**
15 > * A dependency could have changed.
16 > * We need to explicitly ask them if at least one dependency changed.
17 > */
18 > dependenciesMightHaveChanged = 1,
19 >
20 > /**
21 > * A dependency changed and we need to recompute.
22 > */
23 > stale = 2,
24 > upToDate = 3,
25 > }
26 >
27 function autorunStateToString(state: AutorunState): string {
28 switch (state) {
33 }
34 }
36 > export class AutorunObserver<TChangeSummary = any> implements IObserver, IReaderWithStore, IDisposable {
37 > private _state = AutorunState.stale;
38 > private _updateCount = 0;
39 > private _disposed = false;
40 > private _dependencies = new Set<IObservable<any>>();
41 > private _dependenciesToBeRemoved = new Set<IObservable<any>>();
42 > private _changeSummary: TChangeSummary | undefined;
43 > private _isRunning = false;
44 > private _iteration = 0;
45 >
46 > public get debugName(): string {
47 > return this._debugNameData.getDebugName(this) ?? '(anonymous)';
48 > }
49 >
50 > constructor(
51 > public readonly _debugNameData: DebugNameData, autorunImpl.ts
52 > public readonly _runFn: (reader: IReaderWithStore, changeSummary: TChangeSummary) => void,
53 > private readonly _changeTracker: IChangeTracker<TChangeSummary> | undefined,
54 > debugLocation: DebugLocation
55 > ) {
56 > this._changeSummary = this._changeTracker?.createChangeSummary(undefined);
57 > getLogger()?.handleAutorunCreated(this, debugLocation);
58 > this._run();
59 >
60 > trackDisposable(this);
61 > }
63 > public dispose(): void {
64 > if (this._disposed) { autorunImpl.ts
65 return;
66 }
67 > this._disposed = true; autorunImpl.ts
68 > for (const o of this._dependencies) {
69 > o.removeObserver(this); // Warning: external call!
70 > }
71 > this._dependencies.clear();
72 >
73 > if (this._store !== undefined) {
74 > this._store.dispose(); autorunImpl.ts
75 > }
76 > if (this._delayedStore !== undefined) { autorunImpl.ts
77 > this._delayedStore.dispose(); autorunImpl.ts
78 > }
80 > getLogger()?.handleAutorunDisposed(this);
81 > markAsDisposed(this);
82 > }
84 > private _run() {
85 > const emptySet = this._dependenciesToBeRemoved; autorunImpl.ts
86 > this._dependenciesToBeRemoved = this._dependencies;
87 > this._dependencies = emptySet;
88 >
89 > this._state = AutorunState.upToDate;
90 >
91 > try {
92 > if (!this._disposed) {
93 > getLogger()?.handleAutorunStarted(this);
94 > const changeSummary = this._changeSummary!;
95 > const delayedStore = this._delayedStore;
96 > if (delayedStore !== undefined) {
97 this._delayedStore = undefined;
98 }
99 > try { autorunImpl.ts
100 > this._isRunning = true;
101 > if (this._changeTracker) {
102 this._changeTracker.beforeUpdate?.(this, changeSummary);
103 this._changeSummary = this._changeTracker.createChangeSummary(changeSummary); // Warning: external call!
104 }
105 > if (this._store !== undefined) { autorunImpl.ts
106 this._store.dispose();
107 this._store = undefined;
108 }
110 > this._runFn(this, changeSummary); // Warning: external call!
111 > } catch (e) {
112 onBugIndicatingError(e);
113 > } finally { autorunImpl.ts
114 > this._isRunning = false;
115 > if (delayedStore !== undefined) {
116 delayedStore.dispose();
117 }
118 > } autorunImpl.ts
119 > }
120 > } finally {
121 > if (!this._disposed) {
122 > getLogger()?.handleAutorunFinished(this);
123 > }
124 > // We don't want our observed observables to think that they are (not even temporarily) not being observed.
125 > // Thus, we only unsubscribe from observables that are definitely not read anymore.
126 > for (const o of this._dependenciesToBeRemoved) {
127 o.removeObserver(this); // Warning: external call!
128 }
129 > this._dependenciesToBeRemoved.clear(); autorunImpl.ts
130 > }
131 > }
133 > public toString(): string {
134 return `Autorun<${this.debugName}>`;
135 }
137 > // IObserver implementation
138 > public beginUpdate(_observable: IObservable<any>): void {
139 > if (this._state === AutorunState.upToDate) { autorunImpl.ts
140 > this._checkIterations();
141 > this._state = AutorunState.dependenciesMightHaveChanged;
142 > }
143 > this._updateCount++;
144 > }
146 > public endUpdate(_observable: IObservable<any>): void {
147 > try { autorunImpl.ts
148 > if (this._updateCount === 1) {
149 > this._iteration = 1;
150 > do {
151 > if (this._checkIterations()) {
152 return;
153 }
154 > if (this._state === AutorunState.dependenciesMightHaveChanged) { autorunImpl.ts
155 > this._state = AutorunState.upToDate; autorunImpl.ts
156 > for (const d of this._dependencies) {
157 > d.reportChanges(); // Warning: external call!
158 > if (this._state as AutorunState === AutorunState.stale) {
159 > // The other dependencies will refresh on demand autorunImpl.ts
160 > break;
161 > }
162 > } autorunImpl.ts
163 > }
165 > this._iteration++;
166 > if (this._state !== AutorunState.upToDate) {
167 > this._run(); // Warning: indirect external call! autorunImpl.ts
168 > }
169 > } while (this._state !== AutorunState.upToDate); autorunImpl.ts
170 > }
171 > } finally {
172 > this._updateCount--;
173 > }
174 >
175 > assertFn(() => this._updateCount >= 0);
176 > }
178 > public handlePossibleChange(observable: IObservable<any>): void {
179 if (this._state === AutorunState.upToDate && this._isDependency(observable)) {
180 this._checkIterations();
182 }
183 }
185 > public handleChange<T, TChange>(observable: IObservableWithChange<T, TChange>, change: TChange): void {
186 > if (this._isDependency(observable)) { autorunImpl.ts
187 > getLogger()?.handleAutorunDependencyChanged(this, observable, change);
188 > try {
189 > // Warning: external call!
190 > const shouldReact = this._changeTracker ? this._changeTracker.handleChange({
191 changedObservable: observable,
192 change,
193 // eslint-disable-next-line local/code-no-any-casts
194 didChange: (o): this is any => o === observable as any,
195 > }, this._changeSummary!) : true; autorunImpl.ts
196 > if (shouldReact) {
197 > this._checkIterations();
198 > this._state = AutorunState.stale;
199 > }
200 > } catch (e) {
201 onBugIndicatingError(e);
202 }
203 > } autorunImpl.ts
204 > }
206 > private _isDependency(observable: IObservableWithChange<any, any>): boolean {
207 > return this._dependencies.has(observable) && !this._dependenciesToBeRemoved.has(observable); autorunImpl.ts
208 > }
210 > // IReader implementation
211 >
212 > private _ensureNoRunning(): void {
213 > if (!this._isRunning) { throw new BugIndicatingError('The reader object cannot be used outside its compute function!'); } autorunImpl.ts
214 > }
216 > public readObservable<T>(observable: IObservable<T>): T {
217 > this._ensureNoRunning(); autorunImpl.ts
218 >
219 > // In case the run action disposes the autorun
220 > if (this._disposed) {
221 return observable.get(); // warning: external call!
222 }
224 > observable.addObserver(this); // warning: external call!
225 > const value = observable.get(); // warning: external call!
226 > this._dependencies.add(observable);
227 > this._dependenciesToBeRemoved.delete(observable);
228 > return value;
229 > }
231 > private _store: DisposableStore | undefined = undefined;
232 > get store(): DisposableStore {
233 > this._ensureNoRunning(); autorunImpl.ts
234 > if (this._disposed) {
235 throw new BugIndicatingError('Cannot access store after dispose');
236 }
238 > if (this._store === undefined) {
239 > this._store = new DisposableStore();
240 > }
241 > return this._store;
242 > }
244 > private _delayedStore: DisposableStore | undefined = undefined;
245 > get delayedStore(): DisposableStore {
246 > this._ensureNoRunning(); autorunImpl.ts
247 > if (this._disposed) {
248 throw new BugIndicatingError('Cannot access store after dispose');
249 }
251 > if (this._delayedStore === undefined) {
252 > this._delayedStore = new DisposableStore();
253 > }
254 > return this._delayedStore;
255 > }
257 > public debugGetState() {
258 return {
259 isRunning: this._isRunning,
264 };
265 }
267 > public debugRerun(): void {
268 if (!this._isRunning) {
269 this._run();
272 }
273 }
275 > private _checkIterations(): boolean {
276 > if (this._iteration > 100) { autorunImpl.ts
277 onBugIndicatingError(new BugIndicatingError(`Autorun '${this.debugName}' is stuck in an infinite update loop.`));
278 return true;
279 }
280 > return false; autorunImpl.ts
281 > }
282 > } autorunImpl.ts
src/vs/workbench/services/workingCopy/common/workingCopy.ts 221 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workingCopy.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { ISaveOptions, IRevertOptions, SaveReason, SaveSource } from '../../../common/editor.js';
9 > import { CancellationToken } from '../../../../base/common/cancellation.js';
10 > import { VSBufferReadable, VSBufferReadableStream } from '../../../../base/common/buffer.js';
11 >
12 > export const enum WorkingCopyCapabilities {
13 >
14 > /**
15 > * Signals no specific capability for the working copy.
16 > */
17 > None = 0,
18 >
19 > /**
20 > * Signals that the working copy requires
21 > * additional input when saving, e.g. an
22 > * associated path to save to.
23 > */
24 > Untitled = 1 << 1,
25 >
26 > /**
27 > * The working copy will not indicate that
28 > * it is dirty and unsaved content will be
29 > * discarded without prompting if closed.
30 > */
31 > Scratchpad = 1 << 2
32 > }
33 >
34 > /**
35 > * Data to be associated with working copy backups. Use
36 > * `IWorkingCopyBackupService.resolve(workingCopy)` to
37 > * retrieve the backup when loading the working copy.
38 > */
39 > export interface IWorkingCopyBackup {
40 >
41 > /**
42 > * Any serializable metadata to be associated with the backup.
43 > */
44 > meta?: IWorkingCopyBackupMeta;
45 >
46 > /**
47 > * The actual snapshot of the contents of the working copy at
48 > * the time the backup was made.
49 > */
50 > content?: VSBufferReadable | VSBufferReadableStream;
51 > }
52 >
53 > /**
54 > * Working copy backup metadata that can be associated
55 > * with the backup.
56 > *
57 > * Some properties may be reserved as outlined here and
58 > * cannot be used.
59 > */
60 > export interface IWorkingCopyBackupMeta {
61 >
62 > /**
63 > * Any property needs to be serializable through JSON.
64 > */
65 > [key: string]: unknown;
66 >
67 > /**
68 > * `typeId` is a reserved property that cannot be used
69 > * as backup metadata.
70 > */
71 > typeId?: never;
72 > }
73 >
74 > /**
75 > * @deprecated it is important to provide a type identifier
76 > * for working copies to enable all capabilities.
77 > */
78 > export const NO_TYPE_ID = '';
79 >
80 > /**
81 > * Every working copy has in common that it is identified by
82 > * a resource `URI` and a `typeId`. There can only be one
83 > * working copy registered with the same `URI` and `typeId`.
84 > */
85 > export interface IWorkingCopyIdentifier {
86 >
87 > /**
88 > * The type identifier of the working copy for grouping
89 > * working copies of the same domain together.
90 > *
91 > * There can only be one working copy for a given resource
92 > * and type identifier.
93 > */
94 > readonly typeId: string;
95 >
96 > /**
97 > * The resource of the working copy must be unique for
98 > * working copies of the same `typeId`.
99 > */
100 > readonly resource: URI;
101 > }
102 >
103 > export interface IWorkingCopySaveEvent {
104 >
105 > /**
106 > * The reason why the working copy was saved.
107 > */
108 > readonly reason?: SaveReason;
109 >
110 > /**
111 > * The source of the working copy save request.
112 > */
113 > readonly source?: SaveSource;
114 > }
115 >
116 > /**
117 > * A working copy is an abstract concept to unify handling of
118 > * data that can be worked on (e.g. edited) in an editor.
119 > *
120 > *
121 > * A working copy resource may be the backing store of the data
122 > * (e.g. a file on disk), but that is not a requirement. If
123 > * your working copy is file based, consider to use the
124 > * `IFileWorkingCopy` instead that simplifies a lot of things
125 > * when working with file based working copies.
126 > */
127 > export interface IWorkingCopy extends IWorkingCopyIdentifier {
128 >
129 > /**
130 > * Human readable name of the working copy.
131 > */
132 > readonly name: string;
133 >
134 > /**
135 > * The capabilities of the working copy.
136 > */
137 > readonly capabilities: WorkingCopyCapabilities;
138 >
139 >
140 > //#region Events
141 >
142 > /**
143 > * Used by the workbench to signal if the working copy
144 > * is dirty or not. Typically a working copy is dirty
145 > * once changed until saved or reverted.
146 > */
147 > readonly onDidChangeDirty: Event<void>;
148 >
149 > /**
150 > * Used by the workbench e.g. to trigger auto-save
151 > * (unless this working copy is untitled) and backups.
152 > */
153 > readonly onDidChangeContent: Event<void>;
154 >
155 > /**
156 > * Used by the workbench e.g. to track local history
157 > * (unless this working copy is untitled).
158 > */
159 > readonly onDidSave: Event<IWorkingCopySaveEvent>;
160 >
161 > //#endregion
162 >
163 >
164 > //#region Dirty Tracking
165 >
166 > /**
167 > * Indicates that the file has unsaved changes
168 > * and should confirm before closing.
169 > */
170 > isDirty(): boolean;
171 >
172 > /**
173 > * Indicates that the file has unsaved changes.
174 > * Used for backup tracking and accounts for
175 > * working copies that are never dirty e.g.
176 > * scratchpads.
177 > */
178 > isModified(): boolean;
179 >
180 > //#endregion
181 >
182 >
183 > //#region Save / Backup
184 >
185 > /**
186 > * The delay in milliseconds to wait before triggering
187 > * a backup after the content of the model has changed.
188 > *
189 > * If not configured, a sensible default will be taken
190 > * based on user settings.
191 > */
192 > readonly backupDelay?: number;
193 >
194 > /**
195 > * The workbench may call this method often after it receives
196 > * the `onDidChangeContent` event for the working copy. The motivation
197 > * is to allow to quit VSCode with dirty working copies present.
198 > *
199 > * Providers of working copies should use `IWorkingCopyBackupService.resolve(workingCopy)`
200 > * to retrieve the backup metadata associated when loading the working copy.
201 > *
202 > * @param token support for cancellation
203 > */
204 > backup(token: CancellationToken): Promise<IWorkingCopyBackup>;
205 >
206 > /**
207 > * Asks the working copy to save. If the working copy was dirty, it is
208 > * expected to be non-dirty after this operation has finished.
209 > *
210 > * @returns `true` if the operation was successful and `false` otherwise.
211 > */
212 > save(options?: ISaveOptions): Promise<boolean>;
213 >
214 > /**
215 > * Asks the working copy to revert. If the working copy was dirty, it is
216 > * expected to be non-dirty after this operation has finished.
217 > */
218 > revert(options?: IRevertOptions): Promise<void>;
219 >
220 > //#endregion
221 > }
src/vs/workbench/contrib/chat/common/chatService/chatServiceTelemetry.ts 217 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatServiceTelemetry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../base/common/uri.js';
7 > import { isLocation } from '../../../../../editor/common/languages.js';
8 > import { escapeModelIdForTelemetry, ITelemetryService } from '../../../../../platform/telemetry/common/telemetry.js';
9 > import { IChatAgentData } from '../participants/chatAgents.js';
10 > import { ChatRequestModel, IChatRequestVariableData } from '../model/chatModel.js';
11 > import { ChatRequestAgentSubcommandPart, ChatRequestSlashCommandPart } from '../requestParser/chatParserTypes.js';
12 > import { ChatAgentVoteDirection, ChatCopyKind, IChatSendRequestOptions, IChatUserActionEvent } from './chatService.js';
13 > import { isImageVariableEntry } from '../attachments/chatVariableEntries.js';
14 > import { ChatAgentLocation, ChatModeKind, ChatPermissionLevel } from '../constants.js';
15 > import { ILanguageModelsService } from '../languageModels.js';
16 > import { chatSessionResourceToId, getChatSessionType } from '../model/chatUri.js';
17 > import { isRemoteAgentHostSessionType, parseRemoteAgentHostHarness } from '../../../../../platform/agentHost/common/agentHostSessionType.js';
18 >
19 > type ChatVoteEvent = {
20 > direction: 'up' | 'down';
21 > agentId: string;
22 > command: string | undefined;
23 > };
24 >
25 > type ChatVoteClassification = {
26 > direction: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the user voted up or down.' };
27 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the chat agent that this vote is for.' };
28 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the slash command that this vote is for.' };
29 > owner: 'roblourens';
30 > comment: 'Provides insight into the performance of Chat agents.';
31 > };
32 >
33 > type ChatCopyEvent = {
34 > copyKind: 'action' | 'toolbar';
35 > agentId: string;
36 > command: string | undefined;
37 > };
38 >
39 > type ChatCopyClassification = {
40 > copyKind: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'How the copy was initiated.' };
41 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the chat agent that the copy acted on.' };
42 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the slash command the copy acted on.' };
43 > owner: 'roblourens';
44 > comment: 'Provides insight into the usage of Chat features.';
45 > };
46 >
47 > type ChatInsertEvent = {
48 > newFile: boolean;
49 > agentId: string;
50 > command: string | undefined;
51 > };
52 >
53 > type ChatInsertClassification = {
54 > newFile: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the code was inserted into a new untitled file.' };
55 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the chat agent that this insertion is for.' };
56 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the slash command that this insertion is for.' };
57 > owner: 'roblourens';
58 > comment: 'Provides insight into the usage of Chat features.';
59 > };
60 >
61 > type ChatApplyEvent = {
62 > newFile: boolean;
63 > agentId: string;
64 > command: string | undefined;
65 > codeMapper: string | undefined;
66 > editsProposed: boolean;
67 > };
68 >
69 > type ChatApplyClassification = {
70 > newFile: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the code was inserted into a new untitled file.' };
71 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the chat agent that this insertion is for.' };
72 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the slash command that this insertion is for.' };
73 > codeMapper: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The code mapper that wa used to compute the edit.' };
74 > editsProposed: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether there was a change proposed to the user.' };
75 > owner: 'aeschli';
76 > comment: 'Provides insight into the usage of Chat features.';
77 > };
78 >
79 > type ChatFollowupEvent = {
80 > agentId: string;
81 > command: string | undefined;
82 > };
83 >
84 > type ChatFollowupClassification = {
85 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the related chat agent.' };
86 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the related slash command.' };
87 > owner: 'roblourens';
88 > comment: 'Provides insight into the usage of Chat features.';
89 > };
90 >
91 > type ChatTerminalEvent = {
92 > languageId: string;
93 > agentId: string;
94 > command: string | undefined;
95 > };
96 >
97 > type ChatTerminalClassification = {
98 > languageId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The language of the code that was run in the terminal.' };
99 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the related chat agent.' };
100 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the related slash command.' };
101 > owner: 'roblourens';
102 > comment: 'Provides insight into the usage of Chat features.';
103 > };
104 >
105 > type ChatFollowupsRetrievedEvent = {
106 > agentId: string;
107 > command: string | undefined;
108 > numFollowups: number;
109 > };
110 >
111 > type ChatFollowupsRetrievedClassification = {
112 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the related chat agent.' };
113 > command: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the related slash command.' };
114 > numFollowups: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The number of followup prompts returned by the agent.' };
115 > owner: 'roblourens';
116 > comment: 'Provides insight into the usage of Chat features.';
117 > };
118 >
119 > type ChatEditHunkEvent = {
120 > agentId: string;
121 > outcome: 'accepted' | 'rejected';
122 > lineCount: number;
123 > hasRemainingEdits: boolean;
124 > requestId: string;
125 > modelId: string;
126 > modeId: string;
127 > };
128 >
129 > type ChatEditHunkClassification = {
130 > agentId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the related chat agent.' };
131 > outcome: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The outcome of the edit hunk action.' };
132 > lineCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The number of lines in the relevant change.' };
133 > hasRemainingEdits: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether there are remaining edits in the file after this action.' };
134 > requestId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The ID of the chat request that produced the edit.' };
135 > modelId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The AI model used to generate the edit.' };
136 > modeId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The chat mode used for the request (e.g. ask, edit, agent).' };
137 > owner: 'roblourens';
138 > comment: 'Provides insight into the usage of Chat features.';
139 > };
140 >
141 > export type ChatProviderInvokedEvent = {
142 > timeToFirstProgress: number | undefined;
143 > totalTime: number | undefined;
144 > result: 'success' | 'error' | 'errorWithOutput' | 'cancelled' | 'filtered';
145 > requestType: 'string' | 'followup' | 'slashCommand';
146 > chatSessionId: string;
147 > requestId: string;
148 > agent: string;
149 > agentExtensionId: string | undefined;
150 > slashCommand: string | undefined;
151 > location: ChatAgentLocation;
152 > citations: number;
153 > numCodeBlocks: number;
154 > isParticipantDetected: boolean;
155 > enableCommandDetection: boolean;
156 > attachmentKinds: string[];
157 > model: string | undefined;
158 > permissionLevel: ChatPermissionLevel | undefined;
159 > chatMode: string | undefined;
160 > sessionType: string | undefined;
161 > harness: string | undefined;
162 > };
163 >
164 > export type ChatProviderInvokedClassification = {
165 > timeToFirstProgress: { classification: 'SystemMetaData'; purpose: 'PerformanceAndHealth'; comment: 'The time in milliseconds from invoking the provider to getting the first data.' };
166 > totalTime: { classification: 'SystemMetaData'; purpose: 'PerformanceAndHealth'; comment: 'The total time it took to run the provider\'s `provideResponseWithProgress`.' };
167 > result: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether invoking the ChatProvider resulted in an error.' };
168 > requestType: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The type of request that the user made.' };
169 > chatSessionId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'A random ID for the session.' };
170 > requestId: { classification: 'SystemMetaData'; purpose: 'PerformanceAndHealth'; comment: 'The identifier of the chat request, used to correlate workbench and agent host telemetry.' };
171 > agent: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The type of agent used.' };
172 > agentExtensionId: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The extension that contributed the agent.' };
173 > slashCommand?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The type of slashCommand used.' };
174 > location: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The location at which chat request was made.' };
175 > citations: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The number of public code citations that were returned with the response.' };
176 > numCodeBlocks: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The number of code blocks in the response.' };
177 > isParticipantDetected: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether the participant was automatically detected.' };
178 > enableCommandDetection: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Whether participation detection was disabled for this invocation.' };
179 > attachmentKinds: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The types of variables/attachments that the user included with their query.' };
180 > model: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The model used to generate the response.' };
181 > permissionLevel: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The tool auto-approval permission level selected in the permission picker (default, assisted, autoApprove, or autopilot). Undefined when the picker is not applicable (e.g. ask mode or API-driven requests).' };
182 > chatMode: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The chat mode used for the request. Built-in modes (ask, agent, edit), extension-contributed names (e.g. Plan), or a hashed identifier for user-created custom agents.' };
183 > sessionType: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The session type scheme (e.g. vscodeLocalChatSession for local, or remote session scheme).' };
184 > harness: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'For remote agent host sessions, the underlying harness/provider (e.g. copilotcli, claude, codex) so remote activity can be split by harness. Undefined for non-remote sessions.' };
185 > owner: 'roblourens';
186 > comment: 'Provides insight into the performance of Chat agents.';
187 > };
188 >
189 > export class ChatServiceTelemetry {
190 > constructor(
191 > @ITelemetryService private readonly telemetryService: ITelemetryService, chatServiceTelemetry.ts
192 > ) { }
194 > notifyUserAction(action: IChatUserActionEvent): void {
195 if (action.action.kind === 'vote') {
196 this.telemetryService.publicLog2<ChatVoteEvent, ChatVoteClassification>('interactiveSessionVote', {
242 }
243 }
245 > retrievedFollowups(agentId: string, command: string | undefined, numFollowups: number): void {
246 this.telemetryService.publicLog2<ChatFollowupsRetrievedEvent, ChatFollowupsRetrievedClassification>('chatFollowupsRetrieved', {
247 agentId,
250 });
251 }
253 >
254 function getCodeBlocks(text: string): string[] {
255 const lines = text.split('\n');
274 return codeBlockLanguages;
275 }
277 > export class ChatRequestTelemetry {
278 > private isComplete = false;
279 >
280 > constructor(private readonly opts: {
281 agent: IChatAgentData;
282 agentSlashCommandPart: ChatRequestAgentSubcommandPart | undefined;
290 @ILanguageModelsService private readonly languageModelsService: ILanguageModelsService
291 ) { }
293 > complete({ timeToFirstProgress, totalTime, result, requestType, request, detectedAgent }: {
294 timeToFirstProgress: number | undefined;
295 totalTime: number | undefined;
328 });
329 }
331 > private attachmentKindsForTelemetry(variableData: IChatRequestVariableData): string[] {
332 // this shows why attachments still have to be cleaned up somewhat
333 return variableData.variables.map(v => {
366 });
367 }
369 > private resolveModelId(userSelectedModelId: string | undefined): string | undefined {
370 return userSelectedModelId && this.languageModelsService.lookupLanguageModel(userSelectedModelId)?.id;
371 }
373 >
374 function getChatSessionTypeForTelemetry(sessionResource: URI): string {
375 const sessionType = getChatSessionType(sessionResource);
378 return isRemoteAgentHostSessionType(sessionType) ? 'remote-agent-host' : sessionType;
379 }
381 > /**
382 > * For remote agent host sessions, the underlying harness/provider so remote
383 > * activity can be split by harness (the collapsed sessionType cannot). See
384 > * telemetry gap #2 in #8209. Undefined for non-remote sessions.
385 > */
386 function getHarnessForTelemetry(sessionResource: URI): string | undefined {
387 return parseRemoteAgentHostHarness(getChatSessionType(sessionResource));
src/vs/base/test/common/virtualScheduling/processor.ts 215 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- processor.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../common/cancellation.js';
7 > import { Disposable, DisposableStore, IDisposable } from '../../../common/lifecycle.js';
8 > import { Embedding, nextMacrotask } from './embedding.js';
9 > import { TimeApi } from './timeApi.js';
10 > import { ROOT_TRACE, TraceContext } from './trace.js';
11 > import { EventSource, VirtualClock, VirtualEvent, VirtualTime } from './virtualClock.js';
12 >
13 > // ============================================================================
14 > // Termination policy
15 > // ============================================================================
16 >
17 > /**
18 > * When a {@link Run} should terminate.
19 > *
20 > * Greenfield design choice: termination is *always* explicit. There is no
21 > * "bare run()" that terminates on first empty queue, because that creates a
22 > * race with the caller's microtask chain (the run can resolve before the
23 > * caller's `.then` has had a chance to schedule).
24 > */
25 > export type TerminationPolicy =
26 > /** Resolve as soon as the virtual queue is empty. */
27 > | { readonly kind: 'idle' }
28 > /** Resolve when the token is cancelled AND the queue is empty. */
29 > | { readonly kind: 'token'; readonly token: CancellationToken }
30 > /** Resolve when virtual time has reached `time` and all events scheduled
31 > * at or before `time` have been processed. A sentinel event at `time`
32 > * is scheduled by the processor so virtual time always reaches it. */
33 > | { readonly kind: 'time'; readonly time: VirtualTime };
34 >
35 > export const untilIdle: TerminationPolicy = { kind: 'idle' };
36 > export function untilToken(token: CancellationToken): TerminationPolicy { return { kind: 'token', token }; }
37 > export function untilTime(time: VirtualTime): TerminationPolicy { return { kind: 'time', time }; }
38 >
39 > export interface RunOptions {
40 > readonly until: TerminationPolicy;
41 > /** Maximum number of virtual events this run will execute. Default: 100. */
42 > readonly maxEvents?: number;
43 > /** Maximum causal-trace depth this run will tolerate. Useful for catching
44 > * runaway self-rescheduling timers. */
45 > readonly maxTraceDepth?: number;
46 > }
47 >
48 > // ============================================================================
49 > // Run — internal state for a single processor.run() invocation
50 > // ============================================================================
51 >
52 > type RunStatus = 'continue' | 'done' | { readonly error: Error };
53 >
54 > class Run {
55 > private static _idCounter = 0;
56 > public readonly id = ++Run._idCounter;
57 >
58 > public readonly promise: Promise<void>;
59 > private _resolve!: () => void;
60 > private _reject!: (e: Error) => void;
61 > private _settled = false;
62 > public get settled(): boolean { return this._settled; }
63 >
64 > constructor(
65 public readonly options: RunOptions,
66 public readonly executedAtStart: number,
69 this.promise = new Promise<void>((res, rej) => { this._resolve = res; this._reject = rej; });
70 }
72 > settle(error?: Error): void {
73 if (this._settled) { return; }
74 this._settled = true;
75 if (error) { this._reject(error); } else { this._resolve(); }
76 }
78 > evaluate(clock: VirtualClock, executedTotal: number, makeOverflow: () => Error): RunStatus {
79 const local = executedTotal - this.executedAtStart;
80 if (local >= this.maxEvents && clock.hasEvents) {
98 }
99 }
100 > } processor.ts
101 >
102 > // ============================================================================
103 > // Step outcome — what the pure state machine tells the trampoline
104 > // ============================================================================
105 >
106 > type StepOutcome =
107 > /** Either a virtual event was executed, or a run was rejected for a
108 > * bookkeeping reason (depth/event overflow). The trampoline should let
109 > * the embedding decide how to reach the next step. */
110 > | 'progress'
111 > /** No actionable event under any active deadline. The trampoline should
112 > * park until something wakes the processor. */
113 > | 'park'
114 > /** No active runs. The trampoline should stop driving. */
115 > | 'quiesce';
116 >
117 > // ============================================================================
118 > // VirtualTimeProcessor
119 > // ============================================================================
120 >
121 > export interface VirtualTimeProcessorOptions {
122 > readonly defaultMaxEvents?: number;
123 > }
124 >
125 > /**
126 > * # VirtualTimeProcessor
127 > *
128 > * Drives a {@link VirtualClock} from the host event loop. This is the
129 > * **embedding** of a small virtual event loop into the host event loop.
130 > *
131 > * ## Responsibilities, separated
132 > *
133 > * - {@link _step} is a *pure* state-machine advance. It reads the clock,
134 > * decides what to do, optionally executes one virtual event, and returns
135 > * a {@link StepOutcome}. It never touches host time.
136 > *
137 > * - {@link _drive} is the *trampoline*. It calls `_step` and lets the
138 > * {@link Embedding} decide whether to loop in place (`'continueSync'`)
139 > * or schedule the next iteration on the host (`'cbScheduled'`). It is
140 > * the only code that touches host time.
141 > *
142 > * - {@link Run} carries the user's termination predicate. Runs are pure
143 > * over `_step`'s observations; they never schedule.
144 > *
145 > * ## Invariants
146 > *
147 > * 1. **Single driver.** At any moment at most one `_drive` invocation is
148 > * active per processor (the `_inDrive` guard).
149 > *
150 > * 2. **Step is pure w.r.t. host time.** `_step` only reads the clock,
151 > * mutates the run set via settling, and synchronously runs at most one
152 > * virtual event. It never calls into a host time API.
153 > *
154 > * 3. **Embedding chooses the host primitive.** Whether the next step runs
155 > * inline, after a microtask drain, or on a paint frame is entirely the
156 > * embedding's decision — *per event*.
157 > *
158 > * 4. **Park is breakable.** While parked, the processor wakes on
159 > * {@link VirtualClock.onEventScheduled}, on a token cancellation, and
160 > * on a new run being added.
161 > *
162 > * 5. **Disposal is terminal.** After dispose, all runs are rejected and
163 > * `_step`/`_drive` short-circuit to `'quiesce'`.
164 > *
165 > * ## On the trace-reset sink
166 > *
167 > * The trace context's deferred reset (see {@link TraceContext.runAsHandler})
168 > * needs a "fire after the microtask closure" primitive. The processor passes
169 > * its *own* {@link nextMacrotask} as that sink, so the reset goes through
170 > * the same primitive the embedding uses for its own host hops. This removes
171 > * any race between the processor's hops and the trace-reset timer.
172 > */
173 > export class VirtualTimeProcessor extends Disposable {
174 >
175 > private readonly _runs = new Map<Run, IDisposable>();
176 > private readonly _history: VirtualEvent[] = [];
177 > private _executedTotal = 0;
178 > private _disposed = false;
179 >
180 > private _inDrive = false;
181 > private _parkCleanup: IDisposable | undefined;
182 >
183 > private readonly _defaultMaxEvents: number;
184 >
185 > public get history(): readonly VirtualEvent[] { return this._history; }
186 > public get executedTotal(): number { return this._executedTotal; }
187 >
188 > constructor(
189 private readonly _clock: VirtualClock,
190 private readonly _embedding: Embedding,
196 this._register({ dispose: () => this._onDispose() });
197 }
198 > processor.ts
199 > // ---- Public API -----------------------------------------------------
200 >
201 > /** Start a run with the given termination policy. */
202 > run(options: RunOptions): Promise<void> {
203 const run = new Run(options, this._executedTotal, options.maxEvents ?? this._defaultMaxEvents);
204 const cleanup = new DisposableStore();
226 return run.promise;
227 }
228 > processor.ts
229 > // ---- The pure step --------------------------------------------------
230 >
231 > private _step(): StepOutcome {
232 if (this._disposed) { return 'quiesce'; }
233
254 return 'progress';
255 }
256 > processor.ts
257 > private _executeOne(event: VirtualEvent): void {
258 try {
259 TraceContext.instance.runAsHandler(
280 }
281 }
282 > processor.ts
283 > // ---- The trampoline -------------------------------------------------
284 >
285 > private readonly _drive = (): void => {
286 > if (this._inDrive) { return; }
287 > this._inDrive = true;
288 > try {
289 > while (true) {
290 > const outcome = this._step();
291 > if (outcome === 'quiesce') { return; }
292 > if (outcome === 'park') { this._park(); return; } processor.ts
293 >
294 > // 'progress': read the next event so the embedding can pick a
295 > // per-event primitive. If there is none, loop and let the next
296 > // `_step` decide between 'park' and 'quiesce'.
297 > const next = this._clock.peekNext();
298 > if (next === undefined) { continue; }
299 > processor.ts
300 > const choice = this._embedding(next, this._drive);
301 > if (choice === 'cbScheduled') { return; }
302 > // 'continueSync': loop in place. processor.ts
303 > }
304 > } finally {
305 > this._inDrive = false;
306 > }
307 > };
308 >
309 > // ---- Park & wake ----------------------------------------------------
310 >
311 > private _park(): void {
312 this._unpark();
313 const store = new DisposableStore();
315 this._parkCleanup = store;
316 }
317 > processor.ts
318 > private _unpark(): void {
319 this._parkCleanup?.dispose();
320 this._parkCleanup = undefined;
321 }
322 > processor.ts
323 > private _wake(): void {
324 if (this._disposed) { return; }
325 this._unpark();
337 nextMacrotask(this._realApi, this._drive);
338 }
339 > processor.ts
340 > // ---- Run lifecycle --------------------------------------------------
341 >
342 > private _settleFinishedRuns(): void {
343 for (const run of [...this._runs.keys()]) {
344 if (run.settled) { continue; }
351 }
352 }
353 > processor.ts
354 > private _settleRun(run: Run, error?: Error): void {
355 const cleanup = this._runs.get(run);
356 if (!cleanup) { return; }
359 run.settle(error);
360 }
361 > processor.ts
362 > private _buildOverflow(run: Run): Error {
363 const local = this._executedTotal - run.executedAtStart;
364 return new Error(
367 );
368 }
369 > processor.ts
370 > private _buildDepthOverflow(run: Run, depth: number): Error {
371 return new Error(
372 `[VirtualTimeProcessor] Run #${run.id} exceeded maxTraceDepth (${run.options.maxTraceDepth}) — ` +
375 );
376 }
377 > processor.ts
378 > private _onDispose(): void {
379 this._disposed = true;
380 this._unpark();
382 for (const run of [...this._runs.keys()]) { this._settleRun(run, err); }
383 }
384 > } processor.ts
src/vs/workbench/contrib/chat/common/promptSyntax/promptFileParser.ts 215 covered LOC · 38 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptFileParser.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Iterable } from '../../../../../base/common/iterator.js';
7 > import { dirname, joinPath } from '../../../../../base/common/resources.js';
8 > import { splitLinesIncludeSeparators } from '../../../../../base/common/strings.js';
9 > import { URI } from '../../../../../base/common/uri.js';
10 > import { parse, YamlNode, YamlParseError } from '../../../../../base/common/yaml.js';
11 > import { Range } from '../../../../../editor/common/core/range.js';
12 > import { PositionOffsetTransformer } from '../../../../../editor/common/core/text/positionToOffsetImpl.js';
13 >
14 > export class PromptFileParser {
15 > constructor() {
16 }
18 > public parse(uri: URI, content: string): ParsedPromptFile {
19 const linesWithEOL = splitLinesIncludeSeparators(content);
20 if (linesWithEOL.length === 0) {
43 return new ParsedPromptFile(uri, header, body);
44 }
46 >
47 >
48 > export class ParsedPromptFile {
49 > constructor(public readonly uri: URI, public readonly header?: PromptHeader, public readonly body?: PromptBody) {
50 }
52 >
53 > export interface ParseError {
54 > readonly message: string;
55 > readonly range: Range;
56 > readonly code: string;
57 > }
58 >
59 > interface ParsedHeader {
60 > readonly node: YamlNode | undefined;
61 > readonly errors: ParseError[];
62 > readonly attributes: IHeaderAttribute[];
63 > }
64 >
65 > export namespace PromptHeaderAttributes {
66 > export const name = 'name';
67 > export const description = 'description';
68 > export const agent = 'agent';
69 > export const mode = 'mode';
70 > export const model = 'model';
71 > export const applyTo = 'applyTo';
72 > export const paths = 'paths';
73 > export const tools = 'tools';
74 > export const handOffs = 'handoffs';
75 > export const advancedOptions = 'advancedOptions';
76 > export const argumentHint = 'argument-hint';
77 > export const excludeAgent = 'excludeAgent';
78 > export const target = 'target';
79 > export const infer = 'infer';
80 > export const license = 'license';
81 > export const compatibility = 'compatibility';
82 > export const metadata = 'metadata';
83 > export const agents = 'agents';
84 > export const userInvocable = 'user-invocable';
85 > export const disableModelInvocation = 'disable-model-invocation';
86 > export const hooks = 'hooks';
87 > export const context = 'context';
88 > }
89 >
90 > export class PromptHeader {
91 > private _parsed: ParsedHeader | undefined;
92 >
93 > constructor(public readonly range: Range, public readonly uri: URI, private readonly linesWithEOL: string[]) {
94 }
96 > private get _parsedHeader(): ParsedHeader {
97 if (this._parsed === undefined) {
98 const yamlErrors: YamlParseError[] = [];
137 return this._parsed;
138 }
140 > public get attributes(): IHeaderAttribute[] {
141 return this._parsedHeader.attributes;
142 }
144 > public getAttribute(key: string): IHeaderAttribute | undefined {
145 return this._parsedHeader.attributes.find(attr => attr.key === key);
146 }
148 > public get errors(): ParseError[] {
149 return this._parsedHeader.errors;
150 }
152 > private getStringAttribute(key: string): string | undefined {
153 const attribute = this._parsedHeader.attributes.find(attr => attr.key === key);
154 if (attribute?.value.type === 'scalar') {
157 return undefined;
158 }
160 > public get name(): string | undefined {
161 return this.getStringAttribute(PromptHeaderAttributes.name);
162 }
164 > public get description(): string | undefined {
165 return this.getStringAttribute(PromptHeaderAttributes.description);
166 }
168 > public get agent(): string | undefined {
169 return this.getStringAttribute(PromptHeaderAttributes.agent) ?? this.getStringAttribute(PromptHeaderAttributes.mode);
170 }
172 > public get model(): readonly string[] | undefined {
173 return this.getStringOrStringArrayAttribute(PromptHeaderAttributes.model);
174 }
176 > public get applyTo(): string | undefined {
177 return this.getStringAttribute(PromptHeaderAttributes.applyTo);
178 }
180 > /**
181 > * Gets the 'paths' attribute from the header.
182 > * The `paths` field supports a list of glob patterns that scope the instruction
183 > * to specific files (used by Claude rules). Returns a string array or undefined.
184 > */
185 > public get paths(): readonly string[] | undefined {
186 return this.getStringOrStringArrayAttribute(PromptHeaderAttributes.paths);
187 }
189 > public get argumentHint(): string | undefined {
190 return this.getStringAttribute(PromptHeaderAttributes.argumentHint);
191 }
193 > public get target(): string | undefined {
194 return this.getStringAttribute(PromptHeaderAttributes.target);
195 }
197 > public get infer(): boolean | undefined {
198 return this.getBooleanAttribute(PromptHeaderAttributes.infer);
199 }
201 > public get tools(): string[] | undefined {
202 const toolsAttribute = this._parsedHeader.attributes.find(attr => attr.key === PromptHeaderAttributes.tools);
203 if (!toolsAttribute) {
219 return undefined;
220 }
222 > public get handOffs(): IHandOff[] | undefined {
223 const handoffsAttribute = this._parsedHeader.attributes.find(attr => attr.key === PromptHeaderAttributes.handOffs);
224 if (!handoffsAttribute) {
268 return undefined;
269 }
271 > private getStringArrayAttribute(key: string): string[] | undefined {
272 const attribute = this._parsedHeader.attributes.find(attr => attr.key === key);
273 if (!attribute) {
285 return undefined;
286 }
288 > private getStringOrStringArrayAttribute(key: string): readonly string[] | undefined {
289 const attribute = this._parsedHeader.attributes.find(attr => attr.key === key);
290 if (!attribute) {
305 return undefined;
306 }
308 > public get agents(): string[] | undefined {
309 return this.getStringArrayAttribute(PromptHeaderAttributes.agents);
310 }
312 > public get userInvocable(): boolean | undefined {
313 return this.getBooleanAttribute(PromptHeaderAttributes.userInvocable);
314 }
316 > public get disableModelInvocation(): boolean | undefined {
317 return this.getBooleanAttribute(PromptHeaderAttributes.disableModelInvocation);
318 }
320 > public get context(): string | undefined {
321 return this.getStringAttribute(PromptHeaderAttributes.context);
322 }
324 > /**
325 > * Gets the raw 'hooks' attribute value from the header.
326 > * Returns the YAML map value if present, or undefined. The caller is
327 > * responsible for converting this to `ChatRequestHooks` via
328 > * {@link parseSubagentHooksFromYaml}.
329 > */
330 > public get hooksRaw(): IMapValue | undefined {
331 const attr = this._parsedHeader.attributes.find(a => a.key === PromptHeaderAttributes.hooks);
332 if (attr?.value.type === 'map') {
335 return undefined;
336 }
338 > private getBooleanAttribute(key: string): boolean | undefined {
339 const attribute = this._parsedHeader.attributes.find(attr => attr.key === key);
340 if (attribute?.value.type === 'scalar') {
343 return undefined;
344 }
346 >
347 function parseBoolean(stringValue: IScalarValue): boolean | undefined {
348 if (stringValue.value === 'true') {
353 return undefined;
354 }
356 > export interface IHandOff {
357 > readonly agent: string;
358 > readonly label: string;
359 > readonly prompt: string;
360 > readonly send?: boolean;
361 > readonly showContinueOn?: boolean; // treated exactly like send (optional boolean)
362 > readonly model?: string; // qualified model name to switch to (e.g., "GPT-5 (copilot)")
363 > }
364 >
365 > export interface IHeaderAttribute {
366 > readonly range: Range;
367 > readonly key: string;
368 > readonly value: IValue;
369 > }
370 >
371 > export interface IScalarValue {
372 > readonly type: 'scalar';
373 > readonly value: string;
374 > readonly range: Range;
375 > readonly format: 'single' | 'double' | 'none' | 'literal' | 'folded';
376 > }
377 >
378 > export interface ISequenceValue {
379 > readonly type: 'sequence';
380 > readonly items: readonly IValue[];
381 > readonly range: Range;
382 > }
383 >
384 > export interface IMapValue {
385 > readonly type: 'map';
386 > readonly properties: { key: IScalarValue; value: IValue }[];
387 > readonly range: Range;
388 > }
389 >
390 > export type IValue = IScalarValue | ISequenceValue | IMapValue;
391 >
392 >
393 > interface ParsedBody {
394 > readonly fileReferences: readonly IBodyFileReference[];
395 > readonly variableReferences: readonly IBodyVariableReference[];
396 > readonly bodyOffset: number;
397 > }
398 >
399 > export class PromptBody {
400 > private _parsed: ParsedBody | undefined;
401 >
402 > constructor(public readonly range: Range, private readonly linesWithEOL: string[], public readonly uri: URI) {
403 }
405 > public get fileReferences(): readonly IBodyFileReference[] {
406 return this.getParsedBody().fileReferences;
407 }
409 > public get variableReferences(): readonly IBodyVariableReference[] {
410 return this.getParsedBody().variableReferences;
411 }
413 > public get offset(): number {
414 return this.getParsedBody().bodyOffset;
415 }
417 > private getParsedBody(): ParsedBody {
418 if (this._parsed === undefined) {
419 const markdownLinkRanges: Range[] = [];
519 return this._parsed;
520 }
522 > public getContent(): string {
523 return this.linesWithEOL.slice(this.range.startLineNumber - 1, this.range.endLineNumber - 1).join('');
524 }
526 > public resolveFilePath(path: string): URI | undefined {
527 try {
528 if (path.startsWith('/')) {
538 }
539 }
541 >
542 > export interface IBodyFileReference {
543 > readonly content: string;
544 > readonly range: Range;
545 > readonly isMarkdownLink: boolean;
546 > }
547 >
548 > export interface IBodyVariableReference {
549 > readonly name: string;
550 > readonly range: Range;
551 > readonly offset: number;
552 > readonly fullLength: number;
553 > }
554 >
555 > /**
556 > * Parses a comma-separated list of values into an array of strings.
557 > * Values can be unquoted or quoted (single or double quotes).
558 > *
559 > * @param input A string containing comma-separated values
560 > * @returns An ISequenceValue containing the parsed values and their ranges
561 > */
562 > export function parseCommaSeparatedList(stringValue: IScalarValue): ISequenceValue {
563 const result: IScalarValue[] = [];
564 const input = stringValue.value;
625 return { type: 'sequence', items: result, range: stringValue.range };
626 }
628 > /**
629 > * Returns the effective `applyTo` pattern for an instruction file.
630 > * Claude rules use `paths` (defaulting to `**`), while regular instructions use `applyTo`.
631 > */
632 > export function evaluateApplyToPattern(header: PromptHeader | undefined, isClaudeRules: boolean): string | undefined {
633 if (isClaudeRules) {
634 return header?.paths?.join(', ') ?? '**';
src/vs/base/common/filters.ts 212 covered LOC · 47 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- filters.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from './charCode.js';
7 > import { LRUCache } from './map.js';
8 > import { getKoreanAltChars } from './naturalLanguage/korean.js';
9 > import { tryNormalizeToBase } from './normalization.js';
10 > import * as strings from './strings.js';
11 >
12 > export interface IFilter {
13 > // Returns null if word doesn't match.
14 > (word: string, wordToMatchAgainst: string): IMatch[] | null;
15 > }
16 >
17 > export interface IMatch {
18 > start: number;
19 > end: number;
20 > }
21 >
22 > // Combined filters
23 >
24 > /**
25 > * @returns A filter which combines the provided set
26 > * of filters with an or. The *first* filters that
27 > * matches defined the return value of the returned
28 > * filter.
29 > */
30 > export function or(...filter: IFilter[]): IFilter {
31 > return function (word: string, wordToMatchAgainst: string): IMatch[] | null {
32 for (let i = 0, len = filter.length; i < len; i++) {
33 const match = filter[i](word, wordToMatchAgainst);
38 return null;
39 };
40 > } filters.ts
41 >
42 > // Prefix
43 >
44 > export const matchesStrictPrefix: IFilter = _matchesPrefix.bind(undefined, false);
45 > export const matchesPrefix: IFilter = _matchesPrefix.bind(undefined, true);
46 >
47 function _matchesPrefix(ignoreCase: boolean, word: string, wordToMatchAgainst: string): IMatch[] | null {
48 if (!wordToMatchAgainst || wordToMatchAgainst.length < word.length) {
63 return word.length > 0 ? [{ start: 0, end: word.length }] : [];
64 }
65 > filters.ts
66 > // Contiguous Substring
67 >
68 > export function matchesContiguousSubString(word: string, wordToMatchAgainst: string): IMatch[] | null {
69 if (word.length > wordToMatchAgainst.length) {
70 return null;
78 return [{ start: index, end: index + word.length }];
79 }
80 > filters.ts
81 > export function matchesBaseContiguousSubString(word: string, wordToMatchAgainst: string): IMatch[] | null {
82 if (word.length > wordToMatchAgainst.length) {
83 return null;
93 return [{ start: index, end: index + word.length }];
94 }
95 > filters.ts
96 > // Substring
97 >
98 > export function matchesSubString(word: string, wordToMatchAgainst: string): IMatch[] | null {
99 if (word.length > wordToMatchAgainst.length) {
100 return null;
103 return _matchesSubString(word.toLowerCase(), wordToMatchAgainst.toLowerCase(), 0, 0);
104 }
105 > filters.ts
106 function _matchesSubString(word: string, wordToMatchAgainst: string, i: number, j: number): IMatch[] | null {
107 if (i === word.length) {
121 }
122 }
123 > filters.ts
124 > // CamelCase
125 >
126 function isLower(code: number): boolean {
127 return CharCode.a <= code && code <= CharCode.z;
128 }
129 > filters.ts
130 > export function isUpper(code: number): boolean {
131 return CharCode.A <= code && code <= CharCode.Z;
132 }
133 > filters.ts
134 function isNumber(code: number): boolean {
135 return CharCode.Digit0 <= code && code <= CharCode.Digit9;
136 }
137 > filters.ts
138 function isWhitespace(code: number): boolean {
139 return (
144 );
145 }
146 > filters.ts
147 > const wordSeparators = new Set<number>();
148 > // These are chosen as natural word separators based on written text.
149 > // It is a subset of the word separators used by the monaco editor.
150 > '()[]{}<>`\'"-/;:,.?!'
151 > .split('')
152 > .forEach(s => wordSeparators.add(s.charCodeAt(0)));
153 >
154 function isWordSeparator(code: number): boolean {
155 return isWhitespace(code) || wordSeparators.has(code);
156 }
157 > filters.ts
158 function charactersMatch(codeA: number, codeB: number): boolean {
159 return (codeA === codeB) || (isWordSeparator(codeA) && isWordSeparator(codeB));
160 }
161 > filters.ts
162 > const alternateCharsCache: Map<number, ArrayLike<number> | undefined> = new Map();
163 > /**
164 > * Gets alternative codes to the character code passed in. This comes in the
165 > * form of an array of character codes, all of which must match _in order_ to
166 > * successfully match.
167 > *
168 > * @param code The character code to check.
169 > */
170 function getAlternateCodes(code: number): ArrayLike<number> | undefined {
171 if (alternateCharsCache.has(code)) {
186 return result;
187 }
188 > filters.ts
189 function isAlphanumeric(code: number): boolean {
190 return isLower(code) || isUpper(code) || isNumber(code);
191 }
192 > filters.ts
193 function join(head: IMatch, tail: IMatch[]): IMatch[] {
194 if (tail.length === 0) {
201 return tail;
202 }
203 > filters.ts
204 function nextAnchor(camelCaseWord: string, start: number): number {
205 for (let i = start; i < camelCaseWord.length; i++) {
211 return camelCaseWord.length;
212 }
213 > filters.ts
214 function _matchesCamelCase(word: string, camelCaseWord: string, i: number, j: number): IMatch[] | null {
215 if (i === word.length) {
230 }
231 }
232 > filters.ts
233 > interface ICamelCaseAnalysis {
234 > upperPercent: number;
235 > lowerPercent: number;
236 > alphaPercent: number;
237 > numericPercent: number;
238 > }
239 >
240 > // Heuristic to avoid computing camel case matcher for words that don't
241 > // look like camelCaseWords.
242 function analyzeCamelCaseWord(word: string): ICamelCaseAnalysis {
243 let upper = 0, lower = 0, alpha = 0, numeric = 0, code = 0;
259 return { upperPercent, lowerPercent, alphaPercent, numericPercent };
260 }
261 > filters.ts
262 function isUpperCaseWord(analysis: ICamelCaseAnalysis): boolean {
263 const { upperPercent, lowerPercent } = analysis;
264 return lowerPercent === 0 && upperPercent > 0.6;
265 }
266 > filters.ts
267 function isCamelCaseWord(analysis: ICamelCaseAnalysis): boolean {
268 const { upperPercent, lowerPercent, alphaPercent, numericPercent } = analysis;
269 return lowerPercent > 0.2 && upperPercent < 0.8 && alphaPercent > 0.6 && numericPercent < 0.2;
270 }
271 > filters.ts
272 > // Heuristic to avoid computing camel case matcher for words that don't
273 > // look like camel case patterns.
274 function isCamelCasePattern(word: string): boolean {
275 let upper = 0, lower = 0, code = 0, whitespace = 0;
289 }
290 }
291 > filters.ts
292 > export function matchesCamelCase(word: string, camelCaseWord: string): IMatch[] | null {
293 if (!camelCaseWord) {
294 return null;
330 return result;
331 }
332 > filters.ts
333 > // Matches beginning of words supporting non-ASCII languages
334 > // If `contiguous` is true then matches word with beginnings of the words in the target. E.g. "pul" will match "Git: Pull"
335 > // Otherwise also matches sub string of the word with beginnings of the words in the target. E.g. "gp" or "g p" will match "Git: Pull"
336 > // Useful in cases where the target is words (e.g. command labels)
337 >
338 > export function matchesWords(word: string, target: string, contiguous: boolean = false): IMatch[] | null {
339 if (!target || target.length === 0) {
340 return null;
361 return result;
362 }
363 > filters.ts
364 function cloneMatches(matches: IMatch[] | null): IMatch[] | null {
365 if (matches === null) {
372 return result;
373 }
374 > filters.ts
375 function _matchesWords(word: string, target: string, wordIndex: number, targetIndex: number, contiguous: boolean, memo: Map<number, IMatch[] | null>): IMatch[] | null {
376 if (wordIndex === word.length) {
391 return computed;
392 }
393 > filters.ts
394 function _matchesWordsCompute(word: string, target: string, wordIndex: number, targetIndex: number, contiguous: boolean, memo: Map<number, IMatch[] | null>): IMatch[] | null {
395 let targetIndexOffset = 0;
440 return join({ start: targetIndex, end: targetIndex + targetIndexOffset + 1 }, result);
441 }
442 > filters.ts
443 function nextWord(word: string, start: number): number {
444 for (let i = start; i < word.length; i++) {
450 return word.length;
451 }
452 > filters.ts
453 > // Fuzzy
454 >
455 > const fuzzyContiguousFilter = or(matchesPrefix, matchesCamelCase, matchesContiguousSubString);
456 > const fuzzySeparateFilter = or(matchesPrefix, matchesCamelCase, matchesSubString);
457 > const fuzzyRegExpCache = new LRUCache<string, RegExp>(10000); // bounded to 10000 elements
458 >
459 > export function matchesFuzzy(word: string, wordToMatchAgainst: string, enableSeparateSubstringMatching = false): IMatch[] | null {
460 if (typeof word !== 'string' || typeof wordToMatchAgainst !== 'string') {
461 return null; // return early for invalid input
478 return enableSeparateSubstringMatching ? fuzzySeparateFilter(word, wordToMatchAgainst) : fuzzyContiguousFilter(word, wordToMatchAgainst);
479 }
480 > filters.ts
481 > /**
482 > * Match pattern against word in a fuzzy way. As in IntelliSense and faster and more
483 > * powerful than `matchesFuzzy`
484 > */
485 > export function matchesFuzzy2(pattern: string, word: string): IMatch[] | null {
486 const score = fuzzyScore(pattern, pattern.toLowerCase(), 0, word, word.toLowerCase(), 0, { firstMatchCanBeWeak: true, boostFullMatch: true });
487 return score ? createMatches(score) : null;
488 }
489 > filters.ts
490 > export function anyScore(pattern: string, lowPattern: string, patternPos: number, word: string, lowWord: string, wordPos: number): FuzzyScore {
491 const max = Math.min(13, pattern.length);
492 for (; patternPos < max; patternPos++) {
498 return [0, wordPos];
499 }
500 > filters.ts
501 > //#region --- fuzzyScore ---
502 >
503 > export function createMatches(score: undefined | FuzzyScore): IMatch[] {
504 if (typeof score === 'undefined') {
505 return [];
518 return res;
519 }
520 > filters.ts
521 > const _maxLen = 128;
522 >
523 > function initTable() {
524 > const table: number[][] = [];
525 > const row: number[] = [];
526 > for (let i = 0; i <= _maxLen; i++) {
527 > row[i] = 0;
528 > }
529 > for (let i = 0; i <= _maxLen; i++) {
530 > table.push(row.slice(0));
531 > }
532 > return table;
533 > }
534 >
535 > function initArr(maxLen: number) {
536 > const row: number[] = [];
537 > for (let i = 0; i <= maxLen; i++) {
538 > row[i] = 0;
539 > }
540 > return row;
541 > }
542 >
543 > const _minWordMatchPos = initArr(2 * _maxLen); // min word position for a certain pattern position
544 > const _maxWordMatchPos = initArr(2 * _maxLen); // max word position for a certain pattern position
545 > const _diag = initTable(); // the length of a contiguous diagonal match
546 > const _table = initTable();
547 > const _arrows = <Arrow[][]>initTable();
548 > const _debug = false;
549 >
550 function printTable(table: number[][], pattern: string, patternLen: number, word: string, wordLen: number): string {
551 function pad(s: string, n: number, pad = ' ') {
567 return ret;
568 }
569 > filters.ts
570 function printTables(pattern: string, patternStart: number, word: string, wordStart: number): void {
571 pattern = pattern.substr(patternStart);
575 console.log(printTable(_diag, pattern, pattern.length, word, word.length));
576 }
577 > filters.ts
578 function isSeparatorAtPos(value: string, index: number): boolean {
579 if (index < 0 || index >= value.length) {
610 }
611 }
612 > filters.ts
613 function isWhitespaceAtPos(value: string, index: number): boolean {
614 if (index < 0 || index >= value.length) {
624 }
625 }
626 > filters.ts
627 function isUpperCaseAtPos(pos: number, word: string, wordLow: string): boolean {
628 return word[pos] !== wordLow[pos];
629 }
630 > filters.ts
631 > export function isPatternInWord(patternLow: string, patternPos: number, patternLen: number, wordLow: string, wordPos: number, wordLen: number, fillMinWordPosArr = false): boolean {
632 while (patternPos < patternLen && wordPos < wordLen) {
633 if (patternLow[patternPos] === wordLow[wordPos]) {
642 return patternPos === patternLen; // pattern must be exhausted
643 }
644 > filters.ts
645 > const enum Arrow { Diag = 1, Left = 2, LeftLeft = 3 }
646 >
647 > /**
648 > * An array representing a fuzzy match.
649 > *
650 > * 0. the score
651 > * 1. the offset at which matching started
652 > * 2. `<match_pos_N>`
653 > * 3. `<match_pos_1>`
654 > * 4. `<match_pos_0>` etc
655 > */
656 > export type FuzzyScore = [score: number, wordStart: number, ...matches: number[]];
657 >
658 > export namespace FuzzyScore {
659 > /**
660 > * No matches and value `-100`
661 > */
662 > export const Default: FuzzyScore = ([-100, 0]);
663 >
664 > export function isDefault(score?: FuzzyScore): score is [-100, 0] {
665 return !score || (score.length === 2 && score[0] === -100 && score[1] === 0);
666 }
667 > } filters.ts
668 >
669 > export abstract class FuzzyScoreOptions {
670 >
671 > static default = { boostFullMatch: true, firstMatchCanBeWeak: false };
672 >
673 > constructor(
674 readonly firstMatchCanBeWeak: boolean,
675 readonly boostFullMatch: boolean,
676 ) { }
677 > } filters.ts
678 >
679 > export interface FuzzyScorer {
680 > (pattern: string, lowPattern: string, patternPos: number, word: string, lowWord: string, wordPos: number, options?: FuzzyScoreOptions): FuzzyScore | undefined;
681 > }
682 >
683 > export function fuzzyScore(pattern: string, patternLow: string, patternStart: number, word: string, wordLow: string, wordStart: number, options: FuzzyScoreOptions = FuzzyScoreOptions.default): FuzzyScore | undefined {
684
685 const patternLen = pattern.length > _maxLen ? _maxLen : pattern.length;
832 return result;
833 }
834 > filters.ts
835 function _fillInMaxWordMatchPos(patternLen: number, wordLen: number, patternStart: number, wordStart: number, patternLow: string, wordLow: string) {
836 let patternPos = patternLen - 1;
844 }
845 }
846 > filters.ts
847 function _doScore(
848 pattern: string, patternLow: string, patternPos: number, patternStart: number,
913 return score;
914 }
915 > filters.ts
916 > //#endregion
917 >
918 >
919 > //#region --- graceful ---
920 >
921 > export function fuzzyScoreGracefulAggressive(pattern: string, lowPattern: string, patternPos: number, word: string, lowWord: string, wordPos: number, options?: FuzzyScoreOptions): FuzzyScore | undefined {
922 return fuzzyScoreWithPermutations(pattern, lowPattern, patternPos, word, lowWord, wordPos, true, options);
923 }
924 > filters.ts
925 > export function fuzzyScoreGraceful(pattern: string, lowPattern: string, patternPos: number, word: string, lowWord: string, wordPos: number, options?: FuzzyScoreOptions): FuzzyScore | undefined {
926 return fuzzyScoreWithPermutations(pattern, lowPattern, patternPos, word, lowWord, wordPos, false, options);
927 }
928 > filters.ts
929 function fuzzyScoreWithPermutations(pattern: string, lowPattern: string, patternPos: number, word: string, lowWord: string, wordPos: number, aggressive: boolean, options?: FuzzyScoreOptions): FuzzyScore | undefined {
930 let top = fuzzyScore(pattern, lowPattern, patternPos, word, lowWord, wordPos, options);
959 return top;
960 }
961 > filters.ts
962 function nextTypoPermutation(pattern: string, patternPos: number): string | undefined {
963
978 + pattern.slice(patternPos + 2);
979 }
980 > filters.ts
981 > //#endregion
src/vs/workbench/contrib/chat/common/chatDebugServiceImpl.ts 209 covered LOC · 41 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatDebugServiceImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { timeout } from '../../../../base/common/async.js';
7 > import { CancellationToken, CancellationTokenSource } from '../../../../base/common/cancellation.js';
8 > import { Emitter, Event } from '../../../../base/common/event.js';
9 > import { onUnexpectedError } from '../../../../base/common/errors.js';
10 > import { Disposable, IDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
11 > import { ResourceMap } from '../../../../base/common/map.js';
12 > import { extUri } from '../../../../base/common/resources.js';
13 > import { URI } from '../../../../base/common/uri.js';
14 > import { ChatDebugLogLevel, IChatDebugEvent, IChatDebugLogProvider, IChatDebugResolvedEventContent, IChatDebugService } from './chatDebugService.js';
15 > import { isAgentHostTarget, localChatSessionType } from './chatSessionsService.js';
16 > import { getChatSessionType } from './model/chatUri.js';
17 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
18 > import { AgentHostAgentDebugLogMaxEventsSettingId } from './promptSyntax/promptTypes.js';
19 >
20 > /**
21 > * Per-session circular buffer for debug events.
22 > * Stores up to `capacity` events using a ring buffer.
23 > */
24 > class SessionEventBuffer {
25 > private readonly _buffer: (IChatDebugEvent | undefined)[];
26 > private _head = 0;
27 > private _size = 0;
28 >
29 > constructor(readonly capacity: number) {
30 this._buffer = new Array(capacity);
31 }
33 > get size(): number {
34 return this._size;
35 }
37 > push(event: IChatDebugEvent): void {
38 const idx = (this._head + this._size) % this.capacity;
39 this._buffer[idx] = event;
44 }
45 }
47 > /** Return events in insertion order. */
48 > toArray(): IChatDebugEvent[] {
49 const result: IChatDebugEvent[] = [];
50 for (let i = 0; i < this._size; i++) {
56 return result;
57 }
59 > /** Remove events matching the predicate and compact in-place. */
60 > removeWhere(predicate: (event: IChatDebugEvent) => boolean): void {
61 let write = 0;
62 for (let i = 0; i < this._size; i++) {
77 this._size = write;
78 }
80 > clear(): void {
81 this._buffer.fill(undefined);
82 this._head = 0;
83 this._size = 0;
84 }
86 >
87 > export class ChatDebugServiceImpl extends Disposable implements IChatDebugService {
88 > declare readonly _serviceBrand: undefined;
89 >
90 > static readonly MAX_EVENTS_PER_SESSION = 10_000;
91 > static readonly MAX_SESSIONS = 5;
92 >
93 > /** Per-session event buffers. Ordered from oldest to newest session (LRU). */
94 > private readonly _sessionBuffers = new ResourceMap<SessionEventBuffer>();
95 > /** Ordered list of session URIs for LRU eviction. */
96 > private readonly _sessionOrder: URI[] = [];
97 > /** Per-session tracking of seen event IDs to deduplicate events
98 > * that share the same ID (e.g. subagentInvocation + userMessage
99 > * emitted from the same span). Stores id → event kind so we can
100 > * keep the richer event kind on collision. */
101 > private readonly _seenEventIds = new ResourceMap<Map<string, IChatDebugEvent['kind']>>();
102 >
103 > private readonly _onDidAddEvent = this._register(new Emitter<IChatDebugEvent>());
104 > readonly onDidAddEvent: Event<IChatDebugEvent> = this._onDidAddEvent.event;
105 >
106 > private readonly _onDidClearProviderEvents = this._register(new Emitter<URI>());
107 > readonly onDidClearProviderEvents: Event<URI> = this._onDidClearProviderEvents.event;
108 >
109 > private readonly _onDidEndSession = this._register(new Emitter<URI>());
110 > readonly onDidEndSession: Event<URI> = this._onDidEndSession.event;
111 >
112 > private readonly _onDidChangeAvailableSessionResources = this._register(new Emitter<void>());
113 > readonly onDidChangeAvailableSessionResources: Event<void> = this._onDidChangeAvailableSessionResources.event;
114 >
115 > private readonly _providers = new Set<IChatDebugLogProvider>();
116 > private readonly _invocationCts = new ResourceMap<CancellationTokenSource>();
117 >
118 > /**
119 > * Sessions whose provider events should be cleared before the next batch of
120 > * provider events is applied. The clear is deferred until the first new
121 > * provider event actually arrives so that a provider which transiently
122 > * returns nothing (e.g. an Agent Host `events.jsonl` mid-rewrite) does not
123 > * wipe the events currently shown.
124 > */
125 > private readonly _pendingProviderClear = new ResourceMap<boolean>();
126 >
127 > /** Events that were returned by providers (not internally logged). */
128 > private readonly _providerEvents = new WeakSet<IChatDebugEvent>();
129 >
130 > /** Session URIs created via import. */
131 > private readonly _importedSessions = new ResourceMap<boolean>();
132 >
133 > /** Session URIs reported by providers as available on disk (historical sessions). */
134 > private readonly _availableSessionResources: URI[] = [];
135 > private readonly _availableSessionResourceSet = new Set<string>();
136 >
137 > /** Titles for historical sessions discovered from disk. */
138 > private readonly _historicalSessionTitles = new ResourceMap<string>();
139 >
140 > /** Human-readable titles for imported sessions. */
141 > private readonly _importedSessionTitles = new ResourceMap<string>();
142 >
143 > activeSessionResource: URI | undefined;
144 >
145 > constructor(
146 > @IConfigurationService private readonly _configurationService: IConfigurationService, chatDebugServiceImpl.ts
147 > ) {
148 > super();
149 > }
151 > /** Priority for deduplicating events with the same ID: lower = richer. */
152 > private static readonly _eventKindPriority: Record<string, number> = {
153 > subagentInvocation: 0,
154 > modelTurn: 1,
155 > toolCall: 2,
156 > agentResponse: 3,
157 > userMessage: 4,
158 > generic: 5,
159 > };
160 >
161 > /** Session types eligible for debug logging and provider invocation. */
162 > private static readonly _debugEligibleSessionTypes = new Set([
163 > localChatSessionType, // local sessions
164 > 'copilotcli', // Copilot CLI background sessions
165 > 'agent-host-copilotcli', // local Agent Host Copilot CLI sessions
166 > 'claude-code', // Claude Code CLI sessions
167 > ]);
169 > private _isDebugEligibleSession(sessionResource: URI): boolean {
170 const sessionType = getChatSessionType(sessionResource);
171 return ChatDebugServiceImpl._debugEligibleSessionTypes.has(sessionType)
175 || this._importedSessions.has(sessionResource);
176 }
178 > /**
179 > * The in-memory event capacity for a session. Agent host (Copilot CLI)
180 > * sessions honor a dedicated, configurable cap so their (potentially large)
181 > * on-disk logs can be surfaced without changing the local-session default;
182 > * all other sessions use {@link ChatDebugServiceImpl.MAX_EVENTS_PER_SESSION}.
183 > */
184 > private _capacityForSession(sessionResource: URI): number {
185 if (!isAgentHostTarget(getChatSessionType(sessionResource))) {
186 return ChatDebugServiceImpl.MAX_EVENTS_PER_SESSION;
192 return ChatDebugServiceImpl.MAX_EVENTS_PER_SESSION;
193 }
195 > log(sessionResource: URI, name: string, details?: string, level: ChatDebugLogLevel = ChatDebugLogLevel.Info, options?: { id?: string; category?: string; parentEventId?: string }): void {
196 if (!this._isDebugEligibleSession(sessionResource)) {
197 return;
209 });
210 }
212 > addEvent(event: IChatDebugEvent): void {
213 // Resolve the session's buffer (if any) once, and its capacity. New
214 // events during streaming target an existing buffer, so we reuse its
272 this._onDidAddEvent.fire(event);
273 }
275 > addProviderEvent(event: IChatDebugEvent): void {
276 // If a re-invocation is pending for this session, clear the previously
277 // loaded provider events now that fresh data has actually arrived. This
285 this.addEvent(event);
286 }
288 > getEvents(sessionResource?: URI): readonly IChatDebugEvent[] {
289 if (sessionResource) {
290 const buffer = this._sessionBuffers.get(sessionResource);
315 return result;
316 }
318 > private _isSorted(events: IChatDebugEvent[]): boolean {
319 for (let i = 1; i < events.length; i++) {
320 if (events[i].created.getTime() < events[i - 1].created.getTime()) {
324 return true;
325 }
327 > private _deduplicateEvents(events: IChatDebugEvent[]): IChatDebugEvent[] {
328 const seen = new Map<string, number>(); // id → index in result
329 const priority = ChatDebugServiceImpl._eventKindPriority;
347 return result;
348 }
350 > getSessionResources(): readonly URI[] {
351 return [...this._sessionOrder];
352 }
354 > clear(): void {
355 > this._sessionBuffers.clear();
356 > this._sessionOrder.length = 0;
357 > this._seenEventIds.clear();
358 > this._importedSessions.clear();
359 > this._importedSessionTitles.clear();
360 > this._availableSessionResources.length = 0;
361 > this._availableSessionResourceSet.clear();
362 > this._historicalSessionTitles.clear();
363 > }
364 >
365 > /** Remove all ancillary state for an evicted session. */
366 > private _evictSession(sessionResource: URI): void {
367 this._sessionBuffers.delete(sessionResource);
368 this._seenEventIds.delete(sessionResource);
376 }
377 }
379 > registerProvider(provider: IChatDebugLogProvider): IDisposable {
380 this._providers.add(provider);
381
393 });
394 }
396 > hasInvokedProviders(sessionResource: URI): boolean {
397 return this._invocationCts.has(sessionResource);
398 }
400 > async invokeProviders(sessionResource: URI): Promise<void> {
401
402 if (!this._isDebugEligibleSession(sessionResource)) {
434 // or when the service is disposed.
435 }
437 > private async _invokeProvider(provider: IChatDebugLogProvider, sessionResource: URI, token: CancellationToken): Promise<void> {
438 try {
439 const events = await provider.provideChatDebugLog(sessionResource, token);
460 }
461 }
463 > endSession(sessionResource: URI): void {
464 const cts = this._invocationCts.get(sessionResource);
465 if (cts) {
470 this._onDidEndSession.fire(sessionResource);
471 }
473 > private _clearProviderEvents(sessionResource: URI): void {
474 const buffer = this._sessionBuffers.get(sessionResource);
475 if (buffer) {
487 this._onDidClearProviderEvents.fire(sessionResource);
488 }
490 > async resolveEvent(eventId: string): Promise<IChatDebugResolvedEventContent | undefined> {
491 for (const provider of this._providers) {
492 if (provider.resolveChatDebugLogEvent) {
503 return undefined;
504 }
506 > isCoreEvent(event: IChatDebugEvent): boolean {
507 return !this._providerEvents.has(event);
508 }
510 > setImportedSessionTitle(sessionResource: URI, title: string): void {
511 this._importedSessionTitles.set(sessionResource, title);
512 }
514 > getImportedSessionTitle(sessionResource: URI): string | undefined {
515 return this._importedSessionTitles.get(sessionResource);
516 }
518 > addAvailableSessionResources(resources: readonly { uri: URI; title?: string }[]): void {
519 let added = false;
520 for (const { uri, title } of resources) {
533 }
534 }
536 > /** Lazy fetchers for available sessions from providers. Each is invoked at most once. */
537 > private readonly _availableSessionsFetchers = new Set<{ readonly fetcher: (token: CancellationToken) => Promise<{ uri: URI; title?: string }[]>; started: boolean }>();
538 > private _availableSessionsRequested = false;
540 > getAvailableSessionResources(): readonly URI[] {
541 // Trigger lazy fetch when both a fetcher is registered and this getter is called.
542 this._availableSessionsRequested = true;
553 return result;
554 }
556 > registerAvailableSessionsFetcher(fetcher: (token: CancellationToken) => Promise<{ uri: URI; title?: string }[]>): IDisposable {
557 const entry = { fetcher, started: false };
558 this._availableSessionsFetchers.add(entry);
561 return toDisposable(() => this._availableSessionsFetchers.delete(entry));
562 }
564 > private _tryFetchAvailableSessions(): void {
565 if (!this._availableSessionsRequested) {
566 return;
579 }
580 }
582 > getHistoricalSessionTitle(sessionResource: URI): string | undefined {
583 return this._historicalSessionTitles.get(sessionResource);
584 }
586 > async exportLog(sessionResource: URI): Promise<Uint8Array | undefined> {
587 for (const provider of this._providers) {
588 if (provider.provideChatDebugLogExport) {
599 return undefined;
600 }
602 > async importLog(data: Uint8Array): Promise<URI | undefined> {
603 for (const provider of this._providers) {
604 if (provider.resolveChatDebugLogImport) {
src/vs/base/common/observableInternal/base.ts 206 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- base.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DisposableStore, onUnexpectedError } from './commonFacade/deps.js';
7 >
8 > /**
9 > * Represents an observable value.
10 > *
11 > * @template T The type of the values the observable can hold.
12 > */
13 > // This interface exists so that, for example for string observables,
14 > // typescript renders the type as `IObservable<string>` instead of `IObservable<string, unknown>`.
15 > export interface IObservable<T> extends IObservableWithChange<T, unknown> { }
16 >
17 > /**
18 > * Represents an observable value.
19 > *
20 > * @template T The type of the values the observable can hold.
21 > * @template TChange The type used to describe value changes
22 > * (usually `void` and only used in advanced scenarios).
23 > * While observers can miss temporary values of an observable,
24 > * they will receive all change values (as long as they are subscribed)!
25 > */
26 > export interface IObservableWithChange<T, TChange = unknown> {
27 > /**
28 > * Returns the current value.
29 > *
30 > * Calls {@link IObserver.handleChange} if the observable notices that the value changed.
31 > * Must not be called from {@link IObserver.handleChange}!
32 > */
33 > get(): T;
34 >
35 > /**
36 > * Forces the observable to check for changes and report them.
37 > *
38 > * Has the same effect as calling {@link IObservable.get}, but does not force the observable
39 > * to actually construct the value, e.g. if change deltas are used.
40 > * Calls {@link IObserver.handleChange} if the observable notices that the value changed.
41 > * Must not be called from {@link IObserver.handleChange}!
42 > */
43 > reportChanges(): void;
44 >
45 > /**
46 > * Adds the observer to the set of subscribed observers.
47 > * This method is idempotent.
48 > */
49 > addObserver(observer: IObserver): void;
50 >
51 > /**
52 > * Removes the observer from the set of subscribed observers.
53 > * This method is idempotent.
54 > */
55 > removeObserver(observer: IObserver): void;
56 >
57 > // #region These members have a standard implementation and are only part of the interface for convenience.
58 >
59 > /**
60 > * Reads the current value and subscribes the reader to this observable.
61 > *
62 > * Calls {@link IReader.readObservable} if a reader is given, otherwise {@link IObservable.get}
63 > * (see {@link ConvenientObservable.read} for the implementation).
64 > */
65 > read(reader: IReader | undefined): T;
66 >
67 > /**
68 > * Makes sure this value is computed eagerly.
69 > */
70 > recomputeInitiallyAndOnChange(store: DisposableStore, handleValue?: (value: T) => void): IObservable<T>;
71 >
72 > /**
73 > * Makes sure this value is cached.
74 > */
75 > keepObserved(store: DisposableStore): IObservable<T>;
76 >
77 > /**
78 > * Creates a derived observable that depends on this observable.
79 > * Use the reader to read other observables
80 > * (see {@link ConvenientObservable.map} for the implementation).
81 > */
82 > map<TNew>(fn: (value: T, reader: IReader) => TNew): IObservable<TNew>;
83 > map<TNew>(owner: object, fn: (value: T, reader: IReader) => TNew): IObservable<TNew>;
84 >
85 > flatten<TNew>(this: IObservable<IObservable<TNew>>): IObservable<TNew>;
86 >
87 > /**
88 > * ONLY FOR DEBUGGING!
89 > * Logs computations of this derived.
90 > */
91 > log(): IObservableWithChange<T, TChange>;
92 >
93 > /**
94 > * A human-readable name for debugging purposes.
95 > */
96 > readonly debugName: string;
97 >
98 > /**
99 > * This property captures the type of the change object. Do not use it at runtime!
100 > */
101 > readonly TChange: TChange;
102 >
103 > // #endregion
104 > }
105 >
106 > /**
107 > * Represents an observer that can be subscribed to an observable.
108 > *
109 > * If an observer is subscribed to an observable and that observable didn't signal
110 > * a change through one of the observer methods, the observer can assume that the
111 > * observable didn't change.
112 > * If an observable reported a possible change, {@link IObservable.reportChanges} forces
113 > * the observable to report an actual change if there was one.
114 > */
115 > export interface IObserver {
116 > /**
117 > * Signals that the given observable might have changed and a transaction potentially modifying that observable started.
118 > * Before the given observable can call this method again, is must call {@link IObserver.endUpdate}.
119 > *
120 > * Implementations must not get/read the value of other observables, as they might not have received this event yet!
121 > * The method {@link IObservable.reportChanges} can be used to force the observable to report the changes.
122 > */
123 > beginUpdate<T>(observable: IObservable<T>): void;
124 >
125 > /**
126 > * Signals that the transaction that potentially modified the given observable ended.
127 > * This is a good place to react to (potential) changes.
128 > */
129 > endUpdate<T>(observable: IObservable<T>): void;
130 >
131 > /**
132 > * Signals that the given observable might have changed.
133 > * The method {@link IObservable.reportChanges} can be used to force the observable to report the changes.
134 > *
135 > * Implementations must not get/read the value of other observables, as they might not have received this event yet!
136 > * The change should be processed lazily or in {@link IObserver.endUpdate}.
137 > */
138 > handlePossibleChange<T>(observable: IObservable<T>): void;
139 >
140 > /**
141 > * Signals that the given {@link observable} changed.
142 > *
143 > * Implementations must not get/read the value of other observables, as they might not have received this event yet!
144 > * The change should be processed lazily or in {@link IObserver.endUpdate}.
145 > *
146 > * @param change Indicates how or why the value changed.
147 > */
148 > handleChange<T, TChange>(observable: IObservableWithChange<T, TChange>, change: TChange): void;
149 > }
150 >
151 > /**
152 > * A reader allows code to track what it depends on, so the caller knows when the computed value or produced side-effect is no longer valid.
153 > * Use `derived(reader => ...)` to turn code that needs a reader into an observable value.
154 > */
155 > export interface IReader {
156 > /**
157 > * Reads the value of an observable and subscribes to it.
158 > */
159 > readObservable<T>(observable: IObservableWithChange<T, any>): T;
160 > }
161 >
162 > export interface ISettable<T, TChange = void> {
163 > /**
164 > * Sets the value of the observable.
165 > * Use a transaction to batch multiple changes (with a transaction, observers only react at the end of the transaction).
166 > *
167 > * @param transaction When given, value changes are handled on demand or when the transaction ends.
168 > * @param change Describes how or why the value changed.
169 > */
170 > set(value: T, transaction: ITransaction | undefined, change: TChange): void;
171 > }
172 >
173 > export interface ITransaction {
174 > /**
175 > * Calls {@link Observer.beginUpdate} immediately
176 > * and {@link Observer.endUpdate} when the transaction ends.
177 > */
178 > updateObserver(observer: IObserver, observable: IObservableWithChange<any, any>): void;
179 > }
180 >
181 > /**
182 > * This function is used to indicate that the caller recovered from an error that indicates a bug.
183 > */
184 > export function handleBugIndicatingErrorRecovery(message: string) {
185 const err = new Error('BugIndicatingErrorRecovery: ' + message);
186 onUnexpectedError(err);
187 console.error('recovered from an error that indicates a bug', err);
188 }
189 > base.ts
190 > /**
191 > * A settable observable.
192 > */
193 > export interface ISettableObservable<T, TChange = void> extends IObservableWithChange<T, TChange>, ISettable<T, TChange> {
194 > }
195 >
196 > export interface IReaderWithStore extends IReader {
197 > /**
198 > * Items in this store get disposed just before the observable recomputes/reruns or when it becomes unobserved.
199 > */
200 > get store(): DisposableStore;
201 >
202 > /**
203 > * Items in this store get disposed just after the observable recomputes/reruns or when it becomes unobserved.
204 > * This is important if the current run needs the undisposed result from the last run.
205 > *
206 > * Warning: Items in this store might still get disposed before dependents (that read the now disposed value in the past) are recomputed with the new (undisposed) value!
207 > * A clean solution for this is ref counting.
208 > */
209 > get delayedStore(): DisposableStore;
210 > }
src/vs/base/common/observableInternal/logging/debugger/devToolsLogger.ts 204 covered LOC · 24 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- devToolsLogger.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { AutorunObserver, AutorunState } from '../../reactions/autorunImpl.js';
7 > import { TransactionImpl } from '../../transaction.js';
8 > import { IChangeInformation, IObservableLogger } from '../logging.js';
9 > import { formatValue } from '../consoleObservableLogger.js';
10 > import { ObsDebuggerApi, IObsDeclaration, ObsInstanceId, ObsStateUpdate, ITransactionState, ObserverInstanceState } from './debuggerApi.js';
11 > import { registerDebugChannel } from './debuggerRpc.js';
12 > import { deepAssign, deepAssignDeleteNulls, Throttler } from './utils.js';
13 > import { isDefined } from '../../../types.js';
14 > import { FromEventObservable } from '../../observables/observableFromEvent.js';
15 > import { BugIndicatingError, onUnexpectedError } from '../../../errors.js';
16 > import { IObservable, IObserver } from '../../base.js';
17 > import { BaseObservable } from '../../observables/baseObservable.js';
18 > import { Derived, DerivedState } from '../../observables/derivedImpl.js';
19 > import { ObservableValue } from '../../observables/observableValue.js';
20 > import { DebugLocation } from '../../debugLocation.js';
21 >
22 > interface IInstanceInfo {
23 > declarationId: number;
24 > instanceId: number;
25 > }
26 >
27 > interface IObservableInfo extends IInstanceInfo {
28 > listenerCount: number;
29 > lastValue: string | undefined;
30 > updateCount: number;
31 > changedObservables: Set<IObservable<any>>;
32 > }
33 >
34 > interface IAutorunInfo extends IInstanceInfo {
35 > updateCount: number;
36 > changedObservables: Set<IObservable<any>>;
37 > }
38 >
39 > export class DevToolsLogger implements IObservableLogger {
40 > private static _instance: DevToolsLogger | undefined = undefined;
41 > public static getInstance(): DevToolsLogger {
42 if (DevToolsLogger._instance === undefined) {
43 DevToolsLogger._instance = new DevToolsLogger();
45 return DevToolsLogger._instance;
46 }
48 > private _declarationId = 0;
49 > private _instanceId = 0;
50 >
51 > private readonly _declarations = new Map</* declarationId + type */string, IObsDeclaration>();
52 > private readonly _instanceInfos = new WeakMap<object, IObservableInfo | IAutorunInfo>();
53 > private readonly _aliveInstances = new Map<ObsInstanceId, IObservable<any> | AutorunObserver>();
54 > private readonly _activeTransactions = new Set<TransactionImpl>();
55 >
56 > private readonly _channel = registerDebugChannel<ObsDebuggerApi>('observableDevTools', () => {
57 > return {
58 > notifications: {
59 > setDeclarationIdFilter: declarationIds => {
60 >
61 > },
62 > logObservableValue: (observableId) => {
63 > console.log('logObservableValue', observableId);
64 > },
65 > flushUpdates: () => {
66 > this._flushUpdates();
67 > },
68 > resetUpdates: () => {
69 > this._pendingChanges = null;
70 > this._channel.api.notifications.handleChange(this._fullState, true);
71 > },
72 > },
73 > requests: {
74 > getDeclarations: () => {
75 > const result: Record<string, IObsDeclaration> = {};
76 > for (const decl of this._declarations.values()) {
77 > result[decl.id] = decl;
78 > }
79 > return { decls: result };
80 > },
81 > getSummarizedInstances: () => {
82 > return null!;
83 > },
84 > getObservableValueInfo: instanceId => {
85 > const obs = this._aliveInstances.get(instanceId) as BaseObservable<any>;
86 > return {
87 > observers: [...obs.debugGetObservers()].map(d => this._formatObserver(d)).filter(isDefined),
88 > };
89 > },
90 > getDerivedInfo: instanceId => {
91 > const d = this._aliveInstances.get(instanceId) as Derived<any>;
92 > return {
93 > dependencies: [...d.debugGetState().dependencies].map(d => this._formatObservable(d)).filter(isDefined),
94 > observers: [...d.debugGetObservers()].map(d => this._formatObserver(d)).filter(isDefined),
95 > };
96 > },
97 > getAutorunInfo: instanceId => {
98 > const obs = this._aliveInstances.get(instanceId) as AutorunObserver;
99 > return {
100 > dependencies: [...obs.debugGetState().dependencies].map(d => this._formatObservable(d)).filter(isDefined),
101 > };
102 > },
103 > getTransactionState: () => {
104 > return this.getTransactionState();
105 > },
106 > setValue: (instanceId, jsonValue) => {
107 > const obs = this._aliveInstances.get(instanceId) as BaseObservable<any>;
108 >
109 > if (obs instanceof Derived) {
110 > obs.debugSetValue(jsonValue);
111 > } else if (obs instanceof ObservableValue) {
112 > obs.debugSetValue(jsonValue);
113 > } else if (obs instanceof FromEventObservable) {
114 > obs.debugSetValue(jsonValue);
115 > } else {
116 > throw new BugIndicatingError('Observable is not supported');
117 > }
118 >
119 > const observers = [...obs.debugGetObservers()];
120 > for (const d of observers) {
121 > d.beginUpdate(obs);
122 > }
123 > for (const d of observers) {
124 > d.handleChange(obs, undefined);
125 > }
126 > for (const d of observers) {
127 > d.endUpdate(obs);
128 > }
129 > },
130 > getValue: instanceId => {
131 > const obs = this._aliveInstances.get(instanceId) as BaseObservable<any>;
132 > if (obs instanceof Derived) {
133 > return formatValue(obs.debugGetState().value, 200);
134 > } else if (obs instanceof ObservableValue) {
135 > return formatValue(obs.debugGetState().value, 200);
136 > }
137 >
138 > return undefined;
139 > },
140 > logValue: (instanceId) => {
141 > const obs = this._aliveInstances.get(instanceId);
142 > if (obs && 'get' in obs) {
143 > console.log('Logged Value:', obs.get());
144 > } else {
145 > throw new BugIndicatingError('Observable is not supported');
146 > }
147 > },
148 > rerun: (instanceId) => {
149 > const obs = this._aliveInstances.get(instanceId);
150 > if (obs instanceof Derived) {
151 > obs.debugRecompute();
152 > } else if (obs instanceof AutorunObserver) {
153 > obs.debugRerun();
154 > } else {
155 > throw new BugIndicatingError('Observable is not supported');
156 > }
157 > },
158 > }
159 > };
160 > });
161 >
162 > private getTransactionState(): ITransactionState | undefined {
163 const affected: ObserverInstanceState[] = [];
164 const txs = [...this._activeTransactions];
188 return { names: txs.map(t => t.getDebugName() ?? 'tx'), affected };
189 }
191 > private _getObservableInfo(observable: IObservable<any>): IObservableInfo | undefined {
192 const info = this._instanceInfos.get(observable);
193 if (!info) {
197 return info as IObservableInfo;
198 }
200 > private _getAutorunInfo(autorun: AutorunObserver): IAutorunInfo | undefined {
201 const info = this._instanceInfos.get(autorun);
202 if (!info) {
206 return info as IAutorunInfo;
207 }
209 > private _getInfo(observer: IObserver, queue: (observer: IObserver) => void): ObserverInstanceState | undefined {
210 if (observer instanceof Derived) {
211 const observersToUpdate = [...observer.debugGetObservers()];
255 return undefined;
256 }
258 > private _formatObservable(obs: IObservable<any>): { name: string; instanceId: ObsInstanceId } | undefined {
259 const info = this._getObservableInfo(obs);
260 if (!info) { return undefined; }
261 return { name: obs.debugName, instanceId: info.instanceId };
262 }
264 > private _formatObserver(obs: IObserver): { name: string; instanceId: ObsInstanceId } | undefined {
265 if (obs instanceof Derived) {
266 return { name: obs.toString(), instanceId: this._getObservableInfo(obs)?.instanceId! };
273 return undefined;
274 }
276 > private constructor() {
277 DebugLocation.enable();
278 }
280 > private _pendingChanges: ObsStateUpdate | null = null;
281 > private readonly _changeThrottler = new Throttler();
282 >
283 > private readonly _fullState = {};
284 >
285 > private _handleChange(update: ObsStateUpdate): void {
286 deepAssignDeleteNulls(this._fullState, update);
287
294 this._changeThrottler.throttle(this._flushUpdates, 10);
295 }
297 > private readonly _flushUpdates = () => {
298 > if (this._pendingChanges !== null) {
299 > this._channel.api.notifications.handleChange(this._pendingChanges, false);
300 > this._pendingChanges = null;
301 > }
302 > };
303 >
304 > private _getDeclarationId(type: IObsDeclaration['type'], location: DebugLocation): number {
305 if (!location) {
306 return -1;
322 return decInfo.id;
323 }
325 > handleObservableCreated(observable: IObservable<any>, location: DebugLocation): void {
326 const declarationId = this._getDeclarationId('observable/value', location);
327
336 this._instanceInfos.set(observable, info);
337 }
339 > handleOnListenerCountChanged(observable: IObservable<any>, newCount: number): void {
340 const info = this._getObservableInfo(observable);
341 if (!info) { return; }
364 info.listenerCount = newCount;
365 }
367 > handleObservableUpdated(observable: IObservable<any>, changeInfo: IChangeInformation): void {
368 if (observable instanceof Derived) {
369 this._handleDerivedRecomputed(observable, changeInfo);
383 }
384 }
386 > handleAutorunCreated(autorun: AutorunObserver, location: DebugLocation): void {
387 const declarationId = this._getDeclarationId('autorun', location);
388 const info: IAutorunInfo = {
408 }
409 }
410 > handleAutorunDisposed(autorun: AutorunObserver): void { devToolsLogger.ts
411 const info = this._getAutorunInfo(autorun);
412 if (!info) { return; }
418 this._aliveInstances.delete(info.instanceId);
419 }
420 > handleAutorunDependencyChanged(autorun: AutorunObserver, observable: IObservable<any>, change: unknown): void { devToolsLogger.ts
421 const info = this._getAutorunInfo(autorun);
422 if (!info) { return; }
424 info.changedObservables.add(observable);
425 }
426 > handleAutorunStarted(autorun: AutorunObserver): void { devToolsLogger.ts
427
428 }
429 > handleAutorunFinished(autorun: AutorunObserver): void { devToolsLogger.ts
430 const info = this._getAutorunInfo(autorun);
431 if (!info) { return; }
437 });
438 }
440 > handleDerivedDependencyChanged(derived: Derived<any>, observable: IObservable<any>, change: unknown): void {
441 const info = this._getObservableInfo(derived);
442 if (info) {
444 }
445 }
446 > _handleDerivedRecomputed(observable: Derived<any>, changeInfo: IChangeInformation): void { devToolsLogger.ts
447 const info = this._getObservableInfo(observable);
448 if (!info) { return; }
459 }
460 }
461 > handleDerivedCleared(observable: Derived<any>): void { devToolsLogger.ts
462 const info = this._getObservableInfo(observable);
463 if (!info) { return; }
475 }
476 }
477 > handleBeginTransaction(transaction: TransactionImpl): void { devToolsLogger.ts
478 this._activeTransactions.add(transaction);
479 }
480 > handleEndTransaction(transaction: TransactionImpl): void { devToolsLogger.ts
481 this._activeTransactions.delete(transaction);
482 }
src/vs/workbench/contrib/chat/common/actions/chatContextKeys.ts 204 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatContextKeys.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../../../../nls.js';
7 > import { ContextKeyExpr, RawContextKey } from '../../../../../platform/contextkey/common/contextkey.js';
8 > import { IsWebContext } from '../../../../../platform/contextkey/common/contextkeys.js';
9 > import { RemoteNameContext } from '../../../../common/contextkeys.js';
10 > import { ViewContainerLocation } from '../../../../common/views.js';
11 > import { ChatEntitlementContextKeys } from '../../../../services/chat/common/chatEntitlementService.js';
12 > import { ChatAccountPolicyGateActiveContext } from '../../../../services/policies/common/accountPolicyService.js';
13 > import { ChatAgentLocation, ChatModeKind, ChatPermissionLevel } from '../constants.js';
14 >
15 > export namespace ChatContextKeys {
16 > export const responseVote = new RawContextKey<string>('chatSessionResponseVote', '', { type: 'string', description: localize('interactiveSessionResponseVote', "When the response has been voted up, is set to 'up'. When voted down, is set to 'down'. Otherwise an empty string.") });
17 > export const responseDetectedAgentCommand = new RawContextKey<boolean>('chatSessionResponseDetectedAgentOrCommand', false, { type: 'boolean', description: localize('chatSessionResponseDetectedAgentOrCommand', "When the agent or command was automatically detected") });
18 > export const responseSupportsIssueReporting = new RawContextKey<boolean>('chatResponseSupportsIssueReporting', false, { type: 'boolean', description: localize('chatResponseSupportsIssueReporting', "True when the current chat response supports issue reporting.") });
19 > export const responseIsFiltered = new RawContextKey<boolean>('chatSessionResponseFiltered', false, { type: 'boolean', description: localize('chatResponseFiltered', "True when the chat response was filtered out by the server.") });
20 > export const responseHasError = new RawContextKey<boolean>('chatSessionResponseError', false, { type: 'boolean', description: localize('chatResponseErrored', "True when the chat response resulted in an error.") });
21 > export const requestInProgress = new RawContextKey<boolean>('chatSessionRequestInProgress', false, { type: 'boolean', description: localize('interactiveSessionRequestInProgress', "True when the current request is still in progress.") });
22 > export const hasActiveRequest = new RawContextKey<boolean>('chatSessionHasActiveRequest', false, { type: 'boolean', description: localize('chatSessionHasActiveRequest', "True when the current chat response has not completed, regardless of intermediate states like tool calls or elicitations.") });
23 > export const currentlyEditing = new RawContextKey<boolean>('chatSessionCurrentlyEditing', false, { type: 'boolean', description: localize('interactiveSessionCurrentlyEditing', "True when the current request is being edited.") });
24 > export const currentlyEditingInput = new RawContextKey<boolean>('chatSessionCurrentlyEditingInput', false, { type: 'boolean', description: localize('interactiveSessionCurrentlyEditingInput', "True when the current request input at the bottom is being edited.") });
25 >
26 > export const enum EditingRequestType {
27 > Sent = 's',
28 > Queue = 'q',
29 > Steer = 'st',
30 > }
31 > export const editingRequestType = new RawContextKey<EditingRequestType | undefined>('chatEditingSentRequest', undefined, { type: 'string', description: localize('chatEditingSentRequest', "The type of the current editing request.") });
32 >
33 > export const isResponse = new RawContextKey<boolean>('chatResponse', false, { type: 'boolean', description: localize('chatResponse', "The chat item is a response.") });
34 > export const isRequest = new RawContextKey<boolean>('chatRequest', false, { type: 'boolean', description: localize('chatRequest', "The chat item is a request") });
35 > export const isFirstRequest = new RawContextKey<boolean>('chatFirstRequest', false, { type: 'boolean', description: localize('chatFirstRequest', "The chat item is the first request in the session.") });
36 > export const isPendingRequest = new RawContextKey<boolean>('chatRequestIsPending', false, { type: 'boolean', description: localize('chatRequestIsPending', "True when the chat request item is pending in the queue.") });
37 > export const itemId = new RawContextKey<string>('chatItemId', '', { type: 'string', description: localize('chatItemId', "The id of the chat item.") });
38 > export const lastItemId = new RawContextKey<string[]>('chatLastItemId', [], { type: 'string', description: localize('chatLastItemId', "The id of the last chat item.") });
39 >
40 > export const editApplied = new RawContextKey<boolean>('chatEditApplied', false, { type: 'boolean', description: localize('chatEditApplied', "True when the chat text edits have been applied.") });
41 >
42 > export const inputHasText = new RawContextKey<boolean>('chatInputHasText', false, { type: 'boolean', description: localize('interactiveInputHasText', "True when the chat input has text.") });
43 > export const inputHasSendableContent = new RawContextKey<boolean>('chatInputHasSendableContent', false, { type: 'boolean', description: localize('interactiveInputHasSendableContent', "True when the chat input has text or file attachments that can be sent.") });
44 > export const inputHasFocus = new RawContextKey<boolean>('chatInputHasFocus', false, { type: 'boolean', description: localize('interactiveInputHasFocus', "True when the chat input has focus.") });
45 > export const inChatInput = new RawContextKey<boolean>('inChatInput', false, { type: 'boolean', description: localize('inInteractiveInput', "True when focus is in the chat input, false otherwise.") });
46 > export const inChatSession = new RawContextKey<boolean>('inChat', false, { type: 'boolean', description: localize('inChat', "True when focus is in the chat widget, false otherwise.") });
47 > export const inChatQuestionCarousel = new RawContextKey<boolean>('inChatQuestionCarousel', false, { type: 'boolean', description: localize('inChatQuestionCarousel', "True when focus is in the chat question carousel.") });
48 > export const chatQuestionCarouselHasTerminal = new RawContextKey<boolean>('chatQuestionCarouselHasTerminal', false, { type: 'boolean', description: localize('chatQuestionCarouselHasTerminal', "True when the chat question carousel was triggered by a terminal and has a terminal to focus.") });
49 > export const inChatEditor = new RawContextKey<boolean>('inChatEditor', false, { type: 'boolean', description: localize('inChatEditor', "Whether focus is in a chat editor.") });
50 > export const inChatTodoList = new RawContextKey<boolean>('inChatTodoList', false, { type: 'boolean', description: localize('inChatTodoList', "True when focus is in the chat todo list.") });
51 > export const inChatTip = new RawContextKey<boolean>('inChatTip', false, { type: 'boolean', description: localize('inChatTip', "True when focus is in a chat tip.") });
52 > export const multipleChatTips = new RawContextKey<boolean>('multipleChatTips', false, { type: 'boolean', description: localize('multipleChatTips', "True when there are multiple chat tips available.") });
53 > export const inChatTerminalToolOutput = new RawContextKey<boolean>('inChatTerminalToolOutput', false, { type: 'boolean', description: localize('inChatTerminalToolOutput', "True when focus is in the chat terminal output region.") });
54 > export const chatModeKind = new RawContextKey<ChatModeKind>('chatAgentKind', ChatModeKind.Ask, { type: 'string', description: localize('agentKind', "The 'kind' of the current agent.") });
55 > export const chatPermissionLevel = new RawContextKey<ChatPermissionLevel>('chatPermissionLevel', ChatPermissionLevel.Default, { type: 'string', description: localize('chatPermissionLevel', "The current permission level for tool auto-approval.") });
56 > export const chatModeName = new RawContextKey<string>('chatModeName', '', { type: 'string', description: localize('chatModeName', "The name of the current chat mode (e.g. 'Plan' for custom modes).") });
57 > export const chatModelId = new RawContextKey<string>('chatModelId', '', { type: 'string', description: localize('chatModelId', "The short id of the currently selected chat model (for example 'gpt-4.1').") });
58 > export const speechToTextRecording = new RawContextKey<boolean>('chatSpeechToTextRecording', false, { type: 'boolean', description: localize('chatSpeechToTextRecording', "True while the chat input is recording audio for speech-to-text transcription.") });
59 > export const speechToTextConfigured = new RawContextKey<boolean>('chatSpeechToTextConfigured', false, { type: 'boolean', description: localize('chatSpeechToTextConfigured', "True when on-device speech-to-text is available for dictating into the chat input.") });
60 > export const speechToTextPreparing = new RawContextKey<boolean>('chatSpeechToTextPreparing', false, { type: 'boolean', description: localize('chatSpeechToTextPreparing', "True while the selected speech-to-text backend is preparing.") });
61 >
62 > export const supported = ContextKeyExpr.or(IsWebContext.negate(), RemoteNameContext.notEqualsTo(''), ContextKeyExpr.has('config.chat.experimental.serverlessWebEnabled'));
63 > export const enabled = new RawContextKey<boolean>('chatIsEnabled', false, { type: 'boolean', description: localize('chatIsEnabled', "True when chat is enabled because a default chat participant is activated with an implementation.") });
64 > export const accountPolicyGateActive = ChatAccountPolicyGateActiveContext;
65 >
66 > /**
67 > * True when the chat widget is locked to the coding agent session.
68 > */
69 > export const lockedToCodingAgent = new RawContextKey<boolean>('lockedToCodingAgent', false, { type: 'boolean', description: localize('lockedToCodingAgent', "True when the chat widget is locked to the coding agent session.") });
70 > export const lockedCodingAgentId = new RawContextKey<string>('lockedCodingAgentId', '', { type: 'string', description: localize('lockedCodingAgentId', "The agent ID when the chat widget is locked to a coding agent session.") });
71 > /**
72 > * Widget-scoped: true when the chat shown in this widget is read-only (non-interactive),
73 > * e.g. an observable worker chat. Read-only chats hide the composer and do not offer
74 > * mutating actions such as Start Over or Restore Checkpoint.
75 > */
76 > export const readOnly = new RawContextKey<boolean>('chatIsReadonly', false, { type: 'boolean', description: localize('chatIsReadonly', "True when the chat shown in the widget is read-only (non-interactive).") });
77 > /**
78 > * Widget-scoped: true when this chat widget is locked to an Agent Host-backed chat session.
79 > */
80 > export const chatIsAgentHostSession = new RawContextKey<boolean>('chatIsAgentHostSession', false, { type: 'boolean', description: localize('chatIsAgentHostSession', "True when the chat widget is locked to an Agent Host session.") });
81 > /**
82 > * Widget-scoped: logical Agent Host provider ID for this chat widget, e.g. `copilotcli`, `claude`, or `codex`.
83 > */
84 > export const chatAgentHostProviderId = new RawContextKey<string>('chatAgentHostProviderId', '', { type: 'string', description: localize('chatAgentHostProviderId', "The Agent Host provider ID when the chat widget is locked to an Agent Host session.") });
85 > /**
86 > * True when the chat session has a customAgentTarget defined in its contribution,
87 > * which means the mode picker should be shown with filtered custom agents.
88 > */
89 > export const chatSessionHasCustomAgentTarget = new RawContextKey<boolean>('chatSessionHasCustomAgentTarget', false, { type: 'boolean', description: localize('chatSessionHasCustomAgentTarget', "True when the chat session has a customAgentTarget defined to filter modes.") });
90 > /**
91 > * True when the current chat session has models that specifically target it
92 > * via `targetChatSessionType`, which means the model picker should be shown
93 > * even when the widget is locked to a coding agent.
94 > */
95 > export const chatSessionHasTargetedModels = new RawContextKey<boolean>('chatSessionHasTargetedModels', false, { type: 'boolean', description: localize('chatSessionHasTargetedModels', "True when the chat session has language models that target it via targetChatSessionType.") });
96 > export const agentSupportsAttachments = new RawContextKey<boolean>('agentSupportsAttachments', false, { type: 'boolean', description: localize('agentSupportsAttachments', "True when the chat agent supports attachments.") });
97 > export const withinEditSessionDiff = new RawContextKey<boolean>('withinEditSessionDiff', false, { type: 'boolean', description: localize('withinEditSessionDiff', "True when the chat widget dispatches to the edit session chat.") });
98 > export const filePartOfEditSession = new RawContextKey<boolean>('filePartOfEditSession', false, { type: 'boolean', description: localize('filePartOfEditSession', "True when the chat widget is within a file with an edit session.") });
99 >
100 > export const extensionParticipantRegistered = new RawContextKey<boolean>('chatPanelExtensionParticipantRegistered', false, { type: 'boolean', description: localize('chatPanelExtensionParticipantRegistered', "True when a default chat participant is registered for the panel from an extension.") });
101 > export const panelParticipantRegistered = new RawContextKey<boolean>('chatPanelParticipantRegistered', false, { type: 'boolean', description: localize('chatParticipantRegistered', "True when a default chat participant is registered for the panel.") });
102 > export const chatEditingCanUndo = new RawContextKey<boolean>('chatEditingCanUndo', false, { type: 'boolean', description: localize('chatEditingCanUndo', "True when it is possible to undo an interaction in the editing panel.") });
103 > export const chatEditingCanRedo = new RawContextKey<boolean>('chatEditingCanRedo', false, { type: 'boolean', description: localize('chatEditingCanRedo', "True when it is possible to redo an interaction in the editing panel.") });
104 > export const languageModelsAreUserSelectable = new RawContextKey<boolean>('chatModelsAreUserSelectable', false, { type: 'boolean', description: localize('chatModelsAreUserSelectable', "True when the chat model can be selected manually by the user.") });
105 > export const nonCopilotLanguageModelsAreUserSelectable = new RawContextKey<boolean>('chatNonCopilotModelsAreUserSelectable', false, { type: 'boolean', description: localize('chatNonCopilotModelsAreUserSelectable', "True when a user-selectable chat model from a non-Copilot vendor is available.") });
106 > export const chatSessionHasModels = new RawContextKey<boolean>('chatSessionHasModels', false, { type: 'boolean', description: localize('chatSessionHasModels', "True when the chat is in a contributed chat session that has available 'models' to display.") });
107 > export const chatSessionOptionsValid = new RawContextKey<boolean>('chatSessionOptionsValid', true, { type: 'boolean', description: localize('chatSessionOptionsValid', "True when all selected session options exist in their respective option group items.") });
108 > export const extensionInvalid = new RawContextKey<boolean>('chatExtensionInvalid', false, { type: 'boolean', description: localize('chatExtensionInvalid', "True when the installed chat extension is invalid and needs to be updated.") });
109 > export const inputCursorAtTop = new RawContextKey<boolean>('chatCursorAtTop', false);
110 > export const inputHasAgent = new RawContextKey<boolean>('chatInputHasAgent', false);
111 > export const location = new RawContextKey<ChatAgentLocation>('chatLocation', undefined);
112 > export const inQuickChat = new RawContextKey<boolean>('quickChatHasFocus', false, { type: 'boolean', description: localize('inQuickChat', "True when the quick chat UI has focus, false otherwise.") });
113 > export const inAgentSessionsWelcome = new RawContextKey<boolean>('inAgentSessionsWelcome', false, { type: 'boolean', description: localize('inAgentSessionsWelcome', "True when the chat input is within the agent sessions welcome page.") });
114 > export const inAutomationsDialog = new RawContextKey<boolean>('inAutomationsDialog', false, { type: 'boolean', description: localize('inAutomationsDialog', "True when the chat input is within the automations dialog.") });
115 > export const chatSessionType = new RawContextKey<string>('chatSessionType', '', { type: 'string', description: localize('chatSessionType', "The type of the current chat session.") });
116 > export const hasFileAttachments = new RawContextKey<boolean>('chatHasFileAttachments', false, { type: 'boolean', description: localize('chatHasFileAttachments', "True when the chat has file attachments.") });
117 > export const chatSessionIsEmpty = new RawContextKey<boolean>('chatSessionIsEmpty', true, { type: 'boolean', description: localize('chatSessionIsEmpty', "True when the current chat session has no requests.") });
118 > export const hasPendingRequests = new RawContextKey<boolean>('chatHasPendingRequests', false, { type: 'boolean', description: localize('chatHasPendingRequests', "True when there are pending requests in the queue.") });
119 > export const chatSessionHasDebugData = new RawContextKey<boolean>('chatSessionHasDebugData', false, { type: 'boolean', description: localize('chatSessionHasDebugData', "True when the current chat session has debug log data.") });
120 > export const chatSessionHasDebugTools = new RawContextKey<boolean>('chatSessionHasDebugTools', false, { type: 'boolean', description: localize('chatSessionHasDebugTools', "True when debug tools are enabled in the current chat session.") });
121 >
122 > export const remoteJobCreating = new RawContextKey<boolean>('chatRemoteJobCreating', false, { type: 'boolean', description: localize('chatRemoteJobCreating', "True when a remote coding agent job is being created.") });
123 > export const hasRemoteCodingAgent = new RawContextKey<boolean>('hasRemoteCodingAgent', false, localize('hasRemoteCodingAgent', "Whether any remote coding agent is available"));
124 > export const hasCanDelegateProviders = new RawContextKey<boolean>('chatHasCanDelegateProviders', false, { type: 'boolean', description: localize('chatHasCanDelegateProviders', "True when there are chat session providers with delegation support available.") });
125 > export const enableRemoteCodingAgentPromptFileOverlay = new RawContextKey<boolean>('enableRemoteCodingAgentPromptFileOverlay', false, localize('enableRemoteCodingAgentPromptFileOverlay', "Whether the remote coding agent prompt file overlay feature is enabled"));
126 > /** Used by the extension to skip the quit confirmation when #new wants to open a new folder */
127 > export const skipChatRequestInProgressMessage = new RawContextKey<boolean>('chatSkipRequestInProgressMessage', false, { type: 'boolean', description: localize('chatSkipRequestInProgressMessage', "True when the chat request in progress message should be skipped.") });
128 >
129 > // Re-exported from chat entitlement service
130 > export const Setup = ChatEntitlementContextKeys.Setup;
131 > export const Entitlement = ChatEntitlementContextKeys.Entitlement;
132 > export const chatQuotaExceeded = ChatEntitlementContextKeys.chatQuotaExceeded;
133 > export const completionsQuotaExceeded = ChatEntitlementContextKeys.completionsQuotaExceeded;
134 >
135 > export const Editing = {
136 > hasToolConfirmation: new RawContextKey<boolean>('chatHasToolConfirmation', false, { type: 'boolean', description: localize('chatEditingHasToolConfirmation', "True when a tool confirmation is present.") }),
137 > hasElicitationRequest: new RawContextKey<boolean>('chatHasElicitationRequest', false, { type: 'boolean', description: localize('chatEditingHasElicitationRequest', "True when a chat elicitation request is pending.") }),
138 > hasQuestionCarousel: new RawContextKey<boolean>('chatHasQuestionCarousel', false, { type: 'boolean', description: localize('chatEditingHasQuestionCarousel', "True when a question carousel is rendered in the chat input.") }),
139 > };
140 >
141 > export const Tools = {
142 > toolsCount: new RawContextKey<number>('toolsCount', 0, { type: 'number', description: localize('toolsCount', "The count of tools available in the chat.") })
143 > };
144 >
145 > export const foregroundSessionCount = new RawContextKey<number>('chatForegroundSessionCount', 0, { type: 'number', description: localize('chatForegroundSessionCount', "The number of foreground chat sessions visible across chat surfaces.") });
146 >
147 > export const Modes = {
148 > hasCustomChatModes: new RawContextKey<boolean>('chatHasCustomAgents', false, { type: 'boolean', description: localize('chatHasAgents', "True when the chat has custom agents available.") }),
149 > agentModeDisabledByPolicy: new RawContextKey<boolean>('chatAgentModeDisabledByPolicy', false, { type: 'boolean', description: localize('chatAgentModeDisabledByPolicy', "True when agent mode is disabled by organization policy.") }),
150 > };
151 >
152 > export const panelLocation = new RawContextKey<ViewContainerLocation>('chatPanelLocation', undefined, { type: 'number', description: localize('chatPanelLocation', "The location of the chat panel.") });
153 >
154 > export const agentSessionsViewerFocused = new RawContextKey<boolean>('agentSessionsViewerFocused', true, { type: 'boolean', description: localize('agentSessionsViewerFocused', "If the agent sessions view in the chat view is focused.") });
155 > export const agentSessionsViewerOrientation = new RawContextKey<number>('agentSessionsViewerOrientation', undefined, { type: 'number', description: localize('agentSessionsViewerOrientation', "Orientation of the agent sessions view in the chat view.") });
156 > export const agentSessionsViewerPosition = new RawContextKey<number>('agentSessionsViewerPosition', undefined, { type: 'number', description: localize('agentSessionsViewerPosition', "Position of the agent sessions view in the chat view.") });
157 > export const agentSessionsViewerVisible = new RawContextKey<boolean>('agentSessionsViewerVisible', undefined, { type: 'boolean', description: localize('agentSessionsViewerVisible', "Visibility of the agent sessions view in the chat view.") });
158 > export const agentSessionType = new RawContextKey<string>('chatSessionType', '', { type: 'string', description: localize('agentSessionType', "The type of the current agent session item.") });
159 > export const chatSessionSupportsDelegation = new RawContextKey<boolean>('chatSessionSupportsDelegation', true, { type: 'boolean', description: localize('chatSessionSupportsDelegation', "True when the current session type supports delegation.") });
160 > export const hasPendingDelegationTarget = new RawContextKey<boolean>('chatHasPendingDelegationTarget', false, { type: 'boolean', description: localize('chatHasPendingDelegationTarget', "True when a delegation (continue in) target is selected but the request has not been submitted yet.") });
161 > export const chatSessionSupportsFork = new RawContextKey<boolean>('chatSessionSupportsFork', false, { type: 'boolean', description: localize('chatSessionSupportsFork', "True when the current chat session provider supports forking conversations.") });
162 > export const agentSessionSection = new RawContextKey<string>('agentSessionSection', '', { type: 'string', description: localize('agentSessionSection', "The section of the current agent session section item.") });
163 > export const isArchivedAgentSession = new RawContextKey<boolean>('agentSessionIsArchived', false, { type: 'boolean', description: localize('agentSessionIsArchived', "True when the agent session item is archived.") });
164 > export const isPinnedAgentSession = new RawContextKey<boolean>('agentSessionIsPinned', false, { type: 'boolean', description: localize('agentSessionIsPinned', "True when the agent session item is pinned.") });
165 > export const isReadAgentSession = new RawContextKey<boolean>('agentSessionIsRead', false, { type: 'boolean', description: localize('agentSessionIsRead', "True when the agent session item is read.") });
166 > export const hasMultipleAgentSessionsSelected = new RawContextKey<boolean>('agentSessionHasMultipleSelected', false, { type: 'boolean', description: localize('agentSessionHasMultipleSelected', "True when multiple agent sessions are selected.") });
167 > export const hasAgentSessionChanges = new RawContextKey<boolean>('agentSessionHasChanges', false, { type: 'boolean', description: localize('agentSessionHasChanges', "True when the current agent session item has changes.") });
168 >
169 > export const isKatexMathElement = new RawContextKey<boolean>('chatIsKatexMathElement', false, { type: 'boolean', description: localize('chatIsKatexMathElement', "True when focusing a KaTeX math element.") });
170 >
171 > /**
172 > * True when the user has submitted a chat request using any of the `/create-*` slash commands.
173 > * This is persisted in application storage and used to suppress onboarding tips once discovered.
174 > */
175 > export const hasUsedCreateSlashCommands = new RawContextKey<boolean>('chatHasUsedCreateSlashCommands', false, { type: 'boolean', description: localize('chatHasUsedCreateSlashCommands', "True when the user has used any of the /create-* slash commands.") });
176 >
177 > export const contextUsageHasBeenOpened = new RawContextKey<boolean>('chatContextUsageHasBeenOpened', false, { type: 'boolean', description: localize('chatContextUsageHasBeenOpened', "True when the user has opened the context window usage details.") });
178 >
179 > export const newChatButtonExperimentIcon = new RawContextKey<string>('chatNewChatButtonExperimentIcon', '', { type: 'string', description: localize('chatNewChatButtonExperimentIcon', "The icon variant for the new chat button, controlled by experiment. Values: 'copilot', 'new-session', 'comment', or empty for default.") });
180 > }
181 >
182 > export namespace ChatContextKeyExprs {
183 >
184 > export const inEditingMode = ContextKeyExpr.or(
185 > ChatContextKeys.chatModeKind.isEqualTo(ChatModeKind.Edit),
186 > ChatContextKeys.chatModeKind.isEqualTo(ChatModeKind.Agent),
187 > );
188 >
189 > /**
190 > * True when the locked coding agent is an Agent Host session.
191 > * These sessions use {@link AgentHostSnapshotController} which supports checkpoint-based restore.
192 > */
193 > export const isAgentHostSession = ChatContextKeys.chatIsAgentHostSession.isEqualTo(true);
194 >
195 > /**
196 > * True when an agent session item (e.g. in the sessions viewer) is an agent
197 > * host session (agent-host-* or remote-*). Keyed on {@link ChatContextKeys.agentSessionType}
198 > * rather than the locked coding agent, for use in session item menus and keybindings.
199 > */
200 > export const isAgentHostSessionItem = ContextKeyExpr.or(
201 > ContextKeyExpr.regex(ChatContextKeys.agentSessionType.key, /^agent-host-/),
202 > ContextKeyExpr.regex(ChatContextKeys.agentSessionType.key, /^remote-/),
203 > );
204 > }
src/vs/platform/configuration/common/configuration.ts 203 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- configuration.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { assertNever } from '../../../base/common/assert.js';
7 > import { IStringDictionary } from '../../../base/common/collections.js';
8 > import { Event } from '../../../base/common/event.js';
9 > import * as types from '../../../base/common/types.js';
10 > import { URI, UriComponents } from '../../../base/common/uri.js';
11 > import { createDecorator } from '../../instantiation/common/instantiation.js';
12 > import { IWorkspaceFolder } from '../../workspace/common/workspace.js';
13 >
14 > export const IConfigurationService = createDecorator<IConfigurationService>('configurationService');
15 >
16 > export function isConfigurationOverrides(obj: unknown): obj is IConfigurationOverrides {
17 const thing = obj as IConfigurationOverrides;
18 return thing
21 && (!thing.resource || thing.resource instanceof URI);
22 }
24 > export interface IConfigurationOverrides {
25 > overrideIdentifier?: string | null;
26 > resource?: URI | null;
27 > }
28 >
29 > export function isConfigurationUpdateOverrides(obj: unknown): obj is IConfigurationUpdateOverrides {
30 const thing = obj as IConfigurationUpdateOverrides | IConfigurationOverrides;
31 return thing
35 && (!thing.resource || thing.resource instanceof URI);
36 }
38 > export type IConfigurationUpdateOverrides = Omit<IConfigurationOverrides, 'overrideIdentifier'> & { overrideIdentifiers?: string[] | null };
39 >
40 > export const enum ConfigurationTarget {
41 > APPLICATION = 1,
42 > USER,
43 > USER_LOCAL,
44 > USER_REMOTE,
45 > WORKSPACE,
46 > WORKSPACE_FOLDER,
47 > DEFAULT,
48 > MEMORY
49 > }
50 > export function ConfigurationTargetToString(configurationTarget: ConfigurationTarget) {
51 switch (configurationTarget) {
52 case ConfigurationTarget.APPLICATION: return 'APPLICATION';
60 }
61 }
63 > export interface IConfigurationChange {
64 > keys: string[];
65 > overrides: [string, string[]][];
66 > }
67 >
68 > export interface IConfigurationChangeEvent {
69 >
70 > readonly source: ConfigurationTarget;
71 > readonly affectedKeys: ReadonlySet<string>;
72 > readonly change: IConfigurationChange;
73 >
74 > affectsConfiguration(configuration: string, overrides?: IConfigurationOverrides): boolean;
75 > }
76 >
77 > export interface IInspectValue<T> {
78 > readonly value?: T;
79 > readonly override?: T;
80 > readonly overrides?: { readonly identifiers: string[]; readonly value: T }[];
81 > }
82 >
83 > export interface IConfigurationValue<T> {
84 >
85 > readonly defaultValue?: T;
86 > readonly applicationValue?: T;
87 > readonly userValue?: T;
88 > readonly userLocalValue?: T;
89 > readonly userRemoteValue?: T;
90 > readonly workspaceValue?: T;
91 > readonly workspaceFolderValue?: T;
92 > readonly memoryValue?: T;
93 > readonly policyValue?: T;
94 > readonly value?: T;
95 >
96 > readonly default?: IInspectValue<T>;
97 > readonly application?: IInspectValue<T>;
98 > readonly user?: IInspectValue<T>;
99 > readonly userLocal?: IInspectValue<T>;
100 > readonly userRemote?: IInspectValue<T>;
101 > readonly workspace?: IInspectValue<T>;
102 > readonly workspaceFolder?: IInspectValue<T>;
103 > readonly memory?: IInspectValue<T>;
104 > readonly policy?: { value?: T };
105 >
106 > readonly overrideIdentifiers?: string[];
107 > }
108 >
109 > export function getConfigValueInTarget<T>(configValue: IConfigurationValue<T>, scope: ConfigurationTarget): T | undefined {
110 switch (scope) {
111 case ConfigurationTarget.APPLICATION:
129 }
130 }
132 > export function isConfigured<T>(configValue: IConfigurationValue<T>): configValue is IConfigurationValue<T> & { value: T } {
133 return configValue.applicationValue !== undefined ||
134 configValue.userValue !== undefined ||
138 configValue.workspaceFolderValue !== undefined;
139 }
141 > export interface IConfigurationUpdateOptions {
142 > /**
143 > * If `true`, do not notifies the error to user by showing the message box. Default is `false`.
144 > */
145 > donotNotifyError?: boolean;
146 > /**
147 > * How to handle dirty file when updating the configuration.
148 > */
149 > handleDirtyFile?: 'save' | 'revert';
150 > }
151 >
152 > export interface IConfigurationService {
153 > readonly _serviceBrand: undefined;
154 >
155 > readonly onDidChangeConfiguration: Event<IConfigurationChangeEvent>;
156 >
157 > getConfigurationData(): IConfigurationData | null;
158 >
159 > /**
160 > * Fetches the value of the section for the given overrides.
161 > * Value can be of native type or an object keyed off the section name.
162 > *
163 > * @param section - Section of the configuration. Can be `null` or `undefined`.
164 > * @param overrides - Overrides that has to be applied while fetching
165 > *
166 > */
167 > getValue<T>(): T;
168 > getValue<T>(section: string): T;
169 > getValue<T>(overrides: IConfigurationOverrides): T;
170 > getValue<T>(section: string, overrides: IConfigurationOverrides): T;
171 >
172 > /**
173 > * Update a configuration value.
174 > *
175 > * Use `target` to update the configuration in a specific `ConfigurationTarget`.
176 > *
177 > * Use `overrides` to update the configuration for a resource or for override identifiers or both.
178 > *
179 > * Passing a resource through overrides will update the configuration in the workspace folder containing that resource.
180 > *
181 > * *Note 1:* Updating configuration to a default value will remove the configuration from the requested target. If not target is passed, it will be removed from all writeable targets.
182 > *
183 > * *Note 2:* Use `undefined` value to remove the configuration from the given target. If not target is passed, it will be removed from all writeable targets.
184 > *
185 > * Use `donotNotifyError` and set it to `true` to surpresss errors.
186 > *
187 > * @param key setting to be updated
188 > * @param value The new value
189 > */
190 > updateValue(key: string, value: unknown): Promise<void>;
191 > updateValue(key: string, value: unknown, target: ConfigurationTarget): Promise<void>;
192 > updateValue(key: string, value: unknown, overrides: IConfigurationOverrides | IConfigurationUpdateOverrides): Promise<void>;
193 > updateValue(key: string, value: unknown, overrides: IConfigurationOverrides | IConfigurationUpdateOverrides, target: ConfigurationTarget, options?: IConfigurationUpdateOptions): Promise<void>;
194 >
195 > inspect<T>(key: string, overrides?: IConfigurationOverrides): IConfigurationValue<Readonly<T>>;
196 >
197 > reloadConfiguration(target?: ConfigurationTarget | IWorkspaceFolder): Promise<void>;
198 >
199 > keys(): {
200 > default: string[];
201 > policy: string[];
202 > user: string[];
203 > workspace: string[];
204 > workspaceFolder: string[];
205 > memory?: string[];
206 > };
207 > }
208 >
209 > export interface IConfigurationModel {
210 > contents: IStringDictionary<unknown>;
211 > keys: string[];
212 > overrides: IOverrides[];
213 > raw?: ReadonlyArray<IStringDictionary<unknown>> | IStringDictionary<unknown>;
214 > }
215 >
216 > export interface IOverrides {
217 > keys: string[];
218 > contents: IStringDictionary<unknown>;
219 > identifiers: string[];
220 > }
221 >
222 > export interface IConfigurationData {
223 > defaults: IConfigurationModel;
224 > policy: IConfigurationModel;
225 > application: IConfigurationModel;
226 > userLocal: IConfigurationModel;
227 > userRemote: IConfigurationModel;
228 > workspace: IConfigurationModel;
229 > folders: [UriComponents, IConfigurationModel][];
230 > }
231 >
232 > export interface IConfigurationCompareResult {
233 > added: string[];
234 > removed: string[];
235 > updated: string[];
236 > overrides: [string, string[]][];
237 > }
238 >
239 > export function toValuesTree(properties: IStringDictionary<unknown>, conflictReporter: (message: string) => void): IStringDictionary<unknown> {
240 const root = Object.create(null);
241
246 return root;
247 }
249 > export function addToValueTree(settingsTreeRoot: IStringDictionary<unknown>, key: string, value: unknown, conflictReporter: (message: string) => void): void {
250 const segments = key.split('.');
251 const last = segments.pop()!;
282 }
283 }
285 > export function removeFromValueTree(valueTree: IStringDictionary<unknown>, key: string): void {
286 const segments = key.split('.');
287 doRemoveFromValueTree(valueTree, segments);
288 }
290 function doRemoveFromValueTree(valueTree: IStringDictionary<unknown> | unknown, segments: string[]): void {
291 if (!valueTree) {
311 }
312 }
314 > /**
315 > * A helper function to get the configuration value with a specific settings path (e.g. config.some.setting)
316 > */
317 > export function getConfigurationValue<T>(config: IStringDictionary<unknown>, settingPath: string): T | undefined;
318 > export function getConfigurationValue<T>(config: IStringDictionary<unknown>, settingPath: string, defaultValue: T): T;
319 > export function getConfigurationValue<T>(config: IStringDictionary<unknown>, settingPath: string, defaultValue?: T): T | undefined {
320 function accessSetting(config: IStringDictionary<unknown>, path: string[]): unknown {
321 let current: unknown = config;
334 return typeof result === 'undefined' ? defaultValue : result as T;
335 }
337 > export function merge(base: IStringDictionary<unknown>, add: IStringDictionary<unknown>, overwrite: boolean): void {
338 Object.keys(add).forEach(key => {
339 if (key !== '__proto__') {
350 });
351 }
353 > export function getLanguageTagSettingPlainKey(settingKey: string) {
354 return settingKey
355 .replace(/^\[/, '')
src/vs/platform/contextkey/common/scanner.ts 203 covered LOC · 58 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- scanner.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from '../../../base/common/charCode.js';
7 > import { illegalState } from '../../../base/common/errors.js';
8 > import { localize } from '../../../nls.js';
9 >
10 > export const enum TokenType {
11 > LParen,
12 > RParen,
13 > Neg,
14 > Eq,
15 > NotEq,
16 > Lt,
17 > LtEq,
18 > Gt,
19 > GtEq,
20 > RegexOp,
21 > RegexStr,
22 > True,
23 > False,
24 > In,
25 > Not,
26 > And,
27 > Or,
28 > Str,
29 > QuotedStr,
30 > Error,
31 > EOF,
32 > }
33 >
34 > export type Token =
35 > | { type: TokenType.LParen; offset: number }
36 > | { type: TokenType.RParen; offset: number }
37 > | { type: TokenType.Neg; offset: number }
38 > | { type: TokenType.Eq; offset: number; isTripleEq: boolean }
39 > | { type: TokenType.NotEq; offset: number; isTripleEq: boolean }
40 > | { type: TokenType.Lt; offset: number }
41 > | { type: TokenType.LtEq; offset: number }
42 > | { type: TokenType.Gt; offset: number }
43 > | { type: TokenType.GtEq; offset: number }
44 > | { type: TokenType.RegexOp; offset: number }
45 > | { type: TokenType.RegexStr; offset: number; lexeme: string }
46 > | { type: TokenType.True; offset: number }
47 > | { type: TokenType.False; offset: number }
48 > | { type: TokenType.In; offset: number }
49 > | { type: TokenType.Not; offset: number }
50 > | { type: TokenType.And; offset: number }
51 > | { type: TokenType.Or; offset: number }
52 > | { type: TokenType.Str; offset: number; lexeme: string }
53 > | { type: TokenType.QuotedStr; offset: number; lexeme: string }
54 > | { type: TokenType.Error; offset: number; lexeme: string }
55 > | { type: TokenType.EOF; offset: number };
56 >
57 > type KeywordTokenType = TokenType.Not | TokenType.In | TokenType.False | TokenType.True;
58 > type TokenTypeWithoutLexeme =
59 > TokenType.LParen |
60 > TokenType.RParen |
61 > TokenType.Neg |
62 > TokenType.Lt |
63 > TokenType.LtEq |
64 > TokenType.Gt |
65 > TokenType.GtEq |
66 > TokenType.RegexOp |
67 > TokenType.True |
68 > TokenType.False |
69 > TokenType.In |
70 > TokenType.Not |
71 > TokenType.And |
72 > TokenType.Or |
73 > TokenType.EOF;
74 >
75 > /**
76 > * Example:
77 > * `foo == bar'` - note how single quote doesn't have a corresponding closing quote,
78 > * so it's reported as unexpected
79 > */
80 > export type LexingError = {
81 > offset: number; /** note that this doesn't take into account escape characters from the original encoding of the string, e.g., within an extension manifest file's JSON encoding */
82 > lexeme: string;
83 > additionalInfo?: string;
84 > };
85 >
86 function hintDidYouMean(...meant: string[]) {
87 switch (meant.length) {
96 }
97 }
98 > scanner.ts
99 > const hintDidYouForgetToOpenOrCloseQuote = localize('contextkey.scanner.hint.didYouForgetToOpenOrCloseQuote', "Did you forget to open or close the quote?");
100 > const hintDidYouForgetToEscapeSlash = localize('contextkey.scanner.hint.didYouForgetToEscapeSlash', "Did you forget to escape the '/' (slash) character? Put two backslashes before it to escape, e.g., '\\\\/\'.");
101 >
102 > /**
103 > * A simple scanner for context keys.
104 > *
105 > * Example:
106 > *
107 > * ```ts
108 > * const scanner = new Scanner().reset('resourceFileName =~ /docker/ && !config.docker.enabled');
109 > * const tokens = [...scanner];
110 > * if (scanner.errorTokens.length > 0) {
111 > * scanner.errorTokens.forEach(err => console.error(`Unexpected token at ${err.offset}: ${err.lexeme}\nHint: ${err.additional}`));
112 > * } else {
113 > * // process tokens
114 > * }
115 > * ```
116 > */
117 > export class Scanner {
118 >
119 > static getLexeme(token: Token): string {
120 > switch (token.type) {
121 > case TokenType.LParen:
122 > return '('; scanner.ts
123 > case TokenType.RParen: scanner.ts
124 > return ')'; scanner.ts
125 > case TokenType.Neg: scanner.ts
126 > return '!'; scanner.ts
127 > case TokenType.Eq: scanner.ts
128 > return token.isTripleEq ? '===' : '=='; scanner.ts
129 > case TokenType.NotEq: scanner.ts
130 > return token.isTripleEq ? '!==' : '!='; scanner.ts
131 > case TokenType.Lt: scanner.ts
132 > return '<'; scanner.ts
133 > case TokenType.LtEq: scanner.ts
134 > return '<='; scanner.ts
135 > case TokenType.Gt: scanner.ts
136 > return '>'; scanner.ts
137 > case TokenType.GtEq: scanner.ts
138 > return '>='; scanner.ts
139 > case TokenType.RegexOp: scanner.ts
140 > return '=~'; scanner.ts
141 > case TokenType.RegexStr: scanner.ts
142 > return token.lexeme; scanner.ts
143 > case TokenType.True: scanner.ts
144 > return 'true'; scanner.ts
145 > case TokenType.False: scanner.ts
146 > return 'false'; scanner.ts
147 > case TokenType.In: scanner.ts
148 > return 'in'; scanner.ts
149 > case TokenType.Not: scanner.ts
150 > return 'not'; scanner.ts
151 > case TokenType.And: scanner.ts
152 > return '&&'; scanner.ts
153 > case TokenType.Or: scanner.ts
154 > return '||'; scanner.ts
155 > case TokenType.Str: scanner.ts
156 > return token.lexeme; scanner.ts
157 > case TokenType.QuotedStr: scanner.ts
158 > return token.lexeme; scanner.ts
159 > case TokenType.Error: scanner.ts
160 > return token.lexeme; scanner.ts
161 > case TokenType.EOF: scanner.ts
162 > return 'EOF'; scanner.ts
163 > default: scanner.ts
164 > throw illegalState(`unhandled token type: ${JSON.stringify(token)}; have you forgotten to add a case?`); scanner.ts
165 > } scanner.ts
166 > }
167 >
168 > private static _regexFlags = new Set(['i', 'g', 's', 'm', 'y', 'u'].map(ch => ch.charCodeAt(0)));
169 >
170 > private static _keywords = new Map<string, KeywordTokenType>([
171 > ['not', TokenType.Not],
172 > ['in', TokenType.In],
173 > ['false', TokenType.False],
174 > ['true', TokenType.True],
175 > ]);
176 >
177 > private _input: string = '';
178 > private _start: number = 0;
179 > private _current: number = 0;
180 > private _tokens: Token[] = [];
181 > private _errors: LexingError[] = [];
182 >
183 > get errors(): Readonly<LexingError[]> {
184 return this._errors;
185 }
186 > scanner.ts
187 > reset(value: string) {
188 this._input = value;
189
195 return this;
196 }
197 > scanner.ts
198 > scan() {
199 while (!this._isAtEnd()) {
200
267 return Array.from(this._tokens);
268 }
269 > scanner.ts
270 > private _match(expected: number): boolean {
271 if (this._isAtEnd()) {
272 return false;
278 return true;
279 }
280 > scanner.ts
281 > private _advance(): number {
282 return this._input.charCodeAt(this._current++);
283 }
284 > scanner.ts
285 > private _peek(): number {
286 return this._isAtEnd() ? CharCode.Null : this._input.charCodeAt(this._current);
287 }
288 > scanner.ts
289 > private _addToken(type: TokenTypeWithoutLexeme) {
290 this._tokens.push({ type, offset: this._start });
291 }
292 > scanner.ts
293 > private _error(additional?: string) {
294 const offset = this._start;
295 const lexeme = this._input.substring(this._start, this._current);
298 this._tokens.push(errToken);
299 }
300 > scanner.ts
301 > // u - unicode, y - sticky // TODO@ulugbekna: we accept double quotes as part of the string rather than as a delimiter (to preserve old parser's behavior)
302 > private stringRe = /[a-zA-Z0-9_<>\-\./\\:\*\?\+\[\]\^,#@;"%\$\p{L}-]+/uy;
303 > private _string() {
304 this.stringRe.lastIndex = this._start;
305 const match = this.stringRe.exec(this._input);
315 }
316 }
317 > scanner.ts
318 > // captures the lexeme without the leading and trailing '
319 > private _quotedString() {
320 while (this._peek() !== CharCode.SingleQuote && !this._isAtEnd()) { // TODO@ulugbekna: add support for escaping ' ?
321 this._advance();
332 this._tokens.push({ type: TokenType.QuotedStr, lexeme: this._input.substring(this._start + 1, this._current - 1), offset: this._start + 1 });
333 }
334 > scanner.ts
335 > /*
336 > * Lexing a regex expression: /.../[igsmyu]*
337 > * Based on https://github.com/microsoft/TypeScript/blob/9247ef115e617805983740ba795d7a8164babf89/src/compiler/scanner.ts#L2129-L2181
338 > *
339 > * Note that we want slashes within a regex to be escaped, e.g., /file:\\/\\/\\// should match `file:///`
340 > */
341 > private _regex() {
342 let p = this._current;
343
378 this._tokens.push({ type: TokenType.RegexStr, lexeme, offset: this._start });
379 }
380 > scanner.ts
381 > private _isAtEnd() {
382 return this._current >= this._input.length;
383 }
384 > } scanner.ts
src/vs/workbench/contrib/chat/common/model/chatModelStore.ts 203 covered LOC · 42 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatModelStore.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter } from '../../../../../base/common/event.js';
7 > import { Disposable, IReference, ReferenceCollection } from '../../../../../base/common/lifecycle.js';
8 > import { IObservable, ObservableMap } from '../../../../../base/common/observable.js';
9 > import { URI } from '../../../../../base/common/uri.js';
10 > import { ILogService } from '../../../../../platform/log/common/log.js';
11 > import { ChatAgentLocation } from '../constants.js';
12 > import { IChatEditingSession, ModifiedFileEntryState } from '../editing/chatEditingService.js';
13 > import { ChatModel, ISerializableChatModelInputState, ISerializedChatDataReference } from './chatModel.js';
14 >
15 > export interface IStartSessionProps {
16 > readonly initialData?: ISerializedChatDataReference;
17 > readonly location: ChatAgentLocation;
18 > readonly sessionResource: URI;
19 > readonly canUseTools: boolean;
20 > readonly transferEditingSession?: IChatEditingSession;
21 > readonly disableBackgroundKeepAlive?: boolean;
22 > readonly inputState?: ISerializableChatModelInputState;
23 > readonly isReadOnly?: IObservable<boolean>;
24 > }
25 >
26 > export interface ChatModelStoreDelegate {
27 > createModel: (props: IStartSessionProps) => ChatModel;
28 > willDisposeModel: (model: ChatModel) => Promise<void>;
29 > }
30 >
31 > export interface IChatModelReferenceDebugHolder {
32 > readonly holder: string;
33 > readonly count: number;
34 > }
35 >
36 > export interface IChatModelReferenceDebugInfo {
37 > readonly sessionResource: URI;
38 > readonly title: string;
39 > readonly createdBy: string;
40 > readonly initialLocation: ChatAgentLocation;
41 > readonly isImported: boolean;
42 > readonly willKeepAlive: boolean;
43 > readonly hasPendingEdits: boolean;
44 > readonly pendingDisposal: boolean;
45 > readonly referenceCount: number;
46 > readonly holders: readonly IChatModelReferenceDebugHolder[];
47 > }
48 >
49 > export interface IChatModelReferenceDebugSnapshot {
50 > readonly totalModels: number;
51 > readonly totalReferences: number;
52 > readonly models: readonly IChatModelReferenceDebugInfo[];
53 > }
54 >
55 > export class ChatModelStore extends Disposable {
56 > private readonly _refCollection: ReferenceCollection<ChatModel>;
57 >
58 > private readonly _models = new ObservableMap<string, ChatModel>();
59 > private readonly _modelsToDispose = new Set<string>();
60 > private readonly _pendingDisposals = new Set<Promise<void>>();
61 > private readonly _modelCreateOwners = new Map<string, string>();
62 > private readonly _referenceOwners = new Map<string, Map<number, string>>();
63 > private _referenceOwnerIds = 0;
64 >
65 > private readonly _onDidDisposeModel = this._register(new Emitter<ChatModel>());
66 > public readonly onDidDisposeModel = this._onDidDisposeModel.event;
67 >
68 > private readonly _onDidCreateModel = this._register(new Emitter<ChatModel>());
69 > public readonly onDidCreateModel = this._onDidCreateModel.event;
70 >
71 > constructor(
72 > private readonly delegate: ChatModelStoreDelegate, chatModelStore.ts
73 > @ILogService private readonly logService: ILogService,
74 > ) {
75 > super();
76 >
77 > const self = this;
78 > this._refCollection = new class extends ReferenceCollection<ChatModel> {
79 > protected createReferencedObject(key: string, props?: IStartSessionProps, debugOwner?: string): ChatModel {
80 > return self.createReferencedObject(key, props, debugOwner); chatModelStore.ts
81 > }
82 > protected destroyReferencedObject(key: string, object: ChatModel): void { chatModelStore.ts
83 > return self.destroyReferencedObject(key, object); chatModelStore.ts
84 > }
85 > }(); chatModelStore.ts
86 > }
88 > public get observable() {
89 return this._models.observable;
90 }
92 > public values(): Iterable<ChatModel> {
93 return this._models.values();
94 }
96 > /**
97 > * Get a ChatModel directly without acquiring a reference.
98 > */
99 > public get(uri: URI): ChatModel | undefined {
100 > return this._models.get(this.toKey(uri)); chatModelStore.ts
101 > }
103 > public has(uri: URI): boolean {
104 return this._models.has(this.toKey(uri));
105 }
107 > public acquireExisting(uri: URI, debugOwner?: string): IReference<ChatModel> | undefined {
108 > const key = this.toKey(uri); chatModelStore.ts
109 > if (!this._models.has(key)) {
110 > return undefined; chatModelStore.ts
111 > }
112
113 return this.wrapReference(key, this._refCollection.acquire(key, undefined, debugOwner), debugOwner);
116 > public acquireOrCreate(props: IStartSessionProps, debugOwner?: string): IReference<ChatModel> {
117 > const key = this.toKey(props.sessionResource); chatModelStore.ts
118 > return this.wrapReference(key, this._refCollection.acquire(key, props, debugOwner), debugOwner);
119 > }
121 > public getReferenceDebugSnapshot(): IChatModelReferenceDebugSnapshot {
122 const models = Array.from(this._models.values())
123 .map(model => {
154 };
155 }
157 > private createReferencedObject(key: string, props?: IStartSessionProps, debugOwner?: string): ChatModel {
158 > this._modelsToDispose.delete(key); chatModelStore.ts
159 > const existingModel = this._models.get(key);
160 > if (existingModel) {
161 return existingModel;
162 }
164 > if (!props) {
165 throw new Error(`No start session props provided for chat session ${key}`);
166 }
168 > this.logService.trace(`Creating chat session ${key}`);
169 > const model = this.delegate.createModel(props);
170 > this._modelCreateOwners.set(key, debugOwner ?? 'unspecified');
171 > if (model.sessionResource.toString() !== key) {
172 throw new Error(`Chat session key mismatch for ${key}`);
173 }
174 > this._models.set(key, model); chatModelStore.ts
175 > this._onDidCreateModel.fire(model);
176 > return model;
177 > }
179 > private destroyReferencedObject(key: string, object: ChatModel): void {
180 > this._modelsToDispose.add(key); chatModelStore.ts
181 > const promise = this.doDestroyReferencedObject(key, object);
182 > this._pendingDisposals.add(promise);
183 > promise.finally(() => {
184 > this._pendingDisposals.delete(promise);
185 > });
186 > }
188 > private async doDestroyReferencedObject(key: string, object: ChatModel): Promise<void> {
189 > try { chatModelStore.ts
190 > await this.delegate.willDisposeModel(object);
191 > } catch (error) {
192 this.logService.error(error);
193 > } finally { chatModelStore.ts
194 > if (this._modelsToDispose.has(key)) {
195 > this.logService.trace(`Disposing chat session ${key}`); chatModelStore.ts
196 > this._models.delete(key);
197 > this._modelCreateOwners.delete(key);
198 > this._referenceOwners.delete(key);
199 > this._onDidDisposeModel.fire(object);
200 > object.dispose();
201 > }
202 > this._modelsToDispose.delete(key); chatModelStore.ts
203 > }
204 > }
206 > private wrapReference(key: string, reference: IReference<ChatModel>, debugOwner?: string): IReference<ChatModel> {
207 > const ownerId = ++this._referenceOwnerIds; chatModelStore.ts
208 > let ownerEntries = this._referenceOwners.get(key);
209 > if (!ownerEntries) {
210 > ownerEntries = new Map();
211 > this._referenceOwners.set(key, ownerEntries);
212 > }
213 > ownerEntries.set(ownerId, debugOwner ?? 'unspecified');
214 >
215 > let isDisposed = false;
216 > const wrapped: IReference<ChatModel> = {
217 > object: reference.object,
218 > dispose: () => {
219 > if (isDisposed) { chatModelStore.ts
220 return;
221 }
223 > isDisposed = true;
224 > const owners = this._referenceOwners.get(key);
225 > owners?.delete(ownerId);
226 > if (owners?.size === 0) {
227 > this._referenceOwners.delete(key);
228 > }
229 > reference.dispose();
230 >
231 > // Break the reference from this wrapper to the ChatModel so that
232 > // stale holders of this IChatModelReference cannot retain the model.
233 > (wrapped as { object: ChatModel | null }).object = null;
234 > }
236 > return wrapped;
237 > }
239 > /**
240 > * For test use only
241 > */
242 > async waitForModelDisposals(): Promise<void> {
243 > await Promise.all(this._pendingDisposals); chatModelStore.ts
244 > }
246 > private toKey(uri: URI): string {
247 > return uri.toString(); chatModelStore.ts
248 > }
250 > override dispose(): void {
251 > super.dispose(); chatModelStore.ts
252 > this._models.forEach(model => model.dispose());
253 > }
src/vs/workbench/services/mcp/common/mcpWorkbenchManagementService.ts 203 covered LOC · 30 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpWorkbenchManagementService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DisposableStore, IDisposable } from '../../../../base/common/lifecycle.js';
7 > import { ILocalMcpServer, IMcpManagementService, IGalleryMcpServer, InstallOptions, InstallMcpServerEvent, UninstallMcpServerEvent, DidUninstallMcpServerEvent, InstallMcpServerResult, IInstallableMcpServer, IMcpGalleryService, UninstallOptions, IAllowedMcpServersService, RegistryType } from '../../../../platform/mcp/common/mcpManagement.js';
8 > import { IInstantiationService, refineServiceDecorator } from '../../../../platform/instantiation/common/instantiation.js';
9 > import { IUserDataProfileService } from '../../../services/userDataProfile/common/userDataProfile.js';
10 > import { Emitter, Event } from '../../../../base/common/event.js';
11 > import { IMcpResourceScannerService, McpResourceTarget } from '../../../../platform/mcp/common/mcpResourceScannerService.js';
12 > import { isWorkspaceFolder, IWorkspaceContextService, IWorkspaceFolder, IWorkspaceFoldersChangeEvent } from '../../../../platform/workspace/common/workspace.js';
13 > import { IUriIdentityService } from '../../../../platform/uriIdentity/common/uriIdentity.js';
14 > import { MCP_CONFIGURATION_KEY, WORKSPACE_STANDALONE_CONFIGURATIONS } from '../../configuration/common/configuration.js';
15 > import { ILogService } from '../../../../platform/log/common/log.js';
16 > import { IRemoteAgentService } from '../../remote/common/remoteAgentService.js';
17 > import { URI } from '../../../../base/common/uri.js';
18 > import { ConfigurationTarget } from '../../../../platform/configuration/common/configuration.js';
19 > import { IChannel } from '../../../../base/parts/ipc/common/ipc.js';
20 > import { McpManagementChannelClient } from '../../../../platform/mcp/common/mcpManagementIpc.js';
21 > import { IUserDataProfilesService } from '../../../../platform/userDataProfile/common/userDataProfile.js';
22 > import { IRemoteUserDataProfilesService } from '../../userDataProfile/common/remoteUserDataProfiles.js';
23 > import { AbstractMcpManagementService, AbstractMcpResourceManagementService, ILocalMcpServerInfo } from '../../../../platform/mcp/common/mcpManagementService.js';
24 > import { IFileService } from '../../../../platform/files/common/files.js';
25 > import { ResourceMap } from '../../../../base/common/map.js';
26 > import { IMarkdownString } from '../../../../base/common/htmlContent.js';
27 > import { IMcpServerConfiguration } from '../../../../platform/mcp/common/mcpPlatformTypes.js';
28 >
29 > export const USER_CONFIG_ID = 'usrlocal';
30 > export const REMOTE_USER_CONFIG_ID = 'usrremote';
31 > export const WORKSPACE_CONFIG_ID = 'workspace';
32 > export const WORKSPACE_FOLDER_CONFIG_ID_PREFIX = 'ws';
33 >
34 > export interface IWorkbencMcpServerInstallOptions extends InstallOptions {
35 > target?: ConfigurationTarget | IWorkspaceFolder;
36 > }
37 >
38 > export const enum LocalMcpServerScope {
39 > User = 'user',
40 > RemoteUser = 'remoteUser',
41 > Workspace = 'workspace',
42 > }
43 >
44 > export interface IWorkbenchLocalMcpServer extends ILocalMcpServer {
45 > readonly id: string;
46 > readonly scope: LocalMcpServerScope;
47 > }
48 >
49 > export interface InstallWorkbenchMcpServerEvent extends InstallMcpServerEvent {
50 > readonly scope: LocalMcpServerScope;
51 > }
52 >
53 > export interface IWorkbenchMcpServerInstallResult extends InstallMcpServerResult {
54 > readonly local?: IWorkbenchLocalMcpServer;
55 > }
56 >
57 > export interface UninstallWorkbenchMcpServerEvent extends UninstallMcpServerEvent {
58 > readonly scope: LocalMcpServerScope;
59 > }
60 >
61 > export interface DidUninstallWorkbenchMcpServerEvent extends DidUninstallMcpServerEvent {
62 > readonly scope: LocalMcpServerScope;
63 > }
64 >
65 > export const IWorkbenchMcpManagementService = refineServiceDecorator<IMcpManagementService, IWorkbenchMcpManagementService>(IMcpManagementService);
66 > export interface IWorkbenchMcpManagementService extends IMcpManagementService {
67 > readonly _serviceBrand: undefined;
68 >
69 > readonly onInstallMcpServerInCurrentProfile: Event<InstallWorkbenchMcpServerEvent>;
70 > readonly onDidInstallMcpServersInCurrentProfile: Event<readonly IWorkbenchMcpServerInstallResult[]>;
71 > readonly onDidUpdateMcpServersInCurrentProfile: Event<readonly IWorkbenchMcpServerInstallResult[]>;
72 > readonly onUninstallMcpServerInCurrentProfile: Event<UninstallWorkbenchMcpServerEvent>;
73 > readonly onDidUninstallMcpServerInCurrentProfile: Event<DidUninstallWorkbenchMcpServerEvent>;
74 > readonly onDidChangeProfile: Event<void>;
75 >
76 > getInstalled(): Promise<IWorkbenchLocalMcpServer[]>;
77 > install(server: IInstallableMcpServer | URI, options?: IWorkbencMcpServerInstallOptions): Promise<IWorkbenchLocalMcpServer>;
78 > installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<IWorkbenchLocalMcpServer>;
79 > updateMetadata(local: ILocalMcpServer, server: IGalleryMcpServer, profileLocation?: URI): Promise<IWorkbenchLocalMcpServer>;
80 > }
81 >
82 > export class WorkbenchMcpManagementService extends AbstractMcpManagementService implements IWorkbenchMcpManagementService {
83 >
84 > private _onInstallMcpServer = this._register(new Emitter<InstallMcpServerEvent>());
85 > readonly onInstallMcpServer = this._onInstallMcpServer.event;
86 >
87 > private _onDidInstallMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
88 > readonly onDidInstallMcpServers = this._onDidInstallMcpServers.event;
89 >
90 > private _onDidUpdateMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
91 > readonly onDidUpdateMcpServers = this._onDidUpdateMcpServers.event;
92 >
93 > private _onUninstallMcpServer = this._register(new Emitter<UninstallMcpServerEvent>());
94 > readonly onUninstallMcpServer = this._onUninstallMcpServer.event;
95 >
96 > private _onDidUninstallMcpServer = this._register(new Emitter<DidUninstallMcpServerEvent>());
97 > readonly onDidUninstallMcpServer = this._onDidUninstallMcpServer.event;
98 >
99 > private readonly _onInstallMcpServerInCurrentProfile = this._register(new Emitter<InstallWorkbenchMcpServerEvent>());
100 > readonly onInstallMcpServerInCurrentProfile = this._onInstallMcpServerInCurrentProfile.event;
101 >
102 > private readonly _onDidInstallMcpServersInCurrentProfile = this._register(new Emitter<readonly IWorkbenchMcpServerInstallResult[]>());
103 > readonly onDidInstallMcpServersInCurrentProfile = this._onDidInstallMcpServersInCurrentProfile.event;
104 >
105 > private readonly _onDidUpdateMcpServersInCurrentProfile = this._register(new Emitter<readonly IWorkbenchMcpServerInstallResult[]>());
106 > readonly onDidUpdateMcpServersInCurrentProfile = this._onDidUpdateMcpServersInCurrentProfile.event;
107 >
108 > private readonly _onUninstallMcpServerInCurrentProfile = this._register(new Emitter<UninstallWorkbenchMcpServerEvent>());
109 > readonly onUninstallMcpServerInCurrentProfile = this._onUninstallMcpServerInCurrentProfile.event;
110 >
111 > private readonly _onDidUninstallMcpServerInCurrentProfile = this._register(new Emitter<DidUninstallWorkbenchMcpServerEvent>());
112 > readonly onDidUninstallMcpServerInCurrentProfile = this._onDidUninstallMcpServerInCurrentProfile.event;
113 >
114 > private readonly _onDidChangeProfile = this._register(new Emitter<void>());
115 > readonly onDidChangeProfile = this._onDidChangeProfile.event;
116 >
117 > private readonly workspaceMcpManagementService: IMcpManagementService;
118 > private readonly remoteMcpManagementService: IMcpManagementService | undefined;
119 >
120 > constructor(
121 private readonly mcpManagementService: IMcpManagementService,
122 @IAllowedMcpServersService allowedMcpServersService: IAllowedMcpServersService,
237 }));
238 }
240 > private createInstallMcpServerResultsFromEvent(e: readonly InstallMcpServerResult[], scope: LocalMcpServerScope): { mcpServerInstallResult: IWorkbenchMcpServerInstallResult[]; mcpServerInstallResultInCurrentProfile: IWorkbenchMcpServerInstallResult[] } {
241 const mcpServerInstallResult: IWorkbenchMcpServerInstallResult[] = [];
242 const mcpServerInstallResultInCurrentProfile: IWorkbenchMcpServerInstallResult[] = [];
254 return { mcpServerInstallResult, mcpServerInstallResultInCurrentProfile };
255 }
257 > private async handleRemoteInstallMcpServerResultsFromEvent(e: readonly InstallMcpServerResult[], emitter: Emitter<readonly InstallMcpServerResult[]>, currentProfileEmitter: Emitter<readonly IWorkbenchMcpServerInstallResult[]>): Promise<void> {
258 const mcpServerInstallResult: IWorkbenchMcpServerInstallResult[] = [];
259 const mcpServerInstallResultInCurrentProfile: IWorkbenchMcpServerInstallResult[] = [];
275 }
276 }
278 > async getInstalled(): Promise<IWorkbenchLocalMcpServer[]> {
279 const installed: IWorkbenchLocalMcpServer[] = [];
280 const [userServers, remoteServers, workspaceServers] = await Promise.all([
296 return installed;
297 }
299 > private toWorkspaceMcpServer(server: ILocalMcpServer, scope: LocalMcpServerScope): IWorkbenchLocalMcpServer {
300 return { ...server, id: `mcp.config.${this.getConfigId(server, scope)}.${server.name}`, scope };
301 }
303 > private getConfigId(server: ILocalMcpServer, scope: LocalMcpServerScope): string {
304 if (scope === LocalMcpServerScope.User) {
305 return USER_CONFIG_ID;
326 return 'unknown';
327 }
329 > async install(server: IInstallableMcpServer, options?: IWorkbencMcpServerInstallOptions): Promise<IWorkbenchLocalMcpServer> {
330 options = options ?? {};
331
357 return this.toWorkspaceMcpServer(result, LocalMcpServerScope.User);
358 }
360 > async installFromGallery(server: IGalleryMcpServer, options?: IWorkbencMcpServerInstallOptions): Promise<IWorkbenchLocalMcpServer> {
361 options = options ?? {};
362
390 return this.toWorkspaceMcpServer(result, LocalMcpServerScope.User);
391 }
393 > async updateMetadata(local: IWorkbenchLocalMcpServer, server: IGalleryMcpServer, profileLocation: URI): Promise<IWorkbenchLocalMcpServer> {
394 if (local.scope === LocalMcpServerScope.Workspace) {
395 const result = await this.workspaceMcpManagementService.updateMetadata(local, server, profileLocation);
408 return this.toWorkspaceMcpServer(result, LocalMcpServerScope.User);
409 }
411 > async uninstall(server: IWorkbenchLocalMcpServer): Promise<void> {
412 if (server.scope === LocalMcpServerScope.Workspace) {
413 return this.workspaceMcpManagementService.uninstall(server);
423 return this.mcpManagementService.uninstall(server, { mcpResource: this.userDataProfileService.currentProfile.mcpResource });
424 }
426 > private async getRemoteMcpResource(mcpResource?: URI): Promise<URI | undefined> {
427 if (!mcpResource && this.userDataProfileService.currentProfile.isDefault) {
428 return undefined;
437 return profile?.mcpResource;
438 }
440 >
441 > class WorkspaceMcpResourceManagementService extends AbstractMcpResourceManagementService {
442 >
443 > constructor(
444 mcpResource: URI,
445 target: McpResourceTarget,
453 super(mcpResource, target, mcpGalleryService, fileService, uriIdentityService, logService, mcpResourceScannerService, allowedMcpServersService);
454 }
456 > override async installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
457 this.logService.trace('MCP Management Service: installGallery', server.name, server.galleryUrl);
458
493 }
494 }
496 > override updateMetadata(): Promise<ILocalMcpServer> {
497 throw new Error('Not supported');
498 }
500 > protected override installFromUri(): Promise<ILocalMcpServer> {
501 throw new Error('Not supported');
502 }
504 > protected override async getLocalServerInfo(name: string, mcpServerConfig: IMcpServerConfiguration): Promise<ILocalMcpServerInfo | undefined> {
505 if (!mcpServerConfig.gallery) {
506 return undefined;
525 };
526 }
528 > override canInstall(server: IGalleryMcpServer | IInstallableMcpServer): true | IMarkdownString {
529 throw new Error('Not supported');
530 }
532 >
533 > class WorkspaceMcpManagementService extends AbstractMcpManagementService implements IMcpManagementService {
534 >
535 > private readonly _onInstallMcpServer = this._register(new Emitter<InstallMcpServerEvent>());
536 > readonly onInstallMcpServer = this._onInstallMcpServer.event;
537 >
538 > private readonly _onDidInstallMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
539 > readonly onDidInstallMcpServers = this._onDidInstallMcpServers.event;
540 >
541 > private readonly _onDidUpdateMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
542 > readonly onDidUpdateMcpServers = this._onDidUpdateMcpServers.event;
543 >
544 > private readonly _onUninstallMcpServer = this._register(new Emitter<UninstallMcpServerEvent>());
545 > readonly onUninstallMcpServer = this._onUninstallMcpServer.event;
546 >
547 > private readonly _onDidUninstallMcpServer = this._register(new Emitter<DidUninstallMcpServerEvent>());
548 > readonly onDidUninstallMcpServer = this._onDidUninstallMcpServer.event;
549 >
550 > private allMcpServers: ILocalMcpServer[] = [];
551 >
552 > private workspaceConfiguration?: URI | null;
553 > private readonly workspaceMcpManagementServices = new ResourceMap<{ service: WorkspaceMcpResourceManagementService } & IDisposable>();
554 >
555 > constructor(
556 @IAllowedMcpServersService allowedMcpServersService: IAllowedMcpServersService,
557 @IUriIdentityService private readonly uriIdentityService: IUriIdentityService,
563 this.initialize();
564 }
566 > private async initialize(): Promise<void> {
567 try {
568 await this.onDidChangeWorkbenchState();
574 }
575 }
577 > private async onDidChangeWorkbenchState(): Promise<void> {
578 if (this.workspaceConfiguration) {
579 await this.removeWorkspaceService(this.workspaceConfiguration);
584 }
585 }
587 > private async onDidChangeWorkspaceFolders(e: IWorkspaceFoldersChangeEvent): Promise<void> {
588 try {
589 await Promise.allSettled(e.removed.map(folder => this.removeWorkspaceService(folder.toResource(WORKSPACE_STANDALONE_CONFIGURATIONS[MCP_CONFIGURATION_KEY]))));
597 }
598 }
600 > private async addWorkspaceService(mcpResource: URI, target: McpResourceTarget): Promise<void> {
601 if (this.workspaceMcpManagementServices.has(mcpResource)) {
602 return;
651 this.workspaceMcpManagementServices.set(mcpResource, { service, dispose: () => disposables.dispose() });
652 }
654 > private async removeWorkspaceService(mcpResource: URI): Promise<void> {
655 const serviceItem = this.workspaceMcpManagementServices.get(mcpResource);
656 if (serviceItem) {
671 }
672 }
674 > async getInstalled(): Promise<ILocalMcpServer[]> {
675 return this.allMcpServers;
676 }
678 > async install(server: IInstallableMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
679 if (!options?.mcpResource) {
680 throw new Error('MCP resource is required');
688 return mcpManagementServiceItem.service.install(server, options);
689 }
691 > async uninstall(server: ILocalMcpServer, options?: UninstallOptions): Promise<void> {
692 const mcpResource = server.mcpResource;
693
699 return mcpManagementServiceItem.service.uninstall(server, options);
700 }
702 > installFromGallery(gallery: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
703 if (!options?.mcpResource) {
704 throw new Error('MCP resource is required');
712 return mcpManagementServiceItem.service.installFromGallery(gallery, options);
713 }
715 > updateMetadata(): Promise<ILocalMcpServer> {
716 throw new Error('Not supported');
717 }
719 > override dispose(): void {
720 this.workspaceMcpManagementServices.forEach(service => service.dispose());
721 this.workspaceMcpManagementServices.clear();
722 super.dispose();
723 }
src/vs/workbench/services/filesConfiguration/common/filesConfigurationService.ts 201 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- filesConfigurationService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../../../nls.js';
7 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
8 > import { InstantiationType, registerSingleton } from '../../../../platform/instantiation/common/extensions.js';
9 > import { Event, Emitter } from '../../../../base/common/event.js';
10 > import { Disposable, IDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
11 > import { RawContextKey, IContextKeyService, IContextKey } from '../../../../platform/contextkey/common/contextkey.js';
12 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
13 > import { IFilesConfiguration, AutoSaveConfiguration, HotExitConfiguration, FILES_READONLY_INCLUDE_CONFIG, FILES_READONLY_EXCLUDE_CONFIG, IFileStatWithMetadata, IFileService, IBaseFileStat, hasReadonlyCapability, IFilesConfigurationNode } from '../../../../platform/files/common/files.js';
14 > import { equals } from '../../../../base/common/objects.js';
15 > import { URI } from '../../../../base/common/uri.js';
16 > import { isWeb } from '../../../../base/common/platform.js';
17 > import { IWorkspaceContextService } from '../../../../platform/workspace/common/workspace.js';
18 > import { ResourceGlobMatcher } from '../../../common/resources.js';
19 > import { GlobalIdleValue } from '../../../../base/common/async.js';
20 > import { IUriIdentityService } from '../../../../platform/uriIdentity/common/uriIdentity.js';
21 > import { IEnvironmentService } from '../../../../platform/environment/common/environment.js';
22 > import { LRUCache, ResourceMap } from '../../../../base/common/map.js';
23 > import { IMarkdownString } from '../../../../base/common/htmlContent.js';
24 > import { EditorInput } from '../../../common/editor/editorInput.js';
25 > import { EditorResourceAccessor, SaveReason, SideBySideEditor } from '../../../common/editor.js';
26 > import { IMarkerService, MarkerSeverity } from '../../../../platform/markers/common/markers.js';
27 > import { ITextResourceConfigurationService } from '../../../../editor/common/services/textResourceConfiguration.js';
28 > import { IStringDictionary } from '../../../../base/common/collections.js';
29 >
30 > export const AutoSaveAfterShortDelayContext = new RawContextKey<boolean>('autoSaveAfterShortDelayContext', false, true);
31 >
32 > export interface IAutoSaveConfiguration {
33 > autoSave?: 'afterDelay' | 'onFocusChange' | 'onWindowChange';
34 > autoSaveDelay?: number;
35 > autoSaveWorkspaceFilesOnly?: boolean;
36 > autoSaveWhenNoErrors?: boolean;
37 > }
38 >
39 > interface ICachedAutoSaveConfiguration extends IAutoSaveConfiguration {
40 >
41 > // Some extra state that we cache to reduce the amount
42 > // of lookup we have to do since auto save methods
43 > // are being called very often, e.g. when content changes
44 >
45 > isOutOfWorkspace?: boolean;
46 > isShortAutoSaveDelay?: boolean;
47 > }
48 >
49 > export const enum AutoSaveMode {
50 > OFF,
51 > AFTER_SHORT_DELAY,
52 > AFTER_LONG_DELAY,
53 > ON_FOCUS_CHANGE,
54 > ON_WINDOW_CHANGE
55 > }
56 >
57 > export const enum AutoSaveDisabledReason {
58 > SETTINGS = 1,
59 > OUT_OF_WORKSPACE,
60 > ERRORS,
61 > DISABLED
62 > }
63 >
64 > export type IAutoSaveMode = IEnabledAutoSaveMode | IDisabledAutoSaveMode;
65 >
66 > export interface IEnabledAutoSaveMode {
67 > readonly mode: AutoSaveMode.AFTER_SHORT_DELAY | AutoSaveMode.AFTER_LONG_DELAY | AutoSaveMode.ON_FOCUS_CHANGE | AutoSaveMode.ON_WINDOW_CHANGE;
68 > }
69 >
70 > export interface IDisabledAutoSaveMode {
71 > readonly mode: AutoSaveMode.OFF;
72 > readonly reason: AutoSaveDisabledReason;
73 > }
74 >
75 > export const IFilesConfigurationService = createDecorator<IFilesConfigurationService>('filesConfigurationService');
76 >
77 > export interface IFilesConfigurationService {
78 >
79 > readonly _serviceBrand: undefined;
80 >
81 > //#region Auto Save
82 >
83 > readonly onDidChangeAutoSaveConfiguration: Event<void>;
84 >
85 > readonly onDidChangeAutoSaveDisabled: Event<URI>;
86 >
87 > getAutoSaveConfiguration(resourceOrEditor: EditorInput | URI | undefined): IAutoSaveConfiguration;
88 >
89 > hasShortAutoSaveDelay(resourceOrEditor: EditorInput | URI | undefined): boolean;
90 >
91 > getAutoSaveMode(resourceOrEditor: EditorInput | URI | undefined, saveReason?: SaveReason): IAutoSaveMode;
92 >
93 > toggleAutoSave(): Promise<void>;
94 >
95 > enableAutoSaveAfterShortDelay(resourceOrEditor: EditorInput | URI): IDisposable;
96 > disableAutoSave(resourceOrEditor: EditorInput | URI): IDisposable;
97 >
98 > //#endregion
99 >
100 > //#region Configured Readonly
101 >
102 > readonly onDidChangeReadonly: Event<void>;
103 >
104 > isReadonly(resource: URI, stat?: IBaseFileStat): boolean | IMarkdownString;
105 >
106 > updateReadonly(resource: URI, readonly: true | false | 'toggle' | 'reset'): Promise<void>;
107 > updateReadonly(resource: URI[], readonly: true | false | 'reset'): Promise<void>;
108 >
109 > //#endregion
110 >
111 > readonly onDidChangeFilesAssociation: Event<void>;
112 >
113 > readonly isHotExitEnabled: boolean;
114 >
115 > readonly hotExitConfiguration: string | undefined;
116 >
117 > preventSaveConflicts(resource: URI, language?: string): boolean;
118 > }
119 >
120 > export class FilesConfigurationService extends Disposable implements IFilesConfigurationService {
121 >
122 > declare readonly _serviceBrand: undefined;
123 >
124 > private static readonly DEFAULT_AUTO_SAVE_MODE = isWeb ? AutoSaveConfiguration.AFTER_DELAY : AutoSaveConfiguration.OFF;
125 > private static readonly DEFAULT_AUTO_SAVE_DELAY = 1000;
126 >
127 > private static readonly READONLY_MESSAGES = {
128 > providerReadonly: { value: localize('providerReadonly', "Editor is read-only because the file system of the file is read-only."), isTrusted: true },
129 > sessionReadonly: { value: localize({ key: 'sessionReadonly', comment: ['Please do not translate the word "command", it is part of our internal syntax which must not change', '{Locked="](command:{0})"}'] }, "Editor is read-only because the file was set read-only in this session. [Click here](command:{0}) to set writeable.", 'workbench.action.files.setActiveEditorWriteableInSession'), isTrusted: true },
130 > configuredReadonly: { value: localize({ key: 'configuredReadonly', comment: ['Please do not translate the word "command", it is part of our internal syntax which must not change', '{Locked="](command:{0})"}'] }, "Editor is read-only because the file was set read-only via settings. [Click here](command:{0}) to configure or [toggle for this session](command:{1}).", `workbench.action.openSettings?${encodeURIComponent('["files.readonly"]')}`, 'workbench.action.files.toggleActiveEditorReadonlyInSession'), isTrusted: true },
131 > fileLocked: { value: localize({ key: 'fileLocked', comment: ['Please do not translate the word "command", it is part of our internal syntax which must not change', '{Locked="](command:{0})"}'] }, "Editor is read-only because of file permissions. [Click here](command:{0}) to set writeable anyway.", 'workbench.action.files.setActiveEditorWriteableInSession'), isTrusted: true },
132 > fileReadonly: { value: localize('fileReadonly', "Editor is read-only because the file is read-only."), isTrusted: true }
133 > };
134 >
135 > private readonly _onDidChangeAutoSaveConfiguration = this._register(new Emitter<void>());
136 > readonly onDidChangeAutoSaveConfiguration = this._onDidChangeAutoSaveConfiguration.event;
137 >
138 > private readonly _onDidChangeAutoSaveDisabled = this._register(new Emitter<URI>());
139 > readonly onDidChangeAutoSaveDisabled = this._onDidChangeAutoSaveDisabled.event;
140 >
141 > private readonly _onDidChangeFilesAssociation = this._register(new Emitter<void>());
142 > readonly onDidChangeFilesAssociation = this._onDidChangeFilesAssociation.event;
143 >
144 > private readonly _onDidChangeReadonly = this._register(new Emitter<void>());
145 > readonly onDidChangeReadonly = this._onDidChangeReadonly.event;
146 >
147 > private currentGlobalAutoSaveConfiguration: IAutoSaveConfiguration;
148 > private currentFilesAssociationConfiguration: IStringDictionary<string> | undefined;
149 > private currentHotExitConfiguration: string;
150 >
151 > private readonly autoSaveConfigurationCache = new LRUCache<URI, ICachedAutoSaveConfiguration>(1000);
152 >
153 > private readonly autoSaveAfterShortDelayOverrides = new ResourceMap<number /* counter */>();
154 > private readonly autoSaveDisabledOverrides = new ResourceMap<number /* counter */>();
155 >
156 > private readonly autoSaveAfterShortDelayContext: IContextKey<boolean>;
157 >
158 > private readonly readonlyIncludeMatcher = this._register(new GlobalIdleValue(() => this.createReadonlyMatcher(FILES_READONLY_INCLUDE_CONFIG)));
159 > private readonly readonlyExcludeMatcher = this._register(new GlobalIdleValue(() => this.createReadonlyMatcher(FILES_READONLY_EXCLUDE_CONFIG)));
160 > private configuredReadonlyFromPermissions: boolean | undefined;
161 >
162 > private readonly sessionReadonlyOverrides = new ResourceMap<boolean>(resource => this.uriIdentityService.extUri.getComparisonKey(resource));
163 >
164 > constructor(
165 @IContextKeyService contextKeyService: IContextKeyService,
166 @IConfigurationService private readonly configurationService: IConfigurationService,
186 this.registerListeners();
187 }
189 > private createReadonlyMatcher(config: string) {
190 const matcher = this._register(new ResourceGlobMatcher(
191 resource => this.configurationService.getValue(config, { resource }),
199 return matcher;
200 }
202 > isReadonly(resource: URI, stat?: IBaseFileStat): boolean | IMarkdownString {
203
204 // if the entire file system provider is readonly, we respect that
240 return false;
241 }
243 > async updateReadonly(resource: URI | URI[], readonly: true | false | 'toggle' | 'reset'): Promise<void> {
244 if (Array.isArray(resource)) {
245 for (const r of resource) {
266 this._onDidChangeReadonly.fire();
267 }
269 > private applyReadonly(resource: URI, readonly: true | false | 'reset'): void {
270 if (readonly === 'reset') {
271 this.sessionReadonlyOverrides.delete(resource);
284 }));
285 }
287 > protected onFilesConfigurationChange(configuration: IFilesConfiguration, fromEvent: boolean): void {
288
289 // Auto Save
321 }
322 }
324 > getAutoSaveConfiguration(resourceOrEditor: EditorInput | URI | undefined): ICachedAutoSaveConfiguration {
325 const resource = this.toResource(resourceOrEditor);
326 if (resource) {
336 return this.currentGlobalAutoSaveConfiguration;
337 }
339 > private computeAutoSaveConfiguration(resource: URI | undefined, filesConfiguration: IFilesConfigurationNode | undefined): ICachedAutoSaveConfiguration {
340 let autoSave: 'afterDelay' | 'onFocusChange' | 'onWindowChange' | undefined;
341 let autoSaveDelay: number | undefined;
386 };
387 }
389 > private toResource(resourceOrEditor: EditorInput | URI | undefined): URI | undefined {
390 if (resourceOrEditor instanceof EditorInput) {
391 return EditorResourceAccessor.getOriginalUri(resourceOrEditor, { supportSideBySide: SideBySideEditor.PRIMARY });
394 return resourceOrEditor;
395 }
397 > hasShortAutoSaveDelay(resourceOrEditor: EditorInput | URI | undefined): boolean {
398 const resource = this.toResource(resourceOrEditor);
399
408 return false;
409 }
411 > getAutoSaveMode(resourceOrEditor: EditorInput | URI | undefined, saveReason?: SaveReason): IAutoSaveMode {
412 const resource = this.toResource(resourceOrEditor);
413 if (resource && this.autoSaveAfterShortDelayOverrides.has(resource)) {
461 }
462 }
464 > async toggleAutoSave(): Promise<void> {
465 const currentSetting = this.configurationService.getValue('files.autoSave');
466
474 return this.configurationService.updateValue('files.autoSave', newAutoSaveValue);
475 }
477 > enableAutoSaveAfterShortDelay(resourceOrEditor: EditorInput | URI): IDisposable {
478 const resource = this.toResource(resourceOrEditor);
479 if (!resource) {
493 });
494 }
496 > disableAutoSave(resourceOrEditor: EditorInput | URI): IDisposable {
497 const resource = this.toResource(resourceOrEditor);
498 if (!resource) {
517 });
518 }
520 > get isHotExitEnabled(): boolean {
521 if (this.contextService.getWorkspace().transient) {
522 // Transient workspace: hot exit is disabled because
527 return this.currentHotExitConfiguration !== HotExitConfiguration.OFF;
528 }
530 > get hotExitConfiguration(): string {
531 return this.currentHotExitConfiguration;
532 }
534 > preventSaveConflicts(resource: URI, language?: string): boolean {
535 return this.configurationService.getValue('files.saveConflictResolution', { resource, overrideIdentifier: language }) !== 'overwriteFileOnDisk';
536 }
538 >
539 > registerSingleton(IFilesConfigurationService, FilesConfigurationService, InstantiationType.Eager);
src/vs/platform/telemetry/common/telemetryUtils.ts 199 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- telemetryUtils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { cloneAndChange, safeStringify } from '../../../base/common/objects.js';
7 > import { isObject } from '../../../base/common/types.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { localize } from '../../../nls.js';
10 > import { IConfigurationService } from '../../configuration/common/configuration.js';
11 > import { IEnvironmentService } from '../../environment/common/environment.js';
12 > import { LoggerGroup } from '../../log/common/log.js';
13 > import { IProductService } from '../../product/common/productService.js';
14 > import { getRemoteName } from '../../remote/common/remoteHosts.js';
15 > import { verifyMicrosoftInternalDomain } from './commonProperties.js';
16 > import { ICustomEndpointTelemetryService, ITelemetryData, ITelemetryEndpoint, ITelemetryService, TelemetryConfiguration, TelemetryLevel, TELEMETRY_CRASH_REPORTER_SETTING_ID, TELEMETRY_OLD_SETTING_ID, TELEMETRY_SETTING_ID } from './telemetry.js';
17 >
18 > /**
19 > * A special class used to denoting a telemetry value which should not be clean.
20 > * This is because that value is "Trusted" not to contain identifiable information such as paths.
21 > * NOTE: This is used as an API type as well, and should not be changed.
22 > */
23 > export class TelemetryTrustedValue<T> {
24 > // This is merely used as an identifier as the instance will be lost during serialization over the exthost
25 > public readonly isTrustedTelemetryValue = true;
26 > constructor(public readonly value: T) { }
27 > }
28 >
29 > export class NullTelemetryServiceShape implements ITelemetryService {
30 > declare readonly _serviceBrand: undefined;
31 > readonly telemetryLevel = TelemetryLevel.NONE;
32 > readonly sessionId = 'someValue.sessionId';
33 > readonly machineId = 'someValue.machineId';
34 > readonly sqmId = 'someValue.sqmId';
35 > readonly devDeviceId = 'someValue.devDeviceId';
36 > readonly firstSessionDate = 'someValue.firstSessionDate';
37 > readonly sendErrorTelemetry = false;
38 > publicLog() { }
39 > publicLog2() { }
40 > publicLogError() { }
41 > publicLogError2() { }
42 > setExperimentProperty() { }
43 > setCommonProperty() { }
44 > }
45 >
46 > export const NullTelemetryService = new NullTelemetryServiceShape();
47 >
48 > export class NullEndpointTelemetryService implements ICustomEndpointTelemetryService {
49 > _serviceBrand: undefined;
50 >
51 > async publicLog(_endpoint: ITelemetryEndpoint, _eventName: string, _data?: ITelemetryData): Promise<void> {
52 // noop
53 }
55 > async publicLogError(_endpoint: ITelemetryEndpoint, _errorEventName: string, _data?: ITelemetryData): Promise<void> {
56 // noop
57 }
59 >
60 > export const telemetryLogId = 'telemetry';
61 > export const TelemetryLogGroup: LoggerGroup = { id: telemetryLogId, name: localize('telemetryLogName', "Telemetry") };
62 >
63 > export interface ITelemetryAppender {
64 > log(eventName: string, data: ITelemetryData): void;
65 > flush(): Promise<void>;
66 > }
67 >
68 > export const NullAppender: ITelemetryAppender = { log: () => null, flush: () => Promise.resolve(undefined) };
69 >
70 >
71 > /* __GDPR__FRAGMENT__
72 > "URIDescriptor" : {
73 > "mimeType" : { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
74 > "scheme": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
75 > "ext": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
76 > "path": { "classification": "SystemMetaData", "purpose": "FeatureInsight" }
77 > }
78 > */
79 > export interface URIDescriptor {
80 > mimeType?: string;
81 > scheme?: string;
82 > ext?: string;
83 > path?: string;
84 > }
85 >
86 > /**
87 > * Determines whether or not we support logging telemetry.
88 > * This checks if the product is capable of collecting telemetry but not whether or not it can send it
89 > * For checking the user setting and what telemetry you can send please check `getTelemetryLevel`.
90 > * This returns true if `--disable-telemetry` wasn't used, the product.json allows for telemetry, and we're not testing an extension
91 > * If false telemetry is disabled throughout the product
92 > * @param productService
93 > * @param environmentService
94 > * @returns false - telemetry is completely disabled, true - telemetry is logged locally, but may not be sent
95 > */
96 > export function supportsTelemetry(productService: IProductService, environmentService: IEnvironmentService): boolean {
97 // If it's OSS and telemetry isn't disabled via the CLI we will allow it for logging only purposes
98 if (!environmentService.isBuilt && !environmentService.disableTelemetry) {
101 return !(environmentService.disableTelemetry || !productService.enableTelemetry);
102 }
104 > /**
105 > * Checks to see if we're in logging only mode to debug telemetry.
106 > * This is if telemetry is enabled and we're in OSS, but no telemetry key is provided so it's not being sent just logged.
107 > * @param productService
108 > * @param environmentService
109 > * @returns True if telemetry is actually disabled and we're only logging for debug purposes
110 > */
111 > export function isLoggingOnly(productService: IProductService, environmentService: IEnvironmentService): boolean {
112 // If we're testing an extension, log telemetry for debug purposes
113 if (environmentService.extensionTestsLocationURI) {
129 return true;
130 }
132 > /**
133 > * Determines how telemetry is handled based on the user's configuration.
134 > *
135 > * @param configurationService
136 > * @returns OFF, ERROR, ON
137 > */
138 > export function getTelemetryLevel(configurationService: IConfigurationService): TelemetryLevel {
139 const newConfig = configurationService.getValue<TelemetryConfiguration>(TELEMETRY_SETTING_ID);
140 const crashReporterConfig = configurationService.getValue<boolean | undefined>(TELEMETRY_CRASH_REPORTER_SETTING_ID);
158 }
159 }
161 > export interface Properties {
162 > [key: string]: string;
163 > }
164 >
165 > export interface Measurements {
166 > [key: string]: number;
167 > }
168 >
169 > export function validateTelemetryData(data?: unknown): { properties: Properties; measurements: Measurements } {
170
171 const properties: Properties = {};
204 };
205 }
207 > interface IRemoteAuthoringConfig {
208 > remoteExtensionTips?: { readonly [remoteName: string]: unknown };
209 > virtualWorkspaceExtensionTips?: { readonly [remoteName: string]: unknown };
210 > }
211 >
212 > export function cleanRemoteAuthority(remoteAuthority: string | undefined, config: IRemoteAuthoringConfig): string {
213 if (!remoteAuthority) {
214 return 'none';
229 return 'other';
230 }
232 function flatten(obj: unknown, result: Record<string, unknown>, order: number = 0, prefix?: string): void {
233 if (!obj || (typeof obj !== 'object' && typeof obj !== 'function')) {
258 }
259 }
261 > /**
262 > * Whether or not this is an internal user
263 > * @param productService The product service
264 > * @param configService The config servivce
265 > * @returns true if internal, false otherwise
266 > */
267 > export function isInternalTelemetry(productService: IProductService, configService: IConfigurationService) {
268 const msftInternalDomains = productService.msftInternalDomains || [];
269 const internalTesting = configService.getValue<boolean>('telemetry.internalTesting');
270 return verifyMicrosoftInternalDomain(msftInternalDomains) || internalTesting;
271 }
273 > interface IPathEnvironment {
274 > appRoot: string;
275 > extensionsPath: string;
276 > userDataPath: string;
277 > userHome: URI;
278 > tmpDir: URI;
279 > }
280 >
281 > export function getPiiPathsFromEnvironment(paths: IPathEnvironment): string[] {
282 return [paths.appRoot, paths.extensionsPath, paths.userHome.fsPath, paths.tmpDir.fsPath, paths.userDataPath];
283 }
285 > //#region Telemetry Cleaning
286 >
287 > /**
288 > * Cleans a given stack of possible paths
289 > * @param stack The stack to sanitize
290 > * @param cleanupPatterns Cleanup patterns to remove from the stack
291 > * @returns The cleaned stack
292 > */
293 function anonymizeFilePaths(stack: string, cleanupPatterns: RegExp[]): string {
294
356 return updatedStack;
357 }
359 > const userDataRegexes = [
360 > { label: 'URL', regex: /[a-zA-Z][a-zA-Z0-9+.-]*:\/\/[^\s]*/ },
361 > { label: 'Google API Key', regex: /AIza[A-Za-z0-9_\\\-]{35}/ },
362 > { label: 'JWT', regex: /eyJ[0eXAiOiJKV1Qi|hbGci|a-zA-Z0-9\-_]+\.[a-zA-Z0-9\-_]+\.[a-zA-Z0-9\-_]+/ },
363 > { label: 'Slack Token', regex: /xox[pbar]\-[A-Za-z0-9]/ },
364 > { label: 'GitHub Token', regex: /(gh[psuro]_[a-zA-Z0-9]{36}|github_pat_[a-zA-Z0-9]{22}_[a-zA-Z0-9]{59})/ },
365 > { label: 'Generic Secret', regex: /(key|token|sig|secret|signature|password|passwd|pwd|android:value)[^a-zA-Z0-9]/i },
366 > { label: 'CLI Credentials', regex: /((login|psexec|(certutil|psexec)\.exe).{1,50}(\s-u(ser(name)?)?\s+.{3,100})?\s-(admin|user|vm|root)?p(ass(word)?)?\s+["']?[^$\-\/\s]|(^|[\s\r\n\\])net(\.exe)?.{1,5}(user\s+|share\s+\/user:| user -? secrets ? set) \s + [^ $\s \/])/ },
367 > { label: 'Microsoft Entra ID', regex: /eyJ(?:0eXAiOiJKV1Qi|hbGci|[a-zA-Z0-9\-_]+\.[a-zA-Z0-9\-_]+\.)/ },
368 > { label: 'Email', regex: /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/ }
369 > ];
370 >
371 > /**
372 > * Redacts a value if it contains commonly leaked PII.
373 > * @param value The value returned (as-is) when no PII is detected
374 > * @param probe The string actually matched against the PII heuristics. Defaults
375 > * to `value`; callers may pass a value that includes a trailing delimiter (e.g. a
376 > * newline) so that heuristics relying on a non-alphanumeric boundary match the
377 > * same way they would against the original whole string.
378 > * @returns A `<REDACTED: ...>` marker if the probe matched, otherwise `value`
379 > */
380 function redactIfPossibleUserInfo(value: string, probe: string = value): string {
381 for (const secretRegex of userDataRegexes) {
386 return value;
387 }
389 > /**
390 > * Attempts to remove commonly leaked PII.
391 > *
392 > * When a match is found the check is applied per line so that a single suspicious
393 > * frame (e.g. a stack frame containing a function name such as `getStorageKey`
394 > * which matches the broad `Generic Secret` heuristic) only redacts that line —
395 > * replacing it with a `<REDACTED: ...>` marker — instead of wiping the entire
396 > * multi-line value such as a whole callstack.
397 > * @param property The property whose offending lines will be replaced with a redaction marker if they contain user data
398 > * @returns The new value for the property
399 > */
400 function removePropertiesWithPossibleUserInfo(property: string): string {
401 // If for some reason it is undefined we skip it (this shouldn't be possible);
436 return lines.join('\n');
437 }
439 >
440 > /**
441 > * Does a best possible effort to clean a data object from any possible PII.
442 > * @param data The data object to clean
443 > * @param paths Any additional patterns that should be removed from the data set
444 > * @returns A new object with the PII removed
445 > */
446 > export function cleanData(data: ITelemetryData | undefined, cleanUpPatterns: RegExp[]): Record<string, unknown> {
447 if (!data) {
448 return {};
src/vs/platform/mcp/common/mcpManagementService.ts 198 covered LOC · 36 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpManagementService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { RunOnceScheduler } from '../../../base/common/async.js';
7 > import { VSBuffer } from '../../../base/common/buffer.js';
8 > import { CancellationToken } from '../../../base/common/cancellation.js';
9 > import { Emitter, Event } from '../../../base/common/event.js';
10 > import { IMarkdownString, MarkdownString } from '../../../base/common/htmlContent.js';
11 > import { Disposable, DisposableStore, IDisposable } from '../../../base/common/lifecycle.js';
12 > import { ResourceMap } from '../../../base/common/map.js';
13 > import { equals } from '../../../base/common/objects.js';
14 > import { isString } from '../../../base/common/types.js';
15 > import { URI } from '../../../base/common/uri.js';
16 > import { localize } from '../../../nls.js';
17 > import { ConfigurationTarget } from '../../configuration/common/configuration.js';
18 > import { IEnvironmentService } from '../../environment/common/environment.js';
19 > import { IFileService } from '../../files/common/files.js';
20 > import { IInstantiationService } from '../../instantiation/common/instantiation.js';
21 > import { ILogService } from '../../log/common/log.js';
22 > import { IUriIdentityService } from '../../uriIdentity/common/uriIdentity.js';
23 > import { IUserDataProfilesService } from '../../userDataProfile/common/userDataProfile.js';
24 > import { DidUninstallMcpServerEvent, IGalleryMcpServer, ILocalMcpServer, IMcpGalleryService, IMcpManagementService, IMcpServerInput, IGalleryMcpServerConfiguration, InstallMcpServerEvent, InstallMcpServerResult, RegistryType, UninstallMcpServerEvent, InstallOptions, UninstallOptions, IInstallableMcpServer, IAllowedMcpServersService, IMcpServerArgument, IMcpServerKeyValueInput, McpServerConfigurationParseResult } from './mcpManagement.js';
25 > import { IMcpSandboxConfiguration, IMcpServerVariable, McpServerVariableType, IMcpServerConfiguration, McpServerType } from './mcpPlatformTypes.js';
26 > import { IMcpResourceScannerService, McpResourceTarget } from './mcpResourceScannerService.js';
27 >
28 > export interface ILocalMcpServerInfo {
29 > name: string;
30 > version?: string;
31 > displayName?: string;
32 > galleryId?: string;
33 > galleryUrl?: string;
34 > description?: string;
35 > repositoryUrl?: string;
36 > publisher?: string;
37 > publisherDisplayName?: string;
38 > icon?: {
39 > dark: string;
40 > light: string;
41 > };
42 > codicon?: string;
43 > manifest?: IGalleryMcpServerConfiguration;
44 > readmeUrl?: URI;
45 > location?: URI;
46 > licenseUrl?: string;
47 > }
48 >
49 > export abstract class AbstractCommonMcpManagementService extends Disposable implements IMcpManagementService {
50 >
51 > _serviceBrand: undefined;
52 >
53 > abstract onInstallMcpServer: Event<InstallMcpServerEvent>;
54 > abstract onDidInstallMcpServers: Event<readonly InstallMcpServerResult[]>;
55 > abstract onDidUpdateMcpServers: Event<readonly InstallMcpServerResult[]>;
56 > abstract onUninstallMcpServer: Event<UninstallMcpServerEvent>;
57 > abstract onDidUninstallMcpServer: Event<DidUninstallMcpServerEvent>;
58 >
59 > abstract getInstalled(mcpResource?: URI): Promise<ILocalMcpServer[]>;
60 > abstract install(server: IInstallableMcpServer, options?: InstallOptions): Promise<ILocalMcpServer>;
61 > abstract installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer>;
62 > abstract updateMetadata(local: ILocalMcpServer, server: IGalleryMcpServer, profileLocation?: URI): Promise<ILocalMcpServer>;
63 > abstract uninstall(server: ILocalMcpServer, options?: UninstallOptions): Promise<void>;
64 > abstract canInstall(server: IGalleryMcpServer | IInstallableMcpServer): true | IMarkdownString;
65 >
66 > constructor(
67 @ILogService protected readonly logService: ILogService
68 ) {
69 super();
70 }
72 > getMcpServerConfigurationFromManifest(manifest: IGalleryMcpServerConfiguration, packageType: RegistryType): McpServerConfigurationParseResult {
73
74 // remote
180 };
181 }
183 > protected getCommandName(packageType: RegistryType): string {
184 switch (packageType) {
185 case RegistryType.NODE: return 'npx';
190 return packageType;
191 }
193 > protected getVariables(variableInputs: Record<string, IMcpServerInput>): IMcpServerVariable[] {
194 const variables: IMcpServerVariable[] = [];
195 for (const [key, value] of Object.entries(variableInputs)) {
205 return variables;
206 }
208 > private processKeyValueInputs(keyValueInputs: ReadonlyArray<IMcpServerKeyValueInput>): { inputs: Record<string, string>; variables: IMcpServerVariable[]; notices: string[] } {
209 const notices: string[] = [];
210 const inputs: Record<string, string> = {};
239 return { inputs, variables, notices };
240 }
242 > private processArguments(argumentsList: readonly IMcpServerArgument[]): { args: string[]; variables: IMcpServerVariable[]; notices: string[] } {
243 const args: string[] = [];
244 const variables: IMcpServerVariable[] = [];
302 return { args, variables, notices };
303 }
305 > }
306 >
307 > export abstract class AbstractMcpResourceManagementService extends AbstractCommonMcpManagementService {
308 >
309 > private initializePromise: Promise<void> | undefined;
310 > private readonly reloadConfigurationScheduler: RunOnceScheduler;
311 > private local = new Map<string, ILocalMcpServer>();
312 >
313 > protected readonly _onInstallMcpServer = this._register(new Emitter<InstallMcpServerEvent>());
314 > readonly onInstallMcpServer = this._onInstallMcpServer.event;
315 >
316 > protected readonly _onDidInstallMcpServers = this._register(new Emitter<InstallMcpServerResult[]>());
317 > get onDidInstallMcpServers() { return this._onDidInstallMcpServers.event; }
318 >
319 > protected readonly _onDidUpdateMcpServers = this._register(new Emitter<InstallMcpServerResult[]>());
320 > get onDidUpdateMcpServers() { return this._onDidUpdateMcpServers.event; }
321 >
322 > protected readonly _onUninstallMcpServer = this._register(new Emitter<UninstallMcpServerEvent>());
323 > get onUninstallMcpServer() { return this._onUninstallMcpServer.event; }
324 >
325 > protected _onDidUninstallMcpServer = this._register(new Emitter<DidUninstallMcpServerEvent>());
326 > get onDidUninstallMcpServer() { return this._onDidUninstallMcpServer.event; }
327 >
328 > constructor(
329 protected readonly mcpResource: URI,
330 protected readonly target: McpResourceTarget,
339 this.reloadConfigurationScheduler = this._register(new RunOnceScheduler(() => this.updateLocal(), 50));
340 }
342 > /**
343 > * Enforces the enterprise allow/deny policy at the point of persistence. Called by every
344 > * install path (installable and each gallery override) against the fully resolved server
345 > * configuration, so a caller that goes straight to the management API cannot bypass the
346 > * `canInstall` UI check, and a gallery entry cannot slip through if its resolved command/URL
347 > * differs from the pre-resolution metadata.
348 > */
349 > protected ensureServerAllowed(server: IGalleryMcpServer | IInstallableMcpServer): void {
350 const result = this.allowedMcpServersService.isAllowed(server);
351 if (result !== true) {
353 }
354 }
356 > private initialize(): Promise<void> {
357 if (!this.initializePromise) {
358 this.initializePromise = (async () => {
366 return this.initializePromise;
367 }
369 > private async populateLocalServers(): Promise<Map<string, ILocalMcpServer>> {
370 this.logService.trace('AbstractMcpResourceManagementService#populateLocalServers', this.mcpResource.toString());
371 const local = new Map<string, ILocalMcpServer>();
384 return local;
385 }
387 > private startWatching(): void {
388 this._register(this.fileService.watch(this.mcpResource));
389 this._register(this.fileService.onDidFilesChange(e => {
393 }));
394 }
396 > protected async updateLocal(): Promise<void> {
397 try {
398 const current = await this.populateLocalServers();
436 }
437 }
439 > async getInstalled(): Promise<ILocalMcpServer[]> {
440 await this.initialize();
441 return Array.from(this.local.values());
442 }
444 > protected async scanLocalServer(name: string, config: IMcpServerConfiguration, rootSandbox?: IMcpSandboxConfiguration): Promise<ILocalMcpServer> {
445 let mcpServerInfo = await this.getLocalServerInfo(name, config);
446 if (!mcpServerInfo) {
469 };
470 }
472 > async install(server: IInstallableMcpServer, options?: Omit<InstallOptions, 'mcpResource'>): Promise<ILocalMcpServer> {
473 this.logService.trace('MCP Management Service: install', server.name);
474 this.ensureServerAllowed(server);
488 }
489 }
491 > async uninstall(server: ILocalMcpServer, options?: Omit<UninstallOptions, 'mcpResource'>): Promise<void> {
492 this.logService.trace('MCP Management Service: uninstall', server.name);
493 this._onUninstallMcpServer.fire({ name: server.name, mcpResource: this.mcpResource });
508 }
509 }
511 > protected abstract getLocalServerInfo(name: string, mcpServerConfig: IMcpServerConfiguration): Promise<ILocalMcpServerInfo | undefined>;
512 > protected abstract installFromUri(uri: URI, options?: Omit<InstallOptions, 'mcpResource'>): Promise<ILocalMcpServer>;
513 > }
514 >
515 > export class McpUserResourceManagementService extends AbstractMcpResourceManagementService {
516 >
517 > protected readonly mcpLocation: URI;
518 >
519 > constructor(
520 mcpResource: URI,
521 @IMcpGalleryService mcpGalleryService: IMcpGalleryService,
530 this.mcpLocation = uriIdentityService.extUri.joinPath(environmentService.userRoamingDataHome, 'mcp');
531 }
533 > async installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
534 throw new Error('Not supported');
535 }
537 > async updateMetadata(local: ILocalMcpServer, gallery: IGalleryMcpServer): Promise<ILocalMcpServer> {
538 await this.updateMetadataFromGallery(gallery);
539 await this.updateLocal();
544 return updatedLocal;
545 }
547 > protected async updateMetadataFromGallery(gallery: IGalleryMcpServer): Promise<IGalleryMcpServerConfiguration> {
548 const manifest = gallery.configuration;
549 const location = this.getLocation(gallery.name, gallery.version);
573 return manifest;
574 }
576 > protected async getLocalServerInfo(name: string, mcpServerConfig: IMcpServerConfiguration): Promise<ILocalMcpServerInfo | undefined> {
577 let storedMcpServerInfo: ILocalMcpServerInfo | undefined;
578 let location: URI | undefined;
603 return storedMcpServerInfo;
604 }
606 > protected getLocation(name: string, version?: string): URI {
607 name = name.replace('/', '.');
608 return this.uriIdentityService.extUri.joinPath(this.mcpLocation, version ? `${name}-${version}` : name);
609 }
611 > protected override installFromUri(uri: URI, options?: Omit<InstallOptions, 'mcpResource'>): Promise<ILocalMcpServer> {
612 throw new Error('Method not supported.');
613 }
615 > override canInstall(): true | IMarkdownString {
616 throw new Error('Not supported');
617 }
619 > }
620 >
621 > export abstract class AbstractMcpManagementService extends AbstractCommonMcpManagementService implements IMcpManagementService {
622 >
623 > constructor(
624 @IAllowedMcpServersService protected readonly allowedMcpServersService: IAllowedMcpServersService,
625 @ILogService logService: ILogService,
627 super(logService);
628 }
630 > canInstall(server: IGalleryMcpServer | IInstallableMcpServer): true | IMarkdownString {
631 const allowedToInstall = this.allowedMcpServersService.isAllowed(server);
632 if (allowedToInstall !== true) {
635 return true;
636 }
638 >
639 > export class McpManagementService extends AbstractMcpManagementService implements IMcpManagementService {
640 >
641 > private readonly _onInstallMcpServer = this._register(new Emitter<InstallMcpServerEvent>());
642 > readonly onInstallMcpServer = this._onInstallMcpServer.event;
643 >
644 > private readonly _onDidInstallMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
645 > readonly onDidInstallMcpServers = this._onDidInstallMcpServers.event;
646 >
647 > private readonly _onDidUpdateMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
648 > readonly onDidUpdateMcpServers = this._onDidUpdateMcpServers.event;
649 >
650 > private readonly _onUninstallMcpServer = this._register(new Emitter<UninstallMcpServerEvent>());
651 > readonly onUninstallMcpServer = this._onUninstallMcpServer.event;
652 >
653 > private readonly _onDidUninstallMcpServer = this._register(new Emitter<DidUninstallMcpServerEvent>());
654 > readonly onDidUninstallMcpServer = this._onDidUninstallMcpServer.event;
655 >
656 > private readonly mcpResourceManagementServices = new ResourceMap<{ service: McpUserResourceManagementService } & IDisposable>();
657 >
658 > constructor(
659 @IAllowedMcpServersService allowedMcpServersService: IAllowedMcpServersService,
660 @ILogService logService: ILogService,
664 super(allowedMcpServersService, logService);
665 }
667 > private getMcpResourceManagementService(mcpResource: URI): McpUserResourceManagementService {
668 let mcpResourceManagementService = this.mcpResourceManagementServices.get(mcpResource);
669 if (!mcpResourceManagementService) {
679 return mcpResourceManagementService.service;
680 }
682 > async getInstalled(mcpResource?: URI): Promise<ILocalMcpServer[]> {
683 const mcpResourceUri = mcpResource || this.userDataProfilesService.defaultProfile.mcpResource;
684 return this.getMcpResourceManagementService(mcpResourceUri).getInstalled();
685 }
687 > async install(server: IInstallableMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
688 const mcpResourceUri = options?.mcpResource || this.userDataProfilesService.defaultProfile.mcpResource;
689 return this.getMcpResourceManagementService(mcpResourceUri).install(server, options);
690 }
692 > async uninstall(server: ILocalMcpServer, options?: UninstallOptions): Promise<void> {
693 const mcpResourceUri = options?.mcpResource || this.userDataProfilesService.defaultProfile.mcpResource;
694 return this.getMcpResourceManagementService(mcpResourceUri).uninstall(server, options);
695 }
697 > async installFromGallery(server: IGalleryMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
698 const mcpResourceUri = options?.mcpResource || this.userDataProfilesService.defaultProfile.mcpResource;
699 return this.getMcpResourceManagementService(mcpResourceUri).installFromGallery(server, options);
700 }
702 > async updateMetadata(local: ILocalMcpServer, gallery: IGalleryMcpServer, mcpResource?: URI): Promise<ILocalMcpServer> {
703 return this.getMcpResourceManagementService(mcpResource || this.userDataProfilesService.defaultProfile.mcpResource).updateMetadata(local, gallery);
704 }
706 > override dispose(): void {
707 this.mcpResourceManagementServices.forEach(service => service.dispose());
708 this.mcpResourceManagementServices.clear();
709 super.dispose();
710 }
712 > protected createMcpResourceManagementService(mcpResource: URI): McpUserResourceManagementService {
713 return this.instantiationService.createInstance(McpUserResourceManagementService, mcpResource);
714 }
716 > }
src/vs/base/common/jsonSchema.ts 194 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- jsonSchema.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export type JSONSchemaType = 'string' | 'number' | 'integer' | 'boolean' | 'null' | 'array' | 'object';
7 >
8 > export interface IJSONSchema {
9 > id?: string;
10 > $id?: string;
11 > $schema?: string;
12 > type?: JSONSchemaType | JSONSchemaType[];
13 > title?: string;
14 > default?: any;
15 > definitions?: IJSONSchemaMap;
16 > description?: string;
17 > properties?: IJSONSchemaMap;
18 > patternProperties?: IJSONSchemaMap;
19 > additionalProperties?: boolean | IJSONSchema;
20 > minProperties?: number;
21 > maxProperties?: number;
22 > dependencies?: IJSONSchemaMap | { [prop: string]: string[] };
23 > items?: IJSONSchema | IJSONSchema[];
24 > minItems?: number;
25 > maxItems?: number;
26 > uniqueItems?: boolean;
27 > additionalItems?: boolean | IJSONSchema;
28 > pattern?: string;
29 > minLength?: number;
30 > maxLength?: number;
31 > minimum?: number;
32 > maximum?: number;
33 > exclusiveMinimum?: boolean | number;
34 > exclusiveMaximum?: boolean | number;
35 > multipleOf?: number;
36 > required?: string[];
37 > $ref?: string;
38 > anyOf?: IJSONSchema[];
39 > allOf?: IJSONSchema[];
40 > oneOf?: IJSONSchema[];
41 > not?: IJSONSchema;
42 > enum?: any[];
43 > format?: string;
44 >
45 > // schema draft 06
46 > const?: any;
47 > contains?: IJSONSchema;
48 > propertyNames?: IJSONSchema;
49 > examples?: any[];
50 >
51 > // schema draft 07
52 > $comment?: string;
53 > if?: IJSONSchema;
54 > then?: IJSONSchema;
55 > else?: IJSONSchema;
56 >
57 > // schema 2019-09
58 > unevaluatedProperties?: boolean | IJSONSchema;
59 > unevaluatedItems?: boolean | IJSONSchema;
60 > minContains?: number;
61 > maxContains?: number;
62 > deprecated?: boolean;
63 > dependentRequired?: { [prop: string]: string[] };
64 > dependentSchemas?: IJSONSchemaMap;
65 > $defs?: { [name: string]: IJSONSchema };
66 > $anchor?: string;
67 > $recursiveRef?: string;
68 > $recursiveAnchor?: string;
69 > $vocabulary?: any;
70 >
71 > // schema 2020-12
72 > prefixItems?: IJSONSchema[];
73 > $dynamicRef?: string;
74 > $dynamicAnchor?: string;
75 >
76 > // VSCode extensions
77 >
78 > defaultSnippets?: IJSONSchemaSnippet[];
79 > errorMessage?: string;
80 > patternErrorMessage?: string;
81 > deprecationMessage?: string;
82 > markdownDeprecationMessage?: string;
83 > enumDescriptions?: string[];
84 > markdownEnumDescriptions?: string[];
85 > markdownDescription?: string;
86 > doNotSuggest?: boolean;
87 > suggestSortText?: string;
88 > allowComments?: boolean;
89 > allowTrailingCommas?: boolean;
90 > secret?: boolean;
91 > }
92 >
93 > export interface IJSONSchemaMap {
94 > [name: string]: IJSONSchema;
95 > }
96 >
97 > export interface IJSONSchemaSnippet {
98 > label?: string;
99 > description?: string;
100 > body?: any; // a object that will be JSON stringified
101 > bodyText?: string; // an already stringified JSON object that can contain new lines (\n) and tabs (\t)
102 > }
103 >
104 > /**
105 > * Converts a basic JSON schema to a TypeScript type.
106 > */
107 > export type TypeFromJsonSchema<T> =
108 > // enum
109 > T extends { enum: infer EnumValues }
110 > ? UnionOf<EnumValues>
111 >
112 > // Object with list of required properties.
113 > // Values are required or optional based on `required` list.
114 > : T extends { type: 'object'; properties: infer P; required: infer RequiredList }
115 > ? {
116 > [K in keyof P]: IsRequired<K, RequiredList> extends true ? TypeFromJsonSchema<P[K]> : TypeFromJsonSchema<P[K]> | undefined;
117 > } & AdditionalPropertiesType<T>
118 >
119 > // Object with no required properties.
120 > // All values are optional
121 > : T extends { type: 'object'; properties: infer P }
122 > ? { [K in keyof P]: TypeFromJsonSchema<P[K]> | undefined } & AdditionalPropertiesType<T>
123 >
124 > // Array
125 > : T extends { type: 'array'; items: infer Items }
126 > ? Items extends [...infer R]
127 > // If items is an array, we treat it like a tuple
128 > ? { [K in keyof R]: TypeFromJsonSchema<Items[K]> }
129 > : Array<TypeFromJsonSchema<Items>>
130 >
131 > // oneOf / anyof
132 > // These are handled the same way as they both represent a union type.
133 > // However at the validation level, they have different semantics.
134 > : T extends { oneOf: infer I }
135 > ? MapSchemaToType<I>
136 > : T extends { anyOf: infer I }
137 > ? MapSchemaToType<I>
138 >
139 > // Primitive types
140 > : T extends { type: infer Type }
141 > // Basic type
142 > ? Type extends 'string' | 'number' | 'integer' | 'boolean' | 'null'
143 > ? SchemaPrimitiveTypeNameToType<Type>
144 > // Union of primitive types
145 > : Type extends [...infer R]
146 > ? UnionOf<{ [K in keyof R]: SchemaPrimitiveTypeNameToType<R[K]> }>
147 > : never
148 >
149 > // Fallthrough
150 > : never;
151 >
152 > type SchemaPrimitiveTypeNameToType<T> =
153 > T extends 'string' ? string :
154 > T extends 'number' | 'integer' ? number :
155 > T extends 'boolean' ? boolean :
156 > T extends 'null' ? null :
157 > never;
158 >
159 > type UnionOf<T> =
160 > T extends [infer First, ...infer Rest]
161 > ? First | UnionOf<Rest>
162 > : never;
163 >
164 > type IsRequired<K, RequiredList> =
165 > RequiredList extends []
166 > ? false
167 >
168 > : RequiredList extends [K, ...infer _]
169 > ? true
170 >
171 > : RequiredList extends [infer _, ...infer R]
172 > ? IsRequired<K, R>
173 >
174 > : false;
175 >
176 > type AdditionalPropertiesType<Schema> =
177 > Schema extends { additionalProperties: infer AP }
178 > ? AP extends false ? {} : { [key: string]: TypeFromJsonSchema<Schema['additionalProperties']> }
179 > : {};
180 >
181 > type MapSchemaToType<T> = T extends [infer First, ...infer Rest]
182 > ? TypeFromJsonSchema<First> | MapSchemaToType<Rest>
183 > : never;
184 >
185 > interface Equals { schemas: IJSONSchema[]; id?: string }
186 >
187 > export function getCompressedContent(schema: IJSONSchema): string {
188 let hasDups = false;
189
258 return str;
259 }
261 > type IJSONSchemaRef = IJSONSchema | boolean;
262 >
263 function isObject(thing: unknown): thing is object {
264 return typeof thing === 'object' && thing !== null;
265 }
267 > /*
268 > * Traverse a JSON schema and visit each schema node
269 > */
270 function traverseNodes(root: IJSONSchema, visit: (schema: IJSONSchema) => boolean) {
271 if (!root || typeof root !== 'object') {
src/vs/base/common/platform.ts 189 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- platform.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as nls from '../../nls.js';
7 >
8 > export const LANGUAGE_DEFAULT = 'en';
9 >
10 > let _isWindows = false;
11 > let _isMacintosh = false;
12 > let _isLinux = false;
13 > let _isLinuxSnap = false;
14 > let _isNative = false;
15 > let _isWeb = false;
16 > let _isElectron = false;
17 > let _isIOS = false;
18 > let _isCI = false;
19 > let _isMobile = false;
20 > let _locale: string | undefined = undefined;
21 > let _language: string = LANGUAGE_DEFAULT;
22 > let _platformLocale: string = LANGUAGE_DEFAULT;
23 > let _translationsConfigFile: string | undefined = undefined;
24 > let _userAgent: string | undefined = undefined;
25 >
26 > export interface IProcessEnvironment {
27 > [key: string]: string | undefined;
28 > }
29 >
30 > /**
31 > * This interface is intentionally not identical to node.js
32 > * process because it also works in sandboxed environments
33 > * where the process object is implemented differently. We
34 > * define the properties here that we need for `platform`
35 > * to work and nothing else.
36 > */
37 > export interface INodeProcess {
38 > platform: string;
39 > arch: string;
40 > env: IProcessEnvironment;
41 > versions?: {
42 > node?: string;
43 > electron?: string;
44 > chrome?: string;
45 > };
46 > type?: string;
47 > cwd: () => string;
48 > }
49 >
50 > declare const process: INodeProcess;
51 >
52 > const $globalThis: any = globalThis;
53 >
54 > let nodeProcess: INodeProcess | undefined = undefined;
55 > if (typeof $globalThis.vscode !== 'undefined' && typeof $globalThis.vscode.process !== 'undefined') {
56 // Native environment (sandboxed)
57 nodeProcess = $globalThis.vscode.process;
58 > } else if (typeof process !== 'undefined' && typeof process?.versions?.node === 'string') { platform.ts
59 > // Native environment (non-sandboxed)
60 > nodeProcess = process;
61 > }
62 >
63 > const isElectronProcess = typeof nodeProcess?.versions?.electron === 'string';
64 > const isElectronRenderer = isElectronProcess && nodeProcess?.type === 'renderer';
65 >
66 > interface INavigator {
67 > userAgent: string;
68 > maxTouchPoints?: number;
69 > language: string;
70 > }
71 > declare const navigator: INavigator;
72 >
73 > // Native environment
74 > if (typeof nodeProcess === 'object') {
75 > _isWindows = (nodeProcess.platform === 'win32');
76 > _isMacintosh = (nodeProcess.platform === 'darwin');
77 > _isLinux = (nodeProcess.platform === 'linux');
78 > _isLinuxSnap = _isLinux && !!nodeProcess.env['SNAP'] && !!nodeProcess.env['SNAP_REVISION'];
79 > _isElectron = isElectronProcess;
80 > _isCI = !!nodeProcess.env['CI'] || !!nodeProcess.env['BUILD_ARTIFACTSTAGINGDIRECTORY'] || !!nodeProcess.env['GITHUB_WORKSPACE'];
81 > _locale = LANGUAGE_DEFAULT;
82 > _language = LANGUAGE_DEFAULT;
83 > const rawNlsConfig = nodeProcess.env['VSCODE_NLS_CONFIG'];
84 > if (rawNlsConfig) {
85 try {
86 const nlsConfig: nls.INLSConfiguration = JSON.parse(rawNlsConfig);
113 console.error('Unable to resolve platform.');
114 }
115 > platform.ts
116 > export const enum Platform {
117 > Web,
118 > Mac,
119 > Linux,
120 > Windows
121 > }
122 > export type PlatformName = 'Web' | 'Windows' | 'Mac' | 'Linux';
123 >
124 > export function PlatformToString(platform: Platform): PlatformName {
125 switch (platform) {
126 case Platform.Web: return 'Web';
130 }
131 }
132 > platform.ts
133 > let _platform: Platform = Platform.Web;
134 > if (_isMacintosh) {
135 _platform = Platform.Mac;
136 > } else if (_isWindows) { platform.ts
137 _platform = Platform.Windows;
138 > } else if (_isLinux) { platform.ts
139 > _platform = Platform.Linux;
140 > }
141 >
142 > export const isWindows = _isWindows;
143 > export const isMacintosh = _isMacintosh;
144 > export const isLinux = _isLinux;
145 > export const isLinuxSnap = _isLinuxSnap;
146 > export const isNative = _isNative;
147 > export const isElectron = _isElectron;
148 > export const isWeb = _isWeb;
149 > export const isWebWorker = (_isWeb && typeof $globalThis.importScripts === 'function');
150 > export const webWorkerOrigin = isWebWorker ? $globalThis.origin : undefined;
151 > export const isIOS = _isIOS;
152 > export const isMobile = _isMobile;
153 > /**
154 > * Whether we run inside a CI environment, such as
155 > * GH actions or Azure Pipelines.
156 > */
157 > export const isCI = _isCI;
158 > export const platform = _platform;
159 > export const userAgent = _userAgent;
160 >
161 > /**
162 > * The language used for the user interface. The format of
163 > * the string is all lower case (e.g. zh-tw for Traditional
164 > * Chinese or de for German)
165 > */
166 > export const language = _language;
167 >
168 > export namespace Language {
169 >
170 > export function value(): string {
171 return language;
172 }
173 > platform.ts
174 > export function isDefaultVariant(): boolean {
175 if (language.length === 2) {
176 return language === 'en';
181 }
182 }
183 > platform.ts
184 > export function isDefault(): boolean {
185 return language === 'en';
186 }
187 > } platform.ts
188 >
189 > /**
190 > * Desktop: The OS locale or the locale specified by --locale or `argv.json`.
191 > * Web: matches `platformLocale`.
192 > *
193 > * The UI is not necessarily shown in the provided locale.
194 > */
195 > export const locale = _locale;
196 >
197 > /**
198 > * This will always be set to the OS/browser's locale regardless of
199 > * what was specified otherwise. The format of the string is all
200 > * lower case (e.g. zh-tw for Traditional Chinese). The UI is not
201 > * necessarily shown in the provided locale.
202 > */
203 > export const platformLocale = _platformLocale;
204 >
205 > /**
206 > * The translations that are available through language packs.
207 > */
208 > export const translationsConfigFile = _translationsConfigFile;
209 >
210 > export const setTimeout0IsFaster = (typeof $globalThis.postMessage === 'function' && !$globalThis.importScripts);
211 >
212 > /**
213 > * See https://html.spec.whatwg.org/multipage/timers-and-user-prompts.html#:~:text=than%204%2C%20then-,set%20timeout%20to%204,-.
214 > *
215 > * Works similarly to `setTimeout(0)` but doesn't suffer from the 4ms artificial delay
216 > * that browsers set when the nesting level is > 5.
217 > */
218 > export const setTimeout0 = (() => {
219 > if (setTimeout0IsFaster) {
220 interface IQueueElement {
221 id: number;
246 };
247 }
248 > return (callback: () => void) => setTimeout(callback); platform.ts
249 > })();
250 >
251 > export const enum OperatingSystem {
252 > Windows = 1,
253 > Macintosh = 2,
254 > Linux = 3
255 > }
256 > export const OS = (_isMacintosh || _isIOS ? OperatingSystem.Macintosh : (_isWindows ? OperatingSystem.Windows : OperatingSystem.Linux));
257 >
258 > let _isLittleEndian = true;
259 > let _isLittleEndianComputed = false;
260 > export function isLittleEndian(): boolean {
261 if (!_isLittleEndianComputed) {
262 _isLittleEndianComputed = true;
269 return _isLittleEndian;
270 }
271 > platform.ts
272 > export const isChrome = !!(userAgent && userAgent.indexOf('Chrome') >= 0);
273 > export const isFirefox = !!(userAgent && userAgent.indexOf('Firefox') >= 0);
274 > export const isSafari = !!(!isChrome && (userAgent && userAgent.indexOf('Safari') >= 0));
275 > export const isEdge = !!(userAgent && userAgent.indexOf('Edg/') >= 0);
276 > export const isAndroid = !!(userAgent && userAgent.indexOf('Android') >= 0);
277 > export const hasElectronUserAgent = !!(userAgent && userAgent.indexOf('Electron') >= 0);
278 >
279 > export function isTahoeOrNewer(osVersion: string): boolean {
280 return parseFloat(osVersion) >= 25;
281 }
src/vs/workbench/contrib/chat/test/common/mockChatSessionsService.ts 186 covered LOC · 61 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockChatSessionsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
7 > import { Emitter } from '../../../../../base/common/event.js';
8 > import { IDisposable } from '../../../../../base/common/lifecycle.js';
9 > import { ResourceMap } from '../../../../../base/common/map.js';
10 > import { ThemeIcon } from '../../../../../base/common/themables.js';
11 > import { URI } from '../../../../../base/common/uri.js';
12 > import { ReadonlyChatSessionOptionsMap, IChatNewSessionRequest, IChatSession, IChatSessionCommitEvent, IChatSessionContentProvider, IChatSessionCustomizationItemGroup, IChatSessionCustomizationsProvider, IChatSessionItem, IChatSessionItemController, IChatSessionItemsDelta, IChatSessionOptionsChangeEvent, IChatSessionProviderOptionGroup, IChatSessionRequestHistoryItem, IChatSessionsExtensionPoint, IChatSessionsService, ResolvedChatSessionsExtensionPoint, ChatSessionOptionsMap, IChatInputCompletionsParams, IChatInputCompletionsResult } from '../../common/chatSessionsService.js';
13 > import { getChatSessionType } from '../../common/model/chatUri.js';
14 > import { IChatAgentAttachmentCapabilities } from '../../common/participants/chatAgents.js';
15 > import { Target } from '../../common/promptSyntax/promptTypes.js';
16 >
17 > export class MockChatSessionsService implements IChatSessionsService {
18 > _serviceBrand: undefined; mockChatSessionsService.ts
19 >
20 > private readonly _onDidChangeSessionOptions = new Emitter<IChatSessionOptionsChangeEvent>();
21 > readonly onDidChangeSessionOptions = this._onDidChangeSessionOptions.event;
22 >
23 > private readonly _onDidChangeItemsProviders = new Emitter<{ readonly chatSessionType: string }>();
24 > readonly onDidChangeItemsProviders = this._onDidChangeItemsProviders.event;
25 >
26 > private readonly _onDidChangeSessionItems = new Emitter<IChatSessionItemsDelta>();
27 > readonly onDidChangeSessionItems = this._onDidChangeSessionItems.event;
28 >
29 > private readonly _onDidChangeAvailability = new Emitter<void>();
30 > readonly onDidChangeAvailability = this._onDidChangeAvailability.event;
31 >
32 > private readonly _onDidChangeInProgress = new Emitter<void>();
33 > readonly onDidChangeInProgress = this._onDidChangeInProgress.event;
34 >
35 > private readonly _onDidChangeContentProviderSchemes = new Emitter<{ readonly added: string[]; readonly removed: string[] }>();
36 > readonly onDidChangeContentProviderSchemes = this._onDidChangeContentProviderSchemes.event;
37 >
38 > private readonly _onDidChangeOptionGroups = new Emitter<string>();
39 > readonly onDidChangeOptionGroups = this._onDidChangeOptionGroups.event;
40 >
41 > private readonly _onDidCommitSession = new Emitter<IChatSessionCommitEvent>();
42 > readonly onDidCommitSession = this._onDidCommitSession.event;
43 >
44 >
45 > private sessionItemControllers = new Map<string, { readonly controller: IChatSessionItemController; readonly initialRefresh: Promise<void> }>();
46 > private contentProviders = new Map<string, IChatSessionContentProvider>();
47 > private contributions: IChatSessionsExtensionPoint[] = [];
48 > private optionGroups = new Map<string, IChatSessionProviderOptionGroup[]>();
49 > private sessionOptions = new ResourceMap<ChatSessionOptionsMap>();
50 > private inProgress = new Map</* chatSessionType*/ string, number>();
51 >
52 > // For testing: allow triggering events
53 > fireDidChangeItemsProviders(event: { chatSessionType: string }): void {
54 > this._onDidChangeItemsProviders.fire(event);
55 > }
56 >
57 > fireDidChangeSessionItems(event: IChatSessionItemsDelta): void {
58 this._onDidChangeSessionItems.fire(event);
59 }
61 > fireDidChangeAvailability(): void {
62 this._onDidChangeAvailability.fire();
63 }
65 > fireDidChangeInProgress(): void {
66 this._onDidChangeInProgress.fire();
67 }
69 > registerChatSessionItemController(chatSessionType: string, controller: IChatSessionItemController): IDisposable {
70 this.sessionItemControllers.set(chatSessionType, { controller, initialRefresh: controller.refresh(CancellationToken.None) });
71 return {
75 };
76 }
78 > getRegisteredChatSessionItemProviders(): readonly string[] {
79 return Array.from(this.sessionItemControllers.keys());
80 }
82 > getAllChatSessionContributions(): ResolvedChatSessionsExtensionPoint[] {
83 return this.contributions.map(contribution => this.resolveContribution(contribution));
84 }
86 > getChatSessionContribution(chatSessionType: string): ResolvedChatSessionsExtensionPoint | undefined {
87 const contribution = this.contributions.find(c => c.type === chatSessionType);
88 if (!contribution) {
92 return this.resolveContribution(contribution);
93 }
95 > private resolveContribution(contribution: IChatSessionsExtensionPoint): ResolvedChatSessionsExtensionPoint {
96 return {
97 ...contribution,
99 };
100 }
102 > setContributions(contributions: IChatSessionsExtensionPoint[]): void {
103 this.contributions = contributions;
104 }
106 > async activateChatSessionItemProvider(chatSessionType: string): Promise<void> {
107 // Noop, nothing to activate
108 }
110 > async *getChatSessionItems(providerTypeFilter: readonly string[] | undefined, token: CancellationToken): AsyncIterable<{ readonly chatSessionType: string; readonly items: readonly IChatSessionItem[] }> {
111 for (const [chatSessionType, controllerEntry] of this.sessionItemControllers.entries()) {
112 if (!providerTypeFilter || providerTypeFilter.includes(chatSessionType)) {
119 }
120 }
122 > async refreshChatSessionItems(providerTypeFilter: readonly string[] | undefined, token: CancellationToken): Promise<void> {
123 await Promise.all(
124 Array.from(this.sessionItemControllers.entries())
128 }));
129 }
131 > getInProgress(): { chatSessionType: string; count: number }[] {
132 return Array.from(this.inProgress.entries()).map(([chatSessionType, count]) => ({ chatSessionType, count }));
133 }
135 > async resolveChatSessionItem(_chatSessionType: string, _resource: URI, _token: CancellationToken): Promise<IChatSessionItem | undefined> {
136 return undefined;
137 }
139 > canSetChatSessionItemArchived(sessionResource: URI): boolean {
140 const sessionType = getChatSessionType(sessionResource);
141 return typeof this.sessionItemControllers.get(sessionType)?.controller.setChatSessionItemArchived === 'function';
142 }
144 > setChatSessionItemArchived(sessionResource: URI, archived: boolean): void {
145 const sessionType = getChatSessionType(sessionResource);
146 const controller = this.sessionItemControllers.get(sessionType)?.controller;
150 controller.setChatSessionItemArchived(sessionResource, archived);
151 }
153 > registerChatSessionContentProvider(chatSessionType: string, provider: IChatSessionContentProvider): IDisposable {
154 > this.contentProviders.set(chatSessionType, provider); mockChatSessionsService.ts
155 > this._onDidChangeContentProviderSchemes.fire({ added: [chatSessionType], removed: [] });
156 > return {
157 > dispose: () => {
158 > this.contentProviders.delete(chatSessionType);
159 > }
160 > };
161 > }
163 > async canResolveContentProvider(chatSessionType: string): Promise<boolean> {
164 return this.contentProviders.has(chatSessionType);
165 }
167 > async getOrCreateChatSession(sessionResource: URI, token: CancellationToken): Promise<IChatSession> {
168 > const sessionType = getChatSessionType(sessionResource); mockChatSessionsService.ts
169 > const provider = this.contentProviders.get(sessionType);
170 > if (!provider) {
171 throw new Error(`No content provider for ${sessionType}`);
172 }
173 > return provider.provideChatSessionContent(sessionResource, token); mockChatSessionsService.ts
174 > }
176 > async canResolveChatSession(sessionType: string): Promise<boolean> {
177 > return this.contentProviders.has(sessionType); mockChatSessionsService.ts
178 > }
180 > async provideChatInputCompletions(sessionResource: URI, params: IChatInputCompletionsParams, token: CancellationToken): Promise<IChatInputCompletionsResult | undefined> {
181 const sessionType = getChatSessionType(sessionResource);
182 const provider = this.contentProviders.get(sessionType);
186 return provider.provideChatInputCompletions(sessionResource, params, token);
187 }
189 > resolveChatResponseUri(sessionResource: URI, href: string, kind: 'link' | 'image'): string {
190 const sessionType = getChatSessionType(sessionResource);
191 return this.contentProviders.get(sessionType)?.resolveChatResponseUri?.(sessionResource, href, kind) ?? href;
192 }
194 > async getChatInputCompletionTriggerCharacters(sessionType: string): Promise<readonly string[] | undefined> {
195 const provider = this.contentProviders.get(sessionType);
196 if (!provider) {
202 return provider.provideChatInputCompletionTriggerCharacters();
203 }
205 > getOptionGroupsForSessionType(chatSessionType: string): IChatSessionProviderOptionGroup[] | undefined {
206 return this.optionGroups.get(chatSessionType);
207 }
209 > setOptionGroupsForSessionType(chatSessionType: string, handle: number, optionGroups?: IChatSessionProviderOptionGroup[]): void {
210 if (optionGroups) {
211 this.optionGroups.set(chatSessionType, optionGroups);
214 }
215 }
217 > async getNewChatSessionInputState(_chatSessionType: string, _sessionResource: URI): Promise<readonly IChatSessionProviderOptionGroup[] | undefined> {
218 return undefined;
219 }
221 > getSessionOptions(sessionResource: URI): ReadonlyChatSessionOptionsMap | undefined {
222 const options = this.sessionOptions.get(sessionResource);
223 return options && options.size > 0 ? options : undefined;
224 }
226 > getSessionOption(sessionResource: URI, optionId: string): string | undefined {
227 const value = this.sessionOptions.get(sessionResource)?.get(optionId);
228 return typeof value === 'string' ? value : value?.id;
229 }
231 > setSessionOption(sessionResource: URI, optionId: string, value: string): boolean {
232 return this.updateSessionOptions(sessionResource, new Map([[optionId, value]]));
233 }
235 > updateSessionOptions(sessionResource: URI, updates: ReadonlyChatSessionOptionsMap): boolean {
236 if (!this.sessionOptions.has(sessionResource)) {
237 this.sessionOptions.set(sessionResource, new Map());
245 return true;
246 }
248 > getCapabilitiesForSessionType(chatSessionType: string): IChatAgentAttachmentCapabilities | undefined {
249 > return this.contributions.find(c => c.type === chatSessionType)?.capabilities; mockChatSessionsService.ts
250 > }
252 > getCustomAgentTargetForSessionType(chatSessionType: string): Target {
253 return this.contributions.find(c => c.type === chatSessionType)?.customAgentTarget ?? Target.Undefined;
254 }
256 > requiresCustomModelsForSessionType(chatSessionType: string): boolean {
257 return this.contributions.find(c => c.type === chatSessionType)?.requiresCustomModels ?? false;
258 }
260 > supportsAutoModelForSessionType(chatSessionType: string): boolean {
261 return !!this.contributions.find(c => c.type === chatSessionType)?.supportsAutoModel;
262 }
264 > requiresCopilotSignInForSessionType(chatSessionType: string): boolean {
265 return !!this.contributions.find(c => c.type === chatSessionType)?.requiresCopilotSignIn;
266 }
268 > supportsDelegationForSessionType(chatSessionType: string): boolean {
269 return this.contributions.find(c => c.type === chatSessionType)?.supportsDelegation !== false;
270 }
272 > sessionSupportsFork(_sessionResource: URI): boolean {
273 return false;
274 }
276 > async forkChatSession(_sessionResource: URI, _request: IChatSessionRequestHistoryItem | undefined, _token: CancellationToken): Promise<IChatSessionItem> {
277 throw new Error('Not implemented');
278 }
280 > sessionSupportsRename(_sessionResource: URI): boolean {
281 return false;
282 }
284 > async renameChatSession(_sessionResource: URI, _title: string, _token: CancellationToken): Promise<void> {
285 throw new Error('Not implemented');
286 }
288 > getContentProviderSchemes(): string[] {
289 return Array.from(this.contentProviders.keys());
290 }
292 > async createNewChatSessionItem(_chatSessionType: string, _request: IChatNewSessionRequest, _token: CancellationToken): Promise<IChatSessionItem | undefined> {
293 return undefined;
294 }
296 > async deleteChatSessionItem(sessionResource: URI, token: CancellationToken): Promise<void> {
297 const chatSessionType = getChatSessionType(sessionResource);
298 const controllerData = this.sessionItemControllers.get(chatSessionType);
303 return controllerData.controller.deleteChatSessionItem(sessionResource, token);
304 }
306 > private readonly _resourceAliases = new ResourceMap<URI>(); // real -> untitled
307 > private readonly _realResources = new ResourceMap<URI>(); // untitled -> real
308 >
309 > registerSessionResourceAlias(untitledResource: URI, realResource: URI): void {
310 this._resourceAliases.set(realResource, untitledResource);
311 }
313 > setMaterializedSessionResource(untitledResource: URI, realResource: URI): void {
314 this._realResources.set(untitledResource, realResource);
315 }
317 > getMaterializedSessionResource(untitledResource: URI): URI | undefined {
318 > return this._realResources.get(untitledResource); mockChatSessionsService.ts
319 > }
321 > clearMaterializedSessionResource(sessionResource: URI): void {
322 this._realResources.delete(sessionResource);
323 const untitled = this._resourceAliases.get(sessionResource);
326 }
327 }
329 > fireSessionCommitted(_original: URI, _committed: URI): void {
330 // noop
331 }
333 > registerChatSessionContribution(contribution: IChatSessionsExtensionPoint): IDisposable {
334 this.contributions.push(contribution);
335 return {
342 };
343 }
345 > private readonly _onDidChangeCustomizations = new Emitter<{ readonly chatSessionType: string }>();
346 > readonly onDidChangeCustomizations = this._onDidChangeCustomizations.event;
348 > registerCustomizationsProvider(_chatSessionType: string, _provider: IChatSessionCustomizationsProvider): IDisposable {
349 return { dispose: () => { } };
350 }
352 > hasCustomizationsProvider(_chatSessionType: string): boolean {
353 return false;
354 }
356 > async getCustomizations(_chatSessionType: string, _token: CancellationToken): Promise<IChatSessionCustomizationItemGroup[] | undefined> {
357 return undefined;
358 }
360 > }
src/vs/base/common/errors.ts 183 covered LOC · 32 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- errors.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export interface ErrorListenerCallback {
7 > (error: any): void;
8 > }
9 >
10 > export interface ErrorListenerUnbind {
11 > (): void;
12 > }
13 >
14 > // Avoid circular dependency on EventEmitter by implementing a subset of the interface.
15 > export class ErrorHandler {
16 > private unexpectedErrorHandler: (e: any) => void;
17 > private listeners: ErrorListenerCallback[];
18 >
19 > constructor() {
20 >
21 > this.listeners = [];
22 >
23 > this.unexpectedErrorHandler = function (e: any) {
24 setTimeout(() => {
25 if (e.stack) {
34 }, 0);
35 };
36 > } errors.ts
37 >
38 > addListener(listener: ErrorListenerCallback): ErrorListenerUnbind {
39 this.listeners.push(listener);
40
43 };
44 }
45 > errors.ts
46 > private emit(e: any): void {
47 this.listeners.forEach((listener) => {
48 listener(e);
49 });
50 }
51 > errors.ts
52 > private _removeListener(listener: ErrorListenerCallback): void {
53 this.listeners.splice(this.listeners.indexOf(listener), 1);
54 }
55 > errors.ts
56 > setUnexpectedErrorHandler(newUnexpectedErrorHandler: (e: any) => void): void {
57 > this.unexpectedErrorHandler = newUnexpectedErrorHandler;
58 > }
59 >
60 > getUnexpectedErrorHandler(): (e: any) => void {
61 return this.unexpectedErrorHandler;
62 }
63 > errors.ts
64 > onUnexpectedError(e: any): void {
65 this.unexpectedErrorHandler(e);
66 this.emit(e);
67 }
68 > errors.ts
69 > // For external errors, we don't want the listeners to be called
70 > onUnexpectedExternalError(e: any): void {
71 this.unexpectedErrorHandler(e);
72 }
73 > } errors.ts
74 >
75 > export const errorHandler = new ErrorHandler();
76 >
77 > /** @skipMangle */
78 > export function setUnexpectedErrorHandler(newUnexpectedErrorHandler: (e: any) => void): void {
79 > errorHandler.setUnexpectedErrorHandler(newUnexpectedErrorHandler);
80 > }
81 >
82 > /**
83 > * Returns if the error is a SIGPIPE error. SIGPIPE errors should generally be
84 > * logged at most once, to avoid a loop.
85 > *
86 > * @see https://github.com/microsoft/vscode-remote-release/issues/6481
87 > */
88 > export function isSigPipeError(e: unknown): e is Error {
89 if (!e || typeof e !== 'object') {
90 return false;
94 return cast.code === 'EPIPE' && cast.syscall?.toUpperCase() === 'WRITE';
95 }
96 > errors.ts
97 > /**
98 > * This function should only be called with errors that indicate a bug in the product.
99 > * E.g. buggy extensions/invalid user-input/network issues should not be able to trigger this code path.
100 > * If they are, this indicates there is also a bug in the product.
101 > */
102 > export function onBugIndicatingError(e: any): undefined {
103 errorHandler.onUnexpectedError(e);
104 return undefined;
105 }
106 > errors.ts
107 > export function onUnexpectedError(e: any): undefined {
108 // ignore errors from cancelled promises
109 if (!isCancellationError(e)) {
112 return undefined;
113 }
114 > errors.ts
115 > export function onUnexpectedExternalError(e: any): undefined {
116 // ignore errors from cancelled promises
117 if (!isCancellationError(e)) {
120 return undefined;
121 }
122 > errors.ts
123 > type ObjectWithCode = {
124 > readonly code: unknown;
125 > };
126 >
127 function hasErrorCode(error: object): error is ObjectWithCode {
128 return Object.hasOwn(error, 'code');
129 }
130 > errors.ts
131 > export function getErrorCode(error: unknown): string | undefined {
132 if (!error || typeof error !== 'object' || !hasErrorCode(error)) {
133 return undefined;
136 return typeof code === 'string' || typeof code === 'number' ? String(code) : undefined;
137 }
138 > errors.ts
139 > export interface SerializedError {
140 > readonly $isError: true;
141 > readonly name: string;
142 > readonly message: string;
143 > readonly stack: string;
144 > readonly noTelemetry: boolean;
145 > readonly code?: string;
146 > readonly cause?: SerializedError;
147 > }
148 >
149 > type ErrorWithCode = Error & {
150 > code: string | undefined;
151 > };
152 >
153 > export function transformErrorForSerialization(error: Error): SerializedError;
154 > export function transformErrorForSerialization(error: any): any;
155 > export function transformErrorForSerialization(error: any): any {
156 if (error instanceof Error) {
157 const { name, message, cause } = error;
172 return error;
173 }
174 > errors.ts
175 > export function transformErrorFromSerialization(data: SerializedError): Error {
176 let error: Error;
177 if (data.noTelemetry) {
191 return error;
192 }
193 > errors.ts
194 > // see https://github.com/v8/v8/wiki/Stack%20Trace%20API#basic-stack-traces
195 > export interface V8CallSite {
196 > getThis(): unknown;
197 > getTypeName(): string | null;
198 > getFunction(): Function | undefined;
199 > getFunctionName(): string | null;
200 > getMethodName(): string | null;
201 > getFileName(): string | null;
202 > getLineNumber(): number | null;
203 > getColumnNumber(): number | null;
204 > getEvalOrigin(): string | undefined;
205 > isToplevel(): boolean;
206 > isEval(): boolean;
207 > isNative(): boolean;
208 > isConstructor(): boolean;
209 > toString(): string;
210 > }
211 >
212 > export const canceledName = 'Canceled';
213 >
214 > /**
215 > * Checks if the given error is a promise in canceled state
216 > */
217 > export function isCancellationError(error: any): boolean {
218 if (error instanceof CancellationError) {
219 return true;
221 return error instanceof Error && error.name === canceledName && error.message === canceledName;
222 }
223 > errors.ts
224 > // !!!IMPORTANT!!!
225 > // Do NOT change this class because it is also used as an API-type.
226 > export class CancellationError extends Error {
227 > constructor() {
228 super(canceledName);
229 this.name = this.message;
230 }
231 > } errors.ts
232 >
233 > export class PendingMigrationError extends Error {
234 >
235 > private static readonly _name = 'PendingMigrationError';
236 >
237 > static is(error: unknown): error is PendingMigrationError {
238 return error instanceof PendingMigrationError || (error instanceof Error && error.name === PendingMigrationError._name);
239 }
240 > errors.ts
241 > constructor(message: string) {
242 super(message);
243 this.name = PendingMigrationError._name;
244 }
245 > } errors.ts
246 >
247 > /**
248 > * @deprecated use {@link CancellationError `new CancellationError()`} instead
249 > */
250 > export function canceled(): Error {
251 const error = new Error(canceledName);
252 error.name = error.message;
253 return error;
254 }
255 > errors.ts
256 > export function illegalArgument(name?: string): Error {
257 if (name) {
258 return new Error(`Illegal argument: ${name}`);
261 }
262 }
263 > errors.ts
264 > export function illegalState(name?: string): Error {
265 if (name) {
266 return new Error(`Illegal state: ${name}`);
269 }
270 }
271 > errors.ts
272 > export class ReadonlyError extends TypeError {
273 > constructor(name?: string) {
274 super(name ? `${name} is read-only and cannot be changed` : 'Cannot change read-only property');
275 }
276 > } errors.ts
277 >
278 > export function getErrorMessage(err: any): string {
279 if (!err) {
280 return 'Error';
291 return String(err);
292 }
293 > errors.ts
294 > export class NotImplementedError extends Error {
295 > constructor(message?: string) {
296 super('NotImplemented');
297 if (message) {
299 }
300 }
301 > } errors.ts
302 >
303 > export class NotSupportedError extends Error {
304 > constructor(message?: string) {
305 super('NotSupported');
306 if (message) {
308 }
309 }
310 > } errors.ts
311 >
312 > export class ExpectedError extends Error {
313 readonly isExpected = true;
314 > } errors.ts
315 >
316 > /**
317 > * Error that when thrown won't be logged in telemetry as an unhandled error.
318 > */
319 > export class ErrorNoTelemetry extends Error {
320 > override readonly name: string;
321 >
322 > constructor(msg?: string) {
323 super(msg);
324 this.name = 'CodeExpectedError';
325 }
326 > errors.ts
327 > public static fromError(err: Error): ErrorNoTelemetry {
328 if (err instanceof ErrorNoTelemetry) {
329 return err;
335 return result;
336 }
337 > errors.ts
338 > public static isErrorNoTelemetry(err: Error): err is ErrorNoTelemetry {
339 return err.name === 'CodeExpectedError';
340 }
341 > } errors.ts
342 >
343 > /**
344 > * This error indicates a bug.
345 > * Do not throw this for invalid user input.
346 > * Only catch this error to recover gracefully from bugs.
347 > */
348 > export class BugIndicatingError extends Error {
349 > constructor(message?: string) {
350 super(message || 'An unexpected bug occurred.');
351 Object.setPrototypeOf(this, BugIndicatingError.prototype);
src/vs/workbench/contrib/chat/common/promptSyntax/config/config.ts 180 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- config.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import type { IConfigurationService } from '../../../../../../platform/configuration/common/configuration.js';
7 > import { URI } from '../../../../../../base/common/uri.js';
8 > import { PromptFileSource, PromptsType } from '../promptTypes.js';
9 > import { getPromptFileDefaultLocations, IPromptSourceFolder } from './promptFileLocations.js';
10 > import { PromptsStorage } from '../service/promptsService.js';
11 >
12 > /**
13 > * Configuration helper for the `reusable prompts` feature.
14 > * @see {@link PromptsConfig.PROMPT_LOCATIONS_KEY}, {@link PromptsConfig.INSTRUCTIONS_LOCATION_KEY}, {@link PromptsConfig.MODE_LOCATION_KEY}, or {@link PromptsConfig.PROMPT_FILES_SUGGEST_KEY}.
15 > *
16 > * ### Functions
17 > *
18 > * - {@link getLocationsValue} allows to current read configuration value
19 > * - {@link promptSourceFolders} gets list of source folders for prompt files
20 > * - {@link getPromptFilesRecommendationsValue} gets prompt file recommendation configuration
21 > *
22 > * ### File Paths Resolution
23 > *
24 > * We resolve only `*.prompt.md` files inside the resulting source folders. Relative paths are resolved
25 > * relative to:
26 > *
27 > * - the current workspace `root`, if applicable, in other words one of the workspace folders
28 > * can be used as a prompt files source folder
29 > * - root of each top-level folder in the workspace (if there are multiple workspace folders)
30 > * - current root folder (if a single folder is open)
31 > *
32 > * ### Prompt File Suggestions
33 > *
34 > * The `chat.promptFilesRecommendations` setting allows configuring which prompt files to suggest in different contexts:
35 > *
36 > * ```json
37 > * {
38 > * "chat.promptFilesRecommendations": {
39 > * "plan": true, // Always suggest
40 > * "new-page": "resourceExtname == .js", // Suggest for JavaScript files
41 > * "draft-blog": "resourceLangId == markdown", // Suggest for Markdown files
42 > * "debug": false // Never suggest
43 > * }
44 > * }
45 > * ```
46 > */
47 > export namespace PromptsConfig {
48 > /**
49 > * Configuration key for the locations of reusable prompt files.
50 > */
51 > export const PROMPT_LOCATIONS_KEY = 'chat.promptFilesLocations';
52 >
53 > /**
54 > * Configuration key for the locations of instructions files.
55 > */
56 > export const INSTRUCTIONS_LOCATION_KEY = 'chat.instructionsFilesLocations';
57 > /**
58 > * Configuration key for the locations of mode files.
59 > * @deprecated Use {@link AGENTS_LOCATION_KEY} instead
60 > */
61 > export const MODE_LOCATION_KEY = 'chat.modeFilesLocations';
62 >
63 > /**
64 > * Configuration key for the locations of agent files (with simplified path support).
65 > */
66 > export const AGENTS_LOCATION_KEY = 'chat.agentFilesLocations';
67 >
68 > /**
69 > * Configuration key for the locations of skill folders.
70 > */
71 > export const SKILLS_LOCATION_KEY = 'chat.agentSkillsLocations';
72 >
73 > /**
74 > * Configuration key for the locations of hook files.
75 > */
76 > export const HOOKS_LOCATION_KEY = 'chat.hookFilesLocations';
77 >
78 > /**
79 > * Configuration key for prompt file suggestions.
80 > */
81 > export const PROMPT_FILES_SUGGEST_KEY = 'chat.promptFilesRecommendations';
82 >
83 > /**
84 > * Configuration key for use of the copilot instructions file.
85 > */
86 > export const USE_COPILOT_INSTRUCTION_FILES = 'github.copilot.chat.codeGeneration.useInstructionFiles';
87 >
88 > /**
89 > * Configuration key for the AGENTS.md.
90 > */
91 > export const USE_AGENT_MD = 'chat.useAgentsMdFile';
92 >
93 > /**
94 > * Configuration key for nested AGENTS.md files.
95 > */
96 > export const USE_NESTED_AGENT_MD = 'chat.useNestedAgentsMdFiles';
97 >
98 > /**
99 > * Configuration key for the CLAUDE.md.
100 > */
101 > export const USE_CLAUDE_MD = 'chat.useClaudeMdFile';
102 >
103 > /**
104 > * Configuration key for agent skills usage.
105 > */
106 > export const USE_AGENT_SKILLS = 'chat.useAgentSkills';
107 >
108 > /**
109 > * Configuration key for chat hooks usage.
110 > */
111 > export const USE_CHAT_HOOKS = 'chat.useHooks';
112 >
113 > /**
114 > * Configuration key for enabling Claude hooks.
115 > */
116 > export const USE_CLAUDE_HOOKS = 'chat.useClaudeHooks';
117 >
118 > /**
119 > * Configuration key for enabling stronger skill adherence prompt (experimental).
120 > */
121 > export const USE_SKILL_ADHERENCE_PROMPT = 'chat.experimental.useSkillAdherencePrompt';
122 >
123 > /**
124 > * Configuration key for including applying instructions.
125 > */
126 > export const INCLUDE_APPLYING_INSTRUCTIONS = 'chat.includeApplyingInstructions';
127 >
128 > /**
129 > * Configuration key for including referenced instructions.
130 > */
131 > export const INCLUDE_REFERENCED_INSTRUCTIONS = 'chat.includeReferencedInstructions';
132 >
133 > /**
134 > * Search for configuration files in parent repositories of the workspace folder
135 > */
136 > export const USE_CUSTOMIZATIONS_IN_PARENT_REPOS = 'chat.useCustomizationsInParentRepositories';
137 >
138 > /**
139 > * Get value of the `reusable prompt locations` configuration setting.
140 > * @see {@link PROMPT_LOCATIONS_CONFIG_KEY}, {@link INSTRUCTIONS_LOCATIONS_CONFIG_KEY}, {@link MODE_LOCATIONS_CONFIG_KEY}, {@link SKILLS_LOCATION_KEY}.
141 > */
142 > export function getLocationsValue(configService: IConfigurationService, type: PromptsType): Record<string, boolean> | undefined {
143 const key = getPromptFileLocationsConfigKey(type);
144 const configValue = configService.getValue(key);
169 return undefined;
170 }
171 > config.ts
172 > /**
173 > * Gets list of source folders for prompt files.
174 > * Defaults to {@link PROMPT_DEFAULT_SOURCE_FOLDER}, {@link INSTRUCTIONS_DEFAULT_SOURCE_FOLDER}, {@link MODE_DEFAULT_SOURCE_FOLDER} or {@link SKILLS_LOCATION_KEY}.
175 > */
176 > export function promptSourceFolders(configService: IConfigurationService, type: PromptsType): IPromptSourceFolder[] {
177 const value = getLocationsValue(configService, type);
178 const defaultSourceFolders = getPromptFileDefaultLocations(type);
208 return [];
209 }
210 > config.ts
211 > /**
212 > * Get value of the prompt file recommendations configuration setting.
213 > * @param configService Configuration service instance
214 > * @param resource Optional resource URI to get workspace folder-specific settings
215 > * @see {@link PROMPT_FILES_SUGGEST_KEY}.
216 > */
217 > export function getPromptFilesRecommendationsValue(configService: IConfigurationService, resource?: URI): Record<string, boolean | string> | undefined {
218 // Get the merged configuration value (VS Code automatically merges all levels: default → user → workspace → folder)
219 const configValue = configService.getValue(PromptsConfig.PROMPT_FILES_SUGGEST_KEY, { resource });
258 return Object.keys(suggestions).length > 0 ? suggestions : undefined;
259 }
260 > config.ts
261 > }
262 >
263 > export function getPromptFileLocationsConfigKey(type: PromptsType): string {
264 switch (type) {
265 case PromptsType.instructions:
277 }
278 }
279 > config.ts
280 > /**
281 > * Helper to parse an input value of `any` type into a boolean.
282 > *
283 > * @param value - input value to parse
284 > * @returns `true` if the value is the boolean `true` value or a string that can
285 > * be clearly mapped to a boolean (e.g., `"true"`, `"TRUE"`, `"FaLSe"`, etc.),
286 > * `undefined` for rest of the values
287 > */
288 > export function asBoolean(value: unknown): boolean | undefined {
289 if (typeof value === 'boolean') {
290 return value;
306 return undefined;
307 }
308 > config.ts
309 > /**
310 > * Helper to check if a path starts with tilde (user home).
311 > * Only supports Unix-style (`~/`) paths for cross-platform sharing.
312 > * Backslash paths (`~\`) are not supported to ensure paths are shareable in repos.
313 > *
314 > * @param path - path to check
315 > * @returns `true` if the path starts with `~/`
316 > */
317 > export function isTildePath(path: string): boolean {
318 return path.startsWith('~/');
319 }
src/vs/platform/agentHost/common/state/protocol/channels-terminal/state.ts 173 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI } from '../common/state.js';
10 >
11 > // ─── Terminal Types ──────────────────────────────────────────────────────────
12 >
13 > /**
14 > * Lightweight terminal metadata exposed on the root state.
15 > *
16 > * @category Terminal Types
17 > */
18 > export interface TerminalInfo {
19 > /** Terminal URI (subscribable for full terminal state) */
20 > resource: URI;
21 > /** Human-readable terminal title */
22 > title: string;
23 > /** Who currently holds this terminal */
24 > claim: TerminalClaim;
25 > /** Process exit code, if the terminal process has exited */
26 > exitCode?: number;
27 > }
28 >
29 > /**
30 > * Discriminant for terminal claim kinds.
31 > *
32 > * @category Terminal Types
33 > */
34 > export const enum TerminalClaimKind {
35 > Client = 'client',
36 > Session = 'session',
37 > }
38 >
39 > /**
40 > * A terminal claimed by a connected client.
41 > *
42 > * @category Terminal Types
43 > */
44 > export interface TerminalClientClaim {
45 > /** Discriminant */
46 > kind: TerminalClaimKind.Client;
47 > /** The `clientId` of the claiming client */
48 > clientId: string;
49 > }
50 >
51 > /**
52 > * A terminal claimed by a session, optionally scoped to a specific turn or tool call.
53 > *
54 > * @category Terminal Types
55 > */
56 > export interface TerminalSessionClaim {
57 > /** Discriminant */
58 > kind: TerminalClaimKind.Session;
59 > /** Session URI that claimed the terminal */
60 > session: URI;
61 > /** Optional turn identifier within the session */
62 > turnId?: string;
63 > /** Optional tool call identifier within the turn */
64 > toolCallId?: string;
65 > }
66 >
67 > /**
68 > * Describes who currently holds a terminal. A terminal may be claimed by
69 > * either a connected client or a session (e.g. during a tool call).
70 > *
71 > * @category Terminal Types
72 > */
73 > export type TerminalClaim = TerminalClientClaim | TerminalSessionClaim;
74 >
75 > /**
76 > * Full state for a single terminal, loaded when a client subscribes to the terminal's URI.
77 > *
78 > * @category Terminal Types
79 > */
80 > export interface TerminalState {
81 > /** Human-readable terminal title */
82 > title: string;
83 > /** Current working directory of the terminal process */
84 > cwd?: URI;
85 > /** Terminal width in columns */
86 > cols?: number;
87 > /** Terminal height in rows */
88 > rows?: number;
89 > /**
90 > * Typed content parts, replacing the flat `content: string`.
91 > *
92 > * Naive consumers that only need the raw VT stream can reconstruct it with:
93 > * `content.map(p => p.type === 'command' ? p.output : p.value).join('')`
94 > *
95 > * Consumers that need command boundaries can filter by part type.
96 > */
97 > content: TerminalContentPart[];
98 > /** Process exit code, set when the terminal process exits */
99 > exitCode?: number;
100 > /** Who currently holds this terminal */
101 > claim: TerminalClaim;
102 > /**
103 > * Whether this terminal emits `terminal/commandExecuted` and
104 > * `terminal/commandFinished` actions and populates `command`-typed parts.
105 > *
106 > * Clients MUST check this flag before relying on command detection.
107 > * Do NOT use the presence of a `command` part as a feature flag — parts
108 > * are absent in the normal idle state.
109 > */
110 > supportsCommandDetection?: boolean;
111 > /**
112 > * Whether this terminal-style resource is backed by a pseudoterminal.
113 > * When `false`, output is plain text and clients do not need to parse
114 > * VT sequences.
115 > */
116 > isPty?: boolean;
117 > }
118 >
119 > // ─── Terminal Content Parts ──────────────────────────────────────────────────
120 >
121 > /**
122 > * A content part within terminal output.
123 > *
124 > * @category Terminal Types
125 > */
126 > export type TerminalContentPart =
127 > | TerminalUnclassifiedPart
128 > | TerminalCommandPart;
129 >
130 > /**
131 > * Unstructured terminal output — content before, between, or after commands,
132 > * or from terminals that do not support command detection.
133 > *
134 > * @category Terminal Types
135 > */
136 > export interface TerminalUnclassifiedPart {
137 > type: 'unclassified';
138 > /** Accumulated VT output. Appended to by `terminal/data` when no command is executing. */
139 > value: string;
140 > }
141 >
142 > /**
143 > * A single command: its command line and the output it produced.
144 > *
145 > * While `isComplete` is false the command is still executing; `output` grows
146 > * as `terminal/data` actions arrive. At `terminal/commandFinished` the part
147 > * is mutated in-place with `isComplete: true` and the completion metadata.
148 > *
149 > * @category Terminal Types
150 > */
151 > export interface TerminalCommandPart {
152 > type: 'command';
153 > /**
154 > * Stable id matching the `commandId` on the corresponding
155 > * `terminal/commandExecuted` and `terminal/commandFinished` actions.
156 > */
157 > commandId: string;
158 > /** The command line submitted to the shell. */
159 > commandLine: string;
160 > /**
161 > * Accumulated VT output. Appended to by `terminal/data` while `isComplete`
162 > * is false. Shell integration escape sequences are stripped by the server.
163 > */
164 > output: string;
165 > /** Unix timestamp (ms) when execution started, as reported by the server. */
166 > timestamp: number;
167 > /** Whether the command has finished. */
168 > isComplete: boolean;
169 > /** Shell exit code. Set at completion. `undefined` if unknown. */
170 > exitCode?: number;
171 > /** Wall-clock duration in milliseconds. Set at completion. */
172 > durationMs?: number;
173 > }
src/vs/platform/instantiation/common/instantiationService.ts 171 covered LOC · 43 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- instantiationService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { GlobalIdleValue } from '../../../base/common/async.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { illegalState } from '../../../base/common/errors.js';
9 > import { DisposableStore, dispose, IDisposable, isDisposable, toDisposable } from '../../../base/common/lifecycle.js';
10 > import { SyncDescriptor, SyncDescriptor0 } from './descriptors.js';
11 > import { Graph } from './graph.js';
12 > import { GetLeadingNonServiceArgs, IInstantiationService, ServiceIdentifier, ServicesAccessor, _util } from './instantiation.js';
13 > import { ServiceCollection } from './serviceCollection.js';
14 > import { LinkedList } from '../../../base/common/linkedList.js';
15 >
16 > // TRACING
17 > const _enableAllTracing = false
18 > // || "TRUE" // DO NOT CHECK IN!
19 > ;
20 >
21 > class CyclicDependencyError extends Error {
22 > constructor(graph: Graph<any>) {
23 super('cyclic dependency between services');
24 this.message = graph.findCycleSlow() ?? `UNABLE to detect cycle, dumping graph: \n${graph.toString()}`;
25 }
27 >
28 > export class InstantiationService implements IInstantiationService {
29 >
30 > declare readonly _serviceBrand: undefined;
31 >
32 > readonly _globalGraph?: Graph<string>;
33 > private _globalGraphImplicitDependency?: string;
34 >
35 > private _isDisposed = false;
36 > private readonly _servicesToMaybeDispose = new Set<any>();
37 > private readonly _children = new Set<InstantiationService>();
38 >
39 > constructor(
40 > private readonly _services: ServiceCollection = new ServiceCollection(), instantiationService.ts
41 > private readonly _strict: boolean = false,
42 > private readonly _parent?: InstantiationService,
43 > private readonly _enableTracing: boolean = _enableAllTracing
44 > ) {
45 >
46 > this._services.set(IInstantiationService, this);
47 > this._globalGraph = _enableTracing ? _parent?._globalGraph ?? new Graph(e => e) : undefined;
48 > }
50 > dispose(): void {
51 if (!this._isDisposed) {
52 this._isDisposed = true;
86 return result;
87 }
89 > invokeFunction<R, TS extends any[] = []>(fn: (accessor: ServicesAccessor, ...args: TS) => R, ...args: TS): R {
90 this._throwIfDisposed();
91
113 }
114 }
116 > createInstance<T>(descriptor: SyncDescriptor0<T>): T;
117 > createInstance<Ctor extends new (...args: any[]) => unknown, R extends InstanceType<Ctor>>(ctor: Ctor, ...args: GetLeadingNonServiceArgs<ConstructorParameters<Ctor>>): R;
118 > createInstance(ctorOrDescriptor: any | SyncDescriptor<any>, ...rest: unknown[]): unknown {
119 > this._throwIfDisposed(); instantiationService.ts
120 >
121 > let _trace: Trace;
122 > let result: unknown;
123 > if (ctorOrDescriptor instanceof SyncDescriptor) {
124 _trace = Trace.traceCreation(this._enableTracing, ctorOrDescriptor.ctor);
125 result = this._createInstance(ctorOrDescriptor.ctor, ctorOrDescriptor.staticArguments.concat(rest), _trace);
126 > } else { instantiationService.ts
127 > _trace = Trace.traceCreation(this._enableTracing, ctorOrDescriptor);
128 > result = this._createInstance(ctorOrDescriptor, rest, _trace);
129 > }
130 > _trace.stop();
131 > return result;
132 > }
134 > private _createInstance<T>(ctor: any, args: unknown[] = [], _trace: Trace): T {
136 > // arguments defined by service decorators
137 > const serviceDependencies = _util.getServiceDependencies(ctor).sort((a, b) => a.index - b.index);
138 > const serviceArgs: unknown[] = [];
139 > for (const dependency of serviceDependencies) {
140 > const service = this._getOrCreateServiceInstance(dependency.id, _trace); instantiationService.ts
141 > if (!service) {
142 > this._throwIfStrict(`[createInstance] ${ctor.name} depends on UNKNOWN service ${dependency.id}.`, false); instantiationService.ts
143 > }
144 > serviceArgs.push(service); instantiationService.ts
145 > }
147 > const firstServiceArgPos = serviceDependencies.length > 0 ? serviceDependencies[0].index : args.length;
148 >
149 > // check for argument mismatches, adjust static args if needed
150 > if (args.length !== firstServiceArgPos) {
151 console.trace(`[createInstance] First service dependency of ${ctor.name} at position ${firstServiceArgPos + 1} conflicts with ${args.length} static arguments`);
152
158 }
159 }
161 > // now create the instance
162 > return Reflect.construct<any, T>(ctor, args.concat(serviceArgs));
163 > }
165 > private _setCreatedServiceInstance<T>(id: ServiceIdentifier<T>, instance: T): void {
166 if (this._services.get(id) instanceof SyncDescriptor) {
167 this._services.set(id, instance);
172 }
173 }
175 > private _getServiceInstanceOrDescriptor<T>(id: ServiceIdentifier<T>): T | SyncDescriptor<T> {
176 > const instanceOrDesc = this._services.get(id); instantiationService.ts
177 > if (!instanceOrDesc && this._parent) {
178 return this._parent._getServiceInstanceOrDescriptor(id);
179 > } else { instantiationService.ts
180 > return instanceOrDesc;
181 > }
182 > }
184 > protected _getOrCreateServiceInstance<T>(id: ServiceIdentifier<T>, _trace: Trace): T {
185 > if (this._globalGraph && this._globalGraphImplicitDependency) { instantiationService.ts
186 this._globalGraph.insertEdge(this._globalGraphImplicitDependency, String(id));
187 }
188 > const thing = this._getServiceInstanceOrDescriptor(id); instantiationService.ts
189 > if (thing instanceof SyncDescriptor) {
190 return this._safeCreateAndCacheServiceInstance(id, thing, _trace.branch(id, true));
191 > } else { instantiationService.ts
192 > _trace.branch(id, false); instantiationService.ts
193 > return thing;
194 > }
197 > private readonly _activeInstantiations = new Set<ServiceIdentifier<any>>();
198 >
199 >
200 > private _safeCreateAndCacheServiceInstance<T>(id: ServiceIdentifier<T>, desc: SyncDescriptor<T>, _trace: Trace): T {
201 if (this._activeInstantiations.has(id)) {
202 throw new Error(`illegal state - RECURSIVELY instantiating service '${id}'`);
209 }
210 }
212 > private _createAndCacheServiceInstance<T>(id: ServiceIdentifier<T>, desc: SyncDescriptor<T>, _trace: Trace): T {
213
214 type Triple = { id: ServiceIdentifier<any>; desc: SyncDescriptor<any>; _trace: Trace };
279 return <T>this._getServiceInstanceOrDescriptor(id);
280 }
282 > private _createServiceInstanceWithOwner<T>(id: ServiceIdentifier<T>, ctor: any, args: unknown[] = [], supportsDelayedInstantiation: boolean, _trace: Trace): T {
283 if (this._services.get(id) instanceof SyncDescriptor) {
284 return this._createServiceInstance(id, ctor, args, supportsDelayedInstantiation, _trace, this._servicesToMaybeDispose);
289 }
290 }
292 > private _createServiceInstance<T>(id: ServiceIdentifier<T>, ctor: any, args: unknown[] = [], supportsDelayedInstantiation: boolean, _trace: Trace, disposeBucket: Set<any>): T {
293 if (!supportsDelayedInstantiation) {
294 // eager instantiation
384 }
385 }
387 > private _throwIfStrict(msg: string, printWarning: boolean): void {
388 > if (printWarning) { instantiationService.ts
389 console.warn(msg);
390 }
391 > if (this._strict) { instantiationService.ts
392 throw new Error(msg);
393 }
396 >
397 > //#region -- tracing ---
398 >
399 > const enum TraceType {
400 > None = 0,
401 > Creation = 1,
402 > Invocation = 2,
403 > Branch = 3,
404 > }
405 >
406 > export class Trace {
407 >
408 > static all = new Set<string>();
409 >
410 > private static readonly _None = new class extends Trace {
411 > constructor() { super(TraceType.None, null); }
412 > override stop() { }
413 > override branch() { return this; }
414 > };
415 >
416 > static traceInvocation(_enableTracing: boolean, ctor: any): Trace {
417 return !_enableTracing ? Trace._None : new Trace(TraceType.Invocation, ctor.name || new Error().stack!.split('\n').slice(3, 4).join('\n'));
418 }
420 > static traceCreation(_enableTracing: boolean, ctor: any): Trace {
421 > return !_enableTracing ? Trace._None : new Trace(TraceType.Creation, ctor.name); instantiationService.ts
422 > }
424 > private static _totals: number = 0;
425 > private readonly _start: number = Date.now();
426 > private readonly _dep: [ServiceIdentifier<any>, boolean, Trace?][] = [];
427 >
428 > private constructor(
429 > readonly type: TraceType,
430 > readonly name: string | null
431 > ) { }
432 >
433 > branch(id: ServiceIdentifier<any>, first: boolean): Trace {
434 const child = new Trace(TraceType.Branch, id.toString());
435 this._dep.push([id, first, child]);
436 return child;
437 }
439 > stop() {
440 const dur = Date.now() - this._start;
441 Trace._totals += dur;
src/vs/platform/progress/common/progress.ts 170 covered LOC · 9 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- progress.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IAction } from '../../../base/common/actions.js';
7 > import { DeferredPromise } from '../../../base/common/async.js';
8 > import { CancellationToken, CancellationTokenSource } from '../../../base/common/cancellation.js';
9 > import { Disposable, DisposableStore, toDisposable } from '../../../base/common/lifecycle.js';
10 > import { createDecorator } from '../../instantiation/common/instantiation.js';
11 > import { INotificationSource, NotificationPriority } from '../../notification/common/notification.js';
12 >
13 > export const IProgressService = createDecorator<IProgressService>('progressService');
14 >
15 > /**
16 > * A progress service that can be used to report progress to various locations of the UI.
17 > */
18 > export interface IProgressService {
19 >
20 > readonly _serviceBrand: undefined;
21 >
22 > withProgress<R>(
23 > options: IProgressOptions | IProgressDialogOptions | IProgressNotificationOptions | IProgressWindowOptions | IProgressCompositeOptions,
24 > task: (progress: IProgress<IProgressStep>) => Promise<R>,
25 > onDidCancel?: (choice?: number) => void
26 > ): Promise<R>;
27 > }
28 >
29 > export interface IProgressIndicator {
30 >
31 > /**
32 > * Show progress customized with the provided flags.
33 > */
34 > show(infinite: true, delay?: number): IProgressRunner;
35 > show(total: number, delay?: number): IProgressRunner;
36 >
37 > /**
38 > * Indicate progress for the duration of the provided promise. Progress will stop in
39 > * any case of promise completion, error or cancellation.
40 > */
41 > showWhile(promise: Promise<unknown>, delay?: number): Promise<void>;
42 > }
43 >
44 > export const enum ProgressLocation {
45 > Explorer = 1,
46 > Scm = 3,
47 > Extensions = 5,
48 > Window = 10,
49 > Notification = 15,
50 > Dialog = 20
51 > }
52 >
53 > export interface IProgressOptions {
54 > readonly location: ProgressLocation | string;
55 > readonly title?: string;
56 > readonly source?: string | INotificationSource;
57 > readonly total?: number;
58 > readonly cancellable?: boolean | string;
59 > readonly buttons?: string[];
60 > }
61 >
62 > export interface IProgressNotificationOptions extends IProgressOptions {
63 > readonly location: ProgressLocation.Notification;
64 > readonly primaryActions?: readonly IAction[];
65 > readonly secondaryActions?: readonly IAction[];
66 > readonly delay?: number;
67 > readonly priority?: NotificationPriority;
68 > readonly type?: 'loading' | 'syncing';
69 > }
70 >
71 > export interface IProgressDialogOptions extends IProgressOptions {
72 > readonly delay?: number;
73 > readonly detail?: string;
74 > readonly sticky?: boolean;
75 > }
76 >
77 > export interface IProgressWindowOptions extends IProgressOptions {
78 > readonly location: ProgressLocation.Window;
79 > readonly command?: string;
80 > readonly type?: 'loading' | 'syncing';
81 > }
82 >
83 > export interface IProgressCompositeOptions extends IProgressOptions {
84 > readonly location: ProgressLocation.Explorer | ProgressLocation.Extensions | ProgressLocation.Scm | string;
85 > readonly delay?: number;
86 > }
87 >
88 > export interface IProgressStep {
89 > message?: string;
90 > increment?: number;
91 > total?: number;
92 > }
93 >
94 > export interface IProgressRunner {
95 > total(value: number): void;
96 > worked(value: number): void;
97 > done(): void;
98 > }
99 >
100 > export const emptyProgressRunner = Object.freeze<IProgressRunner>({
101 > total() { },
102 > worked() { },
103 > done() { }
104 > });
105 >
106 > export interface IProgress<T> {
107 > report(item: T): void;
108 > }
109 >
110 > export class Progress<T> implements IProgress<T> {
111 >
112 > static readonly None = Object.freeze<IProgress<unknown>>({ report() { } });
113 >
114 > private _value?: T;
115 > get value(): T | undefined { return this._value; }
116 >
117 > constructor(private callback: (data: T) => unknown) {
118 }
119 > progress.ts
120 > report(item: T) {
121 this._value = item;
122 this.callback(this._value);
123 }
124 > } progress.ts
125 >
126 > /**
127 > * A helper to show progress during a long running operation. If the operation
128 > * is started multiple times, only the last invocation will drive the progress.
129 > */
130 > export interface IOperation {
131 > id: number;
132 > isCurrent: () => boolean;
133 > token: CancellationToken;
134 > stop(): void;
135 > }
136 >
137 > /**
138 > * RAII-style progress instance that allows imperative reporting and hides
139 > * once `dispose()` is called.
140 > */
141 > export class UnmanagedProgress extends Disposable {
142 > private readonly deferred = new DeferredPromise<void>();
143 > private reporter?: IProgress<IProgressStep>;
144 > private lastStep?: IProgressStep;
145 >
146 > constructor(
147 options: IProgressOptions | IProgressDialogOptions | IProgressNotificationOptions | IProgressWindowOptions | IProgressCompositeOptions,
148 @IProgressService progressService: IProgressService,
160 this._register(toDisposable(() => this.deferred.complete()));
161 }
162 > progress.ts
163 > report(step: IProgressStep) {
164 if (this.reporter) {
165 this.reporter.report(step);
168 }
169 }
170 > } progress.ts
171 >
172 > export class LongRunningOperation extends Disposable {
173 > private currentOperationId = 0;
174 > private readonly currentOperationDisposables = this._register(new DisposableStore());
175 > private currentProgressRunner: IProgressRunner | undefined;
176 > private currentProgressTimeout: Timeout | undefined = undefined;
177 >
178 > constructor(
179 private progressIndicator: IProgressIndicator
180 ) {
181 super();
182 }
183 > progress.ts
184 > start(progressDelay: number): IOperation {
185
186 // Stop any previous operation
207 };
208 }
209 > progress.ts
210 > stop(): void {
211 this.doStop(this.currentOperationId);
212 }
213 > progress.ts
214 > private doStop(operationId: number): void {
215 if (this.currentOperationId === operationId) {
216 this.currentOperationDisposables.clear();
217 }
218 }
219 > } progress.ts
220 >
221 > export const IEditorProgressService = createDecorator<IEditorProgressService>('editorProgressService');
222 >
223 > /**
224 > * A progress service that will report progress local to the editor triggered from.
225 > */
226 > export interface IEditorProgressService extends IProgressIndicator {
227 >
228 > readonly _serviceBrand: undefined;
229 > }
src/vs/workbench/contrib/chat/common/requestParser/chatParserTypes.ts 169 covered LOC · 26 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatParserTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { revive } from '../../../../../base/common/marshalling.js';
7 > import { ThemeIcon } from '../../../../../base/common/themables.js';
8 > import { IOffsetRange, OffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
9 > import { IRange, Range } from '../../../../../editor/common/core/range.js';
10 > import { IChatAgentCommand, IChatAgentData, IChatAgentService, reviveSerializedAgent } from '../participants/chatAgents.js';
11 > import { IChatSlashData } from '../participants/chatSlashCommands.js';
12 > import { IChatRequestProblemsVariable, IChatRequestVariableValue } from '../attachments/chatVariables.js';
13 > import { ChatAgentLocation } from '../constants.js';
14 > import { IToolData } from '../tools/languageModelToolsService.js';
15 > import { IChatRequestToolEntry, IChatRequestToolSetEntry, IChatRequestVariableEntry, IDiagnosticVariableEntryFilterData } from '../attachments/chatVariableEntries.js';
16 > import { arrayEquals } from '../../../../../base/common/equals.js';
17 >
18 > // These are in a separate file to avoid circular dependencies with the dependencies of the parser
19 >
20 > export interface IParsedChatRequest {
21 > readonly parts: ReadonlyArray<IParsedChatRequestPart>;
22 > readonly text: string;
23 > }
24 >
25 > export namespace IParsedChatRequest {
26 > export function equals(a: IParsedChatRequest, b: IParsedChatRequest): boolean {
27 return a.text === b.text && arrayEquals(a.parts, b.parts, (p1, p2) =>
28 p1.kind === p2.kind &&
32 );
33 }
35 >
36 > export interface IParsedChatRequestPart {
37 > readonly kind: string; // for serialization
38 > readonly range: IOffsetRange;
39 > readonly editorRange: IRange;
40 > readonly text: string;
41 > /** How this part is represented in the prompt going to the agent */
42 > readonly promptText: string;
43 > }
44 >
45 > export function getPromptText(request: IParsedChatRequest): { message: string; diff: number } {
46 const message = request.parts.map(r => r.promptText).join('').trimStart();
47 const diff = request.text.length - message.length;
49 return { message, diff };
50 }
52 > export class ChatRequestTextPart implements IParsedChatRequestPart {
53 > static readonly Kind = 'text';
54 > readonly kind = ChatRequestTextPart.Kind;
55 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly text: string) { }
56 >
57 > get promptText(): string {
58 return this.text;
59 }
61 >
62 > // warning, these also show up in a regex in the parser
63 > export const chatVariableLeader = '#';
64 > export const chatAgentLeader = '@';
65 > export const chatSubcommandLeader = '/';
66 >
67 > /**
68 > * An invocation of a static variable that can be resolved by the variable service
69 > * @deprecated, but kept for backwards compatibility with old persisted chat requests
70 > */
71 > class ChatRequestVariablePart implements IParsedChatRequestPart {
72 > static readonly Kind = 'var';
73 > readonly kind = ChatRequestVariablePart.Kind;
74 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly variableName: string, readonly variableArg: string, readonly variableId: string) { }
75 >
76 > get text(): string {
77 const argPart = this.variableArg ? `:${this.variableArg}` : '';
78 return `${chatVariableLeader}${this.variableName}${argPart}`;
79 }
81 > get promptText(): string {
82 return this.text;
83 }
85 >
86 > /**
87 > * An invocation of a tool
88 > */
89 > export class ChatRequestToolPart implements IParsedChatRequestPart {
90 > static readonly Kind = 'tool';
91 > readonly kind = ChatRequestToolPart.Kind;
92 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly toolName: string, readonly toolId: string, readonly displayName?: string, readonly icon?: IToolData['icon']) { }
93 >
94 > get text(): string {
95 return `${chatVariableLeader}${this.toolName}`;
96 }
98 > get promptText(): string {
99 return this.text;
100 }
102 > toVariableEntry(): IChatRequestToolEntry {
103 return { kind: 'tool', id: this.toolId, name: this.toolName, range: this.range, value: undefined, icon: ThemeIcon.isThemeIcon(this.icon) ? this.icon : undefined, fullName: this.displayName };
104 }
106 >
107 > /**
108 > * An invocation of a tool
109 > */
110 > export class ChatRequestToolSetPart implements IParsedChatRequestPart {
111 > static readonly Kind = 'toolset';
112 > readonly kind = ChatRequestToolSetPart.Kind;
113 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly id: string, readonly name: string, readonly icon: ThemeIcon, readonly tools: IChatRequestToolEntry[]) { }
114 >
115 > get text(): string {
116 return `${chatVariableLeader}${this.name}`;
117 }
119 > get promptText(): string {
120 return this.text;
121 }
123 > toVariableEntry(): IChatRequestToolSetEntry {
124 return { kind: 'toolset', id: this.id, name: this.name, range: this.range, icon: this.icon, value: this.tools };
125 }
127 >
128 > /**
129 > * An invocation of an agent that can be resolved by the agent service
130 > */
131 > export class ChatRequestAgentPart implements IParsedChatRequestPart {
132 > static readonly Kind = 'agent';
133 > readonly kind = ChatRequestAgentPart.Kind;
134 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly agent: IChatAgentData) { }
135 >
136 > get text(): string {
137
138 return `${chatAgentLeader}${this.agent.name}`;
139 }
141 > get promptText(): string {
142 return '';
143 }
145 >
146 > /**
147 > * An invocation of an agent's subcommand
148 > */
149 > export class ChatRequestAgentSubcommandPart implements IParsedChatRequestPart {
150 > static readonly Kind = 'subcommand';
151 > readonly kind = ChatRequestAgentSubcommandPart.Kind;
152 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly command: IChatAgentCommand) { }
153 >
154 > get text(): string {
155 return `${chatSubcommandLeader}${this.command.name}`;
156 }
158 > get promptText(): string {
159 return '';
160 }
162 >
163 > /**
164 > * An invocation of a standalone slash command
165 > */
166 > export class ChatRequestSlashCommandPart implements IParsedChatRequestPart {
167 > static readonly Kind = 'slash';
168 > readonly kind = ChatRequestSlashCommandPart.Kind;
169 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly slashCommand: IChatSlashData) { }
170 >
171 > get text(): string {
172 return `${chatSubcommandLeader}${this.slashCommand.command}`;
173 }
175 > get promptText(): string {
176 return `${chatSubcommandLeader}${this.slashCommand.command}`;
177 }
179 >
180 > /**
181 > * An invocation of a standalone slash command
182 > */
183 > export class ChatRequestSlashPromptPart implements IParsedChatRequestPart {
184 > static readonly Kind = 'prompt';
185 > readonly kind = ChatRequestSlashPromptPart.Kind;
186 > readonly displayText?: string;
187 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly name: string, displayText?: string) {
188 // Only set the own property when provided so the canonical
189 // form (no `displayText` field) stays compatible with snapshots
193 }
194 }
196 > get text(): string {
197 return this.displayText ?? `${chatSubcommandLeader}${this.name}`;
198 }
200 > get promptText(): string {
201 return this.displayText ?? `${chatSubcommandLeader}${this.name}`;
202 }
204 >
205 > /**
206 > * An invocation of a dynamic reference like '#file:'
207 > */
208 > export class ChatRequestDynamicVariablePart implements IParsedChatRequestPart {
209 > static readonly Kind = 'dynamic';
210 > readonly kind = ChatRequestDynamicVariablePart.Kind;
211 > constructor(readonly range: OffsetRange, readonly editorRange: IRange, readonly text: string, readonly id: string, readonly modelDescription: string | undefined, readonly data: IChatRequestVariableValue, readonly fullName?: string, readonly icon?: ThemeIcon, readonly isFile?: boolean, readonly isDirectory?: boolean, readonly _meta?: Record<string, unknown>, readonly isAttachmentReference?: boolean) { }
212 >
213 > get referenceText(): string {
214 return this.text.replace(chatVariableLeader, '');
215 }
217 > get promptText(): string {
218 return this.text;
219 }
221 > toVariableEntry(): IChatRequestVariableEntry {
222 if (this.id === 'vscode.problems') {
223 return IDiagnosticVariableEntryFilterData.toEntry((this.data as IChatRequestProblemsVariable).filter);
226 return { kind: this.isDirectory ? 'directory' : this.isFile ? 'file' : 'generic', id: this.id, name: this.referenceText, range: this.range, value: this.data, fullName: this.fullName, icon: this.icon, _meta: this._meta };
227 }
229 >
230 > export function reviveParsedChatRequest(serialized: IParsedChatRequest): IParsedChatRequest {
231 return {
232 text: serialized.text,
313 };
314 }
316 > export function extractAgentAndCommand(parsed: IParsedChatRequest): { agentPart: ChatRequestAgentPart | undefined; commandPart: ChatRequestAgentSubcommandPart | undefined } {
317 const agentPart = parsed.parts.find((r): r is ChatRequestAgentPart => r instanceof ChatRequestAgentPart);
318 const commandPart = parsed.parts.find((r): r is ChatRequestAgentSubcommandPart => r instanceof ChatRequestAgentSubcommandPart);
319 return { agentPart, commandPart };
320 }
322 > export function formatChatQuestion(chatAgentService: IChatAgentService, location: ChatAgentLocation, prompt: string, participant: string | null = null, command: string | null = null): string | undefined {
323 let question = '';
324 if (participant && participant !== chatAgentService.getDefaultAgent(location)?.id) {
src/vs/editor/common/languages/language.ts 168 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- language.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { IDisposable } from '../../../base/common/lifecycle.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { ILanguageIdCodec } from '../languages.js';
10 > import { createDecorator } from '../../../platform/instantiation/common/instantiation.js';
11 >
12 > export const ILanguageService = createDecorator<ILanguageService>('languageService');
13 >
14 > export interface ILanguageExtensionPoint {
15 > id: string;
16 > extensions?: string[];
17 > filenames?: string[];
18 > filenamePatterns?: string[];
19 > firstLine?: string;
20 > aliases?: string[];
21 > mimetypes?: string[];
22 > configuration?: URI;
23 > /**
24 > * @internal
25 > */
26 > icon?: ILanguageIcon;
27 > }
28 >
29 > export interface ILanguageSelection {
30 > readonly languageId: string;
31 > readonly onDidChange: Event<string>;
32 > }
33 >
34 > export interface ILanguageNameIdPair {
35 > readonly languageName: string;
36 > readonly languageId: string;
37 > }
38 >
39 > export interface ILanguageIcon {
40 > readonly light: URI;
41 > readonly dark: URI;
42 > }
43 >
44 > export interface ILanguageService {
45 > readonly _serviceBrand: undefined;
46 >
47 > /**
48 > * A codec which can encode and decode a string `languageId` as a number.
49 > */
50 > readonly languageIdCodec: ILanguageIdCodec;
51 >
52 > /**
53 > * An event emitted when basic language features are requested for the first time.
54 > * This event is emitted when embedded languages are encountered (e.g. JS code block inside Markdown)
55 > * or when a language is associated to a text model.
56 > *
57 > * **Note**: Basic language features refers to language configuration related features.
58 > * **Note**: This event is a superset of `onDidRequestRichLanguageFeatures`
59 > */
60 > readonly onDidRequestBasicLanguageFeatures: Event<string>;
61 >
62 > /**
63 > * An event emitted when rich language features are requested for the first time.
64 > * This event is emitted when a language is associated to a text model.
65 > *
66 > * **Note**: Rich language features refers to tokenizers, language features based on providers, etc.
67 > * **Note**: This event is a subset of `onDidRequestRichLanguageFeatures`
68 > */
69 > readonly onDidRequestRichLanguageFeatures: Event<string>;
70 >
71 > /**
72 > * An event emitted when languages have changed.
73 > */
74 > readonly onDidChange: Event<void>;
75 >
76 > /**
77 > * Register a language.
78 > */
79 > registerLanguage(def: ILanguageExtensionPoint): IDisposable;
80 >
81 > /**
82 > * Check if `languageId` is registered.
83 > */
84 > isRegisteredLanguageId(languageId: string): boolean;
85 >
86 > /**
87 > * Get a list of all registered languages.
88 > */
89 > getRegisteredLanguageIds(): string[];
90 >
91 > /**
92 > * Get a list of all registered languages with a name.
93 > * If a language is explicitly registered without a name, it will not be part of the result.
94 > * The result is sorted using by name case insensitive.
95 > */
96 > getSortedRegisteredLanguageNames(): ILanguageNameIdPair[];
97 >
98 > /**
99 > * Get the preferred language name for a language.
100 > */
101 > getLanguageName(languageId: string): string | null;
102 >
103 > /**
104 > * Get the mimetype for a language.
105 > */
106 > getMimeType(languageId: string): string | null;
107 >
108 > /**
109 > * Get the default icon for the language.
110 > */
111 > getIcon(languageId: string): ILanguageIcon | null;
112 >
113 > /**
114 > * Get all file extensions for a language.
115 > */
116 > getExtensions(languageId: string): ReadonlyArray<string>;
117 >
118 > /**
119 > * Get all file names for a language.
120 > */
121 > getFilenames(languageId: string): ReadonlyArray<string>;
122 >
123 > /**
124 > * Get all language configuration files for a language.
125 > */
126 > getConfigurationFiles(languageId: string): ReadonlyArray<URI>;
127 >
128 > /**
129 > * Look up a language by its name case insensitive.
130 > */
131 > getLanguageIdByLanguageName(languageName: string): string | null;
132 >
133 > /**
134 > * Look up a language by its mime type.
135 > */
136 > getLanguageIdByMimeType(mimeType: string | null | undefined): string | null;
137 >
138 > /**
139 > * Guess the language id for a resource.
140 > */
141 > guessLanguageIdByFilepathOrFirstLine(resource: URI, firstLine?: string): string | null;
142 >
143 > /**
144 > * Will fall back to 'plaintext' if `languageId` is unknown.
145 > */
146 > createById(languageId: string | null | undefined): ILanguageSelection;
147 >
148 > /**
149 > * Will fall back to 'plaintext' if `mimeType` is unknown.
150 > */
151 > createByMimeType(mimeType: string | null | undefined): ILanguageSelection;
152 >
153 > /**
154 > * Will fall back to 'plaintext' if the `languageId` cannot be determined.
155 > */
156 > createByFilepathOrFirstLine(resource: URI | null, firstLine?: string): ILanguageSelection;
157 >
158 > /**
159 > * Request basic language features for a language.
160 > */
161 > requestBasicLanguageFeatures(languageId: string): void;
162 >
163 > /**
164 > * Request rich language features for a language.
165 > */
166 > requestRichLanguageFeatures(languageId: string): void;
167 >
168 > }
src/vs/base/common/glob.ts 167 covered LOC · 24 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- glob.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { equals } from './arrays.js';
7 > import { isThenable } from './async.js';
8 > import { CharCode } from './charCode.js';
9 > import { isEqualOrParent } from './extpath.js';
10 > import { LRUCache } from './map.js';
11 > import { basename, extname, posix, sep } from './path.js';
12 > import { isLinux } from './platform.js';
13 > import { endsWithIgnoreCase, equalsIgnoreCase, escapeRegExpCharacters, ltrim } from './strings.js';
14 >
15 > export interface IRelativePattern {
16 >
17 > /**
18 > * A base file path to which this pattern will be matched against relatively.
19 > */
20 > readonly base: string;
21 >
22 > /**
23 > * A file glob pattern like `*.{ts,js}` that will be matched on file paths
24 > * relative to the base path.
25 > *
26 > * Example: Given a base of `/home/work/folder` and a file path of `/home/work/folder/index.js`,
27 > * the file glob pattern will match on `index.js`.
28 > */
29 > readonly pattern: string;
30 > }
31 >
32 > export interface IExpression {
33 > [pattern: string]: boolean | SiblingClause;
34 > }
35 >
36 > export function getEmptyExpression(): IExpression {
37 return Object.create(null);
38 }
39 > glob.ts
40 > interface SiblingClause {
41 > when: string;
42 > }
43 >
44 > export const GLOBSTAR = '**';
45 > export const GLOB_SPLIT = '/';
46 >
47 > const PATH_REGEX = '[/\\\\]'; // any slash or backslash
48 > const NO_PATH_REGEX = '[^/\\\\]'; // any non-slash and non-backslash
49 > const ALL_FORWARD_SLASHES = /\//g;
50 >
51 function starsToRegExp(starCount: number, isLastPattern?: boolean): string {
52 switch (starCount) {
63 }
64 }
65 > glob.ts
66 > export function splitGlobAware(pattern: string, splitChar: string): string[] {
67 if (!pattern) {
68 return [];
109 return segments;
110 }
111 > glob.ts
112 function parseRegExp(pattern: string): string {
113 if (!pattern) {
256 return regEx;
257 }
258 > glob.ts
259 > // regexes to check for trivial glob patterns that just check for String#endsWith
260 > const T1 = /^\*\*\/\*\.[\w\.-]+$/; // **/*.something
261 > const T2 = /^\*\*\/([\w\.-]+)\/?$/; // **/something
262 > const T3 = /^{\*\*\/\*?[\w\.-]+\/?(,\*\*\/\*?[\w\.-]+\/?)*}$/; // {**/*.something,**/*.else} or {**/package.json,**/project.json}
263 > const T3_2 = /^{\*\*\/\*?[\w\.-]+(\/(\*\*)?)?(,\*\*\/\*?[\w\.-]+(\/(\*\*)?)?)*}$/; // Like T3, with optional trailing /**
264 > const T4 = /^\*\*((\/[\w\.-]+)+)\/?$/; // **/something/else
265 > const T5 = /^([\w\.-]+(\/[\w\.-]+)*)\/?$/; // something/else
266 >
267 > export type ParsedPattern = (path: string, basename?: string) => boolean;
268 >
269 > // The `ParsedExpression` returns a `Promise`
270 > // iff `hasSibling` returns a `Promise`.
271 > export type ParsedExpression = (path: string, basename?: string, hasSibling?: (name: string) => boolean | Promise<boolean>) => string | null | Promise<string | null> /* the matching pattern */;
272 >
273 > export interface IGlobOptions {
274 >
275 > /**
276 > * Simplify patterns for use as exclusion filters during
277 > * tree traversal to skip entire subtrees. Cannot be used
278 > * outside of a tree traversal.
279 > */
280 > trimForExclusions?: boolean;
281 >
282 > /**
283 > * Whether glob pattern matching should be case insensitive.
284 > */
285 > ignoreCase?: boolean;
286 > }
287 >
288 > interface IGlobOptionsInternal extends IGlobOptions {
289 > equals: (a: string, b: string) => boolean;
290 > endsWith: (str: string, candidate: string) => boolean;
291 > isEqualOrParent: (base: string, candidate: string) => boolean;
292 > }
293 >
294 > interface ParsedStringPattern {
295 > (path: string, basename?: string): string | null | Promise<string | null> /* the matching pattern */;
296 > basenames?: string[];
297 > patterns?: string[];
298 > allBasenames?: string[];
299 > allPaths?: string[];
300 > }
301 >
302 > interface ParsedExpressionPattern {
303 > (path: string, basename?: string, name?: string, hasSibling?: (name: string) => boolean | Promise<boolean>): string | null | Promise<string | null> /* the matching pattern */;
304 > requiresSiblings?: boolean;
305 > allBasenames?: string[];
306 > allPaths?: string[];
307 > }
308 >
309 > const CACHE = new LRUCache<string, ParsedStringPattern>(10000); // bounded to 10000 elements
310 >
311 > const FALSE = function () {
312 return false;
313 };
314 > glob.ts
315 > const NULL = function (): string | null {
316 return null;
317 };
318 > glob.ts
319 > /**
320 > * Check if a provided parsed pattern or expression
321 > * is empty - hence it won't ever match anything.
322 > *
323 > * See {@link FALSE} and {@link NULL}.
324 > */
325 > export function isEmptyPattern(pattern: ParsedPattern | ParsedExpression): pattern is (typeof FALSE | typeof NULL) {
326 if (pattern === FALSE) {
327 return true;
334 return false;
335 }
336 > glob.ts
337 function parsePattern(arg1: string | IRelativePattern, options: IGlobOptions): ParsedStringPattern {
338 if (!arg1) {
390 return wrapRelativePattern(parsedPattern, arg1, internalOptions);
391 }
392 > glob.ts
393 function wrapRelativePattern(parsedPattern: ParsedStringPattern, arg2: string | IRelativePattern, options: IGlobOptionsInternal): ParsedStringPattern {
394 if (typeof arg2 === 'string') {
421 return wrappedPattern;
422 }
423 > glob.ts
424 function trimForExclusions(pattern: string, options: IGlobOptions): string {
425 return options.trimForExclusions && pattern.endsWith('/**') ? pattern.substring(0, pattern.length - 2) : pattern; // dropping **, tailing / is dropped later
426 }
427 > glob.ts
428 > // common pattern: **/*.txt just need endsWith check
429 function trivia1(base: string, pattern: string, options: IGlobOptionsInternal): ParsedStringPattern {
430 return function (path: string, basename?: string) {
432 };
433 }
434 > glob.ts
435 > // common pattern: **/some.txt just need basename check
436 function trivia2(base: string, pattern: string, options: IGlobOptionsInternal): ParsedStringPattern {
437 const slashBase = `/${base}`;
457 return parsedPattern;
458 }
459 > glob.ts
460 > // repetition of common patterns (see above) {**/*.txt,**/*.png}
461 function trivia3(pattern: string, options: IGlobOptionsInternal): ParsedStringPattern {
462 const parsedPatterns = aggregateBasenameMatches(pattern.slice(1, -1)
496 return parsedPattern;
497 }
498 > glob.ts
499 > // common patterns: **/something/else just need endsWith check, something/else just needs and equals check
500 function trivia4and5(targetPath: string, pattern: string, matchPathEnds: boolean, options: IGlobOptionsInternal): ParsedStringPattern {
501 const usingPosixSep = sep === posix.sep;
522 return parsedPattern;
523 }
524 > glob.ts
525 function toRegExp(pattern: string, options: IGlobOptions): ParsedStringPattern {
526 try {
535 }
536 }
537 > glob.ts
538 > /**
539 > * Simplified glob matching. Supports a subset of glob patterns:
540 > * * `*` to match zero or more characters in a path segment
541 > * * `?` to match on one character in a path segment
542 > * * `**` to match any number of path segments, including none
543 > * * `{}` to group conditions (e.g. *.{ts,js} matches all TypeScript and JavaScript files)
544 > * * `[]` to declare a range of characters to match in a path segment (e.g., `example.[0-9]` to match on `example.0`, `example.1`, …)
545 > * * `[!...]` to negate a range of characters to match in a path segment (e.g., `example.[!0-9]` to match on `example.a`, `example.b`, but not `example.0`)
546 > */
547 > export function match(pattern: string | IRelativePattern, path: string, options?: IGlobOptions): boolean;
548 > export function match(expression: IExpression, path: string, options?: IGlobOptions): boolean;
549 > export function match(arg1: string | IExpression | IRelativePattern, path: string, options?: IGlobOptions): boolean {
550 if (!arg1 || typeof path !== 'string') {
551 return false;
554 return parse(arg1, options)(path) as boolean;
555 }
556 > glob.ts
557 > /**
558 > * Simplified glob matching. Supports a subset of glob patterns:
559 > * * `*` to match zero or more characters in a path segment
560 > * * `?` to match on one character in a path segment
561 > * * `**` to match any number of path segments, including none
562 > * * `{}` to group conditions (e.g. *.{ts,js} matches all TypeScript and JavaScript files)
563 > * * `[]` to declare a range of characters to match in a path segment (e.g., `example.[0-9]` to match on `example.0`, `example.1`, …)
564 > * * `[!...]` to negate a range of characters to match in a path segment (e.g., `example.[!0-9]` to match on `example.a`, `example.b`, but not `example.0`)
565 > */
566 > export function parse(pattern: string | IRelativePattern, options?: IGlobOptions): ParsedPattern;
567 > export function parse(expression: IExpression, options?: IGlobOptions): ParsedExpression;
568 > export function parse(arg1: string | IExpression | IRelativePattern, options?: IGlobOptions): ParsedPattern | ParsedExpression;
569 > export function parse(arg1: string | IExpression | IRelativePattern, options: IGlobOptions = {}): ParsedPattern | ParsedExpression {
570 if (!arg1) {
571 return FALSE;
597 return parsedExpression(arg1, options);
598 }
599 > glob.ts
600 > export function isRelativePattern(obj: unknown): obj is IRelativePattern {
601 const rp = obj as IRelativePattern | undefined | null;
602 if (!rp) {
606 return typeof rp.base === 'string' && typeof rp.pattern === 'string';
607 }
608 > glob.ts
609 > export function getBasenameTerms(patternOrExpression: ParsedPattern | ParsedExpression): string[] {
610 return (<ParsedStringPattern>patternOrExpression).allBasenames || [];
611 }
612 > glob.ts
613 > export function getPathTerms(patternOrExpression: ParsedPattern | ParsedExpression): string[] {
614 return (<ParsedStringPattern>patternOrExpression).allPaths || [];
615 }
616 > glob.ts
617 function parsedExpression(expression: IExpression, options: IGlobOptions): ParsedExpression {
618 const parsedPatterns = aggregateBasenameMatches(Object.getOwnPropertyNames(expression)
745 return resultExpression;
746 }
747 > glob.ts
748 function parseExpressionPattern(pattern: string, value: boolean | SiblingClause, options: IGlobOptions): (ParsedStringPattern | ParsedExpressionPattern) {
749 if (value === false) {
786 return parsedPattern;
787 }
788 > glob.ts
789 function aggregateBasenameMatches(parsedPatterns: Array<ParsedStringPattern | ParsedExpressionPattern>, result?: string): Array<ParsedStringPattern | ParsedExpressionPattern> {
790 const basenamePatterns = parsedPatterns.filter(parsedPattern => !!(<ParsedStringPattern>parsedPattern).basenames);
844 return aggregatedPatterns;
845 }
846 > glob.ts
847 > // NOTE: This is not used for actual matching, only for resetting watcher when patterns change.
848 > // That is why it's ok to avoid case-insensitive comparison here.
849 > export function patternsEquals(patternsA: Array<string | IRelativePattern> | undefined, patternsB: Array<string | IRelativePattern> | undefined): boolean {
850 return equals(patternsA, patternsB, (a, b) => {
851 if (typeof a === 'string' && typeof b === 'string') {
src/vs/base/common/buffer.ts 166 covered LOC · 45 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- buffer.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Lazy } from './lazy.js';
7 > import * as streams from './stream.js';
8 >
9 > interface NodeBuffer {
10 > allocUnsafe(size: number): Uint8Array;
11 > isBuffer(obj: unknown): obj is NodeBuffer;
12 > from(arrayBuffer: ArrayBufferLike, byteOffset?: number, length?: number): Uint8Array;
13 > from(data: string): Uint8Array;
14 > }
15 >
16 > declare const Buffer: NodeBuffer;
17 >
18 > const hasBuffer = (typeof Buffer !== 'undefined');
19 > const indexOfTable = new Lazy(() => new Uint8Array(256));
20 >
21 > let textEncoder: { encode: (input: string) => Uint8Array } | null;
22 > let textDecoder: { decode: (input: Uint8Array) => string } | null;
23 >
24 > export class VSBuffer {
25 >
26 > /**
27 > * When running in a nodejs context, the backing store for the returned `VSBuffer` instance
28 > * might use a nodejs Buffer allocated from node's Buffer pool, which is not transferrable.
29 > */
30 > static alloc(byteLength: number): VSBuffer {
31 if (hasBuffer) {
32 return new VSBuffer(Buffer.allocUnsafe(byteLength));
35 }
36 }
37 > buffer.ts
38 > /**
39 > * When running in a nodejs context, if `actual` is not a nodejs Buffer, the backing store for
40 > * the returned `VSBuffer` instance might use a nodejs Buffer allocated from node's Buffer pool,
41 > * which is not transferrable.
42 > */
43 > static wrap(actual: Uint8Array): VSBuffer {
44 if (hasBuffer && !(Buffer.isBuffer(actual))) {
45 // https://nodejs.org/dist/latest-v10.x/docs/api/buffer.html#buffer_class_method_buffer_from_arraybuffer_byteoffset_length
49 return new VSBuffer(actual);
50 }
51 > buffer.ts
52 > /**
53 > * When running in a nodejs context, the backing store for the returned `VSBuffer` instance
54 > * might use a nodejs Buffer allocated from node's Buffer pool, which is not transferrable.
55 > */
56 > static fromString(source: string, options?: { dontUseNodeBuffer?: boolean }): VSBuffer {
57 > const dontUseNodeBuffer = options?.dontUseNodeBuffer || false; buffer.ts
58 > if (!dontUseNodeBuffer && hasBuffer) {
59 > return new VSBuffer(Buffer.from(source));
60 > } else {
61 if (!textEncoder) {
62 textEncoder = new TextEncoder();
64 return new VSBuffer(textEncoder.encode(source));
65 }
66 > } buffer.ts
67 > buffer.ts
68 > /**
69 > * When running in a nodejs context, the backing store for the returned `VSBuffer` instance
70 > * might use a nodejs Buffer allocated from node's Buffer pool, which is not transferrable.
71 > */
72 > static fromByteArray(source: number[]): VSBuffer {
73 const result = VSBuffer.alloc(source.length);
74 for (let i = 0, len = source.length; i < len; i++) {
77 return result;
78 }
79 > buffer.ts
80 > /**
81 > * When running in a nodejs context, the backing store for the returned `VSBuffer` instance
82 > * might use a nodejs Buffer allocated from node's Buffer pool, which is not transferrable.
83 > */
84 > static concat(buffers: VSBuffer[], totalLength?: number): VSBuffer {
85 if (typeof totalLength === 'undefined') {
86 totalLength = 0;
100 return ret;
101 }
102 > buffer.ts
103 > static isNativeBuffer(buffer: unknown): boolean {
104 return hasBuffer && Buffer.isBuffer(buffer);
105 }
106 > buffer.ts
107 > readonly buffer: Uint8Array;
108 > readonly byteLength: number;
109 >
110 > private constructor(buffer: Uint8Array) {
111 > this.buffer = buffer; buffer.ts
112 > this.byteLength = this.buffer.byteLength;
113 > }
114 > buffer.ts
115 > /**
116 > * When running in a nodejs context, the backing store for the returned `VSBuffer` instance
117 > * might use a nodejs Buffer allocated from node's Buffer pool, which is not transferrable.
118 > */
119 > clone(): VSBuffer {
120 const result = VSBuffer.alloc(this.byteLength);
121 result.set(this);
122 return result;
123 }
124 > buffer.ts
125 > toString(): string {
126 if (hasBuffer) {
127 return this.buffer.toString();
133 }
134 }
135 > buffer.ts
136 > slice(start?: number, end?: number): VSBuffer {
137 // IMPORTANT: use subarray instead of slice because TypedArray#slice
138 // creates shallow copy and NodeBuffer#slice doesn't. The use of subarray
140 return new VSBuffer(this.buffer.subarray(start, end));
141 }
142 > buffer.ts
143 > set(array: VSBuffer, offset?: number): void;
144 > set(array: Uint8Array, offset?: number): void;
145 > set(array: ArrayBuffer, offset?: number): void;
146 > set(array: ArrayBufferView, offset?: number): void;
147 > set(array: VSBuffer | Uint8Array | ArrayBuffer | ArrayBufferView, offset?: number): void;
148 > set(array: VSBuffer | Uint8Array | ArrayBuffer | ArrayBufferView, offset?: number): void {
149 if (array instanceof VSBuffer) {
150 this.buffer.set(array.buffer, offset);
159 }
160 }
161 > buffer.ts
162 > readUInt32BE(offset: number): number {
163 return readUInt32BE(this.buffer, offset);
164 }
165 > buffer.ts
166 > writeUInt32BE(value: number, offset: number): void {
167 writeUInt32BE(this.buffer, value, offset);
168 }
169 > buffer.ts
170 > readUInt32LE(offset: number): number {
171 return readUInt32LE(this.buffer, offset);
172 }
173 > buffer.ts
174 > writeUInt32LE(value: number, offset: number): void {
175 writeUInt32LE(this.buffer, value, offset);
176 }
177 > buffer.ts
178 > readUInt8(offset: number): number {
179 return readUInt8(this.buffer, offset);
180 }
181 > buffer.ts
182 > writeUInt8(value: number, offset: number): void {
183 writeUInt8(this.buffer, value, offset);
184 }
185 > buffer.ts
186 > indexOf(subarray: VSBuffer | Uint8Array, offset = 0) {
187 return binaryIndexOf(this.buffer, subarray instanceof VSBuffer ? subarray.buffer : subarray, offset);
188 }
189 > buffer.ts
190 > equals(other: VSBuffer): boolean {
191 if (this === other) {
192 return true;
199 return this.buffer.every((value, index) => value === other.buffer[index]);
200 }
201 > } buffer.ts
202 >
203 > /**
204 > * Like String.indexOf, but works on Uint8Arrays.
205 > * Uses the boyer-moore-horspool algorithm to be reasonably speedy.
206 > */
207 > export function binaryIndexOf(haystack: Uint8Array, needle: Uint8Array, offset = 0): number {
208 const needleLen = needle.byteLength;
209 const haystackLen = haystack.byteLength;
248 return result;
249 }
250 > buffer.ts
251 > export function readUInt16LE(source: Uint8Array, offset: number): number {
252 return (
253 ((source[offset + 0] << 0) >>> 0) |
255 );
256 }
257 > buffer.ts
258 > export function writeUInt16LE(destination: Uint8Array, value: number, offset: number): void {
259 destination[offset + 0] = (value & 0b11111111);
260 value = value >>> 8;
261 destination[offset + 1] = (value & 0b11111111);
262 }
263 > buffer.ts
264 > export function readUInt32BE(source: Uint8Array, offset: number): number {
265 return (
266 source[offset] * 2 ** 24
270 );
271 }
272 > buffer.ts
273 > export function writeUInt32BE(destination: Uint8Array, value: number, offset: number): void {
274 destination[offset + 3] = value;
275 value = value >>> 8;
280 destination[offset] = value;
281 }
282 > buffer.ts
283 > export function readUInt32LE(source: Uint8Array, offset: number): number {
284 return (
285 ((source[offset + 0] << 0) >>> 0) |
289 );
290 }
291 > buffer.ts
292 > export function writeUInt32LE(destination: Uint8Array, value: number, offset: number): void {
293 destination[offset + 0] = (value & 0b11111111);
294 value = value >>> 8;
299 destination[offset + 3] = (value & 0b11111111);
300 }
301 > buffer.ts
302 > export function readUInt8(source: Uint8Array, offset: number): number {
303 return source[offset];
304 }
305 > buffer.ts
306 > export function writeUInt8(destination: Uint8Array, value: number, offset: number): void {
307 destination[offset] = value;
308 }
309 > buffer.ts
310 > export interface VSBufferReadable extends streams.Readable<VSBuffer> { }
311 >
312 > export interface VSBufferReadableStream extends streams.ReadableStream<VSBuffer> { }
313 >
314 > export interface VSBufferWriteableStream extends streams.WriteableStream<VSBuffer> { }
315 >
316 > export interface VSBufferReadableBufferedStream extends streams.ReadableBufferedStream<VSBuffer> { }
317 >
318 > export function readableToBuffer(readable: VSBufferReadable): VSBuffer {
319 return streams.consumeReadable<VSBuffer>(readable, chunks => VSBuffer.concat(chunks));
320 }
321 > buffer.ts
322 > export function bufferToReadable(buffer: VSBuffer): VSBufferReadable {
323 return streams.toReadable<VSBuffer>(buffer);
324 }
325 > buffer.ts
326 > export function streamToBuffer(stream: streams.ReadableStream<VSBuffer>): Promise<VSBuffer> {
327 return streams.consumeStream<VSBuffer>(stream, chunks => VSBuffer.concat(chunks));
328 }
329 > buffer.ts
330 export async function bufferedStreamToBuffer(bufferedStream: streams.ReadableBufferedStream<VSBuffer>): Promise<VSBuffer> {
331 if (bufferedStream.ended) {
342 ]);
343 }
344 > buffer.ts
345 > export function bufferToStream(buffer: VSBuffer): streams.ReadableStream<VSBuffer> {
346 return streams.toStream<VSBuffer>(buffer, chunks => VSBuffer.concat(chunks));
347 }
348 > buffer.ts
349 > export function streamToBufferReadableStream(stream: streams.ReadableStreamEvents<Uint8Array | string>): streams.ReadableStream<VSBuffer> {
350 return streams.transform<Uint8Array | string, VSBuffer>(stream, { data: data => typeof data === 'string' ? VSBuffer.fromString(data) : VSBuffer.wrap(data) }, chunks => VSBuffer.concat(chunks));
351 }
352 > buffer.ts
353 > export function newWriteableBufferStream(options?: streams.WriteableStreamOptions): streams.WriteableStream<VSBuffer> {
354 return streams.newWriteableStream<VSBuffer>(chunks => VSBuffer.concat(chunks), options);
355 }
356 > buffer.ts
357 > export function prefixedBufferReadable(prefix: VSBuffer, readable: VSBufferReadable): VSBufferReadable {
358 return streams.prefixedReadable(prefix, readable, chunks => VSBuffer.concat(chunks));
359 }
360 > buffer.ts
361 > export function prefixedBufferStream(prefix: VSBuffer, stream: VSBufferReadableStream): VSBufferReadableStream {
362 return streams.prefixedStream(prefix, stream, chunks => VSBuffer.concat(chunks));
363 }
364 > buffer.ts
365 > /** Decodes base64 to a uint8 array. URL-encoded and unpadded base64 is allowed. */
366 > export function decodeBase64(encoded: string) {
367 let building = 0;
368 let remainder = 0;
424 return VSBuffer.wrap(buffer).slice(0, unpadded);
425 }
426 > buffer.ts
427 > const base64Alphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/';
428 > const base64UrlSafeAlphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_';
429 >
430 > /** Encodes a buffer to a base64 string. */
431 > export function encodeBase64({ buffer }: VSBuffer, padded = true, urlSafe = false) {
432 const dictionary = urlSafe ? base64UrlSafeAlphabet : base64Alphabet;
433 let output = '';
463 return output;
464 }
465 > buffer.ts
466 > const hexChars = '0123456789abcdef';
467 > export function encodeHex({ buffer }: VSBuffer): string {
468 let result = '';
469 for (let i = 0; i < buffer.length; i++) {
474 return result;
475 }
476 > buffer.ts
477 > export function decodeHex(hex: string): VSBuffer {
478 if (hex.length % 2 !== 0) {
479 throw new SyntaxError('Hex string must have an even length');
485 return VSBuffer.wrap(out);
486 }
487 > buffer.ts
488 function decodeHexChar(str: string, position: number) {
489 const s = str.charCodeAt(position);
src/vs/platform/environment/common/environment.ts 165 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- environment.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../base/common/uri.js';
7 > import { NativeParsedArgs } from './argv.js';
8 > import { createDecorator, refineServiceDecorator } from '../../instantiation/common/instantiation.js';
9 >
10 > export const IEnvironmentService = createDecorator<IEnvironmentService>('environmentService');
11 > export const INativeEnvironmentService = refineServiceDecorator<IEnvironmentService, INativeEnvironmentService>(IEnvironmentService);
12 >
13 > export interface IDebugParams {
14 > port: number | null;
15 > break: boolean;
16 > }
17 >
18 > export interface IExtensionHostDebugParams extends IDebugParams {
19 > debugId?: string;
20 > env?: Record<string, string>;
21 > }
22 >
23 > /**
24 > * Type of extension.
25 > *
26 > * **NOTE**: This is defined in `platform/environment` because it can appear as a CLI argument.
27 > */
28 > export type ExtensionKind = 'ui' | 'workspace' | 'web';
29 >
30 > /**
31 > * A basic environment service that can be used in various processes,
32 > * such as main, renderer and shared process. Use subclasses of this
33 > * service for specific environment.
34 > */
35 > export interface IEnvironmentService {
36 >
37 > readonly _serviceBrand: undefined;
38 >
39 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
40 > //
41 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE.
42 > //
43 > // AS SUCH:
44 > // - PUT NON-WEB PROPERTIES INTO NATIVE ENVIRONMENT SERVICE
45 > // - PUT WORKBENCH ONLY PROPERTIES INTO WORKBENCH ENVIRONMENT SERVICE
46 > // - PUT ELECTRON-MAIN ONLY PROPERTIES INTO MAIN ENVIRONMENT SERVICE
47 > //
48 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
49 >
50 > // --- user roaming data
51 > stateResource: URI;
52 > userRoamingDataHome: URI;
53 > keyboardLayoutResource: URI;
54 > argvResource: URI;
55 >
56 > // --- data paths
57 > untitledWorkspacesHome: URI;
58 > workspaceStorageHome: URI;
59 > localHistoryHome: URI;
60 > cacheHome: URI;
61 > appSharedDataHome: URI;
62 >
63 > // --- settings sync
64 > userDataSyncHome: URI;
65 > sync: 'on' | 'off' | undefined;
66 >
67 > // --- continue edit session
68 > continueOn?: string;
69 > editSessionId?: string;
70 >
71 > // --- extension development
72 > debugExtensionHost: IExtensionHostDebugParams;
73 > isExtensionDevelopment: boolean;
74 > disableExtensions: boolean | string[];
75 > skipBuiltinExtensions?: readonly string[];
76 > enableExtensions?: readonly string[];
77 > extensionDevelopmentLocationURI?: URI[];
78 > extensionDevelopmentKind?: ExtensionKind[];
79 > extensionTestsLocationURI?: URI;
80 >
81 > // --- logging
82 > logsHome: URI;
83 > logLevel?: string;
84 > extensionLogLevel?: [string, string][];
85 > verbose: boolean;
86 > isBuilt: boolean;
87 >
88 > // --- telemetry/exp
89 > disableTelemetry: boolean;
90 > disableExperiments: boolean;
91 > serviceMachineIdResource: URI;
92 >
93 > // --- agent sessions workspace
94 > agentSessionsWorkspace: URI;
95 > // --- Policy
96 > policyFile?: URI;
97 >
98 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
99 > //
100 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE.
101 > //
102 > // AS SUCH:
103 > // - PUT NON-WEB PROPERTIES INTO NATIVE ENVIRONMENT SERVICE
104 > // - PUT WORKBENCH ONLY PROPERTIES INTO WORKBENCH ENVIRONMENT SERVICE
105 > // - PUT ELECTRON-MAIN ONLY PROPERTIES INTO MAIN ENVIRONMENT SERVICE
106 > //
107 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
108 > }
109 >
110 > /**
111 > * A subclass of the `IEnvironmentService` to be used only in native
112 > * environments (Windows, Linux, macOS) but not e.g. web.
113 > */
114 > export interface INativeEnvironmentService extends IEnvironmentService {
115 >
116 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
117 > //
118 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE.
119 > //
120 > // AS SUCH:
121 > // - PUT WORKBENCH ONLY PROPERTIES INTO WORKBENCH ENVIRONMENT SERVICE
122 > // - PUT ELECTRON-MAIN ONLY PROPERTIES INTO MAIN ENVIRONMENT SERVICE
123 > //
124 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
125 >
126 > // --- CLI Arguments
127 > args: NativeParsedArgs;
128 >
129 > // --- data paths
130 > /**
131 > * Root path of the JavaScript sources.
132 > *
133 > * Note: This is NOT the installation root
134 > * directory itself but contained in it at
135 > * a level that is platform dependent.
136 > */
137 > appRoot: string;
138 > userHome: URI;
139 > appSettingsHome: URI;
140 > tmpDir: URI;
141 > userDataPath: string;
142 >
143 > // --- extensions
144 > extensionsPath: string;
145 > extensionsDownloadLocation: URI;
146 > builtinExtensionsPath: string;
147 >
148 > // --- use in-memory Secret Storage
149 > useInMemorySecretStorage?: boolean;
150 >
151 > crossOriginIsolated?: boolean;
152 > exportPolicyData?: string;
153 > exportDefaultKeybindings?: string;
154 >
155 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
156 > //
157 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE.
158 > //
159 > // AS SUCH:
160 > // - PUT NON-WEB PROPERTIES INTO NATIVE ENVIRONMENT SERVICE
161 > // - PUT WORKBENCH ONLY PROPERTIES INTO WORKBENCH ENVIRONMENT SERVICE
162 > // - PUT ELECTRON-MAIN ONLY PROPERTIES INTO MAIN ENVIRONMENT SERVICE
163 > //
164 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
165 > }
src/vs/base/common/policy.ts 161 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- policy.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../nls.js';
7 > import { IPolicyData } from './defaultAccount.js';
8 >
9 > /**
10 > * System-wide policy file path for Linux systems.
11 > */
12 > export const LINUX_SYSTEM_POLICY_FILE_PATH = '/etc/vscode/policy.json';
13 >
14 > export type PolicyName = string;
15 > export type LocalizedValue = {
16 > key: string;
17 > value: string;
18 > };
19 >
20 > export type PolicyValue = string | number | boolean;
21 > export type ManagedSettingValue = PolicyValue;
22 > export type ManagedSettingsData = Readonly<Record<string, ManagedSettingValue>>;
23 >
24 > export interface IManagedSettingPolicyDefinition {
25 > readonly type: 'string' | 'number' | 'boolean';
26 > }
27 >
28 > export type IManagedSettingsPolicyDefinitions = Readonly<Record<string, IManagedSettingPolicyDefinition>>;
29 >
30 > export enum PolicyCategory {
31 > Extensions = 'Extensions',
32 > IntegratedTerminal = 'IntegratedTerminal',
33 > InteractiveSession = 'InteractiveSession',
34 > Telemetry = 'Telemetry',
35 > Update = 'Update',
36 > }
37 >
38 > export const PolicyCategoryData: {
39 > [key in PolicyCategory]: { name: LocalizedValue }
40 > } = {
41 > [PolicyCategory.Extensions]: {
42 > name: {
43 > key: 'extensionsConfigurationTitle', value: localize('extensionsConfigurationTitle', "Extensions"),
44 > }
45 > },
46 > [PolicyCategory.IntegratedTerminal]: {
47 > name: {
48 > key: 'terminalIntegratedConfigurationTitle', value: localize('terminalIntegratedConfigurationTitle', "Integrated Terminal"),
49 > }
50 > },
51 > [PolicyCategory.InteractiveSession]: {
52 > name: {
53 > key: 'interactiveSessionConfigurationTitle', value: localize('interactiveSessionConfigurationTitle', "Chat"),
54 > }
55 > },
56 > [PolicyCategory.Telemetry]: {
57 > name: {
58 > key: 'telemetryConfigurationTitle', value: localize('telemetryConfigurationTitle', "Telemetry"),
59 > }
60 > },
61 > [PolicyCategory.Update]: {
62 > name: {
63 > key: 'updateConfigurationTitle', value: localize('updateConfigurationTitle', "Update"),
64 > }
65 > }
66 > };
67 >
68 > export interface IPolicy {
69 >
70 > /**
71 > * The policy name.
72 > */
73 > readonly name: PolicyName;
74 >
75 > /**
76 > * The policy category.
77 > */
78 > readonly category: PolicyCategory;
79 >
80 > /**
81 > * The Code version in which this policy was introduced.
82 > */
83 > readonly minimumVersion: `${number}.${number}`;
84 >
85 > /**
86 > * Localization info for the policy.
87 > *
88 > * IMPORTANT: the key values for these must be unique to avoid collisions, as during the export time the module information is not available.
89 > */
90 > readonly localization: {
91 > /** The localization key or key value pair. If only a key is provided, the default value will fallback to the parent configuration's description property. */
92 > description: LocalizedValue;
93 > /** List of localization key or key value pair. If only a key is provided, the default value will fallback to the parent configuration's enumDescriptions property. */
94 > enumDescriptions?: LocalizedValue[];
95 > };
96 >
97 > /**
98 > * The value that an ACCOUNT-based feature will use when its corresponding policy is active.
99 > *
100 > * Only applicable when policy is tagged with ACCOUNT. When an account-based feature's policy is enabled,
101 > * this value determines what value the feature receives.
102 > *
103 > * For example:
104 > * - If evaluated value is `true`, the feature's setting is locked to `true` WHEN the policy is in effect.
105 > * - If evaluated value is `foo`, the feature's setting is locked to 'foo' WHEN the policy is in effect.
106 > *
107 > * If `undefined`, the feature's setting is not locked and can be overridden by other means.
108 > */
109 > readonly value?: (policyData: IPolicyData) => string | number | boolean | undefined;
110 >
111 > /**
112 > * Declares Copilot managed-settings keys this policy's value callback reads.
113 > * Keys are dot-separated managed-settings paths, for example
114 > * `permissions.disableBypassPermissionsMode`.
115 > */
116 > readonly managedSettings?: IManagedSettingsPolicyDefinitions;
117 >
118 > /**
119 > * The most-restrictive value that should be applied when the user is subject to the
120 > * "Require Approved Account" gate but the gate is not yet satisfied (i.e. no approved
121 > * GitHub account is signed in or the account-side policy data has not yet resolved).
122 > *
123 > * If omitted, the gate falls back to a type-driven safe default
124 > * (`false` for boolean, `0` for number, `''` for string).
125 > *
126 > * Only consulted while the gate is active and unsatisfied; ignored otherwise.
127 > */
128 > readonly restrictedValue?: string | number | boolean;
129 > }
130 >
131 > /**
132 > * A subordinate attachment to an existing {@link IPolicy} (the "owner"). A setting may declare a
133 > * `policyReference` instead of a full `policy` to be governed by a policy owned by another setting,
134 > * letting a single enterprise policy lock more than one setting (e.g. gating an agent in both the
135 > * editor window and the Agents window).
136 > *
137 > * A reference is a pure pointer: it carries no policy semantics of its own. The owner is the single
138 > * source of truth for the policy's catalog metadata *and* its runtime behaviour (type, value
139 > * callback, etc.); a reference only contributes the policy name so the setting is gated and the OS
140 > * policy watcher observes the name in processes where the owner is not loaded.
141 > */
142 > export interface IPolicyReference {
143 >
144 > /** The name of the owning {@link IPolicy} this setting attaches to. */
145 > readonly name: PolicyName;
146 > }
147 >
148 > /**
149 > * A `product.json` `extensionConfigurationPolicy` entry that attaches its setting to a policy
150 > * *owned* by an in-code setting, instead of declaring a full owner {@link IPolicy}. This mirrors the
151 > * in-code `policyReference` configuration field, so the same indirection can be expressed from
152 > * `product.json` — where the owner's runtime behaviour (notably its `value` callback) cannot live.
153 > *
154 > * An `extensionConfigurationPolicy` entry is therefore either a full {@link IPolicy} (the setting
155 > * "parents"/owns the policy, the current syntax) or this reference wrapper.
156 > */
157 > export interface IExtensionConfigurationPolicyReference {
158 >
159 > /** Pointer to the owning {@link IPolicy} declared by an in-code setting. */
160 > readonly policyReference: IPolicyReference;
161 > }
src/vs/base/common/actions.ts 151 covered LOC · 25 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- actions.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from './event.js';
7 > import { Disposable, IDisposable } from './lifecycle.js';
8 > import * as nls from '../../nls.js';
9 >
10 > export interface ITelemetryData {
11 > readonly from?: string;
12 > readonly target?: string;
13 > [key: string]: unknown;
14 > }
15 >
16 > export type WorkbenchActionExecutedClassification = {
17 > id: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The identifier of the action that was run.' };
18 > from: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'The name of the component the action was run from.' };
19 > detail?: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; comment: 'Optional details about how the action was run, e.g which keybinding was used.' };
20 > owner: 'isidorn';
21 > comment: 'Provides insight into actions that are executed within the workbench.';
22 > };
23 >
24 > export type WorkbenchActionExecutedEvent = {
25 > id: string;
26 > from: string;
27 > detail?: string;
28 > };
29 >
30 > export interface IAction {
31 > readonly id: string;
32 > label: string;
33 > tooltip: string;
34 > class: string | undefined;
35 > enabled: boolean;
36 > checked?: boolean;
37 > run(...args: unknown[]): unknown;
38 > }
39 >
40 > export interface IActionRunner extends IDisposable {
41 > readonly onDidRun: Event<IRunEvent>;
42 > readonly onWillRun: Event<IRunEvent>;
43 >
44 > run(action: IAction, context?: unknown): unknown;
45 > }
46 >
47 > export interface IActionChangeEvent {
48 > readonly label?: string;
49 > readonly tooltip?: string;
50 > readonly class?: string;
51 > readonly enabled?: boolean;
52 > readonly checked?: boolean;
53 > }
54 >
55 > /**
56 > * A concrete implementation of {@link IAction}.
57 > *
58 > * Note that in most cases you should use the lighter-weight {@linkcode toAction} function instead.
59 > */
60 > export class Action extends Disposable implements IAction {
61 >
62 > protected _onDidChange = this._register(new Emitter<IActionChangeEvent>());
63 > get onDidChange() { return this._onDidChange.event; }
64 >
65 > protected readonly _id: string;
66 > protected _label: string;
67 > protected _tooltip: string | undefined;
68 > protected _cssClass: string | undefined;
69 > protected _enabled: boolean = true;
70 > protected _checked?: boolean;
71 > protected readonly _actionCallback?: (event?: unknown) => unknown;
72 >
73 > constructor(id: string, label: string = '', cssClass: string = '', enabled: boolean = true, actionCallback?: (event?: unknown) => unknown) {
74 super();
75 this._id = id;
79 this._actionCallback = actionCallback;
80 }
81 > actions.ts
82 > get id(): string {
83 return this._id;
84 }
85 > actions.ts
86 > get label(): string {
87 return this._label;
88 }
89 > actions.ts
90 > set label(value: string) {
91 this._setLabel(value);
92 }
93 > actions.ts
94 > private _setLabel(value: string): void {
95 if (this._label !== value) {
96 this._label = value;
98 }
99 }
100 > actions.ts
101 > get tooltip(): string {
102 return this._tooltip || '';
103 }
104 > actions.ts
105 > set tooltip(value: string) {
106 this._setTooltip(value);
107 }
108 > actions.ts
109 > protected _setTooltip(value: string): void {
110 if (this._tooltip !== value) {
111 this._tooltip = value;
113 }
114 }
115 > actions.ts
116 > get class(): string | undefined {
117 return this._cssClass;
118 }
119 > actions.ts
120 > set class(value: string | undefined) {
121 this._setClass(value);
122 }
123 > actions.ts
124 > protected _setClass(value: string | undefined): void {
125 if (this._cssClass !== value) {
126 this._cssClass = value;
128 }
129 }
130 > actions.ts
131 > get enabled(): boolean {
132 return this._enabled;
133 }
134 > actions.ts
135 > set enabled(value: boolean) {
136 this._setEnabled(value);
137 }
138 > actions.ts
139 > protected _setEnabled(value: boolean): void {
140 if (this._enabled !== value) {
141 this._enabled = value;
143 }
144 }
145 > actions.ts
146 > get checked(): boolean | undefined {
147 return this._checked;
148 }
149 > actions.ts
150 > set checked(value: boolean | undefined) {
151 this._setChecked(value);
152 }
153 > actions.ts
154 > protected _setChecked(value: boolean | undefined): void {
155 if (this._checked !== value) {
156 this._checked = value;
158 }
159 }
160 > actions.ts
161 > async run(event?: unknown, data?: ITelemetryData): Promise<void> {
162 if (this._actionCallback) {
163 await this._actionCallback(event);
164 }
165 }
166 > } actions.ts
167 >
168 > export interface IRunEvent {
169 > readonly action: IAction;
170 > readonly error?: Error;
171 > }
172 >
173 > export class ActionRunner extends Disposable implements IActionRunner {
174
175 private readonly _onWillRun = this._register(new Emitter<IRunEvent>());
177
178 private readonly _onDidRun = this._register(new Emitter<IRunEvent>());
179 > get onDidRun() { return this._onDidRun.event; } actions.ts
180 >
181 > async run(action: IAction, context?: unknown): Promise<void> {
182 if (!action.enabled) {
183 return;
195 this._onDidRun.fire({ action, error });
196 }
197 > actions.ts
198 > protected async runAction(action: IAction, context?: unknown): Promise<void> {
199 await action.run(context);
200 }
201 > } actions.ts
202 >
203 > export class Separator implements IAction {
204
205 /**
248 readonly enabled: boolean = false;
249 readonly checked: undefined = undefined;
250 > async run() { } actions.ts
251 > }
252 >
253 > export class SubmenuAction implements IAction {
254 >
255 > readonly id: string;
256 > readonly label: string;
257 > readonly class: string | undefined;
258 > readonly tooltip: string = '';
259 > readonly enabled: boolean = true;
260 > readonly checked: undefined = undefined;
261 >
262 > private readonly _actions: readonly IAction[];
263 > get actions(): readonly IAction[] { return this._actions; }
264 >
265 > constructor(id: string, label: string, actions: readonly IAction[], cssClass?: string) {
266 this.id = id;
267 this.label = label;
269 this._actions = actions;
270 }
271 > actions.ts
272 > async run(): Promise<void> { }
273 > }
274 >
275 > export class EmptySubmenuAction extends Action {
276 >
277 > static readonly ID = 'vs.actions.empty';
278 >
279 > constructor() {
280 super(EmptySubmenuAction.ID, nls.localize('submenu.empty', '(empty)'), undefined, false);
281 }
282 > } actions.ts
283 >
284 > export function toAction(props: { id: string; label: string; tooltip?: string; enabled?: boolean; checked?: boolean; class?: string; run: Function }): IAction {
285 return {
286 id: props.id,
src/vs/base/common/keybindings.ts 151 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- keybindings.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { illegalArgument } from './errors.js';
7 > import { KeyCode, ScanCode } from './keyCodes.js';
8 > import { OperatingSystem } from './platform.js';
9 >
10 > /**
11 > * Binary encoding strategy:
12 > * ```
13 > * 1111 11
14 > * 5432 1098 7654 3210
15 > * ---- CSAW KKKK KKKK
16 > * C = bit 11 = ctrlCmd flag
17 > * S = bit 10 = shift flag
18 > * A = bit 9 = alt flag
19 > * W = bit 8 = winCtrl flag
20 > * K = bits 0-7 = key code
21 > * ```
22 > */
23 > const enum BinaryKeybindingsMask {
24 > CtrlCmd = (1 << 11) >>> 0,
25 > Shift = (1 << 10) >>> 0,
26 > Alt = (1 << 9) >>> 0,
27 > WinCtrl = (1 << 8) >>> 0,
28 > KeyCode = 0x000000FF
29 > }
30 >
31 > export function decodeKeybinding(keybinding: number | number[], OS: OperatingSystem): Keybinding | null {
32 if (typeof keybinding === 'number') {
33 if (keybinding === 0) {
51 }
52 }
54 > export function createSimpleKeybinding(keybinding: number, OS: OperatingSystem): KeyCodeChord {
55
56 const ctrlCmd = (keybinding & BinaryKeybindingsMask.CtrlCmd ? true : false);
65 return new KeyCodeChord(ctrlKey, shiftKey, altKey, metaKey, keyCode);
66 }
68 > export interface Modifiers {
69 > readonly ctrlKey: boolean;
70 > readonly shiftKey: boolean;
71 > readonly altKey: boolean;
72 > readonly metaKey: boolean;
73 > }
74 >
75 > /**
76 > * Represents a chord which uses the `keyCode` field of keyboard events.
77 > * A chord is a combination of keys pressed simultaneously.
78 > */
79 > export class KeyCodeChord implements Modifiers {
80 >
81 > constructor(
82 public readonly ctrlKey: boolean,
83 public readonly shiftKey: boolean,
86 public readonly keyCode: KeyCode
87 ) { }
89 > public equals(other: Chord): boolean {
90 return (
91 other instanceof KeyCodeChord
97 );
98 }
100 > public getHashCode(): string {
101 const ctrl = this.ctrlKey ? '1' : '0';
102 const shift = this.shiftKey ? '1' : '0';
105 return `K${ctrl}${shift}${alt}${meta}${this.keyCode}`;
106 }
108 > public isModifierKey(): boolean {
109 return (
110 this.keyCode === KeyCode.Unknown
115 );
116 }
118 > public toKeybinding(): Keybinding {
119 return new Keybinding([this]);
120 }
122 > /**
123 > * Does this keybinding refer to the key code of a modifier and it also has the modifier flag?
124 > */
125 > public isDuplicateModifierCase(): boolean {
126 return (
127 (this.ctrlKey && this.keyCode === KeyCode.Ctrl)
131 );
132 }
133 > } keybindings.ts
134 >
135 > /**
136 > * Represents a chord which uses the `code` field of keyboard events.
137 > * A chord is a combination of keys pressed simultaneously.
138 > */
139 > export class ScanCodeChord implements Modifiers {
140 >
141 > constructor(
142 public readonly ctrlKey: boolean,
143 public readonly shiftKey: boolean,
146 public readonly scanCode: ScanCode
147 ) { }
149 > public equals(other: Chord): boolean {
150 return (
151 other instanceof ScanCodeChord
157 );
158 }
160 > public getHashCode(): string {
161 const ctrl = this.ctrlKey ? '1' : '0';
162 const shift = this.shiftKey ? '1' : '0';
165 return `S${ctrl}${shift}${alt}${meta}${this.scanCode}`;
166 }
168 > /**
169 > * Does this keybinding refer to the key code of a modifier and it also has the modifier flag?
170 > */
171 > public isDuplicateModifierCase(): boolean {
172 return (
173 (this.ctrlKey && (this.scanCode === ScanCode.ControlLeft || this.scanCode === ScanCode.ControlRight))
177 );
178 }
179 > } keybindings.ts
180 >
181 > export type Chord = KeyCodeChord | ScanCodeChord;
182 >
183 > /**
184 > * A keybinding is a sequence of chords.
185 > */
186 > export class Keybinding {
187 >
188 > public readonly chords: Chord[];
189 >
190 > constructor(chords: Chord[]) {
191 if (chords.length === 0) {
192 throw illegalArgument(`chords`);
194 this.chords = chords;
195 }
197 > public getHashCode(): string {
198 let result = '';
199 for (let i = 0, len = this.chords.length; i < len; i++) {
205 return result;
206 }
208 > public equals(other: Keybinding | null): boolean {
209 if (other === null) {
210 return false;
220 return true;
221 }
222 > } keybindings.ts
223 >
224 > export class ResolvedChord {
225 > constructor(
226 public readonly ctrlKey: boolean,
227 public readonly shiftKey: boolean,
231 public readonly keyAriaLabel: string | null
232 ) { }
233 > } keybindings.ts
234 >
235 > export type SingleModifierChord = 'ctrl' | 'shift' | 'alt' | 'meta';
236 >
237 > /**
238 > * A resolved keybinding. Consists of one or multiple chords.
239 > */
240 > export abstract class ResolvedKeybinding {
241 > /**
242 > * This prints the binding in a format suitable for displaying in the UI.
243 > */
244 > public abstract getLabel(): string | null;
245 > /**
246 > * This prints the binding in a format suitable for ARIA.
247 > */
248 > public abstract getAriaLabel(): string | null;
249 > /**
250 > * This prints the binding in a format suitable for electron's accelerators.
251 > * See https://github.com/electron/electron/blob/master/docs/api/accelerator.md
252 > */
253 > public abstract getElectronAccelerator(): string | null;
254 > /**
255 > * This prints the binding in a format suitable for user settings.
256 > */
257 > public abstract getUserSettingsLabel(): string | null;
258 > /**
259 > * Is the user settings label reflecting the label?
260 > */
261 > public abstract isWYSIWYG(): boolean;
262 > /**
263 > * Does the keybinding consist of more than one chord?
264 > */
265 > public abstract hasMultipleChords(): boolean;
266 > /**
267 > * Returns the chords that comprise of the keybinding.
268 > */
269 > public abstract getChords(): ResolvedChord[];
270 > /**
271 > * Returns the chords as strings useful for dispatching.
272 > * Returns null for modifier only chords.
273 > * @example keybinding "Shift" -> null
274 > * @example keybinding ("D" with shift == true) -> "shift+D"
275 > */
276 > public abstract getDispatchChords(): (string | null)[];
277 > /**
278 > * Returns the modifier only chords as strings useful for dispatching.
279 > * Returns null for chords that contain more than one modifier or a regular key.
280 > * @example keybinding "Shift" -> "shift"
281 > * @example keybinding ("D" with shift == true") -> null
282 > */
283 > public abstract getSingleModifierDispatchChords(): (SingleModifierChord | null)[];
284 > }
src/vs/platform/agentHost/common/state/protocol/channels-chat/commands.ts 147 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI } from '../common/state.js';
10 > import type { BaseParams } from '../common/commands.js';
11 > import type { Message, SideChatSelection } from './state.js';
12 >
13 > // ─── createChat ──────────────────────────────────────────────────────────────
14 >
15 > /**
16 > * How a new chat uses its source chat and turn.
17 > */
18 > export const enum ChatSourceKind {
19 > /** Copy source history through the referenced turn into the new chat. */
20 > Fork = 'fork',
21 > /** Supply source context without copying it into the new chat's visible history. */
22 > SideChat = 'sideChat',
23 > }
24 >
25 > /**
26 > * Copies source history through a completed turn into the new chat.
27 > */
28 > export interface ForkChatSource {
29 > /** Discriminant */
30 > kind: ChatSourceKind.Fork;
31 > /** URI of the existing source chat. */
32 > chat: URI;
33 > /**
34 > * Completed turn identifier in the source chat.
35 > *
36 > * Content through this turn is copied into the new chat's visible `turns`.
37 > */
38 > turnId: string;
39 > }
40 >
41 > /**
42 > * Supplies source context to a new side chat without copying it into the side
43 > * chat's visible history.
44 > */
45 > export interface SideChatSource {
46 > /** Discriminant */
47 > kind: ChatSourceKind.SideChat;
48 > /** URI of the existing source chat. */
49 > chat: URI;
50 > /**
51 > * Stable source-turn identifier in the source chat.
52 > *
53 > * Hosts resolve this id against the source chat's current `activeTurn` or its
54 > * retained `turns` when accepting `createChat`. If it names the current
55 > * active turn, the host snapshots the source chat's retained history plus
56 > * that turn's current user message and any partial assistant response already
57 > * available. Once that turn later becomes historical, it is still referenced
58 > * by this same identifier.
59 > */
60 > turnId: string;
61 > /**
62 > * Optional immutable selected-text snapshot to carry into the created side
63 > * chat's origin.
64 > *
65 > * When present, the host MUST snapshot and preserve this exact selection when
66 > * it accepts `createChat`; later source-turn deltas do not alter it.
67 > */
68 > selection?: SideChatSelection;
69 > }
70 >
71 > /**
72 > * Identifies a source chat for a new chat.
73 > */
74 > export type ChatSource =
75 > | ForkChatSource
76 > | SideChatSource;
77 >
78 > /**
79 > * Creates a new chat within a session.
80 > *
81 > * @category Commands
82 > * @method createChat
83 > * @direction Client → Server
84 > * @messageType Request
85 > * @version 1
86 > */
87 > export interface CreateChatParams extends BaseParams {
88 > /** Session URI containing the new chat. */
89 > channel: URI;
90 > /** Chat URI (client-chosen, e.g. `ahp-chat:/<uuid>`). */
91 > chat: URI;
92 > /** Optional initial message for the new chat. */
93 > initialMessage?: Message;
94 > /**
95 > * Optional source chat and source turn.
96 > *
97 > * The source chat MUST belong to this session. Clients MUST only request
98 > * `kind: "fork"` when the selected agent advertises
99 > * `capabilities.multipleChats.fork`, and `kind: "sideChat"` when the
100 > * selected agent advertises `capabilities.multipleChats.sideChat`. Both
101 > * source forms carry a stable top-level `turnId`. Forks target completed
102 > * turns. Side chats also carry a stable `turnId`, which the host resolves
103 > * against the source chat's current active turn or retained history. If it
104 > * resolves to the active turn, the host snapshots the currently available
105 > * partial response when accepting `createChat`. When
106 > * `source.kind === "sideChat"` and `source.selection` is present, the host
107 > * also snapshots and preserves that exact selected text in the created chat's
108 > * origin; any `responsePartId` there is provenance only, not a live range.
109 > */
110 > source?: ChatSource;
111 > /**
112 > * Initial working-directory subset for this chat. Every entry MUST be
113 > * present in the owning session's `workingDirectories`; the server MUST
114 > * reject any entry that is not. When absent, the chat inherits the full
115 > * session set. Forked chats (those whose `source.kind` is `"fork"`) inherit
116 > * the source chat's `workingDirectories`; this field is ignored for forks.
117 > *
118 > * A client MUST NOT supply this field unless the agent advertises
119 > * {@link AgentCapabilities.multipleWorkingDirectories}.
120 > */
121 > workingDirectories?: URI[];
122 > /**
123 > * The chat's primary working directory — the distinguished root this chat is
124 > * centered on. When set, it MUST be one of the chat's effective working
125 > * directories ({@link workingDirectories}, or the session's set when that is
126 > * omitted). A client SHOULD supply this when the agent advertises
127 > * {@link MultipleWorkingDirectoriesCapability.requiresPrimary}; a host MAY
128 > * reject creation that omits it, or fall back to the first of the chat's
129 > * directories. Fixed at creation and reported (read-only) on
130 > * {@link ChatState.primaryWorkingDirectory}. Ignored for forks (a chat whose
131 > * `source.kind` is `"fork"` inherits the source chat's primary).
132 > */
133 > primaryWorkingDirectory?: URI;
134 > }
135 >
136 > // ─── disposeChat ─────────────────────────────────────────────────────────────
137 >
138 > /**
139 > * Disposes a chat and cleans up server-side resources.
140 > *
141 > * @category Commands
142 > * @method disposeChat
143 > * @direction Client → Server
144 > * @messageType Request
145 > * @version 1
146 > */
147 > export interface DisposeChatParams extends BaseParams { }
src/vs/base/common/labels.ts 146 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- labels.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { hasDriveLetter, toSlashes } from './extpath.js';
7 > import { posix, sep, win32 } from './path.js';
8 > import { isMacintosh, isWindows, OperatingSystem, OS } from './platform.js';
9 > import { extUri, extUriIgnorePathCase } from './resources.js';
10 > import { rtrim, startsWithIgnoreCase } from './strings.js';
11 > import { URI } from './uri.js';
12 >
13 > export interface IPathLabelFormatting {
14 >
15 > /**
16 > * The OS the path label is from to produce a label
17 > * that matches OS expectations.
18 > */
19 > readonly os: OperatingSystem;
20 >
21 > /**
22 > * Whether to add a `~` when the path is in the
23 > * user home directory.
24 > *
25 > * Note: this only applies to Linux, macOS but not
26 > * Windows.
27 > */
28 > readonly tildify?: IUserHomeProvider;
29 >
30 > /**
31 > * Whether to convert to a relative path if the path
32 > * is within any of the opened workspace folders.
33 > */
34 > readonly relative?: IRelativePathProvider;
35 > }
36 >
37 > export interface IRelativePathProvider {
38 >
39 > /**
40 > * Whether to not add a prefix when in multi-root workspace.
41 > */
42 > readonly noPrefix?: boolean;
43 >
44 > getWorkspace(): { folders: { uri: URI; name?: string }[] };
45 > getWorkspaceFolder(resource: URI): { uri: URI; name?: string } | null;
46 > }
47 >
48 > export interface IUserHomeProvider {
49 > userHome: URI;
50 > }
51 >
52 > export function getPathLabel(resource: URI, formatting: IPathLabelFormatting): string {
53 const { os, tildify: tildifier, relative: relatifier } = formatting;
54
93 return pathLib.normalize(normalizeDriveLetter(absolutePath, os === OperatingSystem.Windows));
94 }
95 > labels.ts
96 function getRelativePathLabel(resource: URI, relativePathProvider: IRelativePathProvider, os: OperatingSystem): string | undefined {
97 const pathLib = os === OperatingSystem.Windows ? win32 : posix;
137 return relativePathLabel;
138 }
139 > labels.ts
140 > export function normalizeDriveLetter(path: string, isWindowsOS: boolean = isWindows): string {
141 if (hasDriveLetter(path, isWindowsOS)) {
142 return path.charAt(0).toUpperCase() + path.slice(1);
145 return path;
146 }
147 > labels.ts
148 > let normalizedUserHomeCached: { original: string; normalized: string } = Object.create(null);
149 > export function tildify(path: string, userHome: string, os = OS): string {
150 if (os === OperatingSystem.Windows || !path || !userHome) {
151 return path; // unsupported on Windows
174 return path;
175 }
176 > labels.ts
177 > export function untildify(path: string, userHome: string): string {
178 return path.replace(/^~($|\/|\\)/, `${userHome}$1`);
179 }
180 > labels.ts
181 > /**
182 > * Shortens the paths but keeps them easy to distinguish.
183 > * Replaces not important parts with ellipsis.
184 > * Every shorten path matches only one original path and vice versa.
185 > *
186 > * Algorithm for shortening paths is as follows:
187 > * 1. For every path in list, find unique substring of that path.
188 > * 2. Unique substring along with ellipsis is shortened path of that path.
189 > * 3. To find unique substring of path, consider every segment of length from 1 to path.length of path from end of string
190 > * and if present segment is not substring to any other paths then present segment is unique path,
191 > * else check if it is not present as suffix of any other path and present segment is suffix of path itself,
192 > * if it is true take present segment as unique path.
193 > * 4. Apply ellipsis to unique segment according to whether segment is present at start/in-between/end of path.
194 > *
195 > * Example 1
196 > * 1. consider 2 paths i.e. ['a\\b\\c\\d', 'a\\f\\b\\c\\d']
197 > * 2. find unique path of first path,
198 > * a. 'd' is present in path2 and is suffix of path2, hence not unique of present path.
199 > * b. 'c' is present in path2 and 'c' is not suffix of present path, similarly for 'b' and 'a' also.
200 > * c. 'd\\c' is suffix of path2.
201 > * d. 'b\\c' is not suffix of present path.
202 > * e. 'a\\b' is not present in path2, hence unique path is 'a\\b...'.
203 > * 3. for path2, 'f' is not present in path1 hence unique is '...\\f\\...'.
204 > *
205 > * Example 2
206 > * 1. consider 2 paths i.e. ['a\\b', 'a\\b\\c'].
207 > * a. Even if 'b' is present in path2, as 'b' is suffix of path1 and is not suffix of path2, unique path will be '...\\b'.
208 > * 2. for path2, 'c' is not present in path1 hence unique path is '..\\c'.
209 > */
210 > const ellipsis = '\u2026';
211 > const unc = '\\\\';
212 > const urlSchemaRegexp = /^[^:/\\?#]+?:\/\//;
213 > const home = '~';
214 > export function shorten(paths: string[], defaultPathSeparator: string = sep): string[] {
215 const shortenedPaths: string[] = new Array(paths.length);
216
323 return shortenedPaths;
324 }
325 > labels.ts
326 > export interface ISeparator {
327 > label: string;
328 > }
329 >
330 > enum Type {
331 > TEXT,
332 > VARIABLE,
333 > SEPARATOR
334 > }
335 >
336 > interface ISegment {
337 > value: string;
338 > type: Type;
339 > }
340 >
341 > /**
342 > * Helper to insert values for specific template variables into the string. E.g. "this $(is) a $(template)" can be
343 > * passed to this function together with an object that maps "is" and "template" to strings to have them replaced.
344 > * @param value string to which template is applied
345 > * @param values the values of the templates to use
346 > */
347 > export function template(template: string, values: { [key: string]: string | ISeparator | undefined | null } = Object.create(null)): string {
348 const segments: ISegment[] = [];
349
409 }).map(segment => segment.value).join('');
410 }
411 > labels.ts
412 > /**
413 > * Handles mnemonics for menu items. Depending on OS:
414 > * - Windows: Supported via & character (replace && with &)
415 > * - Linux: Supported via & character (replace && with &)
416 > * - macOS: Unsupported (replace && with empty string)
417 > */
418 > export function mnemonicMenuLabel(label: string, forceDisableMnemonics?: boolean): string {
419 if (isMacintosh || forceDisableMnemonics) {
420 return label.replace(/\(&&\w\)|&&/g, '').replace(/&/g, isMacintosh ? '&' : '&&');
423 return label.replace(/&&|&/g, m => m === '&' ? '&&' : '&');
424 }
425 > labels.ts
426 > /**
427 > * Handles mnemonics for buttons. Depending on OS:
428 > * - Windows: Supported via & character (replace && with & and & with && for escaping)
429 > * - Linux: Supported via _ character (replace && with _)
430 > * - macOS: Unsupported (replace && with empty string)
431 > * When forceDisableMnemonics is set, returns just the label without mnemonics.
432 > */
433 > export function mnemonicButtonLabel(label: string, forceDisableMnemonics: true): string;
434 > export function mnemonicButtonLabel(label: string, forceDisableMnemonics?: false): { readonly withMnemonic: string; readonly withoutMnemonic: string };
435 > export function mnemonicButtonLabel(label: string, forceDisableMnemonics?: boolean): { readonly withMnemonic: string; readonly withoutMnemonic: string } | string {
436 const withoutMnemonic = label.replace(/\(&&\w\)|&&/g, '');
437
451 return { withMnemonic, withoutMnemonic };
452 }
453 > labels.ts
454 > export function unmnemonicLabel(label: string): string {
455 return label.replace(/&/g, '&&');
456 }
457 > labels.ts
458 > /**
459 > * Splits a recent label in name and parent path, supporting both '/' and '\' and workspace suffixes.
460 > * If the location is remote, the remote name is included in the name part.
461 > */
462 > export function splitRecentLabel(recentLabel: string): { name: string; parentPath: string } {
463 if (recentLabel.endsWith(']')) {
464 // label with workspace suffix
472 return splitName(recentLabel);
473 }
474 > labels.ts
475 function splitName(fullPath: string): { name: string; parentPath: string } {
476 const p = fullPath.indexOf('/') !== -1 ? posix : win32;
src/vs/base/common/observableInternal/observables/baseObservable.ts 146 covered LOC · 28 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- baseObservable.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservableWithChange, IObserver, IReader, IObservable } from '../base.js';
7 > import { DisposableStore } from '../commonFacade/deps.js';
8 > import { DebugLocation } from '../debugLocation.js';
9 > import { DebugOwner, getFunctionName } from '../debugName.js';
10 > import { debugGetObservableGraph } from '../logging/debugGetDependencyGraph.js';
11 > import { getLogger, logObservable } from '../logging/logging.js';
12 > import type { keepObserved, recomputeInitiallyAndOnChange } from '../utils/utils.js';
13 > import { derivedOpts } from './derived.js';
14 >
15 > let _derived: typeof derivedOpts;
16 > /**
17 > * @internal
18 > * This is to allow splitting files.
19 > */
20 > export function _setDerivedOpts(derived: typeof _derived) {
21 > _derived = derived;
22 > }
23 >
24 > let _recomputeInitiallyAndOnChange: typeof recomputeInitiallyAndOnChange;
25 > export function _setRecomputeInitiallyAndOnChange(recomputeInitiallyAndOnChange: typeof _recomputeInitiallyAndOnChange) {
26 > _recomputeInitiallyAndOnChange = recomputeInitiallyAndOnChange;
27 > }
28 >
29 > let _keepObserved: typeof keepObserved;
30 > export function _setKeepObserved(keepObserved: typeof _keepObserved) {
31 > _keepObserved = keepObserved;
32 > }
33 >
34 > let _debugGetObservableGraph: typeof debugGetObservableGraph;
35 > export function _setDebugGetObservableGraph(debugGetObservableGraph: typeof _debugGetObservableGraph) {
36 > _debugGetObservableGraph = debugGetObservableGraph;
37 > }
38 >
39 > export abstract class ConvenientObservable<T, TChange> implements IObservableWithChange<T, TChange> {
40 > get TChange(): TChange { return null!; }
41 >
42 > public abstract get(): T;
43 >
44 > public reportChanges(): void {
45 > this.get(); baseObservable.ts
46 > }
48 > public abstract addObserver(observer: IObserver): void;
49 > public abstract removeObserver(observer: IObserver): void;
50 >
51 > /** @sealed */
52 > public read(reader: IReader | undefined): T {
53 > if (reader) { baseObservable.ts
54 > return reader.readObservable(this); baseObservable.ts
55 > } else { baseObservable.ts
56 return this.get();
57 }
60 > /** @sealed */
61 > public map<TNew>(fn: (value: T, reader: IReader) => TNew): IObservable<TNew>;
62 > public map<TNew>(owner: DebugOwner, fn: (value: T, reader: IReader) => TNew): IObservable<TNew>;
63 > public map<TNew>(fnOrOwner: DebugOwner | ((value: T, reader: IReader) => TNew), fnOrUndefined?: (value: T, reader: IReader) => TNew, debugLocation: DebugLocation = DebugLocation.ofCaller()): IObservable<TNew> {
64 > const owner = fnOrUndefined === undefined ? undefined : fnOrOwner as DebugOwner; baseObservable.ts
65 > const fn = fnOrUndefined === undefined ? fnOrOwner as (value: T, reader: IReader) => TNew : fnOrUndefined;
66 >
67 > return _derived(
68 > {
69 > owner,
70 > debugName: () => {
71 const name = getFunctionName(fn);
72 if (name !== undefined) {
85 return undefined;
86 },
87 > debugReferenceFn: fn, baseObservable.ts
88 > },
89 > (reader) => fn(this.read(reader), reader),
90 > debugLocation,
91 > );
92 > }
94 > public abstract log(): IObservableWithChange<T, TChange>;
95 >
96 > /**
97 > * @sealed
98 > * Converts an observable of an observable value into a direct observable of the value.
99 > */
100 > public flatten<TNew>(this: IObservable<IObservableWithChange<TNew, any>>): IObservable<TNew> {
101 return _derived(
102 {
107 );
108 }
110 > public recomputeInitiallyAndOnChange(store: DisposableStore, handleValue?: (value: T) => void): IObservable<T> {
111 > store.add(_recomputeInitiallyAndOnChange!(this, handleValue)); baseObservable.ts
112 > return this;
113 > }
115 > /**
116 > * Ensures that this observable is observed. This keeps the cache alive.
117 > * However, in case of deriveds, it does not force eager evaluation (only when the value is read/get).
118 > * Use `recomputeInitiallyAndOnChange` for eager evaluation.
119 > */
120 > public keepObserved(store: DisposableStore): IObservable<T> {
121 store.add(_keepObserved!(this));
122 return this;
123 }
125 > public abstract get debugName(): string;
126 >
127 > protected get debugValue() {
128 return this.get();
129 }
131 > get debug(): DebugHelper {
132 return new DebugHelper(this);
133 }
135 >
136 > class DebugHelper {
137 > constructor(public readonly observable: IObservableWithChange<any, any>) {
138 }
140 > getDependencyGraph(): string {
141 return _debugGetObservableGraph(this.observable, { type: 'dependencies' });
142 }
144 > getObserverGraph(): string {
145 return _debugGetObservableGraph(this.observable, { type: 'observers' });
146 }
148 >
149 > export abstract class BaseObservable<T, TChange = void> extends ConvenientObservable<T, TChange> {
150 > protected readonly _observers = new Set<IObserver>();
151 >
152 > constructor(debugLocation: DebugLocation) {
153 > super(); baseObservable.ts
154 > getLogger()?.handleObservableCreated(this, debugLocation);
155 > }
157 > public addObserver(observer: IObserver): void {
158 > const len = this._observers.size; baseObservable.ts
159 > this._observers.add(observer);
160 > if (len === 0) {
161 > this.onFirstObserverAdded();
162 > }
163 > if (len !== this._observers.size) {
164 > getLogger()?.handleOnListenerCountChanged(this, this._observers.size);
165 > }
166 > }
168 > public removeObserver(observer: IObserver): void {
169 > const deleted = this._observers.delete(observer); baseObservable.ts
170 > if (deleted && this._observers.size === 0) {
171 > this.onLastObserverRemoved();
172 > }
173 > if (deleted) {
174 > getLogger()?.handleOnListenerCountChanged(this, this._observers.size);
175 > }
176 > }
178 > protected onFirstObserverAdded(): void { }
179 > protected onLastObserverRemoved(): void { }
180 >
181 > public override log(): IObservableWithChange<T, TChange> {
182 const hadLogger = !!getLogger();
183 logObservable(this);
src/vs/workbench/contrib/chat/common/aiCustomizationWorkspaceService.ts 146 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- aiCustomizationWorkspaceService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../base/common/cancellation.js';
7 > import { IObservable } from '../../../../base/common/observable.js';
8 > import { URI } from '../../../../base/common/uri.js';
9 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
10 > import { PromptsType } from './promptSyntax/promptTypes.js';
11 > import { IChatPromptSlashCommand } from './promptSyntax/service/promptsService.js';
12 >
13 > export const IAICustomizationWorkspaceService = createDecorator<IAICustomizationWorkspaceService>('aiCustomizationWorkspaceService');
14 >
15 > /**
16 > * Extended storage type for AI Customization that includes built-in prompts
17 > * shipped with the application, alongside the core `PromptsStorage` values.
18 > */
19 > export type AICustomizationSource = 'local' | 'user' | 'extension' | 'plugin' | 'builtin';
20 >
21 > export namespace AICustomizationSources {
22 > export const local: AICustomizationSource = 'local';
23 > export const user: AICustomizationSource = 'user';
24 > export const extension: AICustomizationSource = 'extension';
25 > export const plugin: AICustomizationSource = 'plugin';
26 > export const builtin: AICustomizationSource = 'builtin';
27 > export const all: AICustomizationSource[] = [local, user, extension, plugin, builtin];
28 > }
29 >
30 > /**
31 > * Storage type discriminator for built-in customizations shipped with the application.
32 > */
33 > export const BUILTIN_STORAGE = AICustomizationSources.builtin;
34 >
35 > /**
36 > * Possible section IDs for the AI Customization Management Editor sidebar.
37 > */
38 > export const AICustomizationManagementSection = {
39 > Agents: 'agents',
40 > Skills: 'skills',
41 > Instructions: 'instructions',
42 > Prompts: 'prompts',
43 > Hooks: 'hooks',
44 > Automations: 'automations',
45 > McpServers: 'mcpServers',
46 > Plugins: 'plugins',
47 > Models: 'models',
48 > Tools: 'tools',
49 > HarnessSettings: 'harnessSettings',
50 > } as const;
51 >
52 > export type AICustomizationManagementSection = typeof AICustomizationManagementSection[keyof typeof AICustomizationManagementSection];
53 >
54 > /**
55 > * Per-type filter policy controlling which storage sources are visible
56 > * for a given customization type.
57 > */
58 > export interface IWelcomePageFeatures {
59 > /** Show the "Configure Your AI" getting-started banner. */
60 > readonly showGettingStartedBanner: boolean;
61 > }
62 >
63 > /**
64 > * Provides workspace context for AI Customization views.
65 > */
66 > export interface IAICustomizationWorkspaceService {
67 > readonly _serviceBrand: undefined;
68 >
69 > /**
70 > * Observable that fires when the active project root changes.
71 > */
72 > readonly activeProjectRoot: IObservable<URI | undefined>;
73 >
74 > /**
75 > * Returns the current active project root, if any.
76 > */
77 > getActiveProjectRoot(): URI | undefined;
78 >
79 > /**
80 > * The sections to show in the AI Customization Management Editor sidebar.
81 > */
82 > readonly managementSections: readonly AICustomizationManagementSection[];
83 >
84 > /**
85 > * Whether this is a sessions window (vs core VS Code).
86 > */
87 > readonly isSessionsWindow: boolean;
88 >
89 > /**
90 > * Controls which features are displayed on the welcome page.
91 > */
92 > readonly welcomePageFeatures: IWelcomePageFeatures;
93 >
94 > /**
95 > * Commits files in the active project.
96 > */
97 > commitFiles(projectRoot: URI, fileUris: URI[]): Promise<void>;
98 >
99 > /**
100 > * Commits the deletion of resources that have already been removed from disk.
101 > * The URIs may point to individual files or to directories (for example, when
102 > * deleting a skill, the entire customization folder is removed). Implementations
103 > * should ensure that directory deletions are handled recursively as needed.
104 > * In sessions this stages and commits the removal in the relevant repositories.
105 > */
106 > deleteFiles(projectRoot: URI, fileUris: URI[]): Promise<void>;
107 >
108 > /**
109 > * Launches the AI-guided creation flow for the given customization type.
110 > */
111 > generateCustomization(type: PromptsType): Promise<void>;
112 >
113 > /**
114 > * Whether a transient project root override is currently active.
115 > */
116 > readonly hasOverrideProjectRoot: IObservable<boolean>;
117 >
118 > /**
119 > * Sets a transient override for the active project root.
120 > * While set, `activeProjectRoot` returns this value instead of the
121 > * session- or workspace-derived root. Call `clearOverrideProjectRoot()` to revert.
122 > */
123 > setOverrideProjectRoot(root: URI): void;
124 >
125 > /**
126 > * Clears the transient project root override, reverting to the
127 > * session-derived (or workspace-derived) root.
128 > */
129 > clearOverrideProjectRoot(): void;
130 >
131 > /**
132 > * Returns prompt/skill slash commands filtered through the workspace
133 > * service's storage source policy, ensuring the results match the
134 > * customizations visible in the AI Customization views.
135 > */
136 > getFilteredPromptSlashCommands(token: CancellationToken): Promise<readonly IChatPromptSlashCommand[]>;
137 >
138 > /**
139 > * Returns a map of built-in skill names that have direct UI integrations
140 > * (toolbar buttons, menu items, etc.) to a tooltip describing the
141 > * integration. Used to display a 'UI Integration' badge in the
142 > * customizations editor, especially important when users override a
143 > * built-in skill that drives a UI surface.
144 > */
145 > getSkillUIIntegrations(): ReadonlyMap<string, string>;
146 > }
src/vs/workbench/contrib/chat/common/promptSyntax/service/extensionPromptFileService.ts 144 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensionPromptFileService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../../base/common/cancellation.js';
7 > import { CancellationError } from '../../../../../../base/common/errors.js';
8 > import { Emitter, Event } from '../../../../../../base/common/event.js';
9 > import { Disposable, DisposableStore, IDisposable } from '../../../../../../base/common/lifecycle.js';
10 > import { ResourceMap } from '../../../../../../base/common/map.js';
11 > import { URI } from '../../../../../../base/common/uri.js';
12 > import { IModelService } from '../../../../../../editor/common/services/model.js';
13 > import { ContextKeyExpr, IContextKeyService } from '../../../../../../platform/contextkey/common/contextkey.js';
14 > import { IExtensionDescription } from '../../../../../../platform/extensions/common/extensions.js';
15 > import { IFileService } from '../../../../../../platform/files/common/files.js';
16 > import { ILogService } from '../../../../../../platform/log/common/log.js';
17 > import { IExtensionService } from '../../../../../services/extensions/common/extensions.js';
18 > import { IFilesConfigurationService } from '../../../../../services/filesConfiguration/common/filesConfigurationService.js';
19 > import { getSkillFolderName } from '../config/promptFileLocations.js';
20 > import { ParsedPromptFile, PromptFileParser } from '../promptFileParser.js';
21 > import { PromptFileSource, PromptsType } from '../promptTypes.js';
22 > import {
23 > CUSTOM_AGENT_PROVIDER_ACTIVATION_EVENT,
24 > IExtensionPromptPath,
25 > INSTRUCTIONS_PROVIDER_ACTIVATION_EVENT,
26 > IPromptFileContext,
27 > IPromptFileResource,
28 > PROMPT_FILE_PROVIDER_ACTIVATION_EVENT,
29 > PromptsStorage,
30 > SKILL_PROVIDER_ACTIVATION_EVENT,
31 > } from './promptsService.js';
32 >
33 > /**
34 > * Event payload emitted by {@link ExtensionPromptFileService.onDidChange}.
35 > */
36 > export interface IExtensionPromptFilesChangeEvent {
37 > readonly type: PromptsType;
38 > }
39 >
40 > type PromptFileProviderEntry = {
41 > readonly extension: IExtensionDescription;
42 > readonly type: PromptsType;
43 > readonly onDidChangePromptFiles?: Event<void>;
44 > readonly providePromptFiles: (context: IPromptFileContext, token: CancellationToken) => Promise<IPromptFileResource[] | undefined>;
45 > };
46 >
47 > const ALL_PROMPT_TYPES: readonly PromptsType[] = [
48 > PromptsType.prompt,
49 > PromptsType.instructions,
50 > PromptsType.agent,
51 > PromptsType.skill,
52 > PromptsType.hook,
53 > ];
54 >
55 > /**
56 > * Owns the registry of prompt files contributed by extensions, both via
57 > * static contribution points (see {@link registerContributedFile}) and via
58 > * dynamic providers registered through the proposed extension API (see
59 > * {@link registerPromptFileProvider}).
60 > *
61 > * Exposes a per-type getter ({@link getExtensionPromptFiles}) that merges
62 > * both sources and applies any `when` clauses, plus a single change event
63 > * ({@link onDidChange}) carrying the affected {@link PromptsType}.
64 > */
65 > export class ExtensionPromptFileService extends Disposable {
66 >
67 > /**
68 > * Files contributed via extension contribution points, keyed by type then URI.
69 > */
70 > private readonly contributedFiles = {
71 > [PromptsType.prompt]: new ResourceMap<Promise<IExtensionPromptPath>>(),
72 > [PromptsType.instructions]: new ResourceMap<Promise<IExtensionPromptPath>>(),
73 > [PromptsType.agent]: new ResourceMap<Promise<IExtensionPromptPath>>(),
74 > [PromptsType.skill]: new ResourceMap<Promise<IExtensionPromptPath>>(),
75 > [PromptsType.hook]: new ResourceMap<Promise<IExtensionPromptPath>>(),
76 > };
77 >
78 > /**
79 > * Providers registered via the proposed extension API.
80 > */
81 > private readonly _promptFileProviders: PromptFileProviderEntry[] = [];
82 >
83 > /**
84 > * Context keys referenced by tracked `when` clauses (from contributed
85 > * files and provider results). Used to know when to re-evaluate.
86 > */
87 > private readonly _contributedWhenKeys = new Set<string>();
88 > private readonly _contributedWhenClauses = new Map<string, string>();
89 > private readonly _providerWhenClauses = new Map<PromptFileProviderEntry, readonly string[]>();
90 >
91 > private readonly _onDidChange = this._register(new Emitter<IExtensionPromptFilesChangeEvent>());
92 > public readonly onDidChange: Event<IExtensionPromptFilesChangeEvent> = this._onDidChange.event;
93 >
94 > /**
95 > * Pending URIs to mark as readonly, flushed on the next microtask.
96 > * Batches multiple `registerContributedFile` calls (which happen
97 > * synchronously in the extension point handler) into a single
98 > * `updateReadonly` call to avoid firing `onDidChangeReadonly` per file.
99 > */
100 > private _pendingReadonlyUris: URI[] = [];
101 > private _pendingReadonlyFlush = false;
102 >
103 > constructor(
104 @ILogService private readonly logger: ILogService,
105 @IFileService private readonly fileService: IFileService,
122 }));
123 }
125 > /**
126 > * Returns the merged list of extension-contributed prompt files for the
127 > * given type, filtered by their `when` clause.
128 > */
129 > public async getExtensionPromptFiles(type: PromptsType, token: CancellationToken): Promise<readonly IExtensionPromptPath[]> {
130 await this.extensionService.whenInstalledExtensionsRegistered();
131 const settledResults = await Promise.allSettled(this.contributedFiles[type].values());
149 });
150 }
152 > /**
153 > * Registers a file contributed via a static contribution point. Returns
154 > * a disposable that removes the contribution.
155 > */
156 > public registerContributedFile(type: PromptsType, uri: URI, extension: IExtensionDescription, name?: string, description?: string, when?: string, sessionTypes?: readonly string[]): IDisposable {
157 const bucket = this.contributedFiles[type];
158 if (bucket.has(uri)) {
198 };
199 }
201 > /**
202 > * Registers a prompt file provider (CustomAgentProvider, InstructionsProvider, or PromptFileProvider).
203 > * This is called by the extension host bridge when an extension registers a provider via
204 > * vscode.chat.registerCustomAgentProvider(), registerInstructionsProvider(), or
205 > * registerPromptFileProvider().
206 > */
207 > public registerPromptFileProvider(extension: IExtensionDescription, type: PromptsType, provider: {
208 onDidChangePromptFiles?: Event<void>;
209 providePromptFiles: (context: IPromptFileContext, token: CancellationToken) => Promise<IPromptFileResource[] | undefined>;
236 return disposables;
237 }
239 > private async _listFromProviders(type: PromptsType, activationEvent: string, token: CancellationToken): Promise<IExtensionPromptPath[]> {
240 const result: IExtensionPromptPath[] = [];
241 const readonlyUris: URI[] = [];
284 return result;
285 }
287 > private _getProviderActivationEvent(type: PromptsType): string | undefined {
288 switch (type) {
289 case PromptsType.agent:
299 }
300 }
302 > private _enqueueReadonlyUpdate(uri: URI): void {
303 this._pendingReadonlyUris.push(uri);
304 if (!this._pendingReadonlyFlush) {
312 }
313 }
315 > private _updateContributedWhenKeys(): void {
316 this._contributedWhenKeys.clear();
317 for (const whenClause of this._contributedWhenClauses.values()) {
330 }
331 }
333 > // Skill validation
334 >
335 > private async _validateAndSanitizeSkillFile(uri: URI, token: CancellationToken): Promise<{ name: string; description: string | undefined }> {
336 const parsedFile = await this._parsePromptFile(uri, token);
337 const folderName = getSkillFolderName(uri);
357 return { name: sanitizedName, description: sanitizedDescription };
358 }
360 > private async _parsePromptFile(uri: URI, token: CancellationToken): Promise<ParsedPromptFile> {
361 const model = this.modelService.getModel(uri);
362 if (model) {
369 return new PromptFileParser().parse(uri, fileContent.value.toString());
370 }
372 > private _sanitizeAgentSkillText(text: string): string {
373 // Remove XML tags
374 return text.replace(/<[^>]+>/g, '');
375 }
377 > private _truncateAgentSkillName(name: string, uri: URI): string {
378 const MAX_NAME_LENGTH = 64;
379 const sanitized = this._sanitizeAgentSkillText(name);
387 return sanitized;
388 }
390 > private _truncateAgentSkillDescription(description: string, uri: URI): string {
391 const MAX_DESCRIPTION_LENGTH = 1024;
392 const sanitized = this._sanitizeAgentSkillText(description);
src/vs/platform/markers/common/markers.ts 140 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- markers.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { IDisposable } from '../../../base/common/lifecycle.js';
8 > import Severity from '../../../base/common/severity.js';
9 > import { URI } from '../../../base/common/uri.js';
10 > import { localize } from '../../../nls.js';
11 > import { createDecorator } from '../../instantiation/common/instantiation.js';
12 >
13 > export interface IMarkerReadOptions {
14 > owner?: string;
15 > resource?: URI;
16 > severities?: number;
17 > take?: number;
18 > ignoreResourceFilters?: boolean;
19 > }
20 >
21 > export interface IMarkerService {
22 > readonly _serviceBrand: undefined;
23 >
24 > getStatistics(): MarkerStatistics;
25 >
26 > changeOne(owner: string, resource: URI, markers: IMarkerData[]): void;
27 >
28 > changeAll(owner: string, data: IResourceMarker[]): void;
29 >
30 > remove(owner: string, resources: URI[]): void;
31 >
32 > read(filter?: IMarkerReadOptions): IMarker[];
33 >
34 > installResourceFilter(resource: URI, reason: string): IDisposable;
35 >
36 > readonly onMarkerChanged: Event<readonly URI[]>;
37 > }
38 >
39 > /**
40 > *
41 > */
42 > export interface IRelatedInformation {
43 > resource: URI;
44 > message: string;
45 > startLineNumber: number;
46 > startColumn: number;
47 > endLineNumber: number;
48 > endColumn: number;
49 > }
50 >
51 > export const enum MarkerTag {
52 > Unnecessary = 1,
53 > Deprecated = 2
54 > }
55 >
56 > export enum MarkerSeverity {
57 > Hint = 1,
58 > Info = 2,
59 > Warning = 4,
60 > Error = 8,
61 > }
62 >
63 > export namespace MarkerSeverity {
64 >
65 > export function compare(a: MarkerSeverity, b: MarkerSeverity): number {
66 return b - a;
67 }
68 > markers.ts
69 > const _displayStrings: { [value: number]: string } = Object.create(null);
70 > _displayStrings[MarkerSeverity.Error] = localize('sev.error', "Error");
71 > _displayStrings[MarkerSeverity.Warning] = localize('sev.warning', "Warning");
72 > _displayStrings[MarkerSeverity.Info] = localize('sev.info', "Info");
73 >
74 > export function toString(a: MarkerSeverity): string {
75 return _displayStrings[a] || '';
76 }
77 > markers.ts
78 > const _displayStringsPlural: { [value: number]: string } = Object.create(null);
79 > _displayStringsPlural[MarkerSeverity.Error] = localize('sev.errors', "Errors");
80 > _displayStringsPlural[MarkerSeverity.Warning] = localize('sev.warnings', "Warnings");
81 > _displayStringsPlural[MarkerSeverity.Info] = localize('sev.infos', "Infos");
82 >
83 > export function toStringPlural(a: MarkerSeverity): string {
84 return _displayStringsPlural[a] || '';
85 }
86 > markers.ts
87 > export function fromSeverity(severity: Severity): MarkerSeverity {
88 switch (severity) {
89 case Severity.Error: return MarkerSeverity.Error;
93 }
94 }
95 > markers.ts
96 > export function toSeverity(severity: MarkerSeverity): Severity {
97 switch (severity) {
98 case MarkerSeverity.Error: return Severity.Error;
102 }
103 }
104 > } markers.ts
105 >
106 > /**
107 > * A structure defining a problem/warning/etc.
108 > */
109 > export interface IMarkerData {
110 > code?: string | { value: string; target: URI };
111 > severity: MarkerSeverity;
112 > message: string;
113 > source?: string;
114 > startLineNumber: number;
115 > startColumn: number;
116 > endLineNumber: number;
117 > endColumn: number;
118 > modelVersionId?: number;
119 > relatedInformation?: IRelatedInformation[];
120 > tags?: MarkerTag[];
121 > origin?: string | undefined;
122 > }
123 >
124 > export interface IResourceMarker {
125 > resource: URI;
126 > marker: IMarkerData;
127 > }
128 >
129 > export interface IMarker {
130 > owner: string;
131 > resource: URI;
132 > severity: MarkerSeverity;
133 > code?: string | { value: string; target: URI };
134 > message: string;
135 > source?: string;
136 > startLineNumber: number;
137 > startColumn: number;
138 > endLineNumber: number;
139 > endColumn: number;
140 > modelVersionId?: number;
141 > relatedInformation?: IRelatedInformation[];
142 > tags?: MarkerTag[];
143 > origin?: string | undefined;
144 > }
145 >
146 > export interface MarkerStatistics {
147 > errors: number;
148 > warnings: number;
149 > infos: number;
150 > unknowns: number;
151 > }
152 >
153 > export namespace IMarkerData {
154 > const emptyString = '';
155 > export function makeKey(markerData: IMarkerData): string {
156 return makeKeyOptionalMessage(markerData, true);
157 }
158 > markers.ts
159 > export function makeKeyOptionalMessage(markerData: IMarkerData, useMessage: boolean): string {
160 const result: string[] = [emptyString];
161 if (markerData.source) {
209 return result.join('¦');
210 }
211 > } markers.ts
212 >
213 > export const IMarkerService = createDecorator<IMarkerService>('markerService');
src/vs/platform/opener/common/opener.ts 139 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- opener.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../base/common/cancellation.js';
7 > import { IDisposable } from '../../../base/common/lifecycle.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { IEditorOptions, ITextEditorSelection } from '../../editor/common/editor.js';
10 > import { createDecorator } from '../../instantiation/common/instantiation.js';
11 >
12 > export const IOpenerService = createDecorator<IOpenerService>('openerService');
13 >
14 > export type OpenInternalOptions = {
15 >
16 > /**
17 > * Signals that the intent is to open an editor to the side
18 > * of the currently active editor.
19 > */
20 > readonly openToSide?: boolean;
21 >
22 > /**
23 > * Extra editor options to apply in case an editor is used to open.
24 > */
25 > readonly editorOptions?: IEditorOptions;
26 >
27 > /**
28 > * Signals that the editor to open was triggered through a user
29 > * action, such as keyboard or mouse usage.
30 > */
31 > readonly fromUserGesture?: boolean;
32 >
33 > /**
34 > * Allow command links to be handled.
35 > *
36 > * If this is an array, then only the commands included in the array can be run.
37 > */
38 > readonly allowCommands?: boolean | readonly string[];
39 > };
40 >
41 > export type OpenExternalOptions = {
42 > readonly openExternal?: boolean;
43 > readonly allowTunneling?: boolean;
44 > readonly allowContributedOpeners?: boolean | string;
45 > readonly fromWorkspace?: boolean;
46 > readonly skipValidation?: boolean;
47 > };
48 >
49 > export type OpenOptions = OpenInternalOptions & OpenExternalOptions;
50 >
51 > export type ResolveExternalUriOptions = { readonly allowTunneling?: boolean };
52 >
53 > export interface IResolvedExternalUri extends IDisposable {
54 > resolved: URI;
55 > }
56 >
57 > export interface IOpener {
58 > open(resource: URI | string, options?: OpenInternalOptions | OpenExternalOptions): Promise<boolean>;
59 > }
60 >
61 > export interface IExternalOpener {
62 > openExternal(href: string, ctx: { sourceUri: URI; preferredOpenerId?: string }, token: CancellationToken): Promise<boolean>;
63 > dispose?(): void;
64 > }
65 >
66 > export interface IValidator {
67 > shouldOpen(resource: URI | string, openOptions?: OpenOptions): Promise<boolean>;
68 > }
69 >
70 > export interface IExternalUriResolver {
71 > resolveExternalUri(resource: URI, options?: OpenOptions): Promise<{ resolved: URI; dispose(): void } | undefined>;
72 > }
73 >
74 > export interface IOpenerService {
75 >
76 > readonly _serviceBrand: undefined;
77 >
78 > /**
79 > * Register a participant that can handle the open() call.
80 > */
81 > registerOpener(opener: IOpener): IDisposable;
82 >
83 > /**
84 > * Register a participant that can validate if the URI resource be opened.
85 > * Validators are run before openers.
86 > */
87 > registerValidator(validator: IValidator): IDisposable;
88 >
89 > /**
90 > * Register a participant that can resolve an external URI resource to be opened.
91 > */
92 > registerExternalUriResolver(resolver: IExternalUriResolver): IDisposable;
93 >
94 > /**
95 > * Sets the handler for opening externally. If not provided,
96 > * a default handler will be used.
97 > */
98 > setDefaultExternalOpener(opener: IExternalOpener): void;
99 >
100 > /**
101 > * Registers a new opener external resources openers.
102 > */
103 > registerExternalOpener(opener: IExternalOpener): IDisposable;
104 >
105 > /**
106 > * Opens a resource, like a webaddress, a document uri, or executes command.
107 > *
108 > * @param resource A resource
109 > * @return A promise that resolves when the opening is done.
110 > */
111 > open(resource: URI | string, options?: OpenInternalOptions | OpenExternalOptions): Promise<boolean>;
112 >
113 > /**
114 > * Resolve a resource to its external form.
115 > * @throws whenever resolvers couldn't resolve this resource externally.
116 > */
117 > resolveExternalUri(resource: URI, options?: ResolveExternalUriOptions): Promise<IResolvedExternalUri>;
118 > }
119 >
120 > /**
121 > * Encodes selection into the `URI`.
122 > *
123 > * IMPORTANT: you MUST use `extractSelection` to separate the selection
124 > * again from the original `URI` before passing the `URI` into any
125 > * component that is not aware of selections.
126 > */
127 > export function withSelection(uri: URI, selection: ITextEditorSelection): URI {
128 return uri.with({ fragment: `${selection.startLineNumber},${selection.startColumn}${selection.endLineNumber ? `-${selection.endLineNumber}${selection.endColumn ? `,${selection.endColumn}` : ''}` : ''}` });
129 }
130 > opener.ts
131 > /**
132 > * file:///some/file.js#73
133 > * file:///some/file.js#L73
134 > * file:///some/file.js#73,84
135 > * file:///some/file.js#L73,84
136 > * file:///some/file.js#73-83
137 > * file:///some/file.js#L73-L83
138 > * file:///some/file.js#73,84-83,52
139 > * file:///some/file.js#L73,84-L83,52
140 > */
141 > export function extractSelection(uri: URI): { selection: ITextEditorSelection | undefined; uri: URI } {
142 let selection: ITextEditorSelection | undefined = undefined;
143 const match = /^L?(\d+)(?:,(\d+))?(-L?(\d+)(?:,(\d+))?)?/.exec(uri.fragment);
src/vs/workbench/services/userDataProfile/common/userDataProfile.ts 139 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- userDataProfile.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isUndefined } from '../../../../base/common/types.js';
7 > import { Event } from '../../../../base/common/event.js';
8 > import { localize, localize2 } from '../../../../nls.js';
9 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
10 > import { IUserDataProfile, IUserDataProfileOptions, IUserDataProfileUpdateOptions, ProfileResourceType, ProfileResourceTypeFlags } from '../../../../platform/userDataProfile/common/userDataProfile.js';
11 > import { RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
12 > import { URI } from '../../../../base/common/uri.js';
13 > import { registerIcon } from '../../../../platform/theme/common/iconRegistry.js';
14 > import { Codicon } from '../../../../base/common/codicons.js';
15 > import { ITreeItem, ITreeItemLabel } from '../../../common/views.js';
16 > import { CancellationToken } from '../../../../base/common/cancellation.js';
17 > import { IDisposable } from '../../../../base/common/lifecycle.js';
18 > import { IProductService } from '../../../../platform/product/common/productService.js';
19 >
20 > export interface DidChangeUserDataProfileEvent {
21 > readonly previous: IUserDataProfile;
22 > readonly profile: IUserDataProfile;
23 > join(promise: Promise<void>): void;
24 > }
25 >
26 > export const IUserDataProfileService = createDecorator<IUserDataProfileService>('IUserDataProfileService');
27 > export interface IUserDataProfileService {
28 > readonly _serviceBrand: undefined;
29 > readonly currentProfile: IUserDataProfile;
30 > readonly onDidChangeCurrentProfile: Event<DidChangeUserDataProfileEvent>;
31 > updateCurrentProfile(currentProfile: IUserDataProfile): Promise<void>;
32 > }
33 >
34 > export interface IProfileTemplateInfo {
35 > readonly name: string;
36 > readonly url: string;
37 > }
38 >
39 > export const IUserDataProfileManagementService = createDecorator<IUserDataProfileManagementService>('IUserDataProfileManagementService');
40 > export interface IUserDataProfileManagementService {
41 > readonly _serviceBrand: undefined;
42 >
43 > createProfile(name: string, options?: IUserDataProfileOptions): Promise<IUserDataProfile>;
44 > createAndEnterProfile(name: string, options?: IUserDataProfileOptions): Promise<IUserDataProfile>;
45 > createAndEnterTransientProfile(): Promise<IUserDataProfile>;
46 > removeProfile(profile: IUserDataProfile): Promise<void>;
47 > updateProfile(profile: IUserDataProfile, updateOptions: IUserDataProfileUpdateOptions): Promise<IUserDataProfile>;
48 > switchProfile(profile: IUserDataProfile): Promise<void>;
49 > getBuiltinProfileTemplates(): Promise<IProfileTemplateInfo[]>;
50 > getDefaultProfileToUse(): IUserDataProfile;
51 > }
52 >
53 > export interface IUserDataProfileTemplate {
54 > readonly name: string;
55 > readonly icon?: string;
56 > readonly settings?: string;
57 > readonly keybindings?: string;
58 > readonly tasks?: string;
59 > readonly snippets?: string;
60 > readonly globalState?: string;
61 > readonly extensions?: string;
62 > readonly mcp?: string;
63 > }
64 >
65 > export function isUserDataProfileTemplate(thing: unknown): thing is IUserDataProfileTemplate {
66 const candidate = thing as IUserDataProfileTemplate | undefined;
67
72 && (isUndefined(candidate.mcp) || typeof candidate.mcp === 'string'));
73 }
75 > export const PROFILE_URL_AUTHORITY = 'profile';
76 > export function toUserDataProfileUri(path: string, productService: IProductService): URI {
77 return URI.from({
78 scheme: productService.urlProtocol,
81 });
82 }
84 > export const PROFILE_URL_AUTHORITY_PREFIX = 'profile-';
85 > export function isProfileURL(uri: URI): boolean {
86 return uri.authority === PROFILE_URL_AUTHORITY || new RegExp(`^${PROFILE_URL_AUTHORITY_PREFIX}`).test(uri.authority);
87 }
89 > export interface IUserDataProfileCreateOptions extends IUserDataProfileOptions {
90 > readonly name?: string;
91 > readonly resourceTypeFlags?: ProfileResourceTypeFlags;
92 > }
93 >
94 > export interface IProfileImportOptions extends IUserDataProfileCreateOptions {
95 > readonly name?: string;
96 > readonly icon?: string;
97 > readonly mode?: 'apply';
98 > }
99 >
100 > export const IUserDataProfileImportExportService = createDecorator<IUserDataProfileImportExportService>('IUserDataProfileImportExportService');
101 > export interface IUserDataProfileImportExportService {
102 > readonly _serviceBrand: undefined;
103 >
104 > registerProfileContentHandler(id: string, profileContentHandler: IUserDataProfileContentHandler): IDisposable;
105 > unregisterProfileContentHandler(id: string): void;
106 >
107 > resolveProfileTemplate(uri: URI): Promise<IUserDataProfileTemplate | null>;
108 > exportProfile(profile: IUserDataProfile, exportFlags?: ProfileResourceTypeFlags): Promise<void>;
109 > createFromProfile(from: IUserDataProfile, options: IUserDataProfileCreateOptions, token: CancellationToken): Promise<IUserDataProfile | undefined>;
110 > createProfileFromTemplate(profileTemplate: IUserDataProfileTemplate, options: IUserDataProfileCreateOptions, token: CancellationToken): Promise<IUserDataProfile | undefined>;
111 > createTroubleshootProfile(): Promise<void>;
112 > }
113 >
114 > export interface IProfileResourceInitializer {
115 > initialize(content: string): Promise<void>;
116 > }
117 >
118 > export interface IProfileResource {
119 > getContent(profile: IUserDataProfile): Promise<string>;
120 > apply(content: string, profile: IUserDataProfile): Promise<void>;
121 > }
122 >
123 > export interface IProfileResourceTreeItem extends ITreeItem {
124 > readonly type: ProfileResourceType;
125 > readonly label: ITreeItemLabel;
126 > isFromDefaultProfile(): boolean;
127 > getChildren(): Promise<IProfileResourceChildTreeItem[] | undefined>;
128 > getContent(): Promise<string>;
129 > }
130 >
131 > export interface IProfileResourceChildTreeItem extends ITreeItem {
132 > parent: IProfileResourceTreeItem;
133 > }
134 >
135 > export interface ISaveProfileResult {
136 > readonly id: string;
137 > readonly link: URI;
138 > }
139 >
140 > export interface IUserDataProfileContentHandler {
141 > readonly name: string;
142 > readonly description?: string;
143 > readonly extensionId?: string;
144 > saveProfile(name: string, content: string, token: CancellationToken): Promise<ISaveProfileResult | null>;
145 > readProfile(idOrUri: string | URI, token: CancellationToken): Promise<string | null>;
146 > }
147 >
148 > export const defaultUserDataProfileIcon = registerIcon('defaultProfile-icon', Codicon.settings, localize('defaultProfileIcon', 'Icon for Default Profile.'));
149 >
150 > export const PROFILES_TITLE = localize2('profiles', 'Profiles');
151 > export const PROFILES_CATEGORY = { ...PROFILES_TITLE };
152 > export const PROFILE_EXTENSION = 'code-profile';
153 > export const PROFILE_FILTER = [{ name: localize('profile', "Profile"), extensions: [PROFILE_EXTENSION] }];
154 > export const CURRENT_PROFILE_CONTEXT = new RawContextKey<string>('currentProfile', '');
155 > export const HAS_PROFILES_CONTEXT = new RawContextKey<boolean>('hasProfiles', false);
src/vs/workbench/contrib/chat/test/common/chatService/mockChatService.ts 138 covered LOC · 35 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockChatService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../../base/common/cancellation.js';
7 > import { Emitter, Event } from '../../../../../../base/common/event.js';
8 > import { ResourceMap } from '../../../../../../base/common/map.js';
9 > import { ISettableObservable, observableValue } from '../../../../../../base/common/observable.js';
10 > import { URI } from '../../../../../../base/common/uri.js';
11 > import { ChatRequestQueueKind, ChatSendResult, IChatDetail, IChatModelReference, IChatProgress, IChatSendRequestOptions, IChatService, IChatSessionStartOptions, IChatUserActionEvent, IRemotePendingRequest } from '../../../common/chatService/chatService.js';
12 > import { ChatAgentLocation } from '../../../common/constants.js';
13 > import { IChatModel, IChatRequestModel, IExportableChatData, ISerializableChatData } from '../../../common/model/chatModel.js';
14 > import type { IChatModelReferenceDebugSnapshot } from '../../../common/model/chatModelStore.js';
15 >
16 > export class MockChatService implements IChatService {
17 > private readonly _chatModels: ISettableObservable<Iterable<IChatModel>> = observableValue('chatModels', []); mockChatService.ts
18 > readonly chatModels = this._chatModels;
19 > requestInProgressObs = observableValue('name', false);
20 > _serviceBrand: undefined;
21 > editingSessions = [];
22 > transferredSessionResource = undefined;
23 > readonly onDidSubmitRequest = Event.None;
24 >
25 > private readonly _onDidCreateModel = new Emitter<IChatModel>();
26 > readonly onDidCreateModel = this._onDidCreateModel.event;
27 >
28 > private readonly sessions = new ResourceMap<IChatModel>();
29 > private liveSessionItems: IChatDetail[] = [];
30 > private historySessionItems: IChatDetail[] = [];
31 >
32 > private readonly _onDidDisposeSession = new Emitter<{ sessionResources: URI[]; reason: 'cleared' }>();
33 > readonly onDidDisposeSession = this._onDidDisposeSession.event;
34 >
35 > fireDidDisposeSession(sessionResources: URI[]): void {
36 > this._onDidDisposeSession.fire({ sessionResources, reason: 'cleared' });
37 > }
38 >
39 > setSaveModelsEnabled(enabled: boolean): void {
40
41 }
43 > processPendingRequests(sessionResource: URI): void {
44
45 }
47 > setLiveSessionItems(items: IChatDetail[]): void {
48 this.liveSessionItems = items;
49 }
51 > setHistorySessionItems(items: IChatDetail[]): void {
52 this.historySessionItems = items;
53 }
55 > addSession(session: IChatModel): void {
56 this.sessions.set(session.sessionResource, session);
57 // Update the chatModels observable
59 this._onDidCreateModel.fire(session);
60 }
62 > removeSession(sessionResource: URI): void {
63 this.sessions.delete(sessionResource);
64 // Update the chatModels observable
65 this._chatModels.set([...this.sessions.values()], undefined);
66 }
68 > isEnabled(_location: ChatAgentLocation): boolean {
69 return true;
70 }
72 > hasSessions(): boolean {
73 return this.sessions.size > 0;
74 }
76 > getProviderInfos() {
77 return [];
78 }
80 > startNewLocalSession(_location: ChatAgentLocation, _options?: IChatSessionStartOptions): IChatModelReference {
81 throw new Error('Method not implemented.');
82 }
84 > getSession(sessionResource: URI): IChatModel | undefined {
85 return this.sessions.get(sessionResource);
86 }
88 > getLatestRequest(): IChatRequestModel | undefined {
89 return undefined;
90 }
92 > acquireOrRestoreSession(_sessionResource: URI): Promise<IChatModelReference | undefined> {
93 throw new Error('Method not implemented.');
94 }
96 > getSessionTitle(_sessionResource: URI): string | undefined {
97 return undefined;
98 }
100 > loadSessionFromData(data: IExportableChatData | ISerializableChatData, _debugOwner?: string): IChatModelReference {
101 throw new Error('Method not implemented.');
102 }
104 > getChatModelReferenceDebugInfo(): IChatModelReferenceDebugSnapshot {
105 return { totalModels: 0, totalReferences: 0, models: [] };
106 }
108 > acquireOrLoadSession(_resource: URI, _position: ChatAgentLocation, _token: CancellationToken, _debugOwner?: string): Promise<IChatModelReference | undefined> {
109 throw new Error('Method not implemented.');
110 }
112 > acquireExistingSession(_sessionResource: URI, _debugOwner?: string): IChatModelReference | undefined {
113 > return undefined; mockChatService.ts
114 > }
116 > setSessionTitle(_sessionResource: URI, _title: string): void { }
117 >
118 > appendProgress(_request: IChatRequestModel, _progress: IChatProgress): void { }
119 >
120 > sendRequest(_sessionResource: URI, _message: string): Promise<ChatSendResult> {
121 throw new Error('Method not implemented.');
122 }
124 > resendRequest(_request: IChatRequestModel, _options?: IChatSendRequestOptions): Promise<void> {
125 throw new Error('Method not implemented.');
126 }
128 > adoptRequest(_sessionResource: URI, _request: IChatRequestModel): Promise<void> {
129 throw new Error('Method not implemented.');
130 }
132 > removeRequest(_sessionResource: URI, _requestId: string): Promise<void> {
133 throw new Error('Method not implemented.');
134 }
136 > async cancelCurrentRequestForSession(_sessionResource: URI, _source?: string): Promise<void> { }
137 >
138 > migrateRequests(_originalResource: URI, _targetResource: URI): void { }
139 >
140 > setYieldRequested(_sessionResource: URI): void { }
141 >
142 > removePendingRequest(_sessionResource: URI, _requestId: string): void { }
143 >
144 > setPendingRequests(_sessionResource: URI, _requests: readonly { requestId: string; kind: ChatRequestQueueKind }[]): void { }
145 >
146 > syncPendingRequestsFromRemote(_sessionResource: URI, _requests: readonly IRemotePendingRequest[]): void { }
147 >
148 > sendPendingRequestImmediately(_sessionResource: URI, _requestId: string): Promise<void> {
149 throw new Error('Method not implemented.');
150 }
152 > addCompleteRequest(): void { }
153 >
154 > async getLocalSessionHistory(): Promise<IChatDetail[]> {
155 return this.historySessionItems;
156 }
158 > async clearAllHistoryEntries(): Promise<void> { }
159 >
160 > async removeHistoryEntry(_resource: URI): Promise<void> { }
161 >
162 > readonly onDidPerformUserAction = Event.None;
163 >
164 > notifyUserAction(_event: IChatUserActionEvent): void { }
165 >
166 > readonly onDidReceiveQuestionCarouselAnswer = Event.None;
168 > notifyQuestionCarouselAnswer(_requestId: string, _resolveId: string, _answers: Record<string, unknown> | undefined): void { }
169 >
170 > async transferChatSession(): Promise<void> { }
171 >
172 > setChatSessionTitle(): void { }
173 >
174 > isEditingLocation(_location: ChatAgentLocation): boolean {
175 return false;
176 }
178 > getChatStorageFolder(): URI {
179 return URI.file('/tmp');
180 }
182 > logChatIndex(): void { }
183 >
184 > activateDefaultAgent(_location: ChatAgentLocation): Promise<void> {
185 return Promise.resolve();
186 }
188 > async getLiveSessionItems(): Promise<IChatDetail[]> {
189 return this.liveSessionItems;
190 }
192 > async getHistorySessionItems(): Promise<IChatDetail[]> {
193 return this.historySessionItems;
194 }
196 > waitForModelDisposals(): Promise<void> {
197 return Promise.resolve();
198 }
200 > getMetadataForSession(sessionResource: URI): Promise<IChatDetail | undefined> {
201 return Promise.resolve(this.liveSessionItems.find(item => item.sessionResource.toString() === sessionResource.toString()));
202 }
src/vs/base/common/keybindingLabels.ts 137 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- keybindingLabels.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Modifiers } from './keybindings.js';
7 > import { OperatingSystem } from './platform.js';
8 > import * as nls from '../../nls.js';
9 >
10 > export interface ModifierLabels {
11 > readonly ctrlKey: string;
12 > readonly shiftKey: string;
13 > readonly altKey: string;
14 > readonly metaKey: string;
15 > readonly separator: string;
16 > }
17 >
18 > export interface KeyLabelProvider<T extends Modifiers> {
19 > (keybinding: T): string | null;
20 > }
21 >
22 > export class ModifierLabelProvider {
23 >
24 > public readonly modifierLabels: ModifierLabels[];
25 >
26 > constructor(mac: ModifierLabels, windows: ModifierLabels, linux: ModifierLabels = windows) {
27 > this.modifierLabels = [null!]; // index 0 will never me accessed.
28 > this.modifierLabels[OperatingSystem.Macintosh] = mac;
29 > this.modifierLabels[OperatingSystem.Windows] = windows;
30 > this.modifierLabels[OperatingSystem.Linux] = linux;
31 > }
32 >
33 > public toLabel<T extends Modifiers>(OS: OperatingSystem, chords: readonly T[], keyLabelProvider: KeyLabelProvider<T>): string | null {
34 if (chords.length === 0) {
35 return null;
48 return result.join(' ');
49 }
51 >
52 > /**
53 > * A label provider that prints modifiers in a suitable format for displaying in the UI.
54 > */
55 > export const UILabelProvider = new ModifierLabelProvider(
56 > {
57 > ctrlKey: '\u2303',
58 > shiftKey: '⇧',
59 > altKey: '⌥',
60 > metaKey: '⌘',
61 > separator: '',
62 > },
63 > {
64 > ctrlKey: nls.localize({ key: 'ctrlKey', comment: ['This is the short form for the Control key on the keyboard'] }, "Ctrl"),
65 > shiftKey: nls.localize({ key: 'shiftKey', comment: ['This is the short form for the Shift key on the keyboard'] }, "Shift"),
66 > altKey: nls.localize({ key: 'altKey', comment: ['This is the short form for the Alt key on the keyboard'] }, "Alt"),
67 > metaKey: nls.localize({ key: 'windowsKey', comment: ['This is the short form for the Windows key on the keyboard'] }, "Windows"),
68 > separator: '+',
69 > },
70 > {
71 > ctrlKey: nls.localize({ key: 'ctrlKey', comment: ['This is the short form for the Control key on the keyboard'] }, "Ctrl"),
72 > shiftKey: nls.localize({ key: 'shiftKey', comment: ['This is the short form for the Shift key on the keyboard'] }, "Shift"),
73 > altKey: nls.localize({ key: 'altKey', comment: ['This is the short form for the Alt key on the keyboard'] }, "Alt"),
74 > metaKey: nls.localize({ key: 'superKey', comment: ['This is the short form for the Super key on the keyboard'] }, "Super"),
75 > separator: '+',
76 > }
77 > );
78 >
79 > /**
80 > * A label provider that prints modifiers in a suitable format for ARIA.
81 > */
82 > export const AriaLabelProvider = new ModifierLabelProvider(
83 > {
84 > ctrlKey: nls.localize({ key: 'ctrlKey.long', comment: ['This is the long form for the Control key on the keyboard'] }, "Control"),
85 > shiftKey: nls.localize({ key: 'shiftKey.long', comment: ['This is the long form for the Shift key on the keyboard'] }, "Shift"),
86 > altKey: nls.localize({ key: 'optKey.long', comment: ['This is the long form for the Alt/Option key on the keyboard'] }, "Option"),
87 > metaKey: nls.localize({ key: 'cmdKey.long', comment: ['This is the long form for the Command key on the keyboard'] }, "Command"),
88 > separator: '+',
89 > },
90 > {
91 > ctrlKey: nls.localize({ key: 'ctrlKey.long', comment: ['This is the long form for the Control key on the keyboard'] }, "Control"),
92 > shiftKey: nls.localize({ key: 'shiftKey.long', comment: ['This is the long form for the Shift key on the keyboard'] }, "Shift"),
93 > altKey: nls.localize({ key: 'altKey.long', comment: ['This is the long form for the Alt key on the keyboard'] }, "Alt"),
94 > metaKey: nls.localize({ key: 'windowsKey.long', comment: ['This is the long form for the Windows key on the keyboard'] }, "Windows"),
95 > separator: '+',
96 > },
97 > {
98 > ctrlKey: nls.localize({ key: 'ctrlKey.long', comment: ['This is the long form for the Control key on the keyboard'] }, "Control"),
99 > shiftKey: nls.localize({ key: 'shiftKey.long', comment: ['This is the long form for the Shift key on the keyboard'] }, "Shift"),
100 > altKey: nls.localize({ key: 'altKey.long', comment: ['This is the long form for the Alt key on the keyboard'] }, "Alt"),
101 > metaKey: nls.localize({ key: 'superKey.long', comment: ['This is the long form for the Super key on the keyboard'] }, "Super"),
102 > separator: '+',
103 > }
104 > );
105 >
106 > /**
107 > * A label provider that prints modifiers in a suitable format for Electron Accelerators.
108 > * See https://github.com/electron/electron/blob/master/docs/api/accelerator.md
109 > */
110 > export const ElectronAcceleratorLabelProvider = new ModifierLabelProvider(
111 > {
112 > ctrlKey: 'Ctrl',
113 > shiftKey: 'Shift',
114 > altKey: 'Alt',
115 > metaKey: 'Cmd',
116 > separator: '+',
117 > },
118 > {
119 > ctrlKey: 'Ctrl',
120 > shiftKey: 'Shift',
121 > altKey: 'Alt',
122 > metaKey: 'Super',
123 > separator: '+',
124 > }
125 > );
126 >
127 > /**
128 > * A label provider that prints modifiers in a suitable format for user settings.
129 > */
130 > export const UserSettingsLabelProvider = new ModifierLabelProvider(
131 > {
132 > ctrlKey: 'ctrl',
133 > shiftKey: 'shift',
134 > altKey: 'alt',
135 > metaKey: 'cmd',
136 > separator: '+',
137 > },
138 > {
139 > ctrlKey: 'ctrl',
140 > shiftKey: 'shift',
141 > altKey: 'alt',
142 > metaKey: 'win',
143 > separator: '+',
144 > },
145 > {
146 > ctrlKey: 'ctrl',
147 > shiftKey: 'shift',
148 > altKey: 'alt',
149 > metaKey: 'meta',
150 > separator: '+',
151 > }
152 > );
153 >
154 function _simpleAsString(modifiers: Modifiers, key: string, labels: ModifierLabels): string {
155 if (key === null) {
src/vs/workbench/contrib/chat/common/model/chatSessionOperationLog.ts 136 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatSessionOperationLog.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { softAssertNever } from '../../../../../base/common/assert.js';
7 > import { isMarkdownString } from '../../../../../base/common/htmlContent.js';
8 > import { equals as objectsEqual } from '../../../../../base/common/objects.js';
9 > import { isEqual as _urisEqual } from '../../../../../base/common/resources.js';
10 > import { hasKey } from '../../../../../base/common/types.js';
11 > import { URI, UriComponents } from '../../../../../base/common/uri.js';
12 > import { IChatRequestVariableEntry } from '../attachments/chatVariableEntries.js';
13 > import { IChatMarkdownContent, IChatMcpAuthenticationRequired, IChatMcpServersStartingSlow, ResponseModelState } from '../chatService/chatService.js';
14 > import { ModifiedFileEntryState } from '../editing/chatEditingService.js';
15 > import { IParsedChatRequest } from '../requestParser/chatParserTypes.js';
16 > import { IChatAgentEditedFileEvent, IChatDataSerializerLog, IChatModel, IChatPendingRequest, IChatProgressResponseContent, IChatRequestModel, IChatRequestVariableData, ISerializableChatData, ISerializableChatModelInputState, ISerializableChatRequestData, ISerializablePendingRequestData, SerializedChatResponsePart, serializeSendOptions } from './chatModel.js';
17 > import * as Adapt from './objectMutationLog.js';
18 >
19 > /**
20 > * ChatModel has lots of properties and lots of ways those properties can mutate.
21 > * The naive way to store the ChatModel is serializing it to JSON and calling it
22 > * a day. However, chats can get very, very long, and thus doing so is slow.
23 > *
24 > * In this file, we define a `storageSchema` that adapters from the `IChatModel`
25 > * into the serializable format. This schema tells us what properties in the chat
26 > * model correspond to the serialized properties, *and how they change*. For
27 > * example, `Adapt.constant(...)` defines a property that will never be checked
28 > * for changes after it's written, and `Adapt.primitive(...)` defines a property
29 > * that will be checked for changes using strict equality each time we store it.
30 > *
31 > * We can then use this to generate a log of mutations that we can append to
32 > * cheaply without rewriting and reserializing the entire request each time.
33 > */
34 >
35 > const toJson = <T>(obj: T): T extends { toJSON?(): infer R } ? R : T => {
36 const cast = obj as { toJSON?: () => T };
37 // eslint-disable-next-line local/code-no-any-casts, @typescript-eslint/no-explicit-any
38 return (cast && typeof cast.toJSON === 'function' ? cast.toJSON() : obj) as any;
39 };
41 > const responsePartSchema = Adapt.v<Exclude<IChatProgressResponseContent, IChatMcpAuthenticationRequired | IChatMcpServersStartingSlow>, SerializedChatResponsePart>(
42 > (obj): SerializedChatResponsePart => obj.kind === 'markdownContent' ? obj.content : toJson(obj),
43 > (a, b) => {
44 if (isMarkdownString(a) && isMarkdownString(b)) {
45 return a.value === b.value;
106 return false;
107 }
109 >
110 > const urisEqual = (a: UriComponents, b: UriComponents): boolean => {
111 return _urisEqual(URI.from(a), URI.from(b));
112 };
114 > const messageSchema = Adapt.object<IParsedChatRequest, IParsedChatRequest>({
115 > text: Adapt.v(m => m.text),
116 > parts: Adapt.v(m => m.parts, (a, b) => a.length === b.length && a.every((part, i) => part.text === b[i].text)),
117 > });
118 >
119 > const agentEditedFileEventSchema = Adapt.object<IChatAgentEditedFileEvent, IChatAgentEditedFileEvent>({
120 > uri: Adapt.v(e => e.uri, urisEqual),
121 > eventKind: Adapt.v(e => e.eventKind),
122 > });
123 >
124 > const chatVariableSchema = Adapt.object<IChatRequestVariableData, IChatRequestVariableData>({
125 > variables: Adapt.t(v => v.variables.map(IChatRequestVariableEntry.toExport), Adapt.array(Adapt.value((a, b) => a.name === b.name))),
126 > });
127 >
128 > const requestSchema = Adapt.object<IChatRequestModel, ISerializableChatRequestData>({
129 > // request parts
130 > requestId: Adapt.t(m => m.id, Adapt.key()),
131 > timestamp: Adapt.v(m => m.requestTimestamp),
132 > confirmation: Adapt.v(m => m.confirmation),
133 > message: Adapt.t(m => m.message, messageSchema),
134 > shouldBeRemovedOnSend: Adapt.v(m => m.shouldBeRemovedOnSend, objectsEqual),
135 > agent: Adapt.v(m => m.response?.agent, (a, b) => a?.id === b?.id),
136 > modelId: Adapt.v(m => m.modelId),
137 > editedFileEvents: Adapt.t(m => m.editedFileEvents, Adapt.array(agentEditedFileEventSchema)),
138 > variableData: Adapt.t(m => m.variableData, chatVariableSchema),
139 > isHidden: Adapt.v(() => undefined), // deprecated, always undefined for new data
140 > isCanceled: Adapt.v(() => undefined), // deprecated, modelState is used instead
141 >
142 > response: Adapt.t(m => m.response?.entireResponse.value.filter((p): p is Exclude<IChatProgressResponseContent, IChatMcpAuthenticationRequired | IChatMcpServersStartingSlow> => p.kind !== 'mcpAuthenticationRequired' && p.kind !== 'mcpServersStartingSlow'), Adapt.array(responsePartSchema)),
143 > responseId: Adapt.v(m => m.response?.id),
144 > responseTimestamp: Adapt.v(m => m.response?.timestamp),
145 > result: Adapt.v(m => m.response?.result, objectsEqual),
146 > responseMarkdownInfo: Adapt.v(
147 > m => m.response?.codeBlockInfos?.map(info => ({ suggestionId: info.suggestionId })),
148 > objectsEqual,
149 > ),
150 > followups: Adapt.v(m => m.response?.followups, objectsEqual),
151 > modelState: Adapt.v(m => m.response?.stateT, objectsEqual),
152 > vote: Adapt.v(m => m.response?.vote),
153 > slashCommand: Adapt.t(m => m.response?.slashCommand, Adapt.value((a, b) => a?.name === b?.name)),
154 > usedContext: Adapt.v(m => m.response?.usedContext, objectsEqual),
155 > contentReferences: Adapt.v(m => m.response?.contentReferences, objectsEqual),
156 > codeCitations: Adapt.v(m => m.response?.codeCitations, objectsEqual),
157 > timeSpentWaiting: Adapt.v(m => m.response?.timestamp), // based on response timestamp
158 > completionTokens: Adapt.v(m => m.response?.completionTokenCount),
159 > promptTokens: Adapt.v(m => m.response?.usage?.promptTokens),
160 > outputBuffer: Adapt.v(m => m.response?.usage?.outputBuffer),
161 > promptTokenDetails: Adapt.v(m => m.response?.usage?.promptTokenDetails, objectsEqual),
162 > copilotCredits: Adapt.v(m => m.response?.usage?.copilotCredits),
163 > elapsedMs: Adapt.v(m => m.response?.elapsedMs ?? (m.response?.completedAt ? Math.max(0, m.response.completedAt - m.response.confirmationAdjustedTimestamp.get()) : undefined)),
164 > modeInfo: Adapt.v(m => m.modeInfo, objectsEqual),
165 > isSystemInitiated: Adapt.v(m => m.isSystemInitiated),
166 > systemInitiatedLabel: Adapt.v(m => m.systemInitiatedLabel),
167 > terminalExecutionId: Adapt.v(m => m.terminalExecutionId),
168 > }, {
169 > sealed: (o) => o.modelState?.value === ResponseModelState.Cancelled || o.modelState?.value === ResponseModelState.Failed || o.modelState?.value === ResponseModelState.Complete,
170 > });
171 >
172 > const inputStateSchema = Adapt.object<ISerializableChatModelInputState, ISerializableChatModelInputState>({
173 > attachments: Adapt.v(i => i.attachments.map(IChatRequestVariableEntry.toExport), objectsEqual),
174 > mode: Adapt.v(i => i.mode, (a, b) => a.id === b.id),
175 > selectedModel: Adapt.v(i => i.selectedModel, (a, b) => a?.identifier === b?.identifier && objectsEqual(a?.modelConfiguration, b?.modelConfiguration)),
176 > inputText: Adapt.v(i => i.inputText),
177 > selections: Adapt.v(i => i.selections, objectsEqual),
178 > permissionLevel: Adapt.v(i => i.permissionLevel),
179 > contrib: Adapt.v(i => i.contrib, objectsEqual),
180 > });
181 >
182 > const pendingRequestSchema = Adapt.object<IChatPendingRequest, ISerializablePendingRequestData>({
183 > id: Adapt.t(p => p.request.id, Adapt.key()),
184 > request: Adapt.t(p => p.request, requestSchema),
185 > kind: Adapt.v(p => p.kind),
186 > sendOptions: Adapt.v(p => serializeSendOptions(p.sendOptions), objectsEqual),
187 > });
188 >
189 > export const storageSchema = Adapt.object<IChatModel, ISerializableChatData>({
190 > version: Adapt.v(() => 3),
191 > creationDate: Adapt.v(m => m.timestamp),
192 > customTitle: Adapt.v(m => m.hasCustomTitle ? m.title : undefined),
193 > initialLocation: Adapt.v(m => m.initialLocation),
194 > inputState: Adapt.t(m => m.inputModel.toJSON(), inputStateSchema),
195 > responderUsername: Adapt.v(m => m.responderUsername),
196 > sessionId: Adapt.v(m => m.sessionId),
197 > requests: Adapt.t(m => m.getRequests(), Adapt.array(requestSchema)),
198 > hasPendingEdits: Adapt.v(m => m.editingSession?.entries.get().some(e => e.state.get() === ModifiedFileEntryState.Modified)),
199 > repoData: Adapt.v(m => m.repoData, objectsEqual),
200 > pendingRequests: Adapt.t(m => m.getPendingRequests(), Adapt.array(pendingRequestSchema)),
201 > workingDirectory: Adapt.v(m => m.workingDirectory?.toString()),
202 > });
203 >
204 > export class ChatSessionOperationLog extends Adapt.ObjectMutationLog<IChatModel, ISerializableChatData> implements IChatDataSerializerLog {
205 > constructor() {
206 super(storageSchema, 1024);
207 }
src/vs/platform/assignment/common/assignment.ts 135 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- assignment.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import * as platform from '../../../base/common/platform.js';
8 > import type { IExperimentationFilterProvider } from 'tas-client';
9 >
10 > export const ASSIGNMENT_STORAGE_KEY = 'VSCode.ABExp.FeatureData';
11 > export const ASSIGNMENT_REFETCH_INTERVAL = 60 * 60 * 1000; // 1 hour
12 >
13 > export interface IAssignmentService {
14 > readonly _serviceBrand: undefined;
15 >
16 > readonly onDidRefetchAssignments: Event<void>;
17 > getTreatment<T extends string | number | boolean>(name: string): Promise<T | undefined>;
18 > }
19 >
20 > export enum TargetPopulation {
21 > Insiders = 'insider',
22 > Public = 'public',
23 > Exploration = 'exploration'
24 > }
25 >
26 > /*
27 > Based upon the official VSCode currently existing filters in the
28 > ExP backend for the VSCode cluster.
29 > https://experimentation.visualstudio.com/Analysis%20and%20Experimentation/_git/AnE.ExP.TAS.TachyonHost.Configuration?path=%2FConfigurations%2Fvscode%2Fvscode.json&version=GBmaster
30 > "X-MSEdge-Market": "detection.market",
31 > "X-FD-Corpnet": "detection.corpnet",
32 > "X-VSCode-AppVersion": "appversion",
33 > "X-VSCode-Build": "build",
34 > "X-MSEdge-ClientId": "clientid",
35 > "X-VSCode-ExtensionName": "extensionname",
36 > "X-VSCode-ExtensionVersion": "extensionversion",
37 > "X-VSCode-TargetPopulation": "targetpopulation",
38 > "X-VSCode-Language": "language",
39 > "X-VSCode-Platform": "platform",
40 > "X-VSCode-ReleaseDate": "releasedate",
41 > "X-VSCode-WindowKind": "windowkind"
42 > */
43 > export enum Filters {
44 > /**
45 > * The market in which the extension is distributed.
46 > */
47 > Market = 'X-MSEdge-Market',
48 >
49 > /**
50 > * The corporation network.
51 > */
52 > CorpNet = 'X-FD-Corpnet',
53 >
54 > /**
55 > * Version of the application which uses experimentation service.
56 > */
57 > ApplicationVersion = 'X-VSCode-AppVersion',
58 >
59 > /**
60 > * Insiders vs Stable.
61 > */
62 > Build = 'X-VSCode-Build',
63 >
64 > /**
65 > * Client Id which is used as primary unit for the experimentation.
66 > */
67 > ClientId = 'X-MSEdge-ClientId',
68 >
69 > /**
70 > * Developer Device Id which can be used as an alternate unit for experimentation.
71 > */
72 > DeveloperDeviceId = 'X-VSCode-DevDeviceId',
73 >
74 > /**
75 > * Extension header.
76 > */
77 > ExtensionName = 'X-VSCode-ExtensionName',
78 >
79 > /**
80 > * The version of the extension.
81 > */
82 > ExtensionVersion = 'X-VSCode-ExtensionVersion',
83 >
84 > /**
85 > * The language in use by VS Code
86 > */
87 > Language = 'X-VSCode-Language',
88 >
89 > /**
90 > * The target population.
91 > * This is used to separate internal, early preview, GA, etc.
92 > */
93 > TargetPopulation = 'X-VSCode-TargetPopulation',
94 >
95 > /**
96 > * The platform (OS) on which VS Code is running.
97 > */
98 > Platform = 'X-VSCode-Platform',
99 >
100 > /**
101 > * The release/build date of VS Code (UTC) in the format yyyymmddHH.
102 > */
103 > ReleaseDate = 'X-VSCode-ReleaseDate',
104 >
105 > /**
106 > * The kind of window VS Code is running in (`editor` or `agents`).
107 > */
108 > WindowKind = 'X-VSCode-WindowKind',
109 > }
110 >
111 > export const enum WindowKind {
112 > Editor = 'editor',
113 > Agents = 'agents',
114 > }
115 >
116 > export class AssignmentFilterProvider implements IExperimentationFilterProvider {
117 > constructor(
118 private version: string,
119 private appName: string,
124 private windowKind: WindowKind
125 ) { }
127 > /**
128 > * Returns a version string that can be parsed by the TAS client.
129 > * The tas client cannot handle suffixes lke "-insider"
130 > * Ref: https://github.com/microsoft/tas-client/blob/30340d5e1da37c2789049fcf45928b954680606f/vscode-tas-client/src/vscode-tas-client/VSCodeFilterProvider.ts#L35
131 > *
132 > * @param version Version string to be trimmed.
133 > */
134 > private static trimVersionSuffix(version: string): string {
135 const regex = /\-[a-zA-Z0-9]+$/;
136 const result = version.split(regex);
138 return result[0];
139 }
141 > getFilterValue(filter: string): string | null {
142 switch (filter) {
143 case Filters.ApplicationVersion:
167 }
168 }
170 > private static formatReleaseDate(iso: string): string {
171 // Expect ISO format, fall back to empty string if not provided
172 if (!iso) {
181 return match.slice(1, 5).join('');
182 }
184 > getFilters(): Map<string, unknown> {
185 const filters: Map<string, unknown> = new Map<string, unknown>();
186 const filterValues = Object.values(Filters);
191 return filters;
192 }
193 > } assignment.ts
194 >
195 > export function getInternalOrg(organisations: string[] | undefined): 'vscode' | 'github' | 'microsoft' | undefined {
196 const isVSCodeInternal = organisations?.includes('Visual-Studio-Code');
197 const isGitHubInternal = organisations?.includes('github');
src/vs/editor/common/core/ranges/offsetRange.ts 134 covered LOC · 42 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- offsetRange.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { BugIndicatingError } from '../../../../base/common/errors.js';
7 >
8 > export interface IOffsetRange {
9 > readonly start: number;
10 > readonly endExclusive: number;
11 > }
12 >
13 > /**
14 > * A range of offsets (0-based).
15 > */
16 > export class OffsetRange implements IOffsetRange {
17 > public static fromTo(start: number, endExclusive: number): OffsetRange {
18 > return new OffsetRange(start, endExclusive);
19 > }
20 >
21 > public static equals(r1: IOffsetRange, r2: IOffsetRange): boolean {
22 return r1.start === r2.start && r1.endExclusive === r2.endExclusive;
23 }
25 > public static addRange(range: OffsetRange, sortedRanges: OffsetRange[]): void {
26 let i = 0;
27 while (i < sortedRanges.length && sortedRanges[i].endExclusive < range.start) {
40 }
41 }
43 > public static tryCreate(start: number, endExclusive: number): OffsetRange | undefined {
44 if (start > endExclusive) {
45 return undefined;
47 return new OffsetRange(start, endExclusive);
48 }
50 > public static ofLength(length: number): OffsetRange {
51 return new OffsetRange(0, length);
52 }
54 > public static ofStartAndLength(start: number, length: number): OffsetRange {
55 return new OffsetRange(start, start + length);
56 }
58 > public static emptyAt(offset: number): OffsetRange {
59 return new OffsetRange(offset, offset);
60 }
62 > constructor(public readonly start: number, public readonly endExclusive: number) {
63 > if (start > endExclusive) { offsetRange.ts
64 throw new BugIndicatingError(`Invalid range: ${this.toString()}`);
65 }
68 > get isEmpty(): boolean {
69 return this.start === this.endExclusive;
70 }
72 > public delta(offset: number): OffsetRange {
73 return new OffsetRange(this.start + offset, this.endExclusive + offset);
74 }
76 > public deltaStart(offset: number): OffsetRange {
77 return new OffsetRange(this.start + offset, this.endExclusive);
78 }
80 > public deltaEnd(offset: number): OffsetRange {
81 return new OffsetRange(this.start, this.endExclusive + offset);
82 }
84 > public get length(): number {
85 return this.endExclusive - this.start;
86 }
88 > public toString() {
89 return `[${this.start}, ${this.endExclusive})`;
90 }
92 > public equals(other: OffsetRange): boolean {
93 return this.start === other.start && this.endExclusive === other.endExclusive;
94 }
96 > public containsRange(other: OffsetRange): boolean {
97 return this.start <= other.start && other.endExclusive <= this.endExclusive;
98 }
100 > public contains(offset: number): boolean {
101 return this.start <= offset && offset < this.endExclusive;
102 }
104 > /**
105 > * for all numbers n: range1.contains(n) or range2.contains(n) => range1.join(range2).contains(n)
106 > * The joined range is the smallest range that contains both ranges.
107 > */
108 > public join(other: OffsetRange): OffsetRange {
109 return new OffsetRange(Math.min(this.start, other.start), Math.max(this.endExclusive, other.endExclusive));
110 }
112 > /**
113 > * for all numbers n: range1.contains(n) and range2.contains(n) <=> range1.intersect(range2).contains(n)
114 > *
115 > * The resulting range is empty if the ranges do not intersect, but touch.
116 > * If the ranges don't even touch, the result is undefined.
117 > */
118 > public intersect(other: OffsetRange): OffsetRange | undefined {
119 const start = Math.max(this.start, other.start);
120 const end = Math.min(this.endExclusive, other.endExclusive);
124 return undefined;
125 }
127 > public intersectionLength(range: OffsetRange): number {
128 const start = Math.max(this.start, range.start);
129 const end = Math.min(this.endExclusive, range.endExclusive);
130 return Math.max(0, end - start);
131 }
133 > /**
134 > * `a.intersects(b)` iff there exists a number n so that `a.contains(n)` and `b.contains(n)`.
135 > * Warning: If one range is empty, this method returns always false.
136 > */
137 > public intersects(other: OffsetRange): boolean {
138 const start = Math.max(this.start, other.start);
139 const end = Math.min(this.endExclusive, other.endExclusive);
140 return start < end;
141 }
143 > public intersectsOrTouches(other: OffsetRange): boolean {
144 const start = Math.max(this.start, other.start);
145 const end = Math.min(this.endExclusive, other.endExclusive);
146 return start <= end;
147 }
149 > public isBefore(other: OffsetRange): boolean {
150 return this.endExclusive <= other.start;
151 }
153 > public isAfter(other: OffsetRange): boolean {
154 return this.start >= other.endExclusive;
155 }
157 > public slice<T>(arr: readonly T[]): T[] {
158 return arr.slice(this.start, this.endExclusive);
159 }
161 > public substring(str: string): string {
162 return str.substring(this.start, this.endExclusive);
163 }
165 > /**
166 > * Returns the given value if it is contained in this instance, otherwise the closest value that is contained.
167 > * The range must not be empty.
168 > */
169 > public clip(value: number): number {
170 if (this.isEmpty) {
171 throw new BugIndicatingError(`Invalid clipping range: ${this.toString()}`);
173 return Math.max(this.start, Math.min(this.endExclusive - 1, value));
174 }
176 > /**
177 > * Returns `r := value + k * length` such that `r` is contained in this range.
178 > * The range must not be empty.
179 > *
180 > * E.g. `[5, 10).clipCyclic(10) === 5`, `[5, 10).clipCyclic(11) === 6` and `[5, 10).clipCyclic(4) === 9`.
181 > */
182 > public clipCyclic(value: number): number {
183 if (this.isEmpty) {
184 throw new BugIndicatingError(`Invalid clipping range: ${this.toString()}`);
192 return value;
193 }
195 > public map<T>(f: (offset: number) => T): T[] {
196 const result: T[] = [];
197 for (let i = this.start; i < this.endExclusive; i++) {
200 return result;
201 }
203 > public forEach(f: (offset: number) => void): void {
204 for (let i = this.start; i < this.endExclusive; i++) {
205 f(i);
206 }
207 }
209 > /**
210 > * this: [ 5, 10), range: [10, 15) => [5, 15)]
211 > * Throws if the ranges are not touching.
212 > */
213 > public joinRightTouching(range: OffsetRange): OffsetRange {
214 if (this.endExclusive !== range.start) {
215 throw new BugIndicatingError(`Invalid join: ${this.toString()} and ${range.toString()}`);
217 return new OffsetRange(this.start, range.endExclusive);
218 }
220 > public withMargin(margin: number): OffsetRange;
221 > public withMargin(marginStart: number, marginEnd: number): OffsetRange;
222 > public withMargin(marginStart: number, marginEnd?: number): OffsetRange {
223 if (marginEnd === undefined) {
224 marginEnd = marginStart;
226 return new OffsetRange(this.start - marginStart, this.endExclusive + marginEnd);
227 }
228 > } offsetRange.ts
229 >
230 > export class OffsetRangeSet {
231 private readonly _sortedRanges: OffsetRange[] = [];
233 > public get ranges(): OffsetRange[] {
234 return [...this._sortedRanges];
235 }
237 > public addRange(range: OffsetRange): void {
238 let i = 0;
239 while (i < this._sortedRanges.length && this._sortedRanges[i].endExclusive < range.start) {
252 }
253 }
255 > public toString(): string {
256 return this._sortedRanges.map(r => r.toString()).join(', ');
257 }
259 > /**
260 > * Returns if there is a value that is contained in this instance and the given range.
261 > */
262 > public intersectsStrict(other: OffsetRange): boolean {
263 // TODO use binary search
264 let i = 0;
268 return i < this._sortedRanges.length && this._sortedRanges[i].start < other.endExclusive;
269 }
271 > public intersectWithRange(other: OffsetRange): OffsetRangeSet {
272 // TODO use binary search + slice
273 const result = new OffsetRangeSet();
280 return result;
281 }
283 > public intersectWithRangeLength(other: OffsetRange): number {
284 return this.intersectWithRange(other).length;
285 }
287 > public get length(): number {
288 return this._sortedRanges.reduce((prev, cur) => prev + cur.length, 0);
289 }
290 > } offsetRange.ts
src/vs/base/common/observableInternal/observables/observableFromEvent.ts 133 covered LOC · 27 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableFromEvent.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, ITransaction } from '../base.js';
7 > import { subtransaction } from '../transaction.js';
8 > import { EqualityComparer, Event, IDisposable, strictEquals } from '../commonFacade/deps.js';
9 > import { DebugOwner, DebugNameData, IDebugNameData } from '../debugName.js';
10 > import { getLogger } from '../logging/logging.js';
11 > import { BaseObservable } from './baseObservable.js';
12 > import { DebugLocation } from '../debugLocation.js';
13 >
14 >
15 > export function observableFromEvent<T, TArgs = unknown>(
16 > owner: DebugOwner,
17 > event: Event<TArgs>,
18 > getValue: (args: TArgs | undefined) => T,
19 > debugLocation?: DebugLocation,
20 > ): IObservable<T>;
21 > export function observableFromEvent<T, TArgs = unknown>(
22 > event: Event<TArgs>,
23 > getValue: (args: TArgs | undefined) => T,
24 > ): IObservable<T>;
25 > export function observableFromEvent(...args:
26 > [owner: DebugOwner, event: Event<any>, getValue: (args: any | undefined) => any, debugLocation?: DebugLocation] | observableFromEvent.ts
27 > [event: Event<any>, getValue: (args: any | undefined) => any]
28 > ): IObservable<any> {
29 > let owner;
30 > let event;
31 > let getValue;
32 > let debugLocation;
33 > if (args.length === 2) {
34 [event, getValue] = args;
35 > } else { observableFromEvent.ts
36 > [owner, event, getValue, debugLocation] = args; observableFromEvent.ts
37 > }
38 > return new FromEventObservable( observableFromEvent.ts
39 > new DebugNameData(owner, undefined, getValue),
40 > event,
41 > getValue,
42 > () => FromEventObservable.globalTransaction,
43 > strictEquals,
44 > debugLocation ?? DebugLocation.ofCaller()
45 > );
46 > }
48 > export function observableFromEventOpts<T, TArgs = unknown>(
49 options: IDebugNameData & {
50 equalsFn?: EqualityComparer<T>;
64 );
65 }
67 > export class FromEventObservable<TArgs, T> extends BaseObservable<T> {
68 > public static globalTransaction: ITransaction | undefined;
69 >
70 > private _value: T | undefined;
71 > private _hasValue = false;
72 > private _subscription: IDisposable | undefined;
73 >
74 > constructor(
75 > private readonly _debugNameData: DebugNameData, observableFromEvent.ts
76 > private readonly event: Event<TArgs>,
77 > public readonly _getValue: (args: TArgs | undefined) => T,
78 > private readonly _getTransaction: () => ITransaction | undefined,
79 > private readonly _equalityComparator: EqualityComparer<T>,
80 > debugLocation: DebugLocation
81 > ) {
82 > super(debugLocation);
83 > }
84 >
85 > private getDebugName(): string | undefined {
86 return this._debugNameData.getDebugName(this);
87 }
89 > public get debugName(): string {
90 const name = this.getDebugName();
91 return 'From Event' + (name ? `: ${name}` : '');
92 }
94 > protected override onFirstObserverAdded(): void {
95 > this._subscription = this.event(this.handleEvent); observableFromEvent.ts
96 > }
98 > private readonly handleEvent = (args: TArgs | undefined) => {
99 > const newValue = this._getValue(args); observableFromEvent.ts
100 > const oldValue = this._value;
101 >
102 > const didChange = !this._hasValue || !(this._equalityComparator(oldValue!, newValue));
103 > let didRunTransaction = false;
104 >
105 > if (didChange) {
106 > this._value = newValue;
107 >
108 > if (this._hasValue) {
109 > didRunTransaction = true; observableFromEvent.ts
110 > subtransaction(
111 > this._getTransaction(),
112 > (tx) => {
113 > getLogger()?.handleObservableUpdated(this, { oldValue, newValue, change: undefined, didChange, hadValue: this._hasValue });
114 >
115 > for (const o of this._observers) {
116 > tx.updateObserver(o, this);
117 > o.handleChange(this, undefined);
118 > }
119 > },
120 > () => {
121 const name = this.getDebugName();
122 return 'Event fired' + (name ? `: ${name}` : '');
123 }
125 > }
126 > this._hasValue = true; observableFromEvent.ts
127 > }
128 >
129 > if (!didRunTransaction) {
130 > getLogger()?.handleObservableUpdated(this, { oldValue, newValue, change: undefined, didChange, hadValue: this._hasValue });
131 > }
132 > };
134 > protected override onLastObserverRemoved(): void {
135 > this._subscription!.dispose(); observableFromEvent.ts
136 > this._subscription = undefined;
137 > this._hasValue = false;
138 > this._value = undefined;
139 > }
141 > public get(): T {
142 > if (this._subscription) { observableFromEvent.ts
143 > if (!this._hasValue) { observableFromEvent.ts
144 > this.handleEvent(undefined);
145 > }
146 > return this._value!;
147 > } else { observableFromEvent.ts
148 // no cache, as there are no subscribers to keep it updated
149 const value = this._getValue(undefined);
150 return value;
151 }
154 > public debugSetValue(value: unknown): void {
155 // eslint-disable-next-line local/code-no-any-casts
156 this._value = value as any;
157 }
159 > public debugGetState() {
160 return { value: this._value, hasValue: this._hasValue };
161 }
163 >
164 > export namespace observableFromEvent {
165 > export const Observer = FromEventObservable;
166 >
167 > export function batchEventsGlobally(tx: ITransaction, fn: () => void): void {
168 let didSet = false;
169 if (FromEventObservable.globalTransaction === undefined) {
src/vs/workbench/contrib/chat/common/promptSyntax/promptTypes.ts 131 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { LanguageSelector } from '../../../../../editor/common/languageSelector.js';
7 > import { localize } from '../../../../../nls.js';
8 >
9 > /**
10 > * Documentation link for the reusable prompts feature.
11 > */
12 > export const PROMPT_DOCUMENTATION_URL = 'https://aka.ms/vscode-ghcp-prompt-snippets';
13 > export const INSTRUCTIONS_DOCUMENTATION_URL = 'https://aka.ms/vscode-ghcp-custom-instructions';
14 > export const AGENT_DOCUMENTATION_URL = 'https://aka.ms/vscode-ghcp-custom-chat-modes'; // todo
15 > export const SKILL_DOCUMENTATION_URL = 'https://aka.ms/vscode-agent-skills';
16 > // TODO: update link when available
17 > export const HOOK_DOCUMENTATION_URL = 'https://aka.ms/vscode-chat-hooks';
18 >
19 > /**
20 > * Language ID for the reusable prompt syntax.
21 > */
22 > export const PROMPT_LANGUAGE_ID = 'prompt';
23 >
24 > /**
25 > * Language ID for instructions syntax.
26 > */
27 > export const INSTRUCTIONS_LANGUAGE_ID = 'instructions';
28 >
29 > /**
30 > * Language ID for agent syntax.
31 > */
32 > export const AGENT_LANGUAGE_ID = 'chatagent';
33 >
34 > /**
35 > * Language ID for skill syntax.
36 > */
37 > export const SKILL_LANGUAGE_ID = 'skill';
38 >
39 > /**
40 > * Prompt and instructions files language selector.
41 > */
42 > export const ALL_PROMPTS_LANGUAGE_SELECTOR: LanguageSelector = [PROMPT_LANGUAGE_ID, INSTRUCTIONS_LANGUAGE_ID, AGENT_LANGUAGE_ID, SKILL_LANGUAGE_ID];
43 >
44 > /**
45 > * Configuration key for enabling the agent debug log feature.
46 > */
47 > export const AGENT_DEBUG_LOG_ENABLED_SETTING = 'github.copilot.chat.agentDebugLog.enabled';
48 >
49 > /**
50 > * Configuration key for enabling file logging for the agent debug log.
51 > */
52 > export const AGENT_DEBUG_LOG_FILE_LOGGING_ENABLED_SETTING = 'github.copilot.chat.agentDebugLog.fileLogging.enabled';
53 >
54 > /**
55 > * Configuration key for enabling agent debug logging for agent host (Copilot CLI) sessions.
56 > * Registered in core (see `chat.shared.contribution.ts`) since only core consumes it.
57 > */
58 > export const AgentHostAgentDebugLogEnabledSettingId = 'chat.agentHost.agentDebugLog.enabled';
59 >
60 > /**
61 > * Configuration key for the maximum number of debug events kept in memory for
62 > * agent host (Copilot CLI) sessions. Registered in core (see
63 > * `chat.shared.contribution.ts`) since only core consumes it.
64 > */
65 > export const AgentHostAgentDebugLogMaxEventsSettingId = 'chat.agentHost.agentDebugLog.maxEventsInMemory';
66 >
67 > /**
68 > * The name of the troubleshoot slash command / skill.
69 > */
70 > export const TROUBLESHOOT_COMMAND_NAME = 'troubleshoot';
71 >
72 > /**
73 > * URI scheme used by the Copilot extension for built-in skills.
74 > */
75 > export const COPILOT_SKILL_URI_SCHEME = 'copilot-skill';
76 >
77 > /**
78 > * Path fragment that identifies the troubleshoot skill in a URI.
79 > */
80 > export const TROUBLESHOOT_SKILL_PATH = 'troubleshoot/SKILL.md';
81 >
82 > /**
83 > * The language id for a prompts type.
84 > */
85 > export function getLanguageIdForPromptsType(type: PromptsType): string {
86 switch (type) {
87 case PromptsType.prompt:
100 }
101 }
103 > export function getPromptsTypeForLanguageId(languageId: string): PromptsType | undefined {
104 switch (languageId) {
105 case PROMPT_LANGUAGE_ID:
116 }
117 }
119 >
120 > /**
121 > * What the prompt is used for.
122 > */
123 > export enum PromptsType {
124 > instructions = 'instructions',
125 > prompt = 'prompt',
126 > agent = 'agent',
127 > skill = 'skill',
128 > hook = 'hook'
129 > }
130 > export function isValidPromptType(type: string): type is PromptsType {
131 return Object.values(PromptsType).includes(type as PromptsType);
132 }
134 > export enum Target {
135 > VSCode = 'vscode',
136 > GitHubCopilot = 'github-copilot',
137 > Claude = 'claude',
138 > Undefined = 'undefined',
139 > }
140 >
141 > /**
142 > * Tracks where prompt files originate from.
143 > */
144 > export enum PromptFileSource {
145 > GitHubWorkspace = 'github-workspace',
146 > CopilotPersonal = 'copilot-personal',
147 > ClaudePersonal = 'claude-personal',
148 > ClaudeWorkspace = 'claude-workspace',
149 > ClaudeWorkspaceLocal = 'claude-workspace-local',
150 > AgentsWorkspace = 'agents-workspace',
151 > AgentsPersonal = 'agents-personal',
152 > ConfigWorkspace = 'config-workspace',
153 > ConfigPersonal = 'config-personal',
154 > UserData = 'user-data',
155 > ExtensionContribution = 'extension-contribution',
156 > ExtensionAPI = 'extension-api',
157 > Plugin = 'plugin',
158 > }
159 >
160 > /**
161 > * Returns a human-readable description for a prompt file source.
162 > */
163 > export function getSourceDescription(source: PromptFileSource): string | undefined {
164 switch (source) {
165 case PromptFileSource.AgentsWorkspace:
src/vs/workbench/common/configuration.ts 130 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- configuration.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../nls.js';
7 > import { ConfigurationScope, IConfigurationNode, IConfigurationRegistry, Extensions as ConfigurationExtensions } from '../../platform/configuration/common/configurationRegistry.js';
8 > import { Registry } from '../../platform/registry/common/platform.js';
9 > import { IWorkbenchContribution } from './contributions.js';
10 > import { IWorkspaceContextService, IWorkspaceFolder, WorkbenchState } from '../../platform/workspace/common/workspace.js';
11 > import { ConfigurationTarget, IConfigurationService, IConfigurationValue, IInspectValue } from '../../platform/configuration/common/configuration.js';
12 > import { Disposable } from '../../base/common/lifecycle.js';
13 > import { Emitter } from '../../base/common/event.js';
14 > import { IRemoteAgentService } from '../services/remote/common/remoteAgentService.js';
15 > import { OperatingSystem, isWindows } from '../../base/common/platform.js';
16 > import { URI } from '../../base/common/uri.js';
17 > import { equals } from '../../base/common/objects.js';
18 > import { DeferredPromise } from '../../base/common/async.js';
19 > import { IUserDataProfile, IUserDataProfilesService } from '../../platform/userDataProfile/common/userDataProfile.js';
20 >
21 > export const applicationConfigurationNodeBase = Object.freeze<IConfigurationNode>({
22 > 'id': 'application',
23 > 'order': 100,
24 > 'title': localize('applicationConfigurationTitle', "Application"),
25 > 'type': 'object'
26 > });
27 >
28 > export const workbenchConfigurationNodeBase = Object.freeze<IConfigurationNode>({
29 > 'id': 'workbench',
30 > 'order': 7,
31 > 'title': localize('workbenchConfigurationTitle', "Workbench"),
32 > 'type': 'object',
33 > });
34 >
35 > export const securityConfigurationNodeBase = Object.freeze<IConfigurationNode>({
36 > 'id': 'security',
37 > 'scope': ConfigurationScope.APPLICATION,
38 > 'title': localize('securityConfigurationTitle', "Security"),
39 > 'type': 'object',
40 > 'order': 7
41 > });
42 >
43 > export const problemsConfigurationNodeBase = Object.freeze<IConfigurationNode>({
44 > 'id': 'problems',
45 > 'title': localize('problemsConfigurationTitle', "Problems"),
46 > 'type': 'object',
47 > 'order': 101
48 > });
49 >
50 > export const windowConfigurationNodeBase = Object.freeze<IConfigurationNode>({
51 > 'id': 'window',
52 > 'order': 8,
53 > 'title': localize('windowConfigurationTitle', "Window"),
54 > 'type': 'object',
55 > });
56 >
57 > export const Extensions = {
58 > ConfigurationMigration: 'base.contributions.configuration.migration'
59 > };
60 >
61 > type ConfigurationValue = { value: unknown | undefined /* Remove */ };
62 > export type ConfigurationKeyValuePairs = [string, ConfigurationValue][];
63 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
64 > export type ConfigurationMigrationFn = (value: any, valueAccessor: (key: string) => any) => ConfigurationValue | ConfigurationKeyValuePairs | Promise<ConfigurationValue | ConfigurationKeyValuePairs>;
65 > export type ConfigurationMigration = { key: string; migrateFn: ConfigurationMigrationFn };
66 >
67 > export interface IConfigurationMigrationRegistry {
68 > registerConfigurationMigrations(configurationMigrations: ConfigurationMigration[]): void;
69 > }
70 >
71 > class ConfigurationMigrationRegistry implements IConfigurationMigrationRegistry {
72 >
73 > readonly migrations: ConfigurationMigration[] = [];
74 >
75 > private readonly _onDidRegisterConfigurationMigrations = new Emitter<ConfigurationMigration[]>();
76 > readonly onDidRegisterConfigurationMigration = this._onDidRegisterConfigurationMigrations.event;
77 >
78 > registerConfigurationMigrations(configurationMigrations: ConfigurationMigration[]): void {
79 this.migrations.push(...configurationMigrations);
80 }
82 > }
83 >
84 > const configurationMigrationRegistry = new ConfigurationMigrationRegistry();
85 > Registry.add(Extensions.ConfigurationMigration, configurationMigrationRegistry);
86 >
87 > export class ConfigurationMigrationWorkbenchContribution extends Disposable implements IWorkbenchContribution {
88 >
89 > static readonly ID = 'workbench.contrib.configurationMigration';
90 >
91 > constructor(
92 @IConfigurationService private readonly configurationService: IConfigurationService,
93 @IWorkspaceContextService private readonly workspaceService: IWorkspaceContextService,
102 this._register(configurationMigrationRegistry.onDidRegisterConfigurationMigration(migration => this.migrateConfigurations(migration)));
103 }
105 > private async migrateConfigurations(migrations: ConfigurationMigration[]): Promise<void> {
106 await this.migrateConfigurationsForFolder(undefined, migrations);
107 for (const folder of this.workspaceService.getWorkspace().folders) {
109 }
110 }
112 > private async migrateConfigurationsForFolder(folder: IWorkspaceFolder | undefined, migrations: ConfigurationMigration[]): Promise<void> {
113 await Promise.all([migrations.map(migration => this.migrateConfigurationsForFolderAndOverride(migration, folder?.uri))]);
114 }
116 > private async migrateConfigurationsForFolderAndOverride(migration: ConfigurationMigration, resource?: URI): Promise<void> {
117 const inspectData = this.configurationService.inspect(migration.key, { resource });
118
160 }
161 }
163 > private async runMigration(migration: ConfigurationMigration, dataKey: keyof IConfigurationValue<unknown>, value: unknown, resource: URI | undefined, overrideIdentifiers: string[] | undefined): Promise<ConfigurationKeyValuePairs | undefined> {
164 const valueAccessor = (key: string) => {
165 const inspectData = this.configurationService.inspect(key, { resource });
176 return Array.isArray(result) ? result : [[migration.key, result]];
177 }
179 >
180 > export class DynamicWorkbenchSecurityConfiguration extends Disposable implements IWorkbenchContribution {
181 >
182 > static readonly ID = 'workbench.contrib.dynamicWorkbenchSecurityConfiguration';
183 >
184 > private readonly _ready = new DeferredPromise<void>();
185 > readonly ready = this._ready.p;
186 >
187 > constructor(
188 @IRemoteAgentService private readonly remoteAgentService: IRemoteAgentService
189 ) {
192 this.create();
193 }
195 > private async create(): Promise<void> {
196 try {
197 await this.doCreate();
200 }
201 }
203 > private async doCreate(): Promise<void> {
204 if (!isWindows) {
205 const remoteEnvironment = await this.remoteAgentService.getEnvironment();
234 });
235 }
237 >
238 > export const CONFIG_NEW_WINDOW_PROFILE = 'window.newWindowProfile';
239 >
240 > export class DynamicWindowConfiguration extends Disposable implements IWorkbenchContribution {
241 >
242 > static readonly ID = 'workbench.contrib.dynamicWindowConfiguration';
243 >
244 > private configurationNode: IConfigurationNode | undefined;
245 > private newWindowProfile: IUserDataProfile | undefined;
246 >
247 > constructor(
248 @IUserDataProfilesService private readonly userDataProfilesService: IUserDataProfilesService,
249 @IConfigurationService private readonly configurationService: IConfigurationService,
263 this._register(this.userDataProfilesService.onDidChangeProfiles(() => this.checkAndResetNewWindowProfileConfig()));
264 }
266 > private registerNewWindowProfileConfiguration(): void {
267 const registry = Registry.as<IConfigurationRegistry>(ConfigurationExtensions.Configuration);
268 const configurationNode: IConfigurationNode = {
286 this.configurationNode = configurationNode;
287 }
289 > private setNewWindowProfile(): void {
290 const newWindowProfileName = this.configurationService.getValue(CONFIG_NEW_WINDOW_PROFILE);
291 this.newWindowProfile = newWindowProfileName ? this.userDataProfilesService.profiles.find(profile => profile.name === newWindowProfileName) : undefined;
292 }
294 > private checkAndResetNewWindowProfileConfig(): void {
295 const newWindowProfileName = this.configurationService.getValue(CONFIG_NEW_WINDOW_PROFILE);
296 if (!newWindowProfileName) {
303 this.configurationService.updateValue(CONFIG_NEW_WINDOW_PROFILE, profile?.name);
304 }
src/vs/workbench/services/assignment/common/assignmentService.ts 130 covered LOC · 16 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- assignmentService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { localize } from '../../../../nls.js';
7 > import { createDecorator, IInstantiationService } from '../../../../platform/instantiation/common/instantiation.js';
8 > import type { IKeyValueStorage, IExperimentationTelemetry, ExperimentationService as TASClient } from 'tas-client';
9 > import { Memento } from '../../../common/memento.js';
10 > import { ITelemetryService } from '../../../../platform/telemetry/common/telemetry.js';
11 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
12 > import { ITelemetryData } from '../../../../base/common/actions.js';
13 > import { InstantiationType, registerSingleton } from '../../../../platform/instantiation/common/extensions.js';
14 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
15 > import { IProductService } from '../../../../platform/product/common/productService.js';
16 > import { ASSIGNMENT_REFETCH_INTERVAL, ASSIGNMENT_STORAGE_KEY, AssignmentFilterProvider, IAssignmentService, TargetPopulation, WindowKind } from '../../../../platform/assignment/common/assignment.js';
17 > import { Registry } from '../../../../platform/registry/common/platform.js';
18 > import { workbenchConfigurationNodeBase } from '../../../common/configuration.js';
19 > import { IConfigurationRegistry, Extensions as ConfigurationExtensions, ConfigurationScope } from '../../../../platform/configuration/common/configurationRegistry.js';
20 > import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.js';
21 > import { importAMDNodeModule } from '../../../../amdX.js';
22 > import { timeout } from '../../../../base/common/async.js';
23 > import { StopWatch } from '../../../../base/common/stopwatch.js';
24 > import { CopilotAssignmentFilterProvider } from './assignmentFilters.js';
25 > import { AssignmentContextFilter } from './assignmentContextFilter.js';
26 > import { Disposable, DisposableStore } from '../../../../base/common/lifecycle.js';
27 > import { Emitter, Event } from '../../../../base/common/event.js';
28 > import { experimentsEnabled } from '../../telemetry/common/workbenchTelemetryUtils.js';
29 >
30 > export interface IAssignmentFilter {
31 > /**
32 > * Stable identifier for this filter. Used to persist and reconcile the set of
33 > * assignment-context ids this filter has excluded, independently of other filters.
34 > */
35 > readonly id: string;
36 > exclude(assignment: string): boolean;
37 > onDidChange: Event<void>;
38 > }
39 >
40 > export const IWorkbenchAssignmentService = createDecorator<IWorkbenchAssignmentService>('assignmentService');
41 >
42 > export interface IWorkbenchAssignmentService extends IAssignmentService {
43 > getCurrentExperiments(): Promise<string[] | undefined>;
44 > addTelemetryAssignmentFilter(filter: IAssignmentFilter): void;
45 > }
46 >
47 > class MementoKeyValueStorage implements IKeyValueStorage {
48 >
49 > private readonly mementoObj: Record<string, unknown>;
50 >
51 > constructor(private readonly memento: Memento<Record<string, unknown>>) {
52 this.mementoObj = memento.getMemento(StorageScope.APPLICATION, StorageTarget.MACHINE);
53 }
55 > async getValue<T>(key: string, defaultValue?: T | undefined): Promise<T | undefined> {
56 const value = await this.mementoObj[key] as T | undefined;
57
58 return value || defaultValue;
59 }
61 > setValue<T>(key: string, value: T): void {
62 this.mementoObj[key] = value;
63 this.memento.saveMemento();
64 }
66 >
67 > class WorkbenchAssignmentServiceTelemetry extends Disposable implements IExperimentationTelemetry {
68 >
69 > private readonly _onDidUpdateAssignmentContext = this._register(new Emitter<void>());
70 > readonly onDidUpdateAssignmentContext = this._onDidUpdateAssignmentContext.event;
71 >
72 > private _previousAssignmentContext: string | undefined;
73 > private _lastAssignmentContext: string | undefined;
74 > get assignmentContext(): string[] | undefined {
75 > return this._lastAssignmentContext?.split(';');
76 > }
77 >
78 > constructor(
79 private readonly telemetryService: ITelemetryService,
80 private readonly productService: IProductService,
90 }));
91 }
93 > private _setAssignmentContext(value: string): void {
94 const filteredValue = this.contextFilter.filter(value);
95 this._lastAssignmentContext = filteredValue;
100 }
101 }
103 > // __GDPR__COMMON__ "abexp.assignmentcontext" : { "classification": "SystemMetaData", "purpose": "FeatureInsight" }
104 > setSharedProperty(name: string, value: string): void {
105 if (name === this.productService.tasConfig?.assignmentContextTelemetryPropertyName) {
106 this._previousAssignmentContext = value;
110 this.telemetryService.setExperimentProperty(name, value);
111 }
113 > postEvent(eventName: string, props: Map<string, string>): void {
114 const data: ITelemetryData = {};
115 for (const [key, value] of props.entries()) {
126 this.telemetryService.publicLog(eventName, data);
127 }
129 >
130 > export class WorkbenchAssignmentService extends Disposable implements IAssignmentService {
131 >
132 > declare readonly _serviceBrand: undefined;
133 >
134 > private readonly tasClient: Promise<TASClient> | undefined;
135 > private readonly tasSetupDisposables = new DisposableStore();
136 >
137 > private networkInitialized = false;
138 > private readonly overrideInitDelay: Promise<void>;
139 >
140 > private readonly contextFilter: AssignmentContextFilter;
141 > private readonly telemetry: WorkbenchAssignmentServiceTelemetry;
142 > private readonly keyValueStorage: IKeyValueStorage;
143 >
144 > private readonly experimentsEnabled: boolean;
145 >
146 > private readonly _onDidRefetchAssignments = this._register(new Emitter<void>());
147 > public readonly onDidRefetchAssignments = this._onDidRefetchAssignments.event;
148 >
149 > constructor(
150 @ITelemetryService private readonly telemetryService: ITelemetryService,
151 @IStorageService storageService: IStorageService,
179 this.overrideInitDelay = timeout(overrideDelay);
180 }
182 > async getTreatment<T extends string | number | boolean>(name: string): Promise<T | undefined> {
183 const result = await this.doGetTreatment<T>(name);
184
202 return result;
203 }
205 > private async doGetTreatment<T extends string | number | boolean>(name: string): Promise<T | undefined> {
206 await this.overrideInitDelay; // For development purposes, allow overriding tas assignments to test variants locally.
207
235 return result;
236 }
238 > private async setupTASClient(): Promise<TASClient> {
239 this.tasSetupDisposables.clear();
240
285 return tasClient;
286 }
288 > private logFetchLatency(fetchType: 'initial' | 'refetch', durationMs: number): void {
289 type TASClientFetchLatencyData = {
290 fetchType: string;
304 });
305 }
307 > private async refetchAssignments(): Promise<void> {
308 if (!this.tasClient) {
309 return; // Setup has not started, assignments will use latest filters
319 this.logFetchLatency('refetch', refetchStopWatch.elapsed());
320 }
322 > async getCurrentExperiments(): Promise<string[] | undefined> {
323 if (!this.tasClient) {
324 return undefined;
333 return this.telemetry.assignmentContext;
334 }
336 > addTelemetryAssignmentFilter(filter: IAssignmentFilter): void {
337 this.contextFilter.addFilter(filter);
338 }
340 >
341 > registerSingleton(IWorkbenchAssignmentService, WorkbenchAssignmentService, InstantiationType.Delayed);
342 >
343 > const registry = Registry.as<IConfigurationRegistry>(ConfigurationExtensions.Configuration);
344 > registry.registerConfiguration({
345 > ...workbenchConfigurationNodeBase,
346 > 'properties': {
347 > 'workbench.enableExperiments': {
348 > 'type': 'boolean',
349 > 'description': localize('workbench.enableExperiments', "Fetches experiments to run from a Microsoft online service."),
350 > 'default': true,
351 > 'scope': ConfigurationScope.APPLICATION,
352 > 'restricted': true,
353 > 'tags': ['usesOnlineServices']
354 > }
355 > }
356 > });
src/vs/editor/common/encodedTokenAttributes.ts 128 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- encodedTokenAttributes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * Open ended enum at runtime
8 > */
9 > export const enum LanguageId {
10 > Null = 0,
11 > PlainText = 1
12 > }
13 >
14 > /**
15 > * A font style. Values are 2^x such that a bit mask can be used.
16 > */
17 > export const enum FontStyle {
18 > NotSet = -1,
19 > None = 0,
20 > Italic = 1,
21 > Bold = 2,
22 > Underline = 4,
23 > Strikethrough = 8,
24 > }
25 >
26 > /**
27 > * Open ended enum at runtime
28 > */
29 > export const enum ColorId {
30 > None = 0,
31 > DefaultForeground = 1,
32 > DefaultBackground = 2
33 > }
34 >
35 > /**
36 > * A standard token type.
37 > */
38 > export const enum StandardTokenType {
39 > Other = 0,
40 > Comment = 1,
41 > String = 2,
42 > RegEx = 3
43 > }
44 >
45 > /**
46 > * Helpers to manage the "collapsed" metadata of an entire StackElement stack.
47 > * The following assumptions have been made:
48 > * - languageId < 256 => needs 8 bits
49 > * - unique color count < 512 => needs 9 bits
50 > *
51 > * The binary format is:
52 > * - -------------------------------------------
53 > * 3322 2222 2222 1111 1111 1100 0000 0000
54 > * 1098 7654 3210 9876 5432 1098 7654 3210
55 > * - -------------------------------------------
56 > * xxxx xxxx xxxx xxxx xxxx xxxx xxxx xxxx
57 > * bbbb bbbb ffff ffff fFFF FBTT LLLL LLLL
58 > * - -------------------------------------------
59 > * - L = LanguageId (8 bits)
60 > * - T = StandardTokenType (2 bits)
61 > * - B = Balanced bracket (1 bit)
62 > * - F = FontStyle (4 bits)
63 > * - f = foreground color (9 bits)
64 > * - b = background color (8 bits)
65 > *
66 > */
67 > export const enum MetadataConsts {
68 > LANGUAGEID_MASK /* */ = 0b00000000_00000000_00000000_11111111,
69 > TOKEN_TYPE_MASK /* */ = 0b00000000_00000000_00000011_00000000,
70 > BALANCED_BRACKETS_MASK /* */ = 0b00000000_00000000_00000100_00000000,
71 > FONT_STYLE_MASK /* */ = 0b00000000_00000000_01111000_00000000,
72 > FOREGROUND_MASK /* */ = 0b00000000_11111111_10000000_00000000,
73 > BACKGROUND_MASK /* */ = 0b11111111_00000000_00000000_00000000,
74 >
75 > ITALIC_MASK /* */ = 0b00000000_00000000_00001000_00000000,
76 > BOLD_MASK /* */ = 0b00000000_00000000_00010000_00000000,
77 > UNDERLINE_MASK /* */ = 0b00000000_00000000_00100000_00000000,
78 > STRIKETHROUGH_MASK /* */ = 0b00000000_00000000_01000000_00000000,
79 >
80 > // Semantic tokens cannot set the language id, so we can
81 > // use the first 8 bits for control purposes
82 > SEMANTIC_USE_ITALIC /* */ = 0b00000000_00000000_00000000_00000001,
83 > SEMANTIC_USE_BOLD /* */ = 0b00000000_00000000_00000000_00000010,
84 > SEMANTIC_USE_UNDERLINE /* */ = 0b00000000_00000000_00000000_00000100,
85 > SEMANTIC_USE_STRIKETHROUGH /* */ = 0b00000000_00000000_00000000_00001000,
86 > SEMANTIC_USE_FOREGROUND /* */ = 0b00000000_00000000_00000000_00010000,
87 > SEMANTIC_USE_BACKGROUND /* */ = 0b00000000_00000000_00000000_00100000,
88 >
89 > LANGUAGEID_OFFSET = 0,
90 > TOKEN_TYPE_OFFSET = 8,
91 > BALANCED_BRACKETS_OFFSET = 10,
92 > FONT_STYLE_OFFSET = 11,
93 > FOREGROUND_OFFSET = 15,
94 > BACKGROUND_OFFSET = 24
95 > }
96 >
97 > /**
98 > */
99 > export class TokenMetadata {
100 >
101 > public static getLanguageId(metadata: number): LanguageId {
102 return (metadata & MetadataConsts.LANGUAGEID_MASK) >>> MetadataConsts.LANGUAGEID_OFFSET;
103 }
105 > public static getTokenType(metadata: number): StandardTokenType {
106 return (metadata & MetadataConsts.TOKEN_TYPE_MASK) >>> MetadataConsts.TOKEN_TYPE_OFFSET;
107 }
109 > public static containsBalancedBrackets(metadata: number): boolean {
110 return (metadata & MetadataConsts.BALANCED_BRACKETS_MASK) !== 0;
111 }
113 > public static getFontStyle(metadata: number): FontStyle {
114 return (metadata & MetadataConsts.FONT_STYLE_MASK) >>> MetadataConsts.FONT_STYLE_OFFSET;
115 }
117 > public static getForeground(metadata: number): ColorId {
118 return (metadata & MetadataConsts.FOREGROUND_MASK) >>> MetadataConsts.FOREGROUND_OFFSET;
119 }
121 > public static getBackground(metadata: number): ColorId {
122 return (metadata & MetadataConsts.BACKGROUND_MASK) >>> MetadataConsts.BACKGROUND_OFFSET;
123 }
125 > public static getClassNameFromMetadata(metadata: number): string {
126 const foreground = this.getForeground(metadata);
127 let className = 'mtk' + foreground;
143 return className;
144 }
146 > public static getInlineStyleFromMetadata(metadata: number, colorMap: string[]): string {
147 const foreground = this.getForeground(metadata);
148 const fontStyle = this.getFontStyle(metadata);
168 return result;
169 }
171 > public static getPresentationFromMetadata(metadata: number): ITokenPresentation {
172 const foreground = this.getForeground(metadata);
173 const fontStyle = this.getFontStyle(metadata);
181 };
182 }
184 >
185 > /**
186 > */
187 > export interface ITokenPresentation {
188 > foreground: ColorId;
189 > italic: boolean;
190 > bold: boolean;
191 > underline: boolean;
192 > strikethrough: boolean;
193 > }
src/vs/workbench/contrib/chat/common/model/chatProgressTypes/chatToolInvocation.ts 127 covered LOC · 20 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatToolInvocation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { encodeBase64 } from '../../../../../../base/common/buffer.js';
7 > import { IMarkdownString } from '../../../../../../base/common/htmlContent.js';
8 > import { IObservable, ISettableObservable, observableValue } from '../../../../../../base/common/observable.js';
9 > import { ThemeIcon } from '../../../../../../base/common/themables.js';
10 > import { localize } from '../../../../../../nls.js';
11 > import { ConfirmedReason, IChatAgentFeedbackReviewConfirmationData, IChatAutomationConfigurationData, IChatAutomationConfiguredData, IChatExtensionsContent, IChatModifiedFilesConfirmationData, IChatSearchToolInvocationData, IChatSessionCreatedData, IChatSimpleToolInvocationData, IChatSubagentToolInvocationData, IChatTodoListContent, IChatToolInputInvocationData, IChatToolInvocation, IChatToolInvocationOtherClientData, IChatToolInvocationSerialized, ToolConfirmKind, type IChatMcpAuthenticationRequiredServer, type IChatTerminalToolInvocationData } from '../../chatService/chatService.js';
12 > import { IPreparedToolInvocation, isToolResultOutputDetails, IToolConfirmationMessages, IToolData, IToolProgressStep, IToolResult, ToolDataSource } from '../../tools/languageModelToolsService.js';
13 >
14 > export interface IStreamingToolCallOptions {
15 > toolCallId: string;
16 > toolId: string;
17 > toolData: IToolData;
18 > subagentInvocationId?: string;
19 > chatRequestId?: string;
20 > }
21 >
22 > export class ChatToolInvocation implements IChatToolInvocation {
23 > public readonly kind: 'toolInvocation' = 'toolInvocation';
24 >
25 > public invocationMessage: string | IMarkdownString;
26 > public readonly originMessage: string | IMarkdownString | undefined;
27 > public pastTenseMessage: string | IMarkdownString | undefined;
28 > public confirmationMessages: IToolConfirmationMessages | undefined;
29 > public presentation: IPreparedToolInvocation['presentation'];
30 > public readonly toolId: string;
31 > public readonly icon?: ThemeIcon;
32 > public source: ToolDataSource;
33 > public readonly subAgentInvocationId: string | undefined;
34 > public parameters: unknown;
35 > public generatedTitle?: string;
36 > public readonly chatRequestId?: string;
37 > public isAttachedToThinking: boolean = false;
38 > public otherClientToolCall?: IChatToolInvocationOtherClientData;
39 >
40 > private _toolSpecificData?: IChatTerminalToolInvocationData | IChatToolInputInvocationData | IChatExtensionsContent | IChatTodoListContent | IChatSubagentToolInvocationData | IChatSimpleToolInvocationData | IChatSearchToolInvocationData | IChatModifiedFilesConfirmationData | IChatAgentFeedbackReviewConfirmationData | IChatSessionCreatedData | IChatAutomationConfigurationData | IChatAutomationConfiguredData;
41 > private readonly _toolSpecificDataKind = observableValue<string | undefined>(this, undefined);
42 > public readonly toolSpecificDataKind: IObservable<string | undefined> = this._toolSpecificDataKind;
43 >
44 > public get toolSpecificData() {
45 > return this._toolSpecificData;
46 > }
47 >
48 > public set toolSpecificData(value: typeof this._toolSpecificData) {
49 this._toolSpecificData = value;
50 this._toolSpecificDataKind.set(value?.kind, undefined);
51 }
53 > private readonly _progress = observableValue<{ message?: string | IMarkdownString; progress: number | undefined }>(this, { progress: 0 });
54 > private readonly _state: ISettableObservable<IChatToolInvocation.State>;
55 >
56 > // Streaming-related observables
57 > private readonly _partialInput = observableValue<unknown>(this, undefined);
58 > private readonly _streamingMessage = observableValue<string | IMarkdownString | undefined>(this, undefined);
59 >
60 > public get state(): IObservable<IChatToolInvocation.State> {
61 return this._state;
62 }
64 > /**
65 > * Create a tool invocation in streaming state.
66 > * Use this when the tool call is beginning to stream partial input from the LM.
67 > */
68 > public static createStreaming(options: IStreamingToolCallOptions): ChatToolInvocation {
69 return new ChatToolInvocation(undefined, options.toolData, options.toolCallId, options.subagentInvocationId, undefined, { startInStreaming: true }, options.chatRequestId);
70 }
72 > /**
73 > * Create a tool invocation already in cancelled state.
74 > * Use this when a hook denies tool execution before it even starts.
75 > */
76 > public static createCancelled(options: IStreamingToolCallOptions, parameters: unknown, reason: ToolConfirmKind.Denied | ToolConfirmKind.Skipped, reasonMessage?: string | IMarkdownString): ChatToolInvocation {
77 return new ChatToolInvocation(undefined, options.toolData, options.toolCallId, options.subagentInvocationId, parameters, { startInCancelled: true, cancelReason: reason, cancelReasonMessage: reasonMessage }, options.chatRequestId);
78 }
80 > constructor(
81 preparedInvocation: IPreparedToolInvocation | undefined,
82 toolData: IToolData,
140 }
141 }
143 > /**
144 > * Shared confirmation handler used by every `WaitingForConfirmation` state
145 > * this invocation can enter (initial construction, transition out of
146 > * streaming, and re-arming via {@link requestConfirmation}). Denials/skips
147 > * cancel; anything else moves to executing.
148 > */
149 > private _confirm(reason: ConfirmedReason): void {
150 if (reason.type === ToolConfirmKind.Denied || reason.type === ToolConfirmKind.Skipped) {
151 this._state.set({
165 }
166 }
168 > /**
169 > * Update the partial input observable during streaming.
170 > */
171 > public updatePartialInput(input: unknown): void {
172 if (this._state.get().type !== IChatToolInvocation.StateKind.Streaming) {
173 return; // Only update in streaming state
175 this._partialInput.set(input, undefined);
176 }
178 > /**
179 > * Update the streaming message (from handleToolStream).
180 > */
181 > public updateStreamingMessage(message: string | IMarkdownString): void {
182 const state = this._state.get();
183 if (state.type !== IChatToolInvocation.StateKind.Streaming) {
186 this._streamingMessage.set(message, undefined);
187 }
189 > /**
190 > * Notifies state observers that `toolSpecificData` has been mutated.
191 > * Since `toolSpecificData` isn't observable, this re-sets the internal
192 > * state to trigger autoruns that need to re-read tool metadata.
193 > */
194 > public notifyToolSpecificDataChanged(): void {
195 const current = this._state.get();
196 this._state.set({ ...current }, undefined);
197 }
199 > public updateConfirmationMessages(confirmationMessages: IToolConfirmationMessages): void {
200 const current = this._state.get();
201 if (current.type !== IChatToolInvocation.StateKind.WaitingForConfirmation) {
205 this._state.set({ ...current, confirmationMessages }, undefined);
206 }
208 > /**
209 > * Cancel a streaming invocation directly (e.g., when preToolUse hook denies).
210 > * Only works when in Streaming state.
211 > * @returns true if the cancellation was applied, false if not in streaming state
212 > */
213 > public cancelFromStreaming(reason: ToolConfirmKind.Denied | ToolConfirmKind.Skipped, reasonMessage?: string | IMarkdownString): boolean {
214 const currentState = this._state.get();
215 if (currentState.type !== IChatToolInvocation.StateKind.Streaming) {
226 return true;
227 }
229 > /**
230 > * Transition from streaming state to prepared/executing state.
231 > * Called when the full tool call is ready.
232 > */
233 > public transitionFromStreaming(preparedInvocation: IPreparedToolInvocation | undefined, parameters: unknown, autoConfirmed: ConfirmedReason | undefined): void {
234 const currentState = this._state.get();
235 if (currentState.type !== IChatToolInvocation.StateKind.Streaming) {
275 }
276 }
278 > /** Moves an active invocation into confirmation while preserving the same tool card. */
279 > public requestConfirmation(preparedInvocation: IPreparedToolInvocation): void {
280 const currentType = this._state.get().type;
281 if (currentType === IChatToolInvocation.StateKind.Streaming) {
308 }, undefined);
309 }
311 > private _setCompleted(result: IToolResult | undefined, postConfirmed?: ConfirmedReason | undefined) {
312 if (postConfirmed && (postConfirmed.type === ToolConfirmKind.Denied || postConfirmed.type === ToolConfirmKind.Skipped)) {
313 this._state.set({
330 }, undefined);
331 }
333 > public async didExecuteTool(result: IToolResult | undefined, final?: boolean, checkIfResultAutoApproved?: () => Promise<ConfirmedReason | undefined>): Promise<IChatToolInvocation.State> {
334 if (result?.toolSpecificData) {
335 this.toolSpecificData = result.toolSpecificData;
362 return this._state.get();
363 }
365 > public setAuthenticationRequired(server: IChatMcpAuthenticationRequiredServer, cancel: () => void = () => { }): void {
366 const state = this._state.get();
367 if (state.type !== IChatToolInvocation.StateKind.Executing && state.type !== IChatToolInvocation.StateKind.WaitingForAuthentication) {
377 }, undefined);
378 }
380 > public setAuthenticationResolved(): void {
381 const state = this._state.get();
382 if (state.type !== IChatToolInvocation.StateKind.WaitingForAuthentication) {
391 }, undefined);
392 }
394 > public acceptProgress(step: IToolProgressStep) {
395 const prev = this._progress.get();
396 this._progress.set({
399 }, undefined);
400 }
402 > public toJSON(): IChatToolInvocationSerialized {
403 // persist the serialized call as 'skipped' if we were waiting for postapproval
404 const waitingForPostApproval = this.state.get().type === IChatToolInvocation.StateKind.WaitingForPostApproval;
src/vs/platform/agentHost/common/agentHostConnectionsService.ts 126 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentHostConnectionsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import type { URI } from '../../../base/common/uri.js';
8 > import { createDecorator } from '../../instantiation/common/instantiation.js';
9 > import type { IAgentConnection } from './agentService.js';
10 >
11 > /**
12 > * Chat-session resource scheme prefix for the window's ambient/local agent
13 > * host: `agent-host-<provider>`.
14 > *
15 > * Remote agent host session schemes (`remote-<authority>-<provider>`) are
16 > * handled by `agentHostSessionType.ts` (`isRemoteAgentHostSessionType` and
17 > * friends), which also owns the authority disambiguation.
18 > */
19 > export const LOCAL_AGENT_HOST_SCHEME_PREFIX = 'agent-host-';
20 >
21 > /**
22 > * Reserved connection authority for the window's ambient/primary agent host.
23 > *
24 > * NOTE: "ambient" is not the same as "local". In a local window the ambient
25 > * host is the in-process utility agent host; in a window attached to a remote
26 > * authority the ambient host is itself remote (the `EditorRemoteAgentHostServiceClient`).
27 > * The `'local'` string is reserved/canonical for this ambient connection — it
28 > * is already used as the agent-host URI authority for in-window resources
29 > * (see `toAgentHostUri` in `agentHostUri.ts`).
30 > */
31 > export const AMBIENT_AGENT_HOST_AUTHORITY = 'local';
32 >
33 > /**
34 > * A descriptor for a single agent-host connection exposed by
35 > * {@link IAgentHostConnectionsService}. Covers both the window's ambient host
36 > * and every connected remote host with one uniform shape.
37 > */
38 > export interface IAgentHostConnectionInfo {
39 > /**
40 > * Sanitized connection authority. The ambient host uses
41 > * {@link AMBIENT_AGENT_HOST_AUTHORITY}; remotes use the authority derived
42 > * from their address via `agentHostAuthority`.
43 > */
44 > readonly authority: string;
45 > /** Raw remote address. `undefined` for the ambient host. */
46 > readonly address: string | undefined;
47 > /** Human-readable label for the connection. */
48 > readonly name: string;
49 > /**
50 > * `true` for the window's ambient/primary host. Remember this host may
51 > * itself be remote in a remote window — see {@link AMBIENT_AGENT_HOST_AUTHORITY}.
52 > */
53 > readonly isAmbient: boolean;
54 > /** The live connection, or `undefined` when not currently connected. */
55 > readonly connection: IAgentConnection | undefined;
56 > }
57 >
58 > /**
59 > * The result of resolving a chat-session resource to its backing agent host:
60 > * the owning {@link IAgentConnection} and the canonical backend agent-session
61 > * URI (`<provider>:/<rawId>`) used for protocol operations on that connection.
62 > */
63 > export interface IAgentHostSessionResolution {
64 > readonly connection: IAgentConnection;
65 > readonly backendSession: URI;
66 > }
67 >
68 > export const IAgentHostConnectionsService = createDecorator<IAgentHostConnectionsService>('agentHostConnectionsService');
69 >
70 > /**
71 > * A thin, read-only facade over the window's ambient agent host
72 > * (`IAgentHostService`) and the registry of remote agent hosts
73 > * (`IRemoteAgentHostService`), so consumers can enumerate and resolve
74 > * {@link IAgentConnection}s without branching on local-vs-remote or
75 > * fanning out over "1 ambient + N remote" themselves.
76 > *
77 > * This service deliberately does NOT expose lifecycle/management operations:
78 > * ambient-process concerns (restart, inspect, auth-pending) stay on
79 > * `IAgentHostService`, and remote-registry mutations (add/remove/reconnect/
80 > * upgrade) stay on `IRemoteAgentHostService`. This facade only answers
81 > * "which connections exist?" and "give me the connection for X".
82 > */
83 > export interface IAgentHostConnectionsService {
84 > readonly _serviceBrand: undefined;
85 >
86 > /** Fires when the set of connections changes (ambient lifecycle or remotes added/removed). */
87 > readonly onDidChangeConnections: Event<void>;
88 >
89 > /**
90 > * All known connections as `[ambient, ...remotes]`. The ambient entry is
91 > * always present with a live `connection`; only remote entries may have
92 > * `connection: undefined` (e.g. while connecting/disconnected). Remote
93 > * entries reflect the current `IRemoteAgentHostService` registry.
94 > */
95 > readonly connections: readonly IAgentHostConnectionInfo[];
96 >
97 > /** The window's ambient/primary connection (local, or the window-remote bridge). */
98 > readonly ambientConnection: IAgentConnection;
99 >
100 > /**
101 > * Resolves a live connection by sanitized authority, including
102 > * {@link AMBIENT_AGENT_HOST_AUTHORITY} for the ambient host. Returns
103 > * `undefined` when no connected host matches.
104 > */
105 > getConnectionByAuthority(authority: string): IAgentConnection | undefined;
106 >
107 > /**
108 > * Resolves a live remote connection by raw address. The ambient host has no
109 > * address and is never returned here — use {@link ambientConnection} or
110 > * {@link getConnectionByAuthority} with {@link AMBIENT_AGENT_HOST_AUTHORITY}.
111 > */
112 > getConnectionByAddress(address: string): IAgentConnection | undefined;
113 >
114 > /**
115 > * Resolves an agent-host chat-session resource to its owning connection and
116 > * backend session URI. Handles both local schemes
117 > * (`agent-host-<provider>`) — backed by the ambient connection — and remote
118 > * schemes (`remote-<authority>-<provider>`) — resolved against the live
119 > * remote registry. Returns `undefined` when the resource is not an
120 > * agent-host session, or when the matching remote host is not connected.
121 > *
122 > * NOTE: provisional/untitled sessions are a workbench concern and are NOT
123 > * handled here — callers that support them should resolve those first.
124 > */
125 > resolveSessionResource(sessionResource: URI): IAgentHostSessionResolution | undefined;
126 > }
src/vs/platform/instantiation/common/instantiation.ts 126 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- instantiation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DisposableStore } from '../../../base/common/lifecycle.js';
7 > import * as descriptors from './descriptors.js';
8 > import { ServiceCollection } from './serviceCollection.js';
9 >
10 > // ------ internal util
11 >
12 > export namespace _util {
13 >
14 > export const serviceIds = new Map<string, ServiceIdentifier<any>>();
15 >
16 > export const DI_TARGET = '$di$target';
17 > export const DI_DEPENDENCIES = '$di$dependencies';
18 >
19 > export function getServiceDependencies(ctor: DI_TARGET_OBJ): { id: ServiceIdentifier<any>; index: number }[] {
20 > return ctor[DI_DEPENDENCIES] || []; instantiation.ts
21 > }
23 > export interface DI_TARGET_OBJ extends Function {
24 > [DI_TARGET]: Function;
25 > [DI_DEPENDENCIES]: { id: ServiceIdentifier<any>; index: number }[];
26 > }
27 > }
28 >
29 > // --- interfaces ------
30 >
31 > export type BrandedService = { _serviceBrand: undefined };
32 >
33 > export interface IConstructorSignature<T, Args extends any[] = []> {
34 > new <Services extends BrandedService[]>(...args: [...Args, ...Services]): T;
35 > }
36 >
37 > export interface ServicesAccessor {
38 > get<T>(id: ServiceIdentifier<T>): T;
39 > }
40 >
41 > export const IInstantiationService = createDecorator<IInstantiationService>('instantiationService');
42 >
43 > /**
44 > * Given a list of arguments as a tuple, attempt to extract the leading, non-service arguments
45 > * to their own tuple.
46 > */
47 > export type GetLeadingNonServiceArgs<TArgs extends any[]> =
48 > TArgs extends [] ? []
49 > : TArgs extends [...infer TFirst, BrandedService] ? GetLeadingNonServiceArgs<TFirst>
50 > : TArgs;
51 >
52 > export interface IInstantiationService {
53 >
54 > readonly _serviceBrand: undefined;
55 >
56 > /**
57 > * Synchronously creates an instance that is denoted by the descriptor
58 > */
59 > createInstance<T>(descriptor: descriptors.SyncDescriptor0<T>): T;
60 > createInstance<Ctor extends new (...args: any[]) => unknown, R extends InstanceType<Ctor>>(ctor: Ctor, ...args: GetLeadingNonServiceArgs<ConstructorParameters<Ctor>>): R;
61 >
62 > /**
63 > * Calls a function with a service accessor.
64 > */
65 > invokeFunction<R, TS extends any[] = []>(fn: (accessor: ServicesAccessor, ...args: TS) => R, ...args: TS): R;
66 >
67 > /**
68 > * Creates a child of this service which inherits all current services
69 > * and adds/overwrites the given services.
70 > *
71 > * NOTE that the returned child is `disposable` and should be disposed when not used
72 > * anymore. This will also dispose all the services that this service has created.
73 > */
74 > createChild(services: ServiceCollection, store?: DisposableStore): IInstantiationService;
75 >
76 > /**
77 > * Disposes this instantiation service.
78 > *
79 > * - Will dispose all services that this instantiation service has created.
80 > * - Will dispose all its children but not its parent.
81 > * - Will NOT dispose services-instances that this service has been created with
82 > * - Will NOT dispose consumer-instances this service has created
83 > */
84 > dispose(): void;
85 > }
86 >
87 >
88 > /**
89 > * Identifies a service of type `T`.
90 > */
91 > export interface ServiceIdentifier<T> {
92 > (...args: any[]): void;
93 > type: T;
94 > }
95 >
96 >
97 > function storeServiceDependency(id: ServiceIdentifier<unknown>, target: Function, index: number): void { instantiation.ts
98 > if ((target as _util.DI_TARGET_OBJ)[_util.DI_TARGET] === target) {
99 > (target as _util.DI_TARGET_OBJ)[_util.DI_DEPENDENCIES].push({ id, index }); instantiation.ts
100 > } else { instantiation.ts
101 > (target as _util.DI_TARGET_OBJ)[_util.DI_DEPENDENCIES] = [{ id, index }];
102 > (target as _util.DI_TARGET_OBJ)[_util.DI_TARGET] = target;
103 > }
104 > }
106 > /**
107 > * The *only* valid way to create a {{ServiceIdentifier}}.
108 > */
109 > export function createDecorator<T>(serviceId: string): ServiceIdentifier<T> {
110 >
111 > if (_util.serviceIds.has(serviceId)) {
112 return _util.serviceIds.get(serviceId)!;
113 }
115 > const id = function (target: Function, key: string, index: number) {
116 > if (arguments.length !== 3) { instantiation.ts
117 throw new Error('@IServiceName-decorator can only be used to decorate a parameter');
118 }
119 > storeServiceDependency(id, target, index); instantiation.ts
120 > } as ServiceIdentifier<T>;
122 > id.toString = () => serviceId;
123 >
124 > _util.serviceIds.set(serviceId, id);
125 > return id;
126 > }
127 >
128 > export function refineServiceDecorator<T1, T extends T1>(serviceIdentifier: ServiceIdentifier<T1>): ServiceIdentifier<T> {
129 > return <ServiceIdentifier<T>>serviceIdentifier; instantiation.ts
130 > }
src/vs/base/test/common/virtualScheduling/trace.ts 122 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- trace.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { BugIndicatingError } from '../../../common/errors.js';
7 >
8 > /**
9 > * # Trace — causal-chain attribution for scheduled work
10 > *
11 > * A {@link Trace} is an immutable value identifying a causal chain. Every
12 > * non-root trace carries a `parent`; the head of the chain has no parent.
13 > * Use {@link child} to extend a chain when scheduling follow-up work.
14 > *
15 > * Traces are used to answer "who caused this?" for any virtual event:
16 > * useful for debugging, for per-owner termination, and for attribution in
17 > * error messages.
18 > */
19 > export class Trace {
20 > private static _idCounter = 0;
21 > public readonly id: number = ++Trace._idCounter;
22 > public readonly root: Trace;
23 > public readonly depth: number;
24 >
25 > constructor(
26 > public readonly parent: Trace | undefined,
27 > public readonly label: string,
28 > public readonly stack: string | undefined = undefined,
29 > ) {
30 > this.root = parent?.root ?? this;
31 > this.depth = (parent?.depth ?? -1) + 1;
32 > }
33 >
34 > child(label: string, stack?: string): Trace {
35 return new Trace(this, label, stack);
36 }
37 > trace.ts
38 > /** "#id label ← #id label ← … ← #id label" */
39 > describe(): string {
40 const parts: string[] = [];
41 for (let t: Trace | undefined = this; t; t = t.parent) {
44 return parts.join(' ← ');
45 }
46 > trace.ts
47 > toString(): string { return this.describe(); }
48 > }
49 >
50 > /** Sentinel for "no known causal predecessor". */
51 > export const ROOT_TRACE: Trace = new Trace(undefined, '<root>');
52 >
53 > export function createTraceRoot(label: string, stack?: string): Trace {
54 return new Trace(undefined, label, stack);
55 }
56 > trace.ts
57 > interface Frame {
58 > readonly trace: Trace;
59 > readonly prev: Frame | undefined;
60 > }
61 >
62 > const ROOT_FRAME: Frame = { trace: ROOT_TRACE, prev: undefined };
63 >
64 > /**
65 > * Options for {@link TraceContext.runAsHandler}.
66 > *
67 > * # Why this is a per-call option
68 > *
69 > * `runAsHandler` cannot restore the previous trace synchronously: microtasks
70 > * enqueued by `fn` (including awaited continuations) must observe the new
71 > * trace. So the reset is deferred — but it must fire after the *closure* of
72 > * the microtask queue (the current microtask plus every microtask it
73 > * recursively enqueues), not just one drain.
74 > *
75 > * Per spec, the host doesn't run a macrotask until the microtask queue is
76 > * empty, so any macrotask primitive (`setTimeout(0)`, `setImmediate`, the
77 > * `setTimeout0` shim) achieves this. Letting the *caller* supply the sink
78 > * means:
79 > *
80 > * - the {@link VirtualTimeProcessor} can route the reset through the same
81 > * primitive its embedding uses for its own host hops, eliminating any
82 > * race between the processor's hops and the trace-reset timer;
83 > *
84 > * - production code without a processor can still use a real
85 > * `setTimeout(0)`-based sink and get the same semantics;
86 > *
87 > * - tests can install a deterministic sink (e.g. a hand-driven queue) for
88 > * fully synchronous assertions.
89 > */
90 > export interface RunAsHandlerOptions {
91 > /**
92 > * Sink for the deferred trace-reset.
93 > *
94 > * Must invoke `reset` after the microtask closure that follows the
95 > * `runAsHandler` call returns — i.e. on the next host macrotask.
96 > */
97 > readonly afterMicrotaskClosure: (reset: () => void) => void;
98 > }
99 >
100 > /**
101 > * Holds the mutable "current trace frame" slot. Construct fresh instances
102 > * for test isolation, or use {@link TraceContext.instance} for shared state.
103 > */
104 > export class TraceContext {
105 > public static readonly instance = new TraceContext();
106 >
107 > private _current: Frame = ROOT_FRAME;
108 > private _isHandlerRunning = false;
109 >
110 > currentTrace(): Trace { return this._current.trace; }
111 >
112 > /**
113 > * Install `t` as current for the synchronous duration of `fn`, then
114 > * restore. Nestable. Microtasks enqueued by fn that run after fn returns
115 > * see the *restored* trace — use {@link runAsHandler} when continuation
116 > * inheritance is wanted.
117 > */
118 > runWithTrace<T>(t: Trace, fn: () => T): T {
119 const prev = this._current;
120 const next: Frame = { trace: t, prev };
132 }
133 }
134 > trace.ts
135 > /**
136 > * Install `t` as current and run `fn`. The trace stays current through
137 > * the microtask closure that follows `fn`, so awaited continuations
138 > * inside fn observe `t`. The reset is dispatched via
139 > * `opts.afterMicrotaskClosure`.
140 > *
141 > * Throws on synchronous re-entry: timer callbacks never nest on the
142 > * same JS stack frame, so this only fires for misuse.
143 > */
144 > runAsHandler<T>(t: Trace, fn: () => T, opts: RunAsHandlerOptions): T {
145 if (this._isHandlerRunning) {
146 throw new Error(
165 }
166 }
167 > trace.ts
168 > _resetForTesting(): void {
169 this._current = ROOT_FRAME;
170 this._isHandlerRunning = false;
171 }
172 > } trace.ts
src/vs/workbench/contrib/chat/common/promptSyntax/hookTypes.ts 122 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hookTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as nls from '../../../../../nls.js';
7 > import { Target } from './promptTypes.js';
8 >
9 > /**
10 > * Enum of hook types across all targets. For the set of supported hooks per target, see HOOKS_BY_TARGET.
11 > */
12 > export enum HookType {
13 > SessionStart = 'SessionStart',
14 > SessionEnd = 'SessionEnd',
15 > UserPromptSubmit = 'UserPromptSubmit',
16 > PreToolUse = 'PreToolUse',
17 > PostToolUse = 'PostToolUse',
18 > PreCompact = 'PreCompact',
19 > SubagentStart = 'SubagentStart',
20 > SubagentStop = 'SubagentStop',
21 > Stop = 'Stop',
22 > ErrorOccurred = 'ErrorOccurred',
23 > }
24 >
25 > /**
26 > * String literal type derived from HookType enum values.
27 > */
28 > export type HookTypeValue = `${HookType}`;
29 >
30 > export const HOOKS_BY_TARGET: Record<Target, Record<string, HookType>> = {
31 > // see https://code.visualstudio.com/docs/copilot/customization/hooks#_hook-lifecycle-events
32 > [Target.VSCode]: {
33 > 'SessionStart': HookType.SessionStart,
34 > 'UserPromptSubmit': HookType.UserPromptSubmit,
35 > 'PreToolUse': HookType.PreToolUse,
36 > 'PostToolUse': HookType.PostToolUse,
37 > 'PreCompact': HookType.PreCompact,
38 > 'SubagentStart': HookType.SubagentStart,
39 > 'SubagentStop': HookType.SubagentStop,
40 > 'Stop': HookType.Stop,
41 > },
42 > // see https://docs.github.com/en/copilot/concepts/agents/coding-agent/about-hooks#types-of-hooks
43 > [Target.GitHubCopilot]: {
44 > 'sessionStart': HookType.SessionStart,
45 > 'sessionEnd': HookType.SessionEnd,
46 > 'userPromptSubmitted': HookType.UserPromptSubmit,
47 > 'preToolUse': HookType.PreToolUse,
48 > 'postToolUse': HookType.PostToolUse,
49 > 'agentStop': HookType.Stop,
50 > 'subagentStop': HookType.SubagentStop,
51 > 'errorOccurred': HookType.ErrorOccurred
52 > },
53 > // see https://docs.anthropic.com/en/docs/claude-code/hooks
54 > [Target.Claude]: {
55 > 'SessionStart': HookType.SessionStart,
56 > 'UserPromptSubmit': HookType.UserPromptSubmit,
57 > 'PreToolUse': HookType.PreToolUse,
58 > 'PostToolUse': HookType.PostToolUse,
59 > 'PreCompact': HookType.PreCompact,
60 > 'SubagentStart': HookType.SubagentStart,
61 > 'SubagentStop': HookType.SubagentStop,
62 > 'Stop': HookType.Stop,
63 > },
64 > // if no target, just list all known hook types.
65 > [Target.Undefined]: Object.fromEntries(
66 > Object.values(HookType).map(h => [h, h])
67 > ) as Record<string, HookType>
68 > };
69 >
70 > /**
71 > * Metadata for a hook type including localized label and description.
72 > */
73 > export interface IHookTypeMeta {
74 > readonly label: string;
75 > readonly description: string;
76 > }
77 >
78 > /**
79 > * Metadata for hook types including localized labels and descriptions
80 > */
81 > export const HOOK_METADATA: { [key in HookType]: IHookTypeMeta } = {
82 > [HookType.SessionStart]: {
83 > label: nls.localize('hookType.sessionStart.label', "Session Start"),
84 > description: nls.localize('hookType.sessionStart.description', "Executed when a new agent session begins.")
85 > },
86 > [HookType.UserPromptSubmit]: {
87 > label: nls.localize('hookType.userPromptSubmit.label', "User Prompt Submit"),
88 > description: nls.localize('hookType.userPromptSubmit.description', "Executed when the user submits a prompt to the agent.")
89 > },
90 > [HookType.PreToolUse]: {
91 > label: nls.localize('hookType.preToolUse.label', "Pre-Tool Use"),
92 > description: nls.localize('hookType.preToolUse.description', "Executed before the agent uses any tool.")
93 > },
94 > [HookType.PostToolUse]: {
95 > label: nls.localize('hookType.postToolUse.label', "Post-Tool Use"),
96 > description: nls.localize('hookType.postToolUse.description', "Executed after a tool completes execution successfully.")
97 > },
98 > [HookType.PreCompact]: {
99 > label: nls.localize('hookType.preCompact.label', "Pre-Compact"),
100 > description: nls.localize('hookType.preCompact.description', "Executed before the agent compacts the conversation context.")
101 > },
102 > [HookType.SubagentStart]: {
103 > label: nls.localize('hookType.subagentStart.label', "Subagent Start"),
104 > description: nls.localize('hookType.subagentStart.description', "Executed when a subagent is started.")
105 > },
106 > [HookType.SubagentStop]: {
107 > label: nls.localize('hookType.subagentStop.label', "Subagent Stop"),
108 > description: nls.localize('hookType.subagentStop.description', "Executed when a subagent stops.")
109 > },
110 > [HookType.Stop]: {
111 > label: nls.localize('hookType.stop.label', "Stop"),
112 > description: nls.localize('hookType.stop.description', "Executed when the agent stops.")
113 > },
114 > [HookType.SessionEnd]: {
115 > label: nls.localize('hookType.sessionEnd.label', "Session End"),
116 > description: nls.localize('hookType.sessionEnd.description', "Executed when an agent session ends.")
117 > },
118 > [HookType.ErrorOccurred]: {
119 > label: nls.localize('hookType.errorOccurred.label', "Error Occurred"),
120 > description: nls.localize('hookType.errorOccurred.description', "Executed when an error occurs during the agent session.")
121 > }
122 > };
src/vs/platform/instantiation/test/common/instantiationServiceMock.ts 121 covered LOC · 33 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- instantiationServiceMock.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as sinon from 'sinon';
7 > import { DisposableStore, IDisposable, toDisposable } from '../../../../base/common/lifecycle.js';
8 > import { SyncDescriptor, SyncDescriptor0 } from '../../common/descriptors.js';
9 > import { GetLeadingNonServiceArgs, ServiceIdentifier, ServicesAccessor } from '../../common/instantiation.js';
10 > import { InstantiationService, Trace } from '../../common/instantiationService.js';
11 > import { ServiceCollection } from '../../common/serviceCollection.js';
12 >
13 > interface IServiceMock<T> {
14 > id: ServiceIdentifier<T>;
15 > service: any;
16 > }
17 >
18 > const isSinonSpyLike = (fn: Function): fn is sinon.SinonSpy => fn && 'callCount' in fn;
19 >
20 > export class TestInstantiationService extends InstantiationService implements IDisposable, ServicesAccessor {
21 >
22 > private _servciesMap: Map<ServiceIdentifier<any>, any>;
23 > private readonly _classStubs: Map<Function, any> = new Map();
24 > private readonly _parentTestService: TestInstantiationService | undefined;
25 >
26 > constructor(private _serviceCollection: ServiceCollection = new ServiceCollection(), strict: boolean = false, parent?: InstantiationService, private _properDispose?: boolean) {
27 > super(_serviceCollection, strict, parent); instantiationServiceMock.ts
28 >
29 > this._servciesMap = new Map<ServiceIdentifier<any>, any>();
30 > if (parent instanceof TestInstantiationService) {
31 this._parentTestService = parent;
32 }
35 > public get<T>(service: ServiceIdentifier<T>): T {
36 return super._getOrCreateServiceInstance(service, Trace.traceCreation(false, TestInstantiationService));
37 }
39 > public set<T>(service: ServiceIdentifier<T>, instance: T): T {
40 return <T>this._serviceCollection.set(service, instance);
41 }
43 > public mock<T>(service: ServiceIdentifier<T>): T | sinon.SinonMock {
44 return <T>this._create(service, { mock: true });
45 }
47 > public stubInstance<T>(ctor: new (...args: any[]) => T, instance: Partial<T>): void {
48 this._classStubs.set(ctor, instance);
49 }
51 > protected _getClassStub(ctor: Function): unknown {
52 > return this._classStubs.get(ctor) ?? this._parentTestService?._getClassStub(ctor); instantiationServiceMock.ts
53 > }
55 > public override createInstance<T>(descriptor: SyncDescriptor0<T>): T;
56 > public override createInstance<Ctor extends new (...args: any[]) => unknown, R extends InstanceType<Ctor>>(ctor: Ctor, ...args: GetLeadingNonServiceArgs<ConstructorParameters<Ctor>>): R;
57 > public override createInstance(ctorOrDescriptor: any | SyncDescriptor<any>, ...rest: unknown[]): unknown {
58 > const stub = this._getClassStub(ctorOrDescriptor as Function); instantiationServiceMock.ts
59 > if (stub) {
60 return stub;
61 }
62 > return super.createInstance(ctorOrDescriptor, ...rest); instantiationServiceMock.ts
63 > }
65 > public stub<T>(service: ServiceIdentifier<T>, obj: Partial<NoInfer<T>> | Function): T;
66 > public stub<T, V>(service: ServiceIdentifier<T>, obj: Partial<NoInfer<T>> | Function, property: string, value: V): V extends Function ? sinon.SinonSpy : sinon.SinonStub;
67 > public stub<T, V>(service: ServiceIdentifier<T>, property: string, value: V): V extends Function ? sinon.SinonSpy : sinon.SinonStub;
68 > public stub<T>(serviceIdentifier: ServiceIdentifier<T>, arg2: any, arg3?: string, arg4?: any): sinon.SinonStub | sinon.SinonSpy {
69 > const service = typeof arg2 !== 'string' ? arg2 : undefined; instantiationServiceMock.ts
70 > const serviceMock: IServiceMock<any> = { id: serviceIdentifier, service: service };
71 > const property = typeof arg2 === 'string' ? arg2 : arg3;
72 > const value = typeof arg2 === 'string' ? arg3 : arg4;
73 >
74 > const stubObject = this._create(serviceMock, { stub: true }, service && !property);
75 > if (property) {
76 if (stubObject[property]) {
77 if (stubObject[property].hasOwnProperty('restore')) {
91 }
92 }
93 > return stubObject; instantiationServiceMock.ts
94 > }
96 > public stubPromise<T>(service?: ServiceIdentifier<T>, fnProperty?: string, value?: any): T | sinon.SinonStub;
97 > public stubPromise<T, V>(service?: ServiceIdentifier<T>, ctor?: any, fnProperty?: string, value?: V): V extends Function ? sinon.SinonSpy : sinon.SinonStub;
98 > public stubPromise<T, V>(service?: ServiceIdentifier<T>, obj?: any, fnProperty?: string, value?: V): V extends Function ? sinon.SinonSpy : sinon.SinonStub;
99 > public stubPromise(arg1?: any, arg2?: any, arg3?: any, arg4?: any): sinon.SinonStub | sinon.SinonSpy {
100 arg3 = typeof arg2 === 'string' ? Promise.resolve(arg3) : arg3;
101 arg4 = typeof arg2 !== 'string' && typeof arg3 === 'string' ? Promise.resolve(arg4) : arg4;
102 return this.stub(arg1, arg2, arg3, arg4);
103 }
105 > public spy<T>(service: ServiceIdentifier<T>, fnProperty: string): sinon.SinonSpy {
106 const spy = sinon.spy();
107 this.stub(service, fnProperty, spy);
108 return spy;
109 }
111 > private _create<T>(serviceMock: IServiceMock<T>, options: SinonOptions, reset?: boolean): any;
112 > private _create<T>(ctor: any, options: SinonOptions): any;
113 > private _create(arg1: any, options: SinonOptions, reset: boolean = false): any {
114 > if (this.isServiceMock(arg1)) { instantiationServiceMock.ts
115 > const service = this._getOrCreateService(arg1, options, reset);
116 > this._serviceCollection.set(arg1.id, service);
117 > return service;
118 > }
119 > return options.mock ? sinon.mock(arg1) : this._createStub(arg1);
120 > }
122 > private _getOrCreateService<T>(serviceMock: IServiceMock<T>, opts: SinonOptions, reset?: boolean): any {
123 > const service: any = this._serviceCollection.get(serviceMock.id); instantiationServiceMock.ts
124 > if (!reset && service) {
125 if (opts.mock && service['sinonOptions'] && !!service['sinonOptions'].mock) {
126 return service;
130 }
131 }
132 > return this._createService(serviceMock, opts); instantiationServiceMock.ts
133 > }
135 > private _createService(serviceMock: IServiceMock<any>, opts: SinonOptions): any {
136 > serviceMock.service = serviceMock.service ? serviceMock.service : this._servciesMap.get(serviceMock.id); instantiationServiceMock.ts
137 > const service = opts.mock ? sinon.mock(serviceMock.service) : this._createStub(serviceMock.service);
138 > service['sinonOptions'] = opts;
139 > return service;
140 > }
142 > private _createStub(arg: any): any {
143 > return typeof arg === 'object' ? arg : sinon.createStubInstance(arg); instantiationServiceMock.ts
144 > }
146 > private isServiceMock(arg1: any): boolean {
147 > return typeof arg1 === 'object' && arg1.hasOwnProperty('id'); instantiationServiceMock.ts
148 > }
150 > override createChild(services: ServiceCollection): TestInstantiationService {
151 return new TestInstantiationService(services, false, this);
152 }
154 > override dispose() {
155 > sinon.restore(); instantiationServiceMock.ts
156 > if (this._properDispose) {
157 super.dispose();
158 }
161 >
162 > interface SinonOptions {
163 > mock?: boolean;
164 > stub?: boolean;
165 > }
166 >
167 > export type ServiceIdCtorPair<T> = [id: ServiceIdentifier<T>, ctorOrInstance: T | (new (...args: any[]) => T)];
168 >
169 > export function createServices(disposables: DisposableStore, services: ServiceIdCtorPair<any>[]): TestInstantiationService {
170 const serviceIdentifiers: ServiceIdentifier<any>[] = [];
171 const serviceCollection = new ServiceCollection();
src/vs/base/common/observableInternal/reactions/autorun.ts 120 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- autorun.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IReaderWithStore, IReader, IObservable, ISettableObservable } from '../base.js';
7 > import { IChangeTracker } from '../changeTracker.js';
8 > import { DisposableStore, IDisposable, toDisposable } from '../commonFacade/deps.js';
9 > import { DebugNameData, IDebugNameData } from '../debugName.js';
10 > import { AutorunObserver } from './autorunImpl.js';
11 > import { DebugLocation } from '../debugLocation.js';
12 > import { observableValue } from '../observables/observableValue.js';
13 > import { transaction } from '../transaction.js';
14 >
15 > /**
16 > * Runs immediately and whenever a transaction ends and an observed observable changed.
17 > * {@link fn} should start with a JS Doc using `@description` to name the autorun.
18 > */
19 > export function autorun(fn: (reader: IReaderWithStore) => void, debugLocation = DebugLocation.ofCaller()): IDisposable {
20 > return new AutorunObserver( autorun.ts
21 > new DebugNameData(undefined, undefined, fn),
22 > fn,
23 > undefined,
24 > debugLocation
25 > );
26 > }
27 > autorun.ts
28 > /**
29 > * Runs immediately and whenever a transaction ends and an observed observable changed.
30 > * {@link fn} should start with a JS Doc using `@description` to name the autorun.
31 > */
32 > export function autorunOpts(options: IDebugNameData & {}, fn: (reader: IReaderWithStore) => void, debugLocation = DebugLocation.ofCaller()): IDisposable {
33 return new AutorunObserver(
34 new DebugNameData(options.owner, options.debugName, options.debugReferenceFn ?? fn),
38 );
39 }
40 > autorun.ts
41 > /**
42 > * Runs immediately and whenever a transaction ends and an observed observable changed.
43 > * {@link fn} should start with a JS Doc using `@description` to name the autorun.
44 > *
45 > * Use `changeTracker.createChangeSummary` to create a "change summary" that can collect the changes.
46 > * Use `changeTracker.handleChange` to add a reported change to the change summary.
47 > * The run function is given the last change summary.
48 > * The change summary is discarded after the run function was called.
49 > *
50 > * @see autorun
51 > */
52 > export function autorunHandleChanges<TChangeSummary>(
53 options: IDebugNameData & {
54 changeTracker: IChangeTracker<TChangeSummary>;
64 );
65 }
66 > autorun.ts
67 > /**
68 > * @see autorunHandleChanges (but with a disposable store that is cleared before the next run or on dispose)
69 > */
70 > export function autorunWithStoreHandleChanges<TChangeSummary>(
71 options: IDebugNameData & {
72 changeTracker: IChangeTracker<TChangeSummary>;
92 });
93 }
94 > autorun.ts
95 > /**
96 > * @see autorun (but with a disposable store that is cleared before the next run or on dispose)
97 > *
98 > * @deprecated Use `autorun(reader => { reader.store.add(...) })` instead!
99 > */
100 > export function autorunWithStore(fn: (reader: IReader, store: DisposableStore) => void): IDisposable {
101 const store = new DisposableStore();
102 const disposable = autorunOpts(
116 });
117 }
118 > autorun.ts
119 > export function autorunDelta<T>(
120 observable: IObservable<T>,
121 handler: (args: { lastValue: T | undefined; newValue: T }) => void
129 });
130 }
131 > autorun.ts
132 > export function autorunIterableDelta<T>(
133 getValue: (reader: IReader) => Iterable<T>,
134 handler: (args: { addedValues: T[]; removedValues: T[] }) => void,
157 });
158 }
159 > autorun.ts
160 > /**
161 > * For each key-stable item in {@link items}, runs {@link setup} once when the
162 > * key is first observed and disposes the per-key {@link DisposableStore} when
163 > * the key is no longer present in the array (or when the returned disposable
164 > * is disposed).
165 > *
166 > * The {@link IObservable} handed to {@link setup} fires whenever the array
167 > * still contains an item with the same key but the item value itself has
168 > * changed (e.g. because the upstream state is immutable and produced a new
169 > * object with the same id). All per-key value updates triggered by a single
170 > * change to {@link items} are batched into one transaction, so dependent
171 > * autoruns observe a consistent snapshot.
172 > *
173 > * Per-key state should be stored in closures or in disposables registered
174 > * against the per-key {@link DisposableStore}. {@link setup} should not call
175 > * `.read()` on the outer {@link items} observable from its body (use the
176 > * provided per-key value observable, or create inner autoruns).
177 > */
178 > export function autorunPerKeyedItem<TIn, TKey>(
179 items: IObservable<readonly TIn[]>,
180 keyFn: (input: TIn) => TKey,
227 });
228 }
229 > autorun.ts
230 > export interface IReaderWithDispose extends IReaderWithStore, IDisposable { }
231 >
232 > /**
233 > * An autorun with a `dispose()` method on its `reader` which cancels the autorun.
234 > * It it safe to call `dispose()` synchronously.
235 > * @deprecated Use autorunSelfDisposable2
236 > */
237 > export function autorunSelfDisposable(fn: (reader: IReaderWithDispose) => void, debugLocation = DebugLocation.ofCaller()): IDisposable {
238 > let ar: IDisposable | undefined; autorun.ts
239 > let disposed = false;
240 >
241 > // eslint-disable-next-line prefer-const
242 > ar = autorun(reader => {
243 > fn({
244 > delayedStore: reader.delayedStore,
245 > store: reader.store,
246 > readObservable: reader.readObservable.bind(reader),
247 > dispose: () => {
248 > ar?.dispose();
249 > disposed = true;
250 > }
251 > });
252 > }, debugLocation);
253 >
254 > if (disposed) {
255 > ar.dispose();
256 > }
257 >
258 > return ar;
259 > }
260 > autorun.ts
261 >
262 > /**
263 > * An autorun with a `dispose()` method on its `reader` which cancels the autorun.
264 > * It it safe to call `dispose()` synchronously.
265 > * TODO@hediet/copilot: rename to delete autorunSelfDisposable, and rename autorunSelfDisposable2 to autorunSelfDisposable.
266 > */
267 > export function registerAutorunSelfDisposable(store: DisposableStore, fn: (reader: IReaderWithDispose) => void, debugLocation = DebugLocation.ofCaller()): void {
268 let ar: IDisposable | undefined;
269 let disposeSync = false;
src/vs/editor/common/core/position.ts 116 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- position.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * A position in the editor. This interface is suitable for serialization.
8 > */
9 > export interface IPosition {
10 > /**
11 > * line number (starts at 1)
12 > */
13 > readonly lineNumber: number;
14 > /**
15 > * column (the first character in a line is between column 1 and column 2)
16 > */
17 > readonly column: number;
18 > }
19 >
20 > /**
21 > * A position in the editor.
22 > */
23 > export class Position {
24 > /**
25 > * line number (starts at 1)
26 > */
27 > public readonly lineNumber: number;
28 > /**
29 > * column (the first character in a line is between column 1 and column 2)
30 > */
31 > public readonly column: number;
32 >
33 > constructor(lineNumber: number, column: number) {
34 > this.lineNumber = lineNumber; position.ts
35 > this.column = column;
36 > }
38 > /**
39 > * Create a new position from this position.
40 > *
41 > * @param newLineNumber new line number
42 > * @param newColumn new column
43 > */
44 > with(newLineNumber: number = this.lineNumber, newColumn: number = this.column): Position {
45 if (newLineNumber === this.lineNumber && newColumn === this.column) {
46 return this;
49 }
50 }
52 > /**
53 > * Derive a new position from this position.
54 > *
55 > * @param deltaLineNumber line number delta
56 > * @param deltaColumn column delta
57 > */
58 > delta(deltaLineNumber: number = 0, deltaColumn: number = 0): Position {
59 return this.with(Math.max(1, this.lineNumber + deltaLineNumber), Math.max(1, this.column + deltaColumn));
60 }
62 > /**
63 > * Test if this position equals other position
64 > */
65 > public equals(other: IPosition): boolean {
66 return Position.equals(this, other);
67 }
69 > /**
70 > * Test if position `a` equals position `b`
71 > */
72 > public static equals(a: IPosition | null, b: IPosition | null): boolean {
73 if (!a && !b) {
74 return true;
81 );
82 }
84 > /**
85 > * Test if this position is before other position.
86 > * If the two positions are equal, the result will be false.
87 > */
88 > public isBefore(other: IPosition): boolean {
89 return Position.isBefore(this, other);
90 }
92 > /**
93 > * Test if position `a` is before position `b`.
94 > * If the two positions are equal, the result will be false.
95 > */
96 > public static isBefore(a: IPosition, b: IPosition): boolean {
97 if (a.lineNumber < b.lineNumber) {
98 return true;
103 return a.column < b.column;
104 }
105 > position.ts
106 > /**
107 > * Test if this position is before other position.
108 > * If the two positions are equal, the result will be true.
109 > */
110 > public isBeforeOrEqual(other: IPosition): boolean {
111 return Position.isBeforeOrEqual(this, other);
112 }
113 > position.ts
114 > /**
115 > * Test if position `a` is before position `b`.
116 > * If the two positions are equal, the result will be true.
117 > */
118 > public static isBeforeOrEqual(a: IPosition, b: IPosition): boolean {
119 if (a.lineNumber < b.lineNumber) {
120 return true;
125 return a.column <= b.column;
126 }
127 > position.ts
128 > /**
129 > * A function that compares positions, useful for sorting
130 > */
131 > public static compare(a: IPosition, b: IPosition): number {
132 const aLineNumber = a.lineNumber | 0;
133 const bLineNumber = b.lineNumber | 0;
141 return aLineNumber - bLineNumber;
142 }
143 > position.ts
144 > /**
145 > * Clone this position.
146 > */
147 > public clone(): Position {
148 return new Position(this.lineNumber, this.column);
149 }
150 > position.ts
151 > /**
152 > * Convert to a human-readable representation.
153 > */
154 > public toString(): string {
155 return '(' + this.lineNumber + ',' + this.column + ')';
156 }
157 > position.ts
158 > // ---
159 >
160 > /**
161 > * Create a `Position` from an `IPosition`.
162 > */
163 > public static lift(pos: IPosition): Position {
164 return new Position(pos.lineNumber, pos.column);
165 }
166 > position.ts
167 > /**
168 > * Test if `obj` is an `IPosition`.
169 > */
170 > public static isIPosition(obj: unknown): obj is IPosition {
171 return (
172 !!obj
175 );
176 }
177 > position.ts
178 > public toJSON(): IPosition {
179 return {
180 lineNumber: this.lineNumber,
src/vs/platform/telemetry/common/telemetry.ts 115 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- telemetry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { createDecorator } from '../../instantiation/common/instantiation.js';
7 > import { ClassifiedEvent, IGDPRProperty, OmitMetadata, StrictPropertyCheck } from './gdprTypings.js';
8 >
9 > export const ITelemetryService = createDecorator<ITelemetryService>('telemetryService');
10 >
11 > export interface ITelemetryData {
12 > from?: string;
13 > target?: string;
14 > [key: string]: string | unknown | undefined;
15 > }
16 >
17 > export interface ITelemetryService {
18 >
19 > readonly _serviceBrand: undefined;
20 >
21 > readonly telemetryLevel: TelemetryLevel;
22 >
23 > readonly sessionId: string;
24 > readonly machineId: string;
25 > readonly sqmId: string;
26 > readonly devDeviceId: string;
27 > readonly firstSessionDate: string;
28 > readonly msftInternal?: boolean;
29 >
30 > /**
31 > * Whether error telemetry will get sent. If false, `publicLogError` will no-op.
32 > */
33 > readonly sendErrorTelemetry: boolean;
34 >
35 > /**
36 > * @deprecated Use publicLog2 and the typescript GDPR annotation where possible
37 > */
38 > publicLog(eventName: string, data?: ITelemetryData): void;
39 >
40 > /**
41 > * Sends a telemetry event that has been privacy approved.
42 > * Do not call this unless you have been given approval.
43 > */
44 > publicLog2<E extends ClassifiedEvent<OmitMetadata<T>> = never, T extends IGDPRProperty = never>(eventName: string, data?: StrictPropertyCheck<T, E>): void;
45 >
46 > /**
47 > * @deprecated Use publicLogError2 and the typescript GDPR annotation where possible
48 > */
49 > publicLogError(errorEventName: string, data?: ITelemetryData): void;
50 >
51 > publicLogError2<E extends ClassifiedEvent<OmitMetadata<T>> = never, T extends IGDPRProperty = never>(eventName: string, data?: StrictPropertyCheck<T, E>): void;
52 >
53 > setExperimentProperty(name: string, value: string): void;
54 >
55 > /**
56 > * Sets a common property that will be attached to all telemetry events.
57 > * Common properties are added after PII cleaning and cannot be overridden by event data.
58 > */
59 > setCommonProperty(name: string, value: string | boolean): void;
60 > }
61 >
62 > export function telemetryLevelEnabled(service: ITelemetryService, level: TelemetryLevel): boolean {
63 return service.telemetryLevel >= level;
64 }
66 > /**
67 > * Replaces `/` and `\` with `|` in model identifiers to prevent the
68 > * telemetry pipeline from redacting them as file paths.
69 > */
70 > export function escapeModelIdForTelemetry(modelId: string | undefined): string | undefined {
71 return modelId?.replace(/[\/\\]/g, '|');
72 }
74 > export interface ITelemetryEndpoint {
75 > id: string;
76 > aiKey: string;
77 > sendErrorTelemetry: boolean;
78 > }
79 >
80 > export const ICustomEndpointTelemetryService = createDecorator<ICustomEndpointTelemetryService>('customEndpointTelemetryService');
81 >
82 > export interface ICustomEndpointTelemetryService {
83 > readonly _serviceBrand: undefined;
84 >
85 > publicLog(endpoint: ITelemetryEndpoint, eventName: string, data?: ITelemetryData): void;
86 > publicLogError(endpoint: ITelemetryEndpoint, errorEventName: string, data?: ITelemetryData): void;
87 > }
88 >
89 > // Keys
90 > export const currentSessionDateStorageKey = 'telemetry.currentSessionDate';
91 > export const firstSessionDateStorageKey = 'telemetry.firstSessionDate';
92 > export const lastSessionDateStorageKey = 'telemetry.lastSessionDate';
93 > export const machineIdKey = 'telemetry.machineId';
94 > export const sqmIdKey = 'telemetry.sqmId';
95 > export const devDeviceIdKey = 'telemetry.devDeviceId';
96 >
97 > // Configuration Keys
98 > export const TELEMETRY_SECTION_ID = 'telemetry';
99 > export const TELEMETRY_SETTING_ID = 'telemetry.telemetryLevel';
100 > export const TELEMETRY_CRASH_REPORTER_SETTING_ID = 'telemetry.enableCrashReporter';
101 > export const TELEMETRY_OLD_SETTING_ID = 'telemetry.enableTelemetry';
102 >
103 > export const enum TelemetryLevel {
104 > NONE = 0,
105 > CRASH = 1,
106 > ERROR = 2,
107 > USAGE = 3
108 > }
109 >
110 > export const enum TelemetryConfiguration {
111 > OFF = 'off',
112 > CRASH = 'crash',
113 > ERROR = 'error',
114 > ON = 'all'
115 > }
116 >
117 > export interface ICommonProperties {
118 > [name: string]: string | boolean | undefined;
119 > }
src/vs/base/common/observableInternal/utils/utils.ts 114 covered LOC · 30 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- utils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { autorun } from '../reactions/autorun.js';
7 > import { IObservable, IObservableWithChange, IObserver, IReader, ITransaction } from '../base.js';
8 > import { observableValue } from '../observables/observableValue.js';
9 > import { DebugOwner } from '../debugName.js';
10 > import { DisposableStore, Event, IDisposable, toDisposable } from '../commonFacade/deps.js';
11 > import { derived, derivedOpts } from '../observables/derived.js';
12 > import { observableFromEvent } from '../observables/observableFromEvent.js';
13 > import { observableSignal } from '../observables/observableSignal.js';
14 > import { _setKeepObserved, _setRecomputeInitiallyAndOnChange } from '../observables/baseObservable.js';
15 > import { DebugLocation } from '../debugLocation.js';
16 >
17 > export function observableFromPromise<T>(promise: Promise<T>): IObservable<{ value?: T }> {
18 const observable = observableValue<{ value?: T }>('promiseValue', {});
19 promise.then((value) => {
22 return observable;
23 }
24 > utils.ts
25 > export function signalFromObservable<T>(owner: DebugOwner | undefined, observable: IObservable<T>): IObservable<void> {
26 return derivedOpts({
27 owner,
31 });
32 }
33 > utils.ts
34 > /**
35 > * Creates an observable that debounces the input observable.
36 > */
37 > export function debouncedObservable<T>(observable: IObservable<T>, debounceMs: number | ((lastValue: T | undefined, newValue: T) => number), debugLocation = DebugLocation.ofCaller()): IObservable<T> {
38 let hasValue = false;
39 let lastValue: T | undefined;
79 }, debugLocation);
80 }
81 > utils.ts
82 > /**
83 > * Creates an observable that throttles the input observable.
84 > * Unlike {@link debouncedObservable}, the timer starts on the first change
85 > * and is not reset by subsequent changes, preventing starvation.
86 > */
87 > export function throttledObservable<T>(observable: IObservable<T>, throttleMs: number, debugLocation = DebugLocation.ofCaller()): IObservable<T> {
88 let hasValue = false;
89 let lastValue: T | undefined;
126 }, debugLocation);
127 }
128 > utils.ts
129 > /**
130 > * Creates an observable that debounces the input observable.
131 > */
132 > export function debouncedObservable2<T>(observable: IObservable<T>, debounceMs: number | ((currentValue: T | undefined, newValue: T) => number), debugLocation = DebugLocation.ofCaller()): IObservable<T> {
133 const s = observableSignal('handleTimeout');
134
167 return d;
168 }
169 > utils.ts
170 > export function wasEventTriggeredRecently(event: Event<any>, timeoutMs: number, disposableStore: DisposableStore): IObservable<boolean> {
171 const observable = observableValue('triggeredRecently', false);
172
186 return observable;
187 }
188 > utils.ts
189 > /**
190 > * This makes sure the observable is being observed and keeps its cache alive.
191 > */
192 > export function keepObserved<T>(observable: IObservable<T>): IDisposable {
193 const o = new KeepAliveObserver(false, undefined);
194 observable.addObserver(o);
197 });
198 }
199 > utils.ts
200 > _setKeepObserved(keepObserved);
201 >
202 > /**
203 > * This converts the given observable into an autorun.
204 > */
205 > export function recomputeInitiallyAndOnChange<T>(observable: IObservable<T>, handleValue?: (value: T) => void): IDisposable {
206 > const o = new KeepAliveObserver(true, handleValue); utils.ts
207 > observable.addObserver(o);
208 > try {
209 > o.beginUpdate(observable);
210 > } finally {
211 > o.endUpdate(observable);
212 > }
213 >
214 > return toDisposable(() => {
215 > observable.removeObserver(o);
216 > });
217 > }
218 > utils.ts
219 > _setRecomputeInitiallyAndOnChange(recomputeInitiallyAndOnChange);
220 >
221 > export class KeepAliveObserver implements IObserver {
222 > private _counter = 0;
223 >
224 > constructor(
225 > private readonly _forceRecompute: boolean, utils.ts
226 > private readonly _handleValue: ((value: any) => void) | undefined,
227 > ) { }
228 > utils.ts
229 > beginUpdate<T>(observable: IObservable<T>): void {
230 > this._counter++; utils.ts
231 > }
232 > utils.ts
233 > endUpdate<T>(observable: IObservable<T>): void {
234 > if (this._counter === 1 && this._forceRecompute) { utils.ts
235 > if (this._handleValue) { utils.ts
236 this._handleValue(observable.get());
237 > } else { utils.ts
238 > observable.reportChanges(); utils.ts
239 > }
240 > } utils.ts
241 > this._counter--; utils.ts
242 > }
243 > utils.ts
244 > handlePossibleChange<T>(observable: IObservable<T>): void {
245 // NO OP
246 }
247 > utils.ts
248 > handleChange<T, TChange>(observable: IObservableWithChange<T, TChange>, change: TChange): void {
249 // NO OP
250 }
251 > } utils.ts
252 >
253 > export function derivedObservableWithCache<T>(owner: DebugOwner, computeFn: (reader: IReader, lastValue: T | undefined) => T): IObservable<T> {
254 let lastValue: T | undefined = undefined;
255 const observable = derivedOpts({ owner, debugReferenceFn: computeFn }, reader => {
259 return observable;
260 }
261 > utils.ts
262 > export function derivedObservableWithWritableCache<T>(owner: object, computeFn: (reader: IReader, lastValue: T | undefined) => T): IObservable<T>
263 & { clearCache(transaction: ITransaction): void; setCache(newValue: T | undefined, tx: ITransaction | undefined): void } {
264 let lastValue: T | undefined = undefined;
280 });
281 }
282 > utils.ts
283 > /**
284 > * When the items array changes, referential equal items are not mapped again.
285 > */
286 > export function mapObservableArrayCached<TIn, TOut, TKey = TIn>(owner: DebugOwner, items: IObservable<readonly TIn[]>, map: (input: TIn, store: DisposableStore) => TOut, keySelector?: (input: TIn) => TKey): IObservable<readonly TOut[]> {
287 let m = new ArrayMap(map, keySelector);
288 const self = derivedOpts({
300 return self;
301 }
302 > utils.ts
303 > class ArrayMap<TIn, TOut, TKey> implements IDisposable {
304 > private readonly _cache = new Map<TKey, { out: TOut; store: DisposableStore }>();
305 > private _items: TOut[] = [];
306 > constructor(
307 private readonly _map: (input: TIn, store: DisposableStore) => TOut,
308 private readonly _keySelector?: (input: TIn) => TKey,
309 ) {
310 }
311 > utils.ts
312 > public dispose(): void {
313 this._cache.forEach(entry => entry.store.dispose());
314 this._cache.clear();
315 }
316 > utils.ts
317 > public setItems(items: readonly TIn[]): void {
318 const newItems: TOut[] = [];
319 const itemsToRemove = new Set(this._cache.keys());
342 this._items = newItems;
343 }
344 > utils.ts
345 > public getItems(): TOut[] {
346 return this._items;
347 }
348 > } utils.ts
349 >
350 > export function isObservable<T>(obj: unknown): obj is IObservable<T> {
351 return !!obj && (<IObservable<T>>obj).read !== undefined && (<IObservable<T>>obj).reportChanges !== undefined;
352 }
src/vs/workbench/contrib/chat/test/common/tools/mockLanguageModelToolsService.ts 112 covered LOC · 37 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockLanguageModelToolsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../../base/common/cancellation.js';
7 > import { Codicon } from '../../../../../../base/common/codicons.js';
8 > import { Emitter, Event } from '../../../../../../base/common/event.js';
9 > import { Disposable, IDisposable } from '../../../../../../base/common/lifecycle.js';
10 > import { constObservable, IObservable, IReader } from '../../../../../../base/common/observable.js';
11 > import { ThemeIcon } from '../../../../../../base/common/themables.js';
12 > import { URI } from '../../../../../../base/common/uri.js';
13 > import { MockContextKeyService } from '../../../../../../platform/keybinding/test/common/mockKeybindingService.js';
14 > import { IProgressStep } from '../../../../../../platform/progress/common/progress.js';
15 > import { ChatRequestToolReferenceEntry } from '../../../common/attachments/chatVariableEntries.js';
16 > import { IVariableReference } from '../../../common/chatModes.js';
17 > import { IChatToolInvocation } from '../../../common/chatService/chatService.js';
18 > import { ILanguageModelChatMetadata } from '../../../common/languageModels.js';
19 > import { CountTokensCallback, IBeginToolCallOptions, ILanguageModelToolsService, ToolAndToolSetEnablementMap, IToolData, IToolImpl, IToolInvocation, IToolInvokedEvent, IToolResult, IToolSet, ToolDataSource, ToolSet } from '../../../common/tools/languageModelToolsService.js';
20 >
21 > export class MockLanguageModelToolsService extends Disposable implements ILanguageModelToolsService {
22 > _serviceBrand: undefined;
23 > vscodeToolSet: ToolSet = new ToolSet('vscode', 'vscode', ThemeIcon.fromId(Codicon.code.id), ToolDataSource.Internal, undefined, undefined, undefined, undefined, undefined, new MockContextKeyService());
24 > executeToolSet: ToolSet = new ToolSet('execute', 'execute', ThemeIcon.fromId(Codicon.terminal.id), ToolDataSource.Internal, undefined, undefined, undefined, undefined, undefined, new MockContextKeyService());
25 > readToolSet: ToolSet = new ToolSet('read', 'read', ThemeIcon.fromId(Codicon.book.id), ToolDataSource.Internal, undefined, undefined, undefined, undefined, undefined, new MockContextKeyService());
26 > agentToolSet: ToolSet = new ToolSet('agent', 'agent', ThemeIcon.fromId(Codicon.agent.id), ToolDataSource.Internal, undefined, undefined, undefined, undefined, undefined, new MockContextKeyService());
27 >
28 > private readonly _onDidInvokeTool = this._register(new Emitter<IToolInvokedEvent>());
29 >
30 > private readonly _registeredToolIds = new Set<string>();
31 > private readonly _registeredToolSetNames = new Set<string>();
32 > private readonly _toolSetTools = new Map<string, IToolData[]>();
33 >
34 > constructor() {
36 > }
37 >
38 > readonly onDidChangeTools: Event<void> = Event.None;
39 > readonly onDidPrepareToolCallBecomeUnresponsive: Event<{ sessionResource: URI; toolData: IToolData }> = Event.None;
40 > readonly onDidInvokeTool = this._onDidInvokeTool.event;
41 >
42 > fireOnDidInvokeTool(event: IToolInvokedEvent): void {
43 this._onDidInvokeTool.fire(event);
44 }
46 > registerToolData(toolData: IToolData): IDisposable {
47 return Disposable.None;
48 }
50 > resetToolAutoConfirmation(): void {
51
52 }
54 > getToolPostExecutionAutoConfirmation(toolId: string): 'workspace' | 'profile' | 'session' | 'never' {
55 return 'never';
56 }
58 > resetToolPostExecutionAutoConfirmation(): void {
59
60 }
62 > flushToolUpdates(): void {
63
64 }
66 > cancelToolCallsForRequest(requestId: string): void {
67
68 }
70 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
71 > setToolAutoConfirmation(toolId: string, scope: any): void {
72
73 }
75 > getToolAutoConfirmation(toolId: string): 'never' {
76 return 'never';
77 }
79 > registerToolImplementation(name: string, tool: IToolImpl): IDisposable {
80 return Disposable.None;
81 }
83 > registerTool(toolData: IToolData, tool: IToolImpl): IDisposable {
84 return Disposable.None;
85 }
87 > getTools(): Iterable<IToolData> {
88 return [];
89 }
91 > getAllToolsIncludingDisabled(): Iterable<IToolData> {
92 return [];
93 }
95 > addRegisteredToolId(id: string): void {
96 this._registeredToolIds.add(id);
97 }
99 > getTool(id: string): IToolData | undefined {
100 if (this._registeredToolIds.has(id)) {
101 return { id, source: ToolDataSource.Internal, displayName: id, modelDescription: id };
103 return undefined;
104 }
106 > observeTools(): IObservable<readonly IToolData[]> {
107 return constObservable([]);
108 }
110 > getToolByName(name: string): IToolData | undefined {
111 return undefined;
112 }
114 > acceptProgress(sessionId: string | undefined, callId: string, progress: IProgressStep): void {
115
116 }
118 > async invokeTool(dto: IToolInvocation, countTokens: CountTokensCallback, token: CancellationToken): Promise<IToolResult> {
119 return {
120 content: [{ kind: 'text', value: 'result' }]
121 };
122 }
124 > beginToolCall(_options: IBeginToolCallOptions): IChatToolInvocation | undefined {
125 // Mock implementation - return undefined
126 return undefined;
127 }
129 > async updateToolStream(_toolCallId: string, _partialInput: unknown, _token: CancellationToken): Promise<void> {
130 // Mock implementation - do nothing
131 }
133 > toolSets: IObservable<readonly IToolSet[]> = constObservable([]);
135 > getToolSetsForModel(model: ILanguageModelChatMetadata | undefined, reader?: IReader): Iterable<IToolSet> {
136 return [];
137 }
139 > addRegisteredToolSetName(name: string, tools?: IToolData[]): void {
140 this._registeredToolSetNames.add(name);
141 if (tools) {
143 }
144 }
146 > getToolSetByName(name: string): IToolSet | undefined {
147 if (this._registeredToolSetNames.has(name)) {
148 const tools = this._toolSetTools.get(name) ?? [];
151 return undefined;
152 }
154 > getToolSet(id: string): IToolSet | undefined {
155 return undefined;
156 }
158 > createToolSet(): ToolSet & IDisposable {
159 throw new Error('Method not implemented.');
160 }
162 > toToolAndToolSetEnablementMap(toolOrToolSetNames: readonly string[]): ToolAndToolSetEnablementMap {
163 throw new Error('Method not implemented.');
164 }
166 > toToolReferences(variableReferences: readonly IVariableReference[]): ChatRequestToolReferenceEntry[] {
167 throw new Error('Method not implemented.');
168 }
170 > getFullReferenceNames(): Iterable<string> {
171 throw new Error('Method not implemented.');
172 }
174 > getToolByFullReferenceName(qualifiedName: string): IToolData | IToolSet | undefined {
175 throw new Error('Method not implemented.');
176 }
178 > getFullReferenceName(tool: IToolData | IToolSet, set?: IToolSet): string {
179 throw new Error('Method not implemented.');
180 }
182 > getFullReferenceNameMap(): Map<IToolData | IToolSet, string> {
183 throw new Error('Method not implemented.');
184 }
186 > toFullReferenceNames(map: ToolAndToolSetEnablementMap): string[] {
187 throw new Error('Method not implemented.');
188 }
190 > getDeprecatedFullReferenceNames(): Map<string, Set<string>> {
191 throw new Error('Method not implemented.');
192 }
src/vs/base/common/arraysFind.ts 111 covered LOC · 22 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- arraysFind.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Comparator } from './arrays.js';
7 >
8 > export function findLast<T, R extends T>(array: readonly T[], predicate: (item: T, index: number) => item is R, fromIndex?: number): R | undefined;
9 > export function findLast<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex?: number): T | undefined;
10 > export function findLast<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex = array.length - 1): T | undefined {
11 > const idx = findLastIdx(array, predicate, fromIndex); arraysFind.ts
12 > if (idx === -1) {
13 > return undefined; arraysFind.ts
14 > }
15 > return array[idx]; arraysFind.ts
16 > }
18 > export function findLastIdx<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex = array.length - 1): number {
19 > for (let i = fromIndex; i >= 0; i--) { arraysFind.ts
20 > const element = array[i]; arraysFind.ts
21 >
22 > if (predicate(element, i)) {
23 > return i;
24 > }
25 > }
27 > return -1;
28 > }
30 > export function findFirst<T, R extends T>(array: readonly T[], predicate: (item: T, index: number) => item is R, fromIndex?: number): R | undefined;
31 > export function findFirst<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex?: number): T | undefined;
32 > export function findFirst<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex = 0): T | undefined {
33 const idx = findFirstIdx(array, predicate, fromIndex);
34 if (idx === -1) {
37 return array[idx];
38 }
40 > export function findFirstIdx<T>(array: readonly T[], predicate: (item: T, index: number) => unknown, fromIndex = 0): number {
41 for (let i = fromIndex; i < array.length; i++) {
42 const element = array[i];
49 return -1;
50 }
52 > /**
53 > * Finds the last item where predicate is true using binary search.
54 > * `predicate` must be monotonous, i.e. `arr.map(predicate)` must be like `[true, ..., true, false, ..., false]`!
55 > *
56 > * @returns `undefined` if no item matches, otherwise the last item that matches the predicate.
57 > */
58 > export function findLastMonotonous<T>(array: readonly T[], predicate: (item: T) => boolean): T | undefined {
59 const idx = findLastIdxMonotonous(array, predicate);
60 return idx === -1 ? undefined : array[idx];
61 }
63 > /**
64 > * Finds the last item where predicate is true using binary search.
65 > * `predicate` must be monotonous, i.e. `arr.map(predicate)` must be like `[true, ..., true, false, ..., false]`!
66 > *
67 > * @returns `startIdx - 1` if predicate is false for all items, otherwise the index of the last item that matches the predicate.
68 > */
69 > export function findLastIdxMonotonous<T>(array: readonly T[], predicate: (item: T) => boolean, startIdx = 0, endIdxEx = array.length): number {
70 let i = startIdx;
71 let j = endIdxEx;
80 return i - 1;
81 }
83 > /**
84 > * Finds the first item where predicate is true using binary search.
85 > * `predicate` must be monotonous, i.e. `arr.map(predicate)` must be like `[false, ..., false, true, ..., true]`!
86 > *
87 > * @returns `undefined` if no item matches, otherwise the first item that matches the predicate.
88 > */
89 > export function findFirstMonotonous<T>(array: readonly T[], predicate: (item: T) => boolean): T | undefined {
90 const idx = findFirstIdxMonotonousOrArrLen(array, predicate);
91 return idx === array.length ? undefined : array[idx];
92 }
94 > /**
95 > * Finds the first item where predicate is true using binary search.
96 > * `predicate` must be monotonous, i.e. `arr.map(predicate)` must be like `[false, ..., false, true, ..., true]`!
97 > *
98 > * @returns `endIdxEx` if predicate is false for all items, otherwise the index of the first item that matches the predicate.
99 > */
100 > export function findFirstIdxMonotonousOrArrLen<T>(array: readonly T[], predicate: (item: T) => boolean, startIdx = 0, endIdxEx = array.length): number {
101 let i = startIdx;
102 let j = endIdxEx;
111 return i;
112 }
114 > export function findFirstIdxMonotonous<T>(array: readonly T[], predicate: (item: T) => boolean, startIdx = 0, endIdxEx = array.length): number {
115 const idx = findFirstIdxMonotonousOrArrLen(array, predicate, startIdx, endIdxEx);
116 return idx === array.length ? -1 : idx;
117 }
119 > /**
120 > * Use this when
121 > * * You have a sorted array
122 > * * You query this array with a monotonous predicate to find the last item that has a certain property.
123 > * * You query this array multiple times with monotonous predicates that get weaker and weaker.
124 > */
125 > export class MonotonousArray<T> {
126 > public static assertInvariants = false;
127 >
128 > private _findLastMonotonousLastIdx = 0;
129 > private _prevFindLastPredicate: ((item: T) => boolean) | undefined;
130 >
131 > constructor(private readonly _array: readonly T[]) {
132 }
134 > /**
135 > * The predicate must be monotonous, i.e. `arr.map(predicate)` must be like `[true, ..., true, false, ..., false]`!
136 > * For subsequent calls, current predicate must be weaker than (or equal to) the previous predicate, i.e. more entries must be `true`.
137 > */
138 > findLastMonotonous(predicate: (item: T) => boolean): T | undefined {
139 if (MonotonousArray.assertInvariants) {
140 if (this._prevFindLastPredicate) {
152 return idx === -1 ? undefined : this._array[idx];
153 }
154 > } arraysFind.ts
155 >
156 > /**
157 > * Returns the first item that is equal to or greater than every other item.
158 > */
159 > export function findFirstMax<T>(array: readonly T[], comparator: Comparator<T>): T | undefined {
160 if (array.length === 0) {
161 return undefined;
171 return max;
172 }
174 > /**
175 > * Returns the last item that is equal to or greater than every other item.
176 > */
177 > export function findLastMax<T>(array: readonly T[], comparator: Comparator<T>): T | undefined {
178 if (array.length === 0) {
179 return undefined;
189 return max;
190 }
192 > /**
193 > * Returns the first item that is equal to or less than every other item.
194 > */
195 > export function findFirstMin<T>(array: readonly T[], comparator: Comparator<T>): T | undefined {
196 return findFirstMax(array, (a, b) => -comparator(a, b));
197 }
199 > export function findMaxIdx<T>(array: readonly T[], comparator: Comparator<T>): number {
200 if (array.length === 0) {
201 return -1;
211 return maxIdx;
212 }
214 > /**
215 > * Returns the first mapped value of the array which is not undefined.
216 > */
217 > export function mapFindFirst<T, R>(items: Iterable<T>, mapFn: (value: T) => R | undefined): R | undefined {
218 for (const value of items) {
219 const mapped = mapFn(value);
src/vs/workbench/services/path/common/pathService.ts 111 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- pathService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isValidBasename } from '../../../../base/common/extpath.js';
7 > import { Schemas } from '../../../../base/common/network.js';
8 > import { IPath, win32, posix } from '../../../../base/common/path.js';
9 > import { OperatingSystem, OS } from '../../../../base/common/platform.js';
10 > import { basename } from '../../../../base/common/resources.js';
11 > import { URI } from '../../../../base/common/uri.js';
12 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
13 > import { getVirtualWorkspaceScheme } from '../../../../platform/workspace/common/virtualWorkspace.js';
14 > import { IWorkspaceContextService } from '../../../../platform/workspace/common/workspace.js';
15 > import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.js';
16 > import { IRemoteAgentService } from '../../remote/common/remoteAgentService.js';
17 >
18 > export const IPathService = createDecorator<IPathService>('pathService');
19 >
20 > /**
21 > * Provides access to path related properties that will match the
22 > * environment. If the environment is connected to a remote, the
23 > * path properties will match that of the remotes operating system.
24 > */
25 > export interface IPathService {
26 >
27 > readonly _serviceBrand: undefined;
28 >
29 > /**
30 > * The correct path library to use for the target environment. If
31 > * the environment is connected to a remote, this will be the
32 > * path library of the remote file system. Otherwise it will be
33 > * the local file system's path library depending on the OS.
34 > */
35 > readonly path: Promise<IPath>;
36 >
37 > /**
38 > * Determines the best default URI scheme for the current workspace.
39 > * It uses information about whether we're running remote, in browser,
40 > * or native combined with information about the current workspace to
41 > * find the best default scheme.
42 > */
43 > readonly defaultUriScheme: string;
44 >
45 > /**
46 > * Converts the given path to a file URI to use for the target
47 > * environment. If the environment is connected to a remote, it
48 > * will use the path separators according to the remote file
49 > * system. Otherwise it will use the local file system's path
50 > * separators.
51 > */
52 > fileURI(path: string): Promise<URI>;
53 >
54 > /**
55 > * Resolves the user-home directory for the target environment.
56 > * If the envrionment is connected to a remote, this will be the
57 > * remote's user home directory, otherwise the local one unless
58 > * `preferLocal` is set to `true`.
59 > */
60 > userHome(options: { preferLocal: true }): URI;
61 > userHome(options?: { preferLocal: boolean }): Promise<URI>;
62 >
63 > /**
64 > * Figures out if the provided resource has a valid file name
65 > * for the operating system the file is saved to.
66 > *
67 > * Note: this currently only supports `file` and `vscode-file`
68 > * protocols where we know the limits of the file systems behind
69 > * these OS. Other remotes are not supported and this method
70 > * will always return `true` for them.
71 > */
72 > hasValidBasename(resource: URI, basename?: string): Promise<boolean>;
73 > hasValidBasename(resource: URI, os: OperatingSystem, basename?: string): boolean;
74 >
75 > /**
76 > * @deprecated use `userHome` instead.
77 > */
78 > readonly resolvedUserHome: URI | undefined;
79 > }
80 >
81 > export abstract class AbstractPathService implements IPathService {
82 >
83 > declare readonly _serviceBrand: undefined;
84 >
85 > private resolveOS: Promise<OperatingSystem>;
86 >
87 > private resolveUserHome: Promise<URI>;
88 > private maybeUnresolvedUserHome: URI | undefined;
89 >
90 > constructor(
91 private localUserHome: URI,
92 @IRemoteAgentService private readonly remoteAgentService: IRemoteAgentService,
110 })();
111 }
113 > hasValidBasename(resource: URI, basename?: string): Promise<boolean>;
114 > hasValidBasename(resource: URI, os: OperatingSystem, basename?: string): boolean;
115 > hasValidBasename(resource: URI, arg2?: string | OperatingSystem, basename?: string): boolean | Promise<boolean> {
116
117 // async version
123 return this.doHasValidBasename(resource, arg2, basename);
124 }
126 > private doHasValidBasename(resource: URI, os: OperatingSystem, name?: string): boolean {
127
128 // Our `isValidBasename` method only works with our
135 return true;
136 }
138 > get defaultUriScheme(): string {
139 return AbstractPathService.findDefaultUriScheme(this.environmentService, this.contextService);
140 }
142 > static findDefaultUriScheme(environmentService: IWorkbenchEnvironmentService, contextService: IWorkspaceContextService): string {
143 if (environmentService.remoteAuthority) {
144 return Schemas.vscodeRemote;
162 return Schemas.file;
163 }
165 > userHome(options?: { preferLocal: boolean }): Promise<URI>;
166 > userHome(options: { preferLocal: true }): URI;
167 > userHome(options?: { preferLocal: boolean }): Promise<URI> | URI {
168 return options?.preferLocal ? this.localUserHome : this.resolveUserHome;
169 }
171 > get resolvedUserHome(): URI | undefined {
172 return this.maybeUnresolvedUserHome;
173 }
175 > get path(): Promise<IPath> {
176 return this.resolveOS.then(os => {
177 return os === OperatingSystem.Windows ?
180 });
181 }
183 > async fileURI(_path: string): Promise<URI> {
184 let authority = '';
185
src/vs/base/common/equals.ts 110 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- equals.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as arrays from './arrays.js';
7 >
8 > /*
9 > * Each function in this file which offers an equality comparison, has an accompanying
10 > * `*C` variant which returns an EqualityComparer function.
11 > *
12 > * The `*C` variant allows for easier composition of equality comparers and improved type-inference.
13 > */
14 >
15 >
16 > /** Represents a function that decides if two values are equal. */
17 > export type EqualityComparer<T> = (a: T, b: T) => boolean;
18 >
19 > export interface IEquatable<T> {
20 > equals(other: T): boolean;
21 > }
22 >
23 > /**
24 > * Compares two items for equality using strict equality.
25 > */
26 > export function strictEquals<T>(a: T, b: T): boolean {
27 > return a === b; equals.ts
28 > }
29 > equals.ts
30 > export function strictEqualsC<T>(): EqualityComparer<T> {
31 return (a, b) => a === b;
32 }
33 > equals.ts
34 > /**
35 > * Checks if the items of two arrays are equal.
36 > * By default, strict equality is used to compare elements, but a custom equality comparer can be provided.
37 > */
38 > export function arrayEquals<T>(a: readonly T[], b: readonly T[], itemEquals?: EqualityComparer<T>): boolean {
39 return arrays.equals(a, b, itemEquals ?? strictEquals);
40 }
41 > equals.ts
42 > /**
43 > * Checks if the items of two arrays are equal.
44 > * By default, strict equality is used to compare elements, but a custom equality comparer can be provided.
45 > */
46 > export function arrayEqualsC<T>(itemEquals?: EqualityComparer<T>): EqualityComparer<readonly T[]> {
47 return (a, b) => arrays.equals(a, b, itemEquals ?? strictEquals);
48 }
49 > equals.ts
50 > /**
51 > * Drills into arrays (items ordered) and objects (keys unordered) and uses strict equality on everything else.
52 > */
53 > export function structuralEquals<T>(a: T, b: T): boolean {
54 if (a === b) {
55 return true;
95 return false;
96 }
97 > equals.ts
98 > export function structuralEqualsC<T>(): EqualityComparer<T> {
99 return (a, b) => structuralEquals(a, b);
100 }
101 > equals.ts
102 > /**
103 > * `getStructuralKey(a) === getStructuralKey(b) <=> structuralEquals(a, b)`
104 > * (assuming that a and b are not cyclic structures and nothing extends globalThis Array).
105 > */
106 > export function getStructuralKey(t: unknown): string {
107 return JSON.stringify(toNormalizedJsonStructure(t));
108 }
109 > equals.ts
110 > let objectId = 0;
111 > const objIds = new WeakMap<object, number>();
112 >
113 function toNormalizedJsonStructure(t: unknown): unknown {
114 if (Array.isArray(t)) {
136 return t;
137 }
138 > equals.ts
139 >
140 > /**
141 > * Two items are considered equal, if their stringified representations are equal.
142 > */
143 > export function jsonStringifyEquals<T>(a: T, b: T): boolean {
144 return JSON.stringify(a) === JSON.stringify(b);
145 }
146 > equals.ts
147 > /**
148 > * Two items are considered equal, if their stringified representations are equal.
149 > */
150 > export function jsonStringifyEqualsC<T>(): EqualityComparer<T> {
151 return (a, b) => JSON.stringify(a) === JSON.stringify(b);
152 }
153 > equals.ts
154 > /**
155 > * Uses `item.equals(other)` to determine equality.
156 > */
157 > export function thisEqualsC<T extends IEquatable<T>>(): EqualityComparer<T> {
158 return (a, b) => a.equals(b);
159 }
160 > equals.ts
161 > /**
162 > * Checks if two items are both null or undefined, or are equal according to the provided equality comparer.
163 > */
164 > export function equalsIfDefined<T>(v1: T | undefined | null, v2: T | undefined | null, equals: EqualityComparer<T>): boolean {
165 if (v1 === undefined || v1 === null || v2 === undefined || v2 === null) {
166 return v2 === v1;
168 return equals(v1, v2);
169 }
170 > equals.ts
171 > /**
172 > * Returns an equality comparer that checks if two items are both null or undefined, or are equal according to the provided equality comparer.
173 > */
174 > export function equalsIfDefinedC<T>(equals: EqualityComparer<T>): EqualityComparer<T | undefined | null> {
175 return (v1, v2) => {
176 if (v1 === undefined || v1 === null || v2 === undefined || v2 === null) {
180 };
181 }
182 > equals.ts
183 > /**
184 > * Each function in this file which offers an equality comparison, has an accompanying
185 > * `*C` variant which returns an EqualityComparer function.
186 > *
187 > * The `*C` variant allows for easier composition of equality comparers and improved type-inference.
188 > */
189 > export namespace equals {
190 > export const strict = strictEquals;
191 > export const strictC = strictEqualsC;
192 >
193 > export const array = arrayEquals;
194 > export const arrayC = arrayEqualsC;
195 >
196 > export const structural = structuralEquals;
197 > export const structuralC = structuralEqualsC;
198 >
199 > export const jsonStringify = jsonStringifyEquals;
200 > export const jsonStringifyC = jsonStringifyEqualsC;
201 >
202 > export const thisC = thisEqualsC;
203 >
204 > export const ifDefined = equalsIfDefined;
205 > export const ifDefinedC = equalsIfDefinedC;
206 > }
src/vs/workbench/contrib/chat/common/participants/chatSlashCommands.ts 109 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatSlashCommands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
7 > import { Emitter, Event } from '../../../../../base/common/event.js';
8 > import { Disposable, IDisposable, toDisposable } from '../../../../../base/common/lifecycle.js';
9 > import { ContextKeyExpression } from '../../../../../platform/contextkey/common/contextkey.js';
10 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
11 > import { IProgress } from '../../../../../platform/progress/common/progress.js';
12 > import { IChatMessage } from '../languageModels.js';
13 > import { IChatFollowup, IChatProgress, IChatResponseProgressFileTreeData, IChatSendRequestOptions } from '../chatService/chatService.js';
14 > import { IExtensionService } from '../../../../services/extensions/common/extensions.js';
15 > import { ChatAgentLocation, ChatModeKind } from '../constants.js';
16 > import { URI } from '../../../../../base/common/uri.js';
17 > import { getChatSessionType } from '../model/chatUri.js';
18 > import { matchesSessionType } from '../promptSyntax/service/promptsService.js';
19 >
20 > //#region slash service, commands etc
21 >
22 > export interface IChatSlashData {
23 > command: string;
24 > detail: string;
25 > sortText?: string;
26 > /**
27 > * Whether the command should execute as soon
28 > * as it is entered. Defaults to `false`.
29 > */
30 > executeImmediately?: boolean;
31 >
32 > /**
33 > * Whether a silent command can execute independently while the chat has a request in progress.
34 > */
35 > executeDuringRequest?: boolean;
36 >
37 > /**
38 > * Whether the command should be added as a request/response
39 > * turn to the chat history. Defaults to `false`.
40 > *
41 > * For instance, the `/save` command opens an untitled document
42 > * to the side hence does not contain any chatbot responses.
43 > */
44 > silent?: boolean;
45 >
46 > locations: ChatAgentLocation[];
47 > modes?: ChatModeKind[];
48 > sessionTypes?: string[];
49 >
50 > /**
51 > * Optional context key expression that controls visibility of this command.
52 > * When set, the command is only shown if the expression evaluates to true.
53 > */
54 > when?: ContextKeyExpression;
55 > }
56 >
57 > export interface IChatSlashFragment {
58 > content: string | { treeData: IChatResponseProgressFileTreeData };
59 > }
60 > export type IChatSlashCallback = { (prompt: string, progress: IProgress<IChatProgress>, history: IChatMessage[], location: ChatAgentLocation, sessionResource: URI, token: CancellationToken, options?: IChatSendRequestOptions): Promise<{ followUp: IChatFollowup[] } | void> };
61 >
62 > export const IChatSlashCommandService = createDecorator<IChatSlashCommandService>('chatSlashCommandService');
63 >
64 > /**
65 > * This currently only exists to drive /clear and /help
66 > */
67 > export interface IChatSlashCommandService {
68 > _serviceBrand: undefined;
69 > readonly onDidChangeCommands: Event<void>;
70 > registerSlashCommand(data: IChatSlashData, command: IChatSlashCallback): IDisposable;
71 > executeCommand(id: string, prompt: string, progress: IProgress<IChatProgress>, history: IChatMessage[], location: ChatAgentLocation, sessionResource: URI, token: CancellationToken, options?: IChatSendRequestOptions): Promise<{ followUp: IChatFollowup[] } | void>;
72 > getCommands(location: ChatAgentLocation, mode: ChatModeKind): Array<IChatSlashData>;
73 > hasCommand(id: string, sessionType: string): boolean;
74 > }
75 >
76 > type RegisteredSlashCommand = { data: IChatSlashData; command?: IChatSlashCallback };
77 >
78 > export class ChatSlashCommandService extends Disposable implements IChatSlashCommandService {
79 >
80 > declare _serviceBrand: undefined;
81 >
82 > private readonly _commands = new Map<string, RegisteredSlashCommand[]>();
83 >
84 > private readonly _onDidChangeCommands = this._register(new Emitter<void>());
85 > readonly onDidChangeCommands: Event<void> = this._onDidChangeCommands.event;
86 >
87 > constructor(@IExtensionService private readonly _extensionService: IExtensionService) {
88 > super(); chatSlashCommands.ts
89 > }
91 > override dispose(): void {
92 > super.dispose(); chatSlashCommands.ts
93 > this._commands.clear();
94 > }
96 > private getSessionScopedCommands(id: string): RegisteredSlashCommand[] {
97 return this._commands.get(id) ?? [];
98 }
100 > private commandsOverlap(dataA: IChatSlashData, dataB: IChatSlashData): boolean {
101 if (dataA.sessionTypes === undefined || dataB.sessionTypes === undefined) {
102 return true;
105 return dataA.sessionTypes.some(sessionType => dataB.sessionTypes?.includes(sessionType));
106 }
108 > private getCommand(id: string, sessionType: string | undefined): RegisteredSlashCommand | undefined {
109 return this.getSessionScopedCommands(id).find(candidate => matchesSessionType(candidate.data.sessionTypes, sessionType));
110 }
112 > registerSlashCommand(data: IChatSlashData, command: IChatSlashCallback): IDisposable {
113 const commandsForId = this.getSessionScopedCommands(data.command);
114 if (commandsForId.some(candidate => this.commandsOverlap(candidate.data, data))) {
140 });
141 }
143 > getCommands(location: ChatAgentLocation, mode: ChatModeKind): Array<IChatSlashData> {
144 return Array
145 .from(this._commands.values())
147 .filter(c => c.locations.includes(location) && (!c.modes || c.modes.includes(mode)));
148 }
150 > hasCommand(id: string, sessionType: string): boolean {
151 return !!this.getCommand(id, sessionType);
152 }
154 > async executeCommand(id: string, prompt: string, progress: IProgress<IChatProgress>, history: IChatMessage[], location: ChatAgentLocation, sessionResource: URI, token: CancellationToken, options?: IChatSendRequestOptions): Promise<{ followUp: IChatFollowup[] } | void> {
155 const data = this.getCommand(id, getChatSessionType(sessionResource));
156 if (!data) {
166 return await data.command(prompt, progress, history, location, sessionResource, token, options);
167 }
src/vs/platform/workspace/common/workspaceTrust.ts 108 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workspaceTrust.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { IDisposable } from '../../../base/common/lifecycle.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { createDecorator } from '../../instantiation/common/instantiation.js';
10 >
11 > export enum WorkspaceTrustScope {
12 > Local = 0,
13 > Remote = 1
14 > }
15 >
16 > export interface WorkspaceTrustRequestButton {
17 > readonly label: string;
18 > readonly type: 'ContinueWithTrust' | 'ContinueWithoutTrust' | 'Manage' | 'Cancel';
19 > }
20 >
21 > export interface ResourceTrustRequestOptions {
22 > readonly uri: URI;
23 > readonly message?: string;
24 > }
25 >
26 > export interface WorkspaceTrustRequestOptions {
27 > readonly buttons?: WorkspaceTrustRequestButton[];
28 > readonly message?: string;
29 > }
30 >
31 > export const IWorkspaceTrustEnablementService = createDecorator<IWorkspaceTrustEnablementService>('workspaceTrustEnablementService');
32 >
33 > export interface IWorkspaceTrustEnablementService {
34 > readonly _serviceBrand: undefined;
35 >
36 > isWorkspaceTrustEnabled(): boolean;
37 > }
38 >
39 > export const IWorkspaceTrustManagementService = createDecorator<IWorkspaceTrustManagementService>('workspaceTrustManagementService');
40 >
41 > export interface IWorkspaceTrustManagementService {
42 > readonly _serviceBrand: undefined;
43 >
44 > readonly onDidChangeTrust: Event<boolean>;
45 > readonly onDidChangeTrustedFolders: Event<void>;
46 >
47 > readonly workspaceResolved: Promise<void>;
48 > readonly workspaceTrustInitialized: Promise<void>;
49 > acceptsOutOfWorkspaceFiles: boolean;
50 >
51 > isWorkspaceTrusted(): boolean;
52 > isWorkspaceTrustForced(): boolean;
53 >
54 > canSetParentFolderTrust(): boolean;
55 > setParentFolderTrust(trusted: boolean): Promise<void>;
56 >
57 > canSetWorkspaceTrust(): boolean;
58 > setWorkspaceTrust(trusted: boolean): Promise<void>;
59 >
60 > getUriTrustInfo(uri: URI): Promise<IWorkspaceTrustUriInfo>;
61 > setUrisTrust(uri: URI[], trusted: boolean): Promise<void>;
62 >
63 > getTrustedUris(): URI[];
64 > setTrustedUris(uris: URI[]): Promise<void>;
65 >
66 > addWorkspaceTrustTransitionParticipant(participant: IWorkspaceTrustTransitionParticipant): IDisposable;
67 > }
68 >
69 > export const enum WorkspaceTrustUriResponse {
70 > Open = 1,
71 > OpenInNewWindow = 2,
72 > Cancel = 3
73 > }
74 >
75 > export const IWorkspaceTrustRequestService = createDecorator<IWorkspaceTrustRequestService>('workspaceTrustRequestService');
76 >
77 > export interface IWorkspaceTrustRequestService {
78 > readonly _serviceBrand: undefined;
79 >
80 > readonly onDidInitiateOpenFilesTrustRequest: Event<void>;
81 > readonly onDidInitiateWorkspaceTrustRequest: Event<WorkspaceTrustRequestOptions | undefined>;
82 > readonly onDidInitiateWorkspaceTrustRequestOnStartup: Event<void>;
83 > readonly onDidInitiateResourcesTrustRequest: Event<ResourceTrustRequestOptions>;
84 >
85 > completeOpenFilesTrustRequest(result: WorkspaceTrustUriResponse, saveResponse?: boolean): Promise<void>;
86 > requestOpenFilesTrust(openFiles: URI[]): Promise<WorkspaceTrustUriResponse>;
87 >
88 > completeResourcesTrustRequest(uri: URI, result: WorkspaceTrustUriResponse): Promise<void>;
89 > requestResourcesTrust(options: ResourceTrustRequestOptions): Promise<boolean | undefined>;
90 >
91 > cancelWorkspaceTrustRequest(): void;
92 > completeWorkspaceTrustRequest(trusted?: boolean): Promise<void>;
93 > requestWorkspaceTrust(options?: WorkspaceTrustRequestOptions): Promise<boolean | undefined>;
94 > requestWorkspaceTrustOnStartup(): void;
95 > }
96 >
97 > export interface IWorkspaceTrustTransitionParticipant {
98 > participate(trusted: boolean): Promise<void>;
99 > }
100 >
101 > export interface IWorkspaceTrustUriInfo {
102 > uri: URI;
103 > trusted: boolean;
104 > }
105 >
106 > export interface IWorkspaceTrustInfo {
107 > uriTrustInfo: IWorkspaceTrustUriInfo[];
108 > }
src/vs/base/common/htmlContent.ts 106 covered LOC · 23 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- htmlContent.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { illegalArgument } from './errors.js';
7 > import { escapeIcons } from './iconLabels.js';
8 > import { Schemas } from './network.js';
9 > import { isEqual } from './resources.js';
10 > import { escapeRegExpCharacters } from './strings.js';
11 > import { URI, UriComponents } from './uri.js';
12 >
13 > export interface MarkdownStringTrustedOptions {
14 > readonly enabledCommands: readonly string[];
15 > }
16 >
17 > export interface IMarkdownString {
18 > readonly value: string;
19 > readonly isTrusted?: boolean | MarkdownStringTrustedOptions;
20 > readonly supportThemeIcons?: boolean;
21 > readonly supportHtml?: boolean;
22 > /** @internal */
23 > readonly supportAlertSyntax?: boolean;
24 > readonly baseUri?: UriComponents;
25 > uris?: { [href: string]: UriComponents };
26 > }
27 >
28 > export const enum MarkdownStringTextNewlineStyle {
29 > Paragraph = 0,
30 > Break = 1,
31 > }
32 >
33 > export class MarkdownString implements IMarkdownString {
34 >
35 > public value: string;
36 > public isTrusted?: boolean | MarkdownStringTrustedOptions;
37 > public supportThemeIcons?: boolean;
38 > public supportHtml?: boolean;
39 > public supportAlertSyntax?: boolean;
40 > public baseUri?: URI;
41 > public uris?: { [href: string]: UriComponents } | undefined;
42 >
43 > public static lift(dto: IMarkdownString): MarkdownString {
44 const markdownString = new MarkdownString(dto.value, dto);
45 markdownString.uris = dto.uris;
47 return markdownString;
48 }
50 > constructor(
51 value: string = '',
52 isTrustedOrOptions: boolean | { isTrusted?: boolean | MarkdownStringTrustedOptions; supportThemeIcons?: boolean; supportHtml?: boolean; supportAlertSyntax?: boolean } = false,
70 }
71 }
73 > appendText(value: string, newlineStyle: MarkdownStringTextNewlineStyle = MarkdownStringTextNewlineStyle.Paragraph): MarkdownString {
74 this.value += escapeMarkdownSyntaxTokens(this.supportThemeIcons ? escapeIcons(value) : value) // CodeQL [SM02383] The Markdown is fully sanitized after being rendered.
75 .replace(/([ \t]+)/g, (_match, g1) => '&nbsp;'.repeat(g1.length)) // CodeQL [SM02383] The Markdown is fully sanitized after being rendered.
79 return this;
80 }
82 > appendMarkdown(value: string): MarkdownString {
83 this.value += value;
84 return this;
85 }
87 > appendCodeblock(langId: string, code: string): MarkdownString {
88 this.value += `\n${appendEscapedMarkdownCodeBlockFence(code, langId)}\n`;
89 return this;
90 }
92 > appendLink(target: URI | string, label: string, title?: string): MarkdownString {
93 this.value += '[';
94 this.value += this._escape(label, ']');
101 return this;
102 }
104 > private _escape(value: string, ch: string): string {
105 const r = new RegExp(escapeRegExpCharacters(ch), 'g');
106 return value.replace(r, (match, offset) => {
112 });
113 }
114 > } htmlContent.ts
115 >
116 > export function isEmptyMarkdownString(oneOrMany: IMarkdownString | IMarkdownString[] | null | undefined): boolean {
117 if (isMarkdownString(oneOrMany)) {
118 return !oneOrMany.value;
123 }
124 }
126 > export function isMarkdownString(thing: unknown): thing is IMarkdownString {
127 > if (thing instanceof MarkdownString) { htmlContent.ts
128 return true;
129 > } else if (thing && typeof thing === 'object') { htmlContent.ts
130 > return typeof (<IMarkdownString>thing).value === 'string'
131 && (typeof (<IMarkdownString>thing).isTrusted === 'boolean' || typeof (<IMarkdownString>thing).isTrusted === 'object' || (<IMarkdownString>thing).isTrusted === undefined)
132 && (typeof (<IMarkdownString>thing).supportThemeIcons === 'boolean' || (<IMarkdownString>thing).supportThemeIcons === undefined)
133 && (typeof (<IMarkdownString>thing).supportAlertSyntax === 'boolean' || (<IMarkdownString>thing).supportAlertSyntax === undefined);
134 > } htmlContent.ts
135 return false;
136 }
138 > export function markdownStringEqual(a: IMarkdownString, b: IMarkdownString): boolean {
139 if (a === b) {
140 return true;
150 }
151 }
153 > export function escapeMarkdownSyntaxTokens(text: string): string {
154 // escape markdown syntax tokens: http://daringfireball.net/projects/markdown/syntax#backslash
155 return text
157 .replace(/^([ \t]*)-/gm, '$1\\-'); // CodeQL [SM02383] Backslash is escaped in the character class
158 }
160 > /**
161 > * Escapes only the characters that would break out of markdown link text
162 > * (`[label](url)`) syntax: `\` and `]`. Use this when the escaped string is
163 > * displayed as the visible label of a link, since renderers that extract the
164 > * link text without re-parsing markdown (e.g. the chat inline anchor / skill
165 > * pill) would otherwise show full `escapeMarkdownSyntaxTokens` backslashes
166 > * (`\-`, `\.`, ...) verbatim.
167 > */
168 > export function escapeMarkdownLinkLabel(text: string): string {
169 return text.replace(/[\\\]]/g, '\\$&');
170 }
172 > /**
173 > * @see https://github.com/microsoft/vscode/issues/193746
174 > */
175 > export function appendEscapedMarkdownCodeBlockFence(code: string, langId: string) {
176 const longestFenceLength =
177 code.match(/^`+/gm)?.reduce((a, b) => (a.length > b.length ? a : b)).length ??
187 ].join('\n');
188 }
190 > /**
191 > * Wraps arbitrary text in a markdown inline code span using a backtick fence
192 > * long enough to safely contain any backtick sequences present in the text.
193 > *
194 > * Backticks inside an inline code span cannot be backslash-escaped per the
195 > * CommonMark spec — the only safe way is to choose a delimiter run longer
196 > * than any run of backticks in the content (and pad with spaces if the
197 > * content begins or ends with a backtick).
198 > */
199 > export function appendEscapedMarkdownInlineCode(text: string): string {
200 const longestBacktickRun = Math.max(0, ...(text.match(/`+/g) ?? []).map(m => m.length));
201 const fence = '`'.repeat(longestBacktickRun + 1);
204 return `${fence}${content}${fence}`;
205 }
207 > export function escapeDoubleQuotes(input: string) {
208 return input.replace(/"/g, '&quot;');
209 }
211 > export function removeMarkdownEscapes(text: string): string {
212 if (!text) {
213 return text;
215 return text.replace(/\\([\\`*_{}[\]()#+\-.!~])/g, '$1');
216 }
218 > export function parseHrefAndDimensions(href: string): { href: string; dimensions: string[] } {
219 const dimensions: string[] = [];
220 const splitted = href.split('|').map(s => s.trim());
237 return { href, dimensions };
238 }
240 > export function createMarkdownLink(text: string, href: string, title?: string, escapeTokens = true): string {
241 return `[${escapeTokens ? escapeMarkdownSyntaxTokens(text) : text}](${href}${title ? ` "${escapeMarkdownSyntaxTokens(title)}"` : ''})`;
242 }
244 > export function createMarkdownCommandLink(command: { text: string; id: string; arguments?: unknown[]; tooltip: string }, escapeTokens = true): string {
245 const uri = createCommandUri(command.id, ...(command.arguments || [])).toString();
246 return createMarkdownLink(command.text, uri, command.tooltip, escapeTokens);
247 }
249 > export function createCommandUri(commandId: string, ...commandArgs: unknown[]): URI {
250 return URI.from({
251 scheme: Schemas.command,
src/vs/base/common/observableInternal/debugName.ts 106 covered LOC · 26 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- debugName.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export interface IDebugNameData {
7 > /**
8 > * The owner object of an observable.
9 > * Used for debugging only, such as computing a name for the observable by iterating over the fields of the owner.
10 > */
11 > readonly owner?: DebugOwner | undefined;
12 >
13 > /**
14 > * A string or function that returns a string that represents the name of the observable.
15 > * Used for debugging only.
16 > */
17 > readonly debugName?: DebugNameSource | undefined;
18 >
19 > /**
20 > * A function that points to the defining function of the object.
21 > * Used for debugging only.
22 > */
23 > readonly debugReferenceFn?: Function | undefined;
24 > }
25 >
26 > export class DebugNameData {
27 > constructor(
28 > public readonly owner: DebugOwner | undefined, debugName.ts
29 > public readonly debugNameSource: DebugNameSource | undefined,
30 > public readonly referenceFn: Function | undefined,
31 > ) { }
33 > public getDebugName(target: object): string | undefined {
34 > return getDebugName(target, this); debugName.ts
35 > }
36 > } debugName.ts
37 >
38 > /**
39 > * The owning object of an observable.
40 > * Is only used for debugging purposes, such as computing a name for the observable by iterating over the fields of the owner.
41 > */
42 > export type DebugOwner = object | undefined;
43 > export type DebugNameSource = string | (() => string | undefined);
44 >
45 > const countPerName = new Map<string, number>();
46 > const cachedDebugName = new WeakMap<object, string>();
47 >
48 > export function getDebugName(target: object, data: DebugNameData): string | undefined {
49 > const cached = cachedDebugName.get(target); debugName.ts
50 > if (cached) {
51 return cached;
52 }
54 > const dbgName = computeDebugName(target, data);
55 > if (dbgName) {
56 let count = countPerName.get(dbgName) ?? 0;
57 count++;
61 return result;
62 }
63 > return undefined; debugName.ts
64 > }
66 > function computeDebugName(self: object, data: DebugNameData): string | undefined { debugName.ts
67 > const cached = cachedDebugName.get(self);
68 > if (cached) {
69 return cached;
70 }
72 > const ownerStr = data.owner ? formatOwner(data.owner) + `.` : '';
73 >
74 > let result: string | undefined;
75 > const debugNameSource = data.debugNameSource;
76 > if (debugNameSource !== undefined) {
77 if (typeof debugNameSource === 'function') {
78 result = debugNameSource();
84 }
85 }
87 > const referenceFn = data.referenceFn;
88 > if (referenceFn !== undefined) {
89 result = getFunctionName(referenceFn);
90 if (result !== undefined) {
92 }
93 }
95 > if (data.owner !== undefined) {
96 > const key = findKey(data.owner, self); debugName.ts
97 > if (key !== undefined) {
98 return ownerStr + key;
99 }
100 > } debugName.ts
101 > return undefined; debugName.ts
102 > }
103 > debugName.ts
104 > function findKey(obj: object, value: object): string | undefined { debugName.ts
105 > for (const key in obj) {
106 > if ((obj as Record<string, unknown>)[key] === value) {
107 return key;
108 }
109 > } debugName.ts
110 > return undefined;
111 > }
112 > debugName.ts
113 > const countPerClassName = new Map<string, number>();
114 > const ownerId = new WeakMap<object, string>();
115 >
116 > function formatOwner(owner: object): string { debugName.ts
117 > const id = ownerId.get(owner);
118 > if (id) {
119 return id;
120 }
121 > const className = getClassName(owner) ?? 'Object'; debugName.ts
122 > let count = countPerClassName.get(className) ?? 0;
123 > count++;
124 > countPerClassName.set(className, count);
125 > const result = count === 1 ? className : `${className}#${count}`;
126 > ownerId.set(owner, result);
127 > return result;
128 > }
129 > debugName.ts
130 > export function getClassName(obj: object): string | undefined {
131 > const ctor = obj.constructor; debugName.ts
132 > if (ctor) {
133 > if (ctor.name === 'Object') {
134 return undefined;
135 }
136 > return ctor.name; debugName.ts
137 > }
138 return undefined;
139 }
140 > debugName.ts
141 > export function getFunctionName(fn: Function): string | undefined {
142 const fnSrc = fn.toString();
143 // Pattern: /** @description ... */
src/vs/workbench/services/assignment/common/assignmentFilters.ts 106 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- assignmentFilters.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import type { IExperimentationFilterProvider } from 'tas-client';
7 > import { Emitter } from '../../../../base/common/event.js';
8 > import { Disposable } from '../../../../base/common/lifecycle.js';
9 > import { getInternalOrg } from '../../../../platform/assignment/common/assignment.js';
10 > import { IDefaultAccountService } from '../../../../platform/defaultAccount/common/defaultAccount.js';
11 > import { ExtensionIdentifier } from '../../../../platform/extensions/common/extensions.js';
12 > import { ILogService } from '../../../../platform/log/common/log.js';
13 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
14 > import { IChatEntitlementService } from '../../chat/common/chatEntitlementService.js';
15 > import { IExtensionService } from '../../extensions/common/extensions.js';
16 >
17 > export enum ExtensionsFilter {
18 >
19 > /**
20 > * Version of the github.copilot extension.
21 > */
22 > CopilotExtensionVersion = 'X-Copilot-RelatedPluginVersion-githubcopilot',
23 >
24 > /**
25 > * Version of the github.copilot-chat extension.
26 > */
27 > CopilotChatExtensionVersion = 'X-Copilot-RelatedPluginVersion-githubcopilotchat',
28 >
29 > /**
30 > * Version of the completions version.
31 > */
32 > CompletionsVersionInCopilotChat = 'X-VSCode-CompletionsInChatExtensionVersion',
33 >
34 > /**
35 > * SKU of the copilot entitlement.
36 > */
37 > CopilotSku = 'X-GitHub-Copilot-SKU',
38 >
39 > /**
40 > * The internal org of the user.
41 > */
42 > MicrosoftInternalOrg = 'X-Microsoft-Internal-Org',
43 >
44 > /**
45 > * The tracking ID of the user from Copilot entitlement API.
46 > */
47 > CopilotTrackingId = 'X-Copilot-CopilotTrackingId',
48 >
49 > /**
50 > * Whether the `sn` flag is set to `'1'` in the copilot token.
51 > */
52 > CopilotIsSn = 'X-GitHub-Copilot-IsSn',
53 >
54 > /**
55 > * Whether the `fcv1` flag is set to `'1'` in the copilot token.
56 > */
57 > CopilotIsFcv1 = 'X-GitHub-Copilot-IsFcv1',
58 > }
59 >
60 > enum StorageVersionKeys {
61 > CopilotExtensionVersion = 'extensionsAssignmentFilterProvider.copilotExtensionVersion',
62 > CopilotChatExtensionVersion = 'extensionsAssignmentFilterProvider.copilotChatExtensionVersion',
63 > CompletionsVersion = 'extensionsAssignmentFilterProvider.copilotCompletionsVersion',
64 > CopilotSku = 'extensionsAssignmentFilterProvider.copilotSku',
65 > CopilotInternalOrg = 'extensionsAssignmentFilterProvider.copilotInternalOrg',
66 > CopilotTrackingId = 'extensionsAssignmentFilterProvider.copilotTrackingId',
67 > CopilotIsSn = 'extensionsAssignmentFilterProvider.copilotIsSn',
68 > CopilotIsFcv1 = 'extensionsAssignmentFilterProvider.copilotIsFcv1',
69 > }
70 >
71 > export class CopilotAssignmentFilterProvider extends Disposable implements IExperimentationFilterProvider {
72 > private copilotChatExtensionVersion: string | undefined;
73 > private copilotExtensionVersion: string | undefined;
74 > // TODO@benibenj remove this when completions have been ported to chat
75 > private copilotCompletionsVersion: string | undefined;
76 >
77 > private copilotInternalOrg: string | undefined;
78 > private copilotSku: string | undefined;
79 > private copilotTrackingId: string | undefined;
80 > private copilotIsSn: string | undefined;
81 > private copilotIsFcv1: string | undefined;
82 >
83 > private readonly _onDidChangeFilters = this._register(new Emitter<void>());
84 > readonly onDidChangeFilters = this._onDidChangeFilters.event;
85 >
86 > constructor(
87 @IExtensionService private readonly _extensionService: IExtensionService,
88 @ILogService private readonly _logService: ILogService,
120 this.updateCopilotTokenInfo();
121 }
123 > private async updateExtensionVersions() {
124 let copilotExtensionVersion;
125 let copilotChatExtensionVersion;
156 this._onDidChangeFilters.fire();
157 }
159 > private updateCopilotEntitlementInfo() {
160 const newSku = this._chatEntitlementService.sku;
161 const newTrackingId = this._chatEntitlementService.copilotTrackingId;
177 this._onDidChangeFilters.fire();
178 }
180 > private updateCopilotTokenInfo() {
181 const tokenInfo = this._defaultAccountService.copilotTokenInfo;
182 const newIsSn = tokenInfo?.sn === '1' ? '1' : '0';
196 this._onDidChangeFilters.fire();
197 }
199 > /**
200 > * Returns a version string that can be parsed by the TAS client.
201 > * The tas client cannot handle suffixes lke "-insider"
202 > * Ref: https://github.com/microsoft/tas-client/blob/30340d5e1da37c2789049fcf45928b954680606f/vscode-tas-client/src/vscode-tas-client/VSCodeFilterProvider.ts#L35
203 > *
204 > * @param version Version string to be trimmed.
205 > */
206 > private static trimVersionSuffix(version: string): string {
207 const regex = /\-[a-zA-Z0-9]+$/;
208 const result = version.split(regex);
210 return result[0];
211 }
213 > getFilterValue(filter: string): string | null {
214 switch (filter) {
215 case ExtensionsFilter.CopilotExtensionVersion:
233 }
234 }
236 > getFilters(): Map<string, string | null> {
237 const filters = new Map<string, string | null>();
238 const filterValues = Object.values(ExtensionsFilter);
src/vs/platform/agentHost/common/state/protocol/channels-changeset/commands.ts 105 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI, ContentRef, StringOrMarkdown, TextRange } from '../common/state.js';
10 > import type { BaseParams } from '../common/commands.js';
11 >
12 > // ─── invokeChangesetOperation ────────────────────────────────────────────────
13 >
14 > /**
15 > * Discriminator for {@link ChangesetOperationTarget}. Mirrors the
16 > * non-`Changeset` members of {@link ChangesetOperationScope} — the
17 > * `Changeset` scope has no target.
18 > *
19 > * @category Commands
20 > */
21 > export const enum ChangesetOperationTargetKind {
22 > /** Operation acts on a single file. */
23 > Resource = 'resource',
24 > /** Operation acts on a line range within a single file. */
25 > Range = 'range',
26 > }
27 >
28 > /**
29 > * Identifies the file or range a {@link ChangesetOperation} should act on.
30 > *
31 > * The `kind` MUST match one of the operation's declared
32 > * {@link ChangesetOperation.scopes}.
33 > *
34 > * @category Commands
35 > */
36 > export type ChangesetOperationTarget =
37 > | { kind: ChangesetOperationTargetKind.Resource; resource: URI; side?: 'before' | 'after' }
38 > | { kind: ChangesetOperationTargetKind.Range; resource: URI; side?: 'before' | 'after'; range: TextRange };
39 >
40 > /**
41 > * Optional follow-up surfaced by the server after an operation completes —
42 > * a {@link ContentRef} the client can fetch and display.
43 > *
44 > * Set `external` to `true` to open the content in the user's preferred
45 > * external handler (e.g. browser); otherwise the client is expected to
46 > * surface it inline.
47 > *
48 > * @category Commands
49 > */
50 > export interface ChangesetOperationFollowUp {
51 > content: ContentRef;
52 > /** When `true`, open in an external handler rather than inline. */
53 > external?: boolean;
54 > }
55 >
56 > /**
57 > * Invokes a server-defined {@link ChangesetOperation} against a changeset,
58 > * a single file, or a line range.
59 > *
60 > * The server validates that `operationId` exists in the changeset's
61 > * current `operations` list and that the requested `target.kind` is
62 > * contained in the operation's `scopes`. Invalid combinations result in a
63 > * JSON-RPC error.
64 > *
65 > * State changes resulting from invocation flow back through the normal
66 > * `changeset/*` action stream on the relevant changeset URIs. Clients
67 > * SHOULD NOT synthesise local optimistic changes for invocations unless
68 > * the server explicitly opts in via a future capability.
69 > *
70 > * @category Commands
71 > * @method invokeChangesetOperation
72 > * @direction Client → Server
73 > * @messageType Request
74 > * @version 2
75 > */
76 > export interface InvokeChangesetOperationParams extends BaseParams {
77 > /** The expanded changeset URI. */
78 > channel: URI;
79 > /** Matches {@link ChangesetOperation.id} from the changeset's `operations` list. */
80 > operationId: string;
81 > /**
82 > * Target of the operation. Required iff the chosen scope is
83 > * `'resource'` or `'range'`. Omit for changeset-scoped operations.
84 > */
85 > target?: ChangesetOperationTarget;
86 > }
87 >
88 > /**
89 > * Result of the {@link InvokeChangesetOperationParams | `invokeChangesetOperation`}
90 > * command.
91 > *
92 > * Success is implicit: the server returns this result when it accepted
93 > * the operation. Failure is signalled by rejecting the JSON-RPC request
94 > * with an appropriate error code, not by any field on this result. The
95 > * operation MAY still produce subsequent failure feedback through the
96 > * {@link ChangesetStatusChangedAction | `changeset/statusChanged`} stream.
97 > *
98 > * @category Commands
99 > */
100 > export interface InvokeChangesetOperationResult {
101 > /** Optional human-readable message describing the result. */
102 > message?: StringOrMarkdown;
103 > /** Optional follow-up: a URI to open (e.g. a PR), a content ref, etc. */
104 > followUp?: ChangesetOperationFollowUp;
105 > }
src/vs/platform/keybinding/test/common/mockKeybindingService.ts 105 covered LOC · 42 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockKeybindingService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { KeyCodeChord, Keybinding, ResolvedKeybinding } from '../../../../base/common/keybindings.js';
8 > import { Disposable } from '../../../../base/common/lifecycle.js';
9 > import { OS } from '../../../../base/common/platform.js';
10 > import { ContextKeyExpression, ContextKeyValue, IContextKey, IContextKeyChangeEvent, IContextKeyService, IContextKeyServiceTarget, IScopedContextKeyService } from '../../../contextkey/common/contextkey.js';
11 > import { IKeybindingService, IKeyboardEvent } from '../../common/keybinding.js';
12 > import { NoMatchingKb, ResolutionResult } from '../../common/keybindingResolver.js';
13 > import { ResolvedKeybindingItem } from '../../common/resolvedKeybindingItem.js';
14 > import { USLayoutResolvedKeybinding } from '../../common/usLayoutResolvedKeybinding.js';
15 >
16 > class MockKeybindingContextKey<T extends ContextKeyValue = ContextKeyValue> implements IContextKey<T> {
17 > private _defaultValue: T | undefined;
18 > private _value: T | undefined;
19 >
20 > constructor(defaultValue: T | undefined) {
21 > this._defaultValue = defaultValue; mockKeybindingService.ts
22 > this._value = this._defaultValue;
23 > }
25 > public set(value: T | undefined): void {
26 > this._value = value; mockKeybindingService.ts
27 > }
29 > public reset(): void {
30 this._value = this._defaultValue;
31 }
33 > public get(): T | undefined {
34 return this._value;
35 }
37 >
38 > export class MockContextKeyService implements IContextKeyService {
40 > public _serviceBrand: undefined;
41 > private _keys = new Map<string, IContextKey<any>>();
43 > public dispose(): void {
44 //
45 }
46 > public createKey<T extends ContextKeyValue = ContextKeyValue>(key: string, defaultValue: T | undefined): IContextKey<T> { mockKeybindingService.ts
47 > const ret = new MockKeybindingContextKey(defaultValue); mockKeybindingService.ts
48 > this._keys.set(key, ret);
49 > return ret;
50 > }
51 > public contextMatchesRules(rules: ContextKeyExpression): boolean { mockKeybindingService.ts
52 return false;
53 }
54 > public get onDidChangeContext(): Event<IContextKeyChangeEvent> { mockKeybindingService.ts
55 > return Event.None; mockKeybindingService.ts
56 > }
57 > public bufferChangeEvents(callback: () => void) { callback(); } mockKeybindingService.ts
58 > public getContextKeyValue(key: string) {
59 const value = this._keys.get(key);
60 if (value) {
62 }
63 }
64 > public getContext(domNode: HTMLElement): any { mockKeybindingService.ts
65 return null;
66 }
67 > public createScoped(domNode: HTMLElement): IScopedContextKeyService { mockKeybindingService.ts
68 return this;
69 }
70 > public createOverlay(): IContextKeyService { mockKeybindingService.ts
71 return this;
72 }
73 > updateParent(_parentContextKeyService: IContextKeyService): void { mockKeybindingService.ts
74 // no-op
75 }
77 >
78 > export class MockScopableContextKeyService extends MockContextKeyService {
79 > /**
80 > * Don't implement this for all tests since we rarely depend on this behavior and it isn't implemented fully
81 > */
82 > public override createScoped(domNote: HTMLElement): IScopedContextKeyService {
83 return new MockScopableContextKeyService();
84 }
86 >
87 > export class MockKeybindingService implements IKeybindingService {
88 public _serviceBrand: undefined;
89
90 public readonly inChordMode: boolean = false;
92 > public get onDidUpdateKeybindings(): Event<void> {
93 return Event.None;
94 }
96 > public getDefaultKeybindingsContent(): string {
97 return '';
98 }
100 > public getDefaultKeybindings(): ResolvedKeybindingItem[] {
101 return [];
102 }
104 > public getKeybindings(): ResolvedKeybindingItem[] {
105 return [];
106 }
108 > public resolveKeybinding(keybinding: Keybinding): ResolvedKeybinding[] {
109 return USLayoutResolvedKeybinding.resolveKeybinding(keybinding, OS);
110 }
112 > public resolveKeyboardEvent(keyboardEvent: IKeyboardEvent): ResolvedKeybinding {
113 const chord = new KeyCodeChord(
114 keyboardEvent.ctrlKey,
120 return this.resolveKeybinding(chord.toKeybinding())[0];
121 }
123 > public resolveUserBinding(userBinding: string): ResolvedKeybinding[] {
124 return [];
125 }
127 > public lookupKeybindings(commandId: string): ResolvedKeybinding[] {
128 return [];
129 }
131 > public lookupKeybinding(commandId: string): ResolvedKeybinding | undefined {
132 return undefined;
133 }
135 > public customKeybindingsCount(): number {
136 return 0;
137 }
139 > public softDispatch(keybinding: IKeyboardEvent, target: IContextKeyServiceTarget): ResolutionResult {
140 return NoMatchingKb;
141 }
143 > public dispatchByUserSettingsLabel(userSettingsLabel: string, target: IContextKeyServiceTarget): void {
144
145 }
147 > public dispatchEvent(e: IKeyboardEvent, target: IContextKeyServiceTarget): boolean {
148 return false;
149 }
151 > public enableKeybindingHoldMode(commandId: string): undefined {
152 return undefined;
153 }
155 > public mightProducePrintableCharacter(e: IKeyboardEvent): boolean {
156 return false;
157 }
159 > public toggleLogging(): boolean {
160 return false;
161 }
163 > public _dumpDebugInfo(): string {
164 return '';
165 }
167 > public _dumpDebugInfoJSON(): string {
168 return '';
169 }
171 > public registerSchemaContribution() {
172 return Disposable.None;
173 }
175 > public appendKeybinding(label: string, _commandId: string, _context?: IContextKeyService, _enforceContextCheck?: boolean): string {
176 return label;
177 }
src/vs/base/common/observableInternal/utils/promise.ts 104 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promise.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { DisposableStore } from '../../lifecycle.js';
6 > import { IObservable, ISettableObservable } from '../base.js';
7 > import { autorun } from '../reactions/autorun.js';
8 > import { transaction } from '../transaction.js';
9 > import { derived } from '../observables/derived.js';
10 > import { observableValue } from '../observables/observableValue.js';
11 >
12 > export class ObservableLazy<T> {
13 > private readonly _value = observableValue<T | undefined>(this, undefined);
14 >
15 > /**
16 > * The cached value.
17 > * Does not force a computation of the value.
18 > */
19 > public get cachedValue(): IObservable<T | undefined> { return this._value; }
20 >
21 > constructor(private readonly _computeValue: () => T) {
22 }
23 > promise.ts
24 > /**
25 > * Returns the cached value.
26 > * Computes the value if the value has not been cached yet.
27 > */
28 > public getValue(): T {
29 let v = this._value.get();
30 if (!v) {
34 return v;
35 }
36 > } promise.ts
37 >
38 > /**
39 > * A promise whose state is observable.
40 > */
41 > export class ObservablePromise<T> {
42 > public static fromFn<T>(fn: () => Promise<T>): ObservablePromise<T> {
43 > return new ObservablePromise(fn());
44 > }
45 >
46 > public static resolved<T>(value: T): ObservablePromise<T> {
47 return new ObservablePromise(Promise.resolve(value));
48 }
49 > promise.ts
50 > private readonly _value = observableValue<PromiseResult<T> | undefined>(this, undefined);
51 >
52 > /**
53 > * The promise that this object wraps.
54 > */
55 > public readonly promise: Promise<T>;
56 >
57 > /**
58 > * The current state of the promise.
59 > * Is `undefined` if the promise didn't resolve yet.
60 > */
61 > public readonly promiseResult: IObservable<PromiseResult<T> | undefined> = this._value;
62 >
63 > constructor(promise: Promise<T>) {
64 this.promise = promise.then(value => {
65 transaction(tx => {
76 });
77 }
78 > promise.ts
79 > public readonly resolvedValue = derived(this, reader => {
80 > const result = this.promiseResult.read(reader); promise.ts
81 > if (!result) {
82 > return undefined;
83 > }
84 > return result.getDataOrThrow();
85 > }); promise.ts
86 > }
87 >
88 > export class PromiseResult<T> {
89 > constructor(
90 /**
91 * The value of the resolved promise.
101 ) {
102 }
103 > promise.ts
104 > /**
105 > * Returns the value if the promise resolved, otherwise throws the error.
106 > */
107 > public getDataOrThrow(): T {
108 if (this.error) {
109 throw this.error;
111 return this.data!;
112 }
113 > } promise.ts
114 >
115 > /**
116 > * Tracks a changing {@link ObservablePromise}, exposing the last resolved value
117 > * and whether a newer promise is still pending.
118 > */
119 > export class ObservableResolvedPromise<T> {
120 > private readonly _lastResolved: ISettableObservable<T>;
121 > public readonly lastResolved: IObservable<T>;
122 >
123 > private readonly _isResolving = observableValue<boolean>(this, false);
124 > public readonly isResolving: IObservable<boolean> = this._isResolving;
125 >
126 > private _runningPromise: ObservablePromise<T> | undefined;
127 >
128 > constructor(
129 source: IObservable<ObservablePromise<T>>,
130 initialValue: T,
149 }));
150 }
151 > } promise.ts
152 >
153 > /**
154 > * A lazy promise whose state is observable.
155 > */
156 > export class ObservableLazyPromise<T> {
157 > private readonly _lazyValue = new ObservableLazy(() => new ObservablePromise(this._computePromise()));
158 >
159 > /**
160 > * Does not enforce evaluation of the promise compute function.
161 > * Is undefined if the promise has not been computed yet.
162 > */
163 > public readonly cachedPromiseResult = derived(this, reader => this._lazyValue.cachedValue.read(reader)?.promiseResult.read(reader));
164 >
165 > constructor(private readonly _computePromise: () => Promise<T>) {
166 }
167 > promise.ts
168 > public getPromise(): Promise<T> {
169 return this._lazyValue.getValue().promise;
170 }
171 > } promise.ts
src/vs/base/common/observableInternal/observables/observableValue.ts 103 covered LOC · 31 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableValue.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ISettableObservable, ITransaction } from '../base.js';
7 > import { TransactionImpl } from '../transaction.js';
8 > import { BaseObservable } from './baseObservable.js';
9 > import { EqualityComparer, IDisposable, strictEquals } from '../commonFacade/deps.js';
10 > import { DebugNameData } from '../debugName.js';
11 > import { getLogger } from '../logging/logging.js';
12 > import { DebugLocation } from '../debugLocation.js';
13 >
14 > /**
15 > * Creates an observable value.
16 > * Observers get informed when the value changes.
17 > * @template TChange An arbitrary type to describe how or why the value changed. Defaults to `void`.
18 > * Observers will receive every single change value.
19 > */
20 >
21 > export function observableValue<T, TChange = void>(name: string, initialValue: T): ISettableObservable<T, TChange>;
22 > export function observableValue<T, TChange = void>(owner: object, initialValue: T): ISettableObservable<T, TChange>;
23 > export function observableValue<T, TChange = void>(nameOrOwner: string | object, initialValue: T, debugLocation = DebugLocation.ofCaller()): ISettableObservable<T, TChange> {
24 > let debugNameData: DebugNameData; observableValue.ts
25 > if (typeof nameOrOwner === 'string') {
26 > debugNameData = new DebugNameData(undefined, nameOrOwner, undefined); observableValue.ts
27 > } else { observableValue.ts
28 > debugNameData = new DebugNameData(nameOrOwner, undefined, undefined); observableValue.ts
29 > }
30 > return new ObservableValue(debugNameData, initialValue, strictEquals, debugLocation); observableValue.ts
31 > }
33 > export class ObservableValue<T, TChange = void>
34 > extends BaseObservable<T, TChange>
35 > implements ISettableObservable<T, TChange> {
36 > protected _value: T;
37 >
38 > get debugName() {
39 > return this._debugNameData.getDebugName(this) ?? 'ObservableValue';
40 > }
41 >
42 > constructor(
43 > private readonly _debugNameData: DebugNameData, observableValue.ts
44 > initialValue: T,
45 > private readonly _equalityComparator: EqualityComparer<T>,
46 > debugLocation: DebugLocation
47 > ) {
48 > super(debugLocation);
49 > this._value = initialValue;
50 >
51 > getLogger()?.handleObservableUpdated(this, { hadValue: false, newValue: initialValue, change: undefined, didChange: true, oldValue: undefined });
52 > }
53 > public override get(): T { observableValue.ts
54 > return this._value; observableValue.ts
55 > }
57 > public set(value: T, tx: ITransaction | undefined, change: TChange): void {
58 > if (change === undefined && this._equalityComparator(this._value, value)) { observableValue.ts
59 > return; observableValue.ts
60 > }
62 > let _tx: TransactionImpl | undefined;
63 > if (!tx) {
64 > tx = _tx = new TransactionImpl(() => { }, () => `Setting ${this.debugName}`); observableValue.ts
65 > }
66 > try { observableValue.ts
67 > const oldValue = this._value;
68 > this._setValue(value);
69 > getLogger()?.handleObservableUpdated(this, { oldValue, newValue: value, change, didChange: true, hadValue: true }); observableValue.ts
70 >
71 > for (const observer of this._observers) {
72 > tx.updateObserver(observer, this); observableValue.ts
73 > observer.handleChange(this, change);
74 > }
75 > } finally { observableValue.ts
76 > if (_tx) {
77 > _tx.finish(); observableValue.ts
78 > }
82 > override toString(): string {
83 return `${this.debugName}: ${this._value}`;
84 }
86 > protected _setValue(newValue: T): void {
87 > this._value = newValue; observableValue.ts
88 > }
90 > public debugGetState() {
91 return {
92 value: this._value,
93 };
94 }
96 > public debugSetValue(value: unknown) {
97 this._value = value as T;
98 }
100 > /**
101 > * A disposable observable. When disposed, its value is also disposed.
102 > * When a new value is set, the previous value is disposed.
103 > */
104 >
105 > export function disposableObservableValue<T extends IDisposable | undefined, TChange = void>(nameOrOwner: string | object, initialValue: T, debugLocation = DebugLocation.ofCaller()): ISettableObservable<T, TChange> & IDisposable {
106 let debugNameData: DebugNameData;
107 if (typeof nameOrOwner === 'string') {
112 return new DisposableObservableValue(debugNameData, initialValue, strictEquals, debugLocation);
113 }
115 > export class DisposableObservableValue<T extends IDisposable | undefined, TChange = void> extends ObservableValue<T, TChange> implements IDisposable {
116 > protected override _setValue(newValue: T): void {
117 if (this._value === newValue) {
118 return;
src/vs/base/common/mime.ts 102 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mime.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { extname } from './path.js';
7 >
8 > export const Mimes = Object.freeze({
9 > text: 'text/plain',
10 > binary: 'application/octet-stream',
11 > unknown: 'application/unknown',
12 > markdown: 'text/markdown',
13 > latex: 'text/latex',
14 > uriList: 'text/uri-list',
15 > html: 'text/html',
16 > });
17 >
18 > interface MapExtToMediaMimes {
19 > [index: string]: string | string[];
20 > }
21 >
22 > const mapExtToTextMimes: Record<string, string> = {
23 > '.css': 'text/css',
24 > '.csv': 'text/csv',
25 > '.htm': 'text/html',
26 > '.html': 'text/html',
27 > '.ics': 'text/calendar',
28 > '.js': 'text/javascript',
29 > '.mjs': 'text/javascript',
30 > '.txt': 'text/plain',
31 > '.xml': 'text/xml'
32 > };
33 >
34 > // Known media mimes that we can handle
35 > const mapExtToMediaMimes: MapExtToMediaMimes = {
36 > '.aac': 'audio/x-aac',
37 > '.avi': 'video/x-msvideo',
38 > '.bmp': 'image/bmp',
39 > '.flv': 'video/x-flv',
40 > '.gif': 'image/gif',
41 > '.ico': 'image/x-icon',
42 > '.jpe': ['image/jpg', 'image/jpeg'],
43 > '.jpeg': ['image/jpg', 'image/jpeg'],
44 > '.jpg': ['image/jpg', 'image/jpeg'],
45 > '.m1v': 'video/mpeg',
46 > '.m2a': 'audio/mpeg',
47 > '.m2v': 'video/mpeg',
48 > '.m3a': 'audio/mpeg',
49 > '.mid': 'audio/midi',
50 > '.midi': 'audio/midi',
51 > '.mk3d': 'video/x-matroska',
52 > '.mks': 'video/x-matroska',
53 > '.mkv': 'video/x-matroska',
54 > '.mov': 'video/quicktime',
55 > '.movie': 'video/x-sgi-movie',
56 > '.mp2': 'audio/mpeg',
57 > '.mp2a': 'audio/mpeg',
58 > '.mp3': 'audio/mpeg',
59 > '.mp4': 'video/mp4',
60 > '.mp4a': 'audio/mp4',
61 > '.mp4v': 'video/mp4',
62 > '.mpe': 'video/mpeg',
63 > '.mpeg': 'video/mpeg',
64 > '.mpg': 'video/mpeg',
65 > '.mpg4': 'video/mp4',
66 > '.mpga': 'audio/mpeg',
67 > '.oga': 'audio/ogg',
68 > '.ogg': 'audio/ogg',
69 > '.opus': 'audio/opus',
70 > '.ogv': 'video/ogg',
71 > '.png': 'image/png',
72 > '.psd': 'image/vnd.adobe.photoshop',
73 > '.qt': 'video/quicktime',
74 > '.spx': 'audio/ogg',
75 > '.svg': 'image/svg+xml',
76 > '.tga': 'image/x-tga',
77 > '.tif': 'image/tiff',
78 > '.tiff': 'image/tiff',
79 > '.wav': 'audio/x-wav',
80 > '.webm': 'video/webm',
81 > '.webp': 'image/webp',
82 > '.wma': 'audio/x-ms-wma',
83 > '.wmv': 'video/x-ms-wmv',
84 > '.woff': 'application/font-woff',
85 > };
86 >
87 > export function getMediaOrTextMime(path: string): string | undefined {
88 const ext = extname(path);
89 const textMime = mapExtToTextMimes[ext.toLowerCase()];
94 }
95 }
96 > mime.ts
97 > export function getMediaMime(path: string): string | undefined {
98 const ext = extname(path);
99 const mimeType = mapExtToMediaMimes[ext.toLowerCase()];
100 return Array.isArray(mimeType) ? mimeType[0] : mimeType;
101 }
102 > mime.ts
103 > export function getExtensionForMimeType(mimeType: string): string | undefined {
104 for (const extension in mapExtToMediaMimes) {
105 const value = mapExtToMediaMimes[extension];
111 return undefined;
112 }
113 > mime.ts
114 > const _simplePattern = /^(.+)\/(.+?)(;.+)?$/;
115 >
116 > export function normalizeMimeType(mimeType: string): string;
117 > export function normalizeMimeType(mimeType: string, strict: true): string | undefined;
118 > export function normalizeMimeType(mimeType: string, strict?: true): string | undefined {
119
120 const match = _simplePattern.exec(mimeType);
128 return `${match[1].toLowerCase()}/${match[2].toLowerCase()}${match[3] ?? ''}`;
129 }
130 > mime.ts
131 > /**
132 > * Whether the provided mime type is a text stream like `stdout`, `stderr`.
133 > */
134 > export function isTextStreamMime(mimeType: string) {
135 return ['application/vnd.code.notebook.stdout', 'application/vnd.code.notebook.stderr'].includes(mimeType);
136 }
src/vs/platform/agentHost/common/agentHostUri.ts 102 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentHostUri.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { decodeBase64, encodeBase64, VSBuffer } from '../../../base/common/buffer.js';
7 > import { Schemas } from '../../../base/common/network.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import type { ResourceLabelFormatter } from '../../label/common/label.js';
10 >
11 > /**
12 > * The URI scheme for accessing files on a remote agent host.
13 > *
14 > * The original file path is kept verbatim as the URI path so resource
15 > * labels, language detection, and path comparisons see a real path. The
16 > * original scheme, authority, and query are carried in a single
17 > * url-safe-base64 `_ah` query parameter so any remote resource can be
18 > * represented without assuming `file://`:
19 > *
20 > * ```
21 > * vscode-agent-host://[connectionAuthority][originalPath]?_ah=[meta]#[originalFragment]
22 > * ```
23 > *
24 > * where `meta` is {@link IAgentHostUriMeta} as url-safe-base64-encoded
25 > * JSON. Encoding the metadata as a single opaque parameter (rather than
26 > * raw JSON) keeps the query a well-formed parameter list, so unrelated
27 > * query parameters such as `vscodeLinkType` can coexist on the wrapped
28 > * URI without corrupting the metadata. For example,
29 > * `file:///home/user/foo.ts` on remote `my-server` becomes:
30 > * ```
31 > * vscode-agent-host://my-server/home/user/foo.ts?_ah=eyJzY2hlbWUiOiJmaWxlIn0
32 > * ```
33 > */
34 > export const AGENT_HOST_SCHEME = 'vscode-agent-host';
35 >
36 > /**
37 > * Query parameter that carries the {@link IAgentHostUriMeta} payload.
38 > */
39 > const AGENT_HOST_META_PARAM = '_ah';
40 >
41 > /**
42 > * Metadata carried in the query of a {@link AGENT_HOST_SCHEME} URI so the
43 > * original URI can be reconstructed while keeping the path label-friendly.
44 > */
45 > interface IAgentHostUriMeta {
46 > /** Original URI scheme (e.g. `file`, `git-blob`). */
47 > readonly scheme: string;
48 > /** Original URI authority, omitted when empty. */
49 > readonly authority?: string;
50 > /** Original URI query, omitted when empty. */
51 > readonly query?: string;
52 > }
53 >
54 > /**
55 > * Wraps a remote URI into a {@link AGENT_HOST_SCHEME} URI that can be
56 > * resolved through the agent host filesystem provider.
57 > *
58 > * @param originalUri The URI on the remote (e.g. `file:///path` or
59 > * `agenthost-content:///sessionId/...`)
60 > * @param connectionAuthority The sanitized connection identifier used as
61 > * the URI authority (from {@link agentHostAuthority}).
62 > */
63 > export function toAgentHostUri(originalUri: URI, connectionAuthority: string): URI {
64 if (connectionAuthority === 'local' && originalUri.scheme === Schemas.file) {
65 return originalUri;
81 });
82 }
84 > /**
85 > * Extracts the original URI from a {@link AGENT_HOST_SCHEME} URI.
86 > *
87 > * The inverse of {@link toAgentHostUri}.
88 > */
89 > export function fromAgentHostUri(agentHostUri: URI): URI {
90 if (agentHostUri.scheme !== AGENT_HOST_SCHEME) {
91 return agentHostUri;
116 });
117 }
119 > /**
120 > * Strips the redundant `ws://` scheme from an address. The transport layer
121 > * already defaults to `ws://`, so only `wss://` needs to be preserved.
122 > */
123 > export function normalizeRemoteAgentHostAddress(address: string): string {
124 if (address.startsWith('ws://')) {
125 return address.slice('ws://'.length);
127 return address;
128 }
130 > /**
131 > * Encode a remote address into an identifier that is safe for use in
132 > * both URI schemes and URI authorities, and is collision-free.
133 > *
134 > * Three tiers:
135 > * 1. Purely alphanumeric addresses are returned as-is.
136 > * 2. "Normal" addresses containing only `[a-zA-Z0-9.:-]` get colons
137 > * replaced with `__` (double underscore) for human readability.
138 > * Addresses containing `_` skip this tier to keep the encoding
139 > * collision-free (`__` can only appear from colon replacement).
140 > * 3. Everything else is url-safe base64-encoded with a `b64-` prefix.
141 > */
142 > export function agentHostAuthority(address: string): string {
143 const normalized = normalizeRemoteAgentHostAddress(address);
144 if (/^[a-zA-Z0-9]+$/.test(normalized)) {
150 return `b64-${encodeBase64(VSBuffer.fromString(normalized), false, true)}`;
151 }
153 > /**
154 > * Label formatter for {@link AGENT_HOST_SCHEME} URIs. The URI path is
155 > * already the original resource path, so the label is the path verbatim.
156 > */
157 > export const AGENT_HOST_LABEL_FORMATTER: ResourceLabelFormatter = {
158 > scheme: AGENT_HOST_SCHEME,
159 > formatting: {
160 > label: '${path}',
161 > separator: '/',
162 > },
163 > };
src/vs/workbench/contrib/chat/common/chatPerf.ts 102 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatPerf.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { mark, clearMarks } from '../../../../base/common/performance.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { chatSessionResourceToId } from './model/chatUri.js';
9 >
10 > const chatPerfPrefix = 'code/chat/';
11 >
12 > /** Tracks all mark names emitted per session so they can be cleared individually. */
13 > const chatMarksBySession = new Map<string, Set<string>>();
14 >
15 > /**
16 > * Well-defined perf scenarios for chat request lifecycle.
17 > * Each mark is a boundary of a measurable scenario — don't add marks
18 > * without defining what scenario they belong to.
19 > *
20 > * ## Scenarios
21 > *
22 > * **Time to UI Feedback** (perceived input lag):
23 > * `request/start` → `request/uiUpdated`
24 > *
25 > * **Instruction Collection Overhead**:
26 > * `request/willCollectInstructions` → `request/didCollectInstructions`
27 > *
28 > * **Extension Activation Wait** (first-request cold start):
29 > * `code/chat/willWaitForActivation` → `code/chat/didWaitForActivation`
30 > * (global marks, not session-scoped — emitted via {@link markChatGlobal})
31 > *
32 > * **Time to First Token** (the headline metric):
33 > * `request/start` → `request/firstToken`
34 > *
35 > * **Total Request Duration**:
36 > * `request/start` → `request/complete`
37 > *
38 > * **Agent Invocation Time** (LLM round-trip):
39 > * `agent/willInvoke` → `agent/didInvoke`
40 > */
41 > export const ChatPerfMark = {
42 > /** User pressed Enter / request initiated */
43 > RequestStart: 'request/start',
44 > /** Request added to model → UI shows the message */
45 > RequestUiUpdated: 'request/uiUpdated',
46 > /** Begin collecting .instructions.md / skills / hooks */
47 > WillCollectInstructions: 'request/willCollectInstructions',
48 > /** Done collecting instructions */
49 > DidCollectInstructions: 'request/didCollectInstructions',
50 > /** First streamed response content received */
51 > FirstToken: 'request/firstToken',
52 > /** Response fully complete */
53 > RequestComplete: 'request/complete',
54 > /** Agent invoke begins (LLM round-trip start) */
55 > AgentWillInvoke: 'agent/willInvoke',
56 > /** Agent invoke returns (LLM round-trip end) */
57 > AgentDidInvoke: 'agent/didInvoke',
58 > } as const;
59 >
60 > /**
61 > * Emits a performance mark scoped to a chat session:
62 > * `code/chat/<sessionResource>/<name>`
63 > *
64 > * Marks are automatically cleaned up when the corresponding chat model is
65 > * disposed — see {@link clearChatMarks}.
66 > */
67 > export function markChat(sessionResource: URI, name: string): void {
68 > const sessionId = chatSessionResourceToId(sessionResource); chatPerf.ts
69 > const fullName = `${chatPerfPrefix}${sessionId}/${name}`;
70 > let names = chatMarksBySession.get(sessionId);
71 > if (!names) {
72 > names = new Set();
73 > chatMarksBySession.set(sessionId, names);
74 > }
75 > names.add(fullName);
76 > mark(fullName);
77 > }
79 > /**
80 > * Clears all performance marks for the given chat session.
81 > * Called when the chat model is disposed.
82 > */
83 > export function clearChatMarks(sessionResource: URI): void {
84 const sessionId = chatSessionResourceToId(sessionResource);
85 const names = chatMarksBySession.get(sessionId);
91 }
92 }
94 > /**
95 > * Well-defined one-time global perf marks (not scoped to a session).
96 > * These are emitted via {@link markChatGlobal} and are never cleared.
97 > */
98 > export const ChatGlobalPerfMark = {
99 > /** Begin waiting for chat extension activation (SetupAgent) */
100 > WillWaitForActivation: 'willWaitForActivation',
101 > /** Extension activation + readiness complete (SetupAgent) */
102 > DidWaitForActivation: 'didWaitForActivation',
103 > } as const;
104 >
105 > /**
106 > * Emits a global (non-session-scoped) performance mark:
107 > * `code/chat/<name>`
108 > *
109 > * Used for one-time marks like activation that should persist across requests.
110 > */
111 > export function markChatGlobal(name: string): void {
112 mark(`${chatPerfPrefix}${name}`);
113 }
src/vs/workbench/contrib/chat/common/requestParser/chatRequestParser.ts 101 covered LOC · 25 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatRequestParser.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../base/common/uri.js';
7 > import { IPosition, Position } from '../../../../../editor/common/core/position.js';
8 > import { Range } from '../../../../../editor/common/core/range.js';
9 > import { OffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
10 > import { IChatVariablesService, IDynamicVariable } from '../attachments/chatVariables.js';
11 > import { ChatAgentLocation, ChatModeKind } from '../constants.js';
12 > import { getChatSessionType } from '../model/chatUri.js';
13 > import { IChatAgentAttachmentCapabilities, IChatAgentData, IChatAgentService } from '../participants/chatAgents.js';
14 > import { IChatSlashCommandService } from '../participants/chatSlashCommands.js';
15 > import { IPromptsService, matchesSessionType } from '../promptSyntax/service/promptsService.js';
16 > import { ToolAndToolSetEnablementMap, IToolData, IToolSet, isToolSet } from '../tools/languageModelToolsService.js';
17 > import { ChatRequestAgentPart, ChatRequestAgentSubcommandPart, ChatRequestDynamicVariablePart, ChatRequestSlashCommandPart, ChatRequestSlashPromptPart, ChatRequestTextPart, ChatRequestToolPart, ChatRequestToolSetPart, IParsedChatRequest, IParsedChatRequestPart, chatAgentLeader, chatSubcommandLeader, chatVariableLeader } from './chatParserTypes.js';
18 >
19 > export const agentReg = /^@([\w_\-\.]+)(?=(\s|$|\b))/i; // An @-agent
20 > export const variableReg = /^#([\w_\-]+)(:\d+)?(?=(\s|$|\b))/i; // A #-variable with an optional numeric : arg (@response:2)
21 > export const slashReg = /^\/([\p{L}\d_\-\.:]+)(?=(\s|$|\b))/iu; // A / command
22 >
23 > export interface IChatParserContext {
24 > /** Used only as a disambiguator, when the query references an agent that has a duplicate with the same name. */
25 > selectedAgent?: IChatAgentData;
26 > mode?: ChatModeKind;
27 > /** Parse as this agent, even when it does not appear in the query text */
28 > forcedAgent?: IChatAgentData;
29 > attachmentCapabilities?: IChatAgentAttachmentCapabilities;
30 > sessionType?: string;
31 > }
32 >
33 > export class ChatRequestParser {
34 > constructor(
35 > @IChatAgentService private readonly agentService: IChatAgentService, chatRequestParser.ts
36 > @IChatVariablesService private readonly variableService: IChatVariablesService,
37 > @IChatSlashCommandService private readonly slashCommandService: IChatSlashCommandService,
38 > @IPromptsService private readonly promptsService: IPromptsService,
39 > ) { }
41 > parseChatRequest(sessionResource: URI, message: string, location: ChatAgentLocation = ChatAgentLocation.Chat, context: IChatParserContext = {}): IParsedChatRequest {
42 const references = this.variableService.getDynamicVariables(sessionResource); // must access this list before any async calls
43 const selectedToolAndToolSets = this.variableService.getSelectedToolAndToolSets(sessionResource);
47 return this.parseChatRequestWithReferences(references, selectedToolAndToolSets, message, location, context);
48 }
50 > parseChatRequestWithReferences(references: ReadonlyArray<IDynamicVariable>, selectedToolAndToolSets: ToolAndToolSetEnablementMap, message: string, location: ChatAgentLocation = ChatAgentLocation.Chat, context?: IChatParserContext): IParsedChatRequest {
51 > const parts: IParsedChatRequestPart[] = []; chatRequestParser.ts
52 > const toolsByName = new Map<string, IToolData>();
53 > const toolSetsByName = new Map<string, IToolSet>();
54 > for (const [entry, enabled] of selectedToolAndToolSets) {
55 if (enabled) {
56 if (isToolSet(entry)) {
61 }
62 }
64 > let lineNumber = 1;
65 > let column = 1;
66 > for (let i = 0; i < message.length; i++) {
67 > const previousChar = message.charAt(i - 1); chatRequestParser.ts
68 > const char = message.charAt(i);
69 > let newPart: IParsedChatRequestPart | undefined;
70 > if (previousChar.match(/\s/) || i === 0) {
71 > if (char === chatVariableLeader) {
72 newPart = this.tryToParseVariable(message.slice(i), i, new Position(lineNumber, column), parts, toolsByName, toolSetsByName);
73 > } else if (char === chatAgentLeader) { chatRequestParser.ts
74 newPart = this.tryToParseAgent(message.slice(i), message, i, new Position(lineNumber, column), parts, location, context);
75 > } else if (char === chatSubcommandLeader) { chatRequestParser.ts
76 newPart = this.tryToParseSlashCommand(message.slice(i), message, i, new Position(lineNumber, column), parts, location, context);
77 }
79 > if (!newPart) {
80 > newPart = this.tryToParseDynamicVariable(message.slice(i), i, new Position(lineNumber, column), references); chatRequestParser.ts
81 > }
83 >
84 > if (newPart) {
85 if (i !== 0) {
86 // Insert a part for all the text we passed over, then insert the new parsed part
97 parts.push(newPart);
98 }
100 > if (char === '\n') {
101 lineNumber++;
102 column = 1;
103 > } else { chatRequestParser.ts
104 > column++;
105 > }
106 > }
108 > const lastPart = parts.at(-1);
109 > const lastPartEnd = lastPart?.range.endExclusive ?? 0;
110 > if (lastPartEnd < message.length) {
111 > parts.push(new ChatRequestTextPart( chatRequestParser.ts
112 > new OffsetRange(lastPartEnd, message.length),
113 > new Range(lastPart?.editorRange.endLineNumber ?? 1, lastPart?.editorRange.endColumn ?? 1, lineNumber, column),
114 > message.slice(lastPartEnd, message.length)));
115 > }
117 > return {
118 > parts,
119 > text: message,
120 > };
121 > }
123 > private tryToParseAgent(message: string, fullMessage: string, offset: number, position: IPosition, parts: Array<IParsedChatRequestPart>, location: ChatAgentLocation, context: IChatParserContext | undefined): ChatRequestAgentPart | undefined {
124 const nextAgentMatch = message.match(agentReg);
125 if (!nextAgentMatch) {
171 return new ChatRequestAgentPart(agentRange, agentEditorRange, agent);
172 }
174 > private tryToParseVariable(message: string, offset: number, position: IPosition, parts: ReadonlyArray<IParsedChatRequestPart>, toolsByName: ReadonlyMap<string, IToolData>, toolSetsByName: ReadonlyMap<string, IToolSet>): ChatRequestToolPart | ChatRequestToolSetPart | undefined {
175 const nextVariableMatch = message.match(variableReg);
176 if (!nextVariableMatch) {
195 return;
196 }
198 > private tryToParseSlashCommand(remainingMessage: string, fullMessage: string, offset: number, position: IPosition, parts: ReadonlyArray<IParsedChatRequestPart>, location: ChatAgentLocation, context?: IChatParserContext): ChatRequestSlashCommandPart | ChatRequestAgentSubcommandPart | ChatRequestSlashPromptPart | undefined {
199 const nextSlashMatch = remainingMessage.match(slashReg);
200 if (!nextSlashMatch) {
276 return;
277 }
279 > private tryToParseDynamicVariable(message: string, offset: number, position: IPosition, references: ReadonlyArray<IDynamicVariable>): ChatRequestDynamicVariablePart | undefined {
280 > const refAtThisPosition = references.find(r => chatRequestParser.ts
281 r.range.startLineNumber === position.lineNumber &&
282 > r.range.startColumn === position.column); chatRequestParser.ts
283 > if (refAtThisPosition) {
284 const length = refAtThisPosition.range.endColumn - refAtThisPosition.range.startColumn;
285 const text = message.substring(0, length);
287 return new ChatRequestDynamicVariablePart(range, refAtThisPosition.range, text, refAtThisPosition.id, refAtThisPosition.modelDescription, refAtThisPosition.data, refAtThisPosition.fullName, refAtThisPosition.icon, refAtThisPosition.isFile, refAtThisPosition.isDirectory, refAtThisPosition._meta, refAtThisPosition.isAttachmentReference);
288 }
290 > return;
291 > }
src/vs/workbench/services/configuration/common/configuration.ts 101 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- configuration.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ConfigurationScope } from '../../../../platform/configuration/common/configurationRegistry.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
9 > import { refineServiceDecorator } from '../../../../platform/instantiation/common/instantiation.js';
10 > import { Event } from '../../../../base/common/event.js';
11 > import { ResourceMap } from '../../../../base/common/map.js';
12 > import { IAnyWorkspaceIdentifier } from '../../../../platform/workspace/common/workspace.js';
13 >
14 > export const FOLDER_CONFIG_FOLDER_NAME = '.vscode';
15 > export const FOLDER_SETTINGS_NAME = 'settings';
16 > export const FOLDER_SETTINGS_PATH = `${FOLDER_CONFIG_FOLDER_NAME}/${FOLDER_SETTINGS_NAME}.json`;
17 >
18 > export const defaultSettingsSchemaId = 'vscode://schemas/settings/default';
19 > export const userSettingsSchemaId = 'vscode://schemas/settings/user';
20 > export const profileSettingsSchemaId = 'vscode://schemas/settings/profile';
21 > export const machineSettingsSchemaId = 'vscode://schemas/settings/machine';
22 > export const workspaceSettingsSchemaId = 'vscode://schemas/settings/workspace';
23 > export const folderSettingsSchemaId = 'vscode://schemas/settings/folder';
24 > export const launchSchemaId = 'vscode://schemas/launch';
25 > export const tasksSchemaId = 'vscode://schemas/tasks';
26 > export const mcpSchemaId = 'vscode://schemas/mcp';
27 >
28 > export const APPLICATION_SCOPES = [ConfigurationScope.APPLICATION, ConfigurationScope.APPLICATION_MACHINE];
29 > export const PROFILE_SCOPES = [ConfigurationScope.MACHINE, ConfigurationScope.WINDOW, ConfigurationScope.RESOURCE, ConfigurationScope.LANGUAGE_OVERRIDABLE, ConfigurationScope.MACHINE_OVERRIDABLE];
30 > export const LOCAL_MACHINE_PROFILE_SCOPES = [ConfigurationScope.WINDOW, ConfigurationScope.RESOURCE, ConfigurationScope.LANGUAGE_OVERRIDABLE];
31 > export const LOCAL_MACHINE_SCOPES = [ConfigurationScope.APPLICATION, ...LOCAL_MACHINE_PROFILE_SCOPES];
32 > export const REMOTE_MACHINE_SCOPES = [ConfigurationScope.MACHINE, ConfigurationScope.APPLICATION_MACHINE, ConfigurationScope.WINDOW, ConfigurationScope.RESOURCE, ConfigurationScope.LANGUAGE_OVERRIDABLE, ConfigurationScope.MACHINE_OVERRIDABLE];
33 > export const WORKSPACE_SCOPES = [ConfigurationScope.WINDOW, ConfigurationScope.RESOURCE, ConfigurationScope.LANGUAGE_OVERRIDABLE, ConfigurationScope.MACHINE_OVERRIDABLE];
34 > export const FOLDER_SCOPES = [ConfigurationScope.RESOURCE, ConfigurationScope.LANGUAGE_OVERRIDABLE, ConfigurationScope.MACHINE_OVERRIDABLE];
35 >
36 > export const TASKS_CONFIGURATION_KEY = 'tasks';
37 > export const LAUNCH_CONFIGURATION_KEY = 'launch';
38 > export const MCP_CONFIGURATION_KEY = 'mcp';
39 >
40 > export const WORKSPACE_STANDALONE_CONFIGURATIONS = Object.create(null);
41 > WORKSPACE_STANDALONE_CONFIGURATIONS[TASKS_CONFIGURATION_KEY] = `${FOLDER_CONFIG_FOLDER_NAME}/${TASKS_CONFIGURATION_KEY}.json`;
42 > WORKSPACE_STANDALONE_CONFIGURATIONS[LAUNCH_CONFIGURATION_KEY] = `${FOLDER_CONFIG_FOLDER_NAME}/${LAUNCH_CONFIGURATION_KEY}.json`;
43 > WORKSPACE_STANDALONE_CONFIGURATIONS[MCP_CONFIGURATION_KEY] = `${FOLDER_CONFIG_FOLDER_NAME}/${MCP_CONFIGURATION_KEY}.json`;
44 > export const USER_STANDALONE_CONFIGURATIONS = Object.create(null);
45 > USER_STANDALONE_CONFIGURATIONS[TASKS_CONFIGURATION_KEY] = `${TASKS_CONFIGURATION_KEY}.json`;
46 > USER_STANDALONE_CONFIGURATIONS[MCP_CONFIGURATION_KEY] = `${MCP_CONFIGURATION_KEY}.json`;
47 >
48 > export type ConfigurationKey = { type: 'defaults' | 'user' | 'workspaces' | 'folder'; key: string };
49 >
50 > export interface IConfigurationCache {
51 >
52 > needsCaching(resource: URI): boolean;
53 > read(key: ConfigurationKey): Promise<string>;
54 > write(key: ConfigurationKey, content: string): Promise<void>;
55 > remove(key: ConfigurationKey): Promise<void>;
56 >
57 > }
58 >
59 > export type RestrictedSettings = {
60 > default: ReadonlyArray<string>;
61 > application?: ReadonlyArray<string>;
62 > userLocal?: ReadonlyArray<string>;
63 > userRemote?: ReadonlyArray<string>;
64 > workspace?: ReadonlyArray<string>;
65 > workspaceFolder?: ResourceMap<ReadonlyArray<string>>;
66 > };
67 >
68 > export const IWorkbenchConfigurationService = refineServiceDecorator<IConfigurationService, IWorkbenchConfigurationService>(IConfigurationService);
69 > export interface IWorkbenchConfigurationService extends IConfigurationService {
70 > /**
71 > * Restricted settings defined in each configuration target
72 > */
73 > readonly restrictedSettings: RestrictedSettings;
74 >
75 > /**
76 > * Event that triggers when the restricted settings changes
77 > */
78 > readonly onDidChangeRestrictedSettings: Event<RestrictedSettings>;
79 >
80 > /**
81 > * A promise that resolves when the remote configuration is loaded in a remote window.
82 > * The promise is resolved immediately if the window is not remote.
83 > */
84 > whenRemoteConfigurationLoaded(): Promise<void>;
85 >
86 > /**
87 > * Initialize configuration service for the given workspace
88 > * @param arg workspace Identifier
89 > */
90 > initialize(arg: IAnyWorkspaceIdentifier): Promise<void>;
91 >
92 > /**
93 > * Returns true if the setting can be applied for all profiles otherwise false.
94 > * @param setting
95 > */
96 > isSettingAppliedForAllProfiles(setting: string): boolean;
97 > }
98 >
99 > export const TASKS_DEFAULT = '{\n\t\"version\": \"2.0.0\",\n\t\"tasks\": []\n}';
100 >
101 > export const APPLY_ALL_PROFILES_SETTING = 'workbench.settings.applyToAllProfiles';
src/vs/base/common/cancellation.ts 100 covered LOC · 23 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- cancellation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from './event.js';
7 > import { DisposableStore, IDisposable } from './lifecycle.js';
8 >
9 > export interface CancellationToken {
10 >
11 > /**
12 > * A flag signalling is cancellation has been requested.
13 > */
14 > readonly isCancellationRequested: boolean;
15 >
16 > /**
17 > * An event which fires when cancellation is requested. This event
18 > * only ever fires `once` as cancellation can only happen once. Listeners
19 > * that are registered after cancellation will be called (next event loop run),
20 > * but also only once.
21 > *
22 > * @event
23 > */
24 > readonly onCancellationRequested: (listener: (e: void) => unknown, thisArgs?: unknown, disposables?: IDisposable[]) => IDisposable;
25 > }
26 >
27 > const shortcutEvent: Event<void> = Object.freeze(function (callback, context?): IDisposable {
28 const handle = setTimeout(callback.bind(context), 0);
29 return { dispose() { clearTimeout(handle); } };
30 });
32 > export namespace CancellationToken {
33 >
34 > export function isCancellationToken(thing: unknown): thing is CancellationToken {
35 if (thing === CancellationToken.None || thing === CancellationToken.Cancelled) {
36 return true;
45 && typeof (thing as CancellationToken).onCancellationRequested === 'function';
46 }
48 >
49 > export const None = Object.freeze<CancellationToken>({
50 > isCancellationRequested: false,
51 > onCancellationRequested: Event.None
52 > });
53 >
54 > export const Cancelled = Object.freeze<CancellationToken>({
55 > isCancellationRequested: true,
56 > onCancellationRequested: shortcutEvent
57 > });
58 > }
59 >
60 class MutableToken implements CancellationToken {
61
62 private _isCancelled: boolean = false;
63 private _emitter: Emitter<void> | null = null;
65 > public cancel() {
66 if (!this._isCancelled) {
67 this._isCancelled = true;
72 }
73 }
75 > get isCancellationRequested(): boolean {
76 return this._isCancelled;
77 }
79 > get onCancellationRequested(): Event<void> {
80 if (this._isCancelled) {
81 return shortcutEvent;
86 return this._emitter.event;
87 }
89 > public dispose(): void {
90 if (this._emitter) {
91 this._emitter.dispose();
93 }
94 }
96 >
97 > export class CancellationTokenSource {
98 >
99 > private _token?: CancellationToken = undefined;
100 > private _parentListener?: IDisposable = undefined;
101 >
102 > constructor(parent?: CancellationToken) {
103 > this._parentListener = parent && parent.onCancellationRequested(this.cancel, this); cancellation.ts
104 > }
106 > get token(): CancellationToken {
107 if (!this._token) {
108 // be lazy and create the token only when
112 return this._token;
113 }
115 > cancel(): void {
116 > if (!this._token) { cancellation.ts
117 > // save an object by returning the default cancellation.ts
118 > // cancelled token when cancellation happens
119 > // before someone asks for the token
120 > this._token = CancellationToken.Cancelled;
121 >
122 > } else if (this._token instanceof MutableToken) { cancellation.ts
123 // actually cancel
124 this._token.cancel();
125 }
126 > } cancellation.ts
128 > dispose(cancel: boolean = false): void {
129 if (cancel) {
130 this.cancel();
140 }
141 }
142 > } cancellation.ts
143 >
144 > export function cancelOnDispose(store: DisposableStore): CancellationToken {
145 const source = new CancellationTokenSource();
146 store.add({ dispose() { source.cancel(); } });
147 return source.token;
148 }
150 > /**
151 > * A pool that aggregates multiple cancellation tokens. The pool's own token
152 > * (accessible via `pool.token`) is cancelled only after every token added
153 > * to the pool has been cancelled. Adding tokens after the pool token has
154 > * been cancelled has no effect.
155 > */
156 > export class CancellationTokenPool {
157
158 private readonly _source = new CancellationTokenSource();
162 private _cancelled: number = 0;
163 private _isDone: boolean = false;
165 > get token(): CancellationToken {
166 return this._source.token;
167 }
169 > /**
170 > * Add a token to the pool. If the token is already cancelled it is counted
171 > * immediately. Tokens added after the pool token has been cancelled are ignored.
172 > */
173 > add(token: CancellationToken): void {
174 if (this._isDone) {
175 return;
191 this._listeners.add(d);
192 }
194 > private _check(): void {
195 if (!this._isDone && this._total > 0 && this._total === this._cancelled) {
196 this._isDone = true;
src/vs/platform/jsonschemas/common/jsonContributionRegistry.ts 100 covered LOC · 9 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- jsonContributionRegistry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from '../../../base/common/event.js';
7 > import { getCompressedContent, IJSONSchema } from '../../../base/common/jsonSchema.js';
8 > import { Disposable, DisposableStore, IDisposable, toDisposable } from '../../../base/common/lifecycle.js';
9 > import * as platform from '../../registry/common/platform.js';
10 >
11 > export const Extensions = {
12 > JSONContribution: 'base.contributions.json'
13 > };
14 >
15 > export interface ISchemaContributions {
16 > schemas: { [id: string]: IJSONSchema };
17 > }
18 >
19 > export interface IJSONContributionRegistry {
20 >
21 > readonly onDidChangeSchema: Event<string>;
22 > readonly onDidChangeSchemaAssociations: Event<void>;
23 >
24 > /**
25 > * Register a schema to the registry.
26 > */
27 > registerSchema(uri: string, unresolvedSchemaContent: IJSONSchema, store?: DisposableStore): void;
28 >
29 > registerSchemaAssociation(uri: string, glob: string): IDisposable;
30 >
31 > /**
32 > * Notifies all listeners that the content of the given schema has changed.
33 > * @param uri The id of the schema
34 > */
35 > notifySchemaChanged(uri: string): void;
36 >
37 > /**
38 > * Get all schemas
39 > */
40 > getSchemaContributions(): ISchemaContributions;
41 >
42 > getSchemaAssociations(): { [uri: string]: string[] };
43 >
44 > /**
45 > * Gets the (compressed) content of the schema with the given schema ID (if any)
46 > * @param uri The id of the schema
47 > */
48 > getSchemaContent(uri: string): string | undefined;
49 >
50 > /**
51 > * Returns true if there's a schema that matches the given schema ID
52 > * @param uri The id of the schema
53 > */
54 > hasSchemaContent(uri: string): boolean;
55 > }
56 >
57 >
58 >
59 > function normalizeId(id: string) {
60 > if (id.length > 0 && id.charAt(id.length - 1) === '#') {
61 return id.substring(0, id.length - 1);
62 }
63 > return id; jsonContributionRegistry.ts
64 > }
65 >
66 >
67 >
68 > class JSONContributionRegistry extends Disposable implements IJSONContributionRegistry {
69 >
70 > private readonly schemasById: { [id: string]: IJSONSchema } = {};
71 > private readonly schemaAssociations: { [uri: string]: string[] } = {};
72 >
73 > private readonly _onDidChangeSchema = this._register(new Emitter<string>());
74 > readonly onDidChangeSchema: Event<string> = this._onDidChangeSchema.event;
75 >
76 > private readonly _onDidChangeSchemaAssociations = this._register(new Emitter<void>());
77 > readonly onDidChangeSchemaAssociations: Event<void> = this._onDidChangeSchemaAssociations.event;
78 >
79 > public registerSchema(uri: string, unresolvedSchemaContent: IJSONSchema, store?: DisposableStore): void {
80 > const normalizedUri = normalizeId(uri);
81 > this.schemasById[normalizedUri] = unresolvedSchemaContent;
82 > this._onDidChangeSchema.fire(uri);
83 >
84 > if (store) {
85 store.add(toDisposable(() => {
86 delete this.schemasById[normalizedUri];
88 }));
89 }
91 >
92 > public registerSchemaAssociation(uri: string, glob: string): IDisposable {
93 const normalizedUri = normalizeId(uri);
94 if (!this.schemaAssociations[normalizedUri]) {
114 });
115 }
117 > public notifySchemaChanged(uri: string): void {
118 this._onDidChangeSchema.fire(uri);
119 }
121 > public getSchemaContributions(): ISchemaContributions {
122 return {
123 schemas: this.schemasById,
124 };
125 }
127 > public getSchemaContent(uri: string): string | undefined {
128 const schema = this.schemasById[uri];
129 return schema ? getCompressedContent(schema) : undefined;
130 }
132 > public hasSchemaContent(uri: string): boolean {
133 return !!this.schemasById[uri];
134 }
136 > public getSchemaAssociations(): { [uri: string]: string[] } {
137 return this.schemaAssociations;
138 }
140 > }
141 >
142 > const jsonContributionRegistry = new JSONContributionRegistry();
143 > platform.Registry.add(Extensions.JSONContribution, jsonContributionRegistry);
src/vs/platform/secrets/common/secrets.ts 99 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- secrets.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { SequencerByKey } from '../../../base/common/async.js';
7 > import { IEncryptionService } from '../../encryption/common/encryptionService.js';
8 > import { createDecorator } from '../../instantiation/common/instantiation.js';
9 > import { IStorageService, IStorageValueChangeEvent, InMemoryStorageService, StorageScope, StorageTarget } from '../../storage/common/storage.js';
10 > import { Emitter, Event } from '../../../base/common/event.js';
11 > import { ILogService } from '../../log/common/log.js';
12 > import { Disposable, DisposableStore } from '../../../base/common/lifecycle.js';
13 > import { Lazy } from '../../../base/common/lazy.js';
14 > import { isWindows } from '../../../base/common/platform.js';
15 >
16 > /**
17 > * The storage key prefix used for all secrets.
18 > */
19 > export const SECRET_STORAGE_PREFIX = 'secret://';
20 >
21 > /**
22 > * Builds the full storage key for a secret.
23 > */
24 > export function secretStorageKey(key: string): string {
25 return `${SECRET_STORAGE_PREFIX}${key}`;
26 }
27 > secrets.ts
28 > /**
29 > * Reads an encrypted secret from storage and decrypts it.
30 > * @param key The secret key (without the `secret://` prefix).
31 > * @param storageGet A function that reads the encrypted value from storage given a full storage key.
32 > * @param decrypt A function that decrypts the encrypted value.
33 > * @param logService Optional logger for trace output.
34 > */
35 export async function readEncryptedSecret(
36 key: string,
51 return result;
52 }
53 > secrets.ts
54 > /**
55 > * Encrypts a secret value and writes it to storage.
56 > * @param key The secret key (without the `secret://` prefix).
57 > * @param value The plaintext secret value.
58 > * @param storageSet A function that writes the encrypted value to storage given a full storage key.
59 > * @param encrypt A function that encrypts the plaintext value.
60 > * @param logService Optional logger for trace output.
61 > */
62 export async function writeEncryptedSecret(
63 key: string,
74 logService?.trace('[secrets] stored encrypted secret for key:', fullKey);
75 }
76 > secrets.ts
77 > /**
78 > * Secret keys that should be shared between the VS Code app and the agents app.
79 > * When the agents app starts and doesn't have these secrets, it requests them
80 > * from VS Code via crossAppIPC.
81 > */
82 > export const CROSS_APP_SHARED_SECRET_KEYS: readonly string[] = [
83 > '{"extensionId":"vscode.github-authentication","key":"github.auth"}',
84 > ];
85 >
86 > export const ISecretStorageService = createDecorator<ISecretStorageService>('secretStorageService');
87 >
88 > export interface ISecretStorageProvider {
89 > type: 'in-memory' | 'persisted' | 'unknown';
90 > get(key: string): Promise<string | undefined>;
91 > set(key: string, value: string): Promise<void>;
92 > delete(key: string): Promise<void>;
93 > keys?(): Promise<string[]>;
94 > }
95 >
96 > export interface ISecretStorageService extends ISecretStorageProvider {
97 > readonly _serviceBrand: undefined;
98 > readonly onDidChangeSecret: Event<string>;
99 > }
100 >
101 > export class BaseSecretStorageService extends Disposable implements ISecretStorageService {
102 > declare readonly _serviceBrand: undefined;
103 >
104 > protected readonly onDidChangeSecretEmitter = this._register(new Emitter<string>());
105 > readonly onDidChangeSecret: Event<string> = this.onDidChangeSecretEmitter.event;
106 >
107 > protected readonly _sequencer = new SequencerByKey<string>();
108 >
109 > private _type: 'in-memory' | 'persisted' | 'unknown' = 'unknown';
110 >
111 > private readonly _onDidChangeValueDisposable = this._register(new DisposableStore());
112 >
113 > constructor(
114 private readonly _useInMemoryStorage: boolean,
115 @IStorageService private _storageService: IStorageService,
133
134 private _lazyStorageService: Lazy<Promise<IStorageService>> = new Lazy(() => this.initialize());
135 > protected get resolvedStorageService() { secrets.ts
136 return this._lazyStorageService.value;
137 }
138 > secrets.ts
139 > get(key: string): Promise<string | undefined> {
140 return this._sequencer.queue(key, async () => {
141 const storageService = await this.resolvedStorageService;
156 });
157 }
158 > secrets.ts
159 > set(key: string, value: string): Promise<void> {
160 return this._sequencer.queue(key, async () => {
161 const storageService = await this.resolvedStorageService;
176 });
177 }
178 > secrets.ts
179 > delete(key: string): Promise<void> {
180 return this._sequencer.queue(key, async () => {
181 const storageService = await this.resolvedStorageService;
188 });
189 }
190 > secrets.ts
191 > keys(): Promise<string[]> {
192 return this._sequencer.queue('__keys__', async () => {
193 const storageService = await this.resolvedStorageService;
198 });
199 }
200 > secrets.ts
201 > private getValueFromStorage(key: string, fullKey: string, storageService: IStorageService): string | undefined {
202 if (this.useSharedStorage(key)) {
203 this._logService.trace(`[SecretStorageService] Fetching value for cross-app shared secret: ${fullKey}`);
206 return storageService.get(fullKey, StorageScope.APPLICATION);
207 }
208 > secrets.ts
209 > private setValueInStorage(key: string, fullKey: string, value: string, storageService: IStorageService): void {
210 if (this.useSharedStorage(key)) {
211 this._logService.trace(`[SecretStorageService] Setting value for cross-app shared secret: ${fullKey}`);
215 storageService.store(fullKey, value, StorageScope.APPLICATION, StorageTarget.MACHINE);
216 }
217 > secrets.ts
218 > private async initialize(): Promise<IStorageService> {
219 let storageService;
220 if (!this._useInMemoryStorage && await this._encryptionService.isEncryptionAvailable()) {
241 return storageService;
242 }
243 > secrets.ts
244 > protected reinitialize(): void {
245 this._lazyStorageService = new Lazy(() => this.initialize());
246 }
247 > secrets.ts
248 > private onDidChangeValue(key: string): void {
249 if (!key.startsWith(SECRET_STORAGE_PREFIX)) {
250 return;
256 this.onDidChangeSecretEmitter.fire(secretKey);
257 }
258 > } secrets.ts
src/vs/workbench/services/remote/common/remoteAgentService.ts 99 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- remoteAgentService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
7 > import { RemoteAgentConnectionContext, IRemoteAgentEnvironment } from '../../../../platform/remote/common/remoteAgentEnvironment.js';
8 > import { IChannel, IServerChannel } from '../../../../base/parts/ipc/common/ipc.js';
9 > import { IDiagnosticInfoOptions, IDiagnosticInfo } from '../../../../platform/diagnostics/common/diagnostics.js';
10 > import { Event } from '../../../../base/common/event.js';
11 > import { PersistentConnectionEvent } from '../../../../platform/remote/common/remoteAgentConnection.js';
12 > import { ITelemetryData, TelemetryLevel } from '../../../../platform/telemetry/common/telemetry.js';
13 > import { timeout } from '../../../../base/common/async.js';
14 >
15 > export const IRemoteAgentService = createDecorator<IRemoteAgentService>('remoteAgentService');
16 >
17 > export interface IRemoteAgentService {
18 > readonly _serviceBrand: undefined;
19 >
20 > getConnection(): IRemoteAgentConnection | null;
21 > /**
22 > * Get the remote environment. In case of an error, returns `null`.
23 > */
24 > getEnvironment(): Promise<IRemoteAgentEnvironment | null>;
25 > /**
26 > * Get the remote environment. Can return an error.
27 > */
28 > getRawEnvironment(): Promise<IRemoteAgentEnvironment | null>;
29 > /**
30 > * Get exit information for a remote extension host.
31 > */
32 > getExtensionHostExitInfo(reconnectionToken: string): Promise<IExtensionHostExitInfo | null>;
33 >
34 > /**
35 > * Gets the round trip time from the remote extension host. Note that this
36 > * may be delayed if the extension host is busy.
37 > */
38 > getRoundTripTime(): Promise<number | undefined>;
39 >
40 > /**
41 > * Gracefully ends the current connection, if any.
42 > */
43 > endConnection(): Promise<void>;
44 >
45 > getDiagnosticInfo(options: IDiagnosticInfoOptions): Promise<IDiagnosticInfo | undefined>;
46 > updateTelemetryLevel(telemetryLevel: TelemetryLevel): Promise<void>;
47 > logTelemetry(eventName: string, data?: ITelemetryData): Promise<void>;
48 > flushTelemetry(): Promise<void>;
49 > }
50 >
51 > export interface IExtensionHostExitInfo {
52 > code: number;
53 > signal: string;
54 > }
55 >
56 > export interface IRemoteAgentConnection {
57 > readonly remoteAuthority: string;
58 >
59 > readonly onReconnecting: Event<void>;
60 > readonly onDidStateChange: Event<PersistentConnectionEvent>;
61 >
62 > end(): Promise<void>;
63 > dispose(): void;
64 > getChannel<T extends IChannel>(channelName: string): T;
65 > withChannel<T extends IChannel, R>(channelName: string, callback: (channel: T) => Promise<R>): Promise<R>;
66 > registerChannel<T extends IServerChannel<RemoteAgentConnectionContext>>(channelName: string, channel: T): void;
67 > getInitialConnectionTimeMs(): Promise<number>;
68 > updateGraceTime(graceTime: number): void;
69 > }
70 >
71 > export interface IRemoteConnectionLatencyMeasurement {
72 >
73 > readonly initial: number | undefined;
74 > readonly current: number;
75 > readonly average: number;
76 >
77 > readonly high: boolean;
78 > }
79 >
80 > export const remoteConnectionLatencyMeasurer = new class {
81 >
82 > readonly maxSampleCount = 5;
83 > readonly sampleDelay = 2000;
84 >
85 > readonly initial: number[] = [];
86 > readonly maxInitialCount = 3;
87 >
88 > readonly average: number[] = [];
89 > readonly maxAverageCount = 100;
90 >
91 > readonly highLatencyMultiple = 2;
92 > readonly highLatencyMinThreshold = 500;
93 > readonly highLatencyMaxThreshold = 1500;
94 >
95 > lastMeasurement: IRemoteConnectionLatencyMeasurement | undefined = undefined;
96 > get latency() { return this.lastMeasurement; }
97 >
98 > async measure(remoteAgentService: IRemoteAgentService): Promise<IRemoteConnectionLatencyMeasurement | undefined> {
99 let currentLatency = Infinity;
100
src/vs/base/common/observableInternal/observables/derived.ts 98 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- derived.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IReader, ITransaction, ISettableObservable, IObservableWithChange } from '../base.js';
7 > import { IChangeTracker } from '../changeTracker.js';
8 > import { DisposableStore, EqualityComparer, IDisposable, strictEquals } from '../commonFacade/deps.js';
9 > import { DebugLocation } from '../debugLocation.js';
10 > import { DebugOwner, DebugNameData, IDebugNameData } from '../debugName.js';
11 > import { _setDerivedOpts } from './baseObservable.js';
12 > import { IDerivedReader, Derived, DerivedWithSetter } from './derivedImpl.js';
13 >
14 > /**
15 > * Creates an observable that is derived from other observables.
16 > * The value is only recomputed when absolutely needed.
17 > *
18 > * {@link computeFn} should start with a JS Doc using `@description` to name the derived.
19 > */
20 > export function derived<T, TChange = void>(computeFn: (reader: IDerivedReader<TChange>, debugLocation?: DebugLocation) => T): IObservableWithChange<T, TChange>;
21 > export function derived<T, TChange = void>(owner: DebugOwner, computeFn: (reader: IDerivedReader<TChange>) => T, debugLocation?: DebugLocation): IObservableWithChange<T, TChange>;
22 > export function derived<T, TChange = void>(
23 > computeFnOrOwner: ((reader: IDerivedReader<TChange>) => T) | DebugOwner, derived.ts
24 > computeFn?: ((reader: IDerivedReader<TChange>) => T) | undefined,
25 > debugLocation = DebugLocation.ofCaller()
26 > ): IObservable<T> {
27 > if (computeFn !== undefined) {
28 > return new Derived( derived.ts
29 > new DebugNameData(computeFnOrOwner, undefined, computeFn),
30 > computeFn,
31 > undefined,
32 > undefined,
33 > strictEquals,
34 > debugLocation,
35 > );
36 > }
37 > return new Derived( derived.ts
38 > // eslint-disable-next-line local/code-no-any-casts
39 > new DebugNameData(undefined, undefined, computeFnOrOwner as any),
40 > // eslint-disable-next-line local/code-no-any-casts
41 > computeFnOrOwner as any,
42 > undefined,
43 > undefined,
44 > strictEquals,
45 > debugLocation,
46 > );
47 > }
48 > derived.ts
49 > export function derivedWithSetter<T>(owner: DebugOwner | undefined, computeFn: (reader: IReader) => T, setter: (value: T, transaction: ITransaction | undefined) => void, debugLocation = DebugLocation.ofCaller()): ISettableObservable<T> {
50 return new DerivedWithSetter(
51 new DebugNameData(owner, undefined, computeFn),
58 );
59 }
60 > derived.ts
61 > export function derivedOpts<T>(
62 > options: IDebugNameData & { derived.ts
63 > equalsFn?: EqualityComparer<T>;
64 > onLastObserverRemoved?: (() => void);
65 > },
66 > computeFn: (reader: IReader) => T,
67 > debugLocation = DebugLocation.ofCaller()
68 > ): IObservable<T> {
69 > return new Derived(
70 > new DebugNameData(options.owner, options.debugName, options.debugReferenceFn),
71 > computeFn,
72 > undefined,
73 > options.onLastObserverRemoved,
74 > options.equalsFn ?? strictEquals,
75 > debugLocation
76 > );
77 > }
78 > _setDerivedOpts(derivedOpts); derived.ts
79 >
80 > /**
81 > * Represents an observable that is derived from other observables.
82 > * The value is only recomputed when absolutely needed.
83 > *
84 > * {@link computeFn} should start with a JS Doc using `@description` to name the derived.
85 > *
86 > * Use `createEmptyChangeSummary` to create a "change summary" that can collect the changes.
87 > * Use `handleChange` to add a reported change to the change summary.
88 > * The compute function is given the last change summary.
89 > * The change summary is discarded after the compute function was called.
90 > *
91 > * @see derived
92 > */
93 > export function derivedHandleChanges<T, TDelta, TChangeSummary>(
94 options: IDebugNameData & {
95 changeTracker: IChangeTracker<TChangeSummary>;
108 );
109 }
110 > derived.ts
111 > /**
112 > * @deprecated Use `derived(reader => { reader.store.add(...) })` instead!
113 > */
114 > export function derivedWithStore<T>(computeFn: (reader: IReader, store: DisposableStore) => T): IObservable<T>;
115 >
116 > /**
117 > * @deprecated Use `derived(reader => { reader.store.add(...) })` instead!
118 > */
119 > export function derivedWithStore<T>(owner: DebugOwner, computeFn: (reader: IReader, store: DisposableStore) => T): IObservable<T>;
120 > export function derivedWithStore<T>(computeFnOrOwner: ((reader: IReader, store: DisposableStore) => T) | DebugOwner, computeFnOrUndefined?: ((reader: IReader, store: DisposableStore) => T), debugLocation = DebugLocation.ofCaller()): IObservable<T> {
121 let computeFn: (reader: IReader, store: DisposableStore) => T;
122 let owner: DebugOwner;
151 );
152 }
153 > derived.ts
154 > export function derivedDisposable<T extends IDisposable | undefined>(computeFn: (reader: IReader) => T): IObservable<T>;
155 > export function derivedDisposable<T extends IDisposable | undefined>(owner: DebugOwner, computeFn: (reader: IReader) => T): IObservable<T>;
156 > export function derivedDisposable<T extends IDisposable | undefined>(computeFnOrOwner: ((reader: IReader) => T) | DebugOwner, computeFnOrUndefined?: ((reader: IReader) => T), debugLocation = DebugLocation.ofCaller()): IObservable<T> {
157 let computeFn: (reader: IReader) => T;
158 let owner: DebugOwner;
src/vs/workbench/contrib/chat/test/common/languageModels.ts 98 covered LOC · 29 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- languageModels.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
7 > import { IStringDictionary } from '../../../../../base/common/collections.js';
8 > import { Event } from '../../../../../base/common/event.js';
9 > import { Disposable, IDisposable } from '../../../../../base/common/lifecycle.js';
10 > import { observableValue } from '../../../../../base/common/observable.js';
11 > import { IAction } from '../../../../../base/common/actions.js';
12 > import { ExtensionIdentifier } from '../../../../../platform/extensions/common/extensions.js';
13 > import { IChatMessage, IModelsControlManifest, ILanguageModelChatMetadata, ILanguageModelChatMetadataAndIdentifier, ILanguageModelChatProvider, ILanguageModelChatRequestOptions, ILanguageModelChatResponse, ILanguageModelChatSelector, ILanguageModelProviderDescriptor, ILanguageModelsGroup, ILanguageModelsService, IUserFriendlyLanguageModel } from '../../common/languageModels.js';
14 > import { ILanguageModelsProviderGroup } from '../../common/languageModelsConfiguration.js';
15 >
16 > export class NullLanguageModelsService implements ILanguageModelsService {
17 > _serviceBrand: undefined; languageModels.ts
18 >
19 > registerLanguageModelProvider(vendor: string, provider: ILanguageModelChatProvider): IDisposable {
20 > return Disposable.None;
21 > }
22 >
23 > deltaLanguageModelChatProviderDescriptors(added: IUserFriendlyLanguageModel[], removed: IUserFriendlyLanguageModel[]): void {
24 }
26 > onDidChangeLanguageModels = Event.None;
27 > onDidChangeLanguageModelVendors = Event.None;
28 > onDidChangeModelsControlManifest = Event.None;
29 > onDidChangePinnedModels = Event.None;
30 > onDidChangeModelVisibility = Event.None;
31 >
32 > getVendors(): ILanguageModelProviderDescriptor[] {
33 return [];
34 }
36 > getLanguageModelIds(): string[] {
37 return [];
38 }
40 > lookupLanguageModel(identifier: string): ILanguageModelChatMetadata | undefined {
41 return undefined;
42 }
44 > lookupLanguageModelByQualifiedName(qualifiedName: string) {
45 return undefined;
46 }
48 > getLanguageModels(): ILanguageModelChatMetadataAndIdentifier[] {
49 return [];
50 }
52 > setContributedSessionModels(): void {
53 return;
54 }
56 > clearContributedSessionModels(): void {
57 return;
58 }
60 > getLanguageModelGroups(vendor: string): ILanguageModelsGroup[] {
61 return [];
62 }
64 > hasResolvedVendor(vendor: string): boolean {
65 return false;
66 }
68 > async selectLanguageModels(selector: ILanguageModelChatSelector): Promise<string[]> {
69 return [];
70 }
72 > sendChatRequest(identifier: string, from: ExtensionIdentifier | undefined, messages: IChatMessage[], options: ILanguageModelChatRequestOptions, token: CancellationToken): Promise<ILanguageModelChatResponse> {
73 throw new Error('Method not implemented.');
74 }
76 > computeTokenLength(identifier: string, message: string | IChatMessage, token: CancellationToken): Promise<number> {
77 throw new Error('Method not implemented.');
78 }
80 > getModelConfiguration(_modelId: string): IStringDictionary<unknown> | undefined {
81 return undefined;
82 }
84 > async setModelConfiguration(_modelId: string, _values: IStringDictionary<unknown>): Promise<void> {
85 }
87 > getModelConfigurationActions(_modelId: string): IAction[] {
88 return [];
89 }
91 > async configureLanguageModelsProviderGroup(vendorId: string, name?: string): Promise<void> {
92
93 }
95 > async renameLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void> {
96 }
98 > async updateLanguageModelsProviderGroupApiKey(vendorId: string, providerGroupName: string): Promise<void> {
99 }
101 > async addLanguageModelsProviderGroupModel(vendorId: string, providerGroupName: string): Promise<void> {
102 }
104 > async openLanguageModelsProviderGroupSettings(vendorId: string, providerGroupName: string): Promise<void> {
105 }
107 > async configureModel(_modelId: string): Promise<void> {
108 }
110 > async addLanguageModelsProviderGroup(name: string, vendorId: string, configuration: IStringDictionary<unknown> | undefined): Promise<void> {
111
112 }
114 > async removeLanguageModelsProviderGroup(vendorId: string, providerGroupName: string): Promise<void> {
115 }
117 > async migrateLanguageModelsProviderGroup(languageModelsProviderGroup: ILanguageModelsProviderGroup): Promise<void> { }
118 >
119 > getRecentlyUsedModelIds(): string[] {
120 return [];
121 }
123 > addToRecentlyUsedList(): void { }
124 > clearRecentlyUsedList(): void { }
125 >
126 > getPinnedModelIds(): string[] { return []; }
127 > pinModel(_modelIdentifier: string): void { }
128 > unpinModel(_modelIdentifier: string): void { }
129 > isModelPinned(_modelIdentifier: string): boolean { return false; }
130 >
131 > isModelHidden(_modelIdentifier: string): boolean { return false; }
132 > isGroupHidden(_vendor: string, _groupName: string): boolean { return false; }
133 > setModelHidden(_modelIdentifier: string, _hidden: boolean): void { }
134 > setGroupHidden(_vendor: string, _groupName: string, _hidden: boolean): void { }
135 > getHiddenModelIds(): string[] { return []; }
136 >
137 > getModelsControlManifest(): IModelsControlManifest {
138 return { free: {}, paid: {} };
139 }
141 > restrictedChatParticipants = observableValue('restrictedChatParticipants', Object.create(null));
src/vs/base/test/common/virtualScheduling/virtualClock.ts 94 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- virtualClock.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { compareBy, numberComparator, tieBreakComparators } from '../../../common/arrays.js';
7 > import { Emitter } from '../../../common/event.js';
8 > import { IDisposable } from '../../../common/lifecycle.js';
9 > import { Trace } from './trace.js';
10 >
11 > export type VirtualTime = number;
12 >
13 > /** Debug source description for an event. */
14 > export interface EventSource {
15 > toString(): string;
16 > readonly stackTrace?: string;
17 > }
18 >
19 > /**
20 > * A unit of work scheduled at a point in virtual time.
21 > *
22 > * Timer callbacks are events. External completions (e.g. fake fs reads) can
23 > * also be modelled as events whose virtual completion time is chosen by a
24 > * scheduling policy — to the {@link VirtualClock} they are indistinguishable.
25 > */
26 > export interface VirtualEvent {
27 > readonly time: VirtualTime;
28 > readonly source: EventSource;
29 > readonly trace?: Trace;
30 > /**
31 > * Hint for the {@link Embedding}: this event prefers to run on a real
32 > * animation frame (e.g. so DOM measurements after it observe a real
33 > * reflow). Pure-time tests can ignore the hint.
34 > */
35 > readonly preferRealAnimationFrame?: boolean;
36 > run(): void;
37 > }
38 >
39 > interface QueuedEvent extends VirtualEvent { readonly id: number }
40 >
41 > const eventComparator = tieBreakComparators<QueuedEvent>(
42 > compareBy(e => e.time, numberComparator),
43 > compareBy(e => e.id, numberComparator),
44 > );
45 >
46 > /**
47 > * A pure data structure: a virtual clock + a priority queue of events.
48 > *
49 > * The clock has no concept of "real time". It is advanced exclusively by
50 > * {@link runNext}, which sets `now` to the next event's `time` before running
51 > * it. The {@link VirtualTimeProcessor} is the only intended driver, but the
52 > * clock is useful in isolation (e.g. for unit-testing a scheduler or for
53 > * stepping a scenario manually).
54 > */
55 > export class VirtualClock {
56 > private _now: VirtualTime;
57 > private _idCounter = 0;
58 > private readonly _queue = new SimplePriorityQueue<QueuedEvent>(eventComparator);
59 > private readonly _onEventScheduled = new Emitter<VirtualEvent>();
60 >
61 > public readonly onEventScheduled = this._onEventScheduled.event;
62 >
63 > constructor(startTime: VirtualTime = 0) {
64 this._now = startTime;
65 }
67 > get now(): VirtualTime { return this._now; }
68 > get hasEvents(): boolean { return this._queue.length > 0; }
69 >
70 > schedule(event: VirtualEvent): IDisposable {
71 if (event.time < this._now) {
72 throw new Error(`Scheduled time (${event.time}) must be >= now (${this._now}).`);
77 return { dispose: () => this._queue.remove(queued) };
78 }
80 > peekNext(): VirtualEvent | undefined { return this._queue.getMin(); }
81 >
82 > runNext(): VirtualEvent | undefined {
83 const e = this._queue.removeMin();
84 if (e) {
88 return e;
89 }
91 > getEvents(): readonly VirtualEvent[] { return this._queue.toSortedArray(); }
92 > }
93 >
94 > class SimplePriorityQueue<T> {
95 > private _items: T[] = [];
96 > private _sorted = true;
97 >
98 > constructor(private readonly _compare: (a: T, b: T) => number) { }
99 >
100 > get length(): number { return this._items.length; }
101 >
102 > add(value: T): void {
103 this._items.push(value);
104 this._sorted = false;
105 }
107 > remove(value: T): void {
108 const i = this._items.indexOf(value);
109 if (i !== -1) { this._items.splice(i, 1); }
110 }
112 > getMin(): T | undefined { this._ensureSorted(); return this._items[0]; }
113 > removeMin(): T | undefined { this._ensureSorted(); return this._items.shift(); }
114 > toSortedArray(): T[] { this._ensureSorted(); return [...this._items]; }
115 >
116 > private _ensureSorted(): void {
117 if (this._sorted) { return; }
118 this._items.sort(this._compare);
119 this._sorted = true;
120 }
121 > } virtualClock.ts
src/vs/platform/agentHost/common/meta/agentToolCallMeta.ts 92 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentToolCallMeta.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import type { Mutable } from '../../../../base/common/types.js';
7 >
8 > /** Anything carrying a tool call's `_meta` bag (persisted state or wire actions). */
9 > interface IHasToolCallMeta {
10 > readonly _meta?: Record<string, unknown>;
11 > }
12 >
13 > /**
14 > * Well-known typed view over a tool call's open `_meta` bag. Producers and
15 > * consumers agree on these keys here so the two sides can't drift; always read
16 > * the bag through {@link readToolCallMeta}, which validates each field and drops
17 > * wrong-typed values.
18 > */
19 > export interface IToolCallMeta {
20 > /**
21 > * VS Code rendering hint. `terminal` routes the call to the command/output
22 > * renderer, `subagent` to the subagent UI, `search` to the search renderer;
23 > * everything else falls through to the generic invocation renderer. Set by
24 > * the agent adapter, never matched on raw tool name by the renderer.
25 > */
26 > readonly toolKind?: ToolKind;
27 > /** Shell language for a `terminal` tool call (drives syntax highlighting). */
28 > readonly language?: string;
29 > /** Short task description for a `subagent` tool call (e.g. "Find related files"). */
30 > readonly subagentDescription?: string;
31 > /** Agent name for a `subagent` tool call (e.g. "explore"). */
32 > readonly subagentAgentName?: string;
33 > /** Chat URI of the subagent this tool call spawns, stamped by the host (see {@link buildSubagentChatUri}); the resource may not be registered yet. */
34 > readonly subagentChatUri?: string;
35 > /** Raw, pre-stringified tool arguments captured for display/debugging. */
36 > readonly toolArguments?: unknown;
37 > /** Originating MCP server name, when the call came from an MCP server. */
38 > readonly mcpServerName?: string;
39 > /** Originating MCP tool name, when the call came from an MCP server. */
40 > readonly mcpToolName?: string;
41 > /** MCP App render data, when the call exposes an interactive App surface. */
42 > readonly ui?: IToolCallUiMeta;
43 > /**
44 > * Set by the host's side-effect layer when the call was auto-approved
45 > * because of an `autoApprove` session-config setting (rather than an
46 > * explicit user action), so the client can render it as setting-driven.
47 > */
48 > readonly autoApproveBySetting?: boolean;
49 > /** Transient runtime corpus for the local client tool-search invocation. */
50 > readonly toolSearchCandidates?: readonly IToolSearchCandidate[];
51 > }
52 >
53 > /** Minimal metadata needed to embed and rank a deferred tool. */
54 > export interface IToolSearchCandidate {
55 > readonly name: string;
56 > readonly description: string;
57 > }
58 >
59 > /**
60 > * The set of VS Code-recognized tool-call rendering kinds. Add a new value here
61 > * (and teach the renderer to handle it) rather than matching on tool name.
62 > */
63 > export type ToolKind = 'terminal' | 'subagent' | 'search';
64 >
65 > /**
66 > * MCP App render data carried under {@link IToolCallMeta.ui}. Clients gate
67 > * mounting the App webview on both a `resourceUri` and a `channel` being
68 > * present.
69 > */
70 > export interface IToolCallUiMeta {
71 > /** The MCP App's UI resource URI (an `ui://` resource the App renders). */
72 > readonly resourceUri: string;
73 > /** AHP `mcp://` channel the App's sub-RPCs route back through, when ready. */
74 > readonly channel?: string;
75 > }
76 >
77 function isToolKind(value: unknown): value is ToolKind {
78 return value === 'terminal' || value === 'subagent' || value === 'search';
79 }
81 function readToolCallUiMeta(value: unknown): IToolCallUiMeta | undefined {
82 if (!value || typeof value !== 'object' || Array.isArray(value)) {
93 return result;
94 }
96 function readToolSearchCandidates(value: unknown): readonly IToolSearchCandidate[] | undefined {
97 if (!Array.isArray(value)) {
114 return result;
115 }
117 > /**
118 > * Reads the well-known {@link IToolCallMeta} keys from a tool call's `_meta`
119 > * bag, dropping unknown keys and wrong-typed values.
120 > */
121 > export function readToolCallMeta(source: IHasToolCallMeta): IToolCallMeta {
122 const meta = source._meta;
123 if (!meta) {
140 return result;
141 }
143 > /**
144 > * Serializes a typed {@link IToolCallMeta} into the `_meta` record, dropping
145 > * `undefined` entries and returning `undefined` when empty. Build a tool call's
146 > * `_meta` through this so producers stay in lock-step with
147 > * {@link readToolCallMeta}.
148 > */
149 > export function toToolCallMeta(meta: IToolCallMeta): Record<string, unknown> | undefined {
150 const result: Record<string, unknown> = {};
151 for (const [key, value] of Object.entries(meta)) {
src/vs/base/common/extpath.ts 91 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extpath.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from './charCode.js';
7 > import { isAbsolute, join, normalize, posix, sep } from './path.js';
8 > import { isWindows } from './platform.js';
9 > import { equalsIgnoreCase, rtrim, startsWithIgnoreCase } from './strings.js';
10 > import { isNumber } from './types.js';
11 >
12 > export function isPathSeparator(code: number) {
13 return code === CharCode.Slash || code === CharCode.Backslash;
14 }
15 > extpath.ts
16 > /**
17 > * Takes a Windows OS path and changes backward slashes to forward slashes.
18 > * This should only be done for OS paths from Windows (or user provided paths potentially from Windows).
19 > * Using it on a Linux or MaxOS path might change it.
20 > */
21 > export function toSlashes(osPath: string) {
22 return osPath.replace(/[\\/]/g, posix.sep);
23 }
24 > extpath.ts
25 > /**
26 > * Takes a Windows OS path (using backward or forward slashes) and turns it into a posix path:
27 > * - turns backward slashes into forward slashes
28 > * - makes it absolute if it starts with a drive letter
29 > * This should only be done for OS paths from Windows (or user provided paths potentially from Windows).
30 > * Using it on a Linux or MaxOS path might change it.
31 > */
32 > export function toPosixPath(osPath: string) {
33 if (osPath.indexOf('/') === -1) {
34 osPath = toSlashes(osPath);
39 return osPath;
40 }
41 > extpath.ts
42 > /**
43 > * Computes the _root_ this path, like `getRoot('c:\files') === c:\`,
44 > * `getRoot('files:///files/path') === files:///`,
45 > * or `getRoot('\\server\shares\path') === \\server\shares\`
46 > */
47 > export function getRoot(path: string, sep: string = posix.sep): string {
48 if (!path) {
49 return '';
111 return '';
112 }
113 > extpath.ts
114 > /**
115 > * Check if the path follows this pattern: `\\hostname\sharename`.
116 > *
117 > * @see https://msdn.microsoft.com/en-us/library/gg465305.aspx
118 > * @return A boolean indication if the path is a UNC path, on none-windows
119 > * always false.
120 > */
121 > export function isUNC(path: string): boolean {
122 if (!isWindows) {
123 // UNC is a windows concept
162 return true;
163 }
164 > extpath.ts
165 > // Reference: https://en.wikipedia.org/wiki/Filename
166 > const WINDOWS_INVALID_FILE_CHARS = /[\\/:\*\?"<>\|]/g;
167 > const UNIX_INVALID_FILE_CHARS = /[/]/g;
168 > const WINDOWS_FORBIDDEN_NAMES = /^(con|prn|aux|clock\$|nul|lpt[0-9]|com[0-9])(\.(.*?))?$/i;
169 > export function isValidBasename(name: string | null | undefined, isWindowsOS: boolean = isWindows): boolean {
170 const invalidFileChars = isWindowsOS ? WINDOWS_INVALID_FILE_CHARS : UNIX_INVALID_FILE_CHARS;
171
201 return true;
202 }
203 > extpath.ts
204 > /**
205 > * @deprecated please use `IUriIdentityService.extUri.isEqual` instead. If you are
206 > * in a context without services, consider to pass down the `extUri` from the outside
207 > * or use `extUriBiasedIgnorePathCase` if you know what you are doing.
208 > */
209 > export function isEqual(pathA: string, pathB: string, ignoreCase?: boolean): boolean {
210 const identityEquals = (pathA === pathB);
211 if (!ignoreCase || identityEquals) {
219 return equalsIgnoreCase(pathA, pathB);
220 }
221 > extpath.ts
222 > /**
223 > * @deprecated please use `IUriIdentityService.extUri.isEqualOrParent` instead. If
224 > * you are in a context without services, consider to pass down the `extUri` from the
225 > * outside, or use `extUriBiasedIgnorePathCase` if you know what you are doing.
226 > */
227 > export function isEqualOrParent(base: string, parentCandidate: string, ignoreCase?: boolean, forcePosixSemantics = false): boolean {
228 const separator = forcePosixSemantics ? posix.sep : sep;
229
272 return base.indexOf(parentCandidate) === 0;
273 }
274 > extpath.ts
275 > export function isWindowsDriveLetter(char0: number): boolean {
276 return char0 >= CharCode.A && char0 <= CharCode.Z || char0 >= CharCode.a && char0 <= CharCode.z;
277 }
278 > extpath.ts
279 > export function sanitizeFilePath(candidate: string, cwd: string): string {
280
281 // Special case: allow to open a drive letter without trailing backslash
295 return removeTrailingPathSeparator(candidate);
296 }
297 > extpath.ts
298 > export function removeTrailingPathSeparator(candidate: string): string {
299 if (isWindows) {
300 candidate = rtrim(candidate, sep);
316 return candidate;
317 }
318 > extpath.ts
319 > export function isRootOrDriveLetter(path: string): boolean {
320 const pathNormalized = normalize(path);
321
331 return pathNormalized === posix.sep;
332 }
333 > extpath.ts
334 > export function hasDriveLetter(path: string, isWindowsOS: boolean = isWindows): boolean {
335 if (isWindowsOS) {
336 return isWindowsDriveLetter(path.charCodeAt(0)) && path.charCodeAt(1) === CharCode.Colon;
339 return false;
340 }
341 > extpath.ts
342 > export function getDriveLetter(path: string, isWindowsOS: boolean = isWindows): string | undefined {
343 return hasDriveLetter(path, isWindowsOS) ? path[0] : undefined;
344 }
345 > extpath.ts
346 > export function indexOfPath(path: string, candidate: string, ignoreCase?: boolean): number {
347 if (candidate.length > path.length) {
348 return -1;
360 return path.indexOf(candidate);
361 }
362 > extpath.ts
363 > export interface IPathWithLineAndColumn {
364 > path: string;
365 > line?: number;
366 > column?: number;
367 > }
368 >
369 > export function parseLineAndColumnAware(rawPath: string): IPathWithLineAndColumn {
370 const segments = rawPath.split(':'); // C:\file.txt:<line>:<column>
371
395 };
396 }
397 > extpath.ts
398 > const pathChars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
399 > const windowsSafePathFirstChars = 'BDEFGHIJKMOQRSTUVWXYZbdefghijkmoqrstuvwxyz0123456789';
400 >
401 > export function randomPath(parent?: string, prefix?: string, randomLength = 8): string {
402 let suffix = '';
403 for (let i = 0; i < randomLength; i++) {
src/vs/workbench/contrib/chat/common/promptSyntax/computeAutomaticInstructions.ts 91 covered LOC · 13 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- computeAutomaticInstructions.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
7 > import { match, splitGlobAware } from '../../../../../base/common/glob.js';
8 > import { ResourceMap, ResourceSet } from '../../../../../base/common/map.js';
9 > import { Schemas } from '../../../../../base/common/network.js';
10 > import { OperatingSystem } from '../../../../../base/common/platform.js';
11 > import { basename, dirname } from '../../../../../base/common/resources.js';
12 > import { URI } from '../../../../../base/common/uri.js';
13 > import { localize } from '../../../../../nls.js';
14 > import { IConfigurationService } from '../../../../../platform/configuration/common/configuration.js';
15 > import { IFileService } from '../../../../../platform/files/common/files.js';
16 > import { ILabelService } from '../../../../../platform/label/common/label.js';
17 > import { ILogService } from '../../../../../platform/log/common/log.js';
18 > import { IRemoteAgentService } from '../../../../services/remote/common/remoteAgentService.js';
19 > import { ITelemetryService } from '../../../../../platform/telemetry/common/telemetry.js';
20 > import { IWorkspaceContextService } from '../../../../../platform/workspace/common/workspace.js';
21 > import { ChatRequestVariableSet, IChatRequestVariableEntry, isPromptFileVariableEntry, toPromptFileVariableEntry, toPromptTextVariableEntry, PromptFileVariableKind, IPromptTextVariableEntry, ChatRequestToolReferenceEntry, toToolVariableEntry } from '../attachments/chatVariableEntries.js';
22 > import { ILanguageModelToolsService, IToolData, VSCodeToolReference } from '../tools/languageModelToolsService.js';
23 > import { PromptsConfig } from './config/config.js';
24 > import { isInClaudeAgentsFolder, isInClaudeRulesFolder, isPromptOrInstructionsFile } from './config/promptFileLocations.js';
25 > import { ParsedPromptFile } from './promptFileParser.js';
26 > import { AgentInstructionFileType, IAgentSkill, ICustomAgent, IInstructionFile, IPromptsService, matchesSessionType, newInstructionsCollectionEvent, newInstructionsCollectionDebugInfo, type InstructionsCollectionEvent, type InstructionsCollectionDebugInfo } from './service/promptsService.js';
27 > export type { InstructionsCollectionEvent, InstructionsCollectionDebugInfo } from './service/promptsService.js';
28 > export { newInstructionsCollectionEvent, newInstructionsCollectionDebugInfo } from './service/promptsService.js';
29 > import { AGENT_DEBUG_LOG_FILE_LOGGING_ENABLED_SETTING, TROUBLESHOOT_SKILL_PATH } from './promptTypes.js';
30 > import { OffsetRange } from '../../../../../editor/common/core/ranges/offsetRange.js';
31 > import { ChatModeKind } from '../constants.js';
32 > import { UserSelectedTools } from '../participants/chatAgents.js';
33 > import { hash } from '../../../../../base/common/hash.js';
34 > import { IAgentPlugin, IAgentPluginService } from '../plugins/agentPluginService.js';
35 >
36 > export interface InstructionsCollectionResult {
37 > readonly telemetryEvent: InstructionsCollectionEvent;
38 > readonly debugInfo: InstructionsCollectionDebugInfo;
39 > }
40 >
41 > /**
42 > * The result of the most recent {@link ComputeAutomaticInstructions.collect} call.
43 > * Consumed by debug contributions for logging; not sent as telemetry.
44 > */
45 > export let lastInstructionsCollectionResult: InstructionsCollectionResult | undefined;
46 >
47 > type InstructionsCollectionClassification = {
48 > applyingInstructionsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of instructions added via pattern matching.' };
49 > referencedInstructionsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of instructions added via references from other instruction files.' };
50 > agentInstructionsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of agent instructions added (copilot-instructions.md and agents.md).' };
51 > listedInstructionsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of instruction patterns added.' };
52 > totalInstructionsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Total number of instruction entries added to variables.' };
53 > claudeRulesCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of Claude rules files (.claude/rules/) added via pattern matching.' };
54 > claudeMdCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of CLAUDE.md agent instruction files added.' };
55 > claudeAgentsCount: { classification: 'SystemMetaData'; purpose: 'FeatureInsight'; isMeasurement: true; comment: 'Number of Claude agent files (.claude/agents/) listed as subagents.' };
56 > owner: 'digitarald';
57 > comment: 'Tracks automatic instruction collection usage in chat prompt system.';
58 > };
59 >
60 > export class ComputeAutomaticInstructions {
61 >
62 > private _parseResults: ResourceMap<ParsedPromptFile> = new ResourceMap();
63 >
64 > constructor(
65 private readonly _modeKind: ChatModeKind,
66 private readonly _enabledTools: UserSelectedTools | undefined,
79 ) {
80 }
82 > private async _parseInstructionsFile(uri: URI, token: CancellationToken): Promise<ParsedPromptFile | undefined> {
83 if (this._parseResults.has(uri)) {
84 return this._parseResults.get(uri)!;
94
95 }
97 > public async collect(variables: ChatRequestVariableSet, token: CancellationToken): Promise<void> {
98
99 const startTime = performance.now();
125 lastInstructionsCollectionResult = { telemetryEvent, debugInfo };
126 }
128 > private sendTelemetry(telemetryEvent: InstructionsCollectionEvent): void {
129 // Emit telemetry
130 telemetryEvent.totalInstructionsCount = telemetryEvent.agentInstructionsCount + telemetryEvent.referencedInstructionsCount + telemetryEvent.applyingInstructionsCount + telemetryEvent.listedInstructionsCount;
131 this._telemetryService.publicLog2<InstructionsCollectionEvent, InstructionsCollectionClassification>('instructionsCollected', telemetryEvent);
132 }
134 > private async _logSkillLoadedTelemetry(skills: readonly IAgentSkill[]): Promise<void> {
135 type SkillLoadedIntoContextEvent = {
136 skillNameHash: string;
180 }
181 }
183 > /** public for testing */
184 > public async addApplyingInstructions(instructionFiles: readonly IInstructionFile[], context: { files: ResourceSet; instructions: ResourceSet }, variables: ChatRequestVariableSet, telemetryEvent: InstructionsCollectionEvent, debugInfo: InstructionsCollectionDebugInfo, token: CancellationToken): Promise<void> {
185 const includeApplyingInstructions = this._configurationService.getValue(PromptsConfig.INCLUDE_APPLYING_INSTRUCTIONS);
186 if (!includeApplyingInstructions && this._modeKind !== ChatModeKind.Edit) {
237 }
238 }
240 > private _getContext(attachedContext: ChatRequestVariableSet): { files: ResourceSet; instructions: ResourceSet } {
241 const files = new ResourceSet();
242 const instructions = new ResourceSet();
254 return { files, instructions };
255 }
257 > private async _addAgentInstructions(variables: ChatRequestVariableSet, telemetryEvent: InstructionsCollectionEvent, debugInfo: InstructionsCollectionDebugInfo, token: CancellationToken): Promise<void> {
258 const logger = {
259 logInfo: (message: string) => this._logService.trace(`[InstructionsContextComputer] ${message}`)
291 }
292 }
294 > private _matches(files: ResourceSet, applyToPattern: string): { pattern: string; file?: URI } | undefined {
295 const patterns = splitGlobAware(applyToPattern, ',');
296 const patterMatches = (pattern: string): { pattern: string; file?: URI } | undefined => {
328 return undefined;
329 }
331 > private _getTool(referenceName: string): { tool: IToolData; variable: string } | undefined {
332 if (!this._enabledTools) {
333 return undefined;
339 return undefined;
340 }
342 > private async _getCustomizationsIndex(instructionFiles: readonly IInstructionFile[], _existingVariables: ChatRequestVariableSet, telemetryEvent: InstructionsCollectionEvent, debugInfo: InstructionsCollectionDebugInfo, token: CancellationToken): Promise<IPromptTextVariableEntry | undefined> {
343 const readTool = this._getTool('readFile');
344 const runInTerminalTool = this._getTool('runInTerminal');
568 return toPromptTextVariableEntry(content, true, toolReferences);
569 }
571 > private async _addReferencedInstructions(attachedContext: ChatRequestVariableSet, telemetryEvent: InstructionsCollectionEvent, debugInfo: InstructionsCollectionDebugInfo, token: CancellationToken): Promise<void> {
572 const includeReferencedInstructions = this._configurationService.getValue(PromptsConfig.INCLUDE_REFERENCED_INSTRUCTIONS);
573 if (!includeReferencedInstructions && this._modeKind !== ChatModeKind.Edit) {
621 }
622 }
624 >
625 >
626 > export function getFilePath(uri: URI, remoteOS: OperatingSystem | undefined, isRemote = false): string {
627 // When connected to a remote, local file:// URIs must be represented using
628 // the vscode-local scheme so the remote extension host can read them via the
src/vs/platform/keybinding/common/keybindingResolver.ts 90 covered LOC · 18 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- keybindingResolver.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ContextKeyExpression, ContextKeyExprType, IContext, IContextKeyService, implies } from '../../contextkey/common/contextkey.js';
7 > import { ResolvedKeybindingItem } from './resolvedKeybindingItem.js';
8 >
9 > //#region resolution-result
10 >
11 > export const enum ResultKind {
12 > /** No keybinding found this sequence of chords */
13 > NoMatchingKb,
14 >
15 > /** There're several keybindings that have the given sequence of chords as a prefix */
16 > MoreChordsNeeded,
17 >
18 > /** A single keybinding found to be dispatched/invoked */
19 > KbFound
20 > }
21 >
22 > export type ResolutionResult =
23 > | { kind: ResultKind.NoMatchingKb }
24 > | { kind: ResultKind.MoreChordsNeeded }
25 > | { kind: ResultKind.KbFound; commandId: string | null; commandArgs: any; isBubble: boolean };
26 >
27 >
28 > // util definitions to make working with the above types easier within this module:
29 >
30 > export const NoMatchingKb: ResolutionResult = { kind: ResultKind.NoMatchingKb };
31 > const MoreChordsNeeded: ResolutionResult = { kind: ResultKind.MoreChordsNeeded };
32 function KbFound(commandId: string | null, commandArgs: any, isBubble: boolean): ResolutionResult {
33 return { kind: ResultKind.KbFound, commandId, commandArgs, isBubble };
34 }
36 > //#endregion
37 >
38 > /**
39 > * Stores mappings from keybindings to commands and from commands to keybindings.
40 > * Given a sequence of chords, `resolve`s which keybinding it matches
41 > */
42 > export class KeybindingResolver {
43 > private readonly _log: (str: string) => void;
44 > private readonly _defaultKeybindings: ResolvedKeybindingItem[];
45 > private readonly _keybindings: ResolvedKeybindingItem[];
46 > private readonly _defaultBoundCommands: Map</* commandId */ string, boolean>;
47 > private readonly _map: Map</* 1st chord's keypress */ string, ResolvedKeybindingItem[]>;
48 > private readonly _lookupMap: Map</* commandId */ string, ResolvedKeybindingItem[]>;
49 >
50 > constructor(
51 /** built-in and extension-provided keybindings */
52 defaultKeybindings: ResolvedKeybindingItem[],
88 }
89 }
91 > private static _isTargetedForRemoval(defaultKb: ResolvedKeybindingItem, keypress: string[] | null, when: ContextKeyExpression | undefined): boolean {
92 if (keypress) {
93 for (let i = 0; i < keypress.length; i++) {
118
119 }
121 > /**
122 > * Looks for rules containing "-commandId" and removes them.
123 > */
124 > public static handleRemovals(rules: ResolvedKeybindingItem[]): ResolvedKeybindingItem[] {
125 // Do a first pass and construct a hash-map for removals
126 const removals = new Map</* commandId */ string, ResolvedKeybindingItem[]>();
174 return result;
175 }
177 > private _addKeyPress(keypress: string, item: ResolvedKeybindingItem): void {
178
179 const conflicts = this._map.get(keypress);
217 this._addToLookupMap(item);
218 }
220 > private _addToLookupMap(item: ResolvedKeybindingItem): void {
221 if (!item.command) {
222 return;
231 }
232 }
234 > private _removeFromLookupMap(item: ResolvedKeybindingItem): void {
235 if (!item.command) {
236 return;
247 }
248 }
250 > /**
251 > * Returns true if it is provable `a` implies `b`.
252 > */
253 > public static whenIsEntirelyIncluded(a: ContextKeyExpression | null | undefined, b: ContextKeyExpression | null | undefined): boolean {
254 if (!b || b.type === ContextKeyExprType.True) {
255 return true;
261 return implies(a, b);
262 }
264 > public getDefaultBoundCommands(): Map<string, boolean> {
265 return this._defaultBoundCommands;
266 }
268 > public getDefaultKeybindings(): readonly ResolvedKeybindingItem[] {
269 return this._defaultKeybindings;
270 }
272 > public getKeybindings(): readonly ResolvedKeybindingItem[] {
273 return this._keybindings;
274 }
276 > public lookupKeybindings(commandId: string): ResolvedKeybindingItem[] {
277 const items = this._lookupMap.get(commandId);
278 if (typeof items === 'undefined' || items.length === 0) {
288 return result;
289 }
291 > public lookupPrimaryKeybinding(commandId: string, context: IContextKeyService, enforceContextCheck = false): ResolvedKeybindingItem | null {
292 const items = this._lookupMap.get(commandId);
293 if (typeof items === 'undefined' || items.length === 0) {
311 return items[items.length - 1];
312 }
314 > /**
315 > * Looks up a keybinding trigged as a result of pressing a sequence of chords - `[...currentChords, keypress]`
316 > *
317 > * Example: resolving 3 chords pressed sequentially - `cmd+k cmd+p cmd+i`:
318 > * `currentChords = [ 'cmd+k' , 'cmd+p' ]` and `keypress = `cmd+i` - last pressed chord
319 > */
320 > public resolve(context: IContext, currentChords: string[], keypress: string): ResolutionResult {
321
322 const pressedChords = [...currentChords, keypress];
377 return KbFound(result.command, result.commandArgs, result.bubble);
378 }
380 > private _findCommand(context: IContext, matches: ResolvedKeybindingItem[]): ResolvedKeybindingItem | null {
381 for (let i = matches.length - 1; i >= 0; i--) {
382 const k = matches[i];
391 return null;
392 }
394 > private static _contextMatchesRules(context: IContext, rules: ContextKeyExpression | null | undefined): boolean {
395 if (!rules) {
396 return true;
398 return rules.evaluate(context);
399 }
401 >
402 function printWhenExplanation(when: ContextKeyExpression | undefined): string {
403 if (!when) {
406 return `${when.serialize()}`;
407 }
409 function printSourceExplanation(kb: ResolvedKeybindingItem): string {
410 return (
src/vs/editor/common/services/textResourceConfiguration.ts 89 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- textResourceConfiguration.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { URI } from '../../../base/common/uri.js';
8 > import { IPosition } from '../core/position.js';
9 > import { ConfigurationTarget, IConfigurationValue } from '../../../platform/configuration/common/configuration.js';
10 > import { createDecorator } from '../../../platform/instantiation/common/instantiation.js';
11 >
12 > export const ITextResourceConfigurationService = createDecorator<ITextResourceConfigurationService>('textResourceConfigurationService');
13 >
14 > export interface ITextResourceConfigurationChangeEvent {
15 >
16 > /**
17 > * All affected keys. Also includes language overrides and keys changed under language overrides.
18 > */
19 > readonly affectedKeys: ReadonlySet<string>;
20 >
21 > /**
22 > * Returns `true` if the given section has changed for the given resource.
23 > *
24 > * Example: To check if the configuration section has changed for a given resource use `e.affectsConfiguration(resource, section)`.
25 > *
26 > * @param resource Resource for which the configuration has to be checked.
27 > * @param section Section of the configuration
28 > */
29 > affectsConfiguration(resource: URI | undefined, section: string): boolean;
30 > }
31 >
32 > export interface ITextResourceConfigurationService {
33 >
34 > readonly _serviceBrand: undefined;
35 >
36 > /**
37 > * Event that fires when the configuration changes.
38 > */
39 > readonly onDidChangeConfiguration: Event<ITextResourceConfigurationChangeEvent>;
40 >
41 > /**
42 > * Fetches the value of the section for the given resource by applying language overrides.
43 > * Value can be of native type or an object keyed off the section name.
44 > *
45 > * @param resource - Resource for which the configuration has to be fetched.
46 > * @param position - Position in the resource for which configuration has to be fetched.
47 > * @param section - Section of the configuration.
48 > *
49 > */
50 > getValue<T>(resource: URI | undefined, section?: string): T;
51 > getValue<T>(resource: URI | undefined, position?: IPosition, section?: string): T;
52 >
53 > /**
54 > * Inspects the values of the section for the given resource by applying language overrides.
55 > *
56 > * @param resource - Resource for which the configuration has to be fetched.
57 > * @param position - Position in the resource for which configuration has to be fetched.
58 > * @param section - Section of the configuration.
59 > *
60 > */
61 > inspect<T>(resource: URI | undefined, position: IPosition | null, section: string): IConfigurationValue<Readonly<T>>;
62 >
63 > /**
64 > * Update the configuration value for the given resource at the effective location.
65 > *
66 > * - If configurationTarget is not specified, target will be derived by checking where the configuration is defined.
67 > * - If the language overrides for the give resource contains the configuration, then it is updated.
68 > *
69 > * @param resource Resource for which the configuration has to be updated
70 > * @param key Configuration key
71 > * @param value Configuration value
72 > * @param configurationTarget Optional target into which the configuration has to be updated.
73 > * If not specified, target will be derived by checking where the configuration is defined.
74 > */
75 > updateValue(resource: URI | undefined, key: string, value: unknown, configurationTarget?: ConfigurationTarget): Promise<void>;
76 >
77 > }
78 >
79 > export const ITextResourcePropertiesService = createDecorator<ITextResourcePropertiesService>('textResourcePropertiesService');
80 >
81 > export interface ITextResourcePropertiesService {
82 >
83 > readonly _serviceBrand: undefined;
84 >
85 > /**
86 > * Returns the End of Line characters for the given resource
87 > */
88 > getEOL(resource: URI, language?: string): string;
89 > }
src/vs/workbench/contrib/chat/common/promptSyntax/hookCompatibility.ts 89 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hookCompatibility.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../base/common/uri.js';
7 > import { basename, dirname } from '../../../../../base/common/path.js';
8 > import { IHookCommand, toHookType } from './hookSchema.js';
9 > import { parseClaudeHooks, extractHookCommandsFromItem } from './hookClaudeCompat.js';
10 > import { resolveCopilotCliHookType } from './hookCopilotCliCompat.js';
11 > import { HookType } from './hookTypes.js';
12 >
13 > /**
14 > * Represents a hook source with its original and normalized properties.
15 > * Used to display hooks from different formats in a unified view.
16 > */
17 > export interface IResolvedHookEntry {
18 > /** The normalized hook type (our canonical HookType enum) */
19 > readonly hookType: HookType;
20 > /** The original hook type ID as it appears in the source file */
21 > readonly originalHookTypeId: string;
22 > /** The source format this hook came from */
23 > readonly sourceFormat: HookSourceFormat;
24 > /** The resolved hook command */
25 > readonly command: IHookCommand;
26 > /** The index of this hook in its array (for editing) */
27 > readonly index: number;
28 > }
29 >
30 > /**
31 > * Supported hook file formats.
32 > */
33 > export enum HookSourceFormat {
34 > /** GitHub Copilot hooks .json format */
35 > Copilot = 'copilot',
36 > /** Claude settings.json / settings.local.json format */
37 > Claude = 'claude',
38 > }
39 >
40 > /**
41 > * Determines the hook source format based on the file URI.
42 > */
43 > export function getHookSourceFormat(fileUri: URI): HookSourceFormat {
44 const filename = basename(fileUri.path).toLowerCase();
45 const dir = dirname(fileUri.path);
53 return HookSourceFormat.Copilot;
54 }
56 > /**
57 > * Checks if a file is read-only based on its source format.
58 > * Claude settings files should be read-only from our perspective since they have a different format.
59 > */
60 > export function isReadOnlyHookSource(format: HookSourceFormat): boolean {
61 return format === HookSourceFormat.Claude;
62 }
64 > /**
65 > * Parses hooks from a Copilot hooks .json file (our native format).
66 > */
67 > export function parseCopilotHooks(
68 json: unknown,
69 workspaceRootUri: URI | undefined,
111 return result;
112 }
114 > /**
115 > * Result of parsing hooks from a file.
116 > */
117 > export interface IParseHooksFromFileResult {
118 > readonly format: HookSourceFormat;
119 > readonly hooks: Map<HookType, { hooks: IHookCommand[]; originalId: string }>;
120 > /**
121 > * Whether all hooks from this file were disabled via `disableAllHooks: true`.
122 > */
123 > readonly disabledAllHooks: boolean;
124 > }
125 >
126 > /**
127 > * Parses hooks from any supported format, auto-detecting the format from the file URI.
128 > */
129 > export function parseHooksFromFile(
130 fileUri: URI,
131 json: unknown,
153 return { format, hooks, disabledAllHooks };
154 }
156 > /**
157 > * Parses hooks from a file, ignoring the `disableAllHooks` flag.
158 > * Used by diagnostics to show which hooks are hidden when `disableAllHooks: true` is set.
159 > */
160 > export function parseHooksIgnoringDisableAll(
161 fileUri: URI,
162 json: unknown,
188 return { format, hooks, disabledAllHooks: true };
189 }
191 > /**
192 > * Gets a human-readable label for a hook source format.
193 > */
194 > export function getHookSourceFormatLabel(format: HookSourceFormat): string {
195 switch (format) {
196 case HookSourceFormat.Claude:
200 }
201 }
203 > /**
204 > * Builds a new hook entry object in the appropriate format for the given source format.
205 > * - Copilot format: `{ type: 'command', command: '' }`
206 > * - Claude format: `{ matcher: '', hooks: [{ type: 'command', command: '' }] }`
207 > */
208 > export function buildNewHookEntry(format: HookSourceFormat): Record<string, unknown> {
209 const commandEntry = { type: 'command', command: '' };
210 if (format === HookSourceFormat.Claude) {
src/vs/base/common/cache.ts 88 covered LOC · 9 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- cache.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken, CancellationTokenSource } from './cancellation.js';
7 > import { IDisposable } from './lifecycle.js';
8 >
9 > export interface CacheResult<T> extends IDisposable {
10 > promise: Promise<T>;
11 > }
12 >
13 > export class Cache<T> {
14 >
15 > private result: CacheResult<T> | null = null;
16 > constructor(private task: (ct: CancellationToken) => Promise<T>) { }
17 >
18 > get(): CacheResult<T> {
19 if (this.result) {
20 return this.result;
35 return this.result;
36 }
37 > } cache.ts
38 >
39 > export function identity<T>(t: T): T {
40 return t;
41 }
42 > cache.ts
43 > interface ICacheOptions<TArg> {
44 > /**
45 > * The cache key is used to identify the cache entry.
46 > * Strict equality is used to compare cache keys.
47 > */
48 > getCacheKey: (arg: TArg) => unknown;
49 > }
50 >
51 > /**
52 > * Uses a LRU cache to make a given parametrized function cached.
53 > * Caches just the last key/value.
54 > */
55 > export class LRUCachedFunction<TArg, TComputed> {
56 > private lastCache: TComputed | undefined = undefined;
57 > private lastArgKey: unknown | undefined = undefined;
58 >
59 > private readonly _fn: (arg: TArg) => TComputed;
60 > private readonly _computeKey: (arg: TArg) => unknown;
61 >
62 > constructor(fn: (arg: TArg) => TComputed);
63 > constructor(options: ICacheOptions<TArg>, fn: (arg: TArg) => TComputed);
64 > constructor(arg1: ICacheOptions<TArg> | ((arg: TArg) => TComputed), arg2?: (arg: TArg) => TComputed) {
65 > if (typeof arg1 === 'function') {
66 > this._fn = arg1;
67 > this._computeKey = identity;
68 > } else {
69 this._fn = arg2!;
70 this._computeKey = arg1.getCacheKey;
71 }
72 > } cache.ts
73 >
74 > public get(arg: TArg): TComputed {
75 const key = this._computeKey(arg);
76 if (this.lastArgKey !== key) {
80 return this.lastCache!;
81 }
82 > } cache.ts
83 >
84 > /**
85 > * Uses an unbounded cache to memoize the results of the given function.
86 > */
87 > export class CachedFunction<TArg, TComputed> {
88 > private readonly _map = new Map<TArg, TComputed>();
89 > private readonly _map2 = new Map<unknown, TComputed>();
90 > public get cachedValues(): ReadonlyMap<TArg, TComputed> {
91 > return this._map;
92 > }
93 >
94 > private readonly _fn: (arg: TArg) => TComputed;
95 > private readonly _computeKey: (arg: TArg) => unknown;
96 >
97 > constructor(fn: (arg: TArg) => TComputed);
98 > constructor(options: ICacheOptions<TArg>, fn: (arg: TArg) => TComputed);
99 > constructor(arg1: ICacheOptions<TArg> | ((arg: TArg) => TComputed), arg2?: (arg: TArg) => TComputed) {
100 if (typeof arg1 === 'function') {
101 this._fn = arg1;
106 }
107 }
108 > cache.ts
109 > public get(arg: TArg): TComputed {
110 const key = this._computeKey(arg);
111 if (this._map2.has(key)) {
118 return value;
119 }
120 > } cache.ts
121 >
122 > /**
123 > * Uses an unbounded cache to memoize the results of the given function.
124 > */
125 > export class WeakCachedFunction<TArg, TComputed> {
126 > private readonly _map = new WeakMap<WeakKey, TComputed>();
127 >
128 > private readonly _fn: (arg: TArg) => TComputed;
129 > private readonly _computeKey: (arg: TArg) => unknown;
130 >
131 > constructor(fn: (arg: TArg) => TComputed);
132 > constructor(options: ICacheOptions<TArg>, fn: (arg: TArg) => TComputed);
133 > constructor(arg1: ICacheOptions<TArg> | ((arg: TArg) => TComputed), arg2?: (arg: TArg) => TComputed) {
134 if (typeof arg1 === 'function') {
135 this._fn = arg1;
140 }
141 }
142 > cache.ts
143 > public get(arg: TArg): TComputed {
144 const key = this._computeKey(arg) as WeakKey;
145 if (this._map.has(key)) {
151 return value;
152 }
153 > } cache.ts
src/vs/platform/defaultAccount/common/defaultAccount.ts 88 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- defaultAccount.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ICopilotTokenInfo, IDefaultAccount, IDefaultAccountAuthenticationProvider, IPolicyData } from '../../../base/common/defaultAccount.js';
7 > import { Event } from '../../../base/common/event.js';
8 > import { createDecorator } from '../../instantiation/common/instantiation.js';
9 >
10 > /**
11 > * Well-known GitHub URL paths used with {@link IDefaultAccountService.resolveGitHubUrl}.
12 > */
13 > export const GitHubPaths = {
14 > copilotSettings: 'settings/copilot/features',
15 > billingBudgets: 'settings/copilot/features?utm_source=vscode',
16 > copilotUpgrade: 'github-copilot/upgrade?utm_source=vscode',
17 > } as const;
18 >
19 > /**
20 > * Outcome of the last `/copilot_internal/managed_settings` fetch.
21 > * - A numeric HTTP status code indicates the server responded with that code.
22 > * - `'ok'`: response parsed and adapted successfully (including an empty `{}` body).
23 > * - `'no-url'`: no `managedSettingsUrl` configured in product.json.
24 > * - `'no-response'`: network error, all sessions rejected, or active rate-limit backoff.
25 > * - `'parse-error'`: response received but JSON parsing failed.
26 > * - `null`: never fetched.
27 > */
28 > export type ManagedSettingsFetchStatus = number | 'ok' | 'no-url' | 'no-response' | 'parse-error' | null;
29 >
30 > export interface IDefaultAccountProvider {
31 > readonly defaultAccount: IDefaultAccount | null;
32 > readonly onDidChangeDefaultAccount: Event<IDefaultAccount | null>;
33 > readonly policyData: IPolicyData | null;
34 > readonly onDidChangePolicyData: Event<IPolicyData | null>;
35 > readonly copilotTokenInfo: ICopilotTokenInfo | null;
36 > readonly onDidChangeCopilotTokenInfo: Event<ICopilotTokenInfo | null>;
37 > readonly managedSettingsFetchStatus: ManagedSettingsFetchStatus;
38 > /** Timestamp (ms) of the last managed-settings fetch, or `null` if never fetched. */
39 > readonly managedSettingsFetchedAt: number | null;
40 > /** The raw JSON response from the managed-settings endpoint, for diagnostics. */
41 > readonly managedSettingsRawResponse: unknown;
42 > getDefaultAccountAuthenticationProvider(): IDefaultAccountAuthenticationProvider;
43 >
44 > /**
45 > * Resolves a GitHub URL path to a full URL, using the GitHub Enterprise
46 > * base URL when the user is authenticated via a GHE provider, or
47 > * `https://github.com` otherwise.
48 > *
49 > * @param path The path portion of the URL (e.g. `settings/copilot/features`).
50 > */
51 > resolveGitHubUrl(path: string): string;
52 >
53 > refresh(options?: { forceRefresh?: boolean }): Promise<IDefaultAccount | null>;
54 > signIn(options?: { additionalScopes?: readonly string[];[key: string]: unknown }): Promise<IDefaultAccount | null>;
55 > signOut(): Promise<void>;
56 > }
57 >
58 > export const IDefaultAccountService = createDecorator<IDefaultAccountService>('defaultAccountService');
59 >
60 > export interface IDefaultAccountService {
61 > readonly _serviceBrand: undefined;
62 > readonly onDidChangeDefaultAccount: Event<IDefaultAccount | null>;
63 > readonly onDidChangePolicyData: Event<IPolicyData | null>;
64 > readonly policyData: IPolicyData | null;
65 > readonly currentDefaultAccount: IDefaultAccount | null;
66 > readonly copilotTokenInfo: ICopilotTokenInfo | null;
67 > readonly onDidChangeCopilotTokenInfo: Event<ICopilotTokenInfo | null>;
68 > readonly managedSettingsFetchStatus: ManagedSettingsFetchStatus;
69 > /** Timestamp (ms) of the last managed-settings fetch, or `null` if never fetched. */
70 > readonly managedSettingsFetchedAt: number | null;
71 > /** The raw JSON response from the managed-settings endpoint, for diagnostics. */
72 > readonly managedSettingsRawResponse: unknown;
73 > getDefaultAccount(): Promise<IDefaultAccount | null>;
74 > getDefaultAccountAuthenticationProvider(): IDefaultAccountAuthenticationProvider;
75 > setDefaultAccountProvider(provider: IDefaultAccountProvider): void;
76 > refresh(options?: { forceRefresh?: boolean }): Promise<IDefaultAccount | null>;
77 > signIn(options?: { additionalScopes?: readonly string[];[key: string]: unknown }): Promise<IDefaultAccount | null>;
78 > signOut(): Promise<void>;
79 >
80 > /**
81 > * Resolves a GitHub URL path to a full URL, using the GitHub Enterprise
82 > * base URL when the user is authenticated via a GHE provider, or
83 > * `https://github.com` otherwise.
84 > *
85 > * @param path The path portion of the URL (e.g. `settings/copilot/features`).
86 > */
87 > resolveGitHubUrl(path: string): string;
88 > }
src/vs/workbench/contrib/chat/common/plugins/agentPluginService.ts 88 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentPluginService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IDisposable, toDisposable } from '../../../../../base/common/lifecycle.js';
7 > import { IObservable } from '../../../../../base/common/observable.js';
8 > import { basename } from '../../../../../base/common/resources.js';
9 > import { URI } from '../../../../../base/common/uri.js';
10 > import { SyncDescriptor0 } from '../../../../../platform/instantiation/common/descriptors.js';
11 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
12 > import { type INamedPluginResource, type IMcpServerDefinition, type IParsedHookCommand, type PluginFormat } from '../../../../../platform/agentPlugins/common/pluginParsers.js';
13 > import { ContributionEnablementState, IEnablementModel } from '../enablement.js';
14 > import { HookType } from '../promptSyntax/hookTypes.js';
15 > import { IMarketplacePlugin } from './pluginMarketplaceService.js';
16 >
17 > export const IAgentPluginService = createDecorator<IAgentPluginService>('agentPluginService');
18 >
19 > export interface IAgentPluginHook {
20 > readonly type: HookType;
21 > readonly hooks: readonly IParsedHookCommand[];
22 > /** URI where this hook is defined -- not unique, multiple hooks may be in a manifest */
23 > readonly uri: URI;
24 > readonly originalId: string;
25 > }
26 >
27 > export type IAgentPluginCommand = INamedPluginResource;
28 > export type IAgentPluginSkill = INamedPluginResource;
29 > export type IAgentPluginAgent = INamedPluginResource;
30 > export type IAgentPluginInstruction = INamedPluginResource;
31 > export type IAgentPluginMcpServerDefinition = IMcpServerDefinition;
32 >
33 > export interface IAgentPlugin {
34 > readonly uri: URI;
35 > readonly format: PluginFormat;
36 > /** Human-readable display name for the plugin. */
37 > readonly label: string;
38 > readonly enablement: IObservable<ContributionEnablementState>;
39 > /**
40 > * When `true`, the plugin is blocked by enterprise policy. It remains
41 > * visible (shown as disabled) but its contributions are inactive and the
42 > * user cannot re-enable it. Folded into {@link enablement} so all gating
43 > * consumers honor it automatically.
44 > */
45 > readonly policyBlocked?: IObservable<boolean>;
46 > /** Removes this plugin from its discovery source (config or installed storage). Undefined for policy-managed plugins that cannot be removed by the user. */
47 > remove?(): void;
48 > readonly hooks: IObservable<readonly IAgentPluginHook[]>;
49 > readonly commands: IObservable<readonly IAgentPluginCommand[]>;
50 > readonly skills: IObservable<readonly IAgentPluginSkill[]>;
51 > readonly agents: IObservable<readonly IAgentPluginAgent[]>;
52 > readonly instructions: IObservable<readonly IAgentPluginInstruction[]>;
53 > readonly mcpServerDefinitions: IObservable<readonly IAgentPluginMcpServerDefinition[]>;
54 > /** Set when the plugin was installed from a marketplace repository. */
55 > readonly fromMarketplace?: IMarketplacePlugin;
56 > }
57 >
58 > export interface IAgentPluginService {
59 > readonly _serviceBrand: undefined;
60 > readonly plugins: IObservable<readonly IAgentPlugin[]>;
61 > readonly enablementModel: IEnablementModel;
62 > }
63 >
64 > export interface IAgentPluginDiscovery extends IDisposable {
65 > readonly plugins: IObservable<readonly IAgentPlugin[] | undefined>;
66 > start(enablementModel: IEnablementModel): void;
67 > }
68 >
69 > export const enum AgentPluginDiscoveryPriority {
70 > Configured = 10,
71 > Marketplace = 20,
72 > Extension = 30,
73 > CopilotCli = 40,
74 > }
75 >
76 > export function getCanonicalPluginCommandId(plugin: { readonly uri: URI; readonly label?: string }, commandName: string): string {
77 const prefix = (plugin.label ? normalizePluginToken(plugin.label) : '') || normalizePluginToken(basename(plugin.uri));
78 const normalizedCommand = normalizePluginToken(commandName);
90 return `${prefix}:${normalizedCommand}`;
91 }
93 function normalizePluginToken(value: string): string {
94 return value
100 .replace(/^[-:.]+|[-:.]+$/g, '');
101 }
103 > class AgentPluginDiscoveryRegistry {
104 > private readonly _discovery: { readonly descriptor: SyncDescriptor0<IAgentPluginDiscovery>; readonly priority: AgentPluginDiscoveryPriority; readonly order: number }[] = [];
105 > private _order = 0;
106 >
107 > register(descriptor: SyncDescriptor0<IAgentPluginDiscovery>, priority: AgentPluginDiscoveryPriority): IDisposable {
108 const registration = { descriptor, priority, order: this._order++ };
109 this._discovery.push(registration);
115 });
116 }
118 > getAll(): readonly { readonly descriptor: SyncDescriptor0<IAgentPluginDiscovery>; readonly priority: AgentPluginDiscoveryPriority; readonly order: number }[] {
119 return [...this._discovery].sort((a, b) => a.priority - b.priority || a.order - b.order);
120 }
122 >
123 > export const agentPluginDiscoveryRegistry = new AgentPluginDiscoveryRegistry();
src/vs/workbench/services/policies/common/accountPolicyService.ts 88 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- accountPolicyService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStringDictionary } from '../../../../base/common/collections.js';
7 > import { IPolicyData } from '../../../../base/common/defaultAccount.js';
8 > import { Emitter, Event } from '../../../../base/common/event.js';
9 > import { ManagedSettingsData } from '../../../../base/common/policy.js';
10 > import { localize } from '../../../../nls.js';
11 > import { RawContextKey } from '../../../../platform/contextkey/common/contextkey.js';
12 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
13 > import { ILogService } from '../../../../platform/log/common/log.js';
14 > import { INativeManagedSettingsService, IFileManagedSettingsService, collectManagedSettingsDefinitions, hasManagedSettingsDefinitions, projectManagedSettings, pickManagedSettings } from '../../../../platform/policy/common/copilotManagedSettings.js';
15 > import { AbstractPolicyService, getRestrictedPolicyValue, IPolicyService, PolicyDefinition, PolicyValue } from '../../../../platform/policy/common/policy.js';
16 > import { IDefaultAccountService } from '../../../../platform/defaultAccount/common/defaultAccount.js';
17 >
18 > /**
19 > * Policy name (declared by `chat.approvedAccountOrganizations`) holding the list of
20 > * GitHub organization logins that satisfy the gate. The token `*` is a wildcard.
21 > */
22 > export const APPROVED_ACCOUNT_ORGANIZATIONS_POLICY_NAME = 'ChatApprovedAccountOrganizations';
23 >
24 > export const enum AccountPolicyGateState {
25 > Inactive = 'inactive',
26 > Satisfied = 'satisfied',
27 > /** Gate active and NOT satisfied — restricted values are applied to all gated policies. */
28 > Restricted = 'restricted',
29 > }
30 >
31 > export const enum AccountPolicyGateUnsatisfiedReason {
32 > NoAccount = 'noAccount',
33 > WrongProvider = 'wrongProvider',
34 > OrgNotApproved = 'orgNotApproved',
35 > PolicyNotResolved = 'policyNotResolved',
36 > }
37 >
38 > export interface IAccountPolicyGateInfo {
39 > readonly state: AccountPolicyGateState;
40 > readonly reason?: AccountPolicyGateUnsatisfiedReason;
41 > readonly approvedOrganizations?: readonly string[];
42 > }
43 >
44 > export const ChatAccountPolicyGateActiveContext = new RawContextKey<boolean>(
45 > 'chatAccountPolicyGateActive',
46 > false,
47 > { type: 'boolean', description: localize('chatAccountPolicyGateActive', "True when the 'Require Approved Account' policy is in effect and the user is not yet signed into an approved GitHub organization, so all AI features are disabled until they sign in.") }
48 > );
49 >
50 > /**
51 > * Read-only accessor for the Account Policy gate state. Backed by the same
52 > * `AccountPolicyService` instance that drives policy enforcement, so UX consumers
53 > * (notifications, context keys, telemetry) cannot drift from the authoritative
54 > * gate decision.
55 > */
56 > export const IAccountPolicyGateService = createDecorator<IAccountPolicyGateService>('accountPolicyGateService');
57 > export interface IAccountPolicyGateService {
58 > readonly _serviceBrand: undefined;
59 > readonly gateInfo: IAccountPolicyGateInfo;
60 > readonly onDidChangeGateInfo: Event<IAccountPolicyGateInfo>;
61 > }
62 >
63 > export class AccountPolicyService extends AbstractPolicyService implements IPolicyService, IAccountPolicyGateService {
64 >
65 > declare readonly _serviceBrand: undefined;
66 >
67 > private _gateInfo: IAccountPolicyGateInfo = { state: AccountPolicyGateState.Inactive };
68 > get gateInfo(): IAccountPolicyGateInfo { return this._gateInfo; }
69 >
70 > private readonly _onDidChangeGateInfo = this._register(new Emitter<IAccountPolicyGateInfo>());
71 > readonly onDidChangeGateInfo = this._onDidChangeGateInfo.event;
72 >
73 > // Read-only — the MultiplexPolicyService owns calling updatePolicyDefinitions.
74 > private readonly managedPolicyReader?: IPolicyService;
75 > private readonly nativeManagedSettingsService?: INativeManagedSettingsService;
76 > private readonly fileManagedSettingsService?: IFileManagedSettingsService;
77 >
78 > constructor(
79 @ILogService private readonly logService: ILogService,
80 @IDefaultAccountService private readonly defaultAccountService: IDefaultAccountService,
121 });
122 }
124 > protected async _updatePolicyDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<void> {
125 this.logService.trace(`AccountPolicyService#_updatePolicyDefinitions: Got ${Object.keys(policyDefinitions).length} policy definitions`);
126 const managedSettings = await this.updateCopilotManagedSettingDefinitions(policyDefinitions);
177 }
178 }
180 > private async updateCopilotManagedSettingDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<ManagedSettingsData | undefined> {
181 if (!this.nativeManagedSettingsService || !hasManagedSettingsDefinitions(policyDefinitions)) {
182 return this.nativeManagedSettingsService?.managedSettings;
185 return this.nativeManagedSettingsService.updatePolicyDefinitions(policyDefinitions);
186 }
188 > private getPolicyData(mdmManagedSettings?: ManagedSettingsData): IPolicyData | undefined {
189 const accountPolicyData = this.defaultAccountService.policyData ?? undefined;
190 const nativeManagedSettings = mdmManagedSettings ?? this.nativeManagedSettingsService?.managedSettings;
212 };
213 }
215 > private computeGateInfo(): IAccountPolicyGateInfo {
216 if (!this.managedPolicyReader) {
217 return { state: AccountPolicyGateState.Inactive };
252 return { state: AccountPolicyGateState.Satisfied, approvedOrganizations: approvedOrgs };
253 }
255 >
256 function parseApprovedOrganizations(raw: PolicyValue | undefined): string[] {
257 // Array-typed policies are delivered as JSON-stringified arrays — see
src/vs/base/common/performance.ts 86 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- performance.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import type { INodeProcess } from './platform.js';
7 >
8 > function _definePolyfillMarks(timeOrigin?: number) {
9 > const _data: [string?, number?] = [];
10 > if (typeof timeOrigin === 'number') {
11 > _data.push('code/timeOrigin', timeOrigin);
12 > }
13 >
14 > function mark(name: string, markOptions?: { startTime?: number }) {
15 > _data.push(name, markOptions?.startTime ?? Date.now()); performance.ts
16 > }
17 > function getMarks() { performance.ts
18 const result = [];
19 for (let i = 0; i < _data.length; i += 2) {
25 return result;
26 }
27 > function clearMarks(name?: string) { performance.ts
28 if (typeof name === 'undefined') {
29 const hasTimeOrigin = _data.length >= 2 && _data[0] === 'code/timeOrigin';
41 }
42 }
43 > return { mark, getMarks, clearMarks }; performance.ts
44 > }
45 >
46 > declare const process: INodeProcess;
47 >
48 > interface IPerformanceEntry {
49 > readonly name: string;
50 > readonly startTime: number;
51 > }
52 >
53 > interface IPerformanceTiming {
54 > readonly navigationStart?: number;
55 > readonly redirectStart?: number;
56 > readonly fetchStart?: number;
57 > }
58 >
59 > interface IPerformance {
60 > mark(name: string, markOptions?: { startTime?: number }): void;
61 > clearMarks(name?: string): void;
62 > getEntriesByType(type: string): IPerformanceEntry[];
63 > readonly timeOrigin: number;
64 > readonly timing: IPerformanceTiming;
65 > readonly nodeTiming?: any;
66 > }
67 >
68 > declare const performance: IPerformance;
69 >
70 > function _define() {
71 >
72 > // Identify browser environment when following property is not present
73 > // https://nodejs.org/dist/latest-v16.x/docs/api/perf_hooks.html#performancenodetiming
74 > // @ts-ignore
75 > if (typeof performance === 'object' && typeof performance.mark === 'function' && !performance.nodeTiming) {
76 // in a browser context, reuse performance-util
77
109 }
110
111 > } else if (typeof process === 'object') { performance.ts
112 > // node.js: use the normal polyfill but add the timeOrigin
113 > // from the node perf_hooks API as very first mark
114 > const timeOrigin = performance?.timeOrigin;
115 > return _definePolyfillMarks(timeOrigin);
116 >
117 > } else {
118 // unknown environment
119 console.trace('perf-util loaded in UNKNOWN environment');
120 return _definePolyfillMarks();
121 }
122 > } performance.ts
123 >
124 > function _factory(sharedObj: any) {
125 > if (!sharedObj.MonacoPerformanceMarks) {
126 > sharedObj.MonacoPerformanceMarks = _define();
127 > }
128 > return sharedObj.MonacoPerformanceMarks;
129 > }
130 >
131 > const perf = _factory(globalThis);
132 >
133 > export const mark: (name: string, markOptions?: { startTime?: number }) => void = perf.mark;
134 >
135 > /**
136 > * Clears performance marks. If a name is given, only marks with that exact
137 > * name are removed. If no name is given, all marks are removed.
138 > */
139 > export const clearMarks: (name?: string) => void = perf.clearMarks;
140 >
141 > export interface PerformanceMark {
142 > readonly name: string;
143 > readonly startTime: number;
144 > }
145 >
146 > /**
147 > * Returns all marks, sorted by `startTime`.
148 > */
149 > export const getMarks: () => PerformanceMark[] = perf.getMarks;
src/vs/platform/mcp/common/mcpResourceScannerService.ts 86 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpResourceScannerService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { assertNever } from '../../../base/common/assert.js';
7 > import { Queue } from '../../../base/common/async.js';
8 > import { VSBuffer } from '../../../base/common/buffer.js';
9 > import { IStringDictionary } from '../../../base/common/collections.js';
10 > import { parse, ParseError } from '../../../base/common/json.js';
11 > import { Disposable } from '../../../base/common/lifecycle.js';
12 > import { ResourceMap } from '../../../base/common/map.js';
13 > import { Mutable } from '../../../base/common/types.js';
14 > import { URI } from '../../../base/common/uri.js';
15 > import { ConfigurationTarget, ConfigurationTargetToString } from '../../configuration/common/configuration.js';
16 > import { FileOperationResult, IFileService, toFileOperationResult } from '../../files/common/files.js';
17 > import { InstantiationType, registerSingleton } from '../../instantiation/common/extensions.js';
18 > import { createDecorator } from '../../instantiation/common/instantiation.js';
19 > import { IUriIdentityService } from '../../uriIdentity/common/uriIdentity.js';
20 > import { IInstallableMcpServer } from './mcpManagement.js';
21 > import { ICommonMcpServerConfiguration, IMcpSandboxConfiguration, IMcpServerConfiguration, IMcpServerVariable, IMcpStdioServerConfiguration, McpServerType } from './mcpPlatformTypes.js';
22 >
23 > interface IScannedMcpServers {
24 > servers?: IStringDictionary<Mutable<IMcpServerConfiguration>>;
25 > inputs?: IMcpServerVariable[];
26 > sandbox?: IMcpSandboxConfiguration;
27 > }
28 >
29 > interface IOldScannedMcpServer {
30 > id: string;
31 > name: string;
32 > version?: string;
33 > gallery?: boolean;
34 > config: Mutable<IMcpServerConfiguration>;
35 > }
36 >
37 > interface IScannedWorkspaceMcpServers {
38 > settings?: {
39 > mcp?: IScannedMcpServers;
40 > };
41 > }
42 >
43 > export type McpResourceTarget = ConfigurationTarget.USER | ConfigurationTarget.WORKSPACE | ConfigurationTarget.WORKSPACE_FOLDER;
44 >
45 > export const IMcpResourceScannerService = createDecorator<IMcpResourceScannerService>('IMcpResourceScannerService');
46 > export interface IMcpResourceScannerService {
47 > readonly _serviceBrand: undefined;
48 > scanMcpServers(mcpResource: URI, target?: McpResourceTarget): Promise<IScannedMcpServers>;
49 > addMcpServers(servers: IInstallableMcpServer[], mcpResource: URI, target?: McpResourceTarget): Promise<void>;
50 > updateSandboxConfig(updateFn: (data: IScannedMcpServers) => IScannedMcpServers, mcpResource: URI, target?: McpResourceTarget): Promise<void>;
51 > removeMcpServers(serverNames: string[], mcpResource: URI, target?: McpResourceTarget): Promise<void>;
52 > }
53 >
54 > export class McpResourceScannerService extends Disposable implements IMcpResourceScannerService {
55 > readonly _serviceBrand: undefined;
56 >
57 > private readonly resourcesAccessQueueMap = new ResourceMap<Queue<IScannedMcpServers>>();
58 >
59 > constructor(
60 @IFileService private readonly fileService: IFileService,
61 @IUriIdentityService protected readonly uriIdentityService: IUriIdentityService,
63 super();
64 }
66 > async scanMcpServers(mcpResource: URI, target?: McpResourceTarget): Promise<IScannedMcpServers> {
67 return this.withProfileMcpServers(mcpResource, target);
68 }
70 > async addMcpServers(servers: IInstallableMcpServer[], mcpResource: URI, target?: McpResourceTarget): Promise<void> {
71 await this.withProfileMcpServers(mcpResource, target, scannedMcpServers => {
72 let updatedInputs = scannedMcpServers.inputs ?? [];
83 });
84 }
86 > async updateSandboxConfig(updateFn: (data: IScannedMcpServers) => IScannedMcpServers, mcpResource: URI, target?: McpResourceTarget): Promise<void> {
87 await this.withProfileMcpServers(mcpResource, target, updateFn);
88 }
90 > async removeMcpServers(serverNames: string[], mcpResource: URI, target?: McpResourceTarget): Promise<void> {
91 await this.withProfileMcpServers(mcpResource, target, scannedMcpServers => {
92 for (const serverName of serverNames) {
98 });
99 }
101 > private async withProfileMcpServers(mcpResource: URI, target?: McpResourceTarget, updateFn?: (data: IScannedMcpServers) => IScannedMcpServers): Promise<IScannedMcpServers> {
102 return this.getResourceAccessQueue(mcpResource)
103 .queue(async (): Promise<IScannedMcpServers> => {
143 });
144 }
146 > private async writeScannedMcpServers(mcpResource: URI, scannedMcpServers: IScannedMcpServers): Promise<void> {
147 if ((scannedMcpServers.servers && Object.keys(scannedMcpServers.servers).length > 0)
148 || (scannedMcpServers.inputs && scannedMcpServers.inputs.length > 0)
153 }
154 }
156 > private async writeScannedMcpServersToWorkspaceFolder(mcpResource: URI, scannedMcpServers: IScannedMcpServers): Promise<void> {
157 await this.fileService.writeFile(mcpResource, VSBuffer.fromString(JSON.stringify(scannedMcpServers, null, '\t')));
158 }
160 > private async writeScannedMcpServersToWorkspace(mcpResource: URI, scannedMcpServers: IScannedMcpServers): Promise<void> {
161 let scannedWorkspaceMcpServers: IScannedWorkspaceMcpServers | undefined;
162 try {
179 await this.fileService.writeFile(mcpResource, VSBuffer.fromString(JSON.stringify(scannedWorkspaceMcpServers, null, '\t')));
180 }
182 > private fromUserMcpServers(scannedMcpServers: IScannedMcpServers): IScannedMcpServers {
183 const userMcpServers: IScannedMcpServers = {
184 inputs: scannedMcpServers.inputs,
194 return userMcpServers;
195 }
197 > private fromWorkspaceFolderMcpServers(scannedWorkspaceFolderMcpServers: IScannedMcpServers): IScannedMcpServers {
198 const scannedMcpServers: IScannedMcpServers = {
199 inputs: scannedWorkspaceFolderMcpServers.inputs,
210 return scannedMcpServers;
211 }
213 > private sanitizeServer(serverOrConfig: IOldScannedMcpServer | Mutable<IMcpServerConfiguration>): IMcpServerConfiguration {
214 let server: IMcpServerConfiguration;
215 if ((<IOldScannedMcpServer>serverOrConfig).config) {
229 return server;
230 }
232 > private getResourceAccessQueue(file: URI): Queue<IScannedMcpServers> {
233 let resourceQueue = this.resourcesAccessQueueMap.get(file);
234 if (!resourceQueue) {
238 return resourceQueue;
239 }
241 >
242 > registerSingleton(IMcpResourceScannerService, McpResourceScannerService, InstantiationType.Delayed);
src/vs/platform/extensionManagement/common/extensionManagementUtil.ts 82 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensionManagementUtil.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { compareIgnoreCase } from '../../../base/common/strings.js';
7 > import { IExtensionIdentifier, IGalleryExtension, ILocalExtension, MaliciousExtensionInfo, getTargetPlatform } from './extensionManagement.js';
8 > import { ExtensionIdentifier, IExtension, TargetPlatform, UNDEFINED_PUBLISHER } from '../../extensions/common/extensions.js';
9 > import { IFileService } from '../../files/common/files.js';
10 > import { isLinux, platform } from '../../../base/common/platform.js';
11 > import { URI } from '../../../base/common/uri.js';
12 > import { getErrorMessage } from '../../../base/common/errors.js';
13 > import { ILogService } from '../../log/common/log.js';
14 > import { arch } from '../../../base/common/process.js';
15 > import { TelemetryTrustedValue } from '../../telemetry/common/telemetryUtils.js';
16 > import { isString } from '../../../base/common/types.js';
17 >
18 > export function areSameExtensions(a: IExtensionIdentifier, b: IExtensionIdentifier): boolean {
19 if (a.uuid && b.uuid) {
20 return a.uuid === b.uuid;
25 return compareIgnoreCase(a.id, b.id) === 0;
26 }
28 > const ExtensionKeyRegex = /^([^.]+\..+)-(\d+\.\d+\.\d+)(-(.+))?$/;
29 >
30 > export class ExtensionKey {
31 >
32 > static create(extension: IExtension | IGalleryExtension): ExtensionKey {
33 > const version = (extension as IExtension).manifest ? (extension as IExtension).manifest.version : (extension as IGalleryExtension).version;
34 > const targetPlatform = (extension as IExtension).manifest ? (extension as IExtension).targetPlatform : (extension as IGalleryExtension).properties.targetPlatform;
35 > return new ExtensionKey(extension.identifier, version, targetPlatform);
36 > }
37 >
38 > static parse(key: string): ExtensionKey | null {
39 const matches = ExtensionKeyRegex.exec(key);
40 return matches && matches[1] && matches[2] ? new ExtensionKey({ id: matches[1] }, matches[2], matches[4] as TargetPlatform || undefined) : null;
41 }
43 > readonly id: string;
44 >
45 > constructor(
46 readonly identifier: IExtensionIdentifier,
47 readonly version: string,
50 this.id = identifier.id;
51 }
53 > toString(): string {
54 return `${this.id}-${this.version}${this.targetPlatform !== TargetPlatform.UNDEFINED ? `-${this.targetPlatform}` : ''}`;
55 }
57 > equals(o: unknown): boolean {
58 if (!(o instanceof ExtensionKey)) {
59 return false;
61 return areSameExtensions(this, o) && this.version === o.version && this.targetPlatform === o.targetPlatform;
62 }
64 >
65 > const EXTENSION_IDENTIFIER_WITH_VERSION_REGEX = /^([^.]+\..+)@((prerelease)|(\d+\.\d+\.\d+(-.*)?))$/;
66 > export function getIdAndVersion(id: string): [string, string | undefined] {
67 const matches = EXTENSION_IDENTIFIER_WITH_VERSION_REGEX.exec(id);
68 if (matches && matches[1]) {
71 return [adoptToGalleryExtensionId(id), undefined];
72 }
74 > export function getExtensionId(publisher: string, name: string): string {
75 return `${publisher}.${name}`;
76 }
78 > export function adoptToGalleryExtensionId(id: string): string {
79 return id.toLowerCase();
80 }
82 > export function getGalleryExtensionId(publisher: string | undefined, name: string): string {
83 return adoptToGalleryExtensionId(getExtensionId(publisher ?? UNDEFINED_PUBLISHER, name));
84 }
86 > export function groupByExtension<T>(extensions: T[], getExtensionIdentifier: (t: T) => IExtensionIdentifier): T[][] {
87 const byExtension: T[][] = [];
88 const findGroup = (extension: T) => {
104 return byExtension;
105 }
107 > export function getLocalExtensionTelemetryData(extension: ILocalExtension) {
108 return {
109 id: extension.identifier.id,
116 };
117 }
119 >
120 > /* __GDPR__FRAGMENT__
121 > "GalleryExtensionTelemetryData" : {
122 > "id" : { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
123 > "name": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
124 > "extensionVersion": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
125 > "galleryId": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
126 > "publisherId": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
127 > "publisherName": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
128 > "publisherDisplayName": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
129 > "isPreReleaseVersion": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
130 > "dependencies": { "classification": "SystemMetaData", "purpose": "FeatureInsight", "isMeasurement": true },
131 > "isSigned": { "classification": "SystemMetaData", "purpose": "FeatureInsight" },
132 > "${include}": [
133 > "${GalleryExtensionTelemetryData2}"
134 > ]
135 > }
136 > */
137 > export function getGalleryExtensionTelemetryData(extension: IGalleryExtension) {
138 return {
139 id: new TelemetryTrustedValue(extension.identifier.id),
150 };
151 }
153 > export const BetterMergeId = new ExtensionIdentifier('pprice.better-merge');
154 >
155 > export function getExtensionDependencies(installedExtensions: ReadonlyArray<IExtension>, extension: IExtension): IExtension[] {
156 const dependencies: IExtension[] = [];
157 const extensions = extension.manifest.extensionDependencies?.slice(0) ?? [];
171 return dependencies;
172 }
174 async function isAlpineLinux(fileService: IFileService, logService: ILogService): Promise<boolean> {
175 if (!isLinux) {
191 return !!content && (content.match(/^ID=([^\u001b\r\n]*)/m) || [])[1] === 'alpine';
192 }
194 export async function computeTargetPlatform(fileService: IFileService, logService: ILogService): Promise<TargetPlatform> {
195 const alpineLinux = await isAlpineLinux(fileService, logService);
198 return targetPlatform;
199 }
201 > export function isMalicious(identifier: IExtensionIdentifier, malicious: ReadonlyArray<MaliciousExtensionInfo>): boolean {
202 return findMatchingMaliciousEntry(identifier, malicious) !== undefined;
203 }
205 > export function findMatchingMaliciousEntry(identifier: IExtensionIdentifier, malicious: ReadonlyArray<MaliciousExtensionInfo>): MaliciousExtensionInfo | undefined {
206 return malicious.find(({ extensionOrPublisher }) => {
207 if (isString(extensionOrPublisher)) {
src/vs/platform/mcp/common/mcpPlatformTypes.ts 81 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpPlatformTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStringDictionary } from '../../../base/common/collections.js';
7 >
8 > export interface IMcpDevModeConfig {
9 > /** Pattern or list of glob patterns to watch relative to the workspace folder. */
10 > watch?: string | string[];
11 > /** Whether to debug the MCP server when it's started. */
12 > debug?: { type: 'node' } | { type: 'debugpy'; debugpyPath?: string };
13 > }
14 >
15 > export interface IMcpSandboxConfiguration {
16 > network?: {
17 > allowedDomains?: string[];
18 > deniedDomains?: string[];
19 > };
20 > filesystem?: {
21 > denyRead?: string[];
22 > allowWrite?: string[];
23 > denyWrite?: string[];
24 > };
25 > }
26 >
27 > export const enum McpServerVariableType {
28 > PROMPT = 'promptString',
29 > PICK = 'pickString',
30 > }
31 >
32 > export interface IMcpServerVariable {
33 > readonly id: string;
34 > readonly type: McpServerVariableType;
35 > readonly description: string;
36 > readonly password: boolean;
37 > readonly default?: string;
38 > readonly options?: readonly string[];
39 > readonly serverName?: string;
40 > }
41 >
42 > export const enum McpServerType {
43 > LOCAL = 'stdio',
44 > REMOTE = 'http',
45 > }
46 >
47 > export interface ICommonMcpServerConfiguration {
48 > readonly type: McpServerType;
49 > readonly version?: string;
50 > readonly gallery?: boolean | string;
51 > }
52 >
53 > export interface IMcpStdioServerConfiguration extends ICommonMcpServerConfiguration {
54 > readonly type: McpServerType.LOCAL;
55 > readonly command: string;
56 > readonly args?: readonly string[];
57 > readonly env?: Record<string, string | number | null>;
58 > readonly envFile?: string;
59 > readonly cwd?: string;
60 > readonly sandboxEnabled?: boolean;
61 > readonly dev?: IMcpDevModeConfig;
62 > }
63 >
64 > export interface IMcpRemoteServerOAuthConfiguration {
65 > readonly clientId?: string;
66 > }
67 >
68 > export interface IMcpRemoteServerConfiguration extends ICommonMcpServerConfiguration {
69 > readonly type: McpServerType.REMOTE;
70 > readonly url: string;
71 > readonly headers?: Record<string, string>;
72 > readonly oauth?: IMcpRemoteServerOAuthConfiguration;
73 > readonly dev?: IMcpDevModeConfig;
74 > }
75 >
76 > export type IMcpServerConfiguration = IMcpStdioServerConfiguration | IMcpRemoteServerConfiguration;
77 >
78 > export interface IMcpServersConfiguration {
79 > servers?: IStringDictionary<IMcpServerConfiguration>;
80 > inputs?: IMcpServerVariable[];
81 > }
src/vs/workbench/services/workspaces/common/workspaceEditing.ts 80 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workspaceEditing.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
8 > import { IWorkspaceFolderCreationData } from '../../../../platform/workspaces/common/workspaces.js';
9 > import { URI } from '../../../../base/common/uri.js';
10 > import { IAnyWorkspaceIdentifier, IWorkspaceIdentifier } from '../../../../platform/workspace/common/workspace.js';
11 >
12 > export const IWorkspaceEditingService = createDecorator<IWorkspaceEditingService>('workspaceEditingService');
13 >
14 > /**
15 > * An event that is fired after entering a workspace. Clients can join the entering
16 > * by providing a promise from the join method. This allows for long running operations
17 > * to complete (e.g. to migrate data into the new workspace) before the workspace
18 > * is fully entered.
19 > */
20 > export interface IDidEnterWorkspaceEvent {
21 > readonly oldWorkspace: IAnyWorkspaceIdentifier;
22 > readonly newWorkspace: IAnyWorkspaceIdentifier;
23 >
24 > join(promise: Promise<void>): void;
25 > }
26 >
27 > export interface IWorkspaceEditingService {
28 >
29 > readonly _serviceBrand: undefined;
30 >
31 > /**
32 > * Fired after the workspace is entered. Allows listeners to join the
33 > * entering with a promise to migrate data into this new workspace.
34 > */
35 > readonly onDidEnterWorkspace: Event<IDidEnterWorkspaceEvent>;
36 >
37 > /**
38 > * Add folders to the existing workspace.
39 > * When `donotNotifyError` is `true`, error will be bubbled up otherwise, the service handles the error with proper message and action
40 > */
41 > addFolders(folders: IWorkspaceFolderCreationData[], donotNotifyError?: boolean): Promise<void>;
42 >
43 > /**
44 > * Remove folders from the existing workspace
45 > * When `donotNotifyError` is `true`, error will be bubbled up otherwise, the service handles the error with proper message and action
46 > */
47 > removeFolders(folders: URI[], donotNotifyError?: boolean): Promise<void>;
48 >
49 > /**
50 > * Allows to add and remove folders to the existing workspace at once.
51 > * When `donotNotifyError` is `true`, error will be bubbled up otherwise, the service handles the error with proper message and action
52 > */
53 > updateFolders(index: number, deleteCount?: number, foldersToAdd?: IWorkspaceFolderCreationData[], donotNotifyError?: boolean): Promise<void>;
54 >
55 > /**
56 > * Enters the workspace with the provided path.
57 > */
58 > enterWorkspace(path: URI): Promise<void>;
59 >
60 > /**
61 > * Creates a new workspace with the provided folders and opens it. if path is provided
62 > * the workspace will be saved into that location.
63 > */
64 > createAndEnterWorkspace(folders: IWorkspaceFolderCreationData[], path?: URI): Promise<void>;
65 >
66 > /**
67 > * Saves the current workspace to the provided path and opens it. requires a workspace to be opened.
68 > */
69 > saveAndEnterWorkspace(path: URI): Promise<void>;
70 >
71 > /**
72 > * Copies current workspace settings to the target workspace.
73 > */
74 > copyWorkspaceSettings(toWorkspace: IWorkspaceIdentifier): Promise<void>;
75 >
76 > /**
77 > * Picks a new workspace path
78 > */
79 > pickNewWorkspacePath(): Promise<URI | undefined>;
80 > }
src/vs/base/test/common/virtualScheduling/embedding.ts 76 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- embedding.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { setTimeout0, setTimeout0IsFaster } from '../../../common/platform.js';
7 > import { TimeApi } from './timeApi.js';
8 > import { VirtualEvent } from './virtualClock.js';
9 >
10 > /**
11 > * # The processor/host embedding
12 > *
13 > * An {@link Embedding} is the contract between the processor's pure state
14 > * machine and the host event loop. It is invoked once per virtual step that
15 > * produced progress, and decides *how* the processor reaches the host before
16 > * the next step.
17 > *
18 > * ## Contract
19 > *
20 > * On each invocation the embedding MUST do exactly one of:
21 > *
22 > * 1. Return `'continueSync'` **without** calling `then`. The processor will
23 > * loop in place on the same host stack frame.
24 > *
25 > * 2. Schedule `then` on a host primitive (microtask, macrotask, paint frame)
26 > * and return `'cbScheduled'`. The processor will return and wait for the
27 > * callback to re-enter the trampoline.
28 > *
29 > * The embedding MUST NOT call `then` synchronously and also return
30 > * `'cbScheduled'` (that would re-enter the trampoline before this call
31 > * completed). Likewise, returning `'continueSync'` while having scheduled
32 > * `then` async would cause `then` to fire after the trampoline already
33 > * looped — also a bug.
34 > *
35 > * ## Why a callback contract instead of async/await
36 > *
37 > * Every `await` is an implicit microtask hop. For code whose job is to
38 > * decide host hops, that's the wrong abstraction: the reader has to mentally
39 > * compile the `await` to a boundary. With this contract, every host hop is
40 > * a named call to a single primitive (`api.setTimeout`, `setTimeout0`,
41 > * `api.requestAnimationFrame`, …) at exactly one site in this file.
42 > */
43 > export type Embedding = (
44 > nextEvent: VirtualEvent,
45 > then: () => void,
46 > ) => 'continueSync' | 'cbScheduled';
47 >
48 > /**
49 > * Tasks never schedule via promise chains. The processor runs virtual events
50 > * back-to-back on a single host stack frame — fastest possible, but starves
51 > * the host event loop for the duration of the run.
52 > *
53 > * Use only for tests where no `await` / `.then` chains are involved between
54 > * scheduling and execution of virtual events.
55 > */
56 > export const syncEmbedding: Embedding = () => 'continueSync';
57 >
58 > /**
59 > * Tasks may schedule via `await` / `.then`. Between virtual events, yield to
60 > * the host so the *microtask closure* — the current microtask plus every
61 > * microtask it transitively enqueues — drains before the next event runs.
62 > *
63 > * This is the embedding to use for almost all integration-style tests.
64 > */
65 > export function drainMicrotasksEmbedding(realApi: TimeApi): Embedding {
66 return (next, then) => {
67 if (next.preferRealAnimationFrame && realApi.requestAnimationFrame) {
73 };
74 }
76 > /**
77 > * Schedule `cb` after the closure of the current microtask queue: `cb`
78 > * fires only after the current microtask AND every microtask it
79 > * (recursively, transitively) enqueues has settled.
80 > *
81 > * Per the HTML spec, a macrotask runs only when the microtask queue is
82 > * empty, so any macrotask primitive achieves this. We pick the fastest
83 > * one available on the host.
84 > */
85 > export function nextMacrotask(api: TimeApi, cb: () => void): void {
86 if (setTimeout0IsFaster) { setTimeout0(cb); return; }
87 if (api.setImmediate) { api.setImmediate(cb); return; }
src/vs/base/common/observableInternal/logging/debugGetDependencyGraph.ts 75 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- debugGetDependencyGraph.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IObserver } from '../base.js';
7 > import { Derived } from '../observables/derivedImpl.js';
8 > import { FromEventObservable } from '../observables/observableFromEvent.js';
9 > import { ObservableValue } from '../observables/observableValue.js';
10 > import { AutorunObserver } from '../reactions/autorunImpl.js';
11 > import { formatValue } from './consoleObservableLogger.js';
12 >
13 > interface IOptions {
14 > type: 'dependencies' | 'observers';
15 > debugNamePostProcessor?: (name: string) => string;
16 > }
17 >
18 > export function debugGetObservableGraph(obs: IObservable<any> | IObserver, options: IOptions): string {
19 const debugNamePostProcessor = options?.debugNamePostProcessor ?? ((str: string) => str);
20 const info = Info.from(obs, debugNamePostProcessor);
31 }
32 }
34 function formatObservableInfoWithDependencies(info: Info, indentLevel: number, alreadyListed: Set<IObservable<any> | IObserver>, options: IOptions): string {
35 const indent = '\t\t'.repeat(indentLevel);
58 return lines.join('\n');
59 }
61 function formatObservableInfoWithObservers(info: Info, indentLevel: number, alreadyListed: Set<IObservable<any> | IObserver>, options: IOptions): string {
62 const indent = '\t\t'.repeat(indentLevel);
85 return lines.join('\n');
86 }
88 > class Info {
89 > public static from(obs: IObservable<any> | IObserver, debugNamePostProcessor: (name: string) => string): Info | undefined {
90 > if (obs instanceof AutorunObserver) {
91 > const state = obs.debugGetState(); debugGetDependencyGraph.ts
92 > return new Info(
93 > obs,
94 > debugNamePostProcessor(obs.debugName),
95 > 'autorun',
96 > undefined,
97 > state.stateStr,
98 > Array.from(state.dependencies),
99 > []
100 > );
101 > } else if (obs instanceof Derived) { debugGetDependencyGraph.ts
102 > const state = obs.debugGetState();
103 > return new Info(
104 > obs,
105 > debugNamePostProcessor(obs.debugName),
106 > 'derived',
107 > state.value,
108 > state.stateStr,
109 > Array.from(state.dependencies),
110 > Array.from(obs.debugGetObservers())
111 > );
112 > } else if (obs instanceof ObservableValue) {
113 > const state = obs.debugGetState();
114 > return new Info(
115 > obs,
116 > debugNamePostProcessor(obs.debugName),
117 > 'observableValue',
118 > state.value,
119 > 'upToDate',
120 > [],
121 > Array.from(obs.debugGetObservers())
122 > );
123 > } else if (obs instanceof FromEventObservable) {
124 > const state = obs.debugGetState(); debugGetDependencyGraph.ts
125 > return new Info(
126 > obs,
127 > debugNamePostProcessor(obs.debugName),
128 > 'fromEvent',
129 > state.value,
130 > state.hasValue ? 'upToDate' : 'initial',
131 > [],
132 > Array.from(obs.debugGetObservers())
133 > );
134 > }
135 > return undefined;
137 >
138 > public static unknown(obs: IObservable<any> | IObserver): Info {
139 return new Info(
140 obs,
147 );
148 }
150 > constructor(
151 public readonly sourceObj: IObservable<any> | IObserver,
152 public readonly name: string,
src/vs/base/common/themables.ts 75 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- themables.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Codicon } from './codicons.js';
7 >
8 > export type ColorIdentifier = string;
9 >
10 > export type IconIdentifier = string;
11 >
12 > export interface ThemeColor {
13 > id: string;
14 > }
15 >
16 > export namespace ThemeColor {
17 > export function isThemeColor(obj: unknown): obj is ThemeColor {
18 return !!obj && typeof obj === 'object' && typeof (<ThemeColor>obj).id === 'string';
19 }
20 > } themables.ts
21 >
22 > export function themeColorFromId(id: ColorIdentifier) {
23 return { id };
24 }
26 >
27 > export interface ThemeIcon {
28 > readonly id: string;
29 > readonly color?: ThemeColor;
30 > }
31 >
32 > export namespace ThemeIcon {
33 > export const iconNameSegment = '[A-Za-z0-9]+';
34 > export const iconNameExpression = '[A-Za-z0-9-]+';
35 > export const iconModifierExpression = '~[A-Za-z]+';
36 > export const iconNameCharacter = '[A-Za-z0-9~-]';
37 >
38 > const ThemeIconIdRegex = new RegExp(`^(${iconNameExpression})(${iconModifierExpression})?$`);
39 >
40 > export function asClassNameArray(icon: ThemeIcon): string[] {
41 const match = ThemeIconIdRegex.exec(icon.id);
42 if (!match) {
50 return classNames;
51 }
53 > export function asClassName(icon: ThemeIcon): string {
54 return asClassNameArray(icon).join(' ');
55 }
57 > export function asCSSSelector(icon: ThemeIcon): string {
58 return '.' + asClassNameArray(icon).join('.');
59 }
61 > export function isThemeIcon(obj: unknown): obj is ThemeIcon {
62 return !!obj && typeof obj === 'object' && typeof (<ThemeIcon>obj).id === 'string' && (typeof (<ThemeIcon>obj).color === 'undefined' || ThemeColor.isThemeColor((<ThemeIcon>obj).color));
63 }
65 > const _regexFromString = new RegExp(`^\\$\\((${ThemeIcon.iconNameExpression}(?:${ThemeIcon.iconModifierExpression})?)\\)$`);
66 >
67 > export function fromString(str: string): ThemeIcon | undefined {
68 const match = _regexFromString.exec(str);
69 if (!match) {
73 return { id: name };
74 }
76 > export function fromId(id: string): ThemeIcon {
77 > return { id }; themables.ts
78 > }
80 > export function modify(icon: ThemeIcon, modifier: 'disabled' | 'spin' | undefined): ThemeIcon {
81 > let id = icon.id; themables.ts
82 > const tildeIndex = id.lastIndexOf('~');
83 > if (tildeIndex !== -1) {
84 id = id.substring(0, tildeIndex);
85 }
86 > if (modifier) { themables.ts
87 > id = `${id}~${modifier}`;
88 > }
89 > return { id };
90 > }
92 > export function getModifier(icon: ThemeIcon): string | undefined {
93 const tildeIndex = icon.id.lastIndexOf('~');
94 if (tildeIndex !== -1) {
97 return undefined;
98 }
100 > export function isEqual(ti1: ThemeIcon, ti2: ThemeIcon): boolean {
101 return ti1.id === ti2.id && ti1.color?.id === ti2.color?.id;
102 }
103 > themables.ts
104 > /**
105 > * Returns whether specified icon is defined and has 'file' ID.
106 > */
107 > export function isFile(icon: ThemeIcon | undefined): boolean {
108 return icon?.id === Codicon.file.id;
109 }
110 > themables.ts
111 > /**
112 > * Returns whether specified icon is defined and has 'folder' ID.
113 > */
114 > export function isFolder(icon: ThemeIcon | undefined): boolean {
115 return icon?.id === Codicon.folder.id;
116 }
117 > } themables.ts
src/vs/platform/label/common/label.ts 75 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- label.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { IDisposable } from '../../../base/common/lifecycle.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { createDecorator } from '../../instantiation/common/instantiation.js';
10 > import { IWorkspace, ISingleFolderWorkspaceIdentifier, IWorkspaceIdentifier } from '../../workspace/common/workspace.js';
11 >
12 > export const ILabelService = createDecorator<ILabelService>('labelService');
13 >
14 > export interface ILabelService {
15 >
16 > readonly _serviceBrand: undefined;
17 >
18 > /**
19 > * Gets the human readable label for a uri.
20 > * If `relative` is passed returns a label relative to the workspace root that the uri belongs to.
21 > * If `noPrefix` is passed does not tildify the label and also does not prepand the root name for relative labels in a multi root scenario.
22 > * If `separator` is passed, will use that over the defined path separator of the formatter.
23 > * If `appendWorkspaceSuffix` is passed, will append the name of the workspace to the label.
24 > */
25 > getUriLabel(resource: URI, options?: { relative?: boolean; noPrefix?: boolean; separator?: '/' | '\\'; appendWorkspaceSuffix?: boolean }): string;
26 > getUriBasenameLabel(resource: URI): string;
27 > getWorkspaceLabel(workspace: (IWorkspaceIdentifier | ISingleFolderWorkspaceIdentifier | URI | IWorkspace), options?: { verbose: Verbosity }): string;
28 > getHostLabel(scheme: string, authority?: string): string;
29 > getHostTooltip(scheme: string, authority?: string): string | undefined;
30 > getSeparator(scheme: string, authority?: string): '/' | '\\';
31 >
32 > registerFormatter(formatter: ResourceLabelFormatter): IDisposable;
33 > readonly onDidChangeFormatters: Event<IFormatterChangeEvent>;
34 >
35 > /**
36 > * Registers a formatter that's cached for the machine beyond the lifecycle
37 > * of the current window. Disposing the formatter _will not_ remove it from
38 > * the cache.
39 > */
40 > registerCachedFormatter(formatter: ResourceLabelFormatter): IDisposable;
41 > }
42 >
43 > export const enum Verbosity {
44 > SHORT,
45 > MEDIUM,
46 > LONG
47 > }
48 >
49 > export interface IFormatterChangeEvent {
50 > scheme: string;
51 > }
52 >
53 > export interface ResourceLabelFormatter {
54 > scheme: string;
55 > authority?: string;
56 > priority?: boolean;
57 > formatting: ResourceLabelFormatting;
58 > }
59 >
60 > export interface ResourceLabelFormatting {
61 > label: string; // myLabel:/${path}
62 > separator: '/' | '\\' | '';
63 > tildify?: boolean;
64 > normalizeDriveLetter?: boolean;
65 > workspaceSuffix?: string;
66 > workspaceTooltip?: string;
67 > authorityPrefix?: string;
68 > stripPathStartingSeparator?: boolean;
69 > /**
70 > * Number of leading path segments to strip from `${path}` before
71 > * substitution. For example, a value of `2` turns
72 > * `/scheme/authority/rest/of/path` into `/rest/of/path`.
73 > */
74 > stripPathSegments?: number;
75 > }
src/vs/workbench/contrib/chat/test/common/promptSyntax/service/mockPromptsService.ts 75 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockPromptsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../../../base/common/cancellation.js';
7 > import { Emitter, Event } from '../../../../../../../base/common/event.js';
8 > import { IDisposable } from '../../../../../../../base/common/lifecycle.js';
9 > import { URI } from '../../../../../../../base/common/uri.js';
10 > import { ITextModel } from '../../../../../../../editor/common/model.js';
11 > import { IExtensionDescription } from '../../../../../../../platform/extensions/common/extensions.js';
12 > import { PromptsType } from '../../../../common/promptSyntax/promptTypes.js';
13 > import { ParsedPromptFile } from '../../../../common/promptSyntax/promptFileParser.js';
14 > import { IAgentSkill, ICustomAgent, IPromptDiscoveryInfo, IPromptFileContext, IPromptFileResource, IPromptPath, IPromptsService, IAgentInstructionFile, IInstructionFile, PromptsStorage } from '../../../../common/promptSyntax/service/promptsService.js';
15 > import { ResourceSet } from '../../../../../../../base/common/map.js';
16 >
17 > export class MockPromptsService implements IPromptsService {
19 > _serviceBrand: undefined;
20 >
21 > private readonly _onDidChangeCustomAgents = new Emitter<void>();
22 > readonly onDidChangeCustomAgents = this._onDidChangeCustomAgents.event;
23 >
24 > private _customModes: ICustomAgent[] = [];
25 >
26 > setCustomModes(modes: ICustomAgent[]): void {
27 > this._customModes = modes;
28 > this._onDidChangeCustomAgents.fire();
29 > }
30 >
31 > async getCustomAgents(token: CancellationToken): Promise<readonly ICustomAgent[]> {
32 return this._customModes;
33 }
35 > // Stub implementations for required interface methods
36 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
37 > getSyntaxParserFor(_model: any): any { throw new Error('Not implemented'); }
38 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
39 > listPromptFiles(_type: any): Promise<readonly any[]> { throw new Error('Not implemented'); }
40 > listPromptFilesForStorage(type: PromptsType, storage: PromptsStorage, token: CancellationToken): Promise<readonly IPromptPath[]> { throw new Error('Not implemented'); }
41 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
42 > getSourceFolders(_type: any): Promise<readonly any[]> { throw new Error('Not implemented'); }
43 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
44 > getResolvedSourceFolders(_type: any): Promise<readonly any[]> { throw new Error('Not implemented'); }
45 > isValidSlashCommandName(_command: string): boolean { return false; }
46 > hasPromptSlashCommand(_name: string): boolean { return false; }
47 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
48 > resolvePromptSlashCommand(command: string, _sessionType: string | undefined, _token: CancellationToken): Promise<any> { throw new Error('Not implemented'); }
49 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
50 > getPromptSlashCommands(_token: CancellationToken): Promise<any[]> { throw new Error('Not implemented'); }
51 > getPromptSlashCommandName(uri: URI, _token: CancellationToken): Promise<string> { throw new Error('Not implemented'); }
52 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
53 > parse(_uri: URI, _type: any, _token: CancellationToken): Promise<any> { throw new Error('Not implemented'); }
54 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
55 > parseNew(uri: URI, _token: CancellationToken): Promise<any> { return Promise.resolve({ uri }); }
56 > getParsedPromptFile(textModel: ITextModel): ParsedPromptFile { throw new Error('Not implemented'); }
57 > registerContributedFile(type: PromptsType, uri: URI, extension: IExtensionDescription, name: string | undefined, description: string | undefined, when?: string, sessionTypes?: readonly string[]): IDisposable { throw new Error('Not implemented'); }
58 > getPromptLocationLabel(promptPath: IPromptPath): string { throw new Error('Not implemented'); }
59 > listNestedAgentMDs(token: CancellationToken): Promise<IAgentInstructionFile[]> { throw new Error('Not implemented'); }
60 > listAgentInstructions(token: CancellationToken): Promise<IAgentInstructionFile[]> { throw new Error('Not implemented'); }
61 > getAgentFileURIFromModeFile(oldURI: URI): URI | undefined { throw new Error('Not implemented'); }
62 > getDisabledPromptFiles(type: PromptsType): ResourceSet { throw new Error('Method not implemented.'); }
63 > setDisabledPromptFiles(type: PromptsType, uris: ResourceSet): void { throw new Error('Method not implemented.'); }
64 > registerPromptFileProvider(extension: IExtensionDescription, type: PromptsType, provider: { providePromptFiles: (context: IPromptFileContext, token: CancellationToken) => Promise<IPromptFileResource[] | undefined> }): IDisposable { throw new Error('Method not implemented.'); }
65 > findAgentSkills(token: CancellationToken): Promise<IAgentSkill[] | undefined> { throw new Error('Method not implemented.'); }
66 > // eslint-disable-next-line @typescript-eslint/no-explicit-any
67 > getHooks(_token: CancellationToken): Promise<any> { throw new Error('Method not implemented.'); }
68 > getInstructionFiles(_token: CancellationToken): Promise<readonly IInstructionFile[]> { throw new Error('Method not implemented.'); }
69 > getDiscoveryInfo(_type: PromptsType, _token: CancellationToken): Promise<IPromptDiscoveryInfo> { throw new Error('Method not implemented.'); }
70 > dispose(): void { }
71 > onDidChangeInstructions: Event<void> = Event.None;
72 > onDidChangeAgentInstructions: Event<void> = Event.None;
73 > onDidChangePromptFiles: Event<void> = Event.None;
74 > onDidChangeSkills: Event<void> = Event.None;
75 > onDidChangeHooks: Event<void> = Event.None;
76 > onDidChangeSlashCommands: Event<void> = Event.None;
src/vs/base/common/collections.ts 74 covered LOC · 18 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- collections.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * An interface for a JavaScript object that
8 > * acts a dictionary. The keys are strings.
9 > */
10 > export type IStringDictionary<V> = Record<string, V>;
11 >
12 > /**
13 > * An interface for a JavaScript object that
14 > * acts a dictionary. The keys are numbers.
15 > */
16 > export type INumberDictionary<V> = Record<number, V>;
17 >
18 > /**
19 > * Groups the collection into a dictionary based on the provided
20 > * group function.
21 > */
22 > export function groupBy<K extends string | number | symbol, V>(data: readonly V[], groupFn: (element: V) => K): Partial<Record<K, V[]>> {
23 const result: Partial<Record<K, V[]>> = Object.create(null);
24 for (const element of data) {
32 return result;
33 }
35 > export function groupByMap<K, V>(data: V[], groupFn: (element: V) => K): Map<K, V[]> {
36 const result = new Map<K, V[]>();
37 for (const element of data) {
46 return result;
47 }
49 > export function diffSets<T>(before: ReadonlySet<T>, after: ReadonlySet<T>): { removed: T[]; added: T[] } {
50 const removed: T[] = [];
51 const added: T[] = [];
62 return { removed, added };
63 }
65 > /**
66 > * Checks whether two sets contain exactly the same elements.
67 > *
68 > * @param a - The first set.
69 > * @param b - The second set.
70 > * @returns `true` if both sets have the same size and every element of `a` is also in `b`.
71 > */
72 > export function equalSets<T>(a: ReadonlySet<T>, b: ReadonlySet<T>): boolean {
73 if (a === b) {
74 return true;
84 return true;
85 }
87 > export function diffMaps<K, V>(before: Map<K, V>, after: Map<K, V>): { removed: V[]; added: V[] } {
88 const removed: V[] = [];
89 const added: V[] = [];
100 return { removed, added };
101 }
103 > /**
104 > * Computes the intersection of two sets.
105 > *
106 > * @param setA - The first set.
107 > * @param setB - The second iterable.
108 > * @returns A new set containing the elements that are in both `setA` and `setB`.
109 > */
110 > export function intersection<T>(setA: Set<T>, setB: Iterable<T>): Set<T> {
111 const result = new Set<T>();
112 for (const elem of setB) {
117 return result;
118 }
120 > export class SetWithKey<T> implements Set<T> {
121 > private _map = new Map<unknown, T>();
122 >
123 > constructor(values: T[], private toKey: (t: T) => unknown) {
124 for (const value of values) {
125 this.add(value);
126 }
127 }
129 > get size(): number {
130 return this._map.size;
131 }
133 > add(value: T): this {
134 const key = this.toKey(value);
135 this._map.set(key, value);
136 return this;
137 }
139 > delete(value: T): boolean {
140 return this._map.delete(this.toKey(value));
141 }
143 > has(value: T): boolean {
144 return this._map.has(this.toKey(value));
145 }
147 > *entries(): SetIterator<[T, T]> {
148 for (const entry of this._map.values()) {
149 yield [entry, entry];
150 }
151 }
153 > keys(): SetIterator<T> {
154 return this.values();
155 }
157 > *values(): SetIterator<T> {
158 for (const entry of this._map.values()) {
159 yield entry;
160 }
161 }
163 > clear(): void {
164 this._map.clear();
165 }
167 > forEach(callbackfn: (value: T, value2: T, set: Set<T>) => void, thisArg?: unknown): void {
168 this._map.forEach(entry => callbackfn.call(thisArg, entry, entry, this));
169 }
171 > [Symbol.iterator](): SetIterator<T> {
172 return this.values();
173 }
175 > [Symbol.toStringTag]: string = 'SetWithKey';
176 > }
src/vs/base/common/hash.ts 74 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hash.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { encodeHex, VSBuffer } from './buffer.js';
7 > import * as strings from './strings.js';
8 >
9 > type NotSyncHashable = ArrayBufferLike | ArrayBufferView;
10 >
11 > /**
12 > * Return a hash value for an object.
13 > *
14 > * Note that this should not be used for binary data types. Instead,
15 > * prefer {@link hashAsync}.
16 > */
17 > export function hash<T>(obj: T extends NotSyncHashable ? never : T): number {
18 return doHash(obj, 0);
19 }
20 > hash.ts
21 > export function doHash(obj: unknown, hashVal: number): number {
22 switch (typeof obj) {
23 case 'object':
40 }
41 }
42 > hash.ts
43 > export function numberHash(val: number, initialHashVal: number): number {
44 return (((initialHashVal << 5) - initialHashVal) + val) | 0; // hashVal * 31 + ch, keep as int32
45 }
46 > hash.ts
47 function booleanHash(b: boolean, initialHashVal: number): number {
48 return numberHash(b ? 433 : 863, initialHashVal);
49 }
50 > hash.ts
51 > export function stringHash(s: string, hashVal: number) {
52 hashVal = numberHash(149417, hashVal);
53 for (let i = 0, length = s.length; i < length; i++) {
56 return hashVal;
57 }
58 > hash.ts
59 function arrayHash(arr: unknown[], initialHashVal: number): number {
60 initialHashVal = numberHash(104579, initialHashVal);
61 return arr.reduce<number>((hashVal, item) => doHash(item, hashVal), initialHashVal);
62 }
63 > hash.ts
64 function objectHash(obj: object, initialHashVal: number): number {
65 initialHashVal = numberHash(181387, initialHashVal);
69 }, initialHashVal);
70 }
71 > hash.ts
72 >
73 >
74 > /** Hashes the input as SHA-1, returning a hex-encoded string. */
75 > export const hashAsync = (input: string | ArrayBufferView | VSBuffer) => {
76 // Note: I would very much like to expose a streaming interface for hashing
77 // generally, but this is not available in web crypto yet, see
96 return crypto.subtle.digest('sha-1', buff as ArrayBufferView<ArrayBuffer>).then(toHexString); // CodeQL [SM04514] we use sha1 here for validating old stored client state, not for security
97 };
98 > hash.ts
99 > const enum SHA1Constant {
100 > BLOCK_SIZE = 64, // 512 / 8
101 > UNICODE_REPLACEMENT = 0xFFFD,
102 > }
103 >
104 function leftRotate(value: number, bits: number, totalBits: number = 32): number {
105 // delta + bits = totalBits
112 return ((value << bits) | ((mask & value) >>> delta)) >>> 0;
113 }
114 > hash.ts
115 > function toHexString(buffer: ArrayBuffer): string;
116 > function toHexString(value: number, bitsize?: number): string;
117 function toHexString(bufferOrValue: ArrayBuffer | number, bitsize: number = 32): string {
118 if (bufferOrValue instanceof ArrayBuffer) {
122 return (bufferOrValue >>> 0).toString(16).padStart(bitsize / 4, '0');
123 }
124 > hash.ts
125 > /**
126 > * A SHA1 implementation that works with strings and does not allocate.
127 > *
128 > * Prefer to use {@link hashAsync} in async contexts
129 > */
130 > export class StringSHA1 {
131 > private static _bigBlock32 = new DataView(new ArrayBuffer(320)); // 80 * 4 = 320
132 >
133 > private _h0 = 0x67452301;
134 > private _h1 = 0xEFCDAB89;
135 > private _h2 = 0x98BADCFE;
136 > private _h3 = 0x10325476;
137 > private _h4 = 0xC3D2E1F0;
138 >
139 > private readonly _buff: Uint8Array;
140 > private readonly _buffDV: DataView;
141 > private _buffLen: number;
142 > private _totalLen: number;
143 > private _leftoverHighSurrogate: number;
144 > private _finished: boolean;
145 >
146 > constructor() {
147 this._buff = new Uint8Array(SHA1Constant.BLOCK_SIZE + 3 /* to fit any utf-8 */);
148 this._buffDV = new DataView(this._buff.buffer);
152 this._finished = false;
153 }
154 > hash.ts
155 > public update(str: string): void {
156 const strLen = str.length;
157 if (strLen === 0) {
208 this._leftoverHighSurrogate = leftoverHighSurrogate;
209 }
210 > hash.ts
211 > private _push(buff: Uint8Array, buffLen: number, codePoint: number): number {
212 if (codePoint < 0x0080) {
213 buff[buffLen++] = codePoint;
238 return buffLen;
239 }
240 > hash.ts
241 > public digest(): string {
242 if (!this._finished) {
243 this._finished = true;
253 return toHexString(this._h0) + toHexString(this._h1) + toHexString(this._h2) + toHexString(this._h3) + toHexString(this._h4);
254 }
255 > hash.ts
256 > private _wrapUp(): void {
257 this._buff[this._buffLen++] = 0x80;
258 this._buff.subarray(this._buffLen).fill(0);
271 this._step();
272 }
273 > hash.ts
274 > private _step(): void {
275 const bigBlock32 = StringSHA1._bigBlock32;
276 const data = this._buffDV;
322 this._h4 = (this._h4 + e) & 0xffffffff;
323 }
324 > } hash.ts
src/vs/base/common/iterator.ts 73 covered LOC · 28 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- iterator.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isIterable } from './types.js';
7 >
8 > export namespace Iterable {
9 >
10 > export function is<T = unknown>(thing: unknown): thing is Iterable<T> {
11 > return !!thing && typeof thing === 'object' && typeof (thing as Iterable<T>)[Symbol.iterator] === 'function'; iterator.ts
12 > }
14 > const _empty: Iterable<never> = Object.freeze([]);
15 > export function empty<T = never>(): readonly never[] {
16 return _empty as readonly never[];
17 }
19 > export function* single<T>(element: T): Iterable<T> {
20 yield element;
21 }
23 > export function wrap<T>(iterableOrElement: Iterable<T> | T): Iterable<T> {
24 if (is(iterableOrElement)) {
25 return iterableOrElement;
28 }
29 }
31 > export function from<T>(iterable: Iterable<T> | undefined | null): Iterable<T> {
32 return iterable ?? (_empty as Iterable<T>);
33 }
35 > export function* reverse<T>(array: ReadonlyArray<T>): Iterable<T> {
36 for (let i = array.length - 1; i >= 0; i--) {
37 yield array[i];
38 }
39 }
41 > export function isEmpty<T>(iterable: Iterable<T> | undefined | null): boolean {
42 return !iterable || iterable[Symbol.iterator]().next().done === true;
43 }
45 > export function first<T>(iterable: Iterable<T>): T | undefined {
46 return iterable[Symbol.iterator]().next().value;
47 }
49 > export function some<T>(iterable: Iterable<T>, predicate: (t: T, i: number) => unknown): boolean {
50 > let i = 0; iterator.ts
51 > for (const element of iterable) {
52 > if (predicate(element, i++)) { iterator.ts
53 > return true; iterator.ts
54 > }
55 > } iterator.ts
56 > return false; iterator.ts
57 > }
59 > export function every<T>(iterable: Iterable<T>, predicate: (t: T, i: number) => unknown): boolean {
60 let i = 0;
61 for (const element of iterable) {
66 return true;
67 }
69 > export function find<T, R extends T>(iterable: Iterable<T>, predicate: (t: T) => t is R): R | undefined;
70 > export function find<T>(iterable: Iterable<T>, predicate: (t: T) => boolean): T | undefined;
71 > export function find<T>(iterable: Iterable<T>, predicate: (t: T) => boolean): T | undefined {
72 for (const element of iterable) {
73 if (predicate(element)) {
78 return undefined;
79 }
81 > export function filter<T, R extends T>(iterable: Iterable<T>, predicate: (t: T) => t is R): Iterable<R>;
82 > export function filter<T>(iterable: Iterable<T>, predicate: (t: T) => boolean): Iterable<T>;
83 > export function* filter<T>(iterable: Iterable<T>, predicate: (t: T) => boolean): Iterable<T> {
84 for (const element of iterable) {
85 if (predicate(element)) {
88 }
89 }
91 > export function* map<T, R>(iterable: Iterable<T>, fn: (t: T, index: number) => R): Iterable<R> {
92 let index = 0;
93 for (const element of iterable) {
95 }
96 }
98 > export function* flatMap<T, R>(iterable: Iterable<T>, fn: (t: T, index: number) => Iterable<R>): Iterable<R> {
99 let index = 0;
100 for (const element of iterable) {
102 }
103 }
104 > iterator.ts
105 > export function* concat<T>(...iterables: (Iterable<T> | T)[]): Iterable<T> {
106 for (const item of iterables) {
107 if (isIterable(item)) {
112 }
113 }
114 > iterator.ts
115 > export function reduce<T, R>(iterable: Iterable<T>, reducer: (previousValue: R, currentValue: T) => R, initialValue: R): R {
116 let value = initialValue;
117 for (const element of iterable) {
120 return value;
121 }
122 > iterator.ts
123 > export function length<T>(iterable: Iterable<T>): number {
124 let count = 0;
125 for (const _ of iterable) {
128 return count;
129 }
130 > iterator.ts
131 > /**
132 > * Returns an iterable slice of the array, with the same semantics as `array.slice()`.
133 > */
134 > export function* slice<T>(arr: ReadonlyArray<T>, from: number, to = arr.length): Iterable<T> {
135 if (from < -arr.length) {
136 from = 0;
150 }
151 }
152 > iterator.ts
153 > /**
154 > * Consumes `atMost` elements from iterable and returns the consumed elements,
155 > * and an iterable for the rest of the elements.
156 > */
157 > export function consume<T>(iterable: Iterable<T>, atMost: number = Number.POSITIVE_INFINITY): [T[], Iterable<T>] {
158 const consumed: T[] = [];
159
176 return [consumed, { [Symbol.iterator]() { return iterator; } }];
177 }
178 > iterator.ts
179 > export async function asyncToArray<T>(iterable: AsyncIterable<T>): Promise<T[]> {
180 const result: T[] = [];
181 for await (const item of iterable) {
184 return result;
185 }
186 > iterator.ts
187 > export async function asyncToArrayFlat<T>(iterable: AsyncIterable<T[]>): Promise<T[]> {
188 let result: T[] = [];
189 for await (const item of iterable) {
src/vs/platform/agentHost/common/state/protocol/channels-resource-watch/state.ts 73 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > import type { URI } from '../common/state.js';
10 >
11 > // ─── Resource Watch Types ────────────────────────────────────────────────────
12 >
13 > /**
14 > * Full state for a single resource watch, returned when a client subscribes
15 > * to an `ahp-resource-watch:` URI.
16 > *
17 > * Watches are otherwise stateless: the watcher exists to deliver
18 > * {@link ResourceWatchChangedAction} events. The state carries only the
19 > * descriptor of what is being watched so a re-subscribing client can
20 > * recover the watch configuration after reconnecting.
21 > *
22 > * @category Resource Watch Types
23 > */
24 > export interface ResourceWatchState {
25 > /**
26 > * The URI being watched. For recursive watches this is the root of the
27 > * subtree; for non-recursive watches this is the single file or
28 > * directory.
29 > */
30 > root: URI;
31 > /**
32 > * `true` if the watcher reports changes for descendants of `root`;
33 > * `false` if it only reports changes to `root` itself (and, when
34 > * `root` is a directory, its direct children).
35 > */
36 > recursive: boolean;
37 > /**
38 > * Optional glob patterns or paths relative to `root` to exclude from
39 > * change reporting.
40 > */
41 > excludes?: { items: string[] };
42 > /**
43 > * Optional glob patterns or paths relative to `root` to restrict
44 > * change reporting to. Omit to report every change under `root`
45 > * subject to `excludes`.
46 > */
47 > includes?: { items: string[] };
48 > }
49 >
50 > // ─── Resource Change ─────────────────────────────────────────────────────────
51 >
52 > /**
53 > * Discriminant for {@link ResourceChange.type}.
54 > *
55 > * @category Resource Watch Types
56 > */
57 > export const enum ResourceChangeType {
58 > Added = 'added',
59 > Updated = 'updated',
60 > Deleted = 'deleted',
61 > }
62 >
63 > /**
64 > * A single change observed by a resource watcher.
65 > *
66 > * @category Resource Watch Types
67 > */
68 > export interface ResourceChange {
69 > /** The URI of the resource that changed. */
70 > uri: URI;
71 > /** The kind of change observed. */
72 > type: ResourceChangeType;
73 > }
src/vs/base/test/common/snapshot.ts 71 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- snapshot.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Lazy } from '../../common/lazy.js';
7 > import { FileAccess } from '../../common/network.js';
8 > import { URI } from '../../common/uri.js';
9 >
10 > declare const __readFileInTests: (path: string) => Promise<string>;
11 > declare const __writeFileInTests: (path: string, contents: string) => Promise<void>;
12 > declare const __readDirInTests: (path: string) => Promise<string[]>;
13 > declare const __unlinkInTests: (path: string) => Promise<void>;
14 > declare const __mkdirPInTests: (path: string) => Promise<void>;
15 >
16 > // setup on import so assertSnapshot has the current context without explicit passing
17 > let context: Lazy<SnapshotContext> | undefined;
18 > const sanitizeName = (name: string) => name.replace(/[^a-z0-9_-]/gi, '_');
19 > const normalizeCrlf = (str: string) => str.replace(/\r\n/g, '\n');
20 >
21 > export interface ISnapshotOptions {
22 > /** Name for snapshot file, rather than an incremented number */
23 > name?: string;
24 > /** Extension name of the snapshot file, defaults to `.snap` */
25 > extension?: string;
26 > }
27 >
28 > /**
29 > * This is exported only for tests against the snapshotting itself! Use
30 > * {@link assertSnapshot} as a consumer!
31 > */
32 > export class SnapshotContext {
33 > private nextIndex = 0;
34 > protected snapshotsDir: URI;
35 > private readonly namePrefix: string;
36 > private readonly usedNames = new Set();
37 >
38 > constructor(private readonly test: Mocha.Test | undefined) {
39 if (!test) {
40 throw new Error('assertSnapshot can only be used in a test');
51 this.snapshotsDir = URI.joinPath(src, ...[...parts.slice(0, -1), '__snapshots__']);
52 }
54 > public async assert(value: unknown, options?: ISnapshotOptions) {
55 const originalStack = new Error().stack!; // save to make the stack nicer on failure
56 const nameOrIndex = (options?.name ? sanitizeName(options.name) : this.nextIndex++);
85 }
86 }
88 > public async removeOldSnapshots() {
89 const contents = await __readDirInTests(this.snapshotsDir.fsPath);
90 const toDelete = contents.filter(f => f.startsWith(this.namePrefix) && !this.usedNames.has(f));
95 await Promise.all(toDelete.map(f => __unlinkInTests(URI.joinPath(this.snapshotsDir, f).fsPath)));
96 }
97 > } snapshot.ts
98 >
99 > const debugDescriptionSymbol = Symbol.for('debug.description');
100 >
101 function formatValue(value: unknown, level = 0, seen: unknown[] = []): string {
102 switch (typeof value) {
155 }
156 }
157 > snapshot.ts
158 > setup(function () {
159 > const currentTest = this.currentTest;
160 > context = new Lazy(() => new SnapshotContext(currentTest));
161 > });
162 > teardown(async function () {
163 > if (this.currentTest?.state === 'passed') {
164 > await context?.rawValue?.removeOldSnapshots();
165 > }
166 > context = undefined;
167 > });
168 >
169 > /**
170 > * Implements a snapshot testing utility. ⚠️ This is async! ⚠️
171 > *
172 > * The first time a snapshot test is run, it'll record the value it's called
173 > * with as the expected value. Subsequent runs will fail if the value differs,
174 > * but the snapshot can be regenerated by hand or using the Selfhost Test
175 > * Provider Extension which'll offer to update it.
176 > *
177 > * The snapshot will be associated with the currently running test and stored
178 > * in a `__snapshots__` directory next to the test file, which is expected to
179 > * be the first `.test.js` file in the callstack.
180 > */
181 > export function assertSnapshot(value: unknown, options?: ISnapshotOptions): Promise<void> {
182 if (!context) {
183 throw new Error('assertSnapshot can only be used in a test');
src/vs/editor/common/core/text/positionToOffsetImpl.ts 71 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- positionToOffsetImpl.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { findLastIdxMonotonous } from '../../../../base/common/arraysFind.js';
7 > import { StringEdit, StringReplacement } from '../edits/stringEdit.js';
8 > import { OffsetRange } from '../ranges/offsetRange.js';
9 > import { Position } from '../position.js';
10 > import { Range } from '../range.js';
11 > import type { TextReplacement, TextEdit } from '../edits/textEdit.js';
12 > import type { TextLength } from '../text/textLength.js';
13 >
14 > export abstract class PositionOffsetTransformerBase {
15 > abstract getOffset(position: Position): number;
16 >
17 > getOffsetRange(range: Range): OffsetRange {
18 return new OffsetRange(
19 this.getOffset(range.getStartPosition()),
21 );
22 }
24 > abstract getPosition(offset: number): Position;
25 >
26 > getRange(offsetRange: OffsetRange): Range {
27 return Range.fromPositions(
28 this.getPosition(offsetRange.start),
30 );
31 }
33 > getStringEdit(edit: TextEdit): StringEdit {
34 const edits = edit.replacements.map(e => this.getStringReplacement(e));
35 return new Deps.deps.StringEdit(edits);
36 }
38 > getStringReplacement(edit: TextReplacement): StringReplacement {
39 return new Deps.deps.StringReplacement(this.getOffsetRange(edit.range), edit.text);
40 }
42 > getTextReplacement(edit: StringReplacement): TextReplacement {
43 return new Deps.deps.TextReplacement(this.getRange(edit.replaceRange), edit.newText);
44 }
46 > getTextEdit(edit: StringEdit): TextEdit {
47 const edits = edit.replacements.map(e => this.getTextReplacement(e));
48 return new Deps.deps.TextEdit(edits);
49 }
51 >
52 > interface IDeps {
53 > StringEdit: typeof StringEdit;
54 > StringReplacement: typeof StringReplacement;
55 > TextReplacement: typeof TextReplacement;
56 > TextEdit: typeof TextEdit;
57 > TextLength: typeof TextLength;
58 > }
59 >
60 > class Deps {
61 > static _deps: IDeps | undefined = undefined;
62 > static get deps(): IDeps {
63 if (!this._deps) {
64 throw new Error('Dependencies not set. Call _setDependencies first.');
66 return this._deps;
67 }
69 >
70 > /** This is to break circular module dependencies. */
71 > export function _setPositionOffsetTransformerDependencies(deps: IDeps): void {
72 Deps._deps = deps;
73 }
75 > export class PositionOffsetTransformer extends PositionOffsetTransformerBase {
76 > private _lineStartOffsetByLineIdx: number[] | undefined;
77 > private _lineEndOffsetByLineIdx: number[] | undefined;
78 >
79 > constructor(public readonly text: string) {
80 super();
81 }
83 > private get lineStartOffsetByLineIdx(): number[] {
84 if (!this._lineStartOffsetByLineIdx) {
85 this._computeLineOffsets();
87 return this._lineStartOffsetByLineIdx!;
88 }
90 > private get lineEndOffsetByLineIdx(): number[] {
91 if (!this._lineEndOffsetByLineIdx) {
92 this._computeLineOffsets();
94 return this._lineEndOffsetByLineIdx!;
95 }
97 > private _computeLineOffsets(): void {
98 this._lineStartOffsetByLineIdx = [];
99 this._lineEndOffsetByLineIdx = [];
112 this._lineEndOffsetByLineIdx.push(this.text.length);
113 }
115 > override getOffset(position: Position): number {
116 const valPos = this._validatePosition(position);
117 return this.lineStartOffsetByLineIdx[valPos.lineNumber - 1] + valPos.column - 1;
118 }
120 > private _validatePosition(position: Position): Position {
121 if (position.lineNumber < 1) {
122 return new Position(1, 1);
136 return position;
137 }
139 > override getPosition(offset: number): Position {
140 const idx = findLastIdxMonotonous(this.lineStartOffsetByLineIdx, i => i <= offset);
141 const lineNumber = idx + 1;
143 return new Position(lineNumber, column);
144 }
146 > getTextLength(offsetRange: OffsetRange): TextLength {
147 return Deps.deps.TextLength.ofRange(this.getRange(offsetRange));
148 }
150 > get textLength(): TextLength {
151 const lineIdx = this.lineStartOffsetByLineIdx.length - 1;
152 return new Deps.deps.TextLength(lineIdx, this.text.length - this.lineStartOffsetByLineIdx[lineIdx]);
153 }
155 > getLineLength(lineNumber: number): number {
156 return this.lineEndOffsetByLineIdx[lineNumber - 1] - this.lineStartOffsetByLineIdx[lineNumber - 1];
157 }
src/vs/workbench/contrib/chat/common/attachments/chatVariables.ts 71 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatVariables.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationToken } from '../../../../../base/common/cancellation.js';
7 > import { ThemeIcon } from '../../../../../base/common/themables.js';
8 > import { URI } from '../../../../../base/common/uri.js';
9 > import { IRange } from '../../../../../editor/common/core/range.js';
10 > import { Location } from '../../../../../editor/common/languages.js';
11 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
12 > import { IChatModel } from '../model/chatModel.js';
13 > import { IChatContentReference, IChatProgressMessage } from '../chatService/chatService.js';
14 > import { IDiagnosticVariableEntryFilterData, StringChatContextValue, type IChatRequestVariableEntry } from './chatVariableEntries.js';
15 > import { ToolAndToolSetEnablementMap } from '../tools/languageModelToolsService.js';
16 >
17 > export interface IChatVariableData {
18 > id: string;
19 > name: string;
20 > icon?: ThemeIcon;
21 > fullName?: string;
22 > description: string;
23 > modelDescription?: string;
24 > canTakeArgument?: boolean;
25 > }
26 >
27 > export interface IChatRequestProblemsVariable {
28 > id: 'vscode.problems';
29 > filter: IDiagnosticVariableEntryFilterData;
30 > }
31 >
32 > export const isIChatRequestProblemsVariable = (obj: unknown): obj is IChatRequestProblemsVariable =>
33 typeof obj === 'object' && obj !== null && 'id' in obj && (obj as IChatRequestProblemsVariable).id === 'vscode.problems';
35 > export type IChatRequestVariableValue = string | URI | Location | Uint8Array | IChatRequestProblemsVariable | StringChatContextValue | unknown;
36 >
37 > export type IChatVariableResolverProgress =
38 > | IChatContentReference
39 > | IChatProgressMessage;
40 >
41 > export interface IChatVariableResolver {
42 > (messageText: string, arg: string | undefined, model: IChatModel, progress: (part: IChatVariableResolverProgress) => void, token: CancellationToken): Promise<IChatRequestVariableValue | undefined>;
43 > }
44 >
45 > export const IChatVariablesService = createDecorator<IChatVariablesService>('IChatVariablesService');
46 >
47 > export interface IChatVariablesService {
48 > _serviceBrand: undefined;
49 > getDynamicVariables(sessionResource: URI): ReadonlyArray<IDynamicVariable>;
50 > getSelectedToolAndToolSets(sessionResource: URI): ToolAndToolSetEnablementMap;
51 > }
52 >
53 > export interface IDynamicVariable {
54 > range: IRange;
55 > id: string;
56 > fullName?: string;
57 > icon?: ThemeIcon;
58 > modelDescription?: string;
59 > isFile?: boolean;
60 > isDirectory?: boolean;
61 > isAttachmentReference?: boolean;
62 > data: IChatRequestVariableValue;
63 > /**
64 > * Implementation-defined metadata that flows through to the resulting
65 > * {@link IChatRequestVariableEntry} and any {@link MessageAttachment}
66 > * derived from it. Used to round-trip provider-specific data attached
67 > * to chat input completions.
68 > */
69 > _meta?: Record<string, unknown>;
70 > }
71 >
72 > export function toAttachedContextDynamicVariable(entry: IChatRequestVariableEntry, range: IRange): IDynamicVariable {
73 return {
74 id: entry.id,
src/vs/workbench/contrib/chat/common/enablement.ts 71 covered LOC · 17 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- enablement.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Disposable } from '../../../../base/common/lifecycle.js';
7 > import { IObservable, IReader, ITransaction, transaction } from '../../../../base/common/observable.js';
8 > import { ObservableMemento, observableMemento } from '../../../../platform/observable/common/observableMemento.js';
9 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
10 >
11 > export const enum ContributionEnablementState {
12 > DisabledProfile,
13 > DisabledWorkspace,
14 > EnabledProfile,
15 > EnabledWorkspace,
16 > }
17 >
18 > export function isContributionEnabled(state: ContributionEnablementState): boolean {
19 return state === ContributionEnablementState.EnabledProfile || state === ContributionEnablementState.EnabledWorkspace;
20 }
22 > export function isContributionDisabled(state: ContributionEnablementState): boolean {
23 return !isContributionEnabled(state);
24 }
26 > export interface IEnablementModel {
27 > readEnabled(key: string, reader?: IReader): ContributionEnablementState;
28 > setEnabled(key: string, state: ContributionEnablementState, tx?: ITransaction): void;
29 > remove(key: string): void;
30 > }
31 >
32 > type EnablementMap = ReadonlyMap<string, boolean>;
33 >
34 function mapToStorage(value: EnablementMap): string {
35 return JSON.stringify([...value]);
36 }
38 function mapFromStorage(value: string): EnablementMap {
39 const parsed = JSON.parse(value);
40 return new Map(Array.isArray(parsed) ? parsed : []);
41 }
43 > /**
44 > * A reusable enablement model for string-keyed contributions. Uses
45 > * `observableMemento` to persist enable/disable state in both profile-scoped
46 > * and workspace-scoped storage.
47 > *
48 > * Resolution order: if a workspace-scoped entry exists for a key, it wins.
49 > * Otherwise, the profile-scoped entry is used. The default (absence of any
50 > * entry) is {@link ContributionEnablementState.EnabledProfile}.
51 > */
52 > export class EnablementModel extends Disposable implements IEnablementModel {
53 > private readonly _profileState: ObservableMemento<EnablementMap>;
54 > private readonly _workspaceState: ObservableMemento<EnablementMap>;
55 >
56 > constructor(
57 storageKey: string,
58 @IStorageService storageService: IStorageService,
75 );
76 }
78 > readEnabled(key: string, reader?: IReader): ContributionEnablementState {
79 return this.readEnabledWithWorkspaceKey(key, key, reader);
80 }
82 > readEnabledWithWorkspaceKey(profileKey: string, workspaceKey: string | undefined, reader?: IReader): ContributionEnablementState {
83 const wsMap = this._workspaceState.read(reader);
84 if (workspaceKey !== undefined && wsMap.has(workspaceKey)) {
97 return ContributionEnablementState.EnabledProfile;
98 }
100 > setEnabled(key: string, state: ContributionEnablementState, tx?: ITransaction): void {
101 this.setEnabledWithWorkspaceKey(key, key, state, tx);
102 }
104 > setEnabledWithWorkspaceKey(profileKey: string, workspaceKey: string | undefined, state: ContributionEnablementState, tx?: ITransaction): void {
105 switch (state) {
106 case ContributionEnablementState.EnabledProfile: {
140
141 }
143 > remove(key: string): void {
144 this._deleteFromMap(this._profileState, key);
145 this._deleteFromMap(this._workspaceState, key);
146 }
148 > private _setInMap(memento: ObservableMemento<EnablementMap>, key: string, value: boolean, tx?: ITransaction): void {
149 const current = memento.get();
150 if (current.get(key) === value) {
155 memento.set(next, tx);
156 }
158 > private _deleteFromMap(memento: ObservableMemento<EnablementMap>, key: string, tx?: ITransaction): void {
159 const current = memento.get();
160 if (!current.has(key)) {
165 memento.set(next, tx);
166 }
167 > } enablement.ts
168 >
169 > export class CollisionEnablementModel implements IEnablementModel {
170 >
171 > constructor(
172 private readonly _base: IEnablementModel,
173 private readonly _collisionGroups: IObservable<ReadonlyMap<string, readonly string[]>>,
174 ) { }
176 > readEnabled(key: string, reader?: IReader): ContributionEnablementState {
177 const baseState = this._base.readEnabled(key, reader);
178
196 return baseState;
197 }
199 > setEnabled(key: string, state: ContributionEnablementState, tx?: ITransaction): void {
200 const isEnabling = state === ContributionEnablementState.EnabledProfile || state === ContributionEnablementState.EnabledWorkspace;
201 const group = isEnabling ? this._collisionGroups.get().get(key) : undefined;
src/vs/workbench/contrib/chat/common/model/chatUri.ts 70 covered LOC · 18 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatUri.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { encodeBase64, VSBuffer, decodeBase64 } from '../../../../../base/common/buffer.js';
7 > import { Schemas } from '../../../../../base/common/network.js';
8 > import { URI } from '../../../../../base/common/uri.js';
9 > import { localChatSessionType } from '../chatSessionsService.js';
10 >
11 > type ChatSessionIdentifier = {
12 > readonly chatSessionType: string;
13 > readonly sessionId: string;
14 > };
15 >
16 >
17 > export namespace LocalChatSessionUri {
18 >
19 > export const scheme = Schemas.vscodeLocalChatSession;
20 >
21 > export function forSession(sessionId: string): URI {
22 const encodedId = encodeBase64(VSBuffer.wrap(new TextEncoder().encode(sessionId)), false, true);
23 return URI.from({ scheme, authority: localChatSessionType, path: '/' + encodedId });
24 }
25 > chatUri.ts
26 > export function getNewSessionUri(): URI {
27 const handle = Math.floor(Math.random() * 1e9);
28 return forSession(`chat-${handle}`);
29 }
30 > chatUri.ts
31 > export function parseLocalSessionId(resource: URI): string | undefined {
32 > const parsed = parse(resource); chatUri.ts
33 > return parsed?.chatSessionType === localChatSessionType ? parsed.sessionId : undefined;
34 > }
35 > chatUri.ts
36 > export function isLocalSession(resource: URI): boolean {
37 > return !!parseLocalSessionId(resource); chatUri.ts
38 > }
39 > chatUri.ts
40 > function parse(resource: URI): ChatSessionIdentifier | undefined {
41 > if (resource.scheme !== scheme) { chatUri.ts
42 > return undefined; chatUri.ts
43 > }
44
45 if (!resource.authority) {
55 const decodedSessionId = decodeBase64(parts[1]);
56 return { chatSessionType, sessionId: new TextDecoder().decode(decodedSessionId.buffer) };
57 > } chatUri.ts
58 > } chatUri.ts
59 >
60 > /**
61 > * Converts a chat session resource URI to a string ID.
62 > *
63 > * This exists mainly for backwards compatibility with existing code that uses string IDs in telemetry and storage.
64 > */
65 > export function chatSessionResourceToId(resource: URI): string {
66 > // If we have a local session, prefer using just the id part chatUri.ts
67 > const localId = LocalChatSessionUri.parseLocalSessionId(resource);
68 > if (localId) {
69 return localId;
70 }
71 > chatUri.ts
72 > return resource.toString();
73 > }
74 > chatUri.ts
75 > /**
76 > * Extracts the chat session type from a resource URI.
77 > *
78 > * @param resource - The chat session resource URI
79 > * @returns The session type string. Returns `localChatSessionType` for local sessions
80 > * (vscodeChatEditor and vscodeLocalChatSession schemes), or the scheme/authority
81 > * for contributed sessions.
82 > */
83 > export function getChatSessionType(resource: URI): string {
84 > if (resource.scheme === Schemas.vscodeChatEditor) { chatUri.ts
85 return localChatSessionType;
86 }
87 > chatUri.ts
88 > if (resource.scheme === LocalChatSessionUri.scheme) {
89 return resource.authority || localChatSessionType;
90 }
91 > chatUri.ts
92 > return resource.scheme;
93 > }
94 > chatUri.ts
95 > export function isUntitledChatSession(resource: URI): boolean {
96 return resource.path.startsWith('/untitled-');
97 }
src/vs/base/common/observableInternal/logging/logging.ts 69 covered LOC · 18 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- logging.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { AutorunObserver } from '../reactions/autorunImpl.js';
7 > import { IObservable } from '../base.js';
8 > import { TransactionImpl } from '../transaction.js';
9 > import type { Derived } from '../observables/derivedImpl.js';
10 > import { DebugLocation } from '../debugLocation.js';
11 >
12 > let globalObservableLogger: IObservableLogger | undefined;
13 >
14 > export function addLogger(logger: IObservableLogger): void {
15 if (!globalObservableLogger) {
16 globalObservableLogger = logger;
21 }
22 }
23 > logging.ts
24 > export function getLogger(): IObservableLogger | undefined {
25 > return globalObservableLogger; logging.ts
26 > }
27 > logging.ts
28 > let globalObservableLoggerFn: ((obs: IObservable<any>) => void) | undefined = undefined;
29 > export function setLogObservableFn(fn: (obs: IObservable<any>) => void): void {
30 > globalObservableLoggerFn = fn;
31 > }
32 >
33 > export function logObservable(obs: IObservable<any>): void {
34 if (globalObservableLoggerFn) {
35 globalObservableLoggerFn(obs);
36 }
37 }
38 > logging.ts
39 > export interface IChangeInformation {
40 > oldValue: unknown;
41 > newValue: unknown;
42 > change: unknown;
43 > didChange: boolean;
44 > hadValue: boolean;
45 > }
46 >
47 > export interface IObservableLogger {
48 > handleObservableCreated(observable: IObservable<any>, location: DebugLocation): void;
49 > handleOnListenerCountChanged(observable: IObservable<any>, newCount: number): void;
50 >
51 > handleObservableUpdated(observable: IObservable<any>, info: IChangeInformation): void;
52 >
53 > handleAutorunCreated(autorun: AutorunObserver, location: DebugLocation): void;
54 > handleAutorunDisposed(autorun: AutorunObserver): void;
55 > handleAutorunDependencyChanged(autorun: AutorunObserver, observable: IObservable<any>, change: unknown): void;
56 > handleAutorunStarted(autorun: AutorunObserver): void;
57 > handleAutorunFinished(autorun: AutorunObserver): void;
58 >
59 > handleDerivedDependencyChanged(derived: Derived<any, any, any>, observable: IObservable<any>, change: unknown): void;
60 > handleDerivedCleared(observable: Derived<any, any, any>): void;
61 >
62 > handleBeginTransaction(transaction: TransactionImpl): void;
63 > handleEndTransaction(transaction: TransactionImpl): void;
64 > }
65 >
66 > class ComposedLogger implements IObservableLogger {
67 > constructor(
68 public readonly loggers: IObservableLogger[],
69 ) { }
70 > logging.ts
71 > handleObservableCreated(observable: IObservable<any>, location: DebugLocation): void {
72 for (const logger of this.loggers) {
73 logger.handleObservableCreated(observable, location);
74 }
75 }
76 > handleOnListenerCountChanged(observable: IObservable<any>, newCount: number): void { logging.ts
77 for (const logger of this.loggers) {
78 logger.handleOnListenerCountChanged(observable, newCount);
79 }
80 }
81 > handleObservableUpdated(observable: IObservable<any>, info: IChangeInformation): void { logging.ts
82 for (const logger of this.loggers) {
83 logger.handleObservableUpdated(observable, info);
84 }
85 }
86 > handleAutorunCreated(autorun: AutorunObserver, location: DebugLocation): void { logging.ts
87 for (const logger of this.loggers) {
88 logger.handleAutorunCreated(autorun, location);
89 }
90 }
91 > handleAutorunDisposed(autorun: AutorunObserver): void { logging.ts
92 for (const logger of this.loggers) {
93 logger.handleAutorunDisposed(autorun);
94 }
95 }
96 > handleAutorunDependencyChanged(autorun: AutorunObserver, observable: IObservable<any>, change: unknown): void { logging.ts
97 for (const logger of this.loggers) {
98 logger.handleAutorunDependencyChanged(autorun, observable, change);
99 }
100 }
101 > handleAutorunStarted(autorun: AutorunObserver): void { logging.ts
102 for (const logger of this.loggers) {
103 logger.handleAutorunStarted(autorun);
104 }
105 }
106 > handleAutorunFinished(autorun: AutorunObserver): void { logging.ts
107 for (const logger of this.loggers) {
108 logger.handleAutorunFinished(autorun);
109 }
110 }
111 > handleDerivedDependencyChanged(derived: Derived<any>, observable: IObservable<any>, change: unknown): void { logging.ts
112 for (const logger of this.loggers) {
113 logger.handleDerivedDependencyChanged(derived, observable, change);
114 }
115 }
116 > handleDerivedCleared(observable: Derived<any>): void { logging.ts
117 for (const logger of this.loggers) {
118 logger.handleDerivedCleared(observable);
119 }
120 }
121 > handleBeginTransaction(transaction: TransactionImpl): void { logging.ts
122 for (const logger of this.loggers) {
123 logger.handleBeginTransaction(transaction);
124 }
125 }
126 > handleEndTransaction(transaction: TransactionImpl): void { logging.ts
127 for (const logger of this.loggers) {
128 logger.handleEndTransaction(transaction);
129 }
130 }
131 > } logging.ts
src/vs/base/common/observableInternal/transaction.ts 68 covered LOC · 19 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- transaction.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { handleBugIndicatingErrorRecovery, IObservable, IObserver, ITransaction } from './base.js';
7 > import { getFunctionName } from './debugName.js';
8 > import { getLogger } from './logging/logging.js';
9 >
10 > /**
11 > * Starts a transaction in which many observables can be changed at once.
12 > * {@link fn} should start with a JS Doc using `@description` to give the transaction a debug name.
13 > * Reaction run on demand or when the transaction ends.
14 > */
15 >
16 > export function transaction(fn: (tx: ITransaction) => void, getDebugName?: () => string): void {
17 > const tx = new TransactionImpl(fn, getDebugName); transaction.ts
18 > try {
19 > fn(tx);
20 > } finally {
21 > tx.finish();
22 > }
23 > }
24 > let _globalTransaction: ITransaction | undefined = undefined; transaction.ts
25 >
26 > export function globalTransaction(fn: (tx: ITransaction) => void) {
27 if (_globalTransaction) {
28 fn(_globalTransaction);
40 }
41 }
42 > /** @deprecated */ transaction.ts
43 >
44 export async function asyncTransaction(fn: (tx: ITransaction) => Promise<void>, getDebugName?: () => string): Promise<void> {
45 const tx = new TransactionImpl(fn, getDebugName);
50 }
51 }
52 > /** transaction.ts
53 > * Allows to chain transactions.
54 > */
55 >
56 > export function subtransaction(tx: ITransaction | undefined, fn: (tx: ITransaction) => void, getDebugName?: () => string): void {
57 > if (!tx) { transaction.ts
58 > transaction(fn, getDebugName);
59 > } else {
60 fn(tx);
61 }
62 > } export class TransactionImpl implements ITransaction { transaction.ts
63 > private _updatingObservers: { observer: IObserver; observable: IObservable<any> }[] | null = [];
64 >
65 > constructor(public readonly _fn: Function, private readonly _getDebugName?: () => string) {
66 > getLogger()?.handleBeginTransaction(this); transaction.ts
67 > }
69 > public getDebugName(): string | undefined {
70 if (this._getDebugName) {
71 return this._getDebugName();
73 return getFunctionName(this._fn);
74 }
76 > public updateObserver(observer: IObserver, observable: IObservable<any>): void {
77 > if (!this._updatingObservers) { transaction.ts
78 // This happens when a transaction is used in a callback or async function.
79 // If an async transaction is used, make sure the promise awaits all users of the transaction (e.g. no race).
85 return;
86 }
88 > // When this gets called while finish is active, they will still get considered
89 > this._updatingObservers.push({ observer, observable });
90 > observer.beginUpdate(observable);
91 > }
93 > public finish(): void {
94 > const updatingObservers = this._updatingObservers; transaction.ts
95 > if (!updatingObservers) {
96 handleBugIndicatingErrorRecovery('transaction.finish() has already been called!');
97 return;
98 }
100 > for (let i = 0; i < updatingObservers.length; i++) {
101 > const { observer, observable } = updatingObservers[i]; transaction.ts
102 > observer.endUpdate(observable);
103 > }
104 > // Prevent anyone from updating observers from now on. transaction.ts
105 > this._updatingObservers = null;
106 > getLogger()?.handleEndTransaction(this);
107 > }
109 > public debugGetUpdatingObservers() {
110 return this._updatingObservers;
111 }
112 > } transaction.ts
113
src/vs/platform/mcp/common/mcpManagementIpc.ts 67 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpManagementIpc.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from '../../../base/common/event.js';
7 > import { cloneAndChange } from '../../../base/common/objects.js';
8 > import { URI, UriComponents } from '../../../base/common/uri.js';
9 > import { DefaultURITransformer, IURITransformer, transformAndReviveIncomingURIs } from '../../../base/common/uriIpc.js';
10 > import { IChannel, IServerChannel } from '../../../base/parts/ipc/common/ipc.js';
11 > import { ILogService } from '../../log/common/log.js';
12 > import { RemoteAgentConnectionContext } from '../../remote/common/remoteAgentEnvironment.js';
13 > import { DidUninstallMcpServerEvent, IGalleryMcpServer, ILocalMcpServer, IMcpManagementService, IInstallableMcpServer, InstallMcpServerEvent, InstallMcpServerResult, InstallOptions, UninstallMcpServerEvent, UninstallOptions, IAllowedMcpServersService } from './mcpManagement.js';
14 > import { AbstractMcpManagementService } from './mcpManagementService.js';
15 >
16 > function transformIncomingURI(uri: UriComponents, transformer: IURITransformer | null): URI;
17 > function transformIncomingURI(uri: UriComponents | undefined, transformer: IURITransformer | null): URI | undefined;
18 function transformIncomingURI(uri: UriComponents | undefined, transformer: IURITransformer | null): URI | undefined {
19 return uri ? URI.revive(transformer ? transformer.transformIncoming(uri) : uri) : undefined;
20 }
22 function transformIncomingServer(mcpServer: ILocalMcpServer, transformer: IURITransformer | null): ILocalMcpServer {
23 transformer = transformer ? transformer : DefaultURITransformer;
26 return { ...transformed, ...{ manifest } };
27 }
29 function transformIncomingOptions<O extends { mcpResource?: UriComponents }>(options: O | undefined, transformer: IURITransformer | null): O | undefined {
30 return options?.mcpResource ? transformAndReviveIncomingURIs(options, transformer ?? DefaultURITransformer) : options;
31 }
33 function transformOutgoingExtension(extension: ILocalMcpServer, transformer: IURITransformer | null): ILocalMcpServer {
34 return transformer ? cloneAndChange(extension, value => value instanceof URI ? transformer.transformOutgoingURI(value) : undefined) : extension;
35 }
37 function transformOutgoingURI(uri: URI, transformer: IURITransformer | null): URI {
38 return transformer ? transformer.transformOutgoingURI(uri) : uri;
39 }
41 > export class McpManagementChannel<TContext = RemoteAgentConnectionContext | string> implements IServerChannel<TContext> {
42 > readonly onInstallMcpServer: Event<InstallMcpServerEvent>;
43 > readonly onDidInstallMcpServers: Event<readonly InstallMcpServerResult[]>;
44 > readonly onDidUpdateMcpServers: Event<readonly InstallMcpServerResult[]>;
45 > readonly onUninstallMcpServer: Event<UninstallMcpServerEvent>;
46 > readonly onDidUninstallMcpServer: Event<DidUninstallMcpServerEvent>;
47 >
48 > constructor(private service: IMcpManagementService, private getUriTransformer: (requestContext: TContext) => IURITransformer | null) {
49 this.onInstallMcpServer = Event.buffer(service.onInstallMcpServer, 'onInstallMcpServer', true);
50 this.onDidInstallMcpServers = Event.buffer(service.onDidInstallMcpServers, 'onDidInstallMcpServers', true);
53 this.onDidUninstallMcpServer = Event.buffer(service.onDidUninstallMcpServer, 'onDidUninstallMcpServer', true);
54 }
56 > listen<T>(context: TContext, event: string): Event<T> {
57 const uriTransformer = this.getUriTransformer(context);
58 switch (event) {
92 throw new Error('Invalid listen');
93 }
95 > async call<T>(context: TContext, command: string, args?: unknown): Promise<T> {
96 const uriTransformer: IURITransformer | null = this.getUriTransformer(context);
97 const argsArray = Array.isArray(args) ? args : [];
117 throw new Error('Invalid call');
118 }
120 >
121 > export class McpManagementChannelClient extends AbstractMcpManagementService implements IMcpManagementService {
122 >
123 > declare readonly _serviceBrand: undefined;
124 >
125 > private readonly _onInstallMcpServer = this._register(new Emitter<InstallMcpServerEvent>());
126 > get onInstallMcpServer() { return this._onInstallMcpServer.event; }
127 >
128 > private readonly _onDidInstallMcpServers = this._register(new Emitter<readonly InstallMcpServerResult[]>());
129 > get onDidInstallMcpServers() { return this._onDidInstallMcpServers.event; }
130 >
131 > private readonly _onUninstallMcpServer = this._register(new Emitter<UninstallMcpServerEvent>());
132 > get onUninstallMcpServer() { return this._onUninstallMcpServer.event; }
133 >
134 > private readonly _onDidUninstallMcpServer = this._register(new Emitter<DidUninstallMcpServerEvent>());
135 > get onDidUninstallMcpServer() { return this._onDidUninstallMcpServer.event; }
136 >
137 > private readonly _onDidUpdateMcpServers = this._register(new Emitter<InstallMcpServerResult[]>());
138 > get onDidUpdateMcpServers() { return this._onDidUpdateMcpServers.event; }
139 >
140 > constructor(
141 private readonly channel: IChannel,
142 @IAllowedMcpServersService allowedMcpServersService: IAllowedMcpServersService,
150 this._register(this.channel.listen<DidUninstallMcpServerEvent>('onDidUninstallMcpServer')(e => this._onDidUninstallMcpServer.fire(({ ...e, mcpResource: transformIncomingURI(e.mcpResource, null) }))));
151 }
153 > install(server: IInstallableMcpServer, options?: InstallOptions): Promise<ILocalMcpServer> {
154 return Promise.resolve(this.channel.call<ILocalMcpServer>('install', [server, options])).then(local => transformIncomingServer(local, null));
155 }
157 > installFromGallery(extension: IGalleryMcpServer, installOptions?: InstallOptions): Promise<ILocalMcpServer> {
158 return Promise.resolve(this.channel.call<ILocalMcpServer>('installFromGallery', [extension, installOptions])).then(local => transformIncomingServer(local, null));
159 }
161 > uninstall(extension: ILocalMcpServer, options?: UninstallOptions): Promise<void> {
162 return Promise.resolve(this.channel.call<void>('uninstall', [extension, options]));
163 }
165 > getInstalled(mcpResource?: URI): Promise<ILocalMcpServer[]> {
166 return Promise.resolve(this.channel.call<ILocalMcpServer[]>('getInstalled', [mcpResource]))
167 .then(servers => servers.map(server => transformIncomingServer(server, null)));
168 }
170 > updateMetadata(local: ILocalMcpServer, gallery: IGalleryMcpServer, mcpResource?: URI): Promise<ILocalMcpServer> {
171 return Promise.resolve(this.channel.call<ILocalMcpServer>('updateMetadata', [local, gallery, mcpResource])).then(local => transformIncomingServer(local, null));
172 }
src/vs/base/common/objects.ts 65 covered LOC · 13 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- objects.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isTypedArray, isObject, isUndefinedOrNull } from './types.js';
7 >
8 > export function deepClone<T>(obj: T): T {
9 if (!obj || typeof obj !== 'object') {
10 return obj;
19 return result;
20 }
21 > objects.ts
22 > export function deepFreeze<T>(obj: T): T {
23 if (!obj || typeof obj !== 'object') {
24 return obj;
39 return obj;
40 }
41 > objects.ts
42 > const _hasOwnProperty = Object.prototype.hasOwnProperty;
43 >
44 >
45 > export function cloneAndChange(obj: any, changer: (orig: any) => any): any {
46 return _cloneAndChange(obj, changer, new Set());
47 }
48 > objects.ts
49 function _cloneAndChange(obj: any, changer: (orig: any) => any, seen: Set<any>): any {
50 if (isUndefinedOrNull(obj)) {
82 return obj;
83 }
84 > objects.ts
85 > /**
86 > * Copies all properties of source into destination. The optional parameter "overwrite" allows to control
87 > * if existing properties on the destination should be overwritten or not. Defaults to true (overwrite).
88 > */
89 > export function mixin(destination: any, source: any, overwrite: boolean = true): any {
90 if (!isObject(destination)) {
91 return source;
109 return destination;
110 }
111 > objects.ts
112 > export function equals(one: any, other: any): boolean {
113 if (one === other) {
114 return true;
162 return true;
163 }
164 > objects.ts
165 > /**
166 > * Calls `JSON.Stringify` with a replacer to break apart any circular references.
167 > * This prevents `JSON`.stringify` from throwing the exception
168 > * "Uncaught TypeError: Converting circular structure to JSON"
169 > */
170 > export function safeStringify(obj: any): string {
171 const seen = new Set<any>();
172 return JSON.stringify(obj, (key, value) => {
184 });
185 }
186 > objects.ts
187 > /**
188 > * Like `JSON.stringify`, but with deterministic ordering of object keys so that
189 > * structurally equal inputs always produce the same string. Useful for cache
190 > * keys derived from arbitrary object payloads.
191 > *
192 > * - Object keys are sorted at every level of nesting.
193 > * - Properties whose value is `undefined` are omitted (matching `JSON.stringify`).
194 > * - Circular references are replaced with the string `"[Circular]"` to avoid
195 > * throwing.
196 > * - A top-level `undefined` returns the string `'undefined'`; any other
197 > * stringification failure returns the empty string.
198 > */
199 > export function stableStringify(value: unknown): string {
200 if (value === undefined) {
201 return 'undefined';
207 }
208 }
209 > objects.ts
210 function _stableStringify(value: unknown, seen: WeakSet<object>): string {
211 if (value === null || typeof value !== 'object') {
230 return '{' + parts.join(',') + '}';
231 }
232 > objects.ts
233 > type obj = { [key: string]: any };
234 > /**
235 > * Returns an object that has keys for each value that is different in the base object. Keys
236 > * that do not exist in the target but in the base object are not considered.
237 > *
238 > * Note: This is not a deep-diffing method, so the values are strictly taken into the resulting
239 > * object if they differ.
240 > *
241 > * @param base the object to diff against
242 > * @param obj the object to use for diffing
243 > */
244 > export function distinct(base: obj, target: obj): obj {
245 const result = Object.create(null);
246
261 return result;
262 }
263 > objects.ts
264 > export function getCaseInsensitive(target: obj, key: string): unknown {
265 const lowercaseKey = key.toLowerCase();
266 const equivalentKey = Object.keys(target).find(k => k.toLowerCase() === lowercaseKey);
267 return equivalentKey ? target[equivalentKey] : target[key];
268 }
269 > objects.ts
270 > export function filter(obj: obj, predicate: (key: string, value: any) => boolean): obj {
271 const result = Object.create(null);
272 for (const [key, value] of Object.entries(obj)) {
277 return result;
278 }
279 > objects.ts
280 > export function mapValues<T extends {}, R>(obj: T, fn: (value: T[keyof T], key: string) => R): { [K in keyof T]: R } {
281 const result: { [key: string]: R } = {};
282 for (const [key, value] of Object.entries(obj)) {
src/vs/base/common/uriIpc.ts 65 covered LOC · 15 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- uriIpc.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer } from './buffer.js';
7 > import { MarshalledObject } from './marshalling.js';
8 > import { MarshalledId } from './marshallingIds.js';
9 > import { URI, UriComponents } from './uri.js';
10 >
11 > export interface IURITransformer {
12 > transformIncoming(uri: UriComponents): UriComponents;
13 > transformOutgoing(uri: UriComponents): UriComponents;
14 > transformOutgoingURI(uri: URI): URI;
15 > transformOutgoingScheme(scheme: string): string;
16 > }
17 >
18 > export interface UriParts {
19 > scheme: string;
20 > authority?: string;
21 > path?: string;
22 > query?: string;
23 > fragment?: string;
24 > }
25 >
26 > export interface IRawURITransformer {
27 > transformIncoming(uri: UriParts): UriParts;
28 > transformOutgoing(uri: UriParts): UriParts;
29 > transformOutgoingScheme(scheme: string): string;
30 > }
31 >
32 function toJSON(uri: URI): UriComponents {
33 return uri.toJSON();
34 }
35 > uriIpc.ts
36 > export class URITransformer implements IURITransformer {
37 >
38 > private readonly _uriTransformer: IRawURITransformer;
39 >
40 > constructor(uriTransformer: IRawURITransformer) {
41 this._uriTransformer = uriTransformer;
42 }
43 > uriIpc.ts
44 > public transformIncoming(uri: UriComponents): UriComponents {
45 const result = this._uriTransformer.transformIncoming(uri);
46 return (result === uri ? uri : toJSON(URI.from(result)));
47 }
48 > uriIpc.ts
49 > public transformOutgoing(uri: UriComponents): UriComponents {
50 const result = this._uriTransformer.transformOutgoing(uri);
51 return (result === uri ? uri : toJSON(URI.from(result)));
52 }
53 > uriIpc.ts
54 > public transformOutgoingURI(uri: URI): URI {
55 const result = this._uriTransformer.transformOutgoing(uri);
56 return (result === uri ? uri : URI.from(result));
57 }
58 > uriIpc.ts
59 > public transformOutgoingScheme(scheme: string): string {
60 return this._uriTransformer.transformOutgoingScheme(scheme);
61 }
62 > } uriIpc.ts
63 >
64 > export const DefaultURITransformer: IURITransformer = new class {
65 > transformIncoming(uri: UriComponents) {
66 return uri;
67 }
68 > uriIpc.ts
69 > transformOutgoing(uri: UriComponents): UriComponents {
70 return uri;
71 }
72 > uriIpc.ts
73 > transformOutgoingURI(uri: URI): URI {
74 return uri;
75 }
76 > uriIpc.ts
77 > transformOutgoingScheme(scheme: string): string {
78 return scheme;
79 }
80 > }; uriIpc.ts
81 >
82 function _transformOutgoingURIs(obj: any, transformer: IURITransformer, depth: number): any {
83
104 return null;
105 }
106 > uriIpc.ts
107 > export function transformOutgoingURIs<T>(obj: T, transformer: IURITransformer): T {
108 const result = _transformOutgoingURIs(obj, transformer, 0);
109 if (result === null) {
113 return result;
114 }
115 > uriIpc.ts
116 >
117 function _transformIncomingURIs(obj: any, transformer: IURITransformer, revive: boolean, depth: number): any {
118
144 return null;
145 }
146 > uriIpc.ts
147 > export function transformIncomingURIs<T>(obj: T, transformer: IURITransformer): T {
148 const result = _transformIncomingURIs(obj, transformer, false, 0);
149 if (result === null) {
153 return result;
154 }
155 > uriIpc.ts
156 > export function transformAndReviveIncomingURIs<T>(obj: T, transformer: IURITransformer): T {
157 const result = _transformIncomingURIs(obj, transformer, true, 0);
158 if (result === null) {
src/vs/workbench/contrib/chat/common/promptSyntax/hookClaudeCompat.ts 65 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hookClaudeCompat.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../../../base/common/uri.js';
7 > import { toHookType, IHookCommand, extractHookCommandsFromItem } from './hookSchema.js';
8 > import { HOOKS_BY_TARGET, HookType } from './hookTypes.js';
9 > import { Target } from './promptTypes.js';
10 >
11 > export { extractHookCommandsFromItem };
12 >
13 > /**
14 > * Cached inverse mapping from HookType to Claude hook type name.
15 > * Lazily computed on first access.
16 > */
17 > let _hookTypeToClaudeName: Map<HookType, string> | undefined;
18 >
19 function getHookTypeToClaudeNameMap(): Map<HookType, string> {
20 if (!_hookTypeToClaudeName) {
26 return _hookTypeToClaudeName;
27 }
29 > /**
30 > * Resolves a Claude hook type name to our abstract HookType.
31 > */
32 > export function resolveClaudeHookType(name: string): HookType | undefined {
33 return HOOKS_BY_TARGET[Target.Claude][name];
34 }
36 > /**
37 > * Gets the Claude hook type name for a given abstract HookType.
38 > * Returns undefined if the hook type is not supported in Claude.
39 > */
40 > export function getClaudeHookTypeName(hookType: HookType): string | undefined {
41 return getHookTypeToClaudeNameMap().get(hookType);
42 }
44 > /**
45 > * Result of parsing Claude hooks file.
46 > */
47 > export interface IParseClaudeHooksResult {
48 > /**
49 > * The parsed hooks by type.
50 > */
51 > readonly hooks: Map<HookType, { hooks: IHookCommand[]; originalId: string }>;
52 > /**
53 > * Whether all hooks from this file were disabled via `disableAllHooks: true`.
54 > */
55 > readonly disabledAllHooks: boolean;
56 > }
57 >
58 > /**
59 > * Parses hooks from a Claude settings.json file.
60 > * Claude format:
61 > * {
62 > * "hooks": {
63 > * "PreToolUse": [
64 > * { "matcher": "Bash", "hooks": [{ "type": "command", "command": "..." }] }
65 > * ]
66 > * }
67 > * }
68 > *
69 > * Or simpler format:
70 > * {
71 > * "hooks": {
72 > * "PreToolUse": [{ "type": "command", "command": "..." }]
73 > * }
74 > * }
75 > *
76 > * If the file has `disableAllHooks: true` at the top level, all hooks are filtered out.
77 > */
78 > export function parseClaudeHooks(
79 json: unknown,
80 workspaceRootUri: URI | undefined,
src/vs/base/common/codicons.ts 64 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- codicons.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { ThemeIcon } from './themables.js';
6 > import { register } from './codiconsUtil.js';
7 > import { codiconsLibrary } from './codiconsLibrary.js';
8 >
9 >
10 > /**
11 > * Only to be used by the iconRegistry.
12 > */
13 > export function getAllCodicons(): ThemeIcon[] {
14 return Object.values(Codicon);
15 }
17 > /**
18 > * Derived icons, that could become separate icons.
19 > * These mappings should be moved into the mapping file in the vscode-codicons repo at some point.
20 > */
21 > export const codiconsDerived = {
22 > dialogError: register('dialog-error', 'error'),
23 > dialogWarning: register('dialog-warning', 'warning'),
24 > dialogInfo: register('dialog-info', 'info'),
25 > dialogClose: register('dialog-close', 'close'),
26 > treeItemExpanded: register('tree-item-expanded', 'chevron-down'), // collapsed is done with rotation
27 > treeFilterOnTypeOn: register('tree-filter-on-type-on', 'list-filter'),
28 > treeFilterOnTypeOff: register('tree-filter-on-type-off', 'list-selection'),
29 > treeFilterClear: register('tree-filter-clear', 'close'),
30 > treeItemLoading: register('tree-item-loading', 'loading'),
31 > menuSelection: register('menu-selection', 'check'),
32 > menuSubmenu: register('menu-submenu', 'chevron-right'),
33 > menuBarMore: register('menubar-more', 'more'),
34 > scrollbarButtonLeft: register('scrollbar-button-left', 'triangle-left'),
35 > scrollbarButtonRight: register('scrollbar-button-right', 'triangle-right'),
36 > scrollbarButtonUp: register('scrollbar-button-up', 'triangle-up'),
37 > scrollbarButtonDown: register('scrollbar-button-down', 'triangle-down'),
38 > toolBarMore: register('toolbar-more', 'more'),
39 > quickInputBack: register('quick-input-back', 'arrow-left'),
40 > dropDownButton: register('drop-down-button', 0xeab4),
41 > symbolCustomColor: register('symbol-customcolor', 0xeb5c),
42 > exportIcon: register('export', 0xebac),
43 > workspaceUnspecified: register('workspace-unspecified', 0xebc3),
44 > newLine: register('newline', 0xebea),
45 > thumbsDownFilled: register('thumbsdown-filled', 0xec13),
46 > thumbsUpFilled: register('thumbsup-filled', 0xec14),
47 > gitFetch: register('git-fetch', 0xec1d),
48 > lightbulbSparkleAutofix: register('lightbulb-sparkle-autofix', 0xec1f),
49 > debugBreakpointPending: register('debug-breakpoint-pending', 0xebd9),
50 > chatImport: register('chat-import', 0xec86),
51 > chatExport: register('chat-export', 0xec87),
52 >
53 > } as const;
54 >
55 > /**
56 > * The Codicon library is a set of default icons that are built-in in VS Code.
57 > *
58 > * In the product (outside of base) Codicons should only be used as defaults. In order to have all icons in VS Code
59 > * themeable, component should define new, UI component specific icons using `iconRegistry.registerIcon`.
60 > * In that call a Codicon can be named as default.
61 > */
62 > export const Codicon = {
63 > ...codiconsLibrary,
64 > ...codiconsDerived
65 >
66 > } as const;
src/vs/platform/encryption/common/encryptionService.ts 63 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- encryptionService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { createDecorator } from '../../instantiation/common/instantiation.js';
7 >
8 > export const IEncryptionService = createDecorator<IEncryptionService>('encryptionService');
9 > export interface IEncryptionService extends ICommonEncryptionService {
10 > setUsePlainTextEncryption(): Promise<void>;
11 > getKeyStorageProvider(): Promise<KnownStorageProvider>;
12 > }
13 >
14 > export const IEncryptionMainService = createDecorator<IEncryptionMainService>('encryptionMainService');
15 > export interface IEncryptionMainService extends IEncryptionService { }
16 >
17 > export interface ICommonEncryptionService {
18 >
19 > readonly _serviceBrand: undefined;
20 >
21 > encrypt(value: string): Promise<string>;
22 >
23 > decrypt(value: string): Promise<string>;
24 >
25 > isEncryptionAvailable(): Promise<boolean>;
26 > }
27 >
28 > // The values provided to the `password-store` command line switch.
29 > // Notice that they are not the same as the values returned by
30 > // `getSelectedStorageBackend` in the `safeStorage` API.
31 > export const enum PasswordStoreCLIOption {
32 > kwallet = 'kwallet',
33 > kwallet5 = 'kwallet5',
34 > gnomeLibsecret = 'gnome-libsecret',
35 > basic = 'basic'
36 > }
37 >
38 > // The values returned by `getSelectedStorageBackend` in the `safeStorage` API.
39 > export const enum KnownStorageProvider {
40 > unknown = 'unknown',
41 > basicText = 'basic_text',
42 >
43 > // Linux
44 > gnomeAny = 'gnome_any',
45 > gnomeLibsecret = 'gnome_libsecret',
46 > gnomeKeyring = 'gnome_keyring',
47 > kwallet = 'kwallet',
48 > kwallet5 = 'kwallet5',
49 > kwallet6 = 'kwallet6',
50 >
51 > // The rest of these are not returned by `getSelectedStorageBackend`
52 > // but these were added for platform completeness.
53 >
54 > // Windows
55 > dplib = 'dpapi',
56 >
57 > // macOS
58 > keychainAccess = 'keychain_access',
59 > }
60 >
61 > export function isKwallet(backend: string): boolean {
62 return backend === KnownStorageProvider.kwallet
63 || backend === KnownStorageProvider.kwallet5
64 || backend === KnownStorageProvider.kwallet6;
65 }
67 > export function isGnome(backend: string): boolean {
68 return backend === KnownStorageProvider.gnomeAny
69 || backend === KnownStorageProvider.gnomeLibsecret
src/vs/base/common/observableInternal/map.ts 62 covered LOC · 22 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- map.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, ITransaction } from '../observable.js';
7 > import { observableValueOpts } from './observables/observableValueOpts.js';
8 >
9 >
10 > export class ObservableMap<K, V> implements Map<K, V> {
11 > private readonly _data = new Map<K, V>(); map.ts
12 >
13 > private readonly _obs = observableValueOpts({ equalsFn: () => false }, this);
14 >
15 > readonly observable: IObservable<Map<K, V>> = this._obs;
16 > map.ts
17 > get size(): number {
18 return this._data.size;
19 }
20 > map.ts
21 > has(key: K): boolean {
22 > return this._data.has(key); map.ts
23 > }
24 > map.ts
25 > get(key: K): V | undefined {
26 > return this._data.get(key); map.ts
27 > }
28 > map.ts
29 > set(key: K, value: V, tx?: ITransaction): this {
30 > const hadKey = this._data.has(key); map.ts
31 > const oldValue = this._data.get(key);
32 > if (!hadKey || oldValue !== value) {
33 > this._data.set(key, value);
34 > this._obs.set(this, tx);
35 > }
36 > return this;
37 > }
38 > map.ts
39 > delete(key: K, tx?: ITransaction): boolean {
40 > const result = this._data.delete(key); map.ts
41 > if (result) {
42 > this._obs.set(this, tx);
43 > }
44 > return result;
45 > }
46 > map.ts
47 > clear(tx?: ITransaction): void {
48 if (this._data.size > 0) {
49 this._data.clear();
51 }
52 }
53 > map.ts
54 > forEach(callbackfn: (value: V, key: K, map: Map<K, V>) => void, thisArg?: unknown): void {
55 > this._data.forEach((value, key, _map) => { map.ts
56 > callbackfn.call(thisArg, value, key, this); map.ts
57 > }); map.ts
58 > }
59 > map.ts
60 > *entries(): MapIterator<[K, V]> {
61 yield* this._data.entries();
62 }
63 > map.ts
64 > *keys(): MapIterator<K> {
65 yield* this._data.keys();
66 }
67 > map.ts
68 > *values(): MapIterator<V> {
69 yield* this._data.values();
70 }
71 > map.ts
72 > [Symbol.iterator](): MapIterator<[K, V]> {
73 return this.entries();
74 }
75 > map.ts
76 > get [Symbol.toStringTag](): string {
77 return 'ObservableMap';
78 }
79 > } map.ts
src/vs/editor/common/tokenizationRegistry.ts 62 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- tokenizationRegistry.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Color } from '../../base/common/color.js';
7 > import { Emitter, Event } from '../../base/common/event.js';
8 > import { Disposable, IDisposable, toDisposable } from '../../base/common/lifecycle.js';
9 > import { ITokenizationRegistry, ITokenizationSupportChangedEvent, ILazyTokenizationSupport } from './languages.js';
10 > import { ColorId } from './encodedTokenAttributes.js';
11 >
12 > export class TokenizationRegistry<TSupport> implements ITokenizationRegistry<TSupport> {
13 >
14 > private readonly _tokenizationSupports = new Map<string, TSupport>();
15 > private readonly _factories = new Map<string, TokenizationSupportFactoryData<TSupport>>();
16 >
17 > private readonly _onDidChange = new Emitter<ITokenizationSupportChangedEvent>();
18 > public readonly onDidChange: Event<ITokenizationSupportChangedEvent> = this._onDidChange.event;
19 >
20 > private _colorMap: Color[] | null;
21 >
22 > constructor() {
23 > this._colorMap = null;
24 > }
25 >
26 > public handleChange(languageIds: string[]): void {
27 this._onDidChange.fire({
28 changedLanguages: languageIds,
30 });
31 }
33 > public register(languageId: string, support: TSupport): IDisposable {
34 this._tokenizationSupports.set(languageId, support);
35 this.handleChange([languageId]);
42 });
43 }
45 > public get(languageId: string): TSupport | null {
46 return this._tokenizationSupports.get(languageId) || null;
47 }
49 > public registerFactory(languageId: string, factory: ILazyTokenizationSupport<TSupport>): IDisposable {
50 this._factories.get(languageId)?.dispose();
51 const myData = new TokenizationSupportFactoryData(this, languageId, factory);
60 });
61 }
63 > public async getOrCreate(languageId: string): Promise<TSupport | null> {
64 // check first if the support is already set
65 const tokenizationSupport = this.get(languageId);
78 return this.get(languageId);
79 }
81 > public isResolved(languageId: string): boolean {
82 const tokenizationSupport = this.get(languageId);
83 if (tokenizationSupport) {
92 return false;
93 }
95 > public setColorMap(colorMap: Color[]): void {
96 this._colorMap = colorMap;
97 this._onDidChange.fire({
100 });
101 }
103 > public getColorMap(): Color[] | null {
104 return this._colorMap;
105 }
107 > public getDefaultBackground(): Color | null {
108 if (this._colorMap && this._colorMap.length > ColorId.DefaultBackground) {
109 return this._colorMap[ColorId.DefaultBackground];
111 return null;
112 }
114 >
115 > class TokenizationSupportFactoryData<TSupport> extends Disposable {
116 >
117 > private _isDisposed: boolean = false;
118 > private _resolvePromise: Promise<void> | null = null;
119 > private _isResolved: boolean = false;
120 >
121 > public get isResolved(): boolean {
122 > return this._isResolved;
123 > }
124 >
125 > constructor(
126 private readonly _registry: TokenizationRegistry<TSupport>,
127 private readonly _languageId: string,
130 super();
131 }
133 > public override dispose(): void {
134 this._isDisposed = true;
135 super.dispose();
136 }
138 > public async resolve(): Promise<void> {
139 if (!this._resolvePromise) {
140 this._resolvePromise = this._create();
142 return this._resolvePromise;
143 }
145 > private async _create(): Promise<void> {
146 const value = await this._factory.tokenizationSupport;
147 this._isResolved = true;
src/vs/amdX.ts 59 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- amdX.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { AppResourcePath, FileAccess, nodeModulesAsarPath, nodeModulesPath, Schemas, VSCODE_AUTHORITY } from './base/common/network.js';
7 > import * as platform from './base/common/platform.js';
8 > import { IProductConfiguration } from './base/common/product.js';
9 > import { URI } from './base/common/uri.js';
10 > import { generateUuid } from './base/common/uuid.js';
11 >
12 > declare const window: any;
13 > declare const document: any;
14 > declare const self: any;
15 > declare const globalThis: any;
16 >
17 > class DefineCall {
18 > constructor(
19 public readonly id: string | null | undefined,
20 public readonly dependencies: string[] | null | undefined,
21 public readonly callback: any
22 ) { }
23 > } amdX.ts
24 >
25 > enum AMDModuleImporterState {
26 > Uninitialized = 1,
27 > InitializedInternal,
28 > InitializedExternal
29 > }
30 >
31 > class AMDModuleImporter {
32 > public static INSTANCE = new AMDModuleImporter();
33 >
34 > private readonly _isWebWorker = (typeof self === 'object' && self.constructor && self.constructor.name === 'DedicatedWorkerGlobalScope');
35 > private readonly _isRenderer = typeof document === 'object';
36 >
37 > private readonly _defineCalls: DefineCall[] = [];
38 > private _state = AMDModuleImporterState.Uninitialized;
39 > private _amdPolicy: Pick<TrustedTypePolicy, 'name' | 'createScriptURL'> | undefined;
40 >
41 > constructor() { }
42 >
43 > private _initialize(): void {
44 if (this._state === AMDModuleImporterState.Uninitialized) {
45 if (globalThis.define) {
91 }
92 }
93 > amdX.ts
94 > public async load<T>(scriptSrc: string): Promise<T> {
95 this._initialize();
96
134 }
135 }
136 > amdX.ts
137 > private _rendererLoadScript(scriptSrc: string): Promise<DefineCall | undefined> {
138 return new Promise<DefineCall | undefined>((resolve, reject) => {
139 const scriptElement = document.createElement('script');
165 });
166 }
167 > amdX.ts
168 > private async _workerLoadScript(scriptSrc: string): Promise<DefineCall | undefined> {
169 if (this._amdPolicy) {
170 scriptSrc = this._amdPolicy.createScriptURL(scriptSrc) as unknown as string;
173 return this._defineCalls.pop();
174 }
175 > amdX.ts
176 > private async _nodeJSLoadScript(scriptSrc: string): Promise<DefineCall | undefined> {
177 try {
178 // `import('module')` is not remapped (only `fs` is), so it yields the real
198 }
199 }
200 > } amdX.ts
201 >
202 > const cache = new Map<string, Promise<any>>();
203 >
204 > /**
205 > * Utility for importing an AMD node module. This util supports AMD and ESM contexts and should be used while the ESM adoption
206 > * is on its way.
207 > *
208 > * e.g. pass in `vscode-textmate/release/main.js`
209 > */
210 export async function importAMDNodeModule<T>(nodeModuleName: string, pathInsideNodeModule: string, isBuilt?: boolean): Promise<T> {
211 if (isBuilt === undefined) {
233 return result;
234 }
235 > amdX.ts
236 > export function resolveAmdNodeModulePath(nodeModuleName: string, pathInsideNodeModule: string): string {
237 const product = globalThis._VSCODE_PRODUCT_JSON as unknown as IProductConfiguration;
238 const isBuilt = Boolean((product ?? globalThis.vscode?.context?.configuration()?.product)?.commit);
src/vs/base/common/observableInternal/observables/observableSignalFromEvent.ts 59 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableSignalFromEvent.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable } from '../base.js';
7 > import { transaction } from '../transaction.js';
8 > import { Event, IDisposable } from '../commonFacade/deps.js';
9 > import { DebugOwner, DebugNameData } from '../debugName.js';
10 > import { BaseObservable } from './baseObservable.js';
11 > import { DebugLocation } from '../debugLocation.js';
12 >
13 > export function observableSignalFromEvent(
14 > owner: DebugOwner | string, observableSignalFromEvent.ts
15 > event: Event<any>,
16 > debugLocation = DebugLocation.ofCaller()
17 > ): IObservable<void> {
18 > return new FromEventObservableSignal(typeof owner === 'string' ? owner : new DebugNameData(owner, undefined, undefined), event, debugLocation);
19 > }
21 > class FromEventObservableSignal extends BaseObservable<void> {
22 > private subscription: IDisposable | undefined;
23 >
24 > public readonly debugName: string;
25 > constructor(
26 > debugNameDataOrName: DebugNameData | string, observableSignalFromEvent.ts
27 > private readonly event: Event<any>,
28 > debugLocation: DebugLocation
29 > ) {
30 > super(debugLocation);
31 > this.debugName = typeof debugNameDataOrName === 'string'
32 ? debugNameDataOrName
33 > : debugNameDataOrName.getDebugName(this) ?? 'Observable Signal From Event'; observableSignalFromEvent.ts
34 > }
36 > protected override onFirstObserverAdded(): void {
37 > this.subscription = this.event(this.handleEvent); observableSignalFromEvent.ts
38 > }
40 > private readonly handleEvent = () => {
41 > transaction( observableSignalFromEvent.ts
42 > (tx) => {
43 > for (const o of this._observers) {
44 > tx.updateObserver(o, this);
45 > o.handleChange(this, undefined);
46 > }
47 > },
48 > () => this.debugName
49 > );
50 > };
52 > protected override onLastObserverRemoved(): void {
53 > this.subscription!.dispose(); observableSignalFromEvent.ts
54 > this.subscription = undefined;
55 > }
57 > public override get(): void {
59 > }
src/vs/workbench/services/assignment/common/assignmentContextFilter.ts 59 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- assignmentContextFilter.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from '../../../../base/common/event.js';
7 > import { Disposable, DisposableStore } from '../../../../base/common/lifecycle.js';
8 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
9 > import type { IAssignmentFilter } from './assignmentService.js';
10 >
11 > /**
12 > * Applies the registered {@link IAssignmentFilter assignment filters} to the assignment context
13 > * and remembers, per filter, which assignment-context ids each filter excluded.
14 > *
15 > * The excluded ids are persisted so that they stay filtered out across reloads even before the
16 > * owning filter has been registered again. Once a filter is registered it becomes authoritative
17 > * for its own id: the persisted state is repopulated from the ids it currently excludes and any
18 > * id that is no longer excluded is dropped from storage.
19 > */
20 > export class AssignmentContextFilter extends Disposable {
21 >
22 > private static readonly STORAGE_KEY = 'workbench.assignment.filteredOutAssignmentContextIds';
23 >
24 > private readonly _filters: IAssignmentFilter[] = [];
25 > private readonly _filterDisposables = this._register(new DisposableStore());
26 >
27 > /**
28 > * Assignment-context ids that were filtered out, keyed by the id of the filter that
29 > * excluded them. Entries owned by a filter that has not been registered yet are honored
30 > * across reloads so those ids stay excluded until the filter takes over.
31 > */
32 > private _cachedFilteredOutIds: Map<string, Set<string>>;
33 >
34 > /**
35 > * Ids of filters that have been registered this session. Once a filter is registered it
36 > * becomes authoritative for its own id and the persisted cache is repopulated from the
37 > * ids it actually excludes.
38 > */
39 > private readonly _registeredFilterIds = new Set<string>();
40 >
41 > private readonly _onDidChange = this._register(new Emitter<void>());
42 > /** Fires when a filter is added or an already registered filter changes its exclusions. */
43 > readonly onDidChange: Event<void> = this._onDidChange.event;
44 >
45 > constructor(
46 private readonly storageService: IStorageService
47 ) {
50 this._cachedFilteredOutIds = this._loadFilteredOutIds();
51 }
53 > addFilter(filter: IAssignmentFilter): void {
54 this._filters.push(filter);
55
66 this._onDidChange.fire();
67 }
69 > /**
70 > * Removes the excluded assignment-context ids from the given context and persists the
71 > * reconciled per-filter cache.
72 > */
73 > filter(assignmentContext: string): string {
74 const assignments = assignmentContext.split(';');
75
121 return filteredAssignments.join(';');
122 }
124 > private _loadFilteredOutIds(): Map<string, Set<string>> {
125 const result = new Map<string, Set<string>>();
126 const raw = this.storageService.get(AssignmentContextFilter.STORAGE_KEY, StorageScope.APPLICATION);
147 return result;
148 }
150 > private _storeFilteredOutIds(next: Map<string, Set<string>>): void {
151 // Drop empty entries so an id disappears from storage once nothing is filtered out for it.
152 const normalized = new Map<string, Set<string>>();
173 }
174 }
176 >
177 function areCachesEqual(a: Map<string, Set<string>>, b: Map<string, Set<string>>): boolean {
178 if (a.size !== b.size) {
src/vs/workbench/services/userDataProfile/common/remoteUserDataProfiles.ts 59 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- remoteUserDataProfiles.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Disposable } from '../../../../base/common/lifecycle.js';
7 > import { InstantiationType, registerSingleton } from '../../../../platform/instantiation/common/extensions.js';
8 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
9 > import { DidChangeProfilesEvent, IUserDataProfile, IUserDataProfilesService } from '../../../../platform/userDataProfile/common/userDataProfile.js';
10 > import { IRemoteAgentService } from '../../remote/common/remoteAgentService.js';
11 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
12 > import { IStringDictionary } from '../../../../base/common/collections.js';
13 > import { ILogService } from '../../../../platform/log/common/log.js';
14 > import { IUserDataProfileService } from './userDataProfile.js';
15 > import { distinct } from '../../../../base/common/arrays.js';
16 > import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.js';
17 > import { UserDataProfilesService } from '../../../../platform/userDataProfile/common/userDataProfileIpc.js';
18 > import { ErrorNoTelemetry } from '../../../../base/common/errors.js';
19 >
20 > const associatedRemoteProfilesKey = 'associatedRemoteProfiles';
21 >
22 > export const IRemoteUserDataProfilesService = createDecorator<IRemoteUserDataProfilesService>('IRemoteUserDataProfilesService');
23 > export interface IRemoteUserDataProfilesService {
24 > readonly _serviceBrand: undefined;
25 > getRemoteProfiles(): Promise<readonly IUserDataProfile[]>;
26 > getRemoteProfile(localProfile: IUserDataProfile): Promise<IUserDataProfile>;
27 > }
28 >
29 > class RemoteUserDataProfilesService extends Disposable implements IRemoteUserDataProfilesService {
30 >
31 > readonly _serviceBrand: undefined;
32 >
33 > private readonly initPromise: Promise<void>;
34 >
35 > private remoteUserDataProfilesService: IUserDataProfilesService | undefined;
36 >
37 > constructor(
38 @IWorkbenchEnvironmentService private readonly environmentService: IWorkbenchEnvironmentService,
39 @IRemoteAgentService private readonly remoteAgentService: IRemoteAgentService,
46 this.initPromise = this.init();
47 }
49 > private async init(): Promise<void> {
50 const connection = this.remoteAgentService.getConnection();
51 if (!connection) {
69 this.cleanUp();
70 }
72 > private async onDidChangeLocalProfiles(e: DidChangeProfilesEvent): Promise<void> {
73 for (const profile of e.removed) {
74 const remoteProfile = this.remoteUserDataProfilesService?.profiles.find(p => p.id === profile.id);
78 }
79 }
81 > async getRemoteProfiles(): Promise<readonly IUserDataProfile[]> {
82 await this.initPromise;
83
88 return this.remoteUserDataProfilesService.profiles;
89 }
91 > async getRemoteProfile(localProfile: IUserDataProfile): Promise<IUserDataProfile> {
92 await this.initPromise;
93
98 return this.getAssociatedRemoteProfile(localProfile, this.remoteUserDataProfilesService);
99 }
101 > private async getAssociatedRemoteProfile(localProfile: IUserDataProfile, remoteUserDataProfilesService: IUserDataProfilesService): Promise<IUserDataProfile> {
102 // If the local profile is the default profile, return the remote default profile
103 if (localProfile.isDefault) {
115 return profile;
116 }
118 > private getAssociatedRemoteProfiles(): string[] {
119 if (this.environmentService.remoteAuthority) {
120 const remotes = this.parseAssociatedRemoteProfiles();
123 return [];
124 }
126 > private setAssociatedRemoteProfiles(profiles: string[]): void {
127 if (this.environmentService.remoteAuthority) {
128 const remotes = this.parseAssociatedRemoteProfiles();
140 }
141 }
143 > private parseAssociatedRemoteProfiles(): IStringDictionary<string[]> {
144 if (this.environmentService.remoteAuthority) {
145 const value = this.storageService.get(associatedRemoteProfilesKey, StorageScope.APPLICATION);
152 return {};
153 }
155 > private async cleanUp(): Promise<void> {
156 const associatedRemoteProfiles: string[] = [];
157 for (const profileId of this.getAssociatedRemoteProfiles()) {
175 this.setAssociatedRemoteProfiles(associatedRemoteProfiles);
176 }
178 > }
179 >
180 > registerSingleton(IRemoteUserDataProfilesService, RemoteUserDataProfilesService, InstantiationType.Delayed);
src/vs/base/common/observableInternal/index.ts 57 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- index.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // This is a facade for the observable implementation. Only import from here!
7 >
8 > export { observableValueOpts } from './observables/observableValueOpts.js';
9 > export { autorun, autorunDelta, autorunHandleChanges, autorunOpts, autorunWithStore, autorunWithStoreHandleChanges, autorunIterableDelta, autorunPerKeyedItem, autorunSelfDisposable, registerAutorunSelfDisposable } from './reactions/autorun.js';
10 > export { type IObservable, type IObservableWithChange, type IObserver, type IReader, type ISettable, type IReaderWithStore, type ISettableObservable, type ITransaction } from './base.js';
11 > export { disposableObservableValue } from './observables/observableValue.js';
12 > export { derived, derivedDisposable, derivedHandleChanges, derivedOpts, derivedWithSetter, derivedWithStore } from './observables/derived.js';
13 > export { type IDerivedReader } from './observables/derivedImpl.js';
14 > export { ObservableLazy, ObservableLazyPromise, ObservablePromise, ObservableResolvedPromise, PromiseResult, } from './utils/promise.js';
15 > export { derivedWithCancellationToken, waitForState } from './utils/utilsCancellation.js';
16 > export {
17 > debouncedObservable, debouncedObservable2, derivedObservableWithCache,
18 > derivedObservableWithWritableCache, keepObserved, mapObservableArrayCached, observableFromPromise,
19 > recomputeInitiallyAndOnChange,
20 > signalFromObservable, throttledObservable, wasEventTriggeredRecently,
21 > isObservable,
22 > } from './utils/utils.js';
23 > export { type DebugOwner } from './debugName.js';
24 > export { type IChangeContext, type IChangeTracker, recordChanges, recordChangesLazy } from './changeTracker.js';
25 > export { constObservable } from './observables/constObservable.js';
26 > export { type IObservableSignal, observableSignal } from './observables/observableSignal.js';
27 > export { observableFromEventOpts } from './observables/observableFromEvent.js';
28 > export { observableSignalFromEvent } from './observables/observableSignalFromEvent.js';
29 > export { asyncTransaction, globalTransaction, subtransaction, transaction, TransactionImpl } from './transaction.js';
30 > export { observableFromValueWithChangeEvent, ValueWithChangeEventFromObservable } from './utils/valueWithChangeEvent.js';
31 > export { runOnChange, runOnChangeWithCancellationToken, runOnChangeWithStore, type RemoveUndefined } from './utils/runOnChange.js';
32 > export { derivedConstOnceDefined, latestChangedValue } from './experimental/utils.js';
33 > export { observableFromEvent } from './observables/observableFromEvent.js';
34 > export { observableValue } from './observables/observableValue.js';
35 >
36 > export { ObservableSet } from './set.js';
37 > export { ObservableMap } from './map.js';
38 > export { DebugLocation } from './debugLocation.js';
39 >
40 > import { addLogger, setLogObservableFn } from './logging/logging.js';
41 > import { ConsoleObservableLogger, logObservableToConsole } from './logging/consoleObservableLogger.js';
42 > import { DevToolsLogger } from './logging/debugger/devToolsLogger.js';
43 > import { env } from '../process.js';
44 > import { _setDebugGetObservableGraph } from './observables/baseObservable.js';
45 > import { debugGetObservableGraph } from './logging/debugGetDependencyGraph.js';
46 >
47 > _setDebugGetObservableGraph(debugGetObservableGraph);
48 > setLogObservableFn(logObservableToConsole);
49 >
50 > // Remove "//" in the next line to enable logging
51 > const enableLogging = false
52 > // || Boolean("true") // done "weirdly" so that a lint warning prevents you from pushing this
53 > ;
54 >
55 > if (enableLogging) {
56 addLogger(new ConsoleObservableLogger());
57 }
58 > index.ts
59 > if (env && env['VSCODE_DEV_DEBUG_OBSERVABLES']) {
60 // To debug observables you also need the extension "ms-vscode.debug-value-editor"
61 addLogger(DevToolsLogger.getInstance());
src/vs/base/test/common/utils.ts 56 covered LOC · 13 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- utils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DisposableStore, DisposableTracker, IDisposable, setDisposableTracker } from '../../common/lifecycle.js';
7 > import { join } from '../../common/path.js';
8 > import { isWindows } from '../../common/platform.js';
9 > import { URI } from '../../common/uri.js';
10 >
11 > export type ValueCallback<T = any> = (value: T | Promise<T>) => void;
12 >
13 > export function toResource(this: any, path: string): URI {
14 if (isWindows) {
15 return URI.file(join('C:\\', btoa(this.test.fullTitle()), path));
18 return URI.file(join('/', btoa(this.test.fullTitle()), path));
19 }
20 > utils.ts
21 > export function suiteRepeat(n: number, description: string, callback: (this: any) => void): void {
22 for (let i = 0; i < n; i++) {
23 suite(`${description} (iteration ${i})`, callback);
24 }
25 }
26 > utils.ts
27 > export function testRepeat(n: number, description: string, callback: (this: any) => any): void {
28 for (let i = 0; i < n; i++) {
29 test(`${description} (iteration ${i})`, callback);
30 }
31 }
32 > utils.ts
33 export async function assertThrowsAsync(block: () => any, message: string | Error = 'Missing expected exception'): Promise<void> {
34 try {
41 throw err;
42 }
43 > utils.ts
44 > /**
45 > * Use this function to ensure that all disposables are cleaned up at the end of each test in the current suite.
46 > *
47 > * Use `markAsSingleton` if disposable singletons are created lazily that are allowed to outlive the test.
48 > * Make sure that the singleton properly registers all child disposables so that they are excluded too.
49 > *
50 > * @returns A {@link DisposableStore} that can optionally be used to track disposables in the test.
51 > * This will be automatically disposed on test teardown.
52 > */
53 > export function ensureNoDisposablesAreLeakedInTestSuite(): Pick<DisposableStore, 'add'> {
54 > let tracker: DisposableTracker | undefined;
55 > let store: DisposableStore;
56 > setup(() => {
57 > store = new DisposableStore(); utils.ts
58 > tracker = new DisposableTracker();
59 > setDisposableTracker(tracker);
60 > }); utils.ts
61 >
62 > teardown(function (this: import('mocha').Context) {
63 > store.dispose(); utils.ts
64 > setDisposableTracker(null);
65 > if (this.currentTest?.state !== 'failed') {
66 > const result = tracker!.computeLeakingDisposables();
67 > if (result) {
68 console.error(result.details);
69 throw new Error(`There are ${result.leaks.length} undisposed disposables!${result.details}`);
70 }
71 > } utils.ts
72 > }); utils.ts
73 >
74 > // Wrap store as the suite function is called before it's initialized
75 > const testContext = {
76 > add<T extends IDisposable>(o: T): T {
77 return store.add(o);
78 }
79 > }; utils.ts
80 > return testContext;
81 > }
82 >
83 > export function throwIfDisposablesAreLeaked(body: () => void, logToConsole = true): void {
84 const tracker = new DisposableTracker();
85 setDisposableTracker(tracker);
88 computeLeakingDisposables(tracker, logToConsole);
89 }
90 > utils.ts
91 export async function throwIfDisposablesAreLeakedAsync(body: () => Promise<void>): Promise<void> {
92 const tracker = new DisposableTracker();
96 computeLeakingDisposables(tracker);
97 }
98 > utils.ts
99 function computeLeakingDisposables(tracker: DisposableTracker, logToConsole = true) {
100 const result = tracker.computeLeakingDisposables();
src/vs/platform/policy/common/policy.ts 56 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- policy.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStringDictionary } from '../../../base/common/collections.js';
7 > import { IPolicyData } from '../../../base/common/defaultAccount.js';
8 > import { Emitter, Event } from '../../../base/common/event.js';
9 > import { Iterable } from '../../../base/common/iterator.js';
10 > import { Disposable } from '../../../base/common/lifecycle.js';
11 > import { IManagedSettingsPolicyDefinitions, PolicyName } from '../../../base/common/policy.js';
12 > import { createDecorator } from '../../instantiation/common/instantiation.js';
13 >
14 > export type PolicyValue = string | number | boolean;
15 > export type PolicyDefinition = {
16 > type: 'string' | 'number' | 'boolean';
17 > value?: (policyData: IPolicyData) => string | number | boolean | undefined;
18 > managedSettings?: IManagedSettingsPolicyDefinitions;
19 > restrictedValue?: PolicyValue;
20 > };
21 >
22 > /** Returns a structured-clone-safe copy of `definition`, dropping the non-cloneable `value` callback. */
23 > export function toSerializablePolicyDefinition(definition: PolicyDefinition): PolicyDefinition {
24 return { type: definition.type, managedSettings: definition.managedSettings, restrictedValue: definition.restrictedValue };
25 }
26 > policy.ts
27 > /**
28 > * Returns the value to apply for `definition` when the account-policy gate is active
29 > * but not satisfied. Uses `definition.restrictedValue` when specified, otherwise falls
30 > * back to a type-driven safe default.
31 > */
32 > export function getRestrictedPolicyValue(definition: PolicyDefinition): PolicyValue {
33 if (definition.restrictedValue !== undefined) {
34 return definition.restrictedValue;
40 }
41 }
42 > policy.ts
43 > export const IPolicyService = createDecorator<IPolicyService>('policy');
44 >
45 > export interface IPolicyService {
46 > readonly _serviceBrand: undefined;
47 >
48 > readonly onDidChange: Event<readonly PolicyName[]>;
49 > updatePolicyDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<IStringDictionary<PolicyValue>>;
50 > getPolicyValue(name: PolicyName): PolicyValue | undefined;
51 > serialize(): IStringDictionary<{ definition: PolicyDefinition; value: PolicyValue }> | undefined;
52 > readonly policyDefinitions: IStringDictionary<PolicyDefinition>;
53 > }
54 >
55 > export abstract class AbstractPolicyService extends Disposable implements IPolicyService {
56 readonly _serviceBrand: undefined;
57
61 protected readonly _onDidChange = this._register(new Emitter<readonly PolicyName[]>());
62 readonly onDidChange = this._onDidChange.event;
63 > policy.ts
64 > async updatePolicyDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<IStringDictionary<PolicyValue>> {
65 // Replace existing definitions; identity comparison avoids redundant watcher churn.
66 let changed = false;
78 return Iterable.reduce(this.policies.entries(), (r, [name, value]) => ({ ...r, [name]: value }), {});
79 }
80 > policy.ts
81 > getPolicyValue(name: PolicyName): PolicyValue | undefined {
82 return this.policies.get(name);
83 }
84 > policy.ts
85 > serialize(): IStringDictionary<{ definition: PolicyDefinition; value: PolicyValue }> {
86 return Iterable.reduce<[PolicyName, PolicyDefinition], IStringDictionary<{ definition: PolicyDefinition; value: PolicyValue }>>(Object.entries(this.policyDefinitions), (r, [name, definition]) => ({ ...r, [name]: { definition: toSerializablePolicyDefinition(definition), value: this.policies.get(name)! } }), {});
87 }
88 > policy.ts
89 > protected abstract _updatePolicyDefinitions(policyDefinitions: IStringDictionary<PolicyDefinition>): Promise<void>;
90 > }
91 >
92 > export class NullPolicyService implements IPolicyService {
93 readonly _serviceBrand: undefined;
94 readonly onDidChange = Event.None;
97 serialize() { return undefined; }
98 policyDefinitions: IStringDictionary<PolicyDefinition> = {};
99 > } policy.ts
src/vs/workbench/services/environment/common/environmentService.ts 56 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- environmentService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { refineServiceDecorator } from '../../../../platform/instantiation/common/instantiation.js';
7 > import { IPath } from '../../../../platform/window/common/window.js';
8 > import { IEnvironmentService } from '../../../../platform/environment/common/environment.js';
9 > import { URI } from '../../../../base/common/uri.js';
10 >
11 > export const IWorkbenchEnvironmentService = refineServiceDecorator<IEnvironmentService, IWorkbenchEnvironmentService>(IEnvironmentService);
12 >
13 > /**
14 > * A workbench specific environment service that is only present in workbench
15 > * layer.
16 > */
17 > export interface IWorkbenchEnvironmentService extends IEnvironmentService {
18 >
19 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
20 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE. AS SUCH:
21 > // PUT NON-WEB PROPERTIES INTO THE NATIVE WORKBENCH
22 > // ENVIRONMENT SERVICE
23 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
24 >
25 > // --- Paths
26 > readonly logFile: URI;
27 > readonly windowLogsPath: URI;
28 > readonly extHostLogsPath: URI;
29 >
30 > // --- Extensions
31 > readonly extensionEnabledProposedApi?: string[];
32 >
33 > // --- Config
34 > readonly remoteAuthority?: string;
35 > readonly skipReleaseNotes: boolean;
36 > readonly skipWelcome: boolean;
37 > readonly disableWorkspaceTrust: boolean;
38 > readonly isSessionsWindow: boolean;
39 > readonly webviewExternalEndpoint: string;
40 >
41 > // --- Development
42 > readonly debugRenderer: boolean;
43 > readonly logExtensionHostCommunication?: boolean;
44 > readonly enableSmokeTestDriver?: boolean;
45 > readonly profDurationMarkers?: string[];
46 >
47 > // --- Editors to open
48 > readonly filesToOpenOrCreate?: IPath[] | undefined;
49 > readonly filesToDiff?: IPath[] | undefined;
50 > readonly filesToMerge?: IPath[] | undefined;
51 >
52 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
53 > // NOTE: KEEP THIS INTERFACE AS SMALL AS POSSIBLE. AS SUCH:
54 > // - PUT NON-WEB PROPERTIES INTO NATIVE WB ENV SERVICE
55 > // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
56 > }
src/vs/platform/userDataProfile/common/userDataProfileIpc.ts 55 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- userDataProfileIpc.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter, Event } from '../../../base/common/event.js';
7 > import { IChannel, IServerChannel } from '../../../base/parts/ipc/common/ipc.js';
8 > import { URI, UriDto } from '../../../base/common/uri.js';
9 > import { DidChangeProfilesEvent, IUserDataProfile, IUserDataProfileOptions, IUserDataProfilesService, IUserDataProfileUpdateOptions, reviveProfile } from './userDataProfile.js';
10 > import { IAnyWorkspaceIdentifier } from '../../workspace/common/workspace.js';
11 > import { IURITransformer, transformIncomingURIs, transformOutgoingURIs } from '../../../base/common/uriIpc.js';
12 > import { Disposable } from '../../../base/common/lifecycle.js';
13 >
14 > export class RemoteUserDataProfilesServiceChannel implements IServerChannel {
15 >
16 > constructor(
17 private readonly service: IUserDataProfilesService,
18 private readonly getUriTransformer: (requestContext: any) => IURITransformer
19 ) { }
21 > listen(context: any, event: string): Event<any> {
22 const uriTransformer = this.getUriTransformer(context);
23 switch (event) {
33 throw new Error(`Invalid listen ${event}`);
34 }
36 > async call(context: any, command: string, args?: any): Promise<any> {
37 const uriTransformer = this.getUriTransformer(context);
38 switch (command) {
53 throw new Error(`Invalid call ${command}`);
54 }
56 >
57 > export class UserDataProfilesService extends Disposable implements IUserDataProfilesService {
58 >
59 > readonly _serviceBrand: undefined;
60 >
61 > get defaultProfile(): IUserDataProfile { return this.profiles[0]; }
62 > private _profiles: IUserDataProfile[] = [];
63 > get profiles(): IUserDataProfile[] { return this._profiles; }
64 >
65 > private readonly _onDidChangeProfiles = this._register(new Emitter<DidChangeProfilesEvent>());
66 > readonly onDidChangeProfiles = this._onDidChangeProfiles.event;
67 >
68 > readonly onDidResetWorkspaces: Event<void>;
69 >
70 > constructor(
71 profiles: readonly UriDto<IUserDataProfile>[],
72 readonly profilesHome: URI,
84 this.onDidResetWorkspaces = this.channel.listen<void>('onDidResetWorkspaces');
85 }
87 > async createNamedProfile(name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
88 const result = await this.channel.call<UriDto<IUserDataProfile>>('createNamedProfile', [name, options, workspaceIdentifier]);
89 return reviveProfile(result, this.profilesHome.scheme);
90 }
92 > async createProfile(id: string, name: string, options?: IUserDataProfileOptions, workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
93 const result = await this.channel.call<UriDto<IUserDataProfile>>('createProfile', [id, name, options, workspaceIdentifier]);
94 return reviveProfile(result, this.profilesHome.scheme);
95 }
97 > async createTransientProfile(workspaceIdentifier?: IAnyWorkspaceIdentifier): Promise<IUserDataProfile> {
98 const result = await this.channel.call<UriDto<IUserDataProfile>>('createTransientProfile', [workspaceIdentifier]);
99 return reviveProfile(result, this.profilesHome.scheme);
100 }
102 > async setProfileForWorkspace(workspaceIdentifier: IAnyWorkspaceIdentifier, profile: IUserDataProfile): Promise<void> {
103 await this.channel.call<UriDto<IUserDataProfile>>('setProfileForWorkspace', [workspaceIdentifier, profile]);
104 }
106 > removeProfile(profile: IUserDataProfile): Promise<void> {
107 return this.channel.call('removeProfile', [profile]);
108 }
110 > async updateProfile(profile: IUserDataProfile, updateOptions: IUserDataProfileUpdateOptions): Promise<IUserDataProfile> {
111 const result = await this.channel.call<UriDto<IUserDataProfile>>('updateProfile', [profile, updateOptions]);
112 return reviveProfile(result, this.profilesHome.scheme);
113 }
115 > resetWorkspaces(): Promise<void> {
116 return this.channel.call('resetWorkspaces');
117 }
119 > cleanUp(): Promise<void> {
120 return this.channel.call('cleanUp');
121 }
123 > cleanUpTransientProfiles(): Promise<void> {
124 return this.channel.call('cleanUpTransientProfiles');
125 }
src/vs/workbench/contrib/chat/common/tools/promptTsxTypes.ts 55 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- promptTsxTypes.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * This is a subset of the types export from jsonTypes.d.ts in @vscode/prompt-tsx.
8 > * It's just the types needed to stringify prompt-tsx tool results.
9 > * It should be kept in sync with the types in that file.
10 > *
11 > * Note: do NOT use `declare` with const enums, esbuild doesn't inline them.
12 > * See https://github.com/evanw/esbuild/issues/4394
13 > */
14 >
15 > export const enum PromptNodeType {
16 > Piece = 1,
17 > Text = 2
18 > }
19 > export interface TextJSON {
20 > type: PromptNodeType.Text;
21 > text: string;
22 > lineBreakBefore: boolean | undefined;
23 > }
24 > /**
25 > * Constructor kind of the node represented by {@link PieceJSON}. This is
26 > * less descriptive than the actual constructor, as we only care to preserve
27 > * the element data that the renderer cares about.
28 > */
29 > export const enum PieceCtorKind {
30 > BaseChatMessage = 1,
31 > Other = 2,
32 > ImageChatMessage = 3
33 > }
34 > export interface BasePieceJSON {
35 > type: PromptNodeType.Piece;
36 > ctor: PieceCtorKind.BaseChatMessage | PieceCtorKind.Other;
37 > children: PromptNodeJSON[];
38 > }
39 > export interface ImageChatMessagePieceJSON {
40 > type: PromptNodeType.Piece;
41 > ctor: PieceCtorKind.ImageChatMessage;
42 > children: PromptNodeJSON[];
43 > props: {
44 > src: string;
45 > detail?: 'low' | 'high';
46 > };
47 > }
48 > export type PieceJSON = BasePieceJSON | ImageChatMessagePieceJSON;
49 > export type PromptNodeJSON = PieceJSON | TextJSON;
50 > export interface PromptElementJSON {
51 > node: PieceJSON;
52 > }
53 >
54 > export function stringifyPromptElementJSON(element: PromptElementJSON): string {
55 const strs: string[] = [];
56 stringifyPromptNodeJSON(element.node, strs);
57 return strs.join('');
58 }
60 function stringifyPromptNodeJSON(node: PromptNodeJSON, strs: string[]): void {
61 if (node.type === PromptNodeType.Text) {
src/vs/platform/registry/common/platform.ts 54 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- platform.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as Assert from '../../../base/common/assert.js';
7 > import * as Types from '../../../base/common/types.js';
8 >
9 > export interface IRegistry {
10 >
11 > /**
12 > * Adds the extension functions and properties defined by data to the
13 > * platform. The provided id must be unique.
14 > * @param id a unique identifier
15 > * @param data a contribution
16 > */
17 > add(id: string, data: any): void;
18 >
19 > /**
20 > * Returns true iff there is an extension with the provided id.
21 > * @param id an extension identifier
22 > */
23 > knows(id: string): boolean;
24 >
25 > /**
26 > * Returns the extension functions and properties defined by the specified key or null.
27 > * @param id an extension identifier
28 > */
29 > as<T>(id: string): T;
30 > }
31 >
32 > class RegistryImpl implements IRegistry {
33 >
34 > private readonly data = new Map<string, any>();
35 >
36 > public add(id: string, data: any): void {
37 > Assert.ok(Types.isString(id)); platform.ts
38 > Assert.ok(Types.isObject(data));
39 > Assert.ok(!this.data.has(id), 'There is already an extension with this id');
40 >
41 > this.data.set(id, data);
42 > }
44 > public knows(id: string): boolean {
45 return this.data.has(id);
46 }
48 > public as(id: string): any {
49 > return this.data.get(id) || null; platform.ts
50 > }
52 > public dispose() {
53 this.data.forEach((value) => {
54 if (Types.isFunction(value.dispose)) {
58 this.data.clear();
59 }
61 > }
62 >
63 > export const Registry: IRegistry = new RegistryImpl();
src/vs/base/common/observableInternal/logging/debugger/rpc.ts 53 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- rpc.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export type ChannelFactory = (handler: IChannelHandler) => IChannel;
7 >
8 > export interface IChannel {
9 > sendNotification(data: unknown): void;
10 > sendRequest(data: unknown): Promise<RpcRequestResult>;
11 > }
12 >
13 > export interface IChannelHandler {
14 > handleNotification(notificationData: unknown): void;
15 > handleRequest(requestData: unknown): Promise<RpcRequestResult> | RpcRequestResult;
16 > }
17 >
18 > export type RpcRequestResult = { type: 'result'; value: unknown } | { type: 'error'; value: unknown };
19 >
20 > export type API = {
21 > host: Side;
22 > client: Side;
23 > };
24 >
25 > export type Side = {
26 > notifications: Record<string, (...args: any[]) => void>;
27 > requests: Record<string, (...args: any[]) => Promise<unknown> | unknown>;
28 > };
29 >
30 > type MakeAsyncIfNot<TFn> = TFn extends (...args: infer TArgs) => infer TResult ? TResult extends Promise<unknown> ? TFn : (...args: TArgs) => Promise<TResult> : never;
31 >
32 > export type MakeSideAsync<T extends Side> = {
33 > notifications: T['notifications'];
34 > requests: { [K in keyof T['requests']]: MakeAsyncIfNot<T['requests'][K]> };
35 > };
36 >
37 > export class SimpleTypedRpcConnection<T extends Side> {
38 > public static createHost<T extends API>(channelFactory: ChannelFactory, getHandler: () => T['host']): SimpleTypedRpcConnection<MakeSideAsync<T['client']>> {
39 > return new SimpleTypedRpcConnection(channelFactory, getHandler);
40 > }
41 >
42 > public static createClient<T extends API>(channelFactory: ChannelFactory, getHandler: () => T['client']): SimpleTypedRpcConnection<MakeSideAsync<T['host']>> {
43 return new SimpleTypedRpcConnection(channelFactory, getHandler);
44 }
45 > rpc.ts
46 > public readonly api: T;
47 > private readonly _channel: IChannel;
48 >
49 > private constructor(
50 private readonly _channelFactory: ChannelFactory,
51 private readonly _getHandler: () => Side,
95 this.api = { notifications: notifications, requests: requests } as any;
96 }
97 > } rpc.ts
98 >
99 > type OutgoingMessage = [
100 > method: string,
101 > args: unknown[],
102 > ];
src/vs/base/common/process.ts 53 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- process.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { INodeProcess, isMacintosh, isWindows } from './platform.js';
7 >
8 > let safeProcess: Omit<INodeProcess, 'arch'> & { arch: string | undefined };
9 > declare const process: INodeProcess;
10 >
11 > // Native sandbox environment
12 > const vscodeGlobal = (globalThis as { vscode?: { process?: INodeProcess } }).vscode;
13 > if (typeof vscodeGlobal !== 'undefined' && typeof vscodeGlobal.process !== 'undefined') {
14 const sandboxProcess: INodeProcess = vscodeGlobal.process;
15 safeProcess = {
20 };
21 }
22 > process.ts
23 > // Native node.js environment
24 > else if (typeof process !== 'undefined' && typeof process?.versions?.node === 'string') {
25 > safeProcess = {
26 > get platform() { return process.platform; },
27 > get arch() { return process.arch; },
28 > get env() { return process.env; },
29 > cwd() { return process.env['VSCODE_CWD'] || process.cwd(); }
30 > };
31 }
32
44 };
45 }
46 > process.ts
47 > /**
48 > * Provides safe access to the `cwd` property in node.js, sandboxed or web
49 > * environments.
50 > *
51 > * Note: in web, this property is hardcoded to be `/`.
52 > *
53 > * @skipMangle
54 > */
55 > export const cwd = safeProcess.cwd;
56 >
57 > /**
58 > * Provides safe access to the `env` property in node.js, sandboxed or web
59 > * environments.
60 > *
61 > * Note: in web, this property is hardcoded to be `{}`.
62 > */
63 > export const env = safeProcess.env;
64 >
65 > /**
66 > * Provides safe access to the `platform` property in node.js, sandboxed or web
67 > * environments.
68 > */
69 > export const platform = safeProcess.platform;
70 >
71 > /**
72 > * Provides safe access to the `arch` method in node.js, sandboxed or web
73 > * environments.
74 > * Note: `arch` is `undefined` in web
75 > */
76 > export const arch = safeProcess.arch;
src/vs/platform/agentHost/common/agentHostSessionType.ts 53 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentHostSessionType.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { type AgentProvider } from './agentService.js';
7 >
8 > const REMOTE_AGENT_HOST_SESSION_TYPE_PREFIX = 'remote-';
9 >
10 > /**
11 > * Builds the unique per-connection identifier for a remote agent host.
12 > *
13 > * This string is used as the resource URI scheme registered via
14 > * `registerChatSessionContentProvider` and as the language model vendor /
15 > * `targetChatSessionType` published by `AgentHostLanguageModelProvider`.
16 > */
17 > export function remoteAgentHostSessionTypeId(connectionAuthority: string, agentProvider: AgentProvider): string {
18 return `${remoteAgentHostSessionTypeAuthorityPrefix(connectionAuthority)}${agentProvider}`;
19 }
21 > /**
22 > * Builds the authority-specific prefix for remote agent host session types.
23 > */
24 > export function remoteAgentHostSessionTypeAuthorityPrefix(connectionAuthority: string): string {
25 return `${REMOTE_AGENT_HOST_SESSION_TYPE_PREFIX}${connectionAuthority}-`;
26 }
28 > /**
29 > * Returns whether the given session type uses the remote agent host scheme.
30 > */
31 > export function isRemoteAgentHostSessionType(sessionType: string): boolean {
32 > return sessionType.startsWith(REMOTE_AGENT_HOST_SESSION_TYPE_PREFIX); agentHostSessionType.ts
33 > }
35 > /**
36 > * Finds the best matching remote agent host authority from a known candidate set.
37 > *
38 > * Remote session types are formatted as `remote-{authority}-{provider}` and
39 > * authorities may contain `-`, so callers should match against the full set of
40 > * known authorities instead of splitting the session type.
41 > */
42 > export function findRemoteAgentHostSessionTypeAuthority(sessionType: string, connectionAuthorities: Iterable<string>): string | undefined {
43 if (!isRemoteAgentHostSessionType(sessionType)) {
44 return undefined;
53 return bestMatch;
54 }
56 function isRemoteAgentHostSessionTypeForAuthority(sessionType: string, connectionAuthority: string): boolean {
57 return !!connectionAuthority && sessionType.startsWith(remoteAgentHostSessionTypeAuthorityPrefix(connectionAuthority));
58 }
60 > /**
61 > * Extracts the harness/provider suffix from a remote agent host session type.
62 > *
63 > * Remote session types are formatted as `remote-{authority}-{provider}`. The
64 > * authority may contain `-`, but provider names do not, so the harness is the
65 > * final `-`-delimited segment. Returns `undefined` for non-remote session types.
66 > */
67 > export function parseRemoteAgentHostHarness(sessionType: string): string | undefined {
68 if (!isRemoteAgentHostSessionType(sessionType)) {
69 return undefined;
73 return harness || undefined;
74 }
76 > /**
77 > * Extracts the connection authority from a remote agent host session type when the provider is known.
78 > */
79 > export function parseRemoteAgentHostSessionTypeAuthority(sessionType: string, agentProvider: AgentProvider): string | undefined {
80 if (!isRemoteAgentHostSessionType(sessionType)) {
81 return undefined;
src/vs/workbench/contrib/chat/common/model/chatProgressTypes/chatQuestionCarouselData.ts 53 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatQuestionCarouselData.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DeferredPromise } from '../../../../../../base/common/async.js';
7 > import { IMarkdownString } from '../../../../../../base/common/htmlContent.js';
8 > import { IChatQuestion, IChatQuestionAnswers, IChatQuestionCarousel } from '../../chatService/chatService.js';
9 > import { ToolDataSource } from '../../tools/languageModelToolsService.js';
10 >
11 > /**
12 > * Runtime representation of a question carousel with a {@link DeferredPromise}
13 > * that is resolved when the user submits answers. {@link toJSON} strips the
14 > * completion so only serialisable data is persisted.
15 > */
16 > export class ChatQuestionCarouselData implements IChatQuestionCarousel {
17 > public readonly kind = 'questionCarousel' as const;
18 > public readonly completion = new DeferredPromise<{ answers: IChatQuestionAnswers | undefined }>();
19 > public draftAnswers: IChatQuestionAnswers | undefined;
20 > public draftCurrentIndex: number | undefined;
21 > public draftCollapsed: boolean | undefined;
22 > /**
23 > * Set to `true` when the carousel was dismissed because the user typed
24 > * directly in the associated terminal instead of using the carousel UI.
25 > */
26 > public dismissedByTerminalInput?: boolean;
27 >
28 > /**
29 > * True when the input was accepted/answered outside the carousel UI (e.g.
30 > * via voice) without structured answers, so the summary reads "Answered".
31 > */
32 > public answeredExternally?: boolean;
33 >
34 > /**
35 > * Marks the carousel as dismissed with the given answers and clears draft
36 > * state. Safe to call multiple times — subsequent calls are no-ops.
37 > */
38 > dismiss(answers: IChatQuestionAnswers | undefined): void {
39 > if (this.isUsed) {
41 > }
42 > this.data = answers ?? {}; chatQuestionCarouselData.ts
43 > this.isUsed = true;
44 > this.draftAnswers = undefined;
45 > this.draftCurrentIndex = undefined;
46 > this.draftCollapsed = undefined;
47 > void this.completion.complete({ answers });
48 > }
49 >
50 > constructor(
51 public questions: IChatQuestion[],
52 public allowSkip: boolean,
58 public terminalId?: string,
59 ) { }
61 > toJSON(): IChatQuestionCarousel {
62 return {
63 kind: this.kind,
src/vs/base/common/assert.ts 52 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- assert.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { BugIndicatingError, onUnexpectedError } from './errors.js';
7 >
8 > /**
9 > * Throws an error with the provided message if the provided value does not evaluate to a true Javascript value.
10 > *
11 > * @deprecated Use `assert(...)` instead.
12 > * This method is usually used like this:
13 > * ```ts
14 > * import * as assert from 'vs/base/common/assert';
15 > * assert.ok(...);
16 > * ```
17 > *
18 > * However, `assert` in that example is a user chosen name.
19 > * There is no tooling for generating such an import statement.
20 > * Thus, the `assert(...)` function should be used instead.
21 > */
22 > export function ok(value?: unknown, message?: string) {
23 > if (!value) { assert.ts
24 throw new Error(message ? `Assertion failed (${message})` : 'Assertion Failed');
25 }
26 > } assert.ts
27 > assert.ts
28 > export function assertNever(value: never, message = 'Unreachable'): never {
29 throw new Error(message);
30 }
31 > assert.ts
32 > export function softAssertNever(value: never): void {
33 // no-op
34 }
35 > assert.ts
36 > /**
37 > * Asserts that a condition is `truthy`.
38 > *
39 > * @throws provided {@linkcode messageOrError} if the {@linkcode condition} is `falsy`.
40 > *
41 > * @param condition The condition to assert.
42 > * @param messageOrError An error message or error object to throw if condition is `falsy`.
43 > */
44 > export function assert(
45 condition: boolean,
46 messageOrError: string | Error = 'unexpected state',
55 }
56 }
57 > assert.ts
58 > /**
59 > * Like assert, but doesn't throw.
60 > */
61 > export function softAssert(condition: boolean, message = 'Soft Assertion Failed'): void {
62 if (!condition) {
63 onUnexpectedError(new BugIndicatingError(message));
64 }
65 }
66 > assert.ts
67 > /**
68 > * condition must be side-effect free!
69 > */
70 > export function assertFn(condition: () => boolean): void {
71 > if (!condition()) { assert.ts
72 // eslint-disable-next-line no-debugger
73 debugger;
76 onUnexpectedError(new BugIndicatingError('Assertion Failed'));
77 }
78 > } assert.ts
79 > assert.ts
80 > export function checkAdjacentItems<T>(items: readonly T[], predicate: (item1: T, item2: T) => boolean): boolean {
81 let i = 0;
82 while (i < items.length - 1) {
src/vs/base/common/observableInternal/logging/consoleObservableLogger.ts 52 covered LOC · 20 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- consoleObservableLogger.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable } from '../base.js';
7 > import { TransactionImpl } from '../transaction.js';
8 > import { IObservableLogger, IChangeInformation, addLogger } from './logging.js';
9 > import { FromEventObservable } from '../observables/observableFromEvent.js';
10 > import { getClassName } from '../debugName.js';
11 > import { Derived } from '../observables/derivedImpl.js';
12 > import { AutorunObserver } from '../reactions/autorunImpl.js';
13 >
14 > let consoleObservableLogger: ConsoleObservableLogger | undefined;
15 >
16 > export function logObservableToConsole(obs: IObservable<any>): void {
17 if (!consoleObservableLogger) {
18 consoleObservableLogger = new ConsoleObservableLogger();
21 consoleObservableLogger.addFilteredObj(obs);
22 }
24 > export class ConsoleObservableLogger implements IObservableLogger {
25 private indentation = 0;
26
118
119 private readonly changedObservablesSets = new WeakMap<object, Set<IObservable<any>>>();
121 > formatChanges(changes: Set<IObservable<any>>): ConsoleText | undefined {
122 if (changes.size === 0) {
123 return undefined;
130 );
131 }
133 > handleDerivedDependencyChanged(derived: Derived<any>, observable: IObservable<any>, change: unknown): void {
134 if (!this._isIncluded(derived)) { return; }
135
136 this.changedObservablesSets.get(derived)?.add(observable);
137 }
139 > _handleDerivedRecomputed(derived: Derived<unknown>, info: IChangeInformation): void {
140 if (!this._isIncluded(derived)) { return; }
141
151 changedObservables.clear();
152 }
154 > handleDerivedCleared(derived: Derived<unknown>): void {
155 if (!this._isIncluded(derived)) { return; }
156
160 ]));
161 }
163 > handleFromEventObservableTriggered(observable: FromEventObservable<any, any>, info: IChangeInformation): void {
164 if (!this._isIncluded(observable)) { return; }
165
171 ]));
172 }
174 > handleAutorunCreated(autorun: AutorunObserver): void {
175 if (!this._isIncluded(autorun)) { return; }
176
177 this.changedObservablesSets.set(autorun, new Set());
178 }
180 > handleAutorunDisposed(autorun: AutorunObserver): void {
181 }
183 > handleAutorunDependencyChanged(autorun: AutorunObserver, observable: IObservable<any>, change: unknown): void {
184 if (!this._isIncluded(autorun)) { return; }
185
186 this.changedObservablesSets.get(autorun)!.add(observable);
187 }
189 > handleAutorunStarted(autorun: AutorunObserver): void {
190 const changedObservables = this.changedObservablesSets.get(autorun);
191 if (!changedObservables) { return; }
202 this.indentation++;
203 }
205 > handleAutorunFinished(autorun: AutorunObserver): void {
206 this.indentation--;
207 }
209 > handleBeginTransaction(transaction: TransactionImpl): void {
210 let transactionName = transaction.getDebugName();
211 if (transactionName === undefined) {
221 this.indentation++;
222 }
224 > handleEndTransaction(): void {
225 this.indentation--;
226 }
228 > type ConsoleText = (ConsoleText | undefined)[] |
229 > { text: string; style: string; data?: unknown[] } |
230 > { data: unknown[] };
231 function consoleTextToArgs(text: ConsoleText): unknown[] {
232 const styles = new Array<any>();
294 };
295 }
297 > export function formatValue(value: unknown, availableLen: number): string {
298 switch (typeof value) {
299 case 'number':
325 }
326 }
328 function formatArray(value: unknown[], availableLen: number): string {
329 let result = '[ ';
343 return result;
344 }
346 function formatObject(value: object, availableLen: number): string {
347 if (typeof value.toString === 'function' && value.toString !== Object.prototype.toString) {
371 return result;
372 }
374 function repeat(str: string, count: number): string {
375 let result = '';
379 return result;
380 }
382 function padStr(str: string, length: number): string {
383 while (str.length < length) {
src/vs/platform/remote/common/remoteHosts.ts 52 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- remoteHosts.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Schemas } from '../../../base/common/network.js';
7 > import { URI } from '../../../base/common/uri.js';
8 >
9 > export function getRemoteAuthority(uri: URI): string | undefined {
10 return uri.scheme === Schemas.vscodeRemote ? uri.authority : undefined;
11 }
13 > export function getRemoteName(authority: string): string;
14 > export function getRemoteName(authority: undefined): undefined;
15 > export function getRemoteName(authority: string | undefined): string | undefined;
16 > export function getRemoteName(authority: string | undefined): string | undefined {
17 if (!authority) {
18 return undefined;
25 return authority.substr(0, pos);
26 }
28 > /**
29 > * Returns the suffix part of the authority after the '+' character.
30 > * For remote connections, this is typically the server/tunnel identifier.
31 > * Examples:
32 > * - For tunnels: `tunnel+myTunnel` returns `myTunnel`
33 > * - For SSH: `ssh+myserver` returns `myserver`
34 > * - For localhost: `localhost:8000` returns `undefined`
35 > * @param authority The remote authority string.
36 > * @returns The suffix after the '+' character, or undefined if there is no '+' character.
37 > */
38 > export function getRemoteServerRootPath(authority: string): string | undefined;
39 > export function getRemoteServerRootPath(authority: undefined): undefined;
40 > export function getRemoteServerRootPath(authority: string | undefined): string | undefined;
41 > export function getRemoteServerRootPath(authority: string | undefined): string | undefined {
42 if (!authority) {
43 return undefined;
49 return authority.substring(pos + 1);
50 }
52 > export function parseAuthorityWithPort(authority: string): { host: string; port: number } {
53 const { host, port } = parseAuthority(authority);
54 if (typeof port === 'undefined') {
57 return { host, port };
58 }
60 > export function parseAuthorityWithOptionalPort(authority: string, defaultPort: number): { host: string; port: number } {
61 let { host, port } = parseAuthority(authority);
62 if (typeof port === 'undefined') {
65 return { host, port };
66 }
68 function parseAuthority(authority: string): { host: string; port: number | undefined } {
69 // check for ipv6 with port
88 return { host: authority, port: undefined };
89 }
91 > const loopbackHosts = new Set([
92 > 'localhost',
93 > '127.0.0.1',
94 > '::1',
95 > '[::1]',
96 > '0000:0000:0000:0000:0000:0000:0000:0001',
97 > '[0000:0000:0000:0000:0000:0000:0000:0001]'
98 > ]);
99 >
100 > /**
101 > * Returns whether the given host (as found in a direct `<host>:<port>` remote
102 > * authority) refers to the local loopback interface. The check is intentionally
103 > * strict: only `localhost` and the IPv4/IPv6 loopback literals are considered
104 > * local. Any other host (a routable IP address or a hostname) is treated as a
105 > * connection that leaves the local machine.
106 > */
107 > export function isLoopbackHost(host: string): boolean {
108 return loopbackHosts.has(host.toLowerCase());
109 }
src/vs/workbench/services/notebook/common/notebookDocumentService.ts 52 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- notebookDocumentService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer, decodeBase64, encodeBase64 } from '../../../../base/common/buffer.js';
7 > import { ResourceMap } from '../../../../base/common/map.js';
8 > import { Schemas } from '../../../../base/common/network.js';
9 > import { URI } from '../../../../base/common/uri.js';
10 > import { InstantiationType, registerSingleton } from '../../../../platform/instantiation/common/extensions.js';
11 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
12 >
13 > export const INotebookDocumentService = createDecorator<INotebookDocumentService>('notebookDocumentService');
14 >
15 > export interface INotebookDocument {
16 > readonly uri: URI;
17 > getCellIndex(cellUri: URI): number | undefined;
18 > }
19 >
20 > const _lengths = ['W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f'];
21 > const _padRegexp = new RegExp(`^[${_lengths.join('')}]+`);
22 > const _radix = 7;
23 > export function parse(cell: URI): { notebook: URI; handle: number } | undefined {
24 if (cell.scheme !== Schemas.vscodeNotebookCell) {
25 return undefined;
42 };
43 }
45 > export function generate(notebook: URI, handle: number): URI {
46
47 const s = handle.toString(_radix);
51 return notebook.with({ scheme: Schemas.vscodeNotebookCell, fragment });
52 }
54 > export function parseMetadataUri(metadata: URI): URI | undefined {
55 if (metadata.scheme !== Schemas.vscodeNotebookMetadata) {
56 return undefined;
61 return metadata.with({ scheme: _scheme, fragment: null });
62 }
64 > export function generateMetadataUri(notebook: URI): URI {
65 const fragment = `${encodeBase64(VSBuffer.fromString(notebook.scheme), true, true)}`;
66 return notebook.with({ scheme: Schemas.vscodeNotebookMetadata, fragment });
67 }
69 > export function extractCellOutputDetails(uri: URI): { notebook: URI; openIn: string; outputId?: string; cellFragment?: string; outputIndex?: number; cellHandle?: number; cellIndex?: number } | undefined {
70 if (uri.scheme !== Schemas.vscodeNotebookCellOutput) {
71 return;
97 };
98 }
100 >
101 > export interface INotebookDocumentService {
102 > readonly _serviceBrand: undefined;
103 >
104 > getNotebook(uri: URI): INotebookDocument | undefined;
105 > addNotebookDocument(document: INotebookDocument): void;
106 > removeNotebookDocument(document: INotebookDocument): void;
107 > }
108 >
109 > export class NotebookDocumentWorkbenchService implements INotebookDocumentService {
110 declare readonly _serviceBrand: undefined;
111
112 private readonly _documents = new ResourceMap<INotebookDocument>();
114 > getNotebook(uri: URI): INotebookDocument | undefined {
115 if (uri.scheme === Schemas.vscodeNotebookCell) {
116 const cellUri = parse(uri);
134 return this._documents.get(uri);
135 }
137 > addNotebookDocument(document: INotebookDocument) {
138 this._documents.set(document.uri, document);
139 }
141 > removeNotebookDocument(document: INotebookDocument) {
142 this._documents.delete(document.uri);
143 }
145 > }
146 >
147 > registerSingleton(INotebookDocumentService, NotebookDocumentWorkbenchService, InstantiationType.Delayed);
src/vs/platform/mcp/common/mcpGalleryManifest.ts 50 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpGalleryManifest.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { createDecorator } from '../../instantiation/common/instantiation.js';
8 >
9 > export const enum McpGalleryResourceType {
10 > McpServersQueryService = 'McpServersQueryService',
11 > McpServerWebUri = 'McpServerWebUriTemplate',
12 > McpServerVersionUri = 'McpServerVersionUriTemplate',
13 > McpServerIdUri = 'McpServerIdUriTemplate',
14 > McpServerLatestVersionUri = 'McpServerLatestVersionUriTemplate',
15 > McpServerNamedResourceUri = 'McpServerNamedResourceUriTemplate',
16 > PublisherUriTemplate = 'PublisherUriTemplate',
17 > ContactSupportUri = 'ContactSupportUri',
18 > PrivacyPolicyUri = 'PrivacyPolicyUri',
19 > TermsOfServiceUri = 'TermsOfServiceUri',
20 > ReportUri = 'ReportUri',
21 > }
22 >
23 > export type McpGalleryManifestResource = {
24 > readonly id: string;
25 > readonly type: string;
26 > };
27 >
28 > export interface IMcpGalleryManifest {
29 > readonly version: string;
30 > readonly url: string;
31 > readonly resources: readonly McpGalleryManifestResource[];
32 > }
33 >
34 > export const enum McpGalleryManifestStatus {
35 > Available = 'available',
36 > Unavailable = 'unavailable'
37 > }
38 >
39 > export const IMcpGalleryManifestService = createDecorator<IMcpGalleryManifestService>('IMcpGalleryManifestService');
40 >
41 > export interface IMcpGalleryManifestService {
42 > readonly _serviceBrand: undefined;
43 >
44 > readonly mcpGalleryManifestStatus: McpGalleryManifestStatus;
45 > readonly onDidChangeMcpGalleryManifestStatus: Event<McpGalleryManifestStatus>;
46 > readonly onDidChangeMcpGalleryManifest: Event<IMcpGalleryManifest | null>;
47 > getMcpGalleryManifest(): Promise<IMcpGalleryManifest | null>;
48 > }
49 >
50 > export function getMcpGalleryManifestResourceUri(manifest: IMcpGalleryManifest, type: string): string | undefined {
51 const [name, version] = type.split('/');
52 for (const resource of manifest.resources) {
src/vs/base/common/observableInternal/observables/lazyObservableValue.ts 49 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- lazyObservableValue.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { EqualityComparer } from '../commonFacade/deps.js';
7 > import { IObserver, ISettableObservable, ITransaction } from '../base.js';
8 > import { TransactionImpl } from '../transaction.js';
9 > import { DebugNameData } from '../debugName.js';
10 > import { getLogger } from '../logging/logging.js';
11 > import { BaseObservable } from './baseObservable.js';
12 > import { DebugLocation } from '../debugLocation.js';
13 >
14 > /**
15 > * Holds off updating observers until the value is actually read.
16 > */
17 > export class LazyObservableValue<T, TChange = void>
18 > extends BaseObservable<T, TChange>
19 > implements ISettableObservable<T, TChange> {
20 > protected _value: T;
21 > private _isUpToDate = true;
22 > private readonly _deltas: TChange[] = [];
23 >
24 > get debugName() {
25 > return this._debugNameData.getDebugName(this) ?? 'LazyObservableValue';
26 > }
27 >
28 > constructor(
29 private readonly _debugNameData: DebugNameData,
30 initialValue: T,
35 this._value = initialValue;
36 }
38 > public override get(): T {
39 this._update();
40 return this._value;
41 }
43 > private _update(): void {
44 if (this._isUpToDate) {
45 return;
62 }
63 }
65 > private _updateCounter = 0;
66 >
67 > private _beginUpdate(): void {
68 this._updateCounter++;
69 if (this._updateCounter === 1) {
73 }
74 }
76 > private _endUpdate(): void {
77 this._updateCounter--;
78 if (this._updateCounter === 0) {
86 }
87 }
89 > public override addObserver(observer: IObserver): void {
90 const shouldCallBeginUpdate = !this._observers.has(observer) && this._updateCounter > 0;
91 super.addObserver(observer);
95 }
96 }
98 > public override removeObserver(observer: IObserver): void {
99 const shouldCallEndUpdate = this._observers.has(observer) && this._updateCounter > 0;
100 super.removeObserver(observer);
105 }
106 }
108 > public set(value: T, tx: ITransaction | undefined, change: TChange): void {
109 if (change === undefined && this._equalityComparator(this._value, value)) {
110 return;
src/vs/workbench/common/memento.ts 49 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- memento.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStorageService, IStorageValueChangeEvent, StorageScope, StorageTarget } from '../../platform/storage/common/storage.js';
7 > import { isEmptyObject } from '../../base/common/types.js';
8 > import { onUnexpectedError } from '../../base/common/errors.js';
9 > import { DisposableStore } from '../../base/common/lifecycle.js';
10 > import { Event } from '../../base/common/event.js';
11 >
12 > export class Memento<T extends object> {
13 >
14 > private static readonly applicationMementos = new Map<string, ScopedMemento<unknown>>();
15 > private static readonly applicationSharedMementos = new Map<string, ScopedMemento<unknown>>();
16 > private static readonly profileMementos = new Map<string, ScopedMemento<unknown>>();
17 > private static readonly workspaceMementos = new Map<string, ScopedMemento<unknown>>();
18 >
19 > private static readonly COMMON_PREFIX = 'memento/';
20 >
21 > private readonly id: string;
22 >
23 > constructor(id: string, private storageService: IStorageService) {
24 this.id = Memento.COMMON_PREFIX + id;
25 }
26 > memento.ts
27 > getMemento(scope: StorageScope, target: StorageTarget): Partial<T> {
28 switch (scope) {
29 case StorageScope.WORKSPACE: {
68 }
69 }
70 > memento.ts
71 > onDidChangeValue(scope: StorageScope, disposables: DisposableStore): Event<IStorageValueChangeEvent> {
72 return this.storageService.onDidChangeValue(scope, this.id, disposables);
73 }
74 > memento.ts
75 > saveMemento(): void {
76 Memento.workspaceMementos.get(this.id)?.save();
77 Memento.profileMementos.get(this.id)?.save();
79 Memento.applicationSharedMementos.get(this.id)?.save();
80 }
81 > memento.ts
82 > reloadMemento(scope: StorageScope): void {
83 let memento: ScopedMemento<unknown> | undefined;
84 switch (scope) {
99 memento?.reload();
100 }
101 > memento.ts
102 > static clear(scope: StorageScope): void {
103 switch (scope) {
104 case StorageScope.WORKSPACE:
116 }
117 }
118 > } memento.ts
119 >
120 > class ScopedMemento<T> {
121 >
122 > private mementoObj: Partial<T>;
123 >
124 > constructor(private id: string, private scope: StorageScope, private target: StorageTarget, private storageService: IStorageService) {
125 this.mementoObj = this.doLoad();
126 }
127 > memento.ts
128 > private doLoad(): Partial<T> {
129 try {
130 return this.storageService.getObject(this.id, this.scope, {});
139 return {};
140 }
141 > memento.ts
142 > getMemento(): Partial<T> {
143 return this.mementoObj;
144 }
145 > memento.ts
146 > reload(): void {
147
148 // Clear old
154 Object.assign(this.mementoObj, this.doLoad());
155 }
156 > memento.ts
157 > save(): void {
158 if (!isEmptyObject(this.mementoObj)) {
159 this.storageService.store(this.id, this.mementoObj, this.scope, this.target);
src/vs/base/common/linkedList.ts 48 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- linkedList.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > class Node<E> {
7 >
8 > static readonly Undefined = new Node<unknown>(undefined);
9 >
10 > element: E;
11 > next: Node<E> | typeof Node.Undefined;
12 > prev: Node<E> | typeof Node.Undefined;
13 >
14 > constructor(element: E) {
15 > this.element = element;
16 > this.next = Node.Undefined;
17 > this.prev = Node.Undefined;
18 > }
19 > }
20 >
21 > export class LinkedList<E> {
23 > private _first: Node<E> | typeof Node.Undefined = Node.Undefined;
24 > private _last: Node<E> | typeof Node.Undefined = Node.Undefined;
25 > private _size: number = 0;
27 > get size(): number {
28 return this._size;
29 }
31 > isEmpty(): boolean {
32 return this._first === Node.Undefined;
33 }
35 > clear(): void {
36 let node = this._first;
37 while (node !== Node.Undefined) {
46 this._size = 0;
47 }
49 > unshift(element: E): () => void {
50 return this._insert(element, false);
51 }
53 > push(element: E): () => void {
54 return this._insert(element, true);
55 }
57 > private _insert(element: E, atTheEnd: boolean): () => void {
58 const newNode = new Node(element);
59 if (this._first === Node.Undefined) {
85 };
86 }
88 > shift(): E | undefined {
89 if (this._first === Node.Undefined) {
90 return undefined;
95 }
96 }
98 > pop(): E | undefined {
99 if (this._last === Node.Undefined) {
100 return undefined;
105 }
106 }
108 > peek(): E | undefined {
109 if (this._last === Node.Undefined) {
110 return undefined;
114 }
115 }
117 > private _remove(node: Node<E> | typeof Node.Undefined): void {
118 if (node.prev !== Node.Undefined && node.next !== Node.Undefined) {
119 // middle
141 this._size -= 1;
142 }
144 > *[Symbol.iterator](): Iterator<E> {
145 let node = this._first;
146 while (node !== Node.Undefined) {
src/vs/base/common/managedSettings.ts 48 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- managedSettings.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * A single enterprise-managed marketplace entry, preserving the marketplace
8 > * name (used as `displayLabel`) and the original `source` discriminator.
9 > */
10 > export type IExtraKnownMarketplaceEntry =
11 > | { readonly name: string; readonly source: { readonly source: 'github'; readonly repo: string; readonly ref?: string } }
12 > | { readonly name: string; readonly source: { readonly source: 'git'; readonly url: string; readonly ref?: string } };
13 >
14 > /**
15 > * A single entry in the enterprise-managed `strictKnownMarketplaces` allowlist
16 > * (the `chat.plugins.strictMarketplaces` setting), a discriminated union on
17 > * `source`. Delivered as JSON via managed settings (the server endpoint or
18 > * native MDM) and validated at match time, so the optional fields are only
19 > * meaningful for their corresponding `source`.
20 > */
21 > export interface IStrictMarketplaceSource {
22 > readonly source: 'github' | 'git' | 'url' | 'npm' | 'file' | 'directory' | 'hostPattern' | 'pathPattern';
23 > readonly repo?: string;
24 > readonly url?: string;
25 > readonly ref?: string;
26 > readonly path?: string;
27 > readonly package?: string;
28 > readonly hostPattern?: string;
29 > readonly pathPattern?: string;
30 > readonly headers?: Readonly<Record<string, string>>;
31 > }
32 >
33 > /**
34 > * Converts an {@link IExtraKnownMarketplaceEntry} array into the
35 > * `{ [name]: url-or-shorthand }` dict stored on the `chat.plugins.extraMarketplaces`
36 > * setting (and carried as the canonical JSON value of the `extraKnownMarketplaces`
37 > * managed setting across both the server endpoint and native MDM delivery).
38 > *
39 > * Plain-string entries (allowed by the policy schema but unnamed) are stored with
40 > * the value used as both key and value so they survive the round-trip intact.
41 > *
42 > * Marketplace names come from managed settings (untrusted input) and are written as object keys,
43 > * so `__proto__` / `constructor` / `prototype` keys are skipped to avoid prototype pollution
44 > * (mirroring the guard in the managed-settings normalizer's string-map encoder).
45 > */
46 > export function extraKnownMarketplacesToConfigDict(entries: readonly (string | IExtraKnownMarketplaceEntry)[] | undefined): Record<string, string> | undefined {
47 if (!entries?.length) {
48 return undefined;
66 return obj;
67 }
69 > /** Whether a marketplace name would pollute the prototype chain if used as an object key. */
70 function isUnsafeMarketplaceKey(key: string): boolean {
71 return key === '__proto__' || key === 'constructor' || key === 'prototype';
src/vs/base/test/common/virtualScheduling/timeApi.ts 48 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- timeApi.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export interface TimeoutId { readonly _timeoutIdBrand: void }
7 > export interface IntervalId { readonly _intervalIdBrand: void }
8 > export interface ImmediateId { readonly _immediateIdBrand: void }
9 > export type AnimationFrameId = number & { readonly _animationFrameIdBrand: void };
10 >
11 > /**
12 > * The subset of host time APIs the processor and embeddings need.
13 > *
14 > * Used both for the real host API (captured via {@link captureGlobalTimeApi})
15 > * and for the virtual replacement that runs through a {@link VirtualClock}.
16 > *
17 > * Keeping this as a plain interface means the processor never reaches into
18 > * `globalThis` directly: the boundary between "real time" and "virtual time"
19 > * is exactly which `TimeApi` instance is in use.
20 > */
21 > export interface TimeApi {
22 > setTimeout(handler: () => void, timeout?: number): TimeoutId;
23 > clearTimeout(id: TimeoutId): void;
24 > setInterval(handler: () => void, interval: number): IntervalId;
25 > clearInterval(id: IntervalId): void;
26 > setImmediate?: ((handler: () => void) => ImmediateId);
27 > clearImmediate?: ((id: ImmediateId) => void);
28 > requestAnimationFrame?: ((cb: (time: number) => void) => AnimationFrameId);
29 > cancelAnimationFrame?: ((id: AnimationFrameId) => void);
30 > Date: DateConstructor;
31 > }
32 >
33 > export function captureGlobalTimeApi(): TimeApi {
34 > return {
35 > setTimeout: globalThis.setTimeout.bind(globalThis) as unknown as TimeApi['setTimeout'],
36 > clearTimeout: globalThis.clearTimeout.bind(globalThis) as unknown as TimeApi['clearTimeout'],
37 > setInterval: globalThis.setInterval.bind(globalThis) as unknown as TimeApi['setInterval'],
38 > clearInterval: globalThis.clearInterval.bind(globalThis) as unknown as TimeApi['clearInterval'],
39 > setImmediate: globalThis.setImmediate?.bind(globalThis) as unknown as TimeApi['setImmediate'],
40 > clearImmediate: globalThis.clearImmediate?.bind(globalThis) as unknown as TimeApi['clearImmediate'],
41 > requestAnimationFrame: globalThis.requestAnimationFrame?.bind(globalThis) as unknown as TimeApi['requestAnimationFrame'],
42 > cancelAnimationFrame: globalThis.cancelAnimationFrame?.bind(globalThis) as unknown as TimeApi['cancelAnimationFrame'],
43 > Date: globalThis.Date,
44 > };
45 > }
46 >
47 > /** A snapshot of the real host time API at module-load time. */
48 > export const realTimeApi: TimeApi = captureGlobalTimeApi();
src/vs/base/common/iconLabels.ts 47 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- iconLabels.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IMatch, matchesFuzzy } from './filters.js';
7 > import { ltrim } from './strings.js';
8 > import { ThemeIcon } from './themables.js';
9 >
10 > const iconStartMarker = '$(';
11 >
12 > const iconsRegex = new RegExp(`\\$\\(${ThemeIcon.iconNameExpression}(?:${ThemeIcon.iconModifierExpression})?\\)`, 'g'); // no capturing groups
13 >
14 > const escapeIconsRegex = new RegExp(`(\\\\)?${iconsRegex.source}`, 'g');
15 > export function escapeIcons(text: string): string {
16 return text.replace(escapeIconsRegex, (match, escaped) => escaped ? match : `\\${match}`);
17 }
19 > const markdownEscapedIconsRegex = new RegExp(`\\\\${iconsRegex.source}`, 'g');
20 > export function markdownEscapeEscapedIcons(text: string): string {
21 // Need to add an extra \ for escaping in markdown
22 return text.replace(markdownEscapedIconsRegex, match => `\\${match}`);
23 }
25 > const stripIconsRegex = new RegExp(`(\\s)?(\\\\)?${iconsRegex.source}(\\s)?`, 'g');
26 >
27 > /**
28 > * Takes a label with icons (`$(iconId)xyz`) and strips the icons out (`xyz`)
29 > */
30 > export function stripIcons(text: string): string {
31 if (text.indexOf(iconStartMarker) === -1) {
32 return text;
35 return text.replace(stripIconsRegex, (match, preWhitespace, escaped, postWhitespace) => escaped ? match : preWhitespace || postWhitespace || '');
36 }
38 >
39 > /**
40 > * Takes a label with icons (`$(iconId)xyz`), removes the icon syntax adds whitespace so that screen readers can read the text better.
41 > */
42 > export function getCodiconAriaLabel(text: string | undefined) {
43 if (!text) {
44 return '';
47 return text.replace(/\$\((.*?)\)/g, (_match, codiconName) => ` ${codiconName} `).trim();
48 }
50 >
51 > export interface IParsedLabelWithIcons {
52 > readonly text: string;
53 > readonly iconOffsets?: readonly number[];
54 > }
55 >
56 > const _parseIconsRegex = new RegExp(`\\$\\(${ThemeIcon.iconNameCharacter}+\\)`, 'g');
57 >
58 > /**
59 > * Takes a label with icons (`abc $(iconId)xyz`) and returns the text (`abc xyz`) and the offsets of the icons (`[3]`)
60 > */
61 > export function parseLabelWithIcons(input: string): IParsedLabelWithIcons {
62
63 _parseIconsRegex.lastIndex = 0;
86 return { text, iconOffsets };
87 }
89 >
90 > export function matchesFuzzyIconAware(query: string, target: IParsedLabelWithIcons, enableSeparateSubstringMatching = false): IMatch[] | null {
91 const { text, iconOffsets } = target;
92
src/vs/base/test/common/virtualScheduling/traceLogger.ts 47 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- traceLogger.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { LogEntryLike } from '../executionGraph.js';
7 > import { Trace, TraceContext } from './trace.js';
8 >
9 > /**
10 > * A minimal logger for tests that captures the active {@link Trace} at log
11 > * time so messages can later be woven into a swimlane diagram next to the
12 > * timer events that produced them.
13 > */
14 > export interface ITraceLogger {
15 > log(message: string): void;
16 > warn(message: string): void;
17 > error(message: string): void;
18 > /**
19 > * Run `fn` synchronously and log it as a marker in the trace. The
20 > * marker text is `fn.toString()` so call sites read naturally as e.g.
21 > * `logger.logRun(() => model.trigger())`. Returns whatever `fn`
22 > * returns.
23 > */
24 > logRun<T>(fn: () => T): T;
25 > }
26 >
27 > /**
28 > * One log entry produced by an {@link ITraceLogger}. Extends
29 > * {@link LogEntryLike} (consumed by `buildHistoryFromTasks`) with a level so
30 > * renderers can differentiate `log` / `warn` / `error`.
31 > */
32 > export interface ITraceLogEntry extends LogEntryLike {
33 > readonly trace: Trace;
34 > readonly level: 'log' | 'warn' | 'error';
35 > }
36 >
37 > /**
38 > * Build an {@link ITraceLogger} that pushes every call into `buffer`,
39 > * tagging each entry with the trace that's current at call time.
40 > *
41 > * Pass the same `buffer` to `buildHistoryFromTasks(history, startTime,
42 > * buffer)` to interleave log lines with the timer swimlane.
43 > */
44 > export function createTraceLogger(buffer: ITraceLogEntry[]): ITraceLogger {
45 const make = (level: 'log' | 'warn' | 'error') => (message: string) => {
46 buffer.push({
61 };
62 }
64 > /** Best-effort one-line description of `fn` for trace log markers. */
65 function _describeFn(fn: () => unknown): string {
66 const src = fn.toString();
72 return _collapseWhitespace(src);
73 }
75 function _collapseWhitespace(s: string): string {
76 return s.replace(/\s+/g, ' ').trim();
src/vs/workbench/contrib/chat/common/languageModelsConfiguration.ts 47 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- languageModelsConfiguration.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
9 > import { IRange } from '../../../../editor/common/core/range.js';
10 > import { IStringDictionary } from '../../../../base/common/collections.js';
11 >
12 > export const ILanguageModelsConfigurationService = createDecorator<ILanguageModelsConfigurationService>('ILanguageModelsConfigurationService');
13 >
14 > export interface ConfigureLanguageModelsOptions {
15 > group: ILanguageModelsProviderGroup;
16 > snippet?: string;
17 > snippetTarget?: 'group' | 'models';
18 > }
19 >
20 > export interface ILanguageModelsConfigurationService {
21 > readonly _serviceBrand: undefined;
22 >
23 > readonly configurationFile: URI;
24 >
25 > readonly onDidChangeLanguageModelGroups: Event<readonly ILanguageModelsProviderGroup[]>;
26 >
27 > /** Resolves after the first config-file load attempt (success or failure), so callers can distinguish empty from not-yet-loaded. Never rejects. */
28 > readonly whenReady: Promise<void>;
29 >
30 > getLanguageModelsProviderGroups(): readonly ILanguageModelsProviderGroup[];
31 >
32 > addLanguageModelsProviderGroup(languageModelsProviderGroup: ILanguageModelsProviderGroup): Promise<ILanguageModelsProviderGroup>;
33 >
34 > updateLanguageModelsProviderGroup(from: ILanguageModelsProviderGroup, to: ILanguageModelsProviderGroup): Promise<ILanguageModelsProviderGroup>;
35 >
36 > removeLanguageModelsProviderGroup(languageModelGroup: ILanguageModelsProviderGroup): Promise<void>;
37 >
38 > configureLanguageModels(options?: ConfigureLanguageModelsOptions): Promise<void>;
39 > }
40 >
41 > export interface ILanguageModelsProviderGroup extends IStringDictionary<unknown> {
42 > readonly name: string;
43 > readonly vendor: string;
44 > readonly range?: IRange;
45 > readonly modelsRange?: IRange;
46 > readonly settings?: IStringDictionary<IStringDictionary<unknown>>;
47 > }
src/vs/base/common/observableInternal/debugLocation.ts 46 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- debugLocation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export type DebugLocation = DebugLocationImpl | undefined;
7 >
8 > export namespace DebugLocation {
9 > let enabled = false;
10 >
11 > export function enable(): void {
12 enabled = true;
13 }
15 > export function ofCaller(): DebugLocation {
16 > if (!enabled) { debugLocation.ts
17 > return undefined;
18 > }
19 const Err = Error as ErrorConstructor & { stackTraceLimit: number };
20
25
26 return DebugLocationImpl.fromStack(stack, 2);
29 >
30 > class DebugLocationImpl implements ILocation {
31 > public static fromStack(stack: string, parentIdx: number): DebugLocationImpl | undefined {
32 > const lines = stack.split('\n');
33 > const location = parseLine(lines[parentIdx + 1]);
34 > if (location) {
35 > return new DebugLocationImpl(
36 > location.fileName,
37 > location.line,
38 > location.column,
39 > location.id
40 > );
41 > } else {
42 > return undefined;
43 > }
44 > }
45 >
46 > constructor(
47 public readonly fileName: string,
48 public readonly line: number,
51 ) {
52 }
54 >
55 >
56 > export interface ILocation {
57 > fileName: string;
58 > line: number;
59 > column: number;
60 > id: string;
61 > }
62 >
63 function parseLine(stackLine: string): ILocation | undefined {
64 const match = stackLine.match(/\((.*):(\d+):(\d+)\)/);
src/vs/platform/uriIdentity/common/uriIdentity.ts 46 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- uriIdentity.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../../base/common/uri.js';
7 > import { createDecorator } from '../../instantiation/common/instantiation.js';
8 > import { IExtUri } from '../../../base/common/resources.js';
9 >
10 >
11 > export const IUriIdentityService = createDecorator<IUriIdentityService>('IUriIdentityService');
12 >
13 > export interface IUriIdentityService {
14 >
15 > readonly _serviceBrand: undefined;
16 >
17 > /**
18 > * Uri extensions that are aware of casing.
19 > */
20 > readonly extUri: IExtUri;
21 >
22 > /**
23 > * Returns a canonical uri for the given resource. Different uris can point to the same
24 > * resource. That's because of casing or missing normalization, e.g the following uris
25 > * are different but refer to the same document (because windows paths are not case-sensitive)
26 > *
27 > * ```txt
28 > * file:///c:/foo/bar.txt
29 > * file:///c:/FOO/BAR.txt
30 > * ```
31 > *
32 > * This function should be invoked when feeding uris into the system that represent the truth,
33 > * e.g document uris or marker-to-document associations etc. This function should NOT be called
34 > * to pretty print a label nor to sanitize a uri.
35 > *
36 > * Samples:
37 > *
38 > * | in | out | |
39 > * |---|---|---|
40 > * | `file:///foo/bar/../bar` | `file:///foo/bar` | n/a |
41 > * | `file:///foo/bar/../bar#frag` | `file:///foo/bar#frag` | keep fragment |
42 > * | `file:///foo/BAR` | `file:///foo/bar` | assume ignore case |
43 > * | `file:///foo/bar/../BAR?q=2` | `file:///foo/BAR?q=2` | query makes it a different document |
44 > */
45 > asCanonicalUri(uri: URI): URI;
46 > }
src/vs/base/test/common/virtualScheduling/virtualTimeApi.ts 45 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- virtualTimeApi.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IDisposable } from '../../../common/lifecycle.js';
7 > import { realTimeApi, TimeApi } from './timeApi.js';
8 > import { ROOT_TRACE, TraceContext } from './trace.js';
9 > import { VirtualClock } from './virtualClock.js';
10 >
11 > // V8 default `Error.stackTraceLimit` of 10 swallows everything past the
12 > // first async boundary in the stacks we capture for trace diagnostics.
13 > // Bump it so swimlane callers actually see the user code that scheduled a
14 > // timer rather than just the Promise wrapper.
15 > if (typeof Error.stackTraceLimit === 'number' && Error.stackTraceLimit < 50) {
16 > Error.stackTraceLimit = 50;
17 > }
18 >
19 > /** Virtual timer IDs are `IDisposable`s. Recover one from an opaque id. */
20 function asDisposable(id: unknown): IDisposable | undefined {
21 if (id === null || typeof id !== 'object') { return undefined; }
23 return typeof maybe.dispose === 'function' ? id as IDisposable : undefined;
24 }
26 > export interface CreateVirtualTimeApiOptions {
27 > /**
28 > * If `true`, `requestAnimationFrame` is faked: callbacks are scheduled
29 > * onto the virtual queue at `now + 16ms` and the resulting event hints
30 > * the embedding to use a real `requestAnimationFrame` so the host can
31 > * reflow before the callback runs. Useful for fixtures that need DOM
32 > * measurements after rAF callbacks.
33 > *
34 > * If `false` (default), `requestAnimationFrame` is left to the host.
35 > */
36 > readonly fakeRequestAnimationFrame?: boolean;
37 > }
38 >
39 > /**
40 > * Build a {@link TimeApi} that schedules every timer call into `clock`'s
41 > * virtual queue, capturing the current trace at schedule time so that
42 > * causal chains (`setTimeout` → `setTimeout`, etc.) are preserved.
43 > *
44 > * The returned API is suitable to install with {@link pushGlobalTimeApi},
45 > * which is what {@link runWithFakedTimers} does internally.
46 > */
47 > export function createVirtualTimeApi(
48 clock: VirtualClock,
49 options?: CreateVirtualTimeApiOptions,
190 return api;
191 }
193 > // Re-exported for convenience: many tests want to install both at once.
194 > export { pushGlobalTimeApi } from './globalTimeApi.js';
src/vs/platform/keybinding/common/baseResolvedKeybinding.ts 45 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- baseResolvedKeybinding.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { illegalArgument } from '../../../base/common/errors.js';
7 > import { AriaLabelProvider, ElectronAcceleratorLabelProvider, UILabelProvider, UserSettingsLabelProvider } from '../../../base/common/keybindingLabels.js';
8 > import { Chord, SingleModifierChord, ResolvedKeybinding, ResolvedChord } from '../../../base/common/keybindings.js';
9 > import { OperatingSystem } from '../../../base/common/platform.js';
10 >
11 > export abstract class BaseResolvedKeybinding<T extends Chord> extends ResolvedKeybinding {
12 >
13 > protected readonly _os: OperatingSystem;
14 > protected readonly _chords: readonly T[];
15 >
16 > constructor(os: OperatingSystem, chords: readonly T[]) {
17 super();
18 if (chords.length === 0) {
22 this._chords = chords;
23 }
25 > public getLabel(): string | null {
26 return UILabelProvider.toLabel(this._os, this._chords, (keybinding) => this._getLabel(keybinding));
27 }
29 > public getAriaLabel(): string | null {
30 return AriaLabelProvider.toLabel(this._os, this._chords, (keybinding) => this._getAriaLabel(keybinding));
31 }
33 > public getElectronAccelerator(): string | null {
34 if (this._chords.length > 1) {
35 // [Electron Accelerators] Electron cannot handle chords
43 return ElectronAcceleratorLabelProvider.toLabel(this._os, this._chords, (keybinding) => this._getElectronAccelerator(keybinding));
44 }
46 > public getUserSettingsLabel(): string | null {
47 return UserSettingsLabelProvider.toLabel(this._os, this._chords, (keybinding) => this._getUserSettingsLabel(keybinding));
48 }
50 > public isWYSIWYG(): boolean {
51 return this._chords.every((keybinding) => this._isWYSIWYG(keybinding));
52 }
54 > public hasMultipleChords(): boolean {
55 return (this._chords.length > 1);
56 }
58 > public getChords(): ResolvedChord[] {
59 return this._chords.map((keybinding) => this._getChord(keybinding));
60 }
62 > private _getChord(keybinding: T): ResolvedChord {
63 return new ResolvedChord(
64 keybinding.ctrlKey,
70 );
71 }
73 > public getDispatchChords(): (string | null)[] {
74 return this._chords.map((keybinding) => this._getChordDispatch(keybinding));
75 }
77 > public getSingleModifierDispatchChords(): (SingleModifierChord | null)[] {
78 return this._chords.map((keybinding) => this._getSingleModifierChordDispatch(keybinding));
79 }
81 > protected abstract _getLabel(keybinding: T): string | null;
82 > protected abstract _getAriaLabel(keybinding: T): string | null;
83 > protected abstract _getElectronAccelerator(keybinding: T): string | null;
84 > protected abstract _getUserSettingsLabel(keybinding: T): string | null;
85 > protected abstract _isWYSIWYG(keybinding: T): boolean;
86 > protected abstract _getChordDispatch(keybinding: T): string | null;
87 > protected abstract _getSingleModifierChordDispatch(keybinding: T): SingleModifierChord | null;
88 > }
src/vs/platform/keybinding/common/usLayoutResolvedKeybinding.ts 45 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- usLayoutResolvedKeybinding.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { KeyCode, KeyCodeUtils, IMMUTABLE_CODE_TO_KEY_CODE, ScanCode } from '../../../base/common/keyCodes.js';
7 > import { SingleModifierChord, Chord, KeyCodeChord, Keybinding } from '../../../base/common/keybindings.js';
8 > import { OperatingSystem } from '../../../base/common/platform.js';
9 > import { BaseResolvedKeybinding } from './baseResolvedKeybinding.js';
10 > import { toEmptyArrayIfContainsNull } from './resolvedKeybindingItem.js';
11 >
12 > /**
13 > * Do not instantiate. Use KeybindingService to get a ResolvedKeybinding seeded with information about the current kb layout.
14 > */
15 > export class USLayoutResolvedKeybinding extends BaseResolvedKeybinding<KeyCodeChord> {
16 >
17 > constructor(chords: KeyCodeChord[], os: OperatingSystem) {
18 super(os, chords);
19 }
21 > private _keyCodeToUILabel(keyCode: KeyCode): string {
22 if (this._os === OperatingSystem.Macintosh) {
23 switch (keyCode) {
34 return KeyCodeUtils.toString(keyCode);
35 }
37 > protected _getLabel(chord: KeyCodeChord): string | null {
38 if (chord.isDuplicateModifierCase()) {
39 return '';
41 return this._keyCodeToUILabel(chord.keyCode);
42 }
44 > protected _getAriaLabel(chord: KeyCodeChord): string | null {
45 if (chord.isDuplicateModifierCase()) {
46 return '';
48 return KeyCodeUtils.toString(chord.keyCode);
49 }
51 > protected _getElectronAccelerator(chord: KeyCodeChord): string | null {
52 return KeyCodeUtils.toElectronAccelerator(chord.keyCode);
53 }
55 > protected _getUserSettingsLabel(chord: KeyCodeChord): string | null {
56 if (chord.isDuplicateModifierCase()) {
57 return '';
60 return (result ? result.toLowerCase() : result);
61 }
63 > protected _isWYSIWYG(): boolean {
64 return true;
65 }
67 > protected _getChordDispatch(chord: KeyCodeChord): string | null {
68 return USLayoutResolvedKeybinding.getDispatchStr(chord);
69 }
71 > public static getDispatchStr(chord: KeyCodeChord): string | null {
72 if (chord.isModifierKey()) {
73 return null;
91 return result;
92 }
94 > protected _getSingleModifierChordDispatch(keybinding: KeyCodeChord): SingleModifierChord | null {
95 if (keybinding.keyCode === KeyCode.Ctrl && !keybinding.shiftKey && !keybinding.altKey && !keybinding.metaKey) {
96 return 'ctrl';
107 return null;
108 }
110 > /**
111 > * *NOTE*: Check return value for `KeyCode.Unknown`.
112 > */
113 > private static _scanCodeToKeyCode(scanCode: ScanCode): KeyCode {
114 const immutableKeyCode = IMMUTABLE_CODE_TO_KEY_CODE[scanCode];
115 if (immutableKeyCode !== KeyCode.DependsOnKbLayout) {
170 return KeyCode.Unknown;
171 }
173 > private static _toKeyCodeChord(chord: Chord | null): KeyCodeChord | null {
174 if (!chord) {
175 return null;
184 return new KeyCodeChord(chord.ctrlKey, chord.shiftKey, chord.altKey, chord.metaKey, keyCode);
185 }
187 > public static resolveKeybinding(keybinding: Keybinding, os: OperatingSystem): USLayoutResolvedKeybinding[] {
188 const chords: KeyCodeChord[] = toEmptyArrayIfContainsNull(keybinding.chords.map(chord => this._toKeyCodeChord(chord)));
189 if (chords.length > 0) {
src/vs/workbench/common/resources.ts 45 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- resources.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { URI } from '../../base/common/uri.js';
7 > import { equals } from '../../base/common/objects.js';
8 > import { isAbsolute } from '../../base/common/path.js';
9 > import { Emitter } from '../../base/common/event.js';
10 > import { relativePath } from '../../base/common/resources.js';
11 > import { Disposable } from '../../base/common/lifecycle.js';
12 > import { ParsedExpression, IExpression, parse } from '../../base/common/glob.js';
13 > import { IWorkspaceContextService } from '../../platform/workspace/common/workspace.js';
14 > import { IConfigurationService, IConfigurationChangeEvent } from '../../platform/configuration/common/configuration.js';
15 > import { Schemas } from '../../base/common/network.js';
16 > import { ResourceSet } from '../../base/common/map.js';
17 > import { getDriveLetter } from '../../base/common/extpath.js';
18 >
19 > interface IConfiguredExpression {
20 > readonly expression: IExpression;
21 > readonly hasAbsolutePath: boolean;
22 > }
23 >
24 > export class ResourceGlobMatcher extends Disposable {
25 >
26 > private static readonly NO_FOLDER = null;
27 >
28 > private readonly _onExpressionChange = this._register(new Emitter<void>());
29 > readonly onExpressionChange = this._onExpressionChange.event;
30 >
31 > private readonly mapFolderToParsedExpression = new Map<string | null, ParsedExpression>();
32 > private readonly mapFolderToConfiguredExpression = new Map<string | null, IConfiguredExpression>();
33 >
34 > constructor(
35 private getExpression: (folder?: URI) => IExpression | undefined,
36 private shouldUpdate: (event: IConfigurationChangeEvent) => boolean,
44 this.registerListeners();
45 }
47 > private registerListeners(): void {
48 this._register(this.configurationService.onDidChangeConfiguration(e => {
49 if (this.shouldUpdate(e)) {
54 this._register(this.contextService.onDidChangeWorkspaceFolders(() => this.updateExpressions(true)));
55 }
57 > private updateExpressions(fromEvent: boolean): void {
58 let changed = false;
59
120 }
121 }
122 > resources.ts
123 > private doGetExpression(resource: URI | undefined): IConfiguredExpression | undefined {
124 const expression = this.getExpression(resource);
125 if (!expression) {
164 };
165 }
166 > resources.ts
167 > matches(
168 resource: URI,
169 hasSibling?: (name: string) => boolean
214 return false;
215 }
216 > resources.ts
217 > private uriToPath(uri: URI): string {
218 if (uri.scheme === Schemas.file) {
219 return uri.fsPath;
src/vs/base/test/common/virtualScheduling/runWithFakedTimers.ts 44 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- runWithFakedTimers.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CancellationTokenSource } from '../../../common/cancellation.js';
7 > import { drainMicrotasksEmbedding } from './embedding.js';
8 > import { pushGlobalTimeApi } from './globalTimeApi.js';
9 > import { realTimeApi } from './timeApi.js';
10 > import { untilToken, VirtualTimeProcessor } from './processor.js';
11 > import { createRecordingRealTimeApi, RecordedTimerEvent } from './recordingTimeApi.js';
12 > import { VirtualClock } from './virtualClock.js';
13 > import { createVirtualTimeApi } from './virtualTimeApi.js';
14 >
15 > export interface RunWithFakedTimersOptions {
16 > readonly startTime?: number;
17 > /** Default `true`. Set `false` to bypass virtual time entirely (for
18 > * cases where the same test is parameterised over real/virtual time). */
19 > readonly useFakeTimers?: boolean;
20 > /** No effect in the new processor; accepted for legacy compatibility.
21 > * The drain-microtasks embedding picks the fastest available macrotask
22 > * primitive automatically. */
23 > readonly useSetImmediate?: boolean;
24 > /** Maximum number of virtual events the run is allowed to execute
25 > * before being rejected. Default 100. */
26 > readonly maxTaskCount?: number;
27 > /**
28 > * If set, called once `fn` resolves with the recorded timer events.
29 > * In virtual mode the events come from the {@link VirtualTimeProcessor}'s
30 > * own history; in real mode a recording wrapper around the host time
31 > * API is installed for the duration of `fn`. Useful for swimlane
32 > * diagnostics.
33 > */
34 > readonly onHistory?: (history: readonly RecordedTimerEvent[]) => void;
35 > }
36 >
37 > /**
38 > * Run `fn` with a virtual clock installed as the global time API.
39 > *
40 > * After `fn` resolves, the virtual queue is drained (so any timers `fn`
41 > * scheduled and `await`ed for, transitively, complete deterministically).
42 > * If `fn` throws, the queue is *not* drained — the original error is
43 > * re-thrown immediately.
44 > */
45 export async function runWithFakedTimers<T>(
46 options: RunWithFakedTimersOptions,
src/vs/base/common/uint.ts 43 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- uint.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export const enum Constants {
7 > /**
8 > * MAX SMI (SMall Integer) as defined in v8.
9 > * one bit is lost for boxing/unboxing flag.
10 > * one bit is lost for sign flag.
11 > * See https://thibaultlaurens.github.io/javascript/2013/04/29/how-the-v8-engine-works/#tagged-values
12 > */
13 > MAX_SAFE_SMALL_INTEGER = 1 << 30,
14 >
15 > /**
16 > * MIN SMI (SMall Integer) as defined in v8.
17 > * one bit is lost for boxing/unboxing flag.
18 > * one bit is lost for sign flag.
19 > * See https://thibaultlaurens.github.io/javascript/2013/04/29/how-the-v8-engine-works/#tagged-values
20 > */
21 > MIN_SAFE_SMALL_INTEGER = -(1 << 30),
22 >
23 > /**
24 > * Max unsigned integer that fits on 8 bits.
25 > */
26 > MAX_UINT_8 = 255, // 2^8 - 1
27 >
28 > /**
29 > * Max unsigned integer that fits on 16 bits.
30 > */
31 > MAX_UINT_16 = 65535, // 2^16 - 1
32 >
33 > /**
34 > * Max unsigned integer that fits on 32 bits.
35 > */
36 > MAX_UINT_32 = 4294967295, // 2^32 - 1
37 >
38 > UNICODE_SUPPLEMENTARY_PLANE_BEGIN = 0x010000
39 > }
40 >
41 > export function toUint8(v: number): number {
42 if (v < 0) {
43 return 0;
48 return v | 0;
49 }
50 > uint.ts
51 > export function toUint32(v: number): number {
52 if (v < 0) {
53 return 0;
src/vs/platform/agentHost/common/state/protocol/mcpAppDefaults.ts 43 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mcpAppDefaults.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import type { AhpMcpUiHostCapabilities, McpServerCustomizationApps } from './channels-session/state.js';
7 >
8 > /**
9 > * MCP App capabilities the agent host proxies for every MCP server it
10 > * advertises over the `mcp://` side-channel.
11 > *
12 > * - `serverTools.listChanged` is `true`: we forward
13 > * `notifications/tools/list_changed` whenever the SDK signals that the
14 > * tool inventory has refreshed (see `CopilotAgentSession`).
15 > * - `serverResources` is advertised as an empty object: we serve the
16 > * `resources/*` methods over the channel but do not promise
17 > * `notifications/resources/list_changed` forwarding (no `listChanged`).
18 > * - `sampling` is advertised as an empty object: we serve
19 > * `sampling/createMessage` requests from the App over the `mcp://`
20 > * channel (the agent host handler forwards them to
21 > * `session.rpc.mcp.executeSampling`). The SEP-1577 `tools`
22 > * sub-flag is NOT set — we don't pass through tool content blocks.
23 > *
24 > * Per the AHP spec, `mcpApp` is a static capability declaration —
25 > * "SHOULD be present whenever the server can host Apps" — so this
26 > * constant is set on every MCP customization at construction time,
27 > * regardless of the server's current lifecycle state.
28 > */
29 > export const DEFAULT_MCP_APP_CAPABILITIES: AhpMcpUiHostCapabilities = {
30 > serverTools: { listChanged: true },
31 > serverResources: {},
32 > sampling: {},
33 > };
34 >
35 > /**
36 > * The full `mcpApp` shape applied to a {@link McpServerCustomization}.
37 > * Wraps {@link DEFAULT_MCP_APP_CAPABILITIES} so callers can drop it in
38 > * directly without re-allocating the same wrapper object at every call
39 > * site.
40 > */
41 > export const DEFAULT_MCP_APP: McpServerCustomizationApps = {
42 > capabilities: DEFAULT_MCP_APP_CAPABILITIES,
43 > };
src/vs/platform/observable/common/observableMemento.ts 43 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableMemento.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { strictEquals } from '../../../base/common/equals.js';
7 > import { DisposableStore, IDisposable } from '../../../base/common/lifecycle.js';
8 > import { DebugLocation } from '../../../base/common/observable.js';
9 > // eslint-disable-next-line local/code-no-deep-import-of-internal
10 > import { DebugNameData } from '../../../base/common/observableInternal/debugName.js';
11 > // eslint-disable-next-line local/code-no-deep-import-of-internal
12 > import { ObservableValue } from '../../../base/common/observableInternal/observables/observableValue.js';
13 > import { IStorageService, StorageScope, StorageTarget } from '../../storage/common/storage.js';
14 >
15 > interface IObservableMementoOpts<T> {
16 > defaultValue: T;
17 > key: string;
18 > toStorage: (value: T) => string;
19 > fromStorage: (value: string) => T;
20 > }
21 >
22 > /**
23 > * Defines an observable memento. Returns a function that can be called with
24 > * the specific storage scope, target, and service to use in a class.
25 > *
26 > * Note that the returned Observable is a disposable, because it interacts
27 > * with storage service events, and must be tracked appropriately.
28 > */
29 > export function observableMemento<T>(opts: IObservableMementoOpts<T>) {
30 return (scope: StorageScope, target: StorageTarget, storageService: IStorageService): ObservableMemento<T> => {
31 return new ObservableMemento<T>(opts, scope, target, storageService);
32 };
33 }
35 > /**
36 > * A value that is stored, and is also observable. Note: T should be readonly.
37 > */
38 > export class ObservableMemento<T> extends ObservableValue<T> implements IDisposable {
39 > private readonly _store = new DisposableStore();
40 > private _noStorageUpdateNeeded = false;
41 >
42 > constructor(
43 private readonly opts: IObservableMementoOpts<T>,
44 private readonly storageScope: StorageScope,
73 }));
74 }
76 > protected override _setValue(newValue: T): void {
77 super._setValue(newValue);
78 if (this._noStorageUpdateNeeded) {
82 this.storageService.store(this.opts.key, valueToStore, this.storageScope, this.storageTarget);
83 }
85 > dispose(): void {
86 this._store.dispose();
87 }
src/vs/base/test/common/virtualScheduling/globalTimeApi.ts 42 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- globalTimeApi.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IDisposable } from '../../../common/lifecycle.js';
7 > import { captureGlobalTimeApi, realTimeApi, TimeApi } from './timeApi.js';
8 >
9 > /** Cast through `unknown` so we don't widen our typed `TimeApi` shapes to `any`. */
10 > type AsGlobal<K extends keyof typeof globalThis> = (typeof globalThis)[K];
11 >
12 > /**
13 > * Ensure `fn` carries an `originalFn` back-door pointing at the real
14 > * (non-virtual) `setTimeout`. We prefer the existing tag on `fn`, then a tag
15 > * inherited from `previousFn` (which may itself be a wrapper that already
16 > * carried the back-door), and finally fall back to `realTimeApi.setTimeout`
17 > * — which has its own `originalFn` set at module load.
18 > */
19 function ensureSetTimeoutOriginalFn(fn: TimeApi['setTimeout'], previousFn: TimeApi['setTimeout']): TimeApi['setTimeout'] {
20 const tagged = fn as TimeApi['setTimeout'] & { originalFn?: TimeApi['setTimeout'] };
26 return tagged;
27 }
29 > /**
30 > * Replace the global time APIs (`setTimeout`, `setInterval`, …, `Date`,
31 > * optionally `requestAnimationFrame`) with the ones from `api`. Returns a
32 > * disposable that restores the previous globals.
33 > *
34 > * The previous globals are captured *at install time*, so nested installs
35 > * compose correctly (the disposable restores to whatever was current when
36 > * this call was made, not to the original real values).
37 > *
38 > * `setTimeout.originalFn` is preserved on the installed function so callers
39 > * like the component-explorer host can escape virtual time when polling.
40 > * If `api.setTimeout` does not already carry `originalFn`, it is copied from
41 > * the previous global (or defaulted to the real `setTimeout`) so wrapping
42 > * APIs such as a logging wrapper don't drop the back-door.
43 > */
44 > export function pushGlobalTimeApi(api: TimeApi): IDisposable {
45 const previous = captureGlobalTimeApi();
46
74 };
75 }
77 > // One-shot tag on the *real* setTimeout: lets callers (e.g. the
78 > // component-explorer host's polling loop) escape virtual time even after
79 > // pushGlobalTimeApi has installed a virtual version on top. The `originalFn`
80 > // property is not on the `setTimeout` signature by design — it's a back-door
81 > // convention shared with the polling code.
82 > (realTimeApi.setTimeout as unknown as { originalFn: TimeApi['setTimeout'] }).originalFn = realTimeApi.setTimeout;
src/vs/platform/configuration/test/common/testConfigurationService.ts 42 covered LOC · 11 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- testConfigurationService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Emitter } from '../../../../base/common/event.js';
7 > import { TernarySearchTree } from '../../../../base/common/ternarySearchTree.js';
8 > import { URI } from '../../../../base/common/uri.js';
9 > import { getConfigurationValue, IConfigurationChangeEvent, IConfigurationOverrides, IConfigurationService, IConfigurationValue, isConfigurationOverrides } from '../../common/configuration.js';
10 > import { Extensions, IConfigurationRegistry } from '../../common/configurationRegistry.js';
11 > import { Registry } from '../../../registry/common/platform.js';
12 >
13 > export class TestConfigurationService implements IConfigurationService {
14 > public _serviceBrand: undefined;
15 >
16 > private configuration: Record<string, unknown>;
17 > readonly onDidChangeConfigurationEmitter = new Emitter<IConfigurationChangeEvent>();
18 > readonly onDidChangeConfiguration = this.onDidChangeConfigurationEmitter.event;
19 >
20 > constructor(configuration?: Record<string, unknown>) {
21 > this.configuration = configuration || Object.create(null); testConfigurationService.ts
22 > }
24 > private configurationByRoot: TernarySearchTree<string, Record<string, unknown>> = TernarySearchTree.forPaths<Record<string, unknown>>();
25 >
26 > public reloadConfiguration<T>(): Promise<T> {
27 return Promise.resolve(this.getValue() as T);
28 }
30 > public getValue<T>(arg1?: string | IConfigurationOverrides, arg2?: IConfigurationOverrides): T | undefined {
31 let configuration;
32 const overrides = isConfigurationOverrides(arg1) ? arg1 : isConfigurationOverrides(arg2) ? arg2 : undefined;
42 return configuration as T;
43 }
45 > public updateValue(key: string, value: unknown): Promise<void> {
46 return Promise.resolve(undefined);
47 }
49 > public setUserConfiguration(key: string, value: unknown, root?: URI): Promise<void> {
50 if (root) {
51 const configForRoot = this.configurationByRoot.get(root.fsPath) || Object.create(null);
58 return Promise.resolve(undefined);
59 }
61 > private overrideIdentifiers: Map<string, string[]> = new Map();
62 > public setOverrideIdentifiers(key: string, identifiers: string[]): void {
63 this.overrideIdentifiers.set(key, identifiers);
64 }
66 > public inspect<T>(key: string, overrides?: IConfigurationOverrides): IConfigurationValue<T> {
67 const value = this.getValue(key, overrides) as T;
68
75 };
76 }
78 > public keys() {
79 return {
80 default: Object.keys(Registry.as<IConfigurationRegistry>(Extensions.Configuration).getConfigurationProperties()),
src/vs/platform/instantiation/common/graph.ts 42 covered LOC · 12 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- graph.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export class Node<T> {
7 >
8 >
9 > readonly incoming = new Map<string, Node<T>>();
10 > readonly outgoing = new Map<string, Node<T>>();
11 >
12 > constructor(
13 readonly key: string,
14 readonly data: T
15 ) { }
16 > } graph.ts
17 >
18 > export class Graph<T> {
19 >
20 > private readonly _nodes = new Map<string, Node<T>>();
21 >
22 > constructor(private readonly _hashFn: (element: T) => string) {
23 // empty
24 }
25 > graph.ts
26 > roots(): Node<T>[] {
27 const ret: Node<T>[] = [];
28 for (const node of this._nodes.values()) {
33 return ret;
34 }
35 > graph.ts
36 > insertEdge(from: T, to: T): void {
37 const fromNode = this.lookupOrInsertNode(from);
38 const toNode = this.lookupOrInsertNode(to);
41 toNode.incoming.set(fromNode.key, fromNode);
42 }
43 > graph.ts
44 > removeNode(data: T): void {
45 const key = this._hashFn(data);
46 this._nodes.delete(key);
50 }
51 }
52 > graph.ts
53 > lookupOrInsertNode(data: T): Node<T> {
54 const key = this._hashFn(data);
55 let node = this._nodes.get(key);
62 return node;
63 }
64 > graph.ts
65 > lookup(data: T): Node<T> | undefined {
66 return this._nodes.get(this._hashFn(data));
67 }
68 > graph.ts
69 > isEmpty(): boolean {
70 return this._nodes.size === 0;
71 }
72 > graph.ts
73 > toString(): string {
74 const data: string[] = [];
75 for (const [key, value] of this._nodes) {
79 return data.join('\n');
80 }
81 > graph.ts
82 > /**
83 > * This is brute force and slow and **only** be used
84 > * to trouble shoot.
85 > */
86 > findCycleSlow() {
87 for (const [id, node] of this._nodes) {
88 const seen = new Set<string>([id]);
94 return undefined;
95 }
96 > graph.ts
97 > private _findCycle(node: Node<T>, seen: Set<string>): string | undefined {
98 for (const [id, outgoing] of node.outgoing) {
99 if (seen.has(id)) {
109 return undefined;
110 }
111 > } graph.ts
src/vs/base/common/lazy.ts 39 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- lazy.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > enum LazyValueState {
7 > Uninitialized,
8 > Running,
9 > Completed,
10 > }
11 >
12 > export class Lazy<T> {
13 >
14 > private _state = LazyValueState.Uninitialized;
15 > private _value?: T;
16 > private _error: Error | undefined;
17 >
18 > constructor(
19 > private readonly executor: () => T, lazy.ts
20 > ) { }
21 > lazy.ts
22 > /**
23 > * True if the lazy value has been resolved.
24 > */
25 > get hasValue(): boolean { return this._state === LazyValueState.Completed; }
26 >
27 > /**
28 > * Get the wrapped value.
29 > *
30 > * This will force evaluation of the lazy value if it has not been resolved yet. Lazy values are only
31 > * resolved once. `getValue` will re-throw exceptions that are hit while resolving the value
32 > */
33 > get value(): T {
34 if (this._state === LazyValueState.Uninitialized) {
35 this._state = LazyValueState.Running;
50 return this._value!;
51 }
52 > lazy.ts
53 > /**
54 > * Get the wrapped value without forcing evaluation.
55 > */
56 > get rawValue(): T | undefined { return this._value; }
57 > }
src/vs/platform/product/common/product.ts 39 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- product.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { env } from '../../../base/common/process.js';
7 > import { IProductConfiguration } from '../../../base/common/product.js';
8 > import { ISandboxConfiguration } from '../../../base/parts/sandbox/common/sandboxTypes.js';
9 >
10 > /**
11 > * @deprecated It is preferred that you use `IProductService` if you can. This
12 > * allows web embedders to override our defaults. But for things like `product.quality`,
13 > * the use is fine because that property is not overridable.
14 > */
15 > let product: IProductConfiguration;
16 >
17 > // Native sandbox environment
18 > const vscodeGlobal = (globalThis as { vscode?: { context?: { configuration(): ISandboxConfiguration | undefined } } }).vscode;
19 > if (typeof vscodeGlobal !== 'undefined' && typeof vscodeGlobal.context !== 'undefined') {
20 const configuration: ISandboxConfiguration | undefined = vscodeGlobal.context.configuration();
21 if (configuration) {
25 }
26 }
27 > // _VSCODE environment product.ts
28 > else if (globalThis._VSCODE_PRODUCT_JSON && globalThis._VSCODE_PACKAGE_JSON) {
29 > // Obtain values from product.json and package.json-data
30 > product = globalThis._VSCODE_PRODUCT_JSON as unknown as IProductConfiguration;
31 >
32 > // Running out of sources
33 > if (env['VSCODE_DEV']) {
34 Object.assign(product, {
35 nameShort: `${product.nameShort} Dev`,
39 });
40 }
41 > product.ts
42 > // Version is added during built time, but we still
43 > // want to have it running out of sources so we
44 > // read it from package.json only when we need it.
45 > if (!product.version) {
46 > const pkg = globalThis._VSCODE_PACKAGE_JSON as { version: string };
47 >
48 > Object.assign(product, {
49 > version: pkg.version
50 > });
51 > }
52 }
53
90 }
91 }
92 > product.ts
93 > export default product;
src/vs/base/common/observableInternal/set.ts 38 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- set.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, ITransaction } from '../observable.js';
7 > import { observableValueOpts } from './observables/observableValueOpts.js';
8 >
9 > export class ObservableSet<T> implements Set<T> {
10 > set.ts
11 > private readonly _data = new Set<T>();
12 >
13 > private _obs = observableValueOpts({ equalsFn: () => false }, this);
14 >
15 > readonly observable: IObservable<Set<T>> = this._obs;
16 > set.ts
17 > get size(): number {
18 return this._data.size;
19 }
20 > set.ts
21 > has(value: T): boolean {
22 return this._data.has(value);
23 }
24 > set.ts
25 > add(value: T, tx?: ITransaction): this {
26 const hadValue = this._data.has(value);
27 if (!hadValue) {
31 return this;
32 }
33 > set.ts
34 > delete(value: T, tx?: ITransaction): boolean {
35 const result = this._data.delete(value);
36 if (result) {
39 return result;
40 }
41 > set.ts
42 > clear(tx?: ITransaction): void {
43 if (this._data.size > 0) {
44 this._data.clear();
46 }
47 }
48 > set.ts
49 > forEach(callbackfn: (value: T, value2: T, set: Set<T>) => void, thisArg?: unknown): void {
50 this._data.forEach((value, value2, _set) => {
51 callbackfn.call(thisArg, value, value2, this);
52 });
53 }
54 > set.ts
55 > *entries(): SetIterator<[T, T]> {
56 for (const value of this._data) {
57 yield [value, value];
58 }
59 }
60 > set.ts
61 > *keys(): SetIterator<T> {
62 yield* this._data.keys();
63 }
64 > set.ts
65 > *values(): SetIterator<T> {
66 yield* this._data.values();
67 }
68 > set.ts
69 > [Symbol.iterator](): SetIterator<T> {
70 return this.values();
71 }
72 > set.ts
73 > get [Symbol.toStringTag](): string {
74 return 'ObservableSet';
75 }
76 > } set.ts
src/vs/editor/common/core/editOperation.ts 38 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- editOperation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Position } from './position.js';
7 > import { IRange, Range } from './range.js';
8 >
9 > /**
10 > * A single edit operation, that acts as a simple replace.
11 > * i.e. Replace text at `range` with `text` in model.
12 > */
13 > export interface ISingleEditOperation {
14 > /**
15 > * The range to replace. This can be empty to emulate a simple insert.
16 > */
17 > range: IRange;
18 > /**
19 > * The text to replace with. This can be null to emulate a simple delete.
20 > */
21 > text: string | null;
22 > /**
23 > * This indicates that this operation has "insert" semantics.
24 > * i.e. forceMoveMarkers = true => if `range` is collapsed, all markers at the position will be moved.
25 > */
26 > forceMoveMarkers?: boolean;
27 > }
28 >
29 > export class EditOperation {
30 >
31 > public static insert(position: Position, text: string): ISingleEditOperation {
32 return {
33 range: new Range(position.lineNumber, position.column, position.lineNumber, position.column),
36 };
37 }
39 > public static delete(range: Range): ISingleEditOperation {
40 return {
41 range: range,
43 };
44 }
46 > public static replace(range: Range, text: string | null): ISingleEditOperation {
47 return {
48 range: range,
50 };
51 }
53 > public static replaceMove(range: Range, text: string | null): ISingleEditOperation {
54 return {
55 range: range,
src/vs/editor/common/services/model.ts 38 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- model.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../base/common/event.js';
7 > import { URI } from '../../../base/common/uri.js';
8 > import { ITextBufferFactory, ITextModel, ITextModelCreationOptions } from '../model.js';
9 > import { ILanguageSelection } from '../languages/language.js';
10 > import { createDecorator } from '../../../platform/instantiation/common/instantiation.js';
11 > import { DocumentSemanticTokensProvider, DocumentRangeSemanticTokensProvider } from '../languages.js';
12 > import { TextModelEditSource } from '../textModelEditSource.js';
13 >
14 > export const IModelService = createDecorator<IModelService>('modelService');
15 >
16 > export type DocumentTokensProvider = DocumentSemanticTokensProvider | DocumentRangeSemanticTokensProvider;
17 >
18 > export interface IModelService {
19 > readonly _serviceBrand: undefined;
20 >
21 > createModel(value: string | ITextBufferFactory, languageSelection: ILanguageSelection | null, resource?: URI, isForSimpleWidget?: boolean): ITextModel;
22 >
23 > updateModel(model: ITextModel, value: string | ITextBufferFactory, reason?: TextModelEditSource): void;
24 >
25 > destroyModel(resource: URI): void;
26 >
27 > getModels(): ITextModel[];
28 >
29 > getCreationOptions(language: string, resource: URI, isForSimpleWidget: boolean): ITextModelCreationOptions;
30 >
31 > getModel(resource: URI): ITextModel | null;
32 >
33 > readonly onModelAdded: Event<ITextModel>;
34 >
35 > readonly onModelRemoved: Event<ITextModel>;
36 >
37 > readonly onModelLanguageChanged: Event<{ readonly model: ITextModel; readonly oldLanguageId: string }>;
38 > }
src/vs/platform/extensionManagement/common/implicitActivationEvents.ts 38 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- implicitActivationEvents.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStringDictionary } from '../../../base/common/collections.js';
7 > import { onUnexpectedError } from '../../../base/common/errors.js';
8 > import { ExtensionIdentifier, IExtensionDescription } from '../../extensions/common/extensions.js';
9 >
10 > export interface IActivationEventsGenerator<T> {
11 > (contributions: readonly T[]): Iterable<string>;
12 > }
13 >
14 > export class ImplicitActivationEventsImpl {
15 >
16 > private readonly _generators = new Map<string, IActivationEventsGenerator<unknown>>();
17 > private readonly _cache = new WeakMap<IExtensionDescription, string[]>();
18 >
19 > public register<T>(extensionPointName: string, generator: IActivationEventsGenerator<T>): void {
20 > this._generators.set(extensionPointName, generator as IActivationEventsGenerator<unknown>); implicitActivationEvents.ts
21 > }
23 > /**
24 > * This can run correctly only on the renderer process because that is the only place
25 > * where all extension points and all implicit activation events generators are known.
26 > */
27 > public readActivationEvents(extensionDescription: IExtensionDescription): string[] {
28 if (!this._cache.has(extensionDescription)) {
29 this._cache.set(extensionDescription, this._readActivationEvents(extensionDescription));
31 return this._cache.get(extensionDescription)!;
32 }
34 > /**
35 > * This can run correctly only on the renderer process because that is the only place
36 > * where all extension points and all implicit activation events generators are known.
37 > */
38 > public createActivationEventsMap(extensionDescriptions: IExtensionDescription[]): { [extensionId: string]: string[] } {
39 const result: { [extensionId: string]: string[] } = Object.create(null);
40 for (const extensionDescription of extensionDescriptions) {
46 return result;
47 }
49 > private _readActivationEvents(desc: IExtensionDescription): string[] {
50 if (typeof desc.main === 'undefined' && typeof desc.browser === 'undefined') {
51 return [];
83 return activationEvents;
84 }
86 >
87 > export const ImplicitActivationEvents: ImplicitActivationEventsImpl = new ImplicitActivationEventsImpl();
src/vs/base/common/observableInternal/observables/observableSignal.ts 37 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableSignal.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservableWithChange, ITransaction } from '../base.js';
7 > import { transaction } from '../transaction.js';
8 > import { DebugNameData } from '../debugName.js';
9 > import { BaseObservable } from './baseObservable.js';
10 > import { DebugLocation } from '../debugLocation.js';
11 >
12 > /**
13 > * Creates a signal that can be triggered to invalidate observers.
14 > * Signals don't have a value - when they are triggered they indicate a change.
15 > * However, signals can carry a delta that is passed to observers.
16 > */
17 > export function observableSignal<TDelta = void>(debugName: string): IObservableSignal<TDelta>;
18 > export function observableSignal<TDelta = void>(owner: object): IObservableSignal<TDelta>;
19 > export function observableSignal<TDelta = void>(debugNameOrOwner: string | object, debugLocation = DebugLocation.ofCaller()): IObservableSignal<TDelta> {
20 if (typeof debugNameOrOwner === 'string') {
21 return new ObservableSignal<TDelta>(debugNameOrOwner, undefined, debugLocation);
24 }
25 }
27 > export interface IObservableSignal<TChange> extends IObservableWithChange<void, TChange> {
28 > trigger(tx: ITransaction | undefined, change: TChange): void;
29 > }
30 >
31 > class ObservableSignal<TChange> extends BaseObservable<void, TChange> implements IObservableSignal<TChange> {
32 > public get debugName() {
33 > return new DebugNameData(this._owner, this._debugName, undefined).getDebugName(this) ?? 'Observable Signal';
34 > }
35 >
36 > public override toString(): string {
37 return this.debugName;
38 }
40 > constructor(
41 private readonly _debugName: string | undefined,
42 private readonly _owner: object | undefined,
45 super(debugLocation);
46 }
48 > public trigger(tx: ITransaction | undefined, change: TChange): void {
49 if (!tx) {
50 transaction(tx => {
src/vs/workbench/services/views/common/viewsService.ts 37 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- viewsService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../base/common/event.js';
7 > import { createDecorator } from '../../../../platform/instantiation/common/instantiation.js';
8 > import { IProgressIndicator } from '../../../../platform/progress/common/progress.js';
9 > import { IPaneComposite } from '../../../common/panecomposite.js';
10 > import { IView, IViewDescriptor, IViewPaneContainer, ViewContainer, ViewContainerLocation } from '../../../common/views.js';
11 >
12 > export const IViewsService = createDecorator<IViewsService>('viewsService');
13 > export interface IViewsService {
14 >
15 > readonly _serviceBrand: undefined;
16 >
17 > // View Container APIs
18 > readonly onDidChangeViewContainerVisibility: Event<{ id: string; visible: boolean; location: ViewContainerLocation }>;
19 > isViewContainerVisible(id: string): boolean;
20 > isViewContainerActive(id: string): boolean;
21 > openViewContainer(id: string, focus?: boolean): Promise<IPaneComposite | null>;
22 > closeViewContainer(id: string): void;
23 > getVisibleViewContainer(location: ViewContainerLocation): ViewContainer | null;
24 > getActiveViewPaneContainerWithId(viewContainerId: string): IViewPaneContainer | null;
25 > getFocusedView(): IViewDescriptor | null;
26 > getFocusedViewName(): string;
27 >
28 > // View APIs
29 > readonly onDidChangeViewVisibility: Event<{ id: string; visible: boolean }>;
30 > readonly onDidChangeFocusedView: Event<void>;
31 > isViewVisible(id: string): boolean;
32 > openView<T extends IView>(id: string, focus?: boolean): Promise<T | null>;
33 > closeView(id: string): void;
34 > getActiveViewWithId<T extends IView>(id: string): T | null;
35 > getViewWithId<T extends IView>(id: string): T | null;
36 > getViewProgressIndicator(id: string): IProgressIndicator | undefined;
37 > }
src/vs/base/common/observableInternal/observables/constObservable.ts 36 covered LOC · 14 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- constObservable.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IObserver, IObservableWithChange } from '../base.js';
7 > import { ConvenientObservable } from './baseObservable.js';
8 >
9 > /**
10 > * Represents an efficient observable whose value never changes.
11 > */
12 >
13 > export function constObservable<T>(value: T): IObservable<T> {
14 > return new ConstObservable(value); constObservable.ts
15 > }
16 > class ConstObservable<T> extends ConvenientObservable<T, void> { constObservable.ts
17 > constructor(private readonly value: T) {
18 > super(); constObservable.ts
19 > }
21 > public override get debugName(): string {
22 return this.toString();
23 }
25 > public get(): T {
26 > return this.value; constObservable.ts
27 > }
28 > public addObserver(observer: IObserver): void { constObservable.ts
29 > // NO OP constObservable.ts
30 > }
31 > public removeObserver(observer: IObserver): void { constObservable.ts
32 > // NO OP constObservable.ts
33 > }
35 > override log(): IObservableWithChange<T, void> {
36 return this;
37 }
39 > override toString(): string {
40 return `Const: ${this.value}`;
41 }
src/vs/base/test/common/virtualScheduling/index.ts 36 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- index.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // Greenfield virtual scheduling primitives.
7 > //
8 > // This folder is the new home for virtual-time scheduling. It supersedes
9 > // `timeTravelScheduler.ts` and `traceableTimeApi.ts`, both of which are
10 > // retained as @deprecated re-export shims.
11 >
12 > export type { TimeApi } from './timeApi.js';
13 > export { captureGlobalTimeApi, realTimeApi } from './timeApi.js';
14 >
15 > export type { EventSource, VirtualEvent, VirtualTime } from './virtualClock.js';
16 > export { VirtualClock } from './virtualClock.js';
17 >
18 > export type { RunAsHandlerOptions } from './trace.js';
19 > export { ROOT_TRACE, Trace, TraceContext, createTraceRoot } from './trace.js';
20 >
21 > export type { Embedding } from './embedding.js';
22 > export { drainMicrotasksEmbedding, nextMacrotask, syncEmbedding } from './embedding.js';
23 >
24 > export type { RunOptions, TerminationPolicy, VirtualTimeProcessorOptions } from './processor.js';
25 > export { VirtualTimeProcessor, untilIdle, untilTime, untilToken } from './processor.js';
26 >
27 > export { pushGlobalTimeApi } from './globalTimeApi.js';
28 > export type { CreateVirtualTimeApiOptions } from './virtualTimeApi.js';
29 > export { createVirtualTimeApi } from './virtualTimeApi.js';
30 > export { createLoggingTimeApi } from './loggingTimeApi.js';
31 > export type { RecordedTimerEvent } from './recordingTimeApi.js';
32 > export { createRecordingRealTimeApi } from './recordingTimeApi.js';
33 > export type { ITraceLogEntry, ITraceLogger } from './traceLogger.js';
34 > export { createTraceLogger } from './traceLogger.js';
35 > export type { RunWithFakedTimersOptions } from './runWithFakedTimers.js';
36 > export { runWithFakedTimers } from './runWithFakedTimers.js';
src/vs/base/common/observableInternal/changeTracker.ts 35 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- changeTracker.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { BugIndicatingError } from './commonFacade/deps.js';
7 > import { IObservableWithChange, IReader } from './base.js';
8 >
9 > export interface IChangeTracker<TChangeSummary> {
10 > createChangeSummary(previousChangeSummary: TChangeSummary | undefined): TChangeSummary;
11 > handleChange(ctx: IChangeContext, change: TChangeSummary): boolean;
12 > beforeUpdate?(reader: IReader, change: TChangeSummary): void;
13 > }
14 >
15 > export interface IChangeContext {
16 > readonly changedObservable: IObservableWithChange<any, any>;
17 > readonly change: unknown;
18 >
19 > /**
20 > * Returns if the given observable caused the change.
21 > */
22 > didChange<T, TChange>(observable: IObservableWithChange<T, TChange>): this is { change: TChange };
23 > }
24 >
25 > /**
26 > * Subscribes to and records changes and the last value of the given observables.
27 > * Don't use the key "changes", as it is reserved for the changes array!
28 > */
29 > export function recordChanges<TObs extends Record<any, IObservableWithChange<any, any>>>(obs: TObs):
30 IChangeTracker<{ [TKey in keyof TObs]: ReturnType<TObs[TKey]['get']> }
31 & { changes: readonly ({ [TKey in keyof TObs]: { key: TKey; change: TObs[TKey]['TChange'] } }[keyof TObs])[] }> {
56 };
57 }
59 > /**
60 > * Subscribes to and records changes and the last value of the given observables.
61 > * Don't use the key "changes", as it is reserved for the changes array!
62 > */
63 > export function recordChangesLazy<TObs extends Record<any, IObservableWithChange<any, any>>>(getObs: () => TObs):
64 IChangeTracker<{ [TKey in keyof TObs]: ReturnType<TObs[TKey]['get']> }
65 & { changes: readonly ({ [TKey in keyof TObs]: { key: TKey; change: TObs[TKey]['TChange'] } }[keyof TObs])[] }> {
src/vs/base/test/common/timeTravelScheduler.ts 35 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- timeTravelScheduler.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * @deprecated The contents of this file have moved to
8 > * `./virtualScheduling/index.js`. This re-export is kept for backwards
9 > * compatibility and will be removed once all callers have migrated.
10 > *
11 > * Notes for migration:
12 > * - `TimeTravelScheduler` is now {@link VirtualClock} (same constructor).
13 > * - `AsyncSchedulerProcessor` is now {@link VirtualTimeProcessor}; its
14 > * constructor takes an explicit {@link Embedding}, and {@link Run.options}
15 > * use `until` (a {@link TerminationPolicy}) instead of the implicit
16 > * "drain queue" behaviour. See {@link runWithFakedTimers} for a
17 > * drop-in helper.
18 > * - `originalGlobalValues` is now {@link realTimeApi}.
19 > */
20 >
21 > export {
22 > captureGlobalTimeApi,
23 > createLoggingTimeApi,
24 > createVirtualTimeApi,
25 > pushGlobalTimeApi,
26 > realTimeApi as originalGlobalValues,
27 > runWithFakedTimers,
28 > VirtualClock as TimeTravelScheduler,
29 > } from './virtualScheduling/index.js';
30 >
31 > export type {
32 > CreateVirtualTimeApiOptions,
33 > RunWithFakedTimersOptions,
34 > TimeApi,
35 > } from './virtualScheduling/index.js';
src/vs/platform/instantiation/common/extensions.ts 35 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- extensions.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { SyncDescriptor } from './descriptors.js';
7 > import { BrandedService, ServiceIdentifier } from './instantiation.js';
8 >
9 > const _registry: [ServiceIdentifier<any>, SyncDescriptor<any>][] = [];
10 >
11 > export const enum InstantiationType {
12 > /**
13 > * Instantiate this service as soon as a consumer depends on it. _Note_ that this
14 > * is more costly as some upfront work is done that is likely not needed
15 > */
16 > Eager = 0,
17 >
18 > /**
19 > * Instantiate this service as soon as a consumer uses it. This is the _better_
20 > * way of registering a service.
21 > */
22 > Delayed = 1
23 > }
24 >
25 > export function registerSingleton<T, Services extends BrandedService[]>(id: ServiceIdentifier<T>, ctor: new (...services: Services) => T, supportsDelayedInstantiation: InstantiationType): void;
26 > export function registerSingleton<T, Services extends BrandedService[]>(id: ServiceIdentifier<T>, descriptor: SyncDescriptor<any>): void;
27 > export function registerSingleton<T, Services extends BrandedService[]>(id: ServiceIdentifier<T>, ctorOrDescriptor: { new(...services: Services): T } | SyncDescriptor<any>, supportsDelayedInstantiation?: boolean | InstantiationType): void {
28 > if (!(ctorOrDescriptor instanceof SyncDescriptor)) {
29 > ctorOrDescriptor = new SyncDescriptor<T>(ctorOrDescriptor as new (...args: unknown[]) => T, [], Boolean(supportsDelayedInstantiation));
30 > }
31 >
32 > _registry.push([id, ctorOrDescriptor]);
33 > }
34 >
35 > export function getSingletonServiceDescriptors(): [ServiceIdentifier<any>, SyncDescriptor<any>][] {
36 return _registry;
37 }
src/vs/workbench/contrib/chat/common/model/chatTransferService.ts 35 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatTransferService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { IWorkspaceContextService } from '../../../../../platform/workspace/common/workspace.js';
6 > import { IStorageService, StorageScope, StorageTarget } from '../../../../../platform/storage/common/storage.js';
7 > import { IFileService } from '../../../../../platform/files/common/files.js';
8 > import { IWorkspaceTrustManagementService } from '../../../../../platform/workspace/common/workspaceTrust.js';
9 > import { areWorkspaceFoldersEmpty } from '../../../../services/workspaces/common/workspaceUtils.js';
10 > import { createDecorator } from '../../../../../platform/instantiation/common/instantiation.js';
11 > import { URI } from '../../../../../base/common/uri.js';
12 >
13 > export const IChatTransferService = createDecorator<IChatTransferService>('chatTransferService');
14 > const transferredWorkspacesKey = 'chat.transferedWorkspaces';
15 >
16 > export interface IChatTransferService {
17 > readonly _serviceBrand: undefined;
18 >
19 > checkAndSetTransferredWorkspaceTrust(): Promise<void>;
20 > addWorkspaceToTransferred(workspace: URI): void;
21 > }
22 >
23 > export class ChatTransferService implements IChatTransferService {
24 > _serviceBrand: undefined;
25 >
26 > constructor(
27 @IWorkspaceContextService private readonly workspaceService: IWorkspaceContextService,
28 @IStorageService private readonly storageService: IStorageService,
30 @IWorkspaceTrustManagementService private readonly workspaceTrustManagementService: IWorkspaceTrustManagementService
31 ) { }
33 > private deleteWorkspaceFromTransferredList(workspace: URI): void {
34 const transferredWorkspaces = this.storageService.getObject<string[]>(transferredWorkspacesKey, StorageScope.PROFILE, []);
35 const updatedWorkspaces = transferredWorkspaces.filter(uri => uri !== workspace.toString());
36 this.storageService.store(transferredWorkspacesKey, updatedWorkspaces, StorageScope.PROFILE, StorageTarget.MACHINE);
37 }
39 > addWorkspaceToTransferred(workspace: URI): void {
40 const transferredWorkspaces = this.storageService.getObject<string[]>(transferredWorkspacesKey, StorageScope.PROFILE, []);
41 transferredWorkspaces.push(workspace.toString());
42 this.storageService.store(transferredWorkspacesKey, transferredWorkspaces, StorageScope.PROFILE, StorageTarget.MACHINE);
43 }
45 > async checkAndSetTransferredWorkspaceTrust(): Promise<void> {
46 const workspace = this.workspaceService.getWorkspace();
47 const currentWorkspaceUri = workspace.folders[0]?.uri;
54 }
55 }
57 > private isChatTransferredWorkspace(workspace: URI, storageService: IStorageService): boolean {
58 if (!workspace) {
59 return false;
62 return chatWorkspaceTransfer.some(item => item.toString() === workspace.toString());
63 }
src/vs/base/common/marshallingIds.ts 33 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- marshallingIds.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export const enum MarshalledId {
7 > Uri = 1,
8 > Regexp,
9 > ScmResource,
10 > ScmResourceGroup,
11 > ScmProvider,
12 > CommentController,
13 > CommentThread,
14 > CommentThreadInstance,
15 > CommentThreadReply,
16 > CommentNode,
17 > CommentThreadNode,
18 > TimelineActionContext,
19 > NotebookCellActionContext,
20 > NotebookActionContext,
21 > TerminalContext,
22 > TestItemContext,
23 > Date,
24 > TestMessageMenuArgs,
25 > ChatViewContext,
26 > LanguageModelToolResult,
27 > LanguageModelTextPart,
28 > LanguageModelThinkingPart,
29 > LanguageModelPromptTsxPart,
30 > LanguageModelDataPart,
31 > AgentSessionContext,
32 > ChatResponsePullRequestPart,
33 > }
src/vs/base/test/common/virtualScheduling/recordingTimeApi.ts 33 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- recordingTimeApi.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { realTimeApi, TimeApi } from './timeApi.js';
7 > import { Trace, TraceContext } from './trace.js';
8 > import { EventSource } from './virtualClock.js';
9 >
10 > /**
11 > * One entry in a real-time trace recording. Structurally compatible with
12 > * `VirtualEvent` (and `ScheduledTaskLike` consumed by
13 > * `buildHistoryFromTasks`), so the same swimlane renderer can plot both.
14 > */
15 > export interface RecordedTimerEvent {
16 > readonly time: number;
17 > readonly source: EventSource;
18 > readonly trace?: Trace;
19 > }
20 >
21 > /**
22 > * Wrap the real host time API so every `setTimeout` / `setInterval` /
23 > * `requestAnimationFrame` call is tagged with a child {@link Trace} and
24 > * pushes a {@link RecordedTimerEvent} into `history` when the handler
25 > * actually runs.
26 > *
27 > * Handlers are invoked through {@link TraceContext.runAsHandler} so causal
28 > * chains carry across awaits inside a handler. Note: because each handler's
29 > * deferred trace-reset fires as its own real macrotask, attribution can
30 > * drift slightly when many handlers fire in quick succession — accurate
31 > * enough for diagnostics, not for assertions.
32 > */
33 > export function createRecordingRealTimeApi(history: RecordedTimerEvent[]): TimeApi {
34 const realSetTimeout = realTimeApi.setTimeout;
35
src/vs/base/common/severity.ts 32 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- severity.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as strings from './strings.js';
7 >
8 > enum Severity {
9 > Ignore = 0,
10 > Info = 1,
11 > Warning = 2,
12 > Error = 3
13 > }
14 >
15 > namespace Severity {
16 >
17 > const _error = 'error';
18 > const _warning = 'warning';
19 > const _warn = 'warn';
20 > const _info = 'info';
21 > const _ignore = 'ignore';
22 >
23 > /**
24 > * Parses 'error', 'warning', 'warn', 'info' in call casings
25 > * and falls back to ignore.
26 > */
27 > export function fromValue(value: string): Severity {
28 if (!value) {
29 return Severity.Ignore;
43 return Severity.Ignore;
44 }
46 > export function toString(severity: Severity): string {
47 switch (severity) {
48 case Severity.Error: return _error;
52 }
53 }
54 > } severity.ts
55 >
56 > export default Severity;
src/vs/platform/keybinding/common/resolvedKeybindingItem.ts 32 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- resolvedKeybindingItem.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { CharCode } from '../../../base/common/charCode.js';
7 > import { ResolvedKeybinding } from '../../../base/common/keybindings.js';
8 > import { ContextKeyExpression } from '../../contextkey/common/contextkey.js';
9 >
10 > export class ResolvedKeybindingItem {
11 > _resolvedKeybindingItemBrand: void = undefined;
12 >
13 > public readonly resolvedKeybinding: ResolvedKeybinding | undefined;
14 > public readonly chords: string[];
15 > public readonly bubble: boolean;
16 > public readonly command: string | null;
17 > public readonly commandArgs: any;
18 > public readonly when: ContextKeyExpression | undefined;
19 > public readonly isDefault: boolean;
20 > public readonly extensionId: string | null;
21 > public readonly isBuiltinExtension: boolean;
22 > /**
23 > * Whether this keybinding was declared as a system-wide (OS global) shortcut via
24 > * `keybindings.json`. Only ever `true` for user keybindings; defaults/extension keybindings
25 > * are always `false`.
26 > */
27 > public readonly systemWide: boolean;
28 >
29 > constructor(resolvedKeybinding: ResolvedKeybinding | undefined, command: string | null, commandArgs: any, when: ContextKeyExpression | undefined, isDefault: boolean, extensionId: string | null, isBuiltinExtension: boolean, systemWide: boolean = false) {
30 this.resolvedKeybinding = resolvedKeybinding;
31 this.chords = resolvedKeybinding ? toEmptyArrayIfContainsNull(resolvedKeybinding.getDispatchChords()) : [];
43 this.systemWide = systemWide;
44 }
46 >
47 > export function toEmptyArrayIfContainsNull<T>(arr: (T | null)[]): T[] {
48 const result: T[] = [];
49 for (let i = 0, len = arr.length; i < len; i++) {
src/vs/workbench/contrib/mcp/test/common/testMcpService.ts 32 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- testMcpService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { observableValue } from '../../../../../base/common/observable.js';
7 > import { ContributionEnablementState, IEnablementModel } from '../../../chat/common/enablement.js';
8 > import { IAutostartResult, IMcpServer, IMcpService, LazyCollectionState } from '../../common/mcpTypes.js';
9 >
10 > export class TestEnablementModel implements IEnablementModel {
11 > readEnabled(_key: string): ContributionEnablementState {
12 return ContributionEnablementState.EnabledProfile;
13 }
14 > remove(_key: string): void { } testMcpService.ts
15 > setEnabled(_key: string, _state: ContributionEnablementState): void { }
16 > }
17 >
18 > export class TestMcpService implements IMcpService {
19 > declare readonly _serviceBrand: undefined; testMcpService.ts
20 > public servers = observableValue<readonly IMcpServer[]>(this, []);
21 > public readonly enablementModel: IEnablementModel = new TestEnablementModel();
22 > resetCaches(): void {
23 >
24 > }
25 > resetTrust(): void {
26
27 }
29 > cancelAutostart(): void {
30
31 }
33 > autostart() {
34 return observableValue<IAutostartResult>(this, { working: false, starting: [], serversRequiringInteraction: [] });
35 }
37 > public lazyCollectionState = observableValue(this, { state: LazyCollectionState.AllKnown, collections: [] });
39 > activateCollections(): Promise<void> {
40 return Promise.resolve();
41 }
src/vs/platform/instantiation/common/serviceCollection.ts 30 covered LOC · 10 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- serviceCollection.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ServiceIdentifier } from './instantiation.js';
7 > import { SyncDescriptor } from './descriptors.js';
8 >
9 > export class ServiceCollection {
10 >
11 > private _entries = new Map<ServiceIdentifier<any>, any>();
12 >
13 > constructor(...entries: [ServiceIdentifier<any>, any][]) {
14 > for (const [id, service] of entries) { serviceCollection.ts
15 > this.set(id, service); serviceCollection.ts
16 > }
19 > set<T>(id: ServiceIdentifier<T>, instanceOrDescriptor: T | SyncDescriptor<T>): T | SyncDescriptor<T> {
20 > const result = this._entries.get(id); serviceCollection.ts
21 > this._entries.set(id, instanceOrDescriptor);
22 > return result;
23 > }
25 > has(id: ServiceIdentifier<any>): boolean {
26 return this._entries.has(id);
27 }
29 > get<T>(id: ServiceIdentifier<T>): T | SyncDescriptor<T> {
30 > return this._entries.get(id); serviceCollection.ts
31 > }
src/vs/base/common/errorMessage.ts 29 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- errorMessage.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import * as arrays from './arrays.js';
7 > import * as types from './types.js';
8 > import * as nls from '../../nls.js';
9 > import { IAction } from './actions.js';
10 >
11 function exceptionToErrorMessage(exception: any, verbose: boolean): string {
12 if (verbose && (exception.stack || exception.stacktrace)) {
16 return detectSystemErrorMessage(exception);
17 }
19 function stackToString(stack: string[] | string | undefined): string | undefined {
20 if (Array.isArray(stack)) {
24 return stack;
25 }
27 function detectSystemErrorMessage(exception: any): string {
28
39 return exception.message || nls.localize('error.defaultMessage', "An unknown error occurred. Please consult the log for more details.");
40 }
42 > /**
43 > * Tries to generate a human readable error message out of the error. If the verbose parameter
44 > * is set to true, the error message will include stacktrace details if provided.
45 > *
46 > * @returns A string containing the error message.
47 > */
48 > export function toErrorMessage(error: any = null, verbose: boolean = false): string {
49 if (!error) {
50 return nls.localize('error.defaultMessage', "An unknown error occurred. Please consult the log for more details.");
88 return nls.localize('error.defaultMessage', "An unknown error occurred. Please consult the log for more details.");
89 }
91 >
92 > export interface IErrorWithActions extends Error {
93 > actions: IAction[];
94 > }
95 >
96 > export function isErrorWithActions(obj: unknown): obj is IErrorWithActions {
97 const candidate = obj as IErrorWithActions | undefined;
98
99 return candidate instanceof Error && Array.isArray(candidate.actions);
100 }
102 > export function createErrorWithActions(messageOrError: string | Error, actions: IAction[]): IErrorWithActions {
103 let error: IErrorWithActions;
104 if (typeof messageOrError === 'string') {
src/vs/workbench/contrib/chat/common/model/chatProgressTypes/chatPlanReviewData.ts 29 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatPlanReviewData.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { DeferredPromise } from '../../../../../../base/common/async.js';
7 > import { UriComponents } from '../../../../../../base/common/uri.js';
8 > import { IChatPlanApprovalAction, IChatPlanReview, IChatPlanReviewResult } from '../../chatService/chatService.js';
9 > import { ToolDataSource } from '../../tools/languageModelToolsService.js';
10 >
11 > /**
12 > * Runtime representation of a plan review widget with a {@link DeferredPromise}
13 > * that is resolved when the user approves or rejects the plan. {@link toJSON}
14 > * strips the completion so only serialisable data is persisted.
15 > */
16 > export class ChatPlanReviewData implements IChatPlanReview {
17 > public readonly kind = 'planReview' as const;
18 > public readonly completion = new DeferredPromise<IChatPlanReviewResult | undefined>();
19 >
20 > public draftFeedback: string | undefined;
21 > public draftCollapsed: boolean | undefined;
22 >
23 > constructor(
24 public title: string,
25 public content: string,
32 public source?: ToolDataSource,
33 ) { }
35 > /** Dismiss without a user choice (e.g. the response was cancelled). */
36 > dismiss(): void {
37 if (this.isUsed) {
38 return;
43 void this.completion.complete(undefined);
44 }
46 > toJSON(): IChatPlanReview {
47 return {
48 kind: this.kind,
src/vs/base/common/marshalling.ts 28 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- marshalling.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { VSBuffer } from './buffer.js';
7 > import { URI, UriComponents } from './uri.js';
8 > import { MarshalledId } from './marshallingIds.js';
9 >
10 > export function stringify(obj: unknown): string {
11 return JSON.stringify(obj, replacer);
12 }
14 > export function parse(text: string): any {
15 let data = JSON.parse(text);
16 data = revive(data);
17 return data;
18 }
20 > export interface MarshalledObject {
21 > $mid: MarshalledId;
22 > }
23 >
24 function replacer(key: string, value: any): any {
25 // URI is done via toJSON-member
33 return value;
34 }
36 >
37 > type Deserialize<T> = T extends UriComponents ? URI
38 > : T extends VSBuffer ? VSBuffer
39 > : T extends object
40 > ? Revived<T>
41 > : T;
42 >
43 > export type Revived<T> = { [K in keyof T]: Deserialize<T[K]> };
44 >
45 > export function revive<T = any>(obj: any, depth = 0): Revived<T> {
46 if (!obj || depth > 200) {
47 return obj;
src/vs/base/common/observableInternal/observables/observableValueOpts.ts 28 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observableValueOpts.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ISettableObservable } from '../base.js';
7 > import { DebugNameData, IDebugNameData } from '../debugName.js';
8 > import { EqualityComparer, strictEquals } from '../commonFacade/deps.js';
9 > import { ObservableValue } from './observableValue.js';
10 > import { LazyObservableValue } from './lazyObservableValue.js';
11 > import { DebugLocation } from '../debugLocation.js';
12 >
13 > export function observableValueOpts<T, TChange = void>(
14 > options: IDebugNameData & { observableValueOpts.ts
15 > equalsFn?: EqualityComparer<T>;
16 > lazy?: boolean;
17 > },
18 > initialValue: T,
19 > debugLocation = DebugLocation.ofCaller(),
20 > ): ISettableObservable<T, TChange> {
21 > if (options.lazy) {
22 return new LazyObservableValue(
23 new DebugNameData(options.owner, options.debugName, undefined),
27 );
28 }
29 > return new ObservableValue( observableValueOpts.ts
30 > new DebugNameData(options.owner, options.debugName, undefined),
31 > initialValue,
32 > options.equalsFn ?? strictEquals,
33 > debugLocation
34 > );
35 > }
src/vs/base/common/normalization.ts 27 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- normalization.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { LRUCache } from './map.js';
7 >
8 > const nfcCache = new LRUCache<string, string>(10000); // bounded to 10000 elements
9 > export function normalizeNFC(str: string): string {
10 return normalize(str, 'NFC', nfcCache);
11 }
13 > const nfdCache = new LRUCache<string, string>(10000); // bounded to 10000 elements
14 > export function normalizeNFD(str: string): string {
15 return normalize(str, 'NFD', nfdCache);
16 }
18 > const nonAsciiCharactersPattern = /[^\u0000-\u0080]/;
19 function normalize(str: string, form: string, normalizedCache: LRUCache<string, string>): string {
20 if (!str) {
39 return res;
40 }
42 > /**
43 > * Attempts to normalize the string to Unicode base format (NFD -> remove accents -> lower case).
44 > * When original string contains accent characters directly, only lower casing will be performed.
45 > * This is done so as to keep the string length the same and not affect indices.
46 > *
47 > * @see https://stackoverflow.com/questions/990904/remove-accents-diacritics-in-a-string-in-javascript/37511463#37511463
48 > */
49 > export const tryNormalizeToBase: (str: string) => string = function () {
50 > const cache = new LRUCache<string, string>(10000); // bounded to 10000 elements
51 > const accentsRegex = /[\u0300-\u036f]/g;
52 > return function (str: string): string {
53 const cached = cache.get(str);
54 if (cached) {
src/vs/workbench/contrib/chat/common/promptSyntax/hookCopilotCliCompat.ts 27 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- hookCopilotCliCompat.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { HOOKS_BY_TARGET, HookType } from './hookTypes.js';
7 > import { Target } from './promptTypes.js';
8 >
9 > const COPILOT_CLI_HOOK_TYPE_MAP: Record<string, HookType> = HOOKS_BY_TARGET[Target.GitHubCopilot];
10 >
11 > /**
12 > * Cached inverse mapping from HookType to Copilot CLI hook type name.
13 > * Lazily computed on first access.
14 > */
15 > let _hookTypeToCopilotCliName: Map<HookType, string> | undefined;
16 >
17 function getHookTypeToCopilotCliNameMap(): Map<HookType, string> {
18 if (!_hookTypeToCopilotCliName) {
24 return _hookTypeToCopilotCliName;
25 }
27 > /**
28 > * Resolves a Copilot CLI hook type name to our abstract HookType.
29 > */
30 > export function resolveCopilotCliHookType(name: string): HookType | undefined {
31 return (COPILOT_CLI_HOOK_TYPE_MAP as Record<string, HookType>)[name];
32 }
34 > /**
35 > * Gets the Copilot CLI hook type name for a given abstract HookType.
36 > * Returns undefined if the hook type is not supported in Copilot CLI.
37 > */
38 > export function getCopilotCliHookTypeName(hookType: HookType): string | undefined {
39 return getHookTypeToCopilotCliNameMap().get(hookType);
40 }
src/vs/base/common/codiconsUtil.ts 26 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- codiconsUtil.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { ThemeIcon } from './themables.js';
6 > import { isString } from './types.js';
7 >
8 >
9 > const _codiconFontCharacters: { [id: string]: number } = Object.create(null);
10 >
11 > export function register(id: string, fontCharacter: number | string): ThemeIcon {
12 > if (isString(fontCharacter)) {
13 > const val = _codiconFontCharacters[fontCharacter];
14 > if (val === undefined) {
15 throw new Error(`${id} references an unknown codicon: ${fontCharacter}`);
16 }
17 > fontCharacter = val; codiconsUtil.ts
18 > }
19 > _codiconFontCharacters[id] = fontCharacter;
20 > return { id };
21 > }
22 >
23 > /**
24 > * Only to be used by the iconRegistry.
25 > */
26 > export function getCodiconFontCharacters(): { [id: string]: number } {
27 > return _codiconFontCharacters; codiconsUtil.ts
28 > }
src/vs/base/common/observableInternal/experimental/utils.ts 26 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- utils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable, IReader } from '../base.js';
7 > import { BugIndicatingError, DisposableStore } from '../commonFacade/deps.js';
8 > import { DebugOwner, getDebugName, DebugNameData } from '../debugName.js';
9 > import { observableFromEvent } from '../observables/observableFromEvent.js';
10 > import { autorunOpts } from '../reactions/autorun.js';
11 > import { derivedObservableWithCache } from '../utils/utils.js';
12 >
13 > /**
14 > * Creates an observable that has the latest changed value of the given observables.
15 > * Initially (and when not observed), it has the value of the last observable.
16 > * When observed and any of the observables change, it has the value of the last changed observable.
17 > * If multiple observables change in the same transaction, the last observable wins.
18 > */
19 > export function latestChangedValue<T extends IObservable<any>[]>(owner: DebugOwner, observables: T): IObservable<ReturnType<T[number]['get']>> {
20 if (observables.length === 0) {
21 throw new BugIndicatingError();
50 return result;
51 }
52 > utils.ts
53 > /**
54 > * Works like a derived.
55 > * However, if the value is not undefined, it is cached and will not be recomputed anymore.
56 > * In that case, the derived will unsubscribe from its dependencies.
57 > */
58 > export function derivedConstOnceDefined<T>(owner: DebugOwner, fn: (reader: IReader) => T): IObservable<T | undefined> {
59 return derivedObservableWithCache<T | undefined>(owner, (reader, lastValue) => lastValue ?? fn(reader));
60 }
src/vs/platform/agentPlugins/common/agentPluginParser.ts 26 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- agentPluginParser.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { parse as parseJSONC } from '../../../base/common/json.js';
7 > import { joinPath } from '../../../base/common/resources.js';
8 > import { URI } from '../../../base/common/uri.js';
9 > import { IFileService } from '../../files/common/files.js';
10 >
11 > export const AGENT_PLUGIN_SCHEMA = 'https://agent-plugins.org/schemas/1.0.0/plugin.schema.json';
12 > export const AGENT_PLUGIN_MCP_SCHEMA = 'https://agent-plugins.org/schemas/1.0.0/mcp.schema.json';
13 >
14 > const agentPluginSchemaPrefix = 'https://agent-plugins.org/schemas/';
15 >
16 > export interface IAgentPluginManifest {
17 > readonly $schema: string;
18 > readonly name?: string;
19 > readonly version?: string;
20 > readonly description?: string;
21 > }
22 >
23 export async function readAgentPluginManifest(pluginUri: URI, fileService: IFileService): Promise<IAgentPluginManifest | undefined> {
24 const manifestUri = joinPath(pluginUri, 'plugin.json');
49 };
50 }
52 function isAgentPluginSchema(value: unknown): value is string {
53 return typeof value === 'string'
55 && value.endsWith('/plugin.schema.json');
56 }
58 function asString(value: unknown): string | undefined {
59 return typeof value === 'string' ? value : undefined;
60 }
62 function asNonEmptyString(value: unknown): string | undefined {
63 return typeof value === 'string' && value.trim().length > 0 ? value.trim() : undefined;
64 }
66 function isRecord(value: unknown): value is Record<string, unknown> {
67 return typeof value === 'object' && value !== null && !Array.isArray(value);
src/vs/platform/workspace/test/common/testWorkspace.ts 26 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- testWorkspace.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isLinux, isWindows } from '../../../../base/common/platform.js';
7 > import { URI } from '../../../../base/common/uri.js';
8 > import { toWorkspaceFolder, Workspace as BaseWorkspace, WorkspaceFolder } from '../../common/workspace.js';
9 >
10 > export class Workspace extends BaseWorkspace {
11 > constructor(
12 > id: string,
13 > folders: WorkspaceFolder[] = [],
14 > configuration: URI | null = null,
15 > ignorePathCasing: (key: URI) => boolean = () => !isLinux
16 > ) {
17 > super(id, folders, false, configuration, ignorePathCasing);
18 > }
19 > }
20 >
21 > const wsUri = URI.file(isWindows ? 'C:\\testWorkspace' : '/testWorkspace');
22 > export const TestWorkspace = testWorkspace(wsUri);
23 >
24 > export function testWorkspace(...resource: URI[]): Workspace {
25 > return new Workspace('test-workspace', resource.map(toWorkspaceFolder));
26 > }
src/vs/base/common/functional.ts 25 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- functional.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * Given a function, returns a function that is only calling that function once.
8 > */
9 > export function createSingleCallFunction<T extends Function>(this: unknown, fn: T, fnDidRunCallback?: () => void): T {
10 > const _this = this; functional.ts
11 > let didCall = false;
12 > let result: unknown;
13 >
14 > return function () {
15 > if (didCall) { functional.ts
16 return result;
17 }
19 > didCall = true;
20 > if (fnDidRunCallback) {
21 try {
22 result = fn.apply(_this, arguments);
24 fnDidRunCallback();
25 }
26 > } else { functional.ts
27 > result = fn.apply(_this, arguments);
28 > }
29 >
30 > return result;
31 > } as unknown as T;
32 > } functional.ts
src/vs/base/common/stopwatch.ts 25 covered LOC · 6 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- stopwatch.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > declare const globalThis: { performance: { now(): number } };
7 > const performanceNow = globalThis.performance.now.bind(globalThis.performance);
8 >
9 > export class StopWatch {
10 >
11 > private _startTime: number;
12 > private _stopTime: number;
13 >
14 > private readonly _now: () => number;
15 >
16 > public static create(highResolution?: boolean): StopWatch {
17 return new StopWatch(highResolution);
18 }
20 > constructor(highResolution?: boolean) {
21 this._now = highResolution === false ? Date.now : performanceNow;
22 this._startTime = this._now();
23 this._stopTime = -1;
24 }
26 > public stop(): void {
27 this._stopTime = this._now();
28 }
30 > public reset(): void {
31 this._startTime = this._now();
32 this._stopTime = -1;
33 }
35 > public elapsed(): number {
36 if (this._stopTime !== -1) {
37 return this._stopTime - this._startTime;
39 return this._now() - this._startTime;
40 }
41 > } stopwatch.ts
src/vs/base/common/uuid.ts 25 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- uuid.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 >
7 > const _UUIDPattern = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
8 >
9 > export function isUUID(value: string): boolean {
10 return _UUIDPattern.test(value);
11 }
12 > uuid.ts
13 > export const generateUuid = (function (): () => string {
14 >
15 > // use `randomUUID` if possible
16 > if (typeof crypto.randomUUID === 'function') {
17 > // see https://developer.mozilla.org/en-US/docs/Web/API/Window/crypto
18 > // > Although crypto is available on all windows, the returned Crypto object only has one
19 > // > usable feature in insecure contexts: the getRandomValues() method.
20 > // > In general, you should use this API only in secure contexts.
21 >
22 > return crypto.randomUUID.bind(crypto);
23 > }
24
25 // prep-work
63 return result;
64 };
65 > })(); uuid.ts
66 >
67 > /** Namespace should be 3 letters, e.g. `abc-<uuid>`. */
68 > export function prefixedUuid(namespace: string): string {
69 return `${namespace}-${generateUuid()}`;
70 }
src/vs/base/test/common/mock.ts 25 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mock.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { SinonStub, stub } from 'sinon';
7 > import { DeepPartial } from '../../common/types.js';
8 >
9 > export interface Ctor<T> {
10 > new(): T;
11 > }
12 >
13 > export function mock<T>(): Ctor<T> {
14 // eslint-disable-next-line local/code-no-any-casts
15 return function () { } as any;
16 }
17 > mock.ts
18 > export type MockObject<T, ExceptProps = never> = { [K in keyof T]: K extends ExceptProps ? T[K] : SinonStub };
19 >
20 > // Creates an object object that returns sinon mocks for every property. Optionally
21 > // takes base properties.
22 > export const mockObject = <T extends object>() => <TP extends Partial<T> = {}>(properties?: TP): MockObject<T, keyof TP> => {
23 // eslint-disable-next-line local/code-no-any-casts
24 return new Proxy({ ...properties } as any, {
36 });
37 };
38 > mock.ts
39 > /**
40 > * Shortcut for type-safe partials in mocks. A shortcut for `obj as Partial<T> as T`.
41 > */
42 > export function upcastPartial<T>(partial: Partial<T>): T {
43 return partial as T;
44 }
45 > export function upcastDeepPartial<T>(partial: DeepPartial<T>): T { mock.ts
46 return partial as T;
47 }
src/vs/base/common/observableInternal/logging/debugger/utils.ts 24 covered LOC · 8 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- utils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IDisposable } from '../../../lifecycle.js';
7 >
8 > export class Debouncer implements IDisposable {
9 private _timeout: Timeout | undefined = undefined;
10 > utils.ts
11 > public debounce(fn: () => void, timeoutMs: number): void {
12 if (this._timeout !== undefined) {
13 clearTimeout(this._timeout);
18 }, timeoutMs);
19 }
20 > utils.ts
21 > dispose(): void {
22 if (this._timeout !== undefined) {
23 clearTimeout(this._timeout);
24 }
25 }
26 > } utils.ts
27 >
28 > export class Throttler implements IDisposable {
29 private _timeout: Timeout | undefined = undefined;
30 > utils.ts
31 > public throttle(fn: () => void, timeoutMs: number): void {
32 if (this._timeout === undefined) {
33 this._timeout = setTimeout(() => {
37 }
38 }
39 > utils.ts
40 > dispose(): void {
41 if (this._timeout !== undefined) {
42 clearTimeout(this._timeout);
43 }
44 }
45 > } utils.ts
46 >
47 > export function deepAssign<T>(target: T, source: T): void {
48 for (const key in source) {
49 if (!!target[key] && typeof target[key] === 'object' && !!source[key] && typeof source[key] === 'object') {
54 }
55 }
56 > utils.ts
57 > export function deepAssignDeleteNulls<T>(target: T, source: T): void {
58 for (const key in source) {
59 if (source[key] === null) {
src/vs/base/common/observableInternal/utils/utilsCancellation.ts 24 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- utilsCancellation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IReader, IObservable } from '../base.js';
7 > import { DebugOwner, DebugNameData } from '../debugName.js';
8 > import { CancellationError, CancellationToken, CancellationTokenSource } from '../commonFacade/cancellation.js';
9 > import { strictEquals } from '../commonFacade/deps.js';
10 > import { autorun } from '../reactions/autorun.js';
11 > import { Derived } from '../observables/derivedImpl.js';
12 > import { DebugLocation } from '../debugLocation.js';
13 >
14 > /**
15 > * Resolves the promise when the observables state matches the predicate.
16 > */
17 > export function waitForState<T>(observable: IObservable<T | null | undefined>): Promise<T>;
18 > export function waitForState<T, TState extends T>(observable: IObservable<T>, predicate: (state: T) => state is TState, isError?: (state: T) => boolean | unknown | undefined, cancellationToken?: CancellationToken): Promise<TState>;
19 > export function waitForState<T>(observable: IObservable<T>, predicate: (state: T) => boolean, isError?: (state: T) => boolean | unknown | undefined, cancellationToken?: CancellationToken): Promise<T>;
20 > export function waitForState<T>(observable: IObservable<T>, predicate?: (state: T) => boolean, isError?: (state: T) => boolean | unknown | undefined, cancellationToken?: CancellationToken): Promise<T> {
21 if (!predicate) {
22 predicate = state => state !== null && state !== undefined;
69 });
70 }
72 > export function derivedWithCancellationToken<T>(computeFn: (reader: IReader, cancellationToken: CancellationToken) => T): IObservable<T>;
73 > export function derivedWithCancellationToken<T>(owner: object, computeFn: (reader: IReader, cancellationToken: CancellationToken) => T): IObservable<T>;
74 > export function derivedWithCancellationToken<T>(computeFnOrOwner: ((reader: IReader, cancellationToken: CancellationToken) => T) | object, computeFnOrUndefined?: ((reader: IReader, cancellationToken: CancellationToken) => T)): IObservable<T> {
75 let computeFn: (reader: IReader, store: CancellationToken) => T;
76 let owner: DebugOwner;
src/vs/workbench/contrib/chat/test/common/mockChatVariables.ts 24 covered LOC · 7 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- mockChatVariables.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ResourceMap } from '../../../../../base/common/map.js';
7 > import { URI } from '../../../../../base/common/uri.js';
8 > import { IChatVariablesService, IDynamicVariable } from '../../common/attachments/chatVariables.js';
9 > import { ToolAndToolSetEnablementMap } from '../../common/tools/languageModelToolsService.js';
10 >
11 > export class MockChatVariablesService implements IChatVariablesService {
12 > _serviceBrand: undefined; mockChatVariables.ts
13 >
14 > private _dynamicVariables = new ResourceMap<readonly IDynamicVariable[]>();
15 > private _selectedToolAndToolSets = new ResourceMap<ToolAndToolSetEnablementMap>();
17 > getDynamicVariables(sessionResource: URI): readonly IDynamicVariable[] {
18 return this._dynamicVariables.get(sessionResource) ?? [];
19 }
21 > getSelectedToolAndToolSets(sessionResource: URI): ToolAndToolSetEnablementMap {
22 return this._selectedToolAndToolSets.get(sessionResource) ?? ToolAndToolSetEnablementMap.fromEntries([]);
23 }
25 > setDynamicVariables(sessionResource: URI, variables: readonly IDynamicVariable[]): void {
26 this._dynamicVariables.set(sessionResource, variables);
27 }
29 > setSelectedToolAndToolSets(sessionResource: URI, tools: ToolAndToolSetEnablementMap): void {
30 this._selectedToolAndToolSets.set(sessionResource, tools);
31 }
src/vs/platform/contextkey/common/contextkeys.ts 23 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- contextkeys.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isIOS, isLinux, isMacintosh, isMobile, isWeb, isWindows } from '../../../base/common/platform.js';
7 > import { localize } from '../../../nls.js';
8 > import { RawContextKey } from './contextkey.js';
9 >
10 > export const IsMacContext = new RawContextKey<boolean>('isMac', isMacintosh, localize('isMac', "Whether the operating system is macOS"));
11 > export const IsLinuxContext = new RawContextKey<boolean>('isLinux', isLinux, localize('isLinux', "Whether the operating system is Linux"));
12 > export const IsWindowsContext = new RawContextKey<boolean>('isWindows', isWindows, localize('isWindows', "Whether the operating system is Windows"));
13 >
14 > export const IsWebContext = new RawContextKey<boolean>('isWeb', isWeb, localize('isWeb', "Whether the platform is a web browser"));
15 > export const IsMacNativeContext = new RawContextKey<boolean>('isMacNative', isMacintosh && !isWeb, localize('isMacNative', "Whether the operating system is macOS on a non-browser platform"));
16 > export const IsIOSContext = new RawContextKey<boolean>('isIOS', isIOS, localize('isIOS', "Whether the operating system is iOS"));
17 > export const IsMobileContext = new RawContextKey<boolean>('isMobile', isMobile, localize('isMobile', "Whether the platform is a mobile web browser"));
18 >
19 > export const IsDevelopmentContext = new RawContextKey<boolean>('isDevelopment', false, true);
20 > export const ProductQualityContext = new RawContextKey<string>('productQualityType', '', localize('productQualityType', "Quality type of VS Code"));
21 >
22 > export const InputFocusedContextKey = 'inputFocus';
23 > export const InputFocusedContext = new RawContextKey<boolean>(InputFocusedContextKey, false, localize('inputFocus', "Whether keyboard focus is inside an input box"));
src/vs/platform/instantiation/common/descriptors.ts 21 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- descriptors.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export class SyncDescriptor<T> {
7 >
8 > readonly ctor: any;
9 > readonly staticArguments: unknown[];
10 > readonly supportsDelayedInstantiation: boolean;
11 >
12 > constructor(ctor: new (...args: any[]) => T, staticArguments: unknown[] = [], supportsDelayedInstantiation: boolean = false) {
13 > this.ctor = ctor; descriptors.ts
14 > this.staticArguments = staticArguments;
15 > this.supportsDelayedInstantiation = supportsDelayedInstantiation;
16 > }
18 >
19 > export interface SyncDescriptor0<T> {
20 > readonly ctor: new () => T;
21 > }
src/vs/base/common/observableInternal/utils/valueWithChangeEvent.ts 19 covered LOC · 4 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- valueWithChangeEvent.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservable } from '../base.js';
7 > import { Event, IValueWithChangeEvent } from '../commonFacade/deps.js';
8 > import { DebugOwner } from '../debugName.js';
9 > import { observableFromEvent } from '../observables/observableFromEvent.js';
10 >
11 > export class ValueWithChangeEventFromObservable<T> implements IValueWithChangeEvent<T> {
12 > constructor(public readonly observable: IObservable<T>) {
13 }
15 > get onDidChange(): Event<void> {
16 return Event.fromObservableLight(this.observable);
17 }
19 > get value(): T {
20 return this.observable.get();
21 }
23 >
24 > export function observableFromValueWithChangeEvent<T>(owner: DebugOwner, value: IValueWithChangeEvent<T>): IObservable<T> {
25 if (value instanceof ValueWithChangeEventFromObservable) {
26 return value.observable;
src/vs/workbench/contrib/chat/common/chat.ts 19 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chat.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ResourceSet } from '../../../../base/common/map.js';
7 > import { chatEditingSessionIsReady } from './editing/chatEditingService.js';
8 > import { IChatModel } from './model/chatModel.js';
9 > import { isLegacyChatTerminalToolInvocationData, type IChatSessionStats, type IChatTerminalToolInvocationData, type ILegacyChatTerminalToolInvocationData } from './chatService/chatService.js';
10 > import { ChatModeKind } from './constants.js';
11 >
12 > export function checkModeOption(mode: ChatModeKind, option: boolean | ((mode: ChatModeKind) => boolean) | undefined): boolean | undefined {
13 if (option === undefined) {
14 return undefined;
19 return option;
20 }
21 > chat.ts
22 > /**
23 > * @deprecated This is the old API shape, we should support this for a while before removing it so
24 > * we don't break existing chats
25 > */
26 > export function migrateLegacyTerminalToolSpecificData(data: IChatTerminalToolInvocationData | ILegacyChatTerminalToolInvocationData): IChatTerminalToolInvocationData {
27 if (isLegacyChatTerminalToolInvocationData(data)) {
28 data = {
38 return data;
39 }
40 > chat.ts
41 export async function awaitStatsForSession(model: IChatModel): Promise<IChatSessionStats | undefined> {
42 if (!model.editingSession) {
src/vs/workbench/contrib/chat/common/chatSessionTypePreference.ts 19 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- chatSessionTypePreference.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IStorageService, StorageScope, StorageTarget } from '../../../../platform/storage/common/storage.js';
7 >
8 > export const CHAT_USER_SELECTED_SESSION_TYPE_STORAGE_KEY = 'chat.userSelectedSessionType';
9 > const CHAT_PREFERRED_COPILOT_HARNESS_STORAGE_KEY = 'chat.preferredCopilotHarness';
10 >
11 > export function getRememberedSessionType(storageService: IStorageService): string | undefined {
12 return storageService.get(CHAT_USER_SELECTED_SESSION_TYPE_STORAGE_KEY, StorageScope.PROFILE);
13 }
15 > export function storeUserSelectedSessionType(storageService: IStorageService, sessionType: string): void {
16 storageService.store(CHAT_USER_SELECTED_SESSION_TYPE_STORAGE_KEY, sessionType, StorageScope.PROFILE, StorageTarget.MACHINE);
17 }
19 > export function clearUserSelectedSessionType(storageService: IStorageService): void {
20 storageService.remove(CHAT_USER_SELECTED_SESSION_TYPE_STORAGE_KEY, StorageScope.PROFILE);
21 }
23 > export function hasPreferredCopilotHarness(storageService: IStorageService): boolean {
24 return storageService.getBoolean(CHAT_PREFERRED_COPILOT_HARNESS_STORAGE_KEY, StorageScope.PROFILE, false);
25 }
27 > export function markPreferredCopilotHarness(storageService: IStorageService): void {
28 storageService.store(CHAT_PREFERRED_COPILOT_HARNESS_STORAGE_KEY, true, StorageScope.PROFILE, StorageTarget.MACHINE);
29 }
src/vs/workbench/services/assignment/test/common/nullAssignmentService.ts 19 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- nullAssignmentService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Event } from '../../../../../base/common/event.js';
7 > import { IAssignmentFilter, IWorkbenchAssignmentService } from '../../common/assignmentService.js';
8 >
9 > export class NullWorkbenchAssignmentService implements IWorkbenchAssignmentService {
10 > _serviceBrand: undefined; nullAssignmentService.ts
11 >
12 > readonly onDidRefetchAssignments: Event<void> = Event.None;
14 > async getCurrentExperiments(): Promise<string[] | undefined> {
15 return [];
16 }
18 > async getTreatment<T extends string | number | boolean>(name: string): Promise<T | undefined> {
19 return undefined;
20 }
22 > addTelemetryAssignmentFilter(filter: IAssignmentFilter): void { }
23 > }
src/vs/platform/workspace/common/virtualWorkspace.ts 18 covered LOC · 5 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- virtualWorkspace.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { Schemas } from '../../../base/common/network.js';
7 > import { URI } from '../../../base/common/uri.js';
8 > import { IWorkspace } from './workspace.js';
9 >
10 > export function isVirtualResource(resource: URI) {
11 return resource.scheme !== Schemas.file && resource.scheme !== Schemas.vscodeRemote;
12 }
14 > export function getVirtualWorkspaceLocation(workspace: IWorkspace): { scheme: string; authority: string } | undefined {
15 if (workspace.folders.length) {
16 return workspace.folders.every(f => isVirtualResource(f.uri)) ? workspace.folders[0].uri : undefined;
20 return undefined;
21 }
23 > export function getVirtualWorkspaceScheme(workspace: IWorkspace): string | undefined {
24 return getVirtualWorkspaceLocation(workspace)?.scheme;
25 }
27 > export function getVirtualWorkspaceAuthority(workspace: IWorkspace): string | undefined {
28 return getVirtualWorkspaceLocation(workspace)?.authority;
29 }
31 > export function isVirtualWorkspace(workspace: IWorkspace): boolean {
32 return getVirtualWorkspaceLocation(workspace) !== undefined;
33 }
src/vs/base/common/observableInternal/logging/debugger/debuggerRpc.ts 17 covered LOC · 2 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- debuggerRpc.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { ChannelFactory, IChannelHandler, API, SimpleTypedRpcConnection, MakeSideAsync } from './rpc.js';
7 >
8 > export function registerDebugChannel<T extends { channelId: string } & API>(
9 channelId: T['channelId'],
10 createClient: () => T['client'],
43 });
44 }
46 > interface GlobalObj {
47 > $$debugValueEditor_debugChannels: Record<string, (host: IHost) => { handleRequest: (data: unknown) => unknown }>;
48 > }
49 >
50 > interface IHost {
51 > sendNotification: (data: unknown) => void;
52 > }
53 >
54 function createChannelFactoryFromDebugChannel(host: IHost): { channel: ChannelFactory; handler: { handleRequest: (data: unknown) => unknown } } {
55 let h: IChannelHandler | undefined;
src/vs/base/common/observableInternal/utils/runOnChange.ts 17 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- runOnChange.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IObservableWithChange } from '../base.js';
7 > import { CancellationToken, cancelOnDispose } from '../commonFacade/cancellation.js';
8 > import { DisposableStore, IDisposable } from '../commonFacade/deps.js';
9 > import { autorunWithStoreHandleChanges } from '../reactions/autorun.js';
10 >
11 > export type RemoveUndefined<T> = T extends undefined ? never : T;
12 >
13 > export function runOnChange<T, TChange>(observable: IObservableWithChange<T, TChange>, cb: (value: T, previousValue: T, deltas: RemoveUndefined<TChange>[]) => void): IDisposable {
14 let _previousValue: T | undefined;
15 let _firstRun = true;
42 });
43 }
45 > export function runOnChangeWithStore<T, TChange>(observable: IObservableWithChange<T, TChange>, cb: (value: T, previousValue: T, deltas: RemoveUndefined<TChange>[], store: DisposableStore) => void): IDisposable {
46 const store = new DisposableStore();
47 const disposable = runOnChange(observable, (value, previousValue: T, deltas) => {
56 };
57 }
59 > export function runOnChangeWithCancellationToken<T, TChange>(observable: IObservableWithChange<T, TChange>, cb: (value: T, previousValue: T, deltas: RemoveUndefined<TChange>[], token: CancellationToken) => Promise<void>): IDisposable {
60 return runOnChangeWithStore(observable, (value, previousValue, deltas, store) => {
61 cb(value, previousValue, deltas, cancelOnDispose(store));
src/vs/platform/agentHost/common/state/protocol/state.ts 17 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- state.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > export * from './common/state.js';
10 > export * from './channels-root/state.js';
11 > export * from './channels-session/state.js';
12 > export * from './channels-chat/state.js';
13 > export * from './channels-terminal/state.js';
14 > export * from './channels-changeset/state.js';
15 > export * from './channels-annotations/state.js';
16 > export * from './channels-otlp/state.js';
17 > export * from './channels-resource-watch/state.js';
src/vs/platform/product/common/productService.ts 17 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- productService.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IProductConfiguration } from '../../../base/common/product.js';
7 > import { createDecorator } from '../../instantiation/common/instantiation.js';
8 >
9 > export const IProductService = createDecorator<IProductService>('productService');
10 >
11 > export interface IProductService extends Readonly<IProductConfiguration> {
12 >
13 > readonly _serviceBrand: undefined;
14 >
15 > }
16 >
17 > export const productSchemaId = 'vscode://schemas/vscode-product';
src/vs/workbench/services/telemetry/common/workbenchTelemetryUtils.ts 16 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workbenchTelemetryUtils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { IConfigurationService } from '../../../../platform/configuration/common/configuration.js';
7 > import { IProductService } from '../../../../platform/product/common/productService.js';
8 > import { TelemetryLevel } from '../../../../platform/telemetry/common/telemetry.js';
9 > import { getTelemetryLevel } from '../../../../platform/telemetry/common/telemetryUtils.js';
10 > import { IWorkbenchEnvironmentService } from '../../environment/common/environmentService.js';
11 >
12 > /**
13 > * Determines if experiment properties will be set on telemetry events.
14 > * When true, TelemetryService should buffer events until setExperimentProperty is called.
15 > */
16 > export function experimentsEnabled(
17 configurationService: IConfigurationService,
18 productService: IProductService,
src/vs/base/test/common/virtualScheduling/loggingTimeApi.ts 15 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- loggingTimeApi.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { TimeApi } from './timeApi.js';
7 >
8 > /**
9 > * Wrap `underlying` so that every call to `setTimeout`, `setInterval`,
10 > * `setImmediate` or `requestAnimationFrame` invokes `onCall` first.
11 > *
12 > * Useful for diagnostics — e.g. logging timer registrations made outside of
13 > * virtual time, to find leaks of real-time scheduling into a fixture.
14 > */
15 > export function createLoggingTimeApi(
16 underlying: TimeApi,
17 onCall: (name: string, stack: string | undefined, handler?: () => void) => void,
src/vs/platform/agentHost/common/state/protocol/commands.ts 15 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commands.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // allow-any-unicode-comment-file
7 > // DO NOT EDIT -- auto-generated by scripts/sync-agent-host-protocol.ts
8 >
9 > export * from './common/commands.js';
10 > export * from './channels-root/commands.js';
11 > export * from './channels-session/commands.js';
12 > export * from './channels-chat/commands.js';
13 > export * from './channels-terminal/commands.js';
14 > export * from './channels-changeset/commands.js';
15 > export * from './channels-resource-watch/commands.js';
src/vs/platform/telemetry/common/commonProperties.ts 14 covered LOC · 3 ranges

Open complete file

1 > /*--------------------------------------------------------------------------------------------- commonProperties.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > import { isLinuxSnap, platform, Platform, PlatformToString } from '../../../base/common/platform.js';
7 > import { env, platform as nodePlatform } from '../../../base/common/process.js';
8 > import { generateUuid } from '../../../base/common/uuid.js';
9 > import { ICommonProperties } from './telemetry.js';
10 >
11 function getPlatformDetail(hostname: string): string | undefined {
12 if (platform === Platform.Linux && /^penguin(\.|$)/i.test(hostname)) {
16 return undefined;
17 }
19 > export function resolveCommonProperties(
20 release: string,
21 hostname: string,
97 return result;
98 }
100 > export function verifyMicrosoftInternalDomain(domainList: readonly string[]): boolean {
101 const userDnsDomain = env['USERDNSDOMAIN'];
102 if (!userDnsDomain) {
src/vs/base/common/jsonErrorMessages.ts 12 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- jsonErrorMessages.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * Extracted from json.ts to keep json nls free.
8 > */
9 > import { localize } from '../../nls.js';
10 > import { ParseErrorCode } from './json.js';
11 >
12 > export function getParseErrorMessage(errorCode: ParseErrorCode): string {
13 switch (errorCode) {
14 case ParseErrorCode.InvalidSymbol: return localize('error.invalidSymbol', 'Invalid symbol');
src/vs/base/common/observableInternal/commonFacade/deps.ts 10 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- deps.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export { assertFn } from '../../assert.js';
7 > export { type EqualityComparer, strictEquals } from '../../equals.js';
8 > export { BugIndicatingError, onBugIndicatingError, onUnexpectedError } from '../../errors.js';
9 > export { Event, type IValueWithChangeEvent } from '../../event.js';
10 > export { DisposableStore, type IDisposable, markAsDisposed, toDisposable, trackDisposable } from '../../lifecycle.js';
src/vs/base/common/symbols.ts 9 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- symbols.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > /**
7 > * Can be passed into the Delayed to defer using a microtask
8 > * */
9 > export const MicrotaskDelay = Symbol('MicrotaskDelay');
src/vs/base/common/observable.ts 8 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- observable.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > // This is a facade for the observable implementation. Only import from here!
7 >
8 > export * from './observableInternal/index.js';
src/vs/base/common/observableInternal/commonFacade/cancellation.ts 7 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- cancellation.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export { CancellationError } from '../../errors.js';
7 > export { CancellationToken, CancellationTokenSource, cancelOnDispose } from '../../cancellation.js';
src/vs/workbench/services/workspaces/common/workspaceUtils.ts 7 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- workspaceUtils.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 > import { IWorkspace } from '../../../../platform/workspace/common/workspace.js';
6 > import { IFileService } from '../../../../platform/files/common/files.js';
7 >
8 export async function areWorkspaceFoldersEmpty(workspace: IWorkspace, fileService: IFileService): Promise<boolean> {
9 for (const folder of workspace.folders) {
src/vs/workbench/contrib/mcp/common/modelContextProtocol.ts 6 covered LOC · 1 range

Open complete file

1 > /*--------------------------------------------------------------------------------------------- modelContextProtocol.ts
2 > * Copyright (c) Microsoft Corporation. All rights reserved.
3 > * Licensed under the MIT License. See License.txt in the project root for license information.
4 > *--------------------------------------------------------------------------------------------*/
5 >
6 > export { MCP } from '../../../../platform/mcp/common/modelContextProtocol.js';