temp sending messages [ci skip]

Author: benfdking

vscode/extension/src/webviews/lineagePanel.ts

@@ -30,8 +30,9 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
 
     window.onDidChangeActiveTextEditor((editor) => {
       if (editor) {
+        console.log("sending message to web view", editor.document.uri.fsPath, "panel is present", this.panel?.webview);
         this.panel?.webview.postMessage({
-          key: "vscode_callback",
+          key: "vscode_send",
           payload: {
             key: "changeFocusOnFile",
             payload: { path: editor.document.uri.fsPath },
@@ -82,9 +83,6 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
                 const document = await workspace.openTextDocument(fullPath);
                 await window.showTextDocument(document);
                 break;
-              case "formatProject":
-                console.log("formatProject", message.payload);
-                break;
               default:
                 console.log(`Unhandled message type in key: ${message.key}`);
             }
@@ -129,10 +127,19 @@ export class LineagePanel implements WebviewViewProvider, Disposable {
     
     // Define a function to handle events from the extension and send them to the iframe
     const vscode = acquireVsCodeApi();
+    vscode.postMessage = function(message) {
+      // Forward the message to the iframe
+      const iframe = document.querySelector('iframe');
+      if (iframe && iframe.contentWindow) {
+        iframe.contentWindow.postMessage(message, '*');
+      }
+    };
+
+    // Listen for messages from the extension
     window.addEventListener('message', (event) => {
-      // Check if the message is from the extension
-      if (event.source === window && event.data) {
-        // Forward the message to the iframe
+      console.log("message received in web view", event);
+      // Only handle messages that are from the extension
+      if (event.data && event.data.key === 'vscode_send') {
         const iframe = document.querySelector('iframe');
         if (iframe && iframe.contentWindow) {
           iframe.contentWindow.postMessage(event.data, '*');

vscode/react/src/routes/lineage.tsx

@@ -20,11 +20,8 @@ function Wrappper() {
   React.useEffect(() => {
     const handleMessage = (event: MessageEvent) => {
       // Ensure the message is from VSCode
-      if (event.data && event.data.key === 'vscode_callback') {
+      if (event.data && event.data.key === 'vscode_send') {
         const payload = event.data.payload
-        console.log('Received message from VSCode:', payload)
-        
-        // Handle different types of messages
         switch (payload.key) {
           case 'changeFocusOnFile':
             console.log('Changing focus to file:', payload.payload.path)
@@ -35,10 +32,7 @@ function Wrappper() {
         }
       }
     }
-
     window.addEventListener('message', handleMessage)
-    
-    // Clean up event listener on component unmount
     return () => {
       window.removeEventListener('message', handleMessage)
     }