/**

 * 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

 *

 * 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);

		}

	}

}