ParamsTableModel.java

/*
 * Zed Attack Proxy (ZAP) and its related class files.
 * 
 * ZAP is an HTTP/HTTPS proxy for assessing web application security.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at 
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0 
 *   
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
 * See the License for the specific language governing permissions and 
 * limitations under the License. 
 */
package org.zaproxy.zap.extension.params;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Vector;

import javax.swing.table.AbstractTableModel;

import org.parosproxy.paros.Constant;

public class ParamsTableModel extends AbstractTableModel {

	private static final long serialVersionUID = 1L;
	private static final int COLUMN_COUNT = 7;
	
	private final Vector<String> columnNames;
	private List<HtmlParameterStats> paramStats;

	private int lastAddedRow;
	private int lastEditedRow;
	
	public ParamsTableModel() {
		super();
		columnNames = new Vector<>(COLUMN_COUNT);
		columnNames.add(Constant.messages.getString("params.table.header.type"));
		columnNames.add(Constant.messages.getString("params.table.header.name"));
		columnNames.add(Constant.messages.getString("params.table.header.used"));
		columnNames.add(Constant.messages.getString("params.table.header.numvals"));
		columnNames.add(Constant.messages.getString("params.table.header.pcchange"));
		columnNames.add(Constant.messages.getString("params.table.header.flags"));
		columnNames.add(Constant.messages.getString("params.table.header.values"));

		paramStats = Collections.synchronizedList(new ArrayList<HtmlParameterStats>());
		
		lastAddedRow = -1;
		lastEditedRow = -1;
	}

	@Override
	public int getColumnCount() {
		return COLUMN_COUNT;
	}

	@Override
	public int getRowCount() {
		return paramStats.size();
	}

	@Override
	public String getColumnName(int col) {
		return columnNames.get(col);
	}

	@Override
	public Object getValueAt(int row, int col) {
		Object obj = null;
		if (row >= paramStats.size()) {
			return null;
		}
		// TODO this doesnt work if resorted??
		HtmlParameterStats param = paramStats.get(row);
		switch (col) {
		case 0: obj = Constant.messages.getString("params.type." + param.getType().name()); break;
		case 1:	obj = param.getName(); break;
		case 2:	obj = param.getTimesUsed(); break;
		case 3:	obj = param.getValues().size(); break;
		case 4:	obj = getPercentChange(param); break;
		case 5: obj = param.getAllFlags(); break;
		case 6: obj = param.getValuesSummary(); break;
		}
		return obj;
	}
	
	private int getPercentChange(HtmlParameterStats param) {
		if (param.getValues().size() == 1) {
			return 0;
		}
		return (param.getValues().size() * 100 / param.getTimesUsed());
	}

	public HtmlParameterStats getHtmlParameterStatsAtRow(int row) {
		// TODO this doesnt work if resorted??
		return paramStats.get(row);
	}

	public void addHtmlParameterStats(HtmlParameterStats param) {
		lastAddedRow = -1;
		
		for (int i = 0; i < paramStats.size(); i++) {
			int cmp = param.compareTo(paramStats.get(i));
			if (cmp < 0) {
				paramStats.add(i, param);
				this.fireTableRowsInserted(i, i);

				lastAddedRow = i;
				return;
			} else if (cmp == 0) {
				// Already matches, so ignore
				lastAddedRow = i;
				return;
			}
		}

		paramStats.add(param);
		this.fireTableRowsInserted(paramStats.size()-1, paramStats.size()-1);
		
		lastAddedRow = paramStats.size()-1;
	}
	
	public int getLastAddedRow() {
		return lastAddedRow;
	}
	
	public int getLastEditedRow() {
		return lastEditedRow;
	}
	
	@Override
	public boolean isCellEditable(int row, int col) {
		return false;
	}

	@Override
	public Class<? extends Object> getColumnClass(int c) {
		switch (c) {
		case 0: return String.class;
		case 1:	return String.class;
		case 2:	return Integer.class;
		case 3:	return Integer.class;
		case 4:	return Integer.class;
		case 5: return String.class;
		case 6: return String.class;
		}
		return null;
	}

	public void removeAllElements() {
		paramStats.clear();
	}

}