#include "FileDBLink.hpp"

#include "EditDistance.hpp"

#include "Exception/PermissionException.hpp"

#include <QtCore/QtConcurrentMap>

#include <boost/bind.hpp>

void FileDBLink::updateIfModified(const QString& path)
{
  QFileInfo fileinfo(path);
  FileDBLink::DBStatus status = existsWithMtime(path, fileinfo.lastModified());

  switch (status) {
  case FileDBLink::NONE: {
    addFile(fileinfo);
    break;
  }
  case FileDBLink::MTIME_DIFFERENT: {
    updateFile(fileinfo);
  }
  default: {
  }
  }
}

void FileDBLink::addFile(const QFileInfo& fileinfo)
{
  addFile(fileinfo.absoluteFilePath(), fileinfo.size(), fileinfo.lastModified());
}

QByteArray FileDBLink::computeHash(const QString& path, QCryptographicHash::Algorithm algorithm)
{
  const static uint buffersize = 32768;
  QCryptographicHash hash(algorithm);
  QFile file(path);
  if ( file.open( QIODevice::ReadOnly ) ) {
    while(!file.atEnd()){
      hash.addData(file.read(buffersize));
    }
  }
  else {
    QString errorMsg = path + ": " + file.errorString();

    switch (file.error()) {
    case QFile::PermissionsError:
      throw PermissionException(errorMsg);
    default:
      throw IOException(errorMsg);
    }
  }
  return hash.result();
}


void FileDBLink::addFile(const QString& path, qint64 size, const QDateTime& lastModified)
{
  addFile(path, size, lastModified, computeHash(path));
}

void FileDBLink::updateFile(const QFileInfo& fileinfo)
{
  updateFile(fileinfo.absoluteFilePath(), fileinfo.size(), fileinfo.lastModified());
}

void FileDBLink::updateFile(const QString& path, qint64 size, const QDateTime& lastModified)
{
  updateFile(path, size, lastModified, computeHash(path));
}

const QList<QSharedPointer<FileDBLink::DBInfo> > FileDBLink::sortOn(const QString& prefix, SORTORDER order, bool extended)
{
  QList<QSharedPointer<DBInfo> > list = (extended) ? computedValues(prefix) : values(prefix);

  switch (order) {
  case PATH:
    {
      QList<QSharedPointer<FileDBLink::DBInfo> > oList;
      foreach(QSharedPointer<DBInfo> info, list) {
	oList.push_back(info);
      }
      return oList;
    }
  case SIZE:
    {
      QMultiMap<quint64, QSharedPointer<DBInfo> > oList;
      foreach(QSharedPointer<DBInfo> info, list) {
	oList.insert(info->size(), info);
      }
      return oList.values();
    }
  case MTIME:
    {
      QMultiMap<QDateTime, QSharedPointer<DBInfo> > oList;
      foreach(QSharedPointer<DBInfo> info, list) {
	oList.insert(info->mtime(), info);
      }
      return oList.values();
    }
  case CHECKSUM:
    {
      QMultiMap<QByteArray, QSharedPointer<DBInfo> > oList;
      foreach(QSharedPointer<DBInfo> info, list) {
	oList.insert(info->checksum(), info);
      }
      return oList.values();
    }
  case EDIT:
    {
      assert(extended);
      QMultiMap<int, QSharedPointer<DBInfo> > oList;
      foreach(QSharedPointer<DBInfo> info, list) {
	QSharedPointer<ExtendedDBInfo> ptr;
	ptr = info.dynamicCast<ExtendedDBInfo>();
	oList.insert(ptr->editDistance(), info);
      }
      return oList.values();

    }
  }
  abort();
}

QSharedPointer<FileDBLink::DBInfo> FileDBLink::computedValue(const QSharedPointer<DBInfo>& info,
							     const QList<QSharedPointer<DBInfo> >& entries)
{
  QString p1 = info->name();
  int minDist = 100000;
  QString other;
  for (QList<QSharedPointer<DBInfo> >::const_iterator it2 = entries.begin();
       it2 != entries.end(); ++it2) {
    if (info == *it2)
      continue;
    QString p2 = (*it2)->name();
    int dist = EditDistance::Compute(p1, p2, false);
    if (dist < minDist) {
      minDist = dist;
      other = (*it2)->path();
    }
  }
  return QSharedPointer<DBInfo>(new ExtendedDBInfo(*info, other, minDist));
}

const QList<QSharedPointer<FileDBLink::DBInfo> > FileDBLink::computedValues(const QString& prefix) const
{
  QList<QSharedPointer<DBInfo> > list;
  QList<QSharedPointer<DBInfo> > entries = values(prefix);

#if 1
  list = QtConcurrent::blockingMapped(entries, boost::bind( &FileDBLink::computedValue, _1, entries));
#else
  for (QList<QSharedPointer<DBInfo> >::const_iterator it1 = entries.begin();
       it1 != entries.end(); ++it1) {
    QSharedPointer<DBInfo> ext = computedValue(*it1, entries);
    list.push_back(ext);
  }
#endif
  return list;
}