180814287SRaphael Isemann //===-- CommandObjectCommands.cpp -----------------------------------------===//
2ebc09c36SJim Ingham //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6ebc09c36SJim Ingham //
7ebc09c36SJim Ingham //===----------------------------------------------------------------------===//
8ebc09c36SJim Ingham 
90e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h"
100e5e5a79SGreg Clayton 
116e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h"
1246d4aa21SEnrico Granata #include "CommandObjectHelp.h"
13ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1444d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
157594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
16ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
17de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h"
18ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
1947cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
20012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
2145d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h"
227594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
23ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
2499f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
25145d95c9SPavel Labath #include "lldb/Utility/Args.h"
26573ab909SZachary Turner #include "lldb/Utility/StringList.h"
27ebc09c36SJim Ingham 
28ebc09c36SJim Ingham using namespace lldb;
29ebc09c36SJim Ingham using namespace lldb_private;
30ebc09c36SJim Ingham 
31ebc09c36SJim Ingham // CommandObjectCommandsSource
32ebc09c36SJim Ingham 
3364becc11SRaphael Isemann #define LLDB_OPTIONS_source
3464becc11SRaphael Isemann #include "CommandOptions.inc"
351f0f5b5bSZachary Turner 
36b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed {
375a988416SJim Ingham public:
387428a18cSKate Stone   CommandObjectCommandsSource(CommandInterpreter &interpreter)
39b9c1b51eSKate Stone       : CommandObjectParsed(
40b9c1b51eSKate Stone             interpreter, "command source",
41b9c1b51eSKate Stone             "Read and execute LLDB commands from the file <filename>.",
426e3d8e7fSEugene Zelenko             nullptr),
43b9c1b51eSKate Stone         m_options() {
445a988416SJim Ingham     CommandArgumentEntry arg;
455a988416SJim Ingham     CommandArgumentData file_arg;
465a988416SJim Ingham 
475a988416SJim Ingham     // Define the first (and only) variant of this arg.
485a988416SJim Ingham     file_arg.arg_type = eArgTypeFilename;
495a988416SJim Ingham     file_arg.arg_repetition = eArgRepeatPlain;
505a988416SJim Ingham 
51b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
52b9c1b51eSKate Stone     // argument entry.
535a988416SJim Ingham     arg.push_back(file_arg);
545a988416SJim Ingham 
555a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
565a988416SJim Ingham     m_arguments.push_back(arg);
575a988416SJim Ingham   }
585a988416SJim Ingham 
596e3d8e7fSEugene Zelenko   ~CommandObjectCommandsSource() override = default;
605a988416SJim Ingham 
61b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
62b9c1b51eSKate Stone                                uint32_t index) override {
635a988416SJim Ingham     return "";
645a988416SJim Ingham   }
655a988416SJim Ingham 
66ae34ed2cSRaphael Isemann   void
67ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
682443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
69b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
70b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
71a2e76c0bSRaphael Isemann         request, nullptr);
725a988416SJim Ingham   }
735a988416SJim Ingham 
74b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
755a988416SJim Ingham 
765a988416SJim Ingham protected:
77b9c1b51eSKate Stone   class CommandOptions : public Options {
78e16c50a1SJim Ingham   public:
79b9c1b51eSKate Stone     CommandOptions()
80b9c1b51eSKate Stone         : Options(), m_stop_on_error(true), m_silent_run(false),
81b9c1b51eSKate Stone           m_stop_on_continue(true) {}
82e16c50a1SJim Ingham 
836e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
84e16c50a1SJim Ingham 
8597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
86b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
8797206d57SZachary Turner       Status error;
883bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
89e16c50a1SJim Ingham 
90b9c1b51eSKate Stone       switch (short_option) {
91e16c50a1SJim Ingham       case 'e':
92fe11483bSZachary Turner         error = m_stop_on_error.SetValueFromString(option_arg);
93e16c50a1SJim Ingham         break;
94340b0309SGreg Clayton 
95e16c50a1SJim Ingham       case 'c':
96fe11483bSZachary Turner         error = m_stop_on_continue.SetValueFromString(option_arg);
97e16c50a1SJim Ingham         break;
98340b0309SGreg Clayton 
9960986174SMichael Sartain       case 's':
100fe11483bSZachary Turner         error = m_silent_run.SetValueFromString(option_arg);
10160986174SMichael Sartain         break;
102340b0309SGreg Clayton 
103e16c50a1SJim Ingham       default:
10436162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
105e16c50a1SJim Ingham       }
106e16c50a1SJim Ingham 
107e16c50a1SJim Ingham       return error;
108e16c50a1SJim Ingham     }
109e16c50a1SJim Ingham 
110b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
111012d4fcaSEnrico Granata       m_stop_on_error.Clear();
112340b0309SGreg Clayton       m_silent_run.Clear();
113340b0309SGreg Clayton       m_stop_on_continue.Clear();
114e16c50a1SJim Ingham     }
115e16c50a1SJim Ingham 
1161f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
11770602439SZachary Turner       return llvm::makeArrayRef(g_source_options);
1181f0f5b5bSZachary Turner     }
119e16c50a1SJim Ingham 
120e16c50a1SJim Ingham     // Instance variables to hold the values for command options.
121e16c50a1SJim Ingham 
122012d4fcaSEnrico Granata     OptionValueBoolean m_stop_on_error;
123340b0309SGreg Clayton     OptionValueBoolean m_silent_run;
124340b0309SGreg Clayton     OptionValueBoolean m_stop_on_continue;
125e16c50a1SJim Ingham   };
126e16c50a1SJim Ingham 
127b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1284574a890SZachary Turner     if (command.GetArgumentCount() != 1) {
1294574a890SZachary Turner       result.AppendErrorWithFormat(
1304574a890SZachary Turner           "'%s' takes exactly one executable filename argument.\n",
1314574a890SZachary Turner           GetCommandName().str().c_str());
1324574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
1334574a890SZachary Turner       return false;
1344574a890SZachary Turner     }
135ebc09c36SJim Ingham 
1360d9a201eSRaphael Isemann     FileSpec cmd_file(command[0].ref());
1378f3be7a3SJonas Devlieghere     FileSystem::Instance().Resolve(cmd_file);
1386e3d8e7fSEugene Zelenko     ExecutionContext *exe_ctx = nullptr; // Just use the default context.
139ebc09c36SJim Ingham 
140340b0309SGreg Clayton     // If any options were set, then use them
141340b0309SGreg Clayton     if (m_options.m_stop_on_error.OptionWasSet() ||
142340b0309SGreg Clayton         m_options.m_silent_run.OptionWasSet() ||
143b9c1b51eSKate Stone         m_options.m_stop_on_continue.OptionWasSet()) {
144340b0309SGreg Clayton       // Use user set settings
14526c7bf93SJim Ingham       CommandInterpreterRunOptions options;
1461c19b74cSJonas Devlieghere 
1471c19b74cSJonas Devlieghere       if (m_options.m_stop_on_continue.OptionWasSet())
1481c19b74cSJonas Devlieghere         options.SetStopOnContinue(
1491c19b74cSJonas Devlieghere             m_options.m_stop_on_continue.GetCurrentValue());
1501c19b74cSJonas Devlieghere 
1511c19b74cSJonas Devlieghere       if (m_options.m_stop_on_error.OptionWasSet())
15226c7bf93SJim Ingham         options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue());
153c678ed77SStefan Granitz 
154c678ed77SStefan Granitz       // Individual silent setting is override for global command echo settings.
155c678ed77SStefan Granitz       if (m_options.m_silent_run.GetCurrentValue()) {
156c678ed77SStefan Granitz         options.SetSilent(true);
157c678ed77SStefan Granitz       } else {
158c678ed77SStefan Granitz         options.SetPrintResults(true);
159c0b48ab6SJonas Devlieghere         options.SetPrintErrors(true);
160c678ed77SStefan Granitz         options.SetEchoCommands(m_interpreter.GetEchoCommands());
161c678ed77SStefan Granitz         options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands());
162c678ed77SStefan Granitz       }
16326c7bf93SJim Ingham 
1644574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
165b9c1b51eSKate Stone     } else {
16605097246SAdrian Prantl       // No options were set, inherit any settings from nested "command source"
16705097246SAdrian Prantl       // commands, or set to sane default settings...
16826c7bf93SJim Ingham       CommandInterpreterRunOptions options;
1694574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
170ebc09c36SJim Ingham     }
171ebc09c36SJim Ingham     return result.Succeeded();
172ebc09c36SJim Ingham   }
1736e3d8e7fSEugene Zelenko 
1745a988416SJim Ingham   CommandOptions m_options;
175ebc09c36SJim Ingham };
176ebc09c36SJim Ingham 
177ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
178ebc09c36SJim Ingham // CommandObjectCommandsAlias
179ebc09c36SJim Ingham 
18064becc11SRaphael Isemann #define LLDB_OPTIONS_alias
18164becc11SRaphael Isemann #include "CommandOptions.inc"
1821f0f5b5bSZachary Turner 
183b9c1b51eSKate Stone static const char *g_python_command_instructions =
184b9c1b51eSKate Stone     "Enter your Python command(s). Type 'DONE' to end.\n"
185be93a35aSEnrico Granata     "You must define a Python function with this signature:\n"
18644d93782SGreg Clayton     "def my_command_impl(debugger, args, result, internal_dict):\n";
187be93a35aSEnrico Granata 
188b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw {
18945d0e238SEnrico Granata protected:
190b9c1b51eSKate Stone   class CommandOptions : public OptionGroup {
191ebc09c36SJim Ingham   public:
192b9c1b51eSKate Stone     CommandOptions() : OptionGroup(), m_help(), m_long_help() {}
19345d0e238SEnrico Granata 
19445d0e238SEnrico Granata     ~CommandOptions() override = default;
19545d0e238SEnrico Granata 
1961f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
19770602439SZachary Turner       return llvm::makeArrayRef(g_alias_options);
1981f0f5b5bSZachary Turner     }
19945d0e238SEnrico Granata 
20097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value,
201b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
20297206d57SZachary Turner       Status error;
20345d0e238SEnrico Granata 
2041f0f5b5bSZachary Turner       const int short_option = GetDefinitions()[option_idx].short_option;
2058cef4b0bSZachary Turner       std::string option_str(option_value);
20645d0e238SEnrico Granata 
207b9c1b51eSKate Stone       switch (short_option) {
20845d0e238SEnrico Granata       case 'h':
2098cef4b0bSZachary Turner         m_help.SetCurrentValue(option_str);
21045d0e238SEnrico Granata         m_help.SetOptionWasSet();
21145d0e238SEnrico Granata         break;
21245d0e238SEnrico Granata 
21345d0e238SEnrico Granata       case 'H':
2148cef4b0bSZachary Turner         m_long_help.SetCurrentValue(option_str);
21545d0e238SEnrico Granata         m_long_help.SetOptionWasSet();
21645d0e238SEnrico Granata         break;
21745d0e238SEnrico Granata 
21845d0e238SEnrico Granata       default:
21936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
22045d0e238SEnrico Granata       }
22145d0e238SEnrico Granata 
22245d0e238SEnrico Granata       return error;
22345d0e238SEnrico Granata     }
22445d0e238SEnrico Granata 
225b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
22645d0e238SEnrico Granata       m_help.Clear();
22745d0e238SEnrico Granata       m_long_help.Clear();
22845d0e238SEnrico Granata     }
22945d0e238SEnrico Granata 
23045d0e238SEnrico Granata     OptionValueString m_help;
23145d0e238SEnrico Granata     OptionValueString m_long_help;
23245d0e238SEnrico Granata   };
23345d0e238SEnrico Granata 
23445d0e238SEnrico Granata   OptionGroupOptions m_option_group;
23545d0e238SEnrico Granata   CommandOptions m_command_options;
23645d0e238SEnrico Granata 
23745d0e238SEnrico Granata public:
238b9c1b51eSKate Stone   Options *GetOptions() override { return &m_option_group; }
23945d0e238SEnrico Granata 
2407428a18cSKate Stone   CommandObjectCommandsAlias(CommandInterpreter &interpreter)
241b9c1b51eSKate Stone       : CommandObjectRaw(
242b9c1b51eSKate Stone             interpreter, "command alias",
243a449698cSZachary Turner             "Define a custom command in terms of an existing command."),
244b9c1b51eSKate Stone         m_option_group(), m_command_options() {
24545d0e238SEnrico Granata     m_option_group.Append(&m_command_options);
24645d0e238SEnrico Granata     m_option_group.Finalize();
24745d0e238SEnrico Granata 
248ebc09c36SJim Ingham     SetHelpLong(
249ea671fbdSKate Stone         "'alias' allows the user to create a short-cut or abbreviation for long \
250ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
251b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:"
252b9c1b51eSKate Stone         R"(
253ea671fbdSKate Stone 
254ea671fbdSKate Stone (lldb) command alias sc script
255ea671fbdSKate Stone 
256ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
257ea671fbdSKate Stone 
258ea671fbdSKate Stone (lldb) command alias bp breakpoint
259ea671fbdSKate Stone 
260b9c1b51eSKate Stone )"
261b9c1b51eSKate Stone         "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
262ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
263b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'."
264b9c1b51eSKate Stone         R"(
265ea671fbdSKate Stone 
266ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
267ea671fbdSKate Stone 
268ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
269ea671fbdSKate Stone 
270b9c1b51eSKate Stone )"
271b9c1b51eSKate Stone         "An alias can include some options for the command, with the values either \
272ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
273ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
274b9c1b51eSKate Stone shows how to create aliases with options:"
275b9c1b51eSKate Stone         R"(
276ea671fbdSKate Stone 
277ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
278ea671fbdSKate Stone 
279b9c1b51eSKate Stone )"
280b9c1b51eSKate Stone         "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
281ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
282ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
283ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
284ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
285ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
286ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
287ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
288ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
289ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
290b9c1b51eSKate Stone launch' example below)."
291b9c1b51eSKate Stone         R"(
292ea671fbdSKate Stone 
293b9c1b51eSKate Stone )"
294b9c1b51eSKate Stone         "Note: the positional arguments must substitute as whole words in the resultant \
295ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
296b9c1b51eSKate Stone \".cpp\":"
297b9c1b51eSKate Stone         R"(
298ea671fbdSKate Stone 
299ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
300ea671fbdSKate Stone 
301b9c1b51eSKate Stone )"
302b9c1b51eSKate Stone         "For more complex aliasing, use the \"command regex\" command instead.  In the \
303ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
304ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
305b9c1b51eSKate Stone argument.  The user would use this alias as follows:"
306b9c1b51eSKate Stone         R"(
307ea671fbdSKate Stone 
308ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
309ea671fbdSKate Stone (lldb) bfl my-file.c 137
310ea671fbdSKate Stone 
311ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
312ea671fbdSKate Stone 
313ea671fbdSKate Stone Another example:
314ea671fbdSKate Stone 
315ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
316ea671fbdSKate Stone (lldb) pltty /dev/tty0
317ea671fbdSKate Stone 
318ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
319ea671fbdSKate Stone 
320b9c1b51eSKate Stone )"
321b9c1b51eSKate Stone         "If the user always wanted to pass the same value to a particular option, the \
322ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
323b9c1b51eSKate Stone rather than using a positional placeholder:"
324b9c1b51eSKate Stone         R"(
325ea671fbdSKate Stone 
326ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
327ea671fbdSKate Stone 
328b9c1b51eSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)");
329ebc09c36SJim Ingham 
330405fe67fSCaroline Tice     CommandArgumentEntry arg1;
331405fe67fSCaroline Tice     CommandArgumentEntry arg2;
332405fe67fSCaroline Tice     CommandArgumentEntry arg3;
333405fe67fSCaroline Tice     CommandArgumentData alias_arg;
334405fe67fSCaroline Tice     CommandArgumentData cmd_arg;
335405fe67fSCaroline Tice     CommandArgumentData options_arg;
336405fe67fSCaroline Tice 
337405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
338405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
339405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
340405fe67fSCaroline Tice 
341b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
342b9c1b51eSKate Stone     // argument entry.
343405fe67fSCaroline Tice     arg1.push_back(alias_arg);
344405fe67fSCaroline Tice 
345405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
346405fe67fSCaroline Tice     cmd_arg.arg_type = eArgTypeCommandName;
347405fe67fSCaroline Tice     cmd_arg.arg_repetition = eArgRepeatPlain;
348405fe67fSCaroline Tice 
349b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
350b9c1b51eSKate Stone     // argument entry.
351405fe67fSCaroline Tice     arg2.push_back(cmd_arg);
352405fe67fSCaroline Tice 
353405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
354405fe67fSCaroline Tice     options_arg.arg_type = eArgTypeAliasOptions;
355405fe67fSCaroline Tice     options_arg.arg_repetition = eArgRepeatOptional;
356405fe67fSCaroline Tice 
357b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
358b9c1b51eSKate Stone     // argument entry.
359405fe67fSCaroline Tice     arg3.push_back(options_arg);
360405fe67fSCaroline Tice 
361405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
362405fe67fSCaroline Tice     m_arguments.push_back(arg1);
363405fe67fSCaroline Tice     m_arguments.push_back(arg2);
364405fe67fSCaroline Tice     m_arguments.push_back(arg3);
365ebc09c36SJim Ingham   }
366ebc09c36SJim Ingham 
3676e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAlias() override = default;
368ebc09c36SJim Ingham 
3695a988416SJim Ingham protected:
3704d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
371b9c1b51eSKate Stone                  CommandReturnObject &result) override {
3724d51a902SRaphael Isemann     if (raw_command_line.empty()) {
373d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
37445d0e238SEnrico Granata       return false;
37545d0e238SEnrico Granata     }
37645d0e238SEnrico Granata 
377e1cfbc79STodd Fiala     ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext();
378e1cfbc79STodd Fiala     m_option_group.NotifyOptionParsingStarting(&exe_ctx);
37945d0e238SEnrico Granata 
3803a0e1270SRaphael Isemann     OptionsWithRaw args_with_suffix(raw_command_line);
38145d0e238SEnrico Granata 
3823a0e1270SRaphael Isemann     if (args_with_suffix.HasArgs())
3833a0e1270SRaphael Isemann       if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result,
3843a0e1270SRaphael Isemann                                  m_option_group, exe_ctx))
38545d0e238SEnrico Granata         return false;
38645d0e238SEnrico Granata 
387daed98e5SShivam Mittal     llvm::StringRef raw_command_string = args_with_suffix.GetRawPart();
388a01bccdbSZachary Turner     Args args(raw_command_string);
389844d2303SCaroline Tice 
39011eb9c64SZachary Turner     if (args.GetArgumentCount() < 2) {
391d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
392844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
393844d2303SCaroline Tice       return false;
394844d2303SCaroline Tice     }
395844d2303SCaroline Tice 
396844d2303SCaroline Tice     // Get the alias command.
397844d2303SCaroline Tice 
3980d9a201eSRaphael Isemann     auto alias_command = args[0].ref();
3994574a890SZachary Turner     if (alias_command.startswith("-")) {
400d72e412fSEnrico Granata       result.AppendError("aliases starting with a dash are not supported");
401b9c1b51eSKate Stone       if (alias_command == "--help" || alias_command == "--long-help") {
402b9c1b51eSKate Stone         result.AppendWarning("if trying to pass options to 'command alias' add "
403b9c1b51eSKate Stone                              "a -- at the end of the options");
404d72e412fSEnrico Granata       }
405d72e412fSEnrico Granata       result.SetStatus(eReturnStatusFailed);
406d72e412fSEnrico Granata       return false;
407d72e412fSEnrico Granata     }
408844d2303SCaroline Tice 
409b9c1b51eSKate Stone     // Strip the new alias name off 'raw_command_string'  (leave it on args,
41005097246SAdrian Prantl     // which gets passed to 'Execute', which does the stripping itself.
411844d2303SCaroline Tice     size_t pos = raw_command_string.find(alias_command);
412b9c1b51eSKate Stone     if (pos == 0) {
413844d2303SCaroline Tice       raw_command_string = raw_command_string.substr(alias_command.size());
414844d2303SCaroline Tice       pos = raw_command_string.find_first_not_of(' ');
415844d2303SCaroline Tice       if ((pos != std::string::npos) && (pos > 0))
416844d2303SCaroline Tice         raw_command_string = raw_command_string.substr(pos);
417b9c1b51eSKate Stone     } else {
418844d2303SCaroline Tice       result.AppendError("Error parsing command string.  No alias created.");
419844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
420844d2303SCaroline Tice       return false;
421844d2303SCaroline Tice     }
422844d2303SCaroline Tice 
423844d2303SCaroline Tice     // Verify that the command is alias-able.
424771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
425b9c1b51eSKate Stone       result.AppendErrorWithFormat(
426b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
4274574a890SZachary Turner           args[0].c_str());
428844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
429844d2303SCaroline Tice       return false;
430844d2303SCaroline Tice     }
431844d2303SCaroline Tice 
432b9c1b51eSKate Stone     // Get CommandObject that is being aliased. The command name is read from
433a01bccdbSZachary Turner     // the front of raw_command_string. raw_command_string is returned with the
434a01bccdbSZachary Turner     // name of the command object stripped off the front.
435a01bccdbSZachary Turner     llvm::StringRef original_raw_command_string = raw_command_string;
436b9c1b51eSKate Stone     CommandObject *cmd_obj =
437b9c1b51eSKate Stone         m_interpreter.GetCommandObjectForCommand(raw_command_string);
438844d2303SCaroline Tice 
439b9c1b51eSKate Stone     if (!cmd_obj) {
440b9c1b51eSKate Stone       result.AppendErrorWithFormat("invalid command given to 'command alias'. "
441b9c1b51eSKate Stone                                    "'%s' does not begin with a valid command."
442b9c1b51eSKate Stone                                    "  No alias created.",
443a01bccdbSZachary Turner                                    original_raw_command_string.str().c_str());
444844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
445844d2303SCaroline Tice       return false;
446b9c1b51eSKate Stone     } else if (!cmd_obj->WantsRawCommandString()) {
447b9c1b51eSKate Stone       // Note that args was initialized with the original command, and has not
44805097246SAdrian Prantl       // been updated to this point. Therefore can we pass it to the version of
44905097246SAdrian Prantl       // Execute that does not need/expect raw input in the alias.
4505a988416SJim Ingham       return HandleAliasingNormalCommand(args, result);
451b9c1b51eSKate Stone     } else {
452b9c1b51eSKate Stone       return HandleAliasingRawCommand(alias_command, raw_command_string,
453b9c1b51eSKate Stone                                       *cmd_obj, result);
4545a988416SJim Ingham     }
4555a988416SJim Ingham     return result.Succeeded();
4565a988416SJim Ingham   }
4575a988416SJim Ingham 
458a01bccdbSZachary Turner   bool HandleAliasingRawCommand(llvm::StringRef alias_command,
459a01bccdbSZachary Turner                                 llvm::StringRef raw_command_string,
460b9c1b51eSKate Stone                                 CommandObject &cmd_obj,
461b9c1b51eSKate Stone                                 CommandReturnObject &result) {
462844d2303SCaroline Tice     // Verify & handle any options/arguments passed to the alias command
463844d2303SCaroline Tice 
464b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
465b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
466844d2303SCaroline Tice 
467b9c1b51eSKate Stone     if (CommandObjectSP cmd_obj_sp =
468b9c1b51eSKate Stone             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) {
469a01bccdbSZachary Turner       if (m_interpreter.AliasExists(alias_command) ||
470a01bccdbSZachary Turner           m_interpreter.UserCommandExists(alias_command)) {
471b9c1b51eSKate Stone         result.AppendWarningWithFormat(
472b9c1b51eSKate Stone             "Overwriting existing definition for '%s'.\n",
473a01bccdbSZachary Turner             alias_command.str().c_str());
474844d2303SCaroline Tice       }
475b9c1b51eSKate Stone       if (CommandAlias *alias = m_interpreter.AddAlias(
476a01bccdbSZachary Turner               alias_command, cmd_obj_sp, raw_command_string)) {
47745d0e238SEnrico Granata         if (m_command_options.m_help.OptionWasSet())
47845d0e238SEnrico Granata           alias->SetHelp(m_command_options.m_help.GetCurrentValue());
47945d0e238SEnrico Granata         if (m_command_options.m_long_help.OptionWasSet())
48045d0e238SEnrico Granata           alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
481844d2303SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
482b9c1b51eSKate Stone       } else {
483472362e6SCaroline Tice         result.AppendError("Unable to create requested alias.\n");
484472362e6SCaroline Tice         result.SetStatus(eReturnStatusFailed);
485472362e6SCaroline Tice       }
486212130acSEnrico Granata 
487b9c1b51eSKate Stone     } else {
488212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
489212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
490212130acSEnrico Granata     }
491212130acSEnrico Granata 
492844d2303SCaroline Tice     return result.Succeeded();
493844d2303SCaroline Tice   }
494ebc09c36SJim Ingham 
495b9c1b51eSKate Stone   bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) {
496867b185dSCaroline Tice     size_t argc = args.GetArgumentCount();
497ebc09c36SJim Ingham 
498b9c1b51eSKate Stone     if (argc < 2) {
499d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
500ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
501ebc09c36SJim Ingham       return false;
502ebc09c36SJim Ingham     }
503ebc09c36SJim Ingham 
5044574a890SZachary Turner     // Save these in std::strings since we're going to shift them off.
505adcd0268SBenjamin Kramer     const std::string alias_command(std::string(args[0].ref()));
506adcd0268SBenjamin Kramer     const std::string actual_command(std::string(args[1].ref()));
507ebc09c36SJim Ingham 
508ebc09c36SJim Ingham     args.Shift(); // Shift the alias command word off the argument vector.
509ebc09c36SJim Ingham     args.Shift(); // Shift the old command word off the argument vector.
510ebc09c36SJim Ingham 
511b9c1b51eSKate Stone     // Verify that the command is alias'able, and get the appropriate command
512b9c1b51eSKate Stone     // object.
513ebc09c36SJim Ingham 
514771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
515b9c1b51eSKate Stone       result.AppendErrorWithFormat(
516b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
517ebc09c36SJim Ingham           alias_command.c_str());
518ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
5194574a890SZachary Turner       return false;
5204574a890SZachary Turner     }
5214574a890SZachary Turner 
522b9c1b51eSKate Stone     CommandObjectSP command_obj_sp(
523a449698cSZachary Turner         m_interpreter.GetCommandSPExact(actual_command, true));
524ebc09c36SJim Ingham     CommandObjectSP subcommand_obj_sp;
525ebc09c36SJim Ingham     bool use_subcommand = false;
5264574a890SZachary Turner     if (!command_obj_sp) {
5274574a890SZachary Turner       result.AppendErrorWithFormat("'%s' is not an existing command.\n",
5284574a890SZachary Turner                                    actual_command.c_str());
5294574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
5304574a890SZachary Turner       return false;
5314574a890SZachary Turner     }
532ebc09c36SJim Ingham     CommandObject *cmd_obj = command_obj_sp.get();
5336e3d8e7fSEugene Zelenko     CommandObject *sub_cmd_obj = nullptr;
534b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
535b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
536ebc09c36SJim Ingham 
53711eb9c64SZachary Turner     while (cmd_obj->IsMultiwordObject() && !args.empty()) {
5380d9a201eSRaphael Isemann       auto sub_command = args[0].ref();
53911eb9c64SZachary Turner       assert(!sub_command.empty());
5404574a890SZachary Turner       subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command);
5414574a890SZachary Turner       if (!subcommand_obj_sp) {
542b9c1b51eSKate Stone         result.AppendErrorWithFormat(
543b9c1b51eSKate Stone             "'%s' is not a valid sub-command of '%s'.  "
544f415eeb4SCaroline Tice             "Unable to create alias.\n",
5454574a890SZachary Turner             args[0].c_str(), actual_command.c_str());
546ebc09c36SJim Ingham         result.SetStatus(eReturnStatusFailed);
547ebc09c36SJim Ingham         return false;
548ebc09c36SJim Ingham       }
5494574a890SZachary Turner 
5504574a890SZachary Turner       sub_cmd_obj = subcommand_obj_sp.get();
5514574a890SZachary Turner       use_subcommand = true;
5524574a890SZachary Turner       args.Shift(); // Shift the sub_command word off the argument vector.
5534574a890SZachary Turner       cmd_obj = sub_cmd_obj;
554ebc09c36SJim Ingham     }
555ebc09c36SJim Ingham 
556ebc09c36SJim Ingham     // Verify & handle any options/arguments passed to the alias command
557ebc09c36SJim Ingham 
558212130acSEnrico Granata     std::string args_string;
559212130acSEnrico Granata 
56011eb9c64SZachary Turner     if (!args.empty()) {
561b9c1b51eSKate Stone       CommandObjectSP tmp_sp =
562b9c1b51eSKate Stone           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false);
563ebc09c36SJim Ingham       if (use_subcommand)
5644574a890SZachary Turner         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(),
5654574a890SZachary Turner                                                  false);
566ca90c47eSCaroline Tice 
567ca90c47eSCaroline Tice       args.GetCommandString(args_string);
568867b185dSCaroline Tice     }
569ebc09c36SJim Ingham 
570771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
571771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
572b9c1b51eSKate Stone       result.AppendWarningWithFormat(
5734574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
574ebc09c36SJim Ingham     }
575ebc09c36SJim Ingham 
576b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
5774574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
578771ef6d4SMalcolm Parsons             args_string)) {
57945d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
58045d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
58145d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
58245d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
583ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
584b9c1b51eSKate Stone     } else {
585212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
586212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
587212130acSEnrico Granata       return false;
588212130acSEnrico Granata     }
589ebc09c36SJim Ingham 
590ebc09c36SJim Ingham     return result.Succeeded();
591ebc09c36SJim Ingham   }
592ebc09c36SJim Ingham };
593ebc09c36SJim Ingham 
594ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
595ebc09c36SJim Ingham // CommandObjectCommandsUnalias
596ebc09c36SJim Ingham 
597b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
598ebc09c36SJim Ingham public:
5997428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
600b9c1b51eSKate Stone       : CommandObjectParsed(
601b9c1b51eSKate Stone             interpreter, "command unalias",
602b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
603b9c1b51eSKate Stone             nullptr) {
604405fe67fSCaroline Tice     CommandArgumentEntry arg;
605405fe67fSCaroline Tice     CommandArgumentData alias_arg;
606405fe67fSCaroline Tice 
607405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
608405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
609405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
610405fe67fSCaroline Tice 
611b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
612b9c1b51eSKate Stone     // argument entry.
613405fe67fSCaroline Tice     arg.push_back(alias_arg);
614405fe67fSCaroline Tice 
615405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
616405fe67fSCaroline Tice     m_arguments.push_back(arg);
617ebc09c36SJim Ingham   }
618ebc09c36SJim Ingham 
6196e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
620ebc09c36SJim Ingham 
621*31fd64acSGongyu Deng   void
622*31fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
623*31fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
624*31fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
625*31fd64acSGongyu Deng       return;
626*31fd64acSGongyu Deng 
627*31fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetAliases()) {
628*31fd64acSGongyu Deng       request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
629*31fd64acSGongyu Deng     }
630*31fd64acSGongyu Deng   }
631*31fd64acSGongyu Deng 
6325a988416SJim Ingham protected:
633b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
634ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
635ebc09c36SJim Ingham     CommandObject *cmd_obj;
636ebc09c36SJim Ingham 
63711eb9c64SZachary Turner     if (args.empty()) {
63811eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
63911eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
64011eb9c64SZachary Turner       return false;
64111eb9c64SZachary Turner     }
64211eb9c64SZachary Turner 
6430d9a201eSRaphael Isemann     auto command_name = args[0].ref();
644a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
6454574a890SZachary Turner     if (!cmd_obj) {
6464574a890SZachary Turner       result.AppendErrorWithFormat(
6474574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
6484574a890SZachary Turner           "current list of commands.\n",
649867e7d17SZachary Turner           args[0].c_str());
6504574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
6514574a890SZachary Turner       return false;
6524574a890SZachary Turner     }
6534574a890SZachary Turner 
654b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
655b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
656b9c1b51eSKate Stone         result.AppendErrorWithFormat(
657b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
658b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
659867e7d17SZachary Turner             args[0].c_str());
660b9c1b51eSKate Stone       } else {
661b9c1b51eSKate Stone         result.AppendErrorWithFormat(
662b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
663867e7d17SZachary Turner             args[0].c_str());
664b547278cSGreg Clayton       }
665ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6664574a890SZachary Turner       return false;
6674574a890SZachary Turner     }
6684574a890SZachary Turner 
669b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
670a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
671b9c1b51eSKate Stone         result.AppendErrorWithFormat(
672867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
673867e7d17SZachary Turner             args[0].c_str());
674ebc09c36SJim Ingham       else
675b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
676867e7d17SZachary Turner                                      args[0].c_str());
677ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6784574a890SZachary Turner       return false;
679ebc09c36SJim Ingham     }
680ebc09c36SJim Ingham 
6814574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
682ebc09c36SJim Ingham     return result.Succeeded();
683ebc09c36SJim Ingham   }
684ebc09c36SJim Ingham };
685ebc09c36SJim Ingham 
686b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
687b547278cSGreg Clayton // CommandObjectCommandsDelete
688b547278cSGreg Clayton 
689b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
690b547278cSGreg Clayton public:
6917428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
692b9c1b51eSKate Stone       : CommandObjectParsed(
693b9c1b51eSKate Stone             interpreter, "command delete",
694b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
695b9c1b51eSKate Stone             nullptr) {
696b547278cSGreg Clayton     CommandArgumentEntry arg;
697b547278cSGreg Clayton     CommandArgumentData alias_arg;
698b547278cSGreg Clayton 
699b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
700b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
701b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
702b547278cSGreg Clayton 
703b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
704b9c1b51eSKate Stone     // argument entry.
705b547278cSGreg Clayton     arg.push_back(alias_arg);
706b547278cSGreg Clayton 
707b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
708b547278cSGreg Clayton     m_arguments.push_back(arg);
709b547278cSGreg Clayton   }
710b547278cSGreg Clayton 
7116e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
712b547278cSGreg Clayton 
713*31fd64acSGongyu Deng   void
714*31fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
715*31fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
716*31fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
717*31fd64acSGongyu Deng       return;
718*31fd64acSGongyu Deng 
719*31fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetCommands()) {
720*31fd64acSGongyu Deng       if (ent.second->IsRemovable())
721*31fd64acSGongyu Deng         request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
722*31fd64acSGongyu Deng     }
723*31fd64acSGongyu Deng   }
724*31fd64acSGongyu Deng 
725b547278cSGreg Clayton protected:
726b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
727b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
728b547278cSGreg Clayton 
72911eb9c64SZachary Turner     if (args.empty()) {
73011eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
73111eb9c64SZachary Turner                                    "defined regular expression command names",
732a449698cSZachary Turner                                    GetCommandName().str().c_str());
73311eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
734d77ea5b2SRaphael Isemann       return false;
73511eb9c64SZachary Turner     }
73611eb9c64SZachary Turner 
7370d9a201eSRaphael Isemann     auto command_name = args[0].ref();
7384574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
73946d4aa21SEnrico Granata       StreamString error_msg_stream;
740d5b44036SJonas Devlieghere       const bool generate_upropos = true;
74146d4aa21SEnrico Granata       const bool generate_type_lookup = false;
742b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
7434574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
744d5b44036SJonas Devlieghere           generate_upropos, generate_type_lookup);
745c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
746b547278cSGreg Clayton       result.SetStatus(eReturnStatusFailed);
7474574a890SZachary Turner       return false;
748b547278cSGreg Clayton     }
749b547278cSGreg Clayton 
7504574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
7514574a890SZachary Turner       result.AppendErrorWithFormat(
7524574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
753867e7d17SZachary Turner           args[0].c_str());
7544574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
7554574a890SZachary Turner       return false;
7564574a890SZachary Turner     }
7574574a890SZachary Turner 
7584574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
7594574a890SZachary Turner     return true;
760b547278cSGreg Clayton   }
761b547278cSGreg Clayton };
762b547278cSGreg Clayton 
763de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
7641f0f5b5bSZachary Turner 
76564becc11SRaphael Isemann #define LLDB_OPTIONS_regex
76664becc11SRaphael Isemann #include "CommandOptions.inc"
7671f0f5b5bSZachary Turner 
7685a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
769de164aaaSGreg Clayton 
770b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
771b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
772de164aaaSGreg Clayton public:
7737428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
774b9c1b51eSKate Stone       : CommandObjectParsed(
775a925974bSAdrian Prantl             interpreter, "command regex",
776a925974bSAdrian Prantl             "Define a custom command in terms of "
777b9c1b51eSKate Stone             "existing commands by matching "
778b9c1b51eSKate Stone             "regular expressions.",
7790e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
780b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
781b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
782b9c1b51eSKate Stone         m_options() {
783b9c1b51eSKate Stone     SetHelpLong(
784b9c1b51eSKate Stone         R"(
785b9c1b51eSKate Stone )"
786b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
787ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
788b9c1b51eSKate Stone using the regular expression substitution format of:"
789b9c1b51eSKate Stone         R"(
790ea671fbdSKate Stone 
791ea671fbdSKate Stone     s/<regex>/<subst>/
792ea671fbdSKate Stone 
793b9c1b51eSKate Stone )"
794b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
795ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
796b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
797b9c1b51eSKate Stone         R"(
798ea671fbdSKate Stone 
799b9c1b51eSKate Stone )"
800b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
801ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
802ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
803b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
804b9c1b51eSKate Stone         R"(
805ea671fbdSKate Stone 
806ea671fbdSKate Stone EXAMPLES
807ea671fbdSKate Stone 
808b9c1b51eSKate Stone )"
809b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
810ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
811b9c1b51eSKate Stone a number follows 'f':"
812b9c1b51eSKate Stone         R"(
813ea671fbdSKate Stone 
814b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
815de164aaaSGreg Clayton   }
816de164aaaSGreg Clayton 
8176e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
818de164aaaSGreg Clayton 
8195a988416SJim Ingham protected:
8200affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
8217ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
8220affb582SDave Lee     if (output_sp && interactive) {
8230affb582SDave Lee       output_sp->PutCString("Enter one or more sed substitution commands in "
824b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
825b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
82644d93782SGreg Clayton       output_sp->Flush();
82744d93782SGreg Clayton     }
82844d93782SGreg Clayton   }
82944d93782SGreg Clayton 
830b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
831b9c1b51eSKate Stone                               std::string &data) override {
83244d93782SGreg Clayton     io_handler.SetIsDone(true);
833d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
83444d93782SGreg Clayton       StringList lines;
835b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
83644d93782SGreg Clayton         bool check_only = false;
8374c78b788SRaphael Isemann         for (const std::string &line : lines) {
8384c78b788SRaphael Isemann           Status error = AppendRegexSubstitution(line, check_only);
839b9c1b51eSKate Stone           if (error.Fail()) {
84057179860SJonas Devlieghere             if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) {
84157179860SJonas Devlieghere               StreamSP out_stream = GetDebugger().GetAsyncOutputStream();
84244d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
84344d93782SGreg Clayton             }
84444d93782SGreg Clayton           }
84544d93782SGreg Clayton         }
84644d93782SGreg Clayton       }
847d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
848d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
84944d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
85044d93782SGreg Clayton       }
85144d93782SGreg Clayton     }
85244d93782SGreg Clayton   }
85344d93782SGreg Clayton 
854b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
8555a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
856b9c1b51eSKate Stone     if (argc == 0) {
857b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
858b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
8590e5e5a79SGreg Clayton       result.SetStatus(eReturnStatusFailed);
86011eb9c64SZachary Turner       return false;
86111eb9c64SZachary Turner     }
86211eb9c64SZachary Turner 
86397206d57SZachary Turner     Status error;
8640d9a201eSRaphael Isemann     auto name = command[0].ref();
865a8f3ae7cSJonas Devlieghere     m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>(
8664574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
8674574a890SZachary Turner         true);
8680e5e5a79SGreg Clayton 
869b9c1b51eSKate Stone     if (argc == 1) {
87057179860SJonas Devlieghere       Debugger &debugger = GetDebugger();
871e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
87244d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
873b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
874b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
87573d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
876514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
877514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
878b9c1b51eSKate Stone           multiple_lines, color_prompt,
879f6913cd7SGreg Clayton           0, // Don't show line numbers
880d77c2e09SJonas Devlieghere           *this, nullptr));
88144d93782SGreg Clayton 
882b9c1b51eSKate Stone       if (io_handler_sp) {
8837ce2de2cSJonas Devlieghere         debugger.RunIOHandlerAsync(io_handler_sp);
884de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
885de164aaaSGreg Clayton       }
886b9c1b51eSKate Stone     } else {
88797d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
88844d93782SGreg Clayton         bool check_only = false;
8890d9a201eSRaphael Isemann         error = AppendRegexSubstitution(entry.ref(), check_only);
8900e5e5a79SGreg Clayton         if (error.Fail())
8910e5e5a79SGreg Clayton           break;
8920e5e5a79SGreg Clayton       }
8930e5e5a79SGreg Clayton 
894b9c1b51eSKate Stone       if (error.Success()) {
8950e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
8960e5e5a79SGreg Clayton       }
8970e5e5a79SGreg Clayton     }
898b9c1b51eSKate Stone     if (error.Fail()) {
8990e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
900de164aaaSGreg Clayton       result.SetStatus(eReturnStatusFailed);
901de164aaaSGreg Clayton     }
9020e5e5a79SGreg Clayton 
903de164aaaSGreg Clayton     return result.Succeeded();
904de164aaaSGreg Clayton   }
905de164aaaSGreg Clayton 
90697206d57SZachary Turner   Status AppendRegexSubstitution(const llvm::StringRef &regex_sed,
907b9c1b51eSKate Stone                                  bool check_only) {
90897206d57SZachary Turner     Status error;
9090e5e5a79SGreg Clayton 
910d5b44036SJonas Devlieghere     if (!m_regex_cmd_up) {
911b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
912b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
913b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9140e5e5a79SGreg Clayton       return error;
915de164aaaSGreg Clayton     }
9160e5e5a79SGreg Clayton 
9170e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
9180e5e5a79SGreg Clayton 
919b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
920b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
921b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
922b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9230e5e5a79SGreg Clayton       return error;
9240e5e5a79SGreg Clayton     }
9250e5e5a79SGreg Clayton 
926b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
927b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
928b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
929b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
9300e5e5a79SGreg Clayton       return error;
9310e5e5a79SGreg Clayton     }
9320e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
93305097246SAdrian Prantl     // use the char that follows 's' as the regex separator character so we can
93405097246SAdrian Prantl     // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9350e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
936b9c1b51eSKate Stone     const size_t second_separator_char_pos =
937b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
9380e5e5a79SGreg Clayton 
939b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
940b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
941b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
9420e5e5a79SGreg Clayton           separator_char,
9430e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
944ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
945b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9460e5e5a79SGreg Clayton       return error;
9470e5e5a79SGreg Clayton     }
9480e5e5a79SGreg Clayton 
949b9c1b51eSKate Stone     const size_t third_separator_char_pos =
950b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
9510e5e5a79SGreg Clayton 
952b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
953b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
954b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
9550e5e5a79SGreg Clayton           separator_char,
9560e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
957ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
958b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9590e5e5a79SGreg Clayton       return error;
9600e5e5a79SGreg Clayton     }
9610e5e5a79SGreg Clayton 
962b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
96305097246SAdrian Prantl       // Make sure that everything that follows the last regex separator char
964b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
965b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
966b9c1b51eSKate Stone           std::string::npos) {
967b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
968b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
969b9c1b51eSKate Stone             "string: '%.*s'",
970b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
9710e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
9720e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
9730e5e5a79SGreg Clayton         return error;
9740e5e5a79SGreg Clayton       }
975b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
976b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
977b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
978b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9790e5e5a79SGreg Clayton           regex_sed.data());
9800e5e5a79SGreg Clayton       return error;
981b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
982b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
983b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
984b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9850e5e5a79SGreg Clayton           regex_sed.data());
9860e5e5a79SGreg Clayton       return error;
9870e5e5a79SGreg Clayton     }
98844d93782SGreg Clayton 
989b9c1b51eSKate Stone     if (!check_only) {
990adcd0268SBenjamin Kramer       std::string regex(std::string(regex_sed.substr(
991adcd0268SBenjamin Kramer           first_separator_char_pos + 1,
992adcd0268SBenjamin Kramer           second_separator_char_pos - first_separator_char_pos - 1)));
993adcd0268SBenjamin Kramer       std::string subst(std::string(regex_sed.substr(
994adcd0268SBenjamin Kramer           second_separator_char_pos + 1,
995adcd0268SBenjamin Kramer           third_separator_char_pos - second_separator_char_pos - 1)));
99643224195SRaphael Isemann       m_regex_cmd_up->AddRegexCommand(regex, subst);
99744d93782SGreg Clayton     }
9980e5e5a79SGreg Clayton     return error;
999de164aaaSGreg Clayton   }
1000de164aaaSGreg Clayton 
1001b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
1002d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
1003d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
1004d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
1005de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1006de164aaaSGreg Clayton       }
1007de164aaaSGreg Clayton     }
1008de164aaaSGreg Clayton   }
1009de164aaaSGreg Clayton 
1010de164aaaSGreg Clayton private:
1011d5b44036SJonas Devlieghere   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up;
1012de164aaaSGreg Clayton 
1013b9c1b51eSKate Stone   class CommandOptions : public Options {
1014de164aaaSGreg Clayton   public:
1015b9c1b51eSKate Stone     CommandOptions() : Options() {}
1016de164aaaSGreg Clayton 
10176e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1018de164aaaSGreg Clayton 
101997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1020b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
102197206d57SZachary Turner       Status error;
10223bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1023de164aaaSGreg Clayton 
1024b9c1b51eSKate Stone       switch (short_option) {
1025de164aaaSGreg Clayton       case 'h':
1026adcd0268SBenjamin Kramer         m_help.assign(std::string(option_arg));
1027de164aaaSGreg Clayton         break;
1028de164aaaSGreg Clayton       case 's':
1029adcd0268SBenjamin Kramer         m_syntax.assign(std::string(option_arg));
1030de164aaaSGreg Clayton         break;
1031de164aaaSGreg Clayton       default:
103236162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1033de164aaaSGreg Clayton       }
1034de164aaaSGreg Clayton 
1035de164aaaSGreg Clayton       return error;
1036de164aaaSGreg Clayton     }
1037de164aaaSGreg Clayton 
1038b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1039de164aaaSGreg Clayton       m_help.clear();
1040de164aaaSGreg Clayton       m_syntax.clear();
1041de164aaaSGreg Clayton     }
1042de164aaaSGreg Clayton 
10431f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
104470602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
10451f0f5b5bSZachary Turner     }
1046de164aaaSGreg Clayton 
1047daed98e5SShivam Mittal     llvm::StringRef GetHelp() { return m_help; }
10486e3d8e7fSEugene Zelenko 
1049daed98e5SShivam Mittal     llvm::StringRef GetSyntax() { return m_syntax; }
10506e3d8e7fSEugene Zelenko 
1051de164aaaSGreg Clayton   protected:
10526e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
10536e3d8e7fSEugene Zelenko 
1054de164aaaSGreg Clayton     std::string m_help;
1055de164aaaSGreg Clayton     std::string m_syntax;
1056de164aaaSGreg Clayton   };
1057de164aaaSGreg Clayton 
1058b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1059de164aaaSGreg Clayton 
10605a988416SJim Ingham   CommandOptions m_options;
1061de164aaaSGreg Clayton };
1062de164aaaSGreg Clayton 
1063b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1064223383edSEnrico Granata public:
1065b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1066b9c1b51eSKate Stone                               std::string funct, std::string help,
1067b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1068a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_function_name(funct),
1069a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_long(false) {
1070735152e3SEnrico Granata     if (!help.empty())
1071442f6530SZachary Turner       SetHelp(help);
1072b9c1b51eSKate Stone     else {
1073735152e3SEnrico Granata       StreamString stream;
1074735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1075c156427dSZachary Turner       SetHelp(stream.GetString());
1076735152e3SEnrico Granata     }
1077223383edSEnrico Granata   }
1078223383edSEnrico Granata 
10796e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1080223383edSEnrico Granata 
1081b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
10825a988416SJim Ingham 
1083b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
10845a988416SJim Ingham 
1085b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
10865a988416SJim Ingham 
1087442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1088442f6530SZachary Turner     if (m_fetched_help_long)
1089442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1090442f6530SZachary Turner 
10912b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1092442f6530SZachary Turner     if (!scripter)
1093442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1094442f6530SZachary Turner 
1095fac939e9SEnrico Granata     std::string docstring;
1096442f6530SZachary Turner     m_fetched_help_long =
1097442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1098fac939e9SEnrico Granata     if (!docstring.empty())
1099442f6530SZachary Turner       SetHelpLong(docstring);
1100fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1101fac939e9SEnrico Granata   }
1102fac939e9SEnrico Granata 
11035a988416SJim Ingham protected:
11044d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1105b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11062b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1107223383edSEnrico Granata 
110897206d57SZachary Turner     Status error;
1109223383edSEnrico Granata 
111070f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
111170f11f88SJim Ingham 
1112a925974bSAdrian Prantl     if (!scripter || !scripter->RunScriptBasedCommand(
1113a925974bSAdrian Prantl                          m_function_name.c_str(), raw_command_line, m_synchro,
1114a925974bSAdrian Prantl                          result, error, m_exe_ctx)) {
1115223383edSEnrico Granata       result.AppendError(error.AsCString());
1116223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1117b9c1b51eSKate Stone     } else {
111870f11f88SJim Ingham       // Don't change the status if the command already set it...
1119b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1120c156427dSZachary Turner         if (result.GetOutputData().empty())
1121223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
112270f11f88SJim Ingham         else
112370f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
112470f11f88SJim Ingham       }
112570f11f88SJim Ingham     }
1126223383edSEnrico Granata 
1127223383edSEnrico Granata     return result.Succeeded();
1128223383edSEnrico Granata   }
1129223383edSEnrico Granata 
11306e3d8e7fSEugene Zelenko private:
11316e3d8e7fSEugene Zelenko   std::string m_function_name;
11326e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11336e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1134223383edSEnrico Granata };
1135223383edSEnrico Granata 
1136b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
11379fe00e52SEnrico Granata public:
11389fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
11399fe00e52SEnrico Granata                                std::string name,
11400641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1141b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1142a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp),
1143a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_short(false),
1144b9c1b51eSKate Stone         m_fetched_help_long(false) {
11459fe00e52SEnrico Granata     StreamString stream;
11469fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1147c156427dSZachary Turner     SetHelp(stream.GetString());
11482b29b432SJonas Devlieghere     if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter())
1149e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
11509fe00e52SEnrico Granata   }
11519fe00e52SEnrico Granata 
11526e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
11539fe00e52SEnrico Granata 
1154b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
11559fe00e52SEnrico Granata 
1156b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
11579fe00e52SEnrico Granata 
1158442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1159442f6530SZachary Turner     if (m_fetched_help_short)
1160442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11612b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1162442f6530SZachary Turner     if (!scripter)
1163442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11646f79bb2dSEnrico Granata     std::string docstring;
1165b9c1b51eSKate Stone     m_fetched_help_short =
1166b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
11676f79bb2dSEnrico Granata     if (!docstring.empty())
1168442f6530SZachary Turner       SetHelp(docstring);
1169442f6530SZachary Turner 
11706f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
11716f79bb2dSEnrico Granata   }
11726f79bb2dSEnrico Granata 
1173442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1174442f6530SZachary Turner     if (m_fetched_help_long)
1175442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1176442f6530SZachary Turner 
11772b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1178442f6530SZachary Turner     if (!scripter)
1179442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1180442f6530SZachary Turner 
11816f79bb2dSEnrico Granata     std::string docstring;
1182b9c1b51eSKate Stone     m_fetched_help_long =
1183b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
11846f79bb2dSEnrico Granata     if (!docstring.empty())
1185442f6530SZachary Turner       SetHelpLong(docstring);
11869fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
11879fe00e52SEnrico Granata   }
11889fe00e52SEnrico Granata 
11899fe00e52SEnrico Granata protected:
11904d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1191b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11922b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
11939fe00e52SEnrico Granata 
119497206d57SZachary Turner     Status error;
11959fe00e52SEnrico Granata 
11969fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
11979fe00e52SEnrico Granata 
1198b9c1b51eSKate Stone     if (!scripter ||
1199b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1200b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
12019fe00e52SEnrico Granata       result.AppendError(error.AsCString());
12029fe00e52SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1203b9c1b51eSKate Stone     } else {
12049fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1205b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1206c156427dSZachary Turner         if (result.GetOutputData().empty())
12079fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
12089fe00e52SEnrico Granata         else
12099fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
12109fe00e52SEnrico Granata       }
12119fe00e52SEnrico Granata     }
12129fe00e52SEnrico Granata 
12139fe00e52SEnrico Granata     return result.Succeeded();
12149fe00e52SEnrico Granata   }
12159fe00e52SEnrico Granata 
12166e3d8e7fSEugene Zelenko private:
12176e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
12186e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
12196e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
12206e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
12219fe00e52SEnrico Granata };
12229fe00e52SEnrico Granata 
1223a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
122464becc11SRaphael Isemann #define LLDB_OPTIONS_script_import
122564becc11SRaphael Isemann #include "CommandOptions.inc"
12261f0f5b5bSZachary Turner 
1227b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
12285a988416SJim Ingham public:
1229b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1230b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1231b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1232b9c1b51eSKate Stone         m_options() {
12335a988416SJim Ingham     CommandArgumentEntry arg1;
12345a988416SJim Ingham     CommandArgumentData cmd_arg;
12355a988416SJim Ingham 
12365a988416SJim Ingham     // Define the first (and only) variant of this arg.
12375a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
12383b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
12395a988416SJim Ingham 
1240b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1241b9c1b51eSKate Stone     // argument entry.
12425a988416SJim Ingham     arg1.push_back(cmd_arg);
12435a988416SJim Ingham 
12445a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
12455a988416SJim Ingham     m_arguments.push_back(arg1);
12465a988416SJim Ingham   }
12475a988416SJim Ingham 
12486e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
12495a988416SJim Ingham 
1250ae34ed2cSRaphael Isemann   void
1251ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
12522443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
1253b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1254b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1255a2e76c0bSRaphael Isemann         request, nullptr);
12565a988416SJim Ingham   }
12575a988416SJim Ingham 
1258b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
12595a988416SJim Ingham 
12605a988416SJim Ingham protected:
1261b9c1b51eSKate Stone   class CommandOptions : public Options {
12620a305db7SEnrico Granata   public:
1263b9c1b51eSKate Stone     CommandOptions() : Options() {}
12640a305db7SEnrico Granata 
12656e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
12660a305db7SEnrico Granata 
126797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1268b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
126997206d57SZachary Turner       Status error;
12703bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
12710a305db7SEnrico Granata 
1272b9c1b51eSKate Stone       switch (short_option) {
12730a305db7SEnrico Granata       case 'r':
127415625112SJonas Devlieghere         // NO-OP
12750a305db7SEnrico Granata         break;
12760a305db7SEnrico Granata       default:
127736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
12780a305db7SEnrico Granata       }
12790a305db7SEnrico Granata 
12800a305db7SEnrico Granata       return error;
12810a305db7SEnrico Granata     }
12820a305db7SEnrico Granata 
1283b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
12840a305db7SEnrico Granata     }
12850a305db7SEnrico Granata 
12861f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
128770602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
12881f0f5b5bSZachary Turner     }
12890a305db7SEnrico Granata   };
12900a305db7SEnrico Granata 
1291b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
129211eb9c64SZachary Turner     if (command.empty()) {
12933b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1294a9dbf432SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1295a9dbf432SEnrico Granata       return false;
1296a9dbf432SEnrico Granata     }
1297a9dbf432SEnrico Granata 
129811eb9c64SZachary Turner     for (auto &entry : command.entries()) {
129997206d57SZachary Turner       Status error;
1300a9dbf432SEnrico Granata 
1301c9d645d3SGreg Clayton       const bool init_session = true;
1302b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
130311eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
130411eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
130505097246SAdrian Prantl       // script imports" in __lldb_init_module the real fix is to have
130605097246SAdrian Prantl       // recursive commands possible with a CommandInvocation object separate
130705097246SAdrian Prantl       // from the CommandObject itself, so that recursive command invocations
130805097246SAdrian Prantl       // won't stomp on each other (wrt to execution contents, options, and
130905097246SAdrian Prantl       // more)
1310078551c7SEnrico Granata       m_exe_ctx.Clear();
13112b29b432SJonas Devlieghere       if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule(
131215625112SJonas Devlieghere               entry.c_str(), init_session, error)) {
1313a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1314b9c1b51eSKate Stone       } else {
1315b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1316b9c1b51eSKate Stone                                      error.AsCString());
1317a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusFailed);
1318a9dbf432SEnrico Granata       }
13193b00e35bSEnrico Granata     }
1320a9dbf432SEnrico Granata 
1321a9dbf432SEnrico Granata     return result.Succeeded();
1322a9dbf432SEnrico Granata   }
13230a305db7SEnrico Granata 
13245a988416SJim Ingham   CommandOptions m_options;
1325a9dbf432SEnrico Granata };
1326223383edSEnrico Granata 
1327223383edSEnrico Granata // CommandObjectCommandsScriptAdd
13288fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = {
1329e063ecccSJonas Devlieghere     {
1330e063ecccSJonas Devlieghere         eScriptedCommandSynchronicitySynchronous,
1331e063ecccSJonas Devlieghere         "synchronous",
1332e063ecccSJonas Devlieghere         "Run synchronous",
1333e063ecccSJonas Devlieghere     },
1334e063ecccSJonas Devlieghere     {
1335e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityAsynchronous,
1336e063ecccSJonas Devlieghere         "asynchronous",
1337e063ecccSJonas Devlieghere         "Run asynchronous",
1338e063ecccSJonas Devlieghere     },
1339e063ecccSJonas Devlieghere     {
1340e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityCurrentValue,
1341e063ecccSJonas Devlieghere         "current",
1342e063ecccSJonas Devlieghere         "Do not alter current setting",
1343e063ecccSJonas Devlieghere     },
1344e063ecccSJonas Devlieghere };
13451f0f5b5bSZachary Turner 
13468fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() {
13478fe53c49STatyana Krasnukha   return OptionEnumValues(g_script_synchro_type);
13488fe53c49STatyana Krasnukha }
13498fe53c49STatyana Krasnukha 
135064becc11SRaphael Isemann #define LLDB_OPTIONS_script_add
135164becc11SRaphael Isemann #include "CommandOptions.inc"
13521f0f5b5bSZachary Turner 
1353b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1354b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
13555a988416SJim Ingham public:
1356b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1357b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
13585a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
13596e3d8e7fSEugene Zelenko                             nullptr),
1360b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
13615a988416SJim Ingham     CommandArgumentEntry arg1;
13625a988416SJim Ingham     CommandArgumentData cmd_arg;
13635a988416SJim Ingham 
13645a988416SJim Ingham     // Define the first (and only) variant of this arg.
13655a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
13665a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
13675a988416SJim Ingham 
1368b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1369b9c1b51eSKate Stone     // argument entry.
13705a988416SJim Ingham     arg1.push_back(cmd_arg);
13715a988416SJim Ingham 
13725a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
13735a988416SJim Ingham     m_arguments.push_back(arg1);
13745a988416SJim Ingham   }
13755a988416SJim Ingham 
13766e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
13775a988416SJim Ingham 
1378b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13795a988416SJim Ingham 
13805a988416SJim Ingham protected:
1381b9c1b51eSKate Stone   class CommandOptions : public Options {
1382223383edSEnrico Granata   public:
1383b9c1b51eSKate Stone     CommandOptions()
1384b9c1b51eSKate Stone         : Options(), m_class_name(), m_funct_name(), m_short_help(),
1385b9c1b51eSKate Stone           m_synchronicity(eScriptedCommandSynchronicitySynchronous) {}
1386223383edSEnrico Granata 
13876e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1388223383edSEnrico Granata 
138997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1390b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
139197206d57SZachary Turner       Status error;
13923bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1393223383edSEnrico Granata 
1394b9c1b51eSKate Stone       switch (short_option) {
1395223383edSEnrico Granata       case 'f':
1396fe11483bSZachary Turner         if (!option_arg.empty())
1397adcd0268SBenjamin Kramer           m_funct_name = std::string(option_arg);
1398735152e3SEnrico Granata         break;
13999fe00e52SEnrico Granata       case 'c':
1400fe11483bSZachary Turner         if (!option_arg.empty())
1401adcd0268SBenjamin Kramer           m_class_name = std::string(option_arg);
14029fe00e52SEnrico Granata         break;
1403735152e3SEnrico Granata       case 'h':
1404fe11483bSZachary Turner         if (!option_arg.empty())
1405adcd0268SBenjamin Kramer           m_short_help = std::string(option_arg);
1406223383edSEnrico Granata         break;
14070a305db7SEnrico Granata       case 's':
1408b9c1b51eSKate Stone         m_synchronicity =
140947cbf4a0SPavel Labath             (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum(
1410fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
14110a305db7SEnrico Granata         if (!error.Success())
1412b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1413fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1414fe11483bSZachary Turner               option_arg.str().c_str());
14150a305db7SEnrico Granata         break;
1416223383edSEnrico Granata       default:
141736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1418223383edSEnrico Granata       }
1419223383edSEnrico Granata 
1420223383edSEnrico Granata       return error;
1421223383edSEnrico Granata     }
1422223383edSEnrico Granata 
1423b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
14249fe00e52SEnrico Granata       m_class_name.clear();
1425735152e3SEnrico Granata       m_funct_name.clear();
1426735152e3SEnrico Granata       m_short_help.clear();
142744d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1428223383edSEnrico Granata     }
1429223383edSEnrico Granata 
14301f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
143170602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
14321f0f5b5bSZachary Turner     }
1433223383edSEnrico Granata 
1434223383edSEnrico Granata     // Instance variables to hold the values for command options.
1435223383edSEnrico Granata 
14369fe00e52SEnrico Granata     std::string m_class_name;
1437223383edSEnrico Granata     std::string m_funct_name;
1438735152e3SEnrico Granata     std::string m_short_help;
143944d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1440223383edSEnrico Granata   };
1441223383edSEnrico Granata 
14420affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
14437ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
14440affb582SDave Lee     if (output_sp && interactive) {
144544d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
144644d93782SGreg Clayton       output_sp->Flush();
1447223383edSEnrico Granata     }
1448223383edSEnrico Granata   }
1449223383edSEnrico Granata 
1450b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1451b9c1b51eSKate Stone                               std::string &data) override {
14527ca15ba7SLawrence D'Anna     StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
145344d93782SGreg Clayton 
14542b29b432SJonas Devlieghere     ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1455b9c1b51eSKate Stone     if (interpreter) {
145644d93782SGreg Clayton 
145744d93782SGreg Clayton       StringList lines;
145844d93782SGreg Clayton       lines.SplitIntoLines(data);
1459b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1460a73b7df7SEnrico Granata         std::string funct_name_str;
1461b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1462b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1463b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1464b9c1b51eSKate Stone                              "add python command.\n");
146544d93782SGreg Clayton             error_sp->Flush();
1466b9c1b51eSKate Stone           } else {
1467223383edSEnrico Granata             // everything should be fine now, let's add this alias
1468223383edSEnrico Granata 
1469b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1470771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
147144d93782SGreg Clayton                 m_synchronicity));
1472223383edSEnrico Granata 
1473b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1474b9c1b51eSKate Stone                                               true)) {
1475b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1476b9c1b51eSKate Stone                                "add python command.\n");
147744d93782SGreg Clayton               error_sp->Flush();
1478223383edSEnrico Granata             }
1479223383edSEnrico Granata           }
1480b9c1b51eSKate Stone         } else {
1481b9c1b51eSKate Stone           error_sp->Printf(
1482b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
148344d93782SGreg Clayton           error_sp->Flush();
148444d93782SGreg Clayton         }
1485b9c1b51eSKate Stone       } else {
148644d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
148744d93782SGreg Clayton         error_sp->Flush();
148844d93782SGreg Clayton       }
1489b9c1b51eSKate Stone     } else {
1490b9c1b51eSKate Stone       error_sp->Printf(
1491b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
149244d93782SGreg Clayton       error_sp->Flush();
149344d93782SGreg Clayton     }
149444d93782SGreg Clayton 
149544d93782SGreg Clayton     io_handler.SetIsDone(true);
149644d93782SGreg Clayton   }
1497223383edSEnrico Granata 
1498b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
149957179860SJonas Devlieghere     if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) {
1500b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1501b9c1b51eSKate Stone                          "commands is currently Python");
150299f0b8f9SEnrico Granata       result.SetStatus(eReturnStatusFailed);
150399f0b8f9SEnrico Granata       return false;
150499f0b8f9SEnrico Granata     }
150599f0b8f9SEnrico Granata 
150611eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1507223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1508223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1509223383edSEnrico Granata       return false;
1510223383edSEnrico Granata     }
1511223383edSEnrico Granata 
1512735152e3SEnrico Granata     // Store the options in case we get multi-line input
1513adcd0268SBenjamin Kramer     m_cmd_name = std::string(command[0].ref());
1514735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
151544d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1516223383edSEnrico Granata 
1517b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1518b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1519b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1520b9c1b51eSKate Stone             "     ", // Prompt
1521a6faf851SJonas Devlieghere             *this);  // IOHandlerDelegate
1522b9c1b51eSKate Stone       } else {
1523b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1524b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1525b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1526b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1527223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1528b9c1b51eSKate Stone         } else {
1529223383edSEnrico Granata           result.AppendError("cannot add command");
1530223383edSEnrico Granata           result.SetStatus(eReturnStatusFailed);
1531223383edSEnrico Granata         }
1532223383edSEnrico Granata       }
1533b9c1b51eSKate Stone     } else {
15342b29b432SJonas Devlieghere       ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1535b9c1b51eSKate Stone       if (!interpreter) {
15369fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
15379fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15389fe00e52SEnrico Granata         return false;
15399fe00e52SEnrico Granata       }
15409fe00e52SEnrico Granata 
1541b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1542b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1543b9c1b51eSKate Stone       if (!cmd_obj_sp) {
15449fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
15459fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15469fe00e52SEnrico Granata         return false;
15479fe00e52SEnrico Granata       }
15489fe00e52SEnrico Granata 
1549b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1550b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1551b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
15529fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1553b9c1b51eSKate Stone       } else {
15549fe00e52SEnrico Granata         result.AppendError("cannot add command");
15559fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15569fe00e52SEnrico Granata       }
15579fe00e52SEnrico Granata     }
1558223383edSEnrico Granata 
1559223383edSEnrico Granata     return result.Succeeded();
1560223383edSEnrico Granata   }
15615a988416SJim Ingham 
15625a988416SJim Ingham   CommandOptions m_options;
156344d93782SGreg Clayton   std::string m_cmd_name;
1564735152e3SEnrico Granata   std::string m_short_help;
156544d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1566223383edSEnrico Granata };
1567223383edSEnrico Granata 
1568223383edSEnrico Granata // CommandObjectCommandsScriptList
1569223383edSEnrico Granata 
1570b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1571223383edSEnrico Granata public:
1572b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1573b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1574b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1575223383edSEnrico Granata 
15766e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1577223383edSEnrico Granata 
1578b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1579d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1580d77ea5b2SRaphael Isemann       result.AppendError("'command script list' doesn't take any arguments");
1581d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1582d77ea5b2SRaphael Isemann       return false;
1583d77ea5b2SRaphael Isemann     }
1584d77ea5b2SRaphael Isemann 
1585b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1586223383edSEnrico Granata 
1587223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1588223383edSEnrico Granata 
1589223383edSEnrico Granata     return true;
1590223383edSEnrico Granata   }
1591223383edSEnrico Granata };
1592223383edSEnrico Granata 
1593223383edSEnrico Granata // CommandObjectCommandsScriptClear
1594223383edSEnrico Granata 
1595b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1596223383edSEnrico Granata public:
1597b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1598b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1599b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1600223383edSEnrico Granata 
16016e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1602223383edSEnrico Granata 
16035a988416SJim Ingham protected:
1604b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1605d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1606d77ea5b2SRaphael Isemann       result.AppendError("'command script clear' doesn't take any arguments");
1607d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1608d77ea5b2SRaphael Isemann       return false;
1609d77ea5b2SRaphael Isemann     }
1610d77ea5b2SRaphael Isemann 
1611223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1612223383edSEnrico Granata 
1613223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1614223383edSEnrico Granata 
1615223383edSEnrico Granata     return true;
1616223383edSEnrico Granata   }
1617223383edSEnrico Granata };
1618223383edSEnrico Granata 
1619223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1620223383edSEnrico Granata 
1621b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1622223383edSEnrico Granata public:
1623b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1624b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1625b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1626223383edSEnrico Granata     CommandArgumentEntry arg1;
1627223383edSEnrico Granata     CommandArgumentData cmd_arg;
1628223383edSEnrico Granata 
1629223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1630223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1631223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1632223383edSEnrico Granata 
1633b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1634b9c1b51eSKate Stone     // argument entry.
1635223383edSEnrico Granata     arg1.push_back(cmd_arg);
1636223383edSEnrico Granata 
1637223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1638223383edSEnrico Granata     m_arguments.push_back(arg1);
1639223383edSEnrico Granata   }
1640223383edSEnrico Granata 
16416e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1642223383edSEnrico Granata 
16432e8f304fSGongyu Deng   void
16442e8f304fSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
16452e8f304fSGongyu Deng                            OptionElementVector &opt_element_vector) override {
16462e8f304fSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
16472e8f304fSGongyu Deng       return;
16482e8f304fSGongyu Deng 
16492ebe30c6SRaphael Isemann     for (const auto &c : m_interpreter.GetUserCommands())
16502ebe30c6SRaphael Isemann       request.TryCompleteCurrentArg(c.first, c.second->GetHelp());
16512e8f304fSGongyu Deng   }
16522e8f304fSGongyu Deng 
16535a988416SJim Ingham protected:
1654b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1655223383edSEnrico Granata 
165611eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1657223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1658223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1659223383edSEnrico Granata       return false;
1660223383edSEnrico Granata     }
1661223383edSEnrico Granata 
16620d9a201eSRaphael Isemann     auto cmd_name = command[0].ref();
1663223383edSEnrico Granata 
16644574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
16654574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1666867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
1667223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
16684574a890SZachary Turner       return false;
1669223383edSEnrico Granata     }
1670223383edSEnrico Granata 
16714574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
16724574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
16734574a890SZachary Turner     return true;
1674223383edSEnrico Granata   }
1675223383edSEnrico Granata };
1676223383edSEnrico Granata 
1677223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1678223383edSEnrico Granata 
1679223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1680223383edSEnrico Granata 
1681b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1682223383edSEnrico Granata public:
16837428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1684b9c1b51eSKate Stone       : CommandObjectMultiword(
1685a925974bSAdrian Prantl             interpreter, "command script",
1686a925974bSAdrian Prantl             "Commands for managing custom "
1687b9c1b51eSKate Stone             "commands implemented by "
1688b9c1b51eSKate Stone             "interpreter scripts.",
1689b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1690b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1691b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1692b9c1b51eSKate Stone     LoadSubCommand(
1693b9c1b51eSKate Stone         "delete",
1694b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1695b9c1b51eSKate Stone     LoadSubCommand(
1696b9c1b51eSKate Stone         "clear",
1697b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1698b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1699b9c1b51eSKate Stone                                interpreter)));
1700b9c1b51eSKate Stone     LoadSubCommand(
1701b9c1b51eSKate Stone         "import",
1702b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1703223383edSEnrico Granata   }
1704223383edSEnrico Granata 
17056e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1706223383edSEnrico Granata };
1707223383edSEnrico Granata 
1708ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1709ebc09c36SJim Ingham 
1710ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1711ebc09c36SJim Ingham 
1712b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1713b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1714b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1715b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1716b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1717b9c1b51eSKate Stone   LoadSubCommand("source",
1718b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1719b9c1b51eSKate Stone   LoadSubCommand("alias",
1720b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1721b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1722b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1723b9c1b51eSKate Stone   LoadSubCommand("delete",
1724b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1725b9c1b51eSKate Stone   LoadSubCommand(
1726b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1727b9c1b51eSKate Stone   LoadSubCommand(
1728b9c1b51eSKate Stone       "script",
1729b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1730ebc09c36SJim Ingham }
1731ebc09c36SJim Ingham 
17326e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1733