Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor(vscode): separate the workspace lsp server request from the …
…chat feature
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| import EventEmitter from "events"; | ||
| import { | ||
| ApplyWorkspaceEditParams, | ||
| ApplyWorkspaceEditRequest, | ||
| RevealEditorRangeParams, | ||
| RevealEditorRangeRequest, | ||
| } from "tabby-agent"; | ||
| import { BaseLanguageClient, FeatureState, RegistrationData, StaticFeature, TextEdit } from "vscode-languageclient"; | ||
| import { Disposable, Position, Range, TextDocument, TextEditorEdit, window, workspace } from "vscode"; | ||
| import { diffLines } from "diff"; | ||
| import { getLogger } from "../logger"; | ||
|
Check failure on line 11 in clients/vscode/src/lsp/WorkspaceFeature.ts
|
||
| export class WorkSpaceFeature extends EventEmitter implements StaticFeature { | ||
| private registration: string | undefined = undefined; | ||
| private disposables: Disposable[] = []; | ||
|
|
||
| constructor(private readonly client: BaseLanguageClient) { | ||
| super(); | ||
| } | ||
| getState(): FeatureState { | ||
| return { kind: "static" }; | ||
| } | ||
|
|
||
| fillInitializeParams() { | ||
| // nothing | ||
| } | ||
|
|
||
| fillClientCapabilities(): void { | ||
| // nothing | ||
| } | ||
|
|
||
| preInitialize(): void { | ||
| // nothing | ||
| } | ||
|
|
||
| initialize(): void { | ||
| this.disposables.push( | ||
| this.client.onRequest(ApplyWorkspaceEditRequest.type, (params: ApplyWorkspaceEditParams) => { | ||
| return this.handleApplyWorkspaceEdit(params); | ||
| }), | ||
| this.client.onRequest(RevealEditorRangeRequest.type, (params: RevealEditorRangeParams) => { | ||
| return this.handleRevealEditorRange(params); | ||
| }), | ||
| ); | ||
| } | ||
|
|
||
| register(data: RegistrationData<unknown>): void { | ||
| this.registration = data.id; | ||
| this.emit("didChangeAvailability", true); | ||
| } | ||
|
|
||
| unregister(id: string): void { | ||
| if (this.registration === id) { | ||
| this.registration = undefined; | ||
| this.emit("didChangeAvailability", false); | ||
| } | ||
| } | ||
|
|
||
| clear(): void { | ||
| this.disposables.forEach((disposable) => disposable.dispose()); | ||
| this.disposables = []; | ||
| } | ||
|
|
||
| get isAvailable(): boolean { | ||
| return !!this.registration; | ||
| } | ||
|
|
||
| private async handleApplyWorkspaceEdit(params: ApplyWorkspaceEditParams): Promise<boolean> { | ||
| const { edit, options } = params; | ||
| const activeEditor = window.activeTextEditor; | ||
| if (!activeEditor) { | ||
| return false; | ||
| } | ||
|
|
||
| try { | ||
| const success = await activeEditor.edit( | ||
| (editBuilder: TextEditorEdit) => { | ||
| Object.entries(edit.changes || {}).forEach(([uri, textEdits]) => { | ||
| const document = workspace.textDocuments.find((doc) => doc.uri.toString() === uri); | ||
| if (document && document === activeEditor.document) { | ||
| const textEdit = textEdits[0]; | ||
| if (textEdits.length === 1 && textEdit) { | ||
| applyTextEditMinimalLineChange(editBuilder, textEdit, document); | ||
| } else { | ||
| textEdits.forEach((textEdit) => { | ||
| const range = new Range( | ||
| new Position(textEdit.range.start.line, textEdit.range.start.character), | ||
| new Position(textEdit.range.end.line, textEdit.range.end.character), | ||
| ); | ||
| editBuilder.replace(range, textEdit.newText); | ||
| }); | ||
| } | ||
| } | ||
| }); | ||
| }, | ||
| { | ||
| undoStopBefore: options?.undoStopBefore ?? false, | ||
| undoStopAfter: options?.undoStopAfter ?? false, | ||
| }, | ||
| ); | ||
|
|
||
| return success; | ||
| } catch (error) { | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| handleRevealEditorRange(params: RevealEditorRangeParams): boolean { | ||
| const { range, revealType } = params; | ||
| const activeEditor = window.activeTextEditor; | ||
| if (!activeEditor) { | ||
| return false; | ||
| } | ||
|
|
||
| activeEditor.revealRange( | ||
| new Range( | ||
| new Position(range.start.line, range.start.character), | ||
| new Position(range.end.line, range.end.character), | ||
| ), | ||
| revealType, | ||
| ); | ||
|
|
||
| return true; | ||
| } | ||
| } | ||
|
|
||
| function applyTextEditMinimalLineChange(editBuilder: TextEditorEdit, textEdit: TextEdit, document: TextDocument) { | ||
| const documentRange = new Range( | ||
| new Position(textEdit.range.start.line, textEdit.range.start.character), | ||
| new Position(textEdit.range.end.line, textEdit.range.end.character), | ||
| ); | ||
|
|
||
| const text = document.getText(documentRange); | ||
| const newText = textEdit.newText; | ||
| const diffs = diffLines(text, newText); | ||
|
|
||
| let line = documentRange.start.line; | ||
| for (const diff of diffs) { | ||
| if (!diff.count) { | ||
| continue; | ||
| } | ||
|
|
||
| if (diff.added) { | ||
| editBuilder.insert(new Position(line, 0), diff.value); | ||
| } else if (diff.removed) { | ||
| const range = new Range(new Position(line + 0, 0), new Position(line + diff.count, 0)); | ||
| editBuilder.delete(range); | ||
| line += diff.count; | ||
| } else { | ||
| line += diff.count; | ||
| } | ||
| } | ||
| } | ||
