/**

  * SQL-DK

  * Copyright © 2014 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, version 3 of the License.

  *

  * 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.ColorfulPrintWriter.TerminalColor;

 import java.util.ArrayList;

 import java.util.List;

 import static info.globalcode.sql.dk.Functions.lpad;

 import static info.globalcode.sql.dk.Functions.rpad;

 import static info.globalcode.sql.dk.Functions.repeat;

 /**

  * Longer values are line-wrapped – the cell then contains multiple lines. Marks are added to

  * signalize forced line ends (not present in original data).

  *

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

  */

 public class TabularWrappingFormatter extends TabularFormatter {

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

 	private List<String[]> currentRow;

 	public TabularWrappingFormatter(FormatterContext formatterContext) {

 		super(formatterContext);

 	}

 	@Override

 	public void writeStartResultSet(ColumnsHeader header) {

 		super.writeStartResultSet(header);

 		currentRow = new ArrayList<>(header.getColumnCount());

 	}

 	@Override

 	protected void writeColumnValueInternal(Object value) {

 		boolean rightAlign = value instanceof Number || value instanceof Boolean;

 		String valueString = String.valueOf(value);

 		int columnWidth = getColumnWidth(getCurrentColumnsCount()) - 1;  // -1 = space for new line symbol

 		currentRow.add(wrapLines(valueString, columnWidth, rightAlign));

 	}

 	@Override

 	public void writeEndRow() {

 		super.writeEndRow();

 		int wrappedLine = 0;

 		boolean hasMoreWrappedLines;

 		do {

 			hasMoreWrappedLines = false;

 			for (int i = 0; i < currentRow.size(); i++) {

 				if (i == 0) {

 					printTableIndent();

 					printTableBorder("│ ");

 				} else {

 					printTableBorder(" │ ");

 				}

 				String[] columnArray = currentRow.get(i);

 				if (wrappedLine < columnArray.length) {

 					printValueWithWhitespaceReplaced(columnArray[wrappedLine]);

 					if (wrappedLine < columnArray.length - 1) {

 						out.print(TerminalColor.Red, "↩");

 						hasMoreWrappedLines = true;

 					} else {

 						out.print(" ");

 					}

 				} else {

 					out.print(repeat(' ', getColumnWidth(i + 1)));

 				}

 				if (i == (currentRow.size() - 1)) {

 					printTableBorder(" │");

 				}

 			}

 			out.println();

 			out.flush();

 			wrappedLine++;

 		} while (hasMoreWrappedLines);

 		currentRow.clear();

 	}

 	@Override

 	public void writeEndRowInternal() {

 		// already done – wrapped row ends

 	}

 	private static String[] wrapLines(String s, int width, boolean rightAlign) {

 		String[] array = new String[(s.length() - 1) / width + 1];

 		for (int i = 0; i < array.length; i++) {

 			if (i == array.length - 1) {

 				String part = s.substring(i * width, s.length());

 				array[i] = rightAlign ? lpad(part, width) : rpad(part, width);

 			} else {

 				array[i] = s.substring(i * width, (i + 1) * width);

 			}

 		}

 		return array;

 	}

 }