Applied patch [ 581139 ] Misc wxCmdLineParser changes/fixes
[wxWidgets.git] / src / common / cmdline.cpp
1 ///////////////////////////////////////////////////////////////////////////////
2 // Name: common/cmdline.cpp
3 // Purpose: wxCmdLineParser implementation
4 // Author: Vadim Zeitlin
5 // Modified by:
6 // Created: 05.01.00
7 // RCS-ID: $Id$
8 // Copyright: (c) 2000 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
9 // Licence: wxWindows license
10 ///////////////////////////////////////////////////////////////////////////////
11
12 // ============================================================================
13 // declarations
14 // ============================================================================
15
16 // ----------------------------------------------------------------------------
17 // headers
18 // ----------------------------------------------------------------------------
19
20 #ifdef __GNUG__
21 #pragma implementation "cmdline.h"
22 #endif
23
24 // For compilers that support precompilation, includes "wx.h".
25 #include "wx/wxprec.h"
26
27 #ifdef __BORLANDC__
28 #pragma hdrstop
29 #endif
30
31 #include "wx/cmdline.h"
32
33 #if wxUSE_CMDLINE_PARSER
34
35 #ifndef WX_PRECOMP
36 #include "wx/string.h"
37 #include "wx/log.h"
38 #include "wx/intl.h"
39 #include "wx/app.h"
40 #include "wx/dynarray.h"
41 #include "wx/filefn.h"
42 #endif //WX_PRECOMP
43
44 #include <ctype.h>
45
46 #include "wx/datetime.h"
47
48 // ----------------------------------------------------------------------------
49 // private functions
50 // ----------------------------------------------------------------------------
51
52 static wxString GetTypeName(wxCmdLineParamType type);
53
54 static wxString GetOptionName(const wxChar *p, const wxChar *allowedChars);
55
56 static wxString GetShortOptionName(const wxChar *p);
57
58 static wxString GetLongOptionName(const wxChar *p);
59
60 // ----------------------------------------------------------------------------
61 // private structs
62 // ----------------------------------------------------------------------------
63
64 // an internal representation of an option
65 struct wxCmdLineOption
66 {
67 wxCmdLineOption(wxCmdLineEntryType k,
68 const wxString& shrt,
69 const wxString& lng,
70 const wxString& desc,
71 wxCmdLineParamType typ,
72 int fl)
73 {
74 wxASSERT_MSG( !shrt.empty() || !lng.empty(),
75 _T("option should have at least one name") );
76
77 wxASSERT_MSG
78 (
79 GetShortOptionName(shrt).Len() == shrt.Len(),
80 wxT("Short option contains invalid characters")
81 );
82
83 wxASSERT_MSG
84 (
85 GetLongOptionName(lng).Len() == lng.Len(),
86 wxT("Long option contains invalid characters")
87 );
88
89 kind = k;
90
91 shortName = shrt;
92 longName = lng;
93 description = desc;
94
95 type = typ;
96 flags = fl;
97
98 m_hasVal = FALSE;
99 }
100
101 // can't use union easily here, so just store all possible data fields, we
102 // don't waste much (might still use union later if the number of supported
103 // types increases, so always use the accessor functions and don't access
104 // the fields directly!)
105
106 void Check(wxCmdLineParamType WXUNUSED_UNLESS_DEBUG(typ)) const
107 {
108 wxASSERT_MSG( type == typ, _T("type mismatch in wxCmdLineOption") );
109 }
110
111 long GetLongVal() const
112 { Check(wxCMD_LINE_VAL_NUMBER); return m_longVal; }
113 const wxString& GetStrVal() const
114 { Check(wxCMD_LINE_VAL_STRING); return m_strVal; }
115 const wxDateTime& GetDateVal() const
116 { Check(wxCMD_LINE_VAL_DATE); return m_dateVal; }
117
118 void SetLongVal(long val)
119 { Check(wxCMD_LINE_VAL_NUMBER); m_longVal = val; m_hasVal = TRUE; }
120 void SetStrVal(const wxString& val)
121 { Check(wxCMD_LINE_VAL_STRING); m_strVal = val; m_hasVal = TRUE; }
122 void SetDateVal(const wxDateTime val)
123 { Check(wxCMD_LINE_VAL_DATE); m_dateVal = val; m_hasVal = TRUE; }
124
125 void SetHasValue(bool hasValue = TRUE) { m_hasVal = hasValue; }
126 bool HasValue() const { return m_hasVal; }
127
128 public:
129 wxCmdLineEntryType kind;
130 wxString shortName,
131 longName,
132 description;
133 wxCmdLineParamType type;
134 int flags;
135
136 private:
137 bool m_hasVal;
138
139 long m_longVal;
140 wxString m_strVal;
141 wxDateTime m_dateVal;
142 };
143
144 struct wxCmdLineParam
145 {
146 wxCmdLineParam(const wxString& desc,
147 wxCmdLineParamType typ,
148 int fl)
149 : description(desc)
150 {
151 type = typ;
152 flags = fl;
153 }
154
155 wxString description;
156 wxCmdLineParamType type;
157 int flags;
158 };
159
160 WX_DECLARE_OBJARRAY(wxCmdLineOption, wxArrayOptions);
161 WX_DECLARE_OBJARRAY(wxCmdLineParam, wxArrayParams);
162
163 #include "wx/arrimpl.cpp"
164
165 WX_DEFINE_OBJARRAY(wxArrayOptions);
166 WX_DEFINE_OBJARRAY(wxArrayParams);
167
168 // the parser internal state
169 struct wxCmdLineParserData
170 {
171 // options
172 wxString m_switchChars; // characters which may start an option
173 bool m_enableLongOptions; // TRUE if long options are enabled
174 wxString m_logo; // some extra text to show in Usage()
175
176 // cmd line data
177 wxArrayString m_arguments; // == argv, argc == m_arguments.GetCount()
178 wxArrayOptions m_options; // all possible options and switchrs
179 wxArrayParams m_paramDesc; // description of all possible params
180 wxArrayString m_parameters; // all params found
181
182 // methods
183 wxCmdLineParserData();
184 void SetArguments(int argc, wxChar **argv);
185 void SetArguments(const wxString& cmdline);
186
187 int FindOption(const wxString& name);
188 int FindOptionByLongName(const wxString& name);
189 };
190
191 // ============================================================================
192 // implementation
193 // ============================================================================
194
195 // ----------------------------------------------------------------------------
196 // wxCmdLineParserData
197 // ----------------------------------------------------------------------------
198
199 wxCmdLineParserData::wxCmdLineParserData()
200 {
201 m_enableLongOptions = TRUE;
202 #ifdef __UNIX_LIKE__
203 m_switchChars = _T("-");
204 #else // !Unix
205 m_switchChars = _T("/-");
206 #endif
207 }
208
209 void wxCmdLineParserData::SetArguments(int argc, wxChar **argv)
210 {
211 m_arguments.Empty();
212
213 for ( int n = 0; n < argc; n++ )
214 {
215 m_arguments.Add(argv[n]);
216 }
217 }
218
219 void wxCmdLineParserData::SetArguments(const wxString& cmdLine)
220 {
221 m_arguments.Empty();
222
223 m_arguments.Add(wxTheApp->GetAppName());
224
225 wxArrayString args = wxCmdLineParser::ConvertStringToArgs(cmdLine);
226
227 WX_APPEND_ARRAY(m_arguments, args);
228 }
229
230 int wxCmdLineParserData::FindOption(const wxString& name)
231 {
232 if ( !name.empty() )
233 {
234 size_t count = m_options.GetCount();
235 for ( size_t n = 0; n < count; n++ )
236 {
237 if ( m_options[n].shortName == name )
238 {
239 // found
240 return n;
241 }
242 }
243 }
244
245 return wxNOT_FOUND;
246 }
247
248 int wxCmdLineParserData::FindOptionByLongName(const wxString& name)
249 {
250 size_t count = m_options.GetCount();
251 for ( size_t n = 0; n < count; n++ )
252 {
253 if ( m_options[n].longName == name )
254 {
255 // found
256 return n;
257 }
258 }
259
260 return wxNOT_FOUND;
261 }
262
263 // ----------------------------------------------------------------------------
264 // construction and destruction
265 // ----------------------------------------------------------------------------
266
267 void wxCmdLineParser::Init()
268 {
269 m_data = new wxCmdLineParserData;
270 }
271
272 void wxCmdLineParser::SetCmdLine(int argc, wxChar **argv)
273 {
274 m_data->SetArguments(argc, argv);
275 }
276
277 void wxCmdLineParser::SetCmdLine(const wxString& cmdline)
278 {
279 m_data->SetArguments(cmdline);
280 }
281
282 wxCmdLineParser::~wxCmdLineParser()
283 {
284 delete m_data;
285 }
286
287 // ----------------------------------------------------------------------------
288 // options
289 // ----------------------------------------------------------------------------
290
291 void wxCmdLineParser::SetSwitchChars(const wxString& switchChars)
292 {
293 m_data->m_switchChars = switchChars;
294 }
295
296 void wxCmdLineParser::EnableLongOptions(bool enable)
297 {
298 m_data->m_enableLongOptions = enable;
299 }
300
301 bool wxCmdLineParser::AreLongOptionsEnabled()
302 {
303 return m_data->m_enableLongOptions;
304 }
305
306 void wxCmdLineParser::SetLogo(const wxString& logo)
307 {
308 m_data->m_logo = logo;
309 }
310
311 // ----------------------------------------------------------------------------
312 // command line construction
313 // ----------------------------------------------------------------------------
314
315 void wxCmdLineParser::SetDesc(const wxCmdLineEntryDesc *desc)
316 {
317 for ( ;; desc++ )
318 {
319 switch ( desc->kind )
320 {
321 case wxCMD_LINE_SWITCH:
322 AddSwitch(desc->shortName, desc->longName, desc->description,
323 desc->flags);
324 break;
325
326 case wxCMD_LINE_OPTION:
327 AddOption(desc->shortName, desc->longName, desc->description,
328 desc->type, desc->flags);
329 break;
330
331 case wxCMD_LINE_PARAM:
332 AddParam(desc->description, desc->type, desc->flags);
333 break;
334
335 default:
336 wxFAIL_MSG( _T("unknown command line entry type") );
337 // still fall through
338
339 case wxCMD_LINE_NONE:
340 return;
341 }
342 }
343 }
344
345 void wxCmdLineParser::AddSwitch(const wxString& shortName,
346 const wxString& longName,
347 const wxString& desc,
348 int flags)
349 {
350 wxASSERT_MSG( m_data->FindOption(shortName) == wxNOT_FOUND,
351 _T("duplicate switch") );
352
353 wxCmdLineOption *option = new wxCmdLineOption(wxCMD_LINE_SWITCH,
354 shortName, longName, desc,
355 wxCMD_LINE_VAL_NONE, flags);
356
357 m_data->m_options.Add(option);
358 }
359
360 void wxCmdLineParser::AddOption(const wxString& shortName,
361 const wxString& longName,
362 const wxString& desc,
363 wxCmdLineParamType type,
364 int flags)
365 {
366 wxASSERT_MSG( m_data->FindOption(shortName) == wxNOT_FOUND,
367 _T("duplicate option") );
368
369 wxCmdLineOption *option = new wxCmdLineOption(wxCMD_LINE_OPTION,
370 shortName, longName, desc,
371 type, flags);
372
373 m_data->m_options.Add(option);
374 }
375
376 void wxCmdLineParser::AddParam(const wxString& desc,
377 wxCmdLineParamType type,
378 int flags)
379 {
380 // do some consistency checks: a required parameter can't follow an
381 // optional one and nothing should follow a parameter with MULTIPLE flag
382 #ifdef __WXDEBUG__
383 if ( !m_data->m_paramDesc.IsEmpty() )
384 {
385 wxCmdLineParam& param = m_data->m_paramDesc.Last();
386
387 wxASSERT_MSG( !(param.flags & wxCMD_LINE_PARAM_MULTIPLE),
388 _T("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style will be ignored") );
389
390 if ( !(flags & wxCMD_LINE_PARAM_OPTIONAL) )
391 {
392 wxASSERT_MSG( !(param.flags & wxCMD_LINE_PARAM_OPTIONAL),
393 _T("a required parameter can't follow an optional one") );
394 }
395 }
396 #endif // Debug
397
398 wxCmdLineParam *param = new wxCmdLineParam(desc, type, flags);
399
400 m_data->m_paramDesc.Add(param);
401 }
402
403 // ----------------------------------------------------------------------------
404 // access to parse command line
405 // ----------------------------------------------------------------------------
406
407 bool wxCmdLineParser::Found(const wxString& name) const
408 {
409 int i = m_data->FindOption(name);
410 if ( i == wxNOT_FOUND )
411 i = m_data->FindOptionByLongName(name);
412
413 wxCHECK_MSG( i != wxNOT_FOUND, FALSE, _T("unknown switch") );
414
415 wxCmdLineOption& opt = m_data->m_options[(size_t)i];
416 if ( !opt.HasValue() )
417 return FALSE;
418
419 return TRUE;
420 }
421
422 bool wxCmdLineParser::Found(const wxString& name, wxString *value) const
423 {
424 int i = m_data->FindOption(name);
425 if ( i == wxNOT_FOUND )
426 i = m_data->FindOptionByLongName(name);
427
428 wxCHECK_MSG( i != wxNOT_FOUND, FALSE, _T("unknown option") );
429
430 wxCmdLineOption& opt = m_data->m_options[(size_t)i];
431 if ( !opt.HasValue() )
432 return FALSE;
433
434 wxCHECK_MSG( value, FALSE, _T("NULL pointer in wxCmdLineOption::Found") );
435
436 *value = opt.GetStrVal();
437
438 return TRUE;
439 }
440
441 bool wxCmdLineParser::Found(const wxString& name, long *value) const
442 {
443 int i = m_data->FindOption(name);
444 if ( i == wxNOT_FOUND )
445 i = m_data->FindOptionByLongName(name);
446
447 wxCHECK_MSG( i != wxNOT_FOUND, FALSE, _T("unknown option") );
448
449 wxCmdLineOption& opt = m_data->m_options[(size_t)i];
450 if ( !opt.HasValue() )
451 return FALSE;
452
453 wxCHECK_MSG( value, FALSE, _T("NULL pointer in wxCmdLineOption::Found") );
454
455 *value = opt.GetLongVal();
456
457 return TRUE;
458 }
459
460 bool wxCmdLineParser::Found(const wxString& name, wxDateTime *value) const
461 {
462 int i = m_data->FindOption(name);
463 if ( i == wxNOT_FOUND )
464 i = m_data->FindOptionByLongName(name);
465
466 wxCHECK_MSG( i != wxNOT_FOUND, FALSE, _T("unknown option") );
467
468 wxCmdLineOption& opt = m_data->m_options[(size_t)i];
469 if ( !opt.HasValue() )
470 return FALSE;
471
472 wxCHECK_MSG( value, FALSE, _T("NULL pointer in wxCmdLineOption::Found") );
473
474 *value = opt.GetDateVal();
475
476 return TRUE;
477 }
478
479 size_t wxCmdLineParser::GetParamCount() const
480 {
481 return m_data->m_parameters.GetCount();
482 }
483
484 wxString wxCmdLineParser::GetParam(size_t n) const
485 {
486 wxCHECK_MSG( n < GetParamCount(), wxEmptyString, _T("invalid param index") );
487
488 return m_data->m_parameters[n];
489 }
490
491 // Resets switches and options
492 void wxCmdLineParser::Reset()
493 {
494 for ( size_t i = 0; i < m_data->m_options.Count(); i++ )
495 {
496 wxCmdLineOption& opt = m_data->m_options[i];
497 opt.SetHasValue(FALSE);
498 }
499 }
500
501
502 // ----------------------------------------------------------------------------
503 // the real work is done here
504 // ----------------------------------------------------------------------------
505
506 int wxCmdLineParser::Parse(bool showUsage)
507 {
508 bool maybeOption = TRUE; // can the following arg be an option?
509 bool ok = TRUE; // TRUE until an error is detected
510 bool helpRequested = FALSE; // TRUE if "-h" was given
511 bool hadRepeatableParam = FALSE; // TRUE if found param with MULTIPLE flag
512
513 size_t currentParam = 0; // the index in m_paramDesc
514
515 size_t countParam = m_data->m_paramDesc.GetCount();
516
517 Reset();
518
519 // parse everything
520 wxString arg;
521 size_t count = m_data->m_arguments.GetCount();
522 for ( size_t n = 1; ok && (n < count); n++ ) // 0 is program name
523 {
524 arg = m_data->m_arguments[n];
525
526 // special case: "--" should be discarded and all following arguments
527 // should be considered as parameters, even if they start with '-' and
528 // not like options (this is POSIX-like)
529 if ( arg == _T("--") )
530 {
531 maybeOption = FALSE;
532
533 continue;
534 }
535
536 // empty argument or just '-' is not an option but a parameter
537 if ( maybeOption && arg.length() > 1 &&
538 wxStrchr(m_data->m_switchChars, arg[0u]) )
539 {
540 bool isLong;
541 wxString name;
542 int optInd = wxNOT_FOUND; // init to suppress warnings
543
544 // an option or a switch: find whether it's a long or a short one
545 if ( arg[0u] == _T('-') && arg[1u] == _T('-') )
546 {
547 // a long one
548 isLong = TRUE;
549
550 // Skip leading "--"
551 const wxChar *p = arg.c_str() + 2;
552
553 bool longOptionsEnabled = AreLongOptionsEnabled();
554
555 name = GetLongOptionName(p);
556
557 if (longOptionsEnabled)
558 {
559 optInd = m_data->FindOptionByLongName(name);
560 if ( optInd == wxNOT_FOUND )
561 {
562 wxLogError(_("Unknown long option '%s'"), name.c_str());
563 }
564 }
565 else
566 {
567 optInd = wxNOT_FOUND; // Sanity check
568
569 // Print the argument including leading "--"
570 name.Prepend( wxT("--") );
571 wxLogError(_("Unknown option '%s'"), name.c_str());
572 }
573
574 }
575 else
576 {
577 isLong = FALSE;
578
579 // a short one: as they can be cumulated, we try to find the
580 // longest substring which is a valid option
581 const wxChar *p = arg.c_str() + 1;
582
583 name = GetShortOptionName(p);
584
585 size_t len = name.length();
586 do
587 {
588 if ( len == 0 )
589 {
590 // we couldn't find a valid option name in the
591 // beginning of this string
592 wxLogError(_("Unknown option '%s'"), name.c_str());
593
594 break;
595 }
596 else
597 {
598 optInd = m_data->FindOption(name.Left(len));
599
600 // will try with one character less the next time
601 len--;
602 }
603 }
604 while ( optInd == wxNOT_FOUND );
605
606 len++; // compensates extra len-- above
607 if ( (optInd != wxNOT_FOUND) && (len != name.length()) )
608 {
609 // first of all, the option name is only part of this
610 // string
611 name = name.Left(len);
612
613 // our option is only part of this argument, there is
614 // something else in it - it is either the value of this
615 // option or other switches if it is a switch
616 if ( m_data->m_options[(size_t)optInd].kind
617 == wxCMD_LINE_SWITCH )
618 {
619 // pretend that all the rest of the argument is the
620 // next argument, in fact
621 wxString arg2 = arg[0u];
622 arg2 += arg.Mid(len + 1); // +1 for leading '-'
623
624 m_data->m_arguments.Insert(arg2, n + 1);
625 count++;
626 }
627 //else: it's our value, we'll deal with it below
628 }
629 }
630
631 if ( optInd == wxNOT_FOUND )
632 {
633 ok = FALSE;
634
635 continue; // will break, in fact
636 }
637
638 wxCmdLineOption& opt = m_data->m_options[(size_t)optInd];
639 if ( opt.kind == wxCMD_LINE_SWITCH )
640 {
641 // nothing more to do
642 opt.SetHasValue();
643
644 if ( opt.flags & wxCMD_LINE_OPTION_HELP )
645 {
646 helpRequested = TRUE;
647
648 // it's not an error, but we still stop here
649 ok = FALSE;
650 }
651 }
652 else
653 {
654 // get the value
655
656 // +1 for leading '-'
657 const wxChar *p = arg.c_str() + 1 + name.length();
658 if ( isLong )
659 {
660 p++; // for another leading '-'
661
662 if ( *p++ != _T('=') )
663 {
664 wxLogError(_("Option '%s' requires a value, '=' expected."), name.c_str());
665
666 ok = FALSE;
667 }
668 }
669 else
670 {
671 switch ( *p )
672 {
673 case _T('='):
674 case _T(':'):
675 // the value follows
676 p++;
677 break;
678
679 case 0:
680 // the value is in the next argument
681 if ( ++n == count )
682 {
683 // ... but there is none
684 wxLogError(_("Option '%s' requires a value."),
685 name.c_str());
686
687 ok = FALSE;
688 }
689 else
690 {
691 // ... take it from there
692 p = m_data->m_arguments[n].c_str();
693 }
694 break;
695
696 default:
697 // the value is right here: this may be legal or
698 // not depending on the option style
699 if ( opt.flags & wxCMD_LINE_NEEDS_SEPARATOR )
700 {
701 wxLogError(_("Separator expected after the option '%s'."),
702 name.c_str());
703
704 ok = FALSE;
705 }
706 }
707 }
708
709 if ( ok )
710 {
711 wxString value = p;
712 switch ( opt.type )
713 {
714 default:
715 wxFAIL_MSG( _T("unknown option type") );
716 // still fall through
717
718 case wxCMD_LINE_VAL_STRING:
719 opt.SetStrVal(value);
720 break;
721
722 case wxCMD_LINE_VAL_NUMBER:
723 {
724 long val;
725 if ( value.ToLong(&val) )
726 {
727 opt.SetLongVal(val);
728 }
729 else
730 {
731 wxLogError(_("'%s' is not a correct numeric value for option '%s'."),
732 value.c_str(), name.c_str());
733
734 ok = FALSE;
735 }
736 }
737 break;
738
739 case wxCMD_LINE_VAL_DATE:
740 {
741 wxDateTime dt;
742 const wxChar *res = dt.ParseDate(value);
743 if ( !res || *res )
744 {
745 wxLogError(_("Option '%s': '%s' cannot be converted to a date."),
746 name.c_str(), value.c_str());
747
748 ok = FALSE;
749 }
750 else
751 {
752 opt.SetDateVal(dt);
753 }
754 }
755 break;
756 }
757 }
758 }
759 }
760 else
761 {
762 // a parameter
763 if ( currentParam < countParam )
764 {
765 wxCmdLineParam& param = m_data->m_paramDesc[currentParam];
766
767 // TODO check the param type
768
769 m_data->m_parameters.Add(arg);
770
771 if ( !(param.flags & wxCMD_LINE_PARAM_MULTIPLE) )
772 {
773 currentParam++;
774 }
775 else
776 {
777 wxASSERT_MSG( currentParam == countParam - 1,
778 _T("all parameters after the one with wxCMD_LINE_PARAM_MULTIPLE style are ignored") );
779
780 // remember that we did have this last repeatable parameter
781 hadRepeatableParam = TRUE;
782 }
783 }
784 else
785 {
786 wxLogError(_("Unexpected parameter '%s'"), arg.c_str());
787
788 ok = FALSE;
789 }
790 }
791 }
792
793 // verify that all mandatory options were given
794 if ( ok )
795 {
796 size_t countOpt = m_data->m_options.GetCount();
797 for ( size_t n = 0; ok && (n < countOpt); n++ )
798 {
799 wxCmdLineOption& opt = m_data->m_options[n];
800 if ( (opt.flags & wxCMD_LINE_OPTION_MANDATORY) && !opt.HasValue() )
801 {
802 wxString optName;
803 if ( !opt.longName )
804 {
805 optName = opt.shortName;
806 }
807 else
808 {
809 optName.Printf(_("%s (or %s)"),
810 opt.shortName.c_str(),
811 opt.longName.c_str());
812 }
813
814 wxLogError(_("The value for the option '%s' must be specified."),
815 optName.c_str());
816
817 ok = FALSE;
818 }
819 }
820
821 for ( ; ok && (currentParam < countParam); currentParam++ )
822 {
823 wxCmdLineParam& param = m_data->m_paramDesc[currentParam];
824 if ( (currentParam == countParam - 1) &&
825 (param.flags & wxCMD_LINE_PARAM_MULTIPLE) &&
826 hadRepeatableParam )
827 {
828 // special case: currentParam wasn't incremented, but we did
829 // have it, so don't give error
830 continue;
831 }
832
833 if ( !(param.flags & wxCMD_LINE_PARAM_OPTIONAL) )
834 {
835 wxLogError(_("The required parameter '%s' was not specified."),
836 param.description.c_str());
837
838 ok = FALSE;
839 }
840 }
841 }
842
843 if ( !ok && showUsage )
844 {
845 Usage();
846 }
847
848 return ok ? 0 : helpRequested ? -1 : 1;
849 }
850
851 // ----------------------------------------------------------------------------
852 // give the usage message
853 // ----------------------------------------------------------------------------
854
855 void wxCmdLineParser::Usage()
856 {
857 wxString appname = wxTheApp->GetAppName();
858 if ( !appname )
859 {
860 wxCHECK_RET( !m_data->m_arguments.IsEmpty(), _T("no program name") );
861
862 appname = wxFileNameFromPath(m_data->m_arguments[0]);
863 wxStripExtension(appname);
864 }
865
866 // we construct the brief cmd line desc on the fly, but not the detailed
867 // help message below because we want to align the options descriptions
868 // and for this we must first know the longest one of them
869 wxString brief;
870 wxArrayString namesOptions, descOptions;
871 brief.Printf(_("Usage: %s"), appname.c_str());
872
873 // the switch char is usually '-' but this can be changed with
874 // SetSwitchChars() and then the first one of possible chars is used
875 wxChar chSwitch = !m_data->m_switchChars ? _T('-')
876 : m_data->m_switchChars[0u];
877
878 bool areLongOptionsEnabled = AreLongOptionsEnabled();
879 size_t n, count = m_data->m_options.GetCount();
880 for ( n = 0; n < count; n++ )
881 {
882 wxCmdLineOption& opt = m_data->m_options[n];
883
884 brief << _T(' ');
885 if ( !(opt.flags & wxCMD_LINE_OPTION_MANDATORY) )
886 {
887 brief << _T('[');
888 }
889
890 if ( !opt.shortName.empty() )
891 {
892 brief << chSwitch << opt.shortName;
893 }
894 else if ( areLongOptionsEnabled && !opt.longName.empty() )
895 {
896 brief << _T("--") << opt.longName;
897 }
898 else
899 {
900 if (!opt.longName.empty())
901 {
902 wxFAIL_MSG( wxT("option with only a long name while long ")
903 wxT("options are disabled") );
904 }
905 else
906 {
907 wxFAIL_MSG( _T("option without neither short nor long name") );
908 }
909 }
910
911 wxString option;
912
913 if ( !opt.shortName.empty() )
914 {
915 option << _T(" ") << chSwitch << opt.shortName;
916 }
917
918 if ( areLongOptionsEnabled && !opt.longName.empty() )
919 {
920 option << (option.empty() ? _T(" ") : _T(", "))
921 << _T("--") << opt.longName;
922 }
923
924 if ( opt.kind != wxCMD_LINE_SWITCH )
925 {
926 wxString val;
927 val << _T('<') << GetTypeName(opt.type) << _T('>');
928 brief << _T(' ') << val;
929 option << (!opt.longName ? _T(':') : _T('=')) << val;
930 }
931
932 if ( !(opt.flags & wxCMD_LINE_OPTION_MANDATORY) )
933 {
934 brief << _T(']');
935 }
936
937 namesOptions.Add(option);
938 descOptions.Add(opt.description);
939 }
940
941 count = m_data->m_paramDesc.GetCount();
942 for ( n = 0; n < count; n++ )
943 {
944 wxCmdLineParam& param = m_data->m_paramDesc[n];
945
946 brief << _T(' ');
947 if ( param.flags & wxCMD_LINE_PARAM_OPTIONAL )
948 {
949 brief << _T('[');
950 }
951
952 brief << param.description;
953
954 if ( param.flags & wxCMD_LINE_PARAM_MULTIPLE )
955 {
956 brief << _T("...");
957 }
958
959 if ( param.flags & wxCMD_LINE_PARAM_OPTIONAL )
960 {
961 brief << _T(']');
962 }
963 }
964
965 if ( !!m_data->m_logo )
966 {
967 wxLogMessage(m_data->m_logo);
968 }
969
970 // in console mode we want to show the brief usage message first, then the
971 // detailed one but in GUI build we give the details first and then the
972 // summary - like this, the brief message appears in the wxLogGui dialog,
973 // as expected
974 #if !wxUSE_GUI
975 wxLogMessage(brief);
976 #endif // !wxUSE_GUI
977
978 // now construct the detailed help message
979 size_t len, lenMax = 0;
980 count = namesOptions.GetCount();
981 for ( n = 0; n < count; n++ )
982 {
983 len = namesOptions[n].length();
984 if ( len > lenMax )
985 lenMax = len;
986 }
987
988 wxString detailed;
989 for ( n = 0; n < count; n++ )
990 {
991 len = namesOptions[n].length();
992 detailed << namesOptions[n]
993 << wxString(_T(' '), lenMax - len) << _T('\t')
994 << descOptions[n]
995 << _T('\n');
996 }
997
998 wxLogMessage(detailed);
999
1000 // do it now if not done above
1001 #if wxUSE_GUI
1002 wxLogMessage(brief);
1003 #endif // wxUSE_GUI
1004 }
1005
1006 // ----------------------------------------------------------------------------
1007 // private functions
1008 // ----------------------------------------------------------------------------
1009
1010 static wxString GetTypeName(wxCmdLineParamType type)
1011 {
1012 wxString s;
1013 switch ( type )
1014 {
1015 default:
1016 wxFAIL_MSG( _T("unknown option type") );
1017 // still fall through
1018
1019 case wxCMD_LINE_VAL_STRING: s = _("str"); break;
1020 case wxCMD_LINE_VAL_NUMBER: s = _("num"); break;
1021 case wxCMD_LINE_VAL_DATE: s = _("date"); break;
1022 }
1023
1024 return s;
1025 }
1026
1027 /*
1028 Returns a string which is equal to the string pointed to by p, but up to the
1029 point where p contains an character that's not allowed.
1030 Allowable characters are letters and numbers, and characters pointed to by
1031 the parameter allowedChars.
1032
1033 For example, if p points to "abcde-@-_", and allowedChars is "-_",
1034 this function returns "abcde-".
1035 */
1036 static wxString GetOptionName(const wxChar *p,
1037 const wxChar *allowedChars)
1038 {
1039 wxString argName;
1040
1041 while ( *p && (wxIsalnum(*p) || wxStrchr(allowedChars, *p)) )
1042 {
1043 argName += *p++;
1044 }
1045
1046 return argName;
1047 }
1048
1049 // Besides alphanumeric characters, short and long options can
1050 // have other characters.
1051
1052 // A short option additionally can have these
1053 #define wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION wxT("_?")
1054
1055 // A long option can have the same characters as a short option and a '-'.
1056 #define wxCMD_LINE_CHARS_ALLOWED_BY_LONG_OPTION \
1057 wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION wxT("-")
1058
1059 static wxString GetShortOptionName(const wxChar *p)
1060 {
1061 return GetOptionName(p, wxCMD_LINE_CHARS_ALLOWED_BY_SHORT_OPTION);
1062 }
1063
1064 static wxString GetLongOptionName(const wxChar *p)
1065 {
1066 return GetOptionName(p, wxCMD_LINE_CHARS_ALLOWED_BY_LONG_OPTION);
1067 }
1068
1069 #endif // wxUSE_CMDLINE_PARSER
1070
1071 // ----------------------------------------------------------------------------
1072 // global functions
1073 // ----------------------------------------------------------------------------
1074
1075 /*
1076 This function is mainly used under Windows (as under Unix we always get the
1077 command line arguments as argc/argv anyhow) and so it tries to handle the
1078 Windows path names (separated by backslashes) correctly. For this it only
1079 considers that a backslash may be used to escape another backslash (but
1080 normally this is _not_ needed) or a quote but nothing else.
1081
1082 In particular, to pass a single argument containing a space to the program
1083 it should be quoted:
1084
1085 myprog.exe foo bar -> argc = 3, argv[1] = "foo", argv[2] = "bar"
1086 myprog.exe "foo bar" -> argc = 2, argv[1] = "foo bar"
1087
1088 To pass an argument containing spaces and quotes, the latter should be
1089 escaped with a backslash:
1090
1091 myprog.exe "foo \"bar\"" -> argc = 2, argv[1] = "foo "bar""
1092
1093 This hopefully matches the conventions used by Explorer/command line
1094 interpreter under Windows. If not, this function should be fixed.
1095 */
1096
1097 /* static */
1098 wxArrayString wxCmdLineParser::ConvertStringToArgs(const wxChar *p)
1099 {
1100 wxArrayString args;
1101
1102 wxString arg;
1103 arg.reserve(1024);
1104
1105 bool isInsideQuotes = FALSE;
1106 for ( ;; )
1107 {
1108 // skip white space
1109 while ( *p == _T(' ') || *p == _T('\t') )
1110 p++;
1111
1112 // anything left?
1113 if ( *p == _T('\0') )
1114 break;
1115
1116 // parse this parameter
1117 arg.clear();
1118 for ( ;; p++ )
1119 {
1120 // do we have a (lone) backslash?
1121 bool isQuotedByBS = FALSE;
1122 while ( *p == _T('\\') )
1123 {
1124 p++;
1125
1126 // if we have 2 backslashes in a row, output one
1127 // unless it looks like a UNC path \\machine\dir\file.ext
1128 if ( isQuotedByBS || arg.Len() == 0 )
1129 {
1130 arg += _T('\\');
1131 isQuotedByBS = FALSE;
1132 }
1133 else // the next char is quoted
1134 {
1135 isQuotedByBS = TRUE;
1136 }
1137 }
1138
1139 bool skipChar = FALSE,
1140 endParam = FALSE;
1141 switch ( *p )
1142 {
1143 case _T('"'):
1144 if ( !isQuotedByBS )
1145 {
1146 // don't put the quote itself in the arg
1147 skipChar = TRUE;
1148
1149 isInsideQuotes = !isInsideQuotes;
1150 }
1151 //else: insert a literal quote
1152
1153 break;
1154
1155 case _T(' '):
1156 case _T('\t'):
1157 // we intentionally don't check for preceding backslash
1158 // here as if we allowed it to be used to escape spaces the
1159 // cmd line of the form "foo.exe a:\ c:\bar" wouldn't be
1160 // parsed correctly
1161 if ( isInsideQuotes )
1162 {
1163 // preserve it, skip endParam below
1164 break;
1165 }
1166 //else: fall through
1167
1168 case _T('\0'):
1169 endParam = TRUE;
1170 break;
1171
1172 default:
1173 if ( isQuotedByBS )
1174 {
1175 // ignore backslash before an ordinary character - this
1176 // is needed to properly handle the file names under
1177 // Windows appearing in the command line
1178 arg += _T('\\');
1179 }
1180 }
1181
1182 // end of argument?
1183 if ( endParam )
1184 break;
1185
1186 // otherwise copy this char to arg
1187 if ( !skipChar )
1188 {
1189 arg += *p;
1190 }
1191 }
1192
1193 args.Add(arg);
1194 }
1195
1196 return args;
1197 }
1198