transmission/qt/watchdir.cc

143 lines
3.2 KiB
C++

/*
* This file Copyright (C) 2009-2014 Mnemosyne LLC
*
* It may be used under the GNU Public License v2 or v3 licenses,
* 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(const QString&)),
this, SLOT(watcherActivated(const QString&)));
//std::cerr << "watching " << qPrintable(path) << " for new .torrent files" << std::endl;
watcherActivated (path); // 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;
}