search.cpp

#include "search.h"
#include "ui_search.h"
#include "utils.h"
#include "dicttabscontainer.h"

#include <QLabel>
#include <QCheckBox>
#include <QLineEdit>
#include <QProgressBar>
#include <QTreeWidget>
#include <QSqlDatabase>
#include <QSqlQuery>

#include <QFormLayout>
#include <QComboBox>
#include <QSpinBox>
#include <QDebug>

Search::Search(const QSet<DictGlobalAttributes> * attrs, QWidget *parent) :
  openedDicts(attrs),
  QDialog(parent),
//  dictsToSearch(&(ui->verticalLayout_2)),
  ui(new Ui::search)
{
  ui->setupUi(this);

  layout = new QGridLayout(ui->whereToSearchBox);

  QSet<int> set;
  set.insert(1);
  set.insert(2);
  set.insert(3);

  int counter = 0;
  for (QSet<DictGlobalAttributes>::const_iterator i = attrs->begin(); i != attrs->end(); ++i, counter++) {
      QCheckBox * checkbox = new QCheckBox(tr((*i).getDictname().toStdString().c_str()));
      checkbox->setProperty("row", counter);
      checkbox->setProperty("connection name", (*i).getFilename());
      checkbox->setProperty("dictionary name", (*i).getDictname());
      checkbox->setChecked(true);
      layout->addWidget(checkbox, counter, 0);
//      layout->addWidget(new QLabel(tr(QVariant(counter).toString().toStdString().c_str())), counter, 1);
      QLineEdit * line = new QLineEdit();

      line->setToolTip("В этом поле вы можете задать локальные переменные поиска для каждого из словарей. \nО переменных подробнее написано в всплывающей подсказке в поле 'глобальные переменные поиска'");
      layout->addWidget(line, counter, 1);
  }

  ui->whereToSearchBox->setLayout(layout);

  connect(ui->findButton, SIGNAL(clicked()), this, SLOT(onSearchClick()));
}


bool Search::checkVars(QString line) {
  QStringList varsList = line.split(" ");
  QRegExp rx("(.=(.+(,)?)+)*");
  foreach (const QString &item, varsList) {
      if (!rx.exactMatch(item)) return false;
  }

  return true;
}

bool Search::checkQuery(QString line) {
  if (line.contains("delete", Qt::CaseInsensitive) || line.contains("update", Qt::CaseInsensitive) || line.contains("insert", Qt::CaseInsensitive)) {
      errorMsg("Пожалуйста, не пытайтесь сломать программу (я знаю, это довольно легко, но в то же время абсолютно бесполезно).");
      return false;
    }
  return true;
}

QMap<QString, QList<QString> > Search::parseVars(QString line) {
  QMap<QString, QList<QString> > vars;
  return vars;
}

QStringList constructDictionary(const QStringList& initialLine, QString vars, bool isglobal) {
  QStringList dictionary;
  QStringList plainDictionary; //this list contains lines that don't need to have replaces

  if (vars.trimmed().isEmpty()) {
      plainDictionary = initialLine;
      for (int i = 0; i < plainDictionary.size(); i++) {
          plainDictionary[i].append("%");
          plainDictionary[i].prepend("%");
          if (isglobal) {
            plainDictionary[i].replace(QRegularExpression("\\$."), "_");
            qDebug() << "Replaced unused variable, that's what we get: " << plainDictionary[i];
            }
        }
      return plainDictionary;
    }

  foreach (const QString &line, initialLine) {
    if (!line.contains("$")) {
        plainDictionary.append(line);
      } else {
        dictionary.append(line);
      }
    }

  foreach(const QString& item, dictionary) {
      qDebug() << "For replace: " << item;
    }

  foreach(const QString& item, plainDictionary) {
      qDebug() << "Don't need changes: " << item;
    }

  foreach(const QString& item, dictionary) {
      qDebug() << "Replacing: " << item;
    }

  QStringList varsSeparate = vars.split(" ");

  foreach(const QString& item, varsSeparate) {
      qDebug() << "Vars separate: " << item;
    }

  foreach (const QString& line, dictionary) {
    foreach (const QString& item, varsSeparate) {
        QString varname = item[0];
        varname.prepend("$");
        if (!line.contains(varname)) {
            qDebug() << line << " doesn't contain" << varname;
            continue;
          } else {
            QString onlyValues = item.mid(2);
            QStringList valuesList = onlyValues.split(",");
            long int sizeBefore = dictionary.size();
            foreach (const QString& value, valuesList) {
                if (value.trimmed().isEmpty()) {
                    continue;
                  }
                qDebug() << "replacing " << varname << " with " << value;

                foreach (const QString& word, dictionary) {
                    QString tempWord = word;
                    qDebug() << tempWord;
                    tempWord.replace(varname, value);
                    if (tempWord != word) {
                      dictionary.append(tempWord);
                      qDebug() << tempWord << " " << tempWord.replace(varname, value);
                      }
                  }
              }
            for (int i = 0; i < sizeBefore; i++) {
                dictionary.removeFirst();
              }
          }
      }
    }

  for (int i = 0; i < plainDictionary.size(); i++) {
      plainDictionary[i].append("%");
      plainDictionary[i].prepend("%");
      if (isglobal) {
        plainDictionary[i].replace(QRegularExpression("\\$."), "_");
        }
    }

  for (int i = 0; i < dictionary.size(); i++) {
      dictionary[i].append("%");
      dictionary[i].prepend("%");
      if (isglobal) {
        dictionary[i].replace(QRegularExpression("\\$."), "_");
        }
    }

  dictionary.append(plainDictionary);
  return dictionary;

}

