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

using namespace Pgsql;

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);
		auto value = result->get(col, rij);
		switch (o) {
		case BOOLOID:
			r = bool(value); //s = (s == "t") ? "TRUE" : "FALSE";
			break;
		default: {
			QString s = value;
			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;
//}