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),
80*3bf3b966SJim Ingham           m_stop_on_continue(true), m_cmd_relative_to_command_file(false) {}
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 
98*3bf3b966SJim Ingham       case 'C':
99*3bf3b966SJim Ingham         m_cmd_relative_to_command_file = true;
100*3bf3b966SJim Ingham         break;
101*3bf3b966SJim Ingham 
10260986174SMichael Sartain       case 's':
103fe11483bSZachary Turner         error = m_silent_run.SetValueFromString(option_arg);
10460986174SMichael Sartain         break;
105340b0309SGreg Clayton 
106e16c50a1SJim Ingham       default:
10736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
108e16c50a1SJim Ingham       }
109e16c50a1SJim Ingham 
110e16c50a1SJim Ingham       return error;
111e16c50a1SJim Ingham     }
112e16c50a1SJim Ingham 
113b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
114012d4fcaSEnrico Granata       m_stop_on_error.Clear();
115340b0309SGreg Clayton       m_silent_run.Clear();
116340b0309SGreg Clayton       m_stop_on_continue.Clear();
117*3bf3b966SJim Ingham       m_cmd_relative_to_command_file.Clear();
118e16c50a1SJim Ingham     }
119e16c50a1SJim Ingham 
1201f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
12170602439SZachary Turner       return llvm::makeArrayRef(g_source_options);
1221f0f5b5bSZachary Turner     }
123e16c50a1SJim Ingham 
124e16c50a1SJim Ingham     // Instance variables to hold the values for command options.
125e16c50a1SJim Ingham 
126012d4fcaSEnrico Granata     OptionValueBoolean m_stop_on_error;
127340b0309SGreg Clayton     OptionValueBoolean m_silent_run;
128340b0309SGreg Clayton     OptionValueBoolean m_stop_on_continue;
129*3bf3b966SJim Ingham     OptionValueBoolean m_cmd_relative_to_command_file;
130e16c50a1SJim Ingham   };
131e16c50a1SJim Ingham 
132b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1334574a890SZachary Turner     if (command.GetArgumentCount() != 1) {
1344574a890SZachary Turner       result.AppendErrorWithFormat(
1354574a890SZachary Turner           "'%s' takes exactly one executable filename argument.\n",
1364574a890SZachary Turner           GetCommandName().str().c_str());
1374574a890SZachary Turner       return false;
1384574a890SZachary Turner     }
139ebc09c36SJim Ingham 
140*3bf3b966SJim Ingham     FileSpec source_dir = {};
141*3bf3b966SJim Ingham     if (m_options.m_cmd_relative_to_command_file) {
142*3bf3b966SJim Ingham       source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir();
143*3bf3b966SJim Ingham       if (!source_dir) {
144*3bf3b966SJim Ingham         result.AppendError("command source -C can only be specified "
145*3bf3b966SJim Ingham                            "from a command file");
146*3bf3b966SJim Ingham         result.SetStatus(eReturnStatusFailed);
147*3bf3b966SJim Ingham         return false;
148*3bf3b966SJim Ingham       }
149*3bf3b966SJim Ingham     }
150*3bf3b966SJim Ingham 
1510d9a201eSRaphael Isemann     FileSpec cmd_file(command[0].ref());
152*3bf3b966SJim Ingham     if (source_dir) {
153*3bf3b966SJim Ingham       // Prepend the source_dir to the cmd_file path:
154*3bf3b966SJim Ingham       if (!cmd_file.IsRelative()) {
155*3bf3b966SJim Ingham         result.AppendError("command source -C can only be used "
156*3bf3b966SJim Ingham                            "with a relative path.");
157*3bf3b966SJim Ingham         result.SetStatus(eReturnStatusFailed);
158*3bf3b966SJim Ingham         return false;
159*3bf3b966SJim Ingham       }
160*3bf3b966SJim Ingham       cmd_file.MakeAbsolute(source_dir);
161*3bf3b966SJim Ingham     }
162*3bf3b966SJim Ingham 
1638f3be7a3SJonas Devlieghere     FileSystem::Instance().Resolve(cmd_file);
164ebc09c36SJim Ingham 
16536de94cfSTatyana Krasnukha     CommandInterpreterRunOptions options;
166340b0309SGreg Clayton     // If any options were set, then use them
167340b0309SGreg Clayton     if (m_options.m_stop_on_error.OptionWasSet() ||
168340b0309SGreg Clayton         m_options.m_silent_run.OptionWasSet() ||
169b9c1b51eSKate Stone         m_options.m_stop_on_continue.OptionWasSet()) {
1701c19b74cSJonas Devlieghere       if (m_options.m_stop_on_continue.OptionWasSet())
1711c19b74cSJonas Devlieghere         options.SetStopOnContinue(
1721c19b74cSJonas Devlieghere             m_options.m_stop_on_continue.GetCurrentValue());
1731c19b74cSJonas Devlieghere 
1741c19b74cSJonas Devlieghere       if (m_options.m_stop_on_error.OptionWasSet())
17526c7bf93SJim Ingham         options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue());
176c678ed77SStefan Granitz 
177c678ed77SStefan Granitz       // Individual silent setting is override for global command echo settings.
178c678ed77SStefan Granitz       if (m_options.m_silent_run.GetCurrentValue()) {
179c678ed77SStefan Granitz         options.SetSilent(true);
180c678ed77SStefan Granitz       } else {
181c678ed77SStefan Granitz         options.SetPrintResults(true);
182c0b48ab6SJonas Devlieghere         options.SetPrintErrors(true);
183c678ed77SStefan Granitz         options.SetEchoCommands(m_interpreter.GetEchoCommands());
184c678ed77SStefan Granitz         options.SetEchoCommentCommands(m_interpreter.GetEchoCommentCommands());
185c678ed77SStefan Granitz       }
186122a4ebdSPavel Labath     }
18736de94cfSTatyana Krasnukha 
18836de94cfSTatyana Krasnukha     m_interpreter.HandleCommandsFromFile(cmd_file, options, result);
189ebc09c36SJim Ingham     return result.Succeeded();
190ebc09c36SJim Ingham   }
1916e3d8e7fSEugene Zelenko 
1925a988416SJim Ingham   CommandOptions m_options;
193ebc09c36SJim Ingham };
194ebc09c36SJim Ingham 
195ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
196ebc09c36SJim Ingham // CommandObjectCommandsAlias
197ebc09c36SJim Ingham 
19864becc11SRaphael Isemann #define LLDB_OPTIONS_alias
19964becc11SRaphael Isemann #include "CommandOptions.inc"
2001f0f5b5bSZachary Turner 
201b9c1b51eSKate Stone static const char *g_python_command_instructions =
202b9c1b51eSKate Stone     "Enter your Python command(s). Type 'DONE' to end.\n"
203be93a35aSEnrico Granata     "You must define a Python function with this signature:\n"
20444d93782SGreg Clayton     "def my_command_impl(debugger, args, result, internal_dict):\n";
205be93a35aSEnrico Granata 
206b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw {
20745d0e238SEnrico Granata protected:
208b9c1b51eSKate Stone   class CommandOptions : public OptionGroup {
209ebc09c36SJim Ingham   public:
210b9c1b51eSKate Stone     CommandOptions() : OptionGroup(), m_help(), m_long_help() {}
21145d0e238SEnrico Granata 
21245d0e238SEnrico Granata     ~CommandOptions() override = default;
21345d0e238SEnrico Granata 
2141f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
21570602439SZachary Turner       return llvm::makeArrayRef(g_alias_options);
2161f0f5b5bSZachary Turner     }
21745d0e238SEnrico Granata 
21897206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_value,
219b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
22097206d57SZachary Turner       Status error;
22145d0e238SEnrico Granata 
2221f0f5b5bSZachary Turner       const int short_option = GetDefinitions()[option_idx].short_option;
2238cef4b0bSZachary Turner       std::string option_str(option_value);
22445d0e238SEnrico Granata 
225b9c1b51eSKate Stone       switch (short_option) {
22645d0e238SEnrico Granata       case 'h':
2278cef4b0bSZachary Turner         m_help.SetCurrentValue(option_str);
22845d0e238SEnrico Granata         m_help.SetOptionWasSet();
22945d0e238SEnrico Granata         break;
23045d0e238SEnrico Granata 
23145d0e238SEnrico Granata       case 'H':
2328cef4b0bSZachary Turner         m_long_help.SetCurrentValue(option_str);
23345d0e238SEnrico Granata         m_long_help.SetOptionWasSet();
23445d0e238SEnrico Granata         break;
23545d0e238SEnrico Granata 
23645d0e238SEnrico Granata       default:
23736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
23845d0e238SEnrico Granata       }
23945d0e238SEnrico Granata 
24045d0e238SEnrico Granata       return error;
24145d0e238SEnrico Granata     }
24245d0e238SEnrico Granata 
243b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
24445d0e238SEnrico Granata       m_help.Clear();
24545d0e238SEnrico Granata       m_long_help.Clear();
24645d0e238SEnrico Granata     }
24745d0e238SEnrico Granata 
24845d0e238SEnrico Granata     OptionValueString m_help;
24945d0e238SEnrico Granata     OptionValueString m_long_help;
25045d0e238SEnrico Granata   };
25145d0e238SEnrico Granata 
25245d0e238SEnrico Granata   OptionGroupOptions m_option_group;
25345d0e238SEnrico Granata   CommandOptions m_command_options;
25445d0e238SEnrico Granata 
25545d0e238SEnrico Granata public:
256b9c1b51eSKate Stone   Options *GetOptions() override { return &m_option_group; }
25745d0e238SEnrico Granata 
2587428a18cSKate Stone   CommandObjectCommandsAlias(CommandInterpreter &interpreter)
259b9c1b51eSKate Stone       : CommandObjectRaw(
260b9c1b51eSKate Stone             interpreter, "command alias",
261a449698cSZachary Turner             "Define a custom command in terms of an existing command."),
262b9c1b51eSKate Stone         m_option_group(), m_command_options() {
26345d0e238SEnrico Granata     m_option_group.Append(&m_command_options);
26445d0e238SEnrico Granata     m_option_group.Finalize();
26545d0e238SEnrico Granata 
266ebc09c36SJim Ingham     SetHelpLong(
267ea671fbdSKate Stone         "'alias' allows the user to create a short-cut or abbreviation for long \
268ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
269b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:"
270b9c1b51eSKate Stone         R"(
271ea671fbdSKate Stone 
272ea671fbdSKate Stone (lldb) command alias sc script
273ea671fbdSKate Stone 
274ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
275ea671fbdSKate Stone 
276ea671fbdSKate Stone (lldb) command alias bp breakpoint
277ea671fbdSKate Stone 
278b9c1b51eSKate Stone )"
279b9c1b51eSKate Stone         "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
280ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
281b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'."
282b9c1b51eSKate Stone         R"(
283ea671fbdSKate Stone 
284ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
285ea671fbdSKate Stone 
286ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
287ea671fbdSKate Stone 
288b9c1b51eSKate Stone )"
289b9c1b51eSKate Stone         "An alias can include some options for the command, with the values either \
290ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
291ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
292b9c1b51eSKate Stone shows how to create aliases with options:"
293b9c1b51eSKate Stone         R"(
294ea671fbdSKate Stone 
295ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
296ea671fbdSKate Stone 
297b9c1b51eSKate Stone )"
298b9c1b51eSKate Stone         "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
299ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
300ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
301ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
302ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
303ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
304ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
305ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
306ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
307ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
308b9c1b51eSKate Stone launch' example below)."
309b9c1b51eSKate Stone         R"(
310ea671fbdSKate Stone 
311b9c1b51eSKate Stone )"
312b9c1b51eSKate Stone         "Note: the positional arguments must substitute as whole words in the resultant \
313ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
314b9c1b51eSKate Stone \".cpp\":"
315b9c1b51eSKate Stone         R"(
316ea671fbdSKate Stone 
317ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
318ea671fbdSKate Stone 
319b9c1b51eSKate Stone )"
320b9c1b51eSKate Stone         "For more complex aliasing, use the \"command regex\" command instead.  In the \
321ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
322ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
323b9c1b51eSKate Stone argument.  The user would use this alias as follows:"
324b9c1b51eSKate Stone         R"(
325ea671fbdSKate Stone 
326ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
327ea671fbdSKate Stone (lldb) bfl my-file.c 137
328ea671fbdSKate Stone 
329ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
330ea671fbdSKate Stone 
331ea671fbdSKate Stone Another example:
332ea671fbdSKate Stone 
333ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
334ea671fbdSKate Stone (lldb) pltty /dev/tty0
335ea671fbdSKate Stone 
336ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
337ea671fbdSKate Stone 
338b9c1b51eSKate Stone )"
339b9c1b51eSKate Stone         "If the user always wanted to pass the same value to a particular option, the \
340ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
341b9c1b51eSKate Stone rather than using a positional placeholder:"
342b9c1b51eSKate Stone         R"(
343ea671fbdSKate Stone 
344ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
345ea671fbdSKate Stone 
346b9c1b51eSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)");
347ebc09c36SJim Ingham 
348405fe67fSCaroline Tice     CommandArgumentEntry arg1;
349405fe67fSCaroline Tice     CommandArgumentEntry arg2;
350405fe67fSCaroline Tice     CommandArgumentEntry arg3;
351405fe67fSCaroline Tice     CommandArgumentData alias_arg;
352405fe67fSCaroline Tice     CommandArgumentData cmd_arg;
353405fe67fSCaroline Tice     CommandArgumentData options_arg;
354405fe67fSCaroline Tice 
355405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
356405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
357405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
358405fe67fSCaroline Tice 
359b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
360b9c1b51eSKate Stone     // argument entry.
361405fe67fSCaroline Tice     arg1.push_back(alias_arg);
362405fe67fSCaroline Tice 
363405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
364405fe67fSCaroline Tice     cmd_arg.arg_type = eArgTypeCommandName;
365405fe67fSCaroline Tice     cmd_arg.arg_repetition = eArgRepeatPlain;
366405fe67fSCaroline Tice 
367b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
368b9c1b51eSKate Stone     // argument entry.
369405fe67fSCaroline Tice     arg2.push_back(cmd_arg);
370405fe67fSCaroline Tice 
371405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
372405fe67fSCaroline Tice     options_arg.arg_type = eArgTypeAliasOptions;
373405fe67fSCaroline Tice     options_arg.arg_repetition = eArgRepeatOptional;
374405fe67fSCaroline Tice 
375b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
376b9c1b51eSKate Stone     // argument entry.
377405fe67fSCaroline Tice     arg3.push_back(options_arg);
378405fe67fSCaroline Tice 
379405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
380405fe67fSCaroline Tice     m_arguments.push_back(arg1);
381405fe67fSCaroline Tice     m_arguments.push_back(arg2);
382405fe67fSCaroline Tice     m_arguments.push_back(arg3);
383ebc09c36SJim Ingham   }
384ebc09c36SJim Ingham 
3856e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAlias() override = default;
386ebc09c36SJim Ingham 
3875a988416SJim Ingham protected:
3884d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
389b9c1b51eSKate Stone                  CommandReturnObject &result) override {
3904d51a902SRaphael Isemann     if (raw_command_line.empty()) {
391d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
39245d0e238SEnrico Granata       return false;
39345d0e238SEnrico Granata     }
39445d0e238SEnrico Granata 
395e1cfbc79STodd Fiala     ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext();
396e1cfbc79STodd Fiala     m_option_group.NotifyOptionParsingStarting(&exe_ctx);
39745d0e238SEnrico Granata 
3983a0e1270SRaphael Isemann     OptionsWithRaw args_with_suffix(raw_command_line);
39945d0e238SEnrico Granata 
4003a0e1270SRaphael Isemann     if (args_with_suffix.HasArgs())
4013a0e1270SRaphael Isemann       if (!ParseOptionsAndNotify(args_with_suffix.GetArgs(), result,
4023a0e1270SRaphael Isemann                                  m_option_group, exe_ctx))
40345d0e238SEnrico Granata         return false;
40445d0e238SEnrico Granata 
405daed98e5SShivam Mittal     llvm::StringRef raw_command_string = args_with_suffix.GetRawPart();
406a01bccdbSZachary Turner     Args args(raw_command_string);
407844d2303SCaroline Tice 
40811eb9c64SZachary Turner     if (args.GetArgumentCount() < 2) {
409d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
410844d2303SCaroline Tice       return false;
411844d2303SCaroline Tice     }
412844d2303SCaroline Tice 
413844d2303SCaroline Tice     // Get the alias command.
414844d2303SCaroline Tice 
4150d9a201eSRaphael Isemann     auto alias_command = args[0].ref();
4164574a890SZachary Turner     if (alias_command.startswith("-")) {
417d72e412fSEnrico Granata       result.AppendError("aliases starting with a dash are not supported");
418b9c1b51eSKate Stone       if (alias_command == "--help" || alias_command == "--long-help") {
419b9c1b51eSKate Stone         result.AppendWarning("if trying to pass options to 'command alias' add "
420b9c1b51eSKate Stone                              "a -- at the end of the options");
421d72e412fSEnrico Granata       }
422d72e412fSEnrico Granata       return false;
423d72e412fSEnrico Granata     }
424844d2303SCaroline Tice 
425b9c1b51eSKate Stone     // Strip the new alias name off 'raw_command_string'  (leave it on args,
42605097246SAdrian Prantl     // which gets passed to 'Execute', which does the stripping itself.
427844d2303SCaroline Tice     size_t pos = raw_command_string.find(alias_command);
428b9c1b51eSKate Stone     if (pos == 0) {
429844d2303SCaroline Tice       raw_command_string = raw_command_string.substr(alias_command.size());
430844d2303SCaroline Tice       pos = raw_command_string.find_first_not_of(' ');
431844d2303SCaroline Tice       if ((pos != std::string::npos) && (pos > 0))
432844d2303SCaroline Tice         raw_command_string = raw_command_string.substr(pos);
433b9c1b51eSKate Stone     } else {
434844d2303SCaroline Tice       result.AppendError("Error parsing command string.  No alias created.");
435844d2303SCaroline Tice       return false;
436844d2303SCaroline Tice     }
437844d2303SCaroline Tice 
438844d2303SCaroline Tice     // Verify that the command is alias-able.
439771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
440b9c1b51eSKate Stone       result.AppendErrorWithFormat(
441b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
4424574a890SZachary Turner           args[0].c_str());
443844d2303SCaroline Tice       return false;
444844d2303SCaroline Tice     }
445844d2303SCaroline Tice 
446b9c1b51eSKate Stone     // Get CommandObject that is being aliased. The command name is read from
447a01bccdbSZachary Turner     // the front of raw_command_string. raw_command_string is returned with the
448a01bccdbSZachary Turner     // name of the command object stripped off the front.
449a01bccdbSZachary Turner     llvm::StringRef original_raw_command_string = raw_command_string;
450b9c1b51eSKate Stone     CommandObject *cmd_obj =
451b9c1b51eSKate Stone         m_interpreter.GetCommandObjectForCommand(raw_command_string);
452844d2303SCaroline Tice 
453b9c1b51eSKate Stone     if (!cmd_obj) {
454b9c1b51eSKate Stone       result.AppendErrorWithFormat("invalid command given to 'command alias'. "
455b9c1b51eSKate Stone                                    "'%s' does not begin with a valid command."
456b9c1b51eSKate Stone                                    "  No alias created.",
457a01bccdbSZachary Turner                                    original_raw_command_string.str().c_str());
458844d2303SCaroline Tice       return false;
459b9c1b51eSKate Stone     } else if (!cmd_obj->WantsRawCommandString()) {
460b9c1b51eSKate Stone       // Note that args was initialized with the original command, and has not
46105097246SAdrian Prantl       // been updated to this point. Therefore can we pass it to the version of
46205097246SAdrian Prantl       // Execute that does not need/expect raw input in the alias.
4635a988416SJim Ingham       return HandleAliasingNormalCommand(args, result);
464b9c1b51eSKate Stone     } else {
465b9c1b51eSKate Stone       return HandleAliasingRawCommand(alias_command, raw_command_string,
466b9c1b51eSKate Stone                                       *cmd_obj, result);
4675a988416SJim Ingham     }
4685a988416SJim Ingham     return result.Succeeded();
4695a988416SJim Ingham   }
4705a988416SJim Ingham 
471a01bccdbSZachary Turner   bool HandleAliasingRawCommand(llvm::StringRef alias_command,
472a01bccdbSZachary Turner                                 llvm::StringRef raw_command_string,
473b9c1b51eSKate Stone                                 CommandObject &cmd_obj,
474b9c1b51eSKate Stone                                 CommandReturnObject &result) {
475844d2303SCaroline Tice     // Verify & handle any options/arguments passed to the alias command
476844d2303SCaroline Tice 
477b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
478b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
479844d2303SCaroline Tice 
480b9c1b51eSKate Stone     if (CommandObjectSP cmd_obj_sp =
481a9448872SJonas Devlieghere             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName())) {
482a01bccdbSZachary Turner       if (m_interpreter.AliasExists(alias_command) ||
483a01bccdbSZachary Turner           m_interpreter.UserCommandExists(alias_command)) {
484b9c1b51eSKate Stone         result.AppendWarningWithFormat(
485b9c1b51eSKate Stone             "Overwriting existing definition for '%s'.\n",
486a01bccdbSZachary Turner             alias_command.str().c_str());
487844d2303SCaroline Tice       }
488b9c1b51eSKate Stone       if (CommandAlias *alias = m_interpreter.AddAlias(
489a01bccdbSZachary Turner               alias_command, cmd_obj_sp, raw_command_string)) {
49045d0e238SEnrico Granata         if (m_command_options.m_help.OptionWasSet())
49145d0e238SEnrico Granata           alias->SetHelp(m_command_options.m_help.GetCurrentValue());
49245d0e238SEnrico Granata         if (m_command_options.m_long_help.OptionWasSet())
49345d0e238SEnrico Granata           alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
494844d2303SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
495b9c1b51eSKate Stone       } else {
496472362e6SCaroline Tice         result.AppendError("Unable to create requested alias.\n");
497472362e6SCaroline Tice       }
498212130acSEnrico Granata 
499b9c1b51eSKate Stone     } else {
500212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
501212130acSEnrico Granata     }
502212130acSEnrico Granata 
503844d2303SCaroline Tice     return result.Succeeded();
504844d2303SCaroline Tice   }
505ebc09c36SJim Ingham 
506b9c1b51eSKate Stone   bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) {
507867b185dSCaroline Tice     size_t argc = args.GetArgumentCount();
508ebc09c36SJim Ingham 
509b9c1b51eSKate Stone     if (argc < 2) {
510d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
511ebc09c36SJim Ingham       return false;
512ebc09c36SJim Ingham     }
513ebc09c36SJim Ingham 
5144574a890SZachary Turner     // Save these in std::strings since we're going to shift them off.
515adcd0268SBenjamin Kramer     const std::string alias_command(std::string(args[0].ref()));
516adcd0268SBenjamin Kramer     const std::string actual_command(std::string(args[1].ref()));
517ebc09c36SJim Ingham 
518ebc09c36SJim Ingham     args.Shift(); // Shift the alias command word off the argument vector.
519ebc09c36SJim Ingham     args.Shift(); // Shift the old command word off the argument vector.
520ebc09c36SJim Ingham 
521b9c1b51eSKate Stone     // Verify that the command is alias'able, and get the appropriate command
522b9c1b51eSKate Stone     // object.
523ebc09c36SJim Ingham 
524771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
525b9c1b51eSKate Stone       result.AppendErrorWithFormat(
526b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
527ebc09c36SJim Ingham           alias_command.c_str());
5284574a890SZachary Turner       return false;
5294574a890SZachary Turner     }
5304574a890SZachary Turner 
531b9c1b51eSKate Stone     CommandObjectSP command_obj_sp(
532a449698cSZachary Turner         m_interpreter.GetCommandSPExact(actual_command, true));
533ebc09c36SJim Ingham     CommandObjectSP subcommand_obj_sp;
534ebc09c36SJim Ingham     bool use_subcommand = false;
5354574a890SZachary Turner     if (!command_obj_sp) {
5364574a890SZachary Turner       result.AppendErrorWithFormat("'%s' is not an existing command.\n",
5374574a890SZachary Turner                                    actual_command.c_str());
5384574a890SZachary Turner       return false;
5394574a890SZachary Turner     }
540ebc09c36SJim Ingham     CommandObject *cmd_obj = command_obj_sp.get();
5416e3d8e7fSEugene Zelenko     CommandObject *sub_cmd_obj = nullptr;
542b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
543b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
544ebc09c36SJim Ingham 
54511eb9c64SZachary Turner     while (cmd_obj->IsMultiwordObject() && !args.empty()) {
5460d9a201eSRaphael Isemann       auto sub_command = args[0].ref();
54711eb9c64SZachary Turner       assert(!sub_command.empty());
5484574a890SZachary Turner       subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command);
5494574a890SZachary Turner       if (!subcommand_obj_sp) {
550b9c1b51eSKate Stone         result.AppendErrorWithFormat(
551b9c1b51eSKate Stone             "'%s' is not a valid sub-command of '%s'.  "
552f415eeb4SCaroline Tice             "Unable to create alias.\n",
5534574a890SZachary Turner             args[0].c_str(), actual_command.c_str());
554ebc09c36SJim Ingham         return false;
555ebc09c36SJim Ingham       }
5564574a890SZachary Turner 
5574574a890SZachary Turner       sub_cmd_obj = subcommand_obj_sp.get();
5584574a890SZachary Turner       use_subcommand = true;
5594574a890SZachary Turner       args.Shift(); // Shift the sub_command word off the argument vector.
5604574a890SZachary Turner       cmd_obj = sub_cmd_obj;
561ebc09c36SJim Ingham     }
562ebc09c36SJim Ingham 
563ebc09c36SJim Ingham     // Verify & handle any options/arguments passed to the alias command
564ebc09c36SJim Ingham 
565212130acSEnrico Granata     std::string args_string;
566212130acSEnrico Granata 
56711eb9c64SZachary Turner     if (!args.empty()) {
568b9c1b51eSKate Stone       CommandObjectSP tmp_sp =
569a9448872SJonas Devlieghere           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName());
570ebc09c36SJim Ingham       if (use_subcommand)
571a9448872SJonas Devlieghere         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName());
572ca90c47eSCaroline Tice 
573ca90c47eSCaroline Tice       args.GetCommandString(args_string);
574867b185dSCaroline Tice     }
575ebc09c36SJim Ingham 
576771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
577771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
578b9c1b51eSKate Stone       result.AppendWarningWithFormat(
5794574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
580ebc09c36SJim Ingham     }
581ebc09c36SJim Ingham 
582b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
5834574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
584771ef6d4SMalcolm Parsons             args_string)) {
58545d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
58645d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
58745d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
58845d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
589ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
590b9c1b51eSKate Stone     } else {
591212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
592212130acSEnrico Granata       return false;
593212130acSEnrico Granata     }
594ebc09c36SJim Ingham 
595ebc09c36SJim Ingham     return result.Succeeded();
596ebc09c36SJim Ingham   }
597ebc09c36SJim Ingham };
598ebc09c36SJim Ingham 
599ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
600ebc09c36SJim Ingham // CommandObjectCommandsUnalias
601ebc09c36SJim Ingham 
602b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
603ebc09c36SJim Ingham public:
6047428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
605b9c1b51eSKate Stone       : CommandObjectParsed(
606b9c1b51eSKate Stone             interpreter, "command unalias",
607b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
608b9c1b51eSKate Stone             nullptr) {
609405fe67fSCaroline Tice     CommandArgumentEntry arg;
610405fe67fSCaroline Tice     CommandArgumentData alias_arg;
611405fe67fSCaroline Tice 
612405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
613405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
614405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
615405fe67fSCaroline Tice 
616b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
617b9c1b51eSKate Stone     // argument entry.
618405fe67fSCaroline Tice     arg.push_back(alias_arg);
619405fe67fSCaroline Tice 
620405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
621405fe67fSCaroline Tice     m_arguments.push_back(arg);
622ebc09c36SJim Ingham   }
623ebc09c36SJim Ingham 
6246e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
625ebc09c36SJim Ingham 
62631fd64acSGongyu Deng   void
62731fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
62831fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
62931fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
63031fd64acSGongyu Deng       return;
63131fd64acSGongyu Deng 
63231fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetAliases()) {
63331fd64acSGongyu Deng       request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
63431fd64acSGongyu Deng     }
63531fd64acSGongyu Deng   }
63631fd64acSGongyu Deng 
6375a988416SJim Ingham protected:
638b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
639ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
640ebc09c36SJim Ingham     CommandObject *cmd_obj;
641ebc09c36SJim Ingham 
64211eb9c64SZachary Turner     if (args.empty()) {
64311eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
64411eb9c64SZachary Turner       return false;
64511eb9c64SZachary Turner     }
64611eb9c64SZachary Turner 
6470d9a201eSRaphael Isemann     auto command_name = args[0].ref();
648a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
6494574a890SZachary Turner     if (!cmd_obj) {
6504574a890SZachary Turner       result.AppendErrorWithFormat(
6514574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
6524574a890SZachary Turner           "current list of commands.\n",
653867e7d17SZachary Turner           args[0].c_str());
6544574a890SZachary Turner       return false;
6554574a890SZachary Turner     }
6564574a890SZachary Turner 
657b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
658b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
659b9c1b51eSKate Stone         result.AppendErrorWithFormat(
660b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
661b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
662867e7d17SZachary Turner             args[0].c_str());
663b9c1b51eSKate Stone       } else {
664b9c1b51eSKate Stone         result.AppendErrorWithFormat(
665b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
666867e7d17SZachary Turner             args[0].c_str());
667b547278cSGreg Clayton       }
6684574a890SZachary Turner       return false;
6694574a890SZachary Turner     }
6704574a890SZachary Turner 
671b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
672a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
673b9c1b51eSKate Stone         result.AppendErrorWithFormat(
674867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
675867e7d17SZachary Turner             args[0].c_str());
676ebc09c36SJim Ingham       else
677b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
678867e7d17SZachary Turner                                      args[0].c_str());
6794574a890SZachary Turner       return false;
680ebc09c36SJim Ingham     }
681ebc09c36SJim Ingham 
6824574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
683ebc09c36SJim Ingham     return result.Succeeded();
684ebc09c36SJim Ingham   }
685ebc09c36SJim Ingham };
686ebc09c36SJim Ingham 
687b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
688b547278cSGreg Clayton // CommandObjectCommandsDelete
689b547278cSGreg Clayton 
690b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
691b547278cSGreg Clayton public:
6927428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
693b9c1b51eSKate Stone       : CommandObjectParsed(
694b9c1b51eSKate Stone             interpreter, "command delete",
695b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
696b9c1b51eSKate Stone             nullptr) {
697b547278cSGreg Clayton     CommandArgumentEntry arg;
698b547278cSGreg Clayton     CommandArgumentData alias_arg;
699b547278cSGreg Clayton 
700b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
701b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
702b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
703b547278cSGreg Clayton 
704b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
705b9c1b51eSKate Stone     // argument entry.
706b547278cSGreg Clayton     arg.push_back(alias_arg);
707b547278cSGreg Clayton 
708b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
709b547278cSGreg Clayton     m_arguments.push_back(arg);
710b547278cSGreg Clayton   }
711b547278cSGreg Clayton 
7126e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
713b547278cSGreg Clayton 
71431fd64acSGongyu Deng   void
71531fd64acSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
71631fd64acSGongyu Deng                            OptionElementVector &opt_element_vector) override {
71731fd64acSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
71831fd64acSGongyu Deng       return;
71931fd64acSGongyu Deng 
72031fd64acSGongyu Deng     for (const auto &ent : m_interpreter.GetCommands()) {
72131fd64acSGongyu Deng       if (ent.second->IsRemovable())
72231fd64acSGongyu Deng         request.TryCompleteCurrentArg(ent.first, ent.second->GetHelp());
72331fd64acSGongyu Deng     }
72431fd64acSGongyu Deng   }
72531fd64acSGongyu Deng 
726b547278cSGreg Clayton protected:
727b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
728b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
729b547278cSGreg Clayton 
73011eb9c64SZachary Turner     if (args.empty()) {
73111eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
73211eb9c64SZachary Turner                                    "defined regular expression command names",
733a449698cSZachary Turner                                    GetCommandName().str().c_str());
734d77ea5b2SRaphael Isemann       return false;
73511eb9c64SZachary Turner     }
73611eb9c64SZachary Turner 
7370d9a201eSRaphael Isemann     auto command_name = args[0].ref();
7384574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
73946d4aa21SEnrico Granata       StreamString error_msg_stream;
740d5b44036SJonas Devlieghere       const bool generate_upropos = true;
74146d4aa21SEnrico Granata       const bool generate_type_lookup = false;
742b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
7434574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
744d5b44036SJonas Devlieghere           generate_upropos, generate_type_lookup);
745c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
7464574a890SZachary Turner       return false;
747b547278cSGreg Clayton     }
748b547278cSGreg Clayton 
7494574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
7504574a890SZachary Turner       result.AppendErrorWithFormat(
7514574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
752867e7d17SZachary Turner           args[0].c_str());
7534574a890SZachary Turner       return false;
7544574a890SZachary Turner     }
7554574a890SZachary Turner 
7564574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
7574574a890SZachary Turner     return true;
758b547278cSGreg Clayton   }
759b547278cSGreg Clayton };
760b547278cSGreg Clayton 
761de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
7621f0f5b5bSZachary Turner 
76364becc11SRaphael Isemann #define LLDB_OPTIONS_regex
76464becc11SRaphael Isemann #include "CommandOptions.inc"
7651f0f5b5bSZachary Turner 
7665a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
767de164aaaSGreg Clayton 
768b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
769b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
770de164aaaSGreg Clayton public:
7717428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
772b9c1b51eSKate Stone       : CommandObjectParsed(
773a925974bSAdrian Prantl             interpreter, "command regex",
774a925974bSAdrian Prantl             "Define a custom command in terms of "
775b9c1b51eSKate Stone             "existing commands by matching "
776b9c1b51eSKate Stone             "regular expressions.",
7770e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
778b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
779b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
780b9c1b51eSKate Stone         m_options() {
781b9c1b51eSKate Stone     SetHelpLong(
782b9c1b51eSKate Stone         R"(
783b9c1b51eSKate Stone )"
784b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
785ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
786b9c1b51eSKate Stone using the regular expression substitution format of:"
787b9c1b51eSKate Stone         R"(
788ea671fbdSKate Stone 
789ea671fbdSKate Stone     s/<regex>/<subst>/
790ea671fbdSKate Stone 
791b9c1b51eSKate Stone )"
792b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
793ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
794b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
795b9c1b51eSKate Stone         R"(
796ea671fbdSKate Stone 
797b9c1b51eSKate Stone )"
798b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
799ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
800ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
801b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
802b9c1b51eSKate Stone         R"(
803ea671fbdSKate Stone 
804ea671fbdSKate Stone EXAMPLES
805ea671fbdSKate Stone 
806b9c1b51eSKate Stone )"
807b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
808ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
809b9c1b51eSKate Stone a number follows 'f':"
810b9c1b51eSKate Stone         R"(
811ea671fbdSKate Stone 
812b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
813de164aaaSGreg Clayton   }
814de164aaaSGreg Clayton 
8156e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
816de164aaaSGreg Clayton 
8175a988416SJim Ingham protected:
8180affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
8197ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
8200affb582SDave Lee     if (output_sp && interactive) {
8210affb582SDave Lee       output_sp->PutCString("Enter one or more sed substitution commands in "
822b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
823b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
82444d93782SGreg Clayton       output_sp->Flush();
82544d93782SGreg Clayton     }
82644d93782SGreg Clayton   }
82744d93782SGreg Clayton 
828b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
829b9c1b51eSKate Stone                               std::string &data) override {
83044d93782SGreg Clayton     io_handler.SetIsDone(true);
831d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
83244d93782SGreg Clayton       StringList lines;
833b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
83444d93782SGreg Clayton         bool check_only = false;
8354c78b788SRaphael Isemann         for (const std::string &line : lines) {
8364c78b788SRaphael Isemann           Status error = AppendRegexSubstitution(line, check_only);
837b9c1b51eSKate Stone           if (error.Fail()) {
83857179860SJonas Devlieghere             if (!GetDebugger().GetCommandInterpreter().GetBatchCommandMode()) {
83957179860SJonas Devlieghere               StreamSP out_stream = GetDebugger().GetAsyncOutputStream();
84044d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
84144d93782SGreg Clayton             }
84244d93782SGreg Clayton           }
84344d93782SGreg Clayton         }
84444d93782SGreg Clayton       }
845d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
846d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
84744d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
84844d93782SGreg Clayton       }
84944d93782SGreg Clayton     }
85044d93782SGreg Clayton   }
85144d93782SGreg Clayton 
852b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
8535a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
854b9c1b51eSKate Stone     if (argc == 0) {
855b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
856b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
85711eb9c64SZachary Turner       return false;
85811eb9c64SZachary Turner     }
85911eb9c64SZachary Turner 
86097206d57SZachary Turner     Status error;
8610d9a201eSRaphael Isemann     auto name = command[0].ref();
862a8f3ae7cSJonas Devlieghere     m_regex_cmd_up = std::make_unique<CommandObjectRegexCommand>(
8634574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
8644574a890SZachary Turner         true);
8650e5e5a79SGreg Clayton 
866b9c1b51eSKate Stone     if (argc == 1) {
86757179860SJonas Devlieghere       Debugger &debugger = GetDebugger();
868e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
86944d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
870b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
871b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
87273d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
873514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
874514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
875b9c1b51eSKate Stone           multiple_lines, color_prompt,
876f6913cd7SGreg Clayton           0, // Don't show line numbers
877d77c2e09SJonas Devlieghere           *this, nullptr));
87844d93782SGreg Clayton 
879b9c1b51eSKate Stone       if (io_handler_sp) {
8807ce2de2cSJonas Devlieghere         debugger.RunIOHandlerAsync(io_handler_sp);
881de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
882de164aaaSGreg Clayton       }
883b9c1b51eSKate Stone     } else {
88497d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
88544d93782SGreg Clayton         bool check_only = false;
8860d9a201eSRaphael Isemann         error = AppendRegexSubstitution(entry.ref(), check_only);
8870e5e5a79SGreg Clayton         if (error.Fail())
8880e5e5a79SGreg Clayton           break;
8890e5e5a79SGreg Clayton       }
8900e5e5a79SGreg Clayton 
891b9c1b51eSKate Stone       if (error.Success()) {
8920e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
8930e5e5a79SGreg Clayton       }
8940e5e5a79SGreg Clayton     }
895b9c1b51eSKate Stone     if (error.Fail()) {
8960e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
897de164aaaSGreg Clayton     }
8980e5e5a79SGreg Clayton 
899de164aaaSGreg Clayton     return result.Succeeded();
900de164aaaSGreg Clayton   }
901de164aaaSGreg Clayton 
90297206d57SZachary Turner   Status AppendRegexSubstitution(const llvm::StringRef &regex_sed,
903b9c1b51eSKate Stone                                  bool check_only) {
90497206d57SZachary Turner     Status error;
9050e5e5a79SGreg Clayton 
906d5b44036SJonas Devlieghere     if (!m_regex_cmd_up) {
907b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
908b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
909b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9100e5e5a79SGreg Clayton       return error;
911de164aaaSGreg Clayton     }
9120e5e5a79SGreg Clayton 
9130e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
9140e5e5a79SGreg Clayton 
915b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
916b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
917b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
918b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9190e5e5a79SGreg Clayton       return error;
9200e5e5a79SGreg Clayton     }
9210e5e5a79SGreg Clayton 
922b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
923b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
924b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
925b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
9260e5e5a79SGreg Clayton       return error;
9270e5e5a79SGreg Clayton     }
9280e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
92905097246SAdrian Prantl     // use the char that follows 's' as the regex separator character so we can
93005097246SAdrian Prantl     // have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
9310e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
932b9c1b51eSKate Stone     const size_t second_separator_char_pos =
933b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
9340e5e5a79SGreg Clayton 
935b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
936b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
937b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
9380e5e5a79SGreg Clayton           separator_char,
9390e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
940ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
941b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9420e5e5a79SGreg Clayton       return error;
9430e5e5a79SGreg Clayton     }
9440e5e5a79SGreg Clayton 
945b9c1b51eSKate Stone     const size_t third_separator_char_pos =
946b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
9470e5e5a79SGreg Clayton 
948b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
949b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
950b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
9510e5e5a79SGreg Clayton           separator_char,
9520e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
953ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
954b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
9550e5e5a79SGreg Clayton       return error;
9560e5e5a79SGreg Clayton     }
9570e5e5a79SGreg Clayton 
958b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
95905097246SAdrian Prantl       // Make sure that everything that follows the last regex separator char
960b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
961b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
962b9c1b51eSKate Stone           std::string::npos) {
963b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
964b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
965b9c1b51eSKate Stone             "string: '%.*s'",
966b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
9670e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
9680e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
9690e5e5a79SGreg Clayton         return error;
9700e5e5a79SGreg Clayton       }
971b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
972b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
973b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
974b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9750e5e5a79SGreg Clayton           regex_sed.data());
9760e5e5a79SGreg Clayton       return error;
977b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
978b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
979b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
980b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
9810e5e5a79SGreg Clayton           regex_sed.data());
9820e5e5a79SGreg Clayton       return error;
9830e5e5a79SGreg Clayton     }
98444d93782SGreg Clayton 
985b9c1b51eSKate Stone     if (!check_only) {
986adcd0268SBenjamin Kramer       std::string regex(std::string(regex_sed.substr(
987adcd0268SBenjamin Kramer           first_separator_char_pos + 1,
988adcd0268SBenjamin Kramer           second_separator_char_pos - first_separator_char_pos - 1)));
989adcd0268SBenjamin Kramer       std::string subst(std::string(regex_sed.substr(
990adcd0268SBenjamin Kramer           second_separator_char_pos + 1,
991adcd0268SBenjamin Kramer           third_separator_char_pos - second_separator_char_pos - 1)));
99243224195SRaphael Isemann       m_regex_cmd_up->AddRegexCommand(regex, subst);
99344d93782SGreg Clayton     }
9940e5e5a79SGreg Clayton     return error;
995de164aaaSGreg Clayton   }
996de164aaaSGreg Clayton 
997b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
998d5b44036SJonas Devlieghere     if (m_regex_cmd_up) {
999d5b44036SJonas Devlieghere       if (m_regex_cmd_up->HasRegexEntries()) {
1000d5b44036SJonas Devlieghere         CommandObjectSP cmd_sp(m_regex_cmd_up.release());
1001de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1002de164aaaSGreg Clayton       }
1003de164aaaSGreg Clayton     }
1004de164aaaSGreg Clayton   }
1005de164aaaSGreg Clayton 
1006de164aaaSGreg Clayton private:
1007d5b44036SJonas Devlieghere   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_up;
1008de164aaaSGreg Clayton 
1009b9c1b51eSKate Stone   class CommandOptions : public Options {
1010de164aaaSGreg Clayton   public:
1011b9c1b51eSKate Stone     CommandOptions() : Options() {}
1012de164aaaSGreg Clayton 
10136e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1014de164aaaSGreg Clayton 
101597206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1016b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
101797206d57SZachary Turner       Status error;
10183bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1019de164aaaSGreg Clayton 
1020b9c1b51eSKate Stone       switch (short_option) {
1021de164aaaSGreg Clayton       case 'h':
1022adcd0268SBenjamin Kramer         m_help.assign(std::string(option_arg));
1023de164aaaSGreg Clayton         break;
1024de164aaaSGreg Clayton       case 's':
1025adcd0268SBenjamin Kramer         m_syntax.assign(std::string(option_arg));
1026de164aaaSGreg Clayton         break;
1027de164aaaSGreg Clayton       default:
102836162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1029de164aaaSGreg Clayton       }
1030de164aaaSGreg Clayton 
1031de164aaaSGreg Clayton       return error;
1032de164aaaSGreg Clayton     }
1033de164aaaSGreg Clayton 
1034b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1035de164aaaSGreg Clayton       m_help.clear();
1036de164aaaSGreg Clayton       m_syntax.clear();
1037de164aaaSGreg Clayton     }
1038de164aaaSGreg Clayton 
10391f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
104070602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
10411f0f5b5bSZachary Turner     }
1042de164aaaSGreg Clayton 
1043daed98e5SShivam Mittal     llvm::StringRef GetHelp() { return m_help; }
10446e3d8e7fSEugene Zelenko 
1045daed98e5SShivam Mittal     llvm::StringRef GetSyntax() { return m_syntax; }
10466e3d8e7fSEugene Zelenko 
1047de164aaaSGreg Clayton   protected:
10486e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
10496e3d8e7fSEugene Zelenko 
1050de164aaaSGreg Clayton     std::string m_help;
1051de164aaaSGreg Clayton     std::string m_syntax;
1052de164aaaSGreg Clayton   };
1053de164aaaSGreg Clayton 
1054b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1055de164aaaSGreg Clayton 
10565a988416SJim Ingham   CommandOptions m_options;
1057de164aaaSGreg Clayton };
1058de164aaaSGreg Clayton 
1059b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1060223383edSEnrico Granata public:
1061b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1062b9c1b51eSKate Stone                               std::string funct, std::string help,
1063b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1064a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_function_name(funct),
1065a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_long(false) {
1066735152e3SEnrico Granata     if (!help.empty())
1067442f6530SZachary Turner       SetHelp(help);
1068b9c1b51eSKate Stone     else {
1069735152e3SEnrico Granata       StreamString stream;
1070735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1071c156427dSZachary Turner       SetHelp(stream.GetString());
1072735152e3SEnrico Granata     }
1073223383edSEnrico Granata   }
1074223383edSEnrico Granata 
10756e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1076223383edSEnrico Granata 
1077b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
10785a988416SJim Ingham 
1079b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
10805a988416SJim Ingham 
1081b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
10825a988416SJim Ingham 
1083442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1084442f6530SZachary Turner     if (m_fetched_help_long)
1085442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1086442f6530SZachary Turner 
10872b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1088442f6530SZachary Turner     if (!scripter)
1089442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1090442f6530SZachary Turner 
1091fac939e9SEnrico Granata     std::string docstring;
1092442f6530SZachary Turner     m_fetched_help_long =
1093442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1094fac939e9SEnrico Granata     if (!docstring.empty())
1095442f6530SZachary Turner       SetHelpLong(docstring);
1096fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1097fac939e9SEnrico Granata   }
1098fac939e9SEnrico Granata 
10995a988416SJim Ingham protected:
11004d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1101b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11022b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1103223383edSEnrico Granata 
110497206d57SZachary Turner     Status error;
1105223383edSEnrico Granata 
110670f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
110770f11f88SJim Ingham 
1108a925974bSAdrian Prantl     if (!scripter || !scripter->RunScriptBasedCommand(
1109a925974bSAdrian Prantl                          m_function_name.c_str(), raw_command_line, m_synchro,
1110a925974bSAdrian Prantl                          result, error, m_exe_ctx)) {
1111223383edSEnrico Granata       result.AppendError(error.AsCString());
1112b9c1b51eSKate Stone     } else {
111370f11f88SJim Ingham       // Don't change the status if the command already set it...
1114b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1115c156427dSZachary Turner         if (result.GetOutputData().empty())
1116223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
111770f11f88SJim Ingham         else
111870f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
111970f11f88SJim Ingham       }
112070f11f88SJim Ingham     }
1121223383edSEnrico Granata 
1122223383edSEnrico Granata     return result.Succeeded();
1123223383edSEnrico Granata   }
1124223383edSEnrico Granata 
11256e3d8e7fSEugene Zelenko private:
11266e3d8e7fSEugene Zelenko   std::string m_function_name;
11276e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
11286e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1129223383edSEnrico Granata };
1130223383edSEnrico Granata 
1131b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
11329fe00e52SEnrico Granata public:
11339fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
11349fe00e52SEnrico Granata                                std::string name,
11350641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1136b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1137a925974bSAdrian Prantl       : CommandObjectRaw(interpreter, name), m_cmd_obj_sp(cmd_obj_sp),
1138a925974bSAdrian Prantl         m_synchro(synch), m_fetched_help_short(false),
1139b9c1b51eSKate Stone         m_fetched_help_long(false) {
11409fe00e52SEnrico Granata     StreamString stream;
11419fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1142c156427dSZachary Turner     SetHelp(stream.GetString());
11432b29b432SJonas Devlieghere     if (ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter())
1144e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
11459fe00e52SEnrico Granata   }
11469fe00e52SEnrico Granata 
11476e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
11489fe00e52SEnrico Granata 
1149b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
11509fe00e52SEnrico Granata 
1151b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
11529fe00e52SEnrico Granata 
1153442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1154442f6530SZachary Turner     if (m_fetched_help_short)
1155442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11562b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1157442f6530SZachary Turner     if (!scripter)
1158442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
11596f79bb2dSEnrico Granata     std::string docstring;
1160b9c1b51eSKate Stone     m_fetched_help_short =
1161b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
11626f79bb2dSEnrico Granata     if (!docstring.empty())
1163442f6530SZachary Turner       SetHelp(docstring);
1164442f6530SZachary Turner 
11656f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
11666f79bb2dSEnrico Granata   }
11676f79bb2dSEnrico Granata 
1168442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1169442f6530SZachary Turner     if (m_fetched_help_long)
1170442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1171442f6530SZachary Turner 
11722b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
1173442f6530SZachary Turner     if (!scripter)
1174442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1175442f6530SZachary Turner 
11766f79bb2dSEnrico Granata     std::string docstring;
1177b9c1b51eSKate Stone     m_fetched_help_long =
1178b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
11796f79bb2dSEnrico Granata     if (!docstring.empty())
1180442f6530SZachary Turner       SetHelpLong(docstring);
11819fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
11829fe00e52SEnrico Granata   }
11839fe00e52SEnrico Granata 
11849fe00e52SEnrico Granata protected:
11854d51a902SRaphael Isemann   bool DoExecute(llvm::StringRef raw_command_line,
1186b9c1b51eSKate Stone                  CommandReturnObject &result) override {
11872b29b432SJonas Devlieghere     ScriptInterpreter *scripter = GetDebugger().GetScriptInterpreter();
11889fe00e52SEnrico Granata 
118997206d57SZachary Turner     Status error;
11909fe00e52SEnrico Granata 
11919fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
11929fe00e52SEnrico Granata 
1193b9c1b51eSKate Stone     if (!scripter ||
1194b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1195b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
11969fe00e52SEnrico Granata       result.AppendError(error.AsCString());
1197b9c1b51eSKate Stone     } else {
11989fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1199b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1200c156427dSZachary Turner         if (result.GetOutputData().empty())
12019fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
12029fe00e52SEnrico Granata         else
12039fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
12049fe00e52SEnrico Granata       }
12059fe00e52SEnrico Granata     }
12069fe00e52SEnrico Granata 
12079fe00e52SEnrico Granata     return result.Succeeded();
12089fe00e52SEnrico Granata   }
12099fe00e52SEnrico Granata 
12106e3d8e7fSEugene Zelenko private:
12116e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
12126e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
12136e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
12146e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
12159fe00e52SEnrico Granata };
12169fe00e52SEnrico Granata 
1217a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
121864becc11SRaphael Isemann #define LLDB_OPTIONS_script_import
121964becc11SRaphael Isemann #include "CommandOptions.inc"
12201f0f5b5bSZachary Turner 
1221b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
12225a988416SJim Ingham public:
1223b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1224b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1225b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1226b9c1b51eSKate Stone         m_options() {
12275a988416SJim Ingham     CommandArgumentEntry arg1;
12285a988416SJim Ingham     CommandArgumentData cmd_arg;
12295a988416SJim Ingham 
12305a988416SJim Ingham     // Define the first (and only) variant of this arg.
12315a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
12323b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
12335a988416SJim Ingham 
1234b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1235b9c1b51eSKate Stone     // argument entry.
12365a988416SJim Ingham     arg1.push_back(cmd_arg);
12375a988416SJim Ingham 
12385a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
12395a988416SJim Ingham     m_arguments.push_back(arg1);
12405a988416SJim Ingham   }
12415a988416SJim Ingham 
12426e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
12435a988416SJim Ingham 
1244ae34ed2cSRaphael Isemann   void
1245ae34ed2cSRaphael Isemann   HandleArgumentCompletion(CompletionRequest &request,
12462443bbd4SRaphael Isemann                            OptionElementVector &opt_element_vector) override {
1247b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1248b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1249a2e76c0bSRaphael Isemann         request, nullptr);
12505a988416SJim Ingham   }
12515a988416SJim Ingham 
1252b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
12535a988416SJim Ingham 
12545a988416SJim Ingham protected:
1255b9c1b51eSKate Stone   class CommandOptions : public Options {
12560a305db7SEnrico Granata   public:
1257b9c1b51eSKate Stone     CommandOptions() : Options() {}
12580a305db7SEnrico Granata 
12596e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
12600a305db7SEnrico Granata 
126197206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1262b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
126397206d57SZachary Turner       Status error;
12643bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
12650a305db7SEnrico Granata 
1266b9c1b51eSKate Stone       switch (short_option) {
12670a305db7SEnrico Granata       case 'r':
126815625112SJonas Devlieghere         // NO-OP
12690a305db7SEnrico Granata         break;
127000bb397bSJonas Devlieghere       case 'c':
127100bb397bSJonas Devlieghere         relative_to_command_file = true;
127200bb397bSJonas Devlieghere         break;
1273f9517353SJonas Devlieghere       case 's':
1274f9517353SJonas Devlieghere         silent = true;
1275f9517353SJonas Devlieghere         break;
12760a305db7SEnrico Granata       default:
127736162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
12780a305db7SEnrico Granata       }
12790a305db7SEnrico Granata 
12800a305db7SEnrico Granata       return error;
12810a305db7SEnrico Granata     }
12820a305db7SEnrico Granata 
1283b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
128400bb397bSJonas Devlieghere       relative_to_command_file = false;
12850a305db7SEnrico Granata     }
12860a305db7SEnrico Granata 
12871f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
128870602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
12891f0f5b5bSZachary Turner     }
129000bb397bSJonas Devlieghere     bool relative_to_command_file = false;
1291f9517353SJonas Devlieghere     bool silent = false;
12920a305db7SEnrico Granata   };
12930a305db7SEnrico Granata 
1294b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
129511eb9c64SZachary Turner     if (command.empty()) {
12963b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1297a9dbf432SEnrico Granata       return false;
1298a9dbf432SEnrico Granata     }
1299a9dbf432SEnrico Granata 
130000bb397bSJonas Devlieghere     FileSpec source_dir = {};
130100bb397bSJonas Devlieghere     if (m_options.relative_to_command_file) {
130200bb397bSJonas Devlieghere       source_dir = GetDebugger().GetCommandInterpreter().GetCurrentSourceDir();
130300bb397bSJonas Devlieghere       if (!source_dir) {
130400bb397bSJonas Devlieghere         result.AppendError("command script import -c can only be specified "
130500bb397bSJonas Devlieghere                            "from a command file");
130600bb397bSJonas Devlieghere         return false;
130700bb397bSJonas Devlieghere       }
130800bb397bSJonas Devlieghere     }
130900bb397bSJonas Devlieghere 
131011eb9c64SZachary Turner     for (auto &entry : command.entries()) {
131197206d57SZachary Turner       Status error;
1312a9dbf432SEnrico Granata 
1313f9517353SJonas Devlieghere       LoadScriptOptions options;
1314f9517353SJonas Devlieghere       options.SetInitSession(true);
1315f9517353SJonas Devlieghere       options.SetSilent(m_options.silent);
1316f9517353SJonas Devlieghere 
1317b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
131811eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
131911eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
132005097246SAdrian Prantl       // script imports" in __lldb_init_module the real fix is to have
132105097246SAdrian Prantl       // recursive commands possible with a CommandInvocation object separate
132205097246SAdrian Prantl       // from the CommandObject itself, so that recursive command invocations
132305097246SAdrian Prantl       // won't stomp on each other (wrt to execution contents, options, and
132405097246SAdrian Prantl       // more)
1325078551c7SEnrico Granata       m_exe_ctx.Clear();
13262b29b432SJonas Devlieghere       if (GetDebugger().GetScriptInterpreter()->LoadScriptingModule(
1327f9517353SJonas Devlieghere               entry.c_str(), options, error, /*module_sp=*/nullptr,
1328f9517353SJonas Devlieghere               source_dir)) {
1329a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1330b9c1b51eSKate Stone       } else {
1331b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1332b9c1b51eSKate Stone                                      error.AsCString());
1333a9dbf432SEnrico Granata       }
13343b00e35bSEnrico Granata     }
1335a9dbf432SEnrico Granata 
1336a9dbf432SEnrico Granata     return result.Succeeded();
1337a9dbf432SEnrico Granata   }
13380a305db7SEnrico Granata 
13395a988416SJim Ingham   CommandOptions m_options;
1340a9dbf432SEnrico Granata };
1341223383edSEnrico Granata 
1342223383edSEnrico Granata // CommandObjectCommandsScriptAdd
13438fe53c49STatyana Krasnukha static constexpr OptionEnumValueElement g_script_synchro_type[] = {
1344e063ecccSJonas Devlieghere     {
1345e063ecccSJonas Devlieghere         eScriptedCommandSynchronicitySynchronous,
1346e063ecccSJonas Devlieghere         "synchronous",
1347e063ecccSJonas Devlieghere         "Run synchronous",
1348e063ecccSJonas Devlieghere     },
1349e063ecccSJonas Devlieghere     {
1350e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityAsynchronous,
1351e063ecccSJonas Devlieghere         "asynchronous",
1352e063ecccSJonas Devlieghere         "Run asynchronous",
1353e063ecccSJonas Devlieghere     },
1354e063ecccSJonas Devlieghere     {
1355e063ecccSJonas Devlieghere         eScriptedCommandSynchronicityCurrentValue,
1356e063ecccSJonas Devlieghere         "current",
1357e063ecccSJonas Devlieghere         "Do not alter current setting",
1358e063ecccSJonas Devlieghere     },
1359e063ecccSJonas Devlieghere };
13601f0f5b5bSZachary Turner 
13618fe53c49STatyana Krasnukha static constexpr OptionEnumValues ScriptSynchroType() {
13628fe53c49STatyana Krasnukha   return OptionEnumValues(g_script_synchro_type);
13638fe53c49STatyana Krasnukha }
13648fe53c49STatyana Krasnukha 
136564becc11SRaphael Isemann #define LLDB_OPTIONS_script_add
136664becc11SRaphael Isemann #include "CommandOptions.inc"
13671f0f5b5bSZachary Turner 
1368b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1369b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
13705a988416SJim Ingham public:
1371b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1372b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
13735a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
13746e3d8e7fSEugene Zelenko                             nullptr),
1375b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
13765a988416SJim Ingham     CommandArgumentEntry arg1;
13775a988416SJim Ingham     CommandArgumentData cmd_arg;
13785a988416SJim Ingham 
13795a988416SJim Ingham     // Define the first (and only) variant of this arg.
13805a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
13815a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
13825a988416SJim Ingham 
1383b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1384b9c1b51eSKate Stone     // argument entry.
13855a988416SJim Ingham     arg1.push_back(cmd_arg);
13865a988416SJim Ingham 
13875a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
13885a988416SJim Ingham     m_arguments.push_back(arg1);
13895a988416SJim Ingham   }
13905a988416SJim Ingham 
13916e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
13925a988416SJim Ingham 
1393b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
13945a988416SJim Ingham 
13955a988416SJim Ingham protected:
1396b9c1b51eSKate Stone   class CommandOptions : public Options {
1397223383edSEnrico Granata   public:
1398b9c1b51eSKate Stone     CommandOptions()
13999494c510SJonas Devlieghere         : Options(), m_class_name(), m_funct_name(), m_short_help() {}
1400223383edSEnrico Granata 
14016e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1402223383edSEnrico Granata 
140397206d57SZachary Turner     Status SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1404b9c1b51eSKate Stone                           ExecutionContext *execution_context) override {
140597206d57SZachary Turner       Status error;
14063bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1407223383edSEnrico Granata 
1408b9c1b51eSKate Stone       switch (short_option) {
1409223383edSEnrico Granata       case 'f':
1410fe11483bSZachary Turner         if (!option_arg.empty())
1411adcd0268SBenjamin Kramer           m_funct_name = std::string(option_arg);
1412735152e3SEnrico Granata         break;
14139fe00e52SEnrico Granata       case 'c':
1414fe11483bSZachary Turner         if (!option_arg.empty())
1415adcd0268SBenjamin Kramer           m_class_name = std::string(option_arg);
14169fe00e52SEnrico Granata         break;
1417735152e3SEnrico Granata       case 'h':
1418fe11483bSZachary Turner         if (!option_arg.empty())
1419adcd0268SBenjamin Kramer           m_short_help = std::string(option_arg);
1420223383edSEnrico Granata         break;
14210a305db7SEnrico Granata       case 's':
1422b9c1b51eSKate Stone         m_synchronicity =
142347cbf4a0SPavel Labath             (ScriptedCommandSynchronicity)OptionArgParser::ToOptionEnum(
1424fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
14250a305db7SEnrico Granata         if (!error.Success())
1426b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1427fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1428fe11483bSZachary Turner               option_arg.str().c_str());
14290a305db7SEnrico Granata         break;
1430223383edSEnrico Granata       default:
143136162014SRaphael Isemann         llvm_unreachable("Unimplemented option");
1432223383edSEnrico Granata       }
1433223383edSEnrico Granata 
1434223383edSEnrico Granata       return error;
1435223383edSEnrico Granata     }
1436223383edSEnrico Granata 
1437b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
14389fe00e52SEnrico Granata       m_class_name.clear();
1439735152e3SEnrico Granata       m_funct_name.clear();
1440735152e3SEnrico Granata       m_short_help.clear();
144144d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1442223383edSEnrico Granata     }
1443223383edSEnrico Granata 
14441f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
144570602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
14461f0f5b5bSZachary Turner     }
1447223383edSEnrico Granata 
1448223383edSEnrico Granata     // Instance variables to hold the values for command options.
1449223383edSEnrico Granata 
14509fe00e52SEnrico Granata     std::string m_class_name;
1451223383edSEnrico Granata     std::string m_funct_name;
1452735152e3SEnrico Granata     std::string m_short_help;
14539494c510SJonas Devlieghere     ScriptedCommandSynchronicity m_synchronicity =
14549494c510SJonas Devlieghere         eScriptedCommandSynchronicitySynchronous;
1455223383edSEnrico Granata   };
1456223383edSEnrico Granata 
14570affb582SDave Lee   void IOHandlerActivated(IOHandler &io_handler, bool interactive) override {
14587ca15ba7SLawrence D'Anna     StreamFileSP output_sp(io_handler.GetOutputStreamFileSP());
14590affb582SDave Lee     if (output_sp && interactive) {
146044d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
146144d93782SGreg Clayton       output_sp->Flush();
1462223383edSEnrico Granata     }
1463223383edSEnrico Granata   }
1464223383edSEnrico Granata 
1465b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1466b9c1b51eSKate Stone                               std::string &data) override {
14677ca15ba7SLawrence D'Anna     StreamFileSP error_sp = io_handler.GetErrorStreamFileSP();
146844d93782SGreg Clayton 
14692b29b432SJonas Devlieghere     ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1470b9c1b51eSKate Stone     if (interpreter) {
147144d93782SGreg Clayton 
147244d93782SGreg Clayton       StringList lines;
147344d93782SGreg Clayton       lines.SplitIntoLines(data);
1474b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1475a73b7df7SEnrico Granata         std::string funct_name_str;
1476b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1477b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1478b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1479b9c1b51eSKate Stone                              "add python command.\n");
148044d93782SGreg Clayton             error_sp->Flush();
1481b9c1b51eSKate Stone           } else {
1482223383edSEnrico Granata             // everything should be fine now, let's add this alias
1483223383edSEnrico Granata 
1484b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1485771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
148644d93782SGreg Clayton                 m_synchronicity));
1487223383edSEnrico Granata 
1488b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1489b9c1b51eSKate Stone                                               true)) {
1490b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1491b9c1b51eSKate Stone                                "add python command.\n");
149244d93782SGreg Clayton               error_sp->Flush();
1493223383edSEnrico Granata             }
1494223383edSEnrico Granata           }
1495b9c1b51eSKate Stone         } else {
1496b9c1b51eSKate Stone           error_sp->Printf(
1497b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
149844d93782SGreg Clayton           error_sp->Flush();
149944d93782SGreg Clayton         }
1500b9c1b51eSKate Stone       } else {
150144d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
150244d93782SGreg Clayton         error_sp->Flush();
150344d93782SGreg Clayton       }
1504b9c1b51eSKate Stone     } else {
1505b9c1b51eSKate Stone       error_sp->Printf(
1506b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
150744d93782SGreg Clayton       error_sp->Flush();
150844d93782SGreg Clayton     }
150944d93782SGreg Clayton 
151044d93782SGreg Clayton     io_handler.SetIsDone(true);
151144d93782SGreg Clayton   }
1512223383edSEnrico Granata 
1513b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
151457179860SJonas Devlieghere     if (GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython) {
1515b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1516b9c1b51eSKate Stone                          "commands is currently Python");
151799f0b8f9SEnrico Granata       return false;
151899f0b8f9SEnrico Granata     }
151999f0b8f9SEnrico Granata 
152011eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1521223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1522223383edSEnrico Granata       return false;
1523223383edSEnrico Granata     }
1524223383edSEnrico Granata 
1525735152e3SEnrico Granata     // Store the options in case we get multi-line input
1526adcd0268SBenjamin Kramer     m_cmd_name = std::string(command[0].ref());
1527735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
152844d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1529223383edSEnrico Granata 
1530b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1531b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1532b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1533b9c1b51eSKate Stone             "     ", // Prompt
1534a6faf851SJonas Devlieghere             *this);  // IOHandlerDelegate
1535b9c1b51eSKate Stone       } else {
1536b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1537b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1538b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1539b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1540223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1541b9c1b51eSKate Stone         } else {
1542223383edSEnrico Granata           result.AppendError("cannot add command");
1543223383edSEnrico Granata         }
1544223383edSEnrico Granata       }
1545b9c1b51eSKate Stone     } else {
15462b29b432SJonas Devlieghere       ScriptInterpreter *interpreter = GetDebugger().GetScriptInterpreter();
1547b9c1b51eSKate Stone       if (!interpreter) {
15489fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
15499fe00e52SEnrico Granata         return false;
15509fe00e52SEnrico Granata       }
15519fe00e52SEnrico Granata 
1552b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1553b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1554b9c1b51eSKate Stone       if (!cmd_obj_sp) {
15559fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
15569fe00e52SEnrico Granata         return false;
15579fe00e52SEnrico Granata       }
15589fe00e52SEnrico Granata 
1559b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1560b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1561b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
15629fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1563b9c1b51eSKate Stone       } else {
15649fe00e52SEnrico Granata         result.AppendError("cannot add command");
15659fe00e52SEnrico Granata       }
15669fe00e52SEnrico Granata     }
1567223383edSEnrico Granata 
1568223383edSEnrico Granata     return result.Succeeded();
1569223383edSEnrico Granata   }
15705a988416SJim Ingham 
15715a988416SJim Ingham   CommandOptions m_options;
157244d93782SGreg Clayton   std::string m_cmd_name;
1573735152e3SEnrico Granata   std::string m_short_help;
157444d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1575223383edSEnrico Granata };
1576223383edSEnrico Granata 
1577223383edSEnrico Granata // CommandObjectCommandsScriptList
1578223383edSEnrico Granata 
1579b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1580223383edSEnrico Granata public:
1581b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1582b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1583b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1584223383edSEnrico Granata 
15856e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1586223383edSEnrico Granata 
1587b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1588d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1589d77ea5b2SRaphael Isemann       result.AppendError("'command script list' doesn't take any arguments");
1590d77ea5b2SRaphael Isemann       return false;
1591d77ea5b2SRaphael Isemann     }
1592d77ea5b2SRaphael Isemann 
1593b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1594223383edSEnrico Granata 
1595223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1596223383edSEnrico Granata 
1597223383edSEnrico Granata     return true;
1598223383edSEnrico Granata   }
1599223383edSEnrico Granata };
1600223383edSEnrico Granata 
1601223383edSEnrico Granata // CommandObjectCommandsScriptClear
1602223383edSEnrico Granata 
1603b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1604223383edSEnrico Granata public:
1605b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1606b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1607b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1608223383edSEnrico Granata 
16096e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1610223383edSEnrico Granata 
16115a988416SJim Ingham protected:
1612b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1613d77ea5b2SRaphael Isemann     if (command.GetArgumentCount() != 0) {
1614d77ea5b2SRaphael Isemann       result.AppendError("'command script clear' doesn't take any arguments");
1615d77ea5b2SRaphael Isemann       return false;
1616d77ea5b2SRaphael Isemann     }
1617d77ea5b2SRaphael Isemann 
1618223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1619223383edSEnrico Granata 
1620223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1621223383edSEnrico Granata 
1622223383edSEnrico Granata     return true;
1623223383edSEnrico Granata   }
1624223383edSEnrico Granata };
1625223383edSEnrico Granata 
1626223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1627223383edSEnrico Granata 
1628b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1629223383edSEnrico Granata public:
1630b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1631b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1632b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1633223383edSEnrico Granata     CommandArgumentEntry arg1;
1634223383edSEnrico Granata     CommandArgumentData cmd_arg;
1635223383edSEnrico Granata 
1636223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1637223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1638223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1639223383edSEnrico Granata 
1640b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1641b9c1b51eSKate Stone     // argument entry.
1642223383edSEnrico Granata     arg1.push_back(cmd_arg);
1643223383edSEnrico Granata 
1644223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1645223383edSEnrico Granata     m_arguments.push_back(arg1);
1646223383edSEnrico Granata   }
1647223383edSEnrico Granata 
16486e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1649223383edSEnrico Granata 
16502e8f304fSGongyu Deng   void
16512e8f304fSGongyu Deng   HandleArgumentCompletion(CompletionRequest &request,
16522e8f304fSGongyu Deng                            OptionElementVector &opt_element_vector) override {
16532e8f304fSGongyu Deng     if (!m_interpreter.HasCommands() || request.GetCursorIndex() != 0)
16542e8f304fSGongyu Deng       return;
16552e8f304fSGongyu Deng 
16562ebe30c6SRaphael Isemann     for (const auto &c : m_interpreter.GetUserCommands())
16572ebe30c6SRaphael Isemann       request.TryCompleteCurrentArg(c.first, c.second->GetHelp());
16582e8f304fSGongyu Deng   }
16592e8f304fSGongyu Deng 
16605a988416SJim Ingham protected:
1661b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1662223383edSEnrico Granata 
166311eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1664223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1665223383edSEnrico Granata       return false;
1666223383edSEnrico Granata     }
1667223383edSEnrico Granata 
16680d9a201eSRaphael Isemann     auto cmd_name = command[0].ref();
1669223383edSEnrico Granata 
16704574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
16714574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1672867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
16734574a890SZachary Turner       return false;
1674223383edSEnrico Granata     }
1675223383edSEnrico Granata 
16764574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
16774574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
16784574a890SZachary Turner     return true;
1679223383edSEnrico Granata   }
1680223383edSEnrico Granata };
1681223383edSEnrico Granata 
1682223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1683223383edSEnrico Granata 
1684223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1685223383edSEnrico Granata 
1686b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1687223383edSEnrico Granata public:
16887428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1689b9c1b51eSKate Stone       : CommandObjectMultiword(
1690a925974bSAdrian Prantl             interpreter, "command script",
1691a925974bSAdrian Prantl             "Commands for managing custom "
1692b9c1b51eSKate Stone             "commands implemented by "
1693b9c1b51eSKate Stone             "interpreter scripts.",
1694b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1695b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1696b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1697b9c1b51eSKate Stone     LoadSubCommand(
1698b9c1b51eSKate Stone         "delete",
1699b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1700b9c1b51eSKate Stone     LoadSubCommand(
1701b9c1b51eSKate Stone         "clear",
1702b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1703b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1704b9c1b51eSKate Stone                                interpreter)));
1705b9c1b51eSKate Stone     LoadSubCommand(
1706b9c1b51eSKate Stone         "import",
1707b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1708223383edSEnrico Granata   }
1709223383edSEnrico Granata 
17106e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1711223383edSEnrico Granata };
1712223383edSEnrico Granata 
1713ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1714ebc09c36SJim Ingham 
1715ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1716ebc09c36SJim Ingham 
1717b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1718b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1719b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1720b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1721b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1722b9c1b51eSKate Stone   LoadSubCommand("source",
1723b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1724b9c1b51eSKate Stone   LoadSubCommand("alias",
1725b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1726b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1727b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1728b9c1b51eSKate Stone   LoadSubCommand("delete",
1729b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1730b9c1b51eSKate Stone   LoadSubCommand(
1731b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1732b9c1b51eSKate Stone   LoadSubCommand(
1733b9c1b51eSKate Stone       "script",
1734b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1735ebc09c36SJim Ingham }
1736ebc09c36SJim Ingham 
17376e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1738