Context menu entries can now be executed when at least one of the selected messages is valid.

Added text into context menu entry which shows how may of the selected messages are valid (e.g. 2/5).

Author: bytebutcher

burp-send-to-extension/src/main/java/net/bytebutcher/burpsendtoextension/gui/SendToContextMenu.java

@@ -42,7 +42,13 @@ public CommandObjectMenuItem(Action a, CommandObject commandObject) {
         }
 
         public void refreshStatus(List<Map<String, IPlaceholder>> placeholders) {
-            setEnabled(commandObject.isValid(placeholders));
+            List<Map<String, IPlaceholder>> validEntries = commandObject.getValid(placeholders);
+            if (placeholders.size() > 1) {
+                setText(commandObject.getName() + " (" + validEntries.size() + "/" + placeholders.size() + ")");
+            } else {
+                setText(commandObject.getName());
+            }
+            setEnabled(validEntries.size() > 0);
         }
     }
 

burp-send-to-extension/src/main/java/net/bytebutcher/burpsendtoextension/models/CommandObject.java

@@ -1,5 +1,6 @@
 package net.bytebutcher.burpsendtoextension.models;
 
+import com.google.common.collect.Lists;
 import com.google.common.collect.Sets;
 import net.bytebutcher.burpsendtoextension.models.placeholder.IPlaceholder;
 import net.bytebutcher.burpsendtoextension.utils.StringUtils;
@@ -94,13 +95,21 @@ public int hashCode() {
         return Objects.hash(id);
     }
 
-    public boolean isValid(List<Map<String, IPlaceholder>> placeholders) {
+    public List<Map<String, IPlaceholder>> getValid(List<Map<String, IPlaceholder>> placeholders) {
+        List<Map<String, IPlaceholder>> validItems = Lists.newArrayList();
         for (Map<String, IPlaceholder> placeholderMap : placeholders) {
-            for (String placeholder : getInternalPlaceHolders()) {
-                if (placeholderMap.containsKey(placeholder)) {
-                    if (!placeholderMap.get(placeholder).isValid()) {
-                        return false;
-                    }
+            if (isValid(placeholderMap)) {
+                validItems.add(placeholderMap);
+            }
+        }
+        return validItems;
+    }
+
+    private boolean isValid(Map<String, IPlaceholder> placeholderMap) {
+        for (String placeholder : getInternalPlaceHolders()) {
+            if (placeholderMap.containsKey(placeholder)) {
+                if (!placeholderMap.get(placeholder).isValid()) {
+                    return false;
                 }
             }
         }
@@ -115,7 +124,7 @@ public String getFormattedCommand(List<Map<String, IPlaceholder>> placeholderMap
         try {
             String originalCommand = getCommand();
             for (String internalPlaceHolder : getInternalPlaceHolders()) {
-                String value = placeholderMap.stream().map(m -> m.get(internalPlaceHolder)).map(iPlaceholder -> iPlaceholder.getValue(internalPlaceHolder)).collect(Collectors.joining(","));
+                String value = getValid(placeholderMap).stream().map(m -> m.get(internalPlaceHolder)).map(iPlaceholder -> iPlaceholder.getValue(internalPlaceHolder)).collect(Collectors.joining(","));
                 boolean doesRequireShellEscape = placeholderMap.get(0).get(internalPlaceHolder).doesRequireShellEscape();
                 originalCommand = originalCommand.replace(internalPlaceHolder, doesRequireShellEscape ? "'" + StringUtils.shellEscape(value) + "'" : value);
             }