#include "PgDatabaseCatalogue.h"
#include "PgTypeContainer.h"
#include "PgDatabaseContainer.h"
#include "PgAuthIdContainer.h"
#include "Pgsql_Connection.h"


QString getRoleNameFromOid(std::shared_ptr<const PgDatabaseCatalogue> cat, Oid oid)
{
	QString name;
	auto auth_ids = cat->authIds();
	if (auth_ids) {
		const PgAuthId& auth_id = auth_ids->getByOid(oid);
		if (auth_id.valid()) {
			name = auth_id.name;
		}
	}
	return name;
}

QString getRoleDisplayString(std::shared_ptr<const PgDatabaseCatalogue> cat, Oid oid)
{
	QString name = getRoleNameFromOid(cat, oid);
	return QString("%1 (%2)").arg(name).arg(oid);
}


PgDatabaseCatalogue::PgDatabaseCatalogue()
{
}

PgDatabaseCatalogue::~PgDatabaseCatalogue()
{
}

void PgDatabaseCatalogue::loadAll(Pgsql::Connection &conn)
{
	loadTypes(conn);
	loadDatabases(conn);
	loadAuthIds(conn);
}

void PgDatabaseCatalogue::loadInfo(Pgsql::Connection &conn)
{
	Pgsql::Result r = conn.query("SHOW server_version_num");
	if (r && r.resultStatus() == PGRES_TUPLES_OK)
		if (r.rows() == 1)
			m_serverVersion << r.get(0, 0);

	r = conn.query("SELECT version()");
	if (r && r.resultStatus() == PGRES_TUPLES_OK)
		if (r.rows() == 1)
			m_serverVersionString = r.get(0, 0).asQString();
}

void load(Pgsql::Connection &conn, IPgContainter &pg_cont)
{
	std::string q = pg_cont.getLoadQuery();
	Pgsql::Result result = conn.query(q.c_str());
	if (result && result.resultStatus() == PGRES_TUPLES_OK)
		pg_cont.load(result);
	else
		throw std::runtime_error("Query failed");

}

void PgDatabaseCatalogue::loadTypes(Pgsql::Connection &conn)
{
	if (!m_types)
		m_types = std::make_shared<PgTypeContainer>(shared_from_this());

	load(conn, *m_types);
}

void PgDatabaseCatalogue::loadDatabases(Pgsql::Connection &conn)
{
	if (!m_databases)
		m_databases = std::make_shared<PgDatabaseContainer>(shared_from_this());

	load(conn, *m_databases);
}

void PgDatabaseCatalogue::loadAuthIds(Pgsql::Connection &conn)
{
	if (!m_authIds)
		m_authIds = std::make_shared<PgAuthIdContainer>(shared_from_this());

	load(conn, *m_authIds);
}

const QString& PgDatabaseCatalogue::serverVersionString() const
{
	return m_serverVersionString;
}

int PgDatabaseCatalogue::serverVersion() const
{
	return m_serverVersion;
}

std::shared_ptr<const PgTypeContainer> PgDatabaseCatalogue::types() const
{
	return m_types;
}

std::shared_ptr<const PgDatabaseContainer> PgDatabaseCatalogue::databases() const
{
	return m_databases;
}

std::shared_ptr<const PgAuthIdContainer> PgDatabaseCatalogue::authIds() const
{
	return m_authIds;
}