transmission/qt/watchdir.cc

153 lines
3.4 KiB
C++

/*
* This file Copyright (C) 2009-2014 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
* $Id$
*/
#include <iostream>
#include <QDir>
#include <QFileSystemWatcher>
#include <QTimer>
#include <libtransmission/transmission.h>
#include "prefs.h"
#include "torrent-model.h"
#include "watchdir.h"
/***
****
***/
WatchDir::WatchDir (const TorrentModel& model):
myModel (model),
myWatcher (0)
{
}
WatchDir::~WatchDir ()
{
}
/***
****
***/
int
WatchDir::metainfoTest (const QString& filename) const
{
int ret;
tr_info inf;
tr_ctor * ctor = tr_ctorNew (0);
// parse
tr_ctorSetMetainfoFromFile (ctor, filename.toUtf8().constData());
const int err = tr_torrentParse( ctor, &inf );
if (err)
ret = ERROR;
else if (myModel.hasTorrent (QString::fromUtf8 (inf.hashString)))
ret = DUPLICATE;
else
ret = OK;
// cleanup
if (!err)
tr_metainfoFree (&inf);
tr_ctorFree (ctor);
return ret;
}
void
WatchDir::onTimeout ()
{
QTimer * t = qobject_cast<QTimer*>(sender());
const QString filename = t->objectName ();
if (metainfoTest (filename) == OK)
emit torrentFileAdded( filename );
t->deleteLater( );
}
void
WatchDir::setPath (const QString& path, bool isEnabled)
{
// clear out any remnants of the previous watcher, if any
myWatchDirFiles.clear ();
if (myWatcher)
{
delete myWatcher;
myWatcher = 0;
}
// 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 (const QString& path)
{
const QDir dir(path);
// get the list of files currently in the watch directory
QSet<QString> files;
foreach (QString str, dir.entryList (QDir::Readable|QDir::Files))
files.insert (str);
// try to add any new files which end in .torrent
const QSet<QString> newFiles (files - myWatchDirFiles);
const QString torrentSuffix = QString::fromUtf8 (".torrent");
foreach (QString name, newFiles)
{
if (name.endsWith (torrentSuffix, Qt::CaseInsensitive))
{
const QString 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;
foreach (const QString& path, myWatcher->directories ())
watcherActivated (path);
}