simplecommandlineapp.cpp

/* This file is part of the KDE project
   Copyright (C) 2006 Jaroslaw Staniek <js@iidea.pl>
 
   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 "simplecommandlineapp.h"
 
#include <tqfileinfo.h>
 
#include <tdecmdlineargs.h>
#include <kdebug.h>
 
#include <kexidb/connectiondata.h>
#include <kexidb/drivermanager.h>
 
using namespace KexiDB;
 
static TDECmdLineOptions predefinedOptions[] =
{
    { "drv", 0, 0 },
    { "driver <name>", I18N_NOOP("Database driver name"), 0 },
    { "u", 0, 0 },
    { "user <name>", I18N_NOOP("Database user name"), 0 },
    { "p", 0, 0 },
    { "password", I18N_NOOP("Prompt for password"), 0 },
    { "h", 0, 0 },
    { "host <name>", I18N_NOOP("Host (server) name"), 0 },
    { "port <number>", I18N_NOOP("Server's port number"), 0 },
    { "s", 0, 0 },
    { "local-socket <filename>", I18N_NOOP("Server's local socket filename"), 0 },
    TDECmdLineLastOption
};
 
//-----------------------------------------
 
class SimpleCommandLineApp::Private
{
public:
    Private()
     : conn(0)
    {}
    ~Private()
    {
        if (conn) {
            conn->disconnect();
            delete (Connection*)conn;
        }
        delete instance;
 
        for (TDECmdLineOptions *optionsPtr = allOptions; optionsPtr->name; optionsPtr++) {
            delete optionsPtr->name;
            delete optionsPtr->description;
            delete optionsPtr->def;
        }
        delete allOptions;
    }
 
    KexiDB::DriverManager manager;
    TDECmdLineOptions *allOptions;
    TDEInstance* instance;
    ConnectionData connData;
    TQGuardedPtr<Connection> conn;
};
 
//-----------------------------------------
 
SimpleCommandLineApp::SimpleCommandLineApp(
    int argc, char** argv, TDECmdLineOptions *options,
    const char *programName, const char *version, 
    const char *shortDescription, int licenseType, 
    const char *copyrightStatement, const char *text, 
    const char *homePageAddress, const char *bugsEmailAddress)
 : Object()
 , d( new Private() )
{
    TQFileInfo fi(argv[0]);
    TQCString appName( fi.baseName().latin1() );
    TDECmdLineArgs::init(argc, argv, 
        new TDEAboutData( appName, programName,
            version, shortDescription, licenseType, copyrightStatement, text, 
            homePageAddress, bugsEmailAddress));
 
    int predefinedOptionsCount = 0;
    for (TDECmdLineOptions *optionsPtr = predefinedOptions; optionsPtr->name; optionsPtr++, predefinedOptionsCount++)
        ;
    int userOptionsCount = 0;
    for (TDECmdLineOptions *optionsPtr = options; optionsPtr->name; optionsPtr++, userOptionsCount++)
        ;
 
    d->instance = new TDEInstance(appName);
 
    // join the predefined options and user options
    d->allOptions = new TDECmdLineOptions[predefinedOptionsCount + userOptionsCount + 1];
    TDECmdLineOptions *allOptionsPtr = d->allOptions;
    for (TDECmdLineOptions *optionsPtr = predefinedOptions; optionsPtr->name; optionsPtr++, allOptionsPtr++) {
        allOptionsPtr->name = tqstrdup(optionsPtr->name);
        allOptionsPtr->description = tqstrdup(optionsPtr->description);
        if (optionsPtr == predefinedOptions) //first row == drv
            allOptionsPtr->def = tqstrdup(KexiDB::Driver::defaultFileBasedDriverName().latin1());
        else
            allOptionsPtr->def = tqstrdup(optionsPtr->def);
    }
    for (TDECmdLineOptions *optionsPtr = options; optionsPtr->name; optionsPtr++, allOptionsPtr++) {
        allOptionsPtr->name = tqstrdup(optionsPtr->name);
        allOptionsPtr->description = tqstrdup(optionsPtr->description);
        allOptionsPtr->def = tqstrdup(optionsPtr->def);
    }
    allOptionsPtr->name = 0; //end
    allOptionsPtr->description = 0;
    allOptionsPtr->def = 0;
    TDECmdLineArgs::addCmdLineOptions( d->allOptions );
 
    TDECmdLineArgs *args = TDECmdLineArgs::parsedArgs();
 
    d->connData.driverName = args->getOption("driver");
    d->connData.userName = args->getOption("user");
    d->connData.hostName = args->getOption("host");
    d->connData.localSocketFileName = args->getOption("local-socket");
    d->connData.port = args->getOption("port").toInt();
    d->connData.useLocalSocketFile = args->isSet("local-socket");
 
    if (args->isSet("password")) {
        TQString userAtHost = d->connData.userName;
        if (!d->connData.userName.isEmpty())
            userAtHost += "@";
        userAtHost += (d->connData.hostName.isEmpty() ? "localhost" : d->connData.hostName);
        TQTextStream cout(stdout,IO_WriteOnly);
        cout << i18n("Enter password for %1: ").arg(userAtHost);
        TQTextStream cin(stdin,IO_ReadOnly);
        cin >> d->connData.password;
        KexiDBDbg << d->connData.password << endl;
    }
}
 
SimpleCommandLineApp::~SimpleCommandLineApp()
{
    closeDatabase();
    delete d;
}
 
bool SimpleCommandLineApp::openDatabase(const TQString& databaseName)
{
    if (!d->conn) {
        if (d->manager.error()) {
            setError(&d->manager);
            return false;
        }
 
        //get the driver
        KexiDB::Driver *driver = d->manager.driver(d->connData.driverName);
        if (!driver || d->manager.error()) {
            setError(&d->manager);
            return false;
        }
 
        if (driver->isFileDriver())
            d->connData.setFileName( databaseName );
 
        d->conn = driver->createConnection(d->connData);
        if (!d->conn || driver->error()) {
            setError(driver);
            return false;
        }
    }
    if (d->conn->isConnected()) {
        // db already opened
        if (d->conn->isDatabaseUsed() && d->conn->currentDatabase()==databaseName) //the same: do nothing
            return true;
        if (!closeDatabase()) // differs: close the first
            return false;
    }
    if (!d->conn->connect()) {
        setError(d->conn);
        delete d->conn;
        d->conn = 0;
        return false;
    }
 
    if (!d->conn->useDatabase( databaseName )) {
        setError(d->conn);
        delete d->conn;
        d->conn = 0;
        return false;
    }
    return true;
}
 
bool SimpleCommandLineApp::closeDatabase()
{
    if (!d->conn)
        return true;
    if (!d->conn->disconnect()) {
        setError(d->conn);
        return false;
    }
    return true;
}
 
TDEInstance* SimpleCommandLineApp::instance() const
{
    return d->instance;
}
 
KexiDB::ConnectionData* SimpleCommandLineApp::connectionData() const
{
    return &d->connData;
}
 
KexiDB::Connection* SimpleCommandLineApp::connection() const
{
    return d->conn;
}