Update TextAreaMenu.java

bit4woo · Sep 3, 2024 · c6a1c3f · c6a1c3f
1 parent 6887cb7
commit c6a1c3f
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/src/Tools/TextAreaMenu.java b/src/Tools/TextAreaMenu.java
@@ -6,9 +6,12 @@
 
 import javax.swing.AbstractAction;
 import javax.swing.JMenuItem;
+import javax.swing.JOptionPane;
 import javax.swing.JPopupMenu;
 import javax.swing.JTextArea;
 
+import org.apache.commons.lang3.StringUtils;
+
 import com.bit4woo.utilbox.utils.IPAddressUtils;
 import com.bit4woo.utilbox.utils.TextUtils;
 
@@ -136,6 +139,32 @@ public void actionPerformed(ActionEvent actionEvent) {
 				guiMain.getDomainPanel().saveDomainDataToDB();
 			}
 		});
+
+
+		JMenuItem addToTargetWithComment = new JMenuItem(new AbstractAction("Add To Target With Comment") {
+			@Override
+			public void actionPerformed(ActionEvent actionEvent) {
+
+				String comment = JOptionPane.showInputDialog("Comment", "");
+				if (StringUtils.isBlank(comment)) {
+					return;
+				}
+
+				DomainManager domainResult = guiMain.getDomainPanel().getDomainResult();
+				for (String item:selectedItems) {
+					try {
+						if (IPAddressUtils.isValidIPv4MayPort(item)) {
+							domainResult.getSpecialPortTargets().add(item);
+						}else {
+							domainResult.addToTargetAndSubDomain(item,true,AssetTrustLevel.Confirm,comment);
+						}
+					} catch (Exception e2) {
+						e2.printStackTrace(stderr);
+					}
+				}
+				guiMain.getDomainPanel().saveDomainDataToDB();
+			}
+		});
 
 		JMenuItem doOnlineSearch = new JMenuItem(new AbstractAction("Do Online Search") {
 			@Override
@@ -153,6 +182,7 @@ public void actionPerformed(ActionEvent actionEvent) {
 		this.add(genPortScanCmd);
 		this.add(addToTarget);
 		this.add(addToTargetConfirm);
+		this.add(addToTargetWithComment);
 		this.add(doOnlineSearch);
 	}
 }