/*************************************************************************** krsearchmod.cpp ------------------- copyright : (C) 2001 by Shie Erlich & Rafi Yanai email : krusader@users.sourceforge.net web site : http://krusader.sourceforge.net --------------------------------------------------------------------------- Description *************************************************************************** A db dD d8888b. db db .d8888. .d8b. d8888b. d88888b d8888b. 88 ,8P' 88 `8D 88 88 88' YP d8' `8b 88 `8D 88' 88 `8D 88,8P 88oobY' 88 88 `8bo. 88ooo88 88 88 88ooooo 88oobY' 88`8b 88`8b 88 88 `Y8b. 88~~~88 88 88 88~~~~~ 88`8b 88 `88. 88 `88. 88b d88 db 8D 88 88 88 .8D 88. 88 `88. YP YD 88 YD ~Y8888P' `8888Y' YP YP Y8888D' Y88888P 88 YD S o u r c e F i l e *************************************************************************** * * * 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. * * * ***************************************************************************/ #include "krsearchmod.h" #include "../VFS/krquery.h" #include "../krusader.h" #include "../resources.h" #include "../VFS/vfile.h" #include "../VFS/krpermhandler.h" #include "../VFS/krarchandler.h" #include #include #include #include #include #include #include #include #include #include #include #include #define EVENT_PROCESS_DELAY 250 KRSearchMod::KRSearchMod( const KRQuery* q ) { stopSearch = false; /// ===> added query = new KRQuery( *q ); connect( query, TQ_SIGNAL( status( const TQString & ) ), this, TQ_SIGNAL( searching(const TQString&) ) ); connect( query, TQ_SIGNAL( processEvents( bool & ) ), this, TQ_SLOT ( slotProcessEvents( bool & ) ) ); remote_vfs = 0; virtual_vfs = 0; } KRSearchMod::~KRSearchMod() { delete query; if( remote_vfs ) delete remote_vfs; if( virtual_vfs ) delete virtual_vfs; } void KRSearchMod::start() { unScannedUrls.clear(); scannedUrls.clear(); timer.start(); KURL::List whereToSearch = query->searchInDirs(); // search every dir that needs to be searched for ( unsigned int i = 0; i < whereToSearch.count(); ++i ) scanURL( whereToSearch [ i ] ); emit finished(); } void KRSearchMod::stop() { stopSearch = true; } void KRSearchMod::scanURL( KURL url ) { if( stopSearch ) return; unScannedUrls.push( url ); while ( !unScannedUrls.isEmpty() ) { KURL urlToCheck = unScannedUrls.pop(); if( stopSearch ) return; if( query->isExcluded( urlToCheck ) ) { if( !query->searchInDirs().contains( urlToCheck ) ) continue; } if( scannedUrls.contains( urlToCheck ) ) continue; scannedUrls.push( urlToCheck ); emit searching( vfs::pathOrURL( urlToCheck ) ); if ( urlToCheck.isLocalFile() ) scanLocalDir( urlToCheck ); else scanRemoteDir( urlToCheck ); } } void KRSearchMod::scanLocalDir( KURL urlToScan ) { TQString dir = urlToScan.path( 1 ); DIR* d = opendir( dir.local8Bit() ); if ( !d ) return ; struct dirent* dirEnt; while ( ( dirEnt = readdir( d ) ) != NULL ) { TQString name = TQString::fromLocal8Bit( dirEnt->d_name ); // we dont scan the ".",".." enteries if ( name == "." || name == ".." ) continue; KDE_struct_stat stat_p; KDE_lstat( ( dir + name ).local8Bit(), &stat_p ); KURL url = vfs::fromPathOrURL( dir + name ); TQString mime = TQString(); if ( query->searchInArchives() || !query->hasMimeType() ) mime = KMimeType::findByURL( url, stat_p.st_mode, true, false ) ->name(); // creating a vfile object for matching with krquery vfile * vf = new vfile(name, (TDEIO::filesize_t)stat_p.st_size, KRpermHandler::mode2TQString(stat_p.st_mode), stat_p.st_mtime, S_ISLNK(stat_p.st_mode), stat_p.st_uid, stat_p.st_gid, mime, "", stat_p.st_mode); vf->vfile_setUrl( url ); if ( query->isRecursive() ) { if ( S_ISLNK( stat_p.st_mode ) && query->followLinks() ) unScannedUrls.push( vfs::fromPathOrURL( TQDir( dir + name ).canonicalPath() ) ); else if ( S_ISDIR( stat_p.st_mode ) ) unScannedUrls.push( url ); } if ( query->searchInArchives() ) { TQString type = mime.right( 4 ); if ( mime.contains( "-rar" ) ) type = "-rar"; if ( KRarcHandler::arcSupported( type ) ) { KURL archiveURL = url; bool encrypted; TQString realType = KRarcHandler::getType( encrypted, url.path(), mime ); if( !encrypted ) { if ( realType == "-tbz" || realType == "-tgz" || realType == "tarz" || realType == "-tar" ) archiveURL.setProtocol( "tar" ); else archiveURL.setProtocol( "krarc" ); unScannedUrls.push( archiveURL ); } } } if( query->match( vf ) ) { // if we got here - we got a winner results.append( dir + name ); emit found( name, dir, ( TDEIO::filesize_t ) stat_p.st_size, stat_p.st_mtime, KRpermHandler::mode2TQString( stat_p.st_mode ), query->foundText() ); } delete vf; if( timer.elapsed() >= EVENT_PROCESS_DELAY ) { tqApp->processEvents(); timer.start(); if( stopSearch ) return; } } // clean up closedir( d ); } void KRSearchMod::scanRemoteDir( KURL url ) { vfs * vfs_; if( url.protocol() == "virt" ) { if( virtual_vfs == 0 ) virtual_vfs = new virt_vfs( 0 ); vfs_ = virtual_vfs; } else { if( remote_vfs == 0 ) remote_vfs = new ftp_vfs( 0 ); vfs_ = remote_vfs; } if ( !vfs_->vfs_refresh( url ) ) return ; for ( vfile * vf = vfs_->vfs_getFirstFile(); vf != 0 ; vf = vfs_->vfs_getNextFile() ) { TQString name = vf->vfile_getName(); KURL fileURL = vfs_->vfs_getFile( name ); if ( query->isRecursive() && (( vf->vfile_isSymLink() && query->followLinks() ) || vf->vfile_isDir() ) ) unScannedUrls.push( fileURL ); if( query->match( vf ) ) { // if we got here - we got a winner results.append( vfs::pathOrURL( fileURL, -1 ) ); emit found( fileURL.fileName(), vfs::pathOrURL( fileURL.upURL(), -1 ), vf->vfile_getSize(), vf->vfile_getTime_t(), vf->vfile_getPerm(), query->foundText() ); } if( timer.elapsed() >= EVENT_PROCESS_DELAY ) { tqApp->processEvents(); timer.start(); if( stopSearch ) return; } } } void KRSearchMod::slotProcessEvents( bool & stopped ) { tqApp->processEvents(); stopped = stopSearch; } #include "krsearchmod.moc"