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       return false;
1324574a890SZachary Turner     }
133ebc09c36SJim Ingham 
1340d9a201eSRaphael Isemann     FileSpec cmd_file(command[0].ref());
1358f3be7a3SJonas Devlieghere     FileSystem::Instance().Resolve(cmd_file);
136ebc09c36SJim Ingham 
13736de94cfSTatyana Krasnukha     CommandInterpreterRunOptions options;
138340b0309SGreg Clayton     // If any options were set, then use them
139340b0309SGreg Clayton     if (m_options.m_stop_on_error.OptionWasSet() ||
140340b0309SGreg Clayton         m_options.m_silent_run.OptionWasSet() ||
141b9c1b51eSKate Stone         m_options.m_stop_on_continue.OptionWasSet()) {
1421c19b74cSJonas Devlieghere       if (m_options.m_stop_on_continue.OptionWasSet())
1431c19b74cSJonas Devlieghere         options.SetStopOnContinue(
1441c19b74cSJonas Devlieghere             m_options.m_stop_on_continue.GetCurrentValue());
1451c19b74cSJonas Devlieghere 
1461c19b74cSJonas Devlieghere       if (m_options.m_stop_on_error.OptionWasSet())
14726c7bf93SJim Ingham         options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue());
148c678ed77SStefan Granitz 
149c678ed77SStefan Granitz       // Individual silent setting is override for global command echo settings.
150c678ed77SStefan Granitz       if (m_options.m_silent_run.GetCurrentValue()) {
151c678ed77SStefan Granitz         options.SetSilent(true);
152c678ed77SStefan Granitz       } else {
153c678ed77SStefan Granitz         options.SetPrintResults(true);
154c0b48ab6SJonas Devlieghere         options.SetPrintErrors(true);
155c678ed77SStefan Granitz         options.SetEchoCommands(m_interpreter.GetEchoCommands());
156c678ed77SStefan Granitz         options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands());
157c678ed77SStefan Granitz       }
158122a4ebdSPavel Labath     }
15936de94cfSTatyana Krasnukha 
16036de94cfSTatyana Krasnukha     m_interpreter.HandleCommandsFromFile(cmd_file, options, result);
161ebc09c36SJim Ingham     return result.Succeeded();
162ebc09c36SJim Ingham   }
1636e3d8e7fSEugene Zelenko 
1645a988416SJim Ingham   CommandOptions m_options;
165ebc09c36SJim Ingham };
166ebc09c36SJim Ingham 
167ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
168ebc09c36SJim Ingham // CommandObjectCommandsAlias
169ebc09c36SJim Ingham 
17064becc11SRaphael Isemann #define LLDB_OPTIONS_alias
17164becc11SRaphael Isemann #include "CommandOptions.inc"
1721f0f5b5bSZachary Turner 
173b9c1b51eSKate Stone static const char *g_python_command_instructions =
174b9c1b51eSKate Stone     "Enter your Python command(s). Type 'DONE' to end.\n"
175be93a35aSEnrico Granata     "You must define a Python function with this signature:\n"
17644d93782SGreg Clayton     "def my_command_impl(debugger, args, result, internal_dict):\n";
177be93a35aSEnrico Granata 
178b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw {
17945d0e238SEnrico Granata protected:
180b9c1b51eSKate Stone   class CommandOptions : public OptionGroup {
181ebc09c36SJim Ingham   public:
182b9c1b51eSKate Stone     CommandOptions() : OptionGroup(), m_help(), m_long_help() {}
18345d0e238SEnrico Granata 
18445d0e238SEnrico Granata     ~CommandOptions() override = default;
18545d0e238SEnrico Granata 
1861f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
18770602439SZachary Turner       return llvm::makeArrayRef(g_alias_options);
1881f0f5b5bSZachary Turner     }
18945d0e238SEnrico Granata 
19097206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value,
191b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
19297206d57SZachary Turner       Status error;
19345d0e238SEnrico Granata 
1941f0f5b5bSZachary Turner       const int short_option = GetDefinitions()[option_idx].short_option;
1958cef4b0bSZachary Turner       std::string option_str(option_value);
19645d0e238SEnrico Granata 
197b9c1b51eSKate Stone       switch (short_option) {
19845d0e238SEnrico Granata       case 'h':
1998cef4b0bSZachary Turner         m_help.SetCurrentValue(option_str);
20045d0e238SEnrico Granata         m_help.SetOptionWasSet();
20145d0e238SEnrico Granata         break;
20245d0e238SEnrico Granata 
20345d0e238SEnrico Granata       case 'H':
2048cef4b0bSZachary Turner         m_long_help.SetCurrentValue(option_str);
20545d0e238SEnrico Granata         m_long_help.SetOptionWasSet();
20645d0e238SEnrico Granata         break;
20745d0e238SEnrico Granata 
20845d0e238SEnrico Granata       default:
20936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
21045d0e238SEnrico Granata       }
21145d0e238SEnrico Granata 
21245d0e238SEnrico Granata       return error;
21345d0e238SEnrico Granata     }
21445d0e238SEnrico Granata 
215b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
21645d0e238SEnrico Granata       m_help.Clear();
21745d0e238SEnrico Granata       m_long_help.Clear();
21845d0e238SEnrico Granata     }
21945d0e238SEnrico Granata 
22045d0e238SEnrico Granata     OptionValueString m_help;
22145d0e238SEnrico Granata     OptionValueString m_long_help;
22245d0e238SEnrico Granata   };
22345d0e238SEnrico Granata 
22445d0e238SEnrico Granata   OptionGroupOptions m_option_group;
22545d0e238SEnrico Granata   CommandOptions m_command_options;
22645d0e238SEnrico Granata 
22745d0e238SEnrico Granata public:
228b9c1b51eSKate Stone   Options *GetOptions() override { return &m_option_group; }
22945d0e238SEnrico Granata 
2307428a18cSKate Stone   CommandObjectCommandsAlias(CommandInterpreter &interpreter)
231b9c1b51eSKate Stone       : CommandObjectRaw(
232b9c1b51eSKate Stone             interpreter, "command alias",
233a449698cSZachary Turner             "Define a custom command in terms of an existing command."),
234b9c1b51eSKate Stone         m_option_group(), m_command_options() {
23545d0e238SEnrico Granata     m_option_group.Append(&m_command_options);
23645d0e238SEnrico Granata     m_option_group.Finalize();
23745d0e238SEnrico Granata 
238ebc09c36SJim Ingham     SetHelpLong(
239ea671fbdSKate Stone         "'alias' allows the user to create a short-cut or abbreviation for long \
240ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
241b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:"
242b9c1b51eSKate Stone         R"(
243ea671fbdSKate Stone 
244ea671fbdSKate Stone (lldb) command alias sc script
245ea671fbdSKate Stone 
246ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
247ea671fbdSKate Stone 
248ea671fbdSKate Stone (lldb) command alias bp breakpoint
249ea671fbdSKate Stone 
250b9c1b51eSKate Stone )"
251b9c1b51eSKate Stone         "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
252ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
253b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'."
254b9c1b51eSKate Stone         R"(
255ea671fbdSKate Stone 
256ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
257ea671fbdSKate Stone 
258ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
259ea671fbdSKate Stone 
260b9c1b51eSKate Stone )"
261b9c1b51eSKate Stone         "An alias can include some options for the command, with the values either \
262ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
263ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
264b9c1b51eSKate Stone shows how to create aliases with options:"
265b9c1b51eSKate Stone         R"(
266ea671fbdSKate Stone 
267ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
268ea671fbdSKate Stone 
269b9c1b51eSKate Stone )"
270b9c1b51eSKate Stone         "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
271ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
272ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
273ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
274ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
275ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
276ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
277ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
278ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
279ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
280b9c1b51eSKate Stone launch' example below)."
281b9c1b51eSKate Stone         R"(
282ea671fbdSKate Stone 
283b9c1b51eSKate Stone )"
284b9c1b51eSKate Stone         "Note: the positional arguments must substitute as whole words in the resultant \
285ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
286b9c1b51eSKate Stone \".cpp\":"
287b9c1b51eSKate Stone         R"(
288ea671fbdSKate Stone 
289ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
290ea671fbdSKate Stone 
291b9c1b51eSKate Stone )"
292b9c1b51eSKate Stone         "For more complex aliasing, use the \"command regex\" command instead.  In the \
293ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
294ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
295b9c1b51eSKate Stone argument.  The user would use this alias as follows:"
296b9c1b51eSKate Stone         R"(
297ea671fbdSKate Stone 
298ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
299ea671fbdSKate Stone (lldb) bfl my-file.c 137
300ea671fbdSKate Stone 
301ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
302ea671fbdSKate Stone 
303ea671fbdSKate Stone Another example:
304ea671fbdSKate Stone 
305ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
306ea671fbdSKate Stone (lldb) pltty /dev/tty0
307ea671fbdSKate Stone 
308ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
309ea671fbdSKate Stone 
310b9c1b51eSKate Stone )"
311b9c1b51eSKate Stone         "If the user always wanted to pass the same value to a particular option, the \
312ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
313b9c1b51eSKate Stone rather than using a positional placeholder:"
314b9c1b51eSKate Stone         R"(
315ea671fbdSKate Stone 
316ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
317ea671fbdSKate Stone 
318b9c1b51eSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)");
319ebc09c36SJim Ingham 
320405fe67fSCaroline Tice     CommandArgumentEntry arg1;
321405fe67fSCaroline Tice     CommandArgumentEntry arg2;
322405fe67fSCaroline Tice     CommandArgumentEntry arg3;
323405fe67fSCaroline Tice     CommandArgumentData alias_arg;
324405fe67fSCaroline Tice     CommandArgumentData cmd_arg;
325405fe67fSCaroline Tice     CommandArgumentData options_arg;
326405fe67fSCaroline Tice 
327405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
328405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
329405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
330405fe67fSCaroline Tice 
331b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
332b9c1b51eSKate Stone     // argument entry.
333405fe67fSCaroline Tice     arg1.push_back(alias_arg);
334405fe67fSCaroline Tice 
335405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
336405fe67fSCaroline Tice     cmd_arg.arg_type = eArgTypeCommandName;
337405fe67fSCaroline Tice     cmd_arg.arg_repetition = eArgRepeatPlain;
338405fe67fSCaroline Tice 
339b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
340b9c1b51eSKate Stone     // argument entry.
341405fe67fSCaroline Tice     arg2.push_back(cmd_arg);
342405fe67fSCaroline Tice 
343405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
344405fe67fSCaroline Tice     options_arg.arg_type = eArgTypeAliasOptions;
345405fe67fSCaroline Tice     options_arg.arg_repetition = eArgRepeatOptional;
346405fe67fSCaroline Tice 
347b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
348b9c1b51eSKate Stone     // argument entry.
349405fe67fSCaroline Tice     arg3.push_back(options_arg);
350405fe67fSCaroline Tice 
351405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
352405fe67fSCaroline Tice     m_arguments.push_back(arg1);
353405fe67fSCaroline Tice     m_arguments.push_back(arg2);
354405fe67fSCaroline Tice     m_arguments.push_back(arg3);
355ebc09c36SJim Ingham   }
356ebc09c36SJim Ingham 
3576e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAlias() override = default;
358ebc09c36SJim Ingham 
3595a988416SJim Ingham protected:
3604d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
361b9c1b51eSKate Stone                  CommandReturnObject &result) override {
3624d51a902SRaphael Isemann     if (raw_command_line.empty()) {
363d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
36445d0e238SEnrico Granata       return false;
36545d0e238SEnrico Granata     }
36645d0e238SEnrico Granata 
367e1cfbc79STodd Fiala     ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext();
368e1cfbc79STodd Fiala     m_option_group.NotifyOptionParsingStarting(&exe_ctx);
36945d0e238SEnrico Granata 
3703a0e1270SRaphael Isemann     OptionsWithRaw args_with_suffix(raw_command_line);
37145d0e238SEnrico Granata 
3723a0e1270SRaphael Isemann     if (args_with_suffix.HasArgs())
3733a0e1270SRaphael Isemann       if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result,
3743a0e1270SRaphael Isemann                                  m_option_group, exe_ctx))
37545d0e238SEnrico Granata         return false;
37645d0e238SEnrico Granata 
377daed98e5SShivam Mittal     llvm::StringRef raw_command_string = args_with_suffix.GetRawPart();
378a01bccdbSZachary Turner     Args args(raw_command_string);
379844d2303SCaroline Tice 
38011eb9c64SZachary Turner     if (args.GetArgumentCount() < 2) {
381d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
382844d2303SCaroline Tice       return false;
383844d2303SCaroline Tice     }
384844d2303SCaroline Tice 
385844d2303SCaroline Tice     // Get the alias command.
386844d2303SCaroline Tice 
3870d9a201eSRaphael Isemann     auto alias_command = args[0].ref();
3884574a890SZachary Turner     if (alias_command.startswith("-")) {
389d72e412fSEnrico Granata       result.AppendError("aliases starting with a dash are not supported");
390b9c1b51eSKate Stone       if (alias_command == "--help" || alias_command == "--long-help") {
391b9c1b51eSKate Stone         result.AppendWarning("if trying to pass options to 'command alias' add "
392b9c1b51eSKate Stone                              "a -- at the end of the options");
393d72e412fSEnrico Granata       }
394d72e412fSEnrico Granata       return false;
395d72e412fSEnrico Granata     }
396844d2303SCaroline Tice 
397b9c1b51eSKate Stone     // Strip the new alias name off 'raw_command_string'  (leave it on args,
39805097246SAdrian Prantl     // which gets passed to 'Execute', which does the stripping itself.
399844d2303SCaroline Tice     size_t pos = raw_command_string.find(alias_command);
400b9c1b51eSKate Stone     if (pos == 0) {
401844d2303SCaroline Tice       raw_command_string = raw_command_string.substr(alias_command.size());
402844d2303SCaroline Tice       pos = raw_command_string.find_first_not_of(' ');
403844d2303SCaroline Tice       if ((pos != std::string::npos) && (pos > 0))
404844d2303SCaroline Tice         raw_command_string = raw_command_string.substr(pos);
405b9c1b51eSKate Stone     } else {
406844d2303SCaroline Tice       result.AppendError("Error parsing command string.  No alias created.");
407844d2303SCaroline Tice       return false;
408844d2303SCaroline Tice     }
409844d2303SCaroline Tice 
410844d2303SCaroline Tice     // Verify that the command is alias-able.
411771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
412b9c1b51eSKate Stone       result.AppendErrorWithFormat(
413b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
4144574a890SZachary Turner           args[0].c_str());
415844d2303SCaroline Tice       return false;
416844d2303SCaroline Tice     }
417844d2303SCaroline Tice 
418b9c1b51eSKate Stone     // Get CommandObject that is being aliased. The command name is read from
419a01bccdbSZachary Turner     // the front of raw_command_string. raw_command_string is returned with the
420a01bccdbSZachary Turner     // name of the command object stripped off the front.
421a01bccdbSZachary Turner     llvm::StringRef original_raw_command_string = raw_command_string;
422b9c1b51eSKate Stone     CommandObject *cmd_obj =
423b9c1b51eSKate Stone         m_interpreter.GetCommandObjectForCommand(raw_command_string);
424844d2303SCaroline Tice 
425b9c1b51eSKate Stone     if (!cmd_obj) {
426b9c1b51eSKate Stone       result.AppendErrorWithFormat("invalid command given to 'command alias'. "
427b9c1b51eSKate Stone                                    "'%s' does not begin with a valid command."
428b9c1b51eSKate Stone                                    "  No alias created.",
429a01bccdbSZachary Turner                                    original_raw_command_string.str().c_str());
430844d2303SCaroline Tice       return false;
431b9c1b51eSKate Stone     } else if (!cmd_obj->WantsRawCommandString()) {
432b9c1b51eSKate Stone       // Note that args was initialized with the original command, and has not
43305097246SAdrian Prantl       // been updated to this point. Therefore can we pass it to the version of
43405097246SAdrian Prantl       // Execute that does not need/expect raw input in the alias.
4355a988416SJim Ingham       return HandleAliasingNormalCommand(args, result);
436b9c1b51eSKate Stone     } else {
437b9c1b51eSKate Stone       return HandleAliasingRawCommand(alias_command, raw_command_string,
438b9c1b51eSKate Stone                                       *cmd_obj, result);
4395a988416SJim Ingham     }
4405a988416SJim Ingham     return result.Succeeded();
4415a988416SJim Ingham   }
4425a988416SJim Ingham 
443a01bccdbSZachary Turner   bool HandleAliasingRawCommand(llvm::StringRef alias_command,
444a01bccdbSZachary Turner                                 llvm::StringRef raw_command_string,
445b9c1b51eSKate Stone                                 CommandObject &cmd_obj,
446b9c1b51eSKate Stone                                 CommandReturnObject &result) {
447844d2303SCaroline Tice     // Verify & handle any options/arguments passed to the alias command
448844d2303SCaroline Tice 
449b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
450b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
451844d2303SCaroline Tice 
452b9c1b51eSKate Stone     if (CommandObjectSP cmd_obj_sp =
453a9448872SJonas Devlieghere             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName())) {
454a01bccdbSZachary Turner       if (m_interpreter.AliasExists(alias_command) ||
455a01bccdbSZachary Turner           m_interpreter.UserCommandExists(alias_command)) {
456b9c1b51eSKate Stone         result.AppendWarningWithFormat(
457b9c1b51eSKate Stone             "Overwriting existing definition for '%s'.\n",
458a01bccdbSZachary Turner             alias_command.str().c_str());
459844d2303SCaroline Tice       }
460b9c1b51eSKate Stone       if (CommandAlias *alias = m_interpreter.AddAlias(
461a01bccdbSZachary Turner               alias_command, cmd_obj_sp, raw_command_string)) {
46245d0e238SEnrico Granata         if (m_command_options.m_help.OptionWasSet())
46345d0e238SEnrico Granata           alias->SetHelp(m_command_options.m_help.GetCurrentValue());
46445d0e238SEnrico Granata         if (m_command_options.m_long_help.OptionWasSet())
46545d0e238SEnrico Granata           alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
466844d2303SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
467b9c1b51eSKate Stone       } else {
468472362e6SCaroline Tice         result.AppendError("Unable to create requested alias.\n");
469472362e6SCaroline Tice       }
470212130acSEnrico Granata 
471b9c1b51eSKate Stone     } else {
472212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
473212130acSEnrico Granata     }
474212130acSEnrico Granata 
475844d2303SCaroline Tice     return result.Succeeded();
476844d2303SCaroline Tice   }
477ebc09c36SJim Ingham 
478b9c1b51eSKate Stone   bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) {
479867b185dSCaroline Tice     size_t argc = args.GetArgumentCount();
480ebc09c36SJim Ingham 
481b9c1b51eSKate Stone     if (argc < 2) {
482d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
483ebc09c36SJim Ingham       return false;
484ebc09c36SJim Ingham     }
485ebc09c36SJim Ingham 
4864574a890SZachary Turner     // Save these in std::strings since we're going to shift them off.
487adcd0268SBenjamin Kramer     const std::string alias_command(std::string(args[0].ref()));
488adcd0268SBenjamin Kramer     const std::string actual_command(std::string(args[1].ref()));
489ebc09c36SJim Ingham 
490ebc09c36SJim Ingham     args.Shift(); // Shift the alias command word off the argument vector.
491ebc09c36SJim Ingham     args.Shift(); // Shift the old command word off the argument vector.
492ebc09c36SJim Ingham 
493b9c1b51eSKate Stone     // Verify that the command is alias'able, and get the appropriate command
494b9c1b51eSKate Stone     // object.
495ebc09c36SJim Ingham 
496771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
497b9c1b51eSKate Stone       result.AppendErrorWithFormat(
498b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
499ebc09c36SJim Ingham           alias_command.c_str());
5004574a890SZachary Turner       return false;
5014574a890SZachary Turner     }
5024574a890SZachary Turner 
503b9c1b51eSKate Stone     CommandObjectSP command_obj_sp(
504a449698cSZachary Turner         m_interpreter.GetCommandSPExact(actual_command, true));
505ebc09c36SJim Ingham     CommandObjectSP subcommand_obj_sp;
506ebc09c36SJim Ingham     bool use_subcommand = false;
5074574a890SZachary Turner     if (!command_obj_sp) {
5084574a890SZachary Turner       result.AppendErrorWithFormat("'%s' is not an existing command.\n",
5094574a890SZachary Turner                                    actual_command.c_str());
5104574a890SZachary Turner       return false;
5114574a890SZachary Turner     }
512ebc09c36SJim Ingham     CommandObject *cmd_obj = command_obj_sp.get();
5136e3d8e7fSEugene Zelenko     CommandObject *sub_cmd_obj = nullptr;
514b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
515b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
516ebc09c36SJim Ingham 
51711eb9c64SZachary Turner     while (cmd_obj->IsMultiwordObject() && !args.empty()) {
5180d9a201eSRaphael Isemann       auto sub_command = args[0].ref();
51911eb9c64SZachary Turner       assert(!sub_command.empty());
5204574a890SZachary Turner       subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command);
5214574a890SZachary Turner       if (!subcommand_obj_sp) {
522b9c1b51eSKate Stone         result.AppendErrorWithFormat(
523b9c1b51eSKate Stone             "'%s' is not a valid sub-command of '%s'.  "
524f415eeb4SCaroline Tice             "Unable to create alias.\n",
5254574a890SZachary Turner             args[0].c_str(), actual_command.c_str());
526ebc09c36SJim Ingham         return false;
527ebc09c36SJim Ingham       }
5284574a890SZachary Turner 
5294574a890SZachary Turner       sub_cmd_obj = subcommand_obj_sp.get();
5304574a890SZachary Turner       use_subcommand = true;
5314574a890SZachary Turner       args.Shift(); // Shift the sub_command word off the argument vector.
5324574a890SZachary Turner       cmd_obj = sub_cmd_obj;
533ebc09c36SJim Ingham     }
534ebc09c36SJim Ingham 
535ebc09c36SJim Ingham     // Verify & handle any options/arguments passed to the alias command
536ebc09c36SJim Ingham 
537212130acSEnrico Granata     std::string args_string;
538212130acSEnrico Granata 
53911eb9c64SZachary Turner     if (!args.empty()) {
540b9c1b51eSKate Stone       CommandObjectSP tmp_sp =
541a9448872SJonas Devlieghere           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName());
542ebc09c36SJim Ingham       if (use_subcommand)
543a9448872SJonas Devlieghere         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName());
544ca90c47eSCaroline Tice 
545ca90c47eSCaroline Tice       args.GetCommandString(args_string);
546867b185dSCaroline Tice     }
547ebc09c36SJim Ingham 
548771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
549771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
550b9c1b51eSKate Stone       result.AppendWarningWithFormat(
5514574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
552ebc09c36SJim Ingham     }
553ebc09c36SJim Ingham 
554b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
5554574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
556771ef6d4SMalcolm Parsons             args_string)) {
55745d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
55845d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
55945d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
56045d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
561ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
562b9c1b51eSKate Stone     } else {
563212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
564212130acSEnrico Granata       return false;
565212130acSEnrico Granata     }
566ebc09c36SJim Ingham 
567ebc09c36SJim Ingham     return result.Succeeded();
568ebc09c36SJim Ingham   }
569ebc09c36SJim Ingham };
570ebc09c36SJim Ingham 
571ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
572ebc09c36SJim Ingham // CommandObjectCommandsUnalias
573ebc09c36SJim Ingham 
574b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
575ebc09c36SJim Ingham public:
5767428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
577b9c1b51eSKate Stone       : CommandObjectParsed(
578b9c1b51eSKate Stone             interpreter, "command unalias",
579b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
580b9c1b51eSKate Stone             nullptr) {
581405fe67fSCaroline Tice     CommandArgumentEntry arg;
582405fe67fSCaroline Tice     CommandArgumentData alias_arg;
583405fe67fSCaroline Tice 
584405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
585405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
586405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
587405fe67fSCaroline Tice 
588b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
589b9c1b51eSKate Stone     // argument entry.
590405fe67fSCaroline Tice     arg.push_back(alias_arg);
591405fe67fSCaroline Tice 
592405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
593405fe67fSCaroline Tice     m_arguments.push_back(arg);
594ebc09c36SJim Ingham   }
595ebc09c36SJim Ingham 
5966e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
597ebc09c36SJim Ingham 
59831fd64acSGongyu Deng   void
59931fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
60031fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
60131fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
60231fd64acSGongyu Deng       return;
60331fd64acSGongyu Deng 
60431fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetAliases()) {
60531fd64acSGongyu Deng       request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
60631fd64acSGongyu Deng     }
60731fd64acSGongyu Deng   }
60831fd64acSGongyu Deng 
6095a988416SJim Ingham protected:
610b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
611ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
612ebc09c36SJim Ingham     CommandObject *cmd_obj;
613ebc09c36SJim Ingham 
61411eb9c64SZachary Turner     if (args.empty()) {
61511eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
61611eb9c64SZachary Turner       return false;
61711eb9c64SZachary Turner     }
61811eb9c64SZachary Turner 
6190d9a201eSRaphael Isemann     auto command_name = args[0].ref();
620a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
6214574a890SZachary Turner     if (!cmd_obj) {
6224574a890SZachary Turner       result.AppendErrorWithFormat(
6234574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
6244574a890SZachary Turner           "current list of commands.\n",
625867e7d17SZachary Turner           args[0].c_str());
6264574a890SZachary Turner       return false;
6274574a890SZachary Turner     }
6284574a890SZachary Turner 
629b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
630b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
631b9c1b51eSKate Stone         result.AppendErrorWithFormat(
632b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
633b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
634867e7d17SZachary Turner             args[0].c_str());
635b9c1b51eSKate Stone       } else {
636b9c1b51eSKate Stone         result.AppendErrorWithFormat(
637b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
638867e7d17SZachary Turner             args[0].c_str());
639b547278cSGreg Clayton       }
6404574a890SZachary Turner       return false;
6414574a890SZachary Turner     }
6424574a890SZachary Turner 
643b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
644a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
645b9c1b51eSKate Stone         result.AppendErrorWithFormat(
646867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
647867e7d17SZachary Turner             args[0].c_str());
648ebc09c36SJim Ingham       else
649b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
650867e7d17SZachary Turner                                      args[0].c_str());
6514574a890SZachary Turner       return false;
652ebc09c36SJim Ingham     }
653ebc09c36SJim Ingham 
6544574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
655ebc09c36SJim Ingham     return result.Succeeded();
656ebc09c36SJim Ingham   }
657ebc09c36SJim Ingham };
658ebc09c36SJim Ingham 
659b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
660b547278cSGreg Clayton // CommandObjectCommandsDelete
661b547278cSGreg Clayton 
662b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
663b547278cSGreg Clayton public:
6647428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
665b9c1b51eSKate Stone       : CommandObjectParsed(
666b9c1b51eSKate Stone             interpreter, "command delete",
667b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
668b9c1b51eSKate Stone             nullptr) {
669b547278cSGreg Clayton     CommandArgumentEntry arg;
670b547278cSGreg Clayton     CommandArgumentData alias_arg;
671b547278cSGreg Clayton 
672b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
673b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
674b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
675b547278cSGreg Clayton 
676b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
677b9c1b51eSKate Stone     // argument entry.
678b547278cSGreg Clayton     arg.push_back(alias_arg);
679b547278cSGreg Clayton 
680b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
681b547278cSGreg Clayton     m_arguments.push_back(arg);
682b547278cSGreg Clayton   }
683b547278cSGreg Clayton 
6846e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
685b547278cSGreg Clayton 
68631fd64acSGongyu Deng   void
68731fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
68831fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
68931fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
69031fd64acSGongyu Deng       return;
69131fd64acSGongyu Deng 
69231fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetCommands()) {
69331fd64acSGongyu Deng       if (ent.second->IsRemovable())
69431fd64acSGongyu Deng         request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
69531fd64acSGongyu Deng     }
69631fd64acSGongyu Deng   }
69731fd64acSGongyu Deng 
698b547278cSGreg Clayton protected:
699b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
700b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
701b547278cSGreg Clayton 
70211eb9c64SZachary Turner     if (args.empty()) {
70311eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
70411eb9c64SZachary Turner                                    "defined regular expression command names",
705a449698cSZachary Turner                                    GetCommandName().str().c_str());
706d77ea5b2SRaphael Isemann       return false;
70711eb9c64SZachary Turner     }
70811eb9c64SZachary Turner 
7090d9a201eSRaphael Isemann     auto command_name = args[0].ref();
7104574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
71146d4aa21SEnrico Granata       StreamString error_msg_stream;
712d5b44036SJonas Devlieghere       const bool generate_upropos = true;
71346d4aa21SEnrico Granata       const bool generate_type_lookup = false;
714b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
7154574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
716d5b44036SJonas Devlieghere           generate_upropos, generate_type_lookup);
717c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
7184574a890SZachary Turner       return false;
719b547278cSGreg Clayton     }
720b547278cSGreg Clayton 
7214574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
7224574a890SZachary Turner       result.AppendErrorWithFormat(
7234574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
724867e7d17SZachary Turner           args[0].c_str());
7254574a890SZachary Turner       return false;
7264574a890SZachary Turner     }
7274574a890SZachary Turner 
7284574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
7294574a890SZachary Turner     return true;
730b547278cSGreg Clayton   }
731b547278cSGreg Clayton };
732b547278cSGreg Clayton 
733de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
7341f0f5b5bSZachary Turner 
73564becc11SRaphael Isemann #define LLDB_OPTIONS_regex
73664becc11SRaphael Isemann #include "CommandOptions.inc"
7371f0f5b5bSZachary Turner 
7385a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
739de164aaaSGreg Clayton 
740b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
741b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
742de164aaaSGreg Clayton public:
7437428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
744b9c1b51eSKate Stone       : CommandObjectParsed(
745a925974bSAdrian Prantl             interpreter, "command regex",
746a925974bSAdrian Prantl             "Define a custom command in terms of "
747b9c1b51eSKate Stone             "existing commands by matching "
748b9c1b51eSKate Stone             "regular expressions.",
7490e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
750b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
751b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
752b9c1b51eSKate Stone         m_options() {
753b9c1b51eSKate Stone     SetHelpLong(
754b9c1b51eSKate Stone         R"(
755b9c1b51eSKate Stone )"
756b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
757ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
758b9c1b51eSKate Stone using the regular expression substitution format of:"
759b9c1b51eSKate Stone         R"(
760ea671fbdSKate Stone 
761ea671fbdSKate Stone     s/<regex>/<subst>/
762ea671fbdSKate Stone 
763b9c1b51eSKate Stone )"
764b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
765ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
766b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
767b9c1b51eSKate Stone         R"(
768ea671fbdSKate Stone 
769b9c1b51eSKate Stone )"
770b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
771ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
772ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
773b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
774b9c1b51eSKate Stone         R"(
775ea671fbdSKate Stone 
776ea671fbdSKate Stone EXAMPLES
777ea671fbdSKate Stone 
778b9c1b51eSKate Stone )"
779b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
780ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
781b9c1b51eSKate Stone a number follows 'f':"
782b9c1b51eSKate Stone         R"(
783ea671fbdSKate Stone 
784b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
785de164aaaSGreg Clayton   }
786de164aaaSGreg Clayton 
7876e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
788de164aaaSGreg Clayton 
7895a988416SJim Ingham protected:
7900affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
7917ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
7920affb582SDave Lee     if (output_sp && interactive) {
7930affb582SDave Lee       output_sp->PutCString("Enter one or more sed substitution commands in "
794b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
795b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
79644d93782SGreg Clayton       output_sp->Flush();
79744d93782SGreg Clayton     }
79844d93782SGreg Clayton   }
79944d93782SGreg Clayton 
800b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
801b9c1b51eSKate Stone                               std::string &data) override {
80244d93782SGreg Clayton     io_handler.SetIsDone(true);
803d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
80444d93782SGreg Clayton       StringList lines;
805b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
80644d93782SGreg Clayton         bool check_only = false;
8074c78b788SRaphael Isemann         for (const std::string &line : lines) {
8084c78b788SRaphael Isemann           Status error = AppendRegexSubstitution(line, check_only);
809b9c1b51eSKate Stone           if (error.Fail()) {
81057179860SJonas Devlieghere             if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) {
81157179860SJonas Devlieghere               StreamSP out_stream = GetDebugger().GetAsyncOutputStream();
81244d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
81344d93782SGreg Clayton             }
81444d93782SGreg Clayton           }
81544d93782SGreg Clayton         }
81644d93782SGreg Clayton       }
817d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
818d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
81944d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
82044d93782SGreg Clayton       }
82144d93782SGreg Clayton     }
82244d93782SGreg Clayton   }
82344d93782SGreg Clayton 
824b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
8255a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
826b9c1b51eSKate Stone     if (argc == 0) {
827b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
828b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
82911eb9c64SZachary Turner       return false;
83011eb9c64SZachary Turner     }
83111eb9c64SZachary Turner 
83297206d57SZachary Turner     Status error;
8330d9a201eSRaphael Isemann     auto name = command[0].ref();
834a8f3ae7cSJonas Devlieghere     m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>(
8354574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
8364574a890SZachary Turner         true);
8370e5e5a79SGreg Clayton 
838b9c1b51eSKate Stone     if (argc == 1) {
83957179860SJonas Devlieghere       Debugger &debugger = GetDebugger();
840e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
84144d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
842b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
843b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
84473d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
845514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
846514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
847b9c1b51eSKate Stone           multiple_lines, color_prompt,
848f6913cd7SGreg Clayton           0, // Don't show line numbers
849d77c2e09SJonas Devlieghere           *this, nullptr));
85044d93782SGreg Clayton 
851b9c1b51eSKate Stone       if (io_handler_sp) {
8527ce2de2cSJonas Devlieghere         debugger.RunIOHandlerAsync(io_handler_sp);
853de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
854de164aaaSGreg Clayton       }
855b9c1b51eSKate Stone     } else {
85697d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
85744d93782SGreg Clayton         bool check_only = false;
8580d9a201eSRaphael Isemann         error = AppendRegexSubstitution(entry.ref(), check_only);
8590e5e5a79SGreg Clayton         if (error.Fail())
8600e5e5a79SGreg Clayton           break;
8610e5e5a79SGreg Clayton       }
8620e5e5a79SGreg Clayton 
863b9c1b51eSKate Stone       if (error.Success()) {
8640e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
8650e5e5a79SGreg Clayton       }
8660e5e5a79SGreg Clayton     }
867b9c1b51eSKate Stone     if (error.Fail()) {
8680e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
869de164aaaSGreg Clayton     }
8700e5e5a79SGreg Clayton 
871de164aaaSGreg Clayton     return result.Succeeded();
872de164aaaSGreg Clayton   }
873de164aaaSGreg Clayton 
87497206d57SZachary Turner   Status AppendRegexSubstitution(const llvm::StringRef &regex_sed,
875b9c1b51eSKate Stone                                  bool check_only) {
87697206d57SZachary Turner     Status error;
8770e5e5a79SGreg Clayton 
878d5b44036SJonas Devlieghere     if (!m_regex_cmd_up) {
879b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
880b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
881b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
8820e5e5a79SGreg Clayton       return error;
883de164aaaSGreg Clayton     }
8840e5e5a79SGreg Clayton 
8850e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
8860e5e5a79SGreg Clayton 
887b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
888b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
889b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
890b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
8910e5e5a79SGreg Clayton       return error;
8920e5e5a79SGreg Clayton     }
8930e5e5a79SGreg Clayton 
894b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
895b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
896b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
897b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
8980e5e5a79SGreg Clayton       return error;
8990e5e5a79SGreg Clayton     }
9000e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
90105097246SAdrian Prantl     // use the char that follows 's' as the regex separator character so we can
90205097246SAdrian Prantl     // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9030e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
904b9c1b51eSKate Stone     const size_t second_separator_char_pos =
905b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
9060e5e5a79SGreg Clayton 
907b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
908b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
909b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
9100e5e5a79SGreg Clayton           separator_char,
9110e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
912ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
913b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9140e5e5a79SGreg Clayton       return error;
9150e5e5a79SGreg Clayton     }
9160e5e5a79SGreg Clayton 
917b9c1b51eSKate Stone     const size_t third_separator_char_pos =
918b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
9190e5e5a79SGreg Clayton 
920b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
921b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
922b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
9230e5e5a79SGreg Clayton           separator_char,
9240e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
925ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
926b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9270e5e5a79SGreg Clayton       return error;
9280e5e5a79SGreg Clayton     }
9290e5e5a79SGreg Clayton 
930b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
93105097246SAdrian Prantl       // Make sure that everything that follows the last regex separator char
932b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
933b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
934b9c1b51eSKate Stone           std::string::npos) {
935b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
936b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
937b9c1b51eSKate Stone             "string: '%.*s'",
938b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
9390e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
9400e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
9410e5e5a79SGreg Clayton         return error;
9420e5e5a79SGreg Clayton       }
943b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
944b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
945b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
946b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9470e5e5a79SGreg Clayton           regex_sed.data());
9480e5e5a79SGreg Clayton       return error;
949b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
950b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
951b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
952b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9530e5e5a79SGreg Clayton           regex_sed.data());
9540e5e5a79SGreg Clayton       return error;
9550e5e5a79SGreg Clayton     }
95644d93782SGreg Clayton 
957b9c1b51eSKate Stone     if (!check_only) {
958adcd0268SBenjamin Kramer       std::string regex(std::string(regex_sed.substr(
959adcd0268SBenjamin Kramer           first_separator_char_pos + 1,
960adcd0268SBenjamin Kramer           second_separator_char_pos - first_separator_char_pos - 1)));
961adcd0268SBenjamin Kramer       std::string subst(std::string(regex_sed.substr(
962adcd0268SBenjamin Kramer           second_separator_char_pos + 1,
963adcd0268SBenjamin Kramer           third_separator_char_pos - second_separator_char_pos - 1)));
96443224195SRaphael Isemann       m_regex_cmd_up->AddRegexCommand(regex, subst);
96544d93782SGreg Clayton     }
9660e5e5a79SGreg Clayton     return error;
967de164aaaSGreg Clayton   }
968de164aaaSGreg Clayton 
969b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
970d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
971d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
972d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
973de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
974de164aaaSGreg Clayton       }
975de164aaaSGreg Clayton     }
976de164aaaSGreg Clayton   }
977de164aaaSGreg Clayton 
978de164aaaSGreg Clayton private:
979d5b44036SJonas Devlieghere   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up;
980de164aaaSGreg Clayton 
981b9c1b51eSKate Stone   class CommandOptions : public Options {
982de164aaaSGreg Clayton   public:
983b9c1b51eSKate Stone     CommandOptions() : Options() {}
984de164aaaSGreg Clayton 
9856e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
986de164aaaSGreg Clayton 
98797206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
988b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
98997206d57SZachary Turner       Status error;
9903bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
991de164aaaSGreg Clayton 
992b9c1b51eSKate Stone       switch (short_option) {
993de164aaaSGreg Clayton       case 'h':
994adcd0268SBenjamin Kramer         m_help.assign(std::string(option_arg));
995de164aaaSGreg Clayton         break;
996de164aaaSGreg Clayton       case 's':
997adcd0268SBenjamin Kramer         m_syntax.assign(std::string(option_arg));
998de164aaaSGreg Clayton         break;
999de164aaaSGreg Clayton       default:
100036162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1001de164aaaSGreg Clayton       }
1002de164aaaSGreg Clayton 
1003de164aaaSGreg Clayton       return error;
1004de164aaaSGreg Clayton     }
1005de164aaaSGreg Clayton 
1006b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1007de164aaaSGreg Clayton       m_help.clear();
1008de164aaaSGreg Clayton       m_syntax.clear();
1009de164aaaSGreg Clayton     }
1010de164aaaSGreg Clayton 
10111f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
101270602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
10131f0f5b5bSZachary Turner     }
1014de164aaaSGreg Clayton 
1015daed98e5SShivam Mittal     llvm::StringRef GetHelp() { return m_help; }
10166e3d8e7fSEugene Zelenko 
1017daed98e5SShivam Mittal     llvm::StringRef GetSyntax() { return m_syntax; }
10186e3d8e7fSEugene Zelenko 
1019de164aaaSGreg Clayton   protected:
10206e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
10216e3d8e7fSEugene Zelenko 
1022de164aaaSGreg Clayton     std::string m_help;
1023de164aaaSGreg Clayton     std::string m_syntax;
1024de164aaaSGreg Clayton   };
1025de164aaaSGreg Clayton 
1026b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1027de164aaaSGreg Clayton 
10285a988416SJim Ingham   CommandOptions m_options;
1029de164aaaSGreg Clayton };
1030de164aaaSGreg Clayton 
1031b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1032223383edSEnrico Granata public:
1033b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1034b9c1b51eSKate Stone                               std::string funct, std::string help,
1035b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1036a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_function_name(funct),
1037a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_long(false) {
1038735152e3SEnrico Granata     if (!help.empty())
1039442f6530SZachary Turner       SetHelp(help);
1040b9c1b51eSKate Stone     else {
1041735152e3SEnrico Granata       StreamString stream;
1042735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1043c156427dSZachary Turner       SetHelp(stream.GetString());
1044735152e3SEnrico Granata     }
1045223383edSEnrico Granata   }
1046223383edSEnrico Granata 
10476e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1048223383edSEnrico Granata 
1049b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
10505a988416SJim Ingham 
1051b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
10525a988416SJim Ingham 
1053b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
10545a988416SJim Ingham 
1055442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1056442f6530SZachary Turner     if (m_fetched_help_long)
1057442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1058442f6530SZachary Turner 
10592b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1060442f6530SZachary Turner     if (!scripter)
1061442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1062442f6530SZachary Turner 
1063fac939e9SEnrico Granata     std::string docstring;
1064442f6530SZachary Turner     m_fetched_help_long =
1065442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1066fac939e9SEnrico Granata     if (!docstring.empty())
1067442f6530SZachary Turner       SetHelpLong(docstring);
1068fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1069fac939e9SEnrico Granata   }
1070fac939e9SEnrico Granata 
10715a988416SJim Ingham protected:
10724d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1073b9c1b51eSKate Stone                  CommandReturnObject &result) override {
10742b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1075223383edSEnrico Granata 
107697206d57SZachary Turner     Status error;
1077223383edSEnrico Granata 
107870f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
107970f11f88SJim Ingham 
1080a925974bSAdrian Prantl     if (!scripter || !scripter->RunScriptBasedCommand(
1081a925974bSAdrian Prantl                          m_function_name.c_str(), raw_command_line, m_synchro,
1082a925974bSAdrian Prantl                          result, error, m_exe_ctx)) {
1083223383edSEnrico Granata       result.AppendError(error.AsCString());
1084b9c1b51eSKate Stone     } else {
108570f11f88SJim Ingham       // Don't change the status if the command already set it...
1086b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1087c156427dSZachary Turner         if (result.GetOutputData().empty())
1088223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
108970f11f88SJim Ingham         else
109070f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
109170f11f88SJim Ingham       }
109270f11f88SJim Ingham     }
1093223383edSEnrico Granata 
1094223383edSEnrico Granata     return result.Succeeded();
1095223383edSEnrico Granata   }
1096223383edSEnrico Granata 
10976e3d8e7fSEugene Zelenko private:
10986e3d8e7fSEugene Zelenko   std::string m_function_name;
10996e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11006e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1101223383edSEnrico Granata };
1102223383edSEnrico Granata 
1103b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
11049fe00e52SEnrico Granata public:
11059fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
11069fe00e52SEnrico Granata                                std::string name,
11070641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1108b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1109a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp),
1110a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_short(false),
1111b9c1b51eSKate Stone         m_fetched_help_long(false) {
11129fe00e52SEnrico Granata     StreamString stream;
11139fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1114c156427dSZachary Turner     SetHelp(stream.GetString());
11152b29b432SJonas Devlieghere     if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter())
1116e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
11179fe00e52SEnrico Granata   }
11189fe00e52SEnrico Granata 
11196e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
11209fe00e52SEnrico Granata 
1121b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
11229fe00e52SEnrico Granata 
1123b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
11249fe00e52SEnrico Granata 
1125442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1126442f6530SZachary Turner     if (m_fetched_help_short)
1127442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11282b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1129442f6530SZachary Turner     if (!scripter)
1130442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11316f79bb2dSEnrico Granata     std::string docstring;
1132b9c1b51eSKate Stone     m_fetched_help_short =
1133b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
11346f79bb2dSEnrico Granata     if (!docstring.empty())
1135442f6530SZachary Turner       SetHelp(docstring);
1136442f6530SZachary Turner 
11376f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
11386f79bb2dSEnrico Granata   }
11396f79bb2dSEnrico Granata 
1140442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1141442f6530SZachary Turner     if (m_fetched_help_long)
1142442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1143442f6530SZachary Turner 
11442b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1145442f6530SZachary Turner     if (!scripter)
1146442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1147442f6530SZachary Turner 
11486f79bb2dSEnrico Granata     std::string docstring;
1149b9c1b51eSKate Stone     m_fetched_help_long =
1150b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
11516f79bb2dSEnrico Granata     if (!docstring.empty())
1152442f6530SZachary Turner       SetHelpLong(docstring);
11539fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
11549fe00e52SEnrico Granata   }
11559fe00e52SEnrico Granata 
11569fe00e52SEnrico Granata protected:
11574d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1158b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11592b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
11609fe00e52SEnrico Granata 
116197206d57SZachary Turner     Status error;
11629fe00e52SEnrico Granata 
11639fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
11649fe00e52SEnrico Granata 
1165b9c1b51eSKate Stone     if (!scripter ||
1166b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1167b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
11689fe00e52SEnrico Granata       result.AppendError(error.AsCString());
1169b9c1b51eSKate Stone     } else {
11709fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1171b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1172c156427dSZachary Turner         if (result.GetOutputData().empty())
11739fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
11749fe00e52SEnrico Granata         else
11759fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
11769fe00e52SEnrico Granata       }
11779fe00e52SEnrico Granata     }
11789fe00e52SEnrico Granata 
11799fe00e52SEnrico Granata     return result.Succeeded();
11809fe00e52SEnrico Granata   }
11819fe00e52SEnrico Granata 
11826e3d8e7fSEugene Zelenko private:
11836e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
11846e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11856e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
11866e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
11879fe00e52SEnrico Granata };
11889fe00e52SEnrico Granata 
1189a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
119064becc11SRaphael Isemann #define LLDB_OPTIONS_script_import
119164becc11SRaphael Isemann #include "CommandOptions.inc"
11921f0f5b5bSZachary Turner 
1193b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
11945a988416SJim Ingham public:
1195b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1196b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1197b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1198b9c1b51eSKate Stone         m_options() {
11995a988416SJim Ingham     CommandArgumentEntry arg1;
12005a988416SJim Ingham     CommandArgumentData cmd_arg;
12015a988416SJim Ingham 
12025a988416SJim Ingham     // Define the first (and only) variant of this arg.
12035a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
12043b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
12055a988416SJim Ingham 
1206b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1207b9c1b51eSKate Stone     // argument entry.
12085a988416SJim Ingham     arg1.push_back(cmd_arg);
12095a988416SJim Ingham 
12105a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
12115a988416SJim Ingham     m_arguments.push_back(arg1);
12125a988416SJim Ingham   }
12135a988416SJim Ingham 
12146e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
12155a988416SJim Ingham 
1216ae34ed2cSRaphael Isemann   void
1217ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
12182443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
1219b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1220b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1221a2e76c0bSRaphael Isemann         request, nullptr);
12225a988416SJim Ingham   }
12235a988416SJim Ingham 
1224b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
12255a988416SJim Ingham 
12265a988416SJim Ingham protected:
1227b9c1b51eSKate Stone   class CommandOptions : public Options {
12280a305db7SEnrico Granata   public:
1229b9c1b51eSKate Stone     CommandOptions() : Options() {}
12300a305db7SEnrico Granata 
12316e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
12320a305db7SEnrico Granata 
123397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1234b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
123597206d57SZachary Turner       Status error;
12363bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
12370a305db7SEnrico Granata 
1238b9c1b51eSKate Stone       switch (short_option) {
12390a305db7SEnrico Granata       case 'r':
124015625112SJonas Devlieghere         // NO-OP
12410a305db7SEnrico Granata         break;
124200bb397bSJonas Devlieghere       case 'c':
124300bb397bSJonas Devlieghere         relative_to_command_file = true;
124400bb397bSJonas Devlieghere         break;
1245*f9517353SJonas Devlieghere       case 's':
1246*f9517353SJonas Devlieghere         silent = true;
1247*f9517353SJonas Devlieghere         break;
12480a305db7SEnrico Granata       default:
124936162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
12500a305db7SEnrico Granata       }
12510a305db7SEnrico Granata 
12520a305db7SEnrico Granata       return error;
12530a305db7SEnrico Granata     }
12540a305db7SEnrico Granata 
1255b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
125600bb397bSJonas Devlieghere       relative_to_command_file = false;
12570a305db7SEnrico Granata     }
12580a305db7SEnrico Granata 
12591f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
126070602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
12611f0f5b5bSZachary Turner     }
126200bb397bSJonas Devlieghere     bool relative_to_command_file = false;
1263*f9517353SJonas Devlieghere     bool silent = false;
12640a305db7SEnrico Granata   };
12650a305db7SEnrico Granata 
1266b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
126711eb9c64SZachary Turner     if (command.empty()) {
12683b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1269a9dbf432SEnrico Granata       return false;
1270a9dbf432SEnrico Granata     }
1271a9dbf432SEnrico Granata 
127200bb397bSJonas Devlieghere     FileSpec source_dir = {};
127300bb397bSJonas Devlieghere     if (m_options.relative_to_command_file) {
127400bb397bSJonas Devlieghere       source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir();
127500bb397bSJonas Devlieghere       if (!source_dir) {
127600bb397bSJonas Devlieghere         result.AppendError("command script import -c can only be specified "
127700bb397bSJonas Devlieghere                            "from a command file");
127800bb397bSJonas Devlieghere         return false;
127900bb397bSJonas Devlieghere       }
128000bb397bSJonas Devlieghere     }
128100bb397bSJonas Devlieghere 
128211eb9c64SZachary Turner     for (auto &entry : command.entries()) {
128397206d57SZachary Turner       Status error;
1284a9dbf432SEnrico Granata 
1285*f9517353SJonas Devlieghere       LoadScriptOptions options;
1286*f9517353SJonas Devlieghere       options.SetInitSession(true);
1287*f9517353SJonas Devlieghere       options.SetSilent(m_options.silent);
1288*f9517353SJonas Devlieghere 
1289b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
129011eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
129111eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
129205097246SAdrian Prantl       // script imports" in __lldb_init_module the real fix is to have
129305097246SAdrian Prantl       // recursive commands possible with a CommandInvocation object separate
129405097246SAdrian Prantl       // from the CommandObject itself, so that recursive command invocations
129505097246SAdrian Prantl       // won't stomp on each other (wrt to execution contents, options, and
129605097246SAdrian Prantl       // more)
1297078551c7SEnrico Granata       m_exe_ctx.Clear();
12982b29b432SJonas Devlieghere       if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule(
1299*f9517353SJonas Devlieghere               entry.c_str(), options, error, /*module_sp=*/nullptr,
1300*f9517353SJonas Devlieghere               source_dir)) {
1301a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1302b9c1b51eSKate Stone       } else {
1303b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1304b9c1b51eSKate Stone                                      error.AsCString());
1305a9dbf432SEnrico Granata       }
13063b00e35bSEnrico Granata     }
1307a9dbf432SEnrico Granata 
1308a9dbf432SEnrico Granata     return result.Succeeded();
1309a9dbf432SEnrico Granata   }
13100a305db7SEnrico Granata 
13115a988416SJim Ingham   CommandOptions m_options;
1312a9dbf432SEnrico Granata };
1313223383edSEnrico Granata 
1314223383edSEnrico Granata // CommandObjectCommandsScriptAdd
13158fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = {
1316e063ecccSJonas Devlieghere     {
1317e063ecccSJonas Devlieghere         eScriptedCommandSynchronicitySynchronous,
1318e063ecccSJonas Devlieghere         "synchronous",
1319e063ecccSJonas Devlieghere         "Run synchronous",
1320e063ecccSJonas Devlieghere     },
1321e063ecccSJonas Devlieghere     {
1322e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityAsynchronous,
1323e063ecccSJonas Devlieghere         "asynchronous",
1324e063ecccSJonas Devlieghere         "Run asynchronous",
1325e063ecccSJonas Devlieghere     },
1326e063ecccSJonas Devlieghere     {
1327e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityCurrentValue,
1328e063ecccSJonas Devlieghere         "current",
1329e063ecccSJonas Devlieghere         "Do not alter current setting",
1330e063ecccSJonas Devlieghere     },
1331e063ecccSJonas Devlieghere };
13321f0f5b5bSZachary Turner 
13338fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() {
13348fe53c49STatyana Krasnukha   return OptionEnumValues(g_script_synchro_type);
13358fe53c49STatyana Krasnukha }
13368fe53c49STatyana Krasnukha 
133764becc11SRaphael Isemann #define LLDB_OPTIONS_script_add
133864becc11SRaphael Isemann #include "CommandOptions.inc"
13391f0f5b5bSZachary Turner 
1340b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1341b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
13425a988416SJim Ingham public:
1343b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1344b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
13455a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
13466e3d8e7fSEugene Zelenko                             nullptr),
1347b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
13485a988416SJim Ingham     CommandArgumentEntry arg1;
13495a988416SJim Ingham     CommandArgumentData cmd_arg;
13505a988416SJim Ingham 
13515a988416SJim Ingham     // Define the first (and only) variant of this arg.
13525a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
13535a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
13545a988416SJim Ingham 
1355b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1356b9c1b51eSKate Stone     // argument entry.
13575a988416SJim Ingham     arg1.push_back(cmd_arg);
13585a988416SJim Ingham 
13595a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
13605a988416SJim Ingham     m_arguments.push_back(arg1);
13615a988416SJim Ingham   }
13625a988416SJim Ingham 
13636e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
13645a988416SJim Ingham 
1365b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13665a988416SJim Ingham 
13675a988416SJim Ingham protected:
1368b9c1b51eSKate Stone   class CommandOptions : public Options {
1369223383edSEnrico Granata   public:
1370b9c1b51eSKate Stone     CommandOptions()
13719494c510SJonas Devlieghere         : Options(), m_class_name(), m_funct_name(), m_short_help() {}
1372223383edSEnrico Granata 
13736e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1374223383edSEnrico Granata 
137597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1376b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
137797206d57SZachary Turner       Status error;
13783bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1379223383edSEnrico Granata 
1380b9c1b51eSKate Stone       switch (short_option) {
1381223383edSEnrico Granata       case 'f':
1382fe11483bSZachary Turner         if (!option_arg.empty())
1383adcd0268SBenjamin Kramer           m_funct_name = std::string(option_arg);
1384735152e3SEnrico Granata         break;
13859fe00e52SEnrico Granata       case 'c':
1386fe11483bSZachary Turner         if (!option_arg.empty())
1387adcd0268SBenjamin Kramer           m_class_name = std::string(option_arg);
13889fe00e52SEnrico Granata         break;
1389735152e3SEnrico Granata       case 'h':
1390fe11483bSZachary Turner         if (!option_arg.empty())
1391adcd0268SBenjamin Kramer           m_short_help = std::string(option_arg);
1392223383edSEnrico Granata         break;
13930a305db7SEnrico Granata       case 's':
1394b9c1b51eSKate Stone         m_synchronicity =
139547cbf4a0SPavel Labath             (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum(
1396fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
13970a305db7SEnrico Granata         if (!error.Success())
1398b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1399fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1400fe11483bSZachary Turner               option_arg.str().c_str());
14010a305db7SEnrico Granata         break;
1402223383edSEnrico Granata       default:
140336162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1404223383edSEnrico Granata       }
1405223383edSEnrico Granata 
1406223383edSEnrico Granata       return error;
1407223383edSEnrico Granata     }
1408223383edSEnrico Granata 
1409b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
14109fe00e52SEnrico Granata       m_class_name.clear();
1411735152e3SEnrico Granata       m_funct_name.clear();
1412735152e3SEnrico Granata       m_short_help.clear();
141344d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1414223383edSEnrico Granata     }
1415223383edSEnrico Granata 
14161f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
141770602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
14181f0f5b5bSZachary Turner     }
1419223383edSEnrico Granata 
1420223383edSEnrico Granata     // Instance variables to hold the values for command options.
1421223383edSEnrico Granata 
14229fe00e52SEnrico Granata     std::string m_class_name;
1423223383edSEnrico Granata     std::string m_funct_name;
1424735152e3SEnrico Granata     std::string m_short_help;
14259494c510SJonas Devlieghere     ScriptedCommandSynchronicity m_synchronicity =
14269494c510SJonas Devlieghere         eScriptedCommandSynchronicitySynchronous;
1427223383edSEnrico Granata   };
1428223383edSEnrico Granata 
14290affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
14307ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
14310affb582SDave Lee     if (output_sp && interactive) {
143244d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
143344d93782SGreg Clayton       output_sp->Flush();
1434223383edSEnrico Granata     }
1435223383edSEnrico Granata   }
1436223383edSEnrico Granata 
1437b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1438b9c1b51eSKate Stone                               std::string &data) override {
14397ca15ba7SLawrence D'Anna     StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
144044d93782SGreg Clayton 
14412b29b432SJonas Devlieghere     ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1442b9c1b51eSKate Stone     if (interpreter) {
144344d93782SGreg Clayton 
144444d93782SGreg Clayton       StringList lines;
144544d93782SGreg Clayton       lines.SplitIntoLines(data);
1446b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1447a73b7df7SEnrico Granata         std::string funct_name_str;
1448b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1449b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1450b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1451b9c1b51eSKate Stone                              "add python command.\n");
145244d93782SGreg Clayton             error_sp->Flush();
1453b9c1b51eSKate Stone           } else {
1454223383edSEnrico Granata             // everything should be fine now, let's add this alias
1455223383edSEnrico Granata 
1456b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1457771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
145844d93782SGreg Clayton                 m_synchronicity));
1459223383edSEnrico Granata 
1460b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1461b9c1b51eSKate Stone                                               true)) {
1462b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1463b9c1b51eSKate Stone                                "add python command.\n");
146444d93782SGreg Clayton               error_sp->Flush();
1465223383edSEnrico Granata             }
1466223383edSEnrico Granata           }
1467b9c1b51eSKate Stone         } else {
1468b9c1b51eSKate Stone           error_sp->Printf(
1469b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
147044d93782SGreg Clayton           error_sp->Flush();
147144d93782SGreg Clayton         }
1472b9c1b51eSKate Stone       } else {
147344d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
147444d93782SGreg Clayton         error_sp->Flush();
147544d93782SGreg Clayton       }
1476b9c1b51eSKate Stone     } else {
1477b9c1b51eSKate Stone       error_sp->Printf(
1478b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
147944d93782SGreg Clayton       error_sp->Flush();
148044d93782SGreg Clayton     }
148144d93782SGreg Clayton 
148244d93782SGreg Clayton     io_handler.SetIsDone(true);
148344d93782SGreg Clayton   }
1484223383edSEnrico Granata 
1485b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
148657179860SJonas Devlieghere     if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) {
1487b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1488b9c1b51eSKate Stone                          "commands is currently Python");
148999f0b8f9SEnrico Granata       return false;
149099f0b8f9SEnrico Granata     }
149199f0b8f9SEnrico Granata 
149211eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1493223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1494223383edSEnrico Granata       return false;
1495223383edSEnrico Granata     }
1496223383edSEnrico Granata 
1497735152e3SEnrico Granata     // Store the options in case we get multi-line input
1498adcd0268SBenjamin Kramer     m_cmd_name = std::string(command[0].ref());
1499735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
150044d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1501223383edSEnrico Granata 
1502b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1503b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1504b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1505b9c1b51eSKate Stone             "     ", // Prompt
1506a6faf851SJonas Devlieghere             *this);  // IOHandlerDelegate
1507b9c1b51eSKate Stone       } else {
1508b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1509b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1510b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1511b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1512223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1513b9c1b51eSKate Stone         } else {
1514223383edSEnrico Granata           result.AppendError("cannot add command");
1515223383edSEnrico Granata         }
1516223383edSEnrico Granata       }
1517b9c1b51eSKate Stone     } else {
15182b29b432SJonas Devlieghere       ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1519b9c1b51eSKate Stone       if (!interpreter) {
15209fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
15219fe00e52SEnrico Granata         return false;
15229fe00e52SEnrico Granata       }
15239fe00e52SEnrico Granata 
1524b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1525b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1526b9c1b51eSKate Stone       if (!cmd_obj_sp) {
15279fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
15289fe00e52SEnrico Granata         return false;
15299fe00e52SEnrico Granata       }
15309fe00e52SEnrico Granata 
1531b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1532b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1533b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
15349fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1535b9c1b51eSKate Stone       } else {
15369fe00e52SEnrico Granata         result.AppendError("cannot add command");
15379fe00e52SEnrico Granata       }
15389fe00e52SEnrico Granata     }
1539223383edSEnrico Granata 
1540223383edSEnrico Granata     return result.Succeeded();
1541223383edSEnrico Granata   }
15425a988416SJim Ingham 
15435a988416SJim Ingham   CommandOptions m_options;
154444d93782SGreg Clayton   std::string m_cmd_name;
1545735152e3SEnrico Granata   std::string m_short_help;
154644d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1547223383edSEnrico Granata };
1548223383edSEnrico Granata 
1549223383edSEnrico Granata // CommandObjectCommandsScriptList
1550223383edSEnrico Granata 
1551b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1552223383edSEnrico Granata public:
1553b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1554b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1555b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1556223383edSEnrico Granata 
15576e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1558223383edSEnrico Granata 
1559b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1560d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1561d77ea5b2SRaphael Isemann       result.AppendError("'command script list' doesn't take any arguments");
1562d77ea5b2SRaphael Isemann       return false;
1563d77ea5b2SRaphael Isemann     }
1564d77ea5b2SRaphael Isemann 
1565b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1566223383edSEnrico Granata 
1567223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1568223383edSEnrico Granata 
1569223383edSEnrico Granata     return true;
1570223383edSEnrico Granata   }
1571223383edSEnrico Granata };
1572223383edSEnrico Granata 
1573223383edSEnrico Granata // CommandObjectCommandsScriptClear
1574223383edSEnrico Granata 
1575b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1576223383edSEnrico Granata public:
1577b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1578b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1579b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1580223383edSEnrico Granata 
15816e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1582223383edSEnrico Granata 
15835a988416SJim Ingham protected:
1584b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1585d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1586d77ea5b2SRaphael Isemann       result.AppendError("'command script clear' doesn't take any arguments");
1587d77ea5b2SRaphael Isemann       return false;
1588d77ea5b2SRaphael Isemann     }
1589d77ea5b2SRaphael Isemann 
1590223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1591223383edSEnrico Granata 
1592223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1593223383edSEnrico Granata 
1594223383edSEnrico Granata     return true;
1595223383edSEnrico Granata   }
1596223383edSEnrico Granata };
1597223383edSEnrico Granata 
1598223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1599223383edSEnrico Granata 
1600b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1601223383edSEnrico Granata public:
1602b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1603b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1604b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1605223383edSEnrico Granata     CommandArgumentEntry arg1;
1606223383edSEnrico Granata     CommandArgumentData cmd_arg;
1607223383edSEnrico Granata 
1608223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1609223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1610223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1611223383edSEnrico Granata 
1612b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1613b9c1b51eSKate Stone     // argument entry.
1614223383edSEnrico Granata     arg1.push_back(cmd_arg);
1615223383edSEnrico Granata 
1616223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1617223383edSEnrico Granata     m_arguments.push_back(arg1);
1618223383edSEnrico Granata   }
1619223383edSEnrico Granata 
16206e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1621223383edSEnrico Granata 
16222e8f304fSGongyu Deng   void
16232e8f304fSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
16242e8f304fSGongyu Deng                            OptionElementVector &opt_element_vector) override {
16252e8f304fSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
16262e8f304fSGongyu Deng       return;
16272e8f304fSGongyu Deng 
16282ebe30c6SRaphael Isemann     for (const auto &c : m_interpreter.GetUserCommands())
16292ebe30c6SRaphael Isemann       request.TryCompleteCurrentArg(c.first, c.second->GetHelp());
16302e8f304fSGongyu Deng   }
16312e8f304fSGongyu Deng 
16325a988416SJim Ingham protected:
1633b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1634223383edSEnrico Granata 
163511eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1636223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1637223383edSEnrico Granata       return false;
1638223383edSEnrico Granata     }
1639223383edSEnrico Granata 
16400d9a201eSRaphael Isemann     auto cmd_name = command[0].ref();
1641223383edSEnrico Granata 
16424574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
16434574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1644867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
16454574a890SZachary Turner       return false;
1646223383edSEnrico Granata     }
1647223383edSEnrico Granata 
16484574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
16494574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
16504574a890SZachary Turner     return true;
1651223383edSEnrico Granata   }
1652223383edSEnrico Granata };
1653223383edSEnrico Granata 
1654223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1655223383edSEnrico Granata 
1656223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1657223383edSEnrico Granata 
1658b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1659223383edSEnrico Granata public:
16607428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1661b9c1b51eSKate Stone       : CommandObjectMultiword(
1662a925974bSAdrian Prantl             interpreter, "command script",
1663a925974bSAdrian Prantl             "Commands for managing custom "
1664b9c1b51eSKate Stone             "commands implemented by "
1665b9c1b51eSKate Stone             "interpreter scripts.",
1666b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1667b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1668b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1669b9c1b51eSKate Stone     LoadSubCommand(
1670b9c1b51eSKate Stone         "delete",
1671b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1672b9c1b51eSKate Stone     LoadSubCommand(
1673b9c1b51eSKate Stone         "clear",
1674b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1675b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1676b9c1b51eSKate Stone                                interpreter)));
1677b9c1b51eSKate Stone     LoadSubCommand(
1678b9c1b51eSKate Stone         "import",
1679b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1680223383edSEnrico Granata   }
1681223383edSEnrico Granata 
16826e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1683223383edSEnrico Granata };
1684223383edSEnrico Granata 
1685ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1686ebc09c36SJim Ingham 
1687ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1688ebc09c36SJim Ingham 
1689b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1690b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1691b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1692b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1693b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1694b9c1b51eSKate Stone   LoadSubCommand("source",
1695b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1696b9c1b51eSKate Stone   LoadSubCommand("alias",
1697b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1698b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1699b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1700b9c1b51eSKate Stone   LoadSubCommand("delete",
1701b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1702b9c1b51eSKate Stone   LoadSubCommand(
1703b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1704b9c1b51eSKate Stone   LoadSubCommand(
1705b9c1b51eSKate Stone       "script",
1706b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1707ebc09c36SJim Ingham }
1708ebc09c36SJim Ingham 
17096e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1710