130fdc8d8SChris Lattner //===-- CommandObjectHelp.cpp -----------------------------------*- C++ -*-===//
230fdc8d8SChris Lattner //
330fdc8d8SChris Lattner //                     The LLVM Compiler Infrastructure
430fdc8d8SChris Lattner //
530fdc8d8SChris Lattner // This file is distributed under the University of Illinois Open Source
630fdc8d8SChris Lattner // License. See LICENSE.TXT for details.
730fdc8d8SChris Lattner //
830fdc8d8SChris Lattner //===----------------------------------------------------------------------===//
930fdc8d8SChris Lattner 
1030fdc8d8SChris Lattner // C Includes
1130fdc8d8SChris Lattner // C++ Includes
1230fdc8d8SChris Lattner // Other libraries and framework includes
1330fdc8d8SChris Lattner // Project includes
1426cac3afSEugene Zelenko #include "CommandObjectHelp.h"
1530fdc8d8SChris Lattner #include "lldb/Interpreter/CommandInterpreter.h"
16b9c1b51eSKate Stone #include "lldb/Interpreter/CommandObjectMultiword.h"
1730fdc8d8SChris Lattner #include "lldb/Interpreter/CommandReturnObject.h"
18b9c1b51eSKate Stone #include "lldb/Interpreter/Options.h"
1930fdc8d8SChris Lattner 
2030fdc8d8SChris Lattner using namespace lldb;
2130fdc8d8SChris Lattner using namespace lldb_private;
2230fdc8d8SChris Lattner 
2330fdc8d8SChris Lattner //-------------------------------------------------------------------------
2430fdc8d8SChris Lattner // CommandObjectHelp
2530fdc8d8SChris Lattner //-------------------------------------------------------------------------
2630fdc8d8SChris Lattner 
27b9c1b51eSKate Stone void CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
28*a49c2019SZachary Turner     Stream *s, llvm::StringRef command, llvm::StringRef prefix, llvm::StringRef subcommand,
29b9c1b51eSKate Stone     bool include_apropos, bool include_type_lookup) {
30*a49c2019SZachary Turner   if (!s || command.empty())
31*a49c2019SZachary Turner     return;
32*a49c2019SZachary Turner 
33*a49c2019SZachary Turner   std::string command_str = command.str();
34*a49c2019SZachary Turner   std::string prefix_str = prefix.str();
35*a49c2019SZachary Turner   std::string subcommand_str = subcommand.str();
36*a49c2019SZachary Turner   const std::string &lookup_str = !subcommand_str.empty() ? subcommand_str : command_str;
37*a49c2019SZachary Turner   s->Printf("'%s' is not a known command.\n", command_str.c_str());
38b9c1b51eSKate Stone   s->Printf("Try '%shelp' to see a current list of commands.\n",
39*a49c2019SZachary Turner             prefix.str().c_str());
40b9c1b51eSKate Stone   if (include_apropos) {
4125d6072aSKate Stone     s->Printf("Try '%sapropos %s' for a list of related commands.\n",
42*a49c2019SZachary Turner       prefix_str.c_str(), lookup_str.c_str());
4346d4aa21SEnrico Granata   }
44b9c1b51eSKate Stone   if (include_type_lookup) {
45b9c1b51eSKate Stone     s->Printf("Try '%stype lookup %s' for information on types, methods, "
46b9c1b51eSKate Stone               "functions, modules, etc.",
47*a49c2019SZachary Turner       prefix_str.c_str(), lookup_str.c_str());
4846d4aa21SEnrico Granata   }
4946d4aa21SEnrico Granata }
5046d4aa21SEnrico Granata 
517428a18cSKate Stone CommandObjectHelp::CommandObjectHelp(CommandInterpreter &interpreter)
52b9c1b51eSKate Stone     : CommandObjectParsed(interpreter, "help", "Show a list of all debugger "
53b9c1b51eSKate Stone                                                "commands, or give details "
54b9c1b51eSKate Stone                                                "about a specific command.",
557428a18cSKate Stone                           "help [<cmd-name>]"),
56b9c1b51eSKate Stone       m_options() {
57405fe67fSCaroline Tice   CommandArgumentEntry arg;
58405fe67fSCaroline Tice   CommandArgumentData command_arg;
59405fe67fSCaroline Tice 
60405fe67fSCaroline Tice   // Define the first (and only) variant of this arg.
61405fe67fSCaroline Tice   command_arg.arg_type = eArgTypeCommandName;
62405fe67fSCaroline Tice   command_arg.arg_repetition = eArgRepeatStar;
63405fe67fSCaroline Tice 
64b9c1b51eSKate Stone   // There is only one variant this argument could be; put it into the argument
65b9c1b51eSKate Stone   // entry.
66405fe67fSCaroline Tice   arg.push_back(command_arg);
67405fe67fSCaroline Tice 
68405fe67fSCaroline Tice   // Push the data for the first argument into the m_arguments vector.
69405fe67fSCaroline Tice   m_arguments.push_back(arg);
7030fdc8d8SChris Lattner }
7130fdc8d8SChris Lattner 
7226cac3afSEugene Zelenko CommandObjectHelp::~CommandObjectHelp() = default;
7330fdc8d8SChris Lattner 
741f0f5b5bSZachary Turner static OptionDefinition g_help_options[] = {
75ac9c3a62SKate Stone     // clang-format off
7626cac3afSEugene Zelenko   {LLDB_OPT_SET_ALL, false, "hide-aliases",         'a', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone, "Hide aliases in the command list."},
7726cac3afSEugene Zelenko   {LLDB_OPT_SET_ALL, false, "hide-user-commands",   'u', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone, "Hide user-defined commands from the list."},
7826cac3afSEugene Zelenko   {LLDB_OPT_SET_ALL, false, "show-hidden-commands", 'h', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone, "Include commands prefixed with an underscore."},
79ac9c3a62SKate Stone     // clang-format on
8008633eeaSEnrico Granata };
8108633eeaSEnrico Granata 
821f0f5b5bSZachary Turner llvm::ArrayRef<OptionDefinition>
831f0f5b5bSZachary Turner CommandObjectHelp::CommandOptions::GetDefinitions() {
8470602439SZachary Turner   return llvm::makeArrayRef(g_help_options);
851f0f5b5bSZachary Turner }
861f0f5b5bSZachary Turner 
87b9c1b51eSKate Stone bool CommandObjectHelp::DoExecute(Args &command, CommandReturnObject &result) {
8830fdc8d8SChris Lattner   CommandObject::CommandMap::iterator pos;
8930fdc8d8SChris Lattner   CommandObject *cmd_obj;
90c7bece56SGreg Clayton   const size_t argc = command.GetArgumentCount();
9130fdc8d8SChris Lattner 
92b9c1b51eSKate Stone   // 'help' doesn't take any arguments, other than command names.  If argc is 0,
93b9c1b51eSKate Stone   // we show the user
94b9c1b51eSKate Stone   // all commands (aliases and user commands if asked for).  Otherwise every
95b9c1b51eSKate Stone   // argument must be the name of a command or a sub-command.
96b9c1b51eSKate Stone   if (argc == 0) {
9708633eeaSEnrico Granata     uint32_t cmd_types = CommandInterpreter::eCommandTypesBuiltin;
9808633eeaSEnrico Granata     if (m_options.m_show_aliases)
9908633eeaSEnrico Granata       cmd_types |= CommandInterpreter::eCommandTypesAliases;
10008633eeaSEnrico Granata     if (m_options.m_show_user_defined)
10108633eeaSEnrico Granata       cmd_types |= CommandInterpreter::eCommandTypesUserDef;
102a487aa4cSKate Stone     if (m_options.m_show_hidden)
103a487aa4cSKate Stone       cmd_types |= CommandInterpreter::eCommandTypesHidden;
10408633eeaSEnrico Granata 
10530fdc8d8SChris Lattner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
10608633eeaSEnrico Granata     m_interpreter.GetHelp(result, cmd_types); // General help
107b9c1b51eSKate Stone   } else {
108b9c1b51eSKate Stone     // Get command object for the first command argument. Only search built-in
109b9c1b51eSKate Stone     // command dictionary.
110279a6c26SJim Ingham     StringList matches;
111b9c1b51eSKate Stone     cmd_obj =
112b9c1b51eSKate Stone         m_interpreter.GetCommandObject(command.GetArgumentAtIndex(0), &matches);
113b9c1b51eSKate Stone     bool is_alias_command =
114b9c1b51eSKate Stone         m_interpreter.AliasExists(command.GetArgumentAtIndex(0));
115e7941795SCaroline Tice     std::string alias_name = command.GetArgumentAtIndex(0);
11630fdc8d8SChris Lattner 
117b9c1b51eSKate Stone     if (cmd_obj != nullptr) {
118271ad29aSJim Ingham       StringList matches;
11930fdc8d8SChris Lattner       bool all_okay = true;
12030fdc8d8SChris Lattner       CommandObject *sub_cmd_obj = cmd_obj;
121b9c1b51eSKate Stone       // Loop down through sub_command dictionaries until we find the command
12297d2c401SZachary Turner       // object that corresponds to the help command entered.
12346d4aa21SEnrico Granata       std::string sub_command;
12497d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
12597d2c401SZachary Turner         sub_command = entry.ref;
126271ad29aSJim Ingham         matches.Clear();
127bef55ac8SEnrico Granata         if (sub_cmd_obj->IsAlias())
128b9c1b51eSKate Stone           sub_cmd_obj =
129b9c1b51eSKate Stone               ((CommandAlias *)sub_cmd_obj)->GetUnderlyingCommand().get();
130b9c1b51eSKate Stone         if (!sub_cmd_obj->IsMultiwordObject()) {
13130fdc8d8SChris Lattner           all_okay = false;
13297d2c401SZachary Turner           break;
133b9c1b51eSKate Stone         } else {
134271ad29aSJim Ingham           CommandObject *found_cmd;
135b9c1b51eSKate Stone           found_cmd =
136b9c1b51eSKate Stone               sub_cmd_obj->GetSubcommandObject(sub_command.c_str(), &matches);
13797d2c401SZachary Turner           if (found_cmd == nullptr || matches.GetSize() > 1) {
13830fdc8d8SChris Lattner             all_okay = false;
13997d2c401SZachary Turner             break;
14097d2c401SZachary Turner           } else
141271ad29aSJim Ingham             sub_cmd_obj = found_cmd;
14230fdc8d8SChris Lattner         }
14330fdc8d8SChris Lattner       }
14430fdc8d8SChris Lattner 
145b9c1b51eSKate Stone       if (!all_okay || (sub_cmd_obj == nullptr)) {
14630fdc8d8SChris Lattner         std::string cmd_string;
14730fdc8d8SChris Lattner         command.GetCommandString(cmd_string);
148b9c1b51eSKate Stone         if (matches.GetSize() >= 2) {
149271ad29aSJim Ingham           StreamString s;
150271ad29aSJim Ingham           s.Printf("ambiguous command %s", cmd_string.c_str());
151271ad29aSJim Ingham           size_t num_matches = matches.GetSize();
152b9c1b51eSKate Stone           for (size_t match_idx = 0; match_idx < num_matches; match_idx++) {
153271ad29aSJim Ingham             s.Printf("\n\t%s", matches.GetStringAtIndex(match_idx));
154271ad29aSJim Ingham           }
155271ad29aSJim Ingham           s.Printf("\n");
156c156427dSZachary Turner           result.AppendError(s.GetString());
15730fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
1589b62d1d5SEnrico Granata           return false;
159b9c1b51eSKate Stone         } else if (!sub_cmd_obj) {
16046d4aa21SEnrico Granata           StreamString error_msg_stream;
161b9c1b51eSKate Stone           GenerateAdditionalHelpAvenuesMessage(
162b9c1b51eSKate Stone               &error_msg_stream, cmd_string.c_str(),
163b9c1b51eSKate Stone               m_interpreter.GetCommandPrefix(), sub_command.c_str());
164c156427dSZachary Turner           result.AppendError(error_msg_stream.GetString());
1659b62d1d5SEnrico Granata           result.SetStatus(eReturnStatusFailed);
1669b62d1d5SEnrico Granata           return false;
167b9c1b51eSKate Stone         } else {
168b9c1b51eSKate Stone           GenerateAdditionalHelpAvenuesMessage(
169b9c1b51eSKate Stone               &result.GetOutputStream(), cmd_string.c_str(),
170b9c1b51eSKate Stone               m_interpreter.GetCommandPrefix(), sub_command.c_str());
171b9c1b51eSKate Stone           result.GetOutputStream().Printf(
172b9c1b51eSKate Stone               "\nThe closest match is '%s'. Help on it follows.\n\n",
173a449698cSZachary Turner               sub_cmd_obj->GetCommandName().str().c_str());
174e139cf23SCaroline Tice         }
175c007e846SJim Ingham       }
176c007e846SJim Ingham 
177998255bfSGreg Clayton       sub_cmd_obj->GenerateHelpText(result);
178e7941795SCaroline Tice 
179b9c1b51eSKate Stone       if (is_alias_command) {
180e7941795SCaroline Tice         StreamString sstr;
181a483f579SZachary Turner         m_interpreter.GetAlias(alias_name)->GetAliasExpansion(sstr);
182b9c1b51eSKate Stone         result.GetOutputStream().Printf("\n'%s' is an abbreviation for %s\n",
183b9c1b51eSKate Stone                                         alias_name.c_str(), sstr.GetData());
184e7941795SCaroline Tice       }
185b9c1b51eSKate Stone     } else if (matches.GetSize() > 0) {
186279a6c26SJim Ingham       Stream &output_strm = result.GetOutputStream();
187b9c1b51eSKate Stone       output_strm.Printf("Help requested with ambiguous command name, possible "
188b9c1b51eSKate Stone                          "completions:\n");
189c7bece56SGreg Clayton       const size_t match_count = matches.GetSize();
190b9c1b51eSKate Stone       for (size_t i = 0; i < match_count; i++) {
191279a6c26SJim Ingham         output_strm.Printf("\t%s\n", matches.GetStringAtIndex(i));
192279a6c26SJim Ingham       }
193b9c1b51eSKate Stone     } else {
194b9c1b51eSKate Stone       // Maybe the user is asking for help about a command argument rather than
195b9c1b51eSKate Stone       // a command.
196b9c1b51eSKate Stone       const CommandArgumentType arg_type =
197b9c1b51eSKate Stone           CommandObject::LookupArgumentName(command.GetArgumentAtIndex(0));
198b9c1b51eSKate Stone       if (arg_type != eArgTypeLastArg) {
199e139cf23SCaroline Tice         Stream &output_strm = result.GetOutputStream();
200e139cf23SCaroline Tice         CommandObject::GetArgumentHelp(output_strm, arg_type, m_interpreter);
201e139cf23SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
202b9c1b51eSKate Stone       } else {
20346d4aa21SEnrico Granata         StreamString error_msg_stream;
204b9c1b51eSKate Stone         GenerateAdditionalHelpAvenuesMessage(&error_msg_stream,
205b9c1b51eSKate Stone                                              command.GetArgumentAtIndex(0),
206*a49c2019SZachary Turner                                              m_interpreter.GetCommandPrefix(), "");
207c156427dSZachary Turner         result.AppendError(error_msg_stream.GetString());
20830fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
20930fdc8d8SChris Lattner       }
21030fdc8d8SChris Lattner     }
211e139cf23SCaroline Tice   }
21230fdc8d8SChris Lattner 
21330fdc8d8SChris Lattner   return result.Succeeded();
21430fdc8d8SChris Lattner }
21530fdc8d8SChris Lattner 
216b9c1b51eSKate Stone int CommandObjectHelp::HandleCompletion(Args &input, int &cursor_index,
21730fdc8d8SChris Lattner                                         int &cursor_char_position,
21830fdc8d8SChris Lattner                                         int match_start_point,
21930fdc8d8SChris Lattner                                         int max_return_elements,
220558ce124SJim Ingham                                         bool &word_complete,
221b9c1b51eSKate Stone                                         StringList &matches) {
22230fdc8d8SChris Lattner   // Return the completions of the commands in the help system:
223b9c1b51eSKate Stone   if (cursor_index == 0) {
224b9c1b51eSKate Stone     return m_interpreter.HandleCompletionMatches(
225b9c1b51eSKate Stone         input, cursor_index, cursor_char_position, match_start_point,
226b9c1b51eSKate Stone         max_return_elements, word_complete, matches);
227b9c1b51eSKate Stone   } else {
228b9c1b51eSKate Stone     CommandObject *cmd_obj =
229b9c1b51eSKate Stone         m_interpreter.GetCommandObject(input.GetArgumentAtIndex(0));
2301d18ebf6SJim Ingham 
231b9c1b51eSKate Stone     // The command that they are getting help on might be ambiguous, in which
232b9c1b51eSKate Stone     // case we should complete that,
2331d18ebf6SJim Ingham     // otherwise complete with the command the user is getting help on...
2341d18ebf6SJim Ingham 
235b9c1b51eSKate Stone     if (cmd_obj) {
23630fdc8d8SChris Lattner       input.Shift();
23730fdc8d8SChris Lattner       cursor_index--;
238b9c1b51eSKate Stone       return cmd_obj->HandleCompletion(
239b9c1b51eSKate Stone           input, cursor_index, cursor_char_position, match_start_point,
240b9c1b51eSKate Stone           max_return_elements, word_complete, matches);
241b9c1b51eSKate Stone     } else {
242b9c1b51eSKate Stone       return m_interpreter.HandleCompletionMatches(
243b9c1b51eSKate Stone           input, cursor_index, cursor_char_position, match_start_point,
244b9c1b51eSKate Stone           max_return_elements, word_complete, matches);
2451d18ebf6SJim Ingham     }
2461d18ebf6SJim Ingham   }
24730fdc8d8SChris Lattner }
248