Merge pull request #2247 from digma-ai/changes-in-jcef-component

dispose some jcef components

Author: shalom938

src/main/java/org/digma/intellij/plugin/jaegerui/JaegerUIFileEditor.java

@@ -13,11 +13,13 @@
 
 public class JaegerUIFileEditor extends UserDataHolderBase implements FileEditor {
 
-    private final VirtualFile file;
+    private final JaegerUIVirtualFile file;
 
     @Nullable
     private JCefComponent jCefComponent;
 
+    private boolean disposed = false;
+
     public JaegerUIFileEditor(Project project, JaegerUIVirtualFile file) {
         this.file = file;
         jCefComponent = createJcefComponent(project, file);
@@ -81,7 +83,7 @@ public boolean isModified() {
 
     @Override
     public boolean isValid() {
-        return true;
+        return !disposed;
     }
 
     @Override
@@ -100,6 +102,8 @@ public void dispose() {
             jCefComponent.dispose();
             jCefComponent = null;
         }
+        disposed = true;
+        file.setValid(false);
     }
 
 }

src/main/java/org/digma/intellij/plugin/jaegerui/JaegerUIFileEditorProvider.java

@@ -22,12 +22,6 @@ public boolean accept(@NotNull Project project, @NotNull VirtualFile file) {
         return new JaegerUIFileEditor(project, (JaegerUIVirtualFile) file);
     }
 
-    @Override
-    public void disposeEditor(@NotNull FileEditor editor) {
-        FileEditorProvider.super.disposeEditor(editor);
-    }
-
-
     @Override
     public @NotNull @NonNls String getEditorTypeId() {
         return JAEGER_UI_EDITOR_TYPE;

src/main/java/org/digma/intellij/plugin/jaegerui/JaegerUIVirtualFile.java

@@ -13,6 +13,7 @@
 
 public class JaegerUIVirtualFile extends LightVirtualFile implements DigmaVirtualFileMarker {
 
+    //todo: not sure this key is necessary, it is not used
     public static final Key<String> JAEGER_UI_EDITOR_KEY = Key.create("Digma.JAEGER_UI_EDITOR_KEY");
     private String jaegerBaseUrl;
     private String traceId;

src/main/kotlin/org/digma/intellij/plugin/ui/jcef/JCefComponent.kt

@@ -373,19 +373,24 @@ private constructor(
 
             jbCefClient.cefClient.addDisplayHandler(JCefDisplayHandler(name))
 
-            jbCefBrowser.jbCefClient.addLifeSpanHandler(LifeSpanHandle(schemeHandlerFactory), jbCefBrowser.cefBrowser)
+            val lifeSpanHandle = LifeSpanHandle(schemeHandlerFactory)
+            jbCefClient.addLifeSpanHandler(lifeSpanHandle, jbCefBrowser.cefBrowser)
 
             downloadAdapterRef?.get()?.let {
-                jbCefClient.cefClient.addDownloadHandler(it)
+                jbCefClient.addDownloadHandler(it, jbCefBrowser.cefBrowser)
             }
 
             val jCefComponent =
                 JCefComponent(project, parentDisposable, name, jbCefBrowser, cefMessageRouter)
 
-            //usually the component that holds a reference to JCefComponent needs to call JCefComponent.dispose.
-            //when a parentDisposable is supplied then use it also to dispose, worst case dispose will be called twice.
             Disposer.register(parentDisposable) {
-                jCefComponent.dispose()
+                cefMessageRouter.removeHandler(messageRouterHandler)
+                cefMessageRouter.dispose()
+                jbCefClient.cefClient.removeMessageRouter(cefMessageRouter)
+                jbCefClient.removeLifeSpanHandler(lifeSpanHandle, jbCefBrowser.cefBrowser)
+                downloadAdapterRef?.get()?.let {
+                    jbCefClient.removeDownloadHandle(it, jbCefBrowser.cefBrowser)
+                }
             }
 
 
@@ -421,4 +426,5 @@ class LifeSpanHandle(private val schemeHandlerFactory: BaseSchemeHandlerFactory)
             schemeHandlerFactory.getSchema(), null, schemeHandlerFactory
         )
     }
+
 }