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


QString getRoleNameFromOid(const PgDatabaseCatalogue *cat, Oid oid)
{
	QString name;
	const PgAuthIdContainer *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;
}

PgDatabaseCatalogue::PgDatabaseCatalogue()
{
}

PgDatabaseCatalogue::~PgDatabaseCatalogue()
{
	delete m_types;
}

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 PgDatabaseCatalogue::loadTypes(Pgsql::Connection &conn)
{
	if (m_types == nullptr)
		m_types = new PgTypeContainer(this);

	std::string q = m_types->getLoadQuery();
	Pgsql::Result result = conn.query(q.c_str());
	if (result && result.resultStatus() == PGRES_TUPLES_OK)
		m_types->load(result);
	else
		throw std::runtime_error("Query failed");

}

void PgDatabaseCatalogue::loadDatabases(Pgsql::Connection &conn)
{
	if (m_databases == nullptr)
		m_databases = new PgDatabaseContainer(this);


	std::string q = m_databases->getLoadQuery();
	Pgsql::Result result = conn.query(q.c_str());
	if (result && result.resultStatus() == PGRES_TUPLES_OK)
		m_databases->load(result);
	else
		throw std::runtime_error("Query failed");
}

void PgDatabaseCatalogue::loadAuthIds(Pgsql::Connection &conn)
{
	if (m_authIds == nullptr)
		m_authIds = new PgAuthIdContainer(this);

	std::string q = m_authIds->getLoadQuery();
	Pgsql::Result result = conn.query(q.c_str());
	if (result && result.resultStatus() == PGRES_TUPLES_OK)
		m_authIds->load(result);
	else
		throw std::runtime_error("Query failed");
}

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

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

const PgTypeContainer* PgDatabaseCatalogue::types() const
{
	return m_types;
}

const PgDatabaseContainer *PgDatabaseCatalogue::databases() const
{
	return m_databases;
}

const PgAuthIdContainer *PgDatabaseCatalogue::authIds() const
{
	return m_authIds;
}