138 lines
2.6 KiB
C++
138 lines
2.6 KiB
C++
/*
|
|
* 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 <QDir>
|
|
#include <QFile>
|
|
|
|
#include <libtransmission/crypto-utils.h> // tr_base64_encode()
|
|
#include <libtransmission/transmission.h>
|
|
|
|
#include "AddData.h"
|
|
#include "Utils.h"
|
|
|
|
namespace
|
|
{
|
|
|
|
QString getNameFromMetainfo(QByteArray const& metainfo)
|
|
{
|
|
QString name;
|
|
|
|
tr_ctor* ctor = tr_ctorNew(nullptr);
|
|
tr_ctorSetMetainfo(ctor, metainfo.constData(), metainfo.size());
|
|
|
|
tr_info inf;
|
|
if (tr_torrentParse(ctor, &inf) == TR_PARSE_OK)
|
|
{
|
|
name = QString::fromUtf8(inf.name); // metainfo is required to be UTF-8
|
|
tr_metainfoFree(&inf);
|
|
}
|
|
|
|
tr_ctorFree(ctor);
|
|
return name;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
int AddData::set(QString const& 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 = QStringLiteral("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 != nullptr)
|
|
{
|
|
metainfo.append(static_cast<char const*>(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<char const*>(b64), int(len));
|
|
tr_free(b64);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
QString AddData::readableName() const
|
|
{
|
|
switch (type)
|
|
{
|
|
case FILENAME:
|
|
return filename;
|
|
|
|
case MAGNET:
|
|
return magnet;
|
|
|
|
case URL:
|
|
return url.toString();
|
|
|
|
case METAINFO:
|
|
return getNameFromMetainfo(metainfo);
|
|
|
|
default: // NONE
|
|
return {};
|
|
}
|
|
}
|
|
|
|
QString AddData::readableShortName() const
|
|
{
|
|
switch (type)
|
|
{
|
|
case FILENAME:
|
|
return QFileInfo(filename).baseName();
|
|
|
|
case URL:
|
|
return url.path().split(QLatin1Char('/')).last();
|
|
|
|
default:
|
|
return readableName();
|
|
}
|
|
}
|