1*b1cdbd2cSJim Jagielski /**************************************************************
2*b1cdbd2cSJim Jagielski  *
3*b1cdbd2cSJim Jagielski  * Licensed to the Apache Software Foundation (ASF) under one
4*b1cdbd2cSJim Jagielski  * or more contributor license agreements.  See the NOTICE file
5*b1cdbd2cSJim Jagielski  * distributed with this work for additional information
6*b1cdbd2cSJim Jagielski  * regarding copyright ownership.  The ASF licenses this file
7*b1cdbd2cSJim Jagielski  * to you under the Apache License, Version 2.0 (the
8*b1cdbd2cSJim Jagielski  * "License"); you may not use this file except in compliance
9*b1cdbd2cSJim Jagielski  * with the License.  You may obtain a copy of the License at
10*b1cdbd2cSJim Jagielski  *
11*b1cdbd2cSJim Jagielski  *   http://www.apache.org/licenses/LICENSE-2.0
12*b1cdbd2cSJim Jagielski  *
13*b1cdbd2cSJim Jagielski  * Unless required by applicable law or agreed to in writing,
14*b1cdbd2cSJim Jagielski  * software distributed under the License is distributed on an
15*b1cdbd2cSJim Jagielski  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16*b1cdbd2cSJim Jagielski  * KIND, either express or implied.  See the License for the
17*b1cdbd2cSJim Jagielski  * specific language governing permissions and limitations
18*b1cdbd2cSJim Jagielski  * under the License.
19*b1cdbd2cSJim Jagielski  *
20*b1cdbd2cSJim Jagielski  *************************************************************/
21*b1cdbd2cSJim Jagielski 
22*b1cdbd2cSJim Jagielski 
23*b1cdbd2cSJim Jagielski 
24*b1cdbd2cSJim Jagielski #include <precomp.h>
25*b1cdbd2cSJim Jagielski #include <cosv/commandline.hxx>
26*b1cdbd2cSJim Jagielski 
27*b1cdbd2cSJim Jagielski // NOT FULLY DECLARED SERVICES
28*b1cdbd2cSJim Jagielski #include <cosv/file.hxx>
29*b1cdbd2cSJim Jagielski 
30*b1cdbd2cSJim Jagielski 
31*b1cdbd2cSJim Jagielski namespace csv
32*b1cdbd2cSJim Jagielski {
33*b1cdbd2cSJim Jagielski 
34*b1cdbd2cSJim Jagielski namespace
35*b1cdbd2cSJim Jagielski {
36*b1cdbd2cSJim Jagielski 
37*b1cdbd2cSJim Jagielski const intt C_nNoOption = -1;
38*b1cdbd2cSJim Jagielski 
39*b1cdbd2cSJim Jagielski const char * sIncludeOptionShort = "-A:";
40*b1cdbd2cSJim Jagielski const char * sIncludeOptionLong  = "--Arguments:";
41*b1cdbd2cSJim Jagielski const uintt nIncludeOptionShort_Length = strlen(sIncludeOptionShort);
42*b1cdbd2cSJim Jagielski const uintt nIncludeOptionLong_Length = strlen(sIncludeOptionLong);
43*b1cdbd2cSJim Jagielski 
44*b1cdbd2cSJim Jagielski 
45*b1cdbd2cSJim Jagielski /** Analyses, if an option is the one to include a file with
46*b1cdbd2cSJim Jagielski     further command line arguments.
47*b1cdbd2cSJim Jagielski */
48*b1cdbd2cSJim Jagielski bool                IsIncludeOption(
49*b1cdbd2cSJim Jagielski                         const String &      i_option );
50*b1cdbd2cSJim Jagielski 
51*b1cdbd2cSJim Jagielski /** Gets the file name from an include-arguments-option.
52*b1cdbd2cSJim Jagielski */
53*b1cdbd2cSJim Jagielski String              IncludeFile_fromIncludeOption(
54*b1cdbd2cSJim Jagielski                         const String &      i_option );
55*b1cdbd2cSJim Jagielski 
56*b1cdbd2cSJim Jagielski 
57*b1cdbd2cSJim Jagielski bool
IsIncludeOption(const String & i_option)58*b1cdbd2cSJim Jagielski IsIncludeOption(const String & i_option)
59*b1cdbd2cSJim Jagielski {
60*b1cdbd2cSJim Jagielski     return strncmp(i_option, sIncludeOptionShort, nIncludeOptionShort_Length) == 0
61*b1cdbd2cSJim Jagielski            OR
62*b1cdbd2cSJim Jagielski            strncmp(i_option, sIncludeOptionLong, nIncludeOptionLong_Length) == 0;
63*b1cdbd2cSJim Jagielski }
64*b1cdbd2cSJim Jagielski 
65*b1cdbd2cSJim Jagielski String
IncludeFile_fromIncludeOption(const String & i_option)66*b1cdbd2cSJim Jagielski IncludeFile_fromIncludeOption(const String & i_option)
67*b1cdbd2cSJim Jagielski {
68*b1cdbd2cSJim Jagielski     if ( strncmp(i_option, sIncludeOptionShort, nIncludeOptionShort_Length)
69*b1cdbd2cSJim Jagielski          == 0 )
70*b1cdbd2cSJim Jagielski     {
71*b1cdbd2cSJim Jagielski         return String(i_option, nIncludeOptionShort_Length, str::maxsize);
72*b1cdbd2cSJim Jagielski     }
73*b1cdbd2cSJim Jagielski     else
74*b1cdbd2cSJim Jagielski     if ( strncmp(i_option, sIncludeOptionLong, nIncludeOptionLong_Length)
75*b1cdbd2cSJim Jagielski          == 0 )
76*b1cdbd2cSJim Jagielski     {
77*b1cdbd2cSJim Jagielski         return String(i_option, nIncludeOptionLong_Length, str::maxsize);
78*b1cdbd2cSJim Jagielski     }
79*b1cdbd2cSJim Jagielski     return String::Null_();
80*b1cdbd2cSJim Jagielski }
81*b1cdbd2cSJim Jagielski 
82*b1cdbd2cSJim Jagielski 
83*b1cdbd2cSJim Jagielski }   // end anonymous namespace
84*b1cdbd2cSJim Jagielski 
85*b1cdbd2cSJim Jagielski 
86*b1cdbd2cSJim Jagielski 
87*b1cdbd2cSJim Jagielski 
88*b1cdbd2cSJim Jagielski /** Local helper class for searching a possible option name in a vector of
89*b1cdbd2cSJim Jagielski     ->OptionDescription.
90*b1cdbd2cSJim Jagielski */
91*b1cdbd2cSJim Jagielski struct CommandLine::
92*b1cdbd2cSJim Jagielski FindOptionByText
93*b1cdbd2cSJim Jagielski {
operator ()csv::CommandLine::FindOptionByText94*b1cdbd2cSJim Jagielski     bool                operator()(
95*b1cdbd2cSJim Jagielski                             const CommandLine::OptionDescription &
96*b1cdbd2cSJim Jagielski                                                 i_option )
97*b1cdbd2cSJim Jagielski                         { return i_option.sText == sOption; }
98*b1cdbd2cSJim Jagielski 
99*b1cdbd2cSJim Jagielski     /// @param i_searchText [i_searchText != ""]
FindOptionByTextcsv::CommandLine::FindOptionByText100*b1cdbd2cSJim Jagielski                         FindOptionByText(
101*b1cdbd2cSJim Jagielski                             const String &      i_option )
102*b1cdbd2cSJim Jagielski                         :   sOption(i_option)   { }
103*b1cdbd2cSJim Jagielski   private:
104*b1cdbd2cSJim Jagielski     const String        sOption;
105*b1cdbd2cSJim Jagielski };
106*b1cdbd2cSJim Jagielski 
107*b1cdbd2cSJim Jagielski 
108*b1cdbd2cSJim Jagielski typedef std::vector<StringVector::const_iterator>   StringCIteratorList;
109*b1cdbd2cSJim Jagielski typedef std::vector<intt>                           OptionIdList;
110*b1cdbd2cSJim Jagielski 
111*b1cdbd2cSJim Jagielski bool
Interpret(int argc,char * argv[])112*b1cdbd2cSJim Jagielski CommandLine::Interpret( int    argc,
113*b1cdbd2cSJim Jagielski                         char * argv[] )
114*b1cdbd2cSJim Jagielski {
115*b1cdbd2cSJim Jagielski     Get_Arguments(argc,argv);
116*b1cdbd2cSJim Jagielski     csv_assert(aOptionPoints.size() == aOptionIds.size());
117*b1cdbd2cSJim Jagielski 
118*b1cdbd2cSJim Jagielski     StringVector::const_iterator
119*b1cdbd2cSJim Jagielski         itNext          = aCommandLine.begin();
120*b1cdbd2cSJim Jagielski         ++itNext;       // Move 1 forward from program name.
121*b1cdbd2cSJim Jagielski     StringVector::const_iterator
122*b1cdbd2cSJim Jagielski         itEnd           = aCommandLine.end();
123*b1cdbd2cSJim Jagielski     StringCIteratorList::const_iterator
124*b1cdbd2cSJim Jagielski         itOptPtsEnd     = aOptionPoints.end();
125*b1cdbd2cSJim Jagielski 
126*b1cdbd2cSJim Jagielski     OptionIdList::const_iterator
127*b1cdbd2cSJim Jagielski         itOptIds = aOptionIds.begin();
128*b1cdbd2cSJim Jagielski     for ( StringCIteratorList::const_iterator itOptPts = aOptionPoints.begin();
129*b1cdbd2cSJim Jagielski           itOptPts != itOptPtsEnd AND bIsOk;
130*b1cdbd2cSJim Jagielski           ++itOptPts, ++itOptIds )
131*b1cdbd2cSJim Jagielski     {
132*b1cdbd2cSJim Jagielski         // May be, there are arguments which do not belong to the last option:
133*b1cdbd2cSJim Jagielski         // itNext != *is
134*b1cdbd2cSJim Jagielski         Handle_FreeArguments(itNext, *itOptPts);
135*b1cdbd2cSJim Jagielski 
136*b1cdbd2cSJim Jagielski         itNext = do_HandleOption( *itOptIds,
137*b1cdbd2cSJim Jagielski                                   *itOptPts + 1,
138*b1cdbd2cSJim Jagielski                                   itOptPts+1 == itOptPtsEnd ? itEnd : *(itOptPts+1) );
139*b1cdbd2cSJim Jagielski         csv_assert(itNext <= itEnd);
140*b1cdbd2cSJim Jagielski     }   // end for (is)
141*b1cdbd2cSJim Jagielski     Handle_FreeArguments(itNext, itEnd);
142*b1cdbd2cSJim Jagielski 
143*b1cdbd2cSJim Jagielski     return bIsOk;
144*b1cdbd2cSJim Jagielski }
145*b1cdbd2cSJim Jagielski 
CommandLine()146*b1cdbd2cSJim Jagielski CommandLine::CommandLine()
147*b1cdbd2cSJim Jagielski     :   aOptions(),
148*b1cdbd2cSJim Jagielski         aCommandLine(),
149*b1cdbd2cSJim Jagielski         bIsOk(false)
150*b1cdbd2cSJim Jagielski {
151*b1cdbd2cSJim Jagielski }
152*b1cdbd2cSJim Jagielski 
153*b1cdbd2cSJim Jagielski void
Add_Option(intt i_id,String i_text)154*b1cdbd2cSJim Jagielski CommandLine::Add_Option( intt                i_id,
155*b1cdbd2cSJim Jagielski                          String              i_text )
156*b1cdbd2cSJim Jagielski {
157*b1cdbd2cSJim Jagielski     aOptions.push_back(OptionDescription( i_id,
158*b1cdbd2cSJim Jagielski                                           i_text ));
159*b1cdbd2cSJim Jagielski }
160*b1cdbd2cSJim Jagielski 
161*b1cdbd2cSJim Jagielski void
Get_Arguments(int argc,char * argv[])162*b1cdbd2cSJim Jagielski CommandLine::Get_Arguments( int    argc,
163*b1cdbd2cSJim Jagielski                             char * argv[] )
164*b1cdbd2cSJim Jagielski {
165*b1cdbd2cSJim Jagielski     aCommandLine.erase(aCommandLine.begin(),aCommandLine.end());
166*b1cdbd2cSJim Jagielski     aCommandLine.reserve(argc);
167*b1cdbd2cSJim Jagielski 
168*b1cdbd2cSJim Jagielski     char ** pArgEnd = argv + argc;
169*b1cdbd2cSJim Jagielski     for ( char ** pArg = &argv[0];
170*b1cdbd2cSJim Jagielski           pArg != pArgEnd;
171*b1cdbd2cSJim Jagielski           ++pArg )
172*b1cdbd2cSJim Jagielski     {
173*b1cdbd2cSJim Jagielski         Store_Argument(*pArg);
174*b1cdbd2cSJim Jagielski     }   // end for
175*b1cdbd2cSJim Jagielski     Find_OptionPoints();
176*b1cdbd2cSJim Jagielski     bIsOk = true;
177*b1cdbd2cSJim Jagielski }
178*b1cdbd2cSJim Jagielski 
179*b1cdbd2cSJim Jagielski intt
Find_Option(const String & i_text) const180*b1cdbd2cSJim Jagielski CommandLine::Find_Option( const String & i_text ) const
181*b1cdbd2cSJim Jagielski {
182*b1cdbd2cSJim Jagielski     if (i_text.empty())
183*b1cdbd2cSJim Jagielski         return C_nNoOption;
184*b1cdbd2cSJim Jagielski 
185*b1cdbd2cSJim Jagielski     FindOptionByText aSearch(i_text);
186*b1cdbd2cSJim Jagielski     OptionList::const_iterator
187*b1cdbd2cSJim Jagielski         itFound = std::find_if( aOptions.begin(),
188*b1cdbd2cSJim Jagielski                                 aOptions.end(),
189*b1cdbd2cSJim Jagielski                                 aSearch );
190*b1cdbd2cSJim Jagielski     if (itFound != aOptions.end())
191*b1cdbd2cSJim Jagielski     {
192*b1cdbd2cSJim Jagielski         return (*itFound).nId;
193*b1cdbd2cSJim Jagielski     }
194*b1cdbd2cSJim Jagielski     return C_nNoOption;
195*b1cdbd2cSJim Jagielski }
196*b1cdbd2cSJim Jagielski 
197*b1cdbd2cSJim Jagielski bool
Store_Argument(const String & i_arg)198*b1cdbd2cSJim Jagielski CommandLine::Store_Argument( const String & i_arg )
199*b1cdbd2cSJim Jagielski {
200*b1cdbd2cSJim Jagielski     if ( NOT IsIncludeOption(i_arg) )
201*b1cdbd2cSJim Jagielski     {
202*b1cdbd2cSJim Jagielski         aCommandLine.push_back(i_arg);
203*b1cdbd2cSJim Jagielski         return true;
204*b1cdbd2cSJim Jagielski     }
205*b1cdbd2cSJim Jagielski 
206*b1cdbd2cSJim Jagielski     return Try2Include_Options(i_arg);
207*b1cdbd2cSJim Jagielski }
208*b1cdbd2cSJim Jagielski 
209*b1cdbd2cSJim Jagielski void
Find_OptionPoints()210*b1cdbd2cSJim Jagielski CommandLine::Find_OptionPoints()
211*b1cdbd2cSJim Jagielski {
212*b1cdbd2cSJim Jagielski     StringVector::const_iterator    itEnd   = aCommandLine.end();
213*b1cdbd2cSJim Jagielski     for ( StringVector::const_iterator it = aCommandLine.begin() + 1;
214*b1cdbd2cSJim Jagielski           it != itEnd;
215*b1cdbd2cSJim Jagielski           ++it )
216*b1cdbd2cSJim Jagielski     {
217*b1cdbd2cSJim Jagielski         intt    nOption = Find_Option(*it);
218*b1cdbd2cSJim Jagielski         if (nOption != C_nNoOption)
219*b1cdbd2cSJim Jagielski         {
220*b1cdbd2cSJim Jagielski             aOptionPoints.push_back(it);
221*b1cdbd2cSJim Jagielski             aOptionIds.push_back(nOption);
222*b1cdbd2cSJim Jagielski         }
223*b1cdbd2cSJim Jagielski     }   // end for (i)
224*b1cdbd2cSJim Jagielski }
225*b1cdbd2cSJim Jagielski 
226*b1cdbd2cSJim Jagielski void
Handle_FreeArguments(StringVector::const_iterator i_begin,StringVector::const_iterator i_end)227*b1cdbd2cSJim Jagielski CommandLine::Handle_FreeArguments( StringVector::const_iterator i_begin,
228*b1cdbd2cSJim Jagielski                                    StringVector::const_iterator i_end )
229*b1cdbd2cSJim Jagielski {
230*b1cdbd2cSJim Jagielski     for ( StringVector::const_iterator it = i_begin;
231*b1cdbd2cSJim Jagielski           it != i_end AND bIsOk;
232*b1cdbd2cSJim Jagielski           ++it )
233*b1cdbd2cSJim Jagielski     {
234*b1cdbd2cSJim Jagielski         do_HandleFreeArgument(*it);
235*b1cdbd2cSJim Jagielski     }
236*b1cdbd2cSJim Jagielski }
237*b1cdbd2cSJim Jagielski 
238*b1cdbd2cSJim Jagielski bool
Try2Include_Options(const String & i_includeOption)239*b1cdbd2cSJim Jagielski CommandLine::Try2Include_Options(const String & i_includeOption)
240*b1cdbd2cSJim Jagielski {
241*b1cdbd2cSJim Jagielski     static StringVector
242*b1cdbd2cSJim Jagielski         aIncludedOptionFiles_;
243*b1cdbd2cSJim Jagielski 
244*b1cdbd2cSJim Jagielski     const String
245*b1cdbd2cSJim Jagielski         aOptionFile(IncludeFile_fromIncludeOption(i_includeOption));
246*b1cdbd2cSJim Jagielski 
247*b1cdbd2cSJim Jagielski     // Avoid recursion deadlock 1
248*b1cdbd2cSJim Jagielski     if ( std::find( aIncludedOptionFiles_.begin(),
249*b1cdbd2cSJim Jagielski                     aIncludedOptionFiles_.end(),
250*b1cdbd2cSJim Jagielski                     aOptionFile )
251*b1cdbd2cSJim Jagielski          != aIncludedOptionFiles_.end() )
252*b1cdbd2cSJim Jagielski     {
253*b1cdbd2cSJim Jagielski         Cerr() << "\nError: Self inclusion of option file "
254*b1cdbd2cSJim Jagielski                << aOptionFile
255*b1cdbd2cSJim Jagielski                << ".\n"
256*b1cdbd2cSJim Jagielski                << Endl();
257*b1cdbd2cSJim Jagielski         return false;
258*b1cdbd2cSJim Jagielski     }
259*b1cdbd2cSJim Jagielski 
260*b1cdbd2cSJim Jagielski     // Avoid recursion deadlock 2
261*b1cdbd2cSJim Jagielski     aIncludedOptionFiles_.push_back(aOptionFile);
262*b1cdbd2cSJim Jagielski 
263*b1cdbd2cSJim Jagielski     bool ok = Include_Options(aOptionFile);
264*b1cdbd2cSJim Jagielski 
265*b1cdbd2cSJim Jagielski     // Avoid recursion deadlock 3
266*b1cdbd2cSJim Jagielski     aIncludedOptionFiles_.pop_back();
267*b1cdbd2cSJim Jagielski 
268*b1cdbd2cSJim Jagielski     return ok;
269*b1cdbd2cSJim Jagielski }
270*b1cdbd2cSJim Jagielski 
271*b1cdbd2cSJim Jagielski bool
Include_Options(const String & i_optionsFile)272*b1cdbd2cSJim Jagielski CommandLine::Include_Options( const String & i_optionsFile )
273*b1cdbd2cSJim Jagielski {
274*b1cdbd2cSJim Jagielski     StreamStr
275*b1cdbd2cSJim Jagielski         aIncludedText(500);
276*b1cdbd2cSJim Jagielski     bool ok = Load_Options(aIncludedText, i_optionsFile);
277*b1cdbd2cSJim Jagielski     if (NOT ok)
278*b1cdbd2cSJim Jagielski         return false;
279*b1cdbd2cSJim Jagielski 
280*b1cdbd2cSJim Jagielski     StringVector
281*b1cdbd2cSJim Jagielski         aIncludedOptions;
282*b1cdbd2cSJim Jagielski     Split(aIncludedOptions, aIncludedText.c_str());
283*b1cdbd2cSJim Jagielski 
284*b1cdbd2cSJim Jagielski     StringVector::const_iterator itEnd = aIncludedOptions.end();
285*b1cdbd2cSJim Jagielski     for ( StringVector::const_iterator it = aIncludedOptions.begin();
286*b1cdbd2cSJim Jagielski           it != itEnd;
287*b1cdbd2cSJim Jagielski           ++it )
288*b1cdbd2cSJim Jagielski     {
289*b1cdbd2cSJim Jagielski         Store_Argument(*it);
290*b1cdbd2cSJim Jagielski     }   // end for
291*b1cdbd2cSJim Jagielski 
292*b1cdbd2cSJim Jagielski     return true;
293*b1cdbd2cSJim Jagielski }
294*b1cdbd2cSJim Jagielski 
295*b1cdbd2cSJim Jagielski bool
Load_Options(StreamStr & o_text,const String & i_optionsFile)296*b1cdbd2cSJim Jagielski CommandLine::Load_Options( StreamStr &      o_text,
297*b1cdbd2cSJim Jagielski                            const String &   i_optionsFile )
298*b1cdbd2cSJim Jagielski {
299*b1cdbd2cSJim Jagielski     if (i_optionsFile.empty())
300*b1cdbd2cSJim Jagielski         return false;
301*b1cdbd2cSJim Jagielski 
302*b1cdbd2cSJim Jagielski     File
303*b1cdbd2cSJim Jagielski         aOptionsFile(i_optionsFile, CFM_READ);
304*b1cdbd2cSJim Jagielski     OpenCloseGuard
305*b1cdbd2cSJim Jagielski         aOFGuard(aOptionsFile);
306*b1cdbd2cSJim Jagielski     if (NOT aOFGuard)
307*b1cdbd2cSJim Jagielski     {
308*b1cdbd2cSJim Jagielski         Cerr() << "\nError: Options file "
309*b1cdbd2cSJim Jagielski                << i_optionsFile
310*b1cdbd2cSJim Jagielski                << " not found.\n"
311*b1cdbd2cSJim Jagielski                << Endl();
312*b1cdbd2cSJim Jagielski         return false;
313*b1cdbd2cSJim Jagielski     }
314*b1cdbd2cSJim Jagielski 
315*b1cdbd2cSJim Jagielski     StreamStr
316*b1cdbd2cSJim Jagielski         aLoad(aOptionsFile);
317*b1cdbd2cSJim Jagielski     o_text.swap(aLoad);
318*b1cdbd2cSJim Jagielski     return true;
319*b1cdbd2cSJim Jagielski }
320*b1cdbd2cSJim Jagielski 
321*b1cdbd2cSJim Jagielski 
322*b1cdbd2cSJim Jagielski 
323*b1cdbd2cSJim Jagielski 
324*b1cdbd2cSJim Jagielski /******************         OptionDescription       ***********************/
325*b1cdbd2cSJim Jagielski 
326*b1cdbd2cSJim Jagielski 
327*b1cdbd2cSJim Jagielski CommandLine::
OptionDescription(intt i_id,String i_text)328*b1cdbd2cSJim Jagielski OptionDescription::OptionDescription( intt          i_id,
329*b1cdbd2cSJim Jagielski                                       String        i_text )
330*b1cdbd2cSJim Jagielski     :   nId(i_id),
331*b1cdbd2cSJim Jagielski         sText(i_text)
332*b1cdbd2cSJim Jagielski {
333*b1cdbd2cSJim Jagielski }
334*b1cdbd2cSJim Jagielski 
335*b1cdbd2cSJim Jagielski 
336*b1cdbd2cSJim Jagielski 
337*b1cdbd2cSJim Jagielski 
338*b1cdbd2cSJim Jagielski }   // namespace csv
339