#include "QueryResultModel.h"
#include "ResultTableModelUtil.h"
#include "Pgsql_declare.h"
#include <QBrush>
#include <QColor>

QueryResultModel::QueryResultModel(QObject *parent, std::shared_ptr<Pgsql::Result> r)
	: BaseTableModel(parent)
	, result(std::move(r))
{}


int QueryResultModel::rowCount(const QModelIndex &) const
{
	int r = result->rows();
	return r;
}

int QueryResultModel::columnCount(const QModelIndex &) const
{
	int r = result->cols();
	return r;
}

Oid QueryResultModel::getType(int column) const
{
	return result->type(column);
}

QVariant QueryResultModel::getData(const QModelIndex &index) const
{
	QVariant r;
	int rij = index.row();
	int col = index.column();
	if (result->null(col, rij)) {
		r = "null";
	}
	else {
//		Oid o = result->type(col);
		QString s(result->val(col, rij));
//		switch (o) {
//		case BOOLOID:
//			s = (s == "t") ? "TRUE" : "FALSE";
//			// intentional fall through
//		default:
			if (s.length() > 256) {
				s.truncate(256);
			}
			r = s;
//		}
	}
	return r;
}

QVariant QueryResultModel::data(const QModelIndex &index, int role) const
{
	if (role == Qt::ForegroundRole) {
		int rij = index.row();
		int col = index.column();
		if (result->null(col, rij)) {
			return QBrush(Qt::gray);
		}
	}
	return BaseTableModel::data(index, role);
}

QVariant QueryResultModel::headerData(int section, Qt::Orientation orientation, int role) const
{
	QVariant r;
	if (role == Qt::DisplayRole) {
		if (orientation == Qt::Horizontal) {
			r = QString(result->getColName(section));
		}
		else {
			r = QString::number(section + 1);
		}
	}
	return r;
}

//Qt::ItemFlags QueryResultModel::flags(const QModelIndex &) const
//{
//	return Qt::ItemIsUserCheckable;
//}