quotajobs.cpp

 
#include "quotajobs.h"
#include <tdeio/scheduler.h>
#include <kdebug.h>
 
using namespace KMail;
 
QuotaJobs::GetQuotarootJob* QuotaJobs::getQuotaroot(
    TDEIO::Slave* slave, const KURL& url )
{
  TQByteArray packedArgs;
  TQDataStream stream( packedArgs, IO_WriteOnly );
  stream << (int)'Q' << (int)'R' << url;
 
  GetQuotarootJob* job = new GetQuotarootJob( url, packedArgs, false );
  TDEIO::Scheduler::assignJobToSlave( slave, job );
  return job;
}
 
QuotaJobs::GetQuotarootJob::GetQuotarootJob( const KURL& url,
                                             const TQByteArray &packedArgs,
                                             bool showProgressInfo )
  : TDEIO::SimpleJob( url, TDEIO::CMD_SPECIAL, packedArgs, showProgressInfo )
{
  connect( this, TQ_SIGNAL(infoMessage(TDEIO::Job*,const TQString&)),
           TQ_SLOT(slotInfoMessage(TDEIO::Job*,const TQString&)) );
}
 
void QuotaJobs::GetQuotarootJob::slotInfoMessage( TDEIO::Job*, const TQString& str )
{
  // Parse the result
  TQStringList results = TQStringList::split("\r", str);
  TQStringList roots;
  QuotaInfoList quotas;
  if ( results.size() > 0 ) {
    // the first line is the available roots
    roots = TQStringList::split(" ", results.front() );
    results.pop_front();
    // the rest are pairs of root -> list of triplets
    while ( results.size() > 0 ) {
      TQString root = results.front(); results.pop_front();
      // and the quotas
      if ( results.size() > 0 ) {
        TQStringList triplets = TQStringList::split(" ", results.front() );
        results.pop_front();
        while ( triplets.size() > 0 ) {
          // there's always three, the label, current and max values
          TQString name = triplets.front(); triplets.pop_front();
          TQString current = triplets.front(); triplets.pop_front();
          TQString max = triplets.front(); triplets.pop_front();
          QuotaInfo info( name, root, current, max );
          quotas.append( info );
        }
      }
    }
  }
  if ( !quotas.isEmpty() ) {
    emit quotaInfoReceived( quotas );
  }
  emit quotaRootResult( roots );
}
 
QuotaJobs::GetStorageQuotaJob* QuotaJobs::getStorageQuota(
    TDEIO::Slave* slave, const KURL& url )
{
  GetStorageQuotaJob* job = new GetStorageQuotaJob( slave, url );
  return job;
}
 
 
QuotaJobs::GetStorageQuotaJob::GetStorageQuotaJob( TDEIO::Slave* slave, const KURL& url )
  : TDEIO::Job( false )
{
    TQByteArray packedArgs;
    TQDataStream stream( packedArgs, IO_WriteOnly );
    stream << (int)'Q' << (int)'R' << url;
 
    QuotaJobs::GetQuotarootJob *job =
        new QuotaJobs::GetQuotarootJob( url, packedArgs, false );
    connect(job, TQ_SIGNAL(quotaInfoReceived(const QuotaInfoList&)),
            TQ_SLOT(slotQuotaInfoReceived(const QuotaInfoList&)));
    connect(job, TQ_SIGNAL(quotaRootResult(const TQStringList&)),
            TQ_SLOT(slotQuotarootResult(const TQStringList&)));
    TDEIO::Scheduler::assignJobToSlave( slave, job );
    addSubjob( job );
}
 
void QuotaJobs::GetStorageQuotaJob::slotQuotarootResult( const TQStringList& roots )
{
    Q_UNUSED(roots); // we only support one for now
    if ( !mStorageQuotaInfo.isValid() && !error() ) {
      // No error, so the account supports quota, but no usable info
      // was transmitted => no quota set on the folder. Make the info
      // valid, bit leave it empty.
      mStorageQuotaInfo.setName( "STORAGE" );
    }
    if ( mStorageQuotaInfo.isValid() )
      emit storageQuotaResult( mStorageQuotaInfo );
}
 
void QuotaJobs::GetStorageQuotaJob::slotQuotaInfoReceived( const QuotaInfoList& infos )
{
    QuotaInfoList::ConstIterator it( infos.begin() );
    while ( it != infos.end() ) {
      // FIXME we only use the first storage quota, for now
      if ( it->name() == "STORAGE" && !mStorageQuotaInfo.isValid() ) {
          mStorageQuotaInfo = *it;
      }
      ++it;
    }
}
 
QuotaInfo QuotaJobs::GetStorageQuotaJob::storageQuotaInfo() const
{
  return mStorageQuotaInfo;
}
 
#include "quotajobs.moc"