runtime/kioslave/trash/discspaceutil.cpp

   1 /*
   2    This file is part of the KDE project
   3
   4    Copyright (C) 2008 Tobias Koenig <tokoe@kde.org>
   5
   6    This library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Library General Public
   8    License as published by the Free Software Foundation; either
   9    version 2 of the License, or (at your option) any later version.
  10
  11    This library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Library General Public License for more details.
  15
  16    You should have received a copy of the GNU Library General Public License
  17    along with this library; see the file COPYING.LIB.  If not, write to
  18    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  19    Boston, MA 02110-1301, USA.
  20 */
  21
  22 #include <QtCore/QDirIterator>
  23 #include <QtCore/QFileInfo>
  24
  25 #include <kdiskfreespaceinfo.h>
  26 #include <kdebug.h>
  27 #include <kio/netaccess.h>
  28 #include <kio/directorysizejob.h>
  29
  30 #include "discspaceutil.h"
  31
  32 DiscSpaceUtil::DiscSpaceUtil( const QString &directory )
  33     : mDirectory( directory ),
  34       mFullSize( 0 )
  35 {
  36     calculateFullSize();
  37 }
  38
  39 qulonglong DiscSpaceUtil::sizeOfPath( const QString &path )
  40 {
  41     KIO::DirectorySizeJob* job = KIO::directorySize( KUrl(path) );
  42     job->setUiDelegate( 0 );
  43     bool ok = KIO::NetAccess::synchronousRun( job, 0 );
  44     return ok?job->totalSize():0;
  45 }
  46
  47 double DiscSpaceUtil::usage( qulonglong additional ) const
  48 {
  49     if ( mFullSize == 0 )
  50         return 0;
  51
  52     qulonglong sum = sizeOfPath( mDirectory );
  53     sum += additional;
  54
  55     return (((double)sum*100)/(double)mFullSize);
  56 }
  57
  58 qulonglong DiscSpaceUtil::size() const
  59 {
  60     return mFullSize;
  61 }
  62
  63 QString DiscSpaceUtil::mountPoint() const
  64 {
  65     return mMountPoint;
  66 }
  67
  68 void DiscSpaceUtil::calculateFullSize()
  69 {
  70     KDiskFreeSpaceInfo info = KDiskFreeSpaceInfo::freeSpaceInfo( mDirectory );
  71     if ( info.isValid() ) {
  72         mFullSize = info.size();
  73         mMountPoint = info.mountPoint();
  74     }
  75 }