transmission/qt/watchdir.cc

/*
 * This file Copyright (C) 2009-2010 Mnemosyne LLC
 *
 * This file is licensed by the GPL version 2.  Works owned by the
 * Transmission project are granted a special exemption to clause 2(b)
 * so that the bulk of its code can remain under the MIT license.
 * This exemption does not extend to derived works not owned by
 * the Transmission project.
 *
 * $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( 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 );
    foreach( QString name, newFiles ) {
        if( name.endsWith( ".torrent", 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;
}