/*
 * Bittorrent Client using Qt4 and libtorrent.
 * Copyright (C) 2006  Christophe Dumez
 *
 * This program 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 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * In addition, as a special exception, the copyright holders give permission to
 * link this program with the OpenSSL project's "OpenSSL" library (or with
 * modified versions of it that use the same license as the "OpenSSL" library),
 * and distribute the linked executables. You must obey the GNU General Public
 * License in all respects for all of the code used other than "OpenSSL".  If you
 * modify file(s), you may extend this exception to your version of the file(s),
 * but you are not obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * Contact : chris@qbittorrent.org
 */

#include "torrentpersistentdata.h"

#include <QDateTime>
#include <QDebug>
#include <QVariant>

#include "qinisettings.h"
#include "misc.h"
#include "qtorrenthandle.h"

#include <libtorrent/magnet_uri.hpp>

QHash<QString, TorrentTempData::TorrentData> TorrentTempData::data = QHash<QString, TorrentTempData::TorrentData>();
QHash<QString, TorrentTempData::TorrentMoveState> TorrentTempData::torrentMoveStates = QHash<QString, TorrentTempData::TorrentMoveState>();
QHash<QString, bool> HiddenData::data = QHash<QString, bool>();
unsigned int HiddenData::metadata_counter = 0;

bool TorrentTempData::hasTempData(const QString &hash) {
  return data.contains(hash);
}

void TorrentTempData::deleteTempData(const QString &hash) {
  data.remove(hash);
}

void TorrentTempData::setFilesPriority(const QString &hash,  const std::vector<int> &pp) {
  data[hash].files_priority = pp;
}

void TorrentTempData::setFilesPath(const QString &hash, const QStringList &path_list) {
  data[hash].path_list = path_list;
}

void TorrentTempData::setSavePath(const QString &hash, const QString &save_path) {
  data[hash].save_path = save_path;
}

void TorrentTempData::setLabel(const QString &hash, const QString &label) {
  data[hash].label = label;
}

void TorrentTempData::setSequential(const QString &hash, const bool &sequential) {
  data[hash].sequential = sequential;
}

bool TorrentTempData::isSequential(const QString &hash) {
  return data.value(hash).sequential;
}

void TorrentTempData::setSeedingMode(const QString &hash, const bool &seed) {
  data[hash].seed = seed;
}

bool TorrentTempData::isSeedingMode(const QString &hash) {
  return data.value(hash).seed;
}

QString TorrentTempData::getSavePath(const QString &hash) {
  return data.value(hash).save_path;
}

QStringList TorrentTempData::getFilesPath(const QString &hash) {
  return data.value(hash).path_list;
}

QString TorrentTempData::getLabel(const QString &hash) {
  return data.value(hash).label;
}

void TorrentTempData::getFilesPriority(const QString &hash, std::vector<int> &fp) {
  fp = data.value(hash).files_priority;
}

bool TorrentTempData::isMoveInProgress(const QString &hash) {
  return torrentMoveStates.find(hash) != torrentMoveStates.end();
}

void TorrentTempData::enqueueMove(const QString &hash, const QString &queuedPath) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i == torrentMoveStates.end()) {
    Q_ASSERT(false);
    return;
  }
  i->queuedPath = queuedPath;
}

void TorrentTempData::startMove(const QString &hash, const QString &oldPath, const QString& newPath) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i != torrentMoveStates.end()) {
    Q_ASSERT(false);
    return;
  }

  torrentMoveStates.insert(hash, TorrentMoveState(oldPath, newPath));
}

void TorrentTempData::finishMove(const QString &hash) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i == torrentMoveStates.end()) {
    Q_ASSERT(false);
    return;
  }
  torrentMoveStates.erase(i);
}

QString TorrentTempData::getOldPath(const QString &hash) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i == torrentMoveStates.end()) {
    Q_ASSERT(false);
    return QString();
  }
  return i->oldPath;
}

QString TorrentTempData::getNewPath(const QString &hash) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i == torrentMoveStates.end()) {
    Q_ASSERT(false);
    return QString();
  }
  return i->newPath;
}

QString TorrentTempData::getQueuedPath(const QString &hash) {
  QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
  if (i == torrentMoveStates.end()) {
    Q_ASSERT(false);
    return QString();
  }
  return i->queuedPath;
}

void TorrentTempData::setAddPaused(const QString &hash, const bool &paused) {
  data[hash].add_paused = paused;
}

bool TorrentTempData::isAddPaused(const QString &hash) {
  return data.value(hash).add_paused;
}

void HiddenData::addData(const QString &hash) {
  data[hash] = false;
}

bool HiddenData::hasData(const QString &hash) {
  return data.contains(hash);
}

void HiddenData::deleteData(const QString &hash) {
  if (data.value(hash, false))
    metadata_counter--;
  data.remove(hash);
}

int HiddenData::getSize() {
  return data.size();
}

int HiddenData::getDownloadingSize() {
  return data.size() - metadata_counter;
}

void HiddenData::gotMetadata(const QString &hash) {
  if (!data.contains(hash))
      return;
  data[hash] = true;
  metadata_counter++;
}

bool TorrentPersistentData::isKnownTorrent(QString hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  return all_data.contains(hash);
}

QStringList TorrentPersistentData::knownTorrents() {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  return all_data.keys();
}

void TorrentPersistentData::setRatioLimit(const QString &hash, const qreal &ratio) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  data["max_ratio"] = ratio;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

qreal TorrentPersistentData::getRatioLimit(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("max_ratio", USE_GLOBAL_RATIO).toReal();
}

