/*
 * This file Copyright (C) 2012-2016 Mnemosyne LLC
 *
 * It may be used under the GNU GPL versions 2 or 3
 * or any future license endorsed by Mnemosyne LLC.
 *
 */

#include <QFile>
#include <QDir>

#include <libtransmission/transmission.h>
#include <libtransmission/crypto-utils.h> // tr_base64_encode()

#include "AddData.h"
#include "Utils.h"

int
AddData::set (const QString& key)
{
  if (Utils::isMagnetLink (key))
    {
      magnet = key;
      type = MAGNET;
    }
  else if  (Utils::isUriWithSupportedScheme (key))
    {
      url = key;
      type = URL;
    }
  else if (QFile(key).exists ())
    {
      filename = QDir::fromNativeSeparators (key);
      type = FILENAME;

      QFile file (key);
      file.open (QIODevice::ReadOnly);
      metainfo = file.readAll ();
      file.close ();
    }
  else if (Utils::isHexHashcode (key))
    {
      magnet = QString::fromUtf8("magnet:?xt=urn:btih:") + key;
      type = MAGNET;
    }
  else
    {
      size_t len;
      void * raw = tr_base64_decode (key.toUtf8().constData(), key.toUtf8().size(), &len);
      if (raw)
        {
          metainfo.append (static_cast<const char*> (raw), int(len));
          tr_free (raw);
          type = METAINFO;
        }
      else
        {
          type = NONE;
        }
    }

  return type;
}

QByteArray
AddData::toBase64 () const
{
  QByteArray ret;

  if (!metainfo.isEmpty ())
    {
      size_t len;
      void * b64 = tr_base64_encode (metainfo.constData(), metainfo.size(), &len);
      ret = QByteArray (static_cast<const char*> (b64), int(len));
      tr_free (b64);
    }

  return ret;
}

QString
AddData::readableName () const
{
  QString ret;

  switch (type)
    {
      case FILENAME:
        ret = filename;
        break;

      case MAGNET:
        ret = magnet;
        break;

      case URL:
        ret = url.toString();
        break;

      case METAINFO:
        {
          tr_info inf;
          tr_ctor * ctor = tr_ctorNew (NULL);
          tr_ctorSetMetainfo (ctor, reinterpret_cast<const quint8*> (metainfo.constData()), metainfo.size());
          if (tr_torrentParse (ctor, &inf) == TR_PARSE_OK )
            {
              ret = QString::fromUtf8 (inf.name); // metainfo is required to be UTF-8
              tr_metainfoFree (&inf);
            }
          tr_ctorFree (ctor);
          break;
        }
    }

  return ret;
}

QString
AddData::readableShortName () const
{
  switch (type)
    {
      case FILENAME:
        return QFileInfo (filename).fileName ();

      case URL:
        return url.path ().split (QLatin1Char ('/')).last ();

      default:
        return readableName ();
    }
}