reload jaeger and dashboard

Author: shalom938

src/main/java/org/digma/intellij/plugin/dashboard/DashboardService.java

@@ -1,5 +1,7 @@
 package org.digma.intellij.plugin.dashboard;
 
+import com.intellij.openapi.Disposable;
+import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.components.Service;
 import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.openapi.fileEditor.FileEditorManager;
@@ -9,20 +11,29 @@
 import org.digma.intellij.plugin.dashboard.incoming.GoToSpan;
 import org.digma.intellij.plugin.log.Log;
 import org.digma.intellij.plugin.posthog.*;
+import org.digma.intellij.plugin.reload.*;
 import org.digma.intellij.plugin.scope.*;
 import org.jetbrains.annotations.NotNull;
 
 import java.io.InputStream;
 import java.util.*;
 
+import static org.digma.intellij.plugin.dashboard.DashboardVirtualFile.DASHBOARD_EDITOR_KEY;
+
 @Service(Service.Level.PROJECT)
-public final class DashboardService {
+public final class DashboardService implements Disposable, ReloadableJCefContainer {
 
     private final Logger logger = Logger.getInstance(DashboardService.class);
     private final Project project;
 
     public DashboardService(Project project) {
         this.project = project;
+        ApplicationManager.getApplication().getService(ReloadService.class).register(this, this);
+    }
+
+    @Override
+    public void dispose() {
+        //nothing to do, used as parent disposable
     }
 
     public static DashboardService getInstance(Project project) {
@@ -98,6 +109,35 @@ public void goToSpan(GoToSpan goToSpan) {
 
         var span = goToSpan.payload();
 
-        ScopeManager.getInstance(project).changeScope(new SpanScope(span.spanCodeObjectId()),false,null,null,span.environment());
+        ScopeManager.getInstance(project).changeScope(new SpanScope(span.spanCodeObjectId()), false, null, null, span.environment());
+    }
+
+
+    @Override
+    public void reload() {
+
+        var files = Arrays.stream(FileEditorManager.getInstance(project).getOpenFiles()).filter(DashboardVirtualFile::isDashboardVirtualFile).toList();
+
+        var newFiles = new ArrayList<DashboardVirtualFile>();
+
+        files.forEach(oldFile -> {
+            if (oldFile instanceof DashboardVirtualFile dashboardVirtualFile) {
+                var newFile = new DashboardVirtualFile(dashboardVirtualFile.getName());
+                newFile.setDashboardName(dashboardVirtualFile.getDashboardName());
+                newFile.setPath(dashboardVirtualFile.getPath());
+                DASHBOARD_EDITOR_KEY.set(newFile, DashboardFileEditorProvider.DASHBOARD_EDITOR_TYPE);
+                newFiles.add(newFile);
+            }
+        });
+
+        files.forEach(file -> FileEditorManager.getInstance(project).closeFile(file));
+
+        newFiles.forEach(file -> FileEditorManager.getInstance(project).openFile(file));
+    }
+
+    @NotNull
+    @Override
+    public Project getProject() {
+        return project;
     }
 }

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

@@ -1,6 +1,7 @@
 package org.digma.intellij.plugin.jaegerui;
 
 import com.intellij.openapi.Disposable;
+import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.openapi.fileEditor.FileEditorManager;
 import com.intellij.openapi.project.Project;
@@ -14,15 +15,18 @@
 import org.digma.intellij.plugin.navigation.codenavigation.CodeNavigator;
 import org.digma.intellij.plugin.posthog.*;
 import org.digma.intellij.plugin.psi.*;
+import org.digma.intellij.plugin.reload.*;
 import org.digma.intellij.plugin.scope.*;
 import org.digma.intellij.plugin.settings.SettingsState;
 import org.digma.intellij.plugin.ui.model.TraceSample;
 import org.jetbrains.annotations.*;
 
 import java.util.*;
 
+import static org.digma.intellij.plugin.jaegerui.JaegerUIVirtualFile.JAEGER_UI_EDITOR_KEY;
 
-public class JaegerUIService implements Disposable {
+
+public class JaegerUIService implements Disposable, ReloadableJCefContainer {
 
     private final Logger logger = Logger.getInstance(JaegerUIService.class);
 
@@ -31,6 +35,7 @@ public class JaegerUIService implements Disposable {
 
     public JaegerUIService(Project project) {
         this.project = project;
+        ApplicationManager.getApplication().getService(ReloadService.class).register(this, this);
     }
 
     @Override
@@ -237,4 +242,35 @@ private Map<String, List<Insight>> getInsights(@NotNull List<String> spanIds) {
     }
 
 
+    @Override
+    public void reload() {
+
+        var files = Arrays.stream(FileEditorManager.getInstance(project).getOpenFiles()).filter(JaegerUIVirtualFile::isJaegerUIVirtualFile).toList();
+
+        var newFiles = new ArrayList<JaegerUIVirtualFile>();
+
+        files.forEach(oldFile -> {
+            if (oldFile instanceof JaegerUIVirtualFile jaegerUIVirtualFile) {
+                var newFile = new JaegerUIVirtualFile(jaegerUIVirtualFile.getName());
+                newFile.setJaegerBaseUrl(jaegerUIVirtualFile.getJaegerBaseUrl());
+                newFile.setTraceId(jaegerUIVirtualFile.getTraceId());
+                newFile.setTraceSamples(jaegerUIVirtualFile.getTraceSamples());
+                newFile.setSpanName(jaegerUIVirtualFile.getSpanName());
+                newFile.setSpanCodeObjectId(jaegerUIVirtualFile.getSpanCodeObjectId());
+                newFile.setSendSearchQuery(jaegerUIVirtualFile.isSendSearchQuery());
+                JAEGER_UI_EDITOR_KEY.set(newFile, JaegerUIFileEditorProvider.JAEGER_UI_EDITOR_TYPE);
+                newFiles.add(newFile);
+            }
+        });
+
+        files.forEach(filo -> FileEditorManager.getInstance(project).closeFile(filo));
+
+        newFiles.forEach(file -> FileEditorManager.getInstance(project).openFile(file));
+    }
+
+    @NotNull
+    @Override
+    public Project getProject() {
+        return project;
+    }
 }

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

@@ -6,8 +6,7 @@
 import com.intellij.testFramework.LightVirtualFile;
 import org.digma.intellij.plugin.common.DigmaVirtualFileMarker;
 import org.digma.intellij.plugin.ui.model.TraceSample;
-import org.jetbrains.annotations.NotNull;
-import org.jetbrains.annotations.Nullable;
+import org.jetbrains.annotations.*;
 
 import java.util.List;
 
@@ -57,15 +56,15 @@ public static VirtualFile createVirtualFile(String jaegerBaseUrl, @NotNull List<
         return file;
     }
 
-    private void setTraceSamples(List<TraceSample> traceSamples) {
+    public void setTraceSamples(List<TraceSample> traceSamples) {
         this.traceSamples = traceSamples;
     }
 
-    private void setTraceId(String traceId) {
+    public void setTraceId(String traceId) {
         this.traceId = traceId;
     }
 
-    private void setJaegerBaseUrl(String jaegerBaseUrl) {
+    public void setJaegerBaseUrl(String jaegerBaseUrl) {
         this.jaegerBaseUrl = jaegerBaseUrl;
     }
 
@@ -81,7 +80,7 @@ public List<TraceSample> getTraceSamples() {
         return traceSamples;
     }
 
-    private void setSpanName(String spanName) {
+    public void setSpanName(String spanName) {
         this.spanName = spanName;
     }