1696bd635SAlexander Shaposhnikov //===-- CommandObjectCommands.cpp -------------------------------*- C++ -*-===//
2ebc09c36SJim Ingham //
3ebc09c36SJim Ingham //                     The LLVM Compiler Infrastructure
4ebc09c36SJim Ingham //
5ebc09c36SJim Ingham // This file is distributed under the University of Illinois Open Source
6ebc09c36SJim Ingham // License. See LICENSE.TXT for details.
7ebc09c36SJim Ingham //
8ebc09c36SJim Ingham //===----------------------------------------------------------------------===//
9ebc09c36SJim Ingham 
10ebc09c36SJim Ingham // C Includes
11ebc09c36SJim Ingham // C++ Includes
12ebc09c36SJim Ingham // Other libraries and framework includes
130e5e5a79SGreg Clayton #include "llvm/ADT/StringRef.h"
140e5e5a79SGreg Clayton 
15ebc09c36SJim Ingham // Project includes
166e3d8e7fSEugene Zelenko #include "CommandObjectCommands.h"
1746d4aa21SEnrico Granata #include "CommandObjectHelp.h"
18ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1944d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
203eb2b44dSZachary Turner #include "lldb/Host/OptionParser.h"
21de164aaaSGreg Clayton #include "lldb/Interpreter/Args.h"
227594f14fSEnrico Granata #include "lldb/Interpreter/CommandHistory.h"
23ebc09c36SJim Ingham #include "lldb/Interpreter/CommandInterpreter.h"
24de164aaaSGreg Clayton #include "lldb/Interpreter/CommandObjectRegexCommand.h"
25ebc09c36SJim Ingham #include "lldb/Interpreter/CommandReturnObject.h"
26012d4fcaSEnrico Granata #include "lldb/Interpreter/OptionValueBoolean.h"
2745d0e238SEnrico Granata #include "lldb/Interpreter/OptionValueString.h"
287594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
29ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
3099f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
31573ab909SZachary Turner #include "lldb/Utility/StringList.h"
32ebc09c36SJim Ingham 
33ebc09c36SJim Ingham using namespace lldb;
34ebc09c36SJim Ingham using namespace lldb_private;
35ebc09c36SJim Ingham 
36ebc09c36SJim Ingham //-------------------------------------------------------------------------
37ebc09c36SJim Ingham // CommandObjectCommandsSource
38ebc09c36SJim Ingham //-------------------------------------------------------------------------
39ebc09c36SJim Ingham 
401f0f5b5bSZachary Turner static OptionDefinition g_history_options[] = {
411f0f5b5bSZachary Turner     // clang-format off
421f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "count",       'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "How many history commands to print." },
431f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "start-index", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Index at which to start printing history commands (or end to mean tail mode)." },
441f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "end-index",   'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger, "Index at which to stop printing history commands." },
451f0f5b5bSZachary Turner   { LLDB_OPT_SET_2, false, "clear",       'C', OptionParser::eNoArgument,       nullptr, nullptr, 0, eArgTypeBoolean,         "Clears the current command history." },
461f0f5b5bSZachary Turner     // clang-format on
471f0f5b5bSZachary Turner };
481f0f5b5bSZachary Turner 
49b9c1b51eSKate Stone class CommandObjectCommandsHistory : public CommandObjectParsed {
505a988416SJim Ingham public:
51b9c1b51eSKate Stone   CommandObjectCommandsHistory(CommandInterpreter &interpreter)
52b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command history",
53*aab5be05SJim Ingham                             "Dump the history of commands in this session.\n"
54*aab5be05SJim Ingham                             "Commands in the history list can be run again "
55*aab5be05SJim Ingham                             "using \"!<INDEX>\".   \"!-<OFFSET>\" will re-run "
56*aab5be05SJim Ingham                             "the command that is <OFFSET> commands from the end"
57*aab5be05SJim Ingham                             " of the list (counting the current command).",
586e3d8e7fSEugene Zelenko                             nullptr),
59b9c1b51eSKate Stone         m_options() {}
605a988416SJim Ingham 
616e3d8e7fSEugene Zelenko   ~CommandObjectCommandsHistory() override = default;
625a988416SJim Ingham 
63b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
645a988416SJim Ingham 
655a988416SJim Ingham protected:
66b9c1b51eSKate Stone   class CommandOptions : public Options {
67a5a97ebeSJim Ingham   public:
68b9c1b51eSKate Stone     CommandOptions()
69b9c1b51eSKate Stone         : Options(), m_start_idx(0), m_stop_idx(0), m_count(0), m_clear(false) {
70a5a97ebeSJim Ingham     }
71a5a97ebeSJim Ingham 
726e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
73a5a97ebeSJim Ingham 
74fe11483bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
75b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
76a5a97ebeSJim Ingham       Error error;
773bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
78a5a97ebeSJim Ingham 
79b9c1b51eSKate Stone       switch (short_option) {
80a5a97ebeSJim Ingham       case 'c':
81fe11483bSZachary Turner         error = m_count.SetValueFromString(option_arg, eVarSetOperationAssign);
82a5a97ebeSJim Ingham         break;
83a5a97ebeSJim Ingham       case 's':
84fe11483bSZachary Turner         if (option_arg == "end") {
857594f14fSEnrico Granata           m_start_idx.SetCurrentValue(UINT64_MAX);
867594f14fSEnrico Granata           m_start_idx.SetOptionWasSet();
87b9c1b51eSKate Stone         } else
88fe11483bSZachary Turner           error = m_start_idx.SetValueFromString(option_arg,
89b9c1b51eSKate Stone                                                  eVarSetOperationAssign);
907594f14fSEnrico Granata         break;
917594f14fSEnrico Granata       case 'e':
92fe11483bSZachary Turner         error =
93fe11483bSZachary Turner             m_stop_idx.SetValueFromString(option_arg, eVarSetOperationAssign);
947594f14fSEnrico Granata         break;
9563123b64SEnrico Granata       case 'C':
9663123b64SEnrico Granata         m_clear.SetCurrentValue(true);
9763123b64SEnrico Granata         m_clear.SetOptionWasSet();
98a5a97ebeSJim Ingham         break;
99a5a97ebeSJim Ingham       default:
100b9c1b51eSKate Stone         error.SetErrorStringWithFormat("unrecognized option '%c'",
101b9c1b51eSKate Stone                                        short_option);
102a5a97ebeSJim Ingham         break;
103a5a97ebeSJim Ingham       }
104a5a97ebeSJim Ingham 
105a5a97ebeSJim Ingham       return error;
106a5a97ebeSJim Ingham     }
107a5a97ebeSJim Ingham 
108b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1097594f14fSEnrico Granata       m_start_idx.Clear();
1107594f14fSEnrico Granata       m_stop_idx.Clear();
1117594f14fSEnrico Granata       m_count.Clear();
11263123b64SEnrico Granata       m_clear.Clear();
113a5a97ebeSJim Ingham     }
114a5a97ebeSJim Ingham 
1151f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
11670602439SZachary Turner       return llvm::makeArrayRef(g_history_options);
1171f0f5b5bSZachary Turner     }
118a5a97ebeSJim Ingham 
119a5a97ebeSJim Ingham     // Instance variables to hold the values for command options.
120a5a97ebeSJim Ingham 
1217594f14fSEnrico Granata     OptionValueUInt64 m_start_idx;
1227594f14fSEnrico Granata     OptionValueUInt64 m_stop_idx;
1237594f14fSEnrico Granata     OptionValueUInt64 m_count;
12463123b64SEnrico Granata     OptionValueBoolean m_clear;
125a5a97ebeSJim Ingham   };
126a5a97ebeSJim Ingham 
127b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
128b9c1b51eSKate Stone     if (m_options.m_clear.GetCurrentValue() &&
129b9c1b51eSKate Stone         m_options.m_clear.OptionWasSet()) {
1307594f14fSEnrico Granata       m_interpreter.GetCommandHistory().Clear();
1317594f14fSEnrico Granata       result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
132b9c1b51eSKate Stone     } else {
133b9c1b51eSKate Stone       if (m_options.m_start_idx.OptionWasSet() &&
134b9c1b51eSKate Stone           m_options.m_stop_idx.OptionWasSet() &&
135b9c1b51eSKate Stone           m_options.m_count.OptionWasSet()) {
136b9c1b51eSKate Stone         result.AppendError("--count, --start-index and --end-index cannot be "
137b9c1b51eSKate Stone                            "all specified in the same invocation");
1387594f14fSEnrico Granata         result.SetStatus(lldb::eReturnStatusFailed);
139b9c1b51eSKate Stone       } else {
140b9c1b51eSKate Stone         std::pair<bool, uint64_t> start_idx(
141b9c1b51eSKate Stone             m_options.m_start_idx.OptionWasSet(),
142b9c1b51eSKate Stone             m_options.m_start_idx.GetCurrentValue());
143b9c1b51eSKate Stone         std::pair<bool, uint64_t> stop_idx(
144b9c1b51eSKate Stone             m_options.m_stop_idx.OptionWasSet(),
145b9c1b51eSKate Stone             m_options.m_stop_idx.GetCurrentValue());
146b9c1b51eSKate Stone         std::pair<bool, uint64_t> count(m_options.m_count.OptionWasSet(),
147b9c1b51eSKate Stone                                         m_options.m_count.GetCurrentValue());
148a5a97ebeSJim Ingham 
1497594f14fSEnrico Granata         const CommandHistory &history(m_interpreter.GetCommandHistory());
1507594f14fSEnrico Granata 
151b9c1b51eSKate Stone         if (start_idx.first && start_idx.second == UINT64_MAX) {
152b9c1b51eSKate Stone           if (count.first) {
1537594f14fSEnrico Granata             start_idx.second = history.GetSize() - count.second;
1547594f14fSEnrico Granata             stop_idx.second = history.GetSize() - 1;
155b9c1b51eSKate Stone           } else if (stop_idx.first) {
1567594f14fSEnrico Granata             start_idx.second = stop_idx.second;
1577594f14fSEnrico Granata             stop_idx.second = history.GetSize() - 1;
158b9c1b51eSKate Stone           } else {
1597594f14fSEnrico Granata             start_idx.second = 0;
1607594f14fSEnrico Granata             stop_idx.second = history.GetSize() - 1;
1617594f14fSEnrico Granata           }
162b9c1b51eSKate Stone         } else {
163b9c1b51eSKate Stone           if (!start_idx.first && !stop_idx.first && !count.first) {
1647594f14fSEnrico Granata             start_idx.second = 0;
1657594f14fSEnrico Granata             stop_idx.second = history.GetSize() - 1;
166b9c1b51eSKate Stone           } else if (start_idx.first) {
167b9c1b51eSKate Stone             if (count.first) {
1687594f14fSEnrico Granata               stop_idx.second = start_idx.second + count.second - 1;
169b9c1b51eSKate Stone             } else if (!stop_idx.first) {
1707594f14fSEnrico Granata               stop_idx.second = history.GetSize() - 1;
1717594f14fSEnrico Granata             }
172b9c1b51eSKate Stone           } else if (stop_idx.first) {
173b9c1b51eSKate Stone             if (count.first) {
1747594f14fSEnrico Granata               if (stop_idx.second >= count.second)
1757594f14fSEnrico Granata                 start_idx.second = stop_idx.second - count.second + 1;
1767594f14fSEnrico Granata               else
1777594f14fSEnrico Granata                 start_idx.second = 0;
1787594f14fSEnrico Granata             }
179b9c1b51eSKate Stone           } else /* if (count.first) */
1807594f14fSEnrico Granata           {
1817594f14fSEnrico Granata             start_idx.second = 0;
1827594f14fSEnrico Granata             stop_idx.second = count.second - 1;
1837594f14fSEnrico Granata           }
1847594f14fSEnrico Granata         }
185b9c1b51eSKate Stone         history.Dump(result.GetOutputStream(), start_idx.second,
186b9c1b51eSKate Stone                      stop_idx.second);
1877594f14fSEnrico Granata       }
1887594f14fSEnrico Granata     }
189a5a97ebeSJim Ingham     return result.Succeeded();
190a5a97ebeSJim Ingham   }
1915a988416SJim Ingham 
1925a988416SJim Ingham   CommandOptions m_options;
193a5a97ebeSJim Ingham };
194a5a97ebeSJim Ingham 
195a5a97ebeSJim Ingham //-------------------------------------------------------------------------
196a5a97ebeSJim Ingham // CommandObjectCommandsSource
197a5a97ebeSJim Ingham //-------------------------------------------------------------------------
198a5a97ebeSJim Ingham 
1991f0f5b5bSZachary Turner static OptionDefinition g_source_options[] = {
2001f0f5b5bSZachary Turner     // clang-format off
2011f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "stop-on-error",    'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on error." },
2021f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on continue." },
2031f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "silent-run",       's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true don't echo commands while executing." },
2041f0f5b5bSZachary Turner     // clang-format on
2051f0f5b5bSZachary Turner };
2061f0f5b5bSZachary Turner 
207b9c1b51eSKate Stone class CommandObjectCommandsSource : public CommandObjectParsed {
2085a988416SJim Ingham public:
2097428a18cSKate Stone   CommandObjectCommandsSource(CommandInterpreter &interpreter)
210b9c1b51eSKate Stone       : CommandObjectParsed(
211b9c1b51eSKate Stone             interpreter, "command source",
212b9c1b51eSKate Stone             "Read and execute LLDB commands from the file <filename>.",
2136e3d8e7fSEugene Zelenko             nullptr),
214b9c1b51eSKate Stone         m_options() {
2155a988416SJim Ingham     CommandArgumentEntry arg;
2165a988416SJim Ingham     CommandArgumentData file_arg;
2175a988416SJim Ingham 
2185a988416SJim Ingham     // Define the first (and only) variant of this arg.
2195a988416SJim Ingham     file_arg.arg_type = eArgTypeFilename;
2205a988416SJim Ingham     file_arg.arg_repetition = eArgRepeatPlain;
2215a988416SJim Ingham 
222b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
223b9c1b51eSKate Stone     // argument entry.
2245a988416SJim Ingham     arg.push_back(file_arg);
2255a988416SJim Ingham 
2265a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
2275a988416SJim Ingham     m_arguments.push_back(arg);
2285a988416SJim Ingham   }
2295a988416SJim Ingham 
2306e3d8e7fSEugene Zelenko   ~CommandObjectCommandsSource() override = default;
2315a988416SJim Ingham 
232b9c1b51eSKate Stone   const char *GetRepeatCommand(Args &current_command_args,
233b9c1b51eSKate Stone                                uint32_t index) override {
2345a988416SJim Ingham     return "";
2355a988416SJim Ingham   }
2365a988416SJim Ingham 
237b9c1b51eSKate Stone   int HandleArgumentCompletion(Args &input, int &cursor_index,
2385a988416SJim Ingham                                int &cursor_char_position,
2395a988416SJim Ingham                                OptionElementVector &opt_element_vector,
240b9c1b51eSKate Stone                                int match_start_point, int max_return_elements,
2415a988416SJim Ingham                                bool &word_complete,
242b9c1b51eSKate Stone                                StringList &matches) override {
2434574a890SZachary Turner     auto completion_str = input[cursor_index].ref;
2444574a890SZachary Turner     completion_str = completion_str.take_front(cursor_char_position);
2455a988416SJim Ingham 
246b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
247b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
2484574a890SZachary Turner         completion_str, match_start_point, max_return_elements, nullptr,
249b9c1b51eSKate Stone         word_complete, matches);
2505a988416SJim Ingham     return matches.GetSize();
2515a988416SJim Ingham   }
2525a988416SJim Ingham 
253b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
2545a988416SJim Ingham 
2555a988416SJim Ingham protected:
256b9c1b51eSKate Stone   class CommandOptions : public Options {
257e16c50a1SJim Ingham   public:
258b9c1b51eSKate Stone     CommandOptions()
259b9c1b51eSKate Stone         : Options(), m_stop_on_error(true), m_silent_run(false),
260b9c1b51eSKate Stone           m_stop_on_continue(true) {}
261e16c50a1SJim Ingham 
2626e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
263e16c50a1SJim Ingham 
264fe11483bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
265b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
266e16c50a1SJim Ingham       Error error;
2673bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
268e16c50a1SJim Ingham 
269b9c1b51eSKate Stone       switch (short_option) {
270e16c50a1SJim Ingham       case 'e':
271fe11483bSZachary Turner         error = m_stop_on_error.SetValueFromString(option_arg);
272e16c50a1SJim Ingham         break;
273340b0309SGreg Clayton 
274e16c50a1SJim Ingham       case 'c':
275fe11483bSZachary Turner         error = m_stop_on_continue.SetValueFromString(option_arg);
276e16c50a1SJim Ingham         break;
277340b0309SGreg Clayton 
27860986174SMichael Sartain       case 's':
279fe11483bSZachary Turner         error = m_silent_run.SetValueFromString(option_arg);
28060986174SMichael Sartain         break;
281340b0309SGreg Clayton 
282e16c50a1SJim Ingham       default:
283b9c1b51eSKate Stone         error.SetErrorStringWithFormat("unrecognized option '%c'",
284b9c1b51eSKate Stone                                        short_option);
285e16c50a1SJim Ingham         break;
286e16c50a1SJim Ingham       }
287e16c50a1SJim Ingham 
288e16c50a1SJim Ingham       return error;
289e16c50a1SJim Ingham     }
290e16c50a1SJim Ingham 
291b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
292012d4fcaSEnrico Granata       m_stop_on_error.Clear();
293340b0309SGreg Clayton       m_silent_run.Clear();
294340b0309SGreg Clayton       m_stop_on_continue.Clear();
295e16c50a1SJim Ingham     }
296e16c50a1SJim Ingham 
2971f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
29870602439SZachary Turner       return llvm::makeArrayRef(g_source_options);
2991f0f5b5bSZachary Turner     }
300e16c50a1SJim Ingham 
301e16c50a1SJim Ingham     // Instance variables to hold the values for command options.
302e16c50a1SJim Ingham 
303012d4fcaSEnrico Granata     OptionValueBoolean m_stop_on_error;
304340b0309SGreg Clayton     OptionValueBoolean m_silent_run;
305340b0309SGreg Clayton     OptionValueBoolean m_stop_on_continue;
306e16c50a1SJim Ingham   };
307e16c50a1SJim Ingham 
308b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
3094574a890SZachary Turner     if (command.GetArgumentCount() != 1) {
3104574a890SZachary Turner       result.AppendErrorWithFormat(
3114574a890SZachary Turner           "'%s' takes exactly one executable filename argument.\n",
3124574a890SZachary Turner           GetCommandName().str().c_str());
3134574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
3144574a890SZachary Turner       return false;
3154574a890SZachary Turner     }
316ebc09c36SJim Ingham 
3174574a890SZachary Turner     FileSpec cmd_file(command[0].ref, true);
3186e3d8e7fSEugene Zelenko     ExecutionContext *exe_ctx = nullptr; // Just use the default context.
319ebc09c36SJim Ingham 
320340b0309SGreg Clayton     // If any options were set, then use them
321340b0309SGreg Clayton     if (m_options.m_stop_on_error.OptionWasSet() ||
322340b0309SGreg Clayton         m_options.m_silent_run.OptionWasSet() ||
323b9c1b51eSKate Stone         m_options.m_stop_on_continue.OptionWasSet()) {
324340b0309SGreg Clayton       // Use user set settings
32526c7bf93SJim Ingham       CommandInterpreterRunOptions options;
3264574a890SZachary Turner       options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue());
32726c7bf93SJim Ingham       options.SetStopOnError(m_options.m_stop_on_error.GetCurrentValue());
3287d8555c4SJim Ingham       options.SetEchoCommands(!m_options.m_silent_run.GetCurrentValue());
3297d8555c4SJim Ingham       options.SetPrintResults(!m_options.m_silent_run.GetCurrentValue());
33026c7bf93SJim Ingham 
3314574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
332b9c1b51eSKate Stone     } else {
333b9c1b51eSKate Stone       // No options were set, inherit any settings from nested "command
334b9c1b51eSKate Stone       // source" commands,
335340b0309SGreg Clayton       // or set to sane default settings...
33626c7bf93SJim Ingham       CommandInterpreterRunOptions options;
3374574a890SZachary Turner       m_interpreter.HandleCommandsFromFile(cmd_file, exe_ctx, options, result);
338ebc09c36SJim Ingham     }
339ebc09c36SJim Ingham     return result.Succeeded();
340ebc09c36SJim Ingham   }
3416e3d8e7fSEugene Zelenko 
3425a988416SJim Ingham   CommandOptions m_options;
343ebc09c36SJim Ingham };
344ebc09c36SJim Ingham 
345ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
346ebc09c36SJim Ingham //-------------------------------------------------------------------------
347ebc09c36SJim Ingham // CommandObjectCommandsAlias
348ebc09c36SJim Ingham //-------------------------------------------------------------------------
349ebc09c36SJim Ingham 
3501f0f5b5bSZachary Turner static OptionDefinition g_alias_options[] = {
3511f0f5b5bSZachary Turner     // clang-format off
3521f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "help",      'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "Help text for this command" },
3531f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "long-help", 'H', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "Long help text for this command" },
3541f0f5b5bSZachary Turner     // clang-format on
3551f0f5b5bSZachary Turner };
3561f0f5b5bSZachary Turner 
357b9c1b51eSKate Stone static const char *g_python_command_instructions =
358b9c1b51eSKate Stone     "Enter your Python command(s). Type 'DONE' to end.\n"
359be93a35aSEnrico Granata     "You must define a Python function with this signature:\n"
36044d93782SGreg Clayton     "def my_command_impl(debugger, args, result, internal_dict):\n";
361be93a35aSEnrico Granata 
362b9c1b51eSKate Stone class CommandObjectCommandsAlias : public CommandObjectRaw {
36345d0e238SEnrico Granata protected:
364b9c1b51eSKate Stone   class CommandOptions : public OptionGroup {
365ebc09c36SJim Ingham   public:
366b9c1b51eSKate Stone     CommandOptions() : OptionGroup(), m_help(), m_long_help() {}
36745d0e238SEnrico Granata 
36845d0e238SEnrico Granata     ~CommandOptions() override = default;
36945d0e238SEnrico Granata 
3701f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
37170602439SZachary Turner       return llvm::makeArrayRef(g_alias_options);
3721f0f5b5bSZachary Turner     }
37345d0e238SEnrico Granata 
3748cef4b0bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_value,
375b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
37645d0e238SEnrico Granata       Error error;
37745d0e238SEnrico Granata 
3781f0f5b5bSZachary Turner       const int short_option = GetDefinitions()[option_idx].short_option;
3798cef4b0bSZachary Turner       std::string option_str(option_value);
38045d0e238SEnrico Granata 
381b9c1b51eSKate Stone       switch (short_option) {
38245d0e238SEnrico Granata       case 'h':
3838cef4b0bSZachary Turner         m_help.SetCurrentValue(option_str);
38445d0e238SEnrico Granata         m_help.SetOptionWasSet();
38545d0e238SEnrico Granata         break;
38645d0e238SEnrico Granata 
38745d0e238SEnrico Granata       case 'H':
3888cef4b0bSZachary Turner         m_long_help.SetCurrentValue(option_str);
38945d0e238SEnrico Granata         m_long_help.SetOptionWasSet();
39045d0e238SEnrico Granata         break;
39145d0e238SEnrico Granata 
39245d0e238SEnrico Granata       default:
393b9c1b51eSKate Stone         error.SetErrorStringWithFormat("invalid short option character '%c'",
394b9c1b51eSKate Stone                                        short_option);
39545d0e238SEnrico Granata         break;
39645d0e238SEnrico Granata       }
39745d0e238SEnrico Granata 
39845d0e238SEnrico Granata       return error;
39945d0e238SEnrico Granata     }
40045d0e238SEnrico Granata 
401b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
40245d0e238SEnrico Granata       m_help.Clear();
40345d0e238SEnrico Granata       m_long_help.Clear();
40445d0e238SEnrico Granata     }
40545d0e238SEnrico Granata 
40645d0e238SEnrico Granata     OptionValueString m_help;
40745d0e238SEnrico Granata     OptionValueString m_long_help;
40845d0e238SEnrico Granata   };
40945d0e238SEnrico Granata 
41045d0e238SEnrico Granata   OptionGroupOptions m_option_group;
41145d0e238SEnrico Granata   CommandOptions m_command_options;
41245d0e238SEnrico Granata 
41345d0e238SEnrico Granata public:
414b9c1b51eSKate Stone   Options *GetOptions() override { return &m_option_group; }
41545d0e238SEnrico Granata 
4167428a18cSKate Stone   CommandObjectCommandsAlias(CommandInterpreter &interpreter)
417b9c1b51eSKate Stone       : CommandObjectRaw(
418b9c1b51eSKate Stone             interpreter, "command alias",
419a449698cSZachary Turner             "Define a custom command in terms of an existing command."),
420b9c1b51eSKate Stone         m_option_group(), m_command_options() {
42145d0e238SEnrico Granata     m_option_group.Append(&m_command_options);
42245d0e238SEnrico Granata     m_option_group.Finalize();
42345d0e238SEnrico Granata 
424ebc09c36SJim Ingham     SetHelpLong(
425ea671fbdSKate Stone         "'alias' allows the user to create a short-cut or abbreviation for long \
426ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
427b9c1b51eSKate Stone Below are some simple examples of how one might use the 'alias' command:"
428b9c1b51eSKate Stone         R"(
429ea671fbdSKate Stone 
430ea671fbdSKate Stone (lldb) command alias sc script
431ea671fbdSKate Stone 
432ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
433ea671fbdSKate Stone 
434ea671fbdSKate Stone (lldb) command alias bp breakpoint
435ea671fbdSKate Stone 
436b9c1b51eSKate Stone )"
437b9c1b51eSKate Stone         "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
438ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
439b9c1b51eSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'."
440b9c1b51eSKate Stone         R"(
441ea671fbdSKate Stone 
442ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
443ea671fbdSKate Stone 
444ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
445ea671fbdSKate Stone 
446b9c1b51eSKate Stone )"
447b9c1b51eSKate Stone         "An alias can include some options for the command, with the values either \
448ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
449ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
450b9c1b51eSKate Stone shows how to create aliases with options:"
451b9c1b51eSKate Stone         R"(
452ea671fbdSKate Stone 
453ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
454ea671fbdSKate Stone 
455b9c1b51eSKate Stone )"
456b9c1b51eSKate Stone         "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
457ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
458ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
459ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
460ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
461ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
462ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
463ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
464ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
465ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
466b9c1b51eSKate Stone launch' example below)."
467b9c1b51eSKate Stone         R"(
468ea671fbdSKate Stone 
469b9c1b51eSKate Stone )"
470b9c1b51eSKate Stone         "Note: the positional arguments must substitute as whole words in the resultant \
471ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
472b9c1b51eSKate Stone \".cpp\":"
473b9c1b51eSKate Stone         R"(
474ea671fbdSKate Stone 
475ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
476ea671fbdSKate Stone 
477b9c1b51eSKate Stone )"
478b9c1b51eSKate Stone         "For more complex aliasing, use the \"command regex\" command instead.  In the \
479ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
480ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
481b9c1b51eSKate Stone argument.  The user would use this alias as follows:"
482b9c1b51eSKate Stone         R"(
483ea671fbdSKate Stone 
484ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
485ea671fbdSKate Stone (lldb) bfl my-file.c 137
486ea671fbdSKate Stone 
487ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
488ea671fbdSKate Stone 
489ea671fbdSKate Stone Another example:
490ea671fbdSKate Stone 
491ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
492ea671fbdSKate Stone (lldb) pltty /dev/tty0
493ea671fbdSKate Stone 
494ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
495ea671fbdSKate Stone 
496b9c1b51eSKate Stone )"
497b9c1b51eSKate Stone         "If the user always wanted to pass the same value to a particular option, the \
498ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
499b9c1b51eSKate Stone rather than using a positional placeholder:"
500b9c1b51eSKate Stone         R"(
501ea671fbdSKate Stone 
502ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
503ea671fbdSKate Stone 
504b9c1b51eSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)");
505ebc09c36SJim Ingham 
506405fe67fSCaroline Tice     CommandArgumentEntry arg1;
507405fe67fSCaroline Tice     CommandArgumentEntry arg2;
508405fe67fSCaroline Tice     CommandArgumentEntry arg3;
509405fe67fSCaroline Tice     CommandArgumentData alias_arg;
510405fe67fSCaroline Tice     CommandArgumentData cmd_arg;
511405fe67fSCaroline Tice     CommandArgumentData options_arg;
512405fe67fSCaroline Tice 
513405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
514405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
515405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
516405fe67fSCaroline Tice 
517b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
518b9c1b51eSKate Stone     // argument entry.
519405fe67fSCaroline Tice     arg1.push_back(alias_arg);
520405fe67fSCaroline Tice 
521405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
522405fe67fSCaroline Tice     cmd_arg.arg_type = eArgTypeCommandName;
523405fe67fSCaroline Tice     cmd_arg.arg_repetition = eArgRepeatPlain;
524405fe67fSCaroline Tice 
525b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
526b9c1b51eSKate Stone     // argument entry.
527405fe67fSCaroline Tice     arg2.push_back(cmd_arg);
528405fe67fSCaroline Tice 
529405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
530405fe67fSCaroline Tice     options_arg.arg_type = eArgTypeAliasOptions;
531405fe67fSCaroline Tice     options_arg.arg_repetition = eArgRepeatOptional;
532405fe67fSCaroline Tice 
533b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
534b9c1b51eSKate Stone     // argument entry.
535405fe67fSCaroline Tice     arg3.push_back(options_arg);
536405fe67fSCaroline Tice 
537405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
538405fe67fSCaroline Tice     m_arguments.push_back(arg1);
539405fe67fSCaroline Tice     m_arguments.push_back(arg2);
540405fe67fSCaroline Tice     m_arguments.push_back(arg3);
541ebc09c36SJim Ingham   }
542ebc09c36SJim Ingham 
5436e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAlias() override = default;
544ebc09c36SJim Ingham 
5455a988416SJim Ingham protected:
546b9c1b51eSKate Stone   bool DoExecute(const char *raw_command_line,
547b9c1b51eSKate Stone                  CommandReturnObject &result) override {
548b9c1b51eSKate Stone     if (!raw_command_line || !raw_command_line[0]) {
549d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
55045d0e238SEnrico Granata       return false;
55145d0e238SEnrico Granata     }
55245d0e238SEnrico Granata 
553e1cfbc79STodd Fiala     ExecutionContext exe_ctx = GetCommandInterpreter().GetExecutionContext();
554e1cfbc79STodd Fiala     m_option_group.NotifyOptionParsingStarting(&exe_ctx);
55545d0e238SEnrico Granata 
55645d0e238SEnrico Granata     const char *remainder = nullptr;
55745d0e238SEnrico Granata 
558b9c1b51eSKate Stone     if (raw_command_line[0] == '-') {
55945d0e238SEnrico Granata       // We have some options and these options MUST end with --.
56045d0e238SEnrico Granata       const char *end_options = nullptr;
56145d0e238SEnrico Granata       const char *s = raw_command_line;
562b9c1b51eSKate Stone       while (s && s[0]) {
56345d0e238SEnrico Granata         end_options = ::strstr(s, "--");
564b9c1b51eSKate Stone         if (end_options) {
56545d0e238SEnrico Granata           end_options += 2; // Get past the "--"
566b9c1b51eSKate Stone           if (::isspace(end_options[0])) {
56745d0e238SEnrico Granata             remainder = end_options;
56845d0e238SEnrico Granata             while (::isspace(*remainder))
56945d0e238SEnrico Granata               ++remainder;
57045d0e238SEnrico Granata             break;
57145d0e238SEnrico Granata           }
57245d0e238SEnrico Granata         }
57345d0e238SEnrico Granata         s = end_options;
57445d0e238SEnrico Granata       }
57545d0e238SEnrico Granata 
576b9c1b51eSKate Stone       if (end_options) {
577b9c1b51eSKate Stone         Args args(
578b9c1b51eSKate Stone             llvm::StringRef(raw_command_line, end_options - raw_command_line));
57945d0e238SEnrico Granata         if (!ParseOptions(args, result))
58045d0e238SEnrico Granata           return false;
58145d0e238SEnrico Granata 
582e1cfbc79STodd Fiala         Error error(m_option_group.NotifyOptionParsingFinished(&exe_ctx));
583b9c1b51eSKate Stone         if (error.Fail()) {
58445d0e238SEnrico Granata           result.AppendError(error.AsCString());
58545d0e238SEnrico Granata           result.SetStatus(eReturnStatusFailed);
58645d0e238SEnrico Granata           return false;
58745d0e238SEnrico Granata         }
58845d0e238SEnrico Granata       }
58945d0e238SEnrico Granata     }
59045d0e238SEnrico Granata     if (nullptr == remainder)
59145d0e238SEnrico Granata       remainder = raw_command_line;
59245d0e238SEnrico Granata 
593a01bccdbSZachary Turner     llvm::StringRef raw_command_string(remainder);
594a01bccdbSZachary Turner     Args args(raw_command_string);
595844d2303SCaroline Tice 
59611eb9c64SZachary Turner     if (args.GetArgumentCount() < 2) {
597d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
598844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
599844d2303SCaroline Tice       return false;
600844d2303SCaroline Tice     }
601844d2303SCaroline Tice 
602844d2303SCaroline Tice     // Get the alias command.
603844d2303SCaroline Tice 
6044574a890SZachary Turner     auto alias_command = args[0].ref;
6054574a890SZachary Turner     if (alias_command.startswith("-")) {
606d72e412fSEnrico Granata       result.AppendError("aliases starting with a dash are not supported");
607b9c1b51eSKate Stone       if (alias_command == "--help" || alias_command == "--long-help") {
608b9c1b51eSKate Stone         result.AppendWarning("if trying to pass options to 'command alias' add "
609b9c1b51eSKate Stone                              "a -- at the end of the options");
610d72e412fSEnrico Granata       }
611d72e412fSEnrico Granata       result.SetStatus(eReturnStatusFailed);
612d72e412fSEnrico Granata       return false;
613d72e412fSEnrico Granata     }
614844d2303SCaroline Tice 
615b9c1b51eSKate Stone     // Strip the new alias name off 'raw_command_string'  (leave it on args,
616b9c1b51eSKate Stone     // which gets passed to 'Execute', which
617844d2303SCaroline Tice     // does the stripping itself.
618844d2303SCaroline Tice     size_t pos = raw_command_string.find(alias_command);
619b9c1b51eSKate Stone     if (pos == 0) {
620844d2303SCaroline Tice       raw_command_string = raw_command_string.substr(alias_command.size());
621844d2303SCaroline Tice       pos = raw_command_string.find_first_not_of(' ');
622844d2303SCaroline Tice       if ((pos != std::string::npos) && (pos > 0))
623844d2303SCaroline Tice         raw_command_string = raw_command_string.substr(pos);
624b9c1b51eSKate Stone     } else {
625844d2303SCaroline Tice       result.AppendError("Error parsing command string.  No alias created.");
626844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
627844d2303SCaroline Tice       return false;
628844d2303SCaroline Tice     }
629844d2303SCaroline Tice 
630844d2303SCaroline Tice     // Verify that the command is alias-able.
631771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
632b9c1b51eSKate Stone       result.AppendErrorWithFormat(
633b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
6344574a890SZachary Turner           args[0].c_str());
635844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
636844d2303SCaroline Tice       return false;
637844d2303SCaroline Tice     }
638844d2303SCaroline Tice 
639b9c1b51eSKate Stone     // Get CommandObject that is being aliased. The command name is read from
640a01bccdbSZachary Turner     // the front of raw_command_string. raw_command_string is returned with the
641a01bccdbSZachary Turner     // name of the command object stripped off the front.
642a01bccdbSZachary Turner     llvm::StringRef original_raw_command_string = raw_command_string;
643b9c1b51eSKate Stone     CommandObject *cmd_obj =
644b9c1b51eSKate Stone         m_interpreter.GetCommandObjectForCommand(raw_command_string);
645844d2303SCaroline Tice 
646b9c1b51eSKate Stone     if (!cmd_obj) {
647b9c1b51eSKate Stone       result.AppendErrorWithFormat("invalid command given to 'command alias'. "
648b9c1b51eSKate Stone                                    "'%s' does not begin with a valid command."
649b9c1b51eSKate Stone                                    "  No alias created.",
650a01bccdbSZachary Turner                                    original_raw_command_string.str().c_str());
651844d2303SCaroline Tice       result.SetStatus(eReturnStatusFailed);
652844d2303SCaroline Tice       return false;
653b9c1b51eSKate Stone     } else if (!cmd_obj->WantsRawCommandString()) {
654b9c1b51eSKate Stone       // Note that args was initialized with the original command, and has not
655b9c1b51eSKate Stone       // been updated to this point.
656b9c1b51eSKate Stone       // Therefore can we pass it to the version of Execute that does not
657b9c1b51eSKate Stone       // need/expect raw input in the alias.
6585a988416SJim Ingham       return HandleAliasingNormalCommand(args, result);
659b9c1b51eSKate Stone     } else {
660b9c1b51eSKate Stone       return HandleAliasingRawCommand(alias_command, raw_command_string,
661b9c1b51eSKate Stone                                       *cmd_obj, result);
6625a988416SJim Ingham     }
6635a988416SJim Ingham     return result.Succeeded();
6645a988416SJim Ingham   }
6655a988416SJim Ingham 
666a01bccdbSZachary Turner   bool HandleAliasingRawCommand(llvm::StringRef alias_command,
667a01bccdbSZachary Turner                                 llvm::StringRef raw_command_string,
668b9c1b51eSKate Stone                                 CommandObject &cmd_obj,
669b9c1b51eSKate Stone                                 CommandReturnObject &result) {
670844d2303SCaroline Tice     // Verify & handle any options/arguments passed to the alias command
671844d2303SCaroline Tice 
672b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
673b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
674844d2303SCaroline Tice 
675b9c1b51eSKate Stone     if (CommandObjectSP cmd_obj_sp =
676b9c1b51eSKate Stone             m_interpreter.GetCommandSPExact(cmd_obj.GetCommandName(), false)) {
677a01bccdbSZachary Turner       if (m_interpreter.AliasExists(alias_command) ||
678a01bccdbSZachary Turner           m_interpreter.UserCommandExists(alias_command)) {
679b9c1b51eSKate Stone         result.AppendWarningWithFormat(
680b9c1b51eSKate Stone             "Overwriting existing definition for '%s'.\n",
681a01bccdbSZachary Turner             alias_command.str().c_str());
682844d2303SCaroline Tice       }
683b9c1b51eSKate Stone       if (CommandAlias *alias = m_interpreter.AddAlias(
684a01bccdbSZachary Turner               alias_command, cmd_obj_sp, raw_command_string)) {
68545d0e238SEnrico Granata         if (m_command_options.m_help.OptionWasSet())
68645d0e238SEnrico Granata           alias->SetHelp(m_command_options.m_help.GetCurrentValue());
68745d0e238SEnrico Granata         if (m_command_options.m_long_help.OptionWasSet())
68845d0e238SEnrico Granata           alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
689844d2303SCaroline Tice         result.SetStatus(eReturnStatusSuccessFinishNoResult);
690b9c1b51eSKate Stone       } else {
691472362e6SCaroline Tice         result.AppendError("Unable to create requested alias.\n");
692472362e6SCaroline Tice         result.SetStatus(eReturnStatusFailed);
693472362e6SCaroline Tice       }
694212130acSEnrico Granata 
695b9c1b51eSKate Stone     } else {
696212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
697212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
698212130acSEnrico Granata     }
699212130acSEnrico Granata 
700844d2303SCaroline Tice     return result.Succeeded();
701844d2303SCaroline Tice   }
702ebc09c36SJim Ingham 
703b9c1b51eSKate Stone   bool HandleAliasingNormalCommand(Args &args, CommandReturnObject &result) {
704867b185dSCaroline Tice     size_t argc = args.GetArgumentCount();
705ebc09c36SJim Ingham 
706b9c1b51eSKate Stone     if (argc < 2) {
707d72e412fSEnrico Granata       result.AppendError("'command alias' requires at least two arguments");
708ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
709ebc09c36SJim Ingham       return false;
710ebc09c36SJim Ingham     }
711ebc09c36SJim Ingham 
7124574a890SZachary Turner     // Save these in std::strings since we're going to shift them off.
7134574a890SZachary Turner     const std::string alias_command(args[0].ref);
7144574a890SZachary Turner     const std::string actual_command(args[1].ref);
715ebc09c36SJim Ingham 
716ebc09c36SJim Ingham     args.Shift(); // Shift the alias command word off the argument vector.
717ebc09c36SJim Ingham     args.Shift(); // Shift the old command word off the argument vector.
718ebc09c36SJim Ingham 
719b9c1b51eSKate Stone     // Verify that the command is alias'able, and get the appropriate command
720b9c1b51eSKate Stone     // object.
721ebc09c36SJim Ingham 
722771ef6d4SMalcolm Parsons     if (m_interpreter.CommandExists(alias_command)) {
723b9c1b51eSKate Stone       result.AppendErrorWithFormat(
724b9c1b51eSKate Stone           "'%s' is a permanent debugger command and cannot be redefined.\n",
725ebc09c36SJim Ingham           alias_command.c_str());
726ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
7274574a890SZachary Turner       return false;
7284574a890SZachary Turner     }
7294574a890SZachary Turner 
730b9c1b51eSKate Stone     CommandObjectSP command_obj_sp(
731a449698cSZachary Turner         m_interpreter.GetCommandSPExact(actual_command, true));
732ebc09c36SJim Ingham     CommandObjectSP subcommand_obj_sp;
733ebc09c36SJim Ingham     bool use_subcommand = false;
7344574a890SZachary Turner     if (!command_obj_sp) {
7354574a890SZachary Turner       result.AppendErrorWithFormat("'%s' is not an existing command.\n",
7364574a890SZachary Turner                                    actual_command.c_str());
7374574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
7384574a890SZachary Turner       return false;
7394574a890SZachary Turner     }
740ebc09c36SJim Ingham     CommandObject *cmd_obj = command_obj_sp.get();
7416e3d8e7fSEugene Zelenko     CommandObject *sub_cmd_obj = nullptr;
742b9c1b51eSKate Stone     OptionArgVectorSP option_arg_vector_sp =
743b9c1b51eSKate Stone         OptionArgVectorSP(new OptionArgVector);
744ebc09c36SJim Ingham 
74511eb9c64SZachary Turner     while (cmd_obj->IsMultiwordObject() && !args.empty()) {
7464574a890SZachary Turner       auto sub_command = args[0].ref;
74711eb9c64SZachary Turner       assert(!sub_command.empty());
7484574a890SZachary Turner       subcommand_obj_sp = cmd_obj->GetSubcommandSP(sub_command);
7494574a890SZachary Turner       if (!subcommand_obj_sp) {
750b9c1b51eSKate Stone         result.AppendErrorWithFormat(
751b9c1b51eSKate Stone             "'%s' is not a valid sub-command of '%s'.  "
752f415eeb4SCaroline Tice             "Unable to create alias.\n",
7534574a890SZachary Turner             args[0].c_str(), actual_command.c_str());
754ebc09c36SJim Ingham         result.SetStatus(eReturnStatusFailed);
755ebc09c36SJim Ingham         return false;
756ebc09c36SJim Ingham       }
7574574a890SZachary Turner 
7584574a890SZachary Turner       sub_cmd_obj = subcommand_obj_sp.get();
7594574a890SZachary Turner       use_subcommand = true;
7604574a890SZachary Turner       args.Shift(); // Shift the sub_command word off the argument vector.
7614574a890SZachary Turner       cmd_obj = sub_cmd_obj;
762ebc09c36SJim Ingham     }
763ebc09c36SJim Ingham 
764ebc09c36SJim Ingham     // Verify & handle any options/arguments passed to the alias command
765ebc09c36SJim Ingham 
766212130acSEnrico Granata     std::string args_string;
767212130acSEnrico Granata 
76811eb9c64SZachary Turner     if (!args.empty()) {
769b9c1b51eSKate Stone       CommandObjectSP tmp_sp =
770b9c1b51eSKate Stone           m_interpreter.GetCommandSPExact(cmd_obj->GetCommandName(), false);
771ebc09c36SJim Ingham       if (use_subcommand)
7724574a890SZachary Turner         tmp_sp = m_interpreter.GetCommandSPExact(sub_cmd_obj->GetCommandName(),
7734574a890SZachary Turner                                                  false);
774ca90c47eSCaroline Tice 
775ca90c47eSCaroline Tice       args.GetCommandString(args_string);
776867b185dSCaroline Tice     }
777ebc09c36SJim Ingham 
778771ef6d4SMalcolm Parsons     if (m_interpreter.AliasExists(alias_command) ||
779771ef6d4SMalcolm Parsons         m_interpreter.UserCommandExists(alias_command)) {
780b9c1b51eSKate Stone       result.AppendWarningWithFormat(
7814574a890SZachary Turner           "Overwriting existing definition for '%s'.\n", alias_command.c_str());
782ebc09c36SJim Ingham     }
783ebc09c36SJim Ingham 
784b9c1b51eSKate Stone     if (CommandAlias *alias = m_interpreter.AddAlias(
7854574a890SZachary Turner             alias_command, use_subcommand ? subcommand_obj_sp : command_obj_sp,
786771ef6d4SMalcolm Parsons             args_string)) {
78745d0e238SEnrico Granata       if (m_command_options.m_help.OptionWasSet())
78845d0e238SEnrico Granata         alias->SetHelp(m_command_options.m_help.GetCurrentValue());
78945d0e238SEnrico Granata       if (m_command_options.m_long_help.OptionWasSet())
79045d0e238SEnrico Granata         alias->SetHelpLong(m_command_options.m_long_help.GetCurrentValue());
791ebc09c36SJim Ingham       result.SetStatus(eReturnStatusSuccessFinishNoResult);
792b9c1b51eSKate Stone     } else {
793212130acSEnrico Granata       result.AppendError("Unable to create requested alias.\n");
794212130acSEnrico Granata       result.SetStatus(eReturnStatusFailed);
795212130acSEnrico Granata       return false;
796212130acSEnrico Granata     }
797ebc09c36SJim Ingham 
798ebc09c36SJim Ingham     return result.Succeeded();
799ebc09c36SJim Ingham   }
800ebc09c36SJim Ingham };
801ebc09c36SJim Ingham 
802ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
803ebc09c36SJim Ingham //-------------------------------------------------------------------------
804ebc09c36SJim Ingham // CommandObjectCommandsUnalias
805ebc09c36SJim Ingham //-------------------------------------------------------------------------
806ebc09c36SJim Ingham 
807b9c1b51eSKate Stone class CommandObjectCommandsUnalias : public CommandObjectParsed {
808ebc09c36SJim Ingham public:
8097428a18cSKate Stone   CommandObjectCommandsUnalias(CommandInterpreter &interpreter)
810b9c1b51eSKate Stone       : CommandObjectParsed(
811b9c1b51eSKate Stone             interpreter, "command unalias",
812b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command alias'.",
813b9c1b51eSKate Stone             nullptr) {
814405fe67fSCaroline Tice     CommandArgumentEntry arg;
815405fe67fSCaroline Tice     CommandArgumentData alias_arg;
816405fe67fSCaroline Tice 
817405fe67fSCaroline Tice     // Define the first (and only) variant of this arg.
818405fe67fSCaroline Tice     alias_arg.arg_type = eArgTypeAliasName;
819405fe67fSCaroline Tice     alias_arg.arg_repetition = eArgRepeatPlain;
820405fe67fSCaroline Tice 
821b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
822b9c1b51eSKate Stone     // argument entry.
823405fe67fSCaroline Tice     arg.push_back(alias_arg);
824405fe67fSCaroline Tice 
825405fe67fSCaroline Tice     // Push the data for the first argument into the m_arguments vector.
826405fe67fSCaroline Tice     m_arguments.push_back(arg);
827ebc09c36SJim Ingham   }
828ebc09c36SJim Ingham 
8296e3d8e7fSEugene Zelenko   ~CommandObjectCommandsUnalias() override = default;
830ebc09c36SJim Ingham 
8315a988416SJim Ingham protected:
832b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
833ebc09c36SJim Ingham     CommandObject::CommandMap::iterator pos;
834ebc09c36SJim Ingham     CommandObject *cmd_obj;
835ebc09c36SJim Ingham 
83611eb9c64SZachary Turner     if (args.empty()) {
83711eb9c64SZachary Turner       result.AppendError("must call 'unalias' with a valid alias");
83811eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
83911eb9c64SZachary Turner       return false;
84011eb9c64SZachary Turner     }
84111eb9c64SZachary Turner 
8424574a890SZachary Turner     auto command_name = args[0].ref;
843a7015092SGreg Clayton     cmd_obj = m_interpreter.GetCommandObject(command_name);
8444574a890SZachary Turner     if (!cmd_obj) {
8454574a890SZachary Turner       result.AppendErrorWithFormat(
8464574a890SZachary Turner           "'%s' is not a known command.\nTry 'help' to see a "
8474574a890SZachary Turner           "current list of commands.\n",
848867e7d17SZachary Turner           args[0].c_str());
8494574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
8504574a890SZachary Turner       return false;
8514574a890SZachary Turner     }
8524574a890SZachary Turner 
853b9c1b51eSKate Stone     if (m_interpreter.CommandExists(command_name)) {
854b9c1b51eSKate Stone       if (cmd_obj->IsRemovable()) {
855b9c1b51eSKate Stone         result.AppendErrorWithFormat(
856b9c1b51eSKate Stone             "'%s' is not an alias, it is a debugger command which can be "
857b9c1b51eSKate Stone             "removed using the 'command delete' command.\n",
858867e7d17SZachary Turner             args[0].c_str());
859b9c1b51eSKate Stone       } else {
860b9c1b51eSKate Stone         result.AppendErrorWithFormat(
861b9c1b51eSKate Stone             "'%s' is a permanent debugger command and cannot be removed.\n",
862867e7d17SZachary Turner             args[0].c_str());
863b547278cSGreg Clayton       }
864ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
8654574a890SZachary Turner       return false;
8664574a890SZachary Turner     }
8674574a890SZachary Turner 
868b9c1b51eSKate Stone     if (!m_interpreter.RemoveAlias(command_name)) {
869a7015092SGreg Clayton       if (m_interpreter.AliasExists(command_name))
870b9c1b51eSKate Stone         result.AppendErrorWithFormat(
871867e7d17SZachary Turner             "Error occurred while attempting to unalias '%s'.\n",
872867e7d17SZachary Turner             args[0].c_str());
873ebc09c36SJim Ingham       else
874b9c1b51eSKate Stone         result.AppendErrorWithFormat("'%s' is not an existing alias.\n",
875867e7d17SZachary Turner                                      args[0].c_str());
876ebc09c36SJim Ingham       result.SetStatus(eReturnStatusFailed);
8774574a890SZachary Turner       return false;
878ebc09c36SJim Ingham     }
879ebc09c36SJim Ingham 
8804574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
881ebc09c36SJim Ingham     return result.Succeeded();
882ebc09c36SJim Ingham   }
883ebc09c36SJim Ingham };
884ebc09c36SJim Ingham 
885b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
886b547278cSGreg Clayton //-------------------------------------------------------------------------
887b547278cSGreg Clayton // CommandObjectCommandsDelete
888b547278cSGreg Clayton //-------------------------------------------------------------------------
889b547278cSGreg Clayton 
890b9c1b51eSKate Stone class CommandObjectCommandsDelete : public CommandObjectParsed {
891b547278cSGreg Clayton public:
8927428a18cSKate Stone   CommandObjectCommandsDelete(CommandInterpreter &interpreter)
893b9c1b51eSKate Stone       : CommandObjectParsed(
894b9c1b51eSKate Stone             interpreter, "command delete",
895b9c1b51eSKate Stone             "Delete one or more custom commands defined by 'command regex'.",
896b9c1b51eSKate Stone             nullptr) {
897b547278cSGreg Clayton     CommandArgumentEntry arg;
898b547278cSGreg Clayton     CommandArgumentData alias_arg;
899b547278cSGreg Clayton 
900b547278cSGreg Clayton     // Define the first (and only) variant of this arg.
901b547278cSGreg Clayton     alias_arg.arg_type = eArgTypeCommandName;
902b547278cSGreg Clayton     alias_arg.arg_repetition = eArgRepeatPlain;
903b547278cSGreg Clayton 
904b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
905b9c1b51eSKate Stone     // argument entry.
906b547278cSGreg Clayton     arg.push_back(alias_arg);
907b547278cSGreg Clayton 
908b547278cSGreg Clayton     // Push the data for the first argument into the m_arguments vector.
909b547278cSGreg Clayton     m_arguments.push_back(arg);
910b547278cSGreg Clayton   }
911b547278cSGreg Clayton 
9126e3d8e7fSEugene Zelenko   ~CommandObjectCommandsDelete() override = default;
913b547278cSGreg Clayton 
914b547278cSGreg Clayton protected:
915b9c1b51eSKate Stone   bool DoExecute(Args &args, CommandReturnObject &result) override {
916b547278cSGreg Clayton     CommandObject::CommandMap::iterator pos;
917b547278cSGreg Clayton 
91811eb9c64SZachary Turner     if (args.empty()) {
91911eb9c64SZachary Turner       result.AppendErrorWithFormat("must call '%s' with one or more valid user "
92011eb9c64SZachary Turner                                    "defined regular expression command names",
921a449698cSZachary Turner                                    GetCommandName().str().c_str());
92211eb9c64SZachary Turner       result.SetStatus(eReturnStatusFailed);
92311eb9c64SZachary Turner     }
92411eb9c64SZachary Turner 
9254574a890SZachary Turner     auto command_name = args[0].ref;
9264574a890SZachary Turner     if (!m_interpreter.CommandExists(command_name)) {
92746d4aa21SEnrico Granata       StreamString error_msg_stream;
92846d4aa21SEnrico Granata       const bool generate_apropos = true;
92946d4aa21SEnrico Granata       const bool generate_type_lookup = false;
930b9c1b51eSKate Stone       CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(
9314574a890SZachary Turner           &error_msg_stream, command_name, llvm::StringRef(), llvm::StringRef(),
9324574a890SZachary Turner           generate_apropos, generate_type_lookup);
933c156427dSZachary Turner       result.AppendError(error_msg_stream.GetString());
934b547278cSGreg Clayton       result.SetStatus(eReturnStatusFailed);
9354574a890SZachary Turner       return false;
936b547278cSGreg Clayton     }
937b547278cSGreg Clayton 
9384574a890SZachary Turner     if (!m_interpreter.RemoveCommand(command_name)) {
9394574a890SZachary Turner       result.AppendErrorWithFormat(
9404574a890SZachary Turner           "'%s' is a permanent debugger command and cannot be removed.\n",
941867e7d17SZachary Turner           args[0].c_str());
9424574a890SZachary Turner       result.SetStatus(eReturnStatusFailed);
9434574a890SZachary Turner       return false;
9444574a890SZachary Turner     }
9454574a890SZachary Turner 
9464574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishNoResult);
9474574a890SZachary Turner     return true;
948b547278cSGreg Clayton   }
949b547278cSGreg Clayton };
950b547278cSGreg Clayton 
951de164aaaSGreg Clayton //-------------------------------------------------------------------------
952de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
953de164aaaSGreg Clayton //-------------------------------------------------------------------------
9541f0f5b5bSZachary Turner 
9551f0f5b5bSZachary Turner static OptionDefinition g_regex_options[] = {
9561f0f5b5bSZachary Turner     // clang-format off
9571f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "The help text to display for this command." },
9581f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "A syntax string showing the typical usage syntax." },
9591f0f5b5bSZachary Turner     // clang-format on
9601f0f5b5bSZachary Turner };
9611f0f5b5bSZachary Turner 
9625a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
963de164aaaSGreg Clayton 
964b9c1b51eSKate Stone class CommandObjectCommandsAddRegex : public CommandObjectParsed,
965b9c1b51eSKate Stone                                       public IOHandlerDelegateMultiline {
966de164aaaSGreg Clayton public:
9677428a18cSKate Stone   CommandObjectCommandsAddRegex(CommandInterpreter &interpreter)
968b9c1b51eSKate Stone       : CommandObjectParsed(
969b9c1b51eSKate Stone             interpreter, "command regex", "Define a custom command in terms of "
970b9c1b51eSKate Stone                                           "existing commands by matching "
971b9c1b51eSKate Stone                                           "regular expressions.",
9720e5e5a79SGreg Clayton             "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
973b9c1b51eSKate Stone         IOHandlerDelegateMultiline("",
974b9c1b51eSKate Stone                                    IOHandlerDelegate::Completion::LLDBCommand),
975b9c1b51eSKate Stone         m_options() {
976b9c1b51eSKate Stone     SetHelpLong(
977b9c1b51eSKate Stone         R"(
978b9c1b51eSKate Stone )"
979b9c1b51eSKate Stone         "This command allows the user to create powerful regular expression commands \
980ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
981b9c1b51eSKate Stone using the regular expression substitution format of:"
982b9c1b51eSKate Stone         R"(
983ea671fbdSKate Stone 
984ea671fbdSKate Stone     s/<regex>/<subst>/
985ea671fbdSKate Stone 
986b9c1b51eSKate Stone )"
987b9c1b51eSKate Stone         "<regex> is a regular expression that can use parenthesis to capture regular \
988ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
989b9c1b51eSKate Stone for the first match, %2 for the second, and so on."
990b9c1b51eSKate Stone         R"(
991ea671fbdSKate Stone 
992b9c1b51eSKate Stone )"
993b9c1b51eSKate Stone         "The regular expressions can all be specified on the command line if more than \
994ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
995ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
996b9c1b51eSKate Stone lines, followed by an empty line to terminate the command definition."
997b9c1b51eSKate Stone         R"(
998ea671fbdSKate Stone 
999ea671fbdSKate Stone EXAMPLES
1000ea671fbdSKate Stone 
1001b9c1b51eSKate Stone )"
1002b9c1b51eSKate Stone         "The following example will define a regular expression command named 'f' that \
1003ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
1004b9c1b51eSKate Stone a number follows 'f':"
1005b9c1b51eSKate Stone         R"(
1006ea671fbdSKate Stone 
1007b9c1b51eSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')");
1008de164aaaSGreg Clayton   }
1009de164aaaSGreg Clayton 
10106e3d8e7fSEugene Zelenko   ~CommandObjectCommandsAddRegex() override = default;
1011de164aaaSGreg Clayton 
10125a988416SJim Ingham protected:
1013b9c1b51eSKate Stone   void IOHandlerActivated(IOHandler &io_handler) override {
101444d93782SGreg Clayton     StreamFileSP output_sp(io_handler.GetOutputStreamFile());
1015b9c1b51eSKate Stone     if (output_sp) {
1016b9c1b51eSKate Stone       output_sp->PutCString("Enter one of more sed substitution commands in "
1017b9c1b51eSKate Stone                             "the form: 's/<regex>/<subst>/'.\nTerminate the "
1018b9c1b51eSKate Stone                             "substitution list with an empty line.\n");
101944d93782SGreg Clayton       output_sp->Flush();
102044d93782SGreg Clayton     }
102144d93782SGreg Clayton   }
102244d93782SGreg Clayton 
1023b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1024b9c1b51eSKate Stone                               std::string &data) override {
102544d93782SGreg Clayton     io_handler.SetIsDone(true);
1026b9c1b51eSKate Stone     if (m_regex_cmd_ap) {
102744d93782SGreg Clayton       StringList lines;
1028b9c1b51eSKate Stone       if (lines.SplitIntoLines(data)) {
102944d93782SGreg Clayton         const size_t num_lines = lines.GetSize();
103044d93782SGreg Clayton         bool check_only = false;
1031b9c1b51eSKate Stone         for (size_t i = 0; i < num_lines; ++i) {
103244d93782SGreg Clayton           llvm::StringRef bytes_strref(lines[i]);
103344d93782SGreg Clayton           Error error = AppendRegexSubstitution(bytes_strref, check_only);
1034b9c1b51eSKate Stone           if (error.Fail()) {
1035b9c1b51eSKate Stone             if (!m_interpreter.GetDebugger()
1036b9c1b51eSKate Stone                      .GetCommandInterpreter()
1037b9c1b51eSKate Stone                      .GetBatchCommandMode()) {
1038b9c1b51eSKate Stone               StreamSP out_stream =
1039b9c1b51eSKate Stone                   m_interpreter.GetDebugger().GetAsyncOutputStream();
104044d93782SGreg Clayton               out_stream->Printf("error: %s\n", error.AsCString());
104144d93782SGreg Clayton             }
104244d93782SGreg Clayton           }
104344d93782SGreg Clayton         }
104444d93782SGreg Clayton       }
1045b9c1b51eSKate Stone       if (m_regex_cmd_ap->HasRegexEntries()) {
104644d93782SGreg Clayton         CommandObjectSP cmd_sp(m_regex_cmd_ap.release());
104744d93782SGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
104844d93782SGreg Clayton       }
104944d93782SGreg Clayton     }
105044d93782SGreg Clayton   }
105144d93782SGreg Clayton 
1052b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
10535a988416SJim Ingham     const size_t argc = command.GetArgumentCount();
1054b9c1b51eSKate Stone     if (argc == 0) {
1055b9c1b51eSKate Stone       result.AppendError("usage: 'command regex <command-name> "
1056b9c1b51eSKate Stone                          "[s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
10570e5e5a79SGreg Clayton       result.SetStatus(eReturnStatusFailed);
105811eb9c64SZachary Turner       return false;
105911eb9c64SZachary Turner     }
106011eb9c64SZachary Turner 
10610e5e5a79SGreg Clayton     Error error;
10624574a890SZachary Turner     auto name = command[0].ref;
10634574a890SZachary Turner     m_regex_cmd_ap = llvm::make_unique<CommandObjectRegexCommand>(
10644574a890SZachary Turner         m_interpreter, name, m_options.GetHelp(), m_options.GetSyntax(), 10, 0,
10654574a890SZachary Turner         true);
10660e5e5a79SGreg Clayton 
1067b9c1b51eSKate Stone     if (argc == 1) {
106844d93782SGreg Clayton       Debugger &debugger = m_interpreter.GetDebugger();
1069e30f11d9SKate Stone       bool color_prompt = debugger.GetUseColor();
107044d93782SGreg Clayton       const bool multiple_lines = true; // Get multiple lines
1071b9c1b51eSKate Stone       IOHandlerSP io_handler_sp(new IOHandlerEditline(
1072b9c1b51eSKate Stone           debugger, IOHandler::Type::Other,
107373d80faaSGreg Clayton           "lldb-regex",          // Name of input reader for history
1074514d8cd8SZachary Turner           llvm::StringRef("> "), // Prompt
1075514d8cd8SZachary Turner           llvm::StringRef(),     // Continuation prompt
1076b9c1b51eSKate Stone           multiple_lines, color_prompt,
1077f6913cd7SGreg Clayton           0, // Don't show line numbers
107844d93782SGreg Clayton           *this));
107944d93782SGreg Clayton 
1080b9c1b51eSKate Stone       if (io_handler_sp) {
108144d93782SGreg Clayton         debugger.PushIOHandler(io_handler_sp);
1082de164aaaSGreg Clayton         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1083de164aaaSGreg Clayton       }
1084b9c1b51eSKate Stone     } else {
108597d2c401SZachary Turner       for (auto &entry : command.entries().drop_front()) {
108644d93782SGreg Clayton         bool check_only = false;
108797d2c401SZachary Turner         error = AppendRegexSubstitution(entry.ref, check_only);
10880e5e5a79SGreg Clayton         if (error.Fail())
10890e5e5a79SGreg Clayton           break;
10900e5e5a79SGreg Clayton       }
10910e5e5a79SGreg Clayton 
1092b9c1b51eSKate Stone       if (error.Success()) {
10930e5e5a79SGreg Clayton         AddRegexCommandToInterpreter();
10940e5e5a79SGreg Clayton       }
10950e5e5a79SGreg Clayton     }
1096b9c1b51eSKate Stone     if (error.Fail()) {
10970e5e5a79SGreg Clayton       result.AppendError(error.AsCString());
1098de164aaaSGreg Clayton       result.SetStatus(eReturnStatusFailed);
1099de164aaaSGreg Clayton     }
11000e5e5a79SGreg Clayton 
1101de164aaaSGreg Clayton     return result.Succeeded();
1102de164aaaSGreg Clayton   }
1103de164aaaSGreg Clayton 
1104b9c1b51eSKate Stone   Error AppendRegexSubstitution(const llvm::StringRef &regex_sed,
1105b9c1b51eSKate Stone                                 bool check_only) {
11060e5e5a79SGreg Clayton     Error error;
11070e5e5a79SGreg Clayton 
1108b9c1b51eSKate Stone     if (!m_regex_cmd_ap) {
1109b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1110b9c1b51eSKate Stone           "invalid regular expression command object for: '%.*s'",
1111b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
11120e5e5a79SGreg Clayton       return error;
1113de164aaaSGreg Clayton     }
11140e5e5a79SGreg Clayton 
11150e5e5a79SGreg Clayton     size_t regex_sed_size = regex_sed.size();
11160e5e5a79SGreg Clayton 
1117b9c1b51eSKate Stone     if (regex_sed_size <= 1) {
1118b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1119b9c1b51eSKate Stone           "regular expression substitution string is too short: '%.*s'",
1120b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
11210e5e5a79SGreg Clayton       return error;
11220e5e5a79SGreg Clayton     }
11230e5e5a79SGreg Clayton 
1124b9c1b51eSKate Stone     if (regex_sed[0] != 's') {
1125b9c1b51eSKate Stone       error.SetErrorStringWithFormat("regular expression substitution string "
1126b9c1b51eSKate Stone                                      "doesn't start with 's': '%.*s'",
1127b9c1b51eSKate Stone                                      (int)regex_sed.size(), regex_sed.data());
11280e5e5a79SGreg Clayton       return error;
11290e5e5a79SGreg Clayton     }
11300e5e5a79SGreg Clayton     const size_t first_separator_char_pos = 1;
11310e5e5a79SGreg Clayton     // use the char that follows 's' as the regex separator character
11320e5e5a79SGreg Clayton     // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
11330e5e5a79SGreg Clayton     const char separator_char = regex_sed[first_separator_char_pos];
1134b9c1b51eSKate Stone     const size_t second_separator_char_pos =
1135b9c1b51eSKate Stone         regex_sed.find(separator_char, first_separator_char_pos + 1);
11360e5e5a79SGreg Clayton 
1137b9c1b51eSKate Stone     if (second_separator_char_pos == std::string::npos) {
1138b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1139b9c1b51eSKate Stone           "missing second '%c' separator char after '%.*s' in '%.*s'",
11400e5e5a79SGreg Clayton           separator_char,
11410e5e5a79SGreg Clayton           (int)(regex_sed.size() - first_separator_char_pos - 1),
1142ea508635SGreg Clayton           regex_sed.data() + (first_separator_char_pos + 1),
1143b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
11440e5e5a79SGreg Clayton       return error;
11450e5e5a79SGreg Clayton     }
11460e5e5a79SGreg Clayton 
1147b9c1b51eSKate Stone     const size_t third_separator_char_pos =
1148b9c1b51eSKate Stone         regex_sed.find(separator_char, second_separator_char_pos + 1);
11490e5e5a79SGreg Clayton 
1150b9c1b51eSKate Stone     if (third_separator_char_pos == std::string::npos) {
1151b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1152b9c1b51eSKate Stone           "missing third '%c' separator char after '%.*s' in '%.*s'",
11530e5e5a79SGreg Clayton           separator_char,
11540e5e5a79SGreg Clayton           (int)(regex_sed.size() - second_separator_char_pos - 1),
1155ea508635SGreg Clayton           regex_sed.data() + (second_separator_char_pos + 1),
1156b9c1b51eSKate Stone           (int)regex_sed.size(), regex_sed.data());
11570e5e5a79SGreg Clayton       return error;
11580e5e5a79SGreg Clayton     }
11590e5e5a79SGreg Clayton 
1160b9c1b51eSKate Stone     if (third_separator_char_pos != regex_sed_size - 1) {
11610e5e5a79SGreg Clayton       // Make sure that everything that follows the last regex
11620e5e5a79SGreg Clayton       // separator char
1163b9c1b51eSKate Stone       if (regex_sed.find_first_not_of("\t\n\v\f\r ",
1164b9c1b51eSKate Stone                                       third_separator_char_pos + 1) !=
1165b9c1b51eSKate Stone           std::string::npos) {
1166b9c1b51eSKate Stone         error.SetErrorStringWithFormat(
1167b9c1b51eSKate Stone             "extra data found after the '%.*s' regular expression substitution "
1168b9c1b51eSKate Stone             "string: '%.*s'",
1169b9c1b51eSKate Stone             (int)third_separator_char_pos + 1, regex_sed.data(),
11700e5e5a79SGreg Clayton             (int)(regex_sed.size() - third_separator_char_pos - 1),
11710e5e5a79SGreg Clayton             regex_sed.data() + (third_separator_char_pos + 1));
11720e5e5a79SGreg Clayton         return error;
11730e5e5a79SGreg Clayton       }
1174b9c1b51eSKate Stone     } else if (first_separator_char_pos + 1 == second_separator_char_pos) {
1175b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1176b9c1b51eSKate Stone           "<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
1177b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
11780e5e5a79SGreg Clayton           regex_sed.data());
11790e5e5a79SGreg Clayton       return error;
1180b9c1b51eSKate Stone     } else if (second_separator_char_pos + 1 == third_separator_char_pos) {
1181b9c1b51eSKate Stone       error.SetErrorStringWithFormat(
1182b9c1b51eSKate Stone           "<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
1183b9c1b51eSKate Stone           separator_char, separator_char, separator_char, (int)regex_sed.size(),
11840e5e5a79SGreg Clayton           regex_sed.data());
11850e5e5a79SGreg Clayton       return error;
11860e5e5a79SGreg Clayton     }
118744d93782SGreg Clayton 
1188b9c1b51eSKate Stone     if (!check_only) {
1189b9c1b51eSKate Stone       std::string regex(regex_sed.substr(first_separator_char_pos + 1,
1190b9c1b51eSKate Stone                                          second_separator_char_pos -
1191b9c1b51eSKate Stone                                              first_separator_char_pos - 1));
1192b9c1b51eSKate Stone       std::string subst(regex_sed.substr(second_separator_char_pos + 1,
1193b9c1b51eSKate Stone                                          third_separator_char_pos -
1194b9c1b51eSKate Stone                                              second_separator_char_pos - 1));
1195b9c1b51eSKate Stone       m_regex_cmd_ap->AddRegexCommand(regex.c_str(), subst.c_str());
119644d93782SGreg Clayton     }
11970e5e5a79SGreg Clayton     return error;
1198de164aaaSGreg Clayton   }
1199de164aaaSGreg Clayton 
1200b9c1b51eSKate Stone   void AddRegexCommandToInterpreter() {
1201b9c1b51eSKate Stone     if (m_regex_cmd_ap) {
1202b9c1b51eSKate Stone       if (m_regex_cmd_ap->HasRegexEntries()) {
1203de164aaaSGreg Clayton         CommandObjectSP cmd_sp(m_regex_cmd_ap.release());
1204de164aaaSGreg Clayton         m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1205de164aaaSGreg Clayton       }
1206de164aaaSGreg Clayton     }
1207de164aaaSGreg Clayton   }
1208de164aaaSGreg Clayton 
1209de164aaaSGreg Clayton private:
12107b0992d9SGreg Clayton   std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1211de164aaaSGreg Clayton 
1212b9c1b51eSKate Stone   class CommandOptions : public Options {
1213de164aaaSGreg Clayton   public:
1214b9c1b51eSKate Stone     CommandOptions() : Options() {}
1215de164aaaSGreg Clayton 
12166e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1217de164aaaSGreg Clayton 
1218fe11483bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1219b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
1220de164aaaSGreg Clayton       Error error;
12213bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1222de164aaaSGreg Clayton 
1223b9c1b51eSKate Stone       switch (short_option) {
1224de164aaaSGreg Clayton       case 'h':
1225de164aaaSGreg Clayton         m_help.assign(option_arg);
1226de164aaaSGreg Clayton         break;
1227de164aaaSGreg Clayton       case 's':
1228de164aaaSGreg Clayton         m_syntax.assign(option_arg);
1229de164aaaSGreg Clayton         break;
1230de164aaaSGreg Clayton       default:
1231b9c1b51eSKate Stone         error.SetErrorStringWithFormat("unrecognized option '%c'",
1232b9c1b51eSKate Stone                                        short_option);
1233de164aaaSGreg Clayton         break;
1234de164aaaSGreg Clayton       }
1235de164aaaSGreg Clayton 
1236de164aaaSGreg Clayton       return error;
1237de164aaaSGreg Clayton     }
1238de164aaaSGreg Clayton 
1239b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1240de164aaaSGreg Clayton       m_help.clear();
1241de164aaaSGreg Clayton       m_syntax.clear();
1242de164aaaSGreg Clayton     }
1243de164aaaSGreg Clayton 
12441f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
124570602439SZachary Turner       return llvm::makeArrayRef(g_regex_options);
12461f0f5b5bSZachary Turner     }
1247de164aaaSGreg Clayton 
124811eb9c64SZachary Turner     // TODO: Convert these functions to return StringRefs.
1249b9c1b51eSKate Stone     const char *GetHelp() {
12506e3d8e7fSEugene Zelenko       return (m_help.empty() ? nullptr : m_help.c_str());
1251de164aaaSGreg Clayton     }
12526e3d8e7fSEugene Zelenko 
1253b9c1b51eSKate Stone     const char *GetSyntax() {
12546e3d8e7fSEugene Zelenko       return (m_syntax.empty() ? nullptr : m_syntax.c_str());
1255de164aaaSGreg Clayton     }
12566e3d8e7fSEugene Zelenko 
1257de164aaaSGreg Clayton   protected:
12586e3d8e7fSEugene Zelenko     // Instance variables to hold the values for command options.
12596e3d8e7fSEugene Zelenko 
1260de164aaaSGreg Clayton     std::string m_help;
1261de164aaaSGreg Clayton     std::string m_syntax;
1262de164aaaSGreg Clayton   };
1263de164aaaSGreg Clayton 
1264b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
1265de164aaaSGreg Clayton 
12665a988416SJim Ingham   CommandOptions m_options;
1267de164aaaSGreg Clayton };
1268de164aaaSGreg Clayton 
1269b9c1b51eSKate Stone class CommandObjectPythonFunction : public CommandObjectRaw {
1270223383edSEnrico Granata public:
1271b9c1b51eSKate Stone   CommandObjectPythonFunction(CommandInterpreter &interpreter, std::string name,
1272b9c1b51eSKate Stone                               std::string funct, std::string help,
1273b9c1b51eSKate Stone                               ScriptedCommandSynchronicity synch)
1274a449698cSZachary Turner       : CommandObjectRaw(interpreter, name),
1275b9c1b51eSKate Stone         m_function_name(funct), m_synchro(synch), m_fetched_help_long(false) {
1276735152e3SEnrico Granata     if (!help.empty())
1277442f6530SZachary Turner       SetHelp(help);
1278b9c1b51eSKate Stone     else {
1279735152e3SEnrico Granata       StreamString stream;
1280735152e3SEnrico Granata       stream.Printf("For more information run 'help %s'", name.c_str());
1281c156427dSZachary Turner       SetHelp(stream.GetString());
1282735152e3SEnrico Granata     }
1283223383edSEnrico Granata   }
1284223383edSEnrico Granata 
12856e3d8e7fSEugene Zelenko   ~CommandObjectPythonFunction() override = default;
1286223383edSEnrico Granata 
1287b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
12885a988416SJim Ingham 
1289b9c1b51eSKate Stone   const std::string &GetFunctionName() { return m_function_name; }
12905a988416SJim Ingham 
1291b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
12925a988416SJim Ingham 
1293442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1294442f6530SZachary Turner     if (m_fetched_help_long)
1295442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1296442f6530SZachary Turner 
1297fac939e9SEnrico Granata     ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter();
1298442f6530SZachary Turner     if (!scripter)
1299442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1300442f6530SZachary Turner 
1301fac939e9SEnrico Granata     std::string docstring;
1302442f6530SZachary Turner     m_fetched_help_long =
1303442f6530SZachary Turner         scripter->GetDocumentationForItem(m_function_name.c_str(), docstring);
1304fac939e9SEnrico Granata     if (!docstring.empty())
1305442f6530SZachary Turner       SetHelpLong(docstring);
1306fac939e9SEnrico Granata     return CommandObjectRaw::GetHelpLong();
1307fac939e9SEnrico Granata   }
1308fac939e9SEnrico Granata 
13095a988416SJim Ingham protected:
1310b9c1b51eSKate Stone   bool DoExecute(const char *raw_command_line,
1311b9c1b51eSKate Stone                  CommandReturnObject &result) override {
1312223383edSEnrico Granata     ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter();
1313223383edSEnrico Granata 
1314223383edSEnrico Granata     Error error;
1315223383edSEnrico Granata 
131670f11f88SJim Ingham     result.SetStatus(eReturnStatusInvalid);
131770f11f88SJim Ingham 
1318b9c1b51eSKate Stone     if (!scripter ||
1319b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_function_name.c_str(),
1320b9c1b51eSKate Stone                                          raw_command_line, m_synchro, result,
1321b9c1b51eSKate Stone                                          error, m_exe_ctx)) {
1322223383edSEnrico Granata       result.AppendError(error.AsCString());
1323223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1324b9c1b51eSKate Stone     } else {
132570f11f88SJim Ingham       // Don't change the status if the command already set it...
1326b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1327c156427dSZachary Turner         if (result.GetOutputData().empty())
1328223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
132970f11f88SJim Ingham         else
133070f11f88SJim Ingham           result.SetStatus(eReturnStatusSuccessFinishResult);
133170f11f88SJim Ingham       }
133270f11f88SJim Ingham     }
1333223383edSEnrico Granata 
1334223383edSEnrico Granata     return result.Succeeded();
1335223383edSEnrico Granata   }
1336223383edSEnrico Granata 
13376e3d8e7fSEugene Zelenko private:
13386e3d8e7fSEugene Zelenko   std::string m_function_name;
13396e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
13406e3d8e7fSEugene Zelenko   bool m_fetched_help_long;
1341223383edSEnrico Granata };
1342223383edSEnrico Granata 
1343b9c1b51eSKate Stone class CommandObjectScriptingObject : public CommandObjectRaw {
13449fe00e52SEnrico Granata public:
13459fe00e52SEnrico Granata   CommandObjectScriptingObject(CommandInterpreter &interpreter,
13469fe00e52SEnrico Granata                                std::string name,
13470641ca1aSZachary Turner                                StructuredData::GenericSP cmd_obj_sp,
1348b9c1b51eSKate Stone                                ScriptedCommandSynchronicity synch)
1349a449698cSZachary Turner       : CommandObjectRaw(interpreter, name),
1350b9c1b51eSKate Stone         m_cmd_obj_sp(cmd_obj_sp), m_synchro(synch), m_fetched_help_short(false),
1351b9c1b51eSKate Stone         m_fetched_help_long(false) {
13529fe00e52SEnrico Granata     StreamString stream;
13539fe00e52SEnrico Granata     stream.Printf("For more information run 'help %s'", name.c_str());
1354c156427dSZachary Turner     SetHelp(stream.GetString());
1355e87764f2SEnrico Granata     if (ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter())
1356e87764f2SEnrico Granata       GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
13579fe00e52SEnrico Granata   }
13589fe00e52SEnrico Granata 
13596e3d8e7fSEugene Zelenko   ~CommandObjectScriptingObject() override = default;
13609fe00e52SEnrico Granata 
1361b9c1b51eSKate Stone   bool IsRemovable() const override { return true; }
13629fe00e52SEnrico Granata 
1363b9c1b51eSKate Stone   StructuredData::GenericSP GetImplementingObject() { return m_cmd_obj_sp; }
13649fe00e52SEnrico Granata 
1365b9c1b51eSKate Stone   ScriptedCommandSynchronicity GetSynchronicity() { return m_synchro; }
13669fe00e52SEnrico Granata 
1367442f6530SZachary Turner   llvm::StringRef GetHelp() override {
1368442f6530SZachary Turner     if (m_fetched_help_short)
1369442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
13706f79bb2dSEnrico Granata     ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter();
1371442f6530SZachary Turner     if (!scripter)
1372442f6530SZachary Turner       return CommandObjectRaw::GetHelp();
13736f79bb2dSEnrico Granata     std::string docstring;
1374b9c1b51eSKate Stone     m_fetched_help_short =
1375b9c1b51eSKate Stone         scripter->GetShortHelpForCommandObject(m_cmd_obj_sp, docstring);
13766f79bb2dSEnrico Granata     if (!docstring.empty())
1377442f6530SZachary Turner       SetHelp(docstring);
1378442f6530SZachary Turner 
13796f79bb2dSEnrico Granata     return CommandObjectRaw::GetHelp();
13806f79bb2dSEnrico Granata   }
13816f79bb2dSEnrico Granata 
1382442f6530SZachary Turner   llvm::StringRef GetHelpLong() override {
1383442f6530SZachary Turner     if (m_fetched_help_long)
1384442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1385442f6530SZachary Turner 
13866f79bb2dSEnrico Granata     ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter();
1387442f6530SZachary Turner     if (!scripter)
1388442f6530SZachary Turner       return CommandObjectRaw::GetHelpLong();
1389442f6530SZachary Turner 
13906f79bb2dSEnrico Granata     std::string docstring;
1391b9c1b51eSKate Stone     m_fetched_help_long =
1392b9c1b51eSKate Stone         scripter->GetLongHelpForCommandObject(m_cmd_obj_sp, docstring);
13936f79bb2dSEnrico Granata     if (!docstring.empty())
1394442f6530SZachary Turner       SetHelpLong(docstring);
13959fe00e52SEnrico Granata     return CommandObjectRaw::GetHelpLong();
13969fe00e52SEnrico Granata   }
13979fe00e52SEnrico Granata 
13989fe00e52SEnrico Granata protected:
1399b9c1b51eSKate Stone   bool DoExecute(const char *raw_command_line,
1400b9c1b51eSKate Stone                  CommandReturnObject &result) override {
14019fe00e52SEnrico Granata     ScriptInterpreter *scripter = m_interpreter.GetScriptInterpreter();
14029fe00e52SEnrico Granata 
14039fe00e52SEnrico Granata     Error error;
14049fe00e52SEnrico Granata 
14059fe00e52SEnrico Granata     result.SetStatus(eReturnStatusInvalid);
14069fe00e52SEnrico Granata 
1407b9c1b51eSKate Stone     if (!scripter ||
1408b9c1b51eSKate Stone         !scripter->RunScriptBasedCommand(m_cmd_obj_sp, raw_command_line,
1409b9c1b51eSKate Stone                                          m_synchro, result, error, m_exe_ctx)) {
14109fe00e52SEnrico Granata       result.AppendError(error.AsCString());
14119fe00e52SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1412b9c1b51eSKate Stone     } else {
14139fe00e52SEnrico Granata       // Don't change the status if the command already set it...
1414b9c1b51eSKate Stone       if (result.GetStatus() == eReturnStatusInvalid) {
1415c156427dSZachary Turner         if (result.GetOutputData().empty())
14169fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
14179fe00e52SEnrico Granata         else
14189fe00e52SEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishResult);
14199fe00e52SEnrico Granata       }
14209fe00e52SEnrico Granata     }
14219fe00e52SEnrico Granata 
14229fe00e52SEnrico Granata     return result.Succeeded();
14239fe00e52SEnrico Granata   }
14249fe00e52SEnrico Granata 
14256e3d8e7fSEugene Zelenko private:
14266e3d8e7fSEugene Zelenko   StructuredData::GenericSP m_cmd_obj_sp;
14276e3d8e7fSEugene Zelenko   ScriptedCommandSynchronicity m_synchro;
14286e3d8e7fSEugene Zelenko   bool m_fetched_help_short : 1;
14296e3d8e7fSEugene Zelenko   bool m_fetched_help_long : 1;
14309fe00e52SEnrico Granata };
14319fe00e52SEnrico Granata 
1432a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1433a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1434a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1435a9dbf432SEnrico Granata 
14361f0f5b5bSZachary Turner OptionDefinition g_script_import_options[] = {
14371f0f5b5bSZachary Turner     // clang-format off
14381f0f5b5bSZachary Turner   { LLDB_OPT_SET_1, false, "allow-reload", 'r', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeNone, "Allow the script to be loaded even if it was already loaded before. This argument exists for backwards compatibility, but reloading is always allowed, whether you specify it or not." },
14391f0f5b5bSZachary Turner     // clang-format on
14401f0f5b5bSZachary Turner };
14411f0f5b5bSZachary Turner 
1442b9c1b51eSKate Stone class CommandObjectCommandsScriptImport : public CommandObjectParsed {
14435a988416SJim Ingham public:
1444b9c1b51eSKate Stone   CommandObjectCommandsScriptImport(CommandInterpreter &interpreter)
1445b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script import",
1446b9c1b51eSKate Stone                             "Import a scripting module in LLDB.", nullptr),
1447b9c1b51eSKate Stone         m_options() {
14485a988416SJim Ingham     CommandArgumentEntry arg1;
14495a988416SJim Ingham     CommandArgumentData cmd_arg;
14505a988416SJim Ingham 
14515a988416SJim Ingham     // Define the first (and only) variant of this arg.
14525a988416SJim Ingham     cmd_arg.arg_type = eArgTypeFilename;
14533b00e35bSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlus;
14545a988416SJim Ingham 
1455b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1456b9c1b51eSKate Stone     // argument entry.
14575a988416SJim Ingham     arg1.push_back(cmd_arg);
14585a988416SJim Ingham 
14595a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
14605a988416SJim Ingham     m_arguments.push_back(arg1);
14615a988416SJim Ingham   }
14625a988416SJim Ingham 
14636e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptImport() override = default;
14645a988416SJim Ingham 
1465b9c1b51eSKate Stone   int HandleArgumentCompletion(Args &input, int &cursor_index,
14665a988416SJim Ingham                                int &cursor_char_position,
14675a988416SJim Ingham                                OptionElementVector &opt_element_vector,
1468b9c1b51eSKate Stone                                int match_start_point, int max_return_elements,
14695a988416SJim Ingham                                bool &word_complete,
1470b9c1b51eSKate Stone                                StringList &matches) override {
1471d6a24757SZachary Turner     llvm::StringRef completion_str = input[cursor_index].ref;
1472d6a24757SZachary Turner     completion_str = completion_str.take_front(cursor_char_position);
14735a988416SJim Ingham 
1474b9c1b51eSKate Stone     CommandCompletions::InvokeCommonCompletionCallbacks(
1475b9c1b51eSKate Stone         GetCommandInterpreter(), CommandCompletions::eDiskFileCompletion,
1476d6a24757SZachary Turner         completion_str, match_start_point, max_return_elements, nullptr,
1477b9c1b51eSKate Stone         word_complete, matches);
14785a988416SJim Ingham     return matches.GetSize();
14795a988416SJim Ingham   }
14805a988416SJim Ingham 
1481b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
14825a988416SJim Ingham 
14835a988416SJim Ingham protected:
1484b9c1b51eSKate Stone   class CommandOptions : public Options {
14850a305db7SEnrico Granata   public:
1486b9c1b51eSKate Stone     CommandOptions() : Options() {}
14870a305db7SEnrico Granata 
14886e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
14890a305db7SEnrico Granata 
1490fe11483bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1491b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
14920a305db7SEnrico Granata       Error error;
14933bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
14940a305db7SEnrico Granata 
1495b9c1b51eSKate Stone       switch (short_option) {
14960a305db7SEnrico Granata       case 'r':
14970a305db7SEnrico Granata         m_allow_reload = true;
14980a305db7SEnrico Granata         break;
14990a305db7SEnrico Granata       default:
1500b9c1b51eSKate Stone         error.SetErrorStringWithFormat("unrecognized option '%c'",
1501b9c1b51eSKate Stone                                        short_option);
15020a305db7SEnrico Granata         break;
15030a305db7SEnrico Granata       }
15040a305db7SEnrico Granata 
15050a305db7SEnrico Granata       return error;
15060a305db7SEnrico Granata     }
15070a305db7SEnrico Granata 
1508b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
1509e0c70f1bSEnrico Granata       m_allow_reload = true;
15100a305db7SEnrico Granata     }
15110a305db7SEnrico Granata 
15121f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
151370602439SZachary Turner       return llvm::makeArrayRef(g_script_import_options);
15141f0f5b5bSZachary Turner     }
15150a305db7SEnrico Granata 
15160a305db7SEnrico Granata     // Instance variables to hold the values for command options.
15170a305db7SEnrico Granata 
15180a305db7SEnrico Granata     bool m_allow_reload;
15190a305db7SEnrico Granata   };
15200a305db7SEnrico Granata 
1521b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1522b9c1b51eSKate Stone     if (m_interpreter.GetDebugger().GetScriptLanguage() !=
1523b9c1b51eSKate Stone         lldb::eScriptLanguagePython) {
1524b9c1b51eSKate Stone       result.AppendError("only scripting language supported for module "
1525b9c1b51eSKate Stone                          "importing is currently Python");
1526a9dbf432SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1527a9dbf432SEnrico Granata       return false;
1528a9dbf432SEnrico Granata     }
1529a9dbf432SEnrico Granata 
153011eb9c64SZachary Turner     if (command.empty()) {
15313b00e35bSEnrico Granata       result.AppendError("command script import needs one or more arguments");
1532a9dbf432SEnrico Granata       result.SetStatus(eReturnStatusFailed);
1533a9dbf432SEnrico Granata       return false;
1534a9dbf432SEnrico Granata     }
1535a9dbf432SEnrico Granata 
153611eb9c64SZachary Turner     for (auto &entry : command.entries()) {
1537a9dbf432SEnrico Granata       Error error;
1538a9dbf432SEnrico Granata 
1539c9d645d3SGreg Clayton       const bool init_session = true;
1540b9c1b51eSKate Stone       // FIXME: this is necessary because CommandObject::CheckRequirements()
154111eb9c64SZachary Turner       // assumes that commands won't ever be recursively invoked, but it's
154211eb9c64SZachary Turner       // actually possible to craft a Python script that does other "command
154311eb9c64SZachary Turner       // script imports" in __lldb_init_module the real fix is to have recursive
154411eb9c64SZachary Turner       // commands possible with a CommandInvocation object separate from the
154511eb9c64SZachary Turner       // CommandObject itself, so that recursive command invocations won't stomp
154611eb9c64SZachary Turner       // on each other (wrt to execution contents, options, and more)
1547078551c7SEnrico Granata       m_exe_ctx.Clear();
1548b9c1b51eSKate Stone       if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(
154911eb9c64SZachary Turner               entry.c_str(), m_options.m_allow_reload, init_session, error)) {
1550a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1551b9c1b51eSKate Stone       } else {
1552b9c1b51eSKate Stone         result.AppendErrorWithFormat("module importing failed: %s",
1553b9c1b51eSKate Stone                                      error.AsCString());
1554a9dbf432SEnrico Granata         result.SetStatus(eReturnStatusFailed);
1555a9dbf432SEnrico Granata       }
15563b00e35bSEnrico Granata     }
1557a9dbf432SEnrico Granata 
1558a9dbf432SEnrico Granata     return result.Succeeded();
1559a9dbf432SEnrico Granata   }
15600a305db7SEnrico Granata 
15615a988416SJim Ingham   CommandOptions m_options;
1562a9dbf432SEnrico Granata };
1563223383edSEnrico Granata 
1564223383edSEnrico Granata //-------------------------------------------------------------------------
1565223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1566223383edSEnrico Granata //-------------------------------------------------------------------------
1567223383edSEnrico Granata 
15681f0f5b5bSZachary Turner static OptionEnumValueElement g_script_synchro_type[] = {
15691f0f5b5bSZachary Turner     {eScriptedCommandSynchronicitySynchronous, "synchronous",
15701f0f5b5bSZachary Turner      "Run synchronous"},
15711f0f5b5bSZachary Turner     {eScriptedCommandSynchronicityAsynchronous, "asynchronous",
15721f0f5b5bSZachary Turner      "Run asynchronous"},
15731f0f5b5bSZachary Turner     {eScriptedCommandSynchronicityCurrentValue, "current",
15741f0f5b5bSZachary Turner      "Do not alter current setting"},
15751f0f5b5bSZachary Turner     {0, nullptr, nullptr}};
15761f0f5b5bSZachary Turner 
15771f0f5b5bSZachary Turner static OptionDefinition g_script_add_options[] = {
15781f0f5b5bSZachary Turner     // clang-format off
15791f0f5b5bSZachary Turner   { LLDB_OPT_SET_1,   false, "function",      'f', OptionParser::eRequiredArgument, nullptr, nullptr,               0, eArgTypePythonFunction,               "Name of the Python function to bind to this command name." },
15801f0f5b5bSZachary Turner   { LLDB_OPT_SET_2,   false, "class",         'c', OptionParser::eRequiredArgument, nullptr, nullptr,               0, eArgTypePythonClass,                  "Name of the Python class to bind to this command name." },
15811f0f5b5bSZachary Turner   { LLDB_OPT_SET_1,   false, "help"  ,        'h', OptionParser::eRequiredArgument, nullptr, nullptr,               0, eArgTypeHelpText,                     "The help text to display for this command." },
15821f0f5b5bSZachary Turner   { LLDB_OPT_SET_ALL, false, "synchronicity", 's', OptionParser::eRequiredArgument, nullptr, g_script_synchro_type, 0, eArgTypeScriptedCommandSynchronicity, "Set the synchronicity of this command's executions with regard to LLDB event system." },
15831f0f5b5bSZachary Turner     // clang-format on
15841f0f5b5bSZachary Turner };
15851f0f5b5bSZachary Turner 
1586b9c1b51eSKate Stone class CommandObjectCommandsScriptAdd : public CommandObjectParsed,
1587b9c1b51eSKate Stone                                        public IOHandlerDelegateMultiline {
15885a988416SJim Ingham public:
1589b9c1b51eSKate Stone   CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter)
1590b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script add",
15915a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
15926e3d8e7fSEugene Zelenko                             nullptr),
1593b9c1b51eSKate Stone         IOHandlerDelegateMultiline("DONE"), m_options() {
15945a988416SJim Ingham     CommandArgumentEntry arg1;
15955a988416SJim Ingham     CommandArgumentData cmd_arg;
15965a988416SJim Ingham 
15975a988416SJim Ingham     // Define the first (and only) variant of this arg.
15985a988416SJim Ingham     cmd_arg.arg_type = eArgTypeCommandName;
15995a988416SJim Ingham     cmd_arg.arg_repetition = eArgRepeatPlain;
16005a988416SJim Ingham 
1601b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1602b9c1b51eSKate Stone     // argument entry.
16035a988416SJim Ingham     arg1.push_back(cmd_arg);
16045a988416SJim Ingham 
16055a988416SJim Ingham     // Push the data for the first argument into the m_arguments vector.
16065a988416SJim Ingham     m_arguments.push_back(arg1);
16075a988416SJim Ingham   }
16085a988416SJim Ingham 
16096e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptAdd() override = default;
16105a988416SJim Ingham 
1611b9c1b51eSKate Stone   Options *GetOptions() override { return &m_options; }
16125a988416SJim Ingham 
16135a988416SJim Ingham protected:
1614b9c1b51eSKate Stone   class CommandOptions : public Options {
1615223383edSEnrico Granata   public:
1616b9c1b51eSKate Stone     CommandOptions()
1617b9c1b51eSKate Stone         : Options(), m_class_name(), m_funct_name(), m_short_help(),
1618b9c1b51eSKate Stone           m_synchronicity(eScriptedCommandSynchronicitySynchronous) {}
1619223383edSEnrico Granata 
16206e3d8e7fSEugene Zelenko     ~CommandOptions() override = default;
1621223383edSEnrico Granata 
1622fe11483bSZachary Turner     Error SetOptionValue(uint32_t option_idx, llvm::StringRef option_arg,
1623b9c1b51eSKate Stone                          ExecutionContext *execution_context) override {
1624223383edSEnrico Granata       Error error;
16253bcdfc0eSGreg Clayton       const int short_option = m_getopt_table[option_idx].val;
1626223383edSEnrico Granata 
1627b9c1b51eSKate Stone       switch (short_option) {
1628223383edSEnrico Granata       case 'f':
1629fe11483bSZachary Turner         if (!option_arg.empty())
1630fe11483bSZachary Turner           m_funct_name = option_arg;
1631735152e3SEnrico Granata         break;
16329fe00e52SEnrico Granata       case 'c':
1633fe11483bSZachary Turner         if (!option_arg.empty())
1634fe11483bSZachary Turner           m_class_name = option_arg;
16359fe00e52SEnrico Granata         break;
1636735152e3SEnrico Granata       case 'h':
1637fe11483bSZachary Turner         if (!option_arg.empty())
1638fe11483bSZachary Turner           m_short_help = option_arg;
1639223383edSEnrico Granata         break;
16400a305db7SEnrico Granata       case 's':
1641b9c1b51eSKate Stone         m_synchronicity =
1642b9c1b51eSKate Stone             (ScriptedCommandSynchronicity)Args::StringToOptionEnum(
1643fe11483bSZachary Turner                 option_arg, GetDefinitions()[option_idx].enum_values, 0, error);
16440a305db7SEnrico Granata         if (!error.Success())
1645b9c1b51eSKate Stone           error.SetErrorStringWithFormat(
1646fe11483bSZachary Turner               "unrecognized value for synchronicity '%s'",
1647fe11483bSZachary Turner               option_arg.str().c_str());
16480a305db7SEnrico Granata         break;
1649223383edSEnrico Granata       default:
1650b9c1b51eSKate Stone         error.SetErrorStringWithFormat("unrecognized option '%c'",
1651b9c1b51eSKate Stone                                        short_option);
1652223383edSEnrico Granata         break;
1653223383edSEnrico Granata       }
1654223383edSEnrico Granata 
1655223383edSEnrico Granata       return error;
1656223383edSEnrico Granata     }
1657223383edSEnrico Granata 
1658b9c1b51eSKate Stone     void OptionParsingStarting(ExecutionContext *execution_context) override {
16599fe00e52SEnrico Granata       m_class_name.clear();
1660735152e3SEnrico Granata       m_funct_name.clear();
1661735152e3SEnrico Granata       m_short_help.clear();
166244d93782SGreg Clayton       m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1663223383edSEnrico Granata     }
1664223383edSEnrico Granata 
16651f0f5b5bSZachary Turner     llvm::ArrayRef<OptionDefinition> GetDefinitions() override {
166670602439SZachary Turner       return llvm::makeArrayRef(g_script_add_options);
16671f0f5b5bSZachary Turner     }
1668223383edSEnrico Granata 
1669223383edSEnrico Granata     // Instance variables to hold the values for command options.
1670223383edSEnrico Granata 
16719fe00e52SEnrico Granata     std::string m_class_name;
1672223383edSEnrico Granata     std::string m_funct_name;
1673735152e3SEnrico Granata     std::string m_short_help;
167444d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1675223383edSEnrico Granata   };
1676223383edSEnrico Granata 
1677b9c1b51eSKate Stone   void IOHandlerActivated(IOHandler &io_handler) override {
167844d93782SGreg Clayton     StreamFileSP output_sp(io_handler.GetOutputStreamFile());
1679b9c1b51eSKate Stone     if (output_sp) {
168044d93782SGreg Clayton       output_sp->PutCString(g_python_command_instructions);
168144d93782SGreg Clayton       output_sp->Flush();
1682223383edSEnrico Granata     }
1683223383edSEnrico Granata   }
1684223383edSEnrico Granata 
1685b9c1b51eSKate Stone   void IOHandlerInputComplete(IOHandler &io_handler,
1686b9c1b51eSKate Stone                               std::string &data) override {
168744d93782SGreg Clayton     StreamFileSP error_sp = io_handler.GetErrorStreamFile();
168844d93782SGreg Clayton 
168944d93782SGreg Clayton     ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter();
1690b9c1b51eSKate Stone     if (interpreter) {
169144d93782SGreg Clayton 
169244d93782SGreg Clayton       StringList lines;
169344d93782SGreg Clayton       lines.SplitIntoLines(data);
1694b9c1b51eSKate Stone       if (lines.GetSize() > 0) {
1695a73b7df7SEnrico Granata         std::string funct_name_str;
1696b9c1b51eSKate Stone         if (interpreter->GenerateScriptAliasFunction(lines, funct_name_str)) {
1697b9c1b51eSKate Stone           if (funct_name_str.empty()) {
1698b9c1b51eSKate Stone             error_sp->Printf("error: unable to obtain a function name, didn't "
1699b9c1b51eSKate Stone                              "add python command.\n");
170044d93782SGreg Clayton             error_sp->Flush();
1701b9c1b51eSKate Stone           } else {
1702223383edSEnrico Granata             // everything should be fine now, let's add this alias
1703223383edSEnrico Granata 
1704b9c1b51eSKate Stone             CommandObjectSP command_obj_sp(new CommandObjectPythonFunction(
1705771ef6d4SMalcolm Parsons                 m_interpreter, m_cmd_name, funct_name_str, m_short_help,
170644d93782SGreg Clayton                 m_synchronicity));
1707223383edSEnrico Granata 
1708b9c1b51eSKate Stone             if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp,
1709b9c1b51eSKate Stone                                               true)) {
1710b9c1b51eSKate Stone               error_sp->Printf("error: unable to add selected command, didn't "
1711b9c1b51eSKate Stone                                "add python command.\n");
171244d93782SGreg Clayton               error_sp->Flush();
1713223383edSEnrico Granata             }
1714223383edSEnrico Granata           }
1715b9c1b51eSKate Stone         } else {
1716b9c1b51eSKate Stone           error_sp->Printf(
1717b9c1b51eSKate Stone               "error: unable to create function, didn't add python command.\n");
171844d93782SGreg Clayton           error_sp->Flush();
171944d93782SGreg Clayton         }
1720b9c1b51eSKate Stone       } else {
172144d93782SGreg Clayton         error_sp->Printf("error: empty function, didn't add python command.\n");
172244d93782SGreg Clayton         error_sp->Flush();
172344d93782SGreg Clayton       }
1724b9c1b51eSKate Stone     } else {
1725b9c1b51eSKate Stone       error_sp->Printf(
1726b9c1b51eSKate Stone           "error: script interpreter missing, didn't add python command.\n");
172744d93782SGreg Clayton       error_sp->Flush();
172844d93782SGreg Clayton     }
172944d93782SGreg Clayton 
173044d93782SGreg Clayton     io_handler.SetIsDone(true);
173144d93782SGreg Clayton   }
1732223383edSEnrico Granata 
17335a988416SJim Ingham protected:
1734b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1735b9c1b51eSKate Stone     if (m_interpreter.GetDebugger().GetScriptLanguage() !=
1736b9c1b51eSKate Stone         lldb::eScriptLanguagePython) {
1737b9c1b51eSKate Stone       result.AppendError("only scripting language supported for scripted "
1738b9c1b51eSKate Stone                          "commands is currently Python");
173999f0b8f9SEnrico Granata       result.SetStatus(eReturnStatusFailed);
174099f0b8f9SEnrico Granata       return false;
174199f0b8f9SEnrico Granata     }
174299f0b8f9SEnrico Granata 
174311eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1744223383edSEnrico Granata       result.AppendError("'command script add' requires one argument");
1745223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1746223383edSEnrico Granata       return false;
1747223383edSEnrico Granata     }
1748223383edSEnrico Granata 
1749735152e3SEnrico Granata     // Store the options in case we get multi-line input
17504574a890SZachary Turner     m_cmd_name = command[0].ref;
1751735152e3SEnrico Granata     m_short_help.assign(m_options.m_short_help);
175244d93782SGreg Clayton     m_synchronicity = m_options.m_synchronicity;
1753223383edSEnrico Granata 
1754b9c1b51eSKate Stone     if (m_options.m_class_name.empty()) {
1755b9c1b51eSKate Stone       if (m_options.m_funct_name.empty()) {
1756b9c1b51eSKate Stone         m_interpreter.GetPythonCommandsFromIOHandler(
1757b9c1b51eSKate Stone             "     ",  // Prompt
175844d93782SGreg Clayton             *this,    // IOHandlerDelegate
175944d93782SGreg Clayton             true,     // Run IOHandler in async mode
1760b9c1b51eSKate Stone             nullptr); // Baton for the "io_handler" that will be passed back
1761b9c1b51eSKate Stone                       // into our IOHandlerDelegate functions
1762b9c1b51eSKate Stone       } else {
1763b9c1b51eSKate Stone         CommandObjectSP new_cmd(new CommandObjectPythonFunction(
1764b9c1b51eSKate Stone             m_interpreter, m_cmd_name, m_options.m_funct_name,
1765b9c1b51eSKate Stone             m_options.m_short_help, m_synchronicity));
1766b9c1b51eSKate Stone         if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
1767223383edSEnrico Granata           result.SetStatus(eReturnStatusSuccessFinishNoResult);
1768b9c1b51eSKate Stone         } else {
1769223383edSEnrico Granata           result.AppendError("cannot add command");
1770223383edSEnrico Granata           result.SetStatus(eReturnStatusFailed);
1771223383edSEnrico Granata         }
1772223383edSEnrico Granata       }
1773b9c1b51eSKate Stone     } else {
1774b9c1b51eSKate Stone       ScriptInterpreter *interpreter =
1775b9c1b51eSKate Stone           GetCommandInterpreter().GetScriptInterpreter();
1776b9c1b51eSKate Stone       if (!interpreter) {
17779fe00e52SEnrico Granata         result.AppendError("cannot find ScriptInterpreter");
17789fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
17799fe00e52SEnrico Granata         return false;
17809fe00e52SEnrico Granata       }
17819fe00e52SEnrico Granata 
1782b9c1b51eSKate Stone       auto cmd_obj_sp = interpreter->CreateScriptCommandObject(
1783b9c1b51eSKate Stone           m_options.m_class_name.c_str());
1784b9c1b51eSKate Stone       if (!cmd_obj_sp) {
17859fe00e52SEnrico Granata         result.AppendError("cannot create helper object");
17869fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
17879fe00e52SEnrico Granata         return false;
17889fe00e52SEnrico Granata       }
17899fe00e52SEnrico Granata 
1790b9c1b51eSKate Stone       CommandObjectSP new_cmd(new CommandObjectScriptingObject(
1791b9c1b51eSKate Stone           m_interpreter, m_cmd_name, cmd_obj_sp, m_synchronicity));
1792b9c1b51eSKate Stone       if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true)) {
17939fe00e52SEnrico Granata         result.SetStatus(eReturnStatusSuccessFinishNoResult);
1794b9c1b51eSKate Stone       } else {
17959fe00e52SEnrico Granata         result.AppendError("cannot add command");
17969fe00e52SEnrico Granata         result.SetStatus(eReturnStatusFailed);
17979fe00e52SEnrico Granata       }
17989fe00e52SEnrico Granata     }
1799223383edSEnrico Granata 
1800223383edSEnrico Granata     return result.Succeeded();
1801223383edSEnrico Granata   }
18025a988416SJim Ingham 
18035a988416SJim Ingham   CommandOptions m_options;
180444d93782SGreg Clayton   std::string m_cmd_name;
1805735152e3SEnrico Granata   std::string m_short_help;
180644d93782SGreg Clayton   ScriptedCommandSynchronicity m_synchronicity;
1807223383edSEnrico Granata };
1808223383edSEnrico Granata 
1809223383edSEnrico Granata //-------------------------------------------------------------------------
1810223383edSEnrico Granata // CommandObjectCommandsScriptList
1811223383edSEnrico Granata //-------------------------------------------------------------------------
1812223383edSEnrico Granata 
1813b9c1b51eSKate Stone class CommandObjectCommandsScriptList : public CommandObjectParsed {
1814223383edSEnrico Granata public:
1815b9c1b51eSKate Stone   CommandObjectCommandsScriptList(CommandInterpreter &interpreter)
1816b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script list",
1817b9c1b51eSKate Stone                             "List defined scripted commands.", nullptr) {}
1818223383edSEnrico Granata 
18196e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptList() override = default;
1820223383edSEnrico Granata 
1821b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1822b9c1b51eSKate Stone     m_interpreter.GetHelp(result, CommandInterpreter::eCommandTypesUserDef);
1823223383edSEnrico Granata 
1824223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1825223383edSEnrico Granata 
1826223383edSEnrico Granata     return true;
1827223383edSEnrico Granata   }
1828223383edSEnrico Granata };
1829223383edSEnrico Granata 
1830223383edSEnrico Granata //-------------------------------------------------------------------------
1831223383edSEnrico Granata // CommandObjectCommandsScriptClear
1832223383edSEnrico Granata //-------------------------------------------------------------------------
1833223383edSEnrico Granata 
1834b9c1b51eSKate Stone class CommandObjectCommandsScriptClear : public CommandObjectParsed {
1835223383edSEnrico Granata public:
1836b9c1b51eSKate Stone   CommandObjectCommandsScriptClear(CommandInterpreter &interpreter)
1837b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script clear",
1838b9c1b51eSKate Stone                             "Delete all scripted commands.", nullptr) {}
1839223383edSEnrico Granata 
18406e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptClear() override = default;
1841223383edSEnrico Granata 
18425a988416SJim Ingham protected:
1843b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1844223383edSEnrico Granata     m_interpreter.RemoveAllUser();
1845223383edSEnrico Granata 
1846223383edSEnrico Granata     result.SetStatus(eReturnStatusSuccessFinishResult);
1847223383edSEnrico Granata 
1848223383edSEnrico Granata     return true;
1849223383edSEnrico Granata   }
1850223383edSEnrico Granata };
1851223383edSEnrico Granata 
1852223383edSEnrico Granata //-------------------------------------------------------------------------
1853223383edSEnrico Granata // CommandObjectCommandsScriptDelete
1854223383edSEnrico Granata //-------------------------------------------------------------------------
1855223383edSEnrico Granata 
1856b9c1b51eSKate Stone class CommandObjectCommandsScriptDelete : public CommandObjectParsed {
1857223383edSEnrico Granata public:
1858b9c1b51eSKate Stone   CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter)
1859b9c1b51eSKate Stone       : CommandObjectParsed(interpreter, "command script delete",
1860b9c1b51eSKate Stone                             "Delete a scripted command.", nullptr) {
1861223383edSEnrico Granata     CommandArgumentEntry arg1;
1862223383edSEnrico Granata     CommandArgumentData cmd_arg;
1863223383edSEnrico Granata 
1864223383edSEnrico Granata     // Define the first (and only) variant of this arg.
1865223383edSEnrico Granata     cmd_arg.arg_type = eArgTypeCommandName;
1866223383edSEnrico Granata     cmd_arg.arg_repetition = eArgRepeatPlain;
1867223383edSEnrico Granata 
1868b9c1b51eSKate Stone     // There is only one variant this argument could be; put it into the
1869b9c1b51eSKate Stone     // argument entry.
1870223383edSEnrico Granata     arg1.push_back(cmd_arg);
1871223383edSEnrico Granata 
1872223383edSEnrico Granata     // Push the data for the first argument into the m_arguments vector.
1873223383edSEnrico Granata     m_arguments.push_back(arg1);
1874223383edSEnrico Granata   }
1875223383edSEnrico Granata 
18766e3d8e7fSEugene Zelenko   ~CommandObjectCommandsScriptDelete() override = default;
1877223383edSEnrico Granata 
18785a988416SJim Ingham protected:
1879b9c1b51eSKate Stone   bool DoExecute(Args &command, CommandReturnObject &result) override {
1880223383edSEnrico Granata 
188111eb9c64SZachary Turner     if (command.GetArgumentCount() != 1) {
1882223383edSEnrico Granata       result.AppendError("'command script delete' requires one argument");
1883223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
1884223383edSEnrico Granata       return false;
1885223383edSEnrico Granata     }
1886223383edSEnrico Granata 
18874574a890SZachary Turner     auto cmd_name = command[0].ref;
1888223383edSEnrico Granata 
18894574a890SZachary Turner     if (cmd_name.empty() || !m_interpreter.HasUserCommands() ||
18904574a890SZachary Turner         !m_interpreter.UserCommandExists(cmd_name)) {
1891867e7d17SZachary Turner       result.AppendErrorWithFormat("command %s not found", command[0].c_str());
1892223383edSEnrico Granata       result.SetStatus(eReturnStatusFailed);
18934574a890SZachary Turner       return false;
1894223383edSEnrico Granata     }
1895223383edSEnrico Granata 
18964574a890SZachary Turner     m_interpreter.RemoveUser(cmd_name);
18974574a890SZachary Turner     result.SetStatus(eReturnStatusSuccessFinishResult);
18984574a890SZachary Turner     return true;
1899223383edSEnrico Granata   }
1900223383edSEnrico Granata };
1901223383edSEnrico Granata 
1902223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
1903223383edSEnrico Granata 
1904223383edSEnrico Granata //-------------------------------------------------------------------------
1905223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
1906223383edSEnrico Granata //-------------------------------------------------------------------------
1907223383edSEnrico Granata 
1908b9c1b51eSKate Stone class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword {
1909223383edSEnrico Granata public:
19107428a18cSKate Stone   CommandObjectMultiwordCommandsScript(CommandInterpreter &interpreter)
1911b9c1b51eSKate Stone       : CommandObjectMultiword(
1912b9c1b51eSKate Stone             interpreter, "command script", "Commands for managing custom "
1913b9c1b51eSKate Stone                                            "commands implemented by "
1914b9c1b51eSKate Stone                                            "interpreter scripts.",
1915b9c1b51eSKate Stone             "command script <subcommand> [<subcommand-options>]") {
1916b9c1b51eSKate Stone     LoadSubCommand("add", CommandObjectSP(
1917b9c1b51eSKate Stone                               new CommandObjectCommandsScriptAdd(interpreter)));
1918b9c1b51eSKate Stone     LoadSubCommand(
1919b9c1b51eSKate Stone         "delete",
1920b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptDelete(interpreter)));
1921b9c1b51eSKate Stone     LoadSubCommand(
1922b9c1b51eSKate Stone         "clear",
1923b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptClear(interpreter)));
1924b9c1b51eSKate Stone     LoadSubCommand("list", CommandObjectSP(new CommandObjectCommandsScriptList(
1925b9c1b51eSKate Stone                                interpreter)));
1926b9c1b51eSKate Stone     LoadSubCommand(
1927b9c1b51eSKate Stone         "import",
1928b9c1b51eSKate Stone         CommandObjectSP(new CommandObjectCommandsScriptImport(interpreter)));
1929223383edSEnrico Granata   }
1930223383edSEnrico Granata 
19316e3d8e7fSEugene Zelenko   ~CommandObjectMultiwordCommandsScript() override = default;
1932223383edSEnrico Granata };
1933223383edSEnrico Granata 
1934ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
1935ebc09c36SJim Ingham 
1936ebc09c36SJim Ingham //-------------------------------------------------------------------------
1937ebc09c36SJim Ingham // CommandObjectMultiwordCommands
1938ebc09c36SJim Ingham //-------------------------------------------------------------------------
1939ebc09c36SJim Ingham 
1940b9c1b51eSKate Stone CommandObjectMultiwordCommands::CommandObjectMultiwordCommands(
1941b9c1b51eSKate Stone     CommandInterpreter &interpreter)
1942b9c1b51eSKate Stone     : CommandObjectMultiword(interpreter, "command",
1943b9c1b51eSKate Stone                              "Commands for managing custom LLDB commands.",
1944b9c1b51eSKate Stone                              "command <subcommand> [<subcommand-options>]") {
1945b9c1b51eSKate Stone   LoadSubCommand("source",
1946b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsSource(interpreter)));
1947b9c1b51eSKate Stone   LoadSubCommand("alias",
1948b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsAlias(interpreter)));
1949b9c1b51eSKate Stone   LoadSubCommand("unalias", CommandObjectSP(
1950b9c1b51eSKate Stone                                 new CommandObjectCommandsUnalias(interpreter)));
1951b9c1b51eSKate Stone   LoadSubCommand("delete",
1952b9c1b51eSKate Stone                  CommandObjectSP(new CommandObjectCommandsDelete(interpreter)));
1953b9c1b51eSKate Stone   LoadSubCommand(
1954b9c1b51eSKate Stone       "regex", CommandObjectSP(new CommandObjectCommandsAddRegex(interpreter)));
1955b9c1b51eSKate Stone   LoadSubCommand("history", CommandObjectSP(
1956b9c1b51eSKate Stone                                 new CommandObjectCommandsHistory(interpreter)));
1957b9c1b51eSKate Stone   LoadSubCommand(
1958b9c1b51eSKate Stone       "script",
1959b9c1b51eSKate Stone       CommandObjectSP(new CommandObjectMultiwordCommandsScript(interpreter)));
1960ebc09c36SJim Ingham }
1961ebc09c36SJim Ingham 
19626e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
1963