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);
1376e3d8e7fSEugene Zelenko     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()) {
143340b0309SGreg Clayton       // Use user set settings
14426c7bf93SJim Ingham       CommandInterpreterRunOptions options;
1451c19b74cSJonas Devlieghere 
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       }
16226c7bf93SJim Ingham 
1634574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
164b9c1b51eSKate Stone     } else {
16505097246SAdrian Prantl       // No options were set, inherit any settings from nested "command source"
16605097246SAdrian Prantl       // commands, or set to sane default settings...
16726c7bf93SJim Ingham       CommandInterpreterRunOptions options;
1684574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
169ebc09c36SJim Ingham     }
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 =
467b9c1b51eSKate Stone             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) {
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 =
561b9c1b51eSKate Stone           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false);
562ebc09c36SJim Ingham       if (use_subcommand)
5634574a890SZachary Turner         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(),
5644574a890SZachary Turner                                                  false);
565ca90c47eSCaroline Tice 
566ca90c47eSCaroline Tice       args.GetCommandString(args_string);
567867b185dSCaroline Tice     }
568ebc09c36SJim Ingham 
569771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
570771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
571b9c1b51eSKate Stone       result.AppendWarningWithFormat(
5724574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
573ebc09c36SJim Ingham     }
574ebc09c36SJim Ingham 
575b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
5764574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
577771ef6d4SMalcolm Parsons             args_string)) {
57845d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
57945d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
58045d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
58145d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
582ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
583b9c1b51eSKate Stone     } else {
584212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
585212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
586212130acSEnrico Granata       return false;
587212130acSEnrico Granata     }
588ebc09c36SJim Ingham 
589ebc09c36SJim Ingham     return result.Succeeded();
590ebc09c36SJim Ingham   }
591ebc09c36SJim Ingham };
592ebc09c36SJim Ingham 
593ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
594ebc09c36SJim Ingham // CommandObjectCommandsUnalias
595ebc09c36SJim Ingham 
596b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
597ebc09c36SJim Ingham public:
5987428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
599b9c1b51eSKate Stone       : CommandObjectParsed(
600b9c1b51eSKate Stone             interpreter, "command unalias",
601b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
602b9c1b51eSKate Stone             nullptr) {
603405fe67fSCaroline Tice     CommandArgumentEntry arg;
604405fe67fSCaroline Tice     CommandArgumentData alias_arg;
605405fe67fSCaroline Tice 
606405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
607405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
608405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
609405fe67fSCaroline Tice 
610b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
611b9c1b51eSKate Stone     // argument entry.
612405fe67fSCaroline Tice     arg.push_back(alias_arg);
613405fe67fSCaroline Tice 
614405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
615405fe67fSCaroline Tice     m_arguments.push_back(arg);
616ebc09c36SJim Ingham   }
617ebc09c36SJim Ingham 
6186e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
619ebc09c36SJim Ingham 
62031fd64acSGongyu Deng   void
62131fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
62231fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
62331fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
62431fd64acSGongyu Deng       return;
62531fd64acSGongyu Deng 
62631fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetAliases()) {
62731fd64acSGongyu Deng       request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
62831fd64acSGongyu Deng     }
62931fd64acSGongyu Deng   }
63031fd64acSGongyu Deng 
6315a988416SJim Ingham protected:
632b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
633ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
634ebc09c36SJim Ingham     CommandObject *cmd_obj;
635ebc09c36SJim Ingham 
63611eb9c64SZachary Turner     if (args.empty()) {
63711eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
63811eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
63911eb9c64SZachary Turner       return false;
64011eb9c64SZachary Turner     }
64111eb9c64SZachary Turner 
6420d9a201eSRaphael Isemann     auto command_name = args[0].ref();
643a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
6444574a890SZachary Turner     if (!cmd_obj) {
6454574a890SZachary Turner       result.AppendErrorWithFormat(
6464574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
6474574a890SZachary Turner           "current list of commands.\n",
648867e7d17SZachary Turner           args[0].c_str());
6494574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
6504574a890SZachary Turner       return false;
6514574a890SZachary Turner     }
6524574a890SZachary Turner 
653b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
654b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
655b9c1b51eSKate Stone         result.AppendErrorWithFormat(
656b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
657b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
658867e7d17SZachary Turner             args[0].c_str());
659b9c1b51eSKate Stone       } else {
660b9c1b51eSKate Stone         result.AppendErrorWithFormat(
661b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
662867e7d17SZachary Turner             args[0].c_str());
663b547278cSGreg Clayton       }
664ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6654574a890SZachary Turner       return false;
6664574a890SZachary Turner     }
6674574a890SZachary Turner 
668b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
669a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
670b9c1b51eSKate Stone         result.AppendErrorWithFormat(
671867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
672867e7d17SZachary Turner             args[0].c_str());
673ebc09c36SJim Ingham       else
674b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
675867e7d17SZachary Turner                                      args[0].c_str());
676ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
6774574a890SZachary Turner       return false;
678ebc09c36SJim Ingham     }
679ebc09c36SJim Ingham 
6804574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
681ebc09c36SJim Ingham     return result.Succeeded();
682ebc09c36SJim Ingham   }
683ebc09c36SJim Ingham };
684ebc09c36SJim Ingham 
685b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
686b547278cSGreg Clayton // CommandObjectCommandsDelete
687b547278cSGreg Clayton 
688b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
689b547278cSGreg Clayton public:
6907428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
691b9c1b51eSKate Stone       : CommandObjectParsed(
692b9c1b51eSKate Stone             interpreter, "command delete",
693b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
694b9c1b51eSKate Stone             nullptr) {
695b547278cSGreg Clayton     CommandArgumentEntry arg;
696b547278cSGreg Clayton     CommandArgumentData alias_arg;
697b547278cSGreg Clayton 
698b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
699b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
700b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
701b547278cSGreg Clayton 
702b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
703b9c1b51eSKate Stone     // argument entry.
704b547278cSGreg Clayton     arg.push_back(alias_arg);
705b547278cSGreg Clayton 
706b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
707b547278cSGreg Clayton     m_arguments.push_back(arg);
708b547278cSGreg Clayton   }
709b547278cSGreg Clayton 
7106e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
711b547278cSGreg Clayton 
71231fd64acSGongyu Deng   void
71331fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
71431fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
71531fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
71631fd64acSGongyu Deng       return;
71731fd64acSGongyu Deng 
71831fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetCommands()) {
71931fd64acSGongyu Deng       if (ent.second->IsRemovable())
72031fd64acSGongyu Deng         request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
72131fd64acSGongyu Deng     }
72231fd64acSGongyu Deng   }
72331fd64acSGongyu Deng 
724b547278cSGreg Clayton protected:
725b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
726b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
727b547278cSGreg Clayton 
72811eb9c64SZachary Turner     if (args.empty()) {
72911eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
73011eb9c64SZachary Turner                                    "defined regular expression command names",
731a449698cSZachary Turner                                    GetCommandName().str().c_str());
73211eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
733d77ea5b2SRaphael Isemann       return false;
73411eb9c64SZachary Turner     }
73511eb9c64SZachary Turner 
7360d9a201eSRaphael Isemann     auto command_name = args[0].ref();
7374574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
73846d4aa21SEnrico Granata       StreamString error_msg_stream;
739d5b44036SJonas Devlieghere       const bool generate_upropos = true;
74046d4aa21SEnrico Granata       const bool generate_type_lookup = false;
741b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
7424574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
743d5b44036SJonas Devlieghere           generate_upropos, generate_type_lookup);
744c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
745b547278cSGreg Clayton       result.SetStatus(eReturnStatusFailed);
7464574a890SZachary Turner       return false;
747b547278cSGreg Clayton     }
748b547278cSGreg Clayton 
7494574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
7504574a890SZachary Turner       result.AppendErrorWithFormat(
7514574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
752867e7d17SZachary Turner           args[0].c_str());
7534574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
7544574a890SZachary Turner       return false;
7554574a890SZachary Turner     }
7564574a890SZachary Turner 
7574574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
7584574a890SZachary Turner     return true;
759b547278cSGreg Clayton   }
760b547278cSGreg Clayton };
761b547278cSGreg Clayton 
762de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
7631f0f5b5bSZachary Turner 
76464becc11SRaphael Isemann #define LLDB_OPTIONS_regex
76564becc11SRaphael Isemann #include "CommandOptions.inc"
7661f0f5b5bSZachary Turner 
7675a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
768de164aaaSGreg Clayton 
769b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
770b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
771de164aaaSGreg Clayton public:
7727428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
773b9c1b51eSKate Stone       : CommandObjectParsed(
774a925974bSAdrian Prantl             interpreter, "command regex",
775a925974bSAdrian Prantl             "Define a custom command in terms of "
776b9c1b51eSKate Stone             "existing commands by matching "
777b9c1b51eSKate Stone             "regular expressions.",
7780e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
779b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
780b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
781b9c1b51eSKate Stone         m_options() {
782b9c1b51eSKate Stone     SetHelpLong(
783b9c1b51eSKate Stone         R"(
784b9c1b51eSKate Stone )"
785b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
786ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
787b9c1b51eSKate Stone using the regular expression substitution format of:"
788b9c1b51eSKate Stone         R"(
789ea671fbdSKate Stone 
790ea671fbdSKate Stone     s/<regex>/<subst>/
791ea671fbdSKate Stone 
792b9c1b51eSKate Stone )"
793b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
794ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
795b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
796b9c1b51eSKate Stone         R"(
797ea671fbdSKate Stone 
798b9c1b51eSKate Stone )"
799b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
800ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
801ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
802b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
803b9c1b51eSKate Stone         R"(
804ea671fbdSKate Stone 
805ea671fbdSKate Stone EXAMPLES
806ea671fbdSKate Stone 
807b9c1b51eSKate Stone )"
808b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
809ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
810b9c1b51eSKate Stone a number follows 'f':"
811b9c1b51eSKate Stone         R"(
812ea671fbdSKate Stone 
813b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
814de164aaaSGreg Clayton   }
815de164aaaSGreg Clayton 
8166e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
817de164aaaSGreg Clayton 
8185a988416SJim Ingham protected:
8190affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
8207ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
8210affb582SDave Lee     if (output_sp && interactive) {
8220affb582SDave Lee       output_sp->PutCString("Enter one or more sed substitution commands in "
823b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
824b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
82544d93782SGreg Clayton       output_sp->Flush();
82644d93782SGreg Clayton     }
82744d93782SGreg Clayton   }
82844d93782SGreg Clayton 
829b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
830b9c1b51eSKate Stone                               std::string &data) override {
83144d93782SGreg Clayton     io_handler.SetIsDone(true);
832d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
83344d93782SGreg Clayton       StringList lines;
834b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
83544d93782SGreg Clayton         bool check_only = false;
8364c78b788SRaphael Isemann         for (const std::string &line : lines) {
8374c78b788SRaphael Isemann           Status error = AppendRegexSubstitution(line, check_only);
838b9c1b51eSKate Stone           if (error.Fail()) {
83957179860SJonas Devlieghere             if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) {
84057179860SJonas Devlieghere               StreamSP out_stream = GetDebugger().GetAsyncOutputStream();
84144d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
84244d93782SGreg Clayton             }
84344d93782SGreg Clayton           }
84444d93782SGreg Clayton         }
84544d93782SGreg Clayton       }
846d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
847d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
84844d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
84944d93782SGreg Clayton       }
85044d93782SGreg Clayton     }
85144d93782SGreg Clayton   }
85244d93782SGreg Clayton 
853b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
8545a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
855b9c1b51eSKate Stone     if (argc == 0) {
856b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
857b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
8580e5e5a79SGreg Clayton       result.SetStatus(eReturnStatusFailed);
85911eb9c64SZachary Turner       return false;
86011eb9c64SZachary Turner     }
86111eb9c64SZachary Turner 
86297206d57SZachary Turner     Status error;
8630d9a201eSRaphael Isemann     auto name = command[0].ref();
864a8f3ae7cSJonas Devlieghere     m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>(
8654574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
8664574a890SZachary Turner         true);
8670e5e5a79SGreg Clayton 
868b9c1b51eSKate Stone     if (argc == 1) {
86957179860SJonas Devlieghere       Debugger &debugger = GetDebugger();
870e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
87144d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
872b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
873b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
87473d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
875514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
876514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
877b9c1b51eSKate Stone           multiple_lines, color_prompt,
878f6913cd7SGreg Clayton           0, // Don't show line numbers
879d77c2e09SJonas Devlieghere           *this, nullptr));
88044d93782SGreg Clayton 
881b9c1b51eSKate Stone       if (io_handler_sp) {
8827ce2de2cSJonas Devlieghere         debugger.RunIOHandlerAsync(io_handler_sp);
883de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
884de164aaaSGreg Clayton       }
885b9c1b51eSKate Stone     } else {
88697d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
88744d93782SGreg Clayton         bool check_only = false;
8880d9a201eSRaphael Isemann         error = AppendRegexSubstitution(entry.ref(), check_only);
8890e5e5a79SGreg Clayton         if (error.Fail())
8900e5e5a79SGreg Clayton           break;
8910e5e5a79SGreg Clayton       }
8920e5e5a79SGreg Clayton 
893b9c1b51eSKate Stone       if (error.Success()) {
8940e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
8950e5e5a79SGreg Clayton       }
8960e5e5a79SGreg Clayton     }
897b9c1b51eSKate Stone     if (error.Fail()) {
8980e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
899de164aaaSGreg Clayton       result.SetStatus(eReturnStatusFailed);
900de164aaaSGreg Clayton     }
9010e5e5a79SGreg Clayton 
902de164aaaSGreg Clayton     return result.Succeeded();
903de164aaaSGreg Clayton   }
904de164aaaSGreg Clayton 
90597206d57SZachary Turner   Status AppendRegexSubstitution(const llvm::StringRef &regex_sed,
906b9c1b51eSKate Stone                                  bool check_only) {
90797206d57SZachary Turner     Status error;
9080e5e5a79SGreg Clayton 
909d5b44036SJonas Devlieghere     if (!m_regex_cmd_up) {
910b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
911b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
912b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9130e5e5a79SGreg Clayton       return error;
914de164aaaSGreg Clayton     }
9150e5e5a79SGreg Clayton 
9160e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
9170e5e5a79SGreg Clayton 
918b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
919b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
920b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
921b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9220e5e5a79SGreg Clayton       return error;
9230e5e5a79SGreg Clayton     }
9240e5e5a79SGreg Clayton 
925b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
926b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
927b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
928b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
9290e5e5a79SGreg Clayton       return error;
9300e5e5a79SGreg Clayton     }
9310e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
93205097246SAdrian Prantl     // use the char that follows 's' as the regex separator character so we can
93305097246SAdrian Prantl     // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9340e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
935b9c1b51eSKate Stone     const size_t second_separator_char_pos =
936b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
9370e5e5a79SGreg Clayton 
938b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
939b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
940b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
9410e5e5a79SGreg Clayton           separator_char,
9420e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
943ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
944b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9450e5e5a79SGreg Clayton       return error;
9460e5e5a79SGreg Clayton     }
9470e5e5a79SGreg Clayton 
948b9c1b51eSKate Stone     const size_t third_separator_char_pos =
949b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
9500e5e5a79SGreg Clayton 
951b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
952b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
953b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
9540e5e5a79SGreg Clayton           separator_char,
9550e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
956ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
957b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9580e5e5a79SGreg Clayton       return error;
9590e5e5a79SGreg Clayton     }
9600e5e5a79SGreg Clayton 
961b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
96205097246SAdrian Prantl       // Make sure that everything that follows the last regex separator char
963b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
964b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
965b9c1b51eSKate Stone           std::string::npos) {
966b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
967b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
968b9c1b51eSKate Stone             "string: '%.*s'",
969b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
9700e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
9710e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
9720e5e5a79SGreg Clayton         return error;
9730e5e5a79SGreg Clayton       }
974b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
975b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
976b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
977b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9780e5e5a79SGreg Clayton           regex_sed.data());
9790e5e5a79SGreg Clayton       return error;
980b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
981b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
982b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
983b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9840e5e5a79SGreg Clayton           regex_sed.data());
9850e5e5a79SGreg Clayton       return error;
9860e5e5a79SGreg Clayton     }
98744d93782SGreg Clayton 
988b9c1b51eSKate Stone     if (!check_only) {
989adcd0268SBenjamin Kramer       std::string regex(std::string(regex_sed.substr(
990adcd0268SBenjamin Kramer           first_separator_char_pos + 1,
991adcd0268SBenjamin Kramer           second_separator_char_pos - first_separator_char_pos - 1)));
992adcd0268SBenjamin Kramer       std::string subst(std::string(regex_sed.substr(
993adcd0268SBenjamin Kramer           second_separator_char_pos + 1,
994adcd0268SBenjamin Kramer           third_separator_char_pos - second_separator_char_pos - 1)));
99543224195SRaphael Isemann       m_regex_cmd_up->AddRegexCommand(regex, subst);
99644d93782SGreg Clayton     }
9970e5e5a79SGreg Clayton     return error;
998de164aaaSGreg Clayton   }
999de164aaaSGreg Clayton 
1000b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
1001d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
1002d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
1003d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
1004de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1005de164aaaSGreg Clayton       }
1006de164aaaSGreg Clayton     }
1007de164aaaSGreg Clayton   }
1008de164aaaSGreg Clayton 
1009de164aaaSGreg Clayton private:
1010d5b44036SJonas Devlieghere   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up;
1011de164aaaSGreg Clayton 
1012b9c1b51eSKate Stone   class CommandOptions : public Options {
1013de164aaaSGreg Clayton   public:
1014b9c1b51eSKate Stone     CommandOptions() : Options() {}
1015de164aaaSGreg Clayton 
10166e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1017de164aaaSGreg Clayton 
101897206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1019b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
102097206d57SZachary Turner       Status error;
10213bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1022de164aaaSGreg Clayton 
1023b9c1b51eSKate Stone       switch (short_option) {
1024de164aaaSGreg Clayton       case 'h':
1025adcd0268SBenjamin Kramer         m_help.assign(std::string(option_arg));
1026de164aaaSGreg Clayton         break;
1027de164aaaSGreg Clayton       case 's':
1028adcd0268SBenjamin Kramer         m_syntax.assign(std::string(option_arg));
1029de164aaaSGreg Clayton         break;
1030de164aaaSGreg Clayton       default:
103136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1032de164aaaSGreg Clayton       }
1033de164aaaSGreg Clayton 
1034de164aaaSGreg Clayton       return error;
1035de164aaaSGreg Clayton     }
1036de164aaaSGreg Clayton 
1037b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1038de164aaaSGreg Clayton       m_help.clear();
1039de164aaaSGreg Clayton       m_syntax.clear();
1040de164aaaSGreg Clayton     }
1041de164aaaSGreg Clayton 
10421f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
104370602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
10441f0f5b5bSZachary Turner     }
1045de164aaaSGreg Clayton 
1046daed98e5SShivam Mittal     llvm::StringRef GetHelp() { return m_help; }
10476e3d8e7fSEugene Zelenko 
1048daed98e5SShivam Mittal     llvm::StringRef GetSyntax() { return m_syntax; }
10496e3d8e7fSEugene Zelenko 
1050de164aaaSGreg Clayton   protected:
10516e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
10526e3d8e7fSEugene Zelenko 
1053de164aaaSGreg Clayton     std::string m_help;
1054de164aaaSGreg Clayton     std::string m_syntax;
1055de164aaaSGreg Clayton   };
1056de164aaaSGreg Clayton 
1057b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1058de164aaaSGreg Clayton 
10595a988416SJim Ingham   CommandOptions m_options;
1060de164aaaSGreg Clayton };
1061de164aaaSGreg Clayton 
1062b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1063223383edSEnrico Granata public:
1064b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1065b9c1b51eSKate Stone                               std::string funct, std::string help,
1066b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1067a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_function_name(funct),
1068a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_long(false) {
1069735152e3SEnrico Granata     if (!help.empty())
1070442f6530SZachary Turner       SetHelp(help);
1071b9c1b51eSKate Stone     else {
1072735152e3SEnrico Granata       StreamString stream;
1073735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1074c156427dSZachary Turner       SetHelp(stream.GetString());
1075735152e3SEnrico Granata     }
1076223383edSEnrico Granata   }
1077223383edSEnrico Granata 
10786e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1079223383edSEnrico Granata 
1080b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
10815a988416SJim Ingham 
1082b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
10835a988416SJim Ingham 
1084b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
10855a988416SJim Ingham 
1086442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1087442f6530SZachary Turner     if (m_fetched_help_long)
1088442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1089442f6530SZachary Turner 
10902b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1091442f6530SZachary Turner     if (!scripter)
1092442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1093442f6530SZachary Turner 
1094fac939e9SEnrico Granata     std::string docstring;
1095442f6530SZachary Turner     m_fetched_help_long =
1096442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1097fac939e9SEnrico Granata     if (!docstring.empty())
1098442f6530SZachary Turner       SetHelpLong(docstring);
1099fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1100fac939e9SEnrico Granata   }
1101fac939e9SEnrico Granata 
11025a988416SJim Ingham protected:
11034d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1104b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11052b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1106223383edSEnrico Granata 
110797206d57SZachary Turner     Status error;
1108223383edSEnrico Granata 
110970f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
111070f11f88SJim Ingham 
1111a925974bSAdrian Prantl     if (!scripter || !scripter->RunScriptBasedCommand(
1112a925974bSAdrian Prantl                          m_function_name.c_str(), raw_command_line, m_synchro,
1113a925974bSAdrian Prantl                          result, error, m_exe_ctx)) {
1114223383edSEnrico Granata       result.AppendError(error.AsCString());
1115223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1116b9c1b51eSKate Stone     } else {
111770f11f88SJim Ingham       // Don't change the status if the command already set it...
1118b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1119c156427dSZachary Turner         if (result.GetOutputData().empty())
1120223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
112170f11f88SJim Ingham         else
112270f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
112370f11f88SJim Ingham       }
112470f11f88SJim Ingham     }
1125223383edSEnrico Granata 
1126223383edSEnrico Granata     return result.Succeeded();
1127223383edSEnrico Granata   }
1128223383edSEnrico Granata 
11296e3d8e7fSEugene Zelenko private:
11306e3d8e7fSEugene Zelenko   std::string m_function_name;
11316e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11326e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1133223383edSEnrico Granata };
1134223383edSEnrico Granata 
1135b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
11369fe00e52SEnrico Granata public:
11379fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
11389fe00e52SEnrico Granata                                std::string name,
11390641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1140b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1141a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp),
1142a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_short(false),
1143b9c1b51eSKate Stone         m_fetched_help_long(false) {
11449fe00e52SEnrico Granata     StreamString stream;
11459fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1146c156427dSZachary Turner     SetHelp(stream.GetString());
11472b29b432SJonas Devlieghere     if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter())
1148e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
11499fe00e52SEnrico Granata   }
11509fe00e52SEnrico Granata 
11516e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
11529fe00e52SEnrico Granata 
1153b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
11549fe00e52SEnrico Granata 
1155b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
11569fe00e52SEnrico Granata 
1157442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1158442f6530SZachary Turner     if (m_fetched_help_short)
1159442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11602b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1161442f6530SZachary Turner     if (!scripter)
1162442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11636f79bb2dSEnrico Granata     std::string docstring;
1164b9c1b51eSKate Stone     m_fetched_help_short =
1165b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
11666f79bb2dSEnrico Granata     if (!docstring.empty())
1167442f6530SZachary Turner       SetHelp(docstring);
1168442f6530SZachary Turner 
11696f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
11706f79bb2dSEnrico Granata   }
11716f79bb2dSEnrico Granata 
1172442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1173442f6530SZachary Turner     if (m_fetched_help_long)
1174442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1175442f6530SZachary Turner 
11762b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1177442f6530SZachary Turner     if (!scripter)
1178442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1179442f6530SZachary Turner 
11806f79bb2dSEnrico Granata     std::string docstring;
1181b9c1b51eSKate Stone     m_fetched_help_long =
1182b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
11836f79bb2dSEnrico Granata     if (!docstring.empty())
1184442f6530SZachary Turner       SetHelpLong(docstring);
11859fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
11869fe00e52SEnrico Granata   }
11879fe00e52SEnrico Granata 
11889fe00e52SEnrico Granata protected:
11894d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1190b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11912b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
11929fe00e52SEnrico Granata 
119397206d57SZachary Turner     Status error;
11949fe00e52SEnrico Granata 
11959fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
11969fe00e52SEnrico Granata 
1197b9c1b51eSKate Stone     if (!scripter ||
1198b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1199b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
12009fe00e52SEnrico Granata       result.AppendError(error.AsCString());
12019fe00e52SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1202b9c1b51eSKate Stone     } else {
12039fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1204b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1205c156427dSZachary Turner         if (result.GetOutputData().empty())
12069fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
12079fe00e52SEnrico Granata         else
12089fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
12099fe00e52SEnrico Granata       }
12109fe00e52SEnrico Granata     }
12119fe00e52SEnrico Granata 
12129fe00e52SEnrico Granata     return result.Succeeded();
12139fe00e52SEnrico Granata   }
12149fe00e52SEnrico Granata 
12156e3d8e7fSEugene Zelenko private:
12166e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
12176e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
12186e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
12196e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
12209fe00e52SEnrico Granata };
12219fe00e52SEnrico Granata 
1222a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
122364becc11SRaphael Isemann #define LLDB_OPTIONS_script_import
122464becc11SRaphael Isemann #include "CommandOptions.inc"
12251f0f5b5bSZachary Turner 
1226b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
12275a988416SJim Ingham public:
1228b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1229b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1230b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1231b9c1b51eSKate Stone         m_options() {
12325a988416SJim Ingham     CommandArgumentEntry arg1;
12335a988416SJim Ingham     CommandArgumentData cmd_arg;
12345a988416SJim Ingham 
12355a988416SJim Ingham     // Define the first (and only) variant of this arg.
12365a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
12373b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
12385a988416SJim Ingham 
1239b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1240b9c1b51eSKate Stone     // argument entry.
12415a988416SJim Ingham     arg1.push_back(cmd_arg);
12425a988416SJim Ingham 
12435a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
12445a988416SJim Ingham     m_arguments.push_back(arg1);
12455a988416SJim Ingham   }
12465a988416SJim Ingham 
12476e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
12485a988416SJim Ingham 
1249ae34ed2cSRaphael Isemann   void
1250ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
12512443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
1252b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1253b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1254a2e76c0bSRaphael Isemann         request, nullptr);
12555a988416SJim Ingham   }
12565a988416SJim Ingham 
1257b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
12585a988416SJim Ingham 
12595a988416SJim Ingham protected:
1260b9c1b51eSKate Stone   class CommandOptions : public Options {
12610a305db7SEnrico Granata   public:
1262b9c1b51eSKate Stone     CommandOptions() : Options() {}
12630a305db7SEnrico Granata 
12646e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
12650a305db7SEnrico Granata 
126697206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1267b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
126897206d57SZachary Turner       Status error;
12693bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
12700a305db7SEnrico Granata 
1271b9c1b51eSKate Stone       switch (short_option) {
12720a305db7SEnrico Granata       case 'r':
127315625112SJonas Devlieghere         // NO-OP
12740a305db7SEnrico Granata         break;
1275*00bb397bSJonas Devlieghere       case 'c':
1276*00bb397bSJonas Devlieghere         relative_to_command_file = true;
1277*00bb397bSJonas Devlieghere         break;
12780a305db7SEnrico Granata       default:
127936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
12800a305db7SEnrico Granata       }
12810a305db7SEnrico Granata 
12820a305db7SEnrico Granata       return error;
12830a305db7SEnrico Granata     }
12840a305db7SEnrico Granata 
1285b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1286*00bb397bSJonas Devlieghere       relative_to_command_file = false;
12870a305db7SEnrico Granata     }
12880a305db7SEnrico Granata 
12891f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
129070602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
12911f0f5b5bSZachary Turner     }
1292*00bb397bSJonas Devlieghere     bool relative_to_command_file = false;
12930a305db7SEnrico Granata   };
12940a305db7SEnrico Granata 
1295b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
129611eb9c64SZachary Turner     if (command.empty()) {
12973b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1298a9dbf432SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1299a9dbf432SEnrico Granata       return false;
1300a9dbf432SEnrico Granata     }
1301a9dbf432SEnrico Granata 
1302*00bb397bSJonas Devlieghere     FileSpec source_dir = {};
1303*00bb397bSJonas Devlieghere     if (m_options.relative_to_command_file) {
1304*00bb397bSJonas Devlieghere       source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir();
1305*00bb397bSJonas Devlieghere       if (!source_dir) {
1306*00bb397bSJonas Devlieghere         result.AppendError("command script import -c can only be specified "
1307*00bb397bSJonas Devlieghere                            "from a command file");
1308*00bb397bSJonas Devlieghere         result.SetStatus(eReturnStatusFailed);
1309*00bb397bSJonas Devlieghere         return false;
1310*00bb397bSJonas Devlieghere       }
1311*00bb397bSJonas Devlieghere     }
1312*00bb397bSJonas Devlieghere 
131311eb9c64SZachary Turner     for (auto &entry : command.entries()) {
131497206d57SZachary Turner       Status error;
1315a9dbf432SEnrico Granata 
1316c9d645d3SGreg Clayton       const bool init_session = true;
1317b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
131811eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
131911eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
132005097246SAdrian Prantl       // script imports" in __lldb_init_module the real fix is to have
132105097246SAdrian Prantl       // recursive commands possible with a CommandInvocation object separate
132205097246SAdrian Prantl       // from the CommandObject itself, so that recursive command invocations
132305097246SAdrian Prantl       // won't stomp on each other (wrt to execution contents, options, and
132405097246SAdrian Prantl       // more)
1325078551c7SEnrico Granata       m_exe_ctx.Clear();
13262b29b432SJonas Devlieghere       if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule(
1327*00bb397bSJonas Devlieghere               entry.c_str(), init_session, error, nullptr, source_dir)) {
1328a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1329b9c1b51eSKate Stone       } else {
1330b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1331b9c1b51eSKate Stone                                      error.AsCString());
1332a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusFailed);
1333a9dbf432SEnrico Granata       }
13343b00e35bSEnrico Granata     }
1335a9dbf432SEnrico Granata 
1336a9dbf432SEnrico Granata     return result.Succeeded();
1337a9dbf432SEnrico Granata   }
13380a305db7SEnrico Granata 
13395a988416SJim Ingham   CommandOptions m_options;
1340a9dbf432SEnrico Granata };
1341223383edSEnrico Granata 
1342223383edSEnrico Granata // CommandObjectCommandsScriptAdd
13438fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = {
1344e063ecccSJonas Devlieghere     {
1345e063ecccSJonas Devlieghere         eScriptedCommandSynchronicitySynchronous,
1346e063ecccSJonas Devlieghere         "synchronous",
1347e063ecccSJonas Devlieghere         "Run synchronous",
1348e063ecccSJonas Devlieghere     },
1349e063ecccSJonas Devlieghere     {
1350e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityAsynchronous,
1351e063ecccSJonas Devlieghere         "asynchronous",
1352e063ecccSJonas Devlieghere         "Run asynchronous",
1353e063ecccSJonas Devlieghere     },
1354e063ecccSJonas Devlieghere     {
1355e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityCurrentValue,
1356e063ecccSJonas Devlieghere         "current",
1357e063ecccSJonas Devlieghere         "Do not alter current setting",
1358e063ecccSJonas Devlieghere     },
1359e063ecccSJonas Devlieghere };
13601f0f5b5bSZachary Turner 
13618fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() {
13628fe53c49STatyana Krasnukha   return OptionEnumValues(g_script_synchro_type);
13638fe53c49STatyana Krasnukha }
13648fe53c49STatyana Krasnukha 
136564becc11SRaphael Isemann #define LLDB_OPTIONS_script_add
136664becc11SRaphael Isemann #include "CommandOptions.inc"
13671f0f5b5bSZachary Turner 
1368b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1369b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
13705a988416SJim Ingham public:
1371b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1372b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
13735a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
13746e3d8e7fSEugene Zelenko                             nullptr),
1375b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
13765a988416SJim Ingham     CommandArgumentEntry arg1;
13775a988416SJim Ingham     CommandArgumentData cmd_arg;
13785a988416SJim Ingham 
13795a988416SJim Ingham     // Define the first (and only) variant of this arg.
13805a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
13815a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
13825a988416SJim Ingham 
1383b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1384b9c1b51eSKate Stone     // argument entry.
13855a988416SJim Ingham     arg1.push_back(cmd_arg);
13865a988416SJim Ingham 
13875a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
13885a988416SJim Ingham     m_arguments.push_back(arg1);
13895a988416SJim Ingham   }
13905a988416SJim Ingham 
13916e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
13925a988416SJim Ingham 
1393b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13945a988416SJim Ingham 
13955a988416SJim Ingham protected:
1396b9c1b51eSKate Stone   class CommandOptions : public Options {
1397223383edSEnrico Granata   public:
1398b9c1b51eSKate Stone     CommandOptions()
1399b9c1b51eSKate Stone         : Options(), m_class_name(), m_funct_name(), m_short_help(),
1400b9c1b51eSKate Stone           m_synchronicity(eScriptedCommandSynchronicitySynchronous) {}
1401223383edSEnrico Granata 
14026e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1403223383edSEnrico Granata 
140497206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1405b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
140697206d57SZachary Turner       Status error;
14073bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1408223383edSEnrico Granata 
1409b9c1b51eSKate Stone       switch (short_option) {
1410223383edSEnrico Granata       case 'f':
1411fe11483bSZachary Turner         if (!option_arg.empty())
1412adcd0268SBenjamin Kramer           m_funct_name = std::string(option_arg);
1413735152e3SEnrico Granata         break;
14149fe00e52SEnrico Granata       case 'c':
1415fe11483bSZachary Turner         if (!option_arg.empty())
1416adcd0268SBenjamin Kramer           m_class_name = std::string(option_arg);
14179fe00e52SEnrico Granata         break;
1418735152e3SEnrico Granata       case 'h':
1419fe11483bSZachary Turner         if (!option_arg.empty())
1420adcd0268SBenjamin Kramer           m_short_help = std::string(option_arg);
1421223383edSEnrico Granata         break;
14220a305db7SEnrico Granata       case 's':
1423b9c1b51eSKate Stone         m_synchronicity =
142447cbf4a0SPavel Labath             (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum(
1425fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
14260a305db7SEnrico Granata         if (!error.Success())
1427b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1428fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1429fe11483bSZachary Turner               option_arg.str().c_str());
14300a305db7SEnrico Granata         break;
1431223383edSEnrico Granata       default:
143236162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1433223383edSEnrico Granata       }
1434223383edSEnrico Granata 
1435223383edSEnrico Granata       return error;
1436223383edSEnrico Granata     }
1437223383edSEnrico Granata 
1438b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
14399fe00e52SEnrico Granata       m_class_name.clear();
1440735152e3SEnrico Granata       m_funct_name.clear();
1441735152e3SEnrico Granata       m_short_help.clear();
144244d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1443223383edSEnrico Granata     }
1444223383edSEnrico Granata 
14451f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
144670602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
14471f0f5b5bSZachary Turner     }
1448223383edSEnrico Granata 
1449223383edSEnrico Granata     // Instance variables to hold the values for command options.
1450223383edSEnrico Granata 
14519fe00e52SEnrico Granata     std::string m_class_name;
1452223383edSEnrico Granata     std::string m_funct_name;
1453735152e3SEnrico Granata     std::string m_short_help;
145444d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1455223383edSEnrico Granata   };
1456223383edSEnrico Granata 
14570affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
14587ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
14590affb582SDave Lee     if (output_sp && interactive) {
146044d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
146144d93782SGreg Clayton       output_sp->Flush();
1462223383edSEnrico Granata     }
1463223383edSEnrico Granata   }
1464223383edSEnrico Granata 
1465b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1466b9c1b51eSKate Stone                               std::string &data) override {
14677ca15ba7SLawrence D'Anna     StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
146844d93782SGreg Clayton 
14692b29b432SJonas Devlieghere     ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1470b9c1b51eSKate Stone     if (interpreter) {
147144d93782SGreg Clayton 
147244d93782SGreg Clayton       StringList lines;
147344d93782SGreg Clayton       lines.SplitIntoLines(data);
1474b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1475a73b7df7SEnrico Granata         std::string funct_name_str;
1476b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1477b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1478b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1479b9c1b51eSKate Stone                              "add python command.\n");
148044d93782SGreg Clayton             error_sp->Flush();
1481b9c1b51eSKate Stone           } else {
1482223383edSEnrico Granata             // everything should be fine now, let's add this alias
1483223383edSEnrico Granata 
1484b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1485771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
148644d93782SGreg Clayton                 m_synchronicity));
1487223383edSEnrico Granata 
1488b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1489b9c1b51eSKate Stone                                               true)) {
1490b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1491b9c1b51eSKate Stone                                "add python command.\n");
149244d93782SGreg Clayton               error_sp->Flush();
1493223383edSEnrico Granata             }
1494223383edSEnrico Granata           }
1495b9c1b51eSKate Stone         } else {
1496b9c1b51eSKate Stone           error_sp->Printf(
1497b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
149844d93782SGreg Clayton           error_sp->Flush();
149944d93782SGreg Clayton         }
1500b9c1b51eSKate Stone       } else {
150144d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
150244d93782SGreg Clayton         error_sp->Flush();
150344d93782SGreg Clayton       }
1504b9c1b51eSKate Stone     } else {
1505b9c1b51eSKate Stone       error_sp->Printf(
1506b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
150744d93782SGreg Clayton       error_sp->Flush();
150844d93782SGreg Clayton     }
150944d93782SGreg Clayton 
151044d93782SGreg Clayton     io_handler.SetIsDone(true);
151144d93782SGreg Clayton   }
1512223383edSEnrico Granata 
1513b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
151457179860SJonas Devlieghere     if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) {
1515b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1516b9c1b51eSKate Stone                          "commands is currently Python");
151799f0b8f9SEnrico Granata       result.SetStatus(eReturnStatusFailed);
151899f0b8f9SEnrico Granata       return false;
151999f0b8f9SEnrico Granata     }
152099f0b8f9SEnrico Granata 
152111eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1522223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1523223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1524223383edSEnrico Granata       return false;
1525223383edSEnrico Granata     }
1526223383edSEnrico Granata 
1527735152e3SEnrico Granata     // Store the options in case we get multi-line input
1528adcd0268SBenjamin Kramer     m_cmd_name = std::string(command[0].ref());
1529735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
153044d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1531223383edSEnrico Granata 
1532b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1533b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1534b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1535b9c1b51eSKate Stone             "     ", // Prompt
1536a6faf851SJonas Devlieghere             *this);  // IOHandlerDelegate
1537b9c1b51eSKate Stone       } else {
1538b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1539b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1540b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1541b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1542223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1543b9c1b51eSKate Stone         } else {
1544223383edSEnrico Granata           result.AppendError("cannot add command");
1545223383edSEnrico Granata           result.SetStatus(eReturnStatusFailed);
1546223383edSEnrico Granata         }
1547223383edSEnrico Granata       }
1548b9c1b51eSKate Stone     } else {
15492b29b432SJonas Devlieghere       ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1550b9c1b51eSKate Stone       if (!interpreter) {
15519fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
15529fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15539fe00e52SEnrico Granata         return false;
15549fe00e52SEnrico Granata       }
15559fe00e52SEnrico Granata 
1556b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1557b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1558b9c1b51eSKate Stone       if (!cmd_obj_sp) {
15599fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
15609fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15619fe00e52SEnrico Granata         return false;
15629fe00e52SEnrico Granata       }
15639fe00e52SEnrico Granata 
1564b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1565b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1566b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
15679fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1568b9c1b51eSKate Stone       } else {
15699fe00e52SEnrico Granata         result.AppendError("cannot add command");
15709fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
15719fe00e52SEnrico Granata       }
15729fe00e52SEnrico Granata     }
1573223383edSEnrico Granata 
1574223383edSEnrico Granata     return result.Succeeded();
1575223383edSEnrico Granata   }
15765a988416SJim Ingham 
15775a988416SJim Ingham   CommandOptions m_options;
157844d93782SGreg Clayton   std::string m_cmd_name;
1579735152e3SEnrico Granata   std::string m_short_help;
158044d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1581223383edSEnrico Granata };
1582223383edSEnrico Granata 
1583223383edSEnrico Granata // CommandObjectCommandsScriptList
1584223383edSEnrico Granata 
1585b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1586223383edSEnrico Granata public:
1587b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1588b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1589b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1590223383edSEnrico Granata 
15916e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1592223383edSEnrico Granata 
1593b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1594d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1595d77ea5b2SRaphael Isemann       result.AppendError("'command script list' doesn't take any arguments");
1596d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1597d77ea5b2SRaphael Isemann       return false;
1598d77ea5b2SRaphael Isemann     }
1599d77ea5b2SRaphael Isemann 
1600b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1601223383edSEnrico Granata 
1602223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1603223383edSEnrico Granata 
1604223383edSEnrico Granata     return true;
1605223383edSEnrico Granata   }
1606223383edSEnrico Granata };
1607223383edSEnrico Granata 
1608223383edSEnrico Granata // CommandObjectCommandsScriptClear
1609223383edSEnrico Granata 
1610b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1611223383edSEnrico Granata public:
1612b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1613b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1614b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1615223383edSEnrico Granata 
16166e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1617223383edSEnrico Granata 
16185a988416SJim Ingham protected:
1619b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1620d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1621d77ea5b2SRaphael Isemann       result.AppendError("'command script clear' doesn't take any arguments");
1622d77ea5b2SRaphael Isemann       result.SetStatus(eReturnStatusFailed);
1623d77ea5b2SRaphael Isemann       return false;
1624d77ea5b2SRaphael Isemann     }
1625d77ea5b2SRaphael Isemann 
1626223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1627223383edSEnrico Granata 
1628223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1629223383edSEnrico Granata 
1630223383edSEnrico Granata     return true;
1631223383edSEnrico Granata   }
1632223383edSEnrico Granata };
1633223383edSEnrico Granata 
1634223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1635223383edSEnrico Granata 
1636b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1637223383edSEnrico Granata public:
1638b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1639b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1640b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1641223383edSEnrico Granata     CommandArgumentEntry arg1;
1642223383edSEnrico Granata     CommandArgumentData cmd_arg;
1643223383edSEnrico Granata 
1644223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1645223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1646223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1647223383edSEnrico Granata 
1648b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1649b9c1b51eSKate Stone     // argument entry.
1650223383edSEnrico Granata     arg1.push_back(cmd_arg);
1651223383edSEnrico Granata 
1652223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1653223383edSEnrico Granata     m_arguments.push_back(arg1);
1654223383edSEnrico Granata   }
1655223383edSEnrico Granata 
16566e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1657223383edSEnrico Granata 
16582e8f304fSGongyu Deng   void
16592e8f304fSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
16602e8f304fSGongyu Deng                            OptionElementVector &opt_element_vector) override {
16612e8f304fSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
16622e8f304fSGongyu Deng       return;
16632e8f304fSGongyu Deng 
16642ebe30c6SRaphael Isemann     for (const auto &c : m_interpreter.GetUserCommands())
16652ebe30c6SRaphael Isemann       request.TryCompleteCurrentArg(c.first, c.second->GetHelp());
16662e8f304fSGongyu Deng   }
16672e8f304fSGongyu Deng 
16685a988416SJim Ingham protected:
1669b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1670223383edSEnrico Granata 
167111eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1672223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1673223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1674223383edSEnrico Granata       return false;
1675223383edSEnrico Granata     }
1676223383edSEnrico Granata 
16770d9a201eSRaphael Isemann     auto cmd_name = command[0].ref();
1678223383edSEnrico Granata 
16794574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
16804574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1681867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
1682223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
16834574a890SZachary Turner       return false;
1684223383edSEnrico Granata     }
1685223383edSEnrico Granata 
16864574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
16874574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
16884574a890SZachary Turner     return true;
1689223383edSEnrico Granata   }
1690223383edSEnrico Granata };
1691223383edSEnrico Granata 
1692223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1693223383edSEnrico Granata 
1694223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1695223383edSEnrico Granata 
1696b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1697223383edSEnrico Granata public:
16987428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1699b9c1b51eSKate Stone       : CommandObjectMultiword(
1700a925974bSAdrian Prantl             interpreter, "command script",
1701a925974bSAdrian Prantl             "Commands for managing custom "
1702b9c1b51eSKate Stone             "commands implemented by "
1703b9c1b51eSKate Stone             "interpreter scripts.",
1704b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1705b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1706b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1707b9c1b51eSKate Stone     LoadSubCommand(
1708b9c1b51eSKate Stone         "delete",
1709b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1710b9c1b51eSKate Stone     LoadSubCommand(
1711b9c1b51eSKate Stone         "clear",
1712b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1713b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1714b9c1b51eSKate Stone                                interpreter)));
1715b9c1b51eSKate Stone     LoadSubCommand(
1716b9c1b51eSKate Stone         "import",
1717b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1718223383edSEnrico Granata   }
1719223383edSEnrico Granata 
17206e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1721223383edSEnrico Granata };
1722223383edSEnrico Granata 
1723ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1724ebc09c36SJim Ingham 
1725ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1726ebc09c36SJim Ingham 
1727b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1728b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1729b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1730b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1731b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1732b9c1b51eSKate Stone   LoadSubCommand("source",
1733b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1734b9c1b51eSKate Stone   LoadSubCommand("alias",
1735b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1736b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1737b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1738b9c1b51eSKate Stone   LoadSubCommand("delete",
1739b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1740b9c1b51eSKate Stone   LoadSubCommand(
1741b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1742b9c1b51eSKate Stone   LoadSubCommand(
1743b9c1b51eSKate Stone       "script",
1744b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1745ebc09c36SJim Ingham }
1746ebc09c36SJim Ingham 
17476e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1748