package test.personen;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;

import javax.swing.event.TableModelListener;
import javax.swing.table.TableModel;

public class ResultSetTableModel implements TableModel {
    ResultSet data;
    ResultSetMetaData title;
    int colCount, rowCount;

    ResultSetTableModel (ResultSet rs) throws SQLException {
        data = rs;
        title = rs.getMetaData();

        data.last();
        rowCount = data.getRow();
        colCount = title.getColumnCount();
    }

    public void close() {
        try
        {
            data.getStatement().close();
        }
        catch (SQLException e)
        {};
    }

    @Override
    public int getColumnCount() {
        return colCount;
    }

    @Override
    public int getRowCount() {
        return rowCount;
    }

    @Override
    public String getColumnName(int column) {
        try
        {
            return title.getColumnLabel(column + 1);
        }
        catch (SQLException e)
        {
            return e.toString();
        }
    }

    @Override
    public Class<?> getColumnClass(int column) {
        return String.class;
    }

    @Override
    public Object getValueAt(int row, int column) {
        try
        {
            data.absolute(row + 1); // Go to the specified row
            Object o = data.getObject(column + 1); // Get value of the column
            if (o == null)
            {
                return null;
            }
            else
            {
                return o.toString();
            }
        }
        catch (SQLException e)
        {
            return e.toString();
        }
    }

    @Override
    public boolean isCellEditable(int row, int column) {
        return false;
    }

    @Override
    public void addTableModelListener(TableModelListener l) {}

    @Override
    public void removeTableModelListener(TableModelListener l) {}

    @Override
    public void setValueAt(Object aValue, int rowIndex, int columnIndex) {}

}