Clementine-audio-player-Mac.../src/smartplaylists/search.cpp

103 lines
2.6 KiB
C++

/* This file is part of Clementine.
Clementine is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Clementine is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Clementine. If not, see <http://www.gnu.org/licenses/>.
*/
#include "search.h"
#include "core/song.h"
#include <QStringList>
namespace smart_playlists {
Search::Search() {
Reset();
}
Search::Search(
SearchType type, TermList terms, SortType sort_type,
SearchTerm::Field sort_field, int limit)
: search_type_(type),
terms_(terms),
sort_type_(sort_type),
sort_field_(sort_field),
limit_(limit)
{
}
void Search::Reset() {
search_type_ = Type_And;
terms_.clear();
sort_type_ = Sort_Random;
sort_field_ = SearchTerm::Field_Title;
limit_ = -1;
}
QString Search::ToSql(const QString& songs_table) const {
QString sql = "SELECT ROWID," + Song::kColumnSpec + " FROM " + songs_table;
// Add search terms
QStringList term_sql;
foreach (const SearchTerm& term, terms_) {
term_sql += term.ToSql();
}
if (!terms_.isEmpty() && search_type_ != Type_All) {
QString boolean_op = search_type_ == Type_And ? " AND " : " OR ";
sql += " WHERE " + term_sql.join(boolean_op);
}
// Add sort by
if (sort_type_ == Sort_Random) {
sql += " ORDER BY random()";
} else {
sql += " ORDER BY " + SearchTerm::FieldColumnName(sort_field_)
+ (sort_type_ == Sort_FieldAsc ? " ASC" : " DESC");
}
// Add limit
if (limit_ != -1) {
sql += " LIMIT " + QString::number(limit_);
}
return sql;
}
bool Search::is_valid() const {
if (search_type_ == Type_All)
return true;
return !terms_.isEmpty();
}
} // namespace
QDataStream& operator <<(QDataStream& s, const smart_playlists::Search& search) {
s << search.terms_;
s << quint8(search.sort_type_);
s << quint8(search.sort_field_);
s << qint32(search.limit_);
return s;
}
QDataStream& operator >>(QDataStream& s, smart_playlists::Search& search) {
quint8 sort_type, sort_field;
qint32 limit;
s >> search.terms_ >> sort_type >> sort_field >> limit;
search.sort_type_ = smart_playlists::Search::SortType(sort_type);
search.sort_field_ = smart_playlists::SearchTerm::Field(sort_field);
search.limit_ = limit;
return s;
}