Mercurialsql-dk / file revision
summary | shortlog | changelog | graph | tags | bookmarks | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
java/sql-dk/src/info/globalcode/sql/dk/formatting/TabularFormatter.java
author František Kučera <franta-hg@frantovo.cz>
Mon, 23 Dec 2013 18:27:57 +0100
branchv_0
changeset 43 2813d3409afd
parent 40 a9db7fb3ce65
child 76 fe23cea7542f
permissions -rw-r--r--
TabularFormatter: also booleans will be aligned to the right 
     1 /**

     2  * SQL-DK

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

     4  *

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

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

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

     8  * (at your option) any later version.

     9  *

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

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

    12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

    13  * GNU General Public License for more details.

    14  *

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

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

    17  */

    18 package info.globalcode.sql.dk.formatting;

    19 

    20 import info.globalcode.sql.dk.ColorfulPrintWriter;

    21 import static info.globalcode.sql.dk.ColorfulPrintWriter.*;

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

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

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

    25 

    26 /**

    27  *

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

    29  */

    30 public class TabularFormatter extends AbstractFormatter {

    31 

    32 	public static final String NAME = "tabular";

    33 	private static final String HEADER_TYPE_PREFIX = " (";

    34 	private static final String HEADER_TYPE_SUFFIX = ")";

    35 	private ColorfulPrintWriter out;

    36 	private boolean firstResult = true;

    37 	private int[] columnWidth;

    38 

    39 	public TabularFormatter(FormatterContext formatterContext) {

    40 		super(formatterContext);

    41 		out = new ColorfulPrintWriter(formatterContext.getOutputStream());

    42 	}

    43 

    44 	@Override

    45 	public void writeStartResultSet() {

    46 		super.writeStartResultSet();

    47 		printResultSeparator();

    48 	}

    49 

    50 	@Override

    51 	public void writeColumnsHeader(ColumnsHeader header) {

    52 		super.writeColumnsHeader(header);

    53 

    54 		columnWidth = new int[header.getColumnCount()];

    55 

    56 		printTableIndent();

    57 		printTableBorder("╭");

    58 		for (ColumnDescriptor cd : header.getColumnDescriptors()) {

    59 			setColumnWidth(cd.getColumnNumber(), cd.getLabel().length() + cd.getTypeName().length() + HEADER_TYPE_PREFIX.length() + HEADER_TYPE_SUFFIX.length());

    60 			if (!cd.isFirstColumn()) {

    61 				printTableBorder("┬");

    62 			}

    63 			printTableBorder(repeat('─', getColumnWidth(cd.getColumnNumber()) + 2));

    64 		}

    65 		printTableBorder("╮");

    66 		out.println();

    67 

    68 		for (ColumnDescriptor cd : header.getColumnDescriptors()) {

    69 			if (cd.isFirstColumn()) {

    70 				printTableIndent();

    71 				printTableBorder("│ ");

    72 			} else {

    73 				printTableBorder(" │ ");

    74 			}

    75 			out.print(TerminalStyle.Bright, cd.getLabel());

    76 			out.print(HEADER_TYPE_PREFIX);

    77 			out.print(cd.getTypeName());

    78 			out.print(HEADER_TYPE_SUFFIX);

    79 			if (cd.isLastColumn()) {

    80 				printTableBorder(" │");

    81 			}

    82 

    83 			setColumnWidth(cd.getColumnNumber(), cd.getLabel().length() + cd.getTypeName().length() + HEADER_TYPE_PREFIX.length() + HEADER_TYPE_SUFFIX.length());

    84 		}

    85 		out.println();

    86 

    87 		printTableIndent();

    88 		printTableBorder("├");

    89 		for (int i = 1; i <= header.getColumnCount(); i++) {

    90 			if (i > 1) {

    91 				printTableBorder("┼");

    92 			}

    93 			printTableBorder(repeat('─', getColumnWidth(i) + 2));

    94 		}

    95 		printTableBorder("┤");

    96 		out.println();

    97 

    98 		out.flush();

    99 	}

   100 

   101 	@Override

   102 	public void writeColumnValue(Object value) {

   103 		super.writeColumnValue(value);

   104 

   105 		if (isCurrentColumnFirst()) {

   106 			printTableIndent();

   107 			printTableBorder("│ ");

   108 		} else {

   109 			printTableBorder(" │ ");

   110 		}

   111 

   112 		out.print(TerminalColor.Cyan, toString(value));

   113 

   114 		if (isCurrentColumnLast()) {

   115 			printTableBorder(" │");

   116 		}

   117 

   118 	}

   119 

   120 	private int getColumnWidth(int columnNumber) {

   121 		return columnWidth[columnNumber - 1];

   122 	}

   123 

   124 	private void setColumnWidth(int columnNumber, int width) {

   125 		columnWidth[columnNumber - 1] = width;

   126 	}

   127 

   128 	private void updateColumnWidth(int columnNumber, int width) {

   129 		int oldWidth = getColumnWidth(columnNumber);

   130 		setColumnWidth(columnNumber, Math.max(width, oldWidth));

   131 

   132 	}

   133 

   134 	@Override

   135 	protected String toString(Object value) {

   136 		final int width = getColumnWidth(getCurrentColumnsCount());

   137 		if (value instanceof Number || value instanceof Boolean) {

   138 			return lpad(super.toString(value), width);

   139 		} else {

   140 			return rpad(super.toString(value), width);

   141 		}

   142 		// ?	value = (boolean) value ? "✔" : "✗";

   143 	}

   144 

   145 	@Override

   146 	public void writeEndRow() {

   147 		super.writeEndRow();

   148 		out.println();

   149 		out.flush();

   150 	}

   151 

   152 	@Override

   153 	public void writeEndResultSet() {

   154 		int columnCount = getCurrentColumnsHeader().getColumnCount();

   155 		super.writeEndResultSet();

   156 

   157 		printTableIndent();

   158 		printTableBorder("╰");

   159 		for (int i = 1; i <= columnCount; i++) {

   160 			if (i > 1) {

   161 				printTableBorder("┴");

   162 			}

   163 			printTableBorder(repeat('─', getColumnWidth(i) + 2));

   164 		}

   165 		printTableBorder("╯");

   166 		out.println();

   167 

   168 

   169 		out.print(TerminalColor.Yellow, "Record count: ");

   170 		out.println(getCurrentRowCount());

   171 		out.bell();

   172 		out.flush();

   173 	}

   174 

   175 	@Override

   176 	public void writeStartUpdatesResult() {

   177 		super.writeStartUpdatesResult();

   178 		printResultSeparator();

   179 	}

   180 

   181 	@Override

   182 	public void writeUpdatedRowsCount(int updatedRowsCount) {

   183 		super.writeUpdatedRowsCount(updatedRowsCount);

   184 		out.print(TerminalColor.Red, "Updated records: ");

   185 		out.println(updatedRowsCount);

   186 		out.bell();

   187 		out.flush();

   188 	}

   189 

   190 	@Override

   191 	public void writeEndDatabase() {

   192 		super.writeEndDatabase();

   193 		out.flush();

   194 	}

   195 

   196 	private void printResultSeparator() {

   197 		if (firstResult) {

   198 			firstResult = false;

   199 		} else {

   200 			out.println();

   201 		}

   202 	}

   203 

   204 	private void printTableBorder(String border) {

   205 		out.print(TerminalColor.Green, border);

   206 	}

   207 

   208 	private void printTableIndent() {

   209 		out.print(" ");

   210 	}

   211 }
sql-dk