Amebis/wxWidgets
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
f18d7097da0296b39e7758f2ae7d4af362717beb
wxWidgets/src/common/cmdline.cpp

1674 lines
51 KiB
C++
Raw Blame History

///////////////////////////////////////////////////////////////////////////////
// Name: src/common/cmdline.cpp
// Purpose: wxCmdLineParser implementation
// Author: Vadim Zeitlin
// Modified by:
// Created: 05.01.00
// Copyright: (c) 2000 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
// Licence: wxWindows licence
///////////////////////////////////////////////////////////////////////////////
// ============================================================================
// declarations
// ============================================================================
// ----------------------------------------------------------------------------
// headers
// ----------------------------------------------------------------------------
// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
#ifdef __BORLANDC__
#pragma hdrstop
#endif
#ifndef WX_PRECOMP
#include "wx/dynarray.h"
#include "wx/string.h"
#include "wx/log.h"
#include "wx/intl.h"
#include "wx/app.h"
#endif //WX_PRECOMP
#include "wx/cmdline.h"
#if wxUSE_CMDLINE_PARSER
#include <ctype.h>
#include <locale.h> // for LC_ALL
#include "wx/datetime.h"
#include "wx/msgout.h"
#include "wx/filename.h"
#include "wx/apptrait.h"
#include "wx/scopeguard.h"
// ----------------------------------------------------------------------------
// private functions
// ----------------------------------------------------------------------------
static wxString GetTypeName(wxCmdLineParamType type);
static wxString GetOptionName(wxString::const_iterator p,
wxString::const_iterator end,
const wxChar *allowedChars);
static wxString GetShortOptionName(wxString::const_iterator p,
wxString::const_iterator end);
static wxString GetLongOptionName(wxString::const_iterator p,
wxString::const_iterator end);
// ----------------------------------------------------------------------------
// private structs
// ----------------------------------------------------------------------------
class wxCmdLineArgImpl: public wxCmdLineArg
{
public:
wxCmdLineArgImpl(wxCmdLineEntryType k,
const wxString& shrt,
const wxString& lng,
wxCmdLineParamType typ);
wxCmdLineArgImpl& SetDoubleVal(double val);
wxCmdLineArgImpl& SetLongVal(long val);
wxCmdLineArgImpl& SetStrVal(const wxString& val);
#if wxUSE_DATETIME
wxCmdLineArgImpl& SetDateVal(const wxDateTime& val);
#endif // wxUSE_DATETIME
bool HasValue() const { return m_hasVal; }
wxCmdLineArgImpl& SetHasValue() { m_hasVal = true; return *this; }
wxCmdLineArgImpl& SetNegated() { m_isNegated = true; return *this; }
// Reset to the initial state, called before parsing another command line.
void Reset()
{
m_hasVal =
m_isNegated = false;
}
public:
wxCmdLineEntryType kind;
wxString shortName,
longName;
wxCmdLineParamType type;
// from wxCmdLineArg
virtual wxCmdLineEntryType GetKind() const wxOVERRIDE { return kind; }
virtual wxString GetShortName() const wxOVERRIDE {
wxASSERT_MSG( kind == wxCMD_LINE_OPTION || kind == wxCMD_LINE_SWITCH,
wxT("kind mismatch in wxCmdLineArg") );
return shortName;
}
virtual wxString GetLongName() const wxOVERRIDE {
wxASSERT_MSG( kind == wxCMD_LINE_OPTION || kind == wxCMD_LINE_SWITCH,
wxT("kind mismatch in wxCmdLineArg") );
return longName;
}
virtual wxCmdLineParamType GetType() const wxOVERRIDE {
wxASSERT_MSG( kind == wxCMD_LINE_OPTION,
wxT("kind mismatch in wxCmdLineArg") );
return type;
}
double GetDoubleVal() const wxOVERRIDE;
long GetLongVal() const wxOVERRIDE;
const wxString& GetStrVal() const wxOVERRIDE;
#if wxUSE_DATETIME
const wxDateTime& GetDateVal() const wxOVERRIDE;
#endif // wxUSE_DATETIME
bool IsNegated() const wxOVERRIDE {
wxASSERT_MSG( kind == wxCMD_LINE_SWITCH,
wxT("kind mismatch in wxCmdLineArg") );
return m_isNegated;
}
private:
// can't use union easily here, so just store all possible data fields, we
// don't waste much (might still use union later if the number of supported
// types increases)
void Check(wxCmdLineParamType typ) const;
bool m_hasVal;
bool m_isNegated;
double m_doubleVal;
long m_longVal;
wxString m_strVal;
#if wxUSE_DATETIME
wxDateTime m_dateVal;
#endif // wxUSE_DATETIME
};
// an internal representation of an option
struct wxCmdLineOption: public wxCmdLineArgImpl
{
wxCmdLineOption(wxCmdLineEntryType k,
const wxString& shrt,
const wxString& lng,
const wxString& desc,
wxCmdLineParamType typ,
int fl)
: wxCmdLineArgImpl(k, shrt, lng, typ)
{
description = desc;
flags = fl;
}
wxString description;
int flags;
};
struct wxCmdLineParam
{
wxCmdLineParam(const wxString& desc,
wxCmdLineParamType typ,
int fl)
: description(desc)
{
type = typ;
flags = fl;
}
wxString description;
wxCmdLineParamType type;
int flags;
};
WX_DECLARE_OBJARRAY(wxCmdLineOption, wxArrayOptions);
WX_DECLARE_OBJARRAY(wxCmdLineParam, wxArrayParams);
WX_DECLARE_OBJARRAY(wxCmdLineArgImpl, wxArrayArgs);
#include "wx/arrimpl.cpp"
WX_DEFINE_OBJARRAY(wxArrayOptions)
WX_DEFINE_OBJARRAY(wxArrayParams)
WX_DEFINE_OBJARRAY(wxArrayArgs)
// the parser internal state
struct wxCmdLineParserData
{
// options
wxString m_switchChars; // characters which may start an option
bool m_enableLongOptions; // true if long options are enabled
wxString m_logo; // some extra text to show in Usage()
// cmd line data
wxArrayString m_arguments; // == argv, argc == m_arguments.GetCount()
wxArrayOptions m_options; // all possible options and switches
wxArrayParams m_paramDesc; // description of all possible params
wxArrayString m_parameters; // all params found
wxArrayArgs m_parsedArguments; // all options and parameters in parsing order
// methods
wxCmdLineParserData();
void SetArguments(int argc, char **argv);
#if wxUSE_UNICODE
void SetArguments(int argc, wxChar **argv);
void SetArguments(int argc, const wxCmdLineArgsArray& argv);
#endif // wxUSE_UNICODE
void SetArguments(const wxString& cmdline);
int FindOption(const wxString& name);
int FindOptionByLongName(const wxString& name);
// Find the option by either its short or long name.
//
// Asserts and returns NULL if option with this name is not found.
const wxCmdLineOption* FindOptionByAnyName(const wxString& name);
};
// ============================================================================
// implementation
// ============================================================================
// ----------------------------------------------------------------------------
// wxCmdLineArg
// ----------------------------------------------------------------------------
wxCmdLineArgImpl::wxCmdLineArgImpl(wxCmdLineEntryType k,
const wxString& shrt,
const wxString& lng,
wxCmdLineParamType typ)
{
// wxCMD_LINE_USAGE_TEXT uses only description, shortName and longName is empty
if ( k != wxCMD_LINE_USAGE_TEXT && k != wxCMD_LINE_PARAM)
{
wxASSERT_MSG
(
!shrt.empty() || !lng.empty(),
wxT("option should have at least one name")
);
wxASSERT_MSG
(
GetShortOptionName(shrt.begin(), shrt.end()).Len() == shrt.Len(),
wxT("Short option contains invalid characters")
);
wxASSERT_MSG
(
GetLongOptionName(lng.begin(), lng.end()).Len() == lng.Len(),
wxT("Long option contains invalid characters")
);
}
kind = k;
shortName = shrt;
longName = lng;
type = typ;
Reset();
}
void wxCmdLineArgImpl::Check(wxCmdLineParamType WXUNUSED_UNLESS_DEBUG(typ)) const
{
// NB: Type is always wxCMD_LINE_VAL_NONE for booleans, so mismatch between
// switches / options / params is well checked by this test
// The parameters have type == wxCMD_LINE_VAL_STRING and thus can be
// retrieved only by GetStrVal()
wxASSERT_MSG( type == typ, wxT("type mismatch in wxCmdLineArg") );
}
double wxCmdLineArgImpl::GetDoubleVal() const
{
Check(wxCMD_LINE_VAL_DOUBLE);
return m_doubleVal;
}
long wxCmdLineArgImpl::GetLongVal() const
{
Check(wxCMD_LINE_VAL_NUMBER);
return m_longVal;
}
const wxString& wxCmdLineArgImpl::GetStrVal() const
{
Check(wxCMD_LINE_VAL_STRING);
return m_strVal;
}
#if wxUSE_DATETIME
const wxDateTime& wxCmdLineArgImpl::GetDateVal() const
{
Check(wxCMD_LINE_VAL_DATE);
return m_dateVal;
}
#endif // wxUSE_DATETIME
wxCmdLineArgImpl& wxCmdLineArgImpl::SetDoubleVal(double val)
{
Check(wxCMD_LINE_VAL_DOUBLE);
m_doubleVal = val;
m_hasVal = true;
return *this;
}
wxCmdLineArgImpl& wxCmdLineArgImpl::SetLongVal(long val)
{
Check(wxCMD_LINE_VAL_NUMBER);
m_longVal = val;
m_hasVal = true;
return *this;
}
wxCmdLineArgImpl& wxCmdLineArgImpl::SetStrVal(const wxString& val)
{
Check(wxCMD_LINE_VAL_STRING);
m_strVal = val;
m_hasVal = true;
return *this;
}
#if wxUSE_DATETIME
wxCmdLineArgImpl& wxCmdLineArgImpl::SetDateVal(const wxDateTime& val)
{
Check(wxCMD_LINE_VAL_DATE);
m_dateVal = val;
m_hasVal = true;
return *this;
}
#endif // wxUSE_DATETIME
// ----------------------------------------------------------------------------
// wxCmdLineArgsArrayRef
// ----------------------------------------------------------------------------
size_t wxCmdLineArgs::size() const
{
return m_parser.m_data->m_parsedArguments.GetCount();
}
// ----------------------------------------------------------------------------
// wxCmdLineArgsArrayRef::const_iterator
// ----------------------------------------------------------------------------
wxCmdLineArgs::const_iterator::reference
wxCmdLineArgs::const_iterator::operator *() const
{
return m_parser->m_data->m_parsedArguments[m_index];
}
wxCmdLineArgs::const_iterator::pointer
wxCmdLineArgs::const_iterator::operator ->() const
{
return &**this;
}
wxCmdLineArgs::const_iterator &wxCmdLineArgs::const_iterator::operator ++ ()
{
++m_index;
return *this;
}
wxCmdLineArgs::const_iterator wxCmdLineArgs::const_iterator::operator ++ (int)
{
wxCmdLineArgs::const_iterator tmp(*this);
++*this;
return tmp;
}
wxCmdLineArgs::const_iterator &wxCmdLineArgs::const_iterator::operator -- ()
{
--m_index;
return *this;
}
wxCmdLineArgs::const_iterator wxCmdLineArgs::const_iterator::operator -- (int)
{
wxCmdLineArgs::const_iterator tmp(*this);
--*this;
return tmp;
}
// ----------------------------------------------------------------------------
// wxCmdLineParserData
// ----------------------------------------------------------------------------
wxCmdLineParserData::wxCmdLineParserData()
{
m_enableLongOptions = true;
#ifdef __UNIX_LIKE__
m_switchChars = wxT("-");
#else // !Unix
m_switchChars = wxT("/-");
#endif
}
namespace
{
// Small helper function setting locale for all categories.
//
// We define it because wxSetlocale() can't be easily used with wxScopeGuard as
// it has several overloads -- while this one can.
inline char *SetAllLocaleFacets(const char *loc)
{
return wxSetlocale(LC_ALL, loc);
}
} // private namespace
void wxCmdLineParserData::SetArguments(int argc, char **argv)
{
m_arguments.clear();
// Command-line arguments are supposed to be in the user locale encoding
// (what else?) but wxLocale probably wasn't initialized yet as we're
// called early during the program startup and so our locale might not have
// been set from the environment yet. To work around this problem we
// temporarily change the locale here. The only drawback is that changing
// the locale is thread-unsafe but precisely because we're called so early
// it's hopefully safe to assume that no other threads had been created yet.
char * const locOld = SetAllLocaleFacets("");
wxON_BLOCK_EXIT1( SetAllLocaleFacets, locOld );
for ( int n = 0; n < argc; n++ )
{
// try to interpret the string as being in the current locale
wxString arg(argv[n]);
// but just in case we guessed wrongly and the conversion failed, do
// try to salvage at least something
if ( arg.empty() && argv[n][0] != '\0' )
arg = wxString(argv[n], wxConvISO8859_1);
m_arguments.push_back(arg);
}
}
#if wxUSE_UNICODE
void wxCmdLineParserData::SetArguments(int argc, wxChar **argv)
{
m_arguments.clear();
for ( int n = 0; n < argc; n++ )
{
m_arguments.push_back(argv[n]);
}
}
void wxCmdLineParserData::SetArguments(int WXUNUSED(argc),
const wxCmdLineArgsArray& argv)
{
m_arguments = argv.GetArguments();
}
#endif // wxUSE_UNICODE
void wxCmdLineParserData::SetArguments(const wxString& cmdLine)
{
m_arguments.clear();
if(wxTheApp && wxTheApp->argc > 0)
m_arguments.push_back(wxTheApp->argv[0]);
else
m_arguments.push_back(wxEmptyString);
wxArrayString args = wxCmdLineParser::ConvertStringToArgs(cmdLine);
WX_APPEND_ARRAY(m_arguments, args);
}
int wxCmdLineParserData::FindOption(const wxString& name)
{
if ( !name.empty() )
{
size_t count = m_options.GetCount();
for ( size_t n = 0; n < count; n++ )
{
if ( m_options[n].shortName == name )
{
// found
return n;
}
}
}
return wxNOT_FOUND;
}
int wxCmdLineParserData::FindOptionByLongName(const wxString& name)
{
size_t count = m_options.GetCount();
for ( size_t n = 0; n < count; n++ )
{
if ( m_options[n].longName == name )
{
// found
return n;
}
}
return wxNOT_FOUND;
}
const wxCmdLineOption*
wxCmdLineParserData::FindOptionByAnyName(const wxString& name)
{
int i = FindOption(name);
if ( i == wxNOT_FOUND )
{
i = FindOptionByLongName(name);
if ( i == wxNOT_FOUND )
{
wxFAIL_MSG( wxS("Unknown option ") + name );
return NULL;
}
}
return &m_options[(size_t)i];
}
// ----------------------------------------------------------------------------
// construction and destruction
// ----------------------------------------------------------------------------
void wxCmdLineParser::Init()
{
m_data = new wxCmdLineParserData;
}
void wxCmdLineParser::SetCmdLine(int argc, char **argv)
{
m_data->SetArguments(argc, argv);
}
#if wxUSE_UNICODE
void wxCmdLineParser::SetCmdLine(int argc, wxChar **argv)
{
m_data->SetArguments(argc, argv);
}
void wxCmdLineParser::SetCmdLine(int argc, const wxCmdLineArgsArray& argv)
{
m_data->SetArguments(argc, argv);
}
#endif // wxUSE_UNICODE
void wxCmdLineParser::SetCmdLine(const wxString& cmdline)
{
m_data->SetArguments(cmdline);
}
wxCmdLineParser::~wxCmdLineParser()
{
delete m_data;
}
// ----------------------------------------------------------------------------
// options
// ----------------------------------------------------------------------------
void wxCmdLineParser::SetSwitchChars(const wxString& switchChars)
{
m_data->m_switchChars = switchChars;
}
void wxCmdLineParser::EnableLongOptions(bool enable)
{
m_data->m_enableLongOptions = enable;
}
bool wxCmdLineParser::AreLongOptionsEnabled() const
{
return m_data->m_enableLongOptions;
}
void wxCmdLineParser::SetLogo(const wxString& logo)
{
m_data->m_logo = logo;
}
// ----------------------------------------------------------------------------
// command line construction
// ----------------------------------------------------------------------------
void wxCmdLineParser::SetDesc(const wxCmdLineEntryDesc *desc)
{
for ( ;; desc++ )
{
switch ( desc->kind )
{
case wxCMD_LINE_SWITCH:
AddSwitch(desc->shortName, desc->longName,
wxGetTranslation(desc->description),
desc->flags);
break;
case wxCMD_LINE_OPTION:
AddOption(desc->shortName, desc->longName,
wxGetTranslation(desc->description),
desc->type, desc->flags);
break;
case wxCMD_LINE_PARAM:
AddParam(wxGetTranslation(desc->description),
desc->type, desc->flags);
break;
case wxCMD_LINE_USAGE_TEXT:
AddUsageText(wxGetTranslation(desc->description));
break;
default:
wxFAIL_MSG( wxT("unknown command line entry type") );
wxFALLTHROUGH;
case wxCMD_LINE_NONE:
return;
}
}
}
void wxCmdLineParser::AddSwitch(const wxString& shortName,
const wxString& longName,
const wxString& desc,
int flags)
{
wxASSERT_MSG( m_data->FindOption(shortName) == wxNOT_FOUND,
wxT("duplicate switch") );
wxCmdLineOption *option = new wxCmdLineOption(wxCMD_LINE_SWITCH,
shortName, longName, desc,
wxCMD_LINE_VAL_NONE, flags);
m_data->m_options.Add(option);
}
void wxCmdLineParser::AddOption(const wxString& shortName,
const wxString& longName,
const wxString& desc,
wxCmdLineParamType type,
int flags)
{
wxASSERT_MSG( m_data->FindOption(shortName) == wxNOT_FOUND,
wxT("duplicate option") );
wxCmdLineOption *option = new wxCmdLineOption(wxCMD_LINE_OPTION,
shortName, longName, desc,
type, flags);
m_data->m_options.Add(option);
}
void wxCmdLineParser::AddParam(const wxString& desc,
wxCmdLineParamType type,
int flags)
{
// do some consistency checks: a required parameter can't follow an
// optional one and nothing should follow a parameter with MULTIPLE flag
#if wxDEBUG_LEVEL
if ( !m_data->m_paramDesc.IsEmpty() )
{
wxCmdLineParam& param = m_data->m_paramDesc.Last();
wxASSERT_MSG( !(param.flags & wxCMD_LINE_PARAM_MULTIPLE),
wxT("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style will be ignored") );
if ( !(flags & wxCMD_LINE_PARAM_OPTIONAL) )
{
wxASSERT_MSG( !(param.flags & wxCMD_LINE_PARAM_OPTIONAL),
wxT("a required parameter can't follow an optional one") );
}
}
#endif // wxDEBUG_LEVEL
wxCmdLineParam *param = new wxCmdLineParam(desc, type, flags);
m_data->m_paramDesc.Add(param);
}
void wxCmdLineParser::AddUsageText(const wxString& text)
{
wxASSERT_MSG( !text.empty(), wxT("text can't be empty") );
wxCmdLineOption *option = new wxCmdLineOption(wxCMD_LINE_USAGE_TEXT,
wxEmptyString, wxEmptyString,
text, wxCMD_LINE_VAL_NONE, 0);
m_data->m_options.Add(option);
}
// ----------------------------------------------------------------------------
// access to parse command line
// ----------------------------------------------------------------------------
bool wxCmdLineParser::Found(const wxString& name) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
return opt && opt->HasValue();
}
wxCmdLineSwitchState wxCmdLineParser::FoundSwitch(const wxString& name) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
if ( !opt || !opt->HasValue() )
return wxCMD_SWITCH_NOT_FOUND;
return opt->IsNegated() ? wxCMD_SWITCH_OFF : wxCMD_SWITCH_ON;
}
bool wxCmdLineParser::Found(const wxString& name, wxString *value) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
if ( !opt || !opt->HasValue() )
return false;
wxCHECK_MSG( value, false, wxT("NULL pointer in wxCmdLineOption::Found") );
*value = opt->GetStrVal();
return true;
}
bool wxCmdLineParser::Found(const wxString& name, long *value) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
if ( !opt || !opt->HasValue() )
return false;
wxCHECK_MSG( value, false, wxT("NULL pointer in wxCmdLineOption::Found") );
*value = opt->GetLongVal();
return true;
}
bool wxCmdLineParser::Found(const wxString& name, double *value) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
if ( !opt || !opt->HasValue() )
return false;
wxCHECK_MSG( value, false, wxT("NULL pointer in wxCmdLineOption::Found") );
*value = opt->GetDoubleVal();
return true;
}
#if wxUSE_DATETIME
bool wxCmdLineParser::Found(const wxString& name, wxDateTime *value) const
{
const wxCmdLineOption* const opt = m_data->FindOptionByAnyName(name);
if ( !opt || !opt->HasValue() )
return false;
wxCHECK_MSG( value, false, wxT("NULL pointer in wxCmdLineOption::Found") );
*value = opt->GetDateVal();
return true;
}
#endif // wxUSE_DATETIME
size_t wxCmdLineParser::GetParamCount() const
{
return m_data->m_parameters.size();
}
wxString wxCmdLineParser::GetParam(size_t n) const
{
wxCHECK_MSG( n < GetParamCount(), wxEmptyString, wxT("invalid param index") );
return m_data->m_parameters[n];
}
// Resets switches and options
void wxCmdLineParser::Reset()
{
for ( size_t i = 0; i < m_data->m_options.GetCount(); i++ )
{
m_data->m_options[i].Reset();
}
m_data->m_parsedArguments.Empty();
}
// ----------------------------------------------------------------------------
// the real work is done here
// ----------------------------------------------------------------------------
int wxCmdLineParser::Parse(bool showUsage)
{
bool maybeOption = true; // can the following arg be an option?
bool ok = true; // true until an error is detected
bool helpRequested = false; // true if "-h" was given
bool hadRepeatableParam = false; // true if found param with MULTIPLE flag
size_t currentParam = 0; // the index in m_paramDesc
size_t countParam = m_data->m_paramDesc.GetCount();
wxString errorMsg;
Reset();
// parse everything
wxString arg;
size_t count = m_data->m_arguments.size();
for ( size_t n = 1; ok && (n < count); n++ ) // 0 is program name
{
arg = m_data->m_arguments[n];
// special case: "--" should be discarded and all following arguments
// should be considered as parameters, even if they start with '-' and
// not like options (this is POSIX-like)
if ( arg == wxT("--") )
{
maybeOption = false;
continue;
}
#ifdef __WXOSX__
if ( arg == wxT("-ApplePersistenceIgnoreState") )
{
maybeOption = false;
continue;
}
#endif
// empty argument or just '-' is not an option but a parameter
if ( maybeOption && arg.length() > 1 &&
// FIXME-UTF8: use wc_str() after removing ANSI build
wxStrchr(m_data->m_switchChars.c_str(), arg[0u]) )
{
bool isLong;
wxString name;
int optInd = wxNOT_FOUND; // init to suppress warnings
// an option or a switch: find whether it's a long or a short one
if ( arg.length() >= 3 && arg[0u] == wxT('-') && arg[1u] == wxT('-') )
{
// a long one
isLong = true;
// Skip leading "--"
wxString::const_iterator p = arg.begin() + 2;
bool longOptionsEnabled = AreLongOptionsEnabled();
name = GetLongOptionName(p, arg.end());
if (longOptionsEnabled)
{
wxString errorOpt;
optInd = m_data->FindOptionByLongName(name);
if ( optInd == wxNOT_FOUND )
{
// Check if this could be a negatable long option.
if ( name.Last() == '-' )
{
name.RemoveLast();
optInd = m_data->FindOptionByLongName(name);
if ( optInd != wxNOT_FOUND )
{
if ( !(m_data->m_options[optInd].flags &
wxCMD_LINE_SWITCH_NEGATABLE) )
{
errorOpt.Printf
(
_("Option '%s' can't be negated"),
name
);
optInd = wxNOT_FOUND;
}
}
}
if ( optInd == wxNOT_FOUND )
{
if ( errorOpt.empty() )
{
errorOpt.Printf
(
_("Unknown long option '%s'"),
name
);
}
errorMsg << errorOpt << wxT('\n');
}
}
}
else
{
optInd = wxNOT_FOUND; // Sanity check
// Print the argument including leading "--"
name.Prepend( wxT("--") );
errorMsg << wxString::Format(_("Unknown option '%s'"), name.c_str())
<< wxT('\n');
}
}
else // not a long option
{
isLong = false;
// a short one: as they can be cumulated, we try to find the
// longest substring which is a valid option
wxString::const_iterator p = arg.begin() + 1;
name = GetShortOptionName(p, arg.end());
size_t len = name.length();
do
{
if ( len == 0 )
{
// we couldn't find a valid option name in the
// beginning of this string
errorMsg << wxString::Format(_("Unknown option '%s'"), name.c_str())
<< wxT('\n');
break;
}
else
{
optInd = m_data->FindOption(name.Left(len));
// will try with one character less the next time
len--;
}
}
while ( optInd == wxNOT_FOUND );
len++; // compensates extra len-- above
if ( (optInd != wxNOT_FOUND) && (len != name.length()) )
{
// first of all, the option name is only part of this
// string
name = name.Left(len);
// our option is only part of this argument, there is
// something else in it - it is either the value of this
// option or other switches if it is a switch
if ( m_data->m_options[(size_t)optInd].kind
== wxCMD_LINE_SWITCH )
{
// if the switch is negatable and it is just followed
// by '-' the '-' is considered to be part of this
// switch
if ( (m_data->m_options[(size_t)optInd].flags &
wxCMD_LINE_SWITCH_NEGATABLE) &&
arg[len] == '-' )
++len;
// pretend that all the rest of the argument is the
// next argument, in fact
wxString arg2 = arg[0u];
arg2 += arg.Mid(len + 1); // +1 for leading '-'
m_data->m_arguments.insert
(m_data->m_arguments.begin() + n + 1, arg2);
count++;
// only leave the part which wasn't extracted into the
// next argument in this one
arg = arg.Left(len + 1);
}
//else: it's our value, we'll deal with it below
}
}
if ( optInd == wxNOT_FOUND )
{
ok = false;
continue; // will break, in fact
}
// look at what follows:
// +1 for leading '-'
wxString::const_iterator p = arg.begin() + 1 + name.length();
wxString::const_iterator end = arg.end();
if ( isLong )
++p; // for another leading '-'
wxCmdLineOption& opt = m_data->m_options[(size_t)optInd];
if ( opt.kind == wxCMD_LINE_SWITCH )
{
// we must check that there is no value following the switch
bool negated = (opt.flags & wxCMD_LINE_SWITCH_NEGATABLE) &&
p != arg.end() && *p == '-';
if ( !negated && p != arg.end() )
{
errorMsg << wxString::Format(_("Unexpected characters following option '%s'."), name.c_str())
<< wxT('\n');
ok = false;
}
else // no value, as expected
{
// nothing more to do
opt.SetHasValue();
if ( negated )
opt.SetNegated();
if ( opt.flags & wxCMD_LINE_OPTION_HELP )
{
helpRequested = true;
// it's not an error, but we still stop here
ok = false;
}
}
}
else // it's an option. not a switch
{
switch ( p == end ? '\0' : (*p).GetValue() )
{
case '=':
case ':':
// the value follows
++p;
break;
case '\0':
// the value is in the next argument
if ( ++n == count )
{
// ... but there is none
errorMsg << wxString::Format(_("Option '%s' requires a value."),
name.c_str())
<< wxT('\n');
ok = false;
}
else
{
// ... take it from there
p = m_data->m_arguments[n].begin();
end = m_data->m_arguments[n].end();
}
break;
default:
// the value is right here: this may be legal or
// not depending on the option style
if ( opt.flags & wxCMD_LINE_NEEDS_SEPARATOR )
{
errorMsg << wxString::Format(_("Separator expected after the option '%s'."),
name.c_str())
<< wxT('\n');
ok = false;
}
}
if ( ok )
{
wxString value(p, end);
switch ( opt.type )
{
default:
wxFAIL_MSG( wxT("unknown option type") );
wxFALLTHROUGH;
case wxCMD_LINE_VAL_STRING:
opt.SetStrVal(value);
break;
case wxCMD_LINE_VAL_NUMBER:
{
long val;
if ( value.ToLong(&val) )
{
opt.SetLongVal(val);
}
else
{
errorMsg << wxString::Format(_("'%s' is not a correct numeric value for option '%s'."),
value.c_str(), name.c_str())
<< wxT('\n');
ok = false;
}
}
break;
case wxCMD_LINE_VAL_DOUBLE:
{
double val;
if ( value.ToDouble(&val) )
{
opt.SetDoubleVal(val);
}
else
{
errorMsg << wxString::Format(_("'%s' is not a correct numeric value for option '%s'."),
value.c_str(), name.c_str())
<< wxT('\n');
ok = false;
}
}
break;
#if wxUSE_DATETIME
case wxCMD_LINE_VAL_DATE:
{
wxDateTime dt;
wxString::const_iterator endDate;
if ( !dt.ParseDate(value, &endDate) || endDate != value.end() )
{
errorMsg << wxString::Format(_("Option '%s': '%s' cannot be converted to a date."),
name.c_str(), value.c_str())
<< wxT('\n');
ok = false;
}
else
{
opt.SetDateVal(dt);
}
}
break;
#endif // wxUSE_DATETIME
}
}
}
if (ok)
m_data->m_parsedArguments.push_back (opt);
}
else // not an option, must be a parameter
{
if ( currentParam < countParam )
{
wxCmdLineParam& param = m_data->m_paramDesc[currentParam];
// TODO check the param type
m_data->m_parameters.push_back(arg);
m_data->m_parsedArguments.push_back (
wxCmdLineArgImpl(wxCMD_LINE_PARAM, wxString(), wxString(),
wxCMD_LINE_VAL_STRING).SetStrVal(arg));
if ( !(param.flags & wxCMD_LINE_PARAM_MULTIPLE) )
{
currentParam++;
}
else
{
wxASSERT_MSG( currentParam == countParam - 1,
wxT("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style are ignored") );
// remember that we did have this last repeatable parameter
hadRepeatableParam = true;
}
}
else
{
errorMsg << wxString::Format(_("Unexpected parameter '%s'"), arg.c_str())
<< wxT('\n');
ok = false;
}
}
}
// verify that all mandatory options were given
if ( ok )
{
size_t countOpt = m_data->m_options.GetCount();
for ( size_t n = 0; ok && (n < countOpt); n++ )
{
wxCmdLineOption& opt = m_data->m_options[n];
if ( (opt.flags & wxCMD_LINE_OPTION_MANDATORY) && !opt.HasValue() )
{
wxString optName;
if ( !opt.longName )
{
optName = opt.shortName;
}
else
{
if ( AreLongOptionsEnabled() )
{
optName.Printf( _("%s (or %s)"),
opt.shortName.c_str(),
opt.longName.c_str() );
}
else
{
optName.Printf( wxT("%s"),
opt.shortName.c_str() );
}
}
errorMsg << wxString::Format(_("The value for the option '%s' must be specified."),
optName.c_str())
<< wxT('\n');
ok = false;
}
}
for ( ; ok && (currentParam < countParam); currentParam++ )
{
wxCmdLineParam& param = m_data->m_paramDesc[currentParam];
if ( (currentParam == countParam - 1) &&
(param.flags & wxCMD_LINE_PARAM_MULTIPLE) &&
hadRepeatableParam )
{
// special case: currentParam wasn't incremented, but we did
// have it, so don't give error
continue;
}
if ( !(param.flags & wxCMD_LINE_PARAM_OPTIONAL) )
{
errorMsg << wxString::Format(_("The required parameter '%s' was not specified."),
param.description.c_str())
<< wxT('\n');
ok = false;
}
}
}
// if there was an error during parsing the command line, show this error
// and also the usage message if it had been requested
if ( !ok && (!errorMsg.empty() || (helpRequested && showUsage)) )
{
wxMessageOutput* msgOut = wxMessageOutput::Get();
if ( msgOut )
{
wxString usage;
if ( showUsage )
usage = GetUsageString();
msgOut->Printf( wxT("%s%s"), usage.c_str(), errorMsg.c_str() );
}
else
{
wxFAIL_MSG( wxT("no wxMessageOutput object?") );
}
}
return ok ? 0 : helpRequested ? -1 : 1;
}
// ----------------------------------------------------------------------------
// give the usage message
// ----------------------------------------------------------------------------
void wxCmdLineParser::Usage() const
{
wxMessageOutput* msgOut = wxMessageOutput::Get();
if ( msgOut )
{
msgOut->Printf( wxT("%s"), GetUsageString().c_str() );
}
else
{
wxFAIL_MSG( wxT("no wxMessageOutput object?") );
}
}
wxString wxCmdLineParser::GetUsageString() const
{
wxString appname;
if ( m_data->m_arguments.empty() )
{
if ( wxTheApp )
appname = wxTheApp->GetAppName();
}
else // use argv[0]
{
appname = wxFileName(m_data->m_arguments[0]).GetName();
}
// we construct the brief cmd line desc on the fly, but not the detailed
// help message below because we want to align the options descriptions
// and for this we must first know the longest one of them
wxString usage;
wxArrayString namesOptions, descOptions;
if ( !m_data->m_logo.empty() )
{
usage << m_data->m_logo << wxT('\n');
}
usage << wxString::Format(_("Usage: %s"), appname.c_str());
// the switch char is usually '-' but this can be changed with
// SetSwitchChars() and then the first one of possible chars is used
wxChar chSwitch = !m_data->m_switchChars ? wxT('-')
: m_data->m_switchChars[0u];
bool areLongOptionsEnabled = AreLongOptionsEnabled();
size_t n, count = m_data->m_options.GetCount();
for ( n = 0; n < count; n++ )
{
wxCmdLineOption& opt = m_data->m_options[n];
wxString option, negator;
if ( opt.kind != wxCMD_LINE_USAGE_TEXT )
{
usage << wxT(' ');
if ( !(opt.flags & wxCMD_LINE_OPTION_MANDATORY) )
{
usage << wxT('[');
}
if ( opt.flags & wxCMD_LINE_SWITCH_NEGATABLE )
negator = wxT("[-]");
if ( !opt.shortName.empty() )
{
usage << chSwitch << opt.shortName << negator;
}
else if ( areLongOptionsEnabled && !opt.longName.empty() )
{
usage << wxT("--") << opt.longName << negator;
}
else
{
if (!opt.longName.empty())
{
wxFAIL_MSG( wxT("option with only a long name while long ")
wxT("options are disabled") );
}
else
{
wxFAIL_MSG( wxT("option without neither short nor long name") );
}
}
if ( !opt.shortName.empty() )
{
option << wxT(" ") << chSwitch << opt.shortName;
}
if ( areLongOptionsEnabled && !opt.longName.empty() )
{
option << (option.empty() ? wxT(" ") : wxT(", "))
<< wxT("--") << opt.longName;
}
if ( opt.kind != wxCMD_LINE_SWITCH )
{
wxString val;
val << wxT('<') << GetTypeName(opt.type) << wxT('>');
usage << wxT(' ') << val;
option << (!opt.longName ? wxT(':') : wxT('=')) << val;
}
if ( !(opt.flags & wxCMD_LINE_OPTION_MANDATORY) )
{
usage << wxT(']');
}
}
namesOptions.push_back(option);
descOptions.push_back(opt.description);
}
count = m_data->m_paramDesc.GetCount();
for ( n = 0; n < count; n++ )
{
wxCmdLineParam& param = m_data->m_paramDesc[n];
usage << wxT(' ');
if ( param.flags & wxCMD_LINE_PARAM_OPTIONAL )
{
usage << wxT('[');
}
usage << param.description;
if ( param.flags & wxCMD_LINE_PARAM_MULTIPLE )
{
usage << wxT("...");
}
if ( param.flags & wxCMD_LINE_PARAM_OPTIONAL )
{
usage << wxT(']');
}
}
usage << wxT('\n');
// set to number of our own options, not counting the standard ones
count = namesOptions.size();
// get option names & descriptions for standard options, if any:
wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
wxString stdDesc;
if ( traits )
stdDesc = traits->GetStandardCmdLineOptions(namesOptions, descOptions);
// now construct the detailed help message
size_t len, lenMax = 0;
for ( n = 0; n < namesOptions.size(); n++ )
{
len = namesOptions[n].length();
if ( len > lenMax )
lenMax = len;
}
for ( n = 0; n < namesOptions.size(); n++ )
{
if ( n == count )
usage << wxT('\n') << stdDesc;
len = namesOptions[n].length();
// desc contains text if name is empty
if (len == 0)
{
usage << descOptions[n] << wxT('\n');
}
else
{
usage << namesOptions[n]
<< wxString(wxT(' '), lenMax - len) << wxT('\t')
<< descOptions[n]
<< wxT('\n');
}
}
return usage;
}
// ----------------------------------------------------------------------------
// private functions
// ----------------------------------------------------------------------------
static wxString GetTypeName(wxCmdLineParamType type)
{
wxString s;
switch ( type )
{
default:
wxFAIL_MSG( wxT("unknown option type") );
wxFALLTHROUGH;
case wxCMD_LINE_VAL_STRING:
s = _("str");
break;
case wxCMD_LINE_VAL_NUMBER:
s = _("num");
break;
case wxCMD_LINE_VAL_DOUBLE:
s = _("double");
break;
case wxCMD_LINE_VAL_DATE:
s = _("date");
break;
}
return s;
}
/*
Returns a string which is equal to the string pointed to by p, but up to the
point where p contains an character that's not allowed.
Allowable characters are letters and numbers, and characters pointed to by
the parameter allowedChars.
For example, if p points to "abcde-@-_", and allowedChars is "-_",
this function returns "abcde-".
*/
static wxString GetOptionName(wxString::const_iterator p,
wxString::const_iterator end,
const wxChar *allowedChars)
{
wxString argName;
while ( p != end && (wxIsalnum(*p) || wxStrchr(allowedChars, *p)) )
{
argName += *p++;
}
return argName;
}
// Besides alphanumeric characters, short and long options can
// have other characters.
// A short option additionally can have these
#define wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION wxT("_?")
// A long option can have the same characters as a short option and a '-'.
#define wxCMD_LINE_CHARS_ALLOWED_BY_LONG_OPTION \
wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION wxT("-")
static wxString GetShortOptionName(wxString::const_iterator p,
wxString::const_iterator end)
{
return GetOptionName(p, end, wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION);
}
static wxString GetLongOptionName(wxString::const_iterator p,
wxString::const_iterator end)
{
return GetOptionName(p, end, wxCMD_LINE_CHARS_ALLOWED_BY_LONG_OPTION);
}
#endif // wxUSE_CMDLINE_PARSER
// ----------------------------------------------------------------------------
// global functions
// ----------------------------------------------------------------------------
/*
This function is mainly used under Windows (as under Unix we always get the
command line arguments as argc/argv anyhow) and so it tries to follow
Windows conventions for the command line handling, not Unix ones. For
instance, backslash is not special except when it precedes double quote when
it does quote it.
TODO: Rewrite this to follow the even more complicated rule used by Windows
CommandLineToArgv():
* A string of backslashes not followed by a quotation mark has no special
meaning.
* An even number of backslashes followed by a quotation mark is treated as
pairs of protected backslashes, followed by a word terminator.
* An odd number of backslashes followed by a quotation mark is treated as
pairs of protected backslashes, followed by a protected quotation mark.
See http://blogs.msdn.com/b/oldnewthing/archive/2010/09/17/10063629.aspx
It could also be useful to provide a converse function which is also
non-trivial, see
http://blogs.msdn.com/b/twistylittlepassagesallalike/archive/2011/04/23/everyone-quotes-arguments-the-wrong-way.aspx
*/
/* static */
wxArrayString
wxCmdLineParser::ConvertStringToArgs(const wxString& cmdline,
wxCmdLineSplitType type)
{
wxArrayString args;
wxString arg;
arg.reserve(1024);
const wxString::const_iterator end = cmdline.end();
wxString::const_iterator p = cmdline.begin();
for ( ;; )
{
// skip white space
while ( p != end && (*p == ' ' || *p == '\t') )
++p;
// anything left?
if ( p == end )
break;
// parse this parameter
bool lastBS = false,
isInsideQuotes = false;
wxChar chDelim = '\0';
for ( arg.clear(); p != end; ++p )
{
const wxChar ch = *p;
if ( type == wxCMD_LINE_SPLIT_DOS )
{
if ( ch == '"' )
{
if ( !lastBS )
{
isInsideQuotes = !isInsideQuotes;
// don't put quote in arg
continue;
}
//else: quote has no special meaning but the backslash
// still remains -- makes no sense but this is what
// Windows does
}
// note that backslash does *not* quote the space, only quotes do
else if ( !isInsideQuotes && (ch == ' ' || ch == '\t') )
{
++p; // skip this space anyhow
break;
}
lastBS = !lastBS && ch == '\\';
}
else // type == wxCMD_LINE_SPLIT_UNIX
{
if ( !lastBS )
{
if ( isInsideQuotes )
{
if ( ch == chDelim )
{
isInsideQuotes = false;
continue; // don't use the quote itself
}
}
else // not in quotes and not escaped
{
if ( ch == '\'' || ch == '"' )
{
isInsideQuotes = true;
chDelim = ch;
continue; // don't use the quote itself
}
if ( ch == ' ' || ch == '\t' )
{
++p; // skip this space anyhow
break;
}
}
lastBS = ch == '\\';
if ( lastBS )
continue;
}
else // escaped by backslash, just use as is
{
lastBS = false;
}
}
arg += ch;
}
args.push_back(arg);
}
return args;
}
Reference in New Issue View Git Blame Copy Permalink