tdelibs/kabc/tests/testlock.cpp

/*
    This file is part of libkabc.

    Copyright (c) 2003 Cornelius Schumacher <schumacher@kde.org>

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to
    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
    Boston, MA 02110-1301, USA.
*/

#include "testlock.h"

#include "stdaddressbook.h"

#include <kaboutdata.h>
#include <kapplication.h>
#include <kdebug.h>
#include <klocale.h>
#include <kcmdlineargs.h>
#include <kdirwatch.h>

#include <kmessagebox.h>
#include <kdialog.h>

#include <tqwidget.h>
#include <tqlabel.h>
#include <tqlayout.h>
#include <tqpushbutton.h>
#include <tqlistview.h>
#include <tqdir.h>

#include <iostream>

#include <sys/types.h>
#include <unistd.h>

using namespace KABC;

LockWidget::LockWidget( const TQString &identifier )
{
  TQVBoxLayout *topLayout = new TQVBoxLayout( this );
  topLayout->setMargin( KDialog::marginHint() );
  topLayout->setSpacing( KDialog::spacingHint() );

  if ( identifier.isEmpty() ) {
    mLock = 0;
  } else {
    mLock = new Lock( identifier );

    int pid = getpid();

    TQLabel *pidLabel = new TQLabel( "Process ID: " + TQString::number( pid ),
                                   this );
    topLayout->addWidget( pidLabel );

    TQHBoxLayout *identifierLayout = new TQHBoxLayout( topLayout );

    TQLabel *resourceLabel = new TQLabel( "Identifier:", this );
    identifierLayout->addWidget( resourceLabel );

    TQLabel *resourceIdentifier = new TQLabel( identifier, this );
    identifierLayout->addWidget( resourceIdentifier );

    mStatus = new TQLabel( "Status: Unlocked", this );
    topLayout->addWidget( mStatus );

    TQPushButton *button = new TQPushButton( "Lock", this );
    topLayout->addWidget( button );
    connect( button, TQT_SIGNAL( clicked() ), TQT_SLOT( lock() ) );

    button = new TQPushButton( "Unlock", this );
    topLayout->addWidget( button );
    connect( button, TQT_SIGNAL( clicked() ), TQT_SLOT( unlock() ) );
  }

  mLockView = new TQListView( this );
  topLayout->addWidget( mLockView );
  mLockView->addColumn( "Lock File" );
  mLockView->addColumn( "PID" );
  mLockView->addColumn( "Locking App" );

  updateLockView();

  TQPushButton *quitButton = new TQPushButton( "Quit", this );
  topLayout->addWidget( quitButton );
  connect( quitButton, TQT_SIGNAL( clicked() ), TQT_SLOT( close() ) );
  
  KDirWatch *watch = KDirWatch::self();
  connect( watch, TQT_SIGNAL( dirty( const TQString & ) ),
           TQT_SLOT( updateLockView() ) );
  connect( watch, TQT_SIGNAL( created( const TQString & ) ),
           TQT_SLOT( updateLockView() ) );
  connect( watch, TQT_SIGNAL( deleted( const TQString & ) ),
           TQT_SLOT( updateLockView() ) );
  watch->addDir( Lock::locksDir() );
  watch->startScan();
}

LockWidget::~LockWidget()
{
  delete mLock;
}

void LockWidget::updateLockView()
{
  mLockView->clear();
  
  TQDir dir( Lock::locksDir() );
  
  TQStringList files = dir.entryList( "*.lock" );
  
  TQStringList::ConstIterator it;
  for( it = files.begin(); it != files.end(); ++it ) {
    if ( *it == "." || *it == ".." ) continue;
    
    TQString app;
    int pid;
    if ( !Lock::readLockFile( dir.filePath( *it ), pid, app ) ) {
      kdWarning() << "Unable to open lock file '" << *it << "'" << endl; 
    } else {
      new TQListViewItem( mLockView, *it, TQString::number( pid ), app );
    }
  }
}

void LockWidget::lock()
{
  if ( !mLock->lock() ) {
    KMessageBox::sorry( this, mLock->error() );
  } else {
    mStatus->setText( "Status: Locked" );
  }
}

void LockWidget::unlock()
{
  if ( !mLock->unlock() ) {
    KMessageBox::sorry( this, mLock->error() );
  } else {
    mStatus->setText( "Status: Unlocked" );
  }
}


static const KCmdLineOptions options[] =
{
  { "a", 0, 0 },
  { "addressbook", "Standard address book", 0 },
  { "d", 0, 0 },
  { "diraddressbook", "Standard address book directory resource", 0 },
  { "+identifier", "Identifier of resource to be locked, e.g. filename", 0 },
  KCmdLineLastOption
};

int main(int argc,char **argv)
{
  KAboutData aboutData("testlock",I18N_NOOP("Test libkabc Lock"),"0.1");
  TDECmdLineArgs::init(argc,argv,&aboutData);
  TDECmdLineArgs::addCmdLineOptions( options );

  KApplication app;

  TQString identifier;

  TDECmdLineArgs *args = TDECmdLineArgs::parsedArgs();
  if ( args->count() == 1 ) {
    identifier = args->arg( 0 );
  } else if ( args->count() != 0 ) {
    std::cerr << "Usage: testlock <identifier>" << std::endl;
    return 1;
  }

  if ( args->isSet( "addressbook" ) ) {
    if ( args->count() == 1 ) {
      std::cerr << "Ignoring resource identifier" << std::endl;
    }
    identifier = StdAddressBook::fileName(); 
  }

  if ( args->isSet( "diraddressbook" ) ) {
    if ( args->count() == 1 ) {
      std::cerr << "Ignoring resource identifier" << std::endl;
    }
    identifier = StdAddressBook::directoryName(); 
  }

  LockWidget mainWidget( identifier );

  kapp->setMainWidget( &mainWidget );
  mainWidget.show();

  return app.exec();  
}

#include "testlock.moc"