admin/HaE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Version: 4.0.3 Update

Browse Source
This commit is contained in:
gh0stkey
2025-01-10 17:45:59 +08:00
parent a06ef8e25e
commit 452f297f55
2 changed files with 9 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/hae/HaE.java
View File

@@ -20,7 +20,7 @@ public class HaE implements BurpExtension {
public void initialize(MontoyaApi api) { public void initialize(MontoyaApi api) {
// 设置扩展名称 // 设置扩展名称
api.extension().setName("HaE - Highlighter and Extractor"); api.extension().setName("HaE - Highlighter and Extractor");
String version = "4.0.2"; String version = "4.0.3";
// 加载扩展后输出的项目信息 // 加载扩展后输出的项目信息
Logging logging = api.logging(); Logging logging = api.logging();

51
src/main/java/hae/component/board/table/Datatable.java
View File

@@ -51,6 +51,8 @@ public class Datatable extends JPanel {
} }
private void initComponents(List<String> dataList) { private void initComponents(List<String> dataList) {
dataTable.setRowSorter(sorter);
// 设置ID排序 // 设置ID排序
sorter.setComparator(0, new Comparator<Integer>() { sorter.setComparator(0, new Comparator<Integer>() {
@Override @Override
@@ -107,7 +109,6 @@ public class Datatable extends JPanel {
JScrollPane scrollPane = new JScrollPane(dataTable); JScrollPane scrollPane = new JScrollPane(dataTable);
scrollPane.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS); scrollPane.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS);
dataTable.setRowSorter(sorter);
TableColumn idColumn = dataTable.getColumnModel().getColumn(0); TableColumn idColumn = dataTable.getColumnModel().getColumn(0);
idColumn.setPreferredWidth(50); idColumn.setPreferredWidth(50);
idColumn.setMaxWidth(100); idColumn.setMaxWidth(100);
@@ -162,8 +163,8 @@ public class Datatable extends JPanel {
} }
private void performSearch() { private void performSearch() {
RowFilter<Object, Object> firstRowFilter = applyFirstSearchFilter(); RowFilter<Object, Object> firstRowFilter = getObjectObjectRowFilter(searchField);
RowFilter<Object, Object> secondRowFilter = applySecondFilter(); RowFilter<Object, Object> secondRowFilter = getObjectObjectRowFilter(secondSearchField);
if (searchField.getForeground().equals(Color.BLACK)) { if (searchField.getForeground().equals(Color.BLACK)) {
sorter.setRowFilter(firstRowFilter); sorter.setRowFilter(firstRowFilter);
if (secondSearchField.getForeground().equals(Color.BLACK)) { if (secondSearchField.getForeground().equals(Color.BLACK)) {
@@ -175,7 +176,7 @@ public class Datatable extends JPanel {
} }
} }
private RowFilter<Object, Object> applyFirstSearchFilter() { private RowFilter<Object, Object> getObjectObjectRowFilter(JTextField searchField) {
return new RowFilter<Object, Object>() { return new RowFilter<Object, Object>() {
public boolean include(Entry<?, ?> entry) { public boolean include(Entry<?, ?> entry) {
String searchFieldTextText = searchField.getText(); String searchFieldTextText = searchField.getText();
@@ -187,32 +188,12 @@ public class Datatable extends JPanel {
String entryValue = ((String) entry.getValue(1)).toLowerCase(); String entryValue = ((String) entry.getValue(1)).toLowerCase();
searchFieldTextText = searchFieldTextText.toLowerCase(); searchFieldTextText = searchFieldTextText.toLowerCase();
boolean filterReturn = searchFieldTextText.isEmpty();
if (pattern != null) { if (pattern != null) {
return searchFieldTextText.isEmpty() || pattern.matcher(entryValue).find() != searchMode.isSelected(); filterReturn = filterReturn || pattern.matcher(entryValue).find() != searchMode.isSelected();
} else {
return searchFieldTextText.isEmpty() || entryValue.contains(searchFieldTextText) != searchMode.isSelected();
}
}
};
}
private RowFilter<Object, Object> applySecondFilter() {
return new RowFilter<Object, Object>() {
public boolean include(Entry<?, ?> entry) {
String searchFieldTextText = secondSearchField.getText();
Pattern pattern = null;
try {
pattern = Pattern.compile(searchFieldTextText, Pattern.CASE_INSENSITIVE);
} catch (Exception ignored) {
} }
String entryValue = ((String) entry.getValue(1)).toLowerCase(); return filterReturn || entryValue.contains(searchFieldTextText) != searchMode.isSelected();
searchFieldTextText = searchFieldTextText.toLowerCase();
if (pattern != null) {
return searchFieldTextText.isEmpty() || pattern.matcher(entryValue).find();
} else {
return searchFieldTextText.isEmpty() || entryValue.contains(searchFieldTextText);
}
} }
}; };
} }
@@ -248,22 +229,6 @@ public class Datatable extends JPanel {
}); });
} }
private String getTableData(JTable table) {
StringBuilder selectData = new StringBuilder();
int rowCount = table.getRowCount();
for (int i = 0; i < rowCount; i++) {
selectData.append(table.getValueAt(i, 1).toString()).append("\r\n");
}
if (!selectData.isEmpty()) {
selectData.delete(selectData.length() - 2, selectData.length());
} else {
return "";
}
return selectData.toString();
}
public String getSelectedDataAtTable(JTable table) { public String getSelectedDataAtTable(JTable table) {
int[] selectRows = table.getSelectedRows(); int[] selectRows = table.getSelectedRows();
StringBuilder selectData = new StringBuilder(); StringBuilder selectData = new StringBuilder();