@vscode/chat-lib 0.0.1

package/src/_internal/platform/inlineEdits/common/statelessNextEditProvider.ts

@@ -0,0 +1,468 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Raw } from '@vscode/prompt-tsx';
+import { Result } from '../../../util/common/result';
+import { assert, assertNever } from '../../../util/vs/base/common/assert';
+import { DeferredPromise } from '../../../util/vs/base/common/async';
+import { CancellationToken, CancellationTokenSource } from '../../../util/vs/base/common/cancellation';
+import { URI } from '../../../util/vs/base/common/uri';
+import { LineEdit, LineReplacement, SerializedLineEdit } from '../../../util/vs/editor/common/core/edits/lineEdit';
+import { StringEdit } from '../../../util/vs/editor/common/core/edits/stringEdit';
+import { OffsetRange } from '../../../util/vs/editor/common/core/ranges/offsetRange';
+import { StringText } from '../../../util/vs/editor/common/core/text/abstractText';
+import { ChatFetchResponseType, FetchResponse } from '../../chat/common/commonTypes';
+import { ISerializedOffsetRange, LogEntry, serializeOffsetRange } from '../../workspaceRecorder/common/workspaceLog';
+import { DocumentId } from './dataTypes/documentId';
+import { Edits } from './dataTypes/edit';
+import { SerializedEdit } from './dataTypes/editUtils';
+import { LanguageId } from './dataTypes/languageId';
+import { DebugRecorderBookmark } from './debugRecorderBookmark';
+import { InlineEditRequestLogContext } from './inlineEditLogContext';
+import { stringifyChatMessages } from './utils/stringifyChatMessages';
+import { IXtabHistoryEntry } from './workspaceEditTracker/nesXtabHistoryTracker';
+
+export const enum ShowNextEditPreference {
+	Always = 'always',
+	AroundEdit = 'aroundEdit',
+}
+
+export type PushEdit = (edit: Result<{ edit: LineReplacement; window?: OffsetRange; targetDocument?: DocumentId }, NoNextEditReason>) => void;
+
+export interface IStatelessNextEditProvider {
+	readonly ID: string;
+	readonly dependsOnSelection?: boolean;
+	readonly showNextEditPreference?: ShowNextEditPreference;
+	provideNextEdit(request: StatelessNextEditRequest, pushEdit: PushEdit, logContext: InlineEditRequestLogContext, cancellationToken: CancellationToken): Promise<StatelessNextEditResult>;
+	handleAcceptance?(): void;
+	handleRejection?(): void;
+}
+
+export class StatelessNextEditRequest<TFirstEdit = any> {
+
+	private static ID = 0;
+	public readonly seqid = String(++StatelessNextEditRequest.ID);
+
+	public readonly cancellationTokenSource = new CancellationTokenSource();
+	public liveDependentants = 0; // number of invocations which haven't been canceled and depend on this request
+	public fetchIssued = false;
+	public intermediateUserEdit: StringEdit | undefined = StringEdit.empty;
+
+	private readonly _result: DeferredPromise<StatelessNextEditResult> = new DeferredPromise<StatelessNextEditResult>();
+	public get result(): Promise<StatelessNextEditResult> {
+		return this._result.p;
+	}
+
+	constructor(
+		public readonly id: string,
+		public readonly opportunityId: string,
+		public readonly documentBeforeEdits: StringText,
+		public readonly documents: readonly StatelessNextEditDocument[],
+		public readonly activeDocumentIdx: number,
+		public readonly xtabEditHistory: IXtabHistoryEntry[],
+		public readonly firstEdit: DeferredPromise<Result<TFirstEdit, NoNextEditReason>>,
+		public readonly expandedEditWindowNLines: number | undefined,
+		public readonly logContext: InlineEditRequestLogContext,
+		public readonly recordingBookmark: DebugRecorderBookmark | undefined,
+		public readonly recording: LogEntry[] | undefined,
+		public readonly providerRequestStartDateTime: number | undefined,
+	) {
+		assert(documents.length > 0);
+		assert(activeDocumentIdx >= 0 && activeDocumentIdx < documents.length);
+	}
+
+	public setResult(nextEditResult: StatelessNextEditResult) {
+		this._result.complete(nextEditResult);
+	}
+
+	public setResultError(err: any) {
+		this._result.error(err);
+	}
+
+	public hasDocument(docId: DocumentId): boolean {
+		return this.documents.find(d => d.id === docId) !== undefined;
+	}
+
+	getActiveDocument(): StatelessNextEditDocument {
+		return this.documents[this.activeDocumentIdx];
+	}
+
+	serialize(): ISerializedNextEditRequest {
+		return {
+			id: this.id,
+			documents: this.documents.map(d => d.serialize()),
+			activeDocumentIdx: this.activeDocumentIdx,
+			recording: this.recording,
+		};
+	}
+
+	toString(): string {
+		return this.toMarkdown();
+	}
+
+	toMarkdown(): string {
+		const docs = this.documents.map((d, idx) => ` * [${idx + 1}/${this.documents.length}] ${idx === this.activeDocumentIdx ? '(active document) ' : ''}` + d.toMarkdown()).join('\n\n');
+		return `### StatelessNextEditRequest\n\n${docs}`;
+	}
+}
+
+export interface ISerializedNextEditRequest {
+	id: string;
+	documents: ISerializedNextEditDocument[];
+	activeDocumentIdx: number;
+	recording: LogEntry[] | undefined;
+}
+
+export class StatelessNextEditDocument {
+	public readonly documentAfterEdits = new StringText(this.recentEdits.apply(this.documentBeforeEdits.value));
+	public readonly documentAfterEditsLines: string[] = this.documentAfterEdits.getLines();
+
+	/**
+	 * NOTE: if you add new public fields to this class, please also update {@link ISerializedNextEditDocument} and {@link serialize()} methods,
+	 * which are used to send this to http-server-powered NES provider.
+	 */
+	constructor(
+		public readonly id: DocumentId,
+		public readonly workspaceRoot: URI | undefined,
+		public readonly languageId: LanguageId,
+		public readonly documentLinesBeforeEdit: string[],
+		public readonly recentEdit: LineEdit,
+		public readonly documentBeforeEdits: StringText,
+		public readonly recentEdits: Edits,
+		public readonly lastSelectionInAfterEdit: OffsetRange | undefined = undefined,
+	) { }
+
+	serialize(): ISerializedNextEditDocument {
+		return {
+			id: this.id.uri,
+			workspaceRoot: this.workspaceRoot?.toString(),
+			languageId: this.languageId,
+			documentLinesBeforeEdit: this.documentLinesBeforeEdit,
+			recentEdit: this.recentEdit.serialize(),
+			documentBeforeEdits: this.documentBeforeEdits.value,
+			recentEdits: this.recentEdits.serialize(),
+			lastSelectionInAfterEdit: this.lastSelectionInAfterEdit === undefined ? undefined : serializeOffsetRange(this.lastSelectionInAfterEdit),
+		};
+	}
+
+	toString(): string {
+		return this.toMarkdown();
+	}
+
+	toMarkdown(): string {
+		const lines: string[] = [];
+
+		lines.push(`StatelessNextEditDocument: **${this.id.uri}**\n`);
+		lines.push('```patch');
+		lines.push(this.recentEdit.humanReadablePatch(this.documentLinesBeforeEdit));
+		lines.push('```');
+		lines.push('');
+
+		return lines.join('\n');
+	}
+}
+
+export interface ISerializedNextEditDocument {
+	id: string;
+	workspaceRoot: string | undefined;
+	languageId: string;
+	documentLinesBeforeEdit: string[];
+	recentEdit: SerializedLineEdit;
+	documentBeforeEdits: string;
+	recentEdits: SerializedEdit[];
+	lastSelectionInAfterEdit: ISerializedOffsetRange | undefined;
+}
+
+export enum FilteredOutReason {
+	LowLogProbSuggestions = 'lowLogProbSuggestions',
+	EnforcingNextEditOptions = 'enforcingNextEditOptions',
+	PromptTooLarge = 'promptTooLarge',
+	Uncategorized = 'uncategorized',
+}
+
+export namespace NoNextEditReason {
+	export class ActiveDocumentHasNoEdits {
+		public readonly kind = 'activeDocumentHasNoEdits';
+	}
+	export class NoSuggestions {
+		public readonly kind = 'noSuggestions';
+		constructor(
+			public readonly documentBeforeEdits: StringText,
+			public readonly window: OffsetRange | undefined
+		) {
+		}
+	}
+	export class GotCancelled {
+		public readonly kind = 'gotCancelled';
+		constructor(public readonly message: 'afterDebounce' | 'afterGettingEndpoint' | 'afterPromptConstruction' | 'afterFetchCall' | 'duringStreaming' | 'afterResponse' | 'afterFailedRebase' | 'beforeExecutingNewRequest') {
+		}
+	}
+	export class FetchFailure {
+		public readonly kind = 'fetchFailure';
+		constructor(public readonly error: Error) {
+		}
+	}
+	export class FilteredOut {
+		public readonly kind = 'filteredOut';
+		constructor(public readonly message: FilteredOutReason | string) {
+		}
+	}
+	export class Uncategorized {
+		public readonly kind = 'uncategorized';
+		constructor(public readonly error: Error) {
+		}
+	}
+	export class Unexpected {
+		public readonly kind = 'unexpected';
+		constructor(public readonly error: Error) {
+		}
+	}
+}
+
+export type NoNextEditReason =
+	| NoNextEditReason.ActiveDocumentHasNoEdits
+	| NoNextEditReason.NoSuggestions
+	| NoNextEditReason.GotCancelled
+	| NoNextEditReason.FetchFailure
+	| NoNextEditReason.FilteredOut
+	| NoNextEditReason.Uncategorized
+	| NoNextEditReason.Unexpected
+	;
+
+export class StatelessNextEditResult {
+	public static noEdit(reason: NoNextEditReason, telemetryBuilder: StatelessNextEditTelemetryBuilder): StatelessNextEditResult {
+		const result = Result.error(reason);
+		const telemetry = telemetryBuilder.build(result);
+		return new StatelessNextEditResult(result, telemetry);
+	}
+
+	public static streaming(telemetryBuilder: StatelessNextEditTelemetryBuilder): StatelessNextEditResult {
+		const result = Result.ok<void>(undefined);
+		const telemetry = telemetryBuilder.build(result);
+		return new StatelessNextEditResult(result, telemetry);
+	}
+
+	constructor(
+		public readonly nextEdit: Result<void, NoNextEditReason>,
+		public readonly telemetry: IStatelessNextEditTelemetry,
+	) {
+	}
+}
+
+export interface IStatelessNextEditTelemetry {
+
+	readonly hadStatelessNextEditProviderCall: boolean;
+
+	/* general info */
+	readonly statelessNextEditProviderDuration: number;
+	readonly isCursorAtEndOfLine: boolean | undefined;
+	readonly nLinesOfCurrentFileInPrompt: number | undefined;
+	readonly modelName: string | undefined;
+
+	/* options info */
+	readonly logProbThreshold: number | undefined;
+
+	/* prompt info */
+
+	readonly prompt: string | undefined;
+	readonly promptLineCount: number | undefined;
+	readonly promptCharCount: number | undefined;
+
+	/* fetch request info */
+
+	readonly debounceTime: number | undefined;
+	/** This's only used to compute time from inline edit provider call to fetch init. Not included in telemetry. */
+	readonly fetchStartedAt: number | undefined;
+
+	/* response info */
+
+	/** Artificial delay (aka backoff) on the response based on previous user acceptance/rejection in milliseconds */
+	readonly artificialDelay: number | undefined;
+
+	readonly hadLowLogProbSuggestion: boolean | undefined;
+	readonly response: undefined | Promise<FetchResultWithStats>;
+
+	/* suggestions info */
+
+	readonly nEditsSuggested: number | undefined;
+	readonly lineDistanceToMostRecentEdit: number | undefined;
+
+	/* result info */
+	readonly nextEditLogprob: number | undefined;
+	readonly noNextEditReasonKind: string | undefined;
+	readonly noNextEditReasonMessage: string | undefined;
+	readonly summarizedEditWindow: any;
+}
+
+export type FetchResultWithStats = {
+	readonly ttft: number | undefined;
+	readonly response: FetchResponse<string>;
+	readonly fetchTime: number;
+	readonly fetchResult: ChatFetchResponseType;
+}
+
+export class StatelessNextEditTelemetryBuilder {
+
+	public readonly startTime: number;
+	public readonly requestUuid: string;
+
+	/**
+	 * It takes a request to automatically capture some properties from the request.
+	 */
+	constructor(request: StatelessNextEditRequest) {
+		this.startTime = Date.now();
+		this.requestUuid = request.id;
+	}
+
+	public build(result: Result<void, NoNextEditReason>): IStatelessNextEditTelemetry {
+		const endTime = Date.now();
+		const timeSpent = endTime - this.startTime;
+
+		const prompt = this._prompt ? JSON.stringify(this._prompt.map(({ role, content }) => ({ role, content }))) : undefined;
+		const promptText = this._prompt ? stringifyChatMessages(this._prompt) : undefined;
+		const promptLineCount = promptText?.split('\n').length;
+		const promptCharCount = promptText?.length;
+
+		const noNextEditReasonKind = result.isOk() ? undefined : result.err.kind;
+
+		let noNextEditReasonMessage: string | undefined;
+		if (result.isError()) {
+			if (result.err instanceof NoNextEditReason.ActiveDocumentHasNoEdits || result.err instanceof NoNextEditReason.NoSuggestions) {
+				// ignore
+			} else if (result.err instanceof NoNextEditReason.GotCancelled || result.err instanceof NoNextEditReason.FilteredOut) {
+				noNextEditReasonMessage = result.err.message;
+			} else if (result.err instanceof NoNextEditReason.FetchFailure || result.err instanceof NoNextEditReason.Uncategorized || result.err instanceof NoNextEditReason.Unexpected) {
+				noNextEditReasonMessage = result.err.error.stack ? result.err.error.stack : result.err.error.message;
+			} else {
+				assertNever(result.err);
+			}
+		}
+
+		return {
+			hadStatelessNextEditProviderCall: true,
+
+			noNextEditReasonKind,
+			noNextEditReasonMessage,
+
+			statelessNextEditProviderDuration: timeSpent,
+			logProbThreshold: this._logProbThreshold,
+			nLinesOfCurrentFileInPrompt: this._nLinesOfCurrentFileInPrompt,
+			modelName: this._modelName,
+			prompt,
+			promptLineCount,
+			promptCharCount,
+			isCursorAtEndOfLine: this._isCursorAtLineEnd,
+			debounceTime: this._debounceTime,
+			artificialDelay: this._artificialDelay,
+			fetchStartedAt: this._fetchStartedAt,
+			hadLowLogProbSuggestion: this._hadLowLogProbSuggestion,
+			response: this._response,
+			nEditsSuggested: this._nEditsSuggested,
+			nextEditLogprob: this._nextEditLogProb,
+			lineDistanceToMostRecentEdit: this._lineDistanceToMostRecentEdit,
+			summarizedEditWindow: this._summarizedEditWindow,
+		};
+	}
+
+	private _logProbThreshold: number | undefined;
+	public setLogProbThreshold(logProbThreshold: number): this {
+		this._logProbThreshold = logProbThreshold;
+		return this;
+	}
+
+	private _hadLowLogProbSuggestion: boolean | undefined;
+	public setHadLowLogProbSuggestion(hadLowLogProbSuggestions: boolean): this {
+		this._hadLowLogProbSuggestion = hadLowLogProbSuggestions;
+		return this;
+	}
+
+	private _nLinesOfCurrentFileInPrompt: number | undefined;
+	public setNLinesOfCurrentFileInPrompt(nLines: number): this {
+		this._nLinesOfCurrentFileInPrompt = nLines;
+		return this;
+	}
+
+	private _modelName: string | undefined;
+	public setModelName(modelName: string): this {
+		this._modelName = modelName;
+		return this;
+	}
+
+	private _prompt: Raw.ChatMessage[] | undefined;
+	public setPrompt(prompt: Raw.ChatMessage[]): this {
+		this._prompt = prompt;
+		return this;
+	}
+
+	private _isCursorAtLineEnd: boolean | undefined;
+	public setIsCursorAtLineEnd(isCursorAtLineEnd: boolean): this {
+		this._isCursorAtLineEnd = isCursorAtLineEnd;
+		return this;
+	}
+
+	private _debounceTime: number | undefined;
+	public setDebounceTime(debounceTime: number): this {
+		this._debounceTime = debounceTime;
+		return this;
+	}
+
+	private _artificialDelay: number | undefined;
+	public setArtificialDelay(artificialDelay: number): this {
+		this._artificialDelay = artificialDelay;
+		return this;
+	}
+
+	private _fetchStartedAt: number | undefined;
+	public setFetchStartedAt(): this {
+		this._fetchStartedAt = Date.now();
+		return this;
+	}
+	public get fetchStartedAt(): number | undefined {
+		return this._fetchStartedAt;
+	}
+
+	private _response: Promise<FetchResultWithStats> | undefined;
+	public setResponse(response: Promise<{ ttft: number | undefined; response: FetchResponse<string> }>): this {
+		this._response = response.then(({ response, ttft }) => {
+
+			const fetchTime = Date.now() - this._fetchStartedAt!;
+
+			const fetchResult = response.type;
+
+			return {
+				ttft,
+				response,
+				fetchTime,
+				fetchResult,
+			};
+		});
+
+		return this;
+	}
+
+	private _nextEditLogProb: number | undefined;
+	public setNextEditLogProb(logProb: number): this {
+		this._nextEditLogProb = logProb;
+		return this;
+	}
+
+	private _nEditsSuggested: number | undefined;
+	public setNEditsSuggested(nEditsSuggested: number): this {
+		this._nEditsSuggested = nEditsSuggested;
+		return this;
+	}
+
+	private _lineDistanceToMostRecentEdit: number | undefined;
+	public setLineDistanceToMostRecentEdit(distanceToMostRecentEdit: number): this {
+		this._lineDistanceToMostRecentEdit = distanceToMostRecentEdit;
+		return this;
+	}
+
+	private _summarizedEditWindow: any;
+	public setSummarizedEditWindow(summarizedEditWindow: any): this {
+		this._summarizedEditWindow = summarizedEditWindow;
+		return this;
+	}
+}

