Update SecuCheck notification with action to open SARIF file

secure-software-engineering · May 7, 2024 · 8081d8d · 8081d8d
1 parent 3d6fb24
commit 8081d8d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/dev-assist/src/main/java/de/fraunhofer/iem/devassist/ui/MethodListTree.java b/dev-assist/src/main/java/de/fraunhofer/iem/devassist/ui/MethodListTree.java
@@ -12,6 +12,7 @@
 import com.intellij.notification.*;
 import com.intellij.openapi.actionSystem.ActionManager;
 import com.intellij.openapi.actionSystem.ActionPopupMenu;
+import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.fileEditor.*;
@@ -54,6 +55,7 @@
 import java.awt.event.KeyListener;
 import java.awt.event.MouseAdapter;
 import java.awt.event.MouseEvent;
+import java.io.File;
 import java.text.DecimalFormat;
 import java.util.*;
 
@@ -389,8 +391,23 @@ public void launchSwan(HashMap<String, String> values) {
         bus.connect().subscribe(SecucheckNotifier.END_SECUCHECK_PROCESS_TOPIC, new SecucheckNotifier() {
             @Override
             public void launchSecuCheck() {
-                String notificationContent = "SecuCheck results exported to "+PropertiesComponent.getInstance(project).getValue(Constants.OUTPUT_DIRECTORY);
-                NotificationGroupManager.getInstance().getNotificationGroup("Process_Completed").createNotification(notificationContent, NotificationType.INFORMATION).notify(project);
+                String notificationContent = "Taint analysis results exported successfully.";
+                NotificationGroupManager.getInstance()
+                        .getNotificationGroup("Analysis Notification")
+                        .createNotification(notificationContent, NotificationType.INFORMATION)
+                        .addAction(new NotificationAction("Open Results") {
+                            @Override
+                            public void actionPerformed(@NotNull AnActionEvent anActionEvent, @NotNull Notification notification) {
+
+                                File results = new File(Objects.requireNonNull(PropertiesComponent.getInstance(project).getValue(Constants.LAST_SARIF_FILE)));
+                                Optional<VirtualFile> file = FilenameIndex
+                                        .getVirtualFilesByName(results.getName(),
+                                                GlobalSearchScope.projectScope(project)).stream().findFirst();
+
+                                file.ifPresent(virtualFile -> new OpenFileDescriptor(project, virtualFile).navigate(true));
+                            }
+                        })
+                        .notify(project);
             }
         });
 

diff --git a/dev-assist/src/main/resources/META-INF/plugin.xml b/dev-assist/src/main/resources/META-INF/plugin.xml
@@ -22,8 +22,8 @@
         <codeInsight.lineMarkerProvider language="JAVA"
                                         implementationClass="de.fraunhofer.iem.devassist.ui.markers.ErrorLineMarker"/>
         <toolWindow id="Dev-Assist" anchor="right" factoryClass="de.fraunhofer.iem.devassist.ui.SummaryToolWindow" icon="PluginIcons.DEV_ASSIST"/>
-        <notificationGroup id="Process_Completed" displayType="TOOL_WINDOW"/>
         <notificationGroup id="SRM Notification" displayType="BALLOON"/>
+        <notificationGroup id="Analysis Notification" displayType="STICKY_BALLOON"/>
     </extensions>
 
     <actions>