QString Search::constructQuery(int dict_size) {
  QString query("SELECT * FROM 'dictionary' WHERE ");
//  QString query("SELECT id, word, transcription, translation FROM 'dictionary' WHERE ");
  if (ui->etimCheck->isChecked()) {
      query.append("(etimology_tag NOT NULL) AND ");
    }

  if (ui->paradigmCheck->isChecked()) {
      query.append("id = (SELECT regular_form FROM 'dictionary') AND ");
    }

  if (ui->praatCheck->isChecked()) {
      query.append("id = (SELECT wordid FROM 'dict_blobs_description' WHERE type=2) AND ");
    }

  if (ui->audioCheck->isChecked()) {
      query.append("id = (SELECT wordid FROM 'dict_blobs_description' WHERE type=1) AND ");
    }


  if (ui->whereToSearchCombo->currentIndex() == 0) {
      query.append("regular_form IS NULL AND (");
      query.append("(transcription LIKE ?)");
      for (int i = 1; i < dict_size; i++) {
          query.append(" OR (transcription LIKE ?)");
        }
      query.append(")");
    } else if (ui->whereToSearchCombo->currentIndex() == 1) {
      query.append("(");
      query.append("(translation LIKE ?)");
      for (int i = 1; i < dict_size; i++) {
          query.append(" OR (translation LIKE ?)");
        }
      query.append(")");
    } else if (ui->whereToSearchCombo->currentIndex() == 2) {
      query.append("regular_form NOT NULL AND (");
      query.append("(transcription LIKE ?)");
      for (int i = 1; i < dict_size; i++) {
          query.append(" OR (transcription LIKE ?)");
        }
      query.append(")");
    } else if (ui->whereToSearchCombo->currentIndex() == 3) {
      query.append("(");
      query.append("(word LIKE ?)");
      for (int i = 1; i < dict_size; i++) {
          query.append(" OR (word LIKE ?)");
        }
      query.append(")");
    }

  return query;
}

