/******************************************************************************** Implementation of SQLite driver classes.**** Copyright (C) 1992-2003 Trolltech AS. All rights reserved.**** This file is part of the sql module of the Qt GUI Toolkit.** EDITIONS: FREE, ENTERPRISE**** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.******************************************************************************/#include "qsql_sqlite.h"#include <qdatetime.h>#include <qregexp.h>#include <qfile.h>#if (QT_VERSION-0 < 0x030000)#  include <qvector.h>#  if !defined Q_WS_WIN32#    include <unistd.h>#  endif#  include "../../../3rdparty/libraries/sqlite/sqlite.h"#else#  include <qptrvector.h>#  if !defined Q_WS_WIN32#    include <unistd.h>#  endif#  include <sqlite.h>#endiftypedef struct sqlite_vm sqlite_vm;#define QSQLITE_DRIVER_NAME "QSQLITE"static QSqlVariant::Type nameToType(const QString& typeName){    QString tName = typeName.upper();    if (tName.startsWith("INT"))        return QSqlVariant::Int;    if (tName.startsWith("FLOAT") || tName.startsWith("NUMERIC"))        return QSqlVariant::Double;    if (tName.startsWith("BOOL"))        return QSqlVariant::Bool;    // SQLite is typeless - consider everything else as string    return QSqlVariant::String;}class QSQLiteDriverPrivate{public:    QSQLiteDriverPrivate();    sqlite *access;    bool utf8;};QSQLiteDriverPrivate::QSQLiteDriverPrivate() : access(0){    utf8 = (qstrcmp(sqlite_encoding, "UTF-8") == 0);}class QSQLiteResultPrivate{public:    QSQLiteResultPrivate(QSQLiteResult *res);    void cleanup();    bool fetchNext(QtSqlCachedResult::RowCache *row);    bool isSelect();    // initializes the recordInfo and the cache    void init(const char **cnames, int numCols, QtSqlCachedResult::RowCache **row = 0);    void finalize();    QSQLiteResult* q;    sqlite *access;    // and we have too keep our own struct for the data (sqlite works via    // callback.    const char *currentTail;    sqlite_vm *currentMachine;    uint skippedStatus: 1; // the status of the fetchNext() that's skipped    QtSqlCachedResult::RowCache *skipRow;    uint utf8: 1;    QSqlRecordInfo rInf;};static const uint initial_cache_size = 128;QSQLiteResultPrivate::QSQLiteResultPrivate(QSQLiteResult* res) : q(res), access(0), currentTail(0),    currentMachine(0), skippedStatus(FALSE), skipRow(0), utf8(FALSE){}void QSQLiteResultPrivate::cleanup(){    finalize();    rInf.clear();    currentTail = 0;    currentMachine = 0;    skippedStatus = FALSE;    delete skipRow;    skipRow = 0;    q->setAt(QSql::BeforeFirst);    q->setActive(FALSE);    q->cleanup();}void QSQLiteResultPrivate::finalize(){    if (!currentMachine)        return;    char* err = 0;    int res = sqlite_finalize(currentMachine, &err);    if (err) {        q->setLastError(QSqlError("Unable to fetch results", err, QSqlError::Statement, res));        sqlite_freemem(err);    }    currentMachine = 0;}// called on first fetchvoid QSQLiteResultPrivate::init(const char **cnames, int numCols, QtSqlCachedResult::RowCache **row){    if (!cnames)        return;    rInf.clear();    if (numCols <= 0)        return;    for (int i = 0; i < numCols; ++i) {        const char* lastDot = strrchr(cnames[i], '.');        const char* fieldName = lastDot ? lastDot + 1 : cnames[i];        rInf.append(QSqlFieldInfo(fieldName, nameToType(cnames[i+numCols])));    }    // skip the first fetch    if (row && !*row) {	*row = new QtSqlCachedResult::RowCache(numCols);	skipRow = *row;    }}bool QSQLiteResultPrivate::fetchNext(QtSqlCachedResult::RowCache* row){    // may be caching.    const char **fvals;    const char **cnames;    int colNum;    int res;    int i;    if (skipRow) {	// already fetched	if (row)	    *row = *skipRow;	delete skipRow;	skipRow = 0;	return skippedStatus;    }    if (!currentMachine)	return FALSE;    // keep trying while busy, wish I could implement this better.    while ((res = sqlite_step(currentMachine, &colNum, &fvals, &cnames)) == SQLITE_BUSY) {	// sleep instead requesting result again immidiately.#if defined Q_WS_WIN32	Sleep(1000);#else	sleep(1);#endif    }    switch(res) {    case SQLITE_ROW:	// check to see if should fill out columns	if (rInf.isEmpty())	    // must be first call.	    init(cnames, colNum, &row);	if (!fvals)	    return FALSE;	if (!row)	    return TRUE;	for (i = 0; i < colNum; ++i)	    (*row)[i] = utf8 ? QString::fromUtf8(fvals[i]) : QString(fvals[i]);	return TRUE;    case SQLITE_DONE:	if (rInf.isEmpty())	    // must be first call.	    init(cnames, colNum);	q->setAt(QSql::AfterLast);	return FALSE;    case SQLITE_ERROR:    case SQLITE_MISUSE:    default:	// something wrong, don't get col info, but still return false	finalize(); // finalize to get the error message.	q->setAt(QSql::AfterLast);	return FALSE;    }    return FALSE;}QSQLiteResult::QSQLiteResult(const QSQLiteDriver* db): QtSqlCachedResult(db){    d = new QSQLiteResultPrivate(this);    d->access = db->d->access;    d->utf8 = db->d->utf8;}QSQLiteResult::~QSQLiteResult(){    d->cleanup();    delete d;}/*   Execute \a query.*/bool QSQLiteResult::reset (const QString& query){    // this is where we build a query.    if (!driver())        return FALSE;    if (!driver()-> isOpen() || driver()->isOpenError())        return FALSE;    d->cleanup();    // Um, ok.  callback based so.... pass private static function for this.    setSelect(FALSE);    char *err = 0;    int res = sqlite_compile(d->access,                                d->utf8 ? (const char*)query.utf8().data() : query.ascii(),                                &(d->currentTail),                                &(d->currentMachine),                                &err);    if (res != SQLITE_OK || err) {        setLastError(QSqlError("Unable to execute statement", err, QSqlError::Statement, res));        sqlite_freemem(err);    }    //if (*d->currentTail != '\000' then there is more sql to eval    if (!d->currentMachine) {	setActive(FALSE);	return FALSE;    }    // we have to fetch one row to find out about    // the structure of the result set    d->skippedStatus = d->fetchNext(0);    setSelect(!d->rInf.isEmpty());    if (isSelect())	init(d->rInf.count());    setActive(TRUE);    return TRUE;}bool QSQLiteResult::gotoNext(QtSqlCachedResult::RowCache* row){    return d->fetchNext(row);}int QSQLiteResult::size(){    return -1;}int QSQLiteResult::numRowsAffected(){    return sqlite_changes(d->access);}/////////////////////////////////////////////////////////QSQLiteDriver::QSQLiteDriver(QObject * parent, const char * name)    : QSqlDriver(parent, name ? name : QSQLITE_DRIVER_NAME){    d = new QSQLiteDriverPrivate();}QSQLiteDriver::QSQLiteDriver(sqlite *connection, QObject *parent, const char *name)    : QSqlDriver(parent, name ? name : QSQLITE_DRIVER_NAME){    d = new QSQLiteDriverPrivate();    d->access = connection;    setOpen(TRUE);    setOpenError(FALSE);}QSQLiteDriver::~QSQLiteDriver(){    delete d;}bool QSQLiteDriver::hasFeature(DriverFeature f) const{    switch (f) {    case Transactions:        return TRUE;#if (QT_VERSION-0 >= 0x030000)    case Unicode:        return d->utf8;#endif//   case BLOB:    default:        return FALSE;    }}/*   SQLite dbs have no user name, passwords, hosts or ports.   just file names.*/bool QSQLiteDriver::open(const QString & db, const QString &, const QString &, const QString &, int, const QString &){    if (isOpen())        close();    if (db.isEmpty())	return FALSE;    char* err = 0;    d->access = sqlite_open(QFile::encodeName(db), 0, &err);    if (err) {        setLastError(QSqlError("Error to open database", err, QSqlError::Connection));        sqlite_freemem(err);        err = 0;    }    if (d->access) {        setOpen(TRUE);	setOpenError(FALSE);        return TRUE;    }    setOpenError(TRUE);    return FALSE;}void QSQLiteDriver::close(){    if (isOpen()) {        sqlite_close(d->access);        d->access = 0;        setOpen(FALSE);        setOpenError(FALSE);    }}QSqlQuery QSQLiteDriver::createQuery() const{    return QSqlQuery(new QSQLiteResult(this));}bool QSQLiteDriver::beginTransaction(){    if (!isOpen() || isOpenError())        return FALSE;    char* err;    int res = sqlite_exec(d->access, "BEGIN", 0, this, &err);    if (res == SQLITE_OK)        return TRUE;    setLastError(QSqlError("Unable to begin transaction", err, QSqlError::Transaction, res));    sqlite_freemem(err);    return FALSE;}bool QSQLiteDriver::commitTransaction(){    if (!isOpen() || isOpenError())        return FALSE;    char* err;    int res = sqlite_exec(d->access, "COMMIT", 0, this, &err);    if (res == SQLITE_OK)        return TRUE;    setLastError(QSqlError("Unable to commit transaction", err, QSqlError::Transaction, res));    sqlite_freemem(err);    return FALSE;}bool QSQLiteDriver::rollbackTransaction(){    if (!isOpen() || isOpenError())        return FALSE;    char* err;    int res = sqlite_exec(d->access, "ROLLBACK", 0, this, &err);    if (res == SQLITE_OK)        return TRUE;    setLastError(QSqlError("Unable to rollback Transaction", err, QSqlError::Transaction, res));    sqlite_freemem(err);    return FALSE;}QStringList QSQLiteDriver::tables(const QString &typeName) const{    QStringList res;    if (!isOpen())        return res;    int type = typeName.toInt();    QSqlQuery q = createQuery();    q.setForwardOnly(TRUE);#if (QT_VERSION-0 >= 0x030000)    if ((type & (int)QSql::Tables) && (type & (int)QSql::Views))        q.exec("SELECT name FROM sqlite_master WHERE type='table' OR type='view'");    else if (typeName.isEmpty() || (type & (int)QSql::Tables))        q.exec("SELECT name FROM sqlite_master WHERE type='table'");    else if (type & (int)QSql::Views)        q.exec("SELECT name FROM sqlite_master WHERE type='view'");#else        q.exec("SELECT name FROM sqlite_master WHERE type='table' OR type='view'");#endif    if (q.isActive()) {        while(q.next())            res.append(q.value(0).toString());    }#if (QT_VERSION-0 >= 0x030000)    if (type & (int)QSql::SystemTables) {        // there are no internal tables beside this one:        res.append("sqlite_master");    }#endif    return res;}QSqlIndex QSQLiteDriver::primaryIndex(const QString &tblname) const{    QSqlRecordInfo rec(recordInfo(tblname)); // expensive :(    if (!isOpen())        return QSqlIndex();    QSqlQuery q = createQuery();    q.setForwardOnly(TRUE);    // finrst find a UNIQUE INDEX    q.exec("PRAGMA index_list('" + tblname + "');");    QString indexname;    while(q.next()) {	if (q.value(2).toInt()==1) {	    indexname = q.value(1).toString();	    break;	}    }    if (indexname.isEmpty())	return QSqlIndex();    q.exec("PRAGMA index_info('" + indexname + "');");    QSqlIndex index(indexname);    while(q.next()) {	QString name = q.value(2).toString();	QSqlVariant::Type type = QSqlVariant::Invalid;	if (rec.contains(name))	    type = rec.find(name).type();	index.append(QSqlField(name, type));    }    return index;}QSqlRecordInfo QSQLiteDriver::recordInfo(const QString &tbl) const{    if (!isOpen())        return QSqlRecordInfo();    QSqlQuery q = createQuery();    q.setForwardOnly(TRUE);    q.exec("SELECT * FROM " + tbl + " LIMIT 1");    return recordInfo(q);}QSqlRecord QSQLiteDriver::record(const QString &tblname) const{    if (!isOpen())        return QSqlRecord();    return recordInfo(tblname).toRecord();}QSqlRecord QSQLiteDriver::record(const QSqlQuery& query) const{    if (query.isActive() && query.driver() == this) {        QSQLiteResult* result = (QSQLiteResult*)query.result();        return result->d->rInf.toRecord();    }    return QSqlRecord();}QSqlRecordInfo QSQLiteDriver::recordInfo(const QSqlQuery& query) const{    if (query.isActive() && query.driver() == this) {        QSQLiteResult* result = (QSQLiteResult*)query.result();        return result->d->rInf;    }    return QSqlRecordInfo();}