From 11dfe0e5b8592e8746d476946a97387e8ca9e456 Mon Sep 17 00:00:00 2001 From: bit4woo Date: Thu, 25 Apr 2024 17:49:54 +0800 Subject: [PATCH] update --- src/config/ConfigTable.java | 251 ++++++++++++++----------------- src/config/ConfigTableModel.java | 2 - 2 files changed, 115 insertions(+), 138 deletions(-) diff --git a/src/config/ConfigTable.java b/src/config/ConfigTable.java index ba985b9..974face 100644 --- a/src/config/ConfigTable.java +++ b/src/config/ConfigTable.java @@ -1,8 +1,6 @@ package config; -import java.awt.Color; -import java.awt.Font; -import java.awt.FontMetrics; +import java.awt.*; import java.awt.event.MouseAdapter; import java.awt.event.MouseEvent; import java.io.PrintWriter; @@ -21,137 +19,118 @@ import burp.BurpExtender; - -public class ConfigTable extends JTable -{ - /** - * - */ - private static final long serialVersionUID = 1L; - - - public ConfigTable(ConfigTableModel ConfigTableModel) - { - super(ConfigTableModel); - this.setColumnModel(columnModel); - this.setFillsViewportHeight(true);//在table的空白区域显示右键菜单 - //https://stackoverflow.com/questions/8903040/right-click-mouselistener-on-whole-jtable-component - this.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION); - this.setBorder(new LineBorder(new Color(0, 0, 0))); - - addClickSort(); - registerListeners(); - //switchEnable();//no need - //table.setupTypeColumn()//can't set here, only can after table data loaded. - //tableHeaderLengthInit();//can't set here, only can after table data loaded. - } - - @Override - public void changeSelection(int row, int col, boolean toggle, boolean extend) - { - super.changeSelection(row, col, toggle, extend); - } - - public int[] getSelectedModelRows() { - int[] rows = getSelectedRows(); - - for (int i=0; i < rows.length; i++){ - rows[i] = convertRowIndexToModel(rows[i]);//转换为Model的索引，否则排序后索引不对应〿 - } - Arrays.sort(rows);//升序 - return rows; - } - - private void addClickSort() { - TableRowSorter sorter = new TableRowSorter((ConfigTableModel) this.getModel()); - ConfigTable.this.setRowSorter(sorter); - - JTableHeader header = this.getTableHeader(); - header.addMouseListener(new MouseAdapter() { - @Override - public void mouseClicked(MouseEvent e) { - try { - //ConfigTable.this.getRowSorter().getSortKeys().get(0).getColumn(); - sorter.getSortKeys().get(0).getColumn(); - ////当Jtable中无数据时，jtable.getRowSorter()是null - } catch (Exception e1) { - e1.printStackTrace(new PrintWriter(BurpExtender.callbacks.getStderr(), true));//working? - } - } - }); - } - - /** - * 需要在数据加载后，即setModel后才有效果! - */ - public void tableHeaderLengthInit(){ - Font f = this.getFont(); - FontMetrics fm = this.getFontMetrics(f); - int width = fm.stringWidth("A");//一个字符的宽度 - for (int index=0;index comboBox = new JComboBox<>(); - - String[] items = new ConfigEntry().listAllConfigType(); - for (String item:items) { - comboBox.addItem(item); - } - TableColumnModel model = this.getColumnModel(); - - int col = Arrays.asList(config.ConfigTableModel.titles).indexOf("Type"); - DefaultCellEditor editor = new DefaultCellEditor(comboBox); - editor.setClickCountToStart(2); - model.getColumn(col).setCellEditor(editor); - - JCheckBox jc1 = new JCheckBox(); - int col1 = Arrays.asList(config.ConfigTableModel.titles).indexOf("Enable"); - DefaultCellEditor editor1 = new DefaultCellEditor(jc1); - //editor1.setClickCountToStart(2);//没作用 - model.getColumn(col1).setCellEditor(new DefaultCellEditor(jc1)); - // //Set up tool tips for the sport cells. - // DefaultTableCellRenderer renderer = - // new DefaultTableCellRenderer(); - // renderer.setToolTipText("Click for combo box"); - // typeColumn.setCellRenderer(renderer); - } +import static config.ConfigTableModel.titles; + + +public class ConfigTable extends JTable { + /** + * + */ + private static final long serialVersionUID = 1L; + + + public ConfigTable(ConfigTableModel ConfigTableModel) { + super(ConfigTableModel); + this.setColumnModel(columnModel); + this.setFillsViewportHeight(true);//在table的空白区域显示右键菜单 + //https://stackoverflow.com/questions/8903040/right-click-mouselistener-on-whole-jtable-component + this.setSelectionMode(ListSelectionModel.MULTIPLE_INTERVAL_SELECTION); + this.setBorder(new LineBorder(new Color(0, 0, 0))); + + addClickSort(); + registerListeners(); + //switchEnable();//no need + //table.setupTypeColumn()//can't set here, only can after table data loaded. + //tableHeaderLengthInit();//can't set here, only can after table data loaded. + } + + @Override + public void changeSelection(int row, int col, boolean toggle, boolean extend) { + super.changeSelection(row, col, toggle, extend); + } + + public int[] getSelectedModelRows() { + int[] rows = getSelectedRows(); + + for (int i = 0; i < rows.length; i++) { + rows[i] = convertRowIndexToModel(rows[i]);//转换为Model的索引，否则排序后索引不对应〿 + } + Arrays.sort(rows);//升序 + return rows; + } + + private void addClickSort() { + TableRowSorter sorter = new TableRowSorter<>((ConfigTableModel) this.getModel()); + ConfigTable.this.setRowSorter(sorter); + } + + /** + * 需要在数据加载后，即setModel后才有效果! + */ + public void tableHeaderLengthInit() { + Font f = this.getFont(); + FontMetrics fm = this.getFontMetrics(f); + int width = fm.stringWidth("A");//一个字符的宽度 + for (int index = 0; index < this.getColumnCount(); index++) { + TableColumn column = this.getColumnModel().getColumn(index); + + if (column.getIdentifier().equals("#")) { + column.setMaxWidth(width * "100".length()); + } + + if (column.getIdentifier().equals("Enable")) { + column.setMaxWidth(width * "Enable++".length()); + //需要预留排序时箭头符合的位置，2个字符宽度 + } + + if (column.getIdentifier().equals("Type")) { + column.setPreferredWidth(width * ConfigEntry.Action_If_Base_URL_Matches_Append_To_header_value.length()); + } + } + //this.setAutoResizeMode(JTable.AUTO_RESIZE_OFF);//配合横向滚动条 + } + + + private void registerListeners() { + this.addMouseListener(new MouseAdapter() { + @Override + public void mouseClicked(MouseEvent e) { + if (e.getClickCount() == 2) { + ConfigTable target = (ConfigTable) e.getSource(); + int row = target.getSelectedRow(); + int column = target.getSelectedColumn(); + if (titles[column].equals("Enable")) { + boolean value = (boolean) getValueAt(row, column); + setValueAt(!value, row, column); + } + } + } + + @Override + public void mouseReleased(MouseEvent e) { + } + + @Override + public void mousePressed(MouseEvent e) { + //no need + } + }); + } + + + public void setupTypeColumn() { + //call this function must after table data loaded !!!! + JComboBox comboBox = new JComboBox<>(); + + String[] items = new ConfigEntry().listAllConfigType(); + for (String item : items) { + comboBox.addItem(item); + } + TableColumnModel model = this.getColumnModel(); + + int col = Arrays.asList(titles).indexOf("Type"); + DefaultCellEditor editor = new DefaultCellEditor(comboBox); + editor.setClickCountToStart(2); + model.getColumn(col).setCellEditor(editor); + } } diff --git a/src/config/ConfigTableModel.java b/src/config/ConfigTableModel.java index 03f1f86..ed7d87f 100644 --- a/src/config/ConfigTableModel.java +++ b/src/config/ConfigTableModel.java @@ -438,6 +438,4 @@ public void delRuleWithSameKey(ConfigEntry newrule) { } } } - - } \ No newline at end of file