package/src/_internal/platform/inlineEdits/common/statelessNextEditProviders.ts

@@ -0,0 +1,59 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { LineReplacement } from '../../../util/vs/editor/common/core/edits/lineEdit';
+import { StatelessNextEditDocument } from './statelessNextEditProvider';
+
+export class IgnoreEmptyLineAndLeadingTrailingWhitespaceChanges {
+	public static filterEdit(resultDocument: StatelessNextEditDocument, singleEdits: readonly LineReplacement[]): readonly LineReplacement[] {
+		const filteredEdits = singleEdits.filter(e => !IgnoreEmptyLineAndLeadingTrailingWhitespaceChanges._isWhitespaceOnlyChange(e, resultDocument.documentAfterEditsLines));
+		return filteredEdits;
+	}
+
+	private static _isWhitespaceOnlyChange(edit: LineReplacement, baseLines: string[]): boolean {
+		const originalLines = edit.lineRange.toOffsetRange().slice(baseLines);
+		const newLines = edit.newLines;
+
+		const isRemoval = newLines.length === 0;
+
+		// is removing empty lines
+		if (isRemoval && originalLines.every(line => line.trim() === '')) {
+			return true;
+		}
+
+		// is adding empty lines
+		if (!isRemoval && newLines.every(line => line.trim() === '')) {
+			return true;
+		}
+
+		if (originalLines.length !== newLines.length) {
+			return false;
+		}
+
+		for (let i = 0; i < originalLines.length; i++) {
+			const originalLine = originalLines[i];
+			const newLine = newLines[i];
+			if (originalLine.trim() !== newLine.trim()) {
+				return false;
+			}
+		}
+		return true;
+	}
+}
+
+export class IgnoreWhitespaceOnlyChanges {
+	public static filterEdit(resultDocument: StatelessNextEditDocument, singleEdits: readonly LineReplacement[]): readonly LineReplacement[] {
+		return singleEdits.filter(e => !IgnoreWhitespaceOnlyChanges._isFormattingOnlyChange(resultDocument.documentAfterEditsLines, e));
+	}
+
+	/**
+	 * @remarks public only for testing
+	 */
+	public static _isFormattingOnlyChange(baseLines: string[], singleEdit: LineReplacement): boolean {
+		const originalLines = singleEdit.lineRange.toOffsetRange().slice(baseLines).join('').replace(/\s/g, '');
+		const newLines = singleEdit.newLines.join('').replace(/\s/g, '');
+		return originalLines === newLines;
+	}
+}

