/**
 * @package pgObjects
 */

/****************************************************************************
**
** Definition of PostgreSQL driver classes
**
** Created : 001103
**
** Copyright (C) 1992-2000 Trolltech AS.  All rights reserved.
**
** This file is part of the sql module of the Qt GUI Toolkit.
**
** This file may be distributed under the terms of the Q Public License
** as defined by Trolltech AS of Norway and appearing in the file
** LICENSE.QPL included in the packaging of this file.
**
** This file may be distributed and/or modified under the terms of the
** GNU General Public License version 2 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPL included in the
** packaging of this file.
**
** Licensees holding valid Qt Enterprise Edition licenses may use this
** file in accordance with the Qt Commercial License Agreement provided
** with the Software.
**
** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
**
** See http://www.trolltech.com/pricing.html or email sales@trolltech.com for
**   information about Qt Commercial License Agreements.
** See http://www.trolltech.com/qpl/ for QPL licensing information.
** See http://www.trolltech.com/gpl/ for GPL licensing information.
**
** Contact info@trolltech.com if any conditions of this licensing are
** not clear to you.
**
**********************************************************************/

#ifndef QPGSQL_DRIVER_H
#define QPGSQL_DRIVER_H

#include <qobject.h>
#include <qsignal.h>
#include <qvariant.h>
#include <qsqlresult.h>
#include <qsqlfield.h>
#include <qsqldriver.h>
#include <libpq-fe.h>

#ifdef QT_PLUGIN
#define Q_EXPORT_SQLDRIVER_PGSQL
#else
#define Q_EXPORT_SQLDRIVER_PGSQL Q_EXPORT
#endif

class QPGSQLPrivate;
class QPGSQLDriver;
class QSqlRecordInfo;
/*================================QPGSQLPrivate===============================================*/
class QPGSQLPrivate
{
public:
	QPGSQLPrivate() {connection=NULL; result=NULL; isUtf8=FALSE; }
	PGconn*		connection;
	PGresult*	result;
	bool        isUtf8;
};

/*================================QPGSQLResult================================================*/
class QPGSQLResult : public QSqlResult
{
	friend class QPGSQLDriver;
public:
	QPGSQLResult( const QPGSQLDriver* db, const QPGSQLPrivate* p );
	~QPGSQLResult();
	PGresult* 	result() 			{return d->result;};
	int 		lastOid()			{return PQoidValue(d->result);};
	int			fieldCount()		{return PQnfields(d->result);};
	QString 	fieldName(int nCol) {return PQfname(d->result,nCol);};
	int 		fieldColumn(const QString fieldName);
	QString 	escapeString(const QString& strData);
	int			numRowsAffected();

	bool			beginTransaction();
	bool			commitTransaction();
	bool			rollbackTransaction();


	QPGSQLDriver* pgDriver();
	//Blob stuff
	int createBlob();
	QByteArray readBlob(int oid);
	bool writeBlob(int oid, QByteArray blobData);

protected:
	void		cleanup();
	bool		fetch( int i );
	bool		fetchFirst();
	bool		fetchLast();
	QVariant	data( int i );
	QVariant	data( const QString fieldName );
	bool		isNull( int field );
	bool		reset ( const QString& query );
	int			size();
private:
	int			currentSize;
	QPGSQLPrivate*	d;
};


class Q_EXPORT_SQLDRIVER_PGSQL QPGSQLDriver : public QSqlDriver
{
public:
	enum Protocol {
		Version6 = 6,
		Version7 = 7,
		Version71 = 8,
		Version72 = 9,
		Version73 = 10
	};

	QPGSQLDriver( QObject * parent=0, const char * name=0 );
	~QPGSQLDriver();
	bool			hasFeature( DriverFeature f ) const;
	bool				open( const QString & db,
						const QString & user = QString::null,
						const QString & password = QString::null,
						const QString & host = QString::null,
						int port = -1 );
	void			close();
	QSqlQuery		createQuery() const;
	QStringList		tables( const QString& user ) const;
	QSqlIndex		primaryIndex( const QString& tablename ) const;
	QSqlRecord		record( const QString& tablename ) const;
	QSqlRecord		record( const QSqlQuery& query ) const;
	QSqlRecordInfo	recordInfo( const QString& tablename ) const;
	QSqlRecordInfo	recordInfo( const QSqlQuery& query ) const;

	QString 		version();
	PGconn*			connection();
	int				backendPID();

	void 			addNoticeMonitor(QObject* mon, const char* member);
	void 			removeNoticeMonitor(QObject* mon);
	void 			sendNotice(const QString notType, const QString msg);
	void 			checkSqlNotify();

	Protocol        protocol() const { return pro; }
	QString			formatValue( const QSqlField* field, bool trimStrings );
	QString 		escapeString(const QString& strData);

	bool			beginTransaction();
	bool			commitTransaction();
	bool			rollbackTransaction();

	//Blob stuff
	int createBlob();
	QByteArray readBlob(int oid);
	bool writeBlob(int oid, QByteArray blobData);

protected:

private:
	QSignal*		FSqlNotify;
	void			init();
	Protocol        pro;
	QPGSQLPrivate*	d;
};

#endif //QPGSQL_DRIVER_H