/*
 * Strawberry Music Player
 * This file was part of Clementine.
 * Copyright 2012, David Sansome <me@davidsansome.com>
 *
 * Strawberry 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.
 *
 * Strawberry 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 Strawberry.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include "config.h"

#include <algorithm>
#include <cmath>

#include <QList>
#include <QMap>
#include <QSet>
#include <QChar>
#include <QScopedPointer>
#include <QString>
#include <QtAlgorithms>
#include <QAbstractItemModel>

#include "playlist.h"
#include "playlistfilterparser.h"
#include "utilities/searchparserutils.h"

class SearchTermComparator {
 public:
  SearchTermComparator() = default;
  virtual ~SearchTermComparator() = default;
  virtual bool Matches(const QString &element) const = 0;
 private:
  Q_DISABLE_COPY(SearchTermComparator)
};

// "compares" by checking if the field contains the search term
class DefaultComparator : public SearchTermComparator {
 public:
  explicit DefaultComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.contains(search_term_);
  }
 private:
  QString search_term_;

  Q_DISABLE_COPY(DefaultComparator)
};

class EqComparator : public SearchTermComparator {
 public:
  explicit EqComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return search_term_ == element;
  }
 private:
  QString search_term_;
};

class NeComparator : public SearchTermComparator {
 public:
  explicit NeComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return search_term_ != element;
  }
 private:
  QString search_term_;
};

class LexicalGtComparator : public SearchTermComparator {
 public:
  explicit LexicalGtComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element > search_term_;
  }
 private:
  QString search_term_;
};

class LexicalGeComparator : public SearchTermComparator {
 public:
  explicit LexicalGeComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element >= search_term_;
  }
 private:
  QString search_term_;
};

class LexicalLtComparator : public SearchTermComparator {
 public:
  explicit LexicalLtComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element < search_term_;
  }
 private:
  QString search_term_;
};

class LexicalLeComparator : public SearchTermComparator {
 public:
  explicit LexicalLeComparator(const QString &value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element <= search_term_;
  }
 private:
  QString search_term_;
};

// Float Comparators are for the rating
class FloatEqComparator : public SearchTermComparator {
 public:
  explicit FloatEqComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return search_term_ == element.toFloat();
  }
 private:
  float search_term_;
};

class FloatNeComparator : public SearchTermComparator {
 public:
  explicit FloatNeComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return search_term_ != element.toFloat();
  }
 private:
  float search_term_;
};

class FloatGtComparator : public SearchTermComparator {
 public:
  explicit FloatGtComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toFloat() > search_term_;
  }
 private:
  float search_term_;
};

class FloatGeComparator : public SearchTermComparator {
 public:
  explicit FloatGeComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toFloat() >= search_term_;
  }
 private:
  float search_term_;
};

class FloatLtComparator : public SearchTermComparator {
 public:
  explicit FloatLtComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toFloat() < search_term_;
  }
 private:
  float search_term_;
};

class FloatLeComparator : public SearchTermComparator {
 public:
  explicit FloatLeComparator(const float value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toFloat() <= search_term_;
  }
 private:
  float search_term_;
};

class GtComparator : public SearchTermComparator {
 public:
  explicit GtComparator(const int value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toInt() > search_term_;
  }
 private:
  int search_term_;
};

class GeComparator : public SearchTermComparator {
 public:
  explicit GeComparator(const int value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toInt() >= search_term_;
  }
 private:
  int search_term_;
};

class LtComparator : public SearchTermComparator {
 public:
  explicit LtComparator(const int value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toInt() < search_term_;
  }
 private:
  int search_term_;
};

class LeComparator : public SearchTermComparator {
 public:
  explicit LeComparator(const int value) : search_term_(value) {}
  bool Matches(const QString &element) const override {
    return element.toInt() <= search_term_;
  }
 private:
  int search_term_;
};

// The length field of the playlist (entries) contains a song's running time in nanoseconds.
// However, We don't really care about nanoseconds, just seconds.
// Thus, with this decorator we drop the last 9 digits, if that many are present.
class DropTailComparatorDecorator : public SearchTermComparator {
 public:
  explicit DropTailComparatorDecorator(SearchTermComparator *cmp) : cmp_(cmp) {}

  bool Matches(const QString &element) const override {
    if (element.length() > 9) {
      return cmp_->Matches(element.left(element.length() - 9));
    }
    else {
      return cmp_->Matches(element);
    }
  }
 private:
  QScopedPointer<SearchTermComparator> cmp_;
};

class RatingComparatorDecorator : public SearchTermComparator {
 public:
  explicit RatingComparatorDecorator(SearchTermComparator *cmp) : cmp_(cmp) {}
  bool Matches(const QString &element) const override {
    return cmp_->Matches(QString::number(lround(element.toDouble() * 10.0)));
  }
 private:
  QScopedPointer<SearchTermComparator> cmp_;
};

// filter that applies a SearchTermComparator to all fields of a playlist entry
class FilterTerm : public FilterTree {
 public:
  explicit FilterTerm(SearchTermComparator *comparator, const QList<int> &columns) : cmp_(comparator), columns_(columns) {}

  bool accept(int row, const QModelIndex &parent, const QAbstractItemModel *const model) const override {
    for (int i : columns_) {
      QModelIndex idx(model->index(row, i, parent));
      if (cmp_->Matches(idx.data().toString().toLower())) return true;
    }
    return false;
  }
  FilterType type() override { return FilterType::Term; }
 private:
  QScopedPointer<SearchTermComparator> cmp_;
  QList<int> columns_;
};

// filter that applies a SearchTermComparator to one specific field of a playlist entry
class FilterColumnTerm : public FilterTree {
 public:
  FilterColumnTerm(const int column, SearchTermComparator *comparator) : col(column), cmp_(comparator) {}

  bool accept(int row, const QModelIndex &parent, const QAbstractItemModel *const model) const override {
    QModelIndex idx(model->index(row, col, parent));
    return cmp_->Matches(idx.data().toString().toLower());
  }
  FilterType type() override { return FilterType::Column; }
 private:
  int col;
  QScopedPointer<SearchTermComparator> cmp_;
};

class NotFilter : public FilterTree {
 public:
  explicit NotFilter(const FilterTree *inv) : child_(inv) {}

  bool accept(int row, const QModelIndex &parent, const QAbstractItemModel *const model) const override {
    return !child_->accept(row, parent, model);
  }
  FilterType type() override { return FilterType::Not; }
 private:
  QScopedPointer<const FilterTree> child_;
};

class OrFilter : public FilterTree {
 public:
  ~OrFilter() override { qDeleteAll(children_); }
  virtual void add(FilterTree *child) { children_.append(child); }
  bool accept(int row, const QModelIndex &parent, const QAbstractItemModel *const model) const override {
    return std::any_of(children_.begin(), children_.end(), [row, parent, model](FilterTree *child) { return child->accept(row, parent, model); });
  }
  FilterType type() override { return FilterType::Or; }
 private:
  QList<FilterTree*> children_;
};

class AndFilter : public FilterTree {
 public:
  ~AndFilter() override { qDeleteAll(children_); }
  virtual void add(FilterTree *child) { children_.append(child); }
  bool accept(int row, const QModelIndex &parent, const QAbstractItemModel *const model) const override {
    return !std::any_of(children_.begin(), children_.end(), [row, parent, model](FilterTree *child) { return !child->accept(row, parent, model); });
  }
  FilterType type() override { return FilterType::And; }
 private:
  QList<FilterTree*> children_;
};

FilterParser::FilterParser(const QString &filter, const QMap<QString, int> &columns, const QSet<int> &numerical_cols) : iter_{}, end_{}, filterstring_(filter), columns_(columns), numerical_columns_(numerical_cols) {}

FilterTree *FilterParser::parse() {
  iter_ = filterstring_.constBegin();
  end_ = filterstring_.constEnd();
  return parseOrGroup();
}

void FilterParser::advance() {

  while (iter_ != end_ && iter_->isSpace()) {
    ++iter_;
  }

}

FilterTree *FilterParser::parseOrGroup() {

  advance();
  if (iter_ == end_) return new NopFilter;

  OrFilter *group = new OrFilter;
  group->add(parseAndGroup());
  advance();
  while (checkOr()) {
    group->add(parseAndGroup());
    advance();
  }
  return group;

}

FilterTree *FilterParser::parseAndGroup() {

  advance();
  if (iter_ == end_) return new NopFilter;

  AndFilter *group = new AndFilter();
  do {
    group->add(parseSearchExpression());
    advance();
    if (iter_ != end_ && *iter_ == QLatin1Char(')')) break;
    if (checkOr(false)) {
      break;
    }
    checkAnd();  // if there's no 'AND', we'll add the term anyway...
  } while (iter_ != end_);

  return group;

}

bool FilterParser::checkAnd() {

  if (iter_ != end_) {
    if (*iter_ == QLatin1Char('A')) {
      buf_ += *iter_;
      ++iter_;
      if (iter_ != end_ && *iter_ == QLatin1Char('N')) {
        buf_ += *iter_;
        ++iter_;
        if (iter_ != end_ && *iter_ == QLatin1Char('D')) {
          buf_ += *iter_;
          ++iter_;
          if (iter_ != end_ && (iter_->isSpace() || *iter_ == QLatin1Char('-') || *iter_ == QLatin1Char('('))) {
            advance();
            buf_.clear();
            return true;
          }
        }
      }
    }
  }
  return false;

}

bool FilterParser::checkOr(const bool step_over) {

  if (!buf_.isEmpty()) {
    if (buf_ == QLatin1String("OR")) {
      if (step_over) {
        buf_.clear();
        advance();
      }
      return true;
    }
  }
  else {
    if (iter_ != end_) {
      if (*iter_ == QLatin1Char('O')) {
        buf_ += *iter_;
        ++iter_;
        if (iter_ != end_ && *iter_ == QLatin1Char('R')) {
          buf_ += *iter_;
          ++iter_;
          if (iter_ != end_ && (iter_->isSpace() || *iter_ == QLatin1Char('-') || *iter_ == QLatin1Char('('))) {
            if (step_over) {
              buf_.clear();
              advance();
            }
            return true;
          }
        }
      }
    }
  }
  return false;

}

FilterTree *FilterParser::parseSearchExpression() {

  advance();
  if (iter_ == end_) return new NopFilter;
  if (*iter_ == QLatin1Char('(')) {
    ++iter_;
    advance();
    FilterTree *tree = parseOrGroup();
    advance();
    if (iter_ != end_) {
      if (*iter_ == QLatin1Char(')')) {
        ++iter_;
      }
    }
    return tree;
  }
  else if (*iter_ == QLatin1Char('-')) {
    ++iter_;
    FilterTree *tree = parseSearchExpression();
    if (tree->type() != FilterTree::FilterType::Nop) return new NotFilter(tree);
    return tree;
  }
  else {
    return parseSearchTerm();
  }

}

FilterTree *FilterParser::parseSearchTerm() {

  QString col;
  QString search;
  QString prefix;
  bool inQuotes = false;
  for (; iter_ != end_; ++iter_) {
    if (inQuotes) {
      if (*iter_ == QLatin1Char('"')) {
        inQuotes = false;
      }
      else {
        buf_ += *iter_;
      }
    }
    else {
      if (*iter_ == QLatin1Char('"')) {
        inQuotes = true;
      }
      else if (col.isEmpty() && *iter_ == QLatin1Char(':')) {
        col = buf_.toLower();
        buf_.clear();
        prefix.clear();  // prefix isn't allowed here - let's ignore it
      }
      else if (iter_->isSpace() || *iter_ == QLatin1Char('(') || *iter_ == QLatin1Char(')') || *iter_ == QLatin1Char('-')) {
        break;
      }
      else if (buf_.isEmpty()) {
        // we don't know whether there is a column part in this search term thus we assume the latter and just try and read a prefix
        if (prefix.isEmpty() && (*iter_ == QLatin1Char('>') || *iter_ == QLatin1Char('<') || *iter_ == QLatin1Char('=') || *iter_ == QLatin1Char('!'))) {
          prefix += *iter_;
        }
        else if (prefix != QLatin1Char('=') && *iter_ == QLatin1Char('=')) {
          prefix += *iter_;
        }
        else {
          buf_ += *iter_;
        }
      }
      else {
        buf_ += *iter_;
      }
    }
  }

  search = buf_.toLower();
  buf_.clear();

  return createSearchTermTreeNode(col, prefix, search);

}

FilterTree *FilterParser::createSearchTermTreeNode(const QString &col, const QString &prefix, const QString &search) const {

  if (search.isEmpty() && prefix != QLatin1Char('=')) {
    return new NopFilter;
  }
  // here comes a mess :/
  // well, not that much of a mess, but so many options -_-
  SearchTermComparator *cmp = nullptr;

  // Handle the float based Rating Column
  if (columns_[col] == Playlist::Column_Rating) {
    float parsed_search = Utilities::ParseSearchRating(search);

    if (prefix == QLatin1Char('=')) {
      cmp = new FloatEqComparator(parsed_search);
    }
    else if (prefix == QLatin1String("!=") || prefix == QLatin1String("<>")) {
      cmp = new FloatNeComparator(parsed_search);
    }
    else if (prefix == QLatin1Char('>')) {
      cmp = new FloatGtComparator(parsed_search);
    }
    else if (prefix == QLatin1String(">=")) {
      cmp = new FloatGeComparator(parsed_search);
    }
    else if (prefix == QLatin1Char('<')) {
      cmp = new FloatLtComparator(parsed_search);
    }
    else if (prefix == QLatin1String("<=")) {
      cmp = new FloatLeComparator(parsed_search);
    }
    else {
      cmp = new FloatEqComparator(parsed_search);
    }
  }
  else if (prefix == QLatin1String("!=") || prefix == QLatin1String("<>")) {
    cmp = new NeComparator(search);
  }
  else if (!col.isEmpty() && columns_.contains(col) && numerical_columns_.contains(columns_[col])) {
    // the length column contains the time in seconds (nanoseconds, actually - the "nano" part is handled by the DropTailComparatorDecorator,  though).
    int search_value = 0;
    if (columns_[col] == Playlist::Column_Length) {
      search_value = Utilities::ParseSearchTime(search);
    }
    else {
      search_value = search.toInt();
    }
    // alright, back to deciding which comparator we'll use
    if (prefix == QLatin1Char('>')) {
      cmp = new GtComparator(search_value);
    }
    else if (prefix == QLatin1String(">=")) {
      cmp = new GeComparator(search_value);
    }
    else if (prefix == QLatin1Char('<')) {
      cmp = new LtComparator(search_value);
    }
    else if (prefix == QLatin1String("<=")) {
      cmp = new LeComparator(search_value);
    }
    else {
      // convert back because for time/rating
      cmp = new EqComparator(QString::number(search_value));
    }
  }
  else {
    if (prefix == QLatin1Char('=')) {
      cmp = new EqComparator(search);
    }
    else if (prefix == QLatin1Char('>')) {
      cmp = new LexicalGtComparator(search);
    }
    else if (prefix == QLatin1String(">=")) {
      cmp = new LexicalGeComparator(search);
    }
    else if (prefix == QLatin1Char('<')) {
      cmp = new LexicalLtComparator(search);
    }
    else if (prefix == QLatin1String("<=")) {
      cmp = new LexicalLeComparator(search);
    }
    else {
      cmp = new DefaultComparator(search);
    }
  }

  if (columns_.contains(col)) {
    if (columns_[col] == Playlist::Column_Length) {
      cmp = new DropTailComparatorDecorator(cmp);
    }
    return new FilterColumnTerm(columns_[col], cmp);
  }
  else {
    return new FilterTerm(cmp, columns_.values());
  }

}