: QAbstractTableModel(parent)

, m_db(db)

, m_rowCount(0)

, m_iSortColumn(0)

, m_sSortOrder("ASC")

, m_chunkSize(chunkSize)

, m_valid(false)

{

{

m_sTable = table;

m_headers.clear();

QString rowid = "rowid";

if(m_db->getObjectByName(table).gettype() == "table")

rowid = sqlb::Table::parseSQL(m_db->getObjectByName(table).getsql()).rowidColumn();

m_headers.push_back(rowid);

m_headers.append(m_db->getTableFields(table));

m_mWhere.clear();

buildQuery();

QString r = s.trimmed();

while(r.endsWith(c))

r.chop(1);

return r;

{

// Feel free to fix all the bugs this probably contains or just replace this function entirely by

// a 'simple' regular expression. I know there're better ways to do this...

// This function removes any single line comments (starting with '--') from a given string. It does

// so by going through the string character by character and trying to keep track of wether we currently

// are in a string or identifier and only removing those parts starting with '--' which are in neither.

QChar lastChar = 0;

QList<QChar> stringChars;

for(int i=0; i < s.length(); ++i)

{

if(lastChar != '\\' && (s.at(i) == '\'' || s.at(i) == '"' || s.at(i) == '`'))

{

if(!stringChars.empty() && stringChars.last() == s.at(i))

stringChars.removeLast();

else if(!(!stringChars.empty() && (stringChars.last() != '\'' || stringChars.last() != '"')) || stringChars.empty())

stringChars.push_back(s.at(i));

} else if(stringChars.empty() && s.at(i) == '-' && lastChar == '-') {

int nextNL = s.indexOf('\n', i);

if(nextNL >= 0)

return removeComments(s.remove(i-1, nextNL - i + 2));

else

return s.left(i-1);

}

lastChar = s.at(i);

}

return s;

{

// clear

m_mWhere.clear();

if(!dontClearHeaders)

m_headers.clear();

if(!m_db->isOpen())

return;

m_sQuery = removeComments(sQuery).trimmed();

// do a count query to get the full row count in a fast manner

m_rowCount = getQueryRowCount();

if(m_rowCount == -1)

{

m_valid = false;

return;

}

// headers

if(!dontClearHeaders)

{

sqlite3_stmt* stmt;

QByteArray utf8Query = sQuery.toUtf8();

int status = sqlite3_prepare_v2(m_db->_db, utf8Query, utf8Query.size(), &stmt, NULL);

if(SQLITE_OK == status)

{

status = sqlite3_step(stmt);

int columns = sqlite3_data_count(stmt);

for(int i = 0; i < columns; ++i)

m_headers.append(QString::fromUtf8((const char*)sqlite3_column_name(stmt, i)));

}

sqlite3_finalize(stmt);

}

// now fetch the first entries

clearCache();

fetchData(0, m_chunkSize);

m_valid = true;

emit layoutChanged();

{

// Return -1 if there is an error

int retval = -1;

// Use a different approach of determining the row count when a EXPLAIN or a PRAGMA statement is used because a COUNT fails on these queries

if(m_sQuery.startsWith("EXPLAIN", Qt::CaseInsensitive) || m_sQuery.startsWith("PRAGMA", Qt::CaseInsensitive))

{

// So just execute the statement as it is and fetch all results counting the rows

sqlite3_stmt* stmt;

QByteArray utf8Query = m_sQuery.toUtf8();

if(sqlite3_prepare_v2(m_db->_db, utf8Query, utf8Query.size(), &stmt, NULL) == SQLITE_OK)

{

retval = 0;

while(sqlite3_step(stmt) == SQLITE_ROW)

retval++;

sqlite3_finalize(stmt);

// Return the results but also set the chunk size the number of rows to prevent the lazy population mechanism to kick in as using LIMIT

// fails on this kind of queries as well

m_chunkSize = retval;

return retval;

}

} else {

// If it is a normal query - hopefully starting with SELECT - just do a COUNT on it and return the results

QString sCountQuery = QString("SELECT COUNT(*) FROM (%1);").arg(rtrimChar(m_sQuery, ';'));

m_db->logSQL(sCountQuery, kLogMsg_App);

QByteArray utf8Query = sCountQuery.toUtf8();

sqlite3_stmt* stmt;

int status = sqlite3_prepare_v2(m_db->_db, utf8Query, utf8Query.size(), &stmt, NULL);

if(status == SQLITE_OK)

{

status = sqlite3_step(stmt);

if(status == SQLITE_ROW)

{

QString sCount = QString::fromUtf8((const char*)sqlite3_column_text(stmt, 0));

retval = sCount.toInt();

}

sqlite3_finalize(stmt);

} else {

qWarning() << "Count query failed: " << sCountQuery;

}

}

return retval;

{

if (role != Qt::DisplayRole)

return QVariant();

if (orientation == Qt::Horizontal)

{

// if we have a VIRTUAL table the model will not be valid, with no header data

if(section < m_headers.size())

return m_headers.at(section);

return QString("%1").arg(section + 1);

}

else

return QString("%1").arg(section + 1);

{

if (!index.isValid())

return QVariant();

if (index.row() >= m_rowCount)

return QVariant();

if(role == Qt::DisplayRole || role == Qt::EditRole)

{

// If this row is not in the cache yet get it first

while(index.row() >= m_data.size() && canFetchMore())

const_cast<SqliteTableModel*>(this)->fetchMore(); // Nothing evil to see here, move along

if(role == Qt::DisplayRole && isBinary(index))

return "(BLOB)";

else

return m_data.at(index.row()).at(index.column());

} else {

return QVariant();

}

{

if(index.isValid() && role == Qt::EditRole)

{

// Don't do anything if the data hasn't changed

if(m_data.at(index.row()).at(index.column()) == value)

return true;

if(m_db->updateRecord(m_sTable, m_headers.at(index.column()), m_data[index.row()].at(0).toInt(), value.toByteArray()))

{

// Only update the cache if this row has already been read, if not there's no need to do any changes to the cache

if(index.row() < m_data.size())

m_data[index.row()].replace(index.column(), value.toByteArray());

emit(dataChanged(index, index));

return true;

} else {

QMessageBox::warning(0, qApp->applicationName(), tr("Error changing data:\n%1").arg(m_db->lastErrorMessage));

return false;

}

}

return false;

{

int row = m_data.size();

fetchData(row, row + m_chunkSize);

{

if(!index.isValid())

return Qt::ItemIsEnabled;

Qt::ItemFlags ret = QAbstractTableModel::flags(index);

if(!isBinary(index))

ret |= Qt::ItemIsEditable;

return ret;

{

// Don't do anything when the sort order hasn't changed

if(m_iSortColumn == column && m_sSortOrder == (order == Qt::AscendingOrder ? "ASC" : "DESC"))

return;

// Save sort order

m_iSortColumn = column;

m_sSortOrder = (order == Qt::AscendingOrder ? "ASC" : "DESC");

// Set the new query (but only if a table has already been set

if(m_sTable != "")

buildQuery();

{

beginInsertRows(parent, row, row + count - 1);

QByteArrayList blank_data;

for(int i=0; i < m_headers.size(); ++i)

blank_data.push_back("");

for(int i=0; i < count; ++i)

{

m_data.insert(row, blank_data);

m_data[row].replace(0, QByteArray::number(m_db->addRecord(m_sTable)));

}

m_rowCount += count;

endInsertRows();

return true;

{

beginRemoveRows(parent, row, row + count - 1);

for(int i=count-1;i>=0;i--)

{

m_db->deleteRecord(m_sTable, m_data.at(row + i).at(0).toInt());

m_data.removeAt(row + i);

}

m_rowCount -= count;

endRemoveRows();

return true;

{

int currentsize = m_data.size();

QString sLimitQuery;

if(m_sQuery.startsWith("PRAGMA", Qt::CaseInsensitive) || m_sQuery.startsWith("EXPLAIN", Qt::CaseInsensitive))

{

sLimitQuery = m_sQuery;

} else {

// Remove trailing trailing semicolon

QString queryTemp = rtrimChar(m_sQuery, ';');

// If the query ends with a LIMIT statement take it as it is, if not append our own LIMIT part for lazy population

if(queryTemp.contains(QRegExp("LIMIT\\s+\\d+\\s*(,\\s*\\d+\\s*)?$", Qt::CaseInsensitive)))

sLimitQuery = queryTemp;

else

sLimitQuery = QString("%1 LIMIT %2, %3;").arg(queryTemp).arg(from).arg(to-from);

}

m_db->logSQL(sLimitQuery, kLogMsg_App);

QByteArray utf8Query = sLimitQuery.toUtf8();

sqlite3_stmt *stmt;

int status = sqlite3_prepare_v2(m_db->_db, utf8Query, utf8Query.size(), &stmt, NULL);

if(SQLITE_OK == status)

{

while(sqlite3_step(stmt) == SQLITE_ROW)

{

QByteArrayList rowdata;

for (int i = 0; i < m_headers.size(); ++i)

rowdata.append(QByteArray(static_cast<const char*>(sqlite3_column_blob(stmt, i)), sqlite3_column_bytes(stmt, i)));

m_data.push_back(rowdata);

}

}

sqlite3_finalize(stmt);

beginInsertRows(QModelIndex(), currentsize, m_data.size()-1);

endInsertRows();

{

QString where;

if(m_mWhere.size())

{

where = "WHERE 1=1";

for(QMap<int, QString>::const_iterator i=m_mWhere.constBegin();i!=m_mWhere.constEnd();++i)

where.append(QString(" AND `%1` %2").arg(m_headers.at(i.key())).arg(i.value()));

}

QString sql = QString("SELECT `%1`,* FROM `%2` %3 ORDER BY `%4` %5").arg(m_headers.at(0)).arg(m_sTable).arg(where).arg(m_headers.at(m_iSortColumn)).arg(m_sSortOrder);

setQuery(sql, true);

{

// Check for any special comparison operators at the beginning of the value string. If there are none default to LIKE.

QString op = "LIKE";

QString val;

bool numeric = false;

if(value.left(2) == ">=" || value.left(2) == "<=" || value.left(2) == "<>")

{

bool ok;

value.mid(2).toFloat(&ok);

if(ok)

{

op = value.left(2);

val = value.mid(2);

numeric = true;

}

} else if(value.left(1) == ">" || value.left(1) == "<") {

bool ok;

value.mid(1).toFloat(&ok);

if(ok)

{

op = value.left(1);

val = value.mid(1);

numeric = true;

}

} else {

if(value.left(1) == "=")

{

op = "=";

val = value.mid(1);

}

}

if(val.isEmpty())

val = value;

if(!numeric)

val = QString("'%1'").arg(val.replace("'", "''"));

// If the value was set to an empty string remove any filter for this column. Otherwise insert a new filter rule or replace the old one if there is already one

if(val == "''")

m_mWhere.remove(column);

else

m_mWhere.insert(column, QString("%1 %2").arg(op).arg(val));

// Build the new query

buildQuery();

{

beginRemoveRows(QModelIndex(), 0, m_data.size()-1);

m_data.clear();

endRemoveRows();

{

QByteArray val = m_data.at(index.row()).at(index.column()).left(1024);

return val.contains('\0'); // Cheap BLOB test here...