1 // Copyright (c) 2008, Google Inc.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 // * Redistributions of source code must retain the above copyright
9 // notice, this list of conditions and the following disclaimer.
10 // * Redistributions in binary form must reproduce the above
11 // copyright notice, this list of conditions and the following disclaimer
12 // in the documentation and/or other materials provided with the
13 // distribution.
14 // * Neither the name of Google Inc. nor the names of its
15 // contributors may be used to endorse or promote products derived from
16 // this software without specific prior written permission.
17 //
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 //
30 // ---
31
32 // Bash-style command line flag completion for C++ binaries
33 //
34 // This module implements bash-style completions. It achieves this
35 // goal in the following broad chunks:
36 //
37 // 1) Take a to-be-completed word, and examine it for search hints
38 // 2) Identify all potentially matching flags
39 // 2a) If there are no matching flags, do nothing.
40 // 2b) If all matching flags share a common prefix longer than the
41 // completion word, output just that matching prefix
42 // 3) Categorize those flags to produce a rough ordering of relevence.
43 // 4) Potentially trim the set of flags returned to a smaller number
44 // that bash is happier with
45 // 5) Output the matching flags in groups ordered by relevence.
46 // 5a) Force bash to place most-relevent groups at the top of the list
47 // 5b) Trim most flag's descriptions to fit on a single terminal line
48
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h> // for strlen
52
53 #include <set>
54 #include <string>
55 #include <utility>
56 #include <vector>
57
58 #include "config.h"
59 #include "gflags/gflags.h"
60 #include "gflags/gflags_completions.h"
61 #include "util.h"
62
63 using std::set;
64 using std::string;
65 using std::vector;
66
67
68 DEFINE_string(tab_completion_word, "",
69 "If non-empty, HandleCommandLineCompletions() will hijack the "
70 "process and attempt to do bash-style command line flag "
71 "completion on this value.");
72 DEFINE_int32(tab_completion_columns, 80,
73 "Number of columns to use in output for tab completion");
74
75
76 namespace GFLAGS_NAMESPACE {
77
78
79 namespace {
80 // Function prototypes and Type forward declarations. Code may be
81 // more easily understood if it is roughly ordered according to
82 // control flow, rather than by C's "declare before use" ordering
83 struct CompletionOptions;
84 struct NotableFlags;
85
86 // The entry point if flag completion is to be used.
87 static void PrintFlagCompletionInfo(void);
88
89
90 // 1) Examine search word
91 static void CanonicalizeCursorWordAndSearchOptions(
92 const string &cursor_word,
93 string *canonical_search_token,
94 CompletionOptions *options);
95
96 static bool RemoveTrailingChar(string *str, char c);
97
98
99 // 2) Find all matches
100 static void FindMatchingFlags(
101 const vector<CommandLineFlagInfo> &all_flags,
102 const CompletionOptions &options,
103 const string &match_token,
104 set<const CommandLineFlagInfo *> *all_matches,
105 string *longest_common_prefix);
106
107 static bool DoesSingleFlagMatch(
108 const CommandLineFlagInfo &flag,
109 const CompletionOptions &options,
110 const string &match_token);
111
112
113 // 3) Categorize matches
114 static void CategorizeAllMatchingFlags(
115 const set<const CommandLineFlagInfo *> &all_matches,
116 const string &search_token,
117 const string &module,
118 const string &package_dir,
119 NotableFlags *notable_flags);
120
121 static void TryFindModuleAndPackageDir(
122 const vector<CommandLineFlagInfo> &all_flags,
123 string *module,
124 string *package_dir);
125
126
127 // 4) Decide which flags to use
128 static void FinalizeCompletionOutput(
129 const set<const CommandLineFlagInfo *> &matching_flags,
130 CompletionOptions *options,
131 NotableFlags *notable_flags,
132 vector<string> *completions);
133
134 static void RetrieveUnusedFlags(
135 const set<const CommandLineFlagInfo *> &matching_flags,
136 const NotableFlags ¬able_flags,
137 set<const CommandLineFlagInfo *> *unused_flags);
138
139
140 // 5) Output matches
141 static void OutputSingleGroupWithLimit(
142 const set<const CommandLineFlagInfo *> &group,
143 const string &line_indentation,
144 const string &header,
145 const string &footer,
146 bool long_output_format,
147 int *remaining_line_limit,
148 size_t *completion_elements_added,
149 vector<string> *completions);
150
151 // (helpers for #5)
152 static string GetShortFlagLine(
153 const string &line_indentation,
154 const CommandLineFlagInfo &info);
155
156 static string GetLongFlagLine(
157 const string &line_indentation,
158 const CommandLineFlagInfo &info);
159
160
161 //
162 // Useful types
163
164 // Try to deduce the intentions behind this completion attempt. Return the
165 // canonical search term in 'canonical_search_token'. Binary search options
166 // are returned in the various booleans, which should all have intuitive
167 // semantics, possibly except:
168 // - return_all_matching_flags: Generally, we'll trim the number of
169 // returned candidates to some small number, showing those that are
170 // most likely to be useful first. If this is set, however, the user
171 // really does want us to return every single flag as an option.
172 // - force_no_update: Any time we output lines, all of which share a
173 // common prefix, bash will 'helpfully' not even bother to show the
174 // output, instead changing the current word to be that common prefix.
175 // If it's clear this shouldn't happen, we'll set this boolean
176 struct CompletionOptions {
177 bool flag_name_substring_search;
178 bool flag_location_substring_search;
179 bool flag_description_substring_search;
180 bool return_all_matching_flags;
181 bool force_no_update;
CompletionOptionsGFLAGS_NAMESPACE::__anonc80db8280111::CompletionOptions182 CompletionOptions(): flag_name_substring_search(false),
183 flag_location_substring_search(false),
184 flag_description_substring_search(false),
185 return_all_matching_flags(false),
186 force_no_update(false) { }
187 };
188
189 // Notable flags are flags that are special or preferred for some
190 // reason. For example, flags that are defined in the binary's module
191 // are expected to be much more relevent than flags defined in some
192 // other random location. These sets are specified roughly in precedence
193 // order. Once a flag is placed in one of these 'higher' sets, it won't
194 // be placed in any of the 'lower' sets.
195 struct NotableFlags {
196 typedef set<const CommandLineFlagInfo *> FlagSet;
197 FlagSet perfect_match_flag;
198 FlagSet module_flags; // Found in module file
199 FlagSet package_flags; // Found in same directory as module file
200 FlagSet most_common_flags; // One of the XXX most commonly supplied flags
201 FlagSet subpackage_flags; // Found in subdirectories of package
202 };
203
204
205 //
206 // Tab completion implementation - entry point
PrintFlagCompletionInfo(void)207 static void PrintFlagCompletionInfo(void) {
208 string cursor_word = FLAGS_tab_completion_word;
209 string canonical_token;
210 CompletionOptions options = CompletionOptions();
211 CanonicalizeCursorWordAndSearchOptions(
212 cursor_word,
213 &canonical_token,
214 &options);
215
216 DVLOG(1) << "Identified canonical_token: '" << canonical_token << "'";
217
218 vector<CommandLineFlagInfo> all_flags;
219 set<const CommandLineFlagInfo *> matching_flags;
220 GetAllFlags(&all_flags);
221 DVLOG(2) << "Found " << all_flags.size() << " flags overall";
222
223 string longest_common_prefix;
224 FindMatchingFlags(
225 all_flags,
226 options,
227 canonical_token,
228 &matching_flags,
229 &longest_common_prefix);
230 DVLOG(1) << "Identified " << matching_flags.size() << " matching flags";
231 DVLOG(1) << "Identified " << longest_common_prefix
232 << " as longest common prefix.";
233 if (longest_common_prefix.size() > canonical_token.size()) {
234 // There's actually a shared common prefix to all matching flags,
235 // so may as well output that and quit quickly.
236 DVLOG(1) << "The common prefix '" << longest_common_prefix
237 << "' was longer than the token '" << canonical_token
238 << "'. Returning just this prefix for completion.";
239 fprintf(stdout, "--%s", longest_common_prefix.c_str());
240 return;
241 }
242 if (matching_flags.empty()) {
243 VLOG(1) << "There were no matching flags, returning nothing.";
244 return;
245 }
246
247 string module;
248 string package_dir;
249 TryFindModuleAndPackageDir(all_flags, &module, &package_dir);
250 DVLOG(1) << "Identified module: '" << module << "'";
251 DVLOG(1) << "Identified package_dir: '" << package_dir << "'";
252
253 NotableFlags notable_flags;
254 CategorizeAllMatchingFlags(
255 matching_flags,
256 canonical_token,
257 module,
258 package_dir,
259 ¬able_flags);
260 DVLOG(2) << "Categorized matching flags:";
261 DVLOG(2) << " perfect_match: " << notable_flags.perfect_match_flag.size();
262 DVLOG(2) << " module: " << notable_flags.module_flags.size();
263 DVLOG(2) << " package: " << notable_flags.package_flags.size();
264 DVLOG(2) << " most common: " << notable_flags.most_common_flags.size();
265 DVLOG(2) << " subpackage: " << notable_flags.subpackage_flags.size();
266
267 vector<string> completions;
268 FinalizeCompletionOutput(
269 matching_flags,
270 &options,
271 ¬able_flags,
272 &completions);
273
274 if (options.force_no_update)
275 completions.push_back("~");
276
277 DVLOG(1) << "Finalized with " << completions.size()
278 << " chosen completions";
279
280 for (vector<string>::const_iterator it = completions.begin();
281 it != completions.end();
282 ++it) {
283 DVLOG(9) << " Completion entry: '" << *it << "'";
284 fprintf(stdout, "%s\n", it->c_str());
285 }
286 }
287
288
289 // 1) Examine search word (and helper method)
CanonicalizeCursorWordAndSearchOptions(const string & cursor_word,string * canonical_search_token,CompletionOptions * options)290 static void CanonicalizeCursorWordAndSearchOptions(
291 const string &cursor_word,
292 string *canonical_search_token,
293 CompletionOptions *options) {
294 *canonical_search_token = cursor_word;
295 if (canonical_search_token->empty()) return;
296
297 // Get rid of leading quotes and dashes in the search term
298 if ((*canonical_search_token)[0] == '"')
299 *canonical_search_token = canonical_search_token->substr(1);
300 while ((*canonical_search_token)[0] == '-')
301 *canonical_search_token = canonical_search_token->substr(1);
302
303 options->flag_name_substring_search = false;
304 options->flag_location_substring_search = false;
305 options->flag_description_substring_search = false;
306 options->return_all_matching_flags = false;
307 options->force_no_update = false;
308
309 // Look for all search options we can deduce now. Do this by walking
310 // backwards through the term, looking for up to three '?' and up to
311 // one '+' as suffixed characters. Consume them if found, and remove
312 // them from the canonical search token.
313 int found_question_marks = 0;
314 int found_plusses = 0;
315 while (true) {
316 if (found_question_marks < 3 &&
317 RemoveTrailingChar(canonical_search_token, '?')) {
318 ++found_question_marks;
319 continue;
320 }
321 if (found_plusses < 1 &&
322 RemoveTrailingChar(canonical_search_token, '+')) {
323 ++found_plusses;
324 continue;
325 }
326 break;
327 }
328
329 switch (found_question_marks) { // all fallthroughs
330 case 3: options->flag_description_substring_search = true;
331 case 2: options->flag_location_substring_search = true;
332 case 1: options->flag_name_substring_search = true;
333 };
334
335 options->return_all_matching_flags = (found_plusses > 0);
336 }
337
338 // Returns true if a char was removed
RemoveTrailingChar(string * str,char c)339 static bool RemoveTrailingChar(string *str, char c) {
340 if (str->empty()) return false;
341 if ((*str)[str->size() - 1] == c) {
342 *str = str->substr(0, str->size() - 1);
343 return true;
344 }
345 return false;
346 }
347
348
349 // 2) Find all matches (and helper methods)
FindMatchingFlags(const vector<CommandLineFlagInfo> & all_flags,const CompletionOptions & options,const string & match_token,set<const CommandLineFlagInfo * > * all_matches,string * longest_common_prefix)350 static void FindMatchingFlags(
351 const vector<CommandLineFlagInfo> &all_flags,
352 const CompletionOptions &options,
353 const string &match_token,
354 set<const CommandLineFlagInfo *> *all_matches,
355 string *longest_common_prefix) {
356 all_matches->clear();
357 bool first_match = true;
358 for (vector<CommandLineFlagInfo>::const_iterator it = all_flags.begin();
359 it != all_flags.end();
360 ++it) {
361 if (DoesSingleFlagMatch(*it, options, match_token)) {
362 all_matches->insert(&*it);
363 if (first_match) {
364 first_match = false;
365 *longest_common_prefix = it->name;
366 } else {
367 if (longest_common_prefix->empty() || it->name.empty()) {
368 longest_common_prefix->clear();
369 continue;
370 }
371 string::size_type pos = 0;
372 while (pos < longest_common_prefix->size() &&
373 pos < it->name.size() &&
374 (*longest_common_prefix)[pos] == it->name[pos])
375 ++pos;
376 longest_common_prefix->erase(pos);
377 }
378 }
379 }
380 }
381
382 // Given the set of all flags, the parsed match options, and the
383 // canonical search token, produce the set of all candidate matching
384 // flags for subsequent analysis or filtering.
DoesSingleFlagMatch(const CommandLineFlagInfo & flag,const CompletionOptions & options,const string & match_token)385 static bool DoesSingleFlagMatch(
386 const CommandLineFlagInfo &flag,
387 const CompletionOptions &options,
388 const string &match_token) {
389 // Is there a prefix match?
390 string::size_type pos = flag.name.find(match_token);
391 if (pos == 0) return true;
392
393 // Is there a substring match if we want it?
394 if (options.flag_name_substring_search &&
395 pos != string::npos)
396 return true;
397
398 // Is there a location match if we want it?
399 if (options.flag_location_substring_search &&
400 flag.filename.find(match_token) != string::npos)
401 return true;
402
403 // TODO(user): All searches should probably be case-insensitive
404 // (especially this one...)
405 if (options.flag_description_substring_search &&
406 flag.description.find(match_token) != string::npos)
407 return true;
408
409 return false;
410 }
411
412 // 3) Categorize matches (and helper method)
413
414 // Given a set of matching flags, categorize them by
415 // likely relevence to this specific binary
CategorizeAllMatchingFlags(const set<const CommandLineFlagInfo * > & all_matches,const string & search_token,const string & module,const string & package_dir,NotableFlags * notable_flags)416 static void CategorizeAllMatchingFlags(
417 const set<const CommandLineFlagInfo *> &all_matches,
418 const string &search_token,
419 const string &module, // empty if we couldn't find any
420 const string &package_dir, // empty if we couldn't find any
421 NotableFlags *notable_flags) {
422 notable_flags->perfect_match_flag.clear();
423 notable_flags->module_flags.clear();
424 notable_flags->package_flags.clear();
425 notable_flags->most_common_flags.clear();
426 notable_flags->subpackage_flags.clear();
427
428 for (set<const CommandLineFlagInfo *>::const_iterator it =
429 all_matches.begin();
430 it != all_matches.end();
431 ++it) {
432 DVLOG(2) << "Examining match '" << (*it)->name << "'";
433 DVLOG(7) << " filename: '" << (*it)->filename << "'";
434 string::size_type pos = string::npos;
435 if (!package_dir.empty())
436 pos = (*it)->filename.find(package_dir);
437 string::size_type slash = string::npos;
438 if (pos != string::npos) // candidate for package or subpackage match
439 slash = (*it)->filename.find(
440 PATH_SEPARATOR,
441 pos + package_dir.size() + 1);
442
443 if ((*it)->name == search_token) {
444 // Exact match on some flag's name
445 notable_flags->perfect_match_flag.insert(*it);
446 DVLOG(3) << "Result: perfect match";
447 } else if (!module.empty() && (*it)->filename == module) {
448 // Exact match on module filename
449 notable_flags->module_flags.insert(*it);
450 DVLOG(3) << "Result: module match";
451 } else if (!package_dir.empty() &&
452 pos != string::npos && slash == string::npos) {
453 // In the package, since there was no slash after the package portion
454 notable_flags->package_flags.insert(*it);
455 DVLOG(3) << "Result: package match";
456 } else if (false) {
457 // In the list of the XXX most commonly supplied flags overall
458 // TODO(user): Compile this list.
459 DVLOG(3) << "Result: most-common match";
460 } else if (!package_dir.empty() &&
461 pos != string::npos && slash != string::npos) {
462 // In a subdirectory of the package
463 notable_flags->subpackage_flags.insert(*it);
464 DVLOG(3) << "Result: subpackage match";
465 }
466
467 DVLOG(3) << "Result: not special match";
468 }
469 }
470
PushNameWithSuffix(vector<string> * suffixes,const char * suffix)471 static void PushNameWithSuffix(vector<string>* suffixes, const char* suffix) {
472 suffixes->push_back(
473 StringPrintf("/%s%s", ProgramInvocationShortName(), suffix));
474 }
475
TryFindModuleAndPackageDir(const vector<CommandLineFlagInfo> & all_flags,string * module,string * package_dir)476 static void TryFindModuleAndPackageDir(
477 const vector<CommandLineFlagInfo> &all_flags,
478 string *module,
479 string *package_dir) {
480 module->clear();
481 package_dir->clear();
482
483 vector<string> suffixes;
484 // TODO(user): There's some inherant ambiguity here - multiple directories
485 // could share the same trailing folder and file structure (and even worse,
486 // same file names), causing us to be unsure as to which of the two is the
487 // actual package for this binary. In this case, we'll arbitrarily choose.
488 PushNameWithSuffix(&suffixes, ".");
489 PushNameWithSuffix(&suffixes, "-main.");
490 PushNameWithSuffix(&suffixes, "_main.");
491 // These four are new but probably merited?
492 PushNameWithSuffix(&suffixes, "-test.");
493 PushNameWithSuffix(&suffixes, "_test.");
494 PushNameWithSuffix(&suffixes, "-unittest.");
495 PushNameWithSuffix(&suffixes, "_unittest.");
496
497 for (vector<CommandLineFlagInfo>::const_iterator it = all_flags.begin();
498 it != all_flags.end();
499 ++it) {
500 for (vector<string>::const_iterator suffix = suffixes.begin();
501 suffix != suffixes.end();
502 ++suffix) {
503 // TODO(user): Make sure the match is near the end of the string
504 if (it->filename.find(*suffix) != string::npos) {
505 *module = it->filename;
506 string::size_type sep = it->filename.rfind(PATH_SEPARATOR);
507 *package_dir = it->filename.substr(0, (sep == string::npos) ? 0 : sep);
508 return;
509 }
510 }
511 }
512 }
513
514 // Can't specialize template type on a locally defined type. Silly C++...
515 struct DisplayInfoGroup {
516 const char* header;
517 const char* footer;
518 set<const CommandLineFlagInfo *> *group;
519
SizeInLinesGFLAGS_NAMESPACE::__anonc80db8280111::DisplayInfoGroup520 int SizeInLines() const {
521 int size_in_lines = static_cast<int>(group->size()) + 1;
522 if (strlen(header) > 0) {
523 size_in_lines++;
524 }
525 if (strlen(footer) > 0) {
526 size_in_lines++;
527 }
528 return size_in_lines;
529 }
530 };
531
532 // 4) Finalize and trim output flag set
FinalizeCompletionOutput(const set<const CommandLineFlagInfo * > & matching_flags,CompletionOptions * options,NotableFlags * notable_flags,vector<string> * completions)533 static void FinalizeCompletionOutput(
534 const set<const CommandLineFlagInfo *> &matching_flags,
535 CompletionOptions *options,
536 NotableFlags *notable_flags,
537 vector<string> *completions) {
538
539 // We want to output lines in groups. Each group needs to be indented
540 // the same to keep its lines together. Unless otherwise required,
541 // only 99 lines should be output to prevent bash from harassing the
542 // user.
543
544 // First, figure out which output groups we'll actually use. For each
545 // nonempty group, there will be ~3 lines of header & footer, plus all
546 // output lines themselves.
547 int max_desired_lines = // "999999 flags should be enough for anyone. -dave"
548 (options->return_all_matching_flags ? 999999 : 98);
549 int lines_so_far = 0;
550
551 vector<DisplayInfoGroup> output_groups;
552 bool perfect_match_found = false;
553 if (!notable_flags->perfect_match_flag.empty()) {
554 perfect_match_found = true;
555 DisplayInfoGroup group =
556 { "",
557 "==========",
558 ¬able_flags->perfect_match_flag };
559 lines_so_far += group.SizeInLines();
560 output_groups.push_back(group);
561 }
562 if (lines_so_far < max_desired_lines &&
563 !notable_flags->module_flags.empty()) {
564 DisplayInfoGroup group = {
565 "-* Matching module flags *-",
566 "===========================",
567 ¬able_flags->module_flags };
568 lines_so_far += group.SizeInLines();
569 output_groups.push_back(group);
570 }
571 if (lines_so_far < max_desired_lines &&
572 !notable_flags->package_flags.empty()) {
573 DisplayInfoGroup group = {
574 "-* Matching package flags *-",
575 "============================",
576 ¬able_flags->package_flags };
577 lines_so_far += group.SizeInLines();
578 output_groups.push_back(group);
579 }
580 if (lines_so_far < max_desired_lines &&
581 !notable_flags->most_common_flags.empty()) {
582 DisplayInfoGroup group = {
583 "-* Commonly used flags *-",
584 "=========================",
585 ¬able_flags->most_common_flags };
586 lines_so_far += group.SizeInLines();
587 output_groups.push_back(group);
588 }
589 if (lines_so_far < max_desired_lines &&
590 !notable_flags->subpackage_flags.empty()) {
591 DisplayInfoGroup group = {
592 "-* Matching sub-package flags *-",
593 "================================",
594 ¬able_flags->subpackage_flags };
595 lines_so_far += group.SizeInLines();
596 output_groups.push_back(group);
597 }
598
599 set<const CommandLineFlagInfo *> obscure_flags; // flags not notable
600 if (lines_so_far < max_desired_lines) {
601 RetrieveUnusedFlags(matching_flags, *notable_flags, &obscure_flags);
602 if (!obscure_flags.empty()) {
603 DisplayInfoGroup group = {
604 "-* Other flags *-",
605 "",
606 &obscure_flags };
607 lines_so_far += group.SizeInLines();
608 output_groups.push_back(group);
609 }
610 }
611
612 // Second, go through each of the chosen output groups and output
613 // as many of those flags as we can, while remaining below our limit
614 int remaining_lines = max_desired_lines;
615 size_t completions_output = 0;
616 int indent = static_cast<int>(output_groups.size()) - 1;
617 for (vector<DisplayInfoGroup>::const_iterator it =
618 output_groups.begin();
619 it != output_groups.end();
620 ++it, --indent) {
621 OutputSingleGroupWithLimit(
622 *it->group, // group
623 string(indent, ' '), // line indentation
624 string(it->header), // header
625 string(it->footer), // footer
626 perfect_match_found, // long format
627 &remaining_lines, // line limit - reduces this by number printed
628 &completions_output, // completions (not lines) added
629 completions); // produced completions
630 perfect_match_found = false;
631 }
632
633 if (completions_output != matching_flags.size()) {
634 options->force_no_update = false;
635 completions->push_back("~ (Remaining flags hidden) ~");
636 } else {
637 options->force_no_update = true;
638 }
639 }
640
RetrieveUnusedFlags(const set<const CommandLineFlagInfo * > & matching_flags,const NotableFlags & notable_flags,set<const CommandLineFlagInfo * > * unused_flags)641 static void RetrieveUnusedFlags(
642 const set<const CommandLineFlagInfo *> &matching_flags,
643 const NotableFlags ¬able_flags,
644 set<const CommandLineFlagInfo *> *unused_flags) {
645 // Remove from 'matching_flags' set all members of the sets of
646 // flags we've already printed (specifically, those in notable_flags)
647 for (set<const CommandLineFlagInfo *>::const_iterator it =
648 matching_flags.begin();
649 it != matching_flags.end();
650 ++it) {
651 if (notable_flags.perfect_match_flag.count(*it) ||
652 notable_flags.module_flags.count(*it) ||
653 notable_flags.package_flags.count(*it) ||
654 notable_flags.most_common_flags.count(*it) ||
655 notable_flags.subpackage_flags.count(*it))
656 continue;
657 unused_flags->insert(*it);
658 }
659 }
660
661 // 5) Output matches (and helper methods)
662
OutputSingleGroupWithLimit(const set<const CommandLineFlagInfo * > & group,const string & line_indentation,const string & header,const string & footer,bool long_output_format,int * remaining_line_limit,size_t * completion_elements_output,vector<string> * completions)663 static void OutputSingleGroupWithLimit(
664 const set<const CommandLineFlagInfo *> &group,
665 const string &line_indentation,
666 const string &header,
667 const string &footer,
668 bool long_output_format,
669 int *remaining_line_limit,
670 size_t *completion_elements_output,
671 vector<string> *completions) {
672 if (group.empty()) return;
673 if (!header.empty()) {
674 if (*remaining_line_limit < 2) return;
675 *remaining_line_limit -= 2;
676 completions->push_back(line_indentation + header);
677 completions->push_back(line_indentation + string(header.size(), '-'));
678 }
679 for (set<const CommandLineFlagInfo *>::const_iterator it = group.begin();
680 it != group.end() && *remaining_line_limit > 0;
681 ++it) {
682 --*remaining_line_limit;
683 ++*completion_elements_output;
684 completions->push_back(
685 (long_output_format
686 ? GetLongFlagLine(line_indentation, **it)
687 : GetShortFlagLine(line_indentation, **it)));
688 }
689 if (!footer.empty()) {
690 if (*remaining_line_limit < 1) return;
691 --*remaining_line_limit;
692 completions->push_back(line_indentation + footer);
693 }
694 }
695
GetShortFlagLine(const string & line_indentation,const CommandLineFlagInfo & info)696 static string GetShortFlagLine(
697 const string &line_indentation,
698 const CommandLineFlagInfo &info) {
699 string prefix;
700 bool is_string = (info.type == "string");
701 SStringPrintf(&prefix, "%s--%s [%s%s%s] ",
702 line_indentation.c_str(),
703 info.name.c_str(),
704 (is_string ? "'" : ""),
705 info.default_value.c_str(),
706 (is_string ? "'" : ""));
707 int remainder =
708 FLAGS_tab_completion_columns - static_cast<int>(prefix.size());
709 string suffix;
710 if (remainder > 0)
711 suffix =
712 (static_cast<int>(info.description.size()) > remainder ?
713 (info.description.substr(0, remainder - 3) + "...").c_str() :
714 info.description.c_str());
715 return prefix + suffix;
716 }
717
GetLongFlagLine(const string & line_indentation,const CommandLineFlagInfo & info)718 static string GetLongFlagLine(
719 const string &line_indentation,
720 const CommandLineFlagInfo &info) {
721
722 string output = DescribeOneFlag(info);
723
724 // Replace '-' with '--', and remove trailing newline before appending
725 // the module definition location.
726 string old_flagname = "-" + info.name;
727 output.replace(
728 output.find(old_flagname),
729 old_flagname.size(),
730 "-" + old_flagname);
731 // Stick a newline and indentation in front of the type and default
732 // portions of DescribeOneFlag()s description
733 static const char kNewlineWithIndent[] = "\n ";
734 output.replace(output.find(" type:"), 1, string(kNewlineWithIndent));
735 output.replace(output.find(" default:"), 1, string(kNewlineWithIndent));
736 output = StringPrintf("%s Details for '--%s':\n"
737 "%s defined: %s",
738 line_indentation.c_str(),
739 info.name.c_str(),
740 output.c_str(),
741 info.filename.c_str());
742
743 // Eliminate any doubled newlines that crept in. Specifically, if
744 // DescribeOneFlag() decided to break the line just before "type"
745 // or "default", we don't want to introduce an extra blank line
746 static const string line_of_spaces(FLAGS_tab_completion_columns, ' ');
747 static const char kDoubledNewlines[] = "\n \n";
748 for (string::size_type newlines = output.find(kDoubledNewlines);
749 newlines != string::npos;
750 newlines = output.find(kDoubledNewlines))
751 // Replace each 'doubled newline' with a single newline
752 output.replace(newlines, sizeof(kDoubledNewlines) - 1, string("\n"));
753
754 for (string::size_type newline = output.find('\n');
755 newline != string::npos;
756 newline = output.find('\n')) {
757 int newline_pos = static_cast<int>(newline) % FLAGS_tab_completion_columns;
758 int missing_spaces = FLAGS_tab_completion_columns - newline_pos;
759 output.replace(newline, 1, line_of_spaces, 1, missing_spaces);
760 }
761 return output;
762 }
763 } // anonymous
764
HandleCommandLineCompletions(void)765 void HandleCommandLineCompletions(void) {
766 if (FLAGS_tab_completion_word.empty()) return;
767 PrintFlagCompletionInfo();
768 gflags_exitfunc(0);
769 }
770
771
772 } // namespace GFLAGS_NAMESPACE
773