transmission/qt/WatchDir.cc

168 lines
3.6 KiB
C++

/*
* This file Copyright (C) 2009-2015 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
*/
#include <iostream>
#include <QDir>
#include <QFileSystemWatcher>
#include <QTimer>
#include <libtransmission/transmission.h>
#include "Prefs.h"
#include "TorrentModel.h"
#include "WatchDir.h"
/***
****
***/
WatchDir::WatchDir(TorrentModel const& model) :
myModel(model),
myWatcher(nullptr)
{
}
WatchDir::~WatchDir()
{
}
/***
****
***/
int WatchDir::metainfoTest(QString const& filename) const
{
int ret;
tr_info inf;
tr_ctor* ctor = tr_ctorNew(nullptr);
// parse
tr_ctorSetMetainfoFromFile(ctor, filename.toUtf8().constData());
int const err = tr_torrentParse(ctor, &inf);
if (err != 0)
{
ret = ERROR;
}
else if (myModel.hasTorrent(QString::fromUtf8(inf.hashString)))
{
ret = DUPLICATE;
}
else
{
ret = OK;
}
// cleanup
if (err == 0)
{
tr_metainfoFree(&inf);
}
tr_ctorFree(ctor);
return ret;
}
void WatchDir::onTimeout()
{
QTimer* t = qobject_cast<QTimer*>(sender());
QString const filename = t->objectName();
if (metainfoTest(filename) == OK)
{
emit torrentFileAdded(filename);
}
t->deleteLater();
}
void WatchDir::setPath(QString const& path, bool isEnabled)
{
// clear out any remnants of the previous watcher, if any
myWatchDirFiles.clear();
if (myWatcher != nullptr)
{
delete myWatcher;
myWatcher = nullptr;
}
// maybe create a new watcher
if (isEnabled)
{
myWatcher = new QFileSystemWatcher();
myWatcher->addPath(path);
connect(myWatcher, SIGNAL(directoryChanged(QString)), this, SLOT(watcherActivated(QString)));
// std::cerr << "watching " << qPrintable(path) << " for new .torrent files" << std::endl;
QTimer::singleShot(0, this, SLOT(rescanAllWatchedDirectories())); // trigger the watchdir for .torrent files in there already
}
}
void WatchDir::watcherActivated(QString const& path)
{
QDir const dir(path);
// get the list of files currently in the watch directory
QSet<QString> files;
for (QString const& str : dir.entryList(QDir::Readable | QDir::Files))
{
files.insert(str);
}
// try to add any new files which end in .torrent
QSet<QString> const newFiles(files - myWatchDirFiles);
QString const torrentSuffix = QString::fromUtf8(".torrent");
for (QString const& name : newFiles)
{
if (name.endsWith(torrentSuffix, Qt::CaseInsensitive))
{
QString const filename = dir.absoluteFilePath(name);
switch (metainfoTest(filename))
{
case OK:
emit torrentFileAdded(filename);
break;
case DUPLICATE:
break;
case ERROR:
{
// give the .torrent a few seconds to finish downloading
QTimer* t = new QTimer(this);
t->setObjectName(dir.absoluteFilePath(name));
t->setSingleShot(true);
connect(t, SIGNAL(timeout()), this, SLOT(onTimeout()));
t->start(5000);
}
}
}
}
// update our file list so that we can use it
// for comparison the next time around
myWatchDirFiles = files;
}
void WatchDir::rescanAllWatchedDirectories()
{
if (myWatcher == nullptr)
{
return;
}
for (QString const& path : myWatcher->directories())
{
watcherActivated(path);
}
}