lxqt-sudo-packaging/main.cpp

/* BEGIN_COMMON_COPYRIGHT_HEADER
 * (c)LGPL2+
 *
 * LXQt - a lightweight, Qt based, desktop toolset
 * http://lxqt.org
 *
 * Copyright: 2015 LXQt team
 * Authors:
 *   Palo Kisa <palo.kisa@gmail.com>
 *
 * This program or library is free software; you can redistribute it
 * and/or modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 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
 * Lesser General Public License for more details.

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

#include <LXQt/Application>
#include "passworddialog.h"
#include <QTextStream>
#include <QMessageBox>
#include <QDebug>

extern const QString app_master;
const QString app_master{QStringLiteral(LXQTSUDO)};
const QString app_version{QStringLiteral(LXQT_VERSION)};

extern const QString sudo_prog;
extern int sudo(QStringList const & args, PasswordDialog & dlg);

extern const QString su_prog;
extern int su(QStringList const & args, PasswordDialog & dlg);

void usage(QString const & err = QString())
{
    if (!err.isEmpty())
        QTextStream(stderr) << err << '\n';
    QTextStream(stdout)
        << QObject::tr("Usage: %1 [option] [command [arguments...]]\n\n"
                "GUI frontend for %2/%3\n\n"
                "Arguments:\n"
                "  option:\n"
                "    -h|--help      Print this help.\n"
                "    -v|--version   Print version information.\n"
                "    -s|--su        Use %3(1) as backend (instead of the default %2(8)).\n"
                "  command          Command to run.\n"
                "  arguments        Optional arguments for command.\n\n").arg(app_master).arg(sudo_prog).arg(su_prog);
    if (!err.isEmpty())
        QMessageBox(QMessageBox::Critical, app_master, err, QMessageBox::Ok).exec();
}

void version()
{
    QTextStream(stdout)
        << QObject::tr("%1 version %2\n").arg(app_master).arg(app_version);
}

enum backend_t
{
    BACK_SUDO
        , BACK_SU
};

int main(int argc, char **argv)
{
    LXQt::Application app(argc, argv, true);
    app.setQuitOnLastWindowClosed(false);
    backend_t backend = BACK_SUDO;

    QStringList args = app.arguments();
    args.removeAt(0);
    if (1 >= argc)
    {
        usage(QObject::tr("%1: no command to run provided!").arg(app_master));
        return 1;
    } else
    {
        //simple help check
        std::string arg1(argv[1]);
        if ("-h" == arg1 || "--help" == arg1)
        {
            usage();
            return 0;
        } else if ("-v" == arg1 || "--version" == arg1)
        {
            version();
            return 0;
        } else if ("-s" == arg1 || "--su" == arg1)
        {
            backend = BACK_SU;
            args.removeAt(0);
        }
        //any other arguments we simply forward to sudo
    }

    PasswordDialog dlg(args);
    dlg.setModal(true);
    lxqtApp->setActiveWindow(&dlg);

    switch (backend)
    {
        case BACK_SUDO:
            return sudo(args, dlg);
        case BACK_SU:
            return su(args, dlg);
    }
}