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(
28a49c2019SZachary Turner     Stream *s, llvm::StringRef command, llvm::StringRef prefix, llvm::StringRef subcommand,
29b9c1b51eSKate Stone     bool include_apropos, bool include_type_lookup) {
30a49c2019SZachary Turner   if (!s || command.empty())
31a49c2019SZachary Turner     return;
32a49c2019SZachary Turner 
33a49c2019SZachary Turner   std::string command_str = command.str();
34a49c2019SZachary Turner   std::string prefix_str = prefix.str();
35a49c2019SZachary Turner   std::string subcommand_str = subcommand.str();
36a49c2019SZachary Turner   const std::string &lookup_str = !subcommand_str.empty() ? subcommand_str : command_str;
37a49c2019SZachary 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",
39a49c2019SZachary Turner             prefix.str().c_str());
40b9c1b51eSKate Stone   if (include_apropos) {
4125d6072aSKate Stone     s->Printf("Try '%sapropos %s' for a list of related commands.\n",
42a49c2019SZachary 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.",
47a49c2019SZachary 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 
9205097246SAdrian Prantl   // 'help' doesn't take any arguments, other than command names.  If argc is
9305097246SAdrian Prantl   // 0, we show the user all commands (aliases and user commands if asked for).
9405097246SAdrian Prantl   // Otherwise every argument must be the name of a command or a sub-command.
95b9c1b51eSKate Stone   if (argc == 0) {
9608633eeaSEnrico Granata     uint32_t cmd_types = CommandInterpreter::eCommandTypesBuiltin;
9708633eeaSEnrico Granata     if (m_options.m_show_aliases)
9808633eeaSEnrico Granata       cmd_types |= CommandInterpreter::eCommandTypesAliases;
9908633eeaSEnrico Granata     if (m_options.m_show_user_defined)
10008633eeaSEnrico Granata       cmd_types |= CommandInterpreter::eCommandTypesUserDef;
101a487aa4cSKate Stone     if (m_options.m_show_hidden)
102a487aa4cSKate Stone       cmd_types |= CommandInterpreter::eCommandTypesHidden;
10308633eeaSEnrico Granata 
10430fdc8d8SChris Lattner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
10508633eeaSEnrico Granata     m_interpreter.GetHelp(result, cmd_types); // General help
106b9c1b51eSKate Stone   } else {
107b9c1b51eSKate Stone     // Get command object for the first command argument. Only search built-in
108b9c1b51eSKate Stone     // command dictionary.
109279a6c26SJim Ingham     StringList matches;
11014f6b2c0SZachary Turner     auto command_name = command[0].ref;
11114f6b2c0SZachary Turner     cmd_obj = m_interpreter.GetCommandObject(command_name, &matches);
11230fdc8d8SChris Lattner 
113b9c1b51eSKate Stone     if (cmd_obj != nullptr) {
114271ad29aSJim Ingham       StringList matches;
11530fdc8d8SChris Lattner       bool all_okay = true;
11630fdc8d8SChris Lattner       CommandObject *sub_cmd_obj = cmd_obj;
117b9c1b51eSKate Stone       // Loop down through sub_command dictionaries until we find the command
11897d2c401SZachary Turner       // object that corresponds to the help command entered.
11946d4aa21SEnrico Granata       std::string sub_command;
12097d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
12197d2c401SZachary Turner         sub_command = entry.ref;
122271ad29aSJim Ingham         matches.Clear();
123bef55ac8SEnrico Granata         if (sub_cmd_obj->IsAlias())
124b9c1b51eSKate Stone           sub_cmd_obj =
125b9c1b51eSKate Stone               ((CommandAlias *)sub_cmd_obj)->GetUnderlyingCommand().get();
126b9c1b51eSKate Stone         if (!sub_cmd_obj->IsMultiwordObject()) {
12730fdc8d8SChris Lattner           all_okay = false;
12897d2c401SZachary Turner           break;
129b9c1b51eSKate Stone         } else {
130271ad29aSJim Ingham           CommandObject *found_cmd;
131b9c1b51eSKate Stone           found_cmd =
132b9c1b51eSKate Stone               sub_cmd_obj->GetSubcommandObject(sub_command.c_str(), &matches);
13397d2c401SZachary Turner           if (found_cmd == nullptr || matches.GetSize() > 1) {
13430fdc8d8SChris Lattner             all_okay = false;
13597d2c401SZachary Turner             break;
13697d2c401SZachary Turner           } else
137271ad29aSJim Ingham             sub_cmd_obj = found_cmd;
13830fdc8d8SChris Lattner         }
13930fdc8d8SChris Lattner       }
14030fdc8d8SChris Lattner 
141b9c1b51eSKate Stone       if (!all_okay || (sub_cmd_obj == nullptr)) {
14230fdc8d8SChris Lattner         std::string cmd_string;
14330fdc8d8SChris Lattner         command.GetCommandString(cmd_string);
144b9c1b51eSKate Stone         if (matches.GetSize() >= 2) {
145271ad29aSJim Ingham           StreamString s;
146271ad29aSJim Ingham           s.Printf("ambiguous command %s", cmd_string.c_str());
147271ad29aSJim Ingham           size_t num_matches = matches.GetSize();
148b9c1b51eSKate Stone           for (size_t match_idx = 0; match_idx < num_matches; match_idx++) {
149271ad29aSJim Ingham             s.Printf("\n\t%s", matches.GetStringAtIndex(match_idx));
150271ad29aSJim Ingham           }
151271ad29aSJim Ingham           s.Printf("\n");
152c156427dSZachary Turner           result.AppendError(s.GetString());
15330fdc8d8SChris Lattner           result.SetStatus(eReturnStatusFailed);
1549b62d1d5SEnrico Granata           return false;
155b9c1b51eSKate Stone         } else if (!sub_cmd_obj) {
15646d4aa21SEnrico Granata           StreamString error_msg_stream;
157b9c1b51eSKate Stone           GenerateAdditionalHelpAvenuesMessage(
158b9c1b51eSKate Stone               &error_msg_stream, cmd_string.c_str(),
159b9c1b51eSKate Stone               m_interpreter.GetCommandPrefix(), sub_command.c_str());
160c156427dSZachary Turner           result.AppendError(error_msg_stream.GetString());
1619b62d1d5SEnrico Granata           result.SetStatus(eReturnStatusFailed);
1629b62d1d5SEnrico Granata           return false;
163b9c1b51eSKate Stone         } else {
164b9c1b51eSKate Stone           GenerateAdditionalHelpAvenuesMessage(
165b9c1b51eSKate Stone               &result.GetOutputStream(), cmd_string.c_str(),
166b9c1b51eSKate Stone               m_interpreter.GetCommandPrefix(), sub_command.c_str());
167b9c1b51eSKate Stone           result.GetOutputStream().Printf(
168b9c1b51eSKate Stone               "\nThe closest match is '%s'. Help on it follows.\n\n",
169a449698cSZachary Turner               sub_cmd_obj->GetCommandName().str().c_str());
170e139cf23SCaroline Tice         }
171c007e846SJim Ingham       }
172c007e846SJim Ingham 
173998255bfSGreg Clayton       sub_cmd_obj->GenerateHelpText(result);
174e7941795SCaroline Tice 
17514f6b2c0SZachary Turner       if (m_interpreter.AliasExists(command_name)) {
176e7941795SCaroline Tice         StreamString sstr;
17714f6b2c0SZachary Turner         m_interpreter.GetAlias(command_name)->GetAliasExpansion(sstr);
178b9c1b51eSKate Stone         result.GetOutputStream().Printf("\n'%s' is an abbreviation for %s\n",
17914f6b2c0SZachary Turner                                         command[0].c_str(), sstr.GetData());
180e7941795SCaroline Tice       }
181b9c1b51eSKate Stone     } else if (matches.GetSize() > 0) {
182279a6c26SJim Ingham       Stream &output_strm = result.GetOutputStream();
183b9c1b51eSKate Stone       output_strm.Printf("Help requested with ambiguous command name, possible "
184b9c1b51eSKate Stone                          "completions:\n");
185c7bece56SGreg Clayton       const size_t match_count = matches.GetSize();
186b9c1b51eSKate Stone       for (size_t i = 0; i < match_count; i++) {
187279a6c26SJim Ingham         output_strm.Printf("\t%s\n", matches.GetStringAtIndex(i));
188279a6c26SJim Ingham       }
189b9c1b51eSKate Stone     } else {
190b9c1b51eSKate Stone       // Maybe the user is asking for help about a command argument rather than
191b9c1b51eSKate Stone       // a command.
192b9c1b51eSKate Stone       const CommandArgumentType arg_type =
19314f6b2c0SZachary Turner           CommandObject::LookupArgumentName(command_name);
194b9c1b51eSKate Stone       if (arg_type != eArgTypeLastArg) {
195e139cf23SCaroline Tice         Stream &output_strm = result.GetOutputStream();
196e139cf23SCaroline Tice         CommandObject::GetArgumentHelp(output_strm, arg_type, m_interpreter);
197e139cf23SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
198b9c1b51eSKate Stone       } else {
19946d4aa21SEnrico Granata         StreamString error_msg_stream;
20014f6b2c0SZachary Turner         GenerateAdditionalHelpAvenuesMessage(&error_msg_stream, command_name,
20114f6b2c0SZachary Turner                                              m_interpreter.GetCommandPrefix(),
20214f6b2c0SZachary Turner                                              "");
203c156427dSZachary Turner         result.AppendError(error_msg_stream.GetString());
20430fdc8d8SChris Lattner         result.SetStatus(eReturnStatusFailed);
20530fdc8d8SChris Lattner       }
20630fdc8d8SChris Lattner     }
207e139cf23SCaroline Tice   }
20830fdc8d8SChris Lattner 
20930fdc8d8SChris Lattner   return result.Succeeded();
21030fdc8d8SChris Lattner }
21130fdc8d8SChris Lattner 
212*2443bbd4SRaphael Isemann int CommandObjectHelp::HandleCompletion(CompletionRequest &request) {
21330fdc8d8SChris Lattner   // Return the completions of the commands in the help system:
214*2443bbd4SRaphael Isemann   if (request.GetCursorIndex() == 0) {
215*2443bbd4SRaphael Isemann     return m_interpreter.HandleCompletionMatches(request);
216b9c1b51eSKate Stone   } else {
217*2443bbd4SRaphael Isemann     CommandObject *cmd_obj =
218*2443bbd4SRaphael Isemann         m_interpreter.GetCommandObject(request.GetParsedLine()[0].ref);
2191d18ebf6SJim Ingham 
220b9c1b51eSKate Stone     // The command that they are getting help on might be ambiguous, in which
22105097246SAdrian Prantl     // case we should complete that, otherwise complete with the command the
22205097246SAdrian Prantl     // user is getting help on...
2231d18ebf6SJim Ingham 
224b9c1b51eSKate Stone     if (cmd_obj) {
225*2443bbd4SRaphael Isemann       request.GetParsedLine().Shift();
226*2443bbd4SRaphael Isemann       request.SetCursorIndex(request.GetCursorIndex() - 1);
227*2443bbd4SRaphael Isemann       return cmd_obj->HandleCompletion(request);
228b9c1b51eSKate Stone     } else {
229*2443bbd4SRaphael Isemann       return m_interpreter.HandleCompletionMatches(request);
2301d18ebf6SJim Ingham     }
2311d18ebf6SJim Ingham   }
23230fdc8d8SChris Lattner }
233