1ebc09c36SJim Ingham //===-- CommandObjectSource.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"
17*46d4aa21SEnrico Granata #include "CommandObjectHelp.h"
18ebc09c36SJim Ingham #include "lldb/Core/Debugger.h"
1944d93782SGreg Clayton #include "lldb/Core/IOHandler.h"
20be93a35aSEnrico Granata #include "lldb/Core/StringList.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"
277594f14fSEnrico Granata #include "lldb/Interpreter/OptionValueUInt64.h"
28ebc09c36SJim Ingham #include "lldb/Interpreter/Options.h"
2999f0b8f9SEnrico Granata #include "lldb/Interpreter/ScriptInterpreter.h"
30ebc09c36SJim Ingham 
31ebc09c36SJim Ingham using namespace lldb;
32ebc09c36SJim Ingham using namespace lldb_private;
33ebc09c36SJim Ingham 
34ebc09c36SJim Ingham //-------------------------------------------------------------------------
35ebc09c36SJim Ingham // CommandObjectCommandsSource
36ebc09c36SJim Ingham //-------------------------------------------------------------------------
37ebc09c36SJim Ingham 
385a988416SJim Ingham class CommandObjectCommandsHistory : public CommandObjectParsed
39a5a97ebeSJim Ingham {
405a988416SJim Ingham public:
415a988416SJim Ingham     CommandObjectCommandsHistory(CommandInterpreter &interpreter) :
425a988416SJim Ingham         CommandObjectParsed(interpreter,
435a988416SJim Ingham                             "command history",
445a988416SJim Ingham                             "Dump the history of commands in this session.",
456e3d8e7fSEugene Zelenko                             nullptr),
465a988416SJim Ingham         m_options (interpreter)
475a988416SJim Ingham     {
485a988416SJim Ingham     }
495a988416SJim Ingham 
506e3d8e7fSEugene Zelenko     ~CommandObjectCommandsHistory() override = default;
515a988416SJim Ingham 
5213d21e9aSBruce Mitchener     Options *
5313d21e9aSBruce Mitchener     GetOptions () override
545a988416SJim Ingham     {
555a988416SJim Ingham         return &m_options;
565a988416SJim Ingham     }
575a988416SJim Ingham 
585a988416SJim Ingham protected:
59a5a97ebeSJim Ingham     class CommandOptions : public Options
60a5a97ebeSJim Ingham     {
61a5a97ebeSJim Ingham     public:
62a5a97ebeSJim Ingham         CommandOptions (CommandInterpreter &interpreter) :
637594f14fSEnrico Granata             Options (interpreter),
647594f14fSEnrico Granata             m_start_idx(0),
657594f14fSEnrico Granata             m_stop_idx(0),
667594f14fSEnrico Granata             m_count(0),
6763123b64SEnrico Granata             m_clear(false)
68a5a97ebeSJim Ingham         {
69a5a97ebeSJim Ingham         }
70a5a97ebeSJim Ingham 
716e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
72a5a97ebeSJim Ingham 
7313d21e9aSBruce Mitchener         Error
7413d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
75a5a97ebeSJim Ingham         {
76a5a97ebeSJim Ingham             Error error;
773bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
78a5a97ebeSJim Ingham 
79a5a97ebeSJim Ingham             switch (short_option)
80a5a97ebeSJim Ingham             {
81a5a97ebeSJim Ingham                 case 'c':
82c95f7e2aSPavel Labath                     error = m_count.SetValueFromString(option_arg,eVarSetOperationAssign);
83a5a97ebeSJim Ingham                     break;
84a5a97ebeSJim Ingham                 case 's':
857594f14fSEnrico Granata                     if (option_arg && strcmp("end", option_arg) == 0)
867594f14fSEnrico Granata                     {
877594f14fSEnrico Granata                         m_start_idx.SetCurrentValue(UINT64_MAX);
887594f14fSEnrico Granata                         m_start_idx.SetOptionWasSet();
897594f14fSEnrico Granata                     }
907594f14fSEnrico Granata                     else
91c95f7e2aSPavel Labath                         error = m_start_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
927594f14fSEnrico Granata                     break;
937594f14fSEnrico Granata                 case 'e':
94c95f7e2aSPavel Labath                     error = m_stop_idx.SetValueFromString(option_arg,eVarSetOperationAssign);
957594f14fSEnrico Granata                     break;
9663123b64SEnrico Granata                 case 'C':
9763123b64SEnrico Granata                     m_clear.SetCurrentValue(true);
9863123b64SEnrico Granata                     m_clear.SetOptionWasSet();
99a5a97ebeSJim Ingham                     break;
100a5a97ebeSJim Ingham                 default:
10186edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
102a5a97ebeSJim Ingham                     break;
103a5a97ebeSJim Ingham             }
104a5a97ebeSJim Ingham 
105a5a97ebeSJim Ingham             return error;
106a5a97ebeSJim Ingham         }
107a5a97ebeSJim Ingham 
108a5a97ebeSJim Ingham         void
10913d21e9aSBruce Mitchener         OptionParsingStarting () override
110a5a97ebeSJim Ingham         {
1117594f14fSEnrico Granata             m_start_idx.Clear();
1127594f14fSEnrico Granata             m_stop_idx.Clear();
1137594f14fSEnrico Granata             m_count.Clear();
11463123b64SEnrico Granata             m_clear.Clear();
115a5a97ebeSJim Ingham         }
116a5a97ebeSJim Ingham 
117a5a97ebeSJim Ingham         const OptionDefinition*
11813d21e9aSBruce Mitchener         GetDefinitions () override
119a5a97ebeSJim Ingham         {
120a5a97ebeSJim Ingham             return g_option_table;
121a5a97ebeSJim Ingham         }
122a5a97ebeSJim Ingham 
123a5a97ebeSJim Ingham         // Options table: Required for subclasses of Options.
124a5a97ebeSJim Ingham 
125a5a97ebeSJim Ingham         static OptionDefinition g_option_table[];
126a5a97ebeSJim Ingham 
127a5a97ebeSJim Ingham         // Instance variables to hold the values for command options.
128a5a97ebeSJim Ingham 
1297594f14fSEnrico Granata         OptionValueUInt64 m_start_idx;
1307594f14fSEnrico Granata         OptionValueUInt64 m_stop_idx;
1317594f14fSEnrico Granata         OptionValueUInt64 m_count;
13263123b64SEnrico Granata         OptionValueBoolean m_clear;
133a5a97ebeSJim Ingham     };
134a5a97ebeSJim Ingham 
135a5a97ebeSJim Ingham     bool
13613d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
137a5a97ebeSJim Ingham     {
13863123b64SEnrico Granata         if (m_options.m_clear.GetCurrentValue() && m_options.m_clear.OptionWasSet())
1397594f14fSEnrico Granata         {
1407594f14fSEnrico Granata             m_interpreter.GetCommandHistory().Clear();
1417594f14fSEnrico Granata             result.SetStatus(lldb::eReturnStatusSuccessFinishNoResult);
1427594f14fSEnrico Granata         }
1437594f14fSEnrico Granata         else
1447594f14fSEnrico Granata         {
1457594f14fSEnrico Granata             if (m_options.m_start_idx.OptionWasSet() && m_options.m_stop_idx.OptionWasSet() && m_options.m_count.OptionWasSet())
1467594f14fSEnrico Granata             {
1477594f14fSEnrico Granata                 result.AppendError("--count, --start-index and --end-index cannot be all specified in the same invocation");
1487594f14fSEnrico Granata                 result.SetStatus(lldb::eReturnStatusFailed);
1497594f14fSEnrico Granata             }
1507594f14fSEnrico Granata             else
1517594f14fSEnrico Granata             {
15284400ec7SVirgile Bello                 std::pair<bool,uint64_t> start_idx(m_options.m_start_idx.OptionWasSet(),m_options.m_start_idx.GetCurrentValue());
15384400ec7SVirgile Bello                 std::pair<bool,uint64_t> stop_idx(m_options.m_stop_idx.OptionWasSet(),m_options.m_stop_idx.GetCurrentValue());
15484400ec7SVirgile Bello                 std::pair<bool,uint64_t> count(m_options.m_count.OptionWasSet(),m_options.m_count.GetCurrentValue());
155a5a97ebeSJim Ingham 
1567594f14fSEnrico Granata                 const CommandHistory& history(m_interpreter.GetCommandHistory());
1577594f14fSEnrico Granata 
1587594f14fSEnrico Granata                 if (start_idx.first && start_idx.second == UINT64_MAX)
1597594f14fSEnrico Granata                 {
1607594f14fSEnrico Granata                     if (count.first)
1617594f14fSEnrico Granata                     {
1627594f14fSEnrico Granata                         start_idx.second = history.GetSize() - count.second;
1637594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1647594f14fSEnrico Granata                     }
1657594f14fSEnrico Granata                     else if (stop_idx.first)
1667594f14fSEnrico Granata                     {
1677594f14fSEnrico Granata                         start_idx.second = stop_idx.second;
1687594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1697594f14fSEnrico Granata                     }
1707594f14fSEnrico Granata                     else
1717594f14fSEnrico Granata                     {
1727594f14fSEnrico Granata                         start_idx.second = 0;
1737594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1747594f14fSEnrico Granata                     }
1757594f14fSEnrico Granata                 }
1767594f14fSEnrico Granata                 else
1777594f14fSEnrico Granata                 {
1787594f14fSEnrico Granata                     if (!start_idx.first && !stop_idx.first && !count.first)
1797594f14fSEnrico Granata                     {
1807594f14fSEnrico Granata                         start_idx.second = 0;
1817594f14fSEnrico Granata                         stop_idx.second = history.GetSize() - 1;
1827594f14fSEnrico Granata                     }
1837594f14fSEnrico Granata                     else if (start_idx.first)
1847594f14fSEnrico Granata                     {
1857594f14fSEnrico Granata                         if (count.first)
1867594f14fSEnrico Granata                         {
1877594f14fSEnrico Granata                             stop_idx.second = start_idx.second + count.second - 1;
1887594f14fSEnrico Granata                         }
1897594f14fSEnrico Granata                         else if (!stop_idx.first)
1907594f14fSEnrico Granata                         {
1917594f14fSEnrico Granata                             stop_idx.second = history.GetSize() - 1;
1927594f14fSEnrico Granata                         }
1937594f14fSEnrico Granata                     }
1947594f14fSEnrico Granata                     else if (stop_idx.first)
1957594f14fSEnrico Granata                     {
1967594f14fSEnrico Granata                         if (count.first)
1977594f14fSEnrico Granata                         {
1987594f14fSEnrico Granata                             if (stop_idx.second >= count.second)
1997594f14fSEnrico Granata                                 start_idx.second = stop_idx.second - count.second + 1;
2007594f14fSEnrico Granata                             else
2017594f14fSEnrico Granata                                 start_idx.second = 0;
2027594f14fSEnrico Granata                         }
2037594f14fSEnrico Granata                     }
2047594f14fSEnrico Granata                     else /* if (count.first) */
2057594f14fSEnrico Granata                     {
2067594f14fSEnrico Granata                         start_idx.second = 0;
2077594f14fSEnrico Granata                         stop_idx.second = count.second - 1;
2087594f14fSEnrico Granata                     }
2097594f14fSEnrico Granata                 }
2107594f14fSEnrico Granata                 history.Dump(result.GetOutputStream(), start_idx.second, stop_idx.second);
2117594f14fSEnrico Granata             }
2127594f14fSEnrico Granata         }
213a5a97ebeSJim Ingham         return result.Succeeded();
214a5a97ebeSJim Ingham 
215a5a97ebeSJim Ingham     }
2165a988416SJim Ingham 
2175a988416SJim Ingham     CommandOptions m_options;
218a5a97ebeSJim Ingham };
219a5a97ebeSJim Ingham 
220a5a97ebeSJim Ingham OptionDefinition
221a5a97ebeSJim Ingham CommandObjectCommandsHistory::CommandOptions::g_option_table[] =
222a5a97ebeSJim Ingham {
2236e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "count", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,        "How many history commands to print."},
2246e3d8e7fSEugene Zelenko { 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)."},
2256e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "end-index", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeUnsignedInteger,    "Index at which to stop printing history commands."},
2266e3d8e7fSEugene Zelenko { LLDB_OPT_SET_2, false, "clear", 'C', OptionParser::eNoArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "Clears the current command history."},
2276e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
228a5a97ebeSJim Ingham };
229a5a97ebeSJim Ingham 
230a5a97ebeSJim Ingham //-------------------------------------------------------------------------
231a5a97ebeSJim Ingham // CommandObjectCommandsSource
232a5a97ebeSJim Ingham //-------------------------------------------------------------------------
233a5a97ebeSJim Ingham 
2345a988416SJim Ingham class CommandObjectCommandsSource : public CommandObjectParsed
235ebc09c36SJim Ingham {
2365a988416SJim Ingham public:
2375a988416SJim Ingham     CommandObjectCommandsSource(CommandInterpreter &interpreter) :
2385a988416SJim Ingham         CommandObjectParsed(interpreter,
2395a988416SJim Ingham                             "command source",
2405a988416SJim Ingham                             "Read in debugger commands from the file <filename> and execute them.",
2416e3d8e7fSEugene Zelenko                             nullptr),
2425a988416SJim Ingham         m_options (interpreter)
2435a988416SJim Ingham     {
2445a988416SJim Ingham         CommandArgumentEntry arg;
2455a988416SJim Ingham         CommandArgumentData file_arg;
2465a988416SJim Ingham 
2475a988416SJim Ingham         // Define the first (and only) variant of this arg.
2485a988416SJim Ingham         file_arg.arg_type = eArgTypeFilename;
2495a988416SJim Ingham         file_arg.arg_repetition = eArgRepeatPlain;
2505a988416SJim Ingham 
2515a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
2525a988416SJim Ingham         arg.push_back (file_arg);
2535a988416SJim Ingham 
2545a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
2555a988416SJim Ingham         m_arguments.push_back (arg);
2565a988416SJim Ingham     }
2575a988416SJim Ingham 
2586e3d8e7fSEugene Zelenko     ~CommandObjectCommandsSource() override = default;
2595a988416SJim Ingham 
26013d21e9aSBruce Mitchener     const char*
26113d21e9aSBruce Mitchener     GetRepeatCommand (Args &current_command_args, uint32_t index) override
2625a988416SJim Ingham     {
2635a988416SJim Ingham         return "";
2645a988416SJim Ingham     }
2655a988416SJim Ingham 
26613d21e9aSBruce Mitchener     int
2675a988416SJim Ingham     HandleArgumentCompletion (Args &input,
2685a988416SJim Ingham                               int &cursor_index,
2695a988416SJim Ingham                               int &cursor_char_position,
2705a988416SJim Ingham                               OptionElementVector &opt_element_vector,
2715a988416SJim Ingham                               int match_start_point,
2725a988416SJim Ingham                               int max_return_elements,
2735a988416SJim Ingham                               bool &word_complete,
27413d21e9aSBruce Mitchener                               StringList &matches) override
2755a988416SJim Ingham     {
2765a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
2775a988416SJim Ingham         completion_str.erase (cursor_char_position);
2785a988416SJim Ingham 
2795a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
2805a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
2815a988416SJim Ingham                                                             completion_str.c_str(),
2825a988416SJim Ingham                                                             match_start_point,
2835a988416SJim Ingham                                                             max_return_elements,
2846e3d8e7fSEugene Zelenko                                                             nullptr,
2855a988416SJim Ingham                                                             word_complete,
2865a988416SJim Ingham                                                             matches);
2875a988416SJim Ingham         return matches.GetSize();
2885a988416SJim Ingham     }
2895a988416SJim Ingham 
29013d21e9aSBruce Mitchener     Options *
29113d21e9aSBruce Mitchener     GetOptions () override
2925a988416SJim Ingham     {
2935a988416SJim Ingham         return &m_options;
2945a988416SJim Ingham     }
2955a988416SJim Ingham 
2965a988416SJim Ingham protected:
297e16c50a1SJim Ingham     class CommandOptions : public Options
298e16c50a1SJim Ingham     {
299e16c50a1SJim Ingham     public:
300eb0103f2SGreg Clayton         CommandOptions (CommandInterpreter &interpreter) :
301012d4fcaSEnrico Granata             Options (interpreter),
302340b0309SGreg Clayton             m_stop_on_error (true),
303340b0309SGreg Clayton             m_silent_run (false),
304340b0309SGreg Clayton             m_stop_on_continue (true)
305eb0103f2SGreg Clayton         {
306eb0103f2SGreg Clayton         }
307e16c50a1SJim Ingham 
3086e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
309e16c50a1SJim Ingham 
31013d21e9aSBruce Mitchener         Error
31113d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
312e16c50a1SJim Ingham         {
313e16c50a1SJim Ingham             Error error;
3143bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
315e16c50a1SJim Ingham 
316e16c50a1SJim Ingham             switch (short_option)
317e16c50a1SJim Ingham             {
318e16c50a1SJim Ingham                 case 'e':
319c95f7e2aSPavel Labath                     error = m_stop_on_error.SetValueFromString(option_arg);
320e16c50a1SJim Ingham                     break;
321340b0309SGreg Clayton 
322e16c50a1SJim Ingham                 case 'c':
323c95f7e2aSPavel Labath                     error = m_stop_on_continue.SetValueFromString(option_arg);
324e16c50a1SJim Ingham                     break;
325340b0309SGreg Clayton 
32660986174SMichael Sartain                 case 's':
327c95f7e2aSPavel Labath                     error = m_silent_run.SetValueFromString(option_arg);
32860986174SMichael Sartain                     break;
329340b0309SGreg Clayton 
330e16c50a1SJim Ingham                 default:
33186edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
332e16c50a1SJim Ingham                     break;
333e16c50a1SJim Ingham             }
334e16c50a1SJim Ingham 
335e16c50a1SJim Ingham             return error;
336e16c50a1SJim Ingham         }
337e16c50a1SJim Ingham 
338e16c50a1SJim Ingham         void
33913d21e9aSBruce Mitchener         OptionParsingStarting () override
340e16c50a1SJim Ingham         {
341012d4fcaSEnrico Granata             m_stop_on_error.Clear();
342340b0309SGreg Clayton             m_silent_run.Clear();
343340b0309SGreg Clayton             m_stop_on_continue.Clear();
344e16c50a1SJim Ingham         }
345e16c50a1SJim Ingham 
346e0d378b3SGreg Clayton         const OptionDefinition*
34713d21e9aSBruce Mitchener         GetDefinitions () override
348e16c50a1SJim Ingham         {
349e16c50a1SJim Ingham             return g_option_table;
350e16c50a1SJim Ingham         }
351e16c50a1SJim Ingham 
352e16c50a1SJim Ingham         // Options table: Required for subclasses of Options.
353e16c50a1SJim Ingham 
354e0d378b3SGreg Clayton         static OptionDefinition g_option_table[];
355e16c50a1SJim Ingham 
356e16c50a1SJim Ingham         // Instance variables to hold the values for command options.
357e16c50a1SJim Ingham 
358012d4fcaSEnrico Granata         OptionValueBoolean m_stop_on_error;
359340b0309SGreg Clayton         OptionValueBoolean m_silent_run;
360340b0309SGreg Clayton         OptionValueBoolean m_stop_on_continue;
361e16c50a1SJim Ingham     };
362e16c50a1SJim Ingham 
363ebc09c36SJim Ingham     bool
36413d21e9aSBruce Mitchener     DoExecute(Args& command, CommandReturnObject &result) override
365ebc09c36SJim Ingham     {
366c7bece56SGreg Clayton         const size_t argc = command.GetArgumentCount();
367ebc09c36SJim Ingham         if (argc == 1)
368ebc09c36SJim Ingham         {
3695a988416SJim Ingham             const char *filename = command.GetArgumentAtIndex(0);
370ebc09c36SJim Ingham 
3711ee3853fSJohnny Chen             FileSpec cmd_file (filename, true);
3726e3d8e7fSEugene Zelenko             ExecutionContext *exe_ctx = nullptr;  // Just use the default context.
373ebc09c36SJim Ingham 
374340b0309SGreg Clayton             // If any options were set, then use them
375340b0309SGreg Clayton             if (m_options.m_stop_on_error.OptionWasSet()    ||
376340b0309SGreg Clayton                 m_options.m_silent_run.OptionWasSet()       ||
377340b0309SGreg Clayton                 m_options.m_stop_on_continue.OptionWasSet())
378340b0309SGreg Clayton             {
379340b0309SGreg Clayton                 // Use user set settings
38026c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
38126c7bf93SJim Ingham                 options.SetStopOnContinue(m_options.m_stop_on_continue.GetCurrentValue());
38226c7bf93SJim Ingham                 options.SetStopOnError (m_options.m_stop_on_error.GetCurrentValue());
3837d8555c4SJim Ingham                 options.SetEchoCommands (!m_options.m_silent_run.GetCurrentValue());
3847d8555c4SJim Ingham                 options.SetPrintResults (!m_options.m_silent_run.GetCurrentValue());
38526c7bf93SJim Ingham 
386e16c50a1SJim Ingham                 m_interpreter.HandleCommandsFromFile (cmd_file,
387e16c50a1SJim Ingham                                                       exe_ctx,
38826c7bf93SJim Ingham                                                       options,
389e16c50a1SJim Ingham                                                       result);
390340b0309SGreg Clayton             }
391340b0309SGreg Clayton             else
392340b0309SGreg Clayton             {
393340b0309SGreg Clayton                 // No options were set, inherit any settings from nested "command source" commands,
394340b0309SGreg Clayton                 // or set to sane default settings...
39526c7bf93SJim Ingham                 CommandInterpreterRunOptions options;
396340b0309SGreg Clayton                 m_interpreter.HandleCommandsFromFile (cmd_file,
397340b0309SGreg Clayton                                                       exe_ctx,
39826c7bf93SJim Ingham                                                       options,
399340b0309SGreg Clayton                                                       result);
400340b0309SGreg Clayton             }
401ebc09c36SJim Ingham         }
402ebc09c36SJim Ingham         else
403ebc09c36SJim Ingham         {
404ebc09c36SJim Ingham             result.AppendErrorWithFormat("'%s' takes exactly one executable filename argument.\n", GetCommandName());
405ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
406ebc09c36SJim Ingham         }
407ebc09c36SJim Ingham         return result.Succeeded();
408ebc09c36SJim Ingham     }
4096e3d8e7fSEugene Zelenko 
4105a988416SJim Ingham     CommandOptions m_options;
411ebc09c36SJim Ingham };
412ebc09c36SJim Ingham 
413e0d378b3SGreg Clayton OptionDefinition
414e16c50a1SJim Ingham CommandObjectCommandsSource::CommandOptions::g_option_table[] =
415e16c50a1SJim Ingham {
4166e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-error", 'e', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean,    "If true, stop executing commands on error."},
4176e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "stop-on-continue", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true, stop executing commands on continue."},
4186e3d8e7fSEugene Zelenko { LLDB_OPT_SET_ALL, false, "silent-run", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeBoolean, "If true don't echo commands while executing."},
4196e3d8e7fSEugene Zelenko { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
420e16c50a1SJim Ingham };
421e16c50a1SJim Ingham 
422ebc09c36SJim Ingham #pragma mark CommandObjectCommandsAlias
423ebc09c36SJim Ingham //-------------------------------------------------------------------------
424ebc09c36SJim Ingham // CommandObjectCommandsAlias
425ebc09c36SJim Ingham //-------------------------------------------------------------------------
426ebc09c36SJim Ingham 
427be93a35aSEnrico Granata static const char *g_python_command_instructions =   "Enter your Python command(s). Type 'DONE' to end.\n"
428be93a35aSEnrico Granata                                                      "You must define a Python function with this signature:\n"
42944d93782SGreg Clayton                                                      "def my_command_impl(debugger, args, result, internal_dict):\n";
430be93a35aSEnrico Granata 
4315a988416SJim Ingham class CommandObjectCommandsAlias : public CommandObjectRaw
432ebc09c36SJim Ingham {
433ebc09c36SJim Ingham public:
434a7015092SGreg Clayton     CommandObjectCommandsAlias (CommandInterpreter &interpreter) :
4355a988416SJim Ingham         CommandObjectRaw(interpreter,
4360e5e5a79SGreg Clayton                          "command alias",
437e3d26315SCaroline Tice                          "Allow users to define their own debugger command abbreviations.",
4386e3d8e7fSEugene Zelenko                          nullptr)
439ebc09c36SJim Ingham     {
440ebc09c36SJim Ingham         SetHelpLong(
441ea671fbdSKate Stone "'alias' allows the user to create a short-cut or abbreviation for long \
442ea671fbdSKate Stone commands, multi-word commands, and commands that take particular options.  \
443ea671fbdSKate Stone Below are some simple examples of how one might use the 'alias' command:" R"(
444ea671fbdSKate Stone 
445ea671fbdSKate Stone (lldb) command alias sc script
446ea671fbdSKate Stone 
447ea671fbdSKate Stone     Creates the abbreviation 'sc' for the 'script' command.
448ea671fbdSKate Stone 
449ea671fbdSKate Stone (lldb) command alias bp breakpoint
450ea671fbdSKate Stone 
451ea671fbdSKate Stone )" "    Creates the abbreviation 'bp' for the 'breakpoint' command.  Since \
452ea671fbdSKate Stone breakpoint commands are two-word commands, the user would still need to \
453ea671fbdSKate Stone enter the second word after 'bp', e.g. 'bp enable' or 'bp delete'." R"(
454ea671fbdSKate Stone 
455ea671fbdSKate Stone (lldb) command alias bpl breakpoint list
456ea671fbdSKate Stone 
457ea671fbdSKate Stone     Creates the abbreviation 'bpl' for the two-word command 'breakpoint list'.
458ea671fbdSKate Stone 
459ea671fbdSKate Stone )" "An alias can include some options for the command, with the values either \
460ea671fbdSKate Stone filled in at the time the alias is created, or specified as positional \
461ea671fbdSKate Stone arguments, to be filled in when the alias is invoked.  The following example \
462ea671fbdSKate Stone shows how to create aliases with options:" R"(
463ea671fbdSKate Stone 
464ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
465ea671fbdSKate Stone 
466ea671fbdSKate Stone )" "    Creates the abbreviation 'bfl' (for break-file-line), with the -f and -l \
467ea671fbdSKate Stone options already part of the alias.  So if the user wants to set a breakpoint \
468ea671fbdSKate Stone by file and line without explicitly having to use the -f and -l options, the \
469ea671fbdSKate Stone user can now use 'bfl' instead.  The '%1' and '%2' are positional placeholders \
470ea671fbdSKate Stone for the actual arguments that will be passed when the alias command is used.  \
471ea671fbdSKate Stone The number in the placeholder refers to the position/order the actual value \
472ea671fbdSKate Stone occupies when the alias is used.  All the occurrences of '%1' in the alias \
473ea671fbdSKate Stone will be replaced with the first argument, all the occurrences of '%2' in the \
474ea671fbdSKate Stone alias will be replaced with the second argument, and so on.  This also allows \
475ea671fbdSKate Stone actual arguments to be used multiple times within an alias (see 'process \
476ea671fbdSKate Stone launch' example below)." R"(
477ea671fbdSKate Stone 
478ea671fbdSKate Stone )" "Note: the positional arguments must substitute as whole words in the resultant \
479ea671fbdSKate Stone command, so you can't at present do something like this to append the file extension \
480ea671fbdSKate Stone \".cpp\":" R"(
481ea671fbdSKate Stone 
482ea671fbdSKate Stone (lldb) command alias bcppfl breakpoint set -f %1.cpp -l %2
483ea671fbdSKate Stone 
484ea671fbdSKate Stone )" "For more complex aliasing, use the \"command regex\" command instead.  In the \
485ea671fbdSKate Stone 'bfl' case above, the actual file value will be filled in with the first argument \
486ea671fbdSKate Stone following 'bfl' and the actual line number value will be filled in with the second \
487ea671fbdSKate Stone argument.  The user would use this alias as follows:" R"(
488ea671fbdSKate Stone 
489ea671fbdSKate Stone (lldb) command alias bfl breakpoint set -f %1 -l %2
490ea671fbdSKate Stone (lldb) bfl my-file.c 137
491ea671fbdSKate Stone 
492ea671fbdSKate Stone This would be the same as if the user had entered 'breakpoint set -f my-file.c -l 137'.
493ea671fbdSKate Stone 
494ea671fbdSKate Stone Another example:
495ea671fbdSKate Stone 
496ea671fbdSKate Stone (lldb) command alias pltty process launch -s -o %1 -e %1
497ea671fbdSKate Stone (lldb) pltty /dev/tty0
498ea671fbdSKate Stone 
499ea671fbdSKate Stone     Interpreted as 'process launch -s -o /dev/tty0 -e /dev/tty0'
500ea671fbdSKate Stone 
501ea671fbdSKate Stone )" "If the user always wanted to pass the same value to a particular option, the \
502ea671fbdSKate Stone alias could be defined with that value directly in the alias as a constant, \
503ea671fbdSKate Stone rather than using a positional placeholder:" R"(
504ea671fbdSKate Stone 
505ea671fbdSKate Stone (lldb) command alias bl3 breakpoint set -f %1 -l 3
506ea671fbdSKate Stone 
507ea671fbdSKate Stone     Always sets a breakpoint on line 3 of whatever file is indicated.)"
508ea671fbdSKate Stone         );
509ebc09c36SJim Ingham 
510405fe67fSCaroline Tice         CommandArgumentEntry arg1;
511405fe67fSCaroline Tice         CommandArgumentEntry arg2;
512405fe67fSCaroline Tice         CommandArgumentEntry arg3;
513405fe67fSCaroline Tice         CommandArgumentData alias_arg;
514405fe67fSCaroline Tice         CommandArgumentData cmd_arg;
515405fe67fSCaroline Tice         CommandArgumentData options_arg;
516405fe67fSCaroline Tice 
517405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
518405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
519405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
520405fe67fSCaroline Tice 
521405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
522405fe67fSCaroline Tice         arg1.push_back (alias_arg);
523405fe67fSCaroline Tice 
524405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
525405fe67fSCaroline Tice         cmd_arg.arg_type = eArgTypeCommandName;
526405fe67fSCaroline Tice         cmd_arg.arg_repetition = eArgRepeatPlain;
527405fe67fSCaroline Tice 
528405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
529405fe67fSCaroline Tice         arg2.push_back (cmd_arg);
530405fe67fSCaroline Tice 
531405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
532405fe67fSCaroline Tice         options_arg.arg_type = eArgTypeAliasOptions;
533405fe67fSCaroline Tice         options_arg.arg_repetition = eArgRepeatOptional;
534405fe67fSCaroline Tice 
535405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
536405fe67fSCaroline Tice         arg3.push_back (options_arg);
537405fe67fSCaroline Tice 
538405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
539405fe67fSCaroline Tice         m_arguments.push_back (arg1);
540405fe67fSCaroline Tice         m_arguments.push_back (arg2);
541405fe67fSCaroline Tice         m_arguments.push_back (arg3);
542ebc09c36SJim Ingham     }
543ebc09c36SJim Ingham 
5446e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAlias() override = default;
545ebc09c36SJim Ingham 
5465a988416SJim Ingham protected:
54713d21e9aSBruce Mitchener     bool
54813d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
549844d2303SCaroline Tice     {
550844d2303SCaroline Tice         Args args (raw_command_line);
551844d2303SCaroline Tice         std::string raw_command_string (raw_command_line);
552844d2303SCaroline Tice 
553844d2303SCaroline Tice         size_t argc = args.GetArgumentCount();
554844d2303SCaroline Tice 
555844d2303SCaroline Tice         if (argc < 2)
556844d2303SCaroline Tice         {
557844d2303SCaroline Tice             result.AppendError ("'alias' requires at least two arguments");
558844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
559844d2303SCaroline Tice             return false;
560844d2303SCaroline Tice         }
561844d2303SCaroline Tice 
562844d2303SCaroline Tice         // Get the alias command.
563844d2303SCaroline Tice 
564844d2303SCaroline Tice         const std::string alias_command = args.GetArgumentAtIndex (0);
565844d2303SCaroline Tice 
566844d2303SCaroline Tice         // Strip the new alias name off 'raw_command_string'  (leave it on args, which gets passed to 'Execute', which
567844d2303SCaroline Tice         // does the stripping itself.
568844d2303SCaroline Tice         size_t pos = raw_command_string.find (alias_command);
569844d2303SCaroline Tice         if (pos == 0)
570844d2303SCaroline Tice         {
571844d2303SCaroline Tice             raw_command_string = raw_command_string.substr (alias_command.size());
572844d2303SCaroline Tice             pos = raw_command_string.find_first_not_of (' ');
573844d2303SCaroline Tice             if ((pos != std::string::npos) && (pos > 0))
574844d2303SCaroline Tice                 raw_command_string = raw_command_string.substr (pos);
575844d2303SCaroline Tice         }
576844d2303SCaroline Tice         else
577844d2303SCaroline Tice         {
578844d2303SCaroline Tice             result.AppendError ("Error parsing command string.  No alias created.");
579844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
580844d2303SCaroline Tice             return false;
581844d2303SCaroline Tice         }
582844d2303SCaroline Tice 
583844d2303SCaroline Tice 
584844d2303SCaroline Tice         // Verify that the command is alias-able.
585844d2303SCaroline Tice         if (m_interpreter.CommandExists (alias_command.c_str()))
586844d2303SCaroline Tice         {
587844d2303SCaroline Tice             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
588844d2303SCaroline Tice                                           alias_command.c_str());
589844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
590844d2303SCaroline Tice             return false;
591844d2303SCaroline Tice         }
592844d2303SCaroline Tice 
593844d2303SCaroline Tice         // Get CommandObject that is being aliased. The command name is read from the front of raw_command_string.
594844d2303SCaroline Tice         // raw_command_string is returned with the name of the command object stripped off the front.
595844d2303SCaroline Tice         CommandObject *cmd_obj = m_interpreter.GetCommandObjectForCommand (raw_command_string);
596844d2303SCaroline Tice 
597844d2303SCaroline Tice         if (!cmd_obj)
598844d2303SCaroline Tice         {
59986edbf41SGreg Clayton             result.AppendErrorWithFormat ("invalid command given to 'alias'. '%s' does not begin with a valid command."
600844d2303SCaroline Tice                                           "  No alias created.", raw_command_string.c_str());
601844d2303SCaroline Tice             result.SetStatus (eReturnStatusFailed);
602844d2303SCaroline Tice             return false;
603844d2303SCaroline Tice         }
604844d2303SCaroline Tice         else if (!cmd_obj->WantsRawCommandString ())
605844d2303SCaroline Tice         {
606844d2303SCaroline Tice             // Note that args was initialized with the original command, and has not been updated to this point.
607844d2303SCaroline Tice             // Therefore can we pass it to the version of Execute that does not need/expect raw input in the alias.
6085a988416SJim Ingham             return HandleAliasingNormalCommand (args, result);
609844d2303SCaroline Tice         }
610844d2303SCaroline Tice         else
611844d2303SCaroline Tice         {
6125a988416SJim Ingham             return HandleAliasingRawCommand (alias_command, raw_command_string, *cmd_obj, result);
6135a988416SJim Ingham         }
6145a988416SJim Ingham         return result.Succeeded();
6155a988416SJim Ingham     }
6165a988416SJim Ingham 
6175a988416SJim Ingham     bool
6185a988416SJim Ingham     HandleAliasingRawCommand (const std::string &alias_command, std::string &raw_command_string, CommandObject &cmd_obj, CommandReturnObject &result)
6195a988416SJim Ingham     {
620844d2303SCaroline Tice             // Verify & handle any options/arguments passed to the alias command
621844d2303SCaroline Tice 
622844d2303SCaroline Tice             OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
623844d2303SCaroline Tice             OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
624844d2303SCaroline Tice 
6255a988416SJim Ingham             CommandObjectSP cmd_obj_sp = m_interpreter.GetCommandSPExact (cmd_obj.GetCommandName(), false);
626844d2303SCaroline Tice 
627ca90c47eSCaroline Tice             if (!m_interpreter.ProcessAliasOptionsArgs (cmd_obj_sp, raw_command_string.c_str(), option_arg_vector_sp))
628844d2303SCaroline Tice             {
629844d2303SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
630ca90c47eSCaroline Tice                 result.SetStatus (eReturnStatusFailed);
631844d2303SCaroline Tice                 return false;
632844d2303SCaroline Tice             }
633844d2303SCaroline Tice 
634844d2303SCaroline Tice             // Create the alias
635844d2303SCaroline Tice             if (m_interpreter.AliasExists (alias_command.c_str())
636844d2303SCaroline Tice                 || m_interpreter.UserCommandExists (alias_command.c_str()))
637844d2303SCaroline Tice             {
638844d2303SCaroline Tice                 OptionArgVectorSP temp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
6396e3d8e7fSEugene Zelenko                 if (temp_option_arg_sp)
640844d2303SCaroline Tice                 {
6416e3d8e7fSEugene Zelenko                     if (option_arg_vector->empty())
642844d2303SCaroline Tice                         m_interpreter.RemoveAliasOptions (alias_command.c_str());
643844d2303SCaroline Tice                 }
644844d2303SCaroline Tice                 result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
645844d2303SCaroline Tice                                                 alias_command.c_str());
646844d2303SCaroline Tice             }
647844d2303SCaroline Tice 
648472362e6SCaroline Tice             if (cmd_obj_sp)
649472362e6SCaroline Tice             {
650844d2303SCaroline Tice                 m_interpreter.AddAlias (alias_command.c_str(), cmd_obj_sp);
6516e3d8e7fSEugene Zelenko                 if (!option_arg_vector->empty())
652844d2303SCaroline Tice                     m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
653844d2303SCaroline Tice                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
654844d2303SCaroline Tice             }
655472362e6SCaroline Tice             else
656472362e6SCaroline Tice             {
657472362e6SCaroline Tice                 result.AppendError ("Unable to create requested alias.\n");
658472362e6SCaroline Tice                 result.SetStatus (eReturnStatusFailed);
659472362e6SCaroline Tice             }
660844d2303SCaroline Tice             return result.Succeeded ();
661844d2303SCaroline Tice     }
662ebc09c36SJim Ingham 
663ebc09c36SJim Ingham     bool
6645a988416SJim Ingham     HandleAliasingNormalCommand (Args& args, CommandReturnObject &result)
665ebc09c36SJim Ingham     {
666867b185dSCaroline Tice         size_t argc = args.GetArgumentCount();
667ebc09c36SJim Ingham 
668ebc09c36SJim Ingham         if (argc < 2)
669ebc09c36SJim Ingham         {
670ebc09c36SJim Ingham             result.AppendError ("'alias' requires at least two arguments");
671ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
672ebc09c36SJim Ingham             return false;
673ebc09c36SJim Ingham         }
674ebc09c36SJim Ingham 
675ebc09c36SJim Ingham         const std::string alias_command = args.GetArgumentAtIndex(0);
676ebc09c36SJim Ingham         const std::string actual_command = args.GetArgumentAtIndex(1);
677ebc09c36SJim Ingham 
678ebc09c36SJim Ingham         args.Shift();  // Shift the alias command word off the argument vector.
679ebc09c36SJim Ingham         args.Shift();  // Shift the old command word off the argument vector.
680ebc09c36SJim Ingham 
681ebc09c36SJim Ingham         // Verify that the command is alias'able, and get the appropriate command object.
682ebc09c36SJim Ingham 
683a7015092SGreg Clayton         if (m_interpreter.CommandExists (alias_command.c_str()))
684ebc09c36SJim Ingham         {
685ebc09c36SJim Ingham             result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be redefined.\n",
686ebc09c36SJim Ingham                                          alias_command.c_str());
687ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
688ebc09c36SJim Ingham         }
689ebc09c36SJim Ingham         else
690ebc09c36SJim Ingham         {
691a7015092SGreg Clayton              CommandObjectSP command_obj_sp(m_interpreter.GetCommandSPExact (actual_command.c_str(), true));
692ebc09c36SJim Ingham              CommandObjectSP subcommand_obj_sp;
693ebc09c36SJim Ingham              bool use_subcommand = false;
6946e3d8e7fSEugene Zelenko              if (command_obj_sp)
695ebc09c36SJim Ingham              {
696ebc09c36SJim Ingham                  CommandObject *cmd_obj = command_obj_sp.get();
6976e3d8e7fSEugene Zelenko                  CommandObject *sub_cmd_obj = nullptr;
698ebc09c36SJim Ingham                  OptionArgVectorSP option_arg_vector_sp = OptionArgVectorSP (new OptionArgVector);
699ebc09c36SJim Ingham                  OptionArgVector *option_arg_vector = option_arg_vector_sp.get();
700ebc09c36SJim Ingham 
701844d2303SCaroline Tice                  while (cmd_obj->IsMultiwordObject() && args.GetArgumentCount() > 0)
702ebc09c36SJim Ingham                  {
703ebc09c36SJim Ingham                      if (argc >= 3)
704ebc09c36SJim Ingham                      {
705ebc09c36SJim Ingham                          const std::string sub_command = args.GetArgumentAtIndex(0);
706ebc09c36SJim Ingham                          assert (sub_command.length() != 0);
707998255bfSGreg Clayton                          subcommand_obj_sp = cmd_obj->GetSubcommandSP (sub_command.c_str());
7086e3d8e7fSEugene Zelenko                          if (subcommand_obj_sp)
709ebc09c36SJim Ingham                          {
710ebc09c36SJim Ingham                              sub_cmd_obj = subcommand_obj_sp.get();
711ebc09c36SJim Ingham                              use_subcommand = true;
712ebc09c36SJim Ingham                              args.Shift();  // Shift the sub_command word off the argument vector.
713844d2303SCaroline Tice                              cmd_obj = sub_cmd_obj;
714ebc09c36SJim Ingham                          }
715ebc09c36SJim Ingham                          else
716ebc09c36SJim Ingham                          {
717f415eeb4SCaroline Tice                              result.AppendErrorWithFormat("'%s' is not a valid sub-command of '%s'.  "
718f415eeb4SCaroline Tice                                                           "Unable to create alias.\n",
719f415eeb4SCaroline Tice                                                           sub_command.c_str(), actual_command.c_str());
720ebc09c36SJim Ingham                              result.SetStatus (eReturnStatusFailed);
721ebc09c36SJim Ingham                              return false;
722ebc09c36SJim Ingham                          }
723ebc09c36SJim Ingham                      }
724ebc09c36SJim Ingham                  }
725ebc09c36SJim Ingham 
726ebc09c36SJim Ingham                  // Verify & handle any options/arguments passed to the alias command
727ebc09c36SJim Ingham 
728ebc09c36SJim Ingham                  if (args.GetArgumentCount () > 0)
729ebc09c36SJim Ingham                  {
730ca90c47eSCaroline Tice                     CommandObjectSP tmp_sp = m_interpreter.GetCommandSPExact (cmd_obj->GetCommandName(), false);
731ebc09c36SJim Ingham                     if (use_subcommand)
732ca90c47eSCaroline Tice                         tmp_sp = m_interpreter.GetCommandSPExact (sub_cmd_obj->GetCommandName(), false);
733ca90c47eSCaroline Tice 
734ca90c47eSCaroline Tice                     std::string args_string;
735ca90c47eSCaroline Tice                     args.GetCommandString (args_string);
736ca90c47eSCaroline Tice 
737ca90c47eSCaroline Tice                     if (!m_interpreter.ProcessAliasOptionsArgs (tmp_sp, args_string.c_str(), option_arg_vector_sp))
738ebc09c36SJim Ingham                     {
739ca90c47eSCaroline Tice                         result.AppendError ("Unable to create requested alias.\n");
740ca90c47eSCaroline Tice                         result.SetStatus (eReturnStatusFailed);
741e7941795SCaroline Tice                         return false;
742867b185dSCaroline Tice                     }
743867b185dSCaroline Tice                  }
744867b185dSCaroline Tice 
745ebc09c36SJim Ingham                  // Create the alias.
746ebc09c36SJim Ingham 
747a7015092SGreg Clayton                  if (m_interpreter.AliasExists (alias_command.c_str())
748a7015092SGreg Clayton                      || m_interpreter.UserCommandExists (alias_command.c_str()))
749ebc09c36SJim Ingham                  {
750a7015092SGreg Clayton                      OptionArgVectorSP tmp_option_arg_sp (m_interpreter.GetAliasOptions (alias_command.c_str()));
7516e3d8e7fSEugene Zelenko                      if (tmp_option_arg_sp)
752ebc09c36SJim Ingham                      {
7536e3d8e7fSEugene Zelenko                          if (option_arg_vector->empty())
754a7015092SGreg Clayton                              m_interpreter.RemoveAliasOptions (alias_command.c_str());
755ebc09c36SJim Ingham                      }
756ebc09c36SJim Ingham                      result.AppendWarningWithFormat ("Overwriting existing definition for '%s'.\n",
757ebc09c36SJim Ingham                                                      alias_command.c_str());
758ebc09c36SJim Ingham                  }
759ebc09c36SJim Ingham 
760ebc09c36SJim Ingham                  if (use_subcommand)
761a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), subcommand_obj_sp);
762ebc09c36SJim Ingham                  else
763a7015092SGreg Clayton                      m_interpreter.AddAlias (alias_command.c_str(), command_obj_sp);
7646e3d8e7fSEugene Zelenko                  if (!option_arg_vector->empty())
765a7015092SGreg Clayton                      m_interpreter.AddOrReplaceAliasOptions (alias_command.c_str(), option_arg_vector_sp);
766ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusSuccessFinishNoResult);
767ebc09c36SJim Ingham              }
768ebc09c36SJim Ingham              else
769ebc09c36SJim Ingham              {
770ebc09c36SJim Ingham                  result.AppendErrorWithFormat ("'%s' is not an existing command.\n", actual_command.c_str());
771ebc09c36SJim Ingham                  result.SetStatus (eReturnStatusFailed);
772e7941795SCaroline Tice                  return false;
773ebc09c36SJim Ingham              }
774ebc09c36SJim Ingham         }
775ebc09c36SJim Ingham 
776ebc09c36SJim Ingham         return result.Succeeded();
777ebc09c36SJim Ingham     }
778ebc09c36SJim Ingham };
779ebc09c36SJim Ingham 
780ebc09c36SJim Ingham #pragma mark CommandObjectCommandsUnalias
781ebc09c36SJim Ingham //-------------------------------------------------------------------------
782ebc09c36SJim Ingham // CommandObjectCommandsUnalias
783ebc09c36SJim Ingham //-------------------------------------------------------------------------
784ebc09c36SJim Ingham 
7855a988416SJim Ingham class CommandObjectCommandsUnalias : public CommandObjectParsed
786ebc09c36SJim Ingham {
787ebc09c36SJim Ingham public:
788a7015092SGreg Clayton     CommandObjectCommandsUnalias (CommandInterpreter &interpreter) :
7895a988416SJim Ingham         CommandObjectParsed(interpreter,
7900e5e5a79SGreg Clayton                             "command unalias",
79186ddae50SCaroline Tice                             "Allow the user to remove/delete a user-defined command abbreviation.",
7926e3d8e7fSEugene Zelenko                             nullptr)
793ebc09c36SJim Ingham     {
794405fe67fSCaroline Tice         CommandArgumentEntry arg;
795405fe67fSCaroline Tice         CommandArgumentData alias_arg;
796405fe67fSCaroline Tice 
797405fe67fSCaroline Tice         // Define the first (and only) variant of this arg.
798405fe67fSCaroline Tice         alias_arg.arg_type = eArgTypeAliasName;
799405fe67fSCaroline Tice         alias_arg.arg_repetition = eArgRepeatPlain;
800405fe67fSCaroline Tice 
801405fe67fSCaroline Tice         // There is only one variant this argument could be; put it into the argument entry.
802405fe67fSCaroline Tice         arg.push_back (alias_arg);
803405fe67fSCaroline Tice 
804405fe67fSCaroline Tice         // Push the data for the first argument into the m_arguments vector.
805405fe67fSCaroline Tice         m_arguments.push_back (arg);
806ebc09c36SJim Ingham     }
807ebc09c36SJim Ingham 
8086e3d8e7fSEugene Zelenko     ~CommandObjectCommandsUnalias() override = default;
809ebc09c36SJim Ingham 
8105a988416SJim Ingham protected:
811ebc09c36SJim Ingham     bool
81213d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
813ebc09c36SJim Ingham     {
814ebc09c36SJim Ingham         CommandObject::CommandMap::iterator pos;
815ebc09c36SJim Ingham         CommandObject *cmd_obj;
816ebc09c36SJim Ingham 
817ebc09c36SJim Ingham         if (args.GetArgumentCount() != 0)
818ebc09c36SJim Ingham         {
819ebc09c36SJim Ingham             const char *command_name = args.GetArgumentAtIndex(0);
820a7015092SGreg Clayton             cmd_obj = m_interpreter.GetCommandObject(command_name);
821ebc09c36SJim Ingham             if (cmd_obj)
822ebc09c36SJim Ingham             {
823a7015092SGreg Clayton                 if (m_interpreter.CommandExists (command_name))
824ebc09c36SJim Ingham                 {
825b547278cSGreg Clayton                     if (cmd_obj->IsRemovable())
826b547278cSGreg Clayton                     {
827b547278cSGreg Clayton                         result.AppendErrorWithFormat ("'%s' is not an alias, it is a debugger command which can be removed using the 'command delete' command.\n",
828b547278cSGreg Clayton                                                       command_name);
829b547278cSGreg Clayton                     }
830b547278cSGreg Clayton                     else
831b547278cSGreg Clayton                     {
832ebc09c36SJim Ingham                         result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
833ebc09c36SJim Ingham                                                       command_name);
834b547278cSGreg Clayton                     }
835ebc09c36SJim Ingham                     result.SetStatus (eReturnStatusFailed);
836ebc09c36SJim Ingham                 }
837ebc09c36SJim Ingham                 else
838ebc09c36SJim Ingham                 {
8396e3d8e7fSEugene Zelenko                     if (!m_interpreter.RemoveAlias(command_name))
840ebc09c36SJim Ingham                     {
841a7015092SGreg Clayton                         if (m_interpreter.AliasExists (command_name))
842ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("Error occurred while attempting to unalias '%s'.\n",
843ebc09c36SJim Ingham                                                           command_name);
844ebc09c36SJim Ingham                         else
845ebc09c36SJim Ingham                             result.AppendErrorWithFormat ("'%s' is not an existing alias.\n", command_name);
846ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusFailed);
847ebc09c36SJim Ingham                     }
848ebc09c36SJim Ingham                     else
849ebc09c36SJim Ingham                         result.SetStatus (eReturnStatusSuccessFinishNoResult);
850ebc09c36SJim Ingham                 }
851ebc09c36SJim Ingham             }
852ebc09c36SJim Ingham             else
853ebc09c36SJim Ingham             {
854ebc09c36SJim Ingham                 result.AppendErrorWithFormat ("'%s' is not a known command.\nTry 'help' to see a "
855ebc09c36SJim Ingham                                               "current list of commands.\n",
856ebc09c36SJim Ingham                                              command_name);
857ebc09c36SJim Ingham                 result.SetStatus (eReturnStatusFailed);
858ebc09c36SJim Ingham             }
859ebc09c36SJim Ingham         }
860ebc09c36SJim Ingham         else
861ebc09c36SJim Ingham         {
862ebc09c36SJim Ingham             result.AppendError ("must call 'unalias' with a valid alias");
863ebc09c36SJim Ingham             result.SetStatus (eReturnStatusFailed);
864ebc09c36SJim Ingham         }
865ebc09c36SJim Ingham 
866ebc09c36SJim Ingham         return result.Succeeded();
867ebc09c36SJim Ingham     }
868ebc09c36SJim Ingham };
869ebc09c36SJim Ingham 
870b547278cSGreg Clayton #pragma mark CommandObjectCommandsDelete
871b547278cSGreg Clayton //-------------------------------------------------------------------------
872b547278cSGreg Clayton // CommandObjectCommandsDelete
873b547278cSGreg Clayton //-------------------------------------------------------------------------
874b547278cSGreg Clayton 
875b547278cSGreg Clayton class CommandObjectCommandsDelete : public CommandObjectParsed
876b547278cSGreg Clayton {
877b547278cSGreg Clayton public:
878b547278cSGreg Clayton     CommandObjectCommandsDelete (CommandInterpreter &interpreter) :
879b547278cSGreg Clayton         CommandObjectParsed(interpreter,
880b547278cSGreg Clayton                             "command delete",
881b547278cSGreg Clayton                             "Allow the user to delete user-defined regular expression, python or multi-word commands.",
8826e3d8e7fSEugene Zelenko                             nullptr)
883b547278cSGreg Clayton     {
884b547278cSGreg Clayton         CommandArgumentEntry arg;
885b547278cSGreg Clayton         CommandArgumentData alias_arg;
886b547278cSGreg Clayton 
887b547278cSGreg Clayton         // Define the first (and only) variant of this arg.
888b547278cSGreg Clayton         alias_arg.arg_type = eArgTypeCommandName;
889b547278cSGreg Clayton         alias_arg.arg_repetition = eArgRepeatPlain;
890b547278cSGreg Clayton 
891b547278cSGreg Clayton         // There is only one variant this argument could be; put it into the argument entry.
892b547278cSGreg Clayton         arg.push_back (alias_arg);
893b547278cSGreg Clayton 
894b547278cSGreg Clayton         // Push the data for the first argument into the m_arguments vector.
895b547278cSGreg Clayton         m_arguments.push_back (arg);
896b547278cSGreg Clayton     }
897b547278cSGreg Clayton 
8986e3d8e7fSEugene Zelenko     ~CommandObjectCommandsDelete() override = default;
899b547278cSGreg Clayton 
900b547278cSGreg Clayton protected:
901b547278cSGreg Clayton     bool
90213d21e9aSBruce Mitchener     DoExecute (Args& args, CommandReturnObject &result) override
903b547278cSGreg Clayton     {
904b547278cSGreg Clayton         CommandObject::CommandMap::iterator pos;
905b547278cSGreg Clayton 
906b547278cSGreg Clayton         if (args.GetArgumentCount() != 0)
907b547278cSGreg Clayton         {
908b547278cSGreg Clayton             const char *command_name = args.GetArgumentAtIndex(0);
909b547278cSGreg Clayton             if (m_interpreter.CommandExists (command_name))
910b547278cSGreg Clayton             {
911b547278cSGreg Clayton                 if (m_interpreter.RemoveCommand (command_name))
912b547278cSGreg Clayton                 {
913b547278cSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
914b547278cSGreg Clayton                 }
915b547278cSGreg Clayton                 else
916b547278cSGreg Clayton                 {
917b547278cSGreg Clayton                     result.AppendErrorWithFormat ("'%s' is a permanent debugger command and cannot be removed.\n",
918b547278cSGreg Clayton                                                   command_name);
919b547278cSGreg Clayton                     result.SetStatus (eReturnStatusFailed);
920b547278cSGreg Clayton                 }
921b547278cSGreg Clayton             }
922b547278cSGreg Clayton             else
923b547278cSGreg Clayton             {
924*46d4aa21SEnrico Granata                 StreamString error_msg_stream;
925*46d4aa21SEnrico Granata                 const bool generate_apropos = true;
926*46d4aa21SEnrico Granata                 const bool generate_type_lookup = false;
927*46d4aa21SEnrico Granata                 CommandObjectHelp::GenerateAdditionalHelpAvenuesMessage(&error_msg_stream,
928*46d4aa21SEnrico Granata                                                                         command_name,
929*46d4aa21SEnrico Granata                                                                         nullptr,
930*46d4aa21SEnrico Granata                                                                         nullptr,
931*46d4aa21SEnrico Granata                                                                         generate_apropos,
932*46d4aa21SEnrico Granata                                                                         generate_type_lookup);
933*46d4aa21SEnrico Granata                 result.AppendErrorWithFormat ("%s", error_msg_stream.GetData());
934b547278cSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
935b547278cSGreg Clayton             }
936b547278cSGreg Clayton         }
937b547278cSGreg Clayton         else
938b547278cSGreg Clayton         {
939b547278cSGreg Clayton             result.AppendErrorWithFormat ("must call '%s' with one or more valid user defined regular expression, python or multi-word command names", GetCommandName ());
940b547278cSGreg Clayton             result.SetStatus (eReturnStatusFailed);
941b547278cSGreg Clayton         }
942b547278cSGreg Clayton 
943b547278cSGreg Clayton         return result.Succeeded();
944b547278cSGreg Clayton     }
945b547278cSGreg Clayton };
946b547278cSGreg Clayton 
947de164aaaSGreg Clayton //-------------------------------------------------------------------------
948de164aaaSGreg Clayton // CommandObjectCommandsAddRegex
949de164aaaSGreg Clayton //-------------------------------------------------------------------------
9505a988416SJim Ingham #pragma mark CommandObjectCommandsAddRegex
951de164aaaSGreg Clayton 
95244d93782SGreg Clayton class CommandObjectCommandsAddRegex :
95344d93782SGreg Clayton     public CommandObjectParsed,
954ea508635SGreg Clayton     public IOHandlerDelegateMultiline
955de164aaaSGreg Clayton {
956de164aaaSGreg Clayton public:
957de164aaaSGreg Clayton     CommandObjectCommandsAddRegex (CommandInterpreter &interpreter) :
9585a988416SJim Ingham         CommandObjectParsed (interpreter,
9590e5e5a79SGreg Clayton                        "command regex",
960de164aaaSGreg Clayton                        "Allow the user to create a regular expression command.",
9610e5e5a79SGreg Clayton                        "command regex <cmd-name> [s/<regex>/<subst>/ ...]"),
962ea508635SGreg Clayton         IOHandlerDelegateMultiline ("", IOHandlerDelegate::Completion::LLDBCommand),
963de164aaaSGreg Clayton         m_options (interpreter)
964de164aaaSGreg Clayton     {
965ea671fbdSKate Stone         SetHelpLong(R"(
966ea671fbdSKate Stone )" "This command allows the user to create powerful regular expression commands \
967ea671fbdSKate Stone with substitutions. The regular expressions and substitutions are specified \
968ea671fbdSKate Stone using the regular expression substitution format of:" R"(
969ea671fbdSKate Stone 
970ea671fbdSKate Stone     s/<regex>/<subst>/
971ea671fbdSKate Stone 
972ea671fbdSKate Stone )" "<regex> is a regular expression that can use parenthesis to capture regular \
973ea671fbdSKate Stone expression input and substitute the captured matches in the output using %1 \
974ea671fbdSKate Stone for the first match, %2 for the second, and so on." R"(
975ea671fbdSKate Stone 
976ea671fbdSKate Stone )" "The regular expressions can all be specified on the command line if more than \
977ea671fbdSKate Stone one argument is provided. If just the command name is provided on the command \
978ea671fbdSKate Stone line, then the regular expressions and substitutions can be entered on separate \
979ea671fbdSKate Stone lines, followed by an empty line to terminate the command definition." R"(
980ea671fbdSKate Stone 
981ea671fbdSKate Stone EXAMPLES
982ea671fbdSKate Stone 
983ea671fbdSKate Stone )" "The following example will define a regular expression command named 'f' that \
984ea671fbdSKate Stone will call 'finish' if there are no arguments, or 'frame select <frame-idx>' if \
985ea671fbdSKate Stone a number follows 'f':" R"(
986ea671fbdSKate Stone 
987ea671fbdSKate Stone     (lldb) command regex f s/^$/finish/ 's/([0-9]+)/frame select %1/')"
9880e5e5a79SGreg Clayton         );
989de164aaaSGreg Clayton     }
990de164aaaSGreg Clayton 
9916e3d8e7fSEugene Zelenko     ~CommandObjectCommandsAddRegex() override = default;
992de164aaaSGreg Clayton 
9935a988416SJim Ingham protected:
994ea508635SGreg Clayton     void
995ea508635SGreg Clayton     IOHandlerActivated (IOHandler &io_handler) override
99644d93782SGreg Clayton     {
99744d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
99844d93782SGreg Clayton         if (output_sp)
99944d93782SGreg Clayton         {
100044d93782SGreg Clayton             output_sp->PutCString("Enter one of more sed substitution commands in the form: 's/<regex>/<subst>/'.\nTerminate the substitution list with an empty line.\n");
100144d93782SGreg Clayton             output_sp->Flush();
100244d93782SGreg Clayton         }
100344d93782SGreg Clayton     }
100444d93782SGreg Clayton 
1005ea508635SGreg Clayton     void
1006ea508635SGreg Clayton     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
100744d93782SGreg Clayton     {
100844d93782SGreg Clayton         io_handler.SetIsDone(true);
10096e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
101044d93782SGreg Clayton         {
101144d93782SGreg Clayton             StringList lines;
101244d93782SGreg Clayton             if (lines.SplitIntoLines (data))
101344d93782SGreg Clayton             {
101444d93782SGreg Clayton                 const size_t num_lines = lines.GetSize();
101544d93782SGreg Clayton                 bool check_only = false;
101644d93782SGreg Clayton                 for (size_t i=0; i<num_lines; ++i)
101744d93782SGreg Clayton                 {
101844d93782SGreg Clayton                     llvm::StringRef bytes_strref (lines[i]);
101944d93782SGreg Clayton                     Error error = AppendRegexSubstitution (bytes_strref, check_only);
102044d93782SGreg Clayton                     if (error.Fail())
102144d93782SGreg Clayton                     {
102244d93782SGreg Clayton                         if (!m_interpreter.GetDebugger().GetCommandInterpreter().GetBatchCommandMode())
102344d93782SGreg Clayton                         {
102444d93782SGreg Clayton                             StreamSP out_stream = m_interpreter.GetDebugger().GetAsyncOutputStream();
102544d93782SGreg Clayton                             out_stream->Printf("error: %s\n", error.AsCString());
102644d93782SGreg Clayton                         }
102744d93782SGreg Clayton                     }
102844d93782SGreg Clayton                 }
102944d93782SGreg Clayton             }
103044d93782SGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
103144d93782SGreg Clayton             {
103244d93782SGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
103344d93782SGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
103444d93782SGreg Clayton             }
103544d93782SGreg Clayton         }
103644d93782SGreg Clayton     }
103744d93782SGreg Clayton 
1038de164aaaSGreg Clayton     bool
1039b0a1814fSEric Christopher     DoExecute (Args& command, CommandReturnObject &result) override
1040de164aaaSGreg Clayton     {
10415a988416SJim Ingham         const size_t argc = command.GetArgumentCount();
10420e5e5a79SGreg Clayton         if (argc == 0)
1043de164aaaSGreg Clayton         {
104469c12ccbSJason Molenda             result.AppendError ("usage: 'command regex <command-name> [s/<regex1>/<subst1>/ s/<regex2>/<subst2>/ ...]'\n");
10450e5e5a79SGreg Clayton             result.SetStatus (eReturnStatusFailed);
10460e5e5a79SGreg Clayton         }
10470e5e5a79SGreg Clayton         else
10480e5e5a79SGreg Clayton         {
10490e5e5a79SGreg Clayton             Error error;
10505a988416SJim Ingham             const char *name = command.GetArgumentAtIndex(0);
1051de164aaaSGreg Clayton             m_regex_cmd_ap.reset (new CommandObjectRegexCommand (m_interpreter,
1052de164aaaSGreg Clayton                                                                  name,
1053de164aaaSGreg Clayton                                                                  m_options.GetHelp (),
1054de164aaaSGreg Clayton                                                                  m_options.GetSyntax (),
1055b547278cSGreg Clayton                                                                  10,
1056b547278cSGreg Clayton                                                                  0,
1057b547278cSGreg Clayton                                                                  true));
10580e5e5a79SGreg Clayton 
10590e5e5a79SGreg Clayton             if (argc == 1)
10600e5e5a79SGreg Clayton             {
106144d93782SGreg Clayton                 Debugger &debugger = m_interpreter.GetDebugger();
1062e30f11d9SKate Stone                 bool color_prompt = debugger.GetUseColor();
106344d93782SGreg Clayton                 const bool multiple_lines = true; // Get multiple lines
106444d93782SGreg Clayton                 IOHandlerSP io_handler_sp(new IOHandlerEditline(debugger,
1065e30f11d9SKate Stone                                                                 IOHandler::Type::Other,
106673d80faaSGreg Clayton                                                                 "lldb-regex", // Name of input reader for history
1067ea508635SGreg Clayton                                                                 "> ",         // Prompt
10686e3d8e7fSEugene Zelenko                                                                 nullptr,      // Continuation prompt
106944d93782SGreg Clayton                                                                 multiple_lines,
1070e30f11d9SKate Stone                                                                 color_prompt,
1071f6913cd7SGreg Clayton                                                                 0,            // Don't show line numbers
107244d93782SGreg Clayton                                                                 *this));
107344d93782SGreg Clayton 
107444d93782SGreg Clayton                 if (io_handler_sp)
1075de164aaaSGreg Clayton                 {
107644d93782SGreg Clayton                     debugger.PushIOHandler(io_handler_sp);
1077de164aaaSGreg Clayton                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1078de164aaaSGreg Clayton                 }
1079de164aaaSGreg Clayton             }
1080de164aaaSGreg Clayton             else
1081de164aaaSGreg Clayton             {
10820e5e5a79SGreg Clayton                 for (size_t arg_idx = 1; arg_idx < argc; ++arg_idx)
10830e5e5a79SGreg Clayton                 {
10845a988416SJim Ingham                     llvm::StringRef arg_strref (command.GetArgumentAtIndex(arg_idx));
108544d93782SGreg Clayton                     bool check_only = false;
108644d93782SGreg Clayton                     error = AppendRegexSubstitution (arg_strref, check_only);
10870e5e5a79SGreg Clayton                     if (error.Fail())
10880e5e5a79SGreg Clayton                         break;
10890e5e5a79SGreg Clayton                 }
10900e5e5a79SGreg Clayton 
10910e5e5a79SGreg Clayton                 if (error.Success())
10920e5e5a79SGreg Clayton                 {
10930e5e5a79SGreg Clayton                     AddRegexCommandToInterpreter();
10940e5e5a79SGreg Clayton                 }
10950e5e5a79SGreg Clayton             }
10960e5e5a79SGreg Clayton             if (error.Fail())
10970e5e5a79SGreg Clayton             {
10980e5e5a79SGreg Clayton                 result.AppendError (error.AsCString());
1099de164aaaSGreg Clayton                 result.SetStatus (eReturnStatusFailed);
1100de164aaaSGreg Clayton             }
11010e5e5a79SGreg Clayton         }
11020e5e5a79SGreg Clayton 
1103de164aaaSGreg Clayton         return result.Succeeded();
1104de164aaaSGreg Clayton     }
1105de164aaaSGreg Clayton 
11060e5e5a79SGreg Clayton     Error
110744d93782SGreg Clayton     AppendRegexSubstitution (const llvm::StringRef &regex_sed, bool check_only)
1108de164aaaSGreg Clayton     {
11090e5e5a79SGreg Clayton         Error error;
11100e5e5a79SGreg Clayton 
11116e3d8e7fSEugene Zelenko         if (!m_regex_cmd_ap)
1112de164aaaSGreg Clayton         {
11130e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("invalid regular expression command object for: '%.*s'",
11140e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11150e5e5a79SGreg Clayton                                            regex_sed.data());
11160e5e5a79SGreg Clayton             return error;
1117de164aaaSGreg Clayton         }
11180e5e5a79SGreg Clayton 
11190e5e5a79SGreg Clayton         size_t regex_sed_size = regex_sed.size();
11200e5e5a79SGreg Clayton 
11210e5e5a79SGreg Clayton         if (regex_sed_size <= 1)
11220e5e5a79SGreg Clayton         {
11230e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string is too short: '%.*s'",
11240e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11250e5e5a79SGreg Clayton                                            regex_sed.data());
11260e5e5a79SGreg Clayton             return error;
11270e5e5a79SGreg Clayton         }
11280e5e5a79SGreg Clayton 
11290e5e5a79SGreg Clayton         if (regex_sed[0] != 's')
11300e5e5a79SGreg Clayton         {
11310e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("regular expression substitution string doesn't start with 's': '%.*s'",
11320e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11330e5e5a79SGreg Clayton                                            regex_sed.data());
11340e5e5a79SGreg Clayton             return error;
11350e5e5a79SGreg Clayton         }
11360e5e5a79SGreg Clayton         const size_t first_separator_char_pos = 1;
11370e5e5a79SGreg Clayton         // use the char that follows 's' as the regex separator character
11380e5e5a79SGreg Clayton         // so we can have "s/<regex>/<subst>/" or "s|<regex>|<subst>|"
11390e5e5a79SGreg Clayton         const char separator_char = regex_sed[first_separator_char_pos];
11400e5e5a79SGreg Clayton         const size_t second_separator_char_pos = regex_sed.find (separator_char, first_separator_char_pos + 1);
11410e5e5a79SGreg Clayton 
11420e5e5a79SGreg Clayton         if (second_separator_char_pos == std::string::npos)
11430e5e5a79SGreg Clayton         {
1144ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing second '%c' separator char after '%.*s' in '%.*s'",
11450e5e5a79SGreg Clayton                                            separator_char,
11460e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - first_separator_char_pos - 1),
1147ea508635SGreg Clayton                                            regex_sed.data() + (first_separator_char_pos + 1),
1148ea508635SGreg Clayton                                            (int)regex_sed.size(),
1149ea508635SGreg Clayton                                            regex_sed.data());
11500e5e5a79SGreg Clayton             return error;
11510e5e5a79SGreg Clayton         }
11520e5e5a79SGreg Clayton 
11530e5e5a79SGreg Clayton         const size_t third_separator_char_pos = regex_sed.find (separator_char, second_separator_char_pos + 1);
11540e5e5a79SGreg Clayton 
11550e5e5a79SGreg Clayton         if (third_separator_char_pos == std::string::npos)
11560e5e5a79SGreg Clayton         {
1157ea508635SGreg Clayton             error.SetErrorStringWithFormat("missing third '%c' separator char after '%.*s' in '%.*s'",
11580e5e5a79SGreg Clayton                                            separator_char,
11590e5e5a79SGreg Clayton                                            (int)(regex_sed.size() - second_separator_char_pos - 1),
1160ea508635SGreg Clayton                                            regex_sed.data() + (second_separator_char_pos + 1),
1161ea508635SGreg Clayton                                            (int)regex_sed.size(),
1162ea508635SGreg Clayton                                            regex_sed.data());
11630e5e5a79SGreg Clayton             return error;
11640e5e5a79SGreg Clayton         }
11650e5e5a79SGreg Clayton 
11660e5e5a79SGreg Clayton         if (third_separator_char_pos != regex_sed_size - 1)
11670e5e5a79SGreg Clayton         {
11680e5e5a79SGreg Clayton             // Make sure that everything that follows the last regex
11690e5e5a79SGreg Clayton             // separator char
11700e5e5a79SGreg Clayton             if (regex_sed.find_first_not_of("\t\n\v\f\r ", third_separator_char_pos + 1) != std::string::npos)
11710e5e5a79SGreg Clayton             {
11720e5e5a79SGreg Clayton                 error.SetErrorStringWithFormat("extra data found after the '%.*s' regular expression substitution string: '%.*s'",
11730e5e5a79SGreg Clayton                                                (int)third_separator_char_pos + 1,
11740e5e5a79SGreg Clayton                                                regex_sed.data(),
11750e5e5a79SGreg Clayton                                                (int)(regex_sed.size() - third_separator_char_pos - 1),
11760e5e5a79SGreg Clayton                                                regex_sed.data() + (third_separator_char_pos + 1));
11770e5e5a79SGreg Clayton                 return error;
11780e5e5a79SGreg Clayton             }
11790e5e5a79SGreg Clayton         }
11800e5e5a79SGreg Clayton         else if (first_separator_char_pos + 1 == second_separator_char_pos)
11810e5e5a79SGreg Clayton         {
11820e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<regex> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11830e5e5a79SGreg Clayton                                            separator_char,
11840e5e5a79SGreg Clayton                                            separator_char,
11850e5e5a79SGreg Clayton                                            separator_char,
11860e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11870e5e5a79SGreg Clayton                                            regex_sed.data());
11880e5e5a79SGreg Clayton             return error;
11890e5e5a79SGreg Clayton         }
11900e5e5a79SGreg Clayton         else if (second_separator_char_pos + 1 == third_separator_char_pos)
11910e5e5a79SGreg Clayton         {
11920e5e5a79SGreg Clayton             error.SetErrorStringWithFormat("<subst> can't be empty in 's%c<regex>%c<subst>%c' string: '%.*s'",
11930e5e5a79SGreg Clayton                                            separator_char,
11940e5e5a79SGreg Clayton                                            separator_char,
11950e5e5a79SGreg Clayton                                            separator_char,
11960e5e5a79SGreg Clayton                                            (int)regex_sed.size(),
11970e5e5a79SGreg Clayton                                            regex_sed.data());
11980e5e5a79SGreg Clayton             return error;
11990e5e5a79SGreg Clayton         }
120044d93782SGreg Clayton 
12016e3d8e7fSEugene Zelenko         if (!check_only)
120244d93782SGreg Clayton         {
12030e5e5a79SGreg Clayton             std::string regex(regex_sed.substr(first_separator_char_pos + 1, second_separator_char_pos - first_separator_char_pos - 1));
12040e5e5a79SGreg Clayton             std::string subst(regex_sed.substr(second_separator_char_pos + 1, third_separator_char_pos - second_separator_char_pos - 1));
12050e5e5a79SGreg Clayton             m_regex_cmd_ap->AddRegexCommand (regex.c_str(),
12060e5e5a79SGreg Clayton                                              subst.c_str());
120744d93782SGreg Clayton         }
12080e5e5a79SGreg Clayton         return error;
1209de164aaaSGreg Clayton     }
1210de164aaaSGreg Clayton 
1211de164aaaSGreg Clayton     void
12120e5e5a79SGreg Clayton     AddRegexCommandToInterpreter()
1213de164aaaSGreg Clayton     {
12146e3d8e7fSEugene Zelenko         if (m_regex_cmd_ap)
1215de164aaaSGreg Clayton         {
1216de164aaaSGreg Clayton             if (m_regex_cmd_ap->HasRegexEntries())
1217de164aaaSGreg Clayton             {
1218de164aaaSGreg Clayton                 CommandObjectSP cmd_sp (m_regex_cmd_ap.release());
1219de164aaaSGreg Clayton                 m_interpreter.AddCommand(cmd_sp->GetCommandName(), cmd_sp, true);
1220de164aaaSGreg Clayton             }
1221de164aaaSGreg Clayton         }
1222de164aaaSGreg Clayton     }
1223de164aaaSGreg Clayton 
1224de164aaaSGreg Clayton private:
12257b0992d9SGreg Clayton     std::unique_ptr<CommandObjectRegexCommand> m_regex_cmd_ap;
1226de164aaaSGreg Clayton 
1227de164aaaSGreg Clayton      class CommandOptions : public Options
1228de164aaaSGreg Clayton      {
1229de164aaaSGreg Clayton      public:
1230de164aaaSGreg Clayton          CommandOptions (CommandInterpreter &interpreter) :
1231de164aaaSGreg Clayton             Options (interpreter)
1232de164aaaSGreg Clayton          {
1233de164aaaSGreg Clayton          }
1234de164aaaSGreg Clayton 
12356e3d8e7fSEugene Zelenko          ~CommandOptions() override = default;
1236de164aaaSGreg Clayton 
123713d21e9aSBruce Mitchener          Error
123813d21e9aSBruce Mitchener          SetOptionValue (uint32_t option_idx, const char *option_arg) override
1239de164aaaSGreg Clayton          {
1240de164aaaSGreg Clayton              Error error;
12413bcdfc0eSGreg Clayton              const int short_option = m_getopt_table[option_idx].val;
1242de164aaaSGreg Clayton 
1243de164aaaSGreg Clayton              switch (short_option)
1244de164aaaSGreg Clayton              {
1245de164aaaSGreg Clayton                  case 'h':
1246de164aaaSGreg Clayton                      m_help.assign (option_arg);
1247de164aaaSGreg Clayton                      break;
1248de164aaaSGreg Clayton                  case 's':
1249de164aaaSGreg Clayton                      m_syntax.assign (option_arg);
1250de164aaaSGreg Clayton                      break;
1251de164aaaSGreg Clayton                  default:
125286edbf41SGreg Clayton                      error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1253de164aaaSGreg Clayton                      break;
1254de164aaaSGreg Clayton              }
1255de164aaaSGreg Clayton 
1256de164aaaSGreg Clayton              return error;
1257de164aaaSGreg Clayton          }
1258de164aaaSGreg Clayton 
1259de164aaaSGreg Clayton          void
126013d21e9aSBruce Mitchener          OptionParsingStarting () override
1261de164aaaSGreg Clayton          {
1262de164aaaSGreg Clayton              m_help.clear();
1263de164aaaSGreg Clayton              m_syntax.clear();
1264de164aaaSGreg Clayton          }
1265de164aaaSGreg Clayton 
1266de164aaaSGreg Clayton          const OptionDefinition*
126713d21e9aSBruce Mitchener          GetDefinitions () override
1268de164aaaSGreg Clayton          {
1269de164aaaSGreg Clayton              return g_option_table;
1270de164aaaSGreg Clayton          }
1271de164aaaSGreg Clayton 
1272de164aaaSGreg Clayton          // Options table: Required for subclasses of Options.
1273de164aaaSGreg Clayton 
1274de164aaaSGreg Clayton          static OptionDefinition g_option_table[];
1275de164aaaSGreg Clayton 
1276de164aaaSGreg Clayton          const char *
1277de164aaaSGreg Clayton          GetHelp()
1278de164aaaSGreg Clayton          {
12796e3d8e7fSEugene Zelenko              return (m_help.empty() ? nullptr : m_help.c_str());
1280de164aaaSGreg Clayton          }
12816e3d8e7fSEugene Zelenko 
1282de164aaaSGreg Clayton          const char *
1283de164aaaSGreg Clayton          GetSyntax ()
1284de164aaaSGreg Clayton          {
12856e3d8e7fSEugene Zelenko              return (m_syntax.empty() ? nullptr : m_syntax.c_str());
1286de164aaaSGreg Clayton          }
12876e3d8e7fSEugene Zelenko 
1288de164aaaSGreg Clayton      protected:
12896e3d8e7fSEugene Zelenko          // Instance variables to hold the values for command options.
12906e3d8e7fSEugene Zelenko 
1291de164aaaSGreg Clayton          std::string m_help;
1292de164aaaSGreg Clayton          std::string m_syntax;
1293de164aaaSGreg Clayton      };
1294de164aaaSGreg Clayton 
1295b0a1814fSEric Christopher      Options *
1296b0a1814fSEric Christopher      GetOptions () override
1297de164aaaSGreg Clayton      {
1298de164aaaSGreg Clayton          return &m_options;
1299de164aaaSGreg Clayton      }
1300de164aaaSGreg Clayton 
13015a988416SJim Ingham      CommandOptions m_options;
1302de164aaaSGreg Clayton };
1303de164aaaSGreg Clayton 
1304de164aaaSGreg Clayton OptionDefinition
1305de164aaaSGreg Clayton CommandObjectCommandsAddRegex::CommandOptions::g_option_table[] =
1306de164aaaSGreg Clayton {
13076e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "The help text to display for this command."},
13086e3d8e7fSEugene Zelenko { LLDB_OPT_SET_1, false, "syntax", 's', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeNone, "A syntax string showing the typical usage syntax."},
13096e3d8e7fSEugene Zelenko { 0             , false,  nullptr   , 0  , 0                , nullptr, nullptr, 0, eArgTypeNone, nullptr }
1310de164aaaSGreg Clayton };
1311de164aaaSGreg Clayton 
13125a988416SJim Ingham class CommandObjectPythonFunction : public CommandObjectRaw
1313223383edSEnrico Granata {
1314223383edSEnrico Granata public:
1315223383edSEnrico Granata     CommandObjectPythonFunction (CommandInterpreter &interpreter,
1316223383edSEnrico Granata                                  std::string name,
13170a305db7SEnrico Granata                                  std::string funct,
1318735152e3SEnrico Granata                                  std::string help,
13190a305db7SEnrico Granata                                  ScriptedCommandSynchronicity synch) :
13205a988416SJim Ingham         CommandObjectRaw(interpreter,
1321223383edSEnrico Granata                          name.c_str(),
13226e3d8e7fSEugene Zelenko                          nullptr,
13236e3d8e7fSEugene Zelenko                          nullptr),
13240a305db7SEnrico Granata         m_function_name(funct),
1325fac939e9SEnrico Granata         m_synchro(synch),
1326fac939e9SEnrico Granata         m_fetched_help_long(false)
1327223383edSEnrico Granata     {
1328735152e3SEnrico Granata         if (!help.empty())
1329735152e3SEnrico Granata             SetHelp(help.c_str());
1330735152e3SEnrico Granata         else
1331735152e3SEnrico Granata         {
1332735152e3SEnrico Granata             StreamString stream;
1333735152e3SEnrico Granata             stream.Printf("For more information run 'help %s'",name.c_str());
1334735152e3SEnrico Granata             SetHelp(stream.GetData());
1335735152e3SEnrico Granata         }
1336223383edSEnrico Granata     }
1337223383edSEnrico Granata 
13386e3d8e7fSEugene Zelenko     ~CommandObjectPythonFunction() override = default;
1339223383edSEnrico Granata 
134013d21e9aSBruce Mitchener     bool
134113d21e9aSBruce Mitchener     IsRemovable () const override
13425a988416SJim Ingham     {
13435a988416SJim Ingham         return true;
13445a988416SJim Ingham     }
13455a988416SJim Ingham 
13465a988416SJim Ingham     const std::string&
13475a988416SJim Ingham     GetFunctionName ()
13485a988416SJim Ingham     {
13495a988416SJim Ingham         return m_function_name;
13505a988416SJim Ingham     }
13515a988416SJim Ingham 
13525a988416SJim Ingham     ScriptedCommandSynchronicity
13535a988416SJim Ingham     GetSynchronicity ()
13545a988416SJim Ingham     {
13555a988416SJim Ingham         return m_synchro;
13565a988416SJim Ingham     }
13575a988416SJim Ingham 
135813d21e9aSBruce Mitchener     const char *
135913d21e9aSBruce Mitchener     GetHelpLong () override
1360fac939e9SEnrico Granata     {
1361fac939e9SEnrico Granata         if (!m_fetched_help_long)
1362fac939e9SEnrico Granata         {
1363fac939e9SEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1364fac939e9SEnrico Granata             if (scripter)
1365fac939e9SEnrico Granata             {
1366fac939e9SEnrico Granata                 std::string docstring;
1367fac939e9SEnrico Granata                 m_fetched_help_long = scripter->GetDocumentationForItem(m_function_name.c_str(),docstring);
1368fac939e9SEnrico Granata                 if (!docstring.empty())
1369fac939e9SEnrico Granata                     SetHelpLong(docstring);
1370fac939e9SEnrico Granata             }
1371fac939e9SEnrico Granata         }
1372fac939e9SEnrico Granata         return CommandObjectRaw::GetHelpLong();
1373fac939e9SEnrico Granata     }
1374fac939e9SEnrico Granata 
13755a988416SJim Ingham protected:
137613d21e9aSBruce Mitchener     bool
137713d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
1378223383edSEnrico Granata     {
1379223383edSEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
1380223383edSEnrico Granata 
1381223383edSEnrico Granata         Error error;
1382223383edSEnrico Granata 
138370f11f88SJim Ingham         result.SetStatus(eReturnStatusInvalid);
138470f11f88SJim Ingham 
13856e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_function_name.c_str(),
1386223383edSEnrico Granata                                                           raw_command_line,
13870a305db7SEnrico Granata                                                           m_synchro,
1388223383edSEnrico Granata                                                           result,
138906be059aSEnrico Granata                                                           error,
13906e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
1391223383edSEnrico Granata         {
1392223383edSEnrico Granata             result.AppendError(error.AsCString());
1393223383edSEnrico Granata             result.SetStatus(eReturnStatusFailed);
1394223383edSEnrico Granata         }
1395223383edSEnrico Granata         else
139670f11f88SJim Ingham         {
139770f11f88SJim Ingham             // Don't change the status if the command already set it...
139870f11f88SJim Ingham             if (result.GetStatus() == eReturnStatusInvalid)
139970f11f88SJim Ingham             {
14006e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
1401223383edSEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
140270f11f88SJim Ingham                 else
140370f11f88SJim Ingham                     result.SetStatus(eReturnStatusSuccessFinishResult);
140470f11f88SJim Ingham             }
140570f11f88SJim Ingham         }
1406223383edSEnrico Granata 
1407223383edSEnrico Granata         return result.Succeeded();
1408223383edSEnrico Granata     }
1409223383edSEnrico Granata 
14106e3d8e7fSEugene Zelenko private:
14116e3d8e7fSEugene Zelenko     std::string m_function_name;
14126e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
14136e3d8e7fSEugene Zelenko     bool m_fetched_help_long;
1414223383edSEnrico Granata };
1415223383edSEnrico Granata 
14169fe00e52SEnrico Granata class CommandObjectScriptingObject : public CommandObjectRaw
14179fe00e52SEnrico Granata {
14189fe00e52SEnrico Granata public:
14199fe00e52SEnrico Granata     CommandObjectScriptingObject (CommandInterpreter &interpreter,
14209fe00e52SEnrico Granata                                   std::string name,
14210641ca1aSZachary Turner                                   StructuredData::GenericSP cmd_obj_sp,
14229fe00e52SEnrico Granata                                   ScriptedCommandSynchronicity synch) :
14239fe00e52SEnrico Granata         CommandObjectRaw(interpreter,
14249fe00e52SEnrico Granata                          name.c_str(),
14256e3d8e7fSEugene Zelenko                          nullptr,
14266e3d8e7fSEugene Zelenko                          nullptr),
14279fe00e52SEnrico Granata         m_cmd_obj_sp(cmd_obj_sp),
14286f79bb2dSEnrico Granata         m_synchro(synch),
14296f79bb2dSEnrico Granata         m_fetched_help_short(false),
14306f79bb2dSEnrico Granata         m_fetched_help_long(false)
14319fe00e52SEnrico Granata     {
14329fe00e52SEnrico Granata         StreamString stream;
14339fe00e52SEnrico Granata         stream.Printf("For more information run 'help %s'",name.c_str());
14349fe00e52SEnrico Granata         SetHelp(stream.GetData());
1435e87764f2SEnrico Granata         if (ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter())
1436e87764f2SEnrico Granata             GetFlags().Set(scripter->GetFlagsForCommandObject(cmd_obj_sp));
14379fe00e52SEnrico Granata     }
14389fe00e52SEnrico Granata 
14396e3d8e7fSEugene Zelenko     ~CommandObjectScriptingObject() override = default;
14409fe00e52SEnrico Granata 
144113d21e9aSBruce Mitchener     bool
144213d21e9aSBruce Mitchener     IsRemovable () const override
14439fe00e52SEnrico Granata     {
14449fe00e52SEnrico Granata         return true;
14459fe00e52SEnrico Granata     }
14469fe00e52SEnrico Granata 
14470641ca1aSZachary Turner     StructuredData::GenericSP
14489fe00e52SEnrico Granata     GetImplementingObject ()
14499fe00e52SEnrico Granata     {
14509fe00e52SEnrico Granata         return m_cmd_obj_sp;
14519fe00e52SEnrico Granata     }
14529fe00e52SEnrico Granata 
14539fe00e52SEnrico Granata     ScriptedCommandSynchronicity
14549fe00e52SEnrico Granata     GetSynchronicity ()
14559fe00e52SEnrico Granata     {
14569fe00e52SEnrico Granata         return m_synchro;
14579fe00e52SEnrico Granata     }
14589fe00e52SEnrico Granata 
145913d21e9aSBruce Mitchener     const char *
146013d21e9aSBruce Mitchener     GetHelp () override
14616f79bb2dSEnrico Granata     {
14626f79bb2dSEnrico Granata         if (!m_fetched_help_short)
14636f79bb2dSEnrico Granata         {
14646f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14656f79bb2dSEnrico Granata             if (scripter)
14666f79bb2dSEnrico Granata             {
14676f79bb2dSEnrico Granata                 std::string docstring;
14686f79bb2dSEnrico Granata                 m_fetched_help_short = scripter->GetShortHelpForCommandObject(m_cmd_obj_sp,docstring);
14696f79bb2dSEnrico Granata                 if (!docstring.empty())
14706f79bb2dSEnrico Granata                     SetHelp(docstring);
14716f79bb2dSEnrico Granata             }
14726f79bb2dSEnrico Granata         }
14736f79bb2dSEnrico Granata         return CommandObjectRaw::GetHelp();
14746f79bb2dSEnrico Granata     }
14756f79bb2dSEnrico Granata 
147613d21e9aSBruce Mitchener     const char *
147713d21e9aSBruce Mitchener     GetHelpLong () override
14789fe00e52SEnrico Granata     {
14796f79bb2dSEnrico Granata         if (!m_fetched_help_long)
14806f79bb2dSEnrico Granata         {
14816f79bb2dSEnrico Granata             ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14826f79bb2dSEnrico Granata             if (scripter)
14836f79bb2dSEnrico Granata             {
14846f79bb2dSEnrico Granata                 std::string docstring;
14856f79bb2dSEnrico Granata                 m_fetched_help_long = scripter->GetLongHelpForCommandObject(m_cmd_obj_sp,docstring);
14866f79bb2dSEnrico Granata                 if (!docstring.empty())
14876f79bb2dSEnrico Granata                     SetHelpLong(docstring);
14886f79bb2dSEnrico Granata             }
14896f79bb2dSEnrico Granata         }
14909fe00e52SEnrico Granata         return CommandObjectRaw::GetHelpLong();
14919fe00e52SEnrico Granata     }
14929fe00e52SEnrico Granata 
14939fe00e52SEnrico Granata protected:
149413d21e9aSBruce Mitchener     bool
149513d21e9aSBruce Mitchener     DoExecute (const char *raw_command_line, CommandReturnObject &result) override
14969fe00e52SEnrico Granata     {
14979fe00e52SEnrico Granata         ScriptInterpreter* scripter = m_interpreter.GetScriptInterpreter();
14989fe00e52SEnrico Granata 
14999fe00e52SEnrico Granata         Error error;
15009fe00e52SEnrico Granata 
15019fe00e52SEnrico Granata         result.SetStatus(eReturnStatusInvalid);
15029fe00e52SEnrico Granata 
15036e3d8e7fSEugene Zelenko         if (!scripter || !scripter->RunScriptBasedCommand(m_cmd_obj_sp,
15049fe00e52SEnrico Granata                                                           raw_command_line,
15059fe00e52SEnrico Granata                                                           m_synchro,
15069fe00e52SEnrico Granata                                                           result,
15079fe00e52SEnrico Granata                                                           error,
15086e3d8e7fSEugene Zelenko                                                           m_exe_ctx))
15099fe00e52SEnrico Granata         {
15109fe00e52SEnrico Granata             result.AppendError(error.AsCString());
15119fe00e52SEnrico Granata             result.SetStatus(eReturnStatusFailed);
15129fe00e52SEnrico Granata         }
15139fe00e52SEnrico Granata         else
15149fe00e52SEnrico Granata         {
15159fe00e52SEnrico Granata             // Don't change the status if the command already set it...
15169fe00e52SEnrico Granata             if (result.GetStatus() == eReturnStatusInvalid)
15179fe00e52SEnrico Granata             {
15186e3d8e7fSEugene Zelenko                 if (result.GetOutputData() == nullptr || result.GetOutputData()[0] == '\0')
15199fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishNoResult);
15209fe00e52SEnrico Granata                 else
15219fe00e52SEnrico Granata                     result.SetStatus(eReturnStatusSuccessFinishResult);
15229fe00e52SEnrico Granata             }
15239fe00e52SEnrico Granata         }
15249fe00e52SEnrico Granata 
15259fe00e52SEnrico Granata         return result.Succeeded();
15269fe00e52SEnrico Granata     }
15279fe00e52SEnrico Granata 
15286e3d8e7fSEugene Zelenko private:
15296e3d8e7fSEugene Zelenko     StructuredData::GenericSP m_cmd_obj_sp;
15306e3d8e7fSEugene Zelenko     ScriptedCommandSynchronicity m_synchro;
15316e3d8e7fSEugene Zelenko     bool m_fetched_help_short: 1;
15326e3d8e7fSEugene Zelenko     bool m_fetched_help_long: 1;
15339fe00e52SEnrico Granata };
15349fe00e52SEnrico Granata 
1535a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1536a9dbf432SEnrico Granata // CommandObjectCommandsScriptImport
1537a9dbf432SEnrico Granata //-------------------------------------------------------------------------
1538a9dbf432SEnrico Granata 
15395a988416SJim Ingham class CommandObjectCommandsScriptImport : public CommandObjectParsed
1540a9dbf432SEnrico Granata {
15415a988416SJim Ingham public:
15425a988416SJim Ingham     CommandObjectCommandsScriptImport (CommandInterpreter &interpreter) :
15435a988416SJim Ingham         CommandObjectParsed(interpreter,
15445a988416SJim Ingham                             "command script import",
15455a988416SJim Ingham                             "Import a scripting module in LLDB.",
15466e3d8e7fSEugene Zelenko                             nullptr),
15475a988416SJim Ingham         m_options(interpreter)
15485a988416SJim Ingham     {
15495a988416SJim Ingham         CommandArgumentEntry arg1;
15505a988416SJim Ingham         CommandArgumentData cmd_arg;
15515a988416SJim Ingham 
15525a988416SJim Ingham         // Define the first (and only) variant of this arg.
15535a988416SJim Ingham         cmd_arg.arg_type = eArgTypeFilename;
15543b00e35bSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlus;
15555a988416SJim Ingham 
15565a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
15575a988416SJim Ingham         arg1.push_back (cmd_arg);
15585a988416SJim Ingham 
15595a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
15605a988416SJim Ingham         m_arguments.push_back (arg1);
15615a988416SJim Ingham     }
15625a988416SJim Ingham 
15636e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptImport() override = default;
15645a988416SJim Ingham 
156513d21e9aSBruce Mitchener     int
15665a988416SJim Ingham     HandleArgumentCompletion (Args &input,
15675a988416SJim Ingham                               int &cursor_index,
15685a988416SJim Ingham                               int &cursor_char_position,
15695a988416SJim Ingham                               OptionElementVector &opt_element_vector,
15705a988416SJim Ingham                               int match_start_point,
15715a988416SJim Ingham                               int max_return_elements,
15725a988416SJim Ingham                               bool &word_complete,
157313d21e9aSBruce Mitchener                               StringList &matches) override
15745a988416SJim Ingham     {
15755a988416SJim Ingham         std::string completion_str (input.GetArgumentAtIndex(cursor_index));
15765a988416SJim Ingham         completion_str.erase (cursor_char_position);
15775a988416SJim Ingham 
15785a988416SJim Ingham         CommandCompletions::InvokeCommonCompletionCallbacks(m_interpreter,
15795a988416SJim Ingham                                                             CommandCompletions::eDiskFileCompletion,
15805a988416SJim Ingham                                                             completion_str.c_str(),
15815a988416SJim Ingham                                                             match_start_point,
15825a988416SJim Ingham                                                             max_return_elements,
15836e3d8e7fSEugene Zelenko                                                             nullptr,
15845a988416SJim Ingham                                                             word_complete,
15855a988416SJim Ingham                                                             matches);
15865a988416SJim Ingham         return matches.GetSize();
15875a988416SJim Ingham     }
15885a988416SJim Ingham 
158913d21e9aSBruce Mitchener     Options *
159013d21e9aSBruce Mitchener     GetOptions () override
15915a988416SJim Ingham     {
15925a988416SJim Ingham         return &m_options;
15935a988416SJim Ingham     }
15945a988416SJim Ingham 
15955a988416SJim Ingham protected:
15960a305db7SEnrico Granata     class CommandOptions : public Options
15970a305db7SEnrico Granata     {
15980a305db7SEnrico Granata     public:
15990a305db7SEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
16000a305db7SEnrico Granata             Options (interpreter)
16010a305db7SEnrico Granata         {
16020a305db7SEnrico Granata         }
16030a305db7SEnrico Granata 
16046e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
16050a305db7SEnrico Granata 
160613d21e9aSBruce Mitchener         Error
160713d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
16080a305db7SEnrico Granata         {
16090a305db7SEnrico Granata             Error error;
16103bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
16110a305db7SEnrico Granata 
16120a305db7SEnrico Granata             switch (short_option)
16130a305db7SEnrico Granata             {
16140a305db7SEnrico Granata                 case 'r':
16150a305db7SEnrico Granata                     m_allow_reload = true;
16160a305db7SEnrico Granata                     break;
16170a305db7SEnrico Granata                 default:
16180a305db7SEnrico Granata                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
16190a305db7SEnrico Granata                     break;
16200a305db7SEnrico Granata             }
16210a305db7SEnrico Granata 
16220a305db7SEnrico Granata             return error;
16230a305db7SEnrico Granata         }
16240a305db7SEnrico Granata 
16250a305db7SEnrico Granata         void
162613d21e9aSBruce Mitchener         OptionParsingStarting () override
16270a305db7SEnrico Granata         {
1628e0c70f1bSEnrico Granata             m_allow_reload = true;
16290a305db7SEnrico Granata         }
16300a305db7SEnrico Granata 
16310a305db7SEnrico Granata         const OptionDefinition*
163213d21e9aSBruce Mitchener         GetDefinitions () override
16330a305db7SEnrico Granata         {
16340a305db7SEnrico Granata             return g_option_table;
16350a305db7SEnrico Granata         }
16360a305db7SEnrico Granata 
16370a305db7SEnrico Granata         // Options table: Required for subclasses of Options.
16380a305db7SEnrico Granata 
16390a305db7SEnrico Granata         static OptionDefinition g_option_table[];
16400a305db7SEnrico Granata 
16410a305db7SEnrico Granata         // Instance variables to hold the values for command options.
16420a305db7SEnrico Granata 
16430a305db7SEnrico Granata         bool m_allow_reload;
16440a305db7SEnrico Granata     };
16450a305db7SEnrico Granata 
1646a9dbf432SEnrico Granata     bool
164713d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1648a9dbf432SEnrico Granata     {
1649a9dbf432SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
1650a9dbf432SEnrico Granata         {
1651a9dbf432SEnrico Granata             result.AppendError ("only scripting language supported for module importing is currently Python");
1652a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1653a9dbf432SEnrico Granata             return false;
1654a9dbf432SEnrico Granata         }
1655a9dbf432SEnrico Granata 
16565a988416SJim Ingham         size_t argc = command.GetArgumentCount();
16573b00e35bSEnrico Granata         if (0 == argc)
1658a9dbf432SEnrico Granata         {
16593b00e35bSEnrico Granata             result.AppendError("command script import needs one or more arguments");
1660a9dbf432SEnrico Granata             result.SetStatus (eReturnStatusFailed);
1661a9dbf432SEnrico Granata             return false;
1662a9dbf432SEnrico Granata         }
1663a9dbf432SEnrico Granata 
16640e978481SEd Maste         for (size_t i = 0;
16653b00e35bSEnrico Granata              i < argc;
16663b00e35bSEnrico Granata              i++)
16673b00e35bSEnrico Granata         {
16683b00e35bSEnrico Granata             std::string path = command.GetArgumentAtIndex(i);
1669a9dbf432SEnrico Granata             Error error;
1670a9dbf432SEnrico Granata 
1671c9d645d3SGreg Clayton             const bool init_session = true;
1672078551c7SEnrico Granata             // FIXME: this is necessary because CommandObject::CheckRequirements() assumes that
1673078551c7SEnrico Granata             // commands won't ever be recursively invoked, but it's actually possible to craft
1674078551c7SEnrico Granata             // a Python script that does other "command script imports" in __lldb_init_module
1675078551c7SEnrico Granata             // the real fix is to have recursive commands possible with a CommandInvocation object
1676078551c7SEnrico Granata             // separate from the CommandObject itself, so that recursive command invocations
1677078551c7SEnrico Granata             // won't stomp on each other (wrt to execution contents, options, and more)
1678078551c7SEnrico Granata             m_exe_ctx.Clear();
1679a9dbf432SEnrico Granata             if (m_interpreter.GetScriptInterpreter()->LoadScriptingModule(path.c_str(),
16800a305db7SEnrico Granata                                                                           m_options.m_allow_reload,
1681c9d645d3SGreg Clayton                                                                           init_session,
1682a9dbf432SEnrico Granata                                                                           error))
1683a9dbf432SEnrico Granata             {
1684a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
1685a9dbf432SEnrico Granata             }
1686a9dbf432SEnrico Granata             else
1687a9dbf432SEnrico Granata             {
1688a9dbf432SEnrico Granata                 result.AppendErrorWithFormat("module importing failed: %s", error.AsCString());
1689a9dbf432SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
1690a9dbf432SEnrico Granata             }
16913b00e35bSEnrico Granata         }
1692a9dbf432SEnrico Granata 
1693a9dbf432SEnrico Granata         return result.Succeeded();
1694a9dbf432SEnrico Granata     }
16950a305db7SEnrico Granata 
16965a988416SJim Ingham     CommandOptions m_options;
1697a9dbf432SEnrico Granata };
1698223383edSEnrico Granata 
16990a305db7SEnrico Granata OptionDefinition
17000a305db7SEnrico Granata CommandObjectCommandsScriptImport::CommandOptions::g_option_table[] =
17010a305db7SEnrico Granata {
17026e3d8e7fSEugene Zelenko     { 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."},
17036e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
17040a305db7SEnrico Granata };
17050a305db7SEnrico Granata 
1706223383edSEnrico Granata //-------------------------------------------------------------------------
1707223383edSEnrico Granata // CommandObjectCommandsScriptAdd
1708223383edSEnrico Granata //-------------------------------------------------------------------------
1709223383edSEnrico Granata 
171044d93782SGreg Clayton class CommandObjectCommandsScriptAdd :
171144d93782SGreg Clayton     public CommandObjectParsed,
171244d93782SGreg Clayton     public IOHandlerDelegateMultiline
1713223383edSEnrico Granata {
17145a988416SJim Ingham public:
17155a988416SJim Ingham     CommandObjectCommandsScriptAdd(CommandInterpreter &interpreter) :
17165a988416SJim Ingham         CommandObjectParsed(interpreter,
17175a988416SJim Ingham                             "command script add",
17185a988416SJim Ingham                             "Add a scripted function as an LLDB command.",
17196e3d8e7fSEugene Zelenko                             nullptr),
1720c3d874a5SGreg Clayton         IOHandlerDelegateMultiline ("DONE"),
17215a988416SJim Ingham         m_options (interpreter)
17225a988416SJim Ingham     {
17235a988416SJim Ingham         CommandArgumentEntry arg1;
17245a988416SJim Ingham         CommandArgumentData cmd_arg;
17255a988416SJim Ingham 
17265a988416SJim Ingham         // Define the first (and only) variant of this arg.
17275a988416SJim Ingham         cmd_arg.arg_type = eArgTypeCommandName;
17285a988416SJim Ingham         cmd_arg.arg_repetition = eArgRepeatPlain;
17295a988416SJim Ingham 
17305a988416SJim Ingham         // There is only one variant this argument could be; put it into the argument entry.
17315a988416SJim Ingham         arg1.push_back (cmd_arg);
17325a988416SJim Ingham 
17335a988416SJim Ingham         // Push the data for the first argument into the m_arguments vector.
17345a988416SJim Ingham         m_arguments.push_back (arg1);
17355a988416SJim Ingham     }
17365a988416SJim Ingham 
17376e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptAdd() override = default;
17385a988416SJim Ingham 
173913d21e9aSBruce Mitchener     Options *
174013d21e9aSBruce Mitchener     GetOptions () override
17415a988416SJim Ingham     {
17425a988416SJim Ingham         return &m_options;
17435a988416SJim Ingham     }
17445a988416SJim Ingham 
17455a988416SJim Ingham protected:
1746223383edSEnrico Granata     class CommandOptions : public Options
1747223383edSEnrico Granata     {
1748223383edSEnrico Granata     public:
1749223383edSEnrico Granata         CommandOptions (CommandInterpreter &interpreter) :
17509fe00e52SEnrico Granata             Options (interpreter),
17519fe00e52SEnrico Granata             m_class_name(),
17529fe00e52SEnrico Granata             m_funct_name(),
17539fe00e52SEnrico Granata             m_short_help(),
17549fe00e52SEnrico Granata             m_synchronicity(eScriptedCommandSynchronicitySynchronous)
1755223383edSEnrico Granata         {
1756223383edSEnrico Granata         }
1757223383edSEnrico Granata 
17586e3d8e7fSEugene Zelenko         ~CommandOptions() override = default;
1759223383edSEnrico Granata 
176013d21e9aSBruce Mitchener         Error
176113d21e9aSBruce Mitchener         SetOptionValue (uint32_t option_idx, const char *option_arg) override
1762223383edSEnrico Granata         {
1763223383edSEnrico Granata             Error error;
17643bcdfc0eSGreg Clayton             const int short_option = m_getopt_table[option_idx].val;
1765223383edSEnrico Granata 
1766223383edSEnrico Granata             switch (short_option)
1767223383edSEnrico Granata             {
1768223383edSEnrico Granata                 case 'f':
1769735152e3SEnrico Granata                     if (option_arg)
1770735152e3SEnrico Granata                         m_funct_name.assign(option_arg);
1771735152e3SEnrico Granata                     break;
17729fe00e52SEnrico Granata                 case 'c':
17739fe00e52SEnrico Granata                     if (option_arg)
17749fe00e52SEnrico Granata                         m_class_name.assign(option_arg);
17759fe00e52SEnrico Granata                     break;
1776735152e3SEnrico Granata                 case 'h':
1777735152e3SEnrico Granata                     if (option_arg)
1778735152e3SEnrico Granata                         m_short_help.assign(option_arg);
1779223383edSEnrico Granata                     break;
17800a305db7SEnrico Granata                 case 's':
178144d93782SGreg Clayton                     m_synchronicity = (ScriptedCommandSynchronicity) Args::StringToOptionEnum(option_arg, g_option_table[option_idx].enum_values, 0, error);
17820a305db7SEnrico Granata                     if (!error.Success())
17830a305db7SEnrico Granata                         error.SetErrorStringWithFormat ("unrecognized value for synchronicity '%s'", option_arg);
17840a305db7SEnrico Granata                     break;
1785223383edSEnrico Granata                 default:
178686edbf41SGreg Clayton                     error.SetErrorStringWithFormat ("unrecognized option '%c'", short_option);
1787223383edSEnrico Granata                     break;
1788223383edSEnrico Granata             }
1789223383edSEnrico Granata 
1790223383edSEnrico Granata             return error;
1791223383edSEnrico Granata         }
1792223383edSEnrico Granata 
1793223383edSEnrico Granata         void
179413d21e9aSBruce Mitchener         OptionParsingStarting () override
1795223383edSEnrico Granata         {
17969fe00e52SEnrico Granata             m_class_name.clear();
1797735152e3SEnrico Granata             m_funct_name.clear();
1798735152e3SEnrico Granata             m_short_help.clear();
179944d93782SGreg Clayton             m_synchronicity = eScriptedCommandSynchronicitySynchronous;
1800223383edSEnrico Granata         }
1801223383edSEnrico Granata 
1802223383edSEnrico Granata         const OptionDefinition*
180313d21e9aSBruce Mitchener         GetDefinitions () override
1804223383edSEnrico Granata         {
1805223383edSEnrico Granata             return g_option_table;
1806223383edSEnrico Granata         }
1807223383edSEnrico Granata 
1808223383edSEnrico Granata         // Options table: Required for subclasses of Options.
1809223383edSEnrico Granata 
1810223383edSEnrico Granata         static OptionDefinition g_option_table[];
1811223383edSEnrico Granata 
1812223383edSEnrico Granata         // Instance variables to hold the values for command options.
1813223383edSEnrico Granata 
18149fe00e52SEnrico Granata         std::string m_class_name;
1815223383edSEnrico Granata         std::string m_funct_name;
1816735152e3SEnrico Granata         std::string m_short_help;
181744d93782SGreg Clayton         ScriptedCommandSynchronicity m_synchronicity;
1818223383edSEnrico Granata     };
1819223383edSEnrico Granata 
182013d21e9aSBruce Mitchener     void
182113d21e9aSBruce Mitchener     IOHandlerActivated (IOHandler &io_handler) override
1822223383edSEnrico Granata     {
182344d93782SGreg Clayton         StreamFileSP output_sp(io_handler.GetOutputStreamFile());
182444d93782SGreg Clayton         if (output_sp)
1825223383edSEnrico Granata         {
182644d93782SGreg Clayton             output_sp->PutCString(g_python_command_instructions);
182744d93782SGreg Clayton             output_sp->Flush();
1828223383edSEnrico Granata         }
1829223383edSEnrico Granata     }
1830223383edSEnrico Granata 
1831223383edSEnrico Granata 
183213d21e9aSBruce Mitchener     void
183313d21e9aSBruce Mitchener     IOHandlerInputComplete (IOHandler &io_handler, std::string &data) override
1834223383edSEnrico Granata     {
183544d93782SGreg Clayton         StreamFileSP error_sp = io_handler.GetErrorStreamFile();
183644d93782SGreg Clayton 
183744d93782SGreg Clayton         ScriptInterpreter *interpreter = m_interpreter.GetScriptInterpreter();
183844d93782SGreg Clayton         if (interpreter)
183944d93782SGreg Clayton         {
184044d93782SGreg Clayton 
184144d93782SGreg Clayton             StringList lines;
184244d93782SGreg Clayton             lines.SplitIntoLines(data);
184344d93782SGreg Clayton             if (lines.GetSize() > 0)
184444d93782SGreg Clayton             {
1845a73b7df7SEnrico Granata                 std::string funct_name_str;
184644d93782SGreg Clayton                 if (interpreter->GenerateScriptAliasFunction (lines, funct_name_str))
1847223383edSEnrico Granata                 {
1848a73b7df7SEnrico Granata                     if (funct_name_str.empty())
1849223383edSEnrico Granata                     {
185044d93782SGreg Clayton                         error_sp->Printf ("error: unable to obtain a function name, didn't add python command.\n");
185144d93782SGreg Clayton                         error_sp->Flush();
1852223383edSEnrico Granata                     }
185344d93782SGreg Clayton                     else
185444d93782SGreg Clayton                     {
1855223383edSEnrico Granata                         // everything should be fine now, let's add this alias
1856223383edSEnrico Granata 
1857223383edSEnrico Granata                         CommandObjectSP command_obj_sp(new CommandObjectPythonFunction (m_interpreter,
1858223383edSEnrico Granata                                                                                         m_cmd_name,
1859a73b7df7SEnrico Granata                                                                                         funct_name_str.c_str(),
1860735152e3SEnrico Granata                                                                                         m_short_help,
186144d93782SGreg Clayton                                                                                         m_synchronicity));
1862223383edSEnrico Granata 
18630a305db7SEnrico Granata                         if (!m_interpreter.AddUserCommand(m_cmd_name, command_obj_sp, true))
1864223383edSEnrico Granata                         {
186544d93782SGreg Clayton                             error_sp->Printf ("error: unable to add selected command, didn't add python command.\n");
186644d93782SGreg Clayton                             error_sp->Flush();
1867223383edSEnrico Granata                         }
1868223383edSEnrico Granata                     }
186944d93782SGreg Clayton                 }
187044d93782SGreg Clayton                 else
187144d93782SGreg Clayton                 {
187244d93782SGreg Clayton                     error_sp->Printf ("error: unable to create function, didn't add python command.\n");
187344d93782SGreg Clayton                     error_sp->Flush();
187444d93782SGreg Clayton                 }
187544d93782SGreg Clayton             }
187644d93782SGreg Clayton             else
187744d93782SGreg Clayton             {
187844d93782SGreg Clayton                 error_sp->Printf ("error: empty function, didn't add python command.\n");
187944d93782SGreg Clayton                 error_sp->Flush();
188044d93782SGreg Clayton             }
188144d93782SGreg Clayton         }
188244d93782SGreg Clayton         else
188344d93782SGreg Clayton         {
188444d93782SGreg Clayton             error_sp->Printf ("error: script interpreter missing, didn't add python command.\n");
188544d93782SGreg Clayton             error_sp->Flush();
188644d93782SGreg Clayton         }
188744d93782SGreg Clayton 
188844d93782SGreg Clayton         io_handler.SetIsDone(true);
188944d93782SGreg Clayton     }
1890223383edSEnrico Granata 
18915a988416SJim Ingham protected:
1892223383edSEnrico Granata     bool
189313d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
1894223383edSEnrico Granata     {
189599f0b8f9SEnrico Granata         if (m_interpreter.GetDebugger().GetScriptLanguage() != lldb::eScriptLanguagePython)
189699f0b8f9SEnrico Granata         {
189799f0b8f9SEnrico Granata             result.AppendError ("only scripting language supported for scripted commands is currently Python");
189899f0b8f9SEnrico Granata             result.SetStatus (eReturnStatusFailed);
189999f0b8f9SEnrico Granata             return false;
190099f0b8f9SEnrico Granata         }
190199f0b8f9SEnrico Granata 
19025a988416SJim Ingham         size_t argc = command.GetArgumentCount();
1903223383edSEnrico Granata 
1904223383edSEnrico Granata         if (argc != 1)
1905223383edSEnrico Granata         {
1906223383edSEnrico Granata             result.AppendError ("'command script add' requires one argument");
1907223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
1908223383edSEnrico Granata             return false;
1909223383edSEnrico Granata         }
1910223383edSEnrico Granata 
1911735152e3SEnrico Granata         // Store the options in case we get multi-line input
191244d93782SGreg Clayton         m_cmd_name = command.GetArgumentAtIndex(0);
1913735152e3SEnrico Granata         m_short_help.assign(m_options.m_short_help);
191444d93782SGreg Clayton         m_synchronicity = m_options.m_synchronicity;
1915223383edSEnrico Granata 
19169fe00e52SEnrico Granata         if (m_options.m_class_name.empty())
19179fe00e52SEnrico Granata         {
1918223383edSEnrico Granata             if (m_options.m_funct_name.empty())
1919223383edSEnrico Granata             {
192044d93782SGreg Clayton                 m_interpreter.GetPythonCommandsFromIOHandler("     ",  // Prompt
192144d93782SGreg Clayton                                                              *this,    // IOHandlerDelegate
192244d93782SGreg Clayton                                                              true,     // Run IOHandler in async mode
19236e3d8e7fSEugene Zelenko                                                              nullptr); // Baton for the "io_handler" that will be passed back into our IOHandlerDelegate functions
1924223383edSEnrico Granata             }
1925223383edSEnrico Granata             else
1926223383edSEnrico Granata             {
19270a305db7SEnrico Granata                 CommandObjectSP new_cmd(new CommandObjectPythonFunction(m_interpreter,
192844d93782SGreg Clayton                                                                         m_cmd_name,
19290a305db7SEnrico Granata                                                                         m_options.m_funct_name,
1930735152e3SEnrico Granata                                                                         m_options.m_short_help,
193144d93782SGreg Clayton                                                                         m_synchronicity));
193244d93782SGreg Clayton                 if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
1933223383edSEnrico Granata                 {
1934223383edSEnrico Granata                     result.SetStatus (eReturnStatusSuccessFinishNoResult);
1935223383edSEnrico Granata                 }
1936223383edSEnrico Granata                 else
1937223383edSEnrico Granata                 {
1938223383edSEnrico Granata                     result.AppendError("cannot add command");
1939223383edSEnrico Granata                     result.SetStatus (eReturnStatusFailed);
1940223383edSEnrico Granata                 }
1941223383edSEnrico Granata             }
19429fe00e52SEnrico Granata         }
19439fe00e52SEnrico Granata         else
19449fe00e52SEnrico Granata         {
19459fe00e52SEnrico Granata             ScriptInterpreter *interpreter = GetCommandInterpreter().GetScriptInterpreter();
19469fe00e52SEnrico Granata             if (!interpreter)
19479fe00e52SEnrico Granata             {
19489fe00e52SEnrico Granata                 result.AppendError("cannot find ScriptInterpreter");
19499fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19509fe00e52SEnrico Granata                 return false;
19519fe00e52SEnrico Granata             }
19529fe00e52SEnrico Granata 
19539fe00e52SEnrico Granata             auto cmd_obj_sp = interpreter->CreateScriptCommandObject(m_options.m_class_name.c_str());
19549fe00e52SEnrico Granata             if (!cmd_obj_sp)
19559fe00e52SEnrico Granata             {
19569fe00e52SEnrico Granata                 result.AppendError("cannot create helper object");
19579fe00e52SEnrico Granata                 result.SetStatus(eReturnStatusFailed);
19589fe00e52SEnrico Granata                 return false;
19599fe00e52SEnrico Granata             }
19609fe00e52SEnrico Granata 
19619fe00e52SEnrico Granata             CommandObjectSP new_cmd(new CommandObjectScriptingObject(m_interpreter,
19629fe00e52SEnrico Granata                                                                      m_cmd_name,
19639fe00e52SEnrico Granata                                                                      cmd_obj_sp,
19649fe00e52SEnrico Granata                                                                      m_synchronicity));
19659fe00e52SEnrico Granata             if (m_interpreter.AddUserCommand(m_cmd_name, new_cmd, true))
19669fe00e52SEnrico Granata             {
19679fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusSuccessFinishNoResult);
19689fe00e52SEnrico Granata             }
19699fe00e52SEnrico Granata             else
19709fe00e52SEnrico Granata             {
19719fe00e52SEnrico Granata                 result.AppendError("cannot add command");
19729fe00e52SEnrico Granata                 result.SetStatus (eReturnStatusFailed);
19739fe00e52SEnrico Granata             }
19749fe00e52SEnrico Granata         }
1975223383edSEnrico Granata 
1976223383edSEnrico Granata         return result.Succeeded();
1977223383edSEnrico Granata     }
19785a988416SJim Ingham 
19795a988416SJim Ingham     CommandOptions m_options;
198044d93782SGreg Clayton     std::string m_cmd_name;
1981735152e3SEnrico Granata     std::string m_short_help;
198244d93782SGreg Clayton     ScriptedCommandSynchronicity m_synchronicity;
1983223383edSEnrico Granata };
1984223383edSEnrico Granata 
19850a305db7SEnrico Granata static OptionEnumValueElement g_script_synchro_type[] =
19860a305db7SEnrico Granata {
19870a305db7SEnrico Granata     { eScriptedCommandSynchronicitySynchronous,      "synchronous",       "Run synchronous"},
19880a305db7SEnrico Granata     { eScriptedCommandSynchronicityAsynchronous,     "asynchronous",      "Run asynchronous"},
19890a305db7SEnrico Granata     { eScriptedCommandSynchronicityCurrentValue,     "current",           "Do not alter current setting"},
19906e3d8e7fSEugene Zelenko     { 0, nullptr, nullptr }
19910a305db7SEnrico Granata };
19920a305db7SEnrico Granata 
1993223383edSEnrico Granata OptionDefinition
1994223383edSEnrico Granata CommandObjectCommandsScriptAdd::CommandOptions::g_option_table[] =
1995223383edSEnrico Granata {
19966e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "function", 'f', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonFunction,        "Name of the Python function to bind to this command name."},
19976e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_2, false, "class", 'c', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypePythonClass,        "Name of the Python class to bind to this command name."},
19986e3d8e7fSEugene Zelenko     { LLDB_OPT_SET_1, false, "help"  , 'h', OptionParser::eRequiredArgument, nullptr, nullptr, 0, eArgTypeHelpText, "The help text to display for this command."},
19996e3d8e7fSEugene Zelenko     { 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."},
20006e3d8e7fSEugene Zelenko     { 0, false, nullptr, 0, 0, nullptr, nullptr, 0, eArgTypeNone, nullptr }
2001223383edSEnrico Granata };
2002223383edSEnrico Granata 
2003223383edSEnrico Granata //-------------------------------------------------------------------------
2004223383edSEnrico Granata // CommandObjectCommandsScriptList
2005223383edSEnrico Granata //-------------------------------------------------------------------------
2006223383edSEnrico Granata 
20075a988416SJim Ingham class CommandObjectCommandsScriptList : public CommandObjectParsed
2008223383edSEnrico Granata {
2009223383edSEnrico Granata public:
2010223383edSEnrico Granata     CommandObjectCommandsScriptList(CommandInterpreter &interpreter) :
20115a988416SJim Ingham         CommandObjectParsed(interpreter,
2012223383edSEnrico Granata                             "command script list",
2013223383edSEnrico Granata                             "List defined scripted commands.",
20146e3d8e7fSEugene Zelenko                             nullptr)
2015223383edSEnrico Granata     {
2016223383edSEnrico Granata     }
2017223383edSEnrico Granata 
20186e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptList() override = default;
2019223383edSEnrico Granata 
2020223383edSEnrico Granata     bool
202113d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2022223383edSEnrico Granata     {
2023223383edSEnrico Granata         m_interpreter.GetHelp(result,
2024223383edSEnrico Granata                               CommandInterpreter::eCommandTypesUserDef);
2025223383edSEnrico Granata 
2026223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2027223383edSEnrico Granata 
2028223383edSEnrico Granata         return true;
2029223383edSEnrico Granata     }
2030223383edSEnrico Granata };
2031223383edSEnrico Granata 
2032223383edSEnrico Granata //-------------------------------------------------------------------------
2033223383edSEnrico Granata // CommandObjectCommandsScriptClear
2034223383edSEnrico Granata //-------------------------------------------------------------------------
2035223383edSEnrico Granata 
20365a988416SJim Ingham class CommandObjectCommandsScriptClear : public CommandObjectParsed
2037223383edSEnrico Granata {
2038223383edSEnrico Granata public:
2039223383edSEnrico Granata     CommandObjectCommandsScriptClear(CommandInterpreter &interpreter) :
20405a988416SJim Ingham         CommandObjectParsed(interpreter,
2041223383edSEnrico Granata                             "command script clear",
2042223383edSEnrico Granata                             "Delete all scripted commands.",
20436e3d8e7fSEugene Zelenko                             nullptr)
2044223383edSEnrico Granata     {
2045223383edSEnrico Granata     }
2046223383edSEnrico Granata 
20476e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptClear() override = default;
2048223383edSEnrico Granata 
20495a988416SJim Ingham protected:
2050223383edSEnrico Granata     bool
205113d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2052223383edSEnrico Granata     {
2053223383edSEnrico Granata         m_interpreter.RemoveAllUser();
2054223383edSEnrico Granata 
2055223383edSEnrico Granata         result.SetStatus (eReturnStatusSuccessFinishResult);
2056223383edSEnrico Granata 
2057223383edSEnrico Granata         return true;
2058223383edSEnrico Granata     }
2059223383edSEnrico Granata };
2060223383edSEnrico Granata 
2061223383edSEnrico Granata //-------------------------------------------------------------------------
2062223383edSEnrico Granata // CommandObjectCommandsScriptDelete
2063223383edSEnrico Granata //-------------------------------------------------------------------------
2064223383edSEnrico Granata 
20655a988416SJim Ingham class CommandObjectCommandsScriptDelete : public CommandObjectParsed
2066223383edSEnrico Granata {
2067223383edSEnrico Granata public:
2068223383edSEnrico Granata     CommandObjectCommandsScriptDelete(CommandInterpreter &interpreter) :
20695a988416SJim Ingham         CommandObjectParsed(interpreter,
2070223383edSEnrico Granata                             "command script delete",
2071223383edSEnrico Granata                             "Delete a scripted command.",
20726e3d8e7fSEugene Zelenko                             nullptr)
2073223383edSEnrico Granata     {
2074223383edSEnrico Granata         CommandArgumentEntry arg1;
2075223383edSEnrico Granata         CommandArgumentData cmd_arg;
2076223383edSEnrico Granata 
2077223383edSEnrico Granata         // Define the first (and only) variant of this arg.
2078223383edSEnrico Granata         cmd_arg.arg_type = eArgTypeCommandName;
2079223383edSEnrico Granata         cmd_arg.arg_repetition = eArgRepeatPlain;
2080223383edSEnrico Granata 
2081223383edSEnrico Granata         // There is only one variant this argument could be; put it into the argument entry.
2082223383edSEnrico Granata         arg1.push_back (cmd_arg);
2083223383edSEnrico Granata 
2084223383edSEnrico Granata         // Push the data for the first argument into the m_arguments vector.
2085223383edSEnrico Granata         m_arguments.push_back (arg1);
2086223383edSEnrico Granata     }
2087223383edSEnrico Granata 
20886e3d8e7fSEugene Zelenko     ~CommandObjectCommandsScriptDelete() override = default;
2089223383edSEnrico Granata 
20905a988416SJim Ingham protected:
2091223383edSEnrico Granata     bool
209213d21e9aSBruce Mitchener     DoExecute (Args& command, CommandReturnObject &result) override
2093223383edSEnrico Granata     {
2094223383edSEnrico Granata 
20955a988416SJim Ingham         size_t argc = command.GetArgumentCount();
2096223383edSEnrico Granata 
2097223383edSEnrico Granata         if (argc != 1)
2098223383edSEnrico Granata         {
2099223383edSEnrico Granata             result.AppendError ("'command script delete' requires one argument");
2100223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2101223383edSEnrico Granata             return false;
2102223383edSEnrico Granata         }
2103223383edSEnrico Granata 
21045a988416SJim Ingham         const char* cmd_name = command.GetArgumentAtIndex(0);
2105223383edSEnrico Granata 
2106223383edSEnrico Granata         if (cmd_name && *cmd_name && m_interpreter.HasUserCommands() && m_interpreter.UserCommandExists(cmd_name))
2107223383edSEnrico Granata         {
2108223383edSEnrico Granata             m_interpreter.RemoveUser(cmd_name);
2109223383edSEnrico Granata             result.SetStatus (eReturnStatusSuccessFinishResult);
2110223383edSEnrico Granata         }
2111223383edSEnrico Granata         else
2112223383edSEnrico Granata         {
2113223383edSEnrico Granata             result.AppendErrorWithFormat ("command %s not found", cmd_name);
2114223383edSEnrico Granata             result.SetStatus (eReturnStatusFailed);
2115223383edSEnrico Granata         }
2116223383edSEnrico Granata 
2117223383edSEnrico Granata         return result.Succeeded();
2118223383edSEnrico Granata     }
2119223383edSEnrico Granata };
2120223383edSEnrico Granata 
2121223383edSEnrico Granata #pragma mark CommandObjectMultiwordCommandsScript
2122223383edSEnrico Granata 
2123223383edSEnrico Granata //-------------------------------------------------------------------------
2124223383edSEnrico Granata // CommandObjectMultiwordCommandsScript
2125223383edSEnrico Granata //-------------------------------------------------------------------------
2126223383edSEnrico Granata 
2127223383edSEnrico Granata class CommandObjectMultiwordCommandsScript : public CommandObjectMultiword
2128223383edSEnrico Granata {
2129223383edSEnrico Granata public:
2130223383edSEnrico Granata     CommandObjectMultiwordCommandsScript (CommandInterpreter &interpreter) :
2131223383edSEnrico Granata     CommandObjectMultiword (interpreter,
2132223383edSEnrico Granata                             "command script",
2133223383edSEnrico Granata                             "A set of commands for managing or customizing script commands.",
2134223383edSEnrico Granata                             "command script <subcommand> [<subcommand-options>]")
2135223383edSEnrico Granata     {
2136223383edSEnrico Granata         LoadSubCommand ("add",    CommandObjectSP (new CommandObjectCommandsScriptAdd (interpreter)));
2137223383edSEnrico Granata         LoadSubCommand ("delete", CommandObjectSP (new CommandObjectCommandsScriptDelete (interpreter)));
2138223383edSEnrico Granata         LoadSubCommand ("clear",  CommandObjectSP (new CommandObjectCommandsScriptClear (interpreter)));
2139223383edSEnrico Granata         LoadSubCommand ("list",   CommandObjectSP (new CommandObjectCommandsScriptList (interpreter)));
2140a9dbf432SEnrico Granata         LoadSubCommand ("import", CommandObjectSP (new CommandObjectCommandsScriptImport (interpreter)));
2141223383edSEnrico Granata     }
2142223383edSEnrico Granata 
21436e3d8e7fSEugene Zelenko     ~CommandObjectMultiwordCommandsScript() override = default;
2144223383edSEnrico Granata };
2145223383edSEnrico Granata 
2146ebc09c36SJim Ingham #pragma mark CommandObjectMultiwordCommands
2147ebc09c36SJim Ingham 
2148ebc09c36SJim Ingham //-------------------------------------------------------------------------
2149ebc09c36SJim Ingham // CommandObjectMultiwordCommands
2150ebc09c36SJim Ingham //-------------------------------------------------------------------------
2151ebc09c36SJim Ingham 
2152ebc09c36SJim Ingham CommandObjectMultiwordCommands::CommandObjectMultiwordCommands (CommandInterpreter &interpreter) :
2153a7015092SGreg Clayton     CommandObjectMultiword (interpreter,
21540e5e5a79SGreg Clayton                             "command",
21553f4c09c1SCaroline Tice                             "A set of commands for managing or customizing the debugger commands.",
21560e5e5a79SGreg Clayton                             "command <subcommand> [<subcommand-options>]")
2157ebc09c36SJim Ingham {
2158a7015092SGreg Clayton     LoadSubCommand ("source",  CommandObjectSP (new CommandObjectCommandsSource (interpreter)));
2159a7015092SGreg Clayton     LoadSubCommand ("alias",   CommandObjectSP (new CommandObjectCommandsAlias (interpreter)));
2160a7015092SGreg Clayton     LoadSubCommand ("unalias", CommandObjectSP (new CommandObjectCommandsUnalias (interpreter)));
2161b547278cSGreg Clayton     LoadSubCommand ("delete",  CommandObjectSP (new CommandObjectCommandsDelete (interpreter)));
2162de164aaaSGreg Clayton     LoadSubCommand ("regex",   CommandObjectSP (new CommandObjectCommandsAddRegex (interpreter)));
2163a5a97ebeSJim Ingham     LoadSubCommand ("history", CommandObjectSP (new CommandObjectCommandsHistory (interpreter)));
2164223383edSEnrico Granata     LoadSubCommand ("script",  CommandObjectSP (new CommandObjectMultiwordCommandsScript (interpreter)));
2165ebc09c36SJim Ingham }
2166ebc09c36SJim Ingham 
21676e3d8e7fSEugene Zelenko CommandObjectMultiwordCommands::~CommandObjectMultiwordCommands() = default;
2168