//////////////////////////////////////////////////////////////////////////
//
// pgAdmin III - PostgreSQL Tools
//
// Copyright (C) 2002 - 2016, The pgAdmin Development Team
// This software is released under the PostgreSQL Licence
//
// pgRole.h - PostgreSQL Role
//
//////////////////////////////////////////////////////////////////////////

#ifndef PGROLE_H
#define PGROLE_H

#include "pgServer.h"


#define PGROLE_ADMINOPTION      wxT("(*)")
#define PGROLE_ADMINOPTION_LEN  3


class pgRoleBaseFactory : public pgServerObjFactory
{
public:
	pgRoleBaseFactory(const wxChar *tn, const wxChar *ns, const wxChar *nls, wxImage *img = 0);
	virtual pgObject *CreateObjects(pgCollection *obj, ctlTree *browser, const wxString &restr);
};

class pgLoginRoleFactory : public pgRoleBaseFactory
{
public:
	pgLoginRoleFactory();
	virtual dlgProperty *CreateDialog(frmMain *frame, pgObject *node, pgObject *parent);
	virtual pgObject *CreateObjects(pgCollection *obj, ctlTree *browser, const wxString &restr);
	virtual pgCollection *CreateCollection(pgObject *obj);
};

class pgGroupRoleFactory : public pgRoleBaseFactory
{
public:
	pgGroupRoleFactory();
	virtual dlgProperty *CreateDialog(frmMain *frame, pgObject *node, pgObject *parent);
	virtual pgObject *CreateObjects(pgCollection *obj, ctlTree *browser, const wxString &restr);
	virtual pgCollection *CreateCollection(pgObject *obj);
};

extern pgLoginRoleFactory loginRoleFactory;
extern pgGroupRoleFactory groupRoleFactory;


// Class declarations
class pgRole : public pgServerObject
{
protected:
	pgRole(pgaFactory &factory, const wxString &newName = wxT(""));

public:
	int GetIconId();


	// Role Specific
	wxDateTime GetAccountExpires() const
	{
		return accountExpires;
	}
	void iSetAccountExpires(const wxDateTime &dt)
	{
		accountExpires = dt;
	}
	wxString GetPassword() const
	{
		return password;
	}
	void iSetPassword(const wxString &s)
	{
		password = s;
	}
	void iSetInherits(const bool b)
	{
		inherits = b;
	}
	bool GetInherits() const
	{
		return inherits;
	}
	void iSetCanLogin(const bool b)
	{
		canLogin = b;
	}
	bool GetCanLogin() const
	{
		return canLogin;
	}
	bool GetCreateDatabase() const
	{
		return createDatabase;
	}
	void iSetCreateDatabase(const bool b)
	{
		createDatabase = b;
	}
	bool GetCreateRole() const
	{
		return createRole;
	}
	void iSetCreateRole(const bool b)
	{
		createRole = b;
	}
	bool GetSuperuser() const
	{
		return superuser;
	}
	void iSetSuperuser(const bool b)
	{
		superuser = b;
	}
	bool GetUpdateCatalog() const
	{
		return updateCatalog;
	}
	void iSetUpdateCatalog(const bool b)
	{
		updateCatalog = b;
	}
	bool GetReplication() const
	{
		return replication;
	}
	void iSetReplication(const bool b)
	{
		replication = b;
	}
	wxString GetRolQueueName() const
	{
		return rolqueuename;
	}
	void iSetRolQueueName(const wxString &newVal)
	{
		rolqueuename = newVal;
	}
	long GetConnectionLimit() const
	{
		return connectionLimit;
	}
	void iSetConnectionLimit(long newVal)
	{
		connectionLimit = newVal;
	}
	void iSetIsValidInfinity(const bool b)
	{
		isValidInfinity = b;
	}
	wxArrayString &GetRolesIn()
	{
		return rolesIn;
	}
	wxArrayString &GetVariables()
	{
		return variables;
	}

	void ReassignDropOwnedTo(frmMain *form);

	// Tree object creation
	void ShowTreeDetail(ctlTree *browser, frmMain *form = 0, ctlListView *properties = 0, ctlSQLBox *sqlPane = 0);
	void ShowDependents(frmMain *form, ctlListView *referencedBy, const wxString &where);
	const wxArrayString GetDbList(ctlListView* referencedBy);
	wxString FillOwnedGrant(ctlTree* browser, const wxString& query);
	static wxString GetOptStr(wxString& rolename);
	// virtual methods
	wxString GetSql(ctlTree *browser);
	
	pgObject *Refresh(ctlTree *browser, const wxTreeItemId item);
	bool DropObject(wxFrame *frame, ctlTree *browser, bool cascaded);

	bool HasStats()
	{
		return false;
	}
	bool HasDepends()
	{
		return true;
	}
	bool HasReferences()
	{
		return true;
	}
	bool GetIsValidInfinity()
	{
		return isValidInfinity;
	}
private:
	wxString password;
	wxString rolqueuename;
	wxDateTime accountExpires;
	bool superuser, createDatabase, createRole, updateCatalog, inherits, canLogin, replication, isValidInfinity;
	long connectionLimit;
	wxArrayString rolesIn;
	wxArrayString variables;
};



class pgLoginRole : public pgRole
{
public:
	pgLoginRole(const wxString &newName = wxT(""));
	wxString GetTranslatedMessage(int kindOfMessage) const;
};


class pgGroupRole : public pgRole
{
public:
	pgGroupRole(const wxString &newName = wxT(""));
	wxString GetTranslatedMessage(int kindOfMessage) const;
};

class pgLoginRoleCollection : public pgServerObjCollection
{
public:
	pgLoginRoleCollection(pgaFactory *factory, pgServer *sv);
	wxString GetTranslatedMessage(int kindOfMessage) const;
};


class pgGroupRoleCollection : public pgServerObjCollection
{
public:
	pgGroupRoleCollection(pgaFactory *factory, pgServer *sv);
	wxString GetTranslatedMessage(int kindOfMessage) const;
};

class reassignDropOwnedFactory : public contextActionFactory
{
public:
	reassignDropOwnedFactory(menuFactoryList *list, wxMenu *mnu, ctlMenuToolbar *toolbar);
	wxWindow *StartDialog(frmMain *form, pgObject *obj);
	bool CheckEnable(pgObject *obj);
};


#endif