package/src/_internal/platform/inlineEdits/common/utils/observable.ts

@@ -0,0 +1,54 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { IDisposable } from '../../../../util/vs/base/common/lifecycle';
+import { autorunHandleChanges, IObservableWithChange } from '../../../../util/vs/base/common/observable';
+
+type ObservableResult<T> = T extends IObservableWithChange<infer U, any> ? U : never;
+
+type ObservableArrayToChangesData<T extends Record<string, IObservableWithChange<any, any>>> = {
+	[Key in keyof T]: {
+		value: ObservableResult<T[Key]>;
+		changes: T[Key]['TChange'][];
+		/**
+		 * The value of the observable before the changes. `undefined` if
+		 */
+		previous: ObservableResult<T[Key]> | undefined;
+	}
+};
+
+export function autorunWithChanges<T extends Record<string, IObservableWithChange<any, any>>>(owner: object, observables: T, handler: (data: ObservableArrayToChangesData<T>) => void): IDisposable {
+	const observableToKey = new Map(Object.entries(observables).map(([key, value]) => [value, key] as const));
+
+	const previousValues = new Map(Object.keys(observables).map(key => [key, undefined]));
+
+	return autorunHandleChanges({
+		owner,
+		changeTracker: {
+			createChangeSummary: () => ({}) as ObservableArrayToChangesData<T>,
+			handleChange: (ctx, changeSummary) => {
+				const key = observableToKey.get(ctx.changedObservable)!;
+
+				if (changeSummary[key] === undefined) {
+					(changeSummary as any)[key] = { value: undefined!, changes: [] };
+				}
+				changeSummary[key].changes.push(ctx.change);
+				return true;
+			}
+		}
+	}, (reader, data) => {
+		for (const [key, value] of Object.entries(observables)) {
+			const v = value.read(reader);
+
+			if (data[key] === undefined) {
+				(data as any)[key] = { value: v, changes: [], previous: previousValues.get(key) };
+			}
+			data[key].value = v;
+			data[key].previous = previousValues.get(key) === undefined ? undefined : previousValues.get(key);
+			previousValues.set(key, v);
+		}
+		handler(data);
+	});
+}

