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.
274 lines
7.6 KiB
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.find(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 !!!
|
|
TDEIO::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"
|