/***************************************************************************
- * Copyright (C) 2005-08 by the Quassel IRC Team *
+ * Copyright (C) 2005-09 by the Quassel Project *
* devel@quassel-irc.org *
* *
* This program is free software; you can redistribute it and/or modify *
***************************************************************************/
#include "cliparser.h"
+#include <QDir>
+#include <QDebug>
#include <QString>
#include <QFileInfo>
-CliParser::CliParser(QStringList arguments)
-{
- argsRaw = arguments;
+CliParser::CliParser() : AbstractCliParser() {
+
}
-void CliParser::addSwitch(QString longName, char shortName, QVariant def) {
- addArgument(CliParserArg::CliArgSwitch, longName, shortName, def);
+void CliParser::addArgument(const QString &longName_, const CliParserArg &arg) {
+ QString longName = longName_;
+ longName.remove(QRegExp("\\s*<.*>\\s*")); // KCmdLineArgs takes args of the form "arg <defval>"
+ if(argsHash.contains(longName)) qWarning() << "Warning: Multiple definition of argument" << longName;
+ if(arg.shortName != 0 && !lnameOfShortArg(arg.shortName).isNull())
+ qWarning().nospace() << "Warning: Redefining shortName '" << arg.shortName << "' for " << longName << " previously defined for " << lnameOfShortArg(arg.shortName);
+ argsHash.insert(longName, arg);
}
-void CliParser::addOption(QString longName, char shortName, QVariant def) {
- addArgument(CliParserArg::CliArgOption, longName, shortName, def);
+bool CliParser::addLongArg(const CliParserArg::CliArgType type, const QString &name, const QString &value) {
+ if(argsHash.contains(name)){
+ if(type == CliParserArg::CliArgOption && argsHash.value(name).type == type) {
+ argsHash[name].value = escapedValue(value);
+ return true;
+ }
+ else if (type == CliParserArg::CliArgSwitch && argsHash.value(name).type == type) {
+ argsHash[name].boolValue = true;
+ return true;
+ }
+ }
+ qWarning() << "Warning: Unrecognized argument:" << name;
+ return false;
}
-void CliParser::addArgument(CliParserArg::CliArgType type, QString longName, char shortName, QVariant def) {
- CliParserArg arg = CliParserArg(type, longName, shortName, def);
- argsHash.insert(longName, arg);
- if(shortName && !shortHash.contains(shortName)) shortHash.insert(shortName, argsHash.find(longName));
+bool CliParser::addShortArg(const CliParserArg::CliArgType type, const char shortName, const QString &value) {
+ QString longName = lnameOfShortArg(shortName);
+ if(!longName.isNull()) {
+ if(type == CliParserArg::CliArgOption && argsHash.value(longName).type == type) {
+ argsHash[longName].value = escapedValue(value);
+ return true;
+ }
+ else if (type == CliParserArg::CliArgSwitch) {
+ if(argsHash.value(longName).type == type) {
+ argsHash[longName].boolValue = true;
+ return true;
+ }
+ // arg is an option but detected as a switch -> argument is missing
+ else {
+ qWarning().nospace() << "Warning: '" << shortName << "' is an option which needs an argument";
+ return false;
+ }
+ }
+ }
+ qWarning().nospace() << "Warning: Unrecognized argument: '" << shortName << "'";
+ return false;
}
-void CliParser::addHelp(QString key, QString txt) {
- if(argsHash.contains(key)) argsHash[key].shortHelp = txt;
- else qWarning("Warning: Helptext for unknown argument '%s' given",key.toLatin1().constData());
+QString CliParser::escapedValue(const QString &value) {
+ QString escapedValue = value;
+ if(escapedValue.startsWith("~"))
+ escapedValue.replace(0, 1, QDir::homePath());
+
+ return escapedValue;
}
-bool CliParser::parse() {
+bool CliParser::init(const QStringList &args) {
+ argsRaw = args;
QStringList::const_iterator currentArg;
for (currentArg = argsRaw.constBegin(); currentArg != argsRaw.constEnd(); ++currentArg) {
if(currentArg->startsWith("--")) {
- QString name;
// long
+ QString name;
if(currentArg->contains("=")) {
// option
QStringList tmp = currentArg->mid(2).split("=");
name = tmp.at(0);
- QString value = tmp.at(1);
- if(argsHash.contains(name) && !value.isEmpty()){
- argsHash[name].value = value;
- }
- else return false;
+ QString value;
+ // this is needed to allow --option=""
+ if(tmp.at(1).isNull()) value = QString("");
+ else value = tmp.at(1);
+ if(!addLongArg(CliParserArg::CliArgOption, name, value)) return false;
}
else {
// switch
name = currentArg->mid(2);
- if(argsHash.contains(name)) {
- argsHash[name].value = true;
- }
- else return false;
+ if(!addLongArg(CliParserArg::CliArgSwitch, name)) return false;
}
}
else if(currentArg->startsWith("-")) {
- char name;
- // short
- bool bla = true;
- if(++currentArg == argsRaw.constEnd()) {
- --currentArg;
- bla = false;
- }
+ // short
+ char name;
+ QStringList::const_iterator nextArg = currentArg+1;
// if next arg is a short/long option/switch the current arg is one too
- if(currentArg->startsWith("-")) {
+ if(nextArg == argsRaw.constEnd() || nextArg->startsWith("-")) {
// switch
- if(bla) --currentArg;
for (int i = 0; i < currentArg->mid(1).toAscii().size(); i++) {
name = currentArg->mid(1).toAscii().at(i);
- if(shortHash.contains(name) && shortHash.value(name).value().type == CliParserArg::CliArgSwitch) {
- shortHash[name].value().value = true;
- }
- else return false;
+ if(!addShortArg(CliParserArg::CliArgSwitch, name)) return false;
}
}
// if next arg is is no option/switch it's an argument to a shortoption
else {
// option
- QString value = currentArg->toUtf8();
- if(bla) --currentArg;
- name = currentArg->mid(1).toAscii().at(0);
- if(bla) currentArg++;
- if(shortHash.contains(name) && shortHash.value(name).value().type == CliParserArg::CliArgOption) {
- shortHash[name].value().value = value;
+ // short options are not freely mixable with other shortargs
+ if(currentArg->mid(1).toAscii().size() > 1) {
+ qWarning() << "Warning: Shortoptions may not be combined with other shortoptions or switches";
+ return false;
}
- else return false;
+ QString value;
+ bool skipNext = false;
+ if(nextArg != argsRaw.constEnd()) {
+ value = nextArg->toLocal8Bit();
+ skipNext = true;
+ }
+ else value = currentArg->toLocal8Bit();
+ name = currentArg->mid(1).toAscii().at(0);
+ // we took one argument as argument to an option so skip it next time
+ if(skipNext) currentArg++;
+ if(!addShortArg(CliParserArg::CliArgOption, name, value)) return false;
}
}
else {
// we don't support plain arguments without -/--
- if(currentArg->toLatin1() != argsRaw.at(0)) {
- return false;
- }
+ if(currentArg->toLatin1() != argsRaw.at(0)) return false;
}
}
return true;
}
void CliParser::usage() {
- qWarning("Usage: %s [arguments]",QFileInfo(argsRaw.at(0)).completeBaseName().toLatin1().constData());
-
+ qWarning() << "Usage:" << QFileInfo(argsRaw.at(0)).completeBaseName() << "[arguments]";
+
// get size of longName field
QStringList keys = argsHash.keys();
uint lnameFieldSize = 0;
size += 8;
if(size > lnameFieldSize) lnameFieldSize = size;
}
-
+
QHash<QString, CliParserArg>::const_iterator arg;
for(arg = argsHash.constBegin(); arg != argsHash.constEnd(); ++arg) {
QString output;
QString lnameField;
-
- if(arg.value().sname) {
- output.append(" -").append(arg.value().sname).append(",");
+
+ if(arg.value().shortName) {
+ output.append(" -").append(arg.value().shortName).append(",");
}
else output.append(" ");
- lnameField.append(" --").append(arg.key().toLatin1().constData());
+ lnameField.append(" --").append(arg.key());
if(arg.value().type == CliParserArg::CliArgOption) {
- lnameField.append("=[").append(arg.value().lname.toUpper()).append("]");
+ lnameField.append("=[").append(arg.key().toUpper()).append("]");
}
output.append(lnameField.leftJustified(lnameFieldSize));
- if(!arg.value().shortHelp.isEmpty()) {
- output.append(arg.value().shortHelp);
+ if(!arg.value().help.isEmpty()) {
+ output.append(arg.value().help);
}
- if(arg.value().type == CliParserArg::CliArgOption) {
- output.append(". Default is: ").append(arg.value().def.toString());
+ if(arg.value().type == CliParserArg::CliArgOption && !arg.value().def.isNull()) {
+ output.append(". Default is: ").append(arg.value().def);
}
- qWarning(output.toLatin1());
+ qWarning() << qPrintable(output);
}
}
-QVariant CliParser::value(QString key) {
- if(argsHash.contains(key)) {
- if(argsHash.value(key).value.isValid())
- return argsHash.value(key).value;
+QString CliParser::value(const QString &longName) {
+ if(argsHash.contains(longName) && argsHash.value(longName).type == CliParserArg::CliArgOption) {
+ if(!argsHash.value(longName).value.isNull())
+ return argsHash.value(longName).value;
else
- return argsHash.value(key).def;
+ return argsHash.value(longName).def;
}
else {
- qWarning("Warning: Requested value of not defined argument '%s'",key.toLatin1().constData());
- return QVariant();
+ qWarning() << "Warning: Requested value of not defined argument" << longName << "or argument is a switch";
+ return QString();
}
}
-CliParserArg::CliParserArg(CliArgType _type, QString longName, char shortName, QVariant _def)
- : type(_type),
- lname(longName),
- sname(shortName),
- shortHelp(QString()),
- def(_def),
- value(QVariant()) {
-}
-
-CliParserArg::CliParserArg(const CliParserArg &other) {
- type = other.type;
- lname = other.lname;
- sname = other.sname;
- shortHelp = other.shortHelp;
- def = other.def;
- value = other.value;
+bool CliParser::isSet(const QString &longName) {
+ if(argsHash.contains(longName)) {
+ if(argsHash.value(longName).type == CliParserArg::CliArgOption) return !argsHash.value(longName).value.isNull();
+ else return argsHash.value(longName).boolValue;
+ }
+ else {
+ qWarning() << "Warning: Requested isSet of not defined argument" << longName;
+ return false;
+ }
}
-CliParserArg &CliParserArg::operator=(const CliParserArg &other) {
- type = other.type;
- lname = other.lname;
- sname = other.sname;
- shortHelp = other.shortHelp;
- def = other.def;
- value = other.value;
- return *this;
+QString CliParser::lnameOfShortArg(const char arg) {
+ QHash<QString, CliParserArg>::const_iterator cur;
+ for (cur = argsHash.constBegin(); cur != argsHash.constEnd(); ++cur) {
+ if(cur.value().shortName == arg) return cur.key();
+ }
+ return QString();
}