fix: webview cefQuery

dineug · Jan 19, 2024 · cbb2172 · cbb2172
1 parent 0f0f296
commit cbb2172
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/packages/erd-editor-jetbrains-webview/src/env.d.ts b/packages/erd-editor-jetbrains-webview/src/env.d.ts
@@ -1,5 +1,11 @@
-declare global {
-  interface Window {
-    cefQuery: (arg: any) => number;
-  }
+interface Window {
+  cefQuery: (query: {
+    request: string;
+    onSuccess: (response: string) => void;
+    onFailure: (errorCode: number, errorMessage: string) => void;
+    context?: any;
+    persistent: boolean;
+    keepAlive?: boolean;
+  }) => number;
+  cefQueryCancel: (requestId: number) => void;
 }
diff --git a/packages/erd-editor-jetbrains-webview/src/main.ts b/packages/erd-editor-jetbrains-webview/src/main.ts
@@ -19,12 +19,16 @@ import {
 import { encode } from 'base64-arraybuffer';
 
 const bridge = new Emitter();
-// const vscode = globalThis.acquireVsCodeApi();
 const editor = document.createElement('erd-editor');
 const sharedStore = editor.getSharedStore({ mouseTracker: false });
 
 const dispatch = (action: AnyAction) => {
-  // vscode.postMessage(action);
+  window.cefQuery({
+    request: JSON.stringify(action),
+    persistent: false,
+    onSuccess: () => {},
+    onFailure: () => {},
+  });
 };
 
 import('@dineug/erd-editor-shiki-worker').then(({ getShikiService }) => {
@@ -95,5 +99,5 @@ bridge.on({
   },
 });
 
-globalThis.addEventListener('message', event => bridge.emit(event.data));
+window.addEventListener('message', event => bridge.emit(event.data));
 dispatch(vscodeInitialAction());