1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: common/cmdline.cpp
3 // Purpose: wxCmdLineParser implementation
4 // Author: Vadim Zeitlin
8 // Copyright: (c) 2000 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows license
10 ///////////////////////////////////////////////////////////////////////////////
12 // ============================================================================
14 // ============================================================================
16 // ----------------------------------------------------------------------------
18 // ----------------------------------------------------------------------------
21 #pragma implementation "cmdline.h"
24 // For compilers that support precompilation, includes "wx.h".
25 #include "wx/wxprec.h"
32 #include "wx/string.h"
36 #include "wx/dynarray.h"
37 #include "wx/filefn.h"
42 #include "wx/datetime.h"
43 #include "wx/cmdline.h"
46 // ----------------------------------------------------------------------------
48 // ----------------------------------------------------------------------------
50 static wxString
GetTypeName(wxCmdLineParamType type
);
52 // ----------------------------------------------------------------------------
54 // ----------------------------------------------------------------------------
56 // an internal representation of an option
57 struct wxCmdLineOption
59 wxCmdLineOption(wxCmdLineEntryType k
,
63 wxCmdLineParamType typ
,
78 // can't use union easily here, so just store all possible data fields, we
79 // don't waste much (might still use union later if the number of supported
80 // types increases, so always use the accessor functions and don't access
81 // the fields directly!)
83 void Check(wxCmdLineParamType
WXUNUSED_UNLESS_DEBUG(typ
)) const
85 wxASSERT_MSG( type
== typ
, _T("type mismatch in wxCmdLineOption") );
88 long GetLongVal() const
89 { Check(wxCMD_LINE_VAL_NUMBER
); return m_longVal
; }
90 const wxString
& GetStrVal() const
91 { Check(wxCMD_LINE_VAL_STRING
); return m_strVal
; }
92 const wxDateTime
& GetDateVal() const
93 { Check(wxCMD_LINE_VAL_DATE
); return m_dateVal
; }
95 void SetLongVal(long val
)
96 { Check(wxCMD_LINE_VAL_NUMBER
); m_longVal
= val
; m_hasVal
= TRUE
; }
97 void SetStrVal(const wxString
& val
)
98 { Check(wxCMD_LINE_VAL_STRING
); m_strVal
= val
; m_hasVal
= TRUE
; }
99 void SetDateVal(const wxDateTime val
)
100 { Check(wxCMD_LINE_VAL_DATE
); m_dateVal
= val
; m_hasVal
= TRUE
; }
102 void SetHasValue() { m_hasVal
= TRUE
; }
103 bool HasValue() const { return m_hasVal
; }
106 wxCmdLineEntryType kind
;
107 wxString shortName
, longName
, description
;
108 wxCmdLineParamType type
;
116 wxDateTime m_dateVal
;
119 struct wxCmdLineParam
121 wxCmdLineParam(const wxString
& desc
,
122 wxCmdLineParamType typ
,
130 wxString description
;
131 wxCmdLineParamType type
;
135 WX_DECLARE_OBJARRAY(wxCmdLineOption
, wxArrayOptions
);
136 WX_DECLARE_OBJARRAY(wxCmdLineParam
, wxArrayParams
);
138 #include "wx/arrimpl.cpp"
140 WX_DEFINE_OBJARRAY(wxArrayOptions
);
141 WX_DEFINE_OBJARRAY(wxArrayParams
);
143 // the parser internal state
144 struct wxCmdLineParserData
147 wxString m_switchChars
; // characters which may start an option
148 bool m_enableLongOptions
; // TRUE if long options are enabled
149 wxString m_logo
; // some extra text to show in Usage()
152 wxArrayString m_arguments
; // == argv, argc == m_arguments.GetCount()
153 wxArrayOptions m_options
; // all possible options and switchrs
154 wxArrayParams m_paramDesc
; // description of all possible params
155 wxArrayString m_parameters
; // all params found
158 wxCmdLineParserData();
159 void SetArguments(int argc
, char **argv
);
160 void SetArguments(const wxString
& cmdline
);
162 int FindOption(const wxString
& name
);
163 int FindOptionByLongName(const wxString
& name
);
166 // ============================================================================
168 // ============================================================================
170 // ----------------------------------------------------------------------------
171 // wxCmdLineParserData
172 // ----------------------------------------------------------------------------
174 wxCmdLineParserData
::wxCmdLineParserData()
176 m_enableLongOptions
= TRUE
;
178 m_switchChars
= _T("-");
180 m_switchChars
= _T("/-");
184 void wxCmdLineParserData
::SetArguments(int argc
, char **argv
)
188 for ( int n
= 0; n
< argc
; n
++ )
190 m_arguments
.Add(argv
[n
]);
194 void wxCmdLineParserData
::SetArguments(const wxString
& cmdLine
)
198 m_arguments
.Add(wxTheApp
->GetAppName());
201 // Treat strings enclosed in double-quotes as single arguments
203 int len
= cmdLine
.Length();
207 while ((i
< len
) && wxIsspace(cmdLine
.GetChar(i
)))
212 if (cmdLine
.GetChar(i
) == wxT('"')) // We found the start of a string
216 while ((i
< len
) && (cmdLine
.GetChar(i
) != wxT('"')))
219 wxString
arg(cmdLine
.Mid(first
, (i
- first
)));
221 m_arguments
.Add(arg
);
224 i
++; // Skip past 2nd quote
226 else // Unquoted argument
229 while ((i
< len
) && !wxIsspace(cmdLine
.GetChar(i
)))
232 wxString
arg(cmdLine
.Mid(first
, (i
- first
)));
234 m_arguments
.Add(arg
);
240 int wxCmdLineParserData
::FindOption(const wxString
& name
)
242 size_t count
= m_options
.GetCount();
243 for ( size_t n
= 0; n
< count
; n
++ )
245 if ( m_options
[n
].shortName
== name
)
255 int wxCmdLineParserData
::FindOptionByLongName(const wxString
& name
)
257 size_t count
= m_options
.GetCount();
258 for ( size_t n
= 0; n
< count
; n
++ )
260 if ( m_options
[n
].longName
== name
)
270 // ----------------------------------------------------------------------------
271 // construction and destruction
272 // ----------------------------------------------------------------------------
274 void wxCmdLineParser
::Init()
276 m_data
= new wxCmdLineParserData
;
279 void wxCmdLineParser
::SetCmdLine(int argc
, char **argv
)
281 m_data
->SetArguments(argc
, argv
);
284 void wxCmdLineParser
::SetCmdLine(const wxString
& cmdline
)
286 m_data
->SetArguments(cmdline
);
289 wxCmdLineParser
::~wxCmdLineParser()
294 // ----------------------------------------------------------------------------
296 // ----------------------------------------------------------------------------
298 void wxCmdLineParser
::SetSwitchChars(const wxString
& switchChars
)
300 m_data
->m_switchChars
= switchChars
;
303 void wxCmdLineParser
::EnableLongOptions(bool enable
)
305 m_data
->m_enableLongOptions
= enable
;
308 void wxCmdLineParser
::SetLogo(const wxString
& logo
)
310 m_data
->m_logo
= logo
;
313 // ----------------------------------------------------------------------------
314 // command line construction
315 // ----------------------------------------------------------------------------
317 void wxCmdLineParser
::SetDesc(const wxCmdLineEntryDesc
*desc
)
321 switch ( desc
->kind
)
323 case wxCMD_LINE_SWITCH
:
324 AddSwitch(desc
->shortName
, desc
->longName
, desc
->description
,
328 case wxCMD_LINE_OPTION
:
329 AddOption(desc
->shortName
, desc
->longName
, desc
->description
,
330 desc
->type
, desc
->flags
);
333 case wxCMD_LINE_PARAM
:
334 AddParam(desc
->description
, desc
->type
, desc
->flags
);
338 wxFAIL_MSG( _T("unknown command line entry type") );
339 // still fall through
341 case wxCMD_LINE_NONE
:
347 void wxCmdLineParser
::AddSwitch(const wxString
& shortName
,
348 const wxString
& longName
,
349 const wxString
& desc
,
352 wxASSERT_MSG( m_data
->FindOption(shortName
) == wxNOT_FOUND
,
353 _T("duplicate switch") );
355 wxCmdLineOption
*option
= new wxCmdLineOption(wxCMD_LINE_SWITCH
,
356 shortName
, longName
, desc
,
357 wxCMD_LINE_VAL_NONE
, flags
);
359 m_data
->m_options
.Add(option
);
362 void wxCmdLineParser
::AddOption(const wxString
& shortName
,
363 const wxString
& longName
,
364 const wxString
& desc
,
365 wxCmdLineParamType type
,
368 wxASSERT_MSG( m_data
->FindOption(shortName
) == wxNOT_FOUND
,
369 _T("duplicate option") );
371 wxCmdLineOption
*option
= new wxCmdLineOption(wxCMD_LINE_OPTION
,
372 shortName
, longName
, desc
,
375 m_data
->m_options
.Add(option
);
378 void wxCmdLineParser
::AddParam(const wxString
& desc
,
379 wxCmdLineParamType type
,
382 // do some consistency checks: a required parameter can't follow an
383 // optional one and nothing should follow a parameter with MULTIPLE flag
385 if ( !m_data
->m_paramDesc
.IsEmpty() )
387 wxCmdLineParam
& param
= m_data
->m_paramDesc
.Last();
389 wxASSERT_MSG( !(param
.flags
& wxCMD_LINE_PARAM_MULTIPLE
),
390 _T("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style will be ignored") );
392 if ( !(flags
& wxCMD_LINE_PARAM_OPTIONAL
) )
394 wxASSERT_MSG( !(param
.flags
& wxCMD_LINE_PARAM_OPTIONAL
),
395 _T("a required parameter can't follow an optional one") );
400 wxCmdLineParam
*param
= new wxCmdLineParam(desc
, type
, flags
);
402 m_data
->m_paramDesc
.Add(param
);
405 // ----------------------------------------------------------------------------
406 // access to parse command line
407 // ----------------------------------------------------------------------------
409 bool wxCmdLineParser
::Found(const wxString
& name
) const
411 int i
= m_data
->FindOption(name
);
412 wxCHECK_MSG( i
!= wxNOT_FOUND
, FALSE
, _T("unknown switch") );
414 wxCmdLineOption
& opt
= m_data
->m_options
[(size_t)i
];
415 if ( !opt
.HasValue() )
421 bool wxCmdLineParser
::Found(const wxString
& name
, wxString
*value
) const
423 int i
= m_data
->FindOption(name
);
424 wxCHECK_MSG( i
!= wxNOT_FOUND
, FALSE
, _T("unknown option") );
426 wxCmdLineOption
& opt
= m_data
->m_options
[(size_t)i
];
427 if ( !opt
.HasValue() )
430 wxCHECK_MSG( value
, FALSE
, _T("NULL pointer in wxCmdLineOption::Found") );
432 *value
= opt
.GetStrVal();
437 bool wxCmdLineParser
::Found(const wxString
& name
, long *value
) const
439 int i
= m_data
->FindOption(name
);
440 wxCHECK_MSG( i
!= wxNOT_FOUND
, FALSE
, _T("unknown option") );
442 wxCmdLineOption
& opt
= m_data
->m_options
[(size_t)i
];
443 if ( !opt
.HasValue() )
446 wxCHECK_MSG( value
, FALSE
, _T("NULL pointer in wxCmdLineOption::Found") );
448 *value
= opt
.GetLongVal();
453 bool wxCmdLineParser
::Found(const wxString
& name
, wxDateTime
*value
) const
455 int i
= m_data
->FindOption(name
);
456 wxCHECK_MSG( i
!= wxNOT_FOUND
, FALSE
, _T("unknown option") );
458 wxCmdLineOption
& opt
= m_data
->m_options
[(size_t)i
];
459 if ( !opt
.HasValue() )
462 wxCHECK_MSG( value
, FALSE
, _T("NULL pointer in wxCmdLineOption::Found") );
464 *value
= opt
.GetDateVal();
469 size_t wxCmdLineParser
::GetParamCount() const
471 return m_data
->m_parameters
.GetCount();
474 wxString wxCmdLineParser
::GetParam(size_t n
) const
476 return m_data
->m_parameters
[n
];
479 // ----------------------------------------------------------------------------
480 // the real work is done here
481 // ----------------------------------------------------------------------------
483 int wxCmdLineParser
::Parse()
485 bool maybeOption
= TRUE
; // can the following arg be an option?
486 bool ok
= TRUE
; // TRUE until an error is detected
487 bool helpRequested
= FALSE
; // TRUE if "-h" was given
488 bool hadRepeatableParam
= FALSE
; // TRUE if found param with MULTIPLE flag
490 size_t currentParam
= 0; // the index in m_paramDesc
492 size_t countParam
= m_data
->m_paramDesc
.GetCount();
496 size_t count
= m_data
->m_arguments
.GetCount();
497 for ( size_t n
= 1; ok
&& (n
< count
); n
++ ) // 0 is program name
499 arg
= m_data
->m_arguments
[n
];
501 // special case: "--" should be discarded and all following arguments
502 // should be considered as parameters, even if they start with '-' and
503 // not like options (this is POSIX-like)
504 if ( arg
== _T("--") )
511 // empty argument or just '-' is not an option but a parameter
512 if ( maybeOption
&& arg
.length() > 1 &&
513 wxStrchr(m_data
->m_switchChars
, arg
[0u]) )
517 int optInd
= wxNOT_FOUND
; // init to suppress warnings
519 // an option or a switch: find whether it's a long or a short one
520 if ( m_data
->m_enableLongOptions
&&
521 arg
[0u] == _T('-') && arg
[1u] == _T('-') )
526 const wxChar
*p
= arg
.c_str() + 2;
527 while ( wxIsalnum(*p
) || (*p
== _T('_')) || (*p
== _T('-')) )
532 optInd
= m_data
->FindOptionByLongName(name
);
533 if ( optInd
== wxNOT_FOUND
)
535 wxLogError(_("Unknown long option '%s'"), name
.c_str());
542 // a short one: as they can be cumulated, we try to find the
543 // longest substring which is a valid option
544 const wxChar
*p
= arg
.c_str() + 1;
545 while ( wxIsalnum(*p
) || (*p
== _T('_')) )
550 size_t len
= name
.length();
555 // we couldn't find a valid option name in the
556 // beginning of this string
557 wxLogError(_("Unknown option '%s'"), name
.c_str());
563 optInd
= m_data
->FindOption(name
.Left(len
));
565 // will try with one character less the next time
569 while ( optInd
== wxNOT_FOUND
);
571 len
++; // compensates extra len-- above
572 if ( (optInd
!= wxNOT_FOUND
) && (len
!= name
.length()) )
574 // first of all, the option name is only part of this
576 name
= name
.Left(len
);
578 // our option is only part of this argument, there is
579 // something else in it - it is either the value of this
580 // option or other switches if it is a switch
581 if ( m_data
->m_options
[(size_t)optInd
].kind
582 == wxCMD_LINE_SWITCH
)
584 // pretend that all the rest of the argument is the
585 // next argument, in fact
586 wxString arg2
= arg
[0u];
587 arg2
+= arg
.Mid(len
+ 1); // +1 for leading '-'
589 m_data
->m_arguments
.Insert(arg2
, n
+ 1);
592 //else: it's our value, we'll deal with it below
596 if ( optInd
== wxNOT_FOUND
)
600 continue; // will break, in fact
603 wxCmdLineOption
& opt
= m_data
->m_options
[(size_t)optInd
];
604 if ( opt
.kind
== wxCMD_LINE_SWITCH
)
606 // nothing more to do
609 if ( opt
.flags
& wxCMD_LINE_OPTION_HELP
)
611 helpRequested
= TRUE
;
613 // it's not an error, but we still stop here
621 // +1 for leading '-'
622 const wxChar
*p
= arg
.c_str() + 1 + name
.length();
625 p
++; // for another leading '-'
627 if ( *p
++ != _T('=') )
629 wxLogError(_("Option '%s' requires a value, '=' expected."), name
.c_str());
645 // the value is in the next argument
648 // ... but there is none
649 wxLogError(_("Option '%s' requires a value."),
656 // ... take it from there
657 p
= m_data
->m_arguments
[n
].c_str();
662 // the value is right here: this may be legal or
663 // not depending on the option style
664 if ( opt
.flags
& wxCMD_LINE_NEEDS_SEPARATOR
)
666 wxLogError(_("Separator expected after the option '%s'."),
680 wxFAIL_MSG( _T("unknown option type") );
681 // still fall through
683 case wxCMD_LINE_VAL_STRING
:
684 opt
.SetStrVal(value
);
687 case wxCMD_LINE_VAL_NUMBER
:
690 if ( value
.ToLong(&val
) )
696 wxLogError(_("'%s' is not a correct numeric value for option '%s'."),
697 value
.c_str(), name
.c_str());
704 case wxCMD_LINE_VAL_DATE
:
707 const wxChar
*res
= dt
.ParseDate(value
);
710 wxLogError(_("Option '%s': '%s' cannot be converted to a date."),
711 name
.c_str(), value
.c_str());
728 if ( currentParam
< countParam
)
730 wxCmdLineParam
& param
= m_data
->m_paramDesc
[currentParam
];
732 // TODO check the param type
734 m_data
->m_parameters
.Add(arg
);
736 if ( !(param
.flags
& wxCMD_LINE_PARAM_MULTIPLE
) )
742 wxASSERT_MSG( currentParam
== countParam
- 1,
743 _T("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style are ignored") );
745 // remember that we did have this last repeatable parameter
746 hadRepeatableParam
= TRUE
;
751 wxLogError(_("Unexpected parameter '%s'"), arg
.c_str());
758 // verify that all mandatory options were given
761 size_t countOpt
= m_data
->m_options
.GetCount();
762 for ( size_t n
= 0; ok
&& (n
< countOpt
); n
++ )
764 wxCmdLineOption
& opt
= m_data
->m_options
[n
];
765 if ( (opt
.flags
& wxCMD_LINE_OPTION_MANDATORY
) && !opt
.HasValue() )
770 optName
= opt
.shortName
;
774 optName
.Printf(_("%s (or %s)"),
775 opt
.shortName
.c_str(),
776 opt
.longName
.c_str());
779 wxLogError(_("The value for the option '%s' must be specified."),
786 for ( ; ok
&& (currentParam
< countParam
); currentParam
++ )
788 wxCmdLineParam
& param
= m_data
->m_paramDesc
[currentParam
];
789 if ( (currentParam
== countParam
- 1) &&
790 (param
.flags
& wxCMD_LINE_PARAM_MULTIPLE
) &&
793 // special case: currentParam wasn't incremented, but we did
794 // have it, so don't give error
798 if ( !(param
.flags
& wxCMD_LINE_PARAM_OPTIONAL
) )
800 wxLogError(_("The required parameter '%s' was not specified."),
801 param
.description
.c_str());
813 return ok ?
0 : helpRequested ?
-1 : 1;
816 // ----------------------------------------------------------------------------
817 // give the usage message
818 // ----------------------------------------------------------------------------
820 void wxCmdLineParser
::Usage()
822 wxString appname
= wxTheApp
->GetAppName();
825 wxCHECK_RET( !m_data
->m_arguments
.IsEmpty(), _T("no program name") );
827 appname
= wxFileNameFromPath(m_data
->m_arguments
[0]);
828 wxStripExtension(appname
);
831 // we construct the brief cmd line desc on the fly, but not the detailed
832 // help message below because we want to align the options descriptions
833 // and for this we must first know the longest one of them
835 wxArrayString namesOptions
, descOptions
;
836 brief
.Printf(_("Usage: %s"), appname
.c_str());
838 // the switch char is usually '-' but this can be changed with
839 // SetSwitchChars() and then the first one of possible chars is used
840 wxChar chSwitch
= !m_data
->m_switchChars ?
_T('-')
841 : m_data
->m_switchChars
[0u];
843 size_t n
, count
= m_data
->m_options
.GetCount();
844 for ( n
= 0; n
< count
; n
++ )
846 wxCmdLineOption
& opt
= m_data
->m_options
[n
];
849 if ( !(opt
.flags
& wxCMD_LINE_OPTION_MANDATORY
) )
854 brief
<< chSwitch
<< opt
.shortName
;
857 option
<< _T(" ") << chSwitch
<< opt
.shortName
;
858 if ( !!opt
.longName
)
860 option
<< _T(" --") << opt
.longName
;
863 if ( opt
.kind
!= wxCMD_LINE_SWITCH
)
866 val
<< _T('<') << GetTypeName(opt
.type
) << _T('>');
867 brief
<< _T(' ') << val
;
868 option
<< (!opt
.longName ?
_T(':') : _T('=')) << val
;
871 if ( !(opt
.flags
& wxCMD_LINE_OPTION_MANDATORY
) )
876 namesOptions
.Add(option
);
877 descOptions
.Add(opt
.description
);
880 count
= m_data
->m_paramDesc
.GetCount();
881 for ( n
= 0; n
< count
; n
++ )
883 wxCmdLineParam
& param
= m_data
->m_paramDesc
[n
];
886 if ( param
.flags
& wxCMD_LINE_PARAM_OPTIONAL
)
891 brief
<< param
.description
;
893 if ( param
.flags
& wxCMD_LINE_PARAM_MULTIPLE
)
898 if ( param
.flags
& wxCMD_LINE_PARAM_OPTIONAL
)
904 if ( !!m_data
->m_logo
)
906 wxLogMessage(m_data
->m_logo
);
911 // now construct the detailed help message
912 size_t len
, lenMax
= 0;
913 count
= namesOptions
.GetCount();
914 for ( n
= 0; n
< count
; n
++ )
916 len
= namesOptions
[n
].length();
922 for ( n
= 0; n
< count
; n
++ )
924 len
= namesOptions
[n
].length();
925 detailed
<< namesOptions
[n
]
926 << wxString(_T(' '), lenMax
- len
) << _T('\t')
931 wxLogMessage(detailed
);
934 // ----------------------------------------------------------------------------
936 // ----------------------------------------------------------------------------
938 static wxString
GetTypeName(wxCmdLineParamType type
)
944 wxFAIL_MSG( _T("unknown option type") );
945 // still fall through
947 case wxCMD_LINE_VAL_STRING
: s
= _("str"); break;
948 case wxCMD_LINE_VAL_NUMBER
: s
= _("num"); break;
949 case wxCMD_LINE_VAL_DATE
: s
= _("date"); break;