Carlos Florêncio
48 lines
1.7 KiB
TypeScript
48 lines
1.7 KiB
TypeScript
import type { JsonRpcConnection } from "./lsp-json-rpc";
|
|
import type { ManagedLspConnection } from "./lsp-client-types";
|
|
import { applyLspProgress, createLspProgressSnapshot, isLspProgressToken } from "./lsp-progress";
|
|
|
|
export function beginLspProgressTracking(
|
|
connection: ManagedLspConnection,
|
|
rpc: JsonRpcConnection,
|
|
isCurrent: () => boolean,
|
|
notifyChange: () => void,
|
|
resumeState?: { progressTokens?: unknown[]; progress?: unknown[] },
|
|
): void {
|
|
connection.progress = createLspProgressSnapshot(resumeState ? null : Date.now());
|
|
connection.registeredProgressTokens.clear();
|
|
connection.registeredProgressTokens.add(connection.ownerId);
|
|
for (const token of resumeState?.progressTokens ?? []) {
|
|
if (isLspProgressToken(token)) connection.registeredProgressTokens.add(token);
|
|
}
|
|
for (const progress of resumeState?.progress ?? []) {
|
|
connection.progress = applyLspProgress(
|
|
connection.progress,
|
|
connection.registeredProgressTokens,
|
|
progress,
|
|
Date.now(),
|
|
);
|
|
}
|
|
notifyChange();
|
|
|
|
rpc.onRequest("window/workDoneProgress/create", (params: unknown) => {
|
|
if (!isCurrent() || typeof params !== "object" || params === null) return null;
|
|
const token = (params as { token?: unknown }).token;
|
|
if (isLspProgressToken(token)) connection.registeredProgressTokens.add(token);
|
|
return null;
|
|
});
|
|
|
|
rpc.onNotification("$/progress", (params: unknown) => {
|
|
if (!isCurrent()) return;
|
|
const progress = applyLspProgress(
|
|
connection.progress,
|
|
connection.registeredProgressTokens,
|
|
params,
|
|
Date.now(),
|
|
);
|
|
if (progress === connection.progress) return;
|
|
connection.progress = progress;
|
|
notifyChange();
|
|
});
|
|
}
|