/**

  * SQL-DK

  * Copyright © 2013 František Kučera (frantovo.cz)

  *

  * This program is free software: you can redistribute it and/or modify

  * it under the terms of the GNU General Public License as published by

  * the Free Software Foundation, either version 3 of the License, or

  * (at your option) any later version.

  *

  * This program is distributed in the hope that it will be useful,

  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

  * GNU General Public License for more details.

  *

  * You should have received a copy of the GNU General Public License

  * along with this program. If not, see <http://www.gnu.org/licenses/>.

  */

 package info.globalcode.sql.dk.formatting;

 import info.globalcode.sql.dk.Parameter;

 import java.util.ArrayList;

 import java.util.List;

 /**

  * Prefetches whole result set and computes column widths. Whole table is flushed at once in

  * {@linkplain #writeEndResultSet()}.

  *

  * Long values will not overflow the cells, but whole result set must be loaded into memory.

  *

  * @author Ing. František Kučera (frantovo.cz)

  */

 public class TabularPrefetchingFormatter extends TabularFormatter {

 	public static final String NAME = "tabular-prefetching"; // bash-completion:formatter

 	private String currentSql;

 	private List<? extends Parameter> currentParameters;

 	private ColumnsHeader currentHeader;

 	private List<Object[]> currentResultSet;

 	private Object[] currentRow;

 	private int currentColumnsCount;

 	private boolean prefetchDone = false;

 	public TabularPrefetchingFormatter(FormatterContext formatterContext) {

 		super(formatterContext);

 	}

 	@Override

 	protected int getCurrentColumnsCount() {

 		if (prefetchDone) {

 			return super.getCurrentColumnsCount();

 		} else {

 			return currentColumnsCount;

 		}

 	}

 	@Override

 	public void writeStartResultSet() {

 		currentResultSet = new ArrayList<>();

 	}

 	@Override

 	public void writeQuery(String sql) {

 		currentSql = sql;

 	}

 	@Override

 	public void writeParameters(List<? extends Parameter> parameters) {

 		currentParameters = parameters;

 	}

 	@Override

 	public void writeColumnsHeader(ColumnsHeader header) {

 		currentHeader = header;

 		initColumnWidths(header.getColumnCount());

 	}

 	@Override

 	public void writeStartRow() {

 		currentRow = new Object[currentHeader.getColumnCount()];

 		currentResultSet.add(currentRow);

 		currentColumnsCount = 0;

 	}

 	@Override

 	public void writeColumnValue(Object value) {

 		currentRow[currentColumnsCount] = value;

 		currentColumnsCount++;

 		String textRepresentation = toString(value);

 		/** TODO: count only printable characters (currently not an issue) */

 		updateColumnWidth(currentColumnsCount, textRepresentation.length());

 	}

 	@Override

 	public void writeEndRow() {

 		// do nothing

 	}

 	@Override

 	public void writeEndResultSet() {

 		prefetchDone = true;

 		super.writeStartResultSet();

 		super.writeQuery(currentSql);

 		super.writeParameters(currentParameters);

 		super.writeColumnsHeader(currentHeader);

 		for (Object[] row : currentResultSet) {

 			super.writeStartRow();

 			for (Object cell : row) {

 				super.writeColumnValue(cell);

 			}

 			super.writeEndRow();

 		}

 		currentColumnsCount = 0;

 		currentSql = null;

 		currentParameters = null;

 		currentHeader = null;

 		currentRow = null;

 		currentResultSet = null;

 		super.writeEndResultSet();

 		prefetchDone = false;

 	}

 }