src/services/mocking/MockTextDocument.ts - main - outbreak/kilocode

src/services/mocking/MockTextDocument.ts
147 строк4 KB
Chris Hasson
feat: Remove experimental autocomplete feature (#1565)
29 июл 2025, 10:46
29 июл 2025, 10:468228c86
import * as vscode from "vscode"

/**
 * A simulated vscode TextDocument for testing.
 */
export class MockTextDocument implements vscode.TextDocument {
	private contentLines: string[]
	public uri: vscode.Uri
	public fileName: string
	public isUntitled: boolean = false
	public languageId: string = "typescript"
	public version: number = 1
	public isDirty: boolean = false
	public isClosed: boolean = false
	public eol: vscode.EndOfLine = 1 // vscode.EndOfLine.LF
	public get encoding(): "utf8" {
		return "utf8"
	}
	public get notebook(): undefined {
		return undefined
	}

	constructor(uri: vscode.Uri, content: string) {
		this.uri = uri
		this.fileName = uri.fsPath
		this.contentLines = content.split("\n")
	}

	updateContent(newContent: string): void {
		this.contentLines = newContent.split("\n")
		this.version++
		this.isDirty = true
	}

	getText(range?: vscode.Range): string {
		if (!range) {
			return this.contentLines.join("\n")
		}

		const startLine = range.start.line
		const endLine = range.end.line

		if (startLine === endLine) {
			return this.contentLines[startLine].substring(range.start.character, range.end.character)
		}

		const lines: string[] = []
		for (let i = startLine; i <= endLine && i < this.contentLines.length; i++) {
			if (i === startLine) {
				lines.push(this.contentLines[i].substring(range.start.character))
			} else if (i === endLine) {
				lines.push(this.contentLines[i].substring(0, range.end.character))
			} else {
				lines.push(this.contentLines[i])
			}
		}

		return lines.join("\n")
	}

	get lineCount(): number {
		return this.contentLines.length
	}

	/**
	 * Returns information about a specific line in the document
	 * @param lineNumber The zero-based line number
	 * @returns A simplified TextLine object containing the text and position information
	 */
	lineAt(positionOrLine: number | vscode.Position): vscode.TextLine {
		const line = typeof positionOrLine === "number" ? positionOrLine : positionOrLine.line
		if (line < 0 || line >= this.contentLines.length) {
			throw new Error(`Invalid line number: ${line}`)
		}

		const text = this.contentLines[line]
		const range = new vscode.Range(new vscode.Position(line, 0), new vscode.Position(line, text.length))

		return {
			text,
			range,
			lineNumber: line,
			rangeIncludingLineBreak: range, // Simplified for mock
			firstNonWhitespaceCharacterIndex: text.search(/\S|$/),
			isEmptyOrWhitespace: !/\S/.test(text),
		}
	}

	// Add other required methods with mock implementations
	offsetAt(position: vscode.Position): number {
		let offset = 0
		for (let i = 0; i < position.line; i++) {
			if (i < this.contentLines.length) {
				offset += this.contentLines[i].length + 1 // +1 for newline
			}
		}
		offset += position.character
		return offset
	}

	positionAt(offset: number): vscode.Position {
		let currentOffset = 0
		for (let i = 0; i < this.contentLines.length; i++) {
			const lineLength = this.contentLines[i].length + 1
			if (currentOffset + lineLength > offset) {
				return new vscode.Position(i, offset - currentOffset)
			}
			currentOffset += lineLength
		}
		// If offset is beyond the end of the document
		const lastLine = this.contentLines.length - 1
		const lastLineLength = this.contentLines[lastLine]?.length || 0
		return new vscode.Position(lastLine, lastLineLength)
	}

	save(): Promise<boolean> {
		this.isDirty = false
		return Promise.resolve(true)
	}

	getWordRangeAtPosition(position: vscode.Position, regex?: RegExp): vscode.Range | undefined {
		const line = this.lineAt(position.line)
		const text = line.text
		const wordRegex =
			regex || /(-?\d*\.\d\w*)|([^\`\~\!\@\#\$\%\^\&\*\(\)\-\=\+\[\{\]\}\\\|\;\:\'\"\,\.\<\>\/\?\s]+)/g

		let match
		while ((match = wordRegex.exec(text)) !== null) {
			const start = match.index
			const end = start + match[0].length
			if (position.character >= start && position.character <= end) {
				return new vscode.Range(position.line, start, position.line, end)
			}
		}
		return undefined
	}

	validateRange(range: vscode.Range): vscode.Range {
		// Simplified validation
		return range
	}

	validatePosition(position: vscode.Position): vscode.Position {
		// Simplified validation
		return position
	}
}