1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
import { useEffect, useState } from "react";
import "vscode-webview";
declare const vscode: any;
export function postVscMessage(type: string, data: any) {
if (typeof vscode === "undefined") {
return;
}
vscode.postMessage({
type,
...data,
});
}
export async function vscRequest(type: string, data: any): Promise<any> {
return new Promise((resolve) => {
const handler = (event: any) => {
if (event.data.type === type) {
window.removeEventListener("message", handler);
resolve(event.data);
}
};
window.addEventListener("message", handler);
postVscMessage(type, data);
});
}
export function useVscMessageValue(
messageType: string | string[],
initialValue?: any
) {
const [value, setValue] = useState<any>(initialValue);
window.addEventListener("message", (event) => {
if (event.data.type === messageType) {
setValue(event.data.value);
}
});
return [value, setValue];
}
export async function withProgress(title: string, fn: () => Promise<any>) {
postVscMessage("withProgress", { title, done: false });
return fn().finally(() => {
postVscMessage("withProgress", { title, done: true });
});
}
|
