You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ktorrent/plugins/scanfolder/scanfolder.cpp

274 lines
7.6 KiB

/***************************************************************************
* Copyright (C) 2006 by Ivan Vasić *
* ivasic@gmail.com *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#include "scanfolder.h"
#include <kdirlister.h>
#include <kfileitem.h>
#include <klocale.h>
#include <kio/job.h>
#include <tqstring.h>
#include <tqobject.h>
#include <tqfile.h>
#include <tqvaluelist.h>
#include <torrent/globals.h>
#include <util/log.h>
#include <util/fileops.h>
#include <util/functions.h>
#include <util/constants.h>
#include <torrent/bnode.h>
#include <torrent/bdecoder.h>
#include <interfaces/coreinterface.h>
using namespace bt;
namespace kt
{
ScanFolder::ScanFolder(CoreInterface* core, TQString& dir, LoadedTorrentAction action, bool openSilently)
: m_core(core), m_dir(0), m_loadedAction(action), m_openSilently(openSilently)
{
m_dir = new KDirLister();
if(!m_dir->openURL(dir)) {
m_valid = false;
return;
} else
m_valid = true;
m_dir->setShowingDotFiles(true);
connect(m_dir, TQT_SIGNAL(newItems( const KFileItemList& )), this, TQT_SLOT(onNewItems( const KFileItemList& )));
connect(m_core, TQT_SIGNAL(loadingFinished( const KURL&, bool, bool )), this, TQT_SLOT(onLoadingFinished( const KURL&, bool, bool )));
connect(&m_incomplePollingTimer,TQT_SIGNAL(timeout()),this,TQT_SLOT(onIncompletePollingTimeout()));
}
ScanFolder::~ScanFolder()
{
// Out() << "UNLOADING SCANFOLDER: " << m_dir->url().path() << endl;
delete m_dir;
}
void ScanFolder::onNewItems(const KFileItemList& items)
{
KFileItemList list = items;
KFileItem* file;
for(file=list.first(); file; file=list.next())
{
TQString name = file->name();
TQString dirname = m_dir->url().path();
TQString filename = dirname + bt::DirSeparator() + name;
if(!name.endsWith(".torrent"))
continue;
if(name.startsWith("."))
{
//Check if corresponding torrent exists
if(!TQFile::exists(m_dir->url().path() + bt::DirSeparator() + name.right(name.length() - 1)) && (m_loadedAction == defaultAction))
TQFile::remove(filename);
continue;
}
KURL source;
source.setPath(filename);
//If torrent has it's hidden complement - skip it.
if(TQFile::exists(dirname + "/." + name))
continue;
if (incomplete(source))
{
// incomplete file, try this again in 10 seconds
bt::Out(SYS_SNF|LOG_NOTICE) << "ScanFolder : incomplete file " << source << endl;
m_incompleteURLs.append(source);
if (m_incompleteURLs.count() == 1)
{
// first URL so start the poll timer
// lets poll every 10 seconds
m_incomplePollingTimer.start(10000,false);
}
}
else
{
bt::Out(SYS_SNF|LOG_NOTICE) << "ScanFolder : found " << source << endl;
//Add pending entry...
m_pendingURLs.push_back(source);
//Load torrent
if(m_openSilently)
m_core->loadSilently(source);
else
m_core->load(source);
}
}
}
void ScanFolder::onLoadingFinished(const KURL & url, bool success, bool canceled)
{
if(m_pendingURLs.empty() || !success)
return;
//search for entry
TQValueList<KURL>::iterator it = m_pendingURLs.tqfind(url);
//if no entry is found than this torrent was not started by this plugin so - quit
if(it == m_pendingURLs.end())
return;
//remove this entry
m_pendingURLs.erase(it);
if(canceled)
return;
TQString name = url.filename(false);
TQString dirname = m_dir->url().path();
TQString filename = dirname + "/" + name;
KURL destination(dirname + "/" + i18n("loaded") + "/" + name);
switch(m_loadedAction) {
case deleteAction:
//If torrent has it's hidden complement - remove it too.
if(TQFile::exists(dirname + "/." + name))
TQFile::remove(dirname + "/." + name);
// Out() << "Deleting: " << name.ascii() << endl;
TQFile::remove(filename);
break;
case moveAction:
// Out() << "Moving: " << name.ascii() << endl;
//If torrent has it's hidden complement - remove it too.
if(TQFile::exists(dirname + "/." + name))
TQFile::remove(dirname + "/." + name);
// NetAccess considered harmfull !!!
KIO::file_move(url, destination);
break;
case defaultAction:
TQFile f(dirname + "/." + name);
f.open(IO_WriteOnly);
f.close();
break;
}
}
void ScanFolder::setOpenSilently(bool theValue)
{
m_openSilently = theValue;
}
void ScanFolder::setLoadedAction(const LoadedTorrentAction& theValue)
{
m_loadedAction = theValue;
TQDir tmp(m_dir->url().path());
if( (m_loadedAction == moveAction) && !tmp.exists(i18n("loaded"), false))
tmp.mkdir(i18n("loaded"), false);
}
void ScanFolder::setFolderUrl(TQString& url)
{
if(!m_dir->openURL(url)) {
m_valid = false;
return;
} else
m_valid = true;
}
bool ScanFolder::incomplete(const KURL & src)
{
// try to decode file, if it is syntactically correct, we can try to load it
TQFile fptr(src.path());
if (!fptr.open(IO_ReadOnly))
return false;
try
{
TQByteArray data(fptr.size());
fptr.readBlock(data.data(),fptr.size());
bt::BDecoder dec(data,false);
bt::BNode* n = dec.decode();
if (n)
{
// valid node, so file is complete
delete n;
return false;
}
else
{
// decoding failed so incomplete
return true;
}
}
catch (...)
{
// any error means shit happened and the file is incomplete
return true;
}
return false;
}
void ScanFolder::onIncompletePollingTimeout()
{
bt::Out(SYS_SNF|LOG_NOTICE) << "ScanFolder : checking incomplete files" << endl;
for (TQValueList<KURL>::iterator i = m_incompleteURLs.begin(); i != m_incompleteURLs.end();)
{
KURL source = *i;
if (!bt::Exists(source.path()))
{
// doesn't exist anymore, so throw out of list
i = m_incompleteURLs.erase(i);
}
else if (!incomplete(source))
{
bt::Out(SYS_SNF|LOG_NOTICE) << "ScanFolder : incomplete file " << source << " appears to be completed " << endl;
//Add pending entry...
m_pendingURLs.push_back(source);
//Load torrent
if(m_openSilently)
m_core->loadSilently(source);
else
m_core->load(source);
// remove from incomplete list
i = m_incompleteURLs.erase(i);
}
else
{
bt::Out(SYS_SNF|LOG_NOTICE) << "ScanFolder : still incomplete : " << source << endl;
i++;
}
}
// stop timer when no incomple URL's are left
if (m_incompleteURLs.count() == 0)
m_incomplePollingTimer.stop();
}
}
#include "scanfolder.moc"