/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */


package org.opensearch.jdbc;

import org.opensearch.jdbc.internal.JdbcWrapper;
import org.opensearch.jdbc.internal.results.Schema;

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

public class ResultSetMetaDataImpl implements ResultSetMetaData, JdbcWrapper {

    private ResultSetImpl resultSet;
    private Schema schema;

    public ResultSetMetaDataImpl(ResultSetImpl resultSet, Schema schema) {
        this.resultSet = resultSet;
        this.schema = schema;
    }

    @Override
    public int getColumnCount() throws SQLException {
        return schema.getNumberOfColumns();
    }

    @Override
    public boolean isAutoIncrement(int column) throws SQLException {
        checkAccessible(column);
        return false;
    }

    @Override
    public boolean isCaseSensitive(int column) throws SQLException {
        checkColumnIndex(column);
        return true;
    }

    @Override
    public boolean isSearchable(int column) throws SQLException {
        checkAccessible(column);
        return true;
    }

    @Override
    public boolean isCurrency(int column) throws SQLException {
        checkAccessible(column);
        return false;
    }

    @Override
    public int isNullable(int column) throws SQLException {
        checkAccessible(column);
        return columnNullableUnknown;
    }

    @Override
    public boolean isSigned(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getOpenSearchType().isSigned();
    }

    @Override
    public int getColumnDisplaySize(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getOpenSearchType().getDisplaySize();
    }

    @Override
    public String getColumnLabel(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getLabel();
    }

    @Override
    public String getColumnName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getName();
    }

    @Override
    public String getSchemaName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getTableSchemaName();
    }

    @Override
    public int getPrecision(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getPrecision();
    }

    @Override
    public int getScale(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getScale();
    }

    @Override
    public String getTableName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getTableName();
    }

    @Override
    public String getCatalogName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getColumnMetaData(column-1).getCatalogName();
    }

    @Override
    public int getColumnType(int column) throws SQLException {
        checkAccessible(column);
        return schema.getOpenSearchType(column-1).getJdbcType().getVendorTypeNumber();
    }

    @Override
    public String getColumnTypeName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getOpenSearchType(column-1).getJdbcType().getName();
    }

    @Override
    public boolean isReadOnly(int column) throws SQLException {
        checkAccessible(column);
        return true;
    }

    @Override
    public boolean isWritable(int column) throws SQLException {
        checkAccessible(column);
        return false;
    }

    @Override
    public boolean isDefinitelyWritable(int column) throws SQLException {
        checkAccessible(column);
        return false;
    }

    @Override
    public String getColumnClassName(int column) throws SQLException {
        checkAccessible(column);
        return schema.getOpenSearchType(column-1).getJavaClassName();
    }

    private void checkAccessible(int columnIndex) throws SQLException {
        checkOpen();
        checkColumnIndex(columnIndex);
    }

    private void checkOpen() throws SQLException {
        resultSet.checkOpen();
    }

    private void checkColumnIndex(int columnIndex) throws SQLException {
        resultSet.checkColumnIndex(columnIndex);
    }
}