/**
 * 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, 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 java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

/**
 *
 * @author Ing. František Kučera (frantovo.cz)
 */
public class ColumnsHeader {

	private final ResultSetMetaData metaData;

	public ColumnsHeader(ResultSetMetaData metaData) {
		this.metaData = metaData;
	}

	public int getColumnCount() {
		try {
			return metaData.getColumnCount();
		} catch (SQLException e) {
			throw new IllegalStateException("Error during getting column count.", e);
		}
	}

	public List<ColumnDescriptor> getColumnDescriptors() {
		try {
			int count = metaData.getColumnCount();
			List<ColumnDescriptor> list = new ArrayList<>(count);

			for (int i = 1; i <= count; i++) {
				ColumnDescriptor cd = new ColumnDescriptor();

				cd.setFirstColumn(i == 1);
				cd.setLastColumn(i == count);
				cd.setColumnNumber(i);

				cd.setLabel(metaData.getColumnLabel(i));
				cd.setName(metaData.getColumnName(i));
				cd.setType(metaData.getColumnType(i));
				cd.setTypeName(metaData.getColumnTypeName(i));
				cd.setTableName(metaData.getTableName(i));
				list.add(cd);
			}

			return list;
		} catch (SQLException e) {
			throw new IllegalStateException("Error during building column descriptors.", e);
		}
	}
}