1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
/*
SPDX-FileCopyrightText: 2011 Joris Guisson <joris.guisson@gmail.com>
SPDX-License-Identifier: GPL-2.0-or-later
*/
#include "torrentloadqueue.h"
#include <QDateTime>
#include <QDir>
#include <QFileInfo>
#include <KIO/FileCopyJob>
#include <KIO/Job>
#include <KLocalizedString>
#include "scanfolderpluginsettings.h"
#include <bcodec/bdecoder.h>
#include <bcodec/bnode.h>
#include <interfaces/coreinterface.h>
#include <util/fileops.h>
#include <util/functions.h>
#include <util/log.h>
namespace kt
{
TorrentLoadQueue::TorrentLoadQueue(CoreInterface *core, QObject *parent)
: QObject(parent)
, core(core)
{
connect(&timer, &QTimer::timeout, this, &TorrentLoadQueue::loadOne);
timer.setSingleShot(true);
}
TorrentLoadQueue::~TorrentLoadQueue()
{
}
void TorrentLoadQueue::add(const QUrl &url)
{
to_load.append(url);
if (!timer.isActive())
timer.start(1000);
}
void TorrentLoadQueue::add(const QList<QUrl> &urls)
{
to_load.append(urls);
if (!timer.isActive())
timer.start(1000);
}
bool TorrentLoadQueue::validateTorrent(const QUrl &url, QByteArray &data)
{
// try to decode file, if it is syntactically correct, we can try to load it
QFile fptr(url.toLocalFile());
if (!fptr.open(QIODevice::ReadOnly))
return false;
try {
data = fptr.readAll();
bt::BDecoder dec(data, false);
bt::BNode *n = dec.decode();
if (n) {
// valid node, so file is complete
delete n;
return true;
} else {
// decoding failed so incomplete
return false;
}
} catch (...) {
// any error means shit happened and the file is incomplete
return false;
}
}
void TorrentLoadQueue::loadOne()
{
if (to_load.isEmpty())
return;
QUrl url = to_load.takeFirst();
QByteArray data;
if (validateTorrent(url, data)) {
// Load it
load(url, data);
} else {
// Not valid, so two options:
// - not a torrent
// - incomplete torrent, still being written
// We use the last modified time to determine this
if (QFileInfo(url.toLocalFile()).lastModified().secsTo(QDateTime::currentDateTime()) < 2) {
// Still being written, lets try again later
to_load.append(url);
}
}
if (!to_load.isEmpty())
timer.start(1000);
}
void TorrentLoadQueue::load(const QUrl &url, const QByteArray &data)
{
bt::Out(SYS_SNF | LOG_NOTICE) << "ScanFolder: loading " << url.toDisplayString() << bt::endl;
QString group;
if (ScanFolderPluginSettings::addToGroup())
group = ScanFolderPluginSettings::group();
if (ScanFolderPluginSettings::openSilently())
core->loadSilently(data, url, group, QString());
else
core->load(data, url, group, QString());
loadingFinished(url);
}
void TorrentLoadQueue::loadingFinished(const QUrl &url)
{
QString name = url.fileName();
QString dirname = QFileInfo(url.toLocalFile()).absolutePath();
if (!dirname.endsWith(bt::DirSeparator()))
dirname += bt::DirSeparator();
switch (action) {
case DeleteAction:
// If torrent has hidden complement - remove it too.
if (bt::Exists(dirname + QLatin1Char('.') + name))
bt::Delete(dirname + QLatin1Char('.') + name, true);
bt::Delete(url.toLocalFile(), true);
break;
case MoveAction:
// If torrent has hidden complement - remove it too.
if (bt::Exists(dirname + QLatin1Char('.') + name))
bt::Delete(dirname + QLatin1Char('.') + name, true);
if (!bt::Exists(dirname + i18nc("folder name part", "loaded")))
bt::MakeDir(dirname + i18nc("folder name part", "loaded"), true);
KIO::file_move(url,
QUrl::fromLocalFile(dirname + i18nc("folder name part", "loaded") + bt::DirSeparator() + name),
-1,
KIO::HideProgressInfo | KIO::Overwrite);
break;
case DefaultAction:
QFile f(dirname + QLatin1Char('.') + name);
f.open(QIODevice::WriteOnly);
f.close();
break;
}
}
}
#include "moc_torrentloadqueue.cpp"
|