package/src/_internal/platform/inlineEdits/common/utils/stringifyChatMessages.ts

@@ -0,0 +1,29 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Raw } from '@vscode/prompt-tsx';
+
+export function stringifyChatMessages(messages: Raw.ChatMessage[]) {
+	return messages.map(stringifyMessage).join('\n');
+}
+
+function stringifyMessage({ role, content }: Raw.ChatMessage) {
+	if (role !== Raw.ChatRole.User && role !== Raw.ChatRole.System) {
+		return 'omitted because of non-user and non-system role'; // should be impossible
+	}
+
+	const roleStr = role === Raw.ChatRole.User ? 'User' : 'System';
+
+	const textContentPart = content.at(0);
+	if (textContentPart?.type !== Raw.ChatCompletionContentPartKind.Text) {
+		return 'omitted because of non-text content'; // should be impossible
+	}
+
+	return (
+		`${roleStr}
+------
+${textContentPart.text}
+==================`);
+}

package/src/_internal/platform/inlineEdits/common/utils/utils.ts

@@ -0,0 +1,56 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { ThemeIcon } from '../../../../util/vs/base/common/themables';
+
+/**
+ * Represents a specific point in time.
+*/
+export type Instant = number;
+
+let overridenNowValue = -1;
+
+export function overrideNowValue(value: number): void {
+	overridenNowValue = value;
+}
+
+export function now(): Instant {
+	if (overridenNowValue !== -1) {
+		return overridenNowValue;
+	}
+	return Date.now();
+}
+
+export namespace Icon {
+	export type t = {
+		themeIcon: ThemeIcon;
+		svg: string;
+	};
+
+	export const circleSlash: t = {
+		themeIcon: ThemeIcon.fromId('circle-slash'),
+		svg: `<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 16 16"><path fill="currentColor" d="M8 1a7 7 0 1 1-7 7a7.01 7.01 0 0 1 7-7M2 8c0 1.418.504 2.79 1.423 3.87l8.447-8.447A5.993 5.993 0 0 0 2 8m12 0c0-1.418-.504-2.79-1.423-3.87L4.13 12.577A5.993 5.993 0 0 0 14 8"/></svg>`,
+	};
+
+	export const error: t = {
+		themeIcon: ThemeIcon.fromId('error'),
+		svg: `<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 16 16"><path fill="currentColor" fill-rule="evenodd" d="M8.6 1c1.6.1 3.1.9 4.2 2c1.3 1.4 2 3.1 2 5.1c0 1.6-.6 3.1-1.6 4.4c-1 1.2-2.4 2.1-4 2.4s-3.2.1-4.6-.7s-2.5-2-3.1-3.5S.8 7.5 1.3 6c.5-1.6 1.4-2.9 2.8-3.8C5.4 1.3 7 .9 8.6 1m.5 12.9c1.3-.3 2.5-1 3.4-2.1c.8-1.1 1.3-2.4 1.2-3.8c0-1.6-.6-3.2-1.7-4.3c-1-1-2.2-1.6-3.6-1.7c-1.3-.1-2.7.2-3.8 1S2.7 4.9 2.3 6.3c-.4 1.3-.4 2.7.2 4q.9 1.95 2.7 3c1.2.7 2.6.9 3.9.6M7.9 7.5L10.3 5l.7.7l-2.4 2.5l2.4 2.5l-.7.7l-2.4-2.5l-2.4 2.5l-.7-.7l2.4-2.5l-2.4-2.5l.7-.7z" clip-rule="evenodd"/></svg>`,
+	};
+
+	export const skipped: t = {
+		themeIcon: ThemeIcon.fromId('testing-skipped-icon'),
+		svg: `<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 16 16"><path fill="currentColor" fill-rule="evenodd" d="M14.25 5.75v-4h-1.5v2.542c-1.145-1.359-2.911-2.209-4.84-2.209c-3.177 0-5.92 2.307-6.16 5.398l-.02.269h1.501l.022-.226c.212-2.195 2.202-3.94 4.656-3.94c1.736 0 3.244.875 4.05 2.166h-2.83v1.5h4.163l.962-.975V5.75zM8 14a2 2 0 1 0 0-4a2 2 0 0 0 0 4" clip-rule="evenodd"/></svg>`,
+	};
+
+	export const lightbulbFull: t = {
+		themeIcon: ThemeIcon.fromId('refactor-preview-view-icon'),
+		svg: `<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 16 16"><path fill="currentColor" fill-rule="evenodd" d="M11.67 8.658a3.7 3.7 0 0 0-.781 1.114a3.3 3.3 0 0 0-.268 1.329v1.6a1.3 1.3 0 0 1-.794 1.197a1.3 1.3 0 0 1-.509.102H7.712a1.3 1.3 0 0 1-.922-.379a1.3 1.3 0 0 1-.38-.92v-1.6q0-.718-.274-1.329a3.6 3.6 0 0 0-.776-1.114a4.7 4.7 0 0 1-1.006-1.437A4.2 4.2 0 0 1 4 5.5a4.43 4.43 0 0 1 .616-2.27q.296-.504.705-.914a4.6 4.6 0 0 1 .911-.702q.508-.294 1.084-.454a4.5 4.5 0 0 1 1.2-.16a4.5 4.5 0 0 1 2.276.614a4.5 4.5 0 0 1 1.622 1.616a4.44 4.44 0 0 1 .616 2.27q0 .926-.353 1.721a4.7 4.7 0 0 1-1.006 1.437zM9.623 10.5H7.409v2.201q-.001.12.09.212a.3.3 0 0 0 .213.09h1.606a.3.3 0 0 0 .213-.09a.3.3 0 0 0 .09-.212V10.5z" clip-rule="evenodd"/></svg>`
+	};
+
+	export const database: t = {
+		themeIcon: ThemeIcon.fromId('database'),
+		svg: `<svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" viewBox="0 0 16 16"><path fill="currentColor" d="M13 3.5C13 2.119 10.761 1 8 1S3 2.119 3 3.5c0 .04.02.077.024.117H3v8.872l.056.357C3.336 14.056 5.429 15 8 15s4.664-.944 4.944-2.154l.056-.357V3.617h-.024c.004-.04.024-.077.024-.117M8 2.032c2.442 0 4 .964 4 1.468s-1.558 1.468-4 1.468S4 4 4 3.5s1.558-1.468 4-1.468m4 10.458l-.03.131C11.855 13.116 10.431 14 8 14s-3.855-.884-3.97-1.379L4 12.49v-7.5A7.4 7.4 0 0 0 8 6a7.4 7.4 0 0 0 4-1.014z"/></svg>`,
+	};
+}