Skip to content

Commit ea79986

Browse files
committed
code refactoring
1 parent 2fcf977 commit ea79986

File tree

2 files changed

+32
-45
lines changed

2 files changed

+32
-45
lines changed

src/main/java/swurg/gui/components/menus/ParametersContextMenu.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -58,15 +58,14 @@ private JMenu createHighlightMenu() {
5858
private JMenuItem createHighlightMenuItem(Color color) {
5959
JMenuItem menuItem = new JMenuItem();
6060

61+
CustomTableCellRenderer renderer = (CustomTableCellRenderer) table.getDefaultRenderer(Object.class);
62+
6163
menuItem.setOpaque(true);
6264
menuItem.setBackground(color);
6365
menuItem.setForeground(Color.BLACK);
6466

65-
CustomTableCellRenderer renderer = (CustomTableCellRenderer) table.getDefaultRenderer(Object.class);
66-
67-
menuItem.addActionListener(e -> processSelectedRows(index -> {
68-
SwingUtilities.invokeLater(() -> renderer.setRowHighlightColor(index, color));
69-
}));
67+
menuItem.addActionListener(e -> processSelectedRows(
68+
index -> SwingUtilities.invokeLater(() -> renderer.setRowHighlightColor(index, color))));
7069

7170
return menuItem;
7271
}

src/main/java/swurg/gui/components/menus/ParserContextMenu.java

Lines changed: 28 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -47,37 +47,28 @@ public ParserContextMenu(MontoyaApi montoyaApi, JTable table) {
4747
}
4848

4949
private void initComponents() {
50-
JMenuItem copyToClipboard = createCopyToClipboardMenuItem();
51-
52-
JMenuItem addToScope = createAddToScopeMenuItem();
53-
JMenuItem addToSiteMap = createAddToSiteMapMenuItem();
54-
JMenuItem sendToActiveScan = createSendToActiveScanMenuItem();
55-
JMenuItem sendToPassiveScan = createSendToPassiveScanMenuItem();
56-
JMenuItem sendToIntruder = createSendToIntruderMenuItem();
57-
JMenuItem sendToRepeater = createSendToRepeaterMenuItem();
58-
JMenuItem sendToOrganizer = createSendToOrganizerMenuItem();
59-
JMenuItem sendToComparer = createSendToComparerMenuItem();
60-
JMenu highlightMenu = createHighlightMenu();
61-
JMenuItem clearItems = createClearItemsMenuItem();
62-
JMenuItem clearAll = createClearAllMenuItem();
63-
64-
this.add(copyToClipboard);
50+
this.add(createCopyToClipboardMenuItem());
6551
this.add(new JSeparator());
66-
this.add(addToScope);
52+
53+
this.add(createAddToScopeMenuItem());
6754
this.add(new JSeparator());
68-
this.add(sendToPassiveScan);
69-
this.add(sendToActiveScan);
55+
56+
this.add(createSendToPassiveScanMenuItem());
57+
this.add(createSendToActiveScanMenuItem());
7058
this.add(new JSeparator());
71-
this.add(sendToIntruder);
72-
this.add(sendToRepeater);
73-
this.add(sendToOrganizer);
74-
this.add(sendToComparer);
59+
60+
this.add(createSendToIntruderMenuItem());
61+
this.add(createSendToRepeaterMenuItem());
62+
this.add(createSendToOrganizerMenuItem());
63+
this.add(createSendToComparerMenuItem());
7564
this.add(new JSeparator());
76-
this.add(highlightMenu);
77-
this.add(clearItems);
78-
this.add(clearAll);
65+
66+
this.add(createHighlightMenu());
67+
this.add(createClearItemsMenuItem());
68+
this.add(createClearAllMenuItem());
7969
this.add(new JSeparator());
80-
this.add(addToSiteMap);
70+
71+
this.add(createAddToSiteMapMenuItem());
8172
}
8273

8374
private HttpRequest getHttpRequestAt(int index) {
@@ -96,7 +87,7 @@ private void processSelectedRows(Consumer<Integer> action) {
9687
private JMenuItem createCopyToClipboardMenuItem() {
9788
JMenuItem copyToClipboard = new JMenuItem();
9889

99-
table.addMouseListener(new MouseAdapter() {
90+
this.table.addMouseListener(new MouseAdapter() {
10091
@Override
10192
public void mousePressed(MouseEvent e) {
10293
JTable source = (JTable) e.getSource();
@@ -199,6 +190,7 @@ private JMenuItem createSendToOrganizerMenuItem() {
199190

200191
private JMenuItem createSendToComparerMenuItem() {
201192
JMenuItem sendToComparer = new JMenuItem("Send to Comparer (request)");
193+
202194
sendToComparer.addActionListener(e -> processSelectedRows(index -> {
203195
HttpRequest httpRequest = getHttpRequestAt(index);
204196
montoyaApi.comparer().sendToComparer(httpRequest.toByteArray());
@@ -223,15 +215,14 @@ private JMenu createHighlightMenu() {
223215
private JMenuItem createHighlightMenuItem(Color color) {
224216
JMenuItem menuItem = new JMenuItem();
225217

218+
CustomTableCellRenderer renderer = (CustomTableCellRenderer) table.getDefaultRenderer(Object.class);
219+
226220
menuItem.setOpaque(true);
227221
menuItem.setBackground(color);
228222
menuItem.setForeground(Color.BLACK);
229223

230-
CustomTableCellRenderer renderer = (CustomTableCellRenderer) table.getDefaultRenderer(Object.class);
231-
232-
menuItem.addActionListener(e -> processSelectedRows(index -> {
233-
SwingUtilities.invokeLater(() -> renderer.setRowHighlightColor(index, color));
234-
}));
224+
menuItem.addActionListener(e -> processSelectedRows(
225+
index -> SwingUtilities.invokeLater(() -> renderer.setRowHighlightColor(index, color))));
235226

236227
return menuItem;
237228
}
@@ -251,7 +242,6 @@ private JMenuItem createClearItemsMenuItem() {
251242
// Remove rows individually from the table.
252243
SwingUtilities.invokeLater(() -> {
253244
ParserTableModel tableModel = (ParserTableModel) table.getModel();
254-
255245
selectedRows.forEach(index -> {
256246
int modelRow = table.convertRowIndexToModel(index);
257247
tableModel.removeRow(modelRow);
@@ -272,13 +262,11 @@ private JMenuItem createClearAllMenuItem() {
272262

273263
CustomTableCellRenderer renderer = (CustomTableCellRenderer) table.getDefaultRenderer(Object.class);
274264

275-
clearAll.addActionListener(e -> {
276-
SwingUtilities.invokeLater(() -> {
277-
ParserTableModel tableModel = (ParserTableModel) table.getModel();
278-
tableModel.clear();
279-
renderer.clearAllRowHighlightColors();
280-
});
281-
});
265+
clearAll.addActionListener(e -> SwingUtilities.invokeLater(() -> {
266+
ParserTableModel tableModel = (ParserTableModel) table.getModel();
267+
tableModel.clear();
268+
renderer.clearAllRowHighlightColors();
269+
}));
282270

283271
return clearAll;
284272
}

0 commit comments

Comments
 (0)