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 
96e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h"
1046d4aa21SEnrico Granata #include "CommandObjectHelp.h"
119390b346SJonas Devlieghere #include "CommandObjectRegexCommand.h"
12ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1344d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
147594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
15ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
16ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
1747cbf4a0SPavel Labath #include "lldb/Interpreter/OptionArgParser.h"
18012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
1945d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h"
207594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
21ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
2299f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
23145d95c9SPavel Labath #include "lldb/Utility/Args.h"
24573ab909SZachary Turner #include "lldb/Utility/StringList.h"
259390b346SJonas Devlieghere #include "llvm/ADT/StringRef.h"
26ebc09c36SJim Ingham 
27ebc09c36SJim Ingham using namespace lldb;
28ebc09c36SJim Ingham using namespace lldb_private;
29ebc09c36SJim Ingham 
30ebc09c36SJim Ingham // CommandObjectCommandsSource
31ebc09c36SJim Ingham 
3264becc11SRaphael Isemann #define LLDB_OPTIONS_source
3364becc11SRaphael Isemann #include "CommandOptions.inc"
341f0f5b5bSZachary Turner 
35b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed {
365a988416SJim Ingham public:
377428a18cSKate Stone   CommandObjectCommandsSource(CommandInterpreter &interpreter)
38b9c1b51eSKate Stone       : CommandObjectParsed(
39b9c1b51eSKate Stone             interpreter, "command source",
40b9c1b51eSKate Stone             "Read and execute LLDB commands from the file <filename>.",
416e3d8e7fSEugene Zelenko             nullptr),
42b9c1b51eSKate Stone         m_options() {
435a988416SJim Ingham     CommandArgumentEntry arg;
445a988416SJim Ingham     CommandArgumentData file_arg;
455a988416SJim Ingham 
465a988416SJim Ingham     // Define the first (and only) variant of this arg.
475a988416SJim Ingham     file_arg.arg_type = eArgTypeFilename;
485a988416SJim Ingham     file_arg.arg_repetition = eArgRepeatPlain;
495a988416SJim Ingham 
50b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
51b9c1b51eSKate Stone     // argument entry.
525a988416SJim Ingham     arg.push_back(file_arg);
535a988416SJim Ingham 
545a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
555a988416SJim Ingham     m_arguments.push_back(arg);
565a988416SJim Ingham   }
575a988416SJim Ingham 
586e3d8e7fSEugene Zelenko   ~CommandObjectCommandsSource() override = default;
595a988416SJim Ingham 
60b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
61b9c1b51eSKate Stone                                uint32_t index) override {
625a988416SJim Ingham     return "";
635a988416SJim Ingham   }
645a988416SJim Ingham 
65ae34ed2cSRaphael Isemann   void
66ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
672443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
68b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
69b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
70a2e76c0bSRaphael Isemann         request, nullptr);
715a988416SJim Ingham   }
725a988416SJim Ingham 
73b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
745a988416SJim Ingham 
755a988416SJim Ingham protected:
76b9c1b51eSKate Stone   class CommandOptions : public Options {
77e16c50a1SJim Ingham   public:
78b9c1b51eSKate Stone     CommandOptions()
79b9c1b51eSKate Stone         : Options(), m_stop_on_error(true), m_silent_run(false),
80b9c1b51eSKate Stone           m_stop_on_continue(true) {}
81e16c50a1SJim Ingham 
826e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
83e16c50a1SJim Ingham 
8497206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
85b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
8697206d57SZachary Turner       Status error;
873bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
88e16c50a1SJim Ingham 
89b9c1b51eSKate Stone       switch (short_option) {
90e16c50a1SJim Ingham       case 'e':
91fe11483bSZachary Turner         error = m_stop_on_error.SetValueFromString(option_arg);
92e16c50a1SJim Ingham         break;
93340b0309SGreg Clayton 
94e16c50a1SJim Ingham       case 'c':
95fe11483bSZachary Turner         error = m_stop_on_continue.SetValueFromString(option_arg);
96e16c50a1SJim Ingham         break;
97340b0309SGreg Clayton 
9860986174SMichael Sartain       case 's':
99fe11483bSZachary Turner         error = m_silent_run.SetValueFromString(option_arg);
10060986174SMichael Sartain         break;
101340b0309SGreg Clayton 
102e16c50a1SJim Ingham       default:
10336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
104e16c50a1SJim Ingham       }
105e16c50a1SJim Ingham 
106e16c50a1SJim Ingham       return error;
107e16c50a1SJim Ingham     }
108e16c50a1SJim Ingham 
109b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
110012d4fcaSEnrico Granata       m_stop_on_error.Clear();
111340b0309SGreg Clayton       m_silent_run.Clear();
112340b0309SGreg Clayton       m_stop_on_continue.Clear();
113e16c50a1SJim Ingham     }
114e16c50a1SJim Ingham 
1151f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
11670602439SZachary Turner       return llvm::makeArrayRef(g_source_options);
1171f0f5b5bSZachary Turner     }
118e16c50a1SJim Ingham 
119e16c50a1SJim Ingham     // Instance variables to hold the values for command options.
120e16c50a1SJim Ingham 
121012d4fcaSEnrico Granata     OptionValueBoolean m_stop_on_error;
122340b0309SGreg Clayton     OptionValueBoolean m_silent_run;
123340b0309SGreg Clayton     OptionValueBoolean m_stop_on_continue;
124e16c50a1SJim Ingham   };
125e16c50a1SJim Ingham 
126b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1274574a890SZachary Turner     if (command.GetArgumentCount() != 1) {
1284574a890SZachary Turner       result.AppendErrorWithFormat(
1294574a890SZachary Turner           "'%s' takes exactly one executable filename argument.\n",
1304574a890SZachary Turner           GetCommandName().str().c_str());
1314574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
1324574a890SZachary Turner       return false;
1334574a890SZachary Turner     }
134ebc09c36SJim Ingham 
1350d9a201eSRaphael Isemann     FileSpec cmd_file(command[0].ref());
1368f3be7a3SJonas Devlieghere     FileSystem::Instance().Resolve(cmd_file);
137122a4ebdSPavel Labath     ExecutionContext *exe_ctx = nullptr; // Just use the default context.
138ebc09c36SJim Ingham 
139340b0309SGreg Clayton     // If any options were set, then use them
140340b0309SGreg Clayton     if (m_options.m_stop_on_error.OptionWasSet() ||
141340b0309SGreg Clayton         m_options.m_silent_run.OptionWasSet() ||
142b9c1b51eSKate Stone         m_options.m_stop_on_continue.OptionWasSet()) {
143122a4ebdSPavel Labath       // Use user set settings
144122a4ebdSPavel Labath       CommandInterpreterRunOptions options;
145122a4ebdSPavel Labath 
1461c19b74cSJonas Devlieghere       if (m_options.m_stop_on_continue.OptionWasSet())
1471c19b74cSJonas Devlieghere         options.SetStopOnContinue(
1481c19b74cSJonas Devlieghere             m_options.m_stop_on_continue.GetCurrentValue());
1491c19b74cSJonas Devlieghere 
1501c19b74cSJonas Devlieghere       if (m_options.m_stop_on_error.OptionWasSet())
15126c7bf93SJim Ingham         options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue());
152c678ed77SStefan Granitz 
153c678ed77SStefan Granitz       // Individual silent setting is override for global command echo settings.
154c678ed77SStefan Granitz       if (m_options.m_silent_run.GetCurrentValue()) {
155c678ed77SStefan Granitz         options.SetSilent(true);
156c678ed77SStefan Granitz       } else {
157c678ed77SStefan Granitz         options.SetPrintResults(true);
158c0b48ab6SJonas Devlieghere         options.SetPrintErrors(true);
159c678ed77SStefan Granitz         options.SetEchoCommands(m_interpreter.GetEchoCommands());
160c678ed77SStefan Granitz         options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands());
161c678ed77SStefan Granitz       }
162a01b26fbSTatyana Krasnukha 
163122a4ebdSPavel Labath       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
164122a4ebdSPavel Labath     } else {
165122a4ebdSPavel Labath       // No options were set, inherit any settings from nested "command source"
166122a4ebdSPavel Labath       // commands, or set to sane default settings...
167122a4ebdSPavel Labath       CommandInterpreterRunOptions options;
168122a4ebdSPavel Labath       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
169122a4ebdSPavel Labath     }
170ebc09c36SJim Ingham     return result.Succeeded();
171ebc09c36SJim Ingham   }
1726e3d8e7fSEugene Zelenko 
1735a988416SJim Ingham   CommandOptions m_options;
174ebc09c36SJim Ingham };
175ebc09c36SJim Ingham 
176ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
177ebc09c36SJim Ingham // CommandObjectCommandsAlias
178ebc09c36SJim Ingham 
17964becc11SRaphael Isemann #define LLDB_OPTIONS_alias
18064becc11SRaphael Isemann #include "CommandOptions.inc"
1811f0f5b5bSZachary Turner 
182b9c1b51eSKate Stone static const char *g_python_command_instructions =
183b9c1b51eSKate Stone     "Enter your Python command(s). Type 'DONE' to end.\n"
184be93a35aSEnrico Granata     "You must define a Python function with this signature:\n"
18544d93782SGreg Clayton     "def my_command_impl(debugger, args, result, internal_dict):\n";
186be93a35aSEnrico Granata 
187b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw {
18845d0e238SEnrico Granata protected:
189b9c1b51eSKate Stone   class CommandOptions : public OptionGroup {
190ebc09c36SJim Ingham   public:
191b9c1b51eSKate Stone     CommandOptions() : OptionGroup(), m_help(), m_long_help() {}
19245d0e238SEnrico Granata 
19345d0e238SEnrico Granata     ~CommandOptions() override = default;
19445d0e238SEnrico Granata 
1951f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
19670602439SZachary Turner       return llvm::makeArrayRef(g_alias_options);
1971f0f5b5bSZachary Turner     }
19845d0e238SEnrico Granata 
19997206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value,
200b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
20197206d57SZachary Turner       Status error;
20245d0e238SEnrico Granata 
2031f0f5b5bSZachary Turner       const int short_option = GetDefinitions()[option_idx].short_option;
2048cef4b0bSZachary Turner       std::string option_str(option_value);
20545d0e238SEnrico Granata 
206b9c1b51eSKate Stone       switch (short_option) {
20745d0e238SEnrico Granata       case 'h':
2088cef4b0bSZachary Turner         m_help.SetCurrentValue(option_str);
20945d0e238SEnrico Granata         m_help.SetOptionWasSet();
21045d0e238SEnrico Granata         break;
21145d0e238SEnrico Granata 
21245d0e238SEnrico Granata       case 'H':
2138cef4b0bSZachary Turner         m_long_help.SetCurrentValue(option_str);
21445d0e238SEnrico Granata         m_long_help.SetOptionWasSet();
21545d0e238SEnrico Granata         break;
21645d0e238SEnrico Granata 
21745d0e238SEnrico Granata       default:
21836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
21945d0e238SEnrico Granata       }
22045d0e238SEnrico Granata 
22145d0e238SEnrico Granata       return error;
22245d0e238SEnrico Granata     }
22345d0e238SEnrico Granata 
224b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
22545d0e238SEnrico Granata       m_help.Clear();
22645d0e238SEnrico Granata       m_long_help.Clear();
22745d0e238SEnrico Granata     }
22845d0e238SEnrico Granata 
22945d0e238SEnrico Granata     OptionValueString m_help;
23045d0e238SEnrico Granata     OptionValueString m_long_help;
23145d0e238SEnrico Granata   };
23245d0e238SEnrico Granata 
23345d0e238SEnrico Granata   OptionGroupOptions m_option_group;
23445d0e238SEnrico Granata   CommandOptions m_command_options;
23545d0e238SEnrico Granata 
23645d0e238SEnrico Granata public:
237b9c1b51eSKate Stone   Options *GetOptions() override { return &m_option_group; }
23845d0e238SEnrico Granata 
2397428a18cSKate Stone   CommandObjectCommandsAlias(CommandInterpreter &interpreter)
240b9c1b51eSKate Stone       : CommandObjectRaw(
241b9c1b51eSKate Stone             interpreter, "command alias",
242a449698cSZachary Turner             "Define a custom command in terms of an existing command."),
243b9c1b51eSKate Stone         m_option_group(), m_command_options() {
24445d0e238SEnrico Granata     m_option_group.Append(&m_command_options);
24545d0e238SEnrico Granata     m_option_group.Finalize();
24645d0e238SEnrico Granata 
247ebc09c36SJim Ingham     SetHelpLong(
248ea671fbdSKate Stone         "'alias' allows the user to create a short-cut or abbreviation for long \
249ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
250b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:"
251b9c1b51eSKate Stone         R"(
252ea671fbdSKate Stone 
253ea671fbdSKate Stone (lldb) command alias sc script
254ea671fbdSKate Stone 
255ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
256ea671fbdSKate Stone 
257ea671fbdSKate Stone (lldb) command alias bp breakpoint
258ea671fbdSKate Stone 
259b9c1b51eSKate Stone )"
260b9c1b51eSKate Stone         "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
261ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
262b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'."
263b9c1b51eSKate Stone         R"(
264ea671fbdSKate Stone 
265ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
266ea671fbdSKate Stone 
267ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
268ea671fbdSKate Stone 
269b9c1b51eSKate Stone )"
270b9c1b51eSKate Stone         "An alias can include some options for the command, with the values either \
271ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
272ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
273b9c1b51eSKate Stone shows how to create aliases with options:"
274b9c1b51eSKate Stone         R"(
275ea671fbdSKate Stone 
276ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
277ea671fbdSKate Stone 
278b9c1b51eSKate Stone )"
279b9c1b51eSKate Stone         "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
280ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
281ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
282ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
283ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
284ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
285ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
286ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
287ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
288ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
289b9c1b51eSKate Stone launch' example below)."
290b9c1b51eSKate Stone         R"(
291ea671fbdSKate Stone 
292b9c1b51eSKate Stone )"
293b9c1b51eSKate Stone         "Note: the positional arguments must substitute as whole words in the resultant \
294ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
295b9c1b51eSKate Stone \".cpp\":"
296b9c1b51eSKate Stone         R"(
297ea671fbdSKate Stone 
298ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
299ea671fbdSKate Stone 
300b9c1b51eSKate Stone )"
301b9c1b51eSKate Stone         "For more complex aliasing, use the \"command regex\" command instead.  In the \
302ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
303ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
304b9c1b51eSKate Stone argument.  The user would use this alias as follows:"
305b9c1b51eSKate Stone         R"(
306ea671fbdSKate Stone 
307ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
308ea671fbdSKate Stone (lldb) bfl my-file.c 137
309ea671fbdSKate Stone 
310ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
311ea671fbdSKate Stone 
312ea671fbdSKate Stone Another example:
313ea671fbdSKate Stone 
314ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
315ea671fbdSKate Stone (lldb) pltty /dev/tty0
316ea671fbdSKate Stone 
317ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
318ea671fbdSKate Stone 
319b9c1b51eSKate Stone )"
320b9c1b51eSKate Stone         "If the user always wanted to pass the same value to a particular option, the \
321ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
322b9c1b51eSKate Stone rather than using a positional placeholder:"
323b9c1b51eSKate Stone         R"(
324ea671fbdSKate Stone 
325ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
326ea671fbdSKate Stone 
327b9c1b51eSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)");
328ebc09c36SJim Ingham 
329405fe67fSCaroline Tice     CommandArgumentEntry arg1;
330405fe67fSCaroline Tice     CommandArgumentEntry arg2;
331405fe67fSCaroline Tice     CommandArgumentEntry arg3;
332405fe67fSCaroline Tice     CommandArgumentData alias_arg;
333405fe67fSCaroline Tice     CommandArgumentData cmd_arg;
334405fe67fSCaroline Tice     CommandArgumentData options_arg;
335405fe67fSCaroline Tice 
336405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
337405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
338405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
339405fe67fSCaroline Tice 
340b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
341b9c1b51eSKate Stone     // argument entry.
342405fe67fSCaroline Tice     arg1.push_back(alias_arg);
343405fe67fSCaroline Tice 
344405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
345405fe67fSCaroline Tice     cmd_arg.arg_type = eArgTypeCommandName;
346405fe67fSCaroline Tice     cmd_arg.arg_repetition = eArgRepeatPlain;
347405fe67fSCaroline Tice 
348b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
349b9c1b51eSKate Stone     // argument entry.
350405fe67fSCaroline Tice     arg2.push_back(cmd_arg);
351405fe67fSCaroline Tice 
352405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
353405fe67fSCaroline Tice     options_arg.arg_type = eArgTypeAliasOptions;
354405fe67fSCaroline Tice     options_arg.arg_repetition = eArgRepeatOptional;
355405fe67fSCaroline Tice 
356b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
357b9c1b51eSKate Stone     // argument entry.
358405fe67fSCaroline Tice     arg3.push_back(options_arg);
359405fe67fSCaroline Tice 
360405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
361405fe67fSCaroline Tice     m_arguments.push_back(arg1);
362405fe67fSCaroline Tice     m_arguments.push_back(arg2);
363405fe67fSCaroline Tice     m_arguments.push_back(arg3);
364ebc09c36SJim Ingham   }
365ebc09c36SJim Ingham 
3666e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAlias() override = default;
367ebc09c36SJim Ingham 
3685a988416SJim Ingham protected:
3694d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
370b9c1b51eSKate Stone                  CommandReturnObject &result) override {
3714d51a902SRaphael Isemann     if (raw_command_line.empty()) {
372d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
37345d0e238SEnrico Granata       return false;
37445d0e238SEnrico Granata     }
37545d0e238SEnrico Granata 
376e1cfbc79STodd Fiala     ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext();
377e1cfbc79STodd Fiala     m_option_group.NotifyOptionParsingStarting(&exe_ctx);
37845d0e238SEnrico Granata 
3793a0e1270SRaphael Isemann     OptionsWithRaw args_with_suffix(raw_command_line);
38045d0e238SEnrico Granata 
3813a0e1270SRaphael Isemann     if (args_with_suffix.HasArgs())
3823a0e1270SRaphael Isemann       if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result,
3833a0e1270SRaphael Isemann                                  m_option_group, exe_ctx))
38445d0e238SEnrico Granata         return false;
38545d0e238SEnrico Granata 
386daed98e5SShivam Mittal     llvm::StringRef raw_command_string = args_with_suffix.GetRawPart();
387a01bccdbSZachary Turner     Args args(raw_command_string);
388844d2303SCaroline Tice 
38911eb9c64SZachary Turner     if (args.GetArgumentCount() < 2) {
390d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
391844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
392844d2303SCaroline Tice       return false;
393844d2303SCaroline Tice     }
394844d2303SCaroline Tice 
395844d2303SCaroline Tice     // Get the alias command.
396844d2303SCaroline Tice 
3970d9a201eSRaphael Isemann     auto alias_command = args[0].ref();
3984574a890SZachary Turner     if (alias_command.startswith("-")) {
399d72e412fSEnrico Granata       result.AppendError("aliases starting with a dash are not supported");
400b9c1b51eSKate Stone       if (alias_command == "--help" || alias_command == "--long-help") {
401b9c1b51eSKate Stone         result.AppendWarning("if trying to pass options to 'command alias' add "
402b9c1b51eSKate Stone                              "a -- at the end of the options");
403d72e412fSEnrico Granata       }
404d72e412fSEnrico Granata       result.SetStatus(eReturnStatusFailed);
405d72e412fSEnrico Granata       return false;
406d72e412fSEnrico Granata     }
407844d2303SCaroline Tice 
408b9c1b51eSKate Stone     // Strip the new alias name off 'raw_command_string'  (leave it on args,
40905097246SAdrian Prantl     // which gets passed to 'Execute', which does the stripping itself.
410844d2303SCaroline Tice     size_t pos = raw_command_string.find(alias_command);
411b9c1b51eSKate Stone     if (pos == 0) {
412844d2303SCaroline Tice       raw_command_string = raw_command_string.substr(alias_command.size());
413844d2303SCaroline Tice       pos = raw_command_string.find_first_not_of(' ');
414844d2303SCaroline Tice       if ((pos != std::string::npos) && (pos > 0))
415844d2303SCaroline Tice         raw_command_string = raw_command_string.substr(pos);
416b9c1b51eSKate Stone     } else {
417844d2303SCaroline Tice       result.AppendError("Error parsing command string.  No alias created.");
418844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
419844d2303SCaroline Tice       return false;
420844d2303SCaroline Tice     }
421844d2303SCaroline Tice 
422844d2303SCaroline Tice     // Verify that the command is alias-able.
423771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
424b9c1b51eSKate Stone       result.AppendErrorWithFormat(
425b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
4264574a890SZachary Turner           args[0].c_str());
427844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
428844d2303SCaroline Tice       return false;
429844d2303SCaroline Tice     }
430844d2303SCaroline Tice 
431b9c1b51eSKate Stone     // Get CommandObject that is being aliased. The command name is read from
432a01bccdbSZachary Turner     // the front of raw_command_string. raw_command_string is returned with the
433a01bccdbSZachary Turner     // name of the command object stripped off the front.
434a01bccdbSZachary Turner     llvm::StringRef original_raw_command_string = raw_command_string;
435b9c1b51eSKate Stone     CommandObject *cmd_obj =
436b9c1b51eSKate Stone         m_interpreter.GetCommandObjectForCommand(raw_command_string);
437844d2303SCaroline Tice 
438b9c1b51eSKate Stone     if (!cmd_obj) {
439b9c1b51eSKate Stone       result.AppendErrorWithFormat("invalid command given to 'command alias'. "
440b9c1b51eSKate Stone                                    "'%s' does not begin with a valid command."
441b9c1b51eSKate Stone                                    "  No alias created.",
442a01bccdbSZachary Turner                                    original_raw_command_string.str().c_str());
443844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
444844d2303SCaroline Tice       return false;
445b9c1b51eSKate Stone     } else if (!cmd_obj->WantsRawCommandString()) {
446b9c1b51eSKate Stone       // Note that args was initialized with the original command, and has not
44705097246SAdrian Prantl       // been updated to this point. Therefore can we pass it to the version of
44805097246SAdrian Prantl       // Execute that does not need/expect raw input in the alias.
4495a988416SJim Ingham       return HandleAliasingNormalCommand(args, result);
450b9c1b51eSKate Stone     } else {
451b9c1b51eSKate Stone       return HandleAliasingRawCommand(alias_command, raw_command_string,
452b9c1b51eSKate Stone                                       *cmd_obj, result);
4535a988416SJim Ingham     }
4545a988416SJim Ingham     return result.Succeeded();
4555a988416SJim Ingham   }
4565a988416SJim Ingham 
457a01bccdbSZachary Turner   bool HandleAliasingRawCommand(llvm::StringRef alias_command,
458a01bccdbSZachary Turner                                 llvm::StringRef raw_command_string,
459b9c1b51eSKate Stone                                 CommandObject &cmd_obj,
460b9c1b51eSKate Stone                                 CommandReturnObject &result) {
461844d2303SCaroline Tice     // Verify & handle any options/arguments passed to the alias command
462844d2303SCaroline Tice 
463b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
464b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
465844d2303SCaroline Tice 
466b9c1b51eSKate Stone     if (CommandObjectSP cmd_obj_sp =
467*a9448872SJonas Devlieghere             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName())) {
468a01bccdbSZachary Turner       if (m_interpreter.AliasExists(alias_command) ||
469a01bccdbSZachary Turner           m_interpreter.UserCommandExists(alias_command)) {
470b9c1b51eSKate Stone         result.AppendWarningWithFormat(
471b9c1b51eSKate Stone             "Overwriting existing definition for '%s'.\n",
472a01bccdbSZachary Turner             alias_command.str().c_str());
473844d2303SCaroline Tice       }
474b9c1b51eSKate Stone       if (CommandAlias *alias = m_interpreter.AddAlias(
475a01bccdbSZachary Turner               alias_command, cmd_obj_sp, raw_command_string)) {
47645d0e238SEnrico Granata         if (m_command_options.m_help.OptionWasSet())
47745d0e238SEnrico Granata           alias->SetHelp(m_command_options.m_help.GetCurrentValue());
47845d0e238SEnrico Granata         if (m_command_options.m_long_help.OptionWasSet())
47945d0e238SEnrico Granata           alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
480844d2303SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
481b9c1b51eSKate Stone       } else {
482472362e6SCaroline Tice         result.AppendError("Unable to create requested alias.\n");
483472362e6SCaroline Tice         result.SetStatus(eReturnStatusFailed);
484472362e6SCaroline Tice       }
485212130acSEnrico Granata 
486b9c1b51eSKate Stone     } else {
487212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
488212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
489212130acSEnrico Granata     }
490212130acSEnrico Granata 
491844d2303SCaroline Tice     return result.Succeeded();
492844d2303SCaroline Tice   }
493ebc09c36SJim Ingham 
494b9c1b51eSKate Stone   bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) {
495867b185dSCaroline Tice     size_t argc = args.GetArgumentCount();
496ebc09c36SJim Ingham 
497b9c1b51eSKate Stone     if (argc < 2) {
498d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
499ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
500ebc09c36SJim Ingham       return false;
501ebc09c36SJim Ingham     }
502ebc09c36SJim Ingham 
5034574a890SZachary Turner     // Save these in std::strings since we're going to shift them off.
504adcd0268SBenjamin Kramer     const std::string alias_command(std::string(args[0].ref()));
505adcd0268SBenjamin Kramer     const std::string actual_command(std::string(args[1].ref()));
506ebc09c36SJim Ingham 
507ebc09c36SJim Ingham     args.Shift(); // Shift the alias command word off the argument vector.
508ebc09c36SJim Ingham     args.Shift(); // Shift the old command word off the argument vector.
509ebc09c36SJim Ingham 
510b9c1b51eSKate Stone     // Verify that the command is alias'able, and get the appropriate command
511b9c1b51eSKate Stone     // object.
512ebc09c36SJim Ingham 
513771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
514b9c1b51eSKate Stone       result.AppendErrorWithFormat(
515b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
516ebc09c36SJim Ingham           alias_command.c_str());
517ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
5184574a890SZachary Turner       return false;
5194574a890SZachary Turner     }
5204574a890SZachary Turner 
521b9c1b51eSKate Stone     CommandObjectSP command_obj_sp(
522a449698cSZachary Turner         m_interpreter.GetCommandSPExact(actual_command, true));
523ebc09c36SJim Ingham     CommandObjectSP subcommand_obj_sp;
524ebc09c36SJim Ingham     bool use_subcommand = false;
5254574a890SZachary Turner     if (!command_obj_sp) {
5264574a890SZachary Turner       result.AppendErrorWithFormat("'%s' is not an existing command.\n",
5274574a890SZachary Turner                                    actual_command.c_str());
5284574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
5294574a890SZachary Turner       return false;
5304574a890SZachary Turner     }
531ebc09c36SJim Ingham     CommandObject *cmd_obj = command_obj_sp.get();
5326e3d8e7fSEugene Zelenko     CommandObject *sub_cmd_obj = nullptr;
533b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
534b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
535ebc09c36SJim Ingham 
53611eb9c64SZachary Turner     while (cmd_obj->IsMultiwordObject() && !args.empty()) {
5370d9a201eSRaphael Isemann       auto sub_command = args[0].ref();
53811eb9c64SZachary Turner       assert(!sub_command.empty());
5394574a890SZachary Turner       subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command);
5404574a890SZachary Turner       if (!subcommand_obj_sp) {
541b9c1b51eSKate Stone         result.AppendErrorWithFormat(
542b9c1b51eSKate Stone             "'%s' is not a valid sub-command of '%s'.  "
543f415eeb4SCaroline Tice             "Unable to create alias.\n",
5444574a890SZachary Turner             args[0].c_str(), actual_command.c_str());
545ebc09c36SJim Ingham         result.SetStatus(eReturnStatusFailed);
546ebc09c36SJim Ingham         return false;
547ebc09c36SJim Ingham       }
5484574a890SZachary Turner 
5494574a890SZachary Turner       sub_cmd_obj = subcommand_obj_sp.get();
5504574a890SZachary Turner       use_subcommand = true;
5514574a890SZachary Turner       args.Shift(); // Shift the sub_command word off the argument vector.
5524574a890SZachary Turner       cmd_obj = sub_cmd_obj;
553ebc09c36SJim Ingham     }
554ebc09c36SJim Ingham 
555ebc09c36SJim Ingham     // Verify & handle any options/arguments passed to the alias command
556ebc09c36SJim Ingham 
557212130acSEnrico Granata     std::string args_string;
558212130acSEnrico Granata 
55911eb9c64SZachary Turner     if (!args.empty()) {
560b9c1b51eSKate Stone       CommandObjectSP tmp_sp =
561*a9448872SJonas Devlieghere           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName());
562ebc09c36SJim Ingham       if (use_subcommand)
563*a9448872SJonas Devlieghere         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName());
564ca90c47eSCaroline Tice 
565ca90c47eSCaroline Tice       args.GetCommandString(args_string);
566867b185dSCaroline Tice     }
567ebc09c36SJim Ingham 
568771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
569771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
570b9c1b51eSKate Stone       result.AppendWarningWithFormat(
5714574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
572ebc09c36SJim Ingham     }
573ebc09c36SJim Ingham 
574b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
5754574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
576771ef6d4SMalcolm Parsons             args_string)) {
57745d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
57845d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
57945d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
58045d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
581ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
582b9c1b51eSKate Stone     } else {
583212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
584212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
585212130acSEnrico Granata       return false;
586212130acSEnrico Granata     }
587ebc09c36SJim Ingham 
588ebc09c36SJim Ingham     return result.Succeeded();
589ebc09c36SJim Ingham   }
590ebc09c36SJim Ingham };
591ebc09c36SJim Ingham 
592ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
593ebc09c36SJim Ingham // CommandObjectCommandsUnalias
594ebc09c36SJim Ingham 
595b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
596ebc09c36SJim Ingham public:
5977428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
598b9c1b51eSKate Stone       : CommandObjectParsed(
599b9c1b51eSKate Stone             interpreter, "command unalias",
600b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
601b9c1b51eSKate Stone             nullptr) {
602405fe67fSCaroline Tice     CommandArgumentEntry arg;
603405fe67fSCaroline Tice     CommandArgumentData alias_arg;
604405fe67fSCaroline Tice 
605405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
606405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
607405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
608405fe67fSCaroline Tice 
609b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
610b9c1b51eSKate Stone     // argument entry.
611405fe67fSCaroline Tice     arg.push_back(alias_arg);
612405fe67fSCaroline Tice 
613405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
614405fe67fSCaroline Tice     m_arguments.push_back(arg);
615ebc09c36SJim Ingham   }
616ebc09c36SJim Ingham 
6176e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
618ebc09c36SJim Ingham 
61931fd64acSGongyu Deng   void
62031fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
62131fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
62231fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
62331fd64acSGongyu Deng       return;
62431fd64acSGongyu Deng 
62531fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetAliases()) {
62631fd64acSGongyu Deng       request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
62731fd64acSGongyu Deng     }
62831fd64acSGongyu Deng   }
62931fd64acSGongyu Deng 
6305a988416SJim Ingham protected:
631b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
632ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
633ebc09c36SJim Ingham     CommandObject *cmd_obj;
634ebc09c36SJim Ingham 
63511eb9c64SZachary Turner     if (args.empty()) {
63611eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
63711eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
63811eb9c64SZachary Turner       return false;
63911eb9c64SZachary Turner     }
64011eb9c64SZachary Turner 
6410d9a201eSRaphael Isemann     auto command_name = args[0].ref();
642a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
6434574a890SZachary Turner     if (!cmd_obj) {
6444574a890SZachary Turner       result.AppendErrorWithFormat(
6454574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
6464574a890SZachary Turner           "current list of commands.\n",
647867e7d17SZachary Turner           args[0].c_str());
6484574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
6494574a890SZachary Turner       return false;
6504574a890SZachary Turner     }
6514574a890SZachary Turner 
652b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
653b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
654b9c1b51eSKate Stone         result.AppendErrorWithFormat(
655b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
656b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
657867e7d17SZachary Turner             args[0].c_str());
658b9c1b51eSKate Stone       } else {
659b9c1b51eSKate Stone         result.AppendErrorWithFormat(
660b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
661867e7d17SZachary Turner             args[0].c_str());
662b547278cSGreg Clayton       }
663ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6644574a890SZachary Turner       return false;
6654574a890SZachary Turner     }
6664574a890SZachary Turner 
667b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
668a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
669b9c1b51eSKate Stone         result.AppendErrorWithFormat(
670867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
671867e7d17SZachary Turner             args[0].c_str());
672ebc09c36SJim Ingham       else
673b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
674867e7d17SZachary Turner                                      args[0].c_str());
675ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6764574a890SZachary Turner       return false;
677ebc09c36SJim Ingham     }
678ebc09c36SJim Ingham 
6794574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
680ebc09c36SJim Ingham     return result.Succeeded();
681ebc09c36SJim Ingham   }
682ebc09c36SJim Ingham };
683ebc09c36SJim Ingham 
684b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
685b547278cSGreg Clayton // CommandObjectCommandsDelete
686b547278cSGreg Clayton 
687b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
688b547278cSGreg Clayton public:
6897428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
690b9c1b51eSKate Stone       : CommandObjectParsed(
691b9c1b51eSKate Stone             interpreter, "command delete",
692b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
693b9c1b51eSKate Stone             nullptr) {
694b547278cSGreg Clayton     CommandArgumentEntry arg;
695b547278cSGreg Clayton     CommandArgumentData alias_arg;
696b547278cSGreg Clayton 
697b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
698b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
699b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
700b547278cSGreg Clayton 
701b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
702b9c1b51eSKate Stone     // argument entry.
703b547278cSGreg Clayton     arg.push_back(alias_arg);
704b547278cSGreg Clayton 
705b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
706b547278cSGreg Clayton     m_arguments.push_back(arg);
707b547278cSGreg Clayton   }
708b547278cSGreg Clayton 
7096e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
710b547278cSGreg Clayton 
71131fd64acSGongyu Deng   void
71231fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
71331fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
71431fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
71531fd64acSGongyu Deng       return;
71631fd64acSGongyu Deng 
71731fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetCommands()) {
71831fd64acSGongyu Deng       if (ent.second->IsRemovable())
71931fd64acSGongyu Deng         request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
72031fd64acSGongyu Deng     }
72131fd64acSGongyu Deng   }
72231fd64acSGongyu Deng 
723b547278cSGreg Clayton protected:
724b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
725b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
726b547278cSGreg Clayton 
72711eb9c64SZachary Turner     if (args.empty()) {
72811eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
72911eb9c64SZachary Turner                                    "defined regular expression command names",
730a449698cSZachary Turner                                    GetCommandName().str().c_str());
73111eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
732d77ea5b2SRaphael Isemann       return false;
73311eb9c64SZachary Turner     }
73411eb9c64SZachary Turner 
7350d9a201eSRaphael Isemann     auto command_name = args[0].ref();
7364574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
73746d4aa21SEnrico Granata       StreamString error_msg_stream;
738d5b44036SJonas Devlieghere       const bool generate_upropos = true;
73946d4aa21SEnrico Granata       const bool generate_type_lookup = false;
740b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
7414574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
742d5b44036SJonas Devlieghere           generate_upropos, generate_type_lookup);
743c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
744b547278cSGreg Clayton       result.SetStatus(eReturnStatusFailed);
7454574a890SZachary Turner       return false;
746b547278cSGreg Clayton     }
747b547278cSGreg Clayton 
7484574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
7494574a890SZachary Turner       result.AppendErrorWithFormat(
7504574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
751867e7d17SZachary Turner           args[0].c_str());
7524574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
7534574a890SZachary Turner       return false;
7544574a890SZachary Turner     }
7554574a890SZachary Turner 
7564574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
7574574a890SZachary Turner     return true;
758b547278cSGreg Clayton   }
759b547278cSGreg Clayton };
760b547278cSGreg Clayton 
761de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
7621f0f5b5bSZachary Turner 
76364becc11SRaphael Isemann #define LLDB_OPTIONS_regex
76464becc11SRaphael Isemann #include "CommandOptions.inc"
7651f0f5b5bSZachary Turner 
7665a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
767de164aaaSGreg Clayton 
768b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
769b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
770de164aaaSGreg Clayton public:
7717428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
772b9c1b51eSKate Stone       : CommandObjectParsed(
773a925974bSAdrian Prantl             interpreter, "command regex",
774a925974bSAdrian Prantl             "Define a custom command in terms of "
775b9c1b51eSKate Stone             "existing commands by matching "
776b9c1b51eSKate Stone             "regular expressions.",
7770e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
778b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
779b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
780b9c1b51eSKate Stone         m_options() {
781b9c1b51eSKate Stone     SetHelpLong(
782b9c1b51eSKate Stone         R"(
783b9c1b51eSKate Stone )"
784b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
785ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
786b9c1b51eSKate Stone using the regular expression substitution format of:"
787b9c1b51eSKate Stone         R"(
788ea671fbdSKate Stone 
789ea671fbdSKate Stone     s/<regex>/<subst>/
790ea671fbdSKate Stone 
791b9c1b51eSKate Stone )"
792b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
793ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
794b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
795b9c1b51eSKate Stone         R"(
796ea671fbdSKate Stone 
797b9c1b51eSKate Stone )"
798b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
799ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
800ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
801b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
802b9c1b51eSKate Stone         R"(
803ea671fbdSKate Stone 
804ea671fbdSKate Stone EXAMPLES
805ea671fbdSKate Stone 
806b9c1b51eSKate Stone )"
807b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
808ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
809b9c1b51eSKate Stone a number follows 'f':"
810b9c1b51eSKate Stone         R"(
811ea671fbdSKate Stone 
812b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
813de164aaaSGreg Clayton   }
814de164aaaSGreg Clayton 
8156e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
816de164aaaSGreg Clayton 
8175a988416SJim Ingham protected:
8180affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
8197ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
8200affb582SDave Lee     if (output_sp && interactive) {
8210affb582SDave Lee       output_sp->PutCString("Enter one or more sed substitution commands in "
822b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
823b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
82444d93782SGreg Clayton       output_sp->Flush();
82544d93782SGreg Clayton     }
82644d93782SGreg Clayton   }
82744d93782SGreg Clayton 
828b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
829b9c1b51eSKate Stone                               std::string &data) override {
83044d93782SGreg Clayton     io_handler.SetIsDone(true);
831d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
83244d93782SGreg Clayton       StringList lines;
833b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
83444d93782SGreg Clayton         bool check_only = false;
8354c78b788SRaphael Isemann         for (const std::string &line : lines) {
8364c78b788SRaphael Isemann           Status error = AppendRegexSubstitution(line, check_only);
837b9c1b51eSKate Stone           if (error.Fail()) {
83857179860SJonas Devlieghere             if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) {
83957179860SJonas Devlieghere               StreamSP out_stream = GetDebugger().GetAsyncOutputStream();
84044d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
84144d93782SGreg Clayton             }
84244d93782SGreg Clayton           }
84344d93782SGreg Clayton         }
84444d93782SGreg Clayton       }
845d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
846d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
84744d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
84844d93782SGreg Clayton       }
84944d93782SGreg Clayton     }
85044d93782SGreg Clayton   }
85144d93782SGreg Clayton 
852b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
8535a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
854b9c1b51eSKate Stone     if (argc == 0) {
855b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
856b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
8570e5e5a79SGreg Clayton       result.SetStatus(eReturnStatusFailed);
85811eb9c64SZachary Turner       return false;
85911eb9c64SZachary Turner     }
86011eb9c64SZachary Turner 
86197206d57SZachary Turner     Status error;
8620d9a201eSRaphael Isemann     auto name = command[0].ref();
863a8f3ae7cSJonas Devlieghere     m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>(
8644574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
8654574a890SZachary Turner         true);
8660e5e5a79SGreg Clayton 
867b9c1b51eSKate Stone     if (argc == 1) {
86857179860SJonas Devlieghere       Debugger &debugger = GetDebugger();
869e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
87044d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
871b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
872b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
87373d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
874514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
875514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
876b9c1b51eSKate Stone           multiple_lines, color_prompt,
877f6913cd7SGreg Clayton           0, // Don't show line numbers
878d77c2e09SJonas Devlieghere           *this, nullptr));
87944d93782SGreg Clayton 
880b9c1b51eSKate Stone       if (io_handler_sp) {
8817ce2de2cSJonas Devlieghere         debugger.RunIOHandlerAsync(io_handler_sp);
882de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
883de164aaaSGreg Clayton       }
884b9c1b51eSKate Stone     } else {
88597d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
88644d93782SGreg Clayton         bool check_only = false;
8870d9a201eSRaphael Isemann         error = AppendRegexSubstitution(entry.ref(), check_only);
8880e5e5a79SGreg Clayton         if (error.Fail())
8890e5e5a79SGreg Clayton           break;
8900e5e5a79SGreg Clayton       }
8910e5e5a79SGreg Clayton 
892b9c1b51eSKate Stone       if (error.Success()) {
8930e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
8940e5e5a79SGreg Clayton       }
8950e5e5a79SGreg Clayton     }
896b9c1b51eSKate Stone     if (error.Fail()) {
8970e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
898de164aaaSGreg Clayton       result.SetStatus(eReturnStatusFailed);
899de164aaaSGreg Clayton     }
9000e5e5a79SGreg Clayton 
901de164aaaSGreg Clayton     return result.Succeeded();
902de164aaaSGreg Clayton   }
903de164aaaSGreg Clayton 
90497206d57SZachary Turner   Status AppendRegexSubstitution(const llvm::StringRef &regex_sed,
905b9c1b51eSKate Stone                                  bool check_only) {
90697206d57SZachary Turner     Status error;
9070e5e5a79SGreg Clayton 
908d5b44036SJonas Devlieghere     if (!m_regex_cmd_up) {
909b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
910b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
911b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9120e5e5a79SGreg Clayton       return error;
913de164aaaSGreg Clayton     }
9140e5e5a79SGreg Clayton 
9150e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
9160e5e5a79SGreg Clayton 
917b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
918b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
919b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
920b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9210e5e5a79SGreg Clayton       return error;
9220e5e5a79SGreg Clayton     }
9230e5e5a79SGreg Clayton 
924b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
925b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
926b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
927b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
9280e5e5a79SGreg Clayton       return error;
9290e5e5a79SGreg Clayton     }
9300e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
93105097246SAdrian Prantl     // use the char that follows 's' as the regex separator character so we can
93205097246SAdrian Prantl     // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9330e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
934b9c1b51eSKate Stone     const size_t second_separator_char_pos =
935b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
9360e5e5a79SGreg Clayton 
937b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
938b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
939b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
9400e5e5a79SGreg Clayton           separator_char,
9410e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
942ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
943b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9440e5e5a79SGreg Clayton       return error;
9450e5e5a79SGreg Clayton     }
9460e5e5a79SGreg Clayton 
947b9c1b51eSKate Stone     const size_t third_separator_char_pos =
948b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
9490e5e5a79SGreg Clayton 
950b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
951b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
952b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
9530e5e5a79SGreg Clayton           separator_char,
9540e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
955ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
956b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9570e5e5a79SGreg Clayton       return error;
9580e5e5a79SGreg Clayton     }
9590e5e5a79SGreg Clayton 
960b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
96105097246SAdrian Prantl       // Make sure that everything that follows the last regex separator char
962b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
963b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
964b9c1b51eSKate Stone           std::string::npos) {
965b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
966b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
967b9c1b51eSKate Stone             "string: '%.*s'",
968b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
9690e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
9700e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
9710e5e5a79SGreg Clayton         return error;
9720e5e5a79SGreg Clayton       }
973b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
974b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
975b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
976b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9770e5e5a79SGreg Clayton           regex_sed.data());
9780e5e5a79SGreg Clayton       return error;
979b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
980b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
981b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
982b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9830e5e5a79SGreg Clayton           regex_sed.data());
9840e5e5a79SGreg Clayton       return error;
9850e5e5a79SGreg Clayton     }
98644d93782SGreg Clayton 
987b9c1b51eSKate Stone     if (!check_only) {
988adcd0268SBenjamin Kramer       std::string regex(std::string(regex_sed.substr(
989adcd0268SBenjamin Kramer           first_separator_char_pos + 1,
990adcd0268SBenjamin Kramer           second_separator_char_pos - first_separator_char_pos - 1)));
991adcd0268SBenjamin Kramer       std::string subst(std::string(regex_sed.substr(
992adcd0268SBenjamin Kramer           second_separator_char_pos + 1,
993adcd0268SBenjamin Kramer           third_separator_char_pos - second_separator_char_pos - 1)));
99443224195SRaphael Isemann       m_regex_cmd_up->AddRegexCommand(regex, subst);
99544d93782SGreg Clayton     }
9960e5e5a79SGreg Clayton     return error;
997de164aaaSGreg Clayton   }
998de164aaaSGreg Clayton 
999b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
1000d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
1001d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
1002d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
1003de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1004de164aaaSGreg Clayton       }
1005de164aaaSGreg Clayton     }
1006de164aaaSGreg Clayton   }
1007de164aaaSGreg Clayton 
1008de164aaaSGreg Clayton private:
1009d5b44036SJonas Devlieghere   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up;
1010de164aaaSGreg Clayton 
1011b9c1b51eSKate Stone   class CommandOptions : public Options {
1012de164aaaSGreg Clayton   public:
1013b9c1b51eSKate Stone     CommandOptions() : Options() {}
1014de164aaaSGreg Clayton 
10156e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1016de164aaaSGreg Clayton 
101797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1018b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
101997206d57SZachary Turner       Status error;
10203bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1021de164aaaSGreg Clayton 
1022b9c1b51eSKate Stone       switch (short_option) {
1023de164aaaSGreg Clayton       case 'h':
1024adcd0268SBenjamin Kramer         m_help.assign(std::string(option_arg));
1025de164aaaSGreg Clayton         break;
1026de164aaaSGreg Clayton       case 's':
1027adcd0268SBenjamin Kramer         m_syntax.assign(std::string(option_arg));
1028de164aaaSGreg Clayton         break;
1029de164aaaSGreg Clayton       default:
103036162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1031de164aaaSGreg Clayton       }
1032de164aaaSGreg Clayton 
1033de164aaaSGreg Clayton       return error;
1034de164aaaSGreg Clayton     }
1035de164aaaSGreg Clayton 
1036b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1037de164aaaSGreg Clayton       m_help.clear();
1038de164aaaSGreg Clayton       m_syntax.clear();
1039de164aaaSGreg Clayton     }
1040de164aaaSGreg Clayton 
10411f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
104270602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
10431f0f5b5bSZachary Turner     }
1044de164aaaSGreg Clayton 
1045daed98e5SShivam Mittal     llvm::StringRef GetHelp() { return m_help; }
10466e3d8e7fSEugene Zelenko 
1047daed98e5SShivam Mittal     llvm::StringRef GetSyntax() { return m_syntax; }
10486e3d8e7fSEugene Zelenko 
1049de164aaaSGreg Clayton   protected:
10506e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
10516e3d8e7fSEugene Zelenko 
1052de164aaaSGreg Clayton     std::string m_help;
1053de164aaaSGreg Clayton     std::string m_syntax;
1054de164aaaSGreg Clayton   };
1055de164aaaSGreg Clayton 
1056b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1057de164aaaSGreg Clayton 
10585a988416SJim Ingham   CommandOptions m_options;
1059de164aaaSGreg Clayton };
1060de164aaaSGreg Clayton 
1061b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1062223383edSEnrico Granata public:
1063b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1064b9c1b51eSKate Stone                               std::string funct, std::string help,
1065b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1066a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_function_name(funct),
1067a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_long(false) {
1068735152e3SEnrico Granata     if (!help.empty())
1069442f6530SZachary Turner       SetHelp(help);
1070b9c1b51eSKate Stone     else {
1071735152e3SEnrico Granata       StreamString stream;
1072735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1073c156427dSZachary Turner       SetHelp(stream.GetString());
1074735152e3SEnrico Granata     }
1075223383edSEnrico Granata   }
1076223383edSEnrico Granata 
10776e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1078223383edSEnrico Granata 
1079b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
10805a988416SJim Ingham 
1081b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
10825a988416SJim Ingham 
1083b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
10845a988416SJim Ingham 
1085442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1086442f6530SZachary Turner     if (m_fetched_help_long)
1087442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1088442f6530SZachary Turner 
10892b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1090442f6530SZachary Turner     if (!scripter)
1091442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1092442f6530SZachary Turner 
1093fac939e9SEnrico Granata     std::string docstring;
1094442f6530SZachary Turner     m_fetched_help_long =
1095442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1096fac939e9SEnrico Granata     if (!docstring.empty())
1097442f6530SZachary Turner       SetHelpLong(docstring);
1098fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1099fac939e9SEnrico Granata   }
1100fac939e9SEnrico Granata 
11015a988416SJim Ingham protected:
11024d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1103b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11042b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1105223383edSEnrico Granata 
110697206d57SZachary Turner     Status error;
1107223383edSEnrico Granata 
110870f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
110970f11f88SJim Ingham 
1110a925974bSAdrian Prantl     if (!scripter || !scripter->RunScriptBasedCommand(
1111a925974bSAdrian Prantl                          m_function_name.c_str(), raw_command_line, m_synchro,
1112a925974bSAdrian Prantl                          result, error, m_exe_ctx)) {
1113223383edSEnrico Granata       result.AppendError(error.AsCString());
1114223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1115b9c1b51eSKate Stone     } else {
111670f11f88SJim Ingham       // Don't change the status if the command already set it...
1117b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1118c156427dSZachary Turner         if (result.GetOutputData().empty())
1119223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
112070f11f88SJim Ingham         else
112170f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
112270f11f88SJim Ingham       }
112370f11f88SJim Ingham     }
1124223383edSEnrico Granata 
1125223383edSEnrico Granata     return result.Succeeded();
1126223383edSEnrico Granata   }
1127223383edSEnrico Granata 
11286e3d8e7fSEugene Zelenko private:
11296e3d8e7fSEugene Zelenko   std::string m_function_name;
11306e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11316e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1132223383edSEnrico Granata };
1133223383edSEnrico Granata 
1134b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
11359fe00e52SEnrico Granata public:
11369fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
11379fe00e52SEnrico Granata                                std::string name,
11380641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1139b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1140a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp),
1141a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_short(false),
1142b9c1b51eSKate Stone         m_fetched_help_long(false) {
11439fe00e52SEnrico Granata     StreamString stream;
11449fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1145c156427dSZachary Turner     SetHelp(stream.GetString());
11462b29b432SJonas Devlieghere     if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter())
1147e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
11489fe00e52SEnrico Granata   }
11499fe00e52SEnrico Granata 
11506e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
11519fe00e52SEnrico Granata 
1152b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
11539fe00e52SEnrico Granata 
1154b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
11559fe00e52SEnrico Granata 
1156442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1157442f6530SZachary Turner     if (m_fetched_help_short)
1158442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11592b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1160442f6530SZachary Turner     if (!scripter)
1161442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11626f79bb2dSEnrico Granata     std::string docstring;
1163b9c1b51eSKate Stone     m_fetched_help_short =
1164b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
11656f79bb2dSEnrico Granata     if (!docstring.empty())
1166442f6530SZachary Turner       SetHelp(docstring);
1167442f6530SZachary Turner 
11686f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
11696f79bb2dSEnrico Granata   }
11706f79bb2dSEnrico Granata 
1171442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1172442f6530SZachary Turner     if (m_fetched_help_long)
1173442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1174442f6530SZachary Turner 
11752b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1176442f6530SZachary Turner     if (!scripter)
1177442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1178442f6530SZachary Turner 
11796f79bb2dSEnrico Granata     std::string docstring;
1180b9c1b51eSKate Stone     m_fetched_help_long =
1181b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
11826f79bb2dSEnrico Granata     if (!docstring.empty())
1183442f6530SZachary Turner       SetHelpLong(docstring);
11849fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
11859fe00e52SEnrico Granata   }
11869fe00e52SEnrico Granata 
11879fe00e52SEnrico Granata protected:
11884d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1189b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11902b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
11919fe00e52SEnrico Granata 
119297206d57SZachary Turner     Status error;
11939fe00e52SEnrico Granata 
11949fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
11959fe00e52SEnrico Granata 
1196b9c1b51eSKate Stone     if (!scripter ||
1197b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1198b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
11999fe00e52SEnrico Granata       result.AppendError(error.AsCString());
12009fe00e52SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1201b9c1b51eSKate Stone     } else {
12029fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1203b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1204c156427dSZachary Turner         if (result.GetOutputData().empty())
12059fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
12069fe00e52SEnrico Granata         else
12079fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
12089fe00e52SEnrico Granata       }
12099fe00e52SEnrico Granata     }
12109fe00e52SEnrico Granata 
12119fe00e52SEnrico Granata     return result.Succeeded();
12129fe00e52SEnrico Granata   }
12139fe00e52SEnrico Granata 
12146e3d8e7fSEugene Zelenko private:
12156e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
12166e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
12176e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
12186e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
12199fe00e52SEnrico Granata };
12209fe00e52SEnrico Granata 
1221a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
122264becc11SRaphael Isemann #define LLDB_OPTIONS_script_import
122364becc11SRaphael Isemann #include "CommandOptions.inc"
12241f0f5b5bSZachary Turner 
1225b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
12265a988416SJim Ingham public:
1227b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1228b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1229b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1230b9c1b51eSKate Stone         m_options() {
12315a988416SJim Ingham     CommandArgumentEntry arg1;
12325a988416SJim Ingham     CommandArgumentData cmd_arg;
12335a988416SJim Ingham 
12345a988416SJim Ingham     // Define the first (and only) variant of this arg.
12355a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
12363b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
12375a988416SJim Ingham 
1238b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1239b9c1b51eSKate Stone     // argument entry.
12405a988416SJim Ingham     arg1.push_back(cmd_arg);
12415a988416SJim Ingham 
12425a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
12435a988416SJim Ingham     m_arguments.push_back(arg1);
12445a988416SJim Ingham   }
12455a988416SJim Ingham 
12466e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
12475a988416SJim Ingham 
1248ae34ed2cSRaphael Isemann   void
1249ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
12502443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
1251b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1252b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1253a2e76c0bSRaphael Isemann         request, nullptr);
12545a988416SJim Ingham   }
12555a988416SJim Ingham 
1256b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
12575a988416SJim Ingham 
12585a988416SJim Ingham protected:
1259b9c1b51eSKate Stone   class CommandOptions : public Options {
12600a305db7SEnrico Granata   public:
1261b9c1b51eSKate Stone     CommandOptions() : Options() {}
12620a305db7SEnrico Granata 
12636e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
12640a305db7SEnrico Granata 
126597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1266b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
126797206d57SZachary Turner       Status error;
12683bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
12690a305db7SEnrico Granata 
1270b9c1b51eSKate Stone       switch (short_option) {
12710a305db7SEnrico Granata       case 'r':
127215625112SJonas Devlieghere         // NO-OP
12730a305db7SEnrico Granata         break;
127400bb397bSJonas Devlieghere       case 'c':
127500bb397bSJonas Devlieghere         relative_to_command_file = true;
127600bb397bSJonas Devlieghere         break;
12770a305db7SEnrico Granata       default:
127836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
12790a305db7SEnrico Granata       }
12800a305db7SEnrico Granata 
12810a305db7SEnrico Granata       return error;
12820a305db7SEnrico Granata     }
12830a305db7SEnrico Granata 
1284b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
128500bb397bSJonas Devlieghere       relative_to_command_file = false;
12860a305db7SEnrico Granata     }
12870a305db7SEnrico Granata 
12881f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
128970602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
12901f0f5b5bSZachary Turner     }
129100bb397bSJonas Devlieghere     bool relative_to_command_file = false;
12920a305db7SEnrico Granata   };
12930a305db7SEnrico Granata 
1294b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
129511eb9c64SZachary Turner     if (command.empty()) {
12963b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1297a9dbf432SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1298a9dbf432SEnrico Granata       return false;
1299a9dbf432SEnrico Granata     }
1300a9dbf432SEnrico Granata 
130100bb397bSJonas Devlieghere     FileSpec source_dir = {};
130200bb397bSJonas Devlieghere     if (m_options.relative_to_command_file) {
130300bb397bSJonas Devlieghere       source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir();
130400bb397bSJonas Devlieghere       if (!source_dir) {
130500bb397bSJonas Devlieghere         result.AppendError("command script import -c can only be specified "
130600bb397bSJonas Devlieghere                            "from a command file");
130700bb397bSJonas Devlieghere         result.SetStatus(eReturnStatusFailed);
130800bb397bSJonas Devlieghere         return false;
130900bb397bSJonas Devlieghere       }
131000bb397bSJonas Devlieghere     }
131100bb397bSJonas Devlieghere 
131211eb9c64SZachary Turner     for (auto &entry : command.entries()) {
131397206d57SZachary Turner       Status error;
1314a9dbf432SEnrico Granata 
1315c9d645d3SGreg Clayton       const bool init_session = true;
1316b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
131711eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
131811eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
131905097246SAdrian Prantl       // script imports" in __lldb_init_module the real fix is to have
132005097246SAdrian Prantl       // recursive commands possible with a CommandInvocation object separate
132105097246SAdrian Prantl       // from the CommandObject itself, so that recursive command invocations
132205097246SAdrian Prantl       // won't stomp on each other (wrt to execution contents, options, and
132305097246SAdrian Prantl       // more)
1324078551c7SEnrico Granata       m_exe_ctx.Clear();
13252b29b432SJonas Devlieghere       if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule(
132600bb397bSJonas Devlieghere               entry.c_str(), init_session, error, nullptr, source_dir)) {
1327a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1328b9c1b51eSKate Stone       } else {
1329b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1330b9c1b51eSKate Stone                                      error.AsCString());
1331a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusFailed);
1332a9dbf432SEnrico Granata       }
13333b00e35bSEnrico Granata     }
1334a9dbf432SEnrico Granata 
1335a9dbf432SEnrico Granata     return result.Succeeded();
1336a9dbf432SEnrico Granata   }
13370a305db7SEnrico Granata 
13385a988416SJim Ingham   CommandOptions m_options;
1339a9dbf432SEnrico Granata };
1340223383edSEnrico Granata 
1341223383edSEnrico Granata // CommandObjectCommandsScriptAdd
13428fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = {
1343e063ecccSJonas Devlieghere     {
1344e063ecccSJonas Devlieghere         eScriptedCommandSynchronicitySynchronous,
1345e063ecccSJonas Devlieghere         "synchronous",
1346e063ecccSJonas Devlieghere         "Run synchronous",
1347e063ecccSJonas Devlieghere     },
1348e063ecccSJonas Devlieghere     {
1349e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityAsynchronous,
1350e063ecccSJonas Devlieghere         "asynchronous",
1351e063ecccSJonas Devlieghere         "Run asynchronous",
1352e063ecccSJonas Devlieghere     },
1353e063ecccSJonas Devlieghere     {
1354e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityCurrentValue,
1355e063ecccSJonas Devlieghere         "current",
1356e063ecccSJonas Devlieghere         "Do not alter current setting",
1357e063ecccSJonas Devlieghere     },
1358e063ecccSJonas Devlieghere };
13591f0f5b5bSZachary Turner 
13608fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() {
13618fe53c49STatyana Krasnukha   return OptionEnumValues(g_script_synchro_type);
13628fe53c49STatyana Krasnukha }
13638fe53c49STatyana Krasnukha 
136464becc11SRaphael Isemann #define LLDB_OPTIONS_script_add
136564becc11SRaphael Isemann #include "CommandOptions.inc"
13661f0f5b5bSZachary Turner 
1367b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1368b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
13695a988416SJim Ingham public:
1370b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1371b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
13725a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
13736e3d8e7fSEugene Zelenko                             nullptr),
1374b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
13755a988416SJim Ingham     CommandArgumentEntry arg1;
13765a988416SJim Ingham     CommandArgumentData cmd_arg;
13775a988416SJim Ingham 
13785a988416SJim Ingham     // Define the first (and only) variant of this arg.
13795a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
13805a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
13815a988416SJim Ingham 
1382b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1383b9c1b51eSKate Stone     // argument entry.
13845a988416SJim Ingham     arg1.push_back(cmd_arg);
13855a988416SJim Ingham 
13865a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
13875a988416SJim Ingham     m_arguments.push_back(arg1);
13885a988416SJim Ingham   }
13895a988416SJim Ingham 
13906e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
13915a988416SJim Ingham 
1392b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13935a988416SJim Ingham 
13945a988416SJim Ingham protected:
1395b9c1b51eSKate Stone   class CommandOptions : public Options {
1396223383edSEnrico Granata   public:
1397b9c1b51eSKate Stone     CommandOptions()
1398b9c1b51eSKate Stone         : Options(), m_class_name(), m_funct_name(), m_short_help(),
1399b9c1b51eSKate Stone           m_synchronicity(eScriptedCommandSynchronicitySynchronous) {}
1400223383edSEnrico Granata 
14016e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1402223383edSEnrico Granata 
140397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1404b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
140597206d57SZachary Turner       Status error;
14063bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1407223383edSEnrico Granata 
1408b9c1b51eSKate Stone       switch (short_option) {
1409223383edSEnrico Granata       case 'f':
1410fe11483bSZachary Turner         if (!option_arg.empty())
1411adcd0268SBenjamin Kramer           m_funct_name = std::string(option_arg);
1412735152e3SEnrico Granata         break;
14139fe00e52SEnrico Granata       case 'c':
1414fe11483bSZachary Turner         if (!option_arg.empty())
1415adcd0268SBenjamin Kramer           m_class_name = std::string(option_arg);
14169fe00e52SEnrico Granata         break;
1417735152e3SEnrico Granata       case 'h':
1418fe11483bSZachary Turner         if (!option_arg.empty())
1419adcd0268SBenjamin Kramer           m_short_help = std::string(option_arg);
1420223383edSEnrico Granata         break;
14210a305db7SEnrico Granata       case 's':
1422b9c1b51eSKate Stone         m_synchronicity =
142347cbf4a0SPavel Labath             (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum(
1424fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
14250a305db7SEnrico Granata         if (!error.Success())
1426b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1427fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1428fe11483bSZachary Turner               option_arg.str().c_str());
14290a305db7SEnrico Granata         break;
1430223383edSEnrico Granata       default:
143136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1432223383edSEnrico Granata       }
1433223383edSEnrico Granata 
1434223383edSEnrico Granata       return error;
1435223383edSEnrico Granata     }
1436223383edSEnrico Granata 
1437b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
14389fe00e52SEnrico Granata       m_class_name.clear();
1439735152e3SEnrico Granata       m_funct_name.clear();
1440735152e3SEnrico Granata       m_short_help.clear();
144144d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1442223383edSEnrico Granata     }
1443223383edSEnrico Granata 
14441f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
144570602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
14461f0f5b5bSZachary Turner     }
1447223383edSEnrico Granata 
1448223383edSEnrico Granata     // Instance variables to hold the values for command options.
1449223383edSEnrico Granata 
14509fe00e52SEnrico Granata     std::string m_class_name;
1451223383edSEnrico Granata     std::string m_funct_name;
1452735152e3SEnrico Granata     std::string m_short_help;
145344d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1454223383edSEnrico Granata   };
1455223383edSEnrico Granata 
14560affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
14577ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
14580affb582SDave Lee     if (output_sp && interactive) {
145944d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
146044d93782SGreg Clayton       output_sp->Flush();
1461223383edSEnrico Granata     }
1462223383edSEnrico Granata   }
1463223383edSEnrico Granata 
1464b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1465b9c1b51eSKate Stone                               std::string &data) override {
14667ca15ba7SLawrence D'Anna     StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
146744d93782SGreg Clayton 
14682b29b432SJonas Devlieghere     ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1469b9c1b51eSKate Stone     if (interpreter) {
147044d93782SGreg Clayton 
147144d93782SGreg Clayton       StringList lines;
147244d93782SGreg Clayton       lines.SplitIntoLines(data);
1473b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1474a73b7df7SEnrico Granata         std::string funct_name_str;
1475b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1476b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1477b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1478b9c1b51eSKate Stone                              "add python command.\n");
147944d93782SGreg Clayton             error_sp->Flush();
1480b9c1b51eSKate Stone           } else {
1481223383edSEnrico Granata             // everything should be fine now, let's add this alias
1482223383edSEnrico Granata 
1483b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1484771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
148544d93782SGreg Clayton                 m_synchronicity));
1486223383edSEnrico Granata 
1487b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1488b9c1b51eSKate Stone                                               true)) {
1489b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1490b9c1b51eSKate Stone                                "add python command.\n");
149144d93782SGreg Clayton               error_sp->Flush();
1492223383edSEnrico Granata             }
1493223383edSEnrico Granata           }
1494b9c1b51eSKate Stone         } else {
1495b9c1b51eSKate Stone           error_sp->Printf(
1496b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
149744d93782SGreg Clayton           error_sp->Flush();
149844d93782SGreg Clayton         }
1499b9c1b51eSKate Stone       } else {
150044d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
150144d93782SGreg Clayton         error_sp->Flush();
150244d93782SGreg Clayton       }
1503b9c1b51eSKate Stone     } else {
1504b9c1b51eSKate Stone       error_sp->Printf(
1505b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
150644d93782SGreg Clayton       error_sp->Flush();
150744d93782SGreg Clayton     }
150844d93782SGreg Clayton 
150944d93782SGreg Clayton     io_handler.SetIsDone(true);
151044d93782SGreg Clayton   }
1511223383edSEnrico Granata 
1512b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
151357179860SJonas Devlieghere     if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) {
1514b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1515b9c1b51eSKate Stone                          "commands is currently Python");
151699f0b8f9SEnrico Granata       result.SetStatus(eReturnStatusFailed);
151799f0b8f9SEnrico Granata       return false;
151899f0b8f9SEnrico Granata     }
151999f0b8f9SEnrico Granata 
152011eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1521223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1522223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1523223383edSEnrico Granata       return false;
1524223383edSEnrico Granata     }
1525223383edSEnrico Granata 
1526735152e3SEnrico Granata     // Store the options in case we get multi-line input
1527adcd0268SBenjamin Kramer     m_cmd_name = std::string(command[0].ref());
1528735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
152944d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1530223383edSEnrico Granata 
1531b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1532b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1533b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1534b9c1b51eSKate Stone             "     ", // Prompt
1535a6faf851SJonas Devlieghere             *this);  // IOHandlerDelegate
1536b9c1b51eSKate Stone       } else {
1537b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1538b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1539b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1540b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1541223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1542b9c1b51eSKate Stone         } else {
1543223383edSEnrico Granata           result.AppendError("cannot add command");
1544223383edSEnrico Granata           result.SetStatus(eReturnStatusFailed);
1545223383edSEnrico Granata         }
1546223383edSEnrico Granata       }
1547b9c1b51eSKate Stone     } else {
15482b29b432SJonas Devlieghere       ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1549b9c1b51eSKate Stone       if (!interpreter) {
15509fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
15519fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15529fe00e52SEnrico Granata         return false;
15539fe00e52SEnrico Granata       }
15549fe00e52SEnrico Granata 
1555b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1556b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1557b9c1b51eSKate Stone       if (!cmd_obj_sp) {
15589fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
15599fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15609fe00e52SEnrico Granata         return false;
15619fe00e52SEnrico Granata       }
15629fe00e52SEnrico Granata 
1563b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1564b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1565b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
15669fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1567b9c1b51eSKate Stone       } else {
15689fe00e52SEnrico Granata         result.AppendError("cannot add command");
15699fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15709fe00e52SEnrico Granata       }
15719fe00e52SEnrico Granata     }
1572223383edSEnrico Granata 
1573223383edSEnrico Granata     return result.Succeeded();
1574223383edSEnrico Granata   }
15755a988416SJim Ingham 
15765a988416SJim Ingham   CommandOptions m_options;
157744d93782SGreg Clayton   std::string m_cmd_name;
1578735152e3SEnrico Granata   std::string m_short_help;
157944d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1580223383edSEnrico Granata };
1581223383edSEnrico Granata 
1582223383edSEnrico Granata // CommandObjectCommandsScriptList
1583223383edSEnrico Granata 
1584b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1585223383edSEnrico Granata public:
1586b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1587b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1588b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1589223383edSEnrico Granata 
15906e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1591223383edSEnrico Granata 
1592b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1593d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1594d77ea5b2SRaphael Isemann       result.AppendError("'command script list' doesn't take any arguments");
1595d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1596d77ea5b2SRaphael Isemann       return false;
1597d77ea5b2SRaphael Isemann     }
1598d77ea5b2SRaphael Isemann 
1599b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1600223383edSEnrico Granata 
1601223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1602223383edSEnrico Granata 
1603223383edSEnrico Granata     return true;
1604223383edSEnrico Granata   }
1605223383edSEnrico Granata };
1606223383edSEnrico Granata 
1607223383edSEnrico Granata // CommandObjectCommandsScriptClear
1608223383edSEnrico Granata 
1609b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1610223383edSEnrico Granata public:
1611b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1612b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1613b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1614223383edSEnrico Granata 
16156e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1616223383edSEnrico Granata 
16175a988416SJim Ingham protected:
1618b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1619d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1620d77ea5b2SRaphael Isemann       result.AppendError("'command script clear' doesn't take any arguments");
1621d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1622d77ea5b2SRaphael Isemann       return false;
1623d77ea5b2SRaphael Isemann     }
1624d77ea5b2SRaphael Isemann 
1625223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1626223383edSEnrico Granata 
1627223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1628223383edSEnrico Granata 
1629223383edSEnrico Granata     return true;
1630223383edSEnrico Granata   }
1631223383edSEnrico Granata };
1632223383edSEnrico Granata 
1633223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1634223383edSEnrico Granata 
1635b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1636223383edSEnrico Granata public:
1637b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1638b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1639b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1640223383edSEnrico Granata     CommandArgumentEntry arg1;
1641223383edSEnrico Granata     CommandArgumentData cmd_arg;
1642223383edSEnrico Granata 
1643223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1644223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1645223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1646223383edSEnrico Granata 
1647b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1648b9c1b51eSKate Stone     // argument entry.
1649223383edSEnrico Granata     arg1.push_back(cmd_arg);
1650223383edSEnrico Granata 
1651223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1652223383edSEnrico Granata     m_arguments.push_back(arg1);
1653223383edSEnrico Granata   }
1654223383edSEnrico Granata 
16556e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1656223383edSEnrico Granata 
16572e8f304fSGongyu Deng   void
16582e8f304fSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
16592e8f304fSGongyu Deng                            OptionElementVector &opt_element_vector) override {
16602e8f304fSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
16612e8f304fSGongyu Deng       return;
16622e8f304fSGongyu Deng 
16632ebe30c6SRaphael Isemann     for (const auto &c : m_interpreter.GetUserCommands())
16642ebe30c6SRaphael Isemann       request.TryCompleteCurrentArg(c.first, c.second->GetHelp());
16652e8f304fSGongyu Deng   }
16662e8f304fSGongyu Deng 
16675a988416SJim Ingham protected:
1668b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1669223383edSEnrico Granata 
167011eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1671223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1672223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1673223383edSEnrico Granata       return false;
1674223383edSEnrico Granata     }
1675223383edSEnrico Granata 
16760d9a201eSRaphael Isemann     auto cmd_name = command[0].ref();
1677223383edSEnrico Granata 
16784574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
16794574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1680867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
1681223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
16824574a890SZachary Turner       return false;
1683223383edSEnrico Granata     }
1684223383edSEnrico Granata 
16854574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
16864574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
16874574a890SZachary Turner     return true;
1688223383edSEnrico Granata   }
1689223383edSEnrico Granata };
1690223383edSEnrico Granata 
1691223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1692223383edSEnrico Granata 
1693223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1694223383edSEnrico Granata 
1695b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1696223383edSEnrico Granata public:
16977428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1698b9c1b51eSKate Stone       : CommandObjectMultiword(
1699a925974bSAdrian Prantl             interpreter, "command script",
1700a925974bSAdrian Prantl             "Commands for managing custom "
1701b9c1b51eSKate Stone             "commands implemented by "
1702b9c1b51eSKate Stone             "interpreter scripts.",
1703b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1704b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1705b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1706b9c1b51eSKate Stone     LoadSubCommand(
1707b9c1b51eSKate Stone         "delete",
1708b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1709b9c1b51eSKate Stone     LoadSubCommand(
1710b9c1b51eSKate Stone         "clear",
1711b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1712b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1713b9c1b51eSKate Stone                                interpreter)));
1714b9c1b51eSKate Stone     LoadSubCommand(
1715b9c1b51eSKate Stone         "import",
1716b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1717223383edSEnrico Granata   }
1718223383edSEnrico Granata 
17196e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1720223383edSEnrico Granata };
1721223383edSEnrico Granata 
1722ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1723ebc09c36SJim Ingham 
1724ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1725ebc09c36SJim Ingham 
1726b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1727b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1728b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1729b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1730b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1731b9c1b51eSKate Stone   LoadSubCommand("source",
1732b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1733b9c1b51eSKate Stone   LoadSubCommand("alias",
1734b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1735b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1736b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1737b9c1b51eSKate Stone   LoadSubCommand("delete",
1738b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1739b9c1b51eSKate Stone   LoadSubCommand(
1740b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1741b9c1b51eSKate Stone   LoadSubCommand(
1742b9c1b51eSKate Stone       "script",
1743b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1744ebc09c36SJim Ingham }
1745ebc09c36SJim Ingham 
17466e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1747