192 lines
4.7 KiB
C++
192 lines
4.7 KiB
C++
#include "searchworker.h"
|
|
|
|
#include <QRegularExpression>
|
|
#include <QDebug>
|
|
#include <QSqlError>
|
|
SearchWorker::SearchWorker()
|
|
{
|
|
}
|
|
SearchWorker::SearchWorker(const QString &dbpath)
|
|
{
|
|
db = QSqlDatabase::addDatabase("QSQLITE");
|
|
db.setDatabaseName(dbpath);
|
|
if(!db.open())
|
|
{
|
|
qDebug() << "failed to open database";
|
|
}
|
|
queryContent = new QSqlQuery(db);
|
|
queryFile = new QSqlQuery(db);
|
|
queryFile->prepare("SELECT path, mtime FROM file WHERE path LIKE ? ORDER BY mtime DESC");
|
|
|
|
queryContent->prepare("SELECT file.path, content.page, file.mtime FROM file INNER JOIN content ON file.id = "
|
|
"content.fileid INNER JOIN content_fts ON content.id = content_fts.ROWID WHERE "
|
|
"content_fts.content MATCH ? ORDER By file.mtime DESC, content.page ASC");
|
|
}
|
|
|
|
QVector<SearchWorker::Command> SearchWorker::tokenize(QString expression)
|
|
{
|
|
// TODO: merge lonewords
|
|
QVector<Command> result;
|
|
QRegularExpression rx("((?<filtername>(\\.|\\w)+):(?<args>\\((?<innerargs>[^\\)]+)\\)|(\\w)+)|(?<boolean>AND|OR|!)|"
|
|
"(?<bracket>\\(|\\))|(?<loneword>\\w+))");
|
|
QRegularExpressionMatchIterator i = rx.globalMatch(expression);
|
|
bool wasbool = true;
|
|
while(i.hasNext())
|
|
{
|
|
QRegularExpressionMatch m = i.next();
|
|
QString boolean = m.captured("boolean");
|
|
QString filtername = m.captured("filtername");
|
|
QString bracket = m.captured("bracket");
|
|
QString loneword = m.captured("loneword");
|
|
if(boolean != "")
|
|
{
|
|
/* if(wasbool)
|
|
{
|
|
throw new std::runtime_error("Bool after Bool is invalid");
|
|
}*/
|
|
wasbool = true;
|
|
result.append(Command(boolean));
|
|
}
|
|
|
|
if(bracket != "")
|
|
{
|
|
if(!wasbool)
|
|
{
|
|
if(bracket == "(")
|
|
{
|
|
result.append(Command("AND"));
|
|
}
|
|
}
|
|
result.append(Command(bracket));
|
|
}
|
|
|
|
if(loneword != "")
|
|
{
|
|
if(!wasbool)
|
|
{
|
|
result.append(Command("AND"));
|
|
}
|
|
wasbool = false;
|
|
result.append(Command("contains", loneword));
|
|
}
|
|
if(filtername != "")
|
|
{
|
|
if(!wasbool)
|
|
{
|
|
result.append(Command("AND"));
|
|
}
|
|
wasbool = false;
|
|
QString value = m.captured("innerargs");
|
|
if(value == "")
|
|
value = m.captured("args");
|
|
result.append(Command(filtername, value));
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
QString SearchWorker::createSql(const SearchWorker::Command &cmd)
|
|
{
|
|
QString key = cmd.key;
|
|
QString value = cmd.value;
|
|
if(key == "AND" || key == "OR" || key == "(" || key == ")")
|
|
{
|
|
return " " + key + " ";
|
|
}
|
|
if(key == "!")
|
|
{
|
|
return " NOT ";
|
|
}
|
|
if(key == "path.starts")
|
|
{
|
|
return " file.path LIKE '" + value + "%' ";
|
|
}
|
|
if(key == "path.ends")
|
|
{
|
|
return " file.path LIKE '%" + value + "' ";
|
|
}
|
|
if(key == "path.contains" || key == "inpath")
|
|
{
|
|
return " file.path LIKE '%" + value + "%' ";
|
|
}
|
|
if(key == "page")
|
|
{
|
|
return " content.page = " + value;
|
|
}
|
|
if(key == "contains")
|
|
{
|
|
return " ( COALESCE( (SELECT 1 FROM content_fts WHERE content_fts.content MATCH '" + value +
|
|
"' AND content_fts.ROWID= content.id), 0 ) )";
|
|
}
|
|
qDebug() << "NOHIT" << key;
|
|
// TODO: exception?
|
|
return "NOTHING";
|
|
}
|
|
|
|
QString SearchWorker::makeSql(const QVector<SearchWorker::Command> &tokens)
|
|
{
|
|
QString result;
|
|
for(const Command &c : tokens)
|
|
{
|
|
result += createSql(c);
|
|
}
|
|
return result;
|
|
}
|
|
void SearchWorker::searchForFile(const QString &query)
|
|
{
|
|
QVector<SearchResult> results;
|
|
queryFile->addBindValue("%" + query + "%");
|
|
queryFile->exec();
|
|
while(queryFile->next())
|
|
{
|
|
SearchResult result;
|
|
result.page = 0;
|
|
result.path = queryFile->value(0).toString();
|
|
result.mtime = queryFile->value(1).toUInt();
|
|
|
|
results.append(result);
|
|
}
|
|
emit searchResultsReady(results);
|
|
}
|
|
void SearchWorker::searchForContent(const QString &query)
|
|
{
|
|
QVector<SearchResult> results;
|
|
queryContent->addBindValue(query);
|
|
queryContent->exec();
|
|
while(queryContent->next())
|
|
{
|
|
SearchResult result;
|
|
|
|
result.path = queryContent->value(0).toString();
|
|
result.page = queryContent->value(1).toUInt();
|
|
result.mtime = queryContent->value(2).toUInt();
|
|
results.append(result);
|
|
}
|
|
emit searchResultsReady(results);
|
|
}
|
|
|
|
void SearchWorker::search(const QString &query)
|
|
{
|
|
QSqlQuery dbquery(db);
|
|
QVector<SearchResult> results;
|
|
QString whereSql = makeSql(tokenize(query));
|
|
QString prep =
|
|
"SELECT file.path AS path, content.page AS page, file.mtime AS mtime FROM file INNER JOIN content ON file.id = "
|
|
"content.fileid INNER JOIN content_fts ON content.id = content_fts.ROWID WHERE 1=1 AND " +
|
|
whereSql + " ORDER By file.mtime DESC, content.page ASC";
|
|
dbquery.prepare(prep);
|
|
dbquery.exec();
|
|
qDebug() << "prepped: " << prep;
|
|
qDebug() << dbquery.lastError();
|
|
while(dbquery.next())
|
|
{
|
|
SearchResult result;
|
|
|
|
result.path = dbquery.value("path").toString();
|
|
result.page = dbquery.value("page").toUInt();
|
|
result.mtime = dbquery.value("mtime").toUInt();
|
|
results.append(result);
|
|
}
|
|
emit searchResultsReady(results);
|
|
}
|