pgLab/pglab/TablesTableModel.h

#ifndef TABLESTABLEMODEL_H
#define TABLESTABLEMODEL_H

#include "BaseTableModel.h"
#include "NamespaceFilter.h"
#include "catalog/PgClass.h"
#include <memory>
#include <vector>

class PgClass;
class PgDatabaseCatalog;



class TablesTableModel: public QAbstractTableModel {
public:
    using RowItem = PgClass;
	enum e_Columns : int {
		NameCol, ///< either table, ns.table or table (ns) depending on settings/filters
		NamespaceCol,
		KindCol,
		OwnerCol,
		TablespaceCol,
		OptionsCol,
		AclCol,
        CommentCol,
        TotalSize,
        TableSize,
        IndexSize,
        ToastSize,

		colCount };

	TablesTableModel(QObject *parent);

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

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

	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;
    RowItem rowItem(int row) const
    {
        return getTable(row);
    }
	Oid getTableOid(int row) const;

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

	std::shared_ptr<const PgDatabaseCatalog> m_catalog;
	NamespaceFilter m_namespaceFilter = NamespaceFilter::User;
	t_Tables m_tables;
	QMetaObject::Connection refreshConnection;


	Oid getType(int column) const;
	QVariant getData(const QModelIndex &index) const;
private slots:
	void refresh();

};

#endif // TABLESTABLEMODEL_H