pgLab/pglab/TablesTableModel.h

#ifndef TABLESTABLEMODEL_H
#define TABLESTABLEMODEL_H

#include "BaseTableModel.h"
#include "PgClass.h"
#include <memory>
#include <vector>

class PgClass;
class PgDatabaseCatalog;

class TablesTableModel: public QAbstractTableModel {
public:
	enum e_Columns : int {
		NameCol, ///< either table, ns.table or table (ns) depending on settings/filters
		NamespaceCol,
		OwnerCol,
		TablespaceCol,
		OptionsCol,
		//AclCol,
		colCount };

	TablesTableModel(QObject *parent);

	void setCatalog(std::shared_ptr<const PgDatabaseCatalog> cat);

	virtual void sort(int column, Qt::SortOrder order = Qt::AscendingOrder) override;
	void setSortOrder(int so);

	// Header:
	QVariant headerData(int section, Qt::Orientation orientation, int role) const override;

	// Basic functionality:
	int rowCount(const QModelIndex &parent) const override;
	int columnCount(const QModelIndex &parent) const override;

	virtual QVariant data(const QModelIndex &index, int role) const override;
	PgClass getTable(int row) const;
	Oid getTableOid(int row) const;

private:
	using t_Tables = std::vector<PgClass>;

	std::shared_ptr<const PgDatabaseCatalog> m_catalog;
	t_Tables m_tables;

	Oid getType(int column) const;
	QVariant getData(const QModelIndex &index) const;
	QString formatTableName(const PgClass &cls) const;
	void doSort(int so);
};

#endif // TABLESTABLEMODEL_H