File indexing completed on 2024-12-08 07:25:58

 /*
     SPDX-FileCopyrightText: 1998-2008 Sebastian Trueg <trueg@k3b.org>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "k3bdirsizejob.h"
 
 #include "k3bthread.h"
 #include "k3bthreadjob.h"
 #include "k3bsimplejobhandler.h"
 #include "k3bglobals.h"
 
 #include <QDebug>
 #include <QDir>
 #include <QFileInfo>
 
 
 class K3b::DirSizeJob::Private
 {
 public:
     Private()
         : followSymlinks(false),
           totalSize(0),
           totalFiles(0),
           totalDirs(0),
           totalSymlinks(0) {
     }
 
     QList<QUrl> urls;
     bool followSymlinks;
 
     KIO::filesize_t totalSize;
     KIO::filesize_t totalFiles;
     KIO::filesize_t totalDirs;
     KIO::filesize_t totalSymlinks;
 };
 
 
 
 K3b::DirSizeJob::DirSizeJob( QObject* parent )
     : K3b::ThreadJob( new K3b::SimpleJobHandler(), parent ),
       d( new Private() )
 {
 }
 
 
 K3b::DirSizeJob::~DirSizeJob()
 {
     delete d;
     delete jobHandler();
 }
 
 
 KIO::filesize_t K3b::DirSizeJob::totalSize() const
 {
     return d->totalSize;
 }
 
 
 KIO::filesize_t K3b::DirSizeJob::totalFiles() const
 {
     return d->totalFiles;
 }
 
 
 KIO::filesize_t K3b::DirSizeJob::totalDirs() const
 {
     return d->totalDirs;
 }
 
 
 KIO::filesize_t K3b::DirSizeJob::totalSymlinks() const
 {
     return d->totalSymlinks;
 }
 
 
 void K3b::DirSizeJob::setUrls( const QList<QUrl>& urls )
 {
     d->urls = urls;
 }
 
 
 void K3b::DirSizeJob::setFollowSymlinks( bool b )
 {
     d->followSymlinks = b;
 }
 
 bool K3b::DirSizeJob::run()
 {
     d->totalSize = 0;
     d->totalFiles = 0;
     d->totalDirs = 0;
     d->totalSymlinks = 0;
 
     QStringList l;
     for( QList<QUrl>::const_iterator it = d->urls.constBegin();
          it != d->urls.constEnd(); ++it ) {
         const QUrl& url = *it;
 
         if( !url.isLocalFile() ) {
             qDebug() << "(K3b::DirSizeJob) no remote support.";
             return false;
         }
 
         l.append( url.toLocalFile() );
     }
 
     return countFiles( l, QString() );
 }
 
 
 bool K3b::DirSizeJob::countDir( const QString& dir )
 {
     QStringList l = QDir(dir).entryList( QDir::AllEntries|QDir::Hidden|QDir::System|QDir::NoDotAndDotDot );
     return countFiles( l, dir );
 }
 
 
 bool K3b::DirSizeJob::countFiles( const QStringList& l, const QString& dir )
 {
     for( QStringList::const_iterator it = l.begin();
          it != l.end(); ++it ) {
 
         if( canceled() )
             return false;
 
         k3b_struct_stat s;
         if( k3b_lstat( QFile::encodeName( dir + *it ), &s ) )
             return false;
 
         if( S_ISLNK( s.st_mode ) ) {
             ++d->totalSymlinks;
             if( d->followSymlinks ) {
                 if( k3b_stat( QFile::encodeName( dir + *it ), &s ) )
                     return false;
             }
         }
 
         if( S_ISDIR( s.st_mode ) ) {
             ++d->totalDirs;
             if( !countDir( dir + *it + '/' ) )
                 return false;
         }
         else if( !S_ISLNK( s.st_mode ) ) {
             ++d->totalFiles;
             d->totalSize += (KIO::filesize_t)s.st_size;
         }
     }
 
     return true;
 }
 
 #include "moc_k3bdirsizejob.cpp"