bool TorrentPersistentData::hasPerTorrentRatioLimit() {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant>::ConstIterator it = all_data.constBegin();
  QHash<QString, QVariant>::ConstIterator itend = all_data.constEnd();
  for ( ; it != itend; ++it) {
    if (it.value().toHash().value("max_ratio", USE_GLOBAL_RATIO).toReal() >= 0) {
      return true;
    }
  }
  return false;
}

void TorrentPersistentData::setAddedDate(const QString &hash, const QDateTime &time) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  if (!data.contains("add_date")) {
    data["add_date"] = time;
    all_data[hash] = data;
    settings.setValue("torrents", all_data);
  }
}

QDateTime TorrentPersistentData::getAddedDate(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  QDateTime dt = data.value("add_date").toDateTime();
  if (!dt.isValid()) {
    setAddedDate(hash, QDateTime::currentDateTime());
    dt = QDateTime::currentDateTime();
  }
  return dt;
}

void TorrentPersistentData::setErrorState(const QString &hash, const bool has_error) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  data["has_error"] = has_error;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

bool TorrentPersistentData::hasError(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("has_error", false).toBool();
}

QDateTime TorrentPersistentData::getSeedDate(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("seed_date").toDateTime();
}

void TorrentPersistentData::deletePersistentData(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  if (all_data.contains(hash)) {
    all_data.remove(hash);
    settings.setValue("torrents", all_data);
  }
}

void TorrentPersistentData::saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path, const bool is_magnet) {
  Q_ASSERT(h.is_valid());
  qDebug("Saving persistent data for %s", qPrintable(h.hash()));
  // Save persistent data
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(h.hash()).toHash();
  data["is_magnet"] = is_magnet;
  if (is_magnet) {
    data["magnet_uri"] = misc::toQString(make_magnet_uri(h));
  }
  data["seed"] = h.is_seed();
  data["priority"] = h.queue_position();
  if (save_path.isEmpty()) {
    qDebug("TorrentPersistantData: save path is %s", qPrintable(h.save_path()));
    data["save_path"] = h.save_path();
  } else {
    qDebug("TorrentPersistantData: overriding save path is %s", qPrintable(save_path));
    data["save_path"] = save_path; // Override torrent save path (e.g. because it is a temp dir)
  }
  // Label
  data["label"] = TorrentTempData::getLabel(h.hash());
  // Save data
  all_data[h.hash()] = data;
  settings.setValue("torrents", all_data);
  qDebug("TorrentPersistentData: Saving save_path %s, hash: %s", qPrintable(h.save_path()), qPrintable(h.hash()));
  // Set Added date
  setAddedDate(h.hash(), QDateTime::currentDateTime());
  // Finally, remove temp data
  TorrentTempData::deleteTempData(h.hash());
}

void TorrentPersistentData::saveSavePath(const QString &hash, const QString &save_path) {
  Q_ASSERT(!hash.isEmpty());
  qDebug("TorrentPersistentData::saveSavePath(%s)", qPrintable(save_path));
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  data["save_path"] = save_path;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
  qDebug("TorrentPersistentData: Saving save_path: %s, hash: %s", qPrintable(save_path), qPrintable(hash));
}

void TorrentPersistentData::saveLabel(const QString &hash, const QString &label) {
  Q_ASSERT(!hash.isEmpty());
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  data["label"] = label;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

void TorrentPersistentData::saveName(const QString &hash, const QString &name) {
  Q_ASSERT(!hash.isEmpty());
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data.value(hash).toHash();
  data["name"] = name;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

void TorrentPersistentData::savePriority(const QTorrentHandle &h) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data[h.hash()].toHash();
  data["priority"] = h.queue_position();
  all_data[h.hash()] = data;
  settings.setValue("torrents", all_data);
}

void TorrentPersistentData::savePriority(const QString &hash, const int &queue_pos) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data[hash].toHash();
  data["priority"] = queue_pos;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

void TorrentPersistentData::saveSeedStatus(const QTorrentHandle &h) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data[h.hash()].toHash();
  bool was_seed = data.value("seed", false).toBool();
  if (was_seed != h.is_seed()) {
    data["seed"] = !was_seed;
    all_data[h.hash()] = data;
    settings.setValue("torrents", all_data);
  }
}

void TorrentPersistentData::saveSeedStatus(const QString &hash, const bool seedStatus) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  QHash<QString, QVariant> data = all_data[hash].toHash();
  data["seed"] = seedStatus;
  all_data[hash] = data;
  settings.setValue("torrents", all_data);
}

QString TorrentPersistentData::getSavePath(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  //qDebug("TorrentPersistentData: getSavePath %s", data["save_path"].toString().toLocal8Bit().data());
  return data.value("save_path").toString();
}

QString TorrentPersistentData::getLabel(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("label", "").toString();
}

QString TorrentPersistentData::getName(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("name", "").toString();
}

int TorrentPersistentData::getPriority(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("priority", -1).toInt();
}

bool TorrentPersistentData::isSeed(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("seed", false).toBool();
}

bool TorrentPersistentData::isMagnet(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  return data.value("is_magnet", false).toBool();
}

QString TorrentPersistentData::getMagnetUri(const QString &hash) {
  QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
  const QHash<QString, QVariant> all_data = settings.value("torrents").toHash();
  const QHash<QString, QVariant> data = all_data.value(hash).toHash();
  Q_ASSERT(data.value("is_magnet", false).toBool());
  return data.value("magnet_uri").toString();
}