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

using namespace Pgsql;

QueryResultModel::QueryResultModel(QObject *parent, std::shared_ptr<Pgsql::Result> r
			, std::shared_ptr<PgDatabaseCatalog> cat)
	: QAbstractTableModel(parent)
	, result(std::move(r))
	, catalog(cat)
{}


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();
//	return r;
//}

QVariant QueryResultModel::data(const QModelIndex &index, int role) const
{
//	static const QString null_str("null");
	QVariant v;

	const int col = index.column();
	const int rij = index.row();
	const Oid oid = result->type(col);
	if (role == Qt::DisplayRole) {
		if (result->null(col, rij)) {
			//v = nullptr;
		}
		else {
			auto value = result->val(col, rij);
			if (oid == bool_oid) {
				bool b = value;
				v = b;
			}
			else {
				QString s = value;
				if (s.length() > 256) {
					s.truncate(256);
				}
				v = s;
			}
		}
	}
	else if (role == Qt::UserRole) {
		v = oid;
	}

	return v;
}

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

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