path: root/src/main.cpp

#include <QtGui/QApplication>
#include "dialog.h"

#include <iostream>
#include <stdlib.h>
#include <QTranslator>
#include <QMap>
#include <QString>
#include <QSettings>
#include <QtAlgorithms>
#include "save_restore_session.h"
#include "xsession.h"
#include "vsession.h"
#include "globals.h"
#include "command_line_options.h"

bool myLessThan(Session* a, Session* b) {
    return *a < *b;
}

int main(int argc, char *argv[]) {
    QApplication a(argc, argv);

    QTranslator translator;
    translator.load(":" + QLocale::system().name());
    a.installTranslator(&translator);

    std::string version = "0.0.13";

    CommandLineOptions cmdOptions(argc, argv);

    std::string usage(a.translate(
            "Console",
            "Usage: vmchooser [ OPTIONS | FILE ]\n\n"
            "  -d, --default   name of default session\n"
            "  -p, --path      path to vmware .xml files\n"
            "  -x, --xpath     path of X Session .desktop files\n"
            "  -s, --size      window size <width>x<height>\n"
            "  -v, --version   print version and exit\n"
            "  -h, --help      print usage information and exit\n"
            "\nFILE can be a vmware .xml or an X .desktop file\n")
                      .toUtf8().data());

    if (cmdOptions.contains("error")) {
        std::cerr << usage;
        return 1;
    }

    if (cmdOptions.contains("usage")) {
        std::cout << usage;
        return 0;
    }

    if (cmdOptions.contains("version")) {
        std::cout << version;
        return 0;
    }

    if (cmdOptions.contains("file")) {
        QString file(cmdOptions.value("file"));

        if (file.endsWith(".desktop")) {
            XSession s;
            return s.init(file) && s.run();
        } else if (file.endsWith(".xml")) {
            // our XML-files can contain multiple sessions
            // let's just take the first one
            Session* s(VSession::readXmlFile(file).value(0));
            return s && s->run();
        } else {
            std::cerr << "not a valid session file" << std::endl;
            return 1;
        }
    }

    // read configuration file:
    //   file supplied as command line option or
    //   user vmchooser.conf or
    //   globel vmchooser.conf
    QString confFile;
    QString userConfFile(QDir::homePath() + "/" +
                         VMCHOOSER_USER_PATH + "/" +
                         VMCHOOSER_CONF_FILE);
    QString globalConfFile(QString(VMCHOOSER_ETC_BASE_PATH) + "/" +
                           VMCHOOSER_CONF_FILE);
    if (cmdOptions.contains("config")) {
        confFile = cmdOptions.value("config");
    } else if (QFileInfo(userConfFile).exists()) {
        confFile = userConfFile;
    } else {
        confFile = globalConfFile;
    }
    QSettings settings(confFile, QSettings::IniFormat);
    settings.setIniCodec("UTF-8");

    QString defaultSession;
    if (cmdOptions.contains("default")) {
        defaultSession = cmdOptions.value("default");
    } else if (settings.contains("default")) {
        defaultSession = settings.value("default").toString();
    } else {
        defaultSession = readSessionName();
    }

    QString vSessionXmlPath;
    if (cmdOptions.contains("path")) {
        vSessionXmlPath = cmdOptions.value("path");
    } else if (settings.contains("path")) {
        vSessionXmlPath = settings.value("path").toString();
    } else {
        vSessionXmlPath = VMCHOOSER_VMPATH;
    }

    QString xSessionPath;
    if (cmdOptions.contains("xpath")) {
        xSessionPath = cmdOptions.value("xpath");
    } else if (settings.contains("xpath")) {
        xSessionPath = settings.value("xpath").toString();
    } else {
        xSessionPath = VMCHOOSER_X_SESSIONS_PATH;
    }

    QString size;
    if (cmdOptions.contains("size")) {
        size = cmdOptions.value("size");
    } else if (settings.contains("size")) {
        size = settings.value("size").toString();
    }

    int width, height;
    QRegExp rx("^(\\d+)x(\\d+)$");
    if (rx.indexIn(size) != -1) {
        QStringList list = rx.capturedTexts();
        width = list.value(1).toInt();
        height = list.value(2).toInt();
    } else if (!size.isEmpty()) {
        std::cerr
                << a.translate("Console",
                               "invlid size argument").toUtf8().data()
                << std::endl;
        exit(1);
    } else {
        width = VMCHOOSER_DEFAULT_WIDTH;
        height = VMCHOOSER_DEFAULT_HEIGHT;
    }

    /* read session files */
    QList<Session*> xsessions(XSession::readSessions(xSessionPath));
    QList<Session*> vsessions(VSession::readXmlDir(vSessionXmlPath));

    Dialog w;
    w.resize(width, height);
    if (xsessions.size()) {
        qSort(xsessions.begin(), xsessions.end(), myLessThan);
        w.addItems(xsessions, a.translate("Dialog", "X Sessions"));
    }
    if (vsessions.size()) {
        qSort(vsessions.begin(), vsessions.end(), myLessThan);
        w.addItems(vsessions, a.translate("Dialog", "Virtual Sessions"));
    }
    w.selectSession(defaultSession);
    w.show();
    return a.exec();
}