bool Search::onSearchClick() {
  if (!this->checkVars(ui->globalVarsLine->text())) {
      errorMsg("Вы неправильно заполнили поле глобальных переменных. Наведите мышь на поле ввода глобальных переменных на пару секунд, и выскочит подсказка, в которой написано, как правильно задавать переменные.");
    }

  QList<QCheckBox *> ch = ui->whereToSearchBox->findChildren<QCheckBox *>();
  qDebug() << "List size:" << ch.size();

  for (QList<QCheckBox *>::iterator i = ch.begin(); i != ch.end(); ++i) {
      bool checked = ( (*i)->isChecked());
      if (checked) {
          qDebug() << (*i)->property("row");
          qDebug() << (*i)->property("connection name");

          QLineEdit * varline = (QLineEdit *) layout->itemAtPosition((*i)->property("row").toInt(), 1)->widget();
          if (varline != NULL) {
            qDebug() << "Checking var line";
            if (!this->checkVars(varline->text())) {
                QString msg = "Вы неправильно заполнили поле переменных. Наведите мышь на поле ввода глобальных переменных на пару секунд, и выскочит подсказка, в которой написано, как правильно задавать переменные. \nОшибка в строке: ";
                msg.append(varline->text());
                errorMsg(msg);
              }
            }
        }
    }

  QTreeWidget * resTree = new QTreeWidget(this);
  QStringList labels;
  resTree->setColumnCount(3);
  resTree->setSortingEnabled(true);
  labels.append("Регулярная форма");
  labels.append("Указатель на регулярную форму");
  labels.append("Транскрипция");
  labels.append("Перевод");
  labels.append("ID");
  resTree->setHeaderLabels(labels);
  resTree->hideColumn(1);
  resTree->hideColumn(4);
  resTree->setColumnWidth(0,350);

  for (QList<QCheckBox *>::iterator i = ch.begin(); i != ch.end(); ++i) {
      bool checked = ( (*i)->isChecked());
      if (checked) {
          qDebug() << (*i)->property("row");
          qDebug() << (*i)->property("connection name");

          QTreeWidgetItem * resultsForDict = new QTreeWidgetItem((*i)->property("dictionary name").toStringList());

          resTree->addTopLevelItem(resultsForDict);
          QLineEdit * varline = (QLineEdit *) layout->itemAtPosition((*i)->property("row").toInt(), 1)->widget();
          if (varline != NULL) {
            QStringList dict;
            dict.push_back(ui->searchLine->text());
            dict = constructDictionary(dict, varline->text(), false);
            foreach(const QString& item, dict) {
                qDebug() << "Replaced by local: " << item;
              }

            dict = constructDictionary(dict, ui->globalVarsLine->text(), true);
            foreach(const QString& item, dict) {
                qDebug() << "Replaced by global: " << item;
              }

            qDebug() << "Dict size" << dict.size();

            qDebug() << constructQuery(dict.size());
            QString queryText = constructQuery(dict.size());
            QSqlQuery query(QSqlDatabase::database((*i)->property("connection name").toString()));
            query.prepare(queryText);
            for (int i = 0; i < dict.size(); i++) {
                query.bindValue(i, QVariant(dict[i]));
              }
            if (query.exec()) {
                while (query.next()) {
                    QStringList item;
                    item.append(query.value("word").toString());
                    item.append(query.value("regular_form").toString());
                    item.append(query.value("transcription").toString());
                    item.append(query.value("translation").toString());
                    item.append(query.value("id").toString());
                    QTreeWidgetItem * entry = new QTreeWidgetItem(item);
                    resultsForDict->addChild(entry);
                  }
              }

//            qDebug() << this->parent()->parent()->metaObject()->className();
          }

        }
    }
  int tabid = (static_cast<QTabWidget *>(this->parentWidget()->parentWidget()))->addTab(resTree, ui->searchLine->text());
  qDebug() << this->parentWidget()->parentWidget()->parentWidget()->parentWidget()->findChild<DictTabsContainer *>();
  connect(this, SIGNAL(showDictAndWord(QString,QVariant)), this->parentWidget()->parentWidget()->parentWidget()->parentWidget()->findChild<DictTabsContainer *>(), SLOT(goToDictAndWord(QString,QVariant)));
  connect(resTree, SIGNAL(doubleClicked(QModelIndex)), this, SLOT(doubleClickTree(QModelIndex)));
  (static_cast<QTabWidget *>(this->parentWidget()->parentWidget()))->setCurrentIndex(tabid);
  return true;
}

bool Search::doubleClickTree(QModelIndex idx) {
  QVariant id = idx.sibling(idx.row(), 4).data();
  if (!id.isValid()) {
      return false;
    }

  //that means that we were looking for paradigm
  if (!idx.sibling(idx.row(), 1).data().toString().isEmpty()) {
      id = idx.sibling(idx.row(), 1).data();
    }

  QString dictName = idx.parent().data().toString();

  qDebug() << "Clicked " << dictName << id;

  emit showDictAndWord(dictName, id);

  return true;
}

Search::~Search()
{
  delete ui;
}